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
23typedef void (*FunctionPtr)();
24
25int 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}
123