third_party/android_crazy_linker/src/tests/test_two_shared_relros.cpp

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 // Same as test_relro_sharing.cpp, but uses two libraries at the same
   6 // time (libfoo_with_relro.so and libbar_with_relro.so), each one of
   7 // them gets its own shared RELRO.
   8
   9 #include <errno.h>
  10 #include <pthread.h>
  11 #include <stdarg.h>
  12 #include <stdio.h>
  13 #include <stdlib.h>
  14 #include <sys/socket.h>
  15 #include <sys/uio.h>
  16 #include <sys/wait.h>
  17 #include <unistd.h>
  18
  19 #include <crazy_linker.h>
  20
  21 #include "test_util.h"
  22
  23 typedef void (*FunctionPtr)();
  24
  25 int main() {
  26
  27   if (!crazy_system_can_share_relro()) {
  28     fprintf(stderr, "WARNING: Test ignored due to broken kernel!!\n");
  29     return 0;
  30   }
  31
  32   crazy_context_t* context = crazy_context_create();
  33
  34   RelroLibrary foo;
  35   RelroLibrary bar;
  36
  37   crazy_context_add_search_path_for_address(context, (void*)&main);
  38
  39   // Load libfoo_with_relro.so
  40   crazy_context_set_load_address(context, 0x20000000);
  41   foo.Init("libfoo_with_relro.so", context);
  42
  43   crazy_context_set_load_address(context, 0x20800000);
  44   bar.Init("libbar_with_relro.so", context);
  45
  46   printf("Libraries loaded\n");
  47
  48   int pipes[2];
  49   if (socketpair(AF_UNIX, SOCK_STREAM, 0, pipes) < 0)
  50     Panic("Could not create socket pair: %s", strerror(errno));
  51
  52   pid_t child = fork();
  53   if (child < 0)
  54     Panic("Could not fork test program!");
  55
  56   if (child == 0) {
  57     // In the child.
  58     printf("Child waiting for foo relro fd\n");
  59
  60     foo.ReceiveRelroInfo(pipes[0]);
  61     foo.UseSharedRelro(context);
  62
  63     printf("Child waiting for bar relro fd\n");
  64     bar.ReceiveRelroInfo(pipes[0]);
  65     bar.UseSharedRelro(context);
  66
  67     printf("RELROs used in child process\n");
  68
  69     CheckRelroMaps(2);
  70
  71     FunctionPtr bar_func;
  72     if (!crazy_library_find_symbol(
  73              bar.library, "Bar", reinterpret_cast<void**>(&bar_func)))
  74       Panic("Could not find 'Bar' in library");
  75
  76     printf("Calling Bar()\n");
  77     (*bar_func)();
  78
  79     printf("Bar() called, exiting\n");
  80
  81     exit(0);
  82
  83   } else {
  84     // In the parent.
  85
  86     printf("Parent enabling foo RELRO sharing\n");
  87
  88     foo.EnableSharedRelro(context);
  89     foo.SendRelroInfo(pipes[1]);
  90
  91     printf("Parent enabling bar RELRO sharing\n");
  92
  93     bar.EnableSharedRelro(context);
  94     bar.SendRelroInfo(pipes[1]);
  95
  96     printf("RELROs enabled and sent to child\n");
  97
  98     CheckRelroMaps(2);
  99
 100     printf("Parent waiting for child\n");
 101
 102     // Wait for child to complete.
 103     int status;
 104     waitpid(child, &status, 0);
 105
 106     if (WIFSIGNALED(status))
 107       Panic("Child terminated by signal!!\n");
 108     else if (WIFEXITED(status)) {
 109       int child_status = WEXITSTATUS(status);
 110       if (child_status != 0)
 111         Panic("Child terminated with status=%d\n", child_status);
 112     } else
 113       Panic("Child exited for unknown reason!!\n");
 114   }
 115
 116   printf("Closing libraries\n");
 117   bar.Close();
 118   foo.Close();
 119
 120   crazy_context_destroy(context);
 121   return 0;
 122 }