7c39136e37c1321795ccf7718eb59b4538e874ae
[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
13 static void write(int fd, const char *str)
14 {
15         size_t len = strlen(str);
16         ssize_t written = write(fd, str, len);
17         if (written < 0 || (size_t)written != len) {
18                 std::cerr << "Error writing pipe." << std::endl;
19                 exit(1);
20         }
21 }
22
23 int main(int argc, const char ** argv)
24 {
25         static const regex regex_ip("\\d{1,3}.\\d{1,3}.\\d{1,3}.\\d{1,3}");
26         static const regex regex_user("[a-zA-Z]+");
27         
28         
29         if (argc != 5) {
30                 std::cerr << "Usage: " << argv[0] << " <username> <password> <domain> <IP address>" << std::endl;
31                 return 1;
32         }
33         
34         /* Obtain and validate inpit */
35         std::string user = argv[1];
36         std::string password = argv[2];
37         std::string domain = argv[3];
38         std::string ip = argv[4];
39         
40         if (!regex_match(ip, regex_ip)) {
41                 std::cerr << "Invalid IP address " << ip << "." << std::endl;
42                 exit(1);
43         }
44         if (!regex_match(user, regex_user)) {
45                 std::cerr << "Invalid username " << user << "." << std::endl;
46                 exit(1);
47         }
48         
49         /* read configuration */
50         property_tree::ptree config;
51         property_tree::ini_parser::read_ini(CONFIG_FILE, config);
52         std::string nsupdate = config.get<std::string>("nsupdate");
53         std::string keyfile = config.get<std::string>("key");
54         
55         /* Check username, password, domain */
56         optional<std::string> correct_password = config.get_optional<std::string>(user+".password");
57         if (!correct_password || *correct_password != password) {
58                 std::cerr << "Username or password incorrect." << std::endl;
59                 exit(1);
60         }
61         if (config.get<std::string>(user+".domain") != domain) {
62                 std::cerr << "Domain incorrect." << std::endl;
63                 exit(1);
64         }
65         
66         /* preapre the pipe */
67         int pipe_ends[] = {0,0};
68         pipe(pipe_ends);
69
70         /* Launch nsupdate */
71         pid_t child_pid = fork();
72         if (child_pid < 0) {
73                 std::cerr << "Error while forking." << std::endl;
74                 exit(1);
75         }
76         if (child_pid == 0) {
77                 /* We're in the child */
78                 /* Close write end, use read and as stdin */
79                 close(pipe_ends[1]);
80                 if (dup2(pipe_ends[0], fileno(stdin)) < 0) {
81                         std::cerr << "There was an error redirecting stdin." << std::endl;
82                         exit(1);
83                 }
84                 /* exec nsupdate */
85                 execl(nsupdate.c_str(), nsupdate.c_str(), "-k", keyfile.c_str(), (char *)NULL);
86                 /* There was an error */
87                 std::cerr << "There was an error executing nsupdate." << std::endl;
88                 exit(1);
89         }
90         
91         /* Send it the command */
92         write(pipe_ends[1], "server localhost\n");
93         
94         write(pipe_ends[1], "update delete ");
95         write(pipe_ends[1], domain.c_str());
96         write(pipe_ends[1], ".\n");
97         
98         write(pipe_ends[1], "update add ");
99         write(pipe_ends[1], domain.c_str());
100         write(pipe_ends[1], ". 60 A ");
101         write(pipe_ends[1], ip.c_str());
102         write(pipe_ends[1], "\n");
103         
104         write(pipe_ends[1], "send\n");
105         
106         /* Close both ends */
107         close(pipe_ends[0]);
108         close(pipe_ends[1]);
109         
110         /* Wait for child to be gone */
111         int child_status;
112         waitpid(child_pid, &child_status, 0);
113         if (child_status != 0) {
114                 std::cerr << "There was an error in the child." << std::endl;
115                 exit(1);
116         }
117         
118         return 0;
119 }