[svn-upgrade] Integrating new upstream version, iodine (0.5.0)
[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 <netdb.h>
25 #include <netinet/in.h>
26 #include <sys/types.h>
27 #include <sys/socket.h>
28 #include <fcntl.h>
29 #include <err.h>
30 #include <arpa/inet.h>
31
32 #include "common.h"
33 #include "encoding.h"
34 #include "user.h"
35
36 struct user users[USERS];
37
38 int
39 init_users(in_addr_t my_ip, int netbits)
40 {
41         int i;
42         int skip = 0;
43         char newip[16];
44         int created_users = 0;
45
46         int maxusers;
47
48         in_addr_t netmask = 0;
49         struct in_addr net;
50         struct in_addr ipstart;
51
52         for (i = 0; i < netbits; i++) {
53                 netmask = (netmask << 1) | 1;
54         }
55         netmask <<= (32 - netbits);
56         net.s_addr = htonl(netmask);
57         ipstart.s_addr = my_ip & net.s_addr;
58
59         maxusers = (1 << (32-netbits)) - 3; /* 3: Net addr, broadcast addr, iodined addr */
60         
61         memset(users, 0, USERS * sizeof(struct user));
62         for (i = 0; i < USERS; i++) {
63                 in_addr_t ip;
64                 users[i].id = i;
65                 snprintf(newip, sizeof(newip), "0.0.0.%d", i + skip + 1);
66                 ip = ipstart.s_addr + inet_addr(newip);
67                 if (ip == my_ip && skip == 0) {
68                         /* This IP was taken by iodined */
69                         skip++;
70                         snprintf(newip, sizeof(newip), "0.0.0.%d", i + skip + 1);
71                         ip = ipstart.s_addr + inet_addr(newip);
72                 }
73                 users[i].tun_ip = ip;
74                 net.s_addr = ip;
75                 if (maxusers--  < 1) {
76                         users[i].disabled = 1;
77                 } else {
78                         users[i].disabled = 0;
79                         created_users++;
80                 }
81                 users[i].inpacket.len = 0;
82                 users[i].inpacket.offset = 0;
83                 users[i].outpacket.len = 0;
84                 users[i].q.id = 0;
85                 users[i].out_acked_seqno = 0;
86                 users[i].out_acked_fragment = 0;
87                 users[i].fragsize = 4096;
88         }
89
90         return created_users;
91 }
92
93 int
94 users_waiting_on_reply()
95 {
96         int ret;
97         int i;
98
99         ret = 0;
100         for (i = 0; i < USERS; i++) {
101                 if (users[i].active && !users[i].disabled && 
102                         users[i].last_pkt + 60 > time(NULL) &&
103                         users[i].q.id != 0) {
104                         ret++;
105                 }
106         }
107         
108         return ret;
109 }
110
111 int
112 find_user_by_ip(uint32_t ip)
113 {
114         int ret;
115         int i;
116
117         ret = -1;
118         for (i = 0; i < USERS; i++) {
119                 if (users[i].active && !users[i].disabled &&
120                         users[i].last_pkt + 60 > time(NULL) &&
121                         ip == users[i].tun_ip) {
122                         ret = i;
123                         break;
124                 }
125         }
126         return ret;
127 }
128
129 int
130 all_users_waiting_to_send()
131 {
132         time_t now;
133         int ret;
134         int i;
135
136         ret = 1;
137         now = time(NULL);
138         for (i = 0; i < USERS; i++) {
139                 if (users[i].active && !users[i].disabled &&
140                         users[i].last_pkt + 60 > now &&
141                         users[i].outpacket.len == 0) {
142                         ret = 0;
143                         break;
144                 }
145         }
146         return ret;
147 }
148
149 int
150 find_available_user()
151 {
152         int ret = -1;
153         int i;
154         for (i = 0; i < USERS; i++) {
155                 /* Not used at all or not used in one minute */
156                 if ((!users[i].active || users[i].last_pkt + 60 < time(NULL)) && !users[i].disabled) {
157                         users[i].active = 1;
158                         users[i].last_pkt = time(NULL);
159                         ret = i;
160                         break;
161                 }
162         }
163         return ret;
164 }
165
166 void
167 user_switch_codec(int userid, struct encoder *enc)
168 {
169         if (userid < 0 || userid >= USERS)
170                 return;
171         
172         users[userid].encoder = enc;
173 }