blob: 885725bc6b4757021092ad122a515998ca43accd [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]);
drh1d081ab2016-12-30 15:16:20 +0000207 assert( zNeedle!=0 );
208 assert( zHaystack!=0 || nHaystack==0 );
dan895decf2016-12-30 14:15:56 +0000209 isText = 0;
210 }else{
211 zHaystack = sqlite3_value_text(argv[0]);
212 zNeedle = sqlite3_value_text(argv[1]);
213 isText = 1;
drh1d081ab2016-12-30 15:16:20 +0000214 if( zHaystack==0 || zNeedle==0 ) return;
dan895decf2016-12-30 14:15:56 +0000215 }
216 while( nNeedle<=nHaystack && memcmp(zHaystack, zNeedle, nNeedle)!=0 ){
217 N++;
218 do{
219 nHaystack--;
220 zHaystack++;
221 }while( isText && (zHaystack[0]&0xc0)==0x80 );
222 }
223 if( nNeedle>nHaystack ) N = 0;
drhd55e0722012-10-25 03:07:29 +0000224 }
drhd55e0722012-10-25 03:07:29 +0000225 sqlite3_result_int(context, N);
226}
227
228/*
drha5c14162013-12-17 15:03:06 +0000229** Implementation of the printf() function.
230*/
231static void printfFunc(
232 sqlite3_context *context,
233 int argc,
234 sqlite3_value **argv
235){
236 PrintfArguments x;
237 StrAccum str;
238 const char *zFormat;
239 int n;
drhc0490572015-05-02 11:45:53 +0000240 sqlite3 *db = sqlite3_context_db_handle(context);
drha5c14162013-12-17 15:03:06 +0000241
242 if( argc>=1 && (zFormat = (const char*)sqlite3_value_text(argv[0]))!=0 ){
243 x.nArg = argc-1;
244 x.nUsed = 0;
245 x.apArg = argv+1;
drhc0490572015-05-02 11:45:53 +0000246 sqlite3StrAccumInit(&str, db, 0, 0, db->aLimit[SQLITE_LIMIT_LENGTH]);
drh5f4a6862016-01-30 12:50:25 +0000247 str.printfFlags = SQLITE_PRINTF_SQLFUNC;
248 sqlite3XPrintf(&str, zFormat, &x);
drha5c14162013-12-17 15:03:06 +0000249 n = str.nChar;
250 sqlite3_result_text(context, sqlite3StrAccumFinish(&str), n,
251 SQLITE_DYNAMIC);
252 }
253}
254
255/*
drhf764e6f2007-05-15 01:13:47 +0000256** Implementation of the substr() function.
257**
258** substr(x,p1,p2) returns p2 characters of x[] beginning with p1.
259** p1 is 1-indexed. So substr(x,1,1) returns the first character
260** of x. If x is text, then we actually count UTF-8 characters.
261** If x is a blob, then we count bytes.
262**
263** If p1 is negative, then we begin abs(p1) from the end of x[].
shaneh779b8f12009-11-12 05:04:50 +0000264**
drhf7b54962013-05-28 12:11:54 +0000265** If p2 is negative, return the p2 characters preceding p1.
drh0bce8352002-02-28 00:41:10 +0000266*/
drhf9b596e2004-05-26 16:54:42 +0000267static void substrFunc(
268 sqlite3_context *context,
269 int argc,
270 sqlite3_value **argv
271){
drh2646da72005-12-09 20:02:05 +0000272 const unsigned char *z;
273 const unsigned char *z2;
drh023ae032007-05-08 12:12:16 +0000274 int len;
drhf764e6f2007-05-15 01:13:47 +0000275 int p0type;
drh023ae032007-05-08 12:12:16 +0000276 i64 p1, p2;
drh65595cd2009-02-02 16:32:55 +0000277 int negP2 = 0;
drhf9b596e2004-05-26 16:54:42 +0000278
drh64f31512007-10-12 19:11:55 +0000279 assert( argc==3 || argc==2 );
drh8198d252009-02-01 19:42:37 +0000280 if( sqlite3_value_type(argv[1])==SQLITE_NULL
281 || (argc==3 && sqlite3_value_type(argv[2])==SQLITE_NULL)
282 ){
283 return;
284 }
drhf764e6f2007-05-15 01:13:47 +0000285 p0type = sqlite3_value_type(argv[0]);
drh4adc4cb2009-11-11 20:53:31 +0000286 p1 = sqlite3_value_int(argv[1]);
drhf764e6f2007-05-15 01:13:47 +0000287 if( p0type==SQLITE_BLOB ){
288 len = sqlite3_value_bytes(argv[0]);
289 z = sqlite3_value_blob(argv[0]);
290 if( z==0 ) return;
drh1f0feef2007-05-15 13:27:07 +0000291 assert( len==sqlite3_value_bytes(argv[0]) );
drhf764e6f2007-05-15 01:13:47 +0000292 }else{
293 z = sqlite3_value_text(argv[0]);
294 if( z==0 ) return;
drh4a919112007-05-15 11:55:09 +0000295 len = 0;
drh4adc4cb2009-11-11 20:53:31 +0000296 if( p1<0 ){
297 for(z2=z; *z2; len++){
298 SQLITE_SKIP_UTF8(z2);
299 }
drh4a919112007-05-15 11:55:09 +0000300 }
drhf764e6f2007-05-15 01:13:47 +0000301 }
drh883ad042015-02-19 00:29:11 +0000302#ifdef SQLITE_SUBSTR_COMPATIBILITY
303 /* If SUBSTR_COMPATIBILITY is defined then substr(X,0,N) work the same as
304 ** as substr(X,1,N) - it returns the first N characters of X. This
305 ** is essentially a back-out of the bug-fix in check-in [5fc125d362df4b8]
306 ** from 2009-02-02 for compatibility of applications that exploited the
307 ** old buggy behavior. */
308 if( p1==0 ) p1 = 1; /* <rdar://problem/6778339> */
309#endif
drh64f31512007-10-12 19:11:55 +0000310 if( argc==3 ){
311 p2 = sqlite3_value_int(argv[2]);
drh65595cd2009-02-02 16:32:55 +0000312 if( p2<0 ){
313 p2 = -p2;
314 negP2 = 1;
315 }
drh64f31512007-10-12 19:11:55 +0000316 }else{
drhbb4957f2008-03-20 14:03:29 +0000317 p2 = sqlite3_context_db_handle(context)->aLimit[SQLITE_LIMIT_LENGTH];
drh64f31512007-10-12 19:11:55 +0000318 }
drh0bce8352002-02-28 00:41:10 +0000319 if( p1<0 ){
drh89425d52002-02-28 03:04:48 +0000320 p1 += len;
drh653bc752002-02-28 03:31:10 +0000321 if( p1<0 ){
322 p2 += p1;
drh65595cd2009-02-02 16:32:55 +0000323 if( p2<0 ) p2 = 0;
drh653bc752002-02-28 03:31:10 +0000324 p1 = 0;
325 }
drh0bce8352002-02-28 00:41:10 +0000326 }else if( p1>0 ){
327 p1--;
drh65595cd2009-02-02 16:32:55 +0000328 }else if( p2>0 ){
329 p2--;
drh0bce8352002-02-28 00:41:10 +0000330 }
drh65595cd2009-02-02 16:32:55 +0000331 if( negP2 ){
332 p1 -= p2;
drh4e79c592009-02-01 19:23:32 +0000333 if( p1<0 ){
334 p2 += p1;
335 p1 = 0;
336 }
337 }
drh65595cd2009-02-02 16:32:55 +0000338 assert( p1>=0 && p2>=0 );
drhf764e6f2007-05-15 01:13:47 +0000339 if( p0type!=SQLITE_BLOB ){
drh4a919112007-05-15 11:55:09 +0000340 while( *z && p1 ){
341 SQLITE_SKIP_UTF8(z);
342 p1--;
drhf764e6f2007-05-15 01:13:47 +0000343 }
drh4a919112007-05-15 11:55:09 +0000344 for(z2=z; *z2 && p2; p2--){
345 SQLITE_SKIP_UTF8(z2);
drhf764e6f2007-05-15 01:13:47 +0000346 }
drhbbf483f2014-09-09 20:30:24 +0000347 sqlite3_result_text64(context, (char*)z, z2-z, SQLITE_TRANSIENT,
348 SQLITE_UTF8);
drhf764e6f2007-05-15 01:13:47 +0000349 }else{
drh4adc4cb2009-11-11 20:53:31 +0000350 if( p1+p2>len ){
351 p2 = len-p1;
352 if( p2<0 ) p2 = 0;
353 }
drhbbf483f2014-09-09 20:30:24 +0000354 sqlite3_result_blob64(context, (char*)&z[p1], (u64)p2, SQLITE_TRANSIENT);
drh0bce8352002-02-28 00:41:10 +0000355 }
drh0bce8352002-02-28 00:41:10 +0000356}
357
358/*
359** Implementation of the round() function
360*/
shanefbd60f82009-02-04 03:59:25 +0000361#ifndef SQLITE_OMIT_FLOATING_POINT
danielk19770ae8b832004-05-25 12:05:56 +0000362static void roundFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
danielk197751ad0ec2004-05-24 12:39:02 +0000363 int n = 0;
drh0bce8352002-02-28 00:41:10 +0000364 double r;
drh50d654d2009-06-03 01:24:54 +0000365 char *zBuf;
drh0bce8352002-02-28 00:41:10 +0000366 assert( argc==1 || argc==2 );
danielk197751ad0ec2004-05-24 12:39:02 +0000367 if( argc==2 ){
drh9c054832004-05-31 18:51:57 +0000368 if( SQLITE_NULL==sqlite3_value_type(argv[1]) ) return;
danielk197751ad0ec2004-05-24 12:39:02 +0000369 n = sqlite3_value_int(argv[1]);
370 if( n>30 ) n = 30;
371 if( n<0 ) n = 0;
372 }
drhd589a922006-03-02 03:02:48 +0000373 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drh4f26d6c2004-05-26 23:25:30 +0000374 r = sqlite3_value_double(argv[0]);
shaneh147e1762010-02-17 04:19:27 +0000375 /* If Y==0 and X will fit in a 64-bit int,
376 ** handle the rounding directly,
377 ** otherwise use printf.
378 */
379 if( n==0 && r>=0 && r<LARGEST_INT64-1 ){
380 r = (double)((sqlite_int64)(r+0.5));
381 }else if( n==0 && r<0 && (-r)<LARGEST_INT64-1 ){
382 r = -(double)((sqlite_int64)((-r)+0.5));
drh50d654d2009-06-03 01:24:54 +0000383 }else{
shaneh147e1762010-02-17 04:19:27 +0000384 zBuf = sqlite3_mprintf("%.*f",n,r);
385 if( zBuf==0 ){
386 sqlite3_result_error_nomem(context);
387 return;
388 }
drh9339da12010-09-30 00:50:49 +0000389 sqlite3AtoF(zBuf, &r, sqlite3Strlen30(zBuf), SQLITE_UTF8);
drh50d654d2009-06-03 01:24:54 +0000390 sqlite3_free(zBuf);
drh50d654d2009-06-03 01:24:54 +0000391 }
shaneh147e1762010-02-17 04:19:27 +0000392 sqlite3_result_double(context, r);
drh0bce8352002-02-28 00:41:10 +0000393}
shanefbd60f82009-02-04 03:59:25 +0000394#endif
drhdc04c582002-02-24 01:55:15 +0000395
danielk197726783a52007-08-29 14:06:22 +0000396/*
drhf3cdcdc2015-04-29 16:50:28 +0000397** Allocate nByte bytes of space using sqlite3Malloc(). If the
danielk197726783a52007-08-29 14:06:22 +0000398** allocation fails, call sqlite3_result_error_nomem() to notify
drh27e62db2009-04-02 10:16:17 +0000399** the database handle that malloc() has failed and return NULL.
400** If nByte is larger than the maximum string or blob length, then
401** raise an SQLITE_TOOBIG exception and return NULL.
danielk197726783a52007-08-29 14:06:22 +0000402*/
drhb1a6c3c2008-03-20 16:30:17 +0000403static void *contextMalloc(sqlite3_context *context, i64 nByte){
drhbb4957f2008-03-20 14:03:29 +0000404 char *z;
drh27e62db2009-04-02 10:16:17 +0000405 sqlite3 *db = sqlite3_context_db_handle(context);
drhef31c6a2009-04-02 09:07:12 +0000406 assert( nByte>0 );
drh27e62db2009-04-02 10:16:17 +0000407 testcase( nByte==db->aLimit[SQLITE_LIMIT_LENGTH] );
408 testcase( nByte==db->aLimit[SQLITE_LIMIT_LENGTH]+1 );
409 if( nByte>db->aLimit[SQLITE_LIMIT_LENGTH] ){
drhbb4957f2008-03-20 14:03:29 +0000410 sqlite3_result_error_toobig(context);
411 z = 0;
412 }else{
drhda4ca9d2014-09-09 17:27:35 +0000413 z = sqlite3Malloc(nByte);
drhef31c6a2009-04-02 09:07:12 +0000414 if( !z ){
drhbb4957f2008-03-20 14:03:29 +0000415 sqlite3_result_error_nomem(context);
416 }
danielk1977a1644fd2007-08-29 12:31:25 +0000417 }
418 return z;
419}
420
drhdc04c582002-02-24 01:55:15 +0000421/*
422** Implementation of the upper() and lower() SQL functions.
423*/
danielk19770ae8b832004-05-25 12:05:56 +0000424static void upperFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drh7a521cf2007-04-25 18:23:52 +0000425 char *z1;
426 const char *z2;
drh9310ef22007-04-27 17:16:20 +0000427 int i, n;
drh1d34fde2009-02-03 15:50:33 +0000428 UNUSED_PARAMETER(argc);
drh7a521cf2007-04-25 18:23:52 +0000429 z2 = (char*)sqlite3_value_text(argv[0]);
drh1f0feef2007-05-15 13:27:07 +0000430 n = sqlite3_value_bytes(argv[0]);
431 /* Verify that the call to _bytes() does not invalidate the _text() pointer */
432 assert( z2==(char*)sqlite3_value_text(argv[0]) );
drh7a521cf2007-04-25 18:23:52 +0000433 if( z2 ){
drhb1a6c3c2008-03-20 16:30:17 +0000434 z1 = contextMalloc(context, ((i64)n)+1);
drh7a521cf2007-04-25 18:23:52 +0000435 if( z1 ){
drhdf901d32011-10-13 18:00:11 +0000436 for(i=0; i<n; i++){
437 z1[i] = (char)sqlite3Toupper(z2[i]);
drh7a521cf2007-04-25 18:23:52 +0000438 }
drhdf901d32011-10-13 18:00:11 +0000439 sqlite3_result_text(context, z1, n, sqlite3_free);
drh7a521cf2007-04-25 18:23:52 +0000440 }
drhdc04c582002-02-24 01:55:15 +0000441 }
442}
danielk19770ae8b832004-05-25 12:05:56 +0000443static void lowerFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drhdf901d32011-10-13 18:00:11 +0000444 char *z1;
drh7a521cf2007-04-25 18:23:52 +0000445 const char *z2;
drh9310ef22007-04-27 17:16:20 +0000446 int i, n;
drh1d34fde2009-02-03 15:50:33 +0000447 UNUSED_PARAMETER(argc);
drh7a521cf2007-04-25 18:23:52 +0000448 z2 = (char*)sqlite3_value_text(argv[0]);
drh1f0feef2007-05-15 13:27:07 +0000449 n = sqlite3_value_bytes(argv[0]);
450 /* Verify that the call to _bytes() does not invalidate the _text() pointer */
451 assert( z2==(char*)sqlite3_value_text(argv[0]) );
drh7a521cf2007-04-25 18:23:52 +0000452 if( z2 ){
drhb1a6c3c2008-03-20 16:30:17 +0000453 z1 = contextMalloc(context, ((i64)n)+1);
drh7a521cf2007-04-25 18:23:52 +0000454 if( z1 ){
drhdf901d32011-10-13 18:00:11 +0000455 for(i=0; i<n; i++){
456 z1[i] = sqlite3Tolower(z2[i]);
drh7a521cf2007-04-25 18:23:52 +0000457 }
drhdf901d32011-10-13 18:00:11 +0000458 sqlite3_result_text(context, z1, n, sqlite3_free);
drh7a521cf2007-04-25 18:23:52 +0000459 }
drhdc04c582002-02-24 01:55:15 +0000460 }
461}
462
drhae6bb952009-11-11 00:24:31 +0000463/*
drhcca9f3d2013-09-06 15:23:29 +0000464** Some functions like COALESCE() and IFNULL() and UNLIKELY() are implemented
465** as VDBE code so that unused argument values do not have to be computed.
466** However, we still need some kind of function implementation for this
467** routines in the function table. The noopFunc macro provides this.
468** noopFunc will never be called so it doesn't matter what the implementation
469** is. We might as well use the "version()" function as a substitute.
drhae6bb952009-11-11 00:24:31 +0000470*/
drhcca9f3d2013-09-06 15:23:29 +0000471#define noopFunc versionFunc /* Substitute function - never called */
drh3212e182002-02-28 00:46:26 +0000472
473/*
drhf9ffac92002-03-02 19:00:31 +0000474** Implementation of random(). Return a random integer.
475*/
drhf9b596e2004-05-26 16:54:42 +0000476static void randomFunc(
477 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000478 int NotUsed,
479 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000480){
drh52fc8492006-02-23 21:43:55 +0000481 sqlite_int64 r;
danielk197762c14b32008-11-19 09:05:26 +0000482 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drh2fa18682008-03-19 14:15:34 +0000483 sqlite3_randomness(sizeof(r), &r);
drh3034e3d2009-04-02 14:05:21 +0000484 if( r<0 ){
485 /* We need to prevent a random number of 0x8000000000000000
486 ** (or -9223372036854775808) since when you do abs() of that
487 ** number of you get the same value back again. To do this
488 ** in a way that is testable, mask the sign bit off of negative
489 ** values, resulting in a positive value. Then take the
490 ** 2s complement of that positive value. The end result can
491 ** therefore be no less than -9223372036854775807.
492 */
drhaf8001b2012-02-11 19:53:24 +0000493 r = -(r & LARGEST_INT64);
drh3034e3d2009-04-02 14:05:21 +0000494 }
drh52fc8492006-02-23 21:43:55 +0000495 sqlite3_result_int64(context, r);
drhf9ffac92002-03-02 19:00:31 +0000496}
497
498/*
drh137c7282007-01-29 17:58:28 +0000499** Implementation of randomblob(N). Return a random blob
500** that is N bytes long.
drh63cf66f2007-01-29 15:50:05 +0000501*/
drh137c7282007-01-29 17:58:28 +0000502static void randomBlob(
drh63cf66f2007-01-29 15:50:05 +0000503 sqlite3_context *context,
504 int argc,
505 sqlite3_value **argv
506){
drh137c7282007-01-29 17:58:28 +0000507 int n;
508 unsigned char *p;
drh63cf66f2007-01-29 15:50:05 +0000509 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000510 UNUSED_PARAMETER(argc);
drh63cf66f2007-01-29 15:50:05 +0000511 n = sqlite3_value_int(argv[0]);
drh023ae032007-05-08 12:12:16 +0000512 if( n<1 ){
513 n = 1;
514 }
danielk1977a1644fd2007-08-29 12:31:25 +0000515 p = contextMalloc(context, n);
drh02d85832007-05-07 19:31:15 +0000516 if( p ){
drh2fa18682008-03-19 14:15:34 +0000517 sqlite3_randomness(n, p);
drh17435752007-08-16 04:30:38 +0000518 sqlite3_result_blob(context, (char*)p, n, sqlite3_free);
drh02d85832007-05-07 19:31:15 +0000519 }
drh63cf66f2007-01-29 15:50:05 +0000520}
521
522/*
drh6ed41ad2002-04-06 14:10:47 +0000523** Implementation of the last_insert_rowid() SQL function. The return
danielk197724b03fd2004-05-10 10:34:34 +0000524** value is the same as the sqlite3_last_insert_rowid() API function.
drh6ed41ad2002-04-06 14:10:47 +0000525*/
danielk197751ad0ec2004-05-24 12:39:02 +0000526static void last_insert_rowid(
danielk19770ae8b832004-05-25 12:05:56 +0000527 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000528 int NotUsed,
529 sqlite3_value **NotUsed2
danielk197751ad0ec2004-05-24 12:39:02 +0000530){
drhfa4a4b92008-03-19 21:45:51 +0000531 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000532 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000533 /* IMP: R-51513-12026 The last_insert_rowid() SQL function is a
534 ** wrapper around the sqlite3_last_insert_rowid() C/C++ interface
535 ** function. */
drhf9b596e2004-05-26 16:54:42 +0000536 sqlite3_result_int64(context, sqlite3_last_insert_rowid(db));
drh6ed41ad2002-04-06 14:10:47 +0000537}
538
rdcf146a772004-02-25 22:51:06 +0000539/*
drhab2f1f92010-01-11 18:26:42 +0000540** Implementation of the changes() SQL function.
541**
542** IMP: R-62073-11209 The changes() SQL function is a wrapper
543** around the sqlite3_changes() C/C++ function and hence follows the same
544** rules for counting changes.
rdcf146a772004-02-25 22:51:06 +0000545*/
danielk1977b28af712004-06-21 06:50:26 +0000546static void changes(
drhf9b596e2004-05-26 16:54:42 +0000547 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000548 int NotUsed,
549 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000550){
drhfa4a4b92008-03-19 21:45:51 +0000551 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000552 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhf4479502004-05-27 03:12:53 +0000553 sqlite3_result_int(context, sqlite3_changes(db));
rdcb0c374f2004-02-20 22:53:38 +0000554}
rdcf146a772004-02-25 22:51:06 +0000555
556/*
danielk1977b28af712004-06-21 06:50:26 +0000557** Implementation of the total_changes() SQL function. The return value is
558** the same as the sqlite3_total_changes() API function.
rdcf146a772004-02-25 22:51:06 +0000559*/
danielk1977b28af712004-06-21 06:50:26 +0000560static void total_changes(
561 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000562 int NotUsed,
563 sqlite3_value **NotUsed2
danielk197751ad0ec2004-05-24 12:39:02 +0000564){
drhfa4a4b92008-03-19 21:45:51 +0000565 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000566 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000567 /* IMP: R-52756-41993 This function is a wrapper around the
568 ** sqlite3_total_changes() C/C++ interface. */
danielk1977b28af712004-06-21 06:50:26 +0000569 sqlite3_result_int(context, sqlite3_total_changes(db));
rdcb0c374f2004-02-20 22:53:38 +0000570}
571
drh6ed41ad2002-04-06 14:10:47 +0000572/*
drh4e5ffc52004-08-31 00:52:37 +0000573** A structure defining how to do GLOB-style comparisons.
danielk1977d02eb1f2004-06-06 09:44:03 +0000574*/
drh4e5ffc52004-08-31 00:52:37 +0000575struct compareInfo {
drh07e83472016-01-11 03:48:18 +0000576 u8 matchAll; /* "*" or "%" */
577 u8 matchOne; /* "?" or "_" */
578 u8 matchSet; /* "[" or 0 */
579 u8 noCase; /* true to ignore case differences */
danielk1977d02eb1f2004-06-06 09:44:03 +0000580};
drh55ef4d92005-08-14 01:20:37 +0000581
drhb9175ae2007-12-07 18:39:04 +0000582/*
583** For LIKE and GLOB matching on EBCDIC machines, assume that every
drhb0870482015-06-17 13:20:54 +0000584** character is exactly one byte in size. Also, provde the Utf8Read()
585** macro for fast reading of the next character in the common case where
586** the next character is ASCII.
drhb9175ae2007-12-07 18:39:04 +0000587*/
588#if defined(SQLITE_EBCDIC)
drh88b33222014-09-25 03:51:37 +0000589# define sqlite3Utf8Read(A) (*((*A)++))
drhb0870482015-06-17 13:20:54 +0000590# define Utf8Read(A) (*(A++))
drhb9175ae2007-12-07 18:39:04 +0000591#else
drhb0870482015-06-17 13:20:54 +0000592# define Utf8Read(A) (A[0]<0x80?*(A++):sqlite3Utf8Read(&A))
drhb9175ae2007-12-07 18:39:04 +0000593#endif
594
drh4e5ffc52004-08-31 00:52:37 +0000595static const struct compareInfo globInfo = { '*', '?', '[', 0 };
drh55ef4d92005-08-14 01:20:37 +0000596/* The correct SQL-92 behavior is for the LIKE operator to ignore
597** case. Thus 'a' LIKE 'A' would be true. */
598static const struct compareInfo likeInfoNorm = { '%', '_', 0, 1 };
599/* If SQLITE_CASE_SENSITIVE_LIKE is defined, then the LIKE operator
600** is case sensitive causing 'a' LIKE 'A' to be false */
601static const struct compareInfo likeInfoAlt = { '%', '_', 0, 0 };
danielk1977d02eb1f2004-06-06 09:44:03 +0000602
603/*
drh698a01c2016-12-01 18:49:40 +0000604** Possible error returns from patternMatch()
605*/
606#define SQLITE_MATCH 0
607#define SQLITE_NOMATCH 1
608#define SQLITE_NOWILDCARDMATCH 2
609
610/*
611** Compare two UTF-8 strings for equality where the first string is
612** a GLOB or LIKE expression. Return values:
613**
614** SQLITE_MATCH: Match
615** SQLITE_NOMATCH: No match
616** SQLITE_NOWILDCARDMATCH: No match in spite of having * or % wildcards.
drh0ac65892002-04-20 14:24:41 +0000617**
drh4e5ffc52004-08-31 00:52:37 +0000618** Globbing rules:
drh0ac65892002-04-20 14:24:41 +0000619**
drh4e5ffc52004-08-31 00:52:37 +0000620** '*' Matches any sequence of zero or more characters.
danielk1977d02eb1f2004-06-06 09:44:03 +0000621**
drh4e5ffc52004-08-31 00:52:37 +0000622** '?' Matches exactly one character.
623**
624** [...] Matches one character from the enclosed list of
625** characters.
626**
627** [^...] Matches one character not in the enclosed list.
628**
629** With the [...] and [^...] matching, a ']' character can be included
630** in the list by making it the first character after '[' or '^'. A
631** range of characters can be specified using '-'. Example:
632** "[a-z]" matches any single lower-case letter. To match a '-', make
633** it the last character in the list.
634**
drh9fdfdc82014-09-25 11:08:57 +0000635** Like matching rules:
636**
637** '%' Matches any sequence of zero or more characters
638**
639*** '_' Matches any one character
640**
641** Ec Where E is the "esc" character and c is any other
642** character, including '%', '_', and esc, match exactly c.
643**
drhb0870482015-06-17 13:20:54 +0000644** The comments within this routine usually assume glob matching.
drh9fdfdc82014-09-25 11:08:57 +0000645**
drh4e5ffc52004-08-31 00:52:37 +0000646** This routine is usually quick, but can be N**2 in the worst case.
drh0ac65892002-04-20 14:24:41 +0000647*/
danielk19777c6303c2004-11-17 16:41:29 +0000648static int patternCompare(
drh4e5ffc52004-08-31 00:52:37 +0000649 const u8 *zPattern, /* The glob pattern */
650 const u8 *zString, /* The string to compare against the glob */
danielk19777c6303c2004-11-17 16:41:29 +0000651 const struct compareInfo *pInfo, /* Information about how to do the compare */
drh698a01c2016-12-01 18:49:40 +0000652 u32 matchOther /* The escape char (LIKE) or '[' (GLOB) */
danielk197751ad0ec2004-05-24 12:39:02 +0000653){
drh9fdfdc82014-09-25 11:08:57 +0000654 u32 c, c2; /* Next pattern and input string chars */
655 u32 matchOne = pInfo->matchOne; /* "?" or "_" */
656 u32 matchAll = pInfo->matchAll; /* "*" or "%" */
drh9fdfdc82014-09-25 11:08:57 +0000657 u8 noCase = pInfo->noCase; /* True if uppercase==lowercase */
658 const u8 *zEscaped = 0; /* One past the last escaped input char */
drh88b33222014-09-25 03:51:37 +0000659
drhb0870482015-06-17 13:20:54 +0000660 while( (c = Utf8Read(zPattern))!=0 ){
drh9fdfdc82014-09-25 11:08:57 +0000661 if( c==matchAll ){ /* Match "*" */
662 /* Skip over multiple "*" characters in the pattern. If there
663 ** are also "?" characters, skip those as well, but consume a
664 ** single character of the input string for each "?" skipped */
drhb0870482015-06-17 13:20:54 +0000665 while( (c=Utf8Read(zPattern)) == matchAll || c == matchOne ){
drh42610962012-09-17 18:56:32 +0000666 if( c==matchOne && sqlite3Utf8Read(&zString)==0 ){
drh698a01c2016-12-01 18:49:40 +0000667 return SQLITE_NOWILDCARDMATCH;
drh4e5ffc52004-08-31 00:52:37 +0000668 }
danielk1977ad7dd422004-06-06 12:41:49 +0000669 }
drh66150952007-07-23 19:12:41 +0000670 if( c==0 ){
drh698a01c2016-12-01 18:49:40 +0000671 return SQLITE_MATCH; /* "*" at the end of the pattern matches */
drh88b33222014-09-25 03:51:37 +0000672 }else if( c==matchOther ){
drh07e83472016-01-11 03:48:18 +0000673 if( pInfo->matchSet==0 ){
drh88b33222014-09-25 03:51:37 +0000674 c = sqlite3Utf8Read(&zPattern);
drh698a01c2016-12-01 18:49:40 +0000675 if( c==0 ) return SQLITE_NOWILDCARDMATCH;
drh88b33222014-09-25 03:51:37 +0000676 }else{
drh9fdfdc82014-09-25 11:08:57 +0000677 /* "[...]" immediately follows the "*". We have to do a slow
678 ** recursive search in this case, but it is an unusual case. */
drh88b33222014-09-25 03:51:37 +0000679 assert( matchOther<0x80 ); /* '[' is a single-byte character */
drh698a01c2016-12-01 18:49:40 +0000680 while( *zString ){
681 int bMatch = patternCompare(&zPattern[-1],zString,pInfo,matchOther);
682 if( bMatch!=SQLITE_NOMATCH ) return bMatch;
drh88b33222014-09-25 03:51:37 +0000683 SQLITE_SKIP_UTF8(zString);
684 }
drh698a01c2016-12-01 18:49:40 +0000685 return SQLITE_NOWILDCARDMATCH;
drh66150952007-07-23 19:12:41 +0000686 }
drh66150952007-07-23 19:12:41 +0000687 }
drh9fdfdc82014-09-25 11:08:57 +0000688
689 /* At this point variable c contains the first character of the
690 ** pattern string past the "*". Search in the input string for the
dan1a4a7372016-12-01 17:34:59 +0000691 ** first matching character and recursively continue the match from
drh9fdfdc82014-09-25 11:08:57 +0000692 ** that point.
693 **
694 ** For a case-insensitive search, set variable cx to be the same as
695 ** c but in the other case and search the input string for either
696 ** c or cx.
697 */
698 if( c<=0x80 ){
699 u32 cx;
drh698a01c2016-12-01 18:49:40 +0000700 int bMatch;
drh9fdfdc82014-09-25 11:08:57 +0000701 if( noCase ){
702 cx = sqlite3Toupper(c);
703 c = sqlite3Tolower(c);
drh66150952007-07-23 19:12:41 +0000704 }else{
drh9fdfdc82014-09-25 11:08:57 +0000705 cx = c;
drh4e5ffc52004-08-31 00:52:37 +0000706 }
drh9fdfdc82014-09-25 11:08:57 +0000707 while( (c2 = *(zString++))!=0 ){
708 if( c2!=c && c2!=cx ) continue;
drh698a01c2016-12-01 18:49:40 +0000709 bMatch = patternCompare(zPattern,zString,pInfo,matchOther);
710 if( bMatch!=SQLITE_NOMATCH ) return bMatch;
drh9fdfdc82014-09-25 11:08:57 +0000711 }
712 }else{
drh698a01c2016-12-01 18:49:40 +0000713 int bMatch;
drhb0870482015-06-17 13:20:54 +0000714 while( (c2 = Utf8Read(zString))!=0 ){
drh9fdfdc82014-09-25 11:08:57 +0000715 if( c2!=c ) continue;
drh698a01c2016-12-01 18:49:40 +0000716 bMatch = patternCompare(zPattern,zString,pInfo,matchOther);
717 if( bMatch!=SQLITE_NOMATCH ) return bMatch;
drh9fdfdc82014-09-25 11:08:57 +0000718 }
drh66150952007-07-23 19:12:41 +0000719 }
drh698a01c2016-12-01 18:49:40 +0000720 return SQLITE_NOWILDCARDMATCH;
drh88b33222014-09-25 03:51:37 +0000721 }
drh88b33222014-09-25 03:51:37 +0000722 if( c==matchOther ){
drh07e83472016-01-11 03:48:18 +0000723 if( pInfo->matchSet==0 ){
drh88b33222014-09-25 03:51:37 +0000724 c = sqlite3Utf8Read(&zPattern);
drh698a01c2016-12-01 18:49:40 +0000725 if( c==0 ) return SQLITE_NOMATCH;
drh9fdfdc82014-09-25 11:08:57 +0000726 zEscaped = zPattern;
drh88b33222014-09-25 03:51:37 +0000727 }else{
728 u32 prior_c = 0;
drh9fdfdc82014-09-25 11:08:57 +0000729 int seen = 0;
730 int invert = 0;
drh88b33222014-09-25 03:51:37 +0000731 c = sqlite3Utf8Read(&zString);
drh698a01c2016-12-01 18:49:40 +0000732 if( c==0 ) return SQLITE_NOMATCH;
drh88b33222014-09-25 03:51:37 +0000733 c2 = sqlite3Utf8Read(&zPattern);
734 if( c2=='^' ){
735 invert = 1;
736 c2 = sqlite3Utf8Read(&zPattern);
737 }
738 if( c2==']' ){
739 if( c==']' ) seen = 1;
740 c2 = sqlite3Utf8Read(&zPattern);
741 }
742 while( c2 && c2!=']' ){
743 if( c2=='-' && zPattern[0]!=']' && zPattern[0]!=0 && prior_c>0 ){
744 c2 = sqlite3Utf8Read(&zPattern);
745 if( c>=prior_c && c<=c2 ) seen = 1;
746 prior_c = 0;
747 }else{
748 if( c==c2 ){
749 seen = 1;
750 }
751 prior_c = c2;
752 }
753 c2 = sqlite3Utf8Read(&zPattern);
754 }
755 if( c2==0 || (seen ^ invert)==0 ){
drh698a01c2016-12-01 18:49:40 +0000756 return SQLITE_NOMATCH;
drh88b33222014-09-25 03:51:37 +0000757 }
758 continue;
759 }
760 }
drhb0870482015-06-17 13:20:54 +0000761 c2 = Utf8Read(zString);
drh88b33222014-09-25 03:51:37 +0000762 if( c==c2 ) continue;
drhc80937a2016-06-06 01:48:14 +0000763 if( noCase && sqlite3Tolower(c)==sqlite3Tolower(c2) && c<0x80 && c2<0x80 ){
drh9fdfdc82014-09-25 11:08:57 +0000764 continue;
765 }
766 if( c==matchOne && zPattern!=zEscaped && c2!=0 ) continue;
drh698a01c2016-12-01 18:49:40 +0000767 return SQLITE_NOMATCH;
danielk197751ad0ec2004-05-24 12:39:02 +0000768 }
drh698a01c2016-12-01 18:49:40 +0000769 return *zString==0 ? SQLITE_MATCH : SQLITE_NOMATCH;
drh0ac65892002-04-20 14:24:41 +0000770}
drh4e5ffc52004-08-31 00:52:37 +0000771
drh55ef4d92005-08-14 01:20:37 +0000772/*
drh698a01c2016-12-01 18:49:40 +0000773** The sqlite3_strglob() interface. Return 0 on a match (like strcmp()) and
774** non-zero if there is no match.
drh56282a52013-04-10 16:13:38 +0000775*/
776int sqlite3_strglob(const char *zGlobPattern, const char *zString){
drh698a01c2016-12-01 18:49:40 +0000777 return patternCompare((u8*)zGlobPattern, (u8*)zString, &globInfo, '[');
drh56282a52013-04-10 16:13:38 +0000778}
779
780/*
drh698a01c2016-12-01 18:49:40 +0000781** The sqlite3_strlike() interface. Return 0 on a match and non-zero for
782** a miss - like strcmp().
drh8b4a94a2015-11-24 21:23:59 +0000783*/
784int sqlite3_strlike(const char *zPattern, const char *zStr, unsigned int esc){
drh698a01c2016-12-01 18:49:40 +0000785 return patternCompare((u8*)zPattern, (u8*)zStr, &likeInfoNorm, esc);
drh8b4a94a2015-11-24 21:23:59 +0000786}
787
788/*
drh55ef4d92005-08-14 01:20:37 +0000789** Count the number of times that the LIKE operator (or GLOB which is
790** just a variation of LIKE) gets called. This is used for testing
791** only.
792*/
793#ifdef SQLITE_TEST
794int sqlite3_like_count = 0;
795#endif
796
danielk19773f6b0872004-06-17 05:36:44 +0000797
798/*
799** Implementation of the like() SQL function. This function implements
800** the build-in LIKE operator. The first argument to the function is the
801** pattern and the second argument is the string. So, the SQL statements:
802**
803** A LIKE B
804**
805** is implemented as like(B,A).
806**
drh55ef4d92005-08-14 01:20:37 +0000807** This same function (with a different compareInfo structure) computes
808** the GLOB operator.
danielk19773f6b0872004-06-17 05:36:44 +0000809*/
810static void likeFunc(
811 sqlite3_context *context,
812 int argc,
813 sqlite3_value **argv
814){
drhbeb818d2007-05-08 15:34:47 +0000815 const unsigned char *zA, *zB;
drh07e83472016-01-11 03:48:18 +0000816 u32 escape;
drh27e62db2009-04-02 10:16:17 +0000817 int nPat;
drhbb4957f2008-03-20 14:03:29 +0000818 sqlite3 *db = sqlite3_context_db_handle(context);
drh07e83472016-01-11 03:48:18 +0000819 struct compareInfo *pInfo = sqlite3_user_data(context);
drhbeb818d2007-05-08 15:34:47 +0000820
drh41d2e662015-12-01 21:23:07 +0000821#ifdef SQLITE_LIKE_DOESNT_MATCH_BLOBS
822 if( sqlite3_value_type(argv[0])==SQLITE_BLOB
823 || sqlite3_value_type(argv[1])==SQLITE_BLOB
824 ){
825#ifdef SQLITE_TEST
826 sqlite3_like_count++;
827#endif
828 sqlite3_result_int(context, 0);
829 return;
830 }
831#endif
drh1f0feef2007-05-15 13:27:07 +0000832 zB = sqlite3_value_text(argv[0]);
833 zA = sqlite3_value_text(argv[1]);
834
drhbeb818d2007-05-08 15:34:47 +0000835 /* Limit the length of the LIKE or GLOB pattern to avoid problems
836 ** of deep recursion and N*N behavior in patternCompare().
837 */
drh27e62db2009-04-02 10:16:17 +0000838 nPat = sqlite3_value_bytes(argv[0]);
839 testcase( nPat==db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH] );
840 testcase( nPat==db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH]+1 );
841 if( nPat > db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH] ){
drhbeb818d2007-05-08 15:34:47 +0000842 sqlite3_result_error(context, "LIKE or GLOB pattern too complex", -1);
843 return;
844 }
drh1f0feef2007-05-15 13:27:07 +0000845 assert( zB==sqlite3_value_text(argv[0]) ); /* Encoding did not change */
drhbeb818d2007-05-08 15:34:47 +0000846
danielk19777c6303c2004-11-17 16:41:29 +0000847 if( argc==3 ){
848 /* The escape character string must consist of a single UTF-8 character.
849 ** Otherwise, return an error.
850 */
851 const unsigned char *zEsc = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +0000852 if( zEsc==0 ) return;
drhee858132007-05-08 20:37:38 +0000853 if( sqlite3Utf8CharLen((char*)zEsc, -1)!=1 ){
danielk19777c6303c2004-11-17 16:41:29 +0000854 sqlite3_result_error(context,
855 "ESCAPE expression must be a single character", -1);
856 return;
857 }
drh42610962012-09-17 18:56:32 +0000858 escape = sqlite3Utf8Read(&zEsc);
drh07e83472016-01-11 03:48:18 +0000859 }else{
860 escape = pInfo->matchSet;
danielk19777c6303c2004-11-17 16:41:29 +0000861 }
danielk19773f6b0872004-06-17 05:36:44 +0000862 if( zA && zB ){
drh55ef4d92005-08-14 01:20:37 +0000863#ifdef SQLITE_TEST
864 sqlite3_like_count++;
865#endif
drh698a01c2016-12-01 18:49:40 +0000866 sqlite3_result_int(context, patternCompare(zB, zA, pInfo, escape)==SQLITE_MATCH);
danielk197751ad0ec2004-05-24 12:39:02 +0000867 }
drh8912d102002-05-26 21:34:58 +0000868}
869
870/*
871** Implementation of the NULLIF(x,y) function. The result is the first
872** argument if the arguments are different. The result is NULL if the
873** arguments are equal to each other.
874*/
drhf9b596e2004-05-26 16:54:42 +0000875static void nullifFunc(
876 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000877 int NotUsed,
drhf9b596e2004-05-26 16:54:42 +0000878 sqlite3_value **argv
879){
danielk1977dc1bdc42004-06-11 10:51:27 +0000880 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk197762c14b32008-11-19 09:05:26 +0000881 UNUSED_PARAMETER(NotUsed);
danielk1977dc1bdc42004-06-11 10:51:27 +0000882 if( sqlite3MemCompare(argv[0], argv[1], pColl)!=0 ){
drhf4479502004-05-27 03:12:53 +0000883 sqlite3_result_value(context, argv[0]);
drh8912d102002-05-26 21:34:58 +0000884 }
drh0ac65892002-04-20 14:24:41 +0000885}
886
drh647cb0e2002-11-04 19:32:25 +0000887/*
drh47baebc2009-08-14 16:01:24 +0000888** Implementation of the sqlite_version() function. The result is the version
drh647cb0e2002-11-04 19:32:25 +0000889** of the SQLite library that is running.
890*/
drhf9b596e2004-05-26 16:54:42 +0000891static void versionFunc(
892 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000893 int NotUsed,
894 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000895){
danielk197762c14b32008-11-19 09:05:26 +0000896 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000897 /* IMP: R-48699-48617 This function is an SQL wrapper around the
898 ** sqlite3_libversion() C-interface. */
899 sqlite3_result_text(context, sqlite3_libversion(), -1, SQLITE_STATIC);
drh647cb0e2002-11-04 19:32:25 +0000900}
901
drh47baebc2009-08-14 16:01:24 +0000902/*
903** Implementation of the sqlite_source_id() function. The result is a string
904** that identifies the particular version of the source code used to build
905** SQLite.
906*/
907static void sourceidFunc(
908 sqlite3_context *context,
909 int NotUsed,
910 sqlite3_value **NotUsed2
911){
912 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000913 /* IMP: R-24470-31136 This function is an SQL wrapper around the
914 ** sqlite3_sourceid() C interface. */
915 sqlite3_result_text(context, sqlite3_sourceid(), -1, SQLITE_STATIC);
drh47baebc2009-08-14 16:01:24 +0000916}
917
shanehbdea6d12010-02-23 04:19:54 +0000918/*
drh3ca84ef2011-04-25 18:03:10 +0000919** Implementation of the sqlite_log() function. This is a wrapper around
920** sqlite3_log(). The return value is NULL. The function exists purely for
921** its side-effects.
922*/
drh840561f2011-04-27 18:08:42 +0000923static void errlogFunc(
drh3ca84ef2011-04-25 18:03:10 +0000924 sqlite3_context *context,
925 int argc,
926 sqlite3_value **argv
927){
928 UNUSED_PARAMETER(argc);
929 UNUSED_PARAMETER(context);
930 sqlite3_log(sqlite3_value_int(argv[0]), "%s", sqlite3_value_text(argv[1]));
931}
932
933/*
shanehdc97a8c2010-02-23 20:08:35 +0000934** Implementation of the sqlite_compileoption_used() function.
935** The result is an integer that identifies if the compiler option
936** was used to build SQLite.
shanehbdea6d12010-02-23 04:19:54 +0000937*/
shanehdc97a8c2010-02-23 20:08:35 +0000938#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
939static void compileoptionusedFunc(
shanehbdea6d12010-02-23 04:19:54 +0000940 sqlite3_context *context,
shanehdc97a8c2010-02-23 20:08:35 +0000941 int argc,
942 sqlite3_value **argv
shanehbdea6d12010-02-23 04:19:54 +0000943){
shanehdc97a8c2010-02-23 20:08:35 +0000944 const char *zOptName;
945 assert( argc==1 );
946 UNUSED_PARAMETER(argc);
drha3e414c2010-08-03 18:06:25 +0000947 /* IMP: R-39564-36305 The sqlite_compileoption_used() SQL
948 ** function is a wrapper around the sqlite3_compileoption_used() C/C++
949 ** function.
950 */
drh264a2d42010-02-25 15:28:41 +0000951 if( (zOptName = (const char*)sqlite3_value_text(argv[0]))!=0 ){
shanehdc97a8c2010-02-23 20:08:35 +0000952 sqlite3_result_int(context, sqlite3_compileoption_used(zOptName));
shanehdc97a8c2010-02-23 20:08:35 +0000953 }
shanehbdea6d12010-02-23 04:19:54 +0000954}
shanehdc97a8c2010-02-23 20:08:35 +0000955#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
956
957/*
958** Implementation of the sqlite_compileoption_get() function.
959** The result is a string that identifies the compiler options
960** used to build SQLite.
961*/
962#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
963static void compileoptiongetFunc(
964 sqlite3_context *context,
965 int argc,
966 sqlite3_value **argv
967){
968 int n;
969 assert( argc==1 );
970 UNUSED_PARAMETER(argc);
drha3e414c2010-08-03 18:06:25 +0000971 /* IMP: R-04922-24076 The sqlite_compileoption_get() SQL function
972 ** is a wrapper around the sqlite3_compileoption_get() C/C++ function.
973 */
shanehdc97a8c2010-02-23 20:08:35 +0000974 n = sqlite3_value_int(argv[0]);
975 sqlite3_result_text(context, sqlite3_compileoption_get(n), -1, SQLITE_STATIC);
976}
977#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
shanehbdea6d12010-02-23 04:19:54 +0000978
drh137c7282007-01-29 17:58:28 +0000979/* Array for converting from half-bytes (nybbles) into ASCII hex
980** digits. */
981static const char hexdigits[] = {
982 '0', '1', '2', '3', '4', '5', '6', '7',
983 '8', '9', 'A', 'B', 'C', 'D', 'E', 'F'
984};
danielk1977d641d642004-11-18 15:44:29 +0000985
drh47394702003-08-20 01:03:33 +0000986/*
drh47394702003-08-20 01:03:33 +0000987** Implementation of the QUOTE() function. This function takes a single
988** argument. If the argument is numeric, the return value is the same as
989** the argument. If the argument is NULL, the return value is the string
990** "NULL". Otherwise, the argument is enclosed in single quotes with
991** single-quote escapes.
992*/
danielk19770ae8b832004-05-25 12:05:56 +0000993static void quoteFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drha0df4cc2009-02-02 17:29:59 +0000994 assert( argc==1 );
drh1d34fde2009-02-03 15:50:33 +0000995 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +0000996 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +0000997 case SQLITE_FLOAT: {
drh72b3fbc2012-06-19 03:11:25 +0000998 double r1, r2;
drh72b3fbc2012-06-19 03:11:25 +0000999 char zBuf[50];
mistachkin2b434a72012-06-19 04:36:48 +00001000 r1 = sqlite3_value_double(argv[0]);
drh72b3fbc2012-06-19 03:11:25 +00001001 sqlite3_snprintf(sizeof(zBuf), zBuf, "%!.15g", r1);
1002 sqlite3AtoF(zBuf, &r2, 20, SQLITE_UTF8);
1003 if( r1!=r2 ){
1004 sqlite3_snprintf(sizeof(zBuf), zBuf, "%!.20e", r1);
1005 }
1006 sqlite3_result_text(context, zBuf, -1, SQLITE_TRANSIENT);
1007 break;
1008 }
1009 case SQLITE_INTEGER: {
drhf4479502004-05-27 03:12:53 +00001010 sqlite3_result_value(context, argv[0]);
drhf9b596e2004-05-26 16:54:42 +00001011 break;
1012 }
danielk19773f41e972004-06-08 00:39:01 +00001013 case SQLITE_BLOB: {
danielk19773f41e972004-06-08 00:39:01 +00001014 char *zText = 0;
danielk19773f41e972004-06-08 00:39:01 +00001015 char const *zBlob = sqlite3_value_blob(argv[0]);
drh1f0feef2007-05-15 13:27:07 +00001016 int nBlob = sqlite3_value_bytes(argv[0]);
1017 assert( zBlob==sqlite3_value_blob(argv[0]) ); /* No encoding change */
drhb1a6c3c2008-03-20 16:30:17 +00001018 zText = (char *)contextMalloc(context, (2*(i64)nBlob)+4);
danielk1977a1644fd2007-08-29 12:31:25 +00001019 if( zText ){
danielk19773f41e972004-06-08 00:39:01 +00001020 int i;
1021 for(i=0; i<nBlob; i++){
1022 zText[(i*2)+2] = hexdigits[(zBlob[i]>>4)&0x0F];
1023 zText[(i*2)+3] = hexdigits[(zBlob[i])&0x0F];
1024 }
1025 zText[(nBlob*2)+2] = '\'';
1026 zText[(nBlob*2)+3] = '\0';
1027 zText[0] = 'X';
1028 zText[1] = '\'';
danielk1977d8123362004-06-12 09:25:12 +00001029 sqlite3_result_text(context, zText, -1, SQLITE_TRANSIENT);
drh17435752007-08-16 04:30:38 +00001030 sqlite3_free(zText);
danielk19773f41e972004-06-08 00:39:01 +00001031 }
1032 break;
1033 }
drh9c054832004-05-31 18:51:57 +00001034 case SQLITE_TEXT: {
drh023ae032007-05-08 12:12:16 +00001035 int i,j;
1036 u64 n;
drh2646da72005-12-09 20:02:05 +00001037 const unsigned char *zArg = sqlite3_value_text(argv[0]);
drhf9b596e2004-05-26 16:54:42 +00001038 char *z;
1039
drh7a521cf2007-04-25 18:23:52 +00001040 if( zArg==0 ) return;
drh023ae032007-05-08 12:12:16 +00001041 for(i=0, n=0; zArg[i]; i++){ if( zArg[i]=='\'' ) n++; }
drhb1a6c3c2008-03-20 16:30:17 +00001042 z = contextMalloc(context, ((i64)i)+((i64)n)+3);
danielk1977a1644fd2007-08-29 12:31:25 +00001043 if( z ){
1044 z[0] = '\'';
1045 for(i=0, j=1; zArg[i]; i++){
1046 z[j++] = zArg[i];
1047 if( zArg[i]=='\'' ){
1048 z[j++] = '\'';
1049 }
drhf9b596e2004-05-26 16:54:42 +00001050 }
danielk1977a1644fd2007-08-29 12:31:25 +00001051 z[j++] = '\'';
1052 z[j] = 0;
1053 sqlite3_result_text(context, z, j, sqlite3_free);
drhf9b596e2004-05-26 16:54:42 +00001054 }
drha0df4cc2009-02-02 17:29:59 +00001055 break;
1056 }
1057 default: {
1058 assert( sqlite3_value_type(argv[0])==SQLITE_NULL );
1059 sqlite3_result_text(context, "NULL", 4, SQLITE_STATIC);
1060 break;
drhf9b596e2004-05-26 16:54:42 +00001061 }
drh47394702003-08-20 01:03:33 +00001062 }
1063}
1064
drh137c7282007-01-29 17:58:28 +00001065/*
drhd495d8c2013-02-22 19:34:25 +00001066** The unicode() function. Return the integer unicode code-point value
1067** for the first character of the input string.
1068*/
1069static void unicodeFunc(
1070 sqlite3_context *context,
1071 int argc,
1072 sqlite3_value **argv
1073){
1074 const unsigned char *z = sqlite3_value_text(argv[0]);
drh1d59d032013-03-01 23:40:26 +00001075 (void)argc;
drhd495d8c2013-02-22 19:34:25 +00001076 if( z && z[0] ) sqlite3_result_int(context, sqlite3Utf8Read(&z));
1077}
1078
1079/*
1080** The char() function takes zero or more arguments, each of which is
1081** an integer. It constructs a string where each character of the string
1082** is the unicode character for the corresponding integer argument.
1083*/
1084static void charFunc(
1085 sqlite3_context *context,
1086 int argc,
1087 sqlite3_value **argv
1088){
1089 unsigned char *z, *zOut;
1090 int i;
drhf3cdcdc2015-04-29 16:50:28 +00001091 zOut = z = sqlite3_malloc64( argc*4+1 );
drhd495d8c2013-02-22 19:34:25 +00001092 if( z==0 ){
1093 sqlite3_result_error_nomem(context);
1094 return;
1095 }
1096 for(i=0; i<argc; i++){
mistachkinc9545442013-02-26 05:42:30 +00001097 sqlite3_int64 x;
drhd495d8c2013-02-22 19:34:25 +00001098 unsigned c;
1099 x = sqlite3_value_int64(argv[i]);
1100 if( x<0 || x>0x10ffff ) x = 0xfffd;
1101 c = (unsigned)(x & 0x1fffff);
drhfe7a5d12013-03-07 14:00:04 +00001102 if( c<0x00080 ){
1103 *zOut++ = (u8)(c&0xFF);
1104 }else if( c<0x00800 ){
1105 *zOut++ = 0xC0 + (u8)((c>>6)&0x1F);
1106 *zOut++ = 0x80 + (u8)(c & 0x3F);
1107 }else if( c<0x10000 ){
1108 *zOut++ = 0xE0 + (u8)((c>>12)&0x0F);
1109 *zOut++ = 0x80 + (u8)((c>>6) & 0x3F);
1110 *zOut++ = 0x80 + (u8)(c & 0x3F);
drhd495d8c2013-02-22 19:34:25 +00001111 }else{
drhfe7a5d12013-03-07 14:00:04 +00001112 *zOut++ = 0xF0 + (u8)((c>>18) & 0x07);
1113 *zOut++ = 0x80 + (u8)((c>>12) & 0x3F);
1114 *zOut++ = 0x80 + (u8)((c>>6) & 0x3F);
1115 *zOut++ = 0x80 + (u8)(c & 0x3F);
1116 } \
drhd495d8c2013-02-22 19:34:25 +00001117 }
drhbbf483f2014-09-09 20:30:24 +00001118 sqlite3_result_text64(context, (char*)z, zOut-z, sqlite3_free, SQLITE_UTF8);
drhd495d8c2013-02-22 19:34:25 +00001119}
1120
1121/*
drh137c7282007-01-29 17:58:28 +00001122** The hex() function. Interpret the argument as a blob. Return
1123** a hexadecimal rendering as text.
1124*/
1125static void hexFunc(
1126 sqlite3_context *context,
1127 int argc,
1128 sqlite3_value **argv
1129){
1130 int i, n;
1131 const unsigned char *pBlob;
1132 char *zHex, *z;
1133 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001134 UNUSED_PARAMETER(argc);
drh1f0feef2007-05-15 13:27:07 +00001135 pBlob = sqlite3_value_blob(argv[0]);
drh137c7282007-01-29 17:58:28 +00001136 n = sqlite3_value_bytes(argv[0]);
drh1f0feef2007-05-15 13:27:07 +00001137 assert( pBlob==sqlite3_value_blob(argv[0]) ); /* No encoding change */
drhb1a6c3c2008-03-20 16:30:17 +00001138 z = zHex = contextMalloc(context, ((i64)n)*2 + 1);
danielk1977a1644fd2007-08-29 12:31:25 +00001139 if( zHex ){
1140 for(i=0; i<n; i++, pBlob++){
1141 unsigned char c = *pBlob;
1142 *(z++) = hexdigits[(c>>4)&0xf];
1143 *(z++) = hexdigits[c&0xf];
1144 }
1145 *z = 0;
1146 sqlite3_result_text(context, zHex, n*2, sqlite3_free);
drh137c7282007-01-29 17:58:28 +00001147 }
drh137c7282007-01-29 17:58:28 +00001148}
1149
drh26b6d902007-03-17 13:27:54 +00001150/*
drh8cff3822007-05-02 02:08:28 +00001151** The zeroblob(N) function returns a zero-filled blob of size N bytes.
1152*/
1153static void zeroblobFunc(
1154 sqlite3_context *context,
1155 int argc,
1156 sqlite3_value **argv
1157){
drh98640a32007-06-07 19:08:32 +00001158 i64 n;
dana4d5ae82015-07-24 16:24:37 +00001159 int rc;
drh8cff3822007-05-02 02:08:28 +00001160 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001161 UNUSED_PARAMETER(argc);
drh98640a32007-06-07 19:08:32 +00001162 n = sqlite3_value_int64(argv[0]);
dana4d5ae82015-07-24 16:24:37 +00001163 if( n<0 ) n = 0;
1164 rc = sqlite3_result_zeroblob64(context, n); /* IMP: R-00293-64994 */
1165 if( rc ){
1166 sqlite3_result_error_code(context, rc);
drh98640a32007-06-07 19:08:32 +00001167 }
drh8cff3822007-05-02 02:08:28 +00001168}
1169
1170/*
drh26b6d902007-03-17 13:27:54 +00001171** The replace() function. Three arguments are all strings: call
1172** them A, B, and C. The result is also a string which is derived
drhf7b54962013-05-28 12:11:54 +00001173** from A by replacing every occurrence of B with C. The match
drh26b6d902007-03-17 13:27:54 +00001174** must be exact. Collating sequences are not used.
1175*/
1176static void replaceFunc(
1177 sqlite3_context *context,
1178 int argc,
1179 sqlite3_value **argv
1180){
1181 const unsigned char *zStr; /* The input string A */
1182 const unsigned char *zPattern; /* The pattern string B */
1183 const unsigned char *zRep; /* The replacement string C */
1184 unsigned char *zOut; /* The output */
1185 int nStr; /* Size of zStr */
1186 int nPattern; /* Size of zPattern */
1187 int nRep; /* Size of zRep */
drh2e6400b2007-05-08 15:46:18 +00001188 i64 nOut; /* Maximum size of zOut */
drh26b6d902007-03-17 13:27:54 +00001189 int loopLimit; /* Last zStr[] that might match zPattern[] */
1190 int i, j; /* Loop counters */
1191
1192 assert( argc==3 );
danielk1977f3d3c272008-11-19 16:52:44 +00001193 UNUSED_PARAMETER(argc);
drh26b6d902007-03-17 13:27:54 +00001194 zStr = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001195 if( zStr==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001196 nStr = sqlite3_value_bytes(argv[0]);
1197 assert( zStr==sqlite3_value_text(argv[0]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001198 zPattern = sqlite3_value_text(argv[1]);
drha605fe82009-02-01 18:08:40 +00001199 if( zPattern==0 ){
drh23336062009-02-03 13:19:12 +00001200 assert( sqlite3_value_type(argv[1])==SQLITE_NULL
1201 || sqlite3_context_db_handle(context)->mallocFailed );
drha605fe82009-02-01 18:08:40 +00001202 return;
1203 }
1204 if( zPattern[0]==0 ){
1205 assert( sqlite3_value_type(argv[1])!=SQLITE_NULL );
1206 sqlite3_result_value(context, argv[0]);
1207 return;
1208 }
drh1f0feef2007-05-15 13:27:07 +00001209 nPattern = sqlite3_value_bytes(argv[1]);
1210 assert( zPattern==sqlite3_value_text(argv[1]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001211 zRep = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +00001212 if( zRep==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001213 nRep = sqlite3_value_bytes(argv[2]);
1214 assert( zRep==sqlite3_value_text(argv[2]) );
drh2e6400b2007-05-08 15:46:18 +00001215 nOut = nStr + 1;
1216 assert( nOut<SQLITE_MAX_LENGTH );
drhb1a6c3c2008-03-20 16:30:17 +00001217 zOut = contextMalloc(context, (i64)nOut);
drh2e6400b2007-05-08 15:46:18 +00001218 if( zOut==0 ){
1219 return;
drh26b6d902007-03-17 13:27:54 +00001220 }
drh26b6d902007-03-17 13:27:54 +00001221 loopLimit = nStr - nPattern;
1222 for(i=j=0; i<=loopLimit; i++){
1223 if( zStr[i]!=zPattern[0] || memcmp(&zStr[i], zPattern, nPattern) ){
1224 zOut[j++] = zStr[i];
1225 }else{
drh4a50aac2007-08-23 02:47:53 +00001226 u8 *zOld;
drhbb4957f2008-03-20 14:03:29 +00001227 sqlite3 *db = sqlite3_context_db_handle(context);
drh2e6400b2007-05-08 15:46:18 +00001228 nOut += nRep - nPattern;
drh27e62db2009-04-02 10:16:17 +00001229 testcase( nOut-1==db->aLimit[SQLITE_LIMIT_LENGTH] );
1230 testcase( nOut-2==db->aLimit[SQLITE_LIMIT_LENGTH] );
1231 if( nOut-1>db->aLimit[SQLITE_LIMIT_LENGTH] ){
drha0206bc2007-05-08 15:15:02 +00001232 sqlite3_result_error_toobig(context);
drhb9755982010-07-24 16:34:37 +00001233 sqlite3_free(zOut);
danielk197717374e82007-05-08 14:39:04 +00001234 return;
1235 }
drh4a50aac2007-08-23 02:47:53 +00001236 zOld = zOut;
drhf3cdcdc2015-04-29 16:50:28 +00001237 zOut = sqlite3_realloc64(zOut, (int)nOut);
drh2e6400b2007-05-08 15:46:18 +00001238 if( zOut==0 ){
danielk1977a1644fd2007-08-29 12:31:25 +00001239 sqlite3_result_error_nomem(context);
drhb9755982010-07-24 16:34:37 +00001240 sqlite3_free(zOld);
drh2e6400b2007-05-08 15:46:18 +00001241 return;
1242 }
drh26b6d902007-03-17 13:27:54 +00001243 memcpy(&zOut[j], zRep, nRep);
1244 j += nRep;
1245 i += nPattern-1;
1246 }
1247 }
drh2e6400b2007-05-08 15:46:18 +00001248 assert( j+nStr-i+1==nOut );
drh26b6d902007-03-17 13:27:54 +00001249 memcpy(&zOut[j], &zStr[i], nStr-i);
1250 j += nStr - i;
1251 assert( j<=nOut );
1252 zOut[j] = 0;
1253 sqlite3_result_text(context, (char*)zOut, j, sqlite3_free);
1254}
1255
drh309b3382007-03-17 17:52:42 +00001256/*
1257** Implementation of the TRIM(), LTRIM(), and RTRIM() functions.
1258** The userdata is 0x1 for left trim, 0x2 for right trim, 0x3 for both.
1259*/
1260static void trimFunc(
1261 sqlite3_context *context,
1262 int argc,
1263 sqlite3_value **argv
1264){
1265 const unsigned char *zIn; /* Input string */
1266 const unsigned char *zCharSet; /* Set of characters to trim */
1267 int nIn; /* Number of bytes in input */
drh7209c692008-04-27 18:40:11 +00001268 int flags; /* 1: trimleft 2: trimright 3: trim */
drhd1e3a612007-04-27 21:59:52 +00001269 int i; /* Loop counter */
drh1bd10f82008-12-10 21:19:56 +00001270 unsigned char *aLen = 0; /* Length of each character in zCharSet */
1271 unsigned char **azChar = 0; /* Individual characters in zCharSet */
drhd1e3a612007-04-27 21:59:52 +00001272 int nChar; /* Number of characters in zCharSet */
1273
drh309b3382007-03-17 17:52:42 +00001274 if( sqlite3_value_type(argv[0])==SQLITE_NULL ){
1275 return;
1276 }
1277 zIn = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001278 if( zIn==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001279 nIn = sqlite3_value_bytes(argv[0]);
1280 assert( zIn==sqlite3_value_text(argv[0]) );
drh309b3382007-03-17 17:52:42 +00001281 if( argc==1 ){
drhd1e3a612007-04-27 21:59:52 +00001282 static const unsigned char lenOne[] = { 1 };
danielk1977a4de4532008-09-02 15:44:08 +00001283 static unsigned char * const azOne[] = { (u8*)" " };
drhd1e3a612007-04-27 21:59:52 +00001284 nChar = 1;
drh8cff3822007-05-02 02:08:28 +00001285 aLen = (u8*)lenOne;
danielk1977bc67da42007-12-11 04:23:19 +00001286 azChar = (unsigned char **)azOne;
drhd1e3a612007-04-27 21:59:52 +00001287 zCharSet = 0;
drh7a521cf2007-04-25 18:23:52 +00001288 }else if( (zCharSet = sqlite3_value_text(argv[1]))==0 ){
drh309b3382007-03-17 17:52:42 +00001289 return;
drhd1e3a612007-04-27 21:59:52 +00001290 }else{
1291 const unsigned char *z;
1292 for(z=zCharSet, nChar=0; *z; nChar++){
drh4a919112007-05-15 11:55:09 +00001293 SQLITE_SKIP_UTF8(z);
drhd1e3a612007-04-27 21:59:52 +00001294 }
1295 if( nChar>0 ){
drhb1a6c3c2008-03-20 16:30:17 +00001296 azChar = contextMalloc(context, ((i64)nChar)*(sizeof(char*)+1));
drhd1e3a612007-04-27 21:59:52 +00001297 if( azChar==0 ){
1298 return;
1299 }
1300 aLen = (unsigned char*)&azChar[nChar];
1301 for(z=zCharSet, nChar=0; *z; nChar++){
danielk1977bc67da42007-12-11 04:23:19 +00001302 azChar[nChar] = (unsigned char *)z;
drh4a919112007-05-15 11:55:09 +00001303 SQLITE_SKIP_UTF8(z);
drh3abbd392008-12-10 23:04:13 +00001304 aLen[nChar] = (u8)(z - azChar[nChar]);
drhd1e3a612007-04-27 21:59:52 +00001305 }
1306 }
drh309b3382007-03-17 17:52:42 +00001307 }
drhd1e3a612007-04-27 21:59:52 +00001308 if( nChar>0 ){
shane1fc41292008-07-08 22:28:48 +00001309 flags = SQLITE_PTR_TO_INT(sqlite3_user_data(context));
drh309b3382007-03-17 17:52:42 +00001310 if( flags & 1 ){
drhd1e3a612007-04-27 21:59:52 +00001311 while( nIn>0 ){
drh1bd10f82008-12-10 21:19:56 +00001312 int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001313 for(i=0; i<nChar; i++){
1314 len = aLen[i];
drh27e62db2009-04-02 10:16:17 +00001315 if( len<=nIn && memcmp(zIn, azChar[i], len)==0 ) break;
drhd1e3a612007-04-27 21:59:52 +00001316 }
1317 if( i>=nChar ) break;
1318 zIn += len;
1319 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001320 }
1321 }
1322 if( flags & 2 ){
drhd1e3a612007-04-27 21:59:52 +00001323 while( nIn>0 ){
drh1bd10f82008-12-10 21:19:56 +00001324 int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001325 for(i=0; i<nChar; i++){
1326 len = aLen[i];
1327 if( len<=nIn && memcmp(&zIn[nIn-len],azChar[i],len)==0 ) break;
1328 }
1329 if( i>=nChar ) break;
1330 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001331 }
1332 }
drhd1e3a612007-04-27 21:59:52 +00001333 if( zCharSet ){
1334 sqlite3_free(azChar);
1335 }
drh309b3382007-03-17 17:52:42 +00001336 }
1337 sqlite3_result_text(context, (char*)zIn, nIn, SQLITE_TRANSIENT);
1338}
drh26b6d902007-03-17 13:27:54 +00001339
danielk1977a4de4532008-09-02 15:44:08 +00001340
drhcc153132016-08-04 12:35:17 +00001341#ifdef SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION
1342/*
1343** The "unknown" function is automatically substituted in place of
1344** any unrecognized function name when doing an EXPLAIN or EXPLAIN QUERY PLAN
1345** when the SQLITE_ENABLE_UNKNOWN_FUNCTION compile-time option is used.
1346** When the "sqlite3" command-line shell is built using this functionality,
1347** that allows an EXPLAIN or EXPLAIN QUERY PLAN for complex queries
1348** involving application-defined functions to be examined in a generic
1349** sqlite3 shell.
1350*/
1351static void unknownFunc(
1352 sqlite3_context *context,
1353 int argc,
1354 sqlite3_value **argv
1355){
1356 /* no-op */
1357}
1358#endif /*SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION*/
1359
1360
drh2ba3ccc2009-12-08 02:06:08 +00001361/* IMP: R-25361-16150 This function is omitted from SQLite by default. It
1362** is only available if the SQLITE_SOUNDEX compile-time option is used
1363** when SQLite is built.
1364*/
drhd24cc422003-03-27 12:51:24 +00001365#ifdef SQLITE_SOUNDEX
1366/*
1367** Compute the soundex encoding of a word.
drh2ba3ccc2009-12-08 02:06:08 +00001368**
1369** IMP: R-59782-00072 The soundex(X) function returns a string that is the
1370** soundex encoding of the string X.
drhd24cc422003-03-27 12:51:24 +00001371*/
drh137c7282007-01-29 17:58:28 +00001372static void soundexFunc(
1373 sqlite3_context *context,
1374 int argc,
1375 sqlite3_value **argv
1376){
drhd24cc422003-03-27 12:51:24 +00001377 char zResult[8];
drh4c755c02004-08-08 20:22:17 +00001378 const u8 *zIn;
drhd24cc422003-03-27 12:51:24 +00001379 int i, j;
1380 static const unsigned char iCode[] = {
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, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1385 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1386 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1387 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1388 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1389 };
1390 assert( argc==1 );
drh4c755c02004-08-08 20:22:17 +00001391 zIn = (u8*)sqlite3_value_text(argv[0]);
drhbdf67e02006-08-19 11:34:01 +00001392 if( zIn==0 ) zIn = (u8*)"";
drhdc86e2b2009-01-24 11:30:42 +00001393 for(i=0; zIn[i] && !sqlite3Isalpha(zIn[i]); i++){}
drhd24cc422003-03-27 12:51:24 +00001394 if( zIn[i] ){
drhbdf67e02006-08-19 11:34:01 +00001395 u8 prevcode = iCode[zIn[i]&0x7f];
danielk197778ca0e72009-01-20 16:53:39 +00001396 zResult[0] = sqlite3Toupper(zIn[i]);
drhd24cc422003-03-27 12:51:24 +00001397 for(j=1; j<4 && zIn[i]; i++){
1398 int code = iCode[zIn[i]&0x7f];
1399 if( code>0 ){
drhbdf67e02006-08-19 11:34:01 +00001400 if( code!=prevcode ){
1401 prevcode = code;
1402 zResult[j++] = code + '0';
1403 }
1404 }else{
1405 prevcode = 0;
drhd24cc422003-03-27 12:51:24 +00001406 }
1407 }
1408 while( j<4 ){
1409 zResult[j++] = '0';
1410 }
1411 zResult[j] = 0;
danielk1977d8123362004-06-12 09:25:12 +00001412 sqlite3_result_text(context, zResult, 4, SQLITE_TRANSIENT);
drhd24cc422003-03-27 12:51:24 +00001413 }else{
drh2ba3ccc2009-12-08 02:06:08 +00001414 /* IMP: R-64894-50321 The string "?000" is returned if the argument
1415 ** is NULL or contains no ASCII alphabetic characters. */
danielk1977d8123362004-06-12 09:25:12 +00001416 sqlite3_result_text(context, "?000", 4, SQLITE_STATIC);
drhd24cc422003-03-27 12:51:24 +00001417 }
1418}
drh2ba3ccc2009-12-08 02:06:08 +00001419#endif /* SQLITE_SOUNDEX */
drhd24cc422003-03-27 12:51:24 +00001420
drhfdb83b22006-06-17 14:12:47 +00001421#ifndef SQLITE_OMIT_LOAD_EXTENSION
1422/*
1423** A function that loads a shared-library extension then returns NULL.
1424*/
1425static void loadExt(sqlite3_context *context, int argc, sqlite3_value **argv){
danielk197765fd59f2006-06-24 11:51:33 +00001426 const char *zFile = (const char *)sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001427 const char *zProc;
drhfa4a4b92008-03-19 21:45:51 +00001428 sqlite3 *db = sqlite3_context_db_handle(context);
drhfdb83b22006-06-17 14:12:47 +00001429 char *zErrMsg = 0;
1430
drh191dd062016-04-21 01:30:09 +00001431 /* Disallow the load_extension() SQL function unless the SQLITE_LoadExtFunc
drh1a55ded2016-04-20 00:30:05 +00001432 ** flag is set. See the sqlite3_enable_load_extension() API.
1433 */
drhf602a162016-04-21 01:58:21 +00001434 if( (db->flags & SQLITE_LoadExtFunc)==0 ){
1435 sqlite3_result_error(context, "not authorized", -1);
1436 return;
1437 }
drh1a55ded2016-04-20 00:30:05 +00001438
drhfdb83b22006-06-17 14:12:47 +00001439 if( argc==2 ){
danielk197765fd59f2006-06-24 11:51:33 +00001440 zProc = (const char *)sqlite3_value_text(argv[1]);
drh7a521cf2007-04-25 18:23:52 +00001441 }else{
1442 zProc = 0;
drhfdb83b22006-06-17 14:12:47 +00001443 }
drh7a521cf2007-04-25 18:23:52 +00001444 if( zFile && sqlite3_load_extension(db, zFile, zProc, &zErrMsg) ){
drhfdb83b22006-06-17 14:12:47 +00001445 sqlite3_result_error(context, zErrMsg, -1);
1446 sqlite3_free(zErrMsg);
1447 }
1448}
1449#endif
1450
danielk197701427a62005-01-11 13:02:33 +00001451
drh0ac65892002-04-20 14:24:41 +00001452/*
drhd3a149e2002-02-24 17:12:53 +00001453** An instance of the following structure holds the context of a
drhdd5baa92002-02-27 19:50:59 +00001454** sum() or avg() aggregate computation.
1455*/
1456typedef struct SumCtx SumCtx;
1457struct SumCtx {
drh8c08e862006-02-11 17:34:00 +00001458 double rSum; /* Floating point sum */
1459 i64 iSum; /* Integer sum */
1460 i64 cnt; /* Number of elements summed */
1461 u8 overflow; /* True if integer overflow seen */
1462 u8 approx; /* True if non-integer value was input to the sum */
drhdd5baa92002-02-27 19:50:59 +00001463};
1464
1465/*
drha97fdd32006-01-12 22:17:50 +00001466** Routines used to compute the sum, average, and total.
1467**
1468** The SUM() function follows the (broken) SQL standard which means
1469** that it returns NULL if it sums over no inputs. TOTAL returns
1470** 0.0 in that case. In addition, TOTAL always returns a float where
1471** SUM might return an integer if it never encounters a floating point
drhc806d852006-05-11 13:25:39 +00001472** value. TOTAL never fails, but SUM might through an exception if
1473** it overflows an integer.
drhdd5baa92002-02-27 19:50:59 +00001474*/
danielk19770ae8b832004-05-25 12:05:56 +00001475static void sumStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drhdd5baa92002-02-27 19:50:59 +00001476 SumCtx *p;
drh3d1d95e2005-09-08 10:37:01 +00001477 int type;
drh3f219f42005-09-08 19:45:57 +00001478 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001479 UNUSED_PARAMETER(argc);
drh4f26d6c2004-05-26 23:25:30 +00001480 p = sqlite3_aggregate_context(context, sizeof(*p));
drh29d72102006-02-09 22:13:41 +00001481 type = sqlite3_value_numeric_type(argv[0]);
drh3d1d95e2005-09-08 10:37:01 +00001482 if( p && type!=SQLITE_NULL ){
drh739105c2002-05-29 23:22:23 +00001483 p->cnt++;
drh29d72102006-02-09 22:13:41 +00001484 if( type==SQLITE_INTEGER ){
drh8c08e862006-02-11 17:34:00 +00001485 i64 v = sqlite3_value_int64(argv[0]);
1486 p->rSum += v;
drh158b9cb2011-03-05 20:59:46 +00001487 if( (p->approx|p->overflow)==0 && sqlite3AddInt64(&p->iSum, v) ){
1488 p->overflow = 1;
drh29d72102006-02-09 22:13:41 +00001489 }
1490 }else{
drh8c08e862006-02-11 17:34:00 +00001491 p->rSum += sqlite3_value_double(argv[0]);
drh29d72102006-02-09 22:13:41 +00001492 p->approx = 1;
drh3f219f42005-09-08 19:45:57 +00001493 }
drh739105c2002-05-29 23:22:23 +00001494 }
drhdd5baa92002-02-27 19:50:59 +00001495}
danielk19770ae8b832004-05-25 12:05:56 +00001496static void sumFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001497 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001498 p = sqlite3_aggregate_context(context, 0);
drhc2bd9132005-09-08 20:37:43 +00001499 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001500 if( p->overflow ){
1501 sqlite3_result_error(context,"integer overflow",-1);
1502 }else if( p->approx ){
1503 sqlite3_result_double(context, p->rSum);
drhc2bd9132005-09-08 20:37:43 +00001504 }else{
drh8c08e862006-02-11 17:34:00 +00001505 sqlite3_result_int64(context, p->iSum);
drhc2bd9132005-09-08 20:37:43 +00001506 }
drh3d1d95e2005-09-08 10:37:01 +00001507 }
drhdd5baa92002-02-27 19:50:59 +00001508}
danielk19770ae8b832004-05-25 12:05:56 +00001509static void avgFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001510 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001511 p = sqlite3_aggregate_context(context, 0);
drh739105c2002-05-29 23:22:23 +00001512 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001513 sqlite3_result_double(context, p->rSum/(double)p->cnt);
drhdd5baa92002-02-27 19:50:59 +00001514 }
1515}
drha97fdd32006-01-12 22:17:50 +00001516static void totalFinalize(sqlite3_context *context){
1517 SumCtx *p;
1518 p = sqlite3_aggregate_context(context, 0);
shanefbd60f82009-02-04 03:59:25 +00001519 /* (double)0 In case of SQLITE_OMIT_FLOATING_POINT... */
1520 sqlite3_result_double(context, p ? p->rSum : (double)0);
drha97fdd32006-01-12 22:17:50 +00001521}
drhdd5baa92002-02-27 19:50:59 +00001522
1523/*
drh0bce8352002-02-28 00:41:10 +00001524** The following structure keeps track of state information for the
1525** count() aggregate function.
1526*/
1527typedef struct CountCtx CountCtx;
1528struct CountCtx {
drhfc6ad392006-02-09 13:38:19 +00001529 i64 n;
drh0bce8352002-02-28 00:41:10 +00001530};
drhdd5baa92002-02-27 19:50:59 +00001531
drh0bce8352002-02-28 00:41:10 +00001532/*
1533** Routines to implement the count() aggregate function.
1534*/
danielk19770ae8b832004-05-25 12:05:56 +00001535static void countStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drh0bce8352002-02-28 00:41:10 +00001536 CountCtx *p;
drh4f26d6c2004-05-26 23:25:30 +00001537 p = sqlite3_aggregate_context(context, sizeof(*p));
drh9c054832004-05-31 18:51:57 +00001538 if( (argc==0 || SQLITE_NULL!=sqlite3_value_type(argv[0])) && p ){
drh0bce8352002-02-28 00:41:10 +00001539 p->n++;
1540 }
drh2e79c3d2009-04-08 23:04:14 +00001541
drhd3264c72009-04-15 13:39:47 +00001542#ifndef SQLITE_OMIT_DEPRECATED
drh2e79c3d2009-04-08 23:04:14 +00001543 /* The sqlite3_aggregate_count() function is deprecated. But just to make
1544 ** sure it still operates correctly, verify that its count agrees with our
1545 ** internal count when using count(*) and when the total count can be
1546 ** expressed as a 32-bit integer. */
1547 assert( argc==1 || p==0 || p->n>0x7fffffff
1548 || p->n==sqlite3_aggregate_count(context) );
drhd3264c72009-04-15 13:39:47 +00001549#endif
drh0bce8352002-02-28 00:41:10 +00001550}
danielk19770ae8b832004-05-25 12:05:56 +00001551static void countFinalize(sqlite3_context *context){
drh0bce8352002-02-28 00:41:10 +00001552 CountCtx *p;
drhabfcea22005-09-06 20:36:48 +00001553 p = sqlite3_aggregate_context(context, 0);
drhfc6ad392006-02-09 13:38:19 +00001554 sqlite3_result_int64(context, p ? p->n : 0);
drh0bce8352002-02-28 00:41:10 +00001555}
1556
1557/*
drh0bce8352002-02-28 00:41:10 +00001558** Routines to implement min() and max() aggregate functions.
1559*/
danielk197762c14b32008-11-19 09:05:26 +00001560static void minmaxStep(
1561 sqlite3_context *context,
1562 int NotUsed,
1563 sqlite3_value **argv
1564){
danielk197788208052004-05-25 01:13:20 +00001565 Mem *pArg = (Mem *)argv[0];
drh9eb516c2004-07-18 20:52:32 +00001566 Mem *pBest;
danielk197762c14b32008-11-19 09:05:26 +00001567 UNUSED_PARAMETER(NotUsed);
drh9eb516c2004-07-18 20:52:32 +00001568
drh9eb516c2004-07-18 20:52:32 +00001569 pBest = (Mem *)sqlite3_aggregate_context(context, sizeof(*pBest));
danielk19773aeab9e2004-06-24 00:20:04 +00001570 if( !pBest ) return;
drh268380c2004-02-25 13:47:31 +00001571
drh94a6d992012-02-02 18:42:09 +00001572 if( sqlite3_value_type(argv[0])==SQLITE_NULL ){
1573 if( pBest->flags ) sqlite3SkipAccumulatorLoad(context);
1574 }else if( pBest->flags ){
drh9eb516c2004-07-18 20:52:32 +00001575 int max;
1576 int cmp;
danielk1977dc1bdc42004-06-11 10:51:27 +00001577 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk19777e18c252004-05-25 11:47:24 +00001578 /* This step function is used for both the min() and max() aggregates,
1579 ** the only difference between the two being that the sense of the
1580 ** comparison is inverted. For the max() aggregate, the
1581 ** sqlite3_user_data() function returns (void *)-1. For min() it
1582 ** returns (void *)db, where db is the sqlite3* database pointer.
1583 ** Therefore the next statement sets variable 'max' to 1 for the max()
1584 ** aggregate, or 0 for min().
1585 */
drh309b3382007-03-17 17:52:42 +00001586 max = sqlite3_user_data(context)!=0;
danielk1977dc1bdc42004-06-11 10:51:27 +00001587 cmp = sqlite3MemCompare(pBest, pArg, pColl);
danielk197788208052004-05-25 01:13:20 +00001588 if( (max && cmp<0) || (!max && cmp>0) ){
drhb21c8cd2007-08-21 19:33:56 +00001589 sqlite3VdbeMemCopy(pBest, pArg);
drh7a957892012-02-02 17:35:43 +00001590 }else{
1591 sqlite3SkipAccumulatorLoad(context);
danielk197788208052004-05-25 01:13:20 +00001592 }
drh268380c2004-02-25 13:47:31 +00001593 }else{
drh035e5632014-09-16 14:16:31 +00001594 pBest->db = sqlite3_context_db_handle(context);
drhb21c8cd2007-08-21 19:33:56 +00001595 sqlite3VdbeMemCopy(pBest, pArg);
drh0bce8352002-02-28 00:41:10 +00001596 }
1597}
danielk19770ae8b832004-05-25 12:05:56 +00001598static void minMaxFinalize(sqlite3_context *context){
danielk197788208052004-05-25 01:13:20 +00001599 sqlite3_value *pRes;
drhabfcea22005-09-06 20:36:48 +00001600 pRes = (sqlite3_value *)sqlite3_aggregate_context(context, 0);
1601 if( pRes ){
drh94a6d992012-02-02 18:42:09 +00001602 if( pRes->flags ){
drhabfcea22005-09-06 20:36:48 +00001603 sqlite3_result_value(context, pRes);
1604 }
1605 sqlite3VdbeMemRelease(pRes);
drh0bce8352002-02-28 00:41:10 +00001606 }
1607}
drhdd5baa92002-02-27 19:50:59 +00001608
drhb0689692007-11-01 17:38:30 +00001609/*
1610** group_concat(EXPR, ?SEPARATOR?)
1611*/
1612static void groupConcatStep(
1613 sqlite3_context *context,
1614 int argc,
1615 sqlite3_value **argv
1616){
1617 const char *zVal;
drhade86482007-11-28 22:36:40 +00001618 StrAccum *pAccum;
drhb0689692007-11-01 17:38:30 +00001619 const char *zSep;
drh07d31172009-02-02 21:57:05 +00001620 int nVal, nSep;
1621 assert( argc==1 || argc==2 );
1622 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drhade86482007-11-28 22:36:40 +00001623 pAccum = (StrAccum*)sqlite3_aggregate_context(context, sizeof(*pAccum));
1624
1625 if( pAccum ){
drhbb4957f2008-03-20 14:03:29 +00001626 sqlite3 *db = sqlite3_context_db_handle(context);
drhc0490572015-05-02 11:45:53 +00001627 int firstTerm = pAccum->mxAlloc==0;
drhbb4957f2008-03-20 14:03:29 +00001628 pAccum->mxAlloc = db->aLimit[SQLITE_LIMIT_LENGTH];
drh8bfd7192009-06-19 16:44:41 +00001629 if( !firstTerm ){
drh07d31172009-02-02 21:57:05 +00001630 if( argc==2 ){
1631 zSep = (char*)sqlite3_value_text(argv[1]);
1632 nSep = sqlite3_value_bytes(argv[1]);
drhb0689692007-11-01 17:38:30 +00001633 }else{
1634 zSep = ",";
drhade86482007-11-28 22:36:40 +00001635 nSep = 1;
drhb0689692007-11-01 17:38:30 +00001636 }
dan3043b532016-12-30 17:40:14 +00001637 if( zSep ) sqlite3StrAccumAppend(pAccum, zSep, nSep);
drhb0689692007-11-01 17:38:30 +00001638 }
drh07d31172009-02-02 21:57:05 +00001639 zVal = (char*)sqlite3_value_text(argv[0]);
1640 nVal = sqlite3_value_bytes(argv[0]);
drh8009c9b2014-05-07 20:24:00 +00001641 if( zVal ) sqlite3StrAccumAppend(pAccum, zVal, nVal);
drhb0689692007-11-01 17:38:30 +00001642 }
1643}
1644static void groupConcatFinalize(sqlite3_context *context){
drhade86482007-11-28 22:36:40 +00001645 StrAccum *pAccum;
1646 pAccum = sqlite3_aggregate_context(context, 0);
1647 if( pAccum ){
drhb49bc862013-08-21 21:12:10 +00001648 if( pAccum->accError==STRACCUM_TOOBIG ){
drhade86482007-11-28 22:36:40 +00001649 sqlite3_result_error_toobig(context);
drhb49bc862013-08-21 21:12:10 +00001650 }else if( pAccum->accError==STRACCUM_NOMEM ){
drhade86482007-11-28 22:36:40 +00001651 sqlite3_result_error_nomem(context);
1652 }else{
1653 sqlite3_result_text(context, sqlite3StrAccumFinish(pAccum), -1,
1654 sqlite3_free);
1655 }
drhb0689692007-11-01 17:38:30 +00001656 }
1657}
drh4e5ffc52004-08-31 00:52:37 +00001658
drhd3a149e2002-02-24 17:12:53 +00001659/*
drha4741842010-04-25 20:58:37 +00001660** This routine does per-connection function registration. Most
1661** of the built-in functions above are part of the global function set.
1662** This routine only deals with those that are not global.
drhdc04c582002-02-24 01:55:15 +00001663*/
drh80738d92016-02-15 00:34:16 +00001664void sqlite3RegisterPerConnectionBuiltinFunctions(sqlite3 *db){
drha4741842010-04-25 20:58:37 +00001665 int rc = sqlite3_overload_function(db, "MATCH", 2);
1666 assert( rc==SQLITE_NOMEM || rc==SQLITE_OK );
1667 if( rc==SQLITE_NOMEM ){
drh4a642b62016-02-05 01:55:27 +00001668 sqlite3OomFault(db);
danielk1977832a58a2007-06-22 15:21:15 +00001669 }
drh55ef4d92005-08-14 01:20:37 +00001670}
1671
1672/*
1673** Set the LIKEOPT flag on the 2-argument function with the given name.
1674*/
drh1bd10f82008-12-10 21:19:56 +00001675static void setLikeOptFlag(sqlite3 *db, const char *zName, u8 flagVal){
drh55ef4d92005-08-14 01:20:37 +00001676 FuncDef *pDef;
drh80738d92016-02-15 00:34:16 +00001677 pDef = sqlite3FindFunction(db, zName, 2, SQLITE_UTF8, 0);
drhd27135a2009-04-02 13:36:37 +00001678 if( ALWAYS(pDef) ){
drhd36e1042013-09-06 13:10:12 +00001679 pDef->funcFlags |= flagVal;
drh55ef4d92005-08-14 01:20:37 +00001680 }
1681}
1682
1683/*
1684** Register the built-in LIKE and GLOB functions. The caseSensitive
1685** parameter determines whether or not the LIKE operator is case
1686** sensitive. GLOB is always case sensitive.
1687*/
1688void sqlite3RegisterLikeFunctions(sqlite3 *db, int caseSensitive){
1689 struct compareInfo *pInfo;
1690 if( caseSensitive ){
1691 pInfo = (struct compareInfo*)&likeInfoAlt;
1692 }else{
1693 pInfo = (struct compareInfo*)&likeInfoNorm;
1694 }
drh901e9942010-12-15 18:54:37 +00001695 sqlite3CreateFunc(db, "like", 2, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0);
1696 sqlite3CreateFunc(db, "like", 3, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0);
1697 sqlite3CreateFunc(db, "glob", 2, SQLITE_UTF8,
dand2199f02010-08-27 17:48:52 +00001698 (struct compareInfo*)&globInfo, likeFunc, 0, 0, 0);
drhd64fe2f2005-08-28 17:00:23 +00001699 setLikeOptFlag(db, "glob", SQLITE_FUNC_LIKE | SQLITE_FUNC_CASE);
1700 setLikeOptFlag(db, "like",
1701 caseSensitive ? (SQLITE_FUNC_LIKE | SQLITE_FUNC_CASE) : SQLITE_FUNC_LIKE);
drh55ef4d92005-08-14 01:20:37 +00001702}
1703
1704/*
1705** pExpr points to an expression which implements a function. If
1706** it is appropriate to apply the LIKE optimization to that function
1707** then set aWc[0] through aWc[2] to the wildcard characters and
1708** return TRUE. If the function is not a LIKE-style function then
1709** return FALSE.
drh16897072015-03-07 00:57:37 +00001710**
1711** *pIsNocase is set to true if uppercase and lowercase are equivalent for
1712** the function (default for LIKE). If the function makes the distinction
1713** between uppercase and lowercase (as does GLOB) then *pIsNocase is set to
1714** false.
drh55ef4d92005-08-14 01:20:37 +00001715*/
drhd64fe2f2005-08-28 17:00:23 +00001716int sqlite3IsLikeFunction(sqlite3 *db, Expr *pExpr, int *pIsNocase, char *aWc){
drh55ef4d92005-08-14 01:20:37 +00001717 FuncDef *pDef;
danielk19776ab3a2e2009-02-19 14:39:25 +00001718 if( pExpr->op!=TK_FUNCTION
1719 || !pExpr->x.pList
1720 || pExpr->x.pList->nExpr!=2
1721 ){
drh55ef4d92005-08-14 01:20:37 +00001722 return 0;
1723 }
danielk19776ab3a2e2009-02-19 14:39:25 +00001724 assert( !ExprHasProperty(pExpr, EP_xIsSelect) );
drh80738d92016-02-15 00:34:16 +00001725 pDef = sqlite3FindFunction(db, pExpr->u.zToken, 2, SQLITE_UTF8, 0);
drhd36e1042013-09-06 13:10:12 +00001726 if( NEVER(pDef==0) || (pDef->funcFlags & SQLITE_FUNC_LIKE)==0 ){
drh55ef4d92005-08-14 01:20:37 +00001727 return 0;
1728 }
1729
1730 /* The memcpy() statement assumes that the wildcard characters are
1731 ** the first three statements in the compareInfo structure. The
1732 ** asserts() that follow verify that assumption
1733 */
1734 memcpy(aWc, pDef->pUserData, 3);
1735 assert( (char*)&likeInfoAlt == (char*)&likeInfoAlt.matchAll );
1736 assert( &((char*)&likeInfoAlt)[1] == (char*)&likeInfoAlt.matchOne );
1737 assert( &((char*)&likeInfoAlt)[2] == (char*)&likeInfoAlt.matchSet );
drhd36e1042013-09-06 13:10:12 +00001738 *pIsNocase = (pDef->funcFlags & SQLITE_FUNC_CASE)==0;
drh55ef4d92005-08-14 01:20:37 +00001739 return 1;
drhdc04c582002-02-24 01:55:15 +00001740}
danielk19778c0a7912008-08-20 14:49:23 +00001741
drh70a8ca32008-08-21 18:49:27 +00001742/*
peter.d.reid60ec9142014-09-06 16:39:46 +00001743** All of the FuncDef structures in the aBuiltinFunc[] array above
drh777c5382008-08-21 20:21:34 +00001744** to the global function hash table. This occurs at start-time (as
1745** a consequence of calling sqlite3_initialize()).
1746**
1747** After this routine runs
drh70a8ca32008-08-21 18:49:27 +00001748*/
drh80738d92016-02-15 00:34:16 +00001749void sqlite3RegisterBuiltinFunctions(void){
danielk197793ce7412008-09-01 19:14:02 +00001750 /*
1751 ** The following array holds FuncDef structures for all of the functions
1752 ** defined in this file.
1753 **
1754 ** The array cannot be constant since changes are made to the
1755 ** FuncDef.pHash elements at start-time. The elements of this array
1756 ** are read-only after initialization is complete.
drh80738d92016-02-15 00:34:16 +00001757 **
1758 ** For peak efficiency, put the most frequently used function last.
danielk197793ce7412008-09-01 19:14:02 +00001759 */
drh80738d92016-02-15 00:34:16 +00001760 static FuncDef aBuiltinFunc[] = {
1761#ifdef SQLITE_SOUNDEX
1762 FUNCTION(soundex, 1, 0, 0, soundexFunc ),
1763#endif
1764#ifndef SQLITE_OMIT_LOAD_EXTENSION
1765 VFUNCTION(load_extension, 1, 0, 0, loadExt ),
1766 VFUNCTION(load_extension, 2, 0, 0, loadExt ),
1767#endif
1768#if SQLITE_USER_AUTHENTICATION
1769 FUNCTION(sqlite_crypt, 2, 0, 0, sqlite3CryptFunc ),
1770#endif
1771#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
1772 DFUNCTION(sqlite_compileoption_used,1, 0, 0, compileoptionusedFunc ),
1773 DFUNCTION(sqlite_compileoption_get, 1, 0, 0, compileoptiongetFunc ),
1774#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
1775 FUNCTION2(unlikely, 1, 0, 0, noopFunc, SQLITE_FUNC_UNLIKELY),
1776 FUNCTION2(likelihood, 2, 0, 0, noopFunc, SQLITE_FUNC_UNLIKELY),
1777 FUNCTION2(likely, 1, 0, 0, noopFunc, SQLITE_FUNC_UNLIKELY),
drh54240752017-01-03 14:39:30 +00001778#ifdef SQLITE_DEBUG
drha1a523a2016-12-26 00:18:36 +00001779 FUNCTION2(affinity, 1, 0, 0, noopFunc, SQLITE_FUNC_AFFINITY),
drh54240752017-01-03 14:39:30 +00001780#endif
danielk197793ce7412008-09-01 19:14:02 +00001781 FUNCTION(ltrim, 1, 1, 0, trimFunc ),
1782 FUNCTION(ltrim, 2, 1, 0, trimFunc ),
1783 FUNCTION(rtrim, 1, 2, 0, trimFunc ),
1784 FUNCTION(rtrim, 2, 2, 0, trimFunc ),
1785 FUNCTION(trim, 1, 3, 0, trimFunc ),
1786 FUNCTION(trim, 2, 3, 0, trimFunc ),
1787 FUNCTION(min, -1, 0, 1, minmaxFunc ),
1788 FUNCTION(min, 0, 0, 1, 0 ),
drh9588ad92014-09-15 14:46:02 +00001789 AGGREGATE2(min, 1, 0, 1, minmaxStep, minMaxFinalize,
1790 SQLITE_FUNC_MINMAX ),
danielk197793ce7412008-09-01 19:14:02 +00001791 FUNCTION(max, -1, 1, 1, minmaxFunc ),
1792 FUNCTION(max, 0, 1, 1, 0 ),
drh9588ad92014-09-15 14:46:02 +00001793 AGGREGATE2(max, 1, 1, 1, minmaxStep, minMaxFinalize,
1794 SQLITE_FUNC_MINMAX ),
drha748fdc2012-03-28 01:34:47 +00001795 FUNCTION2(typeof, 1, 0, 0, typeofFunc, SQLITE_FUNC_TYPEOF),
1796 FUNCTION2(length, 1, 0, 0, lengthFunc, SQLITE_FUNC_LENGTH),
drhd55e0722012-10-25 03:07:29 +00001797 FUNCTION(instr, 2, 0, 0, instrFunc ),
drha5c14162013-12-17 15:03:06 +00001798 FUNCTION(printf, -1, 0, 0, printfFunc ),
drhd495d8c2013-02-22 19:34:25 +00001799 FUNCTION(unicode, 1, 0, 0, unicodeFunc ),
1800 FUNCTION(char, -1, 0, 0, charFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001801 FUNCTION(abs, 1, 0, 0, absFunc ),
shanefbd60f82009-02-04 03:59:25 +00001802#ifndef SQLITE_OMIT_FLOATING_POINT
danielk197793ce7412008-09-01 19:14:02 +00001803 FUNCTION(round, 1, 0, 0, roundFunc ),
1804 FUNCTION(round, 2, 0, 0, roundFunc ),
shanefbd60f82009-02-04 03:59:25 +00001805#endif
danielk197793ce7412008-09-01 19:14:02 +00001806 FUNCTION(upper, 1, 0, 0, upperFunc ),
1807 FUNCTION(lower, 1, 0, 0, lowerFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001808 FUNCTION(hex, 1, 0, 0, hexFunc ),
drhcca9f3d2013-09-06 15:23:29 +00001809 FUNCTION2(ifnull, 2, 0, 0, noopFunc, SQLITE_FUNC_COALESCE),
drhb1fba282013-11-21 14:33:48 +00001810 VFUNCTION(random, 0, 0, 0, randomFunc ),
1811 VFUNCTION(randomblob, 1, 0, 0, randomBlob ),
danielk197793ce7412008-09-01 19:14:02 +00001812 FUNCTION(nullif, 2, 0, 1, nullifFunc ),
drh03bf26d2015-08-31 21:16:36 +00001813 DFUNCTION(sqlite_version, 0, 0, 0, versionFunc ),
1814 DFUNCTION(sqlite_source_id, 0, 0, 0, sourceidFunc ),
drh840561f2011-04-27 18:08:42 +00001815 FUNCTION(sqlite_log, 2, 0, 0, errlogFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001816 FUNCTION(quote, 1, 0, 0, quoteFunc ),
drhb1fba282013-11-21 14:33:48 +00001817 VFUNCTION(last_insert_rowid, 0, 0, 0, last_insert_rowid),
1818 VFUNCTION(changes, 0, 0, 0, changes ),
1819 VFUNCTION(total_changes, 0, 0, 0, total_changes ),
danielk197793ce7412008-09-01 19:14:02 +00001820 FUNCTION(replace, 3, 0, 0, replaceFunc ),
1821 FUNCTION(zeroblob, 1, 0, 0, zeroblobFunc ),
drh80738d92016-02-15 00:34:16 +00001822 FUNCTION(substr, 2, 0, 0, substrFunc ),
1823 FUNCTION(substr, 3, 0, 0, substrFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001824 AGGREGATE(sum, 1, 0, 0, sumStep, sumFinalize ),
1825 AGGREGATE(total, 1, 0, 0, sumStep, totalFinalize ),
1826 AGGREGATE(avg, 1, 0, 0, sumStep, avgFinalize ),
drh9588ad92014-09-15 14:46:02 +00001827 AGGREGATE2(count, 0, 0, 0, countStep, countFinalize,
1828 SQLITE_FUNC_COUNT ),
danielk197793ce7412008-09-01 19:14:02 +00001829 AGGREGATE(count, 1, 0, 0, countStep, countFinalize ),
drh07d31172009-02-02 21:57:05 +00001830 AGGREGATE(group_concat, 1, 0, 0, groupConcatStep, groupConcatFinalize),
1831 AGGREGATE(group_concat, 2, 0, 0, groupConcatStep, groupConcatFinalize),
danielk197793ce7412008-09-01 19:14:02 +00001832
1833 LIKEFUNC(glob, 2, &globInfo, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
drhcc153132016-08-04 12:35:17 +00001834#ifdef SQLITE_CASE_SENSITIVE_LIKE
danielk197793ce7412008-09-01 19:14:02 +00001835 LIKEFUNC(like, 2, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
1836 LIKEFUNC(like, 3, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
drhcc153132016-08-04 12:35:17 +00001837#else
danielk197793ce7412008-09-01 19:14:02 +00001838 LIKEFUNC(like, 2, &likeInfoNorm, SQLITE_FUNC_LIKE),
1839 LIKEFUNC(like, 3, &likeInfoNorm, SQLITE_FUNC_LIKE),
drhcc153132016-08-04 12:35:17 +00001840#endif
1841#ifdef SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION
1842 FUNCTION(unknown, -1, 0, 0, unknownFunc ),
1843#endif
drh80738d92016-02-15 00:34:16 +00001844 FUNCTION(coalesce, 1, 0, 0, 0 ),
1845 FUNCTION(coalesce, 0, 0, 0, 0 ),
1846 FUNCTION2(coalesce, -1, 0, 0, noopFunc, SQLITE_FUNC_COALESCE),
danielk197793ce7412008-09-01 19:14:02 +00001847 };
drh545f5872010-04-24 14:02:59 +00001848#ifndef SQLITE_OMIT_ALTERTABLE
1849 sqlite3AlterFunctions();
1850#endif
dan0106e372013-08-12 16:34:32 +00001851#if defined(SQLITE_ENABLE_STAT3) || defined(SQLITE_ENABLE_STAT4)
1852 sqlite3AnalyzeFunctions();
1853#endif
drh80738d92016-02-15 00:34:16 +00001854 sqlite3RegisterDateTimeFunctions();
1855 sqlite3InsertBuiltinFuncs(aBuiltinFunc, ArraySize(aBuiltinFunc));
1856
1857#if 0 /* Enable to print out how the built-in functions are hashed */
1858 {
1859 int i;
1860 FuncDef *p;
1861 for(i=0; i<SQLITE_FUNC_HASH_SZ; i++){
1862 printf("FUNC-HASH %02d:", i);
1863 for(p=sqlite3BuiltinFunctions.a[i]; p; p=p->u.pHash){
1864 int n = sqlite3Strlen30(p->zName);
1865 int h = p->zName[0] + n;
1866 printf(" %s(%d)", p->zName, h);
1867 }
1868 printf("\n");
1869 }
1870 }
1871#endif
drh70a8ca32008-08-21 18:49:27 +00001872}