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
|
/* ****************************************************************************
* eID Middleware Project.
* Copyright (C) 2008-2009 FedICT.
*
* This is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License version
* 3.0 as published by the Free Software Foundation.
*
* This software is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this software; if not, see
* http://www.gnu.org/licenses/.
**************************************************************************** */
#include "SocketServer.h"
#include "../eidErrors.h"
#include "../Log.h"
#include <errno.h>
#ifndef WIN32
#include <netinet/in.h>
#include <sys/ioctl.h>
#include <errno.h>
#ifndef SOCKET_ERROR
#define SOCKET_ERROR -1
#endif
#endif
using namespace std;
namespace eIDMW
{
CSocketServer::CSocketServer(int port, int connections, e_socketType type) throw(CMWException)
: CSocket()
{
sockaddr_in address;
memset(&address, 0, sizeof(address));
address.sin_family = PF_INET;
address.sin_port = htons(port);
m_socket = socket(AF_INET, SOCK_STREAM, 0);
if (m_socket == INVALID_SOCKET)
{
MWLOG(LEV_ERROR, MOD_CAL, L" CSocketServer: can't open a socket on port %s ", port);
throw CMWEXCEPTION(EIDMW_ERR_SOCKET_SOCKET);
}
if (type == NonBlockingSocket)
{
// enable non-blocking mode
unsigned long arg = 1;
#ifdef WIN32
ioctlsocket(m_socket, FIONBIO, &arg);
#else
if(ioctl(m_socket, FIONBIO, &arg) != 0)
{
MWLOG(LEV_ERROR,MOD_CAL,L" CSocketServer: ioctl failed, %s",strerror(errno));
throw CMWEXCEPTION(EIDMW_ERR_SOCKET_SOCKET);
}
#endif
}
if (bind(m_socket, (sockaddr *)&address, sizeof(sockaddr_in) ) == SOCKET_ERROR)
{
#ifdef WIN32
closesocket(m_socket);
#else
close(m_socket);
#endif
MWLOG(LEV_ERROR, MOD_CAL, L" CSocketServer: error in binding socket at port %d ", port);
throw CMWEXCEPTION(EIDMW_ERR_SOCKET_BIND);
}
listen(m_socket, connections == 0 ? SOMAXCONN : connections);
}
CSocketServer::~CSocketServer()
{
if (m_socket != INVALID_SOCKET)
{
#ifdef WIN32
closesocket(m_socket);
#else
close(m_socket);
#endif
}
}
CSocket *CSocketServer::Accept(struct sockaddr *outSockAddr, int *ioSockAddrLen) throw(CMWException)
{
socklen_t tSockAddrLen = *ioSockAddrLen;
SOCKET new_socket = accept(m_socket, outSockAddr, &tSockAddrLen);
if (new_socket == INVALID_SOCKET)
{
#ifdef WIN32
bool error = (WSAGetLastError() == WSAEWOULDBLOCK) ;
#else
bool error = (errno == EWOULDBLOCK);
#endif
if (error)
{
return 0;
}
else
{
throw CMWEXCEPTION(EIDMW_ERR_SOCKET_ACCEPT);
}
}
CSocket *result = new CSocket(new_socket);
return result;
}
} // namespace eIDMW
|
