189428c8fc
This mostly follows LLVM's InstrProfReader.cpp error handling. Previously, attempting to merge corrupted profile data would result in crashes. See https://crbug.com/1216811#c4. Reviewed By: rnk Differential Revision: https://reviews.llvm.org/D104050
62 lines
1.4 KiB
C
62 lines
1.4 KiB
C
// RUN: rm -f %t.profraw
|
|
// RUN: touch %t.profraw
|
|
// RUN: %clang_profgen -o %t %s
|
|
// RUN: %t %t.profraw
|
|
// RUN: %t %t.profraw modifyfile
|
|
// RUN: cp %t.profraw %t.profraw.old
|
|
// RUN: %t %t.profraw 2>&1 | FileCheck %s
|
|
// RUN: diff %t.profraw %t.profraw.old
|
|
// CHECK: Invalid profile data to merge
|
|
|
|
#include <errno.h>
|
|
#include <stdio.h>
|
|
#include <stdint.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/mman.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/stat.h>
|
|
#include <fcntl.h>
|
|
#include <unistd.h>
|
|
|
|
|
|
void __llvm_profile_set_file_object(FILE* File, int EnableMerge);
|
|
|
|
void bail(const char* str) {
|
|
fprintf(stderr, "%s %s\n", str, strerror(errno));
|
|
exit(1);
|
|
}
|
|
|
|
int main(int argc, char** argv) {
|
|
if (argc == 3) {
|
|
int fd = open(argv[1], O_RDWR);
|
|
if (fd == -1)
|
|
bail("open");
|
|
|
|
struct stat st;
|
|
if (stat(argv[1], &st))
|
|
bail("stat");
|
|
uint64_t FileSize = st.st_size;
|
|
|
|
char* Buf = (char *) mmap(NULL, FileSize, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
|
|
if (Buf == MAP_FAILED)
|
|
bail("mmap");
|
|
|
|
// We're trying to make the first CounterPtr invalid.
|
|
// 10 64-bit words as header.
|
|
// CounterPtr is the third 64-bit word field.
|
|
memset(&Buf[10 * 8 + 2 * 8], 0xAB, 8);
|
|
|
|
if (munmap(Buf, FileSize))
|
|
bail("munmap");
|
|
|
|
if (close(fd))
|
|
bail("close");
|
|
} else {
|
|
FILE* f = fopen(argv[1], "r+b");
|
|
if (!f)
|
|
bail("fopen");
|
|
__llvm_profile_set_file_object(f, 1);
|
|
}
|
|
}
|