Merge Chromium + Blink git repositories
[chromium-blink-merge.git] / third_party / sqlite / sqlite-src-3080704 / src / test5.c
blob952e3325e5f4cf02f413bc9fb8d2b015c1abc7be
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.
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.
18 #include "sqliteInt.h"
19 #include "vdbeInt.h"
20 #include "tcl.h"
21 #include <stdlib.h>
22 #include <string.h>
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.
29 static int binarize(
30 void * clientData,
31 Tcl_Interp *interp,
32 int objc,
33 Tcl_Obj *CONST objv[]
35 int len;
36 char *bytes;
37 Tcl_Obj *pRet;
38 assert(objc==2);
40 bytes = Tcl_GetStringFromObj(objv[1], &len);
41 pRet = Tcl_NewByteArrayObj((u8*)bytes, len+1);
42 Tcl_SetObjResult(interp, pRet);
43 return TCL_OK;
47 ** Usage: test_value_overhead <repeat-count> <do-calls>.
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.
54 ** If <do-calls> is 0, then the calls to sqlite3_value_text() are not
55 ** actually made.
57 static int test_value_overhead(
58 void * clientData,
59 Tcl_Interp *interp,
60 int objc,
61 Tcl_Obj *CONST objv[]
63 int do_calls;
64 int repeat_count;
65 int i;
66 Mem val;
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;
74 if( Tcl_GetIntFromObj(interp, objv[1], &repeat_count) ) return TCL_ERROR;
75 if( Tcl_GetIntFromObj(interp, objv[2], &do_calls) ) return TCL_ERROR;
77 val.flags = MEM_Str|MEM_Term|MEM_Static;
78 val.z = "hello world";
79 val.enc = SQLITE_UTF8;
81 for(i=0; i<repeat_count; i++){
82 if( do_calls ){
83 sqlite3_value_text(&val);
87 return TCL_OK;
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 }
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;
108 if( !pEnc->enc ){
109 Tcl_AppendResult(interp, "No such encoding: ", z, 0);
111 if( pEnc->enc==SQLITE_UTF16 ){
112 return SQLITE_UTF16NATIVE;
114 return pEnc->enc;
118 ** Usage: test_translate <string/blob> <from enc> <to enc> ?<transient>?
121 static int test_translate(
122 void * clientData,
123 Tcl_Interp *interp,
124 int objc,
125 Tcl_Obj *CONST objv[]
127 u8 enc_from;
128 u8 enc_to;
129 sqlite3_value *pVal;
131 char *z;
132 int len;
133 void (*xDel)(void *p) = SQLITE_STATIC;
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
140 return TCL_ERROR;
142 if( objc==5 ){
143 xDel = sqlite3_free;
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;
151 pVal = sqlite3ValueNew(0);
153 if( enc_from==SQLITE_UTF8 ){
154 z = Tcl_GetString(objv[1]);
155 if( objc==5 ){
156 z = sqlite3_mprintf("%s", z);
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);
166 sqlite3ValueSetStr(pVal, -1, z, enc_from, xDel);
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));
173 sqlite3ValueFree(pVal);
175 return TCL_OK;
179 ** Usage: translate_selftest
181 ** Call sqlite3UtfSelfTest() to run the internal tests for unicode
182 ** translation. If there is a problem an assert() will fail.
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[]
191 #ifndef SQLITE_OMIT_UTF16
192 sqlite3UtfSelfTest();
193 #endif
194 return SQLITE_OK;
199 ** Register commands with the TCL interpreter.
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},
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);
215 return SQLITE_OK;