Added prototypes for the Remote Shutdown Api calls.
[wine/hacks.git] / tools / winewrap.c
blob1b6b26e3fc91312c565a721cb98268880ff8032f
1 /*
2 * Wine wrapper: takes care of internal details for linking.
4 * Copyright 2000 Francois Gouget
5 * Copyright 2002 Dimitrie O. Paun
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with this library; if not, write to the Free Software
19 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
22 #include "config.h"
23 #include "wine/port.h"
25 #include <stdio.h>
26 #include <string.h>
27 #include <stdarg.h>
28 #include <stdlib.h>
29 #include <errno.h>
30 #ifdef HAVE_SYS_WAIT_H
31 #include <sys/wait.h>
32 #endif
33 #include <sys/stat.h>
34 #ifdef HAVE_UNISTD_H
35 # include <unistd.h>
36 #endif
38 #ifndef WINEDLLS
39 #define WINEDLLS "/usr/local/lib/wine"
40 #endif
42 static const char *app_loader_script =
43 "#!/bin/sh\n"
44 "\n"
45 "appname=\"%s\"\n"
46 "# determine the application directory\n"
47 "appdir=''\n"
48 "case \"$0\" in\n"
49 " */*)\n"
50 " # $0 contains a path, use it\n"
51 " appdir=`dirname \"$0\"`\n"
52 " ;;\n"
53 " *)\n"
54 " # no directory in $0, search in PATH\n"
55 " saved_ifs=$IFS\n"
56 " IFS=:\n"
57 " for d in $PATH\n"
58 " do\n"
59 " IFS=$saved_ifs\n"
60 " if [ -x \"$d/$appname\" ]; then appdir=\"$d\"; break; fi\n"
61 " done\n"
62 " ;;\n"
63 "esac\n"
64 "\n"
65 "while true; do\n"
66 " case \"$1\" in\n"
67 " --debugmsg)\n"
68 " debugmsg=\"$1 $2\"\n"
69 " shift; shift;\n"
70 " ;;\n"
71 " --dll)\n"
72 " dll=\"$1 $2\"\n"
73 " shift; shift;\n"
74 " ;;\n"
75 " *)\n"
76 " break\n"
77 " ;;\n"
78 " esac\n"
79 "done\n"
80 "\n"
81 "# figure out the full app path\n"
82 "if [ -n \"$appdir\" ]; then\n"
83 " apppath=\"$appdir/$appname.exe.so\"\n"
84 "else\n"
85 " apppath=\"$appname.exe.so\"\n"
86 "fi\n"
87 "\n"
88 "# determine the WINELOADER\n"
89 "if [ ! -x \"$WINELOADER\" ]; then WINELOADER=\"wine\"; fi\n"
90 "\n"
91 "# and try to start the app\n"
92 "exec \"$WINELOADER\" $debugmsg $dll -- \"$apppath\" \"$@\"\n"
95 static const char *app_gui_spec =
96 "@ stdcall WinMain(ptr long ptr long) WinMain\n"
99 static const char *app_cui_spec =
100 "@ stdcall main(long ptr ptr) main\n"
103 static const char *wrapper_code =
104 "/*\n"
105 " * Copyright 2000 Francois Gouget <fgouget@codeweavers.com> for CodeWeavers\n"
106 " * Copyright 2002 Dimitrie O. Paun <dpaun@rogers.com>\n"
107 " */\n"
108 "\n"
109 "#include <dlfcn.h>\n"
110 "#include <windows.h>\n"
111 "\n"
112 "\n"
113 "/*\n"
114 " * Describe the wrapped application\n"
115 " */\n"
116 "\n"
117 "/* The app name */\n"
118 "#define APPNAME \"%s\"\n"
119 "/**\n"
120 " * This is either 0 for a console based application or\n"
121 " * 1 for a regular windows application.\n"
122 " */\n"
123 "#define GUIEXE %d\n"
124 "\n"
125 "/**\n"
126 " * This is the name of the library containing the application,\n"
127 " * e.g. 'hello.dll' if the application is called 'hello.exe'.\n"
128 " */\n"
129 "static char* appName = APPNAME \".dll\";\n"
130 "\n"
131 "/**\n"
132 " * This is the name of the application's Windows module. If left NULL\n"
133 " * then appName is used.\n"
134 " */\n"
135 "static char* appModule = NULL;\n"
136 "\n"
137 "/**\n"
138 " * This is the application's entry point. This is usually 'WinMain' for a\n"
139 " * gui app and 'main' for a console application.\n"
140 " */\n"
141 "#if GUIEXE\n"
142 "static char* appInit = \"WinMain\";\n"
143 "#else\n"
144 "static char* appInit = \"main\";\n"
145 "#endif\n"
146 "\n"
147 "/**\n"
148 " * This is either non-NULL for MFC-based applications and is the name of the\n"
149 " * MFC's module. This is the module in which we will take the 'WinMain'\n"
150 " * function.\n"
151 " */\n"
152 "static char* mfcModule = NULL;\n"
153 "\n"
154 "\n"
155 "void error(const char *format, ...)\n"
156 "{\n"
157 " va_list ap;\n"
158 " char msg[4096];\n"
159 "\n"
160 " va_start(ap, format);\n"
161 " vsnprintf(msg, sizeof(msg), format, ap);\n"
162 " MessageBox(NULL, msg, \"Error\", MB_OK);\n"
163 " va_end(ap);\n"
164 " exit(1);\n"
165 "}\n"
166 "\n"
167 "\n"
168 "#if GUIEXE\n"
169 "typedef int WINAPI (*WinMainFunc)(HINSTANCE hInstance, HINSTANCE hPrevInstance,\n"
170 " PSTR szCmdLine, int iCmdShow);\n"
171 "#else\n"
172 "typedef int WINAPI (*MainFunc)(int argc, char** argv, char** envp);\n"
173 "#endif\n"
174 "\n"
175 "#if GUIEXE\n"
176 "int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance,\n"
177 " PSTR szCmdLine, int iCmdShow)\n"
178 "#else\n"
179 "int WINAPI main(int argc, char** argv, char** envp)\n"
180 "#endif\n"
181 "{\n"
182 " void* appLibrary;\n"
183 " HINSTANCE hApp = 0, hMFC = 0, hMain = 0;\n"
184 " void* appMain;\n"
185 " char* libName;\n"
186 " int retcode;\n"
187 "\n"
188 " /* Load the application's library */\n"
189 " libName=(char*)malloc(2+strlen(appName)+3+1);\n"
190 " /* FIXME: we should get the wrapper's path and use that as the base for\n"
191 " * the library\n"
192 " */\n"
193 " sprintf(libName,\"./%%s.so\",appName);\n"
194 " appLibrary=dlopen(libName,RTLD_NOW);\n"
195 " if (appLibrary==NULL) {\n"
196 " sprintf(libName,\"%%s.so\",appName);\n"
197 " appLibrary=dlopen(libName,RTLD_NOW);\n"
198 " }\n"
199 " if (!appLibrary) error(\"Could not load the %%s library: %%s\", libName, dlerror());\n"
200 "\n"
201 " /* Then if this application is MFC based, load the MFC module */\n"
202 " /* FIXME: I'm not sure this is really necessary */\n"
203 " if (mfcModule) {\n"
204 " hMFC = LoadLibrary(mfcModule);\n"
205 " if (!hMFC) error(\"Could not load the MFC module %%s (%%d)\", mfcModule, GetLastError());\n"
206 " /* MFC is a special case: the WinMain is in the MFC library,\n"
207 " * instead of the application's library.\n"
208 " */\n"
209 " hMain = hMFC;\n"
210 " }\n"
211 "\n"
212 " /* Load the application's module */\n"
213 " if (!appModule) appModule = appName;\n"
214 " hApp = LoadLibrary(appModule);\n"
215 " if (!hApp) error(\"Could not load the application's module %%s (%%d)\", appModule, GetLastError());\n"
216 " if (!hMain) hMain = hApp;\n"
217 "\n"
218 " /* Get the address of the application's entry point */\n"
219 " appMain = GetProcAddress(hMain, appInit);\n"
220 " if (!appMain) error(\"Could not get the address of %%s (%%d)\", appInit, GetLastError());\n"
221 "\n"
222 " /* And finally invoke the application's entry point */\n"
223 "#if GUIEXE\n"
224 " retcode = (*((WinMainFunc)appMain))(hApp, hPrevInstance, szCmdLine, iCmdShow);\n"
225 "#else\n"
226 " retcode = (*((MainFunc)appMain))(argc, argv, envp);\n"
227 "#endif\n"
228 "\n"
229 " /* Cleanup and done */\n"
230 " FreeLibrary(hApp);\n"
231 " FreeLibrary(hMFC);\n"
232 " dlclose(appLibrary);\n"
233 " free(libName);\n"
234 "\n"
235 " return retcode;\n"
236 "}\n"
239 static char *output_name;
240 static char **arh_files, **dll_files, **lib_files, **lib_paths, **obj_files;
241 static int nb_arh_files, nb_dll_files, nb_lib_files, nb_lib_paths, nb_obj_files;
242 static int verbose = 0;
243 static int keep_generated = 0;
245 void error(const char *s, ...)
247 va_list ap;
249 va_start(ap, s);
250 fprintf(stderr, "Error: ");
251 vfprintf(stderr, s, ap);
252 fprintf(stderr, "\n");
253 va_end(ap);
254 exit(2);
257 char *strmake(const char *fmt, ...)
259 int n, size = 100;
260 char *p;
261 va_list ap;
263 if ((p = malloc (size)) == NULL)
264 error("Can not malloc %d bytes.", size);
266 while (1)
268 va_start(ap, fmt);
269 n = vsnprintf (p, size, fmt, ap);
270 va_end(ap);
271 if (n > -1 && n < size) return p;
272 size *= 2;
273 if ((p = realloc (p, size)) == NULL)
274 error("Can not realloc %d bytes.", size);
278 void rm_temp_file(const char *file)
280 if (!keep_generated) unlink(file);
283 void create_file(const char *name, const char *fmt, ...)
285 va_list ap;
286 FILE *file;
288 if (verbose) printf("Creating file %s\n", name);
289 va_start(ap, fmt);
290 if ( !(file = fopen(name, "w")) )
291 error ("Can not create %s.", name);
292 vfprintf(file, fmt, ap);
293 va_end(ap);
294 fclose(file);
297 void spawn(char *const argv[])
299 int pid, status, wret, i;
301 if (verbose)
303 for(i = 0; argv[i]; i++) printf("%s ", argv[i]);
304 printf("\n");
307 if ((pid = fork()) == 0) execvp(argv[0], argv);
308 else if (pid > 0)
310 while (pid != (wret = waitpid(pid, &status, 0)))
311 if (wret == -1 && errno != EINTR) break;
313 if (pid == wret && WIFEXITED(status) && WEXITSTATUS(status) == 0) return;
314 error("%s failed.", argv[0]);
316 perror("Error:");
317 exit(3);
320 int is_resource(const char* file)
322 /* see tools/winebuild/res32.c: check_header for details */
323 static const char res_sig[] = { 0,0,0,0, 32,0,0,0, 0xff,0xff, 0,0, 0xff,0xff, 0,0, 0,0,0,0, 0,0, 0,0, 0,0,0,0, 0,0,0,0 };
324 char buf[sizeof(res_sig)];
325 FILE *fin;
327 if (!(fin = fopen(file, "r"))) error("Can not open %s", file);
328 if (fread(buf, sizeof(buf), 1, fin) != 1) error("Truncated file %s", file);
329 fclose(fin);
331 return !memcmp(buf, res_sig, sizeof(buf));
334 /* open the file for a given name, in a specified path, with the given extension */
335 static char *try_path( const char *path, const char *name, const char *ext )
337 char *fullname;
338 int fd;
340 fullname = strmake("%s/lib%s.%s", path, name, ext);
341 if (verbose > 1) fprintf(stderr, "Try %s...", fullname);
343 /* check if the file exists */
344 if ((fd = open( fullname, O_RDONLY )) != -1)
346 close( fd );
347 if (verbose > 1) fprintf(stderr, "FOUND!\n");
348 return fullname;
350 free( fullname );
351 if (verbose > 1) fprintf(stderr, "\n");
352 return NULL;
355 /* open the .def library for a given dll in a specified path */
356 static char *try_dll_path( const char *path, const char *name )
358 return try_path(path, name, "def");
361 /* open the .a library for a given lib in a specified path */
362 static char *try_lib_path( const char *path, const char *name )
364 return try_path(path, name, "a");
367 /* open the .def library for a given dll */
368 static char *open_dll(const char *name)
370 char *fullname;
371 int i;
373 for (i = 0; i < nb_lib_paths; i++)
375 if ((fullname = try_dll_path( lib_paths[i], name ))) return fullname;
377 return try_dll_path( ".", name );
380 /* find a static library */
381 static char *find_lib(const char *name)
383 static const char* std_paths[] = { ".", "/usr/lib", "/usr/local/lib" };
384 char *fullname;
385 int i;
387 for (i = 0; i < nb_lib_paths; i++)
389 if ((fullname = try_lib_path( lib_paths[i], name ))) return fullname;
392 for (i = 0; i < sizeof(std_paths)/sizeof(std_paths[0]); i++)
394 if ((fullname = try_lib_path( std_paths[i], name ))) return fullname;
397 return 0;
400 void add_lib_path(const char* path)
402 lib_paths = realloc( lib_paths, (nb_lib_paths+1) * sizeof(*lib_paths) );
403 lib_paths[nb_lib_paths++] = strdup(path);
404 dll_files = realloc( dll_files, (nb_dll_files+1) * sizeof(*dll_files) );
405 dll_files[nb_dll_files++] = strmake("-L%s", path);
406 lib_files = realloc( lib_files, (nb_lib_files+1) * sizeof(*lib_files) );
407 lib_files[nb_lib_files++] = strmake("-L%s", path);
410 void add_lib_file(const char* library)
412 char *lib;
414 if (open_dll(library))
416 dll_files = realloc( dll_files, (nb_dll_files+1) * sizeof(*dll_files) );
417 dll_files[nb_dll_files++] = strmake("-l%s", library);
419 else if ((lib = find_lib(library)))
421 arh_files = realloc( arh_files, (nb_arh_files+1) * sizeof(*arh_files) );
422 arh_files[nb_arh_files++] = lib;
424 else
426 lib_files = realloc( lib_files, (nb_lib_files+1) * sizeof(*lib_files) );
427 lib_files[nb_lib_files++] = strmake("-l%s", library);
431 int main(int argc, char **argv)
433 char *library = 0, *path = 0;
434 int i, j, len, cpp = 0, no_opt = 0, gui_mode = 0, create_wrapper = -1;
435 char *base_name, *base_file, *app_temp_name, *wrp_temp_name;
436 char *spec_name, *spec_c_name, *spec_o_name;
437 char *wspec_name, *wspec_c_name, *wspec_o_name;
438 char *wrap_c_name, *wrap_o_name;
439 char **spec_args, **comp_args, **link_args;
440 char **wwrap_args, **wspec_args, **wcomp_args, **wlink_args;
442 /* include the standard DLL path first */
443 add_lib_path(WINEDLLS);
445 for (i = 1; i < argc; i++)
447 if (!no_opt && argv[i][0] == '-')
449 switch (argv[i][1])
451 case 'k':
452 keep_generated = 1;
453 break;
454 case 'o':
455 if (argv[i][2]) output_name = strdup(argv[i]+ 2);
456 else if (i + 1 < argc) output_name = strdup(argv[++i]);
457 else error("The -o switch takes an argument.");
458 break;
459 case 'L':
460 if (argv[i][2]) path = argv[i] + 2;
461 else if (i + 1 < argc) path = argv[++i];
462 else error("The -L switch takes an argument\n.");
463 add_lib_path(path);
464 break;
465 case 'l':
466 if (argv[i][2]) library = argv[i] + 2;
467 else if (i + 1 < argc) library = argv[++i];
468 else error("The -l switch takes an argument\n.");
469 add_lib_file(library);
470 break;
471 case 'm':
472 if (strcmp("-mgui", argv[i]) == 0) gui_mode = 1;
473 else error("Unknown option %s\n", argv[i]);
474 break;
475 case 'v': /* verbose */
476 if (argv[i][2] == 0) verbose++;
477 break;
478 case 'V':
479 printf("winewrap v0.40\n");
480 exit(0);
481 break;
482 case 'C':
483 cpp = 1;
484 break;
485 case 'w':
486 create_wrapper = 1;
487 break;
488 case 'W':
489 create_wrapper = 0;
490 break;
491 case '-':
492 if (argv[i][2]) error("No long option supported.");
493 no_opt = 1;
494 break;
495 default:
496 error("Unknown option -%c", argv[i][1]);
498 continue;
501 /* it's a filename, add it to its list */
502 obj_files = realloc( obj_files, (nb_obj_files+1) * sizeof(*obj_files) );
503 obj_files[nb_obj_files++] = strdup(argv[i]);
506 /* create wrapper only in C++ by default */
507 if (create_wrapper == -1) create_wrapper = cpp;
509 /* link in by default the big three */
510 if (gui_mode) add_lib_file("gdi32");
511 add_lib_file("user32");
512 add_lib_file("kernel32");
514 app_temp_name = tempnam(0, "wapp");
515 wrp_temp_name = tempnam(0, "wwrp");
517 /* get base filename by removing the .exe extension, if present */
518 base_file = strdup(output_name);
519 len = strlen(base_file);
520 if (len > 4 && strcmp(base_file + len - 4, ".exe") == 0)
521 base_file[len - 4 ] = 0; /* remove the .exe extension */
523 /* we need to get rid of the directory part to get the base name */
524 if ( (base_name = strrchr(base_file, '/')) ) base_name++;
525 else base_name = base_file;
527 spec_name = strmake("%s.spec", app_temp_name);
528 spec_c_name = strmake("%s.c", spec_name);
529 spec_o_name = strmake("%s.o", spec_name);
531 wspec_name = strmake("%s.spec", wrp_temp_name);
532 wspec_c_name = strmake("%s.c", wspec_name);
533 wspec_o_name = strmake("%s.o", wspec_name);
535 wrap_c_name = strmake("%s.c", wrp_temp_name);
536 wrap_o_name = strmake("%s.o", wrp_temp_name);
538 /* build winebuild's argument list */
539 spec_args = malloc( (nb_arh_files + nb_dll_files + nb_obj_files + 20) * sizeof (char *) );
540 j = 0;
541 spec_args[j++] = BINDIR "/winebuild";
542 spec_args[j++] = "-o";
543 spec_args[j++] = spec_c_name;
544 if (create_wrapper)
546 spec_args[j++] = "-F";
547 spec_args[j++] = strmake("%s.dll", base_name);
548 spec_args[j++] = "--spec";
549 spec_args[j++] = spec_name;
551 else
553 spec_args[j++] = "--exe";
554 spec_args[j++] = strmake("%s.exe", base_name);
555 spec_args[j++] = gui_mode ? "-mgui" : "-mcui";
557 for (i = 0; i < nb_dll_files; i++)
558 spec_args[j++] = dll_files[i];
559 for (i = 0; i < nb_obj_files; i++)
560 spec_args[j++] = obj_files[i];
561 for (i = 0; i < nb_arh_files; i++)
562 spec_args[j++] = arh_files[i];
563 spec_args[j] = 0;
565 /* build gcc's argument list */
566 comp_args = malloc ( 20 * sizeof (char *) );
567 j = 0;
568 comp_args[j++] = "gcc";
569 comp_args[j++] = "-fPIC";
570 comp_args[j++] = "-o";
571 comp_args[j++] = spec_o_name;
572 comp_args[j++] = "-c";
573 comp_args[j++] = spec_c_name;
574 comp_args[j] = 0;
576 /* build ld's argument list */
577 link_args = malloc( (nb_arh_files + nb_obj_files + nb_lib_files + 20) * sizeof (char *) );
578 j = 0;
579 link_args[j++] = cpp ? "g++" : "gcc";
580 link_args[j++] = "-shared";
581 link_args[j++] = "-Wl,-Bsymbolic,-z,defs";
582 link_args[j++] = "-lwine";
583 link_args[j++] = "-lm";
584 for (i = 0; i < nb_lib_files; i++)
585 link_args[j++] = lib_files[i];
586 link_args[j++] = "-o";
587 link_args[j++] = strmake("%s.%s.so", base_file, create_wrapper ? "dll" : "exe");
588 link_args[j++] = spec_o_name;
589 for (i = 0; i < nb_obj_files; i++)
590 if (!is_resource(obj_files[i])) link_args[j++] = obj_files[i];
591 for (i = 0; i < nb_arh_files; i++)
592 link_args[j++] = arh_files[i];
593 link_args[j] = 0;
595 /* build wrapper compile argument list */
596 wwrap_args = malloc ( 20 * sizeof (char *) );
597 j = 0;
598 wwrap_args[j++] = "gcc";
599 wwrap_args[j++] = "-fPIC";
600 wwrap_args[j++] = "-I" INCLUDEDIR "/windows";
601 wwrap_args[j++] = "-o";
602 wwrap_args[j++] = wrap_o_name;
603 wwrap_args[j++] = "-c";
604 wwrap_args[j++] = wrap_c_name;
605 wwrap_args[j] = 0;
607 /* build wrapper winebuild's argument list */
608 wspec_args = malloc( (nb_dll_files + 20) * sizeof (char *) );
609 j = 0;
610 wspec_args[j++] = BINDIR "/winebuild";
611 wspec_args[j++] = "-o";
612 wspec_args[j++] = wspec_c_name;
613 wspec_args[j++] = "--exe";
614 wspec_args[j++] = strmake("%s.exe", base_name);
615 wspec_args[j++] = gui_mode ? "-mgui" : "-mcui";
616 wspec_args[j++] = wrap_o_name;
617 for (i = 0; i < nb_dll_files; i++)
618 wspec_args[j++] = dll_files[i];
619 wspec_args[j] = 0;
621 /* build wrapper gcc's argument list */
622 wcomp_args = malloc ( 20 * sizeof (char *) );
623 j = 0;
624 wcomp_args[j++] = "gcc";
625 wcomp_args[j++] = "-fPIC";
626 wcomp_args[j++] = "-o";
627 wcomp_args[j++] = wspec_o_name;
628 wcomp_args[j++] = "-c";
629 wcomp_args[j++] = wspec_c_name;
630 wcomp_args[j] = 0;
632 /* build wrapper ld's argument list */
633 wlink_args = malloc( 20 * sizeof (char *) );
634 j = 0;
635 wlink_args[j++] = cpp ? "g++" : "gcc";
636 wlink_args[j++] = "-shared";
637 wlink_args[j++] = "-Wl,-Bsymbolic,-z,defs";
638 wlink_args[j++] = "-lwine";
639 wlink_args[j++] = "-ldl";
640 wlink_args[j++] = "-o";
641 wlink_args[j++] = strmake("%s.exe.so", base_file);
642 wlink_args[j++] = wspec_o_name;
643 wlink_args[j++] = wrap_o_name;
644 wlink_args[j] = 0;
646 /* run winebuild to get the .spec.c file */
647 if (create_wrapper)
649 create_file(spec_name, gui_mode ? app_gui_spec : app_cui_spec);
650 spawn(spec_args);
651 rm_temp_file(spec_name);
652 spawn(comp_args);
653 rm_temp_file(spec_c_name);
654 spawn(link_args);
655 rm_temp_file(spec_o_name);
657 create_file(wrap_c_name, wrapper_code, base_name, gui_mode);
658 spawn(wwrap_args);
659 rm_temp_file(wrap_c_name);
660 spawn(wspec_args);
661 spawn(wcomp_args);
662 rm_temp_file(wspec_c_name);
663 spawn(wlink_args);
664 rm_temp_file(wspec_o_name);
665 rm_temp_file(wrap_o_name);
667 else
669 spawn(spec_args);
670 spawn(comp_args);
671 rm_temp_file(spec_c_name);
672 spawn(link_args);
673 rm_temp_file(spec_o_name);
676 /* create the loader script */
677 create_file(base_file, app_loader_script, base_name);
678 chmod(base_file, S_IRUSR | S_IWUSR | S_IXUSR | S_IRGRP | S_IXGRP | S_IROTH | S_IXOTH);
680 return 0;