ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/Urlader/urlader.c
(Generate patch)

Comparing Urlader/urlader.c (file contents):
Revision 1.4 by root, Fri Dec 30 05:27:32 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
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
74static u_handle pack_handle; 14static u_handle pack_handle;
15static u_handle lock_handle;
75static char tmppath[MAX_PATH]; 16static char tmppath[MAX_PATH];
76static char currdir[MAX_PATH];
77static char datadir[MAX_PATH]; // %AppData%/urlader
78static char exe_dir[MAX_PATH]; // %AppData%/urlader/EXE_ID
79static char execdir[MAX_PATH]; // %AppData%/urlader/EXE_ID/EXE_VER
80static char exe_id[MAX_PATH];
81static char exe_ver[MAX_PATH];
82 17
83static int exe_argc; 18static int exe_argc;
84static const char *exe_argv[32]; 19static const char *exe_argv[MAX_ARGC];
85 20static char exe_args[MAX_ARGS]; /* actual arguments strings copied here */
86static void 21static unsigned int exe_argo;
87fatal (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 22
98static void 23static void
99tmpdir (const char *dir) 24tmpdir (const char *dir)
100{ 25{
101 static int cnt; 26 static int cnt;
… …
112 return; 37 return;
113 } 38 }
114} 39}
115 40
116static void 41static void
117systemv (const char *const argv[], int dowait) 42systemv (const char *const argv[])
118{ 43{
119#ifdef _WIN32 44#ifdef _WIN32
120 _spawnv (P_WAIT, argv [0], argv); 45 _spawnv (P_WAIT, argv [0], argv);
121#else 46#else
122 pid_t pid = dowait ? fork () : 0; 47 pid_t pid = fork ();
123 48
124 if (pid < 0) 49 if (pid < 0)
125 fatal ("fork failure"); 50 u_fatal ("fork failure");
126 51
127 if (!pid) 52 if (!pid)
128 { 53 {
129 execv (argv [0], (void *)argv); 54 execv (argv [0], (void *)argv);
130 _exit (124); 55 _exit (124);
… …
143 const char *argv[] = { getenv ("COMSPEC"), "/c", "rd", "/s", "/q", buf, 0 }; 68 const char *argv[] = { getenv ("COMSPEC"), "/c", "rd", "/s", "/q", buf, 0 };
144 sprintf (buf, "\"%s\"", path); 69 sprintf (buf, "\"%s\"", path);
145 #else 70 #else
146 const char *argv[] = { "/bin/rm", "-rf", path, 0 }; 71 const char *argv[] = { "/bin/rm", "-rf", path, 0 };
147 #endif 72 #endif
148 systemv (argv, 1); 73 systemv (argv);
149} 74}
150
151enum
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
162enum
163{
164 F_EXEC = 1,
165 F_LZF = 2,
166 F_NULL = 0
167};
168
169struct hdr
170{
171 unsigned char type;
172 unsigned char flags;
173 unsigned char namelen[2];
174 unsigned char datalen[4];
175};
176
177struct tail {
178 unsigned char max_filesize[4];
179 unsigned char size[4];
180 char magic[12];
181};
182 75
183static char *pack_base, *pack_end; 76static char *pack_base, *pack_end;
184static struct hdr *pack_cur; 77static struct u_pack_hdr *pack_cur;
185static char *scratch; 78static char *scratch;
186static unsigned int scratch_len; 79static unsigned int scratch_size;
187 80
188#define PACK_NAME ((char *)(pack_cur + 1)) 81#define PACK_NAME ((char *)(pack_cur + 1))
189#define PACK_DATA (PACK_NAME + u_16 (pack_cur->namelen) + 1) 82#define PACK_DATA (PACK_NAME + u_16 (pack_cur->namelen) + 1)
190#define PACK_VALID pack_cur->type 83#define PACK_VALID pack_cur->type
191 84
192static void 85static void
193pack_next (void) 86pack_next (void)
194{ 87{
195 unsigned int d = u_32 (pack_cur->datalen); 88 unsigned int d = u_32 (pack_cur->datalen);
196 89
197 pack_cur = (struct hdr *)(PACK_DATA + d); 90 pack_cur = (struct u_pack_hdr *)(PACK_DATA + d);
198} 91}
199 92
200static void 93static void
201pack_unmap (void) 94pack_unmap (void)
202{ 95{
203 if (!pack_base) 96 if (pack_base)
204 return; 97 {
205
206 free (scratch);
207
208#ifdef _WIN32
209 UnmapViewOfFile (pack_base);
210#else
211 munmap (pack_base, pack_end - pack_base); 98 u_munmap (pack_base, pack_end - pack_base);
212#endif 99 pack_base = 0;
100 }
101
102 if (scratch)
103 {
104 u_free (scratch, scratch_size);
105 scratch = 0;
106 }
213} 107}
214 108
215static int 109static int
216pack_map (void) 110pack_map (void)
217{ 111{
112 char *addr;
113 unsigned int size;
114
218#ifdef _WIN32 115#ifdef _WIN32
219 BY_HANDLE_FILE_INFORMATION fi; 116 BY_HANDLE_FILE_INFORMATION fi;
220 117
221 if (!GetFileInformationByHandle (pack_handle, &fi)) 118 if (!GetFileInformationByHandle (pack_handle, &fi))
222 return 0; 119 return 0;
223 120
224 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)
225 return 0; 128 return 0;
226 129
227 HANDLE handle = CreateFileMapping (pack_handle, 0, PAGE_READONLY, 0, fi.nFileSizeLow, NULL);
228
229 if (!handle)
230 return 0;
231
232 pack_unmap (); 130 pack_unmap ();
233 131
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; 132 pack_base = addr;
247
248 pack_end = pack_base + len; 133 pack_end = pack_base + size;
249#endif
250 134
251 if (!pack_base)
252 return 0;
253
254 struct tail *tail; 135 struct u_pack_tail *tail;
255 136
256 tail = (void *)(pack_end - sizeof (*tail)); 137 tail = (void *)(pack_end - sizeof (*tail));
257 138
258 if (memcmp (tail->magic, TAIL_MAGIC, sizeof (TAIL_MAGIC) - 1)) 139 if (memcmp (tail->magic, TAIL_MAGIC, sizeof (TAIL_MAGIC) - 1))
259 return 0; 140 return 0;
260 141
261 pack_cur = (struct hdr *)(pack_end - u_32 (tail->size)); 142 pack_cur = (struct u_pack_hdr *)(pack_end - u_32 (tail->size));
262 143
263 free (scratch);
264 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;
265 147
266 return 1; 148 return 1;
267} 149}
268 150
269static void 151static void
270exe_info (void) 152exe_info (void)
271{ 153{
272 if (!pack_map ()) 154 if (!pack_map ())
273 fatal ("unable to locate packfile in executable - executable corrupted?"); 155 u_fatal ("unable to locate packfile in executable - executable corrupted?");
274 156
275 if (pack_cur->type != T_META) 157 if (pack_cur->type != T_META)
276 fatal ("unable to locate executable metadata - executable corrupted?"); 158 u_fatal ("unable to locate executable metadata - executable corrupted?");
277 159
278 strcpy (exe_id, PACK_NAME); 160 strcpy (exe_id, PACK_NAME);
279} 161}
280 162
281static void 163static void
282load (void) 164load (void)
283{ 165{
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)) 166 if (u_chdir (exe_dir))
291 fatal ("unable to change to application data directory"); 167 u_fatal ("unable to change to application data directory");
292 168
293 u_handle h = u_open ("override"); 169 u_handle h = u_open ("override");
294 if (h) 170 if (u_valid (h))
295 { 171 {
296 u_handle oh = pack_handle; 172 u_handle oh = pack_handle;
297 173
298 pack_handle = h; 174 pack_handle = h;
299 if (!pack_map ()) 175 if (pack_map ())
176 u_setenv ("URLADER_OVERRIDE", "override");
177 else
300 { 178 {
301 pack_handle = oh; 179 pack_handle = oh;
302 oh = h; 180 oh = h;
303 } 181 }
304 182
305 u_close (oh); 183 u_close (oh);
306 } 184 }
307 185
308 if (pack_cur->type != T_META) 186 if (pack_cur->type != T_META)
309 fatal ("unable to locate override metadata"); 187 u_fatal ("unable to locate override metadata");
310 188
311 strcpy (exe_ver, PACK_DATA); 189 strcpy (exe_ver, PACK_DATA);
190 u_set_exe_info ();
312 pack_next (); 191 pack_next ();
313 192
314 for (;;) 193 for (;;)
315 { 194 {
316 if (pack_cur->type == T_ENV) 195 if (pack_cur->type == T_ENV)
317 u_setenv (PACK_NAME, PACK_DATA); 196 u_setenv (PACK_NAME, PACK_DATA);
318 else if (pack_cur->type == T_ARG) 197 else if (pack_cur->type == T_ARG)
319 exe_argv [exe_argc++] = strdup (PACK_NAME); 198 {
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 }
320 else 204 else
321 break; 205 break;
322 206
323 pack_next (); 207 pack_next ();
324 } 208 }
325 209
326done_env_arg: 210done_env_arg:
327 strcpy (execdir, exe_dir); 211
328 u_append (execdir, "i-"); 212 strcat (strcpy (tmppath, execdir), ".lck");
329 strcat (execdir, exe_ver); 213 lock_handle = u_lock (tmppath, 0, 1);
214 if (!lock_handle)
215 u_fatal ("unable to lock application instance");
330 216
331 if (access (execdir, F_OK)) 217 if (access (execdir, F_OK))
332 { 218 {
333 // does not exist yet, so unpack and move 219 // does not exist yet, so unpack and move
334 tmpdir (exe_dir); 220 tmpdir (exe_dir);
335 221
336 if (u_chdir (tmppath)) 222 if (u_chdir (tmppath))
337 fatal ("unable to change to new instance directory"); 223 u_fatal ("unable to change to new instance directory");
338 224
339 for (;;) 225 for (;;)
340 { 226 {
341 switch (pack_cur->type) 227 switch (pack_cur->type)
342 { 228 {
… …
349 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);
350 unsigned int dlen, len = u_32 (pack_cur->datalen); 236 unsigned int dlen, len = u_32 (pack_cur->datalen);
351 char *data = PACK_DATA; 237 char *data = PACK_DATA;
352 238
353 if (pack_cur->flags & F_LZF) 239 if (pack_cur->flags & F_LZF)
354 if (dlen = lzf_decompress (data, len, scratch, scratch_len)) 240 if (dlen = lzf_decompress (data, len, scratch, scratch_size))
355 { 241 {
356 data = scratch; 242 data = scratch;
357 len = dlen; 243 len = dlen;
358 } 244 }
359 else 245 else
360 fatal ("unable to uncompress file data - pack corrupted?"); 246 u_fatal ("unable to uncompress file data - pack corrupted?");
361 247
362 if (!h) 248 if (!u_valid (h))
363 fatal ("unable to unpack file from packfile - disk full?"); 249 u_fatal ("unable to unpack file from packfile - disk full?");
364 250
365 if (u_write (h, data, len) != len) 251 if (u_write (h, data, len) != len)
366 fatal ("unable to unpack file from packfile - disk full?"); 252 u_fatal ("unable to unpack file from packfile - disk full?");
367 253
368 u_fsync (h); 254 u_fsync (h);
369 u_close (h); 255 u_close (h);
370 } 256 }
371 break; 257 break;
… …
377 pack_next (); 263 pack_next ();
378 } 264 }
379 265
380done: 266done:
381 if (u_chdir (datadir)) 267 if (u_chdir (datadir))
382 fatal ("unable to change to data directory"); 268 u_fatal ("unable to change to data directory");
383 269
384 u_sync (); 270 u_sync ();
385 271
386 if (u_rename (tmppath, execdir)) 272 if (u_rename (tmppath, execdir))
387 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
… …
389 275
390 pack_unmap (); 276 pack_unmap ();
391 u_close (pack_handle); 277 u_close (pack_handle);
392 278
393 if (u_chdir (execdir)) 279 if (u_chdir (execdir))
394 fatal ("unable to change to application instance directory"); 280 u_fatal ("unable to change to application instance directory");
395 281
396 u_setenv ("URLADER_VERSION", URLADER_VERSION); 282 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 283
284#if 0
403 // yes, this is overkill 285 // yes, this is overkill
404 u_setenv ("SHLIB_PATH" , execdir); // hpux 286 u_setenv ("SHLIB_PATH" , execdir); // hpux
405 u_setenv ("LIBPATH" , execdir); // aix 287 u_setenv ("LIBPATH" , execdir); // aix
406 u_setenv ("LD_LIBRARY_PATH" , execdir); // most elf systems 288 u_setenv ("LD_LIBRARY_PATH" , execdir); // most elf systems
407 u_setenv ("LD_LIBRARY_PATH_32", execdir); // solaris 289 u_setenv ("LD_LIBRARY_PATH_32", execdir); // solaris
408 u_setenv ("LD_LIBRARY_PATH_64", execdir); // solaris 290 u_setenv ("LD_LIBRARY_PATH_64", execdir); // solaris
409 u_setenv ("LD_LIBRARYN32_PATH", execdir); // irix 291 u_setenv ("LD_LIBRARYN32_PATH", execdir); // irix
410 u_setenv ("LD_LIBRARY64_PATH" , execdir); // irix 292 u_setenv ("LD_LIBRARY64_PATH" , execdir); // irix
411 u_setenv ("DYLD_LIBRARY_PATH" , execdir); // os sucks from apple 293 u_setenv ("DYLD_LIBRARY_PATH" , execdir); // os sucks from apple
294#endif
412} 295}
413 296
414static void 297static void
415execute (void) 298execute (void)
416{ 299{
417 exe_argv [exe_argc] = 0; 300 exe_argv [exe_argc] = 0;
418 systemv (exe_argv, 0); 301 systemv (exe_argv);
302}
303
304static void
305doit (void)
306{
307 u_set_datadir ();
308 u_mkdir (datadir);
309
310 exe_info ();
311 u_set_exe_info ();
312
313 load ();
314 execute ();
419} 315}
420 316
421#ifdef _WIN32 317#ifdef _WIN32
422 318
423int APIENTRY 319int APIENTRY
424WinMain (HINSTANCE hI, HINSTANCE hP, LPSTR argv, int command_show) 320WinMain (HINSTANCE hI, HINSTANCE hP, LPSTR argv, int command_show)
425{ 321{
426 if (!GetModuleFileName (hI, tmppath, sizeof (tmppath))) 322 if (!GetModuleFileName (hI, tmppath, sizeof (tmppath)))
427 fatal ("unable to find executable pack"); 323 u_fatal ("unable to find executable pack");
428 324
429 if (!(pack_handle = u_open (tmppath))) 325 pack_handle = u_open (tmppath);
326 if (!u_valid (pack_handle))
430 fatal ("unable to open executable pack"); 327 u_fatal ("unable to open executable pack");
431
432 exe_info ();
433 328
434 if (!GetCurrentDirectory (sizeof (currdir), currdir)) 329 if (!GetCurrentDirectory (sizeof (currdir), currdir))
435 strcpy (currdir, "."); 330 strcpy (currdir, ".");
436 331
437 if (SHGetFolderPath (0, CSIDL_APPDATA | CSIDL_FLAG_CREATE, NULL, SHGFP_TYPE_CURRENT, datadir) != S_OK) 332 doit ();
438 fatal ("unable to find application data directory");
439
440 u_mkdir (datadir);
441 u_append (datadir, URLADER);
442
443 load ();
444 execute ();
445 333
446 return 0; 334 return 0;
447} 335}
448 336
449#else 337#else
450 338
451int 339int
452main (int argc, char *argv[]) 340main (int argc, char *argv[])
453{ 341{
454 char *home = getenv ("HOME");
455
456 // we assume fd 0 and 2 are "normal"
457
458 if (!(pack_handle = u_open (argv [0]))) 342 pack_handle = u_open (argv [0]);
343 if (!u_valid (pack_handle))
459 fatal ("unable to open executable pack"); 344 u_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 345
473 if (!getcwd (currdir, sizeof (currdir))) 346 if (!getcwd (currdir, sizeof (currdir)))
474 strcpy (currdir, "."); 347 strcpy (currdir, ".");
475 348
476 u_mkdir (home); 349#if 0
477 sprintf (datadir, "%s/.%s", home, URLADER); 350 /* intersperse hostname, for whatever reason */
478 u_mkdir (datadir);
479 351
480 if (gethostname (tmppath, sizeof (tmppath))) 352 if (gethostname (tmppath, sizeof (tmppath)))
481 strcpy (tmppath, "default"); 353 strcpy (tmppath, "default");
482 354
483 u_append (datadir, tmppath); 355 u_append (datadir, tmppath);
356#endif
484 357
485 load (); 358 doit ();
486 execute ();
487 359
488 return 0; 360 return 0;
489} 361}
490 362
491#endif 363#endif

Diff Legend

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