pppctl.c revision 49376 
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 *	$Id: pppctl.c,v 1.18 1999/04/26 08:53:52 brian Exp $
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
50static char Buffer[LINELEN], Command[LINELEN];
51
52static int
53usage()
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
65static int TimedOut = 0;
66static void
67Timeout(int Sig)
68{
69    TimedOut = 1;
70}
71
72#define REC_PASSWD  (1)
73#define REC_SHOW    (2)
74#define REC_VERBOSE (4)
75
76static char *passwd;
77static char *prompt;
78
79static char *
80GetPrompt(EditLine *e)
81{
82    if (prompt == NULL)
83        prompt = "";
84    return prompt;
85}
86
87static int
88Receive(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
153static int data = -1;
154static jmp_buf pppdead;
155
156static void
157check_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
178static const char *
179smartgets(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
194int
195main(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#else
373                history(hist, H_EVENT, size);
374#endif
375                edit = el_init("pppctl", stdin, stdout);
376                el_source(edit, NULL);
377                el_set(edit, EL_PROMPT, GetPrompt);
378                if ((env = getenv("EL_EDITOR"))) {
379                    if (!strcmp(env, "vi"))
380                        el_set(edit, EL_EDITOR, "vi");
381                    else if (!strcmp(env, "emacs"))
382                        el_set(edit, EL_EDITOR, "emacs");
383                }
384                el_set(edit, EL_SIGNAL, 1);
385                el_set(edit, EL_HIST, history, (const char *)hist);
386                while ((l = smartgets(edit, &len, fd))) {
387                    if (len > 1)
388#ifdef __NetBSD__
389                        history(hist, NULL, H_ENTER, l);
390#else
391                        history(hist, H_ENTER, l);
392#endif
393                    write(fd, l, len);
394                    if (Receive(fd, REC_SHOW) != 0)
395                        break;
396                }
397                fprintf(stderr, "Connection closed\n");
398                el_end(edit);
399                history_end(hist);
400            } else {
401                start = Command;
402                do {
403                    next = strchr(start, ';');
404                    while (*start == ' ' || *start == '\t')
405                        start++;
406                    if (next)
407                        *next = '\0';
408                    strcpy(Buffer, start);
409                    Buffer[sizeof(Buffer)-2] = '\0';
410                    strcat(Buffer, "\n");
411                    if (verbose)
412                        write(1, Buffer, strlen(Buffer));
413                    write(fd, Buffer, strlen(Buffer));
414                    if (Receive(fd, verbose | REC_SHOW) != 0) {
415                        fprintf(stderr, "Connection closed\n");
416                        break;
417                    }
418                    if (next)
419                        start = ++next;
420                } while (next && *next);
421                if (verbose)
422                    puts("");
423            }
424            break;
425
426        default:
427            warnx("ppp is not responding");
428            break;
429    }
430
431    close(fd);
432
433    return 0;
434}
435