1 /* $Id: mdocml.c,v 1.17 2008/12/04 11:25:29 kristaps Exp $ */
3 * Copyright (c) 2008 Kristaps Dzonsons <kristaps@kth.se>
5 * Permission to use, copy, modify, and distribute this software for any
6 * purpose with or without fee is hereby granted, provided that the
7 * above copyright notice and this permission notice appear in all
10 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL
11 * WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED
12 * WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE
13 * AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL
14 * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
15 * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER
16 * TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
17 * PERFORMANCE OF THIS SOFTWARE.
19 #include <sys/param.h>
31 #include "libmdocml.h"
33 #define BUFFER_IN_DEF BUFSIZ /* See begin_bufs. */
34 #define BUFFER_OUT_DEF BUFSIZ /* See begin_bufs. */
37 #define CSS "mdocml.css"
39 #define CSS "/usr/local/share/mdocml/mdocml.css"
42 static void usage(void);
44 static int begin_io(const struct md_args
*,
46 static int leave_io(const struct md_buf
*,
47 const struct md_buf
*, int);
48 static int begin_bufs(const struct md_args
*,
49 struct md_buf
*, struct md_buf
*);
50 static int leave_bufs(const struct md_buf
*,
51 const struct md_buf
*, int);
54 main(int argc
, char *argv
[])
65 (void)memset(&args
, 0, sizeof(struct md_args
));
69 while (-1 != (c
= getopt(argc
, argv
, "c:ef:o:vW")))
72 if (args
.type
!= MD_HTML
)
73 errx(1, "-c only valid for -fhtml");
74 args
.params
.html
.css
= optarg
;
77 if (args
.type
!= MD_HTML
)
78 errx(1, "-e only valid for -fhtml");
79 args
.params
.html
.flags
|= HTML_CSS_EMBED
;
82 if (0 == strcmp(optarg
, "html"))
84 else if (0 == strcmp(optarg
, "xml"))
87 errx(1, "invalid filter type");
96 args
.warnings
|= MD_WARN_ALL
;
103 if (MD_HTML
== args
.type
)
104 if (NULL
== args
.params
.html
.css
)
105 args
.params
.html
.css
= CSS
;
113 return(begin_io(&args
, out
? out
: "-", in
? in
: "-"));
118 * Close out file descriptors opened in begin_io. If the descriptor
119 * refers to stdin/stdout, then do nothing.
122 leave_io(const struct md_buf
*out
,
123 const struct md_buf
*in
, int c
)
128 if (-1 != in
->fd
&& -1 == close(in
->fd
)) {
130 warn("%s", in
->name
);
133 if (-1 != out
->fd
&& STDOUT_FILENO
!= out
->fd
&&
134 -1 == close(out
->fd
)) {
136 warn("%s", out
->name
);
139 if (1 == c
&& STDOUT_FILENO
!= out
->fd
)
140 if (-1 == unlink(out
->name
))
141 warn("%s", out
->name
);
148 * Open file descriptors or assign stdin/stdout, if dictated by the "-"
149 * token instead of a filename.
152 begin_io(const struct md_args
*args
, char *out
, char *in
)
157 #define FI_FL O_RDONLY
158 #define FO_FL O_WRONLY|O_CREAT|O_TRUNC
164 bzero(&fi
, sizeof(struct md_buf
));
165 bzero(&fo
, sizeof(struct md_buf
));
167 fi
.fd
= STDIN_FILENO
;
168 fo
.fd
= STDOUT_FILENO
;
173 if (0 != strncmp(fi
.name
, "-", 1))
174 if (-1 == (fi
.fd
= open(fi
.name
, FI_FL
, 0))) {
176 return(leave_io(&fo
, &fi
, 1));
179 if (0 != strncmp(fo
.name
, "-", 1))
180 if (-1 == (fo
.fd
= open(fo
.name
, FO_FL
, 0644))) {
182 return(leave_io(&fo
, &fi
, 1));
185 return(leave_io(&fo
, &fi
, begin_bufs(args
, &fo
, &fi
)));
190 * Free buffers allocated in begin_bufs.
193 leave_bufs(const struct md_buf
*out
,
194 const struct md_buf
*in
, int c
)
207 * Allocate buffers to the maximum of either the input file's blocksize
208 * or BUFFER_IN_DEF/BUFFER_OUT_DEF, which should be around BUFSIZE.
211 begin_bufs(const struct md_args
*args
,
212 struct md_buf
*out
, struct md_buf
*in
)
214 struct stat stin
, stout
;
221 if (-1 == fstat(in
->fd
, &stin
)) {
222 warn("%s", in
->name
);
224 } else if (STDIN_FILENO
!= in
->fd
&& 0 == stin
.st_size
) {
225 warnx("%s: empty file", in
->name
);
227 } else if (-1 == fstat(out
->fd
, &stout
)) {
228 warn("%s", out
->name
);
232 in
->bufsz
= MAX(stin
.st_blksize
, BUFFER_IN_DEF
);
233 out
->bufsz
= MAX(stout
.st_blksize
, BUFFER_OUT_DEF
);
235 if (NULL
== (in
->buf
= malloc(in
->bufsz
))) {
237 return(leave_bufs(out
, in
, 1));
238 } else if (NULL
== (out
->buf
= malloc(out
->bufsz
))) {
240 return(leave_bufs(out
, in
, 1));
243 c
= md_run(args
, out
, in
);
244 return(leave_bufs(out
, in
, -1 == c
? 1 : 0));
251 extern char *__progname
;
253 (void)printf("usage: %s [-vW] [-f filter] [-o outfile] "
254 "[infile]\n", __progname
);