src/user.c

   1 /*
   2  * Copyright (c) 2006-2009 Bjorn Andersson <flex@kryo.se>, Erik Ekman <yarrick@kryo.se>
   3  *
   4  * Permission to use, copy, modify, and distribute this software for any
   5  * purpose with or without fee is hereby granted, provided that the above
   6  * copyright notice and this permission notice appear in all copies.
   7  *
   8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
   9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  15  */
  16
  17 #include <stdio.h>
  18 #include <stdint.h>
  19 #include <time.h>
  20 #include <stdlib.h>
  21 #include <string.h>
  22 #include <signal.h>
  23 #include <unistd.h>
  24 #include <sys/types.h>
  25 #include <fcntl.h>
  26
  27 #ifdef WINDOWS32
  28 #include <winsock2.h>
  29 #else
  30 #include <err.h>
  31 #include <arpa/inet.h>
  32 #include <netinet/in.h>
  33 #include <sys/socket.h>
  34 #include <netdb.h>
  35 #endif
  36
  37 #include "common.h"
  38 #include "encoding.h"
  39 #include "user.h"
  40
  41 struct user users[USERS];
  42
  43 int
  44 init_users(in_addr_t my_ip, int netbits)
  45 {
  46         int i;
  47         int skip = 0;
  48         char newip[16];
  49         int created_users = 0;
  50
  51         int maxusers;
  52
  53         in_addr_t netmask = 0;
  54         struct in_addr net;
  55         struct in_addr ipstart;
  56
  57         for (i = 0; i < netbits; i++) {
  58                 netmask = (netmask << 1) | 1;
  59         }
  60         netmask <<= (32 - netbits);
  61         net.s_addr = htonl(netmask);
  62         ipstart.s_addr = my_ip & net.s_addr;
  63
  64         maxusers = (1 << (32-netbits)) - 3; /* 3: Net addr, broadcast addr, iodined addr */
  65
  66         memset(users, 0, USERS * sizeof(struct user));
  67         for (i = 0; i < USERS; i++) {
  68                 in_addr_t ip;
  69                 users[i].id = i;
  70                 snprintf(newip, sizeof(newip), "0.0.0.%d", i + skip + 1);
  71                 ip = ipstart.s_addr + inet_addr(newip);
  72                 if (ip == my_ip && skip == 0) {
  73                         /* This IP was taken by iodined */
  74                         skip++;
  75                         snprintf(newip, sizeof(newip), "0.0.0.%d", i + skip + 1);
  76                         ip = ipstart.s_addr + inet_addr(newip);
  77                 }
  78                 users[i].tun_ip = ip;
  79                 net.s_addr = ip;
  80                 if (maxusers--  < 1) {
  81                         users[i].disabled = 1;
  82                 } else {
  83                         users[i].disabled = 0;
  84                         created_users++;
  85                 }
  86                 users[i].inpacket.len = 0;
  87                 users[i].inpacket.offset = 0;
  88                 users[i].outpacket.len = 0;
  89                 users[i].q.id = 0;
  90                 users[i].out_acked_seqno = 0;
  91                 users[i].out_acked_fragment = 0;
  92                 users[i].fragsize = 4096;
  93         }
  94
  95         return created_users;
  96 }
  97
  98 int
  99 users_waiting_on_reply()
 100 {
 101         int ret;
 102         int i;
 103
 104         ret = 0;
 105         for (i = 0; i < USERS; i++) {
 106                 if (users[i].active && !users[i].disabled &&
 107                         users[i].last_pkt + 60 > time(NULL) &&
 108                         users[i].q.id != 0) {
 109                         ret++;
 110                 }
 111         }
 112
 113         return ret;
 114 }
 115
 116 int
 117 find_user_by_ip(uint32_t ip)
 118 {
 119         int ret;
 120         int i;
 121
 122         ret = -1;
 123         for (i = 0; i < USERS; i++) {
 124                 if (users[i].active && !users[i].disabled &&
 125                         users[i].last_pkt + 60 > time(NULL) &&
 126                         ip == users[i].tun_ip) {
 127                         ret = i;
 128                         break;
 129                 }
 130         }
 131         return ret;
 132 }
 133
 134 int
 135 all_users_waiting_to_send()
 136 {
 137         time_t now;
 138         int ret;
 139         int i;
 140
 141         ret = 1;
 142         now = time(NULL);
 143         for (i = 0; i < USERS; i++) {
 144                 if (users[i].active && !users[i].disabled &&
 145                         users[i].last_pkt + 60 > now &&
 146                         users[i].outpacket.len == 0) {
 147                         ret = 0;
 148                         break;
 149                 }
 150         }
 151         return ret;
 152 }
 153
 154 int
 155 find_available_user()
 156 {
 157         int ret = -1;
 158         int i;
 159         for (i = 0; i < USERS; i++) {
 160                 /* Not used at all or not used in one minute */
 161                 if ((!users[i].active || users[i].last_pkt + 60 < time(NULL)) && !users[i].disabled) {
 162                         users[i].active = 1;
 163                         users[i].last_pkt = time(NULL);
 164                         ret = i;
 165                         break;
 166                 }
 167         }
 168         return ret;
 169 }
 170
 171 void
 172 user_switch_codec(int userid, struct encoder *enc)
 173 {
 174         if (userid < 0 || userid >= USERS)
 175                 return;
 176
 177         users[userid].encoder = enc;
 178 }