1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
// RUN: %clang_cc1 -std=gnu11 -triple x86_64-unknown-linux-gnu -O2 -emit-llvm -o - %s | FileCheck %s
// Test that counted_by on void* in GNU mode treats void as having size 1 (byte count)
#define __counted_by(f) __attribute__((counted_by(f)))
#define __sized_by(f) __attribute__((sized_by(f)))
struct with_counted_by_void {
int count;
void* buf __counted_by(count);
};
struct with_sized_by_void {
int size;
void* buf __sized_by(size);
};
struct with_counted_by_int {
int count;
int* buf __counted_by(count);
};
// CHECK-LABEL: define dso_local {{.*}}@test_counted_by_void(
// CHECK: %[[COUNT:.*]] = load i32, ptr %s
// CHECK: %[[NARROW:.*]] = tail call i32 @llvm.smax.i32(i32 %[[COUNT]], i32 0)
// CHECK: %[[ZEXT:.*]] = zext nneg i32 %[[NARROW]] to i64
// CHECK: ret i64 %[[ZEXT]]
//
// Verify: counted_by on void* returns the count directly (count * 1 byte)
long long test_counted_by_void(struct with_counted_by_void *s) {
return __builtin_dynamic_object_size(s->buf, 0);
}
// CHECK-LABEL: define dso_local {{.*}}@test_sized_by_void(
// CHECK: %[[SIZE:.*]] = load i32, ptr %s
// CHECK: %[[NARROW:.*]] = tail call i32 @llvm.smax.i32(i32 %[[SIZE]], i32 0)
// CHECK: %[[ZEXT:.*]] = zext nneg i32 %[[NARROW]] to i64
// CHECK: ret i64 %[[ZEXT]]
//
// Verify: sized_by on void* returns the size directly
long long test_sized_by_void(struct with_sized_by_void *s) {
return __builtin_dynamic_object_size(s->buf, 0);
}
// CHECK-LABEL: define dso_local {{.*}}@test_counted_by_int(
// CHECK: %[[COUNT:.*]] = load i32, ptr %s
// CHECK: %[[SEXT:.*]] = sext i32 %[[COUNT]] to i64
// CHECK: %[[SIZE:.*]] = shl nsw i64 %[[SEXT]], 2
// CHECK: ret i64
//
// Verify: counted_by on int* returns count * sizeof(int) = count * 4
long long test_counted_by_int(struct with_counted_by_int *s) {
return __builtin_dynamic_object_size(s->buf, 0);
}
// CHECK-LABEL: define dso_local ptr @test_void_ptr_arithmetic(
// CHECK: %[[BUF:.*]] = load ptr, ptr
// CHECK: %[[EXT:.*]] = sext i32 %offset to i64
// CHECK: %[[PTR:.*]] = getelementptr inbounds i8, ptr %[[BUF]], i64 %[[EXT]]
// CHECK: ret ptr %[[PTR]]
//
// Verify: pointer arithmetic on void* uses i8 (byte offsets), not i32 or other sizes
void* test_void_ptr_arithmetic(struct with_counted_by_void *s, int offset) {
return s->buf + offset; // GNU extension: void* arithmetic
}
|