#include <windows.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "wrapcommon.h"
static int winsock_idx;
static struct winsock_type {
char *version;
char *name;
char *original_name;
} winsock_info[] = {
#define IDN_IDX_WS11 0
{ "1.1", "WSOCK32", "WSOCK32O" },
#define IDN_IDX_WS20 1
{ "2.0", "WS2_32", "WS2_32O" },
{ NULL, NULL, NULL },
};
static HINSTANCE load_original_dll(void);
static BOOL check_idn_processing(void);
static BOOL check_dll(const char *name);
BOOL
idnWinsockVersion(const char *version) {
int i;
for (i = 0; winsock_info[i].version != NULL; i++) {
if (strcmp(winsock_info[i].version, version) == 0) {
winsock_idx = i;
idnLogPrintf(idn_log_level_trace,
"idnWinsockVersion: version %s\n",
version);
return (TRUE);
}
}
idnLogPrintf(idn_log_level_fatal,
"idnWinsockVersion: unknown winsock version %s\n",
version);
return (FALSE);
}
HINSTANCE
idnWinsockHandle(void) {
static HINSTANCE dll_handle = NULL;
static int initialized = 0;
if (!initialized) {
idnLogPrintf(idn_log_level_trace,
"idnWinsockHandle: loading original DLL..\n");
dll_handle = load_original_dll();
}
initialized = 1;
return (dll_handle);
}
idn_resconf_t
idnGetContext(void) {
static int initialized = 0;
static idn_resconf_t ctx = NULL;
if (!initialized) {
idnLogPrintf(idn_log_level_trace,
"idnGetContext: checking IDN status..\n");
if (check_idn_processing()) {
ctx = idnConvInit();
idnLogPrintf(idn_log_level_info,
"Processing context: %08x\n", ctx);
} else {
idnLogPrintf(idn_log_level_info,
"NOT process IDN here\n");
ctx = NULL;
}
initialized = 1;
}
return (ctx);
}
static HINSTANCE
load_original_dll(void) {
char dllpath[MAX_PATH];
const char *dll_name = winsock_info[winsock_idx].original_name;
HINSTANCE handle;
dllpath[0] = '\0';
if (idnGetInstallDir(dllpath, sizeof(dllpath)) != TRUE) {
idnLogPrintf(idn_log_level_fatal,
"idnWinsockHandle: cannot find idn wrapper's "
"install directory\n");
abort();
return (NULL);
}
if (dllpath[0] != '\0' &&
dllpath[strlen(dllpath) - 1] == '\\') {
dllpath[strlen(dllpath) - 1] = '\0';
}
if (strlen(dllpath) + strlen(dll_name) + 1 + 4 >= sizeof(dllpath)) {
idnLogPrintf(idn_log_level_fatal,
"idnWinsockHandle: idn wrapper's install path is "
"too long to be true\n");
abort();
return (NULL);
}
strcat(dllpath, "\\");
strcat(dllpath, dll_name);
strcat(dllpath, ".DLL");
idnLogPrintf(idn_log_level_trace,
"idnWinsockHandle: loading original winsock DLL (%s)\n",
dllpath);
if ((handle = LoadLibrary(dllpath)) == NULL) {
idnLogPrintf(idn_log_level_fatal,
"idnWinsockHandle: no DLL %-.100s\n", dllpath);
abort();
return (NULL);
}
return (handle);
}
static BOOL
check_idn_processing(void) {
int where = idnEncodeWhere();
BOOL here = FALSE;
idnLogPrintf(idn_log_level_trace,
"idnGetContext: Winsock%s, where=%d\n",
winsock_info[winsock_idx].version, where);
switch (winsock_idx) {
case IDN_IDX_WS11:
switch (where) {
case IDN_ENCODE_ALWAYS:
case IDN_ENCODE_ONLY11:
return (TRUE);
case IDN_ENCODE_CHECK:
if (!check_dll(winsock_info[winsock_idx].name)) {
return (TRUE);
}
break;
}
break;
case IDN_IDX_WS20:
switch (where) {
case IDN_ENCODE_ALWAYS:
case IDN_ENCODE_ONLY20:
case IDN_ENCODE_CHECK:
return (TRUE);
break;
}
break;
}
return (FALSE);
}
static BOOL
check_dll(const char *name) {
HINSTANCE hdll = NULL;
#if 1
hdll = LoadLibrary(name);
#else
hdll = LoadLibraryEx(name, NULL, LOAD_LIBRARY_AS_DATAFILE);
#endif
if (hdll == NULL) {
idnLogPrintf(idn_log_level_trace,
"idnGetContext: DLL %s does not exist\n");
return (FALSE);
} else {
idnLogPrintf(idn_log_level_trace,
"idnGetContext: DLL %s exists\n");
FreeLibrary(hdll);
return (TRUE);
}
}