c2html: src/c2html.c@398a7589297f (annotated)

src/c2html.c@398a7589297f (annotated)

src/c2html.c

Wed, 10 Jul 2013 14:38:56 +0200

author: Mike Becker <universe@uap-core.de>
date: Wed, 10 Jul 2013 14:38:56 +0200
changeset 15: 398a7589297f
parent 14: b33629bf4b58
child 16: fa0bcd0444eb
permissions: -rw-r--r--

double free fix

 /*
  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
  *
  * Copyright 2013 Mike Becker. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are met:
  *
  *   1. Redistributions of source code must retain the above copyright
  *      notice, this list of conditions and the following disclaimer.
  *
  *   2. Redistributions in binary form must reproduce the above copyright
  *      notice, this list of conditions and the following disclaimer in the
  *      documentation and/or other materials provided with the distribution.
  *
  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  * POSSIBILITY OF SUCH DAMAGE.
  *
  */
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include <fcntl.h>
 #include <unistd.h>
 #include <ctype.h>
 #define INPUTBUF_SIZE 2048
 #define WORDBUF_SIZE 16
 #define istype(word, len) (word[len-2] == '_' && word[len-1] == 't')
 #define isdirective(word) (word[0] == '#')
 const char* keywords[] = {
   "auto", "break", "case", "char", "const", "continue", "default", "do",
   "double", "else", "enum", "extern", "float", "for", "goto", "if", "int",
   "long", "register", "return", "short", "signed", "sizeof", "static", "struct",
   "switch", "typedef", "union", "unsigned", "void", "volatile", "while", NULL
 };
 typedef struct {
   char* outfilename;
   char* infilename;
   int highlight;
 } settings_t;
 typedef struct {
   size_t count;
   size_t capacity;
   size_t maxlinewidth;
   char** lines;
 } inputfile_t;
 inputfile_t *inputfilebuffer(size_t capacity) {
   inputfile_t *inputfile = (inputfile_t*) malloc(sizeof(inputfile_t));
   inputfile->lines = (char**) malloc(capacity * sizeof(char*));
   inputfile->capacity = capacity;
   inputfile->count = 0;
   inputfile->maxlinewidth = 0;
   return inputfile;
 }
 void addline(inputfile_t *inputfile, char* line, size_t width) {
   char *l = (char*) malloc(width+1);
   memcpy(l, line, width);
   l[width] = 0;
   if (inputfile->count >= inputfile->capacity) {
     inputfile->capacity <<= 1;
     inputfile->lines = realloc(inputfile->lines, inputfile->capacity);
   }
   inputfile->lines[inputfile->count] = l;
   inputfile->maxlinewidth =
           width > inputfile->maxlinewidth ? width : inputfile->maxlinewidth;
   inputfile->count++;
 }
 void freeinputfilebuffer(inputfile_t *inputfile) {
   for (int i = 0 ; i < inputfile->count ; i++) {
     free(inputfile->lines[i]);
   }
   free(inputfile->lines);
   free(inputfile);
 }
 inputfile_t *readinput(char *filename) {
   int fd = open(filename, O_RDONLY);
   if (fd == -1) return NULL;
   inputfile_t *inputfile = inputfilebuffer(512);
   char buf[INPUTBUF_SIZE];
   ssize_t r;
   size_t maxlinewidth = 256;
   char *line = (char*) malloc(maxlinewidth);
   size_t col = 0;
   while ((r = read(fd, buf, INPUTBUF_SIZE)) > 0) {
     for (size_t i = 0 ; i < r ; i++) {
       if (col >= maxlinewidth-4) {
         maxlinewidth <<= 1;
         line = realloc(line, maxlinewidth);
       }
       if (buf[i] == '\n') {
         line[col++] = '\n';
         line[col] = 0;
         addline(inputfile, line, col);
         col = 0;
       } else {
         line[col++] = buf[i];
       }
     }
   }
   free(line);
   close(fd);
   return inputfile;
 }
 size_t writeescapedchar(char *dest, size_t dp, char c) {
   if (c == '>') {
     dest[dp++] = '&'; dest[dp++] = 'g';
     dest[dp++] = 't'; dest[dp++] = ';';
   } else if (c == '<') {
     dest[dp++] = '&'; dest[dp++] = 'l';
     dest[dp++] = 't'; dest[dp++] = ';';
   } else {
     dest[dp++] = c;
   }
   return dp;
 }
 int iskeyword(char *word) {
   for (int i = 0 ; keywords[i] ; i++) {
     if (strncmp(keywords[i], word, WORDBUF_SIZE) == 0) {
       return 1;
     }
   }
   return 0;
 }
 int iscapsonly(char *word, size_t wp) {
   for (size_t i = 0 ; i < wp ; i++) {
     if (!isupper(word[i]) && word[i] != '_') {
       return 0;
     }
   }
   return 1;
 }
 void parseline(char *src, char *dest) {
   size_t sp = 0, dp = 0;
   /* indent */
   while (isspace(src[sp])) {
     dest[dp++] = src[sp++];
   }
   static char word[WORDBUF_SIZE];
   static char includefile[FILENAME_MAX];
   memset(word, 0, WORDBUF_SIZE);
   size_t wp = 0, ifp = 0;
   int isstring = 0, iscomment = 0, isinclude = 0, parseinclude = 0;
   static int iscommentml;
   int isescaping = 0;
   if (iscommentml) {
     iscomment = 1;
     memcpy(&(dest[dp]), "<span class=\"c2html-comment\">", 29);
     dp += 29;
   }
   for (char c = src[sp] ; c ; c=src[++sp]) {
     /* comments */
     if (c == '/') {
       if (iscommentml && sp > 0 && src[sp-1] == '*') {
         iscomment = 0;
         iscommentml = 0;
         memcpy(&(dest[dp]), "/</span>", 8);
         dp += 8;
         continue;
       } else if (!iscomment && (src[sp+1] == '/' || src[sp+1] == '*')) {
         iscomment = 1;
         iscommentml = (src[sp+1] == '*');
         memcpy(&(dest[dp]), "<span class=\"c2html-comment\">", 29);
         dp += 29;
       }
     }
     if (iscomment) {
       if (c == '\n') {
         memcpy(&(dest[dp]), "</span>", 7);
         dp += 7;
       }
       dp = writeescapedchar(dest, dp, c);
     } else if (isinclude) {
       if (c == '<') {
         memcpy(&(dest[dp]), "<span class=\"c2html-stdinclude\">", 32);
         dp += 32;
         dp = writeescapedchar(dest, dp, c);
       } else if (c == '\"') {
         if (parseinclude) {
           dest[dp++] = '\"';
           dest[dp++] = '>';
           memcpy(&(dest[dp]), includefile, ifp);
           dp += ifp;
           dp = writeescapedchar(dest, dp, c);
           memcpy(&(dest[dp]), "</a>", 4);
           dp += 4;
           parseinclude = 0;
         } else {
           memcpy(&(dest[dp]), "<a class=\"c2html-userinclude\" href=", 35);
           dp += 35;
           dp = writeescapedchar(dest, dp, c);
           ifp = 0;
           includefile[ifp++] = '\"';
           parseinclude = 1;
         }
       } else if (c == '>') {
         dp = writeescapedchar(dest, dp, c);
         memcpy(&(dest[dp]), "</span>", 7);
         dp += 7;
       } else {
         if (parseinclude) {
           includefile[ifp++] = c;
         }
         dp = writeescapedchar(dest, dp, c);
       }
     } else {
       /* strings */
       if (!isescaping && (c == '\'' || c == '\"')) {
         isstring ^= 1;
         if (isstring) {
           memcpy(&(dest[dp]), "<span class=\"c2html-string\">", 28);
           dp += 28;
           dp = writeescapedchar(dest, dp, c);
         } else {
           dp = writeescapedchar(dest, dp, c);
           memcpy(&(dest[dp]), "</span>", 7);
           dp += 7;
         }
       } else {
         if (isstring) {
           dp = writeescapedchar(dest, dp, c);
         } else if (!isalnum(c) && c != '_' && c != '#' && c != '.') {
           /* interpret word int_t */
           if (wp > 0 && wp < WORDBUF_SIZE) {
             int closespan = 1;
             if (iskeyword(word)) {
               memcpy(&(dest[dp]), "<span class=\"c2html-keyword\">", 29);
               dp += 29;
             } else if (istype(word, wp)) {
               memcpy(&(dest[dp]), "<span class=\"c2html-type\">", 26);
               dp += 26;
             } else if (isdirective(word)) {
               isinclude = !strncmp("#include", word, WORDBUF_SIZE);
               memcpy(&(dest[dp]), "<span class=\"c2html-directive\">", 31);
               dp += 31;
             } else if (iscapsonly(word, wp)) {
               memcpy(&(dest[dp]), "<span class=\"c2html-macroconst\">", 32);
               dp += 32;
             } else {
               closespan = 0;
             }
             for (int i = 0 ; i < wp ; i++) {
               dp = writeescapedchar(dest, dp, word[i]);
             }
             if (closespan) {
               memcpy(&(dest[dp]), "</span>", 7);
               dp += 7;
             }
           }
           memset(word, 0, WORDBUF_SIZE);
           wp = 0;
           dp = writeescapedchar(dest, dp, c);
         } else {
           /* read word */
           if (wp < WORDBUF_SIZE) {
             word[wp++] = c;
           } else if (wp == WORDBUF_SIZE) {
             for (int i = 0 ; i < WORDBUF_SIZE ; i++) {
               dp = writeescapedchar(dest, dp, word[i]);
             }
             wp++;
             dp = writeescapedchar(dest, dp, c);
           } else {
             dp = writeescapedchar(dest, dp, c);
           }
         }
       }
       isescaping = !isescaping & (c == '\\');
     }
   }
   dest[dp] = 0;
 }
 void printhelp() {
   printf("Formats source code using HTML.\n\nUsage:\n"
       "  c2html [Options] FILE\n\n"
       " Options:\n"
       "  -h                    Prints this help message\n"
       "  -o <output>           Output file (if not specified, stdout is used)\n"
       "  -p                    Disable highlighting (plain text)\n"
       "\n");
 }
 int lnint(size_t lnc) {
   int w = 1, p = 1;
   while ((p*=10) < lnc) w++;
   return w;
 }
 int main(int argc, char** argv) {
   settings_t settings;
   settings.outfilename = NULL;
   settings.highlight = 1;
   char optc;
   while ((optc = getopt(argc, argv, "ho:p")) != -1) {
     switch (optc) {
       case 'o':
         if (!(optarg[0] == '-' && optarg[1] == 0)) {
           settings.outfilename = optarg;
         }
         break;
       case 'p':
         settings.highlight = 0;
         break;
       case 'h':
         printhelp();
         return 0;
       default:
         return 1;
     }
   }
   if (optind != argc-1) {
     printhelp();
     return 1;
   } else {
     settings.infilename = argv[optind];
     inputfile_t *inputfile = readinput(settings.infilename);
     if (inputfile) {
       FILE *fout;
       char *line;
       if (settings.highlight) {
         line = (char*) malloc(inputfile->maxlinewidth*64);
       } else {
         line = NULL;
       }
       if (settings.outfilename) {
         fout = fopen(settings.outfilename, "w");
       } else {
         fout = stdout;
       }
       fprintf(fout, "<pre>\n");
       int lnw = lnint(inputfile->count);
       for (int i = 0 ; i < inputfile->count ; i++) {
         if (settings.highlight) {
           parseline(inputfile->lines[i], line);
         } else {
           line = inputfile->lines[i];
         }
         fprintf(fout, "<span class=\"c2html-lineno\">%*d:</span> %s",
             lnw, i+1, line);
       }
       if (settings.highlight) {
         free(line);
       }
       fprintf(fout, "</pre>\n");
       if (fout != stdout) {
         fclose(fout);
       }
       freeinputfilebuffer(inputfile);
     }
     return 0;
   }
 }

Mercurial > hg > c2html / annotate

src/c2html.c@398a7589297f (annotated)

src/c2html.c