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
122
123
124
125
126
127
128
129
130
131
132
133
134
|
/*
* Label Manager tests.
* Copyright (C) 2020 Volta Networks
* Patrick Ruddy
*
* 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.
*
* 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.
*
* You should have received a copy of the GNU General Public License along
* with this program; see the file COPYING; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*/
#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);
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;
}
|
