usr.bin/chat/chat.c

   1 /*
   2  *      Chat -- a program for automatic session establishment (i.e. dial
   3  *              the phone and log in).
   4  *
   5  * Standard termination codes:
   6  *  0 - successful completion of the script
   7  *  1 - invalid argument, expect string too large, etc.
   8  *  2 - error on an I/O operation or fatal error condition.
   9  *  3 - timeout waiting for a simple string.
  10  *  4 - the first string declared as "ABORT"
  11  *  5 - the second string declared as "ABORT"
  12  *  6 - ... and so on for successive ABORT strings.
  13  *
  14  *      This software is in the public domain.
  15  *
  16  * -----------------
  17  *      added -T and -U option and \T and \U substitution to pass a phone
  18  *      number into chat script. Two are needed for some ISDN TA applications.
  19  *      Keith Dart <kdart@cisco.com>
  20  *
  21  *
  22  *      Added SAY keyword to send output to stderr.
  23  *      This allows to turn ECHO OFF and to output specific, user selected,
  24  *      text to give progress messages. This best works when stderr
  25  *      exists (i.e.: pppd in nodetach mode).
  26  *
  27  *      Added HANGUP directives to allow for us to be called
  28  *      back. When HANGUP is set to NO, chat will not hangup at HUP signal.
  29  *      We rely on timeouts in that case.
  30  *
  31  *      Added CLR_ABORT to clear previously set ABORT string. This has been
  32  *      dictated by the HANGUP above as "NO CARRIER" (for example) must be
  33  *      an ABORT condition until we know the other host is going to close
  34  *      the connection for call back. As soon as we have completed the
  35  *      first stage of the call back sequence, "NO CARRIER" is a valid, non
  36  *      fatal string. As soon as we got called back (probably get "CONNECT"),
  37  *      we should re-arm the ABORT "NO CARRIER". Hence the CLR_ABORT command.
  38  *      Note that CLR_ABORT packs the abort_strings[] array so that we do not
  39  *      have unused entries not being reclaimed.
  40  *
  41  *      In the same vein as above, added CLR_REPORT keyword.
  42  *
  43  *      Allow for comments. Line starting with '#' are comments and are
  44  *      ignored. If a '#' is to be expected as the first character, the
  45  *      expect string must be quoted.
  46  *
  47  *
  48  *              Francis Demierre <Francis@SwissMail.Com>
  49  *              Thu May 15 17:15:40 MET DST 1997
  50  *
  51  *
  52  *      Added -r "report file" switch & REPORT keyword.
  53  *              Robert Geer <bgeer@xmission.com>
  54  *
  55  *      Added -s "use stderr" and -S "don't use syslog" switches.
  56  *              June 18, 1997
  57  *              Karl O. Pinc <kop@meme.com>
  58  *
  59  *
  60  *      Added -e "echo" switch & ECHO keyword
  61  *              Dick Streefland <dicks@tasking.nl>
  62  *
  63  *
  64  *      Considerable updates and modifications by
  65  *              Al Longyear <longyear@pobox.com>
  66  *              Paul Mackerras <paulus@cs.anu.edu.au>
  67  *
  68  *
  69  *      The original author is:
  70  *
  71  *              Karl Fox <karl@MorningStar.Com>
  72  *              Morning Star Technologies, Inc.
  73  *              1760 Zollinger Road
  74  *              Columbus, OH  43221
  75  *              (614)451-1883
  76  *
  77  *
  78  * $FreeBSD: src/usr.bin/chat/chat.c,v 1.21 2003/10/31 06:22:03 kientzle Exp $
  79  * $DragonFly: src/usr.bin/chat/chat.c,v 1.11 2005/01/01 00:13:49 cpressey Exp $
  80  */
  81
  82 #include <sys/cdefs.h>
  83 #include <sys/types.h>
  84 #include <sys/stat.h>
  85 #include <ctype.h>
  86 #include <errno.h>
  87 #include <fcntl.h>
  88 #include <signal.h>
  89 #include <stdarg.h>
  90 #include <stdio.h>
  91 #include <stdlib.h>
  92 #include <string.h>
  93 #include <syslog.h>
  94 #include <termios.h>
  95 #include <time.h>
  96 #include <unistd.h>
  97
  98 #define STR_LEN 1024
  99
 100 #ifndef SIGTYPE
 101 #define SIGTYPE void
 102 #endif
 103
 104 #ifndef O_NONBLOCK
 105 #define O_NONBLOCK      O_NDELAY
 106 #endif
 107
 108 #define MAX_ABORTS              50
 109 #define MAX_REPORTS             50
 110 #define DEFAULT_CHAT_TIMEOUT    45
 111
 112 int echo          = 0;
 113 int verbose       = 0;
 114 int to_log        = 1;
 115 int to_stderr     = 0;
 116 int Verbose       = 0;
 117 int quiet         = 0;
 118 int exit_code     = 0;
 119 FILE* report_fp   = NULL;
 120 char *report_file = NULL;
 121 char *chat_file   = NULL;
 122 char *phone_num   = NULL;
 123 char *phone_num2  = NULL;
 124 int timeout       = DEFAULT_CHAT_TIMEOUT;
 125
 126 static char blank[] = "";
 127
 128 int have_tty_parameters = 0;
 129
 130 #define term_parms struct termios
 131 #define get_term_param(param) tcgetattr(0, param)
 132 #define set_term_param(param) tcsetattr(0, TCSANOW, param)
 133 struct termios saved_tty_parameters;
 134
 135 char *abort_string[MAX_ABORTS], *fail_reason = NULL,
 136         fail_buffer[50];
 137 int n_aborts = 0, abort_next = 0, timeout_next = 0, echo_next = 0;
 138 int clear_abort_next = 0;
 139
 140 char *report_string[MAX_REPORTS] ;
 141 char  report_buffer[50] ;
 142 int n_reports = 0, report_next = 0, report_gathering = 0 ;
 143 int clear_report_next = 0;
 144
 145 int say_next = 0, hup_next = 0;
 146
 147 void *dup_mem(void *b, size_t c);
 148 void *copy_of(char *s);
 149 static void usage(void);
 150 void chat_logf(const char *fmt, ...);
 151 void fatal(int code, const char *fmt, ...);
 152 SIGTYPE sigalrm(int signo);
 153 SIGTYPE sigint(int signo);
 154 SIGTYPE sigterm(int signo);
 155 SIGTYPE sighup(int signo);
 156 void init(void);
 157 void set_tty_parameters(void);
 158 void echo_stderr(int);
 159 void break_sequence(void);
 160 void terminate(int status);
 161 void do_file(char *chatfile);
 162 int  get_string(char *string);
 163 int  put_string(char *s);
 164 int  write_char(int c);
 165 int  put_char(int c);
 166 int  get_char(void);
 167 void chat_send(char *s);
 168 char *character(int c);
 169 void chat_expect(char *s);
 170 char *clean(char *s, int sending);
 171 void pack_array(char **array, int end);
 172 char *expect_strtok(char *, const char *);
 173 int vfmtmsg(char *, int, const char *, va_list);        /* vsprintf++ */
 174
 175 void *
 176 dup_mem(void *b, size_t c)
 177 {
 178     void *ans = malloc (c);
 179     if (!ans)
 180         fatal(2, "memory error!");
 181
 182     memcpy (ans, b, c);
 183     return ans;
 184 }
 185
 186 void *
 187 copy_of(char *s)
 188 {
 189     return dup_mem (s, strlen (s) + 1);
 190 }
 191
 192 /*
 193  * chat [-esSvV] [-f chat-file] [-r report-file] [-t timeout]
 194  *      [-T phone-number] [-U phone-number2] [chat-script]
 195  * where chat-script has the form:
 196  *      [...[[expect[-send[-expect...]] send expect[-send[-expect]] ...]]]
 197  *
 198  * Perform a UUCP-dialer-like chat script on stdin and stdout.
 199  */
 200 int
 201 main(int argc, char *argv[])
 202 {
 203     int option;
 204
 205     tzset();
 206
 207     while ((option = getopt(argc, argv, "ef:r:sSt:T:U:vV")) != -1) {
 208         switch (option) {
 209         case 'e':
 210             ++echo;
 211             break;
 212
 213         case 'f':
 214             if (chat_file != NULL)
 215                 free(chat_file);
 216             chat_file = copy_of(optarg);
 217             break;
 218
 219         case 'r':
 220             if (report_fp != NULL)
 221                 fclose(report_fp);
 222             if (report_file != NULL)
 223                 free(report_file);
 224             report_file = copy_of(optarg);
 225             report_fp = fopen(report_file, "a");
 226             if (report_fp != NULL) {
 227                 if (verbose)
 228                     fprintf(report_fp, "Opening \"%s\"...\n", report_file);
 229             } else
 230                 fatal(2, "cannot open \"%s\" for appending", report_file);
 231             break;
 232
 233         case 's':
 234             ++to_stderr;
 235             break;
 236
 237         case 'S':
 238             to_log = 0;
 239             break;
 240
 241         case 't':
 242             timeout = atoi(optarg);
 243             break;
 244
 245         case 'T':
 246             if (phone_num != NULL)
 247                 free(phone_num);
 248             phone_num = copy_of(optarg);
 249             break;
 250
 251         case 'U':
 252             if (phone_num2 != NULL)
 253                 free(phone_num2);
 254             phone_num2 = copy_of(optarg);
 255             break;
 256
 257         case 'v':
 258             ++verbose;
 259             break;
 260
 261         case 'V':
 262             ++Verbose;
 263             break;
 264
 265         default:
 266             usage();
 267             break;
 268         }
 269     }
 270
 271     argc -= optind;
 272     argv += optind;
 273
 274 /*
 275  * Default the report file to the stderr location
 276  */
 277     if (report_fp == NULL)
 278         report_fp = stderr;
 279
 280     if (to_log) {
 281         openlog("chat", LOG_PID | LOG_NDELAY, LOG_LOCAL2);
 282
 283         if (verbose)
 284             setlogmask(LOG_UPTO(LOG_INFO));
 285         else
 286             setlogmask(LOG_UPTO(LOG_WARNING));
 287     }
 288
 289     if (chat_file != NULL) {
 290         if (*argv != NULL)
 291             usage();
 292         else {
 293             init();
 294             do_file(chat_file);
 295         }
 296     } else {
 297         init();
 298         while (*argv != NULL && argc > 0) {
 299             chat_expect(*argv);
 300             argv++;
 301             argc--;
 302
 303             if (*argv != NULL && argc > 0) {
 304                 chat_send(*argv);
 305                 argv++;
 306                 argc--;
 307             }
 308         }
 309     }
 310
 311     terminate(0);
 312     return 0;
 313 }
 314
 315 /*
 316  *  Process a chat script when read from a file.
 317  */
 318
 319 void
 320 do_file(char *chatfile)
 321 {
 322     int linect, sendflg;
 323     char *sp, *arg, quote;
 324     char buf [STR_LEN];
 325     FILE *cfp;
 326
 327     cfp = fopen (chatfile, "r");
 328     if (cfp == NULL)
 329         fatal(1, "%s -- open failed: %m", chatfile);
 330
 331     linect = 0;
 332     sendflg = 0;
 333
 334     while (fgets(buf, STR_LEN, cfp) != NULL) {
 335         sp = strchr (buf, '\n');
 336         if (sp)
 337             *sp = '\0';
 338
 339         linect++;
 340         sp = buf;
 341
 342         /* lines starting with '#' are comments. If a real '#'
 343            is to be expected, it should be quoted .... */
 344         if ( *sp == '#' )
 345             continue;
 346
 347         while (*sp != '\0') {
 348             if (*sp == ' ' || *sp == '\t') {
 349                 ++sp;
 350                 continue;
 351             }
 352
 353             if (*sp == '"' || *sp == '\'') {
 354                 quote = *sp++;
 355                 arg = sp;
 356                 while (*sp != quote) {
 357                     if (*sp == '\0')
 358                         fatal(1, "unterminated quote (line %d)", linect);
 359
 360                     if (*sp++ == '\\') {
 361                         if (*sp != '\0')
 362                             ++sp;
 363                     }
 364                 }
 365             }
 366             else {
 367                 arg = sp;
 368                 while (*sp != '\0' && *sp != ' ' && *sp != '\t')
 369                     ++sp;
 370             }
 371
 372             if (*sp != '\0')
 373                 *sp++ = '\0';
 374
 375             if (sendflg)
 376                 chat_send (arg);
 377             else
 378                 chat_expect (arg);
 379             sendflg = !sendflg;
 380         }
 381     }
 382     fclose (cfp);
 383 }
 384
 385 /*
 386  *      We got an error parsing the command line.
 387  */
 388 static void
 389 usage(void)
 390 {
 391     fprintf(stderr,
 392       "Usage: chat [-esSvV] [-f chat-file] [-r report-file] [-t timeout]\n"
 393       "            [-T phone-number] [-U phone-number2] [chat-script]\n"
 394       "where chat-script has the form:\n"
 395       "            [...[[expect[-send[-expect...]] send expect[-send[-expect]] ...]]]\n");
 396     exit(1);
 397 }
 398
 399 char line[1024];
 400
 401 /*
 402  * Send a message to syslog and/or stderr.
 403  */
 404 void
 405 chat_logf(const char *fmt, ...)
 406 {
 407     va_list args;
 408
 409     va_start(args, fmt);
 410     vfmtmsg(line, sizeof(line), fmt, args);
 411     if (to_log)
 412         syslog(LOG_INFO, "%s", line);
 413     if (to_stderr)
 414         fprintf(stderr, "%s\n", line);
 415 }
 416
 417 /*
 418  *      Print an error message and terminate.
 419  */
 420
 421 void
 422 fatal(int code, const char *fmt, ...)
 423 {
 424     va_list args;
 425
 426     va_start(args, fmt);
 427     vfmtmsg(line, sizeof(line), fmt, args);
 428     if (to_log)
 429         syslog(LOG_ERR, "%s", line);
 430     if (to_stderr)
 431         fprintf(stderr, "%s\n", line);
 432     terminate(code);
 433 }
 434
 435 int alarmed = 0;
 436
 437 SIGTYPE sigalrm(int signo __unused)
 438 {
 439     int flags;
 440
 441     alarm(1);
 442     alarmed = 1;                /* Reset alarm to avoid race window */
 443     signal(SIGALRM, sigalrm);   /* that can cause hanging in read() */
 444
 445     if ((flags = fcntl(0, F_GETFL, 0)) == -1)
 446         fatal(2, "Can't get file mode flags on stdin: %m");
 447
 448     if (fcntl(0, F_SETFL, flags | O_NONBLOCK) == -1)
 449         fatal(2, "Can't set file mode flags on stdin: %m");
 450
 451     if (verbose)
 452         chat_logf("alarm");
 453 }
 454
 455 SIGTYPE sigint(int signo __unused)
 456 {
 457     fatal(2, "SIGINT");
 458 }
 459
 460 SIGTYPE sigterm(int signo __unused)
 461 {
 462     fatal(2, "SIGTERM");
 463 }
 464
 465 SIGTYPE sighup(int signo __unused)
 466 {
 467     fatal(2, "SIGHUP");
 468 }
 469
 470 void init(void)
 471 {
 472     signal(SIGINT, sigint);
 473     signal(SIGTERM, sigterm);
 474     signal(SIGHUP, sighup);
 475
 476     set_tty_parameters();
 477     signal(SIGALRM, sigalrm);
 478     alarm(0);
 479     alarmed = 0;
 480 }
 481
 482 void set_tty_parameters(void)
 483 {
 484 #if defined(get_term_param)
 485     term_parms t;
 486
 487     if (get_term_param (&t) < 0)
 488         fatal(2, "Can't get terminal parameters: %m");
 489
 490     saved_tty_parameters = t;
 491     have_tty_parameters  = 1;
 492
 493     t.c_iflag     |= IGNBRK | ISTRIP | IGNPAR;
 494     t.c_oflag      = 0;
 495     t.c_lflag      = 0;
 496     t.c_cc[VERASE] =
 497     t.c_cc[VKILL]  = 0;
 498     t.c_cc[VMIN]   = 1;
 499     t.c_cc[VTIME]  = 0;
 500
 501     if (set_term_param (&t) < 0)
 502         fatal(2, "Can't set terminal parameters: %m");
 503 #endif
 504 }
 505
 506 void break_sequence(void)
 507 {
 508     tcsendbreak (0, 0);
 509 }
 510
 511 void terminate(int status)
 512 {
 513     echo_stderr(-1);
 514     if (report_file != NULL && report_fp != NULL) {
 515 /*
 516  * Allow the last of the report string to be gathered before we terminate.
 517  */
 518         if (report_gathering) {
 519             int c;
 520             size_t rep_len;
 521
 522             rep_len = strlen(report_buffer);
 523             while (rep_len + 1 <= sizeof(report_buffer)) {
 524                 alarm(1);
 525                 c = get_char();
 526                 alarm(0);
 527                 if (c < 0 || iscntrl(c))
 528                     break;
 529                 report_buffer[rep_len] = c;
 530                 ++rep_len;
 531             }
 532             report_buffer[rep_len] = 0;
 533             fprintf (report_fp, "chat:  %s\n", report_buffer);
 534         }
 535         if (verbose)
 536             fprintf (report_fp, "Closing \"%s\".\n", report_file);
 537         fclose (report_fp);
 538         report_fp = NULL;
 539     }
 540
 541 #if defined(get_term_param)
 542     if (have_tty_parameters) {
 543         if (set_term_param (&saved_tty_parameters) < 0)
 544             fatal(2, "Can't restore terminal parameters: %m");
 545     }
 546 #endif
 547
 548     exit(status);
 549 }
 550
 551 /*
 552  *      'Clean up' this string.
 553  */
 554 char *
 555 clean(char *s, int sending)
 556 {
 557     char temp[STR_LEN], cur_chr;
 558     char *s1, *phchar;
 559     int add_return = sending;
 560 #define isoctal(chr) (((chr) >= '0') && ((chr) <= '7'))
 561
 562     s1 = temp;
 563     /* Don't overflow buffer, leave room for chars we append later */
 564     while (*s && s1 - temp < (off_t)(sizeof(temp) - 2 - add_return)) {
 565         cur_chr = *s++;
 566         if (cur_chr == '^') {
 567             cur_chr = *s++;
 568             if (cur_chr == '\0') {
 569                 *s1++ = '^';
 570                 break;
 571             }
 572             cur_chr &= 0x1F;
 573             if (cur_chr != 0) {
 574                 *s1++ = cur_chr;
 575             }
 576             continue;
 577         }
 578
 579         if (cur_chr != '\\') {
 580             *s1++ = cur_chr;
 581             continue;
 582         }
 583
 584         cur_chr = *s++;
 585         if (cur_chr == '\0') {
 586             if (sending) {
 587                 *s1++ = '\\';
 588                 *s1++ = '\\';
 589             }
 590             break;
 591         }
 592
 593         switch (cur_chr) {
 594         case 'b':
 595             *s1++ = '\b';
 596             break;
 597
 598         case 'c':
 599             if (sending && *s == '\0')
 600                 add_return = 0;
 601             else
 602                 *s1++ = cur_chr;
 603             break;
 604
 605         case '\\':
 606         case 'K':
 607         case 'p':
 608         case 'd':
 609             if (sending)
 610                 *s1++ = '\\';
 611
 612             *s1++ = cur_chr;
 613             break;
 614
 615         case 'T':
 616             if (sending && phone_num) {
 617                 for ( phchar = phone_num; *phchar != '\0'; phchar++)
 618                     *s1++ = *phchar;
 619             }
 620             else {
 621                 *s1++ = '\\';
 622                 *s1++ = 'T';
 623             }
 624             break;
 625
 626         case 'U':
 627             if (sending && phone_num2) {
 628                 for ( phchar = phone_num2; *phchar != '\0'; phchar++)
 629                     *s1++ = *phchar;
 630             }
 631             else {
 632                 *s1++ = '\\';
 633                 *s1++ = 'U';
 634             }
 635             break;
 636
 637         case 'q':
 638             quiet = 1;
 639             break;
 640
 641         case 'r':
 642             *s1++ = '\r';
 643             break;
 644
 645         case 'n':
 646             *s1++ = '\n';
 647             break;
 648
 649         case 's':
 650             *s1++ = ' ';
 651             break;
 652
 653         case 't':
 654             *s1++ = '\t';
 655             break;
 656
 657         case 'N':
 658             if (sending) {
 659                 *s1++ = '\\';
 660                 *s1++ = '\0';
 661             }
 662             else
 663                 *s1++ = 'N';
 664             break;
 665
 666         default:
 667             if (isoctal (cur_chr)) {
 668                 cur_chr &= 0x07;
 669                 if (isoctal (*s)) {
 670                     cur_chr <<= 3;
 671                     cur_chr |= *s++ - '0';
 672                     if (isoctal (*s)) {
 673                         cur_chr <<= 3;
 674                         cur_chr |= *s++ - '0';
 675                     }
 676                 }
 677
 678                 if (cur_chr != 0 || sending) {
 679                     if (sending && (cur_chr == '\\' || cur_chr == 0))
 680                         *s1++ = '\\';
 681                     *s1++ = cur_chr;
 682                 }
 683                 break;
 684             }
 685
 686             if (sending)
 687                 *s1++ = '\\';
 688             *s1++ = cur_chr;
 689             break;
 690         }
 691     }
 692
 693     if (add_return)
 694         *s1++ = '\r';
 695
 696     *s1++ = '\0'; /* guarantee closure */
 697     *s1++ = '\0'; /* terminate the string */
 698     return dup_mem (temp, (size_t) (s1 - temp)); /* may have embedded nuls */
 699 }
 700
 701 /*
 702  * A modified version of 'strtok'. This version skips \ sequences.
 703  */
 704
 705 char *
 706 expect_strtok (char *s, const char *term)
 707 {
 708     static  char *str   = blank;
 709     int     escape_flag = 0;
 710     char   *result;
 711
 712 /*
 713  * If a string was specified then do initial processing.
 714  */
 715     if (s)
 716         str = s;
 717
 718 /*
 719  * If this is the escape flag then reset it and ignore the character.
 720  */
 721     if (*str)
 722         result = str;
 723     else
 724         result = NULL;
 725
 726     while (*str) {
 727         if (escape_flag) {
 728             escape_flag = 0;
 729             ++str;
 730             continue;
 731         }
 732
 733         if (*str == '\\') {
 734             ++str;
 735             escape_flag = 1;
 736             continue;
 737         }
 738
 739 /*
 740  * If this is not in the termination string, continue.
 741  */
 742         if (strchr (term, *str) == NULL) {
 743             ++str;
 744             continue;
 745         }
 746
 747 /*
 748  * This is the terminator. Mark the end of the string and stop.
 749  */
 750         *str++ = '\0';
 751         break;
 752     }
 753     return (result);
 754 }
 755
 756 /*
 757  * Process the expect string
 758  */
 759
 760 void
 761 chat_expect(char *s)
 762 {
 763     char *expect;
 764     char *reply;
 765
 766     if (strcmp(s, "HANGUP") == 0) {
 767         ++hup_next;
 768         return;
 769     }
 770
 771     if (strcmp(s, "ABORT") == 0) {
 772         ++abort_next;
 773         return;
 774     }
 775
 776     if (strcmp(s, "CLR_ABORT") == 0) {
 777         ++clear_abort_next;
 778         return;
 779     }
 780
 781     if (strcmp(s, "REPORT") == 0) {
 782         ++report_next;
 783         return;
 784     }
 785
 786     if (strcmp(s, "CLR_REPORT") == 0) {
 787         ++clear_report_next;
 788         return;
 789     }
 790
 791     if (strcmp(s, "TIMEOUT") == 0) {
 792         ++timeout_next;
 793         return;
 794     }
 795
 796     if (strcmp(s, "ECHO") == 0) {
 797         ++echo_next;
 798         return;
 799     }
 800
 801     if (strcmp(s, "SAY") == 0) {
 802         ++say_next;
 803         return;
 804     }
 805
 806 /*
 807  * Fetch the expect and reply string.
 808  */
 809     for (;;) {
 810         expect = expect_strtok (s, "-");
 811         s      = NULL;
 812
 813         if (expect == NULL)
 814             return;
 815
 816         reply = expect_strtok (s, "-");
 817
 818 /*
 819  * Handle the expect string. If successful then exit.
 820  */
 821         if (get_string (expect))
 822             return;
 823
 824 /*
 825  * If there is a sub-reply string then send it. Otherwise any condition
 826  * is terminal.
 827  */
 828         if (reply == NULL || exit_code != 3)
 829             break;
 830
 831         chat_send (reply);
 832     }
 833
 834 /*
 835  * The expectation did not occur. This is terminal.
 836  */
 837     if (fail_reason)
 838         chat_logf("Failed (%s)", fail_reason);
 839     else
 840         chat_logf("Failed");
 841     terminate(exit_code);
 842 }
 843
 844 /*
 845  * Translate the input character to the appropriate string for printing
 846  * the data.
 847  */
 848
 849 char *
 850 character(int c)
 851 {
 852     static char string[10];
 853     const char *meta;
 854
 855     meta = (c & 0x80) ? "M-" : "";
 856     c &= 0x7F;
 857
 858     if (c < 32)
 859         sprintf(string, "%s^%c", meta, (int)c + '@');
 860     else if (c == 127)
 861         sprintf(string, "%s^?", meta);
 862     else
 863         sprintf(string, "%s%c", meta, c);
 864
 865     return (string);
 866 }
 867
 868 /*
 869  *  process the reply string
 870  */
 871 void
 872 chat_send(char *s)
 873 {
 874     if (say_next) {
 875         say_next = 0;
 876         s = clean(s,0);
 877         write(STDERR_FILENO, s, strlen(s));
 878         free(s);
 879         return;
 880     }
 881
 882     if (hup_next) {
 883         hup_next = 0;
 884         if (strcmp(s, "OFF") == 0)
 885            signal(SIGHUP, SIG_IGN);
 886         else
 887            signal(SIGHUP, sighup);
 888         return;
 889     }
 890
 891     if (echo_next) {
 892         echo_next = 0;
 893         echo = (strcmp(s, "ON") == 0);
 894         return;
 895     }
 896
 897     if (abort_next) {
 898         char *s1;
 899
 900         abort_next = 0;
 901
 902         if (n_aborts >= MAX_ABORTS)
 903             fatal(2, "Too many ABORT strings");
 904
 905         s1 = clean(s, 0);
 906
 907         if (strlen(s1) > strlen(s)
 908             || strlen(s1) + 1 > sizeof(fail_buffer))
 909             fatal(1, "Illegal or too-long ABORT string ('%v')", s);
 910
 911         abort_string[n_aborts++] = s1;
 912
 913         if (verbose)
 914             chat_logf("abort on (%v)", s);
 915         return;
 916     }
 917
 918     if (clear_abort_next) {
 919         char *s1;
 920         int   i;
 921         int   old_max;
 922         int   pack = 0;
 923
 924         clear_abort_next = 0;
 925
 926         s1 = clean(s, 0);
 927
 928         if (strlen(s1) > strlen(s)
 929             || strlen(s1) + 1 > sizeof(fail_buffer))
 930             fatal(1, "Illegal or too-long CLR_ABORT string ('%v')", s);
 931
 932         old_max = n_aborts;
 933         for (i=0; i < n_aborts; i++) {
 934             if ( strcmp(s1,abort_string[i]) == 0 ) {
 935                 free(abort_string[i]);
 936                 abort_string[i] = NULL;
 937                 pack++;
 938                 n_aborts--;
 939                 if (verbose)
 940                     chat_logf("clear abort on (%v)", s);
 941             }
 942         }
 943         free(s1);
 944         if (pack)
 945             pack_array(abort_string,old_max);
 946         return;
 947     }
 948
 949     if (report_next) {
 950         char *s1;
 951
 952         report_next = 0;
 953         if (n_reports >= MAX_REPORTS)
 954             fatal(2, "Too many REPORT strings");
 955
 956         s1 = clean(s, 0);
 957
 958         if (strlen(s1) > strlen(s) || strlen(s1) > sizeof fail_buffer - 1)
 959             fatal(1, "Illegal or too-long REPORT string ('%v')", s);
 960
 961         report_string[n_reports++] = s1;
 962
 963         if (verbose)
 964             chat_logf("report (%v)", s);
 965         return;
 966     }
 967
 968     if (clear_report_next) {
 969         char *s1;
 970         int   i;
 971         int   old_max;
 972         int   pack = 0;
 973
 974         clear_report_next = 0;
 975
 976         s1 = clean(s, 0);
 977
 978         if (strlen(s1) > strlen(s) || strlen(s1) > sizeof fail_buffer - 1)
 979             fatal(1, "Illegal or too-long REPORT string ('%v')", s);
 980
 981         old_max = n_reports;
 982         for (i=0; i < n_reports; i++) {
 983             if ( strcmp(s1,report_string[i]) == 0 ) {
 984                 free(report_string[i]);
 985                 report_string[i] = NULL;
 986                 pack++;
 987                 n_reports--;
 988                 if (verbose)
 989                     chat_logf("clear report (%v)", s);
 990             }
 991         }
 992         free(s1);
 993         if (pack)
 994             pack_array(report_string,old_max);
 995
 996         return;
 997     }
 998
 999     if (timeout_next) {
1000         timeout_next = 0;
1001         timeout = atoi(s);
1002
1003         if (timeout <= 0)
1004             timeout = DEFAULT_CHAT_TIMEOUT;
1005
1006         if (verbose)
1007             chat_logf("timeout set to %d seconds", timeout);
1008
1009         return;
1010     }
1011
1012     if (strcmp(s, "EOT") == 0)
1013         s = strdup("^D\\c");
1014     else if (strcmp(s, "BREAK") == 0)
1015         s = strdup("\\K\\c");
1016
1017     if (!put_string(s))
1018         fatal(1, "Failed");
1019 }
1020
1021 int
1022 get_char(void)
1023 {
1024     int status;
1025     char c;
1026
1027     status = read(STDIN_FILENO, &c, 1);
1028
1029     switch (status) {
1030     case 1:
1031         return ((int)c & 0x7F);
1032
1033     default:
1034         chat_logf("warning: read() on stdin returned %d", status);
1035
1036     case -1:
1037         if ((status = fcntl(0, F_GETFL, 0)) == -1)
1038             fatal(2, "Can't get file mode flags on stdin: %m");
1039
1040         if (fcntl(0, F_SETFL, status & ~O_NONBLOCK) == -1)
1041             fatal(2, "Can't set file mode flags on stdin: %m");
1042
1043         return (-1);
1044     }
1045 }
1046
1047 int put_char(int c)
1048 {
1049     int status;
1050     char ch = c;
1051
1052     usleep(10000);              /* inter-character typing delay (?) */
1053
1054     status = write(STDOUT_FILENO, &ch, 1);
1055
1056     switch (status) {
1057     case 1:
1058         return (0);
1059
1060     default:
1061         chat_logf("warning: write() on stdout returned %d", status);
1062
1063     case -1:
1064         if ((status = fcntl(0, F_GETFL, 0)) == -1)
1065             fatal(2, "Can't get file mode flags on stdin, %m");
1066
1067         if (fcntl(0, F_SETFL, status & ~O_NONBLOCK) == -1)
1068             fatal(2, "Can't set file mode flags on stdin: %m");
1069
1070         return (-1);
1071     }
1072 }
1073
1074 int
1075 write_char(int c)
1076 {
1077     if (alarmed || put_char(c) < 0) {
1078         alarm(0);
1079         alarmed = 0;
1080
1081         if (verbose) {
1082             if (errno == EINTR || errno == EWOULDBLOCK)
1083                 chat_logf(" -- write timed out");
1084             else
1085                 chat_logf(" -- write failed: %m");
1086         }
1087         return (0);
1088     }
1089     return (1);
1090 }
1091
1092 int
1093 put_string(char *s)
1094 {
1095     quiet = 0;
1096     s = clean(s, 1);
1097
1098     if (verbose)
1099         chat_logf("send (%v)", quiet ? "??????" : s);
1100
1101     alarm(timeout); alarmed = 0;
1102
1103     while (*s) {
1104         char c = *s++;
1105
1106         if (c != '\\') {
1107             if (!write_char (c))
1108                 return 0;
1109             continue;
1110         }
1111
1112         c = *s++;
1113         switch (c) {
1114         case 'd':
1115             sleep(1);
1116             break;
1117
1118         case 'K':
1119             break_sequence();
1120             break;
1121
1122         case 'p':
1123             usleep(10000);      /* 1/100th of a second (arg is microseconds) */
1124             break;
1125
1126         default:
1127             if (!write_char (c))
1128                 return 0;
1129             break;
1130         }
1131     }
1132
1133     alarm(0);
1134     alarmed = 0;
1135     return (1);
1136 }
1137
1138 /*
1139  *      Echo a character to stderr.
1140  *      When called with -1, a '\n' character is generated when
1141  *      the cursor is not at the beginning of a line.
1142  */
1143 void
1144 echo_stderr(int n)
1145 {
1146     static int need_lf;
1147     char *s;
1148
1149     switch (n) {
1150     case '\r':          /* ignore '\r' */
1151         break;
1152     case -1:
1153         if (need_lf == 0)
1154             break;
1155         /* FALLTHROUGH */
1156     case '\n':
1157         write(STDERR_FILENO, "\n", 1);
1158         need_lf = 0;
1159         break;
1160     default:
1161         s = character(n);
1162         write(STDERR_FILENO, s, strlen(s));
1163         need_lf = 1;
1164         break;
1165     }
1166 }
1167
1168 /*
1169  *      'Wait for' this string to appear on this file descriptor.
1170  */
1171 int
1172 get_string(char *string)
1173 {
1174     char temp[STR_LEN];
1175     int c, printed = 0;
1176     size_t len, minlen;
1177     char *s = temp, *end = s + STR_LEN;
1178     char *logged = temp;
1179
1180     fail_reason = NULL;
1181
1182     if (strlen(string) > STR_LEN) {
1183         chat_logf("expect string is too long");
1184         exit_code = 1;
1185         return 0;
1186     }
1187
1188     string = clean(string, 0);
1189     len = strlen(string);
1190     minlen = (len > sizeof(fail_buffer)? len: sizeof(fail_buffer)) - 1;
1191
1192     if (verbose)
1193         chat_logf("expect (%v)", string);
1194
1195     if (len == 0) {
1196         if (verbose)
1197             chat_logf("got it");
1198         return (1);
1199     }
1200
1201     alarm(timeout);
1202     alarmed = 0;
1203
1204     while ( ! alarmed && (c = get_char()) >= 0) {
1205         int n, abort_len, report_len;
1206
1207         if (echo)
1208             echo_stderr(c);
1209         if (verbose && c == '\n') {
1210             if (s == logged)
1211                 chat_logf("");  /* blank line */
1212             else
1213                 chat_logf("%0.*v", s - logged, logged);
1214             logged = s + 1;
1215         }
1216
1217         *s++ = c;
1218
1219         if (verbose && s >= logged + 80) {
1220             chat_logf("%0.*v", s - logged, logged);
1221             logged = s;
1222         }
1223
1224         if (Verbose) {
1225            if (c == '\n')
1226                fputc( '\n', stderr );
1227            else if (c != '\r')
1228                fprintf( stderr, "%s", character(c) );
1229         }
1230
1231         if (!report_gathering) {
1232             for (n = 0; n < n_reports; ++n) {
1233                 if ((report_string[n] != NULL) &&
1234                     s - temp >= (report_len = strlen(report_string[n])) &&
1235                     strncmp(s - report_len, report_string[n], report_len) == 0) {
1236                     time_t time_now   = time (NULL);
1237                     struct tm* tm_now = localtime (&time_now);
1238
1239                     strftime (report_buffer, 20, "%b %d %H:%M:%S ", tm_now);
1240                     strcat (report_buffer, report_string[n]);
1241
1242                     report_string[n] = NULL;
1243                     report_gathering = 1;
1244                     break;
1245                 }
1246             }
1247         }
1248         else {
1249             if (!iscntrl (c)) {
1250                 int rep_len = strlen (report_buffer);
1251                 report_buffer[rep_len]     = c;
1252                 report_buffer[rep_len + 1] = '\0';
1253             }
1254             else {
1255                 report_gathering = 0;
1256                 fprintf (report_fp, "chat:  %s\n", report_buffer);
1257             }
1258         }
1259
1260         if ((size_t)(s - temp) >= len &&
1261             c == string[len - 1] &&
1262             strncmp(s - len, string, len) == 0) {
1263             if (verbose) {
1264                 if (s > logged)
1265                     chat_logf("%0.*v", s - logged, logged);
1266                 chat_logf(" -- got it\n");
1267             }
1268
1269             alarm(0);
1270             alarmed = 0;
1271             return (1);
1272         }
1273
1274         for (n = 0; n < n_aborts; ++n) {
1275             if (s - temp >= (abort_len = strlen(abort_string[n])) &&
1276                 strncmp(s - abort_len, abort_string[n], abort_len) == 0) {
1277                 if (verbose) {
1278                     if (s > logged)
1279                         chat_logf("%0.*v", s - logged, logged);
1280                     chat_logf(" -- failed");
1281                 }
1282
1283                 alarm(0);
1284                 alarmed = 0;
1285                 exit_code = n + 4;
1286                 strcpy(fail_reason = fail_buffer, abort_string[n]);
1287                 return (0);
1288             }
1289         }
1290
1291         if (s >= end) {
1292             if (logged < s - minlen) {
1293                 chat_logf("%0.*v", s - logged, logged);
1294                 logged = s;
1295             }
1296             s -= minlen;
1297             memmove(temp, s, minlen);
1298             logged = temp + (logged - s);
1299             s = temp + minlen;
1300         }
1301
1302         if (alarmed && verbose)
1303             chat_logf("warning: alarm synchronization problem");
1304     }
1305
1306     alarm(0);
1307
1308     if (verbose && printed) {
1309         if (alarmed)
1310             chat_logf(" -- read timed out");
1311         else
1312             chat_logf(" -- read failed: %m");
1313     }
1314
1315     exit_code = 3;
1316     alarmed   = 0;
1317     return (0);
1318 }
1319
1320 void
1321 pack_array(char **array, int end)
1322 {
1323     int i, j;
1324
1325     for (i = 0; i < end; i++) {
1326         if (array[i] == NULL) {
1327             for (j = i+1; j < end; ++j)
1328                 if (array[j] != NULL)
1329                     array[i++] = array[j];
1330             for (; i < end; ++i)
1331                 array[i] = NULL;
1332             break;
1333         }
1334     }
1335 }
1336
1337 /*
1338  * vfmtmsg - format a message into a buffer.  Like vsprintf except we
1339  * also specify the length of the output buffer, and we handle the
1340  * %m (error message) format.
1341  * Doesn't do floating-point formats.
1342  * Returns the number of chars put into buf.
1343  */
1344 #define OUTCHAR(c)      (buflen > 0? (--buflen, *buf++ = (c)): 0)
1345
1346 int
1347 vfmtmsg(char *buf, int buflen, const char *fmt, va_list args)
1348 {
1349     int c, i, n;
1350     int width, prec, fillch;
1351     int base, len, neg, quoted;
1352     unsigned long val = 0;
1353     char *str, *buf0;
1354     const char *f;
1355     unsigned char *p;
1356     char num[32];
1357     static char hexchars[] = "0123456789abcdef";
1358
1359     buf0 = buf;
1360     --buflen;
1361     while (buflen > 0) {
1362         for (f = fmt; *f != '%' && *f != 0; ++f)
1363             ;
1364         if (f > fmt) {
1365             len = f - fmt;
1366             if (len > buflen)
1367                 len = buflen;
1368             memcpy(buf, fmt, len);
1369             buf += len;
1370             buflen -= len;
1371             fmt = f;
1372         }
1373         if (*fmt == 0)
1374             break;
1375         c = *++fmt;
1376         width = prec = 0;
1377         fillch = ' ';
1378         if (c == '0') {
1379             fillch = '0';
1380             c = *++fmt;
1381         }
1382         if (c == '*') {
1383             width = va_arg(args, int);
1384             c = *++fmt;
1385         } else {
1386             while (isdigit(c)) {
1387                 width = width * 10 + c - '0';
1388                 c = *++fmt;
1389             }
1390         }
1391         if (c == '.') {
1392             c = *++fmt;
1393             if (c == '*') {
1394                 prec = va_arg(args, int);
1395                 c = *++fmt;
1396             } else {
1397                 while (isdigit(c)) {
1398                     prec = prec * 10 + c - '0';
1399                     c = *++fmt;
1400                 }
1401             }
1402         }
1403         str = 0;
1404         base = 0;
1405         neg = 0;
1406         ++fmt;
1407         switch (c) {
1408         case 'd':
1409             i = va_arg(args, int);
1410             if (i < 0) {
1411                 neg = 1;
1412                 val = -i;
1413             } else
1414                 val = i;
1415             base = 10;
1416             break;
1417         case 'o':
1418             val = va_arg(args, unsigned int);
1419             base = 8;
1420             break;
1421         case 'x':
1422             val = va_arg(args, unsigned int);
1423             base = 16;
1424             break;
1425         case 'p':
1426             val = (unsigned long) va_arg(args, void *);
1427             base = 16;
1428             neg = 2;
1429             break;
1430         case 's':
1431             str = va_arg(args, char *);
1432             break;
1433         case 'c':
1434             num[0] = va_arg(args, int);
1435             num[1] = 0;
1436             str = num;
1437             break;
1438         case 'm':
1439             str = strerror(errno);
1440             break;
1441         case 'v':               /* "visible" string */
1442         case 'q':               /* quoted string */
1443             quoted = c == 'q';
1444             p = va_arg(args, unsigned char *);
1445             if (fillch == '0' && prec > 0) {
1446                 n = prec;
1447             } else {
1448                 n = strlen((char *)p);
1449                 if (prec > 0 && prec < n)
1450                     n = prec;
1451             }
1452             while (n > 0 && buflen > 0) {
1453                 c = *p++;
1454                 --n;
1455                 if (!quoted && c >= 0x80) {
1456                     OUTCHAR('M');
1457                     OUTCHAR('-');
1458                     c -= 0x80;
1459                 }
1460                 if (quoted && (c == '"' || c == '\\'))
1461                     OUTCHAR('\\');
1462                 if (c < 0x20 || (0x7f <= c && c < 0xa0)) {
1463                     if (quoted) {
1464                         OUTCHAR('\\');
1465                         switch (c) {
1466                         case '\t':      OUTCHAR('t');   break;
1467                         case '\n':      OUTCHAR('n');   break;
1468                         case '\b':      OUTCHAR('b');   break;
1469                         case '\f':      OUTCHAR('f');   break;
1470                         default:
1471                             OUTCHAR('x');
1472                             OUTCHAR(hexchars[c >> 4]);
1473                             OUTCHAR(hexchars[c & 0xf]);
1474                         }
1475                     } else {
1476                         if (c == '\t')
1477                             OUTCHAR(c);
1478                         else {
1479                             OUTCHAR('^');
1480                             OUTCHAR(c ^ 0x40);
1481                         }
1482                     }
1483                 } else
1484                     OUTCHAR(c);
1485             }
1486             continue;
1487         default:
1488             *buf++ = '%';
1489             if (c != '%')
1490                 --fmt;          /* so %z outputs %z etc. */
1491             --buflen;
1492             continue;
1493         }
1494         if (base != 0) {
1495             str = num + sizeof(num);
1496             *--str = 0;
1497             while (str > num + neg) {
1498                 *--str = hexchars[val % base];
1499                 val = val / base;
1500                 if (--prec <= 0 && val == 0)
1501                     break;
1502             }
1503             switch (neg) {
1504             case 1:
1505                 *--str = '-';
1506                 break;
1507             case 2:
1508                 *--str = 'x';
1509                 *--str = '0';
1510                 break;
1511             }
1512             len = num + sizeof(num) - 1 - str;
1513         } else {
1514             len = strlen(str);
1515             if (prec > 0 && len > prec)
1516                 len = prec;
1517         }
1518         if (width > 0) {
1519             if (width > buflen)
1520                 width = buflen;
1521             if ((n = width - len) > 0) {
1522                 buflen -= n;
1523                 for (; n > 0; --n)
1524                     *buf++ = fillch;
1525             }
1526         }
1527         if (len > buflen)
1528             len = buflen;
1529         memcpy(buf, str, len);
1530         buf += len;
1531         buflen -= len;
1532     }
1533     *buf = 0;
1534     return buf - buf0;
1535 }