usr.sbin/ppp/iplist.c

   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 }