third_party/sqlite/sqlite-src-3080704/src/test5.c

   1 /*
   2 ** 2001 September 15
   3 **
   4 ** The author disclaims copyright to this source code.  In place of
   5 ** a legal notice, here is a blessing:
   6 **
   7 **    May you do good and not evil.
   8 **    May you find forgiveness for yourself and forgive others.
   9 **    May you share freely, never taking more than you give.
  10 **
  11 *************************************************************************
  12 ** Code for testing the utf.c module in SQLite.  This code
  13 ** is not included in the SQLite library.  It is used for automated
  14 ** testing of the SQLite library. Specifically, the code in this file
  15 ** is used for testing the SQLite routines for converting between
  16 ** the various supported unicode encodings.
  17 */
  18 #include "sqliteInt.h"
  19 #include "vdbeInt.h"
  20 #include "tcl.h"
  21 #include <stdlib.h>
  22 #include <string.h>
  23
  24 /*
  25 ** The first argument is a TCL UTF-8 string. Return the byte array
  26 ** object with the encoded representation of the string, including
  27 ** the NULL terminator.
  28 */
  29 static int binarize(
  30   void * clientData,
  31   Tcl_Interp *interp,
  32   int objc,
  33   Tcl_Obj *CONST objv[]
  34 ){
  35   int len;
  36   char *bytes;
  37   Tcl_Obj *pRet;
  38   assert(objc==2);
  39
  40   bytes = Tcl_GetStringFromObj(objv[1], &len);
  41   pRet = Tcl_NewByteArrayObj((u8*)bytes, len+1);
  42   Tcl_SetObjResult(interp, pRet);
  43   return TCL_OK;
  44 }
  45
  46 /*
  47 ** Usage: test_value_overhead <repeat-count> <do-calls>.
  48 **
  49 ** This routine is used to test the overhead of calls to
  50 ** sqlite3_value_text(), on a value that contains a UTF-8 string. The idea
  51 ** is to figure out whether or not it is a problem to use sqlite3_value
  52 ** structures with collation sequence functions.
  53 **
  54 ** If <do-calls> is 0, then the calls to sqlite3_value_text() are not
  55 ** actually made.
  56 */
  57 static int test_value_overhead(
  58   void * clientData,
  59   Tcl_Interp *interp,
  60   int objc,
  61   Tcl_Obj *CONST objv[]
  62 ){
  63   int do_calls;
  64   int repeat_count;
  65   int i;
  66   Mem val;
  67
  68   if( objc!=3 ){
  69     Tcl_AppendResult(interp, "wrong # args: should be \"",
  70         Tcl_GetStringFromObj(objv[0], 0), " <repeat-count> <do-calls>", 0);
  71     return TCL_ERROR;
  72   }
  73
  74   if( Tcl_GetIntFromObj(interp, objv[1], &repeat_count) ) return TCL_ERROR;
  75   if( Tcl_GetIntFromObj(interp, objv[2], &do_calls) ) return TCL_ERROR;
  76
  77   val.flags = MEM_Str|MEM_Term|MEM_Static;
  78   val.z = "hello world";
  79   val.enc = SQLITE_UTF8;
  80
  81   for(i=0; i<repeat_count; i++){
  82     if( do_calls ){
  83       sqlite3_value_text(&val);
  84     }
  85   }
  86
  87   return TCL_OK;
  88 }
  89
  90 static u8 name_to_enc(Tcl_Interp *interp, Tcl_Obj *pObj){
  91   struct EncName {
  92     char *zName;
  93     u8 enc;
  94   } encnames[] = {
  95     { "UTF8", SQLITE_UTF8 },
  96     { "UTF16LE", SQLITE_UTF16LE },
  97     { "UTF16BE", SQLITE_UTF16BE },
  98     { "UTF16", SQLITE_UTF16 },
  99     { 0, 0 }
 100   };
 101   struct EncName *pEnc;
 102   char *z = Tcl_GetString(pObj);
 103   for(pEnc=&encnames[0]; pEnc->zName; pEnc++){
 104     if( 0==sqlite3StrICmp(z, pEnc->zName) ){
 105       break;
 106     }
 107   }
 108   if( !pEnc->enc ){
 109     Tcl_AppendResult(interp, "No such encoding: ", z, 0);
 110   }
 111   if( pEnc->enc==SQLITE_UTF16 ){
 112     return SQLITE_UTF16NATIVE;
 113   }
 114   return pEnc->enc;
 115 }
 116
 117 /*
 118 ** Usage:   test_translate <string/blob> <from enc> <to enc> ?<transient>?
 119 **
 120 */
 121 static int test_translate(
 122   void * clientData,
 123   Tcl_Interp *interp,
 124   int objc,
 125   Tcl_Obj *CONST objv[]
 126 ){
 127   u8 enc_from;
 128   u8 enc_to;
 129   sqlite3_value *pVal;
 130
 131   char *z;
 132   int len;
 133   void (*xDel)(void *p) = SQLITE_STATIC;
 134
 135   if( objc!=4 && objc!=5 ){
 136     Tcl_AppendResult(interp, "wrong # args: should be \"",
 137         Tcl_GetStringFromObj(objv[0], 0),
 138         " <string/blob> <from enc> <to enc>", 0
 139     );
 140     return TCL_ERROR;
 141   }
 142   if( objc==5 ){
 143     xDel = sqlite3_free;
 144   }
 145
 146   enc_from = name_to_enc(interp, objv[2]);
 147   if( !enc_from ) return TCL_ERROR;
 148   enc_to = name_to_enc(interp, objv[3]);
 149   if( !enc_to ) return TCL_ERROR;
 150
 151   pVal = sqlite3ValueNew(0);
 152
 153   if( enc_from==SQLITE_UTF8 ){
 154     z = Tcl_GetString(objv[1]);
 155     if( objc==5 ){
 156       z = sqlite3_mprintf("%s", z);
 157     }
 158     sqlite3ValueSetStr(pVal, -1, z, enc_from, xDel);
 159   }else{
 160     z = (char*)Tcl_GetByteArrayFromObj(objv[1], &len);
 161     if( objc==5 ){
 162       char *zTmp = z;
 163       z = sqlite3_malloc(len);
 164       memcpy(z, zTmp, len);
 165     }
 166     sqlite3ValueSetStr(pVal, -1, z, enc_from, xDel);
 167   }
 168
 169   z = (char *)sqlite3ValueText(pVal, enc_to);
 170   len = sqlite3ValueBytes(pVal, enc_to) + (enc_to==SQLITE_UTF8?1:2);
 171   Tcl_SetObjResult(interp, Tcl_NewByteArrayObj((u8*)z, len));
 172
 173   sqlite3ValueFree(pVal);
 174
 175   return TCL_OK;
 176 }
 177
 178 /*
 179 ** Usage: translate_selftest
 180 **
 181 ** Call sqlite3UtfSelfTest() to run the internal tests for unicode
 182 ** translation. If there is a problem an assert() will fail.
 183 **/
 184 void sqlite3UtfSelfTest(void);
 185 static int test_translate_selftest(
 186   void * clientData,
 187   Tcl_Interp *interp,
 188   int objc,
 189   Tcl_Obj *CONST objv[]
 190 ){
 191 #ifndef SQLITE_OMIT_UTF16
 192   sqlite3UtfSelfTest();
 193 #endif
 194   return SQLITE_OK;
 195 }
 196
 197
 198 /*
 199 ** Register commands with the TCL interpreter.
 200 */
 201 int Sqlitetest5_Init(Tcl_Interp *interp){
 202   static struct {
 203     char *zName;
 204     Tcl_ObjCmdProc *xProc;
 205   } aCmd[] = {
 206     { "binarize",                (Tcl_ObjCmdProc*)binarize },
 207     { "test_value_overhead",     (Tcl_ObjCmdProc*)test_value_overhead },
 208     { "test_translate",          (Tcl_ObjCmdProc*)test_translate     },
 209     { "translate_selftest",      (Tcl_ObjCmdProc*)test_translate_selftest},
 210   };
 211   int i;
 212   for(i=0; i<sizeof(aCmd)/sizeof(aCmd[0]); i++){
 213     Tcl_CreateObjCommand(interp, aCmd[i].zName, aCmd[i].xProc, 0, 0);
 214   }
 215   return SQLITE_OK;
 216 }