vrrpd: use correct mtypes

Signed-off-by: Quentin Young <qlyoung@cumulusnetworks.com>

diff --git a/vrrpd/vrrp.c b/vrrpd/vrrp.c
index cc93b94a9b1f2959694a061b2dcfba22d2406f32..37846d625301f327c0f84cd2d55c704ce0cc16ce 100644 (file)
--- a/vrrpd/vrrp.c
+++ b/vrrpd/vrrp.c
@@ -32,6 +32,7 @@
 
 #include "vrrp.h"
 #include "vrrp_arp.h"
+#include "vrrp_memory.h"
 #include "vrrp_ndisc.h"
 #include "vrrp_packet.h"
 #include "vrrp_zebra.h"
@@ -173,7 +174,7 @@ int vrrp_add_ip(struct vrrp_router *r, struct ipaddr *ip, bool activate)
                return -1;
        }
 
-       struct ipaddr *new = XCALLOC(MTYPE_TMP, sizeof(struct ipaddr));
+       struct ipaddr *new = XCALLOC(MTYPE_VRRP_IP, sizeof(struct ipaddr));
 
        *new = *ip;
        listnode_add(r->addrs, new);
@@ -263,7 +264,7 @@ int vrrp_del_ipv4(struct vrrp_vrouter *vr, struct in_addr v4, bool deactivate)
 static void vrrp_router_addr_list_del_cb(void *val)
 {
        struct ipaddr *ip = val;
-       XFREE(MTYPE_TMP, ip);
+       XFREE(MTYPE_VRRP_IP, ip);
 }
 
 /*
@@ -332,7 +333,8 @@ static bool vrrp_attach_interface(struct vrrp_router *r)
 static struct vrrp_router *vrrp_router_create(struct vrrp_vrouter *vr,
                                              int family)
 {
-       struct vrrp_router *r = XCALLOC(MTYPE_TMP, sizeof(struct vrrp_router));
+       struct vrrp_router *r =
+               XCALLOC(MTYPE_VRRP_RTR, sizeof(struct vrrp_router));
 
        r->family = family;
        r->sock_rx = -1;
@@ -361,7 +363,7 @@ static void vrrp_router_destroy(struct vrrp_router *r)
 
        /* FIXME: also delete list elements */
        list_delete(&r->addrs);
-       XFREE(MTYPE_TMP, r);
+       XFREE(MTYPE_VRRP_RTR, r);
 }
 
 struct vrrp_vrouter *vrrp_vrouter_create(struct interface *ifp, uint8_t vrid)
@@ -371,7 +373,7 @@ struct vrrp_vrouter *vrrp_vrouter_create(struct interface *ifp, uint8_t vrid)
        if (vr)
                return vr;
 
-       vr = XCALLOC(MTYPE_TMP, sizeof(struct vrrp_vrouter));
+       vr = XCALLOC(MTYPE_VRRP_RTR, sizeof(struct vrrp_vrouter));
 
        vr->ifp = ifp;
        vr->vrid = vrid;
@@ -394,7 +396,7 @@ void vrrp_vrouter_destroy(struct vrrp_vrouter *vr)
        vrrp_router_destroy(vr->v4);
        vrrp_router_destroy(vr->v6);
        hash_release(vrrp_vrouters_hash, vr);
-       XFREE(MTYPE_TMP, vr);
+       XFREE(MTYPE_VRRP_RTR, vr);
 }
 
 struct vrrp_vrouter *vrrp_lookup(struct interface *ifp, uint8_t vrid)
@@ -444,7 +446,7 @@ static void vrrp_send_advertisement(struct vrrp_router *r)
        ssize_t sent = sendto(r->sock_tx, pkt, (size_t)pktlen, 0, &dest.sa,
                              sockunion_sizeof(&dest));
 
-       XFREE(MTYPE_TMP, pkt);
+       XFREE(MTYPE_VRRP_PKT, pkt);
 
        if (sent < 0) {
                zlog_warn(VRRP_LOGPFX VRRP_LOGPFX_VRID

diff --git a/vrrpd/vrrp_memory.c b/vrrpd/vrrp_memory.c
index 9ce7c90413ab2f1db114dec3c2627f12edeef264..30eef523cd503fa2af6f6eb5987b95369802a52f 100644 (file)
--- a/vrrpd/vrrp_memory.c
+++ b/vrrpd/vrrp_memory.c
@@ -18,8 +18,12 @@
  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
  */
 #include <zebra.h>
-#include <memory.h>
+
+#include "lib/memory.h"
 
 #include "vrrp_memory.h"
 
-DEFINE_MGROUP(VRRP, "vrrpd")
+DEFINE_MGROUP(VRRPD, "vrrpd");
+DEFINE_MTYPE(VRRPD, VRRP_IP, "VRRP IP address");
+DEFINE_MTYPE(VRRPD, VRRP_PKT, "VRRP packet");
+DEFINE_MTYPE(VRRPD, VRRP_RTR, "VRRP Router");

diff --git a/vrrpd/vrrp_memory.h b/vrrpd/vrrp_memory.h
index ae6975478dd3e57b46c7d293e07a26899fb3266c..c3025d1acb5c9692208b89be24f1cdd196bdd895 100644 (file)
--- a/vrrpd/vrrp_memory.h
+++ b/vrrpd/vrrp_memory.h
@@ -20,8 +20,13 @@
 #ifndef __VRRP_MEMORY_H__
 #define __VRRP_MEMORY_H__
 
+#include <zebra.h>
+
 #include "lib/memory.h"
 
-DECLARE_MGROUP(VRRP)
+DECLARE_MGROUP(VRRPD);
+DECLARE_MTYPE(VRRP_IP);
+DECLARE_MTYPE(VRRP_PKT);
+DECLARE_MTYPE(VRRP_RTR);
 
 #endif /* __VRRP_MEMORY_H__ */

diff --git a/vrrpd/vrrp_packet.c b/vrrpd/vrrp_packet.c
index fd0256d6e3871f4200cf6d11f3842f3462173f76..eca3af0437b6dd04072cdab80b711c943a7825cd 100644 (file)
--- a/vrrpd/vrrp_packet.c
+++ b/vrrpd/vrrp_packet.c
@@ -27,6 +27,7 @@
 #include "lib/memory.h"
 
 #include "vrrp.h"
+#include "vrrp_memory.h"
 #include "vrrp_packet.h"
 
 /* clang-format off */
@@ -58,7 +59,7 @@ ssize_t vrrp_pkt_build(struct vrrp_pkt **pkt, struct ipaddr *src, uint8_t vrid,
 
        size_t addrsz = v6 ? sizeof(struct in6_addr) : sizeof(struct in_addr);
        size_t pktsize = VRRP_PKT_SIZE(v6 ? AF_INET6 : AF_INET, numip);
-       *pkt = XCALLOC(MTYPE_TMP, pktsize);
+       *pkt = XCALLOC(MTYPE_VRRP_PKT, pktsize);
 
        (*pkt)->hdr.vertype |= VRRP_VERSION << 4;
        (*pkt)->hdr.vertype |= VRRP_TYPE_ADVERTISEMENT;