encoding.c

非常好的dns解析软件

/*
 * encoding.c - get DNS/Local encodings
 *
 *      Software\JPNIC\IDN\Where
 *                        \LogFile
 *			  \LogLevel
 *			  \InstallDir   
 *                        \PerProg\<name>\Where
 *                        \PerProg\<name>\Encoding
 */

/*
 * Copyright (c) 2000,2001,2002 Japan Network Information Center.
 * All rights reserved.
 *  
 * By using this file, you agree to the terms and conditions set forth bellow.
 * 
 * 			LICENSE TERMS AND CONDITIONS 
 * 
 * The following License Terms and Conditions apply, unless a different
 * license is obtained from Japan Network Information Center ("JPNIC"),
 * a Japanese association, Kokusai-Kougyou-Kanda Bldg 6F, 2-3-4 Uchi-Kanda,
 * Chiyoda-ku, Tokyo 101-0047, Japan.
 * 
 * 1. Use, Modification and Redistribution (including distribution of any
 *    modified or derived work) in source and/or binary forms is permitted
 *    under this License Terms and Conditions.
 * 
 * 2. Redistribution of source code must retain the copyright notices as they
 *    appear in each source code file, this License Terms and Conditions.
 * 
 * 3. Redistribution in binary form must reproduce the Copyright Notice,
 *    this License Terms and Conditions, in the documentation and/or other
 *    materials provided with the distribution.  For the purposes of binary
 *    distribution the "Copyright Notice" refers to the following language:
 *    "Copyright (c) 2000-2002 Japan Network Information Center.  All rights reserved."
 * 
 * 4. The name of JPNIC may not be used to endorse or promote products
 *    derived from this Software without specific prior written approval of
 *    JPNIC.
 * 
 * 5. Disclaimer/Limitation of Liability: THIS SOFTWARE IS PROVIDED BY JPNIC
 *    "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 *    LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
 *    PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL JPNIC BE LIABLE
 *    FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 *    CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 *    SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
 *    BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
 *    WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
 *    OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
 *    ADVISED OF THE POSSIBILITY OF SUCH DAMAGES.
 */

#include <windows.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <ctype.h>

#include "wrapcommon.h"

#define IDN_GLOBAL	1
#define IDN_PERPROG	2
#define IDN_CURUSER	4

/*
 * Registry of Encodings
 */

#define	IDNKEY_WRAPPER	"Software\\JPNIC\\IDN"
#define	IDNKEY_PERPROG	"Software\\JPNIC\\IDN\\PerProg"
#define	IDNVAL_WHERE	"Where"
#define	IDNVAL_ENCODE	"Encoding"
#define	IDNVAL_LOGLVL	"LogLevel"
#define	IDNVAL_LOGFILE	"LogFile"
#define IDNVAL_INSDIR	"InstallDir"

static int	GetRegistry(HKEY top, const char *key, const char *name,
			    DWORD type, void *param, DWORD length);
static char	*GetPerProgKey(char *buf, size_t len);
static int	GetFromRegistry(const char *name, int where, DWORD type,
				void *param, DWORD length);
static int	GetIntFromRegistry(const char *name, int defvalue, int where);
static BOOL	GetStringFromRegistry(const char *name, char *result,
				      size_t length, int where);

static int
GetRegistry(HKEY top, const char *key, const char *name, DWORD type,
	    void *param, DWORD length)
{
	LONG stat;
	HKEY hk;
	DWORD realtype;

	stat = RegOpenKeyEx(top, key, 0, KEY_READ, &hk);
	if (stat != ERROR_SUCCESS) {
		return 0;
	}

	stat = RegQueryValueEx(hk, (LPCTSTR)name, NULL,
			       &realtype, (LPBYTE)param, &length);

	RegCloseKey(hk);

	if (stat != ERROR_SUCCESS || realtype != type)
		return 0;

	return 1;
}

static char *
GetPerProgKey(char *buf, size_t len)
{
	char exename[256];
	char prgname[256];
	char *p, *last;

	GetModuleFileName(NULL, exename, 256);
     
	for (p = exename, last = NULL; *p != '\0'; p++) {
		if (*p == '/' || *p == '\\') {
			last = p;
		}
	}
	strcpy(prgname, (last == NULL) ? exename : (last + 1));
	if ((p = strrchr(prgname, '.')) != NULL) {
		*p = '\0';
	}

	if (strlen(IDNKEY_PERPROG) + 1 + strlen(prgname) >= len) {
		return (NULL);
	}
	sprintf(buf, "%s\\%s", IDNKEY_PERPROG, prgname);
	return buf;
}

static int
GetFromRegistry(const char *name, int where, DWORD type,
		void *param, DWORD length)
{
	if (where & IDN_PERPROG) {
		/*
		 * First, try program specific setting.
		 */
		char keyname[256];

		/*
		 * Try HKEY_CURRENT_USER and HKEY_LOCAL_MACHINE.
		 */
		if (GetPerProgKey(keyname, sizeof(keyname)) != NULL) {
			if (((where & IDN_CURUSER) &&
			     GetRegistry(HKEY_CURRENT_USER, keyname, name,
					 type, param, length)) ||
			    GetRegistry(HKEY_LOCAL_MACHINE, keyname, name,
					type, param, length)) {
				return (1);
			}
		}
	}

	if (where & IDN_GLOBAL) {
		/*
		 * Try global setting.
		 */
		if (((where & IDN_CURUSER) &&
		     GetRegistry(HKEY_CURRENT_USER, IDNKEY_WRAPPER, name,
				 type, param, length)) ||
		    GetRegistry(HKEY_LOCAL_MACHINE, IDNKEY_WRAPPER, name,
				type, param, length)) {
			return (1);
		}
	}

	/*
	 * Not found.
	 */
	return (0);
}

static int
GetIntFromRegistry(const char *name, int defvalue, int where)
{
    DWORD param;

    if (GetFromRegistry(name, where, REG_DWORD, &param, sizeof(param))) {
	    return ((int)param);
    }
    return (defvalue);
}

static BOOL
GetStringFromRegistry(const char *name, char *result, size_t length, int where)
{
    if (GetFromRegistry(name, where, REG_SZ, result, (DWORD)length)) {
	    return (TRUE);
    }
    return (FALSE);
}

/*
 * idnEncodeWhere - which module should convert domain name
 */
int
idnEncodeWhere(void)
{
	int v = GetIntFromRegistry(IDNVAL_WHERE, IDN_ENCODE_ALWAYS,
				   IDN_GLOBAL|IDN_PERPROG|IDN_CURUSER);

	idnLogPrintf(idn_log_level_trace, "idnEncodeWhere: %d\n", v);
	return (v);
}

/*
 * idnGetLogFile - refer to log file
 */
BOOL
idnGetLogFile(char *file, size_t len)
{
	BOOL v = GetStringFromRegistry(IDNVAL_LOGFILE, file, len,
				       IDN_GLOBAL|IDN_CURUSER);

	idnLogPrintf(idn_log_level_trace, "idnGetLogFile: %-.100s\n",
		     (v == TRUE) ? file : "<none>");
	return (v);
}

/*
 * idnGetPrgEncoding - refer to Program's Local Encoding
 *
 *      use program name as registry key
 */
BOOL
idnGetPrgEncoding(char *enc, size_t len)
{
	if (GetStringFromRegistry(IDNVAL_ENCODE, enc, len,
				  IDN_PERPROG|IDN_CURUSER) != TRUE ||
	    enc[0] == '\0') {
		sprintf(enc, "CP%d", GetACP());
	}
	idnLogPrintf(idn_log_level_trace,
		     "idnGetPrgEncoding: %-.30s\n", enc);
	return (TRUE);
}

/*
 * idnGetLogLevel
 */
int
idnGetLogLevel(void)
{
	int v  = GetIntFromRegistry(IDNVAL_LOGLVL, 0,
				    IDN_GLOBAL|IDN_CURUSER);

	idnLogPrintf(idn_log_level_trace, "idnGetLogLevel: %d\n", v);
	return (v);
}

/*
 * idnGetInstallDir - get idn wrapper install directory
 */
BOOL
idnGetInstallDir(char *dir, size_t len)
{
	/* No need to look at HKEY_CURRENT_USER */
	BOOL v = GetStringFromRegistry(IDNVAL_INSDIR, dir, len, IDN_GLOBAL);

	idnLogPrintf(idn_log_level_trace, "idnGetInstallDir: %-.100s\n",
		     (v == TRUE) ? dir : "<none>");
	return (v);
}