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 * --- 46 unchanged lines hidden (view full) --- 55void restore_term(); 56 57char progname[256]; 58void usage(int); 59 60int 61main(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 == 2) { 78 host = "localhost"; 79 port = argv[1]; 80 } else if (argc == 3) { |
81 host = argv[1]; 82 port = argv[2]; |
83 } else { 84 usage(1); 85 } |
86 |
87 if (!isatty(STDIN_FILENO)) 88 errx(1, "not attached to a terminal"); |
89 |
90 raw_term(); |
91 |
92 /* Initialize addrinfo structure */ 93 memset(&hints, 0, sizeof(struct addrinfo)); 94 hints.ai_family = AF_UNSPEC; 95 hints.ai_socktype = SOCK_STREAM; 96 hints.ai_protocol = IPPROTO_TCP; |
97 |
98 s = remote_connect(host, port, hints); 99 ret = 0; 100 readwrite(s); |
101 |
102 if (s) 103 close(s); |
104 |
105 exit(ret); |
106} 107 108/* 109 * remote_connect() 110 * Return's a socket connected to a remote host. Properly bind's to a local 111 * port or source address if needed. Return's -1 on failure. 112 */ 113int 114remote_connect(char *host, char *port, struct addrinfo hints) 115{ |
116 struct addrinfo *res, *res0; 117 int s, error; |
118 |
119 if ((error = getaddrinfo(host, port, &hints, &res))) 120 errx(1, "getaddrinfo: %s", gai_strerror(error)); |
121 |
122 res0 = res; 123 do { 124 if ((s = socket(res0->ai_family, res0->ai_socktype, 125 res0->ai_protocol)) < 0) 126 continue; |
127 |
128 if (connect(s, res0->ai_addr, res0->ai_addrlen) == 0) 129 break; |
130 |
131 close(s); 132 s = -1; 133 } while ((res0 = res0->ai_next) != NULL); |
134 |
135 freeaddrinfo(res); |
136 |
137 return (s); |
138} 139 140/* 141 * readwrite() 142 * Loop that polls on the network file descriptor and stdin. 143 */ 144void 145readwrite(int nfd) 146{ |
147 struct pollfd pfd[2]; 148 char buf[BUFSIZ]; 149 int wfd = fileno(stdin), n, ret; 150 int lfd = fileno(stdout); 151 int escape = 0; |
152 |
153 /* Setup Network FD */ 154 pfd[0].fd = nfd; 155 pfd[0].events = POLLIN; |
156 |
157 /* Setup STDIN FD */ 158 pfd[1].fd = wfd; 159 pfd[1].events = POLLIN; |
160 |
161 while (pfd[0].fd != -1) { 162 if ((n = poll(pfd, 2, -1)) < 0) { 163 close(nfd); 164 err(1, "Polling Error"); 165 } |
166 |
167 if (n == 0) 168 return; |
169 |
170 if (pfd[0].revents & POLLIN) { 171 if ((n = read(nfd, buf, sizeof(buf))) < 0) 172 return; 173 else if (n == 0) { 174 shutdown(nfd, SHUT_RD); 175 pfd[0].fd = -1; 176 pfd[0].events = 0; 177 } else { 178 if ((ret = atomicio(write, lfd, buf, n)) != n) 179 return; 180 } 181 } |
182 |
183 if (pfd[1].revents & POLLIN) { 184 if ((n = read(wfd, buf, sizeof(buf))) < 0) 185 return; 186 else if (n == 0) { 187 shutdown(nfd, SHUT_WR); 188 pfd[1].fd = -1; 189 pfd[1].events = 0; 190 } else { 191 if (escape) { 192 char buf2[] = "~"; 193 if (*buf == '.') { 194 printf("quit!\n"); 195 return; 196 } 197 escape = 0; 198 if (*buf != '~' && 199 (ret = atomicio(write, nfd, buf2, 1)) != n) 200 return; 201 } else { 202 escape = (*buf == '~'); 203 if (escape) 204 continue; |
205 } |
206 207 if ((ret = atomicio(write, nfd, buf, n)) != n) 208 return; 209 } |
210 } |
211 } |
212} 213 214void 215usage(int ret) 216{ |
217 fprintf(stderr, "usage: %s hostname port\n", progname); 218 if (ret) 219 exit(1); |
220} 221 222/* 223 * Copyright (c) 1995,1999 Theo de Raadt. All rights reserved. 224 * All rights reserved. 225 * 226 * Redistribution and use in source and binary forms, with or without 227 * modification, are permitted provided that the following conditions --- 17 unchanged lines hidden (view full) --- 245 */ 246 247/* 248 * ensure all of data on socket comes through. f==read || f==write 249 */ 250ssize_t 251atomicio(ssize_t (*f) (), int fd, void *_s, size_t n) 252{ |
253 char *s = _s; 254 ssize_t res, pos = 0; |
255 |
256 while (n > pos) { 257 res = (f) (fd, s + pos, n - pos); 258 switch (res) { 259 case -1: 260 if (errno == EINTR || errno == EAGAIN) 261 continue; 262 case 0: 263 return (res); 264 default: 265 pos += res; |
266 } |
267 } 268 return (pos); |
269} 270 271/* 272 * Copyright (c) 2003 Nathan L. Binkert <binkertn@umich.edu> 273 * 274 * Permission to use, copy, modify, and distribute this software for any 275 * purpose with or without fee is hereby granted, provided that the above 276 * copyright notice and this permission notice appear in all copies. --- 5 unchanged lines hidden (view full) --- 282 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN 283 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF 284 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. 285 */ 286 287void 288raw_term() 289{ |
290 struct termios ios; |
291 |
292 if (tcgetattr(STDIN_FILENO, &ios) < 0) 293 errx(1, "tcgetagttr\n"); |
294 |
295 memcpy(&saved_ios, &ios, sizeof(struct termios)); |
296 |
297 ios.c_iflag &= ~(ISTRIP|ICRNL|IGNCR|ICRNL|IXOFF|IXON); 298 ios.c_oflag &= ~(OPOST); 299 ios.c_oflag &= (ONLCR); 300 ios.c_lflag &= ~(ISIG|ICANON|ECHO); 301 ios.c_cc[VMIN] = 1; 302 ios.c_cc[VTIME] = 0; |
303 |
304 if (tcsetattr(STDIN_FILENO, TCSANOW, &ios) < 0) 305 errx(1, "tcsetattr\n"); |
306 |
307 atexit(restore_term); |
308} 309 310void 311restore_term() 312{ |
313 tcsetattr(STDIN_FILENO, TCSANOW, &saved_ios); |
314} |