1 /* 2 ** This is a test interface for the ossfuzz.c module. The ossfuzz.c module 3 ** is an adaptor for OSS-FUZZ. (https://github.com/google/oss-fuzz) 4 ** 5 ** This program links against ossfuzz.c. It reads files named on the 6 ** command line and passes them one by one into ossfuzz.c. 7 */ 8 #include <stddef.h> 9 #include <stdint.h> 10 #include <stdio.h> 11 #include <stdlib.h> 12 #include <string.h> 13 #include "sqlite3.h" 14 15 /* 16 ** The entry point in ossfuzz.c that this routine will be calling 17 */ 18 int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size); 19 20 /* Must match equivalent #defines in ossfuzz.c */ 21 #define FUZZ_SQL_TRACE 0x0001 /* Set an sqlite3_trace() callback */ 22 #define FUZZ_SHOW_MAX_DELAY 0x0002 /* Show maximum progress callback delay */ 23 #define FUZZ_SHOW_ERRORS 0x0004 /* Show SQL errors */ 24 extern void ossfuzz_set_debug_flags(unsigned); 25 26 27 28 /* 29 ** Read files named on the command-line and invoke the fuzzer for 30 ** each one. 31 */ 32 int main(int argc, char **argv){ 33 FILE *in; 34 int i; 35 int nErr = 0; 36 uint8_t *zBuf = 0; 37 size_t sz; 38 unsigned mDebug = 0; 39 40 for(i=1; i<argc; i++){ 41 const char *zFilename = argv[i]; 42 if( zFilename[0]=='-' ){ 43 if( zFilename[1]=='-' ) zFilename++; 44 if( strcmp(zFilename, "-show-errors")==0 ){ 45 mDebug |= FUZZ_SHOW_ERRORS; 46 ossfuzz_set_debug_flags(mDebug); 47 }else 48 if( strcmp(zFilename, "-show-max-delay")==0 ){ 49 mDebug |= FUZZ_SHOW_MAX_DELAY; 50 ossfuzz_set_debug_flags(mDebug); 51 }else 52 if( strcmp(zFilename, "-sql-trace")==0 ){ 53 mDebug |= FUZZ_SQL_TRACE; 54 ossfuzz_set_debug_flags(mDebug); 55 }else 56 { 57 printf("unknown option \"%s\"\n", argv[i]); 58 printf("should be one of: --show-errors --show-max-delay" 59 " --sql-trace\n"); 60 exit(1); 61 } 62 continue; 63 } 64 in = fopen(zFilename, "rb"); 65 if( in==0 ){ 66 fprintf(stderr, "cannot open \"%s\"\n", zFilename); 67 nErr++; 68 continue; 69 } 70 fseek(in, 0, SEEK_END); 71 sz = ftell(in); 72 rewind(in); 73 zBuf = realloc(zBuf, sz); 74 if( zBuf==0 ){ 75 fprintf(stderr, "cannot malloc() for %d bytes\n", (int)sz); 76 exit(1); 77 } 78 if( fread(zBuf, sz, 1, in)!=1 ){ 79 fprintf(stderr, "cannot read %d bytes from \"%s\"\n", 80 (int)sz, zFilename); 81 nErr++; 82 }else{ 83 printf("%s... ", zFilename); 84 if( mDebug ) printf("\n"); 85 fflush(stdout); 86 (void)LLVMFuzzerTestOneInput(zBuf, sz); 87 if( mDebug ) printf("%s: ", zFilename); 88 printf("ok\n"); 89 } 90 fclose(in); 91 } 92 free(zBuf); 93 return nErr; 94 } 95