Create a branch for header work.
[reactos.git] / base / applications / network / whois / whois.c
1 /*
2 * Copyright (c) 1980, 1993
3 * The Regents of the University of California. All rights reserved.
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 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 * 3. All advertising materials mentioning features or use of this software
14 * must display the following acknowledgement:
15 * This product includes software developed by the University of
16 * California, Berkeley and its contributors.
17 * 4. Neither the name of the University nor the names of its contributors
18 * may be used to endorse or promote products derived from this software
19 * without specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31 * SUCH DAMAGE.
32 *
33 * 8/1/97 - Ted Felix <tfelix@fred.net>
34 * Ported to Win32 from 4.4-BSDLITE2 from wcarchive.
35 * Added WSAStartup()/WSACleanup() and switched from the
36 * more convenient fdopen()/fprintf() to send()/recv().
37 */
38
39 #include <sys/types.h>
40 #include <winsock2.h>
41 /* #include <sys/socket.h> */
42 /* #include <netinet/in.h> */
43 /* #include <netdb.h> */
44 #include <stdio.h>
45 #include <stdlib.h>
46
47 /* #include <various.h> */
48 /* #include <getopt.h> */
49 /* #include <io.h> */
50
51 static char NICHOST[] = "whois.internic.net";
52 char *host = NULL;
53 int optset = 0;
54 static void usage();
55 static void cleanup(int iExitCode);
56
57 void getwhoisserver(int argc, char **argv)
58 {
59 int i = 1;
60
61 while(i < argc)
62 {
63 if (!strcmp(argv[i], "-h"))
64 {
65 if (i + 2 < argc)
66 {
67 host = argv[i +1];
68 optset = i + 1;
69 }
70 else
71 {
72 optset = argc;
73 }
74 return;
75 }
76 i++;
77 }
78 host = NICHOST;
79 optset = 1;
80 }
81
82 int main(int argc, char **argv)
83 {
84 char ch;
85 struct sockaddr_in sin;
86 struct hostent *hp;
87 struct servent *sp;
88 SOCKET s;
89
90 WORD wVersionRequested;
91 WSADATA wsaData;
92 int err;
93
94 getwhoisserver(argc, argv);
95 argc -= optset;
96 argv += optset;
97
98 if (!host || !argc)
99 usage();
100
101 /* Start winsock */
102 wVersionRequested = MAKEWORD( 1, 1 );
103 err = WSAStartup( wVersionRequested, &wsaData );
104 if ( err != 0 )
105 {
106 /* Tell the user that we couldn't find a usable */
107 /* WinSock DLL. */
108 perror("whois: WSAStartup failed");
109 cleanup(1);
110 }
111
112 hp = gethostbyname(host);
113 if (hp == NULL) {
114 (void)fprintf(stderr, "whois: %s: ", host);
115 cleanup(1);
116 }
117 host = hp->h_name;
118
119 s = socket(hp->h_addrtype, SOCK_STREAM, 0);
120 if (s == INVALID_SOCKET) {
121 perror("whois: socket");
122 cleanup(1);
123 }
124
125 memset(/*(caddr_t)*/&sin, 0, sizeof(sin));
126 sin.sin_family = hp->h_addrtype;
127 if (bind(s, (struct sockaddr *)&sin, sizeof(sin)) < 0) {
128 perror("whois: bind");
129 cleanup(1);
130 }
131
132 memcpy((char *)&sin.sin_addr, hp->h_addr, hp->h_length);
133 sp = getservbyname("nicname", "tcp");
134 if (sp == NULL) {
135 (void)fprintf(stderr, "whois: nicname/tcp: unknown service\n");
136 cleanup(1);
137 }
138
139 sin.sin_port = sp->s_port;
140
141 /* have network connection; identify the host connected with */
142 (void)printf("[%s]\n", hp->h_name);
143
144 if (connect(s, (struct sockaddr *)&sin, sizeof(sin)) < 0) {
145 fprintf(stderr, "whois: connect error = %d\n", WSAGetLastError());
146 cleanup(1);
147 }
148
149 /* WinSock doesn't allow using a socket as a file descriptor. */
150 /* Have to use send() and recv(). whois will drop connection. */
151
152 /* For each request */
153 while (argc-- > 1)
154 {
155 /* Send the request */
156 send(s, *argv, strlen(*argv), 0);
157 send(s, " ", 1, 0);
158 argv++;
159 }
160 /* Send the last request */
161 send(s, *argv, strlen(*argv), 0);
162 send(s, "\r\n", 2, 0);
163
164 /* Receive anything and print it */
165 while (recv(s, &ch, 1, 0) == 1)
166 putchar(ch);
167
168 cleanup(0);
169 return 0;
170 }
171
172 static void usage()
173 {
174 (void)fprintf(stderr, "usage: whois [-h hostname] name ...\n");
175 cleanup(1);
176 }
177
178 static void cleanup(int iExitCode)
179 {
180 WSACleanup();
181 exit(iExitCode);
182 }