ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/Urlader/urlader.c
Revision: 1.2
Committed: Thu Dec 29 10:20:56 2011 UTC (12 years, 4 months ago) by root
Content type: text/plain
Branch: MAIN
Changes since 1.1: +33 -16 lines
Log Message:
*** empty log message ***

File Contents

# User Rev Content
1 root 1.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 root 1.2 #include "liblzf/lzf_d.c"
15 root 1.1
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 root 1.2 systemv (const char *const argv[], int dowait)
111 root 1.1 {
112     #ifdef _WIN32
113     _spawnv (P_WAIT, argv [0], argv);
114     #else
115 root 1.2 pid_t pid = dowait ? fork () : 0;
116 root 1.1
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 root 1.2 systemv (argv, 1);
142 root 1.1 }
143    
144     enum
145     {
146     T_NULL, // 5
147     T_META, // 1 : exe_id, exe_ver
148     T_ARG, // 2 : arg
149     T_ENV, // 3 : name, value
150     T_DIR, // 4+: path
151     T_FILE, // 4+: path, data
152     T_NUM
153     };
154    
155     enum
156     {
157     F_EXEC = 1,
158     F_LZF = 2,
159     F_NULL = 0
160     };
161    
162     struct hdr
163     {
164     unsigned char type;
165     unsigned char flags;
166     unsigned char namelen[2];
167     unsigned char datalen[4];
168     };
169    
170     struct tail {
171     unsigned char max_filesize[4];
172     unsigned char size[4];
173     char magic[12];
174     };
175    
176     static char *pack_base, *pack_end;
177     static struct hdr *pack_cur;
178 root 1.2 static char *scratch;
179     static unsigned int scratch_len;
180 root 1.1
181     #define PACK_NAME ((char *)(pack_cur + 1))
182     #define PACK_DATA (PACK_NAME + u_16 (pack_cur->namelen) + 1)
183     #define PACK_VALID pack_cur->type
184    
185     static void
186     pack_next (void)
187     {
188     unsigned int d = u_32 (pack_cur->datalen);
189    
190     pack_cur = (struct hdr *)(PACK_DATA + d);
191     }
192    
193     static void
194     pack_unmap (void)
195     {
196     if (!pack_base)
197     return;
198    
199 root 1.2 free (scratch);
200    
201 root 1.1 #ifdef _WIN32
202     UnmapViewOfFile (pack_base);
203     #else
204     munmap (pack_base, pack_end - pack_base);
205     #endif
206     }
207    
208     static int
209     pack_map (void)
210     {
211     #ifdef _WIN32
212     BY_HANDLE_FILE_INFORMATION fi;
213    
214     if (!GetFileInformationByHandle (pack_handle, &fi))
215     return 0;
216    
217     if (fi.nFileSizeHigh) // too big
218     return 0;
219    
220     HANDLE handle = CreateFileMapping (pack_handle, 0, PAGE_READONLY, 0, fi.nFileSizeLow, NULL);
221    
222     if (!handle)
223     return 0;
224    
225     pack_unmap ();
226    
227     pack_base = MapViewOfFile (handle, FILE_MAP_READ, 0, 0, fi.nFileSizeLow);
228    
229     CloseHandle (handle);
230    
231     pack_end = pack_base + fi.nFileSizeLow;
232     #else
233     off_t len = lseek (pack_handle - 1, 0, SEEK_END);
234    
235     pack_unmap ();
236    
237     pack_base = mmap (0, len, PROT_READ, MAP_SHARED, pack_handle - 1, 0);
238     if (pack_base == (void *)-1)
239     pack_base = 0;
240    
241     pack_end = pack_base + len;
242     #endif
243    
244     if (!pack_base)
245     return 0;
246    
247     struct tail *tail;
248    
249     tail = (void *)(pack_end - sizeof (*tail));
250    
251     if (memcmp (tail->magic, TAIL_MAGIC, sizeof (TAIL_MAGIC) - 1))
252     return 0;
253    
254     pack_cur = (struct hdr *)(pack_end - u_32 (tail->size));
255    
256 root 1.2 free (scratch);
257     scratch = malloc (scratch_len = u_32 (tail->max_filesize));
258    
259 root 1.1 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 root 1.2 unsigned int dlen, len = u_32 (pack_cur->datalen);
343     char *data = PACK_DATA;
344    
345     if (pack_cur->flags & F_LZF)
346     if (dlen = lzf_decompress (data, len, scratch, scratch_len))
347     {
348     data = scratch;
349     len = dlen;
350     }
351     else
352     fatal ("unable to uncompress file data - pack corrupted?");
353 root 1.1
354     if (!h)
355     fatal ("unable to unpack file from packfile - disk full?");
356    
357 root 1.2 if (u_write (h, data, len) != len)
358 root 1.1 fatal ("unable to unpack file from packfile - disk full?");
359    
360     u_close (h);
361     }
362     break;
363    
364     case T_NULL:
365     goto done;
366     }
367    
368     pack_next ();
369     }
370    
371     done:
372     if (u_chdir (datadir))
373     fatal ("unable to change to data directory");
374    
375     if (u_rename (tmppath, execdir))
376     deltree (tmppath); // if move fails, delete new, assume other process created it independently
377     }
378    
379     pack_unmap ();
380     u_close (pack_handle);
381    
382 root 1.2 if (u_chdir (execdir))
383     fatal ("unable to change to application instance directory");
384    
385 root 1.1 u_setenv ("URLADER_VERSION", URLADER_VERSION);
386     u_setenv ("URLADER_DATADIR", datadir);
387     u_setenv ("URLADER_EXECDIR", execdir);
388     u_setenv ("URLADER_EXE_ID" , exe_id);
389     u_setenv ("URLADER_EXE_DIR", exe_dir);
390     u_setenv ("URLADER_EXE_VER", exe_ver);
391    
392     // yes, this is overkill
393     u_setenv ("SHLIB_PATH" , execdir); // hpux
394     u_setenv ("LIBPATH" , execdir); // aix
395     u_setenv ("LD_LIBRARY_PATH" , execdir); // most elf systems
396     u_setenv ("LD_LIBRARY_PATH_32", execdir); // solaris
397     u_setenv ("LD_LIBRARY_PATH_64", execdir); // solaris
398     u_setenv ("LD_LIBRARYN32_PATH", execdir); // irix
399     u_setenv ("LD_LIBRARY64_PATH" , execdir); // irix
400     u_setenv ("DYLD_LIBRARY_PATH" , execdir); // os sucks from apple
401     }
402    
403 root 1.2 static void
404     execute (void)
405     {
406     exe_argv [exe_argc] = 0;
407     systemv (exe_argv, 0);
408     }
409    
410 root 1.1 #ifdef _WIN32
411    
412     int APIENTRY
413     WinMain (HINSTANCE hI, HINSTANCE hP, LPSTR argv, int command_show)
414     {
415     if (!GetModuleFileName (hI, tmppath, sizeof (tmppath)))
416     fatal ("unable to find executable pack");
417    
418     if (!(pack_handle = u_open (tmppath)))
419     fatal ("unable to open executable pack");
420    
421     exe_info ();
422    
423     if (!GetCurrentDirectory (sizeof (currdir), currdir))
424     strcpy (currdir, ".");
425    
426     if (SHGetFolderPath (0, CSIDL_APPDATA | CSIDL_FLAG_CREATE, NULL, SHGFP_TYPE_CURRENT, datadir) != S_OK)
427     fatal ("unable to find application data directory");
428    
429     u_mkdir (datadir);
430     u_append (datadir, URLADER);
431    
432     load ();
433     execute ();
434    
435     return 0;
436     }
437    
438     #else
439    
440     int
441     main (int argc, char *argv[])
442     {
443     char *home = getenv ("HOME");
444    
445     // we assume fd 0 and 2 are "normal"
446    
447     if (!(pack_handle = u_open (argv [0])))
448     fatal ("unable to open executable pack");
449    
450     exe_info ();
451    
452     if (!home)
453     {
454     struct passwd *pw;
455    
456     if ((pw = getpwuid (getuid ())))
457     home = pw->pw_dir;
458     else
459     home = "/tmp";
460     }
461    
462     if (!getcwd (currdir, sizeof (currdir)))
463     strcpy (currdir, ".");
464    
465     u_mkdir (home);
466     sprintf (datadir, "%s/.%s", home, URLADER);
467     u_mkdir (datadir);
468    
469     if (gethostname (tmppath, sizeof (tmppath)))
470     strcpy (tmppath, "default");
471    
472     u_append (datadir, tmppath);
473    
474     load ();
475     execute ();
476    
477     return 0;
478     }
479    
480     #endif
481