mprgenfile.c

samba最新软件

/**
 *	@file 	mprGenFile.c
 *	@brief	Generic File services
 *	@overview 
 *	@remarks 
 *		See OS/mprFile.c for the per O/S portions
 */

/******************************************************************************/
/*
 *	@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 **********************************/

#include	"mpr.h"

/****************************** Forward Declarations **************************/
#if !BREW

static int closeDestructor(void *data);

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

int mprStartFileServices(MprCtx ctx)
{
	MprApp		*app;

	app = mprGetApp(ctx);
	app->console = mprAllocTypeZeroed(ctx, MprFile);
	app->error = mprAllocTypeZeroed(ctx, MprFile);

	/*
	 *	We assume that STDOUT is 1 and STDERR is 2
 	 */
	app->console->fd = 1;
	app->error->fd = 2;

	return 0;
}

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

void mprStopFileServices(MprCtx ctx)
{
	MprApp		*app;

	app = mprGetApp(ctx);

	mprFree(app->console);
	app->console = 0;
	mprFree(app->error);
	app->error = 0;
}

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

MprFile *mprOpen(MprCtx ctx, const char *path, int omode, int perms)
{
	MprFile		*file;
	
	mprAssert(path && *path);

	file = mprAllocTypeZeroed(ctx, MprFile);
	
	file->fd = open(path, omode, perms);
	if (file->fd < 0) {
		mprFree(file);
		return 0;
	}

	mprSetDestructor(file, closeDestructor);
	return file;
}

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

static int closeDestructor(void *data)
{
	MprFile	*file = (MprFile*) data;

	mprAssert(file);
	
	mprClose(file);
	return 0;
}

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

void mprClose(MprFile *file)
{
	mprAssert(file);

	if (file < 0) {
		return;
	}

	mprAssert(file->fd >= 0);
	close(file->fd);

	mprSetDestructor(file, 0);
	mprFree(file);
}
 
/******************************************************************************/

int mprRead(MprFile *file, void *buf, uint size)
{
	mprAssert(file);

	if (file == 0) {
		return MPR_ERR_BAD_HANDLE;
	}

	return read(file->fd, buf, size);
}

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

int mprWrite(MprFile *file, const void *buf, uint count)
{
	mprAssert(file);

	if (file == 0) {
		return MPR_ERR_BAD_HANDLE;
	}

	return write(file->fd, buf, count);
}

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

int mprSeek(MprFile *file, int seekType, long distance)
{
	mprAssert(file);

	if (file == 0) {
		return MPR_ERR_BAD_HANDLE;
	}

	return lseek(file->fd, seekType, distance);
}

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

int mprDelete(MprCtx ctx, const char *path)
{
	return unlink(path);
}

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

int mprDeleteDir(MprCtx ctx, const char *path)
{
	return rmdir(path);
}
 
#endif /* !BREW */
/******************************************************************************/

char *mprGets(MprFile *file, char *buf, uint size)
{
	MprBuf	*bp;
	int		count, len, c;

	mprAssert(file);

	if (file == 0) {
		return 0;
	}

	if (file->buf == 0) {
		file->buf = mprCreateBuf(file, MPR_DEFAULT_ALLOC, MPR_MAX_STRING);
	}
	bp = file->buf;

	/*
	 *	Must leave room for null
	 */
	count = 0;
	while (--size > 0) {
		if (mprGetBufLength(bp) == 0) {
			mprFlushBuf(bp);
			len = mprRead(file, mprGetBufEnd(bp), 
				mprGetBufLinearSpace(bp));
			if (len <= 0) {
				return 0;
			}
			mprAdjustBufEnd(bp, len);
			mprAddNullToBuf(bp);
		}
		if ((c = mprGetCharFromBuf(bp)) == '\n') {
			buf[count] = '\0';
			return buf;
		}
		buf[count++] = c;
	}
	buf[count] = '\0';
	return buf;
}

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

int mprPuts(MprFile *file, const char *writeBuf, uint count)
{
	MprBuf	*bp;
	char	*buf;
	int		total, bytes, len;

	mprAssert(file);

	/*
	 *	Buffer output and flush when full.
	 */
	if (file->buf == 0) {
		file->buf = mprCreateBuf(file, MPR_BUFSIZE, 0);
		if (file->buf == 0) {
			return MPR_ERR_CANT_ALLOCATE;
		}
	}
	bp = file->buf;

	if (mprGetBufLength(bp) > 0 && mprGetBufSpace(bp) < (int) count) {
		len = mprGetBufLength(bp);
		if (mprWrite(file, mprGetBufStart(bp), len) != len) {
			return MPR_ERR_CANT_WRITE;
		}
		mprFlushBuf(bp);
	}

	total = 0;
	buf = (char*) writeBuf;

	while (count > 0) {
		bytes = mprPutBlockToBuf(bp, buf, count);
		if (bytes <= 0) {
			return MPR_ERR_CANT_ALLOCATE;
		}
		count -= bytes;
		buf += bytes;
		total += bytes;
		mprAddNullToBuf(bp);

		if (count > 0) {
			len = mprGetBufLength(bp);
			if (mprWrite(file, mprGetBufStart(bp), len) != len) {
				return MPR_ERR_CANT_WRITE;
			}
			mprFlushBuf(bp);
		}
	}
	return total;
}

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

int mprMakeTempFileName(MprCtx ctx, char *buf, int bufsize, const char *tempDir)
{
	MprFile		*file;
	MprTime		now;
	char		*dir;
	int 		seed, i;

	if (tempDir == 0) {
#if WIN
		char	*cp;
		dir = mprStrdup(ctx, getenv("TEMP"));
		for (cp = dir; *cp; cp++) {
			if (*cp == '\\') {
				*cp = '/';
			}
		}
#else
		dir = mprStrdup(ctx, "/tmp");
#endif
	} else {
		dir = mprStrdup(ctx, tempDir);
	}

	mprGetTime(ctx, &now);
	seed = now.msec % 64000;
	file = 0;

	for (i = 0; i < 128; i++) {
		mprSprintf(buf, bufsize, "%s/MPR_%d_%d.tmp", dir, getpid(), seed++);
		file = mprOpen(ctx, buf, O_CREAT | O_EXCL | O_BINARY, 0664);
		if (file) {
			break;
		}
	}

	if (file == 0) {
		return MPR_ERR_CANT_CREATE;
	}

	mprClose(file);
	mprFree(dir);

	return 0;
}

/******************************************************************************/
/*
 * 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
 */