| 16 |
***************************************************************************/ |
***************************************************************************/ |
| 17 |
|
|
| 18 |
#include "common.h" |
#include "common.h" |
| 19 |
|
#include "io.h" |
| 20 |
|
#include "tcplib.h" |
| 21 |
|
#include <sys/socket.h> |
| 22 |
|
#include <netinet/in.h> |
| 23 |
|
#include <arpa/inet.h> |
| 24 |
|
|
| 25 |
int |
int |
| 26 |
net_server (const char *hostaddr, unsigned int port) |
net_server (const char *hostaddr, unsigned int port) |
| 27 |
{ |
{ |
| 28 |
int sock_server, sock_client, namelen, seq, netint; |
int namelen, seq, netint, result, flags; |
| 29 |
struct sockaddr_in sin; |
struct sockaddr_in sin; |
| 30 |
char temp[256]; |
char temp[256]; |
| 31 |
|
fd_set testfds; |
| 32 |
|
struct timeval timeout; |
| 33 |
|
|
| 34 |
sock_server = socket (AF_INET, SOCK_STREAM, IPPROTO_TCP); |
socket_server = socket (AF_INET, SOCK_STREAM, IPPROTO_TCP); |
| 35 |
|
|
| 36 |
if (sock_server < 0) |
if (socket_server < 0) |
| 37 |
{ |
{ |
| 38 |
log_error ("Create socket failed\n"); |
log_error ("Create socket failed\n"); |
| 39 |
exit (1); |
exit (1); |
| 44 |
(strlen (hostaddr) > 0 ? inet_addr (hostaddr) : INADDR_ANY); |
(strlen (hostaddr) > 0 ? inet_addr (hostaddr) : INADDR_ANY); |
| 45 |
sin.sin_port = htons (port); |
sin.sin_port = htons (port); |
| 46 |
|
|
| 47 |
if (bind (sock_server, (struct sockaddr *)&sin, sizeof(sin)) < 0) |
if (bind (socket_server, (struct sockaddr *) &sin, sizeof (sin)) < 0) |
| 48 |
{ |
{ |
| 49 |
sprintf(temp, "Bind address %s:%u failed\n", |
log_error ("Bind address %s:%u failed\n", |
| 50 |
inet_ntoa(sin.sin_addr), ntohs(sin.sin_port)); |
inet_ntoa (sin.sin_addr), ntohs (sin.sin_port)); |
|
log_error (temp); |
|
| 51 |
exit (2); |
exit (2); |
| 52 |
} |
} |
| 53 |
|
|
| 54 |
if (listen (sock_server, 10) < 0) |
if (listen (socket_server, 10) < 0) |
| 55 |
{ |
{ |
| 56 |
log_error ("Socket listen failed\n"); |
log_error ("Socket listen failed\n"); |
| 57 |
exit (3); |
exit (3); |
| 58 |
} |
} |
| 59 |
|
|
| 60 |
socket_server = sock_server; |
strcpy (hostaddr_server, inet_ntoa (sin.sin_addr)); |
| 61 |
strcpy(hostaddr_server,(const char *)inet_ntoa(sin.sin_addr)); |
port_server = ntohs (sin.sin_port); |
| 62 |
port_server = ntohs(sin.sin_port); |
|
| 63 |
|
log_std ("Listening at %s:%d\n", hostaddr_server, port_server); |
|
sprintf(temp, "Listening at %s:%d\n", |
|
|
hostaddr_server, port_server); |
|
|
log_std (temp); |
|
| 64 |
|
|
| 65 |
namelen = sizeof (sin); |
namelen = sizeof (sin); |
| 66 |
while(1) |
while (!SYS_exit) |
| 67 |
{ |
{ |
| 68 |
if ((sock_client = accept (sock_server, (struct sockaddr *) &sin, &namelen)) < 0) |
FD_ZERO (&testfds); |
| 69 |
{ |
FD_SET (socket_server, &testfds); |
| 70 |
log_error ("Accept connection error\n"); |
|
| 71 |
} |
timeout.tv_sec = 1; |
| 72 |
else |
timeout.tv_usec = 0; |
| 73 |
{ |
|
| 74 |
sprintf(temp, "Accept connection from %s:%d\n", |
result = SignalSafeSelect (FD_SETSIZE, &testfds, NULL, NULL, &timeout); |
| 75 |
inet_ntoa(sin.sin_addr),ntohs(sin.sin_port)); |
if (result < 0) |
| 76 |
log_std (temp); |
{ |
| 77 |
} |
log_error ("Accept connection error\n"); |
| 78 |
|
continue; |
| 79 |
if (fork_server(sock_server, sock_client, &sin)<0) |
} |
| 80 |
{ |
|
| 81 |
log_error ("Fork error\n"); |
if (result == 0) |
| 82 |
} |
{ |
| 83 |
|
continue; |
| 84 |
if (close(sock_client) == -1) |
} |
| 85 |
|
|
| 86 |
|
if (FD_ISSET (socket_server, &testfds)) |
| 87 |
{ |
{ |
| 88 |
log_error("Close client socket failed\n"); |
flags = fcntl (socket_server, F_GETFL, 0); |
| 89 |
|
fcntl (socket_server, F_SETFL, flags | O_NONBLOCK); |
| 90 |
|
while ((socket_client = |
| 91 |
|
accept (socket_server, (struct sockaddr *) &sin, &namelen)) < 0) |
| 92 |
|
{ |
| 93 |
|
if (errno != EWOULDBLOCK && errno != ECONNABORTED && errno != EINTR) |
| 94 |
|
{ |
| 95 |
|
log_error ("Accept connection error\n"); |
| 96 |
|
break; |
| 97 |
|
} |
| 98 |
|
} |
| 99 |
|
fcntl (socket_server, F_SETFL, flags); |
| 100 |
} |
} |
| 101 |
|
|
| 102 |
|
if (socket_client < 0) |
| 103 |
|
{ |
| 104 |
|
log_error ("Accept connection error\n"); |
| 105 |
|
continue; |
| 106 |
|
} |
| 107 |
|
|
| 108 |
|
strcpy (hostaddr_client, (const char *) inet_ntoa (sin.sin_addr)); |
| 109 |
|
port_client = ntohs (sin.sin_port); |
| 110 |
|
|
| 111 |
|
log_std ("Accept connection from %s:%d\n", hostaddr_client, |
| 112 |
|
port_client); |
| 113 |
|
|
| 114 |
|
if (fork_server () < 0) |
| 115 |
|
{ |
| 116 |
|
log_error ("Fork error\n"); |
| 117 |
|
} |
| 118 |
|
|
| 119 |
|
if (close (socket_client) == -1) |
| 120 |
|
{ |
| 121 |
|
log_error ("Close client socket failed\n"); |
| 122 |
|
} |
| 123 |
|
} |
| 124 |
|
|
| 125 |
|
if (close (socket_server) == -1) |
| 126 |
|
{ |
| 127 |
|
log_error ("Close server socket failed\n"); |
| 128 |
} |
} |
| 129 |
|
|
| 130 |
return 0; |
return 0; |