usr.sbin/pppctl/pppctl.c

   1 /*-
   2  * Copyright (c) 1997 Brian Somers <brian@Awfulhak.org>
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  *
  14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  24  * SUCH DAMAGE.
  25  *
  26  * $FreeBSD$
  27  */
  28
  29 #include <sys/types.h>
  30
  31 #include <sys/socket.h>
  32 #include <netinet/in.h>
  33 #include <arpa/inet.h>
  34 #include <sys/un.h>
  35 #include <netdb.h>
  36
  37 #include <sys/time.h>
  38 #include <err.h>
  39 #include <errno.h>
  40 #include <histedit.h>
  41 #include <setjmp.h>
  42 #include <signal.h>
  43 #include <stdio.h>
  44 #include <stdlib.h>
  45 #include <string.h>
  46 #include <time.h>
  47 #include <unistd.h>
  48
  49 #define LINELEN 2048
  50 static char Buffer[LINELEN], Command[LINELEN];
  51
  52 static int
  53 usage()
  54 {
  55     fprintf(stderr, "usage: pppctl [-v] [-t n] [-p passwd] "
  56             "Port|LocalSock [command[;command]...]\n");
  57     fprintf(stderr, "              -v tells pppctl to output all"
  58             " conversation\n");
  59     fprintf(stderr, "              -t n specifies a timeout of n"
  60             " seconds when connecting (default 2)\n");
  61     fprintf(stderr, "              -p passwd specifies your password\n");
  62     exit(1);
  63 }
  64
  65 static int TimedOut = 0;
  66 static void
  67 Timeout(int Sig)
  68 {
  69     TimedOut = 1;
  70 }
  71
  72 #define REC_PASSWD  (1)
  73 #define REC_SHOW    (2)
  74 #define REC_VERBOSE (4)
  75
  76 static char *passwd;
  77 static char *prompt;
  78
  79 static char *
  80 GetPrompt(EditLine *e)
  81 {
  82     if (prompt == NULL)
  83         prompt = "";
  84     return prompt;
  85 }
  86
  87 static int
  88 Receive(int fd, int display)
  89 {
  90     int Result;
  91     int len;
  92     char *last;
  93
  94     prompt = Buffer;
  95     len = 0;
  96     while (Result = read(fd, Buffer+len, sizeof(Buffer)-len-1), Result != -1) {
  97         if (Result == 0 && errno != EINTR) {
  98           Result = -1;
  99           break;
 100         }
 101         len += Result;
 102         Buffer[len] = '\0';
 103         if (len > 2 && !strcmp(Buffer+len-2, "> ")) {
 104             prompt = strrchr(Buffer, '\n');
 105             if (display & (REC_SHOW|REC_VERBOSE)) {
 106                 if (display & REC_VERBOSE)
 107                     last = Buffer+len-1;
 108                 else
 109                     last = prompt;
 110                 if (last) {
 111                     last++;
 112                     write(1, Buffer, last-Buffer);
 113                 }
 114             }
 115             prompt = prompt == NULL ? Buffer : prompt+1;
 116             for (last = Buffer+len-2; last > Buffer && *last != ' '; last--)
 117                 ;
 118             if (last > Buffer+3 && !strncmp(last-3, " on", 3)) {
 119                  /* a password is required ! */
 120                  if (display & REC_PASSWD) {
 121                     /* password time */
 122                     if (!passwd)
 123                         passwd = getpass("Password: ");
 124                     sprintf(Buffer, "passwd %s\n", passwd);
 125                     memset(passwd, '\0', strlen(passwd));
 126                     if (display & REC_VERBOSE)
 127                         write(1, Buffer, strlen(Buffer));
 128                     write(fd, Buffer, strlen(Buffer));
 129                     memset(Buffer, '\0', strlen(Buffer));
 130                     return Receive(fd, display & ~REC_PASSWD);
 131                 }
 132                 Result = 1;
 133             } else
 134                 Result = 0;
 135             break;
 136         }
 137         if (len == sizeof Buffer - 1) {
 138             int flush;
 139             if ((last = strrchr(Buffer, '\n')) == NULL)
 140                 /* Yeuch - this is one mother of a line ! */
 141                 flush = sizeof Buffer / 2;
 142             else
 143                 flush = last - Buffer + 1;
 144             write(1, Buffer, flush);
 145             strcpy(Buffer, Buffer + flush);
 146             len -= flush;
 147         }
 148     }
 149
 150     return Result;
 151 }
 152
 153 static int data = -1;
 154 static jmp_buf pppdead;
 155
 156 static void
 157 check_fd(int sig)
 158 {
 159   if (data != -1) {
 160     struct timeval t;
 161     fd_set f;
 162     static char buf[LINELEN];
 163     int len;
 164
 165     FD_ZERO(&f);
 166     FD_SET(data, &f);
 167     t.tv_sec = t.tv_usec = 0;
 168     if (select(data+1, &f, NULL, NULL, &t) > 0) {
 169       len = read(data, buf, sizeof buf);
 170       if (len > 0)
 171         write(1, buf, len);
 172       else
 173         longjmp(pppdead, -1);
 174     }
 175   }
 176 }
 177
 178 static const char *
 179 smartgets(EditLine *e, int *count, int fd)
 180 {
 181   const char *result;
 182
 183   data = fd;
 184   signal(SIGALRM, check_fd);
 185   ualarm(500000, 500000);
 186   result = setjmp(pppdead) ? NULL : el_gets(e, count);
 187   ualarm(0,0);
 188   signal(SIGALRM, SIG_DFL);
 189   data = -1;
 190
 191   return result;
 192 }
 193
 194 int
 195 main(int argc, char **argv)
 196 {
 197     struct servent *s;
 198     struct hostent *h;
 199     struct sockaddr *sock;
 200     struct sockaddr_in ifsin;
 201     struct sockaddr_un ifsun;
 202     int socksz, arg, fd, len, verbose, save_errno;
 203     unsigned TimeoutVal;
 204     char *DoneWord = "x", *next, *start;
 205     struct sigaction act, oact;
 206
 207     verbose = 0;
 208     TimeoutVal = 2;
 209
 210     for (arg = 1; arg < argc; arg++)
 211         if (*argv[arg] == '-') {
 212             for (start = argv[arg] + 1; *start; start++)
 213                 switch (*start) {
 214                     case 't':
 215                         TimeoutVal = (unsigned)atoi
 216                             (start[1] ? start + 1 : argv[++arg]);
 217                         start = DoneWord;
 218                         break;
 219
 220                     case 'v':
 221                         verbose = REC_VERBOSE;
 222                         break;
 223
 224                     case 'p':
 225                         passwd = (start[1] ? start + 1 : argv[++arg]);
 226                         start = DoneWord;
 227                         break;
 228
 229                     default:
 230                         usage();
 231                 }
 232         }
 233         else
 234             break;
 235
 236
 237     if (argc < arg + 1)
 238         usage();
 239
 240     if (*argv[arg] == '/') {
 241         sock = (struct sockaddr *)&ifsun;
 242         socksz = sizeof ifsun;
 243
 244         memset(&ifsun, '\0', sizeof ifsun);
 245         ifsun.sun_len = strlen(argv[arg]);
 246         if (ifsun.sun_len > sizeof ifsun.sun_path - 1) {
 247             warnx("%s: path too long", argv[arg]);
 248             return 1;
 249         }
 250         ifsun.sun_family = AF_LOCAL;
 251         strcpy(ifsun.sun_path, argv[arg]);
 252
 253         if (fd = socket(AF_LOCAL, SOCK_STREAM, 0), fd < 0) {
 254             warnx("cannot create local domain socket");
 255             return 2;
 256         }
 257     } else {
 258         char *port, *host, *colon;
 259         int hlen;
 260
 261         colon = strchr(argv[arg], ':');
 262         if (colon) {
 263             port = colon + 1;
 264             *colon = '\0';
 265             host = argv[arg];
 266         } else {
 267             port = argv[arg];
 268             host = "127.0.0.1";
 269         }
 270         sock = (struct sockaddr *)&ifsin;
 271         socksz = sizeof ifsin;
 272         hlen = strlen(host);
 273
 274         memset(&ifsin, '\0', sizeof ifsin);
 275         if (strspn(host, "0123456789.") == hlen) {
 276             if (!inet_aton(host, &ifsin.sin_addr)) {
 277                 warnx("cannot translate %s", host);
 278                 return 1;
 279             }
 280         } else if ((h = gethostbyname(host)) == 0) {
 281             warnx("cannot resolve %s", host);
 282             return 1;
 283         }
 284         else
 285             ifsin.sin_addr.s_addr = *(u_long *)h->h_addr_list[0];
 286
 287         if (colon)
 288             *colon = ':';
 289
 290         if (strspn(port, "0123456789") == strlen(port))
 291             ifsin.sin_port = htons(atoi(port));
 292         else if (s = getservbyname(port, "tcp"), !s) {
 293             warnx("%s isn't a valid port or service!", port);
 294             usage();
 295         }
 296         else
 297             ifsin.sin_port = s->s_port;
 298
 299         ifsin.sin_len = sizeof(ifsin);
 300         ifsin.sin_family = AF_INET;
 301
 302         if (fd = socket(AF_INET, SOCK_STREAM, 0), fd < 0) {
 303             warnx("cannot create internet socket");
 304             return 2;
 305         }
 306     }
 307
 308     TimedOut = 0;
 309     if (TimeoutVal) {
 310         act.sa_handler = Timeout;
 311         sigemptyset(&act.sa_mask);
 312         act.sa_flags = 0;
 313         sigaction(SIGALRM, &act, &oact);
 314         alarm(TimeoutVal);
 315     }
 316
 317     if (connect(fd, sock, socksz) < 0) {
 318         if (TimeoutVal) {
 319             save_errno = errno;
 320             alarm(0);
 321             sigaction(SIGALRM, &oact, 0);
 322             errno = save_errno;
 323         }
 324         if (TimedOut)
 325             warnx("timeout: cannot connect to socket %s", argv[arg]);
 326         else {
 327             if (errno)
 328                 warn("cannot connect to socket %s", argv[arg]);
 329             else
 330                 warnx("cannot connect to socket %s", argv[arg]);
 331         }
 332         close(fd);
 333         return 3;
 334     }
 335
 336     if (TimeoutVal) {
 337         alarm(0);
 338         sigaction(SIGALRM, &oact, 0);
 339     }
 340
 341     len = 0;
 342     Command[sizeof(Command)-1] = '\0';
 343     for (arg++; arg < argc; arg++) {
 344         if (len && len < sizeof(Command)-1)
 345             strcpy(Command+len++, " ");
 346         strncpy(Command+len, argv[arg], sizeof(Command)-len-1);
 347         len += strlen(Command+len);
 348     }
 349
 350     switch (Receive(fd, verbose | REC_PASSWD))
 351     {
 352         case 1:
 353             fprintf(stderr, "Password incorrect\n");
 354             break;
 355
 356         case 0:
 357             if (len == 0) {
 358                 EditLine *edit;
 359                 History *hist;
 360                 const char *l, *env;
 361                 int size;
 362
 363                 hist = history_init();
 364                 if ((env = getenv("EL_SIZE"))) {
 365                     size = atoi(env);
 366                     if (size < 0)
 367                       size = 20;
 368                 } else
 369                     size = 20;
 370 #ifdef __NetBSD__
 371                 history(hist, NULL, H_SETSIZE, size);
 372                 edit = el_init("pppctl", stdin, stdout, stderr);
 373 #else
 374                 history(hist, H_EVENT, size);
 375                 edit = el_init("pppctl", stdin, stdout);
 376 #endif
 377                 el_source(edit, NULL);
 378                 el_set(edit, EL_PROMPT, GetPrompt);
 379                 if ((env = getenv("EL_EDITOR"))) {
 380                     if (!strcmp(env, "vi"))
 381                         el_set(edit, EL_EDITOR, "vi");
 382                     else if (!strcmp(env, "emacs"))
 383                         el_set(edit, EL_EDITOR, "emacs");
 384                 }
 385                 el_set(edit, EL_SIGNAL, 1);
 386                 el_set(edit, EL_HIST, history, (const char *)hist);
 387                 while ((l = smartgets(edit, &len, fd))) {
 388                     if (len > 1)
 389 #ifdef __NetBSD__
 390                         history(hist, NULL, H_ENTER, l);
 391 #else
 392                         history(hist, H_ENTER, l);
 393 #endif
 394                     write(fd, l, len);
 395                     if (Receive(fd, REC_SHOW) != 0)
 396                         break;
 397                 }
 398                 fprintf(stderr, "Connection closed\n");
 399                 el_end(edit);
 400                 history_end(hist);
 401             } else {
 402                 start = Command;
 403                 do {
 404                     next = strchr(start, ';');
 405                     while (*start == ' ' || *start == '\t')
 406                         start++;
 407                     if (next)
 408                         *next = '\0';
 409                     strcpy(Buffer, start);
 410                     Buffer[sizeof(Buffer)-2] = '\0';
 411                     strcat(Buffer, "\n");
 412                     if (verbose)
 413                         write(1, Buffer, strlen(Buffer));
 414                     write(fd, Buffer, strlen(Buffer));
 415                     if (Receive(fd, verbose | REC_SHOW) != 0) {
 416                         fprintf(stderr, "Connection closed\n");
 417                         break;
 418                     }
 419                     if (next)
 420                         start = ++next;
 421                 } while (next && *next);
 422                 if (verbose)
 423                     puts("");
 424             }
 425             break;
 426
 427         default:
 428             warnx("ppp is not responding");
 429             break;
 430     }
 431
 432     close(fd);
 433
 434     return 0;
 435 }