X-Git-Url: http://git.marmaro.de/?p=mmh;a=blobdiff_plain;f=sbr%2Ffmt_new.c;h=c1386d4c710bb4de354bbbbc9d219b6753ccc5eb;hp=7576ae42b53d880ff4ad3dcc5a12b360a7d65647;hb=18591f8e001ecedbee48a51c1d1f08ebaa1c15c8;hpb=ced6090a330d3d83d0bce709f756aa3d7d65fea4

diff --git a/sbr/fmt_new.c b/sbr/fmt_new.c
index 7576ae4..c1386d4 100644
--- a/sbr/fmt_new.c
+++ b/sbr/fmt_new.c
@@ -8,50 +8,67 @@
 
 #include <h/mh.h>
 #include <h/utils.h>
+#include <unistd.h>
+#include <sys/stat.h>
+#include <sysexits.h>
 
-#define QUOTE '\\'
-
-static char *formats = 0;
+static char *formats = NULL;
 
 /*
 ** static prototypes
 */
-static void normalize (char *);
+static void normalize(char *);
 
 
 /*
-** Get new format string
+** Copy first available format string, store in static memory and normalize it.
 */
-
 char *
-new_fs (char *form, char *format, char *default_fs)
+new_fs(char *form, char *def_form)
 {
 	struct stat st;
-	register FILE *fp;
+	FILE *fp;
 
-	if (formats)
-		free (formats);
+	if (formats) {
+		mh_free0(&formats);
+	}
 
 	if (form) {
-		if ((fp = fopen (etcpath (form), "r")) == NULL)
-			adios (form, "unable to open format file");
-
-		if (fstat (fileno (fp), &st) == -1)
-			adios (form, "unable to stat format file");
-
-		formats = mh_xmalloc ((size_t) st.st_size + 1);
-
-		if (read (fileno(fp), formats, (int) st.st_size) != st.st_size)
-			adios (form, "error reading format file");
-
-		formats[st.st_size] = '\0';
-
-		fclose (fp);
-	} else {
-		formats = getcpy (format ? format : default_fs);
+		if (*form == '=') {
+			formats = mh_xstrdup(form+1);
+		} else {
+			if ((fp = fopen(etcpath(form), "r")) == NULL) {
+				adios(EX_IOERR, form, "unable to open format file");
+			}
+			if (fstat(fileno(fp), &st) == -1) {
+				adios(EX_IOERR, form, "unable to stat format file");
+			}
+			formats = mh_xcalloc(st.st_size + 1, sizeof(char));
+			if (read(fileno(fp), formats, (int)st.st_size) != st.st_size) {
+				adios(EX_IOERR, form, "error reading format file");
+			}
+			formats[st.st_size] = '\0';
+			fclose(fp);
+		}
+	} else if (def_form) {
+		if (*def_form == '=') {
+			formats = mh_xstrdup(def_form+1);
+		} else {
+			if ((fp = fopen(etcpath(def_form), "r")) == NULL) {
+				adios(EX_IOERR, def_form, "unable to open format file");
+			}
+			if (fstat(fileno(fp), &st) == -1) {
+				adios(EX_IOERR, def_form, "unable to stat format file");
+			}
+			formats = mh_xcalloc(st.st_size + 1, sizeof(char));
+			if (read(fileno(fp), formats, (int)st.st_size) != st.st_size) {
+				adios(EX_IOERR, def_form, "error reading format file");
+			}
+			formats[st.st_size] = '\0';
+			fclose(fp);
+		}
 	}
-
-	normalize (formats);  /* expand escapes */
+	normalize(formats);  /* expand escapes */
 
 	return formats;
 }
@@ -60,46 +77,41 @@ new_fs (char *form, char *format, char *default_fs)
 /*
 ** Expand escapes in format strings
 */
-
 static void
-normalize (char *cp)
+normalize(char *cp)
 {
 	char *dp;
 
 	for (dp = cp; *cp; cp++) {
-		if (*cp != QUOTE) {
+		if (*cp != '\\') {
 			*dp++ = *cp;
-		} else {
-			switch (*++cp) {
-				case 'b':
-					*dp++ = '\b';
-					break;
-
-				case 'f':
-					*dp++ = '\f';
-					break;
-
-				case 'n':
-					*dp++ = '\n';
-					break;
-
-				case 'r':
-					*dp++ = '\r';
-					break;
-
-				case 't':
-					*dp++ = '\t';
-					break;
-
-				case '\n':
-					break;
+			continue;
+		}
 
-				case 0:
-					cp--;  /* fall */
-				default:
-					*dp++ = *cp;
-					break;
-			}
+		switch (*++cp) {
+		case 'b':
+			*dp++ = '\b';
+			break;
+		case 'f':
+			*dp++ = '\f';
+			break;
+		case 'n':
+			*dp++ = '\n';
+			break;
+		case 'r':
+			*dp++ = '\r';
+			break;
+		case 't':
+			*dp++ = '\t';
+			break;
+		case '\n':
+			break;
+		case '\0':
+			cp--;
+			/* fall */
+		default:
+			*dp++ = *cp;
+			break;
 		}
 	}
 	*dp = '\0';