samba-mirror/source3/rpcclient/cmd_spoolss.c

/* 
   Unix SMB/Netbios implementation.
   Version 1.9.
   NT Domain Authentication SMB / MSRPC client
   Copyright (C) Andrew Tridgell 1994-1999
   Copyright (C) Luke Kenneth Casson Leighton 1996-1999
   
   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 02139, USA.
*/


#ifdef SYSLOG
#undef SYSLOG
#endif

#include "includes.h"
#include "nterr.h"

extern int DEBUGLEVEL;

#define DEBUG_TESTING

extern FILE* out_hnd;

extern struct cli_state *smb_cli;
extern int smb_tidx;

/****************************************************************************
nt spoolss query
****************************************************************************/
BOOL msrpc_spoolss_enum_printers(struct cli_state *cli,
				const char* srv_name,
				uint32 level,
				uint32 *num,
				void ***ctr,
				PRINT_INFO_FN(fn))
{
	uint16 nt_pipe_fnum;
	BOOL res = True;

	/* open SPOOLSS session. */
	res = cli_nt_session_open(cli, PIPE_SPOOLSS, &nt_pipe_fnum);

	res = res ? spoolss_enum_printers(cli, nt_pipe_fnum, 
	                        0x40, srv_name, level, num, ctr) : False;

	/* close the session */
	cli_nt_session_close(cli, nt_pipe_fnum);

	if (res && fn != NULL)
	{
		fn(srv_name, level, *num, *ctr);
	}

	return res;
}

static void spool_print_info_ctr(const char* srv_name, uint32 level,
				uint32 num, void *const *const ctr)
{
	display_printer_info_ctr(out_hnd, ACTION_HEADER   , level, num, ctr);
	display_printer_info_ctr(out_hnd, ACTION_ENUMERATE, level, num, ctr);
	display_printer_info_ctr(out_hnd, ACTION_FOOTER   , level, num, ctr);
}

/****************************************************************************
nt spoolss query
****************************************************************************/
void cmd_spoolss_enum_printers(struct client_info *info)
{
	void **ctr = NULL;
	uint32 num = 0;
	uint32 level = 1;

	fstring srv_name;
	fstrcpy(srv_name, "\\\\");
	fstrcat(srv_name, smb_cli->desthost);
	strupper(srv_name);

	if (msrpc_spoolss_enum_printers(smb_cli, srv_name, level, &num, &ctr,
	                         spool_print_info_ctr))
	{
		DEBUG(5,("cmd_spoolss_enum_printer: query succeeded\n"));
	}
	else
	{
		report(out_hnd, "FAILED\n");
	}

	free_void_array(num, ctr, free);
}

/****************************************************************************
nt spoolss query
****************************************************************************/
void cmd_spoolss_open_printer_ex(struct client_info *info)
{
	uint16 nt_pipe_fnum;
	fstring srv_name;
	fstring printer_name;
	PRINTER_HND hnd;

	BOOL res = True;

	if (!next_token(NULL, printer_name, NULL, sizeof(printer_name)))
	{
		report(out_hnd, "spoolopen <printer name>\n");
		return;
	}

	fstrcpy(srv_name, "\\\\");
	fstrcat(srv_name, info->myhostname);
	strupper(srv_name);

	DEBUG(4,("spoolopen - printer: %s server: %s user: %s\n",
		printer_name, srv_name, smb_cli->user_name));

	DEBUG(5, ("cmd_spoolss_open_printer_ex: smb_cli->fd:%d\n", smb_cli->fd));

	/* open SPOOLSS session. */
	res = res ? cli_nt_session_open(smb_cli, PIPE_SPOOLSS, &nt_pipe_fnum) : False;

	res = res ? spoolss_open_printer_ex(smb_cli, nt_pipe_fnum, 
	                        printer_name,
	                        0, 0, 0,
	                        srv_name, smb_cli->user_name,
	                        &hnd) : False;

	res = res ? spoolss_closeprinter(smb_cli, nt_pipe_fnum, &hnd) : False;

	/* close the session */
	cli_nt_session_close(smb_cli, nt_pipe_fnum);

	if (res)
	{
		DEBUG(5,("cmd_spoolss_open_printer_ex: query succeeded\n"));
		report(out_hnd, "OK\n");
	}
	else
	{
		DEBUG(5,("cmd_spoolss_open_printer_ex: query failed\n"));
	}
}

/****************************************************************************
nt spoolss query
****************************************************************************/
BOOL msrpc_spoolss_enum_jobs(struct cli_state *cli,
				const char* srv_name,
				const char* user_name,
				const char* printer_name,
				uint32 level,
				uint32 *num,
				void ***ctr,
				JOB_INFO_FN(fn))
{
	uint16 nt_pipe_fnum;
	PRINTER_HND hnd;
	uint32 buf_size = 0x0;
	uint32 status = 0x0;

	BOOL res = True;
	BOOL res1 = True;

	DEBUG(4,("spoolopen - printer: %s server: %s user: %s\n",
		printer_name, srv_name, user_name));

	DEBUG(5, ("cmd_spoolss_open_printer_ex: smb_cli->fd:%d\n", smb_cli->fd));

	/* open SPOOLSS session. */
	res = res ? cli_nt_session_open(smb_cli, PIPE_SPOOLSS, &nt_pipe_fnum) : False;

	res = res ? spoolss_open_printer_ex(smb_cli, nt_pipe_fnum, 
	                        printer_name,
	                        0, 0, 0,
	                        srv_name, user_name,
	                        &hnd) : False;

	if (status == 0x0)
	{
		status = spoolss_enum_jobs(smb_cli, nt_pipe_fnum, 
				&hnd,
	                        0, 1000, level, &buf_size,
	                        num, ctr);
	}

	if (status == ERROR_INSUFFICIENT_BUFFER)
	{
		status = spoolss_enum_jobs(smb_cli, nt_pipe_fnum, 
				&hnd,
	                        0, 1000, level, &buf_size,
	                        num, ctr);
	}

	res1 = (status == 0x0);

	res = res ? spoolss_closeprinter(smb_cli, nt_pipe_fnum, &hnd) : False;

	/* close the session */
	cli_nt_session_close(smb_cli, nt_pipe_fnum);

	if (res1 && fn != NULL)
	{
		fn(srv_name, printer_name, level, *num, *ctr);
	}

	return res1;
}

static void spool_job_info_ctr(const char* srv_name, const char* printer_name,
				uint32 level,
				uint32 num, void *const *const ctr)
{
	display_job_info_ctr(out_hnd, ACTION_HEADER   , level, num, ctr);
	display_job_info_ctr(out_hnd, ACTION_ENUMERATE, level, num, ctr);
	display_job_info_ctr(out_hnd, ACTION_FOOTER   , level, num, ctr);
}

/****************************************************************************
nt spoolss query
****************************************************************************/
void cmd_spoolss_enum_jobs(struct client_info *info)
{
	fstring srv_name;
	fstring printer_name;

	void **ctr = NULL;
	uint32 num = 0;
	uint32 level = 1;

	if (!next_token(NULL, printer_name, NULL, sizeof(printer_name)))
	{
		report(out_hnd, "spoolopen <printer name>\n");
		return;
	}

	fstrcpy(srv_name, "\\\\");
	fstrcat(srv_name, info->myhostname);
	strupper(srv_name);

	DEBUG(4,("spoolopen - printer: %s server: %s user: %s\n",
		printer_name, srv_name, smb_cli->user_name));

	if (msrpc_spoolss_enum_jobs(smb_cli,
				srv_name, smb_cli->user_name, printer_name,
				level, &num, &ctr,
				spool_job_info_ctr))
	{
		DEBUG(5,("cmd_spoolss_enum_jobs: query succeeded\n"));
	}
	else
	{
		report(out_hnd, "FAILED\n");
	}

	free_void_array(num, ctr, free);
}