mirror of
https://github.com/Wind4/vlmcsd.git
synced 2025-05-24 19:54:21 -04:00
vlmcsd-svn812-2015-08-30-Hotbird64
This commit is contained in:
Wind4
commit
1af203d2a8
133 changed files with 102338 additions and 0 deletions
675
network.c
Normal file
675
network.c
Normal file
|
|
@ -0,0 +1,675 @@
|
|||
#ifndef CONFIG
|
||||
#define CONFIG "config.h"
|
||||
#endif // CONFIG
|
||||
#include CONFIG
|
||||
|
||||
#ifndef USE_MSRPC
|
||||
|
||||
#ifndef _GNU_SOURCE
|
||||
#define _GNU_SOURCE
|
||||
#endif
|
||||
|
||||
#include <string.h>
|
||||
#ifndef _WIN32
|
||||
#include <signal.h>
|
||||
#include <unistd.h>
|
||||
#include <fcntl.h>
|
||||
#include <errno.h>
|
||||
#include <netinet/in.h>
|
||||
#endif // WIN32
|
||||
|
||||
#include "network.h"
|
||||
#include "endian.h"
|
||||
#include "output.h"
|
||||
#include "helpers.h"
|
||||
#include "shared_globals.h"
|
||||
#include "rpc.h"
|
||||
|
||||
|
||||
#ifndef _WIN32
|
||||
typedef ssize_t (*sendrecv_t)(int, void*, size_t, int);
|
||||
#else
|
||||
typedef int (WINAPI *sendrecv_t)(SOCKET, void*, int, int);
|
||||
#endif
|
||||
|
||||
|
||||
// Send or receive a fixed number of bytes regardless if received in one or more chunks
|
||||
int_fast8_t sendrecv(SOCKET sock, BYTE *data, int len, int_fast8_t do_send)
|
||||
{
|
||||
int n;
|
||||
sendrecv_t f = do_send
|
||||
? (sendrecv_t) send
|
||||
: (sendrecv_t) recv;
|
||||
|
||||
do
|
||||
{
|
||||
n = f(sock, data, len, 0);
|
||||
}
|
||||
while (
|
||||
( n < 0 && socket_errno == VLMCSD_EINTR ) || ( n > 0 && ( data += n, (len -= n) > 0 ) ));
|
||||
|
||||
return ! len;
|
||||
}
|
||||
|
||||
|
||||
static int_fast8_t ip2str(char *restrict result, const size_t resultLength, const struct sockaddr *const restrict socketAddress, const socklen_t socketLength)
|
||||
{
|
||||
static const char *const fIPv4 = "%s:%s";
|
||||
static const char *const fIPv6 = "[%s]:%s";
|
||||
char ipAddress[64], portNumber[8];
|
||||
|
||||
if (getnameinfo
|
||||
(
|
||||
socketAddress,
|
||||
socketLength,
|
||||
ipAddress,
|
||||
sizeof(ipAddress),
|
||||
portNumber,
|
||||
sizeof(portNumber),
|
||||
NI_NUMERICHOST | NI_NUMERICSERV
|
||||
))
|
||||
{
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
if ((unsigned int)snprintf(result, resultLength, socketAddress->sa_family == AF_INET6 ? fIPv6 : fIPv4, ipAddress, portNumber) > resultLength) return FALSE;
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
|
||||
static int_fast8_t getSocketList(struct addrinfo **saList, const char *const addr, const int flags, const int AddressFamily)
|
||||
{
|
||||
int status;
|
||||
char *szHost, *szPort;
|
||||
size_t len = strlen(addr) + 1;
|
||||
|
||||
// Don't alloca too much
|
||||
if (len > 264) return FALSE;
|
||||
|
||||
char *addrcopy = (char*)alloca(len);
|
||||
memcpy(addrcopy, addr, len);
|
||||
|
||||
parseAddress(addrcopy, &szHost, &szPort);
|
||||
|
||||
struct addrinfo hints;
|
||||
|
||||
memset(&hints, 0, sizeof(struct addrinfo));
|
||||
|
||||
hints.ai_family = AddressFamily;
|
||||
hints.ai_socktype = SOCK_STREAM;
|
||||
hints.ai_protocol = IPPROTO_TCP;
|
||||
hints.ai_flags = flags;
|
||||
|
||||
if ((status = getaddrinfo(szHost, szPort, &hints, saList)))
|
||||
{
|
||||
printerrorf("Warning: %s: %s\n", addr, gai_strerror(status));
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
|
||||
static int_fast8_t setBlockingEnabled(SOCKET fd, int_fast8_t blocking)
|
||||
{
|
||||
if (fd == INVALID_SOCKET) return FALSE;
|
||||
|
||||
#ifdef _WIN32
|
||||
|
||||
unsigned long mode = blocking ? 0 : 1;
|
||||
return (ioctlsocket(fd, FIONBIO, &mode) == 0) ? TRUE : FALSE;
|
||||
|
||||
#else // POSIX
|
||||
|
||||
int flags = fcntl(fd, F_GETFL, 0);
|
||||
|
||||
if (flags < 0) return FALSE;
|
||||
|
||||
flags = blocking ? (flags & ~O_NONBLOCK) : (flags | O_NONBLOCK);
|
||||
return (fcntl(fd, F_SETFL, flags) == 0) ? TRUE : FALSE;
|
||||
|
||||
#endif // POSIX
|
||||
}
|
||||
|
||||
|
||||
int_fast8_t isDisconnected(const SOCKET s)
|
||||
{
|
||||
char buffer[1];
|
||||
|
||||
if (!setBlockingEnabled(s, FALSE)) return TRUE;
|
||||
|
||||
int n = recv(s, buffer, 1, MSG_PEEK);
|
||||
|
||||
if (!setBlockingEnabled(s, TRUE)) return TRUE;
|
||||
if (n == 0) return TRUE;
|
||||
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
|
||||
// Connect to TCP address addr (e.g. "kms.example.com:1688") and return an
|
||||
// open socket for the connection if successful or INVALID_SOCKET otherwise
|
||||
SOCKET connectToAddress(const char *const addr, const int AddressFamily, int_fast8_t showHostName)
|
||||
{
|
||||
struct addrinfo *saList, *sa;
|
||||
SOCKET s = INVALID_SOCKET;
|
||||
char szAddr[128];
|
||||
|
||||
if (!getSocketList(&saList, addr, 0, AddressFamily)) return INVALID_SOCKET;
|
||||
|
||||
for (sa = saList; sa; sa = sa->ai_next)
|
||||
{
|
||||
// struct sockaddr_in* addr4 = (struct sockaddr_in*)sa->ai_addr;
|
||||
// struct sockaddr_in6* addr6 = (struct sockaddr_in6*)sa->ai_addr;
|
||||
|
||||
if (ip2str(szAddr, sizeof(szAddr), sa->ai_addr, sa->ai_addrlen))
|
||||
{
|
||||
if (showHostName)
|
||||
printf("Connecting to %s (%s) ... ", addr, szAddr);
|
||||
else
|
||||
printf("Connecting to %s ... ", szAddr);
|
||||
|
||||
fflush(stdout);
|
||||
}
|
||||
|
||||
s = socket(sa->ai_family, SOCK_STREAM, IPPROTO_TCP);
|
||||
|
||||
# if !defined(NO_TIMEOUT) && !__minix__
|
||||
# ifndef _WIN32 // Standard Posix timeout structure
|
||||
|
||||
struct timeval to;
|
||||
to.tv_sec = 10;
|
||||
to.tv_usec = 0;
|
||||
|
||||
# else // Windows requires a DWORD with milliseconds
|
||||
|
||||
DWORD to = 10000;
|
||||
|
||||
# endif // _WIN32
|
||||
|
||||
setsockopt(s, SOL_SOCKET, SO_RCVTIMEO, (sockopt_t)&to, sizeof(to));
|
||||
setsockopt(s, SOL_SOCKET, SO_SNDTIMEO, (sockopt_t)&to, sizeof(to));
|
||||
# endif // !defined(NO_TIMEOUT) && !__minix__
|
||||
|
||||
if (!connect(s, sa->ai_addr, sa->ai_addrlen))
|
||||
{
|
||||
printf("successful\n");
|
||||
break;
|
||||
}
|
||||
|
||||
errorout("%s\n", socket_errno == VLMCSD_EINPROGRESS ? "Timed out" : vlmcsd_strerror(socket_errno));
|
||||
|
||||
socketclose(s);
|
||||
s = INVALID_SOCKET;
|
||||
}
|
||||
|
||||
freeaddrinfo(saList);
|
||||
return s;
|
||||
}
|
||||
|
||||
|
||||
#ifndef NO_SOCKETS
|
||||
|
||||
// Create a Listening socket for addrinfo sa and return socket s
|
||||
// szHost and szPort are for logging only
|
||||
static int listenOnAddress(const struct addrinfo *const ai, SOCKET *s)
|
||||
{
|
||||
int error;
|
||||
char ipstr[64];
|
||||
|
||||
ip2str(ipstr, sizeof(ipstr), ai->ai_addr, ai->ai_addrlen);
|
||||
|
||||
//*s = socket(ai->ai_family, ai->ai_socktype, ai->ai_protocol);
|
||||
*s = socket(ai->ai_family, SOCK_STREAM, IPPROTO_TCP);
|
||||
|
||||
if (*s == INVALID_SOCKET)
|
||||
{
|
||||
error = socket_errno;
|
||||
printerrorf("Warning: %s error. %s\n", ai->ai_family == AF_INET6 ? cIPv6 : cIPv4, vlmcsd_strerror(error));
|
||||
return error;
|
||||
}
|
||||
|
||||
# if !defined(_WIN32) && !defined(NO_SIGHUP)
|
||||
|
||||
int flags = fcntl(*s, F_GETFD, 0);
|
||||
|
||||
if (flags != -1)
|
||||
{
|
||||
flags |= FD_CLOEXEC;
|
||||
fcntl(*s, F_SETFD, flags);
|
||||
}
|
||||
# ifdef _PEDANTIC
|
||||
else
|
||||
{
|
||||
printerrorf("Warning: Could not set FD_CLOEXEC flag on %s: %s\n", ipstr, vlmcsd_strerror(errno));
|
||||
}
|
||||
# endif // _PEDANTIC
|
||||
|
||||
# endif // !defined(_WIN32) && !defined(NO_SIGHUP)
|
||||
|
||||
BOOL socketOption = TRUE;
|
||||
|
||||
// fix for lame tomato toolchain
|
||||
# ifndef IPV6_V6ONLY
|
||||
# ifdef __linux__
|
||||
# define IPV6_V6ONLY (26)
|
||||
# endif // __linux__
|
||||
# endif // IPV6_V6ONLY
|
||||
|
||||
# ifdef IPV6_V6ONLY
|
||||
if (ai->ai_family == AF_INET6) setsockopt(*s, IPPROTO_IPV6, IPV6_V6ONLY, (sockopt_t)&socketOption, sizeof(socketOption));
|
||||
# endif
|
||||
|
||||
# ifndef _WIN32
|
||||
setsockopt(*s, SOL_SOCKET, SO_REUSEADDR, (sockopt_t)&socketOption, sizeof(socketOption));
|
||||
# endif
|
||||
|
||||
if (bind(*s, ai->ai_addr, ai->ai_addrlen) || listen(*s, SOMAXCONN))
|
||||
{
|
||||
error = socket_errno;
|
||||
printerrorf("Warning: %s: %s\n", ipstr, vlmcsd_strerror(error));
|
||||
socketclose(*s);
|
||||
return error;
|
||||
}
|
||||
|
||||
# ifndef NO_LOG
|
||||
logger("Listening on %s\n", ipstr);
|
||||
# endif
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
// Adds a listening socket for an address string,
|
||||
// e.g. 127.0.0.1:1688 or [2001:db8:dead:beef::1]:1688
|
||||
BOOL addListeningSocket(const char *const addr)
|
||||
{
|
||||
struct addrinfo *aiList, *ai;
|
||||
int result = FALSE;
|
||||
SOCKET *s = SocketList + numsockets;
|
||||
|
||||
if (getSocketList(&aiList, addr, AI_PASSIVE | AI_NUMERICHOST, AF_UNSPEC))
|
||||
{
|
||||
for (ai = aiList; ai; ai = ai->ai_next)
|
||||
{
|
||||
// struct sockaddr_in* addr4 = (struct sockaddr_in*)sa->ai_addr;
|
||||
// struct sockaddr_in6* addr6 = (struct sockaddr_in6*)sa->ai_addr;
|
||||
|
||||
if (numsockets >= FD_SETSIZE)
|
||||
{
|
||||
#ifdef _PEDANTIC // Do not report this error in normal builds to keep file size low
|
||||
printerrorf("Warning: Cannot listen on %s. Your OS only supports %u listening sockets in an FD_SET.\n", addr, FD_SETSIZE);
|
||||
#endif
|
||||
break;
|
||||
}
|
||||
|
||||
if (!listenOnAddress(ai, s))
|
||||
{
|
||||
numsockets++;
|
||||
result = TRUE;
|
||||
}
|
||||
}
|
||||
|
||||
freeaddrinfo(aiList);
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
|
||||
// Just create some dummy sockets to see if we have a specific protocol (IPv4 or IPv6)
|
||||
__pure int_fast8_t checkProtocolStack(const int addressfamily)
|
||||
{
|
||||
SOCKET s; // = INVALID_SOCKET;
|
||||
|
||||
s = socket(addressfamily, SOCK_STREAM, 0);
|
||||
int_fast8_t success = (s != INVALID_SOCKET);
|
||||
|
||||
socketclose(s);
|
||||
return success;
|
||||
}
|
||||
|
||||
|
||||
// Build an fd_set of all listening socket then use select to wait for an incoming connection
|
||||
static SOCKET network_accept_any()
|
||||
{
|
||||
fd_set ListeningSocketsList;
|
||||
SOCKET maxSocket, sock;
|
||||
int i;
|
||||
int status;
|
||||
|
||||
FD_ZERO(&ListeningSocketsList);
|
||||
maxSocket = 0;
|
||||
|
||||
for (i = 0; i < numsockets; i++)
|
||||
{
|
||||
FD_SET(SocketList[i], &ListeningSocketsList);
|
||||
if (SocketList[i] > maxSocket) maxSocket = SocketList[i];
|
||||
}
|
||||
|
||||
status = select(maxSocket + 1, &ListeningSocketsList, NULL, NULL, NULL);
|
||||
|
||||
if (status < 0) return INVALID_SOCKET;
|
||||
|
||||
sock = INVALID_SOCKET;
|
||||
|
||||
for (i = 0; i < numsockets; i++)
|
||||
{
|
||||
if (FD_ISSET(SocketList[i], &ListeningSocketsList))
|
||||
{
|
||||
sock = SocketList[i];
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (sock == INVALID_SOCKET)
|
||||
return INVALID_SOCKET;
|
||||
else
|
||||
return accept(sock, NULL, NULL);
|
||||
}
|
||||
|
||||
|
||||
void closeAllListeningSockets()
|
||||
{
|
||||
int i;
|
||||
|
||||
for (i = 0; i < numsockets; i++)
|
||||
{
|
||||
shutdown(SocketList[i], VLMCSD_SHUT_RDWR);
|
||||
socketclose(SocketList[i]);
|
||||
}
|
||||
}
|
||||
#endif // NO_SOCKETS
|
||||
|
||||
|
||||
static void serveClient(const SOCKET s_client, const DWORD RpcAssocGroup)
|
||||
{
|
||||
# if !defined(NO_TIMEOUT) && !__minix__
|
||||
|
||||
# ifndef _WIN32 // Standard Posix timeout structure
|
||||
|
||||
struct timeval to;
|
||||
to.tv_sec = ServerTimeout;
|
||||
to.tv_usec = 0;
|
||||
|
||||
#else // Windows requires a DWORD with milliseconds
|
||||
|
||||
DWORD to = ServerTimeout * 1000;
|
||||
|
||||
# endif // _WIN32
|
||||
|
||||
# if !defined(NO_LOG) && defined(_PEDANTIC)
|
||||
|
||||
int result =
|
||||
setsockopt(s_client, SOL_SOCKET, SO_RCVTIMEO, (sockopt_t)&to, sizeof(to)) ||
|
||||
setsockopt(s_client, SOL_SOCKET, SO_SNDTIMEO, (sockopt_t)&to, sizeof(to));
|
||||
|
||||
if (result) logger("Warning: Set timeout failed: %s\n", vlmcsd_strerror(socket_errno));
|
||||
|
||||
# else // !(!defined(NO_LOG) && defined(_PEDANTIC))
|
||||
|
||||
setsockopt(s_client, SOL_SOCKET, SO_RCVTIMEO, (sockopt_t)&to, sizeof(to));
|
||||
setsockopt(s_client, SOL_SOCKET, SO_SNDTIMEO, (sockopt_t)&to, sizeof(to));
|
||||
|
||||
# endif // !(!defined(NO_LOG) && defined(_PEDANTIC))
|
||||
|
||||
# endif // !defined(NO_TIMEOUT) && !__minix__
|
||||
|
||||
char ipstr[64];
|
||||
socklen_t len;
|
||||
struct sockaddr_storage addr;
|
||||
|
||||
len = sizeof addr;
|
||||
|
||||
if (getpeername(s_client, (struct sockaddr*)&addr, &len) ||
|
||||
!ip2str(ipstr, sizeof(ipstr), (struct sockaddr*)&addr, len))
|
||||
{
|
||||
# if !defined(NO_LOG) && defined(_PEDANTIC)
|
||||
logger("Fatal: Cannot determine client's IP address: %s\n", vlmcsd_strerror(errno));
|
||||
# endif // !defined(NO_LOG) && defined(_PEDANTIC)
|
||||
socketclose(s_client);
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
# ifndef NO_LOG
|
||||
const char *const connection_type = addr.ss_family == AF_INET6 ? cIPv6 : cIPv4;
|
||||
static const char *const cAccepted = "accepted";
|
||||
static const char *const cClosed = "closed";
|
||||
static const char *const fIP = "%s connection %s: %s.\n";
|
||||
|
||||
logger(fIP, connection_type, cAccepted, ipstr);
|
||||
#endif // NO_LOG
|
||||
|
||||
rpcServer(s_client, RpcAssocGroup, ipstr);
|
||||
|
||||
# ifndef NO_LOG
|
||||
logger(fIP, connection_type, cClosed, ipstr);
|
||||
# endif // NO_LOG
|
||||
|
||||
socketclose(s_client);
|
||||
}
|
||||
|
||||
|
||||
#ifndef NO_SOCKETS
|
||||
static void post_sem(void)
|
||||
{
|
||||
#if !defined(NO_LIMIT) && !__minix__
|
||||
if (!InetdMode && MaxTasks != SEM_VALUE_MAX)
|
||||
{
|
||||
semaphore_post(Semaphore);
|
||||
}
|
||||
#endif // !defined(NO_LIMIT) && !__minix__
|
||||
}
|
||||
|
||||
|
||||
static void wait_sem(void)
|
||||
{
|
||||
#if !defined(NO_LIMIT) && !__minix__
|
||||
if (!InetdMode && MaxTasks != SEM_VALUE_MAX)
|
||||
{
|
||||
semaphore_wait(Semaphore);
|
||||
}
|
||||
#endif // !defined(NO_LIMIT) && !__minix__
|
||||
}
|
||||
#endif // NO_SOCKETS
|
||||
|
||||
#if defined(USE_THREADS) && !defined(NO_SOCKETS)
|
||||
|
||||
#if defined(_WIN32) || defined(__CYGWIN__) // Win32 Threads
|
||||
static DWORD WINAPI serveClientThreadProc(PCLDATA clData)
|
||||
#else // Posix threads
|
||||
static void *serveClientThreadProc (PCLDATA clData)
|
||||
#endif // Thread proc is identical in WIN32 and Posix threads
|
||||
{
|
||||
serveClient(clData->socket, clData->RpcAssocGroup);
|
||||
free(clData);
|
||||
post_sem();
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
#endif // USE_THREADS
|
||||
|
||||
|
||||
#ifndef NO_SOCKETS
|
||||
|
||||
#if defined(USE_THREADS) && (defined(_WIN32) || defined(__CYGWIN__)) // Windows Threads
|
||||
static int serveClientAsyncWinThreads(const PCLDATA thr_CLData)
|
||||
{
|
||||
wait_sem();
|
||||
|
||||
HANDLE h = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)serveClientThreadProc, thr_CLData, 0, NULL);
|
||||
|
||||
if (h)
|
||||
CloseHandle(h);
|
||||
else
|
||||
{
|
||||
socketclose(thr_CLData->socket);
|
||||
free(thr_CLData);
|
||||
post_sem();
|
||||
return GetLastError();
|
||||
}
|
||||
|
||||
return NO_ERROR;
|
||||
}
|
||||
#endif // defined(USE_THREADS) && defined(_WIN32) // Windows Threads
|
||||
|
||||
|
||||
#if defined(USE_THREADS) && !defined(_WIN32) && !defined(__CYGWIN__) // Posix Threads
|
||||
static int ServeClientAsyncPosixThreads(const PCLDATA thr_CLData)
|
||||
{
|
||||
pthread_t p_thr;
|
||||
pthread_attr_t attr;
|
||||
|
||||
wait_sem();
|
||||
|
||||
// Must set detached state to avoid memory leak
|
||||
if (pthread_attr_init(&attr) ||
|
||||
pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED) ||
|
||||
pthread_create(&p_thr, &attr, (void * (*)(void *))serveClientThreadProc, thr_CLData))
|
||||
{
|
||||
socketclose(thr_CLData->socket);
|
||||
free(thr_CLData);
|
||||
post_sem();
|
||||
return !0;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
#endif // defined(USE_THREADS) && !defined(_WIN32) // Posix Threads
|
||||
|
||||
#ifndef USE_THREADS // fork() implementation
|
||||
static void ChildSignalHandler(const int signal)
|
||||
{
|
||||
if (signal == SIGHUP) return;
|
||||
|
||||
post_sem();
|
||||
|
||||
#ifndef NO_LOG
|
||||
logger("Warning: Child killed/crashed by %s\n", strsignal(signal));
|
||||
#endif // NO_LOG
|
||||
|
||||
exit(!0);
|
||||
}
|
||||
|
||||
static int ServeClientAsyncFork(const SOCKET s_client, const DWORD RpcAssocGroup)
|
||||
{
|
||||
int pid;
|
||||
wait_sem();
|
||||
|
||||
if ((pid = fork()) < 0)
|
||||
{
|
||||
return errno;
|
||||
}
|
||||
else if ( pid )
|
||||
{
|
||||
// Parent process
|
||||
socketclose(s_client);
|
||||
return 0;
|
||||
}
|
||||
else
|
||||
{
|
||||
// Child process
|
||||
|
||||
// Setup a Child Handler for most common termination signals
|
||||
struct sigaction sa;
|
||||
|
||||
sa.sa_flags = 0;
|
||||
sa.sa_handler = ChildSignalHandler;
|
||||
|
||||
static int signallist[] = { SIGHUP, SIGINT, SIGTERM, SIGSEGV, SIGILL, SIGFPE, SIGBUS };
|
||||
|
||||
if (!sigemptyset(&sa.sa_mask))
|
||||
{
|
||||
uint_fast8_t i;
|
||||
|
||||
for (i = 0; i < _countof(signallist); i++)
|
||||
{
|
||||
sigaction(signallist[i], &sa, NULL);
|
||||
}
|
||||
}
|
||||
|
||||
serveClient(s_client, RpcAssocGroup);
|
||||
post_sem();
|
||||
exit(0);
|
||||
}
|
||||
}
|
||||
#endif
|
||||
|
||||
|
||||
int serveClientAsync(const SOCKET s_client, const DWORD RpcAssocGroup)
|
||||
{
|
||||
#ifndef USE_THREADS // fork() implementation
|
||||
|
||||
return ServeClientAsyncFork(s_client, RpcAssocGroup);
|
||||
|
||||
#else // threads implementation
|
||||
|
||||
PCLDATA thr_CLData = (PCLDATA)vlmcsd_malloc(sizeof(CLDATA));
|
||||
thr_CLData->socket = s_client;
|
||||
thr_CLData->RpcAssocGroup = RpcAssocGroup;
|
||||
|
||||
#if defined(_WIN32) || defined (__CYGWIN__) // Windows threads
|
||||
|
||||
return serveClientAsyncWinThreads(thr_CLData);
|
||||
|
||||
#else // Posix Threads
|
||||
|
||||
return ServeClientAsyncPosixThreads(thr_CLData);
|
||||
|
||||
#endif // Posix Threads
|
||||
|
||||
#endif // USE_THREADS
|
||||
}
|
||||
|
||||
#endif // NO_SOCKETS
|
||||
|
||||
|
||||
int runServer()
|
||||
{
|
||||
DWORD RpcAssocGroup = rand32();
|
||||
|
||||
// If compiled for inetd-only mode just serve the stdin socket
|
||||
#ifdef NO_SOCKETS
|
||||
serveClient(STDIN_FILENO, RpcAssocGroup);
|
||||
return 0;
|
||||
#else
|
||||
// In inetd mode just handle the stdin socket
|
||||
if (InetdMode)
|
||||
{
|
||||
serveClient(STDIN_FILENO, RpcAssocGroup);
|
||||
return 0;
|
||||
}
|
||||
|
||||
// Standalone mode
|
||||
for (;;)
|
||||
{
|
||||
int error;
|
||||
SOCKET s_client;
|
||||
|
||||
if ( (s_client = network_accept_any()) == INVALID_SOCKET )
|
||||
{
|
||||
error = socket_errno;
|
||||
|
||||
if (error == VLMCSD_EINTR || error == VLMCSD_ECONNABORTED) continue;
|
||||
|
||||
#ifdef _NTSERVICE
|
||||
if (ServiceShutdown) return 0;
|
||||
#endif
|
||||
|
||||
#ifndef NO_LOG
|
||||
logger("Fatal: %s\n",vlmcsd_strerror(error));
|
||||
#endif
|
||||
|
||||
return error;
|
||||
}
|
||||
|
||||
RpcAssocGroup++;
|
||||
serveClientAsync(s_client, RpcAssocGroup);
|
||||
}
|
||||
#endif // NO_SOCKETS
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
#endif // USE_MSRPC
|
||||
Loading…
Add table
Add a link
Reference in a new issue