The index key for user records now omits non-alphanumeric characters, in addition...
[citadel.git] / citadel / user_ops.c
index c5eeabd2cbca8bd9fec607d7d10fc542f15af69d..aaf26a5b3a9b6fe775124121832087a8cfab8061 100644 (file)
@@ -31,8 +31,6 @@
 int chkpwd_write_pipe[2];
 int chkpwd_read_pipe[2];
 
-
-
 /*
  * Trim a string down to the maximum username size and return the new length
  */
@@ -51,19 +49,22 @@ long cutusername(char *username) {
 
 /*
  * makeuserkey() - convert a username into the format used as a database key
- *              (it's just the username converted into lower case)
+ *                 (Key format is the username with all non-alphanumeric characters removed, and converted to lower case.)
  */
 void makeuserkey(char *key, const char *username, long len) {
        int i;
+       int keylen = 0;
 
-       if (len >= USERNAME_SIZE)
-       {
+       if (len >= USERNAME_SIZE) {
                syslog(LOG_INFO, "Username too long: %s", username);
                len = USERNAME_SIZE - 1; 
        }
        for (i=0; i<=len; ++i) {
-               key[i] = tolower(username[i]);
+               if (isalnum((username[i]))) {
+                       key[keylen++] = tolower(username[i]);
+               }
        }
+       key[keylen++] = 0;
 }
 
 
@@ -180,10 +181,8 @@ int rename_user(char *oldname, char *newname) {
                if (CtdlGetUser(&usbuf, oldname) != 0) {
                        retcode = RENAMEUSER_NOT_FOUND;
                }
-
                else {          /* Sanity checks succeeded.  Now rename the user. */
-                       if (usbuf.usernum == 0)
-                       {
+                       if (usbuf.usernum == 0) {
                                syslog(LOG_DEBUG, "user_ops: can not rename user \"Citadel\".");
                                retcode = RENAMEUSER_NOT_FOUND;
                        } else {
@@ -203,6 +202,57 @@ int rename_user(char *oldname, char *newname) {
 }
 
 
+/*
+ * Convert a username into the format used as a database key prior to version 928
+ * (This is only used during database upgrade)
+ */
+void makeuserkey_pre928(char *key, const char *username, long len) {
+       int i;
+
+       if (len >= USERNAME_SIZE) {
+               syslog(LOG_INFO, "Username too long: %s", username);
+               len = USERNAME_SIZE - 1; 
+       }
+       for (i=0; i<=len; ++i) {
+               key[i] = tolower(username[i]);
+       }
+}
+
+
+/*
+ * Read a user record using the pre-v928 index format, and write it back using the v928-and-higher index format.
+ * This ONLY gets called by...
+ */
+void reindex_user_928(char *username, void *out_data) {
+
+       char oldkey[USERNAME_SIZE];
+       char newkey[USERNAME_SIZE];
+       struct cdbdata *cdbus;
+       long len = cutusername(username);
+       struct ctdluser usbuf;
+
+       makeuserkey_pre928(oldkey, username, len);
+       makeuserkey(newkey, username, len);
+
+       syslog(LOG_DEBUG, "user_ops: reindex_user_928 <%s> <%s> <%s>", username, oldkey, newkey);
+
+       // Fetch the user record using the old index format
+       cdbus = cdb_fetch(CDB_USERS, oldkey, strlen(oldkey));
+       if (cdbus == NULL) {
+               syslog(LOG_INFO, "user_ops: <%s> not found, were they already reindexed?", username);
+               return;
+       }
+       memcpy(&usbuf, cdbus->ptr, ((cdbus->len > sizeof(struct ctdluser)) ?  sizeof(struct ctdluser) : cdbus->len));
+       cdb_free(cdbus);
+
+       // delete the old record
+       cdb_delete(CDB_USERS, oldkey, strlen(oldkey));
+
+       // write the new record
+       cdb_store(CDB_USERS, newkey, strlen(newkey), &usbuf, sizeof(struct ctdluser));
+}
+
+
 /*
  * Index-generating function used by Ctdl[Get|Set]Relationship
  */
@@ -211,7 +261,6 @@ int GenerateRelationshipIndex(char *IndexBuf,
                              long RoomGen,
                              long UserID)
 {
-
        struct {
                long iRoomID;
                long iRoomGen;
@@ -249,10 +298,7 @@ void put_visit(visit *newvisit)
 /*
  * Define a relationship between a user and a room
  */
-void CtdlSetRelationship(visit *newvisit,
-                        struct ctdluser *rel_user,
-                        struct ctdlroom *rel_room)
-{
+void CtdlSetRelationship(visit *newvisit, struct ctdluser *rel_user, struct ctdlroom *rel_room) {
        /* We don't use these in Citadel because they're implicit by the
         * index, but they must be present if the database is exported.
         */
@@ -267,10 +313,7 @@ void CtdlSetRelationship(visit *newvisit,
 /*
  * Locate a relationship between a user and a room
  */
-void CtdlGetRelationship(visit *vbuf,
-                        struct ctdluser *rel_user,
-                        struct ctdlroom *rel_room)
-{
+void CtdlGetRelationship(visit *vbuf, struct ctdluser *rel_user, struct ctdlroom *rel_room) {
        char IndexBuf[32];
        int IndexLen;
        struct cdbdata *cdbvisit;
@@ -432,38 +475,12 @@ int CtdlGetUserByNumber(struct ctdluser *usbuf, long number)
 /*
  * Helper function for rebuild_usersbynumber()
  */
-void rebuild_ubn_for_user(struct ctdluser *usbuf, void *data) {
-
-       struct ubnlist {
-               struct ubnlist *next;
-               char username[USERNAME_SIZE];
-               long usernum;
-       };
-
-       static struct ubnlist *u = NULL;
-       struct ubnlist *ptr = NULL;
-
-       /* Lazy programming here.  Call this function as a ForEachUser backend
-        * in order to queue up the room names, or call it with a null user
-        * to make it do the processing.
-        */
-       if (usbuf != NULL) {
-               ptr = (struct ubnlist *) malloc(sizeof (struct ubnlist));
-               if (ptr == NULL) return;
-
-               ptr->usernum = usbuf->usernum;
-               safestrncpy(ptr->username, usbuf->fullname, sizeof ptr->username);
-               ptr->next = u;
-               u = ptr;
-               return;
-       }
+void rebuild_ubn_for_user(char *username, void *data) {
+       struct ctdluser u;
 
-       while (u != NULL) {
-               syslog(LOG_DEBUG, "user_ops: rebuilding usersbynumber index %10ld : %s", u->usernum, u->username);
-               cdb_store(CDB_USERSBYNUMBER, &u->usernum, sizeof(long), u->username, strlen(u->username)+1);
-               ptr = u;
-               u = u->next;
-               free(ptr);
+       syslog(LOG_DEBUG, "user_ops: rebuilding usersbynumber index for %s", username);
+       if (CtdlGetUser(&u, username) == 0) {
+               cdb_store(CDB_USERSBYNUMBER, &(u.usernum), sizeof(long), u.fullname, strlen(u.fullname)+1);
        }
 }
 
@@ -474,7 +491,6 @@ void rebuild_ubn_for_user(struct ctdluser *usbuf, void *data) {
 void rebuild_usersbynumber(void) {
        cdb_trunc(CDB_USERSBYNUMBER);                   /* delete the old indices */
        ForEachUser(rebuild_ubn_for_user, NULL);        /* enumerate the users */
-       rebuild_ubn_for_user(NULL, NULL);               /* and index them */
 }
 
 
@@ -1173,52 +1189,46 @@ int CtdlForgetThisRoom(void) {
 
 
 /* 
- *  Traverse the user file...
+ * Traverse the user file and perform a callback for each user record.
+ * (New improved version that runs in two phases so that callbacks can perform writes without having a r/o cursor open)
  */
-void ForEachUser(void (*CallBack) (struct ctdluser * EachUser, void *out_data),
-                void *in_data)
+void ForEachUser(void (*CallBack) (char *, void *out_data), void *in_data)
 {
-       struct ctdluser usbuf;
+       struct feu {
+               char username[USERNAME_SIZE];
+               int version;
+       };
+
        struct cdbdata *cdbus;
+       struct ctdluser *usptr;
+       int i = 0;
+       struct feu *usernames = NULL;
+       int num_users = 0;
+       int num_users_alloc = 0;
 
        cdb_rewind(CDB_USERS);
 
+       // Phase 1 : load list of usernames
        while (cdbus = cdb_next_item(CDB_USERS), cdbus != NULL) {
-               memset(&usbuf, 0, sizeof(struct ctdluser));
-               memcpy(&usbuf, cdbus->ptr,
-                      ((cdbus->len > sizeof(struct ctdluser)) ?
-                       sizeof(struct ctdluser) : cdbus->len));
-               cdb_free(cdbus);
-               (*CallBack) (&usbuf, in_data);
-       }
-}
-
-
-/*
- * List one user (this works with cmd_list)
- */
-void ListThisUser(struct ctdluser *usbuf, void *data)
-{
-       char *searchstring;
+               usptr = (struct ctdluser *) cdbus->ptr;
 
-       searchstring = (char *)data;
-       if (bmstrcasestr(usbuf->fullname, searchstring) == NULL) {
-               return;
+               if (strlen(usptr->fullname) > 0) {
+                       ++num_users;
+                       if (num_users > num_users_alloc) {
+                               num_users_alloc = ((num_users_alloc == 0) ? 1 : (num_users_alloc * 2));
+                               usernames = realloc(usernames, num_users_alloc * sizeof(struct feu));
+                       }
+                       strcpy(usernames[num_users-1].username, usptr->fullname);
+                       usernames[num_users-1].version = usptr->version;
+               }
        }
 
-       if (usbuf->axlevel > AxDeleted) {
-               if ((CC->user.axlevel >= AxAideU)
-                   || ((usbuf->flags & US_UNLISTED) == 0)
-                   || ((CC->internal_pgm))) {
-                       cprintf("%s|%d|%ld|%ld|%ld|%ld||\n",
-                               usbuf->fullname,
-                               usbuf->axlevel,
-                               usbuf->usernum,
-                               (long)usbuf->lastcall,
-                               usbuf->timescalled,
-                               usbuf->posted);
-               }
+       // Phase 2 : perform the callback for each username
+       for (i=0; i<num_users; ++i) {
+               (*CallBack) (usernames[i].username, in_data);
        }
+
+       free(usernames);
 }