blob: 37f7cd6152fb560da376fad9fd8e261b5b283553 [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]);
dan895decf2016-12-30 14:15:56 +0000203 if( nNeedle>0 ){
204 if( typeHaystack==SQLITE_BLOB && typeNeedle==SQLITE_BLOB ){
205 zHaystack = sqlite3_value_blob(argv[0]);
206 zNeedle = sqlite3_value_blob(argv[1]);
207 isText = 0;
208 }else{
209 zHaystack = sqlite3_value_text(argv[0]);
210 zNeedle = sqlite3_value_text(argv[1]);
211 isText = 1;
212 if( zNeedle==0 ) return;
213 assert( zHaystack );
214 }
215 while( nNeedle<=nHaystack && memcmp(zHaystack, zNeedle, nNeedle)!=0 ){
216 N++;
217 do{
218 nHaystack--;
219 zHaystack++;
220 }while( isText && (zHaystack[0]&0xc0)==0x80 );
221 }
222 if( nNeedle>nHaystack ) N = 0;
drhd55e0722012-10-25 03:07:29 +0000223 }
drhd55e0722012-10-25 03:07:29 +0000224 sqlite3_result_int(context, N);
225}
226
227/*
drha5c14162013-12-17 15:03:06 +0000228** Implementation of the printf() function.
229*/
230static void printfFunc(
231 sqlite3_context *context,
232 int argc,
233 sqlite3_value **argv
234){
235 PrintfArguments x;
236 StrAccum str;
237 const char *zFormat;
238 int n;
drhc0490572015-05-02 11:45:53 +0000239 sqlite3 *db = sqlite3_context_db_handle(context);
drha5c14162013-12-17 15:03:06 +0000240
241 if( argc>=1 && (zFormat = (const char*)sqlite3_value_text(argv[0]))!=0 ){
242 x.nArg = argc-1;
243 x.nUsed = 0;
244 x.apArg = argv+1;
drhc0490572015-05-02 11:45:53 +0000245 sqlite3StrAccumInit(&str, db, 0, 0, db->aLimit[SQLITE_LIMIT_LENGTH]);
drh5f4a6862016-01-30 12:50:25 +0000246 str.printfFlags = SQLITE_PRINTF_SQLFUNC;
247 sqlite3XPrintf(&str, zFormat, &x);
drha5c14162013-12-17 15:03:06 +0000248 n = str.nChar;
249 sqlite3_result_text(context, sqlite3StrAccumFinish(&str), n,
250 SQLITE_DYNAMIC);
251 }
252}
253
254/*
drhf764e6f2007-05-15 01:13:47 +0000255** Implementation of the substr() function.
256**
257** substr(x,p1,p2) returns p2 characters of x[] beginning with p1.
258** p1 is 1-indexed. So substr(x,1,1) returns the first character
259** of x. If x is text, then we actually count UTF-8 characters.
260** If x is a blob, then we count bytes.
261**
262** If p1 is negative, then we begin abs(p1) from the end of x[].
shaneh779b8f12009-11-12 05:04:50 +0000263**
drhf7b54962013-05-28 12:11:54 +0000264** If p2 is negative, return the p2 characters preceding p1.
drh0bce8352002-02-28 00:41:10 +0000265*/
drhf9b596e2004-05-26 16:54:42 +0000266static void substrFunc(
267 sqlite3_context *context,
268 int argc,
269 sqlite3_value **argv
270){
drh2646da72005-12-09 20:02:05 +0000271 const unsigned char *z;
272 const unsigned char *z2;
drh023ae032007-05-08 12:12:16 +0000273 int len;
drhf764e6f2007-05-15 01:13:47 +0000274 int p0type;
drh023ae032007-05-08 12:12:16 +0000275 i64 p1, p2;
drh65595cd2009-02-02 16:32:55 +0000276 int negP2 = 0;
drhf9b596e2004-05-26 16:54:42 +0000277
drh64f31512007-10-12 19:11:55 +0000278 assert( argc==3 || argc==2 );
drh8198d252009-02-01 19:42:37 +0000279 if( sqlite3_value_type(argv[1])==SQLITE_NULL
280 || (argc==3 && sqlite3_value_type(argv[2])==SQLITE_NULL)
281 ){
282 return;
283 }
drhf764e6f2007-05-15 01:13:47 +0000284 p0type = sqlite3_value_type(argv[0]);
drh4adc4cb2009-11-11 20:53:31 +0000285 p1 = sqlite3_value_int(argv[1]);
drhf764e6f2007-05-15 01:13:47 +0000286 if( p0type==SQLITE_BLOB ){
287 len = sqlite3_value_bytes(argv[0]);
288 z = sqlite3_value_blob(argv[0]);
289 if( z==0 ) return;
drh1f0feef2007-05-15 13:27:07 +0000290 assert( len==sqlite3_value_bytes(argv[0]) );
drhf764e6f2007-05-15 01:13:47 +0000291 }else{
292 z = sqlite3_value_text(argv[0]);
293 if( z==0 ) return;
drh4a919112007-05-15 11:55:09 +0000294 len = 0;
drh4adc4cb2009-11-11 20:53:31 +0000295 if( p1<0 ){
296 for(z2=z; *z2; len++){
297 SQLITE_SKIP_UTF8(z2);
298 }
drh4a919112007-05-15 11:55:09 +0000299 }
drhf764e6f2007-05-15 01:13:47 +0000300 }
drh883ad042015-02-19 00:29:11 +0000301#ifdef SQLITE_SUBSTR_COMPATIBILITY
302 /* If SUBSTR_COMPATIBILITY is defined then substr(X,0,N) work the same as
303 ** as substr(X,1,N) - it returns the first N characters of X. This
304 ** is essentially a back-out of the bug-fix in check-in [5fc125d362df4b8]
305 ** from 2009-02-02 for compatibility of applications that exploited the
306 ** old buggy behavior. */
307 if( p1==0 ) p1 = 1; /* <rdar://problem/6778339> */
308#endif
drh64f31512007-10-12 19:11:55 +0000309 if( argc==3 ){
310 p2 = sqlite3_value_int(argv[2]);
drh65595cd2009-02-02 16:32:55 +0000311 if( p2<0 ){
312 p2 = -p2;
313 negP2 = 1;
314 }
drh64f31512007-10-12 19:11:55 +0000315 }else{
drhbb4957f2008-03-20 14:03:29 +0000316 p2 = sqlite3_context_db_handle(context)->aLimit[SQLITE_LIMIT_LENGTH];
drh64f31512007-10-12 19:11:55 +0000317 }
drh0bce8352002-02-28 00:41:10 +0000318 if( p1<0 ){
drh89425d52002-02-28 03:04:48 +0000319 p1 += len;
drh653bc752002-02-28 03:31:10 +0000320 if( p1<0 ){
321 p2 += p1;
drh65595cd2009-02-02 16:32:55 +0000322 if( p2<0 ) p2 = 0;
drh653bc752002-02-28 03:31:10 +0000323 p1 = 0;
324 }
drh0bce8352002-02-28 00:41:10 +0000325 }else if( p1>0 ){
326 p1--;
drh65595cd2009-02-02 16:32:55 +0000327 }else if( p2>0 ){
328 p2--;
drh0bce8352002-02-28 00:41:10 +0000329 }
drh65595cd2009-02-02 16:32:55 +0000330 if( negP2 ){
331 p1 -= p2;
drh4e79c592009-02-01 19:23:32 +0000332 if( p1<0 ){
333 p2 += p1;
334 p1 = 0;
335 }
336 }
drh65595cd2009-02-02 16:32:55 +0000337 assert( p1>=0 && p2>=0 );
drhf764e6f2007-05-15 01:13:47 +0000338 if( p0type!=SQLITE_BLOB ){
drh4a919112007-05-15 11:55:09 +0000339 while( *z && p1 ){
340 SQLITE_SKIP_UTF8(z);
341 p1--;
drhf764e6f2007-05-15 01:13:47 +0000342 }
drh4a919112007-05-15 11:55:09 +0000343 for(z2=z; *z2 && p2; p2--){
344 SQLITE_SKIP_UTF8(z2);
drhf764e6f2007-05-15 01:13:47 +0000345 }
drhbbf483f2014-09-09 20:30:24 +0000346 sqlite3_result_text64(context, (char*)z, z2-z, SQLITE_TRANSIENT,
347 SQLITE_UTF8);
drhf764e6f2007-05-15 01:13:47 +0000348 }else{
drh4adc4cb2009-11-11 20:53:31 +0000349 if( p1+p2>len ){
350 p2 = len-p1;
351 if( p2<0 ) p2 = 0;
352 }
drhbbf483f2014-09-09 20:30:24 +0000353 sqlite3_result_blob64(context, (char*)&z[p1], (u64)p2, SQLITE_TRANSIENT);
drh0bce8352002-02-28 00:41:10 +0000354 }
drh0bce8352002-02-28 00:41:10 +0000355}
356
357/*
358** Implementation of the round() function
359*/
shanefbd60f82009-02-04 03:59:25 +0000360#ifndef SQLITE_OMIT_FLOATING_POINT
danielk19770ae8b832004-05-25 12:05:56 +0000361static void roundFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
danielk197751ad0ec2004-05-24 12:39:02 +0000362 int n = 0;
drh0bce8352002-02-28 00:41:10 +0000363 double r;
drh50d654d2009-06-03 01:24:54 +0000364 char *zBuf;
drh0bce8352002-02-28 00:41:10 +0000365 assert( argc==1 || argc==2 );
danielk197751ad0ec2004-05-24 12:39:02 +0000366 if( argc==2 ){
drh9c054832004-05-31 18:51:57 +0000367 if( SQLITE_NULL==sqlite3_value_type(argv[1]) ) return;
danielk197751ad0ec2004-05-24 12:39:02 +0000368 n = sqlite3_value_int(argv[1]);
369 if( n>30 ) n = 30;
370 if( n<0 ) n = 0;
371 }
drhd589a922006-03-02 03:02:48 +0000372 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drh4f26d6c2004-05-26 23:25:30 +0000373 r = sqlite3_value_double(argv[0]);
shaneh147e1762010-02-17 04:19:27 +0000374 /* If Y==0 and X will fit in a 64-bit int,
375 ** handle the rounding directly,
376 ** otherwise use printf.
377 */
378 if( n==0 && r>=0 && r<LARGEST_INT64-1 ){
379 r = (double)((sqlite_int64)(r+0.5));
380 }else if( n==0 && r<0 && (-r)<LARGEST_INT64-1 ){
381 r = -(double)((sqlite_int64)((-r)+0.5));
drh50d654d2009-06-03 01:24:54 +0000382 }else{
shaneh147e1762010-02-17 04:19:27 +0000383 zBuf = sqlite3_mprintf("%.*f",n,r);
384 if( zBuf==0 ){
385 sqlite3_result_error_nomem(context);
386 return;
387 }
drh9339da12010-09-30 00:50:49 +0000388 sqlite3AtoF(zBuf, &r, sqlite3Strlen30(zBuf), SQLITE_UTF8);
drh50d654d2009-06-03 01:24:54 +0000389 sqlite3_free(zBuf);
drh50d654d2009-06-03 01:24:54 +0000390 }
shaneh147e1762010-02-17 04:19:27 +0000391 sqlite3_result_double(context, r);
drh0bce8352002-02-28 00:41:10 +0000392}
shanefbd60f82009-02-04 03:59:25 +0000393#endif
drhdc04c582002-02-24 01:55:15 +0000394
danielk197726783a52007-08-29 14:06:22 +0000395/*
drhf3cdcdc2015-04-29 16:50:28 +0000396** Allocate nByte bytes of space using sqlite3Malloc(). If the
danielk197726783a52007-08-29 14:06:22 +0000397** allocation fails, call sqlite3_result_error_nomem() to notify
drh27e62db2009-04-02 10:16:17 +0000398** the database handle that malloc() has failed and return NULL.
399** If nByte is larger than the maximum string or blob length, then
400** raise an SQLITE_TOOBIG exception and return NULL.
danielk197726783a52007-08-29 14:06:22 +0000401*/
drhb1a6c3c2008-03-20 16:30:17 +0000402static void *contextMalloc(sqlite3_context *context, i64 nByte){
drhbb4957f2008-03-20 14:03:29 +0000403 char *z;
drh27e62db2009-04-02 10:16:17 +0000404 sqlite3 *db = sqlite3_context_db_handle(context);
drhef31c6a2009-04-02 09:07:12 +0000405 assert( nByte>0 );
drh27e62db2009-04-02 10:16:17 +0000406 testcase( nByte==db->aLimit[SQLITE_LIMIT_LENGTH] );
407 testcase( nByte==db->aLimit[SQLITE_LIMIT_LENGTH]+1 );
408 if( nByte>db->aLimit[SQLITE_LIMIT_LENGTH] ){
drhbb4957f2008-03-20 14:03:29 +0000409 sqlite3_result_error_toobig(context);
410 z = 0;
411 }else{
drhda4ca9d2014-09-09 17:27:35 +0000412 z = sqlite3Malloc(nByte);
drhef31c6a2009-04-02 09:07:12 +0000413 if( !z ){
drhbb4957f2008-03-20 14:03:29 +0000414 sqlite3_result_error_nomem(context);
415 }
danielk1977a1644fd2007-08-29 12:31:25 +0000416 }
417 return z;
418}
419
drhdc04c582002-02-24 01:55:15 +0000420/*
421** Implementation of the upper() and lower() SQL functions.
422*/
danielk19770ae8b832004-05-25 12:05:56 +0000423static void upperFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drh7a521cf2007-04-25 18:23:52 +0000424 char *z1;
425 const char *z2;
drh9310ef22007-04-27 17:16:20 +0000426 int i, n;
drh1d34fde2009-02-03 15:50:33 +0000427 UNUSED_PARAMETER(argc);
drh7a521cf2007-04-25 18:23:52 +0000428 z2 = (char*)sqlite3_value_text(argv[0]);
drh1f0feef2007-05-15 13:27:07 +0000429 n = sqlite3_value_bytes(argv[0]);
430 /* Verify that the call to _bytes() does not invalidate the _text() pointer */
431 assert( z2==(char*)sqlite3_value_text(argv[0]) );
drh7a521cf2007-04-25 18:23:52 +0000432 if( z2 ){
drhb1a6c3c2008-03-20 16:30:17 +0000433 z1 = contextMalloc(context, ((i64)n)+1);
drh7a521cf2007-04-25 18:23:52 +0000434 if( z1 ){
drhdf901d32011-10-13 18:00:11 +0000435 for(i=0; i<n; i++){
436 z1[i] = (char)sqlite3Toupper(z2[i]);
drh7a521cf2007-04-25 18:23:52 +0000437 }
drhdf901d32011-10-13 18:00:11 +0000438 sqlite3_result_text(context, z1, n, sqlite3_free);
drh7a521cf2007-04-25 18:23:52 +0000439 }
drhdc04c582002-02-24 01:55:15 +0000440 }
441}
danielk19770ae8b832004-05-25 12:05:56 +0000442static void lowerFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drhdf901d32011-10-13 18:00:11 +0000443 char *z1;
drh7a521cf2007-04-25 18:23:52 +0000444 const char *z2;
drh9310ef22007-04-27 17:16:20 +0000445 int i, n;
drh1d34fde2009-02-03 15:50:33 +0000446 UNUSED_PARAMETER(argc);
drh7a521cf2007-04-25 18:23:52 +0000447 z2 = (char*)sqlite3_value_text(argv[0]);
drh1f0feef2007-05-15 13:27:07 +0000448 n = sqlite3_value_bytes(argv[0]);
449 /* Verify that the call to _bytes() does not invalidate the _text() pointer */
450 assert( z2==(char*)sqlite3_value_text(argv[0]) );
drh7a521cf2007-04-25 18:23:52 +0000451 if( z2 ){
drhb1a6c3c2008-03-20 16:30:17 +0000452 z1 = contextMalloc(context, ((i64)n)+1);
drh7a521cf2007-04-25 18:23:52 +0000453 if( z1 ){
drhdf901d32011-10-13 18:00:11 +0000454 for(i=0; i<n; i++){
455 z1[i] = sqlite3Tolower(z2[i]);
drh7a521cf2007-04-25 18:23:52 +0000456 }
drhdf901d32011-10-13 18:00:11 +0000457 sqlite3_result_text(context, z1, n, sqlite3_free);
drh7a521cf2007-04-25 18:23:52 +0000458 }
drhdc04c582002-02-24 01:55:15 +0000459 }
460}
461
drhae6bb952009-11-11 00:24:31 +0000462/*
drhcca9f3d2013-09-06 15:23:29 +0000463** Some functions like COALESCE() and IFNULL() and UNLIKELY() are implemented
464** as VDBE code so that unused argument values do not have to be computed.
465** However, we still need some kind of function implementation for this
466** routines in the function table. The noopFunc macro provides this.
467** noopFunc will never be called so it doesn't matter what the implementation
468** is. We might as well use the "version()" function as a substitute.
drhae6bb952009-11-11 00:24:31 +0000469*/
drhcca9f3d2013-09-06 15:23:29 +0000470#define noopFunc versionFunc /* Substitute function - never called */
drh3212e182002-02-28 00:46:26 +0000471
472/*
drhf9ffac92002-03-02 19:00:31 +0000473** Implementation of random(). Return a random integer.
474*/
drhf9b596e2004-05-26 16:54:42 +0000475static void randomFunc(
476 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000477 int NotUsed,
478 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000479){
drh52fc8492006-02-23 21:43:55 +0000480 sqlite_int64 r;
danielk197762c14b32008-11-19 09:05:26 +0000481 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drh2fa18682008-03-19 14:15:34 +0000482 sqlite3_randomness(sizeof(r), &r);
drh3034e3d2009-04-02 14:05:21 +0000483 if( r<0 ){
484 /* We need to prevent a random number of 0x8000000000000000
485 ** (or -9223372036854775808) since when you do abs() of that
486 ** number of you get the same value back again. To do this
487 ** in a way that is testable, mask the sign bit off of negative
488 ** values, resulting in a positive value. Then take the
489 ** 2s complement of that positive value. The end result can
490 ** therefore be no less than -9223372036854775807.
491 */
drhaf8001b2012-02-11 19:53:24 +0000492 r = -(r & LARGEST_INT64);
drh3034e3d2009-04-02 14:05:21 +0000493 }
drh52fc8492006-02-23 21:43:55 +0000494 sqlite3_result_int64(context, r);
drhf9ffac92002-03-02 19:00:31 +0000495}
496
497/*
drh137c7282007-01-29 17:58:28 +0000498** Implementation of randomblob(N). Return a random blob
499** that is N bytes long.
drh63cf66f2007-01-29 15:50:05 +0000500*/
drh137c7282007-01-29 17:58:28 +0000501static void randomBlob(
drh63cf66f2007-01-29 15:50:05 +0000502 sqlite3_context *context,
503 int argc,
504 sqlite3_value **argv
505){
drh137c7282007-01-29 17:58:28 +0000506 int n;
507 unsigned char *p;
drh63cf66f2007-01-29 15:50:05 +0000508 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000509 UNUSED_PARAMETER(argc);
drh63cf66f2007-01-29 15:50:05 +0000510 n = sqlite3_value_int(argv[0]);
drh023ae032007-05-08 12:12:16 +0000511 if( n<1 ){
512 n = 1;
513 }
danielk1977a1644fd2007-08-29 12:31:25 +0000514 p = contextMalloc(context, n);
drh02d85832007-05-07 19:31:15 +0000515 if( p ){
drh2fa18682008-03-19 14:15:34 +0000516 sqlite3_randomness(n, p);
drh17435752007-08-16 04:30:38 +0000517 sqlite3_result_blob(context, (char*)p, n, sqlite3_free);
drh02d85832007-05-07 19:31:15 +0000518 }
drh63cf66f2007-01-29 15:50:05 +0000519}
520
521/*
drh6ed41ad2002-04-06 14:10:47 +0000522** Implementation of the last_insert_rowid() SQL function. The return
danielk197724b03fd2004-05-10 10:34:34 +0000523** value is the same as the sqlite3_last_insert_rowid() API function.
drh6ed41ad2002-04-06 14:10:47 +0000524*/
danielk197751ad0ec2004-05-24 12:39:02 +0000525static void last_insert_rowid(
danielk19770ae8b832004-05-25 12:05:56 +0000526 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000527 int NotUsed,
528 sqlite3_value **NotUsed2
danielk197751ad0ec2004-05-24 12:39:02 +0000529){
drhfa4a4b92008-03-19 21:45:51 +0000530 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000531 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000532 /* IMP: R-51513-12026 The last_insert_rowid() SQL function is a
533 ** wrapper around the sqlite3_last_insert_rowid() C/C++ interface
534 ** function. */
drhf9b596e2004-05-26 16:54:42 +0000535 sqlite3_result_int64(context, sqlite3_last_insert_rowid(db));
drh6ed41ad2002-04-06 14:10:47 +0000536}
537
rdcf146a772004-02-25 22:51:06 +0000538/*
drhab2f1f92010-01-11 18:26:42 +0000539** Implementation of the changes() SQL function.
540**
541** IMP: R-62073-11209 The changes() SQL function is a wrapper
542** around the sqlite3_changes() C/C++ function and hence follows the same
543** rules for counting changes.
rdcf146a772004-02-25 22:51:06 +0000544*/
danielk1977b28af712004-06-21 06:50:26 +0000545static void changes(
drhf9b596e2004-05-26 16:54:42 +0000546 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000547 int NotUsed,
548 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000549){
drhfa4a4b92008-03-19 21:45:51 +0000550 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000551 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhf4479502004-05-27 03:12:53 +0000552 sqlite3_result_int(context, sqlite3_changes(db));
rdcb0c374f2004-02-20 22:53:38 +0000553}
rdcf146a772004-02-25 22:51:06 +0000554
555/*
danielk1977b28af712004-06-21 06:50:26 +0000556** Implementation of the total_changes() SQL function. The return value is
557** the same as the sqlite3_total_changes() API function.
rdcf146a772004-02-25 22:51:06 +0000558*/
danielk1977b28af712004-06-21 06:50:26 +0000559static void total_changes(
560 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000561 int NotUsed,
562 sqlite3_value **NotUsed2
danielk197751ad0ec2004-05-24 12:39:02 +0000563){
drhfa4a4b92008-03-19 21:45:51 +0000564 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000565 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000566 /* IMP: R-52756-41993 This function is a wrapper around the
567 ** sqlite3_total_changes() C/C++ interface. */
danielk1977b28af712004-06-21 06:50:26 +0000568 sqlite3_result_int(context, sqlite3_total_changes(db));
rdcb0c374f2004-02-20 22:53:38 +0000569}
570
drh6ed41ad2002-04-06 14:10:47 +0000571/*
drh4e5ffc52004-08-31 00:52:37 +0000572** A structure defining how to do GLOB-style comparisons.
danielk1977d02eb1f2004-06-06 09:44:03 +0000573*/
drh4e5ffc52004-08-31 00:52:37 +0000574struct compareInfo {
drh07e83472016-01-11 03:48:18 +0000575 u8 matchAll; /* "*" or "%" */
576 u8 matchOne; /* "?" or "_" */
577 u8 matchSet; /* "[" or 0 */
578 u8 noCase; /* true to ignore case differences */
danielk1977d02eb1f2004-06-06 09:44:03 +0000579};
drh55ef4d92005-08-14 01:20:37 +0000580
drhb9175ae2007-12-07 18:39:04 +0000581/*
582** For LIKE and GLOB matching on EBCDIC machines, assume that every
drhb0870482015-06-17 13:20:54 +0000583** character is exactly one byte in size. Also, provde the Utf8Read()
584** macro for fast reading of the next character in the common case where
585** the next character is ASCII.
drhb9175ae2007-12-07 18:39:04 +0000586*/
587#if defined(SQLITE_EBCDIC)
drh88b33222014-09-25 03:51:37 +0000588# define sqlite3Utf8Read(A) (*((*A)++))
drhb0870482015-06-17 13:20:54 +0000589# define Utf8Read(A) (*(A++))
drhb9175ae2007-12-07 18:39:04 +0000590#else
drhb0870482015-06-17 13:20:54 +0000591# define Utf8Read(A) (A[0]<0x80?*(A++):sqlite3Utf8Read(&A))
drhb9175ae2007-12-07 18:39:04 +0000592#endif
593
drh4e5ffc52004-08-31 00:52:37 +0000594static const struct compareInfo globInfo = { '*', '?', '[', 0 };
drh55ef4d92005-08-14 01:20:37 +0000595/* The correct SQL-92 behavior is for the LIKE operator to ignore
596** case. Thus 'a' LIKE 'A' would be true. */
597static const struct compareInfo likeInfoNorm = { '%', '_', 0, 1 };
598/* If SQLITE_CASE_SENSITIVE_LIKE is defined, then the LIKE operator
599** is case sensitive causing 'a' LIKE 'A' to be false */
600static const struct compareInfo likeInfoAlt = { '%', '_', 0, 0 };
danielk1977d02eb1f2004-06-06 09:44:03 +0000601
602/*
drh698a01c2016-12-01 18:49:40 +0000603** Possible error returns from patternMatch()
604*/
605#define SQLITE_MATCH 0
606#define SQLITE_NOMATCH 1
607#define SQLITE_NOWILDCARDMATCH 2
608
609/*
610** Compare two UTF-8 strings for equality where the first string is
611** a GLOB or LIKE expression. Return values:
612**
613** SQLITE_MATCH: Match
614** SQLITE_NOMATCH: No match
615** SQLITE_NOWILDCARDMATCH: No match in spite of having * or % wildcards.
drh0ac65892002-04-20 14:24:41 +0000616**
drh4e5ffc52004-08-31 00:52:37 +0000617** Globbing rules:
drh0ac65892002-04-20 14:24:41 +0000618**
drh4e5ffc52004-08-31 00:52:37 +0000619** '*' Matches any sequence of zero or more characters.
danielk1977d02eb1f2004-06-06 09:44:03 +0000620**
drh4e5ffc52004-08-31 00:52:37 +0000621** '?' Matches exactly one character.
622**
623** [...] Matches one character from the enclosed list of
624** characters.
625**
626** [^...] Matches one character not in the enclosed list.
627**
628** With the [...] and [^...] matching, a ']' character can be included
629** in the list by making it the first character after '[' or '^'. A
630** range of characters can be specified using '-'. Example:
631** "[a-z]" matches any single lower-case letter. To match a '-', make
632** it the last character in the list.
633**
drh9fdfdc82014-09-25 11:08:57 +0000634** Like matching rules:
635**
636** '%' Matches any sequence of zero or more characters
637**
638*** '_' Matches any one character
639**
640** Ec Where E is the "esc" character and c is any other
641** character, including '%', '_', and esc, match exactly c.
642**
drhb0870482015-06-17 13:20:54 +0000643** The comments within this routine usually assume glob matching.
drh9fdfdc82014-09-25 11:08:57 +0000644**
drh4e5ffc52004-08-31 00:52:37 +0000645** This routine is usually quick, but can be N**2 in the worst case.
drh0ac65892002-04-20 14:24:41 +0000646*/
danielk19777c6303c2004-11-17 16:41:29 +0000647static int patternCompare(
drh4e5ffc52004-08-31 00:52:37 +0000648 const u8 *zPattern, /* The glob pattern */
649 const u8 *zString, /* The string to compare against the glob */
danielk19777c6303c2004-11-17 16:41:29 +0000650 const struct compareInfo *pInfo, /* Information about how to do the compare */
drh698a01c2016-12-01 18:49:40 +0000651 u32 matchOther /* The escape char (LIKE) or '[' (GLOB) */
danielk197751ad0ec2004-05-24 12:39:02 +0000652){
drh9fdfdc82014-09-25 11:08:57 +0000653 u32 c, c2; /* Next pattern and input string chars */
654 u32 matchOne = pInfo->matchOne; /* "?" or "_" */
655 u32 matchAll = pInfo->matchAll; /* "*" or "%" */
drh9fdfdc82014-09-25 11:08:57 +0000656 u8 noCase = pInfo->noCase; /* True if uppercase==lowercase */
657 const u8 *zEscaped = 0; /* One past the last escaped input char */
drh88b33222014-09-25 03:51:37 +0000658
drhb0870482015-06-17 13:20:54 +0000659 while( (c = Utf8Read(zPattern))!=0 ){
drh9fdfdc82014-09-25 11:08:57 +0000660 if( c==matchAll ){ /* Match "*" */
661 /* Skip over multiple "*" characters in the pattern. If there
662 ** are also "?" characters, skip those as well, but consume a
663 ** single character of the input string for each "?" skipped */
drhb0870482015-06-17 13:20:54 +0000664 while( (c=Utf8Read(zPattern)) == matchAll || c == matchOne ){
drh42610962012-09-17 18:56:32 +0000665 if( c==matchOne && sqlite3Utf8Read(&zString)==0 ){
drh698a01c2016-12-01 18:49:40 +0000666 return SQLITE_NOWILDCARDMATCH;
drh4e5ffc52004-08-31 00:52:37 +0000667 }
danielk1977ad7dd422004-06-06 12:41:49 +0000668 }
drh66150952007-07-23 19:12:41 +0000669 if( c==0 ){
drh698a01c2016-12-01 18:49:40 +0000670 return SQLITE_MATCH; /* "*" at the end of the pattern matches */
drh88b33222014-09-25 03:51:37 +0000671 }else if( c==matchOther ){
drh07e83472016-01-11 03:48:18 +0000672 if( pInfo->matchSet==0 ){
drh88b33222014-09-25 03:51:37 +0000673 c = sqlite3Utf8Read(&zPattern);
drh698a01c2016-12-01 18:49:40 +0000674 if( c==0 ) return SQLITE_NOWILDCARDMATCH;
drh88b33222014-09-25 03:51:37 +0000675 }else{
drh9fdfdc82014-09-25 11:08:57 +0000676 /* "[...]" immediately follows the "*". We have to do a slow
677 ** recursive search in this case, but it is an unusual case. */
drh88b33222014-09-25 03:51:37 +0000678 assert( matchOther<0x80 ); /* '[' is a single-byte character */
drh698a01c2016-12-01 18:49:40 +0000679 while( *zString ){
680 int bMatch = patternCompare(&zPattern[-1],zString,pInfo,matchOther);
681 if( bMatch!=SQLITE_NOMATCH ) return bMatch;
drh88b33222014-09-25 03:51:37 +0000682 SQLITE_SKIP_UTF8(zString);
683 }
drh698a01c2016-12-01 18:49:40 +0000684 return SQLITE_NOWILDCARDMATCH;
drh66150952007-07-23 19:12:41 +0000685 }
drh66150952007-07-23 19:12:41 +0000686 }
drh9fdfdc82014-09-25 11:08:57 +0000687
688 /* At this point variable c contains the first character of the
689 ** pattern string past the "*". Search in the input string for the
dan1a4a7372016-12-01 17:34:59 +0000690 ** first matching character and recursively continue the match from
drh9fdfdc82014-09-25 11:08:57 +0000691 ** that point.
692 **
693 ** For a case-insensitive search, set variable cx to be the same as
694 ** c but in the other case and search the input string for either
695 ** c or cx.
696 */
697 if( c<=0x80 ){
698 u32 cx;
drh698a01c2016-12-01 18:49:40 +0000699 int bMatch;
drh9fdfdc82014-09-25 11:08:57 +0000700 if( noCase ){
701 cx = sqlite3Toupper(c);
702 c = sqlite3Tolower(c);
drh66150952007-07-23 19:12:41 +0000703 }else{
drh9fdfdc82014-09-25 11:08:57 +0000704 cx = c;
drh4e5ffc52004-08-31 00:52:37 +0000705 }
drh9fdfdc82014-09-25 11:08:57 +0000706 while( (c2 = *(zString++))!=0 ){
707 if( c2!=c && c2!=cx ) continue;
drh698a01c2016-12-01 18:49:40 +0000708 bMatch = patternCompare(zPattern,zString,pInfo,matchOther);
709 if( bMatch!=SQLITE_NOMATCH ) return bMatch;
drh9fdfdc82014-09-25 11:08:57 +0000710 }
711 }else{
drh698a01c2016-12-01 18:49:40 +0000712 int bMatch;
drhb0870482015-06-17 13:20:54 +0000713 while( (c2 = Utf8Read(zString))!=0 ){
drh9fdfdc82014-09-25 11:08:57 +0000714 if( c2!=c ) continue;
drh698a01c2016-12-01 18:49:40 +0000715 bMatch = patternCompare(zPattern,zString,pInfo,matchOther);
716 if( bMatch!=SQLITE_NOMATCH ) return bMatch;
drh9fdfdc82014-09-25 11:08:57 +0000717 }
drh66150952007-07-23 19:12:41 +0000718 }
drh698a01c2016-12-01 18:49:40 +0000719 return SQLITE_NOWILDCARDMATCH;
drh88b33222014-09-25 03:51:37 +0000720 }
drh88b33222014-09-25 03:51:37 +0000721 if( c==matchOther ){
drh07e83472016-01-11 03:48:18 +0000722 if( pInfo->matchSet==0 ){
drh88b33222014-09-25 03:51:37 +0000723 c = sqlite3Utf8Read(&zPattern);
drh698a01c2016-12-01 18:49:40 +0000724 if( c==0 ) return SQLITE_NOMATCH;
drh9fdfdc82014-09-25 11:08:57 +0000725 zEscaped = zPattern;
drh88b33222014-09-25 03:51:37 +0000726 }else{
727 u32 prior_c = 0;
drh9fdfdc82014-09-25 11:08:57 +0000728 int seen = 0;
729 int invert = 0;
drh88b33222014-09-25 03:51:37 +0000730 c = sqlite3Utf8Read(&zString);
drh698a01c2016-12-01 18:49:40 +0000731 if( c==0 ) return SQLITE_NOMATCH;
drh88b33222014-09-25 03:51:37 +0000732 c2 = sqlite3Utf8Read(&zPattern);
733 if( c2=='^' ){
734 invert = 1;
735 c2 = sqlite3Utf8Read(&zPattern);
736 }
737 if( c2==']' ){
738 if( c==']' ) seen = 1;
739 c2 = sqlite3Utf8Read(&zPattern);
740 }
741 while( c2 && c2!=']' ){
742 if( c2=='-' && zPattern[0]!=']' && zPattern[0]!=0 && prior_c>0 ){
743 c2 = sqlite3Utf8Read(&zPattern);
744 if( c>=prior_c && c<=c2 ) seen = 1;
745 prior_c = 0;
746 }else{
747 if( c==c2 ){
748 seen = 1;
749 }
750 prior_c = c2;
751 }
752 c2 = sqlite3Utf8Read(&zPattern);
753 }
754 if( c2==0 || (seen ^ invert)==0 ){
drh698a01c2016-12-01 18:49:40 +0000755 return SQLITE_NOMATCH;
drh88b33222014-09-25 03:51:37 +0000756 }
757 continue;
758 }
759 }
drhb0870482015-06-17 13:20:54 +0000760 c2 = Utf8Read(zString);
drh88b33222014-09-25 03:51:37 +0000761 if( c==c2 ) continue;
drhc80937a2016-06-06 01:48:14 +0000762 if( noCase && sqlite3Tolower(c)==sqlite3Tolower(c2) && c<0x80 && c2<0x80 ){
drh9fdfdc82014-09-25 11:08:57 +0000763 continue;
764 }
765 if( c==matchOne && zPattern!=zEscaped && c2!=0 ) continue;
drh698a01c2016-12-01 18:49:40 +0000766 return SQLITE_NOMATCH;
danielk197751ad0ec2004-05-24 12:39:02 +0000767 }
drh698a01c2016-12-01 18:49:40 +0000768 return *zString==0 ? SQLITE_MATCH : SQLITE_NOMATCH;
drh0ac65892002-04-20 14:24:41 +0000769}
drh4e5ffc52004-08-31 00:52:37 +0000770
drh55ef4d92005-08-14 01:20:37 +0000771/*
drh698a01c2016-12-01 18:49:40 +0000772** The sqlite3_strglob() interface. Return 0 on a match (like strcmp()) and
773** non-zero if there is no match.
drh56282a52013-04-10 16:13:38 +0000774*/
775int sqlite3_strglob(const char *zGlobPattern, const char *zString){
drh698a01c2016-12-01 18:49:40 +0000776 return patternCompare((u8*)zGlobPattern, (u8*)zString, &globInfo, '[');
drh56282a52013-04-10 16:13:38 +0000777}
778
779/*
drh698a01c2016-12-01 18:49:40 +0000780** The sqlite3_strlike() interface. Return 0 on a match and non-zero for
781** a miss - like strcmp().
drh8b4a94a2015-11-24 21:23:59 +0000782*/
783int sqlite3_strlike(const char *zPattern, const char *zStr, unsigned int esc){
drh698a01c2016-12-01 18:49:40 +0000784 return patternCompare((u8*)zPattern, (u8*)zStr, &likeInfoNorm, esc);
drh8b4a94a2015-11-24 21:23:59 +0000785}
786
787/*
drh55ef4d92005-08-14 01:20:37 +0000788** Count the number of times that the LIKE operator (or GLOB which is
789** just a variation of LIKE) gets called. This is used for testing
790** only.
791*/
792#ifdef SQLITE_TEST
793int sqlite3_like_count = 0;
794#endif
795
danielk19773f6b0872004-06-17 05:36:44 +0000796
797/*
798** Implementation of the like() SQL function. This function implements
799** the build-in LIKE operator. The first argument to the function is the
800** pattern and the second argument is the string. So, the SQL statements:
801**
802** A LIKE B
803**
804** is implemented as like(B,A).
805**
drh55ef4d92005-08-14 01:20:37 +0000806** This same function (with a different compareInfo structure) computes
807** the GLOB operator.
danielk19773f6b0872004-06-17 05:36:44 +0000808*/
809static void likeFunc(
810 sqlite3_context *context,
811 int argc,
812 sqlite3_value **argv
813){
drhbeb818d2007-05-08 15:34:47 +0000814 const unsigned char *zA, *zB;
drh07e83472016-01-11 03:48:18 +0000815 u32 escape;
drh27e62db2009-04-02 10:16:17 +0000816 int nPat;
drhbb4957f2008-03-20 14:03:29 +0000817 sqlite3 *db = sqlite3_context_db_handle(context);
drh07e83472016-01-11 03:48:18 +0000818 struct compareInfo *pInfo = sqlite3_user_data(context);
drhbeb818d2007-05-08 15:34:47 +0000819
drh41d2e662015-12-01 21:23:07 +0000820#ifdef SQLITE_LIKE_DOESNT_MATCH_BLOBS
821 if( sqlite3_value_type(argv[0])==SQLITE_BLOB
822 || sqlite3_value_type(argv[1])==SQLITE_BLOB
823 ){
824#ifdef SQLITE_TEST
825 sqlite3_like_count++;
826#endif
827 sqlite3_result_int(context, 0);
828 return;
829 }
830#endif
drh1f0feef2007-05-15 13:27:07 +0000831 zB = sqlite3_value_text(argv[0]);
832 zA = sqlite3_value_text(argv[1]);
833
drhbeb818d2007-05-08 15:34:47 +0000834 /* Limit the length of the LIKE or GLOB pattern to avoid problems
835 ** of deep recursion and N*N behavior in patternCompare().
836 */
drh27e62db2009-04-02 10:16:17 +0000837 nPat = sqlite3_value_bytes(argv[0]);
838 testcase( nPat==db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH] );
839 testcase( nPat==db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH]+1 );
840 if( nPat > db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH] ){
drhbeb818d2007-05-08 15:34:47 +0000841 sqlite3_result_error(context, "LIKE or GLOB pattern too complex", -1);
842 return;
843 }
drh1f0feef2007-05-15 13:27:07 +0000844 assert( zB==sqlite3_value_text(argv[0]) ); /* Encoding did not change */
drhbeb818d2007-05-08 15:34:47 +0000845
danielk19777c6303c2004-11-17 16:41:29 +0000846 if( argc==3 ){
847 /* The escape character string must consist of a single UTF-8 character.
848 ** Otherwise, return an error.
849 */
850 const unsigned char *zEsc = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +0000851 if( zEsc==0 ) return;
drhee858132007-05-08 20:37:38 +0000852 if( sqlite3Utf8CharLen((char*)zEsc, -1)!=1 ){
danielk19777c6303c2004-11-17 16:41:29 +0000853 sqlite3_result_error(context,
854 "ESCAPE expression must be a single character", -1);
855 return;
856 }
drh42610962012-09-17 18:56:32 +0000857 escape = sqlite3Utf8Read(&zEsc);
drh07e83472016-01-11 03:48:18 +0000858 }else{
859 escape = pInfo->matchSet;
danielk19777c6303c2004-11-17 16:41:29 +0000860 }
danielk19773f6b0872004-06-17 05:36:44 +0000861 if( zA && zB ){
drh55ef4d92005-08-14 01:20:37 +0000862#ifdef SQLITE_TEST
863 sqlite3_like_count++;
864#endif
drh698a01c2016-12-01 18:49:40 +0000865 sqlite3_result_int(context, patternCompare(zB, zA, pInfo, escape)==SQLITE_MATCH);
danielk197751ad0ec2004-05-24 12:39:02 +0000866 }
drh8912d102002-05-26 21:34:58 +0000867}
868
869/*
870** Implementation of the NULLIF(x,y) function. The result is the first
871** argument if the arguments are different. The result is NULL if the
872** arguments are equal to each other.
873*/
drhf9b596e2004-05-26 16:54:42 +0000874static void nullifFunc(
875 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000876 int NotUsed,
drhf9b596e2004-05-26 16:54:42 +0000877 sqlite3_value **argv
878){
danielk1977dc1bdc42004-06-11 10:51:27 +0000879 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk197762c14b32008-11-19 09:05:26 +0000880 UNUSED_PARAMETER(NotUsed);
danielk1977dc1bdc42004-06-11 10:51:27 +0000881 if( sqlite3MemCompare(argv[0], argv[1], pColl)!=0 ){
drhf4479502004-05-27 03:12:53 +0000882 sqlite3_result_value(context, argv[0]);
drh8912d102002-05-26 21:34:58 +0000883 }
drh0ac65892002-04-20 14:24:41 +0000884}
885
drh647cb0e2002-11-04 19:32:25 +0000886/*
drh47baebc2009-08-14 16:01:24 +0000887** Implementation of the sqlite_version() function. The result is the version
drh647cb0e2002-11-04 19:32:25 +0000888** of the SQLite library that is running.
889*/
drhf9b596e2004-05-26 16:54:42 +0000890static void versionFunc(
891 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000892 int NotUsed,
893 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000894){
danielk197762c14b32008-11-19 09:05:26 +0000895 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000896 /* IMP: R-48699-48617 This function is an SQL wrapper around the
897 ** sqlite3_libversion() C-interface. */
898 sqlite3_result_text(context, sqlite3_libversion(), -1, SQLITE_STATIC);
drh647cb0e2002-11-04 19:32:25 +0000899}
900
drh47baebc2009-08-14 16:01:24 +0000901/*
902** Implementation of the sqlite_source_id() function. The result is a string
903** that identifies the particular version of the source code used to build
904** SQLite.
905*/
906static void sourceidFunc(
907 sqlite3_context *context,
908 int NotUsed,
909 sqlite3_value **NotUsed2
910){
911 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000912 /* IMP: R-24470-31136 This function is an SQL wrapper around the
913 ** sqlite3_sourceid() C interface. */
914 sqlite3_result_text(context, sqlite3_sourceid(), -1, SQLITE_STATIC);
drh47baebc2009-08-14 16:01:24 +0000915}
916
shanehbdea6d12010-02-23 04:19:54 +0000917/*
drh3ca84ef2011-04-25 18:03:10 +0000918** Implementation of the sqlite_log() function. This is a wrapper around
919** sqlite3_log(). The return value is NULL. The function exists purely for
920** its side-effects.
921*/
drh840561f2011-04-27 18:08:42 +0000922static void errlogFunc(
drh3ca84ef2011-04-25 18:03:10 +0000923 sqlite3_context *context,
924 int argc,
925 sqlite3_value **argv
926){
927 UNUSED_PARAMETER(argc);
928 UNUSED_PARAMETER(context);
929 sqlite3_log(sqlite3_value_int(argv[0]), "%s", sqlite3_value_text(argv[1]));
930}
931
932/*
shanehdc97a8c2010-02-23 20:08:35 +0000933** Implementation of the sqlite_compileoption_used() function.
934** The result is an integer that identifies if the compiler option
935** was used to build SQLite.
shanehbdea6d12010-02-23 04:19:54 +0000936*/
shanehdc97a8c2010-02-23 20:08:35 +0000937#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
938static void compileoptionusedFunc(
shanehbdea6d12010-02-23 04:19:54 +0000939 sqlite3_context *context,
shanehdc97a8c2010-02-23 20:08:35 +0000940 int argc,
941 sqlite3_value **argv
shanehbdea6d12010-02-23 04:19:54 +0000942){
shanehdc97a8c2010-02-23 20:08:35 +0000943 const char *zOptName;
944 assert( argc==1 );
945 UNUSED_PARAMETER(argc);
drha3e414c2010-08-03 18:06:25 +0000946 /* IMP: R-39564-36305 The sqlite_compileoption_used() SQL
947 ** function is a wrapper around the sqlite3_compileoption_used() C/C++
948 ** function.
949 */
drh264a2d42010-02-25 15:28:41 +0000950 if( (zOptName = (const char*)sqlite3_value_text(argv[0]))!=0 ){
shanehdc97a8c2010-02-23 20:08:35 +0000951 sqlite3_result_int(context, sqlite3_compileoption_used(zOptName));
shanehdc97a8c2010-02-23 20:08:35 +0000952 }
shanehbdea6d12010-02-23 04:19:54 +0000953}
shanehdc97a8c2010-02-23 20:08:35 +0000954#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
955
956/*
957** Implementation of the sqlite_compileoption_get() function.
958** The result is a string that identifies the compiler options
959** used to build SQLite.
960*/
961#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
962static void compileoptiongetFunc(
963 sqlite3_context *context,
964 int argc,
965 sqlite3_value **argv
966){
967 int n;
968 assert( argc==1 );
969 UNUSED_PARAMETER(argc);
drha3e414c2010-08-03 18:06:25 +0000970 /* IMP: R-04922-24076 The sqlite_compileoption_get() SQL function
971 ** is a wrapper around the sqlite3_compileoption_get() C/C++ function.
972 */
shanehdc97a8c2010-02-23 20:08:35 +0000973 n = sqlite3_value_int(argv[0]);
974 sqlite3_result_text(context, sqlite3_compileoption_get(n), -1, SQLITE_STATIC);
975}
976#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
shanehbdea6d12010-02-23 04:19:54 +0000977
drh137c7282007-01-29 17:58:28 +0000978/* Array for converting from half-bytes (nybbles) into ASCII hex
979** digits. */
980static const char hexdigits[] = {
981 '0', '1', '2', '3', '4', '5', '6', '7',
982 '8', '9', 'A', 'B', 'C', 'D', 'E', 'F'
983};
danielk1977d641d642004-11-18 15:44:29 +0000984
drh47394702003-08-20 01:03:33 +0000985/*
drh47394702003-08-20 01:03:33 +0000986** Implementation of the QUOTE() function. This function takes a single
987** argument. If the argument is numeric, the return value is the same as
988** the argument. If the argument is NULL, the return value is the string
989** "NULL". Otherwise, the argument is enclosed in single quotes with
990** single-quote escapes.
991*/
danielk19770ae8b832004-05-25 12:05:56 +0000992static void quoteFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drha0df4cc2009-02-02 17:29:59 +0000993 assert( argc==1 );
drh1d34fde2009-02-03 15:50:33 +0000994 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +0000995 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +0000996 case SQLITE_FLOAT: {
drh72b3fbc2012-06-19 03:11:25 +0000997 double r1, r2;
drh72b3fbc2012-06-19 03:11:25 +0000998 char zBuf[50];
mistachkin2b434a72012-06-19 04:36:48 +0000999 r1 = sqlite3_value_double(argv[0]);
drh72b3fbc2012-06-19 03:11:25 +00001000 sqlite3_snprintf(sizeof(zBuf), zBuf, "%!.15g", r1);
1001 sqlite3AtoF(zBuf, &r2, 20, SQLITE_UTF8);
1002 if( r1!=r2 ){
1003 sqlite3_snprintf(sizeof(zBuf), zBuf, "%!.20e", r1);
1004 }
1005 sqlite3_result_text(context, zBuf, -1, SQLITE_TRANSIENT);
1006 break;
1007 }
1008 case SQLITE_INTEGER: {
drhf4479502004-05-27 03:12:53 +00001009 sqlite3_result_value(context, argv[0]);
drhf9b596e2004-05-26 16:54:42 +00001010 break;
1011 }
danielk19773f41e972004-06-08 00:39:01 +00001012 case SQLITE_BLOB: {
danielk19773f41e972004-06-08 00:39:01 +00001013 char *zText = 0;
danielk19773f41e972004-06-08 00:39:01 +00001014 char const *zBlob = sqlite3_value_blob(argv[0]);
drh1f0feef2007-05-15 13:27:07 +00001015 int nBlob = sqlite3_value_bytes(argv[0]);
1016 assert( zBlob==sqlite3_value_blob(argv[0]) ); /* No encoding change */
drhb1a6c3c2008-03-20 16:30:17 +00001017 zText = (char *)contextMalloc(context, (2*(i64)nBlob)+4);
danielk1977a1644fd2007-08-29 12:31:25 +00001018 if( zText ){
danielk19773f41e972004-06-08 00:39:01 +00001019 int i;
1020 for(i=0; i<nBlob; i++){
1021 zText[(i*2)+2] = hexdigits[(zBlob[i]>>4)&0x0F];
1022 zText[(i*2)+3] = hexdigits[(zBlob[i])&0x0F];
1023 }
1024 zText[(nBlob*2)+2] = '\'';
1025 zText[(nBlob*2)+3] = '\0';
1026 zText[0] = 'X';
1027 zText[1] = '\'';
danielk1977d8123362004-06-12 09:25:12 +00001028 sqlite3_result_text(context, zText, -1, SQLITE_TRANSIENT);
drh17435752007-08-16 04:30:38 +00001029 sqlite3_free(zText);
danielk19773f41e972004-06-08 00:39:01 +00001030 }
1031 break;
1032 }
drh9c054832004-05-31 18:51:57 +00001033 case SQLITE_TEXT: {
drh023ae032007-05-08 12:12:16 +00001034 int i,j;
1035 u64 n;
drh2646da72005-12-09 20:02:05 +00001036 const unsigned char *zArg = sqlite3_value_text(argv[0]);
drhf9b596e2004-05-26 16:54:42 +00001037 char *z;
1038
drh7a521cf2007-04-25 18:23:52 +00001039 if( zArg==0 ) return;
drh023ae032007-05-08 12:12:16 +00001040 for(i=0, n=0; zArg[i]; i++){ if( zArg[i]=='\'' ) n++; }
drhb1a6c3c2008-03-20 16:30:17 +00001041 z = contextMalloc(context, ((i64)i)+((i64)n)+3);
danielk1977a1644fd2007-08-29 12:31:25 +00001042 if( z ){
1043 z[0] = '\'';
1044 for(i=0, j=1; zArg[i]; i++){
1045 z[j++] = zArg[i];
1046 if( zArg[i]=='\'' ){
1047 z[j++] = '\'';
1048 }
drhf9b596e2004-05-26 16:54:42 +00001049 }
danielk1977a1644fd2007-08-29 12:31:25 +00001050 z[j++] = '\'';
1051 z[j] = 0;
1052 sqlite3_result_text(context, z, j, sqlite3_free);
drhf9b596e2004-05-26 16:54:42 +00001053 }
drha0df4cc2009-02-02 17:29:59 +00001054 break;
1055 }
1056 default: {
1057 assert( sqlite3_value_type(argv[0])==SQLITE_NULL );
1058 sqlite3_result_text(context, "NULL", 4, SQLITE_STATIC);
1059 break;
drhf9b596e2004-05-26 16:54:42 +00001060 }
drh47394702003-08-20 01:03:33 +00001061 }
1062}
1063
drh137c7282007-01-29 17:58:28 +00001064/*
drhd495d8c2013-02-22 19:34:25 +00001065** The unicode() function. Return the integer unicode code-point value
1066** for the first character of the input string.
1067*/
1068static void unicodeFunc(
1069 sqlite3_context *context,
1070 int argc,
1071 sqlite3_value **argv
1072){
1073 const unsigned char *z = sqlite3_value_text(argv[0]);
drh1d59d032013-03-01 23:40:26 +00001074 (void)argc;
drhd495d8c2013-02-22 19:34:25 +00001075 if( z && z[0] ) sqlite3_result_int(context, sqlite3Utf8Read(&z));
1076}
1077
1078/*
1079** The char() function takes zero or more arguments, each of which is
1080** an integer. It constructs a string where each character of the string
1081** is the unicode character for the corresponding integer argument.
1082*/
1083static void charFunc(
1084 sqlite3_context *context,
1085 int argc,
1086 sqlite3_value **argv
1087){
1088 unsigned char *z, *zOut;
1089 int i;
drhf3cdcdc2015-04-29 16:50:28 +00001090 zOut = z = sqlite3_malloc64( argc*4+1 );
drhd495d8c2013-02-22 19:34:25 +00001091 if( z==0 ){
1092 sqlite3_result_error_nomem(context);
1093 return;
1094 }
1095 for(i=0; i<argc; i++){
mistachkinc9545442013-02-26 05:42:30 +00001096 sqlite3_int64 x;
drhd495d8c2013-02-22 19:34:25 +00001097 unsigned c;
1098 x = sqlite3_value_int64(argv[i]);
1099 if( x<0 || x>0x10ffff ) x = 0xfffd;
1100 c = (unsigned)(x & 0x1fffff);
drhfe7a5d12013-03-07 14:00:04 +00001101 if( c<0x00080 ){
1102 *zOut++ = (u8)(c&0xFF);
1103 }else if( c<0x00800 ){
1104 *zOut++ = 0xC0 + (u8)((c>>6)&0x1F);
1105 *zOut++ = 0x80 + (u8)(c & 0x3F);
1106 }else if( c<0x10000 ){
1107 *zOut++ = 0xE0 + (u8)((c>>12)&0x0F);
1108 *zOut++ = 0x80 + (u8)((c>>6) & 0x3F);
1109 *zOut++ = 0x80 + (u8)(c & 0x3F);
drhd495d8c2013-02-22 19:34:25 +00001110 }else{
drhfe7a5d12013-03-07 14:00:04 +00001111 *zOut++ = 0xF0 + (u8)((c>>18) & 0x07);
1112 *zOut++ = 0x80 + (u8)((c>>12) & 0x3F);
1113 *zOut++ = 0x80 + (u8)((c>>6) & 0x3F);
1114 *zOut++ = 0x80 + (u8)(c & 0x3F);
1115 } \
drhd495d8c2013-02-22 19:34:25 +00001116 }
drhbbf483f2014-09-09 20:30:24 +00001117 sqlite3_result_text64(context, (char*)z, zOut-z, sqlite3_free, SQLITE_UTF8);
drhd495d8c2013-02-22 19:34:25 +00001118}
1119
1120/*
drh137c7282007-01-29 17:58:28 +00001121** The hex() function. Interpret the argument as a blob. Return
1122** a hexadecimal rendering as text.
1123*/
1124static void hexFunc(
1125 sqlite3_context *context,
1126 int argc,
1127 sqlite3_value **argv
1128){
1129 int i, n;
1130 const unsigned char *pBlob;
1131 char *zHex, *z;
1132 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001133 UNUSED_PARAMETER(argc);
drh1f0feef2007-05-15 13:27:07 +00001134 pBlob = sqlite3_value_blob(argv[0]);
drh137c7282007-01-29 17:58:28 +00001135 n = sqlite3_value_bytes(argv[0]);
drh1f0feef2007-05-15 13:27:07 +00001136 assert( pBlob==sqlite3_value_blob(argv[0]) ); /* No encoding change */
drhb1a6c3c2008-03-20 16:30:17 +00001137 z = zHex = contextMalloc(context, ((i64)n)*2 + 1);
danielk1977a1644fd2007-08-29 12:31:25 +00001138 if( zHex ){
1139 for(i=0; i<n; i++, pBlob++){
1140 unsigned char c = *pBlob;
1141 *(z++) = hexdigits[(c>>4)&0xf];
1142 *(z++) = hexdigits[c&0xf];
1143 }
1144 *z = 0;
1145 sqlite3_result_text(context, zHex, n*2, sqlite3_free);
drh137c7282007-01-29 17:58:28 +00001146 }
drh137c7282007-01-29 17:58:28 +00001147}
1148
drh26b6d902007-03-17 13:27:54 +00001149/*
drh8cff3822007-05-02 02:08:28 +00001150** The zeroblob(N) function returns a zero-filled blob of size N bytes.
1151*/
1152static void zeroblobFunc(
1153 sqlite3_context *context,
1154 int argc,
1155 sqlite3_value **argv
1156){
drh98640a32007-06-07 19:08:32 +00001157 i64 n;
dana4d5ae82015-07-24 16:24:37 +00001158 int rc;
drh8cff3822007-05-02 02:08:28 +00001159 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001160 UNUSED_PARAMETER(argc);
drh98640a32007-06-07 19:08:32 +00001161 n = sqlite3_value_int64(argv[0]);
dana4d5ae82015-07-24 16:24:37 +00001162 if( n<0 ) n = 0;
1163 rc = sqlite3_result_zeroblob64(context, n); /* IMP: R-00293-64994 */
1164 if( rc ){
1165 sqlite3_result_error_code(context, rc);
drh98640a32007-06-07 19:08:32 +00001166 }
drh8cff3822007-05-02 02:08:28 +00001167}
1168
1169/*
drh26b6d902007-03-17 13:27:54 +00001170** The replace() function. Three arguments are all strings: call
1171** them A, B, and C. The result is also a string which is derived
drhf7b54962013-05-28 12:11:54 +00001172** from A by replacing every occurrence of B with C. The match
drh26b6d902007-03-17 13:27:54 +00001173** must be exact. Collating sequences are not used.
1174*/
1175static void replaceFunc(
1176 sqlite3_context *context,
1177 int argc,
1178 sqlite3_value **argv
1179){
1180 const unsigned char *zStr; /* The input string A */
1181 const unsigned char *zPattern; /* The pattern string B */
1182 const unsigned char *zRep; /* The replacement string C */
1183 unsigned char *zOut; /* The output */
1184 int nStr; /* Size of zStr */
1185 int nPattern; /* Size of zPattern */
1186 int nRep; /* Size of zRep */
drh2e6400b2007-05-08 15:46:18 +00001187 i64 nOut; /* Maximum size of zOut */
drh26b6d902007-03-17 13:27:54 +00001188 int loopLimit; /* Last zStr[] that might match zPattern[] */
1189 int i, j; /* Loop counters */
1190
1191 assert( argc==3 );
danielk1977f3d3c272008-11-19 16:52:44 +00001192 UNUSED_PARAMETER(argc);
drh26b6d902007-03-17 13:27:54 +00001193 zStr = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001194 if( zStr==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001195 nStr = sqlite3_value_bytes(argv[0]);
1196 assert( zStr==sqlite3_value_text(argv[0]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001197 zPattern = sqlite3_value_text(argv[1]);
drha605fe82009-02-01 18:08:40 +00001198 if( zPattern==0 ){
drh23336062009-02-03 13:19:12 +00001199 assert( sqlite3_value_type(argv[1])==SQLITE_NULL
1200 || sqlite3_context_db_handle(context)->mallocFailed );
drha605fe82009-02-01 18:08:40 +00001201 return;
1202 }
1203 if( zPattern[0]==0 ){
1204 assert( sqlite3_value_type(argv[1])!=SQLITE_NULL );
1205 sqlite3_result_value(context, argv[0]);
1206 return;
1207 }
drh1f0feef2007-05-15 13:27:07 +00001208 nPattern = sqlite3_value_bytes(argv[1]);
1209 assert( zPattern==sqlite3_value_text(argv[1]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001210 zRep = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +00001211 if( zRep==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001212 nRep = sqlite3_value_bytes(argv[2]);
1213 assert( zRep==sqlite3_value_text(argv[2]) );
drh2e6400b2007-05-08 15:46:18 +00001214 nOut = nStr + 1;
1215 assert( nOut<SQLITE_MAX_LENGTH );
drhb1a6c3c2008-03-20 16:30:17 +00001216 zOut = contextMalloc(context, (i64)nOut);
drh2e6400b2007-05-08 15:46:18 +00001217 if( zOut==0 ){
1218 return;
drh26b6d902007-03-17 13:27:54 +00001219 }
drh26b6d902007-03-17 13:27:54 +00001220 loopLimit = nStr - nPattern;
1221 for(i=j=0; i<=loopLimit; i++){
1222 if( zStr[i]!=zPattern[0] || memcmp(&zStr[i], zPattern, nPattern) ){
1223 zOut[j++] = zStr[i];
1224 }else{
drh4a50aac2007-08-23 02:47:53 +00001225 u8 *zOld;
drhbb4957f2008-03-20 14:03:29 +00001226 sqlite3 *db = sqlite3_context_db_handle(context);
drh2e6400b2007-05-08 15:46:18 +00001227 nOut += nRep - nPattern;
drh27e62db2009-04-02 10:16:17 +00001228 testcase( nOut-1==db->aLimit[SQLITE_LIMIT_LENGTH] );
1229 testcase( nOut-2==db->aLimit[SQLITE_LIMIT_LENGTH] );
1230 if( nOut-1>db->aLimit[SQLITE_LIMIT_LENGTH] ){
drha0206bc2007-05-08 15:15:02 +00001231 sqlite3_result_error_toobig(context);
drhb9755982010-07-24 16:34:37 +00001232 sqlite3_free(zOut);
danielk197717374e82007-05-08 14:39:04 +00001233 return;
1234 }
drh4a50aac2007-08-23 02:47:53 +00001235 zOld = zOut;
drhf3cdcdc2015-04-29 16:50:28 +00001236 zOut = sqlite3_realloc64(zOut, (int)nOut);
drh2e6400b2007-05-08 15:46:18 +00001237 if( zOut==0 ){
danielk1977a1644fd2007-08-29 12:31:25 +00001238 sqlite3_result_error_nomem(context);
drhb9755982010-07-24 16:34:37 +00001239 sqlite3_free(zOld);
drh2e6400b2007-05-08 15:46:18 +00001240 return;
1241 }
drh26b6d902007-03-17 13:27:54 +00001242 memcpy(&zOut[j], zRep, nRep);
1243 j += nRep;
1244 i += nPattern-1;
1245 }
1246 }
drh2e6400b2007-05-08 15:46:18 +00001247 assert( j+nStr-i+1==nOut );
drh26b6d902007-03-17 13:27:54 +00001248 memcpy(&zOut[j], &zStr[i], nStr-i);
1249 j += nStr - i;
1250 assert( j<=nOut );
1251 zOut[j] = 0;
1252 sqlite3_result_text(context, (char*)zOut, j, sqlite3_free);
1253}
1254
drh309b3382007-03-17 17:52:42 +00001255/*
1256** Implementation of the TRIM(), LTRIM(), and RTRIM() functions.
1257** The userdata is 0x1 for left trim, 0x2 for right trim, 0x3 for both.
1258*/
1259static void trimFunc(
1260 sqlite3_context *context,
1261 int argc,
1262 sqlite3_value **argv
1263){
1264 const unsigned char *zIn; /* Input string */
1265 const unsigned char *zCharSet; /* Set of characters to trim */
1266 int nIn; /* Number of bytes in input */
drh7209c692008-04-27 18:40:11 +00001267 int flags; /* 1: trimleft 2: trimright 3: trim */
drhd1e3a612007-04-27 21:59:52 +00001268 int i; /* Loop counter */
drh1bd10f82008-12-10 21:19:56 +00001269 unsigned char *aLen = 0; /* Length of each character in zCharSet */
1270 unsigned char **azChar = 0; /* Individual characters in zCharSet */
drhd1e3a612007-04-27 21:59:52 +00001271 int nChar; /* Number of characters in zCharSet */
1272
drh309b3382007-03-17 17:52:42 +00001273 if( sqlite3_value_type(argv[0])==SQLITE_NULL ){
1274 return;
1275 }
1276 zIn = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001277 if( zIn==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001278 nIn = sqlite3_value_bytes(argv[0]);
1279 assert( zIn==sqlite3_value_text(argv[0]) );
drh309b3382007-03-17 17:52:42 +00001280 if( argc==1 ){
drhd1e3a612007-04-27 21:59:52 +00001281 static const unsigned char lenOne[] = { 1 };
danielk1977a4de4532008-09-02 15:44:08 +00001282 static unsigned char * const azOne[] = { (u8*)" " };
drhd1e3a612007-04-27 21:59:52 +00001283 nChar = 1;
drh8cff3822007-05-02 02:08:28 +00001284 aLen = (u8*)lenOne;
danielk1977bc67da42007-12-11 04:23:19 +00001285 azChar = (unsigned char **)azOne;
drhd1e3a612007-04-27 21:59:52 +00001286 zCharSet = 0;
drh7a521cf2007-04-25 18:23:52 +00001287 }else if( (zCharSet = sqlite3_value_text(argv[1]))==0 ){
drh309b3382007-03-17 17:52:42 +00001288 return;
drhd1e3a612007-04-27 21:59:52 +00001289 }else{
1290 const unsigned char *z;
1291 for(z=zCharSet, nChar=0; *z; nChar++){
drh4a919112007-05-15 11:55:09 +00001292 SQLITE_SKIP_UTF8(z);
drhd1e3a612007-04-27 21:59:52 +00001293 }
1294 if( nChar>0 ){
drhb1a6c3c2008-03-20 16:30:17 +00001295 azChar = contextMalloc(context, ((i64)nChar)*(sizeof(char*)+1));
drhd1e3a612007-04-27 21:59:52 +00001296 if( azChar==0 ){
1297 return;
1298 }
1299 aLen = (unsigned char*)&azChar[nChar];
1300 for(z=zCharSet, nChar=0; *z; nChar++){
danielk1977bc67da42007-12-11 04:23:19 +00001301 azChar[nChar] = (unsigned char *)z;
drh4a919112007-05-15 11:55:09 +00001302 SQLITE_SKIP_UTF8(z);
drh3abbd392008-12-10 23:04:13 +00001303 aLen[nChar] = (u8)(z - azChar[nChar]);
drhd1e3a612007-04-27 21:59:52 +00001304 }
1305 }
drh309b3382007-03-17 17:52:42 +00001306 }
drhd1e3a612007-04-27 21:59:52 +00001307 if( nChar>0 ){
shane1fc41292008-07-08 22:28:48 +00001308 flags = SQLITE_PTR_TO_INT(sqlite3_user_data(context));
drh309b3382007-03-17 17:52:42 +00001309 if( flags & 1 ){
drhd1e3a612007-04-27 21:59:52 +00001310 while( nIn>0 ){
drh1bd10f82008-12-10 21:19:56 +00001311 int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001312 for(i=0; i<nChar; i++){
1313 len = aLen[i];
drh27e62db2009-04-02 10:16:17 +00001314 if( len<=nIn && memcmp(zIn, azChar[i], len)==0 ) break;
drhd1e3a612007-04-27 21:59:52 +00001315 }
1316 if( i>=nChar ) break;
1317 zIn += len;
1318 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001319 }
1320 }
1321 if( flags & 2 ){
drhd1e3a612007-04-27 21:59:52 +00001322 while( nIn>0 ){
drh1bd10f82008-12-10 21:19:56 +00001323 int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001324 for(i=0; i<nChar; i++){
1325 len = aLen[i];
1326 if( len<=nIn && memcmp(&zIn[nIn-len],azChar[i],len)==0 ) break;
1327 }
1328 if( i>=nChar ) break;
1329 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001330 }
1331 }
drhd1e3a612007-04-27 21:59:52 +00001332 if( zCharSet ){
1333 sqlite3_free(azChar);
1334 }
drh309b3382007-03-17 17:52:42 +00001335 }
1336 sqlite3_result_text(context, (char*)zIn, nIn, SQLITE_TRANSIENT);
1337}
drh26b6d902007-03-17 13:27:54 +00001338
danielk1977a4de4532008-09-02 15:44:08 +00001339
drhcc153132016-08-04 12:35:17 +00001340#ifdef SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION
1341/*
1342** The "unknown" function is automatically substituted in place of
1343** any unrecognized function name when doing an EXPLAIN or EXPLAIN QUERY PLAN
1344** when the SQLITE_ENABLE_UNKNOWN_FUNCTION compile-time option is used.
1345** When the "sqlite3" command-line shell is built using this functionality,
1346** that allows an EXPLAIN or EXPLAIN QUERY PLAN for complex queries
1347** involving application-defined functions to be examined in a generic
1348** sqlite3 shell.
1349*/
1350static void unknownFunc(
1351 sqlite3_context *context,
1352 int argc,
1353 sqlite3_value **argv
1354){
1355 /* no-op */
1356}
1357#endif /*SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION*/
1358
1359
drh2ba3ccc2009-12-08 02:06:08 +00001360/* IMP: R-25361-16150 This function is omitted from SQLite by default. It
1361** is only available if the SQLITE_SOUNDEX compile-time option is used
1362** when SQLite is built.
1363*/
drhd24cc422003-03-27 12:51:24 +00001364#ifdef SQLITE_SOUNDEX
1365/*
1366** Compute the soundex encoding of a word.
drh2ba3ccc2009-12-08 02:06:08 +00001367**
1368** IMP: R-59782-00072 The soundex(X) function returns a string that is the
1369** soundex encoding of the string X.
drhd24cc422003-03-27 12:51:24 +00001370*/
drh137c7282007-01-29 17:58:28 +00001371static void soundexFunc(
1372 sqlite3_context *context,
1373 int argc,
1374 sqlite3_value **argv
1375){
drhd24cc422003-03-27 12:51:24 +00001376 char zResult[8];
drh4c755c02004-08-08 20:22:17 +00001377 const u8 *zIn;
drhd24cc422003-03-27 12:51:24 +00001378 int i, j;
1379 static const unsigned char iCode[] = {
1380 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1381 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1382 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1383 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1384 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1385 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1386 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1387 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1388 };
1389 assert( argc==1 );
drh4c755c02004-08-08 20:22:17 +00001390 zIn = (u8*)sqlite3_value_text(argv[0]);
drhbdf67e02006-08-19 11:34:01 +00001391 if( zIn==0 ) zIn = (u8*)"";
drhdc86e2b2009-01-24 11:30:42 +00001392 for(i=0; zIn[i] && !sqlite3Isalpha(zIn[i]); i++){}
drhd24cc422003-03-27 12:51:24 +00001393 if( zIn[i] ){
drhbdf67e02006-08-19 11:34:01 +00001394 u8 prevcode = iCode[zIn[i]&0x7f];
danielk197778ca0e72009-01-20 16:53:39 +00001395 zResult[0] = sqlite3Toupper(zIn[i]);
drhd24cc422003-03-27 12:51:24 +00001396 for(j=1; j<4 && zIn[i]; i++){
1397 int code = iCode[zIn[i]&0x7f];
1398 if( code>0 ){
drhbdf67e02006-08-19 11:34:01 +00001399 if( code!=prevcode ){
1400 prevcode = code;
1401 zResult[j++] = code + '0';
1402 }
1403 }else{
1404 prevcode = 0;
drhd24cc422003-03-27 12:51:24 +00001405 }
1406 }
1407 while( j<4 ){
1408 zResult[j++] = '0';
1409 }
1410 zResult[j] = 0;
danielk1977d8123362004-06-12 09:25:12 +00001411 sqlite3_result_text(context, zResult, 4, SQLITE_TRANSIENT);
drhd24cc422003-03-27 12:51:24 +00001412 }else{
drh2ba3ccc2009-12-08 02:06:08 +00001413 /* IMP: R-64894-50321 The string "?000" is returned if the argument
1414 ** is NULL or contains no ASCII alphabetic characters. */
danielk1977d8123362004-06-12 09:25:12 +00001415 sqlite3_result_text(context, "?000", 4, SQLITE_STATIC);
drhd24cc422003-03-27 12:51:24 +00001416 }
1417}
drh2ba3ccc2009-12-08 02:06:08 +00001418#endif /* SQLITE_SOUNDEX */
drhd24cc422003-03-27 12:51:24 +00001419
drhfdb83b22006-06-17 14:12:47 +00001420#ifndef SQLITE_OMIT_LOAD_EXTENSION
1421/*
1422** A function that loads a shared-library extension then returns NULL.
1423*/
1424static void loadExt(sqlite3_context *context, int argc, sqlite3_value **argv){
danielk197765fd59f2006-06-24 11:51:33 +00001425 const char *zFile = (const char *)sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001426 const char *zProc;
drhfa4a4b92008-03-19 21:45:51 +00001427 sqlite3 *db = sqlite3_context_db_handle(context);
drhfdb83b22006-06-17 14:12:47 +00001428 char *zErrMsg = 0;
1429
drh191dd062016-04-21 01:30:09 +00001430 /* Disallow the load_extension() SQL function unless the SQLITE_LoadExtFunc
drh1a55ded2016-04-20 00:30:05 +00001431 ** flag is set. See the sqlite3_enable_load_extension() API.
1432 */
drhf602a162016-04-21 01:58:21 +00001433 if( (db->flags & SQLITE_LoadExtFunc)==0 ){
1434 sqlite3_result_error(context, "not authorized", -1);
1435 return;
1436 }
drh1a55ded2016-04-20 00:30:05 +00001437
drhfdb83b22006-06-17 14:12:47 +00001438 if( argc==2 ){
danielk197765fd59f2006-06-24 11:51:33 +00001439 zProc = (const char *)sqlite3_value_text(argv[1]);
drh7a521cf2007-04-25 18:23:52 +00001440 }else{
1441 zProc = 0;
drhfdb83b22006-06-17 14:12:47 +00001442 }
drh7a521cf2007-04-25 18:23:52 +00001443 if( zFile && sqlite3_load_extension(db, zFile, zProc, &zErrMsg) ){
drhfdb83b22006-06-17 14:12:47 +00001444 sqlite3_result_error(context, zErrMsg, -1);
1445 sqlite3_free(zErrMsg);
1446 }
1447}
1448#endif
1449
danielk197701427a62005-01-11 13:02:33 +00001450
drh0ac65892002-04-20 14:24:41 +00001451/*
drhd3a149e2002-02-24 17:12:53 +00001452** An instance of the following structure holds the context of a
drhdd5baa92002-02-27 19:50:59 +00001453** sum() or avg() aggregate computation.
1454*/
1455typedef struct SumCtx SumCtx;
1456struct SumCtx {
drh8c08e862006-02-11 17:34:00 +00001457 double rSum; /* Floating point sum */
1458 i64 iSum; /* Integer sum */
1459 i64 cnt; /* Number of elements summed */
1460 u8 overflow; /* True if integer overflow seen */
1461 u8 approx; /* True if non-integer value was input to the sum */
drhdd5baa92002-02-27 19:50:59 +00001462};
1463
1464/*
drha97fdd32006-01-12 22:17:50 +00001465** Routines used to compute the sum, average, and total.
1466**
1467** The SUM() function follows the (broken) SQL standard which means
1468** that it returns NULL if it sums over no inputs. TOTAL returns
1469** 0.0 in that case. In addition, TOTAL always returns a float where
1470** SUM might return an integer if it never encounters a floating point
drhc806d852006-05-11 13:25:39 +00001471** value. TOTAL never fails, but SUM might through an exception if
1472** it overflows an integer.
drhdd5baa92002-02-27 19:50:59 +00001473*/
danielk19770ae8b832004-05-25 12:05:56 +00001474static void sumStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drhdd5baa92002-02-27 19:50:59 +00001475 SumCtx *p;
drh3d1d95e2005-09-08 10:37:01 +00001476 int type;
drh3f219f42005-09-08 19:45:57 +00001477 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001478 UNUSED_PARAMETER(argc);
drh4f26d6c2004-05-26 23:25:30 +00001479 p = sqlite3_aggregate_context(context, sizeof(*p));
drh29d72102006-02-09 22:13:41 +00001480 type = sqlite3_value_numeric_type(argv[0]);
drh3d1d95e2005-09-08 10:37:01 +00001481 if( p && type!=SQLITE_NULL ){
drh739105c2002-05-29 23:22:23 +00001482 p->cnt++;
drh29d72102006-02-09 22:13:41 +00001483 if( type==SQLITE_INTEGER ){
drh8c08e862006-02-11 17:34:00 +00001484 i64 v = sqlite3_value_int64(argv[0]);
1485 p->rSum += v;
drh158b9cb2011-03-05 20:59:46 +00001486 if( (p->approx|p->overflow)==0 && sqlite3AddInt64(&p->iSum, v) ){
1487 p->overflow = 1;
drh29d72102006-02-09 22:13:41 +00001488 }
1489 }else{
drh8c08e862006-02-11 17:34:00 +00001490 p->rSum += sqlite3_value_double(argv[0]);
drh29d72102006-02-09 22:13:41 +00001491 p->approx = 1;
drh3f219f42005-09-08 19:45:57 +00001492 }
drh739105c2002-05-29 23:22:23 +00001493 }
drhdd5baa92002-02-27 19:50:59 +00001494}
danielk19770ae8b832004-05-25 12:05:56 +00001495static void sumFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001496 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001497 p = sqlite3_aggregate_context(context, 0);
drhc2bd9132005-09-08 20:37:43 +00001498 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001499 if( p->overflow ){
1500 sqlite3_result_error(context,"integer overflow",-1);
1501 }else if( p->approx ){
1502 sqlite3_result_double(context, p->rSum);
drhc2bd9132005-09-08 20:37:43 +00001503 }else{
drh8c08e862006-02-11 17:34:00 +00001504 sqlite3_result_int64(context, p->iSum);
drhc2bd9132005-09-08 20:37:43 +00001505 }
drh3d1d95e2005-09-08 10:37:01 +00001506 }
drhdd5baa92002-02-27 19:50:59 +00001507}
danielk19770ae8b832004-05-25 12:05:56 +00001508static void avgFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001509 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001510 p = sqlite3_aggregate_context(context, 0);
drh739105c2002-05-29 23:22:23 +00001511 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001512 sqlite3_result_double(context, p->rSum/(double)p->cnt);
drhdd5baa92002-02-27 19:50:59 +00001513 }
1514}
drha97fdd32006-01-12 22:17:50 +00001515static void totalFinalize(sqlite3_context *context){
1516 SumCtx *p;
1517 p = sqlite3_aggregate_context(context, 0);
shanefbd60f82009-02-04 03:59:25 +00001518 /* (double)0 In case of SQLITE_OMIT_FLOATING_POINT... */
1519 sqlite3_result_double(context, p ? p->rSum : (double)0);
drha97fdd32006-01-12 22:17:50 +00001520}
drhdd5baa92002-02-27 19:50:59 +00001521
1522/*
drh0bce8352002-02-28 00:41:10 +00001523** The following structure keeps track of state information for the
1524** count() aggregate function.
1525*/
1526typedef struct CountCtx CountCtx;
1527struct CountCtx {
drhfc6ad392006-02-09 13:38:19 +00001528 i64 n;
drh0bce8352002-02-28 00:41:10 +00001529};
drhdd5baa92002-02-27 19:50:59 +00001530
drh0bce8352002-02-28 00:41:10 +00001531/*
1532** Routines to implement the count() aggregate function.
1533*/
danielk19770ae8b832004-05-25 12:05:56 +00001534static void countStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drh0bce8352002-02-28 00:41:10 +00001535 CountCtx *p;
drh4f26d6c2004-05-26 23:25:30 +00001536 p = sqlite3_aggregate_context(context, sizeof(*p));
drh9c054832004-05-31 18:51:57 +00001537 if( (argc==0 || SQLITE_NULL!=sqlite3_value_type(argv[0])) && p ){
drh0bce8352002-02-28 00:41:10 +00001538 p->n++;
1539 }
drh2e79c3d2009-04-08 23:04:14 +00001540
drhd3264c72009-04-15 13:39:47 +00001541#ifndef SQLITE_OMIT_DEPRECATED
drh2e79c3d2009-04-08 23:04:14 +00001542 /* The sqlite3_aggregate_count() function is deprecated. But just to make
1543 ** sure it still operates correctly, verify that its count agrees with our
1544 ** internal count when using count(*) and when the total count can be
1545 ** expressed as a 32-bit integer. */
1546 assert( argc==1 || p==0 || p->n>0x7fffffff
1547 || p->n==sqlite3_aggregate_count(context) );
drhd3264c72009-04-15 13:39:47 +00001548#endif
drh0bce8352002-02-28 00:41:10 +00001549}
danielk19770ae8b832004-05-25 12:05:56 +00001550static void countFinalize(sqlite3_context *context){
drh0bce8352002-02-28 00:41:10 +00001551 CountCtx *p;
drhabfcea22005-09-06 20:36:48 +00001552 p = sqlite3_aggregate_context(context, 0);
drhfc6ad392006-02-09 13:38:19 +00001553 sqlite3_result_int64(context, p ? p->n : 0);
drh0bce8352002-02-28 00:41:10 +00001554}
1555
1556/*
drh0bce8352002-02-28 00:41:10 +00001557** Routines to implement min() and max() aggregate functions.
1558*/
danielk197762c14b32008-11-19 09:05:26 +00001559static void minmaxStep(
1560 sqlite3_context *context,
1561 int NotUsed,
1562 sqlite3_value **argv
1563){
danielk197788208052004-05-25 01:13:20 +00001564 Mem *pArg = (Mem *)argv[0];
drh9eb516c2004-07-18 20:52:32 +00001565 Mem *pBest;
danielk197762c14b32008-11-19 09:05:26 +00001566 UNUSED_PARAMETER(NotUsed);
drh9eb516c2004-07-18 20:52:32 +00001567
drh9eb516c2004-07-18 20:52:32 +00001568 pBest = (Mem *)sqlite3_aggregate_context(context, sizeof(*pBest));
danielk19773aeab9e2004-06-24 00:20:04 +00001569 if( !pBest ) return;
drh268380c2004-02-25 13:47:31 +00001570
drh94a6d992012-02-02 18:42:09 +00001571 if( sqlite3_value_type(argv[0])==SQLITE_NULL ){
1572 if( pBest->flags ) sqlite3SkipAccumulatorLoad(context);
1573 }else if( pBest->flags ){
drh9eb516c2004-07-18 20:52:32 +00001574 int max;
1575 int cmp;
danielk1977dc1bdc42004-06-11 10:51:27 +00001576 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk19777e18c252004-05-25 11:47:24 +00001577 /* This step function is used for both the min() and max() aggregates,
1578 ** the only difference between the two being that the sense of the
1579 ** comparison is inverted. For the max() aggregate, the
1580 ** sqlite3_user_data() function returns (void *)-1. For min() it
1581 ** returns (void *)db, where db is the sqlite3* database pointer.
1582 ** Therefore the next statement sets variable 'max' to 1 for the max()
1583 ** aggregate, or 0 for min().
1584 */
drh309b3382007-03-17 17:52:42 +00001585 max = sqlite3_user_data(context)!=0;
danielk1977dc1bdc42004-06-11 10:51:27 +00001586 cmp = sqlite3MemCompare(pBest, pArg, pColl);
danielk197788208052004-05-25 01:13:20 +00001587 if( (max && cmp<0) || (!max && cmp>0) ){
drhb21c8cd2007-08-21 19:33:56 +00001588 sqlite3VdbeMemCopy(pBest, pArg);
drh7a957892012-02-02 17:35:43 +00001589 }else{
1590 sqlite3SkipAccumulatorLoad(context);
danielk197788208052004-05-25 01:13:20 +00001591 }
drh268380c2004-02-25 13:47:31 +00001592 }else{
drh035e5632014-09-16 14:16:31 +00001593 pBest->db = sqlite3_context_db_handle(context);
drhb21c8cd2007-08-21 19:33:56 +00001594 sqlite3VdbeMemCopy(pBest, pArg);
drh0bce8352002-02-28 00:41:10 +00001595 }
1596}
danielk19770ae8b832004-05-25 12:05:56 +00001597static void minMaxFinalize(sqlite3_context *context){
danielk197788208052004-05-25 01:13:20 +00001598 sqlite3_value *pRes;
drhabfcea22005-09-06 20:36:48 +00001599 pRes = (sqlite3_value *)sqlite3_aggregate_context(context, 0);
1600 if( pRes ){
drh94a6d992012-02-02 18:42:09 +00001601 if( pRes->flags ){
drhabfcea22005-09-06 20:36:48 +00001602 sqlite3_result_value(context, pRes);
1603 }
1604 sqlite3VdbeMemRelease(pRes);
drh0bce8352002-02-28 00:41:10 +00001605 }
1606}
drhdd5baa92002-02-27 19:50:59 +00001607
drhb0689692007-11-01 17:38:30 +00001608/*
1609** group_concat(EXPR, ?SEPARATOR?)
1610*/
1611static void groupConcatStep(
1612 sqlite3_context *context,
1613 int argc,
1614 sqlite3_value **argv
1615){
1616 const char *zVal;
drhade86482007-11-28 22:36:40 +00001617 StrAccum *pAccum;
drhb0689692007-11-01 17:38:30 +00001618 const char *zSep;
drh07d31172009-02-02 21:57:05 +00001619 int nVal, nSep;
1620 assert( argc==1 || argc==2 );
1621 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drhade86482007-11-28 22:36:40 +00001622 pAccum = (StrAccum*)sqlite3_aggregate_context(context, sizeof(*pAccum));
1623
1624 if( pAccum ){
drhbb4957f2008-03-20 14:03:29 +00001625 sqlite3 *db = sqlite3_context_db_handle(context);
drhc0490572015-05-02 11:45:53 +00001626 int firstTerm = pAccum->mxAlloc==0;
drhbb4957f2008-03-20 14:03:29 +00001627 pAccum->mxAlloc = db->aLimit[SQLITE_LIMIT_LENGTH];
drh8bfd7192009-06-19 16:44:41 +00001628 if( !firstTerm ){
drh07d31172009-02-02 21:57:05 +00001629 if( argc==2 ){
1630 zSep = (char*)sqlite3_value_text(argv[1]);
1631 nSep = sqlite3_value_bytes(argv[1]);
drhb0689692007-11-01 17:38:30 +00001632 }else{
1633 zSep = ",";
drhade86482007-11-28 22:36:40 +00001634 nSep = 1;
drhb0689692007-11-01 17:38:30 +00001635 }
drha9ab4812013-12-11 11:00:44 +00001636 if( nSep ) sqlite3StrAccumAppend(pAccum, zSep, nSep);
drhb0689692007-11-01 17:38:30 +00001637 }
drh07d31172009-02-02 21:57:05 +00001638 zVal = (char*)sqlite3_value_text(argv[0]);
1639 nVal = sqlite3_value_bytes(argv[0]);
drh8009c9b2014-05-07 20:24:00 +00001640 if( zVal ) sqlite3StrAccumAppend(pAccum, zVal, nVal);
drhb0689692007-11-01 17:38:30 +00001641 }
1642}
1643static void groupConcatFinalize(sqlite3_context *context){
drhade86482007-11-28 22:36:40 +00001644 StrAccum *pAccum;
1645 pAccum = sqlite3_aggregate_context(context, 0);
1646 if( pAccum ){
drhb49bc862013-08-21 21:12:10 +00001647 if( pAccum->accError==STRACCUM_TOOBIG ){
drhade86482007-11-28 22:36:40 +00001648 sqlite3_result_error_toobig(context);
drhb49bc862013-08-21 21:12:10 +00001649 }else if( pAccum->accError==STRACCUM_NOMEM ){
drhade86482007-11-28 22:36:40 +00001650 sqlite3_result_error_nomem(context);
1651 }else{
1652 sqlite3_result_text(context, sqlite3StrAccumFinish(pAccum), -1,
1653 sqlite3_free);
1654 }
drhb0689692007-11-01 17:38:30 +00001655 }
1656}
drh4e5ffc52004-08-31 00:52:37 +00001657
drhd3a149e2002-02-24 17:12:53 +00001658/*
drha4741842010-04-25 20:58:37 +00001659** This routine does per-connection function registration. Most
1660** of the built-in functions above are part of the global function set.
1661** This routine only deals with those that are not global.
drhdc04c582002-02-24 01:55:15 +00001662*/
drh80738d92016-02-15 00:34:16 +00001663void sqlite3RegisterPerConnectionBuiltinFunctions(sqlite3 *db){
drha4741842010-04-25 20:58:37 +00001664 int rc = sqlite3_overload_function(db, "MATCH", 2);
1665 assert( rc==SQLITE_NOMEM || rc==SQLITE_OK );
1666 if( rc==SQLITE_NOMEM ){
drh4a642b62016-02-05 01:55:27 +00001667 sqlite3OomFault(db);
danielk1977832a58a2007-06-22 15:21:15 +00001668 }
drh55ef4d92005-08-14 01:20:37 +00001669}
1670
1671/*
1672** Set the LIKEOPT flag on the 2-argument function with the given name.
1673*/
drh1bd10f82008-12-10 21:19:56 +00001674static void setLikeOptFlag(sqlite3 *db, const char *zName, u8 flagVal){
drh55ef4d92005-08-14 01:20:37 +00001675 FuncDef *pDef;
drh80738d92016-02-15 00:34:16 +00001676 pDef = sqlite3FindFunction(db, zName, 2, SQLITE_UTF8, 0);
drhd27135a2009-04-02 13:36:37 +00001677 if( ALWAYS(pDef) ){
drhd36e1042013-09-06 13:10:12 +00001678 pDef->funcFlags |= flagVal;
drh55ef4d92005-08-14 01:20:37 +00001679 }
1680}
1681
1682/*
1683** Register the built-in LIKE and GLOB functions. The caseSensitive
1684** parameter determines whether or not the LIKE operator is case
1685** sensitive. GLOB is always case sensitive.
1686*/
1687void sqlite3RegisterLikeFunctions(sqlite3 *db, int caseSensitive){
1688 struct compareInfo *pInfo;
1689 if( caseSensitive ){
1690 pInfo = (struct compareInfo*)&likeInfoAlt;
1691 }else{
1692 pInfo = (struct compareInfo*)&likeInfoNorm;
1693 }
drh901e9942010-12-15 18:54:37 +00001694 sqlite3CreateFunc(db, "like", 2, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0);
1695 sqlite3CreateFunc(db, "like", 3, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0);
1696 sqlite3CreateFunc(db, "glob", 2, SQLITE_UTF8,
dand2199f02010-08-27 17:48:52 +00001697 (struct compareInfo*)&globInfo, likeFunc, 0, 0, 0);
drhd64fe2f2005-08-28 17:00:23 +00001698 setLikeOptFlag(db, "glob", SQLITE_FUNC_LIKE | SQLITE_FUNC_CASE);
1699 setLikeOptFlag(db, "like",
1700 caseSensitive ? (SQLITE_FUNC_LIKE | SQLITE_FUNC_CASE) : SQLITE_FUNC_LIKE);
drh55ef4d92005-08-14 01:20:37 +00001701}
1702
1703/*
1704** pExpr points to an expression which implements a function. If
1705** it is appropriate to apply the LIKE optimization to that function
1706** then set aWc[0] through aWc[2] to the wildcard characters and
1707** return TRUE. If the function is not a LIKE-style function then
1708** return FALSE.
drh16897072015-03-07 00:57:37 +00001709**
1710** *pIsNocase is set to true if uppercase and lowercase are equivalent for
1711** the function (default for LIKE). If the function makes the distinction
1712** between uppercase and lowercase (as does GLOB) then *pIsNocase is set to
1713** false.
drh55ef4d92005-08-14 01:20:37 +00001714*/
drhd64fe2f2005-08-28 17:00:23 +00001715int sqlite3IsLikeFunction(sqlite3 *db, Expr *pExpr, int *pIsNocase, char *aWc){
drh55ef4d92005-08-14 01:20:37 +00001716 FuncDef *pDef;
danielk19776ab3a2e2009-02-19 14:39:25 +00001717 if( pExpr->op!=TK_FUNCTION
1718 || !pExpr->x.pList
1719 || pExpr->x.pList->nExpr!=2
1720 ){
drh55ef4d92005-08-14 01:20:37 +00001721 return 0;
1722 }
danielk19776ab3a2e2009-02-19 14:39:25 +00001723 assert( !ExprHasProperty(pExpr, EP_xIsSelect) );
drh80738d92016-02-15 00:34:16 +00001724 pDef = sqlite3FindFunction(db, pExpr->u.zToken, 2, SQLITE_UTF8, 0);
drhd36e1042013-09-06 13:10:12 +00001725 if( NEVER(pDef==0) || (pDef->funcFlags & SQLITE_FUNC_LIKE)==0 ){
drh55ef4d92005-08-14 01:20:37 +00001726 return 0;
1727 }
1728
1729 /* The memcpy() statement assumes that the wildcard characters are
1730 ** the first three statements in the compareInfo structure. The
1731 ** asserts() that follow verify that assumption
1732 */
1733 memcpy(aWc, pDef->pUserData, 3);
1734 assert( (char*)&likeInfoAlt == (char*)&likeInfoAlt.matchAll );
1735 assert( &((char*)&likeInfoAlt)[1] == (char*)&likeInfoAlt.matchOne );
1736 assert( &((char*)&likeInfoAlt)[2] == (char*)&likeInfoAlt.matchSet );
drhd36e1042013-09-06 13:10:12 +00001737 *pIsNocase = (pDef->funcFlags & SQLITE_FUNC_CASE)==0;
drh55ef4d92005-08-14 01:20:37 +00001738 return 1;
drhdc04c582002-02-24 01:55:15 +00001739}
danielk19778c0a7912008-08-20 14:49:23 +00001740
drh70a8ca32008-08-21 18:49:27 +00001741/*
peter.d.reid60ec9142014-09-06 16:39:46 +00001742** All of the FuncDef structures in the aBuiltinFunc[] array above
drh777c5382008-08-21 20:21:34 +00001743** to the global function hash table. This occurs at start-time (as
1744** a consequence of calling sqlite3_initialize()).
1745**
1746** After this routine runs
drh70a8ca32008-08-21 18:49:27 +00001747*/
drh80738d92016-02-15 00:34:16 +00001748void sqlite3RegisterBuiltinFunctions(void){
danielk197793ce7412008-09-01 19:14:02 +00001749 /*
1750 ** The following array holds FuncDef structures for all of the functions
1751 ** defined in this file.
1752 **
1753 ** The array cannot be constant since changes are made to the
1754 ** FuncDef.pHash elements at start-time. The elements of this array
1755 ** are read-only after initialization is complete.
drh80738d92016-02-15 00:34:16 +00001756 **
1757 ** For peak efficiency, put the most frequently used function last.
danielk197793ce7412008-09-01 19:14:02 +00001758 */
drh80738d92016-02-15 00:34:16 +00001759 static FuncDef aBuiltinFunc[] = {
1760#ifdef SQLITE_SOUNDEX
1761 FUNCTION(soundex, 1, 0, 0, soundexFunc ),
1762#endif
1763#ifndef SQLITE_OMIT_LOAD_EXTENSION
1764 VFUNCTION(load_extension, 1, 0, 0, loadExt ),
1765 VFUNCTION(load_extension, 2, 0, 0, loadExt ),
1766#endif
1767#if SQLITE_USER_AUTHENTICATION
1768 FUNCTION(sqlite_crypt, 2, 0, 0, sqlite3CryptFunc ),
1769#endif
1770#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
1771 DFUNCTION(sqlite_compileoption_used,1, 0, 0, compileoptionusedFunc ),
1772 DFUNCTION(sqlite_compileoption_get, 1, 0, 0, compileoptiongetFunc ),
1773#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
1774 FUNCTION2(unlikely, 1, 0, 0, noopFunc, SQLITE_FUNC_UNLIKELY),
1775 FUNCTION2(likelihood, 2, 0, 0, noopFunc, SQLITE_FUNC_UNLIKELY),
1776 FUNCTION2(likely, 1, 0, 0, noopFunc, SQLITE_FUNC_UNLIKELY),
danielk197793ce7412008-09-01 19:14:02 +00001777 FUNCTION(ltrim, 1, 1, 0, trimFunc ),
1778 FUNCTION(ltrim, 2, 1, 0, trimFunc ),
1779 FUNCTION(rtrim, 1, 2, 0, trimFunc ),
1780 FUNCTION(rtrim, 2, 2, 0, trimFunc ),
1781 FUNCTION(trim, 1, 3, 0, trimFunc ),
1782 FUNCTION(trim, 2, 3, 0, trimFunc ),
1783 FUNCTION(min, -1, 0, 1, minmaxFunc ),
1784 FUNCTION(min, 0, 0, 1, 0 ),
drh9588ad92014-09-15 14:46:02 +00001785 AGGREGATE2(min, 1, 0, 1, minmaxStep, minMaxFinalize,
1786 SQLITE_FUNC_MINMAX ),
danielk197793ce7412008-09-01 19:14:02 +00001787 FUNCTION(max, -1, 1, 1, minmaxFunc ),
1788 FUNCTION(max, 0, 1, 1, 0 ),
drh9588ad92014-09-15 14:46:02 +00001789 AGGREGATE2(max, 1, 1, 1, minmaxStep, minMaxFinalize,
1790 SQLITE_FUNC_MINMAX ),
drha748fdc2012-03-28 01:34:47 +00001791 FUNCTION2(typeof, 1, 0, 0, typeofFunc, SQLITE_FUNC_TYPEOF),
1792 FUNCTION2(length, 1, 0, 0, lengthFunc, SQLITE_FUNC_LENGTH),
drhd55e0722012-10-25 03:07:29 +00001793 FUNCTION(instr, 2, 0, 0, instrFunc ),
drha5c14162013-12-17 15:03:06 +00001794 FUNCTION(printf, -1, 0, 0, printfFunc ),
drhd495d8c2013-02-22 19:34:25 +00001795 FUNCTION(unicode, 1, 0, 0, unicodeFunc ),
1796 FUNCTION(char, -1, 0, 0, charFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001797 FUNCTION(abs, 1, 0, 0, absFunc ),
shanefbd60f82009-02-04 03:59:25 +00001798#ifndef SQLITE_OMIT_FLOATING_POINT
danielk197793ce7412008-09-01 19:14:02 +00001799 FUNCTION(round, 1, 0, 0, roundFunc ),
1800 FUNCTION(round, 2, 0, 0, roundFunc ),
shanefbd60f82009-02-04 03:59:25 +00001801#endif
danielk197793ce7412008-09-01 19:14:02 +00001802 FUNCTION(upper, 1, 0, 0, upperFunc ),
1803 FUNCTION(lower, 1, 0, 0, lowerFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001804 FUNCTION(hex, 1, 0, 0, hexFunc ),
drhcca9f3d2013-09-06 15:23:29 +00001805 FUNCTION2(ifnull, 2, 0, 0, noopFunc, SQLITE_FUNC_COALESCE),
drhb1fba282013-11-21 14:33:48 +00001806 VFUNCTION(random, 0, 0, 0, randomFunc ),
1807 VFUNCTION(randomblob, 1, 0, 0, randomBlob ),
danielk197793ce7412008-09-01 19:14:02 +00001808 FUNCTION(nullif, 2, 0, 1, nullifFunc ),
drh03bf26d2015-08-31 21:16:36 +00001809 DFUNCTION(sqlite_version, 0, 0, 0, versionFunc ),
1810 DFUNCTION(sqlite_source_id, 0, 0, 0, sourceidFunc ),
drh840561f2011-04-27 18:08:42 +00001811 FUNCTION(sqlite_log, 2, 0, 0, errlogFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001812 FUNCTION(quote, 1, 0, 0, quoteFunc ),
drhb1fba282013-11-21 14:33:48 +00001813 VFUNCTION(last_insert_rowid, 0, 0, 0, last_insert_rowid),
1814 VFUNCTION(changes, 0, 0, 0, changes ),
1815 VFUNCTION(total_changes, 0, 0, 0, total_changes ),
danielk197793ce7412008-09-01 19:14:02 +00001816 FUNCTION(replace, 3, 0, 0, replaceFunc ),
1817 FUNCTION(zeroblob, 1, 0, 0, zeroblobFunc ),
drh80738d92016-02-15 00:34:16 +00001818 FUNCTION(substr, 2, 0, 0, substrFunc ),
1819 FUNCTION(substr, 3, 0, 0, substrFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001820 AGGREGATE(sum, 1, 0, 0, sumStep, sumFinalize ),
1821 AGGREGATE(total, 1, 0, 0, sumStep, totalFinalize ),
1822 AGGREGATE(avg, 1, 0, 0, sumStep, avgFinalize ),
drh9588ad92014-09-15 14:46:02 +00001823 AGGREGATE2(count, 0, 0, 0, countStep, countFinalize,
1824 SQLITE_FUNC_COUNT ),
danielk197793ce7412008-09-01 19:14:02 +00001825 AGGREGATE(count, 1, 0, 0, countStep, countFinalize ),
drh07d31172009-02-02 21:57:05 +00001826 AGGREGATE(group_concat, 1, 0, 0, groupConcatStep, groupConcatFinalize),
1827 AGGREGATE(group_concat, 2, 0, 0, groupConcatStep, groupConcatFinalize),
danielk197793ce7412008-09-01 19:14:02 +00001828
1829 LIKEFUNC(glob, 2, &globInfo, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
drhcc153132016-08-04 12:35:17 +00001830#ifdef SQLITE_CASE_SENSITIVE_LIKE
danielk197793ce7412008-09-01 19:14:02 +00001831 LIKEFUNC(like, 2, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
1832 LIKEFUNC(like, 3, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
drhcc153132016-08-04 12:35:17 +00001833#else
danielk197793ce7412008-09-01 19:14:02 +00001834 LIKEFUNC(like, 2, &likeInfoNorm, SQLITE_FUNC_LIKE),
1835 LIKEFUNC(like, 3, &likeInfoNorm, SQLITE_FUNC_LIKE),
drhcc153132016-08-04 12:35:17 +00001836#endif
1837#ifdef SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION
1838 FUNCTION(unknown, -1, 0, 0, unknownFunc ),
1839#endif
drh80738d92016-02-15 00:34:16 +00001840 FUNCTION(coalesce, 1, 0, 0, 0 ),
1841 FUNCTION(coalesce, 0, 0, 0, 0 ),
1842 FUNCTION2(coalesce, -1, 0, 0, noopFunc, SQLITE_FUNC_COALESCE),
danielk197793ce7412008-09-01 19:14:02 +00001843 };
drh545f5872010-04-24 14:02:59 +00001844#ifndef SQLITE_OMIT_ALTERTABLE
1845 sqlite3AlterFunctions();
1846#endif
dan0106e372013-08-12 16:34:32 +00001847#if defined(SQLITE_ENABLE_STAT3) || defined(SQLITE_ENABLE_STAT4)
1848 sqlite3AnalyzeFunctions();
1849#endif
drh80738d92016-02-15 00:34:16 +00001850 sqlite3RegisterDateTimeFunctions();
1851 sqlite3InsertBuiltinFuncs(aBuiltinFunc, ArraySize(aBuiltinFunc));
1852
1853#if 0 /* Enable to print out how the built-in functions are hashed */
1854 {
1855 int i;
1856 FuncDef *p;
1857 for(i=0; i<SQLITE_FUNC_HASH_SZ; i++){
1858 printf("FUNC-HASH %02d:", i);
1859 for(p=sqlite3BuiltinFunctions.a[i]; p; p=p->u.pHash){
1860 int n = sqlite3Strlen30(p->zName);
1861 int h = p->zName[0] + n;
1862 printf(" %s(%d)", p->zName, h);
1863 }
1864 printf("\n");
1865 }
1866 }
1867#endif
drh70a8ca32008-08-21 18:49:27 +00001868}