pdns/dynloader.cc

   1 /*
   2     PowerDNS Versatile Database Driven Nameserver
   3     Copyright (C) 2002 - 2008  PowerDNS.COM BV
   4
   5     This program is free software; you can redistribute it and/or modify
   6     it under the terms of the GNU General Public License version 2 as
   7     published by the Free Software Foundation
   8
   9     Additionally, the license of this program contains a special
  10     exception which allows to distribute the program in binary form when
  11     it is linked against OpenSSL.
  12
  13     This program is distributed in the hope that it will be useful,
  14     but WITHOUT ANY WARRANTY; without even the implied warranty of
  15     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16     GNU General Public License for more details.
  17
  18     You should have received a copy of the GNU General Public License
  19     along with this program; if not, write to the Free Software
  20     Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
  21 */
  22 #ifdef HAVE_CONFIG_H
  23 #include "config.h"
  24 #endif
  25 #include <iostream>
  26 #include <cstdio>
  27 #include <cstring>
  28 #include <cstdlib>
  29 #include <unistd.h>
  30 #include <errno.h>
  31 #include <climits>
  32 #include <string>
  33 #include <map>
  34 #include <sys/mman.h>
  35 #include <fcntl.h>
  36 #include <sys/types.h>
  37
  38
  39 #include <sys/stat.h>
  40 #include "pdnsexception.hh"
  41 #include "misc.hh"
  42 #include "dynmessenger.hh"
  43 #include "arguments.hh"
  44 #include "statbag.hh"
  45 #include "misc.hh"
  46 #include "namespaces.hh"
  47 #include "namespaces.hh"
  48
  49 ArgvMap &arg()
  50 {
  51   static ArgvMap arg;
  52   return arg;
  53 }
  54
  55 StatBag S;
  56
  57 int main(int argc, char **argv)
  58 {
  59   string s_programname="pdns";
  60
  61   ::arg().set("config-dir","Location of configuration directory (pdns.conf)")=SYSCONFDIR;
  62   ::arg().set("socket-dir",string("Where the controlsocket will live, ")+LOCALSTATEDIR+" when unset and not chrooted" )="";
  63   ::arg().set("remote-address","Remote address to query");
  64   ::arg().set("remote-port","Remote port to query")="53000";
  65   ::arg().set("secret","Secret needed to connect to remote PowerDNS");
  66
  67   ::arg().set("config-name","Name of this virtual configuration - will rename the binary image")="";
  68   ::arg().setCmd("no-config","Don't parse configuration file");
  69   ::arg().set("chroot","")="";
  70   ::arg().setCmd("help","Provide a helpful message");
  71   ::arg().laxParse(argc,argv);
  72
  73   if(::arg().mustDo("help")) {
  74     cout<<"syntax:"<<endl<<endl;
  75     cout<<::arg().helpstring(::arg()["help"])<<endl;
  76     cout<<"In addition, 'pdns_control help' can be used to retrieve a list\nof available commands from PowerDNS"<<endl;
  77     exit(0);
  78   }
  79
  80   const vector<string>commands=::arg().getCommands();
  81
  82   if(commands.empty()) {
  83     cerr<<"No command passed"<<endl;
  84     return 0;
  85   }
  86
  87   if(::arg()["config-name"]!="")
  88     s_programname+="-"+::arg()["config-name"];
  89
  90   string configname=::arg()["config-dir"]+"/"+s_programname+".conf";
  91   cleanSlashes(configname);
  92
  93   if(!::arg().mustDo("no-config")) {
  94     ::arg().laxFile(configname.c_str());
  95     ::arg().laxParse(argc,argv); // reparse so the commandline still wins
  96   }
  97
  98   string socketname=::arg()["socket-dir"];
  99   if (::arg()["socket-dir"].empty()) {
 100     if (::arg()["chroot"].empty())
 101       socketname = LOCALSTATEDIR;
 102     else
 103       socketname = ::arg()["chroot"] + "/";
 104   } else if (!::arg()["socket-dir"].empty() && !::arg()["chroot"].empty()) {
 105     socketname = ::arg()["chroot"] + ::arg()["socket-dir"];
 106   }
 107
 108   socketname += "/" + s_programname + ".controlsocket";
 109   cleanSlashes(socketname);
 110
 111   try {
 112     string command=commands[0];
 113     shared_ptr<DynMessenger> D;
 114     if(::arg()["remote-address"].empty())
 115       D=shared_ptr<DynMessenger>(new DynMessenger(socketname));
 116     else {
 117       uint16_t port;
 118       try {
 119         port = static_cast<uint16_t>(pdns_stou(::arg()["remote-port"]));
 120       }
 121       catch(...) {
 122         cerr<<"Unable to convert '"<<::arg()["remote-port"]<<"' to a port number for connecting to remote PowerDNS\n";
 123         exit(99);
 124       }
 125
 126       D=shared_ptr<DynMessenger>(new DynMessenger(ComboAddress(::arg()["remote-address"], port), ::arg()["secret"]));
 127     }
 128
 129     string message;
 130     for(vector<string>::const_iterator i=commands.begin();i!=commands.end();++i) {
 131       if(i!=commands.begin())
 132         message+=" ";
 133       message+=*i;
 134     }
 135
 136     if(command=="show") {
 137       message="SHOW ";
 138       for(unsigned int n=1;n<commands.size();n++) {
 139         message+=commands[n];
 140         message+=" ";
 141       }
 142     }
 143     else if(command=="list") {
 144       message="SHOW *";
 145       command="show";
 146     }
 147     else if(command=="quit" || command=="QUIT") {
 148       message="QUIT";
 149     }
 150     else if(command=="status" || command=="STATUS") {
 151       message="STATUS";
 152     }
 153     else if(command=="version" || command=="VERSION") {
 154       message="VERSION";
 155     }
 156
 157
 158     if(D->send(message)<0) {
 159       cerr<<"Error sending command"<<endl;
 160       return 1;
 161     }
 162
 163     string resp=D->receive();
 164     if(resp.compare(0, 7, "Unknown") == 0) {
 165       cerr<<resp<<endl;
 166       return 1;
 167     }
 168
 169     cout<<resp<<endl;
 170   }
 171   catch(TimeoutException &ae) {
 172     cerr<<"Timeout error: "<<ae.reason<<endl;
 173     return 2;
 174   }
 175   catch(PDNSException &ae) {
 176     cerr<<"Fatal error: "<<ae.reason<<endl;
 177     return 1;
 178   }
 179   catch(const std::runtime_error& e) {
 180     cerr<<"Runtime error: "<<e.what()<<endl;
 181     return 2;
 182   }
 183   return 0;
 184 }
 185
 186