===================================================================
RCS file: /cvs/mandoc/Attic/mdocml.c,v
retrieving revision 1.1
retrieving revision 1.4
diff -u -p -r1.1 -r1.4
--- mandoc/Attic/mdocml.c	2008/11/22 14:53:29	1.1
+++ mandoc/Attic/mdocml.c	2008/11/22 18:34:06	1.4
@@ -1,4 +1,4 @@
-/* $Id: mdocml.c,v 1.1 2008/11/22 14:53:29 kristaps Exp $ */
+/* $Id: mdocml.c,v 1.4 2008/11/22 18:34:06 kristaps Exp $ */
 /*
  * Copyright (c) 2008 Kristaps Dzonsons <kristaps@kth.se>
  *
@@ -30,30 +30,17 @@
 
 #include "libmdocml.h"
 
-struct md_file {
-	int		 fd;
-	const char	*name;
-};
+#define	BUFFER_IN_DEF	BUFSIZ
+#define	BUFFER_OUT_DEF	BUFSIZ
 
-struct md_buf {
-	struct md_file	*file;
-	char		*buf;
-	size_t		 bufsz;
-	size_t		 line;
-};
-
 static void		 usage(void);
+static int		 begin_io(const char *, const char *);
+static int		 leave_io(const struct md_mbuf *, 
+				const struct md_rbuf *, int);
+static int		 begin_bufs(struct md_mbuf *, struct md_rbuf *);
+static int		 leave_bufs(const struct md_mbuf *, 
+				const struct md_rbuf *, int);
 
-static int		 md_begin(const char *, const char *);
-static int		 md_begin_io(const char *, const char *);
-static int		 md_begin_bufs(struct md_file *, struct md_file *);
-static int		 md_run(struct md_buf *, struct md_buf *);
-static int		 md_line(struct md_buf *, const struct md_buf *,
-				const char *, size_t);
-
-static ssize_t		 md_buf_fill(struct md_buf *);
-
-
 int
 main(int argc, char *argv[])
 {
@@ -63,7 +50,7 @@ main(int argc, char *argv[])
 	extern char	*optarg;
 	extern int	 optind;
 
-	out = NULL;
+	out = in = NULL;
 	
 	while (-1 != (c = getopt(argc, argv, "o:")))
 		switch (c) {
@@ -76,212 +63,124 @@ main(int argc, char *argv[])
 		}
 
 	argv += optind;
-	if (1 != (argc -= optind)) {
-		usage();
-		return(1);
-	}
+	argc -= optind;
 
-	argc--;
-	in = *argv++;
+	if (1 == argc)
+		in = *argv++;
 
-	return(md_begin(out, in));
+	return(begin_io(out ? out : "-", in ? in : "-"));
 }
 
 
 static int
-md_begin(const char *out, const char *in)
+leave_io(const struct md_mbuf *out, 
+		const struct md_rbuf *in, int c)
 {
-	char		 buf[MAXPATHLEN];
-
+	assert(out);
 	assert(in);
-	if (out)
-		return(md_begin_io(out, in));
 
-	if (strlcpy(buf, in, MAXPATHLEN) >= MAXPATHLEN)
-		warnx("output filename too long");
-	else if (strlcat(buf, ".html", MAXPATHLEN) >= MAXPATHLEN)
-		warnx("output filename too long");
-	else 
-		return(md_begin_io(buf, in));
+	if (-1 != in->fd && -1 == close(in->fd)) {
+		assert(in->name);
+		warn("%s", in->name);
+		c = 1;
+	}
+	if (-1 != out->fd && STDOUT_FILENO != out->fd &&
+			-1 == close(out->fd)) {
+		assert(out->name);
+		warn("%s", out->name);
+		c = 1;
+	}
 
-	return(1);
+	return(c);
 }
 
 
 static int
-md_begin_io(const char *out, const char *in)
+begin_io(const char *out, const char *in)
 {
-	int		 c;
-	struct md_file	 fin, fout;
+	struct md_rbuf	 fi;
+	struct md_mbuf	 fo;
 
+#define	FI_FL	O_RDONLY
+#define	FO_FL	O_WRONLY|O_CREAT|O_TRUNC
+
 	assert(out);
 	assert(in);
 
-	/* XXX: put an output file in TMPDIR and switch it out when the
-	 * true file is ready to be written.
-	 */
+	bzero(&fi, sizeof(struct md_rbuf));
+	bzero(&fo, sizeof(struct md_mbuf));
 
-	fin.name = in;
+	fi.fd = STDIN_FILENO;
+	fo.fd = STDOUT_FILENO;
 
-	if (-1 == (fin.fd = open(fin.name, O_RDONLY, 0))) {
-		warn("%s", fin.name);
-		return(1);
-	}
+	fi.name = in;
+	fo.name = out;
 
-	fout.name = out;
+	if (0 != strncmp(fi.name, "-", 1))
+		if (-1 == (fi.fd = open(fi.name, FI_FL, 0))) {
+			warn("%s", fi.name);
+			return(leave_io(&fo, &fi, 1));
+		}
 
-	fout.fd = open(fout.name, O_WRONLY | O_CREAT | O_EXCL, 0644);
-	if (-1 == fout.fd) {
-		warn("%s", fout.name);
-		if (-1 == close(fin.fd))
-			warn("%s", fin.name);
-		return(1);
-	}
+	if (0 != strncmp(fo.name, "-", 1)) 
+		if (-1 == (fo.fd = open(fo.name, FO_FL, 0644))) {
+			warn("%s", fo.name);
+			return(leave_io(&fo, &fi, 1));
+		}
 
-	c = md_begin_bufs(&fout, &fin);
+	return(leave_io(&fo, &fi, begin_bufs(&fo, &fi)));
+}
 
-	if (-1 == close(fin.fd)) {
-		warn("%s", in);
-		c = 1;
-	}
-	if (-1 == close(fout.fd)) {
-		warn("%s", out);
-		c = 1;
-	}
 
+static int
+leave_bufs(const struct md_mbuf *out, 
+		const struct md_rbuf *in, int c)
+{
+	assert(out);
+	assert(in);
+	if (out->buf)
+		free(out->buf);
+	if (in->buf)
+		free(in->buf);
 	return(c);
 }
 
 
 static int
-md_begin_bufs(struct md_file *out, struct md_file *in)
+begin_bufs(struct md_mbuf *out, struct md_rbuf *in)
 {
 	struct stat	 stin, stout;
-	struct md_buf	 inbuf, outbuf;
-	int		 c;
 
 	assert(in);
 	assert(out);
 
 	if (-1 == fstat(in->fd, &stin)) {
-		warn("fstat: %s", in->name);
+		warn("%s", in->name);
 		return(1);
 	} else if (-1 == fstat(out->fd, &stout)) {
-		warn("fstat: %s", out->name);
+		warn("%s", out->name);
 		return(1);
 	}
 
-	inbuf.file = in;
-	inbuf.line = 1;
-	inbuf.bufsz = MAX(stin.st_blksize, BUFSIZ);
+	in->bufsz = MAX(stin.st_blksize, BUFFER_IN_DEF);
+	out->bufsz = MAX(stout.st_blksize, BUFFER_OUT_DEF);
 
-	outbuf.file = out;
-	outbuf.line = 1;
-	outbuf.bufsz = MAX(stout.st_blksize, BUFSIZ);
-
-	if (NULL == (inbuf.buf = malloc(inbuf.bufsz))) {
+	if (NULL == (in->buf = malloc(in->bufsz))) {
 		warn("malloc");
-		return(1);
-	} else if (NULL == (outbuf.buf = malloc(outbuf.bufsz))) {
+		return(leave_bufs(out, in, 1));
+	} else if (NULL == (out->buf = malloc(out->bufsz))) {
 		warn("malloc");
-		free(inbuf.buf);
-		return(1);
+		return(leave_bufs(out, in, 1));
 	}
 
-	c = md_run(&outbuf, &inbuf);
-
-	free(inbuf.buf);
-	free(outbuf.buf);
-
-	return(c);
+	return(leave_bufs(out, in, md_run(MD_DUMMY, out, in)));
 }
 
 
-static ssize_t
-md_buf_fill(struct md_buf *in)
-{
-	ssize_t		 ssz;
-
-	assert(in);
-	assert(in->file);
-	assert(in->buf);
-	assert(in->bufsz > 0);
-	assert(in->file->name);
-
-	if (-1 == (ssz = read(in->file->fd, in->buf, in->bufsz))) 
-		warn("%s", in->file->name);
-
-	return(ssz);
-}
-
-
-static int
-md_run(struct md_buf *out, struct md_buf *in)
-{
-	ssize_t		 sz, i;
-	char		 line[BUFSIZ];
-	size_t		 pos;
-
-	assert(in);
-	assert(out); 
-
-	/* LINTED */
-	for (pos = 0; ; ) {
-		if (-1 == (sz = md_buf_fill(in)))
-			return(1);
-		else if (0 == sz)
-			break;
-
-		for (i = 0; i < sz; i++) {
-			if ('\n' == in->buf[i]) {
-				if (md_line(out, in, line, pos))
-					return(1);
-				in->line++;
-				pos = 0;
-				continue;
-			}
-
-			if (pos < BUFSIZ) {
-				/* LINTED */
-				line[pos++] = in->buf[i];
-				continue;
-			}
-
-			warnx("%s: line %zu too long",
-					in->file->name, in->line);
-			return(1);
-		}
-	}
-
-	if (0 != pos)
-		return(md_line(out, in, line, pos));
-
-	return(0);
-}
-
-
-static int
-md_line(struct md_buf *out, const struct md_buf *in,
-		const char *buf, size_t sz)
-{
-	size_t		 i;
-
-	assert(buf);
-	assert(out);
-	assert(in);
-
-	for (i = 0; i < sz; i++) 
-		(void)putchar(buf[i]);
-
-	(void)putchar('\n');
-	return(0);
-}
-
-
 static void
 usage(void)
 {
 	extern char	*__progname;
 
-	(void)printf("usage: %s [-o outfile] infile\n", __progname);
+	(void)printf("usage: %s [-o outfile] [infile]\n", __progname);
 }