diff options
| -rw-r--r-- | source3/utils/regedit.h | 57 | ||||
| -rw-r--r-- | source3/utils/regedit_wrap.c | 143 |
2 files changed, 200 insertions, 0 deletions
diff --git a/source3/utils/regedit.h b/source3/utils/regedit.h new file mode 100644 index 0000000000..8ae8aed07b --- /dev/null +++ b/source3/utils/regedit.h @@ -0,0 +1,57 @@ +/* + * Samba Unix/Linux SMB client library + * Registry Editor + * Copyright (C) Christopher Davis 2012 + * + * 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 _REGEDIT_H_ +#define _REGEDIT_H_ + +struct registry_context; +struct security_token; +struct registry_key; + +struct samba3_registry_key { + struct registry_key *key; +}; + +WERROR reg_openhive_wrap(TALLOC_CTX *ctx, const char *hive, + struct samba3_registry_key *key); +WERROR reg_openkey_wrap(TALLOC_CTX *ctx, struct samba3_registry_key *parent, + const char *name, struct samba3_registry_key *key); +WERROR reg_enumvalue_wrap(TALLOC_CTX *ctx, struct samba3_registry_key *key, + uint32 idx, char **name, uint32_t *type, DATA_BLOB *data); +WERROR reg_queryvalue_wrap(TALLOC_CTX *ctx, struct samba3_registry_key *key, + const char *name, uint32_t *type, DATA_BLOB *data); +WERROR reg_enumkey_wrap(TALLOC_CTX *ctx, struct samba3_registry_key *key, + uint32 idx, char **name, NTTIME *last_write_time); +WERROR reg_createkey_wrap(TALLOC_CTX *ctx, struct samba3_registry_key *parent, + const char *subkeypath, + struct samba3_registry_key *pkey); +WERROR reg_deletekey_wrap(struct samba3_registry_key *parent, + const char *path); +WERROR reg_deletevalue_wrap(struct samba3_registry_key *key, const char *name); +WERROR reg_queryinfokey_wrap(struct samba3_registry_key *key, + uint32_t *num_subkeys, uint32_t *max_subkeylen, + uint32_t *max_subkeysize, uint32_t *num_values, + uint32_t *max_valnamelen, + uint32_t *max_valbufsize, uint32_t *secdescsize, + NTTIME *last_changed_time); +WERROR reg_setvalue_wrap(struct samba3_registry_key *key, const char *name, + uint32_t type, const DATA_BLOB data); +WERROR reg_init_wrap(void); + +#endif diff --git a/source3/utils/regedit_wrap.c b/source3/utils/regedit_wrap.c new file mode 100644 index 0000000000..00d72d88f1 --- /dev/null +++ b/source3/utils/regedit_wrap.c @@ -0,0 +1,143 @@ +/* + * Samba Unix/Linux SMB client library + * Registry Editor + * Copyright (C) Christopher Davis 2012 + * + * 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/>. + */ + +/* Wrap s3 registry API calls to avoid conflicts with 'struct registry_key', + etc, in s4 libregistry. */ + +#include "includes.h" +#include "registry.h" +#include "registry/reg_api.h" +#include "registry/reg_init_basic.h" +#include "registry/reg_util_token.h" + +#include "regedit.h" + +WERROR reg_openhive_wrap(TALLOC_CTX *ctx, const char *hive, + struct samba3_registry_key *pkey) +{ + struct security_token *token; + WERROR rv; + + SMB_ASSERT(pkey->key == NULL); + + rv = ntstatus_to_werror(registry_create_admin_token(ctx, &token)); + if (!W_ERROR_IS_OK(rv)) { + return rv; + } + + return reg_openhive(ctx, hive, REG_KEY_READ | REG_KEY_WRITE, token, + &pkey->key); +} + +WERROR reg_openkey_wrap(TALLOC_CTX *ctx, struct samba3_registry_key *parent, + const char *name, struct samba3_registry_key *pkey) +{ + SMB_ASSERT(pkey->key == NULL); + return reg_openkey(ctx, parent->key, name, + REG_KEY_READ | REG_KEY_WRITE, &pkey->key); +} + +WERROR reg_enumvalue_wrap(TALLOC_CTX *ctx, struct samba3_registry_key *key, + uint32 idx, char **name, uint32_t *type, + DATA_BLOB *data) +{ + struct registry_value *val = NULL; + WERROR rv; + + rv = reg_enumvalue(ctx, key->key, idx, name, &val); + + if (val && W_ERROR_IS_OK(rv)) { + *type = (uint32_t)val->type; + *data = val->data; + } + + return rv; +} + +WERROR reg_queryvalue_wrap(TALLOC_CTX *ctx, struct samba3_registry_key *key, + const char *name, uint32_t *type, DATA_BLOB *data) +{ + struct registry_value *val = NULL; + WERROR rv; + + rv = reg_queryvalue(ctx, key->key, name, &val); + + if (val && W_ERROR_IS_OK(rv)) { + *type = (uint32_t)val->type; + *data = val->data; + } + + return rv; +} + +WERROR reg_enumkey_wrap(TALLOC_CTX *ctx, struct samba3_registry_key *key, + uint32 idx, char **name, NTTIME *last_write_time) +{ + return reg_enumkey(ctx, key->key, idx, name, last_write_time); +} + +WERROR reg_createkey_wrap(TALLOC_CTX *ctx, struct samba3_registry_key *parent, + const char *subkeypath, + struct samba3_registry_key *pkey) +{ + enum winreg_CreateAction act; + + SMB_ASSERT(pkey->key == NULL); + return reg_createkey(ctx, parent->key, subkeypath, + REG_KEY_READ | REG_KEY_WRITE, &pkey->key, &act); +} + +WERROR reg_deletekey_wrap(struct samba3_registry_key *parent, const char *path) +{ + return reg_deletekey(parent->key, path); +} + +WERROR reg_deletevalue_wrap(struct samba3_registry_key *key, const char *name) +{ + return reg_deletevalue(key->key, name); +} + +WERROR reg_queryinfokey_wrap(struct samba3_registry_key *key, + uint32_t *num_subkeys, uint32_t *max_subkeylen, + uint32_t *max_subkeysize, uint32_t *num_values, + uint32_t *max_valnamelen, + uint32_t *max_valbufsize, uint32_t *secdescsize, + NTTIME *last_changed_time) +{ + return reg_queryinfokey(key->key, num_subkeys, max_subkeylen, + max_subkeysize, num_values, max_valnamelen, + max_valbufsize, secdescsize, + last_changed_time); +} + +WERROR reg_setvalue_wrap(struct samba3_registry_key *key, const char *name, + uint32_t type, const DATA_BLOB data) +{ + struct registry_value val; + + val.type = type; + val.data = data; + + return reg_setvalue(key->key, name, &val); +} + +WERROR reg_init_wrap(void) +{ + return registry_init_basic(); +} |