From 5af82b6d09d700906eb8c1e65ecbf2617033435d Mon Sep 17 00:00:00 2001 From: Gerald Carter Date: Fri, 14 Jul 2000 17:01:49 +0000 Subject: cleaned up access to prs_struct variables. Replaced all calls prs_free_data() => prs_mem_free prs_realloc_data => prs_grow --jerry (This used to be commit 3d00e969df1800dadbb91fd8c410e9ce87188551) --- source3/rpc_client/cli_spoolss.c | 563 ++++++++++++++++++++++++++++++++++++++- 1 file changed, 553 insertions(+), 10 deletions(-) (limited to 'source3/rpc_client') diff --git a/source3/rpc_client/cli_spoolss.c b/source3/rpc_client/cli_spoolss.c index 22d0e8c2b5..2bbfb25bc6 100644 --- a/source3/rpc_client/cli_spoolss.c +++ b/source3/rpc_client/cli_spoolss.c @@ -29,6 +29,74 @@ extern int DEBUGLEVEL; +/**************************************************************************** +do a SPOOLSS Enum Printer Drivers +****************************************************************************/ +uint32 spoolss_enum_printerdrivers(const char * srv_name, + const char *environment, + uint32 level, + NEW_BUFFER *buffer, uint32 offered, + uint32 *needed, uint32 *returned) +{ + prs_struct rbuf; + prs_struct buf; + SPOOL_Q_ENUMPRINTERDRIVERS q_o; + SPOOL_R_ENUMPRINTERDRIVERS r_o; + + struct cli_connection *con = NULL; + + if (!cli_connection_init(srv_name, PIPE_SPOOLSS, &con)) + return False; + + prs_init(&buf , 0, 4, False); + prs_init(&rbuf, 0, 4, True ); + + /* create and send a MSRPC command with api SPOOLSS_ENUM_PRINTERS */ + + DEBUG(5,("SPOOLSS Enum Printer Drivers (Server: %s Environment: %s level: %d)\n", + srv_name, environment, level)); + + make_spoolss_q_enumprinterdrivers(&q_o, srv_name, environment, + level, buffer, offered); + + /* turn parameters into data stream */ + if (!spoolss_io_q_enumprinterdrivers("", &q_o, &buf, 0) ) { + prs_mem_free(&rbuf); + prs_mem_free(&buf ); + + cli_connection_unlink(con); + } + + if(!rpc_con_pipe_req(con, SPOOLSS_ENUMPRINTERDRIVERS, &buf, &rbuf)) { + prs_mem_free(&rbuf); + prs_mem_free(&buf ); + + cli_connection_unlink(con); + } + + prs_mem_free(&buf); + ZERO_STRUCT(r_o); + + prs_switch_type(&buffer->prs, UNMARSHALL); + prs_set_offset(&buffer->prs, 0); + r_o.buffer=buffer; + + if(!new_spoolss_io_r_enumprinterdrivers("", &r_o, &rbuf, 0)) { + prs_mem_free(&rbuf); + cli_connection_unlink(con); + } + + *needed=r_o.needed; + *returned=r_o.returned; + + prs_mem_free(&rbuf); + prs_mem_free(&buf ); + + cli_connection_unlink(con); + + return r_o.status; +} + /**************************************************************************** do a SPOOLSS Enum Printers ****************************************************************************/ @@ -57,39 +125,514 @@ uint32 spoolss_enum_printers(uint32 flags, fstring srv_name, uint32 level, /* turn parameters into data stream */ if (!spoolss_io_q_enumprinters("", &q_o, &buf, 0) ) { - prs_free_data(&rbuf); - prs_free_data(&buf ); + prs_mem_free(&rbuf); + prs_mem_free(&buf ); cli_connection_unlink(con); } if(!rpc_con_pipe_req(con, SPOOLSS_ENUMPRINTERS, &buf, &rbuf)) { - prs_free_data(&rbuf); - prs_free_data(&buf ); + prs_mem_free(&rbuf); + prs_mem_free(&buf ); cli_connection_unlink(con); } - prs_free_data(&buf ); + prs_mem_free(&buf ); ZERO_STRUCT(r_o); - buffer->prs.io=UNMARSHALL; - buffer->prs.data_offset=0; + prs_switch_type(&buffer->prs, UNMARSHALL); + prs_set_offset(&buffer->prs, 0); r_o.buffer=buffer; if(!new_spoolss_io_r_enumprinters("", &r_o, &rbuf, 0)) { - prs_free_data(&rbuf); + prs_mem_free(&rbuf); cli_connection_unlink(con); } *needed=r_o.needed; *returned=r_o.returned; - prs_free_data(&rbuf); - prs_free_data(&buf ); + prs_mem_free(&rbuf); + prs_mem_free(&buf ); + + cli_connection_unlink(con); + + return r_o.status; +} + +/**************************************************************************** +do a SPOOLSS Enum Jobs +****************************************************************************/ +uint32 spoolss_enum_jobs(const POLICY_HND *hnd, uint32 firstjob, uint32 numofjobs, + uint32 level, NEW_BUFFER *buffer, uint32 offered, + uint32 *needed, uint32 *returned) +{ + prs_struct rbuf; + prs_struct buf; + SPOOL_Q_ENUMJOBS q_o; + SPOOL_R_ENUMJOBS r_o; + + if (hnd == NULL) + return NT_STATUS_INVALID_PARAMETER; + + prs_init(&buf , 0, 4, False); + prs_init(&rbuf, 0, 4, True ); + + /* create and send a MSRPC command with api SPOOLSS_ENUMJOBS */ + + DEBUG(5,("SPOOLSS Enum Jobs level: %d)\n", level)); + + make_spoolss_q_enumjobs(&q_o, hnd, firstjob, numofjobs, level, buffer, offered); + + /* turn parameters into data stream */ + if (!spoolss_io_q_enumjobs("", &q_o, &buf, 0)) { + prs_mem_free(&rbuf); + prs_mem_free(&buf ); + } + + if(!rpc_hnd_pipe_req(hnd, SPOOLSS_ENUMJOBS, &buf, &rbuf)) + { + prs_mem_free(&rbuf); + prs_mem_free(&buf ); + } + + ZERO_STRUCT(r_o); + prs_mem_free(&buf ); + + r_o.buffer=buffer; + + if(!spoolss_io_r_enumjobs("", &r_o, &rbuf, 0)) { + prs_mem_free(&rbuf); + } + + *needed=r_o.needed; + *returned=r_o.returned; + + prs_mem_free(&rbuf); + prs_mem_free(&buf ); + + return r_o.status; +} + +/*************************************************************************** +do a SPOOLSS Enum printer datas +****************************************************************************/ +uint32 spoolss_enum_printerdata(const POLICY_HND *hnd, uint32 idx, + uint32 *valuelen, uint16 *value, uint32 *rvaluelen, + uint32 *type, + uint32 *datalen, uint8 *data, uint32 *rdatalen) +{ + prs_struct rbuf; + prs_struct buf; + SPOOL_Q_ENUMPRINTERDATA q_o; + SPOOL_R_ENUMPRINTERDATA r_o; + + if (hnd == NULL) + return NT_STATUS_INVALID_PARAMETER; + + prs_init(&buf , 0, 4, False); + prs_init(&rbuf, 0, 4, True ); + + /* create and send a MSRPC command with api SPOOLSS_ENUMJOBS */ + + DEBUG(5,("SPOOLSS Enum Printer data)\n")); + + make_spoolss_q_enumprinterdata(&q_o, hnd, idx, *valuelen, *datalen); + + /* turn parameters into data stream */ + if (!spoolss_io_q_enumprinterdata("", &q_o, &buf, 0)) { + prs_mem_free(&rbuf); + prs_mem_free(&buf ); + } + + if(!rpc_hnd_pipe_req(hnd, SPOOLSS_ENUMPRINTERDATA, &buf, &rbuf)) { + prs_mem_free(&rbuf); + prs_mem_free(&buf ); + } + + ZERO_STRUCT(r_o); + prs_mem_free(&buf ); + + r_o.data=data; + r_o.value=value; + + if(!spoolss_io_r_enumprinterdata("", &r_o, &rbuf, 0)) { + prs_mem_free(&rbuf); + } + + *valuelen=r_o.valuesize; + *rvaluelen=r_o.realvaluesize; + *type=r_o.type; + *datalen=r_o.datasize; + *rdatalen=r_o.realdatasize; + + prs_mem_free(&rbuf); + prs_mem_free(&buf ); + + return r_o.status; +} + +/**************************************************************************** +do a SPOOLSS Enum printer datas +****************************************************************************/ +uint32 spoolss_getprinter(const POLICY_HND *hnd, uint32 level, + NEW_BUFFER *buffer, uint32 offered, + uint32 *needed) +{ + prs_struct rbuf; + prs_struct buf; + SPOOL_Q_GETPRINTER q_o; + SPOOL_R_GETPRINTER r_o; + + if (hnd == NULL) + return NT_STATUS_INVALID_PARAMETER; + + prs_init(&buf , 0, 4, False); + prs_init(&rbuf, 0, 4, True ); + + /* create and send a MSRPC command with api SPOOLSS_ENUMJOBS */ + + DEBUG(5,("SPOOLSS Enum Printer data)\n")); + + make_spoolss_q_getprinter(&q_o, hnd, level, buffer, offered); + + /* turn parameters into data stream */ + if (!spoolss_io_q_getprinter("", &q_o, &buf, 0)) { + prs_mem_free(&rbuf); + prs_mem_free(&buf ); + } + + if(!rpc_hnd_pipe_req(hnd, SPOOLSS_GETPRINTER, &buf, &rbuf)) { + prs_mem_free(&rbuf); + prs_mem_free(&buf ); + } + + ZERO_STRUCT(r_o); + prs_mem_free(&buf ); + + prs_switch_type(&buffer->prs, UNMARSHALL); + prs_set_offset(&buffer->prs, 0); + r_o.buffer=buffer; + + if(!spoolss_io_r_getprinter("", &r_o, &rbuf, 0)) { + prs_mem_free(&rbuf); + } + + *needed=r_o.needed; + + prs_mem_free(&rbuf); + prs_mem_free(&buf ); + + return r_o.status; +} + +/**************************************************************************** +do a SPOOLSS Enum printer driver +****************************************************************************/ +uint32 spoolss_getprinterdriver(const POLICY_HND *hnd, + const char *environment, uint32 level, + NEW_BUFFER *buffer, uint32 offered, + uint32 *needed) +{ + prs_struct rbuf; + prs_struct buf; + SPOOL_Q_GETPRINTERDRIVER2 q_o; + SPOOL_R_GETPRINTERDRIVER2 r_o; + + if (hnd == NULL) + return NT_STATUS_INVALID_PARAMETER; + + prs_init(&buf , 0, 4, False); + prs_init(&rbuf, 0, 4, True ); + + /* create and send a MSRPC command with api SPOOLSS_ENUMJOBS */ + + DEBUG(5,("SPOOLSS Enum Printer driver)\n")); + + make_spoolss_q_getprinterdriver2(&q_o, hnd, environment, level, 2, 0, buffer, offered); + + /* turn parameters into data stream */ + if (!spoolss_io_q_getprinterdriver2("", &q_o, &buf, 0)) { + prs_mem_free(&rbuf); + prs_mem_free(&buf ); + } + + if(!rpc_hnd_pipe_req(hnd, SPOOLSS_GETPRINTERDRIVER2, &buf, &rbuf)) { + prs_mem_free(&rbuf); + prs_mem_free(&buf ); + } + + ZERO_STRUCT(r_o); + prs_mem_free(&buf ); + + prs_switch_type(&buffer->prs, UNMARSHALL); + prs_set_offset(&buffer->prs, 0); + r_o.buffer=buffer; + + if(!spoolss_io_r_getprinterdriver2("", &r_o, &rbuf, 0)) { + prs_mem_free(&rbuf); + } + + *needed=r_o.needed; + + prs_mem_free(&rbuf); + prs_mem_free(&buf ); + + return r_o.status; +} + + + +/**************************************************************************** +do a SPOOLSS Open Printer Ex +****************************************************************************/ +BOOL spoolss_open_printer_ex( const char *printername, + const char *datatype, uint32 access_required, + const char *station, const char *username, + POLICY_HND *hnd) +{ + prs_struct rbuf; + prs_struct buf; + SPOOL_Q_OPEN_PRINTER_EX q_o; + BOOL valid_pol = False; + fstring srv_name; + char *s; + + struct cli_connection *con = NULL; + + memset(srv_name, 0, sizeof(srv_name)); + fstrcpy(srv_name, printername); + + s = strchr(&srv_name[2], '\\'); + + if (s != NULL) + *s = 0; + + if (!cli_connection_init(srv_name, PIPE_SPOOLSS, &con)) + return False; + + if (hnd == NULL) return False; + + prs_init(&buf , MAX_PDU_FRAG_LEN, 4, MARSHALL); + prs_init(&rbuf, 0, 4, UNMARSHALL); + + /* create and send a MSRPC command with api SPOOLSS_OPENPRINTEREX */ + + DEBUG(5,("SPOOLSS Open Printer Ex\n")); + + make_spoolss_q_open_printer_ex(&q_o, printername, datatype, + access_required, station, username); + + /* turn parameters into data stream */ + if (spoolss_io_q_open_printer_ex("", &q_o, &buf, 0) && + rpc_con_pipe_req(con, SPOOLSS_OPENPRINTEREX, &buf, &rbuf)) + { + SPOOL_R_OPEN_PRINTER_EX r_o; + BOOL p = True; + + spoolss_io_r_open_printer_ex("", &r_o, &rbuf, 0); + + if (prs_offset(&rbuf)!= 0 && r_o.status != 0) + { + /* report error code */ + DEBUG(5,("SPOOLSS_OPENPRINTEREX: %s\n", get_nt_error_msg(r_o.status))); + p = False; + } + + if (p) + { + /* ok, at last: we're happy. return the policy handle */ + *hnd = r_o.handle; + valid_pol = True; + +#if 0 /* JERRY */ + /* *hnd should be valid at this point */ + valid_pol = register_policy_hnd(get_global_hnd_cache(), + cli_con_sec_ctx(con), + hnd, access_required) && + set_policy_con(get_global_hnd_cache(), + hnd, con, + cli_connection_unlink); +#endif /* JERRY */ + } + } + + prs_mem_free(&rbuf); + prs_mem_free(&buf ); + + return valid_pol; +} + +/**************************************************************************** +do a SPOOL Close +****************************************************************************/ +BOOL spoolss_closeprinter(POLICY_HND *hnd) +{ + prs_struct rbuf; + prs_struct buf; + SPOOL_Q_CLOSEPRINTER q_c; + BOOL valid_close = False; + + if (hnd == NULL) return False; + + /* create and send a MSRPC command with api SPOOLSS_CLOSEPRINTER */ + + prs_init(&buf , MAX_PDU_FRAG_LEN, 4, MARSHALL); + prs_init(&rbuf, 0, 4, UNMARSHALL); + + DEBUG(4,("SPOOL Close Printer\n")); + + /* store the parameters */ + make_spoolss_q_closeprinter(&q_c, hnd); + + /* turn parameters into data stream */ + if (spoolss_io_q_closeprinter("", &q_c, &buf, 0) && + rpc_hnd_pipe_req(hnd, SPOOLSS_CLOSEPRINTER, &buf, &rbuf)) + { + SPOOL_R_CLOSEPRINTER r_c; + + spoolss_io_r_closeprinter("", &r_c, &rbuf, 0); + + if (prs_offset(&rbuf)!=0 && r_c.status != 0) + { + /* report error code */ + DEBUG(0,("SPOOL_CLOSEPRINTER: %s\n", get_nt_error_msg(r_c.status))); + } + } + + prs_mem_free(&rbuf); + prs_mem_free(&buf ); + + if ( hnd != NULL ) + { + free (hnd); + hnd = NULL; + } + + /* commented out by JERRY - merge from TNG */ + /* close_policy_hnd(get_global_hnd_cache(), hnd); */ + + return valid_close; +} + +/**************************************************************************** +do a SPOOLSS Get printer datas +****************************************************************************/ +uint32 spoolss_getprinterdata(const POLICY_HND *hnd, const UNISTR2 *valuename, + uint32 in_size, + uint32 *type, + uint32 *out_size, + uint8 *data, + uint32 *needed) +{ + prs_struct rbuf; + prs_struct buf; + SPOOL_Q_GETPRINTERDATA q_o; + SPOOL_R_GETPRINTERDATA r_o; + + if (hnd == NULL) + return NT_STATUS_INVALID_PARAMETER; + + prs_init(&buf , 0, 4, False); + prs_init(&rbuf, 0, 4, True ); + + /* create and send a MSRPC command with api SPOOLSS_GETPRINTERDATA */ + + DEBUG(5,("SPOOLSS Get Printer data)\n")); + + make_spoolss_q_getprinterdata(&q_o, hnd, valuename, in_size); + + /* turn parameters into data stream */ + if (!spoolss_io_q_getprinterdata("", &q_o, &buf, 0)) { + prs_mem_free(&rbuf); + prs_mem_free(&buf ); + } + + if(!rpc_hnd_pipe_req(hnd, SPOOLSS_GETPRINTERDATA, &buf, &rbuf)) { + prs_mem_free(&rbuf); + prs_mem_free(&buf ); + } + + ZERO_STRUCT(r_o); + prs_mem_free(&buf ); + + r_o.data=data; + + if(!spoolss_io_r_getprinterdata("", &r_o, &rbuf, 0)) { + prs_mem_free(&rbuf); + } + + *type=r_o.type; + *out_size=r_o.size; + *needed=r_o.needed; + + prs_mem_free(&rbuf); + prs_mem_free(&buf ); + + return r_o.status; +} + +/**************************************************************************** +do a SPOOLSS Get Printer Driver Direcotry +****************************************************************************/ +uint32 spoolss_getprinterdriverdir(fstring srv_name, fstring env_name, uint32 level, + NEW_BUFFER *buffer, uint32 offered, + uint32 *needed) +{ + prs_struct rbuf; + prs_struct buf; + SPOOL_Q_GETPRINTERDRIVERDIR q_o; + SPOOL_R_GETPRINTERDRIVERDIR r_o; + + struct cli_connection *con = NULL; + + if (!cli_connection_init(srv_name, PIPE_SPOOLSS, &con)) + return False; + + prs_init(&buf , 0, 4, False); + prs_init(&rbuf, 0, 4, True ); + + /* create and send a MSRPC command with api SPOOLSS_ENUM_PRINTERS */ + + DEBUG(5,("SPOOLSS GetPrinterDriverDir (Server: %s Env: %s level: %d)\n", srv_name, env_name, level)); + + make_spoolss_q_getprinterdriverdir(&q_o, srv_name, env_name, level, buffer, offered); + + /* turn parameters into data stream */ + if (!spoolss_io_q_getprinterdriverdir("", &q_o, &buf, 0) ) { + prs_mem_free(&rbuf); + prs_mem_free(&buf ); + + cli_connection_unlink(con); + } + + if(!rpc_con_pipe_req(con, SPOOLSS_GETPRINTERDRIVERDIRECTORY, &buf, &rbuf)) { + prs_mem_free(&rbuf); + prs_mem_free(&buf ); + + cli_connection_unlink(con); + } + + prs_mem_free(&buf ); + ZERO_STRUCT(r_o); + + prs_switch_type(&buffer->prs, UNMARSHALL); + prs_set_offset(&buffer->prs, 0); + r_o.buffer=buffer; + + if(!spoolss_io_r_getprinterdriverdir("", &r_o, &rbuf, 0)) { + prs_mem_free(&rbuf); + cli_connection_unlink(con); + } + *needed=r_o.needed; + + prs_mem_free(&rbuf); + prs_mem_free(&buf ); cli_connection_unlink(con); return r_o.status; } + -- cgit