TheRock/compiler/amd-llvm/clang/test/CodeGen/builtin-counted-by-ref.c

// NOTE: Assertions have been autogenerated by utils/update_cc_test_checks.py UTC_ARGS: --version 5
// RUN: %clang_cc1 -triple x86_64-unknown-unknown -emit-llvm -o - %s | FileCheck %s --check-prefix=X86_64
// RUN: %clang_cc1 -triple i386-unknown-unknown -emit-llvm -o - %s | FileCheck %s --check-prefix=I386

struct a {
  char x;
  short count;
  int array[] __attribute__((counted_by(count)));
};

// X86_64-LABEL: define dso_local ptr @test1(
// X86_64-SAME: i32 noundef [[SIZE:%.*]]) #[[ATTR0:[0-9]+]] {
// X86_64-NEXT:  [[ENTRY:.*:]]
// X86_64-NEXT:    [[SIZE_ADDR:%.*]] = alloca i32, align 4
// X86_64-NEXT:    [[P:%.*]] = alloca ptr, align 8
// X86_64-NEXT:    store i32 [[SIZE]], ptr [[SIZE_ADDR]], align 4
// X86_64-NEXT:    [[TMP0:%.*]] = load i32, ptr [[SIZE_ADDR]], align 4
// X86_64-NEXT:    [[CONV:%.*]] = sext i32 [[TMP0]] to i64
// X86_64-NEXT:    [[MUL:%.*]] = mul i64 4, [[CONV]]
// X86_64-NEXT:    [[ADD:%.*]] = add i64 4, [[MUL]]
// X86_64-NEXT:    [[CALL:%.*]] = call ptr @malloc(i64 noundef [[ADD]]) #[[ATTR2:[0-9]+]]
// X86_64-NEXT:    store ptr [[CALL]], ptr [[P]], align 8
// X86_64-NEXT:    [[TMP1:%.*]] = load i32, ptr [[SIZE_ADDR]], align 4
// X86_64-NEXT:    [[CONV1:%.*]] = trunc i32 [[TMP1]] to i16
// X86_64-NEXT:    [[TMP2:%.*]] = load ptr, ptr [[P]], align 8
// X86_64-NEXT:    [[DOT_COUNTED_BY_GEP:%.*]] = getelementptr inbounds [[STRUCT_A:%.*]], ptr [[TMP2]], i32 0, i32 1
// X86_64-NEXT:    store i16 [[CONV1]], ptr [[DOT_COUNTED_BY_GEP]], align 2
// X86_64-NEXT:    [[TMP3:%.*]] = load ptr, ptr [[P]], align 8
// X86_64-NEXT:    ret ptr [[TMP3]]
//
// I386-LABEL: define dso_local ptr @test1(
// I386-SAME: i32 noundef [[SIZE:%.*]]) #[[ATTR0:[0-9]+]] {
// I386-NEXT:  [[ENTRY:.*:]]
// I386-NEXT:    [[SIZE_ADDR:%.*]] = alloca i32, align 4
// I386-NEXT:    [[P:%.*]] = alloca ptr, align 4
// I386-NEXT:    store i32 [[SIZE]], ptr [[SIZE_ADDR]], align 4
// I386-NEXT:    [[TMP0:%.*]] = load i32, ptr [[SIZE_ADDR]], align 4
// I386-NEXT:    [[MUL:%.*]] = mul i32 4, [[TMP0]]
// I386-NEXT:    [[ADD:%.*]] = add i32 4, [[MUL]]
// I386-NEXT:    [[CALL:%.*]] = call ptr @malloc(i32 noundef [[ADD]]) #[[ATTR2:[0-9]+]]
// I386-NEXT:    store ptr [[CALL]], ptr [[P]], align 4
// I386-NEXT:    [[TMP1:%.*]] = load i32, ptr [[SIZE_ADDR]], align 4
// I386-NEXT:    [[CONV:%.*]] = trunc i32 [[TMP1]] to i16
// I386-NEXT:    [[TMP2:%.*]] = load ptr, ptr [[P]], align 4
// I386-NEXT:    [[DOT_COUNTED_BY_GEP:%.*]] = getelementptr inbounds [[STRUCT_A:%.*]], ptr [[TMP2]], i32 0, i32 1
// I386-NEXT:    store i16 [[CONV]], ptr [[DOT_COUNTED_BY_GEP]], align 2
// I386-NEXT:    [[TMP3:%.*]] = load ptr, ptr [[P]], align 4
// I386-NEXT:    ret ptr [[TMP3]]
//
struct a *test1(int size) {
  struct a *p = __builtin_malloc(sizeof(struct a) + sizeof(int) * size);

  *__builtin_counted_by_ref(p->array) = size;
  return p;
}

struct b {
  int _filler;
  struct {
    int __filler;
    struct {
      int ___filler;
      struct {
        char count;
      };
    };
  };
  struct {
    int filler_;
    struct {
      int filler__;
      struct {
        long array[] __attribute__((counted_by(count)));
      };
    };
  };
};

// X86_64-LABEL: define dso_local ptr @test2(
// X86_64-SAME: i32 noundef [[SIZE:%.*]]) #[[ATTR0]] {
// X86_64-NEXT:  [[ENTRY:.*:]]
// X86_64-NEXT:    [[SIZE_ADDR:%.*]] = alloca i32, align 4
// X86_64-NEXT:    [[P:%.*]] = alloca ptr, align 8
// X86_64-NEXT:    store i32 [[SIZE]], ptr [[SIZE_ADDR]], align 4
// X86_64-NEXT:    [[TMP0:%.*]] = load i32, ptr [[SIZE_ADDR]], align 4
// X86_64-NEXT:    [[CONV:%.*]] = sext i32 [[TMP0]] to i64
// X86_64-NEXT:    [[MUL:%.*]] = mul i64 4, [[CONV]]
// X86_64-NEXT:    [[ADD:%.*]] = add i64 4, [[MUL]]
// X86_64-NEXT:    [[CALL:%.*]] = call ptr @malloc(i64 noundef [[ADD]]) #[[ATTR2]]
// X86_64-NEXT:    store ptr [[CALL]], ptr [[P]], align 8
// X86_64-NEXT:    [[TMP1:%.*]] = load i32, ptr [[SIZE_ADDR]], align 4
// X86_64-NEXT:    [[CONV1:%.*]] = trunc i32 [[TMP1]] to i8
// X86_64-NEXT:    [[TMP2:%.*]] = load ptr, ptr [[P]], align 8
// X86_64-NEXT:    [[DOT_COUNTED_BY_GEP:%.*]] = getelementptr inbounds [[STRUCT_B:%.*]], ptr [[TMP2]], i32 0, i32 1, i32 1, i32 1, i32 0
// X86_64-NEXT:    store i8 [[CONV1]], ptr [[DOT_COUNTED_BY_GEP]], align 1
// X86_64-NEXT:    [[TMP3:%.*]] = load ptr, ptr [[P]], align 8
// X86_64-NEXT:    ret ptr [[TMP3]]
//
// I386-LABEL: define dso_local ptr @test2(
// I386-SAME: i32 noundef [[SIZE:%.*]]) #[[ATTR0]] {
// I386-NEXT:  [[ENTRY:.*:]]
// I386-NEXT:    [[SIZE_ADDR:%.*]] = alloca i32, align 4
// I386-NEXT:    [[P:%.*]] = alloca ptr, align 4
// I386-NEXT:    store i32 [[SIZE]], ptr [[SIZE_ADDR]], align 4
// I386-NEXT:    [[TMP0:%.*]] = load i32, ptr [[SIZE_ADDR]], align 4
// I386-NEXT:    [[MUL:%.*]] = mul i32 4, [[TMP0]]
// I386-NEXT:    [[ADD:%.*]] = add i32 4, [[MUL]]
// I386-NEXT:    [[CALL:%.*]] = call ptr @malloc(i32 noundef [[ADD]]) #[[ATTR2]]
// I386-NEXT:    store ptr [[CALL]], ptr [[P]], align 4
// I386-NEXT:    [[TMP1:%.*]] = load i32, ptr [[SIZE_ADDR]], align 4
// I386-NEXT:    [[CONV:%.*]] = trunc i32 [[TMP1]] to i8
// I386-NEXT:    [[TMP2:%.*]] = load ptr, ptr [[P]], align 4
// I386-NEXT:    [[DOT_COUNTED_BY_GEP:%.*]] = getelementptr inbounds [[STRUCT_B:%.*]], ptr [[TMP2]], i32 0, i32 1, i32 1, i32 1, i32 0
// I386-NEXT:    store i8 [[CONV]], ptr [[DOT_COUNTED_BY_GEP]], align 1
// I386-NEXT:    [[TMP3:%.*]] = load ptr, ptr [[P]], align 4
// I386-NEXT:    ret ptr [[TMP3]]
//
struct b *test2(int size) {
  struct b *p = __builtin_malloc(sizeof(struct a) + sizeof(int) * size);

  *__builtin_counted_by_ref(p->array) = size;
  return p;
}

struct c {
  char x;
  short count;
  int array[];
};

// X86_64-LABEL: define dso_local ptr @test3(
// X86_64-SAME: i32 noundef [[SIZE:%.*]]) #[[ATTR0]] {
// X86_64-NEXT:  [[ENTRY:.*:]]
// X86_64-NEXT:    [[SIZE_ADDR:%.*]] = alloca i32, align 4
// X86_64-NEXT:    [[P:%.*]] = alloca ptr, align 8
// X86_64-NEXT:    [[__IGNORED:%.*]] = alloca i64, align 8
// X86_64-NEXT:    store i32 [[SIZE]], ptr [[SIZE_ADDR]], align 4
// X86_64-NEXT:    [[TMP0:%.*]] = load i32, ptr [[SIZE_ADDR]], align 4
// X86_64-NEXT:    [[CONV:%.*]] = sext i32 [[TMP0]] to i64
// X86_64-NEXT:    [[MUL:%.*]] = mul i64 4, [[CONV]]
// X86_64-NEXT:    [[ADD:%.*]] = add i64 4, [[MUL]]
// X86_64-NEXT:    [[CALL:%.*]] = call ptr @malloc(i64 noundef [[ADD]]) #[[ATTR2]]
// X86_64-NEXT:    store ptr [[CALL]], ptr [[P]], align 8
// X86_64-NEXT:    [[TMP1:%.*]] = load i32, ptr [[SIZE_ADDR]], align 4
// X86_64-NEXT:    [[CONV1:%.*]] = sext i32 [[TMP1]] to i64
// X86_64-NEXT:    store i64 [[CONV1]], ptr [[__IGNORED]], align 8
// X86_64-NEXT:    [[TMP2:%.*]] = load ptr, ptr [[P]], align 8
// X86_64-NEXT:    ret ptr [[TMP2]]
//
// I386-LABEL: define dso_local ptr @test3(
// I386-SAME: i32 noundef [[SIZE:%.*]]) #[[ATTR0]] {
// I386-NEXT:  [[ENTRY:.*:]]
// I386-NEXT:    [[SIZE_ADDR:%.*]] = alloca i32, align 4
// I386-NEXT:    [[P:%.*]] = alloca ptr, align 4
// I386-NEXT:    [[__IGNORED:%.*]] = alloca i32, align 4
// I386-NEXT:    store i32 [[SIZE]], ptr [[SIZE_ADDR]], align 4
// I386-NEXT:    [[TMP0:%.*]] = load i32, ptr [[SIZE_ADDR]], align 4
// I386-NEXT:    [[MUL:%.*]] = mul i32 4, [[TMP0]]
// I386-NEXT:    [[ADD:%.*]] = add i32 4, [[MUL]]
// I386-NEXT:    [[CALL:%.*]] = call ptr @malloc(i32 noundef [[ADD]]) #[[ATTR2]]
// I386-NEXT:    store ptr [[CALL]], ptr [[P]], align 4
// I386-NEXT:    [[TMP1:%.*]] = load i32, ptr [[SIZE_ADDR]], align 4
// I386-NEXT:    store i32 [[TMP1]], ptr [[__IGNORED]], align 4
// I386-NEXT:    [[TMP2:%.*]] = load ptr, ptr [[P]], align 4
// I386-NEXT:    ret ptr [[TMP2]]
//
struct c *test3(int size) {
  struct c *p = __builtin_malloc(sizeof(struct c) + sizeof(int) * size);
  unsigned long int __ignored;

  *_Generic(
    __builtin_counted_by_ref(p->array),
      void *: &__ignored,
      default: __builtin_counted_by_ref(p->array)) = size;

  return p;
}

// Test for pointer member with counted_by attribute
struct d {
  int x;
  short count;
  int *ptr __attribute__((counted_by(count)));
};

// X86_64-LABEL: define dso_local ptr @test4(
// X86_64-SAME: i32 noundef [[SIZE:%.*]], ptr noundef [[DATA:%.*]]) #[[ATTR0]] {
// X86_64-NEXT:  [[ENTRY:.*:]]
// X86_64-NEXT:    [[SIZE_ADDR:%.*]] = alloca i32, align 4
// X86_64-NEXT:    [[DATA_ADDR:%.*]] = alloca ptr, align 8
// X86_64-NEXT:    [[P:%.*]] = alloca ptr, align 8
// X86_64-NEXT:    store i32 [[SIZE]], ptr [[SIZE_ADDR]], align 4
// X86_64-NEXT:    store ptr [[DATA]], ptr [[DATA_ADDR]], align 8
// X86_64-NEXT:    [[CALL:%.*]] = call ptr @malloc(i64 noundef 16) #[[ATTR2]]
// X86_64-NEXT:    store ptr [[CALL]], ptr [[P]], align 8
// X86_64-NEXT:    [[TMP0:%.*]] = load ptr, ptr [[DATA_ADDR]], align 8
// X86_64-NEXT:    [[TMP1:%.*]] = load ptr, ptr [[P]], align 8
// X86_64-NEXT:    [[PTR:%.*]] = getelementptr inbounds nuw [[STRUCT_D:%.*]], ptr [[TMP1]], i32 0, i32 2
// X86_64-NEXT:    store ptr [[TMP0]], ptr [[PTR]], align 8
// X86_64-NEXT:    [[TMP2:%.*]] = load i32, ptr [[SIZE_ADDR]], align 4
// X86_64-NEXT:    [[CONV:%.*]] = trunc i32 [[TMP2]] to i16
// X86_64-NEXT:    [[TMP3:%.*]] = load ptr, ptr [[P]], align 8
// X86_64-NEXT:    [[DOT_COUNTED_BY_GEP:%.*]] = getelementptr inbounds [[STRUCT_D]], ptr [[TMP3]], i32 0, i32 1
// X86_64-NEXT:    store i16 [[CONV]], ptr [[DOT_COUNTED_BY_GEP]], align 2
// X86_64-NEXT:    [[TMP4:%.*]] = load ptr, ptr [[P]], align 8
// X86_64-NEXT:    ret ptr [[TMP4]]
//
// I386-LABEL: define dso_local ptr @test4(
// I386-SAME: i32 noundef [[SIZE:%.*]], ptr noundef [[DATA:%.*]]) #[[ATTR0]] {
// I386-NEXT:  [[ENTRY:.*:]]
// I386-NEXT:    [[SIZE_ADDR:%.*]] = alloca i32, align 4
// I386-NEXT:    [[DATA_ADDR:%.*]] = alloca ptr, align 4
// I386-NEXT:    [[P:%.*]] = alloca ptr, align 4
// I386-NEXT:    store i32 [[SIZE]], ptr [[SIZE_ADDR]], align 4
// I386-NEXT:    store ptr [[DATA]], ptr [[DATA_ADDR]], align 4
// I386-NEXT:    [[CALL:%.*]] = call ptr @malloc(i32 noundef 12) #[[ATTR2]]
// I386-NEXT:    store ptr [[CALL]], ptr [[P]], align 4
// I386-NEXT:    [[TMP0:%.*]] = load ptr, ptr [[DATA_ADDR]], align 4
// I386-NEXT:    [[TMP1:%.*]] = load ptr, ptr [[P]], align 4
// I386-NEXT:    [[PTR:%.*]] = getelementptr inbounds nuw [[STRUCT_D:%.*]], ptr [[TMP1]], i32 0, i32 2
// I386-NEXT:    store ptr [[TMP0]], ptr [[PTR]], align 4
// I386-NEXT:    [[TMP2:%.*]] = load i32, ptr [[SIZE_ADDR]], align 4
// I386-NEXT:    [[CONV:%.*]] = trunc i32 [[TMP2]] to i16
// I386-NEXT:    [[TMP3:%.*]] = load ptr, ptr [[P]], align 4
// I386-NEXT:    [[DOT_COUNTED_BY_GEP:%.*]] = getelementptr inbounds [[STRUCT_D]], ptr [[TMP3]], i32 0, i32 1
// I386-NEXT:    store i16 [[CONV]], ptr [[DOT_COUNTED_BY_GEP]], align 2
// I386-NEXT:    [[TMP4:%.*]] = load ptr, ptr [[P]], align 4
// I386-NEXT:    ret ptr [[TMP4]]
//
struct d *test4(int size, int *data) {
  struct d *p = __builtin_malloc(sizeof(struct d));
  p->ptr = data;
  *__builtin_counted_by_ref(p->ptr) = size;
  return p;
}

// Test for FAM struct defined nested inside another struct (issue #182575).
// The nested struct is named (not anonymous), so getOuterLexicalRecordContext()
// would incorrectly resolve to 'struct outer' instead of 'struct inner'.
struct outer {
  struct inner {
    int counter;
    int ent[] __attribute__((counted_by(counter)));
  } *entries;
};

// X86_64-LABEL: define dso_local ptr @test5(
// X86_64-SAME: i32 noundef [[COUNT:%.*]]) #[[ATTR0]] {
// X86_64:    [[DOT_COUNTED_BY_GEP:%.*]] = getelementptr inbounds [[STRUCT_INNER:%.*]], ptr {{%.*}}, i32 0, i32 0
// X86_64-NEXT:    store i32 [[TMP1:%.*]], ptr [[DOT_COUNTED_BY_GEP]], align 4
//
// I386-LABEL: define dso_local ptr @test5(
// I386-SAME: i32 noundef [[COUNT:%.*]]) #[[ATTR0]] {
// I386:    [[DOT_COUNTED_BY_GEP:%.*]] = getelementptr inbounds [[STRUCT_INNER:%.*]], ptr {{%.*}}, i32 0, i32 0
// I386-NEXT:    store i32 [[TMP1:%.*]], ptr [[DOT_COUNTED_BY_GEP]], align 4
//
struct inner *test5(int count) {
  struct inner *entries = __builtin_malloc(sizeof(*entries) + count * sizeof(*entries->ent));
  if (entries)
    *__builtin_counted_by_ref(entries->ent) = count;
  return entries;
}