src/c2html.c

Thu, 25 Aug 2016 12:16:57 +0200

author
Mike Becker <universe@uap-core.de>
date
Thu, 25 Aug 2016 12:16:57 +0200
changeset 51
f25ba6fd7a08
parent 50
17408c3607ce
child 52
33ded421c512
permissions
-rw-r--r--

replaces stack buffers with UCX buffers

universe@23 1 /*
universe@23 2 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
universe@23 3 *
universe@35 4 * Copyright 2016 Mike Becker. All rights reserved.
universe@23 5 *
universe@23 6 * Redistribution and use in source and binary forms, with or without
universe@23 7 * modification, are permitted provided that the following conditions are met:
universe@23 8 *
universe@23 9 * 1. Redistributions of source code must retain the above copyright
universe@23 10 * notice, this list of conditions and the following disclaimer.
universe@23 11 *
universe@23 12 * 2. Redistributions in binary form must reproduce the above copyright
universe@23 13 * notice, this list of conditions and the following disclaimer in the
universe@23 14 * documentation and/or other materials provided with the distribution.
universe@23 15 *
universe@23 16 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
universe@23 17 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
universe@23 18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
universe@23 19 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
universe@23 20 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
universe@23 21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
universe@23 22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
universe@23 23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
universe@23 24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
universe@23 25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
universe@23 26 * POSSIBILITY OF SUCH DAMAGE.
universe@23 27 *
universe@23 28 */
universe@22 29 #include "c2html.h"
universe@1 30
universe@39 31 #include "ucx/list.h"
universe@1 32
universe@1 33 void printhelp() {
universe@19 34 printf("Formats source code using HTML.\n\nUsage:\n"
universe@19 35 " c2html [Options] FILE\n\n"
universe@19 36 " Options:\n"
universe@19 37 " -h Prints this help message\n"
universe@19 38 " -j Highlight Java instead of C source code\n"
universe@19 39 " -o <output> Output file (stdout, if not specified)\n"
universe@22 40 " -H <header> Prepend header file\n"
universe@22 41 " -F <footer> Append footer file\n"
universe@19 42 " -p Disable highlighting (plain text)\n"
olaf@24 43 " -l Disable line numbers\n"
universe@37 44 " -V, -v Prints version and exits\n"
universe@19 45 "\n");
universe@1 46 }
universe@1 47
universe@51 48 static void plain_highlighter(char *src, UcxBuffer *dest, HighlighterData *hd) {
universe@45 49 while (*src && *src != '\n') {
universe@48 50 put_htmlescaped(dest, *src);
universe@45 51 src++;
universe@45 52 }
universe@48 53 ucx_buffer_putc(dest, '\n');
universe@45 54 }
universe@45 55
universe@50 56 void formatlines(highlighter_func highlighter,
universe@46 57 UcxList *in, write_func out, void *stream, int showlineno) {
olaf@24 58
universe@44 59 /* compute width of line numbering */
universe@44 60 int lnw;
universe@44 61 if (showlineno) {
universe@44 62 size_t lines = ucx_list_size(in);
universe@44 63 lnw = 1;
universe@44 64 int p = 1;
universe@44 65 while ((p*=10) < lines) lnw++;
universe@44 66 }
universe@44 67
universe@44 68 /* start monospace formatting */
universe@44 69 out("<pre>\n", 1, 6, stream);
universe@39 70
universe@44 71 /* process lines */
universe@39 72 size_t lineno = 0;
universe@51 73 HighlighterData *hd = new_highlighter_data();
universe@51 74 UcxBuffer *line = ucx_buffer_new(NULL, 1024, UCX_BUFFER_AUTOEXTEND);
universe@51 75 if(!line || !hd) {
universe@51 76 perror("Error allocating buffer for output");
universe@51 77 return;
universe@51 78 }
universe@46 79
universe@39 80 UCX_FOREACH(sourceline, in) {
universe@44 81 /* increase line number and clean line buffer */
universe@39 82 lineno++;
universe@44 83 ucx_buffer_clear(line);
universe@44 84
universe@44 85 /* write line number */
universe@44 86 if (showlineno) {
universe@44 87 ucx_bprintf(line, "<span class=\"c2html-lineno\">"
universe@44 88 "<a name=\"l%d\" href=\"#l%d\">%*d </a></span> ",
universe@44 89 lineno, lineno, lnw, lineno);
universe@44 90 }
universe@44 91
universe@45 92 /* process code line */
universe@51 93 highlighter(sourceline->data, line, hd);
olaf@24 94
universe@50 95 /* write code line */
universe@44 96 out(line->space, 1, line->size, stream);
olaf@24 97 }
olaf@24 98
universe@44 99 /* end monospace formatting */
universe@44 100 out("</pre>\n", 1, 7, stream);
universe@44 101
universe@44 102 /* cleanup and return */
universe@51 103 free_highlighter_data(hd);
universe@39 104 ucx_buffer_free(line);
olaf@24 105 }
olaf@24 106
universe@42 107 #define FILEBUF_SIZE 4096
universe@42 108
universe@39 109 enum source_type {
universe@39 110 SOURCE_C,
universe@39 111 SOURCE_JAVA,
universe@39 112 SOURCE_PLAIN
universe@39 113 };
universe@39 114
universe@1 115 int main(int argc, char** argv) {
universe@43 116
universe@43 117 /* Default settings */
universe@39 118 Settings settings;
universe@22 119 memset(&settings, 0, sizeof(settings));
olaf@24 120 settings.showlinenumbers = 1;
universe@39 121 enum source_type sourcetype = SOURCE_C;
universe@19 122
universe@43 123 /* Parse command line */
universe@19 124 char optc;
universe@37 125 while ((optc = getopt(argc, argv, "hljo:pH:F:vV")) != -1) {
universe@19 126 switch (optc) {
universe@19 127 case 'o':
universe@19 128 if (!(optarg[0] == '-' && optarg[1] == 0)) {
universe@19 129 settings.outfilename = optarg;
universe@19 130 }
universe@19 131 break;
universe@22 132 case 'F':
universe@22 133 settings.footerfile = optarg;
universe@22 134 break;
universe@22 135 case 'H':
universe@22 136 settings.headerfile = optarg;
universe@22 137 break;
universe@19 138 case 'j':
universe@39 139 sourcetype = SOURCE_JAVA;
universe@19 140 break;
universe@19 141 case 'p':
universe@39 142 sourcetype = SOURCE_PLAIN;
universe@19 143 break;
olaf@24 144 case 'l':
olaf@24 145 settings.showlinenumbers = 0;
olaf@24 146 break;
universe@19 147 case 'h':
universe@19 148 printhelp();
universe@38 149 return EXIT_SUCCESS;
universe@37 150 case 'v':
universe@37 151 case 'V':
universe@37 152 #ifdef VERSION_DEVELOP
universe@37 153 printf("%d.%d (unstable)\n", VERSION_MAJOR, VERSION_MINOR);
universe@37 154 #else
universe@37 155 printf("%d.%d\n", VERSION_MAJOR, VERSION_MINOR);
universe@37 156 #endif
universe@38 157 return EXIT_SUCCESS;
universe@19 158 default:
universe@38 159 return EXIT_FAILURE;
universe@11 160 }
universe@19 161 }
universe@19 162
universe@19 163 if (optind != argc-1) {
universe@11 164 printhelp();
universe@43 165 return EXIT_FAILURE;
universe@19 166 } else {
universe@47 167 /* Choose highlighter */
universe@46 168 highlighter_func hltr = NULL;
universe@39 169 switch (sourcetype) {
universe@39 170 case SOURCE_C:
universe@48 171 hltr = c_highlighter;
olaf@24 172 break;
universe@39 173 case SOURCE_JAVA:
universe@48 174 hltr = java_highlighter;
olaf@24 175 break;
universe@39 176 case SOURCE_PLAIN:
universe@48 177 hltr = plain_highlighter;
olaf@24 178 break;
universe@39 179 default: /* should be unreachable */
universe@39 180 fprintf(stderr, "error in enum source_type\n");
universe@43 181 return EXIT_FAILURE;
olaf@24 182 }
universe@42 183
universe@43 184 /* Open output file */
universe@43 185 settings.infilename = argv[optind];
universe@43 186 FILE *fout;
universe@43 187 if (settings.outfilename) {
universe@43 188 fout = fopen(settings.outfilename, "w");
universe@43 189 if (!fout) {
universe@43 190 perror("Error opening output file");
universe@43 191 return EXIT_FAILURE;
universe@43 192 }
universe@43 193 } else {
universe@43 194 fout = stdout;
universe@43 195 }
universe@43 196
universe@43 197 /* Allocate file buffer */
universe@43 198 char *filebuf = malloc(FILEBUF_SIZE);
universe@43 199 if (!filebuf) {
universe@43 200 perror("Error allocating file buffer");
universe@43 201 return EXIT_FAILURE;
universe@43 202 }
universe@43 203
universe@43 204 /* Prepend header file */
universe@42 205 {
universe@42 206 FILE *headerfile = fopen(settings.headerfile, "r");
universe@42 207 if (!headerfile) {
universe@42 208 perror("Error opening header file");
universe@43 209 if (fout != stdout) {
universe@43 210 fclose(fout);
universe@43 211 }
universe@43 212 return EXIT_FAILURE;
universe@42 213 }
universe@42 214 ucx_stream_copy(headerfile, fout,
universe@42 215 (read_func) fread, (write_func) fwrite,
universe@42 216 filebuf, FILEBUF_SIZE, (size_t)-1);
universe@42 217 fclose(headerfile);
universe@42 218 }
universe@19 219
universe@43 220 /* Process input file */
universe@39 221 FILE *inputfile = fopen(settings.infilename, "r");
universe@19 222 if (inputfile) {
universe@42 223 UcxBuffer *content = ucx_buffer_new(NULL,
universe@42 224 FILEBUF_SIZE*2, UCX_BUFFER_AUTOEXTEND);
universe@39 225 {
universe@42 226 ucx_stream_copy(inputfile, content, (read_func) fread,
universe@39 227 (write_func) ucx_buffer_write,
universe@42 228 filebuf, FILEBUF_SIZE, (size_t)-1);
universe@39 229 }
universe@39 230 fclose(inputfile);
universe@39 231
universe@42 232 UcxList *inputlines = ucx_list_append(NULL, content->space);
universe@42 233 for (size_t i = 1 ; i < content->size ; i++) {
universe@42 234 if (content->space[i] == '\r') {
universe@42 235 content->space[i] = '\n'; i++;
universe@39 236 }
universe@42 237 if (content->space[i] == '\n' && i+1 < content->size) {
universe@42 238 ucx_list_append(inputlines, content->space+i+1);
universe@39 239 }
universe@39 240 }
universe@39 241
universe@46 242 formatlines(hltr, inputlines,
universe@46 243 (write_func) fwrite, fout, settings.showlinenumbers);
universe@45 244
universe@42 245 ucx_buffer_free(content);
universe@22 246 } else {
universe@22 247 perror("Error opening input file");
universe@43 248 if (fout != stdout) {
universe@43 249 fclose(fout);
universe@43 250 }
universe@43 251 return EXIT_FAILURE;
olaf@24 252 }
olaf@24 253
universe@43 254 /* Append footer file */
universe@42 255 {
universe@42 256 FILE *footerfile = fopen(settings.footerfile, "r");
universe@42 257 if (!footerfile) {
universe@42 258 perror("Error opening footer file");
universe@43 259 if (fout != stdout) {
universe@43 260 fclose(fout);
universe@43 261 }
universe@43 262 return EXIT_FAILURE;
universe@42 263 }
universe@42 264 ucx_stream_copy(footerfile, fout,
universe@42 265 (read_func) fread, (write_func) fwrite,
universe@42 266 filebuf, FILEBUF_SIZE, (size_t)-1);
universe@42 267 fclose(footerfile);
universe@22 268 }
universe@43 269
universe@22 270
universe@42 271 free(filebuf);
universe@19 272
universe@43 273 return EXIT_SUCCESS;
universe@11 274 }
universe@1 275 }
universe@1 276

mercurial