/ Hex Artifact Content
Login

Artifact e469138f4be3e92df6173b79b3b216ab6e17b407:


0000: 2f 2a 0a 2a 2a 20 54 68 69 73 20 6d 6f 64 75 6c  /*.** This modul
0010: 65 20 69 6e 74 65 72 66 61 63 65 73 20 53 51 4c  e interfaces SQL
0020: 69 74 65 20 74 6f 20 74 68 65 20 47 6f 6f 67 6c  ite to the Googl
0030: 65 20 4f 53 53 2d 46 75 7a 7a 2c 20 66 75 7a 7a  e OSS-Fuzz, fuzz
0040: 65 72 20 61 73 20 61 20 73 65 72 76 69 63 65 2e  er as a service.
0050: 0a 2a 2a 20 28 68 74 74 70 73 3a 2f 2f 67 69 74  .** (https://git
0060: 68 75 62 2e 63 6f 6d 2f 67 6f 6f 67 6c 65 2f 6f  hub.com/google/o
0070: 73 73 2d 66 75 7a 7a 29 0a 2a 2f 0a 23 69 6e 63  ss-fuzz).*/.#inc
0080: 6c 75 64 65 20 3c 73 74 64 64 65 66 2e 68 3e 0a  lude <stddef.h>.
0090: 23 69 6e 63 6c 75 64 65 20 3c 73 74 64 69 6e 74  #include <stdint
00a0: 2e 68 3e 0a 23 69 6e 63 6c 75 64 65 20 22 73 71  .h>.#include "sq
00b0: 6c 69 74 65 33 2e 68 22 0a 0a 23 69 66 6e 64 65  lite3.h"..#ifnde
00c0: 66 20 53 51 4c 49 54 45 5f 4f 4d 49 54 5f 50 52  f SQLITE_OMIT_PR
00d0: 4f 47 52 45 53 53 5f 43 41 4c 4c 42 41 43 4b 0a  OGRESS_CALLBACK.
00e0: 2f 2a 0a 2a 2a 20 50 72 6f 67 72 65 73 73 20 68  /*.** Progress h
00f0: 61 6e 64 6c 65 72 20 63 61 6c 6c 62 61 63 6b 0a  andler callback.
0100: 2a 2f 0a 73 74 61 74 69 63 20 69 6e 74 20 70 72  */.static int pr
0110: 6f 67 72 65 73 73 5f 68 61 6e 64 6c 65 72 28 76  ogress_handler(v
0120: 6f 69 64 20 2a 70 52 65 74 75 72 6e 29 20 7b 0a  oid *pReturn) {.
0130: 20 20 72 65 74 75 72 6e 20 2a 28 69 6e 74 2a 29    return *(int*)
0140: 70 52 65 74 75 72 6e 3b 0a 7d 0a 23 65 6e 64 69  pReturn;.}.#endi
0150: 66 0a 0a 2f 2a 0a 2a 2a 20 43 61 6c 6c 62 61 63  f../*.** Callbac
0160: 6b 20 66 6f 72 20 73 71 6c 69 74 65 33 5f 65 78  k for sqlite3_ex
0170: 65 63 28 29 2e 0a 2a 2f 0a 73 74 61 74 69 63 20  ec()..*/.static 
0180: 69 6e 74 20 65 78 65 63 5f 68 61 6e 64 6c 65 72  int exec_handler
0190: 28 76 6f 69 64 20 2a 70 43 6e 74 2c 20 69 6e 74  (void *pCnt, int
01a0: 20 61 72 67 63 2c 20 63 68 61 72 20 2a 2a 61 72   argc, char **ar
01b0: 67 76 2c 20 63 68 61 72 20 2a 2a 6e 61 6d 65 76  gv, char **namev
01c0: 29 7b 0a 20 20 69 6e 74 20 69 3b 0a 20 20 69 66  ){.  int i;.  if
01d0: 28 20 61 72 67 76 20 29 7b 0a 20 20 20 20 66 6f  ( argv ){.    fo
01e0: 72 28 69 3d 30 3b 20 69 3c 61 72 67 63 3b 20 69  r(i=0; i<argc; i
01f0: 2b 2b 29 20 73 71 6c 69 74 65 33 5f 66 72 65 65  ++) sqlite3_free
0200: 28 73 71 6c 69 74 65 33 5f 6d 70 72 69 6e 74 66  (sqlite3_mprintf
0210: 28 22 25 73 22 2c 20 61 72 67 76 5b 69 5d 29 29  ("%s", argv[i]))
0220: 3b 0a 20 20 7d 0a 20 20 72 65 74 75 72 6e 20 28  ;.  }.  return (
0230: 28 2a 28 69 6e 74 2a 29 70 43 6e 74 29 2d 2d 29  (*(int*)pCnt)--)
0240: 3c 3d 30 3b 0a 7d 0a 0a 2f 2a 0a 2a 2a 20 4d 61  <=0;.}../*.** Ma
0250: 69 6e 20 65 6e 74 72 79 20 70 6f 69 6e 74 2e 20  in entry point. 
0260: 20 54 68 65 20 66 75 7a 7a 65 72 20 69 6e 76 6f   The fuzzer invo
0270: 6b 65 73 20 74 68 69 73 20 66 75 6e 63 74 69 6f  kes this functio
0280: 6e 20 77 69 74 68 20 65 61 63 68 0a 2a 2a 20 66  n with each.** f
0290: 75 7a 7a 65 64 20 69 6e 70 75 74 2e 0a 2a 2f 0a  uzzed input..*/.
02a0: 69 6e 74 20 4c 4c 56 4d 46 75 7a 7a 65 72 54 65  int LLVMFuzzerTe
02b0: 73 74 4f 6e 65 49 6e 70 75 74 28 63 6f 6e 73 74  stOneInput(const
02c0: 20 75 69 6e 74 38 5f 74 2a 20 64 61 74 61 2c 20   uint8_t* data, 
02d0: 73 69 7a 65 5f 74 20 73 69 7a 65 29 20 7b 0a 20  size_t size) {. 
02e0: 20 69 6e 74 20 70 72 6f 67 72 65 73 73 41 72 67   int progressArg
02f0: 20 3d 20 30 3b 20 20 20 20 20 2f 2a 20 31 20 63   = 0;     /* 1 c
0300: 61 75 73 65 73 20 70 72 6f 67 72 65 73 73 20 68  auses progress h
0310: 61 6e 64 6c 65 72 20 61 62 6f 72 74 20 2a 2f 0a  andler abort */.
0320: 20 20 69 6e 74 20 65 78 65 63 43 6e 74 20 3d 20    int execCnt = 
0330: 30 3b 20 20 20 20 20 20 20 20 20 2f 2a 20 41 62  0;         /* Ab
0340: 6f 72 74 20 72 6f 77 20 63 61 6c 6c 62 61 63 6b  ort row callback
0350: 20 77 68 65 6e 20 63 6f 75 6e 74 20 72 65 61 63   when count reac
0360: 68 65 73 20 7a 65 72 6f 20 2a 2f 0a 20 20 63 68  hes zero */.  ch
0370: 61 72 20 2a 7a 45 72 72 4d 73 67 20 3d 20 30 3b  ar *zErrMsg = 0;
0380: 20 20 20 20 20 20 20 2f 2a 20 45 72 72 6f 72 20         /* Error 
0390: 6d 65 73 73 61 67 65 20 72 65 74 75 72 6e 65 64  message returned
03a0: 20 62 79 20 73 71 6c 69 74 65 5f 65 78 65 63 28   by sqlite_exec(
03b0: 29 20 2a 2f 0a 20 20 73 71 6c 69 74 65 33 20 2a  ) */.  sqlite3 *
03c0: 64 62 3b 20 20 20 20 20 20 20 20 20 20 20 20 20  db;             
03d0: 2f 2a 20 54 68 65 20 64 61 74 61 62 61 73 65 20  /* The database 
03e0: 63 6f 6e 6e 65 63 74 69 6f 6e 20 2a 2f 0a 20 20  connection */.  
03f0: 75 69 6e 74 38 5f 74 20 75 53 65 6c 65 63 74 6f  uint8_t uSelecto
0400: 72 3b 20 20 20 20 20 20 20 2f 2a 20 46 69 72 73  r;       /* Firs
0410: 74 20 62 79 74 65 20 6f 66 20 69 6e 70 75 74 20  t byte of input 
0420: 64 61 74 61 5b 5d 20 2a 2f 0a 20 20 69 6e 74 20  data[] */.  int 
0430: 72 63 3b 20 20 20 20 20 20 20 20 20 20 20 20 20  rc;             
0440: 20 20 20 20 20 2f 2a 20 52 65 74 75 72 6e 20 63       /* Return c
0450: 6f 64 65 20 66 72 6f 6d 20 76 61 72 69 6f 75 73  ode from various
0460: 20 69 6e 74 65 72 66 61 63 65 73 20 2a 2f 0a 20   interfaces */. 
0470: 20 63 68 61 72 20 2a 7a 53 71 6c 3b 20 20 20 20   char *zSql;    
0480: 20 20 20 20 20 20 20 20 20 20 2f 2a 20 5a 65 72            /* Zer
0490: 6f 2d 74 65 72 6d 69 6e 61 74 65 64 20 63 6f 70  o-terminated cop
04a0: 79 20 6f 66 20 64 61 74 61 5b 5d 20 2a 2f 0a 0a  y of data[] */..
04b0: 20 20 69 66 28 20 73 69 7a 65 3c 33 20 29 20 72    if( size<3 ) r
04c0: 65 74 75 72 6e 20 30 3b 20 20 20 2f 2a 20 45 61  eturn 0;   /* Ea
04d0: 72 6c 79 20 6f 75 74 20 69 66 20 75 6e 73 75 66  rly out if unsuf
04e0: 66 69 63 69 65 6e 74 20 64 61 74 61 20 2a 2f 0a  ficient data */.
04f0: 0a 20 20 2f 2a 20 45 78 74 72 61 63 74 20 74 68  .  /* Extract th
0500: 65 20 73 65 6c 65 63 74 6f 72 20 62 79 74 65 20  e selector byte 
0510: 66 72 6f 6d 20 74 68 65 20 62 65 67 69 6e 6e 69  from the beginni
0520: 6e 67 20 6f 66 20 74 68 65 20 69 6e 70 75 74 2e  ng of the input.
0530: 20 20 42 75 74 20 6f 6e 6c 79 0a 20 20 2a 2a 20    But only.  ** 
0540: 64 6f 20 74 68 69 73 20 69 66 20 74 68 65 20 73  do this if the s
0550: 65 63 6f 6e 64 20 62 79 74 65 20 69 73 20 61 20  econd byte is a 
0560: 5c 6e 2e 20 20 49 66 20 74 68 65 20 73 65 63 6f  \n.  If the seco
0570: 6e 64 20 62 79 74 65 20 69 73 20 6e 6f 74 20 5c  nd byte is not \
0580: 6e 2c 0a 20 20 2a 2a 20 74 68 65 6e 20 75 73 65  n,.  ** then use
0590: 20 61 20 64 65 66 61 75 6c 74 20 73 65 6c 65 63   a default selec
05a0: 74 6f 72 20 2a 2f 0a 20 20 69 66 28 20 64 61 74  tor */.  if( dat
05b0: 61 5b 31 5d 3d 3d 27 5c 6e 27 20 29 7b 0a 20 20  a[1]=='\n' ){.  
05c0: 20 20 75 53 65 6c 65 63 74 6f 72 20 3d 20 64 61    uSelector = da
05d0: 74 61 5b 30 5d 3b 20 20 64 61 74 61 20 2b 3d 20  ta[0];  data += 
05e0: 32 3b 20 73 69 7a 65 20 2d 3d 20 32 3b 0a 20 20  2; size -= 2;.  
05f0: 7d 65 6c 73 65 7b 0a 20 20 20 20 75 53 65 6c 65  }else{.    uSele
0600: 63 74 6f 72 20 3d 20 30 78 66 64 3b 0a 20 20 7d  ctor = 0xfd;.  }
0610: 0a 0a 20 20 2f 2a 20 4f 70 65 6e 20 74 68 65 20  ..  /* Open the 
0620: 64 61 74 61 62 61 73 65 20 63 6f 6e 6e 65 63 74  database connect
0630: 69 6f 6e 2e 20 20 4f 6e 6c 79 20 75 73 65 20 61  ion.  Only use a
0640: 6e 20 69 6e 2d 6d 65 6d 6f 72 79 20 64 61 74 61  n in-memory data
0650: 62 61 73 65 2e 20 2a 2f 0a 20 20 72 63 20 3d 20  base. */.  rc = 
0660: 73 71 6c 69 74 65 33 5f 6f 70 65 6e 5f 76 32 28  sqlite3_open_v2(
0670: 22 66 75 7a 7a 2e 64 62 22 2c 20 26 64 62 2c 0a  "fuzz.db", &db,.
0680: 20 20 20 20 20 20 20 20 20 20 20 53 51 4c 49 54             SQLIT
0690: 45 5f 4f 50 45 4e 5f 52 45 41 44 57 52 49 54 45  E_OPEN_READWRITE
06a0: 20 7c 20 53 51 4c 49 54 45 5f 4f 50 45 4e 5f 43   | SQLITE_OPEN_C
06b0: 52 45 41 54 45 20 7c 20 53 51 4c 49 54 45 5f 4f  REATE | SQLITE_O
06c0: 50 45 4e 5f 4d 45 4d 4f 52 59 2c 20 30 29 3b 0a  PEN_MEMORY, 0);.
06d0: 20 20 69 66 28 20 72 63 20 29 20 72 65 74 75 72    if( rc ) retur
06e0: 6e 20 30 3b 0a 0a 23 69 66 6e 64 65 66 20 53 51  n 0;..#ifndef SQ
06f0: 4c 49 54 45 5f 4f 4d 49 54 5f 50 52 4f 47 52 45  LITE_OMIT_PROGRE
0700: 53 53 5f 43 41 4c 4c 42 41 43 4b 0a 20 20 2f 2a  SS_CALLBACK.  /*
0710: 20 42 69 74 20 30 20 6f 66 20 74 68 65 20 73 65   Bit 0 of the se
0720: 6c 65 63 74 6f 72 20 65 6e 61 62 6c 65 73 20 70  lector enables p
0730: 72 6f 67 72 65 73 73 20 63 61 6c 6c 62 61 63 6b  rogress callback
0740: 73 2e 20 20 42 69 74 20 31 20 69 73 20 74 68 65  s.  Bit 1 is the
0750: 0a 20 20 2a 2a 20 72 65 74 75 72 6e 20 63 6f 64  .  ** return cod
0760: 65 20 66 72 6f 6d 20 70 72 6f 67 72 65 73 73 20  e from progress 
0770: 63 61 6c 6c 62 61 63 6b 73 20 2a 2f 0a 20 20 69  callbacks */.  i
0780: 66 28 20 75 53 65 6c 65 63 74 6f 72 20 26 20 31  f( uSelector & 1
0790: 20 29 7b 0a 20 20 20 20 73 71 6c 69 74 65 33 5f   ){.    sqlite3_
07a0: 70 72 6f 67 72 65 73 73 5f 68 61 6e 64 6c 65 72  progress_handler
07b0: 28 64 62 2c 20 34 2c 20 70 72 6f 67 72 65 73 73  (db, 4, progress
07c0: 5f 68 61 6e 64 6c 65 72 2c 20 28 76 6f 69 64 2a  _handler, (void*
07d0: 29 26 70 72 6f 67 72 65 73 73 41 72 67 29 3b 0a  )&progressArg);.
07e0: 20 20 7d 0a 23 65 6e 64 69 66 0a 20 20 75 53 65    }.#endif.  uSe
07f0: 6c 65 63 74 6f 72 20 3e 3e 3d 20 31 3b 0a 20 20  lector >>= 1;.  
0800: 70 72 6f 67 72 65 73 73 41 72 67 20 3d 20 75 53  progressArg = uS
0810: 65 6c 65 63 74 6f 72 20 26 20 31 3b 20 20 75 53  elector & 1;  uS
0820: 65 6c 65 63 74 6f 72 20 3e 3e 3d 20 31 3b 0a 0a  elector >>= 1;..
0830: 20 20 2f 2a 20 42 69 74 20 32 20 6f 66 20 74 68    /* Bit 2 of th
0840: 65 20 73 65 6c 65 63 74 6f 72 20 65 6e 61 62 6c  e selector enabl
0850: 65 73 20 66 6f 72 65 69 67 6e 20 6b 65 79 20 63  es foreign key c
0860: 6f 6e 73 74 72 61 69 6e 74 73 20 2a 2f 0a 20 20  onstraints */.  
0870: 73 71 6c 69 74 65 33 5f 64 62 5f 63 6f 6e 66 69  sqlite3_db_confi
0880: 67 28 64 62 2c 20 53 51 4c 49 54 45 5f 44 42 43  g(db, SQLITE_DBC
0890: 4f 4e 46 49 47 5f 45 4e 41 42 4c 45 5f 46 4b 45  ONFIG_ENABLE_FKE
08a0: 59 2c 20 75 53 65 6c 65 63 74 6f 72 26 31 2c 20  Y, uSelector&1, 
08b0: 26 72 63 29 3b 0a 20 20 75 53 65 6c 65 63 74 6f  &rc);.  uSelecto
08c0: 72 20 3e 3e 3d 20 31 3b 0a 0a 20 20 2f 2a 20 52  r >>= 1;..  /* R
08d0: 65 6d 61 69 6e 69 6e 67 20 62 69 74 73 20 6f 66  emaining bits of
08e0: 20 74 68 65 20 73 65 6c 65 63 74 6f 72 20 64 65   the selector de
08f0: 74 65 72 6d 69 6e 65 20 61 20 6c 69 6d 69 74 20  termine a limit 
0900: 6f 6e 20 74 68 65 20 6e 75 6d 62 65 72 20 6f 66  on the number of
0910: 0a 20 20 2a 2a 20 6f 75 74 70 75 74 20 72 6f 77  .  ** output row
0920: 73 20 2a 2f 0a 20 20 65 78 65 63 43 6e 74 20 3d  s */.  execCnt =
0930: 20 75 53 65 6c 65 63 74 6f 72 20 2b 20 31 3b 0a   uSelector + 1;.
0940: 0a 20 20 2f 2a 20 52 75 6e 20 74 68 65 20 53 51  .  /* Run the SQ
0950: 4c 2e 20 20 54 68 65 20 73 71 6c 69 74 65 5f 65  L.  The sqlite_e
0960: 78 65 63 28 29 20 69 6e 74 65 72 66 61 63 65 20  xec() interface 
0970: 65 78 70 65 63 74 73 20 61 20 7a 65 72 6f 2d 74  expects a zero-t
0980: 65 72 6d 69 6e 61 74 65 64 0a 20 20 2a 2a 20 73  erminated.  ** s
0990: 74 72 69 6e 67 2c 20 73 6f 20 6d 61 6b 65 20 61  tring, so make a
09a0: 20 63 6f 70 79 2e 20 2a 2f 0a 20 20 7a 53 71 6c   copy. */.  zSql
09b0: 20 3d 20 73 71 6c 69 74 65 33 5f 6d 70 72 69 6e   = sqlite3_mprin
09c0: 74 66 28 22 25 2e 2a 73 22 2c 20 28 69 6e 74 29  tf("%.*s", (int)
09d0: 73 69 7a 65 2c 20 64 61 74 61 29 3b 0a 20 20 73  size, data);.  s
09e0: 71 6c 69 74 65 33 5f 65 78 65 63 28 64 62 2c 20  qlite3_exec(db, 
09f0: 7a 53 71 6c 2c 20 65 78 65 63 5f 68 61 6e 64 6c  zSql, exec_handl
0a00: 65 72 2c 20 28 76 6f 69 64 2a 29 26 65 78 65 63  er, (void*)&exec
0a10: 43 6e 74 2c 20 26 7a 45 72 72 4d 73 67 29 3b 0a  Cnt, &zErrMsg);.
0a20: 0a 20 20 2f 2a 20 43 6c 65 61 6e 75 70 20 61 6e  .  /* Cleanup an
0a30: 64 20 72 65 74 75 72 6e 20 2a 2f 0a 20 20 73 71  d return */.  sq
0a40: 6c 69 74 65 33 5f 66 72 65 65 28 7a 45 72 72 4d  lite3_free(zErrM
0a50: 73 67 29 3b 0a 20 20 73 71 6c 69 74 65 33 5f 66  sg);.  sqlite3_f
0a60: 72 65 65 28 7a 53 71 6c 29 3b 0a 20 20 73 71 6c  ree(zSql);.  sql
0a70: 69 74 65 33 5f 63 6c 6f 73 65 28 64 62 29 3b 0a  ite3_close(db);.
0a80: 20 20 72 65 74 75 72 6e 20 30 3b 0a 7d 0a          return 0;.}.