diff options
| -rw-r--r-- | source3/Makefile.in | 3 | ||||
| -rw-r--r-- | source3/rpc_client/cli_winreg.c | 101 | ||||
| -rw-r--r-- | source3/rpc_client/cli_winreg.h | 57 |
3 files changed, 160 insertions, 1 deletions
diff --git a/source3/Makefile.in b/source3/Makefile.in index e247c071c7..294faf1447 100644 --- a/source3/Makefile.in +++ b/source3/Makefile.in @@ -347,7 +347,8 @@ LIBCLI_SPOOLSS_OBJ = librpc/gen_ndr/ndr_spoolss_c.o \ LIBCLI_EVENTLOG_OBJ = librpc/gen_ndr/ndr_eventlog_c.o -LIBCLI_WINREG_OBJ = librpc/gen_ndr/ndr_winreg_c.o +LIBCLI_WINREG_OBJ = librpc/gen_ndr/ndr_winreg_c.o \ + rpc_client/cli_winreg.o LIBCLI_WINREG_INT_OBJ = $(LIBCLI_WINREG_OBJ) \ rpc_client/cli_winreg_int.o 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 <asn@samba.org> + * + * 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 <http://www.gnu.org/licenses/>. + */ + +#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: */ diff --git a/source3/rpc_client/cli_winreg.h b/source3/rpc_client/cli_winreg.h new file mode 100644 index 0000000000..35baa85157 --- /dev/null +++ b/source3/rpc_client/cli_winreg.h @@ -0,0 +1,57 @@ +/* + * Unix SMB/CIFS implementation. + * + * WINREG client routines + * + * Copyright (c) 2011 Andreas Schneider <asn@samba.org> + * + * 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 <http://www.gnu.org/licenses/>. + */ + +#ifndef CLI_WINREG_H +#define CLI_WINREG_H + +/** + * @brief Query a key for the specified dword value. + * + * Get the data that is associated with the named value of a specified registry + * open key. This function ensures that the key is a dword and converts it + * corretly. + * + * @param[in] mem_ctx The memory context to use. + * + * @param[in] h The binding handle for the rpc connection. + * + * @param[in] key_handle A handle to a key that MUST have been opened + * previously. + * + * @param[in] value The name of the value to query. + * + * @param[out] data A pointer to store the data of the value. + * + * @param[out] pwerr A pointer to a WERROR to store result of the query. + * + * @return NT_STATUS_OK on success or a corresponding error if + * there was a problem on the connection. + */ +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); + +#endif /* CLI_WINREG_H */ + +/* vim: set ts=8 sw=8 noet cindent syntax=c.doxygen: */ |