From: Ralf Jung Date: Tue, 14 Oct 2014 12:58:39 +0000 (+0200) Subject: tab -> whitespaces X-Git-Url: https://git.ralfj.de/dyn-nsupdate.git/commitdiff_plain/0c40e9f91f866017b6fe850da4ddf8b9c3a91ac6?hp=229a31d3a318cc66dc82428f1429c3aea787f64f tab -> whitespaces --- diff --git a/dyn-nsupdate.cpp b/dyn-nsupdate.cpp index b0bf885..def39c6 100644 --- a/dyn-nsupdate.cpp +++ b/dyn-nsupdate.cpp @@ -1,30 +1,30 @@ /* Copyright (c) 2014, Ralf Jung - * All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions are met: - * - * 1. Redistributions of source code must retain the above copyright notice, this - * list of conditions and the following disclaimer. - * 2. Redistributions in binary form must reproduce the above copyright notice, - * this list of conditions and the following disclaimer in the documentation - * and/or other materials provided with the distribution. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND - * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED - * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE - * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR - * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES - * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; - * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND - * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT - * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS - * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. - * - * The views and conclusions contained in the software and documentation are those - * of the authors and should not be interpreted as representing official policies, - * either expressed or implied, of the FreeBSD Project. - */ +* All rights reserved. +* +* Redistribution and use in source and binary forms, with or without +* modification, are permitted provided that the following conditions are met: +* +* 1. Redistributions of source code must retain the above copyright notice, this +* list of conditions and the following disclaimer. +* 2. Redistributions in binary form must reproduce the above copyright notice, +* this list of conditions and the following disclaimer in the documentation +* and/or other materials provided with the distribution. +* +* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND +* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED +* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE +* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR +* ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES +* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; +* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND +* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT +* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS +* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. +* +* The views and conclusions contained in the software and documentation are those +* of the authors and should not be interpreted as representing official policies, +* either expressed or implied, of the FreeBSD Project. +*/ #include #include @@ -42,110 +42,110 @@ namespace po = boost::program_options; static void write(int fd, const char *str) { - size_t len = strlen(str); - ssize_t written = write(fd, str, len); - if (written < 0 || (size_t)written != len) { - std::cerr << "Error writing pipe." << std::endl; - exit(1); - } + size_t len = strlen(str); + ssize_t written = write(fd, str, len); + if (written < 0 || (size_t)written != len) { + std::cerr << "Error writing pipe." << std::endl; + exit(1); + } } int main(int argc, const char ** argv) { - static const boost::regex regex_ip("\\d{1,3}.\\d{1,3}.\\d{1,3}.\\d{1,3}"); - static const boost::regex regex_password("[a-zA-Z0-9.:;,_-]+"); - static const boost::regex regex_domain("[a-zA-Z0-9.]+"); - - - if (argc != 4) { - std::cerr << "Usage: " << argv[0] << " " << std::endl; - return 1; - } - - /* Obtain input */ - std::string domain = argv[1]; - std::string password = argv[2]; - std::string ip = argv[3]; - - /* Validate input */ - if (!regex_match(ip, regex_ip)) { - std::cerr << "Invalid IP address " << ip << "." << std::endl; - exit(1); - } - if (!regex_match(domain, regex_domain)) { - std::cerr << "Invalid domain " << domain << "." << std::endl; - exit(1); - } - if (!regex_match(password, regex_password)) { - std::cerr << "Invalid password " << password << "." << std::endl; - exit(1); - } - - /* read configuration */ - pt::ptree config; - pt::ini_parser::read_ini(CONFIG_FILE, config); - std::string nsupdate = config.get("nsupdate"); - unsigned server_port = config.get("port", 53); - - /* Given the domain, check whether the password matches */ - boost::optional correct_password = config.get_optional(pt::ptree::path_type(domain+"/password", '/')); - if (!correct_password || *correct_password != password) { - std::cerr << "Password incorrect." << std::endl; - exit(1); - } - - /* preapre the pipe */ - int pipe_ends[2]; - if (pipe(pipe_ends) < 0) { - std::cerr << "Error opening pipe." << std::endl; - exit(1); - } + static const boost::regex regex_ip("\\d{1,3}.\\d{1,3}.\\d{1,3}.\\d{1,3}"); + static const boost::regex regex_password("[a-zA-Z0-9.:;,_-]+"); + static const boost::regex regex_domain("[a-zA-Z0-9.]+"); + + + if (argc != 4) { + std::cerr << "Usage: " << argv[0] << " " << std::endl; + return 1; + } + + /* Obtain input */ + std::string domain = argv[1]; + std::string password = argv[2]; + std::string ip = argv[3]; + + /* Validate input */ + if (!regex_match(ip, regex_ip)) { + std::cerr << "Invalid IP address " << ip << "." << std::endl; + exit(1); + } + if (!regex_match(domain, regex_domain)) { + std::cerr << "Invalid domain " << domain << "." << std::endl; + exit(1); + } + if (!regex_match(password, regex_password)) { + std::cerr << "Invalid password " << password << "." << std::endl; + exit(1); + } + + /* read configuration */ + pt::ptree config; + pt::ini_parser::read_ini(CONFIG_FILE, config); + std::string nsupdate = config.get("nsupdate"); + unsigned server_port = config.get("port", 53); + + /* Given the domain, check whether the password matches */ + boost::optional correct_password = config.get_optional(pt::ptree::path_type(domain+"/password", '/')); + if (!correct_password || *correct_password != password) { + std::cerr << "Password incorrect." << std::endl; + exit(1); + } + + /* preapre the pipe */ + int pipe_ends[2]; + if (pipe(pipe_ends) < 0) { + std::cerr << "Error opening pipe." << std::endl; + exit(1); + } - /* Launch nsupdate */ - pid_t child_pid = fork(); - if (child_pid < 0) { - std::cerr << "Error while forking." << std::endl; - exit(1); - } - if (child_pid == 0) { - /* We're in the child */ - /* Close write end, use read end as stdin */ - close(pipe_ends[1]); - if (dup2(pipe_ends[0], fileno(stdin)) < 0) { - std::cerr << "There was an error redirecting stdin." << std::endl; - exit(1); - } - /* exec nsupdate */ - execl(nsupdate.c_str(), nsupdate.c_str(), "-p", std::to_string(server_port).c_str(), "-l", (char *)NULL); - /* There was an error */ - std::cerr << "There was an error executing nsupdate." << std::endl; - exit(1); - } - - /* Send it the command */ - write(pipe_ends[1], "update delete "); - write(pipe_ends[1], domain.c_str()); - write(pipe_ends[1], ". A\n"); - - write(pipe_ends[1], "update add "); - write(pipe_ends[1], domain.c_str()); - write(pipe_ends[1], ". 60 A "); - write(pipe_ends[1], ip.c_str()); - write(pipe_ends[1], "\n"); - - write(pipe_ends[1], "send\n"); - - /* Close both ends */ - close(pipe_ends[0]); - close(pipe_ends[1]); - - /* Wait for child to be gone */ - int child_status; - waitpid(child_pid, &child_status, 0); - if (child_status != 0) { - std::cerr << "There was an error in the child." << std::endl; - exit(1); - } - - return 0; + /* Launch nsupdate */ + pid_t child_pid = fork(); + if (child_pid < 0) { + std::cerr << "Error while forking." << std::endl; + exit(1); + } + if (child_pid == 0) { + /* We're in the child */ + /* Close write end, use read end as stdin */ + close(pipe_ends[1]); + if (dup2(pipe_ends[0], fileno(stdin)) < 0) { + std::cerr << "There was an error redirecting stdin." << std::endl; + exit(1); + } + /* exec nsupdate */ + execl(nsupdate.c_str(), nsupdate.c_str(), "-p", std::to_string(server_port).c_str(), "-l", (char *)NULL); + /* There was an error */ + std::cerr << "There was an error executing nsupdate." << std::endl; + exit(1); + } + + /* Send it the command */ + write(pipe_ends[1], "update delete "); + write(pipe_ends[1], domain.c_str()); + write(pipe_ends[1], ". A\n"); + + write(pipe_ends[1], "update add "); + write(pipe_ends[1], domain.c_str()); + write(pipe_ends[1], ". 60 A "); + write(pipe_ends[1], ip.c_str()); + write(pipe_ends[1], "\n"); + + write(pipe_ends[1], "send\n"); + + /* Close both ends */ + close(pipe_ends[0]); + close(pipe_ends[1]); + + /* Wait for child to be gone */ + int child_status; + waitpid(child_pid, &child_status, 0); + if (child_status != 0) { + std::cerr << "There was an error in the child." << std::endl; + exit(1); + } + + return 0; }