blob: 81dfd252937c13fecf4b2a487520e1d22e7c3ee0 [file] [log] [blame]
Eric Andersencc8ed391999-10-05 16:24:54 +00001/* mnc: mini-netcat - built from the ground up for LRP
2 Copyright (C) 1998 Charles P. Wright
3
4 0.0.1 6K It works.
5 0.0.2 5K Smaller and you can also check the exit condition if you wish.
Eric Andersen2ce1edc1999-10-12 15:42:48 +00006 0.0.3 Uses select()
Eric Andersencc8ed391999-10-05 16:24:54 +00007
8 19980918 Busy Boxed! Dave Cinege
Eric Andersen2ce1edc1999-10-12 15:42:48 +00009 19990512 Uses Select. Charles P. Wright
10 19990513 Fixes stdin stupidity and uses buffers. Charles P. Wright
Eric Andersencc8ed391999-10-05 16:24:54 +000011
12 This program is free software; you can redistribute it and/or modify
13 it under the terms of the GNU General Public License as published by
14 the Free Software Foundation; either version 2 of the License, or
15 (at your option) any later version.
16
17 This program is distributed in the hope that it will be useful,
18 but WITHOUT ANY WARRANTY; without even the implied warranty of
19 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
20 GNU General Public License for more details.
21
22 You should have received a copy of the GNU General Public License
23 along with this program; if not, write to the Free Software
24 Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
25
26*/
27#include "internal.h"
28#include <stdio.h>
29#include <stdlib.h>
30#include <unistd.h>
31
32#include <sys/types.h>
33#include <sys/socket.h>
34#include <netinet/in.h>
35#include <arpa/inet.h>
36#include <netdb.h>
37#include <sys/time.h>
38#include <sys/ioctl.h>
39
Eric Andersen2ce1edc1999-10-12 15:42:48 +000040#define BUFSIZE 100
41
Eric Andersene77ae3a1999-10-19 20:03:34 +000042static const char mnc_usage[] =
Eric Andersenb6a44b81999-11-13 04:47:09 +000043"mnc [IP] [port]\n\n"
44"mini-netcat opens a pipe to IP:port\n";
Eric Andersencc8ed391999-10-05 16:24:54 +000045
46int
Eric Andersen2ce1edc1999-10-12 15:42:48 +000047mnc_main(int argc, char **argv)
Eric Andersencc8ed391999-10-05 16:24:54 +000048{
Eric Andersencc8ed391999-10-05 16:24:54 +000049 int sfd;
50 int result;
51 int len;
Eric Andersen2ce1edc1999-10-12 15:42:48 +000052 char ch[BUFSIZE];
Eric Andersencc8ed391999-10-05 16:24:54 +000053
54 struct sockaddr_in address;
55 struct hostent *hostinfo;
56
Eric Andersencc8ed391999-10-05 16:24:54 +000057 fd_set readfds, testfds;
Eric Andersencc8ed391999-10-05 16:24:54 +000058
Eric Andersenb6a44b81999-11-13 04:47:09 +000059 if (argc<=1 || **(argv+1) == '-' ) {
60 usage( mnc_usage);
61 }
62 argc--;
63 argv++;
64
Eric Andersencc8ed391999-10-05 16:24:54 +000065 sfd = socket(AF_INET, SOCK_STREAM, 0);
66
Eric Andersenb6a44b81999-11-13 04:47:09 +000067 hostinfo = (struct hostent *) gethostbyname(*argv);
Eric Andersencc8ed391999-10-05 16:24:54 +000068
69 if (!hostinfo)
70 {
71 exit(1);
72 }
73
74 address.sin_family = AF_INET;
75 address.sin_addr = *(struct in_addr *) *hostinfo->h_addr_list;
Eric Andersenb6a44b81999-11-13 04:47:09 +000076 address.sin_port = htons(atoi(*(++argv)));
Eric Andersencc8ed391999-10-05 16:24:54 +000077
78 len = sizeof(address);
79
80 result = connect(sfd, (struct sockaddr *)&address, len);
81
82 if (result < 0)
83 {
84 exit(2);
85 }
86
Eric Andersencc8ed391999-10-05 16:24:54 +000087 FD_ZERO(&readfds);
88 FD_SET(sfd, &readfds);
89 FD_SET(fileno(stdin), &readfds);
90
91 while(1)
92 {
93 int fd;
Eric Andersen2ce1edc1999-10-12 15:42:48 +000094 int ofd;
Eric Andersencc8ed391999-10-05 16:24:54 +000095 int nread;
96
97 testfds = readfds;
98
99 result = select(FD_SETSIZE, &testfds, (fd_set *) NULL, (fd_set *) NULL, (struct timeval *) 0);
100
101 if(result < 1)
102 {
103 exit(3);
104 }
105
106 for(fd = 0; fd < FD_SETSIZE; fd++)
107 {
108 if(FD_ISSET(fd,&testfds))
109 {
Eric Andersen2ce1edc1999-10-12 15:42:48 +0000110 int trn = 0;
111 int rn;
Eric Andersencc8ed391999-10-05 16:24:54 +0000112
Eric Andersen2ce1edc1999-10-12 15:42:48 +0000113 ioctl(fd, FIONREAD, &nread);
Eric Andersencc8ed391999-10-05 16:24:54 +0000114
115 if(fd == sfd)
116 {
Eric Andersen2ce1edc1999-10-12 15:42:48 +0000117 if (nread == 0)
118 exit(0);
119 ofd = fileno(stdout);
120 }
121 else
122 {
123 ofd = sfd;
124 }
Eric Andersencc8ed391999-10-05 16:24:54 +0000125
Eric Andersencc8ed391999-10-05 16:24:54 +0000126
Eric Andersen2ce1edc1999-10-12 15:42:48 +0000127
128 do
129 {
130 rn = (BUFSIZE < nread - trn) ? BUFSIZE : nread - trn;
131 trn += rn;
132 read(fd, ch, rn);
133 write(ofd, ch, rn);
134 }
135 while (trn < nread);
136 }
137 }
138 }
Eric Andersencc8ed391999-10-05 16:24:54 +0000139}