blob: 30990a30f3f0b5bd3988838c2c97a59825327257 [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){
drha15cc472014-09-25 13:17:30 +000025 VdbeOp *pOp = &context->pVdbe->aOp[context->iOp-1];
26 assert( pOp->opcode==OP_CollSeq );
27 assert( pOp->p4type==P4_COLLSEQ );
28 return pOp->p4.pColl;
danielk1977dc1bdc42004-06-11 10:51:27 +000029}
30
drh0bce8352002-02-28 00:41:10 +000031/*
drh7a957892012-02-02 17:35:43 +000032** Indicate that the accumulator load should be skipped on this
33** iteration of the aggregate loop.
34*/
35static void sqlite3SkipAccumulatorLoad(sqlite3_context *context){
36 context->skipFlag = 1;
37}
38
39/*
drh0bce8352002-02-28 00:41:10 +000040** Implementation of the non-aggregate min() and max() functions
41*/
drhf9b596e2004-05-26 16:54:42 +000042static void minmaxFunc(
43 sqlite3_context *context,
44 int argc,
45 sqlite3_value **argv
46){
drh0bce8352002-02-28 00:41:10 +000047 int i;
drh268380c2004-02-25 13:47:31 +000048 int mask; /* 0 for min() or 0xffffffff for max() */
drhf9b596e2004-05-26 16:54:42 +000049 int iBest;
danielk1977dc1bdc42004-06-11 10:51:27 +000050 CollSeq *pColl;
drh0bce8352002-02-28 00:41:10 +000051
drh65595cd2009-02-02 16:32:55 +000052 assert( argc>1 );
drhc44af712004-09-02 15:53:56 +000053 mask = sqlite3_user_data(context)==0 ? 0 : -1;
danielk1977dc1bdc42004-06-11 10:51:27 +000054 pColl = sqlite3GetFuncCollSeq(context);
55 assert( pColl );
danielk1977c572ef72004-05-27 09:28:41 +000056 assert( mask==-1 || mask==0 );
drhf9b596e2004-05-26 16:54:42 +000057 iBest = 0;
drh9c054832004-05-31 18:51:57 +000058 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drhf9b596e2004-05-26 16:54:42 +000059 for(i=1; i<argc; i++){
drh9c054832004-05-31 18:51:57 +000060 if( sqlite3_value_type(argv[i])==SQLITE_NULL ) return;
danielk1977dc1bdc42004-06-11 10:51:27 +000061 if( (sqlite3MemCompare(argv[iBest], argv[i], pColl)^mask)>=0 ){
drh65595cd2009-02-02 16:32:55 +000062 testcase( mask==0 );
drhf9b596e2004-05-26 16:54:42 +000063 iBest = i;
drh0bce8352002-02-28 00:41:10 +000064 }
65 }
drhf4479502004-05-27 03:12:53 +000066 sqlite3_result_value(context, argv[iBest]);
drh0bce8352002-02-28 00:41:10 +000067}
drh0bce8352002-02-28 00:41:10 +000068
drh268380c2004-02-25 13:47:31 +000069/*
70** Return the type of the argument.
71*/
drhf9b596e2004-05-26 16:54:42 +000072static void typeofFunc(
73 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +000074 int NotUsed,
drhf9b596e2004-05-26 16:54:42 +000075 sqlite3_value **argv
76){
danielk197735bb9d02004-05-24 12:55:54 +000077 const char *z = 0;
danielk197762c14b32008-11-19 09:05:26 +000078 UNUSED_PARAMETER(NotUsed);
danielk197735bb9d02004-05-24 12:55:54 +000079 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +000080 case SQLITE_INTEGER: z = "integer"; break;
81 case SQLITE_TEXT: z = "text"; break;
82 case SQLITE_FLOAT: z = "real"; break;
83 case SQLITE_BLOB: z = "blob"; break;
drh65595cd2009-02-02 16:32:55 +000084 default: z = "null"; break;
danielk197735bb9d02004-05-24 12:55:54 +000085 }
danielk1977d8123362004-06-12 09:25:12 +000086 sqlite3_result_text(context, z, -1, SQLITE_STATIC);
drh0bce8352002-02-28 00:41:10 +000087}
88
drh5708d2d2005-06-22 10:53:59 +000089
90/*
drh0bce8352002-02-28 00:41:10 +000091** Implementation of the length() function
92*/
drhf9b596e2004-05-26 16:54:42 +000093static void lengthFunc(
94 sqlite3_context *context,
95 int argc,
96 sqlite3_value **argv
97){
drh0bce8352002-02-28 00:41:10 +000098 int len;
99
100 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000101 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +0000102 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +0000103 case SQLITE_BLOB:
104 case SQLITE_INTEGER:
105 case SQLITE_FLOAT: {
drhf4479502004-05-27 03:12:53 +0000106 sqlite3_result_int(context, sqlite3_value_bytes(argv[0]));
drhf9b596e2004-05-26 16:54:42 +0000107 break;
108 }
drh9c054832004-05-31 18:51:57 +0000109 case SQLITE_TEXT: {
drh2646da72005-12-09 20:02:05 +0000110 const unsigned char *z = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +0000111 if( z==0 ) return;
drh4a919112007-05-15 11:55:09 +0000112 len = 0;
113 while( *z ){
114 len++;
115 SQLITE_SKIP_UTF8(z);
116 }
drhf4479502004-05-27 03:12:53 +0000117 sqlite3_result_int(context, len);
drhf9b596e2004-05-26 16:54:42 +0000118 break;
119 }
120 default: {
121 sqlite3_result_null(context);
122 break;
123 }
124 }
drh0bce8352002-02-28 00:41:10 +0000125}
126
127/*
drh2ba3ccc2009-12-08 02:06:08 +0000128** Implementation of the abs() function.
129**
130** IMP: R-23979-26855 The abs(X) function returns the absolute value of
131** the numeric argument X.
drh0bce8352002-02-28 00:41:10 +0000132*/
danielk19770ae8b832004-05-25 12:05:56 +0000133static void absFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drh0bce8352002-02-28 00:41:10 +0000134 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000135 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +0000136 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +0000137 case SQLITE_INTEGER: {
danielk1977f93bbbe2004-05-27 10:30:52 +0000138 i64 iVal = sqlite3_value_int64(argv[0]);
drh52fc8492006-02-23 21:43:55 +0000139 if( iVal<0 ){
drh693e6712014-01-24 22:58:00 +0000140 if( iVal==SMALLEST_INT64 ){
drheb091cd2013-11-09 19:47:15 +0000141 /* IMP: R-31676-45509 If X is the integer -9223372036854775808
142 ** then abs(X) throws an integer overflow error since there is no
drh2ba3ccc2009-12-08 02:06:08 +0000143 ** equivalent positive 64-bit two complement value. */
drh52fc8492006-02-23 21:43:55 +0000144 sqlite3_result_error(context, "integer overflow", -1);
145 return;
146 }
147 iVal = -iVal;
148 }
danielk1977f93bbbe2004-05-27 10:30:52 +0000149 sqlite3_result_int64(context, iVal);
drhf9b596e2004-05-26 16:54:42 +0000150 break;
151 }
drh9c054832004-05-31 18:51:57 +0000152 case SQLITE_NULL: {
drh2ba3ccc2009-12-08 02:06:08 +0000153 /* IMP: R-37434-19929 Abs(X) returns NULL if X is NULL. */
drhf9b596e2004-05-26 16:54:42 +0000154 sqlite3_result_null(context);
155 break;
156 }
157 default: {
drh2ba3ccc2009-12-08 02:06:08 +0000158 /* Because sqlite3_value_double() returns 0.0 if the argument is not
159 ** something that can be converted into a number, we have:
drh643091f2014-11-20 23:21:23 +0000160 ** IMP: R-01992-00519 Abs(X) returns 0.0 if X is a string or blob
161 ** that cannot be converted to a numeric value.
drh2ba3ccc2009-12-08 02:06:08 +0000162 */
danielk1977f93bbbe2004-05-27 10:30:52 +0000163 double rVal = sqlite3_value_double(argv[0]);
drh52fc8492006-02-23 21:43:55 +0000164 if( rVal<0 ) rVal = -rVal;
danielk1977f93bbbe2004-05-27 10:30:52 +0000165 sqlite3_result_double(context, rVal);
drhf9b596e2004-05-26 16:54:42 +0000166 break;
167 }
168 }
drh0bce8352002-02-28 00:41:10 +0000169}
170
171/*
drhd55e0722012-10-25 03:07:29 +0000172** Implementation of the instr() function.
173**
174** instr(haystack,needle) finds the first occurrence of needle
175** in haystack and returns the number of previous characters plus 1,
176** or 0 if needle does not occur within haystack.
177**
178** If both haystack and needle are BLOBs, then the result is one more than
179** the number of bytes in haystack prior to the first occurrence of needle,
180** or 0 if needle never occurs in haystack.
181*/
182static void instrFunc(
183 sqlite3_context *context,
184 int argc,
185 sqlite3_value **argv
186){
187 const unsigned char *zHaystack;
188 const unsigned char *zNeedle;
189 int nHaystack;
190 int nNeedle;
191 int typeHaystack, typeNeedle;
192 int N = 1;
193 int isText;
194
drh68c804b2012-12-04 11:03:11 +0000195 UNUSED_PARAMETER(argc);
drhd55e0722012-10-25 03:07:29 +0000196 typeHaystack = sqlite3_value_type(argv[0]);
197 typeNeedle = sqlite3_value_type(argv[1]);
198 if( typeHaystack==SQLITE_NULL || typeNeedle==SQLITE_NULL ) return;
199 nHaystack = sqlite3_value_bytes(argv[0]);
200 nNeedle = sqlite3_value_bytes(argv[1]);
201 if( typeHaystack==SQLITE_BLOB && typeNeedle==SQLITE_BLOB ){
202 zHaystack = sqlite3_value_blob(argv[0]);
203 zNeedle = sqlite3_value_blob(argv[1]);
204 isText = 0;
205 }else{
206 zHaystack = sqlite3_value_text(argv[0]);
207 zNeedle = sqlite3_value_text(argv[1]);
208 isText = 1;
209 }
210 while( nNeedle<=nHaystack && memcmp(zHaystack, zNeedle, nNeedle)!=0 ){
211 N++;
212 do{
213 nHaystack--;
214 zHaystack++;
215 }while( isText && (zHaystack[0]&0xc0)==0x80 );
216 }
217 if( nNeedle>nHaystack ) N = 0;
218 sqlite3_result_int(context, N);
219}
220
221/*
drha5c14162013-12-17 15:03:06 +0000222** Implementation of the printf() function.
223*/
224static void printfFunc(
225 sqlite3_context *context,
226 int argc,
227 sqlite3_value **argv
228){
229 PrintfArguments x;
230 StrAccum str;
231 const char *zFormat;
232 int n;
233
234 if( argc>=1 && (zFormat = (const char*)sqlite3_value_text(argv[0]))!=0 ){
235 x.nArg = argc-1;
236 x.nUsed = 0;
237 x.apArg = argv+1;
238 sqlite3StrAccumInit(&str, 0, 0, SQLITE_MAX_LENGTH);
239 str.db = sqlite3_context_db_handle(context);
240 sqlite3XPrintf(&str, SQLITE_PRINTF_SQLFUNC, zFormat, &x);
241 n = str.nChar;
242 sqlite3_result_text(context, sqlite3StrAccumFinish(&str), n,
243 SQLITE_DYNAMIC);
244 }
245}
246
247/*
drhf764e6f2007-05-15 01:13:47 +0000248** Implementation of the substr() function.
249**
250** substr(x,p1,p2) returns p2 characters of x[] beginning with p1.
251** p1 is 1-indexed. So substr(x,1,1) returns the first character
252** of x. If x is text, then we actually count UTF-8 characters.
253** If x is a blob, then we count bytes.
254**
255** If p1 is negative, then we begin abs(p1) from the end of x[].
shaneh779b8f12009-11-12 05:04:50 +0000256**
drhf7b54962013-05-28 12:11:54 +0000257** If p2 is negative, return the p2 characters preceding p1.
drh0bce8352002-02-28 00:41:10 +0000258*/
drhf9b596e2004-05-26 16:54:42 +0000259static void substrFunc(
260 sqlite3_context *context,
261 int argc,
262 sqlite3_value **argv
263){
drh2646da72005-12-09 20:02:05 +0000264 const unsigned char *z;
265 const unsigned char *z2;
drh023ae032007-05-08 12:12:16 +0000266 int len;
drhf764e6f2007-05-15 01:13:47 +0000267 int p0type;
drh023ae032007-05-08 12:12:16 +0000268 i64 p1, p2;
drh65595cd2009-02-02 16:32:55 +0000269 int negP2 = 0;
drhf9b596e2004-05-26 16:54:42 +0000270
drh64f31512007-10-12 19:11:55 +0000271 assert( argc==3 || argc==2 );
drh8198d252009-02-01 19:42:37 +0000272 if( sqlite3_value_type(argv[1])==SQLITE_NULL
273 || (argc==3 && sqlite3_value_type(argv[2])==SQLITE_NULL)
274 ){
275 return;
276 }
drhf764e6f2007-05-15 01:13:47 +0000277 p0type = sqlite3_value_type(argv[0]);
drh4adc4cb2009-11-11 20:53:31 +0000278 p1 = sqlite3_value_int(argv[1]);
drhf764e6f2007-05-15 01:13:47 +0000279 if( p0type==SQLITE_BLOB ){
280 len = sqlite3_value_bytes(argv[0]);
281 z = sqlite3_value_blob(argv[0]);
282 if( z==0 ) return;
drh1f0feef2007-05-15 13:27:07 +0000283 assert( len==sqlite3_value_bytes(argv[0]) );
drhf764e6f2007-05-15 01:13:47 +0000284 }else{
285 z = sqlite3_value_text(argv[0]);
286 if( z==0 ) return;
drh4a919112007-05-15 11:55:09 +0000287 len = 0;
drh4adc4cb2009-11-11 20:53:31 +0000288 if( p1<0 ){
289 for(z2=z; *z2; len++){
290 SQLITE_SKIP_UTF8(z2);
291 }
drh4a919112007-05-15 11:55:09 +0000292 }
drhf764e6f2007-05-15 01:13:47 +0000293 }
drh883ad042015-02-19 00:29:11 +0000294#ifdef SQLITE_SUBSTR_COMPATIBILITY
295 /* If SUBSTR_COMPATIBILITY is defined then substr(X,0,N) work the same as
296 ** as substr(X,1,N) - it returns the first N characters of X. This
297 ** is essentially a back-out of the bug-fix in check-in [5fc125d362df4b8]
298 ** from 2009-02-02 for compatibility of applications that exploited the
299 ** old buggy behavior. */
300 if( p1==0 ) p1 = 1; /* <rdar://problem/6778339> */
301#endif
drh64f31512007-10-12 19:11:55 +0000302 if( argc==3 ){
303 p2 = sqlite3_value_int(argv[2]);
drh65595cd2009-02-02 16:32:55 +0000304 if( p2<0 ){
305 p2 = -p2;
306 negP2 = 1;
307 }
drh64f31512007-10-12 19:11:55 +0000308 }else{
drhbb4957f2008-03-20 14:03:29 +0000309 p2 = sqlite3_context_db_handle(context)->aLimit[SQLITE_LIMIT_LENGTH];
drh64f31512007-10-12 19:11:55 +0000310 }
drh0bce8352002-02-28 00:41:10 +0000311 if( p1<0 ){
drh89425d52002-02-28 03:04:48 +0000312 p1 += len;
drh653bc752002-02-28 03:31:10 +0000313 if( p1<0 ){
314 p2 += p1;
drh65595cd2009-02-02 16:32:55 +0000315 if( p2<0 ) p2 = 0;
drh653bc752002-02-28 03:31:10 +0000316 p1 = 0;
317 }
drh0bce8352002-02-28 00:41:10 +0000318 }else if( p1>0 ){
319 p1--;
drh65595cd2009-02-02 16:32:55 +0000320 }else if( p2>0 ){
321 p2--;
drh0bce8352002-02-28 00:41:10 +0000322 }
drh65595cd2009-02-02 16:32:55 +0000323 if( negP2 ){
324 p1 -= p2;
drh4e79c592009-02-01 19:23:32 +0000325 if( p1<0 ){
326 p2 += p1;
327 p1 = 0;
328 }
329 }
drh65595cd2009-02-02 16:32:55 +0000330 assert( p1>=0 && p2>=0 );
drhf764e6f2007-05-15 01:13:47 +0000331 if( p0type!=SQLITE_BLOB ){
drh4a919112007-05-15 11:55:09 +0000332 while( *z && p1 ){
333 SQLITE_SKIP_UTF8(z);
334 p1--;
drhf764e6f2007-05-15 01:13:47 +0000335 }
drh4a919112007-05-15 11:55:09 +0000336 for(z2=z; *z2 && p2; p2--){
337 SQLITE_SKIP_UTF8(z2);
drhf764e6f2007-05-15 01:13:47 +0000338 }
drhbbf483f2014-09-09 20:30:24 +0000339 sqlite3_result_text64(context, (char*)z, z2-z, SQLITE_TRANSIENT,
340 SQLITE_UTF8);
drhf764e6f2007-05-15 01:13:47 +0000341 }else{
drh4adc4cb2009-11-11 20:53:31 +0000342 if( p1+p2>len ){
343 p2 = len-p1;
344 if( p2<0 ) p2 = 0;
345 }
drhbbf483f2014-09-09 20:30:24 +0000346 sqlite3_result_blob64(context, (char*)&z[p1], (u64)p2, SQLITE_TRANSIENT);
drh0bce8352002-02-28 00:41:10 +0000347 }
drh0bce8352002-02-28 00:41:10 +0000348}
349
350/*
351** Implementation of the round() function
352*/
shanefbd60f82009-02-04 03:59:25 +0000353#ifndef SQLITE_OMIT_FLOATING_POINT
danielk19770ae8b832004-05-25 12:05:56 +0000354static void roundFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
danielk197751ad0ec2004-05-24 12:39:02 +0000355 int n = 0;
drh0bce8352002-02-28 00:41:10 +0000356 double r;
drh50d654d2009-06-03 01:24:54 +0000357 char *zBuf;
drh0bce8352002-02-28 00:41:10 +0000358 assert( argc==1 || argc==2 );
danielk197751ad0ec2004-05-24 12:39:02 +0000359 if( argc==2 ){
drh9c054832004-05-31 18:51:57 +0000360 if( SQLITE_NULL==sqlite3_value_type(argv[1]) ) return;
danielk197751ad0ec2004-05-24 12:39:02 +0000361 n = sqlite3_value_int(argv[1]);
362 if( n>30 ) n = 30;
363 if( n<0 ) n = 0;
364 }
drhd589a922006-03-02 03:02:48 +0000365 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drh4f26d6c2004-05-26 23:25:30 +0000366 r = sqlite3_value_double(argv[0]);
shaneh147e1762010-02-17 04:19:27 +0000367 /* If Y==0 and X will fit in a 64-bit int,
368 ** handle the rounding directly,
369 ** otherwise use printf.
370 */
371 if( n==0 && r>=0 && r<LARGEST_INT64-1 ){
372 r = (double)((sqlite_int64)(r+0.5));
373 }else if( n==0 && r<0 && (-r)<LARGEST_INT64-1 ){
374 r = -(double)((sqlite_int64)((-r)+0.5));
drh50d654d2009-06-03 01:24:54 +0000375 }else{
shaneh147e1762010-02-17 04:19:27 +0000376 zBuf = sqlite3_mprintf("%.*f",n,r);
377 if( zBuf==0 ){
378 sqlite3_result_error_nomem(context);
379 return;
380 }
drh9339da12010-09-30 00:50:49 +0000381 sqlite3AtoF(zBuf, &r, sqlite3Strlen30(zBuf), SQLITE_UTF8);
drh50d654d2009-06-03 01:24:54 +0000382 sqlite3_free(zBuf);
drh50d654d2009-06-03 01:24:54 +0000383 }
shaneh147e1762010-02-17 04:19:27 +0000384 sqlite3_result_double(context, r);
drh0bce8352002-02-28 00:41:10 +0000385}
shanefbd60f82009-02-04 03:59:25 +0000386#endif
drhdc04c582002-02-24 01:55:15 +0000387
danielk197726783a52007-08-29 14:06:22 +0000388/*
389** Allocate nByte bytes of space using sqlite3_malloc(). If the
390** allocation fails, call sqlite3_result_error_nomem() to notify
drh27e62db2009-04-02 10:16:17 +0000391** the database handle that malloc() has failed and return NULL.
392** If nByte is larger than the maximum string or blob length, then
393** raise an SQLITE_TOOBIG exception and return NULL.
danielk197726783a52007-08-29 14:06:22 +0000394*/
drhb1a6c3c2008-03-20 16:30:17 +0000395static void *contextMalloc(sqlite3_context *context, i64 nByte){
drhbb4957f2008-03-20 14:03:29 +0000396 char *z;
drh27e62db2009-04-02 10:16:17 +0000397 sqlite3 *db = sqlite3_context_db_handle(context);
drhef31c6a2009-04-02 09:07:12 +0000398 assert( nByte>0 );
drh27e62db2009-04-02 10:16:17 +0000399 testcase( nByte==db->aLimit[SQLITE_LIMIT_LENGTH] );
400 testcase( nByte==db->aLimit[SQLITE_LIMIT_LENGTH]+1 );
401 if( nByte>db->aLimit[SQLITE_LIMIT_LENGTH] ){
drhbb4957f2008-03-20 14:03:29 +0000402 sqlite3_result_error_toobig(context);
403 z = 0;
404 }else{
drhda4ca9d2014-09-09 17:27:35 +0000405 z = sqlite3Malloc(nByte);
drhef31c6a2009-04-02 09:07:12 +0000406 if( !z ){
drhbb4957f2008-03-20 14:03:29 +0000407 sqlite3_result_error_nomem(context);
408 }
danielk1977a1644fd2007-08-29 12:31:25 +0000409 }
410 return z;
411}
412
drhdc04c582002-02-24 01:55:15 +0000413/*
414** Implementation of the upper() and lower() SQL functions.
415*/
danielk19770ae8b832004-05-25 12:05:56 +0000416static void upperFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drh7a521cf2007-04-25 18:23:52 +0000417 char *z1;
418 const char *z2;
drh9310ef22007-04-27 17:16:20 +0000419 int i, n;
drh1d34fde2009-02-03 15:50:33 +0000420 UNUSED_PARAMETER(argc);
drh7a521cf2007-04-25 18:23:52 +0000421 z2 = (char*)sqlite3_value_text(argv[0]);
drh1f0feef2007-05-15 13:27:07 +0000422 n = sqlite3_value_bytes(argv[0]);
423 /* Verify that the call to _bytes() does not invalidate the _text() pointer */
424 assert( z2==(char*)sqlite3_value_text(argv[0]) );
drh7a521cf2007-04-25 18:23:52 +0000425 if( z2 ){
drhb1a6c3c2008-03-20 16:30:17 +0000426 z1 = contextMalloc(context, ((i64)n)+1);
drh7a521cf2007-04-25 18:23:52 +0000427 if( z1 ){
drhdf901d32011-10-13 18:00:11 +0000428 for(i=0; i<n; i++){
429 z1[i] = (char)sqlite3Toupper(z2[i]);
drh7a521cf2007-04-25 18:23:52 +0000430 }
drhdf901d32011-10-13 18:00:11 +0000431 sqlite3_result_text(context, z1, n, sqlite3_free);
drh7a521cf2007-04-25 18:23:52 +0000432 }
drhdc04c582002-02-24 01:55:15 +0000433 }
434}
danielk19770ae8b832004-05-25 12:05:56 +0000435static void lowerFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drhdf901d32011-10-13 18:00:11 +0000436 char *z1;
drh7a521cf2007-04-25 18:23:52 +0000437 const char *z2;
drh9310ef22007-04-27 17:16:20 +0000438 int i, n;
drh1d34fde2009-02-03 15:50:33 +0000439 UNUSED_PARAMETER(argc);
drh7a521cf2007-04-25 18:23:52 +0000440 z2 = (char*)sqlite3_value_text(argv[0]);
drh1f0feef2007-05-15 13:27:07 +0000441 n = sqlite3_value_bytes(argv[0]);
442 /* Verify that the call to _bytes() does not invalidate the _text() pointer */
443 assert( z2==(char*)sqlite3_value_text(argv[0]) );
drh7a521cf2007-04-25 18:23:52 +0000444 if( z2 ){
drhb1a6c3c2008-03-20 16:30:17 +0000445 z1 = contextMalloc(context, ((i64)n)+1);
drh7a521cf2007-04-25 18:23:52 +0000446 if( z1 ){
drhdf901d32011-10-13 18:00:11 +0000447 for(i=0; i<n; i++){
448 z1[i] = sqlite3Tolower(z2[i]);
drh7a521cf2007-04-25 18:23:52 +0000449 }
drhdf901d32011-10-13 18:00:11 +0000450 sqlite3_result_text(context, z1, n, sqlite3_free);
drh7a521cf2007-04-25 18:23:52 +0000451 }
drhdc04c582002-02-24 01:55:15 +0000452 }
453}
454
drhae6bb952009-11-11 00:24:31 +0000455/*
drhcca9f3d2013-09-06 15:23:29 +0000456** Some functions like COALESCE() and IFNULL() and UNLIKELY() are implemented
457** as VDBE code so that unused argument values do not have to be computed.
458** However, we still need some kind of function implementation for this
459** routines in the function table. The noopFunc macro provides this.
460** noopFunc will never be called so it doesn't matter what the implementation
461** is. We might as well use the "version()" function as a substitute.
drhae6bb952009-11-11 00:24:31 +0000462*/
drhcca9f3d2013-09-06 15:23:29 +0000463#define noopFunc versionFunc /* Substitute function - never called */
drh3212e182002-02-28 00:46:26 +0000464
465/*
drhf9ffac92002-03-02 19:00:31 +0000466** Implementation of random(). Return a random integer.
467*/
drhf9b596e2004-05-26 16:54:42 +0000468static void randomFunc(
469 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000470 int NotUsed,
471 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000472){
drh52fc8492006-02-23 21:43:55 +0000473 sqlite_int64 r;
danielk197762c14b32008-11-19 09:05:26 +0000474 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drh2fa18682008-03-19 14:15:34 +0000475 sqlite3_randomness(sizeof(r), &r);
drh3034e3d2009-04-02 14:05:21 +0000476 if( r<0 ){
477 /* We need to prevent a random number of 0x8000000000000000
478 ** (or -9223372036854775808) since when you do abs() of that
479 ** number of you get the same value back again. To do this
480 ** in a way that is testable, mask the sign bit off of negative
481 ** values, resulting in a positive value. Then take the
482 ** 2s complement of that positive value. The end result can
483 ** therefore be no less than -9223372036854775807.
484 */
drhaf8001b2012-02-11 19:53:24 +0000485 r = -(r & LARGEST_INT64);
drh3034e3d2009-04-02 14:05:21 +0000486 }
drh52fc8492006-02-23 21:43:55 +0000487 sqlite3_result_int64(context, r);
drhf9ffac92002-03-02 19:00:31 +0000488}
489
490/*
drh137c7282007-01-29 17:58:28 +0000491** Implementation of randomblob(N). Return a random blob
492** that is N bytes long.
drh63cf66f2007-01-29 15:50:05 +0000493*/
drh137c7282007-01-29 17:58:28 +0000494static void randomBlob(
drh63cf66f2007-01-29 15:50:05 +0000495 sqlite3_context *context,
496 int argc,
497 sqlite3_value **argv
498){
drh137c7282007-01-29 17:58:28 +0000499 int n;
500 unsigned char *p;
drh63cf66f2007-01-29 15:50:05 +0000501 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000502 UNUSED_PARAMETER(argc);
drh63cf66f2007-01-29 15:50:05 +0000503 n = sqlite3_value_int(argv[0]);
drh023ae032007-05-08 12:12:16 +0000504 if( n<1 ){
505 n = 1;
506 }
danielk1977a1644fd2007-08-29 12:31:25 +0000507 p = contextMalloc(context, n);
drh02d85832007-05-07 19:31:15 +0000508 if( p ){
drh2fa18682008-03-19 14:15:34 +0000509 sqlite3_randomness(n, p);
drh17435752007-08-16 04:30:38 +0000510 sqlite3_result_blob(context, (char*)p, n, sqlite3_free);
drh02d85832007-05-07 19:31:15 +0000511 }
drh63cf66f2007-01-29 15:50:05 +0000512}
513
514/*
drh6ed41ad2002-04-06 14:10:47 +0000515** Implementation of the last_insert_rowid() SQL function. The return
danielk197724b03fd2004-05-10 10:34:34 +0000516** value is the same as the sqlite3_last_insert_rowid() API function.
drh6ed41ad2002-04-06 14:10:47 +0000517*/
danielk197751ad0ec2004-05-24 12:39:02 +0000518static void last_insert_rowid(
danielk19770ae8b832004-05-25 12:05:56 +0000519 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000520 int NotUsed,
521 sqlite3_value **NotUsed2
danielk197751ad0ec2004-05-24 12:39:02 +0000522){
drhfa4a4b92008-03-19 21:45:51 +0000523 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000524 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000525 /* IMP: R-51513-12026 The last_insert_rowid() SQL function is a
526 ** wrapper around the sqlite3_last_insert_rowid() C/C++ interface
527 ** function. */
drhf9b596e2004-05-26 16:54:42 +0000528 sqlite3_result_int64(context, sqlite3_last_insert_rowid(db));
drh6ed41ad2002-04-06 14:10:47 +0000529}
530
rdcf146a772004-02-25 22:51:06 +0000531/*
drhab2f1f92010-01-11 18:26:42 +0000532** Implementation of the changes() SQL function.
533**
534** IMP: R-62073-11209 The changes() SQL function is a wrapper
535** around the sqlite3_changes() C/C++ function and hence follows the same
536** rules for counting changes.
rdcf146a772004-02-25 22:51:06 +0000537*/
danielk1977b28af712004-06-21 06:50:26 +0000538static void changes(
drhf9b596e2004-05-26 16:54:42 +0000539 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000540 int NotUsed,
541 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000542){
drhfa4a4b92008-03-19 21:45:51 +0000543 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000544 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhf4479502004-05-27 03:12:53 +0000545 sqlite3_result_int(context, sqlite3_changes(db));
rdcb0c374f2004-02-20 22:53:38 +0000546}
rdcf146a772004-02-25 22:51:06 +0000547
548/*
danielk1977b28af712004-06-21 06:50:26 +0000549** Implementation of the total_changes() SQL function. The return value is
550** the same as the sqlite3_total_changes() API function.
rdcf146a772004-02-25 22:51:06 +0000551*/
danielk1977b28af712004-06-21 06:50:26 +0000552static void total_changes(
553 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000554 int NotUsed,
555 sqlite3_value **NotUsed2
danielk197751ad0ec2004-05-24 12:39:02 +0000556){
drhfa4a4b92008-03-19 21:45:51 +0000557 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000558 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000559 /* IMP: R-52756-41993 This function is a wrapper around the
560 ** sqlite3_total_changes() C/C++ interface. */
danielk1977b28af712004-06-21 06:50:26 +0000561 sqlite3_result_int(context, sqlite3_total_changes(db));
rdcb0c374f2004-02-20 22:53:38 +0000562}
563
drh6ed41ad2002-04-06 14:10:47 +0000564/*
drh4e5ffc52004-08-31 00:52:37 +0000565** A structure defining how to do GLOB-style comparisons.
danielk1977d02eb1f2004-06-06 09:44:03 +0000566*/
drh4e5ffc52004-08-31 00:52:37 +0000567struct compareInfo {
568 u8 matchAll;
569 u8 matchOne;
570 u8 matchSet;
571 u8 noCase;
danielk1977d02eb1f2004-06-06 09:44:03 +0000572};
drh55ef4d92005-08-14 01:20:37 +0000573
drhb9175ae2007-12-07 18:39:04 +0000574/*
575** For LIKE and GLOB matching on EBCDIC machines, assume that every
576** character is exactly one byte in size. Also, all characters are
577** able to participate in upper-case-to-lower-case mappings in EBCDIC
578** whereas only characters less than 0x80 do in ASCII.
579*/
580#if defined(SQLITE_EBCDIC)
drh88b33222014-09-25 03:51:37 +0000581# define sqlite3Utf8Read(A) (*((*A)++))
582# define GlobUpperToLower(A) A = sqlite3UpperToLower[A]
583# define GlobUpperToLowerAscii(A) A = sqlite3UpperToLower[A]
drhb9175ae2007-12-07 18:39:04 +0000584#else
drh88b33222014-09-25 03:51:37 +0000585# define GlobUpperToLower(A) if( A<=0x7f ){ A = sqlite3UpperToLower[A]; }
586# define GlobUpperToLowerAscii(A) A = sqlite3UpperToLower[A]
drhb9175ae2007-12-07 18:39:04 +0000587#endif
588
drh4e5ffc52004-08-31 00:52:37 +0000589static const struct compareInfo globInfo = { '*', '?', '[', 0 };
drh55ef4d92005-08-14 01:20:37 +0000590/* The correct SQL-92 behavior is for the LIKE operator to ignore
591** case. Thus 'a' LIKE 'A' would be true. */
592static const struct compareInfo likeInfoNorm = { '%', '_', 0, 1 };
593/* If SQLITE_CASE_SENSITIVE_LIKE is defined, then the LIKE operator
594** is case sensitive causing 'a' LIKE 'A' to be false */
595static const struct compareInfo likeInfoAlt = { '%', '_', 0, 0 };
danielk1977d02eb1f2004-06-06 09:44:03 +0000596
597/*
drh4e5ffc52004-08-31 00:52:37 +0000598** Compare two UTF-8 strings for equality where the first string can
drh9fdfdc82014-09-25 11:08:57 +0000599** potentially be a "glob" or "like" expression. Return true (1) if they
drh4e5ffc52004-08-31 00:52:37 +0000600** are the same and false (0) if they are different.
drh0ac65892002-04-20 14:24:41 +0000601**
drh4e5ffc52004-08-31 00:52:37 +0000602** Globbing rules:
drh0ac65892002-04-20 14:24:41 +0000603**
drh4e5ffc52004-08-31 00:52:37 +0000604** '*' Matches any sequence of zero or more characters.
danielk1977d02eb1f2004-06-06 09:44:03 +0000605**
drh4e5ffc52004-08-31 00:52:37 +0000606** '?' Matches exactly one character.
607**
608** [...] Matches one character from the enclosed list of
609** characters.
610**
611** [^...] Matches one character not in the enclosed list.
612**
613** With the [...] and [^...] matching, a ']' character can be included
614** in the list by making it the first character after '[' or '^'. A
615** range of characters can be specified using '-'. Example:
616** "[a-z]" matches any single lower-case letter. To match a '-', make
617** it the last character in the list.
618**
drh9fdfdc82014-09-25 11:08:57 +0000619** Like matching rules:
620**
621** '%' Matches any sequence of zero or more characters
622**
623*** '_' Matches any one character
624**
625** Ec Where E is the "esc" character and c is any other
626** character, including '%', '_', and esc, match exactly c.
627**
628** The comments through this routine usually assume glob matching.
629**
drh4e5ffc52004-08-31 00:52:37 +0000630** This routine is usually quick, but can be N**2 in the worst case.
drh0ac65892002-04-20 14:24:41 +0000631*/
danielk19777c6303c2004-11-17 16:41:29 +0000632static int patternCompare(
drh4e5ffc52004-08-31 00:52:37 +0000633 const u8 *zPattern, /* The glob pattern */
634 const u8 *zString, /* The string to compare against the glob */
danielk19777c6303c2004-11-17 16:41:29 +0000635 const struct compareInfo *pInfo, /* Information about how to do the compare */
drh0a32fa62011-06-13 12:19:21 +0000636 u32 esc /* The escape character */
danielk197751ad0ec2004-05-24 12:39:02 +0000637){
drh9fdfdc82014-09-25 11:08:57 +0000638 u32 c, c2; /* Next pattern and input string chars */
639 u32 matchOne = pInfo->matchOne; /* "?" or "_" */
640 u32 matchAll = pInfo->matchAll; /* "*" or "%" */
641 u32 matchOther; /* "[" or the escape character */
642 u8 noCase = pInfo->noCase; /* True if uppercase==lowercase */
643 const u8 *zEscaped = 0; /* One past the last escaped input char */
drh88b33222014-09-25 03:51:37 +0000644
645 /* The GLOB operator does not have an ESCAPE clause. And LIKE does not
646 ** have the matchSet operator. So we either have to look for one or
647 ** the other, never both. Hence the single variable matchOther is used
648 ** to store the one we have to look for.
649 */
650 matchOther = esc ? esc : pInfo->matchSet;
danielk1977d02eb1f2004-06-06 09:44:03 +0000651
drh42610962012-09-17 18:56:32 +0000652 while( (c = sqlite3Utf8Read(&zPattern))!=0 ){
drh9fdfdc82014-09-25 11:08:57 +0000653 if( c==matchAll ){ /* Match "*" */
654 /* Skip over multiple "*" characters in the pattern. If there
655 ** are also "?" characters, skip those as well, but consume a
656 ** single character of the input string for each "?" skipped */
drh42610962012-09-17 18:56:32 +0000657 while( (c=sqlite3Utf8Read(&zPattern)) == matchAll
drh66150952007-07-23 19:12:41 +0000658 || c == matchOne ){
drh42610962012-09-17 18:56:32 +0000659 if( c==matchOne && sqlite3Utf8Read(&zString)==0 ){
drh66150952007-07-23 19:12:41 +0000660 return 0;
drh4e5ffc52004-08-31 00:52:37 +0000661 }
danielk1977ad7dd422004-06-06 12:41:49 +0000662 }
drh66150952007-07-23 19:12:41 +0000663 if( c==0 ){
drh9fdfdc82014-09-25 11:08:57 +0000664 return 1; /* "*" at the end of the pattern matches */
drh88b33222014-09-25 03:51:37 +0000665 }else if( c==matchOther ){
666 if( esc ){
667 c = sqlite3Utf8Read(&zPattern);
668 if( c==0 ) return 0;
669 }else{
drh9fdfdc82014-09-25 11:08:57 +0000670 /* "[...]" immediately follows the "*". We have to do a slow
671 ** recursive search in this case, but it is an unusual case. */
drh88b33222014-09-25 03:51:37 +0000672 assert( matchOther<0x80 ); /* '[' is a single-byte character */
673 while( *zString
674 && patternCompare(&zPattern[-1],zString,pInfo,esc)==0 ){
675 SQLITE_SKIP_UTF8(zString);
676 }
677 return *zString!=0;
drh66150952007-07-23 19:12:41 +0000678 }
drh66150952007-07-23 19:12:41 +0000679 }
drh9fdfdc82014-09-25 11:08:57 +0000680
681 /* At this point variable c contains the first character of the
682 ** pattern string past the "*". Search in the input string for the
683 ** first matching character and recursively contine the match from
684 ** that point.
685 **
686 ** For a case-insensitive search, set variable cx to be the same as
687 ** c but in the other case and search the input string for either
688 ** c or cx.
689 */
690 if( c<=0x80 ){
691 u32 cx;
692 if( noCase ){
693 cx = sqlite3Toupper(c);
694 c = sqlite3Tolower(c);
drh66150952007-07-23 19:12:41 +0000695 }else{
drh9fdfdc82014-09-25 11:08:57 +0000696 cx = c;
drh4e5ffc52004-08-31 00:52:37 +0000697 }
drh9fdfdc82014-09-25 11:08:57 +0000698 while( (c2 = *(zString++))!=0 ){
699 if( c2!=c && c2!=cx ) continue;
700 if( patternCompare(zPattern,zString,pInfo,esc) ) return 1;
701 }
702 }else{
703 while( (c2 = sqlite3Utf8Read(&zString))!=0 ){
704 if( c2!=c ) continue;
705 if( patternCompare(zPattern,zString,pInfo,esc) ) return 1;
706 }
drh66150952007-07-23 19:12:41 +0000707 }
708 return 0;
drh88b33222014-09-25 03:51:37 +0000709 }
drh88b33222014-09-25 03:51:37 +0000710 if( c==matchOther ){
711 if( esc ){
712 c = sqlite3Utf8Read(&zPattern);
713 if( c==0 ) return 0;
drh9fdfdc82014-09-25 11:08:57 +0000714 zEscaped = zPattern;
drh88b33222014-09-25 03:51:37 +0000715 }else{
716 u32 prior_c = 0;
drh9fdfdc82014-09-25 11:08:57 +0000717 int seen = 0;
718 int invert = 0;
drh88b33222014-09-25 03:51:37 +0000719 c = sqlite3Utf8Read(&zString);
720 if( c==0 ) return 0;
721 c2 = sqlite3Utf8Read(&zPattern);
722 if( c2=='^' ){
723 invert = 1;
724 c2 = sqlite3Utf8Read(&zPattern);
725 }
726 if( c2==']' ){
727 if( c==']' ) seen = 1;
728 c2 = sqlite3Utf8Read(&zPattern);
729 }
730 while( c2 && c2!=']' ){
731 if( c2=='-' && zPattern[0]!=']' && zPattern[0]!=0 && prior_c>0 ){
732 c2 = sqlite3Utf8Read(&zPattern);
733 if( c>=prior_c && c<=c2 ) seen = 1;
734 prior_c = 0;
735 }else{
736 if( c==c2 ){
737 seen = 1;
738 }
739 prior_c = c2;
740 }
741 c2 = sqlite3Utf8Read(&zPattern);
742 }
743 if( c2==0 || (seen ^ invert)==0 ){
744 return 0;
745 }
746 continue;
747 }
748 }
749 c2 = sqlite3Utf8Read(&zString);
750 if( c==c2 ) continue;
drh9fdfdc82014-09-25 11:08:57 +0000751 if( noCase && c<0x80 && c2<0x80 && sqlite3Tolower(c)==sqlite3Tolower(c2) ){
752 continue;
753 }
754 if( c==matchOne && zPattern!=zEscaped && c2!=0 ) continue;
755 return 0;
danielk197751ad0ec2004-05-24 12:39:02 +0000756 }
drh4e5ffc52004-08-31 00:52:37 +0000757 return *zString==0;
drh0ac65892002-04-20 14:24:41 +0000758}
drh4e5ffc52004-08-31 00:52:37 +0000759
drh55ef4d92005-08-14 01:20:37 +0000760/*
drh56282a52013-04-10 16:13:38 +0000761** The sqlite3_strglob() interface.
762*/
763int sqlite3_strglob(const char *zGlobPattern, const char *zString){
764 return patternCompare((u8*)zGlobPattern, (u8*)zString, &globInfo, 0)==0;
765}
766
767/*
drh55ef4d92005-08-14 01:20:37 +0000768** Count the number of times that the LIKE operator (or GLOB which is
769** just a variation of LIKE) gets called. This is used for testing
770** only.
771*/
772#ifdef SQLITE_TEST
773int sqlite3_like_count = 0;
774#endif
775
danielk19773f6b0872004-06-17 05:36:44 +0000776
777/*
778** Implementation of the like() SQL function. This function implements
779** the build-in LIKE operator. The first argument to the function is the
780** pattern and the second argument is the string. So, the SQL statements:
781**
782** A LIKE B
783**
784** is implemented as like(B,A).
785**
drh55ef4d92005-08-14 01:20:37 +0000786** This same function (with a different compareInfo structure) computes
787** the GLOB operator.
danielk19773f6b0872004-06-17 05:36:44 +0000788*/
789static void likeFunc(
790 sqlite3_context *context,
791 int argc,
792 sqlite3_value **argv
793){
drhbeb818d2007-05-08 15:34:47 +0000794 const unsigned char *zA, *zB;
drh0a32fa62011-06-13 12:19:21 +0000795 u32 escape = 0;
drh27e62db2009-04-02 10:16:17 +0000796 int nPat;
drhbb4957f2008-03-20 14:03:29 +0000797 sqlite3 *db = sqlite3_context_db_handle(context);
drhbeb818d2007-05-08 15:34:47 +0000798
drh1f0feef2007-05-15 13:27:07 +0000799 zB = sqlite3_value_text(argv[0]);
800 zA = sqlite3_value_text(argv[1]);
801
drhbeb818d2007-05-08 15:34:47 +0000802 /* Limit the length of the LIKE or GLOB pattern to avoid problems
803 ** of deep recursion and N*N behavior in patternCompare().
804 */
drh27e62db2009-04-02 10:16:17 +0000805 nPat = sqlite3_value_bytes(argv[0]);
806 testcase( nPat==db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH] );
807 testcase( nPat==db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH]+1 );
808 if( nPat > db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH] ){
drhbeb818d2007-05-08 15:34:47 +0000809 sqlite3_result_error(context, "LIKE or GLOB pattern too complex", -1);
810 return;
811 }
drh1f0feef2007-05-15 13:27:07 +0000812 assert( zB==sqlite3_value_text(argv[0]) ); /* Encoding did not change */
drhbeb818d2007-05-08 15:34:47 +0000813
danielk19777c6303c2004-11-17 16:41:29 +0000814 if( argc==3 ){
815 /* The escape character string must consist of a single UTF-8 character.
816 ** Otherwise, return an error.
817 */
818 const unsigned char *zEsc = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +0000819 if( zEsc==0 ) return;
drhee858132007-05-08 20:37:38 +0000820 if( sqlite3Utf8CharLen((char*)zEsc, -1)!=1 ){
danielk19777c6303c2004-11-17 16:41:29 +0000821 sqlite3_result_error(context,
822 "ESCAPE expression must be a single character", -1);
823 return;
824 }
drh42610962012-09-17 18:56:32 +0000825 escape = sqlite3Utf8Read(&zEsc);
danielk19777c6303c2004-11-17 16:41:29 +0000826 }
danielk19773f6b0872004-06-17 05:36:44 +0000827 if( zA && zB ){
drh55ef4d92005-08-14 01:20:37 +0000828 struct compareInfo *pInfo = sqlite3_user_data(context);
829#ifdef SQLITE_TEST
830 sqlite3_like_count++;
831#endif
drhbeb818d2007-05-08 15:34:47 +0000832
danielk1977b56fe1f2007-05-09 08:24:44 +0000833 sqlite3_result_int(context, patternCompare(zB, zA, pInfo, escape));
danielk197751ad0ec2004-05-24 12:39:02 +0000834 }
drh8912d102002-05-26 21:34:58 +0000835}
836
837/*
838** Implementation of the NULLIF(x,y) function. The result is the first
839** argument if the arguments are different. The result is NULL if the
840** arguments are equal to each other.
841*/
drhf9b596e2004-05-26 16:54:42 +0000842static void nullifFunc(
843 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000844 int NotUsed,
drhf9b596e2004-05-26 16:54:42 +0000845 sqlite3_value **argv
846){
danielk1977dc1bdc42004-06-11 10:51:27 +0000847 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk197762c14b32008-11-19 09:05:26 +0000848 UNUSED_PARAMETER(NotUsed);
danielk1977dc1bdc42004-06-11 10:51:27 +0000849 if( sqlite3MemCompare(argv[0], argv[1], pColl)!=0 ){
drhf4479502004-05-27 03:12:53 +0000850 sqlite3_result_value(context, argv[0]);
drh8912d102002-05-26 21:34:58 +0000851 }
drh0ac65892002-04-20 14:24:41 +0000852}
853
drh647cb0e2002-11-04 19:32:25 +0000854/*
drh47baebc2009-08-14 16:01:24 +0000855** Implementation of the sqlite_version() function. The result is the version
drh647cb0e2002-11-04 19:32:25 +0000856** of the SQLite library that is running.
857*/
drhf9b596e2004-05-26 16:54:42 +0000858static void versionFunc(
859 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000860 int NotUsed,
861 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000862){
danielk197762c14b32008-11-19 09:05:26 +0000863 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000864 /* IMP: R-48699-48617 This function is an SQL wrapper around the
865 ** sqlite3_libversion() C-interface. */
866 sqlite3_result_text(context, sqlite3_libversion(), -1, SQLITE_STATIC);
drh647cb0e2002-11-04 19:32:25 +0000867}
868
drh47baebc2009-08-14 16:01:24 +0000869/*
870** Implementation of the sqlite_source_id() function. The result is a string
871** that identifies the particular version of the source code used to build
872** SQLite.
873*/
874static void sourceidFunc(
875 sqlite3_context *context,
876 int NotUsed,
877 sqlite3_value **NotUsed2
878){
879 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000880 /* IMP: R-24470-31136 This function is an SQL wrapper around the
881 ** sqlite3_sourceid() C interface. */
882 sqlite3_result_text(context, sqlite3_sourceid(), -1, SQLITE_STATIC);
drh47baebc2009-08-14 16:01:24 +0000883}
884
shanehbdea6d12010-02-23 04:19:54 +0000885/*
drh3ca84ef2011-04-25 18:03:10 +0000886** Implementation of the sqlite_log() function. This is a wrapper around
887** sqlite3_log(). The return value is NULL. The function exists purely for
888** its side-effects.
889*/
drh840561f2011-04-27 18:08:42 +0000890static void errlogFunc(
drh3ca84ef2011-04-25 18:03:10 +0000891 sqlite3_context *context,
892 int argc,
893 sqlite3_value **argv
894){
895 UNUSED_PARAMETER(argc);
896 UNUSED_PARAMETER(context);
897 sqlite3_log(sqlite3_value_int(argv[0]), "%s", sqlite3_value_text(argv[1]));
898}
899
900/*
shanehdc97a8c2010-02-23 20:08:35 +0000901** Implementation of the sqlite_compileoption_used() function.
902** The result is an integer that identifies if the compiler option
903** was used to build SQLite.
shanehbdea6d12010-02-23 04:19:54 +0000904*/
shanehdc97a8c2010-02-23 20:08:35 +0000905#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
906static void compileoptionusedFunc(
shanehbdea6d12010-02-23 04:19:54 +0000907 sqlite3_context *context,
shanehdc97a8c2010-02-23 20:08:35 +0000908 int argc,
909 sqlite3_value **argv
shanehbdea6d12010-02-23 04:19:54 +0000910){
shanehdc97a8c2010-02-23 20:08:35 +0000911 const char *zOptName;
912 assert( argc==1 );
913 UNUSED_PARAMETER(argc);
drha3e414c2010-08-03 18:06:25 +0000914 /* IMP: R-39564-36305 The sqlite_compileoption_used() SQL
915 ** function is a wrapper around the sqlite3_compileoption_used() C/C++
916 ** function.
917 */
drh264a2d42010-02-25 15:28:41 +0000918 if( (zOptName = (const char*)sqlite3_value_text(argv[0]))!=0 ){
shanehdc97a8c2010-02-23 20:08:35 +0000919 sqlite3_result_int(context, sqlite3_compileoption_used(zOptName));
shanehdc97a8c2010-02-23 20:08:35 +0000920 }
shanehbdea6d12010-02-23 04:19:54 +0000921}
shanehdc97a8c2010-02-23 20:08:35 +0000922#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
923
924/*
925** Implementation of the sqlite_compileoption_get() function.
926** The result is a string that identifies the compiler options
927** used to build SQLite.
928*/
929#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
930static void compileoptiongetFunc(
931 sqlite3_context *context,
932 int argc,
933 sqlite3_value **argv
934){
935 int n;
936 assert( argc==1 );
937 UNUSED_PARAMETER(argc);
drha3e414c2010-08-03 18:06:25 +0000938 /* IMP: R-04922-24076 The sqlite_compileoption_get() SQL function
939 ** is a wrapper around the sqlite3_compileoption_get() C/C++ function.
940 */
shanehdc97a8c2010-02-23 20:08:35 +0000941 n = sqlite3_value_int(argv[0]);
942 sqlite3_result_text(context, sqlite3_compileoption_get(n), -1, SQLITE_STATIC);
943}
944#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
shanehbdea6d12010-02-23 04:19:54 +0000945
drh137c7282007-01-29 17:58:28 +0000946/* Array for converting from half-bytes (nybbles) into ASCII hex
947** digits. */
948static const char hexdigits[] = {
949 '0', '1', '2', '3', '4', '5', '6', '7',
950 '8', '9', 'A', 'B', 'C', 'D', 'E', 'F'
951};
danielk1977d641d642004-11-18 15:44:29 +0000952
drh47394702003-08-20 01:03:33 +0000953/*
drh47394702003-08-20 01:03:33 +0000954** Implementation of the QUOTE() function. This function takes a single
955** argument. If the argument is numeric, the return value is the same as
956** the argument. If the argument is NULL, the return value is the string
957** "NULL". Otherwise, the argument is enclosed in single quotes with
958** single-quote escapes.
959*/
danielk19770ae8b832004-05-25 12:05:56 +0000960static void quoteFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drha0df4cc2009-02-02 17:29:59 +0000961 assert( argc==1 );
drh1d34fde2009-02-03 15:50:33 +0000962 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +0000963 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +0000964 case SQLITE_FLOAT: {
drh72b3fbc2012-06-19 03:11:25 +0000965 double r1, r2;
drh72b3fbc2012-06-19 03:11:25 +0000966 char zBuf[50];
mistachkin2b434a72012-06-19 04:36:48 +0000967 r1 = sqlite3_value_double(argv[0]);
drh72b3fbc2012-06-19 03:11:25 +0000968 sqlite3_snprintf(sizeof(zBuf), zBuf, "%!.15g", r1);
969 sqlite3AtoF(zBuf, &r2, 20, SQLITE_UTF8);
970 if( r1!=r2 ){
971 sqlite3_snprintf(sizeof(zBuf), zBuf, "%!.20e", r1);
972 }
973 sqlite3_result_text(context, zBuf, -1, SQLITE_TRANSIENT);
974 break;
975 }
976 case SQLITE_INTEGER: {
drhf4479502004-05-27 03:12:53 +0000977 sqlite3_result_value(context, argv[0]);
drhf9b596e2004-05-26 16:54:42 +0000978 break;
979 }
danielk19773f41e972004-06-08 00:39:01 +0000980 case SQLITE_BLOB: {
danielk19773f41e972004-06-08 00:39:01 +0000981 char *zText = 0;
danielk19773f41e972004-06-08 00:39:01 +0000982 char const *zBlob = sqlite3_value_blob(argv[0]);
drh1f0feef2007-05-15 13:27:07 +0000983 int nBlob = sqlite3_value_bytes(argv[0]);
984 assert( zBlob==sqlite3_value_blob(argv[0]) ); /* No encoding change */
drhb1a6c3c2008-03-20 16:30:17 +0000985 zText = (char *)contextMalloc(context, (2*(i64)nBlob)+4);
danielk1977a1644fd2007-08-29 12:31:25 +0000986 if( zText ){
danielk19773f41e972004-06-08 00:39:01 +0000987 int i;
988 for(i=0; i<nBlob; i++){
989 zText[(i*2)+2] = hexdigits[(zBlob[i]>>4)&0x0F];
990 zText[(i*2)+3] = hexdigits[(zBlob[i])&0x0F];
991 }
992 zText[(nBlob*2)+2] = '\'';
993 zText[(nBlob*2)+3] = '\0';
994 zText[0] = 'X';
995 zText[1] = '\'';
danielk1977d8123362004-06-12 09:25:12 +0000996 sqlite3_result_text(context, zText, -1, SQLITE_TRANSIENT);
drh17435752007-08-16 04:30:38 +0000997 sqlite3_free(zText);
danielk19773f41e972004-06-08 00:39:01 +0000998 }
999 break;
1000 }
drh9c054832004-05-31 18:51:57 +00001001 case SQLITE_TEXT: {
drh023ae032007-05-08 12:12:16 +00001002 int i,j;
1003 u64 n;
drh2646da72005-12-09 20:02:05 +00001004 const unsigned char *zArg = sqlite3_value_text(argv[0]);
drhf9b596e2004-05-26 16:54:42 +00001005 char *z;
1006
drh7a521cf2007-04-25 18:23:52 +00001007 if( zArg==0 ) return;
drh023ae032007-05-08 12:12:16 +00001008 for(i=0, n=0; zArg[i]; i++){ if( zArg[i]=='\'' ) n++; }
drhb1a6c3c2008-03-20 16:30:17 +00001009 z = contextMalloc(context, ((i64)i)+((i64)n)+3);
danielk1977a1644fd2007-08-29 12:31:25 +00001010 if( z ){
1011 z[0] = '\'';
1012 for(i=0, j=1; zArg[i]; i++){
1013 z[j++] = zArg[i];
1014 if( zArg[i]=='\'' ){
1015 z[j++] = '\'';
1016 }
drhf9b596e2004-05-26 16:54:42 +00001017 }
danielk1977a1644fd2007-08-29 12:31:25 +00001018 z[j++] = '\'';
1019 z[j] = 0;
1020 sqlite3_result_text(context, z, j, sqlite3_free);
drhf9b596e2004-05-26 16:54:42 +00001021 }
drha0df4cc2009-02-02 17:29:59 +00001022 break;
1023 }
1024 default: {
1025 assert( sqlite3_value_type(argv[0])==SQLITE_NULL );
1026 sqlite3_result_text(context, "NULL", 4, SQLITE_STATIC);
1027 break;
drhf9b596e2004-05-26 16:54:42 +00001028 }
drh47394702003-08-20 01:03:33 +00001029 }
1030}
1031
drh137c7282007-01-29 17:58:28 +00001032/*
drhd495d8c2013-02-22 19:34:25 +00001033** The unicode() function. Return the integer unicode code-point value
1034** for the first character of the input string.
1035*/
1036static void unicodeFunc(
1037 sqlite3_context *context,
1038 int argc,
1039 sqlite3_value **argv
1040){
1041 const unsigned char *z = sqlite3_value_text(argv[0]);
drh1d59d032013-03-01 23:40:26 +00001042 (void)argc;
drhd495d8c2013-02-22 19:34:25 +00001043 if( z && z[0] ) sqlite3_result_int(context, sqlite3Utf8Read(&z));
1044}
1045
1046/*
1047** The char() function takes zero or more arguments, each of which is
1048** an integer. It constructs a string where each character of the string
1049** is the unicode character for the corresponding integer argument.
1050*/
1051static void charFunc(
1052 sqlite3_context *context,
1053 int argc,
1054 sqlite3_value **argv
1055){
1056 unsigned char *z, *zOut;
1057 int i;
danb72cad12014-03-08 19:07:03 +00001058 zOut = z = sqlite3_malloc( argc*4+1 );
drhd495d8c2013-02-22 19:34:25 +00001059 if( z==0 ){
1060 sqlite3_result_error_nomem(context);
1061 return;
1062 }
1063 for(i=0; i<argc; i++){
mistachkinc9545442013-02-26 05:42:30 +00001064 sqlite3_int64 x;
drhd495d8c2013-02-22 19:34:25 +00001065 unsigned c;
1066 x = sqlite3_value_int64(argv[i]);
1067 if( x<0 || x>0x10ffff ) x = 0xfffd;
1068 c = (unsigned)(x & 0x1fffff);
drhfe7a5d12013-03-07 14:00:04 +00001069 if( c<0x00080 ){
1070 *zOut++ = (u8)(c&0xFF);
1071 }else if( c<0x00800 ){
1072 *zOut++ = 0xC0 + (u8)((c>>6)&0x1F);
1073 *zOut++ = 0x80 + (u8)(c & 0x3F);
1074 }else if( c<0x10000 ){
1075 *zOut++ = 0xE0 + (u8)((c>>12)&0x0F);
1076 *zOut++ = 0x80 + (u8)((c>>6) & 0x3F);
1077 *zOut++ = 0x80 + (u8)(c & 0x3F);
drhd495d8c2013-02-22 19:34:25 +00001078 }else{
drhfe7a5d12013-03-07 14:00:04 +00001079 *zOut++ = 0xF0 + (u8)((c>>18) & 0x07);
1080 *zOut++ = 0x80 + (u8)((c>>12) & 0x3F);
1081 *zOut++ = 0x80 + (u8)((c>>6) & 0x3F);
1082 *zOut++ = 0x80 + (u8)(c & 0x3F);
1083 } \
drhd495d8c2013-02-22 19:34:25 +00001084 }
drhbbf483f2014-09-09 20:30:24 +00001085 sqlite3_result_text64(context, (char*)z, zOut-z, sqlite3_free, SQLITE_UTF8);
drhd495d8c2013-02-22 19:34:25 +00001086}
1087
1088/*
drh137c7282007-01-29 17:58:28 +00001089** The hex() function. Interpret the argument as a blob. Return
1090** a hexadecimal rendering as text.
1091*/
1092static void hexFunc(
1093 sqlite3_context *context,
1094 int argc,
1095 sqlite3_value **argv
1096){
1097 int i, n;
1098 const unsigned char *pBlob;
1099 char *zHex, *z;
1100 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001101 UNUSED_PARAMETER(argc);
drh1f0feef2007-05-15 13:27:07 +00001102 pBlob = sqlite3_value_blob(argv[0]);
drh137c7282007-01-29 17:58:28 +00001103 n = sqlite3_value_bytes(argv[0]);
drh1f0feef2007-05-15 13:27:07 +00001104 assert( pBlob==sqlite3_value_blob(argv[0]) ); /* No encoding change */
drhb1a6c3c2008-03-20 16:30:17 +00001105 z = zHex = contextMalloc(context, ((i64)n)*2 + 1);
danielk1977a1644fd2007-08-29 12:31:25 +00001106 if( zHex ){
1107 for(i=0; i<n; i++, pBlob++){
1108 unsigned char c = *pBlob;
1109 *(z++) = hexdigits[(c>>4)&0xf];
1110 *(z++) = hexdigits[c&0xf];
1111 }
1112 *z = 0;
1113 sqlite3_result_text(context, zHex, n*2, sqlite3_free);
drh137c7282007-01-29 17:58:28 +00001114 }
drh137c7282007-01-29 17:58:28 +00001115}
1116
drh26b6d902007-03-17 13:27:54 +00001117/*
drh8cff3822007-05-02 02:08:28 +00001118** The zeroblob(N) function returns a zero-filled blob of size N bytes.
1119*/
1120static void zeroblobFunc(
1121 sqlite3_context *context,
1122 int argc,
1123 sqlite3_value **argv
1124){
drh98640a32007-06-07 19:08:32 +00001125 i64 n;
drh27e62db2009-04-02 10:16:17 +00001126 sqlite3 *db = sqlite3_context_db_handle(context);
drh8cff3822007-05-02 02:08:28 +00001127 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001128 UNUSED_PARAMETER(argc);
drh98640a32007-06-07 19:08:32 +00001129 n = sqlite3_value_int64(argv[0]);
drh27e62db2009-04-02 10:16:17 +00001130 testcase( n==db->aLimit[SQLITE_LIMIT_LENGTH] );
1131 testcase( n==db->aLimit[SQLITE_LIMIT_LENGTH]+1 );
1132 if( n>db->aLimit[SQLITE_LIMIT_LENGTH] ){
drh98640a32007-06-07 19:08:32 +00001133 sqlite3_result_error_toobig(context);
1134 }else{
drhab2f1f92010-01-11 18:26:42 +00001135 sqlite3_result_zeroblob(context, (int)n); /* IMP: R-00293-64994 */
drh98640a32007-06-07 19:08:32 +00001136 }
drh8cff3822007-05-02 02:08:28 +00001137}
1138
1139/*
drh26b6d902007-03-17 13:27:54 +00001140** The replace() function. Three arguments are all strings: call
1141** them A, B, and C. The result is also a string which is derived
drhf7b54962013-05-28 12:11:54 +00001142** from A by replacing every occurrence of B with C. The match
drh26b6d902007-03-17 13:27:54 +00001143** must be exact. Collating sequences are not used.
1144*/
1145static void replaceFunc(
1146 sqlite3_context *context,
1147 int argc,
1148 sqlite3_value **argv
1149){
1150 const unsigned char *zStr; /* The input string A */
1151 const unsigned char *zPattern; /* The pattern string B */
1152 const unsigned char *zRep; /* The replacement string C */
1153 unsigned char *zOut; /* The output */
1154 int nStr; /* Size of zStr */
1155 int nPattern; /* Size of zPattern */
1156 int nRep; /* Size of zRep */
drh2e6400b2007-05-08 15:46:18 +00001157 i64 nOut; /* Maximum size of zOut */
drh26b6d902007-03-17 13:27:54 +00001158 int loopLimit; /* Last zStr[] that might match zPattern[] */
1159 int i, j; /* Loop counters */
1160
1161 assert( argc==3 );
danielk1977f3d3c272008-11-19 16:52:44 +00001162 UNUSED_PARAMETER(argc);
drh26b6d902007-03-17 13:27:54 +00001163 zStr = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001164 if( zStr==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001165 nStr = sqlite3_value_bytes(argv[0]);
1166 assert( zStr==sqlite3_value_text(argv[0]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001167 zPattern = sqlite3_value_text(argv[1]);
drha605fe82009-02-01 18:08:40 +00001168 if( zPattern==0 ){
drh23336062009-02-03 13:19:12 +00001169 assert( sqlite3_value_type(argv[1])==SQLITE_NULL
1170 || sqlite3_context_db_handle(context)->mallocFailed );
drha605fe82009-02-01 18:08:40 +00001171 return;
1172 }
1173 if( zPattern[0]==0 ){
1174 assert( sqlite3_value_type(argv[1])!=SQLITE_NULL );
1175 sqlite3_result_value(context, argv[0]);
1176 return;
1177 }
drh1f0feef2007-05-15 13:27:07 +00001178 nPattern = sqlite3_value_bytes(argv[1]);
1179 assert( zPattern==sqlite3_value_text(argv[1]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001180 zRep = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +00001181 if( zRep==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001182 nRep = sqlite3_value_bytes(argv[2]);
1183 assert( zRep==sqlite3_value_text(argv[2]) );
drh2e6400b2007-05-08 15:46:18 +00001184 nOut = nStr + 1;
1185 assert( nOut<SQLITE_MAX_LENGTH );
drhb1a6c3c2008-03-20 16:30:17 +00001186 zOut = contextMalloc(context, (i64)nOut);
drh2e6400b2007-05-08 15:46:18 +00001187 if( zOut==0 ){
1188 return;
drh26b6d902007-03-17 13:27:54 +00001189 }
drh26b6d902007-03-17 13:27:54 +00001190 loopLimit = nStr - nPattern;
1191 for(i=j=0; i<=loopLimit; i++){
1192 if( zStr[i]!=zPattern[0] || memcmp(&zStr[i], zPattern, nPattern) ){
1193 zOut[j++] = zStr[i];
1194 }else{
drh4a50aac2007-08-23 02:47:53 +00001195 u8 *zOld;
drhbb4957f2008-03-20 14:03:29 +00001196 sqlite3 *db = sqlite3_context_db_handle(context);
drh2e6400b2007-05-08 15:46:18 +00001197 nOut += nRep - nPattern;
drh27e62db2009-04-02 10:16:17 +00001198 testcase( nOut-1==db->aLimit[SQLITE_LIMIT_LENGTH] );
1199 testcase( nOut-2==db->aLimit[SQLITE_LIMIT_LENGTH] );
1200 if( nOut-1>db->aLimit[SQLITE_LIMIT_LENGTH] ){
drha0206bc2007-05-08 15:15:02 +00001201 sqlite3_result_error_toobig(context);
drhb9755982010-07-24 16:34:37 +00001202 sqlite3_free(zOut);
danielk197717374e82007-05-08 14:39:04 +00001203 return;
1204 }
drh4a50aac2007-08-23 02:47:53 +00001205 zOld = zOut;
drh2e6400b2007-05-08 15:46:18 +00001206 zOut = sqlite3_realloc(zOut, (int)nOut);
1207 if( zOut==0 ){
danielk1977a1644fd2007-08-29 12:31:25 +00001208 sqlite3_result_error_nomem(context);
drhb9755982010-07-24 16:34:37 +00001209 sqlite3_free(zOld);
drh2e6400b2007-05-08 15:46:18 +00001210 return;
1211 }
drh26b6d902007-03-17 13:27:54 +00001212 memcpy(&zOut[j], zRep, nRep);
1213 j += nRep;
1214 i += nPattern-1;
1215 }
1216 }
drh2e6400b2007-05-08 15:46:18 +00001217 assert( j+nStr-i+1==nOut );
drh26b6d902007-03-17 13:27:54 +00001218 memcpy(&zOut[j], &zStr[i], nStr-i);
1219 j += nStr - i;
1220 assert( j<=nOut );
1221 zOut[j] = 0;
1222 sqlite3_result_text(context, (char*)zOut, j, sqlite3_free);
1223}
1224
drh309b3382007-03-17 17:52:42 +00001225/*
1226** Implementation of the TRIM(), LTRIM(), and RTRIM() functions.
1227** The userdata is 0x1 for left trim, 0x2 for right trim, 0x3 for both.
1228*/
1229static void trimFunc(
1230 sqlite3_context *context,
1231 int argc,
1232 sqlite3_value **argv
1233){
1234 const unsigned char *zIn; /* Input string */
1235 const unsigned char *zCharSet; /* Set of characters to trim */
1236 int nIn; /* Number of bytes in input */
drh7209c692008-04-27 18:40:11 +00001237 int flags; /* 1: trimleft 2: trimright 3: trim */
drhd1e3a612007-04-27 21:59:52 +00001238 int i; /* Loop counter */
drh1bd10f82008-12-10 21:19:56 +00001239 unsigned char *aLen = 0; /* Length of each character in zCharSet */
1240 unsigned char **azChar = 0; /* Individual characters in zCharSet */
drhd1e3a612007-04-27 21:59:52 +00001241 int nChar; /* Number of characters in zCharSet */
1242
drh309b3382007-03-17 17:52:42 +00001243 if( sqlite3_value_type(argv[0])==SQLITE_NULL ){
1244 return;
1245 }
1246 zIn = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001247 if( zIn==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001248 nIn = sqlite3_value_bytes(argv[0]);
1249 assert( zIn==sqlite3_value_text(argv[0]) );
drh309b3382007-03-17 17:52:42 +00001250 if( argc==1 ){
drhd1e3a612007-04-27 21:59:52 +00001251 static const unsigned char lenOne[] = { 1 };
danielk1977a4de4532008-09-02 15:44:08 +00001252 static unsigned char * const azOne[] = { (u8*)" " };
drhd1e3a612007-04-27 21:59:52 +00001253 nChar = 1;
drh8cff3822007-05-02 02:08:28 +00001254 aLen = (u8*)lenOne;
danielk1977bc67da42007-12-11 04:23:19 +00001255 azChar = (unsigned char **)azOne;
drhd1e3a612007-04-27 21:59:52 +00001256 zCharSet = 0;
drh7a521cf2007-04-25 18:23:52 +00001257 }else if( (zCharSet = sqlite3_value_text(argv[1]))==0 ){
drh309b3382007-03-17 17:52:42 +00001258 return;
drhd1e3a612007-04-27 21:59:52 +00001259 }else{
1260 const unsigned char *z;
1261 for(z=zCharSet, nChar=0; *z; nChar++){
drh4a919112007-05-15 11:55:09 +00001262 SQLITE_SKIP_UTF8(z);
drhd1e3a612007-04-27 21:59:52 +00001263 }
1264 if( nChar>0 ){
drhb1a6c3c2008-03-20 16:30:17 +00001265 azChar = contextMalloc(context, ((i64)nChar)*(sizeof(char*)+1));
drhd1e3a612007-04-27 21:59:52 +00001266 if( azChar==0 ){
1267 return;
1268 }
1269 aLen = (unsigned char*)&azChar[nChar];
1270 for(z=zCharSet, nChar=0; *z; nChar++){
danielk1977bc67da42007-12-11 04:23:19 +00001271 azChar[nChar] = (unsigned char *)z;
drh4a919112007-05-15 11:55:09 +00001272 SQLITE_SKIP_UTF8(z);
drh3abbd392008-12-10 23:04:13 +00001273 aLen[nChar] = (u8)(z - azChar[nChar]);
drhd1e3a612007-04-27 21:59:52 +00001274 }
1275 }
drh309b3382007-03-17 17:52:42 +00001276 }
drhd1e3a612007-04-27 21:59:52 +00001277 if( nChar>0 ){
shane1fc41292008-07-08 22:28:48 +00001278 flags = SQLITE_PTR_TO_INT(sqlite3_user_data(context));
drh309b3382007-03-17 17:52:42 +00001279 if( flags & 1 ){
drhd1e3a612007-04-27 21:59:52 +00001280 while( nIn>0 ){
drh1bd10f82008-12-10 21:19:56 +00001281 int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001282 for(i=0; i<nChar; i++){
1283 len = aLen[i];
drh27e62db2009-04-02 10:16:17 +00001284 if( len<=nIn && memcmp(zIn, azChar[i], len)==0 ) break;
drhd1e3a612007-04-27 21:59:52 +00001285 }
1286 if( i>=nChar ) break;
1287 zIn += len;
1288 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001289 }
1290 }
1291 if( flags & 2 ){
drhd1e3a612007-04-27 21:59:52 +00001292 while( nIn>0 ){
drh1bd10f82008-12-10 21:19:56 +00001293 int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001294 for(i=0; i<nChar; i++){
1295 len = aLen[i];
1296 if( len<=nIn && memcmp(&zIn[nIn-len],azChar[i],len)==0 ) break;
1297 }
1298 if( i>=nChar ) break;
1299 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001300 }
1301 }
drhd1e3a612007-04-27 21:59:52 +00001302 if( zCharSet ){
1303 sqlite3_free(azChar);
1304 }
drh309b3382007-03-17 17:52:42 +00001305 }
1306 sqlite3_result_text(context, (char*)zIn, nIn, SQLITE_TRANSIENT);
1307}
drh26b6d902007-03-17 13:27:54 +00001308
danielk1977a4de4532008-09-02 15:44:08 +00001309
drh2ba3ccc2009-12-08 02:06:08 +00001310/* IMP: R-25361-16150 This function is omitted from SQLite by default. It
1311** is only available if the SQLITE_SOUNDEX compile-time option is used
1312** when SQLite is built.
1313*/
drhd24cc422003-03-27 12:51:24 +00001314#ifdef SQLITE_SOUNDEX
1315/*
1316** Compute the soundex encoding of a word.
drh2ba3ccc2009-12-08 02:06:08 +00001317**
1318** IMP: R-59782-00072 The soundex(X) function returns a string that is the
1319** soundex encoding of the string X.
drhd24cc422003-03-27 12:51:24 +00001320*/
drh137c7282007-01-29 17:58:28 +00001321static void soundexFunc(
1322 sqlite3_context *context,
1323 int argc,
1324 sqlite3_value **argv
1325){
drhd24cc422003-03-27 12:51:24 +00001326 char zResult[8];
drh4c755c02004-08-08 20:22:17 +00001327 const u8 *zIn;
drhd24cc422003-03-27 12:51:24 +00001328 int i, j;
1329 static const unsigned char iCode[] = {
1330 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1331 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1332 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1333 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1334 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1335 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1336 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1337 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1338 };
1339 assert( argc==1 );
drh4c755c02004-08-08 20:22:17 +00001340 zIn = (u8*)sqlite3_value_text(argv[0]);
drhbdf67e02006-08-19 11:34:01 +00001341 if( zIn==0 ) zIn = (u8*)"";
drhdc86e2b2009-01-24 11:30:42 +00001342 for(i=0; zIn[i] && !sqlite3Isalpha(zIn[i]); i++){}
drhd24cc422003-03-27 12:51:24 +00001343 if( zIn[i] ){
drhbdf67e02006-08-19 11:34:01 +00001344 u8 prevcode = iCode[zIn[i]&0x7f];
danielk197778ca0e72009-01-20 16:53:39 +00001345 zResult[0] = sqlite3Toupper(zIn[i]);
drhd24cc422003-03-27 12:51:24 +00001346 for(j=1; j<4 && zIn[i]; i++){
1347 int code = iCode[zIn[i]&0x7f];
1348 if( code>0 ){
drhbdf67e02006-08-19 11:34:01 +00001349 if( code!=prevcode ){
1350 prevcode = code;
1351 zResult[j++] = code + '0';
1352 }
1353 }else{
1354 prevcode = 0;
drhd24cc422003-03-27 12:51:24 +00001355 }
1356 }
1357 while( j<4 ){
1358 zResult[j++] = '0';
1359 }
1360 zResult[j] = 0;
danielk1977d8123362004-06-12 09:25:12 +00001361 sqlite3_result_text(context, zResult, 4, SQLITE_TRANSIENT);
drhd24cc422003-03-27 12:51:24 +00001362 }else{
drh2ba3ccc2009-12-08 02:06:08 +00001363 /* IMP: R-64894-50321 The string "?000" is returned if the argument
1364 ** is NULL or contains no ASCII alphabetic characters. */
danielk1977d8123362004-06-12 09:25:12 +00001365 sqlite3_result_text(context, "?000", 4, SQLITE_STATIC);
drhd24cc422003-03-27 12:51:24 +00001366 }
1367}
drh2ba3ccc2009-12-08 02:06:08 +00001368#endif /* SQLITE_SOUNDEX */
drhd24cc422003-03-27 12:51:24 +00001369
drhfdb83b22006-06-17 14:12:47 +00001370#ifndef SQLITE_OMIT_LOAD_EXTENSION
1371/*
1372** A function that loads a shared-library extension then returns NULL.
1373*/
1374static void loadExt(sqlite3_context *context, int argc, sqlite3_value **argv){
danielk197765fd59f2006-06-24 11:51:33 +00001375 const char *zFile = (const char *)sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001376 const char *zProc;
drhfa4a4b92008-03-19 21:45:51 +00001377 sqlite3 *db = sqlite3_context_db_handle(context);
drhfdb83b22006-06-17 14:12:47 +00001378 char *zErrMsg = 0;
1379
1380 if( argc==2 ){
danielk197765fd59f2006-06-24 11:51:33 +00001381 zProc = (const char *)sqlite3_value_text(argv[1]);
drh7a521cf2007-04-25 18:23:52 +00001382 }else{
1383 zProc = 0;
drhfdb83b22006-06-17 14:12:47 +00001384 }
drh7a521cf2007-04-25 18:23:52 +00001385 if( zFile && sqlite3_load_extension(db, zFile, zProc, &zErrMsg) ){
drhfdb83b22006-06-17 14:12:47 +00001386 sqlite3_result_error(context, zErrMsg, -1);
1387 sqlite3_free(zErrMsg);
1388 }
1389}
1390#endif
1391
danielk197701427a62005-01-11 13:02:33 +00001392
drh0ac65892002-04-20 14:24:41 +00001393/*
drhd3a149e2002-02-24 17:12:53 +00001394** An instance of the following structure holds the context of a
drhdd5baa92002-02-27 19:50:59 +00001395** sum() or avg() aggregate computation.
1396*/
1397typedef struct SumCtx SumCtx;
1398struct SumCtx {
drh8c08e862006-02-11 17:34:00 +00001399 double rSum; /* Floating point sum */
1400 i64 iSum; /* Integer sum */
1401 i64 cnt; /* Number of elements summed */
1402 u8 overflow; /* True if integer overflow seen */
1403 u8 approx; /* True if non-integer value was input to the sum */
drhdd5baa92002-02-27 19:50:59 +00001404};
1405
1406/*
drha97fdd32006-01-12 22:17:50 +00001407** Routines used to compute the sum, average, and total.
1408**
1409** The SUM() function follows the (broken) SQL standard which means
1410** that it returns NULL if it sums over no inputs. TOTAL returns
1411** 0.0 in that case. In addition, TOTAL always returns a float where
1412** SUM might return an integer if it never encounters a floating point
drhc806d852006-05-11 13:25:39 +00001413** value. TOTAL never fails, but SUM might through an exception if
1414** it overflows an integer.
drhdd5baa92002-02-27 19:50:59 +00001415*/
danielk19770ae8b832004-05-25 12:05:56 +00001416static void sumStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drhdd5baa92002-02-27 19:50:59 +00001417 SumCtx *p;
drh3d1d95e2005-09-08 10:37:01 +00001418 int type;
drh3f219f42005-09-08 19:45:57 +00001419 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001420 UNUSED_PARAMETER(argc);
drh4f26d6c2004-05-26 23:25:30 +00001421 p = sqlite3_aggregate_context(context, sizeof(*p));
drh29d72102006-02-09 22:13:41 +00001422 type = sqlite3_value_numeric_type(argv[0]);
drh3d1d95e2005-09-08 10:37:01 +00001423 if( p && type!=SQLITE_NULL ){
drh739105c2002-05-29 23:22:23 +00001424 p->cnt++;
drh29d72102006-02-09 22:13:41 +00001425 if( type==SQLITE_INTEGER ){
drh8c08e862006-02-11 17:34:00 +00001426 i64 v = sqlite3_value_int64(argv[0]);
1427 p->rSum += v;
drh158b9cb2011-03-05 20:59:46 +00001428 if( (p->approx|p->overflow)==0 && sqlite3AddInt64(&p->iSum, v) ){
1429 p->overflow = 1;
drh29d72102006-02-09 22:13:41 +00001430 }
1431 }else{
drh8c08e862006-02-11 17:34:00 +00001432 p->rSum += sqlite3_value_double(argv[0]);
drh29d72102006-02-09 22:13:41 +00001433 p->approx = 1;
drh3f219f42005-09-08 19:45:57 +00001434 }
drh739105c2002-05-29 23:22:23 +00001435 }
drhdd5baa92002-02-27 19:50:59 +00001436}
danielk19770ae8b832004-05-25 12:05:56 +00001437static void sumFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001438 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001439 p = sqlite3_aggregate_context(context, 0);
drhc2bd9132005-09-08 20:37:43 +00001440 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001441 if( p->overflow ){
1442 sqlite3_result_error(context,"integer overflow",-1);
1443 }else if( p->approx ){
1444 sqlite3_result_double(context, p->rSum);
drhc2bd9132005-09-08 20:37:43 +00001445 }else{
drh8c08e862006-02-11 17:34:00 +00001446 sqlite3_result_int64(context, p->iSum);
drhc2bd9132005-09-08 20:37:43 +00001447 }
drh3d1d95e2005-09-08 10:37:01 +00001448 }
drhdd5baa92002-02-27 19:50:59 +00001449}
danielk19770ae8b832004-05-25 12:05:56 +00001450static void avgFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001451 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001452 p = sqlite3_aggregate_context(context, 0);
drh739105c2002-05-29 23:22:23 +00001453 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001454 sqlite3_result_double(context, p->rSum/(double)p->cnt);
drhdd5baa92002-02-27 19:50:59 +00001455 }
1456}
drha97fdd32006-01-12 22:17:50 +00001457static void totalFinalize(sqlite3_context *context){
1458 SumCtx *p;
1459 p = sqlite3_aggregate_context(context, 0);
shanefbd60f82009-02-04 03:59:25 +00001460 /* (double)0 In case of SQLITE_OMIT_FLOATING_POINT... */
1461 sqlite3_result_double(context, p ? p->rSum : (double)0);
drha97fdd32006-01-12 22:17:50 +00001462}
drhdd5baa92002-02-27 19:50:59 +00001463
1464/*
drh0bce8352002-02-28 00:41:10 +00001465** The following structure keeps track of state information for the
1466** count() aggregate function.
1467*/
1468typedef struct CountCtx CountCtx;
1469struct CountCtx {
drhfc6ad392006-02-09 13:38:19 +00001470 i64 n;
drh0bce8352002-02-28 00:41:10 +00001471};
drhdd5baa92002-02-27 19:50:59 +00001472
drh0bce8352002-02-28 00:41:10 +00001473/*
1474** Routines to implement the count() aggregate function.
1475*/
danielk19770ae8b832004-05-25 12:05:56 +00001476static void countStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drh0bce8352002-02-28 00:41:10 +00001477 CountCtx *p;
drh4f26d6c2004-05-26 23:25:30 +00001478 p = sqlite3_aggregate_context(context, sizeof(*p));
drh9c054832004-05-31 18:51:57 +00001479 if( (argc==0 || SQLITE_NULL!=sqlite3_value_type(argv[0])) && p ){
drh0bce8352002-02-28 00:41:10 +00001480 p->n++;
1481 }
drh2e79c3d2009-04-08 23:04:14 +00001482
drhd3264c72009-04-15 13:39:47 +00001483#ifndef SQLITE_OMIT_DEPRECATED
drh2e79c3d2009-04-08 23:04:14 +00001484 /* The sqlite3_aggregate_count() function is deprecated. But just to make
1485 ** sure it still operates correctly, verify that its count agrees with our
1486 ** internal count when using count(*) and when the total count can be
1487 ** expressed as a 32-bit integer. */
1488 assert( argc==1 || p==0 || p->n>0x7fffffff
1489 || p->n==sqlite3_aggregate_count(context) );
drhd3264c72009-04-15 13:39:47 +00001490#endif
drh0bce8352002-02-28 00:41:10 +00001491}
danielk19770ae8b832004-05-25 12:05:56 +00001492static void countFinalize(sqlite3_context *context){
drh0bce8352002-02-28 00:41:10 +00001493 CountCtx *p;
drhabfcea22005-09-06 20:36:48 +00001494 p = sqlite3_aggregate_context(context, 0);
drhfc6ad392006-02-09 13:38:19 +00001495 sqlite3_result_int64(context, p ? p->n : 0);
drh0bce8352002-02-28 00:41:10 +00001496}
1497
1498/*
drh0bce8352002-02-28 00:41:10 +00001499** Routines to implement min() and max() aggregate functions.
1500*/
danielk197762c14b32008-11-19 09:05:26 +00001501static void minmaxStep(
1502 sqlite3_context *context,
1503 int NotUsed,
1504 sqlite3_value **argv
1505){
danielk197788208052004-05-25 01:13:20 +00001506 Mem *pArg = (Mem *)argv[0];
drh9eb516c2004-07-18 20:52:32 +00001507 Mem *pBest;
danielk197762c14b32008-11-19 09:05:26 +00001508 UNUSED_PARAMETER(NotUsed);
drh9eb516c2004-07-18 20:52:32 +00001509
drh9eb516c2004-07-18 20:52:32 +00001510 pBest = (Mem *)sqlite3_aggregate_context(context, sizeof(*pBest));
danielk19773aeab9e2004-06-24 00:20:04 +00001511 if( !pBest ) return;
drh268380c2004-02-25 13:47:31 +00001512
drh94a6d992012-02-02 18:42:09 +00001513 if( sqlite3_value_type(argv[0])==SQLITE_NULL ){
1514 if( pBest->flags ) sqlite3SkipAccumulatorLoad(context);
1515 }else if( pBest->flags ){
drh9eb516c2004-07-18 20:52:32 +00001516 int max;
1517 int cmp;
danielk1977dc1bdc42004-06-11 10:51:27 +00001518 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk19777e18c252004-05-25 11:47:24 +00001519 /* This step function is used for both the min() and max() aggregates,
1520 ** the only difference between the two being that the sense of the
1521 ** comparison is inverted. For the max() aggregate, the
1522 ** sqlite3_user_data() function returns (void *)-1. For min() it
1523 ** returns (void *)db, where db is the sqlite3* database pointer.
1524 ** Therefore the next statement sets variable 'max' to 1 for the max()
1525 ** aggregate, or 0 for min().
1526 */
drh309b3382007-03-17 17:52:42 +00001527 max = sqlite3_user_data(context)!=0;
danielk1977dc1bdc42004-06-11 10:51:27 +00001528 cmp = sqlite3MemCompare(pBest, pArg, pColl);
danielk197788208052004-05-25 01:13:20 +00001529 if( (max && cmp<0) || (!max && cmp>0) ){
drhb21c8cd2007-08-21 19:33:56 +00001530 sqlite3VdbeMemCopy(pBest, pArg);
drh7a957892012-02-02 17:35:43 +00001531 }else{
1532 sqlite3SkipAccumulatorLoad(context);
danielk197788208052004-05-25 01:13:20 +00001533 }
drh268380c2004-02-25 13:47:31 +00001534 }else{
drh035e5632014-09-16 14:16:31 +00001535 pBest->db = sqlite3_context_db_handle(context);
drhb21c8cd2007-08-21 19:33:56 +00001536 sqlite3VdbeMemCopy(pBest, pArg);
drh0bce8352002-02-28 00:41:10 +00001537 }
1538}
danielk19770ae8b832004-05-25 12:05:56 +00001539static void minMaxFinalize(sqlite3_context *context){
danielk197788208052004-05-25 01:13:20 +00001540 sqlite3_value *pRes;
drhabfcea22005-09-06 20:36:48 +00001541 pRes = (sqlite3_value *)sqlite3_aggregate_context(context, 0);
1542 if( pRes ){
drh94a6d992012-02-02 18:42:09 +00001543 if( pRes->flags ){
drhabfcea22005-09-06 20:36:48 +00001544 sqlite3_result_value(context, pRes);
1545 }
1546 sqlite3VdbeMemRelease(pRes);
drh0bce8352002-02-28 00:41:10 +00001547 }
1548}
drhdd5baa92002-02-27 19:50:59 +00001549
drhb0689692007-11-01 17:38:30 +00001550/*
1551** group_concat(EXPR, ?SEPARATOR?)
1552*/
1553static void groupConcatStep(
1554 sqlite3_context *context,
1555 int argc,
1556 sqlite3_value **argv
1557){
1558 const char *zVal;
drhade86482007-11-28 22:36:40 +00001559 StrAccum *pAccum;
drhb0689692007-11-01 17:38:30 +00001560 const char *zSep;
drh07d31172009-02-02 21:57:05 +00001561 int nVal, nSep;
1562 assert( argc==1 || argc==2 );
1563 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drhade86482007-11-28 22:36:40 +00001564 pAccum = (StrAccum*)sqlite3_aggregate_context(context, sizeof(*pAccum));
1565
1566 if( pAccum ){
drhbb4957f2008-03-20 14:03:29 +00001567 sqlite3 *db = sqlite3_context_db_handle(context);
drh8bfd7192009-06-19 16:44:41 +00001568 int firstTerm = pAccum->useMalloc==0;
drhb9755982010-07-24 16:34:37 +00001569 pAccum->useMalloc = 2;
drhbb4957f2008-03-20 14:03:29 +00001570 pAccum->mxAlloc = db->aLimit[SQLITE_LIMIT_LENGTH];
drh8bfd7192009-06-19 16:44:41 +00001571 if( !firstTerm ){
drh07d31172009-02-02 21:57:05 +00001572 if( argc==2 ){
1573 zSep = (char*)sqlite3_value_text(argv[1]);
1574 nSep = sqlite3_value_bytes(argv[1]);
drhb0689692007-11-01 17:38:30 +00001575 }else{
1576 zSep = ",";
drhade86482007-11-28 22:36:40 +00001577 nSep = 1;
drhb0689692007-11-01 17:38:30 +00001578 }
drha9ab4812013-12-11 11:00:44 +00001579 if( nSep ) sqlite3StrAccumAppend(pAccum, zSep, nSep);
drhb0689692007-11-01 17:38:30 +00001580 }
drh07d31172009-02-02 21:57:05 +00001581 zVal = (char*)sqlite3_value_text(argv[0]);
1582 nVal = sqlite3_value_bytes(argv[0]);
drh8009c9b2014-05-07 20:24:00 +00001583 if( zVal ) sqlite3StrAccumAppend(pAccum, zVal, nVal);
drhb0689692007-11-01 17:38:30 +00001584 }
1585}
1586static void groupConcatFinalize(sqlite3_context *context){
drhade86482007-11-28 22:36:40 +00001587 StrAccum *pAccum;
1588 pAccum = sqlite3_aggregate_context(context, 0);
1589 if( pAccum ){
drhb49bc862013-08-21 21:12:10 +00001590 if( pAccum->accError==STRACCUM_TOOBIG ){
drhade86482007-11-28 22:36:40 +00001591 sqlite3_result_error_toobig(context);
drhb49bc862013-08-21 21:12:10 +00001592 }else if( pAccum->accError==STRACCUM_NOMEM ){
drhade86482007-11-28 22:36:40 +00001593 sqlite3_result_error_nomem(context);
1594 }else{
1595 sqlite3_result_text(context, sqlite3StrAccumFinish(pAccum), -1,
1596 sqlite3_free);
1597 }
drhb0689692007-11-01 17:38:30 +00001598 }
1599}
drh4e5ffc52004-08-31 00:52:37 +00001600
drhd3a149e2002-02-24 17:12:53 +00001601/*
drha4741842010-04-25 20:58:37 +00001602** This routine does per-connection function registration. Most
1603** of the built-in functions above are part of the global function set.
1604** This routine only deals with those that are not global.
drhdc04c582002-02-24 01:55:15 +00001605*/
drh9bb575f2004-09-06 17:24:11 +00001606void sqlite3RegisterBuiltinFunctions(sqlite3 *db){
drha4741842010-04-25 20:58:37 +00001607 int rc = sqlite3_overload_function(db, "MATCH", 2);
1608 assert( rc==SQLITE_NOMEM || rc==SQLITE_OK );
1609 if( rc==SQLITE_NOMEM ){
1610 db->mallocFailed = 1;
danielk1977832a58a2007-06-22 15:21:15 +00001611 }
drh55ef4d92005-08-14 01:20:37 +00001612}
1613
1614/*
1615** Set the LIKEOPT flag on the 2-argument function with the given name.
1616*/
drh1bd10f82008-12-10 21:19:56 +00001617static void setLikeOptFlag(sqlite3 *db, const char *zName, u8 flagVal){
drh55ef4d92005-08-14 01:20:37 +00001618 FuncDef *pDef;
drhea678832008-12-10 19:26:22 +00001619 pDef = sqlite3FindFunction(db, zName, sqlite3Strlen30(zName),
1620 2, SQLITE_UTF8, 0);
drhd27135a2009-04-02 13:36:37 +00001621 if( ALWAYS(pDef) ){
drhd36e1042013-09-06 13:10:12 +00001622 pDef->funcFlags |= flagVal;
drh55ef4d92005-08-14 01:20:37 +00001623 }
1624}
1625
1626/*
1627** Register the built-in LIKE and GLOB functions. The caseSensitive
1628** parameter determines whether or not the LIKE operator is case
1629** sensitive. GLOB is always case sensitive.
1630*/
1631void sqlite3RegisterLikeFunctions(sqlite3 *db, int caseSensitive){
1632 struct compareInfo *pInfo;
1633 if( caseSensitive ){
1634 pInfo = (struct compareInfo*)&likeInfoAlt;
1635 }else{
1636 pInfo = (struct compareInfo*)&likeInfoNorm;
1637 }
drh901e9942010-12-15 18:54:37 +00001638 sqlite3CreateFunc(db, "like", 2, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0);
1639 sqlite3CreateFunc(db, "like", 3, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0);
1640 sqlite3CreateFunc(db, "glob", 2, SQLITE_UTF8,
dand2199f02010-08-27 17:48:52 +00001641 (struct compareInfo*)&globInfo, likeFunc, 0, 0, 0);
drhd64fe2f2005-08-28 17:00:23 +00001642 setLikeOptFlag(db, "glob", SQLITE_FUNC_LIKE | SQLITE_FUNC_CASE);
1643 setLikeOptFlag(db, "like",
1644 caseSensitive ? (SQLITE_FUNC_LIKE | SQLITE_FUNC_CASE) : SQLITE_FUNC_LIKE);
drh55ef4d92005-08-14 01:20:37 +00001645}
1646
1647/*
1648** pExpr points to an expression which implements a function. If
1649** it is appropriate to apply the LIKE optimization to that function
1650** then set aWc[0] through aWc[2] to the wildcard characters and
1651** return TRUE. If the function is not a LIKE-style function then
1652** return FALSE.
1653*/
drhd64fe2f2005-08-28 17:00:23 +00001654int sqlite3IsLikeFunction(sqlite3 *db, Expr *pExpr, int *pIsNocase, char *aWc){
drh55ef4d92005-08-14 01:20:37 +00001655 FuncDef *pDef;
danielk19776ab3a2e2009-02-19 14:39:25 +00001656 if( pExpr->op!=TK_FUNCTION
1657 || !pExpr->x.pList
1658 || pExpr->x.pList->nExpr!=2
1659 ){
drh55ef4d92005-08-14 01:20:37 +00001660 return 0;
1661 }
danielk19776ab3a2e2009-02-19 14:39:25 +00001662 assert( !ExprHasProperty(pExpr, EP_xIsSelect) );
drh33e619f2009-05-28 01:00:55 +00001663 pDef = sqlite3FindFunction(db, pExpr->u.zToken,
1664 sqlite3Strlen30(pExpr->u.zToken),
drhb7916a72009-05-27 10:31:29 +00001665 2, SQLITE_UTF8, 0);
drhd36e1042013-09-06 13:10:12 +00001666 if( NEVER(pDef==0) || (pDef->funcFlags & SQLITE_FUNC_LIKE)==0 ){
drh55ef4d92005-08-14 01:20:37 +00001667 return 0;
1668 }
1669
1670 /* The memcpy() statement assumes that the wildcard characters are
1671 ** the first three statements in the compareInfo structure. The
1672 ** asserts() that follow verify that assumption
1673 */
1674 memcpy(aWc, pDef->pUserData, 3);
1675 assert( (char*)&likeInfoAlt == (char*)&likeInfoAlt.matchAll );
1676 assert( &((char*)&likeInfoAlt)[1] == (char*)&likeInfoAlt.matchOne );
1677 assert( &((char*)&likeInfoAlt)[2] == (char*)&likeInfoAlt.matchSet );
drhd36e1042013-09-06 13:10:12 +00001678 *pIsNocase = (pDef->funcFlags & SQLITE_FUNC_CASE)==0;
drh55ef4d92005-08-14 01:20:37 +00001679 return 1;
drhdc04c582002-02-24 01:55:15 +00001680}
danielk19778c0a7912008-08-20 14:49:23 +00001681
drh70a8ca32008-08-21 18:49:27 +00001682/*
peter.d.reid60ec9142014-09-06 16:39:46 +00001683** All of the FuncDef structures in the aBuiltinFunc[] array above
drh777c5382008-08-21 20:21:34 +00001684** to the global function hash table. This occurs at start-time (as
1685** a consequence of calling sqlite3_initialize()).
1686**
1687** After this routine runs
drh70a8ca32008-08-21 18:49:27 +00001688*/
1689void sqlite3RegisterGlobalFunctions(void){
danielk197793ce7412008-09-01 19:14:02 +00001690 /*
1691 ** The following array holds FuncDef structures for all of the functions
1692 ** defined in this file.
1693 **
1694 ** The array cannot be constant since changes are made to the
1695 ** FuncDef.pHash elements at start-time. The elements of this array
1696 ** are read-only after initialization is complete.
1697 */
1698 static SQLITE_WSD FuncDef aBuiltinFunc[] = {
1699 FUNCTION(ltrim, 1, 1, 0, trimFunc ),
1700 FUNCTION(ltrim, 2, 1, 0, trimFunc ),
1701 FUNCTION(rtrim, 1, 2, 0, trimFunc ),
1702 FUNCTION(rtrim, 2, 2, 0, trimFunc ),
1703 FUNCTION(trim, 1, 3, 0, trimFunc ),
1704 FUNCTION(trim, 2, 3, 0, trimFunc ),
1705 FUNCTION(min, -1, 0, 1, minmaxFunc ),
1706 FUNCTION(min, 0, 0, 1, 0 ),
drh9588ad92014-09-15 14:46:02 +00001707 AGGREGATE2(min, 1, 0, 1, minmaxStep, minMaxFinalize,
1708 SQLITE_FUNC_MINMAX ),
danielk197793ce7412008-09-01 19:14:02 +00001709 FUNCTION(max, -1, 1, 1, minmaxFunc ),
1710 FUNCTION(max, 0, 1, 1, 0 ),
drh9588ad92014-09-15 14:46:02 +00001711 AGGREGATE2(max, 1, 1, 1, minmaxStep, minMaxFinalize,
1712 SQLITE_FUNC_MINMAX ),
drha748fdc2012-03-28 01:34:47 +00001713 FUNCTION2(typeof, 1, 0, 0, typeofFunc, SQLITE_FUNC_TYPEOF),
1714 FUNCTION2(length, 1, 0, 0, lengthFunc, SQLITE_FUNC_LENGTH),
drhd55e0722012-10-25 03:07:29 +00001715 FUNCTION(instr, 2, 0, 0, instrFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001716 FUNCTION(substr, 2, 0, 0, substrFunc ),
1717 FUNCTION(substr, 3, 0, 0, substrFunc ),
drha5c14162013-12-17 15:03:06 +00001718 FUNCTION(printf, -1, 0, 0, printfFunc ),
drhd495d8c2013-02-22 19:34:25 +00001719 FUNCTION(unicode, 1, 0, 0, unicodeFunc ),
1720 FUNCTION(char, -1, 0, 0, charFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001721 FUNCTION(abs, 1, 0, 0, absFunc ),
shanefbd60f82009-02-04 03:59:25 +00001722#ifndef SQLITE_OMIT_FLOATING_POINT
danielk197793ce7412008-09-01 19:14:02 +00001723 FUNCTION(round, 1, 0, 0, roundFunc ),
1724 FUNCTION(round, 2, 0, 0, roundFunc ),
shanefbd60f82009-02-04 03:59:25 +00001725#endif
danielk197793ce7412008-09-01 19:14:02 +00001726 FUNCTION(upper, 1, 0, 0, upperFunc ),
1727 FUNCTION(lower, 1, 0, 0, lowerFunc ),
1728 FUNCTION(coalesce, 1, 0, 0, 0 ),
danielk197793ce7412008-09-01 19:14:02 +00001729 FUNCTION(coalesce, 0, 0, 0, 0 ),
drhcca9f3d2013-09-06 15:23:29 +00001730 FUNCTION2(coalesce, -1, 0, 0, noopFunc, SQLITE_FUNC_COALESCE),
danielk197793ce7412008-09-01 19:14:02 +00001731 FUNCTION(hex, 1, 0, 0, hexFunc ),
drhcca9f3d2013-09-06 15:23:29 +00001732 FUNCTION2(ifnull, 2, 0, 0, noopFunc, SQLITE_FUNC_COALESCE),
1733 FUNCTION2(unlikely, 1, 0, 0, noopFunc, SQLITE_FUNC_UNLIKELY),
drhaae0f9e2013-09-11 11:38:58 +00001734 FUNCTION2(likelihood, 2, 0, 0, noopFunc, SQLITE_FUNC_UNLIKELY),
drh03202a92014-06-17 16:11:28 +00001735 FUNCTION2(likely, 1, 0, 0, noopFunc, SQLITE_FUNC_UNLIKELY),
drhb1fba282013-11-21 14:33:48 +00001736 VFUNCTION(random, 0, 0, 0, randomFunc ),
1737 VFUNCTION(randomblob, 1, 0, 0, randomBlob ),
danielk197793ce7412008-09-01 19:14:02 +00001738 FUNCTION(nullif, 2, 0, 1, nullifFunc ),
1739 FUNCTION(sqlite_version, 0, 0, 0, versionFunc ),
drh47baebc2009-08-14 16:01:24 +00001740 FUNCTION(sqlite_source_id, 0, 0, 0, sourceidFunc ),
drh840561f2011-04-27 18:08:42 +00001741 FUNCTION(sqlite_log, 2, 0, 0, errlogFunc ),
drhf442e332014-09-10 19:01:14 +00001742#if SQLITE_USER_AUTHENTICATION
1743 FUNCTION(sqlite_crypt, 2, 0, 0, sqlite3CryptFunc ),
1744#endif
shanehdc97a8c2010-02-23 20:08:35 +00001745#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
drh8bb76d32010-02-26 16:37:47 +00001746 FUNCTION(sqlite_compileoption_used,1, 0, 0, compileoptionusedFunc ),
1747 FUNCTION(sqlite_compileoption_get, 1, 0, 0, compileoptiongetFunc ),
shanehdc97a8c2010-02-23 20:08:35 +00001748#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
danielk197793ce7412008-09-01 19:14:02 +00001749 FUNCTION(quote, 1, 0, 0, quoteFunc ),
drhb1fba282013-11-21 14:33:48 +00001750 VFUNCTION(last_insert_rowid, 0, 0, 0, last_insert_rowid),
1751 VFUNCTION(changes, 0, 0, 0, changes ),
1752 VFUNCTION(total_changes, 0, 0, 0, total_changes ),
danielk197793ce7412008-09-01 19:14:02 +00001753 FUNCTION(replace, 3, 0, 0, replaceFunc ),
1754 FUNCTION(zeroblob, 1, 0, 0, zeroblobFunc ),
1755 #ifdef SQLITE_SOUNDEX
1756 FUNCTION(soundex, 1, 0, 0, soundexFunc ),
1757 #endif
1758 #ifndef SQLITE_OMIT_LOAD_EXTENSION
1759 FUNCTION(load_extension, 1, 0, 0, loadExt ),
1760 FUNCTION(load_extension, 2, 0, 0, loadExt ),
1761 #endif
1762 AGGREGATE(sum, 1, 0, 0, sumStep, sumFinalize ),
1763 AGGREGATE(total, 1, 0, 0, sumStep, totalFinalize ),
1764 AGGREGATE(avg, 1, 0, 0, sumStep, avgFinalize ),
drh9588ad92014-09-15 14:46:02 +00001765 AGGREGATE2(count, 0, 0, 0, countStep, countFinalize,
1766 SQLITE_FUNC_COUNT ),
danielk197793ce7412008-09-01 19:14:02 +00001767 AGGREGATE(count, 1, 0, 0, countStep, countFinalize ),
drh07d31172009-02-02 21:57:05 +00001768 AGGREGATE(group_concat, 1, 0, 0, groupConcatStep, groupConcatFinalize),
1769 AGGREGATE(group_concat, 2, 0, 0, groupConcatStep, groupConcatFinalize),
danielk197793ce7412008-09-01 19:14:02 +00001770
1771 LIKEFUNC(glob, 2, &globInfo, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
1772 #ifdef SQLITE_CASE_SENSITIVE_LIKE
1773 LIKEFUNC(like, 2, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
1774 LIKEFUNC(like, 3, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
1775 #else
1776 LIKEFUNC(like, 2, &likeInfoNorm, SQLITE_FUNC_LIKE),
1777 LIKEFUNC(like, 3, &likeInfoNorm, SQLITE_FUNC_LIKE),
1778 #endif
1779 };
1780
drh70a8ca32008-08-21 18:49:27 +00001781 int i;
danielk197793ce7412008-09-01 19:14:02 +00001782 FuncDefHash *pHash = &GLOBAL(FuncDefHash, sqlite3GlobalFunctions);
drh106cee52008-09-03 17:11:16 +00001783 FuncDef *aFunc = (FuncDef*)&GLOBAL(FuncDef, aBuiltinFunc);
danielk197793ce7412008-09-01 19:14:02 +00001784
drh70a8ca32008-08-21 18:49:27 +00001785 for(i=0; i<ArraySize(aBuiltinFunc); i++){
danielk197793ce7412008-09-01 19:14:02 +00001786 sqlite3FuncDefInsert(pHash, &aFunc[i]);
drh70a8ca32008-08-21 18:49:27 +00001787 }
drh777c5382008-08-21 20:21:34 +00001788 sqlite3RegisterDateTimeFunctions();
drh545f5872010-04-24 14:02:59 +00001789#ifndef SQLITE_OMIT_ALTERTABLE
1790 sqlite3AlterFunctions();
1791#endif
dan0106e372013-08-12 16:34:32 +00001792#if defined(SQLITE_ENABLE_STAT3) || defined(SQLITE_ENABLE_STAT4)
1793 sqlite3AnalyzeFunctions();
1794#endif
drh70a8ca32008-08-21 18:49:27 +00001795}