blob: b927296782b25bd7d4d6fd62d7ed987f2e47f4a8 [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 {
drh07e83472016-01-11 03:48:18 +0000570 u8 matchAll; /* "*" or "%" */
571 u8 matchOne; /* "?" or "_" */
572 u8 matchSet; /* "[" or 0 */
573 u8 noCase; /* true to ignore case differences */
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
drhb0870482015-06-17 13:20:54 +0000578** character is exactly one byte in size. Also, provde the Utf8Read()
579** macro for fast reading of the next character in the common case where
580** the next character is ASCII.
drhb9175ae2007-12-07 18:39:04 +0000581*/
582#if defined(SQLITE_EBCDIC)
drh88b33222014-09-25 03:51:37 +0000583# define sqlite3Utf8Read(A) (*((*A)++))
drhb0870482015-06-17 13:20:54 +0000584# define Utf8Read(A) (*(A++))
drhb9175ae2007-12-07 18:39:04 +0000585#else
drhb0870482015-06-17 13:20:54 +0000586# define Utf8Read(A) (A[0]<0x80?*(A++):sqlite3Utf8Read(&A))
drhb9175ae2007-12-07 18:39:04 +0000587#endif
588
drh4e5ffc52004-08-31 00:52:37 +0000589static const struct compareInfo globInfo = { '*', '?', '[', 0 };
drh55ef4d92005-08-14 01:20:37 +0000590/* The correct SQL-92 behavior is for the LIKE operator to ignore
591** case. Thus 'a' LIKE 'A' would be true. */
592static const struct compareInfo likeInfoNorm = { '%', '_', 0, 1 };
593/* If SQLITE_CASE_SENSITIVE_LIKE is defined, then the LIKE operator
594** is case sensitive causing 'a' LIKE 'A' to be false */
595static const struct compareInfo likeInfoAlt = { '%', '_', 0, 0 };
danielk1977d02eb1f2004-06-06 09:44:03 +0000596
597/*
drh4e5ffc52004-08-31 00:52:37 +0000598** Compare two UTF-8 strings for equality where the first string can
drh9fdfdc82014-09-25 11:08:57 +0000599** potentially be a "glob" or "like" expression. Return true (1) if they
drh4e5ffc52004-08-31 00:52:37 +0000600** are the same and false (0) if they are different.
drh0ac65892002-04-20 14:24:41 +0000601**
drh4e5ffc52004-08-31 00:52:37 +0000602** Globbing rules:
drh0ac65892002-04-20 14:24:41 +0000603**
drh4e5ffc52004-08-31 00:52:37 +0000604** '*' Matches any sequence of zero or more characters.
danielk1977d02eb1f2004-06-06 09:44:03 +0000605**
drh4e5ffc52004-08-31 00:52:37 +0000606** '?' Matches exactly one character.
607**
608** [...] Matches one character from the enclosed list of
609** characters.
610**
611** [^...] Matches one character not in the enclosed list.
612**
613** With the [...] and [^...] matching, a ']' character can be included
614** in the list by making it the first character after '[' or '^'. A
615** range of characters can be specified using '-'. Example:
616** "[a-z]" matches any single lower-case letter. To match a '-', make
617** it the last character in the list.
618**
drh9fdfdc82014-09-25 11:08:57 +0000619** Like matching rules:
620**
621** '%' Matches any sequence of zero or more characters
622**
623*** '_' Matches any one character
624**
625** Ec Where E is the "esc" character and c is any other
626** character, including '%', '_', and esc, match exactly c.
627**
drhb0870482015-06-17 13:20:54 +0000628** The comments within this routine usually assume glob matching.
drh9fdfdc82014-09-25 11:08:57 +0000629**
drh4e5ffc52004-08-31 00:52:37 +0000630** This routine is usually quick, but can be N**2 in the worst case.
drh0ac65892002-04-20 14:24:41 +0000631*/
danielk19777c6303c2004-11-17 16:41:29 +0000632static int patternCompare(
drh4e5ffc52004-08-31 00:52:37 +0000633 const u8 *zPattern, /* The glob pattern */
634 const u8 *zString, /* The string to compare against the glob */
danielk19777c6303c2004-11-17 16:41:29 +0000635 const struct compareInfo *pInfo, /* Information about how to do the compare */
drh07e83472016-01-11 03:48:18 +0000636 u32 matchOther /* The escape char (LIKE) or '[' (GLOB) */
danielk197751ad0ec2004-05-24 12:39:02 +0000637){
drh9fdfdc82014-09-25 11:08:57 +0000638 u32 c, c2; /* Next pattern and input string chars */
639 u32 matchOne = pInfo->matchOne; /* "?" or "_" */
640 u32 matchAll = pInfo->matchAll; /* "*" or "%" */
drh9fdfdc82014-09-25 11:08:57 +0000641 u8 noCase = pInfo->noCase; /* True if uppercase==lowercase */
642 const u8 *zEscaped = 0; /* One past the last escaped input char */
drh88b33222014-09-25 03:51:37 +0000643
drhb0870482015-06-17 13:20:54 +0000644 while( (c = Utf8Read(zPattern))!=0 ){
drh9fdfdc82014-09-25 11:08:57 +0000645 if( c==matchAll ){ /* Match "*" */
646 /* Skip over multiple "*" characters in the pattern. If there
647 ** are also "?" characters, skip those as well, but consume a
648 ** single character of the input string for each "?" skipped */
drhb0870482015-06-17 13:20:54 +0000649 while( (c=Utf8Read(zPattern)) == matchAll || c == matchOne ){
drh42610962012-09-17 18:56:32 +0000650 if( c==matchOne && sqlite3Utf8Read(&zString)==0 ){
drh66150952007-07-23 19:12:41 +0000651 return 0;
drh4e5ffc52004-08-31 00:52:37 +0000652 }
danielk1977ad7dd422004-06-06 12:41:49 +0000653 }
drh66150952007-07-23 19:12:41 +0000654 if( c==0 ){
drh9fdfdc82014-09-25 11:08:57 +0000655 return 1; /* "*" at the end of the pattern matches */
drh88b33222014-09-25 03:51:37 +0000656 }else if( c==matchOther ){
drh07e83472016-01-11 03:48:18 +0000657 if( pInfo->matchSet==0 ){
drh88b33222014-09-25 03:51:37 +0000658 c = sqlite3Utf8Read(&zPattern);
659 if( c==0 ) return 0;
660 }else{
drh9fdfdc82014-09-25 11:08:57 +0000661 /* "[...]" immediately follows the "*". We have to do a slow
662 ** recursive search in this case, but it is an unusual case. */
drh88b33222014-09-25 03:51:37 +0000663 assert( matchOther<0x80 ); /* '[' is a single-byte character */
664 while( *zString
drh07e83472016-01-11 03:48:18 +0000665 && patternCompare(&zPattern[-1],zString,pInfo,matchOther)==0 ){
drh88b33222014-09-25 03:51:37 +0000666 SQLITE_SKIP_UTF8(zString);
667 }
668 return *zString!=0;
drh66150952007-07-23 19:12:41 +0000669 }
drh66150952007-07-23 19:12:41 +0000670 }
drh9fdfdc82014-09-25 11:08:57 +0000671
672 /* At this point variable c contains the first character of the
673 ** pattern string past the "*". Search in the input string for the
674 ** first matching character and recursively contine the match from
675 ** that point.
676 **
677 ** For a case-insensitive search, set variable cx to be the same as
678 ** c but in the other case and search the input string for either
679 ** c or cx.
680 */
681 if( c<=0x80 ){
682 u32 cx;
683 if( noCase ){
684 cx = sqlite3Toupper(c);
685 c = sqlite3Tolower(c);
drh66150952007-07-23 19:12:41 +0000686 }else{
drh9fdfdc82014-09-25 11:08:57 +0000687 cx = c;
drh4e5ffc52004-08-31 00:52:37 +0000688 }
drh9fdfdc82014-09-25 11:08:57 +0000689 while( (c2 = *(zString++))!=0 ){
690 if( c2!=c && c2!=cx ) continue;
drh07e83472016-01-11 03:48:18 +0000691 if( patternCompare(zPattern,zString,pInfo,matchOther) ) return 1;
drh9fdfdc82014-09-25 11:08:57 +0000692 }
693 }else{
drhb0870482015-06-17 13:20:54 +0000694 while( (c2 = Utf8Read(zString))!=0 ){
drh9fdfdc82014-09-25 11:08:57 +0000695 if( c2!=c ) continue;
drh07e83472016-01-11 03:48:18 +0000696 if( patternCompare(zPattern,zString,pInfo,matchOther) ) return 1;
drh9fdfdc82014-09-25 11:08:57 +0000697 }
drh66150952007-07-23 19:12:41 +0000698 }
699 return 0;
drh88b33222014-09-25 03:51:37 +0000700 }
drh88b33222014-09-25 03:51:37 +0000701 if( c==matchOther ){
drh07e83472016-01-11 03:48:18 +0000702 if( pInfo->matchSet==0 ){
drh88b33222014-09-25 03:51:37 +0000703 c = sqlite3Utf8Read(&zPattern);
704 if( c==0 ) return 0;
drh9fdfdc82014-09-25 11:08:57 +0000705 zEscaped = zPattern;
drh88b33222014-09-25 03:51:37 +0000706 }else{
707 u32 prior_c = 0;
drh9fdfdc82014-09-25 11:08:57 +0000708 int seen = 0;
709 int invert = 0;
drh88b33222014-09-25 03:51:37 +0000710 c = sqlite3Utf8Read(&zString);
711 if( c==0 ) return 0;
712 c2 = sqlite3Utf8Read(&zPattern);
713 if( c2=='^' ){
714 invert = 1;
715 c2 = sqlite3Utf8Read(&zPattern);
716 }
717 if( c2==']' ){
718 if( c==']' ) seen = 1;
719 c2 = sqlite3Utf8Read(&zPattern);
720 }
721 while( c2 && c2!=']' ){
722 if( c2=='-' && zPattern[0]!=']' && zPattern[0]!=0 && prior_c>0 ){
723 c2 = sqlite3Utf8Read(&zPattern);
724 if( c>=prior_c && c<=c2 ) seen = 1;
725 prior_c = 0;
726 }else{
727 if( c==c2 ){
728 seen = 1;
729 }
730 prior_c = c2;
731 }
732 c2 = sqlite3Utf8Read(&zPattern);
733 }
734 if( c2==0 || (seen ^ invert)==0 ){
735 return 0;
736 }
737 continue;
738 }
739 }
drhb0870482015-06-17 13:20:54 +0000740 c2 = Utf8Read(zString);
drh88b33222014-09-25 03:51:37 +0000741 if( c==c2 ) continue;
drhfb397262016-01-20 14:22:41 +0000742 if( noCase && c<0x80 && c2<0x80 && sqlite3Tolower(c)==sqlite3Tolower(c2) ){
drh9fdfdc82014-09-25 11:08:57 +0000743 continue;
744 }
745 if( c==matchOne && zPattern!=zEscaped && c2!=0 ) continue;
746 return 0;
danielk197751ad0ec2004-05-24 12:39:02 +0000747 }
drh4e5ffc52004-08-31 00:52:37 +0000748 return *zString==0;
drh0ac65892002-04-20 14:24:41 +0000749}
drh4e5ffc52004-08-31 00:52:37 +0000750
drh55ef4d92005-08-14 01:20:37 +0000751/*
drh56282a52013-04-10 16:13:38 +0000752** The sqlite3_strglob() interface.
753*/
754int sqlite3_strglob(const char *zGlobPattern, const char *zString){
drh07e83472016-01-11 03:48:18 +0000755 return patternCompare((u8*)zGlobPattern, (u8*)zString, &globInfo, '[')==0;
drh56282a52013-04-10 16:13:38 +0000756}
757
758/*
drh8b4a94a2015-11-24 21:23:59 +0000759** The sqlite3_strlike() interface.
760*/
761int sqlite3_strlike(const char *zPattern, const char *zStr, unsigned int esc){
762 return patternCompare((u8*)zPattern, (u8*)zStr, &likeInfoNorm, esc)==0;
763}
764
765/*
drh55ef4d92005-08-14 01:20:37 +0000766** Count the number of times that the LIKE operator (or GLOB which is
767** just a variation of LIKE) gets called. This is used for testing
768** only.
769*/
770#ifdef SQLITE_TEST
771int sqlite3_like_count = 0;
772#endif
773
danielk19773f6b0872004-06-17 05:36:44 +0000774
775/*
776** Implementation of the like() SQL function. This function implements
777** the build-in LIKE operator. The first argument to the function is the
778** pattern and the second argument is the string. So, the SQL statements:
779**
780** A LIKE B
781**
782** is implemented as like(B,A).
783**
drh55ef4d92005-08-14 01:20:37 +0000784** This same function (with a different compareInfo structure) computes
785** the GLOB operator.
danielk19773f6b0872004-06-17 05:36:44 +0000786*/
787static void likeFunc(
788 sqlite3_context *context,
789 int argc,
790 sqlite3_value **argv
791){
drhbeb818d2007-05-08 15:34:47 +0000792 const unsigned char *zA, *zB;
drh07e83472016-01-11 03:48:18 +0000793 u32 escape;
drh27e62db2009-04-02 10:16:17 +0000794 int nPat;
drhbb4957f2008-03-20 14:03:29 +0000795 sqlite3 *db = sqlite3_context_db_handle(context);
drh07e83472016-01-11 03:48:18 +0000796 struct compareInfo *pInfo = sqlite3_user_data(context);
drhbeb818d2007-05-08 15:34:47 +0000797
drh41d2e662015-12-01 21:23:07 +0000798#ifdef SQLITE_LIKE_DOESNT_MATCH_BLOBS
799 if( sqlite3_value_type(argv[0])==SQLITE_BLOB
800 || sqlite3_value_type(argv[1])==SQLITE_BLOB
801 ){
802#ifdef SQLITE_TEST
803 sqlite3_like_count++;
804#endif
805 sqlite3_result_int(context, 0);
806 return;
807 }
808#endif
drh1f0feef2007-05-15 13:27:07 +0000809 zB = sqlite3_value_text(argv[0]);
810 zA = sqlite3_value_text(argv[1]);
811
drhbeb818d2007-05-08 15:34:47 +0000812 /* Limit the length of the LIKE or GLOB pattern to avoid problems
813 ** of deep recursion and N*N behavior in patternCompare().
814 */
drh27e62db2009-04-02 10:16:17 +0000815 nPat = sqlite3_value_bytes(argv[0]);
816 testcase( nPat==db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH] );
817 testcase( nPat==db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH]+1 );
818 if( nPat > db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH] ){
drhbeb818d2007-05-08 15:34:47 +0000819 sqlite3_result_error(context, "LIKE or GLOB pattern too complex", -1);
820 return;
821 }
drh1f0feef2007-05-15 13:27:07 +0000822 assert( zB==sqlite3_value_text(argv[0]) ); /* Encoding did not change */
drhbeb818d2007-05-08 15:34:47 +0000823
danielk19777c6303c2004-11-17 16:41:29 +0000824 if( argc==3 ){
825 /* The escape character string must consist of a single UTF-8 character.
826 ** Otherwise, return an error.
827 */
828 const unsigned char *zEsc = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +0000829 if( zEsc==0 ) return;
drhee858132007-05-08 20:37:38 +0000830 if( sqlite3Utf8CharLen((char*)zEsc, -1)!=1 ){
danielk19777c6303c2004-11-17 16:41:29 +0000831 sqlite3_result_error(context,
832 "ESCAPE expression must be a single character", -1);
833 return;
834 }
drh42610962012-09-17 18:56:32 +0000835 escape = sqlite3Utf8Read(&zEsc);
drh07e83472016-01-11 03:48:18 +0000836 }else{
837 escape = pInfo->matchSet;
danielk19777c6303c2004-11-17 16:41:29 +0000838 }
danielk19773f6b0872004-06-17 05:36:44 +0000839 if( zA && zB ){
drh55ef4d92005-08-14 01:20:37 +0000840#ifdef SQLITE_TEST
841 sqlite3_like_count++;
842#endif
danielk1977b56fe1f2007-05-09 08:24:44 +0000843 sqlite3_result_int(context, patternCompare(zB, zA, pInfo, escape));
danielk197751ad0ec2004-05-24 12:39:02 +0000844 }
drh8912d102002-05-26 21:34:58 +0000845}
846
847/*
848** Implementation of the NULLIF(x,y) function. The result is the first
849** argument if the arguments are different. The result is NULL if the
850** arguments are equal to each other.
851*/
drhf9b596e2004-05-26 16:54:42 +0000852static void nullifFunc(
853 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000854 int NotUsed,
drhf9b596e2004-05-26 16:54:42 +0000855 sqlite3_value **argv
856){
danielk1977dc1bdc42004-06-11 10:51:27 +0000857 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk197762c14b32008-11-19 09:05:26 +0000858 UNUSED_PARAMETER(NotUsed);
danielk1977dc1bdc42004-06-11 10:51:27 +0000859 if( sqlite3MemCompare(argv[0], argv[1], pColl)!=0 ){
drhf4479502004-05-27 03:12:53 +0000860 sqlite3_result_value(context, argv[0]);
drh8912d102002-05-26 21:34:58 +0000861 }
drh0ac65892002-04-20 14:24:41 +0000862}
863
drh647cb0e2002-11-04 19:32:25 +0000864/*
drh47baebc2009-08-14 16:01:24 +0000865** Implementation of the sqlite_version() function. The result is the version
drh647cb0e2002-11-04 19:32:25 +0000866** of the SQLite library that is running.
867*/
drhf9b596e2004-05-26 16:54:42 +0000868static void versionFunc(
869 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000870 int NotUsed,
871 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000872){
danielk197762c14b32008-11-19 09:05:26 +0000873 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000874 /* IMP: R-48699-48617 This function is an SQL wrapper around the
875 ** sqlite3_libversion() C-interface. */
876 sqlite3_result_text(context, sqlite3_libversion(), -1, SQLITE_STATIC);
drh647cb0e2002-11-04 19:32:25 +0000877}
878
drh47baebc2009-08-14 16:01:24 +0000879/*
880** Implementation of the sqlite_source_id() function. The result is a string
881** that identifies the particular version of the source code used to build
882** SQLite.
883*/
884static void sourceidFunc(
885 sqlite3_context *context,
886 int NotUsed,
887 sqlite3_value **NotUsed2
888){
889 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000890 /* IMP: R-24470-31136 This function is an SQL wrapper around the
891 ** sqlite3_sourceid() C interface. */
892 sqlite3_result_text(context, sqlite3_sourceid(), -1, SQLITE_STATIC);
drh47baebc2009-08-14 16:01:24 +0000893}
894
shanehbdea6d12010-02-23 04:19:54 +0000895/*
drh3ca84ef2011-04-25 18:03:10 +0000896** Implementation of the sqlite_log() function. This is a wrapper around
897** sqlite3_log(). The return value is NULL. The function exists purely for
898** its side-effects.
899*/
drh840561f2011-04-27 18:08:42 +0000900static void errlogFunc(
drh3ca84ef2011-04-25 18:03:10 +0000901 sqlite3_context *context,
902 int argc,
903 sqlite3_value **argv
904){
905 UNUSED_PARAMETER(argc);
906 UNUSED_PARAMETER(context);
907 sqlite3_log(sqlite3_value_int(argv[0]), "%s", sqlite3_value_text(argv[1]));
908}
909
910/*
shanehdc97a8c2010-02-23 20:08:35 +0000911** Implementation of the sqlite_compileoption_used() function.
912** The result is an integer that identifies if the compiler option
913** was used to build SQLite.
shanehbdea6d12010-02-23 04:19:54 +0000914*/
shanehdc97a8c2010-02-23 20:08:35 +0000915#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
916static void compileoptionusedFunc(
shanehbdea6d12010-02-23 04:19:54 +0000917 sqlite3_context *context,
shanehdc97a8c2010-02-23 20:08:35 +0000918 int argc,
919 sqlite3_value **argv
shanehbdea6d12010-02-23 04:19:54 +0000920){
shanehdc97a8c2010-02-23 20:08:35 +0000921 const char *zOptName;
922 assert( argc==1 );
923 UNUSED_PARAMETER(argc);
drha3e414c2010-08-03 18:06:25 +0000924 /* IMP: R-39564-36305 The sqlite_compileoption_used() SQL
925 ** function is a wrapper around the sqlite3_compileoption_used() C/C++
926 ** function.
927 */
drh264a2d42010-02-25 15:28:41 +0000928 if( (zOptName = (const char*)sqlite3_value_text(argv[0]))!=0 ){
shanehdc97a8c2010-02-23 20:08:35 +0000929 sqlite3_result_int(context, sqlite3_compileoption_used(zOptName));
shanehdc97a8c2010-02-23 20:08:35 +0000930 }
shanehbdea6d12010-02-23 04:19:54 +0000931}
shanehdc97a8c2010-02-23 20:08:35 +0000932#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
933
934/*
935** Implementation of the sqlite_compileoption_get() function.
936** The result is a string that identifies the compiler options
937** used to build SQLite.
938*/
939#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
940static void compileoptiongetFunc(
941 sqlite3_context *context,
942 int argc,
943 sqlite3_value **argv
944){
945 int n;
946 assert( argc==1 );
947 UNUSED_PARAMETER(argc);
drha3e414c2010-08-03 18:06:25 +0000948 /* IMP: R-04922-24076 The sqlite_compileoption_get() SQL function
949 ** is a wrapper around the sqlite3_compileoption_get() C/C++ function.
950 */
shanehdc97a8c2010-02-23 20:08:35 +0000951 n = sqlite3_value_int(argv[0]);
952 sqlite3_result_text(context, sqlite3_compileoption_get(n), -1, SQLITE_STATIC);
953}
954#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
shanehbdea6d12010-02-23 04:19:54 +0000955
drh137c7282007-01-29 17:58:28 +0000956/* Array for converting from half-bytes (nybbles) into ASCII hex
957** digits. */
958static const char hexdigits[] = {
959 '0', '1', '2', '3', '4', '5', '6', '7',
960 '8', '9', 'A', 'B', 'C', 'D', 'E', 'F'
961};
danielk1977d641d642004-11-18 15:44:29 +0000962
drh47394702003-08-20 01:03:33 +0000963/*
drh47394702003-08-20 01:03:33 +0000964** Implementation of the QUOTE() function. This function takes a single
965** argument. If the argument is numeric, the return value is the same as
966** the argument. If the argument is NULL, the return value is the string
967** "NULL". Otherwise, the argument is enclosed in single quotes with
968** single-quote escapes.
969*/
danielk19770ae8b832004-05-25 12:05:56 +0000970static void quoteFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drha0df4cc2009-02-02 17:29:59 +0000971 assert( argc==1 );
drh1d34fde2009-02-03 15:50:33 +0000972 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +0000973 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +0000974 case SQLITE_FLOAT: {
drh72b3fbc2012-06-19 03:11:25 +0000975 double r1, r2;
drh72b3fbc2012-06-19 03:11:25 +0000976 char zBuf[50];
mistachkin2b434a72012-06-19 04:36:48 +0000977 r1 = sqlite3_value_double(argv[0]);
drh72b3fbc2012-06-19 03:11:25 +0000978 sqlite3_snprintf(sizeof(zBuf), zBuf, "%!.15g", r1);
979 sqlite3AtoF(zBuf, &r2, 20, SQLITE_UTF8);
980 if( r1!=r2 ){
981 sqlite3_snprintf(sizeof(zBuf), zBuf, "%!.20e", r1);
982 }
983 sqlite3_result_text(context, zBuf, -1, SQLITE_TRANSIENT);
984 break;
985 }
986 case SQLITE_INTEGER: {
drhf4479502004-05-27 03:12:53 +0000987 sqlite3_result_value(context, argv[0]);
drhf9b596e2004-05-26 16:54:42 +0000988 break;
989 }
danielk19773f41e972004-06-08 00:39:01 +0000990 case SQLITE_BLOB: {
danielk19773f41e972004-06-08 00:39:01 +0000991 char *zText = 0;
danielk19773f41e972004-06-08 00:39:01 +0000992 char const *zBlob = sqlite3_value_blob(argv[0]);
drh1f0feef2007-05-15 13:27:07 +0000993 int nBlob = sqlite3_value_bytes(argv[0]);
994 assert( zBlob==sqlite3_value_blob(argv[0]) ); /* No encoding change */
drhb1a6c3c2008-03-20 16:30:17 +0000995 zText = (char *)contextMalloc(context, (2*(i64)nBlob)+4);
danielk1977a1644fd2007-08-29 12:31:25 +0000996 if( zText ){
danielk19773f41e972004-06-08 00:39:01 +0000997 int i;
998 for(i=0; i<nBlob; i++){
999 zText[(i*2)+2] = hexdigits[(zBlob[i]>>4)&0x0F];
1000 zText[(i*2)+3] = hexdigits[(zBlob[i])&0x0F];
1001 }
1002 zText[(nBlob*2)+2] = '\'';
1003 zText[(nBlob*2)+3] = '\0';
1004 zText[0] = 'X';
1005 zText[1] = '\'';
danielk1977d8123362004-06-12 09:25:12 +00001006 sqlite3_result_text(context, zText, -1, SQLITE_TRANSIENT);
drh17435752007-08-16 04:30:38 +00001007 sqlite3_free(zText);
danielk19773f41e972004-06-08 00:39:01 +00001008 }
1009 break;
1010 }
drh9c054832004-05-31 18:51:57 +00001011 case SQLITE_TEXT: {
drh023ae032007-05-08 12:12:16 +00001012 int i,j;
1013 u64 n;
drh2646da72005-12-09 20:02:05 +00001014 const unsigned char *zArg = sqlite3_value_text(argv[0]);
drhf9b596e2004-05-26 16:54:42 +00001015 char *z;
1016
drh7a521cf2007-04-25 18:23:52 +00001017 if( zArg==0 ) return;
drh023ae032007-05-08 12:12:16 +00001018 for(i=0, n=0; zArg[i]; i++){ if( zArg[i]=='\'' ) n++; }
drhb1a6c3c2008-03-20 16:30:17 +00001019 z = contextMalloc(context, ((i64)i)+((i64)n)+3);
danielk1977a1644fd2007-08-29 12:31:25 +00001020 if( z ){
1021 z[0] = '\'';
1022 for(i=0, j=1; zArg[i]; i++){
1023 z[j++] = zArg[i];
1024 if( zArg[i]=='\'' ){
1025 z[j++] = '\'';
1026 }
drhf9b596e2004-05-26 16:54:42 +00001027 }
danielk1977a1644fd2007-08-29 12:31:25 +00001028 z[j++] = '\'';
1029 z[j] = 0;
1030 sqlite3_result_text(context, z, j, sqlite3_free);
drhf9b596e2004-05-26 16:54:42 +00001031 }
drha0df4cc2009-02-02 17:29:59 +00001032 break;
1033 }
1034 default: {
1035 assert( sqlite3_value_type(argv[0])==SQLITE_NULL );
1036 sqlite3_result_text(context, "NULL", 4, SQLITE_STATIC);
1037 break;
drhf9b596e2004-05-26 16:54:42 +00001038 }
drh47394702003-08-20 01:03:33 +00001039 }
1040}
1041
drh137c7282007-01-29 17:58:28 +00001042/*
drhd495d8c2013-02-22 19:34:25 +00001043** The unicode() function. Return the integer unicode code-point value
1044** for the first character of the input string.
1045*/
1046static void unicodeFunc(
1047 sqlite3_context *context,
1048 int argc,
1049 sqlite3_value **argv
1050){
1051 const unsigned char *z = sqlite3_value_text(argv[0]);
drh1d59d032013-03-01 23:40:26 +00001052 (void)argc;
drhd495d8c2013-02-22 19:34:25 +00001053 if( z && z[0] ) sqlite3_result_int(context, sqlite3Utf8Read(&z));
1054}
1055
1056/*
1057** The char() function takes zero or more arguments, each of which is
1058** an integer. It constructs a string where each character of the string
1059** is the unicode character for the corresponding integer argument.
1060*/
1061static void charFunc(
1062 sqlite3_context *context,
1063 int argc,
1064 sqlite3_value **argv
1065){
1066 unsigned char *z, *zOut;
1067 int i;
drhf3cdcdc2015-04-29 16:50:28 +00001068 zOut = z = sqlite3_malloc64( argc*4+1 );
drhd495d8c2013-02-22 19:34:25 +00001069 if( z==0 ){
1070 sqlite3_result_error_nomem(context);
1071 return;
1072 }
1073 for(i=0; i<argc; i++){
mistachkinc9545442013-02-26 05:42:30 +00001074 sqlite3_int64 x;
drhd495d8c2013-02-22 19:34:25 +00001075 unsigned c;
1076 x = sqlite3_value_int64(argv[i]);
1077 if( x<0 || x>0x10ffff ) x = 0xfffd;
1078 c = (unsigned)(x & 0x1fffff);
drhfe7a5d12013-03-07 14:00:04 +00001079 if( c<0x00080 ){
1080 *zOut++ = (u8)(c&0xFF);
1081 }else if( c<0x00800 ){
1082 *zOut++ = 0xC0 + (u8)((c>>6)&0x1F);
1083 *zOut++ = 0x80 + (u8)(c & 0x3F);
1084 }else if( c<0x10000 ){
1085 *zOut++ = 0xE0 + (u8)((c>>12)&0x0F);
1086 *zOut++ = 0x80 + (u8)((c>>6) & 0x3F);
1087 *zOut++ = 0x80 + (u8)(c & 0x3F);
drhd495d8c2013-02-22 19:34:25 +00001088 }else{
drhfe7a5d12013-03-07 14:00:04 +00001089 *zOut++ = 0xF0 + (u8)((c>>18) & 0x07);
1090 *zOut++ = 0x80 + (u8)((c>>12) & 0x3F);
1091 *zOut++ = 0x80 + (u8)((c>>6) & 0x3F);
1092 *zOut++ = 0x80 + (u8)(c & 0x3F);
1093 } \
drhd495d8c2013-02-22 19:34:25 +00001094 }
drhbbf483f2014-09-09 20:30:24 +00001095 sqlite3_result_text64(context, (char*)z, zOut-z, sqlite3_free, SQLITE_UTF8);
drhd495d8c2013-02-22 19:34:25 +00001096}
1097
1098/*
drh137c7282007-01-29 17:58:28 +00001099** The hex() function. Interpret the argument as a blob. Return
1100** a hexadecimal rendering as text.
1101*/
1102static void hexFunc(
1103 sqlite3_context *context,
1104 int argc,
1105 sqlite3_value **argv
1106){
1107 int i, n;
1108 const unsigned char *pBlob;
1109 char *zHex, *z;
1110 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001111 UNUSED_PARAMETER(argc);
drh1f0feef2007-05-15 13:27:07 +00001112 pBlob = sqlite3_value_blob(argv[0]);
drh137c7282007-01-29 17:58:28 +00001113 n = sqlite3_value_bytes(argv[0]);
drh1f0feef2007-05-15 13:27:07 +00001114 assert( pBlob==sqlite3_value_blob(argv[0]) ); /* No encoding change */
drhb1a6c3c2008-03-20 16:30:17 +00001115 z = zHex = contextMalloc(context, ((i64)n)*2 + 1);
danielk1977a1644fd2007-08-29 12:31:25 +00001116 if( zHex ){
1117 for(i=0; i<n; i++, pBlob++){
1118 unsigned char c = *pBlob;
1119 *(z++) = hexdigits[(c>>4)&0xf];
1120 *(z++) = hexdigits[c&0xf];
1121 }
1122 *z = 0;
1123 sqlite3_result_text(context, zHex, n*2, sqlite3_free);
drh137c7282007-01-29 17:58:28 +00001124 }
drh137c7282007-01-29 17:58:28 +00001125}
1126
drh26b6d902007-03-17 13:27:54 +00001127/*
drh8cff3822007-05-02 02:08:28 +00001128** The zeroblob(N) function returns a zero-filled blob of size N bytes.
1129*/
1130static void zeroblobFunc(
1131 sqlite3_context *context,
1132 int argc,
1133 sqlite3_value **argv
1134){
drh98640a32007-06-07 19:08:32 +00001135 i64 n;
dana4d5ae82015-07-24 16:24:37 +00001136 int rc;
drh8cff3822007-05-02 02:08:28 +00001137 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001138 UNUSED_PARAMETER(argc);
drh98640a32007-06-07 19:08:32 +00001139 n = sqlite3_value_int64(argv[0]);
dana4d5ae82015-07-24 16:24:37 +00001140 if( n<0 ) n = 0;
1141 rc = sqlite3_result_zeroblob64(context, n); /* IMP: R-00293-64994 */
1142 if( rc ){
1143 sqlite3_result_error_code(context, rc);
drh98640a32007-06-07 19:08:32 +00001144 }
drh8cff3822007-05-02 02:08:28 +00001145}
1146
1147/*
drh26b6d902007-03-17 13:27:54 +00001148** The replace() function. Three arguments are all strings: call
1149** them A, B, and C. The result is also a string which is derived
drhf7b54962013-05-28 12:11:54 +00001150** from A by replacing every occurrence of B with C. The match
drh26b6d902007-03-17 13:27:54 +00001151** must be exact. Collating sequences are not used.
1152*/
1153static void replaceFunc(
1154 sqlite3_context *context,
1155 int argc,
1156 sqlite3_value **argv
1157){
1158 const unsigned char *zStr; /* The input string A */
1159 const unsigned char *zPattern; /* The pattern string B */
1160 const unsigned char *zRep; /* The replacement string C */
1161 unsigned char *zOut; /* The output */
1162 int nStr; /* Size of zStr */
1163 int nPattern; /* Size of zPattern */
1164 int nRep; /* Size of zRep */
drh2e6400b2007-05-08 15:46:18 +00001165 i64 nOut; /* Maximum size of zOut */
drh26b6d902007-03-17 13:27:54 +00001166 int loopLimit; /* Last zStr[] that might match zPattern[] */
1167 int i, j; /* Loop counters */
1168
1169 assert( argc==3 );
danielk1977f3d3c272008-11-19 16:52:44 +00001170 UNUSED_PARAMETER(argc);
drh26b6d902007-03-17 13:27:54 +00001171 zStr = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001172 if( zStr==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001173 nStr = sqlite3_value_bytes(argv[0]);
1174 assert( zStr==sqlite3_value_text(argv[0]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001175 zPattern = sqlite3_value_text(argv[1]);
drha605fe82009-02-01 18:08:40 +00001176 if( zPattern==0 ){
drh23336062009-02-03 13:19:12 +00001177 assert( sqlite3_value_type(argv[1])==SQLITE_NULL
1178 || sqlite3_context_db_handle(context)->mallocFailed );
drha605fe82009-02-01 18:08:40 +00001179 return;
1180 }
1181 if( zPattern[0]==0 ){
1182 assert( sqlite3_value_type(argv[1])!=SQLITE_NULL );
1183 sqlite3_result_value(context, argv[0]);
1184 return;
1185 }
drh1f0feef2007-05-15 13:27:07 +00001186 nPattern = sqlite3_value_bytes(argv[1]);
1187 assert( zPattern==sqlite3_value_text(argv[1]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001188 zRep = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +00001189 if( zRep==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001190 nRep = sqlite3_value_bytes(argv[2]);
1191 assert( zRep==sqlite3_value_text(argv[2]) );
drh2e6400b2007-05-08 15:46:18 +00001192 nOut = nStr + 1;
1193 assert( nOut<SQLITE_MAX_LENGTH );
drhb1a6c3c2008-03-20 16:30:17 +00001194 zOut = contextMalloc(context, (i64)nOut);
drh2e6400b2007-05-08 15:46:18 +00001195 if( zOut==0 ){
1196 return;
drh26b6d902007-03-17 13:27:54 +00001197 }
drh26b6d902007-03-17 13:27:54 +00001198 loopLimit = nStr - nPattern;
1199 for(i=j=0; i<=loopLimit; i++){
1200 if( zStr[i]!=zPattern[0] || memcmp(&zStr[i], zPattern, nPattern) ){
1201 zOut[j++] = zStr[i];
1202 }else{
drh4a50aac2007-08-23 02:47:53 +00001203 u8 *zOld;
drhbb4957f2008-03-20 14:03:29 +00001204 sqlite3 *db = sqlite3_context_db_handle(context);
drh2e6400b2007-05-08 15:46:18 +00001205 nOut += nRep - nPattern;
drh27e62db2009-04-02 10:16:17 +00001206 testcase( nOut-1==db->aLimit[SQLITE_LIMIT_LENGTH] );
1207 testcase( nOut-2==db->aLimit[SQLITE_LIMIT_LENGTH] );
1208 if( nOut-1>db->aLimit[SQLITE_LIMIT_LENGTH] ){
drha0206bc2007-05-08 15:15:02 +00001209 sqlite3_result_error_toobig(context);
drhb9755982010-07-24 16:34:37 +00001210 sqlite3_free(zOut);
danielk197717374e82007-05-08 14:39:04 +00001211 return;
1212 }
drh4a50aac2007-08-23 02:47:53 +00001213 zOld = zOut;
drhf3cdcdc2015-04-29 16:50:28 +00001214 zOut = sqlite3_realloc64(zOut, (int)nOut);
drh2e6400b2007-05-08 15:46:18 +00001215 if( zOut==0 ){
danielk1977a1644fd2007-08-29 12:31:25 +00001216 sqlite3_result_error_nomem(context);
drhb9755982010-07-24 16:34:37 +00001217 sqlite3_free(zOld);
drh2e6400b2007-05-08 15:46:18 +00001218 return;
1219 }
drh26b6d902007-03-17 13:27:54 +00001220 memcpy(&zOut[j], zRep, nRep);
1221 j += nRep;
1222 i += nPattern-1;
1223 }
1224 }
drh2e6400b2007-05-08 15:46:18 +00001225 assert( j+nStr-i+1==nOut );
drh26b6d902007-03-17 13:27:54 +00001226 memcpy(&zOut[j], &zStr[i], nStr-i);
1227 j += nStr - i;
1228 assert( j<=nOut );
1229 zOut[j] = 0;
1230 sqlite3_result_text(context, (char*)zOut, j, sqlite3_free);
1231}
1232
drh309b3382007-03-17 17:52:42 +00001233/*
1234** Implementation of the TRIM(), LTRIM(), and RTRIM() functions.
1235** The userdata is 0x1 for left trim, 0x2 for right trim, 0x3 for both.
1236*/
1237static void trimFunc(
1238 sqlite3_context *context,
1239 int argc,
1240 sqlite3_value **argv
1241){
1242 const unsigned char *zIn; /* Input string */
1243 const unsigned char *zCharSet; /* Set of characters to trim */
1244 int nIn; /* Number of bytes in input */
drh7209c692008-04-27 18:40:11 +00001245 int flags; /* 1: trimleft 2: trimright 3: trim */
drhd1e3a612007-04-27 21:59:52 +00001246 int i; /* Loop counter */
drh1bd10f82008-12-10 21:19:56 +00001247 unsigned char *aLen = 0; /* Length of each character in zCharSet */
1248 unsigned char **azChar = 0; /* Individual characters in zCharSet */
drhd1e3a612007-04-27 21:59:52 +00001249 int nChar; /* Number of characters in zCharSet */
1250
drh309b3382007-03-17 17:52:42 +00001251 if( sqlite3_value_type(argv[0])==SQLITE_NULL ){
1252 return;
1253 }
1254 zIn = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001255 if( zIn==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001256 nIn = sqlite3_value_bytes(argv[0]);
1257 assert( zIn==sqlite3_value_text(argv[0]) );
drh309b3382007-03-17 17:52:42 +00001258 if( argc==1 ){
drhd1e3a612007-04-27 21:59:52 +00001259 static const unsigned char lenOne[] = { 1 };
danielk1977a4de4532008-09-02 15:44:08 +00001260 static unsigned char * const azOne[] = { (u8*)" " };
drhd1e3a612007-04-27 21:59:52 +00001261 nChar = 1;
drh8cff3822007-05-02 02:08:28 +00001262 aLen = (u8*)lenOne;
danielk1977bc67da42007-12-11 04:23:19 +00001263 azChar = (unsigned char **)azOne;
drhd1e3a612007-04-27 21:59:52 +00001264 zCharSet = 0;
drh7a521cf2007-04-25 18:23:52 +00001265 }else if( (zCharSet = sqlite3_value_text(argv[1]))==0 ){
drh309b3382007-03-17 17:52:42 +00001266 return;
drhd1e3a612007-04-27 21:59:52 +00001267 }else{
1268 const unsigned char *z;
1269 for(z=zCharSet, nChar=0; *z; nChar++){
drh4a919112007-05-15 11:55:09 +00001270 SQLITE_SKIP_UTF8(z);
drhd1e3a612007-04-27 21:59:52 +00001271 }
1272 if( nChar>0 ){
drhb1a6c3c2008-03-20 16:30:17 +00001273 azChar = contextMalloc(context, ((i64)nChar)*(sizeof(char*)+1));
drhd1e3a612007-04-27 21:59:52 +00001274 if( azChar==0 ){
1275 return;
1276 }
1277 aLen = (unsigned char*)&azChar[nChar];
1278 for(z=zCharSet, nChar=0; *z; nChar++){
danielk1977bc67da42007-12-11 04:23:19 +00001279 azChar[nChar] = (unsigned char *)z;
drh4a919112007-05-15 11:55:09 +00001280 SQLITE_SKIP_UTF8(z);
drh3abbd392008-12-10 23:04:13 +00001281 aLen[nChar] = (u8)(z - azChar[nChar]);
drhd1e3a612007-04-27 21:59:52 +00001282 }
1283 }
drh309b3382007-03-17 17:52:42 +00001284 }
drhd1e3a612007-04-27 21:59:52 +00001285 if( nChar>0 ){
shane1fc41292008-07-08 22:28:48 +00001286 flags = SQLITE_PTR_TO_INT(sqlite3_user_data(context));
drh309b3382007-03-17 17:52:42 +00001287 if( flags & 1 ){
drhd1e3a612007-04-27 21:59:52 +00001288 while( nIn>0 ){
drh1bd10f82008-12-10 21:19:56 +00001289 int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001290 for(i=0; i<nChar; i++){
1291 len = aLen[i];
drh27e62db2009-04-02 10:16:17 +00001292 if( len<=nIn && memcmp(zIn, azChar[i], len)==0 ) break;
drhd1e3a612007-04-27 21:59:52 +00001293 }
1294 if( i>=nChar ) break;
1295 zIn += len;
1296 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001297 }
1298 }
1299 if( flags & 2 ){
drhd1e3a612007-04-27 21:59:52 +00001300 while( nIn>0 ){
drh1bd10f82008-12-10 21:19:56 +00001301 int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001302 for(i=0; i<nChar; i++){
1303 len = aLen[i];
1304 if( len<=nIn && memcmp(&zIn[nIn-len],azChar[i],len)==0 ) break;
1305 }
1306 if( i>=nChar ) break;
1307 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001308 }
1309 }
drhd1e3a612007-04-27 21:59:52 +00001310 if( zCharSet ){
1311 sqlite3_free(azChar);
1312 }
drh309b3382007-03-17 17:52:42 +00001313 }
1314 sqlite3_result_text(context, (char*)zIn, nIn, SQLITE_TRANSIENT);
1315}
drh26b6d902007-03-17 13:27:54 +00001316
danielk1977a4de4532008-09-02 15:44:08 +00001317
drh2ba3ccc2009-12-08 02:06:08 +00001318/* IMP: R-25361-16150 This function is omitted from SQLite by default. It
1319** is only available if the SQLITE_SOUNDEX compile-time option is used
1320** when SQLite is built.
1321*/
drhd24cc422003-03-27 12:51:24 +00001322#ifdef SQLITE_SOUNDEX
1323/*
1324** Compute the soundex encoding of a word.
drh2ba3ccc2009-12-08 02:06:08 +00001325**
1326** IMP: R-59782-00072 The soundex(X) function returns a string that is the
1327** soundex encoding of the string X.
drhd24cc422003-03-27 12:51:24 +00001328*/
drh137c7282007-01-29 17:58:28 +00001329static void soundexFunc(
1330 sqlite3_context *context,
1331 int argc,
1332 sqlite3_value **argv
1333){
drhd24cc422003-03-27 12:51:24 +00001334 char zResult[8];
drh4c755c02004-08-08 20:22:17 +00001335 const u8 *zIn;
drhd24cc422003-03-27 12:51:24 +00001336 int i, j;
1337 static const unsigned char iCode[] = {
1338 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1339 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1340 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1341 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1342 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1343 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1344 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1345 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1346 };
1347 assert( argc==1 );
drh4c755c02004-08-08 20:22:17 +00001348 zIn = (u8*)sqlite3_value_text(argv[0]);
drhbdf67e02006-08-19 11:34:01 +00001349 if( zIn==0 ) zIn = (u8*)"";
drhdc86e2b2009-01-24 11:30:42 +00001350 for(i=0; zIn[i] && !sqlite3Isalpha(zIn[i]); i++){}
drhd24cc422003-03-27 12:51:24 +00001351 if( zIn[i] ){
drhbdf67e02006-08-19 11:34:01 +00001352 u8 prevcode = iCode[zIn[i]&0x7f];
danielk197778ca0e72009-01-20 16:53:39 +00001353 zResult[0] = sqlite3Toupper(zIn[i]);
drhd24cc422003-03-27 12:51:24 +00001354 for(j=1; j<4 && zIn[i]; i++){
1355 int code = iCode[zIn[i]&0x7f];
1356 if( code>0 ){
drhbdf67e02006-08-19 11:34:01 +00001357 if( code!=prevcode ){
1358 prevcode = code;
1359 zResult[j++] = code + '0';
1360 }
1361 }else{
1362 prevcode = 0;
drhd24cc422003-03-27 12:51:24 +00001363 }
1364 }
1365 while( j<4 ){
1366 zResult[j++] = '0';
1367 }
1368 zResult[j] = 0;
danielk1977d8123362004-06-12 09:25:12 +00001369 sqlite3_result_text(context, zResult, 4, SQLITE_TRANSIENT);
drhd24cc422003-03-27 12:51:24 +00001370 }else{
drh2ba3ccc2009-12-08 02:06:08 +00001371 /* IMP: R-64894-50321 The string "?000" is returned if the argument
1372 ** is NULL or contains no ASCII alphabetic characters. */
danielk1977d8123362004-06-12 09:25:12 +00001373 sqlite3_result_text(context, "?000", 4, SQLITE_STATIC);
drhd24cc422003-03-27 12:51:24 +00001374 }
1375}
drh2ba3ccc2009-12-08 02:06:08 +00001376#endif /* SQLITE_SOUNDEX */
drhd24cc422003-03-27 12:51:24 +00001377
drhfdb83b22006-06-17 14:12:47 +00001378#ifndef SQLITE_OMIT_LOAD_EXTENSION
1379/*
1380** A function that loads a shared-library extension then returns NULL.
1381*/
1382static void loadExt(sqlite3_context *context, int argc, sqlite3_value **argv){
danielk197765fd59f2006-06-24 11:51:33 +00001383 const char *zFile = (const char *)sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001384 const char *zProc;
drhfa4a4b92008-03-19 21:45:51 +00001385 sqlite3 *db = sqlite3_context_db_handle(context);
drhfdb83b22006-06-17 14:12:47 +00001386 char *zErrMsg = 0;
1387
1388 if( argc==2 ){
danielk197765fd59f2006-06-24 11:51:33 +00001389 zProc = (const char *)sqlite3_value_text(argv[1]);
drh7a521cf2007-04-25 18:23:52 +00001390 }else{
1391 zProc = 0;
drhfdb83b22006-06-17 14:12:47 +00001392 }
drh7a521cf2007-04-25 18:23:52 +00001393 if( zFile && sqlite3_load_extension(db, zFile, zProc, &zErrMsg) ){
drhfdb83b22006-06-17 14:12:47 +00001394 sqlite3_result_error(context, zErrMsg, -1);
1395 sqlite3_free(zErrMsg);
1396 }
1397}
1398#endif
1399
danielk197701427a62005-01-11 13:02:33 +00001400
drh0ac65892002-04-20 14:24:41 +00001401/*
drhd3a149e2002-02-24 17:12:53 +00001402** An instance of the following structure holds the context of a
drhdd5baa92002-02-27 19:50:59 +00001403** sum() or avg() aggregate computation.
1404*/
1405typedef struct SumCtx SumCtx;
1406struct SumCtx {
drh8c08e862006-02-11 17:34:00 +00001407 double rSum; /* Floating point sum */
1408 i64 iSum; /* Integer sum */
1409 i64 cnt; /* Number of elements summed */
1410 u8 overflow; /* True if integer overflow seen */
1411 u8 approx; /* True if non-integer value was input to the sum */
drhdd5baa92002-02-27 19:50:59 +00001412};
1413
1414/*
drha97fdd32006-01-12 22:17:50 +00001415** Routines used to compute the sum, average, and total.
1416**
1417** The SUM() function follows the (broken) SQL standard which means
1418** that it returns NULL if it sums over no inputs. TOTAL returns
1419** 0.0 in that case. In addition, TOTAL always returns a float where
1420** SUM might return an integer if it never encounters a floating point
drhc806d852006-05-11 13:25:39 +00001421** value. TOTAL never fails, but SUM might through an exception if
1422** it overflows an integer.
drhdd5baa92002-02-27 19:50:59 +00001423*/
danielk19770ae8b832004-05-25 12:05:56 +00001424static void sumStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drhdd5baa92002-02-27 19:50:59 +00001425 SumCtx *p;
drh3d1d95e2005-09-08 10:37:01 +00001426 int type;
drh3f219f42005-09-08 19:45:57 +00001427 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001428 UNUSED_PARAMETER(argc);
drh4f26d6c2004-05-26 23:25:30 +00001429 p = sqlite3_aggregate_context(context, sizeof(*p));
drh29d72102006-02-09 22:13:41 +00001430 type = sqlite3_value_numeric_type(argv[0]);
drh3d1d95e2005-09-08 10:37:01 +00001431 if( p && type!=SQLITE_NULL ){
drh739105c2002-05-29 23:22:23 +00001432 p->cnt++;
drh29d72102006-02-09 22:13:41 +00001433 if( type==SQLITE_INTEGER ){
drh8c08e862006-02-11 17:34:00 +00001434 i64 v = sqlite3_value_int64(argv[0]);
1435 p->rSum += v;
drh158b9cb2011-03-05 20:59:46 +00001436 if( (p->approx|p->overflow)==0 && sqlite3AddInt64(&p->iSum, v) ){
1437 p->overflow = 1;
drh29d72102006-02-09 22:13:41 +00001438 }
1439 }else{
drh8c08e862006-02-11 17:34:00 +00001440 p->rSum += sqlite3_value_double(argv[0]);
drh29d72102006-02-09 22:13:41 +00001441 p->approx = 1;
drh3f219f42005-09-08 19:45:57 +00001442 }
drh739105c2002-05-29 23:22:23 +00001443 }
drhdd5baa92002-02-27 19:50:59 +00001444}
danielk19770ae8b832004-05-25 12:05:56 +00001445static void sumFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001446 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001447 p = sqlite3_aggregate_context(context, 0);
drhc2bd9132005-09-08 20:37:43 +00001448 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001449 if( p->overflow ){
1450 sqlite3_result_error(context,"integer overflow",-1);
1451 }else if( p->approx ){
1452 sqlite3_result_double(context, p->rSum);
drhc2bd9132005-09-08 20:37:43 +00001453 }else{
drh8c08e862006-02-11 17:34:00 +00001454 sqlite3_result_int64(context, p->iSum);
drhc2bd9132005-09-08 20:37:43 +00001455 }
drh3d1d95e2005-09-08 10:37:01 +00001456 }
drhdd5baa92002-02-27 19:50:59 +00001457}
danielk19770ae8b832004-05-25 12:05:56 +00001458static void avgFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001459 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001460 p = sqlite3_aggregate_context(context, 0);
drh739105c2002-05-29 23:22:23 +00001461 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001462 sqlite3_result_double(context, p->rSum/(double)p->cnt);
drhdd5baa92002-02-27 19:50:59 +00001463 }
1464}
drha97fdd32006-01-12 22:17:50 +00001465static void totalFinalize(sqlite3_context *context){
1466 SumCtx *p;
1467 p = sqlite3_aggregate_context(context, 0);
shanefbd60f82009-02-04 03:59:25 +00001468 /* (double)0 In case of SQLITE_OMIT_FLOATING_POINT... */
1469 sqlite3_result_double(context, p ? p->rSum : (double)0);
drha97fdd32006-01-12 22:17:50 +00001470}
drhdd5baa92002-02-27 19:50:59 +00001471
1472/*
drh0bce8352002-02-28 00:41:10 +00001473** The following structure keeps track of state information for the
1474** count() aggregate function.
1475*/
1476typedef struct CountCtx CountCtx;
1477struct CountCtx {
drhfc6ad392006-02-09 13:38:19 +00001478 i64 n;
drh0bce8352002-02-28 00:41:10 +00001479};
drhdd5baa92002-02-27 19:50:59 +00001480
drh0bce8352002-02-28 00:41:10 +00001481/*
1482** Routines to implement the count() aggregate function.
1483*/
danielk19770ae8b832004-05-25 12:05:56 +00001484static void countStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drh0bce8352002-02-28 00:41:10 +00001485 CountCtx *p;
drh4f26d6c2004-05-26 23:25:30 +00001486 p = sqlite3_aggregate_context(context, sizeof(*p));
drh9c054832004-05-31 18:51:57 +00001487 if( (argc==0 || SQLITE_NULL!=sqlite3_value_type(argv[0])) && p ){
drh0bce8352002-02-28 00:41:10 +00001488 p->n++;
1489 }
drh2e79c3d2009-04-08 23:04:14 +00001490
drhd3264c72009-04-15 13:39:47 +00001491#ifndef SQLITE_OMIT_DEPRECATED
drh2e79c3d2009-04-08 23:04:14 +00001492 /* The sqlite3_aggregate_count() function is deprecated. But just to make
1493 ** sure it still operates correctly, verify that its count agrees with our
1494 ** internal count when using count(*) and when the total count can be
1495 ** expressed as a 32-bit integer. */
1496 assert( argc==1 || p==0 || p->n>0x7fffffff
1497 || p->n==sqlite3_aggregate_count(context) );
drhd3264c72009-04-15 13:39:47 +00001498#endif
drh0bce8352002-02-28 00:41:10 +00001499}
danielk19770ae8b832004-05-25 12:05:56 +00001500static void countFinalize(sqlite3_context *context){
drh0bce8352002-02-28 00:41:10 +00001501 CountCtx *p;
drhabfcea22005-09-06 20:36:48 +00001502 p = sqlite3_aggregate_context(context, 0);
drhfc6ad392006-02-09 13:38:19 +00001503 sqlite3_result_int64(context, p ? p->n : 0);
drh0bce8352002-02-28 00:41:10 +00001504}
1505
1506/*
drh0bce8352002-02-28 00:41:10 +00001507** Routines to implement min() and max() aggregate functions.
1508*/
danielk197762c14b32008-11-19 09:05:26 +00001509static void minmaxStep(
1510 sqlite3_context *context,
1511 int NotUsed,
1512 sqlite3_value **argv
1513){
danielk197788208052004-05-25 01:13:20 +00001514 Mem *pArg = (Mem *)argv[0];
drh9eb516c2004-07-18 20:52:32 +00001515 Mem *pBest;
danielk197762c14b32008-11-19 09:05:26 +00001516 UNUSED_PARAMETER(NotUsed);
drh9eb516c2004-07-18 20:52:32 +00001517
drh9eb516c2004-07-18 20:52:32 +00001518 pBest = (Mem *)sqlite3_aggregate_context(context, sizeof(*pBest));
danielk19773aeab9e2004-06-24 00:20:04 +00001519 if( !pBest ) return;
drh268380c2004-02-25 13:47:31 +00001520
drh94a6d992012-02-02 18:42:09 +00001521 if( sqlite3_value_type(argv[0])==SQLITE_NULL ){
1522 if( pBest->flags ) sqlite3SkipAccumulatorLoad(context);
1523 }else if( pBest->flags ){
drh9eb516c2004-07-18 20:52:32 +00001524 int max;
1525 int cmp;
danielk1977dc1bdc42004-06-11 10:51:27 +00001526 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk19777e18c252004-05-25 11:47:24 +00001527 /* This step function is used for both the min() and max() aggregates,
1528 ** the only difference between the two being that the sense of the
1529 ** comparison is inverted. For the max() aggregate, the
1530 ** sqlite3_user_data() function returns (void *)-1. For min() it
1531 ** returns (void *)db, where db is the sqlite3* database pointer.
1532 ** Therefore the next statement sets variable 'max' to 1 for the max()
1533 ** aggregate, or 0 for min().
1534 */
drh309b3382007-03-17 17:52:42 +00001535 max = sqlite3_user_data(context)!=0;
danielk1977dc1bdc42004-06-11 10:51:27 +00001536 cmp = sqlite3MemCompare(pBest, pArg, pColl);
danielk197788208052004-05-25 01:13:20 +00001537 if( (max && cmp<0) || (!max && cmp>0) ){
drhb21c8cd2007-08-21 19:33:56 +00001538 sqlite3VdbeMemCopy(pBest, pArg);
drh7a957892012-02-02 17:35:43 +00001539 }else{
1540 sqlite3SkipAccumulatorLoad(context);
danielk197788208052004-05-25 01:13:20 +00001541 }
drh268380c2004-02-25 13:47:31 +00001542 }else{
drh035e5632014-09-16 14:16:31 +00001543 pBest->db = sqlite3_context_db_handle(context);
drhb21c8cd2007-08-21 19:33:56 +00001544 sqlite3VdbeMemCopy(pBest, pArg);
drh0bce8352002-02-28 00:41:10 +00001545 }
1546}
danielk19770ae8b832004-05-25 12:05:56 +00001547static void minMaxFinalize(sqlite3_context *context){
danielk197788208052004-05-25 01:13:20 +00001548 sqlite3_value *pRes;
drhabfcea22005-09-06 20:36:48 +00001549 pRes = (sqlite3_value *)sqlite3_aggregate_context(context, 0);
1550 if( pRes ){
drh94a6d992012-02-02 18:42:09 +00001551 if( pRes->flags ){
drhabfcea22005-09-06 20:36:48 +00001552 sqlite3_result_value(context, pRes);
1553 }
1554 sqlite3VdbeMemRelease(pRes);
drh0bce8352002-02-28 00:41:10 +00001555 }
1556}
drhdd5baa92002-02-27 19:50:59 +00001557
drhb0689692007-11-01 17:38:30 +00001558/*
1559** group_concat(EXPR, ?SEPARATOR?)
1560*/
1561static void groupConcatStep(
1562 sqlite3_context *context,
1563 int argc,
1564 sqlite3_value **argv
1565){
1566 const char *zVal;
drhade86482007-11-28 22:36:40 +00001567 StrAccum *pAccum;
drhb0689692007-11-01 17:38:30 +00001568 const char *zSep;
drh07d31172009-02-02 21:57:05 +00001569 int nVal, nSep;
1570 assert( argc==1 || argc==2 );
1571 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drhade86482007-11-28 22:36:40 +00001572 pAccum = (StrAccum*)sqlite3_aggregate_context(context, sizeof(*pAccum));
1573
1574 if( pAccum ){
drhbb4957f2008-03-20 14:03:29 +00001575 sqlite3 *db = sqlite3_context_db_handle(context);
drhc0490572015-05-02 11:45:53 +00001576 int firstTerm = pAccum->mxAlloc==0;
drhbb4957f2008-03-20 14:03:29 +00001577 pAccum->mxAlloc = db->aLimit[SQLITE_LIMIT_LENGTH];
drh8bfd7192009-06-19 16:44:41 +00001578 if( !firstTerm ){
drh07d31172009-02-02 21:57:05 +00001579 if( argc==2 ){
1580 zSep = (char*)sqlite3_value_text(argv[1]);
1581 nSep = sqlite3_value_bytes(argv[1]);
drhb0689692007-11-01 17:38:30 +00001582 }else{
1583 zSep = ",";
drhade86482007-11-28 22:36:40 +00001584 nSep = 1;
drhb0689692007-11-01 17:38:30 +00001585 }
drha9ab4812013-12-11 11:00:44 +00001586 if( nSep ) sqlite3StrAccumAppend(pAccum, zSep, nSep);
drhb0689692007-11-01 17:38:30 +00001587 }
drh07d31172009-02-02 21:57:05 +00001588 zVal = (char*)sqlite3_value_text(argv[0]);
1589 nVal = sqlite3_value_bytes(argv[0]);
drh8009c9b2014-05-07 20:24:00 +00001590 if( zVal ) sqlite3StrAccumAppend(pAccum, zVal, nVal);
drhb0689692007-11-01 17:38:30 +00001591 }
1592}
1593static void groupConcatFinalize(sqlite3_context *context){
drhade86482007-11-28 22:36:40 +00001594 StrAccum *pAccum;
1595 pAccum = sqlite3_aggregate_context(context, 0);
1596 if( pAccum ){
drhb49bc862013-08-21 21:12:10 +00001597 if( pAccum->accError==STRACCUM_TOOBIG ){
drhade86482007-11-28 22:36:40 +00001598 sqlite3_result_error_toobig(context);
drhb49bc862013-08-21 21:12:10 +00001599 }else if( pAccum->accError==STRACCUM_NOMEM ){
drhade86482007-11-28 22:36:40 +00001600 sqlite3_result_error_nomem(context);
1601 }else{
1602 sqlite3_result_text(context, sqlite3StrAccumFinish(pAccum), -1,
1603 sqlite3_free);
1604 }
drhb0689692007-11-01 17:38:30 +00001605 }
1606}
drh4e5ffc52004-08-31 00:52:37 +00001607
drhd3a149e2002-02-24 17:12:53 +00001608/*
drha4741842010-04-25 20:58:37 +00001609** This routine does per-connection function registration. Most
1610** of the built-in functions above are part of the global function set.
1611** This routine only deals with those that are not global.
drhdc04c582002-02-24 01:55:15 +00001612*/
drh9bb575f2004-09-06 17:24:11 +00001613void sqlite3RegisterBuiltinFunctions(sqlite3 *db){
drha4741842010-04-25 20:58:37 +00001614 int rc = sqlite3_overload_function(db, "MATCH", 2);
1615 assert( rc==SQLITE_NOMEM || rc==SQLITE_OK );
1616 if( rc==SQLITE_NOMEM ){
1617 db->mallocFailed = 1;
danielk1977832a58a2007-06-22 15:21:15 +00001618 }
drh55ef4d92005-08-14 01:20:37 +00001619}
1620
1621/*
1622** Set the LIKEOPT flag on the 2-argument function with the given name.
1623*/
drh1bd10f82008-12-10 21:19:56 +00001624static void setLikeOptFlag(sqlite3 *db, const char *zName, u8 flagVal){
drh55ef4d92005-08-14 01:20:37 +00001625 FuncDef *pDef;
drhea678832008-12-10 19:26:22 +00001626 pDef = sqlite3FindFunction(db, zName, sqlite3Strlen30(zName),
1627 2, SQLITE_UTF8, 0);
drhd27135a2009-04-02 13:36:37 +00001628 if( ALWAYS(pDef) ){
drhd36e1042013-09-06 13:10:12 +00001629 pDef->funcFlags |= flagVal;
drh55ef4d92005-08-14 01:20:37 +00001630 }
1631}
1632
1633/*
1634** Register the built-in LIKE and GLOB functions. The caseSensitive
1635** parameter determines whether or not the LIKE operator is case
1636** sensitive. GLOB is always case sensitive.
1637*/
1638void sqlite3RegisterLikeFunctions(sqlite3 *db, int caseSensitive){
1639 struct compareInfo *pInfo;
1640 if( caseSensitive ){
1641 pInfo = (struct compareInfo*)&likeInfoAlt;
1642 }else{
1643 pInfo = (struct compareInfo*)&likeInfoNorm;
1644 }
drh901e9942010-12-15 18:54:37 +00001645 sqlite3CreateFunc(db, "like", 2, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0);
1646 sqlite3CreateFunc(db, "like", 3, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0);
1647 sqlite3CreateFunc(db, "glob", 2, SQLITE_UTF8,
dand2199f02010-08-27 17:48:52 +00001648 (struct compareInfo*)&globInfo, likeFunc, 0, 0, 0);
drhd64fe2f2005-08-28 17:00:23 +00001649 setLikeOptFlag(db, "glob", SQLITE_FUNC_LIKE | SQLITE_FUNC_CASE);
1650 setLikeOptFlag(db, "like",
1651 caseSensitive ? (SQLITE_FUNC_LIKE | SQLITE_FUNC_CASE) : SQLITE_FUNC_LIKE);
drh55ef4d92005-08-14 01:20:37 +00001652}
1653
1654/*
1655** pExpr points to an expression which implements a function. If
1656** it is appropriate to apply the LIKE optimization to that function
1657** then set aWc[0] through aWc[2] to the wildcard characters and
1658** return TRUE. If the function is not a LIKE-style function then
1659** return FALSE.
drh16897072015-03-07 00:57:37 +00001660**
1661** *pIsNocase is set to true if uppercase and lowercase are equivalent for
1662** the function (default for LIKE). If the function makes the distinction
1663** between uppercase and lowercase (as does GLOB) then *pIsNocase is set to
1664** false.
drh55ef4d92005-08-14 01:20:37 +00001665*/
drhd64fe2f2005-08-28 17:00:23 +00001666int sqlite3IsLikeFunction(sqlite3 *db, Expr *pExpr, int *pIsNocase, char *aWc){
drh55ef4d92005-08-14 01:20:37 +00001667 FuncDef *pDef;
danielk19776ab3a2e2009-02-19 14:39:25 +00001668 if( pExpr->op!=TK_FUNCTION
1669 || !pExpr->x.pList
1670 || pExpr->x.pList->nExpr!=2
1671 ){
drh55ef4d92005-08-14 01:20:37 +00001672 return 0;
1673 }
danielk19776ab3a2e2009-02-19 14:39:25 +00001674 assert( !ExprHasProperty(pExpr, EP_xIsSelect) );
drh33e619f2009-05-28 01:00:55 +00001675 pDef = sqlite3FindFunction(db, pExpr->u.zToken,
1676 sqlite3Strlen30(pExpr->u.zToken),
drhb7916a72009-05-27 10:31:29 +00001677 2, SQLITE_UTF8, 0);
drhd36e1042013-09-06 13:10:12 +00001678 if( NEVER(pDef==0) || (pDef->funcFlags & SQLITE_FUNC_LIKE)==0 ){
drh55ef4d92005-08-14 01:20:37 +00001679 return 0;
1680 }
1681
1682 /* The memcpy() statement assumes that the wildcard characters are
1683 ** the first three statements in the compareInfo structure. The
1684 ** asserts() that follow verify that assumption
1685 */
1686 memcpy(aWc, pDef->pUserData, 3);
1687 assert( (char*)&likeInfoAlt == (char*)&likeInfoAlt.matchAll );
1688 assert( &((char*)&likeInfoAlt)[1] == (char*)&likeInfoAlt.matchOne );
1689 assert( &((char*)&likeInfoAlt)[2] == (char*)&likeInfoAlt.matchSet );
drhd36e1042013-09-06 13:10:12 +00001690 *pIsNocase = (pDef->funcFlags & SQLITE_FUNC_CASE)==0;
drh55ef4d92005-08-14 01:20:37 +00001691 return 1;
drhdc04c582002-02-24 01:55:15 +00001692}
danielk19778c0a7912008-08-20 14:49:23 +00001693
drh70a8ca32008-08-21 18:49:27 +00001694/*
peter.d.reid60ec9142014-09-06 16:39:46 +00001695** All of the FuncDef structures in the aBuiltinFunc[] array above
drh777c5382008-08-21 20:21:34 +00001696** to the global function hash table. This occurs at start-time (as
1697** a consequence of calling sqlite3_initialize()).
1698**
1699** After this routine runs
drh70a8ca32008-08-21 18:49:27 +00001700*/
1701void sqlite3RegisterGlobalFunctions(void){
danielk197793ce7412008-09-01 19:14:02 +00001702 /*
1703 ** The following array holds FuncDef structures for all of the functions
1704 ** defined in this file.
1705 **
1706 ** The array cannot be constant since changes are made to the
1707 ** FuncDef.pHash elements at start-time. The elements of this array
1708 ** are read-only after initialization is complete.
1709 */
1710 static SQLITE_WSD FuncDef aBuiltinFunc[] = {
1711 FUNCTION(ltrim, 1, 1, 0, trimFunc ),
1712 FUNCTION(ltrim, 2, 1, 0, trimFunc ),
1713 FUNCTION(rtrim, 1, 2, 0, trimFunc ),
1714 FUNCTION(rtrim, 2, 2, 0, trimFunc ),
1715 FUNCTION(trim, 1, 3, 0, trimFunc ),
1716 FUNCTION(trim, 2, 3, 0, trimFunc ),
1717 FUNCTION(min, -1, 0, 1, minmaxFunc ),
1718 FUNCTION(min, 0, 0, 1, 0 ),
drh9588ad92014-09-15 14:46:02 +00001719 AGGREGATE2(min, 1, 0, 1, minmaxStep, minMaxFinalize,
1720 SQLITE_FUNC_MINMAX ),
danielk197793ce7412008-09-01 19:14:02 +00001721 FUNCTION(max, -1, 1, 1, minmaxFunc ),
1722 FUNCTION(max, 0, 1, 1, 0 ),
drh9588ad92014-09-15 14:46:02 +00001723 AGGREGATE2(max, 1, 1, 1, minmaxStep, minMaxFinalize,
1724 SQLITE_FUNC_MINMAX ),
drha748fdc2012-03-28 01:34:47 +00001725 FUNCTION2(typeof, 1, 0, 0, typeofFunc, SQLITE_FUNC_TYPEOF),
1726 FUNCTION2(length, 1, 0, 0, lengthFunc, SQLITE_FUNC_LENGTH),
drhd55e0722012-10-25 03:07:29 +00001727 FUNCTION(instr, 2, 0, 0, instrFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001728 FUNCTION(substr, 2, 0, 0, substrFunc ),
1729 FUNCTION(substr, 3, 0, 0, substrFunc ),
drha5c14162013-12-17 15:03:06 +00001730 FUNCTION(printf, -1, 0, 0, printfFunc ),
drhd495d8c2013-02-22 19:34:25 +00001731 FUNCTION(unicode, 1, 0, 0, unicodeFunc ),
1732 FUNCTION(char, -1, 0, 0, charFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001733 FUNCTION(abs, 1, 0, 0, absFunc ),
shanefbd60f82009-02-04 03:59:25 +00001734#ifndef SQLITE_OMIT_FLOATING_POINT
danielk197793ce7412008-09-01 19:14:02 +00001735 FUNCTION(round, 1, 0, 0, roundFunc ),
1736 FUNCTION(round, 2, 0, 0, roundFunc ),
shanefbd60f82009-02-04 03:59:25 +00001737#endif
danielk197793ce7412008-09-01 19:14:02 +00001738 FUNCTION(upper, 1, 0, 0, upperFunc ),
1739 FUNCTION(lower, 1, 0, 0, lowerFunc ),
1740 FUNCTION(coalesce, 1, 0, 0, 0 ),
danielk197793ce7412008-09-01 19:14:02 +00001741 FUNCTION(coalesce, 0, 0, 0, 0 ),
drhcca9f3d2013-09-06 15:23:29 +00001742 FUNCTION2(coalesce, -1, 0, 0, noopFunc, SQLITE_FUNC_COALESCE),
danielk197793ce7412008-09-01 19:14:02 +00001743 FUNCTION(hex, 1, 0, 0, hexFunc ),
drhcca9f3d2013-09-06 15:23:29 +00001744 FUNCTION2(ifnull, 2, 0, 0, noopFunc, SQLITE_FUNC_COALESCE),
1745 FUNCTION2(unlikely, 1, 0, 0, noopFunc, SQLITE_FUNC_UNLIKELY),
drhaae0f9e2013-09-11 11:38:58 +00001746 FUNCTION2(likelihood, 2, 0, 0, noopFunc, SQLITE_FUNC_UNLIKELY),
drh03202a92014-06-17 16:11:28 +00001747 FUNCTION2(likely, 1, 0, 0, noopFunc, SQLITE_FUNC_UNLIKELY),
drhb1fba282013-11-21 14:33:48 +00001748 VFUNCTION(random, 0, 0, 0, randomFunc ),
1749 VFUNCTION(randomblob, 1, 0, 0, randomBlob ),
danielk197793ce7412008-09-01 19:14:02 +00001750 FUNCTION(nullif, 2, 0, 1, nullifFunc ),
drh03bf26d2015-08-31 21:16:36 +00001751 DFUNCTION(sqlite_version, 0, 0, 0, versionFunc ),
1752 DFUNCTION(sqlite_source_id, 0, 0, 0, sourceidFunc ),
drh840561f2011-04-27 18:08:42 +00001753 FUNCTION(sqlite_log, 2, 0, 0, errlogFunc ),
drhf442e332014-09-10 19:01:14 +00001754#if SQLITE_USER_AUTHENTICATION
1755 FUNCTION(sqlite_crypt, 2, 0, 0, sqlite3CryptFunc ),
1756#endif
shanehdc97a8c2010-02-23 20:08:35 +00001757#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
drh03bf26d2015-08-31 21:16:36 +00001758 DFUNCTION(sqlite_compileoption_used,1, 0, 0, compileoptionusedFunc ),
1759 DFUNCTION(sqlite_compileoption_get, 1, 0, 0, compileoptiongetFunc ),
shanehdc97a8c2010-02-23 20:08:35 +00001760#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
danielk197793ce7412008-09-01 19:14:02 +00001761 FUNCTION(quote, 1, 0, 0, quoteFunc ),
drhb1fba282013-11-21 14:33:48 +00001762 VFUNCTION(last_insert_rowid, 0, 0, 0, last_insert_rowid),
1763 VFUNCTION(changes, 0, 0, 0, changes ),
1764 VFUNCTION(total_changes, 0, 0, 0, total_changes ),
danielk197793ce7412008-09-01 19:14:02 +00001765 FUNCTION(replace, 3, 0, 0, replaceFunc ),
1766 FUNCTION(zeroblob, 1, 0, 0, zeroblobFunc ),
1767 #ifdef SQLITE_SOUNDEX
1768 FUNCTION(soundex, 1, 0, 0, soundexFunc ),
1769 #endif
1770 #ifndef SQLITE_OMIT_LOAD_EXTENSION
drh03bf26d2015-08-31 21:16:36 +00001771 VFUNCTION(load_extension, 1, 0, 0, loadExt ),
1772 VFUNCTION(load_extension, 2, 0, 0, loadExt ),
danielk197793ce7412008-09-01 19:14:02 +00001773 #endif
1774 AGGREGATE(sum, 1, 0, 0, sumStep, sumFinalize ),
1775 AGGREGATE(total, 1, 0, 0, sumStep, totalFinalize ),
1776 AGGREGATE(avg, 1, 0, 0, sumStep, avgFinalize ),
drh9588ad92014-09-15 14:46:02 +00001777 AGGREGATE2(count, 0, 0, 0, countStep, countFinalize,
1778 SQLITE_FUNC_COUNT ),
danielk197793ce7412008-09-01 19:14:02 +00001779 AGGREGATE(count, 1, 0, 0, countStep, countFinalize ),
drh07d31172009-02-02 21:57:05 +00001780 AGGREGATE(group_concat, 1, 0, 0, groupConcatStep, groupConcatFinalize),
1781 AGGREGATE(group_concat, 2, 0, 0, groupConcatStep, groupConcatFinalize),
danielk197793ce7412008-09-01 19:14:02 +00001782
1783 LIKEFUNC(glob, 2, &globInfo, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
1784 #ifdef SQLITE_CASE_SENSITIVE_LIKE
1785 LIKEFUNC(like, 2, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
1786 LIKEFUNC(like, 3, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
1787 #else
1788 LIKEFUNC(like, 2, &likeInfoNorm, SQLITE_FUNC_LIKE),
1789 LIKEFUNC(like, 3, &likeInfoNorm, SQLITE_FUNC_LIKE),
1790 #endif
1791 };
1792
drh70a8ca32008-08-21 18:49:27 +00001793 int i;
danielk197793ce7412008-09-01 19:14:02 +00001794 FuncDefHash *pHash = &GLOBAL(FuncDefHash, sqlite3GlobalFunctions);
drh106cee52008-09-03 17:11:16 +00001795 FuncDef *aFunc = (FuncDef*)&GLOBAL(FuncDef, aBuiltinFunc);
danielk197793ce7412008-09-01 19:14:02 +00001796
drh70a8ca32008-08-21 18:49:27 +00001797 for(i=0; i<ArraySize(aBuiltinFunc); i++){
danielk197793ce7412008-09-01 19:14:02 +00001798 sqlite3FuncDefInsert(pHash, &aFunc[i]);
drh70a8ca32008-08-21 18:49:27 +00001799 }
drh777c5382008-08-21 20:21:34 +00001800 sqlite3RegisterDateTimeFunctions();
drh545f5872010-04-24 14:02:59 +00001801#ifndef SQLITE_OMIT_ALTERTABLE
1802 sqlite3AlterFunctions();
1803#endif
dan0106e372013-08-12 16:34:32 +00001804#if defined(SQLITE_ENABLE_STAT3) || defined(SQLITE_ENABLE_STAT4)
1805 sqlite3AnalyzeFunctions();
1806#endif
drh70a8ca32008-08-21 18:49:27 +00001807}