The Tor Browser: media/webrtc/trunk/tools/gyp/test/module/src/program.c@925c144e1f1f

     1 #include <stdio.h>

     2 #include <stdlib.h>

     4 #if defined(PLATFORM_WIN)

     5 #include <windows.h>

     6 #elif defined(PLATFORM_MAC) || defined(PLATFORM_LINUX)

     7 #include <dlfcn.h>

     8 #include <libgen.h>

     9 #include <string.h>

    10 #include <sys/param.h>

    11 #define MAX_PATH PATH_MAX

    12 #endif

    14 #if defined(PLATFORM_WIN)

    15 #define MODULE_SUFFIX ".dll"

    16 #elif defined(PLATFORM_MAC)

    17 #define MODULE_SUFFIX ".so"

    18 #elif defined(PLATFORM_LINUX)

    19 #define MODULE_SUFFIX ".so"

    20 #endif

    22 typedef void (*module_symbol)(void);

    23 char bin_path[MAX_PATH + 1];

    26 void CallModule(const char* module) {

    27   char module_path[MAX_PATH + 1];

    28   const char* module_function = "module_main";

    29   module_symbol funcptr;

    30 #if defined(PLATFORM_WIN)

    31   HMODULE dl;

    32   char drive[_MAX_DRIVE];

    33   char dir[_MAX_DIR];

    35   if (_splitpath_s(bin_path, drive, _MAX_DRIVE, dir, _MAX_DIR,

    36                     NULL, 0, NULL, 0)) {

    37     fprintf(stderr, "Failed to split executable path.\n");

    38     return;

    39   }

    40   if (_makepath_s(module_path, MAX_PATH, drive, dir, module, MODULE_SUFFIX)) {

    41     fprintf(stderr, "Failed to calculate module path.\n");

    42     return;

    43   }

    45   dl = LoadLibrary(module_path);

    46   if (!dl) {

    47     fprintf(stderr, "Failed to open module: %s\n", module_path);

    48     return;

    49   }

    51   funcptr = (module_symbol) GetProcAddress(dl, module_function);

    52   if (!funcptr) {

    53     fprintf(stderr, "Failed to find symbol: %s\n", module_function);

    54     return;

    55   }

    56   funcptr();

    58   FreeLibrary(dl);

    59 #elif defined(PLATFORM_MAC) || defined(PLATFORM_LINUX)

    60   void* dl;

    61   char* path_copy = strdup(bin_path);

    62   char* bin_dir = dirname(path_copy);

    63   int path_size = snprintf(module_path, MAX_PATH, "%s/%s%s", bin_dir, module,

    64                            MODULE_SUFFIX);

    65   free(path_copy);

    66   if (path_size < 0 || path_size > MAX_PATH) {

    67     fprintf(stderr, "Failed to calculate module path.\n");

    68     return;

    69   }

    70   module_path[path_size] = 0;

    72   dl = dlopen(module_path, RTLD_LAZY);

    73   if (!dl) {

    74     fprintf(stderr, "Failed to open module: %s\n", module_path);

    75     return;

    76   }

    78   funcptr = dlsym(dl, module_function);

    79   if (!funcptr) {

    80     fprintf(stderr, "Failed to find symbol: %s\n", module_function);

    81     return;

    82   }

    83   funcptr();

    85   dlclose(dl);

    86 #endif

    87 }

    89 int main(int argc, char *argv[])

    90 {

    91   fprintf(stdout, "Hello from program.c\n");

    92   fflush(stdout);

    94 #if defined(PLATFORM_WIN)

    95   if (!GetModuleFileName(NULL, bin_path, MAX_PATH)) {

    96     fprintf(stderr, "Failed to determine executable path.\n");

    97     return;

    98   }

    99 #elif defined(PLATFORM_MAC) || defined(PLATFORM_LINUX)

   100   // Using argv[0] should be OK here since we control how the tests run, and

   101   // can avoid exec and such issues that make it unreliable.

   102   if (!realpath(argv[0], bin_path)) {

   103     fprintf(stderr, "Failed to determine executable path (%s).\n", argv[0]);

   104     return;

   105   }

   106 #endif

   108   CallModule("lib1");

   109   CallModule("lib2");

   110   return 0;

   111 }

The Tor Browser / file revision