/ Hex Artifact Content
Login

Artifact 3ae5101de6cbfda2720152ab659ea84079719241:


0000: 2f 2a 0a 2a 2a 20 32 30 30 38 20 53 65 70 74 65  /*.** 2008 Septe
0010: 6d 62 65 72 20 31 0a 2a 2a 0a 2a 2a 20 54 68 65  mber 1.**.** The
0020: 20 61 75 74 68 6f 72 20 64 69 73 63 6c 61 69 6d   author disclaim
0030: 73 20 63 6f 70 79 72 69 67 68 74 20 74 6f 20 74  s copyright to t
0040: 68 69 73 20 73 6f 75 72 63 65 20 63 6f 64 65 2e  his source code.
0050: 20 20 49 6e 20 70 6c 61 63 65 20 6f 66 0a 2a 2a    In place of.**
0060: 20 61 20 6c 65 67 61 6c 20 6e 6f 74 69 63 65 2c   a legal notice,
0070: 20 68 65 72 65 20 69 73 20 61 20 62 6c 65 73 73   here is a bless
0080: 69 6e 67 3a 0a 2a 2a 0a 2a 2a 20 20 20 20 4d 61  ing:.**.**    Ma
0090: 79 20 79 6f 75 20 64 6f 20 67 6f 6f 64 20 61 6e  y you do good an
00a0: 64 20 6e 6f 74 20 65 76 69 6c 2e 0a 2a 2a 20 20  d not evil..**  
00b0: 20 20 4d 61 79 20 79 6f 75 20 66 69 6e 64 20 66    May you find f
00c0: 6f 72 67 69 76 65 6e 65 73 73 20 66 6f 72 20 79  orgiveness for y
00d0: 6f 75 72 73 65 6c 66 20 61 6e 64 20 66 6f 72 67  ourself and forg
00e0: 69 76 65 20 6f 74 68 65 72 73 2e 0a 2a 2a 20 20  ive others..**  
00f0: 20 20 4d 61 79 20 79 6f 75 20 73 68 61 72 65 20    May you share 
0100: 66 72 65 65 6c 79 2c 20 6e 65 76 65 72 20 74 61  freely, never ta
0110: 6b 69 6e 67 20 6d 6f 72 65 20 74 68 61 6e 20 79  king more than y
0120: 6f 75 20 67 69 76 65 2e 0a 2a 2a 0a 2a 2a 2a 2a  ou give..**.****
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 2a 0a 2a 2a 0a 2a 2a 20 54 68 65 20  *****.**.** The 
0180: 63 6f 64 65 20 69 6e 20 74 68 69 73 20 66 69 6c  code in this fil
0190: 65 20 63 6f 6e 74 61 69 6e 73 20 73 61 6d 70 6c  e contains sampl
01a0: 65 20 69 6d 70 6c 65 6d 65 6e 74 61 74 69 6f 6e  e implementation
01b0: 73 20 6f 66 20 74 68 65 20 0a 2a 2a 20 73 71 6c  s of the .** sql
01c0: 69 74 65 33 5f 77 73 64 5f 69 6e 69 74 28 29 20  ite3_wsd_init() 
01d0: 61 6e 64 20 73 71 6c 69 74 65 33 5f 77 73 64 5f  and sqlite3_wsd_
01e0: 66 69 6e 64 28 29 20 66 75 6e 63 74 69 6f 6e 73  find() functions
01f0: 20 72 65 71 75 69 72 65 64 20 69 66 20 74 68 65   required if the
0200: 0a 2a 2a 20 53 51 4c 49 54 45 5f 4f 4d 49 54 5f  .** SQLITE_OMIT_
0210: 57 53 44 20 73 79 6d 62 6f 6c 20 69 73 20 64 65  WSD symbol is de
0220: 66 69 6e 65 64 20 61 74 20 62 75 69 6c 64 20 74  fined at build t
0230: 69 6d 65 2e 0a 2a 2a 0a 2a 2a 20 24 49 64 3a 20  ime..**.** $Id: 
0240: 74 65 73 74 5f 77 73 64 2e 63 2c 76 20 31 2e 34  test_wsd.c,v 1.4
0250: 20 32 30 30 39 2f 30 33 2f 32 33 20 30 34 3a 33   2009/03/23 04:3
0260: 33 3a 33 33 20 64 61 6e 69 65 6c 6b 31 39 37 37  3:33 danielk1977
0270: 20 45 78 70 20 24 0a 2a 2f 0a 0a 23 69 66 20 64   Exp $.*/..#if d
0280: 65 66 69 6e 65 64 28 53 51 4c 49 54 45 5f 4f 4d  efined(SQLITE_OM
0290: 49 54 5f 57 53 44 29 20 26 26 20 64 65 66 69 6e  IT_WSD) && defin
02a0: 65 64 28 53 51 4c 49 54 45 5f 54 45 53 54 29 0a  ed(SQLITE_TEST).
02b0: 0a 23 69 6e 63 6c 75 64 65 20 22 73 71 6c 69 74  .#include "sqlit
02c0: 65 49 6e 74 2e 68 22 0a 0a 23 64 65 66 69 6e 65  eInt.h"..#define
02d0: 20 50 4c 53 5f 48 41 53 48 53 49 5a 45 20 34 33   PLS_HASHSIZE 43
02e0: 0a 0a 74 79 70 65 64 65 66 20 73 74 72 75 63 74  ..typedef struct
02f0: 20 50 72 6f 63 65 73 73 4c 6f 63 61 6c 53 74 6f   ProcessLocalSto
0300: 72 61 67 65 20 50 72 6f 63 65 73 73 4c 6f 63 61  rage ProcessLoca
0310: 6c 53 74 6f 72 61 67 65 3b 0a 74 79 70 65 64 65  lStorage;.typede
0320: 66 20 73 74 72 75 63 74 20 50 72 6f 63 65 73 73  f struct Process
0330: 4c 6f 63 61 6c 56 61 72 20 50 72 6f 63 65 73 73  LocalVar Process
0340: 4c 6f 63 61 6c 56 61 72 3b 0a 0a 73 74 72 75 63  LocalVar;..struc
0350: 74 20 50 72 6f 63 65 73 73 4c 6f 63 61 6c 53 74  t ProcessLocalSt
0360: 6f 72 61 67 65 20 7b 0a 20 20 50 72 6f 63 65 73  orage {.  Proces
0370: 73 4c 6f 63 61 6c 56 61 72 20 2a 61 44 61 74 61  sLocalVar *aData
0380: 5b 50 4c 53 5f 48 41 53 48 53 49 5a 45 5d 3b 0a  [PLS_HASHSIZE];.
0390: 20 20 69 6e 74 20 6e 46 72 65 65 3b 0a 20 20 75    int nFree;.  u
03a0: 38 20 2a 70 46 72 65 65 3b 0a 7d 3b 0a 0a 73 74  8 *pFree;.};..st
03b0: 72 75 63 74 20 50 72 6f 63 65 73 73 4c 6f 63 61  ruct ProcessLoca
03c0: 6c 56 61 72 20 7b 0a 20 20 76 6f 69 64 20 2a 70  lVar {.  void *p
03d0: 4b 65 79 3b 0a 20 20 50 72 6f 63 65 73 73 4c 6f  Key;.  ProcessLo
03e0: 63 61 6c 56 61 72 20 2a 70 4e 65 78 74 3b 0a 7d  calVar *pNext;.}
03f0: 3b 0a 0a 73 74 61 74 69 63 20 50 72 6f 63 65 73  ;..static Proces
0400: 73 4c 6f 63 61 6c 53 74 6f 72 61 67 65 20 2a 70  sLocalStorage *p
0410: 47 6c 6f 62 61 6c 20 3d 20 30 3b 0a 0a 69 6e 74  Global = 0;..int
0420: 20 73 71 6c 69 74 65 33 5f 77 73 64 5f 69 6e 69   sqlite3_wsd_ini
0430: 74 28 69 6e 74 20 4e 2c 20 69 6e 74 20 4a 29 7b  t(int N, int J){
0440: 0a 20 20 69 66 28 20 21 70 47 6c 6f 62 61 6c 20  .  if( !pGlobal 
0450: 29 7b 0a 20 20 20 20 69 6e 74 20 6e 4d 61 6c 6c  ){.    int nMall
0460: 6f 63 20 3d 20 4e 20 2b 20 73 69 7a 65 6f 66 28  oc = N + sizeof(
0470: 50 72 6f 63 65 73 73 4c 6f 63 61 6c 53 74 6f 72  ProcessLocalStor
0480: 61 67 65 29 20 2b 20 4a 2a 73 69 7a 65 6f 66 28  age) + J*sizeof(
0490: 50 72 6f 63 65 73 73 4c 6f 63 61 6c 56 61 72 29  ProcessLocalVar)
04a0: 3b 0a 20 20 20 20 70 47 6c 6f 62 61 6c 20 3d 20  ;.    pGlobal = 
04b0: 28 50 72 6f 63 65 73 73 4c 6f 63 61 6c 53 74 6f  (ProcessLocalSto
04c0: 72 61 67 65 20 2a 29 6d 61 6c 6c 6f 63 28 6e 4d  rage *)malloc(nM
04d0: 61 6c 6c 6f 63 29 3b 0a 20 20 20 20 69 66 28 20  alloc);.    if( 
04e0: 70 47 6c 6f 62 61 6c 20 29 7b 0a 20 20 20 20 20  pGlobal ){.     
04f0: 20 6d 65 6d 73 65 74 28 70 47 6c 6f 62 61 6c 2c   memset(pGlobal,
0500: 20 30 2c 20 73 69 7a 65 6f 66 28 50 72 6f 63 65   0, sizeof(Proce
0510: 73 73 4c 6f 63 61 6c 53 74 6f 72 61 67 65 29 29  ssLocalStorage))
0520: 3b 0a 20 20 20 20 20 20 70 47 6c 6f 62 61 6c 2d  ;.      pGlobal-
0530: 3e 6e 46 72 65 65 20 3d 20 6e 4d 61 6c 6c 6f 63  >nFree = nMalloc
0540: 20 2d 20 73 69 7a 65 6f 66 28 50 72 6f 63 65 73   - sizeof(Proces
0550: 73 4c 6f 63 61 6c 53 74 6f 72 61 67 65 29 3b 0a  sLocalStorage);.
0560: 20 20 20 20 20 20 70 47 6c 6f 62 61 6c 2d 3e 70        pGlobal->p
0570: 46 72 65 65 20 3d 20 28 75 38 20 2a 29 26 70 47  Free = (u8 *)&pG
0580: 6c 6f 62 61 6c 5b 31 5d 3b 0a 20 20 20 20 7d 0a  lobal[1];.    }.
0590: 20 20 7d 0a 0a 20 20 72 65 74 75 72 6e 20 70 47    }..  return pG
05a0: 6c 6f 62 61 6c 20 3f 20 53 51 4c 49 54 45 5f 4f  lobal ? SQLITE_O
05b0: 4b 20 3a 20 53 51 4c 49 54 45 5f 4e 4f 4d 45 4d  K : SQLITE_NOMEM
05c0: 3b 0a 7d 0a 0a 76 6f 69 64 20 2a 73 71 6c 69 74  ;.}..void *sqlit
05d0: 65 33 5f 77 73 64 5f 66 69 6e 64 28 76 6f 69 64  e3_wsd_find(void
05e0: 20 2a 4b 2c 20 69 6e 74 20 4c 29 7b 0a 20 20 69   *K, int L){.  i
05f0: 6e 74 20 69 3b 0a 20 20 69 6e 74 20 69 48 61 73  nt i;.  int iHas
0600: 68 20 3d 20 30 3b 0a 20 20 50 72 6f 63 65 73 73  h = 0;.  Process
0610: 4c 6f 63 61 6c 56 61 72 20 2a 70 56 61 72 3b 0a  LocalVar *pVar;.
0620: 0a 20 20 2f 2a 20 43 61 6c 63 75 6c 61 74 65 20  .  /* Calculate 
0630: 61 20 68 61 73 68 20 6f 66 20 4b 20 2a 2f 0a 20  a hash of K */. 
0640: 20 66 6f 72 28 69 3d 30 3b 20 69 3c 73 69 7a 65   for(i=0; i<size
0650: 6f 66 28 76 6f 69 64 2a 29 3b 20 69 2b 2b 29 7b  of(void*); i++){
0660: 0a 20 20 20 20 69 48 61 73 68 20 3d 20 28 69 48  .    iHash = (iH
0670: 61 73 68 3c 3c 33 29 20 2b 20 28 28 75 6e 73 69  ash<<3) + ((unsi
0680: 67 6e 65 64 20 63 68 61 72 20 2a 29 26 4b 29 5b  gned char *)&K)[
0690: 69 5d 3b 0a 20 20 7d 0a 20 20 69 48 61 73 68 20  i];.  }.  iHash 
06a0: 3d 20 69 48 61 73 68 25 50 4c 53 5f 48 41 53 48  = iHash%PLS_HASH
06b0: 53 49 5a 45 3b 0a 0a 20 20 2f 2a 20 53 65 61 72  SIZE;..  /* Sear
06c0: 63 68 20 74 68 65 20 68 61 73 68 20 74 61 62 6c  ch the hash tabl
06d0: 65 20 66 6f 72 20 4b 2e 20 2a 2f 0a 20 20 66 6f  e for K. */.  fo
06e0: 72 28 70 56 61 72 3d 70 47 6c 6f 62 61 6c 2d 3e  r(pVar=pGlobal->
06f0: 61 44 61 74 61 5b 69 48 61 73 68 5d 3b 20 70 56  aData[iHash]; pV
0700: 61 72 20 26 26 20 70 56 61 72 2d 3e 70 4b 65 79  ar && pVar->pKey
0710: 21 3d 4b 3b 20 70 56 61 72 3d 70 56 61 72 2d 3e  !=K; pVar=pVar->
0720: 70 4e 65 78 74 29 3b 0a 0a 20 20 2f 2a 20 49 66  pNext);..  /* If
0730: 20 6e 6f 20 65 6e 74 72 79 20 66 6f 72 20 4b 20   no entry for K 
0740: 77 61 73 20 66 6f 75 6e 64 2c 20 63 72 65 61 74  was found, creat
0750: 65 20 61 6e 64 20 70 6f 70 75 6c 61 74 65 20 61  e and populate a
0760: 20 6e 65 77 20 6f 6e 65 2e 20 2a 2f 0a 20 20 69   new one. */.  i
0770: 66 28 20 21 70 56 61 72 20 29 7b 0a 20 20 20 20  f( !pVar ){.    
0780: 69 6e 74 20 6e 42 79 74 65 20 3d 20 52 4f 55 4e  int nByte = ROUN
0790: 44 38 28 73 69 7a 65 6f 66 28 50 72 6f 63 65 73  D8(sizeof(Proces
07a0: 73 4c 6f 63 61 6c 56 61 72 29 20 2b 20 4c 29 3b  sLocalVar) + L);
07b0: 0a 20 20 20 20 61 73 73 65 72 74 28 20 70 47 6c  .    assert( pGl
07c0: 6f 62 61 6c 2d 3e 6e 46 72 65 65 3e 3d 6e 42 79  obal->nFree>=nBy
07d0: 74 65 20 29 3b 0a 20 20 20 20 70 56 61 72 20 3d  te );.    pVar =
07e0: 20 28 50 72 6f 63 65 73 73 4c 6f 63 61 6c 56 61   (ProcessLocalVa
07f0: 72 20 2a 29 70 47 6c 6f 62 61 6c 2d 3e 70 46 72  r *)pGlobal->pFr
0800: 65 65 3b 0a 20 20 20 20 70 56 61 72 2d 3e 70 4b  ee;.    pVar->pK
0810: 65 79 20 3d 20 4b 3b 0a 20 20 20 20 70 56 61 72  ey = K;.    pVar
0820: 2d 3e 70 4e 65 78 74 20 3d 20 70 47 6c 6f 62 61  ->pNext = pGloba
0830: 6c 2d 3e 61 44 61 74 61 5b 69 48 61 73 68 5d 3b  l->aData[iHash];
0840: 0a 20 20 20 20 70 47 6c 6f 62 61 6c 2d 3e 61 44  .    pGlobal->aD
0850: 61 74 61 5b 69 48 61 73 68 5d 20 3d 20 70 56 61  ata[iHash] = pVa
0860: 72 3b 0a 20 20 20 20 70 47 6c 6f 62 61 6c 2d 3e  r;.    pGlobal->
0870: 6e 46 72 65 65 20 2d 3d 20 6e 42 79 74 65 3b 0a  nFree -= nByte;.
0880: 20 20 20 20 70 47 6c 6f 62 61 6c 2d 3e 70 46 72      pGlobal->pFr
0890: 65 65 20 2b 3d 20 6e 42 79 74 65 3b 0a 20 20 20  ee += nByte;.   
08a0: 20 6d 65 6d 63 70 79 28 26 70 56 61 72 5b 31 5d   memcpy(&pVar[1]
08b0: 2c 20 4b 2c 20 4c 29 3b 0a 20 20 7d 0a 0a 20 20  , K, L);.  }..  
08c0: 72 65 74 75 72 6e 20 28 76 6f 69 64 20 2a 29 26  return (void *)&
08d0: 70 56 61 72 5b 31 5d 3b 0a 7d 0a 0a 23 65 6e 64  pVar[1];.}..#end
08e0: 69 66 0a                                         if.