1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
|
/* This file contains test for both libntp/authkeys.c and libntp/authusekey.c */
#include "libntptest.h"
extern "C" {
#ifdef OPENSSL
# include "openssl/err.h"
# include "openssl/rand.h"
# include "openssl/evp.h"
#endif
#include "ntp.h"
#include "ntp_stdlib.h"
};
class authkeysTest : public libntptest {
protected:
static const int KEYTYPE = KEY_TYPE_MD5;
virtual void SetUp() {
/*
* init_auth() is called by tests_main.cpp earlier. It
* does not initialize global variables like
* authnumkeys, so let's reset them to zero here.
*/
authnumkeys = 0;
/*
* Especially, empty the key cache!
*/
cache_keyid = 0;
cache_type = 0;
cache_flags = 0;
cache_secret = NULL;
cache_secretsize = 0;
}
void AddTrustedKey(keyid_t keyno) {
/*
* We need to add a MD5-key in addition to setting the
* trust, because authhavekey() requires type != 0.
*/
MD5auth_setkey(keyno, KEYTYPE, NULL, 0);
authtrust(keyno, TRUE);
}
void AddUntrustedKey(keyid_t keyno) {
authtrust(keyno, FALSE);
}
};
TEST_F(authkeysTest, AddTrustedKeys) {
const keyid_t KEYNO1 = 5;
const keyid_t KEYNO2 = 8;
AddTrustedKey(KEYNO1);
AddTrustedKey(KEYNO2);
EXPECT_TRUE(authistrusted(KEYNO1));
EXPECT_TRUE(authistrusted(KEYNO2));
}
TEST_F(authkeysTest, AddUntrustedKey) {
const keyid_t KEYNO = 3;
AddUntrustedKey(KEYNO);
EXPECT_FALSE(authistrusted(KEYNO));
}
TEST_F(authkeysTest, HaveKeyCorrect) {
const keyid_t KEYNO = 3;
AddTrustedKey(KEYNO);
EXPECT_TRUE(auth_havekey(KEYNO));
EXPECT_TRUE(authhavekey(KEYNO));
}
TEST_F(authkeysTest, HaveKeyIncorrect) {
const keyid_t KEYNO = 2;
EXPECT_FALSE(auth_havekey(KEYNO));
EXPECT_FALSE(authhavekey(KEYNO));
}
TEST_F(authkeysTest, AddWithAuthUseKey) {
const keyid_t KEYNO = 5;
const char* KEY = "52a";
EXPECT_TRUE(authusekey(KEYNO, KEYTYPE, (u_char*)KEY));
}
TEST_F(authkeysTest, EmptyKey) {
const keyid_t KEYNO = 3;
const char* KEY = "";
EXPECT_FALSE(authusekey(KEYNO, KEYTYPE, (u_char*)KEY));
}
|
