diff options
Diffstat (limited to 'src/commandline/cmdlinelib.c')
| -rw-r--r-- | src/commandline/cmdlinelib.c | 564 |
1 files changed, 564 insertions, 0 deletions
diff --git a/src/commandline/cmdlinelib.c b/src/commandline/cmdlinelib.c new file mode 100644 index 000000000..e38d729e3 --- /dev/null +++ b/src/commandline/cmdlinelib.c @@ -0,0 +1,564 @@ +#include "cmdlinelib.h" +#include "command.h" +#include <gpxe/tables.h> +#include <console.h> +#include <malloc.h> +#include <string.h> +#include <stdarg.h> + +static struct command cmd_start[0] __table_start ( commands ); +static struct command cmd_end[0] __table_end ( commands ); + +void cmdl_setgetchar(cmd_line* cmd, cmdl_getchar_t in) +{ + cmd->getchar = in; +} + +void cmdl_setputchar(cmd_line* cmd, cmdl_putchar_t in) +{ + cmd->putchar = in; +} + +void cmdl_setprintf(cmd_line* cmd, cmdl_printf_t in) +{ + cmd->printf = in; +} + +int cmdl_getexit(cmd_line* cmd) +{ + if(cmdl_check(cmd) && !cmd->exit){ + return 0; + }else{ + return 1; + } + +} + +void cmdl_setexit(cmd_line* cmd, int exit) +{ + if(cmdl_check(cmd)){ + cmd->exit = exit; + } +} + +int cmdl_printf(cmd_line* cmd, const char *format, ...) +{ + int ret; + char string[CMDL_BUFFER_SIZE]; + va_list ap; + + va_start(ap, format); + ret = vsprintf(string, format, ap); + cmdl_addstr(cmd, string); + va_end(ap); + return ret; +} + +void cmdl_addstr(cmd_line* cmd, char* str) +{ + unsigned int i; + for(i = 0; i < strlen(str); i++){ + cmdl_addchar(cmd, str[i]); + } +} + +/*void cmdl_addoutput_str(cmd_line* cmd, char output[CMDL_OUTPUT_SIZE]) +{ + if(cmdl_check(cmd) && output != NULL){ + if(!cmd->has_output){ + cmdl_clearoutput(cmd); + } + strncat(cmd->output, output, CMDL_OUTPUT_SIZE); + cmd->has_output = 1; + } +}*/ + +/*char* cmdl_getoutput(cmd_line* cmd) +{ + if(cmdl_check(cmd) && cmd->has_output){ + cmd->has_output = 0; + return cmd->output; + }else{ + return ""; + } +}*/ + +void cmdl_setpropmt(cmd_line* cmd, char prompt[CMDL_PROMPT_SIZE]) +{ + if(cmdl_check(cmd) && prompt != NULL){ + strncat(cmd->prompt, prompt, CMDL_PROMPT_SIZE); + } +} + +char *cmdl_getprompt(cmd_line* cmd) +{ + if(cmdl_check(cmd)){ + return cmd->prompt; + }else{ + return ""; + } +} + +char* cmdl_getbuffer(cmd_line* cmd){ + if(cmdl_check(cmd)){ + return cmd->buffer; + }else{ + return ""; + } +} + +void cmdl_enterloop(cmd_line* cmd) +{ + while(!cmdl_getexit(cmd)){ + if(cmd->refresh){ + cmd->printf("%s %s", cmd->prompt, cmd->buffer); + cmd->refresh = 0; + } +// cmd->printf("Got %d\n", cmd->getchar()); + cmdl_parsechar(cmd, cmd->getchar()); + } +} + +void cmdl_addreplace(cmd_line* cmd, char in) +{ + if(cmd->cursor < CMDL_BUFFER_SIZE - 2){ + cmd->buffer[cmd->cursor] = in; + cmd->cursor++; + cmd->putchar((int)in); + } +} + +void cmdl_addinsert(cmd_line* cmd, char in) +{ + int i; + int to; + if(cmd->cursor < CMDL_BUFFER_SIZE - 2 && cmd->cursor >= 0){ + if(strlen(cmd->buffer) < CMDL_BUFFER_SIZE - 2){ + to = strlen(cmd->buffer); + }else{ + to = CMDL_BUFFER_SIZE - 2; + } + for(i=to; i > cmd->cursor; i--){ + cmd->buffer[i] = cmd->buffer[i-1]; + } + cmd->buffer[cmd->cursor] = in; + + for(i=cmd->cursor; i < to; i++){ + cmd->putchar(cmd->buffer[i]); + } + + for(i=cmd->cursor; i < to - 1; i++){ + cmd->putchar(CMDLK_BS); + } + cmd->cursor++; + //cmdl_movecursor(cmd, CMDL_RIGHT); + } +} + +void cmdl_addchar(cmd_line* cmd, char in){ + if(cmd->insert){ + cmdl_addinsert(cmd, in); + }else{ + cmdl_addreplace(cmd, in); + } +} + +void cmdl_parsechar(cmd_line* cmd, char in) +{ + if(cmdl_check(cmd)){ + if(in >= 32){ + cmdl_addchar(cmd, in); + }else{ + switch(in){ + case CMDLK_BS: + if(cmdl_movecursor(cmd, CMDL_LEFT)){ + cmdl_del(cmd); + } + break; + + case CMDLK_RETURN: + cmd->putchar('\n'); + cmdl_exec(cmd); + cmd->refresh = 1; + break; + + case CMDLK_BW: + cmdl_movecursor(cmd, CMDL_LEFT); + break; + + case CMDLK_FW: + //cmdl_movecursor(cmd, CMDL_RIGHT); + break; + + case CMDLK_TAB: + cmdl_tabcomplete(cmd); + break; + + } + } + } +} + +void cmdl_tabcomplete(cmd_line *cmd) +{ + struct command *ccmd; + int count=0; + char* result[CMDL_MAX_TAB_COMPLETE_RESULT]; + + for ( ccmd = cmd_start ; ccmd < cmd_end ; ccmd++ ) { + if(!strncmp(ccmd->name, cmd->buffer, strlen(cmd->buffer))){ + if(count <= CMDL_MAX_TAB_COMPLETE_RESULT){ + result[count++] = (char*)(ccmd->name); + } + } + } + + + if( count == 1 ){ + cmdl_addstr(cmd, (char*)(result[0] + strlen(cmd->buffer))); + cmd->tabstate = 0; + cmdl_addchar(cmd, ' '); + } else if( count > 1 ) { + int i, i2, minlen=CMDL_BUFFER_SIZE, same=1; + char last; + + for(i = 0; i < count; i ++) { + if(minlen > (int)strlen( result[i] ) ){ + minlen = strlen(result[i]); + } + + } + if((int)strlen(cmd->buffer) < minlen){ + for(i = strlen(cmd->buffer); i < minlen; i++){ + last = result[0][i]; + for(i2 = 1; i2 < count; i2 ++) { + if(result[i2][i] != last){ + same = 0; + break; + } + } + if(same){ + cmdl_addchar(cmd, last); + } + + } + } + cmd->tabstate++; + } + + if(count > 1 && cmd->tabstate > 1){ + int i; + cmd->tabstate = 0; + cmd->refresh = 1; + cmd->putchar('\n'); + for(i = 0; i < count; i ++){ + cmd->printf("%s\t", result[i]); + } + cmd->putchar('\n'); + } + + + +} + + +void cmdl_exec(cmd_line* cmd) +{ + cmdl_param_list* params; + int unknown=1; + struct command *ccmd; + + params = cmdl_getparams(cmd->buffer); + + if(params == NULL){ + cmdl_clearbuffer(cmd); + return; + } + + if(params->argc > 0){ + if(!strcmp(params->argv[0], "exit") || !strcmp(params->argv[0], "quit")){ + cmdl_setexit(cmd, 1); +/* }else if(!strcmp(params->argv[0], "help")){ + if(params->argc > 1){ + cmdl_builtin_help(cmd, params->argv[1]); + }else{ + cmdl_builtin_help(cmd, ""); + }*/ + }else{ + for ( ccmd = cmd_start ; ccmd < cmd_end ; ccmd++ ) { + if(!strcmp(ccmd->name, params->argv[0])){ + unknown = 0; + ccmd->exec(params->argc, params->argv); + break; + } + } + if(unknown){ + cmd->printf("%s: unknown command\n", params->argv[0]); + } + } + } + + free(params); + cmdl_clearbuffer(cmd); +} + +/*void cmdl_builtin_help(cmd_line* cmd, char* command){ + struct command *ccmd; + int unknown = 1; + if(strcmp(command, "") == 0){ + cmd->printf("Built in commands:\n\n\thelp\t\tCommand usage help (\"help help\" for more info)\n\texit, quit\t\tExit the command line and boot\n\nCompiled in commands:\n\n"); + + for ( ccmd = cmd_start ; ccmd < cmd_end ; ccmd++ ) { + cmd->printf ("\t%s\t\t%s\n", ccmd->name, ccmd->desc ); + } + }else{ + if(!strcmp(command, "help")){ + cmd->printf("help - The help command\n\nUsage: help <command>\n\n\tExample:\n\t\thelp help\n"); + }else if(!strcmp(command, "exit") || !strcmp(command, "quit")){ + cmd->printf("exit, quit - The quit command\n\nUsage:\nquit or exit\n\n\tExample:\n\t\texit\n"); + }else{ + for ( ccmd = cmd_start ; ccmd < cmd_end ; ccmd++ ) { + if(!strcmp(ccmd->name, command)){ + unknown = 0; + cmd->printf ("\t%s - %s\n\nUsage:\n%s\n", ccmd->name, ccmd->desc, ccmd->usage ); + break; + } + if(unknown){ + cmd->printf("\"%s\" isn't compiled in (does it exist?).\n", command); + } + } + } + + } +}*/ + +cmdl_param_list* cmdl_getparams(const char* command){ + cmdl_param_list* this; + char *result = NULL; + int count=0; + char *command2; + + this = (cmdl_param_list*)malloc(sizeof(cmdl_param_list)); + + if(this == NULL){ + return NULL; + } + + command2 = malloc(strlen(command) + 1); + + this->argc=0; + + strcpy(command2, command); + result = strtok(command2, " "); + + while( result != NULL ) { + this->argc++; + result = strtok( NULL, " "); + } + + this->argv = (char**)malloc(sizeof(char*) * this->argc); + if(this->argv == NULL){ + free(this); + return NULL; + } + + + strcpy(command2, command); + result = strtok(command2, " "); + + while( result != NULL && this->argc > count) { + this->argv[count] = (char*)malloc(sizeof(char) * (strlen(result) + 1)); + if(this->argv[count] == NULL){ + free(this); + return NULL; + } + strcpy(this->argv[count], result); + count++; + result = strtok( NULL, " "); + } + free(command2); + return this; +} + +/*char* cmdl_parse_getcmd(cmd_line* cmd){ + int i; + char* ret; + ret = (char*)malloc(1); + ret[0] = 0; + + for(i=0; i < CMDL_BUFFER_SIZE - 1; i++){ + if(cmd->buffer[i + 1] == ' ' || cmd->buffer[i + 1] == '\0'){ + free(ret); + ret = (char*)malloc(i+1); + strncat(ret, cmd->buffer, i+1); + break; + } + } + return ret; +}*/ + +void cmdl_clearbuffer(cmd_line* cmd) +{ + if(cmdl_check(cmd)){ + int i; + cmd->cursor = 0; + for(i=0; i < CMDL_BUFFER_SIZE; i++){ + cmd->buffer[i] = 0; + } + } +} + +/*void cmdl_clearoutput(cmd_line* cmd) +{ + if(cmdl_check(cmd)){ + int i; + for(i=0; i < CMDL_OUTPUT_SIZE; i++){ + cmd->output[i] = 0; + } + } +}*/ + +int cmdl_movecursor(cmd_line* cmd, int direction) +{ + if(cmdl_check(cmd)){ + switch(direction){ + case CMDL_LEFT: + if(cmd->cursor > 0){ + cmd->cursor--; + cmd->putchar(CMDLK_BS); + }else{ + return 0; + } + break; + case CMDL_RIGHT: + if(cmd->cursor < CMDL_BUFFER_SIZE - 2){ + cmd->cursor++; + cmd->putchar(' '); + }else{ + return 0; + } + break; + } + } + return 1; +} + +void cmdl_del(cmd_line* cmd) +{ + if(cmdl_check(cmd) && cmd->cursor < CMDL_BUFFER_SIZE - 2 && cmd->cursor >= 0){ + int i; + for(i = cmd->cursor; i < (int)strlen(cmd->buffer); i++){ + cmd->buffer[i] = cmd->buffer[i + 1]; + if(!cmd->buffer[i]){ + cmd->putchar(' '); + }else{ + cmd->putchar(cmd->buffer[i]); + } + } + for(i = cmd->cursor; i < (int)strlen(cmd->buffer) + 1; i++){ + cmd->putchar(CMDLK_BS); + } + } +} + + +int cmdl_check(cmd_line* cmd) +{ + if( + cmd != NULL && + cmd->buffer != NULL && + cmd->prompt != NULL && + cmd->cursor >= 0 && + cmd->cursor < CMDL_BUFFER_SIZE - 1 && + cmd->buffer[CMDL_BUFFER_SIZE - 1] == 0 && + cmd->prompt[CMDL_PROMPT_SIZE - 1] == 0 + ){ + return 1; + }else{ + return 0; + } +} + +cmd_line* cmdl_create() +{ + cmd_line* this; + int i; + + /* Initiate the command line */ + + this = (cmd_line*)malloc(sizeof(cmd_line)); + + if(this == NULL){ + return NULL; + } + + + /* Allocate output buffer */ + + /*this->output = (char*)malloc(CMDL_OUTPUT_SIZE); + if(this->output == NULL){ + free(this); + return NULL; + }*/ + +/* for(i = 0; i < CMDL_OUTPUT_SIZE; i++){ + this->output[i] = 0; + }*/ + + /* Allocate command line buffer */ + + this->buffer = (char*)malloc(CMDL_BUFFER_SIZE); + if(this->buffer == NULL){ + free(this); + return NULL; + } + + for(i = 0; i < CMDL_BUFFER_SIZE; i++){ + this->buffer[i] = 0; + } + + /* Allocate prompt buffer */ + + this->prompt = (char*)malloc(CMDL_PROMPT_SIZE); + if(this->prompt == NULL){ + free(this); + return NULL; + } + + for(i = 0; i < CMDL_PROMPT_SIZE; i++){ + this->prompt[i] = 0; + } + + /* Initiate cursor position etc.*/ + + this->cursor = 0; + //this->has_output = 0; + this->exit = 0; + this->refresh = 1; + this->tabstate = 0; + this->insert = 0; + + /* set callbacks to NULL */ + + this->getchar = NULL; + this->putchar = NULL; + this->printf = NULL; + + /* List the commands */ + + struct command *cmd; + + printf ( "Compiled in commands: "); + for ( cmd = cmd_start ; cmd < cmd_end ; cmd++ ) { + printf("%s ", cmd->name); + } + printf("\n"); + + return this; +} + +void cmdl_free(cmd_line* cmd) +{ + free(cmd); +} + |