#include "common.h" namespace bpftrace { namespace test { namespace codegen { TEST(codegen, struct_string_ptr) { auto expected = R"EXPECTED(; Function Attrs: nounwind declare i64 @llvm.bpf.pseudo(i64, i64) #0 ; Function Attrs: argmemonly nounwind declare void @llvm.lifetime.start.p0i8(i64, i8* nocapture) #1 define i64 @"kprobe:f"(i8* nocapture readnone) local_unnamed_addr section "s_kprobe:f_1" { entry: %"@mystr_key" = alloca i64, align 8 %Foo.str = alloca i64, align 8 %str = alloca [64 x i8], align 1 %"$foo" = alloca i64, align 8 %tmpcast = bitcast i64* %"$foo" to [8 x i8]* %1 = bitcast i64* %"$foo" to i8* call void @llvm.lifetime.start.p0i8(i64 -1, i8* nonnull %1) store i64 0, [8 x i8]* %tmpcast, align 8 call void @llvm.lifetime.start.p0i8(i64 -1, i8* nonnull %1) %2 = load i64, i64 addrspace(64)* null, align 536870912 store i64 %2, i64* %"$foo", align 8 %3 = getelementptr inbounds [64 x i8], [64 x i8]* %str, i64 0, i64 0 call void @llvm.lifetime.start.p0i8(i64 -1, i8* nonnull %3) call void @llvm.memset.p0i8.i64(i8* nonnull %3, i8 0, i64 64, i32 1, i1 false) %4 = bitcast i64* %Foo.str to i8* call void @llvm.lifetime.start.p0i8(i64 -1, i8* nonnull %4) %probe_read = call i64 inttoptr (i64 4 to i64 (i8*, i64, i8*)*)(i64* nonnull %Foo.str, i64 8, [8 x i8]* nonnull %tmpcast) %5 = load i64, i64* %Foo.str, align 8 call void @llvm.lifetime.end.p0i8(i64 -1, i8* nonnull %4) %probe_read_str = call i64 inttoptr (i64 45 to i64 (i8*, i64, i8*)*)([64 x i8]* nonnull %str, i64 64, i64 %5) %6 = bitcast i64* %"@mystr_key" to i8* call void @llvm.lifetime.start.p0i8(i64 -1, i8* nonnull %6) store i64 0, i64* %"@mystr_key", align 8 %pseudo = call i64 @llvm.bpf.pseudo(i64 1, i64 1) %update_elem = call i64 inttoptr (i64 2 to i64 (i8*, i8*, i8*, i64)*)(i64 %pseudo, i64* nonnull %"@mystr_key", [64 x i8]* nonnull %str, i64 0) call void @llvm.lifetime.end.p0i8(i64 -1, i8* nonnull %6) call void @llvm.lifetime.end.p0i8(i64 -1, i8* nonnull %3) ret i64 0 } ; Function Attrs: argmemonly nounwind declare void @llvm.memset.p0i8.i64(i8* nocapture writeonly, i8, i64, i32, i1) #1 ; Function Attrs: argmemonly nounwind declare void @llvm.lifetime.end.p0i8(i64, i8* nocapture) #1 attributes #0 = { nounwind } attributes #1 = { argmemonly nounwind } )EXPECTED"; test("struct Foo { char *str; }" "kprobe:f" "{" " $foo = (Foo)0;" " @mystr = str($foo.str);" "}", expected); expected = R"EXPECTED(; Function Attrs: nounwind declare i64 @llvm.bpf.pseudo(i64, i64) #0 ; Function Attrs: argmemonly nounwind declare void @llvm.lifetime.start.p0i8(i64, i8* nocapture) #1 define i64 @"kprobe:f"(i8* nocapture readnone) local_unnamed_addr section "s_kprobe:f_1" { entry: %"@mystr_key" = alloca i64, align 8 %Foo.str = alloca i64, align 8 %str = alloca [64 x i8], align 1 %1 = getelementptr inbounds [64 x i8], [64 x i8]* %str, i64 0, i64 0 call void @llvm.lifetime.start.p0i8(i64 -1, i8* nonnull %1) call void @llvm.memset.p0i8.i64(i8* nonnull %1, i8 0, i64 64, i32 1, i1 false) %2 = bitcast i64* %Foo.str to i8* call void @llvm.lifetime.start.p0i8(i64 -1, i8* nonnull %2) %probe_read = call i64 inttoptr (i64 4 to i64 (i8*, i64, i8*)*)(i64* nonnull %Foo.str, i64 8, i64 0) %3 = load i64, i64* %Foo.str, align 8 call void @llvm.lifetime.end.p0i8(i64 -1, i8* nonnull %2) %probe_read_str = call i64 inttoptr (i64 45 to i64 (i8*, i64, i8*)*)([64 x i8]* nonnull %str, i64 64, i64 %3) %4 = bitcast i64* %"@mystr_key" to i8* call void @llvm.lifetime.start.p0i8(i64 -1, i8* nonnull %4) store i64 0, i64* %"@mystr_key", align 8 %pseudo = call i64 @llvm.bpf.pseudo(i64 1, i64 1) %update_elem = call i64 inttoptr (i64 2 to i64 (i8*, i8*, i8*, i64)*)(i64 %pseudo, i64* nonnull %"@mystr_key", [64 x i8]* nonnull %str, i64 0) call void @llvm.lifetime.end.p0i8(i64 -1, i8* nonnull %4) call void @llvm.lifetime.end.p0i8(i64 -1, i8* nonnull %1) ret i64 0 } ; Function Attrs: argmemonly nounwind declare void @llvm.memset.p0i8.i64(i8* nocapture writeonly, i8, i64, i32, i1) #1 ; Function Attrs: argmemonly nounwind declare void @llvm.lifetime.end.p0i8(i64, i8* nocapture) #1 attributes #0 = { nounwind } attributes #1 = { argmemonly nounwind } )EXPECTED"; test("struct Foo { char *str; }" "kprobe:f" "{" " $foo = (Foo*)0;" " @mystr = str($foo->str);" "}", expected); } } // namespace codegen } // namespace test } // namespace bpftrace