/*
 * This file and its contents are supplied under the terms of the
 * Common Development and Distribution License ("CDDL"), version 1.0.
 * You may only use this file in accordance with the terms of version
 * 1.0 of the CDDL.
 *
 * A full copy of the text of the CDDL should have accompanied this
 * source.  A copy of the CDDL is also available via the Internet at
 * http://www.illumos.org/license/CDDL.
 */

/*
 * Copyright 2022 Tintri by DDN, Inc. All rights reserved.
 */

/*
 * Test usr/src/common/smbsrv/smb_sid.c with large SIDs
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <smbsrv/smb_sid.h>
#include <limits.h>

void
test_sid(const char *sidstr, uint8_t idauth, const uint32_t *subauths,
    size_t subauth_cnt)
{
        char newstr[1024];
        smb_sid_t *sid;
        int i;

        sid = smb_sid_fromstr(sidstr);
        if (!smb_sid_isvalid(sid)) {
                fprintf(stderr, "SID %s not valid: %p\n", sidstr, sid);
                exit(1);
        }

        smb_sid_tostr(sid, newstr);

        if (strncmp(sidstr, newstr, sizeof (newstr)) != 0) {
                fprintf(stderr, "SID %s did not match decoded SID %s\n",
                    sidstr, newstr);
                exit(5);
        }

        if (subauths == NULL) {
                smb_sid_free(sid);
                return;
        }

        if (sid->sid_authority[5] != idauth) {
                fprintf(stderr, "Wrong SID authority %u (expected %u): %s\n",
                    sid->sid_authority, idauth, sidstr);
                exit(2);
        }

        if (sid->sid_subauthcnt != subauth_cnt) {
                fprintf(stderr, "Wrong subauthcnt %u (expected %u): %s\n",
                    sid->sid_subauthcnt, subauth_cnt, sidstr);
                exit(3);
        }

        for (i = 0; i < subauth_cnt; i++) {
                if (sid->sid_subauth[i] != subauths[i]) {
                        fprintf(stderr,
                            "Wrong subauthcnt %u (expected %u): %s\n",
                            sid->sid_subauthcnt, subauth_cnt, sidstr);
                        exit(4);
                }
        }

        smb_sid_free(sid);
}

int
main(int argc, char *argv[])
{
        char sid[1024];
        uint32_t subauths[NT_SID_SUBAUTH_MAX];
        size_t len = sizeof (sid);
        int off = 0;
        int i, idauth;

        if (argc > 1) {
                test_sid(argv[1], 0, NULL, 0);
                goto out;
        }

        for (idauth = 2; idauth <= UINT8_MAX; idauth += 11) {
                off = snprintf(&sid[0], len, "S-1-%u", idauth);
                for (i = 0; i < NT_SID_SUBAUTH_MAX; i++) {
                        subauths[i] = arc4random();
                        off += snprintf(&sid[off], len - off,
                            "-%u", subauths[i]);
                }
                test_sid(sid, idauth, subauths, NT_SID_SUBAUTH_MAX);
        }

out:
        printf("success!\n");
        return (0);
}