src/UserNode.c

   1 /* UserNode.c - NeonServ v5.0
   2  * Copyright (C) 2011  Philipp Kreil (pk910)
   3  *
   4  * This program is free software: you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation, either version 3 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program. If not, see <http://www.gnu.org/licenses/>.
  16  */
  17 #include "UserNode.h"
  18 #include "ChanUser.h"
  19 #include "tools.h"
  20
  21 static struct UserNode **userList;
  22
  23 void init_UserNode() {
  24     userList = calloc(VALID_NICK_CHARS_FIRST_LEN+1, sizeof(*userList));
  25 }
  26
  27 void free_UserNode() {
  28     //kamikaze free all users
  29     //chanusers will be destroyed in free_ChanNode()
  30     int i;
  31     struct UserNode *user, *next;
  32     for(i = 0; i < VALID_NICK_CHARS_FIRST_LEN+1; i++) {
  33         for(user = userList[i]; user; user = next) {
  34             next = user->next;
  35             free(user);
  36         }
  37     }
  38     free(userList);
  39 }
  40
  41 int is_valid_nick(const char *nick) {
  42     unsigned int i;
  43     //first char must be one of: a-zA-Z{|}~[\]^_`
  44     if (!strchr(VALID_NICK_CHARS_FIRST, *nick))
  45         return 0;
  46     //all other chars must be one of: a-zA-Z0-9{|}~[\]^_`
  47     for (i = 0; nick[i]; ++i)
  48         if (!strchr(VALID_NICK_CHARS, nick[i]))
  49             return 0;
  50     if (strlen(nick) > NICKLEN)
  51         return 0;
  52     return 1;
  53 }
  54
  55 static int get_nicklist_entry(int nick) {
  56     int i;
  57     char *valid_chars = VALID_NICK_CHARS_FIRST;
  58     for(i = 0; i < VALID_NICK_CHARS_FIRST_LEN; i++) {
  59         if(valid_chars[i] == nick)
  60             return i;
  61     }
  62     return -1; //ERROR!
  63 }
  64
  65 struct UserNode* getUserByNick(const char *nick) { //case sensitive
  66     int userListIndex = get_nicklist_entry(*nick);
  67     if(userListIndex == -1 || userList[userListIndex] == NULL)
  68         return NULL;
  69     struct UserNode *user;
  70     for(user = userList[userListIndex]; user; user = user->next) {
  71         if(!stricmp(nick, user->nick))
  72             return user;
  73     }
  74     return NULL;
  75 }
  76
  77 struct UserNode* getUserByMask(const char *mask) { //case sensitive
  78     char cmask[strlen(mask)+1];
  79     strcpy(cmask, mask);
  80     int i;
  81     struct UserNode *user = NULL;
  82     for(i = 0; i < strlen(mask); i++) {
  83         if(cmask[i] == '!') {
  84             cmask[i] = 0;
  85             user = getUserByNick(&cmask[0]);
  86             return user;
  87         } else if(cmask[i] == '.') {
  88             //it's a server
  89             return NULL;
  90         }
  91     }
  92     return NULL;
  93 }
  94
  95 struct UserNode* searchUserByNick(const char *nick) { //case insensitive
  96     if(!isalpha(*nick))
  97         return getUserByNick(nick);
  98
  99     int userListIndex;
 100     struct UserNode *user;
 101
 102     //search in the lower case "section"
 103     userListIndex = get_nicklist_entry(tolower(*nick));
 104     if(userListIndex != -1 && userList[userListIndex] != NULL) {
 105         for(user = userList[userListIndex]; user; user = user->next) {
 106             if(!stricmp(nick, user->nick))
 107                 return user;
 108         }
 109     }
 110     //search in the upper case "section"
 111     userListIndex = get_nicklist_entry(toupper(*nick));
 112     if(userListIndex != -1 && userList[userListIndex] != NULL) {
 113         for(user = userList[userListIndex]; user; user = user->next) {
 114             if(!stricmp(nick, user->nick))
 115                 return user;
 116         }
 117     }
 118     return NULL;
 119 }
 120
 121 int countUsersWithHost(char *host) {
 122     int i, count = 0;
 123     struct UserNode *user;
 124     for(i = 0; i < VALID_NICK_CHARS_FIRST_LEN+1; i++) {
 125         for(user = userList[i]; user; user = user->next) {
 126             if(!strcmp(user->host, host)) {
 127                 count++;
 128             }
 129         }
 130     }
 131     return count;
 132 }
 133
 134 char *getAuthFakehost(char *auth) {
 135     int i;
 136     struct UserNode *user;
 137     for(i = 0; i < VALID_NICK_CHARS_FIRST_LEN+1; i++) {
 138         for(user = userList[i]; user; user = user->next) {
 139             if((user->flags & USERFLAG_ISAUTHED) && !strcmp(user->auth, auth) && isFakeHost(user->host)) {
 140                 return user->host;
 141             }
 142         }
 143     }
 144     return NULL;
 145 }
 146
 147 struct UserNode* getAllUsers(struct UserNode *last) {
 148     if(last == NULL || last->next == NULL) {
 149         int cindex;
 150         if(last == NULL)
 151             cindex = 0;
 152         else
 153             cindex = get_nicklist_entry(last->nick[0]) + 1;
 154         while(userList[cindex] == NULL && cindex <= VALID_NICK_CHARS_FIRST_LEN)
 155             cindex++;
 156         if(cindex > VALID_NICK_CHARS_FIRST_LEN) return NULL;
 157         return userList[cindex];
 158     } else
 159         return last->next;
 160 }
 161
 162 int getUserCount() {
 163     int i, count = 0;
 164     struct UserNode *user;
 165     for(i = 0; i < VALID_NICK_CHARS_FIRST_LEN+1; i++) {
 166         for(user = userList[i]; user; user = user->next) {
 167             count++;
 168         }
 169     }
 170     return count;
 171 }
 172
 173 struct UserNode* addUser(const char *nick) {
 174     int userListIndex = get_nicklist_entry(*nick);
 175     if(userListIndex == -1 || !is_valid_nick(nick))
 176         return NULL;
 177     struct UserNode *user = malloc(sizeof(*user));
 178     if (!user)
 179     {
 180         perror("malloc() failed");
 181         return NULL;
 182     }
 183     strcpy(user->nick, nick);
 184     user->created = time(0);
 185     user->ident[0] = 0;
 186     user->host[0] = 0;
 187     user->realname[0] = 0;
 188     user->flags = 0;
 189     user->channel = NULL;
 190     user->next = userList[userListIndex];
 191     userList[userListIndex] = user;
 192     return user;
 193 }
 194
 195 struct UserNode* addUserMask(const char *mask) {
 196     char cmask[strlen(mask)+1];
 197     strcpy(cmask, mask);
 198     int i, ii = 0;
 199     struct UserNode *user = NULL;
 200     for(i = 0; i < strlen(mask)+1; i++) {
 201         if(cmask[i] == '!') {
 202             cmask[i] = 0;
 203             user = addUser(cmask);
 204             if(user == NULL) return NULL;
 205             ii = i+1;
 206         } else if(cmask[i] == '.' && !user) {
 207             //it's a server
 208             return NULL;
 209         } else if(cmask[i] == '@') {
 210             if(user == NULL) return NULL;
 211             cmask[i] = 0;
 212             strcpy(user->ident, &cmask[ii]);
 213             ii = i+1;
 214         } else if(cmask[i] == '\0') {
 215             if(user == NULL) return NULL;
 216             strcpy(user->host, &cmask[ii]);
 217         }
 218     }
 219     return user;
 220 }
 221
 222 struct UserNode* createTempUser(const char *mask) {
 223     //note: it could also be a server we have to create a temponary user for...
 224     char cmask[strlen(mask)+1];
 225     strcpy(cmask, mask);
 226     int i, ii = 0;
 227     struct UserNode *user = NULL;
 228     for(i = 0; i < strlen(mask)+1; i++) {
 229         if(cmask[i] == '!') {
 230             cmask[i] = 0;
 231             user = malloc(sizeof(*user));
 232             if (!user)
 233             {
 234                 perror("malloc() failed");
 235                 return NULL;
 236             }
 237             strcpy(user->nick, cmask);
 238             user->created = time(0);
 239             user->ident[0] = 0;
 240             user->host[0] = 0;
 241             user->realname[0] = 0;
 242             user->flags = 0;
 243             user->channel = NULL;
 244             ii = i+1;
 245         } else if(cmask[i] == '.' && !user) {
 246             //it's a server
 247             user = malloc(sizeof(*user));
 248             if (!user)
 249             {
 250                 perror("malloc() failed");
 251                 return NULL;
 252             }
 253             strcpy(user->host, cmask);
 254             user->created = time(0);
 255             user->ident[0] = 0;
 256             user->host[0] = 0;
 257             user->realname[0] = 0;
 258             user->flags = USERFLAG_ISSERVER;
 259             user->channel = NULL;
 260             return user;
 261         } else if(cmask[i] == '@') {
 262             if(user == NULL) return NULL;
 263             cmask[i] = 0;
 264             strcpy(user->ident, &cmask[ii]);
 265             ii = i+1;
 266         } else if(cmask[i] == '\0') {
 267             if(user == NULL) {
 268                 //nick only
 269                 user = malloc(sizeof(*user));
 270                 if (!user)
 271                 {
 272                     perror("malloc() failed");
 273                     return NULL;
 274                 }
 275                 strcpy(user->nick, cmask);
 276                 user->created = time(0);
 277                 user->ident[0] = 0;
 278                 user->host[0] = 0;
 279                 user->realname[0] = 0;
 280                 user->flags = 0;
 281                 user->channel = NULL;
 282                 return user;
 283             }
 284             strcpy(user->host, &cmask[ii]);
 285         }
 286     }
 287     return user;
 288 }
 289
 290 int renameUser(struct UserNode* user, const char *new_nick) {
 291     if(!is_valid_nick(new_nick))
 292         return 0;
 293     if(user->nick[0] == *new_nick) {
 294         strcpy(user->nick, new_nick);
 295         return 1;
 296     }
 297     //delUser(user, 0); //EPIC FAIL! This deletes the user from the channel Userlist -.-
 298     //manually remove the user from the old userList
 299     int userListIndex = get_nicklist_entry(user->nick[0]);
 300     if(userListIndex != -1) {
 301         struct UserNode *cuser, *last_user = NULL;
 302         for(cuser = userList[userListIndex]; cuser; cuser = cuser->next) {
 303             if(cuser == user) {
 304                 if(last_user)
 305                     last_user->next = user->next;
 306                 else
 307                     userList[userListIndex] = user->next;
 308                 break;
 309             } else
 310                 last_user = cuser;
 311         }
 312     }
 313     userListIndex = get_nicklist_entry(*new_nick);
 314     strcpy(user->nick, new_nick);
 315     user->next = userList[userListIndex];
 316     userList[userListIndex] = user;
 317     return 1;
 318 }
 319
 320 void delUser(struct UserNode* user, int freeUser) {
 321     int userListIndex = get_nicklist_entry(user->nick[0]);
 322     if(userListIndex == -1) return;
 323     struct UserNode *cuser, *last_user = NULL;
 324     for(cuser = userList[userListIndex]; cuser; cuser = cuser->next) {
 325         if(cuser == user) {
 326             if(last_user)
 327                 last_user->next = user->next;
 328             else
 329                 userList[userListIndex] = user->next;
 330             break;
 331         } else
 332             last_user = cuser;
 333     }
 334     if(user->channel) {
 335         struct ChanUser *chanUser, *next;
 336         for(chanUser = user->channel; chanUser; chanUser = next) {
 337             next = chanUser->next_chan;
 338             removeChanUserFromLists(chanUser, 1, 0, freeUser);
 339         }
 340     }
 341     if(freeUser)
 342         free(user);
 343     else
 344         user->next = NULL;
 345 }
 346
 347 void clearTempUsers() {
 348     int userListIndex = TEMPUSER_LIST_INDEX;
 349     struct UserNode *cuser, *last_user = NULL, *next;
 350     time_t now = time(0);
 351     for(cuser = userList[userListIndex]; cuser; cuser = next) {
 352         next = cuser->next;
 353         if(cuser->flags & USERFLAG_FREETMPUSER || now - cuser->created >= 300) {
 354             if(last_user)
 355                 last_user->next = cuser->next;
 356             else
 357                 userList[userListIndex] = cuser->next;
 358             break;
 359         } else
 360             last_user = cuser;
 361     }
 362 }