rnd-20060112-1-src
[rocksndiamonds.git] / src / netserv.c
1 /***********************************************************
2 * Rocks'n'Diamonds -- McDuffin Strikes Back!               *
3 *----------------------------------------------------------*
4 * (c) 1995-2002 Artsoft Entertainment                      *
5 *               Holger Schemel                             *
6 *               Detmolder Strasse 189                      *
7 *               33604 Bielefeld                            *
8 *               Germany                                    *
9 *               e-mail: info@artsoft.org                   *
10 *----------------------------------------------------------*
11 * netserv.c                                                *
12 ***********************************************************/
13
14 #include "libgame/platform.h"
15
16 #if defined(NETWORK_AVALIABLE)
17
18 #include <fcntl.h>
19 #include <sys/time.h>
20 #include <signal.h>
21 #include <errno.h>
22
23 #if defined(TARGET_SDL)
24 #include "main.h"
25 #else
26 #include <sys/socket.h>
27 #include <netinet/in.h>
28 #include <netinet/tcp.h>
29 #include <arpa/inet.h>
30 #include <netdb.h>
31 #include <sys/select.h>                 /* apparently needed for OS/2 port */
32 #endif
33
34 #include "libgame/libgame.h"
35
36 #include "netserv.h"
37
38 static int clients = 0;
39 static int onceonly = 0;
40
41 struct NetworkServerPlayerInfo
42 {
43 #if defined(TARGET_SDL)
44   TCPsocket fd;
45 #else
46   int fd;
47 #endif
48
49   char player_name[16];
50   unsigned char number;
51   struct NetworkServerPlayerInfo *next;
52   char active;
53   char introduced;
54   unsigned char readbuffer[MAX_BUFFER_SIZE];
55   unsigned char writbuffer[MAX_BUFFER_SIZE];
56   int nread, nwrite;
57   byte action;
58   boolean action_received;
59 };
60
61 static struct NetworkServerPlayerInfo *first_player = NULL;
62
63 #define NEXT(player) ((player)->next ? (player)->next : first_player)
64
65 #if defined(TARGET_SDL)
66 /* TODO: peer address */
67 static TCPsocket lfd;           /* listening socket */
68 static SDLNet_SocketSet fds;    /* socket set */
69 #else
70 static struct sockaddr_in saddr;
71 static int lfd;                 /* listening socket */
72 static fd_set fds;              /* socket set */
73 static int tcp = -1;
74 #endif
75
76 static unsigned char realbuffer[512], *buffer = realbuffer + 4;
77
78 static int interrupt;
79
80 static unsigned long ServerFrameCounter = 0;
81
82 static void addtobuffer(struct NetworkServerPlayerInfo *player,
83                         unsigned char *b, int len)
84 {
85   if (player->nwrite + len >= MAX_BUFFER_SIZE)
86     Error(ERR_EXIT_NETWORK_SERVER,
87           "internal error: network send buffer overflow");
88
89   memcpy(player->writbuffer + player->nwrite, b, len);
90   player->nwrite += len;
91 }
92
93 static void flushuser(struct NetworkServerPlayerInfo *player)
94 {
95   if (player->nwrite)
96   {
97 #if defined(TARGET_SDL)
98     SDLNet_TCP_Send(player->fd, player->writbuffer, player->nwrite);
99 #else
100     write(player->fd, player->writbuffer, player->nwrite);
101 #endif
102     player->nwrite = 0;
103   }
104 }
105
106 static void broadcast(struct NetworkServerPlayerInfo *except,
107                       int len, int activeonly)
108 {
109   struct NetworkServerPlayerInfo *player;
110
111   realbuffer[0] = realbuffer[1] = realbuffer[2] = 0;
112   realbuffer[3] = (unsigned char)len;
113   for (player = first_player; player; player = player->next)
114     if (player != except && player->introduced &&
115         (player->active || !activeonly))
116       addtobuffer(player, realbuffer, 4 + len);
117 }
118
119 static void sendtoone(struct NetworkServerPlayerInfo *to, int len)
120 {
121   realbuffer[0] = realbuffer[1] = realbuffer[2] = 0;
122   realbuffer[3] = (unsigned char)len;
123   addtobuffer(to, realbuffer, 4 + len);
124 }
125
126 static void RemovePlayer(struct NetworkServerPlayerInfo *player)
127 {
128   struct NetworkServerPlayerInfo *v;
129   
130   if (options.verbose)
131     Error(ERR_NETWORK_SERVER, "dropping client %d (%s)",
132           player->number, player->player_name);
133
134   if (player == first_player)
135     first_player = player->next;
136   else
137   {
138     for (v = first_player; v; v = v->next)
139     {
140       if (v->next && v->next == player)
141       {
142         v->next = player->next;
143         break;
144       }
145     }
146   }
147
148 #if defined(TARGET_SDL)
149   SDLNet_TCP_DelSocket(fds, player->fd);
150   SDLNet_TCP_Close(player->fd);
151 #else
152   close(player->fd);
153 #endif
154
155   if (player->introduced)
156   {
157     buffer[0] = player->number;
158     buffer[1] = OP_PLAYER_DISCONNECTED;
159     broadcast(player, 2, 0);
160   }
161
162   free(player);
163   clients--;
164
165   if (onceonly && clients == 0)
166   {
167     if (options.verbose)
168     {
169       Error(ERR_NETWORK_SERVER, "no clients left");
170       Error(ERR_NETWORK_SERVER, "aborting");
171     }
172     exit(0);
173   }
174 }
175
176 #if defined(TARGET_SDL)
177 static void AddPlayer(TCPsocket fd)
178 #else
179 static void AddPlayer(int fd)
180 #endif
181 {
182   struct NetworkServerPlayerInfo *player, *v;
183   unsigned char nxn;
184   boolean again = TRUE;
185
186   player = checked_malloc(sizeof (struct NetworkServerPlayerInfo));
187
188   player->fd = fd;
189   player->player_name[0] = 0;
190   player->next = first_player;
191   player->active = 0;
192   player->nread = 0;
193   player->nwrite = 0;
194   player->introduced = 0;
195   player->action = 0;
196   player->action_received = FALSE;
197
198 #if defined(TARGET_SDL)
199   SDLNet_TCP_AddSocket(fds, fd);
200 #endif
201
202   first_player = player;
203
204   nxn = 1;
205
206   while (again)
207   {
208     again = FALSE;
209     v = player->next;
210
211     while (v)
212     {
213       if (v->number == nxn)
214       {
215         nxn++;
216
217         again = TRUE;
218         break;
219       }
220       v = v->next;
221     }
222   }
223
224   player->number = nxn;
225 #if !defined(TARGET_SDL)
226   if (options.verbose)
227     Error(ERR_NETWORK_SERVER, "client %d connecting from %s",
228           nxn, inet_ntoa(saddr.sin_addr));
229 #endif
230   clients++;
231
232   buffer[0] = 0;
233   buffer[1] = OP_YOUR_NUMBER;
234   buffer[2] = player->number;
235   sendtoone(player, 3);
236 }
237
238 static void Handle_OP_PROTOCOL_VERSION(struct NetworkServerPlayerInfo *player,
239                                        unsigned int len)
240 {
241   if (len != 5 ||
242       buffer[2] != PROTOCOL_VERSION_1 ||
243       buffer[3] != PROTOCOL_VERSION_2)
244   {
245     if (options.verbose)
246       Error(ERR_NETWORK_SERVER,
247             "client %d (%s) has wrong protocol version %d.%d.%d",
248             player->number, player->player_name, buffer[2], buffer[3], buffer[4]);
249
250     buffer[0] = 0;
251     buffer[1] = OP_BAD_PROTOCOL_VERSION;
252     buffer[2] = PROTOCOL_VERSION_1;
253     buffer[3] = PROTOCOL_VERSION_2;
254     buffer[4] = PROTOCOL_VERSION_3;
255     sendtoone(player, 5);
256     flushuser(player);
257
258     RemovePlayer(player);
259     interrupt = 1;
260   }
261   else
262   {
263     if (options.verbose)
264       Error(ERR_NETWORK_SERVER,
265             "client %d (%s) uses protocol version %d.%d.%d",
266             player->number, player->player_name, buffer[2], buffer[3], buffer[4]);
267   }
268 }
269
270 static void Handle_OP_NUMBER_WANTED(struct NetworkServerPlayerInfo *player)
271 {
272   struct NetworkServerPlayerInfo *v;
273   int client_nr = player->number;
274   int nr_wanted = buffer[2];
275   int nr_is_free = 1;
276
277   if (options.verbose)
278       Error(ERR_NETWORK_SERVER, "client %d (%s) wants to switch to # %d",
279             player->number, player->player_name, nr_wanted);
280
281   for (v = first_player; v; v = v->next)
282   {
283     if (v->number == nr_wanted)
284     {
285       nr_is_free = 0;
286       break;
287     }
288   }
289
290   if (options.verbose)
291   {
292     if (nr_is_free)
293       Error(ERR_NETWORK_SERVER, "client %d (%s) switches to # %d",
294             player->number, player->player_name, nr_wanted);
295     else if (player->number == nr_wanted)
296       Error(ERR_NETWORK_SERVER, "client %d (%s) still has # %d",
297             player->number, player->player_name, nr_wanted);
298     else
299       Error(ERR_NETWORK_SERVER,
300             "client %d (%s) cannot switch (client %d still exists)",
301             player->number, player->player_name, nr_wanted);
302   }
303
304   if (nr_is_free)
305     player->number = nr_wanted;
306
307   buffer[0] = client_nr;
308   buffer[1] = OP_NUMBER_WANTED;
309   buffer[2] = nr_wanted;
310   buffer[3] = player->number;
311
312   /*
313   sendtoone(player, 4);
314   */
315
316   broadcast(NULL, 4, 0);
317 }
318
319 static void Handle_OP_PLAYER_NAME(struct NetworkServerPlayerInfo *player,
320                                   unsigned int len)
321 {
322   struct NetworkServerPlayerInfo *v;
323   int i;
324
325   if (len>16)
326     len=16;
327   memcpy(player->player_name, &buffer[2], len-2);
328   player->player_name[len-2] = 0;
329   for (i = 0; i < len - 2; i++)
330   {
331     if (player->player_name[i] < ' ' || 
332         ((unsigned char)(player->player_name[i]) > 0x7e &&
333          (unsigned char)(player->player_name[i]) <= 0xa0))
334     {
335       player->player_name[i] = 0;
336       break;
337     }
338   }
339
340   if (!player->introduced)
341   {
342     buffer[0] = player->number;
343     buffer[1] = OP_PLAYER_CONNECTED;
344     broadcast(player, 2, 0);
345   }
346               
347   if (options.verbose)
348     Error(ERR_NETWORK_SERVER, "client %d calls itself \"%s\"",
349           player->number, player->player_name);
350   buffer[1] = OP_PLAYER_NAME;
351   broadcast(player, len, 0);
352
353   if (!player->introduced)
354   {
355     for (v = first_player; v; v = v->next)
356     {
357       if (v != player && v->introduced)
358       {
359         buffer[0] = v->number;
360         buffer[1] = OP_PLAYER_CONNECTED;
361         sendtoone(player, 2);
362         buffer[1] = OP_PLAYER_NAME;
363         memcpy(&buffer[2], v->player_name, 14);
364         sendtoone(player, 2+strlen(v->player_name));
365       }
366     }
367   }
368
369   player->introduced = 1;
370 }
371
372 static void Handle_OP_START_PLAYING(struct NetworkServerPlayerInfo *player)
373 {
374   struct NetworkServerPlayerInfo *v, *w;
375
376   if (options.verbose)
377     Error(ERR_NETWORK_SERVER,
378           "client %d (%s) starts game [level %d from levedir %d (%s)]",
379           player->number, player->player_name,
380           (buffer[2] << 8) + buffer[3],
381           (buffer[4] << 8) + buffer[5],
382           &buffer[6]);
383
384   for (w = first_player; w; w = w->next)
385     if (w->introduced)
386       w->active = 1;
387
388   /* reset frame counter */
389   ServerFrameCounter = 0;
390
391   /* reset player actions */
392   for (v = first_player; v; v = v->next)
393   {
394     v->action = 0;
395     v->action_received = FALSE;
396   }
397
398   broadcast(NULL, 10 + strlen((char *)&buffer[10])+1, 0);
399 }
400
401 static void Handle_OP_PAUSE_PLAYING(struct NetworkServerPlayerInfo *player)
402 {
403   if (options.verbose)
404     Error(ERR_NETWORK_SERVER, "client %d (%s) pauses game",
405           player->number, player->player_name);
406   broadcast(NULL, 2, 0);
407 }
408
409 static void Handle_OP_CONTINUE_PLAYING(struct NetworkServerPlayerInfo *player)
410 {
411   if (options.verbose)
412     Error(ERR_NETWORK_SERVER, "client %d (%s) continues game",
413           player->number, player->player_name);
414   broadcast(NULL, 2, 0);
415 }
416
417 static void Handle_OP_STOP_PLAYING(struct NetworkServerPlayerInfo *player)
418 {
419   if (options.verbose)
420     Error(ERR_NETWORK_SERVER, "client %d (%s) stops game",
421           player->number, player->player_name);
422   broadcast(NULL, 2, 0);
423 }
424
425 static void Handle_OP_MOVE_PLAYER(struct NetworkServerPlayerInfo *player)
426 {
427   struct NetworkServerPlayerInfo *v;
428   int last_client_nr = 0;
429   int i;
430
431   /* store player action */
432   for (v = first_player; v; v = v->next)
433   {
434     if (v->number == player->number)
435     {
436       v->action = buffer[2];
437       v->action_received = TRUE;
438     }
439   }
440
441   /* check if server received action from each player */
442   for (v = first_player; v; v = v->next)
443   {
444     if (!v->action_received)
445       return;
446
447     if (v->number > last_client_nr)
448       last_client_nr = v->number;
449   }
450
451   /* initialize all player actions to zero */
452   for (i = 0; i < last_client_nr; i++)
453     buffer[6 + i] = 0;
454
455   /* broadcast actions of all players to all players */
456   for (v = first_player; v; v = v->next)
457   {
458     buffer[6 + v->number-1] = v->action;
459     v->action = 0;
460     v->action_received = FALSE;
461   }
462
463   buffer[2] = (unsigned char)((ServerFrameCounter >> 24) & 0xff);
464   buffer[3] = (unsigned char)((ServerFrameCounter >> 16) & 0xff);
465   buffer[4] = (unsigned char)((ServerFrameCounter >>  8) & 0xff);
466   buffer[5] = (unsigned char)((ServerFrameCounter >>  0) & 0xff);
467
468   broadcast(NULL, 6 + last_client_nr, 0);
469
470   ServerFrameCounter++;
471 }
472
473 #if defined(TARGET_SDL)
474 /* the following is not used for a standalone server;
475    the pointer points to an integer containing the port-number */
476 int NetworkServerThread(void *ptr)
477 {
478   NetworkServer(*((int *) ptr), 0);
479
480   /* should never be reached */
481   return 0;
482 }
483 #endif
484
485 void NetworkServer(int port, int serveronly)
486 {
487   int sl;
488   struct NetworkServerPlayerInfo *player;
489   int r; 
490   unsigned int len;
491 #if defined(TARGET_SDL)
492   IPaddress ip;
493 #else
494   int i, on;
495   int is_daemon = 0;
496   struct protoent *tcpproto;
497   struct timeval tv;
498   int mfd;
499 #endif
500
501 #if defined(PLATFORM_UNIX) && !defined(PLATFORM_NEXT)
502   struct sigaction sact;
503 #endif
504
505   if (port == 0)
506     port = DEFAULT_SERVER_PORT;
507
508   if (!serveronly)
509     onceonly = 1;
510
511 #if !defined(TARGET_SDL)
512   if ((tcpproto = getprotobyname("tcp")) != NULL)
513     tcp = tcpproto->p_proto;
514 #endif
515
516 #if defined(PLATFORM_UNIX)
517 #if defined(PLATFORM_NEXT)
518   signal(SIGPIPE, SIG_IGN);
519 #else
520   sact.sa_handler = SIG_IGN;
521   sigemptyset(&sact.sa_mask);
522   sact.sa_flags = 0;
523   sigaction(SIGPIPE, &sact, NULL);
524 #endif
525 #endif
526
527 #if defined(TARGET_SDL)
528
529   /* assume that SDL is already initialized */
530 #if 0
531   if (SDLNet_Init() == -1)
532     Error(ERR_EXIT_NETWORK_SERVER, "SDLNet_Init() failed");
533   atexit(SDLNet_Quit);
534 #endif
535
536   if (SDLNet_ResolveHost(&ip, NULL, port) == -1)
537     Error(ERR_EXIT_NETWORK_SERVER, "SDLNet_ResolveHost() failed");
538
539   lfd = SDLNet_TCP_Open(&ip);
540   if (!lfd)
541     Error(ERR_EXIT_NETWORK_SERVER, "SDLNet_TCP_Open() failed");
542
543   fds = SDLNet_AllocSocketSet(MAX_PLAYERS+1);
544   SDLNet_TCP_AddSocket(fds, lfd);
545
546 #else
547
548   if ((lfd = socket(PF_INET, SOCK_STREAM, 0)) < 0)
549     Error(ERR_EXIT_NETWORK_SERVER, "socket() failed");
550
551   saddr.sin_family = AF_INET;
552   saddr.sin_addr.s_addr = htonl(INADDR_ANY);
553   saddr.sin_port = htons(port);
554
555   on = 1;
556
557   setsockopt(lfd, SOL_SOCKET, SO_REUSEADDR, (char *)&on, sizeof(int));
558   if (bind(lfd, (struct sockaddr *)&saddr, sizeof(saddr)) < 0)
559     Error(ERR_EXIT_NETWORK_SERVER, "bind() failed");
560
561   listen(lfd, 5);
562 #endif
563
564 #if !defined(TARGET_SDL)
565   if (is_daemon)
566   {
567     /* become a daemon, breaking all ties with the controlling terminal */
568     options.verbose = FALSE;
569     for (i = 0; i < 255; i++)
570     {
571       if (i != lfd)
572         close(i);
573     }
574
575     if (fork())
576       exit(0);
577     setsid();
578     if (fork())
579       exit(0);
580     chdir("/");
581
582     /* open a fake stdin, stdout, stderr, just in case */
583     open("/dev/null", O_RDONLY);
584     open("/dev/null", O_WRONLY);
585     open("/dev/null", O_WRONLY);
586   }
587 #endif
588
589   if (options.verbose)
590   {
591     Error(ERR_NETWORK_SERVER, "started up, listening on port %d", port);
592     Error(ERR_NETWORK_SERVER, "using protocol version %d.%d.%d",
593           PROTOCOL_VERSION_1, PROTOCOL_VERSION_2, PROTOCOL_VERSION_3);
594   }
595
596   while (1)
597   {
598     interrupt = 0;
599
600     for (player = first_player; player; player = player->next)
601       flushuser(player);
602
603 #if defined(TARGET_SDL)
604     if ((sl = SDLNet_CheckSockets(fds, 500000)) < 1)
605     {
606       Error(ERR_NETWORK_SERVER, SDLNet_GetError());
607       perror("SDLNet_CheckSockets");
608     }
609
610 #else
611
612     FD_ZERO(&fds);
613     mfd = lfd;
614     player = first_player;
615     while (player)
616     {
617       FD_SET(player->fd, &fds);
618       if (player->fd > mfd)
619         mfd = player->fd;
620       player = player->next;
621     }
622     FD_SET(lfd, &fds);
623     tv.tv_sec = 0;
624     tv.tv_usec = 500000;
625     if ((sl = select(mfd + 1, &fds, NULL, NULL, &tv)) < 0)
626     {
627       if (errno != EINTR)
628         Error(ERR_EXIT_NETWORK_SERVER, "select() failed");
629       else
630         continue;
631     }
632 #endif
633
634     if (sl < 0)
635       continue;
636     
637     if (sl == 0)
638       continue;
639
640     /* accept incoming connections */
641 #if defined(TARGET_SDL)
642     if (SDLNet_SocketReady(lfd))
643     {
644       TCPsocket newsock;
645
646       newsock = SDLNet_TCP_Accept(lfd);
647
648       if (newsock)
649         AddPlayer(newsock);
650     }
651
652 #else
653
654     if (FD_ISSET(lfd, &fds))
655     {
656       int newfd, slen;
657
658       slen = sizeof(saddr);
659       newfd = accept(lfd, (struct sockaddr *)&saddr, &slen);
660       if (newfd < 0)
661       {
662         if (errno != EINTR)
663           Error(ERR_EXIT_NETWORK_SERVER, "accept() failed");
664       }
665       else
666       {
667         if (tcp != -1)
668         {
669           on = 1;
670           setsockopt(newfd, tcp, TCP_NODELAY, (char *)&on, sizeof(int));
671         }
672         AddPlayer(newfd);
673       }
674       continue;
675     }
676 #endif
677
678     player = first_player;
679
680     do
681     {
682 #if defined(TARGET_SDL)
683       if (SDLNet_SocketReady(player->fd))
684 #else
685       if (FD_ISSET(player->fd, &fds))
686 #endif
687       {
688 #if defined(TARGET_SDL)
689         /* read only 1 byte, because SDLNet blocks when we want more than is
690            in the buffer */
691         r = SDLNet_TCP_Recv(player->fd, player->readbuffer + player->nread, 1);
692 #else
693         r = read(player->fd, player->readbuffer + player->nread,
694                  MAX_BUFFER_SIZE - player->nread);
695 #endif
696
697         if (r <= 0)
698         {
699           if (options.verbose)
700             Error(ERR_NETWORK_SERVER, "EOF from client %d (%s)",
701                   player->number, player->player_name);
702           RemovePlayer(player);
703           interrupt = 1;
704           break;
705         }
706         player->nread += r;
707         while (player->nread >= 4 && player->nread >= 4 + player->readbuffer[3])
708         {
709           len = player->readbuffer[3];
710           if (player->readbuffer[0] || player->readbuffer[1] || player->readbuffer[2])
711           {
712             if (options.verbose)
713               Error(ERR_NETWORK_SERVER, "crap from client %d (%s)",
714                     player->number, player->player_name);
715             RemovePlayer(player);
716             interrupt = 1;
717             break;
718           }
719           memcpy(buffer, &player->readbuffer[4], len);
720           player->nread -= 4 + len;
721           memmove(player->readbuffer, player->readbuffer + 4 + len, player->nread);
722
723           buffer[0] = player->number;
724           if (!player->introduced && buffer[1] != OP_PLAYER_NAME)
725           {
726             if (options.verbose)
727               Error(ERR_NETWORK_SERVER, "!(client %d)->introduced && buffer[1]==%d (expected OP_PLAYER_NAME)", buffer[0], buffer[1]);
728
729             RemovePlayer(player);
730             interrupt = 1;
731             break;
732           }
733
734           switch(buffer[1])
735           {
736             case OP_PLAYER_NAME:
737               Handle_OP_PLAYER_NAME(player, len);
738               break;
739
740             case OP_PROTOCOL_VERSION:
741               Handle_OP_PROTOCOL_VERSION(player, len);
742               break;
743
744             case OP_NUMBER_WANTED:
745               Handle_OP_NUMBER_WANTED(player);
746               break;
747
748             case OP_START_PLAYING:
749               Handle_OP_START_PLAYING(player);
750               break;
751
752             case OP_PAUSE_PLAYING:
753               Handle_OP_PAUSE_PLAYING(player);
754               break;
755
756             case OP_CONTINUE_PLAYING:
757               Handle_OP_CONTINUE_PLAYING(player);
758               break;
759
760             case OP_STOP_PLAYING:
761               Handle_OP_STOP_PLAYING(player);
762               break;
763
764             case OP_MOVE_PLAYER:
765               Handle_OP_MOVE_PLAYER(player);
766               break;
767
768             case OP_BROADCAST_MESSAGE:
769               buffer[len] = '\0';
770               if (options.verbose)
771                 Error(ERR_NETWORK_SERVER, "client %d (%s) sends message: %s",
772                       player->number, player->player_name, &buffer[2]);
773               broadcast(player, len, 0);
774               break;
775             
776             default:
777               if (options.verbose)
778                 Error(ERR_NETWORK_SERVER,
779                       "unknown opcode %d from client %d (%s)",
780                       buffer[0], player->number, player->player_name);
781           }
782         }
783       }
784
785       if (player && !interrupt)
786         player = player->next;
787     }
788     while (player && !interrupt);
789   }
790 }
791
792 #endif /* NETWORK_AVALIABLE */