blob: d12810a0c002432d46cebb3c350fa7a1b80064a4 [file] [log] [blame]
// Copyright 2014 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <errno.h>
#include <stdarg.h>
#include <stdio.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <sys/uio.h>
#include <sys/wait.h>
#include <unistd.h>
#include <crazy_linker.h>
#include "test_util.h"
#define PARENT_ADDRESS 0x20000000
#define CHILD_ADDRESS 0x20134000
#define LIB_NAME ""
typedef void (*FunctionPtr)();
int main() {
crazy_context_t* context = crazy_context_create();
RelroLibrary foo;
int pipes[2];
if (socketpair(AF_UNIX, SOCK_STREAM, 0, pipes) < 0)
Panic("Could not create socket pair: %s", strerror(errno));
pid_t child = fork();
if (child < 0)
Panic("Could not fork test program!");
if (child == 0) {
// In the child.
crazy_context_set_load_address(context, CHILD_ADDRESS);
foo.Init(LIB_NAME, context);
printf("Child waiting for foo relro fd\n");
printf("RELRO used in child process\n");
FunctionPtr foo_func;
if (!crazy_library_find_symbol(
foo.library, "Foo", reinterpret_cast<void**>(&foo_func)))
Panic("Could not find 'Foo' in library");
printf("Calling Foo()\n");
printf("Foo called, exiting\n");
} else {
// In the parent.
// Load at fixed address to simplify testing.
crazy_context_set_load_address(context, PARENT_ADDRESS);
foo.Init(LIB_NAME, context);
printf("Library loaded\n");
printf("Parent enabling foo RELRO sharing\n");
foo.CreateSharedRelro(context, CHILD_ADDRESS);
printf("Relocated RELRO sent to child\n");
printf("Parent waiting for child\n");
// Wait for child to complete.
int status;
waitpid(child, &status, 0);
if (WIFSIGNALED(status))
Panic("Child terminated by signal!!\n");
else if (WIFEXITED(status)) {
int child_status = WEXITSTATUS(status);
if (child_status != 0)
Panic("Child terminated with status=%d\n", child_status);
} else
Panic("Child exited for unknown reason!!\n");
return 0;