mprplatform.c

samba最新软件

/**
 *	@file 	mprPlatform.c
 *	@brief	Cross platform routines 
 *	@overview This module provides low level cross platform routines.
 *	@remarks Most routines in this file are not thread-safe. It is the callers 
 *		responsibility to perform all thread synchronization.
 */

/*
 *	@copy	default
 *	
 *	Copyright (c) Mbedthis Software LLC, 2003-2006. All Rights Reserved.
 *	
 *	This software is distributed under commercial and open source licenses.
 *	You may use the GPL open source license described below or you may acquire 
 *	a commercial license from Mbedthis Software. You agree to be fully bound 
 *	by the terms of either license. Consult the LICENSE.TXT distributed with 
 *	this software for full details.
 *	
 *	This software is open source; you can redistribute it and/or modify it 
 *	under the terms of the GNU General Public License as published by the 
 *	Free Software Foundation; either version 2 of the License, or (at your 
 *	option) any later version. See the GNU General Public License for more 
 *	details at: http://www.mbedthis.com/downloads/gplLicense.html
 *	
 *	This program is distributed WITHOUT ANY WARRANTY; without even the 
 *	implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. 
 *	
 *	This GPL license does NOT permit incorporating this software into 
 *	proprietary programs. If you are unable to comply with the GPL, you must
 *	acquire a commercial license to use this software. Commercial licenses 
 *	for this software and support services are available from Mbedthis 
 *	Software at http://www.mbedthis.com 
 *	
 *	@end
 */

/********************************** Includes **********************************/
/*
 *	We need to use the underlying str(cpy) routines to implement our safe
 *	alternatives
 */
#if !DOXYGEN
#define 	UNSAFE_FUNCTIONS_OK 1
#endif

#include	"mpr.h"

/**************************** Forward Declarations ****************************/

static const char *getHive(const char *keyPath, HKEY *hive);

/************************************ Code ************************************/

char *mprInetToStr(char *buffer, int bufsize, const struct in_addr in)
{
#if HAVE_NTOA_R
	inet_ntoa_r(in, buffer, bufsize);
#else
	uchar	*cp;
	cp = (uchar*) ∈
	mprSprintf(buffer, bufsize, "%d.%d.%d.%d", cp[0], cp[1], cp[2], cp[3]);
#endif
	return buffer;
}

/******************************************************************************/

void mprSetShell(MprCtx ctx, void *shell)
{
}

/******************************************************************************/

void *mprGetShell(MprCtx ctx)
{
	return 0;
}

/******************************************************************************/
/*
 *	Sleep. Period given in milliseconds.
 */

void mprSleep(MprCtx ctx, int milliseconds)
{
	Sleep(milliseconds);
}

/******************************************************************************/
/*
 *	Make intervening directories
 */

int mprMakeDirPath(MprCtx ctx, const char *path)
{
	char	dir[MPR_MAX_PATH], buf[MPR_MAX_PATH];
	char	*dirSep;
	char	*next, *tok;

	dir[0] = '\0';
	dirSep = "/\\";

	if (path == 0 || *path == '\0') {
		return MPR_ERR_BAD_ARGS;
	}

	mprStrcpy(buf, sizeof(buf), path);
	next = mprStrTok(buf, dirSep, &tok);
	if (*buf == '/') {
		dir[0] = '/';
	}
	while (next != NULL) {
		if (strcmp(next, ".") == 0 ) {
			next = mprStrTok(NULL, dirSep, &tok);
			continue;
		}
		strcat(dir, next);
		if (access(dir, R_OK) != 0) {
			if (_mkdir(dir) < 0) {
				return MPR_ERR_CANT_CREATE;
			}
		}
		strcat(dir, "/");
		next = mprStrTok(NULL, dirSep, &tok);
	}
	return 0;
}

/******************************************************************************/
/*
 *	Get a fully qualified file name for the given path. Return with forward
 *	slashes always
 */

char *mprGetFullPathName(char *buf, int buflen, const char *path)
{
#if (WIN || NW || OS2) && !BLD_FEATURE_ROMFS
	char	*junk, *cp;
	int	rc;

	--buflen;
	rc = GetFullPathName(path, buflen, buf, &junk);
	for (cp = buf; *cp; cp++) {
		if (*cp == '\\') {
			*cp = '/';
		}
	}
	buf[buflen] = '\0';
#else
	if (mprStrcpy(buf, buflen, path) < 0) {
		mprAssert(0);
		return 0;
	}
#endif
	return buf;
}

/******************************************************************************/
/*
 *	Replacement for gethostbyname that is multi-thread safe
 */

struct hostent *mprGetHostByName(MprCtx ctx, const char *name)
{
	MprApp			*app;
	struct hostent	*hp;
	struct hostent	*ip;
	int				count, i;

	hp = (struct hostent*) mprAlloc(ctx, sizeof(struct hostent));
	memset(hp, 0, sizeof(struct hostent));

	app = mprGetApp(ctx);

	#undef gethostbyname

	mprGlobalLock(app);
	ip = gethostbyname(name);
	mprGlobalUnlock(app);

	if (ip == 0) {
		return 0;
	}
	hp->h_addrtype = ip->h_addrtype;
	hp->h_length = ip->h_length;
	hp->h_name = mprStrdup(hp, ip->h_name);
	hp->h_addr_list = 0;
	hp->h_aliases = 0;

	for (count = 0; ip->h_addr_list[count] != 0; ) {
		count++;
	}
	if (count > 0) {
		count++;
		hp->h_addr_list = mprAlloc(hp, count * sizeof(char*));
		for (i = 0; ip->h_addr_list[i] != 0; i++) {
			memcpy(&hp->h_addr_list[i], &ip->h_addr_list[i], ip->h_length);
		}
		hp->h_addr_list[i] = 0;
	}

	for (count = 0; ip->h_aliases[count] != 0; ) {
		count++;
	}
	if (count > 0) {
		count++;
		hp->h_aliases = mprAlloc(hp, count * sizeof(char*));
		for (i = 0; ip->h_aliases[i] != 0; i++) {
			hp->h_aliases[i] = mprStrdup(hp, ip->h_aliases[i]);
		}
		hp->h_aliases[i] = 0;
	}
	return hp;
}

/******************************************************************************/
/*
 *	Read a registry value. Returns allocated memory in buf.
 */ 

int mprReadRegistry(MprCtx ctx, char **buf, int max, const char *key, 
	const char *name)
{
	HKEY		top, h;
	char		*value;
	ulong		type, size;

	mprAssert(key && *key);
	mprAssert(buf);

	/*
	 *	Get the registry hive
	 */
	if ((key = getHive(key, &top)) == 0) {
		return MPR_ERR_CANT_ACCESS;
	}

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

	/*
	 *	Get the type
	 */
	if (RegQueryValueEx(h, name, 0, &type, 0, &size) != ERROR_SUCCESS) {
		RegCloseKey(h);
		return MPR_ERR_CANT_READ;
	}
	if (type != REG_SZ && type != REG_EXPAND_SZ) {
		RegCloseKey(h);
		return MPR_ERR_BAD_TYPE;
	}

	value = (char*) mprAlloc(ctx, size);
	if ((int) size > max) {
		RegCloseKey(h);
		return MPR_ERR_WONT_FIT;
	}
	if (RegQueryValueEx(h, name, 0, &type, (uchar*) value, &size) != 
			ERROR_SUCCESS) {
		mprFree(value);
		RegCloseKey(h);
		return MPR_ERR_CANT_READ;
	}

    RegCloseKey(h);
	*buf = value;
	return 0;
}

/******************************************************************************/
/*
 *	Write a string registry value. Returns allocated memory in buf.
 */ 

int mprWriteRegistry(MprCtx ctx, const char *key, const char *name, 
	const char *value)
{
	HKEY	top, h, subHandle;
	ulong	disposition;

	mprAssert(key && *key);
	mprAssert(name && *name);
	mprAssert(value && *value);

	/*
	 *	Get the registry hive
	 */
	if ((key = getHive(key, &top)) == 0) {
		return MPR_ERR_CANT_ACCESS;
	}

	if (name) {
		/*
		 *	Write a registry string value
		 */
		if (RegOpenKeyEx(top, key, 0, KEY_ALL_ACCESS, &h) != ERROR_SUCCESS) {
			return MPR_ERR_CANT_ACCESS;
		}
		if (RegSetValueEx(h, name, 0, REG_SZ, value, strlen(value) + 1) 
				!= ERROR_SUCCESS) {
			RegCloseKey(h);
			return MPR_ERR_CANT_READ;
		}

	} else {
		/*
		 *	Create a new sub key
		 */
		if (RegOpenKeyEx(top, key, 0, KEY_CREATE_SUB_KEY, &h) != ERROR_SUCCESS){
			return MPR_ERR_CANT_ACCESS;
		}
		if (RegCreateKeyEx(h, name, 0, NULL, REG_OPTION_NON_VOLATILE,
			KEY_ALL_ACCESS, NULL, &subHandle, &disposition) != ERROR_SUCCESS) {
			return MPR_ERR_CANT_ACCESS;
		}
		RegCloseKey(subHandle);
	}
    RegCloseKey(h);
	return 0;
}

/******************************************************************************/
/*
 *	Determine the registry hive by the first portion of the path. Return 
 *	a pointer to the rest of key path after the hive portion.
 */ 

static const char *getHive(const char *keyPath, HKEY *hive)
{
	char	key[MPR_MAX_STRING], *cp;
	int		len;

	mprAssert(keyPath && *keyPath);

	*hive = 0;

	mprStrcpy(key, sizeof(key), keyPath);
	key[sizeof(key) - 1] = '\0';

	if (cp = strchr(key, '\\')) {
		*cp++ = '\0';
	}
	if (cp == 0 || *cp == '\0') {
		return 0;
	}

	if (!mprStrcmpAnyCase(key, "HKEY_LOCAL_MACHINE")) {
		*hive = HKEY_LOCAL_MACHINE;
	} else if (!mprStrcmpAnyCase(key, "HKEY_CURRENT_USER")) {
		*hive = HKEY_CURRENT_USER;
	} else if (!mprStrcmpAnyCase(key, "HKEY_USERS")) {
		*hive = HKEY_USERS;
	} else if (!mprStrcmpAnyCase(key, "HKEY_CLASSES_ROOT")) {
		*hive = HKEY_CLASSES_ROOT;
	} else {
		return 0;
	}

	if (*hive == 0) {
		return 0;
	}
	len = strlen(key) + 1;
	return keyPath + len;
}

/******************************************************************************/

/*
 * Local variables:
 * tab-width: 4
 * c-basic-offset: 4
 * End:
 * vim:tw=78
 * vim600: sw=4 ts=4 fdm=marker
 * vim<600: sw=4 ts=4
 */