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 }