blob: 62abf13d4d9afca6444c8727b93aa4125764d743 [file] [log] [blame]
drhdc04c582002-02-24 01:55:15 +00001/*
2** 2002 February 23
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*************************************************************************
peter.d.reid60ec9142014-09-06 16:39:46 +000012** This file contains the C-language implementations for many of the SQL
drhede7ae32014-08-06 11:58:40 +000013** functions of SQLite. (Some function, and in particular the date and
14** time functions, are implemented separately.)
drhdc04c582002-02-24 01:55:15 +000015*/
drhb659e9b2005-01-28 01:29:08 +000016#include "sqliteInt.h"
drhd3a149e2002-02-24 17:12:53 +000017#include <stdlib.h>
drh0bce8352002-02-28 00:41:10 +000018#include <assert.h>
danielk197788208052004-05-25 01:13:20 +000019#include "vdbeInt.h"
drh0bce8352002-02-28 00:41:10 +000020
drh55ef4d92005-08-14 01:20:37 +000021/*
22** Return the collating function associated with a function.
23*/
danielk1977dc1bdc42004-06-11 10:51:27 +000024static CollSeq *sqlite3GetFuncCollSeq(sqlite3_context *context){
drha9e03b12015-03-12 06:46:52 +000025 VdbeOp *pOp;
26 assert( context->pVdbe!=0 );
27 pOp = &context->pVdbe->aOp[context->iOp-1];
drha15cc472014-09-25 13:17:30 +000028 assert( pOp->opcode==OP_CollSeq );
29 assert( pOp->p4type==P4_COLLSEQ );
30 return pOp->p4.pColl;
danielk1977dc1bdc42004-06-11 10:51:27 +000031}
32
drh0bce8352002-02-28 00:41:10 +000033/*
drh7a957892012-02-02 17:35:43 +000034** Indicate that the accumulator load should be skipped on this
35** iteration of the aggregate loop.
36*/
37static void sqlite3SkipAccumulatorLoad(sqlite3_context *context){
38 context->skipFlag = 1;
39}
40
41/*
drh0bce8352002-02-28 00:41:10 +000042** Implementation of the non-aggregate min() and max() functions
43*/
drhf9b596e2004-05-26 16:54:42 +000044static void minmaxFunc(
45 sqlite3_context *context,
46 int argc,
47 sqlite3_value **argv
48){
drh0bce8352002-02-28 00:41:10 +000049 int i;
drh268380c2004-02-25 13:47:31 +000050 int mask; /* 0 for min() or 0xffffffff for max() */
drhf9b596e2004-05-26 16:54:42 +000051 int iBest;
danielk1977dc1bdc42004-06-11 10:51:27 +000052 CollSeq *pColl;
drh0bce8352002-02-28 00:41:10 +000053
drh65595cd2009-02-02 16:32:55 +000054 assert( argc>1 );
drhc44af712004-09-02 15:53:56 +000055 mask = sqlite3_user_data(context)==0 ? 0 : -1;
danielk1977dc1bdc42004-06-11 10:51:27 +000056 pColl = sqlite3GetFuncCollSeq(context);
57 assert( pColl );
danielk1977c572ef72004-05-27 09:28:41 +000058 assert( mask==-1 || mask==0 );
drhf9b596e2004-05-26 16:54:42 +000059 iBest = 0;
drh9c054832004-05-31 18:51:57 +000060 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drhf9b596e2004-05-26 16:54:42 +000061 for(i=1; i<argc; i++){
drh9c054832004-05-31 18:51:57 +000062 if( sqlite3_value_type(argv[i])==SQLITE_NULL ) return;
danielk1977dc1bdc42004-06-11 10:51:27 +000063 if( (sqlite3MemCompare(argv[iBest], argv[i], pColl)^mask)>=0 ){
drh65595cd2009-02-02 16:32:55 +000064 testcase( mask==0 );
drhf9b596e2004-05-26 16:54:42 +000065 iBest = i;
drh0bce8352002-02-28 00:41:10 +000066 }
67 }
drhf4479502004-05-27 03:12:53 +000068 sqlite3_result_value(context, argv[iBest]);
drh0bce8352002-02-28 00:41:10 +000069}
drh0bce8352002-02-28 00:41:10 +000070
drh268380c2004-02-25 13:47:31 +000071/*
72** Return the type of the argument.
73*/
drhf9b596e2004-05-26 16:54:42 +000074static void typeofFunc(
75 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +000076 int NotUsed,
drhf9b596e2004-05-26 16:54:42 +000077 sqlite3_value **argv
78){
danielk197735bb9d02004-05-24 12:55:54 +000079 const char *z = 0;
danielk197762c14b32008-11-19 09:05:26 +000080 UNUSED_PARAMETER(NotUsed);
danielk197735bb9d02004-05-24 12:55:54 +000081 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +000082 case SQLITE_INTEGER: z = "integer"; break;
83 case SQLITE_TEXT: z = "text"; break;
84 case SQLITE_FLOAT: z = "real"; break;
85 case SQLITE_BLOB: z = "blob"; break;
drh65595cd2009-02-02 16:32:55 +000086 default: z = "null"; break;
danielk197735bb9d02004-05-24 12:55:54 +000087 }
danielk1977d8123362004-06-12 09:25:12 +000088 sqlite3_result_text(context, z, -1, SQLITE_STATIC);
drh0bce8352002-02-28 00:41:10 +000089}
90
drh5708d2d2005-06-22 10:53:59 +000091
92/*
drh0bce8352002-02-28 00:41:10 +000093** Implementation of the length() function
94*/
drhf9b596e2004-05-26 16:54:42 +000095static void lengthFunc(
96 sqlite3_context *context,
97 int argc,
98 sqlite3_value **argv
99){
drh0bce8352002-02-28 00:41:10 +0000100 int len;
101
102 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000103 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +0000104 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +0000105 case SQLITE_BLOB:
106 case SQLITE_INTEGER:
107 case SQLITE_FLOAT: {
drhf4479502004-05-27 03:12:53 +0000108 sqlite3_result_int(context, sqlite3_value_bytes(argv[0]));
drhf9b596e2004-05-26 16:54:42 +0000109 break;
110 }
drh9c054832004-05-31 18:51:57 +0000111 case SQLITE_TEXT: {
drh2646da72005-12-09 20:02:05 +0000112 const unsigned char *z = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +0000113 if( z==0 ) return;
drh4a919112007-05-15 11:55:09 +0000114 len = 0;
115 while( *z ){
116 len++;
117 SQLITE_SKIP_UTF8(z);
118 }
drhf4479502004-05-27 03:12:53 +0000119 sqlite3_result_int(context, len);
drhf9b596e2004-05-26 16:54:42 +0000120 break;
121 }
122 default: {
123 sqlite3_result_null(context);
124 break;
125 }
126 }
drh0bce8352002-02-28 00:41:10 +0000127}
128
129/*
drh2ba3ccc2009-12-08 02:06:08 +0000130** Implementation of the abs() function.
131**
132** IMP: R-23979-26855 The abs(X) function returns the absolute value of
133** the numeric argument X.
drh0bce8352002-02-28 00:41:10 +0000134*/
danielk19770ae8b832004-05-25 12:05:56 +0000135static void absFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drh0bce8352002-02-28 00:41:10 +0000136 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000137 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +0000138 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +0000139 case SQLITE_INTEGER: {
danielk1977f93bbbe2004-05-27 10:30:52 +0000140 i64 iVal = sqlite3_value_int64(argv[0]);
drh52fc8492006-02-23 21:43:55 +0000141 if( iVal<0 ){
drh693e6712014-01-24 22:58:00 +0000142 if( iVal==SMALLEST_INT64 ){
drheb091cd2013-11-09 19:47:15 +0000143 /* IMP: R-31676-45509 If X is the integer -9223372036854775808
144 ** then abs(X) throws an integer overflow error since there is no
drh2ba3ccc2009-12-08 02:06:08 +0000145 ** equivalent positive 64-bit two complement value. */
drh52fc8492006-02-23 21:43:55 +0000146 sqlite3_result_error(context, "integer overflow", -1);
147 return;
148 }
149 iVal = -iVal;
150 }
danielk1977f93bbbe2004-05-27 10:30:52 +0000151 sqlite3_result_int64(context, iVal);
drhf9b596e2004-05-26 16:54:42 +0000152 break;
153 }
drh9c054832004-05-31 18:51:57 +0000154 case SQLITE_NULL: {
drh2ba3ccc2009-12-08 02:06:08 +0000155 /* IMP: R-37434-19929 Abs(X) returns NULL if X is NULL. */
drhf9b596e2004-05-26 16:54:42 +0000156 sqlite3_result_null(context);
157 break;
158 }
159 default: {
drh2ba3ccc2009-12-08 02:06:08 +0000160 /* Because sqlite3_value_double() returns 0.0 if the argument is not
161 ** something that can be converted into a number, we have:
drh643091f2014-11-20 23:21:23 +0000162 ** IMP: R-01992-00519 Abs(X) returns 0.0 if X is a string or blob
163 ** that cannot be converted to a numeric value.
drh2ba3ccc2009-12-08 02:06:08 +0000164 */
danielk1977f93bbbe2004-05-27 10:30:52 +0000165 double rVal = sqlite3_value_double(argv[0]);
drh52fc8492006-02-23 21:43:55 +0000166 if( rVal<0 ) rVal = -rVal;
danielk1977f93bbbe2004-05-27 10:30:52 +0000167 sqlite3_result_double(context, rVal);
drhf9b596e2004-05-26 16:54:42 +0000168 break;
169 }
170 }
drh0bce8352002-02-28 00:41:10 +0000171}
172
173/*
drhd55e0722012-10-25 03:07:29 +0000174** Implementation of the instr() function.
175**
176** instr(haystack,needle) finds the first occurrence of needle
177** in haystack and returns the number of previous characters plus 1,
178** or 0 if needle does not occur within haystack.
179**
180** If both haystack and needle are BLOBs, then the result is one more than
181** the number of bytes in haystack prior to the first occurrence of needle,
182** or 0 if needle never occurs in haystack.
183*/
184static void instrFunc(
185 sqlite3_context *context,
186 int argc,
187 sqlite3_value **argv
188){
189 const unsigned char *zHaystack;
190 const unsigned char *zNeedle;
191 int nHaystack;
192 int nNeedle;
193 int typeHaystack, typeNeedle;
194 int N = 1;
195 int isText;
196
drh68c804b2012-12-04 11:03:11 +0000197 UNUSED_PARAMETER(argc);
drhd55e0722012-10-25 03:07:29 +0000198 typeHaystack = sqlite3_value_type(argv[0]);
199 typeNeedle = sqlite3_value_type(argv[1]);
200 if( typeHaystack==SQLITE_NULL || typeNeedle==SQLITE_NULL ) return;
201 nHaystack = sqlite3_value_bytes(argv[0]);
202 nNeedle = sqlite3_value_bytes(argv[1]);
203 if( typeHaystack==SQLITE_BLOB && typeNeedle==SQLITE_BLOB ){
204 zHaystack = sqlite3_value_blob(argv[0]);
205 zNeedle = sqlite3_value_blob(argv[1]);
206 isText = 0;
207 }else{
208 zHaystack = sqlite3_value_text(argv[0]);
209 zNeedle = sqlite3_value_text(argv[1]);
210 isText = 1;
211 }
212 while( nNeedle<=nHaystack && memcmp(zHaystack, zNeedle, nNeedle)!=0 ){
213 N++;
214 do{
215 nHaystack--;
216 zHaystack++;
217 }while( isText && (zHaystack[0]&0xc0)==0x80 );
218 }
219 if( nNeedle>nHaystack ) N = 0;
220 sqlite3_result_int(context, N);
221}
222
223/*
drha5c14162013-12-17 15:03:06 +0000224** Implementation of the printf() function.
225*/
226static void printfFunc(
227 sqlite3_context *context,
228 int argc,
229 sqlite3_value **argv
230){
231 PrintfArguments x;
232 StrAccum str;
233 const char *zFormat;
234 int n;
drhc0490572015-05-02 11:45:53 +0000235 sqlite3 *db = sqlite3_context_db_handle(context);
drha5c14162013-12-17 15:03:06 +0000236
237 if( argc>=1 && (zFormat = (const char*)sqlite3_value_text(argv[0]))!=0 ){
238 x.nArg = argc-1;
239 x.nUsed = 0;
240 x.apArg = argv+1;
drhc0490572015-05-02 11:45:53 +0000241 sqlite3StrAccumInit(&str, db, 0, 0, db->aLimit[SQLITE_LIMIT_LENGTH]);
drha5c14162013-12-17 15:03:06 +0000242 sqlite3XPrintf(&str, SQLITE_PRINTF_SQLFUNC, zFormat, &x);
243 n = str.nChar;
244 sqlite3_result_text(context, sqlite3StrAccumFinish(&str), n,
245 SQLITE_DYNAMIC);
246 }
247}
248
249/*
drhf764e6f2007-05-15 01:13:47 +0000250** Implementation of the substr() function.
251**
252** substr(x,p1,p2) returns p2 characters of x[] beginning with p1.
253** p1 is 1-indexed. So substr(x,1,1) returns the first character
254** of x. If x is text, then we actually count UTF-8 characters.
255** If x is a blob, then we count bytes.
256**
257** If p1 is negative, then we begin abs(p1) from the end of x[].
shaneh779b8f12009-11-12 05:04:50 +0000258**
drhf7b54962013-05-28 12:11:54 +0000259** If p2 is negative, return the p2 characters preceding p1.
drh0bce8352002-02-28 00:41:10 +0000260*/
drhf9b596e2004-05-26 16:54:42 +0000261static void substrFunc(
262 sqlite3_context *context,
263 int argc,
264 sqlite3_value **argv
265){
drh2646da72005-12-09 20:02:05 +0000266 const unsigned char *z;
267 const unsigned char *z2;
drh023ae032007-05-08 12:12:16 +0000268 int len;
drhf764e6f2007-05-15 01:13:47 +0000269 int p0type;
drh023ae032007-05-08 12:12:16 +0000270 i64 p1, p2;
drh65595cd2009-02-02 16:32:55 +0000271 int negP2 = 0;
drhf9b596e2004-05-26 16:54:42 +0000272
drh64f31512007-10-12 19:11:55 +0000273 assert( argc==3 || argc==2 );
drh8198d252009-02-01 19:42:37 +0000274 if( sqlite3_value_type(argv[1])==SQLITE_NULL
275 || (argc==3 && sqlite3_value_type(argv[2])==SQLITE_NULL)
276 ){
277 return;
278 }
drhf764e6f2007-05-15 01:13:47 +0000279 p0type = sqlite3_value_type(argv[0]);
drh4adc4cb2009-11-11 20:53:31 +0000280 p1 = sqlite3_value_int(argv[1]);
drhf764e6f2007-05-15 01:13:47 +0000281 if( p0type==SQLITE_BLOB ){
282 len = sqlite3_value_bytes(argv[0]);
283 z = sqlite3_value_blob(argv[0]);
284 if( z==0 ) return;
drh1f0feef2007-05-15 13:27:07 +0000285 assert( len==sqlite3_value_bytes(argv[0]) );
drhf764e6f2007-05-15 01:13:47 +0000286 }else{
287 z = sqlite3_value_text(argv[0]);
288 if( z==0 ) return;
drh4a919112007-05-15 11:55:09 +0000289 len = 0;
drh4adc4cb2009-11-11 20:53:31 +0000290 if( p1<0 ){
291 for(z2=z; *z2; len++){
292 SQLITE_SKIP_UTF8(z2);
293 }
drh4a919112007-05-15 11:55:09 +0000294 }
drhf764e6f2007-05-15 01:13:47 +0000295 }
drh883ad042015-02-19 00:29:11 +0000296#ifdef SQLITE_SUBSTR_COMPATIBILITY
297 /* If SUBSTR_COMPATIBILITY is defined then substr(X,0,N) work the same as
298 ** as substr(X,1,N) - it returns the first N characters of X. This
299 ** is essentially a back-out of the bug-fix in check-in [5fc125d362df4b8]
300 ** from 2009-02-02 for compatibility of applications that exploited the
301 ** old buggy behavior. */
302 if( p1==0 ) p1 = 1; /* <rdar://problem/6778339> */
303#endif
drh64f31512007-10-12 19:11:55 +0000304 if( argc==3 ){
305 p2 = sqlite3_value_int(argv[2]);
drh65595cd2009-02-02 16:32:55 +0000306 if( p2<0 ){
307 p2 = -p2;
308 negP2 = 1;
309 }
drh64f31512007-10-12 19:11:55 +0000310 }else{
drhbb4957f2008-03-20 14:03:29 +0000311 p2 = sqlite3_context_db_handle(context)->aLimit[SQLITE_LIMIT_LENGTH];
drh64f31512007-10-12 19:11:55 +0000312 }
drh0bce8352002-02-28 00:41:10 +0000313 if( p1<0 ){
drh89425d52002-02-28 03:04:48 +0000314 p1 += len;
drh653bc752002-02-28 03:31:10 +0000315 if( p1<0 ){
316 p2 += p1;
drh65595cd2009-02-02 16:32:55 +0000317 if( p2<0 ) p2 = 0;
drh653bc752002-02-28 03:31:10 +0000318 p1 = 0;
319 }
drh0bce8352002-02-28 00:41:10 +0000320 }else if( p1>0 ){
321 p1--;
drh65595cd2009-02-02 16:32:55 +0000322 }else if( p2>0 ){
323 p2--;
drh0bce8352002-02-28 00:41:10 +0000324 }
drh65595cd2009-02-02 16:32:55 +0000325 if( negP2 ){
326 p1 -= p2;
drh4e79c592009-02-01 19:23:32 +0000327 if( p1<0 ){
328 p2 += p1;
329 p1 = 0;
330 }
331 }
drh65595cd2009-02-02 16:32:55 +0000332 assert( p1>=0 && p2>=0 );
drhf764e6f2007-05-15 01:13:47 +0000333 if( p0type!=SQLITE_BLOB ){
drh4a919112007-05-15 11:55:09 +0000334 while( *z && p1 ){
335 SQLITE_SKIP_UTF8(z);
336 p1--;
drhf764e6f2007-05-15 01:13:47 +0000337 }
drh4a919112007-05-15 11:55:09 +0000338 for(z2=z; *z2 && p2; p2--){
339 SQLITE_SKIP_UTF8(z2);
drhf764e6f2007-05-15 01:13:47 +0000340 }
drhbbf483f2014-09-09 20:30:24 +0000341 sqlite3_result_text64(context, (char*)z, z2-z, SQLITE_TRANSIENT,
342 SQLITE_UTF8);
drhf764e6f2007-05-15 01:13:47 +0000343 }else{
drh4adc4cb2009-11-11 20:53:31 +0000344 if( p1+p2>len ){
345 p2 = len-p1;
346 if( p2<0 ) p2 = 0;
347 }
drhbbf483f2014-09-09 20:30:24 +0000348 sqlite3_result_blob64(context, (char*)&z[p1], (u64)p2, SQLITE_TRANSIENT);
drh0bce8352002-02-28 00:41:10 +0000349 }
drh0bce8352002-02-28 00:41:10 +0000350}
351
352/*
353** Implementation of the round() function
354*/
shanefbd60f82009-02-04 03:59:25 +0000355#ifndef SQLITE_OMIT_FLOATING_POINT
danielk19770ae8b832004-05-25 12:05:56 +0000356static void roundFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
danielk197751ad0ec2004-05-24 12:39:02 +0000357 int n = 0;
drh0bce8352002-02-28 00:41:10 +0000358 double r;
drh50d654d2009-06-03 01:24:54 +0000359 char *zBuf;
drh0bce8352002-02-28 00:41:10 +0000360 assert( argc==1 || argc==2 );
danielk197751ad0ec2004-05-24 12:39:02 +0000361 if( argc==2 ){
drh9c054832004-05-31 18:51:57 +0000362 if( SQLITE_NULL==sqlite3_value_type(argv[1]) ) return;
danielk197751ad0ec2004-05-24 12:39:02 +0000363 n = sqlite3_value_int(argv[1]);
364 if( n>30 ) n = 30;
365 if( n<0 ) n = 0;
366 }
drhd589a922006-03-02 03:02:48 +0000367 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drh4f26d6c2004-05-26 23:25:30 +0000368 r = sqlite3_value_double(argv[0]);
shaneh147e1762010-02-17 04:19:27 +0000369 /* If Y==0 and X will fit in a 64-bit int,
370 ** handle the rounding directly,
371 ** otherwise use printf.
372 */
373 if( n==0 && r>=0 && r<LARGEST_INT64-1 ){
374 r = (double)((sqlite_int64)(r+0.5));
375 }else if( n==0 && r<0 && (-r)<LARGEST_INT64-1 ){
376 r = -(double)((sqlite_int64)((-r)+0.5));
drh50d654d2009-06-03 01:24:54 +0000377 }else{
shaneh147e1762010-02-17 04:19:27 +0000378 zBuf = sqlite3_mprintf("%.*f",n,r);
379 if( zBuf==0 ){
380 sqlite3_result_error_nomem(context);
381 return;
382 }
drh9339da12010-09-30 00:50:49 +0000383 sqlite3AtoF(zBuf, &r, sqlite3Strlen30(zBuf), SQLITE_UTF8);
drh50d654d2009-06-03 01:24:54 +0000384 sqlite3_free(zBuf);
drh50d654d2009-06-03 01:24:54 +0000385 }
shaneh147e1762010-02-17 04:19:27 +0000386 sqlite3_result_double(context, r);
drh0bce8352002-02-28 00:41:10 +0000387}
shanefbd60f82009-02-04 03:59:25 +0000388#endif
drhdc04c582002-02-24 01:55:15 +0000389
danielk197726783a52007-08-29 14:06:22 +0000390/*
drhf3cdcdc2015-04-29 16:50:28 +0000391** Allocate nByte bytes of space using sqlite3Malloc(). If the
danielk197726783a52007-08-29 14:06:22 +0000392** allocation fails, call sqlite3_result_error_nomem() to notify
drh27e62db2009-04-02 10:16:17 +0000393** the database handle that malloc() has failed and return NULL.
394** If nByte is larger than the maximum string or blob length, then
395** raise an SQLITE_TOOBIG exception and return NULL.
danielk197726783a52007-08-29 14:06:22 +0000396*/
drhb1a6c3c2008-03-20 16:30:17 +0000397static void *contextMalloc(sqlite3_context *context, i64 nByte){
drhbb4957f2008-03-20 14:03:29 +0000398 char *z;
drh27e62db2009-04-02 10:16:17 +0000399 sqlite3 *db = sqlite3_context_db_handle(context);
drhef31c6a2009-04-02 09:07:12 +0000400 assert( nByte>0 );
drh27e62db2009-04-02 10:16:17 +0000401 testcase( nByte==db->aLimit[SQLITE_LIMIT_LENGTH] );
402 testcase( nByte==db->aLimit[SQLITE_LIMIT_LENGTH]+1 );
403 if( nByte>db->aLimit[SQLITE_LIMIT_LENGTH] ){
drhbb4957f2008-03-20 14:03:29 +0000404 sqlite3_result_error_toobig(context);
405 z = 0;
406 }else{
drhda4ca9d2014-09-09 17:27:35 +0000407 z = sqlite3Malloc(nByte);
drhef31c6a2009-04-02 09:07:12 +0000408 if( !z ){
drhbb4957f2008-03-20 14:03:29 +0000409 sqlite3_result_error_nomem(context);
410 }
danielk1977a1644fd2007-08-29 12:31:25 +0000411 }
412 return z;
413}
414
drhdc04c582002-02-24 01:55:15 +0000415/*
416** Implementation of the upper() and lower() SQL functions.
417*/
danielk19770ae8b832004-05-25 12:05:56 +0000418static void upperFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drh7a521cf2007-04-25 18:23:52 +0000419 char *z1;
420 const char *z2;
drh9310ef22007-04-27 17:16:20 +0000421 int i, n;
drh1d34fde2009-02-03 15:50:33 +0000422 UNUSED_PARAMETER(argc);
drh7a521cf2007-04-25 18:23:52 +0000423 z2 = (char*)sqlite3_value_text(argv[0]);
drh1f0feef2007-05-15 13:27:07 +0000424 n = sqlite3_value_bytes(argv[0]);
425 /* Verify that the call to _bytes() does not invalidate the _text() pointer */
426 assert( z2==(char*)sqlite3_value_text(argv[0]) );
drh7a521cf2007-04-25 18:23:52 +0000427 if( z2 ){
drhb1a6c3c2008-03-20 16:30:17 +0000428 z1 = contextMalloc(context, ((i64)n)+1);
drh7a521cf2007-04-25 18:23:52 +0000429 if( z1 ){
drhdf901d32011-10-13 18:00:11 +0000430 for(i=0; i<n; i++){
431 z1[i] = (char)sqlite3Toupper(z2[i]);
drh7a521cf2007-04-25 18:23:52 +0000432 }
drhdf901d32011-10-13 18:00:11 +0000433 sqlite3_result_text(context, z1, n, sqlite3_free);
drh7a521cf2007-04-25 18:23:52 +0000434 }
drhdc04c582002-02-24 01:55:15 +0000435 }
436}
danielk19770ae8b832004-05-25 12:05:56 +0000437static void lowerFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drhdf901d32011-10-13 18:00:11 +0000438 char *z1;
drh7a521cf2007-04-25 18:23:52 +0000439 const char *z2;
drh9310ef22007-04-27 17:16:20 +0000440 int i, n;
drh1d34fde2009-02-03 15:50:33 +0000441 UNUSED_PARAMETER(argc);
drh7a521cf2007-04-25 18:23:52 +0000442 z2 = (char*)sqlite3_value_text(argv[0]);
drh1f0feef2007-05-15 13:27:07 +0000443 n = sqlite3_value_bytes(argv[0]);
444 /* Verify that the call to _bytes() does not invalidate the _text() pointer */
445 assert( z2==(char*)sqlite3_value_text(argv[0]) );
drh7a521cf2007-04-25 18:23:52 +0000446 if( z2 ){
drhb1a6c3c2008-03-20 16:30:17 +0000447 z1 = contextMalloc(context, ((i64)n)+1);
drh7a521cf2007-04-25 18:23:52 +0000448 if( z1 ){
drhdf901d32011-10-13 18:00:11 +0000449 for(i=0; i<n; i++){
450 z1[i] = sqlite3Tolower(z2[i]);
drh7a521cf2007-04-25 18:23:52 +0000451 }
drhdf901d32011-10-13 18:00:11 +0000452 sqlite3_result_text(context, z1, n, sqlite3_free);
drh7a521cf2007-04-25 18:23:52 +0000453 }
drhdc04c582002-02-24 01:55:15 +0000454 }
455}
456
drhae6bb952009-11-11 00:24:31 +0000457/*
drhcca9f3d2013-09-06 15:23:29 +0000458** Some functions like COALESCE() and IFNULL() and UNLIKELY() are implemented
459** as VDBE code so that unused argument values do not have to be computed.
460** However, we still need some kind of function implementation for this
461** routines in the function table. The noopFunc macro provides this.
462** noopFunc will never be called so it doesn't matter what the implementation
463** is. We might as well use the "version()" function as a substitute.
drhae6bb952009-11-11 00:24:31 +0000464*/
drhcca9f3d2013-09-06 15:23:29 +0000465#define noopFunc versionFunc /* Substitute function - never called */
drh3212e182002-02-28 00:46:26 +0000466
467/*
drhf9ffac92002-03-02 19:00:31 +0000468** Implementation of random(). Return a random integer.
469*/
drhf9b596e2004-05-26 16:54:42 +0000470static void randomFunc(
471 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000472 int NotUsed,
473 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000474){
drh52fc8492006-02-23 21:43:55 +0000475 sqlite_int64 r;
danielk197762c14b32008-11-19 09:05:26 +0000476 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drh2fa18682008-03-19 14:15:34 +0000477 sqlite3_randomness(sizeof(r), &r);
drh3034e3d2009-04-02 14:05:21 +0000478 if( r<0 ){
479 /* We need to prevent a random number of 0x8000000000000000
480 ** (or -9223372036854775808) since when you do abs() of that
481 ** number of you get the same value back again. To do this
482 ** in a way that is testable, mask the sign bit off of negative
483 ** values, resulting in a positive value. Then take the
484 ** 2s complement of that positive value. The end result can
485 ** therefore be no less than -9223372036854775807.
486 */
drhaf8001b2012-02-11 19:53:24 +0000487 r = -(r & LARGEST_INT64);
drh3034e3d2009-04-02 14:05:21 +0000488 }
drh52fc8492006-02-23 21:43:55 +0000489 sqlite3_result_int64(context, r);
drhf9ffac92002-03-02 19:00:31 +0000490}
491
492/*
drh137c7282007-01-29 17:58:28 +0000493** Implementation of randomblob(N). Return a random blob
494** that is N bytes long.
drh63cf66f2007-01-29 15:50:05 +0000495*/
drh137c7282007-01-29 17:58:28 +0000496static void randomBlob(
drh63cf66f2007-01-29 15:50:05 +0000497 sqlite3_context *context,
498 int argc,
499 sqlite3_value **argv
500){
drh137c7282007-01-29 17:58:28 +0000501 int n;
502 unsigned char *p;
drh63cf66f2007-01-29 15:50:05 +0000503 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000504 UNUSED_PARAMETER(argc);
drh63cf66f2007-01-29 15:50:05 +0000505 n = sqlite3_value_int(argv[0]);
drh023ae032007-05-08 12:12:16 +0000506 if( n<1 ){
507 n = 1;
508 }
danielk1977a1644fd2007-08-29 12:31:25 +0000509 p = contextMalloc(context, n);
drh02d85832007-05-07 19:31:15 +0000510 if( p ){
drh2fa18682008-03-19 14:15:34 +0000511 sqlite3_randomness(n, p);
drh17435752007-08-16 04:30:38 +0000512 sqlite3_result_blob(context, (char*)p, n, sqlite3_free);
drh02d85832007-05-07 19:31:15 +0000513 }
drh63cf66f2007-01-29 15:50:05 +0000514}
515
516/*
drh6ed41ad2002-04-06 14:10:47 +0000517** Implementation of the last_insert_rowid() SQL function. The return
danielk197724b03fd2004-05-10 10:34:34 +0000518** value is the same as the sqlite3_last_insert_rowid() API function.
drh6ed41ad2002-04-06 14:10:47 +0000519*/
danielk197751ad0ec2004-05-24 12:39:02 +0000520static void last_insert_rowid(
danielk19770ae8b832004-05-25 12:05:56 +0000521 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000522 int NotUsed,
523 sqlite3_value **NotUsed2
danielk197751ad0ec2004-05-24 12:39:02 +0000524){
drhfa4a4b92008-03-19 21:45:51 +0000525 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000526 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000527 /* IMP: R-51513-12026 The last_insert_rowid() SQL function is a
528 ** wrapper around the sqlite3_last_insert_rowid() C/C++ interface
529 ** function. */
drhf9b596e2004-05-26 16:54:42 +0000530 sqlite3_result_int64(context, sqlite3_last_insert_rowid(db));
drh6ed41ad2002-04-06 14:10:47 +0000531}
532
rdcf146a772004-02-25 22:51:06 +0000533/*
drhab2f1f92010-01-11 18:26:42 +0000534** Implementation of the changes() SQL function.
535**
536** IMP: R-62073-11209 The changes() SQL function is a wrapper
537** around the sqlite3_changes() C/C++ function and hence follows the same
538** rules for counting changes.
rdcf146a772004-02-25 22:51:06 +0000539*/
danielk1977b28af712004-06-21 06:50:26 +0000540static void changes(
drhf9b596e2004-05-26 16:54:42 +0000541 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000542 int NotUsed,
543 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000544){
drhfa4a4b92008-03-19 21:45:51 +0000545 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000546 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhf4479502004-05-27 03:12:53 +0000547 sqlite3_result_int(context, sqlite3_changes(db));
rdcb0c374f2004-02-20 22:53:38 +0000548}
rdcf146a772004-02-25 22:51:06 +0000549
550/*
danielk1977b28af712004-06-21 06:50:26 +0000551** Implementation of the total_changes() SQL function. The return value is
552** the same as the sqlite3_total_changes() API function.
rdcf146a772004-02-25 22:51:06 +0000553*/
danielk1977b28af712004-06-21 06:50:26 +0000554static void total_changes(
555 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000556 int NotUsed,
557 sqlite3_value **NotUsed2
danielk197751ad0ec2004-05-24 12:39:02 +0000558){
drhfa4a4b92008-03-19 21:45:51 +0000559 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000560 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000561 /* IMP: R-52756-41993 This function is a wrapper around the
562 ** sqlite3_total_changes() C/C++ interface. */
danielk1977b28af712004-06-21 06:50:26 +0000563 sqlite3_result_int(context, sqlite3_total_changes(db));
rdcb0c374f2004-02-20 22:53:38 +0000564}
565
drh6ed41ad2002-04-06 14:10:47 +0000566/*
drh4e5ffc52004-08-31 00:52:37 +0000567** A structure defining how to do GLOB-style comparisons.
danielk1977d02eb1f2004-06-06 09:44:03 +0000568*/
drh4e5ffc52004-08-31 00:52:37 +0000569struct compareInfo {
570 u8 matchAll;
571 u8 matchOne;
572 u8 matchSet;
573 u8 noCase;
danielk1977d02eb1f2004-06-06 09:44:03 +0000574};
drh55ef4d92005-08-14 01:20:37 +0000575
drhb9175ae2007-12-07 18:39:04 +0000576/*
577** For LIKE and GLOB matching on EBCDIC machines, assume that every
578** character is exactly one byte in size. Also, all characters are
579** able to participate in upper-case-to-lower-case mappings in EBCDIC
580** whereas only characters less than 0x80 do in ASCII.
581*/
582#if defined(SQLITE_EBCDIC)
drh88b33222014-09-25 03:51:37 +0000583# define sqlite3Utf8Read(A) (*((*A)++))
584# define GlobUpperToLower(A) A = sqlite3UpperToLower[A]
585# define GlobUpperToLowerAscii(A) A = sqlite3UpperToLower[A]
drhb9175ae2007-12-07 18:39:04 +0000586#else
drh88b33222014-09-25 03:51:37 +0000587# define GlobUpperToLower(A) if( A<=0x7f ){ A = sqlite3UpperToLower[A]; }
588# define GlobUpperToLowerAscii(A) A = sqlite3UpperToLower[A]
drhb9175ae2007-12-07 18:39:04 +0000589#endif
590
drh4e5ffc52004-08-31 00:52:37 +0000591static const struct compareInfo globInfo = { '*', '?', '[', 0 };
drh55ef4d92005-08-14 01:20:37 +0000592/* The correct SQL-92 behavior is for the LIKE operator to ignore
593** case. Thus 'a' LIKE 'A' would be true. */
594static const struct compareInfo likeInfoNorm = { '%', '_', 0, 1 };
595/* If SQLITE_CASE_SENSITIVE_LIKE is defined, then the LIKE operator
596** is case sensitive causing 'a' LIKE 'A' to be false */
597static const struct compareInfo likeInfoAlt = { '%', '_', 0, 0 };
danielk1977d02eb1f2004-06-06 09:44:03 +0000598
599/*
drh4e5ffc52004-08-31 00:52:37 +0000600** Compare two UTF-8 strings for equality where the first string can
drh9fdfdc82014-09-25 11:08:57 +0000601** potentially be a "glob" or "like" expression. Return true (1) if they
drh4e5ffc52004-08-31 00:52:37 +0000602** are the same and false (0) if they are different.
drh0ac65892002-04-20 14:24:41 +0000603**
drh4e5ffc52004-08-31 00:52:37 +0000604** Globbing rules:
drh0ac65892002-04-20 14:24:41 +0000605**
drh4e5ffc52004-08-31 00:52:37 +0000606** '*' Matches any sequence of zero or more characters.
danielk1977d02eb1f2004-06-06 09:44:03 +0000607**
drh4e5ffc52004-08-31 00:52:37 +0000608** '?' Matches exactly one character.
609**
610** [...] Matches one character from the enclosed list of
611** characters.
612**
613** [^...] Matches one character not in the enclosed list.
614**
615** With the [...] and [^...] matching, a ']' character can be included
616** in the list by making it the first character after '[' or '^'. A
617** range of characters can be specified using '-'. Example:
618** "[a-z]" matches any single lower-case letter. To match a '-', make
619** it the last character in the list.
620**
drh9fdfdc82014-09-25 11:08:57 +0000621** Like matching rules:
622**
623** '%' Matches any sequence of zero or more characters
624**
625*** '_' Matches any one character
626**
627** Ec Where E is the "esc" character and c is any other
628** character, including '%', '_', and esc, match exactly c.
629**
630** The comments through this routine usually assume glob matching.
631**
drh4e5ffc52004-08-31 00:52:37 +0000632** This routine is usually quick, but can be N**2 in the worst case.
drh0ac65892002-04-20 14:24:41 +0000633*/
danielk19777c6303c2004-11-17 16:41:29 +0000634static int patternCompare(
drh4e5ffc52004-08-31 00:52:37 +0000635 const u8 *zPattern, /* The glob pattern */
636 const u8 *zString, /* The string to compare against the glob */
danielk19777c6303c2004-11-17 16:41:29 +0000637 const struct compareInfo *pInfo, /* Information about how to do the compare */
drh0a32fa62011-06-13 12:19:21 +0000638 u32 esc /* The escape character */
danielk197751ad0ec2004-05-24 12:39:02 +0000639){
drh9fdfdc82014-09-25 11:08:57 +0000640 u32 c, c2; /* Next pattern and input string chars */
641 u32 matchOne = pInfo->matchOne; /* "?" or "_" */
642 u32 matchAll = pInfo->matchAll; /* "*" or "%" */
643 u32 matchOther; /* "[" or the escape character */
644 u8 noCase = pInfo->noCase; /* True if uppercase==lowercase */
645 const u8 *zEscaped = 0; /* One past the last escaped input char */
drh88b33222014-09-25 03:51:37 +0000646
647 /* The GLOB operator does not have an ESCAPE clause. And LIKE does not
648 ** have the matchSet operator. So we either have to look for one or
649 ** the other, never both. Hence the single variable matchOther is used
650 ** to store the one we have to look for.
651 */
652 matchOther = esc ? esc : pInfo->matchSet;
danielk1977d02eb1f2004-06-06 09:44:03 +0000653
drh42610962012-09-17 18:56:32 +0000654 while( (c = sqlite3Utf8Read(&zPattern))!=0 ){
drh9fdfdc82014-09-25 11:08:57 +0000655 if( c==matchAll ){ /* Match "*" */
656 /* Skip over multiple "*" characters in the pattern. If there
657 ** are also "?" characters, skip those as well, but consume a
658 ** single character of the input string for each "?" skipped */
drh42610962012-09-17 18:56:32 +0000659 while( (c=sqlite3Utf8Read(&zPattern)) == matchAll
drh66150952007-07-23 19:12:41 +0000660 || c == matchOne ){
drh42610962012-09-17 18:56:32 +0000661 if( c==matchOne && sqlite3Utf8Read(&zString)==0 ){
drh66150952007-07-23 19:12:41 +0000662 return 0;
drh4e5ffc52004-08-31 00:52:37 +0000663 }
danielk1977ad7dd422004-06-06 12:41:49 +0000664 }
drh66150952007-07-23 19:12:41 +0000665 if( c==0 ){
drh9fdfdc82014-09-25 11:08:57 +0000666 return 1; /* "*" at the end of the pattern matches */
drh88b33222014-09-25 03:51:37 +0000667 }else if( c==matchOther ){
668 if( esc ){
669 c = sqlite3Utf8Read(&zPattern);
670 if( c==0 ) return 0;
671 }else{
drh9fdfdc82014-09-25 11:08:57 +0000672 /* "[...]" immediately follows the "*". We have to do a slow
673 ** recursive search in this case, but it is an unusual case. */
drh88b33222014-09-25 03:51:37 +0000674 assert( matchOther<0x80 ); /* '[' is a single-byte character */
675 while( *zString
676 && patternCompare(&zPattern[-1],zString,pInfo,esc)==0 ){
677 SQLITE_SKIP_UTF8(zString);
678 }
679 return *zString!=0;
drh66150952007-07-23 19:12:41 +0000680 }
drh66150952007-07-23 19:12:41 +0000681 }
drh9fdfdc82014-09-25 11:08:57 +0000682
683 /* At this point variable c contains the first character of the
684 ** pattern string past the "*". Search in the input string for the
685 ** first matching character and recursively contine the match from
686 ** that point.
687 **
688 ** For a case-insensitive search, set variable cx to be the same as
689 ** c but in the other case and search the input string for either
690 ** c or cx.
691 */
692 if( c<=0x80 ){
693 u32 cx;
694 if( noCase ){
695 cx = sqlite3Toupper(c);
696 c = sqlite3Tolower(c);
drh66150952007-07-23 19:12:41 +0000697 }else{
drh9fdfdc82014-09-25 11:08:57 +0000698 cx = c;
drh4e5ffc52004-08-31 00:52:37 +0000699 }
drh9fdfdc82014-09-25 11:08:57 +0000700 while( (c2 = *(zString++))!=0 ){
701 if( c2!=c && c2!=cx ) continue;
702 if( patternCompare(zPattern,zString,pInfo,esc) ) return 1;
703 }
704 }else{
705 while( (c2 = sqlite3Utf8Read(&zString))!=0 ){
706 if( c2!=c ) continue;
707 if( patternCompare(zPattern,zString,pInfo,esc) ) return 1;
708 }
drh66150952007-07-23 19:12:41 +0000709 }
710 return 0;
drh88b33222014-09-25 03:51:37 +0000711 }
drh88b33222014-09-25 03:51:37 +0000712 if( c==matchOther ){
713 if( esc ){
714 c = sqlite3Utf8Read(&zPattern);
715 if( c==0 ) return 0;
drh9fdfdc82014-09-25 11:08:57 +0000716 zEscaped = zPattern;
drh88b33222014-09-25 03:51:37 +0000717 }else{
718 u32 prior_c = 0;
drh9fdfdc82014-09-25 11:08:57 +0000719 int seen = 0;
720 int invert = 0;
drh88b33222014-09-25 03:51:37 +0000721 c = sqlite3Utf8Read(&zString);
722 if( c==0 ) return 0;
723 c2 = sqlite3Utf8Read(&zPattern);
724 if( c2=='^' ){
725 invert = 1;
726 c2 = sqlite3Utf8Read(&zPattern);
727 }
728 if( c2==']' ){
729 if( c==']' ) seen = 1;
730 c2 = sqlite3Utf8Read(&zPattern);
731 }
732 while( c2 && c2!=']' ){
733 if( c2=='-' && zPattern[0]!=']' && zPattern[0]!=0 && prior_c>0 ){
734 c2 = sqlite3Utf8Read(&zPattern);
735 if( c>=prior_c && c<=c2 ) seen = 1;
736 prior_c = 0;
737 }else{
738 if( c==c2 ){
739 seen = 1;
740 }
741 prior_c = c2;
742 }
743 c2 = sqlite3Utf8Read(&zPattern);
744 }
745 if( c2==0 || (seen ^ invert)==0 ){
746 return 0;
747 }
748 continue;
749 }
750 }
751 c2 = sqlite3Utf8Read(&zString);
752 if( c==c2 ) continue;
drh9fdfdc82014-09-25 11:08:57 +0000753 if( noCase && c<0x80 && c2<0x80 && sqlite3Tolower(c)==sqlite3Tolower(c2) ){
754 continue;
755 }
756 if( c==matchOne && zPattern!=zEscaped && c2!=0 ) continue;
757 return 0;
danielk197751ad0ec2004-05-24 12:39:02 +0000758 }
drh4e5ffc52004-08-31 00:52:37 +0000759 return *zString==0;
drh0ac65892002-04-20 14:24:41 +0000760}
drh4e5ffc52004-08-31 00:52:37 +0000761
drh55ef4d92005-08-14 01:20:37 +0000762/*
drh56282a52013-04-10 16:13:38 +0000763** The sqlite3_strglob() interface.
764*/
765int sqlite3_strglob(const char *zGlobPattern, const char *zString){
766 return patternCompare((u8*)zGlobPattern, (u8*)zString, &globInfo, 0)==0;
767}
768
769/*
drh55ef4d92005-08-14 01:20:37 +0000770** Count the number of times that the LIKE operator (or GLOB which is
771** just a variation of LIKE) gets called. This is used for testing
772** only.
773*/
774#ifdef SQLITE_TEST
775int sqlite3_like_count = 0;
776#endif
777
danielk19773f6b0872004-06-17 05:36:44 +0000778
779/*
780** Implementation of the like() SQL function. This function implements
781** the build-in LIKE operator. The first argument to the function is the
782** pattern and the second argument is the string. So, the SQL statements:
783**
784** A LIKE B
785**
786** is implemented as like(B,A).
787**
drh55ef4d92005-08-14 01:20:37 +0000788** This same function (with a different compareInfo structure) computes
789** the GLOB operator.
danielk19773f6b0872004-06-17 05:36:44 +0000790*/
791static void likeFunc(
792 sqlite3_context *context,
793 int argc,
794 sqlite3_value **argv
795){
drhbeb818d2007-05-08 15:34:47 +0000796 const unsigned char *zA, *zB;
drh0a32fa62011-06-13 12:19:21 +0000797 u32 escape = 0;
drh27e62db2009-04-02 10:16:17 +0000798 int nPat;
drhbb4957f2008-03-20 14:03:29 +0000799 sqlite3 *db = sqlite3_context_db_handle(context);
drhbeb818d2007-05-08 15:34:47 +0000800
drh1f0feef2007-05-15 13:27:07 +0000801 zB = sqlite3_value_text(argv[0]);
802 zA = sqlite3_value_text(argv[1]);
803
drhbeb818d2007-05-08 15:34:47 +0000804 /* Limit the length of the LIKE or GLOB pattern to avoid problems
805 ** of deep recursion and N*N behavior in patternCompare().
806 */
drh27e62db2009-04-02 10:16:17 +0000807 nPat = sqlite3_value_bytes(argv[0]);
808 testcase( nPat==db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH] );
809 testcase( nPat==db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH]+1 );
810 if( nPat > db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH] ){
drhbeb818d2007-05-08 15:34:47 +0000811 sqlite3_result_error(context, "LIKE or GLOB pattern too complex", -1);
812 return;
813 }
drh1f0feef2007-05-15 13:27:07 +0000814 assert( zB==sqlite3_value_text(argv[0]) ); /* Encoding did not change */
drhbeb818d2007-05-08 15:34:47 +0000815
danielk19777c6303c2004-11-17 16:41:29 +0000816 if( argc==3 ){
817 /* The escape character string must consist of a single UTF-8 character.
818 ** Otherwise, return an error.
819 */
820 const unsigned char *zEsc = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +0000821 if( zEsc==0 ) return;
drhee858132007-05-08 20:37:38 +0000822 if( sqlite3Utf8CharLen((char*)zEsc, -1)!=1 ){
danielk19777c6303c2004-11-17 16:41:29 +0000823 sqlite3_result_error(context,
824 "ESCAPE expression must be a single character", -1);
825 return;
826 }
drh42610962012-09-17 18:56:32 +0000827 escape = sqlite3Utf8Read(&zEsc);
danielk19777c6303c2004-11-17 16:41:29 +0000828 }
danielk19773f6b0872004-06-17 05:36:44 +0000829 if( zA && zB ){
drh55ef4d92005-08-14 01:20:37 +0000830 struct compareInfo *pInfo = sqlite3_user_data(context);
831#ifdef SQLITE_TEST
832 sqlite3_like_count++;
833#endif
drhbeb818d2007-05-08 15:34:47 +0000834
danielk1977b56fe1f2007-05-09 08:24:44 +0000835 sqlite3_result_int(context, patternCompare(zB, zA, pInfo, escape));
danielk197751ad0ec2004-05-24 12:39:02 +0000836 }
drh8912d102002-05-26 21:34:58 +0000837}
838
839/*
840** Implementation of the NULLIF(x,y) function. The result is the first
841** argument if the arguments are different. The result is NULL if the
842** arguments are equal to each other.
843*/
drhf9b596e2004-05-26 16:54:42 +0000844static void nullifFunc(
845 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000846 int NotUsed,
drhf9b596e2004-05-26 16:54:42 +0000847 sqlite3_value **argv
848){
danielk1977dc1bdc42004-06-11 10:51:27 +0000849 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk197762c14b32008-11-19 09:05:26 +0000850 UNUSED_PARAMETER(NotUsed);
danielk1977dc1bdc42004-06-11 10:51:27 +0000851 if( sqlite3MemCompare(argv[0], argv[1], pColl)!=0 ){
drhf4479502004-05-27 03:12:53 +0000852 sqlite3_result_value(context, argv[0]);
drh8912d102002-05-26 21:34:58 +0000853 }
drh0ac65892002-04-20 14:24:41 +0000854}
855
drh647cb0e2002-11-04 19:32:25 +0000856/*
drh47baebc2009-08-14 16:01:24 +0000857** Implementation of the sqlite_version() function. The result is the version
drh647cb0e2002-11-04 19:32:25 +0000858** of the SQLite library that is running.
859*/
drhf9b596e2004-05-26 16:54:42 +0000860static void versionFunc(
861 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000862 int NotUsed,
863 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000864){
danielk197762c14b32008-11-19 09:05:26 +0000865 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000866 /* IMP: R-48699-48617 This function is an SQL wrapper around the
867 ** sqlite3_libversion() C-interface. */
868 sqlite3_result_text(context, sqlite3_libversion(), -1, SQLITE_STATIC);
drh647cb0e2002-11-04 19:32:25 +0000869}
870
drh47baebc2009-08-14 16:01:24 +0000871/*
872** Implementation of the sqlite_source_id() function. The result is a string
873** that identifies the particular version of the source code used to build
874** SQLite.
875*/
876static void sourceidFunc(
877 sqlite3_context *context,
878 int NotUsed,
879 sqlite3_value **NotUsed2
880){
881 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000882 /* IMP: R-24470-31136 This function is an SQL wrapper around the
883 ** sqlite3_sourceid() C interface. */
884 sqlite3_result_text(context, sqlite3_sourceid(), -1, SQLITE_STATIC);
drh47baebc2009-08-14 16:01:24 +0000885}
886
shanehbdea6d12010-02-23 04:19:54 +0000887/*
drh3ca84ef2011-04-25 18:03:10 +0000888** Implementation of the sqlite_log() function. This is a wrapper around
889** sqlite3_log(). The return value is NULL. The function exists purely for
890** its side-effects.
891*/
drh840561f2011-04-27 18:08:42 +0000892static void errlogFunc(
drh3ca84ef2011-04-25 18:03:10 +0000893 sqlite3_context *context,
894 int argc,
895 sqlite3_value **argv
896){
897 UNUSED_PARAMETER(argc);
898 UNUSED_PARAMETER(context);
899 sqlite3_log(sqlite3_value_int(argv[0]), "%s", sqlite3_value_text(argv[1]));
900}
901
902/*
shanehdc97a8c2010-02-23 20:08:35 +0000903** Implementation of the sqlite_compileoption_used() function.
904** The result is an integer that identifies if the compiler option
905** was used to build SQLite.
shanehbdea6d12010-02-23 04:19:54 +0000906*/
shanehdc97a8c2010-02-23 20:08:35 +0000907#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
908static void compileoptionusedFunc(
shanehbdea6d12010-02-23 04:19:54 +0000909 sqlite3_context *context,
shanehdc97a8c2010-02-23 20:08:35 +0000910 int argc,
911 sqlite3_value **argv
shanehbdea6d12010-02-23 04:19:54 +0000912){
shanehdc97a8c2010-02-23 20:08:35 +0000913 const char *zOptName;
914 assert( argc==1 );
915 UNUSED_PARAMETER(argc);
drha3e414c2010-08-03 18:06:25 +0000916 /* IMP: R-39564-36305 The sqlite_compileoption_used() SQL
917 ** function is a wrapper around the sqlite3_compileoption_used() C/C++
918 ** function.
919 */
drh264a2d42010-02-25 15:28:41 +0000920 if( (zOptName = (const char*)sqlite3_value_text(argv[0]))!=0 ){
shanehdc97a8c2010-02-23 20:08:35 +0000921 sqlite3_result_int(context, sqlite3_compileoption_used(zOptName));
shanehdc97a8c2010-02-23 20:08:35 +0000922 }
shanehbdea6d12010-02-23 04:19:54 +0000923}
shanehdc97a8c2010-02-23 20:08:35 +0000924#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
925
926/*
927** Implementation of the sqlite_compileoption_get() function.
928** The result is a string that identifies the compiler options
929** used to build SQLite.
930*/
931#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
932static void compileoptiongetFunc(
933 sqlite3_context *context,
934 int argc,
935 sqlite3_value **argv
936){
937 int n;
938 assert( argc==1 );
939 UNUSED_PARAMETER(argc);
drha3e414c2010-08-03 18:06:25 +0000940 /* IMP: R-04922-24076 The sqlite_compileoption_get() SQL function
941 ** is a wrapper around the sqlite3_compileoption_get() C/C++ function.
942 */
shanehdc97a8c2010-02-23 20:08:35 +0000943 n = sqlite3_value_int(argv[0]);
944 sqlite3_result_text(context, sqlite3_compileoption_get(n), -1, SQLITE_STATIC);
945}
946#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
shanehbdea6d12010-02-23 04:19:54 +0000947
drh137c7282007-01-29 17:58:28 +0000948/* Array for converting from half-bytes (nybbles) into ASCII hex
949** digits. */
950static const char hexdigits[] = {
951 '0', '1', '2', '3', '4', '5', '6', '7',
952 '8', '9', 'A', 'B', 'C', 'D', 'E', 'F'
953};
danielk1977d641d642004-11-18 15:44:29 +0000954
drh47394702003-08-20 01:03:33 +0000955/*
drh47394702003-08-20 01:03:33 +0000956** Implementation of the QUOTE() function. This function takes a single
957** argument. If the argument is numeric, the return value is the same as
958** the argument. If the argument is NULL, the return value is the string
959** "NULL". Otherwise, the argument is enclosed in single quotes with
960** single-quote escapes.
961*/
danielk19770ae8b832004-05-25 12:05:56 +0000962static void quoteFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drha0df4cc2009-02-02 17:29:59 +0000963 assert( argc==1 );
drh1d34fde2009-02-03 15:50:33 +0000964 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +0000965 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +0000966 case SQLITE_FLOAT: {
drh72b3fbc2012-06-19 03:11:25 +0000967 double r1, r2;
drh72b3fbc2012-06-19 03:11:25 +0000968 char zBuf[50];
mistachkin2b434a72012-06-19 04:36:48 +0000969 r1 = sqlite3_value_double(argv[0]);
drh72b3fbc2012-06-19 03:11:25 +0000970 sqlite3_snprintf(sizeof(zBuf), zBuf, "%!.15g", r1);
971 sqlite3AtoF(zBuf, &r2, 20, SQLITE_UTF8);
972 if( r1!=r2 ){
973 sqlite3_snprintf(sizeof(zBuf), zBuf, "%!.20e", r1);
974 }
975 sqlite3_result_text(context, zBuf, -1, SQLITE_TRANSIENT);
976 break;
977 }
978 case SQLITE_INTEGER: {
drhf4479502004-05-27 03:12:53 +0000979 sqlite3_result_value(context, argv[0]);
drhf9b596e2004-05-26 16:54:42 +0000980 break;
981 }
danielk19773f41e972004-06-08 00:39:01 +0000982 case SQLITE_BLOB: {
danielk19773f41e972004-06-08 00:39:01 +0000983 char *zText = 0;
danielk19773f41e972004-06-08 00:39:01 +0000984 char const *zBlob = sqlite3_value_blob(argv[0]);
drh1f0feef2007-05-15 13:27:07 +0000985 int nBlob = sqlite3_value_bytes(argv[0]);
986 assert( zBlob==sqlite3_value_blob(argv[0]) ); /* No encoding change */
drhb1a6c3c2008-03-20 16:30:17 +0000987 zText = (char *)contextMalloc(context, (2*(i64)nBlob)+4);
danielk1977a1644fd2007-08-29 12:31:25 +0000988 if( zText ){
danielk19773f41e972004-06-08 00:39:01 +0000989 int i;
990 for(i=0; i<nBlob; i++){
991 zText[(i*2)+2] = hexdigits[(zBlob[i]>>4)&0x0F];
992 zText[(i*2)+3] = hexdigits[(zBlob[i])&0x0F];
993 }
994 zText[(nBlob*2)+2] = '\'';
995 zText[(nBlob*2)+3] = '\0';
996 zText[0] = 'X';
997 zText[1] = '\'';
danielk1977d8123362004-06-12 09:25:12 +0000998 sqlite3_result_text(context, zText, -1, SQLITE_TRANSIENT);
drh17435752007-08-16 04:30:38 +0000999 sqlite3_free(zText);
danielk19773f41e972004-06-08 00:39:01 +00001000 }
1001 break;
1002 }
drh9c054832004-05-31 18:51:57 +00001003 case SQLITE_TEXT: {
drh023ae032007-05-08 12:12:16 +00001004 int i,j;
1005 u64 n;
drh2646da72005-12-09 20:02:05 +00001006 const unsigned char *zArg = sqlite3_value_text(argv[0]);
drhf9b596e2004-05-26 16:54:42 +00001007 char *z;
1008
drh7a521cf2007-04-25 18:23:52 +00001009 if( zArg==0 ) return;
drh023ae032007-05-08 12:12:16 +00001010 for(i=0, n=0; zArg[i]; i++){ if( zArg[i]=='\'' ) n++; }
drhb1a6c3c2008-03-20 16:30:17 +00001011 z = contextMalloc(context, ((i64)i)+((i64)n)+3);
danielk1977a1644fd2007-08-29 12:31:25 +00001012 if( z ){
1013 z[0] = '\'';
1014 for(i=0, j=1; zArg[i]; i++){
1015 z[j++] = zArg[i];
1016 if( zArg[i]=='\'' ){
1017 z[j++] = '\'';
1018 }
drhf9b596e2004-05-26 16:54:42 +00001019 }
danielk1977a1644fd2007-08-29 12:31:25 +00001020 z[j++] = '\'';
1021 z[j] = 0;
1022 sqlite3_result_text(context, z, j, sqlite3_free);
drhf9b596e2004-05-26 16:54:42 +00001023 }
drha0df4cc2009-02-02 17:29:59 +00001024 break;
1025 }
1026 default: {
1027 assert( sqlite3_value_type(argv[0])==SQLITE_NULL );
1028 sqlite3_result_text(context, "NULL", 4, SQLITE_STATIC);
1029 break;
drhf9b596e2004-05-26 16:54:42 +00001030 }
drh47394702003-08-20 01:03:33 +00001031 }
1032}
1033
drh137c7282007-01-29 17:58:28 +00001034/*
drhd495d8c2013-02-22 19:34:25 +00001035** The unicode() function. Return the integer unicode code-point value
1036** for the first character of the input string.
1037*/
1038static void unicodeFunc(
1039 sqlite3_context *context,
1040 int argc,
1041 sqlite3_value **argv
1042){
1043 const unsigned char *z = sqlite3_value_text(argv[0]);
drh1d59d032013-03-01 23:40:26 +00001044 (void)argc;
drhd495d8c2013-02-22 19:34:25 +00001045 if( z && z[0] ) sqlite3_result_int(context, sqlite3Utf8Read(&z));
1046}
1047
1048/*
1049** The char() function takes zero or more arguments, each of which is
1050** an integer. It constructs a string where each character of the string
1051** is the unicode character for the corresponding integer argument.
1052*/
1053static void charFunc(
1054 sqlite3_context *context,
1055 int argc,
1056 sqlite3_value **argv
1057){
1058 unsigned char *z, *zOut;
1059 int i;
drhf3cdcdc2015-04-29 16:50:28 +00001060 zOut = z = sqlite3_malloc64( argc*4+1 );
drhd495d8c2013-02-22 19:34:25 +00001061 if( z==0 ){
1062 sqlite3_result_error_nomem(context);
1063 return;
1064 }
1065 for(i=0; i<argc; i++){
mistachkinc9545442013-02-26 05:42:30 +00001066 sqlite3_int64 x;
drhd495d8c2013-02-22 19:34:25 +00001067 unsigned c;
1068 x = sqlite3_value_int64(argv[i]);
1069 if( x<0 || x>0x10ffff ) x = 0xfffd;
1070 c = (unsigned)(x & 0x1fffff);
drhfe7a5d12013-03-07 14:00:04 +00001071 if( c<0x00080 ){
1072 *zOut++ = (u8)(c&0xFF);
1073 }else if( c<0x00800 ){
1074 *zOut++ = 0xC0 + (u8)((c>>6)&0x1F);
1075 *zOut++ = 0x80 + (u8)(c & 0x3F);
1076 }else if( c<0x10000 ){
1077 *zOut++ = 0xE0 + (u8)((c>>12)&0x0F);
1078 *zOut++ = 0x80 + (u8)((c>>6) & 0x3F);
1079 *zOut++ = 0x80 + (u8)(c & 0x3F);
drhd495d8c2013-02-22 19:34:25 +00001080 }else{
drhfe7a5d12013-03-07 14:00:04 +00001081 *zOut++ = 0xF0 + (u8)((c>>18) & 0x07);
1082 *zOut++ = 0x80 + (u8)((c>>12) & 0x3F);
1083 *zOut++ = 0x80 + (u8)((c>>6) & 0x3F);
1084 *zOut++ = 0x80 + (u8)(c & 0x3F);
1085 } \
drhd495d8c2013-02-22 19:34:25 +00001086 }
drhbbf483f2014-09-09 20:30:24 +00001087 sqlite3_result_text64(context, (char*)z, zOut-z, sqlite3_free, SQLITE_UTF8);
drhd495d8c2013-02-22 19:34:25 +00001088}
1089
1090/*
drh137c7282007-01-29 17:58:28 +00001091** The hex() function. Interpret the argument as a blob. Return
1092** a hexadecimal rendering as text.
1093*/
1094static void hexFunc(
1095 sqlite3_context *context,
1096 int argc,
1097 sqlite3_value **argv
1098){
1099 int i, n;
1100 const unsigned char *pBlob;
1101 char *zHex, *z;
1102 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001103 UNUSED_PARAMETER(argc);
drh1f0feef2007-05-15 13:27:07 +00001104 pBlob = sqlite3_value_blob(argv[0]);
drh137c7282007-01-29 17:58:28 +00001105 n = sqlite3_value_bytes(argv[0]);
drh1f0feef2007-05-15 13:27:07 +00001106 assert( pBlob==sqlite3_value_blob(argv[0]) ); /* No encoding change */
drhb1a6c3c2008-03-20 16:30:17 +00001107 z = zHex = contextMalloc(context, ((i64)n)*2 + 1);
danielk1977a1644fd2007-08-29 12:31:25 +00001108 if( zHex ){
1109 for(i=0; i<n; i++, pBlob++){
1110 unsigned char c = *pBlob;
1111 *(z++) = hexdigits[(c>>4)&0xf];
1112 *(z++) = hexdigits[c&0xf];
1113 }
1114 *z = 0;
1115 sqlite3_result_text(context, zHex, n*2, sqlite3_free);
drh137c7282007-01-29 17:58:28 +00001116 }
drh137c7282007-01-29 17:58:28 +00001117}
1118
drh26b6d902007-03-17 13:27:54 +00001119/*
drh8cff3822007-05-02 02:08:28 +00001120** The zeroblob(N) function returns a zero-filled blob of size N bytes.
1121*/
1122static void zeroblobFunc(
1123 sqlite3_context *context,
1124 int argc,
1125 sqlite3_value **argv
1126){
drh98640a32007-06-07 19:08:32 +00001127 i64 n;
drh27e62db2009-04-02 10:16:17 +00001128 sqlite3 *db = sqlite3_context_db_handle(context);
drh8cff3822007-05-02 02:08:28 +00001129 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001130 UNUSED_PARAMETER(argc);
drh98640a32007-06-07 19:08:32 +00001131 n = sqlite3_value_int64(argv[0]);
drh27e62db2009-04-02 10:16:17 +00001132 testcase( n==db->aLimit[SQLITE_LIMIT_LENGTH] );
1133 testcase( n==db->aLimit[SQLITE_LIMIT_LENGTH]+1 );
1134 if( n>db->aLimit[SQLITE_LIMIT_LENGTH] ){
drh98640a32007-06-07 19:08:32 +00001135 sqlite3_result_error_toobig(context);
1136 }else{
drhab2f1f92010-01-11 18:26:42 +00001137 sqlite3_result_zeroblob(context, (int)n); /* IMP: R-00293-64994 */
drh98640a32007-06-07 19:08:32 +00001138 }
drh8cff3822007-05-02 02:08:28 +00001139}
1140
1141/*
drh26b6d902007-03-17 13:27:54 +00001142** The replace() function. Three arguments are all strings: call
1143** them A, B, and C. The result is also a string which is derived
drhf7b54962013-05-28 12:11:54 +00001144** from A by replacing every occurrence of B with C. The match
drh26b6d902007-03-17 13:27:54 +00001145** must be exact. Collating sequences are not used.
1146*/
1147static void replaceFunc(
1148 sqlite3_context *context,
1149 int argc,
1150 sqlite3_value **argv
1151){
1152 const unsigned char *zStr; /* The input string A */
1153 const unsigned char *zPattern; /* The pattern string B */
1154 const unsigned char *zRep; /* The replacement string C */
1155 unsigned char *zOut; /* The output */
1156 int nStr; /* Size of zStr */
1157 int nPattern; /* Size of zPattern */
1158 int nRep; /* Size of zRep */
drh2e6400b2007-05-08 15:46:18 +00001159 i64 nOut; /* Maximum size of zOut */
drh26b6d902007-03-17 13:27:54 +00001160 int loopLimit; /* Last zStr[] that might match zPattern[] */
1161 int i, j; /* Loop counters */
1162
1163 assert( argc==3 );
danielk1977f3d3c272008-11-19 16:52:44 +00001164 UNUSED_PARAMETER(argc);
drh26b6d902007-03-17 13:27:54 +00001165 zStr = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001166 if( zStr==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001167 nStr = sqlite3_value_bytes(argv[0]);
1168 assert( zStr==sqlite3_value_text(argv[0]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001169 zPattern = sqlite3_value_text(argv[1]);
drha605fe82009-02-01 18:08:40 +00001170 if( zPattern==0 ){
drh23336062009-02-03 13:19:12 +00001171 assert( sqlite3_value_type(argv[1])==SQLITE_NULL
1172 || sqlite3_context_db_handle(context)->mallocFailed );
drha605fe82009-02-01 18:08:40 +00001173 return;
1174 }
1175 if( zPattern[0]==0 ){
1176 assert( sqlite3_value_type(argv[1])!=SQLITE_NULL );
1177 sqlite3_result_value(context, argv[0]);
1178 return;
1179 }
drh1f0feef2007-05-15 13:27:07 +00001180 nPattern = sqlite3_value_bytes(argv[1]);
1181 assert( zPattern==sqlite3_value_text(argv[1]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001182 zRep = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +00001183 if( zRep==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001184 nRep = sqlite3_value_bytes(argv[2]);
1185 assert( zRep==sqlite3_value_text(argv[2]) );
drh2e6400b2007-05-08 15:46:18 +00001186 nOut = nStr + 1;
1187 assert( nOut<SQLITE_MAX_LENGTH );
drhb1a6c3c2008-03-20 16:30:17 +00001188 zOut = contextMalloc(context, (i64)nOut);
drh2e6400b2007-05-08 15:46:18 +00001189 if( zOut==0 ){
1190 return;
drh26b6d902007-03-17 13:27:54 +00001191 }
drh26b6d902007-03-17 13:27:54 +00001192 loopLimit = nStr - nPattern;
1193 for(i=j=0; i<=loopLimit; i++){
1194 if( zStr[i]!=zPattern[0] || memcmp(&zStr[i], zPattern, nPattern) ){
1195 zOut[j++] = zStr[i];
1196 }else{
drh4a50aac2007-08-23 02:47:53 +00001197 u8 *zOld;
drhbb4957f2008-03-20 14:03:29 +00001198 sqlite3 *db = sqlite3_context_db_handle(context);
drh2e6400b2007-05-08 15:46:18 +00001199 nOut += nRep - nPattern;
drh27e62db2009-04-02 10:16:17 +00001200 testcase( nOut-1==db->aLimit[SQLITE_LIMIT_LENGTH] );
1201 testcase( nOut-2==db->aLimit[SQLITE_LIMIT_LENGTH] );
1202 if( nOut-1>db->aLimit[SQLITE_LIMIT_LENGTH] ){
drha0206bc2007-05-08 15:15:02 +00001203 sqlite3_result_error_toobig(context);
drhb9755982010-07-24 16:34:37 +00001204 sqlite3_free(zOut);
danielk197717374e82007-05-08 14:39:04 +00001205 return;
1206 }
drh4a50aac2007-08-23 02:47:53 +00001207 zOld = zOut;
drhf3cdcdc2015-04-29 16:50:28 +00001208 zOut = sqlite3_realloc64(zOut, (int)nOut);
drh2e6400b2007-05-08 15:46:18 +00001209 if( zOut==0 ){
danielk1977a1644fd2007-08-29 12:31:25 +00001210 sqlite3_result_error_nomem(context);
drhb9755982010-07-24 16:34:37 +00001211 sqlite3_free(zOld);
drh2e6400b2007-05-08 15:46:18 +00001212 return;
1213 }
drh26b6d902007-03-17 13:27:54 +00001214 memcpy(&zOut[j], zRep, nRep);
1215 j += nRep;
1216 i += nPattern-1;
1217 }
1218 }
drh2e6400b2007-05-08 15:46:18 +00001219 assert( j+nStr-i+1==nOut );
drh26b6d902007-03-17 13:27:54 +00001220 memcpy(&zOut[j], &zStr[i], nStr-i);
1221 j += nStr - i;
1222 assert( j<=nOut );
1223 zOut[j] = 0;
1224 sqlite3_result_text(context, (char*)zOut, j, sqlite3_free);
1225}
1226
drh309b3382007-03-17 17:52:42 +00001227/*
1228** Implementation of the TRIM(), LTRIM(), and RTRIM() functions.
1229** The userdata is 0x1 for left trim, 0x2 for right trim, 0x3 for both.
1230*/
1231static void trimFunc(
1232 sqlite3_context *context,
1233 int argc,
1234 sqlite3_value **argv
1235){
1236 const unsigned char *zIn; /* Input string */
1237 const unsigned char *zCharSet; /* Set of characters to trim */
1238 int nIn; /* Number of bytes in input */
drh7209c692008-04-27 18:40:11 +00001239 int flags; /* 1: trimleft 2: trimright 3: trim */
drhd1e3a612007-04-27 21:59:52 +00001240 int i; /* Loop counter */
drh1bd10f82008-12-10 21:19:56 +00001241 unsigned char *aLen = 0; /* Length of each character in zCharSet */
1242 unsigned char **azChar = 0; /* Individual characters in zCharSet */
drhd1e3a612007-04-27 21:59:52 +00001243 int nChar; /* Number of characters in zCharSet */
1244
drh309b3382007-03-17 17:52:42 +00001245 if( sqlite3_value_type(argv[0])==SQLITE_NULL ){
1246 return;
1247 }
1248 zIn = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001249 if( zIn==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001250 nIn = sqlite3_value_bytes(argv[0]);
1251 assert( zIn==sqlite3_value_text(argv[0]) );
drh309b3382007-03-17 17:52:42 +00001252 if( argc==1 ){
drhd1e3a612007-04-27 21:59:52 +00001253 static const unsigned char lenOne[] = { 1 };
danielk1977a4de4532008-09-02 15:44:08 +00001254 static unsigned char * const azOne[] = { (u8*)" " };
drhd1e3a612007-04-27 21:59:52 +00001255 nChar = 1;
drh8cff3822007-05-02 02:08:28 +00001256 aLen = (u8*)lenOne;
danielk1977bc67da42007-12-11 04:23:19 +00001257 azChar = (unsigned char **)azOne;
drhd1e3a612007-04-27 21:59:52 +00001258 zCharSet = 0;
drh7a521cf2007-04-25 18:23:52 +00001259 }else if( (zCharSet = sqlite3_value_text(argv[1]))==0 ){
drh309b3382007-03-17 17:52:42 +00001260 return;
drhd1e3a612007-04-27 21:59:52 +00001261 }else{
1262 const unsigned char *z;
1263 for(z=zCharSet, nChar=0; *z; nChar++){
drh4a919112007-05-15 11:55:09 +00001264 SQLITE_SKIP_UTF8(z);
drhd1e3a612007-04-27 21:59:52 +00001265 }
1266 if( nChar>0 ){
drhb1a6c3c2008-03-20 16:30:17 +00001267 azChar = contextMalloc(context, ((i64)nChar)*(sizeof(char*)+1));
drhd1e3a612007-04-27 21:59:52 +00001268 if( azChar==0 ){
1269 return;
1270 }
1271 aLen = (unsigned char*)&azChar[nChar];
1272 for(z=zCharSet, nChar=0; *z; nChar++){
danielk1977bc67da42007-12-11 04:23:19 +00001273 azChar[nChar] = (unsigned char *)z;
drh4a919112007-05-15 11:55:09 +00001274 SQLITE_SKIP_UTF8(z);
drh3abbd392008-12-10 23:04:13 +00001275 aLen[nChar] = (u8)(z - azChar[nChar]);
drhd1e3a612007-04-27 21:59:52 +00001276 }
1277 }
drh309b3382007-03-17 17:52:42 +00001278 }
drhd1e3a612007-04-27 21:59:52 +00001279 if( nChar>0 ){
shane1fc41292008-07-08 22:28:48 +00001280 flags = SQLITE_PTR_TO_INT(sqlite3_user_data(context));
drh309b3382007-03-17 17:52:42 +00001281 if( flags & 1 ){
drhd1e3a612007-04-27 21:59:52 +00001282 while( nIn>0 ){
drh1bd10f82008-12-10 21:19:56 +00001283 int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001284 for(i=0; i<nChar; i++){
1285 len = aLen[i];
drh27e62db2009-04-02 10:16:17 +00001286 if( len<=nIn && memcmp(zIn, azChar[i], len)==0 ) break;
drhd1e3a612007-04-27 21:59:52 +00001287 }
1288 if( i>=nChar ) break;
1289 zIn += len;
1290 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001291 }
1292 }
1293 if( flags & 2 ){
drhd1e3a612007-04-27 21:59:52 +00001294 while( nIn>0 ){
drh1bd10f82008-12-10 21:19:56 +00001295 int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001296 for(i=0; i<nChar; i++){
1297 len = aLen[i];
1298 if( len<=nIn && memcmp(&zIn[nIn-len],azChar[i],len)==0 ) break;
1299 }
1300 if( i>=nChar ) break;
1301 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001302 }
1303 }
drhd1e3a612007-04-27 21:59:52 +00001304 if( zCharSet ){
1305 sqlite3_free(azChar);
1306 }
drh309b3382007-03-17 17:52:42 +00001307 }
1308 sqlite3_result_text(context, (char*)zIn, nIn, SQLITE_TRANSIENT);
1309}
drh26b6d902007-03-17 13:27:54 +00001310
danielk1977a4de4532008-09-02 15:44:08 +00001311
drh2ba3ccc2009-12-08 02:06:08 +00001312/* IMP: R-25361-16150 This function is omitted from SQLite by default. It
1313** is only available if the SQLITE_SOUNDEX compile-time option is used
1314** when SQLite is built.
1315*/
drhd24cc422003-03-27 12:51:24 +00001316#ifdef SQLITE_SOUNDEX
1317/*
1318** Compute the soundex encoding of a word.
drh2ba3ccc2009-12-08 02:06:08 +00001319**
1320** IMP: R-59782-00072 The soundex(X) function returns a string that is the
1321** soundex encoding of the string X.
drhd24cc422003-03-27 12:51:24 +00001322*/
drh137c7282007-01-29 17:58:28 +00001323static void soundexFunc(
1324 sqlite3_context *context,
1325 int argc,
1326 sqlite3_value **argv
1327){
drhd24cc422003-03-27 12:51:24 +00001328 char zResult[8];
drh4c755c02004-08-08 20:22:17 +00001329 const u8 *zIn;
drhd24cc422003-03-27 12:51:24 +00001330 int i, j;
1331 static const unsigned char iCode[] = {
1332 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1333 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1334 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1335 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1336 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1337 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1338 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1339 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1340 };
1341 assert( argc==1 );
drh4c755c02004-08-08 20:22:17 +00001342 zIn = (u8*)sqlite3_value_text(argv[0]);
drhbdf67e02006-08-19 11:34:01 +00001343 if( zIn==0 ) zIn = (u8*)"";
drhdc86e2b2009-01-24 11:30:42 +00001344 for(i=0; zIn[i] && !sqlite3Isalpha(zIn[i]); i++){}
drhd24cc422003-03-27 12:51:24 +00001345 if( zIn[i] ){
drhbdf67e02006-08-19 11:34:01 +00001346 u8 prevcode = iCode[zIn[i]&0x7f];
danielk197778ca0e72009-01-20 16:53:39 +00001347 zResult[0] = sqlite3Toupper(zIn[i]);
drhd24cc422003-03-27 12:51:24 +00001348 for(j=1; j<4 && zIn[i]; i++){
1349 int code = iCode[zIn[i]&0x7f];
1350 if( code>0 ){
drhbdf67e02006-08-19 11:34:01 +00001351 if( code!=prevcode ){
1352 prevcode = code;
1353 zResult[j++] = code + '0';
1354 }
1355 }else{
1356 prevcode = 0;
drhd24cc422003-03-27 12:51:24 +00001357 }
1358 }
1359 while( j<4 ){
1360 zResult[j++] = '0';
1361 }
1362 zResult[j] = 0;
danielk1977d8123362004-06-12 09:25:12 +00001363 sqlite3_result_text(context, zResult, 4, SQLITE_TRANSIENT);
drhd24cc422003-03-27 12:51:24 +00001364 }else{
drh2ba3ccc2009-12-08 02:06:08 +00001365 /* IMP: R-64894-50321 The string "?000" is returned if the argument
1366 ** is NULL or contains no ASCII alphabetic characters. */
danielk1977d8123362004-06-12 09:25:12 +00001367 sqlite3_result_text(context, "?000", 4, SQLITE_STATIC);
drhd24cc422003-03-27 12:51:24 +00001368 }
1369}
drh2ba3ccc2009-12-08 02:06:08 +00001370#endif /* SQLITE_SOUNDEX */
drhd24cc422003-03-27 12:51:24 +00001371
drhfdb83b22006-06-17 14:12:47 +00001372#ifndef SQLITE_OMIT_LOAD_EXTENSION
1373/*
1374** A function that loads a shared-library extension then returns NULL.
1375*/
1376static void loadExt(sqlite3_context *context, int argc, sqlite3_value **argv){
danielk197765fd59f2006-06-24 11:51:33 +00001377 const char *zFile = (const char *)sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001378 const char *zProc;
drhfa4a4b92008-03-19 21:45:51 +00001379 sqlite3 *db = sqlite3_context_db_handle(context);
drhfdb83b22006-06-17 14:12:47 +00001380 char *zErrMsg = 0;
1381
1382 if( argc==2 ){
danielk197765fd59f2006-06-24 11:51:33 +00001383 zProc = (const char *)sqlite3_value_text(argv[1]);
drh7a521cf2007-04-25 18:23:52 +00001384 }else{
1385 zProc = 0;
drhfdb83b22006-06-17 14:12:47 +00001386 }
drh7a521cf2007-04-25 18:23:52 +00001387 if( zFile && sqlite3_load_extension(db, zFile, zProc, &zErrMsg) ){
drhfdb83b22006-06-17 14:12:47 +00001388 sqlite3_result_error(context, zErrMsg, -1);
1389 sqlite3_free(zErrMsg);
1390 }
1391}
1392#endif
1393
danielk197701427a62005-01-11 13:02:33 +00001394
drh0ac65892002-04-20 14:24:41 +00001395/*
drhd3a149e2002-02-24 17:12:53 +00001396** An instance of the following structure holds the context of a
drhdd5baa92002-02-27 19:50:59 +00001397** sum() or avg() aggregate computation.
1398*/
1399typedef struct SumCtx SumCtx;
1400struct SumCtx {
drh8c08e862006-02-11 17:34:00 +00001401 double rSum; /* Floating point sum */
1402 i64 iSum; /* Integer sum */
1403 i64 cnt; /* Number of elements summed */
1404 u8 overflow; /* True if integer overflow seen */
1405 u8 approx; /* True if non-integer value was input to the sum */
drhdd5baa92002-02-27 19:50:59 +00001406};
1407
1408/*
drha97fdd32006-01-12 22:17:50 +00001409** Routines used to compute the sum, average, and total.
1410**
1411** The SUM() function follows the (broken) SQL standard which means
1412** that it returns NULL if it sums over no inputs. TOTAL returns
1413** 0.0 in that case. In addition, TOTAL always returns a float where
1414** SUM might return an integer if it never encounters a floating point
drhc806d852006-05-11 13:25:39 +00001415** value. TOTAL never fails, but SUM might through an exception if
1416** it overflows an integer.
drhdd5baa92002-02-27 19:50:59 +00001417*/
danielk19770ae8b832004-05-25 12:05:56 +00001418static void sumStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drhdd5baa92002-02-27 19:50:59 +00001419 SumCtx *p;
drh3d1d95e2005-09-08 10:37:01 +00001420 int type;
drh3f219f42005-09-08 19:45:57 +00001421 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001422 UNUSED_PARAMETER(argc);
drh4f26d6c2004-05-26 23:25:30 +00001423 p = sqlite3_aggregate_context(context, sizeof(*p));
drh29d72102006-02-09 22:13:41 +00001424 type = sqlite3_value_numeric_type(argv[0]);
drh3d1d95e2005-09-08 10:37:01 +00001425 if( p && type!=SQLITE_NULL ){
drh739105c2002-05-29 23:22:23 +00001426 p->cnt++;
drh29d72102006-02-09 22:13:41 +00001427 if( type==SQLITE_INTEGER ){
drh8c08e862006-02-11 17:34:00 +00001428 i64 v = sqlite3_value_int64(argv[0]);
1429 p->rSum += v;
drh158b9cb2011-03-05 20:59:46 +00001430 if( (p->approx|p->overflow)==0 && sqlite3AddInt64(&p->iSum, v) ){
1431 p->overflow = 1;
drh29d72102006-02-09 22:13:41 +00001432 }
1433 }else{
drh8c08e862006-02-11 17:34:00 +00001434 p->rSum += sqlite3_value_double(argv[0]);
drh29d72102006-02-09 22:13:41 +00001435 p->approx = 1;
drh3f219f42005-09-08 19:45:57 +00001436 }
drh739105c2002-05-29 23:22:23 +00001437 }
drhdd5baa92002-02-27 19:50:59 +00001438}
danielk19770ae8b832004-05-25 12:05:56 +00001439static void sumFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001440 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001441 p = sqlite3_aggregate_context(context, 0);
drhc2bd9132005-09-08 20:37:43 +00001442 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001443 if( p->overflow ){
1444 sqlite3_result_error(context,"integer overflow",-1);
1445 }else if( p->approx ){
1446 sqlite3_result_double(context, p->rSum);
drhc2bd9132005-09-08 20:37:43 +00001447 }else{
drh8c08e862006-02-11 17:34:00 +00001448 sqlite3_result_int64(context, p->iSum);
drhc2bd9132005-09-08 20:37:43 +00001449 }
drh3d1d95e2005-09-08 10:37:01 +00001450 }
drhdd5baa92002-02-27 19:50:59 +00001451}
danielk19770ae8b832004-05-25 12:05:56 +00001452static void avgFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001453 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001454 p = sqlite3_aggregate_context(context, 0);
drh739105c2002-05-29 23:22:23 +00001455 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001456 sqlite3_result_double(context, p->rSum/(double)p->cnt);
drhdd5baa92002-02-27 19:50:59 +00001457 }
1458}
drha97fdd32006-01-12 22:17:50 +00001459static void totalFinalize(sqlite3_context *context){
1460 SumCtx *p;
1461 p = sqlite3_aggregate_context(context, 0);
shanefbd60f82009-02-04 03:59:25 +00001462 /* (double)0 In case of SQLITE_OMIT_FLOATING_POINT... */
1463 sqlite3_result_double(context, p ? p->rSum : (double)0);
drha97fdd32006-01-12 22:17:50 +00001464}
drhdd5baa92002-02-27 19:50:59 +00001465
1466/*
drh0bce8352002-02-28 00:41:10 +00001467** The following structure keeps track of state information for the
1468** count() aggregate function.
1469*/
1470typedef struct CountCtx CountCtx;
1471struct CountCtx {
drhfc6ad392006-02-09 13:38:19 +00001472 i64 n;
drh0bce8352002-02-28 00:41:10 +00001473};
drhdd5baa92002-02-27 19:50:59 +00001474
drh0bce8352002-02-28 00:41:10 +00001475/*
1476** Routines to implement the count() aggregate function.
1477*/
danielk19770ae8b832004-05-25 12:05:56 +00001478static void countStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drh0bce8352002-02-28 00:41:10 +00001479 CountCtx *p;
drh4f26d6c2004-05-26 23:25:30 +00001480 p = sqlite3_aggregate_context(context, sizeof(*p));
drh9c054832004-05-31 18:51:57 +00001481 if( (argc==0 || SQLITE_NULL!=sqlite3_value_type(argv[0])) && p ){
drh0bce8352002-02-28 00:41:10 +00001482 p->n++;
1483 }
drh2e79c3d2009-04-08 23:04:14 +00001484
drhd3264c72009-04-15 13:39:47 +00001485#ifndef SQLITE_OMIT_DEPRECATED
drh2e79c3d2009-04-08 23:04:14 +00001486 /* The sqlite3_aggregate_count() function is deprecated. But just to make
1487 ** sure it still operates correctly, verify that its count agrees with our
1488 ** internal count when using count(*) and when the total count can be
1489 ** expressed as a 32-bit integer. */
1490 assert( argc==1 || p==0 || p->n>0x7fffffff
1491 || p->n==sqlite3_aggregate_count(context) );
drhd3264c72009-04-15 13:39:47 +00001492#endif
drh0bce8352002-02-28 00:41:10 +00001493}
danielk19770ae8b832004-05-25 12:05:56 +00001494static void countFinalize(sqlite3_context *context){
drh0bce8352002-02-28 00:41:10 +00001495 CountCtx *p;
drhabfcea22005-09-06 20:36:48 +00001496 p = sqlite3_aggregate_context(context, 0);
drhfc6ad392006-02-09 13:38:19 +00001497 sqlite3_result_int64(context, p ? p->n : 0);
drh0bce8352002-02-28 00:41:10 +00001498}
1499
1500/*
drh0bce8352002-02-28 00:41:10 +00001501** Routines to implement min() and max() aggregate functions.
1502*/
danielk197762c14b32008-11-19 09:05:26 +00001503static void minmaxStep(
1504 sqlite3_context *context,
1505 int NotUsed,
1506 sqlite3_value **argv
1507){
danielk197788208052004-05-25 01:13:20 +00001508 Mem *pArg = (Mem *)argv[0];
drh9eb516c2004-07-18 20:52:32 +00001509 Mem *pBest;
danielk197762c14b32008-11-19 09:05:26 +00001510 UNUSED_PARAMETER(NotUsed);
drh9eb516c2004-07-18 20:52:32 +00001511
drh9eb516c2004-07-18 20:52:32 +00001512 pBest = (Mem *)sqlite3_aggregate_context(context, sizeof(*pBest));
danielk19773aeab9e2004-06-24 00:20:04 +00001513 if( !pBest ) return;
drh268380c2004-02-25 13:47:31 +00001514
drh94a6d992012-02-02 18:42:09 +00001515 if( sqlite3_value_type(argv[0])==SQLITE_NULL ){
1516 if( pBest->flags ) sqlite3SkipAccumulatorLoad(context);
1517 }else if( pBest->flags ){
drh9eb516c2004-07-18 20:52:32 +00001518 int max;
1519 int cmp;
danielk1977dc1bdc42004-06-11 10:51:27 +00001520 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk19777e18c252004-05-25 11:47:24 +00001521 /* This step function is used for both the min() and max() aggregates,
1522 ** the only difference between the two being that the sense of the
1523 ** comparison is inverted. For the max() aggregate, the
1524 ** sqlite3_user_data() function returns (void *)-1. For min() it
1525 ** returns (void *)db, where db is the sqlite3* database pointer.
1526 ** Therefore the next statement sets variable 'max' to 1 for the max()
1527 ** aggregate, or 0 for min().
1528 */
drh309b3382007-03-17 17:52:42 +00001529 max = sqlite3_user_data(context)!=0;
danielk1977dc1bdc42004-06-11 10:51:27 +00001530 cmp = sqlite3MemCompare(pBest, pArg, pColl);
danielk197788208052004-05-25 01:13:20 +00001531 if( (max && cmp<0) || (!max && cmp>0) ){
drhb21c8cd2007-08-21 19:33:56 +00001532 sqlite3VdbeMemCopy(pBest, pArg);
drh7a957892012-02-02 17:35:43 +00001533 }else{
1534 sqlite3SkipAccumulatorLoad(context);
danielk197788208052004-05-25 01:13:20 +00001535 }
drh268380c2004-02-25 13:47:31 +00001536 }else{
drh035e5632014-09-16 14:16:31 +00001537 pBest->db = sqlite3_context_db_handle(context);
drhb21c8cd2007-08-21 19:33:56 +00001538 sqlite3VdbeMemCopy(pBest, pArg);
drh0bce8352002-02-28 00:41:10 +00001539 }
1540}
danielk19770ae8b832004-05-25 12:05:56 +00001541static void minMaxFinalize(sqlite3_context *context){
danielk197788208052004-05-25 01:13:20 +00001542 sqlite3_value *pRes;
drhabfcea22005-09-06 20:36:48 +00001543 pRes = (sqlite3_value *)sqlite3_aggregate_context(context, 0);
1544 if( pRes ){
drh94a6d992012-02-02 18:42:09 +00001545 if( pRes->flags ){
drhabfcea22005-09-06 20:36:48 +00001546 sqlite3_result_value(context, pRes);
1547 }
1548 sqlite3VdbeMemRelease(pRes);
drh0bce8352002-02-28 00:41:10 +00001549 }
1550}
drhdd5baa92002-02-27 19:50:59 +00001551
drhb0689692007-11-01 17:38:30 +00001552/*
1553** group_concat(EXPR, ?SEPARATOR?)
1554*/
1555static void groupConcatStep(
1556 sqlite3_context *context,
1557 int argc,
1558 sqlite3_value **argv
1559){
1560 const char *zVal;
drhade86482007-11-28 22:36:40 +00001561 StrAccum *pAccum;
drhb0689692007-11-01 17:38:30 +00001562 const char *zSep;
drh07d31172009-02-02 21:57:05 +00001563 int nVal, nSep;
1564 assert( argc==1 || argc==2 );
1565 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drhade86482007-11-28 22:36:40 +00001566 pAccum = (StrAccum*)sqlite3_aggregate_context(context, sizeof(*pAccum));
1567
1568 if( pAccum ){
drhbb4957f2008-03-20 14:03:29 +00001569 sqlite3 *db = sqlite3_context_db_handle(context);
drhc0490572015-05-02 11:45:53 +00001570 int firstTerm = pAccum->mxAlloc==0;
drhbb4957f2008-03-20 14:03:29 +00001571 pAccum->mxAlloc = db->aLimit[SQLITE_LIMIT_LENGTH];
drh8bfd7192009-06-19 16:44:41 +00001572 if( !firstTerm ){
drh07d31172009-02-02 21:57:05 +00001573 if( argc==2 ){
1574 zSep = (char*)sqlite3_value_text(argv[1]);
1575 nSep = sqlite3_value_bytes(argv[1]);
drhb0689692007-11-01 17:38:30 +00001576 }else{
1577 zSep = ",";
drhade86482007-11-28 22:36:40 +00001578 nSep = 1;
drhb0689692007-11-01 17:38:30 +00001579 }
drha9ab4812013-12-11 11:00:44 +00001580 if( nSep ) sqlite3StrAccumAppend(pAccum, zSep, nSep);
drhb0689692007-11-01 17:38:30 +00001581 }
drh07d31172009-02-02 21:57:05 +00001582 zVal = (char*)sqlite3_value_text(argv[0]);
1583 nVal = sqlite3_value_bytes(argv[0]);
drh8009c9b2014-05-07 20:24:00 +00001584 if( zVal ) sqlite3StrAccumAppend(pAccum, zVal, nVal);
drhb0689692007-11-01 17:38:30 +00001585 }
1586}
1587static void groupConcatFinalize(sqlite3_context *context){
drhade86482007-11-28 22:36:40 +00001588 StrAccum *pAccum;
1589 pAccum = sqlite3_aggregate_context(context, 0);
1590 if( pAccum ){
drhb49bc862013-08-21 21:12:10 +00001591 if( pAccum->accError==STRACCUM_TOOBIG ){
drhade86482007-11-28 22:36:40 +00001592 sqlite3_result_error_toobig(context);
drhb49bc862013-08-21 21:12:10 +00001593 }else if( pAccum->accError==STRACCUM_NOMEM ){
drhade86482007-11-28 22:36:40 +00001594 sqlite3_result_error_nomem(context);
1595 }else{
1596 sqlite3_result_text(context, sqlite3StrAccumFinish(pAccum), -1,
1597 sqlite3_free);
1598 }
drhb0689692007-11-01 17:38:30 +00001599 }
1600}
drh4e5ffc52004-08-31 00:52:37 +00001601
drhd3a149e2002-02-24 17:12:53 +00001602/*
drha4741842010-04-25 20:58:37 +00001603** This routine does per-connection function registration. Most
1604** of the built-in functions above are part of the global function set.
1605** This routine only deals with those that are not global.
drhdc04c582002-02-24 01:55:15 +00001606*/
drh9bb575f2004-09-06 17:24:11 +00001607void sqlite3RegisterBuiltinFunctions(sqlite3 *db){
drha4741842010-04-25 20:58:37 +00001608 int rc = sqlite3_overload_function(db, "MATCH", 2);
1609 assert( rc==SQLITE_NOMEM || rc==SQLITE_OK );
1610 if( rc==SQLITE_NOMEM ){
1611 db->mallocFailed = 1;
danielk1977832a58a2007-06-22 15:21:15 +00001612 }
drh55ef4d92005-08-14 01:20:37 +00001613}
1614
1615/*
1616** Set the LIKEOPT flag on the 2-argument function with the given name.
1617*/
drh1bd10f82008-12-10 21:19:56 +00001618static void setLikeOptFlag(sqlite3 *db, const char *zName, u8 flagVal){
drh55ef4d92005-08-14 01:20:37 +00001619 FuncDef *pDef;
drhea678832008-12-10 19:26:22 +00001620 pDef = sqlite3FindFunction(db, zName, sqlite3Strlen30(zName),
1621 2, SQLITE_UTF8, 0);
drhd27135a2009-04-02 13:36:37 +00001622 if( ALWAYS(pDef) ){
drhd36e1042013-09-06 13:10:12 +00001623 pDef->funcFlags |= flagVal;
drh55ef4d92005-08-14 01:20:37 +00001624 }
1625}
1626
1627/*
1628** Register the built-in LIKE and GLOB functions. The caseSensitive
1629** parameter determines whether or not the LIKE operator is case
1630** sensitive. GLOB is always case sensitive.
1631*/
1632void sqlite3RegisterLikeFunctions(sqlite3 *db, int caseSensitive){
1633 struct compareInfo *pInfo;
1634 if( caseSensitive ){
1635 pInfo = (struct compareInfo*)&likeInfoAlt;
1636 }else{
1637 pInfo = (struct compareInfo*)&likeInfoNorm;
1638 }
drh901e9942010-12-15 18:54:37 +00001639 sqlite3CreateFunc(db, "like", 2, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0);
1640 sqlite3CreateFunc(db, "like", 3, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0);
1641 sqlite3CreateFunc(db, "glob", 2, SQLITE_UTF8,
dand2199f02010-08-27 17:48:52 +00001642 (struct compareInfo*)&globInfo, likeFunc, 0, 0, 0);
drhd64fe2f2005-08-28 17:00:23 +00001643 setLikeOptFlag(db, "glob", SQLITE_FUNC_LIKE | SQLITE_FUNC_CASE);
1644 setLikeOptFlag(db, "like",
1645 caseSensitive ? (SQLITE_FUNC_LIKE | SQLITE_FUNC_CASE) : SQLITE_FUNC_LIKE);
drh55ef4d92005-08-14 01:20:37 +00001646}
1647
1648/*
1649** pExpr points to an expression which implements a function. If
1650** it is appropriate to apply the LIKE optimization to that function
1651** then set aWc[0] through aWc[2] to the wildcard characters and
1652** return TRUE. If the function is not a LIKE-style function then
1653** return FALSE.
drh16897072015-03-07 00:57:37 +00001654**
1655** *pIsNocase is set to true if uppercase and lowercase are equivalent for
1656** the function (default for LIKE). If the function makes the distinction
1657** between uppercase and lowercase (as does GLOB) then *pIsNocase is set to
1658** false.
drh55ef4d92005-08-14 01:20:37 +00001659*/
drhd64fe2f2005-08-28 17:00:23 +00001660int sqlite3IsLikeFunction(sqlite3 *db, Expr *pExpr, int *pIsNocase, char *aWc){
drh55ef4d92005-08-14 01:20:37 +00001661 FuncDef *pDef;
danielk19776ab3a2e2009-02-19 14:39:25 +00001662 if( pExpr->op!=TK_FUNCTION
1663 || !pExpr->x.pList
1664 || pExpr->x.pList->nExpr!=2
1665 ){
drh55ef4d92005-08-14 01:20:37 +00001666 return 0;
1667 }
danielk19776ab3a2e2009-02-19 14:39:25 +00001668 assert( !ExprHasProperty(pExpr, EP_xIsSelect) );
drh33e619f2009-05-28 01:00:55 +00001669 pDef = sqlite3FindFunction(db, pExpr->u.zToken,
1670 sqlite3Strlen30(pExpr->u.zToken),
drhb7916a72009-05-27 10:31:29 +00001671 2, SQLITE_UTF8, 0);
drhd36e1042013-09-06 13:10:12 +00001672 if( NEVER(pDef==0) || (pDef->funcFlags & SQLITE_FUNC_LIKE)==0 ){
drh55ef4d92005-08-14 01:20:37 +00001673 return 0;
1674 }
1675
1676 /* The memcpy() statement assumes that the wildcard characters are
1677 ** the first three statements in the compareInfo structure. The
1678 ** asserts() that follow verify that assumption
1679 */
1680 memcpy(aWc, pDef->pUserData, 3);
1681 assert( (char*)&likeInfoAlt == (char*)&likeInfoAlt.matchAll );
1682 assert( &((char*)&likeInfoAlt)[1] == (char*)&likeInfoAlt.matchOne );
1683 assert( &((char*)&likeInfoAlt)[2] == (char*)&likeInfoAlt.matchSet );
drhd36e1042013-09-06 13:10:12 +00001684 *pIsNocase = (pDef->funcFlags & SQLITE_FUNC_CASE)==0;
drh55ef4d92005-08-14 01:20:37 +00001685 return 1;
drhdc04c582002-02-24 01:55:15 +00001686}
danielk19778c0a7912008-08-20 14:49:23 +00001687
drh70a8ca32008-08-21 18:49:27 +00001688/*
peter.d.reid60ec9142014-09-06 16:39:46 +00001689** All of the FuncDef structures in the aBuiltinFunc[] array above
drh777c5382008-08-21 20:21:34 +00001690** to the global function hash table. This occurs at start-time (as
1691** a consequence of calling sqlite3_initialize()).
1692**
1693** After this routine runs
drh70a8ca32008-08-21 18:49:27 +00001694*/
1695void sqlite3RegisterGlobalFunctions(void){
danielk197793ce7412008-09-01 19:14:02 +00001696 /*
1697 ** The following array holds FuncDef structures for all of the functions
1698 ** defined in this file.
1699 **
1700 ** The array cannot be constant since changes are made to the
1701 ** FuncDef.pHash elements at start-time. The elements of this array
1702 ** are read-only after initialization is complete.
1703 */
1704 static SQLITE_WSD FuncDef aBuiltinFunc[] = {
1705 FUNCTION(ltrim, 1, 1, 0, trimFunc ),
1706 FUNCTION(ltrim, 2, 1, 0, trimFunc ),
1707 FUNCTION(rtrim, 1, 2, 0, trimFunc ),
1708 FUNCTION(rtrim, 2, 2, 0, trimFunc ),
1709 FUNCTION(trim, 1, 3, 0, trimFunc ),
1710 FUNCTION(trim, 2, 3, 0, trimFunc ),
1711 FUNCTION(min, -1, 0, 1, minmaxFunc ),
1712 FUNCTION(min, 0, 0, 1, 0 ),
drh9588ad92014-09-15 14:46:02 +00001713 AGGREGATE2(min, 1, 0, 1, minmaxStep, minMaxFinalize,
1714 SQLITE_FUNC_MINMAX ),
danielk197793ce7412008-09-01 19:14:02 +00001715 FUNCTION(max, -1, 1, 1, minmaxFunc ),
1716 FUNCTION(max, 0, 1, 1, 0 ),
drh9588ad92014-09-15 14:46:02 +00001717 AGGREGATE2(max, 1, 1, 1, minmaxStep, minMaxFinalize,
1718 SQLITE_FUNC_MINMAX ),
drha748fdc2012-03-28 01:34:47 +00001719 FUNCTION2(typeof, 1, 0, 0, typeofFunc, SQLITE_FUNC_TYPEOF),
1720 FUNCTION2(length, 1, 0, 0, lengthFunc, SQLITE_FUNC_LENGTH),
drhd55e0722012-10-25 03:07:29 +00001721 FUNCTION(instr, 2, 0, 0, instrFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001722 FUNCTION(substr, 2, 0, 0, substrFunc ),
1723 FUNCTION(substr, 3, 0, 0, substrFunc ),
drha5c14162013-12-17 15:03:06 +00001724 FUNCTION(printf, -1, 0, 0, printfFunc ),
drhd495d8c2013-02-22 19:34:25 +00001725 FUNCTION(unicode, 1, 0, 0, unicodeFunc ),
1726 FUNCTION(char, -1, 0, 0, charFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001727 FUNCTION(abs, 1, 0, 0, absFunc ),
shanefbd60f82009-02-04 03:59:25 +00001728#ifndef SQLITE_OMIT_FLOATING_POINT
danielk197793ce7412008-09-01 19:14:02 +00001729 FUNCTION(round, 1, 0, 0, roundFunc ),
1730 FUNCTION(round, 2, 0, 0, roundFunc ),
shanefbd60f82009-02-04 03:59:25 +00001731#endif
danielk197793ce7412008-09-01 19:14:02 +00001732 FUNCTION(upper, 1, 0, 0, upperFunc ),
1733 FUNCTION(lower, 1, 0, 0, lowerFunc ),
1734 FUNCTION(coalesce, 1, 0, 0, 0 ),
danielk197793ce7412008-09-01 19:14:02 +00001735 FUNCTION(coalesce, 0, 0, 0, 0 ),
drhcca9f3d2013-09-06 15:23:29 +00001736 FUNCTION2(coalesce, -1, 0, 0, noopFunc, SQLITE_FUNC_COALESCE),
danielk197793ce7412008-09-01 19:14:02 +00001737 FUNCTION(hex, 1, 0, 0, hexFunc ),
drhcca9f3d2013-09-06 15:23:29 +00001738 FUNCTION2(ifnull, 2, 0, 0, noopFunc, SQLITE_FUNC_COALESCE),
1739 FUNCTION2(unlikely, 1, 0, 0, noopFunc, SQLITE_FUNC_UNLIKELY),
drhaae0f9e2013-09-11 11:38:58 +00001740 FUNCTION2(likelihood, 2, 0, 0, noopFunc, SQLITE_FUNC_UNLIKELY),
drh03202a92014-06-17 16:11:28 +00001741 FUNCTION2(likely, 1, 0, 0, noopFunc, SQLITE_FUNC_UNLIKELY),
drhb1fba282013-11-21 14:33:48 +00001742 VFUNCTION(random, 0, 0, 0, randomFunc ),
1743 VFUNCTION(randomblob, 1, 0, 0, randomBlob ),
danielk197793ce7412008-09-01 19:14:02 +00001744 FUNCTION(nullif, 2, 0, 1, nullifFunc ),
1745 FUNCTION(sqlite_version, 0, 0, 0, versionFunc ),
drh47baebc2009-08-14 16:01:24 +00001746 FUNCTION(sqlite_source_id, 0, 0, 0, sourceidFunc ),
drh840561f2011-04-27 18:08:42 +00001747 FUNCTION(sqlite_log, 2, 0, 0, errlogFunc ),
drhf442e332014-09-10 19:01:14 +00001748#if SQLITE_USER_AUTHENTICATION
1749 FUNCTION(sqlite_crypt, 2, 0, 0, sqlite3CryptFunc ),
1750#endif
shanehdc97a8c2010-02-23 20:08:35 +00001751#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
drh8bb76d32010-02-26 16:37:47 +00001752 FUNCTION(sqlite_compileoption_used,1, 0, 0, compileoptionusedFunc ),
1753 FUNCTION(sqlite_compileoption_get, 1, 0, 0, compileoptiongetFunc ),
shanehdc97a8c2010-02-23 20:08:35 +00001754#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
danielk197793ce7412008-09-01 19:14:02 +00001755 FUNCTION(quote, 1, 0, 0, quoteFunc ),
drhb1fba282013-11-21 14:33:48 +00001756 VFUNCTION(last_insert_rowid, 0, 0, 0, last_insert_rowid),
1757 VFUNCTION(changes, 0, 0, 0, changes ),
1758 VFUNCTION(total_changes, 0, 0, 0, total_changes ),
danielk197793ce7412008-09-01 19:14:02 +00001759 FUNCTION(replace, 3, 0, 0, replaceFunc ),
1760 FUNCTION(zeroblob, 1, 0, 0, zeroblobFunc ),
1761 #ifdef SQLITE_SOUNDEX
1762 FUNCTION(soundex, 1, 0, 0, soundexFunc ),
1763 #endif
1764 #ifndef SQLITE_OMIT_LOAD_EXTENSION
1765 FUNCTION(load_extension, 1, 0, 0, loadExt ),
1766 FUNCTION(load_extension, 2, 0, 0, loadExt ),
1767 #endif
1768 AGGREGATE(sum, 1, 0, 0, sumStep, sumFinalize ),
1769 AGGREGATE(total, 1, 0, 0, sumStep, totalFinalize ),
1770 AGGREGATE(avg, 1, 0, 0, sumStep, avgFinalize ),
drh9588ad92014-09-15 14:46:02 +00001771 AGGREGATE2(count, 0, 0, 0, countStep, countFinalize,
1772 SQLITE_FUNC_COUNT ),
danielk197793ce7412008-09-01 19:14:02 +00001773 AGGREGATE(count, 1, 0, 0, countStep, countFinalize ),
drh07d31172009-02-02 21:57:05 +00001774 AGGREGATE(group_concat, 1, 0, 0, groupConcatStep, groupConcatFinalize),
1775 AGGREGATE(group_concat, 2, 0, 0, groupConcatStep, groupConcatFinalize),
danielk197793ce7412008-09-01 19:14:02 +00001776
1777 LIKEFUNC(glob, 2, &globInfo, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
1778 #ifdef SQLITE_CASE_SENSITIVE_LIKE
1779 LIKEFUNC(like, 2, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
1780 LIKEFUNC(like, 3, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
1781 #else
1782 LIKEFUNC(like, 2, &likeInfoNorm, SQLITE_FUNC_LIKE),
1783 LIKEFUNC(like, 3, &likeInfoNorm, SQLITE_FUNC_LIKE),
1784 #endif
1785 };
1786
drh70a8ca32008-08-21 18:49:27 +00001787 int i;
danielk197793ce7412008-09-01 19:14:02 +00001788 FuncDefHash *pHash = &GLOBAL(FuncDefHash, sqlite3GlobalFunctions);
drh106cee52008-09-03 17:11:16 +00001789 FuncDef *aFunc = (FuncDef*)&GLOBAL(FuncDef, aBuiltinFunc);
danielk197793ce7412008-09-01 19:14:02 +00001790
drh70a8ca32008-08-21 18:49:27 +00001791 for(i=0; i<ArraySize(aBuiltinFunc); i++){
danielk197793ce7412008-09-01 19:14:02 +00001792 sqlite3FuncDefInsert(pHash, &aFunc[i]);
drh70a8ca32008-08-21 18:49:27 +00001793 }
drh777c5382008-08-21 20:21:34 +00001794 sqlite3RegisterDateTimeFunctions();
drh545f5872010-04-24 14:02:59 +00001795#ifndef SQLITE_OMIT_ALTERTABLE
1796 sqlite3AlterFunctions();
1797#endif
dan0106e372013-08-12 16:34:32 +00001798#if defined(SQLITE_ENABLE_STAT3) || defined(SQLITE_ENABLE_STAT4)
1799 sqlite3AnalyzeFunctions();
1800#endif
drh70a8ca32008-08-21 18:49:27 +00001801}