]> git.pond.sub.org Git - empserver/blobdiff - src/client/main.c
Use src/lib/w32/w32sockets.c for client
[empserver] / src / client / main.c
index 7cd3317b3c5156fc41113a24f173c93f89fb5a9d..dea0a01d7795b95e63343ece92a020b949920d81 100644 (file)
@@ -1,6 +1,6 @@
 /*
  *  Empire - A multi-player, client/server Internet based war game.
- *  Copyright (C) 1986-2005, Dave Pare, Jeff Bailey, Thomas Ruschak,
+ *  Copyright (C) 1986-2009, Dave Pare, Jeff Bailey, Thomas Ruschak,
  *                           Ken Stevens, Steve McClure
  *
  *  This program is free software; you can redistribute it and/or modify
  *
  *  ---
  *
- *  See the "LEGAL", "LICENSE", "CREDITS" and "README" files for all the
- *  related information and legal notices. It is expected that any future
- *  projects/authors will amend these files as needed.
+ *  See files README, COPYING and CREDITS in the root of the source
+ *  tree for related information and legal notices.  It is expected
+ *  that future projects/authors will amend these files as needed.
  *
  *  ---
  *
  *  main.c: client main function
- * 
+ *
  *  Known contributors to this file:
  *     Dave Pare, 1986
  *     Steve McClure, 1998
+ *     Ron Koenderink, 2004-2005
+ *     Markus Armbruster, 2005-2009
  */
 
-#ifdef _NTSDK
-#undef _NTSDK
-#endif
+#include <config.h>
 
-#include "misc.h"
-#include "proto.h"
-#include "queue.h"
-#include "ioqueue.h"
-#include "tags.h"
-
-#include <stdio.h>
 #include <stdlib.h>
-#include <string.h>
-#include <sys/types.h>
 #ifndef _WIN32
 #include <pwd.h>
 #endif
-#include <signal.h>
-#include <errno.h>
-#include <time.h>
-#ifndef _WIN32
-#include <sys/socket.h>
-#include <sys/time.h>
-#include <netinet/in.h>
 #include <unistd.h>
-#else
-#include <windows.h>
-#include <winsock.h>
-#include <conio.h>
-#include <io.h>
-#endif
+#include "misc.h"
+#include "version.h"
 
 #ifdef _WIN32
-HANDLE hStdIn;
+#define getuid() 0
+#define getpwuid(uid) ((void)(uid), w32_getpw())
+#define sysdep_init() w32_sysdep_init()
+#else
+#define sysdep_init() ((void)0)
 #endif
 
-#define        RETRY   3
-
-int interrupt;
-int sock;
-
-static void intr(int sig);
-
+static void
+print_usage(char *program_name)
+{
+    printf("Usage: %s [OPTION]...[COUNTRY [PASSWORD]]\n"
+          "  -2 FILE         Append log of session to FILE\n"
+          "  -k              Kill connection\n"
+          "  -u              Use UTF-8\n"
+          "  -h              display this help and exit\n"
+          "  -v              display version information and exit\n",
+          program_name);
+}
 
 int
-main(int ac, char **av)
+main(int argc, char **argv)
 {
-#ifdef _WIN32
-    WSADATA WsaData;
-    int err;
-    fd_set readfds;
-    struct timeval tm;
-    DWORD stdinmode;
-    SECURITY_ATTRIBUTES security;
-    int bRedirected = 0;
-    char unamebuf[128];
-#else
-    fd_set mask;
-    fd_set savemask;
-    int retry = 0;
-#endif
-    struct ioqueue server;
-    char *argv[128];
-    int i, j;
-    char *ptr;
-    char *auxout_fname;
-    FILE *auxout_fp;
-    struct sockaddr_in sin;
-    int n;
-    char *cname;
-    char *pname;
-    char *uname;
+    int opt;
+    char *auxfname = NULL;
     int send_kill = 0;
+    int utf8 = 0;
+    char **ap;
+    char *country;
+    char *passwd;
+    char *uname;
+    char *host;
+    char *port;
+    int sock;
 
-#ifdef _WIN32
-    err = WSAStartup(0x0101, &WsaData);
-    if (err == SOCKET_ERROR) {
-       printf("WSAStartup Failed\n");
-       return FALSE;
-    }
-#else
-    FD_ZERO(&mask);
-    FD_ZERO(&savemask);
-#endif
-    memset(argv, 0, sizeof(argv));
-    saveargv(ac, av, argv);
-    auxout_fname = NULL;
-    auxout_fp = NULL;
-    for (i = j = 1; i < ac; ++i) {
-       ptr = argv[i];
-       if (strcmp(ptr, "-2") == 0) {
-           if (i + 1 >= ac) {
-               fprintf(stderr, "-2: Missing filename!\n");
-               exit(1);
-           }
-           auxout_fname = argv[i + 1];
-           ++i;
-           continue;
-       } else if (strcmp(ptr, "-k") == 0) {
+    while ((opt = getopt(argc, argv, "2:kuhv")) != EOF) {
+       switch (opt) {
+       case '2':
+           auxfname = optarg;
+           break;
+       case 'k':
            send_kill = 1;
-           continue;
+           break;
+       case 'u':
+           utf8 = eight_bit_clean = 1;
+           break;
+       case 'h':
+           print_usage(argv[0]);
+           exit(0);
+       case 'v':
+           printf("%s\n\n%s", version, legal);
+           exit(0);
+       default:
+           print_usage(argv[0]);
+           exit(1);
        }
-       argv[j] = argv[i];
-       ++j;
     }
-    ac = j;
-    if (auxout_fname && (auxout_fp = fopen(auxout_fname, "a")) == NULL) {
-       fprintf(stderr, "Unable to open %s for append\n", auxout_fname);
-       exit(1);
-    }
-    getsose();
-    if (!hostport(getenv("EMPIREPORT"), &sin) &&
-       !hostport("empire", &sin) && !hostport(empireport, &sin)) {
-       fprintf(stderr, "No empire port\n");
-       exit(1);
-    }
-    if (!hostaddr(getenv("EMPIREHOST"), &sin) &&
-       !hostaddr(empirehost, &sin)) {
-       fprintf(stderr, "No empire host\n");
-       exit(1);
-    }
-    if ((sock = hostconnect(&sin)) < 0)
-       exit(1);
-    cname = getenv("COUNTRY");
-    if (ac > 1)
-       cname = argv[1];
-    pname = getenv("PLAYER");
-    if (ac > 2)
-       pname = argv[2];
+
+    ap = argv + optind;
+    if (*ap)
+       country = *ap++;
+    else
+       country = getenv("COUNTRY");
+    if (*ap)
+       passwd = *ap++;
+    else
+       passwd = getenv("PLAYER");
+    port = getenv("EMPIREPORT");
+    if (!port)
+       port = empireport;
+    host = getenv("EMPIREHOST");
+    if (!host)
+       host = empirehost;
     uname = getenv("LOGNAME");
     if (uname == NULL) {
-#ifndef _WIN32
        struct passwd *pwd;
 
        pwd = getpwuid(getuid());
@@ -172,151 +127,23 @@ main(int ac, char **av)
            exit(1);
        }
        uname = pwd->pw_name;
-#else
-       DWORD unamesize;
-
-       unamesize = sizeof(unamebuf);
-       if (GetUserName(unamebuf, &unamesize)) {
-           uname = unamebuf;
-           if ((unamesize <= 0 ) || (strlen(uname) <= 0))
-               uname = "nobody";
-       } else
-           uname = "nobody";
-#endif
     }
-    if (!login(sock, uname, cname, pname, send_kill)) {
-       close(sock);
+
+    getsose();
+    if (auxfname && (auxfp = fopen(auxfname, "a")) == NULL) {
+       fprintf(stderr, "Unable to open %s for append\n", auxfname);
        exit(1);
     }
-    ioq_init(&server, 2048);
-    io_init();
-#ifndef _WIN32
-    FD_ZERO(&mask);
-    FD_SET(0, &savemask);
-    FD_SET(sock, &savemask);
-#endif
-    (void)signal(SIGINT, intr);
-#ifndef _WIN32
-    (void)signal(SIGPIPE, SIG_IGN);
-    while (FD_ISSET(sock, &savemask)) {
-       mask = savemask;
-       n = select(sock + 1, &mask, NULL, NULL, NULL);
-       if (interrupt) {
-           if (!handleintr(sock))
-               break;
-           errno = 0;
-       }
-       if (n <= 0) {
-           if (errno == EINTR) {
-               perror("select");
-               (void)close(sock);
-               FD_CLR(sock, &savemask);
-           }
-       } else {
-           if (FD_ISSET(0, &mask)) {
-               if (!termio(0, sock, auxout_fp)) {
-                   if (retry++ >= RETRY) {
-                       FD_CLR(0, &savemask);
-                   }
-               } else {
-                   retry = 0;
-               }
-           }
-           if (FD_ISSET(sock, &mask)) {
-               if (!serverio(sock, &server))
-                   FD_CLR(sock, &savemask);
-               else
-                   servercmd(&server, auxout_fp);
-           }
-       }
-    }
-#else
-    bRedirected = 0;
-    tm.tv_sec = 0;
-    tm.tv_usec = 1000;
 
-    if (!_isatty(_fileno(stdin)))
-       bRedirected = 1;
-    else {
-       security.nLength = sizeof(SECURITY_ATTRIBUTES);
-       security.lpSecurityDescriptor = NULL;
-       security.bInheritHandle = TRUE;
-       hStdIn = CreateFile("CONIN$",
-                           GENERIC_READ | GENERIC_WRITE,
-                           FILE_SHARE_READ | FILE_SHARE_WRITE,
-                           &security, OPEN_EXISTING, (DWORD) NULL, NULL);
-       
-       if (hStdIn == INVALID_HANDLE_VALUE) {
-           printf("Error getting hStdIn.\n");
-           fflush(stdout);
-           exit(-3);
-       }
-       
-       err = GetConsoleMode(hStdIn, &stdinmode);
-       if (!err) {
-           printf("Error getting console mode.\n");
-           fflush(stdout);
-           exit(-4);
-       } else {
-           stdinmode |= ENABLE_LINE_INPUT | ENABLE_ECHO_INPUT;
-           err = SetConsoleMode(hStdIn, stdinmode);
-           if (!err) {
-               printf("Error setting console mode.\n");
-               fflush(stdout);
-               exit(-5);
-           }
-       }
-    }
-    while (1) {
-       FD_ZERO(&readfds);
-       FD_SET(sock, &readfds);
-       n = select(sock + 1, &readfds, NULL, NULL, &tm);
-       if (interrupt) {
-           if (!handleintr(sock))
-               break;
-           errno = 0;
-       }
-       if (n < 0) {
-           if (errno == EINTR) {
-               errno = WSAGetLastError();
-               perror("select");
-               (void)close(sock);
-               break;
-           }
-       } else {
-           if (bRedirected == 1) {
-               if (!termio(0, sock, auxout_fp))
-                   bRedirected = -1;
-           } else if (bRedirected == 0) {
-               if (WaitForSingleObject(hStdIn, 10) != WAIT_TIMEOUT) {
-                   termio(-1, sock, auxout_fp);
-                   FlushConsoleInputBuffer(hStdIn);
-               }
-           }
-           if (FD_ISSET(sock, &readfds)) {
-               if (!serverio(sock, &server))
-                   break;
-               else
-                   servercmd(&server, auxout_fp);
-           }
-       }
-    }
-    if (bRedirected == 0)
-       CloseHandle(hStdIn);
-#endif
-    ioq_drain(&server);
-    (void)close(sock);
-    return 0;                  /* Shut the compiler up */
-}
+    sysdep_init();
 
-static void
-intr(int sig)
-{
-    interrupt++;
-#ifdef _WIN32
-    signal(SIGINT, intr);
-#endif
-#ifdef hpux
-    signal(SIGINT, intr);
-#endif
+    sock = tcp_connect(host, port);
+
+    if (!login(sock, uname, country, passwd, send_kill, utf8))
+       exit(1);
+
+    if (play(sock) < 0)
+       exit(1);
+
+    return 0;
 }