git.gag.com Git - fw/openocd/blob - src/server/telnet_server.c

   1 /***************************************************************************
   2  *   Copyright (C) 2005 by Dominic Rath                                    *
   3  *   Dominic.Rath@gmx.de                                                   *
   4  *                                                                         *
   5  *   Copyright (C) 2007-2010 Øyvind Harboe                                 *
   6  *   oyvind.harboe@zylin.com                                               *
   7  *                                                                         *
   8  *   Copyright (C) 2008 by Spencer Oliver                                  *
   9  *   spen@spen-soft.co.uk                                                  *
  10  *                                                                         *
  11  *   This program is free software; you can redistribute it and/or modify  *
  12  *   it under the terms of the GNU General Public License as published by  *
  13  *   the Free Software Foundation; either version 2 of the License, or     *
  14  *   (at your option) any later version.                                   *
  15  *                                                                         *
  16  *   This program is distributed in the hope that it will be useful,       *
  17  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  18  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  19  *   GNU General Public License for more details.                          *
  20  *                                                                         *
  21  *   You should have received a copy of the GNU General Public License     *
  22  *   along with this program.  If not, see <http://www.gnu.org/licenses/>. *
  23  ***************************************************************************/
  24
  25 #ifdef HAVE_CONFIG_H
  26 #include "config.h"
  27 #endif
  28
  29 #include "telnet_server.h"
  30 #include <target/target_request.h>
  31 #include <helper/configuration.h>
  32 #include <helper/list.h>
  33
  34 static char *telnet_port;
  35
  36 static char *negotiate =
  37         "\xFF\xFB\x03"                  /* IAC WILL Suppress Go Ahead */
  38         "\xFF\xFB\x01"                  /* IAC WILL Echo */
  39         "\xFF\xFD\x03"                  /* IAC DO Suppress Go Ahead */
  40         "\xFF\xFE\x01";                 /* IAC DON'T Echo */
  41
  42 #define CTRL(c) (c - '@')
  43 #define TELNET_HISTORY  ".openocd_history"
  44
  45 /* The only way we can detect that the socket is closed is the first time
  46  * we write to it, we will fail. Subsequent write operations will
  47  * succeed. Shudder!
  48  */
  49 static int telnet_write(struct connection *connection, const void *data,
  50         int len)
  51 {
  52         struct telnet_connection *t_con = connection->priv;
  53         if (t_con->closed)
  54                 return ERROR_SERVER_REMOTE_CLOSED;
  55
  56         if (connection_write(connection, data, len) == len)
  57                 return ERROR_OK;
  58         t_con->closed = true;
  59         return ERROR_SERVER_REMOTE_CLOSED;
  60 }
  61
  62 /* output an audible bell */
  63 static int telnet_bell(struct connection *connection)
  64 {
  65         /* ("\a" does not work, at least on windows) */
  66         return telnet_write(connection, "\x07", 1);
  67 }
  68
  69 static int telnet_prompt(struct connection *connection)
  70 {
  71         struct telnet_connection *t_con = connection->priv;
  72
  73         return telnet_write(connection, t_con->prompt, strlen(t_con->prompt));
  74 }
  75
  76 static int telnet_outputline(struct connection *connection, const char *line)
  77 {
  78         int len;
  79
  80         /* process lines in buffer */
  81         while (*line) {
  82                 char *line_end = strchr(line, '\n');
  83
  84                 if (line_end)
  85                         len = line_end-line;
  86                 else
  87                         len = strlen(line);
  88
  89                 telnet_write(connection, line, len);
  90                 if (line_end) {
  91                         telnet_write(connection, "\r\n", 2);
  92                         line += len + 1;
  93                 } else
  94                         line += len;
  95         }
  96
  97         return ERROR_OK;
  98 }
  99
 100 static int telnet_output(struct command_context *cmd_ctx, const char *line)
 101 {
 102         struct connection *connection = cmd_ctx->output_handler_priv;
 103
 104         return telnet_outputline(connection, line);
 105 }
 106
 107 static void telnet_log_callback(void *priv, const char *file, unsigned line,
 108         const char *function, const char *string)
 109 {
 110         struct connection *connection = priv;
 111         struct telnet_connection *t_con = connection->priv;
 112         size_t i;
 113         size_t tmp;
 114
 115         /* If the prompt is not visible, simply output the message. */
 116         if (!t_con->prompt_visible) {
 117                 telnet_outputline(connection, string);
 118                 return;
 119         }
 120
 121         /* Clear the command line. */
 122         tmp = strlen(t_con->prompt) + t_con->line_size;
 123
 124         for (i = 0; i < tmp; i += 16)
 125                 telnet_write(connection, "\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b",
 126                         MIN(tmp - i, 16));
 127
 128         for (i = 0; i < tmp; i += 16)
 129                 telnet_write(connection, "                ", MIN(tmp - i, 16));
 130
 131         for (i = 0; i < tmp; i += 16)
 132                 telnet_write(connection, "\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b",
 133                         MIN(tmp - i, 16));
 134
 135         telnet_outputline(connection, string);
 136
 137         /* Put the command line to its previous state. */
 138         telnet_prompt(connection);
 139         telnet_write(connection, t_con->line, t_con->line_size);
 140
 141         for (i = t_con->line_cursor; i < t_con->line_size; i++)
 142                 telnet_write(connection, "\b", 1);
 143 }
 144
 145 static void telnet_load_history(struct telnet_connection *t_con)
 146 {
 147         FILE *histfp;
 148         char buffer[TELNET_BUFFER_SIZE];
 149         int i = 0;
 150
 151         char *history = get_home_dir(TELNET_HISTORY);
 152
 153         if (!history) {
 154                 LOG_INFO("unable to get user home directory, telnet history will be disabled");
 155                 return;
 156         }
 157
 158         histfp = fopen(history, "rb");
 159
 160         if (histfp) {
 161
 162                 while (fgets(buffer, sizeof(buffer), histfp)) {
 163
 164                         char *p = strchr(buffer, '\n');
 165                         if (p)
 166                                 *p = '\0';
 167                         if (buffer[0] && i < TELNET_LINE_HISTORY_SIZE)
 168                                 t_con->history[i++] = strdup(buffer);
 169                 }
 170
 171                 t_con->next_history = i;
 172                 t_con->next_history %= TELNET_LINE_HISTORY_SIZE;
 173                 /* try to set to last entry - 1, that way we skip over any exit/shutdown cmds */
 174                 t_con->current_history = t_con->next_history > 0 ? i - 1 : 0;
 175                 fclose(histfp);
 176         }
 177
 178         free(history);
 179 }
 180
 181 static void telnet_save_history(struct telnet_connection *t_con)
 182 {
 183         FILE *histfp;
 184         int i;
 185         int num;
 186
 187         char *history = get_home_dir(TELNET_HISTORY);
 188
 189         if (!history) {
 190                 LOG_INFO("unable to get user home directory, telnet history will be disabled");
 191                 return;
 192         }
 193
 194         histfp = fopen(history, "wb");
 195
 196         if (histfp) {
 197
 198                 num = TELNET_LINE_HISTORY_SIZE;
 199                 i = t_con->current_history + 1;
 200                 i %= TELNET_LINE_HISTORY_SIZE;
 201
 202                 while (!t_con->history[i] && num > 0) {
 203                         i++;
 204                         i %= TELNET_LINE_HISTORY_SIZE;
 205                         num--;
 206                 }
 207
 208                 if (num > 0) {
 209                         for (; num > 0; num--) {
 210                                 fprintf(histfp, "%s\n", t_con->history[i]);
 211                                 i++;
 212                                 i %= TELNET_LINE_HISTORY_SIZE;
 213                         }
 214                 }
 215                 fclose(histfp);
 216         }
 217
 218         free(history);
 219 }
 220
 221 static int telnet_new_connection(struct connection *connection)
 222 {
 223         struct telnet_connection *telnet_connection;
 224         struct telnet_service *telnet_service = connection->service->priv;
 225         int i;
 226
 227         telnet_connection = malloc(sizeof(struct telnet_connection));
 228
 229         if (!telnet_connection) {
 230                 LOG_ERROR("Failed to allocate telnet connection.");
 231                 return ERROR_FAIL;
 232         }
 233
 234         connection->priv = telnet_connection;
 235
 236         /* initialize telnet connection information */
 237         telnet_connection->closed = false;
 238         telnet_connection->line_size = 0;
 239         telnet_connection->line_cursor = 0;
 240         telnet_connection->prompt = strdup("> ");
 241         telnet_connection->prompt_visible = true;
 242         telnet_connection->state = TELNET_STATE_DATA;
 243
 244         /* output goes through telnet connection */
 245         command_set_output_handler(connection->cmd_ctx, telnet_output, connection);
 246
 247         /* negotiate telnet options */
 248         telnet_write(connection, negotiate, strlen(negotiate));
 249
 250         /* print connection banner */
 251         if (telnet_service->banner) {
 252                 telnet_write(connection, telnet_service->banner, strlen(telnet_service->banner));
 253                 telnet_write(connection, "\r\n", 2);
 254         }
 255
 256         /* the prompt is always placed at the line beginning */
 257         telnet_write(connection, "\r", 1);
 258         telnet_prompt(connection);
 259
 260         /* initialize history */
 261         for (i = 0; i < TELNET_LINE_HISTORY_SIZE; i++)
 262                 telnet_connection->history[i] = NULL;
 263         telnet_connection->next_history = 0;
 264         telnet_connection->current_history = 0;
 265         telnet_load_history(telnet_connection);
 266
 267         log_add_callback(telnet_log_callback, connection);
 268
 269         return ERROR_OK;
 270 }
 271
 272 static void telnet_clear_line(struct connection *connection,
 273         struct telnet_connection *t_con)
 274 {
 275         /* move to end of line */
 276         if (t_con->line_cursor < t_con->line_size)
 277                 telnet_write(connection,
 278                         t_con->line + t_con->line_cursor,
 279                         t_con->line_size - t_con->line_cursor);
 280
 281         /* backspace, overwrite with space, backspace */
 282         while (t_con->line_size > 0) {
 283                 telnet_write(connection, "\b \b", 3);
 284                 t_con->line_size--;
 285         }
 286         t_con->line_cursor = 0;
 287 }
 288
 289 static void telnet_history_go(struct connection *connection, int idx)
 290 {
 291         struct telnet_connection *t_con = connection->priv;
 292
 293         if (t_con->history[idx]) {
 294                 telnet_clear_line(connection, t_con);
 295                 t_con->line_size = strlen(t_con->history[idx]);
 296                 t_con->line_cursor = t_con->line_size;
 297                 memcpy(t_con->line, t_con->history[idx], t_con->line_size);
 298                 telnet_write(connection, t_con->line, t_con->line_size);
 299                 t_con->current_history = idx;
 300         }
 301         t_con->state = TELNET_STATE_DATA;
 302 }
 303
 304 static void telnet_history_up(struct connection *connection)
 305 {
 306         struct telnet_connection *t_con = connection->priv;
 307
 308         size_t last_history = (t_con->current_history > 0) ?
 309                                 t_con->current_history - 1 :
 310                                 TELNET_LINE_HISTORY_SIZE-1;
 311         telnet_history_go(connection, last_history);
 312 }
 313
 314 static void telnet_history_down(struct connection *connection)
 315 {
 316         struct telnet_connection *t_con = connection->priv;
 317         size_t next_history;
 318
 319         next_history = (t_con->current_history + 1) % TELNET_LINE_HISTORY_SIZE;
 320         telnet_history_go(connection, next_history);
 321 }
 322
 323 static int telnet_history_print(struct connection *connection)
 324 {
 325         struct telnet_connection *tc;
 326
 327         tc = connection->priv;
 328
 329         for (size_t i = 1; i < TELNET_LINE_HISTORY_SIZE; i++) {
 330                 char *line;
 331
 332                 /*
 333                  * The tc->next_history line contains empty string (unless NULL), thus
 334                  * it is not printed.
 335                  */
 336                 line = tc->history[(tc->next_history + i) % TELNET_LINE_HISTORY_SIZE];
 337
 338                 if (line) {
 339                         telnet_write(connection, line, strlen(line));
 340                         telnet_write(connection, "\r\n\x00", 3);
 341                 }
 342         }
 343
 344         tc->line_size = 0;
 345         tc->line_cursor = 0;
 346
 347         /* The prompt is always placed at the line beginning. */
 348         telnet_write(connection, "\r", 1);
 349
 350         return telnet_prompt(connection);
 351 }
 352
 353 static void telnet_move_cursor(struct connection *connection, size_t pos)
 354 {
 355         struct telnet_connection *tc;
 356         size_t tmp;
 357
 358         tc = connection->priv;
 359
 360         if (pos < tc->line_cursor) {
 361                 tmp = tc->line_cursor - pos;
 362
 363                 for (size_t i = 0; i < tmp; i += 16)
 364                         telnet_write(connection, "\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b",
 365                                 MIN(tmp - i, 16));
 366         } else {
 367                 tmp = pos - tc->line_cursor;
 368
 369                 for (size_t i = 0; i < tmp; i += 16)
 370                         telnet_write(connection, tc->line + tc->line_cursor + i,
 371                                 MIN(tmp - i, 16));
 372         }
 373
 374         tc->line_cursor = pos;
 375 }
 376
 377 /* check buffer size leaving one spare character for string null termination */
 378 static inline bool telnet_can_insert(struct connection *connection, size_t len)
 379 {
 380         struct telnet_connection *t_con = connection->priv;
 381
 382         return t_con->line_size + len < TELNET_LINE_MAX_SIZE;
 383 }
 384
 385 /* write to telnet console, and update the telnet_connection members
 386  * this function is capable of inserting in the middle of a line
 387  * please ensure that data does not contain special characters (\n, \r, \t, \b ...)
 388  *
 389  * returns false when it fails to insert the requested data
 390  */
 391 static bool telnet_insert(struct connection *connection, const void *data, size_t len)
 392 {
 393         struct telnet_connection *t_con = connection->priv;
 394
 395         if (!telnet_can_insert(connection, len)) {
 396                 telnet_bell(connection);
 397                 return false;
 398         }
 399
 400         if (t_con->line_cursor < t_con->line_size) {
 401                 /* we have some content after the cursor */
 402                 memmove(t_con->line + t_con->line_cursor + len,
 403                                 t_con->line + t_con->line_cursor,
 404                                 t_con->line_size - t_con->line_cursor);
 405         }
 406
 407         strncpy(t_con->line + t_con->line_cursor, data, len);
 408
 409         telnet_write(connection,
 410                         t_con->line + t_con->line_cursor,
 411                         t_con->line_size + len - t_con->line_cursor);
 412
 413         t_con->line_size += len;
 414         t_con->line_cursor += len;
 415
 416         for (size_t i = t_con->line_cursor; i < t_con->line_size; i++)
 417                 telnet_write(connection, "\b", 1);
 418
 419         return true;
 420 }
 421
 422 static void telnet_auto_complete(struct connection *connection)
 423 {
 424         struct telnet_connection *t_con = connection->priv;
 425         struct command_context *command_context = connection->cmd_ctx;
 426
 427         struct cmd_match {
 428                 char *cmd;
 429                 struct list_head lh;
 430         };
 431
 432         LIST_HEAD(matches);
 433
 434         /* user command sequence, either at line beginning
 435          * or we start over after these characters ';', '[', '{' */
 436         size_t seq_start = (t_con->line_cursor == 0) ? 0 : (t_con->line_cursor - 1);
 437         while (seq_start > 0) {
 438                 char c = t_con->line[seq_start];
 439                 if (c == ';' || c == '[' || c == '{') {
 440                         seq_start++;
 441                         break;
 442                 }
 443
 444                 seq_start--;
 445         }
 446
 447         /* user command position in the line, ignore leading spaces */
 448         size_t usr_cmd_pos = seq_start;
 449         while ((usr_cmd_pos < t_con->line_cursor) && isspace(t_con->line[usr_cmd_pos]))
 450                 usr_cmd_pos++;
 451
 452         /* user command length */
 453         size_t usr_cmd_len = t_con->line_cursor - usr_cmd_pos;
 454
 455         /* optimize multiple spaces in the user command,
 456          * because info commands does not tolerate multiple spaces */
 457         size_t optimized_spaces = 0;
 458         char query[usr_cmd_len + 1];
 459         for (size_t i = 0; i < usr_cmd_len; i++) {
 460                 if ((i < usr_cmd_len - 1) && isspace(t_con->line[usr_cmd_pos + i])
 461                                 && isspace(t_con->line[usr_cmd_pos + i + 1])) {
 462                         optimized_spaces++;
 463                         continue;
 464                 }
 465
 466                 query[i - optimized_spaces] = t_con->line[usr_cmd_pos + i];
 467         }
 468
 469         usr_cmd_len -= optimized_spaces;
 470         query[usr_cmd_len] = '\0';
 471
 472         /* filter commands */
 473         char *query_cmd = alloc_printf("_telnet_autocomplete_helper {%s*}", query);
 474
 475         if (!query_cmd) {
 476                 LOG_ERROR("Out of memory");
 477                 return;
 478         }
 479
 480         int retval = Jim_EvalSource(command_context->interp, __FILE__, __LINE__, query_cmd);
 481         free(query_cmd);
 482         if (retval != JIM_OK)
 483                 return;
 484
 485         Jim_Obj *list = Jim_GetResult(command_context->interp);
 486         Jim_IncrRefCount(list);
 487
 488         /* common prefix length of the matched commands */
 489         size_t common_len = 0;
 490         char *first_match = NULL; /* used to compute the common prefix length */
 491
 492         int len = Jim_ListLength(command_context->interp, list);
 493         for (int i = 0; i < len; i++) {
 494                 Jim_Obj *elem = Jim_ListGetIndex(command_context->interp, list, i);
 495                 Jim_IncrRefCount(elem);
 496
 497                 char *name = (char *)Jim_GetString(elem, NULL);
 498
 499                 /* validate the command */
 500                 bool ignore_cmd = false;
 501                 Jim_Cmd *jim_cmd = Jim_GetCommand(command_context->interp, elem, JIM_NONE);
 502
 503                 if (!jim_cmd) {
 504                         /* Why we are here? Let's ignore it! */
 505                         ignore_cmd = true;
 506                 } else if (jimcmd_is_oocd_command(jim_cmd)) {
 507                         struct command *cmd = jimcmd_privdata(jim_cmd);
 508
 509                         if (cmd && !cmd->handler && !cmd->jim_handler) {
 510                                 /* Initial part of a multi-word command. Ignore it! */
 511                                 ignore_cmd = true;
 512                         } else if (cmd && cmd->mode == COMMAND_CONFIG) {
 513                                 /* Not executable after config phase. Ignore it! */
 514                                 ignore_cmd = true;
 515                         }
 516                 }
 517
 518                 /* save the command in the prediction list */
 519                 if (!ignore_cmd) {
 520                         struct cmd_match *match = calloc(1, sizeof(struct cmd_match));
 521                         if (!match) {
 522                                 LOG_ERROR("Out of memory");
 523                                 Jim_DecrRefCount(command_context->interp, elem);
 524                                 break; /* break the for loop */
 525                         }
 526
 527                         if (list_empty(&matches)) {
 528                                 common_len = strlen(name);
 529                                 first_match = name;
 530                         } else {
 531                                 size_t new_common_len = usr_cmd_len; /* save some loops */
 532
 533                                 while (new_common_len < common_len && first_match[new_common_len] == name[new_common_len])
 534                                         new_common_len++;
 535
 536                                 common_len = new_common_len;
 537                         }
 538
 539                         match->cmd = name;
 540                         list_add_tail(&match->lh, &matches);
 541                 }
 542
 543                 Jim_DecrRefCount(command_context->interp, elem);
 544         }
 545         /* end of command filtering */
 546
 547         /* proceed with auto-completion */
 548         if (list_empty(&matches))
 549                 telnet_bell(connection);
 550         else if (common_len == usr_cmd_len && list_is_singular(&matches) && t_con->line_cursor == t_con->line_size)
 551                 telnet_insert(connection, " ", 1);
 552         else if (common_len > usr_cmd_len) {
 553                 int completion_size = common_len - usr_cmd_len;
 554                 if (telnet_insert(connection, first_match + usr_cmd_len, completion_size)) {
 555                         /* in bash this extra space is only added when the cursor in at the end of line */
 556                         if (list_is_singular(&matches) && t_con->line_cursor == t_con->line_size)
 557                                 telnet_insert(connection, " ", 1);
 558                 }
 559         } else if (!list_is_singular(&matches)) {
 560                 telnet_write(connection, "\n\r", 2);
 561
 562                 struct cmd_match *match;
 563                 list_for_each_entry(match, &matches, lh) {
 564                         telnet_write(connection, match->cmd, strlen(match->cmd));
 565                         telnet_write(connection, "\n\r", 2);
 566                 }
 567
 568                 telnet_prompt(connection);
 569                 telnet_write(connection, t_con->line, t_con->line_size);
 570
 571                 /* restore the terminal visible cursor location */
 572                 for (size_t i = t_con->line_cursor; i < t_con->line_size; i++)
 573                         telnet_write(connection, "\b", 1);
 574         }
 575
 576         /* destroy the command_list */
 577         struct cmd_match *tmp, *match;
 578         list_for_each_entry_safe(match, tmp, &matches, lh)
 579                 free(match);
 580
 581         Jim_DecrRefCount(command_context->interp, list);
 582 }
 583
 584 static int telnet_input(struct connection *connection)
 585 {
 586         int bytes_read;
 587         unsigned char buffer[TELNET_BUFFER_SIZE];
 588         unsigned char *buf_p;
 589         struct telnet_connection *t_con = connection->priv;
 590         struct command_context *command_context = connection->cmd_ctx;
 591
 592         bytes_read = connection_read(connection, buffer, TELNET_BUFFER_SIZE);
 593
 594         if (bytes_read == 0)
 595                 return ERROR_SERVER_REMOTE_CLOSED;
 596         else if (bytes_read == -1) {
 597                 LOG_ERROR("error during read: %s", strerror(errno));
 598                 return ERROR_SERVER_REMOTE_CLOSED;
 599         }
 600
 601         buf_p = buffer;
 602         while (bytes_read) {
 603                 switch (t_con->state) {
 604                         case TELNET_STATE_DATA:
 605                                 if (*buf_p == 0xff) {
 606                                         t_con->state = TELNET_STATE_IAC;
 607                                 } else {
 608                                         if (isprint(*buf_p)) {  /* printable character */
 609                                                 telnet_insert(connection, buf_p, 1);
 610                                         } else { /* non-printable */
 611                                                 if (*buf_p == 0x1b) {   /* escape */
 612                                                         t_con->state = TELNET_STATE_ESCAPE;
 613                                                         t_con->last_escape = '\x00';
 614                                                 } else if ((*buf_p == 0xd) || (*buf_p == 0xa)) { /* CR/LF */
 615                                                         int retval;
 616
 617                                                         /* skip over combinations with CR/LF and NUL characters */
 618                                                         if ((bytes_read > 1) && ((*(buf_p + 1) == 0xa) ||
 619                                                                         (*(buf_p + 1) == 0xd))) {
 620                                                                 buf_p++;
 621                                                                 bytes_read--;
 622                                                         }
 623                                                         if ((bytes_read > 1) && (*(buf_p + 1) == 0)) {
 624                                                                 buf_p++;
 625                                                                 bytes_read--;
 626                                                         }
 627                                                         t_con->line[t_con->line_size] = 0;
 628
 629                                                         telnet_write(connection, "\r\n\x00", 3);
 630
 631                                                         if (strcmp(t_con->line, "history") == 0) {
 632                                                                 retval = telnet_history_print(connection);
 633
 634                                                                 if (retval != ERROR_OK)
 635                                                                         return retval;
 636
 637                                                                 continue;
 638                                                         }
 639
 640                                                         /* save only non-blank not repeating lines in the history */
 641                                                         char *prev_line = t_con->history[(t_con->current_history > 0) ?
 642                                                                         t_con->current_history - 1 : TELNET_LINE_HISTORY_SIZE-1];
 643                                                         if (*t_con->line && (!prev_line ||
 644                                                                         strcmp(t_con->line, prev_line))) {
 645                                                                 /* if the history slot is already taken, free it */
 646                                                                 free(t_con->history[t_con->next_history]);
 647
 648                                                                 /* add line to history */
 649                                                                 t_con->history[t_con->next_history] = strdup(t_con->line);
 650
 651                                                                 /* wrap history at TELNET_LINE_HISTORY_SIZE */
 652                                                                 t_con->next_history = (t_con->next_history + 1) %
 653                                                                                 TELNET_LINE_HISTORY_SIZE;
 654
 655                                                                 /* current history line starts at the new entry */
 656                                                                 t_con->current_history =
 657                                                                                 t_con->next_history;
 658
 659                                                                 free(t_con->history[t_con->current_history]);
 660                                                                 t_con->history[t_con->current_history] = strdup("");
 661                                                         }
 662
 663                                                         t_con->line_size = 0;
 664
 665                                                         /* to suppress prompt in log callback during command execution */
 666                                                         t_con->prompt_visible = false;
 667
 668                                                         if (strcmp(t_con->line, "shutdown") == 0)
 669                                                                 telnet_save_history(t_con);
 670
 671                                                         retval = command_run_line(command_context, t_con->line);
 672
 673                                                         t_con->line_cursor = 0;
 674                                                         t_con->prompt_visible = true;
 675
 676                                                         if (retval == ERROR_COMMAND_CLOSE_CONNECTION)
 677                                                                 return ERROR_SERVER_REMOTE_CLOSED;
 678
 679                                                         /* the prompt is always * placed at the line beginning */
 680                                                         telnet_write(connection, "\r", 1);
 681
 682                                                         retval = telnet_prompt(connection);
 683                                                         if (retval == ERROR_SERVER_REMOTE_CLOSED)
 684                                                                 return ERROR_SERVER_REMOTE_CLOSED;
 685
 686                                                 } else if ((*buf_p == 0x7f) || (*buf_p == 0x8)) {       /* delete character */
 687                                                         if (t_con->line_cursor > 0) {
 688                                                                 if (t_con->line_cursor != t_con->line_size) {
 689                                                                         size_t i;
 690                                                                         telnet_write(connection, "\b", 1);
 691                                                                         t_con->line_cursor--;
 692                                                                         t_con->line_size--;
 693                                                                         memmove(t_con->line + t_con->line_cursor,
 694                                                                                         t_con->line + t_con->line_cursor + 1,
 695                                                                                         t_con->line_size -
 696                                                                                         t_con->line_cursor);
 697
 698                                                                         telnet_write(connection,
 699                                                                                         t_con->line + t_con->line_cursor,
 700                                                                                         t_con->line_size -
 701                                                                                         t_con->line_cursor);
 702                                                                         telnet_write(connection, " \b", 2);
 703                                                                         for (i = t_con->line_cursor; i < t_con->line_size; i++)
 704                                                                                 telnet_write(connection, "\b", 1);
 705                                                                 } else {
 706                                                                         t_con->line_size--;
 707                                                                         t_con->line_cursor--;
 708                                                                         /* back space: move the 'printer' head one char
 709                                                                          * back, overwrite with space, move back again */
 710                                                                         telnet_write(connection, "\b \b", 3);
 711                                                                 }
 712                                                         }
 713                                                 } else if (*buf_p == 0x15) {    /* clear line */
 714                                                         telnet_clear_line(connection, t_con);
 715                                                 } else if (*buf_p == CTRL('B')) {       /* cursor left */
 716                                                         if (t_con->line_cursor > 0) {
 717                                                                 telnet_write(connection, "\b", 1);
 718                                                                 t_con->line_cursor--;
 719                                                         }
 720                                                         t_con->state = TELNET_STATE_DATA;
 721                                                 } else if (*buf_p == CTRL('F')) {       /* cursor right */
 722                                                         if (t_con->line_cursor < t_con->line_size)
 723                                                                 telnet_write(connection, t_con->line + t_con->line_cursor++, 1);
 724                                                         t_con->state = TELNET_STATE_DATA;
 725                                                 } else if (*buf_p == CTRL('P')) {       /* cursor up */
 726                                                         telnet_history_up(connection);
 727                                                 } else if (*buf_p == CTRL('N')) {       /* cursor down */
 728                                                         telnet_history_down(connection);
 729                                                 } else if (*buf_p == CTRL('A')) {       /* move the cursor to the beginning of the line */
 730                                                         telnet_move_cursor(connection, 0);
 731                                                 } else if (*buf_p == CTRL('E')) {       /* move the cursor to the end of the line */
 732                                                         telnet_move_cursor(connection, t_con->line_size);
 733                                                 } else if (*buf_p == CTRL('K')) {       /* kill line to end */
 734                                                         if (t_con->line_cursor < t_con->line_size) {
 735                                                                 /* overwrite with space, until end of line, move back */
 736                                                                 for (size_t i = t_con->line_cursor; i < t_con->line_size; i++)
 737                                                                         telnet_write(connection, " ", 1);
 738                                                                 for (size_t i = t_con->line_cursor; i < t_con->line_size; i++)
 739                                                                         telnet_write(connection, "\b", 1);
 740                                                                 t_con->line[t_con->line_cursor] = '\0';
 741                                                                 t_con->line_size = t_con->line_cursor;
 742                                                         }
 743                                                 } else if (*buf_p == '\t') {
 744                                                         telnet_auto_complete(connection);
 745                                                 } else {
 746                                                         LOG_DEBUG("unhandled nonprintable: %2.2x", *buf_p);
 747                                                 }
 748                                         }
 749                                 }
 750                                 break;
 751                         case TELNET_STATE_IAC:
 752                                 switch (*buf_p) {
 753                                 case 0xfe:
 754                                         t_con->state = TELNET_STATE_DONT;
 755                                         break;
 756                                 case 0xfd:
 757                                         t_con->state = TELNET_STATE_DO;
 758                                         break;
 759                                 case 0xfc:
 760                                         t_con->state = TELNET_STATE_WONT;
 761                                         break;
 762                                 case 0xfb:
 763                                         t_con->state = TELNET_STATE_WILL;
 764                                         break;
 765                                 }
 766                                 break;
 767                         case TELNET_STATE_SB:
 768                                 break;
 769                         case TELNET_STATE_SE:
 770                                 break;
 771                         case TELNET_STATE_WILL:
 772                         case TELNET_STATE_WONT:
 773                         case TELNET_STATE_DO:
 774                         case TELNET_STATE_DONT:
 775                                 t_con->state = TELNET_STATE_DATA;
 776                                 break;
 777                         case TELNET_STATE_ESCAPE:
 778                                 if (t_con->last_escape == '[') {
 779                                         if (*buf_p == 'D') {    /* cursor left */
 780                                                 if (t_con->line_cursor > 0) {
 781                                                         telnet_write(connection, "\b", 1);
 782                                                         t_con->line_cursor--;
 783                                                 }
 784                                                 t_con->state = TELNET_STATE_DATA;
 785                                         } else if (*buf_p == 'C') {     /* cursor right */
 786                                                 if (t_con->line_cursor < t_con->line_size)
 787                                                         telnet_write(connection,
 788                                                                         t_con->line + t_con->line_cursor++, 1);
 789                                                 t_con->state = TELNET_STATE_DATA;
 790                                         } else if (*buf_p == 'A') {     /* cursor up */
 791                                                 telnet_history_up(connection);
 792                                         } else if (*buf_p == 'B') {     /* cursor down */
 793                                                 telnet_history_down(connection);
 794                                         } else if (*buf_p == 'F') { /* end key */
 795                                                 telnet_move_cursor(connection, t_con->line_size);
 796                                                 t_con->state = TELNET_STATE_DATA;
 797                                         } else if (*buf_p == 'H') { /* home key */
 798                                                 telnet_move_cursor(connection, 0);
 799                                                 t_con->state = TELNET_STATE_DATA;
 800                                         } else if (*buf_p == '3') {
 801                                                 t_con->last_escape = *buf_p;
 802                                         } else {
 803                                                 t_con->state = TELNET_STATE_DATA;
 804                                         }
 805                                 } else if (t_con->last_escape == '3') {
 806                                         /* Remove character */
 807                                         if (*buf_p == '~') {
 808                                                 if (t_con->line_cursor < t_con->line_size) {
 809                                                         size_t i;
 810                                                         t_con->line_size--;
 811                                                         /* remove char from line buffer */
 812                                                         memmove(t_con->line + t_con->line_cursor,
 813                                                                         t_con->line + t_con->line_cursor + 1,
 814                                                                         t_con->line_size - t_con->line_cursor);
 815
 816                                                         /* print remainder of buffer */
 817                                                         telnet_write(connection, t_con->line + t_con->line_cursor,
 818                                                                         t_con->line_size - t_con->line_cursor);
 819                                                         /* overwrite last char with whitespace */
 820                                                         telnet_write(connection, " \b", 2);
 821
 822                                                         /* move back to cursor position*/
 823                                                         for (i = t_con->line_cursor; i < t_con->line_size; i++)
 824                                                                 telnet_write(connection, "\b", 1);
 825                                                 }
 826
 827                                                 t_con->state = TELNET_STATE_DATA;
 828                                         } else
 829                                                 t_con->state = TELNET_STATE_DATA;
 830                                 } else if (t_con->last_escape == '\x00') {
 831                                         if (*buf_p == '[')
 832                                                 t_con->last_escape = *buf_p;
 833                                         else
 834                                                 t_con->state = TELNET_STATE_DATA;
 835                                 } else {
 836                                         LOG_ERROR("BUG: unexpected value in t_con->last_escape");
 837                                         t_con->state = TELNET_STATE_DATA;
 838                                 }
 839
 840                                 break;
 841                         default:
 842                                 LOG_ERROR("unknown telnet state");
 843                                 return ERROR_FAIL;
 844                 }
 845
 846                 bytes_read--;
 847                 buf_p++;
 848         }
 849
 850         return ERROR_OK;
 851 }
 852
 853 static int telnet_connection_closed(struct connection *connection)
 854 {
 855         struct telnet_connection *t_con = connection->priv;
 856         int i;
 857
 858         log_remove_callback(telnet_log_callback, connection);
 859
 860         free(t_con->prompt);
 861         t_con->prompt = NULL;
 862
 863         /* save telnet history */
 864         telnet_save_history(t_con);
 865
 866         for (i = 0; i < TELNET_LINE_HISTORY_SIZE; i++) {
 867                 free(t_con->history[i]);
 868                 t_con->history[i] = NULL;
 869         }
 870
 871         /* if this connection registered a debug-message receiver delete it */
 872         delete_debug_msg_receiver(connection->cmd_ctx, NULL);
 873
 874         free(connection->priv);
 875         connection->priv = NULL;
 876
 877         return ERROR_OK;
 878 }
 879
 880 int telnet_init(char *banner)
 881 {
 882         if (strcmp(telnet_port, "disabled") == 0) {
 883                 LOG_INFO("telnet server disabled");
 884                 return ERROR_OK;
 885         }
 886
 887         struct telnet_service *telnet_service =
 888                 malloc(sizeof(struct telnet_service));
 889
 890         if (!telnet_service) {
 891                 LOG_ERROR("Failed to allocate telnet service.");
 892                 return ERROR_FAIL;
 893         }
 894
 895         telnet_service->banner = banner;
 896
 897         int ret = add_service("telnet", telnet_port, CONNECTION_LIMIT_UNLIMITED,
 898                 telnet_new_connection, telnet_input, telnet_connection_closed,
 899                 telnet_service);
 900
 901         if (ret != ERROR_OK) {
 902                 free(telnet_service);
 903                 return ret;
 904         }
 905
 906         return ERROR_OK;
 907 }
 908
 909 /* daemon configuration command telnet_port */
 910 COMMAND_HANDLER(handle_telnet_port_command)
 911 {
 912         return CALL_COMMAND_HANDLER(server_pipe_command, &telnet_port);
 913 }
 914
 915 COMMAND_HANDLER(handle_exit_command)
 916 {
 917         return ERROR_COMMAND_CLOSE_CONNECTION;
 918 }
 919
 920 static const struct command_registration telnet_command_handlers[] = {
 921         {
 922                 .name = "exit",
 923                 .handler = handle_exit_command,
 924                 .mode = COMMAND_EXEC,
 925                 .usage = "",
 926                 .help = "exit telnet session",
 927         },
 928         {
 929                 .name = "telnet_port",
 930                 .handler = handle_telnet_port_command,
 931                 .mode = COMMAND_CONFIG,
 932                 .help = "Specify port on which to listen "
 933                         "for incoming telnet connections.  "
 934                         "Read help on 'gdb_port'.",
 935                 .usage = "[port_num]",
 936         },
 937         COMMAND_REGISTRATION_DONE
 938 };
 939
 940 int telnet_register_commands(struct command_context *cmd_ctx)
 941 {
 942         telnet_port = strdup("4444");
 943         return register_commands(cmd_ctx, NULL, telnet_command_handlers);
 944 }
 945
 946 void telnet_service_free(void)
 947 {
 948         free(telnet_port);
 949 }