/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
/*
 * test_certstore.c
 *
 * Test CertStore Type
 *
 */

#include "testutil.h"
#include "testutil_nss.h"

static void *plContext = NULL;

static PKIX_Error *
testCRLCallback(
    PKIX_CertStore *store,
    PKIX_CRLSelector *selector,
    void **pNBIOContext,
    PKIX_List **pCrls, /* list of PKIX_PL_Crl */
    void *plContext)
{
    return (0);
}

static PKIX_Error *
testCRLContinue(
    PKIX_CertStore *store,
    PKIX_CRLSelector *selector,
    void **pNBIOContext,
    PKIX_List **pCrls, /* list of PKIX_PL_Crl */
    void *plContext)
{
    return (0);
}

static PKIX_Error *
testCertCallback(
    PKIX_CertStore *store,
    PKIX_CertSelector *selector,
    void **pNBIOContext,
    PKIX_List **pCerts, /* list of PKIX_PL_Cert */
    void *plContext)
{
    return (0);
}

static PKIX_Error *
testCertContinue(
    PKIX_CertStore *store,
    PKIX_CertSelector *selector,
    void **pNBIOContext,
    PKIX_List **pCerts, /* list of PKIX_PL_Cert */
    void *plContext)
{
    return (0);
}

static char *
catDirName(char *platform, char *dir, void *plContext)
{
    char *pathName = NULL;
    PKIX_UInt32 dirLen;
    PKIX_UInt32 platformLen;

    PKIX_TEST_STD_VARS();

    dirLen = PL_strlen(dir);
    platformLen = PL_strlen(platform);

    PKIX_TEST_EXPECT_NO_ERROR(PKIX_PL_Malloc(platformLen +
                                                 dirLen +
                                                 2,
                                             (void **)&pathName, plContext));

    PL_strcpy(pathName, platform);
    PL_strcat(pathName, "/");
    PL_strcat(pathName, dir);

cleanup:

    PKIX_TEST_RETURN();

    return (pathName);
}

static void
testCertStore(char *crlDir)
{
    PKIX_PL_String *dirString = NULL;
    PKIX_CertStore *certStore = NULL;
    PKIX_PL_Object *getCertStoreContext = NULL;
    PKIX_CertStore_CertCallback certCallback = NULL;
    PKIX_CertStore_CRLCallback crlCallback = NULL;

    PKIX_TEST_STD_VARS();

    PKIX_TEST_EXPECT_NO_ERROR(PKIX_PL_String_Create(PKIX_ESCASCII,
                                                    crlDir,
                                                    0,
                                                    &dirString,
                                                    plContext));

    subTest("PKIX_CertStore_Create");
    PKIX_TEST_EXPECT_NO_ERROR(PKIX_CertStore_Create(testCertCallback,
                                                    testCRLCallback,
                                                    testCertContinue,
                                                    testCRLContinue,
                                                    NULL, /* trustCallback */
                                                    (PKIX_PL_Object *)dirString,
                                                    PKIX_TRUE, /* cacheFlag */
                                                    PKIX_TRUE, /* local */
                                                    &certStore,
                                                    plContext));

    subTest("PKIX_CertStore_GetCertCallback");
    PKIX_TEST_EXPECT_NO_ERROR(PKIX_CertStore_GetCertCallback(certStore, &certCallback, plContext));

    if (certCallback != testCertCallback) {
        testError("PKIX_CertStore_GetCertCallback unexpected mismatch");
    }

    subTest("PKIX_CertStore_GetCRLCallback");
    PKIX_TEST_EXPECT_NO_ERROR(PKIX_CertStore_GetCRLCallback(certStore, &crlCallback, plContext));

    if (crlCallback != testCRLCallback) {
        testError("PKIX_CertStore_GetCRLCallback unexpected mismatch");
    }

    subTest("PKIX_CertStore_GetCertStoreContext");
    PKIX_TEST_EXPECT_NO_ERROR(PKIX_CertStore_GetCertStoreContext(certStore, &getCertStoreContext, plContext));

    if ((PKIX_PL_Object *)dirString != getCertStoreContext) {
        testError("PKIX_CertStore_GetCertStoreContext unexpected mismatch");
    }

cleanup:

    PKIX_TEST_DECREF_AC(dirString);
    PKIX_TEST_DECREF_AC(certStore);
    PKIX_TEST_DECREF_AC(getCertStoreContext);

    PKIX_TEST_RETURN();
}

static void
printUsage(char *pName)
{
    printf("\nUSAGE: %s testName <data-dir> <platform-dir>\n\n", pName);
}

/* Functional tests for CertStore public functions */

int
test_store(int argc, char *argv[])
{

    char *platformDir = NULL;
    char *dataDir = NULL;
    char *combinedDir = NULL;
    PKIX_UInt32 actualMinorVersion;
    PKIX_UInt32 j = 0;

    PKIX_TEST_STD_VARS();

    PKIX_TEST_EXPECT_NO_ERROR(
        PKIX_PL_NssContext_Create(0, PKIX_FALSE, NULL, &plContext));

    if (argc < (3 + j)) {
        printUsage(argv[0]);
        return (0);
    }

    startTests(argv[1 + j]);

    dataDir = argv[2 + j];
    platformDir = argv[3 + j];
    combinedDir = catDirName(platformDir, dataDir, plContext);

    testCertStore(combinedDir);

cleanup:

    pkixTestErrorResult = PKIX_PL_Free(combinedDir, plContext);

    PKIX_Shutdown(plContext);

    PKIX_TEST_RETURN();

    endTests("CertStore");

    return (0);
}