?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
winex11.drv: Map coordinates before calling send_mouse_input.
[wine/zf.git] / tools / winebuild / spec32.c
blob02029056b302e02ce647bc458ed416aa8e932fa5
1 /*
2 * 32-bit spec files
3 *
4 * Copyright 1993 Robert J. Amstadt
5 * Copyright 1995 Martin von Loewis
6 * Copyright 1995, 1996, 1997 Alexandre Julliard
7 * Copyright 1997 Eric Youngdale
8 * Copyright 1999 Ulrich Weigand
9 *
10 * This library is free software; you can redistribute it and/or
11 * modify it under the terms of the GNU Lesser General Public
12 * License as published by the Free Software Foundation; either
13 * version 2.1 of the License, or (at your option) any later version.
14 *
15 * This library is distributed in the hope that it will be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18 * Lesser General Public License for more details.
19 *
20 * You should have received a copy of the GNU Lesser General Public
21 * License along with this library; if not, write to the Free Software
22 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
23 */
24
25 #include "config.h"
26 #include "wine/port.h"
27
28 #include <assert.h>
29 #include <ctype.h>
30 #include <stdarg.h>
31 #include <string.h>
32
33 #include "build.h"
34
35 #define IMAGE_FILE_MACHINE_UNKNOWN 0
36 #define IMAGE_FILE_MACHINE_I386 0x014c
37 #define IMAGE_FILE_MACHINE_POWERPC 0x01f0
38 #define IMAGE_FILE_MACHINE_AMD64 0x8664
39 #define IMAGE_FILE_MACHINE_ARMNT 0x01C4
40 #define IMAGE_FILE_MACHINE_ARM64 0xaa64
41
42 #define IMAGE_SIZEOF_NT_OPTIONAL32_HEADER 224
43 #define IMAGE_SIZEOF_NT_OPTIONAL64_HEADER 240
44
45 #define IMAGE_NT_OPTIONAL_HDR32_MAGIC 0x10b
46 #define IMAGE_NT_OPTIONAL_HDR64_MAGIC 0x20b
47 #define IMAGE_ROM_OPTIONAL_HDR_MAGIC 0x107
48
49 int needs_get_pc_thunk = 0;
50
51 static const char builtin_signature[32] = "Wine builtin DLL";
52 static const char fakedll_signature[32] = "Wine placeholder DLL";
53 static struct strarray spec_extra_ld_symbols = { 0 }; /* list of extra symbols that ld should resolve */
54
55 /* add a symbol to the list of extra symbols that ld must resolve */
56 void add_spec_extra_ld_symbol( const char *name )
57 {
58 strarray_add( &spec_extra_ld_symbols, name, NULL );
59 }
60
61 static unsigned int hash_filename( const char *name )
62 {
63 /* FNV-1 hash */
64 unsigned int ret = 2166136261u;
65 while (*name) ret = (ret * 16777619) ^ *name++;
66 return ret;
67 }
68
69 /* check if entry point needs a relay thunk */
70 static inline int needs_relay( const ORDDEF *odp )
71 {
72 /* skip nonexistent entry points */
73 if (!odp) return 0;
74 /* skip non-functions */
75 switch (odp->type)
76 {
77 case TYPE_STDCALL:
78 case TYPE_CDECL:
79 break;
80 case TYPE_STUB:
81 if (odp->u.func.nb_args != -1) break;
82 /* fall through */
83 default:
84 return 0;
85 }
86 /* skip norelay and forward entry points */
87 if (odp->flags & (FLAG_NORELAY|FLAG_FORWARD)) return 0;
88 return 1;
89 }
90
91 static int is_float_arg( const ORDDEF *odp, int arg )
92 {
93 if (arg >= odp->u.func.nb_args) return 0;
94 return (odp->u.func.args[arg] == ARG_FLOAT || odp->u.func.args[arg] == ARG_DOUBLE);
95 }
96
97 /* check if dll will output relay thunks */
98 static int has_relays( DLLSPEC *spec )
99 {
100 int i;
101
102 if (target_cpu != CPU_x86 && target_cpu != CPU_x86_64 &&
103 target_cpu != CPU_ARM && target_cpu != CPU_ARM64)
104 return 0;
105
106 for (i = spec->base; i <= spec->limit; i++)
107 {
108 ORDDEF *odp = spec->ordinals[i];
109 if (needs_relay( odp )) return 1;
110 }
111 return 0;
112 }
113
114 static int get_exports_count( DLLSPEC *spec )
115 {
116 if (unix_lib) return 0;
117 if (spec->base > spec->limit) return 0;
118 return spec->limit - spec->base + 1;
119 }
120
121 static int cmp_func_args( const void *p1, const void *p2 )
122 {
123 const ORDDEF *odp1 = *(const ORDDEF **)p1;
124 const ORDDEF *odp2 = *(const ORDDEF **)p2;
125
126 return odp2->u.func.nb_args - odp1->u.func.nb_args;
127 }
128
129 static void get_arg_string( ORDDEF *odp, char str[MAX_ARGUMENTS + 1] )
130 {
131 int i;
132
133 for (i = 0; i < odp->u.func.nb_args; i++)
134 {
135 switch (odp->u.func.args[i])
136 {
137 case ARG_STR: str[i] = 's'; break;
138 case ARG_WSTR: str[i] = 'w'; break;
139 case ARG_FLOAT: str[i] = 'f'; break;
140 case ARG_DOUBLE: str[i] = 'd'; break;
141 case ARG_INT64:
142 case ARG_INT128:
143 if (get_ptr_size() == 4)
144 {
145 str[i] = (odp->u.func.args[i] == ARG_INT64) ? 'j' : 'k';
146 break;
147 }
148 /* fall through */
149 case ARG_LONG:
150 case ARG_PTR:
151 default:
152 str[i] = 'i';
153 break;
154 }
155 }
156 if (odp->flags & (FLAG_THISCALL | FLAG_FASTCALL)) str[0] = 't';
157 if ((odp->flags & FLAG_FASTCALL) && odp->u.func.nb_args > 1) str[1] = 't';
158
159 /* append return value */
160 if (get_ptr_size() == 4 && (odp->flags & FLAG_RET64))
161 strcpy( str + i, "J" );
162 else
163 strcpy( str + i, "I" );
164 }
165
166 static void output_data_directories( const char *names[16] )
167 {
168 int i;
169
170 for (i = 0; i < 16; i++)
171 {
172 if (names[i])
173 {
174 output_rva( "%s", names[i] );
175 output( "\t.long %s_end - %s\n", names[i], names[i] );
176 }
177 else output( "\t.long 0,0\n" );
178 }
179 }
180
181 /*******************************************************************
182 * build_args_string
183 */
184 static char *build_args_string( DLLSPEC *spec )
185 {
186 int i, count = 0, len = 1;
187 char *p, *buffer;
188 char str[MAX_ARGUMENTS + 2];
189 ORDDEF **funcs;
190
191 funcs = xmalloc( (spec->limit + 1 - spec->base) * sizeof(*funcs) );
192 for (i = spec->base; i <= spec->limit; i++)
193 {
194 ORDDEF *odp = spec->ordinals[i];
195
196 if (!needs_relay( odp )) continue;
197 funcs[count++] = odp;
198 len += odp->u.func.nb_args + 1;
199 }
200 /* sort functions by decreasing number of arguments */
201 qsort( funcs, count, sizeof(*funcs), cmp_func_args );
202 buffer = xmalloc( len );
203 buffer[0] = 0;
204 /* build the arguments string, reusing substrings where possible */
205 for (i = 0; i < count; i++)
206 {
207 get_arg_string( funcs[i], str );
208 if (!(p = strstr( buffer, str )))
209 {
210 p = buffer + strlen( buffer );
211 strcpy( p, str );
212 }
213 funcs[i]->u.func.args_str_offset = p - buffer;
214 }
215 free( funcs );
216 return buffer;
217 }
218
219 /*******************************************************************
220 * output_relay_debug
221 *
222 * Output entry points for relay debugging
223 */
224 static void output_relay_debug( DLLSPEC *spec )
225 {
226 int i;
227
228 /* first the table of entry point offsets */
229
230 output( "\t%s\n", get_asm_rodata_section() );
231 output( "\t.align %d\n", get_alignment(4) );
232 output( ".L__wine_spec_relay_entry_point_offsets:\n" );
233
234 for (i = spec->base; i <= spec->limit; i++)
235 {
236 ORDDEF *odp = spec->ordinals[i];
237
238 if (needs_relay( odp ))
239 output( "\t.long __wine_spec_relay_entry_point_%d-__wine_spec_relay_entry_points\n", i );
240 else
241 output( "\t.long 0\n" );
242 }
243
244 /* then the strings of argument types */
245
246 output( ".L__wine_spec_relay_args_string:\n" );
247 output( "\t%s \"%s\"\n", get_asm_string_keyword(), build_args_string( spec ));
248
249 /* then the relay thunks */
250
251 output( "\t.text\n" );
252 output( "__wine_spec_relay_entry_points:\n" );
253 output( "\tnop\n" ); /* to avoid 0 offset */
254
255 for (i = spec->base; i <= spec->limit; i++)
256 {
257 ORDDEF *odp = spec->ordinals[i];
258
259 if (!needs_relay( odp )) continue;
260
261 switch (target_cpu)
262 {
263 case CPU_x86:
264 output( "\t.align %d\n", get_alignment(4) );
265 output( "\t.long 0x90909090,0x90909090\n" );
266 output( "__wine_spec_relay_entry_point_%d:\n", i );
267 output_cfi( ".cfi_startproc" );
268 output( "\t.byte 0x8b,0xff,0x55,0x8b,0xec,0x5d\n" ); /* hotpatch prolog */
269 if (odp->flags & (FLAG_THISCALL | FLAG_FASTCALL)) /* add the register arguments */
270 {
271 output( "\tpopl %%eax\n" );
272 if ((odp->flags & FLAG_FASTCALL) && get_args_size( odp ) > 4) output( "\tpushl %%edx\n" );
273 output( "\tpushl %%ecx\n" );
274 output( "\tpushl %%eax\n" );
275 }
276 output( "\tpushl $%u\n", (odp->u.func.args_str_offset << 16) | (i - spec->base) );
277 output_cfi( ".cfi_adjust_cfa_offset 4" );
278
279 if (UsePIC)
280 {
281 output( "\tcall %s\n", asm_name("__wine_spec_get_pc_thunk_eax") );
282 output( "1:\tleal .L__wine_spec_relay_descr-1b(%%eax),%%eax\n" );
283 needs_get_pc_thunk = 1;
284 }
285 else output( "\tmovl $.L__wine_spec_relay_descr,%%eax\n" );
286 output( "\tpushl %%eax\n" );
287 output_cfi( ".cfi_adjust_cfa_offset 4" );
288
289 output( "\tcall *4(%%eax)\n" );
290 output_cfi( ".cfi_adjust_cfa_offset -8" );
291 if (odp->type == TYPE_STDCALL)
292 output( "\tret $%u\n", get_args_size( odp ));
293 else
294 output( "\tret\n" );
295 output_cfi( ".cfi_endproc" );
296 break;
297
298 case CPU_ARM:
299 {
300 int j, has_float = 0;
301
302 if (strcmp( float_abi_option, "soft" ))
303 for (j = 0; j < odp->u.func.nb_args && !has_float; j++)
304 has_float = is_float_arg( odp, j );
305
306 output( "\t.align %d\n", get_alignment(4) );
307 if (thumb_mode) output( "\t.thumb_func\n" );
308 output( "__wine_spec_relay_entry_point_%d:\n", i );
309 output_cfi( ".cfi_startproc" );
310 output( "\tpush {r0-r3}\n" );
311 output( "\tmov r2, SP\n");
312 if (has_float) output( "\tvpush {s0-s15}\n" );
313 output( "\tpush {LR}\n" );
314 output( "\tsub SP, #4\n");
315 output( "\tmov r1,#%u\n", i - spec->base );
316 output( "\tmovt r1,#%u\n", odp->u.func.args_str_offset );
317 if (UsePIC)
318 {
319 output( "\tldr r0, 2f\n");
320 output( "1:\tadd r0, PC\n");
321 }
322 else
323 {
324 output( "\tmovw r0, :lower16:.L__wine_spec_relay_descr\n" );
325 output( "\tmovt r0, :upper16:.L__wine_spec_relay_descr\n" );
326 }
327 output( "\tldr IP, [r0, #4]\n");
328 output( "\tblx IP\n");
329 output( "\tldr IP, [SP, #4]\n" );
330 output( "\tadd SP, #%u\n", 24 + (has_float ? 64 : 0) );
331 output( "\tbx IP\n");
332 if (UsePIC) output( "2:\t.long .L__wine_spec_relay_descr-1b-%u\n", thumb_mode ? 4 : 8 );
333 output_cfi( ".cfi_endproc" );
334 break;
335 }
336
337 case CPU_ARM64:
338 output( "\t.align %d\n", get_alignment(4) );
339 output( "__wine_spec_relay_entry_point_%d:\n", i );
340 output_cfi( ".cfi_startproc" );
341 switch (odp->u.func.nb_args)
342 {
343 default:
344 case 8:
345 case 7: output( "\tstp x6, x7, [SP,#-16]!\n" );
346 /* fall through */
347 case 6:
348 case 5: output( "\tstp x4, x5, [SP,#-16]!\n" );
349 /* fall through */
350 case 4:
351 case 3: output( "\tstp x2, x3, [SP,#-16]!\n" );
352 /* fall through */
353 case 2:
354 case 1: output( "\tstp x0, x1, [SP,#-16]!\n" );
355 /* fall through */
356 case 0: break;
357 }
358 output( "\tmov x2, SP\n");
359 output( "\tstp x29, x30, [SP,#-16]!\n" );
360 output( "\tstp x8, x9, [SP,#-16]!\n" );
361 output( "\tmov w1, #%u\n", odp->u.func.args_str_offset << 16 );
362 if (i - spec->base) output( "\tadd w1, w1, #%u\n", i - spec->base );
363 output( "\tadrp x0, %s\n", arm64_page(".L__wine_spec_relay_descr") );
364 output( "\tadd x0, x0, #%s\n", arm64_pageoff(".L__wine_spec_relay_descr") );
365 output( "\tldr x3, [x0, #8]\n");
366 output( "\tblr x3\n");
367 output( "\tadd SP, SP, #16\n" );
368 output( "\tldp x29, x30, [SP], #16\n" );
369 if (odp->u.func.nb_args)
370 output( "\tadd SP, SP, #%u\n", 8 * ((min(odp->u.func.nb_args, 8) + 1) & ~1) );
371 output( "\tret\n");
372 output_cfi( ".cfi_endproc" );
373 break;
374
375 case CPU_x86_64:
376 output( "\t.align %d\n", get_alignment(4) );
377 output( "\t.long 0x90909090,0x90909090\n" );
378 output( "__wine_spec_relay_entry_point_%d:\n", i );
379 output_cfi( ".cfi_startproc" );
380 switch (odp->u.func.nb_args)
381 {
382 default: output( "\tmovq %%%s,32(%%rsp)\n", is_float_arg( odp, 3 ) ? "xmm3" : "r9" );
383 /* fall through */
384 case 3: output( "\tmovq %%%s,24(%%rsp)\n", is_float_arg( odp, 2 ) ? "xmm2" : "r8" );
385 /* fall through */
386 case 2: output( "\tmovq %%%s,16(%%rsp)\n", is_float_arg( odp, 1 ) ? "xmm1" : "rdx" );
387 /* fall through */
388 case 1: output( "\tmovq %%%s,8(%%rsp)\n", is_float_arg( odp, 0 ) ? "xmm0" : "rcx" );
389 /* fall through */
390 case 0: break;
391 }
392 output( "\tmovl $%u,%%edx\n", (odp->u.func.args_str_offset << 16) | (i - spec->base) );
393 output( "\tleaq .L__wine_spec_relay_descr(%%rip),%%rcx\n" );
394 output( "\tcallq *8(%%rcx)\n" );
395 output( "\tret\n" );
396 output_cfi( ".cfi_endproc" );
397 break;
398
399 default:
400 assert(0);
401 }
402 }
403 }
404
405 /*******************************************************************
406 * output_exports
407 *
408 * Output the export table for a Win32 module.
409 */
410 void output_exports( DLLSPEC *spec )
411 {
412 int i, fwd_size = 0;
413 int needs_imports = 0;
414 int needs_relay = has_relays( spec );
415 int nr_exports = get_exports_count( spec );
416 const char *func_ptr = is_pe() ? ".rva" : get_asm_ptr_keyword();
417 const char *name;
418
419 if (!nr_exports) return;
420
421 output( "\n/* export table */\n\n" );
422 output( "\t%s\n", get_asm_export_section() );
423 output( "\t.align %d\n", get_alignment(4) );
424 output( ".L__wine_spec_exports:\n" );
425
426 /* export directory header */
427
428 output( "\t.long 0\n" ); /* Characteristics */
429 output( "\t.long %u\n", hash_filename(spec->file_name) ); /* TimeDateStamp */
430 output( "\t.long 0\n" ); /* MajorVersion/MinorVersion */
431 output_rva( ".L__wine_spec_exp_names" ); /* Name */
432 output( "\t.long %u\n", spec->base ); /* Base */
433 output( "\t.long %u\n", nr_exports ); /* NumberOfFunctions */
434 output( "\t.long %u\n", spec->nb_names ); /* NumberOfNames */
435 output_rva( ".L__wine_spec_exports_funcs " ); /* AddressOfFunctions */
436 if (spec->nb_names)
437 {
438 output_rva( ".L__wine_spec_exp_name_ptrs" ); /* AddressOfNames */
439 output_rva( ".L__wine_spec_exp_ordinals" ); /* AddressOfNameOrdinals */
440 }
441 else
442 {
443 output( "\t.long 0\n" ); /* AddressOfNames */
444 output( "\t.long 0\n" ); /* AddressOfNameOrdinals */
445 }
446
447 /* output the function pointers */
448
449 output( "\n.L__wine_spec_exports_funcs:\n" );
450 for (i = spec->base; i <= spec->limit; i++)
451 {
452 ORDDEF *odp = spec->ordinals[i];
453 if (!odp) output( "\t%s 0\n", is_pe() ? ".long" : get_asm_ptr_keyword() );
454 else switch(odp->type)
455 {
456 case TYPE_EXTERN:
457 case TYPE_STDCALL:
458 case TYPE_VARARGS:
459 case TYPE_CDECL:
460 if (odp->flags & FLAG_FORWARD)
461 {
462 output( "\t%s .L__wine_spec_forwards+%u\n", func_ptr, fwd_size );
463 fwd_size += strlen(odp->link_name) + 1;
464 }
465 else if ((odp->flags & FLAG_IMPORT) && (target_cpu == CPU_x86 || target_cpu == CPU_x86_64))
466 {
467 name = odp->name ? odp->name : odp->export_name;
468 if (name) output( "\t%s %s_%s\n", func_ptr, asm_name("__wine_spec_imp"), name );
469 else output( "\t%s %s_%u\n", func_ptr, asm_name("__wine_spec_imp"), i );
470 needs_imports = 1;
471 }
472 else if (odp->flags & FLAG_EXT_LINK)
473 {
474 output( "\t%s %s_%s\n", func_ptr, asm_name("__wine_spec_ext_link"), odp->link_name );
475 }
476 else
477 {
478 output( "\t%s %s\n", func_ptr, asm_name( get_link_name( odp )));
479 }
480 break;
481 case TYPE_STUB:
482 output( "\t%s %s\n", func_ptr, asm_name( get_stub_name( odp, spec )) );
483 break;
484 default:
485 assert(0);
486 }
487 }
488
489 if (spec->nb_names)
490 {
491 /* output the function name pointers */
492
493 int namepos = strlen(spec->file_name) + 1;
494
495 output( "\n.L__wine_spec_exp_name_ptrs:\n" );
496 for (i = 0; i < spec->nb_names; i++)
497 {
498 output_rva( ".L__wine_spec_exp_names + %u", namepos );
499 namepos += strlen(spec->names[i]->name) + 1;
500 }
501
502 /* output the function ordinals */
503
504 output( "\n.L__wine_spec_exp_ordinals:\n" );
505 for (i = 0; i < spec->nb_names; i++)
506 {
507 output( "\t.short %d\n", spec->names[i]->ordinal - spec->base );
508 }
509 if (spec->nb_names % 2)
510 {
511 output( "\t.short 0\n" );
512 }
513 }
514
515 if (needs_relay)
516 {
517 output( "\t.long 0xdeb90002\n" ); /* magic */
518 if (is_pe()) output_rva( ".L__wine_spec_relay_descr" );
519 else output( "\t.long 0\n" );
520 }
521
522 /* output the export name strings */
523
524 output( "\n.L__wine_spec_exp_names:\n" );
525 output( "\t%s \"%s\"\n", get_asm_string_keyword(), spec->file_name );
526 for (i = 0; i < spec->nb_names; i++)
527 output( "\t%s \"%s\"\n",
528 get_asm_string_keyword(), spec->names[i]->name );
529
530 /* output forward strings */
531
532 if (fwd_size)
533 {
534 output( "\n.L__wine_spec_forwards:\n" );
535 for (i = spec->base; i <= spec->limit; i++)
536 {
537 ORDDEF *odp = spec->ordinals[i];
538 if (odp && (odp->flags & FLAG_FORWARD))
539 output( "\t%s \"%s\"\n", get_asm_string_keyword(), odp->link_name );
540 }
541 }
542
543 /* output relays */
544
545 if (needs_relay)
546 {
547 if (is_pe())
548 {
549 output( "\t.data\n" );
550 output( "\t.align %d\n", get_alignment(get_ptr_size()) );
551 }
552 else
553 {
554 output( "\t.align %d\n", get_alignment(get_ptr_size()) );
555 output( ".L__wine_spec_exports_end:\n" );
556 }
557
558 output( ".L__wine_spec_relay_descr:\n" );
559 output( "\t%s 0xdeb90002\n", get_asm_ptr_keyword() ); /* magic */
560 output( "\t%s 0\n", get_asm_ptr_keyword() ); /* relay func */
561 output( "\t%s 0\n", get_asm_ptr_keyword() ); /* private data */
562 output( "\t%s __wine_spec_relay_entry_points\n", get_asm_ptr_keyword() );
563 output( "\t%s .L__wine_spec_relay_entry_point_offsets\n", get_asm_ptr_keyword() );
564 output( "\t%s .L__wine_spec_relay_args_string\n", get_asm_ptr_keyword() );
565
566 output_relay_debug( spec );
567 }
568 else if (!is_pe())
569 {
570 output( "\t.align %d\n", get_alignment(get_ptr_size()) );
571 output( ".L__wine_spec_exports_end:\n" );
572 output( "\t%s 0\n", get_asm_ptr_keyword() );
573 }
574
575 /* output import thunks */
576
577 if (!needs_imports) return;
578 output( "\t.text\n" );
579 for (i = spec->base; i <= spec->limit; i++)
580 {
581 ORDDEF *odp = spec->ordinals[i];
582 if (!odp) continue;
583 if (!(odp->flags & FLAG_IMPORT)) continue;
584
585 name = odp->name ? odp->name : odp->export_name;
586
587 output( "\t.align %d\n", get_alignment(4) );
588 output( "\t.long 0x90909090,0x90909090\n" );
589 if (name) output( "%s_%s:\n", asm_name("__wine_spec_imp"), name );
590 else output( "%s_%u:\n", asm_name("__wine_spec_imp"), i );
591 output_cfi( ".cfi_startproc" );
592
593 switch (target_cpu)
594 {
595 case CPU_x86:
596 output( "\t.byte 0x8b,0xff,0x55,0x8b,0xec,0x5d\n" ); /* hotpatch prolog */
597 if (UsePIC)
598 {
599 output( "\tcall %s\n", asm_name("__wine_spec_get_pc_thunk_eax") );
600 output( "1:\tjmp *__imp_%s-1b(%%eax)\n", asm_name( get_link_name( odp )));
601 needs_get_pc_thunk = 1;
602 }
603 else output( "\tjmp *__imp_%s\n", asm_name( get_link_name( odp )));
604 break;
605 case CPU_x86_64:
606 output( "\t.byte 0x48,0x8d,0xa4,0x24,0x00,0x00,0x00,0x00\n" ); /* hotpatch prolog */
607 output( "\tjmp *__imp_%s(%%rip)\n", asm_name( get_link_name( odp )));
608 break;
609 default:
610 assert(0);
611 }
612 output_cfi( ".cfi_endproc" );
613 }
614 }
615
616
617 /*******************************************************************
618 * output_module
619 *
620 * Output the module data.
621 */
622 void output_module( DLLSPEC *spec )
623 {
624 int machine = 0;
625 int i;
626 unsigned int page_size = get_page_size();
627 const char *data_dirs[16] = { NULL };
628
629 /* Reserve some space for the PE header */
630
631 switch (target_platform)
632 {
633 case PLATFORM_MINGW:
634 case PLATFORM_WINDOWS:
635 return; /* nothing to do */
636 case PLATFORM_APPLE:
637 output( "\t.text\n" );
638 output( "\t.align %d\n", get_alignment(page_size) );
639 output( "__wine_spec_pe_header:\n" );
640 output( "\t.space 65536\n" );
641 break;
642 case PLATFORM_SOLARIS:
643 output( "\n\t.section \".text\",\"ax\"\n" );
644 output( "__wine_spec_pe_header:\n" );
645 output( "\t.skip %u\n", 65536 + page_size );
646 break;
647 default:
648 switch(target_cpu)
649 {
650 case CPU_x86:
651 case CPU_x86_64:
652 output( "\n\t.section \".init\",\"ax\"\n" );
653 output( "\tjmp 1f\n" );
654 break;
655 case CPU_ARM:
656 output( "\n\t.section \".text\",\"ax\"\n" );
657 output( "\tb 1f\n" );
658 break;
659 case CPU_ARM64:
660 case CPU_POWERPC:
661 output( "\n\t.section \".init\",\"ax\"\n" );
662 output( "\tb 1f\n" );
663 break;
664 }
665 output( "__wine_spec_pe_header:\n" );
666 output( "\t.skip %u\n", 65536 + page_size );
667 output( "1:\n" );
668 break;
669 }
670
671 /* Output the NT header */
672
673 output( "\n\t.data\n" );
674 output( "\t.align %d\n", get_alignment(get_ptr_size()) );
675 output( "\t.globl %s\n", asm_name("__wine_spec_nt_header") );
676 output( "%s:\n", asm_name("__wine_spec_nt_header") );
677 output( ".L__wine_spec_rva_base:\n" );
678
679 output( "\t.long 0x4550\n" ); /* Signature */
680 switch(target_cpu)
681 {
682 case CPU_x86: machine = IMAGE_FILE_MACHINE_I386; break;
683 case CPU_x86_64: machine = IMAGE_FILE_MACHINE_AMD64; break;
684 case CPU_POWERPC: machine = IMAGE_FILE_MACHINE_POWERPC; break;
685 case CPU_ARM: machine = IMAGE_FILE_MACHINE_ARMNT; break;
686 case CPU_ARM64: machine = IMAGE_FILE_MACHINE_ARM64; break;
687 }
688 output( "\t.short 0x%04x\n", /* Machine */
689 machine );
690 output( "\t.short 0\n" ); /* NumberOfSections */
691 output( "\t.long %u\n", hash_filename(spec->file_name) ); /* TimeDateStamp */
692 output( "\t.long 0\n" ); /* PointerToSymbolTable */
693 output( "\t.long 0\n" ); /* NumberOfSymbols */
694 output( "\t.short %d\n", /* SizeOfOptionalHeader */
695 get_ptr_size() == 8 ? IMAGE_SIZEOF_NT_OPTIONAL64_HEADER : IMAGE_SIZEOF_NT_OPTIONAL32_HEADER );
696 output( "\t.short 0x%04x\n", /* Characteristics */
697 spec->characteristics );
698 output( "\t.short 0x%04x\n", /* Magic */
699 get_ptr_size() == 8 ? IMAGE_NT_OPTIONAL_HDR64_MAGIC : IMAGE_NT_OPTIONAL_HDR32_MAGIC );
700 output( "\t.byte 7\n" ); /* MajorLinkerVersion */
701 output( "\t.byte 10\n" ); /* MinorLinkerVersion */
702 output( "\t.long 0\n" ); /* SizeOfCode */
703 output( "\t.long 0\n" ); /* SizeOfInitializedData */
704 output( "\t.long 0\n" ); /* SizeOfUninitializedData */
705
706 for (i = 0; i < spec_extra_ld_symbols.count; i++)
707 output( "\t.globl %s\n", asm_name(spec_extra_ld_symbols.str[i]) );
708
709 /* note: we expand the AddressOfEntryPoint field on 64-bit by overwriting the BaseOfCode field */
710 output( "\t%s %s\n", /* AddressOfEntryPoint */
711 get_asm_ptr_keyword(), spec->init_func ? asm_name(spec->init_func) : "0" );
712 if (get_ptr_size() == 4)
713 {
714 output( "\t.long 0\n" ); /* BaseOfCode */
715 output( "\t.long 0\n" ); /* BaseOfData */
716 }
717 output( "\t%s __wine_spec_pe_header\n", /* ImageBase */
718 get_asm_ptr_keyword() );
719 output( "\t.long %u\n", page_size ); /* SectionAlignment */
720 output( "\t.long %u\n", page_size ); /* FileAlignment */
721 output( "\t.short 1,0\n" ); /* Major/MinorOperatingSystemVersion */
722 output( "\t.short 0,0\n" ); /* Major/MinorImageVersion */
723 output( "\t.short %u,%u\n", /* Major/MinorSubsystemVersion */
724 spec->subsystem_major, spec->subsystem_minor );
725 output( "\t.long 0\n" ); /* Win32VersionValue */
726 output_rva( "%s", asm_name("_end") ); /* SizeOfImage */
727 output( "\t.long %u\n", page_size ); /* SizeOfHeaders */
728 output( "\t.long 0\n" ); /* CheckSum */
729 output( "\t.short 0x%04x\n", /* Subsystem */
730 spec->subsystem );
731 output( "\t.short 0x%04x\n", /* DllCharacteristics */
732 spec->dll_characteristics );
733 output( "\t%s %u,%u\n", /* SizeOfStackReserve/Commit */
734 get_asm_ptr_keyword(), (spec->stack_size ? spec->stack_size : 1024) * 1024, page_size );
735 output( "\t%s %u,%u\n", /* SizeOfHeapReserve/Commit */
736 get_asm_ptr_keyword(), (spec->heap_size ? spec->heap_size : 1024) * 1024, page_size );
737 output( "\t.long 0\n" ); /* LoaderFlags */
738 output( "\t.long 16\n" ); /* NumberOfRvaAndSizes */
739
740 if (get_exports_count( spec ))
741 data_dirs[0] = ".L__wine_spec_exports"; /* DataDirectory[IMAGE_DIRECTORY_ENTRY_EXPORT] */
742 if (has_imports())
743 data_dirs[1] = ".L__wine_spec_imports"; /* DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT] */
744 if (spec->nb_resources)
745 data_dirs[2] = ".L__wine_spec_resources"; /* DataDirectory[IMAGE_DIRECTORY_ENTRY_RESOURCE] */
746
747 output_data_directories( data_dirs );
748
749 if (target_platform == PLATFORM_APPLE)
750 output( "\t.lcomm %s,4\n", asm_name("_end") );
751 }
752
753
754 /*******************************************************************
755 * output_spec32_file
756 *
757 * Build a Win32 C file from a spec file.
758 */
759 void output_spec32_file( DLLSPEC *spec )
760 {
761 needs_get_pc_thunk = 0;
762 open_output_file();
763 output_standard_file_header();
764 output_module( spec );
765 output_stubs( spec );
766 output_exports( spec );
767 output_imports( spec );
768 output_syscalls( spec );
769 if (needs_get_pc_thunk) output_get_pc_thunk();
770 output_resources( spec );
771 output_gnu_stack_note();
772 close_output_file();
773 }
774
775
776 /*******************************************************************
777 * output_fake_module
778 *
779 * Build a fake binary module from a spec file.
780 */
781 void output_fake_module( DLLSPEC *spec )
782 {
783 static const unsigned char dll_code_section[] = { 0x31, 0xc0, /* xor %eax,%eax */
784 0xc2, 0x0c, 0x00 }; /* ret $12 */
785
786 static const unsigned char exe_code_section[] = { 0xb8, 0x01, 0x00, 0x00, 0x00, /* movl $1,%eax */
787 0xc2, 0x04, 0x00 }; /* ret $4 */
788
789 const unsigned int page_size = get_page_size();
790 const unsigned int section_align = page_size;
791 const unsigned int file_align = 0x200;
792 const unsigned int reloc_size = 8;
793 const unsigned int lfanew = 0x40 + sizeof(fakedll_signature);
794 const unsigned int nb_sections = 2 + (spec->nb_resources != 0);
795 const unsigned int text_size = (spec->characteristics & IMAGE_FILE_DLL) ?
796 sizeof(dll_code_section) : sizeof(exe_code_section);
797 unsigned char *resources;
798 unsigned int resources_size;
799 unsigned int image_size = 3 * section_align;
800
801 resolve_imports( spec );
802 output_bin_resources( spec, 3 * section_align );
803 resources = output_buffer;
804 resources_size = output_buffer_pos;
805 if (resources_size) image_size += (resources_size + section_align - 1) & ~(section_align - 1);
806
807 init_output_buffer();
808
809 put_word( 0x5a4d ); /* e_magic */
810 put_word( 0x40 ); /* e_cblp */
811 put_word( 0x01 ); /* e_cp */
812 put_word( 0 ); /* e_crlc */
813 put_word( lfanew / 16 ); /* e_cparhdr */
814 put_word( 0x0000 ); /* e_minalloc */
815 put_word( 0xffff ); /* e_maxalloc */
816 put_word( 0x0000 ); /* e_ss */
817 put_word( 0x00b8 ); /* e_sp */
818 put_word( 0 ); /* e_csum */
819 put_word( 0 ); /* e_ip */
820 put_word( 0 ); /* e_cs */
821 put_word( lfanew ); /* e_lfarlc */
822 put_word( 0 ); /* e_ovno */
823 put_dword( 0 ); /* e_res */
824 put_dword( 0 );
825 put_word( 0 ); /* e_oemid */
826 put_word( 0 ); /* e_oeminfo */
827 put_dword( 0 ); /* e_res2 */
828 put_dword( 0 );
829 put_dword( 0 );
830 put_dword( 0 );
831 put_dword( 0 );
832 put_dword( lfanew );
833
834 put_data( fakedll_signature, sizeof(fakedll_signature) );
835
836 put_dword( 0x4550 ); /* Signature */
837 switch(target_cpu)
838 {
839 case CPU_x86: put_word( IMAGE_FILE_MACHINE_I386 ); break;
840 case CPU_x86_64: put_word( IMAGE_FILE_MACHINE_AMD64 ); break;
841 case CPU_POWERPC: put_word( IMAGE_FILE_MACHINE_POWERPC ); break;
842 case CPU_ARM: put_word( IMAGE_FILE_MACHINE_ARMNT ); break;
843 case CPU_ARM64: put_word( IMAGE_FILE_MACHINE_ARM64 ); break;
844 }
845 put_word( nb_sections ); /* NumberOfSections */
846 put_dword( hash_filename(spec->file_name) ); /* TimeDateStamp */
847 put_dword( 0 ); /* PointerToSymbolTable */
848 put_dword( 0 ); /* NumberOfSymbols */
849 put_word( get_ptr_size() == 8 ?
850 IMAGE_SIZEOF_NT_OPTIONAL64_HEADER :
851 IMAGE_SIZEOF_NT_OPTIONAL32_HEADER ); /* SizeOfOptionalHeader */
852 put_word( spec->characteristics ); /* Characteristics */
853 put_word( get_ptr_size() == 8 ?
854 IMAGE_NT_OPTIONAL_HDR64_MAGIC :
855 IMAGE_NT_OPTIONAL_HDR32_MAGIC ); /* Magic */
856 put_byte( 7 ); /* MajorLinkerVersion */
857 put_byte( 10 ); /* MinorLinkerVersion */
858 put_dword( text_size ); /* SizeOfCode */
859 put_dword( 0 ); /* SizeOfInitializedData */
860 put_dword( 0 ); /* SizeOfUninitializedData */
861 put_dword( section_align ); /* AddressOfEntryPoint */
862 put_dword( section_align ); /* BaseOfCode */
863 if (get_ptr_size() == 4) put_dword( 0 ); /* BaseOfData */
864 put_pword( 0x10000000 ); /* ImageBase */
865 put_dword( section_align ); /* SectionAlignment */
866 put_dword( file_align ); /* FileAlignment */
867 put_word( 1 ); /* MajorOperatingSystemVersion */
868 put_word( 0 ); /* MinorOperatingSystemVersion */
869 put_word( 0 ); /* MajorImageVersion */
870 put_word( 0 ); /* MinorImageVersion */
871 put_word( spec->subsystem_major ); /* MajorSubsystemVersion */
872 put_word( spec->subsystem_minor ); /* MinorSubsystemVersion */
873 put_dword( 0 ); /* Win32VersionValue */
874 put_dword( image_size ); /* SizeOfImage */
875 put_dword( file_align ); /* SizeOfHeaders */
876 put_dword( 0 ); /* CheckSum */
877 put_word( spec->subsystem ); /* Subsystem */
878 put_word( spec->dll_characteristics ); /* DllCharacteristics */
879 put_pword( (spec->stack_size ? spec->stack_size : 1024) * 1024 ); /* SizeOfStackReserve */
880 put_pword( page_size ); /* SizeOfStackCommit */
881 put_pword( (spec->heap_size ? spec->heap_size : 1024) * 1024 ); /* SizeOfHeapReserve */
882 put_pword( page_size ); /* SizeOfHeapCommit */
883 put_dword( 0 ); /* LoaderFlags */
884 put_dword( 16 ); /* NumberOfRvaAndSizes */
885
886 put_dword( 0 ); put_dword( 0 ); /* DataDirectory[IMAGE_DIRECTORY_ENTRY_EXPORT] */
887 put_dword( 0 ); put_dword( 0 ); /* DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT] */
888 if (resources_size) /* DataDirectory[IMAGE_DIRECTORY_ENTRY_RESOURCE] */
889 {
890 put_dword( 3 * section_align );
891 put_dword( resources_size );
892 }
893 else
894 {
895 put_dword( 0 );
896 put_dword( 0 );
897 }
898
899 put_dword( 0 ); put_dword( 0 ); /* DataDirectory[IMAGE_DIRECTORY_ENTRY_EXCEPTION] */
900 put_dword( 0 ); put_dword( 0 ); /* DataDirectory[IMAGE_DIRECTORY_ENTRY_SECURITY] */
901 put_dword( 2 * section_align ); /* DataDirectory[IMAGE_DIRECTORY_ENTRY_BASERELOC] */
902 put_dword( reloc_size );
903 put_dword( 0 ); put_dword( 0 ); /* DataDirectory[IMAGE_DIRECTORY_ENTRY_DEBUG] */
904 put_dword( 0 ); put_dword( 0 ); /* DataDirectory[IMAGE_DIRECTORY_ENTRY_COPYRIGHT] */
905 put_dword( 0 ); put_dword( 0 ); /* DataDirectory[IMAGE_DIRECTORY_ENTRY_GLOBALPTR] */
906 put_dword( 0 ); put_dword( 0 ); /* DataDirectory[IMAGE_DIRECTORY_ENTRY_TLS] */
907 put_dword( 0 ); put_dword( 0 ); /* DataDirectory[IMAGE_DIRECTORY_ENTRY_LOAD_CONFIG] */
908 put_dword( 0 ); put_dword( 0 ); /* DataDirectory[IMAGE_DIRECTORY_ENTRY_BOUND_IMPORT] */
909 put_dword( 0 ); put_dword( 0 ); /* DataDirectory[IMAGE_DIRECTORY_ENTRY_IAT] */
910 put_dword( 0 ); put_dword( 0 ); /* DataDirectory[IMAGE_DIRECTORY_ENTRY_DELAY_IMPORT] */
911 put_dword( 0 ); put_dword( 0 ); /* DataDirectory[IMAGE_DIRECTORY_ENTRY_COM_DESCRIPTOR] */
912 put_dword( 0 ); put_dword( 0 ); /* DataDirectory[15] */
913
914 /* .text section */
915 put_data( ".text\0\0", 8 ); /* Name */
916 put_dword( section_align ); /* VirtualSize */
917 put_dword( section_align ); /* VirtualAddress */
918 put_dword( text_size ); /* SizeOfRawData */
919 put_dword( file_align ); /* PointerToRawData */
920 put_dword( 0 ); /* PointerToRelocations */
921 put_dword( 0 ); /* PointerToLinenumbers */
922 put_word( 0 ); /* NumberOfRelocations */
923 put_word( 0 ); /* NumberOfLinenumbers */
924 put_dword( 0x60000020 /* CNT_CODE|MEM_EXECUTE|MEM_READ */ ); /* Characteristics */
925
926 /* .reloc section */
927 put_data( ".reloc\0", 8 ); /* Name */
928 put_dword( section_align ); /* VirtualSize */
929 put_dword( 2 * section_align );/* VirtualAddress */
930 put_dword( reloc_size ); /* SizeOfRawData */
931 put_dword( 2 * file_align ); /* PointerToRawData */
932 put_dword( 0 ); /* PointerToRelocations */
933 put_dword( 0 ); /* PointerToLinenumbers */
934 put_word( 0 ); /* NumberOfRelocations */
935 put_word( 0 ); /* NumberOfLinenumbers */
936 put_dword( 0x42000040 /* CNT_INITIALIZED_DATA|MEM_DISCARDABLE|MEM_READ */ ); /* Characteristics */
937
938 /* .rsrc section */
939 if (resources_size)
940 {
941 put_data( ".rsrc\0\0", 8 ); /* Name */
942 put_dword( (resources_size + section_align - 1) & ~(section_align - 1) ); /* VirtualSize */
943 put_dword( 3 * section_align );/* VirtualAddress */
944 put_dword( resources_size ); /* SizeOfRawData */
945 put_dword( 3 * file_align ); /* PointerToRawData */
946 put_dword( 0 ); /* PointerToRelocations */
947 put_dword( 0 ); /* PointerToLinenumbers */
948 put_word( 0 ); /* NumberOfRelocations */
949 put_word( 0 ); /* NumberOfLinenumbers */
950 put_dword( 0x40000040 /* CNT_INITIALIZED_DATA|MEM_READ */ ); /* Characteristics */
951 }
952
953 /* .text contents */
954 align_output( file_align );
955 if (spec->characteristics & IMAGE_FILE_DLL)
956 put_data( dll_code_section, sizeof(dll_code_section) );
957 else
958 put_data( exe_code_section, sizeof(exe_code_section) );
959
960 /* .reloc contents */
961 align_output( file_align );
962 put_dword( 0 ); /* VirtualAddress */
963 put_dword( 0 ); /* SizeOfBlock */
964
965 /* .rsrc contents */
966 if (resources_size)
967 {
968 align_output( file_align );
969 put_data( resources, resources_size );
970 }
971 flush_output_buffer();
972 }
973
974
975 /*******************************************************************
976 * output_def_file
977 *
978 * Build a Win32 def file from a spec file.
979 */
980 void output_def_file( DLLSPEC *spec, int import_only )
981 {
982 DLLSPEC *spec32 = NULL;
983 const char *name;
984 int i, total;
985
986 if (spec->type == SPEC_WIN16)
987 {
988 spec32 = alloc_dll_spec();
989 add_16bit_exports( spec32, spec );
990 spec = spec32;
991 }
992
993 if (spec_file_name)
994 output( "; File generated automatically from %s; do not edit!\n\n",
995 spec_file_name );
996 else
997 output( "; File generated automatically; do not edit!\n\n" );
998
999 output( "LIBRARY %s\n\n", spec->file_name);
1000 output( "EXPORTS\n");
1001
1002 /* Output the exports and relay entry points */
1003
1004 for (i = total = 0; i < spec->nb_entry_points; i++)
1005 {
1006 const ORDDEF *odp = &spec->entry_points[i];
1007 int is_data = 0, is_private = odp->flags & FLAG_PRIVATE;
1008
1009 if (odp->name) name = odp->name;
1010 else if (odp->export_name) name = odp->export_name;
1011 else continue;
1012
1013 if (!is_private) total++;
1014 if (import_only && odp->type == TYPE_STUB) continue;
1015
1016 if ((odp->flags & FLAG_FASTCALL) && is_pe())
1017 name = strmake( "@%s", name );
1018
1019 output( " %s", name );
1020
1021 switch(odp->type)
1022 {
1023 case TYPE_EXTERN:
1024 is_data = 1;
1025 /* fall through */
1026 case TYPE_VARARGS:
1027 case TYPE_CDECL:
1028 /* try to reduce output */
1029 if(!import_only && (strcmp(name, odp->link_name) || (odp->flags & FLAG_FORWARD)))
1030 output( "=%s", odp->link_name );
1031 break;
1032 case TYPE_STDCALL:
1033 {
1034 int at_param = get_args_size( odp );
1035 if (!kill_at && target_cpu == CPU_x86) output( "@%d", at_param );
1036 if (import_only) break;
1037 if (odp->flags & FLAG_FORWARD)
1038 output( "=%s", odp->link_name );
1039 else if (strcmp(name, odp->link_name)) /* try to reduce output */
1040 output( "=%s", get_link_name( odp ));
1041 break;
1042 }
1043 case TYPE_STUB:
1044 if (!kill_at && target_cpu == CPU_x86) output( "@%d", get_args_size( odp ));
1045 is_private = 1;
1046 break;
1047 default:
1048 assert(0);
1049 }
1050 output( " @%d", odp->ordinal );
1051 if (!odp->name || (odp->flags & FLAG_ORDINAL)) output( " NONAME" );
1052 if (is_data) output( " DATA" );
1053 if (is_private) output( " PRIVATE" );
1054 output( "\n" );
1055 }
1056 if (!total) warning( "%s: Import library doesn't export anything\n", spec->file_name );
1057 if (spec32) free_dll_spec( spec32 );
1058 }
1059
1060
1061 /*******************************************************************
1062 * make_builtin_files
1063 */
1064 void make_builtin_files( char *argv[] )
1065 {
1066 int i, fd;
1067 struct
1068 {
1069 unsigned short e_magic;
1070 unsigned short unused[29];
1071 unsigned int e_lfanew;
1072 } header;
1073
1074 for (i = 0; argv[i]; i++)
1075 {
1076 if ((fd = open( argv[i], O_RDWR | O_BINARY )) == -1) fatal_perror( "Cannot open %s", argv[i] );
1077 if (read( fd, &header, sizeof(header) ) == sizeof(header) && !memcmp( &header.e_magic, "MZ", 2 ))
1078 {
1079 if (header.e_lfanew < sizeof(header) + sizeof(builtin_signature))
1080 fatal_error( "%s: Not enough space (%x) for Wine signature\n", argv[i], header.e_lfanew );
1081 write( fd, builtin_signature, sizeof(builtin_signature) );
1082
1083 if (prefer_native)
1084 {
1085 unsigned int pos = header.e_lfanew + 0x5e; /* OptionalHeader.DllCharacteristics */
1086 unsigned short dll_charact;
1087 lseek( fd, pos, SEEK_SET );
1088 if (read( fd, &dll_charact, sizeof(dll_charact) ) == sizeof(dll_charact))
1089 {
1090 dll_charact |= IMAGE_DLLCHARACTERISTICS_PREFER_NATIVE;
1091 lseek( fd, pos, SEEK_SET );
1092 write( fd, &dll_charact, sizeof(dll_charact) );
1093 }
1094 }
1095 }
1096 else fatal_error( "%s: Unrecognized file format\n", argv[i] );
1097 close( fd );
1098 }
1099 }
1100
1101 static void fixup_elf32( const char *name, int fd, void *header, size_t header_size )
1102 {
1103 struct
1104 {
1105 unsigned char e_ident[16];
1106 unsigned short e_type;
1107 unsigned short e_machine;
1108 unsigned int e_version;
1109 unsigned int e_entry;
1110 unsigned int e_phoff;
1111 unsigned int e_shoff;
1112 unsigned int e_flags;
1113 unsigned short e_ehsize;
1114 unsigned short e_phentsize;
1115 unsigned short e_phnum;
1116 unsigned short e_shentsize;
1117 unsigned short e_shnum;
1118 unsigned short e_shstrndx;
1119 } *elf = header;
1120 struct
1121 {
1122 unsigned int p_type;
1123 unsigned int p_offset;
1124 unsigned int p_vaddr;
1125 unsigned int p_paddr;
1126 unsigned int p_filesz;
1127 unsigned int p_memsz;
1128 unsigned int p_flags;
1129 unsigned int p_align;
1130 } *phdr;
1131 struct
1132 {
1133 unsigned int d_tag;
1134 unsigned int d_val;
1135 } *dyn;
1136
1137 unsigned int i, size;
1138
1139 if (header_size < sizeof(*elf)) return;
1140 if (elf->e_ident[6] != 1 /* EV_CURRENT */) return;
1141
1142 size = elf->e_phnum * elf->e_phentsize;
1143 phdr = xmalloc( size );
1144 lseek( fd, elf->e_phoff, SEEK_SET );
1145 if (read( fd, phdr, size ) != size) return;
1146
1147 for (i = 0; i < elf->e_phnum; i++)
1148 {
1149 if (phdr->p_type == 2 /* PT_DYNAMIC */ ) break;
1150 phdr = (void *)((char *)phdr + elf->e_phentsize);
1151 }
1152 if (i == elf->e_phnum) return;
1153
1154 dyn = xmalloc( phdr->p_filesz );
1155 lseek( fd, phdr->p_offset, SEEK_SET );
1156 if (read( fd, dyn, phdr->p_filesz ) != phdr->p_filesz) return;
1157 for (i = 0; i < phdr->p_filesz / sizeof(*dyn) && dyn[i].d_tag; i++)
1158 {
1159 switch (dyn[i].d_tag)
1160 {
1161 case 25: dyn[i].d_tag = 0x60009990; break; /* DT_INIT_ARRAY */
1162 case 27: dyn[i].d_tag = 0x60009991; break; /* DT_INIT_ARRAYSZ */
1163 case 12: dyn[i].d_tag = 0x60009992; break; /* DT_INIT */
1164 }
1165 }
1166 lseek( fd, phdr->p_offset, SEEK_SET );
1167 write( fd, dyn, phdr->p_filesz );
1168 }
1169
1170 static void fixup_elf64( const char *name, int fd, void *header, size_t header_size )
1171 {
1172 struct
1173 {
1174 unsigned char e_ident[16];
1175 unsigned short e_type;
1176 unsigned short e_machine;
1177 unsigned int e_version;
1178 unsigned __int64 e_entry;
1179 unsigned __int64 e_phoff;
1180 unsigned __int64 e_shoff;
1181 unsigned int e_flags;
1182 unsigned short e_ehsize;
1183 unsigned short e_phentsize;
1184 unsigned short e_phnum;
1185 unsigned short e_shentsize;
1186 unsigned short e_shnum;
1187 unsigned short e_shstrndx;
1188 } *elf = header;
1189 struct
1190 {
1191 unsigned int p_type;
1192 unsigned int p_flags;
1193 unsigned __int64 p_offset;
1194 unsigned __int64 p_vaddr;
1195 unsigned __int64 p_paddr;
1196 unsigned __int64 p_filesz;
1197 unsigned __int64 p_memsz;
1198 unsigned __int64 p_align;
1199 } *phdr;
1200 struct
1201 {
1202 unsigned __int64 d_tag;
1203 unsigned __int64 d_val;
1204 } *dyn;
1205
1206 unsigned int i, size;
1207
1208 if (header_size < sizeof(*elf)) return;
1209 if (elf->e_ident[6] != 1 /* EV_CURRENT */) return;
1210
1211 size = elf->e_phnum * elf->e_phentsize;
1212 phdr = xmalloc( size );
1213 lseek( fd, elf->e_phoff, SEEK_SET );
1214 if (read( fd, phdr, size ) != size) return;
1215
1216 for (i = 0; i < elf->e_phnum; i++)
1217 {
1218 if (phdr->p_type == 2 /* PT_DYNAMIC */ ) break;
1219 phdr = (void *)((char *)phdr + elf->e_phentsize);
1220 }
1221 if (i == elf->e_phnum) return;
1222
1223 dyn = xmalloc( phdr->p_filesz );
1224 lseek( fd, phdr->p_offset, SEEK_SET );
1225 if (read( fd, dyn, phdr->p_filesz ) != phdr->p_filesz) return;
1226 for (i = 0; i < phdr->p_filesz / sizeof(*dyn) && dyn[i].d_tag; i++)
1227 {
1228 switch (dyn[i].d_tag)
1229 {
1230 case 25: dyn[i].d_tag = 0x60009990; break; /* DT_INIT_ARRAY */
1231 case 27: dyn[i].d_tag = 0x60009991; break; /* DT_INIT_ARRAYSZ */
1232 case 12: dyn[i].d_tag = 0x60009992; break; /* DT_INIT */
1233 }
1234 }
1235 lseek( fd, phdr->p_offset, SEEK_SET );
1236 write( fd, dyn, phdr->p_filesz );
1237 }
1238
1239 /*******************************************************************
1240 * fixup_constructors
1241 */
1242 void fixup_constructors( char *argv[] )
1243 {
1244 int i, fd, size;
1245 unsigned int header[64];
1246
1247 for (i = 0; argv[i]; i++)
1248 {
1249 if ((fd = open( argv[i], O_RDWR | O_BINARY )) == -1) fatal_perror( "Cannot open %s", argv[i] );
1250 size = read( fd, &header, sizeof(header) );
1251 if (size > 5)
1252 {
1253 if (!memcmp( header, "\177ELF\001", 5 )) fixup_elf32( argv[i], fd, header, size );
1254 else if (!memcmp( header, "\177ELF\002", 5 )) fixup_elf64( argv[i], fd, header, size );
1255 }
1256 close( fd );
1257 }
1258 }
Zebediah Figura's local tree