/* * %CopyrightBegin% * * Copyright Ericsson AB 2003-2009. All Rights Reserved. * * The contents of this file are subject to the Erlang Public License, * Version 1.1, (the "License"); you may not use this file except in * compliance with the License. You should have received a copy of the * Erlang Public License along with this software. If not, it can be * retrieved online at http://www.erlang.org/. * * Software distributed under the License is distributed on an "AS IS" * basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See * the License for the specific language governing rights and limitations * under the License. * * %CopyrightEnd% */ #pragma comment(linker,"/manifestdependency:\"type='win32' "\ "name='Microsoft.Windows.Common-Controls' "\ "version='6.0.0.0' processorArchitecture='*' "\ "publicKeyToken='6595b64144ccf1df' language='*'\"") #include #include #include #include "init_file.h" typedef int ErlexecFunction(int, char **, HANDLE, int); #define INI_FILENAME L"erl.ini" #define INI_SECTION "erlang" #define ERLEXEC_BASENAME L"erlexec.dll" static void get_parameters(void); static void error(char* format, ...); static wchar_t *erlexec_name; static wchar_t *erlexec_dir; #ifdef WIN32_WERL #define WERL 1 int WINAPI wWinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, PWSTR szCmdLine, int iCmdShow) { int argc = __argc; wchar_t **argv = __wargv; #else #define WERL 0 int wmain(int argc, wchar_t **argv) { #endif HANDLE erlexec_handle; /* Instance */ ErlexecFunction *win_erlexec; wchar_t *path = malloc(100*sizeof(wchar_t)); wchar_t *npath; int pathlen; char ** utf8argv; int i, len; get_parameters(); if ((pathlen = GetEnvironmentVariableW(L"PATH",path,100)) == 0) { error("No PATH variable (!)"); } else if (pathlen > 100) { path = realloc(path,pathlen*sizeof(wchar_t)); GetEnvironmentVariableW(L"PATH",path,pathlen); } pathlen = (wcslen(path) + wcslen(erlexec_dir) + 2); npath = (wchar_t *) malloc(pathlen*sizeof(wchar_t)); swprintf(npath,pathlen,L"%s;%s",erlexec_dir,path); SetEnvironmentVariableW(L"PATH",npath); if ((erlexec_handle = LoadLibraryW(erlexec_name)) == NULL) { error("Could not load module %S.",erlexec_name); } if ((win_erlexec = (ErlexecFunction *) GetProcAddress(erlexec_handle,"win_erlexec")) == NULL) { error("Could not find entry point \"win_erlexec\" in %S.", erlexec_name); } /* Convert argv to utf8 */ utf8argv = malloc((argc+1) * sizeof(char*)); for (i=0; i= res && *p != L'\\'; --p) ; *(p+1) =L'\0'; wcscat(res,new_base); return res; } static char *do_lookup_in_section(InitSection *inis, char *name, char *section, wchar_t *filename) { char *p = lookup_init_entry(inis, name); if (p == NULL) { error("Could not find key %s in section %s of file %S", name,section,filename); } return p; } static void copy_latest_vsn(wchar_t *latest_vsn, wchar_t *next_vsn) { /* Copy */ wchar_t *lp; wchar_t *np; /* Find vsn */ for (lp = next_vsn+wcslen(next_vsn)-1 ;lp >= next_vsn && *lp != L'\\'; --lp) ; /* lp =+ length("erts-"); */ for (np = next_vsn+wcslen(next_vsn)-1 ;np >= next_vsn && *np != L'\\'; --np) ; /* np =+ length("erts-"); */ for (; lp && np; ++lp, ++np) { if (*lp == *np) { continue; } if (*np == L'.' || *np == L'\0' || *np <= *lp) { /* */ return; } if (*lp == L'.' || *lp == L'\0') { wcscpy(latest_vsn, next_vsn); return; } } return; } static wchar_t *find_erlexec_dir2(wchar_t *install_dir) { /* List install dir and look for latest erts-vsn */ HANDLE dir_handle; /* Handle to directory. */ wchar_t wildcard[MAX_PATH]; /* Wildcard to search for. */ WIN32_FIND_DATAW find_data; /* Data found by FindFirstFile() or FindNext(). */ wchar_t latest_vsn[MAX_PATH]; /* Setup wildcard */ int length = wcslen(install_dir); wchar_t *p; if (length+3 >= MAX_PATH) { error("Cannot find erlexec.exe"); } wcscpy(wildcard, install_dir); p = wildcard+length-1; if (*p != L'/' && *p != L'\\') *++p = L'\\'; wcscpy(++p, L"erts-*"); /* Find first dir */ dir_handle = FindFirstFileW(wildcard, &find_data); if (dir_handle == INVALID_HANDLE_VALUE) { /* No erts-vsn found*/ return NULL; } wcscpy(latest_vsn, find_data.cFileName); /* Find the rest */ while(FindNextFileW(dir_handle, &find_data)) { copy_latest_vsn(latest_vsn, find_data.cFileName); } FindClose(dir_handle); p = (wchar_t *) malloc((wcslen(install_dir)+1+wcslen(latest_vsn)+4+1)*sizeof(wchar_t)); wcscpy(p,install_dir); wcscat(p,L"\\"); wcscat(p,latest_vsn); wcscat(p,L"\\bin"); return p; } static wchar_t *find_erlexec_dir(wchar_t *erlpath) { /* Assume that the path to erl is absolute and * that it is not a symbolic link*/ wchar_t *dir =_wcsdup(erlpath); wchar_t *p; wchar_t *p2; /* Chop of base name*/ for (p = dir+wcslen(dir)-1 ;p >= dir && *p != L'\\'; --p) ; *p =L'\0'; p--; /* Check if dir path is like ...\install_dir\erts-vsn\bin */ for (;p >= dir && *p != L'\\'; --p) ; p--; for (p2 = p;p2 >= dir && *p2 != '\\'; --p2) ; p2++; if (wcsncmp(p2, L"erts-", wcslen(L"erts-")) == 0) { p = _wcsdup(dir); free(dir); return p; } /* Assume that dir path is like ...\install_dir\bin */ *++p =L'\0'; /* chop off bin dir */ p = find_erlexec_dir2(dir); free(dir); if (p == NULL) { error("Cannot find erlexec.exe"); } else { return p; } } static void get_parameters(void) { wchar_t buffer[MAX_PATH]; wchar_t *ini_filename; HANDLE module = GetModuleHandle(NULL); InitFile *inif; InitSection *inis; char *utf8dir; int len; if (module = NULL) { error("Cannot GetModuleHandle()"); } if (GetModuleFileNameW(module,buffer,MAX_PATH) == 0) { error("Could not GetModuleFileName"); } ini_filename = replace_filename(buffer,INI_FILENAME); if ((inif = load_init_file(ini_filename)) == NULL) { erlexec_dir = find_erlexec_dir(ini_filename); SetEnvironmentVariableW(L"ERLEXEC_DIR", erlexec_dir); } else { if ((inis = lookup_init_section(inif,INI_SECTION)) == NULL) { error("Could not find section %s in init file %S", INI_SECTION, ini_filename); } utf8dir = do_lookup_in_section(inis, "Bindir", INI_SECTION, ini_filename); len = MultiByteToWideChar(CP_UTF8, 0, utf8dir, -1, NULL, 0); erlexec_dir = malloc(len*sizeof(wchar_t)); MultiByteToWideChar(CP_UTF8, 0, utf8dir, -1, erlexec_dir, len); if(len == 0) { error("Bindir is not a valid utf8 '%s' in init file %S", utf8dir, ini_filename); } free_init_file(inif); } erlexec_name = malloc((wcslen(erlexec_dir) + wcslen(ERLEXEC_BASENAME) + 2)*sizeof(wchar_t)); wcscpy(erlexec_name,erlexec_dir); wcscat(erlexec_name, L"\\" ERLEXEC_BASENAME); free(ini_filename); } static void error(char* format, ...) { char sbuf[2048]; va_list ap; va_start(ap, format); vsprintf(sbuf, format, ap); va_end(ap); #ifndef WIN32_WERL fprintf(stderr, "%s\n", sbuf); #else MessageBox(NULL, sbuf, "Werl", MB_OK|MB_ICONERROR); #endif exit(1); }