]> icculus.org git repositories - taylor/freespace2.git/blob - src/network/valid.cpp
merge in some code cleanup
[taylor/freespace2.git] / src / network / valid.cpp
1 /*
2  * Copyright (C) Volition, Inc. 2005.  All rights reserved.
3  * 
4  * All source code herein is the property of Volition, Inc. You may not sell 
5  * or otherwise commercially exploit the source or things you created based on the 
6  * source.
7  *
8 */
9
10
11 //Validate tracker user class
12
13
14 #ifdef PLAT_UNIX
15 #include <arpa/inet.h>
16 #include <netinet/in.h>
17 #include <errno.h>
18 #include <netdb.h>
19 #endif
20
21 #include "multi.h"
22 #include "ptrack.h"
23 #include "valid.h"
24 #include "psnet.h"
25 #include "timer.h"
26
27
28 // check structs for size compatibility
29 SDL_COMPILE_TIME_ASSERT(vmt_validate_mission_req_struct, sizeof(vmt_validate_mission_req_struct) == 104);
30
31
32 // Variables
33 udp_packet_header PacketHeader;
34 validate_id_request *ValidIDReq;
35
36 int ValidState;
37
38 // SOCKET validsock;
39 struct sockaddr_in      rtrackaddr;
40
41 int ValidFirstSent;
42 int ValidLastSent;
43
44 char *Psztracker_id;
45
46 // mission validation
47 int MissionValidState;
48 int MissionValidFirstSent;
49 int MissionValidLastSent;
50
51 // squad war validation
52 int SquadWarValidState;
53 int SquadWarFirstSent;
54 int SquadWarLastSent;
55
56 // squad war response
57 squad_war_response SquadWarValidateResponse;
58
59
60 static int SerializeValidatePacket(const udp_packet_header *uph, ubyte *data)
61 {
62         int packet_size = 0;
63         int i;
64
65         PXO_ADD_DATA(uph->type);
66         PXO_ADD_USHORT(uph->len);
67         PXO_ADD_UINT(uph->code);
68         PXO_ADD_USHORT(uph->xcode);
69         PXO_ADD_UINT(uph->sig);
70         PXO_ADD_UINT(uph->security);
71
72         switch (uph->type) {
73                 // no extra data for this
74                 case UNT_CONTROL:
75                         break;
76
77                 case UNT_LOGIN_AUTH_REQUEST: {
78                         validate_id_request *id_req = (validate_id_request *)&uph->data;
79
80                         PXO_ADD_DATA(id_req->login);
81                         PXO_ADD_DATA(id_req->password);
82                         PXO_ADD_DATA(id_req->tracker_id);       // junk here, just for size
83
84                         break;
85                 }
86
87                 case UNT_VALID_FS2_MSN_REQ: {
88                         vmt_validate_mission_req_struct *mis_req = (vmt_validate_mission_req_struct *)&uph->data;
89
90                         PXO_ADD_UINT(mis_req->checksum);
91
92                         memcpy(data+packet_size, mis_req->file_name, strlen(mis_req->file_name));
93                         packet_size += strlen(mis_req->file_name);
94
95                         data[packet_size] = '\0';
96                         packet_size++;
97
98                         break;
99                 }
100
101                 case UNT_VALID_SW_MSN_REQ: {
102                         squad_war_request *sw_req = (squad_war_request *)&uph->data;
103
104                         for (i = 0; i < MAX_SQUAD_PLAYERS; i++) {
105                                 PXO_ADD_INT(sw_req->squad_plr1[i]);
106                         }
107
108                         for (i = 0; i < MAX_SQUAD_PLAYERS; i++) {
109                                 PXO_ADD_INT(sw_req->squad_plr2[i]);
110                         }
111
112                         PXO_ADD_DATA(sw_req->squad_count1);
113                         PXO_ADD_DATA(sw_req->squad_count2);
114
115                         PXO_ADD_DATA(sw_req->match_code);
116
117                         PXO_ADD_DATA(sw_req->mission_filename);
118                         PXO_ADD_INT(sw_req->mission_checksum);
119
120                         break;
121                 }
122
123                 // we shouldn't be sending any other packet types
124                 default:
125                         Int3();
126                         break;
127         }
128
129         SDL_assert(packet_size >= (int)PACKED_HEADER_ONLY_SIZE);
130         SDL_assert(packet_size == (int)uph->len);
131
132         return packet_size;
133 }
134
135 static void DeserializeValidatePacket(const ubyte *data, const int data_size, udp_packet_header *uph)
136 {
137         int offset = 0;
138
139         memset(uph, 0, sizeof(udp_packet_header));
140
141         // make sure we received a complete base packet
142         if (data_size < (int)PACKED_HEADER_ONLY_SIZE) {
143                 uph->len = 0;
144                 uph->type = 0xff;
145
146                 return;
147         }
148
149         PXO_GET_DATA(uph->type);
150         PXO_GET_USHORT(uph->len);
151         PXO_GET_UINT(uph->code);
152         PXO_GET_USHORT(uph->xcode);
153         PXO_GET_UINT(uph->sig);
154         PXO_GET_UINT(uph->security);
155
156         // sanity check data size to make sure we reveived all of the expected packet
157         // (not exactly sure what -1 is for, but that's how it is later)
158         if ((int)uph->len-1 > data_size) {
159                 uph->len = 0;
160                 uph->type = 0xff;
161
162                 return;
163         }
164
165         switch (uph->type) {
166                 // no extra data for these
167                 case UNT_CONTROL:
168                 case UNT_CONTROL_VALIDATION:
169                 case UNT_LOGIN_NO_AUTH:
170                 case UNT_VALID_FS_MSN_RSP:
171                 case UNT_VALID_FS2_MSN_RSP:
172                         break;
173
174                 case UNT_LOGIN_AUTHENTICATED: {
175                         SDL_strlcpy((char *)uph->data, (const char *)(data+offset), SDL_arraysize(uph->data));
176                         break;
177                 }
178
179                 case UNT_VALID_SW_MSN_RSP: {
180                         squad_war_response *sw_resp = (squad_war_response *)&uph->data;
181
182                         PXO_GET_DATA(sw_resp->reason);
183                         PXO_GET_DATA(sw_resp->accepted);
184
185                         break;
186                 }
187
188                 default:
189                         break;
190         }
191
192         //SDL_assert(offset == data_size);
193 }
194
195
196 int InitValidateClient(void)
197 {
198         struct sockaddr_in sockaddr;
199         in_addr_t iaddr;
200         ValidFirstSent = 0;
201         ValidLastSent = 0;
202         ValidState = VALID_STATE_IDLE;
203         
204         MissionValidFirstSent = 0;
205         MissionValidLastSent = 0;
206         MissionValidState = VALID_STATE_IDLE;
207
208         SquadWarFirstSent = 0;
209         SquadWarLastSent = 0;
210         SquadWarValidState = VALID_STATE_IDLE;
211
212         /*
213         validsock = socket(AF_INET,SOCK_DGRAM,0);       
214         if ( validsock == INVALID_SOCKET )
215         {
216                 printf("Unable to open a socket.\n");
217                 return 0;
218         }
219         */
220         
221         memset( &sockaddr, 0, sizeof(struct sockaddr_in) );
222         sockaddr.sin_family = AF_INET; 
223         sockaddr.sin_addr.s_addr = INADDR_ANY; 
224         sockaddr.sin_port = 0;
225         
226         /*
227         if (SOCKET_ERROR==bind(validsock, (struct sockaddr*)&sockaddr, sizeof (sockaddr)))
228         {       
229                 printf("Unable to bind a socket.\n");
230                 printf("WSAGetLastError() returned %d.\n",WSAGetLastError());
231                 return 0;
232         }
233         */
234
235         rtrackaddr.sin_family = AF_INET; 
236         iaddr = inet_addr( Multi_options_g.user_tracker_ip );
237         if ( iaddr == INADDR_NONE ) {
238                 struct hostent *he;
239                 he = gethostbyname( Multi_options_g.user_tracker_ip );
240                 if(!he)
241                         return 0;
242         /*
243                 {               
244                         // try and resolve by address
245                         unsigned int n_order = inet_addr(Multi_user_tracker_ip_address);
246                         he = gethostbyaddr((char*)&n_order,4,PF_INET);
247
248                         if(!he){
249                                 return 0;
250                         }
251                 }
252         */
253                 iaddr = ((in_addr *)(he->h_addr))->s_addr;
254         }
255         
256         rtrackaddr.sin_addr.s_addr = iaddr;
257         rtrackaddr.sin_port = htons(REGPORT);
258         
259         return 1;
260
261 }
262
263 //Call with a valid struct to validate a user
264 //Call with NULL to poll
265
266 //Return codes:
267 // -3   Still waiting (returned if we were waiting for a tracker response and ValidateUser was called with a non-NULL value
268 // -2 Timeout waiting for tracker to respond
269 // -1   User invalid
270 //  0   Still waiting for response from tracker/Idle
271 //  1   User valid
272 int ValidateUser(validate_id_request *valid_id, char *trackerid)
273 {
274         ubyte packet_data[sizeof(udp_packet_header)];
275         int packet_length = 0;
276
277         ValidIdle();
278         if(valid_id==NULL)
279         {
280                 switch(ValidState)
281                 {
282                 case VALID_STATE_IDLE:
283                         return 0;
284                         break;
285                 case VALID_STATE_WAITING:
286                         return 0;
287                         break;
288                 case VALID_STATE_VALID:
289                         ValidState = VALID_STATE_IDLE;
290                         return 1;
291                         break;
292                 case VALID_STATE_INVALID:
293                         ValidState = VALID_STATE_IDLE;
294                         return -1;
295                 case VALID_STATE_TIMEOUT:
296                         ValidState = VALID_STATE_IDLE;
297                         return -2;
298                 }
299                 return 0;
300         }
301         else
302         {
303                 if(ValidState==VALID_STATE_IDLE)
304                 {
305                         //First, flush the input buffer for the socket
306                         fd_set read_fds;                   
307                         struct timeval timeout;
308                         
309                         timeout.tv_sec=0;            
310                         timeout.tv_usec=0;
311                         
312                         FD_ZERO(&read_fds);
313                         FD_SET(Unreliable_socket, &read_fds);    
314
315                         while(SELECT(Unreliable_socket+1,&read_fds,NULL,NULL,&timeout, PSNET_TYPE_VALIDATION))
316                         {
317                                 int addrsize;
318                                 struct sockaddr_in fromaddr;
319
320                                 udp_packet_header inpacket;
321                                 addrsize = sizeof(struct sockaddr_in);
322                                 RECVFROM(Unreliable_socket, (char *)&inpacket,sizeof(udp_packet_header),0,(struct sockaddr *)&fromaddr,&addrsize, PSNET_TYPE_VALIDATION);
323                         }
324                         Psztracker_id = trackerid;
325
326                         //Build the request packet
327                         PacketHeader.type = UNT_LOGIN_AUTH_REQUEST;
328                         PacketHeader.len = PACKED_HEADER_ONLY_SIZE+sizeof(validate_id_request);
329                         ValidIDReq=(validate_id_request *)&PacketHeader.data;
330                         SDL_strlcpy(ValidIDReq->login, valid_id->login, SDL_arraysize(ValidIDReq->login));
331                         SDL_strlcpy(ValidIDReq->password, valid_id->password, SDL_arraysize(ValidIDReq->password));
332
333                         packet_length = SerializeValidatePacket(&PacketHeader, packet_data);
334                         SENDTO(Unreliable_socket, (char *)&packet_data, packet_length, 0, (struct sockaddr *)&rtrackaddr, sizeof(struct sockaddr), PSNET_TYPE_VALIDATION);
335                         ValidState = VALID_STATE_WAITING;
336                         ValidFirstSent = timer_get_milliseconds();
337                         ValidLastSent = timer_get_milliseconds();
338                         return 0;
339                 }
340                 else
341                 {
342                         return -3;
343                 }
344         }
345 }
346
347
348 void ValidIdle()
349 {
350         fd_set read_fds;                   
351         struct timeval timeout;
352         ubyte packet_data[sizeof(udp_packet_header)];
353         int packet_length = 0;
354
355         PSNET_TOP_LAYER_PROCESS();
356         
357         timeout.tv_sec=0;            
358         timeout.tv_usec=0;
359         
360         FD_ZERO(&read_fds);
361         FD_SET(Unreliable_socket, &read_fds);    
362
363         if(SELECT(Unreliable_socket+1,&read_fds,NULL,NULL,&timeout, PSNET_TYPE_VALIDATION)){
364                 int bytesin;
365                 int addrsize;
366                 struct sockaddr_in fromaddr;
367
368                 udp_packet_header inpacket;
369
370                 SDL_zero(inpacket);
371                 addrsize = sizeof(struct sockaddr_in);
372
373                 bytesin = RECVFROM(Unreliable_socket, (char *)&packet_data, sizeof(udp_packet_header), 0, (struct sockaddr *)&fromaddr, &addrsize, PSNET_TYPE_VALIDATION);
374
375                 if (bytesin > 0) {
376                         DeserializeValidatePacket(packet_data, bytesin, &inpacket);
377
378                         // decrease packet size by 1
379                         inpacket.len--;
380 #ifndef NDEBUG
381                 } else {
382                         int wserr=WSAGetLastError();
383                         mprintf(("recvfrom() failure. WSAGetLastError() returned %d\n",wserr));
384 #endif
385                 }
386
387                 FD_ZERO(&read_fds);
388                 FD_SET(Unreliable_socket, &read_fds);    
389
390                 //Check to make sure the packets ok
391                 if ( (bytesin > 0) && (bytesin == inpacket.len) ) {
392                         switch(inpacket.type)
393                         {
394                                 case UNT_LOGIN_NO_AUTH:
395                                         if(ValidState == VALID_STATE_WAITING)
396                                         {
397                                                 ValidState = VALID_STATE_INVALID;                                               
398                                         }
399                                         break;
400                                 case UNT_LOGIN_AUTHENTICATED:
401                                         if(ValidState == VALID_STATE_WAITING)
402                                         {
403                                                 ValidState = VALID_STATE_VALID;
404                                                 SDL_strlcpy(Psztracker_id, (const char *)&inpacket.data, TRACKER_ID_LEN);
405                                         }
406                                         break;
407                                 // old - this is a Freespace 1 packet type
408                                 case UNT_VALID_FS_MSN_RSP:
409                                         Int3();
410                                         break;
411
412                                 // fs2 mission validation response
413                                 case UNT_VALID_FS2_MSN_RSP:
414                                         if(MissionValidState == VALID_STATE_WAITING){
415                                                 if(inpacket.code==2){
416                                                         MissionValidState = VALID_STATE_VALID;
417                                                 } else {
418                                                         MissionValidState = VALID_STATE_INVALID;
419                                                 }
420                                         }
421                                         break;
422
423                                 // fs2 squad war validation response
424                                 case UNT_VALID_SW_MSN_RSP:
425                                         if(SquadWarValidState == VALID_STATE_WAITING){
426                                                 // copy the data
427                                                 SDL_assert((bytesin - PACKED_HEADER_ONLY_SIZE) == sizeof(squad_war_response));
428                                                 if((bytesin - PACKED_HEADER_ONLY_SIZE) == sizeof(squad_war_response)){
429                                                         memset(&SquadWarValidateResponse, 0, sizeof(squad_war_response));
430                                                         memcpy(&SquadWarValidateResponse, inpacket.data, sizeof(squad_war_response));
431
432                                                         // now check to see if we're good
433                                                         if(SquadWarValidateResponse.accepted){
434                                                                 SquadWarValidState = VALID_STATE_VALID;
435                                                         } else {
436                                                                 SquadWarValidState = VALID_STATE_INVALID;
437                                                         }
438                                                 } else {
439                                                         SquadWarValidState = VALID_STATE_INVALID;
440                                                 }                                               
441                                         }
442                                         break;
443
444                                 case UNT_CONTROL_VALIDATION:
445                                         Int3();
446                                         break;
447
448                                 case UNT_CONTROL:
449                                         Int3();
450                                         break;
451                         }
452                         AckValidServer(inpacket.sig);
453                 }
454         }
455
456         if(ValidState == VALID_STATE_WAITING)
457         {
458                 if((timer_get_milliseconds()-ValidFirstSent)>=PILOT_REQ_TIMEOUT)
459                 {
460                         ValidState = VALID_STATE_TIMEOUT;
461
462                 }               
463                 else if((timer_get_milliseconds()-ValidLastSent)>=PILOT_REQ_RESEND_TIME)
464                 {
465                         //Send 'da packet
466                         packet_length = SerializeValidatePacket(&PacketHeader, packet_data);
467                         SENDTO(Unreliable_socket, (char *)&packet_data, packet_length, 0, (struct sockaddr *)&rtrackaddr, sizeof(struct sockaddr), PSNET_TYPE_VALIDATION);
468                         ValidLastSent = timer_get_milliseconds();
469                 }
470         }
471 }
472
473
474 //Send an ACK to the server
475 void AckValidServer(unsigned int sig)
476 {
477         udp_packet_header ack_pack;
478         ubyte packet_data[sizeof(udp_packet_header)];
479         int packet_length = 0;
480
481         ack_pack.type = UNT_CONTROL;
482         ack_pack.sig = sig;
483         ack_pack.code = CMD_CLIENT_RECEIVED;
484         ack_pack.len = PACKED_HEADER_ONLY_SIZE;
485
486         packet_length = SerializeValidatePacket(&ack_pack, packet_data);
487         SDL_assert(packet_length == PACKED_HEADER_ONLY_SIZE);
488         SENDTO(Unreliable_socket, (char *)&packet_data, packet_length, 0, (struct sockaddr *)&rtrackaddr, sizeof(struct sockaddr_in), PSNET_TYPE_VALIDATION);
489 }
490
491 // call with a valid struct to validate a mission
492 // call with NULL to poll
493
494 // Return codes:
495 // -3   Still waiting (returned if we were waiting for a tracker response and ValidateMission was called with a non-NULL value
496 // -2 Timeout waiting for tracker to respond
497 // -1   User invalid
498 //  0   Still waiting for response from tracker/Idle
499 //  1   User valid
500 int ValidateMission(vmt_validate_mission_req_struct *valid_msn)
501 {
502         ubyte packet_data[sizeof(udp_packet_header)];
503         int packet_length = 0;
504
505         ValidIdle();
506         if(valid_msn==NULL)
507         {
508                 switch(MissionValidState)
509                 {
510                 case VALID_STATE_IDLE:
511                         return 0;
512                         break;
513                 case VALID_STATE_WAITING:
514                         return 0;
515                         break;                  
516                 case VALID_STATE_VALID:
517                         MissionValidState = VALID_STATE_IDLE;
518                         return 1;
519                         break;
520                 case VALID_STATE_INVALID:
521                         MissionValidState = VALID_STATE_IDLE;
522                         return -1;
523                 case VALID_STATE_TIMEOUT:
524                         MissionValidState = VALID_STATE_IDLE;
525                         return -2;
526                 }
527                 return 0;
528         }
529         else
530         {
531                 if(MissionValidState==VALID_STATE_IDLE)
532                 {
533                         //First, flush the input buffer for the socket
534                         fd_set read_fds;                   
535                         struct timeval timeout;
536                         
537                         timeout.tv_sec=0;            
538                         timeout.tv_usec=0;
539                         
540                         FD_ZERO(&read_fds);
541                         FD_SET(Unreliable_socket, &read_fds);    
542
543                         while(SELECT(Unreliable_socket+1,&read_fds,NULL,NULL,&timeout, PSNET_TYPE_VALIDATION))
544                         {
545                                 int addrsize;
546                                 struct sockaddr_in fromaddr;
547
548                                 udp_packet_header inpacket;
549                                 addrsize = sizeof(struct sockaddr_in);
550                                 RECVFROM(Unreliable_socket, (char *)&inpacket,sizeof(udp_packet_header),0,(struct sockaddr *)&fromaddr,&addrsize, PSNET_TYPE_VALIDATION);
551                                 FD_ZERO(&read_fds);
552                                 FD_SET(Unreliable_socket, &read_fds);    
553                         }
554                         //only send the header, the checksum and the string length plus the null
555                         PacketHeader.type = UNT_VALID_FS2_MSN_REQ;
556                         PacketHeader.len = (short)(PACKED_HEADER_ONLY_SIZE + sizeof(int)+1+strlen(valid_msn->file_name));
557                         memcpy(PacketHeader.data,valid_msn,PacketHeader.len-PACKED_HEADER_ONLY_SIZE);
558                         packet_length = SerializeValidatePacket(&PacketHeader, packet_data);
559                         SENDTO(Unreliable_socket, (char *)&packet_data, packet_length, 0, (struct sockaddr *)&rtrackaddr, sizeof(struct sockaddr), PSNET_TYPE_VALIDATION);
560                         MissionValidState = VALID_STATE_WAITING;
561                         MissionValidFirstSent = timer_get_milliseconds();
562                         MissionValidLastSent = timer_get_milliseconds();
563                         return 0;
564                 }
565                 else
566                 {
567                         return -3;
568                 }
569         }
570 }
571
572 // query the usertracker to validate a squad war match
573 // call with a valid struct to validate a mission
574 // call with NULL to poll
575
576 // Return codes:
577 // -3   Still waiting (returned if we were waiting for a tracker response and ValidateSquadWae was called with a non-NULL value
578 // -2 Timeout waiting for tracker to respond
579 // -1   match invalid
580 //  0   Still waiting for response from tracker/Idle
581 //  1   match valid
582 int ValidateSquadWar(squad_war_request *sw_req, squad_war_response *sw_resp)
583 {
584         ubyte packet_data[sizeof(udp_packet_header)];
585         int packet_length = 0;
586
587         ValidIdle();
588         if(sw_req==NULL){
589                 switch(SquadWarValidState){
590                 case VALID_STATE_IDLE:
591                         return 0;
592                         break;
593                 case VALID_STATE_WAITING:
594                         return 0;
595                         break;
596
597                 // fill in the response
598                 case VALID_STATE_VALID:
599                         SquadWarValidState = VALID_STATE_IDLE;
600                         if(sw_resp != NULL){
601                                 memcpy(sw_resp, &SquadWarValidateResponse, sizeof(squad_war_response));
602                         }
603                         return 1;
604                         break;
605                 // fill in the response
606                 case VALID_STATE_INVALID:
607                         SquadWarValidState = VALID_STATE_IDLE;
608                         if(sw_resp != NULL){
609                                 memcpy(sw_resp, &SquadWarValidateResponse, sizeof(squad_war_response));
610                         }
611                         return -1;
612
613                 case VALID_STATE_TIMEOUT:
614                         SquadWarValidState = VALID_STATE_IDLE;
615                         return -2;
616                 }
617                 return 0;
618         } else {
619                 if(SquadWarValidState==VALID_STATE_IDLE){
620                         // First, flush the input buffer for the socket
621                         fd_set read_fds;                   
622                         struct timeval timeout;
623                         
624                         timeout.tv_sec=0;            
625                         timeout.tv_usec=0;
626                         
627                         FD_ZERO(&read_fds);
628                         FD_SET(Unreliable_socket, &read_fds);    
629
630                         while(SELECT(Unreliable_socket+1,&read_fds,NULL,NULL,&timeout, PSNET_TYPE_VALIDATION)){
631                                 int addrsize;
632                                 struct sockaddr_in fromaddr;
633
634                                 udp_packet_header inpacket;
635                                 addrsize = sizeof(struct sockaddr_in);
636                                 RECVFROM(Unreliable_socket, (char *)&inpacket,sizeof(udp_packet_header),0,(struct sockaddr *)&fromaddr,&addrsize, PSNET_TYPE_VALIDATION);
637                                 FD_ZERO(&read_fds);
638                                 FD_SET(Unreliable_socket, &read_fds);    
639                         }
640                         // only send the header, the checksum and the string length plus the null
641                         PacketHeader.type = UNT_VALID_SW_MSN_REQ;
642                         PacketHeader.len = (short)(PACKED_HEADER_ONLY_SIZE + sizeof(squad_war_request));
643                         memcpy(PacketHeader.data, sw_req, PacketHeader.len-PACKED_HEADER_ONLY_SIZE);
644                         packet_length = SerializeValidatePacket(&PacketHeader, packet_data);
645                         SENDTO(Unreliable_socket, (char *)&packet_data, packet_length, 0, (struct sockaddr *)&rtrackaddr, sizeof(struct sockaddr), PSNET_TYPE_VALIDATION);
646                         SquadWarValidState = VALID_STATE_WAITING;
647                         SquadWarFirstSent = timer_get_milliseconds();
648                         SquadWarLastSent = timer_get_milliseconds();
649                         return 0;
650                 } else {
651                         return -3;
652                 }
653         }
654 }