ntoskrnl.exe: Stub for KeWaitForSingleObject.
[wine.git] / tools / winebuild / utils.c
blob8d31fff71255cc4cc535e96595e2d87847ca139a
1 /*
2 * Small utility functions for winebuild
4 * Copyright 2000 Alexandre Julliard
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
21 #include "config.h"
22 #include "wine/port.h"
24 #include <assert.h>
25 #include <ctype.h>
26 #include <stdarg.h>
27 #include <stdio.h>
28 #include <stdlib.h>
29 #include <string.h>
30 #ifdef HAVE_UNISTD_H
31 # include <unistd.h>
32 #endif
34 #include "build.h"
36 #define MAX_TMP_FILES 8
37 static const char *tmp_files[MAX_TMP_FILES];
38 static unsigned int nb_tmp_files;
40 static const struct
42 const char *name;
43 enum target_cpu cpu;
44 } cpu_names[] =
46 { "i386", CPU_x86 },
47 { "i486", CPU_x86 },
48 { "i586", CPU_x86 },
49 { "i686", CPU_x86 },
50 { "i786", CPU_x86 },
51 { "x86_64", CPU_x86_64 },
52 { "sparc", CPU_SPARC },
53 { "alpha", CPU_ALPHA },
54 { "powerpc", CPU_POWERPC }
57 /* atexit handler to clean tmp files */
58 static void cleanup_tmp_files(void)
60 unsigned int i;
61 for (i = 0; i < MAX_TMP_FILES; i++) if (tmp_files[i]) unlink( tmp_files[i] );
65 void *xmalloc (size_t size)
67 void *res;
69 res = malloc (size ? size : 1);
70 if (res == NULL)
72 fprintf (stderr, "Virtual memory exhausted.\n");
73 exit (1);
75 return res;
78 void *xrealloc (void *ptr, size_t size)
80 void *res = realloc (ptr, size);
81 if (size && res == NULL)
83 fprintf (stderr, "Virtual memory exhausted.\n");
84 exit (1);
86 return res;
89 char *xstrdup( const char *str )
91 char *res = strdup( str );
92 if (!res)
94 fprintf (stderr, "Virtual memory exhausted.\n");
95 exit (1);
97 return res;
100 char *strupper(char *s)
102 char *p;
103 for (p = s; *p; p++) *p = toupper(*p);
104 return s;
107 int strendswith(const char* str, const char* end)
109 int l = strlen(str);
110 int m = strlen(end);
111 return l >= m && strcmp(str + l - m, end) == 0;
114 void fatal_error( const char *msg, ... )
116 va_list valist;
117 va_start( valist, msg );
118 if (input_file_name)
120 fprintf( stderr, "%s:", input_file_name );
121 if (current_line)
122 fprintf( stderr, "%d:", current_line );
123 fputc( ' ', stderr );
125 else fprintf( stderr, "winebuild: " );
126 vfprintf( stderr, msg, valist );
127 va_end( valist );
128 exit(1);
131 void fatal_perror( const char *msg, ... )
133 va_list valist;
134 va_start( valist, msg );
135 if (input_file_name)
137 fprintf( stderr, "%s:", input_file_name );
138 if (current_line)
139 fprintf( stderr, "%d:", current_line );
140 fputc( ' ', stderr );
142 vfprintf( stderr, msg, valist );
143 perror( " " );
144 va_end( valist );
145 exit(1);
148 void error( const char *msg, ... )
150 va_list valist;
151 va_start( valist, msg );
152 if (input_file_name)
154 fprintf( stderr, "%s:", input_file_name );
155 if (current_line)
156 fprintf( stderr, "%d:", current_line );
157 fputc( ' ', stderr );
159 vfprintf( stderr, msg, valist );
160 va_end( valist );
161 nb_errors++;
164 void warning( const char *msg, ... )
166 va_list valist;
168 if (!display_warnings) return;
169 va_start( valist, msg );
170 if (input_file_name)
172 fprintf( stderr, "%s:", input_file_name );
173 if (current_line)
174 fprintf( stderr, "%d:", current_line );
175 fputc( ' ', stderr );
177 fprintf( stderr, "warning: " );
178 vfprintf( stderr, msg, valist );
179 va_end( valist );
182 int output( const char *format, ... )
184 int ret;
185 va_list valist;
187 va_start( valist, format );
188 ret = vfprintf( output_file, format, valist );
189 va_end( valist );
190 if (ret < 0) fatal_perror( "Output error" );
191 return ret;
194 /* find a build tool in the path, trying the various names */
195 char *find_tool( const char * const *names )
197 static char **dirs;
198 static unsigned int count, maxlen;
200 char *p, *file;
201 unsigned int i, len;
202 struct stat st;
204 if (!dirs)
206 char *path;
208 /* split the path in directories */
210 if (!getenv( "PATH" )) return NULL;
211 path = xstrdup( getenv( "PATH" ));
212 for (p = path, count = 2; *p; p++) if (*p == ':') count++;
213 dirs = xmalloc( count * sizeof(*dirs) );
214 count = 0;
215 dirs[count++] = p = path;
216 while (*p)
218 while (*p && *p != ':') p++;
219 if (!*p) break;
220 *p++ = 0;
221 dirs[count++] = p;
223 for (i = 0; i < count; i++) maxlen = max( maxlen, strlen(dirs[i])+2 );
226 while (*names)
228 len = strlen(*names) + 1;
229 file = xmalloc( maxlen + len );
231 for (i = 0; i < count; i++)
233 strcpy( file, dirs[i] );
234 p = file + strlen(file);
235 if (p == file) *p++ = '.';
236 if (p[-1] != '/') *p++ = '/';
237 strcpy( p, *names );
239 if (!stat( file, &st ) && S_ISREG(st.st_mode) && (st.st_mode & 0111)) return file;
241 free( file );
242 names++;
244 return NULL;
247 const char *get_as_command(void)
249 if (!as_command)
251 if (target_alias)
253 as_command = xmalloc( strlen(target_alias) + sizeof("-as") );
254 strcpy( as_command, target_alias );
255 strcat( as_command, "-as" );
257 else
259 static const char * const commands[] = { "gas", "as", NULL };
260 if (!(as_command = find_tool( commands ))) as_command = xstrdup("as");
263 if (force_pointer_size)
265 const char *args = (target_platform == PLATFORM_APPLE) ?
266 ((force_pointer_size == 8) ? " -arch x86_64" : " -arch i386") :
267 ((force_pointer_size == 8) ? " --64" : " --32");
268 as_command = xrealloc( as_command, strlen(as_command) + strlen(args) + 1 );
269 strcat( as_command, args );
272 return as_command;
275 const char *get_ld_command(void)
277 if (!ld_command)
279 if (target_alias)
281 ld_command = xmalloc( strlen(target_alias) + sizeof("-ld") );
282 strcpy( ld_command, target_alias );
283 strcat( ld_command, "-ld" );
285 else
287 static const char * const commands[] = { "ld", "gld", NULL };
288 if (!(ld_command = find_tool( commands ))) ld_command = xstrdup("ld");
291 if (force_pointer_size)
293 const char *args = (target_platform == PLATFORM_APPLE) ?
294 ((force_pointer_size == 8) ? " -arch x86_64" : " -arch i386") :
295 ((force_pointer_size == 8) ? " -m elf_x86_64" : " -m elf_i386");
296 ld_command = xrealloc( ld_command, strlen(ld_command) + strlen(args) + 1 );
297 strcat( ld_command, args );
300 return ld_command;
303 const char *get_nm_command(void)
305 if (!nm_command)
307 if (target_alias)
309 nm_command = xmalloc( strlen(target_alias) + sizeof("-nm") );
310 strcpy( nm_command, target_alias );
311 strcat( nm_command, "-nm" );
313 else
315 static const char * const commands[] = { "nm", "gnm", NULL };
316 if (!(nm_command = find_tool( commands ))) nm_command = xstrdup("nm");
319 return nm_command;
322 const char *get_windres_command(void)
324 static char *windres_command;
326 if (!windres_command)
328 if (target_alias)
330 windres_command = xmalloc( strlen(target_alias) + sizeof("-windres") );
331 strcpy( windres_command, target_alias );
332 strcat( windres_command, "-windres" );
334 else
336 static const char * const commands[] = { "windres", NULL };
337 if (!(windres_command = find_tool( commands ))) windres_command = xstrdup("windres");
340 return windres_command;
343 /* get a name for a temp file, automatically cleaned up on exit */
344 char *get_temp_file_name( const char *prefix, const char *suffix )
346 char *name;
347 const char *ext;
348 int fd;
350 assert( nb_tmp_files < MAX_TMP_FILES );
351 if (!nb_tmp_files && !save_temps) atexit( cleanup_tmp_files );
353 if (!prefix || !prefix[0]) prefix = "winebuild";
354 if (!suffix) suffix = "";
355 if (!(ext = strchr( prefix, '.' ))) ext = prefix + strlen(prefix);
356 name = xmalloc( sizeof("/tmp/") + (ext - prefix) + sizeof(".XXXXXX") + strlen(suffix) );
357 strcpy( name, "/tmp/" );
358 memcpy( name + 5, prefix, ext - prefix );
359 strcpy( name + 5 + (ext - prefix), ".XXXXXX" );
360 strcat( name, suffix );
362 /* first try without the /tmp/ prefix */
363 if ((fd = mkstemps( name + 5, strlen(suffix) )) != -1)
364 name += 5;
365 else if ((fd = mkstemps( name, strlen(suffix) )) == -1)
366 fatal_error( "could not generate a temp file\n" );
368 close( fd );
369 tmp_files[nb_tmp_files++] = name;
370 return name;
373 /* output a standard header for generated files */
374 void output_standard_file_header(void)
376 if (spec_file_name)
377 output( "/* File generated automatically from %s; do not edit! */\n", spec_file_name );
378 else
379 output( "/* File generated automatically; do not edit! */\n" );
380 output( "/* This file can be copied, modified and distributed without restriction. */\n\n" );
383 /* dump a byte stream into the assembly code */
384 void dump_bytes( const void *buffer, unsigned int size )
386 unsigned int i;
387 const unsigned char *ptr = buffer;
389 if (!size) return;
390 output( "\t.byte " );
391 for (i = 0; i < size - 1; i++, ptr++)
393 if ((i % 16) == 15) output( "0x%02x\n\t.byte ", *ptr );
394 else output( "0x%02x,", *ptr );
396 output( "0x%02x\n", *ptr );
400 /*******************************************************************
401 * open_input_file
403 * Open a file in the given srcdir and set the input_file_name global variable.
405 FILE *open_input_file( const char *srcdir, const char *name )
407 char *fullname;
408 FILE *file = fopen( name, "r" );
410 if (!file && srcdir)
412 fullname = xmalloc( strlen(srcdir) + strlen(name) + 2 );
413 strcpy( fullname, srcdir );
414 strcat( fullname, "/" );
415 strcat( fullname, name );
416 file = fopen( fullname, "r" );
418 else fullname = xstrdup( name );
420 if (!file) fatal_error( "Cannot open file '%s'\n", fullname );
421 input_file_name = fullname;
422 current_line = 1;
423 return file;
427 /*******************************************************************
428 * close_input_file
430 * Close the current input file (must have been opened with open_input_file).
432 void close_input_file( FILE *file )
434 fclose( file );
435 free( input_file_name );
436 input_file_name = NULL;
437 current_line = 0;
441 /*******************************************************************
442 * remove_stdcall_decoration
444 * Remove a possible @xx suffix from a function name.
445 * Return the numerical value of the suffix, or -1 if none.
447 int remove_stdcall_decoration( char *name )
449 char *p, *end = strrchr( name, '@' );
450 if (!end || !end[1] || end == name) return -1;
451 /* make sure all the rest is digits */
452 for (p = end + 1; *p; p++) if (!isdigit(*p)) return -1;
453 *end = 0;
454 return atoi( end + 1 );
458 /*******************************************************************
459 * assemble_file
461 * Run a file through the assembler.
463 void assemble_file( const char *src_file, const char *obj_file )
465 const char *prog = get_as_command();
466 char *cmd;
467 int err;
469 cmd = xmalloc( strlen(prog) + strlen(obj_file) + strlen(src_file) + 6 );
470 sprintf( cmd, "%s -o %s %s", prog, obj_file, src_file );
471 if (verbose) fprintf( stderr, "%s\n", cmd );
472 err = system( cmd );
473 if (err) fatal_error( "%s failed with status %d\n", prog, err );
474 free( cmd );
478 /*******************************************************************
479 * alloc_dll_spec
481 * Create a new dll spec file descriptor
483 DLLSPEC *alloc_dll_spec(void)
485 DLLSPEC *spec;
487 spec = xmalloc( sizeof(*spec) );
488 spec->file_name = NULL;
489 spec->dll_name = NULL;
490 spec->init_func = NULL;
491 spec->main_module = NULL;
492 spec->type = SPEC_WIN32;
493 spec->base = MAX_ORDINALS;
494 spec->limit = 0;
495 spec->stack_size = 0;
496 spec->heap_size = 0;
497 spec->nb_entry_points = 0;
498 spec->alloc_entry_points = 0;
499 spec->nb_names = 0;
500 spec->nb_resources = 0;
501 spec->characteristics = IMAGE_FILE_EXECUTABLE_IMAGE;
502 if (get_ptr_size() > 4)
503 spec->characteristics |= IMAGE_FILE_LARGE_ADDRESS_AWARE;
504 else
505 spec->characteristics |= IMAGE_FILE_32BIT_MACHINE;
506 spec->dll_characteristics = IMAGE_DLLCHARACTERISTICS_NX_COMPAT;
507 spec->subsystem = 0;
508 spec->subsystem_major = 4;
509 spec->subsystem_minor = 0;
510 spec->entry_points = NULL;
511 spec->names = NULL;
512 spec->ordinals = NULL;
513 spec->resources = NULL;
514 return spec;
518 /*******************************************************************
519 * free_dll_spec
521 * Free dll spec file descriptor
523 void free_dll_spec( DLLSPEC *spec )
525 int i;
527 for (i = 0; i < spec->nb_entry_points; i++)
529 ORDDEF *odp = &spec->entry_points[i];
530 free( odp->name );
531 free( odp->export_name );
532 free( odp->link_name );
534 free( spec->file_name );
535 free( spec->dll_name );
536 free( spec->init_func );
537 free( spec->entry_points );
538 free( spec->names );
539 free( spec->ordinals );
540 free( spec->resources );
541 free( spec );
545 /*******************************************************************
546 * make_c_identifier
548 * Map a string to a valid C identifier.
550 const char *make_c_identifier( const char *str )
552 static char buffer[256];
553 char *p;
555 for (p = buffer; *str && p < buffer+sizeof(buffer)-1; p++, str++)
557 if (isalnum(*str)) *p = *str;
558 else *p = '_';
560 *p = 0;
561 return buffer;
565 /*******************************************************************
566 * get_stub_name
568 * Generate an internal name for a stub entry point.
570 const char *get_stub_name( const ORDDEF *odp, const DLLSPEC *spec )
572 static char buffer[256];
573 if (odp->name || odp->export_name)
575 char *p;
576 sprintf( buffer, "__wine_stub_%s", odp->name ? odp->name : odp->export_name );
577 /* make sure name is a legal C identifier */
578 for (p = buffer; *p; p++) if (!isalnum(*p) && *p != '_') break;
579 if (!*p) return buffer;
581 sprintf( buffer, "__wine_stub_%s_%d", make_c_identifier(spec->file_name), odp->ordinal );
582 return buffer;
585 /* parse a cpu name and return the corresponding value */
586 enum target_cpu get_cpu_from_name( const char *name )
588 unsigned int i;
590 for (i = 0; i < sizeof(cpu_names)/sizeof(cpu_names[0]); i++)
591 if (!strcmp( cpu_names[i].name, name )) return cpu_names[i].cpu;
592 return -1;
595 /*****************************************************************
596 * Function: get_alignment
598 * Description:
599 * According to the info page for gas, the .align directive behaves
600 * differently on different systems. On some architectures, the
601 * argument of a .align directive is the number of bytes to pad to, so
602 * to align on an 8-byte boundary you'd say
603 * .align 8
604 * On other systems, the argument is "the number of low-order zero bits
605 * that the location counter must have after advancement." So to
606 * align on an 8-byte boundary you'd say
607 * .align 3
609 * The reason gas is written this way is that it's trying to mimick
610 * native assemblers for the various architectures it runs on. gas
611 * provides other directives that work consistently across
612 * architectures, but of course we want to work on all arches with or
613 * without gas. Hence this function.
616 * Parameters:
617 * align -- the number of bytes to align to. Must be a power of 2.
619 unsigned int get_alignment(unsigned int align)
621 unsigned int n;
623 assert( !(align & (align - 1)) );
625 switch(target_cpu)
627 case CPU_x86:
628 case CPU_x86_64:
629 case CPU_SPARC:
630 if (target_platform != PLATFORM_APPLE) return align;
631 /* fall through */
632 case CPU_POWERPC:
633 case CPU_ALPHA:
634 n = 0;
635 while ((1u << n) != align) n++;
636 return n;
638 /* unreached */
639 assert(0);
640 return 0;
643 /* return the page size for the target CPU */
644 unsigned int get_page_size(void)
646 switch(target_cpu)
648 case CPU_x86: return 4096;
649 case CPU_x86_64: return 4096;
650 case CPU_POWERPC: return 4096;
651 case CPU_SPARC: return 8192;
652 case CPU_ALPHA: return 8192;
654 /* unreached */
655 assert(0);
656 return 0;
659 /* return the size of a pointer on the target CPU */
660 unsigned int get_ptr_size(void)
662 switch(target_cpu)
664 case CPU_x86:
665 case CPU_POWERPC:
666 case CPU_SPARC:
667 case CPU_ALPHA:
668 return 4;
669 case CPU_x86_64:
670 return 8;
672 /* unreached */
673 assert(0);
674 return 0;
677 /* return the assembly name for a C symbol */
678 const char *asm_name( const char *sym )
680 static char buffer[256];
682 switch (target_platform)
684 case PLATFORM_APPLE:
685 case PLATFORM_WINDOWS:
686 if (sym[0] == '.' && sym[1] == 'L') return sym;
687 buffer[0] = '_';
688 strcpy( buffer + 1, sym );
689 return buffer;
690 default:
691 return sym;
695 /* return an assembly function declaration for a C function name */
696 const char *func_declaration( const char *func )
698 static char buffer[256];
700 switch (target_platform)
702 case PLATFORM_APPLE:
703 return "";
704 case PLATFORM_WINDOWS:
705 sprintf( buffer, ".def _%s; .scl 2; .type 32; .endef", func );
706 break;
707 default:
708 sprintf( buffer, ".type %s,@function", func );
709 break;
711 return buffer;
714 /* output a size declaration for an assembly function */
715 void output_function_size( const char *name )
717 switch (target_platform)
719 case PLATFORM_APPLE:
720 case PLATFORM_WINDOWS:
721 break;
722 default:
723 output( "\t.size %s, .-%s\n", name, name );
724 break;
728 /* output the GNU note for non-exec stack */
729 void output_gnu_stack_note(void)
731 switch (target_platform)
733 case PLATFORM_WINDOWS:
734 case PLATFORM_APPLE:
735 break;
736 default:
737 output( "\t.section .note.GNU-stack,\"\",@progbits\n" );
738 break;
742 /* return a global symbol declaration for an assembly symbol */
743 const char *asm_globl( const char *func )
745 static char buffer[256];
747 switch (target_platform)
749 case PLATFORM_APPLE:
750 sprintf( buffer, "\t.globl _%s\n\t.private_extern _%s\n_%s:", func, func, func );
751 return buffer;
752 case PLATFORM_WINDOWS:
753 sprintf( buffer, "\t.globl _%s\n_%s:", func, func );
754 return buffer;
755 default:
756 sprintf( buffer, "\t.globl %s\n\t.hidden %s\n%s:", func, func, func );
757 return buffer;
761 const char *get_asm_ptr_keyword(void)
763 switch(get_ptr_size())
765 case 4: return ".long";
766 case 8: return ".quad";
768 assert(0);
769 return NULL;
772 const char *get_asm_string_keyword(void)
774 switch (target_platform)
776 case PLATFORM_APPLE:
777 return ".asciz";
778 default:
779 return ".string";
783 const char *get_asm_short_keyword(void)
785 switch (target_platform)
787 default: return ".short";
791 const char *get_asm_rodata_section(void)
793 switch (target_platform)
795 case PLATFORM_APPLE: return ".const";
796 default: return ".section .rodata";
800 const char *get_asm_string_section(void)
802 switch (target_platform)
804 case PLATFORM_APPLE: return ".cstring";
805 default: return ".section .rodata";