/*
* Empire - A multi-player, client/server Internet based war game.
- * Copyright (C) 1986-2005, Dave Pare, Jeff Bailey, Thomas Ruschak,
- * Ken Stevens, Steve McClure
+ * Copyright (C) 1986-2021, Dave Pare, Jeff Bailey, Thomas Ruschak,
+ * Ken Stevens, Steve McClure, Markus Armbruster
*
- * This program is free software; you can redistribute it and/or modify
+ * Empire is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
+ * the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
*
* ---
*
- * 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-2007
+ * Markus Armbruster, 2005-2015
+ * Tom Dickson-Hunt, 2010
+ * Martin Haukeli, 2015
*/
#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
+#ifdef _WIN32
+#include <windows.h>
+#include <shlobj.h>
+#include "sys/socket.h"
+#else
#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 "fnameat.h"
+#include "misc.h"
+#include "version.h"
#ifdef _WIN32
-HANDLE hStdIn;
-#endif
+#define getuid() 0
+#define getpwuid(uid) ((void)(uid), w32_getpw())
+#define sysdep_init() w32_sysdep_init()
-#define RETRY 3
+struct passwd {
+ char *pw_name;
+ char *pw_dir;
+};
-int eight_bit_clean;
-int sock;
-
-static volatile sig_atomic_t interrupt;
-static void intr(int sig);
+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)
+{
+ printf("Usage: %s [OPTION]...[COUNTRY [PASSWORD]]\n"
+ " -2 FILE Append log of session to FILE\n"
+ " -k Kill connection\n"
+ " -r Restricted mode, no redirections\n"
+ " -s [HOST:]PORT Specify server HOST and PORT\n"
+ " -u Use UTF-8\n"
+#ifdef HAVE_LIBREADLINE
+ " -H FILE Load and save command history from FILE\n"
+ " (default ~/.empire_history with -r, none without -r)\n"
+#endif /* HAVE_LIBREADLINE */
+ " -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;
- char *host;
- char *port;
+ int opt;
+ char *auxfname = NULL;
+ char *history_file = NULL;
int send_kill = 0;
+ char *host = NULL;
+ char *port = NULL;
int utf8 = 0;
+ char **ap;
+ char *country;
+ char *passwd;
+ char *uname;
+ char *udir;
+ char *colon;
+ int sock;
-#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);
- 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:H:krs:uhv")) != EOF) {
+ switch (opt) {
+ case '2':
+ auxfname = optarg;
+ break;
+#ifdef HAVE_LIBREADLINE
+ case 'H':
+ history_file = optarg;
+ break;
+#endif /* HAVE_LIBREADLINE */
+ case 'k':
send_kill = 1;
- continue;
- } else if (strcmp(ptr, "-u") == 0) {
+ break;
+ case 'r':
+ restricted = 1;
+ break;
+ case 's':
+ port = strdup(optarg);
+ colon = strrchr(port, ':');
+ if (colon) {
+ *colon = 0;
+ host = port;
+ port = colon + 1;
+ }
+ break;
+ case 'u':
utf8 = eight_bit_clean = 1;
- continue;
+ break;
+ case 'h':
+ print_usage(argv[0]);
+ exit(0);
+ case 'v':
+ printf("%s\n\n%s", version, legal);
+ exit(0);
+ default:
+ fprintf(stderr, "Try -h for help.\n");
+ 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();
- port = getenv("EMPIREPORT");
+
+ ap = argv + optind;
+ if (*ap)
+ country = *ap++;
+ else
+ country = getenv("COUNTRY");
+ if (*ap)
+ passwd = *ap++;
+ else
+ passwd = getenv("PLAYER");
+ if (!port)
+ port = getenv("EMPIREPORT");
if (!port)
port = empireport;
- if (!hostport(port, &sin)) {
- fprintf(stderr, "Can't resolve Empire port %s\n", port);
- exit(1);
- }
- host = getenv("EMPIREHOST");
+ if (!host)
+ host = getenv("EMPIREHOST");
if (!host)
host = empirehost;
- if (!hostaddr(host, &sin)) {
- fprintf(stderr, "Can't resolve Empire host %s\n", host);
- exit(1);
- }
- if ((sock = hostconnect(&sin)) < 0) {
- perror("Can't connect to Empire server");
- exit(1);
- }
- 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
+ udir = getenv("HOME");
+ if (!uname || !udir) {
struct passwd *pwd;
pwd = getpwuid(getuid());
fprintf(stderr, "You don't exist. Go away\n");
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 (!uname)
+ uname = pwd->pw_name;
+ if (!udir)
+ udir = pwd->pw_dir;
}
- if (!login(sock, uname, cname, pname, send_kill, utf8)) {
- close(sock);
+ if (*ap) {
+ fprintf(stderr, "%s: extra operand %s\n", argv[0], *ap);
+ fprintf(stderr, "Try -h for help.\n");
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);
- }
- }
+ getsose();
+ if (auxfname && (auxfp = fopen(auxfname, "a")) == NULL) {
+ fprintf(stderr, "Unable to open %s for append\n", auxfname);
+ exit(1);
}
- if (bRedirected == 0)
- CloseHandle(hStdIn);
-#endif
- ioq_drain(&server);
- (void)close(sock);
- return 0; /* Shut the compiler up */
+
+ sysdep_init();
+
+ sock = tcp_connect(host, port);
+
+ if (!restricted && !history_file)
+ history_file = ".empire_history";
+ if (history_file)
+ history_file = fnameat(history_file, udir);
+
+ if (!login(sock, uname, country, passwd, send_kill, utf8))
+ exit(1);
+
+ if (play(sock, history_file) < 0)
+ exit(1);
+
+ return 0;
}
-static void
-intr(int sig)
-{
- interrupt = 1;
#ifdef _WIN32
- signal(SIGINT, intr);
-#endif
-#ifdef hpux
- signal(SIGINT, intr);
-#endif
+/*
+ * Get Windows user name and directory
+ */
+static struct passwd *
+w32_getpw(void)
+{
+ static char unamebuf[128];
+ static char udirbuf[MAX_PATH];
+ 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";
+ if (SUCCEEDED(SHGetFolderPathA(NULL, CSIDL_PROFILE, NULL, 0, udirbuf))
+ && strlen(udirbuf) == 0)
+ pwd.pw_dir = udirbuf;
+ return &pwd;
}
-static int
-handleintr(int s)
+static void
+w32_sysdep_init(void)
{
- 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;
+ int err;
+
+ /*
+ * 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