check.c

   1 /*
   2  * Example trivial client program that uses the sparse library
   3  * to tokenize, pre-process and parse a C file, and prints out
   4  * the results.
   5  *
   6  * Copyright (C) 2003 Transmeta Corp, all rights reserved.
   7  */
   8 #include <stdarg.h>
   9 #include <stdlib.h>
  10 #include <stdio.h>
  11 #include <string.h>
  12 #include <ctype.h>
  13 #include <unistd.h>
  14 #include <fcntl.h>
  15
  16 #include "lib.h"
  17 #include "token.h"
  18 #include "parse.h"
  19 #include "symbol.h"
  20 #include "expression.h"
  21
  22 static unsigned int pre_buffer_size = 0;
  23 static unsigned char pre_buffer[8192];
  24
  25 static void add_pre_buffer(const char *fmt, ...)
  26 {
  27         va_list args;
  28         unsigned int size;
  29
  30         va_start(args, fmt);
  31         size = pre_buffer_size;
  32         size += vsnprintf(pre_buffer + size,
  33                 sizeof(pre_buffer) - size,
  34                 fmt, args);
  35         pre_buffer_size = size;
  36         va_end(args);
  37 }
  38
  39 static void handle_switch(char *arg)
  40 {
  41         switch (*arg) {
  42         case 'D': {
  43                 const char *name = arg+1;
  44                 const char *value = "";
  45                 for (;;) {
  46                         char c;
  47                         c = *++arg;
  48                         if (!c)
  49                                 break;
  50                         if (isspace(c) || c == '=') {
  51                                 *arg = '\0';
  52                                 value = arg+1;
  53                                 break;
  54                         }
  55                 }
  56                 add_pre_buffer("#define %s %s\n", name, value);
  57                 return;
  58         }
  59
  60         case 'I':
  61                 add_pre_buffer("#add_include \"%s/\"\n", arg+1);
  62                 return;
  63         default:
  64                 fprintf(stderr, "unknown switch '%s'\n", arg);
  65         }
  66 }
  67
  68 static void clean_up_symbol(struct symbol *sym, void *_parent, int flags)
  69 {
  70         evaluate_symbol(sym);
  71 }
  72
  73 int main(int argc, char **argv)
  74 {
  75         int i, fd;
  76         char *filename = NULL;
  77         struct token *token;
  78
  79         // Initialize symbol stream first, so that we can add defines etc
  80         init_symbols();
  81
  82         for (i = 1; i < argc; i++) {
  83                 char *arg = argv[i];
  84                 if (arg[0] == '-') {
  85                         handle_switch(arg+1);
  86                         continue;
  87                 }
  88                 filename = arg;
  89         }
  90
  91
  92         fd = open(filename, O_RDONLY);
  93         if (fd < 0)
  94                 die("No such file: %s", argv[1]);
  95
  96         // Tokenize the input stream
  97         token = tokenize(filename, fd, NULL);
  98         close(fd);
  99
 100         // Prepend the initial built-in stream
 101         token = tokenize_buffer(pre_buffer, pre_buffer_size, token);
 102
 103         // Pre-process the stream
 104         token = preprocess(token);
 105
 106         // Parse the resulting C code
 107         translation_unit(token, &used_list);
 108
 109         // Do type evaluation and simplify
 110         symbol_iterate(used_list, clean_up_symbol, NULL);
 111         return 0;
 112 }