empserver/src/client/main.c

307 lines
6.9 KiB
C

/*
* Empire - A multi-player, client/server Internet based war game.
* Copyright (C) 1986-2000, Dave Pare, Jeff Bailey, Thomas Ruschak,
* Ken Stevens, Steve McClure
*
* This program 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
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* 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
*
* ---
*
* 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.
*
* ---
*
* main.c: client main function
*
* Known contributors to this file:
* Dave Pare, 1986
* Steve McClure, 1998
*/
#ifdef _NTSDK
#undef _NTSDK
#endif
#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
#ifdef _WIN32
HANDLE hStdIn;
#endif
#define RETRY 3
int interrupt;
int sock;
void saveargv(int ac, s_char **src, s_char **dst);
void getsose(void);
int hostport(s_char *name, struct sockaddr_in *addr);
int hostaddr(s_char *name, struct sockaddr_in *addr);
int hostconnect(struct sockaddr_in *addr);
int login(int s, s_char *uname, s_char *cname, s_char *cpass, int kill_proc);
int handleintr(int);
int termio(int fd, int sock, FILE *auxfi);
int serverio(int s, struct ioqueue *ioq);
void servercmd(struct ioqueue *ioq, FILE *auxfi);
static void intr(int sig);
int
main(int ac, s_char **av)
{
#ifdef _WIN32
WSADATA WsaData;
int err;
fd_set readfds;
struct timeval tm;
DWORD stdinmode;
SECURITY_ATTRIBUTES security;
#endif
extern s_char empireport[];
extern s_char empirehost[];
fd_set mask;
struct ioqueue server;
s_char *argv[128];
int i, j;
s_char *ptr;
s_char *auxout_fname;
FILE *auxout_fp;
#ifndef _WIN32
fd_set savemask;
struct passwd *pwd;
#endif
struct sockaddr_in sin;
int n;
s_char *cname;
s_char *pname;
int retry = 0;
int send_kill = 0;
#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 = 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);
exit(1);
}
#ifndef _WIN32
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);
}
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)) {
#else
if (!login(sock, "nobody", cname, pname, send_kill)) {
#endif
close(sock);
exit(1);
}
ioq_init(&server, 2048);
io_init();
FD_ZERO(&mask);
#ifndef _WIN32
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, (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);
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
tm.tv_sec = 0;
tm.tv_usec = 1000;
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);
}
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 setting console mode.\n");
fflush(stdout);
}
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);
}
}
}
CloseHandle(hStdIn);
#endif
ioq_drain(&server);
(void)close(sock);
exit(0);
return 0; /* Shut the compiler up */
}
static void
intr(int sig)
{
interrupt++;
#ifdef hpux
signal(SIGINT, intr);
#endif
}