universe@23: /*
universe@23:  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
universe@23:  *
universe@35:  * Copyright 2016 Mike Becker. All rights reserved.
universe@23:  *
universe@23:  * Redistribution and use in source and binary forms, with or without
universe@23:  * modification, are permitted provided that the following conditions are met:
universe@23:  *
universe@23:  *   1. Redistributions of source code must retain the above copyright
universe@23:  *      notice, this list of conditions and the following disclaimer.
universe@23:  *
universe@23:  *   2. Redistributions in binary form must reproduce the above copyright
universe@23:  *      notice, this list of conditions and the following disclaimer in the
universe@23:  *      documentation and/or other materials provided with the distribution.
universe@23:  *
universe@23:  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
universe@23:  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
universe@23:  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
universe@23:  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
universe@23:  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
universe@23:  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
universe@23:  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
universe@23:  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
universe@23:  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
universe@23:  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
universe@23:  * POSSIBILITY OF SUCH DAMAGE.
universe@23:  *
universe@23:  */
universe@22: #include "c2html.h"
universe@1: 
universe@1: inputfile_t *inputfilebuffer(size_t capacity) {
universe@19:     inputfile_t *inputfile = (inputfile_t*) malloc(sizeof(inputfile_t));
universe@19:     inputfile->lines = (char**) malloc(capacity * sizeof(char*));
universe@19:     inputfile->capacity = capacity;
universe@19:     inputfile->count = 0;
universe@19:     inputfile->maxlinewidth = 0;
universe@19: 
universe@19:     return inputfile;
universe@0: }
universe@0: 
universe@1: void addline(inputfile_t *inputfile, char* line, size_t width) {
universe@19:     char *l = (char*) malloc(width+1);
universe@19:     memcpy(l, line, width);
universe@19:     l[width] = 0;
universe@19:     if (inputfile->count >= inputfile->capacity) {
universe@19:         inputfile->capacity <<= 1;
universe@25:         inputfile->lines = realloc(inputfile->lines,
universe@25:             sizeof(char*)*inputfile->capacity);
universe@19:     }
universe@19:     inputfile->lines[inputfile->count] = l;
universe@19:     inputfile->maxlinewidth =
universe@19:         width > inputfile->maxlinewidth ? width : inputfile->maxlinewidth;
universe@19:     inputfile->count++;
universe@1: }
universe@1: 
universe@1: void freeinputfilebuffer(inputfile_t *inputfile) {
universe@19:     for (int i = 0 ; i < inputfile->count ; i++) {
universe@19:         free(inputfile->lines[i]);
universe@19:     }
universe@19:     free(inputfile->lines);
universe@19:     free(inputfile);
universe@1: }
universe@1: 
universe@1: inputfile_t *readinput(char *filename) {
universe@1: 
universe@19:     int fd = open(filename, O_RDONLY);
universe@19:     if (fd == -1) return NULL;
universe@1: 
universe@19:     inputfile_t *inputfile = inputfilebuffer(512);
universe@19: 
universe@19:     char buf[INPUTBUF_SIZE];
universe@19:     ssize_t r;
universe@19: 
universe@19:     size_t maxlinewidth = 256;
universe@19:     char *line = (char*) malloc(maxlinewidth);
universe@19:     size_t col = 0;
universe@19: 
universe@19:     while ((r = read(fd, buf, INPUTBUF_SIZE)) > 0) {
universe@19:         for (size_t i = 0 ; i < r ; i++) {
universe@19:             if (col >= maxlinewidth-4) {
universe@19:                 maxlinewidth <<= 1;
universe@19:                 line = realloc(line, maxlinewidth);
universe@19:             }
universe@19: 
universe@19:             if (buf[i] == '\n') {
universe@19:                 line[col++] = '\n';
universe@19:                 line[col] = 0;
universe@19:                 addline(inputfile, line, col);
universe@19:                 col = 0;
universe@19:             } else {
universe@19:                 line[col++] = buf[i];
universe@19:             }
universe@19:         }
universe@1:     }
universe@19: 
universe@19:     free(line);
universe@19: 
universe@19:     close(fd);
universe@19: 
universe@19:     return inputfile;
universe@1: }
universe@1: 
universe@1: void printhelp() {
universe@19:     printf("Formats source code using HTML.\n\nUsage:\n"
universe@19:         "  c2html [Options] FILE\n\n"
universe@19:         " Options:\n"
universe@19:         "  -h                    Prints this help message\n"
universe@19:         "  -j                    Highlight Java instead of C source code\n"
universe@19:         "  -o <output>           Output file (stdout, if not specified)\n"
universe@22:         "  -H <header>           Prepend header file\n"
universe@22:         "  -F <footer>           Append footer file\n"
universe@19:         "  -p                    Disable highlighting (plain text)\n"
olaf@24:         "  -l                    Disable line numbers\n"
universe@19:         "\n");
universe@19: 
universe@19: 
universe@1: }
universe@1: 
universe@4: int lnint(size_t lnc) {
universe@19:     int w = 1, p = 1;
universe@19:     while ((p*=10) < lnc) w++;
universe@19:     return w;
universe@1: }
universe@1: 
universe@22: int copyfile(char *filename, FILE *dest) {
universe@22:     if (!filename) {
universe@22:         return 0;
universe@22:     }
universe@22:     
universe@22:     FILE *src = fopen(filename, "r");
universe@22:     if (src) {
universe@22:         char buf[4096];
universe@22:         int r;
universe@22:         while ((r = fread(buf, 1, 4096, src)) > 0) {
universe@22:             fwrite(buf, 1, r, dest);
universe@22:         }
universe@22:         fclose(src);
universe@22:         return 0;
universe@22:     } else {
olaf@24:         return -1;
universe@22:     }
universe@22: }
universe@22: 
olaf@24: #define WRITECONST(stream, out, cstr) out(cstr, 1, sizeof(cstr)-1, stream)
olaf@24: int formatfile(
olaf@24:         highlighter_t *highlighter,
olaf@24:         inputfile_t *in,
olaf@24:         fmt_write_func out,
olaf@24:         void *stream,
olaf@24:         _Bool showln) {
olaf@24:     // formats an input file and writes the result to out
olaf@24:     
olaf@24:     char *line = malloc(in->maxlinewidth*64);
olaf@24:     if(!line) {
olaf@24:         return 1;
olaf@24:     }
olaf@24:     WRITECONST(stream, out, "<pre>\n");
olaf@24:     
olaf@24:     int lnw = lnint(in->count);
olaf@24:     for (int i = 0 ; i < in->count ; i++) {
olaf@24:         if (highlighter) {
olaf@24:             highlighter->parser(in->lines[i], line, highlighter);
olaf@24:         } else {
universe@30:             char *c = in->lines[i];
universe@30:             size_t dp = 0;
universe@30:             while (*c) {
universe@30:                 dp = writeescapedchar(line, dp, *c);
universe@30:                 c++;
universe@30:             }
universe@30:             line[dp] = '\0';
olaf@24:         }
universe@30: 
olaf@24:         // write line number
olaf@24:         if (showln) {
olaf@24:             WRITECONST(stream, out, "<span class=\"c2html-lineno\">");
universe@27:             char lnbuf[128];
universe@27:             int len;
universe@27:             // line number link
universe@27:             len = snprintf(lnbuf, 128, "<a name=\"l%d\" href=\"#l%d\">",
universe@27:                 i+1, i+1);
olaf@24:             out(lnbuf, 1, len, stream);
universe@27:             // justified line number
universe@27:             len = snprintf(lnbuf, 128, "%*d ", lnw, i+1);
universe@27:             out(lnbuf, 1, len, stream);
universe@27:             WRITECONST(stream, out, "</a></span> ");
olaf@24:         }
olaf@24:         
olaf@24:         // write formated (or plain) code line
universe@30:         out(line, 1, strlen(line), stream);
olaf@24:     }
olaf@24:     
olaf@24:     WRITECONST(stream, out, "</pre>\n");
olaf@24:     free(line);
olaf@24:     return 0;
olaf@24: }
olaf@24: 
olaf@24: void init_c_highlighter(highlighter_t *highlighter) {
olaf@24:     memset(highlighter, 0, sizeof(highlighter_t));
universe@36:     highlighter->isdirective = check_cdirective;
universe@36:     highlighter->istype = check_ctype;
olaf@24:     highlighter->keywords = ckeywords;
olaf@24:     highlighter->parser = cparseline;
olaf@24: }
olaf@24: 
olaf@24: void init_java_highlighter(highlighter_t *highlighter) {
olaf@24:     memset(highlighter, 0, sizeof(highlighter_t));
universe@36:     highlighter->isdirective = check_jdirective;
universe@36:     highlighter->istype = check_jtype;
olaf@24:     highlighter->keywords = jkeywords;
olaf@24:     highlighter->parser = jparseline;
olaf@24: }
olaf@24: 
universe@1: int main(int argc, char** argv) {
universe@22:     int retcode = EXIT_SUCCESS;
universe@22:     
universe@19:     settings_t settings;
universe@22:     memset(&settings, 0, sizeof(settings));
universe@19:     settings.highlight = 1;
olaf@24:     settings.showlinenumbers = 1;
universe@22:     
olaf@24:     int lang = C2HTML_C;
universe@19: 
universe@19:     char optc;
olaf@24:     while ((optc = getopt(argc, argv, "hljo:pH:F:")) != -1) {
universe@19:         switch (optc) {
universe@19:             case 'o':
universe@19:                 if (!(optarg[0] == '-' && optarg[1] == 0)) {
universe@19:                     settings.outfilename = optarg;
universe@19:                 }
universe@19:                 break;
universe@22:             case 'F':
universe@22:                 settings.footerfile = optarg;
universe@22:                 break;
universe@22:             case 'H':
universe@22:                 settings.headerfile = optarg;
universe@22:                 break;
universe@19:             case 'j':
olaf@24:                 lang = C2HTML_JAVA;
universe@19:                 break;
universe@19:             case 'p':
universe@19:                 settings.highlight = 0;
universe@19:                 break;
olaf@24:             case 'l':
olaf@24:                 settings.showlinenumbers = 0;
olaf@24:                 break;
universe@19:             case 'h':
universe@19:                 printhelp();
universe@19:                 return 0;
universe@19:             default:
universe@19:                 return 1;
universe@11:         }
universe@19:     }
universe@19: 
universe@19:     if (optind != argc-1) {
universe@11:         printhelp();
universe@19:         return 1;
universe@19:     } else {
universe@19:         settings.infilename = argv[optind];
universe@22:         FILE *fout;
universe@22:         if (settings.outfilename) {
universe@22:             fout = fopen(settings.outfilename, "w");
universe@22:             if (!fout) {
universe@22:                 perror("Error opening output file");
olaf@24:                 return -1;
universe@22:             }
universe@22:         } else {
universe@22:             fout = stdout;
universe@22:         }
universe@22:         
universe@22:         if (copyfile(settings.headerfile, fout)) {
universe@22:             perror("Error opening header file");
olaf@24:             retcode = -1;
universe@22:             goto prog_end;
universe@22:         }
olaf@24:         
olaf@24:         highlighter_t highlighter;
olaf@24:         highlighter_t *hptr = &highlighter;
olaf@24:         switch (lang) {
olaf@24:             case C2HTML_C:
olaf@24:                 init_c_highlighter(&highlighter);
olaf@24:                 break;
olaf@24:             case C2HTML_JAVA:
olaf@24:                 init_java_highlighter(&highlighter);
olaf@24:                 break;
olaf@24:             default:
olaf@24:                 hptr = NULL;
olaf@24:                 break;
olaf@24:         }
olaf@24:         if (!settings.highlight) {
olaf@24:             hptr = NULL;
olaf@24:         }
universe@19: 
universe@19:         inputfile_t *inputfile = readinput(settings.infilename);
universe@19:         if (inputfile) {
olaf@24:             formatfile(
olaf@24:                     hptr,
olaf@24:                     inputfile,
olaf@24:                     (fmt_write_func)fwrite,
olaf@24:                     fout,
olaf@24:                     settings.showlinenumbers);
universe@25:             freeinputfilebuffer(inputfile);
universe@22:         } else {
universe@22:             perror("Error opening input file");
olaf@24:             retcode = -1;
olaf@24:         }
olaf@24:         
olaf@24:         if (copyfile(settings.footerfile, fout)) {
olaf@24:             perror("Error opening footer file");
olaf@24:             retcode = -1;
universe@22:         }
universe@22:         
universe@22:         prog_end:        
universe@22:         if (fout != stdout) {
universe@22:             fclose(fout);
universe@19:         }
universe@19: 
universe@22:         return retcode;
universe@11:     }
universe@1: }
universe@1: