Files
i2p.itoopie/installer/lib/launch4j/head_src/head.c
2005-09-16 04:34:59 +00:00

359 lines
9.1 KiB
C

/*
launch4j :: Cross-platform Java application wrapper for creating Windows native executables
Copyright (C) 2004-2005 Grzegorz Kowal
Compiled with Mingw port of GCC, Bloodshed Dev-C++ IDE (http://www.bloodshed.net/devcpp.html)
This library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public
License as published by the Free Software Foundation; either
version 2.1 of the License, or (at your option) any later version.
This library is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public
License along with this library; if not, write to the Free Software
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*/
#include "resource.h"
#include "head.h"
BOOL console = FALSE;
int foundJava = NO_JAVA_FOUND;
struct _stat statBuf;
PROCESS_INFORMATION pi;
char errTitle[STR] = "launch4j";
char javaMinVer[STR] = "";
char javaMaxVer[STR] = "";
char foundJavaVer[STR] = "";
char workingDir[_MAX_PATH] = "";
char cmd[_MAX_PATH] = "";
char args[BIG_STR * 2] = "";
void setConsoleFlag() {
console = TRUE;
}
void msgBox(const char* text) {
if (console) {
printf("%s: %s\n", errTitle, text);
} else {
MessageBox(NULL, text, errTitle, MB_OK);
}
}
void showJavaWebPage() {
ShellExecute(NULL, "open", "http://java.com", NULL, NULL, SW_SHOWNORMAL);
}
BOOL loadString(HMODULE hLibrary, int resID, char* buffer) {
HRSRC hResource;
HGLOBAL hResourceLoaded;
LPBYTE lpBuffer;
hResource = FindResourceEx(hLibrary, RT_RCDATA, MAKEINTRESOURCE(resID),
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT));
if (NULL != hResource) {
hResourceLoaded = LoadResource(hLibrary, hResource);
if (NULL != hResourceLoaded) {
lpBuffer = (LPBYTE) LockResource(hResourceLoaded);
if (NULL != lpBuffer) {
int x = 0;
do {
buffer[x] = (char) lpBuffer[x];
} while (buffer[x++] != 0);
return TRUE;
}
}
}
return FALSE;
}
BOOL loadBoolString(HMODULE hLibrary, int resID) {
char boolStr[10] = "";
loadString(hLibrary, resID, boolStr);
return strcmp(boolStr, TRUE_STR) == 0;
}
void regSearch(HKEY hKey, char* keyName, int searchType) {
DWORD x = 0;
unsigned long size = BIG_STR;
FILETIME time;
char buffer[BIG_STR] = "";
while (RegEnumKeyEx(
hKey, // handle to key to enumerate
x++, // index of subkey to enumerate
buffer, // address of buffer for subkey name
&size, // address for size of subkey buffer
NULL, // reserved
NULL, // address of buffer for class string
NULL, // address for size of class buffer
&time) == ERROR_SUCCESS) {
if (strcmp(buffer, javaMinVer) >= 0
&& (javaMaxVer[0] == 0 || strcmp(buffer, javaMaxVer) <= 0)
&& strcmp(buffer, foundJavaVer) > 0) {
strcpy(foundJavaVer, buffer);
foundJava = searchType;
}
size = BIG_STR;
}
}
BOOL findJavaHome(char* path) {
HKEY hKey;
char jre[] = "SOFTWARE\\JavaSoft\\Java Runtime Environment";
char sdk[] = "SOFTWARE\\JavaSoft\\Java Development Kit";
if (RegOpenKeyEx(HKEY_LOCAL_MACHINE,
TEXT(jre),
0,
KEY_QUERY_VALUE | KEY_ENUMERATE_SUB_KEYS,
&hKey) == ERROR_SUCCESS) {
regSearch(hKey, jre, FOUND_JRE);
RegCloseKey(hKey);
}
if (RegOpenKeyEx(HKEY_LOCAL_MACHINE,
TEXT(sdk),
0,
KEY_QUERY_VALUE | KEY_ENUMERATE_SUB_KEYS,
&hKey) == ERROR_SUCCESS) {
regSearch(hKey, sdk, FOUND_SDK);
RegCloseKey(hKey);
}
if (foundJava != NO_JAVA_FOUND) {
char keyBuffer[BIG_STR];
unsigned long datatype;
unsigned long bufferlength = BIG_STR;
if (foundJava == FOUND_JRE) {
strcpy(keyBuffer, jre);
} else {
strcpy(keyBuffer, sdk);
}
strcat(keyBuffer, "\\");
strcat(keyBuffer, foundJavaVer);
if (RegOpenKeyEx(HKEY_LOCAL_MACHINE,
TEXT(keyBuffer),
0,
KEY_QUERY_VALUE,
&hKey) == ERROR_SUCCESS) {
unsigned char buffer[BIG_STR];
if (RegQueryValueEx(hKey, "JavaHome", NULL, &datatype, buffer, &bufferlength)
== ERROR_SUCCESS) {
int i = 0;
do {
path[i] = buffer[i];
} while (path[i++] != 0);
if (foundJava == FOUND_SDK) {
strcat(path, "\\jre");
}
RegCloseKey(hKey);
return TRUE;
}
RegCloseKey(hKey);
}
}
return FALSE;
}
/*
* extract the executable name, returns path length.
*/
int getExePath(char* exePath) {
HMODULE hModule = GetModuleHandle(NULL);
if (hModule == 0
|| GetModuleFileName(hModule, exePath, _MAX_PATH) == 0) {
return -1;
}
return strrchr(exePath, '\\') - exePath;
}
void appendJavaw(char* jrePath) {
if (console) {
strcat(jrePath, "\\bin\\java.exe");
} else {
strcat(jrePath, "\\bin\\javaw.exe");
}
}
void appendLauncher(BOOL setProcName, char* exePath, int pathLen, char* cmd) {
if (setProcName) {
char tmpspec[_MAX_PATH] = "";
char tmpfile[_MAX_PATH] = "";
strcpy(tmpspec, cmd);
strcat(tmpspec, LAUNCH4J_TMP_DIR);
tmpspec[strlen(tmpspec) - 1] = 0;
if (_stat(tmpspec, &statBuf) == 0) {
// remove temp launchers
struct _finddata_t c_file;
long hFile;
strcat(tmpspec, "\\*.exe");
strcpy(tmpfile, cmd);
strcat(tmpfile, LAUNCH4J_TMP_DIR);
char* filename = tmpfile + strlen(tmpfile);
if ((hFile = _findfirst(tmpspec, &c_file)) != -1L) {
do {
strcpy(filename, c_file.name);
_unlink(tmpfile);
} while (_findnext(hFile, &c_file) == 0);
}
_findclose(hFile);
} else {
if (_mkdir(tmpspec) != 0) {
appendJavaw(cmd);
return;
}
}
char javaw[_MAX_PATH] = "";
strcpy(javaw, cmd);
appendJavaw(javaw);
strcpy(tmpfile, cmd);
strcat(tmpfile, LAUNCH4J_TMP_DIR);
char* exeFilePart = exePath + pathLen + 1;
strcat(tmpfile, exeFilePart);
if (CopyFile(javaw, tmpfile, FALSE)) {
strcpy(cmd, tmpfile);
return;
} else {
long fs = statBuf.st_size;
if (_stat(tmpfile, &statBuf) == 0 && fs == statBuf.st_size) {
strcpy(cmd, tmpfile);
return;
}
}
}
appendJavaw(cmd);
}
BOOL isJrePathOk(char* path) {
if (!*path) {
return FALSE;
}
char javaw[_MAX_PATH];
strcpy(javaw, path);
appendJavaw(javaw);
return _stat(javaw, &statBuf) == 0;
}
BOOL prepare(HMODULE hLibrary, char *lpCmdLine) {
// open executable
char exePath[_MAX_PATH] = "";
int pathLen = getExePath(exePath);
if (pathLen == -1) {
msgBox("Cannot determinate exe file name.");
return FALSE;
}
hLibrary = LoadLibrary(exePath);
if (hLibrary == NULL) {
char msg[BIG_STR] = "Cannot find file: ";
strcat(msg, exePath);
msgBox(msg);
return FALSE;
}
// error message box title
loadString(hLibrary, ERR_TITLE, errTitle);
// working dir
char tmp_path[_MAX_PATH] = "";
if (loadString(hLibrary, CHDIR, tmp_path)) {
strncpy(workingDir, exePath, pathLen);
strcat(workingDir, "/");
strcat(workingDir, tmp_path);
_chdir(workingDir);
}
// custom process name
const BOOL setProcName = loadBoolString(hLibrary, SET_PROC_NAME);
// use bundled jre or find java
if (loadString(hLibrary, JRE_PATH, tmp_path)) {
strncpy(cmd, exePath, pathLen);
strcat(cmd, "/");
strcat(cmd, tmp_path);
}
if (!isJrePathOk(cmd)) {
if (!loadString(hLibrary, JAVA_MIN_VER, javaMinVer)) {
msgBox("Cannot find bundled JRE or javaw.exe is missing.");
return FALSE;
}
loadString(hLibrary, JAVA_MAX_VER, javaMaxVer);
if (!findJavaHome(cmd)) {
char txt[BIG_STR] = "Cannot find Java ";
strcat(txt, javaMinVer);
if (*javaMaxVer) {
strcat(txt, " - ");
strcat(txt, javaMaxVer);
}
msgBox(txt);
showJavaWebPage();
return FALSE;
}
if (!isJrePathOk(cmd)) {
msgBox("Java found, but javaw.exe seems to be missing.");
return FALSE;
}
}
appendLauncher(setProcName, exePath, pathLen, cmd);
char jarArgs[BIG_STR] = "";
loadString(hLibrary, JAR_ARGS, jarArgs);
loadString(hLibrary, JVM_ARGS, args);
if (*args) {
strcat(args, " ");
}
strcat(args, "-jar \"");
strcat(args, exePath);
strcat(args, "\"");
if (*jarArgs) {
strcat(args, " ");
strcat(args, jarArgs);
}
if (*lpCmdLine) {
strcat(args, " ");
strcat(args, lpCmdLine);
}
// msgBox(cmd);
// msgBox(args);
// msgBox(workingDir);
return TRUE;
}
void closeHandles() {
CloseHandle(pi.hThread);
CloseHandle(pi.hProcess);
}
DWORD execute(BOOL wait) {
STARTUPINFO si;
memset(&pi, 0, sizeof(pi));
memset(&si, 0, sizeof(si));
si.cb = sizeof(si);
DWORD dwExitCode = -1;
char cmdline[_MAX_PATH + BIG_STR] = "\"";
strcat(cmdline, cmd);
strcat(cmdline, "\" ");
strcat(cmdline, args);
if (CreateProcess(NULL, cmdline, NULL, NULL,
FALSE, 0, NULL, NULL, &si, &pi)) {
if (wait) {
WaitForSingleObject(pi.hProcess, INFINITE);
GetExitCodeProcess(pi.hProcess, &dwExitCode);
closeHandles();
} else {
dwExitCode = 0;
}
}
return dwExitCode;
}