UserNode.c

   1 #include "UserNode.h"
   2 #include "ChanUser.h"
   3
   4 static struct UserNode **userList;
   5
   6 void init_UserNode() {
   7     userList = calloc(VALID_NICK_CHARS_FIRST_LEN, sizeof(*userList));
   8
   9 }
  10
  11 int is_valid_nick(const char *nick) {
  12     unsigned int i;
  13     //first char must be one of: a-zA-Z{|}~[\]^_`
  14     if (!strchr(VALID_NICK_CHARS_FIRST, *nick))
  15         return 0;
  16     //all other chars must be one of: a-zA-Z0-9{|}~[\]^_`
  17     for (i = 0; nick[i]; ++i)
  18         if (!strchr(VALID_NICK_CHARS, nick[i]))
  19             return 0;
  20     if (strlen(nick) > NICKLEN)
  21         return 0;
  22     return 1;
  23 }
  24
  25 static int get_nicklist_entry(int nick) {
  26     int i;
  27     char *valid_chars = VALID_NICK_CHARS_FIRST;
  28     for(i = 0; i < VALID_NICK_CHARS_FIRST_LEN; i++) {
  29         if(valid_chars[i] == nick)
  30             return i;
  31     }
  32     return -1; //ERROR!
  33 }
  34
  35 struct UserNode* getUserByNick(const char *nick) { //case sensitive
  36     int userListIndex = get_nicklist_entry(*nick);
  37     if(userListIndex == -1 || userList[userListIndex] == NULL)
  38         return NULL;
  39     struct UserNode *user;
  40     for(user = userList[userListIndex]; user; user = user->next) {
  41         if(!stricmp(nick, user->nick))
  42             return user;
  43     }
  44     return NULL;
  45 }
  46
  47 struct UserNode* getUserByMask(const char *mask) { //case sensitive
  48     char cmask[strlen(mask)+1];
  49     strcpy(cmask, mask);
  50     int i;
  51     struct UserNode *user = NULL;
  52     for(i = 0; i < strlen(mask); i++) {
  53         if(cmask[i] == '!') {
  54             cmask[i] = 0;
  55             user = getUserByNick(&cmask[0]);
  56             return user;
  57         } else if(cmask[i] == '.') {
  58             //it's a server
  59             return NULL;
  60         }
  61     }
  62     return NULL;
  63 }
  64
  65 struct UserNode* searchUserByNick(const char *nick) { //case insensitive
  66     if(!isalpha(*nick))
  67         return getUserByNick(nick);
  68
  69     int userListIndex;
  70     struct UserNode *user;
  71
  72     //search in the lower case "section"
  73     userListIndex = get_nicklist_entry(tolower(*nick));
  74     if(userListIndex != -1 && userList[userListIndex] != NULL) {
  75         for(user = userList[userListIndex]; user; user = user->next) {
  76             if(!stricmp(nick, user->nick))
  77                 return user;
  78         }
  79     }
  80     //search in the upper case "section"
  81     userListIndex = get_nicklist_entry(toupper(*nick));
  82     if(userListIndex != -1 && userList[userListIndex] != NULL) {
  83         for(user = userList[userListIndex]; user; user = user->next) {
  84             if(!stricmp(nick, user->nick))
  85                 return user;
  86         }
  87     }
  88     return NULL;
  89 }
  90
  91 struct UserNode* addUser(const char *nick) {
  92     int userListIndex = get_nicklist_entry(*nick);
  93     if(userListIndex == -1 || !is_valid_nick(nick))
  94         return NULL;
  95     struct UserNode *user = malloc(sizeof(*user));
  96     if (!user)
  97     {
  98         perror("malloc() failed");
  99         return NULL;
 100     }
 101     strcpy(user->nick, nick);
 102     user->ident[0] = 0;
 103     user->host[0] = 0;
 104     user->realname[0] = 0;
 105     user->flags = 0;
 106     user->channel = NULL;
 107     user->next = userList[userListIndex];
 108     userList[userListIndex] = user;
 109     return user;
 110 }
 111
 112
 113 struct UserNode* addUserMask(const char *mask) {
 114     char cmask[strlen(mask)+1];
 115     strcpy(cmask, mask);
 116     int i, ii = 0;
 117     struct UserNode *user = NULL;
 118     for(i = 0; i < strlen(mask)+1; i++) {
 119         if(cmask[i] == '!') {
 120             cmask[i] = 0;
 121             user = addUser(&cmask[0]);
 122             if(user == NULL) return NULL;
 123             ii = i+1;
 124         } else if(cmask[i] == '.' && !user) {
 125             //it's a server
 126             return NULL;
 127         } else if(cmask[i] == '@') {
 128             if(user == NULL) return NULL;
 129             cmask[i] = 0;
 130             strcpy(user->ident, &cmask[ii]);
 131             ii = i+1;
 132         } else if(cmask[i] == '\0') {
 133             if(user == NULL) return NULL;
 134             strcpy(user->host, &cmask[ii]);
 135         }
 136     }
 137     return NULL;
 138 }
 139
 140 int renameUser(struct UserNode* user, const char *new_nick) {
 141     if(!is_valid_nick(new_nick))
 142         return 0;
 143     if(user->nick[0] == *new_nick) {
 144         strcpy(user->nick, new_nick);
 145         return 1;
 146     }
 147     int userListIndex = get_nicklist_entry(*new_nick);
 148     delUser(user, 0);
 149     strcpy(user->nick, new_nick);
 150     user->next = userList[userListIndex];
 151     userList[userListIndex] = user;
 152     return 1;
 153 }
 154
 155 void delUser(struct UserNode* user, int freeUser) {
 156     int userListIndex = get_nicklist_entry(user->nick[0]);
 157     if(userListIndex == -1) return;
 158     struct UserNode *cuser, *last_user = NULL;
 159     for(cuser = userList[userListIndex]; cuser; cuser = cuser->next) {
 160         if(cuser == user) {
 161             if(last_user)
 162                 last_user->next = user->next;
 163             else
 164                 userList[userListIndex] = user->next;
 165             break;
 166         } else
 167             last_user = cuser;
 168     }
 169     if(user->channel) {
 170         struct ChanUser *chanUser, *next;
 171         for(chanUser = user->channel; chanUser; chanUser = next) {
 172             next = chanUser->next_chan;
 173             quitChanUser(chanUser, freeUser);
 174         }
 175     }
 176     if(freeUser)
 177         free(user);
 178     else
 179         user->next = NULL;
 180 }