env.c

用C语言实现的DOS操作系统的源代码

/****************************************************************/
/*								*/
/*			      env.c				*/
/*								*/
/*		    command.com Environment Support 		*/
/*								*/
/*			  August 9, 1991			*/
/*								*/
/*			Copyright (c) 1995			*/
/*			Pasquale J. Villani			*/
/*			All Rights Reserved			*/
/*								*/
/* This file is part of DOS-C.					*/
/*								*/
/* DOS-C is free software; 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, or (at your option) any later version.			*/
/*								*/
/* DOS-C is distributed in the hope that it will be useful, but	*/
/* WITHOUT ANY WARRANTY; without even the implied warranty of	*/
/* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See	*/
/* the GNU General Public License for more details.		*/
/*								*/
/* You should have received a copy of the GNU General Public	*/
/* License along with DOS-C; see the file COPYING.  If not,	*/
/* write to the Free Software Foundation, 675 Mass Ave,		*/
/* Cambridge, MA 02139, USA.					*/
/****************************************************************/


/* $Log:   C:/dos-c/src/command/env.c_v  $
 * 
 *    Rev 1.3   31 Jan 1998  8:02:22   patv
 * Re-written for improved environment handling.  Also fixed EnvSetVar bug.
 *
 *    Rev 1.2   29 Aug 1996 13:06:52   patv
 * Bug fixes for v0.91b
 *
 *    Rev 1.1   01 Sep 1995 18:04:36   patv
 * First GPL release.
 *
 *    Rev 1.0   02 Jul 1995 10:01:54   patv
 * Initial revision.
 */
/* $EndLog$ */

#include "../../hdr/portab.h"
#include "globals.h"
#include "proto.h"

#ifdef VERSION_STRINGS
static BYTE *RcsId = "$Header:   C:/dos-c/src/command/env.c_v   1.3   31 Jan 1998  8:02:22   patv  $";
#endif

BOOL EnvClearVar(BYTE *pszName);
COUNT EnvSizeUp(VOID);
VOID EnvCopy(BYTE FAR *lpszDest, BYTE FAR *lpszSrc);

COUNT EnvSizeUp(VOID)
{
	UWORD uNewEnv;
	BOOL bError;
	BYTE FAR *lpszSrc = lpEnviron;
	COUNT nSize = 0;

	while('\0' != *lpszSrc)
	{
		while(*lpszSrc)
			lpszSrc++, ++nSize;
		++lpszSrc, ++nSize;
	}
	/* Count the terminating entry					*/
	lpszSrc++, ++nSize;


	/* Count the invocation part					*/
	((UWORD FAR *)lpszSrc)++;
	nSize += sizeof(UWORD);
	while(*lpszSrc)
		lpszSrc++, ++nSize;

	/* Count the terminating null ...				*/
	nSize++;

	/* and return the count.					*/
	return nSize;
}


BOOL EnvAlloc(COUNT nSize)
{
	COUNT nKlicks;
	BOOL bError;
	UWORD uSeg;

	/* Check for error						*/
	if(nSize <= EnvSizeUp())
		return FALSE;

	/* Do the allocation, then copy the environment			*/
	nKlicks = (nSize + PARASIZE - 1)/PARASIZE;
	uSeg = DosAllocMem(nKlicks, (BOOL FAR *)&bError);
	if(!bError)
	{
		EnvCopy(MK_FP(uSeg, 0), lpEnviron);
		DosFreeMem(FP_SEG(lpEnviron), &bError);
		if(bError)
		{
			DosFreeMem(uSeg, &bError);
			return FALSE;
		}
		lpEnviron = MK_FP(uSeg, 0);
		return TRUE;
	}
	else
		return FALSE;
}


BOOL EnvFree(BYTE FAR *lpMem)
{
	UWORD uSeg;
	BOOL bError;

	uSeg = FP_SEG(lpMem);
	DosFreeMem(uSeg, (BOOL FAR *)bError);
	return !bError;
}


BYTE *EnvLookup(pszName)
BYTE *pszName;
{
	BYTE FAR *lpszString;
	static BYTE szNameBuffer[MAX_CMDLINE];
	BYTE *pszLclString;
	COUNT nNameLen;

	strcpy(szNameBuffer, pszName);
	nNameLen = strlen(pszName);
	for(pszLclString = szNameBuffer; *pszLclString != '\0'; pszLclString++)
		*pszLclString = toupper(*pszLclString);

	lpszString = lpEnviron;
	while('\0' != *lpszString)
	{
		if((0 == fstrncmp(lpszString, (BYTE FAR *)szNameBuffer, nNameLen))
		|| (0 == fstrncmp(lpszString, (BYTE FAR *)pszName, nNameLen)))
		{
			/* Match, skip the variable name */
			while(*lpszString && '=' != *lpszString)
				++lpszString;
			++lpszString;
			/* Now copy var into static buffer */
			for(pszLclString = szNameBuffer; *lpszString; )
			{
				*pszLclString++ = *lpszString++;
			}
			*pszLclString = '\0';
			return szNameBuffer;
		}
		else
		{
			while(*lpszString)
				*lpszString++;
			++lpszString;
		}
	}
	return (BYTE *)0;
}


BOOL EnvClearVar(BYTE *pszName)
{
	COUNT nNameLen, uArenaSeg;
	UWORD uNewEnv;
	BYTE FAR *lpszSrc, FAR *lpszDest;
	BOOL bError;
	BYTE szNameBuffer[MAX_CMDLINE], *pszLclString;
	mcb FAR *lpArena;

	strcpy(szNameBuffer, pszName);
	for(pszLclString = szNameBuffer; *pszLclString != '\0'; pszLclString++)
		*pszLclString = toupper(*pszLclString);
	nNameLen = strlen(pszName);
	lpszSrc = lpEnviron;
	uArenaSeg = FP_SEG(lpEnviron) - 1;
	lpArena = MK_FP(uArenaSeg, 0);
	uNewEnv = DosAllocMem(lpArena -> m_size, (BOOL FAR *)&bError);
	if(bError)
		return FALSE;
	lpszDest = MK_FP(uNewEnv, 0);

	/* Copy the old environment into the new and skip the one we	*/
	/* want to delete.						*/
	while('\0' != *lpszSrc)
	{
		if((0 == fstrncmp(lpszSrc, (BYTE FAR *)szNameBuffer, nNameLen))
		|| (0 == fstrncmp(lpszSrc, (BYTE FAR *)pszName, nNameLen)))
		{
			/* Match, skip the source */
			while(*lpszSrc)
				++lpszSrc;
			++lpszSrc;
		}
		else
		{
			while(*lpszSrc)
				*lpszDest++ = *lpszSrc++;
			++lpszSrc;
			*lpszDest++ = '\0';
		}
	}
	/* Copy the terminating entry					*/
	*lpszDest++ = *lpszSrc++;


	/* Copy the invocation part					*/
	*((UWORD FAR *)lpszDest)++ = *((UWORD FAR *)lpszSrc)++;
	while(*lpszSrc)
		*lpszDest++ = *lpszSrc++;

	/* and finally null terminate.					*/
	*lpszDest++ = '\0';

	EnvCopy(lpEnviron, MK_FP(uNewEnv, 0));
	DosFreeMem(uNewEnv, (BOOL FAR *)&bError);
	return !bError;
}


VOID EnvCopy(BYTE FAR *lpszDest, BYTE FAR *lpszSrc)
{
	UWORD uNewEnv;
	BOOL bError;

	while('\0' != *lpszSrc)
	{
		while(*lpszSrc)
			*lpszDest++ = *lpszSrc++;
		++lpszSrc;
		*lpszDest++ = '\0';
	}
	/* Copy the terminating entry					*/
	*lpszDest++ = *lpszSrc++;


	/* Copy the invocation part					*/
	*((UWORD FAR *)lpszDest)++ = *((UWORD FAR *)lpszSrc)++;
	while(*lpszSrc)
		*lpszDest++ = *lpszSrc++;

	/* and finally null terminate.					*/
	*lpszDest++ = '\0';
}


BOOL EnvSetVar(BYTE *pszName, BYTE *pszValue)
{
	BYTE *pszOldVal;

	/* See if it's already defined.  If it is, then we have a	*/
	/* operation.  Otherwise, we do an append.			*/
	pszOldVal = EnvLookup(pszName);
	if(pszOldVal)	/* Replace operation				*/
	{
		COUNT nSize, nNameLen;
		UWORD uEnvSize;
		psp FAR *lpPsp;
		mcb FAR *lpArena;
		UWORD uNewEnv, uArenaSeg;
		BYTE FAR *lpszSrc, FAR *lpszDest;
		BOOL bError;
		BYTE szNameBuffer[MAX_CMDLINE], *pszLclString;

		if((strlen(pszName)+strlen(pszValue)+2-strlen(pszOldVal))
		 >= EnvSizeUp())
			return FALSE;

		strcpy(szNameBuffer, pszName);
		nNameLen = strlen(szNameBuffer);
		for(pszLclString = szNameBuffer; *pszLclString != '\0'; pszLclString++)
			*pszLclString = toupper(*pszLclString);
		lpszSrc = lpEnviron;
		uArenaSeg = FP_SEG(lpEnviron) - 1;
		lpArena = MK_FP(uArenaSeg, 0);
		uNewEnv = DosAllocMem(lpArena -> m_size, (BOOL FAR *)&bError);
		if(bError)
			return FALSE;
		lpszDest = MK_FP(uNewEnv, 0);

		/* Copy the old environment into the new and append the	*/
		/* new one.						*/
		while('\0' != *lpszSrc)
		{
			if((0 == fstrncmp(lpszSrc, (BYTE FAR *)szNameBuffer, nNameLen))
			|| (0 == fstrncmp(lpszSrc, (BYTE FAR *)pszName, nNameLen)))
			{
				/* Copy in the new string */
				for(pszLclString = szNameBuffer; *pszLclString; )
					*lpszDest++ = *pszLclString++;
				*lpszDest++ = '=';
				for(pszLclString = pszValue; *pszLclString; )
					*lpszDest++ = *pszLclString++;
				*lpszDest++ = '\0';

				/* Skip past the source */
				while(*lpszSrc)
					lpszSrc++;
				++lpszSrc;
			}
			else
			{
				while(*lpszSrc)
					*lpszDest++ = *lpszSrc++;
				++lpszSrc;
				*lpszDest++ = '\0';
			}
		}

		/* Copy the terminating entry					*/
		*lpszDest++ = *lpszSrc++;

		/* Copy the invocation part					*/
		*((UWORD FAR *)lpszDest)++ = *((UWORD FAR *)lpszSrc)++;
		while(*lpszSrc)
			*lpszDest++ = *lpszSrc++;

		/* and finally null terminate.					*/
		*lpszDest++ = '\0';

		EnvCopy(lpEnviron, MK_FP(uNewEnv, 0));
		DosFreeMem(uNewEnv, (BOOL FAR *)&bError);
		return !bError;
	}
	else		/* Append operation				*/
	{
		COUNT nSize;
		UWORD uEnvSize;
		psp FAR *lpPsp;
		mcb FAR *lpArena;
		UWORD uNewEnv, uArenaSeg;
		BYTE FAR *lpszSrc, FAR *lpszDest;
		BOOL bError;
		BYTE szNameBuffer[MAX_CMDLINE], *pszLclString;

		if((strlen(pszName)+strlen(pszValue)+2) >= EnvSizeUp())
			return FALSE;

		strcpy(szNameBuffer, pszName);
		for(pszLclString = szNameBuffer; *pszLclString != '\0'; pszLclString++)
			*pszLclString = toupper(*pszLclString);
		lpszSrc = lpEnviron;
		uArenaSeg = FP_SEG(lpEnviron) - 1;
		lpArena = MK_FP(uArenaSeg, 0);
		uNewEnv = DosAllocMem(lpArena -> m_size, (BOOL FAR *)&bError);
		if(bError)
			return FALSE;
		lpszDest = MK_FP(uNewEnv, 0);

		/* Copy the old environment into the new and append the	*/
		/* new one.						*/
		while('\0' != *lpszSrc)
		{
			while(*lpszSrc)
				*lpszDest++ = *lpszSrc++;
			++lpszSrc;
			*lpszDest++ = '\0';
		}

		/* Append the new one.					*/
		for(pszLclString = szNameBuffer; *pszLclString; )
			*lpszDest++ = *pszLclString++;
		*lpszDest++ = '=';
		for(pszLclString = pszValue; *pszLclString; )
			*lpszDest++ = *pszLclString++;
		*lpszDest++ = '\0';

		/* Copy the terminating entry					*/
		*lpszDest++ = *lpszSrc++;

		/* Copy the invocation part					*/
		*((UWORD FAR *)lpszDest)++ = *((UWORD FAR *)lpszSrc)++;
		while(*lpszSrc)
			*lpszDest++ = *lpszSrc++;

		/* and finally null terminate.					*/
		*lpszDest++ = '\0';

		EnvCopy(lpEnviron, MK_FP(uNewEnv, 0));
		DosFreeMem(uNewEnv, (BOOL FAR *)&bError);
		return !bError;
	}
	return TRUE;
}


BOOL EnvDump(VOID)
{
	BYTE FAR *lpszEnv;
	BYTE *pszLclBuffer;
	COUNT nCount;

	for(lpszEnv = lpEnviron; *lpszEnv != '\0'; )
	{
		static BYTE szBuffer[MAX_CMDLINE];

		for(pszLclBuffer = szBuffer, nCount = 0;
		 nCount < MAX_CMDLINE; nCount++)
		{
			*pszLclBuffer++ = *lpszEnv++;
			if(!*lpszEnv)
				break;
		}
		*pszLclBuffer++ = '\0';
		++lpszEnv;
		printf("%s\n", szBuffer);
	}
	return TRUE;
}