team names defined in global include
[netris.git] / server.c
1 /*
2  * Netris -- A free networked version of T*tris
3  * Copyright (C) 1994,1995,1996  Mark H. Weaver <mhw@netris.org>
4  * 
5  * This program is free software; you can redistribute it and/or
6  * modify it under the terms of the GNU General Public License
7  * as published by the Free Software Foundation; either version 2
8  * of the License, or (at your option) any later version.
9  * 
10  * This program is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  * GNU General Public License for more details.
14  * 
15  * You should have received a copy of the GNU General Public License
16  * along with this program; if not, write to the Free Software
17  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
18  */
19
20 #define NOEXT
21 #include "netris.h"
22
23 #include <stdlib.h>
24 #include <stdbool.h>
25 #include <ctype.h>
26 #include <string.h>
27 #include <sys/types.h>
28 #include <netinet/in.h>
29 #include <unistd.h>
30 #include <sys/socket.h>
31 #include <netdb.h>
32 #include <errno.h>
33 #include <setjmp.h>
34
35 #include "util.h"
36 #include "msg.h"
37
38 #define HEADER_SIZE sizeof(netint4[3])
39
40 static struct option options[] = {
41         { "wait",        0, 0, 'w' },
42         { "port",        1, 0, 'p' },
43         { "quadra",      1, 0, 'q' },
44         { "min-players", 1, 0, 'm' },
45         { "max-players", 1, 0, 'x' },
46         { "continuous",  1, 0, 'c' },
47         { "speed",       1, 0, 'i' },
48         { "seed",        1, 0, 's' },
49         { "verbose",     0, 0, 'v' },
50         { "info",        0, 0, 'H' },
51         { "help",        0, 0, 'h' },
52         { 0,             0, 0,  0  }
53 };
54
55 static char minplayers = 2;
56 static char maxplayers = 8;
57 static char playercount;
58 static char verbose = 0;
59
60 struct sockaddr_in addr;
61
62 MyEventType NetGenFunc(EventGenRec *gen, MyEvent *event);
63 static EventGenRec netGen[MAX_SCREENS] = {
64         { NULL, 0, FT_read, -1, NetGenFunc, EM_net, 0, "\0", 0, HEADER_SIZE }
65 };
66
67 static MyEventType AlarmGenFunc(EventGenRec *gen, MyEvent *event);
68 static EventGenRec alarmGen = {
69         &alarmGen, 0, FT_read, -1, AlarmGenFunc, EM_alarm
70 };
71
72 static MyEventType ConnGenFunc(EventGenRec *gen, MyEvent *event);
73 static EventGenRec connGen = {
74         NULL, 0, FT_read, -1, ConnGenFunc, EM_connect
75 };
76
77 static EventGenRec *nextGen = &alarmGen;
78
79 static sigjmp_buf close_env;
80
81
82 void SendPacketTo(short playa, short uid, NetPacketType type, int size, void *data)
83 { //send to someone
84         netint4 header[3];
85
86         if (netGen[playa].fd >= 0) {
87                 if (verbose)
88                         fprintf(stderr, MSG_SERVER_DBG_SEND "\n", type, uid, playa);
89                 header[0] = hton4(uid);
90                 header[1] = hton4(type);
91                 header[2] = hton4(size + HEADER_SIZE);
92                 if (MyWrite(netGen[playa].fd, header, HEADER_SIZE) != HEADER_SIZE)
93                         die("write (header)");
94                 if (size > 0 && data && MyWrite(netGen[playa].fd, data, size) != size)
95                         die("write");
96         }
97 }
98
99 static MyEventType AlarmGenFunc(EventGenRec *gen, MyEvent *event)
100 {
101         return E_alarm;
102 }
103
104 void SCloseNet(short playa)
105 { //kick some connection's ass!
106         MyEvent event;
107
108         if (netGen[playa].fd >= 0) {
109                 if (Players[playa].alive >= 0) {
110                         SendPacketTo(playa, 0, NP_endConn, 0, NULL);
111                         do {} while (WaitMyEvent(&event, EM_net) != E_lostConn);
112                 } //say bye to player
113                 close(netGen[playa].fd);
114                 netGen[playa].fd = -1;
115         }
116         if (netGen[playa].next)
117                 RemoveEventGen(&netGen[playa]);
118 }
119
120 void CloseNets(void)
121 { //nou oogjes dicht en snaveltjes toe
122         int i;
123
124         fprintf(stderr, MSG_SERVER_CLOSE_ALL "\n");
125         for (i = 1; i < MAX_SCREENS; i++)
126                 SCloseNet(i); // bye everybuddy
127         fprintf(stderr, MSG_SERVER_CLOSE_ALL_DONE "\n\n");
128 }
129
130 MyEventType NetGenFunc(EventGenRec *gen, MyEvent *event)
131 { //receive
132         int result;
133         short uid, type, size;
134         netint4 *data = (netint4*)&(gen->buf);
135
136         result = MyRead(gen->fd, gen->buf + gen->bufSize,
137                 gen->bufGoal - gen->bufSize);
138         if (result < 0) {
139                 fprintf(stderr, MSG_SERVER_PLAYER_CLOSED "\n", gen->player);
140                 return E_lostConn;
141         }
142         gen->bufSize += result;
143         if (gen->bufSize < gen->bufGoal)
144                 return E_none;
145         // *ugly* memcpy(data, gen->buf, sizeof(data));
146         uid  = ntoh4(data[0]);
147         type = ntoh4(data[1]);
148         size = ntoh4(data[2]);
149         gen->bufGoal = size;
150         if (gen->bufSize < gen->bufGoal)
151                 return E_none;
152         gen->bufSize = 0;
153         gen->bufGoal = HEADER_SIZE;
154         event->u.net.sender = gen->player;
155         event->u.net.uid = uid;
156         event->u.net.type = type;
157         event->u.net.size = size - HEADER_SIZE;
158         event->u.net.data = gen->buf + HEADER_SIZE;
159         if (type == NP_endConn) {
160                 fprintf(stderr, MSG_SERVER_PLAYER_QUIT "\n", gen->player);
161                 return E_lostConn;
162         } //client sent quit signal
163         return E_net;
164 }
165
166
167 static MyEventType ConnGenFunc(EventGenRec *gen, MyEvent *event)
168 {
169         int addrLen;
170         struct linger val2;
171         int new;
172
173         addrLen = sizeof(addr);
174         for (new = 1; new <= MAX_SCREENS; new++)
175                 if (netGen[new].fd < 0) break;
176         if (new > maxplayers) return;
177
178         if ((
179                 netGen[new].fd = accept(gen->fd, (struct sockaddr *)&addr, &addrLen)
180         ) < 0)
181                 die("accept");
182         fprintf(stderr, MSG_SERVER_CONNECT "\n", inet_ntoa(addr.sin_addr));
183         val2.l_onoff = 1;
184         val2.l_linger = 0;
185         setsockopt(netGen[new].fd, SOL_SOCKET, SO_LINGER,(void *)&val2,
186                 sizeof(val2));
187         AddEventGen(&netGen[new]);
188         netGen[new].player = event->u.net.uid = new;
189         { //new connection; netGen already initialized in GenFunc
190                 struct hostent *host;
191
192                 sprintf(Players[new].host, "%s", inet_ntoa(addr.sin_addr));
193                 if (addr.sin_family == AF_INET) {
194                         host = gethostbyaddr(
195                                 (void *)&addr.sin_addr, sizeof(struct in_addr), AF_INET
196                         );
197                         if (host) {
198                                 strncpy(Players[new].host, host->h_name,
199                                         sizeof(Players[new].host) - 1);
200                                 Players[new].host[sizeof(Players[new].host) - 1] = 0;
201                         } //set player's hostname
202                 }
203         } //E_connect
204         return E_connect;
205 }
206
207 void CountPlayers(void)
208 { //count number of players/teams
209         int i, j;
210         playercount = 0;
211         for (i = 1; i < MAX_SCREENS; i++) if (Players[i].alive > 0) {
212                 if (Players[i].team < 128) for (j = 1; j < i; j++) {
213                         if (Players[j].alive > 0 && Players[j].team == Players[i].team) {
214                                 playercount--; // player of same team counted before
215                                 break;
216                         }
217                 }
218                 playercount++;
219         } //player alive
220 }
221
222 int StartServer(void)
223 {
224         MyEvent event;
225         netint2 currentpiece[MAX_SCREENS];
226         int playersReady = 0;
227         bool paused = 1;
228         int i;
229
230         do {
231                 switch (WaitMyEvent(&event, EM_any)) {
232                         case E_lostConn: //client went away :(
233                                 Players[event.u.net.sender].alive = -1;
234                                 for (i = 1; i < MAX_SCREENS; i++)
235                                         if (Players[i].alive >= 0)
236                                                 SendPacketTo(i, event.u.net.sender,
237                                                         NP_part, sizeof(Players[0].alive),
238                                                         &Players[event.u.net.sender].alive);
239                                 SCloseNet(event.u.net.sender);
240                                 break; //NP_endConn
241                         case E_net:
242                                 if (verbose) fprintf(stderr, MSG_SERVER_DBG_RECV "\n",
243                                         netGen[event.u.net.sender].fd, event.u.net.type);
244                                 switch(event.u.net.type) {
245                                 case NP_hello:
246 //                                      if (event.u.net.type != NP_hello) ByeClient(new);
247                                 {
248                                         netint4 versiondata[2];
249                                         char data[255];
250                                         int major;
251                                         int protocolVersion;
252
253                                         memcpy(versiondata, event.u.net.data, sizeof(versiondata));
254                                         major = ntoh4(versiondata[0]);
255                                         protocolVersion = ntoh4(versiondata[1]);
256                                         if (major != MAJOR_VERSION
257                                          || protocolVersion != PROTOCOL_VERSION) {
258                                                 snprintf(data, sizeof(data),
259                                                         "Version mismatch: received %d.%d",
260                                                         major, protocolVersion);
261                                                 fprintf(stderr, MSG_SERVER_PLAYER_EVERSION "\n",
262                                                         event.u.net.sender, data);
263                                                 SendPacketTo(event.u.net.sender, 0, NP_error,
264                                                         strlen(data)+1, data);
265                                                 SCloseNet(event.u.net.sender);
266                                         } //version mismatch
267                                         fprintf(stderr, MSG_SERVER_PLAYER_ACCEPT "\n",
268                                                 event.u.net.sender);
269                                         break;
270                                 } //NP_hello
271                                 case NP_newPlayer:
272                                 //receive player details and return other players
273                                         memcpy(&Players[event.u.net.sender],
274                                                 event.u.net.data, event.u.net.size);
275                                         if (Players[event.u.net.sender].team < 1
276                                          || Players[event.u.net.sender].team > 7) {
277                                                 int team;
278
279                                                 for (team = 1; team < 7; team++) {
280                                                         for (i = 1; i < MAX_SCREENS; i++)
281                                                                 if (Players[i].alive > 0 && Players[i].team == team)
282                                                                         break; //team in use
283                                                         if (i == MAX_SCREENS) break;
284                                                 } //find unused team
285                                                 Players[event.u.net.sender].team = team;
286                                                 SendPacketTo(event.u.net.sender, event.u.net.sender, NP_team,
287                                                         sizeof(Players[event.u.net.sender].team),
288                                                         &Players[event.u.net.sender].team);
289                                         } //invalid team
290                                         if (Game.started < 2)
291                                                 Players[event.u.net.sender].flags |= SCF_paused;
292                                         if (!Game.continuous && Game.started >= 2) {
293                                                 char data[40];
294                                                 strcpy(data, "Can't join: Game has already started");
295                                                 fprintf(stderr, MSG_SERVER_PLAYER_JOIN_ESTARTED,
296                                                         event.u.net.sender);
297                                                 SendPacketTo(event.u.net.sender, 0, NP_error,
298                                                         strlen(data)+1, data);
299 //                                              SCloseNet(event.u.net.sender, 0);
300                                                 break;
301                                         } //can't join started game
302                                         {
303                                                 static struct {
304                                                         int playerflags;
305                                                         int gravity;    //1
306                                                         int started;    //2
307                                                         int continuous; //3
308                                                         long seed;              //4
309                                                         int initspeed;  //5
310                                                 } data;
311
312                                                 memcpy(&data, &Players[event.u.net.sender].flags,
313                                                         sizeof(data.playerflags));
314                                                 memcpy(&data.gravity, &Game.gravity,
315                                                         sizeof(data) - sizeof(data.playerflags));
316                                                 SendPacketTo(event.u.net.sender, 0, NP_gamedata,
317                                                         sizeof(data), &data);
318                                         } //send game options
319                                         for (i = 1; i < MAX_SCREENS; i++)
320                                                 if (netGen[i].fd >= 0 && i != event.u.net.sender) {
321                                                         SendPacketTo(event.u.net.sender, i, NP_newPlayer,
322                                                                 sizeof(_Player), &Players[i]);
323                                                         SendPacketTo(event.u.net.sender, i, NP_newPiece,
324                                                                 sizeof(Players[i].curShape), &Players[i].curShape);
325                                                         SendPacketTo(i, event.u.net.sender, NP_newPlayer,
326                                                                 sizeof(_Player), &Players[event.u.net.sender]);
327                                                 } //send (to) players
328                                         fprintf(stderr, MSG_SERVER_PLAYER_JOIN "\n",
329                                                 event.u.net.sender,
330                                                 Players[event.u.net.sender].name,
331                                                 Players[event.u.net.sender].host,
332                                                 teamname[Players[event.u.net.sender].team]);
333                                         if (++playersReady >= minplayers) {
334                                                 if (Game.started > 1)
335                                                         SendPacketTo(event.u.net.sender, 0,
336                                                                 NP_start, 0, NULL);
337 /*                                              else {
338                                                         fprintf(stderr, "* Starting game (%010d)\n",
339                                                                 Game.seed);
340                                                         for (i = 1; i < MAX_SCREENS; i++)
341                                                                 SendPacketTo(i, 0, NP_start, 0, NULL);
342                                                         Game.started++;
343                                                 } //first goahead (to all)*/
344                                         } //give go ahead
345                                         break; //NP_playerdata
346                                 case NP_newPiece:
347                                         memcpy(&Players[event.u.net.sender].curShape,
348                                                 event.u.net.data, sizeof(Players[0].curShape));
349                                         goto sendtoall;
350                                 case NP_argghhh:
351                                         Players[event.u.net.sender].alive = 0;
352                                         fprintf(stderr, MSG_SERVER_PLAYER_DIE "\n",
353                                                 event.u.net.sender);
354                                         //check for unpaused game
355                                 case NP_pause:
356                                 {
357                                         Players[event.u.net.sender].flags ^= SCF_paused;
358                                         paused = Game.started < 1;
359                                         for (i = 1; i < MAX_SCREENS; i++)
360                                                 if (Players[i].alive > 0)
361                                                         paused |= (Players[i].flags & SCF_paused) != 0;
362                                         fprintf(stderr, MSG_SERVER_PLAYER_PAUSE "\n",
363                                                 event.u.net.sender, paused);
364                                         goto sendtoall;
365                                 } //NP_pause
366                                 default: //relay data to all players
367                                 sendtoall:
368 //                                      if (event.u.net.type >= NP_pause)
369                                         if (event.u.net.type >= NP_rotright
370                                          && Game.started < 2)
371                                                 break;
372                                         for (i = 1; i < MAX_SCREENS; i++)
373                                                 if (i != event.u.net.sender)
374                                                 if (event.u.net.type != NP_giveJunk
375                                                  || Players[i].team != Players[event.u.net.sender].team)
376                                                         SendPacketTo(i, event.u.net.sender,
377                                                                 event.u.net.type, event.u.net.size,
378                                                                 event.u.net.data);
379                                         break; //>=NP_paused
380                                 }
381                                 break; //E_net
382                         case E_connect:
383                         { //new connection; netGen already initialized in GenFunc
384                                 char serverdata[255];
385
386                                 sprintf(serverdata, "Netris server %s", version_string);
387                                 SendPacketTo(event.u.net.uid, event.u.net.uid, NP_hello,
388                                         strlen(serverdata)+1, serverdata);
389                                 break;
390                         } //E_connect
391                 } //event
392                 CountPlayers();
393                 if (Game.started < 1) {
394                         if (playercount >= 2) {
395                                 fprintf(stderr, MSG_SERVER_GAME_READY "\n", Game.seed);
396                                 Game.started++;
397                         } //give goahead
398                 } //game not yet started
399                 else {
400                         if (playercount < 2) {
401                                 fprintf(stderr, MSG_SERVER_GAME_STOP "\n");
402                                 if (Game.seed) Game.seed++;
403                                 if (Game.started > 1) for (i = 1; i < MAX_SCREENS; i++) {
404                                         if (Players[i].alive >= 0) {
405                                                 Players[i].alive = 1;
406                                                 Players[i].flags |= SCF_paused;
407                                                 SendPacketTo(i, 0, NP_stop,
408                                                         sizeof(Game.seed), &Game.seed);
409                                         } //transmit game stop and set players not ready
410                                 }
411                                 paused = 1;
412                                 Game.started = 0;
413                         } //too few players for game
414                         if (Game.started == 1 && !paused) {
415                                 Game.started++;
416                                 fprintf(stderr, MSG_SERVER_GAME_START "\n");
417                                 for (i = 1; i < MAX_SCREENS; i++)
418                                         if (Players[i].alive > 0)
419                                                 SendPacketTo(i, 0, NP_start, 0, NULL);
420                         } //everybody ready to start
421                 } //game (ready to) start(ed)
422         } while (1);
423         fprintf(stderr, "* Exiting server\n");
424 }
425
426
427 void SHeader(void)
428 {
429         fprintf(stderr, MSG_SERVER_TITLE "\n\n", version_string);
430 }
431
432 void SUsage(void)
433 {
434         SHeader();
435         fprintf(stderr,
436                 "Usage: netris <options>\n"
437                 "\n"
438                 "  -h, --help\t\tPrint this usage information\n"
439                 "  -H, --info\t\tShow distribution and warranty information\n"
440                 "\n"
441                 "  -p, --port %d\tSet port number\n"
442                 "\n"
443                 "  -s, --seed <seed>\tStart with given random seed\n"
444                 "  -i, --speed %.2f\tSet the initial step-down interval, in seconds\n"
445                 "  -m, --min-players %d\tNumber of players required before starting the game\n"
446                 "  -x, --max-players %d\tMaximum number of players allowed in the game\n"
447                 "  -c, --continuous\tDon'n quit the game\n"
448                 "\n",
449                 DEFAULT_PORT, DEFAULT_INTERVAL/1e6, minplayers, maxplayers
450         );
451 }
452
453 void HandleOption(char tag, char *value)
454 {
455         switch (tag) {
456         case 'v':  //verbose
457                 verbose = 1;
458                 break;
459         case 'p':  //port
460                 port = atoi(value);
461                 break;
462         case 'c':  //min-players
463                 Game.continuous = atoi(value);
464                 break;
465         case 'm':  //min-players
466                 minplayers = atoi(value);
467                 break;
468         case 'x':  //max-players
469                 maxplayers = MIN(atoi(value), MAX_SCREENS);
470                 break;
471         case 'q':  //quadra-style gravity
472                 Game.gravity ^= 1;
473                 break;
474         case 'i':  //speed (of level 1)
475                 Game.initspeed = atof(value) * 1e6;
476                 break;
477         case 's':  //seed
478                 Game.seed = atoi(value);
479                 break;
480         case 'H':  //info
481                 SHeader();
482                 DistInfo();
483                 exit(0);
484         case 'h':  //help
485                 SUsage();
486                 exit(0);
487         default:
488                 break;
489         }
490 }
491
492 void ReadConf(char *filename)
493 {
494         FILE *file_in;
495         char buf[513];
496         int i;
497         char *ch;
498         char tag[81], value[81];
499
500         file_in = fopen(filename, "r");
501         if (file_in) {
502                 while (fgets(buf, 512, file_in) != NULL) {
503                         if ((ch = strchr(buf, '#')))
504                                 *ch = '\0'; // truncate string from # char
505                         for (i = strlen(buf)-1; i >= 0; i--)
506                                 if (buf[i] == ' '  || buf[i] == '\t'
507                                  || buf[i] == '\n' || buf[i] == 13)
508                                         buf[i] = '\0';
509                                 else break;
510
511                         sscanf(buf, "%80[^= \t] = %80[^\n]", tag, value);
512                         for (i = 0; options[i].name; i++){
513                                 if (!strcasecmp(options[i].name, tag)) {
514                                         HandleOption(options[i].val, value);
515                                         break;
516                                 }
517                         }
518                 }
519                 fclose(file_in);
520         } //read file
521         else {
522                 fprintf(stderr, "Unable to open config file %s.\n", filename);
523         } //defaults
524
525 }
526
527 void CatchInt(int sig)
528 {
529         siglongjmp(close_env, 1);
530 }
531
532 int main(int argc, char **argv)
533 {
534         char ch;
535
536         if (sigsetjmp(close_env, 1)) exit(0);
537         signal(SIGINT, CatchInt);
538         port = DEFAULT_PORT;
539         maxplayers = 8;
540         Game.initspeed = DEFAULT_INTERVAL;
541         Game.seed = time(0);
542         Game.gravity = 0;
543         {
544                 int i;
545
546                 for (i = 1; i < MAX_SCREENS; i++)
547                         Players[i].alive = -1;
548         }
549
550 //      if (getopt(argc, argv, "f:") == 'f')
551 //              ReadConf(optarg);
552 //      else
553         ReadConf(CONFIG_FILE);
554         while ((ch = getopt_long(
555                 argc, argv, "hHvqp:i:s:c:m:x:", options, NULL
556         )) != -1)
557                 HandleOption(ch, optarg);
558         if (optind < argc) {
559                 SUsage();
560                 exit(1);
561         }
562 //      WriteConf();
563
564         SHeader();
565
566         {
567                 int i;
568
569                 for (i = 1; i < MAX_SCREENS; i++)
570                         memcpy(&netGen[i], &netGen[0], sizeof(EventGenRec));
571         } //setup netGen var
572
573         AtExit(CloseNets);
574
575         {
576                 int val1;
577
578                 memset(&addr, 0, sizeof(addr));
579                 addr.sin_family = AF_INET;
580                 addr.sin_port = htons(port);
581                 addr.sin_addr.s_addr = htonl(INADDR_ANY);
582                 if ((connGen.fd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
583                         die("socket");
584                 val1 = 1;
585                 setsockopt(connGen.fd, SOL_SOCKET, SO_REUSEADDR,
586                         (void *)&val1, sizeof(val1));
587                 if (bind(connGen.fd, (struct sockaddr *)&addr, sizeof(addr)) < 0)
588                         die("bind");
589                 if (listen(connGen.fd, 1) < 0)
590                         die("listen");
591
592                 AddEventGen(&connGen);
593         } //setup listening sock
594
595         StartServer(); //server loop
596
597         return 0;
598 }
599