padding-00-mmcs.c   [plain text]

/*
 * Copyright (c) 2017 Apple Inc. All Rights Reserved.
 */

#include "testmore.h"

#include "SecPaddingConfigurationsPriv.h"
#include "shared_regressions.h"
#include "SecCFWrappers.h"

#define kTestTestCount 67

struct test_vector {
	SecPaddingType type;
	uint32_t originalSize;
	int64_t paddedSize;
};
typedef struct test_vector test_vector_t;

test_vector_t test_vectors[] = {
	{SecPaddingTypeMMCS,0,64},
	{SecPaddingTypeMMCS,1,64},
	{SecPaddingTypeMMCS,2,64},
	{SecPaddingTypeMMCS,64,64},
	{SecPaddingTypeMMCS,65,128},
	{SecPaddingTypeMMCS,68,128},
	{SecPaddingTypeMMCS,80,128},
	{SecPaddingTypeMMCS,90,128},
	{SecPaddingTypeMMCS,100,128},
	{SecPaddingTypeMMCS,128,128},
	{SecPaddingTypeMMCS,129,256},
	{SecPaddingTypeMMCS,129,256},
	{SecPaddingTypeMMCS,150,256},
	{SecPaddingTypeMMCS,256,256},
	{SecPaddingTypeMMCS,257,512},
	{SecPaddingTypeMMCS,512,512},
	{SecPaddingTypeMMCS,612,1024},
	{SecPaddingTypeMMCS,1000,1024},
	{SecPaddingTypeMMCS,1024,1024},
	{SecPaddingTypeMMCS,1025,2048},
	{SecPaddingTypeMMCS,1800,2048},
	{SecPaddingTypeMMCS,2000,2048},
	{SecPaddingTypeMMCS,2047,2048},
	{SecPaddingTypeMMCS,2048,2048},
	{SecPaddingTypeMMCS,3072,3072},
	{SecPaddingTypeMMCS,4096,4096},
	{SecPaddingTypeMMCS,31744,31744},
	{SecPaddingTypeMMCS,31999,32768},
	{SecPaddingTypeMMCS,32000,32768},
	{SecPaddingTypeMMCS,32001,32768},
	{SecPaddingTypeMMCS,32769,40960},
	{SecPaddingTypeMMCS,4294967295,4294967296}
};

static void tests_negative(void)
{
	CFErrorRef testError = NULL;

	// Sanity
	ok(SecPaddingCompute(SecPaddingTypeMMCS, 0, &testError) == 64 && testError==NULL);

	// Wrong type
	ok(SecPaddingCompute(100, 20, &testError) == -1);
	ok(testError!=NULL);
	CFReleaseNull(testError);
}

static void tests_positive(void)
{
	CFErrorRef testError = NULL;
	size_t n;

	n = sizeof(test_vectors)/sizeof(test_vectors[0]);
	for(size_t i=0; i<n; i++) {
		is((SecPaddingCompute(test_vectors[i].type, test_vectors[i].originalSize, &testError) + test_vectors[i].originalSize),
		   test_vectors[i].paddedSize, "Test # %zu, type %ld: input %u",i,(long)test_vectors[i].type,
		   test_vectors[i].originalSize);
		ok(testError==NULL);
	}
}

int padding_00_mmcs(int argc, char *const *argv)
{
	plan_tests(kTestTestCount);

	tests_positive();
	tests_negative();

	return 0;
}