blob: 5335872e5a7ca50b4e3fa5b410fbffb333d459b9 [file] [log] [blame]
Erik Andersene49d5ec2000-02-08 19:58:47 +00001/* vi: set sw=4 ts=4: */
Erik Andersen7ab9c7e2000-05-12 19:41:47 +00002/* nc: mini-netcat - built from the ground up for LRP
Eric Andersencc8ed391999-10-05 16:24:54 +00003 Copyright (C) 1998 Charles P. Wright
4
5 0.0.1 6K It works.
6 0.0.2 5K Smaller and you can also check the exit condition if you wish.
Eric Andersen2ce1edc1999-10-12 15:42:48 +00007 0.0.3 Uses select()
Eric Andersencc8ed391999-10-05 16:24:54 +00008
9 19980918 Busy Boxed! Dave Cinege
Eric Andersen2ce1edc1999-10-12 15:42:48 +000010 19990512 Uses Select. Charles P. Wright
11 19990513 Fixes stdin stupidity and uses buffers. Charles P. Wright
Eric Andersencc8ed391999-10-05 16:24:54 +000012
13 This program is free software; you can redistribute it and/or modify
14 it under the terms of the GNU General Public License as published by
15 the Free Software Foundation; either version 2 of the License, or
16 (at your option) any later version.
17
18 This program is distributed in the hope that it will be useful,
19 but WITHOUT ANY WARRANTY; without even the implied warranty of
20 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
21 GNU General Public License for more details.
22
23 You should have received a copy of the GNU General Public License
24 along with this program; if not, write to the Free Software
25 Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
26
27*/
Eric Andersen67059862001-01-22 22:48:42 +000028
Eric Andersencc8ed391999-10-05 16:24:54 +000029#include <stdio.h>
30#include <stdlib.h>
Eric Anderseneba8ed72001-03-09 14:36:42 +000031#include <string.h>
Eric Andersencc8ed391999-10-05 16:24:54 +000032#include <unistd.h>
33
34#include <sys/types.h>
35#include <sys/socket.h>
36#include <netinet/in.h>
37#include <arpa/inet.h>
38#include <netdb.h>
39#include <sys/time.h>
40#include <sys/ioctl.h>
Eric Andersencbe31da2001-02-20 06:14:08 +000041#include "busybox.h"
Eric Andersencc8ed391999-10-05 16:24:54 +000042
Erik Andersen7ab9c7e2000-05-12 19:41:47 +000043int nc_main(int argc, char **argv)
Eric Andersencc8ed391999-10-05 16:24:54 +000044{
Matt Kraai1d702672001-02-07 04:09:23 +000045 int do_listen = 0, lport = 0, tmpfd, opt, sfd;
Matt Kraaibfa79672000-12-15 22:34:34 +000046 char buf[BUFSIZ];
Eric Andersencc8ed391999-10-05 16:24:54 +000047
Erik Andersene49d5ec2000-02-08 19:58:47 +000048 struct sockaddr_in address;
49 struct hostent *hostinfo;
Eric Andersencc8ed391999-10-05 16:24:54 +000050
Erik Andersene49d5ec2000-02-08 19:58:47 +000051 fd_set readfds, testfds;
52
Matt Kraai1d702672001-02-07 04:09:23 +000053 while ((opt = getopt(argc, argv, "lp:")) > 0) {
54 switch (opt) {
55 case 'l':
56 do_listen++;
57 break;
58 case 'p':
59 lport = atoi(optarg);
60 break;
61 default:
Eric Andersen67991cf2001-02-14 21:23:06 +000062 show_usage();
Matt Kraai1d702672001-02-07 04:09:23 +000063 }
Erik Andersen5e1189e2000-04-15 16:34:54 +000064 }
Eric Andersenb6a44b81999-11-13 04:47:09 +000065
Matt Kraai1d702672001-02-07 04:09:23 +000066 if ((do_listen && optind != argc) || (!do_listen && optind + 2 != argc))
Eric Andersen67991cf2001-02-14 21:23:06 +000067 show_usage();
Matt Kraai1d702672001-02-07 04:09:23 +000068
Matt Kraaibfa79672000-12-15 22:34:34 +000069 if ((sfd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
70 perror_msg_and_die("socket");
Eric Andersencc8ed391999-10-05 16:24:54 +000071
Erik Andersene49d5ec2000-02-08 19:58:47 +000072 address.sin_family = AF_INET;
Eric Andersencc8ed391999-10-05 16:24:54 +000073
Matt Kraai1d702672001-02-07 04:09:23 +000074 if (lport != 0) {
75 memset(&address.sin_addr, 0, sizeof(address.sin_addr));
76 address.sin_port = htons(lport);
77
78 if (bind(sfd, (struct sockaddr *) &address, sizeof(address)) < 0)
79 perror_msg_and_die("bind");
80 }
81
82 if (do_listen) {
Matt Kraaibe9f44a2001-05-15 03:05:39 +000083 socklen_t addrlen = sizeof(address);
84
Matt Kraai1d702672001-02-07 04:09:23 +000085 if (listen(sfd, 1) < 0)
86 perror_msg_and_die("listen");
87
Matt Kraaibe9f44a2001-05-15 03:05:39 +000088 if ((tmpfd = accept(sfd, (struct sockaddr *) &address, &addrlen)) < 0)
Matt Kraai1d702672001-02-07 04:09:23 +000089 perror_msg_and_die("accept");
90
91 close(sfd);
92 sfd = tmpfd;
93 } else {
Matt Kraaic55b8d42001-05-16 15:40:51 +000094 hostinfo = xgethostbyname(argv[optind]);
Matt Kraai1d702672001-02-07 04:09:23 +000095
96 address.sin_addr = *(struct in_addr *) *hostinfo->h_addr_list;
97 address.sin_port = htons(atoi(argv[optind+1]));
98
99 if (connect(sfd, (struct sockaddr *) &address, sizeof(address)) < 0)
100 perror_msg_and_die("connect");
101 }
Eric Andersencc8ed391999-10-05 16:24:54 +0000102
Erik Andersene49d5ec2000-02-08 19:58:47 +0000103 FD_ZERO(&readfds);
104 FD_SET(sfd, &readfds);
Matt Kraaibfa79672000-12-15 22:34:34 +0000105 FD_SET(STDIN_FILENO, &readfds);
Eric Andersencc8ed391999-10-05 16:24:54 +0000106
Erik Andersene49d5ec2000-02-08 19:58:47 +0000107 while (1) {
108 int fd;
Eric Andersen2ce1edc1999-10-12 15:42:48 +0000109 int ofd;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000110 int nread;
Eric Andersencc8ed391999-10-05 16:24:54 +0000111
Erik Andersene49d5ec2000-02-08 19:58:47 +0000112 testfds = readfds;
Eric Andersencc8ed391999-10-05 16:24:54 +0000113
Matt Kraaibfa79672000-12-15 22:34:34 +0000114 if (select(FD_SETSIZE, &testfds, NULL, NULL, NULL) < 0)
115 perror_msg_and_die("select");
Eric Andersencc8ed391999-10-05 16:24:54 +0000116
Erik Andersene49d5ec2000-02-08 19:58:47 +0000117 for (fd = 0; fd < FD_SETSIZE; fd++) {
118 if (FD_ISSET(fd, &testfds)) {
Matt Kraaibfa79672000-12-15 22:34:34 +0000119 if ((nread = safe_read(fd, buf, sizeof(buf))) < 0)
120 perror_msg_and_die("read");
Eric Andersencc8ed391999-10-05 16:24:54 +0000121
Erik Andersene49d5ec2000-02-08 19:58:47 +0000122 if (fd == sfd) {
123 if (nread == 0)
124 exit(0);
Matt Kraaibfa79672000-12-15 22:34:34 +0000125 ofd = STDOUT_FILENO;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000126 } else {
Matt Kraai95fa0ea2000-12-14 04:34:58 +0000127 if (nread == 0)
128 shutdown(sfd, 1);
Eric Andersen2ce1edc1999-10-12 15:42:48 +0000129 ofd = sfd;
130 }
Eric Andersencc8ed391999-10-05 16:24:54 +0000131
Matt Kraaibfa79672000-12-15 22:34:34 +0000132 if (full_write(ofd, buf, nread) < 0)
133 perror_msg_and_die("write");
Eric Andersen2ce1edc1999-10-12 15:42:48 +0000134 }
Erik Andersene49d5ec2000-02-08 19:58:47 +0000135 }
136 }
Eric Andersencc8ed391999-10-05 16:24:54 +0000137}