Fix prefix class logic

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 891dc41..09173e4 100644 (file)
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -10,6 +10,11 @@ set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -g -std=c99")
 
 add_definitions(-D_GNU_SOURCE -Wall -Werror -Wextra -DWITH_UBUS)
 
+if(${EXT_PREFIX_CLASS})
+       add_definitions(-DEXT_PREFIX_CLASS=${EXT_PREFIX_CLASS})
+endif(${EXT_PREFIX_CLASS})
+
+
 add_executable(odhcpd src/odhcpd.c src/config.c src/router.c src/dhcpv6.c src/ndp.c src/md5.c src/dhcpv6-ia.c src/dhcpv4.c src/ubus.c)
 target_link_libraries(odhcpd resolv ubus ubox uci)
 

diff --git a/src/dhcpv6-ia.c b/src/dhcpv6-ia.c
index 3d9fe83..b0a31a2 100644 (file)
--- a/src/dhcpv6-ia.c
+++ b/src/dhcpv6-ia.c
@@ -900,8 +900,10 @@ size_t dhcpv6_handle_ia(uint8_t *buf, size_t buflen, struct interface *iface,
                                a->assigned = reqhint;
                                a->all_class = class_oro;
                                a->classes_cnt = classes_cnt;
-                               a->classes = malloc(classes_cnt * sizeof(uint16_t));
-                               memcpy(a->classes, classes, classes_cnt * sizeof(uint16_t));
+                               if (classes_cnt) {
+                                       a->classes = malloc(classes_cnt * sizeof(uint16_t));
+                                       memcpy(a->classes, classes, classes_cnt * sizeof(uint16_t));
+                               }
 
                                if (first)
                                        memcpy(a->key, first->key, sizeof(a->key));

diff --git a/src/odhcpd.c b/src/odhcpd.c
index 5a800b7..d6ca298 100644 (file)
--- a/src/odhcpd.c
+++ b/src/odhcpd.c
@@ -244,10 +244,9 @@ ssize_t odhcpd_get_interface_addresses(int ifindex,
                addrs[ret].class = 0;
 #ifdef WITH_UBUS
                struct interface *iface = odhcpd_get_interface_by_index(ifindex);
-               if (iface) {
-                       addrs[ret].has_class = true;
-                       addrs[ret].class = ubus_get_class(iface->ifname, &addrs[ret].addr);
-               }
+               if (iface)
+                       addrs[ret].has_class = ubus_get_class(iface->ifname,
+                                       &addrs[ret].addr, &addrs[ret].class);
 #endif
                ++ret;
        }

diff --git a/src/odhcpd.h b/src/odhcpd.h
index 510dd6a..727c638 100644 (file)
--- a/src/odhcpd.h
+++ b/src/odhcpd.h
@@ -195,7 +195,7 @@ int init_ubus(void);
 const char* ubus_get_ifname(const char *name);
 void ubus_apply_network(void);
 bool ubus_has_prefix(const char *name, const char *ifname);
-uint16_t ubus_get_class(const char *ifname, const struct in6_addr *addr);
+bool ubus_get_class(const char *ifname, const struct in6_addr *addr, uint16_t *pclass);
 #endif
 
 

diff --git a/src/ubus.c b/src/ubus.c
index 317a9c1..dbb43a6 100644 (file)
--- a/src/ubus.c
+++ b/src/ubus.c
@@ -364,13 +364,13 @@ static const struct blobmsg_policy addr_attrs[ADDR_ATTR_MAX] = {
        [ADDR_ATTR_CLASS] = { .name = "class", .type = BLOBMSG_TYPE_STRING },
 };
 
-uint16_t ubus_get_class(const char *ifname, const struct in6_addr *addr)
+bool ubus_get_class(const char *ifname, const struct in6_addr *addr, uint16_t *pclass)
 {
        struct blob_attr *c, *cur;
        unsigned rem;
 
        if (!dump)
-               return 0;
+               return false;
 
        blobmsg_for_each_attr(c, dump, rem) {
                struct blob_attr *tb[IFACE_ATTR_MAX];
@@ -401,15 +401,17 @@ uint16_t ubus_get_class(const char *ifname, const struct in6_addr *addr)
                                struct in6_addr ip6addr;
                                inet_pton(AF_INET6, addrs, &ip6addr);
 
-                               if (IN6_ARE_ADDR_EQUAL(&ip6addr, addr))
-                                       return atoi(class);
+                               if (IN6_ARE_ADDR_EQUAL(&ip6addr, addr)) {
+                                       *pclass = atoi(class);
+                                       return true;
+                               }
                        }
                }
 
-               return 0;
+               return false;
        }
 
-       return 0;
+       return false;
 }