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
|
#include "server.hpp"
int server::c_server::init( ) {
#if WIN32
if( WSAStartup( MAKEWORD( 2, 2 ), &m_sock_data ) ) // windows
return 1;
#endif
m_socket = socket( AF_INET, SOCK_STREAM, 0 );
if( m_socket == INVALID_SOCKET )
return 2;
sockaddr_in server_address{ };
server_address.sin_addr.s_addr = INADDR_ANY;
server_address.sin_port = htons( PORT_NUM );
server_address.sin_family = AF_INET;
auto result = ::bind( m_socket, ( sockaddr* )( &server_address ), sizeof( server_address ) );
return result != -1 ? 0 : 3;
}
void server::c_server::listen( ) {
::listen( m_socket, 5 );
sockaddr_in client_address{ };
// not needed, this is set in accept, values are input.
// client_address.sin_port = htons( PORT_NUM );
int len = sizeof( client_address );
SOCKET result = accept( m_socket, ( sockaddr* )( &client_address ), &len );
if( result == INVALID_SOCKET ) {
printf( "socket error accepting a connection\n" );
return;
}
printf( "incoming connection from: %s\n", inet_ntoa( client_address.sin_addr ) );
// connection established.
std::lock_guard lock( m_mutex );
auto client = std::make_shared< c_client >( result, client_address.sin_addr );
client->send_message( "hello" );
m_clients.push_back( client );
}
void server::c_server::client_loop( ) {
if( m_clients.size( ) ) {
std::lock_guard lock( m_mutex );
for( auto it = m_clients.begin( ); it != m_clients.end( ); ++it ) {
if( !( *it )->handle( ) ) {
printf( "%s disconnected\n", ( *it )->get_ip( ) );
m_clients.erase( it );
break;
}
}
}
}
|
