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
  54         /* Check username, password, domain */
  55         optional<std::string> correct_password = config.get_optional<std::string>(user+".password");
  56         if (!correct_password || *correct_password != password) {
  57                 std::cerr << "Username or password incorrect." << std::endl;
  58                 exit(1);
  59         }
  60         if (config.get<std::string>(user+".domain") != domain) {
  61                 std::cerr << "Domain incorrect." << std::endl;
  62                 exit(1);
  63         }
  64
  65         /* preapre the pipe */
  66         int pipe_ends[2];
  67         if (pipe(pipe_ends) < 0) {
  68                 std::cerr << "Error opening pipe." << std::endl;
  69                 exit(1);
  70         }
  71
  72         /* Launch nsupdate */
  73         pid_t child_pid = fork();
  74         if (child_pid < 0) {
  75                 std::cerr << "Error while forking." << std::endl;
  76                 exit(1);
  77         }
  78         if (child_pid == 0) {
  79                 /* We're in the child */
  80                 /* Close write end, use read and as stdin */
  81                 close(pipe_ends[1]);
  82                 if (dup2(pipe_ends[0], fileno(stdin)) < 0) {
  83                         std::cerr << "There was an error redirecting stdin." << std::endl;
  84                         exit(1);
  85                 }
  86                 /* exec nsupdate */
  87                 execl(nsupdate.c_str(), nsupdate.c_str(), "-l", (char *)NULL);
  88                 /* There was an error */
  89                 std::cerr << "There was an error executing nsupdate." << std::endl;
  90                 exit(1);
  91         }
  92
  93         /* Send it the command */
  94         write(pipe_ends[1], "update delete ");
  95         write(pipe_ends[1], domain.c_str());
  96         write(pipe_ends[1], ". A\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 }