2 * This file is part of PowerDNS or dnsdist.
3 * Copyright -- PowerDNS.COM B.V. and its contributors
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of version 2 of the GNU General Public License as
7 * published by the Free Software Foundation.
9 * In addition, for the avoidance of any doubt, permission is granted to
10 * link this program with OpenSSL and to (re)distribute the binaries
11 * produced as the result of such linking.
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.
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 Street, Fifth Floor, Boston, MA 02110-1301 USA.
25 #include "rec_channel.hh"
27 #include "pdnsexception.hh"
28 #include "arguments.hh"
30 #include "namespaces.hh"
38 static void initArguments(int argc
, char** argv
)
40 arg().set("config-dir","Location of configuration directory (recursor.conf)")=SYSCONFDIR
;
42 arg().set("socket-dir",string("Where the controlsocket will live, ")+LOCALSTATEDIR
+" when unset and not chrooted" )="";
43 arg().set("chroot","switch to chroot jail")="";
44 arg().set("process","When controlling multiple recursors, the target process number")="";
45 arg().set("timeout", "Number of seconds to wait for the recursor to respond")="5";
46 arg().set("config-name","Name of this virtual configuration - will rename the binary image")="";
47 arg().setCmd("help","Provide this helpful message");
48 arg().setCmd("version","Show the version of this program");
50 arg().laxParse(argc
,argv
);
51 if(arg().mustDo("help") || arg().getCommands().empty()) {
52 cout
<<"syntax: rec_control [options] command, options as below: "<<endl
<<endl
;
53 cout
<<arg().helpstring(arg()["help"])<<endl
;
54 cout
<<"In addition, 'rec_control help' can be used to retrieve a list\nof available commands from PowerDNS"<<endl
;
55 exit(arg().mustDo("help") ? 0 : 99);
58 if(arg().mustDo("version")) {
59 cout
<<"rec_control version "<<VERSION
<<endl
;
63 string configname
=::arg()["config-dir"]+"/recursor.conf";
64 if (::arg()["config-name"] != "")
65 configname
=::arg()["config-dir"]+"/recursor-"+::arg()["config-name"]+".conf";
67 cleanSlashes(configname
);
69 arg().laxFile(configname
.c_str());
71 arg().laxParse(argc
,argv
); // make sure the commandline wins
73 if (::arg()["socket-dir"].empty()) {
74 if (::arg()["chroot"].empty())
75 ::arg().set("socket-dir") = LOCALSTATEDIR
;
77 ::arg().set("socket-dir") = ::arg()["chroot"] + "/";
78 } else if (!::arg()["chroot"].empty()) {
79 ::arg().set("socket-dir") = ::arg()["chroot"] + "/" + ::arg()["socket-dir"];
83 int main(int argc
, char** argv
)
86 initArguments(argc
, argv
);
87 RecursorControlChannel rccS
;
88 string sockname
="pdns_recursor";
90 if (arg()["config-name"] != "")
91 sockname
+="-"+arg()["config-name"];
93 if(!arg()["process"].empty())
94 sockname
+="."+arg()["process"];
96 sockname
.append(".controlsocket");
98 rccS
.connect(arg()["socket-dir"], sockname
);
100 const vector
<string
>&commands
=arg().getCommands();
102 for(unsigned int i
=0; i
< commands
.size(); ++i
) {
105 command
+=commands
[i
];
108 string receive
=rccS
.recv(0, arg().asNum("timeout"));
109 if(receive
.compare(0, 7, "Unknown") == 0) {
116 catch(PDNSException
& ae
)
118 cerr
<<"Fatal: "<<ae
.reason
<<"\n";