| #include <stdio.h> | |
| #include <stdlib.h> | |
| #include <stdint.h> | |
| #include "bar.h" | |
| extern void foo(char*); | |
| void bar(char *ptr) { | |
| if (((uintptr_t)ptr >> 56) != 0x1f) { | |
| fprintf(stderr, "Top byte corrupted on Rust -> C FFI boundary!\n"); | |
| exit(1); | |
| } | |
| } | |
| int main(void) | |
| { | |
| float *ptr = alloc_page(); | |
| if (ptr == MAP_FAILED) | |
| { | |
| perror("mmap() failed"); | |
| return EXIT_FAILURE; | |
| } | |
| // Store an arbitrary tag in bits 56-59 of the pointer (where an MTE tag may be), | |
| // and a different value in the ignored top 4 bits. | |
| ptr = (float *)((uintptr_t)ptr | 0x1fl << 56); | |
| if (mte_enabled()) { | |
| set_tag(ptr); | |
| } | |
| ptr[0] = 2.0f; | |
| ptr[1] = 1.5f; | |
| foo(ptr); // should change the contents of the page and call `bar` | |
| if (ptr[0] != 0.5f || ptr[1] != 0.2f) { | |
| fprintf(stderr, "invalid data in memory; expected '0.5 0.2', got '%f %f'\n", | |
| ptr[0], ptr[1]); | |
| return EXIT_FAILURE; | |
| } | |
| return 0; | |
| } |