--- Urlader/urlader.c 2011/12/30 05:27:32 1.4 +++ Urlader/urlader.c 2012/01/17 18:38:37 1.13 @@ -1,7 +1,41 @@ -#ifndef URLADER -# define URLADER "urlader" -#endif -#define URLADER_VERSION "1.0" +/* + * Copyright (c) 2012 Marc Alexander Lehmann + * + * Redistribution and use in source and binary forms, with or without modifica- + * tion, are permitted provided that the following conditions are met: + * + * 1. Redistributions of source code must retain the above copyright notice, + * this list of conditions and the following disclaimer. + * + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED + * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MER- + * CHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO + * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPE- + * CIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, + * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; + * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, + * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTH- + * ERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED + * OF THE POSSIBILITY OF SUCH DAMAGE. + * + * Alternatively, the contents of this file may be used under the terms of + * the GNU General Public License ("GPL") version 2 or any later version, + * in which case the provisions of the GPL are applicable instead of + * the above. If you wish to allow the use of your version of this file + * only under the terms of the GPL and not to allow others to use your + * version of this file under the BSD license, indicate your decision + * by deleting the provisions above and replace them with the notice + * and other provisions required by the GPL. If you do not delete the + * provisions above, a recipient may use your version of this file under + * either the BSD or the GPL. + */ + +#include "urlib.h" +#include "urlib.c" #include #include @@ -9,91 +43,17 @@ #include #include #include -#include #include "liblzf/lzf_d.c" -#define TAIL_MAGIC "SCHMORPPACK0" - -#ifdef _WIN32 - - #include - //#include - #include - #include - #include - - static DWORD dword; - - #define u_handle HANDLE - #define u_setenv(name,value) SetEnvironmentVariable (name, value) - #define u_mkdir(path) !CreateDirectory (path, NULL) - #define u_chdir(path) !SetCurrentDirectory (path) - #define u_rename(fr,to) !MoveFile (fr, to) - #define u_open(path) CreateFile (path, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL | FILE_FLAG_SEQUENTIAL_SCAN, NULL) - #define u_creat(path,exec) CreateFile (path, GENERIC_WRITE, FILE_SHARE_WRITE, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL | FILE_FLAG_SEQUENTIAL_SCAN, NULL) - #define u_close(handle) CloseHandle (handle) - #define u_append(path,add) PathAppend (path, add) - #define u_write(handle,data,len) (WriteFile (handle, data, len, &dword, 0) ? dword : -1) - - #define u_fsync(handle) FlushFileBuffers (handle) - #define u_sync() - -#else - - #include - #include - #include - - #ifdef PATH_MAX - #define MAX_PATH (PATH_MAX < 4096 ? 4096 : PATH_MAX) - #else - #define MAX_PATH 4096 - #endif - - #define u_handle int - #define u_setenv(name,value) setenv (name, value, 1) - #define u_mkdir(path) mkdir (path, 0777) - #define u_chdir(path) chdir (path) - #define u_rename(fr,to) rename (fr, to) - #define u_open(path) open (path, O_RDONLY) + 1 - #define u_creat(path,exec) open (path, O_WRONLY | O_CREAT | O_TRUNC, (exec) ? 0777 : 0666) + 1 - #define u_close(handle) close (handle - 1) - #define u_append(path,add) strcat (strcat (path, "/"), add) - #define u_write(handle,data,len) write ((handle) - 1, data, len) - - // on a mostly idle system, a sync at the end is certainly faster, hope for the best - #define u_fsync(handle) - #define u_sync() sync () - -#endif - -#define u_16(ptr) (((ptr)[0] << 8) | (ptr)[1]) -#define u_32(ptr) (((ptr)[0] << 24) | ((ptr)[1] << 16) | ((ptr)[2] << 8) | (ptr)[3]) - +/* some simple? dynamic memory management for paths might save ltos of ram */ static u_handle pack_handle; +static u_handle lock_handle; static char tmppath[MAX_PATH]; -static char currdir[MAX_PATH]; -static char datadir[MAX_PATH]; // %AppData%/urlader -static char exe_dir[MAX_PATH]; // %AppData%/urlader/EXE_ID -static char execdir[MAX_PATH]; // %AppData%/urlader/EXE_ID/EXE_VER -static char exe_id[MAX_PATH]; -static char exe_ver[MAX_PATH]; static int exe_argc; -static const char *exe_argv[32]; - -static void -fatal (const char *msg) -{ -#ifdef _WIN32 - MessageBox (0, msg, URLADER, 0); -#else - fprintf (stderr, "%s: %s\n", URLADER, msg); -#endif - - _exit (1); -} +static u_dynbuf exe_argv; +static u_dynbuf exe_args; /* actual arguments strings copied here */ static void tmpdir (const char *dir) @@ -114,15 +74,15 @@ } static void -systemv (const char *const argv[], int dowait) +systemv (const char *const argv[]) { #ifdef _WIN32 _spawnv (P_WAIT, argv [0], argv); #else - pid_t pid = dowait ? fork () : 0; + pid_t pid = fork (); if (pid < 0) - fatal ("fork failure"); + u_fatal ("fork failure"); if (!pid) { @@ -145,45 +105,14 @@ #else const char *argv[] = { "/bin/rm", "-rf", path, 0 }; #endif - systemv (argv, 1); + systemv (argv); } -enum -{ - T_NULL, // 5 - T_META, // 1 : exe_id, exe_ver - T_ENV, // 2 : name, value - T_ARG, // 3 : arg - T_DIR, // 4+: path - T_FILE, // 4+: path, data - T_NUM -}; - -enum -{ - F_EXEC = 1, - F_LZF = 2, - F_NULL = 0 -}; - -struct hdr -{ - unsigned char type; - unsigned char flags; - unsigned char namelen[2]; - unsigned char datalen[4]; -}; - -struct tail { - unsigned char max_filesize[4]; - unsigned char size[4]; - char magic[12]; -}; - static char *pack_base, *pack_end; -static struct hdr *pack_cur; +static struct u_pack_tail *pack_tail; +static struct u_pack_hdr *pack_cur; static char *scratch; -static unsigned int scratch_len; +static unsigned int scratch_size; #define PACK_NAME ((char *)(pack_cur + 1)) #define PACK_DATA (PACK_NAME + u_16 (pack_cur->namelen) + 1) @@ -194,121 +123,110 @@ { unsigned int d = u_32 (pack_cur->datalen); - pack_cur = (struct hdr *)(PACK_DATA + d); + pack_cur = (struct u_pack_hdr *)(PACK_DATA + d); } static void pack_unmap (void) { - if (!pack_base) - return; - - free (scratch); + if (pack_base) + { + u_munmap (pack_base, pack_end - pack_base); + pack_base = 0; + } -#ifdef _WIN32 - UnmapViewOfFile (pack_base); -#else - munmap (pack_base, pack_end - pack_base); -#endif + if (scratch) + { + u_free (scratch, scratch_size); + scratch = 0; + } } static int pack_map (void) { + char *addr; + unsigned int size; + #ifdef _WIN32 BY_HANDLE_FILE_INFORMATION fi; if (!GetFileInformationByHandle (pack_handle, &fi)) return 0; - if (fi.nFileSizeHigh) // too big - return 0; - - HANDLE handle = CreateFileMapping (pack_handle, 0, PAGE_READONLY, 0, fi.nFileSizeLow, NULL); + size = fi.nFileSizeLow; +#else + size = lseek (pack_handle, 0, SEEK_END); +#endif - if (!handle) + addr = u_mmap (pack_handle, size); + if (!addr) return 0; - pack_unmap (); - - pack_base = MapViewOfFile (handle, FILE_MAP_READ, 0, 0, fi.nFileSizeLow); - - CloseHandle (handle); + /*pack_unmap ();*/ - pack_end = pack_base + fi.nFileSizeLow; -#else - off_t len = lseek (pack_handle - 1, 0, SEEK_END); - - pack_unmap (); + pack_base = addr; + pack_end = pack_base + size; - pack_base = mmap (0, len, PROT_READ, MAP_SHARED, pack_handle - 1, 0); - if (pack_base == (void *)-1) - pack_base = 0; + pack_tail = (void *)(pack_end - sizeof (*pack_tail)); - pack_end = pack_base + len; -#endif - - if (!pack_base) + if (memcmp (pack_tail->magic, TAIL_MAGIC, sizeof (TAIL_MAGIC) - 1)) return 0; - struct tail *tail; - - tail = (void *)(pack_end - sizeof (*tail)); + pack_cur = (struct u_pack_hdr *)(pack_end - u_32 (pack_tail->size)); - if (memcmp (tail->magic, TAIL_MAGIC, sizeof (TAIL_MAGIC) - 1)) + if (pack_cur->type != T_META) return 0; - pack_cur = (struct hdr *)(pack_end - u_32 (tail->size)); - - free (scratch); - scratch = malloc (scratch_len = u_32 (tail->max_filesize)); + scratch = u_malloc (scratch_size = u_32 (pack_tail->max_uncompressed)); + if (!scratch) + return 0; return 1; } static void -exe_info (void) +add_arg (char *arg, unsigned int len) { - if (!pack_map ()) - fatal ("unable to locate packfile in executable - executable corrupted?"); - - if (pack_cur->type != T_META) - fatal ("unable to locate executable metadata - executable corrupted?"); - - strcpy (exe_id, PACK_NAME); + char *addr = u_dynbuf_append (&exe_args, arg, len); + u_dynbuf_append (&exe_argv, &addr, sizeof (addr)); } static void load (void) { - u_mkdir (datadir); - - strcpy (exe_dir, datadir); - u_append (exe_dir, exe_id); - u_mkdir (exe_dir); + strcpy (exe_id , PACK_NAME); + strcpy (exe_ver, PACK_DATA); + u_set_exe_info (); if (u_chdir (exe_dir)) - fatal ("unable to change to application data directory"); + u_fatal ("unable to change to application data directory"); u_handle h = u_open ("override"); - if (h) + if (u_valid (h)) { u_handle oh = pack_handle; + pack_unmap (); pack_handle = h; - if (!pack_map ()) + + if (pack_map () + && strcmp (exe_id , PACK_NAME) == 0 + && strcmp (exe_ver, PACK_DATA) <= 0) + u_setenv ("URLADER_OVERRIDE", "override"); + else { + pack_unmap (); pack_handle = oh; oh = h; + pack_map (); } u_close (oh); } - if (pack_cur->type != T_META) - fatal ("unable to locate override metadata"); - strcpy (exe_ver, PACK_DATA); + u_set_exe_info (); pack_next (); for (;;) @@ -316,7 +234,7 @@ if (pack_cur->type == T_ENV) u_setenv (PACK_NAME, PACK_DATA); else if (pack_cur->type == T_ARG) - exe_argv [exe_argc++] = strdup (PACK_NAME); + add_arg (PACK_NAME, u_16 (pack_cur->namelen) + 1); else break; @@ -324,9 +242,11 @@ } done_env_arg: - strcpy (execdir, exe_dir); - u_append (execdir, "i-"); - strcat (execdir, exe_ver); + + strcat (strcpy (tmppath, execdir), ".lck"); + lock_handle = u_lock (tmppath, 0, 1); + if (!lock_handle) + u_fatal ("unable to lock application instance"); if (access (execdir, F_OK)) { @@ -334,7 +254,7 @@ tmpdir (exe_dir); if (u_chdir (tmppath)) - fatal ("unable to change to new instance directory"); + u_fatal ("unable to change to new instance directory"); for (;;) { @@ -351,19 +271,19 @@ char *data = PACK_DATA; if (pack_cur->flags & F_LZF) - if (dlen = lzf_decompress (data, len, scratch, scratch_len)) + if (dlen = lzf_decompress (data, len, scratch, scratch_size)) { data = scratch; len = dlen; } else - fatal ("unable to uncompress file data - pack corrupted?"); + u_fatal ("unable to uncompress file data - pack corrupted?"); - if (!h) - fatal ("unable to unpack file from packfile - disk full?"); + if (!u_valid (h)) + u_fatal ("unable to unpack file from packfile - disk full?"); if (u_write (h, data, len) != len) - fatal ("unable to unpack file from packfile - disk full?"); + u_fatal ("unable to unpack file from packfile - disk full?"); u_fsync (h); u_close (h); @@ -379,7 +299,7 @@ done: if (u_chdir (datadir)) - fatal ("unable to change to data directory"); + u_fatal ("unable to change to data directory"); u_sync (); @@ -391,15 +311,11 @@ u_close (pack_handle); if (u_chdir (execdir)) - fatal ("unable to change to application instance directory"); + u_fatal ("unable to change to application instance directory"); u_setenv ("URLADER_VERSION", URLADER_VERSION); - u_setenv ("URLADER_DATADIR", datadir); - u_setenv ("URLADER_EXECDIR", execdir); - u_setenv ("URLADER_EXE_ID" , exe_id); - u_setenv ("URLADER_EXE_DIR", exe_dir); - u_setenv ("URLADER_EXE_VER", exe_ver); +#if 0 // yes, this is overkill u_setenv ("SHLIB_PATH" , execdir); // hpux u_setenv ("LIBPATH" , execdir); // aix @@ -409,13 +325,40 @@ u_setenv ("LD_LIBRARYN32_PATH", execdir); // irix u_setenv ("LD_LIBRARY64_PATH" , execdir); // irix u_setenv ("DYLD_LIBRARY_PATH" , execdir); // os sucks from apple +#endif } static void execute (void) { - exe_argv [exe_argc] = 0; - systemv (exe_argv, 0); + char *null = 0; + u_dynbuf_append (&exe_argv, &null, sizeof (null)); + systemv ((const char *const *)exe_argv.addr); +} + +// this argc/argv is without argv [0] +static void +doit (int argc, char *argv[]) +{ + int i; + + u_setenv ("URLADER_CURRDIR", currdir); + + u_set_datadir (); + u_mkdir (datadir); + + if (!pack_map ()) + u_fatal ("unable to map pack file - executable corrupted?"); + + load (); + + while (argc--) + { + add_arg (*argv, strlen (*argv) + 1); + ++argv; + } + + execute (); } #ifdef _WIN32 @@ -424,24 +367,18 @@ WinMain (HINSTANCE hI, HINSTANCE hP, LPSTR argv, int command_show) { if (!GetModuleFileName (hI, tmppath, sizeof (tmppath))) - fatal ("unable to find executable pack"); + u_fatal ("unable to find executable pack"); - if (!(pack_handle = u_open (tmppath))) - fatal ("unable to open executable pack"); + u_setenv ("URLADER_EXEPATH", tmppath); - exe_info (); + pack_handle = u_open (tmppath); + if (!u_valid (pack_handle)) + u_fatal ("unable to open executable pack"); if (!GetCurrentDirectory (sizeof (currdir), currdir)) strcpy (currdir, "."); - if (SHGetFolderPath (0, CSIDL_APPDATA | CSIDL_FLAG_CREATE, NULL, SHGFP_TYPE_CURRENT, datadir) != S_OK) - fatal ("unable to find application data directory"); - - u_mkdir (datadir); - u_append (datadir, URLADER); - - load (); - execute (); + doit (1, &argv); return 0; } @@ -451,39 +388,25 @@ int main (int argc, char *argv[]) { - char *home = getenv ("HOME"); + u_setenv ("URLADER_EXEPATH", argv [0]); - // we assume fd 0 and 2 are "normal" - - if (!(pack_handle = u_open (argv [0]))) - fatal ("unable to open executable pack"); - - exe_info (); - - if (!home) - { - struct passwd *pw; - - if ((pw = getpwuid (getuid ()))) - home = pw->pw_dir; - else - home = "/tmp"; - } + pack_handle = u_open (argv [0]); + if (!u_valid (pack_handle)) + u_fatal ("unable to open executable pack"); if (!getcwd (currdir, sizeof (currdir))) strcpy (currdir, "."); - u_mkdir (home); - sprintf (datadir, "%s/.%s", home, URLADER); - u_mkdir (datadir); +#if 0 + /* intersperse hostname, for whatever reason */ if (gethostname (tmppath, sizeof (tmppath))) strcpy (tmppath, "default"); u_append (datadir, tmppath); +#endif - load (); - execute (); + doit (argc - 1, argv + 1); return 0; }