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
|
/*
* Copyright (c) 2017 Cisco and/or its affiliates.
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at:
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include <vnet/ip/ip.h>
#include <vnet/plugin/plugin.h>
#include <vnet/udp/udp.h>
#include <ioam/ipfixcollector/ipfixcollector.h>
ipfix_collector_main_t ipfix_collector_main;
/**
* @brief IP-FIX SetID registration function.
*
* This function can be used by other VPP graph nodes to receive IP-FIX packets
* with a particular setid.
*
* @param vm Vlib main of the graph node which is interested in
* getting IP-Fix packet.
* @param info Structure describing the client node which
* is interested in getting the IP-Fix packets for
* a SetID.
*
* @returns 0 on success.
* @returns Error codes(<0) otherwise.
*/
int
ipfix_collector_reg_setid (vlib_main_t * vm, ipfix_client_add_del_t * info)
{
ipfix_collector_main_t *cm = &ipfix_collector_main;
uword *p = NULL;
int i;
ipfix_client *client = 0;
if ((!info) || (!info->client_name))
return IPFIX_COLLECTOR_ERR_INVALID_PARAM;
p = hash_get (cm->client_reg_table, info->ipfix_setid);
client = p ? pool_elt_at_index (cm->client_reg_pool, (*p)) : NULL;
if (info->del)
{
if (!client)
return 0; //There is no registered handler, so send success
hash_unset (cm->client_reg_table, info->ipfix_setid);
vec_free (client->client_name);
pool_put (cm->client_reg_pool, client);
return 0;
}
if (client)
return IPFIX_COLLECTOR_ERR_REG_EXISTS;
pool_get (cm->client_reg_pool, client);
i = client - cm->client_reg_pool;
client->client_name = vec_dup (info->client_name);
client->client_node = info->client_node;
client->client_next_node = vlib_node_add_next (vm,
ipfix_collector_node.index,
client->client_node);
client->set_id = info->ipfix_setid;
hash_set (cm->client_reg_table, info->ipfix_setid, i);
if (!udp_is_valid_dst_port (UDP_DST_PORT_ipfix, 1))
udp_register_dst_port (vm, UDP_DST_PORT_ipfix,
ipfix_collector_node.index, 1);
return 0;
}
static clib_error_t *
ipfix_collector_init (vlib_main_t * vm)
{
clib_error_t *error = 0;
ipfix_collector_main_t *cm = &ipfix_collector_main;
cm->vlib_main = vm;
cm->vnet_main = vnet_get_main ();
cm->client_reg_pool = NULL;
cm->client_reg_table = hash_create (0, sizeof (uword));
return error;
}
VLIB_INIT_FUNCTION (ipfix_collector_init);
/*
* fd.io coding-style-patch-verification: ON
*
* Local Variables:
* eval: (c-set-style "gnu")
* End:
*/
|
