usr.sbin/traceroute/as.c

   1 /*      $NetBSD: as.c,v 1.1 2001/11/04 23:14:36 atatat Exp $    */
   2
   3 /*
   4  * Copyright (c) 2001 The NetBSD Foundation, Inc.
   5  * All rights reserved.
   6  *
   7  * This code is derived from software contributed to The NetBSD Foundation
   8  * by Andrew Brown.
   9  *
  10  * Redistribution and use in source and binary forms, with or without
  11  * modification, are permitted provided that the following conditions
  12  * are met:
  13  * 1. Redistributions of source code must retain the above copyright
  14  *    notice, this list of conditions and the following disclaimer.
  15  * 2. Redistributions in binary form must reproduce the above copyright
  16  *    notice, this list of conditions and the following disclaimer in the
  17  *    documentation and/or other materials provided with the distribution.
  18  *
  19  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
  20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
  21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
  23  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  29  * POSSIBILITY OF SUCH DAMAGE.
  30  */
  31
  32 #include <sys/cdefs.h>
  33 #include <sys/types.h>
  34 #include <sys/socket.h>
  35 #include <netinet/in.h>
  36 #include <arpa/inet.h>
  37 #include <netdb.h>
  38 #include <unistd.h>
  39 #include <string.h>
  40 #include <stdlib.h>
  41 #include <errno.h>
  42 #include <err.h>
  43 #include <stdio.h>
  44
  45 #include "as.h"
  46
  47 #define DEFAULT_AS_SERVER "whois.radb.net"
  48 #undef AS_DEBUG_FILE
  49
  50 struct aslookup {
  51         FILE *as_f;
  52 #ifdef AS_DEBUG_FILE
  53         FILE *as_debug;
  54 #endif /* AS_DEBUG_FILE */
  55 };
  56
  57 void *
  58 as_setup(server)
  59         char *server;
  60 {
  61         struct aslookup *asn;
  62         struct hostent *he = NULL;
  63         struct servent *se;
  64         struct sockaddr_in in;
  65         FILE *f;
  66         int s;
  67
  68         if (server == NULL)
  69                 server = DEFAULT_AS_SERVER;
  70
  71         (void)memset(&in, 0, sizeof(in));
  72         in.sin_family = AF_INET;
  73         in.sin_len = sizeof(in);
  74         if ((se = getservbyname("whois", "tcp")) == NULL) {
  75                 warnx("warning: whois/tcp service not found");
  76                 in.sin_port = ntohs(43);
  77         } else
  78                 in.sin_port = se->s_port;
  79
  80         if (inet_aton(server, &in.sin_addr) == 0 &&
  81             ((he = gethostbyname(server)) == NULL ||
  82             he->h_addr == NULL)) {
  83                 warnx("%s: %s", server, hstrerror(h_errno));
  84                 return (NULL);
  85         }
  86
  87         if ((s = socket(PF_INET, SOCK_STREAM, 0)) == -1) {
  88                 warn("socket");
  89                 return (NULL);
  90         }
  91
  92         do {
  93                 if (he != NULL) {
  94                         memcpy(&in.sin_addr, he->h_addr, he->h_length);
  95                         he->h_addr_list++;
  96                 }
  97                 if (connect(s, (struct sockaddr *)&in, sizeof(in)) == 0)
  98                         break;
  99                 if (he == NULL || he->h_addr == NULL) {
 100                         close(s);
 101                         s = -1;
 102                         break;
 103                 }
 104         } while (1);
 105
 106         if (s == -1) {
 107                 warn("connect");
 108                 return (NULL);
 109         }
 110
 111         f = fdopen(s, "r+");
 112         (void)fprintf(f, "!!\n");
 113         (void)fflush(f);
 114
 115         asn = malloc(sizeof(struct aslookup));
 116         if (asn == NULL)
 117                 (void)fclose(f);
 118         else
 119                 asn->as_f = f;
 120
 121 #ifdef AS_DEBUG_FILE
 122         asn->as_debug = fopen(AS_DEBUG_FILE, "w");
 123         if (asn->as_debug) {
 124                 (void)fprintf(asn->as_debug, ">> !!\n");
 125                 (void)fflush(asn->as_debug);
 126         }
 127 #endif /* AS_DEBUG_FILE */
 128
 129         return (asn);
 130 }
 131
 132 int
 133 as_lookup(_asn, addr)
 134         void *_asn;
 135         struct in_addr *addr;
 136 {
 137         struct aslookup *asn = _asn;
 138         char buf[1024];
 139         int as, rc, dlen;
 140
 141         as = rc = dlen = 0;
 142         (void)fprintf(asn->as_f, "!r%s/32,l\n", inet_ntoa(*addr));
 143         (void)fflush(asn->as_f);
 144
 145 #ifdef AS_DEBUG_FILE
 146         if (asn->as_debug) {
 147                 (void)fprintf(asn->as_debug, ">> !r%s/32,l\n",
 148                      inet_ntoa(*addr));
 149                 (void)fflush(asn->as_debug);
 150         }
 151 #endif /* AS_DEBUG_FILE */
 152
 153         while (fgets(buf, sizeof(buf), asn->as_f) != NULL) {
 154                 buf[sizeof(buf) - 1] = '\0';
 155
 156 #ifdef AS_DEBUG_FILE
 157                 if (asn->as_debug) {
 158                         (void)fprintf(asn->as_debug, "<< %s", buf);
 159                         (void)fflush(asn->as_debug);
 160                 }
 161 #endif /* AS_DEBUG_FILE */
 162
 163                 if (rc == 0) {
 164                         rc = buf[0];
 165                         switch (rc) {
 166                             case 'A':
 167                                 /* A - followed by # bytes of answer */
 168                                 sscanf(buf, "A%d\n", &dlen);
 169 #ifdef AS_DEBUG_FILE
 170                                 if (asn->as_debug) {
 171                                         (void)fprintf(asn->as_debug,
 172                                              "dlen: %d\n", dlen);
 173                                         (void)fflush(asn->as_debug);
 174                                 }
 175 #endif /* AS_DEBUG_FILE */
 176                                 break;
 177                             case 'C':
 178                             case 'D':
 179                             case 'E':
 180                             case 'F':
 181                                 /* C - no data returned */
 182                                 /* D - key not found */
 183                                 /* E - multiple copies of key */
 184                                 /* F - some other error */
 185                                 break;
 186                         }
 187                         if (rc == 'A')
 188                                 /* skip to next input line */
 189                                 continue;
 190                 }
 191
 192                 if (dlen == 0)
 193                         /* out of data, next char read is end code */
 194                         rc = buf[0];
 195                 if (rc != 'A')
 196                         /* either an error off the bat, or a done code */
 197                         break;
 198
 199                 /* data received, thank you */
 200                 dlen -= strlen(buf);
 201
 202                 /* origin line is the interesting bit */
 203                 if (as == 0 && strncasecmp(buf, "origin:", 7) == 0) {
 204                         sscanf(buf + 7, " AS%d", &as);
 205 #ifdef AS_DEBUG_FILE
 206                         if (asn->as_debug) {
 207                                 (void)fprintf(asn->as_debug, "as: %d\n", as);
 208                                 (void)fflush(asn->as_debug);
 209                         }
 210 #endif /* AS_DEBUG_FILE */
 211                 }
 212         }
 213
 214         return (as);
 215 }
 216
 217 void
 218 as_shutdown(_asn)
 219         void *_asn;
 220 {
 221         struct aslookup *asn = _asn;
 222
 223         (void)fprintf(asn->as_f, "!q\n");
 224         (void)fclose(asn->as_f);
 225
 226 #ifdef AS_DEBUG_FILE
 227         if (asn->as_debug) {
 228                 (void)fprintf(asn->as_debug, ">> !q\n");
 229                 (void)fclose(asn->as_debug);
 230         }
 231 #endif /* AS_DEBUG_FILE */
 232
 233         free(asn);
 234 }