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_ping(struct torture_context *torture)
  59 {
  60         struct timeval tv = timeval_current();
  61         int timelimit = torture_setting_int(torture, "timelimit", 5);
  62         uint32_t total = 0;
  63
  64         torture_comment(torture,
  65                         "Running WINBINDD_PING (struct based) for %d seconds\n",
  66                         timelimit);
  67
  68         while (timeval_elapsed(&tv) < timelimit) {
  69                 DO_STRUCT_REQ_REP(WINBINDD_PING, NULL, NULL);
  70                 total++;
  71         }
  72
  73         torture_comment(torture,
  74                         "%u (%.1f/s) WINBINDD_PING (struct based)\n",
  75                         total, total / timeval_elapsed(&tv));
  76
  77         return true;
  78 }
  79
  80 struct torture_trust_domain {
  81         const char *netbios_name;
  82         const char *dns_name;
  83         struct dom_sid *sid;
  84 };
  85
  86 static bool get_trusted_domains(struct torture_context *torture,
  87                                 struct torture_trust_domain **_d)
  88 {
  89         struct winbindd_request req;
  90         struct winbindd_response rep;
  91         struct torture_trust_domain *d = NULL;
  92         uint32_t dcount = 0;
  93         fstring line;
  94         const char *extra_data;
  95
  96         ZERO_STRUCT(req);
  97         ZERO_STRUCT(rep);
  98
  99         DO_STRUCT_REQ_REP(WINBINDD_LIST_TRUSTDOM, &req, &rep);
 100
 101         extra_data = (char *)rep.extra_data.data;
 102         torture_assert(torture, extra_data, "NULL trust list");
 103
 104         while (next_token(&extra_data, line, "\n", sizeof(fstring))) {
 105                 char *p, *lp;
 106
 107                 d = talloc_realloc(torture, d,
 108                                    struct torture_trust_domain,
 109                                    dcount + 2);
 110                 ZERO_STRUCT(d[dcount+1]);
 111
 112                 lp = line;
 113                 p = strchr(lp, '\\');
 114                 torture_assert(torture, p, "missing 1st '\\' in line");
 115                 *p = 0;
 116                 d[dcount].netbios_name = talloc_strdup(d, lp);
 117                 torture_assert(torture, strlen(d[dcount].netbios_name) > 0,
 118                                "empty netbios_name");
 119
 120                 lp = p+1;
 121                 p = strchr(lp, '\\');
 122                 torture_assert(torture, p, "missing 2nd '\\' in line");
 123                 *p = 0;
 124                 d[dcount].dns_name = talloc_strdup(d, lp);
 125                 /* it's ok to have an empty dns_name */
 126
 127                 lp = p+1;
 128                 d[dcount].sid = dom_sid_parse_talloc(d, lp);
 129                 torture_assert(torture, d[dcount].sid,
 130                                "failed to parse sid");
 131
 132                 dcount++;
 133         }
 134
 135         SAFE_FREE(rep.extra_data.data);
 136         *_d = d;
 137         return true;
 138 }
 139
 140 static bool torture_winbind_struct_list_trustdom(struct torture_context *torture)
 141 {
 142         struct winbindd_request req;
 143         struct winbindd_response rep;
 144         char *list1;
 145         char *list2;
 146         bool ok;
 147         struct torture_trust_domain *listd = NULL;
 148         uint32_t i;
 149
 150         torture_comment(torture, "Running WINBINDD_LIST_TRUSTDOM (struct based)\n");
 151
 152         ZERO_STRUCT(req);
 153         ZERO_STRUCT(rep);
 154
 155         req.data.list_all_domains = false;
 156
 157         DO_STRUCT_REQ_REP(WINBINDD_LIST_TRUSTDOM, &req, &rep);
 158
 159         list1 = (char *)rep.extra_data.data;
 160         torture_assert(torture, list1, "NULL trust list");
 161
 162         torture_comment(torture, "%s\n", list1);
 163
 164         ZERO_STRUCT(req);
 165         ZERO_STRUCT(rep);
 166
 167         req.data.list_all_domains = true;
 168
 169         DO_STRUCT_REQ_REP(WINBINDD_LIST_TRUSTDOM, &req, &rep);
 170
 171         list2 = (char *)rep.extra_data.data;
 172         torture_assert(torture, list2, "NULL trust list");
 173
 174         /*
 175          * The list_all_domains parameter should be ignored
 176          */
 177         torture_assert_str_equal(torture, list2, list1, "list_all_domains not ignored");
 178
 179         SAFE_FREE(list1);
 180         SAFE_FREE(list2);
 181
 182         ok = get_trusted_domains(torture, &listd);
 183         torture_assert(torture, ok, "failed to get trust list");
 184
 185         for (i=0; listd[i].netbios_name; i++) {
 186                 if (i == 0) {
 187                         struct dom_sid *builtin_sid;
 188
 189                         builtin_sid = dom_sid_parse_talloc(torture, SID_BUILTIN);
 190
 191                         torture_assert_str_equal(torture,
 192                                                  listd[i].netbios_name,
 193                                                  NAME_BUILTIN,
 194                                                  "first domain should be 'BUILTIN'");
 195
 196                         torture_assert_str_equal(torture,
 197                                                  listd[i].dns_name,
 198                                                  "",
 199                                                  "BUILTIN domain should not have a dns name");
 200
 201                         ok = dom_sid_equal(builtin_sid,
 202                                            listd[i].sid);
 203                         torture_assert(torture, ok, "BUILTIN domain should have S-1-5-32");
 204
 205                         continue;
 206                 }
 207
 208                 /*
 209                  * TODO: verify the content of the 2nd and 3rd (in member server mode)
 210                  *       domain entries
 211                  */
 212         }
 213
 214         torture_assert(torture, i >= 2,
 215                        "The list of trusted domain should contain 2 entries");
 216
 217         return true;
 218 }
 219
 220 static bool torture_winbind_struct_getdcname(struct torture_context *torture)
 221 {
 222         struct winbindd_request req;
 223         struct winbindd_response rep;
 224
 225         torture_comment(torture, "Running WINBINDD_GETDCNAME (struct based)\n");
 226
 227         ZERO_STRUCT(req);
 228         ZERO_STRUCT(rep);
 229
 230         fstrcpy(req.domain_name, lp_workgroup());
 231
 232         DO_STRUCT_REQ_REP(WINBINDD_GETDCNAME, &req, &rep);
 233
 234         /*
 235          * TODO: test all trusted domains
 236          */
 237
 238         return true;
 239 }
 240
 241 struct torture_suite *torture_winbind_struct_init(void)
 242 {
 243         struct torture_suite *suite = torture_suite_create(talloc_autofree_context(), "STRUCT");
 244
 245         torture_suite_add_simple_test(suite, "PING", torture_winbind_struct_ping);
 246         torture_suite_add_simple_test(suite, "GETDCNAME", torture_winbind_struct_getdcname);
 247         torture_suite_add_simple_test(suite, "LIST_TRUSTDOM", torture_winbind_struct_list_trustdom);
 248
 249         suite->description = talloc_strdup(suite, "WINBIND - struct based protocol tests");
 250
 251         return suite;
 252 }