am f4692640: Add unit tests for netkeystore. (c7ceaf24) · Commits · e / os / platform_frameworks_native

cmds/keystore/keymgmt.c

+31 −7

Original line number	Diff line number	Diff line
		@@ -228,6 +228,11 @@ int remove_key(const char namespace, const char keyname)
		char keyfile[KEYFILE_LEN];

		if (state != UNLOCKED) return -state;
		if ((strlen(namespace) >= MAX_KEY_NAME_LENGTH) \|\|
		(strlen(keyname) >= MAX_KEY_NAME_LENGTH)) {
		LOGE("keyname is too long.");
		return -1;
		}
		sprintf(keyfile, KEYFILE_NAME, namespace, keyname);
		return unlink(keyfile);
		}
		@@ -243,12 +248,12 @@ int put_key(const char namespace, const char keyname,
		LOGE("Can not store key with current state %d\n", state);
		return -state;
		}
		sprintf(keyfile, KEYFILE_NAME, namespace, keyname);
		// flatten the args
		if (strlen(keyname) >= MAX_KEY_NAME_LENGTH) {
		if ((strlen(namespace) >= MAX_KEY_NAME_LENGTH) \|\|
		(strlen(keyname) >= MAX_KEY_NAME_LENGTH)) {
		LOGE("keyname is too long.");
		return -1;
		}
		sprintf(keyfile, KEYFILE_NAME, namespace, keyname);
		strcpy(blob.keyname, keyname);
		blob.value_size = size;
		if (size > MAX_KEY_VALUE_LENGTH) {
		@@ -271,6 +276,11 @@ int get_key(const char namespace, const char keyname,
		LOGE("Can not retrieve key value with current state %d\n", state);
		return -state;
		}
		if ((strlen(namespace) >= MAX_KEY_NAME_LENGTH) \|\|
		(strlen(keyname) >= MAX_KEY_NAME_LENGTH)) {
		LOGE("keyname is too long.");
		return -1;
		}
		sprintf(keyfile, KEYFILE_NAME, namespace, keyname);
		ret = load_n_decrypt(keyname, keyfile, &decryptKey, &blob);
		if (!ret) {
		@@ -299,6 +309,13 @@ int list_keys(const char *namespace, char reply[BUFFER_MAX])
		LOGE("cannot open keystore dir or namespace is null\n");
		return -1;
		}

		if (strlen(namespace) >= MAX_KEY_NAME_LENGTH) {
		LOGE("namespace is too long.");
		return -1;
		}

		reply[0] = 0;
		while ((de = readdir(d))) {
		char prefix, name, *keyfile = de->d_name;
		char *context = NULL;
		@@ -367,6 +384,7 @@ KEYSTORE_STATE get_state()

		int reset_keystore()
		{
		int ret = 0;
		DIR *d;
		struct dirent *de;

		@@ -374,18 +392,24 @@ int reset_keystore()
		LOGE("cannot open keystore dir\n");
		return -1;
		}
		while ((de = readdir(d))) unlink(de->d_name);
		while ((de = readdir(d))) {
		if (unlink(de->d_name) != 0) ret = -1;
		}
		closedir(d);
		state = UNINITIALIZED;
		if (ret == 0) {
		LOGI("keystore is reset.");
		return 0;
		} else {
		LOGI("keystore can not be cleaned up entirely.");
		}
		return ret;
		}

		int init_keystore(const char *dir)
		{
		int fd;

		if (!dir) mkdir(dir, 0770);
		if (dir) mkdir(dir, 0770);
		if (!dir \|\| chdir(dir)) {
		LOGE("Can not open/create the keystore directory %s\n",
		dir ? dir : "(null)");

cmds/keystore/tests/Android.mk

0 → 100644

+28 −0

Original line number	Diff line number	Diff line
		# Copyright (C) 2009 The Android Open Source Project
		#
		# Licensed under the Apache License, Version 2.0 (the "License");
		# you may not use this file except in compliance with the License.
		# You may obtain a copy of the License at
		#
		# http://www.apache.org/licenses/LICENSE-2.0
		#
		# Unless required by applicable law or agreed to in writing, software
		# distributed under the License is distributed on an "AS IS" BASIS,
		# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
		# See the License for the specific language governing permissions and
		# limitations under the License.
		#
		# define the KEYSTORE_TESTS environment variable to build the test programs
		ifdef KEYSTORE_TESTS
		LOCAL_PATH:= $(call my-dir)
		include $(CLEAR_VARS)
		LOCAL_SRC_FILES:= netkeystore_test.c ../keymgmt.c
		LOCAL_SHARED_LIBRARIES := libcutils libssl
		LOCAL_MODULE:= netkeystore_test
		LOCAL_MODULE_TAGS := optional
		LOCAL_C_INCLUDES := external/openssl/include \
		frameworks/base/cmds/keystore
		EXTRA_CFLAGS := -g -O0 -DGTEST_OS_LINUX -DGTEST_HAS_STD_STRING
		include $(BUILD_EXECUTABLE)

		endif #KEYSTORE_TESTS

cmds/keystore/tests/netkeystore_test.c

0 → 100644

+249 −0

Original line number	Diff line number	Diff line
		/*
		* Copyright (C) 2009 The Android Open Source Project
		* All rights reserved.
		*
		* Redistribution and use in source and binary forms, with or without
		* modification, are permitted provided that the following conditions
		* are met:
		* * Redistributions of source code must retain the above copyright
		* notice, this list of conditions and the following disclaimer.
		* * Redistributions in binary form must reproduce the above copyright
		* notice, this list of conditions and the following disclaimer in
		* the documentation and/or other materials provided with the
		* distribution.
		*
		* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
		* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
		* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
		* FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
		* COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
		* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
		* BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
		* OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
		* AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
		* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
		* OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
		* SUCH DAMAGE.
		*/

		#include <stdio.h>
		#include <stdlib.h>
		#include <unistd.h>

		#include "common.h"
		#include "keymgmt.h"

		typedef int FUNC_PTR();
		typedef struct {
		const char *name;
		FUNC_PTR *func;
		} TESTFUNC;

		#define FUNC_NAME(x) { #x, test_##x }
		#define FUNC_BODY(x) int test_##x()

		#define TEST_PASSWD "12345678"
		#define TEST_NPASSWD "11111111"
		#define TEST_DIR "/data/local/tmp/keystore"
		#define READONLY_DIR "/proc/keystore"
		#define TEST_NAMESPACE "test"
		#define TEST_KEYNAME "key"
		#define TEST_KEYNAME2 "key2"
		#define TEST_KEYVALUE "ANDROID"

		void setup()
		{
		if (init_keystore(TEST_DIR) != 0) {
		fprintf(stderr, "Can not create the test directory %s\n", TEST_DIR);
		exit(-1);
		}
		}

		void teardown()
		{
		reset_keystore();
		rmdir(TEST_DIR);
		}

		FUNC_BODY(init_keystore)
		{
		if (init_keystore(READONLY_DIR) == 0) return -1;

		return EXIT_SUCCESS;
		}

		FUNC_BODY(reset_keystore)
		{
		chdir("/procx");
		if (reset_keystore() == 0) return -1;
		chdir(TEST_DIR);
		return EXIT_SUCCESS;
		}

		FUNC_BODY(get_state)
		{
		if (get_state() != UNINITIALIZED) return -1;
		passwd(TEST_PASSWD);
		if (get_state() != UNLOCKED) return -1;
		lock();
		if (get_state() != LOCKED) return -1;
		reset_keystore();
		if (get_state() != UNINITIALIZED) return -1;
		return EXIT_SUCCESS;
		}

		FUNC_BODY(passwd)
		{
		char buf[512];

		if (passwd(" 23432dsfsdf") == 0) return -1;
		if (passwd("dsfsdf") == 0) return -1;
		passwd(TEST_PASSWD);
		lock();
		if (unlock("55555555") == 0) return -1;
		if (unlock(TEST_PASSWD) != 0) return -1;

		// change the password
		sprintf(buf, "%s %s", "klfdjdsklfjg", "abcdefghi");
		if (passwd(buf) == 0) return -1;

		sprintf(buf, "%s %s", TEST_PASSWD, TEST_NPASSWD);
		if (passwd(buf) != 0) return -1;
		lock();

		if (unlock(TEST_PASSWD) == 0) return -1;
		if (unlock(TEST_NPASSWD) != 0) return -1;

		return EXIT_SUCCESS;
		}

		FUNC_BODY(lock)
		{
		if (lock() == 0) return -1;
		passwd(TEST_PASSWD);
		if (lock() != 0) return -1;
		if (lock() != 0) return -1;
		return EXIT_SUCCESS;
		}

		FUNC_BODY(unlock)
		{
		int i = MAX_RETRY_COUNT;
		passwd(TEST_PASSWD);
		lock();
		while (i > 1) {
		if (unlock(TEST_NPASSWD) != --i) return -1;
		}
		if (unlock(TEST_NPASSWD) != -1) return -1;
		return EXIT_SUCCESS;
		}

		FUNC_BODY(put_key)
		{
		int i = 0;
		char keyname[512];

		if (put_key(TEST_NAMESPACE, TEST_KEYNAME, (unsigned char *)TEST_KEYVALUE,
		strlen(TEST_KEYVALUE)) == 0) return -1;
		passwd(TEST_PASSWD);
		if (put_key(TEST_NAMESPACE, TEST_KEYNAME, (unsigned char *)TEST_KEYVALUE,
		strlen(TEST_KEYVALUE)) != 0) return -1;

		for(i = 0; i < 500; i++) keyname[i] = 'K';
		keyname[i] = 0;
		if (put_key(TEST_NAMESPACE, keyname, (unsigned char *)TEST_KEYVALUE,
		strlen(TEST_KEYVALUE)) == 0) return -1;
		if (put_key(TEST_NAMESPACE, TEST_KEYNAME, (unsigned char *)TEST_KEYVALUE,
		MAX_KEY_VALUE_LENGTH + 1) == 0) return -1;
		return EXIT_SUCCESS;
		}

		FUNC_BODY(get_key)
		{
		int size;
		unsigned char data[MAX_KEY_VALUE_LENGTH];

		if (get_key(TEST_NAMESPACE, TEST_KEYNAME, data, &size) == 0) return -1;

		passwd(TEST_PASSWD);
		put_key(TEST_NAMESPACE, TEST_KEYNAME, (unsigned char *)TEST_KEYVALUE,
		strlen(TEST_KEYVALUE));
		if (get_key(TEST_NAMESPACE, TEST_KEYNAME, data, &size) != 0) return -1;
		if (memcmp(data, TEST_KEYVALUE, size) != 0) return -1;

		return EXIT_SUCCESS;
		}

		FUNC_BODY(remove_key)
		{
		if (remove_key(TEST_NAMESPACE, TEST_KEYNAME) == 0) return -1;

		passwd(TEST_PASSWD);
		if (remove_key(TEST_NAMESPACE, TEST_KEYNAME) == 0) return -1;

		put_key(TEST_NAMESPACE, TEST_KEYNAME, (unsigned char *)TEST_KEYVALUE,
		strlen(TEST_KEYVALUE));
		if (remove_key(TEST_NAMESPACE, TEST_KEYNAME) != 0) return -1;

		return EXIT_SUCCESS;
		}

		FUNC_BODY(list_keys)
		{
		int i;
		char buf[128];
		char reply[BUFFER_MAX];

		for(i = 0; i < 100; i++) buf[i] = 'K';
		buf[i] = 0;

		if (list_keys(TEST_NAMESPACE, reply) == 0) return -1;

		passwd(TEST_PASSWD);
		if (list_keys(buf, reply) == 0) return -1;

		if (list_keys(TEST_NAMESPACE, reply) != 0) return -1;
		if (strcmp(reply, "") != 0) return -1;

		put_key(TEST_NAMESPACE, TEST_KEYNAME, (unsigned char *)TEST_KEYVALUE,
		strlen(TEST_KEYVALUE));
		if (list_keys(TEST_NAMESPACE, reply) != 0) return -1;
		if (strcmp(reply, TEST_KEYNAME) != 0) return -1;

		put_key(TEST_NAMESPACE, TEST_KEYNAME2, (unsigned char *)TEST_KEYVALUE,
		strlen(TEST_KEYVALUE));

		if (list_keys(TEST_NAMESPACE, reply) != 0) return -1;
		sprintf(buf, "%s %s", TEST_KEYNAME2, TEST_KEYNAME);
		if (strcmp(reply, buf) != 0) return -1;

		return EXIT_SUCCESS;
		}

		TESTFUNC all_tests[] = {
		FUNC_NAME(init_keystore),
		FUNC_NAME(reset_keystore),
		FUNC_NAME(get_state),
		FUNC_NAME(passwd),
		FUNC_NAME(lock),
		FUNC_NAME(unlock),
		FUNC_NAME(put_key),
		FUNC_NAME(get_key),
		FUNC_NAME(remove_key),
		FUNC_NAME(list_keys),
		};

		int main(int argc, char **argv) {
		int i, ret;
		for (i = 0 ; i < (int)(sizeof(all_tests)/sizeof(TESTFUNC)) ; ++i) {
		setup();
		if ((ret = all_tests[i].func()) != EXIT_SUCCESS) {
		fprintf(stderr, "ERROR in function %s\n", all_tests[i].name);
		return ret;
		} else {
		fprintf(stderr, "function %s PASSED!\n", all_tests[i].name);
		}
		teardown();
		}
		return EXIT_SUCCESS;
		}