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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
|
#include <sys/types.h>
#include <sys/socket.h>
#include <netdb.h>
#include <string.h>
#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <assert.h>
#include "net.h"
/**
* Server side function; start server on specified port
* @param[in] port Port on which server should listen
* @return Listening socket
*/
int server_start(const char* port)
{
int status;
int serversock;
struct addrinfo hints, *result, *tmp;
memset(&hints, 0, sizeof(struct addrinfo));
hints.ai_family = AF_UNSPEC; // IPv4 or IPv6
hints.ai_socktype = SOCK_STREAM; // TCP socket
hints.ai_flags = AI_PASSIVE; // wildcard IP
hints.ai_protocol = 0; // any protocol
status = getaddrinfo(NULL, port, &hints, &result);
if(status != 0)
{
printf("getaddrinfo: %s\n", gai_strerror(status));
exit(EXIT_FAILURE);
}
// iterate over linked addrinfo list and use first useable entry
for(tmp = result; tmp != NULL; tmp = tmp->ai_next)
{
int yes=1;
// create socket
serversock = socket(tmp->ai_family, tmp->ai_socktype, tmp->ai_protocol);
if(serversock == -1)
continue;
// try to reuse still open sockets in TIME_WAIT state
setsockopt(serversock, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(int));
// try to bind to address/port
if(bind(serversock, tmp->ai_addr, tmp->ai_addrlen) == 0)
break; // Success!
close(serversock);
}
if(tmp == NULL)
{
printf("failed to bind\n");
exit(EXIT_FAILURE);
}
freeaddrinfo(result);
// start listening
status = listen(serversock, 1);
if(status == -1)
{
printf("listen: %s\n", strerror(status));
exit(EXIT_FAILURE);
}
return serversock;
}
/**
* Server side function; accepts connections from clients
* @param[in] serversock Socket on which server is listening
* @param[in] count Number of clients that should connect
* @return List of $count open sockets with connections to clients
*/
int* server_get_players(int serversock, const uint8_t count)
{
int* clientsocks;
int i;
assert(count < MAX_PLAYERS && count > 0);
clientsocks = malloc(count*sizeof(int));
assert(clientsocks != NULL);
// accept connections
for(i=0; i<count; i++)
{
int sock;
struct sockaddr_storage addr;
socklen_t addrlen = sizeof(addr);
sock = accept(serversock, (struct sockaddr*) &addr, &addrlen);
if(sock == -1)
{
printf("accept: %s\n", strerror(sock));
exit(EXIT_FAILURE);
}
//printf("new client connected: %s\n", inet_ntop(sock.ss_family, get_in_addr((struct sockaddr*)&sock), INET6_ADDRSTRLEN));
printf("new client connected (%d/%d)\n", i+1, count);
clientsocks[i] = sock;
}
return clientsocks;
}
/**
* Client side function; connects to specified host:port
* @param[in] host Hostname of server
* @param[in] port Port of server
* @return Socket with open connection to server
*/
int client_connect_server(const char* host, const char* port)
{
int status;
int sock;
struct addrinfo hints, *result, *tmp;
memset(&hints, 0, sizeof(hints));
hints.ai_family = AF_UNSPEC; // IPv4 or IPv6
hints.ai_socktype = SOCK_STREAM; // TCP socket
status = getaddrinfo(host, port, &hints, &result);
if(status != 0)
{
printf("getaddrinfo: %s\n", gai_strerror(status));
exit(EXIT_FAILURE);
}
// connect to first result in linked addrinfo list
for(tmp = result; tmp != NULL; tmp = tmp->ai_next)
{
// create socket
sock = socket(tmp->ai_family, tmp->ai_socktype, tmp->ai_protocol);
if(sock == -1)
continue;
// connect!
if(connect(sock, tmp->ai_addr, tmp->ai_addrlen) != -1)
break; // Success!
close(sock);
}
if(tmp == NULL)
{
printf("failed to connect\n");
exit(EXIT_FAILURE);
}
freeaddrinfo(result);
return sock;
}
|
