1
0
mirror of https://github.com/samba-team/samba.git synced 2024-12-22 13:34:15 +03:00

regfio: Add trivial unit test

An upcoming commit will resolve two cases of insufficient handling of
mangled registry hive files and will include unit tests.

BUG: https://bugzilla.samba.org/show_bug.cgi?id=13840

Signed-off-by: Michael Hanselmann <public@hansmi.ch>
Reviewed-by: Douglas Bagnall <douglas.bagnall@catalyst.net.nz>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>
This commit is contained in:
Michael Hanselmann 2019-03-19 00:47:52 +01:00 committed by Andrew Bartlett
parent aa6b355858
commit 9b2cb845b2
3 changed files with 147 additions and 0 deletions

View File

@ -257,3 +257,5 @@ plantestsuite("samba.unittests.ms_fnmatch", "none",
[os.path.join(bindir(), "default/lib/util/test_ms_fnmatch")])
plantestsuite("samba.unittests.ntlm_check", "none",
[os.path.join(bindir(), "default/libcli/auth/test_ntlm_check")])
plantestsuite("samba.unittests.test_registry_regfio", "none",
[os.path.join(bindir(), "default/source3/test_registry_regfio")])

View File

@ -0,0 +1,139 @@
/*
* Unix SMB/CIFS implementation.
*
* Copyright (C) 2019 Michael Hanselmann <public@hansmi.ch>
*
* 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 <stdarg.h>
#include <stddef.h>
#include <setjmp.h>
#include <cmocka.h>
#include <errno.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include "includes.h"
#include "lib/replace/replace.h"
#include "lib/util/samba_util.h"
#include "registry/regfio.h"
struct test_ctx {
char *tmp_regfile;
int tmp_regfile_fd;
REGF_FILE *rb;
};
static int setup_context(void **state)
{
struct test_ctx *test_ctx;
test_ctx = talloc_zero(NULL, struct test_ctx);
assert_non_null(test_ctx);
test_ctx->tmp_regfile_fd = -1;
*state = test_ctx;
return 0;
}
static int setup_context_tempfile(void **state)
{
struct test_ctx *test_ctx;
int ret;
ret = setup_context(state);
if (ret == 0) {
test_ctx = talloc_get_type_abort(*state, struct test_ctx);
test_ctx->tmp_regfile = talloc_strdup(test_ctx, "/tmp/regfio.XXXXXX");
assert_non_null(test_ctx->tmp_regfile);
test_ctx->tmp_regfile_fd = mkstemp(test_ctx->tmp_regfile);
assert_return_code(test_ctx->tmp_regfile_fd, errno);
}
return ret;
}
static int teardown_context(void **state)
{
struct test_ctx *test_ctx =
talloc_get_type_abort(*state, struct test_ctx);
if (test_ctx->rb) {
regfio_close(test_ctx->rb);
}
if (test_ctx->tmp_regfile) {
unlink(test_ctx->tmp_regfile);
}
if (test_ctx->tmp_regfile_fd != -1) {
close(test_ctx->tmp_regfile_fd);
}
talloc_free(test_ctx);
return 0;
}
static void test_regfio_open_new_file(void **state)
{
struct test_ctx *test_ctx =
talloc_get_type_abort(*state, struct test_ctx);
REGF_NK_REC *root;
struct regval_ctr *values;
struct regsubkey_ctr *subkeys;
WERROR werr;
test_ctx->rb = regfio_open(test_ctx->tmp_regfile,
O_RDWR | O_CREAT | O_TRUNC, 0600);
assert_non_null(test_ctx->rb);
root = regfio_rootkey(test_ctx->rb);
assert_null(root);
werr = regsubkey_ctr_init(NULL, &subkeys);
assert_true(W_ERROR_IS_OK(werr));
werr = regval_ctr_init(subkeys, &values);
assert_true(W_ERROR_IS_OK(werr));
// Write root key
regfio_write_key(test_ctx->rb, "", values, subkeys, NULL, NULL);
root = regfio_rootkey(test_ctx->rb);
assert_non_null(root);
assert_memory_equal(root->header, "nk", sizeof(root->header));
assert_int_equal(root->key_type, NK_TYPE_ROOTKEY);
}
int main(void) {
const struct CMUnitTest tests[] = {
cmocka_unit_test_setup_teardown(test_regfio_open_new_file,
setup_context_tempfile,
teardown_context),
};
cmocka_set_message_output(CM_OUTPUT_SUBUNIT);
return cmocka_run_group_tests(tests, NULL, NULL);
}

View File

@ -203,6 +203,12 @@ bld.SAMBA3_SUBSYSTEM('REGFIO',
source='registry/regfio.c',
deps='samba-util REG_PARSE_PRS')
bld.SAMBA_BINARY('test_registry_regfio',
source='registry/tests/test_regfio.c',
deps='cmocka samba3-util smbconf REGFIO',
local_include=False,
install=False)
bld.SAMBA3_SUBSYSTEM('REG_API_REGF',
source='registry/reg_api_regf.c',
deps='samba-util')