diff options
Diffstat (limited to 'command.c')
| -rw-r--r-- | command.c | 74 |
1 files changed, 39 insertions, 35 deletions
@@ -390,25 +390,34 @@ struct cmd_info{ struct cmd_retval (*handler)(struct conn_data *data,const char *tag,const char **args); }; -static const struct cmd_info commands[]={ - {"version",1,false,cmd_version}, - {"register",2,true,cmd_register}, - {"login",2,true,cmd_login}, - {"logout",0,false,cmd_logout}, - {"list_rooms",0,false,cmd_list_rooms}, - {"list_members",1,false,cmd_list_members}, - {"create_room",0,false,cmd_create_room}, - {"invite",2,false,cmd_invite}, - {"send",2,true,cmd_send}, - {"history",2,false,cmd_history}, - {"history_before",3,false,cmd_history_before}, - {"ping",0,false,cmd_ping}, - {"is_online",1,false,cmd_is_online}, - {"firebase_token",1,false,cmd_firebase_token}, - {"delete_firebase_token",1,false,cmd_delete_firebase_token}, - {"user_active",1,false,cmd_user_active}, +// Use CommandHash.hs to re-generate this perfect hash function for a different +// list of commands. +#define COMMAND_HASH_MODULUS 31 +#define COMMAND_HASH(cmd0, len) ((cmd0 + 6 * len) % COMMAND_HASH_MODULUS) + +#define COMMAND_ENTRY(cmd0, cmd, nargs, longlast, handler) \ + [COMMAND_HASH(cmd0, strlen(cmd))] = {cmd, nargs, longlast, handler} + +// First argument to COMMAND_ENTRY must be command[0]. This is because +// apparently, "abc"[0] is not a constant expression, while strlen("abc") is. +static const struct cmd_info commands[COMMAND_HASH_MODULUS] = { + COMMAND_ENTRY('v', "version", 1, false, cmd_version), + COMMAND_ENTRY('r', "register", 2, true, cmd_register), + COMMAND_ENTRY('l', "login", 2, true, cmd_login), + COMMAND_ENTRY('l', "logout", 0, false, cmd_logout), + COMMAND_ENTRY('l', "list_rooms", 0, false, cmd_list_rooms), + COMMAND_ENTRY('l', "list_members", 1, false, cmd_list_members), + COMMAND_ENTRY('c', "create_room", 0, false, cmd_create_room), + COMMAND_ENTRY('i', "invite", 2, false, cmd_invite), + COMMAND_ENTRY('s', "send", 2, true, cmd_send), + COMMAND_ENTRY('h', "history", 2, false, cmd_history), + COMMAND_ENTRY('h', "history_before", 3, false, cmd_history_before), + COMMAND_ENTRY('p', "ping", 0, false, cmd_ping), + COMMAND_ENTRY('i', "is_online", 1, false, cmd_is_online), + COMMAND_ENTRY('f', "firebase_token", 1, false, cmd_firebase_token), + COMMAND_ENTRY('d', "delete_firebase_token", 1, false, cmd_delete_firebase_token), + COMMAND_ENTRY('u', "user_active", 1, false, cmd_user_active), }; -#define NCOMMANDS (sizeof(commands)/sizeof(commands[0])) bool handle_input_line(struct conn_data *data,char *line,size_t linelen){ @@ -427,36 +436,31 @@ bool handle_input_line(struct conn_data *data,char *line,size_t linelen){ sepp=memchr(line,' ',linelen); if(sepp==NULL)sepp=line+linelen; const size_t cmdlen=sepp-line; - size_t cmdi; - for(cmdi=0;cmdi<NCOMMANDS;cmdi++){ - if(cmdlen==strlen(commands[cmdi].cmdname)&& - memcmp(line,commands[cmdi].cmdname,cmdlen)==0){ - break; - } - } - - if(cmdi==NCOMMANDS){ + const struct cmd_info *command=&commands[COMMAND_HASH(line[0],cmdlen)]; + if(!command->cmdname + ||cmdlen!=strlen(command->cmdname) + ||memcmp(line,command->cmdname,cmdlen)!=0){ debug("Unknown command %s on connection %d",line,data->fd); return true; } // Ensure first command is 'version' - if(data->protversion==-1&&strcmp(commands[cmdi].cmdname,"version")!=0){ + if(data->protversion==-1&&command->handler!=cmd_version){ debug("Command %s before version negotiation on connection %d", - commands[cmdi].cmdname,data->fd); + command->cmdname,data->fd); return true; } - int nargs=commands[cmdi].nargs; - char *args[nargs]; + const int nargs=command->nargs; + const char *args[nargs]; size_t cursor=cmdlen+1; for(int i=0;i<nargs;i++){ if(cursor>linelen){ - debug("Connection %d sent too few parameters to command %s",data->fd,commands[cmdi].cmdname); + debug("Connection %d sent too few parameters to command %s",data->fd,command->cmdname); return true; } - if(i==nargs-1&&commands[cmdi].longlast){ + if(i==nargs-1&&command->longlast){ sepp=line+linelen; } else { sepp=memchr(line+cursor,' ',linelen-cursor); @@ -467,11 +471,11 @@ bool handle_input_line(struct conn_data *data,char *line,size_t linelen){ cursor=sepp-line+1; } if(sepp-line<(i64)linelen){ - debug("Connection %d sent too many parameters to command %s",data->fd,commands[cmdi].cmdname); + debug("Connection %d sent too many parameters to command %s",data->fd,command->cmdname); return true; } - struct cmd_retval retval=commands[cmdi].handler(data,tag,(const char**)args); + struct cmd_retval retval=command->handler(data,tag,(const char**)args); if(retval.memzero)sodium_memzero(line,linelen); return retval.socket_close; } |