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 static 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 static void SendPacketTo(short player, short uid, NetPacketType type, int size, void *data)
  83 { //send to someone
  84         netint4 header[3];
  85
  86         if (netGen[player].fd >= 0) {
  87                 if (verbose)
  88                         fprintf(stderr, MSG_SERVER_DBG_SEND "\n", type, uid, player);
  89                 header[0] = hton4(uid);
  90                 header[1] = hton4(type);
  91                 header[2] = hton4(size + HEADER_SIZE);
  92                 if (MyWrite(netGen[player].fd, header, HEADER_SIZE) != HEADER_SIZE)
  93                         die("write (header)");
  94                 if (size > 0 && data && MyWrite(netGen[player].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 static void SCloseNet(short player)
 105 { //kick some connection's ass!
 106         MyEvent event;
 107
 108         if (netGen[player].fd >= 0) {
 109                 if (Players[player].alive >= 0) {
 110                         SendPacketTo(player, 0, NP_endConn, 0, NULL);
 111                         do {} while (WaitMyEvent(&event, EM_net) != E_lostConn);
 112                 } //say bye to player
 113                 close(netGen[player].fd);
 114                 netGen[player].fd = -1;
 115         }
 116         if (netGen[player].next)
 117                 RemoveEventGen(&netGen[player]);
 118 }
 119
 120 static 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 static 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 static 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 static 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                                                         float shapes[7];
 310                                                         int initspeed;  //5
 311                                                 } data;
 312
 313                                                 memcpy(&data, &Players[event.u.net.sender].flags,
 314                                                         sizeof(data.playerflags));
 315                                                 memcpy(&data.gravity, &Game.gravity,
 316                                                         sizeof(data) - sizeof(data.playerflags));
 317                                                 SendPacketTo(event.u.net.sender, 0, NP_gamedata,
 318                                                         sizeof(data), &data);
 319                                         } //send game options
 320                                         for (i = 1; i < MAX_SCREENS; i++)
 321                                                 if (netGen[i].fd >= 0 && i != event.u.net.sender) {
 322                                                         SendPacketTo(event.u.net.sender, i, NP_newPlayer,
 323                                                                 sizeof(player_t), &Players[i]);
 324                                                         SendPacketTo(event.u.net.sender, i, NP_newPiece,
 325                                                                 sizeof(Players[i].curShape), &Players[i].curShape);
 326                                                         SendPacketTo(i, event.u.net.sender, NP_newPlayer,
 327                                                                 sizeof(player_t), &Players[event.u.net.sender]);
 328                                                 } //send (to) players
 329                                         fprintf(stderr, MSG_SERVER_PLAYER_JOIN "\n",
 330                                                 event.u.net.sender,
 331                                                 Players[event.u.net.sender].name,
 332                                                 Players[event.u.net.sender].host,
 333                                                 teamname[Players[event.u.net.sender].team]);
 334                                         if (++playersReady >= minplayers) {
 335                                                 if (Game.started > 1)
 336                                                         SendPacketTo(event.u.net.sender, 0,
 337                                                                 NP_start, 0, NULL);
 338 /*                                              else {
 339                                                         fprintf(stderr, "* Starting game (%010d)\n",
 340                                                                 Game.seed);
 341                                                         for (i = 1; i < MAX_SCREENS; i++)
 342                                                                 SendPacketTo(i, 0, NP_start, 0, NULL);
 343                                                         Game.started++;
 344                                                 } //first goahead (to all)*/
 345                                         } //give go ahead
 346                                         break; //NP_playerdata
 347                                 case NP_newPiece:
 348                                         memcpy(&Players[event.u.net.sender].curShape,
 349                                                 event.u.net.data, sizeof(Players[0].curShape));
 350                                         goto sendtoall;
 351                                 case NP_argghhh:
 352                                         Players[event.u.net.sender].alive = 0;
 353                                         fprintf(stderr, MSG_SERVER_PLAYER_DIE "\n",
 354                                                 event.u.net.sender);
 355                                         //check for unpaused game
 356                                 case NP_pause:
 357                                 {
 358                                         Players[event.u.net.sender].flags ^= SCF_paused;
 359                                         paused = Game.started < 1;
 360                                         for (i = 1; i < MAX_SCREENS; i++)
 361                                                 if (Players[i].alive > 0)
 362                                                         paused |= (Players[i].flags & SCF_paused) != 0;
 363                                         fprintf(stderr, MSG_SERVER_PLAYER_PAUSE "\n",
 364                                                 event.u.net.sender, paused);
 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 >= 2) {
 396                                 fprintf(stderr, MSG_SERVER_GAME_READY "\n", Game.seed);
 397                                 Game.started++;
 398                         } //give goahead
 399                 } //game not yet started
 400                 else {
 401                         if (playercount < 2) {
 402                                 fprintf(stderr, MSG_SERVER_GAME_STOP "\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                                 }
 412                                 paused = 1;
 413                                 Game.started = 0;
 414                         } //too few players for game
 415                         if (Game.started == 1 && !paused) {
 416                                 Game.started++;
 417                                 fprintf(stderr, MSG_SERVER_GAME_START "\n");
 418                                 for (i = 1; i < MAX_SCREENS; i++)
 419                                         if (Players[i].alive > 0)
 420                                                 SendPacketTo(i, 0, NP_start, 0, NULL);
 421                         } //everybody ready to start
 422                 } //game (ready to) start(ed)
 423         } while (1);
 424         fprintf(stderr, "* Exiting server\n");
 425 }
 426
 427
 428 static void SHeader(void)
 429 {
 430         fprintf(stderr, MSG_SERVER_TITLE "\n\n", version_string);
 431 }
 432
 433 static void SUsage(void)
 434 {
 435         SHeader();
 436         fprintf(stderr,
 437                 "Usage: netris <options>\n"
 438                 "\n"
 439                 "  -h, --help\t\tPrint this usage information\n"
 440                 "  -H, --info\t\tShow distribution and warranty information\n"
 441                 "\n"
 442                 "  -p, --port %d\tSet port number\n"
 443                 "\n"
 444                 "  -s, --seed <seed>\tStart with given random seed\n"
 445                 "  -i, --speed %.2f\tSet the initial step-down interval, in seconds\n"
 446                 "  -m, --min-players %d\tNumber of players required before starting the game\n"
 447                 "  -x, --max-players %d\tMaximum number of players allowed in the game\n"
 448                 "  -c, --continuous\tDon'n quit the game\n"
 449                 "\n",
 450                 DEFAULT_PORT, DEFAULT_INTERVAL/1e6, minplayers, maxplayers
 451         );
 452 }
 453
 454 static void HandleOption(char tag, char *value)
 455 {
 456         switch (tag) {
 457         case 'v':  //verbose
 458                 verbose = 1;
 459                 break;
 460         case 'p':  //port
 461                 port = atoi(value);
 462                 break;
 463         case 'c':  //min-players
 464                 Game.continuous = atoi(value);
 465                 break;
 466         case 'm':  //min-players
 467                 minplayers = atoi(value);
 468                 break;
 469         case 'x':  //max-players
 470                 maxplayers = MIN(atoi(value), MAX_SCREENS);
 471                 break;
 472         case 'q':  //quadra-style gravity
 473                 Game.gravity ^= 1;
 474                 break;
 475         case 'i':  //speed (of level 1)
 476                 Game.initspeed = atof(value) * 1e6;
 477                 break;
 478         case 's':  //seed
 479                 Game.seed = atoi(value);
 480                 break;
 481         case 'H':  //info
 482                 SHeader();
 483                 DistInfo();
 484                 exit(0);
 485         case 'h':  //help
 486                 SUsage();
 487                 exit(0);
 488         default:
 489                 break;
 490         }
 491 }
 492
 493 static void ReadConf(char *filename)
 494 {
 495         FILE *file_in;
 496         char buf[513];
 497         int i;
 498         char *ch;
 499         char tag[81], value[81];
 500
 501         file_in = fopen(filename, "r");
 502         if (file_in) {
 503                 while (fgets(buf, 512, file_in) != NULL) {
 504                         if ((ch = strchr(buf, '#')))
 505                                 *ch = '\0'; // truncate string from # char
 506                         for (i = strlen(buf)-1; i >= 0; i--)
 507                                 if (buf[i] == ' '  || buf[i] == '\t'
 508                                  || buf[i] == '\n' || buf[i] == 13)
 509                                         buf[i] = '\0';
 510                                 else break;
 511
 512                         sscanf(buf, "%80[^= \t] = %80[^\n]", tag, value);
 513                         for (i = 0; options[i].name; i++){
 514                                 if (!strcasecmp(options[i].name, tag)) {
 515                                         HandleOption(options[i].val, value);
 516                                         break;
 517                                 }
 518                         }
 519                 }
 520                 fclose(file_in);
 521         } //read file
 522         else {
 523                 fprintf(stderr, "Unable to open config file %s.\n", filename);
 524         } //defaults
 525
 526 }
 527
 528 static void CatchInt(int sig)
 529 {
 530         siglongjmp(close_env, 1);
 531 }
 532
 533 int main(int argc, char **argv)
 534 {
 535         int i;
 536         char ch;
 537
 538         if (sigsetjmp(close_env, 1)) exit(0);
 539         signal(SIGINT, CatchInt);
 540         port = DEFAULT_PORT;
 541         maxplayers = 8;
 542         Game.initspeed = DEFAULT_INTERVAL;
 543         Game.seed = time(0);
 544         for (i = 0; i < sizeof(Game.shapes) / sizeof(Game.shapes[0]); i++)
 545                 Game.shapes[i] = 1;
 546         Game.gravity = 0;
 547         {
 548                 int i;
 549
 550                 for (i = 1; i < MAX_SCREENS; i++)
 551                         Players[i].alive = -1;
 552         }
 553
 554 //      if (getopt(argc, argv, "f:") == 'f')
 555 //              ReadConf(optarg);
 556 //      else
 557         ReadConf(CONFIG_FILE);
 558         while ((ch = getopt_long(
 559                 argc, argv, "hHvqp:i:s:c:m:x:", options, NULL
 560         )) != -1)
 561                 HandleOption(ch, optarg);
 562         if (optind < argc) {
 563                 SUsage();
 564                 exit(1);
 565         }
 566 //      WriteConf();
 567
 568         SHeader();
 569
 570         {
 571                 int i;
 572
 573                 for (i = 1; i < MAX_SCREENS; i++)
 574                         memcpy(&netGen[i], &netGen[0], sizeof(EventGenRec));
 575         } //setup netGen var
 576
 577         AtExit(CloseNets);
 578
 579         {
 580                 int val1;
 581
 582                 memset(&addr, 0, sizeof(addr));
 583                 addr.sin_family = AF_INET;
 584                 addr.sin_port = htons(port);
 585                 addr.sin_addr.s_addr = htonl(INADDR_ANY);
 586                 if ((connGen.fd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
 587                         die("socket");
 588                 val1 = 1;
 589                 setsockopt(connGen.fd, SOL_SOCKET, SO_REUSEADDR,
 590                         (void *)&val1, sizeof(val1));
 591                 if (bind(connGen.fd, (struct sockaddr *)&addr, sizeof(addr)) < 0)
 592                         die("bind");
 593                 if (listen(connGen.fd, 1) < 0)
 594                         die("listen");
 595
 596                 AddEventGen(&connGen);
 597         } //setup listening sock
 598
 599         StartServer(); //server loop
 600
 601         return 0;
 602 }
 603