diff options
Diffstat (limited to 'source4/lib/registry/patchfile.c')
| -rw-r--r-- | source4/lib/registry/patchfile.c | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/source4/lib/registry/patchfile.c b/source4/lib/registry/patchfile.c index 9a4c9a6fb7..2932910ea0 100644 --- a/source4/lib/registry/patchfile.c +++ b/source4/lib/registry/patchfile.c @@ -95,6 +95,11 @@ WERROR reg_generate_diff_key(struct registry_key *oldkey, /* if "error2" is going to be "WERR_BADFILE", then newkey */ /* didn't have such a subkey and therefore add a del diff */ tmppath = talloc_asprintf(mem_ctx, "%s\\%s", path, keyname1); + if (tmppath == NULL) { + DEBUG(0, ("Out of memory\n")); + talloc_free(mem_ctx); + return WERR_NOMEM; + } if (!W_ERROR_IS_OK(error2)) callbacks->del_key(callback_data, tmppath); @@ -156,6 +161,11 @@ WERROR reg_generate_diff_key(struct registry_key *oldkey, /* oldkey didn't have such a subkey, add a add diff */ tmppath = talloc_asprintf(mem_ctx, "%s\\%s", path, keyname1); + if (tmppath == NULL) { + DEBUG(0, ("Out of memory\n")); + talloc_free(mem_ctx); + return WERR_NOMEM; + } callbacks->add_key(callback_data, tmppath); /* perform here also the recursive invocation */ |