semantic tests, codegen tests for LLVM 6

tests/codegen.cpp

@@ -36,6 +36,8 @@
 #include "codegen/call_reg.cpp"
 #include "codegen/call_stats.cpp"
 #include "codegen/call_str.cpp"
+#include "codegen/call_str_2_lit.cpp"
+#include "codegen/call_str_2_expr.cpp"
 #include "codegen/call_sum.cpp"
 #include "codegen/call_system.cpp"
 #include "codegen/call_time.cpp"

tests/codegen/call_str_2_expr.cpp

+#include "common.h"
+
+namespace bpftrace {
+namespace test {
+namespace codegen {
+
+TEST(codegen, call_str_2_expr)
+{
+  test("kprobe:f { @x = str(arg0, arg1) }",
+
+#if LLVM_VERSION_MAJOR < 7
+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*) local_unnamed_addr section "s_kprobe:f_1" {
+entry:
+  %"@x_key" = alloca i64, align 8
+  %arg0 = alloca i64, align 8
+  %str = alloca [64 x i8], align 1
+  %arg1 = alloca i64, align 8
+  %1 = bitcast i64* %arg1 to i8*
+  call void @llvm.lifetime.start.p0i8(i64 -1, i8* nonnull %1)
+  %2 = getelementptr i8, i8* %0, i64 104
+  %probe_read = call i64 inttoptr (i64 4 to i64 (i8*, i64, i8*)*)(i64* nonnull %arg1, i64 8, i8* %2)
+  %3 = load i64, i64* %arg1, align 8
+  call void @llvm.lifetime.end.p0i8(i64 -1, i8* nonnull %1)
+  %4 = add i64 %3, 1
+  %5 = icmp ult i64 %4, 64
+  %str.min.select = select i1 %5, i64 %4, i64 64
+  %6 = getelementptr inbounds [64 x i8], [64 x i8]* %str, i64 0, i64 0
+  call void @llvm.lifetime.start.p0i8(i64 -1, i8* nonnull %6)
+  call void @llvm.memset.p0i8.i64(i8* nonnull %6, i8 0, i64 64, i32 1, i1 false)
+  %7 = bitcast i64* %arg0 to i8*
+  call void @llvm.lifetime.start.p0i8(i64 -1, i8* nonnull %7)
+  %8 = getelementptr i8, i8* %0, i64 112
+  %probe_read1 = call i64 inttoptr (i64 4 to i64 (i8*, i64, i8*)*)(i64* nonnull %arg0, i64 8, i8* %8)
+  %9 = load i64, i64* %arg0, align 8
+  call void @llvm.lifetime.end.p0i8(i64 -1, i8* nonnull %7)
+  %probe_read_str = call i64 inttoptr (i64 45 to i64 (i8*, i64, i8*)*)([64 x i8]* nonnull %str, i64 %str.min.select, i64 %9)
+  %10 = bitcast i64* %"@x_key" to i8*
+  call void @llvm.lifetime.start.p0i8(i64 -1, i8* nonnull %10)
+  store i64 0, i64* %"@x_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 %"@x_key", [64 x i8]* nonnull %str, i64 0)
+  call void @llvm.lifetime.end.p0i8(i64 -1, i8* nonnull %10)
+  call void @llvm.lifetime.end.p0i8(i64 -1, i8* nonnull %6)
+  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");
+#endif
+}
+
+} // namespace codegen
+} // namespace test
+} // namespace bpftrace

tests/codegen/call_str_2_lit.cpp

+#include "common.h"
+
+namespace bpftrace {
+namespace test {
+namespace codegen {
+
+TEST(codegen, call_str_2_lit)
+{
+  test("kprobe:f { @x = str(arg0, 6) }",
+
+#if LLVM_VERSION_MAJOR < 7
+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*) local_unnamed_addr section "s_kprobe:f_1" {
+entry:
+  %"@x_key" = alloca i64, align 8
+  %arg0 = 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* %arg0 to i8*
+  call void @llvm.lifetime.start.p0i8(i64 -1, i8* nonnull %2)
+  %3 = getelementptr i8, i8* %0, i64 112
+  %probe_read = call i64 inttoptr (i64 4 to i64 (i8*, i64, i8*)*)(i64* nonnull %arg0, i64 8, i8* %3)
+  %4 = load i64, i64* %arg0, 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 7, i64 %4)
+  %5 = bitcast i64* %"@x_key" to i8*
+  call void @llvm.lifetime.start.p0i8(i64 -1, i8* nonnull %5)
+  store i64 0, i64* %"@x_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 %"@x_key", [64 x i8]* nonnull %str, i64 0)
+  call void @llvm.lifetime.end.p0i8(i64 -1, i8* nonnull %5)
+  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");
+#endif
+}
+
+} // namespace codegen
+} // namespace test
+} // namespace bpftrace

tests/semantic_analyser.cpp

@@ -230,6 +230,19 @@ TEST(semantic_analyser, call_str)
   test("kprobe:f { str(\"hello\"); }", 10);
 }
 
+TEST(semantic_analyser, call_str_2_lit)
+{
+  test("kprobe:f { str(arg0, 3); }", 0);
+  test("kprobe:f { @x = str(arg0, 3); }", 0);
+  test("kprobe:f { str(arg0, \"hello\"); }", 10);
+}
+
+TEST(semantic_analyser, call_str_2_expr)
+{
+  test("kprobe:f { str(arg0, arg1); }", 0);
+  test("kprobe:f { @x = str(arg0, arg1); }", 0);
+}
+
 TEST(semantic_analyser, call_sym)
 {
   test("kprobe:f { sym(arg0); }", 0);