mirror of
https://github.com/rn10950/RetroZilla.git
synced 2024-11-16 04:20:32 +01:00
498 lines
17 KiB
C
498 lines
17 KiB
C
/* 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_buildchain_resourcelimits.c
|
|
*
|
|
* Test BuildChain function with constraints on resources
|
|
*
|
|
*/
|
|
|
|
#include "testutil.h"
|
|
#include "testutil_nss.h"
|
|
|
|
#define PKIX_TESTUSERCHECKER_TYPE (PKIX_NUMTYPES+30)
|
|
|
|
static void *plContext = NULL;
|
|
static PKIX_Boolean usebind = PKIX_FALSE;
|
|
static PKIX_Boolean useLDAP = PKIX_FALSE;
|
|
static char buf[PR_NETDB_BUF_SIZE];
|
|
static char *serverName = NULL;
|
|
|
|
static void printUsage(void) {
|
|
(void) printf("\nUSAGE:\ttest_buildchain_resourcelimits [-arenas] "
|
|
"[usebind] servername[:port]\\\n\t\t<testName> [ENE|EE]"
|
|
" <certStoreDirectory>\\\n\t\t<targetCert>"
|
|
" <intermediate Certs...> <trustedCert>\n\n");
|
|
(void) printf
|
|
("Builds a chain of certificates from <targetCert> to <trustedCert>\n"
|
|
"using the certs and CRLs in <certStoreDirectory>. "
|
|
"servername[:port] gives\n"
|
|
"the address of an LDAP server. If port is not"
|
|
" specified, port 389 is used.\n\"-\" means no LDAP server.\n\n"
|
|
"If ENE is specified, then an Error is Not Expected.\n"
|
|
"EE indicates an Error is Expected.\n");
|
|
}
|
|
|
|
static PKIX_Error *
|
|
createLdapCertStore(
|
|
char *hostname,
|
|
PRIntervalTime timeout,
|
|
PKIX_CertStore **pLdapCertStore,
|
|
void* plContext)
|
|
{
|
|
PRIntn backlog = 0;
|
|
|
|
char *bindname = "";
|
|
char *auth = "";
|
|
|
|
LDAPBindAPI bindAPI;
|
|
LDAPBindAPI *bindPtr = NULL;
|
|
PKIX_PL_LdapDefaultClient *ldapClient = NULL;
|
|
PKIX_CertStore *ldapCertStore = NULL;
|
|
|
|
PKIX_TEST_STD_VARS();
|
|
|
|
if (usebind) {
|
|
bindPtr = &bindAPI;
|
|
bindAPI.selector = SIMPLE_AUTH;
|
|
bindAPI.chooser.simple.bindName = bindname;
|
|
bindAPI.chooser.simple.authentication = auth;
|
|
}
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR(PKIX_PL_LdapDefaultClient_CreateByName
|
|
(hostname, timeout, bindPtr, &ldapClient, plContext));
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR(PKIX_PL_LdapCertStore_Create
|
|
((PKIX_PL_LdapClient *)ldapClient, &ldapCertStore, plContext));
|
|
|
|
*pLdapCertStore = ldapCertStore;
|
|
cleanup:
|
|
|
|
PKIX_TEST_DECREF_AC(ldapClient);
|
|
|
|
PKIX_TEST_RETURN();
|
|
|
|
return (pkixTestErrorResult);
|
|
|
|
}
|
|
|
|
static void Test_BuildResult(
|
|
PKIX_ProcessingParams *procParams,
|
|
PKIX_Boolean testValid,
|
|
PKIX_List *expectedCerts,
|
|
void *plContext)
|
|
{
|
|
PKIX_PL_Cert *cert = NULL;
|
|
PKIX_List *certs = NULL;
|
|
PKIX_PL_String *actualCertsString = NULL;
|
|
PKIX_PL_String *expectedCertsString = NULL;
|
|
PKIX_BuildResult *buildResult = NULL;
|
|
PKIX_Boolean result;
|
|
PKIX_Boolean supportForward = PKIX_FALSE;
|
|
PKIX_UInt32 numCerts, i;
|
|
char *asciiResult = NULL;
|
|
char *actualCertsAscii = NULL;
|
|
char *expectedCertsAscii = NULL;
|
|
void *state = NULL;
|
|
PRPollDesc *pollDesc = NULL;
|
|
|
|
PKIX_TEST_STD_VARS();
|
|
|
|
pkixTestErrorResult = PKIX_BuildChain
|
|
(procParams,
|
|
(void **)&pollDesc,
|
|
&state,
|
|
&buildResult,
|
|
NULL,
|
|
plContext);
|
|
|
|
while (pollDesc != NULL) {
|
|
|
|
if (PR_Poll(pollDesc, 1, 0) < 0) {
|
|
testError("PR_Poll failed");
|
|
}
|
|
|
|
pkixTestErrorResult = PKIX_BuildChain
|
|
(procParams,
|
|
(void **)&pollDesc,
|
|
&state,
|
|
&buildResult,
|
|
NULL,
|
|
plContext);
|
|
}
|
|
|
|
if (pkixTestErrorResult) {
|
|
if (testValid == PKIX_FALSE) { /* EE */
|
|
(void) printf("EXPECTED ERROR RECEIVED!\n");
|
|
} else { /* ENE */
|
|
testError("UNEXPECTED ERROR RECEIVED!\n");
|
|
}
|
|
PKIX_TEST_DECREF_BC(pkixTestErrorResult);
|
|
goto cleanup;
|
|
}
|
|
|
|
if (testValid == PKIX_TRUE) { /* ENE */
|
|
(void) printf("EXPECTED NON-ERROR RECEIVED!\n");
|
|
} else { /* EE */
|
|
testError("UNEXPECTED NON-ERROR RECEIVED!\n");
|
|
}
|
|
|
|
if (buildResult){
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR
|
|
(PKIX_BuildResult_GetCertChain
|
|
(buildResult, &certs, NULL));
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR
|
|
(PKIX_List_GetLength(certs, &numCerts, plContext));
|
|
|
|
printf("\n");
|
|
|
|
for (i = 0; i < numCerts; i++){
|
|
PKIX_TEST_EXPECT_NO_ERROR
|
|
(PKIX_List_GetItem
|
|
(certs,
|
|
i,
|
|
(PKIX_PL_Object**)&cert,
|
|
plContext));
|
|
|
|
asciiResult = PKIX_Cert2ASCII(cert);
|
|
|
|
printf("CERT[%d]:\n%s\n", i, asciiResult);
|
|
|
|
/* PKIX_Cert2ASCII used PKIX_PL_Malloc(...,,NULL) */
|
|
PKIX_TEST_EXPECT_NO_ERROR
|
|
(PKIX_PL_Free(asciiResult, NULL));
|
|
asciiResult = NULL;
|
|
|
|
PKIX_TEST_DECREF_BC(cert);
|
|
}
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR
|
|
(PKIX_PL_Object_Equals
|
|
((PKIX_PL_Object*)certs,
|
|
(PKIX_PL_Object*)expectedCerts,
|
|
&result,
|
|
plContext));
|
|
|
|
if (!result){
|
|
testError("BUILT CERTCHAIN IS "
|
|
"NOT THE ONE THAT WAS EXPECTED");
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR
|
|
(PKIX_PL_Object_ToString
|
|
((PKIX_PL_Object *)certs,
|
|
&actualCertsString,
|
|
plContext));
|
|
|
|
actualCertsAscii = PKIX_String2ASCII
|
|
(actualCertsString, plContext);
|
|
if (actualCertsAscii == NULL){
|
|
pkixTestErrorMsg = "PKIX_String2ASCII Failed";
|
|
goto cleanup;
|
|
}
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR
|
|
(PKIX_PL_Object_ToString
|
|
((PKIX_PL_Object *)expectedCerts,
|
|
&expectedCertsString,
|
|
plContext));
|
|
|
|
expectedCertsAscii = PKIX_String2ASCII
|
|
(expectedCertsString, plContext);
|
|
if (expectedCertsAscii == NULL){
|
|
pkixTestErrorMsg = "PKIX_String2ASCII Failed";
|
|
goto cleanup;
|
|
}
|
|
|
|
(void) printf("Actual value:\t%s\n", actualCertsAscii);
|
|
(void) printf("Expected value:\t%s\n",
|
|
expectedCertsAscii);
|
|
}
|
|
|
|
}
|
|
|
|
cleanup:
|
|
|
|
PKIX_PL_Free(asciiResult, NULL);
|
|
PKIX_PL_Free(actualCertsAscii, plContext);
|
|
PKIX_PL_Free(expectedCertsAscii, plContext);
|
|
PKIX_TEST_DECREF_AC(state);
|
|
PKIX_TEST_DECREF_AC(buildResult);
|
|
PKIX_TEST_DECREF_AC(certs);
|
|
PKIX_TEST_DECREF_AC(cert);
|
|
PKIX_TEST_DECREF_AC(actualCertsString);
|
|
PKIX_TEST_DECREF_AC(expectedCertsString);
|
|
|
|
PKIX_TEST_RETURN();
|
|
|
|
}
|
|
|
|
int test_buildchain_resourcelimits(int argc, char *argv[])
|
|
{
|
|
PKIX_ComCertSelParams *certSelParams = NULL;
|
|
PKIX_CertSelector *certSelector = NULL;
|
|
PKIX_TrustAnchor *anchor = NULL;
|
|
PKIX_List *anchors = NULL;
|
|
PKIX_ProcessingParams *procParams = NULL;
|
|
PKIX_CertChainChecker *checker = NULL;
|
|
PKIX_ResourceLimits *resourceLimits = NULL;
|
|
char *dirName = NULL;
|
|
PKIX_PL_String *dirNameString = NULL;
|
|
PKIX_PL_Cert *trustedCert = NULL;
|
|
PKIX_PL_Cert *targetCert = NULL;
|
|
PKIX_PL_Cert *dirCert = NULL;
|
|
PKIX_UInt32 actualMinorVersion = 0;
|
|
PKIX_UInt32 j = 0;
|
|
PKIX_UInt32 k = 0;
|
|
PKIX_CertStore *ldapCertStore = NULL;
|
|
PRIntervalTime timeout = 0; /* 0 for non-blocking */
|
|
PKIX_CertStore *certStore = NULL;
|
|
PKIX_List *certStores = NULL;
|
|
PKIX_List *expectedCerts = NULL;
|
|
PKIX_Boolean testValid = PKIX_FALSE;
|
|
PKIX_Boolean usebind = PKIX_FALSE;
|
|
PKIX_Boolean useLDAP = PKIX_FALSE;
|
|
|
|
PKIX_TEST_STD_VARS();
|
|
|
|
if (argc < 5){
|
|
printUsage();
|
|
return (0);
|
|
}
|
|
|
|
startTests("BuildChain_ResourceLimits");
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR(
|
|
PKIX_PL_NssContext_Create(0, PKIX_FALSE, NULL, &plContext));
|
|
|
|
/*
|
|
* arguments:
|
|
* [optional] -arenas
|
|
* [optional] usebind
|
|
* servername or servername:port ( - for no server)
|
|
* testname
|
|
* EE or ENE
|
|
* cert directory
|
|
* target cert (end entity)
|
|
* intermediate certs
|
|
* trust anchor
|
|
*/
|
|
|
|
/* optional argument "usebind" for Ldap CertStore */
|
|
if (argv[j + 1]) {
|
|
if (PORT_Strcmp(argv[j + 1], "usebind") == 0) {
|
|
usebind = PKIX_TRUE;
|
|
j++;
|
|
}
|
|
}
|
|
|
|
if (PORT_Strcmp(argv[++j], "-") == 0) {
|
|
useLDAP = PKIX_FALSE;
|
|
} else {
|
|
serverName = argv[j];
|
|
}
|
|
|
|
subTest(argv[++j]);
|
|
|
|
/* ENE = expect no error; EE = expect error */
|
|
if (PORT_Strcmp(argv[++j], "ENE") == 0) {
|
|
testValid = PKIX_TRUE;
|
|
} else if (PORT_Strcmp(argv[j], "EE") == 0) {
|
|
testValid = PKIX_FALSE;
|
|
} else {
|
|
printUsage();
|
|
return (0);
|
|
}
|
|
|
|
dirName = argv[++j];
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR(PKIX_List_Create(&expectedCerts, plContext));
|
|
|
|
for (k = ++j; k < argc; k++) {
|
|
|
|
dirCert = createCert(dirName, argv[k], plContext);
|
|
|
|
if (k == (argc - 1)) {
|
|
PKIX_TEST_EXPECT_NO_ERROR(PKIX_PL_Object_IncRef
|
|
((PKIX_PL_Object *)dirCert, plContext));
|
|
trustedCert = dirCert;
|
|
} else {
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR
|
|
(PKIX_List_AppendItem
|
|
(expectedCerts,
|
|
(PKIX_PL_Object *)dirCert,
|
|
plContext));
|
|
|
|
if (k == j) {
|
|
PKIX_TEST_EXPECT_NO_ERROR(PKIX_PL_Object_IncRef
|
|
((PKIX_PL_Object *)dirCert, plContext));
|
|
targetCert = dirCert;
|
|
}
|
|
}
|
|
|
|
PKIX_TEST_DECREF_BC(dirCert);
|
|
}
|
|
|
|
/* create processing params with list of trust anchors */
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR(PKIX_TrustAnchor_CreateWithCert
|
|
(trustedCert, &anchor, plContext));
|
|
PKIX_TEST_EXPECT_NO_ERROR(PKIX_List_Create(&anchors, plContext));
|
|
PKIX_TEST_EXPECT_NO_ERROR
|
|
(PKIX_List_AppendItem
|
|
(anchors, (PKIX_PL_Object *)anchor, plContext));
|
|
PKIX_TEST_EXPECT_NO_ERROR(PKIX_ProcessingParams_Create
|
|
(anchors, &procParams, plContext));
|
|
|
|
/* create CertSelector with target certificate in params */
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR
|
|
(PKIX_ComCertSelParams_Create(&certSelParams, plContext));
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR
|
|
(PKIX_ComCertSelParams_SetCertificate
|
|
(certSelParams, targetCert, plContext));
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR
|
|
(PKIX_CertSelector_Create
|
|
(NULL, NULL, &certSelector, plContext));
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR(PKIX_CertSelector_SetCommonCertSelectorParams
|
|
(certSelector, certSelParams, plContext));
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR
|
|
(PKIX_ProcessingParams_SetTargetCertConstraints
|
|
(procParams, certSelector, plContext));
|
|
|
|
/* create CertStores */
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR(PKIX_PL_String_Create
|
|
(PKIX_ESCASCII,
|
|
dirName,
|
|
0,
|
|
&dirNameString,
|
|
plContext));
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR(PKIX_PL_CollectionCertStore_Create
|
|
(dirNameString, &certStore, plContext));
|
|
|
|
#if 0
|
|
PKIX_TEST_EXPECT_NO_ERROR(PKIX_PL_Pk11CertStore_Create
|
|
(&certStore, plContext));
|
|
#endif
|
|
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR(PKIX_List_Create(&certStores, plContext));
|
|
|
|
if (useLDAP == PKIX_TRUE) {
|
|
PKIX_TEST_EXPECT_NO_ERROR(createLdapCertStore
|
|
(serverName, timeout, &ldapCertStore, plContext));
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR
|
|
(PKIX_List_AppendItem
|
|
(certStores,
|
|
(PKIX_PL_Object *)ldapCertStore,
|
|
plContext));
|
|
}
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR
|
|
(PKIX_List_AppendItem
|
|
(certStores, (PKIX_PL_Object *)certStore, plContext));
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR(PKIX_ProcessingParams_SetCertStores
|
|
(procParams, certStores, plContext));
|
|
|
|
/* set resource limits */
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR(PKIX_ResourceLimits_Create
|
|
(&resourceLimits, plContext));
|
|
|
|
/* need longer time when running dbx for memory leak checking */
|
|
PKIX_TEST_EXPECT_NO_ERROR(PKIX_ResourceLimits_SetMaxTime
|
|
(resourceLimits, 60, plContext));
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR(PKIX_ResourceLimits_SetMaxFanout
|
|
(resourceLimits, 2, plContext));
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR(PKIX_ResourceLimits_SetMaxDepth
|
|
(resourceLimits, 2, plContext));
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR(PKIX_ProcessingParams_SetResourceLimits
|
|
(procParams, resourceLimits, plContext));
|
|
|
|
/* build cert chain using processing params and return buildResult */
|
|
|
|
subTest("Testing ResourceLimits MaxFanout & MaxDepth - <pass>");
|
|
Test_BuildResult
|
|
(procParams,
|
|
testValid,
|
|
expectedCerts,
|
|
plContext);
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR(PKIX_ResourceLimits_SetMaxFanout
|
|
(resourceLimits, 1, plContext));
|
|
|
|
subTest("Testing ResourceLimits MaxFanout - <fail>");
|
|
Test_BuildResult
|
|
(procParams,
|
|
PKIX_FALSE,
|
|
expectedCerts,
|
|
plContext);
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR(PKIX_ResourceLimits_SetMaxFanout
|
|
(resourceLimits, 2, plContext));
|
|
PKIX_TEST_EXPECT_NO_ERROR(PKIX_ResourceLimits_SetMaxDepth
|
|
(resourceLimits, 1, plContext));
|
|
|
|
subTest("Testing ResourceLimits MaxDepth - <fail>");
|
|
Test_BuildResult
|
|
(procParams,
|
|
PKIX_FALSE,
|
|
expectedCerts,
|
|
plContext);
|
|
|
|
PKIX_TEST_EXPECT_NO_ERROR(PKIX_ResourceLimits_SetMaxFanout
|
|
(resourceLimits, 0, plContext));
|
|
PKIX_TEST_EXPECT_NO_ERROR(PKIX_ResourceLimits_SetMaxDepth
|
|
(resourceLimits, 0, plContext));
|
|
PKIX_TEST_EXPECT_NO_ERROR(PKIX_ResourceLimits_SetMaxTime
|
|
(resourceLimits, 0, plContext));
|
|
|
|
subTest("Testing ResourceLimits No checking - <pass>");
|
|
Test_BuildResult
|
|
(procParams,
|
|
testValid,
|
|
expectedCerts,
|
|
plContext);
|
|
|
|
cleanup:
|
|
|
|
PKIX_TEST_DECREF_AC(expectedCerts);
|
|
PKIX_TEST_DECREF_AC(procParams);
|
|
PKIX_TEST_DECREF_AC(procParams);
|
|
PKIX_TEST_DECREF_AC(certStores);
|
|
PKIX_TEST_DECREF_AC(certStore);
|
|
PKIX_TEST_DECREF_AC(ldapCertStore);
|
|
PKIX_TEST_DECREF_AC(dirNameString);
|
|
PKIX_TEST_DECREF_AC(trustedCert);
|
|
PKIX_TEST_DECREF_AC(targetCert);
|
|
PKIX_TEST_DECREF_AC(anchors);
|
|
PKIX_TEST_DECREF_AC(anchor);
|
|
PKIX_TEST_DECREF_AC(certSelParams);
|
|
PKIX_TEST_DECREF_AC(certSelector);
|
|
PKIX_TEST_DECREF_AC(checker);
|
|
PKIX_TEST_DECREF_AC(resourceLimits);
|
|
|
|
PKIX_TEST_RETURN();
|
|
|
|
PKIX_Shutdown(plContext);
|
|
|
|
endTests("BuildChain_UserChecker");
|
|
|
|
return (0);
|
|
|
|
}
|