source/smb_server/smb/nttrans.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    NT transaction handling
   4    Copyright (C) Andrew Tridgell 2003
   5    Copyright (C) James J Myers 2003 <myersjj@samba.org>
   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 3 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, see <http://www.gnu.org/licenses/>.
  19 */
  20 /*
  21    This file handles the parsing of transact2 requests
  22 */
  23
  24 #include "includes.h"
  25 #include "smb_server/smb_server.h"
  26 #include "ntvfs/ntvfs.h"
  27 #include "libcli/raw/libcliraw.h"
  28 #include "libcli/raw/raw_proto.h"
  29 #include "librpc/gen_ndr/ndr_security.h"
  30 #include "param/param.h"
  31
  32 /*
  33   hold the state of a nttrans op while in progress. Needed to allow for async backend
  34   functions.
  35 */
  36 struct nttrans_op {
  37         struct smb_nttrans *trans;
  38         NTSTATUS (*send_fn)(struct nttrans_op *);
  39         void *op_info;
  40 };
  41
  42
  43 /* setup a nttrans reply, given the data and params sizes */
  44 static NTSTATUS nttrans_setup_reply(struct nttrans_op *op,
  45                                     struct smb_nttrans *trans,
  46                                     uint32_t param_size, uint32_t data_size,
  47                                     uint8_t setup_count)
  48 {
  49         trans->out.setup_count = setup_count;
  50         if (setup_count != 0) {
  51                 trans->out.setup = talloc_zero_array(op, uint8_t, setup_count*2);
  52                 NT_STATUS_HAVE_NO_MEMORY(trans->out.setup);
  53         }
  54         trans->out.params = data_blob_talloc(op, NULL, param_size);
  55         if (param_size != 0) {
  56                 NT_STATUS_HAVE_NO_MEMORY(trans->out.params.data);
  57         }
  58         trans->out.data = data_blob_talloc(op, NULL, data_size);
  59         if (data_size != 0) {
  60                 NT_STATUS_HAVE_NO_MEMORY(trans->out.data.data);
  61         }
  62         return NT_STATUS_OK;
  63 }
  64
  65 /*
  66   send a nttrans create reply
  67 */
  68 static NTSTATUS nttrans_create_send(struct nttrans_op *op)
  69 {
  70         union smb_open *io = talloc_get_type(op->op_info, union smb_open);
  71         uint8_t *params;
  72         NTSTATUS status;
  73
  74         status = nttrans_setup_reply(op, op->trans, 69, 0, 0);
  75         NT_STATUS_NOT_OK_RETURN(status);
  76         params = op->trans->out.params.data;
  77
  78         SSVAL(params,        0, io->ntcreatex.out.oplock_level);
  79         smbsrv_push_fnum(params, 2, io->ntcreatex.out.file.ntvfs);
  80         SIVAL(params,        4, io->ntcreatex.out.create_action);
  81         SIVAL(params,        8, 0); /* ea error offset */
  82         push_nttime(params, 12, io->ntcreatex.out.create_time);
  83         push_nttime(params, 20, io->ntcreatex.out.access_time);
  84         push_nttime(params, 28, io->ntcreatex.out.write_time);
  85         push_nttime(params, 36, io->ntcreatex.out.change_time);
  86         SIVAL(params,       44, io->ntcreatex.out.attrib);
  87         SBVAL(params,       48, io->ntcreatex.out.alloc_size);
  88         SBVAL(params,       56, io->ntcreatex.out.size);
  89         SSVAL(params,       64, io->ntcreatex.out.file_type);
  90         SSVAL(params,       66, io->ntcreatex.out.ipc_state);
  91         SCVAL(params,       68, io->ntcreatex.out.is_directory);
  92
  93         return NT_STATUS_OK;
  94 }
  95
  96 /*
  97    parse NTTRANS_CREATE request
  98  */
  99 static NTSTATUS nttrans_create(struct smbsrv_request *req,
 100                                struct nttrans_op *op)
 101 {
 102         struct smb_nttrans *trans = op->trans;
 103         union smb_open *io;
 104         uint16_t fname_len;
 105         uint32_t sd_length, ea_length;
 106         NTSTATUS status;
 107         uint8_t *params;
 108         enum ndr_err_code ndr_err;
 109
 110         if (trans->in.params.length < 54) {
 111                 return NT_STATUS_INVALID_PARAMETER;
 112         }
 113
 114         /* parse the request */
 115         io = talloc(op, union smb_open);
 116         NT_STATUS_HAVE_NO_MEMORY(io);
 117
 118         io->ntcreatex.level = RAW_OPEN_NTTRANS_CREATE;
 119
 120         params = trans->in.params.data;
 121
 122         io->ntcreatex.in.flags            = IVAL(params,  0);
 123         io->ntcreatex.in.root_fid         = IVAL(params,  4);
 124         io->ntcreatex.in.access_mask      = IVAL(params,  8);
 125         io->ntcreatex.in.alloc_size       = BVAL(params, 12);
 126         io->ntcreatex.in.file_attr        = IVAL(params, 20);
 127         io->ntcreatex.in.share_access     = IVAL(params, 24);
 128         io->ntcreatex.in.open_disposition = IVAL(params, 28);
 129         io->ntcreatex.in.create_options   = IVAL(params, 32);
 130         sd_length                         = IVAL(params, 36);
 131         ea_length                         = IVAL(params, 40);
 132         fname_len                         = IVAL(params, 44);
 133         io->ntcreatex.in.impersonation    = IVAL(params, 48);
 134         io->ntcreatex.in.security_flags   = CVAL(params, 52);
 135         io->ntcreatex.in.sec_desc         = NULL;
 136         io->ntcreatex.in.ea_list          = NULL;
 137         io->ntcreatex.in.query_maximal_access = false;
 138
 139         req_pull_string(&req->in.bufinfo, &io->ntcreatex.in.fname,
 140                         params + 53,
 141                         MIN(fname_len+1, trans->in.params.length - 53),
 142                         STR_NO_RANGE_CHECK | STR_TERMINATE);
 143         if (!io->ntcreatex.in.fname) {
 144                 return NT_STATUS_INVALID_PARAMETER;
 145         }
 146
 147         if (sd_length > trans->in.data.length ||
 148             ea_length > trans->in.data.length ||
 149             (sd_length+ea_length) > trans->in.data.length) {
 150                 return NT_STATUS_INVALID_PARAMETER;
 151         }
 152
 153         /* this call has an optional security descriptor */
 154         if (sd_length != 0) {
 155                 DATA_BLOB blob;
 156                 blob.data = trans->in.data.data;
 157                 blob.length = sd_length;
 158                 io->ntcreatex.in.sec_desc = talloc(io, struct security_descriptor);
 159                 if (io->ntcreatex.in.sec_desc == NULL) {
 160                         return NT_STATUS_NO_MEMORY;
 161                 }
 162                 ndr_err = ndr_pull_struct_blob(&blob, io, NULL,
 163                                                io->ntcreatex.in.sec_desc,
 164                                                (ndr_pull_flags_fn_t)ndr_pull_security_descriptor);
 165                 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
 166                         return ndr_map_error2ntstatus(ndr_err);
 167                 }
 168         }
 169
 170         /* and an optional ea_list */
 171         if (ea_length > 4) {
 172                 DATA_BLOB blob;
 173                 blob.data = trans->in.data.data + sd_length;
 174                 blob.length = ea_length;
 175                 io->ntcreatex.in.ea_list = talloc(io, struct smb_ea_list);
 176                 if (io->ntcreatex.in.ea_list == NULL) {
 177                         return NT_STATUS_NO_MEMORY;
 178                 }
 179
 180                 status = ea_pull_list_chained(&blob, io,
 181                                               &io->ntcreatex.in.ea_list->num_eas,
 182                                               &io->ntcreatex.in.ea_list->eas);
 183                 if (!NT_STATUS_IS_OK(status)) {
 184                         return status;
 185                 }
 186         }
 187
 188         op->send_fn = nttrans_create_send;
 189         op->op_info = io;
 190
 191         return ntvfs_open(req->ntvfs, io);
 192 }
 193
 194
 195 /*
 196    send NTTRANS_QUERY_SEC_DESC reply
 197  */
 198 static NTSTATUS nttrans_query_sec_desc_send(struct nttrans_op *op)
 199 {
 200         union smb_fileinfo *io = talloc_get_type(op->op_info, union smb_fileinfo);
 201         uint8_t *params;
 202         NTSTATUS status;
 203         enum ndr_err_code ndr_err;
 204
 205         status = nttrans_setup_reply(op, op->trans, 4, 0, 0);
 206         NT_STATUS_NOT_OK_RETURN(status);
 207         params = op->trans->out.params.data;
 208
 209         ndr_err = ndr_push_struct_blob(&op->trans->out.data, op, NULL,
 210                                        io->query_secdesc.out.sd,
 211                                        (ndr_push_flags_fn_t)ndr_push_security_descriptor);
 212         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
 213                 return ndr_map_error2ntstatus(ndr_err);
 214         }
 215
 216         SIVAL(params, 0, op->trans->out.data.length);
 217
 218         return NT_STATUS_OK;
 219 }
 220
 221 /*
 222    parse NTTRANS_QUERY_SEC_DESC request
 223  */
 224 static NTSTATUS nttrans_query_sec_desc(struct smbsrv_request *req,
 225                                        struct nttrans_op *op)
 226 {
 227         struct smb_nttrans *trans = op->trans;
 228         union smb_fileinfo *io;
 229
 230         if (trans->in.params.length < 8) {
 231                 return NT_STATUS_INVALID_PARAMETER;
 232         }
 233
 234         /* parse the request */
 235         io = talloc(op, union smb_fileinfo);
 236         NT_STATUS_HAVE_NO_MEMORY(io);
 237
 238         io->query_secdesc.level            = RAW_FILEINFO_SEC_DESC;
 239         io->query_secdesc.in.file.ntvfs    = smbsrv_pull_fnum(req, trans->in.params.data, 0);
 240         io->query_secdesc.in.secinfo_flags = IVAL(trans->in.params.data, 4);
 241
 242         op->op_info = io;
 243         op->send_fn = nttrans_query_sec_desc_send;
 244
 245         SMBSRV_CHECK_FILE_HANDLE_NTSTATUS(io->query_secdesc.in.file.ntvfs);
 246         return ntvfs_qfileinfo(req->ntvfs, io);
 247 }
 248
 249
 250 /*
 251    parse NTTRANS_SET_SEC_DESC request
 252  */
 253 static NTSTATUS nttrans_set_sec_desc(struct smbsrv_request *req,
 254                                      struct nttrans_op *op)
 255 {
 256         struct smb_nttrans *trans = op->trans;
 257         union smb_setfileinfo *io;
 258         enum ndr_err_code ndr_err;
 259
 260         if (trans->in.params.length < 8) {
 261                 return NT_STATUS_INVALID_PARAMETER;
 262         }
 263
 264         /* parse the request */
 265         io = talloc(req, union smb_setfileinfo);
 266         NT_STATUS_HAVE_NO_MEMORY(io);
 267
 268         io->set_secdesc.level            = RAW_SFILEINFO_SEC_DESC;
 269         io->set_secdesc.in.file.ntvfs    = smbsrv_pull_fnum(req, trans->in.params.data, 0);
 270         io->set_secdesc.in.secinfo_flags = IVAL(trans->in.params.data, 4);
 271
 272         io->set_secdesc.in.sd = talloc(io, struct security_descriptor);
 273         NT_STATUS_HAVE_NO_MEMORY(io->set_secdesc.in.sd);
 274
 275         ndr_err = ndr_pull_struct_blob(&trans->in.data, req, NULL,
 276                                        io->set_secdesc.in.sd,
 277                                        (ndr_pull_flags_fn_t)ndr_pull_security_descriptor);
 278         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
 279                 return ndr_map_error2ntstatus(ndr_err);
 280         }
 281
 282         SMBSRV_CHECK_FILE_HANDLE_NTSTATUS(io->set_secdesc.in.file.ntvfs);
 283         return ntvfs_setfileinfo(req->ntvfs, io);
 284 }
 285
 286
 287 /* parse NTTRANS_RENAME request
 288  */
 289 static NTSTATUS nttrans_rename(struct smbsrv_request *req,
 290                                struct nttrans_op *op)
 291 {
 292         struct smb_nttrans *trans = op->trans;
 293         union smb_rename *io;
 294
 295         if (trans->in.params.length < 5) {
 296                 return NT_STATUS_INVALID_PARAMETER;
 297         }
 298
 299         /* parse the request */
 300         io = talloc(req, union smb_rename);
 301         NT_STATUS_HAVE_NO_MEMORY(io);
 302
 303         io->nttrans.level               = RAW_RENAME_NTTRANS;
 304         io->nttrans.in.file.ntvfs       = smbsrv_pull_fnum(req, trans->in.params.data, 0);
 305         io->nttrans.in.flags            = SVAL(trans->in.params.data, 2);
 306
 307         smbsrv_blob_pull_string(&req->in.bufinfo, &trans->in.params, 4,
 308                                 &io->nttrans.in.new_name,
 309                                 STR_TERMINATE);
 310         if (!io->nttrans.in.new_name) {
 311                 return NT_STATUS_INVALID_PARAMETER;
 312         }
 313
 314         SMBSRV_CHECK_FILE_HANDLE_NTSTATUS(io->nttrans.in.file.ntvfs);
 315         return ntvfs_rename(req->ntvfs, io);
 316 }
 317
 318 /*
 319    parse NTTRANS_IOCTL send
 320  */
 321 static NTSTATUS nttrans_ioctl_send(struct nttrans_op *op)
 322 {
 323         union smb_ioctl *info = talloc_get_type(op->op_info, union smb_ioctl);
 324         NTSTATUS status;
 325
 326         /*
 327          * we pass 0 as data_count here,
 328          * because we reuse the DATA_BLOB from the smb_ioctl
 329          * struct
 330          */
 331         status = nttrans_setup_reply(op, op->trans, 0, 0, 1);
 332         NT_STATUS_NOT_OK_RETURN(status);
 333
 334         op->trans->out.setup[0]         = 0;
 335         op->trans->out.data             = info->ntioctl.out.blob;
 336
 337         return NT_STATUS_OK;
 338 }
 339
 340
 341 /*
 342    parse NTTRANS_IOCTL request
 343  */
 344 static NTSTATUS nttrans_ioctl(struct smbsrv_request *req,
 345                               struct nttrans_op *op)
 346 {
 347         struct smb_nttrans *trans = op->trans;
 348         union smb_ioctl *nt;
 349
 350         /* should have at least 4 setup words */
 351         if (trans->in.setup_count != 4) {
 352                 return NT_STATUS_INVALID_PARAMETER;
 353         }
 354
 355         nt = talloc(op, union smb_ioctl);
 356         NT_STATUS_HAVE_NO_MEMORY(nt);
 357
 358         nt->ntioctl.level               = RAW_IOCTL_NTIOCTL;
 359         nt->ntioctl.in.function         = IVAL(trans->in.setup, 0);
 360         nt->ntioctl.in.file.ntvfs       = smbsrv_pull_fnum(req, (uint8_t *)trans->in.setup, 4);
 361         nt->ntioctl.in.fsctl            = CVAL(trans->in.setup, 6);
 362         nt->ntioctl.in.filter           = CVAL(trans->in.setup, 7);
 363         nt->ntioctl.in.max_data         = trans->in.max_data;
 364         nt->ntioctl.in.blob             = trans->in.data;
 365
 366         op->op_info = nt;
 367         op->send_fn = nttrans_ioctl_send;
 368
 369         SMBSRV_CHECK_FILE_HANDLE_NTSTATUS(nt->ntioctl.in.file.ntvfs);
 370         return ntvfs_ioctl(req->ntvfs, nt);
 371 }
 372
 373
 374 /*
 375    send NTTRANS_NOTIFY_CHANGE reply
 376  */
 377 static NTSTATUS nttrans_notify_change_send(struct nttrans_op *op)
 378 {
 379         union smb_notify *info = talloc_get_type(op->op_info, union smb_notify);
 380         size_t size = 0;
 381         int i;
 382         NTSTATUS status;
 383         uint8_t *p;
 384 #define MAX_BYTES_PER_CHAR 3
 385
 386         /* work out how big the reply buffer could be */
 387         for (i=0;i<info->nttrans.out.num_changes;i++) {
 388                 size += 12 + 3 + (1+strlen(info->nttrans.out.changes[i].name.s)) * MAX_BYTES_PER_CHAR;
 389         }
 390
 391         status = nttrans_setup_reply(op, op->trans, size, 0, 0);
 392         NT_STATUS_NOT_OK_RETURN(status);
 393         p = op->trans->out.params.data;
 394
 395         /* construct the changes buffer */
 396         for (i=0;i<info->nttrans.out.num_changes;i++) {
 397                 uint32_t ofs;
 398                 ssize_t len;
 399
 400                 SIVAL(p, 4, info->nttrans.out.changes[i].action);
 401                 len = push_string(lp_iconv_convenience(global_loadparm), p + 12, info->nttrans.out.changes[i].name.s,
 402                                   op->trans->out.params.length -
 403                                   (p+12 - op->trans->out.params.data), STR_UNICODE);
 404                 SIVAL(p, 8, len);
 405
 406                 ofs = len + 12;
 407
 408                 if (ofs & 3) {
 409                         int pad = 4 - (ofs & 3);
 410                         memset(p+ofs, 0, pad);
 411                         ofs += pad;
 412                 }
 413
 414                 if (i == info->nttrans.out.num_changes-1) {
 415                         SIVAL(p, 0, 0);
 416                 } else {
 417                         SIVAL(p, 0, ofs);
 418                 }
 419
 420                 p += ofs;
 421         }
 422
 423         op->trans->out.params.length = p - op->trans->out.params.data;
 424
 425         return NT_STATUS_OK;
 426 }
 427
 428 /*
 429    parse NTTRANS_NOTIFY_CHANGE request
 430  */
 431 static NTSTATUS nttrans_notify_change(struct smbsrv_request *req,
 432                                       struct nttrans_op *op)
 433 {
 434         struct smb_nttrans *trans = op->trans;
 435         union smb_notify *info;
 436
 437         /* should have at least 4 setup words */
 438         if (trans->in.setup_count != 4) {
 439                 return NT_STATUS_INVALID_PARAMETER;
 440         }
 441
 442         info = talloc(op, union smb_notify);
 443         NT_STATUS_HAVE_NO_MEMORY(info);
 444
 445         info->nttrans.level                     = RAW_NOTIFY_NTTRANS;
 446         info->nttrans.in.completion_filter      = IVAL(trans->in.setup, 0);
 447         info->nttrans.in.file.ntvfs             = smbsrv_pull_fnum(req, (uint8_t *)trans->in.setup, 4);
 448         info->nttrans.in.recursive              = SVAL(trans->in.setup, 6);
 449         info->nttrans.in.buffer_size            = trans->in.max_param;
 450
 451         op->op_info = info;
 452         op->send_fn = nttrans_notify_change_send;
 453
 454         SMBSRV_CHECK_FILE_HANDLE_NTSTATUS(info->nttrans.in.file.ntvfs);
 455         return ntvfs_notify(req->ntvfs, info);
 456 }
 457
 458 /*
 459   backend for nttrans requests
 460 */
 461 static NTSTATUS nttrans_backend(struct smbsrv_request *req,
 462                                 struct nttrans_op *op)
 463 {
 464         /* the nttrans command is in function */
 465         switch (op->trans->in.function) {
 466         case NT_TRANSACT_CREATE:
 467                 return nttrans_create(req, op);
 468         case NT_TRANSACT_IOCTL:
 469                 return nttrans_ioctl(req, op);
 470         case NT_TRANSACT_RENAME:
 471                 return nttrans_rename(req, op);
 472         case NT_TRANSACT_QUERY_SECURITY_DESC:
 473                 return nttrans_query_sec_desc(req, op);
 474         case NT_TRANSACT_SET_SECURITY_DESC:
 475                 return nttrans_set_sec_desc(req, op);
 476         case NT_TRANSACT_NOTIFY_CHANGE:
 477                 return nttrans_notify_change(req, op);
 478         }
 479
 480         /* an unknown nttrans command */
 481         return NT_STATUS_DOS(ERRSRV, ERRerror);
 482 }
 483
 484
 485 static void reply_nttrans_send(struct ntvfs_request *ntvfs)
 486 {
 487         struct smbsrv_request *req;
 488         uint32_t params_left, data_left;
 489         uint8_t *params, *data;
 490         struct smb_nttrans *trans;
 491         struct nttrans_op *op;
 492
 493         SMBSRV_CHECK_ASYNC_STATUS(op, struct nttrans_op);
 494
 495         trans = op->trans;
 496
 497         /* if this function needs work to form the nttrans reply buffer, then
 498            call that now */
 499         if (op->send_fn != NULL) {
 500                 NTSTATUS status;
 501                 status = op->send_fn(op);
 502                 if (!NT_STATUS_IS_OK(status)) {
 503                         smbsrv_send_error(req, status);
 504                         return;
 505                 }
 506         }
 507
 508         smbsrv_setup_reply(req, 18 + trans->out.setup_count, 0);
 509
 510         /* note that we don't check the max_setup count (matching w2k3
 511            behaviour) */
 512
 513         if (trans->out.params.length > trans->in.max_param) {
 514                 smbsrv_setup_error(req, NT_STATUS_BUFFER_TOO_SMALL);
 515                 trans->out.params.length = trans->in.max_param;
 516         }
 517         if (trans->out.data.length > trans->in.max_data) {
 518                 smbsrv_setup_error(req, NT_STATUS_BUFFER_TOO_SMALL);
 519                 trans->out.data.length = trans->in.max_data;
 520         }
 521
 522         params_left = trans->out.params.length;
 523         data_left   = trans->out.data.length;
 524         params      = trans->out.params.data;
 525         data        = trans->out.data.data;
 526
 527         /* we need to divide up the reply into chunks that fit into
 528            the negotiated buffer size */
 529         do {
 530                 uint32_t this_data, this_param, max_bytes;
 531                 uint_t align1 = 1, align2 = (params_left ? 2 : 0);
 532                 struct smbsrv_request *this_req;
 533
 534                 max_bytes = req_max_data(req) - (align1 + align2);
 535
 536                 this_param = params_left;
 537                 if (this_param > max_bytes) {
 538                         this_param = max_bytes;
 539                 }
 540                 max_bytes -= this_param;
 541
 542                 this_data = data_left;
 543                 if (this_data > max_bytes) {
 544                         this_data = max_bytes;
 545                 }
 546
 547                 /* don't destroy unless this is the last chunk */
 548                 if (params_left - this_param != 0 ||
 549                     data_left - this_data != 0) {
 550                         this_req = smbsrv_setup_secondary_request(req);
 551                 } else {
 552                         this_req = req;
 553                 }
 554
 555                 req_grow_data(this_req, this_param + this_data + (align1 + align2));
 556
 557                 SSVAL(this_req->out.vwv, 0, 0); /* reserved */
 558                 SCVAL(this_req->out.vwv, 2, 0); /* reserved */
 559                 SIVAL(this_req->out.vwv, 3, trans->out.params.length);
 560                 SIVAL(this_req->out.vwv, 7, trans->out.data.length);
 561
 562                 SIVAL(this_req->out.vwv, 11, this_param);
 563                 SIVAL(this_req->out.vwv, 15, align1 + PTR_DIFF(this_req->out.data, this_req->out.hdr));
 564                 SIVAL(this_req->out.vwv, 19, PTR_DIFF(params, trans->out.params.data));
 565
 566                 SIVAL(this_req->out.vwv, 23, this_data);
 567                 SIVAL(this_req->out.vwv, 27, align1 + align2 +
 568                       PTR_DIFF(this_req->out.data + this_param, this_req->out.hdr));
 569                 SIVAL(this_req->out.vwv, 31, PTR_DIFF(data, trans->out.data.data));
 570
 571                 SCVAL(this_req->out.vwv, 35, trans->out.setup_count);
 572                 memcpy((char *)(this_req->out.vwv) + VWV(18), trans->out.setup,
 573                        sizeof(uint16_t) * trans->out.setup_count);
 574                 memset(this_req->out.data, 0, align1);
 575                 if (this_param != 0) {
 576                         memcpy(this_req->out.data + align1, params, this_param);
 577                 }
 578                 memset(this_req->out.data+this_param+align1, 0, align2);
 579                 if (this_data != 0) {
 580                         memcpy(this_req->out.data+this_param+align1+align2,
 581                                data, this_data);
 582                 }
 583
 584                 params_left -= this_param;
 585                 data_left -= this_data;
 586                 params += this_param;
 587                 data += this_data;
 588
 589                 smbsrv_send_reply(this_req);
 590         } while (params_left != 0 || data_left != 0);
 591 }
 592
 593 /*
 594   send a continue request
 595 */
 596 static void reply_nttrans_continue(struct smbsrv_request *req, struct smb_nttrans *trans)
 597 {
 598         struct smbsrv_request *req2;
 599         struct smbsrv_trans_partial *tp;
 600         int count;
 601
 602         /* make sure they don't flood us */
 603         for (count=0,tp=req->smb_conn->trans_partial;tp;tp=tp->next) count++;
 604         if (count > 100) {
 605                 smbsrv_send_error(req, NT_STATUS_INSUFFICIENT_RESOURCES);
 606                 return;
 607         }
 608
 609         tp = talloc(req, struct smbsrv_trans_partial);
 610
 611         tp->req = req;
 612         tp->u.nttrans = trans;
 613         tp->command = SMBnttrans;
 614
 615         DLIST_ADD(req->smb_conn->trans_partial, tp);
 616         talloc_set_destructor(tp, smbsrv_trans_partial_destructor);
 617
 618         req2 = smbsrv_setup_secondary_request(req);
 619
 620         /* send a 'please continue' reply */
 621         smbsrv_setup_reply(req2, 0, 0);
 622         smbsrv_send_reply(req2);
 623 }
 624
 625
 626 /*
 627   answer a reconstructed trans request
 628 */
 629 static void reply_nttrans_complete(struct smbsrv_request *req, struct smb_nttrans *trans)
 630 {
 631         struct nttrans_op *op;
 632
 633         SMBSRV_TALLOC_IO_PTR(op, struct nttrans_op);
 634         SMBSRV_SETUP_NTVFS_REQUEST(reply_nttrans_send, NTVFS_ASYNC_STATE_MAY_ASYNC);
 635
 636         op->trans       = trans;
 637         op->op_info     = NULL;
 638         op->send_fn     = NULL;
 639
 640         /* its a full request, give it to the backend */
 641         ZERO_STRUCT(trans->out);
 642         SMBSRV_CALL_NTVFS_BACKEND(nttrans_backend(req, op));
 643 }
 644
 645
 646 /****************************************************************************
 647  Reply to an SMBnttrans request
 648 ****************************************************************************/
 649 void smbsrv_reply_nttrans(struct smbsrv_request *req)
 650 {
 651         struct smb_nttrans *trans;
 652         uint32_t param_ofs, data_ofs;
 653         uint32_t param_count, data_count;
 654         uint32_t param_total, data_total;
 655
 656         /* parse request */
 657         if (req->in.wct < 19) {
 658                 smbsrv_send_error(req, NT_STATUS_FOOBAR);
 659                 return;
 660         }
 661
 662         trans = talloc(req, struct smb_nttrans);
 663         if (trans == NULL) {
 664                 smbsrv_send_error(req, NT_STATUS_NO_MEMORY);
 665                 return;
 666         }
 667
 668         trans->in.max_setup  = CVAL(req->in.vwv, 0);
 669         param_total          = IVAL(req->in.vwv, 3);
 670         data_total           = IVAL(req->in.vwv, 7);
 671         trans->in.max_param  = IVAL(req->in.vwv, 11);
 672         trans->in.max_data   = IVAL(req->in.vwv, 15);
 673         param_count          = IVAL(req->in.vwv, 19);
 674         param_ofs            = IVAL(req->in.vwv, 23);
 675         data_count           = IVAL(req->in.vwv, 27);
 676         data_ofs             = IVAL(req->in.vwv, 31);
 677         trans->in.setup_count= CVAL(req->in.vwv, 35);
 678         trans->in.function   = SVAL(req->in.vwv, 36);
 679
 680         if (req->in.wct != 19 + trans->in.setup_count) {
 681                 smbsrv_send_error(req, NT_STATUS_DOS(ERRSRV, ERRerror));
 682                 return;
 683         }
 684
 685         /* parse out the setup words */
 686         trans->in.setup = talloc_array(req, uint8_t, trans->in.setup_count*2);
 687         if (!trans->in.setup) {
 688                 smbsrv_send_error(req, NT_STATUS_NO_MEMORY);
 689                 return;
 690         }
 691         memcpy(trans->in.setup, (char *)(req->in.vwv) + VWV(19),
 692                sizeof(uint16_t) * trans->in.setup_count);
 693
 694         if (!req_pull_blob(&req->in.bufinfo, req->in.hdr + param_ofs, param_count, &trans->in.params) ||
 695             !req_pull_blob(&req->in.bufinfo, req->in.hdr + data_ofs, data_count, &trans->in.data)) {
 696                 smbsrv_send_error(req, NT_STATUS_FOOBAR);
 697                 return;
 698         }
 699
 700         /* is it a partial request? if so, then send a 'send more' message */
 701         if (param_total > param_count || data_total > data_count) {
 702                 reply_nttrans_continue(req, trans);
 703                 return;
 704         }
 705
 706         reply_nttrans_complete(req, trans);
 707 }
 708
 709
 710 /****************************************************************************
 711  Reply to an SMBnttranss request
 712 ****************************************************************************/
 713 void smbsrv_reply_nttranss(struct smbsrv_request *req)
 714 {
 715         struct smbsrv_trans_partial *tp;
 716         struct smb_nttrans *trans = NULL;
 717         uint32_t param_ofs, data_ofs;
 718         uint32_t param_count, data_count;
 719         uint32_t param_disp, data_disp;
 720         uint32_t param_total, data_total;
 721         DATA_BLOB params, data;
 722
 723         /* parse request */
 724         if (req->in.wct != 18) {
 725                 smbsrv_send_error(req, NT_STATUS_DOS(ERRSRV, ERRerror));
 726                 return;
 727         }
 728
 729         for (tp=req->smb_conn->trans_partial;tp;tp=tp->next) {
 730                 if (tp->command == SMBnttrans &&
 731                     SVAL(tp->req->in.hdr, HDR_MID) == SVAL(req->in.hdr, HDR_MID)) {
 732 /* TODO: check the VUID, PID and TID too? */
 733                         break;
 734                 }
 735         }
 736
 737         if (tp == NULL) {
 738                 smbsrv_send_error(req, NT_STATUS_INVALID_PARAMETER);
 739                 return;
 740         }
 741
 742         trans = tp->u.nttrans;
 743
 744         param_total           = IVAL(req->in.vwv, 3);
 745         data_total            = IVAL(req->in.vwv, 7);
 746         param_count           = IVAL(req->in.vwv, 11);
 747         param_ofs             = IVAL(req->in.vwv, 15);
 748         param_disp            = IVAL(req->in.vwv, 19);
 749         data_count            = IVAL(req->in.vwv, 23);
 750         data_ofs              = IVAL(req->in.vwv, 27);
 751         data_disp             = IVAL(req->in.vwv, 31);
 752
 753         if (!req_pull_blob(&req->in.bufinfo, req->in.hdr + param_ofs, param_count, &params) ||
 754             !req_pull_blob(&req->in.bufinfo, req->in.hdr + data_ofs, data_count, &data)) {
 755                 smbsrv_send_error(req, NT_STATUS_INVALID_PARAMETER);
 756                 return;
 757         }
 758
 759         /* only allow contiguous requests */
 760         if ((param_count != 0 &&
 761              param_disp != trans->in.params.length) ||
 762             (data_count != 0 &&
 763              data_disp != trans->in.data.length)) {
 764                 smbsrv_send_error(req, NT_STATUS_INVALID_PARAMETER);
 765                 return;
 766         }
 767
 768         /* add to the existing request */
 769         if (param_count != 0) {
 770                 trans->in.params.data = talloc_realloc(trans,
 771                                                        trans->in.params.data,
 772                                                        uint8_t,
 773                                                        param_disp + param_count);
 774                 if (trans->in.params.data == NULL) {
 775                         smbsrv_send_error(tp->req, NT_STATUS_NO_MEMORY);
 776                         return;
 777                 }
 778                 trans->in.params.length = param_disp + param_count;
 779         }
 780
 781         if (data_count != 0) {
 782                 trans->in.data.data = talloc_realloc(trans,
 783                                                      trans->in.data.data,
 784                                                      uint8_t,
 785                                                      data_disp + data_count);
 786                 if (trans->in.data.data == NULL) {
 787                         smbsrv_send_error(tp->req, NT_STATUS_NO_MEMORY);
 788                         return;
 789                 }
 790                 trans->in.data.length = data_disp + data_count;
 791         }
 792
 793         memcpy(trans->in.params.data + param_disp, params.data, params.length);
 794         memcpy(trans->in.data.data + data_disp, data.data, data.length);
 795
 796         /* the sequence number of the reply is taken from the last secondary
 797            response */
 798         tp->req->seq_num = req->seq_num;
 799
 800         /* we don't reply to Transs2 requests */
 801         talloc_free(req);
 802
 803         if (trans->in.params.length == param_total &&
 804             trans->in.data.length == data_total) {
 805                 /* its now complete */
 806                 req = tp->req;
 807                 talloc_free(tp);
 808                 reply_nttrans_complete(req, trans);
 809         }
 810         return;
 811 }