Urlader/urlader.c

#ifndef URLADER
# define URLADER "urlader"
#endif
#define URLADER_VERSION "1.0"

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <errno.h>
#include <string.h>
#include <time.h>
#include <fcntl.h>

#include "liblzf/lzf_d.c"

#define TAIL_MAGIC "SCHMORPPACK0"

#ifdef _WIN32

   #include <windows.h>
   //#include <winbase.h>
   #include <shlobj.h>
   #include <shlwapi.h>
   #include <wininet.h>

   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 <sys/mman.h>
   #include <sys/types.h>
   #include <pwd.h>

   #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])

static u_handle pack_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 void
tmpdir (const char *dir)
{
  static int cnt;

  for (;;)
    {
//      #ifdef _WIN32
//        sprintf (tmppath, "%s/%x_%x.tmp", dir, (unsigned int)GetCurrentProcessId (), ++cnt);
//      #else
        sprintf (tmppath, "%s/t-%x_%x.tmp", dir, (unsigned int)getpid ()             , ++cnt);
//      #endif

      if (!u_mkdir (tmppath))
        return;
    }
}

static void
systemv (const char *const argv[], int dowait)
{
#ifdef _WIN32
  _spawnv (P_WAIT, argv [0], argv);
#else
  pid_t pid = dowait ? fork () : 0;

  if (pid < 0)
    fatal ("fork failure");

  if (!pid)
    {
      execv (argv [0], (void *)argv);
      _exit (124);
    }

  int status;
  waitpid (pid, &status, 0);
#endif
}

static void
deltree (const char *path)
{
  #ifdef _WIN32
    char buf[MAX_PATH * 2 + 64];
    const char *argv[] = { getenv ("COMSPEC"), "/c", "rd", "/s", "/q", buf, 0 };
    sprintf (buf, "\"%s\"", path);
  #else
    const char *argv[] = { "/bin/rm", "-rf", path, 0 };
  #endif
  systemv (argv, 1);
}

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 char *scratch;
static unsigned int scratch_len;

#define PACK_NAME ((char *)(pack_cur + 1))
#define PACK_DATA (PACK_NAME + u_16 (pack_cur->namelen) + 1)
#define PACK_VALID pack_cur->type

static void
pack_next (void)
{
  unsigned int d = u_32 (pack_cur->datalen);

  pack_cur = (struct hdr *)(PACK_DATA + d);
}

static void
pack_unmap (void)
{
  if (!pack_base)
    return;

  free (scratch);

#ifdef _WIN32
  UnmapViewOfFile (pack_base);
#else
  munmap (pack_base, pack_end - pack_base);
#endif
}

static int
pack_map (void)
{
#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);

  if (!handle)
    return 0;

  pack_unmap ();

  pack_base = MapViewOfFile (handle, FILE_MAP_READ, 0, 0, fi.nFileSizeLow);

  CloseHandle (handle);

  pack_end  = pack_base + fi.nFileSizeLow;
#else
  off_t len = lseek (pack_handle - 1, 0, SEEK_END);

  pack_unmap ();

  pack_base = mmap (0, len, PROT_READ, MAP_SHARED, pack_handle - 1, 0);
  if (pack_base == (void *)-1)
    pack_base = 0;

  pack_end = pack_base + len;
#endif

  if (!pack_base)
    return 0;

  struct tail *tail;

  tail = (void *)(pack_end - sizeof (*tail));

  if (memcmp (tail->magic, TAIL_MAGIC, sizeof (TAIL_MAGIC) - 1))
    return 0;

  pack_cur = (struct hdr *)(pack_end - u_32 (tail->size));

  free (scratch);
  scratch = malloc (scratch_len = u_32 (tail->max_filesize));

  return 1;
}

static void
exe_info (void)
{
  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);
}

static void
load (void)
{
  u_mkdir (datadir);

  strcpy (exe_dir, datadir);
  u_append (exe_dir, exe_id);
  u_mkdir (exe_dir);

  if (u_chdir (exe_dir))
    fatal ("unable to change to application data directory");

  u_handle h = u_open ("override");
  if (h)
    {
      u_handle oh = pack_handle;

      pack_handle = h;
      if (!pack_map ())
        {
          pack_handle = oh;
          oh = h;
        }

      u_close (oh);
    }

  if (pack_cur->type != T_META)
    fatal ("unable to locate override metadata");

  strcpy (exe_ver, PACK_DATA);
  pack_next ();

  for (;;)
    {
      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);
      else
        break;

      pack_next ();
    }
 
done_env_arg:
  strcpy (execdir, exe_dir);
  u_append (execdir, "i-");
  strcat (execdir, exe_ver);

  if (access (execdir, F_OK))
    {
      // does not exist yet, so unpack and move
      tmpdir (exe_dir);

      if (u_chdir (tmppath))
        fatal ("unable to change to new instance directory");

      for (;;)
        {
          switch (pack_cur->type)
            {
              case T_DIR:
                u_mkdir (PACK_NAME);
                break;

              case T_FILE:
                {
                  u_handle h = u_creat (PACK_NAME, pack_cur->flags & F_EXEC);
                  unsigned int dlen, len = u_32 (pack_cur->datalen);
                  char *data = PACK_DATA;

                  if (pack_cur->flags & F_LZF)
                    if (dlen = lzf_decompress (data, len, scratch, scratch_len))
                      {
                        data = scratch;
                        len  = dlen;
                      }
                    else
                      fatal ("unable to uncompress file data - pack corrupted?");

                  if (!h)
                    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_fsync (h);
                  u_close (h);
                }
                break;

              case T_NULL:
                goto done;
            }

          pack_next ();
        }

done:
      if (u_chdir (datadir))
        fatal ("unable to change to data directory");

      u_sync ();

      if (u_rename (tmppath, execdir))
        deltree (tmppath); // if move fails, delete new, assume other process created it independently
    }

  pack_unmap ();
  u_close (pack_handle);

  if (u_chdir (execdir))
    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);

  // yes, this is overkill
  u_setenv ("SHLIB_PATH"        , execdir); // hpux
  u_setenv ("LIBPATH"           , execdir); // aix
  u_setenv ("LD_LIBRARY_PATH"   , execdir); // most elf systems
  u_setenv ("LD_LIBRARY_PATH_32", execdir); // solaris
  u_setenv ("LD_LIBRARY_PATH_64", execdir); // solaris
  u_setenv ("LD_LIBRARYN32_PATH", execdir); // irix
  u_setenv ("LD_LIBRARY64_PATH" , execdir); // irix
  u_setenv ("DYLD_LIBRARY_PATH" , execdir); // os sucks from apple
}

static void
execute (void)
{
  exe_argv [exe_argc] = 0;
  systemv (exe_argv, 0);
}

#ifdef _WIN32

int APIENTRY
WinMain (HINSTANCE hI, HINSTANCE hP, LPSTR argv, int command_show)
{
  if (!GetModuleFileName (hI, tmppath, sizeof (tmppath)))
    fatal ("unable to find executable pack");

  if (!(pack_handle = u_open (tmppath)))
    fatal ("unable to open executable pack");

  exe_info ();

  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 ();

  return 0;
}

#else

int
main (int argc, char *argv[])
{
  char *home = getenv ("HOME");

  // 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";
    }

  if (!getcwd (currdir, sizeof (currdir)))
    strcpy (currdir, ".");

  u_mkdir (home);
  sprintf (datadir, "%s/.%s", home, URLADER);
  u_mkdir (datadir);

  if (gethostname (tmppath, sizeof (tmppath)))
    strcpy (tmppath, "default");

  u_append (datadir, tmppath);

  load ();
  execute ();

  return 0;
}

#endif

Revision:	1.4
Committed:	Fri Dec 30 05:27:32 2011 UTC (12 years, 4 months ago) by root
Content type:	text/plain
Branch:	MAIN
Changes since 1.3:	+18 -11 lines
Log Message:	* empty log message *
#	Content
1	#ifndef URLADER
2	# define URLADER "urlader"
3	#endif
4	#define URLADER_VERSION "1.0"
5
6	#include <stdio.h>
7	#include <stdlib.h>
8	#include <unistd.h>
9	#include <errno.h>
10	#include <string.h>
11	#include <time.h>
12	#include <fcntl.h>
13
14	#include "liblzf/lzf_d.c"
15
16	#define TAIL_MAGIC "SCHMORPPACK0"
17
18	#ifdef _WIN32
19
20	#include <windows.h>
21	//#include <winbase.h>
22	#include <shlobj.h>
23	#include <shlwapi.h>
24	#include <wininet.h>
25
26	static DWORD dword;
27
28	#define u_handle HANDLE
29	#define u_setenv(name,value) SetEnvironmentVariable (name, value)
30	#define u_mkdir(path) !CreateDirectory (path, NULL)
31	#define u_chdir(path) !SetCurrentDirectory (path)
32	#define u_rename(fr,to) !MoveFile (fr, to)
33	#define u_open(path) CreateFile (path, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL \| FILE_FLAG_SEQUENTIAL_SCAN, NULL)
34	#define u_creat(path,exec) CreateFile (path, GENERIC_WRITE, FILE_SHARE_WRITE, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL \| FILE_FLAG_SEQUENTIAL_SCAN, NULL)
35	#define u_close(handle) CloseHandle (handle)
36	#define u_append(path,add) PathAppend (path, add)
37	#define u_write(handle,data,len) (WriteFile (handle, data, len, &dword, 0) ? dword : -1)
38
39	#define u_fsync(handle) FlushFileBuffers (handle)
40	#define u_sync()
41
42	#else
43
44	#include <sys/mman.h>
45	#include <sys/types.h>
46	#include <pwd.h>
47
48	#ifdef PATH_MAX
49	#define MAX_PATH (PATH_MAX < 4096 ? 4096 : PATH_MAX)
50	#else
51	#define MAX_PATH 4096
52	#endif
53
54	#define u_handle int
55	#define u_setenv(name,value) setenv (name, value, 1)
56	#define u_mkdir(path) mkdir (path, 0777)
57	#define u_chdir(path) chdir (path)
58	#define u_rename(fr,to) rename (fr, to)
59	#define u_open(path) open (path, O_RDONLY) + 1
60	#define u_creat(path,exec) open (path, O_WRONLY \| O_CREAT \| O_TRUNC, (exec) ? 0777 : 0666) + 1
61	#define u_close(handle) close (handle - 1)
62	#define u_append(path,add) strcat (strcat (path, "/"), add)
63	#define u_write(handle,data,len) write ((handle) - 1, data, len)
64
65	// on a mostly idle system, a sync at the end is certainly faster, hope for the best
66	#define u_fsync(handle)
67	#define u_sync() sync ()
68
69	#endif
70
71	#define u_16(ptr) (((ptr)[0] << 8) \| (ptr)[1])
72	#define u_32(ptr) (((ptr)[0] << 24) \| ((ptr)[1] << 16) \| ((ptr)[2] << 8) \| (ptr)[3])
73
74	static u_handle pack_handle;
75	static char tmppath[MAX_PATH];
76	static char currdir[MAX_PATH];
77	static char datadir[MAX_PATH]; // %AppData%/urlader
78	static char exe_dir[MAX_PATH]; // %AppData%/urlader/EXE_ID
79	static char execdir[MAX_PATH]; // %AppData%/urlader/EXE_ID/EXE_VER
80	static char exe_id[MAX_PATH];
81	static char exe_ver[MAX_PATH];
82
83	static int exe_argc;
84	static const char *exe_argv[32];
85
86	static void
87	fatal (const char *msg)
88	{
89	#ifdef _WIN32
90	MessageBox (0, msg, URLADER, 0);
91	#else
92	fprintf (stderr, "%s: %s\n", URLADER, msg);
93	#endif
94
95	_exit (1);
96	}
97
98	static void
99	tmpdir (const char *dir)
100	{
101	static int cnt;
102
103	for (;;)
104	{
105	// #ifdef _WIN32
106	// sprintf (tmppath, "%s/%x_%x.tmp", dir, (unsigned int)GetCurrentProcessId (), ++cnt);
107	// #else
108	sprintf (tmppath, "%s/t-%x_%x.tmp", dir, (unsigned int)getpid () , ++cnt);
109	// #endif
110
111	if (!u_mkdir (tmppath))
112	return;
113	}
114	}
115
116	static void
117	systemv (const char *const argv[], int dowait)
118	{
119	#ifdef _WIN32
120	_spawnv (P_WAIT, argv [0], argv);
121	#else
122	pid_t pid = dowait ? fork () : 0;
123
124	if (pid < 0)
125	fatal ("fork failure");
126
127	if (!pid)
128	{
129	execv (argv [0], (void *)argv);
130	_exit (124);
131	}
132
133	int status;
134	waitpid (pid, &status, 0);
135	#endif
136	}
137
138	static void
139	deltree (const char *path)
140	{
141	#ifdef _WIN32
142	char buf[MAX_PATH * 2 + 64];
143	const char *argv[] = { getenv ("COMSPEC"), "/c", "rd", "/s", "/q", buf, 0 };
144	sprintf (buf, "\"%s\"", path);
145	#else
146	const char *argv[] = { "/bin/rm", "-rf", path, 0 };
147	#endif
148	systemv (argv, 1);
149	}
150
151	enum
152	{
153	T_NULL, // 5
154	T_META, // 1 : exe_id, exe_ver
155	T_ENV, // 2 : name, value
156	T_ARG, // 3 : arg
157	T_DIR, // 4+: path
158	T_FILE, // 4+: path, data
159	T_NUM
160	};
161
162	enum
163	{
164	F_EXEC = 1,
165	F_LZF = 2,
166	F_NULL = 0
167	};
168
169	struct hdr
170	{
171	unsigned char type;
172	unsigned char flags;
173	unsigned char namelen[2];
174	unsigned char datalen[4];
175	};
176
177	struct tail {
178	unsigned char max_filesize[4];
179	unsigned char size[4];
180	char magic[12];
181	};
182
183	static char pack_base, pack_end;
184	static struct hdr *pack_cur;
185	static char *scratch;
186	static unsigned int scratch_len;
187
188	#define PACK_NAME ((char *)(pack_cur + 1))
189	#define PACK_DATA (PACK_NAME + u_16 (pack_cur->namelen) + 1)
190	#define PACK_VALID pack_cur->type
191
192	static void
193	pack_next (void)
194	{
195	unsigned int d = u_32 (pack_cur->datalen);
196
197	pack_cur = (struct hdr *)(PACK_DATA + d);
198	}
199
200	static void
201	pack_unmap (void)
202	{
203	if (!pack_base)
204	return;
205
206	free (scratch);
207
208	#ifdef _WIN32
209	UnmapViewOfFile (pack_base);
210	#else
211	munmap (pack_base, pack_end - pack_base);
212	#endif
213	}
214
215	static int
216	pack_map (void)
217	{
218	#ifdef _WIN32
219	BY_HANDLE_FILE_INFORMATION fi;
220
221	if (!GetFileInformationByHandle (pack_handle, &fi))
222	return 0;
223
224	if (fi.nFileSizeHigh) // too big
225	return 0;
226
227	HANDLE handle = CreateFileMapping (pack_handle, 0, PAGE_READONLY, 0, fi.nFileSizeLow, NULL);
228
229	if (!handle)
230	return 0;
231
232	pack_unmap ();
233
234	pack_base = MapViewOfFile (handle, FILE_MAP_READ, 0, 0, fi.nFileSizeLow);
235
236	CloseHandle (handle);
237
238	pack_end = pack_base + fi.nFileSizeLow;
239	#else
240	off_t len = lseek (pack_handle - 1, 0, SEEK_END);
241
242	pack_unmap ();
243
244	pack_base = mmap (0, len, PROT_READ, MAP_SHARED, pack_handle - 1, 0);
245	if (pack_base == (void *)-1)
246	pack_base = 0;
247
248	pack_end = pack_base + len;
249	#endif
250
251	if (!pack_base)
252	return 0;
253
254	struct tail *tail;
255
256	tail = (void )(pack_end - sizeof (tail));
257
258	if (memcmp (tail->magic, TAIL_MAGIC, sizeof (TAIL_MAGIC) - 1))
259	return 0;
260
261	pack_cur = (struct hdr *)(pack_end - u_32 (tail->size));
262
263	free (scratch);
264	scratch = malloc (scratch_len = u_32 (tail->max_filesize));
265
266	return 1;
267	}
268
269	static void
270	exe_info (void)
271	{
272	if (!pack_map ())
273	fatal ("unable to locate packfile in executable - executable corrupted?");
274
275	if (pack_cur->type != T_META)
276	fatal ("unable to locate executable metadata - executable corrupted?");
277
278	strcpy (exe_id, PACK_NAME);
279	}
280
281	static void
282	load (void)
283	{
284	u_mkdir (datadir);
285
286	strcpy (exe_dir, datadir);
287	u_append (exe_dir, exe_id);
288	u_mkdir (exe_dir);
289
290	if (u_chdir (exe_dir))
291	fatal ("unable to change to application data directory");
292
293	u_handle h = u_open ("override");
294	if (h)
295	{
296	u_handle oh = pack_handle;
297
298	pack_handle = h;
299	if (!pack_map ())
300	{
301	pack_handle = oh;
302	oh = h;
303	}
304
305	u_close (oh);
306	}
307
308	if (pack_cur->type != T_META)
309	fatal ("unable to locate override metadata");
310
311	strcpy (exe_ver, PACK_DATA);
312	pack_next ();
313
314	for (;;)
315	{
316	if (pack_cur->type == T_ENV)
317	u_setenv (PACK_NAME, PACK_DATA);
318	else if (pack_cur->type == T_ARG)
319	exe_argv [exe_argc++] = strdup (PACK_NAME);
320	else
321	break;
322
323	pack_next ();
324	}
325
326	done_env_arg:
327	strcpy (execdir, exe_dir);
328	u_append (execdir, "i-");
329	strcat (execdir, exe_ver);
330
331	if (access (execdir, F_OK))
332	{
333	// does not exist yet, so unpack and move
334	tmpdir (exe_dir);
335
336	if (u_chdir (tmppath))
337	fatal ("unable to change to new instance directory");
338
339	for (;;)
340	{
341	switch (pack_cur->type)
342	{
343	case T_DIR:
344	u_mkdir (PACK_NAME);
345	break;
346
347	case T_FILE:
348	{
349	u_handle h = u_creat (PACK_NAME, pack_cur->flags & F_EXEC);
350	unsigned int dlen, len = u_32 (pack_cur->datalen);
351	char *data = PACK_DATA;
352
353	if (pack_cur->flags & F_LZF)
354	if (dlen = lzf_decompress (data, len, scratch, scratch_len))
355	{
356	data = scratch;
357	len = dlen;
358	}
359	else
360	fatal ("unable to uncompress file data - pack corrupted?");
361
362	if (!h)
363	fatal ("unable to unpack file from packfile - disk full?");
364
365	if (u_write (h, data, len) != len)
366	fatal ("unable to unpack file from packfile - disk full?");
367
368	u_fsync (h);
369	u_close (h);
370	}
371	break;
372
373	case T_NULL:
374	goto done;
375	}
376
377	pack_next ();
378	}
379
380	done:
381	if (u_chdir (datadir))
382	fatal ("unable to change to data directory");
383
384	u_sync ();
385
386	if (u_rename (tmppath, execdir))
387	deltree (tmppath); // if move fails, delete new, assume other process created it independently
388	}
389
390	pack_unmap ();
391	u_close (pack_handle);
392
393	if (u_chdir (execdir))
394	fatal ("unable to change to application instance directory");
395
396	u_setenv ("URLADER_VERSION", URLADER_VERSION);
397	u_setenv ("URLADER_DATADIR", datadir);
398	u_setenv ("URLADER_EXECDIR", execdir);
399	u_setenv ("URLADER_EXE_ID" , exe_id);
400	u_setenv ("URLADER_EXE_DIR", exe_dir);
401	u_setenv ("URLADER_EXE_VER", exe_ver);
402
403	// yes, this is overkill
404	u_setenv ("SHLIB_PATH" , execdir); // hpux
405	u_setenv ("LIBPATH" , execdir); // aix
406	u_setenv ("LD_LIBRARY_PATH" , execdir); // most elf systems
407	u_setenv ("LD_LIBRARY_PATH_32", execdir); // solaris
408	u_setenv ("LD_LIBRARY_PATH_64", execdir); // solaris
409	u_setenv ("LD_LIBRARYN32_PATH", execdir); // irix
410	u_setenv ("LD_LIBRARY64_PATH" , execdir); // irix
411	u_setenv ("DYLD_LIBRARY_PATH" , execdir); // os sucks from apple
412	}
413
414	static void
415	execute (void)
416	{
417	exe_argv [exe_argc] = 0;
418	systemv (exe_argv, 0);
419	}
420
421	#ifdef _WIN32
422
423	int APIENTRY
424	WinMain (HINSTANCE hI, HINSTANCE hP, LPSTR argv, int command_show)
425	{
426	if (!GetModuleFileName (hI, tmppath, sizeof (tmppath)))
427	fatal ("unable to find executable pack");
428
429	if (!(pack_handle = u_open (tmppath)))
430	fatal ("unable to open executable pack");
431
432	exe_info ();
433
434	if (!GetCurrentDirectory (sizeof (currdir), currdir))
435	strcpy (currdir, ".");
436
437	if (SHGetFolderPath (0, CSIDL_APPDATA \| CSIDL_FLAG_CREATE, NULL, SHGFP_TYPE_CURRENT, datadir) != S_OK)
438	fatal ("unable to find application data directory");
439
440	u_mkdir (datadir);
441	u_append (datadir, URLADER);
442
443	load ();
444	execute ();
445
446	return 0;
447	}
448
449	#else
450
451	int
452	main (int argc, char *argv[])
453	{
454	char *home = getenv ("HOME");
455
456	// we assume fd 0 and 2 are "normal"
457
458	if (!(pack_handle = u_open (argv [0])))
459	fatal ("unable to open executable pack");
460
461	exe_info ();
462
463	if (!home)
464	{
465	struct passwd *pw;
466
467	if ((pw = getpwuid (getuid ())))
468	home = pw->pw_dir;
469	else
470	home = "/tmp";
471	}
472
473	if (!getcwd (currdir, sizeof (currdir)))
474	strcpy (currdir, ".");
475
476	u_mkdir (home);
477	sprintf (datadir, "%s/.%s", home, URLADER);
478	u_mkdir (datadir);
479
480	if (gethostname (tmppath, sizeof (tmppath)))
481	strcpy (tmppath, "default");
482
483	u_append (datadir, tmppath);
484
485	load ();
486	execute ();
487
488	return 0;
489	}
490
491	#endif
492