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

Comparing Urlader/urlader.c (file contents):
Revision 1.3 by root, Thu Dec 29 10:39:34 2011 UTC vs.
Revision 1.8 by root, Fri Dec 30 23:34:08 2011 UTC

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

Diff Legend

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