223 lines
8.3 KiB
C
223 lines
8.3 KiB
C
|
#include "CommandLine.h"
|
||
|
|
#include <malloc.h>
|
||
|
|
|
||
|
|
BOOL WINAPI MakeSureDirectoryPathExistsW(LPCWSTR DirPath);
|
||
|
|
|
||
|
|
DWORD WINAPI CheckCommandLineOptions(INT ArgC, LPWSTR* ArgVW) {
|
||
|
|
DWORD dwReturnFlags = 0;
|
||
|
|
INT i;
|
||
|
|
WCHAR* cp;
|
||
|
|
WCHAR wszInstallPath[MAX_PATH+1];
|
||
|
|
|
||
|
|
for (i = 1; i < ArgC && wcschr(L"/-",ArgVW[i][0]) != NULL; ++i) {
|
||
|
|
|
||
|
|
for (cp = &ArgVW[i][1]; *cp != L'\0'; ++cp) {
|
||
|
|
switch (towupper(*cp)) {
|
||
|
|
case L'U': {
|
||
|
|
HKEY hKey;
|
||
|
|
|
||
|
|
LONG lStatus = 0;
|
||
|
|
LONG lCreatedOrOpened = 0;
|
||
|
|
SET_FLAG(dwReturnFlags, FLAG_UNATTENDED_INSTALL);
|
||
|
|
|
||
|
|
// next param isn't a flag (or NULL), so it *must* be the path to install to
|
||
|
|
if ( (i+1 < ArgC) && wcschr(L"/-",ArgVW[i+1][0]) == NULL ) {
|
||
|
|
i++; // account for the parameter removed
|
||
|
|
|
||
|
|
SET_FLAG(dwReturnFlags, FLAG_UNATTENDED_PATH_PROVIDED);
|
||
|
|
StringCchCopyW(wszInstallPath, MAX_PATH+1, ArgVW[i]);
|
||
|
|
|
||
|
|
// make sure path ends in '\'
|
||
|
|
if (wszInstallPath[wcslen(wszInstallPath)]!=L'\\')
|
||
|
|
StringCchCatW(wszInstallPath,MAX_PATH+1,L"\\");
|
||
|
|
|
||
|
|
// make sure the directory exists!
|
||
|
|
if (! MakeSureDirectoryPathExistsW(wszInstallPath) ) {
|
||
|
|
SET_FLAG(dwReturnFlags, FLAG_FATAL_ERROR);
|
||
|
|
} else {
|
||
|
|
// Either create the regkey (if it doesn't exist) or open it (if it
|
||
|
|
// does exist). lCreatedOrOpened can be tested against
|
||
|
|
// REG_CREATED_NEW_KEY or REG_OPENED_EXISTING_KEY to determine which
|
||
|
|
// occurred.
|
||
|
|
lStatus = RegCreateKeyExW(SYMBOLS_REGKEY_ROOT,
|
||
|
|
SYMBOLS_REGKEY_PATH,
|
||
|
|
0,
|
||
|
|
NULL,
|
||
|
|
REG_OPTION_NON_VOLATILE,
|
||
|
|
KEY_ALL_ACCESS,
|
||
|
|
NULL,
|
||
|
|
&hKey,
|
||
|
|
&lCreatedOrOpened);
|
||
|
|
|
||
|
|
if (lStatus != ERROR_SUCCESS) {
|
||
|
|
SET_FLAG(dwReturnFlags, FLAG_FATAL_ERROR);
|
||
|
|
} else {
|
||
|
|
// Write the value of the path to SYMBOLS_REGKEY
|
||
|
|
lStatus = RegSetValueExW( hKey, SYMBOLS_REGKEY, 0, REG_SZ, (BYTE*)wszInstallPath, ((wcslen(wszInstallPath) + 1) * sizeof(WCHAR)));
|
||
|
|
if (lStatus != ERROR_SUCCESS) {
|
||
|
|
SET_FLAG(dwReturnFlags, FLAG_FATAL_ERROR);
|
||
|
|
}
|
||
|
|
// close the regkey
|
||
|
|
lStatus = RegCloseKey( hKey );
|
||
|
|
if (lStatus != ERROR_SUCCESS) {
|
||
|
|
SET_FLAG(dwReturnFlags, FLAG_ERROR);
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
} // else ...
|
||
|
|
// couldn't set the path requests, so use what's
|
||
|
|
// already in the registry.
|
||
|
|
|
||
|
|
} // else ...
|
||
|
|
// no path provided, so don't set anything- setupapi will
|
||
|
|
// nicely use the existing key or default to the value
|
||
|
|
// specified in the INF
|
||
|
|
// StringCchCopyW(wszInstallPath, MAX_PATH+1, DEFAULT_INSTALL_PATH);
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
|
||
|
|
case L'Q':
|
||
|
|
SET_FLAG(dwReturnFlags, FLAG_TOTALLY_QUIET);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case L'?': // explicit fall through
|
||
|
|
case L'H': // explicit fall through
|
||
|
|
default:
|
||
|
|
SET_FLAG(dwReturnFlags, FLAG_USAGE);
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if ( IS_FLAG_SET(dwReturnFlags, FLAG_USAGE) ) {
|
||
|
|
WCHAR UsageBuffer[1024];
|
||
|
|
|
||
|
|
StringCchPrintfW(UsageBuffer,
|
||
|
|
sizeof(UsageBuffer)/sizeof(WCHAR),
|
||
|
|
L"Usage: %s [ /u [<path>] [/q] ]\n\n"
|
||
|
|
L"/u [<path>] \n"
|
||
|
|
L" Unattended install. If <path> is specified install\n"
|
||
|
|
L" symbols to <path>. If no path is specified, symbols\n"
|
||
|
|
L" are installed to the default location.\n"
|
||
|
|
L" NOTE: USING UNATTENDED INSTALL MEANS YOU\n"
|
||
|
|
L" HAVE READ AND AGREED TO THE END USER LICENSE\n"
|
||
|
|
L" AGREEMENT FOR THIS PRODUCT.\n"
|
||
|
|
L"/q\n"
|
||
|
|
L" Valid only when using unattended install. Prevents\n"
|
||
|
|
L" error messages from being display if unattended\n"
|
||
|
|
L" install fails.\n"
|
||
|
|
L"/?\n"
|
||
|
|
L" Show this dialog box.\n\n"
|
||
|
|
L"If no options are specified, the interactive installation\n"
|
||
|
|
L" is started.",
|
||
|
|
ArgVW[0]);
|
||
|
|
|
||
|
|
|
||
|
|
MessageBoxW( NULL,
|
||
|
|
UsageBuffer,
|
||
|
|
L"Microsoft Windows Symbols",
|
||
|
|
0 );
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
return(dwReturnFlags);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Modified from MakeSureDirectoryPathExists from dbghelp.h
|
||
|
|
// The same caveats apply. (see MSDN)
|
||
|
|
BOOL WINAPI MakeSureDirectoryPathExistsW(LPCWSTR DirPath) {
|
||
|
|
LPWSTR p, DirCopy;
|
||
|
|
DWORD dw;
|
||
|
|
|
||
|
|
// Make a copy of the string for editing.
|
||
|
|
|
||
|
|
__try {
|
||
|
|
DirCopy = (LPWSTR)malloc((wcslen(DirPath) + 1) * sizeof(WCHAR));
|
||
|
|
|
||
|
|
if (!DirCopy) {
|
||
|
|
return FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
StringCchCopyW(DirCopy, wcslen(DirPath)+1, DirPath);
|
||
|
|
|
||
|
|
p = DirCopy;
|
||
|
|
|
||
|
|
// If the second character in the path is "\", then this is a UNC
|
||
|
|
// path, and we should skip forward until we reach the 2nd \ in the path.
|
||
|
|
|
||
|
|
if ((*p == L'\\') && (*(p+1) == L'\\')) {
|
||
|
|
p++; // Skip over the first \ in the name.
|
||
|
|
p++; // Skip over the second \ in the name.
|
||
|
|
|
||
|
|
// Skip until we hit the first "\" (\\Server\).
|
||
|
|
|
||
|
|
while (*p && *p != L'\\') {
|
||
|
|
p = CharNextW(p);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Advance over it.
|
||
|
|
|
||
|
|
if (*p) {
|
||
|
|
p++;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Skip until we hit the second "\" (\\Server\Share\).
|
||
|
|
|
||
|
|
while (*p && *p != L'\\') {
|
||
|
|
p = CharNextW(p);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Advance over it also.
|
||
|
|
|
||
|
|
if (*p) {
|
||
|
|
p++;
|
||
|
|
}
|
||
|
|
|
||
|
|
} else
|
||
|
|
// Not a UNC. See if it's <drive>:
|
||
|
|
if (*(p+1) == L':' ) {
|
||
|
|
|
||
|
|
p++;
|
||
|
|
p++;
|
||
|
|
|
||
|
|
// If it exists, skip over the root specifier
|
||
|
|
|
||
|
|
if (*p && (*p == L'\\')) {
|
||
|
|
p++;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
while( *p ) {
|
||
|
|
if ( *p == '\\' ) {
|
||
|
|
*p = '\0';
|
||
|
|
dw = GetFileAttributesW(DirCopy);
|
||
|
|
// Nothing exists with this name. Try to make the directory name and error if unable to.
|
||
|
|
if ( dw == 0xffffffff ) {
|
||
|
|
if ( !CreateDirectoryW(DirCopy,NULL) ) {
|
||
|
|
if( GetLastError() != ERROR_ALREADY_EXISTS ) {
|
||
|
|
free(DirCopy);
|
||
|
|
return FALSE;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
if ( (dw & FILE_ATTRIBUTE_DIRECTORY) != FILE_ATTRIBUTE_DIRECTORY ) {
|
||
|
|
// Something exists with this name, but it's not a directory... Error
|
||
|
|
free(DirCopy);
|
||
|
|
return FALSE;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
*p = L'\\';
|
||
|
|
}
|
||
|
|
p = CharNextW(p);
|
||
|
|
}
|
||
|
|
} __except (EXCEPTION_EXECUTE_HANDLER) {
|
||
|
|
SetLastError( GetExceptionCode() );
|
||
|
|
free(DirCopy);
|
||
|
|
return(FALSE);
|
||
|
|
}
|
||
|
|
|
||
|
|
free(DirCopy);
|
||
|
|
return TRUE;
|
||
|
|
}
|