]> git.pond.sub.org Git - empserver/blobdiff - src/client/main.c
COPYING duplicates information from README. Remove. Move GPL from
[empserver] / src / client / main.c
index cb539c60b668481a08d87f39de072aada8c54d33..11ed0aff4fed7a5e45dc5c190ab43c8e342586ae 100644 (file)
@@ -1,6 +1,6 @@
 /*
  *  Empire - A multi-player, client/server Internet based war game.
- *  Copyright (C) 1986-2000, Dave Pare, Jeff Bailey, Thomas Ruschak,
+ *  Copyright (C) 1986-2006, Dave Pare, Jeff Bailey, Thomas Ruschak,
  *                           Ken Stevens, Steve McClure
  *
  *  This program is free software; you can redistribute it and/or modify
@@ -19,9 +19,9 @@
  *
  *  ---
  *
- *  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.
  *
  *  ---
  *
  *     Steve McClure, 1998
  */
 
-#ifdef _NTSDK
-#undef _NTSDK
-#endif
+#include <config.h>
 
 #include "misc.h"
 #include "proto.h"
 #include "queue.h"
 #include "ioqueue.h"
-#include "bit.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
 
 #ifdef _WIN32
 HANDLE hStdIn;
 #endif
 
-extern int errno;
-
 #define        RETRY   3
 
-int    interrupt;
-int    sock;
-
-void saveargv();
-void getsose();
-int hostport();
-int hostaddr();
-int hostconnect();
-int login();
-void ioq_init();
-void io_init();
-void bit_copy();
-int handleintr();
-int termio();
-int serverio();
-void servercmd();
-void ioq_drain();
-
-
-#ifdef _WIN32
-s_char *
-index(str, c)
-s_char *str;
-char c;
-{
-       static s_char *p;
-
-       p = str;
-       while (p && *p) {
-               if (*p == c)
-                       return (s_char *)p;
-               p++;
-       }
-       return (s_char *)0;
-}
-
-bzero(ptr, size)
-s_char *ptr;
-int size;
-{
-       int i;
-
-       for (i = 0; i < size; i++)
-               ptr[i] = 0;
-}
-
-bcopy(src, dest, size)
-unsigned char *src, *dest;
-int size;
-{
-       int i;
-       unsigned char *d = dest,
-                  *s = src;
+int eight_bit_clean;
+int sock;
 
-       for (i = 0; i < size; i++)
-               *d++ = *s++;
-}
-#endif
+static volatile sig_atomic_t interrupt;
+static void intr(int sig);
+static int handleintr(int);
 
 int
-main(ac, av)
-       int     ac;
-       s_char  *av[];
+main(int ac, char **av)
 {
 #ifdef _WIN32
-       WSADATA WsaData;
-       int err;
-       fd_set  readfds;
-       struct timeval tm;
-       INPUT_RECORD InpBuffer[10];
-       DWORD numevents;
-       DWORD stdinmode;
-       SECURITY_ATTRIBUTES security;
-       void intr(void);
-#endif
-       extern  s_char *getenv();
-       extern  s_char empireport[];
-       extern  s_char empirehost[];
-       bit_fdmask mask;
-       bit_fdmask savemask;
-       struct  ioqueue server;
-       s_char  *argv[128];
-       int     i, j;
-       s_char  *ptr;
-       s_char  *auxout_fname;
-       FILE    *auxout_fp;
-#ifndef _WIN32
-       struct  passwd *pwd;
-       void    intr();
+    WSADATA WsaData;
+    int err;
+    fd_set readfds;
+    struct timeval tm;
+    DWORD stdinmode;
+    SECURITY_ATTRIBUTES security;
+    int bRedirected = 0;
+    char unamebuf[128];
+#else
+    struct sigaction sa;
+    fd_set mask;
+    fd_set savemask;
+    int retry = 0;
 #endif
-       struct  sockaddr_in sin;
-       int     n;
-       s_char  *cname;
-       s_char  *pname;
-       int     retry = 0;
-       int     send_kill = 0;
+    struct ioqueue server;
+    char *argv[128];
+    int i, j;
+    char *ptr;
+    char *auxout_fname;
+    FILE *auxout_fp;
+    int n;
+    char *cname;
+    char *pname;
+    char *uname;
+    char *host;
+    char *port;
+    int send_kill = 0;
+    int utf8 = 0;
 
 #ifdef _WIN32
-       err = WSAStartup (0x0101, &WsaData);
-       if (err == SOCKET_ERROR) {
-               printf ("WSAStartup Failed\n");
-               return FALSE;
-       }
+    /*
+     * stdout is unbuffered under Windows if connected to a character
+     * device, and putchar() screws up when printing multibyte strings
+     * bytewise to an unbuffered stream.  Switch stdout to line-
+     * buffered mode.  Unfortunately, ISO C allows implementations to
+     * screw that up, and of course Windows does.  Manual flushing
+     * after each prompt is required.
+     */
+    setvbuf(stdout, NULL, _IOLBF, 4096);
+    err = WSAStartup(0x0101, &WsaData);
+    if (err == SOCKET_ERROR) {
+       printf("WSAStartup Failed\n");
+       return FALSE;
+    }
 #else
-       mask = bit_newfdmask();
-       savemask = bit_newfdmask();
+    FD_ZERO(&mask);
+    FD_ZERO(&savemask);
 #endif
-       bzero((s_char *)argv, sizeof(argv));
-       saveargv(ac, av, argv);
-       auxout_fname = 0;
-       auxout_fp = 0;
-       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) {
-                       send_kill = 1;
-                       continue;
-               }
-               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);
+    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) {
+           send_kill = 1;
+           continue;
+       } else if (strcmp(ptr, "-u") == 0) {
+           utf8 = eight_bit_clean = 1;
+           continue;
        }
+       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();
+    port = getenv("EMPIREPORT");
+    if (!port)
+       port = empireport;
+    host = getenv("EMPIREHOST");
+    if (!host)
+       host = empirehost;
+    sock = tcp_connect(host, port);
+    cname = getenv("COUNTRY");
+    if (ac > 1)
+       cname = argv[1];
+    pname = getenv("PLAYER");
+    if (ac > 2)
+       pname = argv[2];
+    uname = getenv("LOGNAME");
+    if (uname == NULL) {
 #ifndef _WIN32
+       struct passwd *pwd;
+
        pwd = getpwuid(getuid());
        if (pwd == NULL) {
-               fprintf(stderr, "You don't exist.  Go away\n");
-               exit(1);
-       }
-#endif
-       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);
+           fprintf(stderr, "You don't exist.  Go away\n");
+           exit(1);
        }
-       if ((sock = hostconnect(&sin)) < 0)
-               exit(1);
-       cname = getenv("COUNTRY");
-       pname = getenv("PLAYER");
-       if (ac > 1)
-               cname = argv[1];
-       if (ac > 2)
-               pname = argv[2];
-#ifndef _WIN32
-       if (!login(sock, pwd->pw_name, cname, pname, send_kill)) {
+       uname = pwd->pw_name;
 #else
-       if (!login(sock, "win-empcl2.8", cname, pname, send_kill)) {
-#endif
-               close(sock);
-               exit(1);
-       }
-       ioq_init(&server, 2048);
-       io_init();
-       mask = bit_newfdmask();
-#ifndef _WIN32
-       BIT_SETB(0, savemask);
-       BIT_SETB(sock, savemask);
+       DWORD unamesize;
+
+       unamesize = sizeof(unamebuf);
+       if (GetUserName(unamebuf, &unamesize)) {
+           uname = unamebuf;
+           if ((unamesize <= 0 ) || (strlen(uname) <= 0))
+               uname = "nobody";
+       } else
+           uname = "nobody";
 #endif
-       (void) signal(SIGINT, intr);
+    }
+    if (!login(sock, uname, cname, pname, send_kill, utf8)) {
+       close(sock);
+       exit(1);
+    }
+    ioq_init(&server, 2048);
+    io_init();
 #ifndef _WIN32
-       (void) signal(SIGPIPE, SIG_IGN);
-       while (BIT_ISSETB(sock, savemask)) {
-               bit_copy(savemask, mask);
-               n = select(sock+1, (fd_set *)mask, (fd_set *)0, (fd_set *)0,
-                       (struct timeval *)0);
-               if (interrupt) {
-                       if (!handleintr(sock))
-                               break;
-                       errno = 0;
-               }
-               if (n <= 0) {
-                       if (errno == EINTR) {
-                               perror("select");
-                               (void) close(sock);
-                               BIT_CLRB(sock, savemask);
-                       }
+    FD_ZERO(&mask);
+    FD_SET(0, &savemask);
+    FD_SET(sock, &savemask);
+    sigemptyset(&sa.sa_mask);
+    sa.sa_flags = 0;
+    sa.sa_handler = intr;
+    sigaction(SIGINT, &sa, NULL);
+    sa.sa_handler = SIG_IGN;
+    sigaction(SIGPIPE, &sa, NULL);
+    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 {
-                       if (BIT_ISSETB(0, mask)) {
-                               if (!termio(0, sock, auxout_fp)) {
-                                       if (retry++ >= RETRY)  {
-                                               BIT_CLRB(0, savemask);
-                                       }
-                               } else {
-                                       retry = 0;
-                               }
-                       }
-                       if (BIT_ISSETB(sock, mask)) {
-                               if (!serverio(sock, &server))
-                                       BIT_CLRB(sock, savemask);
-                               else
-                                       servercmd(&server, auxout_fp);
-                       }
+                   retry = 0;
                }
+           }
+           if (FD_ISSET(sock, &mask)) {
+               if (!serverio(sock, &server))
+                   FD_CLR(sock, &savemask);
+               else
+                   servercmd(&server, auxout_fp);
+           }
        }
-#else
-       tm.tv_sec = 0;
-       tm.tv_usec = 1000;
+    }
+#else  /* _WIN32 */
+    signal(SIGINT, intr);
 
+    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);
-
+                           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);
+           printf("Error getting hStdIn.\n");
+           fflush(stdout);
+           exit(-3);
        }
+       
        err = GetConsoleMode(hStdIn, &stdinmode);
        if (!err) {
-               printf("Error getting console mode.\n");
-               fflush(stdout);
-       }
-       stdinmode |= ENABLE_LINE_INPUT|ENABLE_ECHO_INPUT;
-       err = SetConsoleMode(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, (fd_set *)0, (fd_set *)0,
-                                  (struct timeval *)&tm);
-               if (n < 0) {
-                       if (errno == EINTR) {
-                               errno = WSAGetLastError();
-                               perror("select");
-                               (void) close(sock);
-                               break;
-                       }
-               } else {
-                       if (WaitForSingleObject(hStdIn, 10) != WAIT_TIMEOUT) {
-                               if (!termio(-1, sock, auxout_fp)) {
-                                       if (retry++ >= RETRY)  {
-                                               ;
-                                       }
-                               } else {
-                                       retry = 0;
-                               }
-                               FlushConsoleInputBuffer(hStdIn);
-                       }
-                       if (FD_ISSET(sock, &readfds)) {
-                               if (!serverio(sock, &server))
-                                       break;
-                               else
-                                       servercmd(&server, auxout_fp);
-                       }
+    }
+    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);
-       exit(0);
-       return 0; /* Shut the compiler up */
+#endif /* _WIN32 */
+    ioq_drain(&server);
+    (void)close(sock);
+    return 0;                  /* Shut the compiler up */
 }
 
-void
+static void
+intr(int sig)
+{
+    interrupt = 1;
 #ifdef _WIN32
-intr(void)
-#else
-intr()
+    signal(SIGINT, intr);
 #endif
+}
+
+static int
+handleintr(int s)
 {
-       interrupt++;
-#ifdef hpux
-       signal(SIGINT, intr);
+    if (interrupt) {
+       /* tacky, but it works */
+#if !defined(_WIN32)
+       if (write(s, "\naborted\n", 1 + 7 + 1) <= 0)
+#else
+       if (send(s, "\naborted\n", 1 + 7 + 1, 0) <= 0)
 #endif
+           return 0;
+       interrupt = 0;
+    }
+    return 1;
 }