|
Packit |
1fb8d4 |
/**
|
|
Packit |
1fb8d4 |
* WinPR: Windows Portable Runtime
|
|
Packit |
1fb8d4 |
* Process Environment Functions
|
|
Packit |
1fb8d4 |
*
|
|
Packit |
1fb8d4 |
* Copyright 2012 Marc-Andre Moreau <marcandre.moreau@gmail.com>
|
|
Packit |
1fb8d4 |
* Copyright 2013 Thincast Technologies GmbH
|
|
Packit |
1fb8d4 |
* Copyright 2013 DI (FH) Martin Haimberger <martin.haimberger@thincast.com>
|
|
Packit |
1fb8d4 |
*
|
|
Packit |
1fb8d4 |
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
Packit |
1fb8d4 |
* you may not use this file except in compliance with the License.
|
|
Packit |
1fb8d4 |
* You may obtain a copy of the License at
|
|
Packit |
1fb8d4 |
*
|
|
Packit |
1fb8d4 |
* http://www.apache.org/licenses/LICENSE-2.0
|
|
Packit |
1fb8d4 |
*
|
|
Packit |
1fb8d4 |
* Unless required by applicable law or agreed to in writing, software
|
|
Packit |
1fb8d4 |
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
Packit |
1fb8d4 |
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
Packit |
1fb8d4 |
* See the License for the specific language governing permissions and
|
|
Packit |
1fb8d4 |
* limitations under the License.
|
|
Packit |
1fb8d4 |
*/
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
#ifdef HAVE_CONFIG_H
|
|
Packit |
1fb8d4 |
#include "config.h"
|
|
Packit |
1fb8d4 |
#endif
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
#include <winpr/crt.h>
|
|
Packit |
1fb8d4 |
#include <winpr/platform.h>
|
|
Packit |
1fb8d4 |
#include <winpr/error.h>
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
#include <winpr/environment.h>
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
#ifndef _WIN32
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
#define stricmp strcasecmp
|
|
Packit |
1fb8d4 |
#define strnicmp strncasecmp
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
#include <winpr/crt.h>
|
|
Packit |
1fb8d4 |
#include <winpr/platform.h>
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
#ifdef HAVE_UNISTD_H
|
|
Packit |
1fb8d4 |
#include <unistd.h>
|
|
Packit |
1fb8d4 |
#endif
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
#if defined(__IOS__)
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
#elif defined(__MACOSX__)
|
|
Packit |
1fb8d4 |
#include <crt_externs.h>
|
|
Packit |
1fb8d4 |
#define environ (*_NSGetEnviron())
|
|
Packit |
1fb8d4 |
#endif
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
DWORD GetCurrentDirectoryA(DWORD nBufferLength, LPSTR lpBuffer)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
char* cwd;
|
|
Packit Service |
5a9772 |
size_t length;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
cwd = getcwd(NULL, 0);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!cwd)
|
|
Packit |
1fb8d4 |
return 0;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
length = strlen(cwd);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if ((nBufferLength == 0) && (lpBuffer == NULL))
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
free(cwd);
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
return (DWORD)length;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
else
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
if (lpBuffer == NULL)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
free(cwd);
|
|
Packit |
1fb8d4 |
return 0;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if ((length + 1) > nBufferLength)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
free(cwd);
|
|
Packit Service |
5a9772 |
return (DWORD)(length + 1);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
memcpy(lpBuffer, cwd, length + 1);
|
|
Packit Service |
5a9772 |
free(cwd);
|
|
Packit |
1fb8d4 |
return length;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return 0;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
DWORD GetCurrentDirectoryW(DWORD nBufferLength, LPWSTR lpBuffer)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return 0;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL SetCurrentDirectoryA(LPCSTR lpPathName)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return TRUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL SetCurrentDirectoryW(LPCWSTR lpPathName)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return TRUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
DWORD SearchPathA(LPCSTR lpPath, LPCSTR lpFileName, LPCSTR lpExtension, DWORD nBufferLength,
|
|
Packit Service |
5a9772 |
LPSTR lpBuffer, LPSTR* lpFilePart)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return 0;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
DWORD SearchPathW(LPCWSTR lpPath, LPCWSTR lpFileName, LPCWSTR lpExtension, DWORD nBufferLength,
|
|
Packit Service |
5a9772 |
LPWSTR lpBuffer, LPWSTR* lpFilePart)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return 0;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
LPSTR GetCommandLineA(VOID)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
LPWSTR GetCommandLineW(VOID)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL NeedCurrentDirectoryForExePathA(LPCSTR ExeName)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return TRUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL NeedCurrentDirectoryForExePathW(LPCWSTR ExeName)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return TRUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
#endif
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
#if !defined(_WIN32) || defined(_UWP)
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
DWORD GetEnvironmentVariableA(LPCSTR lpName, LPSTR lpBuffer, DWORD nSize)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
#if !defined(_UWP)
|
|
Packit Service |
5a9772 |
size_t length;
|
|
Packit |
1fb8d4 |
char* env = NULL;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
env = getenv(lpName);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!env)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
SetLastError(ERROR_ENVVAR_NOT_FOUND);
|
|
Packit |
1fb8d4 |
return 0;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
length = strlen(env);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if ((length + 1 > nSize) || (!lpBuffer))
|
|
Packit |
1fb8d4 |
return length + 1;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
CopyMemory(lpBuffer, env, length);
|
|
Packit |
1fb8d4 |
lpBuffer[length] = '\0';
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return length;
|
|
Packit |
1fb8d4 |
#else
|
|
Packit |
1fb8d4 |
SetLastError(ERROR_ENVVAR_NOT_FOUND);
|
|
Packit |
1fb8d4 |
return 0;
|
|
Packit |
1fb8d4 |
#endif
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
DWORD GetEnvironmentVariableW(LPCWSTR lpName, LPWSTR lpBuffer, DWORD nSize)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
SetLastError(ERROR_ENVVAR_NOT_FOUND);
|
|
Packit |
1fb8d4 |
return 0;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL SetEnvironmentVariableA(LPCSTR lpName, LPCSTR lpValue)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
#if !defined(_UWP)
|
|
Packit |
1fb8d4 |
if (!lpName)
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (lpValue)
|
|
Packit |
1fb8d4 |
{
|
|
Packit Service |
5a9772 |
if (0 != setenv(lpName, lpValue, 1))
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
else
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
if (0 != unsetenv(lpName))
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return TRUE;
|
|
Packit |
1fb8d4 |
#else
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
#endif
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL SetEnvironmentVariableW(LPCWSTR lpName, LPCWSTR lpValue)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
/**
|
|
Packit |
1fb8d4 |
* GetEnvironmentStrings function:
|
|
Packit |
1fb8d4 |
* http://msdn.microsoft.com/en-us/library/windows/desktop/ms683187/
|
|
Packit |
1fb8d4 |
*
|
|
Packit |
1fb8d4 |
* The GetEnvironmentStrings function returns a pointer to a block of memory
|
|
Packit |
1fb8d4 |
* that contains the environment variables of the calling process (both the
|
|
Packit |
1fb8d4 |
* system and the user environment variables). Each environment block contains
|
|
Packit |
1fb8d4 |
* the environment variables in the following format:
|
|
Packit |
1fb8d4 |
*
|
|
Packit |
1fb8d4 |
* Var1=Value1\0
|
|
Packit |
1fb8d4 |
* Var2=Value2\0
|
|
Packit |
1fb8d4 |
* Var3=Value3\0
|
|
Packit |
1fb8d4 |
* ...
|
|
Packit |
1fb8d4 |
* VarN=ValueN\0\0
|
|
Packit |
1fb8d4 |
*/
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
extern char** environ;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
LPCH GetEnvironmentStringsA(VOID)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
#if !defined(_UWP)
|
|
Packit |
1fb8d4 |
char* p;
|
|
Packit Service |
5a9772 |
size_t offset;
|
|
Packit Service |
5a9772 |
size_t length;
|
|
Packit |
1fb8d4 |
char** envp;
|
|
Packit |
1fb8d4 |
DWORD cchEnvironmentBlock;
|
|
Packit |
1fb8d4 |
LPCH lpszEnvironmentBlock;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
offset = 0;
|
|
Packit |
1fb8d4 |
envp = environ;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
cchEnvironmentBlock = 128;
|
|
Packit Service |
5a9772 |
lpszEnvironmentBlock = (LPCH)calloc(cchEnvironmentBlock, sizeof(CHAR));
|
|
Packit |
1fb8d4 |
if (!lpszEnvironmentBlock)
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
while (*envp)
|
|
Packit |
1fb8d4 |
{
|
|
Packit Service |
5a9772 |
length = strlen(*envp);
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
while ((offset + length + 8) > cchEnvironmentBlock)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
DWORD new_size;
|
|
Packit |
1fb8d4 |
LPCH new_blk;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
new_size = cchEnvironmentBlock * 2;
|
|
Packit Service |
5a9772 |
new_blk = (LPCH)realloc(lpszEnvironmentBlock, new_size * sizeof(CHAR));
|
|
Packit |
1fb8d4 |
if (!new_blk)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
free(lpszEnvironmentBlock);
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
lpszEnvironmentBlock = new_blk;
|
|
Packit |
1fb8d4 |
cchEnvironmentBlock = new_size;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
p = &(lpszEnvironmentBlock[offset]);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
CopyMemory(p, *envp, length * sizeof(CHAR));
|
|
Packit |
1fb8d4 |
p[length] = '\0';
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
offset += (length + 1);
|
|
Packit |
1fb8d4 |
envp++;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
lpszEnvironmentBlock[offset] = '\0';
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return lpszEnvironmentBlock;
|
|
Packit |
1fb8d4 |
#else
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
#endif
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
LPWCH GetEnvironmentStringsW(VOID)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL SetEnvironmentStringsA(LPCH NewEnvironment)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return TRUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL SetEnvironmentStringsW(LPWCH NewEnvironment)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return TRUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
DWORD ExpandEnvironmentStringsA(LPCSTR lpSrc, LPSTR lpDst, DWORD nSize)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return 0;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
DWORD ExpandEnvironmentStringsW(LPCWSTR lpSrc, LPWSTR lpDst, DWORD nSize)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return 0;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL FreeEnvironmentStringsA(LPCH lpszEnvironmentBlock)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
free(lpszEnvironmentBlock);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return TRUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL FreeEnvironmentStringsW(LPWCH lpszEnvironmentBlock)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return TRUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
#endif
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
LPCH MergeEnvironmentStrings(PCSTR original, PCSTR merge)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
const char* cp;
|
|
Packit |
1fb8d4 |
char* p;
|
|
Packit Service |
5a9772 |
size_t offset;
|
|
Packit Service |
5a9772 |
size_t length;
|
|
Packit |
1fb8d4 |
const char* envp;
|
|
Packit |
1fb8d4 |
DWORD cchEnvironmentBlock;
|
|
Packit |
1fb8d4 |
LPCH lpszEnvironmentBlock;
|
|
Packit |
1fb8d4 |
const char** mergeStrings;
|
|
Packit Service |
5a9772 |
size_t mergeStringLength;
|
|
Packit Service |
5a9772 |
size_t mergeArraySize = 128;
|
|
Packit Service |
5a9772 |
size_t run;
|
|
Packit Service |
5a9772 |
size_t mergeLength;
|
|
Packit Service |
5a9772 |
size_t foundMerge;
|
|
Packit |
1fb8d4 |
char* foundEquals;
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
mergeStrings = (LPCSTR*)calloc(mergeArraySize, sizeof(char*));
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!mergeStrings)
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
mergeStringLength = 0;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
cp = merge;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
while (*cp && *(cp + 1))
|
|
Packit |
1fb8d4 |
{
|
|
Packit Service |
5a9772 |
length = strlen(cp);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (mergeStringLength == mergeArraySize)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
const char** new_str;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
mergeArraySize += 128;
|
|
Packit Service |
5a9772 |
new_str = (const char**)realloc((void*)mergeStrings, mergeArraySize * sizeof(char*));
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!new_str)
|
|
Packit |
1fb8d4 |
{
|
|
Packit Service |
5a9772 |
free((void*)mergeStrings);
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
mergeStrings = new_str;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
mergeStrings[mergeStringLength] = cp;
|
|
Packit |
1fb8d4 |
cp += length + 1;
|
|
Packit |
1fb8d4 |
mergeStringLength++;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
offset = 0;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
cchEnvironmentBlock = 128;
|
|
Packit Service |
5a9772 |
lpszEnvironmentBlock = (LPCH)calloc(cchEnvironmentBlock, sizeof(CHAR));
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!lpszEnvironmentBlock)
|
|
Packit |
1fb8d4 |
{
|
|
Packit Service |
5a9772 |
free((void*)mergeStrings);
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
envp = original;
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
while ((original != NULL) && (*envp && *(envp + 1)))
|
|
Packit |
1fb8d4 |
{
|
|
Packit Service |
5a9772 |
size_t old_offset = offset;
|
|
Packit Service |
5a9772 |
length = strlen(envp);
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
while ((offset + length + 8) > cchEnvironmentBlock)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
LPCH tmp;
|
|
Packit |
1fb8d4 |
cchEnvironmentBlock *= 2;
|
|
Packit Service |
5a9772 |
tmp = (LPCH)realloc(lpszEnvironmentBlock, cchEnvironmentBlock * sizeof(CHAR));
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!tmp)
|
|
Packit |
1fb8d4 |
{
|
|
Packit Service |
5a9772 |
free((void*)lpszEnvironmentBlock);
|
|
Packit Service |
5a9772 |
free((void*)mergeStrings);
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
lpszEnvironmentBlock = tmp;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
p = &(lpszEnvironmentBlock[offset]);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
// check if this value is in the mergeStrings
|
|
Packit |
1fb8d4 |
foundMerge = 0;
|
|
Packit Service |
5a9772 |
for (run = 0; run < mergeStringLength; run++)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
if (!mergeStrings[run])
|
|
Packit |
1fb8d4 |
continue;
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
mergeLength = strlen(mergeStrings[run]);
|
|
Packit |
1fb8d4 |
foundEquals = strstr(mergeStrings[run], "=");
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!foundEquals)
|
|
Packit |
1fb8d4 |
continue;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (strncmp(envp, mergeStrings[run], foundEquals - mergeStrings[run] + 1) == 0)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
// found variable in merge list ... use this ....
|
|
Packit |
1fb8d4 |
if (*(foundEquals + 1) == '\0')
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
// check if the argument is set ... if not remove variable ...
|
|
Packit |
1fb8d4 |
foundMerge = 1;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
else
|
|
Packit |
1fb8d4 |
{
|
|
Packit Service |
5a9772 |
while ((offset + mergeLength + 8) > cchEnvironmentBlock)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
LPCH tmp;
|
|
Packit |
1fb8d4 |
cchEnvironmentBlock *= 2;
|
|
Packit Service |
5a9772 |
tmp =
|
|
Packit Service |
5a9772 |
(LPCH)realloc(lpszEnvironmentBlock, cchEnvironmentBlock * sizeof(CHAR));
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!tmp)
|
|
Packit |
1fb8d4 |
{
|
|
Packit Service |
5a9772 |
free((void*)lpszEnvironmentBlock);
|
|
Packit Service |
5a9772 |
free((void*)mergeStrings);
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
lpszEnvironmentBlock = tmp;
|
|
Packit |
1fb8d4 |
p = &(lpszEnvironmentBlock[old_offset]);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
foundMerge = 1;
|
|
Packit |
1fb8d4 |
CopyMemory(p, mergeStrings[run], mergeLength);
|
|
Packit |
1fb8d4 |
mergeStrings[run] = NULL;
|
|
Packit |
1fb8d4 |
p[mergeLength] = '\0';
|
|
Packit |
1fb8d4 |
offset += (mergeLength + 1);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (foundMerge == 0)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
CopyMemory(p, envp, length * sizeof(CHAR));
|
|
Packit |
1fb8d4 |
p[length] = '\0';
|
|
Packit |
1fb8d4 |
offset += (length + 1);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
envp += (length + 1);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
// now merge the not already merged env
|
|
Packit Service |
5a9772 |
for (run = 0; run < mergeStringLength; run++)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
if (!mergeStrings[run])
|
|
Packit |
1fb8d4 |
continue;
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
mergeLength = strlen(mergeStrings[run]);
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
while ((offset + mergeLength + 8) > cchEnvironmentBlock)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
LPCH tmp;
|
|
Packit |
1fb8d4 |
cchEnvironmentBlock *= 2;
|
|
Packit Service |
5a9772 |
tmp = (LPCH)realloc(lpszEnvironmentBlock, cchEnvironmentBlock * sizeof(CHAR));
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!tmp)
|
|
Packit |
1fb8d4 |
{
|
|
Packit Service |
5a9772 |
free((void*)lpszEnvironmentBlock);
|
|
Packit Service |
5a9772 |
free((void*)mergeStrings);
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
lpszEnvironmentBlock = tmp;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
p = &(lpszEnvironmentBlock[offset]);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
CopyMemory(p, mergeStrings[run], mergeLength);
|
|
Packit |
1fb8d4 |
mergeStrings[run] = NULL;
|
|
Packit |
1fb8d4 |
p[mergeLength] = '\0';
|
|
Packit |
1fb8d4 |
offset += (mergeLength + 1);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
lpszEnvironmentBlock[offset] = '\0';
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
free((void*)mergeStrings);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return lpszEnvironmentBlock;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
DWORD GetEnvironmentVariableEBA(LPCSTR envBlock, LPCSTR lpName, LPSTR lpBuffer, DWORD nSize)
|
|
Packit |
1fb8d4 |
{
|
|
Packit Service |
5a9772 |
size_t vLength = 0;
|
|
Packit |
1fb8d4 |
char* env = NULL;
|
|
Packit |
1fb8d4 |
char* foundEquals;
|
|
Packit |
1fb8d4 |
const char* penvb = envBlock;
|
|
Packit Service |
5a9772 |
size_t nLength, fLength, lpNameLength;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!lpName || NULL == envBlock)
|
|
Packit |
1fb8d4 |
return 0;
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
lpNameLength = strlen(lpName);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (lpNameLength < 1)
|
|
Packit |
1fb8d4 |
return 0;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
while (*penvb && *(penvb + 1))
|
|
Packit |
1fb8d4 |
{
|
|
Packit Service |
5a9772 |
fLength = strlen(penvb);
|
|
Packit Service |
5a9772 |
foundEquals = strstr(penvb, "=");
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!foundEquals)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
/* if no = sign is found the envBlock is broken */
|
|
Packit |
1fb8d4 |
return 0;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
nLength = (foundEquals - penvb);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (nLength != lpNameLength)
|
|
Packit |
1fb8d4 |
{
|
|
Packit Service |
5a9772 |
penvb += (fLength + 1);
|
|
Packit |
1fb8d4 |
continue;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (strncmp(penvb, lpName, nLength) == 0)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
env = foundEquals + 1;
|
|
Packit |
1fb8d4 |
break;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
penvb += (fLength + 1);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!env)
|
|
Packit |
1fb8d4 |
return 0;
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
vLength = strlen(env);
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
if ((vLength + 1 > nSize) || (!lpBuffer))
|
|
Packit |
1fb8d4 |
return vLength + 1;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
CopyMemory(lpBuffer, env, vLength + 1);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return vLength;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL SetEnvironmentVariableEBA(LPSTR* envBlock, LPCSTR lpName, LPCSTR lpValue)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
int length;
|
|
Packit |
1fb8d4 |
char* envstr;
|
|
Packit |
1fb8d4 |
char* newEB;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!lpName)
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (lpValue)
|
|
Packit |
1fb8d4 |
{
|
|
Packit Service |
5a9772 |
length = (int)(strlen(lpName) + strlen(lpValue) + 2); /* +2 because of = and \0 */
|
|
Packit Service |
5a9772 |
envstr = (char*)malloc(length + 1); /* +1 because of closing \0 */
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!envstr)
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
sprintf_s(envstr, length, "%s=%s", lpName, lpValue);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
else
|
|
Packit |
1fb8d4 |
{
|
|
Packit Service |
5a9772 |
length = (int)strlen(lpName) + 2; /* +2 because of = and \0 */
|
|
Packit Service |
5a9772 |
envstr = (char*)malloc(length + 1); /* +1 because of closing \0 */
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!envstr)
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
sprintf_s(envstr, length, "%s=", lpName);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
envstr[length] = '\0';
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
newEB = MergeEnvironmentStrings((LPCSTR)*envBlock, envstr);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
free(envstr);
|
|
Packit |
1fb8d4 |
free(*envBlock);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
*envBlock = newEB;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return TRUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
char** EnvironmentBlockToEnvpA(LPCH lpszEnvironmentBlock)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
char* p;
|
|
Packit Service |
5a9772 |
SSIZE_T index;
|
|
Packit Service |
5a9772 |
size_t count;
|
|
Packit Service |
5a9772 |
size_t length;
|
|
Packit |
1fb8d4 |
char** envp = NULL;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
count = 0;
|
|
Packit |
1fb8d4 |
if (!lpszEnvironmentBlock)
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
p = (char*)lpszEnvironmentBlock;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
while (p[0] && p[1])
|
|
Packit |
1fb8d4 |
{
|
|
Packit Service |
5a9772 |
length = strlen(p);
|
|
Packit |
1fb8d4 |
p += (length + 1);
|
|
Packit |
1fb8d4 |
count++;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
index = 0;
|
|
Packit Service |
5a9772 |
p = (char*)lpszEnvironmentBlock;
|
|
Packit |
1fb8d4 |
|
|
Packit Service |
5a9772 |
envp = (char**)calloc(count + 1, sizeof(char*));
|
|
Packit |
1fb8d4 |
if (!envp)
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
envp[count] = NULL;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
while (p[0] && p[1])
|
|
Packit |
1fb8d4 |
{
|
|
Packit Service |
5a9772 |
length = strlen(p);
|
|
Packit |
1fb8d4 |
envp[index] = _strdup(p);
|
|
Packit |
1fb8d4 |
if (!envp[index])
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
for (index -= 1; index >= 0; --index)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
free(envp[index]);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
free(envp);
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
p += (length + 1);
|
|
Packit |
1fb8d4 |
index++;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return envp;
|
|
Packit |
1fb8d4 |
}
|