X-Git-Url: https://diplodocus.org/git/nmh/blobdiff_plain/2adafe760a9c45b417727a3f5d9481de26471a8b..ef1ba39e8dae81091b6c3e73e72825ef6edea3c6:/uip/ap.c

diff --git a/uip/ap.c b/uip/ap.c
index 3dd77b17..89b207b5 100644
--- a/uip/ap.c
+++ b/uip/ap.c
@@ -11,6 +11,7 @@
 #include <h/addrsbr.h>
 #include <h/fmt_scan.h>
 #include <h/mts.h>
+#include <h/utils.h>
 
 #define	NADDRS	100
 
@@ -52,9 +53,9 @@ main (int argc, char **argv)
     char buf[BUFSIZ], **argp;
     char **arguments, *addrs[NADDRS];
 
-    if (nmh_init(argv[0], 1)) { return 1; }
+    if (nmh_init(argv[0], 2)) { return 1; }
 
-    mts_init (invo_name);
+    mts_init ();
     arguments = getarguments (invo_name, argc, argv, 1);
     argp = arguments;
 
@@ -112,11 +113,11 @@ main (int argc, char **argv)
 	if ((width = sc_width ()) < WIDTH / 2) {
 	    /* Default:  width of the terminal, but at least WIDTH/2. */
 	    width = WIDTH / 2;
-	} else if (width == 0) {
-	    /* Unlimited width. */
-	    width = INT_MAX;
 	}
 	width -= 2;
+    } else if (width == 0) {
+	/* Unlimited width.  */
+	width = INT_MAX;
     }
     fmt_compile (nfs, &fmt, 1);
 
@@ -145,20 +146,19 @@ static int
 process (char *arg, int length)
 {
     int	status = 0;
-    register char *cp;
+    char *cp;
     char error[BUFSIZ];
-    register struct comp *cptr;
-    register struct pqpair *p, *q;
+    struct comp *cptr;
+    struct pqpair *p, *q;
     struct pqpair pq;
-    register struct mailname *mp;
+    struct mailname *mp;
 
     (q = &pq)->pq_next = NULL;
     while ((cp = getname (arg))) {
-	if ((p = (struct pqpair *) calloc ((size_t) 1, sizeof(*p))) == NULL)
-	    adios (NULL, "unable to allocate pqpair memory");
+	NEW0(p);
 	if ((mp = getm (cp, NULL, 0, error, sizeof(error))) == NULL) {
-	    p->pq_text = getcpy (cp);
-	    p->pq_error = getcpy (error);
+	    p->pq_text = mh_xstrdup(cp);
+	    p->pq_error = mh_xstrdup(error);
 	    status++;
 	}
 	else {
@@ -169,19 +169,18 @@ process (char *arg, int length)
     }
 
     for (p = pq.pq_next; p; p = q) {
-	charstring_t scanl = charstring_create (length);
+	charstring_t scanl =
+	    charstring_create (length < NMH_BUFSIZ ? length : NMH_BUFSIZ);
 
 	cptr = fmt_findcomp ("text");
 	if (cptr) {
-	    if (cptr->c_text)
-	    	free(cptr->c_text);
+            mh_xfree(cptr->c_text);
 	    cptr->c_text = p->pq_text;
 	    p->pq_text = NULL;
 	}
 	cptr = fmt_findcomp ("error");
 	if (cptr) {
-	    if (cptr->c_text)
-	    	free(cptr->c_text);
+            mh_xfree(cptr->c_text);
 	    cptr->c_text = p->pq_error;
 	    p->pq_error = NULL;
 	}
@@ -190,12 +189,10 @@ process (char *arg, int length)
 	fputs (charstring_buffer (scanl), stdout);
 	charstring_free (scanl);
 
-	if (p->pq_text)
-	    free (p->pq_text);
-	if (p->pq_error)
-	    free (p->pq_error);
+        mh_xfree(p->pq_text);
+        mh_xfree(p->pq_error);
 	q = p->pq_next;
-	free ((char *) p);
+	free(p);
     }
 
     return status;