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 14:19:11 +04:00
# include "system/network.h"
2011-05-05 20:12:07 +04:00
# include "lib/util/tevent_ntstatus.h"
2012-05-15 14:19:11 +04:00
# include "smb_common.h"
# include "smbXcli_base.h"
2011-05-05 20:12:07 +04:00
struct smb2cli_query_directory_state {
uint8_t fixed [ 32 ] ;
2011-09-05 20:22:57 +04:00
uint8_t dyn_pad [ 1 ] ;
2011-07-08 19:57:56 +04:00
struct iovec * recv_iov ;
2011-05-05 20:12:07 +04:00
uint8_t * data ;
uint32_t data_length ;
} ;
static void smb2cli_query_directory_done ( struct tevent_req * subreq ) ;
struct tevent_req * smb2cli_query_directory_send ( TALLOC_CTX * mem_ctx ,
struct tevent_context * ev ,
2012-05-10 20:37:43 +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
uint8_t level ,
uint8_t flags ,
uint32_t file_index ,
uint64_t fid_persistent ,
uint64_t fid_volatile ,
const char * mask ,
uint32_t outbuf_len )
{
struct tevent_req * req , * subreq ;
struct smb2cli_query_directory_state * state ;
uint8_t * fixed ;
uint8_t * dyn ;
size_t dyn_len ;
req = tevent_req_create ( mem_ctx , & state ,
struct smb2cli_query_directory_state ) ;
if ( req = = NULL ) {
return NULL ;
}
if ( ! convert_string_talloc ( state , CH_UNIX , CH_UTF16 ,
2011-09-05 20:22:57 +04:00
mask , strlen ( mask ) ,
2011-05-05 20:12:07 +04:00
& dyn , & dyn_len ) ) {
2011-07-10 11:21:45 +04:00
tevent_req_oom ( req ) ;
2011-05-05 20:12:07 +04:00
return tevent_req_post ( req , ev ) ;
}
2011-09-05 20:22:57 +04:00
if ( strlen ( mask ) = = 0 ) {
TALLOC_FREE ( dyn ) ;
dyn_len = 0 ;
}
2011-05-05 20:12:07 +04:00
fixed = state - > fixed ;
SSVAL ( fixed , 0 , 33 ) ;
SCVAL ( fixed , 2 , level ) ;
SCVAL ( fixed , 3 , flags ) ;
SIVAL ( fixed , 4 , file_index ) ;
SBVAL ( fixed , 8 , fid_persistent ) ;
SBVAL ( fixed , 16 , fid_volatile ) ;
SSVAL ( fixed , 24 , SMB2_HDR_BODY + 32 ) ;
SSVAL ( fixed , 26 , dyn_len ) ;
2017-01-30 08:49:58 +03:00
SIVAL ( fixed , 28 , outbuf_len ) ;
2011-05-05 20:12:07 +04:00
2011-09-05 20:22:57 +04:00
if ( dyn_len = = 0 ) {
dyn = state - > dyn_pad ;
dyn_len = sizeof ( state - > dyn_pad ) ;
}
2015-03-24 17:16:26 +03:00
subreq = smb2cli_req_send ( state , ev , conn , SMB2_OP_QUERY_DIRECTORY ,
2011-08-12 19:26:13 +04:00
0 , 0 , /* flags */
2012-05-10 20:37:43 +04:00
timeout_msec ,
2012-07-25 12:36:27 +04:00
tcon ,
2012-05-10 20:37:43 +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 ,
outbuf_len ) ; /* 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_query_directory_done , req ) ;
return req ;
}
static void smb2cli_query_directory_done ( struct tevent_req * subreq )
{
struct tevent_req * req =
tevent_req_callback_data ( subreq ,
struct tevent_req ) ;
struct smb2cli_query_directory_state * state =
tevent_req_data ( req ,
struct smb2cli_query_directory_state ) ;
NTSTATUS status ;
struct iovec * iov ;
uint16_t data_offset ;
2011-08-31 02:40:06 +04:00
static const struct smb2cli_req_expected_response expected [ ] = {
{
. status = NT_STATUS_OK ,
. body_size = 0x09
}
} ;
2011-05-05 20:12:07 +04:00
2011-08-31 02:40:06 +04:00
status = smb2cli_req_recv ( subreq , state , & iov ,
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 ;
}
data_offset = SVAL ( iov [ 1 ] . iov_base , 2 ) ;
state - > data_length = IVAL ( iov [ 1 ] . iov_base , 4 ) ;
if ( ( data_offset ! = SMB2_HDR_BODY + 8 ) | |
( state - > data_length > iov [ 2 ] . iov_len ) ) {
tevent_req_nterror ( req , NT_STATUS_INVALID_NETWORK_RESPONSE ) ;
return ;
}
2011-07-08 19:57:56 +04:00
state - > recv_iov = iov ;
2011-05-05 20:12:07 +04:00
state - > data = ( uint8_t * ) iov [ 2 ] . iov_base ;
tevent_req_done ( req ) ;
}
NTSTATUS smb2cli_query_directory_recv ( struct tevent_req * req ,
TALLOC_CTX * mem_ctx ,
uint8_t * * data ,
uint32_t * data_length )
{
struct smb2cli_query_directory_state * state =
tevent_req_data ( req ,
struct smb2cli_query_directory_state ) ;
NTSTATUS status ;
if ( tevent_req_is_nterror ( req , & status ) ) {
return status ;
}
2011-07-08 19:57:56 +04:00
talloc_steal ( mem_ctx , state - > recv_iov ) ;
2011-05-05 20:12:07 +04:00
* data_length = state - > data_length ;
* data = state - > data ;
return NT_STATUS_OK ;
}
2012-05-10 20:37:43 +04:00
NTSTATUS smb2cli_query_directory ( 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
uint8_t level ,
uint8_t flags ,
uint32_t file_index ,
uint64_t fid_persistent ,
uint64_t fid_volatile ,
const char * mask ,
uint32_t outbuf_len ,
TALLOC_CTX * mem_ctx ,
uint8_t * * data ,
uint32_t * data_length )
{
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:37:43 +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-05-10 20:37:43 +04:00
req = smb2cli_query_directory_send ( frame , ev , conn , timeout_msec ,
2012-07-24 00:32:49 +04:00
session , tcon ,
level , flags ,
2011-05-05 20:12:07 +04:00
file_index , fid_persistent ,
fid_volatile , mask , outbuf_len ) ;
if ( req = = NULL ) {
goto fail ;
}
if ( ! tevent_req_poll_ntstatus ( req , ev , & status ) ) {
goto fail ;
}
status = smb2cli_query_directory_recv ( req , mem_ctx ,
data , data_length ) ;
fail :
TALLOC_FREE ( frame ) ;
return status ;
}