diff options
Diffstat (limited to 'private/nw/nwscript/env.c')
| -rw-r--r-- | private/nw/nwscript/env.c | 352 |
1 files changed, 352 insertions, 0 deletions
diff --git a/private/nw/nwscript/env.c b/private/nw/nwscript/env.c new file mode 100644 index 000000000..dfece205b --- /dev/null +++ b/private/nw/nwscript/env.c @@ -0,0 +1,352 @@ +/************************************************************************* +* +* ENV.C +* +* Environment export routines +* +* Copyright (c) 1995 Microsoft Corporation +* +* $Log: N:\NT\PRIVATE\NW4\NWSCRIPT\VCS\ENV.C $ +* +* Rev 1.2 10 Apr 1996 14:22:28 terryt +* Hotfix for 21181hq +* +* Rev 1.2 12 Mar 1996 19:53:48 terryt +* Relative NDS names and merge +* +* Rev 1.1 22 Dec 1995 14:24:40 terryt +* Add Microsoft headers +* +* Rev 1.0 15 Nov 1995 18:06:58 terryt +* Initial revision. +* +* Rev 1.2 25 Aug 1995 16:22:50 terryt +* Capture support +* +* Rev 1.1 23 May 1995 19:36:54 terryt +* Spruce up source +* +* Rev 1.0 15 May 1995 19:10:34 terryt +* Initial revision. +* +*************************************************************************/ +#include <stdio.h> +#include <direct.h> +#include <time.h> +#include <stdlib.h> + +#include <nt.h> +#include <ntrtl.h> +#include <nturtl.h> +#include <windows.h> + +#include "nwscript.h" + +#define MAX_PATH_LEN 2048 +#define PATH "Path" +#define LIBPATH "LibPath" +#define OS2LIBPATH "Os2LibPath" + +unsigned char * Path_Value = NULL; +unsigned char * LibPath_Value = NULL; +unsigned char * Os2LibPath_Value = NULL; + + +/******************************************************************** + + GetOldPaths + +Routine Description: + + Save the orginal paths for + Path + LibPath + Os2LibPath + +Arguments: + none + +Return Value: + none + + *******************************************************************/ +void +GetOldPaths( void ) +{ + if (!(Path_Value = (unsigned char *)LocalAlloc( LPTR, MAX_PATH_LEN ))) + { + DisplayMessage(IDR_NOT_ENOUGH_MEMORY); + return; + } + GetEnvironmentVariableA( PATH, Path_Value, MAX_PATH_LEN ); + if (!(LibPath_Value = (unsigned char *)LocalAlloc( LPTR, MAX_PATH_LEN))) + { + DisplayMessage(IDR_NOT_ENOUGH_MEMORY); + return; + } + GetEnvironmentVariableA( LIBPATH, LibPath_Value, MAX_PATH_LEN ); + if (!(Os2LibPath_Value = (unsigned char *)LocalAlloc( LPTR, MAX_PATH_LEN))) + { + DisplayMessage(IDR_NOT_ENOUGH_MEMORY); + return; + } + GetEnvironmentVariableA( OS2LIBPATH, Os2LibPath_Value, MAX_PATH_LEN ); +} + + +/******************************************************************** + + AdjustPath + +Routine Description: + + Given an old path and a new path, merge the two togther. + Basically, the Adjusted path is the old path with the + new values at the end, minus any duplicates. + +Arguments: + + Value - New path + OldPath_Value - Old path + AdjustedValue - New value (allocated) + +Return Value: + none + + *******************************************************************/ +void +AdjustPath( unsigned char * Value, + unsigned char * OldPath_Value, + unsigned char ** AdjustedValue ) +{ + unsigned char * tokenPath; + unsigned char * clipStart; + unsigned char * clipEnd; + unsigned char * tokenSearch; + unsigned char * tokenNext; + + if (!(*AdjustedValue = (unsigned char *)LocalAlloc( LPTR, MAX_PATH_LEN))) + { + DisplayMessage(IDR_NOT_ENOUGH_MEMORY); + return; + } + strncpy( *AdjustedValue, Value, MAX_PATH_LEN ); + + if (!(tokenSearch = (unsigned char *)LocalAlloc( LPTR, MAX_PATH_LEN))) + { + DisplayMessage(IDR_NOT_ENOUGH_MEMORY); + (void) LocalFree((HLOCAL) *AdjustedValue) ; + return; + } + strncpy( tokenSearch, OldPath_Value, MAX_PATH_LEN ); + + tokenNext = tokenSearch; + + if ( !tokenNext || !tokenNext[0] ) + tokenPath = NULL; + else { + tokenPath = tokenNext; + tokenNext = strchr( tokenPath, ';' ); + if ( tokenNext ) { + *tokenNext++ = 0; + } + } + + while ( tokenPath != NULL ) + { + if ( clipStart = strstr( *AdjustedValue, tokenPath ) ) { + if ( clipEnd = strchr( clipStart, ';' ) ) { + memmove( clipStart, clipEnd + 1, strlen( clipEnd + 1 ) + 1 ); + } + else { + clipStart[0] = 0; + } + } + + if ( !tokenNext || !tokenNext[0] ) + tokenPath = NULL; + else { + tokenPath = tokenNext; + tokenNext = strchr( tokenPath, ';' ); + if ( tokenNext ) { + *tokenNext++ = 0; + } + } + } + (void) LocalFree((HLOCAL) tokenSearch) ; + +} + +/******************************************************************** + + ExportEnv + +Routine Description: + + Export environment value to the registry + +Arguments: + + EnvString - Environment string + +Return Value: + none + + *******************************************************************/ +void +ExportEnv( unsigned char * EnvString ) +{ + HKEY ScriptEnvironmentKey; + NTSTATUS Status; + unsigned char * Value; + unsigned char * ValueName; + unsigned char * AdjustedValue = NULL; + + ValueName = EnvString; + Value = strchr( EnvString, '=' ); + + if ( Value == NULL ) { + wprintf(L"Bad Environment string\n"); + + return; + } + Value++; + + if (!(ValueName = (unsigned char *)LocalAlloc( LPTR, Value-EnvString + 1))) + { + DisplayMessage(IDR_NOT_ENOUGH_MEMORY); + return; + } + strncpy( ValueName, EnvString, Value-EnvString - 1 ); + + if ( !_strcmpi( ValueName, PATH ) ) { + AdjustPath( Value, Path_Value, &AdjustedValue ); + Value = AdjustedValue; + } + else if ( !_strcmpi( ValueName, LIBPATH ) ) { + AdjustPath( Value, LibPath_Value, &AdjustedValue ); + Value = AdjustedValue; + } + else if ( !_strcmpi( ValueName, OS2LIBPATH ) ) { + AdjustPath( Value, Os2LibPath_Value, &AdjustedValue ); + Value = AdjustedValue; + } + + Status = RegCreateKeyExW( HKEY_CURRENT_USER, + SCRIPT_ENVIRONMENT_VALUENAME, + 0, + WIN31_CLASS, + REG_OPTION_VOLATILE, + KEY_WRITE, + NULL, // security attr + &ScriptEnvironmentKey, + NULL + ); + + if ( NT_SUCCESS(Status)) { + + Status = RegSetValueExA( ScriptEnvironmentKey, + ValueName, + 0, + REG_SZ, + (LPVOID) Value, + strlen( Value ) + 1 + ); + } + else { + wprintf(L"Cannot create registry key\n"); + } + + (void) LocalFree((HLOCAL) ValueName) ; + + if ( AdjustedValue ) + (void) LocalFree((HLOCAL) AdjustedValue) ; + + RegCloseKey( ScriptEnvironmentKey ); +} + +/******************************************************************** + + ExportCurrentDirectory + +Routine Description: + + Return the first non-local drive + +Arguments: + + DriveNum - Number of drive 1-26 + +Return Value: + none + + *******************************************************************/ +void +ExportCurrentDirectory( int DriveNum ) +{ + char DriveName[10]; + HKEY ScriptEnvironmentKey; + NTSTATUS Status; + char CurrentPath[MAX_PATH_LEN]; + + strcpy( DriveName, "=A:" ); + + DriveName[1] += (DriveNum - 1); + + if ( NTGetCurrentDirectory( (unsigned char)(DriveNum - 1), CurrentPath ) ) + return; + + Status = RegCreateKeyExW( HKEY_CURRENT_USER, + SCRIPT_ENVIRONMENT_VALUENAME, + 0, + WIN31_CLASS, + REG_OPTION_VOLATILE, + KEY_WRITE, + NULL, // security attr + &ScriptEnvironmentKey, + NULL + ); + + if ( NT_SUCCESS(Status)) { + + Status = RegSetValueExA( ScriptEnvironmentKey, + DriveName, + 0, + REG_SZ, + (LPVOID) CurrentPath, + strlen( CurrentPath ) + 1 + ); + } + else { + wprintf(L"Cannot open registry key\n"); + } + + RegCloseKey( ScriptEnvironmentKey ); + +} + + +/******************************************************************** + + ExportCurrentDrive + +Routine Description: + + Export current drive to registry + NOT IMPLEMENTED + +Arguments: + + DriveNum - drive number + +Return Value: + none + + *******************************************************************/ +void +ExportCurrentDrive( int DriveNum ) +{ + /* + * Don't know if we want to do this or how. + */ +} |