tools/winebuild/utils.c

   1 /*
   2  * Small utility functions for winebuild
   3  *
   4  * Copyright 2000 Alexandre Julliard
   5  *
   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.
  10  *
  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.
  15  *
  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
  19  */
  20
  21 #include "config.h"
  22 #include "wine/port.h"
  23
  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
  33
  34 #include "windef.h"
  35 #include "winnt.h"
  36 #include "build.h"
  37
  38 #define MAX_TMP_FILES 8
  39 static const char *tmp_files[MAX_TMP_FILES];
  40 static unsigned int nb_tmp_files;
  41
  42 /* atexit handler to clean tmp files */
  43 static void cleanup_tmp_files(void)
  44 {
  45     unsigned int i;
  46     for (i = 0; i < MAX_TMP_FILES; i++) if (tmp_files[i]) unlink( tmp_files[i] );
  47 }
  48
  49
  50 void *xmalloc (size_t size)
  51 {
  52     void *res;
  53
  54     res = malloc (size ? size : 1);
  55     if (res == NULL)
  56     {
  57         fprintf (stderr, "Virtual memory exhausted.\n");
  58         exit (1);
  59     }
  60     return res;
  61 }
  62
  63 void *xrealloc (void *ptr, size_t size)
  64 {
  65     void *res = realloc (ptr, size);
  66     if (size && res == NULL)
  67     {
  68         fprintf (stderr, "Virtual memory exhausted.\n");
  69         exit (1);
  70     }
  71     return res;
  72 }
  73
  74 char *xstrdup( const char *str )
  75 {
  76     char *res = strdup( str );
  77     if (!res)
  78     {
  79         fprintf (stderr, "Virtual memory exhausted.\n");
  80         exit (1);
  81     }
  82     return res;
  83 }
  84
  85 char *strupper(char *s)
  86 {
  87     char *p;
  88     for (p = s; *p; p++) *p = toupper(*p);
  89     return s;
  90 }
  91
  92 int strendswith(const char* str, const char* end)
  93 {
  94     int l = strlen(str);
  95     int m = strlen(end);
  96     return l >= m && strcmp(str + l - m, end) == 0;
  97 }
  98
  99 void fatal_error( const char *msg, ... )
 100 {
 101     va_list valist;
 102     va_start( valist, msg );
 103     if (input_file_name)
 104     {
 105         fprintf( stderr, "%s:", input_file_name );
 106         if (current_line)
 107             fprintf( stderr, "%d:", current_line );
 108         fputc( ' ', stderr );
 109     }
 110     else fprintf( stderr, "winebuild: " );
 111     vfprintf( stderr, msg, valist );
 112     va_end( valist );
 113     exit(1);
 114 }
 115
 116 void fatal_perror( const char *msg, ... )
 117 {
 118     va_list valist;
 119     va_start( valist, msg );
 120     if (input_file_name)
 121     {
 122         fprintf( stderr, "%s:", input_file_name );
 123         if (current_line)
 124             fprintf( stderr, "%d:", current_line );
 125         fputc( ' ', stderr );
 126     }
 127     vfprintf( stderr, msg, valist );
 128     perror( " " );
 129     va_end( valist );
 130     exit(1);
 131 }
 132
 133 void error( const char *msg, ... )
 134 {
 135     va_list valist;
 136     va_start( valist, msg );
 137     if (input_file_name)
 138     {
 139         fprintf( stderr, "%s:", input_file_name );
 140         if (current_line)
 141             fprintf( stderr, "%d:", current_line );
 142         fputc( ' ', stderr );
 143     }
 144     vfprintf( stderr, msg, valist );
 145     va_end( valist );
 146     nb_errors++;
 147 }
 148
 149 void warning( const char *msg, ... )
 150 {
 151     va_list valist;
 152
 153     if (!display_warnings) return;
 154     va_start( valist, msg );
 155     if (input_file_name)
 156     {
 157         fprintf( stderr, "%s:", input_file_name );
 158         if (current_line)
 159             fprintf( stderr, "%d:", current_line );
 160         fputc( ' ', stderr );
 161     }
 162     fprintf( stderr, "warning: " );
 163     vfprintf( stderr, msg, valist );
 164     va_end( valist );
 165 }
 166
 167 int output( const char *format, ... )
 168 {
 169     int ret;
 170     va_list valist;
 171
 172     va_start( valist, format );
 173     ret = vfprintf( output_file, format, valist );
 174     va_end( valist );
 175     if (ret < 0) fatal_perror( "Output error" );
 176     return ret;
 177 }
 178
 179 /* get a name for a temp file, automatically cleaned up on exit */
 180 char *get_temp_file_name( const char *prefix, const char *suffix )
 181 {
 182     char *name;
 183     const char *ext;
 184     int fd;
 185
 186     assert( nb_tmp_files < MAX_TMP_FILES );
 187     if (!nb_tmp_files && !save_temps) atexit( cleanup_tmp_files );
 188
 189     if (!prefix || !prefix[0]) prefix = "winebuild";
 190     if (!suffix) suffix = "";
 191     if (!(ext = strchr( prefix, '.' ))) ext = prefix + strlen(prefix);
 192     name = xmalloc( sizeof("/tmp/") + (ext - prefix) + sizeof(".XXXXXX") + strlen(suffix) );
 193     strcpy( name, "/tmp/" );
 194     memcpy( name + 5, prefix, ext - prefix );
 195     strcpy( name + 5 + (ext - prefix), ".XXXXXX" );
 196     strcat( name, suffix );
 197
 198     /* first try without the /tmp/ prefix */
 199     if ((fd = mkstemps( name + 5, strlen(suffix) )) != -1)
 200         name += 5;
 201     else if ((fd = mkstemps( name, strlen(suffix) )) == -1)
 202         fatal_error( "could not generate a temp file\n" );
 203
 204     close( fd );
 205     tmp_files[nb_tmp_files++] = name;
 206     return name;
 207 }
 208
 209 /* output a standard header for generated files */
 210 void output_standard_file_header(void)
 211 {
 212     if (spec_file_name)
 213         output( "/* File generated automatically from %s; do not edit! */\n", spec_file_name );
 214     else
 215         output( "/* File generated automatically; do not edit! */\n" );
 216     output( "/* This file can be copied, modified and distributed without restriction. */\n\n" );
 217 }
 218
 219 /* dump a byte stream into the assembly code */
 220 void dump_bytes( const void *buffer, unsigned int size )
 221 {
 222     unsigned int i;
 223     const unsigned char *ptr = buffer;
 224
 225     if (!size) return;
 226     output( "\t.byte " );
 227     for (i = 0; i < size - 1; i++, ptr++)
 228     {
 229         if ((i % 16) == 15) output( "0x%02x\n\t.byte ", *ptr );
 230         else output( "0x%02x,", *ptr );
 231     }
 232     output( "0x%02x\n", *ptr );
 233 }
 234
 235
 236 /*******************************************************************
 237  *         open_input_file
 238  *
 239  * Open a file in the given srcdir and set the input_file_name global variable.
 240  */
 241 FILE *open_input_file( const char *srcdir, const char *name )
 242 {
 243     char *fullname;
 244     FILE *file = fopen( name, "r" );
 245
 246     if (!file && srcdir)
 247     {
 248         fullname = xmalloc( strlen(srcdir) + strlen(name) + 2 );
 249         strcpy( fullname, srcdir );
 250         strcat( fullname, "/" );
 251         strcat( fullname, name );
 252         file = fopen( fullname, "r" );
 253     }
 254     else fullname = xstrdup( name );
 255
 256     if (!file) fatal_error( "Cannot open file '%s'\n", fullname );
 257     input_file_name = fullname;
 258     current_line = 1;
 259     return file;
 260 }
 261
 262
 263 /*******************************************************************
 264  *         close_input_file
 265  *
 266  * Close the current input file (must have been opened with open_input_file).
 267  */
 268 void close_input_file( FILE *file )
 269 {
 270     fclose( file );
 271     free( input_file_name );
 272     input_file_name = NULL;
 273     current_line = 0;
 274 }
 275
 276
 277 /*******************************************************************
 278  *         remove_stdcall_decoration
 279  *
 280  * Remove a possible @xx suffix from a function name.
 281  * Return the numerical value of the suffix, or -1 if none.
 282  */
 283 int remove_stdcall_decoration( char *name )
 284 {
 285     char *p, *end = strrchr( name, '@' );
 286     if (!end || !end[1] || end == name) return -1;
 287     /* make sure all the rest is digits */
 288     for (p = end + 1; *p; p++) if (!isdigit(*p)) return -1;
 289     *end = 0;
 290     return atoi( end + 1 );
 291 }
 292
 293
 294 /*******************************************************************
 295  *         assemble_file
 296  *
 297  * Run a file through the assembler.
 298  */
 299 void assemble_file( const char *src_file, const char *obj_file )
 300 {
 301     char *cmd;
 302     int err;
 303
 304     if (!as_command) as_command = xstrdup("as");
 305     cmd = xmalloc( strlen(as_command) + strlen(obj_file) + strlen(src_file) + 6 );
 306     sprintf( cmd, "%s -o %s %s", as_command, obj_file, src_file );
 307     if (verbose) fprintf( stderr, "%s\n", cmd );
 308     err = system( cmd );
 309     if (err) fatal_error( "%s failed with status %d\n", as_command, err );
 310     free( cmd );
 311 }
 312
 313
 314 /*******************************************************************
 315  *         alloc_dll_spec
 316  *
 317  * Create a new dll spec file descriptor
 318  */
 319 DLLSPEC *alloc_dll_spec(void)
 320 {
 321     DLLSPEC *spec;
 322
 323     spec = xmalloc( sizeof(*spec) );
 324     spec->file_name          = NULL;
 325     spec->dll_name           = NULL;
 326     spec->init_func          = NULL;
 327     spec->type               = SPEC_WIN32;
 328     spec->base               = MAX_ORDINALS;
 329     spec->limit              = 0;
 330     spec->stack_size         = 0;
 331     spec->heap_size          = 0;
 332     spec->nb_entry_points    = 0;
 333     spec->alloc_entry_points = 0;
 334     spec->nb_names           = 0;
 335     spec->nb_resources       = 0;
 336     spec->characteristics    = IMAGE_FILE_EXECUTABLE_IMAGE;
 337     if (get_ptr_size() > 4)
 338         spec->characteristics |= IMAGE_FILE_LARGE_ADDRESS_AWARE;
 339     else
 340         spec->characteristics |= IMAGE_FILE_32BIT_MACHINE;
 341     spec->dll_characteristics = IMAGE_DLLCHARACTERISTICS_NX_COMPAT;
 342     spec->subsystem          = 0;
 343     spec->subsystem_major    = 4;
 344     spec->subsystem_minor    = 0;
 345     spec->entry_points       = NULL;
 346     spec->names              = NULL;
 347     spec->ordinals           = NULL;
 348     spec->resources          = NULL;
 349     return spec;
 350 }
 351
 352
 353 /*******************************************************************
 354  *         free_dll_spec
 355  *
 356  * Free dll spec file descriptor
 357  */
 358 void free_dll_spec( DLLSPEC *spec )
 359 {
 360     int i;
 361
 362     for (i = 0; i < spec->nb_entry_points; i++)
 363     {
 364         ORDDEF *odp = &spec->entry_points[i];
 365         free( odp->name );
 366         free( odp->export_name );
 367         free( odp->link_name );
 368     }
 369     free( spec->file_name );
 370     free( spec->dll_name );
 371     free( spec->init_func );
 372     free( spec->entry_points );
 373     free( spec->names );
 374     free( spec->ordinals );
 375     free( spec->resources );
 376     free( spec );
 377 }
 378
 379
 380 /*******************************************************************
 381  *         make_c_identifier
 382  *
 383  * Map a string to a valid C identifier.
 384  */
 385 const char *make_c_identifier( const char *str )
 386 {
 387     static char buffer[256];
 388     char *p;
 389
 390     for (p = buffer; *str && p < buffer+sizeof(buffer)-1; p++, str++)
 391     {
 392         if (isalnum(*str)) *p = *str;
 393         else *p = '_';
 394     }
 395     *p = 0;
 396     return buffer;
 397 }
 398
 399
 400 /*******************************************************************
 401  *         get_stub_name
 402  *
 403  * Generate an internal name for a stub entry point.
 404  */
 405 const char *get_stub_name( const ORDDEF *odp, const DLLSPEC *spec )
 406 {
 407     static char buffer[256];
 408     if (odp->name || odp->export_name)
 409     {
 410         char *p;
 411         sprintf( buffer, "__wine_stub_%s", odp->name ? odp->name : odp->export_name );
 412         /* make sure name is a legal C identifier */
 413         for (p = buffer; *p; p++) if (!isalnum(*p) && *p != '_') break;
 414         if (!*p) return buffer;
 415     }
 416     sprintf( buffer, "__wine_stub_%s_%d", make_c_identifier(spec->file_name), odp->ordinal );
 417     return buffer;
 418 }
 419
 420
 421 /*****************************************************************
 422  *  Function:    get_alignment
 423  *
 424  *  Description:
 425  *    According to the info page for gas, the .align directive behaves
 426  * differently on different systems.  On some architectures, the
 427  * argument of a .align directive is the number of bytes to pad to, so
 428  * to align on an 8-byte boundary you'd say
 429  *     .align 8
 430  * On other systems, the argument is "the number of low-order zero bits
 431  * that the location counter must have after advancement."  So to
 432  * align on an 8-byte boundary you'd say
 433  *     .align 3
 434  *
 435  * The reason gas is written this way is that it's trying to mimick
 436  * native assemblers for the various architectures it runs on.  gas
 437  * provides other directives that work consistently across
 438  * architectures, but of course we want to work on all arches with or
 439  * without gas.  Hence this function.
 440  *
 441  *
 442  *  Parameters:
 443  *    align  --  the number of bytes to align to. Must be a power of 2.
 444  */
 445 unsigned int get_alignment(unsigned int align)
 446 {
 447     unsigned int n;
 448
 449     assert( !(align & (align - 1)) );
 450
 451     switch(target_cpu)
 452     {
 453     case CPU_x86:
 454     case CPU_x86_64:
 455     case CPU_SPARC:
 456         if (target_platform != PLATFORM_APPLE) return align;
 457         /* fall through */
 458     case CPU_POWERPC:
 459     case CPU_ALPHA:
 460         n = 0;
 461         while ((1u << n) != align) n++;
 462         return n;
 463     }
 464     /* unreached */
 465     assert(0);
 466     return 0;
 467 }
 468
 469 /* return the page size for the target CPU */
 470 unsigned int get_page_size(void)
 471 {
 472     switch(target_cpu)
 473     {
 474     case CPU_x86:     return 4096;
 475     case CPU_x86_64:  return 4096;
 476     case CPU_POWERPC: return 4096;
 477     case CPU_SPARC:   return 8192;
 478     case CPU_ALPHA:   return 8192;
 479     }
 480     /* unreached */
 481     assert(0);
 482     return 0;
 483 }
 484
 485 /* return the size of a pointer on the target CPU */
 486 unsigned int get_ptr_size(void)
 487 {
 488     switch(target_cpu)
 489     {
 490     case CPU_x86:
 491     case CPU_POWERPC:
 492     case CPU_SPARC:
 493     case CPU_ALPHA:
 494         return 4;
 495     case CPU_x86_64:
 496         return 8;
 497     }
 498     /* unreached */
 499     assert(0);
 500     return 0;
 501 }
 502
 503 /* return the assembly name for a C symbol */
 504 const char *asm_name( const char *sym )
 505 {
 506     static char buffer[256];
 507
 508     switch (target_platform)
 509     {
 510     case PLATFORM_APPLE:
 511     case PLATFORM_WINDOWS:
 512         buffer[0] = '_';
 513         strcpy( buffer + 1, sym );
 514         return buffer;
 515     default:
 516         return sym;
 517     }
 518 }
 519
 520 /* return an assembly function declaration for a C function name */
 521 const char *func_declaration( const char *func )
 522 {
 523     static char buffer[256];
 524
 525     switch (target_platform)
 526     {
 527     case PLATFORM_APPLE:
 528         return "";
 529     case PLATFORM_WINDOWS:
 530         sprintf( buffer, ".def _%s; .scl 2; .type 32; .endef", func );
 531         break;
 532     default:
 533         sprintf( buffer, ".type %s,@function", func );
 534         break;
 535     }
 536     return buffer;
 537 }
 538
 539 /* output a size declaration for an assembly function */
 540 void output_function_size( const char *name )
 541 {
 542     switch (target_platform)
 543     {
 544     case PLATFORM_APPLE:
 545     case PLATFORM_WINDOWS:
 546         break;
 547     default:
 548         output( "\t.size %s, .-%s\n", name, name );
 549         break;
 550     }
 551 }
 552
 553 /* output the GNU note for non-exec stack */
 554 void output_gnu_stack_note(void)
 555 {
 556     switch (target_platform)
 557     {
 558     case PLATFORM_WINDOWS:
 559     case PLATFORM_APPLE:
 560         break;
 561     default:
 562         output( "\t.section .note.GNU-stack,\"\",@progbits\n" );
 563         break;
 564     }
 565 }
 566
 567 /* return a global symbol declaration for an assembly symbol */
 568 const char *asm_globl( const char *func )
 569 {
 570     static char buffer[256];
 571
 572     switch (target_platform)
 573     {
 574     case PLATFORM_APPLE:
 575         sprintf( buffer, "\t.globl _%s\n\t.private_extern _%s\n_%s:", func, func, func );
 576         return buffer;
 577     case PLATFORM_WINDOWS:
 578         sprintf( buffer, "\t.globl _%s\n_%s:", func, func );
 579         return buffer;
 580     default:
 581         sprintf( buffer, "\t.globl %s\n\t.hidden %s\n%s:", func, func, func );
 582         return buffer;
 583     }
 584 }
 585
 586 const char *get_asm_ptr_keyword(void)
 587 {
 588     switch(get_ptr_size())
 589     {
 590     case 4: return ".long";
 591     case 8: return ".quad";
 592     }
 593     assert(0);
 594     return NULL;
 595 }
 596
 597 const char *get_asm_string_keyword(void)
 598 {
 599     switch (target_platform)
 600     {
 601     case PLATFORM_APPLE:
 602         return ".asciz";
 603     default:
 604         return ".string";
 605     }
 606 }
 607
 608 const char *get_asm_short_keyword(void)
 609 {
 610     switch (target_platform)
 611     {
 612     default:            return ".short";
 613     }
 614 }
 615
 616 const char *get_asm_rodata_section(void)
 617 {
 618     switch (target_platform)
 619     {
 620     case PLATFORM_APPLE: return ".const";
 621     default:             return ".section .rodata";
 622     }
 623 }
 624
 625 const char *get_asm_string_section(void)
 626 {
 627     switch (target_platform)
 628     {
 629     case PLATFORM_APPLE: return ".cstring";
 630     default:             return ".section .rodata";
 631     }
 632 }