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: src/usr.sbin/ppp/iplist.c,v 1.9 1999/08/28 01:18:31 peter Exp $ 27 * $DragonFly: src/usr.sbin/ppp/iplist.c,v 1.2 2003/06/17 04:30:00 dillon Exp $ 28 */ 29 30 #include <sys/types.h> 31 #include <netinet/in.h> 32 #include <arpa/inet.h> 33 34 #include <stdlib.h> 35 #include <string.h> 36 #include <termios.h> 37 38 #include "log.h" 39 #include "defs.h" 40 #include "iplist.h" 41 42 static int 43 do_inet_aton(const char *start, const char *end, struct in_addr *ip) 44 { 45 char ipstr[16]; 46 47 if (end - start > 15) { 48 log_Printf(LogWARN, "%.*s: Invalid IP address\n", (int)(end-start), start); 49 return 0; 50 } 51 strncpy(ipstr, start, end-start); 52 ipstr[end-start] = '\0'; 53 return inet_aton(ipstr, ip); 54 } 55 56 static void 57 iplist_first(struct iplist *list) 58 { 59 list->cur.pos = -1; 60 } 61 62 static int 63 iplist_setrange(struct iplist *list, char *range) 64 { 65 char *ptr, *to; 66 67 if ((ptr = strpbrk(range, ",-")) == NULL) { 68 if (!inet_aton(range, &list->cur.ip)) 69 return 0; 70 list->cur.lstart = ntohl(list->cur.ip.s_addr); 71 list->cur.nItems = 1; 72 } else { 73 if (!do_inet_aton(range, ptr, &list->cur.ip)) 74 return 0; 75 if (*ptr == ',') { 76 list->cur.lstart = ntohl(list->cur.ip.s_addr); 77 list->cur.nItems = 1; 78 } else { 79 struct in_addr endip; 80 81 to = ptr+1; 82 if ((ptr = strpbrk(to, ",-")) == NULL) 83 ptr = to + strlen(to); 84 if (*to == '-') 85 return 0; 86 if (!do_inet_aton(to, ptr, &endip)) 87 return 0; 88 list->cur.lstart = ntohl(list->cur.ip.s_addr); 89 list->cur.nItems = ntohl(endip.s_addr) - list->cur.lstart + 1; 90 if (list->cur.nItems < 1) 91 return 0; 92 } 93 } 94 list->cur.srcitem = 0; 95 list->cur.srcptr = range; 96 return 1; 97 } 98 99 static int 100 iplist_nextrange(struct iplist *list) 101 { 102 char *ptr, *to, *end; 103 104 ptr = list->cur.srcptr; 105 if (ptr != NULL && (ptr = strchr(ptr, ',')) != NULL) 106 ptr++; 107 else 108 ptr = list->src; 109 110 while (*ptr != '\0' && !iplist_setrange(list, ptr)) { 111 if ((end = strchr(ptr, ',')) == NULL) 112 end = ptr + strlen(ptr); 113 if (end == ptr) 114 return 0; 115 log_Printf(LogWARN, "%.*s: Invalid IP range (skipping)\n", 116 (int)(end - ptr), ptr); 117 to = ptr; 118 do 119 *to = *end++; 120 while (*to++ != '\0'); 121 if (*ptr == '\0') 122 ptr = list->src; 123 } 124 125 return 1; 126 } 127 128 struct in_addr 129 iplist_next(struct iplist *list) 130 { 131 if (list->cur.pos == -1) { 132 list->cur.srcptr = NULL; 133 if (!iplist_nextrange(list)) { 134 list->cur.ip.s_addr = INADDR_ANY; 135 return list->cur.ip; 136 } 137 } else if (++list->cur.srcitem == list->cur.nItems) { 138 if (!iplist_nextrange(list)) { 139 list->cur.ip.s_addr = INADDR_ANY; 140 list->cur.pos = -1; 141 return list->cur.ip; 142 } 143 } else 144 list->cur.ip.s_addr = htonl(list->cur.lstart + list->cur.srcitem); 145 list->cur.pos++; 146 147 return list->cur.ip; 148 } 149 150 int 151 iplist_setsrc(struct iplist *list, const char *src) 152 { 153 strncpy(list->src, src, sizeof list->src - 1); 154 list->src[sizeof list->src - 1] = '\0'; 155 list->cur.srcptr = list->src; 156 do { 157 if (iplist_nextrange(list)) 158 list->nItems += list->cur.nItems; 159 else 160 return 0; 161 } while (list->cur.srcptr != list->src); 162 return 1; 163 } 164 165 void 166 iplist_reset(struct iplist *list) 167 { 168 list->src[0] = '\0'; 169 list->nItems = 0; 170 list->cur.pos = -1; 171 } 172 173 struct in_addr 174 iplist_setcurpos(struct iplist *list, long pos) 175 { 176 if (pos < 0 || pos >= list->nItems) { 177 list->cur.pos = -1; 178 list->cur.ip.s_addr = INADDR_ANY; 179 return list->cur.ip; 180 } 181 182 list->cur.srcptr = NULL; 183 list->cur.pos = 0; 184 while (1) { 185 iplist_nextrange(list); 186 if (pos < list->cur.nItems) { 187 if (pos) { 188 list->cur.srcitem = pos; 189 list->cur.pos += pos; 190 list->cur.ip.s_addr = htonl(list->cur.lstart + list->cur.srcitem); 191 } 192 break; 193 } 194 pos -= list->cur.nItems; 195 list->cur.pos += list->cur.nItems; 196 } 197 198 return list->cur.ip; 199 } 200 201 struct in_addr 202 iplist_setrandpos(struct iplist *list) 203 { 204 randinit(); 205 return iplist_setcurpos(list, random() % list->nItems); 206 } 207 208 int 209 iplist_ip2pos(struct iplist *list, struct in_addr ip) 210 { 211 struct iplist_cur cur; 212 u_long f; 213 int result; 214 215 result = -1; 216 memcpy(&cur, &list->cur, sizeof cur); 217 218 for (iplist_first(list), f = 0; f < list->nItems; f++) 219 if (iplist_next(list).s_addr == ip.s_addr) { 220 result = list->cur.pos; 221 break; 222 } 223 224 memcpy(&list->cur, &cur, sizeof list->cur); 225 return result; 226 } 227