/ Hex Artifact Content
Login

Artifact 76e45655f4046e756064ab10c62e18f2eb846b9f:


0000: 2f 2a 0a 2a 2a 20 32 30 31 34 2d 30 36 2d 31 33  /*.** 2014-06-13
0010: 0a 2a 2a 0a 2a 2a 20 54 68 65 20 61 75 74 68 6f  .**.** The autho
0020: 72 20 64 69 73 63 6c 61 69 6d 73 20 63 6f 70 79  r disclaims copy
0030: 72 69 67 68 74 20 74 6f 20 74 68 69 73 20 73 6f  right to this so
0040: 75 72 63 65 20 63 6f 64 65 2e 20 20 49 6e 20 70  urce code.  In p
0050: 6c 61 63 65 20 6f 66 0a 2a 2a 20 61 20 6c 65 67  lace of.** a leg
0060: 61 6c 20 6e 6f 74 69 63 65 2c 20 68 65 72 65 20  al notice, here 
0070: 69 73 20 61 20 62 6c 65 73 73 69 6e 67 3a 0a 2a  is a blessing:.*
0080: 2a 0a 2a 2a 20 20 20 20 4d 61 79 20 79 6f 75 20  *.**    May you 
0090: 64 6f 20 67 6f 6f 64 20 61 6e 64 20 6e 6f 74 20  do good and not 
00a0: 65 76 69 6c 2e 0a 2a 2a 20 20 20 20 4d 61 79 20  evil..**    May 
00b0: 79 6f 75 20 66 69 6e 64 20 66 6f 72 67 69 76 65  you find forgive
00c0: 6e 65 73 73 20 66 6f 72 20 79 6f 75 72 73 65 6c  ness for yoursel
00d0: 66 20 61 6e 64 20 66 6f 72 67 69 76 65 20 6f 74  f and forgive ot
00e0: 68 65 72 73 2e 0a 2a 2a 20 20 20 20 4d 61 79 20  hers..**    May 
00f0: 79 6f 75 20 73 68 61 72 65 20 66 72 65 65 6c 79  you share freely
0100: 2c 20 6e 65 76 65 72 20 74 61 6b 69 6e 67 20 6d  , never taking m
0110: 6f 72 65 20 74 68 61 6e 20 79 6f 75 20 67 69 76  ore than you giv
0120: 65 2e 0a 2a 2a 0a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a  e..**.**********
0130: 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a  ****************
0140: 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a  ****************
0150: 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a  ****************
0160: 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a 2a  ****************
0170: 2a 2a 2a 2a 0a 2a 2a 0a 2a 2a 20 54 68 69 73 20  ****.**.** This 
0180: 53 51 4c 69 74 65 20 65 78 74 65 6e 73 69 6f 6e  SQLite extension
0190: 20 69 6d 70 6c 65 6d 65 6e 74 73 20 53 51 4c 20   implements SQL 
01a0: 63 6f 6d 70 72 65 73 73 69 6f 6e 20 66 75 6e 63  compression func
01b0: 74 69 6f 6e 73 0a 2a 2a 20 63 6f 6d 70 72 65 73  tions.** compres
01c0: 73 28 29 20 61 6e 64 20 75 6e 63 6f 6d 70 72 65  s() and uncompre
01d0: 73 73 28 29 20 75 73 69 6e 67 20 5a 4c 49 42 2e  ss() using ZLIB.
01e0: 0a 2a 2f 0a 23 69 6e 63 6c 75 64 65 20 22 73 71  .*/.#include "sq
01f0: 6c 69 74 65 33 65 78 74 2e 68 22 0a 53 51 4c 49  lite3ext.h".SQLI
0200: 54 45 5f 45 58 54 45 4e 53 49 4f 4e 5f 49 4e 49  TE_EXTENSION_INI
0210: 54 31 0a 23 69 6e 63 6c 75 64 65 20 3c 7a 6c 69  T1.#include <zli
0220: 62 2e 68 3e 0a 0a 2f 2a 0a 2a 2a 20 49 6d 70 6c  b.h>../*.** Impl
0230: 65 6d 65 6e 74 61 74 69 6f 6e 20 6f 66 20 74 68  ementation of th
0240: 65 20 22 63 6f 6d 70 72 65 73 73 28 58 29 22 20  e "compress(X)" 
0250: 53 51 4c 20 66 75 6e 63 74 69 6f 6e 2e 20 20 54  SQL function.  T
0260: 68 65 20 69 6e 70 75 74 20 58 20 69 73 0a 2a 2a  he input X is.**
0270: 20 63 6f 6d 70 72 65 73 73 65 64 20 75 73 69 6e   compressed usin
0280: 67 20 7a 4c 69 62 20 61 6e 64 20 74 68 65 20 6f  g zLib and the o
0290: 75 74 70 75 74 20 69 73 20 72 65 74 75 72 6e 65  utput is returne
02a0: 64 2e 0a 2a 2a 0a 2a 2a 20 54 68 65 20 6f 75 74  d..**.** The out
02b0: 70 75 74 20 69 73 20 61 20 42 4c 4f 42 20 74 68  put is a BLOB th
02c0: 61 74 20 62 65 67 69 6e 73 20 77 69 74 68 20 61  at begins with a
02d0: 20 76 61 72 69 61 62 6c 65 2d 6c 65 6e 67 74 68   variable-length
02e0: 20 69 6e 74 65 67 65 72 20 74 68 61 74 0a 2a 2a   integer that.**
02f0: 20 69 73 20 74 68 65 20 69 6e 70 75 74 20 73 69   is the input si
0300: 7a 65 20 69 6e 20 62 79 74 65 73 20 28 74 68 65  ze in bytes (the
0310: 20 73 69 7a 65 20 6f 66 20 58 20 62 65 66 6f 72   size of X befor
0320: 65 20 63 6f 6d 70 72 65 73 73 69 6f 6e 29 2e 20  e compression). 
0330: 20 54 68 65 0a 2a 2a 20 76 61 72 69 61 62 6c 65   The.** variable
0340: 2d 6c 65 6e 67 74 68 20 69 6e 74 65 67 65 72 20  -length integer 
0350: 69 73 20 69 6d 70 6c 65 6d 65 6e 74 65 64 20 61  is implemented a
0360: 73 20 31 20 74 6f 20 35 20 62 79 74 65 73 2e 20  s 1 to 5 bytes. 
0370: 20 54 68 65 72 65 20 61 72 65 0a 2a 2a 20 73 65   There are.** se
0380: 76 65 6e 20 62 69 74 73 20 70 65 72 20 69 6e 74  ven bits per int
0390: 65 67 65 72 20 73 74 6f 72 65 64 20 69 6e 20 74  eger stored in t
03a0: 68 65 20 6c 6f 77 65 72 20 73 65 76 65 6e 20 62  he lower seven b
03b0: 69 74 73 20 6f 66 20 65 61 63 68 20 62 79 74 65  its of each byte
03c0: 2e 0a 2a 2a 20 4d 6f 72 65 20 73 69 67 6e 69 66  ..** More signif
03d0: 69 63 61 6e 74 20 62 69 74 73 20 6f 63 63 75 72  icant bits occur
03e0: 20 66 69 72 73 74 2e 20 20 54 68 65 20 6d 6f 73   first.  The mos
03f0: 74 20 73 69 67 6e 69 66 69 63 61 6e 74 20 62 69  t significant bi
0400: 74 20 28 30 78 38 30 29 0a 2a 2a 20 69 73 20 61  t (0x80).** is a
0410: 20 66 6c 61 67 20 74 6f 20 69 6e 64 69 63 61 74   flag to indicat
0420: 65 20 74 68 65 20 65 6e 64 20 6f 66 20 74 68 65  e the end of the
0430: 20 69 6e 74 65 67 65 72 2e 0a 2a 2f 0a 73 74 61   integer..*/.sta
0440: 74 69 63 20 76 6f 69 64 20 63 6f 6d 70 72 65 73  tic void compres
0450: 73 46 75 6e 63 28 0a 20 20 73 71 6c 69 74 65 33  sFunc(.  sqlite3
0460: 5f 63 6f 6e 74 65 78 74 20 2a 63 6f 6e 74 65 78  _context *contex
0470: 74 2c 0a 20 20 69 6e 74 20 61 72 67 63 2c 0a 20  t,.  int argc,. 
0480: 20 73 71 6c 69 74 65 33 5f 76 61 6c 75 65 20 2a   sqlite3_value *
0490: 2a 61 72 67 76 0a 29 7b 0a 20 20 63 6f 6e 73 74  *argv.){.  const
04a0: 20 75 6e 73 69 67 6e 65 64 20 63 68 61 72 20 2a   unsigned char *
04b0: 70 49 6e 3b 0a 20 20 75 6e 73 69 67 6e 65 64 20  pIn;.  unsigned 
04c0: 63 68 61 72 20 2a 70 4f 75 74 3b 0a 20 20 75 6e  char *pOut;.  un
04d0: 73 69 67 6e 65 64 20 69 6e 74 20 6e 49 6e 3b 0a  signed int nIn;.
04e0: 20 20 75 6e 73 69 67 6e 65 64 20 6c 6f 6e 67 20    unsigned long 
04f0: 69 6e 74 20 6e 4f 75 74 3b 0a 20 20 75 6e 73 69  int nOut;.  unsi
0500: 67 6e 65 64 20 63 68 61 72 20 78 5b 38 5d 3b 0a  gned char x[8];.
0510: 20 20 69 6e 74 20 69 2c 20 6a 3b 0a 0a 20 20 70    int i, j;..  p
0520: 49 6e 20 3d 20 73 71 6c 69 74 65 33 5f 76 61 6c  In = sqlite3_val
0530: 75 65 5f 62 6c 6f 62 28 61 72 67 76 5b 30 5d 29  ue_blob(argv[0])
0540: 3b 0a 20 20 6e 49 6e 20 3d 20 73 71 6c 69 74 65  ;.  nIn = sqlite
0550: 33 5f 76 61 6c 75 65 5f 62 79 74 65 73 28 61 72  3_value_bytes(ar
0560: 67 76 5b 30 5d 29 3b 0a 20 20 6e 4f 75 74 20 3d  gv[0]);.  nOut =
0570: 20 31 33 20 2b 20 6e 49 6e 20 2b 20 28 6e 49 6e   13 + nIn + (nIn
0580: 2b 39 39 39 29 2f 31 30 30 30 3b 0a 20 20 70 4f  +999)/1000;.  pO
0590: 75 74 20 3d 20 73 71 6c 69 74 65 33 5f 6d 61 6c  ut = sqlite3_mal
05a0: 6c 6f 63 28 20 6e 4f 75 74 2b 35 20 29 3b 0a 20  loc( nOut+5 );. 
05b0: 20 66 6f 72 28 69 3d 34 3b 20 69 3e 3d 30 3b 20   for(i=4; i>=0; 
05c0: 69 2d 2d 29 7b 0a 20 20 20 20 78 5b 69 5d 20 3d  i--){.    x[i] =
05d0: 20 28 6e 49 6e 20 3e 3e 20 28 37 2a 28 34 2d 69   (nIn >> (7*(4-i
05e0: 29 29 29 26 30 78 37 66 3b 0a 20 20 7d 0a 20 20  )))&0x7f;.  }.  
05f0: 66 6f 72 28 69 3d 30 3b 20 69 3c 34 20 26 26 20  for(i=0; i<4 && 
0600: 78 5b 69 5d 3d 3d 30 3b 20 69 2b 2b 29 7b 7d 0a  x[i]==0; i++){}.
0610: 20 20 66 6f 72 28 6a 3d 30 3b 20 69 3c 3d 34 3b    for(j=0; i<=4;
0620: 20 69 2b 2b 2c 20 6a 2b 2b 29 20 70 4f 75 74 5b   i++, j++) pOut[
0630: 6a 5d 20 3d 20 78 5b 69 5d 3b 0a 20 20 70 4f 75  j] = x[i];.  pOu
0640: 74 5b 6a 2d 31 5d 20 7c 3d 20 30 78 38 30 3b 0a  t[j-1] |= 0x80;.
0650: 20 20 63 6f 6d 70 72 65 73 73 28 26 70 4f 75 74    compress(&pOut
0660: 5b 6a 5d 2c 20 26 6e 4f 75 74 2c 20 70 49 6e 2c  [j], &nOut, pIn,
0670: 20 6e 49 6e 29 3b 0a 20 20 73 71 6c 69 74 65 33   nIn);.  sqlite3
0680: 5f 72 65 73 75 6c 74 5f 62 6c 6f 62 28 63 6f 6e  _result_blob(con
0690: 74 65 78 74 2c 20 70 4f 75 74 2c 20 6e 4f 75 74  text, pOut, nOut
06a0: 2b 6a 2c 20 73 71 6c 69 74 65 33 5f 66 72 65 65  +j, sqlite3_free
06b0: 29 3b 0a 7d 0a 0a 2f 2a 0a 2a 2a 20 49 6d 70 6c  );.}../*.** Impl
06c0: 65 6d 65 6e 74 61 74 69 6f 6e 20 6f 66 20 74 68  ementation of th
06d0: 65 20 22 75 6e 63 6f 6d 70 72 65 73 73 28 58 29  e "uncompress(X)
06e0: 22 20 53 51 4c 20 66 75 6e 63 74 69 6f 6e 2e 20  " SQL function. 
06f0: 20 54 68 65 20 61 72 67 75 6d 65 6e 74 20 58 0a   The argument X.
0700: 2a 2a 20 69 73 20 61 20 62 6c 6f 62 20 77 68 69  ** is a blob whi
0710: 63 68 20 77 61 73 20 6f 62 74 61 69 6e 65 64 20  ch was obtained 
0720: 66 72 6f 6d 20 63 6f 6d 70 72 65 73 73 28 59 29  from compress(Y)
0730: 2e 20 20 54 68 65 20 6f 75 74 70 75 74 20 77 69  .  The output wi
0740: 6c 6c 20 62 65 0a 2a 2a 20 74 68 65 20 76 61 6c  ll be.** the val
0750: 75 65 20 59 2e 0a 2a 2f 0a 73 74 61 74 69 63 20  ue Y..*/.static 
0760: 76 6f 69 64 20 75 6e 63 6f 6d 70 72 65 73 73 46  void uncompressF
0770: 75 6e 63 28 0a 20 20 73 71 6c 69 74 65 33 5f 63  unc(.  sqlite3_c
0780: 6f 6e 74 65 78 74 20 2a 63 6f 6e 74 65 78 74 2c  ontext *context,
0790: 0a 20 20 69 6e 74 20 61 72 67 63 2c 0a 20 20 73  .  int argc,.  s
07a0: 71 6c 69 74 65 33 5f 76 61 6c 75 65 20 2a 2a 61  qlite3_value **a
07b0: 72 67 76 0a 29 7b 0a 20 20 63 6f 6e 73 74 20 75  rgv.){.  const u
07c0: 6e 73 69 67 6e 65 64 20 63 68 61 72 20 2a 70 49  nsigned char *pI
07d0: 6e 3b 0a 20 20 75 6e 73 69 67 6e 65 64 20 63 68  n;.  unsigned ch
07e0: 61 72 20 2a 70 4f 75 74 3b 0a 20 20 75 6e 73 69  ar *pOut;.  unsi
07f0: 67 6e 65 64 20 69 6e 74 20 6e 49 6e 3b 0a 20 20  gned int nIn;.  
0800: 75 6e 73 69 67 6e 65 64 20 6c 6f 6e 67 20 69 6e  unsigned long in
0810: 74 20 6e 4f 75 74 3b 0a 20 20 69 6e 74 20 72 63  t nOut;.  int rc
0820: 3b 0a 20 20 69 6e 74 20 69 3b 0a 0a 20 20 70 49  ;.  int i;..  pI
0830: 6e 20 3d 20 73 71 6c 69 74 65 33 5f 76 61 6c 75  n = sqlite3_valu
0840: 65 5f 62 6c 6f 62 28 61 72 67 76 5b 30 5d 29 3b  e_blob(argv[0]);
0850: 0a 20 20 6e 49 6e 20 3d 20 73 71 6c 69 74 65 33  .  nIn = sqlite3
0860: 5f 76 61 6c 75 65 5f 62 79 74 65 73 28 61 72 67  _value_bytes(arg
0870: 76 5b 30 5d 29 3b 0a 20 20 6e 4f 75 74 20 3d 20  v[0]);.  nOut = 
0880: 30 3b 0a 20 20 66 6f 72 28 69 3d 30 3b 20 69 3c  0;.  for(i=0; i<
0890: 6e 49 6e 20 26 26 20 69 3c 35 3b 20 69 2b 2b 29  nIn && i<5; i++)
08a0: 7b 0a 20 20 20 20 6e 4f 75 74 20 3d 20 28 6e 4f  {.    nOut = (nO
08b0: 75 74 3c 3c 37 29 20 7c 20 28 70 49 6e 5b 69 5d  ut<<7) | (pIn[i]
08c0: 26 30 78 37 66 29 3b 0a 20 20 20 20 69 66 28 20  &0x7f);.    if( 
08d0: 28 70 49 6e 5b 69 5d 26 30 78 38 30 29 21 3d 30  (pIn[i]&0x80)!=0
08e0: 20 29 7b 20 69 2b 2b 3b 20 62 72 65 61 6b 3b 20   ){ i++; break; 
08f0: 7d 0a 20 20 7d 0a 20 20 70 4f 75 74 20 3d 20 73  }.  }.  pOut = s
0900: 71 6c 69 74 65 33 5f 6d 61 6c 6c 6f 63 28 20 6e  qlite3_malloc( n
0910: 4f 75 74 2b 31 20 29 3b 0a 20 20 72 63 20 3d 20  Out+1 );.  rc = 
0920: 75 6e 63 6f 6d 70 72 65 73 73 28 70 4f 75 74 2c  uncompress(pOut,
0930: 20 26 6e 4f 75 74 2c 20 26 70 49 6e 5b 69 5d 2c   &nOut, &pIn[i],
0940: 20 6e 49 6e 2d 69 29 3b 0a 20 20 69 66 28 20 72   nIn-i);.  if( r
0950: 63 3d 3d 5a 5f 4f 4b 20 29 7b 0a 20 20 20 20 73  c==Z_OK ){.    s
0960: 71 6c 69 74 65 33 5f 72 65 73 75 6c 74 5f 62 6c  qlite3_result_bl
0970: 6f 62 28 63 6f 6e 74 65 78 74 2c 20 70 4f 75 74  ob(context, pOut
0980: 2c 20 6e 4f 75 74 2c 20 73 71 6c 69 74 65 33 5f  , nOut, sqlite3_
0990: 66 72 65 65 29 3b 0a 20 20 7d 0a 7d 0a 0a 0a 23  free);.  }.}...#
09a0: 69 66 64 65 66 20 5f 57 49 4e 33 32 0a 5f 5f 64  ifdef _WIN32.__d
09b0: 65 63 6c 73 70 65 63 28 64 6c 6c 65 78 70 6f 72  eclspec(dllexpor
09c0: 74 29 0a 23 65 6e 64 69 66 0a 69 6e 74 20 73 71  t).#endif.int sq
09d0: 6c 69 74 65 33 5f 63 6f 6d 70 72 65 73 73 5f 69  lite3_compress_i
09e0: 6e 69 74 28 0a 20 20 73 71 6c 69 74 65 33 20 2a  nit(.  sqlite3 *
09f0: 64 62 2c 20 0a 20 20 63 68 61 72 20 2a 2a 70 7a  db, .  char **pz
0a00: 45 72 72 4d 73 67 2c 20 0a 20 20 63 6f 6e 73 74  ErrMsg, .  const
0a10: 20 73 71 6c 69 74 65 33 5f 61 70 69 5f 72 6f 75   sqlite3_api_rou
0a20: 74 69 6e 65 73 20 2a 70 41 70 69 0a 29 7b 0a 20  tines *pApi.){. 
0a30: 20 69 6e 74 20 72 63 20 3d 20 53 51 4c 49 54 45   int rc = SQLITE
0a40: 5f 4f 4b 3b 0a 20 20 53 51 4c 49 54 45 5f 45 58  _OK;.  SQLITE_EX
0a50: 54 45 4e 53 49 4f 4e 5f 49 4e 49 54 32 28 70 41  TENSION_INIT2(pA
0a60: 70 69 29 3b 0a 20 20 28 76 6f 69 64 29 70 7a 45  pi);.  (void)pzE
0a70: 72 72 4d 73 67 3b 20 20 2f 2a 20 55 6e 75 73 65  rrMsg;  /* Unuse
0a80: 64 20 70 61 72 61 6d 65 74 65 72 20 2a 2f 0a 20  d parameter */. 
0a90: 20 72 63 20 3d 20 73 71 6c 69 74 65 33 5f 63 72   rc = sqlite3_cr
0aa0: 65 61 74 65 5f 66 75 6e 63 74 69 6f 6e 28 64 62  eate_function(db
0ab0: 2c 20 22 63 6f 6d 70 72 65 73 73 22 2c 20 31 2c  , "compress", 1,
0ac0: 20 53 51 4c 49 54 45 5f 55 54 46 38 2c 20 30 2c   SQLITE_UTF8, 0,
0ad0: 0a 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20  .               
0ae0: 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20                  
0af0: 63 6f 6d 70 72 65 73 73 46 75 6e 63 2c 20 30 2c  compressFunc, 0,
0b00: 20 30 29 3b 0a 20 20 69 66 28 20 72 63 3d 3d 53   0);.  if( rc==S
0b10: 51 4c 49 54 45 5f 4f 4b 20 29 7b 0a 20 20 20 20  QLITE_OK ){.    
0b20: 72 63 20 3d 20 73 71 6c 69 74 65 33 5f 63 72 65  rc = sqlite3_cre
0b30: 61 74 65 5f 66 75 6e 63 74 69 6f 6e 28 64 62 2c  ate_function(db,
0b40: 20 22 75 6e 63 6f 6d 70 72 65 73 73 22 2c 20 31   "uncompress", 1
0b50: 2c 20 53 51 4c 49 54 45 5f 55 54 46 38 2c 20 30  , SQLITE_UTF8, 0
0b60: 2c 0a 20 20 20 20 20 20 20 20 20 20 20 20 20 20  ,.              
0b70: 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20                  
0b80: 20 20 20 75 6e 63 6f 6d 70 72 65 73 73 46 75 6e     uncompressFun
0b90: 63 2c 20 30 2c 20 30 29 3b 0a 20 20 7d 0a 20 20  c, 0, 0);.  }.  
0ba0: 72 65 74 75 72 6e 20 72 63 3b 0a 7d 0a           return rc;.}.