forked from llvm/llvm-project
-
Notifications
You must be signed in to change notification settings - Fork 103
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[rust] Changes needed for 'x86_64-fortanix-unknown-sgx' nightly target.
Code is guarded via defines to enable only if 'RUST_SGX' is present. Main logic is in libunwind/src/AddressSpace.hpp We use 6 symbols to figure out where eh_frame / eh_frame_hdr is at runtime when loaded in an SGX enclave. (EH symbols + IMAGE base) These are set by 'fortanix-sgx-tools'. As notes: - Target above at the moment uses a pre-compiled libunwind.a from forked repo. - Goal of these changes is to use official llvm with patch. - Changes in rust-lang to use this are planned if/when this is accepted. - Ticket: fortanix/rust-sgx#174 - Original ported changes: llvm/llvm-project@release/5.x...fortanix:release/5.x
- Loading branch information
Showing
9 changed files
with
313 additions
and
7 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,22 @@ | ||
# Libunwind customizations for linking with x86_64-fortanix-unknown-sgx Rust target. | ||
|
||
## Description | ||
### Initial Fork | ||
Initial Fork has been made from 5.0 release of llvm (commit: 6a075b6de4) | ||
### Detailed Description | ||
#### Header files that we do not include for this target | ||
1. pthread.h | ||
#### Library that we do not link to for this target. | ||
1. pthread (Locks used by libunwind is provided by rust stdlib for this target) | ||
|
||
## Building unwind for rust-sgx target | ||
### Generate Make files: | ||
* `cd where you want to build libunwind` | ||
* `mkdir build` | ||
* `cd build` | ||
* `cmake -DCMAKE_BUILD_TYPE="RELEASE" -DRUST_SGX=1 -G "Unix Makefiles" -DLLVM_ENABLE_WARNINGS=1 -DLIBUNWIND_ENABLE_PEDANTIC=0 -DLLVM_PATH=<path/to/llvm> <path/to/libunwind>` | ||
* `"DEBUG"` could be used instead of `"RELEASE"` to enable debug logs of libunwind. | ||
|
||
### Build: | ||
* `make unwind_static` | ||
* `build/lib/` will have the built library. |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,125 @@ | ||
//===--------------------- UnwindRustSgx.c ----------------------------------===// | ||
// | ||
//// The LLVM Compiler Infrastructure | ||
//// | ||
//// This file is dual licensed under the MIT and the University of Illinois Open | ||
//// Source Licenses. See LICENSE.TXT for details. | ||
//// | ||
//// | ||
////===----------------------------------------------------------------------===// | ||
|
||
#define _GNU_SOURCE | ||
#include <link.h> | ||
|
||
#include <elf.h> | ||
#include <stdarg.h> | ||
#include <stdio.h> | ||
#include <stddef.h> | ||
#include "UnwindRustSgx.h" | ||
|
||
#define max_log 256 | ||
|
||
__attribute__((weak)) struct _IO_FILE *stderr = (struct _IO_FILE *)-1; | ||
|
||
static int vwrite_err(const char *format, va_list ap) | ||
{ | ||
int len = 0; | ||
#ifndef NDEBUG | ||
char s[max_log]; | ||
s[0]='\0'; | ||
len = vsnprintf(s, max_log, format, ap); | ||
__rust_print_err((uint8_t *)s, len); | ||
#endif | ||
return len; | ||
} | ||
|
||
static int write_err(const char *format, ...) | ||
{ | ||
int ret; | ||
va_list args; | ||
va_start(args, format); | ||
ret = vwrite_err(format, args); | ||
va_end(args); | ||
|
||
|
||
return ret; | ||
} | ||
|
||
__attribute__((weak)) int fprintf (FILE *__restrict __stream, | ||
const char *__restrict __format, ...) | ||
{ | ||
|
||
int ret; | ||
if (__stream != stderr) { | ||
write_err("Rust SGX Unwind supports only writing to stderr\n"); | ||
return -1; | ||
} else { | ||
va_list args; | ||
ret = 0; | ||
va_start(args, __format); | ||
ret += vwrite_err(__format, args); | ||
va_end(args); | ||
} | ||
|
||
return ret; | ||
} | ||
|
||
__attribute__((weak)) int fflush (FILE *__stream) | ||
{ | ||
// We do not need to do anything here. | ||
return 0; | ||
} | ||
|
||
__attribute__((weak)) void __assert_fail(const char * assertion, | ||
const char * file, | ||
unsigned int line, | ||
const char * function) | ||
{ | ||
write_err("%s:%d %s %s\n", file, line, function, assertion); | ||
abort(); | ||
} | ||
|
||
// We do not report stack over flow detected. | ||
// Calling write_err uses more stack due to the way we have implemented it. | ||
// With possible enabling of stack probes, we should not | ||
// get into __stack_chk_fail() at all. | ||
__attribute__((weak)) void __stack_chk_fail() { | ||
abort(); | ||
} | ||
|
||
/* | ||
* Below are defined for all executibles compiled for | ||
* x86_64-fortanix-unknown-sgx rust target. | ||
* Ref: rust/src/libstd/sys/sgx/abi/entry.S | ||
*/ | ||
|
||
struct libwu_rs_alloc_meta { | ||
size_t alloc_size; | ||
// Should we put a signatre guard before ptr for oob access? | ||
unsigned char ptr[0]; | ||
}; | ||
|
||
#define META_FROM_PTR(__PTR) (struct libwu_rs_alloc_meta *) \ | ||
((unsigned char *)__PTR - offsetof(struct libwu_rs_alloc_meta, ptr)) | ||
|
||
void *libuw_malloc(size_t size) | ||
{ | ||
struct libwu_rs_alloc_meta *meta; | ||
size_t alloc_size = size + sizeof(struct libwu_rs_alloc_meta); | ||
meta = (void *)__rust_c_alloc(alloc_size, sizeof(size_t)); | ||
if (!meta) { | ||
return NULL; | ||
} | ||
meta->alloc_size = alloc_size; | ||
return (void *)meta->ptr; | ||
} | ||
|
||
void libuw_free(void *p) | ||
{ | ||
struct libwu_rs_alloc_meta *meta; | ||
if (!p) { | ||
return; | ||
} | ||
meta = META_FROM_PTR(p); | ||
__rust_c_dealloc((unsigned char *)meta, meta->alloc_size, sizeof(size_t)); | ||
} |
Oops, something went wrong.