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