blob: 5b8ed6dd2bf3c51f07adf5b35a54c639a7187693 [file] [log] [blame]
drhdc04c582002-02-24 01:55:15 +00001/*
2** 2002 February 23
3**
4** The author disclaims copyright to this source code. In place of
5** a legal notice, here is a blessing:
6**
7** May you do good and not evil.
8** May you find forgiveness for yourself and forgive others.
9** May you share freely, never taking more than you give.
10**
11*************************************************************************
peter.d.reid60ec9142014-09-06 16:39:46 +000012** This file contains the C-language implementations for many of the SQL
drhede7ae32014-08-06 11:58:40 +000013** functions of SQLite. (Some function, and in particular the date and
14** time functions, are implemented separately.)
drhdc04c582002-02-24 01:55:15 +000015*/
drhb659e9b2005-01-28 01:29:08 +000016#include "sqliteInt.h"
drhd3a149e2002-02-24 17:12:53 +000017#include <stdlib.h>
drh0bce8352002-02-28 00:41:10 +000018#include <assert.h>
danielk197788208052004-05-25 01:13:20 +000019#include "vdbeInt.h"
drh0bce8352002-02-28 00:41:10 +000020
drh55ef4d92005-08-14 01:20:37 +000021/*
22** Return the collating function associated with a function.
23*/
danielk1977dc1bdc42004-06-11 10:51:27 +000024static CollSeq *sqlite3GetFuncCollSeq(sqlite3_context *context){
drha9e03b12015-03-12 06:46:52 +000025 VdbeOp *pOp;
26 assert( context->pVdbe!=0 );
27 pOp = &context->pVdbe->aOp[context->iOp-1];
drha15cc472014-09-25 13:17:30 +000028 assert( pOp->opcode==OP_CollSeq );
29 assert( pOp->p4type==P4_COLLSEQ );
30 return pOp->p4.pColl;
danielk1977dc1bdc42004-06-11 10:51:27 +000031}
32
drh0bce8352002-02-28 00:41:10 +000033/*
drh7a957892012-02-02 17:35:43 +000034** Indicate that the accumulator load should be skipped on this
35** iteration of the aggregate loop.
36*/
37static void sqlite3SkipAccumulatorLoad(sqlite3_context *context){
38 context->skipFlag = 1;
39}
40
41/*
drh0bce8352002-02-28 00:41:10 +000042** Implementation of the non-aggregate min() and max() functions
43*/
drhf9b596e2004-05-26 16:54:42 +000044static void minmaxFunc(
45 sqlite3_context *context,
46 int argc,
47 sqlite3_value **argv
48){
drh0bce8352002-02-28 00:41:10 +000049 int i;
drh268380c2004-02-25 13:47:31 +000050 int mask; /* 0 for min() or 0xffffffff for max() */
drhf9b596e2004-05-26 16:54:42 +000051 int iBest;
danielk1977dc1bdc42004-06-11 10:51:27 +000052 CollSeq *pColl;
drh0bce8352002-02-28 00:41:10 +000053
drh65595cd2009-02-02 16:32:55 +000054 assert( argc>1 );
drhc44af712004-09-02 15:53:56 +000055 mask = sqlite3_user_data(context)==0 ? 0 : -1;
danielk1977dc1bdc42004-06-11 10:51:27 +000056 pColl = sqlite3GetFuncCollSeq(context);
57 assert( pColl );
danielk1977c572ef72004-05-27 09:28:41 +000058 assert( mask==-1 || mask==0 );
drhf9b596e2004-05-26 16:54:42 +000059 iBest = 0;
drh9c054832004-05-31 18:51:57 +000060 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drhf9b596e2004-05-26 16:54:42 +000061 for(i=1; i<argc; i++){
drh9c054832004-05-31 18:51:57 +000062 if( sqlite3_value_type(argv[i])==SQLITE_NULL ) return;
danielk1977dc1bdc42004-06-11 10:51:27 +000063 if( (sqlite3MemCompare(argv[iBest], argv[i], pColl)^mask)>=0 ){
drh65595cd2009-02-02 16:32:55 +000064 testcase( mask==0 );
drhf9b596e2004-05-26 16:54:42 +000065 iBest = i;
drh0bce8352002-02-28 00:41:10 +000066 }
67 }
drhf4479502004-05-27 03:12:53 +000068 sqlite3_result_value(context, argv[iBest]);
drh0bce8352002-02-28 00:41:10 +000069}
drh0bce8352002-02-28 00:41:10 +000070
drh268380c2004-02-25 13:47:31 +000071/*
72** Return the type of the argument.
73*/
drhf9b596e2004-05-26 16:54:42 +000074static void typeofFunc(
75 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +000076 int NotUsed,
drhf9b596e2004-05-26 16:54:42 +000077 sqlite3_value **argv
78){
danielk197735bb9d02004-05-24 12:55:54 +000079 const char *z = 0;
danielk197762c14b32008-11-19 09:05:26 +000080 UNUSED_PARAMETER(NotUsed);
danielk197735bb9d02004-05-24 12:55:54 +000081 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +000082 case SQLITE_INTEGER: z = "integer"; break;
83 case SQLITE_TEXT: z = "text"; break;
84 case SQLITE_FLOAT: z = "real"; break;
85 case SQLITE_BLOB: z = "blob"; break;
drh65595cd2009-02-02 16:32:55 +000086 default: z = "null"; break;
danielk197735bb9d02004-05-24 12:55:54 +000087 }
danielk1977d8123362004-06-12 09:25:12 +000088 sqlite3_result_text(context, z, -1, SQLITE_STATIC);
drh0bce8352002-02-28 00:41:10 +000089}
90
drh5708d2d2005-06-22 10:53:59 +000091
92/*
drh0bce8352002-02-28 00:41:10 +000093** Implementation of the length() function
94*/
drhf9b596e2004-05-26 16:54:42 +000095static void lengthFunc(
96 sqlite3_context *context,
97 int argc,
98 sqlite3_value **argv
99){
drh0bce8352002-02-28 00:41:10 +0000100 int len;
101
102 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000103 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +0000104 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +0000105 case SQLITE_BLOB:
106 case SQLITE_INTEGER:
107 case SQLITE_FLOAT: {
drhf4479502004-05-27 03:12:53 +0000108 sqlite3_result_int(context, sqlite3_value_bytes(argv[0]));
drhf9b596e2004-05-26 16:54:42 +0000109 break;
110 }
drh9c054832004-05-31 18:51:57 +0000111 case SQLITE_TEXT: {
drh2646da72005-12-09 20:02:05 +0000112 const unsigned char *z = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +0000113 if( z==0 ) return;
drh4a919112007-05-15 11:55:09 +0000114 len = 0;
115 while( *z ){
116 len++;
117 SQLITE_SKIP_UTF8(z);
118 }
drhf4479502004-05-27 03:12:53 +0000119 sqlite3_result_int(context, len);
drhf9b596e2004-05-26 16:54:42 +0000120 break;
121 }
122 default: {
123 sqlite3_result_null(context);
124 break;
125 }
126 }
drh0bce8352002-02-28 00:41:10 +0000127}
128
129/*
drh2ba3ccc2009-12-08 02:06:08 +0000130** Implementation of the abs() function.
131**
132** IMP: R-23979-26855 The abs(X) function returns the absolute value of
133** the numeric argument X.
drh0bce8352002-02-28 00:41:10 +0000134*/
danielk19770ae8b832004-05-25 12:05:56 +0000135static void absFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drh0bce8352002-02-28 00:41:10 +0000136 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000137 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +0000138 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +0000139 case SQLITE_INTEGER: {
danielk1977f93bbbe2004-05-27 10:30:52 +0000140 i64 iVal = sqlite3_value_int64(argv[0]);
drh52fc8492006-02-23 21:43:55 +0000141 if( iVal<0 ){
drh693e6712014-01-24 22:58:00 +0000142 if( iVal==SMALLEST_INT64 ){
drheb091cd2013-11-09 19:47:15 +0000143 /* IMP: R-31676-45509 If X is the integer -9223372036854775808
144 ** then abs(X) throws an integer overflow error since there is no
drh2ba3ccc2009-12-08 02:06:08 +0000145 ** equivalent positive 64-bit two complement value. */
drh52fc8492006-02-23 21:43:55 +0000146 sqlite3_result_error(context, "integer overflow", -1);
147 return;
148 }
149 iVal = -iVal;
150 }
danielk1977f93bbbe2004-05-27 10:30:52 +0000151 sqlite3_result_int64(context, iVal);
drhf9b596e2004-05-26 16:54:42 +0000152 break;
153 }
drh9c054832004-05-31 18:51:57 +0000154 case SQLITE_NULL: {
drh2ba3ccc2009-12-08 02:06:08 +0000155 /* IMP: R-37434-19929 Abs(X) returns NULL if X is NULL. */
drhf9b596e2004-05-26 16:54:42 +0000156 sqlite3_result_null(context);
157 break;
158 }
159 default: {
drh2ba3ccc2009-12-08 02:06:08 +0000160 /* Because sqlite3_value_double() returns 0.0 if the argument is not
161 ** something that can be converted into a number, we have:
drh643091f2014-11-20 23:21:23 +0000162 ** IMP: R-01992-00519 Abs(X) returns 0.0 if X is a string or blob
163 ** that cannot be converted to a numeric value.
drh2ba3ccc2009-12-08 02:06:08 +0000164 */
danielk1977f93bbbe2004-05-27 10:30:52 +0000165 double rVal = sqlite3_value_double(argv[0]);
drh52fc8492006-02-23 21:43:55 +0000166 if( rVal<0 ) rVal = -rVal;
danielk1977f93bbbe2004-05-27 10:30:52 +0000167 sqlite3_result_double(context, rVal);
drhf9b596e2004-05-26 16:54:42 +0000168 break;
169 }
170 }
drh0bce8352002-02-28 00:41:10 +0000171}
172
173/*
drhd55e0722012-10-25 03:07:29 +0000174** Implementation of the instr() function.
175**
176** instr(haystack,needle) finds the first occurrence of needle
177** in haystack and returns the number of previous characters plus 1,
178** or 0 if needle does not occur within haystack.
179**
180** If both haystack and needle are BLOBs, then the result is one more than
181** the number of bytes in haystack prior to the first occurrence of needle,
182** or 0 if needle never occurs in haystack.
183*/
184static void instrFunc(
185 sqlite3_context *context,
186 int argc,
187 sqlite3_value **argv
188){
189 const unsigned char *zHaystack;
190 const unsigned char *zNeedle;
191 int nHaystack;
192 int nNeedle;
193 int typeHaystack, typeNeedle;
194 int N = 1;
195 int isText;
196
drh68c804b2012-12-04 11:03:11 +0000197 UNUSED_PARAMETER(argc);
drhd55e0722012-10-25 03:07:29 +0000198 typeHaystack = sqlite3_value_type(argv[0]);
199 typeNeedle = sqlite3_value_type(argv[1]);
200 if( typeHaystack==SQLITE_NULL || typeNeedle==SQLITE_NULL ) return;
201 nHaystack = sqlite3_value_bytes(argv[0]);
202 nNeedle = sqlite3_value_bytes(argv[1]);
203 if( typeHaystack==SQLITE_BLOB && typeNeedle==SQLITE_BLOB ){
204 zHaystack = sqlite3_value_blob(argv[0]);
205 zNeedle = sqlite3_value_blob(argv[1]);
206 isText = 0;
207 }else{
208 zHaystack = sqlite3_value_text(argv[0]);
209 zNeedle = sqlite3_value_text(argv[1]);
210 isText = 1;
dana2bfa042016-11-04 12:05:29 +0000211 if( zNeedle==0 ) return;
212 assert( zHaystack );
drhd55e0722012-10-25 03:07:29 +0000213 }
214 while( nNeedle<=nHaystack && memcmp(zHaystack, zNeedle, nNeedle)!=0 ){
215 N++;
216 do{
217 nHaystack--;
218 zHaystack++;
219 }while( isText && (zHaystack[0]&0xc0)==0x80 );
220 }
221 if( nNeedle>nHaystack ) N = 0;
222 sqlite3_result_int(context, N);
223}
224
225/*
drha5c14162013-12-17 15:03:06 +0000226** Implementation of the printf() function.
227*/
228static void printfFunc(
229 sqlite3_context *context,
230 int argc,
231 sqlite3_value **argv
232){
233 PrintfArguments x;
234 StrAccum str;
235 const char *zFormat;
236 int n;
drhc0490572015-05-02 11:45:53 +0000237 sqlite3 *db = sqlite3_context_db_handle(context);
drha5c14162013-12-17 15:03:06 +0000238
239 if( argc>=1 && (zFormat = (const char*)sqlite3_value_text(argv[0]))!=0 ){
240 x.nArg = argc-1;
241 x.nUsed = 0;
242 x.apArg = argv+1;
drhc0490572015-05-02 11:45:53 +0000243 sqlite3StrAccumInit(&str, db, 0, 0, db->aLimit[SQLITE_LIMIT_LENGTH]);
drh5f4a6862016-01-30 12:50:25 +0000244 str.printfFlags = SQLITE_PRINTF_SQLFUNC;
245 sqlite3XPrintf(&str, zFormat, &x);
drha5c14162013-12-17 15:03:06 +0000246 n = str.nChar;
247 sqlite3_result_text(context, sqlite3StrAccumFinish(&str), n,
248 SQLITE_DYNAMIC);
249 }
250}
251
252/*
drhf764e6f2007-05-15 01:13:47 +0000253** Implementation of the substr() function.
254**
255** substr(x,p1,p2) returns p2 characters of x[] beginning with p1.
256** p1 is 1-indexed. So substr(x,1,1) returns the first character
257** of x. If x is text, then we actually count UTF-8 characters.
258** If x is a blob, then we count bytes.
259**
260** If p1 is negative, then we begin abs(p1) from the end of x[].
shaneh779b8f12009-11-12 05:04:50 +0000261**
drhf7b54962013-05-28 12:11:54 +0000262** If p2 is negative, return the p2 characters preceding p1.
drh0bce8352002-02-28 00:41:10 +0000263*/
drhf9b596e2004-05-26 16:54:42 +0000264static void substrFunc(
265 sqlite3_context *context,
266 int argc,
267 sqlite3_value **argv
268){
drh2646da72005-12-09 20:02:05 +0000269 const unsigned char *z;
270 const unsigned char *z2;
drh023ae032007-05-08 12:12:16 +0000271 int len;
drhf764e6f2007-05-15 01:13:47 +0000272 int p0type;
drh023ae032007-05-08 12:12:16 +0000273 i64 p1, p2;
drh65595cd2009-02-02 16:32:55 +0000274 int negP2 = 0;
drhf9b596e2004-05-26 16:54:42 +0000275
drh64f31512007-10-12 19:11:55 +0000276 assert( argc==3 || argc==2 );
drh8198d252009-02-01 19:42:37 +0000277 if( sqlite3_value_type(argv[1])==SQLITE_NULL
278 || (argc==3 && sqlite3_value_type(argv[2])==SQLITE_NULL)
279 ){
280 return;
281 }
drhf764e6f2007-05-15 01:13:47 +0000282 p0type = sqlite3_value_type(argv[0]);
drh4adc4cb2009-11-11 20:53:31 +0000283 p1 = sqlite3_value_int(argv[1]);
drhf764e6f2007-05-15 01:13:47 +0000284 if( p0type==SQLITE_BLOB ){
285 len = sqlite3_value_bytes(argv[0]);
286 z = sqlite3_value_blob(argv[0]);
287 if( z==0 ) return;
drh1f0feef2007-05-15 13:27:07 +0000288 assert( len==sqlite3_value_bytes(argv[0]) );
drhf764e6f2007-05-15 01:13:47 +0000289 }else{
290 z = sqlite3_value_text(argv[0]);
291 if( z==0 ) return;
drh4a919112007-05-15 11:55:09 +0000292 len = 0;
drh4adc4cb2009-11-11 20:53:31 +0000293 if( p1<0 ){
294 for(z2=z; *z2; len++){
295 SQLITE_SKIP_UTF8(z2);
296 }
drh4a919112007-05-15 11:55:09 +0000297 }
drhf764e6f2007-05-15 01:13:47 +0000298 }
drh883ad042015-02-19 00:29:11 +0000299#ifdef SQLITE_SUBSTR_COMPATIBILITY
300 /* If SUBSTR_COMPATIBILITY is defined then substr(X,0,N) work the same as
301 ** as substr(X,1,N) - it returns the first N characters of X. This
302 ** is essentially a back-out of the bug-fix in check-in [5fc125d362df4b8]
303 ** from 2009-02-02 for compatibility of applications that exploited the
304 ** old buggy behavior. */
305 if( p1==0 ) p1 = 1; /* <rdar://problem/6778339> */
306#endif
drh64f31512007-10-12 19:11:55 +0000307 if( argc==3 ){
308 p2 = sqlite3_value_int(argv[2]);
drh65595cd2009-02-02 16:32:55 +0000309 if( p2<0 ){
310 p2 = -p2;
311 negP2 = 1;
312 }
drh64f31512007-10-12 19:11:55 +0000313 }else{
drhbb4957f2008-03-20 14:03:29 +0000314 p2 = sqlite3_context_db_handle(context)->aLimit[SQLITE_LIMIT_LENGTH];
drh64f31512007-10-12 19:11:55 +0000315 }
drh0bce8352002-02-28 00:41:10 +0000316 if( p1<0 ){
drh89425d52002-02-28 03:04:48 +0000317 p1 += len;
drh653bc752002-02-28 03:31:10 +0000318 if( p1<0 ){
319 p2 += p1;
drh65595cd2009-02-02 16:32:55 +0000320 if( p2<0 ) p2 = 0;
drh653bc752002-02-28 03:31:10 +0000321 p1 = 0;
322 }
drh0bce8352002-02-28 00:41:10 +0000323 }else if( p1>0 ){
324 p1--;
drh65595cd2009-02-02 16:32:55 +0000325 }else if( p2>0 ){
326 p2--;
drh0bce8352002-02-28 00:41:10 +0000327 }
drh65595cd2009-02-02 16:32:55 +0000328 if( negP2 ){
329 p1 -= p2;
drh4e79c592009-02-01 19:23:32 +0000330 if( p1<0 ){
331 p2 += p1;
332 p1 = 0;
333 }
334 }
drh65595cd2009-02-02 16:32:55 +0000335 assert( p1>=0 && p2>=0 );
drhf764e6f2007-05-15 01:13:47 +0000336 if( p0type!=SQLITE_BLOB ){
drh4a919112007-05-15 11:55:09 +0000337 while( *z && p1 ){
338 SQLITE_SKIP_UTF8(z);
339 p1--;
drhf764e6f2007-05-15 01:13:47 +0000340 }
drh4a919112007-05-15 11:55:09 +0000341 for(z2=z; *z2 && p2; p2--){
342 SQLITE_SKIP_UTF8(z2);
drhf764e6f2007-05-15 01:13:47 +0000343 }
drhbbf483f2014-09-09 20:30:24 +0000344 sqlite3_result_text64(context, (char*)z, z2-z, SQLITE_TRANSIENT,
345 SQLITE_UTF8);
drhf764e6f2007-05-15 01:13:47 +0000346 }else{
drh4adc4cb2009-11-11 20:53:31 +0000347 if( p1+p2>len ){
348 p2 = len-p1;
349 if( p2<0 ) p2 = 0;
350 }
drhbbf483f2014-09-09 20:30:24 +0000351 sqlite3_result_blob64(context, (char*)&z[p1], (u64)p2, SQLITE_TRANSIENT);
drh0bce8352002-02-28 00:41:10 +0000352 }
drh0bce8352002-02-28 00:41:10 +0000353}
354
355/*
356** Implementation of the round() function
357*/
shanefbd60f82009-02-04 03:59:25 +0000358#ifndef SQLITE_OMIT_FLOATING_POINT
danielk19770ae8b832004-05-25 12:05:56 +0000359static void roundFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
danielk197751ad0ec2004-05-24 12:39:02 +0000360 int n = 0;
drh0bce8352002-02-28 00:41:10 +0000361 double r;
drh50d654d2009-06-03 01:24:54 +0000362 char *zBuf;
drh0bce8352002-02-28 00:41:10 +0000363 assert( argc==1 || argc==2 );
danielk197751ad0ec2004-05-24 12:39:02 +0000364 if( argc==2 ){
drh9c054832004-05-31 18:51:57 +0000365 if( SQLITE_NULL==sqlite3_value_type(argv[1]) ) return;
danielk197751ad0ec2004-05-24 12:39:02 +0000366 n = sqlite3_value_int(argv[1]);
367 if( n>30 ) n = 30;
368 if( n<0 ) n = 0;
369 }
drhd589a922006-03-02 03:02:48 +0000370 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drh4f26d6c2004-05-26 23:25:30 +0000371 r = sqlite3_value_double(argv[0]);
shaneh147e1762010-02-17 04:19:27 +0000372 /* If Y==0 and X will fit in a 64-bit int,
373 ** handle the rounding directly,
374 ** otherwise use printf.
375 */
376 if( n==0 && r>=0 && r<LARGEST_INT64-1 ){
377 r = (double)((sqlite_int64)(r+0.5));
378 }else if( n==0 && r<0 && (-r)<LARGEST_INT64-1 ){
379 r = -(double)((sqlite_int64)((-r)+0.5));
drh50d654d2009-06-03 01:24:54 +0000380 }else{
shaneh147e1762010-02-17 04:19:27 +0000381 zBuf = sqlite3_mprintf("%.*f",n,r);
382 if( zBuf==0 ){
383 sqlite3_result_error_nomem(context);
384 return;
385 }
drh9339da12010-09-30 00:50:49 +0000386 sqlite3AtoF(zBuf, &r, sqlite3Strlen30(zBuf), SQLITE_UTF8);
drh50d654d2009-06-03 01:24:54 +0000387 sqlite3_free(zBuf);
drh50d654d2009-06-03 01:24:54 +0000388 }
shaneh147e1762010-02-17 04:19:27 +0000389 sqlite3_result_double(context, r);
drh0bce8352002-02-28 00:41:10 +0000390}
shanefbd60f82009-02-04 03:59:25 +0000391#endif
drhdc04c582002-02-24 01:55:15 +0000392
danielk197726783a52007-08-29 14:06:22 +0000393/*
drhf3cdcdc2015-04-29 16:50:28 +0000394** Allocate nByte bytes of space using sqlite3Malloc(). If the
danielk197726783a52007-08-29 14:06:22 +0000395** allocation fails, call sqlite3_result_error_nomem() to notify
drh27e62db2009-04-02 10:16:17 +0000396** the database handle that malloc() has failed and return NULL.
397** If nByte is larger than the maximum string or blob length, then
398** raise an SQLITE_TOOBIG exception and return NULL.
danielk197726783a52007-08-29 14:06:22 +0000399*/
drhb1a6c3c2008-03-20 16:30:17 +0000400static void *contextMalloc(sqlite3_context *context, i64 nByte){
drhbb4957f2008-03-20 14:03:29 +0000401 char *z;
drh27e62db2009-04-02 10:16:17 +0000402 sqlite3 *db = sqlite3_context_db_handle(context);
drhef31c6a2009-04-02 09:07:12 +0000403 assert( nByte>0 );
drh27e62db2009-04-02 10:16:17 +0000404 testcase( nByte==db->aLimit[SQLITE_LIMIT_LENGTH] );
405 testcase( nByte==db->aLimit[SQLITE_LIMIT_LENGTH]+1 );
406 if( nByte>db->aLimit[SQLITE_LIMIT_LENGTH] ){
drhbb4957f2008-03-20 14:03:29 +0000407 sqlite3_result_error_toobig(context);
408 z = 0;
409 }else{
drhda4ca9d2014-09-09 17:27:35 +0000410 z = sqlite3Malloc(nByte);
drhef31c6a2009-04-02 09:07:12 +0000411 if( !z ){
drhbb4957f2008-03-20 14:03:29 +0000412 sqlite3_result_error_nomem(context);
413 }
danielk1977a1644fd2007-08-29 12:31:25 +0000414 }
415 return z;
416}
417
drhdc04c582002-02-24 01:55:15 +0000418/*
419** Implementation of the upper() and lower() SQL functions.
420*/
danielk19770ae8b832004-05-25 12:05:56 +0000421static void upperFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drh7a521cf2007-04-25 18:23:52 +0000422 char *z1;
423 const char *z2;
drh9310ef22007-04-27 17:16:20 +0000424 int i, n;
drh1d34fde2009-02-03 15:50:33 +0000425 UNUSED_PARAMETER(argc);
drh7a521cf2007-04-25 18:23:52 +0000426 z2 = (char*)sqlite3_value_text(argv[0]);
drh1f0feef2007-05-15 13:27:07 +0000427 n = sqlite3_value_bytes(argv[0]);
428 /* Verify that the call to _bytes() does not invalidate the _text() pointer */
429 assert( z2==(char*)sqlite3_value_text(argv[0]) );
drh7a521cf2007-04-25 18:23:52 +0000430 if( z2 ){
drhb1a6c3c2008-03-20 16:30:17 +0000431 z1 = contextMalloc(context, ((i64)n)+1);
drh7a521cf2007-04-25 18:23:52 +0000432 if( z1 ){
drhdf901d32011-10-13 18:00:11 +0000433 for(i=0; i<n; i++){
434 z1[i] = (char)sqlite3Toupper(z2[i]);
drh7a521cf2007-04-25 18:23:52 +0000435 }
drhdf901d32011-10-13 18:00:11 +0000436 sqlite3_result_text(context, z1, n, sqlite3_free);
drh7a521cf2007-04-25 18:23:52 +0000437 }
drhdc04c582002-02-24 01:55:15 +0000438 }
439}
danielk19770ae8b832004-05-25 12:05:56 +0000440static void lowerFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drhdf901d32011-10-13 18:00:11 +0000441 char *z1;
drh7a521cf2007-04-25 18:23:52 +0000442 const char *z2;
drh9310ef22007-04-27 17:16:20 +0000443 int i, n;
drh1d34fde2009-02-03 15:50:33 +0000444 UNUSED_PARAMETER(argc);
drh7a521cf2007-04-25 18:23:52 +0000445 z2 = (char*)sqlite3_value_text(argv[0]);
drh1f0feef2007-05-15 13:27:07 +0000446 n = sqlite3_value_bytes(argv[0]);
447 /* Verify that the call to _bytes() does not invalidate the _text() pointer */
448 assert( z2==(char*)sqlite3_value_text(argv[0]) );
drh7a521cf2007-04-25 18:23:52 +0000449 if( z2 ){
drhb1a6c3c2008-03-20 16:30:17 +0000450 z1 = contextMalloc(context, ((i64)n)+1);
drh7a521cf2007-04-25 18:23:52 +0000451 if( z1 ){
drhdf901d32011-10-13 18:00:11 +0000452 for(i=0; i<n; i++){
453 z1[i] = sqlite3Tolower(z2[i]);
drh7a521cf2007-04-25 18:23:52 +0000454 }
drhdf901d32011-10-13 18:00:11 +0000455 sqlite3_result_text(context, z1, n, sqlite3_free);
drh7a521cf2007-04-25 18:23:52 +0000456 }
drhdc04c582002-02-24 01:55:15 +0000457 }
458}
459
drhae6bb952009-11-11 00:24:31 +0000460/*
drhcca9f3d2013-09-06 15:23:29 +0000461** Some functions like COALESCE() and IFNULL() and UNLIKELY() are implemented
462** as VDBE code so that unused argument values do not have to be computed.
463** However, we still need some kind of function implementation for this
464** routines in the function table. The noopFunc macro provides this.
465** noopFunc will never be called so it doesn't matter what the implementation
466** is. We might as well use the "version()" function as a substitute.
drhae6bb952009-11-11 00:24:31 +0000467*/
drhcca9f3d2013-09-06 15:23:29 +0000468#define noopFunc versionFunc /* Substitute function - never called */
drh3212e182002-02-28 00:46:26 +0000469
470/*
drhf9ffac92002-03-02 19:00:31 +0000471** Implementation of random(). Return a random integer.
472*/
drhf9b596e2004-05-26 16:54:42 +0000473static void randomFunc(
474 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000475 int NotUsed,
476 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000477){
drh52fc8492006-02-23 21:43:55 +0000478 sqlite_int64 r;
danielk197762c14b32008-11-19 09:05:26 +0000479 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drh2fa18682008-03-19 14:15:34 +0000480 sqlite3_randomness(sizeof(r), &r);
drh3034e3d2009-04-02 14:05:21 +0000481 if( r<0 ){
482 /* We need to prevent a random number of 0x8000000000000000
483 ** (or -9223372036854775808) since when you do abs() of that
484 ** number of you get the same value back again. To do this
485 ** in a way that is testable, mask the sign bit off of negative
486 ** values, resulting in a positive value. Then take the
487 ** 2s complement of that positive value. The end result can
488 ** therefore be no less than -9223372036854775807.
489 */
drhaf8001b2012-02-11 19:53:24 +0000490 r = -(r & LARGEST_INT64);
drh3034e3d2009-04-02 14:05:21 +0000491 }
drh52fc8492006-02-23 21:43:55 +0000492 sqlite3_result_int64(context, r);
drhf9ffac92002-03-02 19:00:31 +0000493}
494
495/*
drh137c7282007-01-29 17:58:28 +0000496** Implementation of randomblob(N). Return a random blob
497** that is N bytes long.
drh63cf66f2007-01-29 15:50:05 +0000498*/
drh137c7282007-01-29 17:58:28 +0000499static void randomBlob(
drh63cf66f2007-01-29 15:50:05 +0000500 sqlite3_context *context,
501 int argc,
502 sqlite3_value **argv
503){
drh137c7282007-01-29 17:58:28 +0000504 int n;
505 unsigned char *p;
drh63cf66f2007-01-29 15:50:05 +0000506 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000507 UNUSED_PARAMETER(argc);
drh63cf66f2007-01-29 15:50:05 +0000508 n = sqlite3_value_int(argv[0]);
drh023ae032007-05-08 12:12:16 +0000509 if( n<1 ){
510 n = 1;
511 }
danielk1977a1644fd2007-08-29 12:31:25 +0000512 p = contextMalloc(context, n);
drh02d85832007-05-07 19:31:15 +0000513 if( p ){
drh2fa18682008-03-19 14:15:34 +0000514 sqlite3_randomness(n, p);
drh17435752007-08-16 04:30:38 +0000515 sqlite3_result_blob(context, (char*)p, n, sqlite3_free);
drh02d85832007-05-07 19:31:15 +0000516 }
drh63cf66f2007-01-29 15:50:05 +0000517}
518
519/*
drh6ed41ad2002-04-06 14:10:47 +0000520** Implementation of the last_insert_rowid() SQL function. The return
danielk197724b03fd2004-05-10 10:34:34 +0000521** value is the same as the sqlite3_last_insert_rowid() API function.
drh6ed41ad2002-04-06 14:10:47 +0000522*/
danielk197751ad0ec2004-05-24 12:39:02 +0000523static void last_insert_rowid(
danielk19770ae8b832004-05-25 12:05:56 +0000524 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000525 int NotUsed,
526 sqlite3_value **NotUsed2
danielk197751ad0ec2004-05-24 12:39:02 +0000527){
drhfa4a4b92008-03-19 21:45:51 +0000528 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000529 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000530 /* IMP: R-51513-12026 The last_insert_rowid() SQL function is a
531 ** wrapper around the sqlite3_last_insert_rowid() C/C++ interface
532 ** function. */
drhf9b596e2004-05-26 16:54:42 +0000533 sqlite3_result_int64(context, sqlite3_last_insert_rowid(db));
drh6ed41ad2002-04-06 14:10:47 +0000534}
535
rdcf146a772004-02-25 22:51:06 +0000536/*
drhab2f1f92010-01-11 18:26:42 +0000537** Implementation of the changes() SQL function.
538**
539** IMP: R-62073-11209 The changes() SQL function is a wrapper
540** around the sqlite3_changes() C/C++ function and hence follows the same
541** rules for counting changes.
rdcf146a772004-02-25 22:51:06 +0000542*/
danielk1977b28af712004-06-21 06:50:26 +0000543static void changes(
drhf9b596e2004-05-26 16:54:42 +0000544 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000545 int NotUsed,
546 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000547){
drhfa4a4b92008-03-19 21:45:51 +0000548 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000549 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhf4479502004-05-27 03:12:53 +0000550 sqlite3_result_int(context, sqlite3_changes(db));
rdcb0c374f2004-02-20 22:53:38 +0000551}
rdcf146a772004-02-25 22:51:06 +0000552
553/*
danielk1977b28af712004-06-21 06:50:26 +0000554** Implementation of the total_changes() SQL function. The return value is
555** the same as the sqlite3_total_changes() API function.
rdcf146a772004-02-25 22:51:06 +0000556*/
danielk1977b28af712004-06-21 06:50:26 +0000557static void total_changes(
558 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000559 int NotUsed,
560 sqlite3_value **NotUsed2
danielk197751ad0ec2004-05-24 12:39:02 +0000561){
drhfa4a4b92008-03-19 21:45:51 +0000562 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000563 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000564 /* IMP: R-52756-41993 This function is a wrapper around the
565 ** sqlite3_total_changes() C/C++ interface. */
danielk1977b28af712004-06-21 06:50:26 +0000566 sqlite3_result_int(context, sqlite3_total_changes(db));
rdcb0c374f2004-02-20 22:53:38 +0000567}
568
drh6ed41ad2002-04-06 14:10:47 +0000569/*
drh4e5ffc52004-08-31 00:52:37 +0000570** A structure defining how to do GLOB-style comparisons.
danielk1977d02eb1f2004-06-06 09:44:03 +0000571*/
drh4e5ffc52004-08-31 00:52:37 +0000572struct compareInfo {
drh07e83472016-01-11 03:48:18 +0000573 u8 matchAll; /* "*" or "%" */
574 u8 matchOne; /* "?" or "_" */
575 u8 matchSet; /* "[" or 0 */
576 u8 noCase; /* true to ignore case differences */
danielk1977d02eb1f2004-06-06 09:44:03 +0000577};
drh55ef4d92005-08-14 01:20:37 +0000578
drhb9175ae2007-12-07 18:39:04 +0000579/*
580** For LIKE and GLOB matching on EBCDIC machines, assume that every
drhb0870482015-06-17 13:20:54 +0000581** character is exactly one byte in size. Also, provde the Utf8Read()
582** macro for fast reading of the next character in the common case where
583** the next character is ASCII.
drhb9175ae2007-12-07 18:39:04 +0000584*/
585#if defined(SQLITE_EBCDIC)
drh88b33222014-09-25 03:51:37 +0000586# define sqlite3Utf8Read(A) (*((*A)++))
drhb0870482015-06-17 13:20:54 +0000587# define Utf8Read(A) (*(A++))
drhb9175ae2007-12-07 18:39:04 +0000588#else
drhb0870482015-06-17 13:20:54 +0000589# define Utf8Read(A) (A[0]<0x80?*(A++):sqlite3Utf8Read(&A))
drhb9175ae2007-12-07 18:39:04 +0000590#endif
591
drh4e5ffc52004-08-31 00:52:37 +0000592static const struct compareInfo globInfo = { '*', '?', '[', 0 };
drh55ef4d92005-08-14 01:20:37 +0000593/* The correct SQL-92 behavior is for the LIKE operator to ignore
594** case. Thus 'a' LIKE 'A' would be true. */
595static const struct compareInfo likeInfoNorm = { '%', '_', 0, 1 };
596/* If SQLITE_CASE_SENSITIVE_LIKE is defined, then the LIKE operator
597** is case sensitive causing 'a' LIKE 'A' to be false */
598static const struct compareInfo likeInfoAlt = { '%', '_', 0, 0 };
danielk1977d02eb1f2004-06-06 09:44:03 +0000599
600/*
drh698a01c2016-12-01 18:49:40 +0000601** Possible error returns from patternMatch()
602*/
603#define SQLITE_MATCH 0
604#define SQLITE_NOMATCH 1
605#define SQLITE_NOWILDCARDMATCH 2
606
607/*
608** Compare two UTF-8 strings for equality where the first string is
609** a GLOB or LIKE expression. Return values:
610**
611** SQLITE_MATCH: Match
612** SQLITE_NOMATCH: No match
613** SQLITE_NOWILDCARDMATCH: No match in spite of having * or % wildcards.
drh0ac65892002-04-20 14:24:41 +0000614**
drh4e5ffc52004-08-31 00:52:37 +0000615** Globbing rules:
drh0ac65892002-04-20 14:24:41 +0000616**
drh4e5ffc52004-08-31 00:52:37 +0000617** '*' Matches any sequence of zero or more characters.
danielk1977d02eb1f2004-06-06 09:44:03 +0000618**
drh4e5ffc52004-08-31 00:52:37 +0000619** '?' Matches exactly one character.
620**
621** [...] Matches one character from the enclosed list of
622** characters.
623**
624** [^...] Matches one character not in the enclosed list.
625**
626** With the [...] and [^...] matching, a ']' character can be included
627** in the list by making it the first character after '[' or '^'. A
628** range of characters can be specified using '-'. Example:
629** "[a-z]" matches any single lower-case letter. To match a '-', make
630** it the last character in the list.
631**
drh9fdfdc82014-09-25 11:08:57 +0000632** Like matching rules:
633**
634** '%' Matches any sequence of zero or more characters
635**
636*** '_' Matches any one character
637**
638** Ec Where E is the "esc" character and c is any other
639** character, including '%', '_', and esc, match exactly c.
640**
drhb0870482015-06-17 13:20:54 +0000641** The comments within this routine usually assume glob matching.
drh9fdfdc82014-09-25 11:08:57 +0000642**
drh4e5ffc52004-08-31 00:52:37 +0000643** This routine is usually quick, but can be N**2 in the worst case.
drh0ac65892002-04-20 14:24:41 +0000644*/
danielk19777c6303c2004-11-17 16:41:29 +0000645static int patternCompare(
drh4e5ffc52004-08-31 00:52:37 +0000646 const u8 *zPattern, /* The glob pattern */
647 const u8 *zString, /* The string to compare against the glob */
danielk19777c6303c2004-11-17 16:41:29 +0000648 const struct compareInfo *pInfo, /* Information about how to do the compare */
drh698a01c2016-12-01 18:49:40 +0000649 u32 matchOther /* The escape char (LIKE) or '[' (GLOB) */
danielk197751ad0ec2004-05-24 12:39:02 +0000650){
drh9fdfdc82014-09-25 11:08:57 +0000651 u32 c, c2; /* Next pattern and input string chars */
652 u32 matchOne = pInfo->matchOne; /* "?" or "_" */
653 u32 matchAll = pInfo->matchAll; /* "*" or "%" */
drh9fdfdc82014-09-25 11:08:57 +0000654 u8 noCase = pInfo->noCase; /* True if uppercase==lowercase */
655 const u8 *zEscaped = 0; /* One past the last escaped input char */
drh88b33222014-09-25 03:51:37 +0000656
drhb0870482015-06-17 13:20:54 +0000657 while( (c = Utf8Read(zPattern))!=0 ){
drh9fdfdc82014-09-25 11:08:57 +0000658 if( c==matchAll ){ /* Match "*" */
659 /* Skip over multiple "*" characters in the pattern. If there
660 ** are also "?" characters, skip those as well, but consume a
661 ** single character of the input string for each "?" skipped */
drhb0870482015-06-17 13:20:54 +0000662 while( (c=Utf8Read(zPattern)) == matchAll || c == matchOne ){
drh42610962012-09-17 18:56:32 +0000663 if( c==matchOne && sqlite3Utf8Read(&zString)==0 ){
drh698a01c2016-12-01 18:49:40 +0000664 return SQLITE_NOWILDCARDMATCH;
drh4e5ffc52004-08-31 00:52:37 +0000665 }
danielk1977ad7dd422004-06-06 12:41:49 +0000666 }
drh66150952007-07-23 19:12:41 +0000667 if( c==0 ){
drh698a01c2016-12-01 18:49:40 +0000668 return SQLITE_MATCH; /* "*" at the end of the pattern matches */
drh88b33222014-09-25 03:51:37 +0000669 }else if( c==matchOther ){
drh07e83472016-01-11 03:48:18 +0000670 if( pInfo->matchSet==0 ){
drh88b33222014-09-25 03:51:37 +0000671 c = sqlite3Utf8Read(&zPattern);
drh698a01c2016-12-01 18:49:40 +0000672 if( c==0 ) return SQLITE_NOWILDCARDMATCH;
drh88b33222014-09-25 03:51:37 +0000673 }else{
drh9fdfdc82014-09-25 11:08:57 +0000674 /* "[...]" immediately follows the "*". We have to do a slow
675 ** recursive search in this case, but it is an unusual case. */
drh88b33222014-09-25 03:51:37 +0000676 assert( matchOther<0x80 ); /* '[' is a single-byte character */
drh698a01c2016-12-01 18:49:40 +0000677 while( *zString ){
678 int bMatch = patternCompare(&zPattern[-1],zString,pInfo,matchOther);
679 if( bMatch!=SQLITE_NOMATCH ) return bMatch;
drh88b33222014-09-25 03:51:37 +0000680 SQLITE_SKIP_UTF8(zString);
681 }
drh698a01c2016-12-01 18:49:40 +0000682 return SQLITE_NOWILDCARDMATCH;
drh66150952007-07-23 19:12:41 +0000683 }
drh66150952007-07-23 19:12:41 +0000684 }
drh9fdfdc82014-09-25 11:08:57 +0000685
686 /* At this point variable c contains the first character of the
687 ** pattern string past the "*". Search in the input string for the
dan1a4a7372016-12-01 17:34:59 +0000688 ** first matching character and recursively continue the match from
drh9fdfdc82014-09-25 11:08:57 +0000689 ** that point.
690 **
691 ** For a case-insensitive search, set variable cx to be the same as
692 ** c but in the other case and search the input string for either
693 ** c or cx.
694 */
695 if( c<=0x80 ){
696 u32 cx;
drh698a01c2016-12-01 18:49:40 +0000697 int bMatch;
drh9fdfdc82014-09-25 11:08:57 +0000698 if( noCase ){
699 cx = sqlite3Toupper(c);
700 c = sqlite3Tolower(c);
drh66150952007-07-23 19:12:41 +0000701 }else{
drh9fdfdc82014-09-25 11:08:57 +0000702 cx = c;
drh4e5ffc52004-08-31 00:52:37 +0000703 }
drh9fdfdc82014-09-25 11:08:57 +0000704 while( (c2 = *(zString++))!=0 ){
705 if( c2!=c && c2!=cx ) continue;
drh698a01c2016-12-01 18:49:40 +0000706 bMatch = patternCompare(zPattern,zString,pInfo,matchOther);
707 if( bMatch!=SQLITE_NOMATCH ) return bMatch;
drh9fdfdc82014-09-25 11:08:57 +0000708 }
709 }else{
drh698a01c2016-12-01 18:49:40 +0000710 int bMatch;
drhb0870482015-06-17 13:20:54 +0000711 while( (c2 = Utf8Read(zString))!=0 ){
drh9fdfdc82014-09-25 11:08:57 +0000712 if( c2!=c ) continue;
drh698a01c2016-12-01 18:49:40 +0000713 bMatch = patternCompare(zPattern,zString,pInfo,matchOther);
714 if( bMatch!=SQLITE_NOMATCH ) return bMatch;
drh9fdfdc82014-09-25 11:08:57 +0000715 }
drh66150952007-07-23 19:12:41 +0000716 }
drh698a01c2016-12-01 18:49:40 +0000717 return SQLITE_NOWILDCARDMATCH;
drh88b33222014-09-25 03:51:37 +0000718 }
drh88b33222014-09-25 03:51:37 +0000719 if( c==matchOther ){
drh07e83472016-01-11 03:48:18 +0000720 if( pInfo->matchSet==0 ){
drh88b33222014-09-25 03:51:37 +0000721 c = sqlite3Utf8Read(&zPattern);
drh698a01c2016-12-01 18:49:40 +0000722 if( c==0 ) return SQLITE_NOMATCH;
drh9fdfdc82014-09-25 11:08:57 +0000723 zEscaped = zPattern;
drh88b33222014-09-25 03:51:37 +0000724 }else{
725 u32 prior_c = 0;
drh9fdfdc82014-09-25 11:08:57 +0000726 int seen = 0;
727 int invert = 0;
drh88b33222014-09-25 03:51:37 +0000728 c = sqlite3Utf8Read(&zString);
drh698a01c2016-12-01 18:49:40 +0000729 if( c==0 ) return SQLITE_NOMATCH;
drh88b33222014-09-25 03:51:37 +0000730 c2 = sqlite3Utf8Read(&zPattern);
731 if( c2=='^' ){
732 invert = 1;
733 c2 = sqlite3Utf8Read(&zPattern);
734 }
735 if( c2==']' ){
736 if( c==']' ) seen = 1;
737 c2 = sqlite3Utf8Read(&zPattern);
738 }
739 while( c2 && c2!=']' ){
740 if( c2=='-' && zPattern[0]!=']' && zPattern[0]!=0 && prior_c>0 ){
741 c2 = sqlite3Utf8Read(&zPattern);
742 if( c>=prior_c && c<=c2 ) seen = 1;
743 prior_c = 0;
744 }else{
745 if( c==c2 ){
746 seen = 1;
747 }
748 prior_c = c2;
749 }
750 c2 = sqlite3Utf8Read(&zPattern);
751 }
752 if( c2==0 || (seen ^ invert)==0 ){
drh698a01c2016-12-01 18:49:40 +0000753 return SQLITE_NOMATCH;
drh88b33222014-09-25 03:51:37 +0000754 }
755 continue;
756 }
757 }
drhb0870482015-06-17 13:20:54 +0000758 c2 = Utf8Read(zString);
drh88b33222014-09-25 03:51:37 +0000759 if( c==c2 ) continue;
drhc80937a2016-06-06 01:48:14 +0000760 if( noCase && sqlite3Tolower(c)==sqlite3Tolower(c2) && c<0x80 && c2<0x80 ){
drh9fdfdc82014-09-25 11:08:57 +0000761 continue;
762 }
763 if( c==matchOne && zPattern!=zEscaped && c2!=0 ) continue;
drh698a01c2016-12-01 18:49:40 +0000764 return SQLITE_NOMATCH;
danielk197751ad0ec2004-05-24 12:39:02 +0000765 }
drh698a01c2016-12-01 18:49:40 +0000766 return *zString==0 ? SQLITE_MATCH : SQLITE_NOMATCH;
drh0ac65892002-04-20 14:24:41 +0000767}
drh4e5ffc52004-08-31 00:52:37 +0000768
drh55ef4d92005-08-14 01:20:37 +0000769/*
drh698a01c2016-12-01 18:49:40 +0000770** The sqlite3_strglob() interface. Return 0 on a match (like strcmp()) and
771** non-zero if there is no match.
drh56282a52013-04-10 16:13:38 +0000772*/
773int sqlite3_strglob(const char *zGlobPattern, const char *zString){
drh698a01c2016-12-01 18:49:40 +0000774 return patternCompare((u8*)zGlobPattern, (u8*)zString, &globInfo, '[');
drh56282a52013-04-10 16:13:38 +0000775}
776
777/*
drh698a01c2016-12-01 18:49:40 +0000778** The sqlite3_strlike() interface. Return 0 on a match and non-zero for
779** a miss - like strcmp().
drh8b4a94a2015-11-24 21:23:59 +0000780*/
781int sqlite3_strlike(const char *zPattern, const char *zStr, unsigned int esc){
drh698a01c2016-12-01 18:49:40 +0000782 return patternCompare((u8*)zPattern, (u8*)zStr, &likeInfoNorm, esc);
drh8b4a94a2015-11-24 21:23:59 +0000783}
784
785/*
drh55ef4d92005-08-14 01:20:37 +0000786** Count the number of times that the LIKE operator (or GLOB which is
787** just a variation of LIKE) gets called. This is used for testing
788** only.
789*/
790#ifdef SQLITE_TEST
791int sqlite3_like_count = 0;
792#endif
793
danielk19773f6b0872004-06-17 05:36:44 +0000794
795/*
796** Implementation of the like() SQL function. This function implements
797** the build-in LIKE operator. The first argument to the function is the
798** pattern and the second argument is the string. So, the SQL statements:
799**
800** A LIKE B
801**
802** is implemented as like(B,A).
803**
drh55ef4d92005-08-14 01:20:37 +0000804** This same function (with a different compareInfo structure) computes
805** the GLOB operator.
danielk19773f6b0872004-06-17 05:36:44 +0000806*/
807static void likeFunc(
808 sqlite3_context *context,
809 int argc,
810 sqlite3_value **argv
811){
drhbeb818d2007-05-08 15:34:47 +0000812 const unsigned char *zA, *zB;
drh07e83472016-01-11 03:48:18 +0000813 u32 escape;
drh27e62db2009-04-02 10:16:17 +0000814 int nPat;
drhbb4957f2008-03-20 14:03:29 +0000815 sqlite3 *db = sqlite3_context_db_handle(context);
drh07e83472016-01-11 03:48:18 +0000816 struct compareInfo *pInfo = sqlite3_user_data(context);
drhbeb818d2007-05-08 15:34:47 +0000817
drh41d2e662015-12-01 21:23:07 +0000818#ifdef SQLITE_LIKE_DOESNT_MATCH_BLOBS
819 if( sqlite3_value_type(argv[0])==SQLITE_BLOB
820 || sqlite3_value_type(argv[1])==SQLITE_BLOB
821 ){
822#ifdef SQLITE_TEST
823 sqlite3_like_count++;
824#endif
825 sqlite3_result_int(context, 0);
826 return;
827 }
828#endif
drh1f0feef2007-05-15 13:27:07 +0000829 zB = sqlite3_value_text(argv[0]);
830 zA = sqlite3_value_text(argv[1]);
831
drhbeb818d2007-05-08 15:34:47 +0000832 /* Limit the length of the LIKE or GLOB pattern to avoid problems
833 ** of deep recursion and N*N behavior in patternCompare().
834 */
drh27e62db2009-04-02 10:16:17 +0000835 nPat = sqlite3_value_bytes(argv[0]);
836 testcase( nPat==db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH] );
837 testcase( nPat==db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH]+1 );
838 if( nPat > db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH] ){
drhbeb818d2007-05-08 15:34:47 +0000839 sqlite3_result_error(context, "LIKE or GLOB pattern too complex", -1);
840 return;
841 }
drh1f0feef2007-05-15 13:27:07 +0000842 assert( zB==sqlite3_value_text(argv[0]) ); /* Encoding did not change */
drhbeb818d2007-05-08 15:34:47 +0000843
danielk19777c6303c2004-11-17 16:41:29 +0000844 if( argc==3 ){
845 /* The escape character string must consist of a single UTF-8 character.
846 ** Otherwise, return an error.
847 */
848 const unsigned char *zEsc = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +0000849 if( zEsc==0 ) return;
drhee858132007-05-08 20:37:38 +0000850 if( sqlite3Utf8CharLen((char*)zEsc, -1)!=1 ){
danielk19777c6303c2004-11-17 16:41:29 +0000851 sqlite3_result_error(context,
852 "ESCAPE expression must be a single character", -1);
853 return;
854 }
drh42610962012-09-17 18:56:32 +0000855 escape = sqlite3Utf8Read(&zEsc);
drh07e83472016-01-11 03:48:18 +0000856 }else{
857 escape = pInfo->matchSet;
danielk19777c6303c2004-11-17 16:41:29 +0000858 }
danielk19773f6b0872004-06-17 05:36:44 +0000859 if( zA && zB ){
drh55ef4d92005-08-14 01:20:37 +0000860#ifdef SQLITE_TEST
861 sqlite3_like_count++;
862#endif
drh698a01c2016-12-01 18:49:40 +0000863 sqlite3_result_int(context, patternCompare(zB, zA, pInfo, escape)==SQLITE_MATCH);
danielk197751ad0ec2004-05-24 12:39:02 +0000864 }
drh8912d102002-05-26 21:34:58 +0000865}
866
867/*
868** Implementation of the NULLIF(x,y) function. The result is the first
869** argument if the arguments are different. The result is NULL if the
870** arguments are equal to each other.
871*/
drhf9b596e2004-05-26 16:54:42 +0000872static void nullifFunc(
873 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000874 int NotUsed,
drhf9b596e2004-05-26 16:54:42 +0000875 sqlite3_value **argv
876){
danielk1977dc1bdc42004-06-11 10:51:27 +0000877 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk197762c14b32008-11-19 09:05:26 +0000878 UNUSED_PARAMETER(NotUsed);
danielk1977dc1bdc42004-06-11 10:51:27 +0000879 if( sqlite3MemCompare(argv[0], argv[1], pColl)!=0 ){
drhf4479502004-05-27 03:12:53 +0000880 sqlite3_result_value(context, argv[0]);
drh8912d102002-05-26 21:34:58 +0000881 }
drh0ac65892002-04-20 14:24:41 +0000882}
883
drh647cb0e2002-11-04 19:32:25 +0000884/*
drh47baebc2009-08-14 16:01:24 +0000885** Implementation of the sqlite_version() function. The result is the version
drh647cb0e2002-11-04 19:32:25 +0000886** of the SQLite library that is running.
887*/
drhf9b596e2004-05-26 16:54:42 +0000888static void versionFunc(
889 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000890 int NotUsed,
891 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000892){
danielk197762c14b32008-11-19 09:05:26 +0000893 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000894 /* IMP: R-48699-48617 This function is an SQL wrapper around the
895 ** sqlite3_libversion() C-interface. */
896 sqlite3_result_text(context, sqlite3_libversion(), -1, SQLITE_STATIC);
drh647cb0e2002-11-04 19:32:25 +0000897}
898
drh47baebc2009-08-14 16:01:24 +0000899/*
900** Implementation of the sqlite_source_id() function. The result is a string
901** that identifies the particular version of the source code used to build
902** SQLite.
903*/
904static void sourceidFunc(
905 sqlite3_context *context,
906 int NotUsed,
907 sqlite3_value **NotUsed2
908){
909 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000910 /* IMP: R-24470-31136 This function is an SQL wrapper around the
911 ** sqlite3_sourceid() C interface. */
912 sqlite3_result_text(context, sqlite3_sourceid(), -1, SQLITE_STATIC);
drh47baebc2009-08-14 16:01:24 +0000913}
914
shanehbdea6d12010-02-23 04:19:54 +0000915/*
drh3ca84ef2011-04-25 18:03:10 +0000916** Implementation of the sqlite_log() function. This is a wrapper around
917** sqlite3_log(). The return value is NULL. The function exists purely for
918** its side-effects.
919*/
drh840561f2011-04-27 18:08:42 +0000920static void errlogFunc(
drh3ca84ef2011-04-25 18:03:10 +0000921 sqlite3_context *context,
922 int argc,
923 sqlite3_value **argv
924){
925 UNUSED_PARAMETER(argc);
926 UNUSED_PARAMETER(context);
927 sqlite3_log(sqlite3_value_int(argv[0]), "%s", sqlite3_value_text(argv[1]));
928}
929
930/*
shanehdc97a8c2010-02-23 20:08:35 +0000931** Implementation of the sqlite_compileoption_used() function.
932** The result is an integer that identifies if the compiler option
933** was used to build SQLite.
shanehbdea6d12010-02-23 04:19:54 +0000934*/
shanehdc97a8c2010-02-23 20:08:35 +0000935#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
936static void compileoptionusedFunc(
shanehbdea6d12010-02-23 04:19:54 +0000937 sqlite3_context *context,
shanehdc97a8c2010-02-23 20:08:35 +0000938 int argc,
939 sqlite3_value **argv
shanehbdea6d12010-02-23 04:19:54 +0000940){
shanehdc97a8c2010-02-23 20:08:35 +0000941 const char *zOptName;
942 assert( argc==1 );
943 UNUSED_PARAMETER(argc);
drha3e414c2010-08-03 18:06:25 +0000944 /* IMP: R-39564-36305 The sqlite_compileoption_used() SQL
945 ** function is a wrapper around the sqlite3_compileoption_used() C/C++
946 ** function.
947 */
drh264a2d42010-02-25 15:28:41 +0000948 if( (zOptName = (const char*)sqlite3_value_text(argv[0]))!=0 ){
shanehdc97a8c2010-02-23 20:08:35 +0000949 sqlite3_result_int(context, sqlite3_compileoption_used(zOptName));
shanehdc97a8c2010-02-23 20:08:35 +0000950 }
shanehbdea6d12010-02-23 04:19:54 +0000951}
shanehdc97a8c2010-02-23 20:08:35 +0000952#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
953
954/*
955** Implementation of the sqlite_compileoption_get() function.
956** The result is a string that identifies the compiler options
957** used to build SQLite.
958*/
959#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
960static void compileoptiongetFunc(
961 sqlite3_context *context,
962 int argc,
963 sqlite3_value **argv
964){
965 int n;
966 assert( argc==1 );
967 UNUSED_PARAMETER(argc);
drha3e414c2010-08-03 18:06:25 +0000968 /* IMP: R-04922-24076 The sqlite_compileoption_get() SQL function
969 ** is a wrapper around the sqlite3_compileoption_get() C/C++ function.
970 */
shanehdc97a8c2010-02-23 20:08:35 +0000971 n = sqlite3_value_int(argv[0]);
972 sqlite3_result_text(context, sqlite3_compileoption_get(n), -1, SQLITE_STATIC);
973}
974#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
shanehbdea6d12010-02-23 04:19:54 +0000975
drh137c7282007-01-29 17:58:28 +0000976/* Array for converting from half-bytes (nybbles) into ASCII hex
977** digits. */
978static const char hexdigits[] = {
979 '0', '1', '2', '3', '4', '5', '6', '7',
980 '8', '9', 'A', 'B', 'C', 'D', 'E', 'F'
981};
danielk1977d641d642004-11-18 15:44:29 +0000982
drh47394702003-08-20 01:03:33 +0000983/*
drh47394702003-08-20 01:03:33 +0000984** Implementation of the QUOTE() function. This function takes a single
985** argument. If the argument is numeric, the return value is the same as
986** the argument. If the argument is NULL, the return value is the string
987** "NULL". Otherwise, the argument is enclosed in single quotes with
988** single-quote escapes.
989*/
danielk19770ae8b832004-05-25 12:05:56 +0000990static void quoteFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drha0df4cc2009-02-02 17:29:59 +0000991 assert( argc==1 );
drh1d34fde2009-02-03 15:50:33 +0000992 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +0000993 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +0000994 case SQLITE_FLOAT: {
drh72b3fbc2012-06-19 03:11:25 +0000995 double r1, r2;
drh72b3fbc2012-06-19 03:11:25 +0000996 char zBuf[50];
mistachkin2b434a72012-06-19 04:36:48 +0000997 r1 = sqlite3_value_double(argv[0]);
drh72b3fbc2012-06-19 03:11:25 +0000998 sqlite3_snprintf(sizeof(zBuf), zBuf, "%!.15g", r1);
999 sqlite3AtoF(zBuf, &r2, 20, SQLITE_UTF8);
1000 if( r1!=r2 ){
1001 sqlite3_snprintf(sizeof(zBuf), zBuf, "%!.20e", r1);
1002 }
1003 sqlite3_result_text(context, zBuf, -1, SQLITE_TRANSIENT);
1004 break;
1005 }
1006 case SQLITE_INTEGER: {
drhf4479502004-05-27 03:12:53 +00001007 sqlite3_result_value(context, argv[0]);
drhf9b596e2004-05-26 16:54:42 +00001008 break;
1009 }
danielk19773f41e972004-06-08 00:39:01 +00001010 case SQLITE_BLOB: {
danielk19773f41e972004-06-08 00:39:01 +00001011 char *zText = 0;
danielk19773f41e972004-06-08 00:39:01 +00001012 char const *zBlob = sqlite3_value_blob(argv[0]);
drh1f0feef2007-05-15 13:27:07 +00001013 int nBlob = sqlite3_value_bytes(argv[0]);
1014 assert( zBlob==sqlite3_value_blob(argv[0]) ); /* No encoding change */
drhb1a6c3c2008-03-20 16:30:17 +00001015 zText = (char *)contextMalloc(context, (2*(i64)nBlob)+4);
danielk1977a1644fd2007-08-29 12:31:25 +00001016 if( zText ){
danielk19773f41e972004-06-08 00:39:01 +00001017 int i;
1018 for(i=0; i<nBlob; i++){
1019 zText[(i*2)+2] = hexdigits[(zBlob[i]>>4)&0x0F];
1020 zText[(i*2)+3] = hexdigits[(zBlob[i])&0x0F];
1021 }
1022 zText[(nBlob*2)+2] = '\'';
1023 zText[(nBlob*2)+3] = '\0';
1024 zText[0] = 'X';
1025 zText[1] = '\'';
danielk1977d8123362004-06-12 09:25:12 +00001026 sqlite3_result_text(context, zText, -1, SQLITE_TRANSIENT);
drh17435752007-08-16 04:30:38 +00001027 sqlite3_free(zText);
danielk19773f41e972004-06-08 00:39:01 +00001028 }
1029 break;
1030 }
drh9c054832004-05-31 18:51:57 +00001031 case SQLITE_TEXT: {
drh023ae032007-05-08 12:12:16 +00001032 int i,j;
1033 u64 n;
drh2646da72005-12-09 20:02:05 +00001034 const unsigned char *zArg = sqlite3_value_text(argv[0]);
drhf9b596e2004-05-26 16:54:42 +00001035 char *z;
1036
drh7a521cf2007-04-25 18:23:52 +00001037 if( zArg==0 ) return;
drh023ae032007-05-08 12:12:16 +00001038 for(i=0, n=0; zArg[i]; i++){ if( zArg[i]=='\'' ) n++; }
drhb1a6c3c2008-03-20 16:30:17 +00001039 z = contextMalloc(context, ((i64)i)+((i64)n)+3);
danielk1977a1644fd2007-08-29 12:31:25 +00001040 if( z ){
1041 z[0] = '\'';
1042 for(i=0, j=1; zArg[i]; i++){
1043 z[j++] = zArg[i];
1044 if( zArg[i]=='\'' ){
1045 z[j++] = '\'';
1046 }
drhf9b596e2004-05-26 16:54:42 +00001047 }
danielk1977a1644fd2007-08-29 12:31:25 +00001048 z[j++] = '\'';
1049 z[j] = 0;
1050 sqlite3_result_text(context, z, j, sqlite3_free);
drhf9b596e2004-05-26 16:54:42 +00001051 }
drha0df4cc2009-02-02 17:29:59 +00001052 break;
1053 }
1054 default: {
1055 assert( sqlite3_value_type(argv[0])==SQLITE_NULL );
1056 sqlite3_result_text(context, "NULL", 4, SQLITE_STATIC);
1057 break;
drhf9b596e2004-05-26 16:54:42 +00001058 }
drh47394702003-08-20 01:03:33 +00001059 }
1060}
1061
drh137c7282007-01-29 17:58:28 +00001062/*
drhd495d8c2013-02-22 19:34:25 +00001063** The unicode() function. Return the integer unicode code-point value
1064** for the first character of the input string.
1065*/
1066static void unicodeFunc(
1067 sqlite3_context *context,
1068 int argc,
1069 sqlite3_value **argv
1070){
1071 const unsigned char *z = sqlite3_value_text(argv[0]);
drh1d59d032013-03-01 23:40:26 +00001072 (void)argc;
drhd495d8c2013-02-22 19:34:25 +00001073 if( z && z[0] ) sqlite3_result_int(context, sqlite3Utf8Read(&z));
1074}
1075
1076/*
1077** The char() function takes zero or more arguments, each of which is
1078** an integer. It constructs a string where each character of the string
1079** is the unicode character for the corresponding integer argument.
1080*/
1081static void charFunc(
1082 sqlite3_context *context,
1083 int argc,
1084 sqlite3_value **argv
1085){
1086 unsigned char *z, *zOut;
1087 int i;
drhf3cdcdc2015-04-29 16:50:28 +00001088 zOut = z = sqlite3_malloc64( argc*4+1 );
drhd495d8c2013-02-22 19:34:25 +00001089 if( z==0 ){
1090 sqlite3_result_error_nomem(context);
1091 return;
1092 }
1093 for(i=0; i<argc; i++){
mistachkinc9545442013-02-26 05:42:30 +00001094 sqlite3_int64 x;
drhd495d8c2013-02-22 19:34:25 +00001095 unsigned c;
1096 x = sqlite3_value_int64(argv[i]);
1097 if( x<0 || x>0x10ffff ) x = 0xfffd;
1098 c = (unsigned)(x & 0x1fffff);
drhfe7a5d12013-03-07 14:00:04 +00001099 if( c<0x00080 ){
1100 *zOut++ = (u8)(c&0xFF);
1101 }else if( c<0x00800 ){
1102 *zOut++ = 0xC0 + (u8)((c>>6)&0x1F);
1103 *zOut++ = 0x80 + (u8)(c & 0x3F);
1104 }else if( c<0x10000 ){
1105 *zOut++ = 0xE0 + (u8)((c>>12)&0x0F);
1106 *zOut++ = 0x80 + (u8)((c>>6) & 0x3F);
1107 *zOut++ = 0x80 + (u8)(c & 0x3F);
drhd495d8c2013-02-22 19:34:25 +00001108 }else{
drhfe7a5d12013-03-07 14:00:04 +00001109 *zOut++ = 0xF0 + (u8)((c>>18) & 0x07);
1110 *zOut++ = 0x80 + (u8)((c>>12) & 0x3F);
1111 *zOut++ = 0x80 + (u8)((c>>6) & 0x3F);
1112 *zOut++ = 0x80 + (u8)(c & 0x3F);
1113 } \
drhd495d8c2013-02-22 19:34:25 +00001114 }
drhbbf483f2014-09-09 20:30:24 +00001115 sqlite3_result_text64(context, (char*)z, zOut-z, sqlite3_free, SQLITE_UTF8);
drhd495d8c2013-02-22 19:34:25 +00001116}
1117
1118/*
drh137c7282007-01-29 17:58:28 +00001119** The hex() function. Interpret the argument as a blob. Return
1120** a hexadecimal rendering as text.
1121*/
1122static void hexFunc(
1123 sqlite3_context *context,
1124 int argc,
1125 sqlite3_value **argv
1126){
1127 int i, n;
1128 const unsigned char *pBlob;
1129 char *zHex, *z;
1130 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001131 UNUSED_PARAMETER(argc);
drh1f0feef2007-05-15 13:27:07 +00001132 pBlob = sqlite3_value_blob(argv[0]);
drh137c7282007-01-29 17:58:28 +00001133 n = sqlite3_value_bytes(argv[0]);
drh1f0feef2007-05-15 13:27:07 +00001134 assert( pBlob==sqlite3_value_blob(argv[0]) ); /* No encoding change */
drhb1a6c3c2008-03-20 16:30:17 +00001135 z = zHex = contextMalloc(context, ((i64)n)*2 + 1);
danielk1977a1644fd2007-08-29 12:31:25 +00001136 if( zHex ){
1137 for(i=0; i<n; i++, pBlob++){
1138 unsigned char c = *pBlob;
1139 *(z++) = hexdigits[(c>>4)&0xf];
1140 *(z++) = hexdigits[c&0xf];
1141 }
1142 *z = 0;
1143 sqlite3_result_text(context, zHex, n*2, sqlite3_free);
drh137c7282007-01-29 17:58:28 +00001144 }
drh137c7282007-01-29 17:58:28 +00001145}
1146
drh26b6d902007-03-17 13:27:54 +00001147/*
drh8cff3822007-05-02 02:08:28 +00001148** The zeroblob(N) function returns a zero-filled blob of size N bytes.
1149*/
1150static void zeroblobFunc(
1151 sqlite3_context *context,
1152 int argc,
1153 sqlite3_value **argv
1154){
drh98640a32007-06-07 19:08:32 +00001155 i64 n;
dana4d5ae82015-07-24 16:24:37 +00001156 int rc;
drh8cff3822007-05-02 02:08:28 +00001157 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001158 UNUSED_PARAMETER(argc);
drh98640a32007-06-07 19:08:32 +00001159 n = sqlite3_value_int64(argv[0]);
dana4d5ae82015-07-24 16:24:37 +00001160 if( n<0 ) n = 0;
1161 rc = sqlite3_result_zeroblob64(context, n); /* IMP: R-00293-64994 */
1162 if( rc ){
1163 sqlite3_result_error_code(context, rc);
drh98640a32007-06-07 19:08:32 +00001164 }
drh8cff3822007-05-02 02:08:28 +00001165}
1166
1167/*
drh26b6d902007-03-17 13:27:54 +00001168** The replace() function. Three arguments are all strings: call
1169** them A, B, and C. The result is also a string which is derived
drhf7b54962013-05-28 12:11:54 +00001170** from A by replacing every occurrence of B with C. The match
drh26b6d902007-03-17 13:27:54 +00001171** must be exact. Collating sequences are not used.
1172*/
1173static void replaceFunc(
1174 sqlite3_context *context,
1175 int argc,
1176 sqlite3_value **argv
1177){
1178 const unsigned char *zStr; /* The input string A */
1179 const unsigned char *zPattern; /* The pattern string B */
1180 const unsigned char *zRep; /* The replacement string C */
1181 unsigned char *zOut; /* The output */
1182 int nStr; /* Size of zStr */
1183 int nPattern; /* Size of zPattern */
1184 int nRep; /* Size of zRep */
drh2e6400b2007-05-08 15:46:18 +00001185 i64 nOut; /* Maximum size of zOut */
drh26b6d902007-03-17 13:27:54 +00001186 int loopLimit; /* Last zStr[] that might match zPattern[] */
1187 int i, j; /* Loop counters */
1188
1189 assert( argc==3 );
danielk1977f3d3c272008-11-19 16:52:44 +00001190 UNUSED_PARAMETER(argc);
drh26b6d902007-03-17 13:27:54 +00001191 zStr = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001192 if( zStr==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001193 nStr = sqlite3_value_bytes(argv[0]);
1194 assert( zStr==sqlite3_value_text(argv[0]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001195 zPattern = sqlite3_value_text(argv[1]);
drha605fe82009-02-01 18:08:40 +00001196 if( zPattern==0 ){
drh23336062009-02-03 13:19:12 +00001197 assert( sqlite3_value_type(argv[1])==SQLITE_NULL
1198 || sqlite3_context_db_handle(context)->mallocFailed );
drha605fe82009-02-01 18:08:40 +00001199 return;
1200 }
1201 if( zPattern[0]==0 ){
1202 assert( sqlite3_value_type(argv[1])!=SQLITE_NULL );
1203 sqlite3_result_value(context, argv[0]);
1204 return;
1205 }
drh1f0feef2007-05-15 13:27:07 +00001206 nPattern = sqlite3_value_bytes(argv[1]);
1207 assert( zPattern==sqlite3_value_text(argv[1]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001208 zRep = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +00001209 if( zRep==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001210 nRep = sqlite3_value_bytes(argv[2]);
1211 assert( zRep==sqlite3_value_text(argv[2]) );
drh2e6400b2007-05-08 15:46:18 +00001212 nOut = nStr + 1;
1213 assert( nOut<SQLITE_MAX_LENGTH );
drhb1a6c3c2008-03-20 16:30:17 +00001214 zOut = contextMalloc(context, (i64)nOut);
drh2e6400b2007-05-08 15:46:18 +00001215 if( zOut==0 ){
1216 return;
drh26b6d902007-03-17 13:27:54 +00001217 }
drh26b6d902007-03-17 13:27:54 +00001218 loopLimit = nStr - nPattern;
1219 for(i=j=0; i<=loopLimit; i++){
1220 if( zStr[i]!=zPattern[0] || memcmp(&zStr[i], zPattern, nPattern) ){
1221 zOut[j++] = zStr[i];
1222 }else{
drh4a50aac2007-08-23 02:47:53 +00001223 u8 *zOld;
drhbb4957f2008-03-20 14:03:29 +00001224 sqlite3 *db = sqlite3_context_db_handle(context);
drh2e6400b2007-05-08 15:46:18 +00001225 nOut += nRep - nPattern;
drh27e62db2009-04-02 10:16:17 +00001226 testcase( nOut-1==db->aLimit[SQLITE_LIMIT_LENGTH] );
1227 testcase( nOut-2==db->aLimit[SQLITE_LIMIT_LENGTH] );
1228 if( nOut-1>db->aLimit[SQLITE_LIMIT_LENGTH] ){
drha0206bc2007-05-08 15:15:02 +00001229 sqlite3_result_error_toobig(context);
drhb9755982010-07-24 16:34:37 +00001230 sqlite3_free(zOut);
danielk197717374e82007-05-08 14:39:04 +00001231 return;
1232 }
drh4a50aac2007-08-23 02:47:53 +00001233 zOld = zOut;
drhf3cdcdc2015-04-29 16:50:28 +00001234 zOut = sqlite3_realloc64(zOut, (int)nOut);
drh2e6400b2007-05-08 15:46:18 +00001235 if( zOut==0 ){
danielk1977a1644fd2007-08-29 12:31:25 +00001236 sqlite3_result_error_nomem(context);
drhb9755982010-07-24 16:34:37 +00001237 sqlite3_free(zOld);
drh2e6400b2007-05-08 15:46:18 +00001238 return;
1239 }
drh26b6d902007-03-17 13:27:54 +00001240 memcpy(&zOut[j], zRep, nRep);
1241 j += nRep;
1242 i += nPattern-1;
1243 }
1244 }
drh2e6400b2007-05-08 15:46:18 +00001245 assert( j+nStr-i+1==nOut );
drh26b6d902007-03-17 13:27:54 +00001246 memcpy(&zOut[j], &zStr[i], nStr-i);
1247 j += nStr - i;
1248 assert( j<=nOut );
1249 zOut[j] = 0;
1250 sqlite3_result_text(context, (char*)zOut, j, sqlite3_free);
1251}
1252
drh309b3382007-03-17 17:52:42 +00001253/*
1254** Implementation of the TRIM(), LTRIM(), and RTRIM() functions.
1255** The userdata is 0x1 for left trim, 0x2 for right trim, 0x3 for both.
1256*/
1257static void trimFunc(
1258 sqlite3_context *context,
1259 int argc,
1260 sqlite3_value **argv
1261){
1262 const unsigned char *zIn; /* Input string */
1263 const unsigned char *zCharSet; /* Set of characters to trim */
1264 int nIn; /* Number of bytes in input */
drh7209c692008-04-27 18:40:11 +00001265 int flags; /* 1: trimleft 2: trimright 3: trim */
drhd1e3a612007-04-27 21:59:52 +00001266 int i; /* Loop counter */
drh1bd10f82008-12-10 21:19:56 +00001267 unsigned char *aLen = 0; /* Length of each character in zCharSet */
1268 unsigned char **azChar = 0; /* Individual characters in zCharSet */
drhd1e3a612007-04-27 21:59:52 +00001269 int nChar; /* Number of characters in zCharSet */
1270
drh309b3382007-03-17 17:52:42 +00001271 if( sqlite3_value_type(argv[0])==SQLITE_NULL ){
1272 return;
1273 }
1274 zIn = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001275 if( zIn==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001276 nIn = sqlite3_value_bytes(argv[0]);
1277 assert( zIn==sqlite3_value_text(argv[0]) );
drh309b3382007-03-17 17:52:42 +00001278 if( argc==1 ){
drhd1e3a612007-04-27 21:59:52 +00001279 static const unsigned char lenOne[] = { 1 };
danielk1977a4de4532008-09-02 15:44:08 +00001280 static unsigned char * const azOne[] = { (u8*)" " };
drhd1e3a612007-04-27 21:59:52 +00001281 nChar = 1;
drh8cff3822007-05-02 02:08:28 +00001282 aLen = (u8*)lenOne;
danielk1977bc67da42007-12-11 04:23:19 +00001283 azChar = (unsigned char **)azOne;
drhd1e3a612007-04-27 21:59:52 +00001284 zCharSet = 0;
drh7a521cf2007-04-25 18:23:52 +00001285 }else if( (zCharSet = sqlite3_value_text(argv[1]))==0 ){
drh309b3382007-03-17 17:52:42 +00001286 return;
drhd1e3a612007-04-27 21:59:52 +00001287 }else{
1288 const unsigned char *z;
1289 for(z=zCharSet, nChar=0; *z; nChar++){
drh4a919112007-05-15 11:55:09 +00001290 SQLITE_SKIP_UTF8(z);
drhd1e3a612007-04-27 21:59:52 +00001291 }
1292 if( nChar>0 ){
drhb1a6c3c2008-03-20 16:30:17 +00001293 azChar = contextMalloc(context, ((i64)nChar)*(sizeof(char*)+1));
drhd1e3a612007-04-27 21:59:52 +00001294 if( azChar==0 ){
1295 return;
1296 }
1297 aLen = (unsigned char*)&azChar[nChar];
1298 for(z=zCharSet, nChar=0; *z; nChar++){
danielk1977bc67da42007-12-11 04:23:19 +00001299 azChar[nChar] = (unsigned char *)z;
drh4a919112007-05-15 11:55:09 +00001300 SQLITE_SKIP_UTF8(z);
drh3abbd392008-12-10 23:04:13 +00001301 aLen[nChar] = (u8)(z - azChar[nChar]);
drhd1e3a612007-04-27 21:59:52 +00001302 }
1303 }
drh309b3382007-03-17 17:52:42 +00001304 }
drhd1e3a612007-04-27 21:59:52 +00001305 if( nChar>0 ){
shane1fc41292008-07-08 22:28:48 +00001306 flags = SQLITE_PTR_TO_INT(sqlite3_user_data(context));
drh309b3382007-03-17 17:52:42 +00001307 if( flags & 1 ){
drhd1e3a612007-04-27 21:59:52 +00001308 while( nIn>0 ){
drh1bd10f82008-12-10 21:19:56 +00001309 int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001310 for(i=0; i<nChar; i++){
1311 len = aLen[i];
drh27e62db2009-04-02 10:16:17 +00001312 if( len<=nIn && memcmp(zIn, azChar[i], len)==0 ) break;
drhd1e3a612007-04-27 21:59:52 +00001313 }
1314 if( i>=nChar ) break;
1315 zIn += len;
1316 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001317 }
1318 }
1319 if( flags & 2 ){
drhd1e3a612007-04-27 21:59:52 +00001320 while( nIn>0 ){
drh1bd10f82008-12-10 21:19:56 +00001321 int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001322 for(i=0; i<nChar; i++){
1323 len = aLen[i];
1324 if( len<=nIn && memcmp(&zIn[nIn-len],azChar[i],len)==0 ) break;
1325 }
1326 if( i>=nChar ) break;
1327 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001328 }
1329 }
drhd1e3a612007-04-27 21:59:52 +00001330 if( zCharSet ){
1331 sqlite3_free(azChar);
1332 }
drh309b3382007-03-17 17:52:42 +00001333 }
1334 sqlite3_result_text(context, (char*)zIn, nIn, SQLITE_TRANSIENT);
1335}
drh26b6d902007-03-17 13:27:54 +00001336
danielk1977a4de4532008-09-02 15:44:08 +00001337
drhcc153132016-08-04 12:35:17 +00001338#ifdef SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION
1339/*
1340** The "unknown" function is automatically substituted in place of
1341** any unrecognized function name when doing an EXPLAIN or EXPLAIN QUERY PLAN
1342** when the SQLITE_ENABLE_UNKNOWN_FUNCTION compile-time option is used.
1343** When the "sqlite3" command-line shell is built using this functionality,
1344** that allows an EXPLAIN or EXPLAIN QUERY PLAN for complex queries
1345** involving application-defined functions to be examined in a generic
1346** sqlite3 shell.
1347*/
1348static void unknownFunc(
1349 sqlite3_context *context,
1350 int argc,
1351 sqlite3_value **argv
1352){
1353 /* no-op */
1354}
1355#endif /*SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION*/
1356
1357
drh2ba3ccc2009-12-08 02:06:08 +00001358/* IMP: R-25361-16150 This function is omitted from SQLite by default. It
1359** is only available if the SQLITE_SOUNDEX compile-time option is used
1360** when SQLite is built.
1361*/
drhd24cc422003-03-27 12:51:24 +00001362#ifdef SQLITE_SOUNDEX
1363/*
1364** Compute the soundex encoding of a word.
drh2ba3ccc2009-12-08 02:06:08 +00001365**
1366** IMP: R-59782-00072 The soundex(X) function returns a string that is the
1367** soundex encoding of the string X.
drhd24cc422003-03-27 12:51:24 +00001368*/
drh137c7282007-01-29 17:58:28 +00001369static void soundexFunc(
1370 sqlite3_context *context,
1371 int argc,
1372 sqlite3_value **argv
1373){
drhd24cc422003-03-27 12:51:24 +00001374 char zResult[8];
drh4c755c02004-08-08 20:22:17 +00001375 const u8 *zIn;
drhd24cc422003-03-27 12:51:24 +00001376 int i, j;
1377 static const unsigned char iCode[] = {
1378 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1379 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1380 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1381 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1382 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1383 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1384 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1385 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1386 };
1387 assert( argc==1 );
drh4c755c02004-08-08 20:22:17 +00001388 zIn = (u8*)sqlite3_value_text(argv[0]);
drhbdf67e02006-08-19 11:34:01 +00001389 if( zIn==0 ) zIn = (u8*)"";
drhdc86e2b2009-01-24 11:30:42 +00001390 for(i=0; zIn[i] && !sqlite3Isalpha(zIn[i]); i++){}
drhd24cc422003-03-27 12:51:24 +00001391 if( zIn[i] ){
drhbdf67e02006-08-19 11:34:01 +00001392 u8 prevcode = iCode[zIn[i]&0x7f];
danielk197778ca0e72009-01-20 16:53:39 +00001393 zResult[0] = sqlite3Toupper(zIn[i]);
drhd24cc422003-03-27 12:51:24 +00001394 for(j=1; j<4 && zIn[i]; i++){
1395 int code = iCode[zIn[i]&0x7f];
1396 if( code>0 ){
drhbdf67e02006-08-19 11:34:01 +00001397 if( code!=prevcode ){
1398 prevcode = code;
1399 zResult[j++] = code + '0';
1400 }
1401 }else{
1402 prevcode = 0;
drhd24cc422003-03-27 12:51:24 +00001403 }
1404 }
1405 while( j<4 ){
1406 zResult[j++] = '0';
1407 }
1408 zResult[j] = 0;
danielk1977d8123362004-06-12 09:25:12 +00001409 sqlite3_result_text(context, zResult, 4, SQLITE_TRANSIENT);
drhd24cc422003-03-27 12:51:24 +00001410 }else{
drh2ba3ccc2009-12-08 02:06:08 +00001411 /* IMP: R-64894-50321 The string "?000" is returned if the argument
1412 ** is NULL or contains no ASCII alphabetic characters. */
danielk1977d8123362004-06-12 09:25:12 +00001413 sqlite3_result_text(context, "?000", 4, SQLITE_STATIC);
drhd24cc422003-03-27 12:51:24 +00001414 }
1415}
drh2ba3ccc2009-12-08 02:06:08 +00001416#endif /* SQLITE_SOUNDEX */
drhd24cc422003-03-27 12:51:24 +00001417
drhfdb83b22006-06-17 14:12:47 +00001418#ifndef SQLITE_OMIT_LOAD_EXTENSION
1419/*
1420** A function that loads a shared-library extension then returns NULL.
1421*/
1422static void loadExt(sqlite3_context *context, int argc, sqlite3_value **argv){
danielk197765fd59f2006-06-24 11:51:33 +00001423 const char *zFile = (const char *)sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001424 const char *zProc;
drhfa4a4b92008-03-19 21:45:51 +00001425 sqlite3 *db = sqlite3_context_db_handle(context);
drhfdb83b22006-06-17 14:12:47 +00001426 char *zErrMsg = 0;
1427
drh191dd062016-04-21 01:30:09 +00001428 /* Disallow the load_extension() SQL function unless the SQLITE_LoadExtFunc
drh1a55ded2016-04-20 00:30:05 +00001429 ** flag is set. See the sqlite3_enable_load_extension() API.
1430 */
drhf602a162016-04-21 01:58:21 +00001431 if( (db->flags & SQLITE_LoadExtFunc)==0 ){
1432 sqlite3_result_error(context, "not authorized", -1);
1433 return;
1434 }
drh1a55ded2016-04-20 00:30:05 +00001435
drhfdb83b22006-06-17 14:12:47 +00001436 if( argc==2 ){
danielk197765fd59f2006-06-24 11:51:33 +00001437 zProc = (const char *)sqlite3_value_text(argv[1]);
drh7a521cf2007-04-25 18:23:52 +00001438 }else{
1439 zProc = 0;
drhfdb83b22006-06-17 14:12:47 +00001440 }
drh7a521cf2007-04-25 18:23:52 +00001441 if( zFile && sqlite3_load_extension(db, zFile, zProc, &zErrMsg) ){
drhfdb83b22006-06-17 14:12:47 +00001442 sqlite3_result_error(context, zErrMsg, -1);
1443 sqlite3_free(zErrMsg);
1444 }
1445}
1446#endif
1447
danielk197701427a62005-01-11 13:02:33 +00001448
drh0ac65892002-04-20 14:24:41 +00001449/*
drhd3a149e2002-02-24 17:12:53 +00001450** An instance of the following structure holds the context of a
drhdd5baa92002-02-27 19:50:59 +00001451** sum() or avg() aggregate computation.
1452*/
1453typedef struct SumCtx SumCtx;
1454struct SumCtx {
drh8c08e862006-02-11 17:34:00 +00001455 double rSum; /* Floating point sum */
1456 i64 iSum; /* Integer sum */
1457 i64 cnt; /* Number of elements summed */
1458 u8 overflow; /* True if integer overflow seen */
1459 u8 approx; /* True if non-integer value was input to the sum */
drhdd5baa92002-02-27 19:50:59 +00001460};
1461
1462/*
drha97fdd32006-01-12 22:17:50 +00001463** Routines used to compute the sum, average, and total.
1464**
1465** The SUM() function follows the (broken) SQL standard which means
1466** that it returns NULL if it sums over no inputs. TOTAL returns
1467** 0.0 in that case. In addition, TOTAL always returns a float where
1468** SUM might return an integer if it never encounters a floating point
drhc806d852006-05-11 13:25:39 +00001469** value. TOTAL never fails, but SUM might through an exception if
1470** it overflows an integer.
drhdd5baa92002-02-27 19:50:59 +00001471*/
danielk19770ae8b832004-05-25 12:05:56 +00001472static void sumStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drhdd5baa92002-02-27 19:50:59 +00001473 SumCtx *p;
drh3d1d95e2005-09-08 10:37:01 +00001474 int type;
drh3f219f42005-09-08 19:45:57 +00001475 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001476 UNUSED_PARAMETER(argc);
drh4f26d6c2004-05-26 23:25:30 +00001477 p = sqlite3_aggregate_context(context, sizeof(*p));
drh29d72102006-02-09 22:13:41 +00001478 type = sqlite3_value_numeric_type(argv[0]);
drh3d1d95e2005-09-08 10:37:01 +00001479 if( p && type!=SQLITE_NULL ){
drh739105c2002-05-29 23:22:23 +00001480 p->cnt++;
drh29d72102006-02-09 22:13:41 +00001481 if( type==SQLITE_INTEGER ){
drh8c08e862006-02-11 17:34:00 +00001482 i64 v = sqlite3_value_int64(argv[0]);
1483 p->rSum += v;
drh158b9cb2011-03-05 20:59:46 +00001484 if( (p->approx|p->overflow)==0 && sqlite3AddInt64(&p->iSum, v) ){
1485 p->overflow = 1;
drh29d72102006-02-09 22:13:41 +00001486 }
1487 }else{
drh8c08e862006-02-11 17:34:00 +00001488 p->rSum += sqlite3_value_double(argv[0]);
drh29d72102006-02-09 22:13:41 +00001489 p->approx = 1;
drh3f219f42005-09-08 19:45:57 +00001490 }
drh739105c2002-05-29 23:22:23 +00001491 }
drhdd5baa92002-02-27 19:50:59 +00001492}
danielk19770ae8b832004-05-25 12:05:56 +00001493static void sumFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001494 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001495 p = sqlite3_aggregate_context(context, 0);
drhc2bd9132005-09-08 20:37:43 +00001496 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001497 if( p->overflow ){
1498 sqlite3_result_error(context,"integer overflow",-1);
1499 }else if( p->approx ){
1500 sqlite3_result_double(context, p->rSum);
drhc2bd9132005-09-08 20:37:43 +00001501 }else{
drh8c08e862006-02-11 17:34:00 +00001502 sqlite3_result_int64(context, p->iSum);
drhc2bd9132005-09-08 20:37:43 +00001503 }
drh3d1d95e2005-09-08 10:37:01 +00001504 }
drhdd5baa92002-02-27 19:50:59 +00001505}
danielk19770ae8b832004-05-25 12:05:56 +00001506static void avgFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001507 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001508 p = sqlite3_aggregate_context(context, 0);
drh739105c2002-05-29 23:22:23 +00001509 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001510 sqlite3_result_double(context, p->rSum/(double)p->cnt);
drhdd5baa92002-02-27 19:50:59 +00001511 }
1512}
drha97fdd32006-01-12 22:17:50 +00001513static void totalFinalize(sqlite3_context *context){
1514 SumCtx *p;
1515 p = sqlite3_aggregate_context(context, 0);
shanefbd60f82009-02-04 03:59:25 +00001516 /* (double)0 In case of SQLITE_OMIT_FLOATING_POINT... */
1517 sqlite3_result_double(context, p ? p->rSum : (double)0);
drha97fdd32006-01-12 22:17:50 +00001518}
drhdd5baa92002-02-27 19:50:59 +00001519
1520/*
drh0bce8352002-02-28 00:41:10 +00001521** The following structure keeps track of state information for the
1522** count() aggregate function.
1523*/
1524typedef struct CountCtx CountCtx;
1525struct CountCtx {
drhfc6ad392006-02-09 13:38:19 +00001526 i64 n;
drh0bce8352002-02-28 00:41:10 +00001527};
drhdd5baa92002-02-27 19:50:59 +00001528
drh0bce8352002-02-28 00:41:10 +00001529/*
1530** Routines to implement the count() aggregate function.
1531*/
danielk19770ae8b832004-05-25 12:05:56 +00001532static void countStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drh0bce8352002-02-28 00:41:10 +00001533 CountCtx *p;
drh4f26d6c2004-05-26 23:25:30 +00001534 p = sqlite3_aggregate_context(context, sizeof(*p));
drh9c054832004-05-31 18:51:57 +00001535 if( (argc==0 || SQLITE_NULL!=sqlite3_value_type(argv[0])) && p ){
drh0bce8352002-02-28 00:41:10 +00001536 p->n++;
1537 }
drh2e79c3d2009-04-08 23:04:14 +00001538
drhd3264c72009-04-15 13:39:47 +00001539#ifndef SQLITE_OMIT_DEPRECATED
drh2e79c3d2009-04-08 23:04:14 +00001540 /* The sqlite3_aggregate_count() function is deprecated. But just to make
1541 ** sure it still operates correctly, verify that its count agrees with our
1542 ** internal count when using count(*) and when the total count can be
1543 ** expressed as a 32-bit integer. */
1544 assert( argc==1 || p==0 || p->n>0x7fffffff
1545 || p->n==sqlite3_aggregate_count(context) );
drhd3264c72009-04-15 13:39:47 +00001546#endif
drh0bce8352002-02-28 00:41:10 +00001547}
danielk19770ae8b832004-05-25 12:05:56 +00001548static void countFinalize(sqlite3_context *context){
drh0bce8352002-02-28 00:41:10 +00001549 CountCtx *p;
drhabfcea22005-09-06 20:36:48 +00001550 p = sqlite3_aggregate_context(context, 0);
drhfc6ad392006-02-09 13:38:19 +00001551 sqlite3_result_int64(context, p ? p->n : 0);
drh0bce8352002-02-28 00:41:10 +00001552}
1553
1554/*
drh0bce8352002-02-28 00:41:10 +00001555** Routines to implement min() and max() aggregate functions.
1556*/
danielk197762c14b32008-11-19 09:05:26 +00001557static void minmaxStep(
1558 sqlite3_context *context,
1559 int NotUsed,
1560 sqlite3_value **argv
1561){
danielk197788208052004-05-25 01:13:20 +00001562 Mem *pArg = (Mem *)argv[0];
drh9eb516c2004-07-18 20:52:32 +00001563 Mem *pBest;
danielk197762c14b32008-11-19 09:05:26 +00001564 UNUSED_PARAMETER(NotUsed);
drh9eb516c2004-07-18 20:52:32 +00001565
drh9eb516c2004-07-18 20:52:32 +00001566 pBest = (Mem *)sqlite3_aggregate_context(context, sizeof(*pBest));
danielk19773aeab9e2004-06-24 00:20:04 +00001567 if( !pBest ) return;
drh268380c2004-02-25 13:47:31 +00001568
drh94a6d992012-02-02 18:42:09 +00001569 if( sqlite3_value_type(argv[0])==SQLITE_NULL ){
1570 if( pBest->flags ) sqlite3SkipAccumulatorLoad(context);
1571 }else if( pBest->flags ){
drh9eb516c2004-07-18 20:52:32 +00001572 int max;
1573 int cmp;
danielk1977dc1bdc42004-06-11 10:51:27 +00001574 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk19777e18c252004-05-25 11:47:24 +00001575 /* This step function is used for both the min() and max() aggregates,
1576 ** the only difference between the two being that the sense of the
1577 ** comparison is inverted. For the max() aggregate, the
1578 ** sqlite3_user_data() function returns (void *)-1. For min() it
1579 ** returns (void *)db, where db is the sqlite3* database pointer.
1580 ** Therefore the next statement sets variable 'max' to 1 for the max()
1581 ** aggregate, or 0 for min().
1582 */
drh309b3382007-03-17 17:52:42 +00001583 max = sqlite3_user_data(context)!=0;
danielk1977dc1bdc42004-06-11 10:51:27 +00001584 cmp = sqlite3MemCompare(pBest, pArg, pColl);
danielk197788208052004-05-25 01:13:20 +00001585 if( (max && cmp<0) || (!max && cmp>0) ){
drhb21c8cd2007-08-21 19:33:56 +00001586 sqlite3VdbeMemCopy(pBest, pArg);
drh7a957892012-02-02 17:35:43 +00001587 }else{
1588 sqlite3SkipAccumulatorLoad(context);
danielk197788208052004-05-25 01:13:20 +00001589 }
drh268380c2004-02-25 13:47:31 +00001590 }else{
drh035e5632014-09-16 14:16:31 +00001591 pBest->db = sqlite3_context_db_handle(context);
drhb21c8cd2007-08-21 19:33:56 +00001592 sqlite3VdbeMemCopy(pBest, pArg);
drh0bce8352002-02-28 00:41:10 +00001593 }
1594}
danielk19770ae8b832004-05-25 12:05:56 +00001595static void minMaxFinalize(sqlite3_context *context){
danielk197788208052004-05-25 01:13:20 +00001596 sqlite3_value *pRes;
drhabfcea22005-09-06 20:36:48 +00001597 pRes = (sqlite3_value *)sqlite3_aggregate_context(context, 0);
1598 if( pRes ){
drh94a6d992012-02-02 18:42:09 +00001599 if( pRes->flags ){
drhabfcea22005-09-06 20:36:48 +00001600 sqlite3_result_value(context, pRes);
1601 }
1602 sqlite3VdbeMemRelease(pRes);
drh0bce8352002-02-28 00:41:10 +00001603 }
1604}
drhdd5baa92002-02-27 19:50:59 +00001605
drhb0689692007-11-01 17:38:30 +00001606/*
1607** group_concat(EXPR, ?SEPARATOR?)
1608*/
1609static void groupConcatStep(
1610 sqlite3_context *context,
1611 int argc,
1612 sqlite3_value **argv
1613){
1614 const char *zVal;
drhade86482007-11-28 22:36:40 +00001615 StrAccum *pAccum;
drhb0689692007-11-01 17:38:30 +00001616 const char *zSep;
drh07d31172009-02-02 21:57:05 +00001617 int nVal, nSep;
1618 assert( argc==1 || argc==2 );
1619 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drhade86482007-11-28 22:36:40 +00001620 pAccum = (StrAccum*)sqlite3_aggregate_context(context, sizeof(*pAccum));
1621
1622 if( pAccum ){
drhbb4957f2008-03-20 14:03:29 +00001623 sqlite3 *db = sqlite3_context_db_handle(context);
drhc0490572015-05-02 11:45:53 +00001624 int firstTerm = pAccum->mxAlloc==0;
drhbb4957f2008-03-20 14:03:29 +00001625 pAccum->mxAlloc = db->aLimit[SQLITE_LIMIT_LENGTH];
drh8bfd7192009-06-19 16:44:41 +00001626 if( !firstTerm ){
drh07d31172009-02-02 21:57:05 +00001627 if( argc==2 ){
1628 zSep = (char*)sqlite3_value_text(argv[1]);
1629 nSep = sqlite3_value_bytes(argv[1]);
drhb0689692007-11-01 17:38:30 +00001630 }else{
1631 zSep = ",";
drhade86482007-11-28 22:36:40 +00001632 nSep = 1;
drhb0689692007-11-01 17:38:30 +00001633 }
drha9ab4812013-12-11 11:00:44 +00001634 if( nSep ) sqlite3StrAccumAppend(pAccum, zSep, nSep);
drhb0689692007-11-01 17:38:30 +00001635 }
drh07d31172009-02-02 21:57:05 +00001636 zVal = (char*)sqlite3_value_text(argv[0]);
1637 nVal = sqlite3_value_bytes(argv[0]);
drh8009c9b2014-05-07 20:24:00 +00001638 if( zVal ) sqlite3StrAccumAppend(pAccum, zVal, nVal);
drhb0689692007-11-01 17:38:30 +00001639 }
1640}
1641static void groupConcatFinalize(sqlite3_context *context){
drhade86482007-11-28 22:36:40 +00001642 StrAccum *pAccum;
1643 pAccum = sqlite3_aggregate_context(context, 0);
1644 if( pAccum ){
drhb49bc862013-08-21 21:12:10 +00001645 if( pAccum->accError==STRACCUM_TOOBIG ){
drhade86482007-11-28 22:36:40 +00001646 sqlite3_result_error_toobig(context);
drhb49bc862013-08-21 21:12:10 +00001647 }else if( pAccum->accError==STRACCUM_NOMEM ){
drhade86482007-11-28 22:36:40 +00001648 sqlite3_result_error_nomem(context);
1649 }else{
1650 sqlite3_result_text(context, sqlite3StrAccumFinish(pAccum), -1,
1651 sqlite3_free);
1652 }
drhb0689692007-11-01 17:38:30 +00001653 }
1654}
drh4e5ffc52004-08-31 00:52:37 +00001655
drhd3a149e2002-02-24 17:12:53 +00001656/*
drha4741842010-04-25 20:58:37 +00001657** This routine does per-connection function registration. Most
1658** of the built-in functions above are part of the global function set.
1659** This routine only deals with those that are not global.
drhdc04c582002-02-24 01:55:15 +00001660*/
drh80738d92016-02-15 00:34:16 +00001661void sqlite3RegisterPerConnectionBuiltinFunctions(sqlite3 *db){
drha4741842010-04-25 20:58:37 +00001662 int rc = sqlite3_overload_function(db, "MATCH", 2);
1663 assert( rc==SQLITE_NOMEM || rc==SQLITE_OK );
1664 if( rc==SQLITE_NOMEM ){
drh4a642b62016-02-05 01:55:27 +00001665 sqlite3OomFault(db);
danielk1977832a58a2007-06-22 15:21:15 +00001666 }
drh55ef4d92005-08-14 01:20:37 +00001667}
1668
1669/*
1670** Set the LIKEOPT flag on the 2-argument function with the given name.
1671*/
drh1bd10f82008-12-10 21:19:56 +00001672static void setLikeOptFlag(sqlite3 *db, const char *zName, u8 flagVal){
drh55ef4d92005-08-14 01:20:37 +00001673 FuncDef *pDef;
drh80738d92016-02-15 00:34:16 +00001674 pDef = sqlite3FindFunction(db, zName, 2, SQLITE_UTF8, 0);
drhd27135a2009-04-02 13:36:37 +00001675 if( ALWAYS(pDef) ){
drhd36e1042013-09-06 13:10:12 +00001676 pDef->funcFlags |= flagVal;
drh55ef4d92005-08-14 01:20:37 +00001677 }
1678}
1679
1680/*
1681** Register the built-in LIKE and GLOB functions. The caseSensitive
1682** parameter determines whether or not the LIKE operator is case
1683** sensitive. GLOB is always case sensitive.
1684*/
1685void sqlite3RegisterLikeFunctions(sqlite3 *db, int caseSensitive){
1686 struct compareInfo *pInfo;
1687 if( caseSensitive ){
1688 pInfo = (struct compareInfo*)&likeInfoAlt;
1689 }else{
1690 pInfo = (struct compareInfo*)&likeInfoNorm;
1691 }
drh901e9942010-12-15 18:54:37 +00001692 sqlite3CreateFunc(db, "like", 2, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0);
1693 sqlite3CreateFunc(db, "like", 3, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0);
1694 sqlite3CreateFunc(db, "glob", 2, SQLITE_UTF8,
dand2199f02010-08-27 17:48:52 +00001695 (struct compareInfo*)&globInfo, likeFunc, 0, 0, 0);
drhd64fe2f2005-08-28 17:00:23 +00001696 setLikeOptFlag(db, "glob", SQLITE_FUNC_LIKE | SQLITE_FUNC_CASE);
1697 setLikeOptFlag(db, "like",
1698 caseSensitive ? (SQLITE_FUNC_LIKE | SQLITE_FUNC_CASE) : SQLITE_FUNC_LIKE);
drh55ef4d92005-08-14 01:20:37 +00001699}
1700
1701/*
1702** pExpr points to an expression which implements a function. If
1703** it is appropriate to apply the LIKE optimization to that function
1704** then set aWc[0] through aWc[2] to the wildcard characters and
1705** return TRUE. If the function is not a LIKE-style function then
1706** return FALSE.
drh16897072015-03-07 00:57:37 +00001707**
1708** *pIsNocase is set to true if uppercase and lowercase are equivalent for
1709** the function (default for LIKE). If the function makes the distinction
1710** between uppercase and lowercase (as does GLOB) then *pIsNocase is set to
1711** false.
drh55ef4d92005-08-14 01:20:37 +00001712*/
drhd64fe2f2005-08-28 17:00:23 +00001713int sqlite3IsLikeFunction(sqlite3 *db, Expr *pExpr, int *pIsNocase, char *aWc){
drh55ef4d92005-08-14 01:20:37 +00001714 FuncDef *pDef;
danielk19776ab3a2e2009-02-19 14:39:25 +00001715 if( pExpr->op!=TK_FUNCTION
1716 || !pExpr->x.pList
1717 || pExpr->x.pList->nExpr!=2
1718 ){
drh55ef4d92005-08-14 01:20:37 +00001719 return 0;
1720 }
danielk19776ab3a2e2009-02-19 14:39:25 +00001721 assert( !ExprHasProperty(pExpr, EP_xIsSelect) );
drh80738d92016-02-15 00:34:16 +00001722 pDef = sqlite3FindFunction(db, pExpr->u.zToken, 2, SQLITE_UTF8, 0);
drhd36e1042013-09-06 13:10:12 +00001723 if( NEVER(pDef==0) || (pDef->funcFlags & SQLITE_FUNC_LIKE)==0 ){
drh55ef4d92005-08-14 01:20:37 +00001724 return 0;
1725 }
1726
1727 /* The memcpy() statement assumes that the wildcard characters are
1728 ** the first three statements in the compareInfo structure. The
1729 ** asserts() that follow verify that assumption
1730 */
1731 memcpy(aWc, pDef->pUserData, 3);
1732 assert( (char*)&likeInfoAlt == (char*)&likeInfoAlt.matchAll );
1733 assert( &((char*)&likeInfoAlt)[1] == (char*)&likeInfoAlt.matchOne );
1734 assert( &((char*)&likeInfoAlt)[2] == (char*)&likeInfoAlt.matchSet );
drhd36e1042013-09-06 13:10:12 +00001735 *pIsNocase = (pDef->funcFlags & SQLITE_FUNC_CASE)==0;
drh55ef4d92005-08-14 01:20:37 +00001736 return 1;
drhdc04c582002-02-24 01:55:15 +00001737}
danielk19778c0a7912008-08-20 14:49:23 +00001738
drh70a8ca32008-08-21 18:49:27 +00001739/*
peter.d.reid60ec9142014-09-06 16:39:46 +00001740** All of the FuncDef structures in the aBuiltinFunc[] array above
drh777c5382008-08-21 20:21:34 +00001741** to the global function hash table. This occurs at start-time (as
1742** a consequence of calling sqlite3_initialize()).
1743**
1744** After this routine runs
drh70a8ca32008-08-21 18:49:27 +00001745*/
drh80738d92016-02-15 00:34:16 +00001746void sqlite3RegisterBuiltinFunctions(void){
danielk197793ce7412008-09-01 19:14:02 +00001747 /*
1748 ** The following array holds FuncDef structures for all of the functions
1749 ** defined in this file.
1750 **
1751 ** The array cannot be constant since changes are made to the
1752 ** FuncDef.pHash elements at start-time. The elements of this array
1753 ** are read-only after initialization is complete.
drh80738d92016-02-15 00:34:16 +00001754 **
1755 ** For peak efficiency, put the most frequently used function last.
danielk197793ce7412008-09-01 19:14:02 +00001756 */
drh80738d92016-02-15 00:34:16 +00001757 static FuncDef aBuiltinFunc[] = {
1758#ifdef SQLITE_SOUNDEX
1759 FUNCTION(soundex, 1, 0, 0, soundexFunc ),
1760#endif
1761#ifndef SQLITE_OMIT_LOAD_EXTENSION
1762 VFUNCTION(load_extension, 1, 0, 0, loadExt ),
1763 VFUNCTION(load_extension, 2, 0, 0, loadExt ),
1764#endif
1765#if SQLITE_USER_AUTHENTICATION
1766 FUNCTION(sqlite_crypt, 2, 0, 0, sqlite3CryptFunc ),
1767#endif
1768#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
1769 DFUNCTION(sqlite_compileoption_used,1, 0, 0, compileoptionusedFunc ),
1770 DFUNCTION(sqlite_compileoption_get, 1, 0, 0, compileoptiongetFunc ),
1771#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
1772 FUNCTION2(unlikely, 1, 0, 0, noopFunc, SQLITE_FUNC_UNLIKELY),
1773 FUNCTION2(likelihood, 2, 0, 0, noopFunc, SQLITE_FUNC_UNLIKELY),
1774 FUNCTION2(likely, 1, 0, 0, noopFunc, SQLITE_FUNC_UNLIKELY),
danielk197793ce7412008-09-01 19:14:02 +00001775 FUNCTION(ltrim, 1, 1, 0, trimFunc ),
1776 FUNCTION(ltrim, 2, 1, 0, trimFunc ),
1777 FUNCTION(rtrim, 1, 2, 0, trimFunc ),
1778 FUNCTION(rtrim, 2, 2, 0, trimFunc ),
1779 FUNCTION(trim, 1, 3, 0, trimFunc ),
1780 FUNCTION(trim, 2, 3, 0, trimFunc ),
1781 FUNCTION(min, -1, 0, 1, minmaxFunc ),
1782 FUNCTION(min, 0, 0, 1, 0 ),
drh9588ad92014-09-15 14:46:02 +00001783 AGGREGATE2(min, 1, 0, 1, minmaxStep, minMaxFinalize,
1784 SQLITE_FUNC_MINMAX ),
danielk197793ce7412008-09-01 19:14:02 +00001785 FUNCTION(max, -1, 1, 1, minmaxFunc ),
1786 FUNCTION(max, 0, 1, 1, 0 ),
drh9588ad92014-09-15 14:46:02 +00001787 AGGREGATE2(max, 1, 1, 1, minmaxStep, minMaxFinalize,
1788 SQLITE_FUNC_MINMAX ),
drha748fdc2012-03-28 01:34:47 +00001789 FUNCTION2(typeof, 1, 0, 0, typeofFunc, SQLITE_FUNC_TYPEOF),
1790 FUNCTION2(length, 1, 0, 0, lengthFunc, SQLITE_FUNC_LENGTH),
drhd55e0722012-10-25 03:07:29 +00001791 FUNCTION(instr, 2, 0, 0, instrFunc ),
drha5c14162013-12-17 15:03:06 +00001792 FUNCTION(printf, -1, 0, 0, printfFunc ),
drhd495d8c2013-02-22 19:34:25 +00001793 FUNCTION(unicode, 1, 0, 0, unicodeFunc ),
1794 FUNCTION(char, -1, 0, 0, charFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001795 FUNCTION(abs, 1, 0, 0, absFunc ),
shanefbd60f82009-02-04 03:59:25 +00001796#ifndef SQLITE_OMIT_FLOATING_POINT
danielk197793ce7412008-09-01 19:14:02 +00001797 FUNCTION(round, 1, 0, 0, roundFunc ),
1798 FUNCTION(round, 2, 0, 0, roundFunc ),
shanefbd60f82009-02-04 03:59:25 +00001799#endif
danielk197793ce7412008-09-01 19:14:02 +00001800 FUNCTION(upper, 1, 0, 0, upperFunc ),
1801 FUNCTION(lower, 1, 0, 0, lowerFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001802 FUNCTION(hex, 1, 0, 0, hexFunc ),
drhcca9f3d2013-09-06 15:23:29 +00001803 FUNCTION2(ifnull, 2, 0, 0, noopFunc, SQLITE_FUNC_COALESCE),
drhb1fba282013-11-21 14:33:48 +00001804 VFUNCTION(random, 0, 0, 0, randomFunc ),
1805 VFUNCTION(randomblob, 1, 0, 0, randomBlob ),
danielk197793ce7412008-09-01 19:14:02 +00001806 FUNCTION(nullif, 2, 0, 1, nullifFunc ),
drh03bf26d2015-08-31 21:16:36 +00001807 DFUNCTION(sqlite_version, 0, 0, 0, versionFunc ),
1808 DFUNCTION(sqlite_source_id, 0, 0, 0, sourceidFunc ),
drh840561f2011-04-27 18:08:42 +00001809 FUNCTION(sqlite_log, 2, 0, 0, errlogFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001810 FUNCTION(quote, 1, 0, 0, quoteFunc ),
drhb1fba282013-11-21 14:33:48 +00001811 VFUNCTION(last_insert_rowid, 0, 0, 0, last_insert_rowid),
1812 VFUNCTION(changes, 0, 0, 0, changes ),
1813 VFUNCTION(total_changes, 0, 0, 0, total_changes ),
danielk197793ce7412008-09-01 19:14:02 +00001814 FUNCTION(replace, 3, 0, 0, replaceFunc ),
1815 FUNCTION(zeroblob, 1, 0, 0, zeroblobFunc ),
drh80738d92016-02-15 00:34:16 +00001816 FUNCTION(substr, 2, 0, 0, substrFunc ),
1817 FUNCTION(substr, 3, 0, 0, substrFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001818 AGGREGATE(sum, 1, 0, 0, sumStep, sumFinalize ),
1819 AGGREGATE(total, 1, 0, 0, sumStep, totalFinalize ),
1820 AGGREGATE(avg, 1, 0, 0, sumStep, avgFinalize ),
drh9588ad92014-09-15 14:46:02 +00001821 AGGREGATE2(count, 0, 0, 0, countStep, countFinalize,
1822 SQLITE_FUNC_COUNT ),
danielk197793ce7412008-09-01 19:14:02 +00001823 AGGREGATE(count, 1, 0, 0, countStep, countFinalize ),
drh07d31172009-02-02 21:57:05 +00001824 AGGREGATE(group_concat, 1, 0, 0, groupConcatStep, groupConcatFinalize),
1825 AGGREGATE(group_concat, 2, 0, 0, groupConcatStep, groupConcatFinalize),
danielk197793ce7412008-09-01 19:14:02 +00001826
1827 LIKEFUNC(glob, 2, &globInfo, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
drhcc153132016-08-04 12:35:17 +00001828#ifdef SQLITE_CASE_SENSITIVE_LIKE
danielk197793ce7412008-09-01 19:14:02 +00001829 LIKEFUNC(like, 2, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
1830 LIKEFUNC(like, 3, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
drhcc153132016-08-04 12:35:17 +00001831#else
danielk197793ce7412008-09-01 19:14:02 +00001832 LIKEFUNC(like, 2, &likeInfoNorm, SQLITE_FUNC_LIKE),
1833 LIKEFUNC(like, 3, &likeInfoNorm, SQLITE_FUNC_LIKE),
drhcc153132016-08-04 12:35:17 +00001834#endif
1835#ifdef SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION
1836 FUNCTION(unknown, -1, 0, 0, unknownFunc ),
1837#endif
drh80738d92016-02-15 00:34:16 +00001838 FUNCTION(coalesce, 1, 0, 0, 0 ),
1839 FUNCTION(coalesce, 0, 0, 0, 0 ),
1840 FUNCTION2(coalesce, -1, 0, 0, noopFunc, SQLITE_FUNC_COALESCE),
danielk197793ce7412008-09-01 19:14:02 +00001841 };
drh545f5872010-04-24 14:02:59 +00001842#ifndef SQLITE_OMIT_ALTERTABLE
1843 sqlite3AlterFunctions();
1844#endif
dan0106e372013-08-12 16:34:32 +00001845#if defined(SQLITE_ENABLE_STAT3) || defined(SQLITE_ENABLE_STAT4)
1846 sqlite3AnalyzeFunctions();
1847#endif
drh80738d92016-02-15 00:34:16 +00001848 sqlite3RegisterDateTimeFunctions();
1849 sqlite3InsertBuiltinFuncs(aBuiltinFunc, ArraySize(aBuiltinFunc));
1850
1851#if 0 /* Enable to print out how the built-in functions are hashed */
1852 {
1853 int i;
1854 FuncDef *p;
1855 for(i=0; i<SQLITE_FUNC_HASH_SZ; i++){
1856 printf("FUNC-HASH %02d:", i);
1857 for(p=sqlite3BuiltinFunctions.a[i]; p; p=p->u.pHash){
1858 int n = sqlite3Strlen30(p->zName);
1859 int h = p->zName[0] + n;
1860 printf(" %s(%d)", p->zName, h);
1861 }
1862 printf("\n");
1863 }
1864 }
1865#endif
drh70a8ca32008-08-21 18:49:27 +00001866}