1 /*
2 * Empire - A multi-player, client/server Internet based war game.
3 * Copyright (C) 1986-2021, Dave Pare, Jeff Bailey, Thomas Ruschak,
4 * Ken Stevens, Steve McClure, Markus Armbruster
5 *
6 * Empire is free software: you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation, either version 3 of the License, or
9 * (at your option) any later version.
10 *
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License
17 * along with this program. If not, see <http://www.gnu.org/licenses/>.
18 *
19 * ---
20 *
21 * See files README, COPYING and CREDITS in the root of the source
22 * tree for related information and legal notices. It is expected
23 * that future projects/authors will amend these files as needed.
24 *
25 * ---
26 *
27 * main.c: client main function
28 *
29 * Known contributors to this file:
30 * Dave Pare, 1986
31 * Steve McClure, 1998
32 * Ron Koenderink, 2004-2007
33 * Markus Armbruster, 2005-2015
34 * Tom Dickson-Hunt, 2010
35 * Martin Haukeli, 2015
36 */
37
38 #include <config.h>
39
40 #include <stdlib.h>
41 #include <string.h>
42 #ifdef _WIN32
43 #include <windows.h>
44 #include <shlobj.h>
45 #include "sys/socket.h"
46 #else
47 #include <pwd.h>
48 #endif
49 #include <unistd.h>
50 #include "fnameat.h"
51 #include "misc.h"
52 #include "version.h"
53
54 #ifdef _WIN32
55 #define getuid() 0
56 #define getpwuid(uid) ((void)(uid), w32_getpw())
57 #define sysdep_init() w32_sysdep_init()
58
59 struct passwd {
60 char *pw_name;
61 char *pw_dir;
62 };
63
64 static struct passwd *w32_getpw(void);
65 static void w32_sysdep_init(void);
66 #else /* !_WIN32 */
67 #define sysdep_init() ((void)0)
68 #endif /* !_WIN32 */
69
70 static void
print_usage(char * program_name)71 print_usage(char *program_name)
72 {
73 printf("Usage: %s [OPTION]...[COUNTRY [PASSWORD]]\n"
74 " -2 FILE Append log of session to FILE\n"
75 " -k Kill connection\n"
76 " -r Restricted mode, no redirections\n"
77 " -s [HOST:]PORT Specify server HOST and PORT\n"
78 " -u Use UTF-8\n"
79 #ifdef HAVE_LIBREADLINE
80 " -H FILE Load and save command history from FILE\n"
81 " (default ~/.empire_history with -r, none without -r)\n"
82 #endif /* HAVE_LIBREADLINE */
83 " -h display this help and exit\n"
84 " -v display version information and exit\n",
85 program_name);
86 }
87
88 int
main(int argc,char ** argv)89 main(int argc, char **argv)
90 {
91 int opt;
92 char *auxfname = NULL;
93 char *history_file = NULL;
94 int send_kill = 0;
95 char *host = NULL;
96 char *port = NULL;
97 int utf8 = 0;
98 char **ap;
99 char *country;
100 char *passwd;
101 char *uname;
102 char *udir;
103 char *colon;
104 int sock;
105
106 while ((opt = getopt(argc, argv, "2:H:krs:uhv")) != EOF) {
107 switch (opt) {
108 case '2':
109 auxfname = optarg;
110 break;
111 #ifdef HAVE_LIBREADLINE
112 case 'H':
113 history_file = optarg;
114 break;
115 #endif /* HAVE_LIBREADLINE */
116 case 'k':
117 send_kill = 1;
118 break;
119 case 'r':
120 restricted = 1;
121 break;
122 case 's':
123 port = strdup(optarg);
124 colon = strrchr(port, ':');
125 if (colon) {
126 *colon = 0;
127 host = port;
128 port = colon + 1;
129 }
130 break;
131 case 'u':
132 utf8 = eight_bit_clean = 1;
133 break;
134 case 'h':
135 print_usage(argv[0]);
136 exit(0);
137 case 'v':
138 printf("%s\n\n%s", version, legal);
139 exit(0);
140 default:
141 fprintf(stderr, "Try -h for help.\n");
142 exit(1);
143 }
144 }
145
146 ap = argv + optind;
147 if (*ap)
148 country = *ap++;
149 else
150 country = getenv("COUNTRY");
151 if (*ap)
152 passwd = *ap++;
153 else
154 passwd = getenv("PLAYER");
155 if (!port)
156 port = getenv("EMPIREPORT");
157 if (!port)
158 port = empireport;
159 if (!host)
160 host = getenv("EMPIREHOST");
161 if (!host)
162 host = empirehost;
163 uname = getenv("LOGNAME");
164 udir = getenv("HOME");
165 if (!uname || !udir) {
166 struct passwd *pwd;
167
168 pwd = getpwuid(getuid());
169 if (pwd == NULL) {
170 fprintf(stderr, "You don't exist. Go away\n");
171 exit(1);
172 }
173 if (!uname)
174 uname = pwd->pw_name;
175 if (!udir)
176 udir = pwd->pw_dir;
177 }
178 if (*ap) {
179 fprintf(stderr, "%s: extra operand %s\n", argv[0], *ap);
180 fprintf(stderr, "Try -h for help.\n");
181 exit(1);
182 }
183
184 getsose();
185 if (auxfname && (auxfp = fopen(auxfname, "a")) == NULL) {
186 fprintf(stderr, "Unable to open %s for append\n", auxfname);
187 exit(1);
188 }
189
190 sysdep_init();
191
192 sock = tcp_connect(host, port);
193
194 if (!restricted && !history_file)
195 history_file = ".empire_history";
196 if (history_file)
197 history_file = fnameat(history_file, udir);
198
199 if (!login(sock, uname, country, passwd, send_kill, utf8))
200 exit(1);
201
202 if (play(sock, history_file) < 0)
203 exit(1);
204
205 return 0;
206 }
207
208 #ifdef _WIN32
209 /*
210 * Get Windows user name and directory
211 */
212 static struct passwd *
w32_getpw(void)213 w32_getpw(void)
214 {
215 static char unamebuf[128];
216 static char udirbuf[MAX_PATH];
217 static struct passwd pwd;
218 DWORD unamesize;
219
220 unamesize = sizeof(unamebuf);
221 if (GetUserName(unamebuf, &unamesize)) {
222 pwd.pw_name = unamebuf;
223 if (unamesize == 0 || strlen(unamebuf) == 0)
224 pwd.pw_name = "nobody";
225 } else
226 pwd.pw_name = "nobody";
227 if (SUCCEEDED(SHGetFolderPathA(NULL, CSIDL_PROFILE, NULL, 0, udirbuf))
228 && strlen(udirbuf) == 0)
229 pwd.pw_dir = udirbuf;
230 return &pwd;
231 }
232
233 static void
w32_sysdep_init(void)234 w32_sysdep_init(void)
235 {
236 int err;
237
238 /*
239 * stdout is unbuffered under Windows if connected to a character
240 * device, and putchar() screws up when printing multibyte strings
241 * bytewise to an unbuffered stream. Switch stdout to line-
242 * buffered mode. Unfortunately, ISO C allows implementations to
243 * screw that up, and of course Windows does. Manual flushing
244 * after each prompt is required.
245 */
246 setvbuf(stdout, NULL, _IOLBF, 4096);
247
248 err = w32_socket_init();
249 if (err != 0) {
250 fprintf(stderr, "WSAStartup Failed, error code %d\n", err);
251 exit(1);
252 }
253 }
254 #endif
255