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  */
  80
  81 #include <sys/types.h>
  82 #include <sys/stat.h>
  83 #include <ctype.h>
  84 #include <errno.h>
  85 #include <fcntl.h>
  86 #include <signal.h>
  87 #include <stdarg.h>
  88 #include <stdio.h>
  89 #include <stdlib.h>
  90 #include <string.h>
  91 #include <syslog.h>
  92 #include <termios.h>
  93 #include <time.h>
  94 #include <unistd.h>
  95
  96 #define STR_LEN 1024
  97
  98 #ifndef SIGTYPE
  99 #define SIGTYPE void
 100 #endif
 101
 102 #ifndef O_NONBLOCK
 103 #define O_NONBLOCK      O_NDELAY
 104 #endif
 105
 106 #define MAX_ABORTS              50
 107 #define MAX_REPORTS             50
 108 #define DEFAULT_CHAT_TIMEOUT    45
 109
 110 static int echo          = 0;
 111 static int verbose       = 0;
 112 static int to_log        = 1;
 113 static int to_stderr     = 0;
 114 static int Verbose       = 0;
 115 static int quiet         = 0;
 116 static int exit_code     = 0;
 117 static FILE* report_fp   = NULL;
 118 static char *report_file = NULL;
 119 static char *chat_file   = NULL;
 120 static char *phone_num   = NULL;
 121 static char *phone_num2  = NULL;
 122 static int timeout       = DEFAULT_CHAT_TIMEOUT;
 123
 124 static char blank[] = "";
 125
 126 static int have_tty_parameters = 0;
 127
 128 #define term_parms struct termios
 129 #define get_term_param(param) tcgetattr(0, param)
 130 #define set_term_param(param) tcsetattr(0, TCSANOW, param)
 131 static struct termios saved_tty_parameters;
 132
 133 static char *abort_string[MAX_ABORTS], *fail_reason = NULL,
 134         fail_buffer[50];
 135 static int n_aborts = 0, abort_next = 0, timeout_next = 0, echo_next = 0;
 136 static int clear_abort_next = 0;
 137
 138 static char *report_string[MAX_REPORTS] ;
 139 static char  report_buffer[50] ;
 140 static int n_reports = 0, report_next = 0, report_gathering = 0 ;
 141 static int clear_report_next = 0;
 142
 143 static int say_next = 0, hup_next = 0;
 144
 145 static void *dup_mem(void *b, size_t c);
 146 static void *copy_of(char *s);
 147 static void usage(void);
 148 static void chat_logf(const char *fmt, ...);
 149 static void fatal(int code, const char *fmt, ...) __dead2;
 150 static SIGTYPE sigalrm(int signo);
 151 static SIGTYPE sigint(int signo) __dead2;
 152 static SIGTYPE sigterm(int signo) __dead2;
 153 static SIGTYPE sighup(int signo) __dead2;
 154 static void init(void);
 155 static void set_tty_parameters(void);
 156 static void echo_stderr(int);
 157 static void break_sequence(void);
 158 static void terminate(int status) __dead2;
 159 static void do_file(char *chatfile);
 160 static int  get_string(char *string);
 161 static int  put_string(char *s);
 162 static int  write_char(int c);
 163 static int  put_char(int c);
 164 static int  get_char(void);
 165 static void chat_send(char *s);
 166 static char *character(int c);
 167 static void chat_expect(char *s);
 168 static char *clean(char *s, int sending);
 169 static void pack_array(char **array, int end);
 170 static char *expect_strtok(char *, const char *);
 171 static int vfmtmsg(char *, int, const char *, va_list); /* vsprintf++ */
 172
 173 static void *
 174 dup_mem(void *b, size_t c)
 175 {
 176     void *ans = malloc (c);
 177     if (!ans)
 178         fatal(2, "memory error!");
 179
 180     memcpy (ans, b, c);
 181     return ans;
 182 }
 183
 184 static void *
 185 copy_of(char *s)
 186 {
 187     return dup_mem (s, strlen (s) + 1);
 188 }
 189
 190 /*
 191  * chat [-esSvV] [-f chat-file] [-r report-file] [-t timeout]
 192  *      [-T phone-number] [-U phone-number2] [chat-script]
 193  * where chat-script has the form:
 194  *      [...[[expect[-send[-expect...]] send expect[-send[-expect]] ...]]]
 195  *
 196  * Perform a UUCP-dialer-like chat script on stdin and stdout.
 197  */
 198 int
 199 main(int argc, char *argv[])
 200 {
 201     int option;
 202
 203     tzset();
 204
 205     while ((option = getopt(argc, argv, "ef:r:sSt:T:U:vV")) != -1) {
 206         switch (option) {
 207         case 'e':
 208             ++echo;
 209             break;
 210
 211         case 'f':
 212             if (chat_file != NULL)
 213                 free(chat_file);
 214             chat_file = copy_of(optarg);
 215             break;
 216
 217         case 'r':
 218             if (report_fp != NULL)
 219                 fclose(report_fp);
 220             if (report_file != NULL)
 221                 free(report_file);
 222             report_file = copy_of(optarg);
 223             report_fp = fopen(report_file, "a");
 224             if (report_fp != NULL) {
 225                 if (verbose)
 226                     fprintf(report_fp, "Opening \"%s\"...\n", report_file);
 227             } else
 228                 fatal(2, "cannot open \"%s\" for appending", report_file);
 229             break;
 230
 231         case 's':
 232             ++to_stderr;
 233             break;
 234
 235         case 'S':
 236             to_log = 0;
 237             break;
 238
 239         case 't':
 240             timeout = atoi(optarg);
 241             break;
 242
 243         case 'T':
 244             if (phone_num != NULL)
 245                 free(phone_num);
 246             phone_num = copy_of(optarg);
 247             break;
 248
 249         case 'U':
 250             if (phone_num2 != NULL)
 251                 free(phone_num2);
 252             phone_num2 = copy_of(optarg);
 253             break;
 254
 255         case 'v':
 256             ++verbose;
 257             break;
 258
 259         case 'V':
 260             ++Verbose;
 261             break;
 262
 263         default:
 264             usage();
 265             break;
 266         }
 267     }
 268
 269     argc -= optind;
 270     argv += optind;
 271
 272 /*
 273  * Default the report file to the stderr location
 274  */
 275     if (report_fp == NULL)
 276         report_fp = stderr;
 277
 278     if (to_log) {
 279         openlog("chat", LOG_PID | LOG_NDELAY, LOG_LOCAL2);
 280
 281         if (verbose)
 282             setlogmask(LOG_UPTO(LOG_INFO));
 283         else
 284             setlogmask(LOG_UPTO(LOG_WARNING));
 285     }
 286
 287     if (chat_file != NULL) {
 288         if (*argv != NULL)
 289             usage();
 290         else {
 291             init();
 292             do_file(chat_file);
 293         }
 294     } else {
 295         init();
 296         while (*argv != NULL && argc > 0) {
 297             chat_expect(*argv);
 298             argv++;
 299             argc--;
 300
 301             if (*argv != NULL && argc > 0) {
 302                 chat_send(*argv);
 303                 argv++;
 304                 argc--;
 305             }
 306         }
 307     }
 308
 309     terminate(0);
 310     return 0;
 311 }
 312
 313 /*
 314  *  Process a chat script when read from a file.
 315  */
 316
 317 static void
 318 do_file(char *chatfile)
 319 {
 320     int linect, sendflg;
 321     char *sp, *arg, quote;
 322     char buf [STR_LEN];
 323     FILE *cfp;
 324
 325     cfp = fopen (chatfile, "r");
 326     if (cfp == NULL)
 327         fatal(1, "%s -- open failed: %m", chatfile);
 328
 329     linect = 0;
 330     sendflg = 0;
 331
 332     while (fgets(buf, STR_LEN, cfp) != NULL) {
 333         sp = strchr (buf, '\n');
 334         if (sp)
 335             *sp = '\0';
 336
 337         linect++;
 338         sp = buf;
 339
 340         /* lines starting with '#' are comments. If a real '#'
 341            is to be expected, it should be quoted .... */
 342         if ( *sp == '#' )
 343             continue;
 344
 345         while (*sp != '\0') {
 346             if (*sp == ' ' || *sp == '\t') {
 347                 ++sp;
 348                 continue;
 349             }
 350
 351             if (*sp == '"' || *sp == '\'') {
 352                 quote = *sp++;
 353                 arg = sp;
 354                 while (*sp != quote) {
 355                     if (*sp == '\0')
 356                         fatal(1, "unterminated quote (line %d)", linect);
 357
 358                     if (*sp++ == '\\') {
 359                         if (*sp != '\0')
 360                             ++sp;
 361                     }
 362                 }
 363             }
 364             else {
 365                 arg = sp;
 366                 while (*sp != '\0' && *sp != ' ' && *sp != '\t')
 367                     ++sp;
 368             }
 369
 370             if (*sp != '\0')
 371                 *sp++ = '\0';
 372
 373             if (sendflg)
 374                 chat_send (arg);
 375             else
 376                 chat_expect (arg);
 377             sendflg = !sendflg;
 378         }
 379     }
 380     fclose (cfp);
 381 }
 382
 383 /*
 384  *      We got an error parsing the command line.
 385  */
 386 static void
 387 usage(void)
 388 {
 389     fprintf(stderr,
 390       "Usage: chat [-esSvV] [-f chat-file] [-r report-file] [-t timeout]\n"
 391       "            [-T phone-number] [-U phone-number2] [chat-script]\n"
 392       "where chat-script has the form:\n"
 393       "            [...[[expect[-send[-expect...]] send expect[-send[-expect]] ...]]]\n");
 394     exit(1);
 395 }
 396
 397 static char line[1024];
 398
 399 /*
 400  * Send a message to syslog and/or stderr.
 401  */
 402 static void
 403 chat_logf(const char *fmt, ...)
 404 {
 405     va_list args;
 406
 407     va_start(args, fmt);
 408     vfmtmsg(line, sizeof(line), fmt, args);
 409     if (to_log)
 410         syslog(LOG_INFO, "%s", line);
 411     if (to_stderr)
 412         fprintf(stderr, "%s\n", line);
 413 }
 414
 415 /*
 416  *      Print an error message and terminate.
 417  */
 418
 419 static void
 420 fatal(int code, const char *fmt, ...)
 421 {
 422     va_list args;
 423
 424     va_start(args, fmt);
 425     vfmtmsg(line, sizeof(line), fmt, args);
 426     if (to_log)
 427         syslog(LOG_ERR, "%s", line);
 428     if (to_stderr)
 429         fprintf(stderr, "%s\n", line);
 430     terminate(code);
 431 }
 432
 433 static int alarmed = 0;
 434
 435 static SIGTYPE sigalrm(int signo __unused)
 436 {
 437     int flags;
 438
 439     alarm(1);
 440     alarmed = 1;                /* Reset alarm to avoid race window */
 441     signal(SIGALRM, sigalrm);   /* that can cause hanging in read() */
 442
 443     if ((flags = fcntl(0, F_GETFL, 0)) == -1)
 444         fatal(2, "Can't get file mode flags on stdin: %m");
 445
 446     if (fcntl(0, F_SETFL, flags | O_NONBLOCK) == -1)
 447         fatal(2, "Can't set file mode flags on stdin: %m");
 448
 449     if (verbose)
 450         chat_logf("alarm");
 451 }
 452
 453 static SIGTYPE sigint(int signo __unused)
 454 {
 455     fatal(2, "SIGINT");
 456 }
 457
 458 static SIGTYPE sigterm(int signo __unused)
 459 {
 460     fatal(2, "SIGTERM");
 461 }
 462
 463 static SIGTYPE sighup(int signo __unused)
 464 {
 465     fatal(2, "SIGHUP");
 466 }
 467
 468 static void init(void)
 469 {
 470     signal(SIGINT, sigint);
 471     signal(SIGTERM, sigterm);
 472     signal(SIGHUP, sighup);
 473
 474     set_tty_parameters();
 475     signal(SIGALRM, sigalrm);
 476     alarm(0);
 477     alarmed = 0;
 478 }
 479
 480 static void set_tty_parameters(void)
 481 {
 482 #if defined(get_term_param)
 483     term_parms t;
 484
 485     if (get_term_param (&t) < 0)
 486         fatal(2, "Can't get terminal parameters: %m");
 487
 488     saved_tty_parameters = t;
 489     have_tty_parameters  = 1;
 490
 491     t.c_iflag     |= IGNBRK | ISTRIP | IGNPAR;
 492     t.c_oflag      = 0;
 493     t.c_lflag      = 0;
 494     t.c_cc[VERASE] =
 495     t.c_cc[VKILL]  = 0;
 496     t.c_cc[VMIN]   = 1;
 497     t.c_cc[VTIME]  = 0;
 498
 499     if (set_term_param (&t) < 0)
 500         fatal(2, "Can't set terminal parameters: %m");
 501 #endif
 502 }
 503
 504 static void break_sequence(void)
 505 {
 506     tcsendbreak (0, 0);
 507 }
 508
 509 static void terminate(int status)
 510 {
 511     echo_stderr(-1);
 512     if (report_file != NULL && report_fp != NULL) {
 513 /*
 514  * Allow the last of the report string to be gathered before we terminate.
 515  */
 516         if (report_gathering) {
 517             int c;
 518             size_t rep_len;
 519
 520             rep_len = strlen(report_buffer);
 521             while (rep_len + 1 <= sizeof(report_buffer)) {
 522                 alarm(1);
 523                 c = get_char();
 524                 alarm(0);
 525                 if (c < 0 || iscntrl(c))
 526                     break;
 527                 report_buffer[rep_len] = c;
 528                 ++rep_len;
 529             }
 530             report_buffer[rep_len] = 0;
 531             fprintf (report_fp, "chat:  %s\n", report_buffer);
 532         }
 533         if (verbose)
 534             fprintf (report_fp, "Closing \"%s\".\n", report_file);
 535         fclose (report_fp);
 536         report_fp = NULL;
 537     }
 538
 539 #if defined(get_term_param)
 540     if (have_tty_parameters) {
 541         if (set_term_param (&saved_tty_parameters) < 0)
 542             fatal(2, "Can't restore terminal parameters: %m");
 543     }
 544 #endif
 545
 546     exit(status);
 547 }
 548
 549 /*
 550  *      'Clean up' this string.
 551  */
 552 static char *
 553 clean(char *s, int sending)
 554 {
 555     char temp[STR_LEN], cur_chr;
 556     char *s1, *phchar;
 557     int add_return = sending;
 558 #define isoctal(chr) (((chr) >= '0') && ((chr) <= '7'))
 559
 560     s1 = temp;
 561     /* Don't overflow buffer, leave room for chars we append later */
 562     while (*s && s1 - temp < (off_t)(sizeof(temp) - 2 - add_return)) {
 563         cur_chr = *s++;
 564         if (cur_chr == '^') {
 565             cur_chr = *s++;
 566             if (cur_chr == '\0') {
 567                 *s1++ = '^';
 568                 break;
 569             }
 570             cur_chr &= 0x1F;
 571             if (cur_chr != 0) {
 572                 *s1++ = cur_chr;
 573             }
 574             continue;
 575         }
 576
 577         if (cur_chr != '\\') {
 578             *s1++ = cur_chr;
 579             continue;
 580         }
 581
 582         cur_chr = *s++;
 583         if (cur_chr == '\0') {
 584             if (sending) {
 585                 *s1++ = '\\';
 586                 *s1++ = '\\';
 587             }
 588             break;
 589         }
 590
 591         switch (cur_chr) {
 592         case 'b':
 593             *s1++ = '\b';
 594             break;
 595
 596         case 'c':
 597             if (sending && *s == '\0')
 598                 add_return = 0;
 599             else
 600                 *s1++ = cur_chr;
 601             break;
 602
 603         case '\\':
 604         case 'K':
 605         case 'p':
 606         case 'd':
 607             if (sending)
 608                 *s1++ = '\\';
 609
 610             *s1++ = cur_chr;
 611             break;
 612
 613         case 'T':
 614             if (sending && phone_num) {
 615                 for ( phchar = phone_num; *phchar != '\0'; phchar++)
 616                     *s1++ = *phchar;
 617             }
 618             else {
 619                 *s1++ = '\\';
 620                 *s1++ = 'T';
 621             }
 622             break;
 623
 624         case 'U':
 625             if (sending && phone_num2) {
 626                 for ( phchar = phone_num2; *phchar != '\0'; phchar++)
 627                     *s1++ = *phchar;
 628             }
 629             else {
 630                 *s1++ = '\\';
 631                 *s1++ = 'U';
 632             }
 633             break;
 634
 635         case 'q':
 636             quiet = 1;
 637             break;
 638
 639         case 'r':
 640             *s1++ = '\r';
 641             break;
 642
 643         case 'n':
 644             *s1++ = '\n';
 645             break;
 646
 647         case 's':
 648             *s1++ = ' ';
 649             break;
 650
 651         case 't':
 652             *s1++ = '\t';
 653             break;
 654
 655         case 'N':
 656             if (sending) {
 657                 *s1++ = '\\';
 658                 *s1++ = '\0';
 659             }
 660             else
 661                 *s1++ = 'N';
 662             break;
 663
 664         default:
 665             if (isoctal (cur_chr)) {
 666                 cur_chr &= 0x07;
 667                 if (isoctal (*s)) {
 668                     cur_chr <<= 3;
 669                     cur_chr |= *s++ - '0';
 670                     if (isoctal (*s)) {
 671                         cur_chr <<= 3;
 672                         cur_chr |= *s++ - '0';
 673                     }
 674                 }
 675
 676                 if (cur_chr != 0 || sending) {
 677                     if (sending && (cur_chr == '\\' || cur_chr == 0))
 678                         *s1++ = '\\';
 679                     *s1++ = cur_chr;
 680                 }
 681                 break;
 682             }
 683
 684             if (sending)
 685                 *s1++ = '\\';
 686             *s1++ = cur_chr;
 687             break;
 688         }
 689     }
 690
 691     if (add_return)
 692         *s1++ = '\r';
 693
 694     *s1++ = '\0'; /* guarantee closure */
 695     *s1++ = '\0'; /* terminate the string */
 696     return dup_mem (temp, (size_t) (s1 - temp)); /* may have embedded nuls */
 697 }
 698
 699 /*
 700  * A modified version of 'strtok'. This version skips \ sequences.
 701  */
 702
 703 static char *
 704 expect_strtok (char *s, const char *term)
 705 {
 706     static  char *str   = blank;
 707     int     escape_flag = 0;
 708     char   *result;
 709
 710 /*
 711  * If a string was specified then do initial processing.
 712  */
 713     if (s)
 714         str = s;
 715
 716 /*
 717  * If this is the escape flag then reset it and ignore the character.
 718  */
 719     if (*str)
 720         result = str;
 721     else
 722         result = NULL;
 723
 724     while (*str) {
 725         if (escape_flag) {
 726             escape_flag = 0;
 727             ++str;
 728             continue;
 729         }
 730
 731         if (*str == '\\') {
 732             ++str;
 733             escape_flag = 1;
 734             continue;
 735         }
 736
 737 /*
 738  * If this is not in the termination string, continue.
 739  */
 740         if (strchr (term, *str) == NULL) {
 741             ++str;
 742             continue;
 743         }
 744
 745 /*
 746  * This is the terminator. Mark the end of the string and stop.
 747  */
 748         *str++ = '\0';
 749         break;
 750     }
 751     return (result);
 752 }
 753
 754 /*
 755  * Process the expect string
 756  */
 757
 758 static void
 759 chat_expect(char *s)
 760 {
 761     char *expect;
 762     char *reply;
 763
 764     if (strcmp(s, "HANGUP") == 0) {
 765         ++hup_next;
 766         return;
 767     }
 768
 769     if (strcmp(s, "ABORT") == 0) {
 770         ++abort_next;
 771         return;
 772     }
 773
 774     if (strcmp(s, "CLR_ABORT") == 0) {
 775         ++clear_abort_next;
 776         return;
 777     }
 778
 779     if (strcmp(s, "REPORT") == 0) {
 780         ++report_next;
 781         return;
 782     }
 783
 784     if (strcmp(s, "CLR_REPORT") == 0) {
 785         ++clear_report_next;
 786         return;
 787     }
 788
 789     if (strcmp(s, "TIMEOUT") == 0) {
 790         ++timeout_next;
 791         return;
 792     }
 793
 794     if (strcmp(s, "ECHO") == 0) {
 795         ++echo_next;
 796         return;
 797     }
 798
 799     if (strcmp(s, "SAY") == 0) {
 800         ++say_next;
 801         return;
 802     }
 803
 804 /*
 805  * Fetch the expect and reply string.
 806  */
 807     for (;;) {
 808         expect = expect_strtok (s, "-");
 809         s      = NULL;
 810
 811         if (expect == NULL)
 812             return;
 813
 814         reply = expect_strtok (s, "-");
 815
 816 /*
 817  * Handle the expect string. If successful then exit.
 818  */
 819         if (get_string (expect))
 820             return;
 821
 822 /*
 823  * If there is a sub-reply string then send it. Otherwise any condition
 824  * is terminal.
 825  */
 826         if (reply == NULL || exit_code != 3)
 827             break;
 828
 829         chat_send (reply);
 830     }
 831
 832 /*
 833  * The expectation did not occur. This is terminal.
 834  */
 835     if (fail_reason)
 836         chat_logf("Failed (%s)", fail_reason);
 837     else
 838         chat_logf("Failed");
 839     terminate(exit_code);
 840 }
 841
 842 /*
 843  * Translate the input character to the appropriate string for printing
 844  * the data.
 845  */
 846
 847 static char *
 848 character(int c)
 849 {
 850     static char string[10];
 851     const char *meta;
 852
 853     meta = (c & 0x80) ? "M-" : "";
 854     c &= 0x7F;
 855
 856     if (c < 32)
 857         sprintf(string, "%s^%c", meta, (int)c + '@');
 858     else if (c == 127)
 859         sprintf(string, "%s^?", meta);
 860     else
 861         sprintf(string, "%s%c", meta, c);
 862
 863     return (string);
 864 }
 865
 866 /*
 867  *  process the reply string
 868  */
 869 static void
 870 chat_send(char *s)
 871 {
 872     if (say_next) {
 873         say_next = 0;
 874         s = clean(s,0);
 875         write(STDERR_FILENO, s, strlen(s));
 876         free(s);
 877         return;
 878     }
 879
 880     if (hup_next) {
 881         hup_next = 0;
 882         if (strcmp(s, "OFF") == 0)
 883            signal(SIGHUP, SIG_IGN);
 884         else
 885            signal(SIGHUP, sighup);
 886         return;
 887     }
 888
 889     if (echo_next) {
 890         echo_next = 0;
 891         echo = (strcmp(s, "ON") == 0);
 892         return;
 893     }
 894
 895     if (abort_next) {
 896         char *s1;
 897
 898         abort_next = 0;
 899
 900         if (n_aborts >= MAX_ABORTS)
 901             fatal(2, "Too many ABORT strings");
 902
 903         s1 = clean(s, 0);
 904
 905         if (strlen(s1) > strlen(s)
 906             || strlen(s1) + 1 > sizeof(fail_buffer))
 907             fatal(1, "Illegal or too-long ABORT string ('%v')", s);
 908
 909         abort_string[n_aborts++] = s1;
 910
 911         if (verbose)
 912             chat_logf("abort on (%v)", s);
 913         return;
 914     }
 915
 916     if (clear_abort_next) {
 917         char *s1;
 918         int   i;
 919         int   old_max;
 920         int   pack = 0;
 921
 922         clear_abort_next = 0;
 923
 924         s1 = clean(s, 0);
 925
 926         if (strlen(s1) > strlen(s)
 927             || strlen(s1) + 1 > sizeof(fail_buffer))
 928             fatal(1, "Illegal or too-long CLR_ABORT string ('%v')", s);
 929
 930         old_max = n_aborts;
 931         for (i=0; i < n_aborts; i++) {
 932             if ( strcmp(s1,abort_string[i]) == 0 ) {
 933                 free(abort_string[i]);
 934                 abort_string[i] = NULL;
 935                 pack++;
 936                 n_aborts--;
 937                 if (verbose)
 938                     chat_logf("clear abort on (%v)", s);
 939             }
 940         }
 941         free(s1);
 942         if (pack)
 943             pack_array(abort_string,old_max);
 944         return;
 945     }
 946
 947     if (report_next) {
 948         char *s1;
 949
 950         report_next = 0;
 951         if (n_reports >= MAX_REPORTS)
 952             fatal(2, "Too many REPORT strings");
 953
 954         s1 = clean(s, 0);
 955
 956         if (strlen(s1) > strlen(s) || strlen(s1) > sizeof fail_buffer - 1)
 957             fatal(1, "Illegal or too-long REPORT string ('%v')", s);
 958
 959         report_string[n_reports++] = s1;
 960
 961         if (verbose)
 962             chat_logf("report (%v)", s);
 963         return;
 964     }
 965
 966     if (clear_report_next) {
 967         char *s1;
 968         int   i;
 969         int   old_max;
 970         int   pack = 0;
 971
 972         clear_report_next = 0;
 973
 974         s1 = clean(s, 0);
 975
 976         if (strlen(s1) > strlen(s) || strlen(s1) > sizeof fail_buffer - 1)
 977             fatal(1, "Illegal or too-long REPORT string ('%v')", s);
 978
 979         old_max = n_reports;
 980         for (i=0; i < n_reports; i++) {
 981             if ( strcmp(s1,report_string[i]) == 0 ) {
 982                 free(report_string[i]);
 983                 report_string[i] = NULL;
 984                 pack++;
 985                 n_reports--;
 986                 if (verbose)
 987                     chat_logf("clear report (%v)", s);
 988             }
 989         }
 990         free(s1);
 991         if (pack)
 992             pack_array(report_string,old_max);
 993
 994         return;
 995     }
 996
 997     if (timeout_next) {
 998         timeout_next = 0;
 999         timeout = atoi(s);
1000
1001         if (timeout <= 0)
1002             timeout = DEFAULT_CHAT_TIMEOUT;
1003
1004         if (verbose)
1005             chat_logf("timeout set to %d seconds", timeout);
1006
1007         return;
1008     }
1009
1010     if (strcmp(s, "EOT") == 0)
1011         s = strdup("^D\\c");
1012     else if (strcmp(s, "BREAK") == 0)
1013         s = strdup("\\K\\c");
1014
1015     if (!put_string(s))
1016         fatal(1, "Failed");
1017 }
1018
1019 static int
1020 get_char(void)
1021 {
1022     int status;
1023     char c;
1024
1025     status = read(STDIN_FILENO, &c, 1);
1026
1027     switch (status) {
1028     case 1:
1029         return ((int)c & 0x7F);
1030
1031     default:
1032         chat_logf("warning: read() on stdin returned %d", status);
1033         /* FALLTHROUGH */
1034
1035     case -1:
1036         if ((status = fcntl(0, F_GETFL, 0)) == -1)
1037             fatal(2, "Can't get file mode flags on stdin: %m");
1038
1039         if (fcntl(0, F_SETFL, status & ~O_NONBLOCK) == -1)
1040             fatal(2, "Can't set file mode flags on stdin: %m");
1041
1042         return (-1);
1043     }
1044 }
1045
1046 static int put_char(int c)
1047 {
1048     int status;
1049     char ch = c;
1050
1051     usleep(10000);              /* inter-character typing delay (?) */
1052
1053     status = write(STDOUT_FILENO, &ch, 1);
1054
1055     switch (status) {
1056     case 1:
1057         return (0);
1058
1059     default:
1060         chat_logf("warning: write() on stdout returned %d", status);
1061         /* FALLTHROUGH */
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 static 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 static 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 static 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 static 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 static 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 static 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 = NULL;
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 }