ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/Urlader/urlader.c
Revision: 1.1
Committed: Thu Dec 29 00:44:21 2011 UTC (12 years, 4 months ago) by root
Content type: text/plain
Branch: MAIN
Log Message:
*** empty log message ***

File Contents

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