Add some #include's to fix 'implicit declaration of...' warnings.
[dragonfly.git] / usr.sbin / mrouted / testrsrr / testrsrr.c
1 /*
2  * Copyright 1995 Massachusetts Institute of Technology
3  *
4  * Permission to use, copy, modify, and distribute this software and
5  * its documentation for any purpose and without fee is hereby
6  * granted, provided that both the above copyright notice and this
7  * permission notice appear in all copies, that both the above
8  * copyright notice and this permission notice appear in all
9  * supporting documentation, and that the name of M.I.T. not be used
10  * in advertising or publicity pertaining to distribution of the
11  * software without specific, written prior permission.  M.I.T. makes
12  * no representations about the suitability of this software for any
13  * purpose.  It is provided "as is" without express or implied
14  * warranty.
15  * 
16  * THIS SOFTWARE IS PROVIDED BY M.I.T. ``AS IS''.  M.I.T. DISCLAIMS
17  * ALL EXPRESS OR IMPLIED WARRANTIES WITH REGARD TO THIS SOFTWARE,
18  * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
19  * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE. IN NO EVENT
20  * SHALL M.I.T. BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
21  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
22  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
23  * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
24  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
25  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
26  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27  * SUCH DAMAGE.
28  *
29  * $FreeBSD: src/usr.sbin/mrouted/testrsrr/testrsrr.c,v 1.4 1999/08/28 01:17:17 peter Exp $
30  * $DragonFly: src/usr.sbin/mrouted/testrsrr/testrsrr.c,v 1.5 2007/11/25 01:28:24 swildner Exp $
31  */
32 #include <stddef.h>
33 #include <stdlib.h>
34 #include <stdio.h>
35 #include <string.h>
36 #include <err.h>
37 #include <sysexits.h>
38
39 #include <sys/param.h>
40 #include <sys/types.h>
41 #include <unistd.h>
42 #include <sys/socket.h>
43 #include <sys/un.h>
44
45 #include <netinet/in.h>
46 #include <arpa/inet.h>
47
48 #include "rsrr.h"
49
50 char sunpath[MAXPATHLEN];
51 int s;
52
53 void
54 exitfn(void)
55 {
56         close(s);
57         unlink(sunpath);
58 }
59
60 int 
61 main(void)
62 {
63         struct sockaddr_un sun;
64         char buf[RSRR_MAX_LEN];
65         struct rsrr_header *rh;
66         struct rsrr_vif *rvp;
67         int i;
68
69         s = socket(PF_LOCAL, SOCK_DGRAM, 0);
70         if (s < 0) {
71                 err(EX_OSERR, "socket(PF_LOCAL, SOCK_DGRAM, 0)");
72         }
73
74         sun.sun_family = AF_LOCAL;
75         snprintf(sunpath, sizeof sun.sun_path, "/tmp/testrsrr.%lu", 
76                  (unsigned long)getpid());
77         strcpy(sun.sun_path, sunpath);
78         sun.sun_len = (offsetof(struct sockaddr_un, sun_path) 
79                        + strlen(sunpath));
80
81         if (bind(s, (struct sockaddr *)&sun, sun.sun_len) < 0) {
82                 err(EX_OSERR, "bind: %s", sunpath);
83         }
84
85         atexit(exitfn);         /* clean up if we exit on error */
86
87         strcpy(sun.sun_path, RSRR_SERV_PATH);
88         sun.sun_len = (offsetof(struct sockaddr_un, sun_path) 
89                        + strlen(sunpath));
90
91         if (connect(s, (struct sockaddr *)&sun, sun.sun_len) < 0) {
92                 err(EX_OSERR, "connect: %s", RSRR_SERV_PATH);
93         }
94
95         rh = (struct rsrr_header *)buf;
96         rh->version = RSRR_MAX_VERSION;
97         rh->type = RSRR_INITIAL_QUERY;
98         rh->flags = 0;
99         rh->num = 0;
100
101         if (write(s, rh, sizeof *rh) == (ssize_t)-1) {
102                 err(EX_OSERR, "write(initial query)");
103         }
104
105         if (read(s, buf, sizeof buf) == (ssize_t)-1) {
106                 err(EX_OSERR, "read(initial reply)");
107         }
108
109         if (rh->version != RSRR_MAX_VERSION) {
110                 errx(EX_PROTOCOL, "bad remote version %d", rh->version);
111         }
112
113         if (rh->type != RSRR_INITIAL_REPLY) {
114                 errx(EX_PROTOCOL, "remote returned unexpected message type %d",
115                      rh->type);
116         }
117
118         if (rh->flags) {
119                 printf("confusing flags: %d\n", rh->flags);
120         }
121
122         printf("There are %d vifs configured:\n", rh->num);
123
124         printf(" Vif  Thresh  Status  Local address\n");
125         for(i = 0, rvp = (struct rsrr_vif *)(rh + 1); i < rh->num; i++,rvp++) {
126                 printf(" %3d  %6d  %6d  %s\n", rvp->id, rvp->threshold,
127                        rvp->status, inet_ntoa(rvp->local_addr));
128         }
129         exit(0);
130 }