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 50struct aslookup { 51 FILE *as_f; 52#ifdef AS_DEBUG_FILE 53 FILE *as_debug; 54#endif /* AS_DEBUG_FILE */ 55}; 56 57void * 58as_setup(const char *server) 59{ 60 struct aslookup *asn; 61 struct addrinfo hints, *res0, *res; 62 FILE *f; 63 int s, error; 64 65 s = -1; 66 if (server == NULL) 67 server = getenv("RA_SERVER"); 68 if (server == NULL) 69 server = DEFAULT_AS_SERVER; 70 71 memset(&hints, 0, sizeof(hints)); 72 hints.ai_family = PF_UNSPEC; 73 hints.ai_socktype = SOCK_STREAM; 74 error = getaddrinfo(server, "whois", &hints, &res0); 75 if (error == EAI_SERVICE) { 76 warnx("warning: whois/tcp service not found"); 77 error = getaddrinfo(server, "43", &hints, &res0); 78 } 79 if (error != 0) { 80 warnx("%s: %s", server, gai_strerror(error)); 81 return (NULL); 82 } 83 84 for (res = res0; res; res = res->ai_next) { 85 s = socket(res->ai_family, res->ai_socktype, res->ai_protocol); 86 if (s < 0) 87 continue; 88 if (connect(s, res->ai_addr, res->ai_addrlen) >= 0) 89 break; 90 close(s); 91 s = -1; 92 } 93 freeaddrinfo(res0); 94 if (s < 0) { 95 warn("connect"); 96 return (NULL); 97 } 98 99 f = fdopen(s, "r+"); 100 (void)fprintf(f, "!!\n"); 101 (void)fflush(f); 102 103 asn = malloc(sizeof(struct aslookup)); 104 if (asn == NULL) 105 (void)fclose(f); 106 else 107 asn->as_f = f; 108 109#ifdef AS_DEBUG_FILE 110 asn->as_debug = fopen(AS_DEBUG_FILE, "w"); 111 if (asn->as_debug) { 112 (void)fprintf(asn->as_debug, ">> !!\n"); 113 (void)fflush(asn->as_debug); 114 } 115#endif /* AS_DEBUG_FILE */ 116 117 return (asn); 118} 119 120unsigned int 121as_lookup(void *_asn, char *addr, sa_family_t family) 122{ 123 struct aslookup *asn = _asn; 124 char buf[1024]; 125 unsigned int as; 126 int rc, dlen, plen; 127 128 as = 0; 129 rc = dlen = 0; 130 plen = (family == AF_INET6) ? 128 : 32; 131 (void)fprintf(asn->as_f, "!r%s/%d,l\n", addr, plen); 132 (void)fflush(asn->as_f); 133 134#ifdef AS_DEBUG_FILE 135 if (asn->as_debug) { 136 (void)fprintf(asn->as_debug, ">> !r%s/%d,l\n", addr, plen); 137 (void)fflush(asn->as_debug); 138 } 139#endif /* AS_DEBUG_FILE */ 140 141 while (fgets(buf, sizeof(buf), asn->as_f) != NULL) { 142 buf[sizeof(buf) - 1] = '\0'; 143 144#ifdef AS_DEBUG_FILE 145 if (asn->as_debug) { 146 (void)fprintf(asn->as_debug, "<< %s", buf); 147 (void)fflush(asn->as_debug); 148 } 149#endif /* AS_DEBUG_FILE */ 150 151 if (rc == 0) { 152 rc = buf[0]; 153 switch (rc) { 154 case 'A': 155 /* A - followed by # bytes of answer */ 156 sscanf(buf, "A%d\n", &dlen); 157#ifdef AS_DEBUG_FILE 158 if (asn->as_debug) { 159 (void)fprintf(asn->as_debug, 160 "dlen: %d\n", dlen); 161 (void)fflush(asn->as_debug); 162 } 163#endif /* AS_DEBUG_FILE */ 164 break; 165 case 'C': 166 case 'D': 167 case 'E': 168 case 'F': 169 /* C - no data returned */ 170 /* D - key not found */ 171 /* E - multiple copies of key */ 172 /* F - some other error */ 173 break; 174 } 175 if (rc == 'A') 176 /* skip to next input line */ 177 continue; 178 } 179 180 if (dlen == 0) 181 /* out of data, next char read is end code */ 182 rc = buf[0]; 183 if (rc != 'A') 184 /* either an error off the bat, or a done code */ 185 break; 186 187 /* data received, thank you */ 188 dlen -= strlen(buf); 189 190 /* origin line is the interesting bit */ 191 if (as == 0 && strncasecmp(buf, "origin:", 7) == 0) { 192 sscanf(buf + 7, " AS%u", &as); 193#ifdef AS_DEBUG_FILE 194 if (asn->as_debug) { 195 (void)fprintf(asn->as_debug, "as: %d\n", as); 196 (void)fflush(asn->as_debug); 197 } 198#endif /* AS_DEBUG_FILE */ 199 } 200 } 201 202 return (as); 203} 204 205void 206as_shutdown(void *_asn) 207{ 208 struct aslookup *asn = _asn; 209 210 (void)fprintf(asn->as_f, "!q\n"); 211 (void)fclose(asn->as_f); 212 213#ifdef AS_DEBUG_FILE 214 if (asn->as_debug) { 215 (void)fprintf(asn->as_debug, ">> !q\n"); 216 (void)fclose(asn->as_debug); 217 } 218#endif /* AS_DEBUG_FILE */ 219 220 free(asn); 221} 222