SQLite
Hex Artifact Content
Not logged in

Artifact 1ac6f95f99b575907b9b09c81a349114cf9be45a:


0000: 2f 2a 0a 2a 2a 20 32 30 31 33 2d 30 35 2d 31 35  /*.** 2013-05-15
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 61 20 72 6f   implements a ro
01a0: 74 31 33 28 29 20 66 75 6e 63 74 69 6f 6e 20 61  t13() function a
01b0: 6e 64 20 61 20 72 6f 74 31 33 0a 2a 2a 20 63 6f  nd a rot13.** co
01c0: 6c 6c 61 74 69 6e 67 20 73 65 71 75 65 6e 63 65  llating sequence
01d0: 2e 0a 2a 2f 0a 23 69 6e 63 6c 75 64 65 20 22 73  ..*/.#include "s
01e0: 71 6c 69 74 65 33 65 78 74 2e 68 22 0a 53 51 4c  qlite3ext.h".SQL
01f0: 49 54 45 5f 45 58 54 45 4e 53 49 4f 4e 5f 49 4e  ITE_EXTENSION_IN
0200: 49 54 31 0a 23 69 6e 63 6c 75 64 65 20 3c 61 73  IT1.#include <as
0210: 73 65 72 74 2e 68 3e 0a 23 69 6e 63 6c 75 64 65  sert.h>.#include
0220: 20 3c 73 74 72 69 6e 67 2e 68 3e 0a 0a 2f 2a 0a   <string.h>../*.
0230: 2a 2a 20 50 65 72 66 6f 72 6d 20 72 6f 74 31 33  ** Perform rot13
0240: 20 65 6e 63 6f 64 69 6e 67 20 6f 6e 20 61 20 73   encoding on a s
0250: 69 6e 67 6c 65 20 41 53 43 49 49 20 63 68 61 72  ingle ASCII char
0260: 61 63 74 65 72 2e 0a 2a 2f 0a 73 74 61 74 69 63  acter..*/.static
0270: 20 75 6e 73 69 67 6e 65 64 20 63 68 61 72 20 72   unsigned char r
0280: 6f 74 31 33 28 75 6e 73 69 67 6e 65 64 20 63 68  ot13(unsigned ch
0290: 61 72 20 63 29 7b 0a 20 20 69 66 28 20 63 3e 3d  ar c){.  if( c>=
02a0: 27 61 27 20 26 26 20 63 3c 3d 27 7a 27 20 29 7b  'a' && c<='z' ){
02b0: 0a 20 20 20 20 63 20 2b 3d 20 31 33 3b 0a 20 20  .    c += 13;.  
02c0: 20 20 69 66 28 20 63 3e 27 7a 27 20 29 20 63 20    if( c>'z' ) c 
02d0: 2d 3d 20 32 36 3b 0a 20 20 7d 65 6c 73 65 20 69  -= 26;.  }else i
02e0: 66 28 20 63 3e 3d 27 41 27 20 26 26 20 63 3c 3d  f( c>='A' && c<=
02f0: 27 5a 27 20 29 7b 0a 20 20 20 20 63 20 2b 3d 20  'Z' ){.    c += 
0300: 31 33 3b 0a 20 20 20 20 69 66 28 20 63 3e 27 5a  13;.    if( c>'Z
0310: 27 20 29 20 63 20 2d 3d 20 32 36 3b 0a 20 20 7d  ' ) c -= 26;.  }
0320: 0a 20 20 72 65 74 75 72 6e 20 63 3b 0a 7d 0a 0a  .  return c;.}..
0330: 2f 2a 0a 2a 2a 20 49 6d 70 6c 65 6d 65 6e 74 61  /*.** Implementa
0340: 74 69 6f 6e 20 6f 66 20 74 68 65 20 72 6f 74 31  tion of the rot1
0350: 33 28 29 20 66 75 6e 63 74 69 6f 6e 2e 0a 2a 2a  3() function..**
0360: 0a 2a 2a 20 52 6f 74 61 74 65 20 41 53 43 49 49  .** Rotate ASCII
0370: 20 61 6c 70 68 61 62 65 74 69 63 20 63 68 61 72   alphabetic char
0380: 61 63 74 65 72 73 20 62 79 20 31 33 20 63 68 61  acters by 13 cha
0390: 72 61 63 74 65 72 20 70 6f 73 69 74 69 6f 6e 73  racter positions
03a0: 2e 20 20 0a 2a 2a 20 4e 6f 6e 2d 41 53 43 49 49  .  .** Non-ASCII
03b0: 20 63 68 61 72 61 63 74 65 72 73 20 61 72 65 20   characters are 
03c0: 75 6e 63 68 61 6e 67 65 64 2e 20 20 72 6f 74 31  unchanged.  rot1
03d0: 33 28 72 6f 74 31 33 28 58 29 29 20 73 68 6f 75  3(rot13(X)) shou
03e0: 6c 64 20 61 6c 77 61 79 73 0a 2a 2a 20 65 71 75  ld always.** equ
03f0: 61 6c 20 58 2e 0a 2a 2f 0a 73 74 61 74 69 63 20  al X..*/.static 
0400: 76 6f 69 64 20 72 6f 74 31 33 66 75 6e 63 28 0a  void rot13func(.
0410: 20 20 73 71 6c 69 74 65 33 5f 63 6f 6e 74 65 78    sqlite3_contex
0420: 74 20 2a 63 6f 6e 74 65 78 74 2c 0a 20 20 69 6e  t *context,.  in
0430: 74 20 61 72 67 63 2c 0a 20 20 73 71 6c 69 74 65  t argc,.  sqlite
0440: 33 5f 76 61 6c 75 65 20 2a 2a 61 72 67 76 0a 29  3_value **argv.)
0450: 7b 0a 20 20 63 6f 6e 73 74 20 75 6e 73 69 67 6e  {.  const unsign
0460: 65 64 20 63 68 61 72 20 2a 7a 49 6e 3b 0a 20 20  ed char *zIn;.  
0470: 69 6e 74 20 6e 49 6e 3b 0a 20 20 75 6e 73 69 67  int nIn;.  unsig
0480: 6e 65 64 20 63 68 61 72 20 2a 7a 4f 75 74 3b 0a  ned char *zOut;.
0490: 20 20 63 68 61 72 20 2a 7a 54 6f 46 72 65 65 20    char *zToFree 
04a0: 3d 20 30 3b 0a 20 20 69 6e 74 20 69 3b 0a 20 20  = 0;.  int i;.  
04b0: 63 68 61 72 20 7a 54 65 6d 70 5b 31 30 30 5d 3b  char zTemp[100];
04c0: 0a 20 20 61 73 73 65 72 74 28 20 61 72 67 63 3d  .  assert( argc=
04d0: 3d 31 20 29 3b 0a 20 20 69 66 28 20 73 71 6c 69  =1 );.  if( sqli
04e0: 74 65 33 5f 76 61 6c 75 65 5f 74 79 70 65 28 61  te3_value_type(a
04f0: 72 67 76 5b 30 5d 29 3d 3d 53 51 4c 49 54 45 5f  rgv[0])==SQLITE_
0500: 4e 55 4c 4c 20 29 20 72 65 74 75 72 6e 3b 0a 20  NULL ) return;. 
0510: 20 7a 49 6e 20 3d 20 28 63 6f 6e 73 74 20 75 6e   zIn = (const un
0520: 73 69 67 6e 65 64 20 63 68 61 72 2a 29 73 71 6c  signed char*)sql
0530: 69 74 65 33 5f 76 61 6c 75 65 5f 74 65 78 74 28  ite3_value_text(
0540: 61 72 67 76 5b 30 5d 29 3b 0a 20 20 6e 49 6e 20  argv[0]);.  nIn 
0550: 3d 20 73 71 6c 69 74 65 33 5f 76 61 6c 75 65 5f  = sqlite3_value_
0560: 62 79 74 65 73 28 61 72 67 76 5b 30 5d 29 3b 0a  bytes(argv[0]);.
0570: 20 20 69 66 28 20 6e 49 6e 3c 73 69 7a 65 6f 66    if( nIn<sizeof
0580: 28 7a 54 65 6d 70 29 2d 31 20 29 7b 0a 20 20 20  (zTemp)-1 ){.   
0590: 20 7a 4f 75 74 20 3d 20 7a 54 65 6d 70 3b 0a 20   zOut = zTemp;. 
05a0: 20 7d 65 6c 73 65 7b 0a 20 20 20 20 7a 4f 75 74   }else{.    zOut
05b0: 20 3d 20 7a 54 6f 46 72 65 65 20 3d 20 73 71 6c   = zToFree = sql
05c0: 69 74 65 33 5f 6d 61 6c 6c 6f 63 28 20 6e 49 6e  ite3_malloc( nIn
05d0: 2b 31 20 29 3b 0a 20 20 20 20 69 66 28 20 7a 4f  +1 );.    if( zO
05e0: 75 74 3d 3d 30 20 29 7b 0a 20 20 20 20 20 20 73  ut==0 ){.      s
05f0: 71 6c 69 74 65 33 5f 72 65 73 75 6c 74 5f 65 72  qlite3_result_er
0600: 72 6f 72 5f 6e 6f 6d 65 6d 28 63 6f 6e 74 65 78  ror_nomem(contex
0610: 74 29 3b 0a 20 20 20 20 20 20 72 65 74 75 72 6e  t);.      return
0620: 3b 0a 20 20 20 20 7d 0a 20 20 7d 0a 20 20 66 6f  ;.    }.  }.  fo
0630: 72 28 69 3d 30 3b 20 69 3c 6e 49 6e 3b 20 69 2b  r(i=0; i<nIn; i+
0640: 2b 29 20 7a 4f 75 74 5b 69 5d 20 3d 20 72 6f 74  +) zOut[i] = rot
0650: 31 33 28 7a 49 6e 5b 69 5d 29 3b 0a 20 20 7a 4f  13(zIn[i]);.  zO
0660: 75 74 5b 69 5d 20 3d 20 30 3b 0a 20 20 73 71 6c  ut[i] = 0;.  sql
0670: 69 74 65 33 5f 72 65 73 75 6c 74 5f 74 65 78 74  ite3_result_text
0680: 28 63 6f 6e 74 65 78 74 2c 20 28 63 68 61 72 2a  (context, (char*
0690: 29 7a 4f 75 74 2c 20 69 2c 20 53 51 4c 49 54 45  )zOut, i, SQLITE
06a0: 5f 54 52 41 4e 53 49 45 4e 54 29 3b 0a 20 20 73  _TRANSIENT);.  s
06b0: 71 6c 69 74 65 33 5f 66 72 65 65 28 7a 54 6f 46  qlite3_free(zToF
06c0: 72 65 65 29 3b 0a 7d 0a 0a 2f 2a 0a 2a 2a 20 49  ree);.}../*.** I
06d0: 6d 70 6c 65 6d 65 6e 74 20 74 68 65 20 72 6f 74  mplement the rot
06e0: 31 33 20 63 6f 6c 6c 61 74 69 6e 67 20 73 65 71  13 collating seq
06f0: 75 65 6e 63 65 20 73 6f 20 74 68 61 74 20 69 66  uence so that if
0700: 0a 2a 2a 0a 2a 2a 20 20 20 20 20 20 78 3d 79 20  .**.**      x=y 
0710: 43 4f 4c 4c 41 54 45 20 72 6f 74 31 33 0a 2a 2a  COLLATE rot13.**
0720: 0a 2a 2a 20 54 68 65 6e 20 0a 2a 2a 0a 2a 2a 20  .** Then .**.** 
0730: 20 20 20 20 20 72 6f 74 31 33 28 78 29 3d 72 6f       rot13(x)=ro
0740: 74 31 33 28 79 29 20 43 4f 4c 4c 41 54 45 20 62  t13(y) COLLATE b
0750: 69 6e 61 72 79 0a 2a 2f 0a 73 74 61 74 69 63 20  inary.*/.static 
0760: 69 6e 74 20 72 6f 74 31 33 43 6f 6c 6c 46 75 6e  int rot13CollFun
0770: 63 28 0a 20 20 76 6f 69 64 20 2a 6e 6f 74 55 73  c(.  void *notUs
0780: 65 64 2c 0a 20 20 69 6e 74 20 6e 4b 65 79 31 2c  ed,.  int nKey1,
0790: 20 63 6f 6e 73 74 20 76 6f 69 64 20 2a 70 4b 65   const void *pKe
07a0: 79 31 2c 0a 20 20 69 6e 74 20 6e 4b 65 79 32 2c  y1,.  int nKey2,
07b0: 20 63 6f 6e 73 74 20 76 6f 69 64 20 2a 70 4b 65   const void *pKe
07c0: 79 32 0a 29 7b 0a 20 20 63 6f 6e 73 74 20 63 68  y2.){.  const ch
07d0: 61 72 20 2a 7a 41 20 3d 20 28 63 6f 6e 73 74 20  ar *zA = (const 
07e0: 63 68 61 72 2a 29 70 4b 65 79 31 3b 0a 20 20 63  char*)pKey1;.  c
07f0: 6f 6e 73 74 20 63 68 61 72 20 2a 7a 42 20 3d 20  onst char *zB = 
0800: 28 63 6f 6e 73 74 20 63 68 61 72 2a 29 70 4b 65  (const char*)pKe
0810: 79 32 3b 0a 20 20 69 6e 74 20 69 2c 20 78 3b 0a  y2;.  int i, x;.
0820: 20 20 66 6f 72 28 69 3d 30 3b 20 69 3c 6e 4b 65    for(i=0; i<nKe
0830: 79 31 20 26 26 20 69 3c 6e 4b 65 79 32 3b 20 69  y1 && i<nKey2; i
0840: 2b 2b 29 7b 0a 20 20 20 20 78 20 3d 20 28 69 6e  ++){.    x = (in
0850: 74 29 72 6f 74 31 33 28 7a 41 5b 69 5d 29 20 2d  t)rot13(zA[i]) -
0860: 20 28 69 6e 74 29 72 6f 74 31 33 28 7a 42 5b 69   (int)rot13(zB[i
0870: 5d 29 3b 0a 20 20 20 20 69 66 28 20 78 21 3d 30  ]);.    if( x!=0
0880: 20 29 20 72 65 74 75 72 6e 20 78 3b 0a 20 20 7d   ) return x;.  }
0890: 0a 20 20 72 65 74 75 72 6e 20 6e 4b 65 79 31 20  .  return nKey1 
08a0: 2d 20 6e 4b 65 79 32 3b 0a 7d 0a 0a 0a 23 69 66  - nKey2;.}...#if
08b0: 64 65 66 20 5f 57 49 4e 33 32 0a 5f 5f 64 65 63  def _WIN32.__dec
08c0: 6c 73 70 65 63 28 64 6c 6c 65 78 70 6f 72 74 29  lspec(dllexport)
08d0: 0a 23 65 6e 64 69 66 0a 69 6e 74 20 73 71 6c 69  .#endif.int sqli
08e0: 74 65 33 5f 72 6f 74 5f 69 6e 69 74 28 0a 20 20  te3_rot_init(.  
08f0: 73 71 6c 69 74 65 33 20 2a 64 62 2c 20 0a 20 20  sqlite3 *db, .  
0900: 63 68 61 72 20 2a 2a 70 7a 45 72 72 4d 73 67 2c  char **pzErrMsg,
0910: 20 0a 20 20 63 6f 6e 73 74 20 73 71 6c 69 74 65   .  const sqlite
0920: 33 5f 61 70 69 5f 72 6f 75 74 69 6e 65 73 20 2a  3_api_routines *
0930: 70 41 70 69 0a 29 7b 0a 20 20 69 6e 74 20 72 63  pApi.){.  int rc
0940: 20 3d 20 53 51 4c 49 54 45 5f 4f 4b 3b 0a 20 20   = SQLITE_OK;.  
0950: 53 51 4c 49 54 45 5f 45 58 54 45 4e 53 49 4f 4e  SQLITE_EXTENSION
0960: 5f 49 4e 49 54 32 28 70 41 70 69 29 3b 0a 20 20  _INIT2(pApi);.  
0970: 28 76 6f 69 64 29 70 7a 45 72 72 4d 73 67 3b 20  (void)pzErrMsg; 
0980: 20 2f 2a 20 55 6e 75 73 65 64 20 70 61 72 61 6d   /* Unused param
0990: 65 74 65 72 20 2a 2f 0a 20 20 72 63 20 3d 20 73  eter */.  rc = s
09a0: 71 6c 69 74 65 33 5f 63 72 65 61 74 65 5f 66 75  qlite3_create_fu
09b0: 6e 63 74 69 6f 6e 28 64 62 2c 20 22 72 6f 74 31  nction(db, "rot1
09c0: 33 22 2c 20 31 2c 20 53 51 4c 49 54 45 5f 55 54  3", 1, SQLITE_UT
09d0: 46 38 2c 20 30 2c 0a 20 20 20 20 20 20 20 20 20  F8, 0,.         
09e0: 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20                  
09f0: 20 20 20 20 20 20 72 6f 74 31 33 66 75 6e 63 2c        rot13func,
0a00: 20 30 2c 20 30 29 3b 0a 20 20 69 66 28 20 72 63   0, 0);.  if( rc
0a10: 3d 3d 53 51 4c 49 54 45 5f 4f 4b 20 29 7b 0a 20  ==SQLITE_OK ){. 
0a20: 20 20 20 72 63 20 3d 20 73 71 6c 69 74 65 33 5f     rc = sqlite3_
0a30: 63 72 65 61 74 65 5f 63 6f 6c 6c 61 74 69 6f 6e  create_collation
0a40: 28 64 62 2c 20 22 72 6f 74 31 33 22 2c 20 53 51  (db, "rot13", SQ
0a50: 4c 49 54 45 5f 55 54 46 38 2c 20 30 2c 20 72 6f  LITE_UTF8, 0, ro
0a60: 74 31 33 43 6f 6c 6c 46 75 6e 63 29 3b 0a 20 20  t13CollFunc);.  
0a70: 7d 0a 20 20 72 65 74 75 72 6e 20 72 63 3b 0a 7d  }.  return rc;.}
0a80: 0a                                               .