[ros-diffs] [janderwald] 25057: - fix compiling with msvc - replace getopt with simple parse routine

Author: janderwald
Date: Mon Dec  4 01:29:07 2006
New Revision: 25057

URL: http://svn.reactos.org/svn/reactos?rev=25057&view=rev
Log:
- fix compiling with msvc
- replace getopt with simple parse routine

Modified:
    trunk/reactos/base/applications/network/whois/whois.c

Modified: trunk/reactos/base/applications/network/whois/whois.c
URL:
http://svn.reactos.org/svn/reactos/trunk/reactos/base/applications/network/…
==============================================================================
--- trunk/reactos/base/applications/network/whois/whois.c (original)
+++ trunk/reactos/base/applications/network/whois/whois.c Mon Dec  4 01:29:07 2006
@@ -45,42 +45,57 @@
 #include <stdlib.h>
 
 /* #include <various.h> */
-#include <getopt.h>
-#include <io.h>
+/* #include <getopt.h> */
+/* #include <io.h> */
 
-#define	NICHOST	"whois.internic.net"
+static char NICHOST[] = "whois.internic.net";
+char *host = NULL;
+int optset = 0;
+static void usage();
+static void cleanup(int iExitCode);
 
-static void usage();
-static void leave(int iExitCode);
+void getwhoisserver(int argc, char **argv)
+{
+	int i = 1;
+
+    while(i < argc)
+	{
+		if (!strcmp(argv[i], "-h"))
+		{
+			if (i + 2 < argc)
+			{
+				host = argv[i +1];
+				optset = i + 1;	
+			}
+			else
+			{
+				optset = argc;
+			}
+			return;
+		}
+		i++;
+	}
+	host = NICHOST;
+	optset = 1;
+}
 
 int main(int argc, char **argv)
 {
-	extern char *optarg;
-	extern int optind;
 	char ch;
 	struct sockaddr_in sin;
 	struct hostent *hp;
 	struct servent *sp;
 	int s;
-	const char *host;
+
 	WORD wVersionRequested;
 	WSADATA wsaData;
 	int err;
 
-	host = NICHOST;
-	while ((ch = (char)getopt(argc, argv, "h:")) != EOF)
-		switch((char)ch) {
-		case 'h':
-			host = optarg;
-			break;
-		case '?':
-		default:
-			usage();
-		}
-	argc -= optind;
-	argv += optind;
+	getwhoisserver(argc, argv);
+	argc -= optset;
+	argv += optset;
 
-	if (!argc)
+    if (!host || !argc)
 		usage();
 
 	/* Start winsock */
@@ -91,34 +106,34 @@
 		/* Tell the user that we couldn't find a usable */
 		/* WinSock DLL.                                 */
 		perror("whois: WSAStartup failed");
-		leave(1);
+		cleanup(1);
 	}
 
 	hp = gethostbyname(host);
 	if (hp == NULL) {
 		(void)fprintf(stderr, "whois: %s: ", host);
-		leave(1);
+		cleanup(1);
 	}
 	host = hp->h_name;
 
 	s = socket(hp->h_addrtype, SOCK_STREAM, 0);
-	if (s < 0) {
+	if (s == INVALID_SOCKET) {
 		perror("whois: socket");
-		leave(1);
+		cleanup(1);
 	}
 
 	memset(/*(caddr_t)*/&sin, 0, sizeof(sin));
 	sin.sin_family = hp->h_addrtype;
 	if (bind(s, (struct sockaddr *)&sin, sizeof(sin)) < 0) {
 		perror("whois: bind");
-		leave(1);
+		cleanup(1);
 	}
 
 	memcpy((char *)&sin.sin_addr, hp->h_addr, hp->h_length);
 	sp = getservbyname("whois", "tcp");
 	if (sp == NULL) {
 		(void)fprintf(stderr, "whois: whois/tcp: unknown service\n");
-		leave(1);
+		cleanup(1);
 	}
 
 	sin.sin_port = sp->s_port;
@@ -128,7 +143,7 @@
 
 	if (connect(s, (struct sockaddr *)&sin, sizeof(sin)) < 0) {
 		fprintf(stderr, "whois: connect error = %d\n", WSAGetLastError());
-		leave(1);
+		cleanup(1);
 	}
 
 	/* WinSock doesn't allow using a socket as a file descriptor. */
@@ -150,17 +165,17 @@
 	while (recv(s, &ch, 1, 0) == 1)
 		putchar(ch);
 
-	leave(0);
+	cleanup(0);
 	return 0;
 }
 
 static void usage()
 {
 	(void)fprintf(stderr, "usage: whois [-h hostname] name ...\n");
-	leave(1);
+	cleanup(1);
 }
 
-static void leave(int iExitCode)
+static void cleanup(int iExitCode)
 {
 	WSACleanup();
 	exit(iExitCode);