universe@22: #include <errno.h>
universe@1: 
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@19:         inputfile->lines = realloc(inputfile->lines, 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"
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 {
universe@22:         return errno;
universe@22:     }
universe@22: }
universe@22: 
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;
universe@22:     
universe@19:     highlighter_t highlighter;
universe@20:     memset(&highlighter, 0, sizeof(highlighter));
universe@19:     highlighter.isdirective = iscdirective;
universe@19:     highlighter.istype = isctype;
universe@19:     highlighter.keywords = ckeywords;
universe@21:     highlighter.parser = cparseline;
universe@19: 
universe@19:     char optc;
universe@22:     while ((optc = getopt(argc, argv, "hjo: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':
universe@19:                 highlighter.isdirective = isjdirective;
universe@19:                 highlighter.istype = isjtype;
universe@19:                 highlighter.keywords = jkeywords;
universe@21:                 highlighter.parser = jparseline;
universe@19:                 break;
universe@19:             case 'p':
universe@19:                 settings.highlight = 0;
universe@19:                 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");
universe@22:                 return errno;
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");
universe@22:             retcode = errno;
universe@22:             goto prog_end;
universe@22:         }
universe@19: 
universe@19:         inputfile_t *inputfile = readinput(settings.infilename);
universe@19:         if (inputfile) {
universe@19:             char *line;
universe@19:             if (settings.highlight) {
universe@19:                 line = (char*) malloc(inputfile->maxlinewidth*64);
universe@19:             } else {
universe@19:                 line = NULL;
universe@19:             }
universe@19:             fprintf(fout, "<pre>\n");
universe@19:             int lnw = lnint(inputfile->count);
universe@19:             for (int i = 0 ; i < inputfile->count ; i++) {
universe@19:                 if (settings.highlight) {
universe@21:                     highlighter.parser(inputfile->lines[i], line, &highlighter);
universe@19:                 } else {
universe@19:                     line = inputfile->lines[i];
universe@19:                 }
universe@19:                 fprintf(fout, "<span class=\"c2html-lineno\">%*d:</span> %s",
universe@19:                     lnw, i+1, line);
universe@19:             }
universe@19:             if (settings.highlight) {
universe@19:                 free(line);
universe@19:             }
universe@19:             fprintf(fout, "</pre>\n");
universe@19: 
universe@22:             freeinputfilebuffer(inputfile);
universe@22:             
universe@22:             if (copyfile(settings.footerfile, fout)) {
universe@22:                 perror("Error opening footer file");
universe@22:                 retcode = errno;
universe@19:             }
universe@22:         } else {
universe@22:             perror("Error opening input file");
universe@22:             retcode = errno;
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: