1 files changed, 71 insertions, 0 deletions
diff --git a/src/lib/class.c b/src/lib/class.c
new file mode 100644
index 0000000..93346d5
--- /dev/null
+++ b/src/lib/class.c
@@ -0,0 +1,71 @@
+/*
+ * src/lib/class.c     CLI Class Helpers
+ *
+ *	This library is free software; you can redistribute it and/or
+ *	modify it under the terms of the GNU Lesser General Public
+ *	License as published by the Free Software Foundation version 2.1
+ *	of the License.
+ *
+ * Copyright (c) 2010 Thomas Graf <tgraf@suug.ch>
+ */
+
+/**
+ * @ingroup cli
+ * @defgroup cli_class Traffic Classes
+ * @{
+ */
+
+#include <netlink/cli/utils.h>
+#include <netlink/cli/class.h>
+
+struct rtnl_class *nl_cli_class_alloc(void)
+{
+	struct rtnl_class *class;
+
+	class = rtnl_class_alloc();
+	if (!class)
+		nl_cli_fatal(ENOMEM, "Unable to allocate class object");
+
+	return class;
+}
+
+void nl_cli_class_parse_dev(struct rtnl_class *class, struct nl_cache *link_cache, char *arg)
+{
+	int ival;
+
+	if (!(ival = rtnl_link_name2i(link_cache, arg)))
+		nl_cli_fatal(ENOENT, "Link \"%s\" does not exist", arg);
+
+	rtnl_class_set_ifindex(class, ival);
+}
+
+void nl_cli_class_parse_parent(struct rtnl_class *class, char *arg)
+{
+	uint32_t parent;
+	int err;
+
+	if ((err = rtnl_tc_str2handle(arg, &parent)) < 0)
+		nl_cli_fatal(err, "Unable to parse handle \"%s\": %s",
+		      arg, nl_geterror(err));
+
+	rtnl_class_set_parent(class, parent);
+}
+
+void nl_cli_class_parse_handle(struct rtnl_class *class, char *arg)
+{
+	uint32_t handle;
+	int err;
+
+	if ((err = rtnl_tc_str2handle(arg, &handle)) < 0)
+		nl_cli_fatal(err, "Unable to parse classid \"%s\": %s",
+		      arg, nl_geterror(err));
+
+	rtnl_class_set_handle(class, handle);
+}
+
+void nl_cli_class_parse_kind(struct rtnl_class *class, char *arg)
+{
+	rtnl_class_set_kind(class, arg);
+}
+
+/** @} */