s3-spoolss: Use tmp_ctx in winreg_get_driver_internal.
authorAndreas Schneider <asn@samba.org>
Mon, 25 Jul 2011 19:14:45 +0000 (21:14 +0200)
committerAndreas Schneider <asn@samba.org>
Wed, 27 Jul 2011 06:50:00 +0000 (08:50 +0200)
source3/rpc_server/spoolss/srv_spoolss_util.c

index d536a39105491fe6ff57be34ab43d268d1621d24..394c8d27a745a88d15fd2f5b1fa26f3f1fe21586 100644 (file)
@@ -395,15 +395,28 @@ WERROR winreg_get_driver_internal(TALLOC_CTX *mem_ctx,
 {
        WERROR result;
        struct dcerpc_binding_handle *b;
+       TALLOC_CTX *tmp_ctx;
 
-       result = winreg_printer_binding_handle(mem_ctx, session_info, msg_ctx, &b);
-       W_ERROR_NOT_OK_RETURN(result);
+       tmp_ctx = talloc_stackframe();
+       if (tmp_ctx == NULL) {
+               return WERR_NOMEM;
+       }
 
-       return winreg_get_driver(mem_ctx, b,
-                                architecture,
-                                driver_name,
-                                driver_version,
-                                _info8);
+       result = winreg_printer_binding_handle(tmp_ctx, session_info, msg_ctx, &b);
+       if (!W_ERROR_IS_OK(result)) {
+               talloc_free(tmp_ctx);
+               return result;
+       }
+
+       result = winreg_get_driver(mem_ctx,
+                                  b,
+                                  architecture,
+                                  driver_name,
+                                  driver_version,
+                                  _info8);
+
+       talloc_free(tmp_ctx);
+       return result;
 }
 
 WERROR winreg_get_driver_list_internal(TALLOC_CTX *mem_ctx,