- xfr-tsig, tsig_get_mem function.

diff --git a/services/authzone.c b/services/authzone.c
index 6c9d83444a15dfe262a722dfaa58b955153359bf..c7aff82a1f1d31ee4be9445382f5a4fb84a03aed 100644 (file)
--- a/services/authzone.c
+++ b/services/authzone.c
@@ -8880,12 +8880,14 @@ auth_xfer_get_mem(struct auth_xfer* xfr)
        m += auth_primaries_get_mem(xfr->task_probe->masters);
        m += comm_point_get_mem(xfr->task_probe->cp);
        m += comm_timer_get_mem(xfr->task_probe->timer);
+       m += tsig_get_mem(xfr->task_probe->tsig);
 
        /* auth_transfer */
        m += auth_chunks_get_mem(xfr->task_transfer->chunks_first);
        m += auth_primaries_get_mem(xfr->task_transfer->masters);
        m += comm_point_get_mem(xfr->task_transfer->cp);
        m += comm_timer_get_mem(xfr->task_transfer->timer);
+       m += tsig_get_mem(xfr->task_transfer->tsig);
 
        /* allow_notify_list */
        m += auth_primaries_get_mem(xfr->allow_notify_list);

diff --git a/util/tsig.c b/util/tsig.c
index f54051064ef9c9da5122c9334772609f93b2f099..bd6e4c11e0cd012c1b4ea17a49b4b366aba04f5b 100644 (file)
--- a/util/tsig.c
+++ b/util/tsig.c
@@ -492,6 +492,14 @@ tsig_delete(struct tsig_data* tsig)
        free(tsig);
 }
 
+size_t tsig_get_mem(struct tsig_data* tsig)
+{
+       if(!tsig)
+               return 0;
+       return sizeof(*tsig) + tsig->key_name_len + tsig->algo_name_len
+               + tsig->mac_size;
+}
+
 size_t
 tsig_reserved_space(struct tsig_data* tsig)
 {

diff --git a/util/tsig.h b/util/tsig.h
index 7278df2f1ad3c5670040e61c3a30f8205c3539ce..16cbf08004cbdd94c3ba6102f152eb9a043b08cc 100644 (file)
--- a/util/tsig.h
+++ b/util/tsig.h
@@ -305,6 +305,13 @@ struct tsig_data* tsig_create_fromstr(struct tsig_key_table* key_table,
  */
 void tsig_delete(struct tsig_data* tsig);
 
+/**
+ * Get memory usage of tsig data.
+ * @param rsig: the tsig data.
+ * @return memory use.
+ */
+size_t tsig_get_mem(struct tsig_data* tsig);
+
 /**
  * Sign a query with TSIG. Appends the TSIG record.
  * @param tsig: the tsig data, keeps state to verify reply.