New upstream release.
[debian/iodine.git] / 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 }