1// RUN: %clang_cc1 -fblocks -emit-llvm %s -o %t
2// RUN: FileCheck %s -input-file=%t -check-prefix=CHECK1
3// RUN: FileCheck %s -input-file=%t -check-prefix=CHECK2
4
5struct A {
6  int a;
7  float b;
8  char c;
9};
10
11void test_nest_captured_stmt(int param, int size, int param_arr[size]) {
12  int w;
13  int arr[param][size];
14  // CHECK1: %struct.anon{{.*}} = type { [[INT:i.+]]*, [[INT]]*, [[SIZE_TYPE:i.+]], [[INT]]**, [[INT]]*, [[SIZE_TYPE]], [[SIZE_TYPE]], [[INT]]* }
15  // CHECK1: %struct.anon{{.*}} = type { [[INT]]*, [[INT]]*, [[INT]]**, [[INT]]*, [[SIZE_TYPE]], [[INT]]**, [[INT]]*, [[SIZE_TYPE]], [[SIZE_TYPE]], [[INT]]* }
16  // CHECK1: [[T:%struct.anon.*]] = type { [[INT]]*, [[INT]]*, %struct.A*, [[INT]]**, [[INT]]*, [[SIZE_TYPE]], [[INT]]**, [[INT]]*, [[SIZE_TYPE]], [[SIZE_TYPE]], [[INT]]* }
17  #pragma clang __debug captured
18  {
19    int x;
20    int *y = &w;
21    #pragma clang __debug captured
22    {
23      struct A z;
24      #pragma clang __debug captured
25      {
26        w = x = z.a = 1;
27        *y = param;
28        z.b = 0.1f;
29        z.c = 'c';
30        param_arr[size - 1] = 2;
31        arr[10][z.a] = 12;
32
33        // CHECK1: define internal {{.*}}void @__captured_stmt{{.*}}([[T]]
34        // CHECK1: [[PARAM_ARR_SIZE_REF:%.+]] = getelementptr inbounds [[T]], [[T]]* {{.+}}, i{{[0-9]+}} 0, i{{[0-9]+}} 5
35        // CHECK1: [[PARAM_ARR_SIZE:%.+]] = load [[SIZE_TYPE]], [[SIZE_TYPE]]* [[PARAM_ARR_SIZE_REF]]
36        // CHECK1: [[ARR_SIZE1_REF:%.+]] = getelementptr inbounds [[T]], [[T]]* {{.+}}, i{{[0-9]+}} 0, i{{[0-9]+}} 8
37        // CHECK1: [[ARR_SIZE1:%.+]] = load [[SIZE_TYPE]], [[SIZE_TYPE]]* [[ARR_SIZE1_REF]]
38        // CHECK1: [[ARR_SIZE2_REF:%.+]] = getelementptr inbounds [[T]], [[T]]* {{.+}}, i{{[0-9]+}} 0, i{{[0-9]+}} 9
39        // CHECK1: [[ARR_SIZE2:%.+]] = load [[SIZE_TYPE]], [[SIZE_TYPE]]* [[ARR_SIZE2_REF]]
40        //
41        // CHECK1: getelementptr inbounds [[T]], [[T]]* {{.*}}, i{{[0-9]+}} 0, i{{[0-9]+}} 2
42        // CHECK1-NEXT: load %struct.A*, %struct.A**
43        // CHECK1-NEXT: getelementptr inbounds %struct.A, %struct.A*
44        // CHECK1-NEXT: store i{{.+}} 1
45        //
46        // CHECK1: getelementptr inbounds [[T]], [[T]]* {{.*}}, i{{[0-9]+}} 0, i{{[0-9]+}} 1
47        // CHECK1-NEXT: load i{{[0-9]+}}*, i{{[0-9]+}}**
48        // CHECK1-NEXT: store i{{[0-9]+}} 1
49        //
50        // CHECK1: getelementptr inbounds [[T]], [[T]]* {{.*}}, i{{[0-9]+}} 0, i{{[0-9]+}} 0
51        // CHECK1-NEXT: load i{{[0-9]+}}*, i{{[0-9]+}}**
52        // CHECK1-NEXT: store i{{[0-9]+}} 1
53        //
54        // CHECK1: getelementptr inbounds [[T]], [[T]]* {{.*}}, i{{[0-9]+}} 0, i{{[0-9]+}} 4
55        // CHECK1-NEXT: load i{{[0-9]+}}*, i{{[0-9]+}}**
56        // CHECK1-NEXT: load i{{[0-9]+}}, i{{[0-9]+}}*
57        // CHECK1-NEXT: getelementptr inbounds [[T]], [[T]]* {{.*}}, i{{[0-9]+}} 0, i{{[0-9]+}} 3
58        // CHECK1-NEXT: load i{{[0-9]+}}**, i{{[0-9]+}}***
59        // CHECK1-NEXT: load i{{[0-9]+}}*, i{{[0-9]+}}**
60        // CHECK1-NEXT: store i{{[0-9]+}}
61        //
62        // CHECK1: getelementptr inbounds [[T]], [[T]]* {{.*}}, i{{[0-9]+}} 0, i{{[0-9]+}} 2
63        // CHECK1-NEXT: load %struct.A*, %struct.A**
64        // CHECK1-NEXT: getelementptr inbounds %struct.A, %struct.A*
65        // CHECK1-NEXT: store float
66        //
67        // CHECK1: getelementptr inbounds [[T]], [[T]]* {{.*}}, i{{[0-9]+}} 0, i{{[0-9]+}} 2
68        // CHECK1-NEXT: load %struct.A*, %struct.A**
69        // CHECK1-NEXT: getelementptr inbounds %struct.A, %struct.A*
70        // CHECK1-NEXT: store i8 99
71        //
72        // CHECK1: [[SIZE_ADDR_REF:%.*]] = getelementptr inbounds [[T]], [[T]]* {{.*}}, i{{.+}} 0, i{{.+}} 7
73        // CHECK1-DAG: [[SIZE_ADDR:%.*]] = load i{{.+}}*, i{{.+}}** [[SIZE_ADDR_REF]]
74        // CHECK1-DAG: [[SIZE:%.*]] = load i{{.+}}, i{{.+}}* [[SIZE_ADDR]]
75        // CHECK1-DAG: [[PARAM_ARR_IDX:%.*]] = sub nsw i{{.+}} [[SIZE]], 1
76        // CHECK1-DAG: [[PARAM_ARR_ADDR_REF:%.*]] = getelementptr inbounds [[T]], [[T]]* {{.*}}, i{{.+}} 0, i{{.+}} 6
77        // CHECK1-DAG: [[PARAM_ARR_ADDR:%.*]] = load i{{.+}}**, i{{.+}}*** [[PARAM_ARR_ADDR_REF]]
78        // CHECK1-DAG: [[PARAM_ARR:%.*]] = load i{{.+}}*, i{{.+}}** [[PARAM_ARR_ADDR]]
79        // CHECK1-DAG: [[PARAM_ARR_SIZE_MINUS_1_ADDR:%.*]] = getelementptr inbounds i{{.+}}, i{{.+}}* [[PARAM_ARR]], i{{.*}}
80        // CHECK1: store i{{.+}} 2, i{{.+}}* [[PARAM_ARR_SIZE_MINUS_1_ADDR]]
81        //
82        // CHECK1: [[Z_ADDR_REF:%.*]] = getelementptr inbounds [[T]], [[T]]* {{.*}}, i{{.+}} 0, i{{.+}} 2
83        // CHECK1-DAG: [[Z_ADDR:%.*]] = load %struct.A*, %struct.A** [[Z_ADDR_REF]]
84        // CHECK1-DAG: [[Z_A_ADDR:%.*]] = getelementptr inbounds %struct.A, %struct.A* [[Z_ADDR]], i{{.+}} 0, i{{.+}} 0
85        // CHECK1-DAG: [[ARR_IDX_2:%.*]] = load i{{.+}}, i{{.+}}* [[Z_A_ADDR]]
86        // CHECK1-DAG: [[ARR_ADDR_REF:%.*]] = getelementptr inbounds [[T]], [[T]]* {{.*}}, i{{.+}} 0, i{{.+}} 10
87        // CHECK1-DAG: [[ARR_ADDR:%.*]] = load i{{.+}}*, i{{.+}}** [[ARR_ADDR_REF]]
88        // CHECK1-DAG: [[ARR_IDX_1:%.*]] = mul {{.*}} 10
89        // CHECK1-DAG: [[ARR_10_ADDR:%.*]] = getelementptr inbounds i{{.+}}, i{{.+}}* [[ARR_ADDR]], i{{.*}} [[ARR_IDX_1]]
90        // CHECK1-DAG: [[ARR_10_Z_A_ADDR:%.*]] = getelementptr inbounds i{{.+}}, i{{.+}}* [[ARR_10_ADDR]], i{{.*}}
91        // CHECK1: store i{{.+}} 12, i{{.+}}* [[ARR_10_Z_A_ADDR]]
92      }
93    }
94  }
95}
96
97void test_nest_block() {
98  __block int x;
99  int y;
100  ^{
101    int z;
102    x = z;
103    #pragma clang __debug captured
104    {
105      z = y; // OK
106    }
107  }();
108
109  // CHECK2: define internal {{.*}}void @{{.*}}test_nest_block_block_invoke
110  //
111  // CHECK2: [[Z:%[0-9a-z_]*]] = alloca i{{[0-9]+}},
112  // CHECK2: alloca %struct.anon{{.*}}
113  //
114  // CHECK2: store i{{[0-9]+}}
115  // CHECK2: store i{{[0-9]+}}* [[Z]]
116  //
117  // CHECK2: getelementptr inbounds %struct.anon
118  // CHECK2-NEXT: getelementptr inbounds
119  // CHECK2-NEXT: store i{{[0-9]+}}*
120  //
121  // CHECK2: call void @__captured_stmt
122
123  int a;
124  #pragma clang __debug captured
125  {
126    __block int b;
127    int c;
128    __block int d;
129    ^{
130      b = a;
131      b = c;
132      b = d;
133    }();
134  }
135
136  // CHECK2: alloca %struct.__block_byref_b
137  // CHECK2-NEXT: [[C:%[0-9a-z_]*]] = alloca i{{[0-9]+}}
138  // CHECK2-NEXT: alloca %struct.__block_byref_d
139  //
140  // CHECK2: bitcast %struct.__block_byref_b*
141  // CHECK2-NEXT: store i8*
142  //
143  // CHECK2: [[CapA:%[0-9a-z_.]*]] = getelementptr inbounds {{.*}}, i{{[0-9]+}} 0, i{{[0-9]+}} 7
144  //
145  // CHECK2: getelementptr inbounds %struct.anon{{.*}}, i{{[0-9]+}} 0, i{{[0-9]+}} 0
146  // CHECK2: load i{{[0-9]+}}*, i{{[0-9]+}}**
147  // CHECK2: load i{{[0-9]+}}, i{{[0-9]+}}*
148  // CHECK2: store i{{[0-9]+}} {{.*}}, i{{[0-9]+}}* [[CapA]]
149  //
150  // CHECK2: [[CapC:%[0-9a-z_.]*]] = getelementptr inbounds {{.*}}, i{{[0-9]+}} 0, i{{[0-9]+}} 8
151  // CHECK2-NEXT: [[Val:%[0-9a-z_]*]] = load i{{[0-9]+}}, i{{[0-9]+}}* [[C]]
152  // CHECK2-NEXT: store i{{[0-9]+}} [[Val]], i{{[0-9]+}}* [[CapC]]
153  //
154  // CHECK2: bitcast %struct.__block_byref_d*
155  // CHECK2-NEXT: store i8*
156}
157