/*
* Empire - A multi-player, client/server Internet based war game.
- * Copyright (C) 1986-2007, Dave Pare, Jeff Bailey, Thomas Ruschak,
+ * Copyright (C) 1986-2010, Dave Pare, Jeff Bailey, Thomas Ruschak,
* Ken Stevens, Steve McClure
*
* This program is free software; you can redistribute it and/or modify
* ---
*
* main.c: client main function
- *
+ *
* Known contributors to this file:
* Dave Pare, 1986
* Steve McClure, 1998
- * Ron Koenderink, 2004-2005
- * Markus Armbruster, 2005-2007
+ * Ron Koenderink, 2004-2007
+ * Markus Armbruster, 2005-2009
*/
#include <config.h>
-#include <errno.h>
-#include <signal.h>
-#include <stdio.h>
#include <stdlib.h>
-#include <string.h>
-#if defined(_WIN32)
-#include <io.h>
+#ifdef _WIN32
+#include <windows.h>
+#include "sys/socket.h"
#else
#include <pwd.h>
-#include <sys/time.h>
-#include <sys/types.h>
-#include <unistd.h>
#endif
-
-#include "ioqueue.h"
+#include <unistd.h>
#include "misc.h"
-#include "proto.h"
-#include "tags.h"
#include "version.h"
#ifdef _WIN32
-HANDLE hStdIn;
-#endif
-
-#define RETRY 3
+#define getuid() 0
+#define getpwuid(uid) ((void)(uid), w32_getpw())
+#define sysdep_init() w32_sysdep_init()
-int eight_bit_clean;
-int sock;
+struct passwd {
+ char *pw_name;
+};
-static volatile sig_atomic_t interrupt;
-static void intr(int sig);
-static int handleintr(int);
+static struct passwd *w32_getpw(void);
+static void w32_sysdep_init(void);
+#else /* !_WIN32 */
+#define sysdep_init() ((void)0)
+#endif /* !_WIN32 */
static void
print_usage(char *program_name)
char *uname;
char *host;
char *port;
-#ifdef _WIN32
- WORD wVersionRequested;
- 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 ioqueue server;
- FILE *auxout_fp = NULL;
- int n;
-
-#ifdef _WIN32
- /*
- * 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);
-#else
- FD_ZERO(&mask);
- FD_ZERO(&savemask);
-#endif
+ int sock;
while ((opt = getopt(argc, argv, "2:kuhv")) != EOF) {
switch (opt) {
host = empirehost;
uname = getenv("LOGNAME");
if (uname == NULL) {
-#ifndef _WIN32
struct passwd *pwd;
pwd = getpwuid(getuid());
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
}
getsose();
- if (auxfname && (auxout_fp = fopen(auxfname, "a")) == NULL) {
+ if (auxfname && (auxfp = fopen(auxfname, "a")) == NULL) {
fprintf(stderr, "Unable to open %s for append\n", auxfname);
exit(1);
}
-#ifdef _WIN32
- wVersionRequested = MAKEWORD(2, 0);
- err = WSAStartup(wVersionRequested, &WsaData);
- if (err != 0) {
- printf("WSAStartup Failed, error code %d\n", err);
- exit(1);
- }
-#endif
+ sysdep_init();
sock = tcp_connect(host, port);
if (!login(sock, uname, country, passwd, send_kill, utf8))
exit(1);
- ioq_init(&server, 2048);
- io_init();
-#ifndef _WIN32
- 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 {
- retry = 0;
- }
- }
- if (FD_ISSET(sock, &mask)) {
- if (!serverio(sock, &server))
- FD_CLR(sock, &savemask);
- else
- servercmd(&server, auxout_fp);
- }
- }
- }
-#else /* _WIN32 */
- signal(SIGINT, intr);
+ if (play(sock) < 0)
+ exit(1);
- bRedirected = 0;
- tm.tv_sec = 0;
- tm.tv_usec = 1000;
+ return 0;
+}
- 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)closesocket(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 /* _WIN32 */
- ioq_drain(&server);
#ifdef _WIN32
- (void)closesocket(sock);
-#else
- (void)close(sock);
-#endif
- return 0; /* Shut the compiler up */
+/*
+ * Get Windows user name
+ */
+static struct passwd *
+w32_getpw(void)
+{
+ static char unamebuf[128];
+ static struct passwd pwd;
+ DWORD unamesize;
+
+ unamesize = sizeof(unamebuf);
+ if (GetUserName(unamebuf, &unamesize)) {
+ pwd.pw_name = unamebuf;
+ if (unamesize == 0 || strlen(unamebuf) == 0)
+ pwd.pw_name = "nobody";
+ } else
+ pwd.pw_name = "nobody";
+ return &pwd;
}
static void
-intr(int sig)
+w32_sysdep_init(void)
{
- interrupt = 1;
-#ifdef _WIN32
- signal(SIGINT, intr);
-#endif
-}
+ int err;
-static int
-handleintr(int s)
-{
- 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;
+ /*
+ * 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 = w32_socket_init();
+ if (err != 0) {
+ fprintf(stderr, "WSAStartup Failed, error code %d\n", err);
+ exit(1);
}
- return 1;
}
+#endif
projects / empserver / blobdiff