1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801 |
- #if !defined(_XOPEN_SOURCE) && !defined(__FreeBSD__) && !defined(__NetBSD__) && !defined(__OpenBSD__)
- #define _XOPEN_SOURCE 600
- #endif
- #include <stdio.h>
- #include <stdlib.h>
- #include <esl.h>
- #include <signal.h>
- #include <getopt.h>
- #define CMD_BUFLEN 1024
- #ifndef WIN32
- #include <esl_config_auto.h>
- #include <sys/select.h>
- #include <unistd.h>
- #include <time.h>
- #else
- #define strdup(src) _strdup(src)
- #define fileno _fileno
- #define read _read
- #include <io.h>
- #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 <histedit.h>
- #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];
- 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];
- } 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[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 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;
- 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 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) {
- 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);
- 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 <host>] [-P <port>] [-p <secret>] [-d <level>] [-x command] [-t <timeout_ms>] [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 miliseconds)\n"
- " -T, --connect-timeout Timeout for socket connection (in miliseconds)\n"
- " -n, --no-color Disable color\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)
- {
- #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) 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);
- snprintf(s2, sizeof(s2), "\033[%dC", pos);
- printf("%s%s",s1,s2);
- }
- fflush(stdout);
- return;
- #endif
- #endif
- }
- 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;
- 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")) {
- 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);
- }
- 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++;
- }
- 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++;
- }
- 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", 3)) {
- 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) {
- if (fgets(command_buf, sizeof(command_buf) - 1, stdin) != command_buf) {
- break;
- }
- command_buf[strlen(command_buf)-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 <enter> 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 <cc.h>
- 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=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_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_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_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[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, "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, "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
- }
- 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;
- int argv_use_history_file = 1;
- int use_history_file = 0;
- #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'},
- {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_log_uuid = 0;
- int argv_log_uuid_short = 0;
- int argv_quiet = 0;
- int argv_batch = 0;
- int loops = 2, reconnect = 0;
- char *ccheck;
- 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, "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);
- 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?n", 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':
- argv_use_history_file = 0;
- 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 '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)) {
- 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_batch || profile->batch_mode) {
- profile->batch_mode = 1;
- feature_level=0;
- }
- if (argv_use_history_file && profile->use_history_file) {
- use_history_file = 1;
- }
- 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);
- 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);
- 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);
- }
- 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", 3)) {
- 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;
- }
- #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;
- 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:
- */
|