Add a talloc failure check

[sfrench/samba-autobuild/.git] / source4 / rpc_server / epmapper / rpc_epmapper.c

diff --git a/source4/rpc_server/epmapper/rpc_epmapper.c b/source4/rpc_server/epmapper/rpc_epmapper.c
index 5491bd242a5cb169347aea4e476f1c8f4aa1d637..b9b02962be79add20218a936e20c45ed9cf628c8 100644 (file)
--- a/source4/rpc_server/epmapper/rpc_epmapper.c
+++ b/source4/rpc_server/epmapper/rpc_epmapper.c
@@ -179,6 +179,9 @@ static error_status_t dcesrv_epm_Map(struct dcesrv_call_state *dce_call, TALLOC_
 
        ZERO_STRUCT(*r->out.entry_handle);
        r->out.num_towers = talloc(mem_ctx, uint32_t);
+       if (!r->out.num_towers) {
+               return EPMAPPER_STATUS_NO_MEMORY;
+       }
        *r->out.num_towers = 1;
        r->out.towers = talloc(mem_ctx, struct epm_twr_p_t);
        if (!r->out.towers) {