mirror of
https://github.com/samba-team/samba.git
synced 2024-12-25 23:21:54 +03:00
774a30989a
Jeremy
(This used to be commit 816aea6c1a
)
687 lines
20 KiB
C
687 lines
20 KiB
C
/*
|
|
Unix SMB/CIFS implementation.
|
|
client transaction calls
|
|
Copyright (C) Andrew Tridgell 1994-1998
|
|
|
|
This program is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation; either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "includes.h"
|
|
|
|
|
|
/****************************************************************************
|
|
Send a SMB trans or trans2 request.
|
|
****************************************************************************/
|
|
|
|
bool cli_send_trans(struct cli_state *cli, int trans,
|
|
const char *pipe_name,
|
|
int fid, int flags,
|
|
uint16 *setup, unsigned int lsetup, unsigned int msetup,
|
|
const char *param, unsigned int lparam, unsigned int mparam,
|
|
const char *data, unsigned int ldata, unsigned int mdata)
|
|
{
|
|
unsigned int i;
|
|
unsigned int this_ldata,this_lparam;
|
|
unsigned int tot_data=0,tot_param=0;
|
|
char *outdata,*outparam;
|
|
char *p;
|
|
int pipe_name_len=0;
|
|
uint16 mid;
|
|
|
|
this_lparam = MIN(lparam,cli->max_xmit - (500+lsetup*2)); /* hack */
|
|
this_ldata = MIN(ldata,cli->max_xmit - (500+lsetup*2+this_lparam));
|
|
|
|
memset(cli->outbuf,'\0',smb_size);
|
|
set_message(cli->outbuf,14+lsetup,0,True);
|
|
SCVAL(cli->outbuf,smb_com,trans);
|
|
SSVAL(cli->outbuf,smb_tid, cli->cnum);
|
|
cli_setup_packet(cli);
|
|
|
|
/*
|
|
* Save the mid we're using. We need this for finding
|
|
* signing replies.
|
|
*/
|
|
|
|
mid = cli->mid;
|
|
|
|
if (pipe_name) {
|
|
pipe_name_len = clistr_push(cli, smb_buf(cli->outbuf), pipe_name, -1, STR_TERMINATE);
|
|
}
|
|
|
|
outparam = smb_buf(cli->outbuf)+(trans==SMBtrans ? pipe_name_len : 3);
|
|
outdata = outparam+this_lparam;
|
|
|
|
/* primary request */
|
|
SSVAL(cli->outbuf,smb_tpscnt,lparam); /* tpscnt */
|
|
SSVAL(cli->outbuf,smb_tdscnt,ldata); /* tdscnt */
|
|
SSVAL(cli->outbuf,smb_mprcnt,mparam); /* mprcnt */
|
|
SSVAL(cli->outbuf,smb_mdrcnt,mdata); /* mdrcnt */
|
|
SCVAL(cli->outbuf,smb_msrcnt,msetup); /* msrcnt */
|
|
SSVAL(cli->outbuf,smb_flags,flags); /* flags */
|
|
SIVAL(cli->outbuf,smb_timeout,0); /* timeout */
|
|
SSVAL(cli->outbuf,smb_pscnt,this_lparam); /* pscnt */
|
|
SSVAL(cli->outbuf,smb_psoff,smb_offset(outparam,cli->outbuf)); /* psoff */
|
|
SSVAL(cli->outbuf,smb_dscnt,this_ldata); /* dscnt */
|
|
SSVAL(cli->outbuf,smb_dsoff,smb_offset(outdata,cli->outbuf)); /* dsoff */
|
|
SCVAL(cli->outbuf,smb_suwcnt,lsetup); /* suwcnt */
|
|
for (i=0;i<lsetup;i++) /* setup[] */
|
|
SSVAL(cli->outbuf,smb_setup+i*2,setup[i]);
|
|
p = smb_buf(cli->outbuf);
|
|
if (trans != SMBtrans) {
|
|
*p++ = 0; /* put in a null smb_name */
|
|
*p++ = 'D'; *p++ = ' '; /* observed in OS/2 */
|
|
}
|
|
if (this_lparam) /* param[] */
|
|
memcpy(outparam,param,this_lparam);
|
|
if (this_ldata) /* data[] */
|
|
memcpy(outdata,data,this_ldata);
|
|
cli_setup_bcc(cli, outdata+this_ldata);
|
|
|
|
show_msg(cli->outbuf);
|
|
|
|
if (!cli_send_smb(cli)) {
|
|
return False;
|
|
}
|
|
|
|
if (this_ldata < ldata || this_lparam < lparam) {
|
|
/* receive interim response */
|
|
if (!cli_receive_smb(cli) || cli_is_error(cli)) {
|
|
return(False);
|
|
}
|
|
|
|
tot_data = this_ldata;
|
|
tot_param = this_lparam;
|
|
|
|
while (tot_data < ldata || tot_param < lparam) {
|
|
this_lparam = MIN(lparam-tot_param,cli->max_xmit - 500); /* hack */
|
|
this_ldata = MIN(ldata-tot_data,cli->max_xmit - (500+this_lparam));
|
|
|
|
set_message(cli->outbuf,trans==SMBtrans?8:9,0,True);
|
|
SCVAL(cli->outbuf,smb_com,(trans==SMBtrans ? SMBtranss : SMBtranss2));
|
|
|
|
outparam = smb_buf(cli->outbuf);
|
|
outdata = outparam+this_lparam;
|
|
|
|
/* secondary request */
|
|
SSVAL(cli->outbuf,smb_tpscnt,lparam); /* tpscnt */
|
|
SSVAL(cli->outbuf,smb_tdscnt,ldata); /* tdscnt */
|
|
SSVAL(cli->outbuf,smb_spscnt,this_lparam); /* pscnt */
|
|
SSVAL(cli->outbuf,smb_spsoff,smb_offset(outparam,cli->outbuf)); /* psoff */
|
|
SSVAL(cli->outbuf,smb_spsdisp,tot_param); /* psdisp */
|
|
SSVAL(cli->outbuf,smb_sdscnt,this_ldata); /* dscnt */
|
|
SSVAL(cli->outbuf,smb_sdsoff,smb_offset(outdata,cli->outbuf)); /* dsoff */
|
|
SSVAL(cli->outbuf,smb_sdsdisp,tot_data); /* dsdisp */
|
|
if (trans==SMBtrans2)
|
|
SSVALS(cli->outbuf,smb_sfid,fid); /* fid */
|
|
if (this_lparam) /* param[] */
|
|
memcpy(outparam,param+tot_param,this_lparam);
|
|
if (this_ldata) /* data[] */
|
|
memcpy(outdata,data+tot_data,this_ldata);
|
|
cli_setup_bcc(cli, outdata+this_ldata);
|
|
|
|
/*
|
|
* Save the mid we're using. We need this for finding
|
|
* signing replies.
|
|
*/
|
|
mid = cli->mid;
|
|
|
|
show_msg(cli->outbuf);
|
|
if (!cli_send_smb(cli)) {
|
|
return False;
|
|
}
|
|
|
|
/* Ensure we use the same mid for the secondaries. */
|
|
cli->mid = mid;
|
|
|
|
tot_data += this_ldata;
|
|
tot_param += this_lparam;
|
|
}
|
|
}
|
|
|
|
return(True);
|
|
}
|
|
|
|
/****************************************************************************
|
|
Receive a SMB trans or trans2 response allocating the necessary memory.
|
|
****************************************************************************/
|
|
|
|
bool cli_receive_trans(struct cli_state *cli,int trans,
|
|
char **param, unsigned int *param_len,
|
|
char **data, unsigned int *data_len)
|
|
{
|
|
unsigned int total_data=0;
|
|
unsigned int total_param=0;
|
|
unsigned int this_data,this_param;
|
|
NTSTATUS status;
|
|
bool ret = False;
|
|
|
|
*data_len = *param_len = 0;
|
|
|
|
if (!cli_receive_smb(cli)) {
|
|
return False;
|
|
}
|
|
|
|
show_msg(cli->inbuf);
|
|
|
|
/* sanity check */
|
|
if (CVAL(cli->inbuf,smb_com) != trans) {
|
|
DEBUG(0,("Expected %s response, got command 0x%02x\n",
|
|
trans==SMBtrans?"SMBtrans":"SMBtrans2",
|
|
CVAL(cli->inbuf,smb_com)));
|
|
return False;
|
|
}
|
|
|
|
/*
|
|
* An NT RPC pipe call can return ERRDOS, ERRmoredata
|
|
* to a trans call. This is not an error and should not
|
|
* be treated as such. Note that STATUS_NO_MORE_FILES is
|
|
* returned when a trans2 findfirst/next finishes.
|
|
* When setting up an encrypted transport we can also
|
|
* see NT_STATUS_MORE_PROCESSING_REQUIRED here.
|
|
*
|
|
* Vista returns NT_STATUS_INACCESSIBLE_SYSTEM_SHORTCUT if the folder
|
|
* "<share>/Users/All Users" is enumerated. This is a special pseudo
|
|
* folder, and the response does not have parameters (nor a parameter
|
|
* length).
|
|
*/
|
|
status = cli_nt_error(cli);
|
|
|
|
if (!NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
|
|
if (NT_STATUS_IS_ERR(status) ||
|
|
NT_STATUS_EQUAL(status,STATUS_NO_MORE_FILES) ||
|
|
NT_STATUS_EQUAL(status,NT_STATUS_INACCESSIBLE_SYSTEM_SHORTCUT)) {
|
|
goto out;
|
|
}
|
|
}
|
|
|
|
/* parse out the lengths */
|
|
total_data = SVAL(cli->inbuf,smb_tdrcnt);
|
|
total_param = SVAL(cli->inbuf,smb_tprcnt);
|
|
|
|
/* allocate it */
|
|
if (total_data!=0) {
|
|
/* We know adding 2 is safe as total_data is an
|
|
* SVAL <= 0xFFFF. */
|
|
*data = (char *)SMB_REALLOC(*data,total_data+2);
|
|
if (!(*data)) {
|
|
DEBUG(0,("cli_receive_trans: failed to enlarge data buffer\n"));
|
|
goto out;
|
|
}
|
|
}
|
|
|
|
if (total_param!=0) {
|
|
/* We know adding 2 is safe as total_param is an
|
|
* SVAL <= 0xFFFF. */
|
|
*param = (char *)SMB_REALLOC(*param,total_param+2);
|
|
if (!(*param)) {
|
|
DEBUG(0,("cli_receive_trans: failed to enlarge param buffer\n"));
|
|
goto out;
|
|
}
|
|
}
|
|
|
|
for (;;) {
|
|
this_data = SVAL(cli->inbuf,smb_drcnt);
|
|
this_param = SVAL(cli->inbuf,smb_prcnt);
|
|
|
|
if (this_data + *data_len > total_data ||
|
|
this_param + *param_len > total_param) {
|
|
DEBUG(1,("Data overflow in cli_receive_trans\n"));
|
|
goto out;
|
|
}
|
|
|
|
if (this_data + *data_len < this_data ||
|
|
this_data + *data_len < *data_len ||
|
|
this_param + *param_len < this_param ||
|
|
this_param + *param_len < *param_len) {
|
|
DEBUG(1,("Data overflow in cli_receive_trans\n"));
|
|
goto out;
|
|
}
|
|
|
|
if (this_data) {
|
|
unsigned int data_offset_out = SVAL(cli->inbuf,smb_drdisp);
|
|
unsigned int data_offset_in = SVAL(cli->inbuf,smb_droff);
|
|
|
|
if (data_offset_out > total_data ||
|
|
data_offset_out + this_data > total_data ||
|
|
data_offset_out + this_data < data_offset_out ||
|
|
data_offset_out + this_data < this_data) {
|
|
DEBUG(1,("Data overflow in cli_receive_trans\n"));
|
|
goto out;
|
|
}
|
|
if (data_offset_in > cli->bufsize ||
|
|
data_offset_in + this_data > cli->bufsize ||
|
|
data_offset_in + this_data < data_offset_in ||
|
|
data_offset_in + this_data < this_data) {
|
|
DEBUG(1,("Data overflow in cli_receive_trans\n"));
|
|
goto out;
|
|
}
|
|
|
|
memcpy(*data + data_offset_out, smb_base(cli->inbuf) + data_offset_in, this_data);
|
|
}
|
|
if (this_param) {
|
|
unsigned int param_offset_out = SVAL(cli->inbuf,smb_prdisp);
|
|
unsigned int param_offset_in = SVAL(cli->inbuf,smb_proff);
|
|
|
|
if (param_offset_out > total_param ||
|
|
param_offset_out + this_param > total_param ||
|
|
param_offset_out + this_param < param_offset_out ||
|
|
param_offset_out + this_param < this_param) {
|
|
DEBUG(1,("Param overflow in cli_receive_trans\n"));
|
|
goto out;
|
|
}
|
|
if (param_offset_in > cli->bufsize ||
|
|
param_offset_in + this_param > cli->bufsize ||
|
|
param_offset_in + this_param < param_offset_in ||
|
|
param_offset_in + this_param < this_param) {
|
|
DEBUG(1,("Param overflow in cli_receive_trans\n"));
|
|
goto out;
|
|
}
|
|
|
|
memcpy(*param + param_offset_out, smb_base(cli->inbuf) + param_offset_in, this_param);
|
|
}
|
|
*data_len += this_data;
|
|
*param_len += this_param;
|
|
|
|
if (total_data <= *data_len && total_param <= *param_len) {
|
|
ret = True;
|
|
break;
|
|
}
|
|
|
|
if (!cli_receive_smb(cli)) {
|
|
goto out;
|
|
}
|
|
|
|
show_msg(cli->inbuf);
|
|
|
|
/* sanity check */
|
|
if (CVAL(cli->inbuf,smb_com) != trans) {
|
|
DEBUG(0,("Expected %s response, got command 0x%02x\n",
|
|
trans==SMBtrans?"SMBtrans":"SMBtrans2",
|
|
CVAL(cli->inbuf,smb_com)));
|
|
goto out;
|
|
}
|
|
if (!NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
|
|
if (NT_STATUS_IS_ERR(cli_nt_error(cli))) {
|
|
goto out;
|
|
}
|
|
}
|
|
|
|
/* parse out the total lengths again - they can shrink! */
|
|
if (SVAL(cli->inbuf,smb_tdrcnt) < total_data)
|
|
total_data = SVAL(cli->inbuf,smb_tdrcnt);
|
|
if (SVAL(cli->inbuf,smb_tprcnt) < total_param)
|
|
total_param = SVAL(cli->inbuf,smb_tprcnt);
|
|
|
|
if (total_data <= *data_len && total_param <= *param_len) {
|
|
ret = True;
|
|
break;
|
|
}
|
|
}
|
|
|
|
out:
|
|
|
|
if (ret) {
|
|
/* Ensure the last 2 bytes of param and data are 2 null
|
|
* bytes. These are malloc'ed, but not included in any
|
|
* length counts. This allows cli_XX string reading functions
|
|
* to safely null terminate. */
|
|
if (total_data) {
|
|
SSVAL(*data,total_data,0);
|
|
}
|
|
if (total_param) {
|
|
SSVAL(*param,total_param,0);
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
/****************************************************************************
|
|
Send a SMB nttrans request.
|
|
****************************************************************************/
|
|
|
|
bool cli_send_nt_trans(struct cli_state *cli,
|
|
int function,
|
|
int flags,
|
|
uint16 *setup, unsigned int lsetup, unsigned int msetup,
|
|
char *param, unsigned int lparam, unsigned int mparam,
|
|
char *data, unsigned int ldata, unsigned int mdata)
|
|
{
|
|
unsigned int i;
|
|
unsigned int this_ldata,this_lparam;
|
|
unsigned int tot_data=0,tot_param=0;
|
|
uint16 mid;
|
|
char *outdata,*outparam;
|
|
|
|
this_lparam = MIN(lparam,cli->max_xmit - (500+lsetup*2)); /* hack */
|
|
this_ldata = MIN(ldata,cli->max_xmit - (500+lsetup*2+this_lparam));
|
|
|
|
memset(cli->outbuf,'\0',smb_size);
|
|
set_message(cli->outbuf,19+lsetup,0,True);
|
|
SCVAL(cli->outbuf,smb_com,SMBnttrans);
|
|
SSVAL(cli->outbuf,smb_tid, cli->cnum);
|
|
cli_setup_packet(cli);
|
|
|
|
/*
|
|
* Save the mid we're using. We need this for finding
|
|
* signing replies.
|
|
*/
|
|
|
|
mid = cli->mid;
|
|
|
|
outparam = smb_buf(cli->outbuf)+3;
|
|
outdata = outparam+this_lparam;
|
|
|
|
/* primary request */
|
|
SCVAL(cli->outbuf,smb_nt_MaxSetupCount,msetup);
|
|
SCVAL(cli->outbuf,smb_nt_Flags,flags);
|
|
SIVAL(cli->outbuf,smb_nt_TotalParameterCount, lparam);
|
|
SIVAL(cli->outbuf,smb_nt_TotalDataCount, ldata);
|
|
SIVAL(cli->outbuf,smb_nt_MaxParameterCount, mparam);
|
|
SIVAL(cli->outbuf,smb_nt_MaxDataCount, mdata);
|
|
SIVAL(cli->outbuf,smb_nt_ParameterCount, this_lparam);
|
|
SIVAL(cli->outbuf,smb_nt_ParameterOffset, smb_offset(outparam,cli->outbuf));
|
|
SIVAL(cli->outbuf,smb_nt_DataCount, this_ldata);
|
|
SIVAL(cli->outbuf,smb_nt_DataOffset, smb_offset(outdata,cli->outbuf));
|
|
SIVAL(cli->outbuf,smb_nt_SetupCount, lsetup);
|
|
SIVAL(cli->outbuf,smb_nt_Function, function);
|
|
for (i=0;i<lsetup;i++) /* setup[] */
|
|
SSVAL(cli->outbuf,smb_nt_SetupStart+i*2,setup[i]);
|
|
|
|
if (this_lparam) /* param[] */
|
|
memcpy(outparam,param,this_lparam);
|
|
if (this_ldata) /* data[] */
|
|
memcpy(outdata,data,this_ldata);
|
|
|
|
cli_setup_bcc(cli, outdata+this_ldata);
|
|
|
|
show_msg(cli->outbuf);
|
|
if (!cli_send_smb(cli)) {
|
|
return False;
|
|
}
|
|
|
|
if (this_ldata < ldata || this_lparam < lparam) {
|
|
/* receive interim response */
|
|
if (!cli_receive_smb(cli) || cli_is_error(cli)) {
|
|
return(False);
|
|
}
|
|
|
|
tot_data = this_ldata;
|
|
tot_param = this_lparam;
|
|
|
|
while (tot_data < ldata || tot_param < lparam) {
|
|
this_lparam = MIN(lparam-tot_param,cli->max_xmit - 500); /* hack */
|
|
this_ldata = MIN(ldata-tot_data,cli->max_xmit - (500+this_lparam));
|
|
|
|
set_message(cli->outbuf,18,0,True);
|
|
SCVAL(cli->outbuf,smb_com,SMBnttranss);
|
|
|
|
/* XXX - these should probably be aligned */
|
|
outparam = smb_buf(cli->outbuf);
|
|
outdata = outparam+this_lparam;
|
|
|
|
/* secondary request */
|
|
SIVAL(cli->outbuf,smb_nts_TotalParameterCount,lparam);
|
|
SIVAL(cli->outbuf,smb_nts_TotalDataCount,ldata);
|
|
SIVAL(cli->outbuf,smb_nts_ParameterCount,this_lparam);
|
|
SIVAL(cli->outbuf,smb_nts_ParameterOffset,smb_offset(outparam,cli->outbuf));
|
|
SIVAL(cli->outbuf,smb_nts_ParameterDisplacement,tot_param);
|
|
SIVAL(cli->outbuf,smb_nts_DataCount,this_ldata);
|
|
SIVAL(cli->outbuf,smb_nts_DataOffset,smb_offset(outdata,cli->outbuf));
|
|
SIVAL(cli->outbuf,smb_nts_DataDisplacement,tot_data);
|
|
if (this_lparam) /* param[] */
|
|
memcpy(outparam,param+tot_param,this_lparam);
|
|
if (this_ldata) /* data[] */
|
|
memcpy(outdata,data+tot_data,this_ldata);
|
|
cli_setup_bcc(cli, outdata+this_ldata);
|
|
|
|
/*
|
|
* Save the mid we're using. We need this for finding
|
|
* signing replies.
|
|
*/
|
|
mid = cli->mid;
|
|
|
|
show_msg(cli->outbuf);
|
|
|
|
if (!cli_send_smb(cli)) {
|
|
return False;
|
|
}
|
|
|
|
/* Ensure we use the same mid for the secondaries. */
|
|
cli->mid = mid;
|
|
|
|
tot_data += this_ldata;
|
|
tot_param += this_lparam;
|
|
}
|
|
}
|
|
|
|
return(True);
|
|
}
|
|
|
|
/****************************************************************************
|
|
Receive a SMB nttrans response allocating the necessary memory.
|
|
****************************************************************************/
|
|
|
|
bool cli_receive_nt_trans(struct cli_state *cli,
|
|
char **param, unsigned int *param_len,
|
|
char **data, unsigned int *data_len)
|
|
{
|
|
unsigned int total_data=0;
|
|
unsigned int total_param=0;
|
|
unsigned int this_data,this_param;
|
|
uint8 eclass;
|
|
uint32 ecode;
|
|
bool ret = False;
|
|
|
|
*data_len = *param_len = 0;
|
|
|
|
if (!cli_receive_smb(cli)) {
|
|
return False;
|
|
}
|
|
|
|
show_msg(cli->inbuf);
|
|
|
|
/* sanity check */
|
|
if (CVAL(cli->inbuf,smb_com) != SMBnttrans) {
|
|
DEBUG(0,("Expected SMBnttrans response, got command 0x%02x\n",
|
|
CVAL(cli->inbuf,smb_com)));
|
|
return(False);
|
|
}
|
|
|
|
/*
|
|
* An NT RPC pipe call can return ERRDOS, ERRmoredata
|
|
* to a trans call. This is not an error and should not
|
|
* be treated as such.
|
|
*/
|
|
if (cli_is_dos_error(cli)) {
|
|
cli_dos_error(cli, &eclass, &ecode);
|
|
if (!(eclass == ERRDOS && ecode == ERRmoredata)) {
|
|
goto out;
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Likewise for NT_STATUS_BUFFER_TOO_SMALL
|
|
*/
|
|
if (cli_is_nt_error(cli)) {
|
|
if (!NT_STATUS_EQUAL(cli_nt_error(cli),
|
|
NT_STATUS_BUFFER_TOO_SMALL)) {
|
|
goto out;
|
|
}
|
|
}
|
|
|
|
/* parse out the lengths */
|
|
total_data = IVAL(cli->inbuf,smb_ntr_TotalDataCount);
|
|
total_param = IVAL(cli->inbuf,smb_ntr_TotalParameterCount);
|
|
/* Only allow 16 megs. */
|
|
if (total_param > 16*1024*1024) {
|
|
DEBUG(0,("cli_receive_nt_trans: param buffer too large %d\n",
|
|
total_param));
|
|
goto out;
|
|
}
|
|
if (total_data > 16*1024*1024) {
|
|
DEBUG(0,("cli_receive_nt_trans: data buffer too large %d\n",
|
|
total_data));
|
|
goto out;
|
|
}
|
|
|
|
/* allocate it */
|
|
if (total_data) {
|
|
/* We know adding 2 is safe as total_data is less
|
|
* than 16mb (above). */
|
|
*data = (char *)SMB_REALLOC(*data,total_data+2);
|
|
if (!(*data)) {
|
|
DEBUG(0,("cli_receive_nt_trans: failed to enlarge data buffer to %d\n",total_data));
|
|
goto out;
|
|
}
|
|
}
|
|
|
|
if (total_param) {
|
|
/* We know adding 2 is safe as total_param is less
|
|
* than 16mb (above). */
|
|
*param = (char *)SMB_REALLOC(*param,total_param+2);
|
|
if (!(*param)) {
|
|
DEBUG(0,("cli_receive_nt_trans: failed to enlarge param buffer to %d\n", total_param));
|
|
goto out;
|
|
}
|
|
}
|
|
|
|
while (1) {
|
|
this_data = SVAL(cli->inbuf,smb_ntr_DataCount);
|
|
this_param = SVAL(cli->inbuf,smb_ntr_ParameterCount);
|
|
|
|
if (this_data + *data_len > total_data ||
|
|
this_param + *param_len > total_param) {
|
|
DEBUG(1,("Data overflow in cli_receive_nt_trans\n"));
|
|
goto out;
|
|
}
|
|
|
|
if (this_data + *data_len < this_data ||
|
|
this_data + *data_len < *data_len ||
|
|
this_param + *param_len < this_param ||
|
|
this_param + *param_len < *param_len) {
|
|
DEBUG(1,("Data overflow in cli_receive_nt_trans\n"));
|
|
goto out;
|
|
}
|
|
|
|
if (this_data) {
|
|
unsigned int data_offset_out = SVAL(cli->inbuf,smb_ntr_DataDisplacement);
|
|
unsigned int data_offset_in = SVAL(cli->inbuf,smb_ntr_DataOffset);
|
|
|
|
if (data_offset_out > total_data ||
|
|
data_offset_out + this_data > total_data ||
|
|
data_offset_out + this_data < data_offset_out ||
|
|
data_offset_out + this_data < this_data) {
|
|
DEBUG(1,("Data overflow in cli_receive_nt_trans\n"));
|
|
goto out;
|
|
}
|
|
if (data_offset_in > cli->bufsize ||
|
|
data_offset_in + this_data > cli->bufsize ||
|
|
data_offset_in + this_data < data_offset_in ||
|
|
data_offset_in + this_data < this_data) {
|
|
DEBUG(1,("Data overflow in cli_receive_nt_trans\n"));
|
|
goto out;
|
|
}
|
|
|
|
memcpy(*data + data_offset_out, smb_base(cli->inbuf) + data_offset_in, this_data);
|
|
}
|
|
|
|
if (this_param) {
|
|
unsigned int param_offset_out = SVAL(cli->inbuf,smb_ntr_ParameterDisplacement);
|
|
unsigned int param_offset_in = SVAL(cli->inbuf,smb_ntr_ParameterOffset);
|
|
|
|
if (param_offset_out > total_param ||
|
|
param_offset_out + this_param > total_param ||
|
|
param_offset_out + this_param < param_offset_out ||
|
|
param_offset_out + this_param < this_param) {
|
|
DEBUG(1,("Param overflow in cli_receive_nt_trans\n"));
|
|
goto out;
|
|
}
|
|
if (param_offset_in > cli->bufsize ||
|
|
param_offset_in + this_param > cli->bufsize ||
|
|
param_offset_in + this_param < param_offset_in ||
|
|
param_offset_in + this_param < this_param) {
|
|
DEBUG(1,("Param overflow in cli_receive_nt_trans\n"));
|
|
goto out;
|
|
}
|
|
|
|
memcpy(*param + param_offset_out, smb_base(cli->inbuf) + param_offset_in, this_param);
|
|
}
|
|
|
|
*data_len += this_data;
|
|
*param_len += this_param;
|
|
|
|
if (total_data <= *data_len && total_param <= *param_len) {
|
|
ret = True;
|
|
break;
|
|
}
|
|
|
|
if (!cli_receive_smb(cli)) {
|
|
goto out;
|
|
}
|
|
|
|
show_msg(cli->inbuf);
|
|
|
|
/* sanity check */
|
|
if (CVAL(cli->inbuf,smb_com) != SMBnttrans) {
|
|
DEBUG(0,("Expected SMBnttrans response, got command 0x%02x\n",
|
|
CVAL(cli->inbuf,smb_com)));
|
|
goto out;
|
|
}
|
|
if (cli_is_dos_error(cli)) {
|
|
cli_dos_error(cli, &eclass, &ecode);
|
|
if(!(eclass == ERRDOS && ecode == ERRmoredata)) {
|
|
goto out;
|
|
}
|
|
}
|
|
/*
|
|
* Likewise for NT_STATUS_BUFFER_TOO_SMALL
|
|
*/
|
|
if (cli_is_nt_error(cli)) {
|
|
if (!NT_STATUS_EQUAL(cli_nt_error(cli),
|
|
NT_STATUS_BUFFER_TOO_SMALL)) {
|
|
goto out;
|
|
}
|
|
}
|
|
|
|
/* parse out the total lengths again - they can shrink! */
|
|
if (IVAL(cli->inbuf,smb_ntr_TotalDataCount) < total_data)
|
|
total_data = IVAL(cli->inbuf,smb_ntr_TotalDataCount);
|
|
if (IVAL(cli->inbuf,smb_ntr_TotalParameterCount) < total_param)
|
|
total_param = IVAL(cli->inbuf,smb_ntr_TotalParameterCount);
|
|
|
|
if (total_data <= *data_len && total_param <= *param_len) {
|
|
ret = True;
|
|
break;
|
|
}
|
|
}
|
|
|
|
out:
|
|
|
|
if (ret) {
|
|
/* Ensure the last 2 bytes of param and data are 2 null
|
|
* bytes. These are malloc'ed, but not included in any
|
|
* length counts. This allows cli_XX string reading functions
|
|
* to safely null terminate. */
|
|
if (total_data) {
|
|
SSVAL(*data,total_data,0);
|
|
}
|
|
if (total_param) {
|
|
SSVAL(*param,total_param,0);
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|