mirror of https://github.com/ipxe/ipxe.git
394 lines
6.9 KiB
C
394 lines
6.9 KiB
C
#include "cmdlinelib.h"
|
|
#include <gpxe/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_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_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 "";
|
|
}
|
|
}
|
|
|
|
static int cmdl_exit = 0;
|
|
|
|
static int exit_exec ( int argc __unused, char **argv __unused ) {
|
|
cmdl_exit = 1;
|
|
return 0;
|
|
}
|
|
|
|
struct command exit_command __command = {
|
|
.name = "exit",
|
|
.exec = exit_exec,
|
|
};
|
|
|
|
void cmdl_enterloop(cmd_line* cmd)
|
|
{
|
|
cmdl_exit = 0;
|
|
do {
|
|
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());
|
|
} while ( ! cmdl_exit );
|
|
}
|
|
|
|
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');
|
|
system ( cmd->buffer );
|
|
cmdl_clearbuffer(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_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;
|
|
}
|
|
}
|
|
}
|
|
|
|
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->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 ( "Available commands: ");
|
|
for ( cmd = cmd_start ; cmd < cmd_end ; cmd++ ) {
|
|
printf("%s ", cmd->name);
|
|
}
|
|
printf("\n\n");
|
|
|
|
return this;
|
|
}
|
|
|
|
void cmdl_free(cmd_line* cmd)
|
|
{
|
|
free(cmd);
|
|
}
|
|
|