Move pidl to top-level directory.
[ira/wip.git] / pidl / tests / samba-ndr.pl
1 #!/usr/bin/perl
2 # (C) 2007 Jelmer Vernooij <jelmer@samba.org>
3 # Published under the GNU General Public License
4 use strict;
5 use warnings;
6
7 use Test::More tests => 31;
8 use FindBin qw($RealBin);
9 use lib "$RealBin";
10 use Util;
11 use strict;
12 use Parse::Pidl::Util qw(MyDumper);
13 use Parse::Pidl::Samba4::NDR::Parser qw(check_null_pointer 
14         NeededFunction NeededElement NeededType
15         NeededInterface TypeFunctionName ParseElementPrint); 
16
17 my $output;
18 sub print_fn($) { my $x = shift; $output.=$x; }
19
20 # Test case 1: Simple unique pointer dereference
21
22 $output = "";
23 my $fn = check_null_pointer({ 
24         PARENT => {
25                 ELEMENTS => [
26                         { 
27                                 NAME => "bla",
28                                 LEVELS => [
29                                         { TYPE => "POINTER",
30                                           POINTER_INDEX => 0,
31                                           POINTER_TYPE => "unique" },
32                                         { TYPE => "DATA" }
33                                 ],
34                         },
35                 ]
36         }
37 }, { bla => "r->in.bla" }, \&print_fn, "return;"); 
38
39
40 test_warnings("", sub { $fn->("r->in.bla"); });
41
42 is($output, "if (r->in.bla == NULL) return;");
43
44 # Test case 2: Simple ref pointer dereference
45
46 $output = "";
47 $fn = check_null_pointer({ 
48         PARENT => {
49                 ELEMENTS => [
50                         { 
51                                 NAME => "bla",
52                                 LEVELS => [
53                                         { TYPE => "POINTER",
54                                           POINTER_INDEX => 0,
55                                           POINTER_TYPE => "ref" },
56                                         { TYPE => "DATA" }
57                                 ],
58                         },
59                 ]
60         }
61 }, { bla => "r->in.bla" }, \&print_fn, undef); 
62
63 test_warnings("", sub { $fn->("r->in.bla"); });
64
65 is($output, "");
66
67 # Test case 3: Illegal dereference
68
69 $output = "";
70 $fn = check_null_pointer({ 
71         FILE => "nofile",
72         LINE => 1,
73         PARENT => {
74                 ELEMENTS => [
75                         { 
76                                 NAME => "bla",
77                                 LEVELS => [
78                                         { TYPE => "DATA" }
79                                 ],
80                         },
81                 ]
82         }
83 }, { bla => "r->in.bla" }, \&print_fn, undef); 
84
85 test_warnings("nofile:1: too much dereferences for `bla'\n", 
86                   sub { $fn->("r->in.bla"); });
87
88 is($output, "");
89
90 # Test case 4: Double pointer dereference
91
92 $output = "";
93 $fn = check_null_pointer({ 
94         PARENT => {
95                 ELEMENTS => [
96                         { 
97                                 NAME => "bla",
98                                 LEVELS => [
99                                         { TYPE => "POINTER",
100                                           POINTER_INDEX => 0,
101                                           POINTER_TYPE => "unique" },
102                                         { TYPE => "POINTER",
103                                           POINTER_INDEX => 1,
104                                           POINTER_TYPE => "unique" },
105                                         { TYPE => "DATA" }
106                                 ],
107                         },
108                 ]
109         }
110 }, { bla => "r->in.bla" }, \&print_fn, "return;"); 
111
112 test_warnings("",
113                   sub { $fn->("*r->in.bla"); });
114
115 is($output, "if (*r->in.bla == NULL) return;");
116
117 # Test case 5: Unknown variable
118
119 $output = "";
120 $fn = check_null_pointer({ 
121         FILE => "nofile",
122         LINE => 2,
123         PARENT => {
124                 ELEMENTS => [
125                         { 
126                                 NAME => "bla",
127                                 LEVELS => [
128                                         { TYPE => "DATA" }
129                                 ],
130                         },
131                 ]
132         }
133 }, { }, \&print_fn, "return;"); 
134
135 test_warnings("nofile:2: unknown dereferenced expression `r->in.bla'\n",
136                   sub { $fn->("r->in.bla"); });
137
138 is($output, "if (r->in.bla == NULL) return;");
139
140 my $needed = {};
141 NeededElement({ TYPE => "foo", REPRESENTATION_TYPE => "foo" }, "pull", $needed); 
142 is_deeply($needed, { ndr_pull_foo => 1 });
143
144 # old settings should be kept
145 $needed = { ndr_pull_foo => 0 };
146 NeededElement({ TYPE => "foo", REPRESENTATION_TYPE => "foo" }, "pull", $needed); 
147 is_deeply($needed, { ndr_pull_foo => 0 });
148
149 # print/pull/push are independent of each other
150 $needed = { ndr_pull_foo => 0 };
151 NeededElement({ TYPE => "foo", REPRESENTATION_TYPE => "foo" }, "print", $needed); 
152 is_deeply($needed, { ndr_pull_foo => 0, ndr_print_foo => 1 });
153
154 $needed = { };
155 NeededFunction({ NAME => "foo", ELEMENTS => [ { TYPE => "bar", REPRESENTATION_TYPE => "bar" } ] }, $needed); 
156 is_deeply($needed, { ndr_pull_foo => 1, ndr_print_foo => 1, ndr_push_foo => 1,
157                          ndr_pull_bar => 1, ndr_print_bar => 1, ndr_push_bar => 1});
158
159 # push/pull/print are always set for functions
160 $needed = { ndr_pull_foo => 0 };
161 NeededFunction({ NAME => "foo", ELEMENTS => [ { TYPE => "bar", REPRESENTATION_TYPE => "bar" } ] }, $needed); 
162 is_deeply($needed, { ndr_pull_foo => 1, ndr_print_foo => 1, ndr_push_foo => 1,
163                          ndr_pull_bar => 1, ndr_push_bar => 1, ndr_print_bar => 1});
164
165 # public structs are always needed
166 $needed = {};
167 NeededType({ NAME => "bla", TYPE => "TYPEDEF",
168                 DATA => { TYPE => "STRUCT", ELEMENTS => [] } },
169                           $needed, "pull");
170 is_deeply($needed, { });
171
172 $needed = {};
173 NeededInterface({ TYPES => [ { PROPERTIES => { public => 1 }, NAME => "bla", 
174                                 TYPE => "TYPEDEF",
175                     DATA => { TYPE => "STRUCT", ELEMENTS => [] } } ] },
176                           $needed);
177 is_deeply($needed, { ndr_pull_bla => 1, ndr_push_bla => 1, ndr_print_bla => 1 });
178
179 # make sure types for elements are set too
180 $needed = {};
181 NeededInterface({ TYPES => [ { PROPERTIES => { public => 1 }, NAME => "bla", 
182                                 TYPE => "TYPEDEF",
183                     DATA => { TYPE => "STRUCT", 
184                                                   ELEMENTS => [ { TYPE => "bar", REPRESENTATION_TYPE => "bar" } ] } } ] },
185                           $needed);
186 is_deeply($needed, { ndr_pull_bla => 1, ndr_pull_bar => 1, ndr_push_bla => 1, ndr_push_bar => 1,
187                                          ndr_print_bla => 1, ndr_print_bar => 1});
188
189 $needed = {};
190 NeededInterface({ TYPES => [ { PROPERTIES => { gensize => 1}, NAME => "bla", 
191                                 TYPE => "TYPEDEF",
192                     DATA => { TYPE => "STRUCT", 
193                                                   ELEMENTS => [ { TYPE => "bar", REPRESENTATION_TYPE => "bar" } ] } } ] },
194                           $needed);
195 is_deeply($needed, { ndr_size_bla => 1 });
196                          
197 # make sure types for elements are set too
198 $needed = { ndr_pull_bla => 1 };
199 NeededType({ NAME => "bla", 
200                                 TYPE => "TYPEDEF",
201                     DATA => { TYPE => "STRUCT", 
202                                                   ELEMENTS => [ { TYPE => "bar", REPRESENTATION_TYPE => "bar" } ] } },
203                           $needed, "pull");
204 is_deeply($needed, { ndr_pull_bla => 1, ndr_pull_bar => 1 });
205
206 $needed = {};
207 NeededInterface({ TYPES => [ { PROPERTIES => { public => 1}, 
208                                 NAME => "bla", 
209                                 TYPE => "TYPEDEF",
210                     DATA => { TYPE => "STRUCT", 
211                                                   ELEMENTS => [ { TYPE => "bar", REPRESENTATION_TYPE => "rep" } ] } } ] }, $needed);
212 is_deeply($needed, { ndr_pull_bla => 1, ndr_push_bla => 1, ndr_print_bla => 1, 
213                                          ndr_print_rep => 1,
214                          ndr_pull_bar => 1, ndr_push_bar => 1, 
215                                      ndr_bar_to_rep => 1, ndr_rep_to_bar => 1});
216         
217 my $generator = new Parse::Pidl::Samba4::NDR::Parser();
218 $generator->ParseStructPush({
219                         NAME => "mystruct",
220                         TYPE => "STRUCT",
221                         PROPERTIES => {},
222                         ALIGN => 4,
223                         ELEMENTS => [ ]}, "ndr", "x");
224 is($generator->{res}, "if (ndr_flags & NDR_SCALARS) {
225         NDR_CHECK(ndr_push_align(ndr, 4));
226 }
227 if (ndr_flags & NDR_BUFFERS) {
228 }
229 ");
230
231 $generator = new Parse::Pidl::Samba4::NDR::Parser();
232 my $e = { 
233         NAME => "el1", 
234         TYPE => "mytype",
235         REPRESENTATION_TYPE => "mytype",
236         PROPERTIES => {},
237         LEVELS => [ 
238                 { LEVEL_INDEX => 0, TYPE => "DATA", DATA_TYPE => "mytype" } 
239 ] };
240 $generator->ParseStructPush({
241                         NAME => "mystruct",
242                         TYPE => "STRUCT",
243                         PROPERTIES => {},
244                         ALIGN => 4,
245                         SURROUNDING_ELEMENT => $e,
246                         ELEMENTS => [ $e ]}, "ndr", "x");
247 is($generator->{res}, "if (ndr_flags & NDR_SCALARS) {
248         NDR_CHECK(ndr_push_uint32(ndr, NDR_SCALARS, ndr_string_array_size(ndr, x->el1)));
249         NDR_CHECK(ndr_push_align(ndr, 4));
250         NDR_CHECK(ndr_push_mytype(ndr, NDR_SCALARS, &x->el1));
251 }
252 if (ndr_flags & NDR_BUFFERS) {
253 }
254 ");
255
256 is(TypeFunctionName("ndr_pull", "uint32"), "ndr_pull_uint32");
257 is(TypeFunctionName("ndr_pull", {TYPE => "ENUM", NAME => "bar"}), "ndr_pull_ENUM_bar");
258 is(TypeFunctionName("ndr_pull", {TYPE => "TYPEDEF", NAME => "bar", DATA => undef}), "ndr_pull_bar");
259 is(TypeFunctionName("ndr_push", {TYPE => "STRUCT", NAME => "bar"}), "ndr_push_STRUCT_bar");
260
261 # check noprint works
262 $generator = new Parse::Pidl::Samba4::NDR::Parser();
263 $generator->ParseElementPrint({ NAME => "x", TYPE => "rt", REPRESENTATION_TYPE => "rt", 
264                                     PROPERTIES => { noprint => 1},
265                                     LEVELS => [ { TYPE => "DATA", DATA_TYPE => "rt"} ]},
266                                     "ndr", "var", { "x" => "r->foobar" } );
267 is($generator->{res}, "");
268
269 $generator = new Parse::Pidl::Samba4::NDR::Parser();
270 $generator->ParseElementPrint({ NAME => "x", TYPE => "rt", REPRESENTATION_TYPE => "rt", 
271                                     PROPERTIES => {},
272                                     LEVELS => [ { TYPE => "DATA", DATA_TYPE => "rt" }]},
273                                     "ndr", "var", { "x" => "r->foobar" } );
274 is($generator->{res}, "ndr_print_rt(ndr, \"x\", &var);\n");
275
276 # make sure that a print function for an element with value() set works
277 $generator = new Parse::Pidl::Samba4::NDR::Parser();
278 $generator->ParseElementPrint({ NAME => "x", TYPE => "uint32", REPRESENTATION_TYPE => "uint32", 
279                                     PROPERTIES => { value => "23" },
280                                     LEVELS => [ { TYPE => "DATA", DATA_TYPE => "uint32"} ]},
281                                     "ndr", "var", { "x" => "r->foobar" } );
282 is($generator->{res}, "ndr_print_uint32(ndr, \"x\", (ndr->flags & LIBNDR_PRINT_SET_VALUES)?23:var);\n");
283
284 $generator = new Parse::Pidl::Samba4::NDR::Parser();
285 $generator->AuthServiceStruct("bridge", "\"rot13\",\"onetimepad\"");
286 is($generator->{res}, "static const char * const bridge_authservice_strings[] = {
287         \"rot13\", 
288         \"onetimepad\", 
289 };
290
291 static const struct ndr_interface_string_array bridge_authservices = {
292         .count  = 2,
293         .names  = bridge_authservice_strings
294 };
295
296 ");