blob: 805bbd4d8050a717c421280dec1fb944a55dcf8b [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 Andersen3570a342000-09-25 21:45:58 +000028#include "busybox.h"
Eric Andersencc8ed391999-10-05 16:24:54 +000029#include <stdio.h>
30#include <stdlib.h>
31#include <unistd.h>
32
33#include <sys/types.h>
34#include <sys/socket.h>
35#include <netinet/in.h>
36#include <arpa/inet.h>
37#include <netdb.h>
38#include <sys/time.h>
39#include <sys/ioctl.h>
40
Eric Andersen2ce1edc1999-10-12 15:42:48 +000041#define BUFSIZE 100
42
Erik Andersen7ab9c7e2000-05-12 19:41:47 +000043int nc_main(int argc, char **argv)
Eric Andersencc8ed391999-10-05 16:24:54 +000044{
Erik Andersene49d5ec2000-02-08 19:58:47 +000045 int sfd;
46 int result;
47 int len;
48 char ch[BUFSIZE];
Eric Andersencc8ed391999-10-05 16:24:54 +000049
Erik Andersene49d5ec2000-02-08 19:58:47 +000050 struct sockaddr_in address;
51 struct hostent *hostinfo;
Eric Andersencc8ed391999-10-05 16:24:54 +000052
Erik Andersene49d5ec2000-02-08 19:58:47 +000053 fd_set readfds, testfds;
54
Eric Andersenb6a44b81999-11-13 04:47:09 +000055 argc--;
56 argv++;
Erik Andersen5e1189e2000-04-15 16:34:54 +000057 if (argc < 2 || **(argv + 1) == '-') {
Erik Andersen7ab9c7e2000-05-12 19:41:47 +000058 usage(nc_usage);
Erik Andersen5e1189e2000-04-15 16:34:54 +000059 }
Eric Andersenb6a44b81999-11-13 04:47:09 +000060
Erik Andersene49d5ec2000-02-08 19:58:47 +000061 sfd = socket(AF_INET, SOCK_STREAM, 0);
Eric Andersencc8ed391999-10-05 16:24:54 +000062
Erik Andersene49d5ec2000-02-08 19:58:47 +000063 hostinfo = (struct hostent *) gethostbyname(*argv);
Eric Andersencc8ed391999-10-05 16:24:54 +000064
Erik Andersene49d5ec2000-02-08 19:58:47 +000065 if (!hostinfo) {
Mark Whitleyf57c9442000-12-07 19:56:48 +000066 error_msg_and_die("cannot resolve %s\n", *argv);
Erik Andersene49d5ec2000-02-08 19:58:47 +000067 }
Eric Andersencc8ed391999-10-05 16:24:54 +000068
Erik Andersene49d5ec2000-02-08 19:58:47 +000069 address.sin_family = AF_INET;
70 address.sin_addr = *(struct in_addr *) *hostinfo->h_addr_list;
71 address.sin_port = htons(atoi(*(++argv)));
Eric Andersencc8ed391999-10-05 16:24:54 +000072
Erik Andersene49d5ec2000-02-08 19:58:47 +000073 len = sizeof(address);
Eric Andersencc8ed391999-10-05 16:24:54 +000074
Erik Andersene49d5ec2000-02-08 19:58:47 +000075 result = connect(sfd, (struct sockaddr *) &address, len);
Eric Andersencc8ed391999-10-05 16:24:54 +000076
Erik Andersene49d5ec2000-02-08 19:58:47 +000077 if (result < 0) {
Eric Andersenafeb6832000-06-04 05:21:49 +000078 perror("nc: connect");
Erik Andersene49d5ec2000-02-08 19:58:47 +000079 exit(2);
80 }
Eric Andersencc8ed391999-10-05 16:24:54 +000081
Erik Andersene49d5ec2000-02-08 19:58:47 +000082 FD_ZERO(&readfds);
83 FD_SET(sfd, &readfds);
84 FD_SET(fileno(stdin), &readfds);
Eric Andersencc8ed391999-10-05 16:24:54 +000085
Erik Andersene49d5ec2000-02-08 19:58:47 +000086 while (1) {
87 int fd;
Eric Andersen2ce1edc1999-10-12 15:42:48 +000088 int ofd;
Erik Andersene49d5ec2000-02-08 19:58:47 +000089 int nread;
Eric Andersencc8ed391999-10-05 16:24:54 +000090
Erik Andersene49d5ec2000-02-08 19:58:47 +000091 testfds = readfds;
Eric Andersencc8ed391999-10-05 16:24:54 +000092
Erik Andersene49d5ec2000-02-08 19:58:47 +000093 result =
94 select(FD_SETSIZE, &testfds, (fd_set *) NULL, (fd_set *) NULL,
95 (struct timeval *) 0);
Eric Andersencc8ed391999-10-05 16:24:54 +000096
Erik Andersene49d5ec2000-02-08 19:58:47 +000097 if (result < 1) {
Eric Andersenafeb6832000-06-04 05:21:49 +000098 perror("nc: select");
Erik Andersene49d5ec2000-02-08 19:58:47 +000099 exit(3);
100 }
Eric Andersencc8ed391999-10-05 16:24:54 +0000101
Erik Andersene49d5ec2000-02-08 19:58:47 +0000102 for (fd = 0; fd < FD_SETSIZE; fd++) {
103 if (FD_ISSET(fd, &testfds)) {
Eric Andersen2ce1edc1999-10-12 15:42:48 +0000104 int trn = 0;
105 int rn;
Eric Andersencc8ed391999-10-05 16:24:54 +0000106
Erik Andersene49d5ec2000-02-08 19:58:47 +0000107 ioctl(fd, FIONREAD, &nread);
Eric Andersencc8ed391999-10-05 16:24:54 +0000108
Erik Andersene49d5ec2000-02-08 19:58:47 +0000109 if (fd == sfd) {
110 if (nread == 0)
111 exit(0);
Eric Andersen2ce1edc1999-10-12 15:42:48 +0000112 ofd = fileno(stdout);
Erik Andersene49d5ec2000-02-08 19:58:47 +0000113 } else {
Eric Andersen2ce1edc1999-10-12 15:42:48 +0000114 ofd = sfd;
115 }
Eric Andersencc8ed391999-10-05 16:24:54 +0000116
Eric Andersencc8ed391999-10-05 16:24:54 +0000117
Eric Andersen2ce1edc1999-10-12 15:42:48 +0000118
Erik Andersene49d5ec2000-02-08 19:58:47 +0000119 do {
Eric Andersen2ce1edc1999-10-12 15:42:48 +0000120 rn = (BUFSIZE < nread - trn) ? BUFSIZE : nread - trn;
121 trn += rn;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000122 read(fd, ch, rn);
123 write(ofd, ch, rn);
Eric Andersen2ce1edc1999-10-12 15:42:48 +0000124 }
125 while (trn < nread);
126 }
Erik Andersene49d5ec2000-02-08 19:58:47 +0000127 }
128 }
Eric Andersencc8ed391999-10-05 16:24:54 +0000129}