OLD | NEW |
| (Empty) |
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 } | |
OLD | NEW |