r12558: Support [flag(NULLTERM)] on [charset()] arrays
[samba.git] / source4 / torture / rpc / ndr.c
1 /* 
2    Unix SMB/CIFS implementation.
3    test suite for basic ndr functions
4
5    Copyright (C) Jelmer Vernooij 2005
6    
7    This program is free software; you can redistribute it and/or modify
8    it under the terms of the GNU General Public License as published by
9    the Free Software Foundation; either version 2 of the License, or
10    (at your option) any later version.
11    
12    This program is distributed in the hope that it will be useful,
13    but WITHOUT ANY WARRANTY; without even the implied warranty of
14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15    GNU General Public License for more details.
16    
17    You should have received a copy of the GNU General Public License
18    along with this program; if not, write to the Free Software
19    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20 */
21
22 #include "includes.h"
23
24 BOOL test_check_string_terminator(TALLOC_CTX *mem_ctx)
25 {
26         struct ndr_pull *ndr;
27         DATA_BLOB blob;
28
29         /* Simple test */
30         blob = strhex_to_data_blob("0000");
31         
32         ndr = ndr_pull_init_blob(&blob, mem_ctx);
33
34         if (NT_STATUS_IS_ERR(ndr_check_string_terminator(ndr, 1, 2))) {
35                 DEBUG(0, ("simple check_string_terminator test failed\n"));
36                 return False;
37         }
38
39         if (ndr->offset != 0) {
40                 DEBUG(0, ("check_string_terminator did not reset offset\n"));
41                 return False;
42         }
43
44         if (NT_STATUS_IS_OK(ndr_check_string_terminator(ndr, 1, 3))) {
45                 DEBUG(0, ("check_string_terminator checked beyond string boundaries\n"));
46                 return False;
47         }
48
49         if (ndr->offset != 0) {
50                 DEBUG(0, ("check_string_terminator did not reset offset\n"));
51                 return False;
52         }
53
54         talloc_free(ndr);
55
56         blob = strhex_to_data_blob("11220000");
57         ndr = ndr_pull_init_blob(&blob, mem_ctx);
58
59         if (NT_STATUS_IS_ERR(ndr_check_string_terminator(ndr, 4, 1))) {
60                 DEBUG(0, ("check_string_terminator failed to recognize terminator\n"));
61                 return False;
62         }
63
64         if (NT_STATUS_IS_ERR(ndr_check_string_terminator(ndr, 3, 1))) {
65                 DEBUG(0, ("check_string_terminator failed to recognize terminator\n"));
66                 return False;
67         }
68
69         if (NT_STATUS_IS_OK(ndr_check_string_terminator(ndr, 2, 1))) {
70                 DEBUG(0, ("check_string_terminator erroneously reported terminator\n"));
71                 return False;
72         }
73
74         if (ndr->offset != 0) {
75                 DEBUG(0, ("check_string_terminator did not reset offset\n"));
76                 return False;
77         }
78
79         talloc_free(ndr);
80
81         return True;
82 }
83
84 BOOL torture_rpc_ndr(void)
85 {
86     NTSTATUS status;
87         TALLOC_CTX *mem_ctx;
88         BOOL ret = True;
89
90         mem_ctx = talloc_init("torture_rpc_mgmt");
91
92         ret &= test_check_string_terminator(mem_ctx);
93
94         talloc_free(mem_ctx);
95
96         return ret;
97 }