source4/torture/basic/aliases.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    SMB trans2 alias scanner
   4    Copyright (C) Andrew Tridgell 2003
   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 "../lib/util/dlinklist.h"
  22 #include "libcli/raw/libcliraw.h"
  23 #include "libcli/raw/raw_proto.h"
  24 #include "libcli/libcli.h"
  25 #include "torture/util.h"
  26
  27 int create_complex_file(struct smbcli_state *cli, TALLOC_CTX *mem_ctx, const char *fname);
  28
  29 struct trans2_blobs {
  30         struct trans2_blobs *next, *prev;
  31         uint16_t level;
  32         DATA_BLOB params, data;
  33 };
  34
  35 /* look for aliases for a query */
  36 static bool gen_aliases(struct torture_context *tctx,
  37                                                 struct smbcli_state *cli, struct smb_trans2 *t2,
  38                                                 int level_offset)
  39 {
  40         uint16_t level;
  41         struct trans2_blobs *alias_blobs = NULL;
  42         struct trans2_blobs *t2b, *t2b2;
  43         int count=0, alias_count=0;
  44
  45         for (level=0;level<2000;level++) {
  46                 NTSTATUS status;
  47
  48                 SSVAL(t2->in.params.data, level_offset, level);
  49
  50                 status = smb_raw_trans2(cli->tree, tctx, t2);
  51                 if (!NT_STATUS_IS_OK(status)) continue;
  52
  53                 t2b = talloc(tctx, struct trans2_blobs);
  54                 t2b->level = level;
  55                 t2b->params = t2->out.params;
  56                 t2b->data = t2->out.data;
  57                 DLIST_ADD(alias_blobs, t2b);
  58                 torture_comment(tctx,
  59                                                 "\tFound level %4u (0x%03x) of size %3d (0x%02x)\n",
  60                          level, level,
  61                          (int)t2b->data.length, (int)t2b->data.length);
  62                 count++;
  63         }
  64
  65         torture_comment(tctx, "Found %d levels with success status\n", count);
  66
  67         for (t2b=alias_blobs; t2b; t2b=t2b->next) {
  68                 for (t2b2=alias_blobs; t2b2; t2b2=t2b2->next) {
  69                         if (t2b->level >= t2b2->level) continue;
  70                         if (data_blob_cmp(&t2b->params, &t2b2->params) == 0 &&
  71                             data_blob_cmp(&t2b->data, &t2b2->data) == 0) {
  72                                 torture_comment(tctx,
  73                                 "\tLevel %u (0x%x) and level %u (0x%x) are possible aliases\n",
  74                                        t2b->level, t2b->level, t2b2->level, t2b2->level);
  75                                 alias_count++;
  76                         }
  77                 }
  78         }
  79
  80         torture_comment(tctx, "Found %d aliased levels\n", alias_count);
  81
  82         return true;
  83 }
  84
  85 /* look for qfsinfo aliases */
  86 static bool qfsinfo_aliases(struct torture_context *tctx, struct smbcli_state *cli)
  87 {
  88         struct smb_trans2 t2;
  89         uint16_t setup = TRANSACT2_QFSINFO;
  90
  91         t2.in.max_param = 0;
  92         t2.in.max_data = UINT16_MAX;
  93         t2.in.max_setup = 0;
  94         t2.in.flags = 0;
  95         t2.in.timeout = 0;
  96         t2.in.setup_count = 1;
  97         t2.in.setup = &setup;
  98         t2.in.params = data_blob_talloc_zero(tctx, 2);
  99         t2.in.data = data_blob(NULL, 0);
 100         ZERO_STRUCT(t2.out);
 101
 102         return gen_aliases(tctx, cli, &t2, 0);
 103 }
 104
 105 /* look for qfileinfo aliases */
 106 static bool qfileinfo_aliases(struct torture_context *tctx, struct smbcli_state *cli)
 107 {
 108         struct smb_trans2 t2;
 109         uint16_t setup = TRANSACT2_QFILEINFO;
 110         const char *fname = "\\qfileinfo_aliases.txt";
 111         int fnum;
 112
 113         t2.in.max_param = 2;
 114         t2.in.max_data = UINT16_MAX;
 115         t2.in.max_setup = 0;
 116         t2.in.flags = 0;
 117         t2.in.timeout = 0;
 118         t2.in.setup_count = 1;
 119         t2.in.setup = &setup;
 120         t2.in.params = data_blob_talloc_zero(tctx, 4);
 121         t2.in.data = data_blob(NULL, 0);
 122         ZERO_STRUCT(t2.out);
 123
 124         smbcli_unlink(cli->tree, fname);
 125         fnum = create_complex_file(cli, cli, fname);
 126         torture_assert(tctx, fnum != -1, talloc_asprintf(tctx,
 127                                         "open of %s failed (%s)", fname,
 128                                    smbcli_errstr(cli->tree)));
 129
 130         smbcli_write(cli->tree, fnum, 0, &t2, 0, sizeof(t2));
 131
 132         SSVAL(t2.in.params.data, 0, fnum);
 133
 134         if (!gen_aliases(tctx, cli, &t2, 2))
 135                 return false;
 136
 137         smbcli_close(cli->tree, fnum);
 138         smbcli_unlink(cli->tree, fname);
 139
 140         return true;
 141 }
 142
 143
 144 /* look for qpathinfo aliases */
 145 static bool qpathinfo_aliases(struct torture_context *tctx, struct smbcli_state *cli)
 146 {
 147         struct smb_trans2 t2;
 148         uint16_t setup = TRANSACT2_QPATHINFO;
 149         const char *fname = "\\qpathinfo_aliases.txt";
 150         int fnum;
 151
 152         t2.in.max_param = 2;
 153         t2.in.max_data = UINT16_MAX;
 154         t2.in.max_setup = 0;
 155         t2.in.flags = 0;
 156         t2.in.timeout = 0;
 157         t2.in.setup_count = 1;
 158         t2.in.setup = &setup;
 159         t2.in.params = data_blob_talloc_zero(tctx, 6);
 160         t2.in.data = data_blob(NULL, 0);
 161         ZERO_STRUCT(t2.out);
 162
 163         smbcli_unlink(cli->tree, fname);
 164         fnum = create_complex_file(cli, cli, fname);
 165         torture_assert(tctx, fnum != -1, talloc_asprintf(tctx,
 166                                         "open of %s failed (%s)", fname,
 167                                    smbcli_errstr(cli->tree)));
 168
 169         smbcli_write(cli->tree, fnum, 0, &t2, 0, sizeof(t2));
 170         smbcli_close(cli->tree, fnum);
 171
 172         SIVAL(t2.in.params.data, 2, 0);
 173
 174         smbcli_blob_append_string(cli->session, tctx, &t2.in.params,
 175                                fname, STR_TERMINATE);
 176
 177         if (!gen_aliases(tctx, cli, &t2, 0))
 178                 return false;
 179
 180         smbcli_unlink(cli->tree, fname);
 181
 182         return true;
 183 }
 184
 185
 186 /* look for trans2 findfirst aliases */
 187 static bool findfirst_aliases(struct torture_context *tctx, struct smbcli_state *cli)
 188 {
 189         struct smb_trans2 t2;
 190         uint16_t setup = TRANSACT2_FINDFIRST;
 191         const char *fname = "\\findfirst_aliases.txt";
 192         int fnum;
 193
 194         t2.in.max_param = 16;
 195         t2.in.max_data = UINT16_MAX;
 196         t2.in.max_setup = 0;
 197         t2.in.flags = 0;
 198         t2.in.timeout = 0;
 199         t2.in.setup_count = 1;
 200         t2.in.setup = &setup;
 201         t2.in.params = data_blob_talloc_zero(tctx, 12);
 202         t2.in.data = data_blob(NULL, 0);
 203         ZERO_STRUCT(t2.out);
 204
 205         smbcli_unlink(cli->tree, fname);
 206         fnum = create_complex_file(cli, cli, fname);
 207         torture_assert(tctx, fnum != -1, talloc_asprintf(tctx,
 208                                         "open of %s failed (%s)", fname,
 209                                    smbcli_errstr(cli->tree)));
 210
 211         smbcli_write(cli->tree, fnum, 0, &t2, 0, sizeof(t2));
 212         smbcli_close(cli->tree, fnum);
 213
 214         SSVAL(t2.in.params.data, 0, 0);
 215         SSVAL(t2.in.params.data, 2, 1);
 216         SSVAL(t2.in.params.data, 4, FLAG_TRANS2_FIND_CLOSE);
 217         SSVAL(t2.in.params.data, 6, 0);
 218         SIVAL(t2.in.params.data, 8, 0);
 219
 220         smbcli_blob_append_string(cli->session, tctx, &t2.in.params,
 221                                fname, STR_TERMINATE);
 222
 223         if (!gen_aliases(tctx, cli, &t2, 6))
 224                 return false;
 225
 226         smbcli_unlink(cli->tree, fname);
 227
 228         return true;
 229 }
 230
 231
 232
 233 /* look for aliases for a set function */
 234 static bool gen_set_aliases(struct torture_context *tctx,
 235                                                         struct smbcli_state *cli,
 236                                                         struct smb_trans2 *t2, int level_offset)
 237 {
 238         uint16_t level;
 239         struct trans2_blobs *alias_blobs = NULL;
 240         struct trans2_blobs *t2b;
 241         int count=0, dsize;
 242
 243         for (level=1;level<1100;level++) {
 244                 NTSTATUS status, status1;
 245                 SSVAL(t2->in.params.data, level_offset, level);
 246
 247                 status1 = NT_STATUS_OK;
 248
 249                 for (dsize=2; dsize<1024; dsize += 2) {
 250                         data_blob_free(&t2->in.data);
 251                         t2->in.data = data_blob(NULL, dsize);
 252                         data_blob_clear(&t2->in.data);
 253                         status = smb_raw_trans2(cli->tree, tctx, t2);
 254                         /* some error codes mean that this whole level doesn't exist */
 255                         if (NT_STATUS_EQUAL(NT_STATUS_INVALID_LEVEL, status) ||
 256                             NT_STATUS_EQUAL(NT_STATUS_INVALID_INFO_CLASS, status) ||
 257                             NT_STATUS_EQUAL(NT_STATUS_NOT_SUPPORTED, status)) {
 258                                 break;
 259                         }
 260                         if (NT_STATUS_IS_OK(status)) break;
 261
 262                         /* invalid parameter means that the level exists at this
 263                            size, but the contents are wrong (not surprising with
 264                            all zeros!) */
 265                         if (NT_STATUS_EQUAL(status, NT_STATUS_INVALID_PARAMETER)) break;
 266
 267                         /* this is the usual code for 'wrong size' */
 268                         if (NT_STATUS_EQUAL(status, NT_STATUS_INFO_LENGTH_MISMATCH)) {
 269                                 continue;
 270                         }
 271
 272                         if (!NT_STATUS_EQUAL(status, status1)) {
 273                                 torture_comment(tctx, "level=%d size=%d %s\n", level, dsize, nt_errstr(status));
 274                         }
 275                         status1 = status;
 276                 }
 277
 278                 if (!NT_STATUS_IS_OK(status) &&
 279                     !NT_STATUS_EQUAL(status, NT_STATUS_INVALID_PARAMETER)) continue;
 280
 281                 t2b = talloc(tctx, struct trans2_blobs);
 282                 t2b->level = level;
 283                 t2b->params = t2->out.params;
 284                 t2b->data = t2->out.data;
 285                 DLIST_ADD(alias_blobs, t2b);
 286                 torture_comment(tctx,
 287                                         "\tFound level %4u (0x%03x) of size %3d (0x%02x)\n",
 288                          level, level,
 289                          (int)t2->in.data.length, (int)t2->in.data.length);
 290                 count++;
 291         }
 292
 293         torture_comment(tctx, "Found %d valid levels\n", count);
 294
 295         return true;
 296 }
 297
 298
 299
 300 /* look for setfileinfo aliases */
 301 static bool setfileinfo_aliases(struct torture_context *tctx, struct smbcli_state *cli)
 302 {
 303         struct smb_trans2 t2;
 304         uint16_t setup = TRANSACT2_SETFILEINFO;
 305         const char *fname = "\\setfileinfo_aliases.txt";
 306         int fnum;
 307
 308         t2.in.max_param = 2;
 309         t2.in.max_data = 0;
 310         t2.in.max_setup = 0;
 311         t2.in.flags = 0;
 312         t2.in.timeout = 0;
 313         t2.in.setup_count = 1;
 314         t2.in.setup = &setup;
 315         t2.in.params = data_blob_talloc_zero(tctx, 6);
 316         t2.in.data = data_blob(NULL, 0);
 317         ZERO_STRUCT(t2.out);
 318
 319         smbcli_unlink(cli->tree, fname);
 320         fnum = create_complex_file(cli, cli, fname);
 321         torture_assert(tctx, fnum != -1, talloc_asprintf(tctx,
 322                                    "open of %s failed (%s)", fname,
 323                                    smbcli_errstr(cli->tree)));
 324
 325         smbcli_write(cli->tree, fnum, 0, &t2, 0, sizeof(t2));
 326
 327         SSVAL(t2.in.params.data, 0, fnum);
 328         SSVAL(t2.in.params.data, 4, 0);
 329
 330         gen_set_aliases(tctx, cli, &t2, 2);
 331
 332         smbcli_close(cli->tree, fnum);
 333         smbcli_unlink(cli->tree, fname);
 334
 335         return true;
 336 }
 337
 338 /* look for setpathinfo aliases */
 339 static bool setpathinfo_aliases(struct torture_context *tctx,
 340                                                                 struct smbcli_state *cli)
 341 {
 342         struct smb_trans2 t2;
 343         uint16_t setup = TRANSACT2_SETPATHINFO;
 344         const char *fname = "\\setpathinfo_aliases.txt";
 345         int fnum;
 346
 347         t2.in.max_param = 32;
 348         t2.in.max_data = UINT16_MAX;
 349         t2.in.max_setup = 0;
 350         t2.in.flags = 0;
 351         t2.in.timeout = 0;
 352         t2.in.setup_count = 1;
 353         t2.in.setup = &setup;
 354         t2.in.params = data_blob_talloc_zero(tctx, 4);
 355         t2.in.data = data_blob(NULL, 0);
 356         ZERO_STRUCT(t2.out);
 357
 358         smbcli_unlink(cli->tree, fname);
 359
 360         fnum = create_complex_file(cli, cli, fname);
 361         torture_assert(tctx, fnum != -1, talloc_asprintf(tctx,
 362                                         "open of %s failed (%s)", fname,
 363                                    smbcli_errstr(cli->tree)));
 364
 365         smbcli_write(cli->tree, fnum, 0, &t2, 0, sizeof(t2));
 366         smbcli_close(cli->tree, fnum);
 367
 368         SSVAL(t2.in.params.data, 2, 0);
 369
 370         smbcli_blob_append_string(cli->session, tctx, &t2.in.params,
 371                                fname, STR_TERMINATE);
 372
 373         if (!gen_set_aliases(tctx, cli, &t2, 0))
 374                 return false;
 375
 376         torture_assert_ntstatus_ok(tctx, smbcli_unlink(cli->tree, fname),
 377                 talloc_asprintf(tctx, "unlink: %s", smbcli_errstr(cli->tree)));
 378
 379         return true;
 380 }
 381
 382
 383 /* look for aliased info levels in trans2 calls */
 384 struct torture_suite *torture_trans2_aliases(TALLOC_CTX *mem_ctx)
 385 {
 386         struct torture_suite *suite = torture_suite_create(mem_ctx, "ALIASES");
 387
 388         torture_suite_add_1smb_test(suite, "QFILEINFO aliases", qfsinfo_aliases);
 389         torture_suite_add_1smb_test(suite, "QFSINFO aliases", qfileinfo_aliases);
 390         torture_suite_add_1smb_test(suite, "QPATHINFO aliases", qpathinfo_aliases);
 391         torture_suite_add_1smb_test(suite, "FINDFIRST aliases", findfirst_aliases);
 392         torture_suite_add_1smb_test(suite, "setfileinfo_aliases", setfileinfo_aliases);
 393         torture_suite_add_1smb_test(suite, "setpathinfo_aliases", setpathinfo_aliases);
 394
 395         return suite;
 396 }