From d2794b05a6b8be40ebde4a47b9c03008242b1c46 Mon Sep 17 00:00:00 2001 From: Michael Adam Date: Mon, 20 Sep 2010 03:34:49 +0200 Subject: s3:services_db: rewrite svcctl_set_secdesc to use tmp talloc ctx and add a common exit point --- source3/services/services_db.c | 30 +++++++++++++++--------------- 1 file changed, 15 insertions(+), 15 deletions(-) diff --git a/source3/services/services_db.c b/source3/services/services_db.c index f32b305505..cd05b0e586 100644 --- a/source3/services/services_db.c +++ b/source3/services/services_db.c @@ -581,31 +581,32 @@ bool svcctl_set_secdesc( TALLOC_CTX *ctx, const char *name, struct security_desc char *path = NULL; struct registry_value value; NTSTATUS status; - bool ret = False; + bool ret = false; + TALLOC_CTX *mem_ctx = talloc_stackframe(); /* now add the security descriptor */ - if (asprintf(&path, "%s\\%s\\%s", KEY_SERVICES, name, "Security") < 0) { - return false; + path = talloc_asprintf(mem_ctx, "%s\\%s\\%s", KEY_SERVICES, name, + "Security"); + if (path == NULL) { + goto done; } - wresult = reg_open_path(NULL, path, REG_KEY_ALL, token, &key); + wresult = reg_open_path(mem_ctx, path, REG_KEY_ALL, token, &key); if ( !W_ERROR_IS_OK(wresult) ) { DEBUG(0,("svcctl_get_secdesc: key lookup failed! [%s] (%s)\n", path, win_errstr(wresult))); - SAFE_FREE(path); - return False; + goto done; } - SAFE_FREE(path); /* stream the printer security descriptor */ - status = marshall_sec_desc(ctx, sec_desc, &value.data.data, &value.data.length); + status = marshall_sec_desc(mem_ctx, sec_desc, &value.data.data, + &value.data.length); if (!NT_STATUS_IS_OK(status)) { DEBUG(0,("svcctl_set_secdesc: ndr_push_struct_blob() failed!\n")); - TALLOC_FREE( key ); - return False; + goto done; } value.type = REG_BINARY; @@ -614,14 +615,13 @@ bool svcctl_set_secdesc( TALLOC_CTX *ctx, const char *name, struct security_desc if (!W_ERROR_IS_OK(wresult)) { DEBUG(0, ("svcctl_set_secdesc: reg_setvalue failed: %s\n", win_errstr(wresult))); - talloc_free(key); - return false; + goto done; } - /* cleanup */ - - TALLOC_FREE( key); + ret = true; +done: + talloc_free(mem_ctx); return ret; } -- cgit