From 2ad270f0348da17ffcc1d8ab5d96513bbad82db9 Mon Sep 17 00:00:00 2001
From: "Jean-Marc Pigeon (Delson)" <jmp@safe.ca>
Date: Wed, 14 Aug 2024 14:52:36 -0400
Subject: [PATCH] Working out the unidns module

---
 lib/subrou.c | 27 ++++++++++++++++++++++++++-
 lib/subrou.h |  5 ++++-
 lib/unidns.c | 32 ++++++++++++++++++++++++--------
 3 files changed, 54 insertions(+), 10 deletions(-)

diff --git a/lib/subrou.c b/lib/subrou.c
index 9fa3673..61d85ec 100644
--- a/lib/subrou.c
+++ b/lib/subrou.c
@@ -21,7 +21,7 @@
 
 //version definition 
 #define VERSION "0.4.2"
-#define RELEASE "3"
+#define RELEASE "4"
 
 //Public variables
 PUBLIC  int debug=0;            //debug level
@@ -352,6 +352,31 @@ if ((list!=(void **)0)&&(entry!=(void **)0)) {
 return found;
 }
 /*
+
+*/
+/********************************************************/
+/*							*/
+/*	Procedure to free all list entries, return a    */
+/*      NULL pointer.                                   */
+/*							*/
+/********************************************************/
+PUBLIC void **rou_freelist(void **list,freehandler_t handler)
+
+{
+if (list!=(void **)0) {
+  void **ptr;
+
+  ptr=list;
+  while (*ptr!=(void *)0) {
+    (void) handler(*ptr);
+    ptr++;
+    }
+  (void) free(list);
+  list=(void **)0;
+  }
+return list;
+}
+/*
 ^L
 */
 /********************************************************/
diff --git a/lib/subrou.h b/lib/subrou.h
index 912f0d4..2e583bc 100644
--- a/lib/subrou.h
+++ b/lib/subrou.h
@@ -61,9 +61,12 @@ extern int rou_nbrlist(void **list);
 extern void **rou_addlist(void **list,void *entry);
 
 //Procedure to free one entry part of list
-// return true if found, false otherwize. 
+//return true if found, false otherwise. 
 extern _Bool rou_rmlist(void **list,void *entry,freehandler_t handler);
 
+//Procedure to free all entries of list
+extern void **rou_freelist(void **list,freehandler_t handler);
+
 //change the application name
 extern char *rou_setappname(const char *newname);
 
diff --git a/lib/unidns.c b/lib/unidns.c
index dc07158..649c154 100644
--- a/lib/unidns.c
+++ b/lib/unidns.c
@@ -153,16 +153,13 @@ while((answer-->0)&&(cp<eom)) {
         dynbuf=(char *)calloc(10,sizeof(char));
 	while (collected<dlen) {
           u_int taille;
-	  char *newdynbuf;
 
 	  (void) memset(buf,'\000',sizeof(buf));
           taille=((u_int)(txtptr[0]))&0X0FF; 
 	  txtptr++;
           (void) snprintf(buf,sizeof(buf),"%.*s ",taille,txtptr);
-	  if ((newdynbuf=(char *)realloc(dynbuf,strlen(dynbuf)+strlen(buf)+5))!=(char *)0) {
-            dynbuf=newdynbuf;
-	    (void) strcat(dynbuf,buf);
-            }
+	  dynbuf=(char *)realloc(dynbuf,strlen(dynbuf)+strlen(buf)+5);
+	  (void) strcat(dynbuf,buf);
 	  collected+=taille+1;
 	  txtptr+=taille;
           }
@@ -244,11 +241,29 @@ return inftxt;
 static char **getspf(char *domain)
 
 {
-char **inftxt;
+char **infspf;
+
+if ((infspf=gettxt(domain,T_SPF,"SPF"))==(char **)0) {
+  char **inftxt;
 
-if ((inftxt=gettxt(domain,T_SPF,"SPF"))==(char **)0)
   inftxt=gettxt(domain,T_TXT,"TXT");
-return inftxt;
+  if (inftxt!=(char **)0) {
+    char **ptr;
+
+    ptr=inftxt;
+    while (*ptr!=(char *)0) {
+      if (strstr(*ptr,"v=spf1")==*ptr) {
+        char *dup;
+
+        dup=strdup(*ptr);
+        infspf=(char **)rou_addlist((void **)infspf,(void *)dup);
+        }
+      ptr++;
+      }
+    inftxt=(char **)rou_freelist((void **)inftxt,(freehandler_t)rou_freestr);
+    }
+  }
+return infspf;
 }
 /*
 ^L
@@ -342,6 +357,7 @@ while (proceed==true) {
           (void) printf("JMPDBG <%s>\n",*ptr);
           ptr++;
           }
+        list=(char **)rou_freelist((void **)list,(freehandler_t)rou_freestr);
         }
       break;
     default     :       //SAFE guard
-- 
2.47.3