1 files changed, 129 insertions, 0 deletions
diff --git a/src/nl-class-add.c b/src/nl-class-add.c
new file mode 100644
index 0000000..6e1c402
--- /dev/null
+++ b/src/nl-class-add.c
@@ -0,0 +1,129 @@
+/*
+ * src/nl-class-add.c     Add/Update/Replace Traffic Class
+ *
+ *	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>
+ */
+
+#include <netlink/cli/utils.h>
+#include <netlink/cli/qdisc.h>
+#include <netlink/cli/class.h>
+#include <netlink/cli/link.h>
+
+static int quiet = 0;
+
+static void print_usage(void)
+{
+	printf(
+"Usage: nl-class-add [OPTIONS]... class [CONFIGURATION]...\n"
+"\n"
+"OPTIONS\n"
+" -q, --quiet               Do not print informal notifications.\n"
+" -h, --help                Show this help text.\n"
+" -v, --version             Show versioning information.\n"
+"     --update              Update class if it exists.\n"
+"     --update-only         Only update class, never create it.\n"
+" -d, --dev=DEV             Network device the class should be attached to.\n"
+" -i, --id=ID               ID of new class (default: auto-generated)r\n"
+" -p, --parent=ID           ID of parent { root | ingress | class-ID }\n"
+"\n"
+"CONFIGURATION\n"
+" -h, --help                Show help text of class specific options.\n"
+"\n"
+"EXAMPLE\n"
+"   $ nl-class-add --dev=eth1 --parent=root htb --rate=100mbit\n"
+"\n"
+	);
+	exit(0);
+}
+
+int main(int argc, char *argv[])
+{
+	struct nl_sock *sock;
+	struct rtnl_class *class;
+	struct nl_cache *link_cache;
+	struct nl_dump_params dp = {
+		.dp_type = NL_DUMP_DETAILS,
+		.dp_fd = stdout,
+	};
+	struct nl_cli_qdisc_module *qm;
+	struct rtnl_class_ops *ops;
+	int err, flags = NLM_F_CREATE | NLM_F_EXCL;
+	char *kind;
+ 
+	sock = nl_cli_alloc_socket();
+	nl_cli_connect(sock, NETLINK_ROUTE);
+
+	link_cache = nl_cli_link_alloc_cache(sock);
+
+ 	class = nl_cli_class_alloc();
+ 
+	for (;;) {
+		int c, optidx = 0;
+		enum {
+			ARG_UPDATE = 257,
+			ARG_UPDATE_ONLY = 258,
+		};
+		static struct option long_opts[] = {
+			{ "quiet", 0, 0, 'q' },
+			{ "help", 0, 0, 'h' },
+			{ "version", 0, 0, 'v' },
+			{ "dev", 1, 0, 'd' },
+			{ "parent", 1, 0, 'p' },
+			{ "id", 1, 0, 'i' },
+			{ "update", 0, 0, ARG_UPDATE },
+			{ "update-only", 0, 0, ARG_UPDATE_ONLY },
+			{ 0, 0, 0, 0 }
+		};
+	
+		c = getopt_long(argc, argv, "+qhvd:p:i:",
+				long_opts, &optidx);
+		if (c == -1)
+			break;
+
+		switch (c) {
+		case 'q': quiet = 1; break;
+		case 'h': print_usage(); break;
+		case 'v': nl_cli_print_version(); break;
+		case 'd': nl_cli_class_parse_dev(class, link_cache, optarg); break;
+		case 'p': nl_cli_class_parse_parent(class, optarg); break;
+		case 'i': nl_cli_class_parse_handle(class, optarg); break;
+		case ARG_UPDATE: flags = NLM_F_CREATE; break;
+		case ARG_UPDATE_ONLY: flags = 0; break;
+		}
+ 	}
+
+	if (optind >= argc)
+		print_usage();
+
+	if (!rtnl_class_get_ifindex(class))
+		nl_cli_fatal(EINVAL, "You must specify a network device (--dev=XXX)");
+
+	if (!rtnl_class_get_parent(class))
+		nl_cli_fatal(EINVAL, "You must specify a parent (--parent=XXX)");
+
+	kind = argv[optind++];
+	rtnl_class_set_kind(class, kind);
+
+	if (!(ops = rtnl_class_lookup_ops(class)))
+		nl_cli_fatal(ENOENT, "Unknown class \"%s\"", kind);
+
+	if (!(qm = nl_cli_qdisc_lookup_by_class(ops)))
+		nl_cli_fatal(ENOTSUP, "class type \"%s\" not supported.", kind);
+
+	qm->qm_parse_class_argv(class, argc, argv);
+
+	if (!quiet) {
+		printf("Adding ");
+		nl_object_dump(OBJ_CAST(class), &dp);
+ 	}
+
+	if ((err = rtnl_class_add(sock, class, flags)) < 0)
+		nl_cli_fatal(EINVAL, "Unable to add class: %s", nl_geterror(err));
+
+	return 0;
+}