/*
* Samba Unix/Linux SMB client library
* Distributed SMB/CIFS Server Management Utility
* Local registry interface
*
* Copyright (C) Michael Adam 2008
*
* 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 "registry.h"
#include "registry/reg_api.h"
#include "registry/reg_util_token.h"
#include "registry/reg_init_basic.h"
#include "utils/net.h"
#include "utils/net_registry_util.h"
#include "include/g_lock.h"
#include "registry/reg_backend_db.h"
#include "registry/reg_import.h"
#include "registry/reg_format.h"
#include "registry/reg_api_util.h"
#include
#include "../libcli/security/display_sec.h"
#include "../libcli/security/sddl.h"
#include "../libcli/registry/util_reg.h"
#include "passdb/machine_sid.h"
#include "net_registry_check.h"
#include "lib/util/util_tdb.h"
#include "lib/util/smb_strtox.h"
/*
*
* Helper functions
*
*/
/**
* split given path into hive and remaining path and open the hive key
*/
static WERROR open_hive(TALLOC_CTX *ctx, const char *path,
uint32_t desired_access,
struct registry_key **hive,
char **subkeyname)
{
WERROR werr;
struct security_token *token = NULL;
char *hivename = NULL;
char *tmp_subkeyname = NULL;
TALLOC_CTX *tmp_ctx = talloc_stackframe();
if ((hive == NULL) || (subkeyname == NULL)) {
werr = WERR_INVALID_PARAMETER;
goto done;
}
werr = split_hive_key(tmp_ctx, path, &hivename, &tmp_subkeyname);
if (!W_ERROR_IS_OK(werr)) {
goto done;
}
*subkeyname = talloc_strdup(ctx, tmp_subkeyname);
if (*subkeyname == NULL) {
werr = WERR_NOT_ENOUGH_MEMORY;
goto done;
}
werr = ntstatus_to_werror(registry_create_admin_token(tmp_ctx, &token));
if (!W_ERROR_IS_OK(werr)) {
goto done;
}
werr = reg_openhive(ctx, hivename, desired_access, token, hive);
if (!W_ERROR_IS_OK(werr)) {
goto done;
}
werr = WERR_OK;
done:
TALLOC_FREE(tmp_ctx);
return werr;
}
static WERROR open_key(TALLOC_CTX *ctx, const char *path,
uint32_t desired_access,
struct registry_key **key)
{
WERROR werr;
char *subkey_name = NULL;
struct registry_key *hive = NULL;
TALLOC_CTX *tmp_ctx = talloc_stackframe();
if ((path == NULL) || (key == NULL)) {
return WERR_INVALID_PARAMETER;
}
werr = open_hive(tmp_ctx, path, desired_access, &hive, &subkey_name);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, _("open_hive failed: %s\n"),
win_errstr(werr));
goto done;
}
werr = reg_openkey(ctx, hive, subkey_name, desired_access, key);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, _("reg_openkey failed: %s\n"),
win_errstr(werr));
goto done;
}
werr = WERR_OK;
done:
TALLOC_FREE(tmp_ctx);
return werr;
}
static WERROR registry_enumkey(struct registry_key *parent, const char *keyname,
bool recursive)
{
WERROR werr;
TALLOC_CTX *ctx = talloc_stackframe();
char *subkey_name;
NTTIME modtime;
uint32_t count;
char *valname = NULL;
struct registry_value *valvalue = NULL;
struct registry_key *key = NULL;
werr = reg_openkey(ctx, parent, keyname, REG_KEY_READ, &key);
if (!W_ERROR_IS_OK(werr)) {
goto done;
}
if (recursive) {
printf("[%s]\n\n", key->key->name);
} else {
for (count = 0;
werr = reg_enumkey(ctx, key, count, &subkey_name, &modtime),
W_ERROR_IS_OK(werr);
count++)
{
print_registry_key(subkey_name, &modtime);
}
if (!W_ERROR_EQUAL(WERR_NO_MORE_ITEMS, werr)) {
goto done;
}
}
for (count = 0;
werr = reg_enumvalue(ctx, key, count, &valname, &valvalue),
W_ERROR_IS_OK(werr);
count++)
{
print_registry_value_with_name(valname, valvalue);
}
if (!W_ERROR_EQUAL(WERR_NO_MORE_ITEMS, werr)) {
goto done;
}
if (!recursive) {
werr = WERR_OK;
goto done;
}
for (count = 0;
werr = reg_enumkey(ctx, key, count, &subkey_name, &modtime),
W_ERROR_IS_OK(werr);
count++)
{
werr = registry_enumkey(key, subkey_name, recursive);
if (!W_ERROR_IS_OK(werr)) {
goto done;
}
}
if (!W_ERROR_EQUAL(WERR_NO_MORE_ITEMS, werr)) {
goto done;
}
werr = WERR_OK;
done:
TALLOC_FREE(ctx);
return werr;
}
/*
*
* the main "net registry" function implementations
*
*/
static int net_registry_enumerate(struct net_context *c, int argc,
const char **argv)
{
WERROR werr;
struct registry_key *key = NULL;
char *name = NULL;
TALLOC_CTX *ctx = talloc_stackframe();
int ret = -1;
if (argc != 1 || c->display_usage) {
d_printf("%s\n%s",
_("Usage:"),
_("net registry enumerate \n"));
d_printf("%s\n%s",
_("Example:"),
_("net registry enumerate 'HKLM\\Software\\Samba'\n"));
goto done;
}
werr = open_hive(ctx, argv[0], REG_KEY_READ, &key, &name);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, _("open_key failed: %s\n"), win_errstr(werr));
goto done;
}
werr = registry_enumkey(key, name, c->opt_reboot);
if (W_ERROR_IS_OK(werr)) {
ret = 0;
}
done:
TALLOC_FREE(ctx);
return ret;
}
static int net_registry_enumerate_recursive(struct net_context *c, int argc,
const char **argv)
{
WERROR werr;
struct registry_key *key = NULL;
char *name = NULL;
TALLOC_CTX *ctx = talloc_stackframe();
int ret = -1;
if (argc != 1 || c->display_usage) {
d_printf("%s\n%s",
_("Usage:"),
_("net registry enumerate \n"));
d_printf("%s\n%s",
_("Example:"),
_("net registry enumerate 'HKLM\\Software\\Samba'\n"));
goto done;
}
werr = open_hive(ctx, argv[0], REG_KEY_READ, &key, &name);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, _("open_key failed: %s\n"), win_errstr(werr));
goto done;
}
werr = registry_enumkey(key, name, true);
if (W_ERROR_IS_OK(werr)) {
ret = 0;
}
done:
TALLOC_FREE(ctx);
return ret;
}
static int net_registry_createkey(struct net_context *c, int argc,
const char **argv)
{
WERROR werr;
enum winreg_CreateAction action;
char *subkeyname = NULL;
struct registry_key *hivekey = NULL;
struct registry_key *subkey = NULL;
TALLOC_CTX *ctx = talloc_stackframe();
int ret = -1;
if (argc != 1 || c->display_usage) {
d_printf("%s\n%s",
_("Usage:"),
_("net registry createkey \n"));
d_printf("%s\n%s",
_("Example:"),
_("net registry createkey "
"'HKLM\\Software\\Samba\\smbconf.127.0.0.1'\n"));
goto done;
}
if (strlen(argv[0]) == 0) {
d_fprintf(stderr, _("error: zero length key name given\n"));
goto done;
}
werr = open_hive(ctx, argv[0], REG_KEY_WRITE, &hivekey, &subkeyname);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, _("open_hive failed: %s\n"),
win_errstr(werr));
goto done;
}
werr = reg_createkey(ctx, hivekey, subkeyname, REG_KEY_WRITE,
&subkey, &action);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, _("reg_createkey failed: %s\n"),
win_errstr(werr));
goto done;
}
switch (action) {
case REG_ACTION_NONE:
d_printf(_("createkey did nothing -- huh?\n"));
break;
case REG_CREATED_NEW_KEY:
d_printf(_("createkey created %s\n"), argv[0]);
break;
case REG_OPENED_EXISTING_KEY:
d_printf(_("createkey opened existing %s\n"), argv[0]);
break;
}
ret = 0;
done:
TALLOC_FREE(ctx);
return ret;
}
static int net_registry_deletekey_internal(struct net_context *c, int argc,
const char **argv,
bool recursive)
{
WERROR werr;
char *subkeyname = NULL;
struct registry_key *hivekey = NULL;
TALLOC_CTX *ctx = talloc_stackframe();
int ret = -1;
if (argc != 1 || c->display_usage) {
d_printf("%s\n%s",
_("Usage:"),
_("net registry deletekey \n"));
d_printf("%s\n%s",
_("Example:"),
_("net registry deletekey "
"'HKLM\\Software\\Samba\\smbconf.127.0.0.1'\n"));
goto done;
}
if (strlen(argv[0]) == 0) {
d_fprintf(stderr, _("error: zero length key name given\n"));
goto done;
}
werr = open_hive(ctx, argv[0], REG_KEY_WRITE, &hivekey, &subkeyname);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, "open_hive %s: %s\n", _("failed"),
win_errstr(werr));
goto done;
}
if (recursive) {
werr = reg_deletekey_recursive(hivekey, subkeyname);
} else {
werr = reg_deletekey(hivekey, subkeyname);
}
if (!W_ERROR_IS_OK(werr) &&
!(c->opt_force && W_ERROR_EQUAL(werr, WERR_FILE_NOT_FOUND)))
{
d_fprintf(stderr, "reg_deletekey %s: %s\n", _("failed"),
win_errstr(werr));
goto done;
}
ret = 0;
done:
TALLOC_FREE(ctx);
return ret;
}
static int net_registry_deletekey(struct net_context *c, int argc,
const char **argv)
{
return net_registry_deletekey_internal(c, argc, argv, false);
}
static int net_registry_deletekey_recursive(struct net_context *c, int argc,
const char **argv)
{
return net_registry_deletekey_internal(c, argc, argv, true);
}
static int net_registry_getvalue_internal(struct net_context *c, int argc,
const char **argv, bool raw)
{
WERROR werr;
int ret = -1;
struct registry_key *key = NULL;
struct registry_value *value = NULL;
TALLOC_CTX *ctx = talloc_stackframe();
if (argc != 2 || c->display_usage) {
d_fprintf(stderr, "%s\n%s",
_("Usage:"),
_("net registry getvalue \n"));
goto done;
}
werr = open_key(ctx, argv[0], REG_KEY_READ, &key);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, _("open_key failed: %s\n"), win_errstr(werr));
goto done;
}
werr = reg_queryvalue(ctx, key, argv[1], &value);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, _("reg_queryvalue failed: %s\n"),
win_errstr(werr));
goto done;
}
print_registry_value(value, raw);
ret = 0;
done:
TALLOC_FREE(ctx);
return ret;
}
static int net_registry_getvalue(struct net_context *c, int argc,
const char **argv)
{
return net_registry_getvalue_internal(c, argc, argv, false);
}
static int net_registry_getvalueraw(struct net_context *c, int argc,
const char **argv)
{
return net_registry_getvalue_internal(c, argc, argv, true);
}
static int net_registry_getvaluesraw(struct net_context *c, int argc,
const char **argv)
{
WERROR werr;
int ret = -1;
struct registry_key *key = NULL;
TALLOC_CTX *ctx = talloc_stackframe();
uint32_t idx;
if (argc != 1 || c->display_usage) {
d_fprintf(stderr, "usage: net rpc registry getvaluesraw "
"\n");
goto done;
}
werr = open_key(ctx, argv[0], REG_KEY_READ, &key);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, "open_key failed: %s\n", win_errstr(werr));
goto done;
}
idx = 0;
while (true) {
struct registry_value *val;
werr = reg_enumvalue(talloc_tos(), key, idx, NULL, &val);
if (W_ERROR_EQUAL(werr, WERR_NO_MORE_ITEMS)) {
ret = 0;
break;
}
if (!W_ERROR_IS_OK(werr)) {
break;
}
print_registry_value(val, true);
TALLOC_FREE(val);
idx += 1;
}
done:
TALLOC_FREE(ctx);
return ret;
}
static int net_registry_setvalue(struct net_context *c, int argc,
const char **argv)
{
WERROR werr;
struct registry_value value;
struct registry_key *key = NULL;
int ret = -1;
TALLOC_CTX *ctx = talloc_stackframe();
if (argc < 4 || c->display_usage) {
d_fprintf(stderr, "%s\n%s",
_("Usage:"),
_("net registry setvalue "
" []+\n"));
goto done;
}
if (!strequal(argv[2], "multi_sz") && (argc != 4)) {
d_fprintf(stderr, _("Too many args for type %s\n"), argv[2]);
goto done;
}
if (strequal(argv[2], "dword")) {
int error = 0;
uint32_t v;
v = smb_strtoul(argv[3], NULL, 10, &error, SMB_STR_STANDARD);
if (error != 0) {
goto done;
}
value.type = REG_DWORD;
value.data = data_blob_talloc(ctx, NULL, 4);
SIVAL(value.data.data, 0, v);
} else if (strequal(argv[2], "sz")) {
value.type = REG_SZ;
if (!push_reg_sz(ctx, &value.data, argv[3])) {
goto done;
}
} else if (strequal(argv[2], "multi_sz")) {
const char **array;
int count = argc - 3;
int i;
value.type = REG_MULTI_SZ;
array = talloc_zero_array(ctx, const char *, count + 1);
if (array == NULL) {
goto done;
}
for (i=0; i < count; i++) {
array[i] = talloc_strdup(array, argv[count+i]);
if (array[i] == NULL) {
goto done;
}
}
if (!push_reg_multi_sz(ctx, &value.data, array)) {
goto done;
}
} else {
d_fprintf(stderr, _("type \"%s\" not implemented\n"), argv[2]);
goto done;
}
werr = open_key(ctx, argv[0], REG_KEY_WRITE, &key);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, _("open_key failed: %s\n"), win_errstr(werr));
goto done;
}
werr = reg_setvalue(key, argv[1], &value);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, _("reg_setvalue failed: %s\n"),
win_errstr(werr));
goto done;
}
ret = 0;
done:
TALLOC_FREE(ctx);
return ret;
}
static int net_registry_increment(struct net_context *c, int argc,
const char **argv)
{
TDB_DATA lock_key = string_term_tdb_data("registry_increment_lock");
struct g_lock_ctx *ctx = NULL;
const char *keyname = NULL;
struct registry_key *key = NULL;
const char *valuename = NULL;
struct registry_value *value = NULL;
uint32_t v;
uint32_t increment;
uint32_t newvalue;
NTSTATUS status;
WERROR werr;
int ret = -1;
if (argc < 2 || c->display_usage) {
d_fprintf(stderr, "%s\n%s",
_("Usage:"),
_("net registry increment "
"[]\n"));
goto done;
}
keyname = argv[0];
valuename = argv[1];
increment = 1;
if (argc == 3) {
int error = 0;
increment = smb_strtoul(
argv[2], NULL, 10, &error, SMB_STR_STANDARD);
if (error != 0) {
goto done;
}
}
ctx = g_lock_ctx_init(c, c->msg_ctx);
if (ctx == NULL) {
d_fprintf(stderr, _("g_lock_ctx_init failed\n"));
goto done;
}
status = g_lock_lock(ctx,
lock_key,
G_LOCK_WRITE,
timeval_set(600, 0),
NULL,
NULL);
if (!NT_STATUS_IS_OK(status)) {
d_fprintf(stderr, _("g_lock_lock failed: %s\n"),
nt_errstr(status));
goto done;
}
werr = open_key(c, keyname, REG_KEY_READ|REG_KEY_WRITE, &key);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, _("open_key failed: %s\n"),
win_errstr(werr));
goto done;
}
werr = reg_queryvalue(key, key, valuename, &value);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, _("reg_queryvalue failed: %s\n"),
win_errstr(werr));
goto done;
}
if (value->type != REG_DWORD) {
d_fprintf(stderr, _("value not a DWORD: %s\n"),
str_regtype(value->type));
goto done;
}
if (value->data.length < 4) {
d_fprintf(stderr, _("value too short for regular DWORD\n"));
goto done;
}
v = IVAL(value->data.data, 0);
v += increment;
newvalue = v;
SIVAL(value->data.data, 0, v);
werr = reg_setvalue(key, valuename, value);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, _("reg_setvalue failed: %s\n"),
win_errstr(werr));
goto done;
}
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, _("increment failed: %s\n"),
win_errstr(werr));
goto done;
}
g_lock_unlock(ctx, lock_key);
d_printf(_("%"PRIu32"\n"), newvalue);
ret = 0;
done:
TALLOC_FREE(value);
TALLOC_FREE(key);
TALLOC_FREE(ctx);
return ret;
}
static int net_registry_deletevalue(struct net_context *c, int argc,
const char **argv)
{
WERROR werr;
struct registry_key *key = NULL;
TALLOC_CTX *ctx = talloc_stackframe();
int ret = -1;
if (argc != 2 || c->display_usage) {
d_fprintf(stderr, "%s\n%s",
_("Usage:"),
_("net registry deletevalue \n"));
goto done;
}
werr = open_key(ctx, argv[0], REG_KEY_WRITE, &key);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, _("open_key failed: %s\n"), win_errstr(werr));
goto done;
}
werr = reg_deletevalue(key, argv[1]);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, _("reg_deletevalue failed: %s\n"),
win_errstr(werr));
goto done;
}
ret = 0;
done:
TALLOC_FREE(ctx);
return ret;
}
static WERROR net_registry_getsd_internal(struct net_context *c,
TALLOC_CTX *mem_ctx,
const char *keyname,
struct security_descriptor **sd)
{
WERROR werr;
struct registry_key *key = NULL;
TALLOC_CTX *ctx = talloc_stackframe();
uint32_t access_mask = REG_KEY_READ |
SEC_FLAG_MAXIMUM_ALLOWED |
SEC_FLAG_SYSTEM_SECURITY;
/*
* net_rpc_regsitry uses SEC_FLAG_SYSTEM_SECURITY, but access
* is denied with these perms right now...
*/
access_mask = REG_KEY_READ;
if (sd == NULL) {
d_fprintf(stderr, _("internal error: invalid argument\n"));
werr = WERR_INVALID_PARAMETER;
goto done;
}
if (strlen(keyname) == 0) {
d_fprintf(stderr, _("error: zero length key name given\n"));
werr = WERR_INVALID_PARAMETER;
goto done;
}
werr = open_key(ctx, keyname, access_mask, &key);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, "%s%s\n", _("open_key failed: "),
win_errstr(werr));
goto done;
}
werr = reg_getkeysecurity(mem_ctx, key, sd);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, "%s%s\n", _("reg_getkeysecurity failed: "),
win_errstr(werr));
goto done;
}
werr = WERR_OK;
done:
TALLOC_FREE(ctx);
return werr;
}
static int net_registry_getsd(struct net_context *c, int argc,
const char **argv)
{
WERROR werr;
int ret = -1;
struct security_descriptor *secdesc = NULL;
TALLOC_CTX *ctx = talloc_stackframe();
if (argc != 1 || c->display_usage) {
d_printf("%s\n%s",
_("Usage:"),
_("net registry getsd \n"));
d_printf("%s\n%s",
_("Example:"),
_("net registry getsd 'HKLM\\Software\\Samba'\n"));
goto done;
}
werr = net_registry_getsd_internal(c, ctx, argv[0], &secdesc);
if (!W_ERROR_IS_OK(werr)) {
goto done;
}
display_sec_desc(secdesc);
ret = 0;
done:
TALLOC_FREE(ctx);
return ret;
}
static int net_registry_getsd_sddl(struct net_context *c,
int argc, const char **argv)
{
WERROR werr;
int ret = -1;
struct security_descriptor *secdesc = NULL;
TALLOC_CTX *ctx = talloc_stackframe();
if (argc != 1 || c->display_usage) {
d_printf("%s\n%s",
_("Usage:"),
_("net registry getsd_sddl \n"));
d_printf("%s\n%s",
_("Example:"),
_("net registry getsd_sddl 'HKLM\\Software\\Samba'\n"));
goto done;
}
werr = net_registry_getsd_internal(c, ctx, argv[0], &secdesc);
if (!W_ERROR_IS_OK(werr)) {
goto done;
}
d_printf("%s\n", sddl_encode(ctx, secdesc, get_global_sam_sid()));
ret = 0;
done:
TALLOC_FREE(ctx);
return ret;
}
static WERROR net_registry_setsd_internal(struct net_context *c,
TALLOC_CTX *mem_ctx,
const char *keyname,
struct security_descriptor *sd)
{
WERROR werr;
struct registry_key *key = NULL;
TALLOC_CTX *ctx = talloc_stackframe();
uint32_t access_mask = REG_KEY_WRITE |
SEC_FLAG_MAXIMUM_ALLOWED |
SEC_FLAG_SYSTEM_SECURITY;
/*
* net_rpc_regsitry uses SEC_FLAG_SYSTEM_SECURITY, but access
* is denied with these perms right now...
*/
access_mask = REG_KEY_WRITE;
if (strlen(keyname) == 0) {
d_fprintf(stderr, _("error: zero length key name given\n"));
werr = WERR_INVALID_PARAMETER;
goto done;
}
werr = open_key(ctx, keyname, access_mask, &key);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, "%s%s\n", _("open_key failed: "),
win_errstr(werr));
goto done;
}
werr = reg_setkeysecurity(key, sd);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, "%s%s\n", _("reg_setkeysecurity failed: "),
win_errstr(werr));
goto done;
}
werr = WERR_OK;
done:
TALLOC_FREE(ctx);
return werr;
}
static int net_registry_setsd_sddl(struct net_context *c,
int argc, const char **argv)
{
WERROR werr;
int ret = -1;
struct security_descriptor *secdesc = NULL;
TALLOC_CTX *ctx = talloc_stackframe();
if (argc != 2 || c->display_usage) {
d_printf("%s\n%s",
_("Usage:"),
_("net registry setsd_sddl \n"));
d_printf("%s\n%s",
_("Example:"),
_("net registry setsd_sddl 'HKLM\\Software\\Samba'\n"));
goto done;
}
secdesc = sddl_decode(ctx, argv[1], get_global_sam_sid());
if (secdesc == NULL) {
goto done;
}
werr = net_registry_setsd_internal(c, ctx, argv[0], secdesc);
if (!W_ERROR_IS_OK(werr)) {
goto done;
}
ret = 0;
done:
TALLOC_FREE(ctx);
return ret;
}
/******************************************************************************/
/**
* @defgroup net_registry net registry
*/
/**
* @defgroup net_registry_import Import
* @ingroup net_registry
* @{
*/
struct import_ctx {
TALLOC_CTX *mem_ctx;
};
static WERROR import_create_key(struct import_ctx *ctx,
struct registry_key *parent,
const char *name, void **pkey, bool *existing)
{
WERROR werr;
TALLOC_CTX *mem_ctx = talloc_new(ctx->mem_ctx);
struct registry_key *key = NULL;
enum winreg_CreateAction action;
if (parent == NULL) {
char *subkeyname = NULL;
werr = open_hive(mem_ctx, name, REG_KEY_WRITE,
&parent, &subkeyname);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, _("open_hive failed: %s\n"),
win_errstr(werr));
goto done;
}
name = subkeyname;
}
action = REG_ACTION_NONE;
werr = reg_createkey(mem_ctx, parent, name, REG_KEY_WRITE,
&key, &action);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, _("reg_createkey failed: %s\n"),
win_errstr(werr));
goto done;
}
if (action == REG_ACTION_NONE) {
d_fprintf(stderr, _("createkey did nothing -- huh?\n"));
werr = WERR_CREATE_FAILED;
goto done;
}
if (existing != NULL) {
*existing = (action == REG_OPENED_EXISTING_KEY);
}
if (pkey!=NULL) {
*pkey = talloc_steal(ctx->mem_ctx, key);
}
done:
talloc_free(mem_ctx);
return werr;
}
static WERROR import_close_key(struct import_ctx *ctx,
struct registry_key *key)
{
return WERR_OK;
}
static WERROR import_delete_key(struct import_ctx *ctx,
struct registry_key *parent, const char *name)
{
WERROR werr;
TALLOC_CTX *mem_ctx = talloc_new(talloc_tos());
if (parent == NULL) {
char *subkeyname = NULL;
werr = open_hive(mem_ctx, name, REG_KEY_WRITE,
&parent, &subkeyname);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, _("open_hive failed: %s\n"),
win_errstr(werr));
goto done;
}
name = subkeyname;
}
werr = reg_deletekey_recursive(parent, name);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, "reg_deletekey_recursive %s: %s\n",
_("failed"), win_errstr(werr));
goto done;
}
done:
talloc_free(mem_ctx);
return werr;
}
static WERROR import_create_val (struct import_ctx *ctx,
struct registry_key *parent, const char *name,
const struct registry_value *value)
{
WERROR werr;
if (parent == NULL) {
return WERR_INVALID_PARAMETER;
}
werr = reg_setvalue(parent, name, value);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, _("reg_setvalue failed: %s\n"),
win_errstr(werr));
}
return werr;
}
static WERROR import_delete_val (struct import_ctx *ctx,
struct registry_key *parent, const char *name)
{
WERROR werr;
if (parent == NULL) {
return WERR_INVALID_PARAMETER;
}
werr = reg_deletevalue(parent, name);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, _("reg_deletevalue failed: %s\n"),
win_errstr(werr));
}
return werr;
}
struct precheck_ctx {
TALLOC_CTX *mem_ctx;
bool failed;
};
static WERROR precheck_create_key(struct precheck_ctx *ctx,
struct registry_key *parent,
const char *name, void **pkey, bool *existing)
{
WERROR werr;
TALLOC_CTX *frame = talloc_stackframe();
struct registry_key *key = NULL;
if (parent == NULL) {
char *subkeyname = NULL;
werr = open_hive(frame, name, REG_KEY_READ,
&parent, &subkeyname);
if (!W_ERROR_IS_OK(werr)) {
d_printf("Precheck: open_hive of [%s] failed: %s\n",
name, win_errstr(werr));
goto done;
}
name = subkeyname;
}
werr = reg_openkey(frame, parent, name, 0, &key);
if (!W_ERROR_IS_OK(werr)) {
d_printf("Precheck: openkey [%s] failed: %s\n",
name, win_errstr(werr));
goto done;
}
if (existing != NULL) {
*existing = true;
}
if (pkey != NULL) {
*pkey = talloc_steal(ctx->mem_ctx, key);
}
done:
talloc_free(frame);
ctx->failed = !W_ERROR_IS_OK(werr);
return werr;
}
static WERROR precheck_close_key(struct precheck_ctx *ctx,
struct registry_key *key)
{
talloc_free(key);
return WERR_OK;
}
static WERROR precheck_delete_key(struct precheck_ctx *ctx,
struct registry_key *parent, const char *name)
{
WERROR werr;
TALLOC_CTX *frame = talloc_stackframe();
struct registry_key *key;
if (parent == NULL) {
char *subkeyname = NULL;
werr = open_hive(frame, name, REG_KEY_READ,
&parent, &subkeyname);
if (!W_ERROR_IS_OK(werr)) {
d_printf("Precheck: open_hive of [%s] failed: %s\n",
name, win_errstr(werr));
goto done;
}
name = subkeyname;
}
werr = reg_openkey(ctx->mem_ctx, parent, name, 0, &key);
if (W_ERROR_IS_OK(werr)) {
d_printf("Precheck: key [%s\\%s] should not exist\n",
parent->key->name, name);
werr = WERR_FILE_EXISTS;
} else if (W_ERROR_EQUAL(werr, WERR_FILE_NOT_FOUND)) {
werr = WERR_OK;
} else {
d_printf("Precheck: openkey [%s\\%s] failed: %s\n",
parent->key->name, name, win_errstr(werr));
}
done:
talloc_free(frame);
ctx->failed = !W_ERROR_IS_OK(werr);
return werr;
}
static int registry_value_cmp(
const struct registry_value* v1, const struct registry_value* v2)
{
if (v1->type == v2->type) {
return data_blob_cmp(&v1->data, &v2->data);
}
return v1->type - v2->type;
}
static WERROR precheck_create_val(struct precheck_ctx *ctx,
struct registry_key *parent,
const char *name,
const struct registry_value *value)
{
TALLOC_CTX *frame = talloc_stackframe();
struct registry_value *old;
WERROR werr;
SMB_ASSERT(parent);
werr = reg_queryvalue(frame, parent, name, &old);
if (!W_ERROR_IS_OK(werr)) {
d_printf("Precheck: queryvalue \"%s\" of [%s] failed: %s\n",
name, parent->key->name, win_errstr(werr));
goto done;
}
if (registry_value_cmp(value, old) != 0) {
d_printf("Precheck: unexpected value \"%s\" of key [%s]\n",
name, parent->key->name);
ctx->failed = true;
}
done:
talloc_free(frame);
return werr;
}
static WERROR precheck_delete_val(struct precheck_ctx *ctx,
struct registry_key *parent,
const char *name)
{
TALLOC_CTX *frame = talloc_stackframe();
struct registry_value *old;
WERROR werr;
SMB_ASSERT(parent);
werr = reg_queryvalue(frame, parent, name, &old);
if (W_ERROR_IS_OK(werr)) {
d_printf("Precheck: value \"%s\" of key [%s] should not exist\n",
name, parent->key->name);
werr = WERR_FILE_EXISTS;
} else if (W_ERROR_EQUAL(werr, WERR_FILE_NOT_FOUND)) {
werr = WERR_OK;
} else {
printf("Precheck: queryvalue \"%s\" of key [%s] failed: %s\n",
name, parent->key->name, win_errstr(werr));
}
talloc_free(frame);
ctx->failed = !W_ERROR_IS_OK(werr);
return werr;
}
static bool import_precheck(const char *fname, const char *parse_options)
{
TALLOC_CTX *mem_ctx = talloc_tos();
struct precheck_ctx precheck_ctx = {
.mem_ctx = mem_ctx,
.failed = false,
};
struct reg_import_callback precheck_callback = {
.openkey = NULL,
.closekey = (reg_import_callback_closekey_t)&precheck_close_key,
.createkey = (reg_import_callback_createkey_t)&precheck_create_key,
.deletekey = (reg_import_callback_deletekey_t)&precheck_delete_key,
.deleteval = (reg_import_callback_deleteval_t)&precheck_delete_val,
.setval = {
.registry_value = (reg_import_callback_setval_registry_value_t)
&precheck_create_val,
},
.setval_type = REGISTRY_VALUE,
.data = &precheck_ctx
};
struct reg_parse_callback *parse_callback;
int ret;
if (!fname) {
return true;
}
parse_callback = reg_import_adapter(mem_ctx, precheck_callback);
if (parse_callback == NULL) {
d_printf("talloc failed\n");
return false;
}
ret = reg_parse_file(fname, parse_callback, parse_options);
if (ret < 0 || precheck_ctx.failed) {
d_printf("Precheck failed\n");
return false;
}
return true;
}
static int import_with_precheck_action(const char *import_fname,
const char *precheck_fname,
const char *parse_options)
{
TALLOC_CTX *frame = talloc_stackframe();
struct import_ctx import_ctx = {
.mem_ctx = frame,
};
struct reg_import_callback import_callback = {
.openkey = NULL,
.closekey = (reg_import_callback_closekey_t)&import_close_key,
.createkey = (reg_import_callback_createkey_t)&import_create_key,
.deletekey = (reg_import_callback_deletekey_t)&import_delete_key,
.deleteval = (reg_import_callback_deleteval_t)&import_delete_val,
.setval = {
.registry_value = (reg_import_callback_setval_registry_value_t)
&import_create_val,
},
.setval_type = REGISTRY_VALUE,
.data = &import_ctx
};
struct reg_parse_callback *parse_callback;
int ret = -1;
bool precheck_ok;
precheck_ok = import_precheck(precheck_fname, parse_options);
if (!precheck_ok) {
goto done;
}
parse_callback = reg_import_adapter(frame, import_callback);
if (parse_callback == NULL) {
d_printf("talloc failed\n");
goto done;
}
ret = reg_parse_file(import_fname, parse_callback, parse_options);
done:
talloc_free(frame);
return ret;
}
static int net_registry_import(struct net_context *c, int argc,
const char **argv)
{
const char *parse_options = (argc > 1) ? argv[1] : NULL;
int ret = -1;
WERROR werr;
if (argc < 1 || argc > 2 || c->display_usage) {
d_printf("%s\n%s",
_("Usage:"),
_("net registry import [options]\n"));
d_printf("%s\n%s",
_("Example:"),
_("net registry import file.reg enc=CP1252\n"));
return -1;
}
werr = regdb_open();
if (!W_ERROR_IS_OK(werr)) {
d_printf("Failed to open regdb: %s\n", win_errstr(werr));
return -1;
}
werr = regdb_transaction_start();
if (!W_ERROR_IS_OK(werr)) {
d_printf("Failed to start transaction on regdb: %s\n",
win_errstr(werr));
goto done;
}
ret = import_with_precheck_action(argv[0], c->opt_precheck,
parse_options);
if (ret < 0) {
d_printf("Transaction canceled!\n");
regdb_transaction_cancel();
goto done;
}
SMB_ASSERT(ret == 0);
if (c->opt_testmode) {
d_printf("Testmode: not committing changes.\n");
regdb_transaction_cancel();
goto done;
}
werr = regdb_transaction_commit();
if (!W_ERROR_IS_OK(werr)) {
d_printf("Failed to commit transaction on regdb: %s\n",
win_errstr(werr));
ret = -1;
}
done:
regdb_close();
return ret;
}
/**@}*/
/******************************************************************************/
/**
* @defgroup net_registry_export Export
* @ingroup net_registry
* @{
*/
static int registry_export(TALLOC_CTX *ctx, /*const*/ struct registry_key *key,
struct reg_format *f)
{
int ret=-1;
WERROR werr;
uint32_t count;
struct registry_value *valvalue = NULL;
char *valname = NULL;
char *subkey_name = NULL;
NTTIME modtime = 0;
reg_format_registry_key(f, key, false);
/* print values */
for (count = 0;
werr = reg_enumvalue(ctx, key, count, &valname, &valvalue),
W_ERROR_IS_OK(werr);
count++)
{
reg_format_registry_value(f, valname, valvalue);
}
if (!W_ERROR_EQUAL(WERR_NO_MORE_ITEMS, werr)) {
d_fprintf(stderr, _("reg_enumvalue failed: %s\n"),
win_errstr(werr));
goto done;
}
/* recurse on subkeys */
for (count = 0;
werr = reg_enumkey(ctx, key, count, &subkey_name, &modtime),
W_ERROR_IS_OK(werr);
count++)
{
struct registry_key *subkey = NULL;
werr = reg_openkey(ctx, key, subkey_name, REG_KEY_READ,
&subkey);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, _("reg_openkey failed: %s\n"),
win_errstr(werr));
goto done;
}
registry_export(ctx, subkey, f);
TALLOC_FREE(subkey);
}
if (!W_ERROR_EQUAL(WERR_NO_MORE_ITEMS, werr)) {
d_fprintf(stderr, _("reg_enumkey failed: %s\n"),
win_errstr(werr));
goto done;
}
ret = 0;
done:
return ret;
}
static int net_registry_export(struct net_context *c, int argc,
const char **argv)
{
int ret=-1;
WERROR werr;
struct registry_key *key = NULL;
TALLOC_CTX *ctx = talloc_stackframe();
struct reg_format *f=NULL;
if (argc < 2 || argc > 3 || c->display_usage) {
d_printf("%s\n%s",
_("Usage:"),
_("net registry export [opt]\n"));
d_printf("%s\n%s",
_("Example:"),
_("net registry export 'HKLM\\Software\\Samba' "
"samba.reg regedit5\n"));
goto done;
}
werr = open_key(ctx, argv[0], REG_KEY_READ, &key);
if (!W_ERROR_IS_OK(werr)) {
d_fprintf(stderr, _("open_key failed: %s\n"), win_errstr(werr));
goto done;
}
f = reg_format_file(ctx, argv[1], (argc > 2) ? argv[2] : NULL);
if (f == NULL) {
d_fprintf(stderr, _("open file failed: %s\n"), strerror(errno));
goto done;
}
ret = registry_export(ctx, key, f);
done:
TALLOC_FREE(ctx);
return ret;
}
/**@}*/
/******************************************************************************/
/**
* @defgroup net_registry_convert Convert
* @ingroup net_registry
* @{
*/
static int net_registry_convert(struct net_context *c, int argc,
const char **argv)
{
int ret;
TALLOC_CTX *mem_ctx;
const char *in_opt = NULL;
const char *out_opt = NULL;
if (argc < 2 || argc > 4|| c->display_usage) {
d_printf("%s\n%s",
_("Usage:"),
_("net registry convert [in_opt] [out_opt]\n"
"net registry convert [out_opt]\n"));
d_printf("%s\n%s",
_("Example:"),
_("net registry convert in.reg out.reg regedit4,enc=CP1252\n"));
return -1;
}
mem_ctx = talloc_stackframe();
switch (argc ) {
case 2:
break;
case 3:
out_opt = argv[2];
break;
case 4:
out_opt = argv[3];
in_opt = argv[2];
break;
default:
assert(false);
}
ret = reg_parse_file(argv[0], (struct reg_parse_callback*)
reg_format_file(mem_ctx, argv[1], out_opt),
in_opt);
talloc_free(mem_ctx);
return ret;
}
/**@}*/
static int net_registry_check(struct net_context *c, int argc,
const char **argv)
{
char *dbfile;
struct check_options opts;
int ret;
if (argc > 1|| c->display_usage) {
d_printf("%s\n%s",
_("Usage:"),
_("net registry check [-vraTfl] [-o ] [--wipe] []\n"
" Check a registry database.\n"
" -v|--verbose\t be verbose\n"
" -r|--repair\t\t interactive repair mode\n"
" -a|--auto\t\t noninteractive repair mode\n"
" -T|--test\t\t dry run\n"
" -f|--force\t\t force\n"
" -l|--lock\t\t lock while doing the check\n"
" -o|--output=\t output database\n"
" --reg-version=n\t assume database format version {n|1,2,3}\n"
" --wipe\t\t create a new database from scratch\n"
" --db=\t\t registry database to open\n"));
return c->display_usage ? 0 : -1;
}
if (c->opt_db != NULL) {
dbfile = talloc_strdup(talloc_tos(), c->opt_db);
} else if (argc > 0) {
dbfile = talloc_strdup(talloc_tos(), argv[0]);
} else {
dbfile = state_path(talloc_tos(), "registry.tdb");
}
if (dbfile == NULL) {
return -1;
}
opts = (struct check_options) {
.lock = c->opt_lock || c->opt_long_list_entries,
.test = c->opt_testmode,
.automatic = c->opt_auto,
.verbose = c->opt_verbose,
.force = c->opt_force,
.repair = c->opt_repair || c->opt_reboot,
.version = c->opt_reg_version,
.output = c->opt_output,
.wipe = c->opt_wipe,
.implicit_db = (c->opt_db == NULL) && (argc == 0),
};
ret = net_registry_check_db(dbfile, &opts);
talloc_free(dbfile);
return ret;
}
/******************************************************************************/
int net_registry(struct net_context *c, int argc, const char **argv)
{
int ret = -1;
struct functable func[] = {
{
"enumerate",
net_registry_enumerate,
NET_TRANSPORT_LOCAL,
N_("Enumerate registry keys and values"),
N_("net registry enumerate\n"
" Enumerate registry keys and values")
},
{
"enumerate_recursive",
net_registry_enumerate_recursive,
NET_TRANSPORT_LOCAL,
N_("Enumerate registry keys and values"),
N_("net registry enumerate_recursive\n"
" Enumerate registry keys and values")
},
{
"createkey",
net_registry_createkey,
NET_TRANSPORT_LOCAL,
N_("Create a new registry key"),
N_("net registry createkey\n"
" Create a new registry key")
},
{
"deletekey",
net_registry_deletekey,
NET_TRANSPORT_LOCAL,
N_("Delete a registry key"),
N_("net registry deletekey\n"
" Delete a registry key")
},
{
"deletekey_recursive",
net_registry_deletekey_recursive,
NET_TRANSPORT_LOCAL,
N_("Delete a registry key with subkeys"),
N_("net registry deletekey_recursive\n"
" Delete a registry key with subkeys")
},
{
"getvalue",
net_registry_getvalue,
NET_TRANSPORT_LOCAL,
N_("Print a registry value"),
N_("net registry getvalue\n"
" Print a registry value")
},
{
"getvalueraw",
net_registry_getvalueraw,
NET_TRANSPORT_LOCAL,
N_("Print a registry value (raw format)"),
N_("net registry getvalueraw\n"
" Print a registry value (raw format)")
},
{
"getvaluesraw",
net_registry_getvaluesraw,
NET_TRANSPORT_LOCAL,
"Print all values of a key in raw format",
"net registry getvaluesraw \n"
" Print a registry value (raw format)"
},
{
"setvalue",
net_registry_setvalue,
NET_TRANSPORT_LOCAL,
N_("Set a new registry value"),
N_("net registry setvalue\n"
" Set a new registry value")
},
{
"increment",
net_registry_increment,
NET_TRANSPORT_LOCAL,
N_("Increment a DWORD registry value under a lock"),
N_("net registry increment\n"
" Increment a DWORD registry value under a lock")
},
{
"deletevalue",
net_registry_deletevalue,
NET_TRANSPORT_LOCAL,
N_("Delete a registry value"),
N_("net registry deletevalue\n"
" Delete a registry value")
},
{
"getsd",
net_registry_getsd,
NET_TRANSPORT_LOCAL,
N_("Get security descriptor"),
N_("net registry getsd\n"
" Get security descriptor")
},
{
"getsd_sddl",
net_registry_getsd_sddl,
NET_TRANSPORT_LOCAL,
N_("Get security descriptor in sddl format"),
N_("net registry getsd_sddl\n"
" Get security descriptor in sddl format")
},
{
"setsd_sddl",
net_registry_setsd_sddl,
NET_TRANSPORT_LOCAL,
N_("Set security descriptor from sddl format string"),
N_("net registry setsd_sddl\n"
" Set security descriptor from sddl format string")
},
{
"import",
net_registry_import,
NET_TRANSPORT_LOCAL,
N_("Import .reg file"),
N_("net registry import\n"
" Import .reg file")
},
{
"export",
net_registry_export,
NET_TRANSPORT_LOCAL,
N_("Export .reg file"),
N_("net registry export\n"
" Export .reg file")
},
{
"convert",
net_registry_convert,
NET_TRANSPORT_LOCAL,
N_("Convert .reg file"),
N_("net registry convert\n"
" Convert .reg file")
},
{
"check",
net_registry_check,
NET_TRANSPORT_LOCAL,
N_("Check a registry database"),
N_("net registry check\n"
" Check a registry database")
},
{ NULL, NULL, 0, NULL, NULL }
};
if (!c->display_usage
&& argc > 0
&& (strcasecmp_m(argv[0], "convert") != 0)
&& (strcasecmp_m(argv[0], "check") != 0))
{
if (!W_ERROR_IS_OK(registry_init_basic())) {
return -1;
}
}
ret = net_run_function(c, argc, argv, "net registry", func);
return ret;
}