]> icculus.org git repositories - taylor/freespace2.git/blob - src/network/valid.cpp
silence various compiler warnings (clang:rel)
[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                 addrsize = sizeof(struct sockaddr_in);
370
371                 bytesin = RECVFROM(Unreliable_socket, (char *)&packet_data, sizeof(udp_packet_header), 0, (struct sockaddr *)&fromaddr, &addrsize, PSNET_TYPE_VALIDATION);
372
373                 if (bytesin > 0) {
374                         DeserializeValidatePacket(packet_data, bytesin, &inpacket);
375
376                         // decrease packet size by 1
377                         inpacket.len--;
378 #ifndef NDEBUG
379                 } else {
380                         int wserr=WSAGetLastError();
381                         mprintf(("recvfrom() failure. WSAGetLastError() returned %d\n",wserr));
382 #endif
383                 }
384
385                 FD_ZERO(&read_fds);
386                 FD_SET(Unreliable_socket, &read_fds);    
387
388                 //Check to make sure the packets ok
389                 if ( (bytesin > 0) && (bytesin == inpacket.len) ) {
390                         switch(inpacket.type)
391                         {
392                                 case UNT_LOGIN_NO_AUTH:
393                                         if(ValidState == VALID_STATE_WAITING)
394                                         {
395                                                 ValidState = VALID_STATE_INVALID;                                               
396                                         }
397                                         break;
398                                 case UNT_LOGIN_AUTHENTICATED:
399                                         if(ValidState == VALID_STATE_WAITING)
400                                         {
401                                                 ValidState = VALID_STATE_VALID;
402                                                 SDL_strlcpy(Psztracker_id, (const char *)&inpacket.data, TRACKER_ID_LEN);
403                                         }
404                                         break;
405                                 // old - this is a Freespace 1 packet type
406                                 case UNT_VALID_FS_MSN_RSP:
407                                         Int3();
408                                         break;
409
410                                 // fs2 mission validation response
411                                 case UNT_VALID_FS2_MSN_RSP:
412                                         if(MissionValidState == VALID_STATE_WAITING){
413                                                 if(inpacket.code==2){
414                                                         MissionValidState = VALID_STATE_VALID;
415                                                 } else {
416                                                         MissionValidState = VALID_STATE_INVALID;
417                                                 }
418                                         }
419                                         break;
420
421                                 // fs2 squad war validation response
422                                 case UNT_VALID_SW_MSN_RSP:
423                                         if(SquadWarValidState == VALID_STATE_WAITING){
424                                                 // copy the data
425                                                 SDL_assert((bytesin - PACKED_HEADER_ONLY_SIZE) == sizeof(squad_war_response));
426                                                 if((bytesin - PACKED_HEADER_ONLY_SIZE) == sizeof(squad_war_response)){
427                                                         memset(&SquadWarValidateResponse, 0, sizeof(squad_war_response));
428                                                         memcpy(&SquadWarValidateResponse, inpacket.data, sizeof(squad_war_response));
429
430                                                         // now check to see if we're good
431                                                         if(SquadWarValidateResponse.accepted){
432                                                                 SquadWarValidState = VALID_STATE_VALID;
433                                                         } else {
434                                                                 SquadWarValidState = VALID_STATE_INVALID;
435                                                         }
436                                                 } else {
437                                                         SquadWarValidState = VALID_STATE_INVALID;
438                                                 }                                               
439                                         }
440                                         break;
441
442                                 case UNT_CONTROL_VALIDATION:
443                                         Int3();
444                                         break;
445
446                                 case UNT_CONTROL:
447                                         Int3();
448                                         break;
449                         }
450                         AckValidServer(inpacket.sig);
451                 }
452         }
453
454         if(ValidState == VALID_STATE_WAITING)
455         {
456                 if((timer_get_milliseconds()-ValidFirstSent)>=PILOT_REQ_TIMEOUT)
457                 {
458                         ValidState = VALID_STATE_TIMEOUT;
459
460                 }               
461                 else if((timer_get_milliseconds()-ValidLastSent)>=PILOT_REQ_RESEND_TIME)
462                 {
463                         //Send 'da packet
464                         packet_length = SerializeValidatePacket(&PacketHeader, packet_data);
465                         SENDTO(Unreliable_socket, (char *)&packet_data, packet_length, 0, (struct sockaddr *)&rtrackaddr, sizeof(struct sockaddr), PSNET_TYPE_VALIDATION);
466                         ValidLastSent = timer_get_milliseconds();
467                 }
468         }
469 }
470
471
472 //Send an ACK to the server
473 void AckValidServer(unsigned int sig)
474 {
475         udp_packet_header ack_pack;
476         ubyte packet_data[sizeof(udp_packet_header)];
477         int packet_length = 0;
478
479         ack_pack.type = UNT_CONTROL;
480         ack_pack.sig = sig;
481         ack_pack.code = CMD_CLIENT_RECEIVED;
482         ack_pack.len = PACKED_HEADER_ONLY_SIZE;
483
484         packet_length = SerializeValidatePacket(&ack_pack, packet_data);
485         SDL_assert(packet_length == PACKED_HEADER_ONLY_SIZE);
486         SENDTO(Unreliable_socket, (char *)&packet_data, packet_length, 0, (struct sockaddr *)&rtrackaddr, sizeof(struct sockaddr_in), PSNET_TYPE_VALIDATION);
487 }
488
489 // call with a valid struct to validate a mission
490 // call with NULL to poll
491
492 // Return codes:
493 // -3   Still waiting (returned if we were waiting for a tracker response and ValidateMission was called with a non-NULL value
494 // -2 Timeout waiting for tracker to respond
495 // -1   User invalid
496 //  0   Still waiting for response from tracker/Idle
497 //  1   User valid
498 int ValidateMission(vmt_validate_mission_req_struct *valid_msn)
499 {
500         ubyte packet_data[sizeof(udp_packet_header)];
501         int packet_length = 0;
502
503         ValidIdle();
504         if(valid_msn==NULL)
505         {
506                 switch(MissionValidState)
507                 {
508                 case VALID_STATE_IDLE:
509                         return 0;
510                         break;
511                 case VALID_STATE_WAITING:
512                         return 0;
513                         break;                  
514                 case VALID_STATE_VALID:
515                         MissionValidState = VALID_STATE_IDLE;
516                         return 1;
517                         break;
518                 case VALID_STATE_INVALID:
519                         MissionValidState = VALID_STATE_IDLE;
520                         return -1;
521                 case VALID_STATE_TIMEOUT:
522                         MissionValidState = VALID_STATE_IDLE;
523                         return -2;
524                 }
525                 return 0;
526         }
527         else
528         {
529                 if(MissionValidState==VALID_STATE_IDLE)
530                 {
531                         //First, flush the input buffer for the socket
532                         fd_set read_fds;                   
533                         struct timeval timeout;
534                         
535                         timeout.tv_sec=0;            
536                         timeout.tv_usec=0;
537                         
538                         FD_ZERO(&read_fds);
539                         FD_SET(Unreliable_socket, &read_fds);    
540
541                         while(SELECT(Unreliable_socket+1,&read_fds,NULL,NULL,&timeout, PSNET_TYPE_VALIDATION))
542                         {
543                                 int addrsize;
544                                 struct sockaddr_in fromaddr;
545
546                                 udp_packet_header inpacket;
547                                 addrsize = sizeof(struct sockaddr_in);
548                                 RECVFROM(Unreliable_socket, (char *)&inpacket,sizeof(udp_packet_header),0,(struct sockaddr *)&fromaddr,&addrsize, PSNET_TYPE_VALIDATION);
549                                 FD_ZERO(&read_fds);
550                                 FD_SET(Unreliable_socket, &read_fds);    
551                         }
552                         //only send the header, the checksum and the string length plus the null
553                         PacketHeader.type = UNT_VALID_FS2_MSN_REQ;
554                         PacketHeader.len = (short)(PACKED_HEADER_ONLY_SIZE + sizeof(int)+1+strlen(valid_msn->file_name));
555                         memcpy(PacketHeader.data,valid_msn,PacketHeader.len-PACKED_HEADER_ONLY_SIZE);
556                         packet_length = SerializeValidatePacket(&PacketHeader, packet_data);
557                         SENDTO(Unreliable_socket, (char *)&packet_data, packet_length, 0, (struct sockaddr *)&rtrackaddr, sizeof(struct sockaddr), PSNET_TYPE_VALIDATION);
558                         MissionValidState = VALID_STATE_WAITING;
559                         MissionValidFirstSent = timer_get_milliseconds();
560                         MissionValidLastSent = timer_get_milliseconds();
561                         return 0;
562                 }
563                 else
564                 {
565                         return -3;
566                 }
567         }
568 }
569
570 // query the usertracker to validate a squad war match
571 // call with a valid struct to validate a mission
572 // call with NULL to poll
573
574 // Return codes:
575 // -3   Still waiting (returned if we were waiting for a tracker response and ValidateSquadWae was called with a non-NULL value
576 // -2 Timeout waiting for tracker to respond
577 // -1   match invalid
578 //  0   Still waiting for response from tracker/Idle
579 //  1   match valid
580 int ValidateSquadWar(squad_war_request *sw_req, squad_war_response *sw_resp)
581 {
582         ubyte packet_data[sizeof(udp_packet_header)];
583         int packet_length = 0;
584
585         ValidIdle();
586         if(sw_req==NULL){
587                 switch(SquadWarValidState){
588                 case VALID_STATE_IDLE:
589                         return 0;
590                         break;
591                 case VALID_STATE_WAITING:
592                         return 0;
593                         break;
594
595                 // fill in the response
596                 case VALID_STATE_VALID:
597                         SquadWarValidState = VALID_STATE_IDLE;
598                         if(sw_resp != NULL){
599                                 memcpy(sw_resp, &SquadWarValidateResponse, sizeof(squad_war_response));
600                         }
601                         return 1;
602                         break;
603                 // fill in the response
604                 case VALID_STATE_INVALID:
605                         SquadWarValidState = VALID_STATE_IDLE;
606                         if(sw_resp != NULL){
607                                 memcpy(sw_resp, &SquadWarValidateResponse, sizeof(squad_war_response));
608                         }
609                         return -1;
610
611                 case VALID_STATE_TIMEOUT:
612                         SquadWarValidState = VALID_STATE_IDLE;
613                         return -2;
614                 }
615                 return 0;
616         } else {
617                 if(SquadWarValidState==VALID_STATE_IDLE){
618                         // First, flush the input buffer for the socket
619                         fd_set read_fds;                   
620                         struct timeval timeout;
621                         
622                         timeout.tv_sec=0;            
623                         timeout.tv_usec=0;
624                         
625                         FD_ZERO(&read_fds);
626                         FD_SET(Unreliable_socket, &read_fds);    
627
628                         while(SELECT(Unreliable_socket+1,&read_fds,NULL,NULL,&timeout, PSNET_TYPE_VALIDATION)){
629                                 int addrsize;
630                                 struct sockaddr_in fromaddr;
631
632                                 udp_packet_header inpacket;
633                                 addrsize = sizeof(struct sockaddr_in);
634                                 RECVFROM(Unreliable_socket, (char *)&inpacket,sizeof(udp_packet_header),0,(struct sockaddr *)&fromaddr,&addrsize, PSNET_TYPE_VALIDATION);
635                                 FD_ZERO(&read_fds);
636                                 FD_SET(Unreliable_socket, &read_fds);    
637                         }
638                         // only send the header, the checksum and the string length plus the null
639                         PacketHeader.type = UNT_VALID_SW_MSN_REQ;
640                         PacketHeader.len = (short)(PACKED_HEADER_ONLY_SIZE + sizeof(squad_war_request));
641                         memcpy(PacketHeader.data, sw_req, PacketHeader.len-PACKED_HEADER_ONLY_SIZE);
642                         packet_length = SerializeValidatePacket(&PacketHeader, packet_data);
643                         SENDTO(Unreliable_socket, (char *)&packet_data, packet_length, 0, (struct sockaddr *)&rtrackaddr, sizeof(struct sockaddr), PSNET_TYPE_VALIDATION);
644                         SquadWarValidState = VALID_STATE_WAITING;
645                         SquadWarFirstSent = timer_get_milliseconds();
646                         SquadWarLastSent = timer_get_milliseconds();
647                         return 0;
648                 } else {
649                         return -3;
650                 }
651         }
652 }