llvm-project/compiler-rt/test/profile/Linux/corrupted-profile.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>

enum ValueKind {
#define VALUE_PROF_KIND(Enumerator, Value, Descr) Enumerator = Value,
#include "profile/InstrProfData.inc"
};

typedef struct __llvm_profile_header {
#define INSTR_PROF_RAW_HEADER(Type, Name, Initializer) Type Name;
#include "profile/InstrProfData.inc"
} __llvm_profile_header;

typedef void *IntPtrT;
typedef struct __llvm_profile_data {
#define INSTR_PROF_DATA(Type, LLVMType, Name, Initializer) Type Name;
#include "profile/InstrProfData.inc"
} __llvm_profile_data;

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");

    __llvm_profile_header *Header = (__llvm_profile_header *)Buf;
    __llvm_profile_data *SrcDataStart =
        (__llvm_profile_data *)(Buf + sizeof(__llvm_profile_header) +
                                Header->BinaryIdsSize);
    memset(&SrcDataStart->CounterPtr, 0xAB, sizeof(SrcDataStart->CounterPtr));

    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);
  }
}