SQLite
Hex Artifact Content
Not logged in

Artifact 122faa92d25033d6c3f07c39231de074ab3d2e83:


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 72 63 3b 0a 20 20 69 6e 74 20    int rc;.  int 
0520: 69 2c 20 6a 3b 0a 0a 20 20 70 49 6e 20 3d 20 73  i, j;..  pIn = s
0530: 71 6c 69 74 65 33 5f 76 61 6c 75 65 5f 62 6c 6f  qlite3_value_blo
0540: 62 28 61 72 67 76 5b 30 5d 29 3b 0a 20 20 6e 49  b(argv[0]);.  nI
0550: 6e 20 3d 20 73 71 6c 69 74 65 33 5f 76 61 6c 75  n = sqlite3_valu
0560: 65 5f 62 79 74 65 73 28 61 72 67 76 5b 30 5d 29  e_bytes(argv[0])
0570: 3b 0a 20 20 6e 4f 75 74 20 3d 20 31 33 20 2b 20  ;.  nOut = 13 + 
0580: 6e 49 6e 20 2b 20 28 6e 49 6e 2b 39 39 39 29 2f  nIn + (nIn+999)/
0590: 31 30 30 30 3b 0a 20 20 70 4f 75 74 20 3d 20 73  1000;.  pOut = s
05a0: 71 6c 69 74 65 33 5f 6d 61 6c 6c 6f 63 28 20 6e  qlite3_malloc( n
05b0: 4f 75 74 2b 35 20 29 3b 0a 20 20 66 6f 72 28 69  Out+5 );.  for(i
05c0: 3d 34 3b 20 69 3e 3d 30 3b 20 69 2d 2d 29 7b 0a  =4; i>=0; i--){.
05d0: 20 20 20 20 78 5b 69 5d 20 3d 20 28 6e 49 6e 20      x[i] = (nIn 
05e0: 3e 3e 20 28 37 2a 28 34 2d 69 29 29 29 26 30 78  >> (7*(4-i)))&0x
05f0: 37 66 3b 0a 20 20 7d 0a 20 20 66 6f 72 28 69 3d  7f;.  }.  for(i=
0600: 30 3b 20 69 3c 34 20 26 26 20 78 5b 69 5d 3d 3d  0; i<4 && x[i]==
0610: 30 3b 20 69 2b 2b 29 7b 7d 0a 20 20 66 6f 72 28  0; i++){}.  for(
0620: 6a 3d 30 3b 20 69 3c 3d 34 3b 20 69 2b 2b 2c 20  j=0; i<=4; i++, 
0630: 6a 2b 2b 29 20 70 4f 75 74 5b 6a 5d 20 3d 20 78  j++) pOut[j] = x
0640: 5b 69 5d 3b 0a 20 20 70 4f 75 74 5b 6a 2d 31 5d  [i];.  pOut[j-1]
0650: 20 7c 3d 20 30 78 38 30 3b 0a 20 20 72 63 20 3d   |= 0x80;.  rc =
0660: 20 63 6f 6d 70 72 65 73 73 28 26 70 4f 75 74 5b   compress(&pOut[
0670: 6a 5d 2c 20 26 6e 4f 75 74 2c 20 70 49 6e 2c 20  j], &nOut, pIn, 
0680: 6e 49 6e 29 3b 0a 20 20 69 66 28 20 72 63 3d 3d  nIn);.  if( rc==
0690: 5a 5f 4f 4b 20 29 7b 0a 20 20 20 20 73 71 6c 69  Z_OK ){.    sqli
06a0: 74 65 33 5f 72 65 73 75 6c 74 5f 62 6c 6f 62 28  te3_result_blob(
06b0: 63 6f 6e 74 65 78 74 2c 20 70 4f 75 74 2c 20 6e  context, pOut, n
06c0: 4f 75 74 2b 6a 2c 20 73 71 6c 69 74 65 33 5f 66  Out+j, sqlite3_f
06d0: 72 65 65 29 3b 0a 20 20 7d 65 6c 73 65 7b 0a 20  ree);.  }else{. 
06e0: 20 20 20 73 71 6c 69 74 65 33 5f 66 72 65 65 28     sqlite3_free(
06f0: 70 4f 75 74 29 3b 0a 20 20 7d 0a 7d 0a 0a 2f 2a  pOut);.  }.}../*
0700: 0a 2a 2a 20 49 6d 70 6c 65 6d 65 6e 74 61 74 69  .** Implementati
0710: 6f 6e 20 6f 66 20 74 68 65 20 22 75 6e 63 6f 6d  on of the "uncom
0720: 70 72 65 73 73 28 58 29 22 20 53 51 4c 20 66 75  press(X)" SQL fu
0730: 6e 63 74 69 6f 6e 2e 20 20 54 68 65 20 61 72 67  nction.  The arg
0740: 75 6d 65 6e 74 20 58 0a 2a 2a 20 69 73 20 61 20  ument X.** is a 
0750: 62 6c 6f 62 20 77 68 69 63 68 20 77 61 73 20 6f  blob which was o
0760: 62 74 61 69 6e 65 64 20 66 72 6f 6d 20 63 6f 6d  btained from com
0770: 70 72 65 73 73 28 59 29 2e 20 20 54 68 65 20 6f  press(Y).  The o
0780: 75 74 70 75 74 20 77 69 6c 6c 20 62 65 0a 2a 2a  utput will be.**
0790: 20 74 68 65 20 76 61 6c 75 65 20 59 2e 0a 2a 2f   the value Y..*/
07a0: 0a 73 74 61 74 69 63 20 76 6f 69 64 20 75 6e 63  .static void unc
07b0: 6f 6d 70 72 65 73 73 46 75 6e 63 28 0a 20 20 73  ompressFunc(.  s
07c0: 71 6c 69 74 65 33 5f 63 6f 6e 74 65 78 74 20 2a  qlite3_context *
07d0: 63 6f 6e 74 65 78 74 2c 0a 20 20 69 6e 74 20 61  context,.  int a
07e0: 72 67 63 2c 0a 20 20 73 71 6c 69 74 65 33 5f 76  rgc,.  sqlite3_v
07f0: 61 6c 75 65 20 2a 2a 61 72 67 76 0a 29 7b 0a 20  alue **argv.){. 
0800: 20 63 6f 6e 73 74 20 75 6e 73 69 67 6e 65 64 20   const unsigned 
0810: 63 68 61 72 20 2a 70 49 6e 3b 0a 20 20 75 6e 73  char *pIn;.  uns
0820: 69 67 6e 65 64 20 63 68 61 72 20 2a 70 4f 75 74  igned char *pOut
0830: 3b 0a 20 20 75 6e 73 69 67 6e 65 64 20 69 6e 74  ;.  unsigned int
0840: 20 6e 49 6e 3b 0a 20 20 75 6e 73 69 67 6e 65 64   nIn;.  unsigned
0850: 20 6c 6f 6e 67 20 69 6e 74 20 6e 4f 75 74 3b 0a   long int nOut;.
0860: 20 20 69 6e 74 20 72 63 3b 0a 20 20 69 6e 74 20    int rc;.  int 
0870: 69 3b 0a 0a 20 20 70 49 6e 20 3d 20 73 71 6c 69  i;..  pIn = sqli
0880: 74 65 33 5f 76 61 6c 75 65 5f 62 6c 6f 62 28 61  te3_value_blob(a
0890: 72 67 76 5b 30 5d 29 3b 0a 20 20 6e 49 6e 20 3d  rgv[0]);.  nIn =
08a0: 20 73 71 6c 69 74 65 33 5f 76 61 6c 75 65 5f 62   sqlite3_value_b
08b0: 79 74 65 73 28 61 72 67 76 5b 30 5d 29 3b 0a 20  ytes(argv[0]);. 
08c0: 20 6e 4f 75 74 20 3d 20 30 3b 0a 20 20 66 6f 72   nOut = 0;.  for
08d0: 28 69 3d 30 3b 20 69 3c 6e 49 6e 20 26 26 20 69  (i=0; i<nIn && i
08e0: 3c 35 3b 20 69 2b 2b 29 7b 0a 20 20 20 20 6e 4f  <5; i++){.    nO
08f0: 75 74 20 3d 20 28 6e 4f 75 74 3c 3c 37 29 20 7c  ut = (nOut<<7) |
0900: 20 28 70 49 6e 5b 69 5d 26 30 78 37 66 29 3b 0a   (pIn[i]&0x7f);.
0910: 20 20 20 20 69 66 28 20 28 70 49 6e 5b 69 5d 26      if( (pIn[i]&
0920: 30 78 38 30 29 21 3d 30 20 29 7b 20 69 2b 2b 3b  0x80)!=0 ){ i++;
0930: 20 62 72 65 61 6b 3b 20 7d 0a 20 20 7d 0a 20 20   break; }.  }.  
0940: 70 4f 75 74 20 3d 20 73 71 6c 69 74 65 33 5f 6d  pOut = sqlite3_m
0950: 61 6c 6c 6f 63 28 20 6e 4f 75 74 2b 31 20 29 3b  alloc( nOut+1 );
0960: 0a 20 20 72 63 20 3d 20 75 6e 63 6f 6d 70 72 65  .  rc = uncompre
0970: 73 73 28 70 4f 75 74 2c 20 26 6e 4f 75 74 2c 20  ss(pOut, &nOut, 
0980: 26 70 49 6e 5b 69 5d 2c 20 6e 49 6e 2d 69 29 3b  &pIn[i], nIn-i);
0990: 0a 20 20 69 66 28 20 72 63 3d 3d 5a 5f 4f 4b 20  .  if( rc==Z_OK 
09a0: 29 7b 0a 20 20 20 20 73 71 6c 69 74 65 33 5f 72  ){.    sqlite3_r
09b0: 65 73 75 6c 74 5f 62 6c 6f 62 28 63 6f 6e 74 65  esult_blob(conte
09c0: 78 74 2c 20 70 4f 75 74 2c 20 6e 4f 75 74 2c 20  xt, pOut, nOut, 
09d0: 73 71 6c 69 74 65 33 5f 66 72 65 65 29 3b 0a 20  sqlite3_free);. 
09e0: 20 7d 65 6c 73 65 7b 0a 20 20 20 20 73 71 6c 69   }else{.    sqli
09f0: 74 65 33 5f 66 72 65 65 28 70 4f 75 74 29 3b 0a  te3_free(pOut);.
0a00: 20 20 7d 0a 7d 0a 0a 0a 23 69 66 64 65 66 20 5f    }.}...#ifdef _
0a10: 57 49 4e 33 32 0a 5f 5f 64 65 63 6c 73 70 65 63  WIN32.__declspec
0a20: 28 64 6c 6c 65 78 70 6f 72 74 29 0a 23 65 6e 64  (dllexport).#end
0a30: 69 66 0a 69 6e 74 20 73 71 6c 69 74 65 33 5f 63  if.int sqlite3_c
0a40: 6f 6d 70 72 65 73 73 5f 69 6e 69 74 28 0a 20 20  ompress_init(.  
0a50: 73 71 6c 69 74 65 33 20 2a 64 62 2c 20 0a 20 20  sqlite3 *db, .  
0a60: 63 68 61 72 20 2a 2a 70 7a 45 72 72 4d 73 67 2c  char **pzErrMsg,
0a70: 20 0a 20 20 63 6f 6e 73 74 20 73 71 6c 69 74 65   .  const sqlite
0a80: 33 5f 61 70 69 5f 72 6f 75 74 69 6e 65 73 20 2a  3_api_routines *
0a90: 70 41 70 69 0a 29 7b 0a 20 20 69 6e 74 20 72 63  pApi.){.  int rc
0aa0: 20 3d 20 53 51 4c 49 54 45 5f 4f 4b 3b 0a 20 20   = SQLITE_OK;.  
0ab0: 53 51 4c 49 54 45 5f 45 58 54 45 4e 53 49 4f 4e  SQLITE_EXTENSION
0ac0: 5f 49 4e 49 54 32 28 70 41 70 69 29 3b 0a 20 20  _INIT2(pApi);.  
0ad0: 28 76 6f 69 64 29 70 7a 45 72 72 4d 73 67 3b 20  (void)pzErrMsg; 
0ae0: 20 2f 2a 20 55 6e 75 73 65 64 20 70 61 72 61 6d   /* Unused param
0af0: 65 74 65 72 20 2a 2f 0a 20 20 72 63 20 3d 20 73  eter */.  rc = s
0b00: 71 6c 69 74 65 33 5f 63 72 65 61 74 65 5f 66 75  qlite3_create_fu
0b10: 6e 63 74 69 6f 6e 28 64 62 2c 20 22 63 6f 6d 70  nction(db, "comp
0b20: 72 65 73 73 22 2c 20 31 2c 20 53 51 4c 49 54 45  ress", 1, SQLITE
0b30: 5f 55 54 46 38 2c 20 30 2c 0a 20 20 20 20 20 20  _UTF8, 0,.      
0b40: 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20                  
0b50: 20 20 20 20 20 20 20 20 20 63 6f 6d 70 72 65 73           compres
0b60: 73 46 75 6e 63 2c 20 30 2c 20 30 29 3b 0a 20 20  sFunc, 0, 0);.  
0b70: 69 66 28 20 72 63 3d 3d 53 51 4c 49 54 45 5f 4f  if( rc==SQLITE_O
0b80: 4b 20 29 7b 0a 20 20 20 20 72 63 20 3d 20 73 71  K ){.    rc = sq
0b90: 6c 69 74 65 33 5f 63 72 65 61 74 65 5f 66 75 6e  lite3_create_fun
0ba0: 63 74 69 6f 6e 28 64 62 2c 20 22 75 6e 63 6f 6d  ction(db, "uncom
0bb0: 70 72 65 73 73 22 2c 20 31 2c 20 53 51 4c 49 54  press", 1, SQLIT
0bc0: 45 5f 55 54 46 38 2c 20 30 2c 0a 20 20 20 20 20  E_UTF8, 0,.     
0bd0: 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20                  
0be0: 20 20 20 20 20 20 20 20 20 20 20 20 75 6e 63 6f              unco
0bf0: 6d 70 72 65 73 73 46 75 6e 63 2c 20 30 2c 20 30  mpressFunc, 0, 0
0c00: 29 3b 0a 20 20 7d 0a 20 20 72 65 74 75 72 6e 20  );.  }.  return 
0c10: 72 63 3b 0a 7d 0a                                rc;.}.