src/user.c

   1 /*
   2  * Copyright (c) 2006-2007 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 <string.h>
  18 #include <stdio.h>
  19 #include <time.h>
  20 #include <sys/types.h>
  21 #include <sys/socket.h>
  22 #include <arpa/inet.h>
  23 #include <netinet/in.h>
  24
  25 #include "common.h"
  26 #include "user.h"
  27
  28 struct user users[USERS];
  29
  30 void
  31 init_users(in_addr_t my_ip)
  32 {
  33         int i;
  34         char newip[16];
  35
  36         memset(users, 0, USERS * sizeof(struct user));
  37         for (i = 0; i < USERS; i++) {
  38                 users[i].id = i;
  39                 snprintf(newip, sizeof(newip), "0.0.0.%d", i + 1);
  40                 users[i].tun_ip = my_ip + inet_addr(newip);;
  41                 users[i].inpacket.len = 0;
  42                 users[i].inpacket.offset = 0;
  43                 users[i].outpacket.len = 0;
  44                 users[i].q.id = 0;
  45         }
  46 }
  47
  48 int
  49 users_waiting_on_reply()
  50 {
  51         int ret;
  52         int i;
  53
  54         ret = 0;
  55         for (i = 0; i < USERS; i++) {
  56                 if (users[i].active && users[i].last_pkt + 60 > time(NULL) &&
  57                         users[i].q.id != 0) {
  58                         ret++;
  59                 }
  60         }
  61
  62         return ret;
  63 }
  64
  65 int
  66 find_user_by_ip(uint32_t ip)
  67 {
  68         int ret;
  69         int i;
  70
  71         ret = -1;
  72         for (i = 0; i < USERS; i++) {
  73                 if (users[i].active && users[i].last_pkt + 60 > time(NULL) &&
  74                         ip == users[i].tun_ip) {
  75                         ret = i;
  76                         break;
  77                 }
  78         }
  79         return ret;
  80 }
  81
  82 int
  83 all_users_waiting_to_send()
  84 {
  85         time_t now;
  86         int ret;
  87         int i;
  88
  89         ret = 1;
  90         now = time(NULL);
  91         for (i = 0; i < USERS; i++) {
  92                 if (users[i].active && users[i].last_pkt + 60 > now &&
  93                         users[i].outpacket.len == 0) {
  94                         ret = 0;
  95                         break;
  96                 }
  97         }
  98         return ret;
  99 }
 100
 101 int
 102 find_available_user()
 103 {
 104         int ret = -1;
 105         int i;
 106         for (i = 0; i < USERS; i++) {
 107                 /* Not used at all or not used in one minute */
 108                 if (!users[i].active || users[i].last_pkt + 60 < time(NULL)) {
 109                         users[i].active = 1;
 110                         users[i].last_pkt = time(NULL);
 111                         ret = i;
 112                         break;
 113                 }
 114         }
 115         return ret;
 116 }
 117