src/c2html.c

Wed, 10 Jul 2013 13:45:26 +0200

author
Mike Becker <universe@uap-core.de>
date
Wed, 10 Jul 2013 13:45:26 +0200
changeset 12
7ce5c4b51959
parent 11
c59fe73459fd
child 13
fe74bf2d5f27
permissions
-rw-r--r--

option for plaintext

universe@1 1 /*
universe@1 2 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
universe@1 3 *
universe@1 4 * Copyright 2013 Mike Becker. All rights reserved.
universe@1 5 *
universe@1 6 * Redistribution and use in source and binary forms, with or without
universe@1 7 * modification, are permitted provided that the following conditions are met:
universe@1 8 *
universe@1 9 * 1. Redistributions of source code must retain the above copyright
universe@1 10 * notice, this list of conditions and the following disclaimer.
universe@1 11 *
universe@1 12 * 2. Redistributions in binary form must reproduce the above copyright
universe@1 13 * notice, this list of conditions and the following disclaimer in the
universe@1 14 * documentation and/or other materials provided with the distribution.
universe@1 15 *
universe@1 16 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
universe@1 17 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
universe@1 18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
universe@1 19 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
universe@1 20 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
universe@1 21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
universe@1 22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
universe@1 23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
universe@1 24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
universe@1 25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
universe@1 26 * POSSIBILITY OF SUCH DAMAGE.
universe@1 27 *
universe@1 28 */
universe@1 29
universe@1 30 #include <stdio.h>
universe@1 31 #include <stdlib.h>
universe@1 32 #include <string.h>
universe@1 33 #include <fcntl.h>
universe@1 34 #include <unistd.h>
universe@4 35 #include <ctype.h>
universe@4 36
universe@4 37 #define INPUTBUF_SIZE 2048
universe@5 38 #define WORDBUF_SIZE 16
universe@5 39
universe@5 40 const char* keywords[] = {
universe@5 41 "auto", "break", "case", "char", "const", "continue", "default", "do",
universe@5 42 "double", "else", "enum", "extern", "float", "for", "goto", "if", "int",
universe@5 43 "long", "register", "return", "short", "signed", "sizeof", "static", "struct",
universe@5 44 "switch", "typedef", "union", "unsigned", "void", "volatile", "while", NULL
universe@5 45 };
universe@4 46
universe@11 47 typedef struct {
universe@11 48 char* outfilename;
universe@11 49 char* infilename;
universe@12 50 int highlight;
universe@11 51 } settings_t;
universe@4 52
universe@4 53 typedef struct {
universe@4 54 size_t count;
universe@4 55 size_t capacity;
universe@4 56 size_t maxlinewidth;
universe@4 57 char** lines;
universe@4 58 } inputfile_t;
universe@1 59
universe@1 60 inputfile_t *inputfilebuffer(size_t capacity) {
universe@1 61 inputfile_t *inputfile = (inputfile_t*) malloc(sizeof(inputfile_t));
universe@1 62 inputfile->lines = (char**) malloc(capacity * sizeof(char*));
universe@1 63 inputfile->capacity = capacity;
universe@1 64 inputfile->count = 0;
universe@4 65 inputfile->maxlinewidth = 0;
universe@1 66
universe@1 67 return inputfile;
universe@0 68 }
universe@0 69
universe@1 70 void addline(inputfile_t *inputfile, char* line, size_t width) {
universe@1 71 char *l = (char*) malloc(width+1);
universe@1 72 memcpy(l, line, width);
universe@1 73 l[width] = 0;
universe@1 74 if (inputfile->count >= inputfile->capacity) {
universe@1 75 inputfile->capacity <<= 1;
universe@1 76 inputfile->lines = realloc(inputfile->lines, inputfile->capacity);
universe@1 77 }
universe@1 78 inputfile->lines[inputfile->count] = l;
universe@4 79 inputfile->maxlinewidth =
universe@4 80 width > inputfile->maxlinewidth ? width : inputfile->maxlinewidth;
universe@1 81 inputfile->count++;
universe@1 82 }
universe@1 83
universe@1 84 void freeinputfilebuffer(inputfile_t *inputfile) {
universe@1 85 for (int i = 0 ; i < inputfile->count ; i++) {
universe@1 86 free(inputfile->lines[i]);
universe@1 87 }
universe@1 88 free(inputfile->lines);
universe@1 89 free(inputfile);
universe@1 90 }
universe@1 91
universe@1 92 inputfile_t *readinput(char *filename) {
universe@1 93
universe@1 94 int fd = open(filename, O_RDONLY);
universe@1 95 if (fd == -1) return NULL;
universe@1 96
universe@1 97 inputfile_t *inputfile = inputfilebuffer(512);
universe@1 98
universe@4 99 char buf[INPUTBUF_SIZE];
universe@1 100 ssize_t r;
universe@1 101
universe@4 102 size_t maxlinewidth = 256;
universe@1 103 char *line = (char*) malloc(maxlinewidth);
universe@1 104 size_t col = 0;
universe@1 105
universe@4 106 while ((r = read(fd, buf, INPUTBUF_SIZE)) > 0) {
universe@1 107 for (size_t i = 0 ; i < r ; i++) {
universe@1 108 if (col >= maxlinewidth-4) {
universe@1 109 maxlinewidth <<= 1;
universe@1 110 line = realloc(line, maxlinewidth);
universe@1 111 }
universe@1 112
universe@1 113 if (buf[i] == '\n') {
universe@5 114 line[col++] = '\n';
universe@1 115 line[col] = 0;
universe@1 116 addline(inputfile, line, col);
universe@1 117 col = 0;
universe@1 118 } else {
universe@1 119 line[col++] = buf[i];
universe@1 120 }
universe@1 121 }
universe@1 122 }
universe@1 123
universe@1 124 free(line);
universe@1 125
universe@1 126 close(fd);
universe@1 127
universe@1 128 return inputfile;
universe@1 129 }
universe@1 130
universe@5 131 size_t writeescapedchar(char *dest, size_t dp, char c) {
universe@5 132 if (c == '>') {
universe@5 133 dest[dp++] = '&'; dest[dp++] = 'g';
universe@5 134 dest[dp++] = 't'; dest[dp++] = ';';
universe@5 135 } else if (c == '<') {
universe@5 136 dest[dp++] = '&'; dest[dp++] = 'l';
universe@5 137 dest[dp++] = 't'; dest[dp++] = ';';
universe@5 138 } else {
universe@5 139 dest[dp++] = c;
universe@5 140 }
universe@5 141
universe@5 142 return dp;
universe@5 143 }
universe@5 144
universe@5 145 int iskeyword(char *word) {
universe@5 146 for (int i = 0 ; keywords[i] ; i++) {
universe@5 147 if (strncmp(keywords[i], word, WORDBUF_SIZE) == 0) {
universe@5 148 return 1;
universe@5 149 }
universe@5 150 }
universe@5 151 return 0;
universe@5 152 }
universe@5 153
universe@5 154 #define istype(word, len) (word[len-2] == '_' && word[len-1] == 't')
universe@5 155
universe@4 156 void parseline(char *src, char *dest) {
universe@4 157 size_t sp = 0, dp = 0;
universe@4 158 /* indent */
universe@4 159 while (isspace(src[sp])) {
universe@4 160 dest[dp++] = src[sp++];
universe@4 161 }
universe@5 162 char word[WORDBUF_SIZE];
universe@5 163 memset(word, 0, WORDBUF_SIZE);
universe@5 164 size_t wp = 0;
universe@5 165 int closespan;
universe@4 166 for (char c = src[sp] ; c ; c=src[++sp]) {
universe@5 167 if (!isalnum(c) && c != '_') {
universe@5 168 /* interpret word int_t */
universe@5 169 if (wp > 0 && wp < WORDBUF_SIZE) {
universe@5 170 if (iskeyword(word)) {
universe@5 171 memcpy(&(dest[dp]), "<span class=\"c2html-keyword\">", 29);
universe@5 172 dp += 29;
universe@5 173 closespan = 1;
universe@5 174 } else if (istype(word, wp)) {
universe@5 175 memcpy(&(dest[dp]), "<span class=\"c2html-type\">", 26);
universe@5 176 dp += 26;
universe@5 177 closespan = 1;
universe@5 178 } else {
universe@5 179 closespan = 0;
universe@5 180 }
universe@5 181 for (int i = 0 ; i < wp ; i++) {
universe@5 182 dp = writeescapedchar(dest, dp, word[i]);
universe@5 183 }
universe@5 184 if (closespan) {
universe@5 185 memcpy(&(dest[dp]), "</span>", 7);
universe@5 186 dp += 7;
universe@5 187 }
universe@6 188 memset(word, 0, WORDBUF_SIZE);
universe@6 189 wp = 0;
universe@5 190 }
universe@5 191 dp = writeescapedchar(dest, dp, c);
universe@5 192 } else {
universe@5 193 /* read word */
universe@5 194 if (wp < WORDBUF_SIZE) {
universe@5 195 word[wp++] = c;
universe@5 196 } else if (wp == WORDBUF_SIZE) {
universe@5 197 for (int i = 0 ; i < WORDBUF_SIZE ; i++) {
universe@5 198 dp = writeescapedchar(dest, dp, word[i]);
universe@5 199 }
universe@5 200 wp++;
universe@5 201 dp = writeescapedchar(dest, dp, c);
universe@5 202 } else {
universe@5 203 dp = writeescapedchar(dest, dp, c);
universe@5 204 }
universe@4 205 }
universe@4 206 }
universe@4 207 dest[dp] = 0;
universe@4 208 }
universe@4 209
universe@1 210 void printhelp() {
universe@1 211 printf("Formats source code using HTML.\n\nUsage:\n"
universe@11 212 " c2html [Options] FILE\n\n"
universe@11 213 " Options:\n"
universe@11 214 " -h Prints this help message\n"
universe@11 215 " -o <output> Output file (if not specified, stdout is used)\n"
universe@1 216 "\n");
universe@1 217
universe@1 218
universe@1 219 }
universe@1 220
universe@4 221 int lnint(size_t lnc) {
universe@1 222 int w = 1, p = 1;
universe@1 223 while ((p*=10) < lnc) w++;
universe@1 224 return w;
universe@1 225 }
universe@1 226
universe@1 227 int main(int argc, char** argv) {
universe@1 228
universe@11 229 settings_t settings;
universe@11 230 settings.outfilename = NULL;
universe@12 231 settings.highlight = 1;
universe@11 232
universe@11 233 char optc;
universe@12 234 while ((optc = getopt(argc, argv, "ho:p")) != -1) {
universe@11 235 switch (optc) {
universe@11 236 case 'o':
universe@11 237 if (!(optarg[0] == '-' && optarg[1] == 0)) {
universe@11 238 settings.outfilename = optarg;
universe@11 239 }
universe@11 240 break;
universe@12 241 case 'p':
universe@12 242 settings.highlight = 0;
universe@12 243 break;
universe@11 244 case 'h':
universe@11 245 printhelp();
universe@11 246 return 0;
universe@11 247 default:
universe@11 248 return 1;
universe@11 249 }
universe@11 250 }
universe@11 251
universe@11 252 if (optind != argc-1) {
universe@1 253 printhelp();
universe@11 254 return 1;
universe@1 255 } else {
universe@11 256 settings.infilename = argv[optind];
universe@1 257
universe@11 258 inputfile_t *inputfile = readinput(settings.infilename);
universe@1 259 if (inputfile) {
universe@11 260 FILE *fout;
universe@11 261 if (settings.outfilename) {
universe@11 262 fout = fopen(settings.outfilename, "w");
universe@11 263 } else {
universe@11 264 fout = stdout;
universe@11 265 }
universe@11 266 fprintf(fout, "<pre>\n");
universe@12 267 char *line = (char*) malloc(inputfile->maxlinewidth
universe@12 268 * (settings.highlight?64:0));
universe@4 269 int lnw = lnint(inputfile->count);
universe@1 270 for (int i = 0 ; i < inputfile->count ; i++) {
universe@12 271 if (settings.highlight) {
universe@12 272 parseline(inputfile->lines[i], line);
universe@12 273 } else {
universe@12 274 line = inputfile->lines[i];
universe@12 275 }
universe@11 276 fprintf(fout, "<span class=\"c2html-lineno\">%*d:</span> %s",
universe@4 277 lnw, i, line);
universe@1 278 }
universe@4 279 free(line);
universe@11 280 fprintf(fout, "</pre>\n");
universe@11 281
universe@11 282 if (fout != stdout) {
universe@11 283 fclose(fout);
universe@11 284 }
universe@11 285
universe@1 286 freeinputfilebuffer(inputfile);
universe@1 287 }
universe@1 288
universe@1 289 return 0;
universe@1 290 }
universe@1 291 }
universe@1 292

mercurial