b1bcfd281e23796ef86a0a7697ebca5120e9d044
[dyn-nsupdate.git] / nsupd-wrapper / dyn-nsupdate.cpp
1 /* Copyright (c) 2014, Ralf Jung <post@ralfj.de>
2 * All rights reserved.
3
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions are met:
6
7 * 1. Redistributions of source code must retain the above copyright notice, this
8 *    list of conditions and the following disclaimer. 
9 * 2. Redistributions in binary form must reproduce the above copyright notice,
10 *    this list of conditions and the following disclaimer in the documentation
11 *    and/or other materials provided with the distribution.
12
13 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
14 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
15 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
16 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
17 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
18 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
19 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
20 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
21 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
22 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
23 */
24
25 #include <iostream>
26 #include <fstream>
27 #include <sys/wait.h>
28
29 #include <boost/regex.hpp>
30 #include <boost/program_options.hpp>
31 #include <boost/property_tree/ptree.hpp>
32 #include <boost/property_tree/ini_parser.hpp>
33 #include <boost/iostreams/device/file_descriptor.hpp>
34 #include <boost/iostreams/stream.hpp>
35
36 namespace pt = boost::property_tree;
37 namespace po = boost::program_options;
38 using std::string;
39 using boost::regex;
40 using boost::optional;
41
42 static void write(int fd, const char *str)
43 {
44     size_t len = strlen(str);
45     ssize_t written = write(fd, str, len);
46     if (written < 0 || (size_t)written != len) {
47         std::cerr << "Error writing pipe." << std::endl;
48         exit(1);
49     }
50 }
51
52 int main(int argc, const char ** argv)
53 {
54     try {
55         static const regex regex_ipv4("\\d{1,3}(\\.\\d{1,3}){3}|");
56         static const regex regex_ipv6("[a-fA-F0-9]{1,4}(:[a-fA-F0-9]{1,4}){7}|");
57         static const regex regex_password("[a-zA-Z0-9.:;,_-]+");
58         static const regex regex_domain("[a-zA-Z0-9.]+");
59         
60         // Declare the supported options.
61         po::options_description desc("Allowed options");
62         desc.add_options()
63             ("help", "produce help message")
64             ("domain", po::value<string>()->required(), "the domain to update")
65             ("password", po::value<string>()->required(), "the password for the domain")
66             ("ipv4", po::value<string>(), "the new IPv4 address (empty to delete the A record)")
67             ("ipv6", po::value<string>(), "the new IPv6 address (empty to delete the AAAA record)")
68         ;
69         
70         // parse arguments
71         po::variables_map vm;
72         po::store(po::parse_command_line(argc, argv, desc), vm);
73         po::notify(vm);    
74         if (vm.count("help")) {
75             std::cout << "dyn-nsupdate -- a safe setuid wrapper for nsupdate" << std::endl << std::endl;
76             std::cout << desc << "\n";
77             return 1;
78         }
79         string domain = vm["domain"].as<string>();
80         string password = vm["password"].as<string>();
81         bool haveIPv4 = vm.count("ipv4");
82         string ipv4 = haveIPv4 ? vm["ipv4"].as<string>() : "";
83         bool haveIPv6 = vm.count("ipv6");
84         string ipv6 = haveIPv6 ? vm["ipv6"].as<string>() : "";
85         
86         /* Validate input */
87         if (!regex_match(ipv4, regex_ipv4)) {
88             throw std::runtime_error("Invalid IPv4 address" + ipv4);
89         }
90         if (!regex_match(ipv6, regex_ipv6)) {
91             throw std::runtime_error("Invalid IPv6 address: " + ipv6);
92         }
93         if (!regex_match(domain, regex_domain)) {
94             throw std::runtime_error("Invalid Domain: " + domain);
95         }
96         if (!regex_match(password, regex_password)) {
97            throw std::runtime_error("Invalid Password: " + password);
98         }
99         
100         /* read configuration */
101         pt::ptree config;
102         pt::ini_parser::read_ini(CONFIG_FILE, config);
103         std::string nsupdate = config.get<std::string>("nsupdate");
104         unsigned server_port = config.get<unsigned>("port", 53);
105         
106         /* Given the domain, check whether the password matches */
107         optional<std::string> correct_password = config.get_optional<std::string>(pt::ptree::path_type(domain+"/password", '/'));
108         if (!correct_password || *correct_password != password) {
109             std::cerr << "Password incorrect." << std::endl;
110             exit(1);
111         }
112         
113         /* preapre the pipe */
114         int pipe_ends[2];
115         if (pipe(pipe_ends) < 0) {
116             std::cerr << "Error opening pipe." << std::endl;
117             exit(1);
118         }
119
120         /* Launch nsupdate */
121         pid_t child_pid = fork();
122         if (child_pid < 0) {
123             std::cerr << "Error while forking." << std::endl;
124             exit(1);
125         }
126         if (child_pid == 0) {
127             /* We're in the child */
128             /* Close write end, use read end as stdin */
129             close(pipe_ends[1]);
130             if (dup2(pipe_ends[0], fileno(stdin)) < 0) {
131                 std::cerr << "There was an error redirecting stdin." << std::endl;
132                 exit(1);
133             }
134             /* exec nsupdate */
135             execl(nsupdate.c_str(), nsupdate.c_str(), "-p", std::to_string(server_port).c_str(), "-l", (char *)NULL);
136             /* There was an error */
137             std::cerr << "There was an error executing nsupdate." << std::endl;
138             exit(1);
139         }
140         
141         /* Send it the command */
142         if (haveIPv4) {
143             write(pipe_ends[1], "update delete ");
144             write(pipe_ends[1], domain.c_str());
145             write(pipe_ends[1], ". A\n");
146             
147             if (!ipv4.empty()) {
148                 write(pipe_ends[1], "update add ");
149                 write(pipe_ends[1], domain.c_str());
150                 write(pipe_ends[1], ". 60 A ");
151                 write(pipe_ends[1], ipv4.c_str());
152                 write(pipe_ends[1], "\n");
153             }
154         }
155         
156         if (haveIPv6) {
157             write(pipe_ends[1], "update delete ");
158             write(pipe_ends[1], domain.c_str());
159             write(pipe_ends[1], ". AAAA\n");
160             
161             if (!ipv6.empty()) {
162                 write(pipe_ends[1], "update add ");
163                 write(pipe_ends[1], domain.c_str());
164                 write(pipe_ends[1], ". 60 AAAA ");
165                 write(pipe_ends[1], ipv6.c_str());
166                 write(pipe_ends[1], "\n");
167             }
168         }
169         
170         write(pipe_ends[1], "send\n");
171         
172         /* Close both ends */
173         close(pipe_ends[0]);
174         close(pipe_ends[1]);
175         
176         /* Wait for child to be gone */
177         int child_status;
178         waitpid(child_pid, &child_status, 0);
179         if (child_status != 0) {
180             std::cerr << "There was an error in the child." << std::endl;
181             exit(1);
182         }
183     }
184     catch(std::exception &e) {
185         std::cout << e.what() << "\n";
186         return 1;
187     } 
188     
189     return 0;
190 }