2011-05-05 20:12:07 +04:00
/*
Unix SMB / CIFS implementation .
smb2 lib
Copyright ( C ) Volker Lendecke 2011
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"
2012-05-15 13:11:24 +04:00
# include "system/network.h"
2011-05-05 20:12:07 +04:00
# include "lib/util/tevent_ntstatus.h"
2012-05-15 13:11:24 +04:00
# include "smb_common.h"
# include "smbXcli_base.h"
2011-05-05 20:12:07 +04:00
struct smb2cli_write_state {
uint8_t fixed [ 48 ] ;
2011-09-05 20:22:57 +04:00
uint8_t dyn_pad [ 1 ] ;
2013-08-08 01:41:24 +04:00
uint32_t written ;
2011-05-05 20:12:07 +04:00
} ;
static void smb2cli_write_done ( struct tevent_req * subreq ) ;
struct tevent_req * smb2cli_write_send ( TALLOC_CTX * mem_ctx ,
struct tevent_context * ev ,
2012-05-10 20:40:56 +04:00
struct smbXcli_conn * conn ,
uint32_t timeout_msec ,
struct smbXcli_session * session ,
2012-07-24 00:32:49 +04:00
struct smbXcli_tcon * tcon ,
2011-05-05 20:12:07 +04:00
uint32_t length ,
uint64_t offset ,
uint64_t fid_persistent ,
uint64_t fid_volatile ,
uint32_t remaining_bytes ,
uint32_t flags ,
const uint8_t * data )
{
struct tevent_req * req , * subreq ;
struct smb2cli_write_state * state ;
uint8_t * fixed ;
2011-09-05 20:22:57 +04:00
const uint8_t * dyn ;
size_t dyn_len ;
2011-05-05 20:12:07 +04:00
req = tevent_req_create ( mem_ctx , & state ,
struct smb2cli_write_state ) ;
if ( req = = NULL ) {
return NULL ;
}
fixed = state - > fixed ;
SSVAL ( fixed , 0 , 49 ) ;
SSVAL ( fixed , 2 , SMB2_HDR_BODY + 48 ) ;
SIVAL ( fixed , 4 , length ) ;
SBVAL ( fixed , 8 , offset ) ;
SBVAL ( fixed , 16 , fid_persistent ) ;
SBVAL ( fixed , 24 , fid_volatile ) ;
SIVAL ( fixed , 36 , remaining_bytes ) ;
SIVAL ( fixed , 44 , flags ) ;
2011-09-05 20:22:57 +04:00
if ( length > 0 ) {
dyn = data ;
dyn_len = length ;
} else {
dyn = state - > dyn_pad ; ;
dyn_len = sizeof ( state - > dyn_pad ) ;
}
2012-05-10 20:40:56 +04:00
subreq = smb2cli_req_send ( state , ev , conn , SMB2_OP_WRITE ,
2011-08-12 19:26:13 +04:00
0 , 0 , /* flags */
2012-05-10 20:40:56 +04:00
timeout_msec ,
2012-07-25 12:36:27 +04:00
tcon ,
2012-05-10 20:40:56 +04:00
session ,
2011-05-05 20:12:07 +04:00
state - > fixed , sizeof ( state - > fixed ) ,
2013-08-13 12:25:52 +04:00
dyn , dyn_len ,
0 ) ; /* max_dyn_len */
2011-05-05 20:12:07 +04:00
if ( tevent_req_nomem ( subreq , req ) ) {
return tevent_req_post ( req , ev ) ;
}
tevent_req_set_callback ( subreq , smb2cli_write_done , req ) ;
return req ;
}
static void smb2cli_write_done ( struct tevent_req * subreq )
{
struct tevent_req * req =
tevent_req_callback_data ( subreq ,
struct tevent_req ) ;
2013-08-08 01:41:24 +04:00
struct smb2cli_write_state * state =
tevent_req_data ( req ,
struct smb2cli_write_state ) ;
2011-05-05 20:12:07 +04:00
NTSTATUS status ;
2013-08-08 01:41:24 +04:00
struct iovec * iov ;
2011-08-31 02:40:06 +04:00
static const struct smb2cli_req_expected_response expected [ ] = {
{
. status = NT_STATUS_OK ,
. body_size = 0x11
}
} ;
2011-05-05 20:12:07 +04:00
2013-08-08 01:41:24 +04:00
status = smb2cli_req_recv ( subreq , state , & iov ,
2011-08-31 02:40:06 +04:00
expected , ARRAY_SIZE ( expected ) ) ;
2012-09-28 21:48:26 +04:00
TALLOC_FREE ( subreq ) ;
2011-05-05 20:12:07 +04:00
if ( tevent_req_nterror ( req , status ) ) {
return ;
}
2013-08-08 01:41:24 +04:00
state - > written = IVAL ( iov [ 1 ] . iov_base , 4 ) ;
2011-05-05 20:12:07 +04:00
tevent_req_done ( req ) ;
}
2013-08-08 01:41:24 +04:00
NTSTATUS smb2cli_write_recv ( struct tevent_req * req , uint32_t * written )
2011-05-05 20:12:07 +04:00
{
2013-08-08 01:41:24 +04:00
struct smb2cli_write_state * state =
tevent_req_data ( req ,
struct smb2cli_write_state ) ;
NTSTATUS status ;
if ( tevent_req_is_nterror ( req , & status ) ) {
tevent_req_received ( req ) ;
return status ;
}
if ( written ) {
* written = state - > written ;
}
tevent_req_received ( req ) ;
return NT_STATUS_OK ;
2011-05-05 20:12:07 +04:00
}
2012-05-10 20:40:56 +04:00
NTSTATUS smb2cli_write ( struct smbXcli_conn * conn ,
uint32_t timeout_msec ,
struct smbXcli_session * session ,
2012-07-24 00:32:49 +04:00
struct smbXcli_tcon * tcon ,
2011-05-05 20:12:07 +04:00
uint32_t length ,
uint64_t offset ,
uint64_t fid_persistent ,
uint64_t fid_volatile ,
uint32_t remaining_bytes ,
uint32_t flags ,
2013-08-08 01:41:24 +04:00
const uint8_t * data ,
uint32_t * written )
2011-05-05 20:12:07 +04:00
{
TALLOC_CTX * frame = talloc_stackframe ( ) ;
2012-05-14 11:56:47 +04:00
struct tevent_context * ev ;
2011-05-05 20:12:07 +04:00
struct tevent_req * req ;
NTSTATUS status = NT_STATUS_NO_MEMORY ;
2012-05-10 20:40:56 +04:00
if ( smbXcli_conn_has_async_calls ( conn ) ) {
2011-05-05 20:12:07 +04:00
/*
* Can ' t use sync call while an async call is in flight
*/
status = NT_STATUS_INVALID_PARAMETER ;
goto fail ;
}
2013-02-18 12:07:11 +04:00
ev = samba_tevent_context_init ( frame ) ;
2011-05-05 20:12:07 +04:00
if ( ev = = NULL ) {
goto fail ;
}
2012-07-24 00:32:49 +04:00
req = smb2cli_write_send ( frame , ev , conn , timeout_msec ,
session , tcon ,
length , offset ,
2011-05-05 20:12:07 +04:00
fid_persistent , fid_volatile ,
remaining_bytes , flags , data ) ;
if ( req = = NULL ) {
goto fail ;
}
if ( ! tevent_req_poll_ntstatus ( req , ev , & status ) ) {
goto fail ;
}
2013-08-08 01:41:24 +04:00
status = smb2cli_write_recv ( req , written ) ;
2011-05-05 20:12:07 +04:00
fail :
TALLOC_FREE ( frame ) ;
return status ;
}