Added all builtin commands plus some cleanups.

AUTHORS

@@ -1,6 +1,8 @@
-Authors of uHub
+Authors of uhub
 ===============
 
 Jan Vidar Krey, Design and implementation
 E_zombie, Centos/RedHat customization scripts and heavy load testing
+FleetCommand, Hub topic
+MiMic, Implemented user commands
 

src/core/commands.c

@@ -27,13 +27,13 @@
 
 static int command_access_denied(struct command_base* cbase, struct hub_user* user, const char* prefix);
 static int command_not_found(struct command_base* cbase, struct hub_user* user, const char* prefix);
-static int command_status_user_not_found(struct command_base* cbase, struct hub_user* user, struct hub_command* cmd, const char* nick);
+static int command_syntax_error(struct command_base* cbase, struct hub_user* user);
-static int command_arg_mismatch(struct command_base* cbase, struct hub_user* user, struct hub_command* cmd, struct command_handle* handler);
 
 struct command_base
 {
 	struct hub_info* hub;
 	struct linked_list* handlers;
+	size_t prefix_length_max;
 };
 
 struct command_base* command_initialize(struct hub_info* hub)
@@ -44,13 +44,18 @@ struct command_base* command_initialize(struct hub_info* hub)
 
 	cbase->hub = hub;
 	cbase->handlers = (struct linked_list*) list_create();
+	cbase->prefix_length_max = 0;
 
 	uhub_assert(cbase->handlers != NULL);
+
+	commands_builtin_add(cbase);
+
 	return cbase;
 }
 
 void command_shutdown(struct command_base* cbase)
 {
+	commands_builtin_remove(cbase);
 	assert(list_size(cbase->handlers) == 0);
 	hub_free(cbase);
 }
@@ -63,6 +68,7 @@ int command_add(struct command_base* cbase, struct command_handle* cmd)
 	uhub_assert(cmd->handler != NULL);
 	uhub_assert(cmd->description && *cmd->description);
 	list_append(cbase->handlers, cmd);
+	cbase->prefix_length_max = MAX(cmd->length, cbase->prefix_length_max);
 	return 1;
 }
 
@@ -91,7 +97,7 @@ static void command_destroy(struct hub_command* cmd)
 	hub_free(cmd);
 }
 
-static struct command_handle* command_find_handler(struct command_base* cbase, struct hub_user* user, const char* prefix)
+static struct command_handle* command_handler_lookup(struct command_base* cbase, const char* prefix)
 {
 	struct command_handle* handler = NULL;
 	size_t prefix_len = strlen(prefix);
@@ -102,30 +108,20 @@ static struct command_handle* command_find_handler(struct command_base* cbase, s
 			continue;
 
 		if (!strncmp(prefix, handler->prefix, handler->length))
-		{
-			if (handler->cred <= user->credentials)
 		{
 			return handler;
 		}
-			else
-			{
-				command_access_denied(cbase, user, prefix);
-				return NULL;
 	}
-		}
-	}
-
-	command_not_found(cbase, user, prefix);
 	return NULL;
 }
 
-static struct linked_list* command_extract_arguments(struct command_base* cbase, struct hub_user* user, struct command_handle* command, struct linked_list* tokens)
+static struct linked_list* command_extract_arguments(struct command_base* cbase, struct command_handle* command, struct linked_list* tokens)
 {
 	int arg = 0;
 	int opt = 0;
 	char* token = NULL;
-	char* temp = NULL;
 	struct hub_user* target = NULL;
+	struct command_handle* target_command = NULL;
 	enum auth_credentials cred;
 	struct linked_list* args = list_create();
 
@@ -135,15 +131,12 @@ static struct linked_list* command_extract_arguments(struct command_base* cbase,
 
 	while ((token = list_get_first(tokens)))
 	{
-		user = NULL;
-		temp = NULL;
-
 		switch (command->args[arg++])
 		{
 			case '?':
 				uhub_assert(opt == 0);
 				opt = 1;
-				break;
+				continue;
 
 			case 'n':
 				target = uman_get_user_by_nick(cbase->hub, token);
@@ -176,10 +169,19 @@ static struct linked_list* command_extract_arguments(struct command_base* cbase,
 
 			case 'm':
 			case 'p':
-			case 'c':
 				list_append(args, token);
 				break;
 
+			case 'c':
+				target_command = command_handler_lookup(cbase, token);
+				if (!target_command)
+				{
+					list_destroy(args);
+					return NULL;
+				}
+				list_append(args, target_command);
+				break;
+
 			case 'C':
 				if (!auth_string_to_cred(token, &cred))
 				{
@@ -210,7 +212,7 @@ static struct linked_list* command_extract_arguments(struct command_base* cbase,
 /**
  * Parse a command and break it down into a struct hub_command.
  */
-static void command_parse(struct command_base* cbase, struct hub_user* user, const char* message)
+static int command_parse(struct command_base* cbase, struct hub_user* user, const char* message)
 {
 	char* prefix;
 	int n;
@@ -218,7 +220,7 @@ static void command_parse(struct command_base* cbase, struct hub_user* user, con
 	struct command_handle* handler = NULL;
 	struct linked_list* tokens = NULL;
 
-	if (!cmd) return;
+	if (!cmd) return 0;
 
 	cmd->message = message;
 	cmd->args = NULL;
@@ -228,7 +230,8 @@ static void command_parse(struct command_base* cbase, struct hub_user* user, con
 	if (n <= 0)
 	{
 		command_destroy(cmd);
-		return;
+		// FIXME
+		return 0;
 	}
 
 	// Find a matching command handler
@@ -237,12 +240,17 @@ static void command_parse(struct command_base* cbase, struct hub_user* user, con
 	{
 		cmd->prefix = hub_strdup(&prefix[1]);
 		cmd->prefix_len = strlen(cmd->prefix);
-		handler = command_find_handler(cbase, user, cmd->prefix);
+		handler = command_handler_lookup(cbase, cmd->prefix);
+		if (!handler)
+		{
+			return command_not_found(cbase, user, prefix);
+		}
 	}
 	else
 	{
 		command_destroy(cmd);
-		return;
+		command_syntax_error(cbase, user);
+		return 0;
 	}
 
 	// Remove the first token.
@@ -250,19 +258,28 @@ static void command_parse(struct command_base* cbase, struct hub_user* user, con
 	hub_free(prefix);
 
 	// Parse arguments
-	cmd->args = command_extract_arguments(cbase, user, handler, tokens);
+	cmd->args = command_extract_arguments(cbase, handler, tokens);
 	list_clear(tokens, &hub_free);
 	list_destroy(tokens);
 
 	if (!cmd->args)
 	{
 		command_destroy(cmd);
-		return;
+		// FIXME
+		return 0;
 	}
 
+	if (command_is_available(handler, user))
+	{
 		handler->handler(cbase, user, cmd);
 		command_destroy(cmd);
-	return;
+		return 0;
+	}
+	else
+	{
+		command_destroy(cmd);
+		return command_access_denied(cbase, user, prefix);
+	}
 }
 
 const char* command_get_syntax(struct command_handle* handler)
@@ -299,7 +316,7 @@ const char* command_get_syntax(struct command_handle* handler)
 }
 
 
-static void send_message(struct command_base* cbase, struct hub_user* user, const char* message)
+void send_message(struct command_base* cbase, struct hub_user* user, const char* message)
 {
 	char* buffer = adc_msg_escape(message);
 	struct adc_message* command = adc_msg_construct(ADC_CMD_IMSG, strlen(buffer) + 6);
@@ -325,14 +342,15 @@ static int command_not_found(struct command_base* cbase, struct hub_user* user,
 	return 0;
 }
 
-static int command_status_user_not_found(struct command_base* cbase, struct hub_user* user, struct hub_command* cmd, const char* nick)
+
+static int command_syntax_error(struct command_base* cbase, struct hub_user* user)
 {
-	char temp[128];
+	send_message(cbase, user, "*** Syntax error");
-	snprintf(temp, 128, "*** %s: No user \"%s\"", cmd->prefix, nick);
-	send_message(cbase, user, temp);
 	return 0;
 }
 
+
+
 static size_t command_count_required_args(struct command_handle* handler)
 {
 	size_t n = 0;
@@ -355,16 +373,19 @@ int command_check_args(struct hub_command* cmd, struct command_handle* handler)
 	return 0;
 }
 
-static int command_arg_mismatch(struct command_base* cbase, struct hub_user* user, struct hub_command* cmd, struct command_handle* handler)
+int command_invoke(struct command_base* cbase, struct hub_user* user, const char* message)
 {
-	char temp[256];
+	return command_parse(cbase, user, message);
-	const char* args = command_get_syntax(handler);
-	if (args) snprintf(temp, 256, "*** %s: Use: !%s %s", cmd->prefix, cmd->prefix, args);
-	else      snprintf(temp, 256, "*** %s: Use: !%s", cmd->prefix, cmd->prefix);
-	send_message(cbase, user, temp);
-	return 0;
 }
 
+int command_is_available(struct command_handle* handle, struct hub_user* user)
+{
+	uhub_assert(handle != NULL);
+	uhub_assert(user != NULL);
+	return handle->cred <= user->credentials;
+}
+
+
 static int command_status(struct command_base* cbase, struct hub_user* user, struct hub_command* cmd, const char* message)
 {
 	char temp[1024];
@@ -373,71 +394,51 @@ static int command_status(struct command_base* cbase, struct hub_user* user, str
 	return 0;
 }
 
-int command_invoke(struct command_base* cbase, struct hub_user* user, const char* message)
+static int command_status_user_not_found(struct command_base* cbase, struct hub_user* user, struct hub_command* cmd, const char* nick)
-{
-	command_parse(cbase, user, message);
-	return 0;
-}
-
-
-
-
-static int command_stats(struct command_base* cbase, struct hub_user* user, struct hub_command* cmd)
 {
 	char temp[128];
-	struct hub_info* hub = cbase->hub;
+	snprintf(temp, 128, "*** %s: No user \"%s\"", cmd->prefix, nick);
-
+	send_message(cbase, user, temp);
-	snprintf(temp, 128, PRINTF_SIZE_T " users, peak: " PRINTF_SIZE_T ". Network (up/down): %d/%d KB/s, peak: %d/%d KB/s",
+	return 0;
-	hub->users->count,
-	hub->users->count_peak,
-	(int) hub->stats.net_tx / 1024,
-	(int) hub->stats.net_rx / 1024,
-	(int) hub->stats.net_tx_peak / 1024,
-	(int) hub->stats.net_rx_peak / 1024);
-	return command_status(cbase, user, cmd, temp);
 }
 
 static int command_help(struct command_base* cbase, struct hub_user* user, struct hub_command* cmd)
 {
-#if 0
-	size_t n;
 	char msg[MAX_HELP_MSG];
-	char* command = list_get_first(cmd->args);
+	struct command_handle* command = list_get_first(cmd->args);
+	size_t n = 0;
 	msg[0] = 0;
 
 	if (!command)
 	{
 		strcat(msg, "Available commands:\n");
 
-		for (n = 0; command_handlers[n].prefix; n++)
+		for (command = (struct command_handle*) list_get_first(cbase->handlers); command; command = (struct command_handle*) list_get_next(cbase->handlers))
 		{
-			if (command_handlers[n].cred <= user->credentials)
+			if (command_is_available(command, user))
 			{
 				strcat(msg, "!");
-				strcat(msg, command_handlers[n].prefix);
+				strcat(msg, command->prefix);
-				strcat(msg, " - ");
+				for (n = command->length; n < cbase->prefix_length_max; n++)
-				strcat(msg, command_handlers[n].description);
+				{
+					strcat(msg, " ");
+				}
+				strcat(msg, "- ");
+				strcat(msg, command->description);
 				strcat(msg, "\n");
 			}
 		}
 	}
 	else
 	{
-		int found = 0;
+		if (command_is_available(command, user))
-		for (n = 0; command_handlers[n].prefix; n++)
-		{
-			if (strcmp(command, command_handlers[n].prefix) == 0)
-			{
-				found = 1;
-				if (command_handlers[n].cred <= user->credentials)
 		{
 			strcat(msg, "Usage: !");
-					strcat(msg, command_handlers[n].prefix);
+			strcat(msg, command->prefix);
 			strcat(msg, " ");
-					strcat(msg, command_get_syntax(&command_handlers[n]));
+			strcat(msg, command_get_syntax(command));
 			strcat(msg, "\n");
-
+			strcat(msg, command->description);
-					strcat(msg, command_handlers[n].description);
 			strcat(msg, "\n");
 		}
 		else
@@ -445,16 +446,7 @@ static int command_help(struct command_base* cbase, struct hub_user* user, struc
 			strcat(msg, "This command is not available to you.\n");
 		}
 	}
-		}
-
-		if (!found)
-		{
-			sprintf(msg, "Command \"%s\" not found.\n", command);
-		}
-	}
 	return command_status(cbase, user, cmd, msg);
-#endif
-	return 0;
 }
 
 static int command_uptime(struct command_base* cbase, struct hub_user* user, struct hub_command* cmd)
@@ -977,48 +969,77 @@ static int command_crash(struct command_base* cbase, struct hub_user* user, stru
 }
 #endif
 
-#if 0
+
-int command_dipatcher(struct hub_info* hub, struct hub_user* user, const char* message)
+
+static int command_stats(struct command_base* cbase, struct hub_user* user, struct hub_command* cmd)
 {
-	size_t n = 0;
+	char temp[128];
-	int rc;
+	struct hub_info* hub = cbase->hub;
 
-	/* Parse and validate the command */
+	snprintf(temp, 128, PRINTF_SIZE_T " users, peak: " PRINTF_SIZE_T ". Network (up/down): %d/%d KB/s, peak: %d/%d KB/s",
-	struct hub_command* cmd = command_create(message);
+	hub->users->count,
-	if (!cmd) return 0;
+	hub->users->count_peak,
-
+	(int) hub->stats.net_tx / 1024,
-	for (n = 0; command_handlers[n].prefix; n++)
+	(int) hub->stats.net_rx / 1024,
-	{
+	(int) hub->stats.net_tx_peak / 1024,
-		struct commands_handler* handler = &command_handlers[n];
+	(int) hub->stats.net_rx_peak / 1024);
-		if (cmd->prefix_len != handler->length)
+	return command_status(cbase, user, cmd, temp);
-			continue;
-
-		if (!strncmp(cmd->prefix, handler->prefix, handler->length))
-		{
-			if (handler->cred <= user->credentials)
-			{
-				if (command_check_args(cmd, handler))
-				{
-					rc = handler->handler(hub, user, cmd);
-				}
-				else
-				{
-					rc = command_arg_mismatch(hub, user, cmd, handler);
-				}
-				command_destroy(cmd);
-				return rc;
-			}
-			else
-			{
-				rc = command_access_denied(hub, user, cmd);
-				command_destroy(cmd);
-				return rc;
-			}
-		}
-	}
-
-	command_not_found(hub, user, cmd);
-	command_destroy(cmd);
-	return 0;
 }
+
+static struct command_handle* add_builtin(const char* prefix, const char* args, enum auth_credentials cred, command_handler handler, const char* description)
+{
+	struct command_handle* handle = (struct command_handle*) hub_malloc_zero(sizeof(struct command_handle));
+	handle->prefix = strdup(prefix);
+	handle->length = strlen(prefix);
+	handle->args = args ? strdup(args) : NULL;
+	handle->cred = cred;
+	handle->handler = handler;
+	handle->description = strdup(description);
+	handle->command_origin = "built-in";
+	return handle;
+}
+
+#define ADD_COMMAND(PREFIX, LENGTH, ARGS, CREDENTIALS, FUNCTION, DESCRIPTION) \
+	command_add(cbase, add_builtin(PREFIX, ARGS, CREDENTIALS, FUNCTION, DESCRIPTION))
+
+void commands_builtin_add(struct command_base* cbase)
+{
+	ADD_COMMAND("ban",        3, "n", auth_cred_operator,  command_ban,      "Ban a user"                   );
+	ADD_COMMAND("broadcast",  9, "m", auth_cred_operator,  command_broadcast,"Send a message to all users"  );
+#ifdef CRASH_DEBUG
+	ADD_COMMAND("crash",      5, 0,   auth_cred_admin,     command_crash,    "Crash the hub (DEBUG)."       );
 #endif
+	ADD_COMMAND("getip",      5, "n", auth_cred_operator,  command_getip,    "Show IP address for a user"   );
+	ADD_COMMAND("help",       4, "?c",auth_cred_guest,     command_help,     "Show this help message."      );
+	ADD_COMMAND("history",    7, "?N",auth_cred_guest,     command_history,  "Show the last chat messages." );
+	ADD_COMMAND("kick",       4, "n", auth_cred_operator,  command_kick,     "Kick a user"                  );
+	ADD_COMMAND("log",        3, 0,   auth_cred_operator,  command_log,      "Display log"                  );
+	ADD_COMMAND("motd",       4, 0,   auth_cred_guest,     command_motd,     "Show the message of the day"  );
+	ADD_COMMAND("mute",       4, "n", auth_cred_operator,  command_mute,     "Mute user"                    );
+	ADD_COMMAND("myip",       4, 0,   auth_cred_guest,     command_myip,     "Show your own IP."            );
+	ADD_COMMAND("register",   8, "p", auth_cred_guest,     command_register, "Register your username."      );
+	ADD_COMMAND("reload",     6, 0,   auth_cred_admin,     command_reload,   "Reload configuration files."  );
+	ADD_COMMAND("rules",      5, 0,   auth_cred_guest,     command_rules,    "Show the hub rules"           );
+	ADD_COMMAND("password",   8, "p", auth_cred_user,      command_password, "Change your own password."    );
+	ADD_COMMAND("shutdown",   8, 0,   auth_cred_admin,     command_shutdown_hub, "Shutdown hub."                );
+	ADD_COMMAND("stats",      5, 0,   auth_cred_super,     command_stats,    "Show hub statistics."         );
+	ADD_COMMAND("unban",      5, "n", auth_cred_operator,  command_unban,    "Lift ban on a user"           );
+	ADD_COMMAND("unmute",     6, "n", auth_cred_operator,  command_mute,     "Unmute user"                  );
+	ADD_COMMAND("uptime",     6, 0,   auth_cred_guest,     command_uptime,   "Display hub uptime info."     );
+	ADD_COMMAND("useradd",    7, "np",auth_cred_operator,  command_useradd,  "Register a new user."         );
+	ADD_COMMAND("userdel",    7, "n", auth_cred_operator,  command_userdel,  "Delete a registered user."    );
+	ADD_COMMAND("userinfo",   8, "n", auth_cred_operator,  command_userinfo, "Show registered user info."   );
+	ADD_COMMAND("usermod",    7, "nC",auth_cred_admin,     command_usermod,  "Modify user credentials."     );
+	ADD_COMMAND("userpass",   8, "np",auth_cred_operator,  command_userpass, "Change password for a user."  );
+	ADD_COMMAND("version",    7, 0,   auth_cred_guest,     command_version,  "Show hub version info."       );
+	ADD_COMMAND("whoip",      5, "a", auth_cred_operator,  command_whoip,    "Show users matching IP range" );
+}
+
+void commands_builtin_remove(struct command_base* cbase)
+{
+	struct command_handle* command;
+	while ((command = list_get_first(cbase->handlers)))
+	{
+		command_del(cbase, command);
+	}
+}

src/core/commands.h

@@ -1,6 +1,6 @@
 /*
  * uhub - A tiny ADC p2p connection hub
- * Copyright (C) 2007-2010, Jan Vidar Krey
+ * Copyright (C) 2007-2011, Jan Vidar Krey
  *
  * This program is free software; you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
@@ -57,7 +57,7 @@ struct command_handle
 {
 	const char* prefix;				/**<<< "Command prefix, for instance 'help' would be the prefix for the !help command." */
 	size_t length;					/**<<< "Length of the prefix" */
-	const char* args;				/**<<< "Argument codes (see below)" */
+	const char* args;				/**<<< "Argument codes (see above)" */
 	enum auth_credentials cred;		/**<<< "Minimum access level for the command" */
 	command_handler handler;		/**<<< "Function pointer for the command" */
 	const char* description;		/**<<< "Description for the command" */
@@ -82,6 +82,11 @@ extern int command_add(struct command_base*, struct command_handle*);
  */
 extern int command_del(struct command_base*, struct command_handle*);
 
+/**
+ * Returns 1 if a command is available to a user (user has access to run it.)
+ */
+extern int command_is_available(struct command_handle*, struct hub_user* user);
+
 /**
  * Dispatch a message and forward it as a command.
  * Returns 1 if the message should be forwarded as a chat message, or 0 if

src/core/hub.c

@@ -813,7 +813,6 @@ struct hub_info* hub_start_service(struct hub_config* config)
 
 	// Start the hub command sub-system
 	hub->commands = command_initialize(hub);
-
 	return hub;
 }
 
@@ -974,37 +973,6 @@ void hub_free_variables(struct hub_info* hub)
 	adc_msg_free(hub->command_support);
 }
 
-
-/**
- * @return 1 if nickname is in use, or 0 if not used.
- */
-static int is_nick_in_use(struct hub_info* hub, const char* nick)
-{
-	struct hub_user* lookup = uman_get_user_by_nick(hub, nick);
-	if (lookup)
-	{
-		return 1;
-	}
-	return 0;
-}
-
-
-/**
- * @return 1 if CID is in use, or 0 if not used.
- */
-static int is_cid_in_use(struct hub_info* hub, const char* cid)
-{
-	struct hub_user* lookup = uman_get_user_by_cid(hub, cid);
-	if (lookup)
-	{
-		return 1;
-	}
-	return 0;
-}
-
-
-
-
 static void set_status_code(enum msg_status_level level, int code, char buffer[4])
 {
 	buffer[0] = ('0' + (int) level);

src/uhub.h

@@ -90,6 +90,7 @@ extern "C" {
 #include "core/pluginloader.h"
 #include "core/hub.h"
 #include "core/commands.h"
+#include "core/commands_builtin.h"
 #include "core/inf.h"
 #include "core/hubevent.h"
 #include "core/plugincallback.h"