From 375bdfc7f89d8941c006bc2afb058176d81027e3 Mon Sep 17 00:00:00 2001
From: Simo Sorce <idra@samba.org>
Date: Mon, 26 Apr 2010 10:24:46 -0400
Subject: s3-spoolss: Removed construct_dev_mode() function.

Replace it with spoolss_create_default_devmode() or
copy_devicemode() where needed.

Signed-off-by: Jim McDonough <jmcd@samba.org>
---
 source3/include/proto.h             |  2 --
 source3/rpc_server/srv_spoolss_nt.c | 57 +++++++++++++------------------------
 2 files changed, 20 insertions(+), 39 deletions(-)

(limited to 'source3')

diff --git a/source3/include/proto.h b/source3/include/proto.h
index f9d8262973..02bc54c5b0 100644
--- a/source3/include/proto.h
+++ b/source3/include/proto.h
@@ -5139,8 +5139,6 @@ void reset_all_printerdata(struct messaging_context *msg,
 WERROR set_printer_dataex(NT_PRINTER_INFO_LEVEL *printer,
 			  const char *key, const char *value,
 			  uint32_t type, uint8_t *data, int real_len);
-struct spoolss_DeviceMode *construct_dev_mode(TALLOC_CTX *mem_ctx,
-					      const char *servicename);
 bool add_printer_hook(TALLOC_CTX *ctx, NT_USER_TOKEN *token,
 		      struct spoolss_SetPrinterInfo2 *info2);
 
diff --git a/source3/rpc_server/srv_spoolss_nt.c b/source3/rpc_server/srv_spoolss_nt.c
index 27bd5aef95..05bbdd9a88 100644
--- a/source3/rpc_server/srv_spoolss_nt.c
+++ b/source3/rpc_server/srv_spoolss_nt.c
@@ -3455,36 +3455,6 @@ static WERROR construct_printer_info0(TALLOC_CTX *mem_ctx,
 }
 
 
-/****************************************************************************
- Create a spoolss_DeviceMode struct. Returns talloced memory.
-****************************************************************************/
-
-struct spoolss_DeviceMode *construct_dev_mode(TALLOC_CTX *mem_ctx,
-					      const char *servicename)
-{
-	NT_PRINTER_INFO_LEVEL 	*printer = NULL;
-	struct spoolss_DeviceMode *devmode = NULL;
-
-	DEBUG(7,("construct_dev_mode\n"));
-
-	DEBUGADD(8,("getting printer characteristics\n"));
-
-	if (!W_ERROR_IS_OK(get_a_printer(NULL, &printer, 2, servicename)))
-		return NULL;
-
-	if (!printer->info_2->devmode) {
-		DEBUG(5, ("BONG! There was no device mode!\n"));
-		goto done;
-	}
-
-        devmode = talloc_steal(mem_ctx, printer->info_2->devmode);
-
-done:
-	free_a_printer(&printer,2);
-
-	return devmode;
-}
-
 /********************************************************************
  * construct_printer_info1
  * fill a spoolss_PrinterInfo1 struct
@@ -6062,6 +6032,7 @@ static WERROR enumjobs_level2(TALLOC_CTX *mem_ctx,
 			      union spoolss_JobInfo **info_p,
 			      uint32_t *count)
 {
+	struct spoolss_DeviceMode *devmode;
 	union spoolss_JobInfo *info;
 	int i;
 	WERROR result = WERR_OK;
@@ -6073,11 +6044,17 @@ static WERROR enumjobs_level2(TALLOC_CTX *mem_ctx,
 
 	for (i=0; i<*count; i++) {
 
-		struct spoolss_DeviceMode *devmode;
-
-		devmode = construct_dev_mode(info, lp_const_servicename(snum));
-		if (!devmode) {
-			result = WERR_NOMEM;
+		if (!pinfo2->devmode) {
+			result = spoolss_create_default_devmode(info,
+							pinfo2->printername,
+							&devmode);
+		} else {
+			result = copy_devicemode(info,
+						 pinfo2->devmode,
+						 &devmode);
+		}
+		if (!W_ERROR_IS_OK(result)) {
+			DEBUG(3, ("Can't proceed w/o a devmode!"));
 			goto out;
 		}
 
@@ -8107,6 +8084,7 @@ static WERROR getjob_level_2(TALLOC_CTX *mem_ctx,
 	int i = 0;
 	bool found = false;
 	struct spoolss_DeviceMode *devmode;
+	WERROR result;
 
 	for (i=0; i<count; i++) {
 		if (queue[i].job == (int)jobid) {
@@ -8129,8 +8107,13 @@ static WERROR getjob_level_2(TALLOC_CTX *mem_ctx,
 
 	devmode = print_job_devmode(lp_const_servicename(snum), jobid);
 	if (!devmode) {
-		devmode = construct_dev_mode(mem_ctx, lp_const_servicename(snum));
-		W_ERROR_HAVE_NO_MEMORY(devmode);
+		result = spoolss_create_default_devmode(mem_ctx,
+						pinfo2->printername,
+						&devmode);
+		if (!W_ERROR_IS_OK(result)) {
+			DEBUG(3, ("Can't proceed w/o a devmode!"));
+			return result;
+		}
 	}
 
 	return fill_job_info2(mem_ctx,
-- 
cgit