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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
// SPDX-License-Identifier: GPL-2.0-or-later
/*
* Label Manager tests.
* Copyright (C) 2020 Volta Networks
* Patrick Ruddy
*/
#include <zebra.h>
#include "zebra/zapi_msg.h"
#include "zebra/label_manager.h"
/* shim out unused functions/variables to allow the lablemanager to compile*/
DEFINE_KOOH(zserv_client_close, (struct zserv * client), (client));
unsigned long zebra_debug_packet = 0;
struct zserv *zserv_find_client_session(uint8_t proto, unsigned short instance,
uint32_t session_id)
{
return NULL;
}
int zsend_label_manager_connect_response(struct zserv *client, vrf_id_t vrf_id,
unsigned short result)
{
return 0;
}
int zsend_assign_label_chunk_response(struct zserv *client, vrf_id_t vrf_id,
struct label_manager_chunk *lmc)
{
return 0;
}
static int test_client_connect(struct zserv *client, vrf_id_t vrf_id)
{
return 0;
}
static int test_client_disconnect(struct zserv *client)
{
return 0;
}
/* external test hook functions */
static int lm_get_chunk_pi(struct label_manager_chunk **lmc,
struct zserv *client, uint8_t keep, uint32_t size,
uint32_t base, vrf_id_t vrf_id)
{
if (base == 0)
*lmc = create_label_chunk(10, 55, 0, 1, 50, 50 + size, true);
else
*lmc = assign_label_chunk(10, 55, 0, 1, size, base);
return 0;
}
static int lm_release_chunk_pi(struct zserv *client, uint32_t start,
uint32_t end)
{
return release_label_chunk(client->proto, client->instance,
client->session_id, start, end);
}
/* use external allocations */
static void lp_plugin_init(void)
{
/* register our own hooks */
hook_register(lm_client_connect, test_client_connect);
hook_register(lm_client_disconnect, test_client_disconnect);
hook_register(lm_get_chunk, lm_get_chunk_pi);
hook_register(lm_release_chunk, lm_release_chunk_pi);
}
static void lp_plugin_cleanup(void)
{
/* register our own hooks */
hook_unregister(lm_client_connect, test_client_connect);
hook_unregister(lm_client_disconnect, test_client_disconnect);
hook_unregister(lm_get_chunk, lm_get_chunk_pi);
hook_unregister(lm_release_chunk, lm_release_chunk_pi);
}
/* tests */
static void test_lp_plugin(void)
{
struct label_manager_chunk *lmc;
lmc = assign_label_chunk(10, 55, 0, 1, 50, 0);
fprintf(stdout,
"chunk: start %u end %u proto %u instance %u session %u keep %s\n",
lmc->start, lmc->end, lmc->proto, lmc->instance,
lmc->session_id, lmc->keep ? "yes" : "no");
delete_label_chunk(lmc);
lmc = assign_label_chunk(10, 55, 0, 1, 50, 100);
fprintf(stdout,
"chunk: start %u end %u proto %u instance %u session %u keep %s\n",
lmc->start, lmc->end, lmc->proto, lmc->instance,
lmc->session_id, lmc->keep ? "yes" : "no");
release_label_chunk(10, 55, 0, lmc->start, lmc->end);
}
int main(int argc, char **argv)
{
/* set up label manager and release it's hooks */
label_manager_init();
lm_hooks_unregister();
/* test plugin */
lp_plugin_init();
test_lp_plugin();
lp_plugin_cleanup();
/* this keeps the compiler happy */
hook_call(zserv_client_close, NULL);
return 0;
}
|
