|
|
#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;}
|
