r25282: add WINBIND-STRUCT-INTERFACE_VERSION test
[kai/samba.git] / source4 / torture / winbind / struct_based.c
1 /*
2    Unix SMB/CIFS implementation.
3    SMB torture tester - winbind struct based protocol
4    Copyright (C) Stefan Metzmacher 2007
5
6    This program is free software; you can redistribute it and/or modify
7    it under the terms of the GNU General Public License as published by
8    the Free Software Foundation; either version 3 of the License, or
9    (at your option) any later version.
10
11    This program is distributed in the hope that it will be useful,
12    but WITHOUT ANY WARRANTY; without even the implied warranty of
13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14    GNU General Public License for more details.
15
16    You should have received a copy of the GNU General Public License
17    along with this program.  If not, see <http://www.gnu.org/licenses/>.
18 */
19
20 #include "includes.h"
21 #include "pstring.h"
22 #include "torture/torture.h"
23 #include "torture/winbind/proto.h"
24 #include "nsswitch/winbind_client.h"
25 #include "libcli/security/security.h"
26 #include "param/param.h"
27
28 #define DO_STRUCT_REQ_REP_EXT(op,req,rep,expected,strict,warnaction,cmt) do { \
29         NSS_STATUS __got, __expected = (expected); \
30         __got = winbindd_request_response(op, req, rep); \
31         if (__got != __expected) { \
32                 const char *__cmt = (cmt); \
33                 if (strict) { \
34                         torture_result(torture, TORTURE_FAIL, \
35                                 __location__ ": " __STRING(op) \
36                                 " returned %d, expected %d%s%s", \
37                                 __got, __expected, \
38                                 (__cmt) ? ": " : "", \
39                                 (__cmt) ? (__cmt) : ""); \
40                         return false; \
41                 } else { \
42                         torture_warning(torture, \
43                                 __location__ ": " __STRING(op) \
44                                 " returned %d, expected %d%s%s", \
45                                 __got, __expected, \
46                                 (__cmt) ? ": " : "", \
47                                 (__cmt) ? (__cmt) : ""); \
48                         warnaction; \
49                 } \
50         } \
51 } while(0)
52
53 #define DO_STRUCT_REQ_REP(op,req,rep) do { \
54         bool __noop = false; \
55         DO_STRUCT_REQ_REP_EXT(op,req,rep,NSS_STATUS_SUCCESS,true,__noop=true,NULL); \
56 } while (0)
57
58 static bool torture_winbind_struct_interface_version(struct torture_context *torture)
59 {
60         struct winbindd_request req;
61         struct winbindd_response rep;
62
63         ZERO_STRUCT(req);
64         ZERO_STRUCT(rep);
65
66         torture_comment(torture, "Running WINBINDD_INTERFACE_VERSION (struct based)\n");
67
68         DO_STRUCT_REQ_REP(WINBINDD_INTERFACE_VERSION, &req, &rep);
69
70         torture_assert_int_equal(torture,
71                                  rep.data.interface_version,
72                                  WINBIND_INTERFACE_VERSION,
73                                  "winbind server and client doesn't match");
74
75         return true;
76 }
77
78 static bool torture_winbind_struct_ping(struct torture_context *torture)
79 {
80         struct timeval tv = timeval_current();
81         int timelimit = torture_setting_int(torture, "timelimit", 5);
82         uint32_t total = 0;
83
84         torture_comment(torture,
85                         "Running WINBINDD_PING (struct based) for %d seconds\n",
86                         timelimit);
87
88         while (timeval_elapsed(&tv) < timelimit) {
89                 DO_STRUCT_REQ_REP(WINBINDD_PING, NULL, NULL);
90                 total++;
91         }
92
93         torture_comment(torture,
94                         "%u (%.1f/s) WINBINDD_PING (struct based)\n",
95                         total, total / timeval_elapsed(&tv));
96
97         return true;
98 }
99
100 struct torture_trust_domain {
101         const char *netbios_name;
102         const char *dns_name;
103         struct dom_sid *sid;
104 };
105
106 static bool get_trusted_domains(struct torture_context *torture,
107                                 struct torture_trust_domain **_d)
108 {
109         struct winbindd_request req;
110         struct winbindd_response rep;
111         struct torture_trust_domain *d = NULL;
112         uint32_t dcount = 0;
113         fstring line;
114         const char *extra_data;
115
116         ZERO_STRUCT(req);
117         ZERO_STRUCT(rep);
118
119         DO_STRUCT_REQ_REP(WINBINDD_LIST_TRUSTDOM, &req, &rep);
120
121         extra_data = (char *)rep.extra_data.data;
122         torture_assert(torture, extra_data, "NULL trust list");
123
124         while (next_token(&extra_data, line, "\n", sizeof(fstring))) {
125                 char *p, *lp;
126
127                 d = talloc_realloc(torture, d,
128                                    struct torture_trust_domain,
129                                    dcount + 2);
130                 ZERO_STRUCT(d[dcount+1]);
131
132                 lp = line;
133                 p = strchr(lp, '\\');
134                 torture_assert(torture, p, "missing 1st '\\' in line");
135                 *p = 0;
136                 d[dcount].netbios_name = talloc_strdup(d, lp);
137                 torture_assert(torture, strlen(d[dcount].netbios_name) > 0,
138                                "empty netbios_name");
139
140                 lp = p+1;
141                 p = strchr(lp, '\\');
142                 torture_assert(torture, p, "missing 2nd '\\' in line");
143                 *p = 0;
144                 d[dcount].dns_name = talloc_strdup(d, lp);
145                 /* it's ok to have an empty dns_name */
146
147                 lp = p+1;
148                 d[dcount].sid = dom_sid_parse_talloc(d, lp);
149                 torture_assert(torture, d[dcount].sid,
150                                "failed to parse sid");
151
152                 dcount++;
153         }
154         SAFE_FREE(rep.extra_data.data);
155
156         torture_assert(torture, dcount >= 2,
157                        "The list of trusted domain should contain 2 entries");
158
159         *_d = d;
160         return true;
161 }
162
163 static bool torture_winbind_struct_list_trustdom(struct torture_context *torture)
164 {
165         struct winbindd_request req;
166         struct winbindd_response rep;
167         char *list1;
168         char *list2;
169         bool ok;
170         struct torture_trust_domain *listd = NULL;
171         uint32_t i;
172
173         torture_comment(torture, "Running WINBINDD_LIST_TRUSTDOM (struct based)\n");
174
175         ZERO_STRUCT(req);
176         ZERO_STRUCT(rep);
177
178         req.data.list_all_domains = false;
179
180         DO_STRUCT_REQ_REP(WINBINDD_LIST_TRUSTDOM, &req, &rep);
181
182         list1 = (char *)rep.extra_data.data;
183         torture_assert(torture, list1, "NULL trust list");
184
185         torture_comment(torture, "%s\n", list1);
186
187         ZERO_STRUCT(req);
188         ZERO_STRUCT(rep);
189
190         req.data.list_all_domains = true;
191
192         DO_STRUCT_REQ_REP(WINBINDD_LIST_TRUSTDOM, &req, &rep);
193
194         list2 = (char *)rep.extra_data.data;
195         torture_assert(torture, list2, "NULL trust list");
196
197         /*
198          * The list_all_domains parameter should be ignored
199          */
200         torture_assert_str_equal(torture, list2, list1, "list_all_domains not ignored");
201
202         SAFE_FREE(list1);
203         SAFE_FREE(list2);
204
205         ok = get_trusted_domains(torture, &listd);
206         torture_assert(torture, ok, "failed to get trust list");
207
208         for (i=0; listd[i].netbios_name; i++) {
209                 if (i == 0) {
210                         struct dom_sid *builtin_sid;
211
212                         builtin_sid = dom_sid_parse_talloc(torture, SID_BUILTIN);
213
214                         torture_assert_str_equal(torture,
215                                                  listd[i].netbios_name,
216                                                  NAME_BUILTIN,
217                                                  "first domain should be 'BUILTIN'");
218
219                         torture_assert_str_equal(torture,
220                                                  listd[i].dns_name,
221                                                  "",
222                                                  "BUILTIN domain should not have a dns name");
223
224                         ok = dom_sid_equal(builtin_sid,
225                                            listd[i].sid);
226                         torture_assert(torture, ok, "BUILTIN domain should have S-1-5-32");
227                                        
228                         continue;
229                 }
230
231                 /*
232                  * TODO: verify the content of the 2nd and 3rd (in member server mode)
233                  *       domain entries
234                  */
235         }
236
237         return true;
238 }
239
240 static bool torture_winbind_struct_getdcname(struct torture_context *torture)
241 {
242         bool ok;
243         bool strict = torture_setting_bool(torture, "strict mode", false);
244         struct torture_trust_domain *listd = NULL;
245         uint32_t i;
246
247         torture_comment(torture, "Running WINBINDD_GETDCNAME (struct based)\n");
248
249         ok = get_trusted_domains(torture, &listd);
250         torture_assert(torture, ok, "failed to get trust list");
251
252         for (i=0; listd[i].netbios_name; i++) {
253                 struct winbindd_request req;
254                 struct winbindd_response rep;
255
256                 ZERO_STRUCT(req);
257                 ZERO_STRUCT(rep);
258
259                 fstrcpy(req.domain_name, listd[i].netbios_name);
260
261                 ok = true;
262                 DO_STRUCT_REQ_REP_EXT(WINBINDD_GETDCNAME, &req, &rep,
263                                       NSS_STATUS_SUCCESS,
264                                       (i <2 || strict), ok = false,
265                                       talloc_asprintf(torture, "DOMAIN '%s'",
266                                                       req.domain_name));
267                 if (!ok) continue;
268
269                 /* TODO: check rep.data.dc_name; */
270                 torture_comment(torture, "DOMAIN '%s' => DCNAME '%s'\n",
271                                 req.domain_name, rep.data.dc_name);
272         }
273
274         return true;
275 }
276
277 struct torture_suite *torture_winbind_struct_init(void)
278 {
279         struct torture_suite *suite = torture_suite_create(talloc_autofree_context(), "STRUCT");
280
281         torture_suite_add_simple_test(suite, "INTERFACE_VERSION", torture_winbind_struct_interface_version);
282         torture_suite_add_simple_test(suite, "PING", torture_winbind_struct_ping);
283         torture_suite_add_simple_test(suite, "LIST_TRUSTDOM", torture_winbind_struct_list_trustdom);
284         torture_suite_add_simple_test(suite, "GETDCNAME", torture_winbind_struct_getdcname);
285
286         suite->description = talloc_strdup(suite, "WINBIND - struct based protocol tests");
287
288         return suite;
289 }