util/term/term.c

   1 /* $OpenBSD: netcat.c,v 1.57 2002/12/30 18:00:18 stevesk Exp $ */
   2 /*
   3  * Copyright (c) 2001 Eric Jackson <ericj@monkey.org>
   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  *
   9  * 1. Redistributions of source code must retain the above copyright
  10  *   notice, this list of conditions and the following disclaimer.
  11  * 2. Redistributions in binary form must reproduce the above copyright
  12  *   notice, this list of conditions and the following disclaimer in the
  13  *   documentation and/or other materials provided with the distribution.
  14  * 3. The name of the author may not be used to endorse or promote products
  15  *   derived from this software without specific prior written permission.
  16  *
  17  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  18  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  19  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  20  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  21  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  22  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  23  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  24  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  25  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  26  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  27  */
  28
  29 #include <sys/types.h>
  30 #include <sys/socket.h>
  31 #include <sys/termios.h>
  32 #include <sys/time.h>
  33 #include <sys/un.h>
  34
  35 #include <netinet/in.h>
  36 #include <arpa/telnet.h>
  37
  38 #include <err.h>
  39 #include <errno.h>
  40 #include <netdb.h>
  41 #include <poll.h>
  42 #include <stdarg.h>
  43 #include <stdio.h>
  44 #include <stdlib.h>
  45 #include <string.h>
  46 #include <unistd.h>
  47 #include <fcntl.h>
  48
  49 ssize_t atomicio(ssize_t (*)(), int, void *, size_t);
  50 void    readwrite(int);
  51 int     remote_connect(char *, char *, struct addrinfo);
  52
  53 struct  termios saved_ios;
  54 void    raw_term();
  55 void    restore_term();
  56
  57 char    progname[256];
  58 void    usage(int);
  59
  60 int
  61 main(int argc, char *argv[])
  62 {
  63         int ch, s, ret;
  64         char *host, *port, *endp;
  65         struct addrinfo hints;
  66         socklen_t len;
  67
  68         ret = 1;
  69         s = 0;
  70         host = NULL;
  71         port = NULL;
  72         endp = NULL;
  73
  74         strncpy(progname, argv[0], sizeof progname);
  75
  76         /* Cruft to make sure options are clean, and used properly. */
  77         if (argc != 3 || !argv[1] || !argv[2])
  78                 usage(1);
  79
  80         host = argv[1];
  81         port = argv[2];
  82
  83
  84         if (!isatty(STDIN_FILENO))
  85                 errx(1, "not attached to a terminal");
  86
  87         raw_term();
  88
  89         /* Initialize addrinfo structure */
  90         memset(&hints, 0, sizeof(struct addrinfo));
  91         hints.ai_family = AF_UNSPEC;
  92         hints.ai_socktype = SOCK_STREAM;
  93         hints.ai_protocol = IPPROTO_TCP;
  94
  95         s = remote_connect(host, port, hints);
  96         ret = 0;
  97         readwrite(s);
  98
  99         if (s)
 100                 close(s);
 101
 102         exit(ret);
 103 }
 104
 105 /*
 106  * remote_connect()
 107  * Return's a socket connected to a remote host. Properly bind's to a local
 108  * port or source address if needed. Return's -1 on failure.
 109  */
 110 int
 111 remote_connect(char *host, char *port, struct addrinfo hints)
 112 {
 113         struct addrinfo *res, *res0;
 114         int s, error;
 115
 116         if ((error = getaddrinfo(host, port, &hints, &res)))
 117                 errx(1, "getaddrinfo: %s", gai_strerror(error));
 118
 119         res0 = res;
 120         do {
 121                 if ((s = socket(res0->ai_family, res0->ai_socktype,
 122                     res0->ai_protocol)) < 0)
 123                         continue;
 124
 125                 if (connect(s, res0->ai_addr, res0->ai_addrlen) == 0)
 126                         break;
 127
 128                 close(s);
 129                 s = -1;
 130         } while ((res0 = res0->ai_next) != NULL);
 131
 132         freeaddrinfo(res);
 133
 134         return (s);
 135 }
 136
 137 /*
 138  * readwrite()
 139  * Loop that polls on the network file descriptor and stdin.
 140  */
 141 void
 142 readwrite(int nfd)
 143 {
 144         struct pollfd pfd[2];
 145         char buf[BUFSIZ];
 146         int wfd = fileno(stdin), n, ret;
 147         int lfd = fileno(stdout);
 148         int escape = 0;
 149
 150         /* Setup Network FD */
 151         pfd[0].fd = nfd;
 152         pfd[0].events = POLLIN;
 153
 154         /* Setup STDIN FD */
 155         pfd[1].fd = wfd;
 156         pfd[1].events = POLLIN;
 157
 158         while (pfd[0].fd != -1) {
 159                 if ((n = poll(pfd, 2, -1)) < 0) {
 160                         close(nfd);
 161                         err(1, "Polling Error");
 162                 }
 163
 164                 if (n == 0)
 165                         return;
 166
 167                 if (pfd[0].revents & POLLIN) {
 168                         if ((n = read(nfd, buf, sizeof(buf))) < 0)
 169                                 return;
 170                         else if (n == 0) {
 171                                 shutdown(nfd, SHUT_RD);
 172                                 pfd[0].fd = -1;
 173                                 pfd[0].events = 0;
 174                         } else {
 175                                 if ((ret = atomicio(write, lfd, buf, n)) != n)
 176                                         return;
 177                         }
 178                 }
 179
 180                 if (pfd[1].revents & POLLIN) {
 181                         if ((n = read(wfd, buf, sizeof(buf))) < 0)
 182                                 return;
 183                         else if (n == 0) {
 184                                 shutdown(nfd, SHUT_WR);
 185                                 pfd[1].fd = -1;
 186                                 pfd[1].events = 0;
 187                         } else {
 188                                 if (escape) {
 189                                         char buf2[] = "~";
 190                                         if (*buf == '.') {
 191                                                 printf("quit!\n");
 192                                                 return;
 193                                         }
 194                                         escape = 0;
 195                                         if (*buf != '~' &&
 196                                             (ret = atomicio(write, nfd, buf2, 1)) != n)
 197                                                 return;
 198                                 } else {
 199                                         escape = (*buf == '~');
 200                                         if (escape)
 201                                                 continue;
 202                                 }
 203
 204                                 if ((ret = atomicio(write, nfd, buf, n)) != n)
 205                                         return;
 206                         }
 207                 }
 208         }
 209 }
 210
 211 void
 212 usage(int ret)
 213 {
 214         fprintf(stderr, "usage: %s hostname port\n", progname);
 215         if (ret)
 216                 exit(1);
 217 }
 218
 219 /*
 220  * Copyright (c) 1995,1999 Theo de Raadt.  All rights reserved.
 221  * All rights reserved.
 222  *
 223  * Redistribution and use in source and binary forms, with or without
 224  * modification, are permitted provided that the following conditions
 225  * are met:
 226  * 1. Redistributions of source code must retain the above copyright
 227  *    notice, this list of conditions and the following disclaimer.
 228  * 2. Redistributions in binary form must reproduce the above copyright
 229  *    notice, this list of conditions and the following disclaimer in the
 230  *    documentation and/or other materials provided with the distribution.
 231  *
 232  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 233  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 234  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 235  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 236  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 237  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 238  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 239  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 240  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 241  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 242  */
 243
 244 /*
 245  * ensure all of data on socket comes through. f==read || f==write
 246  */
 247 ssize_t
 248 atomicio(ssize_t (*f) (), int fd, void *_s, size_t n)
 249 {
 250         char *s = _s;
 251         ssize_t res, pos = 0;
 252
 253         while (n > pos) {
 254                 res = (f) (fd, s + pos, n - pos);
 255                 switch (res) {
 256                 case -1:
 257                         if (errno == EINTR || errno == EAGAIN)
 258                                 continue;
 259                 case 0:
 260                         return (res);
 261                 default:
 262                         pos += res;
 263                 }
 264         }
 265         return (pos);
 266 }
 267
 268 /*
 269  * Copyright (c) 2003 Nathan L. Binkert <binkertn@umich.edu>
 270  *
 271  * Permission to use, copy, modify, and distribute this software for any
 272  * purpose with or without fee is hereby granted, provided that the above
 273  * copyright notice and this permission notice appear in all copies.
 274  *
 275  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 276  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 277  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 278  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 279  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 280  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 281  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 282  */
 283
 284 void
 285 raw_term()
 286 {
 287         struct termios ios;
 288
 289         if (tcgetattr(STDIN_FILENO, &ios) < 0)
 290             errx(1, "tcgetagttr\n");
 291
 292         memcpy(&saved_ios, &ios, sizeof(struct termios));
 293
 294         ios.c_iflag &= ~(ISTRIP|ICRNL|IGNCR|ICRNL|IXOFF|IXON);
 295         ios.c_oflag &= ~(OPOST);
 296         ios.c_oflag &= (ONLCR);
 297         ios.c_lflag &= ~(ISIG|ICANON|ECHO);
 298         ios.c_cc[VMIN] = 1;
 299         ios.c_cc[VTIME] = 0;
 300
 301         if (tcsetattr(STDIN_FILENO, TCSANOW, &ios) < 0)
 302             errx(1, "tcsetattr\n");
 303
 304         atexit(restore_term);
 305 }
 306
 307 void
 308 restore_term()
 309 {
 310         tcsetattr(STDIN_FILENO, TCSANOW, &saved_ios);
 311 }