1 /*- 2 * Copyright (c) 1997 Brian Somers <brian@Awfulhak.org> 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 24 * SUCH DAMAGE. 25 * 26 * $FreeBSD$ 27 */ 28 29 #include <sys/types.h> 30 #include <netinet/in.h> 31 #include <arpa/inet.h> 32 33 #include <stdlib.h> 34 #include <string.h> 35 #include <termios.h> 36 37 #include "log.h" 38 #include "defs.h" 39 #include "iplist.h" 40 41 static int 42 do_inet_aton(const char *start, const char *end, struct in_addr *ip) 43 { 44 char ipstr[16]; 45 46 if (end - start > 15) { 47 log_Printf(LogWARN, "%.*s: Invalid IP address\n", (int)(end-start), start); 48 return 0; 49 } 50 strncpy(ipstr, start, end-start); 51 ipstr[end-start] = '\0'; 52 return inet_aton(ipstr, ip); 53 } 54 55 static void 56 iplist_first(struct iplist *list) 57 { 58 list->cur.pos = -1; 59 } 60 61 static int 62 iplist_setrange(struct iplist *list, char *range) 63 { 64 char *ptr, *to; 65 66 if ((ptr = strpbrk(range, ",-")) == NULL) { 67 if (!inet_aton(range, &list->cur.ip)) 68 return 0; 69 list->cur.lstart = ntohl(list->cur.ip.s_addr); 70 list->cur.nItems = 1; 71 } else { 72 if (!do_inet_aton(range, ptr, &list->cur.ip)) 73 return 0; 74 if (*ptr == ',') { 75 list->cur.lstart = ntohl(list->cur.ip.s_addr); 76 list->cur.nItems = 1; 77 } else { 78 struct in_addr endip; 79 80 to = ptr+1; 81 if ((ptr = strpbrk(to, ",-")) == NULL) 82 ptr = to + strlen(to); 83 if (*to == '-') 84 return 0; 85 if (!do_inet_aton(to, ptr, &endip)) 86 return 0; 87 list->cur.lstart = ntohl(list->cur.ip.s_addr); 88 list->cur.nItems = ntohl(endip.s_addr) - list->cur.lstart + 1; 89 if (list->cur.nItems < 1) 90 return 0; 91 } 92 } 93 list->cur.srcitem = 0; 94 list->cur.srcptr = range; 95 return 1; 96 } 97 98 static int 99 iplist_nextrange(struct iplist *list) 100 { 101 char *ptr, *to, *end; 102 103 ptr = list->cur.srcptr; 104 if (ptr != NULL && (ptr = strchr(ptr, ',')) != NULL) 105 ptr++; 106 else 107 ptr = list->src; 108 109 while (*ptr != '\0' && !iplist_setrange(list, ptr)) { 110 if ((end = strchr(ptr, ',')) == NULL) 111 end = ptr + strlen(ptr); 112 if (end == ptr) 113 return 0; 114 log_Printf(LogWARN, "%.*s: Invalid IP range (skipping)\n", 115 (int)(end - ptr), ptr); 116 to = ptr; 117 do 118 *to = *end++; 119 while (*to++ != '\0'); 120 if (*ptr == '\0') 121 ptr = list->src; 122 } 123 124 return 1; 125 } 126 127 struct in_addr 128 iplist_next(struct iplist *list) 129 { 130 if (list->cur.pos == -1) { 131 list->cur.srcptr = NULL; 132 if (!iplist_nextrange(list)) { 133 list->cur.ip.s_addr = INADDR_ANY; 134 return list->cur.ip; 135 } 136 } else if (++list->cur.srcitem == list->cur.nItems) { 137 if (!iplist_nextrange(list)) { 138 list->cur.ip.s_addr = INADDR_ANY; 139 list->cur.pos = -1; 140 return list->cur.ip; 141 } 142 } else 143 list->cur.ip.s_addr = htonl(list->cur.lstart + list->cur.srcitem); 144 list->cur.pos++; 145 146 return list->cur.ip; 147 } 148 149 int 150 iplist_setsrc(struct iplist *list, const char *src) 151 { 152 strncpy(list->src, src, sizeof list->src - 1); 153 list->src[sizeof list->src - 1] = '\0'; 154 list->cur.srcptr = list->src; 155 do { 156 if (iplist_nextrange(list)) 157 list->nItems += list->cur.nItems; 158 else 159 return 0; 160 } while (list->cur.srcptr != list->src); 161 return 1; 162 } 163 164 void 165 iplist_reset(struct iplist *list) 166 { 167 list->src[0] = '\0'; 168 list->nItems = 0; 169 list->cur.pos = -1; 170 } 171 172 struct in_addr 173 iplist_setcurpos(struct iplist *list, long pos) 174 { 175 if (pos < 0 || (unsigned)pos >= list->nItems) { 176 list->cur.pos = -1; 177 list->cur.ip.s_addr = INADDR_ANY; 178 return list->cur.ip; 179 } 180 181 list->cur.srcptr = NULL; 182 list->cur.pos = 0; 183 while (1) { 184 iplist_nextrange(list); 185 if (pos < (int)list->cur.nItems) { 186 if (pos) { 187 list->cur.srcitem = pos; 188 list->cur.pos += pos; 189 list->cur.ip.s_addr = htonl(list->cur.lstart + list->cur.srcitem); 190 } 191 break; 192 } 193 pos -= list->cur.nItems; 194 list->cur.pos += list->cur.nItems; 195 } 196 197 return list->cur.ip; 198 } 199 200 struct in_addr 201 iplist_setrandpos(struct iplist *list) 202 { 203 randinit(); 204 return iplist_setcurpos(list, random() % list->nItems); 205 } 206 207 int 208 iplist_ip2pos(struct iplist *list, struct in_addr ip) 209 { 210 struct iplist_cur cur; 211 u_long f; 212 int result; 213 214 result = -1; 215 memcpy(&cur, &list->cur, sizeof cur); 216 217 for (iplist_first(list), f = 0; f < list->nItems; f++) 218 if (iplist_next(list).s_addr == ip.s_addr) { 219 result = list->cur.pos; 220 break; 221 } 222 223 memcpy(&list->cur, &cur, sizeof list->cur); 224 return result; 225 } 226