s3-rpcclient: properly check and initialize buffers in getdriverpackagepath

[vlendec/samba-autobuild/.git] / source3 / rpcclient / cmd_spoolss.c

diff --git a/source3/rpcclient/cmd_spoolss.c b/source3/rpcclient/cmd_spoolss.c
index 4500ec39f9e50351a5cc8ca593d4ff59a7a4ec97..dbcfd17dba22130c594d8c21e104505328d547ff 100644 (file)
--- a/source3/rpcclient/cmd_spoolss.c
+++ b/source3/rpcclient/cmd_spoolss.c
@@ -1630,7 +1630,10 @@ static WERROR cmd_spoolss_getdriverpackagepath(struct rpc_pipe_client *cli,
        }
 
        offered = 1;
-       cab = talloc_array(mem_ctx, char, offered);
+       cab = talloc_zero_array(mem_ctx, char, offered);
+       if (cab == NULL) {
+               return WERR_NOT_ENOUGH_MEMORY;
+       }
        status = dcerpc_spoolss_GetPrinterDriverPackagePath(b, mem_ctx,
                                                            cli->srv_name_slash,
                                                            env,
@@ -1647,7 +1650,9 @@ static WERROR cmd_spoolss_getdriverpackagepath(struct rpc_pipe_client *cli,
        if (W_ERROR_EQUAL(W_ERROR(WIN32_FROM_HRESULT(hresult)), WERR_INSUFFICIENT_BUFFER)) {
                offered = needed;
                cab = talloc_zero_array(mem_ctx, char, offered);
-
+               if (cab == NULL) {
+                       return WERR_NOT_ENOUGH_MEMORY;
+               }
                status = dcerpc_spoolss_GetPrinterDriverPackagePath(b, mem_ctx,
                                                                    cli->srv_name_slash,
                                                                    env,