diff options
author | Andreas Schneider <asn@samba.org> | 2011-01-31 16:39:34 +0100 |
---|---|---|
committer | Günther Deschner <gd@samba.org> | 2011-02-07 12:32:00 +0100 |
commit | 56963d1f44d137581627eec672b63993585ebf64 (patch) | |
tree | 7ae66f767b557c4a797a971d1c0565cec05e2db4 /source3/rpc_client/cli_winreg.c | |
parent | 69e6db1a5354c09f038ca158e08f4426863dab34 (diff) | |
download | samba-56963d1f44d137581627eec672b63993585ebf64.tar.gz samba-56963d1f44d137581627eec672b63993585ebf64.tar.bz2 samba-56963d1f44d137581627eec672b63993585ebf64.zip |
s3-rpc_client: Added a winreg query sz helper.
Signed-off-by: Günther Deschner <gd@samba.org>
Diffstat (limited to 'source3/rpc_client/cli_winreg.c')
-rw-r--r-- | source3/rpc_client/cli_winreg.c | 75 |
1 files changed, 75 insertions, 0 deletions
diff --git a/source3/rpc_client/cli_winreg.c b/source3/rpc_client/cli_winreg.c index 081f951b2d..be58c1f1f4 100644 --- a/source3/rpc_client/cli_winreg.c +++ b/source3/rpc_client/cli_winreg.c @@ -244,6 +244,81 @@ NTSTATUS dcerpc_winreg_query_multi_sz(TALLOC_CTX *mem_ctx, return status; } +NTSTATUS dcerpc_winreg_query_sz(TALLOC_CTX *mem_ctx, + struct dcerpc_binding_handle *h, + struct policy_handle *key_handle, + const char *value, + const char **data, + WERROR *pwerr) +{ + struct winreg_String wvalue; + enum winreg_Type type; + WERROR result = WERR_OK; + uint32_t value_len = 0; + uint32_t data_size = 0; + NTSTATUS status; + DATA_BLOB blob; + + wvalue.name = value; + + status = dcerpc_winreg_QueryValue(h, + mem_ctx, + key_handle, + &wvalue, + &type, + NULL, + &data_size, + &value_len, + &result); + if (!NT_STATUS_IS_OK(status)) { + return status; + } + if (!W_ERROR_IS_OK(result)) { + *pwerr = result; + return status; + } + + if (type != REG_SZ) { + *pwerr = WERR_INVALID_DATATYPE; + return status; + } + + blob = data_blob_talloc(mem_ctx, NULL, data_size); + if (blob.data == NULL) { + *pwerr = WERR_NOMEM; + return status; + } + value_len = 0; + + status = dcerpc_winreg_QueryValue(h, + mem_ctx, + key_handle, + &wvalue, + &type, + blob.data, + &data_size, + &value_len, + &result); + if (!NT_STATUS_IS_OK(status)) { + return status; + } + if (!W_ERROR_IS_OK(result)) { + *pwerr = result; + return status; + } + + if (data) { + bool ok; + + ok = pull_reg_sz(mem_ctx, &blob, data); + if (!ok) { + *pwerr = WERR_NOMEM; + } + } + + return status; +} + NTSTATUS dcerpc_winreg_set_dword(TALLOC_CTX *mem_ctx, struct dcerpc_binding_handle *h, struct policy_handle *key_handle, |