c33a541d8877351fa9549c01b0ced5512dd8ac10
[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 <ctype.h>
25 #include <string.h>
26 #include <sys/types.h>
27 #include <netinet/in.h>
28 #include <unistd.h>
29 #include <sys/socket.h>
30 #include <netdb.h>
31 #include <errno.h>
32 #include <setjmp.h>
33
34 #include "util.h"
35
36 #define HEADER_SIZE sizeof(netint4[3])
37
38 static struct option options[] = {
39         { "wait",        0, 0, 'w' },
40         { "port",        1, 0, 'p' },
41         { "quadra",      1, 0, 'q' },
42         { "min-players", 1, 0, 'm' },
43         { "max-players", 1, 0, 'x' },
44         { "continuous",  1, 0, 'c' },
45         { "speed",       1, 0, 'i' },
46         { "seed",        1, 0, 's' },
47         { "verbose",     0, 0, 'v' },
48         { "info",        0, 0, 'H' },
49         { "help",        0, 0, 'h' },
50         { 0,             0, 0,  0  }
51 };
52
53 static char minplayers = 2;
54 static char maxplayers = 8;
55 static char playercount;
56 static char verbose = 0;
57
58 struct sockaddr_in addr;
59
60 MyEventType NetGenFunc(EventGenRec *gen, MyEvent *event);
61 static EventGenRec netGen[MAX_SCREENS] = {
62         { NULL, 0, FT_read, -1, NetGenFunc, EM_net, 0, "\0", 0, HEADER_SIZE }
63 };
64
65 static MyEventType AlarmGenFunc(EventGenRec *gen, MyEvent *event);
66 static EventGenRec alarmGen = {
67         &alarmGen, 0, FT_read, -1, AlarmGenFunc, EM_alarm
68 };
69
70 static MyEventType ConnGenFunc(EventGenRec *gen, MyEvent *event);
71 static EventGenRec connGen = {
72         NULL, 0, FT_read, -1, ConnGenFunc, EM_connect
73 };
74
75 static EventGenRec *nextGen = &alarmGen;
76
77 static sigjmp_buf close_env;
78
79
80 void SendPacketTo(short playa, short uid, NetPacketType type, int size, void *data)
81 { //send to someone
82         netint4 header[3];
83
84         if (netGen[playa].fd >= 0) {
85                 if (verbose)
86                         fprintf(stderr, ": send %d from %d to %d\n", type, uid, playa);
87                 header[0] = hton4(uid);
88                 header[1] = hton4(type);
89                 header[2] = hton4(size + HEADER_SIZE);
90                 if (MyWrite(netGen[playa].fd, header, HEADER_SIZE) != HEADER_SIZE)
91                         die("write (header)");
92                 if (size > 0 && data && MyWrite(netGen[playa].fd, data, size) != size)
93                         die("write");
94         }
95 }
96
97 static MyEventType AlarmGenFunc(EventGenRec *gen, MyEvent *event)
98 {
99         return E_alarm;
100 }
101
102 void SCloseNet(short playa)
103 { //kick some connection's ass!
104         MyEvent event;
105
106         if (netGen[playa].fd >= 0) {
107                 if (Players[playa].alive >= 0) {
108                         SendPacketTo(playa, 0, NP_endConn, 0, NULL);
109                         do {} while (WaitMyEvent(&event, EM_net) != E_lostConn);
110                 } //say bye to player
111                 close(netGen[playa].fd);
112                 netGen[playa].fd = -1;
113         }
114         if (netGen[playa].next)
115                 RemoveEventGen(&netGen[playa]);
116 }
117
118 void CloseNets(void)
119 { //nou oogjes dicht en snaveltjes toe
120         int i;
121
122         fprintf(stderr, "- Closing connections...\n");
123         for (i = 1; i < MAX_SCREENS; i++)
124                 SCloseNet(i); //bye everybuddy
125         fprintf(stderr, "* All Done\n\n");
126 }
127
128 MyEventType NetGenFunc(EventGenRec *gen, MyEvent *event)
129 { //receive
130         int result;
131         short uid, type, size;
132         netint4 *data = (netint4*)&(gen->buf);
133
134         result = MyRead(gen->fd, gen->buf + gen->bufSize,
135                 gen->bufGoal - gen->bufSize);
136         if (result < 0) {
137                 fprintf(stderr, "- Closed connection to player #%d\n", gen->player);
138                 return E_lostConn;
139         }
140         gen->bufSize += result;
141         if (gen->bufSize < gen->bufGoal)
142                 return E_none;
143         // *ugly* memcpy(data, gen->buf, sizeof(data));
144         uid = ntoh4(data[0]);
145         type = ntoh4(data[1]);
146         size = ntoh4(data[2]);
147         gen->bufGoal = size;
148         if (gen->bufSize < gen->bufGoal)
149                 return E_none;
150         gen->bufSize = 0;
151         gen->bufGoal = HEADER_SIZE;
152         event->u.net.sender = gen->player;
153         event->u.net.uid = uid;
154         event->u.net.type = type;
155         event->u.net.size = size - HEADER_SIZE;
156         event->u.net.data = gen->buf + HEADER_SIZE;
157         if (type == NP_endConn) {
158                 fprintf(stderr, "- Quit player #%d\n", gen->player);
159                 return E_lostConn;
160         } //client sent quit signal
161         return E_net;
162 }
163
164
165 static MyEventType ConnGenFunc(EventGenRec *gen, MyEvent *event)
166 {
167         int addrLen;
168         struct linger val2;
169         int new;
170
171         addrLen = sizeof(addr);
172         for (new = 1; new <= MAX_SCREENS; new++)
173                 if (netGen[new].fd < 0) break;
174         if (new > maxplayers) return;
175
176         if ((netGen[new].fd =
177                 accept(gen->fd, (struct sockaddr *)&addr, &addrLen)) < 0)
178                         die("accept");
179         fprintf(stderr, "+ Connection: %s\n", inet_ntoa(addr.sin_addr));
180         val2.l_onoff = 1;
181         val2.l_linger = 0;
182         setsockopt(netGen[new].fd, SOL_SOCKET, SO_LINGER,(void *)&val2,
183                 sizeof(val2));
184         AddEventGen(&netGen[new]);
185         netGen[new].player = event->u.net.uid = new;
186                         { //new connection; netGen already initialized in GenFunc
187                                 struct hostent *host;
188
189                                 sprintf(Players[new].host, "%s", inet_ntoa(addr.sin_addr));
190                                 if (addr.sin_family == AF_INET) {
191                                         host = gethostbyaddr((void *)&addr.sin_addr,
192                                                              sizeof(struct in_addr), AF_INET);
193                                         if (host) {
194                                                 strncpy(Players[new].host, host->h_name,
195                                                         sizeof(Players[new].host) - 1);
196                                                 Players[new].host[sizeof(Players[new].host) - 1] = 0;
197                                         } //set player's hostname
198                                 }
199                         } //E_connect
200         return E_connect;
201 }
202
203 void CountPlayers(void)
204 { //count number of players/teams
205         int i, j;
206         playercount = 0;
207         for (i = 1; i < MAX_SCREENS; i++) if (Players[i].alive > 0) {
208                 if (Players[i].team < 128) for (j = 1; j < i; j++)
209                         if (Players[j].alive > 0 && (Players[j].team == Players[i].team)) {
210                                 playercount--;
211                                 break;
212                         } //player of same team counted before
213                 playercount++;
214         } //player alive
215 }
216
217 int StartServer(void)
218 {
219         MyEvent event;
220         netint2 currentpiece[MAX_SCREENS];
221         int playersReady = 0;
222         int paused = 1;
223         int i;
224         char teams[10][7] = {
225                 "", "Green", "Cyan", "Blue", "Purple",
226                 "Red", "Grey", "White", "*Orange"
227         };
228
229         do {
230                 switch (WaitMyEvent(&event, EM_any)) {
231                         case E_lostConn: //client went away :(
232                                 Players[event.u.net.sender].alive = -1;
233                                 for (i = 1; i < MAX_SCREENS; i++)
234                                         if (Players[i].alive >= 0)
235                                                 SendPacketTo(i, event.u.net.sender,
236                                                         NP_part, sizeof(Players[0].alive),
237                                                         &Players[event.u.net.sender].alive);
238                                 SCloseNet(event.u.net.sender);
239                                 break; //NP_endConn
240                         case E_net:
241                                 if (verbose) fprintf(stderr, ": %d sent %d\n",
242                                         netGen[event.u.net.sender].fd, event.u.net.type);
243                                 switch(event.u.net.type) {
244                                         case NP_hello:
245 //                                              if (event.u.net.type != NP_hello) ByeClient(new);
246                                         {
247                                                 netint4 versiondata[2];
248                                                 char data[255];
249                                                 int major;
250                                                 int protocolVersion;
251
252                                                 memcpy(versiondata, event.u.net.data,
253                                                         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, "= Wrong version player #%d (%s)\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, "* Accepted player #%d\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, "- Can't join player #%d in "
296                                                                 "non-continuous game\n", 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,
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,
322                                                                         NP_newPlayer, 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, "> Joined player #%d: %s <%s> (%s)\n",
329                                                         event.u.net.sender,
330                                                         Players[event.u.net.sender].name,
331                                                         Players[event.u.net.sender].host,
332                                                         teams[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, "< Player #%d died\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;
362                                                 fprintf(stderr, "* Player #%d (un)paused (pause=%d)\n",
363                                                         event.u.net.sender, paused);
364                                                 if (paused) paused = 1;
365                                                 goto sendtoall;
366                                         } //NP_pause
367                                         default: //relay data to all players
368                                         sendtoall:
369 //                                              if (event.u.net.type >= NP_pause)
370                                                 if (event.u.net.type >= NP_rotright
371                                                  && Game.started < 2)
372                                                         break;
373                                                 for (i = 1; i < MAX_SCREENS; i++)
374                                                         if (i != event.u.net.sender)
375                                                         if (event.u.net.type != NP_giveJunk ||
376                                                         Players[i].team != Players[event.u.net.sender].team)
377                                                                 SendPacketTo(i, event.u.net.sender,
378                                                                         event.u.net.type, event.u.net.size,
379                                                                         event.u.net.data);
380                                                 break; //>=NP_paused
381                                 }
382                                 break; //E_net
383                         case E_connect:
384                         { //new connection; netGen already initialized in GenFunc
385                                 char serverdata[255];
386
387                                 sprintf(serverdata, "Netris server %s", version_string);
388                                 SendPacketTo(event.u.net.uid, event.u.net.uid, NP_hello,
389                                         strlen(serverdata)+1, serverdata);
390                                 break;
391                         } //E_connect
392                 } //event
393                 CountPlayers();
394                 if (Game.started < 1) {
395                         if (playercount > 1) {
396                                 fprintf(stderr, "* Game (%010d) ready to start\n", Game.seed);
397                                 Game.started++;
398                         } //give goahead
399                 } //game not yet started
400                 else {
401                         if (playercount < 2) {
402                                 fprintf(stderr, "* Stopping game\n");
403                                 if (Game.seed) Game.seed++;
404                                 if (Game.started > 1) for (i = 1; i < MAX_SCREENS; i++)
405                                         if (Players[i].alive >= 0) {
406                                                 Players[i].alive = 1;
407                                                 Players[i].flags |= SCF_paused;
408                                                 SendPacketTo(i, 0, NP_stop,
409                                                         sizeof(Game.seed), &Game.seed);
410                                         } //transmit game stop and set players not ready
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, "* Game starts\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,
430                 "NETRIS Server %s\t(c) 2002 Shiar <shiar@shiar.org>\n\n",
431                 version_string);
432 }
433
434 void SUsage(void)
435 {
436         SHeader();
437         fprintf(stderr,
438                 "Usage: netris <options>\n"
439                 "\n"
440                 "  -h, --help\t\tPrint this usage information\n"
441                 "  -H, --info\t\tShow distribution and warranty information\n"
442                 "\n"
443                 "  -p, --port <port>\tSet port number (default is %d)\n"
444                 "\n"
445                 "  -s, --seed <seed>\tStart with given random seed\n"
446                 "  -i, --speed <sec>\tSet the initial step-down interval, in seconds\n"
447                 "  -m, --min-players <2>\tNumber of players required before starting the game\n"
448                 "  -x, --max-players <8>\tMaximum number of players allowed in the game\n"
449                 "  -c, --continuous\tDon'n quit the game\n"
450                 "\n", DEFAULT_PORT);
451 }
452
453 void WriteConf(void)
454 {
455         FILE *file_out;
456
457         file_out = fopen(CONFIG_FILE, "w");
458         if (file_out == NULL) {
459                 perror("Error writing config file");
460                 exit(1);
461         }
462
463         fprintf(file_out, "### NETRIS %s Config file ###\n\n", version_string);
464
465         fclose(file_out);
466         fprintf(stderr, "Wrote new game configuration to %s\n", CONFIG_FILE);
467 }
468
469 void HandleOption(char tag, char *value)
470 {
471         switch (tag) {
472                 case 'v':  //verbose
473                         verbose = 1;
474                         break;
475                 case 'p':  //port
476                         port = atoi(value);
477                         break;
478                 case 'c':  //min-players
479                         Game.continuous = atoi(value);
480                         break;
481                 case 'm':  //min-players
482                         minplayers = atoi(value);
483                         break;
484                 case 'x':  //max-players
485                         maxplayers = atoi(value);
486                         if (maxplayers >= MAX_SCREENS)
487                                 maxplayers = MAX_SCREENS;
488                         break;
489                 case 'q':  //quadra-style gravity
490                         Game.gravity ^= 1;
491                         break;
492                 case 'i':  //speed (of level 1)
493                         Game.initspeed = atof(value) * 1e6;
494                         break;
495                 case 's':  //seed
496                         Game.seed = atoi(value);
497                         break;
498                 case 'H':  //info
499                         SHeader();
500                         DistInfo(); exit(0);
501                 case 'h':  //help
502                         SUsage(); exit(0);
503                 default:
504                         break;
505         }
506 }
507
508 void ReadConf(char *filename)
509 {
510         FILE *file_in;
511         char buf[513];
512         int i;
513         char *ch;
514         char tag[81], value[81];
515
516         file_in = fopen(filename, "r");
517         if (file_in) {
518                 while (fgets(buf, 512, file_in) != NULL) {
519                         if ((ch = strchr(buf, '#')))
520                                 *ch = '\0'; // truncate string from # char
521                         for (i = strlen(buf)-1; i >= 0; i--)
522                                 if (buf[i] == ' '  || buf[i] == '\t'
523                                  || buf[i] == '\n' || buf[i] == 13)
524                                         buf[i] = '\0';
525                                 else break;
526
527                         sscanf(buf, "%80[^= \t] = %80[^\n]", tag, value);
528                         for (i = 0; options[i].name; i++){
529                                 if (!strcasecmp(options[i].name, tag)) {
530                                         HandleOption(options[i].val, value);
531                                         break;
532                                 }
533                         }
534                 }
535                 fclose(file_in);
536         } //read file
537         else {
538                 fprintf(stderr, "Unable to open config file %s.\n", filename);
539         } //defaults
540
541 }
542
543 void CatchInt(int sig)
544 {
545         siglongjmp(close_env, 1);
546 }
547
548 int main(int argc, char **argv)
549 {
550         char ch;
551
552         if (sigsetjmp(close_env, 1)) exit(0);
553         signal(SIGINT, CatchInt);
554         port = DEFAULT_PORT;
555         maxplayers = 8;
556         Game.initspeed = DEFAULT_INTERVAL;
557         Game.seed = time(0);
558         Game.gravity = 0;
559         {
560                 int i;
561
562                 for (i = 1; i < MAX_SCREENS; i++)
563                         Players[i].alive = -1;
564         }
565
566 //      if (getopt(argc, argv, "f:") == 'f')
567 //              ReadConf(optarg);
568 //      else
569         ReadConf(CONFIG_FILE);
570         while ((ch = getopt_long(
571                 argc, argv, "hHvqp:i:s:c:m:x:", options, NULL
572         )) != -1)
573                 HandleOption(ch, optarg);
574         if (optind < argc) {
575                 SUsage();
576                 exit(1);
577         }
578 //      WriteConf();
579
580         SHeader();
581
582         {
583                 int i;
584
585                 for (i = 1; i < MAX_SCREENS; i++)
586                         memcpy(&netGen[i], &netGen[0], sizeof(EventGenRec));
587         } //setup netGen var
588
589         AtExit(CloseNets);
590
591         {
592                 int val1;
593
594                 memset(&addr, 0, sizeof(addr));
595                 addr.sin_family = AF_INET;
596                 addr.sin_port = htons(port);
597                 addr.sin_addr.s_addr = htonl(INADDR_ANY);
598                 if ((connGen.fd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
599                         die("socket");
600                 val1 = 1;
601                 setsockopt(connGen.fd, SOL_SOCKET, SO_REUSEADDR,
602                         (void *)&val1, sizeof(val1));
603                 if (bind(connGen.fd, (struct sockaddr *)&addr, sizeof(addr)) < 0)
604                         die("bind");
605                 if (listen(connGen.fd, 1) < 0)
606                         die("listen");
607
608                 AddEventGen(&connGen);
609         } //setup listening sock
610
611         StartServer(); //server loop
612
613         return 0;
614 }
615