]>
git.ipfire.org Git - thirdparty/pdns.git/blob - pdns/notify.cc
dace6775c03291eeee7dc14c2247d9c5e69793f7
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.
26 #include "dnsparser.hh"
27 #include "dns_random.hh"
29 #include <boost/program_options.hpp>
31 #include <boost/format.hpp>
32 #include <boost/utility.hpp>
33 #include <boost/multi_index_container.hpp>
34 #include <boost/multi_index/ordered_index.hpp>
35 #include <boost/multi_index/key_extractors.hpp>
39 #include "arguments.hh"
41 #include "namespaces.hh"
42 using namespace ::boost::multi_index
;
43 #include "namespaces.hh"
45 namespace po
= boost::program_options
;
46 po::variables_map g_vm
;
56 cerr
<<"Syntax: pdns_notify IP_ADDRESS/HOSTNAME[:PORT] DOMAIN"<<endl
;
59 int main(int argc
, char** argv
)
62 set
<ComboAddress
> addrs
;
63 ::arg().set("rng")="auto";
64 ::arg().set("entropy-source")="/dev/urandom";
66 for(int n
=1 ; n
< argc
; ++n
) {
67 if ((string
) argv
[n
] == "--help") {
72 if ((string
) argv
[n
] == "--version") {
73 cerr
<<"notify "<<VERSION
<<endl
;
85 // ComboAddress local("127.0.0.1", (int)0);
86 // if(::bind(sock, (struct sockaddr*) &local, local.getSocklen()) < 0)
87 // throw runtime_error("Failed to bind local socket to address "+local.toString()+": "+stringerror());
90 addrs
.emplace(ComboAddress
{argv
[1], 53});
91 } catch (PDNSException
&ex
) {
92 /* needs resolving, maybe */
93 struct addrinfo
*info
;
95 boost::split(parts
, argv
[1], [](char c
){return c
== ':';});
96 if (parts
.size() == 1)
97 parts
.push_back("domain");
98 else if (parts
.size() != 2)
99 throw runtime_error("Invalid hostname:port syntax");
100 if (getaddrinfo(parts
[0].c_str(), parts
[1].c_str(), NULL
, &info
) < 0)
101 throw runtime_error("Cannot resolve '" + string(argv
[1]) +"'");
102 for(auto ptr
= info
; ptr
!= NULL
; ptr
= ptr
->ai_next
)
103 addrs
.emplace(ComboAddress
{ptr
->ai_addr
, ptr
->ai_addrlen
});
106 for(const auto &addr
: addrs
) {
109 sock
= socket(addr
.sin4
.sin_family
, SOCK_DGRAM
, 0);
111 throw runtime_error("Creating socket for incoming packets: "+stringerror());
112 if(connect(sock
, (struct sockaddr
*)&addr
, addr
.getSocklen()) < 0) {
113 cerr
<<"Failed to connect to address "+addr
.toStringWithPort()+": "+stringerror()<<endl
;
116 vector
<uint8_t> outpacket
;
117 DNSPacketWriter
pw(outpacket
, DNSName(argv
[2]), QType::SOA
, 1, Opcode::Notify
);
118 pw
.getHeader()->id
= dns_random_uint16();
120 if(send(sock
, &outpacket
[0], outpacket
.size(), 0) < 0) {
121 cerr
<<"Unable to send notify to "<<addr
.toStringWithPort()<<": "+stringerror()<<endl
;
131 FD_SET(sock
, &errfds
);
136 for(int tries
=0; tries
<60; tries
++) {
139 if ((len
= select(sock
+1, &rfds
, nullptr, &errfds
, &tv
)) > 0) {
140 len
= recv(sock
, buffer
, sizeof(buffer
), 0);
147 cerr
<<"Unable to receive notification response from "<<addr
.toStringWithPort()<<": "+stringerror()<<endl
;
149 } else if (timeout
) {
150 cerr
<<"Unable to receive notification response from "<<addr
.toStringWithPort()<<": Timed out"<<endl
;
152 } else if (len
== 0) {
153 cerr
<<"Unable to receive notification response from "<<addr
.toStringWithPort()<<": EOF"<<endl
;
157 string
packet(buffer
, len
);
158 MOADNSParser
mdp(false, packet
);
160 if (mdp
.d_header
.rcode
== 0)
161 cerr
<<"Successfully notified "<<addr
.toStringWithPort()<<endl
;
163 cerr
<<"Received notification response with error from "<<addr
.toStringWithPort()<<": "<<RCode::to_s(mdp
.d_header
.rcode
)<<endl
;
164 cerr
<<"For: '"<<mdp
.d_qname
<<"'"<<endl
;
167 catch(std::exception
& e
)
169 cerr
<<"Fatal: "<<e
.what()<<endl
;