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