164 lines
3.8 KiB
C++
164 lines
3.8 KiB
C++
/** (c) Nullsoft, Inc. C O N F I D E N T I A L
|
|
** Filename:
|
|
** Project:
|
|
** Description:
|
|
** Author: Ben Allison benski@nullsoft.com
|
|
** Created:
|
|
**/
|
|
|
|
#include <windows.h>
|
|
#include <shlwapi.h>
|
|
#include <strsafe.h>
|
|
#include <shlobj.h>
|
|
#include <shellapi.h>
|
|
#include "../Winamp/in2.h"
|
|
extern In_Module mod;
|
|
#ifdef __cplusplus
|
|
extern "C" BOOL UtilGetSpecialFolderPath(HWND hwnd, TCHAR *path, int folder)
|
|
#else
|
|
BOOL UtilGetSpecialFolderPath(HWND hwnd, TCHAR *path, int folder)
|
|
#endif
|
|
{
|
|
ITEMIDLIST *pidl; // Shell Item ID List ptr
|
|
IMalloc *imalloc; // Shell IMalloc interface ptr
|
|
BOOL result; // Return value
|
|
|
|
if (SHGetSpecialFolderLocation(hwnd, folder, &pidl) != NOERROR)
|
|
return FALSE;
|
|
|
|
result = SHGetPathFromIDList (pidl, path);
|
|
|
|
if (SHGetMalloc (&imalloc) == NOERROR)
|
|
{
|
|
#ifdef __cplusplus
|
|
imalloc->Free(pidl);
|
|
imalloc->Release();
|
|
#else
|
|
imalloc->lpVtbl->Free(imalloc, pidl);
|
|
imalloc->lpVtbl->Release(imalloc);
|
|
#endif
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
|
|
/*
|
|
This ugly function converts our specially coded paths
|
|
environment variables are surrounded with %'s and CSIDLs are surrounded with {}
|
|
note that CSIDLs need to be DECIMAL!
|
|
*/
|
|
#ifdef __cplusplus
|
|
extern "C" void ResolveEnvironmentVariables2(wchar_t *string, wchar_t *destString, size_t stringSize)
|
|
#else
|
|
void ResolveEnvironmentVariables2(TCHAR *string, TCHAR *destString, size_t stringSize)
|
|
#endif
|
|
{
|
|
//char *saveStart = string;
|
|
wchar_t *p = string;
|
|
int inPercent = 0, inBrace = 0;
|
|
wchar_t environString[MAX_PATH] = {0};
|
|
wchar_t *dest = destString;
|
|
wchar_t helper[MAX_PATH] = {0};
|
|
wchar_t *pEnv = environString;
|
|
|
|
*dest = 0;
|
|
|
|
while (p && *p)
|
|
{
|
|
if (*p == L'%')
|
|
{
|
|
if (inPercent)
|
|
{
|
|
*pEnv = 0;
|
|
helper[0] = 0;
|
|
GetEnvironmentVariableW(environString, helper, MAX_PATH);
|
|
StringCchCatW(dest, stringSize, helper);
|
|
pEnv = environString;
|
|
string = CharNextW(p);
|
|
*p = 0;
|
|
p = string;
|
|
}
|
|
else
|
|
{
|
|
wchar_t *newP = CharNextW(p);
|
|
*p = 0;
|
|
StringCchCatW(dest, stringSize, string);
|
|
string = p = newP;
|
|
}
|
|
|
|
inPercent = !inPercent;
|
|
}
|
|
else if (*p == L'{')
|
|
{
|
|
if (inPercent)
|
|
{
|
|
ptrdiff_t count = CharNextW(p) - p;
|
|
while (count-- && (pEnv - environString) < MAX_PATH)
|
|
*pEnv++ = *p++;
|
|
}
|
|
else
|
|
{
|
|
wchar_t *newP = CharNextW(p);
|
|
*p = 0;
|
|
StringCchCatW(dest, stringSize, string);
|
|
string = p = newP;
|
|
inBrace = 1;
|
|
}
|
|
}
|
|
else if (*p == L'}')
|
|
{
|
|
if (inPercent)
|
|
{
|
|
ptrdiff_t count = CharNextW(p) - p;
|
|
while (count-- && (pEnv - environString) < MAX_PATH)
|
|
*pEnv++ = *p++;
|
|
}
|
|
*pEnv = 0;
|
|
//SHGetSpecialFolderPath(NULL, helper, atoi(environString), FALSE);
|
|
#if defined(UNICODE) || defined(_UNICODE)
|
|
SHGetSpecialFolderPath(NULL, helper, StrToInt(environString), FALSE);
|
|
//UtilGetSpecialFolderPath(NULL, helper, _wtoi(environString));
|
|
#else
|
|
SHGetSpecialFolderPathW(NULL, helper, StrToIntW(environString), FALSE);
|
|
//UtilGetSpecialFolderPath(NULL, helper, _aoi(environString));
|
|
#endif
|
|
StringCchCatW(dest, stringSize, helper);
|
|
pEnv = environString;
|
|
string = CharNextW(p);
|
|
*p = 0;
|
|
p = string;
|
|
inBrace = 0;
|
|
}
|
|
else
|
|
{
|
|
if (inPercent || inBrace)
|
|
{
|
|
ptrdiff_t count = CharNextW(p) - p;
|
|
while (count-- && (pEnv - environString) < MAX_PATH)
|
|
*pEnv++ = *p++;
|
|
}
|
|
else
|
|
p = CharNextW(p);
|
|
}
|
|
}
|
|
|
|
StringCchCatW(dest, stringSize, string);
|
|
}
|
|
|
|
#ifndef NO_INPLACE_RESOLVE
|
|
#include <malloc.h>
|
|
// call this function if you want to modify in-place
|
|
#include "main.h"
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" void ResolveEnvironmentVariables(wchar_t *string, size_t stringSize) //
|
|
#else
|
|
void ResolveEnvironmentVariables(wchar_t *string, size_t stringSize) //
|
|
#endif
|
|
{
|
|
wchar_t *dest = (wchar_t *) alloca(stringSize * sizeof(dest[0]));
|
|
ResolveEnvironmentVariables2(string, dest, stringSize);
|
|
StringCchCopyW(string, stringSize, dest);
|
|
}
|
|
#endif |