186 lines
4.5 KiB
C
186 lines
4.5 KiB
C
/**
|
|
* @file ike_sa_manager_test.c
|
|
*
|
|
* @brief Tests for the ike_sa_manager_t class.
|
|
*
|
|
*/
|
|
|
|
/*
|
|
* Copyright (C) 2005 Jan Hutter, Martin Willi
|
|
* Hochschule fuer Technik Rapperswil
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify it
|
|
* under the terms of the GNU General Public License as published by the
|
|
* Free Software Foundation; either version 2 of the License, or (at your
|
|
* option) any later version. See <http://www.fsf.org/copyleft/gpl.txt>.
|
|
*
|
|
* This program is distributed in the hope that it will be useful, but
|
|
* WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
|
|
* or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
|
* for more details.
|
|
*/
|
|
|
|
#include <string.h>
|
|
#include <pthread.h>
|
|
#include <unistd.h>
|
|
|
|
#include "ike_sa_manager_test.h"
|
|
|
|
#include <types.h>
|
|
#include <sa/ike_sa_manager.h>
|
|
|
|
|
|
static struct ike_sa_manager_test_struct_s {
|
|
protected_tester_t *tester;
|
|
ike_sa_manager_t *isam;
|
|
} td;
|
|
|
|
static void test1_thread(ike_sa_id_t *ike_sa_id)
|
|
{
|
|
ike_sa_t *ike_sa;
|
|
status_t status;
|
|
|
|
status = td.isam->checkout(td.isam, ike_sa_id, &ike_sa);
|
|
td.tester->assert_true(td.tester, (status == SUCCESS), "checkout of a blocked ike_sa");
|
|
usleep(10000);
|
|
status = td.isam->checkin(td.isam, ike_sa);
|
|
td.tester->assert_true(td.tester, (status == SUCCESS), "checkin of a requested ike_sa");
|
|
}
|
|
|
|
|
|
static void test3_thread(ike_sa_id_t *ike_sa_id)
|
|
{
|
|
ike_sa_t *ike_sa;
|
|
status_t status;
|
|
|
|
status = td.isam->checkout(td.isam, ike_sa_id, &ike_sa);
|
|
td.tester->assert_true(td.tester, (status == NOT_FOUND), "IKE_SA already deleted");
|
|
}
|
|
|
|
|
|
|
|
|
|
void test_ike_sa_manager(protected_tester_t *tester)
|
|
{
|
|
status_t status;
|
|
u_int64_t initiator, responder;
|
|
ike_sa_id_t *ike_sa_id, *sa_id;
|
|
ike_sa_t *ike_sa;
|
|
int thread_count = 200;
|
|
int sa_count = 100;
|
|
int i;
|
|
pthread_t threads[thread_count];
|
|
|
|
td.tester = tester;
|
|
td.isam = ike_sa_manager_create();
|
|
tester->assert_true(tester, (td.isam != NULL), "ike_sa_manager creation");
|
|
|
|
|
|
|
|
|
|
/* First Test:
|
|
* we play initiator for IKE_SA_INIT first
|
|
* create an IKE_SA,
|
|
*
|
|
*/
|
|
|
|
td.isam->create_and_checkout(td.isam, &ike_sa);
|
|
/* for testing purposes, we manipulate the responder spi.
|
|
* this is usually done be the response from the communication partner,
|
|
* but we don't have one...
|
|
*/
|
|
responder = 123;
|
|
|
|
sa_id = ike_sa->get_id(ike_sa);
|
|
sa_id->set_responder_spi(sa_id, responder);
|
|
|
|
ike_sa_id = sa_id->clone(sa_id);
|
|
|
|
/* check in, so we should have a "completed" sa, specified by ike_sa_id */
|
|
status = td.isam->checkin(td.isam, ike_sa);
|
|
tester->assert_true(tester, (status == SUCCESS), "checkin modified IKE_SA");
|
|
|
|
/* now we check it out and start some other threads */
|
|
status = td.isam->checkout(td.isam, ike_sa_id, &ike_sa);
|
|
tester->assert_true(tester, (status == SUCCESS), "checkout existing IKE_SA 1");
|
|
|
|
for (i = 0; i < thread_count; i++)
|
|
{
|
|
if (pthread_create(&threads[i], NULL, (void*(*)(void*))test1_thread, (void*)ike_sa_id))
|
|
{
|
|
/* failed, decrease list */
|
|
thread_count--;
|
|
i--;
|
|
}
|
|
}
|
|
sleep(1);
|
|
|
|
|
|
status = td.isam->checkin(td.isam, ike_sa);
|
|
tester->assert_true(tester, (status == SUCCESS), "checkin IKE_SA");
|
|
|
|
|
|
sleep(1);
|
|
/* we now delete the IKE_SA, while it is requested by the threads.
|
|
* this should block until the have done their work.*/
|
|
status = td.isam->delete(td.isam, ike_sa_id);
|
|
tester->assert_true(tester, (status == SUCCESS), "delete IKE_SA by id");
|
|
|
|
|
|
for (i = 0; i < thread_count; i++)
|
|
{
|
|
pthread_join(threads[i], NULL);
|
|
}
|
|
|
|
ike_sa_id->destroy(ike_sa_id);
|
|
|
|
|
|
/* Second Test:
|
|
* now we simulate our partner initiates an IKE_SA_INIT,
|
|
* so we are the responder.
|
|
*
|
|
*/
|
|
memset(&initiator, 0, sizeof(initiator));
|
|
memset(&responder, 0, sizeof(responder));
|
|
|
|
initiator = 123;
|
|
ike_sa_id = ike_sa_id_create(initiator, responder, TRUE);
|
|
|
|
status = td.isam->checkout(td.isam, ike_sa_id, &ike_sa);
|
|
tester->assert_false(tester, (status == SUCCESS), "checkout unexisting IKE_SA 2");
|
|
|
|
/* let them go acquiring */
|
|
sleep(1);
|
|
|
|
|
|
ike_sa_id->destroy(ike_sa_id);
|
|
|
|
/* Third Test:
|
|
* put in a lot of IKE_SAs, check it out, set a thread waiting
|
|
* and destroy the manager...
|
|
*/
|
|
thread_count = sa_count;
|
|
|
|
for (i = 0; i < sa_count; i++)
|
|
{
|
|
td.isam->create_and_checkout(td.isam, &ike_sa);
|
|
|
|
if (pthread_create(&threads[i], NULL, (void*(*)(void*))test3_thread, (void*)ike_sa->get_id(ike_sa)))
|
|
{
|
|
/* failed, decrease list */
|
|
thread_count--;
|
|
}
|
|
}
|
|
|
|
/* let them go acquiring */
|
|
sleep(1);
|
|
|
|
td.isam->destroy(td.isam);
|
|
|
|
for (i = 0; i < thread_count; i++)
|
|
{
|
|
pthread_join(threads[i], NULL);
|
|
}
|
|
}
|
|
|