#include "apr.h"
#include "apr_general.h"
#include "apr_pools.h"
#include "apr_signal.h"
#include "apr_arch_misc.h"
#include "apr_arch_proc_mutex.h"
#include "apr_arch_internal_time.h"
#include "apr_ldap.h"
#ifdef USE_WINSOCK
int WSAStartupRTags(WORD wVersionRequested,
LPWSADATA lpWSAData,
rtag_t WSAStartupRTag,
rtag_t WSPSKTRTag,
rtag_t lookUpServiceBeginRTag,
rtag_t WSAEventRTag,
rtag_t WSPCPRTag);
int WSACleanupRTag(rtag_t rTag);
#define WS_LOAD_ENTRY_SIGNATURE (*(unsigned long *) "WLDE")
#define WS_SKT_SIGNATURE (*(unsigned long *) "WSKT")
#define WS_LOOKUP_SERVICE_SIGNATURE (*(unsigned long *) "WLUP")
#define WS_WSAEVENT_SIGNATURE (*(unsigned long *) "WEVT")
#define WS_CPORT_SIGNATURE (*(unsigned long *) "WCPT")
int (*WSAStartupWithNLMHandle)( WORD version, LPWSADATA data, void *handle ) = NULL;
int (*WSACleanupWithNLMHandle)( void *handle ) = NULL;
static int wsa_startup_with_handle (WORD wVersionRequested, LPWSADATA data, void *handle)
{
APP_DATA *app_data;
if (!(app_data = (APP_DATA*) get_app_data(gLibId)))
return APR_EGENERAL;
app_data->gs_startup_rtag = AllocateResourceTag(handle, "WinSock Start-up", WS_LOAD_ENTRY_SIGNATURE);
app_data->gs_socket_rtag = AllocateResourceTag(handle, "WinSock socket()", WS_SKT_SIGNATURE);
app_data->gs_lookup_rtag = AllocateResourceTag(handle, "WinSock Look-up", WS_LOOKUP_SERVICE_SIGNATURE);
app_data->gs_event_rtag = AllocateResourceTag(handle, "WinSock Event", WS_WSAEVENT_SIGNATURE);
app_data->gs_pcp_rtag = AllocateResourceTag(handle, "WinSock C-Port", WS_CPORT_SIGNATURE);
return WSAStartupRTags(wVersionRequested, data,
app_data->gs_startup_rtag,
app_data->gs_socket_rtag,
app_data->gs_lookup_rtag,
app_data->gs_event_rtag,
app_data->gs_pcp_rtag);
}
static int wsa_cleanup_with_handle (void *handle)
{
APP_DATA *app_data;
if (!(app_data = (APP_DATA*) get_app_data(gLibId)))
return APR_EGENERAL;
return WSACleanupRTag(app_data->gs_startup_rtag);
}
static int UnregisterAppWithWinSock (void *nlm_handle)
{
if (!WSACleanupWithNLMHandle)
{
if (!(WSACleanupWithNLMHandle = ImportPublicObject(gLibHandle, "WSACleanupWithNLMHandle")))
WSACleanupWithNLMHandle = wsa_cleanup_with_handle;
}
return (*WSACleanupWithNLMHandle)(nlm_handle);
}
static int RegisterAppWithWinSock (void *nlm_handle)
{
int err;
WSADATA wsaData;
WORD wVersionRequested = MAKEWORD(WSAHighByte, WSALowByte);
if (!WSAStartupWithNLMHandle)
{
if (!(WSAStartupWithNLMHandle = ImportPublicObject(gLibHandle, "WSAStartupWithNLMHandle")))
WSAStartupWithNLMHandle = wsa_startup_with_handle;
}
err = (*WSAStartupWithNLMHandle)(wVersionRequested, &wsaData, nlm_handle);
if (LOBYTE(wsaData.wVersion) != WSAHighByte ||
HIBYTE(wsaData.wVersion) != WSALowByte) {
UnregisterAppWithWinSock (nlm_handle);
return APR_EEXIST;
}
return err;
}
#endif
APR_DECLARE(apr_status_t) apr_app_initialize(int *argc,
const char * const * *argv,
const char * const * *env)
{
return apr_initialize();
}
APR_DECLARE(apr_status_t) apr_initialize(void)
{
apr_pool_t *pool;
void *nlmhandle = getnlmhandle();
if (register_NLM(nlmhandle) != 0) {
return APR_SUCCESS;
}
if (apr_pool_create(&pool, NULL) != APR_SUCCESS) {
return APR_ENOPOOL;
}
apr_pool_tag(pool, "apr_initilialize");
#ifdef USE_WINSOCK
{
int err;
if ((err = RegisterAppWithWinSock (nlmhandle))) {
return err;
}
}
#endif
apr_signal_init(pool);
#if APR_HAS_LDAP
apr_ldap_rebind_init(pool);
#endif
return APR_SUCCESS;
}
APR_DECLARE_NONSTD(void) apr_terminate(void)
{
APP_DATA *app_data;
if (!(app_data = (APP_DATA*) get_app_data(gLibId)))
return;
if (unregister_NLM(app_data->gs_nlmhandle) != 0) {
return;
}
netware_pool_proc_cleanup ();
#ifdef USE_WINSOCK
UnregisterAppWithWinSock (app_data->gs_nlmhandle);
#endif
}
APR_DECLARE(void) apr_terminate2(void)
{
apr_terminate();
}