Home | History | Annotate | Line # | Download | only in libntp
authkeys.c revision 1.1.1.3.2.2 
      1  1.1.1.3.2.2  snj /*	$NetBSD: authkeys.c,v 1.1.1.3.2.2 2015/11/07 22:26:46 snj Exp $	*/
      2  1.1.1.3.2.2  snj 
      3  1.1.1.3.2.2  snj /* This file contains test for both libntp/authkeys.c and libntp/authusekey.c */
      4  1.1.1.3.2.2  snj 
      5  1.1.1.3.2.2  snj #include "config.h"
      6  1.1.1.3.2.2  snj 
      7  1.1.1.3.2.2  snj #include "ntp.h"
      8  1.1.1.3.2.2  snj #include "ntp_stdlib.h"
      9  1.1.1.3.2.2  snj #include "ntp_calendar.h"
     10  1.1.1.3.2.2  snj 
     11  1.1.1.3.2.2  snj #include "unity.h"
     12  1.1.1.3.2.2  snj 
     13  1.1.1.3.2.2  snj #ifdef OPENSSL
     14  1.1.1.3.2.2  snj # include "openssl/err.h"
     15  1.1.1.3.2.2  snj # include "openssl/rand.h"
     16  1.1.1.3.2.2  snj # include "openssl/evp.h"
     17  1.1.1.3.2.2  snj #endif
     18  1.1.1.3.2.2  snj 
     19  1.1.1.3.2.2  snj u_long current_time = 4;
     20  1.1.1.3.2.2  snj int counter = 0;
     21  1.1.1.3.2.2  snj 
     22  1.1.1.3.2.2  snj void setUp(void);
     23  1.1.1.3.2.2  snj void tearDown(void);
     24  1.1.1.3.2.2  snj void AddTrustedKey(keyid_t keyno);
     25  1.1.1.3.2.2  snj void AddUntrustedKey(keyid_t keyno);
     26  1.1.1.3.2.2  snj void test_AddTrustedKeys(void);
     27  1.1.1.3.2.2  snj void test_AddUntrustedKey(void);
     28  1.1.1.3.2.2  snj void test_HaveKeyCorrect(void);
     29  1.1.1.3.2.2  snj void test_HaveKeyIncorrect(void);
     30  1.1.1.3.2.2  snj void test_AddWithAuthUseKey(void);
     31  1.1.1.3.2.2  snj void test_EmptyKey(void);
     32  1.1.1.3.2.2  snj 
     33  1.1.1.3.2.2  snj 
     34  1.1.1.3.2.2  snj void
     35  1.1.1.3.2.2  snj setUp(void)
     36  1.1.1.3.2.2  snj {
     37  1.1.1.3.2.2  snj 	if (counter == 0) {
     38  1.1.1.3.2.2  snj 		counter++;
     39  1.1.1.3.2.2  snj 		init_auth(); // causes segfault if called more than once
     40  1.1.1.3.2.2  snj 	}
     41  1.1.1.3.2.2  snj 	/*
     42  1.1.1.3.2.2  snj 	 * init_auth() is called by tests_main.cpp earlier.  It
     43  1.1.1.3.2.2  snj 	 * does not initialize global variables like
     44  1.1.1.3.2.2  snj 	 * authnumkeys, so let's reset them to zero here.
     45  1.1.1.3.2.2  snj 	 */
     46  1.1.1.3.2.2  snj 	authnumkeys = 0;
     47  1.1.1.3.2.2  snj 
     48  1.1.1.3.2.2  snj 	/*
     49  1.1.1.3.2.2  snj 	 * Especially, empty the key cache!
     50  1.1.1.3.2.2  snj 	 */
     51  1.1.1.3.2.2  snj 	cache_keyid = 0;
     52  1.1.1.3.2.2  snj 	cache_type = 0;
     53  1.1.1.3.2.2  snj 	cache_flags = 0;
     54  1.1.1.3.2.2  snj 	cache_secret = NULL;
     55  1.1.1.3.2.2  snj 	cache_secretsize = 0;
     56  1.1.1.3.2.2  snj }
     57  1.1.1.3.2.2  snj 
     58  1.1.1.3.2.2  snj void
     59  1.1.1.3.2.2  snj tearDown(void)
     60  1.1.1.3.2.2  snj {
     61  1.1.1.3.2.2  snj 
     62  1.1.1.3.2.2  snj }
     63  1.1.1.3.2.2  snj 
     64  1.1.1.3.2.2  snj static const int KEYTYPE = KEY_TYPE_MD5;
     65  1.1.1.3.2.2  snj 
     66  1.1.1.3.2.2  snj void
     67  1.1.1.3.2.2  snj AddTrustedKey(keyid_t keyno) {
     68  1.1.1.3.2.2  snj 	/*
     69  1.1.1.3.2.2  snj 	 * We need to add a MD5-key in addition to setting the
     70  1.1.1.3.2.2  snj 	 * trust, because authhavekey() requires type != 0.
     71  1.1.1.3.2.2  snj 	 */
     72  1.1.1.3.2.2  snj 	MD5auth_setkey(keyno, KEYTYPE, NULL, 0);
     73  1.1.1.3.2.2  snj 
     74  1.1.1.3.2.2  snj 	authtrust(keyno, TRUE);
     75  1.1.1.3.2.2  snj }
     76  1.1.1.3.2.2  snj 
     77  1.1.1.3.2.2  snj void
     78  1.1.1.3.2.2  snj AddUntrustedKey(keyid_t keyno) {
     79  1.1.1.3.2.2  snj 	authtrust(keyno, FALSE);
     80  1.1.1.3.2.2  snj }
     81  1.1.1.3.2.2  snj 
     82  1.1.1.3.2.2  snj void
     83  1.1.1.3.2.2  snj test_AddTrustedKeys(void) {
     84  1.1.1.3.2.2  snj 	const keyid_t KEYNO1 = 5;
     85  1.1.1.3.2.2  snj 	const keyid_t KEYNO2 = 8;
     86  1.1.1.3.2.2  snj 
     87  1.1.1.3.2.2  snj 	AddTrustedKey(KEYNO1);
     88  1.1.1.3.2.2  snj 	AddTrustedKey(KEYNO2);
     89  1.1.1.3.2.2  snj 
     90  1.1.1.3.2.2  snj 	TEST_ASSERT_TRUE(authistrusted(KEYNO1));
     91  1.1.1.3.2.2  snj 	TEST_ASSERT_TRUE(authistrusted(KEYNO2));
     92  1.1.1.3.2.2  snj }
     93  1.1.1.3.2.2  snj 
     94  1.1.1.3.2.2  snj void
     95  1.1.1.3.2.2  snj test_AddUntrustedKey(void) {
     96  1.1.1.3.2.2  snj 	const keyid_t KEYNO = 3;
     97  1.1.1.3.2.2  snj 
     98  1.1.1.3.2.2  snj 	AddUntrustedKey(KEYNO);
     99  1.1.1.3.2.2  snj 
    100  1.1.1.3.2.2  snj 	TEST_ASSERT_FALSE(authistrusted(KEYNO));
    101  1.1.1.3.2.2  snj }
    102  1.1.1.3.2.2  snj 
    103  1.1.1.3.2.2  snj void
    104  1.1.1.3.2.2  snj test_HaveKeyCorrect(void) {
    105  1.1.1.3.2.2  snj 	const keyid_t KEYNO = 3;
    106  1.1.1.3.2.2  snj 
    107  1.1.1.3.2.2  snj 	AddTrustedKey(KEYNO);
    108  1.1.1.3.2.2  snj 
    109  1.1.1.3.2.2  snj 	TEST_ASSERT_TRUE(auth_havekey(KEYNO));
    110  1.1.1.3.2.2  snj 	TEST_ASSERT_TRUE(authhavekey(KEYNO));
    111  1.1.1.3.2.2  snj }
    112  1.1.1.3.2.2  snj 
    113  1.1.1.3.2.2  snj void
    114  1.1.1.3.2.2  snj test_HaveKeyIncorrect(void) {
    115  1.1.1.3.2.2  snj 	const keyid_t KEYNO = 2;
    116  1.1.1.3.2.2  snj 
    117  1.1.1.3.2.2  snj 	TEST_ASSERT_FALSE(auth_havekey(KEYNO));
    118  1.1.1.3.2.2  snj 	TEST_ASSERT_FALSE(authhavekey(KEYNO));
    119  1.1.1.3.2.2  snj }
    120  1.1.1.3.2.2  snj 
    121  1.1.1.3.2.2  snj void
    122  1.1.1.3.2.2  snj test_AddWithAuthUseKey(void) {
    123  1.1.1.3.2.2  snj 	const keyid_t KEYNO = 5;
    124  1.1.1.3.2.2  snj 	const char* KEY = "52a";
    125  1.1.1.3.2.2  snj 
    126  1.1.1.3.2.2  snj 	TEST_ASSERT_TRUE(authusekey(KEYNO, KEYTYPE, (u_char*)KEY));
    127  1.1.1.3.2.2  snj }
    128  1.1.1.3.2.2  snj 
    129  1.1.1.3.2.2  snj void
    130  1.1.1.3.2.2  snj test_EmptyKey(void) {
    131  1.1.1.3.2.2  snj 	const keyid_t KEYNO = 3;
    132  1.1.1.3.2.2  snj 	const char* KEY = "";
    133  1.1.1.3.2.2  snj 
    134  1.1.1.3.2.2  snj 
    135  1.1.1.3.2.2  snj 	TEST_ASSERT_FALSE(authusekey(KEYNO, KEYTYPE, (u_char*)KEY));
    136  1.1.1.3.2.2  snj }
    137