lua: fix memory leak problem
[project/ubus.git] / ubusd_id.c
index 9443db9..8d9fede 100644 (file)
@@ -1,8 +1,23 @@
+/*
+ * Copyright (C) 2011 Felix Fietkau <nbd@openwrt.org>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License version 2.1
+ * as published by the Free Software Foundation
+ *
+ * This program 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.
+ */
+
 #include <stdio.h>
 #include <stdlib.h>
 #include <unistd.h>
 #include <fcntl.h>
+#include <libubox/avl-cmp.h>
 
+#include "ubusmsg.h"
 #include "ubusd_id.h"
 
 static int random_fd = -1;
@@ -17,6 +32,11 @@ static int ubus_cmp_id(const void *k1, const void *k2, void *ptr)
                return *id1 > *id2;
 }
 
+void ubus_init_string_tree(struct avl_tree *tree, bool dup)
+{
+       avl_init(tree, avl_strcmp, dup, NULL);
+}
+
 void ubus_init_id_tree(struct avl_tree *tree)
 {
        if (random_fd < 0) {
@@ -30,14 +50,19 @@ void ubus_init_id_tree(struct avl_tree *tree)
        avl_init(tree, ubus_cmp_id, false, NULL);
 }
 
-bool ubus_alloc_id(struct avl_tree *tree, struct ubus_id *id)
+bool ubus_alloc_id(struct avl_tree *tree, struct ubus_id *id, uint32_t val)
 {
        id->avl.key = &id->id;
+       if (val) {
+               id->id = val;
+               return avl_insert(tree, &id->avl) == 0;
+       }
+
        do {
                if (read(random_fd, &id->id, sizeof(id->id)) != sizeof(id->id))
                        return false;
 
-               if (!id->id)
+               if (id->id < UBUS_SYSTEM_OBJECT_MAX)
                        continue;
        } while (avl_insert(tree, &id->avl) != 0);