/*
* Copyright (c) 2015 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.
*/
/*
*------------------------------------------------------------------
* srv6_localsid_sample.c - Simple SRv6 LocalSID
*------------------------------------------------------------------
*/
#include <vnet/vnet.h>
#include <vnet/plugin/plugin.h>
#include <srv6-localsid/srv6_localsid_sample.h>
#include <vlibapi/api.h>
#include <vlibmemory/api.h>
unsigned char srv6_localsid_name[32] = "Sample-SRv6-LocalSID-plugin";
unsigned char keyword_str[32] = "new_srv6_localsid";
unsigned char def_str[64] = "This is a definition of a sample new_srv6_localsid";
unsigned char params_str[32] = "<fib_table>";
/*****************************************/
/* SRv6 LocalSID instantiation and removal functions */
static int
srv6_localsid_creation_fn (ip6_sr_localsid_t *localsid)
{
/*
* Do you want to do anything fancy upon localsid instantiation?
* You can do it here
* (If return != 0 the localsid creation will be cancelled.)
*/
/* As an example Im going to do a +1 to the fib table inserted by the user */
srv6_localsid_sample_per_sid_memory_t *ls_mem = localsid->plugin_mem;
ls_mem->fib_table += 1;
return 0;
}
static int
srv6_localsid_removal_fn (ip6_sr_localsid_t *localsid)
{
/* Do you want to do anything fancy upon localsid removal?
* You can do it here
* (If return != 0 the localsid removal will be cancelled.)
*/
/*
* BTW if you stored something in localsid->plugin_mem you should clean it now
*/
//In this example we are only cleaning the memory allocated per localsid
clib_mem_free(localsid->plugin_mem);
return 0;
}
/**********************************/
/* SRv6 LocalSID format functions */
/*
* Prints nicely the parameters of a localsid
* Example: print "Table 5"
*/
u8 *
format_srv6_localsid_sample (u8 * s, va_list * args)
{
srv6_localsid_sample_per_sid_memory_t *ls_mem = va_arg (*args, void *);
return (format (s, "Table: %u", ls_mem->fib_table));
}
/*
* Process the parameters of a localsid
* Example: process from:
* sr localsid address cafe::1 behavior new_srv6_localsid 5
* everything from behavior on... so in this case 'new_srv6_localsid 5'
* Notice that it MUST match the keyword_str and params_str defined above.
*/
uword
unformat_srv6_localsid_sample (unformat_input_t * input, va_list * args)
{
void **plugin_mem = va_arg (*args, void **);
srv6_localsid_sample_per_sid_memory_t *ls_mem;
u32 table_id;
if (unformat (input, "new_srv6_localsid %u", &table_id))
{
/* Allocate a portion of memory */
ls_mem = clib_mem_alloc_aligned_at_offset (
sizeof(srv6_localsid_sample_per_sid_memory_t), 0, 0, 1);
/* Set to zero the memory */
memset (ls_mem, 0, sizeof(srv6_localsid_sample_per_sid_memory_t));
/* Our brand-new car is ready */
ls_mem->fib_table = table_id;
/* Dont forget to add it to the localsid */
*plugin_mem = ls_mem;
return 1;
}
return 0;
}
/*************************/
/* SRv6 LocalSID FIB DPO */
static u8 *
format_srv6_localsid_sample_dpo (u8 * s, va_list * args)
{
index_t index = va_arg (*args, index_t);
CLIB_UNUSED (u32 indent) = va_arg (*args, u32);
return (format (s, "SR: localsid_sample_index:[%u]", index));
}
void
srv6_localsid_sample_dpo_lock (dpo_id_t * dpo)
{
}
void
srv6_localsid_sample_dpo_unlock (dpo_id_t * dpo)
{
}
const static dpo_vft_t srv6_localsid_sample_vft = {
.dv_lock = srv6_localsid_sample_dpo_lock,
.dv_unlock = srv6_localsid_sample_dpo_unlock,
.dv_format = format_srv6_localsid_sample_dpo,
};
const static char *const srv6_localsid_sample_ip6_nodes[] = {
"srv6-localsid-sample",
NULL,
};
const static char *const *const srv6_localsid_sample_nodes[DPO_PROTO_NUM] = {
[DPO_PROTO_IP6] = srv6_localsid_sample_ip6_nodes,
};
/**********************/
static clib_error_t * srv6_localsid_sample_init (vlib_main_t * vm)
{
srv6_localsid_sample_main_t * sm = &srv6_localsid_sample_main;
int rv = 0;
/* Create DPO */
sm->srv6_localsid_sample_dpo_type = dpo_register_new_type (
&srv6_localsid_sample_vft, srv6_localsid_sample_nodes);
/* Register SRv6 LocalSID */
rv = sr_localsid_register_function (vm,
srv6_localsid_name,
keyword_str,
def_str,
params_str,
&sm->srv6_localsid_sample_dpo_type,
format_srv6_localsid_sample,
unformat_srv6_localsid_sample,
srv6_localsid_creation_fn,
srv6_localsid_removal_fn);
if (rv < 0)
clib_error_return (0, "SRv6 LocalSID function could not be registered.");
else
sm->srv6_localsid_behavior_id = rv;
return 0;
}
VLIB_INIT_FUNCTION (srv6_localsid_sample_init);
VLIB_PLUGIN_REGISTER () = {
.version = "1.0",
};
