/*
|
* Copyright (C) 2014 The Android Open Source Project
|
*
|
* 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.
|
*/
|
|
#include "stdafx.h"
|
#include "utils.h"
|
|
// Set to true to get some extra debug information
|
bool gIsDebug = false;
|
// Set to true to output errors to stderr (for a Console app)
|
// or to false to output using msg box (for a Windows UI app)
|
bool gIsConsole = false;
|
|
// Application name used in error dialog. Defined using initUtils()
|
static CString gAppName("Find Java 2");
|
|
// Called by the application to initialize the app name used in error dialog boxes.
|
void initUtils(const TCHAR *appName) {
|
if (appName != NULL) {
|
gAppName = CString(appName);
|
return;
|
}
|
|
// Try to get the VERSIONINFO.FileDescription and use as app name
|
// Errors are ignored, in which case the default app name is used.
|
|
// First get the module (aka app instance) filename.
|
TCHAR moduleName[MAX_PATH + 1];
|
DWORD sz = ::GetModuleFileName(NULL /*AfxGetInstanceHandle()*/, moduleName, MAX_PATH);
|
if (sz == 0) {
|
// GetModuleFileName failed. Do nothing.
|
return;
|
}
|
moduleName[sz] = '\0'; // make sure string is properly terminated.
|
|
// Get the size of the FileVersionInfo buffer
|
DWORD obsoleteHandle; // see http://blogs.msdn.com/b/oldnewthing/archive/2007/07/31/4138786.aspx
|
DWORD fviSize = ::GetFileVersionInfoSize(moduleName, &obsoleteHandle);
|
if (fviSize == 0) {
|
return; // do nothing on error
|
}
|
|
char *fviBuffer = new char[fviSize];
|
if (::GetFileVersionInfo(moduleName, 0, fviSize, fviBuffer) != 0) {
|
VOID *vBuffer;
|
UINT vLen;
|
|
struct LANGUAGE_CODEPAGE {
|
WORD mLanguage;
|
WORD mCodePage;
|
} *lgcpBuffer;
|
|
UINT lgcpSize;
|
|
// Read the list of languages and code pages (c.f. MSDN for VerQueryValue)
|
if (::VerQueryValue(fviBuffer, _T("\\VarFileInfo\\Translation"), (LPVOID*)&lgcpBuffer, &lgcpSize) != 0 &&
|
lgcpSize >= sizeof(LANGUAGE_CODEPAGE)) {
|
// Use the first available language and code page
|
CString subBlock;
|
subBlock.Format(_T("\\StringFileInfo\\%04x%04x\\FileDescription"),
|
lgcpBuffer[0].mLanguage,
|
lgcpBuffer[0].mCodePage);
|
if (::VerQueryValue(fviBuffer, subBlock, &vBuffer, &vLen) != 0) {
|
gAppName.SetString((LPCTSTR)vBuffer, vLen);
|
}
|
}
|
}
|
delete[] fviBuffer;
|
}
|
|
CString getAppName() {
|
return gAppName;
|
}
|
|
|
// Displays a message in an ok+info dialog box.
|
void msgBox(const TCHAR* text, ...) {
|
CString formatted;
|
va_list ap;
|
va_start(ap, text);
|
formatted.FormatV(text, ap);
|
va_end(ap);
|
|
// TODO global CString to get app name
|
MessageBox(NULL, formatted, gAppName, MB_OK | MB_ICONINFORMATION);
|
}
|
|
// Sets the string to the message matching Win32 GetLastError.
|
// If message is non-null, it is prepended to the last error string.
|
CString getLastWin32Error(const TCHAR* message) {
|
DWORD err = GetLastError();
|
CString result;
|
LPTSTR errStr;
|
if (FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | /* dwFlags */
|
FORMAT_MESSAGE_FROM_SYSTEM,
|
NULL, /* lpSource */
|
err, /* dwMessageId */
|
0, /* dwLanguageId */
|
(LPTSTR) &errStr, /* out lpBuffer */
|
0, /* nSize */
|
NULL) != 0) { /* va_list args */
|
if (message == NULL) {
|
result.Format(_T("[%d] %s"), err, errStr);
|
} else {
|
result.Format(_T("%s[%d] %s"), message, err, errStr);
|
}
|
LocalFree(errStr);
|
}
|
return result;
|
}
|
|
// Displays GetLastError prefixed with a description in an error dialog box
|
void displayLastError(const TCHAR *description, ...) {
|
CString formatted;
|
va_list ap;
|
va_start(ap, description);
|
formatted.FormatV(description, ap);
|
va_end(ap);
|
|
CString error = getLastWin32Error(NULL);
|
formatted.Append(_T("\r\n"));
|
formatted.Append(error);
|
|
if (gIsConsole) {
|
_ftprintf(stderr, _T("%s\n"), (LPCTSTR) formatted);
|
} else {
|
CString name(gAppName);
|
name.Append(_T(" - Error"));
|
MessageBox(NULL, formatted, name, MB_OK | MB_ICONERROR);
|
}
|
}
|
|
// Executes the command line. Does not wait for the program to finish.
|
// The return code is from CreateProcess (0 means failure), not the running app.
|
int execNoWait(const TCHAR *app, const TCHAR *params, const TCHAR *workDir) {
|
STARTUPINFO startup;
|
PROCESS_INFORMATION pinfo;
|
|
ZeroMemory(&pinfo, sizeof(pinfo));
|
|
ZeroMemory(&startup, sizeof(startup));
|
startup.cb = sizeof(startup);
|
startup.dwFlags = STARTF_USESHOWWINDOW;
|
startup.wShowWindow = SW_SHOWDEFAULT;
|
|
int ret = CreateProcess(
|
app, /* program path */
|
(TCHAR *)params, /* command-line */
|
NULL, /* process handle is not inheritable */
|
NULL, /* thread handle is not inheritable */
|
TRUE, /* yes, inherit some handles */
|
0, /* create flags */
|
NULL, /* use parent's environment block */
|
workDir, /* use parent's starting directory */
|
&startup, /* startup info, i.e. std handles */
|
&pinfo);
|
|
if (ret) {
|
CloseHandle(pinfo.hProcess);
|
CloseHandle(pinfo.hThread);
|
}
|
|
return ret;
|
}
|
|
// Executes command, waits for completion and returns exit code.
|
// As indicated in MSDN for CreateProcess, callers should double-quote the program name
|
// e.g. cmd="\"c:\program files\myapp.exe\" arg1 arg2";
|
int execWait(const TCHAR *cmd) {
|
STARTUPINFO startup;
|
PROCESS_INFORMATION pinfo;
|
|
ZeroMemory(&pinfo, sizeof(pinfo));
|
|
ZeroMemory(&startup, sizeof(startup));
|
startup.cb = sizeof(startup);
|
startup.dwFlags = STARTF_USESHOWWINDOW;
|
startup.wShowWindow = SW_HIDE | SW_MINIMIZE;
|
|
int ret = CreateProcess(
|
NULL, /* program path */
|
(LPTSTR)cmd, /* command-line */
|
NULL, /* process handle is not inheritable */
|
NULL, /* thread handle is not inheritable */
|
TRUE, /* yes, inherit some handles */
|
CREATE_NO_WINDOW, /* we don't want a console */
|
NULL, /* use parent's environment block */
|
NULL, /* use parent's starting directory */
|
&startup, /* startup info, i.e. std handles */
|
&pinfo);
|
|
int result = -1;
|
if (ret) {
|
WaitForSingleObject(pinfo.hProcess, INFINITE);
|
|
DWORD exitCode;
|
if (GetExitCodeProcess(pinfo.hProcess, &exitCode)) {
|
// this should not return STILL_ACTIVE (259)
|
result = exitCode;
|
}
|
CloseHandle(pinfo.hProcess);
|
CloseHandle(pinfo.hThread);
|
}
|
|
return result;
|
}
|
|
bool getModuleDir(CPath *outDir) {
|
TCHAR programDir[MAX_PATH];
|
int ret = GetModuleFileName(NULL, programDir, sizeof(programDir) / sizeof(programDir[0]));
|
if (ret != 0) {
|
CPath dir(programDir);
|
dir.RemoveFileSpec();
|
*outDir = dir;
|
return true;
|
}
|
return false;
|
}
|
|
// Disables the FS redirection done by WOW64.
|
// Because this runs as a 32-bit app, Windows automagically remaps some
|
// folder under the hood (e.g. "Programs Files(x86)" is mapped as "Program Files").
|
// This prevents the app from correctly searching for java.exe in these folders.
|
// The registry is also remapped. This method disables this redirection.
|
// Caller should restore the redirection later by using revertWow64FsRedirection().
|
PVOID disableWow64FsRedirection() {
|
|
// The call we want to make is the following:
|
// PVOID oldWow64Value;
|
// Wow64DisableWow64FsRedirection(&oldWow64Value);
|
// However that method may not exist (e.g. on XP non-64 systems) so
|
// we must not call it directly.
|
|
PVOID oldWow64Value = 0;
|
|
HMODULE hmod = LoadLibrary(_T("kernel32.dll"));
|
if (hmod != NULL) {
|
FARPROC proc = GetProcAddress(hmod, "Wow64DisableWow64FsRedirection");
|
if (proc != NULL) {
|
typedef BOOL(WINAPI *disableWow64FuncType)(PVOID *);
|
disableWow64FuncType funcPtr = (disableWow64FuncType)proc;
|
funcPtr(&oldWow64Value);
|
}
|
|
FreeLibrary(hmod);
|
}
|
|
return oldWow64Value;
|
}
|
|
// Reverts the redirection disabled in disableWow64FsRedirection.
|
void revertWow64FsRedirection(PVOID oldWow64Value) {
|
|
// The call we want to make is the following:
|
// Wow64RevertWow64FsRedirection(oldWow64Value);
|
// However that method may not exist (e.g. on XP non-64 systems) so
|
// we must not call it directly.
|
|
HMODULE hmod = LoadLibrary(_T("kernel32.dll"));
|
if (hmod != NULL) {
|
FARPROC proc = GetProcAddress(hmod, "Wow64RevertWow64FsRedirection");
|
if (proc != NULL) {
|
typedef BOOL(WINAPI *revertWow64FuncType)(PVOID);
|
revertWow64FuncType funcPtr = (revertWow64FuncType)proc;
|
funcPtr(oldWow64Value);
|
}
|
|
FreeLibrary(hmod);
|
}
|
}
|
