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  * $Id: game.c,v 1.39 1999/05/16 06:56:27 mhw Exp $
  20  */
  21
  22 #define NOEXT
  23 #include "netris.h"
  24 #include <stdlib.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 #define HEADER_SIZE sizeof(netint4[3])
  36 #define MAX_CONNECTIONS 3
  37
  38 char *version_string = "0.5.89";
  39
  40 static struct option options[] = {
  41         { "wait",               0, 0, 'w' },
  42         { "port",               1, 0, 'p' },
  43         { "speed",              1, 0, 'i' },
  44         { "seed",               1, 0, 's' },
  45         { "info",               0, 0, 'H' },
  46         { "help",               0, 0, 'h' },
  47         { 0,                    0, 0,  0 }
  48 };
  49
  50 static char *gameNames[GT_len] = { "OnePlayer", "ClassicTwo" };
  51
  52 ExtFunc MyEventType NetGenFunc(EventGenRec *gen, MyEvent *event);
  53
  54 EventGenRec netGen[MAX_SCREENS] = {
  55         { NULL, 0, FT_read, -1, NetGenFunc, EM_net, 0, "\0", 0, HEADER_SIZE } };
  56
  57 static MyEventType AlarmGenFunc(EventGenRec *gen, MyEvent *event);
  58 static EventGenRec alarmGen =
  59                 { &alarmGen, 0, FT_read, -1, AlarmGenFunc, EM_alarm };
  60 static EventGenRec *nextGen = &alarmGen;
  61
  62 static sigjmp_buf close_env;
  63
  64 ExtFunc volatile void die(char *msg)
  65 {
  66         perror(msg);
  67         exit(1);
  68 } //die
  69
  70 ExtFunc int MyRead(int fd, void *data, int len)
  71 {
  72         int result, left;
  73
  74         left = len;
  75         while (left > 0) {
  76                 result = read(fd, data, left);
  77                 if (result > 0) {
  78                         data = ((char *)data) + result;
  79                         left -= result;
  80                 }
  81                 else if (errno != EINTR)
  82                         return result;
  83         }
  84         return len;
  85 } //MyRead
  86
  87 ExtFunc int MyWrite(int fd, void *data, int len)
  88 {
  89         int result, left;
  90
  91         left = len;
  92         while (left > 0) {
  93                 result = write(fd, data, left);
  94                 if (result > 0) {
  95                         data = ((char *)data) + result;
  96                         left -= result;
  97                 }
  98                 else if (errno != EINTR)
  99                         return result;
 100         }
 101         return len;
 102 } //MyWrite
 103
 104 static MyEventType AlarmGenFunc(EventGenRec *gen, MyEvent *event)
 105 {
 106         return E_alarm;
 107 }
 108
 109 ExtFunc void AddEventGen(EventGenRec *gen)
 110 {
 111         assert(gen->next == NULL);
 112         assert(nextGen->next != (void*)0xffffffff);
 113         gen->next = nextGen->next;
 114         nextGen->next = gen;
 115 } //AddEventGen
 116
 117 ExtFunc void RemoveEventGen(EventGenRec *gen)
 118 {
 119         // assert(gen->next != NULL);   /* Be more forgiving, for SIGINTs */
 120         if (gen->next) {
 121                 while (nextGen->next != gen)
 122                         nextGen = nextGen->next;
 123                 nextGen->next = gen->next;
 124                 gen->next = NULL;
 125         }
 126 } //RemoveEventGen
 127
 128 ExtFunc MyEventType WaitMyEvent(MyEvent *event, int mask)
 129 { //poll
 130         int i, retry = 0;
 131         fd_set fds[FT_len];
 132         EventGenRec *gen;
 133         int result, anyReady, anySet;
 134         struct timeval tv;
 135
 136         for (;;) {
 137                 for (i = 0; i < FT_len; ++i)
 138                         FD_ZERO(&fds[i]);
 139                 anyReady = anySet = 0;
 140                 gen = nextGen;
 141                 do {
 142                         if (gen->mask & mask) {
 143                                 if (gen->ready)
 144                                         anyReady = 1;
 145                                 if (gen->fd >= 0) {
 146                                         FD_SET(gen->fd, &fds[gen->fdType]);
 147                                         anySet = 1;
 148                                 }
 149                         }
 150                         gen = gen->next;
 151                 } while (gen != nextGen);
 152                 if (anySet) {
 153                         tv.tv_sec = 0;
 154                         tv.tv_usec = (retry && !anyReady) ? 500000 : 0;
 155                         result = select(FD_SETSIZE, &fds[FT_read], &fds[FT_write],
 156                                         &fds[FT_except], anyReady ? &tv : NULL);
 157                 }
 158                 else {
 159                         if (retry && !anyReady)
 160                                 sleep(1);
 161                         result = 0;
 162                 }
 163                 gen = nextGen;
 164                 do {
 165                         if ((gen->mask & mask)
 166                                         && (gen->ready || (result > 0 && gen->fd >= 0
 167                                                 && FD_ISSET(gen->fd, &fds[gen->fdType])))) {
 168                                 gen->ready = 0;
 169                                 event->type = gen->func(gen, event);
 170                                 if (event->type != E_none) {
 171                                         nextGen = gen->next;
 172                                         return event->type;
 173                                 }
 174                         }
 175                         gen = gen->next;
 176                 } while (gen != nextGen);
 177                 retry = 1;
 178         }
 179 } //WaitMyEvent
 180
 181 ExtFunc void ByeClient(int playa)
 182 { //client went away
 183         fprintf(stderr, "Close connection #%d\n", playa);
 184         close(netGen[playa].fd);
 185         netGen[playa].fd = -1;
 186 } //ByeClient
 187
 188 ExtFunc MyEventType NetGenFunc(EventGenRec *gen, MyEvent *event)
 189 { //receive
 190         int result;
 191         short uid, type, size;
 192         netint4 *data = (netint4*)&(gen->buf);
 193
 194         result = MyRead(gen->fd, gen->buf + gen->bufSize,
 195                 gen->bufGoal - gen->bufSize);
 196         if (result < 0) {
 197                 ByeClient(gen->player);
 198                 type = NP_endConn;
 199                 return E_net;
 200         }
 201         gen->bufSize += result;
 202         if (gen->bufSize < gen->bufGoal)
 203                 return E_none;
 204         // *ugly* memcpy(data, gen->buf, sizeof(data));
 205         uid = ntoh4(data[0]);
 206         type = ntoh4(data[1]);
 207         size = ntoh4(data[2]);
 208         gen->bufGoal = size;
 209         if (gen->bufSize < gen->bufGoal)
 210                 return E_none;
 211         gen->bufSize = 0;
 212         gen->bufGoal = HEADER_SIZE;
 213         event->u.net.sender = gen->player;
 214         event->u.net.uid = uid;
 215         event->u.net.type = type;
 216         event->u.net.size = size - HEADER_SIZE;
 217         event->u.net.data = gen->buf + HEADER_SIZE;
 218         if (type == NP_endConn)
 219                 ByeClient(gen->player);
 220         return E_net;
 221 } //NetGenFunc
 222
 223 ExtFunc void SendPacketTo(short playa, short uid, NetPacketType type, int size, void *data)
 224 { //send to someone
 225         netint4 header[3];
 226
 227         if (netGen[playa].fd >= 0) {
 228                 header[0] = hton4(uid);
 229                 header[1] = hton4(type);
 230                 header[2] = hton4(size + HEADER_SIZE);
 231                 if (MyWrite(netGen[playa].fd, header, HEADER_SIZE) != HEADER_SIZE)
 232                         die("write (header)");
 233                 if (size > 0 && data && MyWrite(netGen[playa].fd, data, size) != size)
 234                         die("write");
 235         }
 236 } //SendPacketTo
 237
 238
 239 ExtFunc void AtExit(void (*handler)(void))
 240 {
 241 #ifdef HAS_ON_EXIT
 242         on_exit((void *)handler, NULL);
 243 #else
 244         atexit(handler);
 245 #endif
 246 }
 247
 248 ExtFunc void SCloseNet(short playa)
 249 { //kick some connection's ass!
 250         MyEvent event;
 251
 252         if (netGen[playa].fd >= 0) {
 253                 SendPacketTo(playa, 0, NP_endConn, 0, NULL);
 254                 do{} while (WaitMyEvent(&event, EM_net) != E_lostConn);
 255                 close(netGen[playa].fd);
 256                 netGen[playa].fd = -1;
 257         }
 258         if (netGen[playa].next)
 259                 RemoveEventGen(&netGen[playa]);
 260 } //SCloseNet
 261
 262 ExtFunc void CloseNets(void)
 263 { //nou oogjes dicht en snaveltjes toe
 264         int i;
 265
 266         for (i = 1; i <= totalPlayers; i++)
 267                 SCloseNet(i);
 268 } //CloseNets
 269
 270 ExtFunc int WaitForConnection(short port)
 271 {
 272         struct sockaddr_in addr;
 273         struct hostent *host;
 274         int sockListen;
 275         int addrLen;
 276         int val1;
 277         struct linger val2;
 278         int i;
 279
 280         AtExit(CloseNets);
 281         for (i = 1; i < MAX_SCREENS; i++)
 282                 memcpy(&netGen[i], &netGen[0], sizeof(EventGenRec));
 283
 284         memset(&addr, 0, sizeof(addr));
 285         addr.sin_family = AF_INET;
 286         addr.sin_port = htons(port);
 287         addr.sin_addr.s_addr = htonl(INADDR_ANY);
 288         if ((sockListen = socket(AF_INET, SOCK_STREAM, 0)) < 0)
 289                 die("socket");
 290         val1 = 1;
 291         setsockopt(sockListen, SOL_SOCKET, SO_REUSEADDR,
 292                         (void *)&val1, sizeof(val1));
 293         if (bind(sockListen, (struct sockaddr *)&addr, sizeof(addr)) < 0)
 294                 die("bind");
 295         if (listen(sockListen, 1) < 0)
 296                 die("listen");
 297
 298         addrLen = sizeof(addr);
 299         for (i = 1; i <= MAX_CONNECTIONS; i++) {
 300                 if ((netGen[i].fd = accept(sockListen, (struct sockaddr *)&addr, &addrLen)) < 0)
 301                         die("accept");
 302                 fprintf(stderr, "Connection: %s\n", inet_ntoa(addr.sin_addr));
 303                 val2.l_onoff = 1;
 304                 val2.l_linger = 0;
 305                 setsockopt(netGen[i].fd, SOL_SOCKET, SO_LINGER, (void *)&val2, sizeof(val2));
 306                 sprintf(Players[i].host, "%s", inet_ntoa(addr.sin_addr));
 307                 if (addr.sin_family == AF_INET) {
 308                         host = gethostbyaddr((void *)&addr.sin_addr,
 309                                         sizeof(struct in_addr), AF_INET);
 310                         if (host) {
 311                                 strncpy(Players[i].host, host->h_name, sizeof(Players[i].host) - 1);
 312                                 Players[i].host[sizeof(Players[i].host) - 1] = 0;
 313                         }
 314                 }
 315                 AddEventGen(&netGen[i]);
 316                 netGen[i].player = i;
 317                 totalPlayers++;
 318                 {
 319                         MyEvent event;
 320
 321                         do {} while ((WaitMyEvent(&event, EM_net) != E_net) ||
 322                                         (event.u.net.sender != i));
 323                         if (event.u.net.type != NP_hello) ByeClient(i);
 324                         else {
 325                                 netint4 versiondata[2];
 326                                 char data[255];
 327                                 int major;
 328                                 memcpy(versiondata, event.u.net.data, sizeof(versiondata));
 329                                 major = ntoh4(versiondata[0]);
 330                                 protocolVersion = ntoh4(versiondata[1]);
 331                                 if (major != MAJOR_VERSION || protocolVersion != PROTOCOL_VERSION) {
 332                                         snprintf(data, sizeof(data), "Version mismatch: received %d.%d",
 333                                                 major, protocolVersion);
 334                                         fprintf(stderr, "Byebye client #%d (%s)\n",
 335                                                 event.u.net.sender, data);
 336                                         SendPacketTo(event.u.net.sender, 0, NP_error, strlen(data)+1, data);
 337                                         SCloseNet(event.u.net.sender);
 338                                 } //version mismatch
 339                                 fprintf(stderr, "Accepted #%d\n", event.u.net.sender);
 340                         } //NP_hello
 341                 }
 342         }
 343         close(sockListen);
 344         return 0;
 345 } //WaitForConnection
 346
 347 ExtFunc int StartServer(char *portStr)
 348 {
 349         MyEvent event;
 350         char serverdata[255];
 351         int playercount;
 352         int i;
 353
 354         {
 355                 short port;
 356
 357                 if (portStr)
 358                         port = atoi(portStr);   /* XXX Error checking */
 359                 else
 360                         port = DEFAULT_PORT;
 361                 WaitForConnection(port);
 362         }
 363
 364         playercount = MAX_CONNECTIONS;
 365
 366         for (i = 1; i <= playercount; i++) {
 367                 sprintf(serverdata, "Netris server %s", version_string);
 368                 SendPacketTo(i, i, NP_hello, strlen(serverdata)+1, serverdata);
 369         }
 370
 371         while(1) {
 372                 if (WaitMyEvent(&event, EM_net) == E_net) {
 373 //                      fprintf(stderr, "in %d: %d\n",
 374 //                              netGen[event.u.net.sender].fd, event.u.net.type);
 375                         switch(event.u.net.type) {
 376                                 case NP_endConn:
 377                                 { //client went away :(
 378                                         //tell the others! :)
 379                                         break;
 380                                 } //NP_endConn
 381                                 case NP_hello:
 382                                         break;
 383                                 case NP_newPlayer:
 384                                 { //receive player details and return other players
 385                                         memcpy(&Players[event.u.net.sender],
 386                                                 event.u.net.data, event.u.net.size);
 387                                         fprintf(stderr, "player %d: %s <%s>\n", event.u.net.sender,
 388                                                 event.u.net.data, //Players[event.u.net.sender].name
 389                                                 Players[event.u.net.sender].host);
 390                                         for (i = 1; i <= totalPlayers; i++)
 391                                                 if (i != event.u.net.sender)
 392                                                         SendPacketTo(i, event.u.net.sender, event.u.net.type,
 393                                                                 sizeof(Player) - sizeof(Players[0].spy),
 394                                                                 &Players[event.u.net.sender]);
 395                                         if (--playercount == 0) {
 396                                                 fprintf(stderr, "Starting game\n");
 397                                                 for (i = 1; i <= totalPlayers; i++)
 398                                                         SendPacketTo(i, 0, NP_goAhead, 0, NULL);
 399                                                 playercount++;
 400                                         } //give go ahead
 401                                         break;
 402                                 } //NP_playerdata
 403                                 default:
 404                                 { //relay data to all players
 405         //                              if (event.u.net.type >= NP_pause)
 406                                         for (i = 1; i <= totalPlayers; i++)
 407                                                 if (i != event.u.net.sender)
 408                                                         SendPacketTo(i, event.u.net.sender, event.u.net.type,
 409                                                                 event.u.net.size, event.u.net.data);
 410                                         break;
 411                                 }
 412                         } //E_net
 413                 }
 414         } //loop
 415 } //StartServer
 416
 417
 418 ExtFunc void Header(void)
 419 {
 420         fprintf(stderr,
 421           "NETRIS Server %s\t(c) 2002 Shiar <shiar@shiar.org>\n\n",
 422           version_string);
 423 } //Header
 424
 425 ExtFunc void Usage(void)
 426 {
 427         Header();
 428         fprintf(stderr,
 429           "Usage: netris <options>\n"
 430           "\n"
 431           "  -h, --help\t\tPrint this usage information\n"
 432           "  -H, --info\t\tShow distribution and warranty information\n"
 433           "\n"
 434           "  -p, --port <port>\tSet port number (default is %d)\n"
 435           "\n"
 436           "  -s, --seed <seed>\tStart with given random seed\n"
 437           "  -i, --speed <sec>\tSet the initial step-down interval, in seconds\n"
 438           "\n", DEFAULT_PORT);
 439 }
 440
 441 ExtFunc void DistInfo(void)
 442 {
 443         Header();
 444         fprintf(stderr,
 445           "This program is free software; you can redistribute it and/or modify\n"
 446           "it under the terms of the GNU General Public License as published by\n"
 447           "the Free Software Foundation; either version 2 of the License, or\n"
 448           "(at your option) any later version.\n"
 449           "\n"
 450           "This program is distributed in the hope that it will be useful,\n"
 451           "but WITHOUT ANY WARRANTY; without even the implied warranty of\n"
 452           "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\n"
 453           "GNU General Public License for more details.\n"
 454           "\n"
 455           "You should have received a copy of the GNU General Public License\n"
 456           "along with this program; if not, write to the Free Software\n"
 457           "Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.\n"
 458           "\n");
 459 } //DistInfo
 460
 461 ExtFunc void WriteConf(void)
 462 {
 463         FILE *file_out;
 464
 465         file_out = fopen(CONFIG_FILE, "w");
 466         if (file_out == NULL) {
 467                 perror("Error writing config file");
 468                 exit(1);
 469         }
 470
 471         fprintf(file_out, "### NETRIS %s Config file ###\n\n", version_string);
 472
 473         fclose(file_out);
 474         fprintf(stderr, "Wrote new game configuration to %s\n", CONFIG_FILE);
 475 } //WriteConf
 476
 477 ExtFunc void HandleOption(char tag, char *value)
 478 {
 479         switch (tag) {
 480                 case 'p':       //port
 481                         portStr = value; break;
 482                 case 'i':       //speed (of level 1)
 483                         Game.initspeed = atof(value) * 1e6;
 484                         break;
 485                 case 's':       //seed
 486                         Game.seed = atoi(value);
 487                         Players[0].flags |= SCF_setSeed;
 488                         break;
 489                 case 'H':       //info
 490                         DistInfo(); exit(0);
 491                 case 'h':       //help
 492                         Usage(); exit(0);
 493                 default:
 494                         break;
 495         }
 496 } //HandleParam
 497
 498 ExtFunc void ReadConf(char *filename)
 499 {
 500         FILE *file_in;
 501         char buf[513];
 502         int i;
 503         char *ch;
 504         char tag[81], value[81];
 505
 506         file_in = fopen(filename, "r");
 507         if (file_in) {
 508                 while (fgets(buf, 512, file_in) != NULL) {
 509                         if ((ch = strchr(buf, '#')))
 510                                 *ch = '\0'; // truncate string from # char
 511                         for (i = strlen(buf)-1; i >= 0; i--)
 512                                 if (buf[i] == ' ' || buf[i] == '\n' || buf[i] == '\t' || buf[i] == 13)
 513                                         buf[i] = '\0';
 514                                 else break;
 515
 516                         sscanf(buf, "%80[^= \t] = %80[^\n]", tag, value);
 517                         for (i = 0; options[i].name; i++){
 518                                 if (!strcasecmp(options[i].name, tag)) {
 519                                         HandleOption(options[i].val, value);
 520                                         break;
 521                                 }
 522                         }
 523                 }
 524                 fclose(file_in);
 525         } //read file
 526         else {
 527                 fprintf(stderr, "Unable to open config file %s.\n", filename);
 528         } //defaults
 529
 530 } //ReadConf
 531
 532 ExtFunc void CatchInt(int sig)
 533 {
 534         siglongjmp(close_env, 1);
 535 }
 536
 537 ExtFunc int main(int argc, char **argv)
 538 {
 539         char ch;
 540
 541         if (sigsetjmp(close_env, 1)) exit(0);
 542         signal(SIGINT, CatchInt);
 543         Game.standout = Game.color = 1;
 544         Game.initspeed = DEFAULT_INTERVAL;
 545
 546 //      if (getopt(argc, argv, "f:") == 'f')
 547 //              ReadConf(optarg);
 548 //      else
 549         ReadConf(CONFIG_FILE);
 550         while ((ch = getopt_long(argc, argv,
 551                         "hHp:i:s:", options, NULL)) != -1)
 552                 HandleOption(ch, optarg);
 553         if (optind < argc) {
 554                 Usage();
 555                 exit(1);
 556         }
 557 //      WriteConf();
 558
 559         Header();
 560         StartServer(portStr);
 561         return 0;
 562 }
 563
 564 /*
 565  * vi: ts=4 ai
 566  * vim: noai si
 567  */