1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
|
#include <sys/types.h>
#include <sys/socket.h>
#include <stdlib.h>
#include <stdio.h>
#include <assert.h>
#include "net.h"
void *net_recv(int sock, msg_type_t type)
{
msg_t m;
void *result;
ssize_t len = recv(sock, &m.hdr, sizeof(msg_header_t), MSG_PEEK); // just peek into packet to determine message header
assert(len != -1);
if(m.hdr.type != type)
{
printf("net_recv: received message type %d instead of %d", m.hdr.type, type);
return NULL;
}
m.payload = malloc(m.hdr.payload_length); // Allocate space for message payload
recv(sock, &m.hdr, sizeof(msg_header_t), 0); // Remove message header from socket
recv(sock, m.payload, m.hdr.payload_length, 0);// And then receive the payload
switch(type)
{
case msg_type_hello:
result = server_recv_hello(m.payload, m.hdr.payload_length);
break;
case msg_type_start_game:
result = client_recv_player_list(m.payload, m.hdr.payload_length);
break;
case msg_type_deal_cards:
result = client_recv_deal_cards(m.payload, m.hdr.payload_length);
break;
case msg_type_selected_card:
result = server_recv_selected_card(m.payload, m.hdr.payload_length);
break;
default:
printf("net_recv: Unknown message type %d received!\n", type);
exit(EXIT_FAILURE);
break;
}
free(m.payload);
return result;
}
|
