#if !defined(_XOPEN_SOURCE) && !defined(__FreeBSD__) && !defined(__NetBSD__) && !defined(__OpenBSD__) #define _XOPEN_SOURCE 600 #endif #include #include #include #include #include #define CMD_BUFLEN 1024 #ifndef WIN32 #include #include #include #else #define strdup(src) _strdup(src) #define fileno _fileno #define read _read #include #define CC_NORM 0 #define CC_NEWLINE 1 #define CC_EOF 2 #define CC_ARGHACK 3 #define CC_REFRESH 4 #define CC_CURSOR 5 #define CC_ERROR 6 #define CC_FATAL 7 #define CC_REDISPLAY 8 #define CC_REFRESH_BEEP 9 #define HISTLEN 10 #define KEY_UP 1 #define KEY_DOWN 2 #define KEY_TAB 3 #define CLEAR_OP 4 #define DELETE_REFRESH_OP 5 #define KEY_LEFT 6 #define KEY_RIGHT 7 #define KEY_INSERT 8 #define PROMPT_OP 9 static int console_bufferInput (char *buf, int len, char *cmd, int key); static unsigned char esl_console_complete(const char *buffer, const char *cursor, const char *lastchar); #endif #ifdef HAVE_EDITLINE #include #endif typedef struct { char name[128]; char host[128]; esl_port_t port; char user[256]; char pass[128]; int debug; const char *console_fnkeys[12]; char loglevel[128]; int quiet; char prompt_color[12]; char input_text_color[12]; char output_text_color[12]; } cli_profile_t; static int warn_stop = 0; static int connected = 0; static int allow_ctl_c = 0; static char bare_prompt_str[512] = ""; static int bare_prompt_str_len = 0; static char prompt_str[512] = ""; static char prompt_color[12] = {ESL_SEQ_DEFAULT_COLOR}; static char input_text_color[12] = {ESL_SEQ_DEFAULT_COLOR}; static char output_text_color[12] = {ESL_SEQ_DEFAULT_COLOR}; static cli_profile_t profiles[128] = {{{0}}}; static cli_profile_t internal_profile = {{ 0 }}; static int pcount = 0; static esl_handle_t *global_handle; static cli_profile_t *global_profile; static int running = 1; static int thread_running = 0; static char *filter_uuid; #ifndef WIN32 static EditLine *el; static History *myhistory; static HistEvent ev; #endif static void _sleep_ns(int secs, long nsecs) { #ifndef WIN32 if (nsecs > 999999999) { secs += nsecs/1000000000; nsecs = nsecs % 1000000000; } { struct timespec ts = { secs, nsecs }; nanosleep(&ts, NULL); } #else Sleep(secs*1000 + nsecs/1000000); #endif } static void sleep_ns(long nsecs) { _sleep_ns(0, nsecs); } static void sleep_ms(int msecs) { sleep_ns(msecs*1000000); } static void sleep_s(int secs) { _sleep_ns(secs, 0); } static int process_command(esl_handle_t *handle, const char *cmd); static void clear_cli(void) { putchar('\r'); printf("\033[%dC", bare_prompt_str_len); printf("\033[K"); fflush(stdout); } /* If a fnkey is configured then process the command */ static unsigned char console_fnkey_pressed(int i) { const char *c; assert((i > 0) && (i <= 12)); if (!(c = global_profile->console_fnkeys[i - 1])) { printf("\n"); esl_log(ESL_LOG_ERROR, "FUNCTION KEY F%d IS NOT BOUND, please edit your config.\n", i); return CC_REDISPLAY; } clear_cli(); printf("%s\n", c); if (process_command(global_handle, c)) { running = thread_running = 0; } return CC_REDISPLAY; } #ifdef HAVE_EDITLINE static char *prompt(EditLine *e) { return prompt_str; } static unsigned char console_f1key(EditLine *el, int ch) { return console_fnkey_pressed(1); } static unsigned char console_f2key(EditLine *el, int ch) { return console_fnkey_pressed(2); } static unsigned char console_f3key(EditLine *el, int ch) { return console_fnkey_pressed(3); } static unsigned char console_f4key(EditLine *el, int ch) { return console_fnkey_pressed(4); } static unsigned char console_f5key(EditLine *el, int ch) { return console_fnkey_pressed(5); } static unsigned char console_f6key(EditLine *el, int ch) { return console_fnkey_pressed(6); } static unsigned char console_f7key(EditLine *el, int ch) { return console_fnkey_pressed(7); } static unsigned char console_f8key(EditLine *el, int ch) { return console_fnkey_pressed(8); } static unsigned char console_f9key(EditLine *el, int ch) { return console_fnkey_pressed(9); } static unsigned char console_f10key(EditLine *el, int ch) { return console_fnkey_pressed(10); } static unsigned char console_f11key(EditLine *el, int ch) { return console_fnkey_pressed(11); } static unsigned char console_f12key(EditLine *el, int ch) { return console_fnkey_pressed(12); } static unsigned char console_eofkey(EditLine *el, int ch) { LineInfo *line; /* only exit if empty line */ line = (LineInfo *)el_line(el); if (line->buffer == line->lastchar) { printf("/exit\n\n"); running = thread_running = 0; return CC_EOF; } else { if (line->cursor != line->lastchar) { line->cursor++; el_deletestr(el, 1); } return CC_REDISPLAY; } } #else #ifdef _MSC_VER char history[HISTLEN][CMD_BUFLEN+1]; int iHistory = 0; int iHistorySel = 0; static int console_history (char *cmd, int direction) { int i; static int first; if (direction == 0) { first = 1; if (iHistory < HISTLEN) { if (iHistory && strcmp(history[iHistory-1], cmd)) { iHistorySel = iHistory; strcpy(history[iHistory++], cmd); } else if (iHistory == 0) { iHistorySel = iHistory; strcpy(history[iHistory++], cmd); } } else { iHistory = HISTLEN-1; for (i = 0; i < HISTLEN-1; i++) { strcpy(history[i], history[i+1]); } iHistorySel = iHistory; strcpy(history[iHistory++], cmd); } } else { if (!first) { iHistorySel += direction; } first = 0; if (iHistorySel < 0) { iHistorySel = 0; } if (iHistory && iHistorySel >= iHistory) { iHistorySel = iHistory-1; } strcpy(cmd, history[iHistorySel]); } return (0); } static int console_bufferInput (char *addchars, int len, char *cmd, int key) { static int iCmdBuffer = 0; static int iCmdCursor = 0; static int ignoreNext = 0; static int insertMode = 1; static COORD orgPosition; static char prompt [80]; int iBuf; int i; HANDLE hOut; CONSOLE_SCREEN_BUFFER_INFO info; COORD position; hOut = GetStdHandle(STD_OUTPUT_HANDLE); GetConsoleScreenBufferInfo(hOut, &info); position = info.dwCursorPosition; if (iCmdCursor == 0) { orgPosition = position; } if (key == PROMPT_OP) { if (strlen(cmd) < sizeof(prompt)) { strcpy(prompt, cmd); } return 0; } if (key == KEY_TAB) { esl_console_complete(cmd, cmd+iCmdBuffer, &cmd[iCmdBuffer-1]); return 0; } if (key == KEY_UP || key == KEY_DOWN || key == CLEAR_OP) { SetConsoleCursorPosition(hOut, orgPosition); for (i = 0; i < (int)strlen(cmd); i++) { printf(" "); } SetConsoleCursorPosition(hOut, orgPosition); iCmdBuffer = 0; iCmdCursor = 0; memset(cmd, 0, CMD_BUFLEN); } if (key == DELETE_REFRESH_OP) { int l = len < (int)strlen(cmd) ? len : (int)strlen(cmd); for (i = 0; i < l; i++) { cmd[--iCmdBuffer] = 0; } iCmdCursor = (int)strlen(cmd); printf("%s", prompt); GetConsoleScreenBufferInfo(hOut, &info); orgPosition = info.dwCursorPosition; printf("%s", cmd); return 0; } if (key == KEY_LEFT) { if (iCmdCursor) { if (position.X == 0) { position.Y -= 1; position.X = info.dwSize.X-1; } else { position.X -= 1; } SetConsoleCursorPosition(hOut, position); iCmdCursor--; } } if (key == KEY_RIGHT) { if (iCmdCursor < (int)strlen(cmd)) { if (position.X == info.dwSize.X-1) { position.Y += 1; position.X = 0; } else { position.X += 1; } SetConsoleCursorPosition(hOut, position); iCmdCursor++; } } if (key == KEY_INSERT) { insertMode = !insertMode; } for (iBuf = 0; iBuf < len; iBuf++) { switch (addchars[iBuf]) { case '\r': case '\n': if (ignoreNext) { ignoreNext = 0; } else { int ret = iCmdBuffer; if (iCmdBuffer == 0) { strcpy(cmd, "Empty"); ret = (int)strlen(cmd); } else { console_history(cmd, 0); cmd[iCmdBuffer] = 0; } iCmdBuffer = 0; iCmdCursor = 0; printf("\n"); return (ret); } break; case '\b': if (iCmdCursor) { if (position.X == 0) { position.Y -= 1; position.X = info.dwSize.X-1; SetConsoleCursorPosition(hOut, position); } else { position.X -= 1; SetConsoleCursorPosition(hOut, position); } printf(" "); if (iCmdCursor < iCmdBuffer) { int pos; iCmdCursor--; for (pos = iCmdCursor; pos < iCmdBuffer; pos++) { cmd[pos] = cmd[pos+1]; } cmd[pos] = 0; iCmdBuffer--; SetConsoleCursorPosition(hOut, position); for (pos = iCmdCursor; pos < iCmdBuffer; pos++) { printf("%c", cmd[pos]); } printf(" "); SetConsoleCursorPosition(hOut, position); } else { SetConsoleCursorPosition(hOut, position); iCmdBuffer--; iCmdCursor--; cmd[iCmdBuffer] = 0; } } break; default: if (!ignoreNext) { if (iCmdCursor < iCmdBuffer) { int pos; if (position.X == info.dwSize.X-1) { position.Y += 1; position.X = 0; } else { position.X += 1; } if (insertMode) { for (pos = iCmdBuffer-1; pos >= iCmdCursor; pos--) { cmd[pos+1] = cmd[pos]; } } iCmdBuffer++; cmd[iCmdCursor++] = addchars[iBuf]; printf("%c", addchars[iBuf]); for (pos = iCmdCursor; pos < iCmdBuffer; pos++) { GetConsoleScreenBufferInfo(hOut, &info); if (info.dwCursorPosition.X == info.dwSize.X-1 && info.dwCursorPosition.Y == info.dwSize.Y-1) { orgPosition.Y -= 1; position.Y -= 1; } printf("%c", cmd[pos]); } SetConsoleCursorPosition(hOut, position); } else { if (position.X == info.dwSize.X-1 && position.Y == info.dwSize.Y-1) { orgPosition.Y -= 1; } cmd[iCmdBuffer++] = addchars[iBuf]; iCmdCursor++; printf("%c", addchars[iBuf]); } } } if (iCmdBuffer == CMD_BUFLEN) { printf("Read Console... BUFFER OVERRUN\n"); iCmdBuffer = 0; ignoreNext = 1; } } return (0); } static BOOL console_readConsole(HANDLE conIn, char *buf, int len, int *pRed, int *key) { DWORD recordIndex, bufferIndex, toRead, red; PINPUT_RECORD pInput; GetNumberOfConsoleInputEvents(conIn, &toRead); if (len < (int)toRead) { toRead = len; } if (toRead == 0) { return(FALSE); } if ((pInput = (PINPUT_RECORD) malloc(toRead * sizeof(INPUT_RECORD))) == NULL) { return (FALSE); } *key = 0; ReadConsoleInput(conIn, pInput, toRead, &red); for (recordIndex = bufferIndex = 0; recordIndex < red; recordIndex++) { KEY_EVENT_RECORD keyEvent = pInput[recordIndex].Event.KeyEvent; if (pInput[recordIndex].EventType == KEY_EVENT && keyEvent.bKeyDown) { if (keyEvent.wVirtualKeyCode == 38 && keyEvent.wVirtualScanCode == 72) { buf[0] = 0; console_history(buf, -1); *key = KEY_UP; bufferIndex += (DWORD)strlen(buf); } if (keyEvent.wVirtualKeyCode == 40 && keyEvent.wVirtualScanCode == 80) { buf[0] = 0; console_history(buf, 1); *key = KEY_DOWN; bufferIndex += (DWORD)strlen(buf); } if (keyEvent.wVirtualKeyCode == 112 && keyEvent.wVirtualScanCode == 59) { console_fnkey_pressed(1); } if (keyEvent.wVirtualKeyCode == 113 && keyEvent.wVirtualScanCode == 60) { console_fnkey_pressed(2); } if (keyEvent.wVirtualKeyCode == 114 && keyEvent.wVirtualScanCode == 61) { console_fnkey_pressed(3); } if (keyEvent.wVirtualKeyCode == 115 && keyEvent.wVirtualScanCode == 62) { console_fnkey_pressed(4); } if (keyEvent.wVirtualKeyCode == 116 && keyEvent.wVirtualScanCode == 63) { console_fnkey_pressed(5); } if (keyEvent.wVirtualKeyCode == 117 && keyEvent.wVirtualScanCode == 64) { console_fnkey_pressed(6); } if (keyEvent.wVirtualKeyCode == 118 && keyEvent.wVirtualScanCode == 65) { console_fnkey_pressed(7); } if (keyEvent.wVirtualKeyCode == 119 && keyEvent.wVirtualScanCode == 66) { console_fnkey_pressed(8); } if (keyEvent.wVirtualKeyCode == 120 && keyEvent.wVirtualScanCode == 67) { console_fnkey_pressed(9); } if (keyEvent.wVirtualKeyCode == 121 && keyEvent.wVirtualScanCode == 68) { console_fnkey_pressed(10); } if (keyEvent.wVirtualKeyCode == 122 && keyEvent.wVirtualScanCode == 87) { console_fnkey_pressed(11); } if (keyEvent.wVirtualKeyCode == 123 && keyEvent.wVirtualScanCode == 88) { console_fnkey_pressed(12); } if (keyEvent.uChar.AsciiChar == 9) { *key = KEY_TAB; break; } if (keyEvent.uChar.AsciiChar == 27) { *key = CLEAR_OP; break; } if (keyEvent.wVirtualKeyCode == 37 && keyEvent.wVirtualScanCode == 75) { *key = KEY_LEFT; } if (keyEvent.wVirtualKeyCode == 39 && keyEvent.wVirtualScanCode == 77) { *key = KEY_RIGHT; } if (keyEvent.wVirtualKeyCode == 45 && keyEvent.wVirtualScanCode == 82) { *key = KEY_INSERT; } while (keyEvent.wRepeatCount && keyEvent.uChar.AsciiChar) { buf[bufferIndex] = keyEvent.uChar.AsciiChar; if (buf[bufferIndex] == '\r') { buf[bufferIndex] = '\n'; } bufferIndex++; keyEvent.wRepeatCount--; } } } free(pInput); *pRed = bufferIndex; return (TRUE); } #endif #endif static void handle_SIGINT(int sig) { if (!connected || allow_ctl_c) { fprintf(stdout, "Interrupted.\n"); exit(1); } warn_stop = 1; signal(SIGINT, handle_SIGINT); #ifdef SIGTSTP signal(SIGTSTP, handle_SIGINT); #endif return; } static void handle_SIGQUIT(int sig) { fprintf(stdout, "Caught SIGQUIT\n"); return; } #ifdef WIN32 static HANDLE hStdout; static WORD wOldColorAttrs; static CONSOLE_SCREEN_BUFFER_INFO csbiInfo; #endif #ifdef WIN32 static WORD colors[] #else static const char *colors[] #endif = { ESL_SEQ_DEFAULT_COLOR, ESL_SEQ_FRED, ESL_SEQ_FRED, ESL_SEQ_FRED, ESL_SEQ_FMAGEN, ESL_SEQ_FCYAN, ESL_SEQ_FGREEN, ESL_SEQ_FYELLOW }; static const char *usage_str = "Usage: %s [-H ] [-P ] [-p ] [-d ] [-x command] [-t ] [profile]\n\n" " -?,-h --help Usage Information\n" " -H, --host=hostname Host to connect\n" " -P, --port=port Port to connect (1 - 65535)\n" " -u, --user=user@domain user@domain\n" " -p, --password=password Password\n" " -i, --interrupt Allow Control-c to interrupt\n" " -x, --execute=command Execute Command and Exit\n" " -l, --loglevel=command Log Level\n" " -q, --quiet Disable logging\n" " -r, --retry Retry connection on failure\n" " -R, --reconnect Reconnect if disconnected\n" " -d, --debug=level Debug Level (0 - 7)\n" " -t, --timeout Timeout for API commands (in miliseconds)\n\n"; static int usage(char *name){ printf(usage_str, name); return 1; } static int stdout_writable(void) { #ifndef WIN32 fd_set set; int fd = fileno(stdout); struct timeval to; memset(&to, 0, sizeof(to)); FD_ZERO(&set); FD_SET(fd, &set); to.tv_sec = 0; to.tv_usec = 100000; if (select(fd + 1, NULL, &set, NULL, &to) > 0) { return FD_ISSET(fd, &set); } else { return 0; } #else return 1; #endif } #ifndef WIN32 static int write_str(const char *s) { int n, left = strlen(s); while (1) { n = write(STDOUT_FILENO, s, left); if (n == left) return 1; if (n < 0) return 0; s += n; left -= n; } return 1; } static int write_char(int c) { char s[2] = { c, 0 }; return write_str(s); } #endif #ifdef WIN32 static void clear_line(void) { putchar('\r'); printf("\033[K"); fflush(stdout); return; } #else static void clear_line(void) { if (!(write_char('\r'))) goto done; if (!(write_str("\033[K"))) goto done; done: return; } #endif static void redisplay(void) { #ifdef WIN32 #else const LineInfo *lf = el_line(el); const char *c = lf->buffer; if (!(write_str(prompt_str))) goto done; while (c < lf->lastchar && *c) { if (!(write_char(*c))) goto done; c++; } { int pos = (int)(lf->cursor - lf->buffer); char s1[12], s2[12]; if (!(write_char('\r'))) goto done; snprintf(s1, sizeof(s1), "\033[%dC", bare_prompt_str_len); snprintf(s2, sizeof(s2), "\033[%dC", pos); if (!(write_str(s1))) goto done; if (pos > 0) if (!(write_str(s2))) goto done; } done: return; #endif } static int output_printf(const char *fmt, ...) { va_list ap; int r; va_start(ap, fmt); #ifndef WIN32 printf("%s", output_text_color); #endif r = vprintf(fmt, ap); va_end(ap); return r; } static void *msg_thread_run(esl_thread_t *me, void *obj) { esl_handle_t *handle = (esl_handle_t *) obj; thread_running = 1; while(thread_running && handle->connected) { int aok = 1; esl_status_t status = esl_recv_event_timed(handle, 10, 1, NULL); if (status == ESL_FAIL) { esl_log(ESL_LOG_WARNING, "Disconnected.\n"); running = -1; thread_running = 0; } else if (status == ESL_SUCCESS) { aok = stdout_writable(); if (handle->last_event) { int known = 1; const char *type = esl_event_get_header(handle->last_event, "content-type"); if (!esl_strlen_zero(type)) { if (!strcasecmp(type, "log/data")) { const char *userdata = esl_event_get_header(handle->last_event, "user-data"); if (esl_strlen_zero(userdata) || esl_strlen_zero(filter_uuid) || !strcasecmp(filter_uuid, userdata)) { int level = 0; const char *lname = esl_event_get_header(handle->last_event, "log-level"); #ifdef WIN32 DWORD len = (DWORD) strlen(handle->last_event->body); DWORD outbytes = 0; #endif if (lname) { level = atoi(lname); } #ifndef WIN32 if (aok) { clear_line(); printf("%s%s", colors[level], handle->last_event->body); redisplay(); } #else if (aok) { SetConsoleTextAttribute(hStdout, colors[level]); WriteFile(hStdout, handle->last_event->body, len, &outbytes, NULL); SetConsoleTextAttribute(hStdout, wOldColorAttrs); } #endif } } else if (!strcasecmp(type, "text/disconnect-notice")) { running = -1; thread_running = 0; } else if (!strcasecmp(type, "text/event-plain")) { char *s; esl_event_serialize(handle->last_ievent, &s, ESL_FALSE); if (aok) { clear_line(); output_printf("RECV EVENT\n%s\n", s); redisplay(); } free(s); } else { known = 0; } } if (aok && !known) { char *s; output_printf("INCOMING DATA [%s]\n%s\n", type, handle->last_event->body ? handle->last_event->body : ""); esl_event_serialize(handle->last_event, &s, ESL_FALSE); output_printf("RECV EVENT\n%s\n", s); redisplay(); free(s); } } } if (warn_stop) { if (aok) { clear_line(); output_printf("Type control-D or /exit or /quit or /bye to exit.\n\n"); redisplay(); } warn_stop = 0; } sleep_ms(1); } thread_running = 0; esl_log(ESL_LOG_DEBUG, "Thread Done\n"); return NULL; } static const char *cli_usage = "Command \tDescription\n" "-----------------------------------------------\n" "/help \tHelp\n" "/exit, /quit, /bye, ... \tExit the program.\n" "/event, /noevents, /nixevent\tEvent commands.\n" "/log, /nolog \tLog commands.\n" "/uuid \tFilter logs for a single call uuid\n" "/filter \tFilter commands.\n" "/debug [0-7] \tSet debug level.\n" "\n"; static int process_command(esl_handle_t *handle, const char *cmd) { while (*cmd == ' ') cmd++; if ((*cmd == '/' && cmd++) || !strncasecmp(cmd, "...", 3)) { if (!strcasecmp(cmd, "help")) { output_printf("%s", cli_usage); goto end; } if (!strcasecmp(cmd, "exit") || !strcasecmp(cmd, "quit") || !strcasecmp(cmd, "...") || !strcasecmp(cmd, "bye") ) { esl_log(ESL_LOG_INFO, "Goodbye!\nSee you at ClueCon http://www.cluecon.com/\n"); return -1; } else if (!strncasecmp(cmd, "uuid", 4)) { cmd += 4; while (*cmd && *cmd == ' ') { cmd++; } if (!esl_strlen_zero(cmd)) { filter_uuid = strdup(cmd); } else { esl_safe_free(filter_uuid); } output_printf("UUID filtering %s\n", filter_uuid ? "enabled" : "disabled"); } else if (!strncasecmp(cmd, "event", 5) || !strncasecmp(cmd, "noevents", 8) || !strncasecmp(cmd, "nixevent", 8) || !strncasecmp(cmd, "log", 3) || !strncasecmp(cmd, "nolog", 5) || !strncasecmp(cmd, "filter", 6) ) { esl_send_recv(handle, cmd); printf("%s\n", handle->last_sr_reply); } else if (!strncasecmp(cmd, "debug", 5)) { int tmp_debug = atoi(cmd+6); if (tmp_debug > -1 && tmp_debug < 8) { esl_global_set_default_logger(tmp_debug); output_printf("fs_cli debug level set to %d\n", tmp_debug); } else { output_printf("fs_cli debug level must be 0 - 7\n"); } } else { output_printf("Unknown command [%s]\n", cmd); } } else { char cmd_str[1024] = ""; const char *err = NULL; snprintf(cmd_str, sizeof(cmd_str), "api %s\nconsole_execute: true\n\n", cmd); if (esl_send_recv(handle, cmd_str)) { output_printf("Socket interrupted, bye!\n"); return -1; } if (handle->last_sr_event) { if (handle->last_sr_event->body) { output_printf("%s\n", handle->last_sr_event->body); } else if ((err = esl_event_get_header(handle->last_sr_event, "reply-text")) && !strncasecmp(err, "-err", 3)) { output_printf("Error: %s!\n", err + 4); } } } end: return 0; } static int get_profile(const char *name, cli_profile_t **profile) { int x; for (x = 0; x < pcount; x++) { if (!strcmp(profiles[x].name, name)) { *profile = &profiles[x]; return 0; } } return -1; } #ifndef HAVE_EDITLINE static char command_buf[CMD_BUFLEN+1] = ""; static const char *basic_gets(int *cnt) { #ifndef _MSC_VER int x = 0; printf("%s", prompt_str); memset(&command_buf, 0, sizeof(command_buf)); for (x = 0; x < (sizeof(command_buf) - 1); x++) { int c = getchar(); if (c < 0) { int y = read(fileno(stdin), command_buf, sizeof(command_buf) - 1); command_buf[y - 1] = '\0'; break; } command_buf[x] = (char) c; if (command_buf[x] == '\n') { command_buf[x] = '\0'; break; } } *cnt = x; #else int read, key; char keys[CMD_BUFLEN]; HANDLE stdinHandle = GetStdHandle(STD_INPUT_HANDLE); console_bufferInput (0, 0, prompt_str, PROMPT_OP); printf("%s", prompt_str); *cnt = 0; memset(&command_buf, 0, sizeof(command_buf)); while (!*cnt) { if (console_readConsole(stdinHandle, keys, (int)sizeof(keys), &read, &key)) { *cnt = console_bufferInput(keys, read, command_buf, key); if (!strcmp(command_buf, "Empty")) { command_buf[0] = 0; } } sleep_ms(20); } #endif return command_buf; } #endif static const char *banner = " _____ ____ ____ _ ___ \n" " | ___/ ___| / ___| | |_ _| \n" " | |_ \\___ \\ | | | | | | \n" " | _| ___) | | |___| |___ | | \n" " |_| |____/ \\____|_____|___| \n" "\n" "*******************************************************\n" "* Anthony Minessale II, Ken Rice, *\n" "* Michael Jerris, Travis Cross *\n" "* FreeSWITCH (http://www.freeswitch.org) *\n" "* Paypal Donations Appreciated: paypal@freeswitch.org *\n" "* Brought to you by ClueCon http://www.cluecon.com/ *\n" "*******************************************************\n" "\n" "Type /help to see a list of commands\n\n\n"; static void print_banner(FILE *stream) { fprintf(stream, "%s%s", output_text_color, banner); } static void set_fn_keys(cli_profile_t *profile) { profile->console_fnkeys[0] = "help"; profile->console_fnkeys[1] = "status"; profile->console_fnkeys[2] = "show channels"; profile->console_fnkeys[3] = "show calls"; profile->console_fnkeys[4] = "sofia status"; profile->console_fnkeys[5] = "reloadxml"; profile->console_fnkeys[6] = "/log console"; profile->console_fnkeys[7] = "/log debug"; profile->console_fnkeys[8] = "sofia status profile internal"; profile->console_fnkeys[9] = "fsctl pause"; profile->console_fnkeys[10] = "fsctl resume"; profile->console_fnkeys[11] = "version"; } static char* end_of_str(char *s) { return (*s == '\0' ? s : s + strlen(s) - 1); } static char* _strndup(const char *s, int n) { char *r = (char*)malloc(n + 1), *d=r; while (n > 0 && *s) { *d = *s; d++; s++; n--; } *d = 0; return r; } static unsigned char esl_console_complete(const char *buffer, const char *cursor, const char *lastchar) { char cmd_str[2048] = ""; unsigned char ret = CC_REDISPLAY; char *dup = _strndup(buffer, (int)(lastchar - buffer)); char *buf = dup; int sc = 0, offset = (int)(cursor - buffer), pos = (offset > 0) ? offset : 0; char *p; if (pos > 0) { *(buf + pos) = '\0'; } if ((p = strchr(buf, '\r')) || (p = strchr(buf, '\n'))) { *p = '\0'; } while (*buf == ' ') { buf++; sc++; } #ifdef HAVE_EDITLINE if (!*buf && sc) { el_deletestr(el, sc); sc = 0; } #endif p = end_of_str(buf); while(p >= buf && *p == ' ') { sc++; p--; } #ifdef HAVE_EDITLINE if (sc > 1) { el_deletestr(el, sc - 1); *(p + 2) = '\0'; } #endif if (*cursor) { snprintf(cmd_str, sizeof(cmd_str), "api console_complete c=%ld;%s\n\n", (long)pos, buf); } else { snprintf(cmd_str, sizeof(cmd_str), "api console_complete %s\n\n", buf); } esl_send_recv(global_handle, cmd_str); if (global_handle->last_sr_event && global_handle->last_sr_event->body) { char *r = global_handle->last_sr_event->body; char *w, *p1; if (r) { if ((w = strstr(r, "\n\nwrite="))) { int len = 0; *w = '\0'; w += 8; len = atoi(w); if ((p1= strchr(w, ':'))) { w = p1+ 1; } printf("%s\n\n\n", r); #ifdef HAVE_EDITLINE el_deletestr(el, len); el_insertstr(el, w); #else #ifdef _MSC_VER console_bufferInput(0, len, (char*)buffer, DELETE_REFRESH_OP); console_bufferInput(w, (int)strlen(w), (char*)buffer, 0); #endif #endif } else { printf("%s\n", r); #ifdef _MSC_VER console_bufferInput(0, 0, (char*)buffer, DELETE_REFRESH_OP); #endif } } fflush(stdout); } esl_safe_free(dup); return ret; } #ifdef HAVE_EDITLINE static unsigned char complete(EditLine *el, int ch) { const LineInfo *lf = el_line(el); return esl_console_complete(lf->buffer, lf->cursor, lf->lastchar); } #endif struct color_map_el { char name[32]; char seq[12]; }; struct color_map_el color_map[] = { {"black", ESL_SEQ_FBLACK}, {"bold-black", ESL_SEQ_BBLACK}, {"red", ESL_SEQ_FRED}, {"bold-red", ESL_SEQ_BRED}, {"green", ESL_SEQ_FGREEN}, {"bold-green", ESL_SEQ_BGREEN}, {"yellow", ESL_SEQ_FYELLOW}, {"bold-yellow", ESL_SEQ_BYELLOW}, {"blue", ESL_SEQ_FBLUE}, {"bold-blue", ESL_SEQ_BBLUE}, {"magenta", ESL_SEQ_FMAGEN}, {"bold-magenta", ESL_SEQ_BMAGEN}, {"cyan", ESL_SEQ_FCYAN}, {"bold-cyan", ESL_SEQ_BCYAN}, {"white", ESL_SEQ_FWHITE}, {"bold-white", ESL_SEQ_BWHITE}, {{0}}}; static const char* match_color(const char *s) { struct color_map_el *map = color_map; while (map->name) { if (!(strcasecmp(s, map->name))) { return map->seq; } map++; } #ifdef WIN32 return "white"; #else return ESL_SEQ_DEFAULT_COLOR; #endif } static void read_config(const char *dft_cfile, const char *cfile) { esl_config_t cfg; if (esl_config_open_file(&cfg, cfile) || esl_config_open_file(&cfg, dft_cfile)) { char *var, *val; char cur_cat[128] = ""; while (esl_config_next_pair(&cfg, &var, &val)) { if (strcmp(cur_cat, cfg.category)) { esl_set_string(cur_cat, cfg.category); esl_set_string(profiles[pcount].name, cur_cat); esl_set_string(profiles[pcount].host, "localhost"); esl_set_string(profiles[pcount].pass, "ClueCon"); profiles[pcount].port = 8021; set_fn_keys(&profiles[pcount]); esl_set_string(profiles[pcount].prompt_color, prompt_color); esl_set_string(profiles[pcount].input_text_color, input_text_color); esl_set_string(profiles[pcount].output_text_color, output_text_color); esl_log(ESL_LOG_DEBUG, "Found Profile [%s]\n", profiles[pcount].name); pcount++; } if (!strcasecmp(var, "host")) { esl_set_string(profiles[pcount-1].host, val); } else if (!strcasecmp(var, "user")) { esl_set_string(profiles[pcount-1].user, val); } else if (!strcasecmp(var, "password")) { esl_set_string(profiles[pcount-1].pass, val); } else if (!strcasecmp(var, "port")) { int pt = atoi(val); if (pt > 0) { profiles[pcount-1].port = (esl_port_t)pt; } } else if (!strcasecmp(var, "debug")) { int dt = atoi(val); if (dt > -1 && dt < 8){ profiles[pcount-1].debug = dt; } } else if(!strcasecmp(var, "loglevel")) { esl_set_string(profiles[pcount-1].loglevel, val); } else if(!strcasecmp(var, "quiet")) { profiles[pcount-1].quiet = esl_true(val); } else if(!strcasecmp(var, "prompt-color")) { esl_set_string(profiles[pcount-1].prompt_color, match_color(val)); } else if(!strcasecmp(var, "input-text-color")) { esl_set_string(profiles[pcount-1].input_text_color, match_color(val)); } else if(!strcasecmp(var, "output-text-color")) { esl_set_string(profiles[pcount-1].output_text_color, match_color(val)); } else if (!strncasecmp(var, "key_F", 5)) { char *key = var + 5; if (key) { int i = atoi(key); if (i > 0 && i < 13) { profiles[pcount-1].console_fnkeys[i - 1] = strdup(val); } } } } esl_config_close_file(&cfg); } } static void clear_el_buffer(void) { #ifdef HAVE_EDITLINE const LineInfo *lf = el_line(el); int len = (int)(lf->lastchar - lf->buffer); el_deletestr(el, len); memset((char*)lf->buffer, 0, len); #endif } int main(int argc, char *argv[]) { esl_handle_t handle = {{0}}; int count = 0; const char *line = NULL; char cmd_str[1024] = ""; cli_profile_t *profile = NULL; #ifndef WIN32 char hfile[512] = "/tmp/fs_cli_history"; char cfile[512] = "/etc/fs_cli.conf"; char dft_cfile[512] = "/etc/fs_cli.conf"; #else char hfile[512] = "fs_cli_history"; char cfile[512] = "fs_cli.conf"; char dft_cfile[512] = "fs_cli.conf"; #endif char *home = getenv("HOME"); /* Vars for optargs */ int opt; static struct option options[] = { {"help", 0, 0, 'h'}, {"host", 1, 0, 'H'}, {"port", 1, 0, 'P'}, {"user", 1, 0, 'u'}, {"password", 1, 0, 'p'}, {"debug", 1, 0, 'd'}, {"execute", 1, 0, 'x'}, {"loglevel", 1, 0, 'l'}, {"quiet", 0, 0, 'q'}, {"retry", 0, 0, 'r'}, {"interrupt", 0, 0, 'i'}, {"reconnect", 0, 0, 'R'}, {"timeout", 1, 0, 't'}, {0, 0, 0, 0} }; char temp_host[128]; int argv_host = 0; char temp_user[256]; char temp_pass[128]; int argv_pass = 0 ; int argv_user = 0 ; int temp_port = 0; int argv_port = 0; int temp_log = -1; int argv_error = 0; int argv_exec = 0; char argv_command[1024] = ""; char argv_loglevel[128] = ""; int argv_quiet = 0; int loops = 2, reconnect = 0, timeout = 0; strncpy(internal_profile.host, "127.0.0.1", sizeof(internal_profile.host)); strncpy(internal_profile.pass, "ClueCon", sizeof(internal_profile.pass)); strncpy(internal_profile.name, "internal", sizeof(internal_profile.name)); internal_profile.port = 8021; set_fn_keys(&internal_profile); esl_set_string(internal_profile.prompt_color, prompt_color); esl_set_string(internal_profile.input_text_color, input_text_color); esl_set_string(internal_profile.output_text_color, output_text_color); if (home) { snprintf(hfile, sizeof(hfile), "%s/.fs_cli_history", home); snprintf(cfile, sizeof(cfile), "%s/.fs_cli_conf", home); } signal(SIGINT, handle_SIGINT); #ifdef SIGTSTP signal(SIGTSTP, handle_SIGINT); #endif #ifdef SIGQUIT signal(SIGQUIT, handle_SIGQUIT); #endif esl_global_set_default_logger(6); /* default debug level to 6 (info) */ for(;;) { int option_index = 0; opt = getopt_long(argc, argv, "H:U:P:S:u:p:d:x:l:t:qrRhi?", options, &option_index); if (opt == -1) break; switch (opt) { case 'H': esl_set_string(temp_host, optarg); argv_host = 1; break; case 'P': temp_port= atoi(optarg); if (temp_port > 0 && temp_port < 65536) { argv_port = 1; } else { printf("ERROR: Port must be in range 1 - 65535\n"); argv_error = 1; } break; case 'u': esl_set_string(temp_user, optarg); argv_user = 1; break; case 'p': esl_set_string(temp_pass, optarg); argv_pass = 1; break; case 'd': temp_log=atoi(optarg); if (temp_log < 0 || temp_log > 7) { printf("ERROR: Debug level should be 0 - 7.\n"); argv_error = 1; } else { esl_global_set_default_logger(temp_log); } break; case 'x': argv_exec = 1; esl_set_string(argv_command, optarg); break; case 'l': esl_set_string(argv_loglevel, optarg); break; case 'q': argv_quiet = 1; break; case 'i': allow_ctl_c = 1; break; case 'r': loops += 120; break; case 'R': reconnect = 1; break; case 't': timeout = atoi(optarg); break; case 'h': case '?': print_banner(stdout); usage(argv[0]); return 0; default: opt = 0; } } if (argv_error) { printf("\n"); return usage(argv[0]); } read_config(dft_cfile, cfile); if (optind < argc) { get_profile(argv[optind], &profile); } if (!profile) { if (get_profile("default", &profile)) { esl_log(ESL_LOG_DEBUG, "profile default does not exist using builtin profile\n"); profile = &internal_profile; } } if (temp_log < 0 ) { esl_global_set_default_logger(profile->debug); } if (argv_host) { esl_set_string(profile->host, temp_host); } if (argv_port) { profile->port = (esl_port_t)temp_port; } if (argv_user) { esl_set_string(profile->user, temp_user); } if (argv_pass) { esl_set_string(profile->pass, temp_pass); } if (*argv_loglevel) { esl_set_string(profile->loglevel, argv_loglevel); profile->quiet = 0; } esl_log(ESL_LOG_DEBUG, "Using profile %s [%s]\n", profile->name, profile->host); esl_set_string(prompt_color, profile->prompt_color); esl_set_string(input_text_color, profile->input_text_color); esl_set_string(output_text_color, profile->output_text_color); if (argv_host) { if (argv_port && profile->port != 8021) { snprintf(bare_prompt_str, sizeof(bare_prompt_str), "freeswitch@%s:%u@%s> ", profile->host, profile->port, profile->name); } else { snprintf(bare_prompt_str, sizeof(bare_prompt_str), "freeswitch@%s@%s> ", profile->host, profile->name); } } else { snprintf(bare_prompt_str, sizeof(bare_prompt_str), "freeswitch@%s> ", profile->name); } bare_prompt_str_len = (int)strlen(bare_prompt_str); #ifdef WIN32 snprintf(prompt_str, sizeof(prompt_str), "%s", bare_prompt_str); /* Not supporting this for now */ #else snprintf(prompt_str, sizeof(prompt_str), "%s%s%s", prompt_color, bare_prompt_str, input_text_color); #endif connect: connected = 0; while (--loops > 0) { memset(&handle, 0, sizeof(handle)); if (esl_connect(&handle, profile->host, profile->port, profile->user, profile->pass)) { esl_global_set_default_logger(7); esl_log(ESL_LOG_ERROR, "Error Connecting [%s]\n", handle.err); if (loops == 1) { if (!argv_exec) usage(argv[0]); return -1; } else { sleep_s(1); esl_log(ESL_LOG_INFO, "Retrying\n"); } } else { connected = 1; if (temp_log < 0 ) { esl_global_set_default_logger(profile->debug); } else { esl_global_set_default_logger(temp_log); } break; } } if (argv_exec) { const char *err = NULL; snprintf(cmd_str, sizeof(cmd_str), "api %s\n\n", argv_command); if (timeout) { esl_status_t status = esl_send_recv_timed(&handle, cmd_str, timeout); if (status != ESL_SUCCESS) { printf("Request timed out.\n"); esl_disconnect(&handle); return -2; } } else { esl_send_recv(&handle, cmd_str); } if (handle.last_sr_event) { if (handle.last_sr_event->body) { printf("%s\n", handle.last_sr_event->body); } else if ((err = esl_event_get_header(handle.last_sr_event, "reply-text")) && !strncasecmp(err, "-err", 3)) { printf("Error: %s!\n", err + 4); } } esl_disconnect(&handle); return 0; } global_handle = &handle; global_profile = profile; esl_thread_create_detached(msg_thread_run, &handle); #ifdef HAVE_EDITLINE el = el_init(__FILE__, stdin, stdout, stderr); el_set(el, EL_PROMPT, &prompt); el_set(el, EL_EDITOR, "emacs"); el_set(el, EL_ADDFN, "f1-key", "F1 KEY PRESS", console_f1key); el_set(el, EL_ADDFN, "f2-key", "F2 KEY PRESS", console_f2key); el_set(el, EL_ADDFN, "f3-key", "F3 KEY PRESS", console_f3key); el_set(el, EL_ADDFN, "f4-key", "F4 KEY PRESS", console_f4key); el_set(el, EL_ADDFN, "f5-key", "F5 KEY PRESS", console_f5key); el_set(el, EL_ADDFN, "f6-key", "F6 KEY PRESS", console_f6key); el_set(el, EL_ADDFN, "f7-key", "F7 KEY PRESS", console_f7key); el_set(el, EL_ADDFN, "f8-key", "F8 KEY PRESS", console_f8key); el_set(el, EL_ADDFN, "f9-key", "F9 KEY PRESS", console_f9key); el_set(el, EL_ADDFN, "f10-key", "F10 KEY PRESS", console_f10key); el_set(el, EL_ADDFN, "f11-key", "F11 KEY PRESS", console_f11key); el_set(el, EL_ADDFN, "f12-key", "F12 KEY PRESS", console_f12key); el_set(el, EL_ADDFN, "EOF-key", "EOF (^D) KEY PRESS", console_eofkey); el_set(el, EL_BIND, "\033OP", "f1-key", NULL); el_set(el, EL_BIND, "\033OQ", "f2-key", NULL); el_set(el, EL_BIND, "\033OR", "f3-key", NULL); el_set(el, EL_BIND, "\033OS", "f4-key", NULL); el_set(el, EL_BIND, "\033[11~", "f1-key", NULL); el_set(el, EL_BIND, "\033[12~", "f2-key", NULL); el_set(el, EL_BIND, "\033[13~", "f3-key", NULL); el_set(el, EL_BIND, "\033[14~", "f4-key", NULL); el_set(el, EL_BIND, "\033[15~", "f5-key", NULL); el_set(el, EL_BIND, "\033[17~", "f6-key", NULL); el_set(el, EL_BIND, "\033[18~", "f7-key", NULL); el_set(el, EL_BIND, "\033[19~", "f8-key", NULL); el_set(el, EL_BIND, "\033[20~", "f9-key", NULL); el_set(el, EL_BIND, "\033[21~", "f10-key", NULL); el_set(el, EL_BIND, "\033[23~", "f11-key", NULL); el_set(el, EL_BIND, "\033[24~", "f12-key", NULL); el_set(el, EL_BIND, "\004", "EOF-key", NULL); el_set(el, EL_ADDFN, "ed-complete", "Complete argument", complete); el_set(el, EL_BIND, "^I", "ed-complete", NULL); if (!(myhistory = history_init())) { esl_log(ESL_LOG_ERROR, "history could not be initialized\n"); goto done; } history(myhistory, &ev, H_SETSIZE, 800); el_set(el, EL_HIST, history, myhistory); history(myhistory, &ev, H_LOAD, hfile); el_source(el, NULL); #endif #ifdef WIN32 hStdout = GetStdHandle(STD_OUTPUT_HANDLE); if (hStdout != INVALID_HANDLE_VALUE && GetConsoleScreenBufferInfo(hStdout, &csbiInfo)) { wOldColorAttrs = csbiInfo.wAttributes; } #endif if (!argv_quiet && !profile->quiet) { snprintf(cmd_str, sizeof(cmd_str), "log %s\n\n", profile->loglevel); esl_send_recv(&handle, cmd_str); } print_banner(stdout); esl_log(ESL_LOG_INFO, "FS CLI Ready.\nenter /help for a list of commands.\n"); output_printf("%s\n", handle.last_sr_reply); while (running > 0) { int r; #ifdef HAVE_EDITLINE line = el_gets(el, &count); #else line = basic_gets(&count); #endif if (count > 1 && !esl_strlen_zero(line)) { char *p, *cmd = strdup(line); assert(cmd); if ((p = strrchr(cmd, '\r')) || (p = strrchr(cmd, '\n'))) { *p = '\0'; } #ifdef HAVE_EDITLINE history(myhistory, &ev, H_ENTER, line); #endif if ((r = process_command(&handle, cmd))) { running = r; } free(cmd); clear_el_buffer(); } sleep_ms(1); } if (running < 0 && reconnect) { running = 1; loops = 120; goto connect; } #ifdef HAVE_EDITLINE done: history(myhistory, &ev, H_SAVE, hfile); history_end(myhistory); el_end(el); #endif esl_disconnect(&handle); thread_running = 0; return 0; } /* For Emacs: * Local Variables: * mode:c * indent-tabs-mode:t * tab-width:4 * c-basic-offset:4 * End: * For VIM: * vim:set softtabstop=4 shiftwidth=4 tabstop=4: */