148 lines
2.7 KiB
C
148 lines
2.7 KiB
C
/**
|
|
* WinPR: Windows Portable Runtime
|
|
* Shell Functions
|
|
*
|
|
* Copyright 2015 Dell Software <Mike.McDonald@software.dell.com>
|
|
*
|
|
* 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/shell.h>
|
|
|
|
/**
|
|
* shell32.dll:
|
|
*
|
|
* GetUserProfileDirectoryA
|
|
* GetUserProfileDirectoryW
|
|
*/
|
|
|
|
#ifndef _WIN32
|
|
|
|
#include <winpr/crt.h>
|
|
|
|
#ifdef HAVE_UNISTD_H
|
|
#include <unistd.h>
|
|
#endif
|
|
|
|
#include <pwd.h>
|
|
#include <grp.h>
|
|
|
|
#include "../handle/handle.h"
|
|
|
|
#include "../security/security.h"
|
|
|
|
BOOL GetUserProfileDirectoryA(HANDLE hToken, LPSTR lpProfileDir, LPDWORD lpcchSize)
|
|
{
|
|
char* buf;
|
|
int buflen;
|
|
int status;
|
|
DWORD cchDirSize;
|
|
struct passwd pwd;
|
|
struct passwd* pw = NULL;
|
|
WINPR_ACCESS_TOKEN* token;
|
|
|
|
token = (WINPR_ACCESS_TOKEN*) hToken;
|
|
|
|
if (!token || (token->Type != HANDLE_TYPE_ACCESS_TOKEN) || !lpcchSize)
|
|
{
|
|
SetLastError(ERROR_INVALID_PARAMETER);
|
|
return FALSE;
|
|
}
|
|
|
|
buflen = sysconf(_SC_GETPW_R_SIZE_MAX);
|
|
|
|
if (buflen == -1)
|
|
buflen = 8196;
|
|
|
|
buf = (char*) malloc(buflen);
|
|
|
|
if (!buf)
|
|
return FALSE;
|
|
|
|
status = getpwnam_r(token->Username, &pwd, buf, buflen, &pw);
|
|
|
|
if ((status != 0) || !pw)
|
|
{
|
|
SetLastError(ERROR_INVALID_PARAMETER);
|
|
free (buf);
|
|
return FALSE;
|
|
}
|
|
|
|
cchDirSize = strlen(pw->pw_dir) + 1;
|
|
|
|
if (!lpProfileDir || (*lpcchSize < cchDirSize))
|
|
{
|
|
*lpcchSize = cchDirSize;
|
|
SetLastError(ERROR_INSUFFICIENT_BUFFER);
|
|
free(buf);
|
|
return FALSE;
|
|
}
|
|
|
|
ZeroMemory(lpProfileDir, *lpcchSize);
|
|
strcpy(lpProfileDir, pw->pw_dir);
|
|
*lpcchSize = cchDirSize;
|
|
free(buf);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL GetUserProfileDirectoryW(HANDLE hToken, LPWSTR lpProfileDir, LPDWORD lpcchSize)
|
|
{
|
|
BOOL bStatus;
|
|
DWORD cchSizeA;
|
|
LPSTR lpProfileDirA;
|
|
|
|
if (!lpcchSize)
|
|
{
|
|
SetLastError(ERROR_INVALID_PARAMETER);
|
|
return FALSE;
|
|
}
|
|
|
|
cchSizeA = *lpcchSize;
|
|
lpProfileDirA = NULL;
|
|
|
|
if (lpProfileDir)
|
|
{
|
|
lpProfileDirA = (LPSTR) malloc(cchSizeA);
|
|
|
|
if (lpProfileDirA == NULL)
|
|
{
|
|
SetLastError(ERROR_OUTOFMEMORY);
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
bStatus = GetUserProfileDirectoryA(hToken, lpProfileDirA, &cchSizeA);
|
|
|
|
if (bStatus)
|
|
{
|
|
MultiByteToWideChar(CP_ACP, 0, lpProfileDirA, cchSizeA, lpProfileDir, *lpcchSize);
|
|
}
|
|
|
|
if (lpProfileDirA)
|
|
{
|
|
free(lpProfileDirA);
|
|
}
|
|
|
|
*lpcchSize = cchSizeA;
|
|
|
|
return bStatus;
|
|
}
|
|
|
|
#endif
|
|
|