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 ((
 177                 netGen[new].fd = accept(gen->fd, (struct sockaddr *)&addr, &addrLen)
 178         ) < 0)
 179                 die("accept");
 180         fprintf(stderr, "+ Connection: %s\n", inet_ntoa(addr.sin_addr));
 181         val2.l_onoff = 1;
 182         val2.l_linger = 0;
 183         setsockopt(netGen[new].fd, SOL_SOCKET, SO_LINGER,(void *)&val2,
 184                 sizeof(val2));
 185         AddEventGen(&netGen[new]);
 186         netGen[new].player = event->u.net.uid = new;
 187         { //new connection; netGen already initialized in GenFunc
 188                 struct hostent *host;
 189
 190                 sprintf(Players[new].host, "%s", inet_ntoa(addr.sin_addr));
 191                 if (addr.sin_family == AF_INET) {
 192                         host = gethostbyaddr(
 193                                 (void *)&addr.sin_addr, sizeof(struct in_addr), AF_INET
 194                         );
 195                         if (host) {
 196                                 strncpy(Players[new].host, host->h_name,
 197                                         sizeof(Players[new].host) - 1);
 198                                 Players[new].host[sizeof(Players[new].host) - 1] = 0;
 199                         } //set player's hostname
 200                 }
 201         } //E_connect
 202         return E_connect;
 203 }
 204
 205 void CountPlayers(void)
 206 { //count number of players/teams
 207         int i, j;
 208         playercount = 0;
 209         for (i = 1; i < MAX_SCREENS; i++) if (Players[i].alive > 0) {
 210                 if (Players[i].team < 128) for (j = 1; j < i; j++) {
 211                         if (Players[j].alive > 0 && Players[j].team == Players[i].team) {
 212                                 playercount--; // player of same team counted before
 213                                 break;
 214                         }
 215                 }
 216                 playercount++;
 217         } //player alive
 218 }
 219
 220 int StartServer(void)
 221 {
 222         MyEvent event;
 223         netint2 currentpiece[MAX_SCREENS];
 224         int playersReady = 0;
 225         int paused = 1;
 226         int i;
 227         char teams[10][7] = {
 228                 "", "Green", "Cyan", "Blue", "Purple",
 229                 "Red", "Grey", "White", "*Orange"
 230         };
 231
 232         do {
 233                 switch (WaitMyEvent(&event, EM_any)) {
 234                         case E_lostConn: //client went away :(
 235                                 Players[event.u.net.sender].alive = -1;
 236                                 for (i = 1; i < MAX_SCREENS; i++)
 237                                         if (Players[i].alive >= 0)
 238                                                 SendPacketTo(i, event.u.net.sender,
 239                                                         NP_part, sizeof(Players[0].alive),
 240                                                         &Players[event.u.net.sender].alive);
 241                                 SCloseNet(event.u.net.sender);
 242                                 break; //NP_endConn
 243                         case E_net:
 244                                 if (verbose) fprintf(stderr, ": %d sent %d\n",
 245                                         netGen[event.u.net.sender].fd, event.u.net.type);
 246                                 switch(event.u.net.type) {
 247                                 case NP_hello:
 248 //                                      if (event.u.net.type != NP_hello) ByeClient(new);
 249                                 {
 250                                         netint4 versiondata[2];
 251                                         char data[255];
 252                                         int major;
 253                                         int protocolVersion;
 254
 255                                         memcpy(versiondata, event.u.net.data, sizeof(versiondata));
 256                                         major = ntoh4(versiondata[0]);
 257                                         protocolVersion = ntoh4(versiondata[1]);
 258                                         if (major != MAJOR_VERSION
 259                                          || protocolVersion != PROTOCOL_VERSION) {
 260                                                 snprintf(data, sizeof(data),
 261                                                         "Version mismatch: received %d.%d",
 262                                                         major, protocolVersion);
 263                                                 fprintf(stderr, "= Wrong version player #%d (%s)\n",
 264                                                         event.u.net.sender, data);
 265                                                 SendPacketTo(event.u.net.sender, 0, NP_error,
 266                                                         strlen(data)+1, data);
 267                                                 SCloseNet(event.u.net.sender);
 268                                         } //version mismatch
 269                                         fprintf(stderr, "* Accepted player #%d\n",
 270                                                 event.u.net.sender);
 271                                         break;
 272                                 } //NP_hello
 273                                 case NP_newPlayer:
 274                                 //receive player details and return other players
 275                                         memcpy(&Players[event.u.net.sender],
 276                                                 event.u.net.data, event.u.net.size);
 277                                         if (Players[event.u.net.sender].team < 1
 278                                          || Players[event.u.net.sender].team > 7) {
 279                                                 int team;
 280
 281                                                 for (team = 1; team < 7; team++) {
 282                                                         for (i = 1; i < MAX_SCREENS; i++)
 283                                                                 if (Players[i].alive > 0 && Players[i].team == team)
 284                                                                         break; //team in use
 285                                                         if (i == MAX_SCREENS) break;
 286                                                 } //find unused team
 287                                                 Players[event.u.net.sender].team = team;
 288                                                 SendPacketTo(event.u.net.sender, event.u.net.sender, NP_team,
 289                                                         sizeof(Players[event.u.net.sender].team),
 290                                                         &Players[event.u.net.sender].team);
 291                                         } //invalid team
 292                                         if (Game.started < 2)
 293                                                 Players[event.u.net.sender].flags |= SCF_paused;
 294                                         if (!Game.continuous && Game.started >= 2) {
 295                                                 char data[40];
 296                                                 strcpy(data, "Can't join: Game has already started");
 297                                                 fprintf(stderr, "- Can't join player #%d in "
 298                                                         "non-continuous game\n", event.u.net.sender);
 299                                                 SendPacketTo(event.u.net.sender, 0, NP_error,
 300                                                         strlen(data)+1, data);
 301 //                                              SCloseNet(event.u.net.sender, 0);
 302                                                 break;
 303                                         } //can't join started game
 304                                         {
 305                                                 static struct {
 306                                                         int playerflags;
 307                                                         int gravity;    //1
 308                                                         int started;    //2
 309                                                         int continuous; //3
 310                                                         long seed;              //4
 311                                                         int initspeed;  //5
 312                                                 } data;
 313
 314                                                 memcpy(&data, &Players[event.u.net.sender].flags,
 315                                                         sizeof(data.playerflags));
 316                                                 memcpy(&data.gravity, &Game,
 317                                                         sizeof(data) - sizeof(data.playerflags));
 318                                                 SendPacketTo(event.u.net.sender, 0, NP_gamedata,
 319                                                         sizeof(data), &data);
 320                                         } //send game options
 321                                         for (i = 1; i < MAX_SCREENS; i++)
 322                                                 if (netGen[i].fd >= 0 && i != event.u.net.sender) {
 323                                                         SendPacketTo(event.u.net.sender, i, NP_newPlayer,
 324                                                                 sizeof(_Player), &Players[i]);
 325                                                         SendPacketTo(event.u.net.sender, i, NP_newPiece,
 326                                                                 sizeof(Players[i].curShape), &Players[i].curShape);
 327                                                         SendPacketTo(i, event.u.net.sender, NP_newPlayer,
 328                                                                 sizeof(_Player), &Players[event.u.net.sender]);
 329                                                 } //send (to) players
 330                                         fprintf(stderr, "> Joined player #%d: %s <%s> (%s)\n",
 331                                                 event.u.net.sender,
 332                                                 Players[event.u.net.sender].name,
 333                                                 Players[event.u.net.sender].host,
 334                                                 teams[Players[event.u.net.sender].team]);
 335                                         if (++playersReady >= minplayers) {
 336                                                 if (Game.started > 1)
 337                                                         SendPacketTo(event.u.net.sender, 0,
 338                                                                 NP_start, 0, NULL);
 339 /*                                              else {
 340                                                         fprintf(stderr, "* Starting game (%010d)\n",
 341                                                                 Game.seed);
 342                                                         for (i = 1; i < MAX_SCREENS; i++)
 343                                                                 SendPacketTo(i, 0, NP_start, 0, NULL);
 344                                                         Game.started++;
 345                                                 } //first goahead (to all)*/
 346                                         } //give go ahead
 347                                         break; //NP_playerdata
 348                                 case NP_newPiece:
 349                                         memcpy(&Players[event.u.net.sender].curShape,
 350                                                 event.u.net.data, sizeof(Players[0].curShape));
 351                                         goto sendtoall;
 352                                 case NP_argghhh:
 353                                         Players[event.u.net.sender].alive = 0;
 354                                         fprintf(stderr, "< Player #%d died\n",
 355                                                 event.u.net.sender);
 356                                         //check for unpaused game
 357                                 case NP_pause:
 358                                 {
 359                                         Players[event.u.net.sender].flags ^= SCF_paused;
 360                                         paused = Game.started < 1;
 361                                         for (i = 1; i < MAX_SCREENS; i++)
 362                                                 if (Players[i].alive > 0)
 363                                                         paused |= Players[i].flags & SCF_paused;
 364                                         fprintf(stderr, "* Player #%d (un)paused (pause=%d)\n",
 365                                                 event.u.net.sender, paused);
 366                                         if (paused) paused = 1;
 367                                         goto sendtoall;
 368                                 } //NP_pause
 369                                 default: //relay data to all players
 370                                 sendtoall:
 371 //                                      if (event.u.net.type >= NP_pause)
 372                                         if (event.u.net.type >= NP_rotright
 373                                          && Game.started < 2)
 374                                                 break;
 375                                         for (i = 1; i < MAX_SCREENS; i++)
 376                                                 if (i != event.u.net.sender)
 377                                                 if (event.u.net.type != NP_giveJunk
 378                                                  || Players[i].team != Players[event.u.net.sender].team)
 379                                                         SendPacketTo(i, event.u.net.sender,
 380                                                                 event.u.net.type, event.u.net.size,
 381                                                                 event.u.net.data);
 382                                         break; //>=NP_paused
 383                                 }
 384                                 break; //E_net
 385                         case E_connect:
 386                         { //new connection; netGen already initialized in GenFunc
 387                                 char serverdata[255];
 388
 389                                 sprintf(serverdata, "Netris server %s", version_string);
 390                                 SendPacketTo(event.u.net.uid, event.u.net.uid, NP_hello,
 391                                         strlen(serverdata)+1, serverdata);
 392                                 break;
 393                         } //E_connect
 394                 } //event
 395                 CountPlayers();
 396                 if (Game.started < 1) {
 397                         if (playercount >= 2) {
 398                                 fprintf(stderr, "* Game (%010d) ready to start\n", Game.seed);
 399                                 Game.started++;
 400                         } //give goahead
 401                 } //game not yet started
 402                 else {
 403                         if (playercount < 2) {
 404                                 fprintf(stderr, "* Stopping game\n");
 405                                 if (Game.seed) Game.seed++;
 406                                 if (Game.started > 1) for (i = 1; i < MAX_SCREENS; i++) {
 407                                         if (Players[i].alive >= 0) {
 408                                                 Players[i].alive = 1;
 409                                                 Players[i].flags |= SCF_paused;
 410                                                 SendPacketTo(i, 0, NP_stop,
 411                                                         sizeof(Game.seed), &Game.seed);
 412                                         } //transmit game stop and set players not ready
 413                                 }
 414                                 paused = 1;
 415                                 Game.started = 0;
 416                         } //too few players for game
 417                         if (Game.started == 1 && !paused) {
 418                                 Game.started++;
 419                                 fprintf(stderr, "* Game starts\n");
 420                                 for (i = 1; i < MAX_SCREENS; i++)
 421                                         if (Players[i].alive > 0)
 422                                                 SendPacketTo(i, 0, NP_start, 0, NULL);
 423                         } //everybody ready to start
 424                 } //game (ready to) start(ed)
 425         } while (1);
 426         fprintf(stderr, "* Exiting server\n");
 427 }
 428
 429
 430 void SHeader(void)
 431 {
 432         fprintf(stderr,
 433                 "NETRIS Server %s\t(c) 2002 Shiar <shiar@shiar.org>\n\n",
 434                 version_string);
 435 }
 436
 437 void SUsage(void)
 438 {
 439         SHeader();
 440         fprintf(stderr,
 441                 "Usage: netris <options>\n"
 442                 "\n"
 443                 "  -h, --help\t\tPrint this usage information\n"
 444                 "  -H, --info\t\tShow distribution and warranty information\n"
 445                 "\n"
 446                 "  -p, --port %d\tSet port number\n"
 447                 "\n"
 448                 "  -s, --seed <seed>\tStart with given random seed\n"
 449                 "  -i, --speed %.2f\tSet the initial step-down interval, in seconds\n"
 450                 "  -m, --min-players %d\tNumber of players required before starting the game\n"
 451                 "  -x, --max-players %d\tMaximum number of players allowed in the game\n"
 452                 "  -c, --continuous\tDon'n quit the game\n"
 453                 "\n",
 454                 DEFAULT_PORT, DEFAULT_INTERVAL/1e6, minplayers, maxplayers
 455         );
 456 }
 457
 458 void HandleOption(char tag, char *value)
 459 {
 460         switch (tag) {
 461         case 'v':  //verbose
 462                 verbose = 1;
 463                 break;
 464         case 'p':  //port
 465                 port = atoi(value);
 466                 break;
 467         case 'c':  //min-players
 468                 Game.continuous = atoi(value);
 469                 break;
 470         case 'm':  //min-players
 471                 minplayers = atoi(value);
 472                 break;
 473         case 'x':  //max-players
 474                 maxplayers = MIN(atoi(value), MAX_SCREENS);
 475                 break;
 476         case 'q':  //quadra-style gravity
 477                 Game.gravity ^= 1;
 478                 break;
 479         case 'i':  //speed (of level 1)
 480                 Game.initspeed = atof(value) * 1e6;
 481                 break;
 482         case 's':  //seed
 483                 Game.seed = atoi(value);
 484                 break;
 485         case 'H':  //info
 486                 SHeader();
 487                 DistInfo();
 488                 exit(0);
 489         case 'h':  //help
 490                 SUsage();
 491                 exit(0);
 492         default:
 493                 break;
 494         }
 495 }
 496
 497 void ReadConf(char *filename)
 498 {
 499         FILE *file_in;
 500         char buf[513];
 501         int i;
 502         char *ch;
 503         char tag[81], value[81];
 504
 505         file_in = fopen(filename, "r");
 506         if (file_in) {
 507                 while (fgets(buf, 512, file_in) != NULL) {
 508                         if ((ch = strchr(buf, '#')))
 509                                 *ch = '\0'; // truncate string from # char
 510                         for (i = strlen(buf)-1; i >= 0; i--)
 511                                 if (buf[i] == ' '  || buf[i] == '\t'
 512                                  || buf[i] == '\n' || 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 }
 531
 532 void CatchInt(int sig)
 533 {
 534         siglongjmp(close_env, 1);
 535 }
 536
 537 int main(int argc, char **argv)
 538 {
 539         char ch;
 540
 541         if (sigsetjmp(close_env, 1)) exit(0);
 542         signal(SIGINT, CatchInt);
 543         port = DEFAULT_PORT;
 544         maxplayers = 8;
 545         Game.initspeed = DEFAULT_INTERVAL;
 546         Game.seed = time(0);
 547         Game.gravity = 0;
 548         {
 549                 int i;
 550
 551                 for (i = 1; i < MAX_SCREENS; i++)
 552                         Players[i].alive = -1;
 553         }
 554
 555 //      if (getopt(argc, argv, "f:") == 'f')
 556 //              ReadConf(optarg);
 557 //      else
 558         ReadConf(CONFIG_FILE);
 559         while ((ch = getopt_long(
 560                 argc, argv, "hHvqp:i:s:c:m:x:", options, NULL
 561         )) != -1)
 562                 HandleOption(ch, optarg);
 563         if (optind < argc) {
 564                 SUsage();
 565                 exit(1);
 566         }
 567 //      WriteConf();
 568
 569         SHeader();
 570
 571         {
 572                 int i;
 573
 574                 for (i = 1; i < MAX_SCREENS; i++)
 575                         memcpy(&netGen[i], &netGen[0], sizeof(EventGenRec));
 576         } //setup netGen var
 577
 578         AtExit(CloseNets);
 579
 580         {
 581                 int val1;
 582
 583                 memset(&addr, 0, sizeof(addr));
 584                 addr.sin_family = AF_INET;
 585                 addr.sin_port = htons(port);
 586                 addr.sin_addr.s_addr = htonl(INADDR_ANY);
 587                 if ((connGen.fd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
 588                         die("socket");
 589                 val1 = 1;
 590                 setsockopt(connGen.fd, SOL_SOCKET, SO_REUSEADDR,
 591                         (void *)&val1, sizeof(val1));
 592                 if (bind(connGen.fd, (struct sockaddr *)&addr, sizeof(addr)) < 0)
 593                         die("bind");
 594                 if (listen(connGen.fd, 1) < 0)
 595                         die("listen");
 596
 597                 AddEventGen(&connGen);
 598         } //setup listening sock
 599
 600         StartServer(); //server loop
 601
 602         return 0;
 603 }
 604