87 lines
3.1 KiB
C
87 lines
3.1 KiB
C
#define _GNU_SOURCE
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
#define ftl(...) do { fprintf( stderr, __VA_ARGS__ ); exit(2); } while (0)
|
|
|
|
#include "mpw-algorithm.h"
|
|
#include "mpw-util.h"
|
|
|
|
#include "mpw-tests-util.h"
|
|
|
|
int main(int argc, char *const argv[]) {
|
|
|
|
int failedTests = 0;
|
|
|
|
xmlNodePtr tests = xmlDocGetRootElement( xmlParseFile( "mpw_tests.xml" ) );
|
|
if (!tests) {
|
|
ftl( "Couldn't find test case: mpw_tests.xml\n" );
|
|
abort();
|
|
}
|
|
|
|
for (xmlNodePtr testCase = tests->children; testCase; testCase = testCase->next) {
|
|
if (testCase->type != XML_ELEMENT_NODE || xmlStrcmp( testCase->name, BAD_CAST "case" ) != 0)
|
|
continue;
|
|
|
|
// Read in the test case.
|
|
xmlChar *id = mpw_xmlTestCaseString( testCase, "id" );
|
|
uint32_t algorithm = mpw_xmlTestCaseInteger( testCase, "algorithm" );
|
|
xmlChar *fullName = mpw_xmlTestCaseString( testCase, "fullName" );
|
|
xmlChar *masterPassword = mpw_xmlTestCaseString( testCase, "masterPassword" );
|
|
xmlChar *keyID = mpw_xmlTestCaseString( testCase, "keyID" );
|
|
xmlChar *siteName = mpw_xmlTestCaseString( testCase, "siteName" );
|
|
uint32_t siteCounter = mpw_xmlTestCaseInteger( testCase, "siteCounter" );
|
|
xmlChar *siteTypeString = mpw_xmlTestCaseString( testCase, "siteType" );
|
|
xmlChar *siteVariantString = mpw_xmlTestCaseString( testCase, "siteVariant" );
|
|
xmlChar *siteContext = mpw_xmlTestCaseString( testCase, "siteContext" );
|
|
xmlChar *result = mpw_xmlTestCaseString( testCase, "result" );
|
|
|
|
MPSiteType siteType = mpw_typeWithName( (char *)siteTypeString );
|
|
MPSiteVariant siteVariant = mpw_variantWithName( (char *)siteVariantString );
|
|
|
|
// Run the test case.
|
|
fprintf( stdout, "test case %s... ", id );
|
|
if (!xmlStrlen( result )) {
|
|
fprintf( stdout, "abstract.\n" );
|
|
continue;
|
|
}
|
|
|
|
// 1. calculate the master key.
|
|
const uint8_t *masterKey = mpw_masterKeyForUser(
|
|
(char *)fullName, (char *)masterPassword, algorithm );
|
|
if (!masterKey)
|
|
ftl( "Couldn't derive master key." );
|
|
|
|
// 2. calculate the site password.
|
|
const char *sitePassword = mpw_passwordForSite(
|
|
masterKey, (char *)siteName, siteType, siteCounter, siteVariant, (char *)siteContext, algorithm );
|
|
mpw_free( masterKey, MP_dkLen );
|
|
if (!sitePassword)
|
|
ftl( "Couldn't derive site password." );
|
|
|
|
// Check the result.
|
|
if (xmlStrcmp( result, BAD_CAST sitePassword ) == 0)
|
|
fprintf( stdout, "pass.\n" );
|
|
|
|
else {
|
|
++failedTests;
|
|
fprintf( stdout, "FAILED! (got %s != expected %s)\n", sitePassword, result );
|
|
}
|
|
|
|
// Free test case.
|
|
mpw_free_string( sitePassword );
|
|
xmlFree( id );
|
|
xmlFree( fullName );
|
|
xmlFree( masterPassword );
|
|
xmlFree( keyID );
|
|
xmlFree( siteName );
|
|
xmlFree( siteTypeString );
|
|
xmlFree( siteVariantString );
|
|
xmlFree( siteContext );
|
|
xmlFree( result );
|
|
}
|
|
|
|
return failedTests;
|
|
}
|