b0a7aed0d8e51185d9838ce64ecff16e79bc2242
[dyn-nsupdate.git] / dyn-nsupdate.cpp
1 #include <iostream>
2 #include <fstream>
3 #include <sys/wait.h>
4
5 #include <boost/regex.hpp>
6 #include <boost/property_tree/ptree.hpp>
7 #include <boost/property_tree/ini_parser.hpp>
8 #include <boost/iostreams/device/file_descriptor.hpp>
9 #include <boost/iostreams/stream.hpp>
10
11 using namespace boost;
12 typedef property_tree::ptree::path_type path;
13
14 static void write(int fd, const char *str)
15 {
16         size_t len = strlen(str);
17         ssize_t written = write(fd, str, len);
18         if (written < 0 || (size_t)written != len) {
19                 std::cerr << "Error writing pipe." << std::endl;
20                 exit(1);
21         }
22 }
23
24 int main(int argc, const char ** argv)
25 {
26         static const regex regex_ip("\\d{1,3}.\\d{1,3}.\\d{1,3}.\\d{1,3}");
27         static const regex regex_password("[a-zA-Z0-9.:;,_-]+");
28         static const regex regex_domain("[a-zA-Z0-9.]+");
29         
30         
31         if (argc != 4) {
32                 std::cerr << "Usage: " << argv[0] << " <domain> <password> <IP address>" << std::endl;
33                 return 1;
34         }
35         
36         /* Obtain input */
37         std::string domain = argv[1];
38         std::string password = argv[2];
39         std::string ip = argv[3];
40         
41         /* Validate input */
42         if (!regex_match(ip, regex_ip)) {
43                 std::cerr << "Invalid IP address " << ip << "." << std::endl;
44                 exit(1);
45         }
46         if (!regex_match(domain, regex_domain)) {
47                 std::cerr << "Invalid domain " << domain << "." << std::endl;
48                 exit(1);
49         }
50         if (!regex_match(password, regex_password)) {
51                 std::cerr << "Invalid password " << password << "." << std::endl;
52                 exit(1);
53         }
54         
55         /* read configuration */
56         property_tree::ptree config;
57         property_tree::ini_parser::read_ini(CONFIG_FILE, config);
58         std::string nsupdate = config.get<std::string>("nsupdate");
59         unsigned server_port = config.get<unsigned>("port", 53);
60         
61         /* Given the domain, check whether the password matches */
62         optional<std::string> correct_password = config.get_optional<std::string>(path(domain+"/password", '/'));
63         if (!correct_password || *correct_password != password) {
64                 std::cerr << "Password incorrect." << std::endl;
65                 exit(1);
66         }
67         
68         /* preapre the pipe */
69         int pipe_ends[2];
70         if (pipe(pipe_ends) < 0) {
71                 std::cerr << "Error opening pipe." << std::endl;
72                 exit(1);
73         }
74
75         /* Launch nsupdate */
76         pid_t child_pid = fork();
77         if (child_pid < 0) {
78                 std::cerr << "Error while forking." << std::endl;
79                 exit(1);
80         }
81         if (child_pid == 0) {
82                 /* We're in the child */
83                 /* Close write end, use read end as stdin */
84                 close(pipe_ends[1]);
85                 if (dup2(pipe_ends[0], fileno(stdin)) < 0) {
86                         std::cerr << "There was an error redirecting stdin." << std::endl;
87                         exit(1);
88                 }
89                 /* exec nsupdate */
90                 execl(nsupdate.c_str(), nsupdate.c_str(), "-p", std::to_string(server_port).c_str(), "-l", (char *)NULL);
91                 /* There was an error */
92                 std::cerr << "There was an error executing nsupdate." << std::endl;
93                 exit(1);
94         }
95         
96         /* Send it the command */
97         write(pipe_ends[1], "update delete ");
98         write(pipe_ends[1], domain.c_str());
99         write(pipe_ends[1], ". A\n");
100         
101         write(pipe_ends[1], "update add ");
102         write(pipe_ends[1], domain.c_str());
103         write(pipe_ends[1], ". 60 A ");
104         write(pipe_ends[1], ip.c_str());
105         write(pipe_ends[1], "\n");
106         
107         write(pipe_ends[1], "send\n");
108         
109         /* Close both ends */
110         close(pipe_ends[0]);
111         close(pipe_ends[1]);
112         
113         /* Wait for child to be gone */
114         int child_status;
115         waitpid(child_pid, &child_status, 0);
116         if (child_status != 0) {
117                 std::cerr << "There was an error in the child." << std::endl;
118                 exit(1);
119         }
120         
121         return 0;
122 }