]> CyberLeo.Net >> Repos - FreeBSD/releng/9.2.git/blob - contrib/bind9/bin/named/statschannel.c
- Copy stable/9 to releng/9.2 as part of the 9.2-RELEASE cycle.
[FreeBSD/releng/9.2.git] / contrib / bind9 / bin / named / statschannel.c
1 /*
2  * Copyright (C) 2008-2012  Internet Systems Consortium, Inc. ("ISC")
3  *
4  * Permission to use, copy, modify, and/or distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  *
8  * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
9  * REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
10  * AND FITNESS.  IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
11  * INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
12  * LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
13  * OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
14  * PERFORMANCE OF THIS SOFTWARE.
15  */
16
17 /* $Id: statschannel.c,v 1.26.150.2 2011/03/12 04:59:14 tbox Exp $ */
18
19 /*! \file */
20
21 #include <config.h>
22
23 #include <isc/buffer.h>
24 #include <isc/httpd.h>
25 #include <isc/mem.h>
26 #include <isc/once.h>
27 #include <isc/print.h>
28 #include <isc/socket.h>
29 #include <isc/stats.h>
30 #include <isc/task.h>
31
32 #include <dns/cache.h>
33 #include <dns/db.h>
34 #include <dns/opcode.h>
35 #include <dns/resolver.h>
36 #include <dns/rdataclass.h>
37 #include <dns/rdatatype.h>
38 #include <dns/stats.h>
39 #include <dns/view.h>
40 #include <dns/zt.h>
41
42 #include <named/log.h>
43 #include <named/server.h>
44 #include <named/statschannel.h>
45
46 #include "bind9.xsl.h"
47
48 struct ns_statschannel {
49         /* Unlocked */
50         isc_httpdmgr_t                          *httpdmgr;
51         isc_sockaddr_t                          address;
52         isc_mem_t                               *mctx;
53
54         /*
55          * Locked by channel lock: can be referenced and modified by both
56          * the server task and the channel task.
57          */
58         isc_mutex_t                             lock;
59         dns_acl_t                               *acl;
60
61         /* Locked by server task */
62         ISC_LINK(struct ns_statschannel)        link;
63 };
64
65 typedef enum { statsformat_file, statsformat_xml } statsformat_t;
66
67 typedef struct
68 stats_dumparg {
69         statsformat_t   type;
70         void            *arg;           /* type dependent argument */
71         int             ncounters;      /* used for general statistics */
72         int             *counterindices; /* used for general statistics */
73         isc_uint64_t    *countervalues;  /* used for general statistics */
74         isc_result_t    result;
75 } stats_dumparg_t;
76
77 static isc_once_t once = ISC_ONCE_INIT;
78
79 /*%
80  * Statistics descriptions.  These could be statistically initialized at
81  * compile time, but we configure them run time in the init_desc() function
82  * below so that they'll be less susceptible to counter name changes.
83  */
84 static const char *nsstats_desc[dns_nsstatscounter_max];
85 static const char *resstats_desc[dns_resstatscounter_max];
86 static const char *zonestats_desc[dns_zonestatscounter_max];
87 static const char *sockstats_desc[isc_sockstatscounter_max];
88 static const char *dnssecstats_desc[dns_dnssecstats_max];
89 #ifdef HAVE_LIBXML2
90 static const char *nsstats_xmldesc[dns_nsstatscounter_max];
91 static const char *resstats_xmldesc[dns_resstatscounter_max];
92 static const char *zonestats_xmldesc[dns_zonestatscounter_max];
93 static const char *sockstats_xmldesc[isc_sockstatscounter_max];
94 static const char *dnssecstats_xmldesc[dns_dnssecstats_max];
95 #else
96 #define nsstats_xmldesc NULL
97 #define resstats_xmldesc NULL
98 #define zonestats_xmldesc NULL
99 #define sockstats_xmldesc NULL
100 #define dnssecstats_xmldesc NULL
101 #endif  /* HAVE_LIBXML2 */
102
103 #define TRY0(a) do { xmlrc = (a); if (xmlrc < 0) goto error; } while(0)
104
105 /*%
106  * Mapping arrays to represent statistics counters in the order of our
107  * preference, regardless of the order of counter indices.  For example,
108  * nsstats_desc[nsstats_index[0]] will be the description that is shown first.
109  */
110 static int nsstats_index[dns_nsstatscounter_max];
111 static int resstats_index[dns_resstatscounter_max];
112 static int zonestats_index[dns_zonestatscounter_max];
113 static int sockstats_index[isc_sockstatscounter_max];
114 static int dnssecstats_index[dns_dnssecstats_max];
115
116 static inline void
117 set_desc(int counter, int maxcounter, const char *fdesc, const char **fdescs,
118          const char *xdesc, const char **xdescs)
119 {
120         REQUIRE(counter < maxcounter);
121         REQUIRE(fdescs[counter] == NULL);
122 #ifdef HAVE_LIBXML2
123         REQUIRE(xdescs[counter] == NULL);
124 #endif
125
126         fdescs[counter] = fdesc;
127 #ifdef HAVE_LIBXML2
128         xdescs[counter] = xdesc;
129 #else
130         UNUSED(xdesc);
131         UNUSED(xdescs);
132 #endif
133 }
134
135 static void
136 init_desc(void) {
137         int i;
138
139         /* Initialize name server statistics */
140         for (i = 0; i < dns_nsstatscounter_max; i++)
141                 nsstats_desc[i] = NULL;
142 #ifdef HAVE_LIBXML2
143         for (i = 0; i < dns_nsstatscounter_max; i++)
144                 nsstats_xmldesc[i] = NULL;
145 #endif
146
147 #define SET_NSSTATDESC(counterid, desc, xmldesc) \
148         do { \
149                 set_desc(dns_nsstatscounter_ ## counterid, \
150                          dns_nsstatscounter_max, \
151                          desc, nsstats_desc, xmldesc, nsstats_xmldesc); \
152                 nsstats_index[i++] = dns_nsstatscounter_ ## counterid; \
153         } while (0)
154
155         i = 0;
156         SET_NSSTATDESC(requestv4, "IPv4 requests received", "Requestv4");
157         SET_NSSTATDESC(requestv6, "IPv6 requests received", "Requestv6");
158         SET_NSSTATDESC(edns0in, "requests with EDNS(0) received", "ReqEdns0");
159         SET_NSSTATDESC(badednsver,
160                        "requests with unsupported EDNS version received",
161                        "ReqBadEDNSVer");
162         SET_NSSTATDESC(tsigin, "requests with TSIG received", "ReqTSIG");
163         SET_NSSTATDESC(sig0in, "requests with SIG(0) received", "ReqSIG0");
164         SET_NSSTATDESC(invalidsig, "requests with invalid signature",
165                        "ReqBadSIG");
166         SET_NSSTATDESC(tcp, "TCP requests received", "ReqTCP");
167         SET_NSSTATDESC(authrej, "auth queries rejected", "AuthQryRej");
168         SET_NSSTATDESC(recurserej, "recursive queries rejected", "RecQryRej");
169         SET_NSSTATDESC(xfrrej, "transfer requests rejected", "XfrRej");
170         SET_NSSTATDESC(updaterej, "update requests rejected", "UpdateRej");
171         SET_NSSTATDESC(response, "responses sent", "Response");
172         SET_NSSTATDESC(truncatedresp, "truncated responses sent",
173                        "TruncatedResp");
174         SET_NSSTATDESC(edns0out, "responses with EDNS(0) sent", "RespEDNS0");
175         SET_NSSTATDESC(tsigout, "responses with TSIG sent", "RespTSIG");
176         SET_NSSTATDESC(sig0out, "responses with SIG(0) sent", "RespSIG0");
177         SET_NSSTATDESC(success, "queries resulted in successful answer",
178                        "QrySuccess");
179         SET_NSSTATDESC(authans, "queries resulted in authoritative answer",
180                        "QryAuthAns");
181         SET_NSSTATDESC(nonauthans,
182                        "queries resulted in non authoritative answer",
183                        "QryNoauthAns");
184         SET_NSSTATDESC(referral, "queries resulted in referral answer",
185                        "QryReferral");
186         SET_NSSTATDESC(nxrrset, "queries resulted in nxrrset", "QryNxrrset");
187         SET_NSSTATDESC(servfail, "queries resulted in SERVFAIL", "QrySERVFAIL");
188         SET_NSSTATDESC(formerr, "queries resulted in FORMERR", "QryFORMERR");
189         SET_NSSTATDESC(nxdomain, "queries resulted in NXDOMAIN", "QryNXDOMAIN");
190         SET_NSSTATDESC(recursion, "queries caused recursion","QryRecursion");
191         SET_NSSTATDESC(duplicate, "duplicate queries received", "QryDuplicate");
192         SET_NSSTATDESC(dropped, "queries dropped", "QryDropped");
193         SET_NSSTATDESC(failure, "other query failures", "QryFailure");
194         SET_NSSTATDESC(xfrdone, "requested transfers completed", "XfrReqDone");
195         SET_NSSTATDESC(updatereqfwd, "update requests forwarded",
196                        "UpdateReqFwd");
197         SET_NSSTATDESC(updaterespfwd, "update responses forwarded",
198                        "UpdateRespFwd");
199         SET_NSSTATDESC(updatefwdfail, "update forward failed", "UpdateFwdFail");
200         SET_NSSTATDESC(updatedone, "updates completed", "UpdateDone");
201         SET_NSSTATDESC(updatefail, "updates failed", "UpdateFail");
202         SET_NSSTATDESC(updatebadprereq,
203                        "updates rejected due to prerequisite failure",
204                        "UpdateBadPrereq");
205         INSIST(i == dns_nsstatscounter_max);
206
207         /* Initialize resolver statistics */
208         for (i = 0; i < dns_resstatscounter_max; i++)
209                 resstats_desc[i] = NULL;
210 #ifdef  HAVE_LIBXML2
211         for (i = 0; i < dns_resstatscounter_max; i++)
212                 resstats_xmldesc[i] = NULL;
213 #endif
214
215 #define SET_RESSTATDESC(counterid, desc, xmldesc) \
216         do { \
217                 set_desc(dns_resstatscounter_ ## counterid, \
218                          dns_resstatscounter_max, \
219                          desc, resstats_desc, xmldesc, resstats_xmldesc); \
220                 resstats_index[i++] = dns_resstatscounter_ ## counterid; \
221         } while (0)
222
223         i = 0;
224         SET_RESSTATDESC(queryv4, "IPv4 queries sent", "Queryv4");
225         SET_RESSTATDESC(queryv6, "IPv6 queries sent", "Queryv6");
226         SET_RESSTATDESC(responsev4, "IPv4 responses received", "Responsev4");
227         SET_RESSTATDESC(responsev6, "IPv6 responses received", "Responsev6");
228         SET_RESSTATDESC(nxdomain, "NXDOMAIN received", "NXDOMAIN");
229         SET_RESSTATDESC(servfail, "SERVFAIL received", "SERVFAIL");
230         SET_RESSTATDESC(formerr, "FORMERR received", "FORMERR");
231         SET_RESSTATDESC(othererror, "other errors received", "OtherError");
232         SET_RESSTATDESC(edns0fail, "EDNS(0) query failures", "EDNS0Fail");
233         SET_RESSTATDESC(mismatch, "mismatch responses received", "Mismatch");
234         SET_RESSTATDESC(truncated, "truncated responses received", "Truncated");
235         SET_RESSTATDESC(lame, "lame delegations received", "Lame");
236         SET_RESSTATDESC(retry, "query retries", "Retry");
237         SET_RESSTATDESC(dispabort, "queries aborted due to quota",
238                         "QueryAbort");
239         SET_RESSTATDESC(dispsockfail, "failures in opening query sockets",
240                         "QuerySockFail");
241         SET_RESSTATDESC(querytimeout, "query timeouts", "QueryTimeout");
242         SET_RESSTATDESC(gluefetchv4, "IPv4 NS address fetches", "GlueFetchv4");
243         SET_RESSTATDESC(gluefetchv6, "IPv6 NS address fetches", "GlueFetchv6");
244         SET_RESSTATDESC(gluefetchv4fail, "IPv4 NS address fetch failed",
245                         "GlueFetchv4Fail");
246         SET_RESSTATDESC(gluefetchv6fail, "IPv6 NS address fetch failed",
247                         "GlueFetchv6Fail");
248         SET_RESSTATDESC(val, "DNSSEC validation attempted", "ValAttempt");
249         SET_RESSTATDESC(valsuccess, "DNSSEC validation succeeded", "ValOk");
250         SET_RESSTATDESC(valnegsuccess, "DNSSEC NX validation succeeded",
251                         "ValNegOk");
252         SET_RESSTATDESC(valfail, "DNSSEC validation failed", "ValFail");
253         SET_RESSTATDESC(queryrtt0, "queries with RTT < "
254                         DNS_RESOLVER_QRYRTTCLASS0STR "ms",
255                         "QryRTT" DNS_RESOLVER_QRYRTTCLASS0STR);
256         SET_RESSTATDESC(queryrtt1, "queries with RTT "
257                         DNS_RESOLVER_QRYRTTCLASS0STR "-"
258                         DNS_RESOLVER_QRYRTTCLASS1STR "ms",
259                         "QryRTT" DNS_RESOLVER_QRYRTTCLASS1STR);
260         SET_RESSTATDESC(queryrtt2, "queries with RTT "
261                         DNS_RESOLVER_QRYRTTCLASS1STR "-"
262                         DNS_RESOLVER_QRYRTTCLASS2STR "ms",
263                         "QryRTT" DNS_RESOLVER_QRYRTTCLASS2STR);
264         SET_RESSTATDESC(queryrtt3, "queries with RTT "
265                         DNS_RESOLVER_QRYRTTCLASS2STR "-"
266                         DNS_RESOLVER_QRYRTTCLASS3STR "ms",
267                         "QryRTT" DNS_RESOLVER_QRYRTTCLASS3STR);
268         SET_RESSTATDESC(queryrtt4, "queries with RTT "
269                         DNS_RESOLVER_QRYRTTCLASS3STR "-"
270                         DNS_RESOLVER_QRYRTTCLASS4STR "ms",
271                         "QryRTT" DNS_RESOLVER_QRYRTTCLASS4STR);
272         SET_RESSTATDESC(queryrtt5, "queries with RTT > "
273                         DNS_RESOLVER_QRYRTTCLASS4STR "ms",
274                         "QryRTT" DNS_RESOLVER_QRYRTTCLASS4STR "+");
275         INSIST(i == dns_resstatscounter_max);
276
277         /* Initialize zone statistics */
278         for (i = 0; i < dns_zonestatscounter_max; i++)
279                 zonestats_desc[i] = NULL;
280 #ifdef  HAVE_LIBXML2
281         for (i = 0; i < dns_zonestatscounter_max; i++)
282                 zonestats_xmldesc[i] = NULL;
283 #endif
284
285 #define SET_ZONESTATDESC(counterid, desc, xmldesc) \
286         do { \
287                 set_desc(dns_zonestatscounter_ ## counterid, \
288                          dns_zonestatscounter_max, \
289                          desc, zonestats_desc, xmldesc, zonestats_xmldesc); \
290                 zonestats_index[i++] = dns_zonestatscounter_ ## counterid; \
291         } while (0)
292
293         i = 0;
294         SET_ZONESTATDESC(notifyoutv4, "IPv4 notifies sent", "NotifyOutv4");
295         SET_ZONESTATDESC(notifyoutv6, "IPv6 notifies sent", "NotifyOutv6");
296         SET_ZONESTATDESC(notifyinv4, "IPv4 notifies received", "NotifyInv4");
297         SET_ZONESTATDESC(notifyinv6, "IPv6 notifies received", "NotifyInv6");
298         SET_ZONESTATDESC(notifyrej, "notifies rejected", "NotifyRej");
299         SET_ZONESTATDESC(soaoutv4, "IPv4 SOA queries sent", "SOAOutv4");
300         SET_ZONESTATDESC(soaoutv6, "IPv6 SOA queries sent", "SOAOutv6");
301         SET_ZONESTATDESC(axfrreqv4, "IPv4 AXFR requested", "AXFRReqv4");
302         SET_ZONESTATDESC(axfrreqv6, "IPv6 AXFR requested", "AXFRReqv6");
303         SET_ZONESTATDESC(ixfrreqv4, "IPv4 IXFR requested", "IXFRReqv4");
304         SET_ZONESTATDESC(ixfrreqv6, "IPv6 IXFR requested", "IXFRReqv6");
305         SET_ZONESTATDESC(xfrsuccess, "transfer requests succeeded","XfrSuccess");
306         SET_ZONESTATDESC(xfrfail, "transfer requests failed", "XfrFail");
307         INSIST(i == dns_zonestatscounter_max);
308
309         /* Initialize socket statistics */
310         for (i = 0; i < isc_sockstatscounter_max; i++)
311                 sockstats_desc[i] = NULL;
312 #ifdef  HAVE_LIBXML2
313         for (i = 0; i < isc_sockstatscounter_max; i++)
314                 sockstats_xmldesc[i] = NULL;
315 #endif
316
317 #define SET_SOCKSTATDESC(counterid, desc, xmldesc) \
318         do { \
319                 set_desc(isc_sockstatscounter_ ## counterid, \
320                          isc_sockstatscounter_max, \
321                          desc, sockstats_desc, xmldesc, sockstats_xmldesc); \
322                 sockstats_index[i++] = isc_sockstatscounter_ ## counterid; \
323         } while (0)
324
325         i = 0;
326         SET_SOCKSTATDESC(udp4open, "UDP/IPv4 sockets opened", "UDP4Open");
327         SET_SOCKSTATDESC(udp6open, "UDP/IPv6 sockets opened", "UDP6Open");
328         SET_SOCKSTATDESC(tcp4open, "TCP/IPv4 sockets opened", "TCP4Open");
329         SET_SOCKSTATDESC(tcp6open, "TCP/IPv6 sockets opened", "TCP6Open");
330         SET_SOCKSTATDESC(unixopen, "Unix domain sockets opened", "UnixOpen");
331         SET_SOCKSTATDESC(udp4openfail, "UDP/IPv4 socket open failures",
332                          "UDP4OpenFail");
333         SET_SOCKSTATDESC(udp6openfail, "UDP/IPv6 socket open failures",
334                          "UDP6OpenFail");
335         SET_SOCKSTATDESC(tcp4openfail, "TCP/IPv4 socket open failures",
336                          "TCP4OpenFail");
337         SET_SOCKSTATDESC(tcp6openfail, "TCP/IPv6 socket open failures",
338                          "TCP6OpenFail");
339         SET_SOCKSTATDESC(unixopenfail, "Unix domain socket open failures",
340                          "UnixOpenFail");
341         SET_SOCKSTATDESC(udp4close, "UDP/IPv4 sockets closed", "UDP4Close");
342         SET_SOCKSTATDESC(udp6close, "UDP/IPv6 sockets closed", "UDP6Close");
343         SET_SOCKSTATDESC(tcp4close, "TCP/IPv4 sockets closed", "TCP4Close");
344         SET_SOCKSTATDESC(tcp6close, "TCP/IPv6 sockets closed", "TCP6Close");
345         SET_SOCKSTATDESC(unixclose, "Unix domain sockets closed", "UnixClose");
346         SET_SOCKSTATDESC(fdwatchclose, "FDwatch sockets closed",
347                          "FDWatchClose");
348         SET_SOCKSTATDESC(udp4bindfail, "UDP/IPv4 socket bind failures",
349                          "UDP4BindFail");
350         SET_SOCKSTATDESC(udp6bindfail, "UDP/IPv6 socket bind failures",
351                          "UDP6BindFail");
352         SET_SOCKSTATDESC(tcp4bindfail, "TCP/IPv4 socket bind failures",
353                          "TCP4BindFail");
354         SET_SOCKSTATDESC(tcp6bindfail, "TCP/IPv6 socket bind failures",
355                          "TCP6BindFail");
356         SET_SOCKSTATDESC(unixbindfail, "Unix domain socket bind failures",
357                          "UnixBindFail");
358         SET_SOCKSTATDESC(fdwatchbindfail, "FDwatch socket bind failures",
359                          "FdwatchBindFail");
360         SET_SOCKSTATDESC(udp4connectfail, "UDP/IPv4 socket connect failures",
361                          "UDP4ConnFail");
362         SET_SOCKSTATDESC(udp6connectfail, "UDP/IPv6 socket connect failures",
363                          "UDP6ConnFail");
364         SET_SOCKSTATDESC(tcp4connectfail, "TCP/IPv4 socket connect failures",
365                          "TCP4ConnFail");
366         SET_SOCKSTATDESC(tcp6connectfail, "TCP/IPv6 socket connect failures",
367                          "TCP6ConnFail");
368         SET_SOCKSTATDESC(unixconnectfail, "Unix domain socket connect failures",
369                          "UnixConnFail");
370         SET_SOCKSTATDESC(fdwatchconnectfail, "FDwatch socket connect failures",
371                          "FDwatchConnFail");
372         SET_SOCKSTATDESC(udp4connect, "UDP/IPv4 connections established",
373                          "UDP4Conn");
374         SET_SOCKSTATDESC(udp6connect, "UDP/IPv6 connections established",
375                          "UDP6Conn");
376         SET_SOCKSTATDESC(tcp4connect, "TCP/IPv4 connections established",
377                          "TCP4Conn");
378         SET_SOCKSTATDESC(tcp6connect, "TCP/IPv6 connections established",
379                          "TCP6Conn");
380         SET_SOCKSTATDESC(unixconnect, "Unix domain connections established",
381                          "UnixConn");
382         SET_SOCKSTATDESC(fdwatchconnect,
383                          "FDwatch domain connections established",
384                          "FDwatchConn");
385         SET_SOCKSTATDESC(tcp4acceptfail, "TCP/IPv4 connection accept failures",
386                          "TCP4AcceptFail");
387         SET_SOCKSTATDESC(tcp6acceptfail, "TCP/IPv6 connection accept failures",
388                          "TCP6AcceptFail");
389         SET_SOCKSTATDESC(unixacceptfail,
390                          "Unix domain connection accept failures",
391                          "UnixAcceptFail");
392         SET_SOCKSTATDESC(tcp4accept, "TCP/IPv4 connections accepted",
393                          "TCP4Accept");
394         SET_SOCKSTATDESC(tcp6accept, "TCP/IPv6 connections accepted",
395                          "TCP6Accept");
396         SET_SOCKSTATDESC(unixaccept, "Unix domain connections accepted",
397                          "UnixAccept");
398         SET_SOCKSTATDESC(udp4sendfail, "UDP/IPv4 send errors", "UDP4SendErr");
399         SET_SOCKSTATDESC(udp6sendfail, "UDP/IPv6 send errors", "UDP6SendErr");
400         SET_SOCKSTATDESC(tcp4sendfail, "TCP/IPv4 send errors", "TCP4SendErr");
401         SET_SOCKSTATDESC(tcp6sendfail, "TCP/IPv6 send errors", "TCP6SendErr");
402         SET_SOCKSTATDESC(unixsendfail, "Unix domain send errors",
403                          "UnixSendErr");
404         SET_SOCKSTATDESC(fdwatchsendfail, "FDwatch send errors",
405                          "FDwatchSendErr");
406         SET_SOCKSTATDESC(udp4recvfail, "UDP/IPv4 recv errors", "UDP4RecvErr");
407         SET_SOCKSTATDESC(udp6recvfail, "UDP/IPv6 recv errors", "UDP6RecvErr");
408         SET_SOCKSTATDESC(tcp4recvfail, "TCP/IPv4 recv errors", "TCP4RecvErr");
409         SET_SOCKSTATDESC(tcp6recvfail, "TCP/IPv6 recv errors", "TCP6RecvErr");
410         SET_SOCKSTATDESC(unixrecvfail, "Unix domain recv errors",
411                          "UnixRecvErr");
412         SET_SOCKSTATDESC(fdwatchrecvfail, "FDwatch recv errors",
413                          "FDwatchRecvErr");
414         INSIST(i == isc_sockstatscounter_max);
415
416         /* Initialize DNSSEC statistics */
417         for (i = 0; i < dns_dnssecstats_max; i++)
418                 dnssecstats_desc[i] = NULL;
419 #ifdef  HAVE_LIBXML2
420         for (i = 0; i < dns_dnssecstats_max; i++)
421                 dnssecstats_xmldesc[i] = NULL;
422 #endif
423
424 #define SET_DNSSECSTATDESC(counterid, desc, xmldesc) \
425         do { \
426                 set_desc(dns_dnssecstats_ ## counterid, \
427                          dns_dnssecstats_max, \
428                          desc, dnssecstats_desc,\
429                          xmldesc, dnssecstats_xmldesc); \
430                 dnssecstats_index[i++] = dns_dnssecstats_ ## counterid; \
431         } while (0)
432
433         i = 0;
434         SET_DNSSECSTATDESC(asis, "dnssec validation success with signer "
435                            "\"as is\"", "DNSSECasis");
436         SET_DNSSECSTATDESC(downcase, "dnssec validation success with signer "
437                            "lower cased", "DNSSECdowncase");
438         SET_DNSSECSTATDESC(wildcard, "dnssec validation of wildcard signature",
439                            "DNSSECwild");
440         SET_DNSSECSTATDESC(fail, "dnssec validation failures", "DNSSECfail");
441         INSIST(i == dns_dnssecstats_max);
442
443         /* Sanity check */
444         for (i = 0; i < dns_nsstatscounter_max; i++)
445                 INSIST(nsstats_desc[i] != NULL);
446         for (i = 0; i < dns_resstatscounter_max; i++)
447                 INSIST(resstats_desc[i] != NULL);
448         for (i = 0; i < dns_zonestatscounter_max; i++)
449                 INSIST(zonestats_desc[i] != NULL);
450         for (i = 0; i < isc_sockstatscounter_max; i++)
451                 INSIST(sockstats_desc[i] != NULL);
452         for (i = 0; i < dns_dnssecstats_max; i++)
453                 INSIST(dnssecstats_desc[i] != NULL);
454 #ifdef  HAVE_LIBXML2
455         for (i = 0; i < dns_nsstatscounter_max; i++)
456                 INSIST(nsstats_xmldesc[i] != NULL);
457         for (i = 0; i < dns_resstatscounter_max; i++)
458                 INSIST(resstats_xmldesc[i] != NULL);
459         for (i = 0; i < dns_zonestatscounter_max; i++)
460                 INSIST(zonestats_xmldesc[i] != NULL);
461         for (i = 0; i < isc_sockstatscounter_max; i++)
462                 INSIST(sockstats_xmldesc[i] != NULL);
463         for (i = 0; i < dns_dnssecstats_max; i++)
464                 INSIST(dnssecstats_xmldesc[i] != NULL);
465 #endif
466 }
467
468 /*%
469  * Dump callback functions.
470  */
471 static void
472 generalstat_dump(isc_statscounter_t counter, isc_uint64_t val, void *arg) {
473         stats_dumparg_t *dumparg = arg;
474
475         REQUIRE(counter < dumparg->ncounters);
476         dumparg->countervalues[counter] = val;
477 }
478
479 static isc_result_t
480 dump_counters(isc_stats_t *stats, statsformat_t type, void *arg,
481               const char *category, const char **desc, int ncounters,
482               int *indices, isc_uint64_t *values, int options)
483 {
484         int i, index;
485         isc_uint64_t value;
486         stats_dumparg_t dumparg;
487         FILE *fp;
488 #ifdef HAVE_LIBXML2
489         xmlTextWriterPtr writer;
490         int xmlrc;
491 #endif
492
493 #ifndef HAVE_LIBXML2
494         UNUSED(category);
495 #endif
496
497         dumparg.type = type;
498         dumparg.ncounters = ncounters;
499         dumparg.counterindices = indices;
500         dumparg.countervalues = values;
501
502         memset(values, 0, sizeof(values[0]) * ncounters);
503         isc_stats_dump(stats, generalstat_dump, &dumparg, options);
504
505         for (i = 0; i < ncounters; i++) {
506                 index = indices[i];
507                 value = values[index];
508
509                 if (value == 0 && (options & ISC_STATSDUMP_VERBOSE) == 0)
510                         continue;
511
512                 switch (dumparg.type) {
513                 case statsformat_file:
514                         fp = arg;
515                         fprintf(fp, "%20" ISC_PRINT_QUADFORMAT "u %s\n",
516                                 value, desc[index]);
517                         break;
518                 case statsformat_xml:
519 #ifdef HAVE_LIBXML2
520                         writer = arg;
521
522                         if (category != NULL) {
523                                 TRY0(xmlTextWriterStartElement(writer,
524                                                                ISC_XMLCHAR
525                                                                category));
526                                 TRY0(xmlTextWriterStartElement(writer,
527                                                                ISC_XMLCHAR
528                                                                "name"));
529                                 TRY0(xmlTextWriterWriteString(writer,
530                                                               ISC_XMLCHAR
531                                                               desc[index]));
532                                 TRY0(xmlTextWriterEndElement(writer)); /* name */
533
534                                 TRY0(xmlTextWriterStartElement(writer,
535                                                                ISC_XMLCHAR
536                                                                "counter"));
537                         } else {
538                                 TRY0(xmlTextWriterStartElement(writer,
539                                                                ISC_XMLCHAR
540                                                                desc[index]));
541                         }
542                         TRY0(xmlTextWriterWriteFormatString(writer,
543                                                             "%"
544                                                             ISC_PRINT_QUADFORMAT
545                                                             "u", value));
546                         TRY0(xmlTextWriterEndElement(writer)); /* counter */
547                         if (category != NULL)
548                                 TRY0(xmlTextWriterEndElement(writer)); /* category */
549 #endif
550                         break;
551                 }
552         }
553         return (ISC_R_SUCCESS);
554 #ifdef HAVE_LIBXML2
555  error:
556         return (ISC_R_FAILURE);
557 #endif
558 }
559
560 static void
561 rdtypestat_dump(dns_rdatastatstype_t type, isc_uint64_t val, void *arg) {
562         char typebuf[64];
563         const char *typestr;
564         stats_dumparg_t *dumparg = arg;
565         FILE *fp;
566 #ifdef HAVE_LIBXML2
567         xmlTextWriterPtr writer;
568         int xmlrc;
569 #endif
570
571         if ((DNS_RDATASTATSTYPE_ATTR(type) & DNS_RDATASTATSTYPE_ATTR_OTHERTYPE)
572             == 0) {
573                 dns_rdatatype_format(DNS_RDATASTATSTYPE_BASE(type), typebuf,
574                                      sizeof(typebuf));
575                 typestr = typebuf;
576         } else
577                 typestr = "Others";
578
579         switch (dumparg->type) {
580         case statsformat_file:
581                 fp = dumparg->arg;
582                 fprintf(fp, "%20" ISC_PRINT_QUADFORMAT "u %s\n", val, typestr);
583                 break;
584         case statsformat_xml:
585 #ifdef HAVE_LIBXML2
586                 writer = dumparg->arg;
587
588                 TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "rdtype"));
589
590                 TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "name"));
591                 TRY0(xmlTextWriterWriteString(writer, ISC_XMLCHAR typestr));
592                 TRY0(xmlTextWriterEndElement(writer)); /* name */
593
594                 TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "counter"));
595                 TRY0(xmlTextWriterWriteFormatString(writer,
596                                                "%" ISC_PRINT_QUADFORMAT "u",
597                                                val));
598                 TRY0(xmlTextWriterEndElement(writer)); /* counter */
599
600                 TRY0(xmlTextWriterEndElement(writer)); /* rdtype */
601 #endif
602                 break;
603         }
604         return;
605 #ifdef HAVE_LIBXML2
606  error:
607         dumparg->result = ISC_R_FAILURE;
608         return;
609 #endif
610 }
611
612 static void
613 rdatasetstats_dump(dns_rdatastatstype_t type, isc_uint64_t val, void *arg) {
614         stats_dumparg_t *dumparg = arg;
615         FILE *fp;
616         char typebuf[64];
617         const char *typestr;
618         isc_boolean_t nxrrset = ISC_FALSE;
619 #ifdef HAVE_LIBXML2
620         xmlTextWriterPtr writer;
621         int xmlrc;
622 #endif
623
624         if ((DNS_RDATASTATSTYPE_ATTR(type) & DNS_RDATASTATSTYPE_ATTR_NXDOMAIN)
625             != 0) {
626                 typestr = "NXDOMAIN";
627         } else if ((DNS_RDATASTATSTYPE_ATTR(type) &
628                     DNS_RDATASTATSTYPE_ATTR_OTHERTYPE) != 0) {
629                 typestr = "Others";
630         } else {
631                 dns_rdatatype_format(DNS_RDATASTATSTYPE_BASE(type), typebuf,
632                                      sizeof(typebuf));
633                 typestr = typebuf;
634         }
635
636         if ((DNS_RDATASTATSTYPE_ATTR(type) & DNS_RDATASTATSTYPE_ATTR_NXRRSET)
637             != 0)
638                 nxrrset = ISC_TRUE;
639
640         switch (dumparg->type) {
641         case statsformat_file:
642                 fp = dumparg->arg;
643                 fprintf(fp, "%20" ISC_PRINT_QUADFORMAT "u %s%s\n", val,
644                         nxrrset ? "!" : "", typestr);
645                 break;
646         case statsformat_xml:
647 #ifdef HAVE_LIBXML2
648                 writer = dumparg->arg;
649
650                 TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "rrset"));
651                 TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "name"));
652                 TRY0(xmlTextWriterWriteFormatString(writer, "%s%s",
653                                                nxrrset ? "!" : "", typestr));
654                 TRY0(xmlTextWriterEndElement(writer)); /* name */
655
656                 TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "counter"));
657                 TRY0(xmlTextWriterWriteFormatString(writer,
658                                                "%" ISC_PRINT_QUADFORMAT "u",
659                                                val));
660                 TRY0(xmlTextWriterEndElement(writer)); /* counter */
661
662                 TRY0(xmlTextWriterEndElement(writer)); /* rrset */
663 #endif
664                 break;
665         }
666         return;
667 #ifdef HAVE_LIBXML2
668  error:
669         dumparg->result = ISC_R_FAILURE;
670 #endif
671
672 }
673
674 static void
675 opcodestat_dump(dns_opcode_t code, isc_uint64_t val, void *arg) {
676         FILE *fp;
677         isc_buffer_t b;
678         char codebuf[64];
679         stats_dumparg_t *dumparg = arg;
680 #ifdef HAVE_LIBXML2
681         xmlTextWriterPtr writer;
682         int xmlrc;
683 #endif
684
685         isc_buffer_init(&b, codebuf, sizeof(codebuf) - 1);
686         dns_opcode_totext(code, &b);
687         codebuf[isc_buffer_usedlength(&b)] = '\0';
688
689         switch (dumparg->type) {
690         case statsformat_file:
691                 fp = dumparg->arg;
692                 fprintf(fp, "%20" ISC_PRINT_QUADFORMAT "u %s\n", val, codebuf);
693                 break;
694         case statsformat_xml:
695 #ifdef HAVE_LIBXML2
696                 writer = dumparg->arg;
697
698                 TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "opcode"));
699
700                 TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "name"));
701                 TRY0(xmlTextWriterWriteString(writer, ISC_XMLCHAR codebuf));
702                 TRY0(xmlTextWriterEndElement(writer)); /* name */
703
704                 TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "counter"));
705                 TRY0(xmlTextWriterWriteFormatString(writer,
706                                                "%" ISC_PRINT_QUADFORMAT "u",
707                                                val));
708                 TRY0(xmlTextWriterEndElement(writer)); /* counter */
709
710                 TRY0(xmlTextWriterEndElement(writer)); /* opcode */
711 #endif
712                 break;
713         }
714         return;
715
716 #ifdef HAVE_LIBXML2
717  error:
718         dumparg->result = ISC_R_FAILURE;
719         return;
720 #endif
721 }
722
723 #ifdef HAVE_LIBXML2
724
725 /* XXXMLG below here sucks. */
726
727
728 static isc_result_t
729 zone_xmlrender(dns_zone_t *zone, void *arg) {
730         char buf[1024 + 32];    /* sufficiently large for zone name and class */
731         dns_rdataclass_t rdclass;
732         isc_uint32_t serial;
733         xmlTextWriterPtr writer = arg;
734         isc_stats_t *zonestats;
735         isc_uint64_t nsstat_values[dns_nsstatscounter_max];
736         int xmlrc;
737         isc_result_t result;
738
739         TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "zone"));
740
741         dns_zone_name(zone, buf, sizeof(buf));
742         TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "name"));
743         TRY0(xmlTextWriterWriteString(writer, ISC_XMLCHAR buf));
744         TRY0(xmlTextWriterEndElement(writer));
745
746         rdclass = dns_zone_getclass(zone);
747         dns_rdataclass_format(rdclass, buf, sizeof(buf));
748         TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "rdataclass"));
749         TRY0(xmlTextWriterWriteString(writer, ISC_XMLCHAR buf));
750         TRY0(xmlTextWriterEndElement(writer));
751
752         TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "serial"));
753         if (dns_zone_getserial2(zone, &serial) == ISC_R_SUCCESS)
754                 TRY0(xmlTextWriterWriteFormatString(writer, "%u", serial));
755         else
756                 TRY0(xmlTextWriterWriteString(writer, ISC_XMLCHAR "-"));
757         TRY0(xmlTextWriterEndElement(writer));
758
759         zonestats = dns_zone_getrequeststats(zone);
760         if (zonestats != NULL) {
761                 TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "counters"));
762                 result = dump_counters(zonestats, statsformat_xml, writer, NULL,
763                                       nsstats_xmldesc, dns_nsstatscounter_max,
764                                       nsstats_index, nsstat_values,
765                                       ISC_STATSDUMP_VERBOSE);
766                 if (result != ISC_R_SUCCESS)
767                         goto error;
768                 TRY0(xmlTextWriterEndElement(writer)); /* counters */
769         }
770
771         TRY0(xmlTextWriterEndElement(writer)); /* zone */
772
773         return (ISC_R_SUCCESS);
774  error:
775         return (ISC_R_FAILURE);
776 }
777
778 static isc_result_t
779 generatexml(ns_server_t *server, int *buflen, xmlChar **buf) {
780         char boottime[sizeof "yyyy-mm-ddThh:mm:ssZ"];
781         char nowstr[sizeof "yyyy-mm-ddThh:mm:ssZ"];
782         isc_time_t now;
783         xmlTextWriterPtr writer = NULL;
784         xmlDocPtr doc = NULL;
785         int xmlrc;
786         dns_view_t *view;
787         stats_dumparg_t dumparg;
788         dns_stats_t *cachestats;
789         isc_uint64_t nsstat_values[dns_nsstatscounter_max];
790         isc_uint64_t resstat_values[dns_resstatscounter_max];
791         isc_uint64_t zonestat_values[dns_zonestatscounter_max];
792         isc_uint64_t sockstat_values[isc_sockstatscounter_max];
793         isc_result_t result;
794
795         isc_time_now(&now);
796         isc_time_formatISO8601(&ns_g_boottime, boottime, sizeof boottime);
797         isc_time_formatISO8601(&now, nowstr, sizeof nowstr);
798
799         writer = xmlNewTextWriterDoc(&doc, 0);
800         if (writer == NULL)
801                 goto error;
802         TRY0(xmlTextWriterStartDocument(writer, NULL, "UTF-8", NULL));
803         TRY0(xmlTextWriterWritePI(writer, ISC_XMLCHAR "xml-stylesheet",
804                         ISC_XMLCHAR "type=\"text/xsl\" href=\"/bind9.xsl\""));
805         TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "isc"));
806         TRY0(xmlTextWriterWriteAttribute(writer, ISC_XMLCHAR "version",
807                                          ISC_XMLCHAR "1.0"));
808
809         TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "bind"));
810         TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "statistics"));
811         TRY0(xmlTextWriterWriteAttribute(writer, ISC_XMLCHAR "version",
812                                          ISC_XMLCHAR "2.2"));
813
814         /* Set common fields for statistics dump */
815         dumparg.type = statsformat_xml;
816         dumparg.arg = writer;
817
818         /*
819          * Start by rendering the views we know of here.  For each view we
820          * know of, call its rendering function.
821          */
822         view = ISC_LIST_HEAD(server->viewlist);
823         TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "views"));
824         while (view != NULL) {
825                 TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "view"));
826
827                 TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "name"));
828                 TRY0(xmlTextWriterWriteString(writer, ISC_XMLCHAR view->name));
829                 TRY0(xmlTextWriterEndElement(writer));
830
831                 TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "zones"));
832                 result = dns_zt_apply(view->zonetable, ISC_TRUE, zone_xmlrender,
833                                       writer);
834                 if (result != ISC_R_SUCCESS)
835                         goto error;
836                 TRY0(xmlTextWriterEndElement(writer));
837
838                 if (view->resquerystats != NULL) {
839                         dumparg.result = ISC_R_SUCCESS;
840                         dns_rdatatypestats_dump(view->resquerystats,
841                                                 rdtypestat_dump, &dumparg, 0);
842                         if (dumparg.result != ISC_R_SUCCESS)
843                                 goto error;
844                 }
845
846                 if (view->resstats != NULL) {
847                         result = dump_counters(view->resstats, statsformat_xml,
848                                                writer, "resstat",
849                                                resstats_xmldesc,
850                                                dns_resstatscounter_max,
851                                                resstats_index, resstat_values,
852                                                ISC_STATSDUMP_VERBOSE);
853                         if (result != ISC_R_SUCCESS)
854                                 goto error;
855                 }
856
857                 cachestats = dns_db_getrrsetstats(view->cachedb);
858                 if (cachestats != NULL) {
859                         TRY0(xmlTextWriterStartElement(writer,
860                                                        ISC_XMLCHAR "cache"));
861                         TRY0(xmlTextWriterWriteAttribute(writer,
862                                          ISC_XMLCHAR "name",
863                                          ISC_XMLCHAR
864                                          dns_cache_getname(view->cache)));
865                         dumparg.result = ISC_R_SUCCESS;
866                         dns_rdatasetstats_dump(cachestats, rdatasetstats_dump,
867                                                &dumparg, 0);
868                         if (dumparg.result != ISC_R_SUCCESS)
869                                 goto error;
870                         TRY0(xmlTextWriterEndElement(writer)); /* cache */
871                 }
872
873                 TRY0(xmlTextWriterEndElement(writer)); /* view */
874
875                 view = ISC_LIST_NEXT(view, link);
876         }
877         TRY0(xmlTextWriterEndElement(writer)); /* views */
878
879         TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "socketmgr"));
880         isc_socketmgr_renderxml(ns_g_socketmgr, writer);
881         TRY0(xmlTextWriterEndElement(writer)); /* socketmgr */
882
883         TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "taskmgr"));
884         isc_taskmgr_renderxml(ns_g_taskmgr, writer);
885         TRY0(xmlTextWriterEndElement(writer)); /* taskmgr */
886
887         TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "server"));
888         TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "boot-time"));
889         TRY0(xmlTextWriterWriteString(writer, ISC_XMLCHAR boottime));
890         TRY0(xmlTextWriterEndElement(writer));
891         TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "current-time"));
892         TRY0(xmlTextWriterWriteString(writer, ISC_XMLCHAR nowstr));
893         TRY0(xmlTextWriterEndElement(writer));
894
895         TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "requests"));
896         dumparg.result = ISC_R_SUCCESS;
897         dns_opcodestats_dump(server->opcodestats, opcodestat_dump, &dumparg,
898                              0);
899         if (dumparg.result != ISC_R_SUCCESS)
900                 goto error;
901         TRY0(xmlTextWriterEndElement(writer)); /* requests */
902
903         TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "queries-in"));
904         dumparg.result = ISC_R_SUCCESS;
905         dns_rdatatypestats_dump(server->rcvquerystats, rdtypestat_dump,
906                                 &dumparg, 0);
907         if (dumparg.result != ISC_R_SUCCESS)
908                 goto error;
909         TRY0(xmlTextWriterEndElement(writer)); /* queries-in */
910
911         result = dump_counters(server->nsstats, statsformat_xml, writer,
912                                "nsstat", nsstats_xmldesc,
913                                 dns_nsstatscounter_max,
914                                 nsstats_index, nsstat_values,
915                                 ISC_STATSDUMP_VERBOSE);
916         if (result != ISC_R_SUCCESS)
917                 goto error;
918
919         result = dump_counters(server->zonestats, statsformat_xml, writer,
920                                "zonestat", zonestats_xmldesc,
921                                dns_zonestatscounter_max, zonestats_index,
922                                zonestat_values, ISC_STATSDUMP_VERBOSE);
923         if (result != ISC_R_SUCCESS)
924                 goto error;
925
926         /*
927          * Most of the common resolver statistics entries are 0, so we don't
928          * use the verbose dump here.
929          */
930         result = dump_counters(server->resolverstats, statsformat_xml, writer,
931                                "resstat", resstats_xmldesc,
932                                dns_resstatscounter_max, resstats_index,
933                                resstat_values, 0);
934         if (result != ISC_R_SUCCESS)
935                 goto error;
936
937         result = dump_counters(server->sockstats, statsformat_xml, writer,
938                                "sockstat", sockstats_xmldesc,
939                                isc_sockstatscounter_max, sockstats_index,
940                                sockstat_values, ISC_STATSDUMP_VERBOSE);
941         if (result != ISC_R_SUCCESS)
942                 goto error;
943
944         TRY0(xmlTextWriterEndElement(writer)); /* server */
945
946         TRY0(xmlTextWriterStartElement(writer, ISC_XMLCHAR "memory"));
947         isc_mem_renderxml(writer);
948         TRY0(xmlTextWriterEndElement(writer)); /* memory */
949
950         TRY0(xmlTextWriterEndElement(writer)); /* statistics */
951         TRY0(xmlTextWriterEndElement(writer)); /* bind */
952         TRY0(xmlTextWriterEndElement(writer)); /* isc */
953
954         TRY0(xmlTextWriterEndDocument(writer));
955
956         xmlFreeTextWriter(writer);
957
958         xmlDocDumpFormatMemoryEnc(doc, buf, buflen, "UTF-8", 1);
959         xmlFreeDoc(doc);
960         return (ISC_R_SUCCESS);
961
962  error:
963         if (writer != NULL)
964                 xmlFreeTextWriter(writer);
965         if (doc != NULL)
966                 xmlFreeDoc(doc);
967         return (ISC_R_FAILURE);
968 }
969
970 static void
971 wrap_xmlfree(isc_buffer_t *buffer, void *arg) {
972         UNUSED(arg);
973
974         xmlFree(isc_buffer_base(buffer));
975 }
976
977 static isc_result_t
978 render_index(const char *url, const char *querystring, void *arg,
979              unsigned int *retcode, const char **retmsg, const char **mimetype,
980              isc_buffer_t *b, isc_httpdfree_t **freecb,
981              void **freecb_args)
982 {
983         unsigned char *msg;
984         int msglen;
985         ns_server_t *server = arg;
986         isc_result_t result;
987
988         UNUSED(url);
989         UNUSED(querystring);
990
991         result = generatexml(server, &msglen, &msg);
992
993         if (result == ISC_R_SUCCESS) {
994                 *retcode = 200;
995                 *retmsg = "OK";
996                 *mimetype = "text/xml";
997                 isc_buffer_reinit(b, msg, msglen);
998                 isc_buffer_add(b, msglen);
999                 *freecb = wrap_xmlfree;
1000                 *freecb_args = NULL;
1001         }
1002
1003         return (result);
1004 }
1005
1006 #endif  /* HAVE_LIBXML2 */
1007
1008 static isc_result_t
1009 render_xsl(const char *url, const char *querystring, void *args,
1010            unsigned int *retcode, const char **retmsg, const char **mimetype,
1011            isc_buffer_t *b, isc_httpdfree_t **freecb,
1012            void **freecb_args)
1013 {
1014         UNUSED(url);
1015         UNUSED(querystring);
1016         UNUSED(args);
1017
1018         *retcode = 200;
1019         *retmsg = "OK";
1020         *mimetype = "text/xslt+xml";
1021         isc_buffer_reinit(b, xslmsg, strlen(xslmsg));
1022         isc_buffer_add(b, strlen(xslmsg));
1023         *freecb = NULL;
1024         *freecb_args = NULL;
1025
1026         return (ISC_R_SUCCESS);
1027 }
1028
1029 static void
1030 shutdown_listener(ns_statschannel_t *listener) {
1031         char socktext[ISC_SOCKADDR_FORMATSIZE];
1032         isc_sockaddr_format(&listener->address, socktext, sizeof(socktext));
1033         isc_log_write(ns_g_lctx, NS_LOGCATEGORY_GENERAL,NS_LOGMODULE_SERVER,
1034                       ISC_LOG_NOTICE, "stopping statistics channel on %s",
1035                       socktext);
1036
1037         isc_httpdmgr_shutdown(&listener->httpdmgr);
1038 }
1039
1040 static isc_boolean_t
1041 client_ok(const isc_sockaddr_t *fromaddr, void *arg) {
1042         ns_statschannel_t *listener = arg;
1043         isc_netaddr_t netaddr;
1044         char socktext[ISC_SOCKADDR_FORMATSIZE];
1045         int match;
1046
1047         REQUIRE(listener != NULL);
1048
1049         isc_netaddr_fromsockaddr(&netaddr, fromaddr);
1050
1051         LOCK(&listener->lock);
1052         if (dns_acl_match(&netaddr, NULL, listener->acl, &ns_g_server->aclenv,
1053                           &match, NULL) == ISC_R_SUCCESS && match > 0) {
1054                 UNLOCK(&listener->lock);
1055                 return (ISC_TRUE);
1056         }
1057         UNLOCK(&listener->lock);
1058
1059         isc_sockaddr_format(fromaddr, socktext, sizeof(socktext));
1060         isc_log_write(ns_g_lctx, NS_LOGCATEGORY_GENERAL,
1061                       NS_LOGMODULE_SERVER, ISC_LOG_WARNING,
1062                       "rejected statistics connection from %s", socktext);
1063
1064         return (ISC_FALSE);
1065 }
1066
1067 static void
1068 destroy_listener(void *arg) {
1069         ns_statschannel_t *listener = arg;
1070
1071         REQUIRE(listener != NULL);
1072         REQUIRE(!ISC_LINK_LINKED(listener, link));
1073
1074         /* We don't have to acquire the lock here since it's already unlinked */
1075         dns_acl_detach(&listener->acl);
1076
1077         DESTROYLOCK(&listener->lock);
1078         isc_mem_putanddetach(&listener->mctx, listener, sizeof(*listener));
1079 }
1080
1081 static isc_result_t
1082 add_listener(ns_server_t *server, ns_statschannel_t **listenerp,
1083              const cfg_obj_t *listen_params, const cfg_obj_t *config,
1084              isc_sockaddr_t *addr, cfg_aclconfctx_t *aclconfctx,
1085              const char *socktext)
1086 {
1087         isc_result_t result;
1088         ns_statschannel_t *listener;
1089         isc_task_t *task = NULL;
1090         isc_socket_t *sock = NULL;
1091         const cfg_obj_t *allow;
1092         dns_acl_t *new_acl = NULL;
1093
1094         listener = isc_mem_get(server->mctx, sizeof(*listener));
1095         if (listener == NULL)
1096                 return (ISC_R_NOMEMORY);
1097
1098         listener->httpdmgr = NULL;
1099         listener->address = *addr;
1100         listener->acl = NULL;
1101         listener->mctx = NULL;
1102         ISC_LINK_INIT(listener, link);
1103
1104         result = isc_mutex_init(&listener->lock);
1105         if (result != ISC_R_SUCCESS) {
1106                 isc_mem_put(server->mctx, listener, sizeof(*listener));
1107                 return (ISC_R_FAILURE);
1108         }
1109
1110         isc_mem_attach(server->mctx, &listener->mctx);
1111
1112         allow = cfg_tuple_get(listen_params, "allow");
1113         if (allow != NULL && cfg_obj_islist(allow)) {
1114                 result = cfg_acl_fromconfig(allow, config, ns_g_lctx,
1115                                             aclconfctx, listener->mctx, 0,
1116                                             &new_acl);
1117         } else
1118                 result = dns_acl_any(listener->mctx, &new_acl);
1119         if (result != ISC_R_SUCCESS)
1120                 goto cleanup;
1121         dns_acl_attach(new_acl, &listener->acl);
1122         dns_acl_detach(&new_acl);
1123
1124         result = isc_task_create(ns_g_taskmgr, 0, &task);
1125         if (result != ISC_R_SUCCESS)
1126                 goto cleanup;
1127         isc_task_setname(task, "statchannel", NULL);
1128
1129         result = isc_socket_create(ns_g_socketmgr, isc_sockaddr_pf(addr),
1130                                    isc_sockettype_tcp, &sock);
1131         if (result != ISC_R_SUCCESS)
1132                 goto cleanup;
1133         isc_socket_setname(sock, "statchannel", NULL);
1134
1135 #ifndef ISC_ALLOW_MAPPED
1136         isc_socket_ipv6only(sock, ISC_TRUE);
1137 #endif
1138
1139         result = isc_socket_bind(sock, addr, ISC_SOCKET_REUSEADDRESS);
1140         if (result != ISC_R_SUCCESS)
1141                 goto cleanup;
1142
1143         result = isc_httpdmgr_create(server->mctx, sock, task, client_ok,
1144                                      destroy_listener, listener, ns_g_timermgr,
1145                                      &listener->httpdmgr);
1146         if (result != ISC_R_SUCCESS)
1147                 goto cleanup;
1148
1149 #ifdef HAVE_LIBXML2
1150         isc_httpdmgr_addurl(listener->httpdmgr, "/", render_index, server);
1151 #endif
1152         isc_httpdmgr_addurl(listener->httpdmgr, "/bind9.xsl", render_xsl,
1153                             server);
1154
1155         *listenerp = listener;
1156         isc_log_write(ns_g_lctx, NS_LOGCATEGORY_GENERAL,
1157                       NS_LOGMODULE_SERVER, ISC_LOG_NOTICE,
1158                       "statistics channel listening on %s", socktext);
1159
1160 cleanup:
1161         if (result != ISC_R_SUCCESS) {
1162                 if (listener->acl != NULL)
1163                         dns_acl_detach(&listener->acl);
1164                 DESTROYLOCK(&listener->lock);
1165                 isc_mem_putanddetach(&listener->mctx, listener,
1166                                      sizeof(*listener));
1167         }
1168         if (task != NULL)
1169                 isc_task_detach(&task);
1170         if (sock != NULL)
1171                 isc_socket_detach(&sock);
1172
1173         return (result);
1174 }
1175
1176 static void
1177 update_listener(ns_server_t *server, ns_statschannel_t **listenerp,
1178                 const cfg_obj_t *listen_params, const cfg_obj_t *config,
1179                 isc_sockaddr_t *addr, cfg_aclconfctx_t *aclconfctx,
1180                 const char *socktext)
1181 {
1182         ns_statschannel_t *listener;
1183         const cfg_obj_t *allow = NULL;
1184         dns_acl_t *new_acl = NULL;
1185         isc_result_t result = ISC_R_SUCCESS;
1186
1187         for (listener = ISC_LIST_HEAD(server->statschannels);
1188              listener != NULL;
1189              listener = ISC_LIST_NEXT(listener, link))
1190                 if (isc_sockaddr_equal(addr, &listener->address))
1191                         break;
1192
1193         if (listener == NULL) {
1194                 *listenerp = NULL;
1195                 return;
1196         }
1197
1198         /*
1199          * Now, keep the old access list unless a new one can be made.
1200          */
1201         allow = cfg_tuple_get(listen_params, "allow");
1202         if (allow != NULL && cfg_obj_islist(allow)) {
1203                 result = cfg_acl_fromconfig(allow, config, ns_g_lctx,
1204                                             aclconfctx, listener->mctx, 0,
1205                                             &new_acl);
1206         } else
1207                 result = dns_acl_any(listener->mctx, &new_acl);
1208
1209         if (result == ISC_R_SUCCESS) {
1210                 LOCK(&listener->lock);
1211
1212                 dns_acl_detach(&listener->acl);
1213                 dns_acl_attach(new_acl, &listener->acl);
1214                 dns_acl_detach(&new_acl);
1215
1216                 UNLOCK(&listener->lock);
1217         } else {
1218                 cfg_obj_log(listen_params, ns_g_lctx, ISC_LOG_WARNING,
1219                             "couldn't install new acl for "
1220                             "statistics channel %s: %s",
1221                             socktext, isc_result_totext(result));
1222         }
1223
1224         *listenerp = listener;
1225 }
1226
1227 isc_result_t
1228 ns_statschannels_configure(ns_server_t *server, const cfg_obj_t *config,
1229                          cfg_aclconfctx_t *aclconfctx)
1230 {
1231         ns_statschannel_t *listener, *listener_next;
1232         ns_statschannellist_t new_listeners;
1233         const cfg_obj_t *statschannellist = NULL;
1234         const cfg_listelt_t *element, *element2;
1235         char socktext[ISC_SOCKADDR_FORMATSIZE];
1236
1237         RUNTIME_CHECK(isc_once_do(&once, init_desc) == ISC_R_SUCCESS);
1238
1239         ISC_LIST_INIT(new_listeners);
1240
1241         /*
1242          * Get the list of named.conf 'statistics-channels' statements.
1243          */
1244         (void)cfg_map_get(config, "statistics-channels", &statschannellist);
1245
1246         /*
1247          * Run through the new address/port list, noting sockets that are
1248          * already being listened on and moving them to the new list.
1249          *
1250          * Identifying duplicate addr/port combinations is left to either
1251          * the underlying config code, or to the bind attempt getting an
1252          * address-in-use error.
1253          */
1254         if (statschannellist != NULL) {
1255 #ifndef HAVE_LIBXML2
1256                 isc_log_write(ns_g_lctx, NS_LOGCATEGORY_GENERAL,
1257                               NS_LOGMODULE_SERVER, ISC_LOG_WARNING,
1258                               "statistics-channels specified but not effective "
1259                               "due to missing XML library");
1260 #endif
1261
1262                 for (element = cfg_list_first(statschannellist);
1263                      element != NULL;
1264                      element = cfg_list_next(element)) {
1265                         const cfg_obj_t *statschannel;
1266                         const cfg_obj_t *listenercfg = NULL;
1267
1268                         statschannel = cfg_listelt_value(element);
1269                         (void)cfg_map_get(statschannel, "inet",
1270                                           &listenercfg);
1271                         if (listenercfg == NULL)
1272                                 continue;
1273
1274                         for (element2 = cfg_list_first(listenercfg);
1275                              element2 != NULL;
1276                              element2 = cfg_list_next(element2)) {
1277                                 const cfg_obj_t *listen_params;
1278                                 const cfg_obj_t *obj;
1279                                 isc_sockaddr_t addr;
1280
1281                                 listen_params = cfg_listelt_value(element2);
1282
1283                                 obj = cfg_tuple_get(listen_params, "address");
1284                                 addr = *cfg_obj_assockaddr(obj);
1285                                 if (isc_sockaddr_getport(&addr) == 0)
1286                                         isc_sockaddr_setport(&addr, NS_STATSCHANNEL_HTTPPORT);
1287
1288                                 isc_sockaddr_format(&addr, socktext,
1289                                                     sizeof(socktext));
1290
1291                                 isc_log_write(ns_g_lctx,
1292                                               NS_LOGCATEGORY_GENERAL,
1293                                               NS_LOGMODULE_SERVER,
1294                                               ISC_LOG_DEBUG(9),
1295                                               "processing statistics "
1296                                               "channel %s",
1297                                               socktext);
1298
1299                                 update_listener(server, &listener,
1300                                                 listen_params, config, &addr,
1301                                                 aclconfctx, socktext);
1302
1303                                 if (listener != NULL) {
1304                                         /*
1305                                          * Remove the listener from the old
1306                                          * list, so it won't be shut down.
1307                                          */
1308                                         ISC_LIST_UNLINK(server->statschannels,
1309                                                         listener, link);
1310                                 } else {
1311                                         /*
1312                                          * This is a new listener.
1313                                          */
1314                                         isc_result_t r;
1315
1316                                         r = add_listener(server, &listener,
1317                                                          listen_params, config,
1318                                                          &addr, aclconfctx,
1319                                                          socktext);
1320                                         if (r != ISC_R_SUCCESS) {
1321                                                 cfg_obj_log(listen_params,
1322                                                             ns_g_lctx,
1323                                                             ISC_LOG_WARNING,
1324                                                             "couldn't allocate "
1325                                                             "statistics channel"
1326                                                             " %s: %s",
1327                                                             socktext,
1328                                                             isc_result_totext(r));
1329                                         }
1330                                 }
1331
1332                                 if (listener != NULL)
1333                                         ISC_LIST_APPEND(new_listeners, listener,
1334                                                         link);
1335                         }
1336                 }
1337         }
1338
1339         for (listener = ISC_LIST_HEAD(server->statschannels);
1340              listener != NULL;
1341              listener = listener_next) {
1342                 listener_next = ISC_LIST_NEXT(listener, link);
1343                 ISC_LIST_UNLINK(server->statschannels, listener, link);
1344                 shutdown_listener(listener);
1345         }
1346
1347         ISC_LIST_APPENDLIST(server->statschannels, new_listeners, link);
1348         return (ISC_R_SUCCESS);
1349 }
1350
1351 void
1352 ns_statschannels_shutdown(ns_server_t *server) {
1353         ns_statschannel_t *listener;
1354
1355         while ((listener = ISC_LIST_HEAD(server->statschannels)) != NULL) {
1356                 ISC_LIST_UNLINK(server->statschannels, listener, link);
1357                 shutdown_listener(listener);
1358         }
1359 }
1360
1361 isc_result_t
1362 ns_stats_dump(ns_server_t *server, FILE *fp) {
1363         isc_stdtime_t now;
1364         isc_result_t result;
1365         dns_view_t *view;
1366         dns_zone_t *zone, *next;
1367         stats_dumparg_t dumparg;
1368         isc_uint64_t nsstat_values[dns_nsstatscounter_max];
1369         isc_uint64_t resstat_values[dns_resstatscounter_max];
1370         isc_uint64_t zonestat_values[dns_zonestatscounter_max];
1371         isc_uint64_t sockstat_values[isc_sockstatscounter_max];
1372
1373         RUNTIME_CHECK(isc_once_do(&once, init_desc) == ISC_R_SUCCESS);
1374
1375         /* Set common fields */
1376         dumparg.type = statsformat_file;
1377         dumparg.arg = fp;
1378
1379         isc_stdtime_get(&now);
1380         fprintf(fp, "+++ Statistics Dump +++ (%lu)\n", (unsigned long)now);
1381
1382         fprintf(fp, "++ Incoming Requests ++\n");
1383         dns_opcodestats_dump(server->opcodestats, opcodestat_dump, &dumparg, 0);
1384
1385         fprintf(fp, "++ Incoming Queries ++\n");
1386         dns_rdatatypestats_dump(server->rcvquerystats, rdtypestat_dump,
1387                                 &dumparg, 0);
1388
1389         fprintf(fp, "++ Outgoing Queries ++\n");
1390         for (view = ISC_LIST_HEAD(server->viewlist);
1391              view != NULL;
1392              view = ISC_LIST_NEXT(view, link)) {
1393                 if (view->resquerystats == NULL)
1394                         continue;
1395                 if (strcmp(view->name, "_default") == 0)
1396                         fprintf(fp, "[View: default]\n");
1397                 else
1398                         fprintf(fp, "[View: %s]\n", view->name);
1399                 dns_rdatatypestats_dump(view->resquerystats, rdtypestat_dump,
1400                                         &dumparg, 0);
1401         }
1402
1403         fprintf(fp, "++ Name Server Statistics ++\n");
1404         (void) dump_counters(server->nsstats, statsformat_file, fp, NULL,
1405                              nsstats_desc, dns_nsstatscounter_max,
1406                              nsstats_index, nsstat_values, 0);
1407
1408         fprintf(fp, "++ Zone Maintenance Statistics ++\n");
1409         (void) dump_counters(server->zonestats, statsformat_file, fp, NULL,
1410                              zonestats_desc, dns_zonestatscounter_max,
1411                              zonestats_index, zonestat_values, 0);
1412
1413         fprintf(fp, "++ Resolver Statistics ++\n");
1414         fprintf(fp, "[Common]\n");
1415         (void) dump_counters(server->resolverstats, statsformat_file, fp, NULL,
1416                              resstats_desc, dns_resstatscounter_max,
1417                              resstats_index, resstat_values, 0);
1418         for (view = ISC_LIST_HEAD(server->viewlist);
1419              view != NULL;
1420              view = ISC_LIST_NEXT(view, link)) {
1421                 if (view->resstats == NULL)
1422                         continue;
1423                 if (strcmp(view->name, "_default") == 0)
1424                         fprintf(fp, "[View: default]\n");
1425                 else
1426                         fprintf(fp, "[View: %s]\n", view->name);
1427                 (void) dump_counters(view->resstats, statsformat_file, fp, NULL,
1428                                      resstats_desc, dns_resstatscounter_max,
1429                                      resstats_index, resstat_values, 0);
1430         }
1431
1432         fprintf(fp, "++ Cache DB RRsets ++\n");
1433         for (view = ISC_LIST_HEAD(server->viewlist);
1434              view != NULL;
1435              view = ISC_LIST_NEXT(view, link)) {
1436                 dns_stats_t *cachestats;
1437
1438                 cachestats = dns_db_getrrsetstats(view->cachedb);
1439                 if (cachestats == NULL)
1440                         continue;
1441                 if (strcmp(view->name, "_default") == 0)
1442                         fprintf(fp, "[View: default]\n");
1443                 else
1444                         fprintf(fp, "[View: %s (Cache: %s)]\n", view->name,
1445                                 dns_cache_getname(view->cache));
1446                 if (dns_view_iscacheshared(view)) {
1447                         /*
1448                          * Avoid dumping redundant statistics when the cache is
1449                          * shared.
1450                          */
1451                         continue;
1452                 }
1453                 dns_rdatasetstats_dump(cachestats, rdatasetstats_dump, &dumparg,
1454                                        0);
1455         }
1456
1457         fprintf(fp, "++ Socket I/O Statistics ++\n");
1458         (void) dump_counters(server->sockstats, statsformat_file, fp, NULL,
1459                              sockstats_desc, isc_sockstatscounter_max,
1460                              sockstats_index, sockstat_values, 0);
1461
1462         fprintf(fp, "++ Per Zone Query Statistics ++\n");
1463         zone = NULL;
1464         for (result = dns_zone_first(server->zonemgr, &zone);
1465              result == ISC_R_SUCCESS;
1466              next = NULL, result = dns_zone_next(zone, &next), zone = next)
1467         {
1468                 isc_stats_t *zonestats = dns_zone_getrequeststats(zone);
1469                 if (zonestats != NULL) {
1470                         char zonename[DNS_NAME_FORMATSIZE];
1471
1472                         dns_name_format(dns_zone_getorigin(zone),
1473                                         zonename, sizeof(zonename));
1474                         view = dns_zone_getview(zone);
1475
1476                         fprintf(fp, "[%s", zonename);
1477                         if (strcmp(view->name, "_default") != 0)
1478                                 fprintf(fp, " (view: %s)", view->name);
1479                         fprintf(fp, "]\n");
1480
1481                         (void) dump_counters(zonestats, statsformat_file, fp,
1482                                              NULL, nsstats_desc,
1483                                              dns_nsstatscounter_max,
1484                                              nsstats_index, nsstat_values, 0);
1485                 }
1486         }
1487
1488         fprintf(fp, "--- Statistics Dump --- (%lu)\n", (unsigned long)now);
1489
1490         return (ISC_R_SUCCESS); /* this function currently always succeeds */
1491 }