SQLite
Hex Artifact Content
Not logged in

Artifact 540a169cb0d74f15522a8930b0cccdcb37a4fd071d219a5a083a319fc6e8db77:


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 75 6e 73 69 67 6e 65 64 20 63 68 61 72 20    unsigned char 
04a0: 2a 7a 54 6f 46 72 65 65 20 3d 20 30 3b 0a 20 20  *zToFree = 0;.  
04b0: 69 6e 74 20 69 3b 0a 20 20 75 6e 73 69 67 6e 65  int i;.  unsigne
04c0: 64 20 63 68 61 72 20 7a 54 65 6d 70 5b 31 30 30  d char zTemp[100
04d0: 5d 3b 0a 20 20 61 73 73 65 72 74 28 20 61 72 67  ];.  assert( arg
04e0: 63 3d 3d 31 20 29 3b 0a 20 20 69 66 28 20 73 71  c==1 );.  if( sq
04f0: 6c 69 74 65 33 5f 76 61 6c 75 65 5f 74 79 70 65  lite3_value_type
0500: 28 61 72 67 76 5b 30 5d 29 3d 3d 53 51 4c 49 54  (argv[0])==SQLIT
0510: 45 5f 4e 55 4c 4c 20 29 20 72 65 74 75 72 6e 3b  E_NULL ) return;
0520: 0a 20 20 7a 49 6e 20 3d 20 28 63 6f 6e 73 74 20  .  zIn = (const 
0530: 75 6e 73 69 67 6e 65 64 20 63 68 61 72 2a 29 73  unsigned char*)s
0540: 71 6c 69 74 65 33 5f 76 61 6c 75 65 5f 74 65 78  qlite3_value_tex
0550: 74 28 61 72 67 76 5b 30 5d 29 3b 0a 20 20 6e 49  t(argv[0]);.  nI
0560: 6e 20 3d 20 73 71 6c 69 74 65 33 5f 76 61 6c 75  n = sqlite3_valu
0570: 65 5f 62 79 74 65 73 28 61 72 67 76 5b 30 5d 29  e_bytes(argv[0])
0580: 3b 0a 20 20 69 66 28 20 6e 49 6e 3c 73 69 7a 65  ;.  if( nIn<size
0590: 6f 66 28 7a 54 65 6d 70 29 2d 31 20 29 7b 0a 20  of(zTemp)-1 ){. 
05a0: 20 20 20 7a 4f 75 74 20 3d 20 7a 54 65 6d 70 3b     zOut = zTemp;
05b0: 0a 20 20 7d 65 6c 73 65 7b 0a 20 20 20 20 7a 4f  .  }else{.    zO
05c0: 75 74 20 3d 20 7a 54 6f 46 72 65 65 20 3d 20 28  ut = zToFree = (
05d0: 75 6e 73 69 67 6e 65 64 20 63 68 61 72 2a 29 73  unsigned char*)s
05e0: 71 6c 69 74 65 33 5f 6d 61 6c 6c 6f 63 36 34 28  qlite3_malloc64(
05f0: 20 6e 49 6e 2b 31 20 29 3b 0a 20 20 20 20 69 66   nIn+1 );.    if
0600: 28 20 7a 4f 75 74 3d 3d 30 20 29 7b 0a 20 20 20  ( zOut==0 ){.   
0610: 20 20 20 73 71 6c 69 74 65 33 5f 72 65 73 75 6c     sqlite3_resul
0620: 74 5f 65 72 72 6f 72 5f 6e 6f 6d 65 6d 28 63 6f  t_error_nomem(co
0630: 6e 74 65 78 74 29 3b 0a 20 20 20 20 20 20 72 65  ntext);.      re
0640: 74 75 72 6e 3b 0a 20 20 20 20 7d 0a 20 20 7d 0a  turn;.    }.  }.
0650: 20 20 66 6f 72 28 69 3d 30 3b 20 69 3c 6e 49 6e    for(i=0; i<nIn
0660: 3b 20 69 2b 2b 29 20 7a 4f 75 74 5b 69 5d 20 3d  ; i++) zOut[i] =
0670: 20 72 6f 74 31 33 28 7a 49 6e 5b 69 5d 29 3b 0a   rot13(zIn[i]);.
0680: 20 20 7a 4f 75 74 5b 69 5d 20 3d 20 30 3b 0a 20    zOut[i] = 0;. 
0690: 20 73 71 6c 69 74 65 33 5f 72 65 73 75 6c 74 5f   sqlite3_result_
06a0: 74 65 78 74 28 63 6f 6e 74 65 78 74 2c 20 28 63  text(context, (c
06b0: 68 61 72 2a 29 7a 4f 75 74 2c 20 69 2c 20 53 51  har*)zOut, i, SQ
06c0: 4c 49 54 45 5f 54 52 41 4e 53 49 45 4e 54 29 3b  LITE_TRANSIENT);
06d0: 0a 20 20 73 71 6c 69 74 65 33 5f 66 72 65 65 28  .  sqlite3_free(
06e0: 7a 54 6f 46 72 65 65 29 3b 0a 7d 0a 0a 2f 2a 0a  zToFree);.}../*.
06f0: 2a 2a 20 49 6d 70 6c 65 6d 65 6e 74 20 74 68 65  ** Implement the
0700: 20 72 6f 74 31 33 20 63 6f 6c 6c 61 74 69 6e 67   rot13 collating
0710: 20 73 65 71 75 65 6e 63 65 20 73 6f 20 74 68 61   sequence so tha
0720: 74 20 69 66 0a 2a 2a 0a 2a 2a 20 20 20 20 20 20  t if.**.**      
0730: 78 3d 79 20 43 4f 4c 4c 41 54 45 20 72 6f 74 31  x=y COLLATE rot1
0740: 33 0a 2a 2a 0a 2a 2a 20 54 68 65 6e 20 0a 2a 2a  3.**.** Then .**
0750: 0a 2a 2a 20 20 20 20 20 20 72 6f 74 31 33 28 78  .**      rot13(x
0760: 29 3d 72 6f 74 31 33 28 79 29 20 43 4f 4c 4c 41  )=rot13(y) COLLA
0770: 54 45 20 62 69 6e 61 72 79 0a 2a 2f 0a 73 74 61  TE binary.*/.sta
0780: 74 69 63 20 69 6e 74 20 72 6f 74 31 33 43 6f 6c  tic int rot13Col
0790: 6c 46 75 6e 63 28 0a 20 20 76 6f 69 64 20 2a 6e  lFunc(.  void *n
07a0: 6f 74 55 73 65 64 2c 0a 20 20 69 6e 74 20 6e 4b  otUsed,.  int nK
07b0: 65 79 31 2c 20 63 6f 6e 73 74 20 76 6f 69 64 20  ey1, const void 
07c0: 2a 70 4b 65 79 31 2c 0a 20 20 69 6e 74 20 6e 4b  *pKey1,.  int nK
07d0: 65 79 32 2c 20 63 6f 6e 73 74 20 76 6f 69 64 20  ey2, const void 
07e0: 2a 70 4b 65 79 32 0a 29 7b 0a 20 20 63 6f 6e 73  *pKey2.){.  cons
07f0: 74 20 63 68 61 72 20 2a 7a 41 20 3d 20 28 63 6f  t char *zA = (co
0800: 6e 73 74 20 63 68 61 72 2a 29 70 4b 65 79 31 3b  nst char*)pKey1;
0810: 0a 20 20 63 6f 6e 73 74 20 63 68 61 72 20 2a 7a  .  const char *z
0820: 42 20 3d 20 28 63 6f 6e 73 74 20 63 68 61 72 2a  B = (const char*
0830: 29 70 4b 65 79 32 3b 0a 20 20 69 6e 74 20 69 2c  )pKey2;.  int i,
0840: 20 78 3b 0a 20 20 66 6f 72 28 69 3d 30 3b 20 69   x;.  for(i=0; i
0850: 3c 6e 4b 65 79 31 20 26 26 20 69 3c 6e 4b 65 79  <nKey1 && i<nKey
0860: 32 3b 20 69 2b 2b 29 7b 0a 20 20 20 20 78 20 3d  2; i++){.    x =
0870: 20 28 69 6e 74 29 72 6f 74 31 33 28 7a 41 5b 69   (int)rot13(zA[i
0880: 5d 29 20 2d 20 28 69 6e 74 29 72 6f 74 31 33 28  ]) - (int)rot13(
0890: 7a 42 5b 69 5d 29 3b 0a 20 20 20 20 69 66 28 20  zB[i]);.    if( 
08a0: 78 21 3d 30 20 29 20 72 65 74 75 72 6e 20 78 3b  x!=0 ) return x;
08b0: 0a 20 20 7d 0a 20 20 72 65 74 75 72 6e 20 6e 4b  .  }.  return nK
08c0: 65 79 31 20 2d 20 6e 4b 65 79 32 3b 0a 7d 0a 0a  ey1 - nKey2;.}..
08d0: 0a 23 69 66 64 65 66 20 5f 57 49 4e 33 32 0a 5f  .#ifdef _WIN32._
08e0: 5f 64 65 63 6c 73 70 65 63 28 64 6c 6c 65 78 70  _declspec(dllexp
08f0: 6f 72 74 29 0a 23 65 6e 64 69 66 0a 69 6e 74 20  ort).#endif.int 
0900: 73 71 6c 69 74 65 33 5f 72 6f 74 5f 69 6e 69 74  sqlite3_rot_init
0910: 28 0a 20 20 73 71 6c 69 74 65 33 20 2a 64 62 2c  (.  sqlite3 *db,
0920: 20 0a 20 20 63 68 61 72 20 2a 2a 70 7a 45 72 72   .  char **pzErr
0930: 4d 73 67 2c 20 0a 20 20 63 6f 6e 73 74 20 73 71  Msg, .  const sq
0940: 6c 69 74 65 33 5f 61 70 69 5f 72 6f 75 74 69 6e  lite3_api_routin
0950: 65 73 20 2a 70 41 70 69 0a 29 7b 0a 20 20 69 6e  es *pApi.){.  in
0960: 74 20 72 63 20 3d 20 53 51 4c 49 54 45 5f 4f 4b  t rc = SQLITE_OK
0970: 3b 0a 20 20 53 51 4c 49 54 45 5f 45 58 54 45 4e  ;.  SQLITE_EXTEN
0980: 53 49 4f 4e 5f 49 4e 49 54 32 28 70 41 70 69 29  SION_INIT2(pApi)
0990: 3b 0a 20 20 28 76 6f 69 64 29 70 7a 45 72 72 4d  ;.  (void)pzErrM
09a0: 73 67 3b 20 20 2f 2a 20 55 6e 75 73 65 64 20 70  sg;  /* Unused p
09b0: 61 72 61 6d 65 74 65 72 20 2a 2f 0a 20 20 72 63  arameter */.  rc
09c0: 20 3d 20 73 71 6c 69 74 65 33 5f 63 72 65 61 74   = sqlite3_creat
09d0: 65 5f 66 75 6e 63 74 69 6f 6e 28 64 62 2c 20 22  e_function(db, "
09e0: 72 6f 74 31 33 22 2c 20 31 2c 20 53 51 4c 49 54  rot13", 1, SQLIT
09f0: 45 5f 55 54 46 38 2c 20 30 2c 0a 20 20 20 20 20  E_UTF8, 0,.     
0a00: 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20                  
0a10: 20 20 20 20 20 20 20 20 20 20 72 6f 74 31 33 66            rot13f
0a20: 75 6e 63 2c 20 30 2c 20 30 29 3b 0a 20 20 69 66  unc, 0, 0);.  if
0a30: 28 20 72 63 3d 3d 53 51 4c 49 54 45 5f 4f 4b 20  ( rc==SQLITE_OK 
0a40: 29 7b 0a 20 20 20 20 72 63 20 3d 20 73 71 6c 69  ){.    rc = sqli
0a50: 74 65 33 5f 63 72 65 61 74 65 5f 63 6f 6c 6c 61  te3_create_colla
0a60: 74 69 6f 6e 28 64 62 2c 20 22 72 6f 74 31 33 22  tion(db, "rot13"
0a70: 2c 20 53 51 4c 49 54 45 5f 55 54 46 38 2c 20 30  , SQLITE_UTF8, 0
0a80: 2c 20 72 6f 74 31 33 43 6f 6c 6c 46 75 6e 63 29  , rot13CollFunc)
0a90: 3b 0a 20 20 7d 0a 20 20 72 65 74 75 72 6e 20 72  ;.  }.  return r
0aa0: 63 3b 0a 7d 0a                                   c;.}.