lib,yang,zebra: add affinity-map support

Add the affinity-map global command to zebra. The syntax is:

> affinity-map NAME bit-position (0-1023)

Signed-off-by: Louis Scalbert <louis.scalbert@6wind.com>

1 files changed, 122 insertions, 0 deletions

diff --git a/lib/affinitymap.c b/lib/affinitymap.c
new file mode 100644
index 0000000000..62c9c0df8c
--- /dev/null
+++ b/lib/affinitymap.c
@@ -0,0 +1,122 @@
+/*
+ * Affinity map function.
+ *
+ * Copyright 2022 Hiroki Shirokura, LINE Corporation
+ * Copyright 2022 Masakazu Asama
+ * Copyright 2022 6WIND S.A.
+ *
+ * This file is part of Free Range Routing (FRR).
+ *
+ * FRR 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, or (at your option) any
+ * later version.
+ *
+ * FRR 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 "linklist.h"
+#include "memory.h"
+#include "command.h"
+#include "vector.h"
+#include "prefix.h"
+#include "vty.h"
+#include "affinitymap.h"
+#include "command.h"
+#include "log.h"
+#include "hash.h"
+#include "libfrr.h"
+#include "lib_errors.h"
+#include "table.h"
+#include "json.h"
+#include "jhash.h"
+
+DEFINE_MTYPE_STATIC(LIB, AFFINITY_MAP, "Affinity map");
+DEFINE_MTYPE(LIB, AFFINITY_MAP_NAME, "Affinity map name");
+DEFINE_MTYPE_STATIC(LIB, AFFINITY_MAP_INDEX, "Affinity map index");
+
+DEFINE_QOBJ_TYPE(affinity_maps);
+DEFINE_QOBJ_TYPE(affinity_map);
+
+struct affinity_maps affinity_map_master = {NULL};
+
+static void affinity_map_free(struct affinity_map *map)
+{
+	XFREE(MTYPE_AFFINITY_MAP, map);
+}
+
+void affinity_map_set(const char *name, int pos)
+{
+	struct listnode *node;
+	struct affinity_map *map;
+
+	if (!affinity_map_master.maps)
+		affinity_map_master.maps = list_new();
+
+	for (ALL_LIST_ELEMENTS_RO(affinity_map_master.maps, node, map)) {
+		if (strncmp(name, map->name, AFFINITY_NAME_SIZE) != 0)
+			continue;
+		map->bit_position = pos;
+		return;
+	}
+
+	map = XCALLOC(MTYPE_AFFINITY_MAP, sizeof(*map));
+	map->bit_position = pos;
+	snprintf(map->name, sizeof(map->name), "%s", name);
+	listnode_add(affinity_map_master.maps, map);
+}
+
+void affinity_map_unset(const char *name)
+{
+	struct listnode *node, *nnode;
+	struct affinity_map *map;
+
+	if (!affinity_map_master.maps)
+		return;
+
+	for (ALL_LIST_ELEMENTS(affinity_map_master.maps, node, nnode, map)) {
+		if (strncmp(name, map->name, AFFINITY_NAME_SIZE) != 0)
+			continue;
+		listnode_delete(affinity_map_master.maps, map);
+		affinity_map_free(map);
+		return;
+	}
+}
+
+struct affinity_map *affinity_map_get(const char *name)
+{
+	struct listnode *node;
+	struct affinity_map *map;
+
+	if (!affinity_map_master.maps)
+		return NULL;
+
+	for (ALL_LIST_ELEMENTS_RO(affinity_map_master.maps, node, map))
+		if (strncmp(name, map->name, AFFINITY_NAME_SIZE) == 0)
+			return map;
+	return NULL;
+}
+
+
+char *affinity_map_name_get(int pos)
+{
+	struct listnode *node;
+	struct affinity_map *map;
+
+	if (!affinity_map_master.maps)
+		return NULL;
+
+	for (ALL_LIST_ELEMENTS_RO(affinity_map_master.maps, node, map))
+		if (map->bit_position == pos)
+			return map->name;
+	return NULL;
+}