12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934 |
- #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[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 <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 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 <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;
- 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:
- */
|