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