#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 #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 #define KEY_DELETE 10 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_LIBEDIT #include #endif typedef struct { char name[256]; char host[128]; esl_port_t port; char user[256]; char pass[128]; int debug; const char *console_fnkeys[12]; const char *console_fnkeys_toggle[12]; int console_fnkeys_state[12]; char loglevel[128]; int log_uuid; int log_uuid_length; int quiet; int use_history_file; int batch_mode; char prompt_color[12]; char input_text_color[12]; char output_text_color[12]; char prompt_string[512]; } cli_profile_t; static const int log_uuid_short_length = 8; static int is_color = 1; static int warn_stop = 0; static int connected = 0; static int allow_ctl_c = 0; static char bare_prompt_str[514] = ""; static int bare_prompt_str_len = 0; static char prompt_str[1024] = ""; 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 int feature_level = 0; 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, thread_up = 0, check_up = 0; static char *filter_uuid; static char *logfilter; static int timeout = 0; static int connect_timeout = 0; #ifdef HAVE_LIBEDIT static EditLine *el; static History *myhistory; static HistEvent ev; #endif static char hostname[256] = ""; static char switchname[256] = ""; static char switch_hostname[256] = ""; static esl_mutex_t *MUTEX = NULL; 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); #if defined(HAVE_LIBEDIT) || defined(WIN32) static void clear_cli(void) { if (global_profile->batch_mode) return; putchar('\r'); printf("\033[%dC", bare_prompt_str_len); printf("\033[K"); fflush(stdout); } #endif static void screen_size(int *x, int *y) { #ifdef WIN32 CONSOLE_SCREEN_BUFFER_INFO csbi; int ret; if ((ret = GetConsoleScreenBufferInfo(GetStdHandle( STD_OUTPUT_HANDLE ), &csbi))) { if (x) *x = csbi.dwSize.X; if (y) *y = csbi.dwSize.Y; } #elif defined(TIOCGWINSZ) struct winsize w; if ( (ioctl(0, TIOCGWINSZ, &w)) >= 0 ) { if (x) *x = w.ws_col; if (y) *y = w.ws_row; } #else if (x) *x = 80; if (y) *y = 24; #endif } #if defined(HAVE_LIBEDIT) || defined(WIN32) /* If a fnkey is configured then process the command */ static unsigned char console_fnkey_pressed(int i) { const char *c; int fnkey; assert((i > 0) && (i <= 12)); fnkey = i - 1; if ((c = global_profile->console_fnkeys_toggle[fnkey]) && global_profile->console_fnkeys_state[fnkey]) { global_profile->console_fnkeys_state[fnkey] = 0; } else if ((c = global_profile->console_fnkeys[fnkey])) { global_profile->console_fnkeys_state[fnkey] = 1; } else { 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; } #endif #ifdef HAVE_LIBEDIT 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) { #ifdef HAVE_EL_CURSOR el_cursor(el, 1); #endif 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); 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; } if (key == KEY_DELETE) { if (iCmdCursor < iCmdBuffer) { int pos; for (pos = iCmdCursor; pos < iCmdBuffer; pos++) { cmd[pos] = cmd[pos + 1]; } cmd[pos] = 0; iCmdBuffer--; for (pos = iCmdCursor; pos < iCmdBuffer; pos++) { printf("%c", cmd[pos]); } printf(" "); SetConsoleCursorPosition(hOut, position); } } 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; if (GetNumberOfConsoleInputEvents(conIn, &toRead) == 0) { return(FALSE); } 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; } if (keyEvent.wVirtualKeyCode == 46 && keyEvent.wVirtualScanCode == 83) { *key = KEY_DELETE; } 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" " -U, --log-uuid Include UUID in log output\n" " -S, --log-uuid-short Include shortened UUID in log output\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" " -b, --batchmode Batch mode\n" " -t, --timeout Timeout for API commands (in milliseconds)\n" " -T, --connect-timeout Timeout for socket connection (in milliseconds)\n" " -n, --no-color Disable color\n" " -s, --set-log-uuid Set UUID to filter log events\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 } static void clear_line(void) { if (global_profile->batch_mode) return; putchar('\r'); printf("\033[K"); fflush(stdout); return; } static void redisplay(void) { esl_mutex_lock(MUTEX); { #ifdef HAVE_LIBEDIT #ifdef HAVE_DECL_EL_REFRESH #ifdef HAVE_EL_WSET /* Current libedit versions don't implement EL_REFRESH in eln.c so * use the wide version instead. */ el_wset(el, EL_REFRESH); #else /* This will work on future libedit versions and versions built * without wide character support. */ el_set(el, EL_REFRESH); #endif #else /* Old libedit versions don't implement EL_REFRESH at all so use * our own implementation instead. */ const LineInfo *lf = el_line(el); const char *c = lf->buffer; if (global_profile->batch_mode) { esl_mutex_unlock(MUTEX); return; } printf("%s",prompt_str); while (c < lf->lastchar && *c) { putchar(*c); c++; } { int pos = (int)(lf->cursor - lf->buffer); char s1[12], s2[12] = ""; putchar('\r'); snprintf(s1, sizeof(s1), "\033[%dC", bare_prompt_str_len); if (pos) snprintf(s2, sizeof(s2), "\033[%dC", pos); printf("%s%s",s1,s2); } fflush(stdout); #endif #endif } esl_mutex_unlock(MUTEX); return; } static int output_printf(const char *fmt, ...) { va_list ap; int r; va_start(ap, fmt); #ifndef WIN32 if (!(global_profile->batch_mode)) { 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; esl_mutex_lock(MUTEX); thread_up = 1; esl_mutex_unlock(MUTEX); while(thread_running && handle->connected) { int aok = 1; esl_status_t status; esl_mutex_lock(MUTEX); status = esl_recv_event_timed(handle, 10, 1, NULL); esl_mutex_unlock(MUTEX); if (status == ESL_BREAK) { sleep_ms(1); } else 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") && handle->last_event->body) { 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 (logfilter) { if (!strstr(handle->last_event->body, logfilter)) { continue; } } if (lname) { level = atoi(lname); } #ifndef WIN32 if (aok) { if (feature_level) clear_line(); if(!(global_profile->batch_mode)) { printf("%s", colors[level]); } if (global_profile->log_uuid && !esl_strlen_zero(userdata)) { if (global_profile->log_uuid_length) { int len = strlen(userdata); int i = (global_profile->log_uuid_length < len) ? global_profile->log_uuid_length : len; if (fwrite(userdata, sizeof(char), i, stdout) < i) { // don't care } printf(" "); } else { printf("%s ", userdata); } } if (strcmp("\n",handle->last_event->body)) { char *c = handle->last_event->body; printf("%s", handle->last_event->body); if (*c) { while (*c) ++c; c--; if (*c != '\n') printf("\n"); } } if(!(global_profile->batch_mode)) { if (!feature_level) printf("%s", ESL_SEQ_DEFAULT_COLOR); } if (feature_level) redisplay(); } #else if (aok) { if(!(global_profile->batch_mode)) { SetConsoleTextAttribute(hStdout, colors[level]); } if (global_profile->log_uuid && !esl_strlen_zero(userdata)) { WriteFile(hStdout, userdata, (DWORD)strlen(userdata), &outbytes, NULL); WriteFile(hStdout, " ", (DWORD)strlen(" "), &outbytes, NULL); } WriteFile(hStdout, handle->last_event->body, len, &outbytes, NULL); if(!(global_profile->batch_mode)) { 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); } esl_mutex_lock(MUTEX); thread_up = 0; esl_mutex_unlock(MUTEX); 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" "/logfilter \tFilter Log for a single string.\n" "/debug [0-7] \tSet debug level.\n" "\n"; static int process_command(esl_handle_t *handle, const char *cmd) { int r = 0; while (*cmd == ' ') cmd++; esl_mutex_lock(MUTEX); 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"); r = -1; goto end; } else if (!strncasecmp(cmd, "logfilter", 9)) { cmd += 9; while (cmd && *cmd && *cmd == ' ') { cmd++; } if (!esl_strlen_zero(cmd)) { esl_safe_free(logfilter); logfilter = strdup(cmd); } else { esl_safe_free(logfilter); } output_printf("Logfilter %s\n", logfilter ? "enabled" : "disabled"); } else if (!strncasecmp(cmd, "uuid", 4)) { cmd += 4; while (cmd && *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; if (!strncasecmp(cmd, "console loglevel ", 17)) { snprintf(cmd_str, sizeof(cmd_str), "log %s", cmd + 17); esl_send_recv(handle, cmd_str); printf("%s\n", handle->last_sr_reply); } 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"); r = -1; goto end; } 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", 4)) { output_printf("Error: %s!\n", err + 4); } } } end: esl_mutex_unlock(MUTEX); return r; } 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; } static char command_buf[CMD_BUFLEN+1] = ""; static const char *basic_gets(int *cnt) { int x = 0; #ifdef _MSC_VER int read, key; char keys[CMD_BUFLEN]; HANDLE stdinHandle; if (global_profile->batch_mode) { #endif printf("%s", prompt_str); if (global_profile->batch_mode) fflush(stdout); memset(&command_buf, 0, sizeof(command_buf)); for (x = 0; x < (sizeof(command_buf) - 1); x++) { int c = getchar(); if (c < 0) { size_t command_buf_len; if (fgets(command_buf, sizeof(command_buf) - 1, stdin) != command_buf) { break; } if ((command_buf_len = strlen(command_buf)) > 0) { command_buf[command_buf_len - 1] = '\0'; /* remove endline */ } break; } command_buf[x] = (char) c; if (command_buf[x] == '\n') { command_buf[x] = '\0'; break; } } *cnt = x; #ifdef _MSC_VER } else { stdinHandle = GetStdHandle(STD_INPUT_HANDLE); console_bufferInput (0, 0, prompt_str, PROMPT_OP); printf("%s", prompt_str); if (global_profile->batch_mode) fflush(stdout); *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 (global_profile->batch_mode) fflush(stdout); if (!strcmp(command_buf, "Empty")) { command_buf[0] = 0; } } sleep_ms(20); } } #endif return command_buf; } static const char *banner = ".=======================================================.\n" "| _____ ____ ____ _ ___ |\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"; static const char *inf = "Type /help to see a list of commands\n\n\n"; static void print_banner(FILE *stream, int color) { int x = 0; const char *use = NULL; #include screen_size(&x, NULL); use = (x > 100) ? cc : cc_s; #ifdef WIN32 /* Print banner in yellow with blue background */ if (color) { SetConsoleTextAttribute(hStdout, ESL_SEQ_FYELLOW | BACKGROUND_BLUE); } WriteFile(hStdout, banner, (DWORD) strlen(banner), NULL, NULL); WriteFile(hStdout, use, (DWORD) strlen(use), NULL, NULL); if (color) { SetConsoleTextAttribute(hStdout, wOldColorAttrs); } /* Print the rest info in default colors */ fprintf(stream, "\n%s\n", inf); #else if (color) { fprintf(stream, "%s%s%s", ESL_SEQ_DEFAULT_COLOR, ESL_SEQ_FYELLOW, ESL_SEQ_BBLUE); } #ifndef DISABLE_CC fprintf(stream, "%s%s", banner, use); #else fprintf(stream, "%s", banner); #endif if (color) { fprintf(stream, "%s", ESL_SEQ_DEFAULT_COLOR); } fprintf(stream, "\n%s\n", inf); if (color) { fprintf(stream, "%s", output_text_color); } #endif if (x < 160) { fprintf(stream, "\n[This app Best viewed at 160x60 or more..]\n"); } } 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"; } #if defined(HAVE_LIBEDIT) || defined(WIN32) 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; assert(r); 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; int sc = 0, offset = (int)(cursor - buffer), pos = (offset > 0) ? offset : 0; char *p; assert(dup); buf = dup; if (pos > 0) { *(buf + pos) = '\0'; } if ((p = strchr(buf, '\r')) || (p = strchr(buf, '\n'))) { *p = '\0'; } while (*buf == ' ') { buf++; sc++; } #ifdef HAVE_LIBEDIT if (!*buf && sc) { el_deletestr(el, sc); sc = 0; } #endif p = end_of_str(buf); while(p >= buf && *p == ' ') { sc++; p--; } #ifdef HAVE_LIBEDIT 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_mutex_lock(MUTEX); esl_send_recv(global_handle, cmd_str); esl_mutex_unlock(MUTEX); if (global_handle->last_sr_event && global_handle->last_sr_event->body) { char *r = global_handle->last_sr_event->body; char *w, *p1; 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_LIBEDIT 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; } #endif /* if defined(HAVE_LIBEDIT) || defined(WIN32) */ #ifdef HAVE_LIBEDIT static unsigned char complete(EditLine *el, int ch) { const LineInfo *lf = el_line(el); int r; esl_mutex_lock(MUTEX); r = esl_console_complete(lf->buffer, lf->cursor, lf->lastchar); esl_mutex_unlock(MUTEX); return r; } #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[256] = ""; 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, "127.0.0.1"); 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); profiles[pcount].use_history_file = 1; 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, "batchmode")) { profiles[pcount-1].batch_mode = esl_true(val); } 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, "log-uuid")) { profiles[pcount-1].log_uuid = esl_true(val); } else if(!strcasecmp(var, "log-uuid-short")) { profiles[pcount-1].log_uuid = esl_true(val); profiles[pcount-1].log_uuid_length = (esl_true(val) ? log_uuid_short_length : 0); } else if(!strcasecmp(var, "log-uuid-length")) { int i; if ((i = atoi(val)) > -1) { profiles[pcount-1].log_uuid_length = i; } } else if(!strcasecmp(var, "quiet")) { profiles[pcount-1].quiet = esl_true(val); } else if(!strcasecmp(var, "no-history-file")) { profiles[pcount-1].use_history_file = !esl_true(val); } else if(!strcasecmp(var, "prompt-color")) { esl_set_string(profiles[pcount-1].prompt_color, match_color(val)); } else if(!strcasecmp(var, "prompt-string")) { esl_set_string(profiles[pcount-1].prompt_string, 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); } } } else if (!strncasecmp(var, "key_toggle_F", 12)) { char *key = var + 12; if (key) { int i = atoi(key); if (i > 0 && i < 13) { profiles[pcount-1].console_fnkeys_toggle[i - 1] = strdup(val); } } } else if (!strcasecmp(var, "timeout")) { timeout = atoi(val); } else if (!strcasecmp(var, "connect-timeout")) { connect_timeout = atoi(val); } } esl_config_close_file(&cfg); } } static void clear_el_buffer(void) { #ifdef HAVE_LIBEDIT const LineInfo *lf = el_line(el); int len = (int)(lf->cursor - lf->buffer); if (global_profile->batch_mode) return; el_deletestr(el, len); memset((char*)lf->buffer, 0, len); #endif } static void expand_prompt(char *s, size_t len, cli_profile_t *profile) { char tmp[512] = ""; char *p, *q = tmp; for (p = s; p && *p; p++) { if (*p == '%') { p++; switch(*p) { case 's': esl_copy_string(q, switchname, len - (q - &tmp[0])); q += strlen(switchname); break; case 'h': esl_copy_string(q, hostname, len - (q - &tmp[0])); q += strlen(hostname); break; case 'H': esl_copy_string(q, switch_hostname, len - (q - &tmp[0])); q += strlen(switch_hostname); break; case 'p': esl_copy_string(q, profile->name, len - (q - &tmp[0])); q += strlen(profile->name); break; case 'o': esl_copy_string(q, profile->host, len - (q - &tmp[0])); q += strlen(profile->host); break; case 'P': { char ptmp[35] = ""; esl_snprintf(ptmp, sizeof(ptmp), "%d", profile->port); esl_copy_string(q, ptmp, len - (q - &tmp[0])); q += strlen(ptmp); } break; case '%': *q++ = '%'; break; } } else { *q++ = *p; } } esl_copy_string(s, tmp, len); } int main(int argc, char *argv[]) { esl_handle_t handle = {{0}}; int count = 0; const char *line = NULL; char cmd_str[2048] = ""; cli_profile_t *profile = NULL; #ifdef HAVE_LIBEDIT int argv_use_history_file = 1; int use_history_file = 0; #endif #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'}, {"no-color", 0, 0, 'n'}, {"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'}, {"log-uuid", 0, 0, 'U'}, {"log-uuid-short", 0, 0, 'S'}, {"quiet", 0, 0, 'q'}, {"batchmode", 0, 0, 'b'}, {"no-history-file", 0, 0, 'Q'}, {"retry", 0, 0, 'r'}, {"interrupt", 0, 0, 'i'}, {"reconnect", 0, 0, 'R'}, {"timeout", 1, 0, 't'}, {"connect-timeout", 1, 0, 'T'}, {"set-log-uuid", 1, 0, 's'}, {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[127] = ""; char argv_filter_uuid[64] = {0}; int argv_log_uuid = 0; int argv_log_uuid_short = 0; int argv_quiet = 0; int argv_batch = 0; int loops = 2, reconnect = 0; char *ccheck; gethostname(hostname, sizeof(hostname)); esl_mutex_create(&MUTEX); #if HAVE_DECL_EL_PROMPT_ESC feature_level = 1; #else { char *term = getenv("TERM"); if (term && (!strncasecmp("screen", term, 6) || !strncasecmp("vt100", term, 5))) { feature_level = 1; } else { feature_level = 0; } } #endif #ifdef WIN32 feature_level = 0; #endif if ((ccheck = getenv("FS_CLI_COLOR"))) { is_color = esl_true(ccheck); } 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, hostname, 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); internal_profile.use_history_file = 1; 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:P:u:p:d:x:l:USt:T:qQrRhib?ns:", 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 'n': is_color = 0; 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 'U': argv_log_uuid = 1; break; case 'S': argv_log_uuid_short = 1; break; case 'q': argv_quiet = 1; break; case 'b': argv_batch = 1; break; case 'Q': #ifdef HAVE_LIBEDIT argv_use_history_file = 0; #endif 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 'T': connect_timeout = atoi(optarg); break; case 's': esl_set_string(argv_filter_uuid, optarg); filter_uuid = strdup(argv_filter_uuid); break; case 'h': case '?': print_banner(stdout, is_color); usage(argv[0]); return 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)) { if (!esl_strlen_zero(profiles[0].name)) { profile = &profiles[0]; } } } if (!profile) { esl_log(ESL_LOG_DEBUG, "no profiles found, 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_batch || profile->batch_mode) { profile->batch_mode = 1; feature_level=0; } #ifdef HAVE_LIBEDIT if (argv_use_history_file && profile->use_history_file) { use_history_file = 1; } #endif if (*argv_loglevel) { esl_set_string(profile->loglevel, argv_loglevel); profile->quiet = 0; } if (argv_log_uuid) { profile->log_uuid = 1; } if (argv_log_uuid_short) { profile->log_uuid = 1; profile->log_uuid_length = log_uuid_short_length; } 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); connect: connected = 0; while (--loops > 0) { memset(&handle, 0, sizeof(handle)); if (esl_connect_timeout(&handle, profile->host, profile->port, profile->user, profile->pass, connect_timeout)) { 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\nconsole_execute: true\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", 4)) { printf("Error: %s!\n", err + 4); } } esl_disconnect(&handle); return 0; } global_handle = &handle; global_profile = profile; if (esl_thread_create_detached(msg_thread_run, &handle) != ESL_SUCCESS) { printf("Error starting thread!\n"); esl_disconnect(&handle); return 0; } snprintf(cmd_str, sizeof(cmd_str), "api switchname\n\n"); esl_send_recv(global_handle, cmd_str); if (global_handle->last_sr_event && global_handle->last_sr_event->body) { esl_set_string(switchname, global_handle->last_sr_event->body); } else { esl_set_string(switchname, profile->name); } snprintf(cmd_str, sizeof(cmd_str), "api hostname\n\n"); esl_send_recv(global_handle, cmd_str); if (global_handle->last_sr_event && global_handle->last_sr_event->body) { esl_set_string(switch_hostname, global_handle->last_sr_event->body); } else { esl_set_string(switch_hostname, profile->name); } if (!esl_strlen_zero(profile->prompt_string)) { expand_prompt(profile->prompt_string, sizeof(profile->prompt_string), profile); snprintf(bare_prompt_str, sizeof(bare_prompt_str), "%s> ", profile->prompt_string); } else { snprintf(bare_prompt_str, sizeof(bare_prompt_str), "freeswitch@%s> ", switchname); } bare_prompt_str_len = (int)strlen(bare_prompt_str); if (feature_level) { #if HAVE_DECL_EL_PROMPT_ESC snprintf(prompt_str, sizeof(prompt_str), "\1%s\1%s\1%s\1", prompt_color, bare_prompt_str, input_text_color); #else snprintf(prompt_str, sizeof(prompt_str), "%s%s%s", prompt_color, bare_prompt_str, input_text_color); #endif } else { snprintf(prompt_str, sizeof(prompt_str), "%s", bare_prompt_str); } #ifdef HAVE_LIBEDIT el = el_init(__FILE__, stdin, stdout, stderr); #if HAVE_DECL_EL_PROMPT_ESC el_set(el, EL_PROMPT_ESC, &prompt, '\1'); #else el_set(el, EL_PROMPT, &prompt); #endif 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, "\033OT", "f5-key", NULL); el_set(el, EL_BIND, "\033OU", "f6-key", NULL); el_set(el, EL_BIND, "\033OV", "f7-key", NULL); el_set(el, EL_BIND, "\033OW", "f8-key", NULL); el_set(el, EL_BIND, "\033OX", "f9-key", NULL); el_set(el, EL_BIND, "\033OY", "f10-key", NULL); el_set(el, EL_BIND, "\033OZ", "f11-key", NULL); el_set(el, EL_BIND, "\033O[", "f12-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); /* "Delete" key. */ el_set(el, EL_BIND, "\033[3~", "ed-delete-next-char", NULL); if (!(myhistory = history_init())) { esl_log(ESL_LOG_ERROR, "history could not be initialized\n"); goto done; } history(myhistory, &ev, H_SETSIZE, 800); /* Ignore duplicate lines */ history(myhistory, &ev, H_SETUNIQUE, 1); el_set(el, EL_HIST, history, myhistory); if (use_history_file) 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); } if (global_profile->batch_mode) { setvbuf(stdout, (char*)NULL, _IONBF, 0); } print_banner(stdout, is_color); 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_LIBEDIT if (!(global_profile->batch_mode)) { line = el_gets(el, &count); } else { #endif line = basic_gets(&count); #ifdef HAVE_LIBEDIT } #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_LIBEDIT 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_LIBEDIT done: if (use_history_file) history(myhistory, &ev, H_SAVE, hfile); history_end(myhistory); el_end(el); #endif esl_disconnect(&handle); global_handle = NULL; thread_running = 0; do { esl_mutex_lock(MUTEX); check_up = thread_up; esl_mutex_unlock(MUTEX); sleep_ms(10); } while (check_up > 0); esl_mutex_destroy(&MUTEX); 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 noet: */