path.c

来自「PocketCMD是与pocketconsole配合实用的命令行解释器（Shel」· C语言 代码 · 共 86 行

/*
 *  PATH.C - path internal command.
 *
 *
 *  History:
 *
 *    17 Jul 1998 (John P Price)
 *        Separated commands into individual files.
 *
 *    27-Jul-1998 (John P Price <[e-mail]>)
 *        added config.h include
 *
 *    09-Dec-1998 (Eric Kohl <[e-mail]>)
 *        Added help text ("/?").
 *
 *    18-Jan-1999 (Eric Kohl <[e-mail]>)
 *        Unicode ready!
 *
 *    18-Jan-1999 (Eric Kohl <[e-mail]>)
 *        Redirection safe!
 *
 *    24-Jan-1999 (Eric Kohl <[e-mail]>)
 *        Fixed Win32 environment handling.
 */

#include "config.h"

#ifdef INCLUDE_CMD_PATH

#include "cmd.h"


/* size of environment variable buffer */
#define ENV_BUFFER_SIZE 1024


INT cmd_path (LPTSTR cmd, LPTSTR param)
{
	if (!_tcsncmp (param, _T("/?"), 2))
	{
		ConOutPuts (_T("Displays or sets a search path for executable files.\n\n")
				   _T("PATH [[drive:]path[;...]]\nPATH ;\n\n")
				   _T("Type PATH ; to clear all search-path settings and direct the command shell\n")
				   _T("to search only in the current directory.\n")
				   _T("Type PATH without parameters to display the current path.\n"));
		return 0;
	}

	/* if param is empty, display the PATH environment variable */
	if (!param || !*param)
	{
		DWORD  dwBuffer;
		LPTSTR pszBuffer;

		pszBuffer = (LPTSTR)malloc (ENV_BUFFER_SIZE * sizeof(TCHAR));
		dwBuffer = GetEnvironmentVariable (_T("PATH"), pszBuffer, ENV_BUFFER_SIZE);
		if (dwBuffer == 0)
		{
			ConErrPrintf (_T("CMD: Not in environment \"PATH\"\n"));
			return 0;
		}
		else if (dwBuffer > ENV_BUFFER_SIZE)
		{
			pszBuffer = (LPTSTR)realloc (pszBuffer, dwBuffer * sizeof (TCHAR));
			GetEnvironmentVariable (_T("PATH"), pszBuffer, ENV_BUFFER_SIZE);
		}

		ConOutPrintf (_T("PATH=%s\n"), pszBuffer);
		free (pszBuffer);

		return 0;
	}

	/* skip leading '=' */
	if (*param == _T('='))
		param++;

	/* set PATH environment variable */
	if (!SetEnvironmentVariable (_T("PATH"), param))
		return 1;

	return 0;
}

#endif