summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
authorSumit Bose <sbose@redhat.com>2012-12-04 13:55:13 +0100
committerJakub Hrozek <jhrozek@redhat.com>2013-01-08 14:42:57 +0100
commit34f64327316a7dafbac593bd9e507aa628a3339b (patch)
tree4affa1e79009de89f057dce70d9f1925e1545844 /src
parent908b6d56158438f3fe96d55cc8166b414729f628 (diff)
downloadsssd-34f64327316a7dafbac593bd9e507aa628a3339b.tar.gz
sssd-34f64327316a7dafbac593bd9e507aa628a3339b.tar.bz2
sssd-34f64327316a7dafbac593bd9e507aa628a3339b.zip
Add tests for get_gids_from_pac()
Diffstat (limited to 'src')
-rw-r--r--src/tests/pac_responder-tests.c256
1 files changed, 256 insertions, 0 deletions
diff --git a/src/tests/pac_responder-tests.c b/src/tests/pac_responder-tests.c
index 81cee3c1..b2a14578 100644
--- a/src/tests/pac_responder-tests.c
+++ b/src/tests/pac_responder-tests.c
@@ -23,11 +23,23 @@
#include <check.h>
#include <stdbool.h>
+#include <math.h>
#include <util/data_blob.h>
#include <gen_ndr/security.h>
#include "tests/common.h"
#include "responder/pac/pacsrv.h"
+#include "lib/idmap/sss_idmap.h"
+
+struct dom_sid test_dom_sid = {1, 4, {0, 0, 0, 0, 0, 5},
+ {21, 2127521184, 1604012920, 1887927527, 0,
+ 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}};
+const char *test_dom_sid_str = "S-1-5-21-2127521184-1604012920-1887927527";
+
+struct dom_sid test_remote_dom_sid = {1, 4, {0, 0, 0, 0, 0, 5},
+ {21, 123, 456, 789, 0,
+ 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}};
+const char *test_remote_dom_sid_str = "S-1-5-21-123-456-789";
struct dom_sid test_smb_sid = {1, 5, {0, 0, 0, 0, 0, 5},
{21, 2127521184, 1604012920, 1887927527, 1123,
@@ -43,6 +55,90 @@ struct local_mapping_ranges test_map = {{1200000, 1399999},
{1000, 200999},
{201000, 400999}};
+static void *idmap_talloc(size_t size, void *pvt)
+{
+ return talloc_size(pvt, size);
+}
+
+static void idmap_talloc_free(void *ptr, void *pvt)
+{
+ talloc_free(ptr);
+}
+
+struct pac_ctx *pac_ctx;
+
+#define IDMAP_RANGE_MIN 1234
+#define IDMAP_RANGE_MAX 9876543
+
+void pac_setup(void) {
+ enum idmap_error_code err;
+ struct sss_idmap_range remote_range = {IDMAP_RANGE_MIN, IDMAP_RANGE_MAX};
+ struct sss_domain_info *sd;
+
+ pac_ctx = talloc_zero(global_talloc_context, struct pac_ctx);
+ fail_unless(pac_ctx != NULL, "talloc_zero failed.\n");
+
+ pac_ctx->rctx = talloc_zero(pac_ctx, struct resp_ctx);
+ fail_unless(pac_ctx->rctx != NULL, "talloc_zero failed.");
+
+ pac_ctx->rctx->domains = talloc_zero(pac_ctx->rctx, struct sss_domain_info);
+ fail_unless(pac_ctx->rctx->domains != NULL, "talloc_zero failed.");
+
+ pac_ctx->rctx->domains->name = talloc_strdup(pac_ctx->rctx->domains,
+ "TEST.DOM");
+ fail_unless(pac_ctx->rctx->domains->name != NULL, "talloc_strdup failed.");
+
+ pac_ctx->rctx->domains->flat_name = talloc_strdup(pac_ctx->rctx->domains,
+ "TESTDOM");
+ fail_unless(pac_ctx->rctx->domains->flat_name != NULL,
+ "talloc_strdup failed.");
+
+ pac_ctx->rctx->domains->domain_id = talloc_strdup(pac_ctx->rctx->domains,
+ test_dom_sid_str);
+ fail_unless(pac_ctx->rctx->domains->domain_id != NULL,
+ "talloc_strdup failed.");
+
+ pac_ctx->rctx->domains->subdomain_count = 1;
+ pac_ctx->rctx->domains->subdomains = talloc_zero_array(pac_ctx->rctx->domains,
+ struct sss_domain_info *,
+ pac_ctx->rctx->domains->subdomain_count);
+ fail_unless(pac_ctx->rctx->domains->subdomains != NULL,
+ "talloc_array_zero failed");
+
+ sd = talloc_zero(pac_ctx->rctx->domains->subdomains,
+ struct sss_domain_info);
+ fail_unless(sd != NULL, "talloc_zero failed.");
+
+ sd->name = talloc_strdup(sd, "remote.dom");
+ fail_unless(sd->name != NULL, "talloc_strdup failed");
+
+ sd->flat_name = talloc_strdup(sd, "REMOTEDOM");
+ fail_unless(sd->flat_name != NULL, "talloc_strdup failed");
+
+ sd->domain_id = talloc_strdup(sd, test_remote_dom_sid_str);
+ fail_unless(sd->domain_id != NULL, "talloc_strdup failed");
+
+ pac_ctx->rctx->domains->subdomains[0] = sd;
+
+ err = sss_idmap_init(idmap_talloc, pac_ctx, idmap_talloc_free,
+ &pac_ctx->idmap_ctx);
+
+ fail_unless(err == IDMAP_SUCCESS, "sss_idmap_init failed.");
+ fail_unless(pac_ctx->idmap_ctx != NULL, "sss_idmap_init returned NULL.");
+
+ err = sss_idmap_add_domain(pac_ctx->idmap_ctx, "remote.dom",
+ test_remote_dom_sid_str, &remote_range);
+
+ pac_ctx->my_dom_sid = &test_dom_sid;
+
+ pac_ctx->range_map = &test_map;
+}
+
+void pac_teardown(void)
+{
+ talloc_free(pac_ctx);
+}
+
START_TEST(pac_test_local_sid_to_id)
{
@@ -226,6 +322,161 @@ START_TEST(pac_test_find_domain_by_id)
}
END_TEST
+START_TEST(pac_test_get_gids_from_pac)
+{
+ int ret;
+ size_t c;
+ size_t g;
+ size_t t;
+ size_t gid_count;
+ struct pac_grp *gids;
+ struct PAC_LOGON_INFO *logon_info;
+ bool found;
+ gid_t exp_gid;
+ struct timeval start_time;
+ struct timeval end_time;
+ struct timeval diff_time;
+
+ ret = get_gids_from_pac(NULL, NULL, NULL, NULL, NULL, NULL, NULL);
+ fail_unless(ret == EINVAL, "Unexpected return value for NULL parameters");
+
+ logon_info = talloc_zero(global_talloc_context, struct PAC_LOGON_INFO);
+ fail_unless(logon_info != NULL, "talloc_zero failed.\n");
+
+ ret = get_gids_from_pac(global_talloc_context, pac_ctx, pac_ctx->range_map,
+ pac_ctx->my_dom_sid, logon_info, &gid_count, &gids);
+ fail_unless(ret == EOK, "Failed with empty PAC");
+ fail_unless(gid_count == 0, "O groups expected, got [%d]", gid_count);
+ fail_unless(gids == NULL, "Expected NULL gid array.");
+
+ logon_info->info3.base.domain_sid = &test_smb_sid_2nd; /* unknown SID */
+ logon_info->info3.base.groups.count = 10;
+ logon_info->info3.base.groups.rids = talloc_array(logon_info,
+ struct samr_RidWithAttribute,
+ logon_info->info3.base.groups.count);
+ fail_unless(logon_info->info3.base.groups.rids != NULL, "talloc_array failed.");
+
+ for (c = 0; c < logon_info->info3.base.groups.count; c++) {
+ logon_info->info3.base.groups.rids[c].rid = 500 + c;
+ }
+
+ ret = get_gids_from_pac(global_talloc_context, pac_ctx, pac_ctx->range_map,
+ pac_ctx->my_dom_sid, logon_info, &gid_count, &gids);
+ fail_unless(ret == EINVAL, "Unexpected return code [%d] with unknown SID.",
+ ret);
+
+ /* change SID to a known one */
+ logon_info->info3.base.domain_sid = &test_remote_dom_sid;
+
+ ret = get_gids_from_pac(global_talloc_context, pac_ctx, pac_ctx->range_map,
+ pac_ctx->my_dom_sid, logon_info, &gid_count, &gids);
+ fail_unless(ret == EOK, "Failed with 10 RIDs in PAC");
+ fail_unless(gid_count == logon_info->info3.base.groups.count,
+ "[%d] groups expected, got [%d]",
+ logon_info->info3.base.groups.count, gid_count);
+ fail_unless(gids != NULL, "Expected gid array.");
+
+ for (c = 0; c < logon_info->info3.base.groups.count; c++) {
+ found = false;
+ exp_gid = IDMAP_RANGE_MIN + 500 + c;
+ for (g = 0; g < gid_count; g++) {
+ if (gids[g].gid == exp_gid) {
+ found = true;
+ break;
+ }
+ }
+ fail_unless(found, "[%d] not found in group list", exp_gid);
+ }
+
+ talloc_free(gids);
+ gids = NULL;
+
+ /* duplicated RIDs */
+ for (c = 0; c < logon_info->info3.base.groups.count; c++) {
+ logon_info->info3.base.groups.rids[c].rid = 500;
+ }
+
+ ret = get_gids_from_pac(global_talloc_context, pac_ctx, pac_ctx->range_map,
+ pac_ctx->my_dom_sid, logon_info, &gid_count, &gids);
+ fail_unless(ret == EOK, "Failed with 10 duplicated RIDs in PAC");
+ fail_unless(gid_count == 1, "[%d] groups expected, got [%d]", 1, gid_count);
+ fail_unless(gids != NULL, "Expected gid array.");
+ fail_unless(gids[0].gid == IDMAP_RANGE_MIN + 500,
+ "Wrong gid returned, got [%d], expected [%d].", gids[0].gid,
+ IDMAP_RANGE_MIN + 500);
+ talloc_free(gids);
+ gids = NULL;
+
+ logon_info->info3.sidcount = 2;
+ logon_info->info3.sids = talloc_zero_array(logon_info, struct netr_SidAttr,
+ logon_info->info3.sidcount);
+ fail_unless(logon_info->info3.sids != NULL, "talloc_zero_array failed.");
+
+ logon_info->info3.sids[0].sid = &test_smb_sid;
+ logon_info->info3.sids[1].sid = &test_smb_sid_2nd;
+
+ ret = get_gids_from_pac(global_talloc_context, pac_ctx, pac_ctx->range_map,
+ pac_ctx->my_dom_sid, logon_info, &gid_count, &gids);
+ fail_unless(ret == EOK, "Failed with 10 duplicated RIDs and local SIDS in PAC");
+ fail_unless(gid_count == 3, "[%d] groups expected, got [%d]", 3, gid_count);
+ fail_unless(gids != NULL, "Expected gid array.");
+
+ gid_t exp_gids[] = {IDMAP_RANGE_MIN + 500, test_id, test_id_2nd, 0};
+
+ for (c = 0; exp_gids[c] != 0; c++) {
+ found = false;
+ for (g = 0; g < gid_count; g++) {
+ if (gids[g].gid == exp_gids[c]) {
+ found = true;
+ break;
+ }
+ }
+ fail_unless(found, "[%d] not found in group list", exp_gids[c]);
+ }
+
+ talloc_free(gids);
+ gids = NULL;
+
+ talloc_free(logon_info->info3.base.groups.rids);
+
+ for (t = 0; t < 7; t++) {
+ logon_info->info3.base.groups.count = powl(10, t);
+ logon_info->info3.base.groups.rids = talloc_array(logon_info,
+ struct samr_RidWithAttribute,
+ logon_info->info3.base.groups.count);
+ fail_unless(logon_info->info3.base.groups.rids != NULL, "talloc_array failed.");
+
+ for (c = 0; c < logon_info->info3.base.groups.count; c++) {
+ logon_info->info3.base.groups.rids[c].rid = 500 + c;
+ }
+
+ ret = gettimeofday(&start_time, NULL);
+ fail_unless(ret == 0, "gettimeofday failed.");
+
+ ret = get_gids_from_pac(global_talloc_context, pac_ctx, pac_ctx->range_map,
+ pac_ctx->my_dom_sid, logon_info, &gid_count, &gids);
+ fail_unless(ret == EOK, "Unexpected return code [%d].", ret);
+
+ ret = gettimeofday(&end_time, NULL);
+ fail_unless(ret == 0, "gettimeofday failed.");
+
+ timersub(&end_time, &start_time, &diff_time);
+ fprintf(stderr, "Testcase [%zu], number of groups [%u], " \
+ "duration [%ds %dus]\n", t,
+ logon_info->info3.base.groups.count,
+ (int) diff_time.tv_sec,
+ (int) diff_time.tv_usec);
+
+ talloc_free(gids);
+ gids = NULL;
+
+ talloc_free(logon_info->info3.base.groups.rids);
+ }
+
+ talloc_free(logon_info);
+}
+END_TEST
+
Suite *idmap_test_suite (void)
{
Suite *s = suite_create ("PAC responder");
@@ -235,10 +486,15 @@ Suite *idmap_test_suite (void)
leak_check_setup,
leak_check_teardown);
+ tcase_add_checked_fixture(tc_pac,
+ pac_setup,
+ pac_teardown);
+
tcase_add_test(tc_pac, pac_test_local_sid_to_id);
tcase_add_test(tc_pac, pac_test_seondary_local_sid_to_id);
tcase_add_test(tc_pac, pac_test_get_gids_to_add_and_remove);
tcase_add_test(tc_pac, pac_test_find_domain_by_id);
+ tcase_add_test(tc_pac, pac_test_get_gids_from_pac);
suite_add_tcase(s, tc_pac);