From b6bd12c3f0303e71b580e5ff0eb8e9adf580d286 Mon Sep 17 00:00:00 2001 From: Kristaps Dzonsons Date: Sat, 22 Nov 2008 18:34:06 +0000 Subject: Putting md_run and friends into libmdocml (needs work to be useful). --- mdocml.c | 302 ++++++++++++++------------------------------------------------- 1 file changed, 65 insertions(+), 237 deletions(-) (limited to 'mdocml.c') diff --git a/mdocml.c b/mdocml.c index 2620a31e..dbcb88d0 100644 --- a/mdocml.c +++ b/mdocml.c @@ -32,40 +32,14 @@ #define BUFFER_IN_DEF BUFSIZ #define BUFFER_OUT_DEF BUFSIZ -#define BUFFER_LINE BUFSIZ - -struct md_rbuf { - int fd; - const char *name; - char *buf; - size_t bufsz; - size_t line; -}; - -struct md_mbuf { - int fd; - const char *name; - char *buf; - size_t bufsz; - size_t pos; -}; static void usage(void); - -static int md_begin(const char *, const char *); -static int md_begin_io(const char *, const char *); -static int md_begin_bufs(struct md_mbuf *, struct md_rbuf *); -static int md_run(struct md_mbuf *, struct md_rbuf *); -static int md_line(struct md_mbuf *, const struct md_rbuf *, - const char *, size_t); - -static ssize_t md_buf_fill(struct md_rbuf *); -static int md_buf_flush(struct md_mbuf *); - -static int md_buf_putchar(struct md_mbuf *, char); -static int md_buf_puts(struct md_mbuf *, - const char *, size_t); - +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); int main(int argc, char *argv[]) @@ -76,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) { @@ -89,87 +63,93 @@ 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_rbuf fin; - struct md_mbuf 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); - /* TODO: accept "-" as both input and output. */ + 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_TRUNC, 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_mbuf *out, struct md_rbuf *in) +begin_bufs(struct md_mbuf *out, struct md_rbuf *in) { struct stat stin, stout; - int c; assert(in); assert(out); @@ -183,169 +163,17 @@ md_begin_bufs(struct md_mbuf *out, struct md_rbuf *in) } in->bufsz = MAX(stin.st_blksize, BUFFER_IN_DEF); - out->bufsz = MAX(stout.st_blksize, BUFFER_OUT_DEF); if (NULL == (in->buf = malloc(in->bufsz))) { warn("malloc"); - return(1); + return(leave_bufs(out, in, 1)); } else if (NULL == (out->buf = malloc(out->bufsz))) { warn("malloc"); - free(in->buf); - return(1); - } - - c = md_run(out, in); - - free(in->buf); - free(out->buf); - - return(c); -} - - -static ssize_t -md_buf_fill(struct md_rbuf *in) -{ - ssize_t ssz; - - assert(in); - assert(in->buf); - assert(in->bufsz > 0); - assert(in->name); - - if (-1 == (ssz = read(in->fd, in->buf, in->bufsz))) - warn("%s", in->name); - - return(ssz); -} - - -static int -md_run(struct md_mbuf *out, struct md_rbuf *in) -{ - ssize_t sz, i; - char line[BUFFER_LINE]; - size_t pos; - - assert(in); - assert(out); - - out->pos = 0; - in->line = 1; - - /* 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 < BUFFER_LINE) { - /* LINTED */ - line[pos++] = in->buf[i]; - continue; - } - - warnx("%s: line %zu too long", - in->name, in->line); - return(1); - } - } - - if (0 != pos && md_line(out, in, line, pos)) - return(1); - - return(md_buf_flush(out) ? 0 : 1); -} - - -static int -md_buf_flush(struct md_mbuf *buf) -{ - ssize_t sz; - - assert(buf); - assert(buf->buf); - assert(buf->name); - - if (0 == buf->pos) - return(1); - - sz = write(buf->fd, buf->buf, buf->pos); - - if (-1 == sz) { - warn("%s", buf->name); - return(0); - } else if ((size_t)sz != buf->pos) { - warnx("%s: short write", buf->name); - return(0); - } - - buf->pos = 0; - return(1); -} - - -static int -md_buf_putchar(struct md_mbuf *buf, char c) -{ - return(md_buf_puts(buf, &c, 1)); -} - - -static int -md_buf_puts(struct md_mbuf *buf, const char *p, size_t sz) -{ - size_t ssz; - - assert(p); - assert(buf); - assert(buf->buf); - - while (buf->pos + sz > buf->bufsz) { - ssz = buf->bufsz - buf->pos; - (void)memcpy(buf->buf + buf->pos, p, ssz); - p += ssz; - sz -= ssz; - buf->pos += ssz; - - if ( ! md_buf_flush(buf)) - return(0); + return(leave_bufs(out, in, 1)); } - (void)memcpy(buf->buf + buf->pos, p, sz); - buf->pos += sz; - return(1); -} - - -static int -md_line(struct md_mbuf *out, const struct md_rbuf *in, - const char *buf, size_t sz) -{ - - /* FIXME: this is just a placeholder function. */ - - assert(buf); - assert(out); - assert(in); - - if ( ! md_buf_puts(out, buf, sz)) - return(1); - if ( ! md_buf_putchar(out, '\n')) - return(1); - - return(0); + return(leave_bufs(out, in, md_run(MD_DUMMY, out, in))); } @@ -354,5 +182,5 @@ usage(void) { extern char *__progname; - (void)printf("usage: %s [-o outfile] infile\n", __progname); + (void)printf("usage: %s [-o outfile] [infile]\n", __progname); } -- cgit