/* 
   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 ntuser_creds *usr_creds;

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

	if (spoolss_enum_printers( 0x40, srv_name, level, num, ctr) &&
	    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, int argc, char *argv[])
{
	void **ctr = NULL;
	uint32 num = 0;
	uint32 level = 1;

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

	if (msrpc_spoolss_enum_printers(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, int argc, char *argv[])
{
	fstring srv_name;
	fstring station;
	char *printer_name;
	POLICY_HND hnd;

	BOOL res = True;

	if (argc < 2)
	{
		report(out_hnd, "spoolopen <printer name>\n");
		return;
	}

	printer_name = argv[1];

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

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

	if (!strnequal("\\\\", printer_name, 2))
	{
		fstrcat(srv_name, "\\");
		fstrcat(srv_name, printer_name);
		printer_name = srv_name;
	}

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

	res = res ? spoolss_open_printer_ex( printer_name,
	                        0, 0, 0,
	                        station, usr_creds->user_name,
	                        &hnd) : False;

	res = res ? spoolss_closeprinter(&hnd) : False;

	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( const char* printer_name,
				const char* station, const char* user_name, 
				uint32 level,
				uint32 *num,
				void ***ctr,
				JOB_INFO_FN(fn))
{
	POLICY_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, station, user_name));

	res = res ? spoolss_open_printer_ex( printer_name,
	                        0, 0, 0,
	                        station, user_name,
	                        &hnd) : False;

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

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

	res1 = (status == 0x0);

	res = res ? spoolss_closeprinter(&hnd) : False;

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

	return res1;
}

static void spool_job_info_ctr( const char* printer_name,
				const char* station,
				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, int argc, char *argv[])
{
	fstring srv_name;
	fstring station;
	char *printer_name;

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

	if (argc < 2)
	{
		report(out_hnd, "spoolenum <printer name>\n");
		return;
	}

	printer_name = argv[1];

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

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

	if (!strnequal("\\\\", printer_name, 2))
	{
		fstrcat(srv_name, "\\");
		fstrcat(srv_name, printer_name);
		printer_name = srv_name;
	}

	DEBUG(4,("spoolopen - printer: %s station: %s user: %s\n",
		printer_name, station, usr_creds->user_name));

	if (msrpc_spoolss_enum_jobs( printer_name, station,
	                        usr_creds->user_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);
}