[ViewVC] Diff of: cvs/Urlader/urlader.c

Comparing Urlader/urlader.c (file contents):
Revision 1.3 by root, Thu Dec 29 10:39:34 2011 UTC vs.
Revision 1.11 by root, Thu Jan 5 06:00:13 2012 UTC

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

Diff Legend

-–
+Removed lines
-+
+Added lines
-<
+Changed lines
->
+Changed lines

Comparing Urlader/urlader.c (file contents): Revision 1.3 by root, Thu Dec 29 10:39:34 2011 UTC vs. Revision 1.11 by root, Thu Jan 5 06:00:13 2012 UTC

Diff Legend

Comparing Urlader/urlader.c (file contents):
Revision 1.3 by root, Thu Dec 29 10:39:34 2011 UTC vs.
Revision 1.11 by root, Thu Jan 5 06:00:13 2012 UTC