inet.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: inet.c,v 1.18 1996/02/09 08:22:13 mhw Exp $
  20  */
  21
  22 #include "netris.h"
  23 #include <sys/types.h>
  24 #include <unistd.h>
  25 #include <stdlib.h>
  26 #include <sys/socket.h>
  27 #include <netinet/in.h>
  28 #include <netdb.h>
  29 #include <string.h>
  30 #include <errno.h>
  31
  32 #define HEADER_SIZE sizeof(netint2[2])
  33 #define HEADER_SIZE3 sizeof(netint4[3])
  34
  35 ExtFunc MyEventType NetGenFunc(EventGenRec *gen, MyEvent *event);
  36
  37 EventGenRec netGen[MAX_SCREENS] = {
  38         { NULL, 0, FT_read, -1, NetGenFunc, EM_net, 0, "\0", 0, HEADER_SIZE3 } };
  39
  40
  41 ExtFunc int InitiateConnection(char *hostStr, short port)
  42 { //connect to host
  43         struct sockaddr_in addr;
  44         struct hostent *host;
  45
  46         AtExit(CloseNet);
  47         host = gethostbyname(hostStr);
  48         if (!host)
  49                 die("gethostbyname");
  50         assert(host->h_addrtype == AF_INET);
  51  again:
  52         memset(&addr, 0, sizeof(addr));
  53         addr.sin_family = host->h_addrtype;
  54         memcpy(&addr.sin_addr, host->h_addr, host->h_length);
  55         addr.sin_port = htons(port);
  56         if ((netGen[0].fd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
  57                 die("socket");
  58         if (connect(netGen[0].fd, (struct sockaddr *)&addr, sizeof(addr)) < 0) {
  59                 if (errno != ECONNREFUSED)
  60                         die("connect");
  61                 close(netGen[0].fd);
  62                 sleep(1);
  63                 goto again;
  64         }
  65         AddEventGen(&netGen[0]);
  66         totalPlayers = 1;
  67         return 0;
  68 } //InitiateConnection
  69
  70 ExtFunc void HandShake(void)
  71 { //talk to your host
  72         MyEvent event;
  73
  74         {
  75                 netint4 versiondata[2];
  76                 versiondata[0] = hton4(MAJOR_VERSION);
  77                 versiondata[1] = hton4(PROTOCOL_VERSION);
  78                 SendPacket(0, NP_hello, sizeof(versiondata), versiondata);
  79         }
  80
  81         do {
  82                 if (WaitMyEvent(&event, EM_net) == E_net)
  83                 switch (event.u.net.type) {
  84                         case NP_hello:
  85                         {
  86                                 me = event.u.net.uid;
  87                                 memcpy(&Players[me], &Players[0], sizeof(Player));
  88                                 fprintf(stderr, "Accepted (%s) as #%d (%s)\n",
  89                                         event.u.net.data, me, Players[me].name);
  90                                 SendPacket(0, NP_newPlayer,
  91                                         sizeof(Player) - sizeof(Players[me].host) - sizeof(Players[me].spy),
  92                                         &Players[me]);
  93                                 break;
  94                         }
  95                         case NP_gamedata:
  96                         {
  97                                 memcpy(&Game, event.u.net.data, event.u.net.size);
  98                                 SRandom(Game.seed);
  99                                 break;
 100                         }
 101                         case NP_newPlayer:
 102                         {
 103                                 totalPlayers++;
 104                                 memcpy(&Players[event.u.net.uid],
 105                                         event.u.net.data, event.u.net.size);
 106                                 fprintf(stderr, "Receiving player #%d (%s)\n",
 107                                         event.u.net.uid, Players[event.u.net.uid].name);
 108                                 break;
 109                         }
 110                         case NP_error:
 111                         {
 112                                 fprintf(stderr, "Rejected by server: %s\n", event.u.net.data);
 113                                 exit(1);
 114                         }
 115                         default:
 116                                 break;
 117                 }
 118                 else
 119                         fatal("Hm, the party apparantly ended prematurely.");
 120         }
 121         while (event.u.net.type != NP_goAhead);
 122
 123         // send Players[0]
 124         // receive seed, initspeed
 125                 // receive #players
 126                 // receive Players[*]
 127
 128         /*
 129                         {
 130                         netint4 data[3];
 131                         int len;
 132                         int seed;
 133
 134                         if (protocolVersion >= 3)
 135                                 len = sizeof(data);
 136                         else
 137                                 len = sizeof(netint4[2]);
 138                         if ((Players[0].flags & SCF_setSeed))
 139                                 seed = Game.seed;
 140                         else
 141                                 seed = time(0);
 142                         if (waitConn)
 143                                 SRandom(seed);
 144                         data[0] = hton4(Players[0].flags);
 145                         data[1] = hton4(seed);
 146                         data[2] = hton4(Game.initspeed);
 147                         SendPackets(0, NP_startConn, len, data);
 148                         if (WaitMyEvent(&event, EM_net) != E_net ||
 149                                         event.u.net.type != NP_startConn)
 150                                 fatal("Network negotiation failed");
 151                         memcpy(data, event.u.net.data, len);
 152                         Players[1].flags = ntoh4(data[0]);
 153                         seed = ntoh4(data[1]);
 154                         if (initConn) {
 155                                 if ((Players[0].flags & SCF_setSeed) != (Players[1].flags & SCF_setSeed))
 156                                         fatal("If one player sets the random number seed, "
 157                                                         "both must.");
 158                                 if ((Players[0].flags & SCF_setSeed) && seed != Game.seed)
 159                                         fatal("Both players have set the random number seed, "
 160                                                         "and they are unequal.");
 161                                 if (protocolVersion >= 3 && Game.initspeed != ntoh4(data[2]))
 162                                         fatal("Your opponent is using a different step-down "
 163                                                 "interval (-i).\nYou must both use the same one.");
 164                                 SRandom(seed);
 165                         }
 166                 }
 167                 */
 168 //              SendPackets(0, NP_initData, strlen(Players[0].name) + 1, Players[0].name);
 169
 170 /*
 171                         if (WaitMyEvent(&event, EM_net) != E_net ||
 172                                         event.u.net.type != NP_userName)
 173                                 fatal("Network negotiation failed");
 174                         strncpy(Players[1].name, event.u.net.data,
 175                                 sizeof(Players[1].name) - 1);
 176                         Players[1].name[sizeof(Players[1].name)-1] = 0;
 177                         for (i = 0; Players[1].name[i]; ++i)
 178                                 if (!isprint(Players[1].name[i]))
 179                                         Players[1].name[i] = '?';
 180                         for (i = 0; Players[1].host[i]; ++i)
 181                                 if (!isprint(Players[1].host[i]))
 182                                         Players[1].host[i] = '?';
 183 */
 184 } //HandShake
 185
 186 ExtFunc void CheckNetConn(void)
 187 { //am I necessary?
 188 }
 189
 190
 191 ExtFunc MyEventType NetGenFunc(EventGenRec *gen, MyEvent *event)
 192 { //receive
 193         int result;
 194         short uid, type, size;
 195         netint4 *data = (netint4*)&(gen->buf);
 196
 197         result = MyRead(gen->fd, gen->buf + gen->bufSize,
 198                 gen->bufGoal - gen->bufSize);
 199         if (result < 0) {
 200                 close(gen->fd);
 201                 gen->fd = -1;
 202                 return E_lostConn;
 203         }
 204         gen->bufSize += result;
 205         if (gen->bufSize < gen->bufGoal)
 206                 return E_none;
 207         // *ugly* memcpy(data, gen->buf, sizeof(data));
 208         uid = ntoh4(data[0]);
 209         type = ntoh4(data[1]);
 210         size = ntoh4(data[2]);
 211         gen->bufGoal = size;
 212         if (gen->bufSize < gen->bufGoal)
 213                 return E_none;
 214         gen->bufSize = 0;
 215         gen->bufGoal = HEADER_SIZE3;
 216         event->u.net.sender = gen->player;
 217         event->u.net.uid = uid;
 218         event->u.net.type = type;
 219         event->u.net.size = size - HEADER_SIZE3;
 220         event->u.net.data = gen->buf + HEADER_SIZE3;
 221         if (type == NP_endConn) {
 222                 fprintf(stderr, "Close connection\n");
 223                 return E_lostConn;
 224         }
 225         return E_net;
 226 } //NetGenFunc
 227
 228 ExtFunc void SendPacket(short uid, NetPacketType type, int size, void *data)
 229 { //send shit to server
 230         netint4 header[3];
 231
 232         header[0] = hton4(uid);
 233         header[1] = hton4(type);
 234         header[2] = hton4(size + HEADER_SIZE3);
 235         if (MyWrite(netGen[0].fd, header, HEADER_SIZE3) != HEADER_SIZE3)
 236                 die("write (header)");
 237         if (size > 0 && data && MyWrite(netGen[0].fd, data, size) != size)
 238                 die("write");
 239 } //SendPacket
 240
 241 ExtFunc void CloseNet(void)
 242 { //kick some connection's ass!
 243         MyEvent event;
 244
 245         if (netGen[0].fd >= 0) {
 246                 SendPacket(0, NP_endConn, 0, NULL);
 247                 close(netGen[0].fd);
 248                 netGen[0].fd = -1;
 249         }
 250         if (netGen[0].next)
 251                 RemoveEventGen(&netGen[0]);
 252 } //CloseNet
 253
 254 /*
 255  * vi: ts=4 ai
 256  * vim: noai si
 257  */