loader/glibc.c

   1 /*
   2  * glibc threading support
   3  *
   4  * Copyright 2003 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  */
  20
  21 #include "config.h"
  22 #include "wine/port.h"
  23
  24 #include <stdio.h>
  25 #include <stdlib.h>
  26 #ifdef HAVE_UNISTD_H
  27 # include <unistd.h>
  28 #endif
  29 #ifdef HAVE_PTHREAD_H
  30 # include <pthread.h>
  31 #endif
  32
  33 #include "wine/library.h"
  34
  35 /* malloc wrapper */
  36 static void *xmalloc( size_t size )
  37 {
  38     void *res;
  39
  40     if (!size) size = 1;
  41     if (!(res = malloc( size )))
  42     {
  43         fprintf( stderr, "wine: virtual memory exhausted\n" );
  44         exit(1);
  45     }
  46     return res;
  47 }
  48
  49 /* separate thread to check for NPTL and TLS features */
  50 static void *needs_pthread( void *arg )
  51 {
  52     pid_t tid = gettid();
  53     /* check for NPTL */
  54     if (tid != -1 && tid != getpid()) return (void *)1;
  55     /* check for TLS glibc */
  56     return (void *)(wine_get_gs() != 0);
  57 }
  58
  59 /* return the name of the Wine threading variant to use */
  60 static const char *get_threading(void)
  61 {
  62     pthread_t id;
  63     void *ret;
  64
  65     pthread_create( &id, NULL, needs_pthread, NULL );
  66     pthread_join( id, &ret );
  67     return ret ? "wine-pthread" : "wine-kthread";
  68 }
  69
  70
  71 /**********************************************************************
  72  *           main
  73  */
  74 int main( int argc, char *argv[] )
  75 {
  76     const char *loader = getenv( "WINELOADER" );
  77     const char *threads = get_threading();
  78
  79     if (loader)
  80     {
  81         const char *path;
  82         char *new_name, *new_loader;
  83
  84         if ((path = strrchr( loader, '/' ))) path++;
  85         else path = loader;
  86
  87         new_name = xmalloc( (path - loader) + strlen(threads) + 1 );
  88         memcpy( new_name, loader, path - loader );
  89         strcpy( new_name + (path - loader), threads );
  90
  91         /* update WINELOADER with the new name */
  92         new_loader = xmalloc( sizeof("WINELOADER=") + strlen(new_name) );
  93         strcpy( new_loader, "WINELOADER=" );
  94         strcat( new_loader, new_name );
  95         putenv( new_loader );
  96         wine_exec_wine_binary( new_name, argv, NULL, TRUE );
  97     }
  98     else
  99     {
 100         wine_init_argv0_path( argv[0] );
 101         wine_exec_wine_binary( threads, argv, NULL, TRUE );
 102     }
 103     fprintf( stderr, "wine: could not exec %s\n", argv[0] );
 104     exit(1);
 105 }