From 9c7607d3d384240b2777aafa3ea634383431ec4d Mon Sep 17 00:00:00 2001 From: Andreas Schneider Date: Thu, 27 Jan 2011 15:20:58 +0100 Subject: s3-rpc_client: Added winreg query dword helper. MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Günther Deschner --- source3/rpc_client/cli_winreg.c | 101 ++++++++++++++++++++++++++++++++++++++++ 1 file changed, 101 insertions(+) create mode 100644 source3/rpc_client/cli_winreg.c (limited to 'source3/rpc_client/cli_winreg.c') diff --git a/source3/rpc_client/cli_winreg.c b/source3/rpc_client/cli_winreg.c new file mode 100644 index 0000000000..62dca39976 --- /dev/null +++ b/source3/rpc_client/cli_winreg.c @@ -0,0 +1,101 @@ +/* + * Unix SMB/CIFS implementation. + * + * WINREG client routines + * + * Copyright (c) 2011 Andreas Schneider + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, see . + */ + +#include "includes.h" +#include "../librpc/gen_ndr/ndr_winreg_c.h" +#include "rpc_client/cli_winreg.h" + +NTSTATUS dcerpc_winreg_query_dword(TALLOC_CTX *mem_ctx, + struct dcerpc_binding_handle *h, + struct policy_handle *key_handle, + const char *value, + uint32_t *data, + WERROR *pwerr) +{ + struct winreg_String wvalue; + enum winreg_Type type; + uint32_t value_len = 0; + uint32_t data_size = 0; + WERROR result = WERR_OK; + 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_DWORD) { + *pwerr = WERR_INVALID_DATATYPE; + return status; + } + + if (data_size != 4) { + *pwerr = WERR_INVALID_DATA; + 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) { + *data = IVAL(blob.data, 0); + } + + return status; +} + +/* vim: set ts=8 sw=8 noet cindent syntax=c.doxygen: */ -- cgit