2003-11-15 08:42:49 +03:00
/*
Unix SMB / CIFS implementation .
test suite for spoolss rpc operations
Copyright ( C ) Tim Potter 2003
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 2 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 , write to the Free Software
Foundation , Inc . , 675 Mass Ave , Cambridge , MA 0213 9 , USA .
*/
# include "includes.h"
2004-11-01 13:30:34 +03:00
# include "librpc/gen_ndr/ndr_spoolss.h"
2003-11-15 08:42:49 +03:00
2003-12-01 02:40:04 +03:00
static BOOL test_GetPrinter ( struct dcerpc_pipe * p , TALLOC_CTX * mem_ctx ,
2003-11-17 07:56:59 +03:00
struct policy_handle * handle )
{
NTSTATUS status ;
struct spoolss_GetPrinter r ;
2004-05-25 21:24:24 +04:00
uint16_t levels [ ] = { 1 , 2 , 3 , 4 , 5 , 6 , 7 } ;
2003-11-17 07:56:59 +03:00
int i ;
BOOL ret = True ;
for ( i = 0 ; i < ARRAY_SIZE ( levels ) ; i + + ) {
2004-05-25 20:24:13 +04:00
uint32_t buf_size = 0 ;
2003-11-17 07:56:59 +03:00
r . in . handle = handle ;
r . in . level = levels [ i ] ;
r . in . buffer = NULL ;
r . in . buf_size = & buf_size ;
r . out . buf_size = & buf_size ;
printf ( " Testing GetPrinter level %u \n " , r . in . level ) ;
status = dcerpc_spoolss_GetPrinter ( p , mem_ctx , & r ) ;
if ( ! NT_STATUS_IS_OK ( status ) ) {
printf ( " GetPrinter failed - %s \n " , nt_errstr ( status ) ) ;
ret = False ;
continue ;
}
if ( W_ERROR_EQUAL ( r . out . result , WERR_INSUFFICIENT_BUFFER ) ) {
2003-11-17 12:34:19 +03:00
DATA_BLOB blob = data_blob_talloc ( mem_ctx , NULL , buf_size ) ;
2003-11-17 07:56:59 +03:00
data_blob_clear ( & blob ) ;
r . in . buffer = & blob ;
status = dcerpc_spoolss_GetPrinter ( p , mem_ctx , & r ) ;
}
if ( ! NT_STATUS_IS_OK ( status ) | |
! W_ERROR_IS_OK ( r . out . result ) ) {
printf ( " GetPrinter failed - %s/%s \n " ,
nt_errstr ( status ) , win_errstr ( r . out . result ) ) ;
ret = False ;
continue ;
}
}
return ret ;
}
2003-12-01 02:40:04 +03:00
static BOOL test_ClosePrinter ( struct dcerpc_pipe * p , TALLOC_CTX * mem_ctx ,
2003-11-17 07:56:59 +03:00
struct policy_handle * handle )
{
NTSTATUS status ;
struct spoolss_ClosePrinter r ;
r . in . handle = handle ;
r . out . handle = handle ;
printf ( " Testing ClosePrinter \n " ) ;
status = dcerpc_spoolss_ClosePrinter ( p , mem_ctx , & r ) ;
if ( ! NT_STATUS_IS_OK ( status ) ) {
printf ( " ClosePrinter failed - %s \n " , nt_errstr ( status ) ) ;
return False ;
}
return True ;
}
2003-12-01 02:40:04 +03:00
static BOOL test_GetForm ( struct dcerpc_pipe * p , TALLOC_CTX * mem_ctx ,
2004-10-28 17:40:50 +04:00
struct policy_handle * handle ,
const char * formname )
2003-11-26 09:26:18 +03:00
{
NTSTATUS status ;
struct spoolss_GetForm r ;
2004-05-25 20:24:13 +04:00
uint32_t buf_size ;
2003-11-26 09:26:18 +03:00
r . in . handle = handle ;
r . in . formname = formname ;
r . in . level = 1 ;
r . in . buffer = NULL ;
buf_size = 0 ;
r . in . buf_size = r . out . buf_size = & buf_size ;
printf ( " Testing GetForm \n " ) ;
status = dcerpc_spoolss_GetForm ( p , mem_ctx , & r ) ;
if ( ! NT_STATUS_IS_OK ( status ) ) {
printf ( " GetForm failed - %s \n " , nt_errstr ( status ) ) ;
return False ;
}
if ( W_ERROR_EQUAL ( r . out . result , WERR_INSUFFICIENT_BUFFER ) ) {
DATA_BLOB blob = data_blob_talloc ( mem_ctx , NULL , buf_size ) ;
data_blob_clear ( & blob ) ;
r . in . buffer = & blob ;
status = dcerpc_spoolss_GetForm ( p , mem_ctx , & r ) ;
if ( ! r . out . info ) {
printf ( " No form info returned " ) ;
return False ;
}
}
return True ;
}
2003-12-01 02:40:04 +03:00
static BOOL test_EnumForms ( struct dcerpc_pipe * p , TALLOC_CTX * mem_ctx ,
2003-11-26 09:26:18 +03:00
struct policy_handle * handle )
{
NTSTATUS status ;
struct spoolss_EnumForms r ;
2004-05-25 20:24:13 +04:00
uint32_t buf_size ;
2003-11-26 09:26:18 +03:00
r . in . handle = handle ;
r . in . level = 1 ;
r . in . buffer = NULL ;
buf_size = 0 ;
r . in . buf_size = & buf_size ;
r . out . buf_size = & buf_size ;
printf ( " Testing EnumForms \n " ) ;
status = dcerpc_spoolss_EnumForms ( p , mem_ctx , & r ) ;
if ( ! NT_STATUS_IS_OK ( status ) ) {
printf ( " EnumForms failed - %s \n " , nt_errstr ( status ) ) ;
return False ;
}
if ( W_ERROR_EQUAL ( r . out . result , WERR_INSUFFICIENT_BUFFER ) ) {
DATA_BLOB blob = data_blob_talloc ( mem_ctx , NULL , buf_size ) ;
union spoolss_FormInfo * info ;
int j ;
data_blob_clear ( & blob ) ;
r . in . buffer = & blob ;
status = dcerpc_spoolss_EnumForms ( p , mem_ctx , & r ) ;
2005-02-21 16:54:06 +03:00
if ( ! r . out . info ) {
2003-11-26 09:26:18 +03:00
printf ( " No forms returned " ) ;
return False ;
}
2005-02-21 16:54:06 +03:00
info = * r . out . info ;
2003-11-26 09:26:18 +03:00
2005-02-21 16:54:06 +03:00
for ( j = 0 ; j < r . out . count ; j + + ) {
2004-10-28 12:37:46 +04:00
test_GetForm ( p , mem_ctx , handle , info [ j ] . info1 . formname ) ;
2005-02-21 16:54:06 +03:00
}
2003-11-26 09:26:18 +03:00
}
if ( ! NT_STATUS_IS_OK ( status ) | | ! W_ERROR_IS_OK ( r . out . result ) ) {
printf ( " EnumForms failed - %s/%s \n " ,
nt_errstr ( status ) , win_errstr ( r . out . result ) ) ;
return False ;
}
return True ;
}
2003-12-01 02:40:04 +03:00
static BOOL test_DeleteForm ( struct dcerpc_pipe * p , TALLOC_CTX * mem_ctx ,
2004-10-28 17:40:50 +04:00
struct policy_handle * handle ,
const char * formname )
2003-11-27 12:50:25 +03:00
{
NTSTATUS status ;
struct spoolss_DeleteForm r ;
r . in . handle = handle ;
r . in . formname = formname ;
status = dcerpc_spoolss_DeleteForm ( p , mem_ctx , & r ) ;
if ( ! NT_STATUS_IS_OK ( status ) | | ! W_ERROR_IS_OK ( r . out . result ) ) {
printf ( " DeleteForm failed - %s/%s \n " ,
nt_errstr ( status ) , win_errstr ( r . out . result ) ) ;
return False ;
}
return True ;
}
2003-12-01 02:40:04 +03:00
static BOOL test_AddForm ( struct dcerpc_pipe * p , TALLOC_CTX * mem_ctx ,
2003-11-27 12:50:25 +03:00
struct policy_handle * handle )
{
struct spoolss_AddForm r ;
struct spoolss_AddFormInfo1 form ;
NTSTATUS status ;
2003-11-28 12:05:44 +03:00
const char * formname = " testform3 " ;
2003-11-27 23:55:13 +03:00
BOOL ret = True ;
2003-11-27 12:50:25 +03:00
r . in . handle = handle ;
r . in . level = 1 ;
form . flags = 2 ; /* User form */
2004-10-28 12:37:46 +04:00
form . formname = formname ;
2003-11-27 12:50:25 +03:00
form . width = 1 ;
form . length = 2 ;
form . left = 3 ;
form . top = 4 ;
form . right = 5 ;
form . bottom = 6 ;
r . in . info . info1 = & form ;
status = dcerpc_spoolss_AddForm ( p , mem_ctx , & r ) ;
if ( ! NT_STATUS_IS_OK ( status ) ) {
printf ( " AddForm failed - %s \n " , nt_errstr ( status ) ) ;
return False ;
}
if ( ! W_ERROR_IS_OK ( r . out . result ) ) {
printf ( " AddForm failed - %s \n " , nt_errstr ( status ) ) ;
2003-11-27 23:55:13 +03:00
goto done ;
}
{
struct spoolss_SetForm sf ;
sf . in . handle = handle ;
2004-10-31 02:35:34 +03:00
sf . in . formname = formname ;
2003-11-27 23:55:13 +03:00
sf . in . level = 1 ;
sf . in . info . info1 = & form ;
form . width = 1234 ;
status = dcerpc_spoolss_SetForm ( p , mem_ctx , & sf ) ;
if ( ! NT_STATUS_IS_OK ( status ) | | ! W_ERROR_IS_OK ( r . out . result ) ) {
printf ( " SetForm failed - %s/%s \n " ,
nt_errstr ( status ) , win_errstr ( r . out . result ) ) ;
ret = False ;
/* Fall through to delete */
}
2003-11-27 12:50:25 +03:00
}
2003-11-27 23:55:13 +03:00
done :
2003-11-27 12:50:25 +03:00
if ( ! test_DeleteForm ( p , mem_ctx , handle , formname ) ) {
printf ( " DeleteForm failed \n " ) ;
2003-11-27 23:55:13 +03:00
ret = False ;
2003-11-27 12:50:25 +03:00
}
2003-11-27 23:55:13 +03:00
return ret ;
2003-11-27 12:50:25 +03:00
}
2004-09-03 17:36:58 +04:00
static BOOL test_EnumPorts ( struct dcerpc_pipe * p , TALLOC_CTX * mem_ctx )
{
NTSTATUS status ;
struct spoolss_EnumPorts r ;
uint32_t buf_size ;
r . in . servername = talloc_asprintf ( mem_ctx , " \\ \\ %s " ,
dcerpc_server_name ( p ) ) ;
r . in . level = 2 ;
r . in . buffer = NULL ;
buf_size = 0 ;
r . in . buf_size = & buf_size ;
r . out . buf_size = & buf_size ;
printf ( " Testing EnumPorts \n " ) ;
status = dcerpc_spoolss_EnumPorts ( p , mem_ctx , & r ) ;
if ( ! NT_STATUS_IS_OK ( status ) ) {
printf ( " EnumPorts failed -- %s \n " , nt_errstr ( status ) ) ;
return False ;
}
if ( W_ERROR_EQUAL ( r . out . result , WERR_INSUFFICIENT_BUFFER ) ) {
DATA_BLOB blob = data_blob_talloc ( mem_ctx , NULL , buf_size ) ;
data_blob_clear ( & blob ) ;
r . in . buffer = & blob ;
status = dcerpc_spoolss_EnumPorts ( p , mem_ctx , & r ) ;
if ( ! NT_STATUS_IS_OK ( status ) ) {
printf ( " EnumPorts failed -- %s \n " , nt_errstr ( status ) ) ;
return False ;
}
2005-02-21 16:54:06 +03:00
if ( ! r . out . info ) {
2004-09-03 17:36:58 +04:00
printf ( " No ports returned " ) ;
return False ;
}
}
return True ;
}
2003-12-01 02:40:04 +03:00
static BOOL test_GetJob ( struct dcerpc_pipe * p , TALLOC_CTX * mem_ctx ,
2004-05-25 20:24:13 +04:00
struct policy_handle * handle , uint32_t job_id )
2003-11-28 08:02:32 +03:00
{
NTSTATUS status ;
struct spoolss_GetJob r ;
2004-05-25 20:24:13 +04:00
uint32_t buf_size ;
2003-11-28 08:02:32 +03:00
r . in . handle = handle ;
r . in . job_id = job_id ;
r . in . level = 1 ;
r . in . buffer = NULL ;
buf_size = 0 ;
r . in . buf_size = r . out . buf_size = & buf_size ;
printf ( " Testing GetJob \n " ) ;
status = dcerpc_spoolss_GetJob ( p , mem_ctx , & r ) ;
if ( ! NT_STATUS_IS_OK ( status ) ) {
printf ( " GetJob failed - %s \n " , nt_errstr ( status ) ) ;
return False ;
}
if ( W_ERROR_EQUAL ( r . out . result , WERR_INSUFFICIENT_BUFFER ) ) {
DATA_BLOB blob = data_blob_talloc ( mem_ctx , NULL , buf_size ) ;
data_blob_clear ( & blob ) ;
r . in . buffer = & blob ;
status = dcerpc_spoolss_GetJob ( p , mem_ctx , & r ) ;
if ( ! r . out . info ) {
printf ( " No job info returned " ) ;
return False ;
}
}
return True ;
}
2003-12-01 02:40:04 +03:00
static BOOL test_SetJob ( struct dcerpc_pipe * p , TALLOC_CTX * mem_ctx ,
2004-05-25 20:24:13 +04:00
struct policy_handle * handle , uint32_t job_id , uint32_t command )
2003-11-28 08:02:32 +03:00
{
NTSTATUS status ;
struct spoolss_SetJob r ;
r . in . handle = handle ;
r . in . job_id = job_id ;
r . in . level = 0 ;
r . in . command = command ;
printf ( " Testing SetJob \n " ) ;
status = dcerpc_spoolss_SetJob ( p , mem_ctx , & r ) ;
if ( ! NT_STATUS_IS_OK ( status ) ) {
printf ( " SetJob failed - %s \n " , nt_errstr ( status ) ) ;
return False ;
}
return True ;
}
2003-12-01 02:40:04 +03:00
static BOOL test_EnumJobs ( struct dcerpc_pipe * p , TALLOC_CTX * mem_ctx ,
2003-11-28 08:02:32 +03:00
struct policy_handle * handle )
{
NTSTATUS status ;
struct spoolss_EnumJobs r ;
2004-05-25 20:24:13 +04:00
uint32_t buf_size ;
2003-11-28 08:02:32 +03:00
r . in . handle = handle ;
r . in . firstjob = 0 ;
r . in . numjobs = 0xffffffff ;
r . in . level = 1 ;
r . in . buffer = NULL ;
buf_size = 0 ;
r . in . buf_size = & buf_size ;
r . out . buf_size = & buf_size ;
printf ( " Testing EnumJobs \n " ) ;
status = dcerpc_spoolss_EnumJobs ( p , mem_ctx , & r ) ;
if ( ! NT_STATUS_IS_OK ( status ) ) {
printf ( " EnumJobs failed - %s \n " , nt_errstr ( status ) ) ;
return False ;
}
if ( W_ERROR_EQUAL ( r . out . result , WERR_INSUFFICIENT_BUFFER ) ) {
DATA_BLOB blob = data_blob_talloc ( mem_ctx , NULL , buf_size ) ;
union spoolss_JobInfo * info ;
int j ;
data_blob_clear ( & blob ) ;
r . in . buffer = & blob ;
status = dcerpc_spoolss_EnumJobs ( p , mem_ctx , & r ) ;
2005-02-21 16:54:06 +03:00
if ( ! r . out . info ) {
2003-11-28 08:02:32 +03:00
printf ( " No jobs returned " ) ;
return True ;
}
2005-02-21 16:54:06 +03:00
info = * r . out . info ;
2003-11-28 08:02:32 +03:00
for ( j = 0 ; j < r . out . count ; j + + ) {
test_GetJob ( p , mem_ctx , handle , info [ j ] . info1 . job_id ) ;
2005-02-21 16:54:06 +03:00
test_SetJob ( p , mem_ctx , handle , info [ j ] . info1 . job_id , 1 ) ;
2003-11-28 08:02:32 +03:00
}
} else if ( ! W_ERROR_IS_OK ( r . out . result ) ) {
printf ( " EnumJobs failed - %s \n " , win_errstr ( r . out . result ) ) ;
return False ;
}
return True ;
}
2003-12-01 02:40:04 +03:00
static BOOL test_GetPrinterData ( struct dcerpc_pipe * p , TALLOC_CTX * mem_ctx ,
2004-10-28 17:40:50 +04:00
struct policy_handle * handle ,
const char * value_name )
2003-11-28 09:39:06 +03:00
{
NTSTATUS status ;
struct spoolss_GetPrinterData r ;
2004-05-25 20:24:13 +04:00
uint32_t buf_size ;
2003-11-28 09:39:06 +03:00
r . in . handle = handle ;
r . in . value_name = value_name ;
buf_size = 0 ;
r . in . buf_size = r . out . buf_size = & buf_size ;
printf ( " Testing GetPrinterData \n " ) ;
status = dcerpc_spoolss_GetPrinterData ( p , mem_ctx , & r ) ;
if ( ! NT_STATUS_IS_OK ( status ) ) {
printf ( " GetPrinterData failed - %s \n " , nt_errstr ( status ) ) ;
return False ;
}
if ( W_ERROR_EQUAL ( r . out . result , WERR_MORE_DATA ) ) {
status = dcerpc_spoolss_GetPrinterData ( p , mem_ctx , & r ) ;
if ( ! NT_STATUS_IS_OK ( status ) | | ! W_ERROR_IS_OK ( r . out . result ) ) {
printf ( " GetPrinterData failed - %s/%s \n " ,
nt_errstr ( status ) , win_errstr ( r . out . result ) ) ;
return False ;
}
}
return True ;
}
2003-12-01 02:40:04 +03:00
static BOOL test_GetPrinterDataEx ( struct dcerpc_pipe * p , TALLOC_CTX * mem_ctx ,
2004-10-28 17:40:50 +04:00
struct policy_handle * handle ,
const char * key_name ,
const char * value_name )
2003-11-28 14:50:33 +03:00
{
NTSTATUS status ;
struct spoolss_GetPrinterDataEx r ;
2004-05-25 20:24:13 +04:00
uint32_t buf_size ;
2003-11-28 14:50:33 +03:00
r . in . handle = handle ;
r . in . key_name = key_name ;
r . in . value_name = value_name ;
buf_size = 0 ;
r . in . buf_size = r . out . buf_size = & buf_size ;
printf ( " Testing GetPrinterDataEx \n " ) ;
status = dcerpc_spoolss_GetPrinterDataEx ( p , mem_ctx , & r ) ;
if ( ! NT_STATUS_IS_OK ( status ) ) {
2005-02-23 18:08:41 +03:00
if ( NT_STATUS_EQUAL ( status , NT_STATUS_NET_WRITE_FAULT ) & &
p - > last_fault_code = = DCERPC_FAULT_OP_RNG_ERROR ) {
printf ( " GetPrinterDataEx not supported by server \n " ) ;
return True ;
}
2003-11-28 14:50:33 +03:00
printf ( " GetPrinterDataEx failed - %s \n " , nt_errstr ( status ) ) ;
return False ;
}
if ( W_ERROR_EQUAL ( r . out . result , WERR_MORE_DATA ) ) {
status = dcerpc_spoolss_GetPrinterDataEx ( p , mem_ctx , & r ) ;
if ( ! NT_STATUS_IS_OK ( status ) | | ! W_ERROR_IS_OK ( r . out . result ) ) {
printf ( " GetPrinterDataEx failed - %s/%s \n " ,
nt_errstr ( status ) , win_errstr ( r . out . result ) ) ;
return False ;
}
}
return True ;
}
2003-12-01 02:40:04 +03:00
static BOOL test_EnumPrinterData ( struct dcerpc_pipe * p , TALLOC_CTX * mem_ctx ,
2003-11-18 08:54:14 +03:00
struct policy_handle * handle )
{
NTSTATUS status ;
struct spoolss_EnumPrinterData r ;
r . in . handle = handle ;
r . in . enum_index = 0 ;
do {
2004-05-25 20:24:13 +04:00
uint32_t data_size ;
2003-11-18 08:54:14 +03:00
r . in . value_offered = 0 ;
data_size = 0 ;
r . in . data_size = & data_size ;
r . out . data_size = & data_size ;
printf ( " Testing EnumPrinterData \n " ) ;
status = dcerpc_spoolss_EnumPrinterData ( p , mem_ctx , & r ) ;
if ( ! NT_STATUS_IS_OK ( status ) ) {
printf ( " EnumPrinterData failed - %s \n " , nt_errstr ( status ) ) ;
return False ;
}
r . in . value_offered = r . out . value_needed ;
status = dcerpc_spoolss_EnumPrinterData ( p , mem_ctx , & r ) ;
if ( ! NT_STATUS_IS_OK ( status ) ) {
printf ( " EnumPrinterData failed - %s \n " , nt_errstr ( status ) ) ;
return False ;
}
2003-11-28 09:39:06 +03:00
test_GetPrinterData ( p , mem_ctx , handle , r . out . value_name ) ;
2003-11-28 14:50:33 +03:00
test_GetPrinterDataEx (
p , mem_ctx , handle , " PrinterDriverData " ,
r . out . value_name ) ;
2003-11-18 08:54:14 +03:00
r . in . enum_index + + ;
2003-11-28 09:39:06 +03:00
} while ( W_ERROR_IS_OK ( r . out . result ) ) ;
2003-11-18 08:54:14 +03:00
return True ;
}
2003-12-01 02:40:04 +03:00
static BOOL test_DeletePrinterData ( struct dcerpc_pipe * p , TALLOC_CTX * mem_ctx ,
2004-10-28 17:40:50 +04:00
struct policy_handle * handle ,
const char * value_name )
2003-11-28 13:34:58 +03:00
{
NTSTATUS status ;
struct spoolss_DeletePrinterData r ;
r . in . handle = handle ;
r . in . value_name = value_name ;
printf ( " Testing DeletePrinterData \n " ) ;
status = dcerpc_spoolss_DeletePrinterData ( p , mem_ctx , & r ) ;
if ( ! NT_STATUS_IS_OK ( status ) ) {
printf ( " DeletePrinterData failed - %s \n " , nt_errstr ( status ) ) ;
return False ;
}
return True ;
}
2003-12-01 02:40:04 +03:00
static BOOL test_SetPrinterData ( struct dcerpc_pipe * p , TALLOC_CTX * mem_ctx ,
2003-11-28 13:34:58 +03:00
struct policy_handle * handle )
{
NTSTATUS status ;
struct spoolss_SetPrinterData r ;
2004-10-28 17:40:50 +04:00
const char * value_name = " spottyfoot " ;
2003-11-28 13:34:58 +03:00
r . in . handle = handle ;
r . in . value_name = value_name ;
r . in . type = 0 ;
r . in . buffer = data_blob_talloc ( mem_ctx , " dog " , 4 ) ;
r . in . real_len = 4 ;
printf ( " Testing SetPrinterData \n " ) ;
status = dcerpc_spoolss_SetPrinterData ( p , mem_ctx , & r ) ;
if ( ! NT_STATUS_IS_OK ( status ) ) {
printf ( " SetPrinterData failed - %s \n " , nt_errstr ( status ) ) ;
return False ;
}
if ( ! test_DeletePrinterData ( p , mem_ctx , handle , value_name ) ) {
return False ;
}
return True ;
}
2004-01-20 09:07:09 +03:00
static BOOL test_SecondaryClosePrinter ( struct dcerpc_pipe * p , TALLOC_CTX * mem_ctx ,
struct policy_handle * handle )
{
NTSTATUS status ;
struct dcerpc_pipe * p2 ;
2004-01-20 09:10:15 +03:00
BOOL ret = True ;
2004-01-20 09:07:09 +03:00
/* only makes sense on SMB */
2005-01-09 11:34:05 +03:00
if ( p - > conn - > transport . transport ! = NCACN_NP ) {
2004-01-20 09:07:09 +03:00
return True ;
}
printf ( " testing close on secondary pipe \n " ) ;
2004-06-06 11:14:10 +04:00
status = dcerpc_secondary_connection ( p , & p2 ,
DCERPC_SPOOLSS_NAME ,
DCERPC_SPOOLSS_UUID ,
DCERPC_SPOOLSS_VERSION ) ;
2004-01-20 09:07:09 +03:00
if ( ! NT_STATUS_IS_OK ( status ) ) {
printf ( " Failed to create secondary connection \n " ) ;
return False ;
}
if ( test_ClosePrinter ( p2 , mem_ctx , handle ) ) {
printf ( " ERROR: Allowed close on secondary connection! \n " ) ;
2004-01-20 09:10:15 +03:00
ret = False ;
}
if ( p2 - > last_fault_code ! = DCERPC_FAULT_CONTEXT_MISMATCH ) {
printf ( " Unexpected fault code 0x%x - expected 0x%x \n " ,
p2 - > last_fault_code , DCERPC_FAULT_CONTEXT_MISMATCH ) ;
ret = False ;
2004-01-20 09:07:09 +03:00
}
dcerpc_pipe_close ( p2 ) ;
2004-01-20 09:10:15 +03:00
return ret ;
2004-01-20 09:07:09 +03:00
}
2005-02-22 14:51:18 +03:00
static BOOL test_OpenPrinter_badname ( struct dcerpc_pipe * p , TALLOC_CTX * mem_ctx , const char * name )
{
NTSTATUS status ;
struct spoolss_OpenPrinter op ;
struct spoolss_OpenPrinterEx opEx ;
struct policy_handle handle ;
BOOL ret = True ;
op . in . printername = name ;
op . in . datatype = NULL ;
op . in . devmode_ctr . size = 0 ;
op . in . devmode_ctr . devmode = NULL ;
op . in . access_mask = 0 ;
op . out . handle = & handle ;
printf ( " \n Testing OpenPrinter(%s) with bad name \n " , op . in . printername ) ;
status = dcerpc_spoolss_OpenPrinter ( p , mem_ctx , & op ) ;
if ( ! NT_STATUS_IS_OK ( status ) ) {
printf ( " OpenPrinter failed - %s \n " , nt_errstr ( status ) ) ;
ret = False ;
}
if ( ! W_ERROR_EQUAL ( WERR_INVALID_PRINTER_NAME , op . out . result ) ) {
printf ( " OpenPrinter(%s) unexpected result[%s] should be WERR_INVALID_PRINTER_NAME \n " ,
name , win_errstr ( op . out . result ) ) ;
}
if ( W_ERROR_IS_OK ( op . out . result ) ) {
ret & = test_ClosePrinter ( p , mem_ctx , & handle ) ;
}
opEx . in . printername = name ;
opEx . in . datatype = NULL ;
opEx . in . devmode_ctr . size = 0 ;
opEx . in . devmode_ctr . devmode = NULL ;
opEx . in . access_mask = 0 ;
opEx . in . level = 1 ;
opEx . in . userlevel . level1 = NULL ;
opEx . out . handle = & handle ;
printf ( " \n Testing OpenPrinter(%s) with bad name \n " , opEx . in . printername ) ;
status = dcerpc_spoolss_OpenPrinterEx ( p , mem_ctx , & opEx ) ;
if ( ! NT_STATUS_IS_OK ( status ) ) {
printf ( " OpenPrinter failed - %s \n " , nt_errstr ( status ) ) ;
ret = False ;
}
if ( ! W_ERROR_EQUAL ( WERR_INVALID_PRINTER_NAME , opEx . out . result ) ) {
printf ( " OpenPrinterEx(%s) unexpected result[%s] should be WERR_INVALID_PRINTER_NAME \n " ,
name , win_errstr ( opEx . out . result ) ) ;
}
if ( W_ERROR_IS_OK ( opEx . out . result ) ) {
ret & = test_ClosePrinter ( p , mem_ctx , & handle ) ;
}
return ret ;
}
static BOOL test_OpenPrinter_badnames ( struct dcerpc_pipe * p , TALLOC_CTX * mem_ctx )
{
BOOL ret = True ;
char * name ;
ret & = test_OpenPrinter_badname ( p , mem_ctx , " __INVALID_PRINTER__ " ) ;
ret & = test_OpenPrinter_badname ( p , mem_ctx , " \\ \\ 127.0.0.1 " ) ;
ret & = test_OpenPrinter_badname ( p , mem_ctx , " \\ \\ localhost " ) ;
name = talloc_asprintf ( mem_ctx , " \\ \\ %s \\ " , dcerpc_server_name ( p ) ) ;
ret & = test_OpenPrinter_badname ( p , mem_ctx , name ) ;
talloc_free ( name ) ;
name = talloc_asprintf ( mem_ctx , " \\ \\ %s \\ __INVALID_PRINTER__ " , dcerpc_server_name ( p ) ) ;
ret & = test_OpenPrinter_badname ( p , mem_ctx , name ) ;
talloc_free ( name ) ;
return ret ;
}
2003-11-17 06:38:13 +03:00
static BOOL test_OpenPrinter ( struct dcerpc_pipe * p , TALLOC_CTX * mem_ctx ,
2003-11-17 07:56:59 +03:00
const char * name )
2003-11-17 06:38:13 +03:00
{
NTSTATUS status ;
struct spoolss_OpenPrinter r ;
struct policy_handle handle ;
2003-11-17 07:56:59 +03:00
BOOL ret = True ;
2003-11-17 06:38:13 +03:00
2005-02-22 11:04:52 +03:00
r . in . printername = talloc_asprintf ( mem_ctx , " \\ \\ %s \\ %s " , dcerpc_server_name ( p ) , name ) ;
r . in . datatype = NULL ;
r . in . devmode_ctr . size = 0 ;
r . in . devmode_ctr . devmode = NULL ;
r . in . access_mask = SEC_FLAG_MAXIMUM_ALLOWED ;
r . out . handle = & handle ;
2003-11-17 06:38:13 +03:00
2005-02-22 11:04:52 +03:00
printf ( " \n Testing OpenPrinter(%s) \n " , r . in . printername ) ;
2003-11-17 06:38:13 +03:00
status = dcerpc_spoolss_OpenPrinter ( p , mem_ctx , & r ) ;
if ( ! NT_STATUS_IS_OK ( status ) | | ! W_ERROR_IS_OK ( r . out . result ) ) {
printf ( " OpenPrinter failed - %s/%s \n " ,
nt_errstr ( status ) , win_errstr ( r . out . result ) ) ;
2005-02-22 14:51:18 +03:00
return False ;
2003-11-17 06:38:13 +03:00
}
2003-11-17 07:56:59 +03:00
if ( ! test_GetPrinter ( p , mem_ctx , & handle ) ) {
ret = False ;
}
2004-01-20 09:07:09 +03:00
if ( ! test_SecondaryClosePrinter ( p , mem_ctx , & handle ) ) {
ret = False ;
}
2003-11-17 07:56:59 +03:00
if ( ! test_ClosePrinter ( p , mem_ctx , & handle ) ) {
ret = False ;
}
2005-02-22 11:04:52 +03:00
2003-11-22 14:49:22 +03:00
return ret ;
2003-11-17 07:56:59 +03:00
}
2003-11-30 14:57:40 +03:00
static BOOL call_OpenPrinterEx ( struct dcerpc_pipe * p , TALLOC_CTX * mem_ctx ,
const char * name , struct policy_handle * handle )
2003-11-17 07:56:59 +03:00
{
struct spoolss_OpenPrinterEx r ;
struct spoolss_UserLevel1 userlevel1 ;
NTSTATUS status ;
2005-02-22 14:51:18 +03:00
if ( name & & name [ 0 ] ) {
2003-11-30 14:57:40 +03:00
r . in . printername = talloc_asprintf ( mem_ctx , " \\ \\ %s \\ %s " ,
dcerpc_server_name ( p ) , name ) ;
2005-02-22 14:51:18 +03:00
} else {
2003-11-30 14:57:40 +03:00
r . in . printername = talloc_asprintf ( mem_ctx , " \\ \\ %s " ,
dcerpc_server_name ( p ) ) ;
2005-02-22 14:51:18 +03:00
}
2003-11-30 14:57:40 +03:00
2005-02-22 14:51:18 +03:00
r . in . datatype = NULL ;
r . in . devmode_ctr . size = 0 ;
r . in . devmode_ctr . devmode = NULL ;
r . in . access_mask = SEC_FLAG_MAXIMUM_ALLOWED ;
r . in . level = 1 ;
r . in . userlevel . level1 = & userlevel1 ;
2003-11-30 14:57:40 +03:00
r . out . handle = handle ;
2003-11-17 07:56:59 +03:00
userlevel1 . size = 1234 ;
userlevel1 . client = " hello " ;
userlevel1 . user = " spottyfoot! " ;
userlevel1 . build = 1 ;
userlevel1 . major = 2 ;
userlevel1 . minor = 3 ;
userlevel1 . processor = 4 ;
printf ( " Testing OpenPrinterEx(%s) \n " , r . in . printername ) ;
status = dcerpc_spoolss_OpenPrinterEx ( p , mem_ctx , & r ) ;
if ( ! NT_STATUS_IS_OK ( status ) ) {
printf ( " OpenPrinterEx failed - %s \n " , nt_errstr ( status ) ) ;
return False ;
}
2003-11-30 14:57:40 +03:00
2003-11-17 07:56:59 +03:00
if ( ! W_ERROR_IS_OK ( r . out . result ) ) {
printf ( " OpenPrinterEx failed - %s \n " , win_errstr ( r . out . result ) ) ;
return False ;
}
2003-11-30 14:57:40 +03:00
return True ;
}
static BOOL test_OpenPrinterEx ( struct dcerpc_pipe * p , TALLOC_CTX * mem_ctx ,
const char * name )
{
struct policy_handle handle ;
BOOL ret = True ;
if ( ! call_OpenPrinterEx ( p , mem_ctx , name , & handle ) ) {
return False ;
}
2003-11-17 07:56:59 +03:00
if ( ! test_GetPrinter ( p , mem_ctx , & handle ) ) {
ret = False ;
}
2003-11-26 09:26:18 +03:00
if ( ! test_EnumForms ( p , mem_ctx , & handle ) ) {
ret = False ;
}
2003-11-27 12:50:25 +03:00
if ( ! test_AddForm ( p , mem_ctx , & handle ) ) {
ret = False ;
}
2003-11-18 08:54:14 +03:00
if ( ! test_EnumPrinterData ( p , mem_ctx , & handle ) ) {
ret = False ;
}
2003-11-28 08:02:32 +03:00
if ( ! test_EnumJobs ( p , mem_ctx , & handle ) ) {
ret = False ;
}
2003-11-28 13:34:58 +03:00
if ( ! test_SetPrinterData ( p , mem_ctx , & handle ) ) {
ret = False ;
}
2004-01-20 09:07:09 +03:00
if ( ! test_SecondaryClosePrinter ( p , mem_ctx , & handle ) ) {
ret = False ;
}
2003-11-17 07:56:59 +03:00
if ( ! test_ClosePrinter ( p , mem_ctx , & handle ) ) {
ret = False ;
}
2003-11-30 14:57:40 +03:00
2003-11-17 07:56:59 +03:00
return ret ;
2003-11-17 06:38:13 +03:00
}
2003-11-16 07:20:29 +03:00
static BOOL test_EnumPrinters ( struct dcerpc_pipe * p , TALLOC_CTX * mem_ctx )
{
struct spoolss_EnumPrinters r ;
NTSTATUS status ;
2004-05-25 21:24:24 +04:00
uint16_t levels [ ] = { 1 , 2 , 4 , 5 } ;
2003-11-16 16:49:14 +03:00
int i ;
BOOL ret = True ;
2003-11-16 07:20:29 +03:00
2003-11-16 16:49:14 +03:00
for ( i = 0 ; i < ARRAY_SIZE ( levels ) ; i + + ) {
2004-05-25 20:24:13 +04:00
uint32_t buf_size = 0 ;
2003-11-17 07:56:59 +03:00
union spoolss_PrinterInfo * info ;
2003-11-17 05:18:11 +03:00
int j ;
2003-11-16 16:49:14 +03:00
r . in . flags = 0x02 ;
r . in . server = " " ;
r . in . level = levels [ i ] ;
2003-11-17 05:18:11 +03:00
r . in . buffer = NULL ;
r . in . buf_size = & buf_size ;
r . out . buf_size = & buf_size ;
2003-11-16 07:20:29 +03:00
2003-11-16 16:49:14 +03:00
printf ( " \n Testing EnumPrinters level %u \n " , r . in . level ) ;
2003-11-16 07:20:29 +03:00
status = dcerpc_spoolss_EnumPrinters ( p , mem_ctx , & r ) ;
2003-11-17 07:56:59 +03:00
if ( ! NT_STATUS_IS_OK ( status ) ) {
2003-11-16 16:49:14 +03:00
printf ( " EnumPrinters failed - %s \n " , nt_errstr ( status ) ) ;
ret = False ;
continue ;
}
2003-11-17 05:58:10 +03:00
if ( W_ERROR_EQUAL ( r . out . result , WERR_INSUFFICIENT_BUFFER ) ) {
2003-11-17 05:18:11 +03:00
DATA_BLOB blob = data_blob_talloc ( mem_ctx , NULL , buf_size ) ;
data_blob_clear ( & blob ) ;
r . in . buffer = & blob ;
2003-11-16 16:49:14 +03:00
status = dcerpc_spoolss_EnumPrinters ( p , mem_ctx , & r ) ;
}
2003-11-17 05:58:10 +03:00
if ( ! NT_STATUS_IS_OK ( status ) | |
! W_ERROR_IS_OK ( r . out . result ) ) {
printf ( " EnumPrinters failed - %s/%s \n " ,
nt_errstr ( status ) , win_errstr ( r . out . result ) ) ;
continue ;
2003-11-16 16:49:14 +03:00
}
2003-11-16 14:36:59 +03:00
2005-02-21 16:54:06 +03:00
if ( ! r . out . info ) {
2003-11-18 12:19:34 +03:00
printf ( " No printers returned " ) ;
continue ;
}
2005-02-21 16:54:06 +03:00
info = * r . out . info ;
2003-11-17 06:38:13 +03:00
for ( j = 0 ; j < r . out . count ; j + + ) {
2003-11-17 07:56:59 +03:00
if ( r . in . level = = 1 ) {
/* the names appear to be comma-separated name lists? */
char * name = talloc_strdup ( mem_ctx , info [ j ] . info1 . name ) ;
char * comma = strchr ( name , ' , ' ) ;
if ( comma ) * comma = 0 ;
if ( ! test_OpenPrinter ( p , mem_ctx , name ) ) {
ret = False ;
}
if ( ! test_OpenPrinterEx ( p , mem_ctx , name ) ) {
ret = False ;
}
2003-11-17 06:38:13 +03:00
}
}
2003-11-16 07:20:29 +03:00
}
2003-11-16 16:49:14 +03:00
return ret ;
2003-11-16 07:20:29 +03:00
}
2003-11-30 14:57:40 +03:00
static BOOL test_GetPrinterDriver2 ( struct dcerpc_pipe * p , TALLOC_CTX * mem_ctx ,
2004-10-28 17:40:50 +04:00
struct policy_handle * handle ,
const char * driver_name )
2003-11-30 14:57:40 +03:00
{
NTSTATUS status ;
struct spoolss_GetPrinterDriver2 r ;
2004-05-25 20:24:13 +04:00
uint32_t buf_size ;
2003-11-30 14:57:40 +03:00
r . in . handle = handle ;
r . in . architecture = " W32X86 " ;
r . in . level = 1 ;
buf_size = 0 ;
2004-09-03 17:20:31 +04:00
r . in . buffer = NULL ;
2003-11-30 14:57:40 +03:00
r . in . buf_size = r . out . buf_size = & buf_size ;
r . in . client_major_version = 0 ;
r . in . client_minor_version = 0 ;
printf ( " Testing GetPrinterDriver2 \n " ) ;
status = dcerpc_spoolss_GetPrinterDriver2 ( p , mem_ctx , & r ) ;
if ( ! NT_STATUS_IS_OK ( status ) ) {
printf ( " GetPrinterDriver2 failed - %s \n " , nt_errstr ( status ) ) ;
return False ;
}
if ( W_ERROR_EQUAL ( r . out . result , WERR_INSUFFICIENT_BUFFER ) ) {
status = dcerpc_spoolss_GetPrinterDriver2 ( p , mem_ctx , & r ) ;
}
if ( ! NT_STATUS_IS_OK ( status ) | |
! W_ERROR_IS_OK ( r . out . result ) ) {
printf ( " GetPrinterDriver2 failed - %s/%s \n " ,
nt_errstr ( status ) , win_errstr ( r . out . result ) ) ;
return False ;
}
return True ;
}
static BOOL test_EnumPrinterDrivers ( struct dcerpc_pipe * p , TALLOC_CTX * mem_ctx )
{
struct spoolss_EnumPrinterDrivers r ;
NTSTATUS status ;
2004-05-25 21:24:24 +04:00
uint16_t levels [ ] = { 1 , 2 , 3 } ;
2003-11-30 14:57:40 +03:00
int i ;
BOOL ret = True ;
for ( i = 0 ; i < ARRAY_SIZE ( levels ) ; i + + ) {
2004-05-25 20:24:13 +04:00
uint32_t buf_size ;
2003-11-30 14:57:40 +03:00
union spoolss_DriverInfo * info ;
2004-05-25 20:24:13 +04:00
uint32_t j ;
2003-11-30 14:57:40 +03:00
2005-02-21 16:54:06 +03:00
r . in . server = talloc_asprintf ( mem_ctx , " \\ \\ %s " , dcerpc_server_name ( p ) ) ;
2003-11-30 14:57:40 +03:00
r . in . environment = " Windows NT x86 " ;
r . in . level = levels [ i ] ;
r . in . buffer = NULL ;
buf_size = 0 ;
r . in . buf_size = & buf_size ;
r . out . buf_size = & buf_size ;
printf ( " \n Testing EnumPrinterDrivers level %u \n " , r . in . level ) ;
status = dcerpc_spoolss_EnumPrinterDrivers ( p , mem_ctx , & r ) ;
if ( ! NT_STATUS_IS_OK ( status ) ) {
printf ( " EnumPrinterDrivers failed - %s \n " ,
nt_errstr ( status ) ) ;
ret = False ;
continue ;
}
2005-02-21 16:54:06 +03:00
2003-11-30 14:57:40 +03:00
if ( W_ERROR_EQUAL ( r . out . result , WERR_INSUFFICIENT_BUFFER ) ) {
DATA_BLOB blob = data_blob_talloc (
mem_ctx , NULL , buf_size ) ;
data_blob_clear ( & blob ) ;
r . in . buffer = & blob ;
2005-02-21 16:54:06 +03:00
status = dcerpc_spoolss_EnumPrinterDrivers ( p , mem_ctx , & r ) ;
2003-11-30 14:57:40 +03:00
}
if ( ! NT_STATUS_IS_OK ( status ) | |
! W_ERROR_IS_OK ( r . out . result ) ) {
printf ( " EnumPrinterDrivers failed - %s/%s \n " ,
nt_errstr ( status ) , win_errstr ( r . out . result ) ) ;
2005-02-21 16:54:06 +03:00
break ;
2003-11-30 14:57:40 +03:00
}
2005-02-21 16:54:06 +03:00
if ( ! r . out . info ) {
2003-11-30 14:57:40 +03:00
printf ( " No printer drivers returned " ) ;
2005-02-21 16:54:06 +03:00
break ;
2003-11-30 14:57:40 +03:00
}
2005-02-21 16:54:06 +03:00
info = * r . out . info ;
2003-11-30 14:57:40 +03:00
2005-02-21 16:54:06 +03:00
if ( r . in . level ! = 1 ) continue ;
2003-11-30 14:57:40 +03:00
for ( j = 0 ; j < r . out . count ; j + + ) {
2005-02-21 16:54:06 +03:00
struct policy_handle handle ;
2003-11-30 14:57:40 +03:00
2005-02-21 16:54:06 +03:00
if ( ! call_OpenPrinterEx ( p , mem_ctx , " " , & handle ) ) {
continue ;
2003-11-30 14:57:40 +03:00
}
2005-02-21 16:54:06 +03:00
ret & = test_GetPrinterDriver2 ( p , mem_ctx , & handle , info [ j ] . info1 . driver_name ) ;
2003-11-30 14:57:40 +03:00
}
}
2005-02-21 16:54:06 +03:00
2003-11-30 14:57:40 +03:00
return ret ;
}
2004-10-28 17:40:50 +04:00
BOOL torture_rpc_spoolss ( void )
2003-11-15 08:42:49 +03:00
{
NTSTATUS status ;
struct dcerpc_pipe * p ;
TALLOC_CTX * mem_ctx ;
BOOL ret = True ;
2003-11-18 08:01:10 +03:00
status = torture_rpc_connection ( & p ,
DCERPC_SPOOLSS_NAME ,
DCERPC_SPOOLSS_UUID ,
DCERPC_SPOOLSS_VERSION ) ;
2003-11-15 08:42:49 +03:00
if ( ! NT_STATUS_IS_OK ( status ) ) {
return False ;
}
2003-11-18 04:18:24 +03:00
2005-02-21 16:54:06 +03:00
mem_ctx = talloc_init ( " torture_rpc_spoolss " ) ;
2005-02-22 14:51:18 +03:00
ret & = test_OpenPrinter_badnames ( p , mem_ctx ) ;
2004-09-03 17:36:58 +04:00
2005-02-22 14:51:18 +03:00
ret & = test_EnumPorts ( p , mem_ctx ) ;
ret & = test_EnumPrinters ( p , mem_ctx ) ;
ret & = test_EnumPrinterDrivers ( p , mem_ctx ) ;
2003-11-16 07:20:29 +03:00
2005-01-27 10:08:20 +03:00
talloc_free ( mem_ctx ) ;
2003-11-22 11:11:32 +03:00
2003-11-15 08:42:49 +03:00
torture_rpc_close ( p ) ;
return ret ;
}