blob: dae85d60a839f8911c91f97f467b8c4cc8b5e0fd [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){
drh21d59782018-01-23 19:24:54 +000038 assert( context->isError<=0 );
39 context->isError = -1;
drh7a957892012-02-02 17:35:43 +000040 context->skipFlag = 1;
41}
42
43/*
drh0bce8352002-02-28 00:41:10 +000044** Implementation of the non-aggregate min() and max() functions
45*/
drhf9b596e2004-05-26 16:54:42 +000046static void minmaxFunc(
47 sqlite3_context *context,
48 int argc,
49 sqlite3_value **argv
50){
drh0bce8352002-02-28 00:41:10 +000051 int i;
drh268380c2004-02-25 13:47:31 +000052 int mask; /* 0 for min() or 0xffffffff for max() */
drhf9b596e2004-05-26 16:54:42 +000053 int iBest;
danielk1977dc1bdc42004-06-11 10:51:27 +000054 CollSeq *pColl;
drh0bce8352002-02-28 00:41:10 +000055
drh65595cd2009-02-02 16:32:55 +000056 assert( argc>1 );
drhc44af712004-09-02 15:53:56 +000057 mask = sqlite3_user_data(context)==0 ? 0 : -1;
danielk1977dc1bdc42004-06-11 10:51:27 +000058 pColl = sqlite3GetFuncCollSeq(context);
59 assert( pColl );
danielk1977c572ef72004-05-27 09:28:41 +000060 assert( mask==-1 || mask==0 );
drhf9b596e2004-05-26 16:54:42 +000061 iBest = 0;
drh9c054832004-05-31 18:51:57 +000062 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drhf9b596e2004-05-26 16:54:42 +000063 for(i=1; i<argc; i++){
drh9c054832004-05-31 18:51:57 +000064 if( sqlite3_value_type(argv[i])==SQLITE_NULL ) return;
danielk1977dc1bdc42004-06-11 10:51:27 +000065 if( (sqlite3MemCompare(argv[iBest], argv[i], pColl)^mask)>=0 ){
drh65595cd2009-02-02 16:32:55 +000066 testcase( mask==0 );
drhf9b596e2004-05-26 16:54:42 +000067 iBest = i;
drh0bce8352002-02-28 00:41:10 +000068 }
69 }
drhf4479502004-05-27 03:12:53 +000070 sqlite3_result_value(context, argv[iBest]);
drh0bce8352002-02-28 00:41:10 +000071}
drh0bce8352002-02-28 00:41:10 +000072
drh268380c2004-02-25 13:47:31 +000073/*
74** Return the type of the argument.
75*/
drhf9b596e2004-05-26 16:54:42 +000076static void typeofFunc(
77 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +000078 int NotUsed,
drhf9b596e2004-05-26 16:54:42 +000079 sqlite3_value **argv
80){
drh9d8e4012017-07-06 13:51:50 +000081 static const char *azType[] = { "integer", "real", "text", "blob", "null" };
82 int i = sqlite3_value_type(argv[0]) - 1;
danielk197762c14b32008-11-19 09:05:26 +000083 UNUSED_PARAMETER(NotUsed);
drh9d8e4012017-07-06 13:51:50 +000084 assert( i>=0 && i<ArraySize(azType) );
85 assert( SQLITE_INTEGER==1 );
86 assert( SQLITE_FLOAT==2 );
87 assert( SQLITE_TEXT==3 );
88 assert( SQLITE_BLOB==4 );
89 assert( SQLITE_NULL==5 );
drh3cef3642017-07-14 19:22:08 +000090 /* EVIDENCE-OF: R-01470-60482 The sqlite3_value_type(V) interface returns
91 ** the datatype code for the initial datatype of the sqlite3_value object
92 ** V. The returned value is one of SQLITE_INTEGER, SQLITE_FLOAT,
93 ** SQLITE_TEXT, SQLITE_BLOB, or SQLITE_NULL. */
drh9d8e4012017-07-06 13:51:50 +000094 sqlite3_result_text(context, azType[i], -1, SQLITE_STATIC);
drh0bce8352002-02-28 00:41:10 +000095}
96
drh5708d2d2005-06-22 10:53:59 +000097
98/*
drh0bce8352002-02-28 00:41:10 +000099** Implementation of the length() function
100*/
drhf9b596e2004-05-26 16:54:42 +0000101static void lengthFunc(
102 sqlite3_context *context,
103 int argc,
104 sqlite3_value **argv
105){
drh0bce8352002-02-28 00:41:10 +0000106 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000107 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +0000108 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +0000109 case SQLITE_BLOB:
110 case SQLITE_INTEGER:
111 case SQLITE_FLOAT: {
drhf4479502004-05-27 03:12:53 +0000112 sqlite3_result_int(context, sqlite3_value_bytes(argv[0]));
drhf9b596e2004-05-26 16:54:42 +0000113 break;
114 }
drh9c054832004-05-31 18:51:57 +0000115 case SQLITE_TEXT: {
drh2646da72005-12-09 20:02:05 +0000116 const unsigned char *z = sqlite3_value_text(argv[0]);
drh7ea34692018-01-23 04:22:33 +0000117 const unsigned char *z0;
118 unsigned char c;
drh7a521cf2007-04-25 18:23:52 +0000119 if( z==0 ) return;
drh7ea34692018-01-23 04:22:33 +0000120 z0 = z;
121 while( (c = *z)!=0 ){
122 z++;
123 if( c>=0xc0 ){
124 while( (*z & 0xc0)==0x80 ){ z++; z0++; }
125 }
drh4a919112007-05-15 11:55:09 +0000126 }
drh7ea34692018-01-23 04:22:33 +0000127 sqlite3_result_int(context, (int)(z-z0));
drhf9b596e2004-05-26 16:54:42 +0000128 break;
129 }
130 default: {
131 sqlite3_result_null(context);
132 break;
133 }
134 }
drh0bce8352002-02-28 00:41:10 +0000135}
136
137/*
drh2ba3ccc2009-12-08 02:06:08 +0000138** Implementation of the abs() function.
139**
140** IMP: R-23979-26855 The abs(X) function returns the absolute value of
141** the numeric argument X.
drh0bce8352002-02-28 00:41:10 +0000142*/
danielk19770ae8b832004-05-25 12:05:56 +0000143static void absFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drh0bce8352002-02-28 00:41:10 +0000144 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000145 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +0000146 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +0000147 case SQLITE_INTEGER: {
danielk1977f93bbbe2004-05-27 10:30:52 +0000148 i64 iVal = sqlite3_value_int64(argv[0]);
drh52fc8492006-02-23 21:43:55 +0000149 if( iVal<0 ){
drh693e6712014-01-24 22:58:00 +0000150 if( iVal==SMALLEST_INT64 ){
drheb091cd2013-11-09 19:47:15 +0000151 /* IMP: R-31676-45509 If X is the integer -9223372036854775808
152 ** then abs(X) throws an integer overflow error since there is no
drh2ba3ccc2009-12-08 02:06:08 +0000153 ** equivalent positive 64-bit two complement value. */
drh52fc8492006-02-23 21:43:55 +0000154 sqlite3_result_error(context, "integer overflow", -1);
155 return;
156 }
157 iVal = -iVal;
158 }
danielk1977f93bbbe2004-05-27 10:30:52 +0000159 sqlite3_result_int64(context, iVal);
drhf9b596e2004-05-26 16:54:42 +0000160 break;
161 }
drh9c054832004-05-31 18:51:57 +0000162 case SQLITE_NULL: {
drh2ba3ccc2009-12-08 02:06:08 +0000163 /* IMP: R-37434-19929 Abs(X) returns NULL if X is NULL. */
drhf9b596e2004-05-26 16:54:42 +0000164 sqlite3_result_null(context);
165 break;
166 }
167 default: {
drh2ba3ccc2009-12-08 02:06:08 +0000168 /* Because sqlite3_value_double() returns 0.0 if the argument is not
169 ** something that can be converted into a number, we have:
drh643091f2014-11-20 23:21:23 +0000170 ** IMP: R-01992-00519 Abs(X) returns 0.0 if X is a string or blob
171 ** that cannot be converted to a numeric value.
drh2ba3ccc2009-12-08 02:06:08 +0000172 */
danielk1977f93bbbe2004-05-27 10:30:52 +0000173 double rVal = sqlite3_value_double(argv[0]);
drh52fc8492006-02-23 21:43:55 +0000174 if( rVal<0 ) rVal = -rVal;
danielk1977f93bbbe2004-05-27 10:30:52 +0000175 sqlite3_result_double(context, rVal);
drhf9b596e2004-05-26 16:54:42 +0000176 break;
177 }
178 }
drh0bce8352002-02-28 00:41:10 +0000179}
180
181/*
drhd55e0722012-10-25 03:07:29 +0000182** Implementation of the instr() function.
183**
184** instr(haystack,needle) finds the first occurrence of needle
185** in haystack and returns the number of previous characters plus 1,
186** or 0 if needle does not occur within haystack.
187**
188** If both haystack and needle are BLOBs, then the result is one more than
189** the number of bytes in haystack prior to the first occurrence of needle,
190** or 0 if needle never occurs in haystack.
191*/
192static void instrFunc(
193 sqlite3_context *context,
194 int argc,
195 sqlite3_value **argv
196){
197 const unsigned char *zHaystack;
198 const unsigned char *zNeedle;
199 int nHaystack;
200 int nNeedle;
201 int typeHaystack, typeNeedle;
202 int N = 1;
203 int isText;
drhc930b402019-01-08 15:18:24 +0000204 unsigned char firstChar;
drhd55e0722012-10-25 03:07:29 +0000205
drh68c804b2012-12-04 11:03:11 +0000206 UNUSED_PARAMETER(argc);
drhd55e0722012-10-25 03:07:29 +0000207 typeHaystack = sqlite3_value_type(argv[0]);
208 typeNeedle = sqlite3_value_type(argv[1]);
209 if( typeHaystack==SQLITE_NULL || typeNeedle==SQLITE_NULL ) return;
210 nHaystack = sqlite3_value_bytes(argv[0]);
211 nNeedle = sqlite3_value_bytes(argv[1]);
dan895decf2016-12-30 14:15:56 +0000212 if( nNeedle>0 ){
213 if( typeHaystack==SQLITE_BLOB && typeNeedle==SQLITE_BLOB ){
214 zHaystack = sqlite3_value_blob(argv[0]);
215 zNeedle = sqlite3_value_blob(argv[1]);
216 isText = 0;
217 }else{
218 zHaystack = sqlite3_value_text(argv[0]);
219 zNeedle = sqlite3_value_text(argv[1]);
220 isText = 1;
dan895decf2016-12-30 14:15:56 +0000221 }
drhb30574b2017-03-16 14:28:52 +0000222 if( zNeedle==0 || (nHaystack && zHaystack==0) ) return;
drhc930b402019-01-08 15:18:24 +0000223 firstChar = zNeedle[0];
224 while( nNeedle<=nHaystack
225 && (zHaystack[0]!=firstChar || memcmp(zHaystack, zNeedle, nNeedle)!=0)
226 ){
dan895decf2016-12-30 14:15:56 +0000227 N++;
228 do{
229 nHaystack--;
230 zHaystack++;
231 }while( isText && (zHaystack[0]&0xc0)==0x80 );
232 }
233 if( nNeedle>nHaystack ) N = 0;
drhd55e0722012-10-25 03:07:29 +0000234 }
drhd55e0722012-10-25 03:07:29 +0000235 sqlite3_result_int(context, N);
236}
237
238/*
drha5c14162013-12-17 15:03:06 +0000239** Implementation of the printf() function.
240*/
241static void printfFunc(
242 sqlite3_context *context,
243 int argc,
244 sqlite3_value **argv
245){
246 PrintfArguments x;
247 StrAccum str;
248 const char *zFormat;
249 int n;
drhc0490572015-05-02 11:45:53 +0000250 sqlite3 *db = sqlite3_context_db_handle(context);
drha5c14162013-12-17 15:03:06 +0000251
252 if( argc>=1 && (zFormat = (const char*)sqlite3_value_text(argv[0]))!=0 ){
253 x.nArg = argc-1;
254 x.nUsed = 0;
255 x.apArg = argv+1;
drhc0490572015-05-02 11:45:53 +0000256 sqlite3StrAccumInit(&str, db, 0, 0, db->aLimit[SQLITE_LIMIT_LENGTH]);
drh5f4a6862016-01-30 12:50:25 +0000257 str.printfFlags = SQLITE_PRINTF_SQLFUNC;
drh0cdbe1a2018-05-09 13:46:26 +0000258 sqlite3_str_appendf(&str, zFormat, &x);
drha5c14162013-12-17 15:03:06 +0000259 n = str.nChar;
260 sqlite3_result_text(context, sqlite3StrAccumFinish(&str), n,
261 SQLITE_DYNAMIC);
262 }
263}
264
265/*
drhf764e6f2007-05-15 01:13:47 +0000266** Implementation of the substr() function.
267**
268** substr(x,p1,p2) returns p2 characters of x[] beginning with p1.
269** p1 is 1-indexed. So substr(x,1,1) returns the first character
270** of x. If x is text, then we actually count UTF-8 characters.
271** If x is a blob, then we count bytes.
272**
273** If p1 is negative, then we begin abs(p1) from the end of x[].
shaneh779b8f12009-11-12 05:04:50 +0000274**
drhf7b54962013-05-28 12:11:54 +0000275** If p2 is negative, return the p2 characters preceding p1.
drh0bce8352002-02-28 00:41:10 +0000276*/
drhf9b596e2004-05-26 16:54:42 +0000277static void substrFunc(
278 sqlite3_context *context,
279 int argc,
280 sqlite3_value **argv
281){
drh2646da72005-12-09 20:02:05 +0000282 const unsigned char *z;
283 const unsigned char *z2;
drh023ae032007-05-08 12:12:16 +0000284 int len;
drhf764e6f2007-05-15 01:13:47 +0000285 int p0type;
drh023ae032007-05-08 12:12:16 +0000286 i64 p1, p2;
drh65595cd2009-02-02 16:32:55 +0000287 int negP2 = 0;
drhf9b596e2004-05-26 16:54:42 +0000288
drh64f31512007-10-12 19:11:55 +0000289 assert( argc==3 || argc==2 );
drh8198d252009-02-01 19:42:37 +0000290 if( sqlite3_value_type(argv[1])==SQLITE_NULL
291 || (argc==3 && sqlite3_value_type(argv[2])==SQLITE_NULL)
292 ){
293 return;
294 }
drhf764e6f2007-05-15 01:13:47 +0000295 p0type = sqlite3_value_type(argv[0]);
drh4adc4cb2009-11-11 20:53:31 +0000296 p1 = sqlite3_value_int(argv[1]);
drhf764e6f2007-05-15 01:13:47 +0000297 if( p0type==SQLITE_BLOB ){
298 len = sqlite3_value_bytes(argv[0]);
299 z = sqlite3_value_blob(argv[0]);
300 if( z==0 ) return;
drh1f0feef2007-05-15 13:27:07 +0000301 assert( len==sqlite3_value_bytes(argv[0]) );
drhf764e6f2007-05-15 01:13:47 +0000302 }else{
303 z = sqlite3_value_text(argv[0]);
304 if( z==0 ) return;
drh4a919112007-05-15 11:55:09 +0000305 len = 0;
drh4adc4cb2009-11-11 20:53:31 +0000306 if( p1<0 ){
307 for(z2=z; *z2; len++){
308 SQLITE_SKIP_UTF8(z2);
309 }
drh4a919112007-05-15 11:55:09 +0000310 }
drhf764e6f2007-05-15 01:13:47 +0000311 }
drh883ad042015-02-19 00:29:11 +0000312#ifdef SQLITE_SUBSTR_COMPATIBILITY
313 /* If SUBSTR_COMPATIBILITY is defined then substr(X,0,N) work the same as
314 ** as substr(X,1,N) - it returns the first N characters of X. This
315 ** is essentially a back-out of the bug-fix in check-in [5fc125d362df4b8]
316 ** from 2009-02-02 for compatibility of applications that exploited the
317 ** old buggy behavior. */
318 if( p1==0 ) p1 = 1; /* <rdar://problem/6778339> */
319#endif
drh64f31512007-10-12 19:11:55 +0000320 if( argc==3 ){
321 p2 = sqlite3_value_int(argv[2]);
drh65595cd2009-02-02 16:32:55 +0000322 if( p2<0 ){
323 p2 = -p2;
324 negP2 = 1;
325 }
drh64f31512007-10-12 19:11:55 +0000326 }else{
drhbb4957f2008-03-20 14:03:29 +0000327 p2 = sqlite3_context_db_handle(context)->aLimit[SQLITE_LIMIT_LENGTH];
drh64f31512007-10-12 19:11:55 +0000328 }
drh0bce8352002-02-28 00:41:10 +0000329 if( p1<0 ){
drh89425d52002-02-28 03:04:48 +0000330 p1 += len;
drh653bc752002-02-28 03:31:10 +0000331 if( p1<0 ){
332 p2 += p1;
drh65595cd2009-02-02 16:32:55 +0000333 if( p2<0 ) p2 = 0;
drh653bc752002-02-28 03:31:10 +0000334 p1 = 0;
335 }
drh0bce8352002-02-28 00:41:10 +0000336 }else if( p1>0 ){
337 p1--;
drh65595cd2009-02-02 16:32:55 +0000338 }else if( p2>0 ){
339 p2--;
drh0bce8352002-02-28 00:41:10 +0000340 }
drh65595cd2009-02-02 16:32:55 +0000341 if( negP2 ){
342 p1 -= p2;
drh4e79c592009-02-01 19:23:32 +0000343 if( p1<0 ){
344 p2 += p1;
345 p1 = 0;
346 }
347 }
drh65595cd2009-02-02 16:32:55 +0000348 assert( p1>=0 && p2>=0 );
drhf764e6f2007-05-15 01:13:47 +0000349 if( p0type!=SQLITE_BLOB ){
drh4a919112007-05-15 11:55:09 +0000350 while( *z && p1 ){
351 SQLITE_SKIP_UTF8(z);
352 p1--;
drhf764e6f2007-05-15 01:13:47 +0000353 }
drh4a919112007-05-15 11:55:09 +0000354 for(z2=z; *z2 && p2; p2--){
355 SQLITE_SKIP_UTF8(z2);
drhf764e6f2007-05-15 01:13:47 +0000356 }
drhbbf483f2014-09-09 20:30:24 +0000357 sqlite3_result_text64(context, (char*)z, z2-z, SQLITE_TRANSIENT,
358 SQLITE_UTF8);
drhf764e6f2007-05-15 01:13:47 +0000359 }else{
drh4adc4cb2009-11-11 20:53:31 +0000360 if( p1+p2>len ){
361 p2 = len-p1;
362 if( p2<0 ) p2 = 0;
363 }
drhbbf483f2014-09-09 20:30:24 +0000364 sqlite3_result_blob64(context, (char*)&z[p1], (u64)p2, SQLITE_TRANSIENT);
drh0bce8352002-02-28 00:41:10 +0000365 }
drh0bce8352002-02-28 00:41:10 +0000366}
367
368/*
369** Implementation of the round() function
370*/
shanefbd60f82009-02-04 03:59:25 +0000371#ifndef SQLITE_OMIT_FLOATING_POINT
danielk19770ae8b832004-05-25 12:05:56 +0000372static void roundFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
danielk197751ad0ec2004-05-24 12:39:02 +0000373 int n = 0;
drh0bce8352002-02-28 00:41:10 +0000374 double r;
drh50d654d2009-06-03 01:24:54 +0000375 char *zBuf;
drh0bce8352002-02-28 00:41:10 +0000376 assert( argc==1 || argc==2 );
danielk197751ad0ec2004-05-24 12:39:02 +0000377 if( argc==2 ){
drh9c054832004-05-31 18:51:57 +0000378 if( SQLITE_NULL==sqlite3_value_type(argv[1]) ) return;
danielk197751ad0ec2004-05-24 12:39:02 +0000379 n = sqlite3_value_int(argv[1]);
380 if( n>30 ) n = 30;
381 if( n<0 ) n = 0;
382 }
drhd589a922006-03-02 03:02:48 +0000383 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drh4f26d6c2004-05-26 23:25:30 +0000384 r = sqlite3_value_double(argv[0]);
shaneh147e1762010-02-17 04:19:27 +0000385 /* If Y==0 and X will fit in a 64-bit int,
386 ** handle the rounding directly,
387 ** otherwise use printf.
388 */
389 if( n==0 && r>=0 && r<LARGEST_INT64-1 ){
390 r = (double)((sqlite_int64)(r+0.5));
391 }else if( n==0 && r<0 && (-r)<LARGEST_INT64-1 ){
392 r = -(double)((sqlite_int64)((-r)+0.5));
drh50d654d2009-06-03 01:24:54 +0000393 }else{
shaneh147e1762010-02-17 04:19:27 +0000394 zBuf = sqlite3_mprintf("%.*f",n,r);
395 if( zBuf==0 ){
396 sqlite3_result_error_nomem(context);
397 return;
398 }
drh9339da12010-09-30 00:50:49 +0000399 sqlite3AtoF(zBuf, &r, sqlite3Strlen30(zBuf), SQLITE_UTF8);
drh50d654d2009-06-03 01:24:54 +0000400 sqlite3_free(zBuf);
drh50d654d2009-06-03 01:24:54 +0000401 }
shaneh147e1762010-02-17 04:19:27 +0000402 sqlite3_result_double(context, r);
drh0bce8352002-02-28 00:41:10 +0000403}
shanefbd60f82009-02-04 03:59:25 +0000404#endif
drhdc04c582002-02-24 01:55:15 +0000405
danielk197726783a52007-08-29 14:06:22 +0000406/*
drhf3cdcdc2015-04-29 16:50:28 +0000407** Allocate nByte bytes of space using sqlite3Malloc(). If the
danielk197726783a52007-08-29 14:06:22 +0000408** allocation fails, call sqlite3_result_error_nomem() to notify
drh27e62db2009-04-02 10:16:17 +0000409** the database handle that malloc() has failed and return NULL.
410** If nByte is larger than the maximum string or blob length, then
411** raise an SQLITE_TOOBIG exception and return NULL.
danielk197726783a52007-08-29 14:06:22 +0000412*/
drhb1a6c3c2008-03-20 16:30:17 +0000413static void *contextMalloc(sqlite3_context *context, i64 nByte){
drhbb4957f2008-03-20 14:03:29 +0000414 char *z;
drh27e62db2009-04-02 10:16:17 +0000415 sqlite3 *db = sqlite3_context_db_handle(context);
drhef31c6a2009-04-02 09:07:12 +0000416 assert( nByte>0 );
drh27e62db2009-04-02 10:16:17 +0000417 testcase( nByte==db->aLimit[SQLITE_LIMIT_LENGTH] );
418 testcase( nByte==db->aLimit[SQLITE_LIMIT_LENGTH]+1 );
419 if( nByte>db->aLimit[SQLITE_LIMIT_LENGTH] ){
drhbb4957f2008-03-20 14:03:29 +0000420 sqlite3_result_error_toobig(context);
421 z = 0;
422 }else{
drhda4ca9d2014-09-09 17:27:35 +0000423 z = sqlite3Malloc(nByte);
drhef31c6a2009-04-02 09:07:12 +0000424 if( !z ){
drhbb4957f2008-03-20 14:03:29 +0000425 sqlite3_result_error_nomem(context);
426 }
danielk1977a1644fd2007-08-29 12:31:25 +0000427 }
428 return z;
429}
430
drhdc04c582002-02-24 01:55:15 +0000431/*
432** Implementation of the upper() and lower() SQL functions.
433*/
danielk19770ae8b832004-05-25 12:05:56 +0000434static void upperFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drh7a521cf2007-04-25 18:23:52 +0000435 char *z1;
436 const char *z2;
drh9310ef22007-04-27 17:16:20 +0000437 int i, n;
drh1d34fde2009-02-03 15:50:33 +0000438 UNUSED_PARAMETER(argc);
drh7a521cf2007-04-25 18:23:52 +0000439 z2 = (char*)sqlite3_value_text(argv[0]);
drh1f0feef2007-05-15 13:27:07 +0000440 n = sqlite3_value_bytes(argv[0]);
441 /* Verify that the call to _bytes() does not invalidate the _text() pointer */
442 assert( z2==(char*)sqlite3_value_text(argv[0]) );
drh7a521cf2007-04-25 18:23:52 +0000443 if( z2 ){
drhb1a6c3c2008-03-20 16:30:17 +0000444 z1 = contextMalloc(context, ((i64)n)+1);
drh7a521cf2007-04-25 18:23:52 +0000445 if( z1 ){
drhdf901d32011-10-13 18:00:11 +0000446 for(i=0; i<n; i++){
447 z1[i] = (char)sqlite3Toupper(z2[i]);
drh7a521cf2007-04-25 18:23:52 +0000448 }
drhdf901d32011-10-13 18:00:11 +0000449 sqlite3_result_text(context, z1, n, sqlite3_free);
drh7a521cf2007-04-25 18:23:52 +0000450 }
drhdc04c582002-02-24 01:55:15 +0000451 }
452}
danielk19770ae8b832004-05-25 12:05:56 +0000453static void lowerFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drhdf901d32011-10-13 18:00:11 +0000454 char *z1;
drh7a521cf2007-04-25 18:23:52 +0000455 const char *z2;
drh9310ef22007-04-27 17:16:20 +0000456 int i, n;
drh1d34fde2009-02-03 15:50:33 +0000457 UNUSED_PARAMETER(argc);
drh7a521cf2007-04-25 18:23:52 +0000458 z2 = (char*)sqlite3_value_text(argv[0]);
drh1f0feef2007-05-15 13:27:07 +0000459 n = sqlite3_value_bytes(argv[0]);
460 /* Verify that the call to _bytes() does not invalidate the _text() pointer */
461 assert( z2==(char*)sqlite3_value_text(argv[0]) );
drh7a521cf2007-04-25 18:23:52 +0000462 if( z2 ){
drhb1a6c3c2008-03-20 16:30:17 +0000463 z1 = contextMalloc(context, ((i64)n)+1);
drh7a521cf2007-04-25 18:23:52 +0000464 if( z1 ){
drhdf901d32011-10-13 18:00:11 +0000465 for(i=0; i<n; i++){
466 z1[i] = sqlite3Tolower(z2[i]);
drh7a521cf2007-04-25 18:23:52 +0000467 }
drhdf901d32011-10-13 18:00:11 +0000468 sqlite3_result_text(context, z1, n, sqlite3_free);
drh7a521cf2007-04-25 18:23:52 +0000469 }
drhdc04c582002-02-24 01:55:15 +0000470 }
471}
472
drhae6bb952009-11-11 00:24:31 +0000473/*
drhcca9f3d2013-09-06 15:23:29 +0000474** Some functions like COALESCE() and IFNULL() and UNLIKELY() are implemented
475** as VDBE code so that unused argument values do not have to be computed.
476** However, we still need some kind of function implementation for this
477** routines in the function table. The noopFunc macro provides this.
478** noopFunc will never be called so it doesn't matter what the implementation
479** is. We might as well use the "version()" function as a substitute.
drhae6bb952009-11-11 00:24:31 +0000480*/
drhcca9f3d2013-09-06 15:23:29 +0000481#define noopFunc versionFunc /* Substitute function - never called */
drh3212e182002-02-28 00:46:26 +0000482
483/*
drhf9ffac92002-03-02 19:00:31 +0000484** Implementation of random(). Return a random integer.
485*/
drhf9b596e2004-05-26 16:54:42 +0000486static void randomFunc(
487 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000488 int NotUsed,
489 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000490){
drh52fc8492006-02-23 21:43:55 +0000491 sqlite_int64 r;
danielk197762c14b32008-11-19 09:05:26 +0000492 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drh2fa18682008-03-19 14:15:34 +0000493 sqlite3_randomness(sizeof(r), &r);
drh3034e3d2009-04-02 14:05:21 +0000494 if( r<0 ){
495 /* We need to prevent a random number of 0x8000000000000000
496 ** (or -9223372036854775808) since when you do abs() of that
497 ** number of you get the same value back again. To do this
498 ** in a way that is testable, mask the sign bit off of negative
499 ** values, resulting in a positive value. Then take the
500 ** 2s complement of that positive value. The end result can
501 ** therefore be no less than -9223372036854775807.
502 */
drhaf8001b2012-02-11 19:53:24 +0000503 r = -(r & LARGEST_INT64);
drh3034e3d2009-04-02 14:05:21 +0000504 }
drh52fc8492006-02-23 21:43:55 +0000505 sqlite3_result_int64(context, r);
drhf9ffac92002-03-02 19:00:31 +0000506}
507
508/*
drh137c7282007-01-29 17:58:28 +0000509** Implementation of randomblob(N). Return a random blob
510** that is N bytes long.
drh63cf66f2007-01-29 15:50:05 +0000511*/
drh137c7282007-01-29 17:58:28 +0000512static void randomBlob(
drh63cf66f2007-01-29 15:50:05 +0000513 sqlite3_context *context,
514 int argc,
515 sqlite3_value **argv
516){
drh3cb79202019-01-18 14:53:15 +0000517 sqlite3_int64 n;
drh137c7282007-01-29 17:58:28 +0000518 unsigned char *p;
drh63cf66f2007-01-29 15:50:05 +0000519 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000520 UNUSED_PARAMETER(argc);
drh3cb79202019-01-18 14:53:15 +0000521 n = sqlite3_value_int64(argv[0]);
drh023ae032007-05-08 12:12:16 +0000522 if( n<1 ){
523 n = 1;
524 }
danielk1977a1644fd2007-08-29 12:31:25 +0000525 p = contextMalloc(context, n);
drh02d85832007-05-07 19:31:15 +0000526 if( p ){
drh2fa18682008-03-19 14:15:34 +0000527 sqlite3_randomness(n, p);
drh17435752007-08-16 04:30:38 +0000528 sqlite3_result_blob(context, (char*)p, n, sqlite3_free);
drh02d85832007-05-07 19:31:15 +0000529 }
drh63cf66f2007-01-29 15:50:05 +0000530}
531
532/*
drh6ed41ad2002-04-06 14:10:47 +0000533** Implementation of the last_insert_rowid() SQL function. The return
danielk197724b03fd2004-05-10 10:34:34 +0000534** value is the same as the sqlite3_last_insert_rowid() API function.
drh6ed41ad2002-04-06 14:10:47 +0000535*/
danielk197751ad0ec2004-05-24 12:39:02 +0000536static void last_insert_rowid(
danielk19770ae8b832004-05-25 12:05:56 +0000537 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000538 int NotUsed,
539 sqlite3_value **NotUsed2
danielk197751ad0ec2004-05-24 12:39:02 +0000540){
drhfa4a4b92008-03-19 21:45:51 +0000541 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000542 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000543 /* IMP: R-51513-12026 The last_insert_rowid() SQL function is a
544 ** wrapper around the sqlite3_last_insert_rowid() C/C++ interface
545 ** function. */
drhf9b596e2004-05-26 16:54:42 +0000546 sqlite3_result_int64(context, sqlite3_last_insert_rowid(db));
drh6ed41ad2002-04-06 14:10:47 +0000547}
548
rdcf146a772004-02-25 22:51:06 +0000549/*
drhab2f1f92010-01-11 18:26:42 +0000550** Implementation of the changes() SQL function.
551**
552** IMP: R-62073-11209 The changes() SQL function is a wrapper
553** around the sqlite3_changes() C/C++ function and hence follows the same
554** rules for counting changes.
rdcf146a772004-02-25 22:51:06 +0000555*/
danielk1977b28af712004-06-21 06:50:26 +0000556static void changes(
drhf9b596e2004-05-26 16:54:42 +0000557 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000558 int NotUsed,
559 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000560){
drhfa4a4b92008-03-19 21:45:51 +0000561 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000562 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhf4479502004-05-27 03:12:53 +0000563 sqlite3_result_int(context, sqlite3_changes(db));
rdcb0c374f2004-02-20 22:53:38 +0000564}
rdcf146a772004-02-25 22:51:06 +0000565
566/*
danielk1977b28af712004-06-21 06:50:26 +0000567** Implementation of the total_changes() SQL function. The return value is
568** the same as the sqlite3_total_changes() API function.
rdcf146a772004-02-25 22:51:06 +0000569*/
danielk1977b28af712004-06-21 06:50:26 +0000570static void total_changes(
571 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000572 int NotUsed,
573 sqlite3_value **NotUsed2
danielk197751ad0ec2004-05-24 12:39:02 +0000574){
drhfa4a4b92008-03-19 21:45:51 +0000575 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000576 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000577 /* IMP: R-52756-41993 This function is a wrapper around the
578 ** sqlite3_total_changes() C/C++ interface. */
danielk1977b28af712004-06-21 06:50:26 +0000579 sqlite3_result_int(context, sqlite3_total_changes(db));
rdcb0c374f2004-02-20 22:53:38 +0000580}
581
drh6ed41ad2002-04-06 14:10:47 +0000582/*
drh4e5ffc52004-08-31 00:52:37 +0000583** A structure defining how to do GLOB-style comparisons.
danielk1977d02eb1f2004-06-06 09:44:03 +0000584*/
drh4e5ffc52004-08-31 00:52:37 +0000585struct compareInfo {
drh07e83472016-01-11 03:48:18 +0000586 u8 matchAll; /* "*" or "%" */
587 u8 matchOne; /* "?" or "_" */
588 u8 matchSet; /* "[" or 0 */
589 u8 noCase; /* true to ignore case differences */
danielk1977d02eb1f2004-06-06 09:44:03 +0000590};
drh55ef4d92005-08-14 01:20:37 +0000591
drhb9175ae2007-12-07 18:39:04 +0000592/*
593** For LIKE and GLOB matching on EBCDIC machines, assume that every
drhb0870482015-06-17 13:20:54 +0000594** character is exactly one byte in size. Also, provde the Utf8Read()
595** macro for fast reading of the next character in the common case where
596** the next character is ASCII.
drhb9175ae2007-12-07 18:39:04 +0000597*/
598#if defined(SQLITE_EBCDIC)
drh88b33222014-09-25 03:51:37 +0000599# define sqlite3Utf8Read(A) (*((*A)++))
drhb0870482015-06-17 13:20:54 +0000600# define Utf8Read(A) (*(A++))
drhb9175ae2007-12-07 18:39:04 +0000601#else
drhb0870482015-06-17 13:20:54 +0000602# define Utf8Read(A) (A[0]<0x80?*(A++):sqlite3Utf8Read(&A))
drhb9175ae2007-12-07 18:39:04 +0000603#endif
604
drh4e5ffc52004-08-31 00:52:37 +0000605static const struct compareInfo globInfo = { '*', '?', '[', 0 };
drh55ef4d92005-08-14 01:20:37 +0000606/* The correct SQL-92 behavior is for the LIKE operator to ignore
607** case. Thus 'a' LIKE 'A' would be true. */
608static const struct compareInfo likeInfoNorm = { '%', '_', 0, 1 };
609/* If SQLITE_CASE_SENSITIVE_LIKE is defined, then the LIKE operator
610** is case sensitive causing 'a' LIKE 'A' to be false */
611static const struct compareInfo likeInfoAlt = { '%', '_', 0, 0 };
danielk1977d02eb1f2004-06-06 09:44:03 +0000612
613/*
drh698a01c2016-12-01 18:49:40 +0000614** Possible error returns from patternMatch()
615*/
616#define SQLITE_MATCH 0
617#define SQLITE_NOMATCH 1
618#define SQLITE_NOWILDCARDMATCH 2
619
620/*
621** Compare two UTF-8 strings for equality where the first string is
622** a GLOB or LIKE expression. Return values:
623**
624** SQLITE_MATCH: Match
625** SQLITE_NOMATCH: No match
626** SQLITE_NOWILDCARDMATCH: No match in spite of having * or % wildcards.
drh0ac65892002-04-20 14:24:41 +0000627**
drh4e5ffc52004-08-31 00:52:37 +0000628** Globbing rules:
drh0ac65892002-04-20 14:24:41 +0000629**
drh4e5ffc52004-08-31 00:52:37 +0000630** '*' Matches any sequence of zero or more characters.
danielk1977d02eb1f2004-06-06 09:44:03 +0000631**
drh4e5ffc52004-08-31 00:52:37 +0000632** '?' Matches exactly one character.
633**
634** [...] Matches one character from the enclosed list of
635** characters.
636**
637** [^...] Matches one character not in the enclosed list.
638**
639** With the [...] and [^...] matching, a ']' character can be included
640** in the list by making it the first character after '[' or '^'. A
641** range of characters can be specified using '-'. Example:
642** "[a-z]" matches any single lower-case letter. To match a '-', make
643** it the last character in the list.
644**
drh9fdfdc82014-09-25 11:08:57 +0000645** Like matching rules:
646**
647** '%' Matches any sequence of zero or more characters
648**
649*** '_' Matches any one character
650**
651** Ec Where E is the "esc" character and c is any other
652** character, including '%', '_', and esc, match exactly c.
653**
drhb0870482015-06-17 13:20:54 +0000654** The comments within this routine usually assume glob matching.
drh9fdfdc82014-09-25 11:08:57 +0000655**
drh4e5ffc52004-08-31 00:52:37 +0000656** This routine is usually quick, but can be N**2 in the worst case.
drh0ac65892002-04-20 14:24:41 +0000657*/
danielk19777c6303c2004-11-17 16:41:29 +0000658static int patternCompare(
drh4e5ffc52004-08-31 00:52:37 +0000659 const u8 *zPattern, /* The glob pattern */
660 const u8 *zString, /* The string to compare against the glob */
danielk19777c6303c2004-11-17 16:41:29 +0000661 const struct compareInfo *pInfo, /* Information about how to do the compare */
drh698a01c2016-12-01 18:49:40 +0000662 u32 matchOther /* The escape char (LIKE) or '[' (GLOB) */
danielk197751ad0ec2004-05-24 12:39:02 +0000663){
drh9fdfdc82014-09-25 11:08:57 +0000664 u32 c, c2; /* Next pattern and input string chars */
665 u32 matchOne = pInfo->matchOne; /* "?" or "_" */
666 u32 matchAll = pInfo->matchAll; /* "*" or "%" */
drh9fdfdc82014-09-25 11:08:57 +0000667 u8 noCase = pInfo->noCase; /* True if uppercase==lowercase */
668 const u8 *zEscaped = 0; /* One past the last escaped input char */
drh88b33222014-09-25 03:51:37 +0000669
drhb0870482015-06-17 13:20:54 +0000670 while( (c = Utf8Read(zPattern))!=0 ){
drh9fdfdc82014-09-25 11:08:57 +0000671 if( c==matchAll ){ /* Match "*" */
672 /* Skip over multiple "*" characters in the pattern. If there
673 ** are also "?" characters, skip those as well, but consume a
674 ** single character of the input string for each "?" skipped */
drhb0870482015-06-17 13:20:54 +0000675 while( (c=Utf8Read(zPattern)) == matchAll || c == matchOne ){
drh42610962012-09-17 18:56:32 +0000676 if( c==matchOne && sqlite3Utf8Read(&zString)==0 ){
drh698a01c2016-12-01 18:49:40 +0000677 return SQLITE_NOWILDCARDMATCH;
drh4e5ffc52004-08-31 00:52:37 +0000678 }
danielk1977ad7dd422004-06-06 12:41:49 +0000679 }
drh66150952007-07-23 19:12:41 +0000680 if( c==0 ){
drh698a01c2016-12-01 18:49:40 +0000681 return SQLITE_MATCH; /* "*" at the end of the pattern matches */
drh88b33222014-09-25 03:51:37 +0000682 }else if( c==matchOther ){
drh07e83472016-01-11 03:48:18 +0000683 if( pInfo->matchSet==0 ){
drh88b33222014-09-25 03:51:37 +0000684 c = sqlite3Utf8Read(&zPattern);
drh698a01c2016-12-01 18:49:40 +0000685 if( c==0 ) return SQLITE_NOWILDCARDMATCH;
drh88b33222014-09-25 03:51:37 +0000686 }else{
drh9fdfdc82014-09-25 11:08:57 +0000687 /* "[...]" immediately follows the "*". We have to do a slow
688 ** recursive search in this case, but it is an unusual case. */
drh88b33222014-09-25 03:51:37 +0000689 assert( matchOther<0x80 ); /* '[' is a single-byte character */
drh698a01c2016-12-01 18:49:40 +0000690 while( *zString ){
691 int bMatch = patternCompare(&zPattern[-1],zString,pInfo,matchOther);
692 if( bMatch!=SQLITE_NOMATCH ) return bMatch;
drh88b33222014-09-25 03:51:37 +0000693 SQLITE_SKIP_UTF8(zString);
694 }
drh698a01c2016-12-01 18:49:40 +0000695 return SQLITE_NOWILDCARDMATCH;
drh66150952007-07-23 19:12:41 +0000696 }
drh66150952007-07-23 19:12:41 +0000697 }
drh9fdfdc82014-09-25 11:08:57 +0000698
699 /* At this point variable c contains the first character of the
700 ** pattern string past the "*". Search in the input string for the
dan1a4a7372016-12-01 17:34:59 +0000701 ** first matching character and recursively continue the match from
drh9fdfdc82014-09-25 11:08:57 +0000702 ** that point.
703 **
704 ** For a case-insensitive search, set variable cx to be the same as
705 ** c but in the other case and search the input string for either
706 ** c or cx.
707 */
708 if( c<=0x80 ){
drheba21f92017-10-30 18:49:11 +0000709 char zStop[3];
drh698a01c2016-12-01 18:49:40 +0000710 int bMatch;
drh9fdfdc82014-09-25 11:08:57 +0000711 if( noCase ){
drheba21f92017-10-30 18:49:11 +0000712 zStop[0] = sqlite3Toupper(c);
713 zStop[1] = sqlite3Tolower(c);
714 zStop[2] = 0;
drh66150952007-07-23 19:12:41 +0000715 }else{
drheba21f92017-10-30 18:49:11 +0000716 zStop[0] = c;
717 zStop[1] = 0;
drh4e5ffc52004-08-31 00:52:37 +0000718 }
drheba21f92017-10-30 18:49:11 +0000719 while(1){
720 zString += strcspn((const char*)zString, zStop);
721 if( zString[0]==0 ) break;
722 zString++;
drh698a01c2016-12-01 18:49:40 +0000723 bMatch = patternCompare(zPattern,zString,pInfo,matchOther);
724 if( bMatch!=SQLITE_NOMATCH ) return bMatch;
drh9fdfdc82014-09-25 11:08:57 +0000725 }
726 }else{
drh698a01c2016-12-01 18:49:40 +0000727 int bMatch;
drhb0870482015-06-17 13:20:54 +0000728 while( (c2 = Utf8Read(zString))!=0 ){
drh9fdfdc82014-09-25 11:08:57 +0000729 if( c2!=c ) continue;
drh698a01c2016-12-01 18:49:40 +0000730 bMatch = patternCompare(zPattern,zString,pInfo,matchOther);
731 if( bMatch!=SQLITE_NOMATCH ) return bMatch;
drh9fdfdc82014-09-25 11:08:57 +0000732 }
drh66150952007-07-23 19:12:41 +0000733 }
drh698a01c2016-12-01 18:49:40 +0000734 return SQLITE_NOWILDCARDMATCH;
drh88b33222014-09-25 03:51:37 +0000735 }
drh88b33222014-09-25 03:51:37 +0000736 if( c==matchOther ){
drh07e83472016-01-11 03:48:18 +0000737 if( pInfo->matchSet==0 ){
drh88b33222014-09-25 03:51:37 +0000738 c = sqlite3Utf8Read(&zPattern);
drh698a01c2016-12-01 18:49:40 +0000739 if( c==0 ) return SQLITE_NOMATCH;
drh9fdfdc82014-09-25 11:08:57 +0000740 zEscaped = zPattern;
drh88b33222014-09-25 03:51:37 +0000741 }else{
742 u32 prior_c = 0;
drh9fdfdc82014-09-25 11:08:57 +0000743 int seen = 0;
744 int invert = 0;
drh88b33222014-09-25 03:51:37 +0000745 c = sqlite3Utf8Read(&zString);
drh698a01c2016-12-01 18:49:40 +0000746 if( c==0 ) return SQLITE_NOMATCH;
drh88b33222014-09-25 03:51:37 +0000747 c2 = sqlite3Utf8Read(&zPattern);
748 if( c2=='^' ){
749 invert = 1;
750 c2 = sqlite3Utf8Read(&zPattern);
751 }
752 if( c2==']' ){
753 if( c==']' ) seen = 1;
754 c2 = sqlite3Utf8Read(&zPattern);
755 }
756 while( c2 && c2!=']' ){
757 if( c2=='-' && zPattern[0]!=']' && zPattern[0]!=0 && prior_c>0 ){
758 c2 = sqlite3Utf8Read(&zPattern);
759 if( c>=prior_c && c<=c2 ) seen = 1;
760 prior_c = 0;
761 }else{
762 if( c==c2 ){
763 seen = 1;
764 }
765 prior_c = c2;
766 }
767 c2 = sqlite3Utf8Read(&zPattern);
768 }
769 if( c2==0 || (seen ^ invert)==0 ){
drh698a01c2016-12-01 18:49:40 +0000770 return SQLITE_NOMATCH;
drh88b33222014-09-25 03:51:37 +0000771 }
772 continue;
773 }
774 }
drhb0870482015-06-17 13:20:54 +0000775 c2 = Utf8Read(zString);
drh88b33222014-09-25 03:51:37 +0000776 if( c==c2 ) continue;
drhc80937a2016-06-06 01:48:14 +0000777 if( noCase && sqlite3Tolower(c)==sqlite3Tolower(c2) && c<0x80 && c2<0x80 ){
drh9fdfdc82014-09-25 11:08:57 +0000778 continue;
779 }
780 if( c==matchOne && zPattern!=zEscaped && c2!=0 ) continue;
drh698a01c2016-12-01 18:49:40 +0000781 return SQLITE_NOMATCH;
danielk197751ad0ec2004-05-24 12:39:02 +0000782 }
drh698a01c2016-12-01 18:49:40 +0000783 return *zString==0 ? SQLITE_MATCH : SQLITE_NOMATCH;
drh0ac65892002-04-20 14:24:41 +0000784}
drh4e5ffc52004-08-31 00:52:37 +0000785
drh55ef4d92005-08-14 01:20:37 +0000786/*
drh698a01c2016-12-01 18:49:40 +0000787** The sqlite3_strglob() interface. Return 0 on a match (like strcmp()) and
788** non-zero if there is no match.
drh56282a52013-04-10 16:13:38 +0000789*/
790int sqlite3_strglob(const char *zGlobPattern, const char *zString){
drh698a01c2016-12-01 18:49:40 +0000791 return patternCompare((u8*)zGlobPattern, (u8*)zString, &globInfo, '[');
drh56282a52013-04-10 16:13:38 +0000792}
793
794/*
drh698a01c2016-12-01 18:49:40 +0000795** The sqlite3_strlike() interface. Return 0 on a match and non-zero for
796** a miss - like strcmp().
drh8b4a94a2015-11-24 21:23:59 +0000797*/
798int sqlite3_strlike(const char *zPattern, const char *zStr, unsigned int esc){
drh698a01c2016-12-01 18:49:40 +0000799 return patternCompare((u8*)zPattern, (u8*)zStr, &likeInfoNorm, esc);
drh8b4a94a2015-11-24 21:23:59 +0000800}
801
802/*
drh55ef4d92005-08-14 01:20:37 +0000803** Count the number of times that the LIKE operator (or GLOB which is
804** just a variation of LIKE) gets called. This is used for testing
805** only.
806*/
807#ifdef SQLITE_TEST
808int sqlite3_like_count = 0;
809#endif
810
danielk19773f6b0872004-06-17 05:36:44 +0000811
812/*
813** Implementation of the like() SQL function. This function implements
814** the build-in LIKE operator. The first argument to the function is the
815** pattern and the second argument is the string. So, the SQL statements:
816**
817** A LIKE B
818**
819** is implemented as like(B,A).
820**
drh55ef4d92005-08-14 01:20:37 +0000821** This same function (with a different compareInfo structure) computes
822** the GLOB operator.
danielk19773f6b0872004-06-17 05:36:44 +0000823*/
824static void likeFunc(
825 sqlite3_context *context,
826 int argc,
827 sqlite3_value **argv
828){
drhbeb818d2007-05-08 15:34:47 +0000829 const unsigned char *zA, *zB;
drh07e83472016-01-11 03:48:18 +0000830 u32 escape;
drh27e62db2009-04-02 10:16:17 +0000831 int nPat;
drhbb4957f2008-03-20 14:03:29 +0000832 sqlite3 *db = sqlite3_context_db_handle(context);
drh07e83472016-01-11 03:48:18 +0000833 struct compareInfo *pInfo = sqlite3_user_data(context);
drhbeb818d2007-05-08 15:34:47 +0000834
drh41d2e662015-12-01 21:23:07 +0000835#ifdef SQLITE_LIKE_DOESNT_MATCH_BLOBS
836 if( sqlite3_value_type(argv[0])==SQLITE_BLOB
837 || sqlite3_value_type(argv[1])==SQLITE_BLOB
838 ){
839#ifdef SQLITE_TEST
840 sqlite3_like_count++;
841#endif
842 sqlite3_result_int(context, 0);
843 return;
844 }
845#endif
drh1f0feef2007-05-15 13:27:07 +0000846
drhbeb818d2007-05-08 15:34:47 +0000847 /* Limit the length of the LIKE or GLOB pattern to avoid problems
848 ** of deep recursion and N*N behavior in patternCompare().
849 */
drh27e62db2009-04-02 10:16:17 +0000850 nPat = sqlite3_value_bytes(argv[0]);
851 testcase( nPat==db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH] );
852 testcase( nPat==db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH]+1 );
853 if( nPat > db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH] ){
drhbeb818d2007-05-08 15:34:47 +0000854 sqlite3_result_error(context, "LIKE or GLOB pattern too complex", -1);
855 return;
856 }
danielk19777c6303c2004-11-17 16:41:29 +0000857 if( argc==3 ){
858 /* The escape character string must consist of a single UTF-8 character.
859 ** Otherwise, return an error.
860 */
861 const unsigned char *zEsc = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +0000862 if( zEsc==0 ) return;
drhee858132007-05-08 20:37:38 +0000863 if( sqlite3Utf8CharLen((char*)zEsc, -1)!=1 ){
danielk19777c6303c2004-11-17 16:41:29 +0000864 sqlite3_result_error(context,
865 "ESCAPE expression must be a single character", -1);
866 return;
867 }
drh42610962012-09-17 18:56:32 +0000868 escape = sqlite3Utf8Read(&zEsc);
drh07e83472016-01-11 03:48:18 +0000869 }else{
870 escape = pInfo->matchSet;
danielk19777c6303c2004-11-17 16:41:29 +0000871 }
drhcf833232019-04-30 11:54:36 +0000872 zB = sqlite3_value_text(argv[0]);
873 zA = sqlite3_value_text(argv[1]);
danielk19773f6b0872004-06-17 05:36:44 +0000874 if( zA && zB ){
drh55ef4d92005-08-14 01:20:37 +0000875#ifdef SQLITE_TEST
876 sqlite3_like_count++;
877#endif
drhf49759b2017-08-25 19:51:51 +0000878 sqlite3_result_int(context,
879 patternCompare(zB, zA, pInfo, escape)==SQLITE_MATCH);
danielk197751ad0ec2004-05-24 12:39:02 +0000880 }
drh8912d102002-05-26 21:34:58 +0000881}
882
883/*
884** Implementation of the NULLIF(x,y) function. The result is the first
885** argument if the arguments are different. The result is NULL if the
886** arguments are equal to each other.
887*/
drhf9b596e2004-05-26 16:54:42 +0000888static void nullifFunc(
889 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000890 int NotUsed,
drhf9b596e2004-05-26 16:54:42 +0000891 sqlite3_value **argv
892){
danielk1977dc1bdc42004-06-11 10:51:27 +0000893 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk197762c14b32008-11-19 09:05:26 +0000894 UNUSED_PARAMETER(NotUsed);
danielk1977dc1bdc42004-06-11 10:51:27 +0000895 if( sqlite3MemCompare(argv[0], argv[1], pColl)!=0 ){
drhf4479502004-05-27 03:12:53 +0000896 sqlite3_result_value(context, argv[0]);
drh8912d102002-05-26 21:34:58 +0000897 }
drh0ac65892002-04-20 14:24:41 +0000898}
899
drh647cb0e2002-11-04 19:32:25 +0000900/*
drh47baebc2009-08-14 16:01:24 +0000901** Implementation of the sqlite_version() function. The result is the version
drh647cb0e2002-11-04 19:32:25 +0000902** of the SQLite library that is running.
903*/
drhf9b596e2004-05-26 16:54:42 +0000904static void versionFunc(
905 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000906 int NotUsed,
907 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000908){
danielk197762c14b32008-11-19 09:05:26 +0000909 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000910 /* IMP: R-48699-48617 This function is an SQL wrapper around the
911 ** sqlite3_libversion() C-interface. */
912 sqlite3_result_text(context, sqlite3_libversion(), -1, SQLITE_STATIC);
drh647cb0e2002-11-04 19:32:25 +0000913}
914
drh47baebc2009-08-14 16:01:24 +0000915/*
916** Implementation of the sqlite_source_id() function. The result is a string
917** that identifies the particular version of the source code used to build
918** SQLite.
919*/
920static void sourceidFunc(
921 sqlite3_context *context,
922 int NotUsed,
923 sqlite3_value **NotUsed2
924){
925 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000926 /* IMP: R-24470-31136 This function is an SQL wrapper around the
927 ** sqlite3_sourceid() C interface. */
928 sqlite3_result_text(context, sqlite3_sourceid(), -1, SQLITE_STATIC);
drh47baebc2009-08-14 16:01:24 +0000929}
930
shanehbdea6d12010-02-23 04:19:54 +0000931/*
drh3ca84ef2011-04-25 18:03:10 +0000932** Implementation of the sqlite_log() function. This is a wrapper around
933** sqlite3_log(). The return value is NULL. The function exists purely for
934** its side-effects.
935*/
drh840561f2011-04-27 18:08:42 +0000936static void errlogFunc(
drh3ca84ef2011-04-25 18:03:10 +0000937 sqlite3_context *context,
938 int argc,
939 sqlite3_value **argv
940){
941 UNUSED_PARAMETER(argc);
942 UNUSED_PARAMETER(context);
943 sqlite3_log(sqlite3_value_int(argv[0]), "%s", sqlite3_value_text(argv[1]));
944}
945
946/*
shanehdc97a8c2010-02-23 20:08:35 +0000947** Implementation of the sqlite_compileoption_used() function.
948** The result is an integer that identifies if the compiler option
949** was used to build SQLite.
shanehbdea6d12010-02-23 04:19:54 +0000950*/
shanehdc97a8c2010-02-23 20:08:35 +0000951#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
952static void compileoptionusedFunc(
shanehbdea6d12010-02-23 04:19:54 +0000953 sqlite3_context *context,
shanehdc97a8c2010-02-23 20:08:35 +0000954 int argc,
955 sqlite3_value **argv
shanehbdea6d12010-02-23 04:19:54 +0000956){
shanehdc97a8c2010-02-23 20:08:35 +0000957 const char *zOptName;
958 assert( argc==1 );
959 UNUSED_PARAMETER(argc);
drha3e414c2010-08-03 18:06:25 +0000960 /* IMP: R-39564-36305 The sqlite_compileoption_used() SQL
961 ** function is a wrapper around the sqlite3_compileoption_used() C/C++
962 ** function.
963 */
drh264a2d42010-02-25 15:28:41 +0000964 if( (zOptName = (const char*)sqlite3_value_text(argv[0]))!=0 ){
shanehdc97a8c2010-02-23 20:08:35 +0000965 sqlite3_result_int(context, sqlite3_compileoption_used(zOptName));
shanehdc97a8c2010-02-23 20:08:35 +0000966 }
shanehbdea6d12010-02-23 04:19:54 +0000967}
shanehdc97a8c2010-02-23 20:08:35 +0000968#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
969
970/*
971** Implementation of the sqlite_compileoption_get() function.
972** The result is a string that identifies the compiler options
973** used to build SQLite.
974*/
975#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
976static void compileoptiongetFunc(
977 sqlite3_context *context,
978 int argc,
979 sqlite3_value **argv
980){
981 int n;
982 assert( argc==1 );
983 UNUSED_PARAMETER(argc);
drha3e414c2010-08-03 18:06:25 +0000984 /* IMP: R-04922-24076 The sqlite_compileoption_get() SQL function
985 ** is a wrapper around the sqlite3_compileoption_get() C/C++ function.
986 */
shanehdc97a8c2010-02-23 20:08:35 +0000987 n = sqlite3_value_int(argv[0]);
988 sqlite3_result_text(context, sqlite3_compileoption_get(n), -1, SQLITE_STATIC);
989}
990#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
shanehbdea6d12010-02-23 04:19:54 +0000991
drh137c7282007-01-29 17:58:28 +0000992/* Array for converting from half-bytes (nybbles) into ASCII hex
993** digits. */
994static const char hexdigits[] = {
995 '0', '1', '2', '3', '4', '5', '6', '7',
996 '8', '9', 'A', 'B', 'C', 'D', 'E', 'F'
997};
danielk1977d641d642004-11-18 15:44:29 +0000998
drh47394702003-08-20 01:03:33 +0000999/*
drh47394702003-08-20 01:03:33 +00001000** Implementation of the QUOTE() function. This function takes a single
1001** argument. If the argument is numeric, the return value is the same as
1002** the argument. If the argument is NULL, the return value is the string
1003** "NULL". Otherwise, the argument is enclosed in single quotes with
1004** single-quote escapes.
1005*/
danielk19770ae8b832004-05-25 12:05:56 +00001006static void quoteFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drha0df4cc2009-02-02 17:29:59 +00001007 assert( argc==1 );
drh1d34fde2009-02-03 15:50:33 +00001008 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +00001009 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +00001010 case SQLITE_FLOAT: {
drh72b3fbc2012-06-19 03:11:25 +00001011 double r1, r2;
drh72b3fbc2012-06-19 03:11:25 +00001012 char zBuf[50];
mistachkin2b434a72012-06-19 04:36:48 +00001013 r1 = sqlite3_value_double(argv[0]);
drh72b3fbc2012-06-19 03:11:25 +00001014 sqlite3_snprintf(sizeof(zBuf), zBuf, "%!.15g", r1);
1015 sqlite3AtoF(zBuf, &r2, 20, SQLITE_UTF8);
1016 if( r1!=r2 ){
1017 sqlite3_snprintf(sizeof(zBuf), zBuf, "%!.20e", r1);
1018 }
1019 sqlite3_result_text(context, zBuf, -1, SQLITE_TRANSIENT);
1020 break;
1021 }
1022 case SQLITE_INTEGER: {
drhf4479502004-05-27 03:12:53 +00001023 sqlite3_result_value(context, argv[0]);
drhf9b596e2004-05-26 16:54:42 +00001024 break;
1025 }
danielk19773f41e972004-06-08 00:39:01 +00001026 case SQLITE_BLOB: {
danielk19773f41e972004-06-08 00:39:01 +00001027 char *zText = 0;
danielk19773f41e972004-06-08 00:39:01 +00001028 char const *zBlob = sqlite3_value_blob(argv[0]);
drh1f0feef2007-05-15 13:27:07 +00001029 int nBlob = sqlite3_value_bytes(argv[0]);
1030 assert( zBlob==sqlite3_value_blob(argv[0]) ); /* No encoding change */
drhb1a6c3c2008-03-20 16:30:17 +00001031 zText = (char *)contextMalloc(context, (2*(i64)nBlob)+4);
danielk1977a1644fd2007-08-29 12:31:25 +00001032 if( zText ){
danielk19773f41e972004-06-08 00:39:01 +00001033 int i;
1034 for(i=0; i<nBlob; i++){
1035 zText[(i*2)+2] = hexdigits[(zBlob[i]>>4)&0x0F];
1036 zText[(i*2)+3] = hexdigits[(zBlob[i])&0x0F];
1037 }
1038 zText[(nBlob*2)+2] = '\'';
1039 zText[(nBlob*2)+3] = '\0';
1040 zText[0] = 'X';
1041 zText[1] = '\'';
danielk1977d8123362004-06-12 09:25:12 +00001042 sqlite3_result_text(context, zText, -1, SQLITE_TRANSIENT);
drh17435752007-08-16 04:30:38 +00001043 sqlite3_free(zText);
danielk19773f41e972004-06-08 00:39:01 +00001044 }
1045 break;
1046 }
drh9c054832004-05-31 18:51:57 +00001047 case SQLITE_TEXT: {
drh023ae032007-05-08 12:12:16 +00001048 int i,j;
1049 u64 n;
drh2646da72005-12-09 20:02:05 +00001050 const unsigned char *zArg = sqlite3_value_text(argv[0]);
drhf9b596e2004-05-26 16:54:42 +00001051 char *z;
1052
drh7a521cf2007-04-25 18:23:52 +00001053 if( zArg==0 ) return;
drh023ae032007-05-08 12:12:16 +00001054 for(i=0, n=0; zArg[i]; i++){ if( zArg[i]=='\'' ) n++; }
drhb1a6c3c2008-03-20 16:30:17 +00001055 z = contextMalloc(context, ((i64)i)+((i64)n)+3);
danielk1977a1644fd2007-08-29 12:31:25 +00001056 if( z ){
1057 z[0] = '\'';
1058 for(i=0, j=1; zArg[i]; i++){
1059 z[j++] = zArg[i];
1060 if( zArg[i]=='\'' ){
1061 z[j++] = '\'';
1062 }
drhf9b596e2004-05-26 16:54:42 +00001063 }
danielk1977a1644fd2007-08-29 12:31:25 +00001064 z[j++] = '\'';
1065 z[j] = 0;
1066 sqlite3_result_text(context, z, j, sqlite3_free);
drhf9b596e2004-05-26 16:54:42 +00001067 }
drha0df4cc2009-02-02 17:29:59 +00001068 break;
1069 }
1070 default: {
1071 assert( sqlite3_value_type(argv[0])==SQLITE_NULL );
1072 sqlite3_result_text(context, "NULL", 4, SQLITE_STATIC);
1073 break;
drhf9b596e2004-05-26 16:54:42 +00001074 }
drh47394702003-08-20 01:03:33 +00001075 }
1076}
1077
drh137c7282007-01-29 17:58:28 +00001078/*
drhd495d8c2013-02-22 19:34:25 +00001079** The unicode() function. Return the integer unicode code-point value
1080** for the first character of the input string.
1081*/
1082static void unicodeFunc(
1083 sqlite3_context *context,
1084 int argc,
1085 sqlite3_value **argv
1086){
1087 const unsigned char *z = sqlite3_value_text(argv[0]);
drh1d59d032013-03-01 23:40:26 +00001088 (void)argc;
drhd495d8c2013-02-22 19:34:25 +00001089 if( z && z[0] ) sqlite3_result_int(context, sqlite3Utf8Read(&z));
1090}
1091
1092/*
1093** The char() function takes zero or more arguments, each of which is
1094** an integer. It constructs a string where each character of the string
1095** is the unicode character for the corresponding integer argument.
1096*/
1097static void charFunc(
1098 sqlite3_context *context,
1099 int argc,
1100 sqlite3_value **argv
1101){
1102 unsigned char *z, *zOut;
1103 int i;
drhf3cdcdc2015-04-29 16:50:28 +00001104 zOut = z = sqlite3_malloc64( argc*4+1 );
drhd495d8c2013-02-22 19:34:25 +00001105 if( z==0 ){
1106 sqlite3_result_error_nomem(context);
1107 return;
1108 }
1109 for(i=0; i<argc; i++){
mistachkinc9545442013-02-26 05:42:30 +00001110 sqlite3_int64 x;
drhd495d8c2013-02-22 19:34:25 +00001111 unsigned c;
1112 x = sqlite3_value_int64(argv[i]);
1113 if( x<0 || x>0x10ffff ) x = 0xfffd;
1114 c = (unsigned)(x & 0x1fffff);
drhfe7a5d12013-03-07 14:00:04 +00001115 if( c<0x00080 ){
1116 *zOut++ = (u8)(c&0xFF);
1117 }else if( c<0x00800 ){
1118 *zOut++ = 0xC0 + (u8)((c>>6)&0x1F);
1119 *zOut++ = 0x80 + (u8)(c & 0x3F);
1120 }else if( c<0x10000 ){
1121 *zOut++ = 0xE0 + (u8)((c>>12)&0x0F);
1122 *zOut++ = 0x80 + (u8)((c>>6) & 0x3F);
1123 *zOut++ = 0x80 + (u8)(c & 0x3F);
drhd495d8c2013-02-22 19:34:25 +00001124 }else{
drhfe7a5d12013-03-07 14:00:04 +00001125 *zOut++ = 0xF0 + (u8)((c>>18) & 0x07);
1126 *zOut++ = 0x80 + (u8)((c>>12) & 0x3F);
1127 *zOut++ = 0x80 + (u8)((c>>6) & 0x3F);
1128 *zOut++ = 0x80 + (u8)(c & 0x3F);
1129 } \
drhd495d8c2013-02-22 19:34:25 +00001130 }
drhbbf483f2014-09-09 20:30:24 +00001131 sqlite3_result_text64(context, (char*)z, zOut-z, sqlite3_free, SQLITE_UTF8);
drhd495d8c2013-02-22 19:34:25 +00001132}
1133
1134/*
drh137c7282007-01-29 17:58:28 +00001135** The hex() function. Interpret the argument as a blob. Return
1136** a hexadecimal rendering as text.
1137*/
1138static void hexFunc(
1139 sqlite3_context *context,
1140 int argc,
1141 sqlite3_value **argv
1142){
1143 int i, n;
1144 const unsigned char *pBlob;
1145 char *zHex, *z;
1146 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001147 UNUSED_PARAMETER(argc);
drh1f0feef2007-05-15 13:27:07 +00001148 pBlob = sqlite3_value_blob(argv[0]);
drh137c7282007-01-29 17:58:28 +00001149 n = sqlite3_value_bytes(argv[0]);
drh1f0feef2007-05-15 13:27:07 +00001150 assert( pBlob==sqlite3_value_blob(argv[0]) ); /* No encoding change */
drhb1a6c3c2008-03-20 16:30:17 +00001151 z = zHex = contextMalloc(context, ((i64)n)*2 + 1);
danielk1977a1644fd2007-08-29 12:31:25 +00001152 if( zHex ){
1153 for(i=0; i<n; i++, pBlob++){
1154 unsigned char c = *pBlob;
1155 *(z++) = hexdigits[(c>>4)&0xf];
1156 *(z++) = hexdigits[c&0xf];
1157 }
1158 *z = 0;
1159 sqlite3_result_text(context, zHex, n*2, sqlite3_free);
drh137c7282007-01-29 17:58:28 +00001160 }
drh137c7282007-01-29 17:58:28 +00001161}
1162
drh26b6d902007-03-17 13:27:54 +00001163/*
drh8cff3822007-05-02 02:08:28 +00001164** The zeroblob(N) function returns a zero-filled blob of size N bytes.
1165*/
1166static void zeroblobFunc(
1167 sqlite3_context *context,
1168 int argc,
1169 sqlite3_value **argv
1170){
drh98640a32007-06-07 19:08:32 +00001171 i64 n;
dana4d5ae82015-07-24 16:24:37 +00001172 int rc;
drh8cff3822007-05-02 02:08:28 +00001173 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001174 UNUSED_PARAMETER(argc);
drh98640a32007-06-07 19:08:32 +00001175 n = sqlite3_value_int64(argv[0]);
dana4d5ae82015-07-24 16:24:37 +00001176 if( n<0 ) n = 0;
1177 rc = sqlite3_result_zeroblob64(context, n); /* IMP: R-00293-64994 */
1178 if( rc ){
1179 sqlite3_result_error_code(context, rc);
drh98640a32007-06-07 19:08:32 +00001180 }
drh8cff3822007-05-02 02:08:28 +00001181}
1182
1183/*
drh26b6d902007-03-17 13:27:54 +00001184** The replace() function. Three arguments are all strings: call
1185** them A, B, and C. The result is also a string which is derived
drhf7b54962013-05-28 12:11:54 +00001186** from A by replacing every occurrence of B with C. The match
drh26b6d902007-03-17 13:27:54 +00001187** must be exact. Collating sequences are not used.
1188*/
1189static void replaceFunc(
1190 sqlite3_context *context,
1191 int argc,
1192 sqlite3_value **argv
1193){
1194 const unsigned char *zStr; /* The input string A */
1195 const unsigned char *zPattern; /* The pattern string B */
1196 const unsigned char *zRep; /* The replacement string C */
1197 unsigned char *zOut; /* The output */
1198 int nStr; /* Size of zStr */
1199 int nPattern; /* Size of zPattern */
1200 int nRep; /* Size of zRep */
drh2e6400b2007-05-08 15:46:18 +00001201 i64 nOut; /* Maximum size of zOut */
drh26b6d902007-03-17 13:27:54 +00001202 int loopLimit; /* Last zStr[] that might match zPattern[] */
1203 int i, j; /* Loop counters */
drhf3139522018-02-09 23:25:14 +00001204 unsigned cntExpand; /* Number zOut expansions */
1205 sqlite3 *db = sqlite3_context_db_handle(context);
drh26b6d902007-03-17 13:27:54 +00001206
1207 assert( argc==3 );
danielk1977f3d3c272008-11-19 16:52:44 +00001208 UNUSED_PARAMETER(argc);
drh26b6d902007-03-17 13:27:54 +00001209 zStr = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001210 if( zStr==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001211 nStr = sqlite3_value_bytes(argv[0]);
1212 assert( zStr==sqlite3_value_text(argv[0]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001213 zPattern = sqlite3_value_text(argv[1]);
drha605fe82009-02-01 18:08:40 +00001214 if( zPattern==0 ){
drh23336062009-02-03 13:19:12 +00001215 assert( sqlite3_value_type(argv[1])==SQLITE_NULL
1216 || sqlite3_context_db_handle(context)->mallocFailed );
drha605fe82009-02-01 18:08:40 +00001217 return;
1218 }
1219 if( zPattern[0]==0 ){
1220 assert( sqlite3_value_type(argv[1])!=SQLITE_NULL );
1221 sqlite3_result_value(context, argv[0]);
1222 return;
1223 }
drh1f0feef2007-05-15 13:27:07 +00001224 nPattern = sqlite3_value_bytes(argv[1]);
1225 assert( zPattern==sqlite3_value_text(argv[1]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001226 zRep = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +00001227 if( zRep==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001228 nRep = sqlite3_value_bytes(argv[2]);
1229 assert( zRep==sqlite3_value_text(argv[2]) );
drh2e6400b2007-05-08 15:46:18 +00001230 nOut = nStr + 1;
1231 assert( nOut<SQLITE_MAX_LENGTH );
drhb1a6c3c2008-03-20 16:30:17 +00001232 zOut = contextMalloc(context, (i64)nOut);
drh2e6400b2007-05-08 15:46:18 +00001233 if( zOut==0 ){
1234 return;
drh26b6d902007-03-17 13:27:54 +00001235 }
drh26b6d902007-03-17 13:27:54 +00001236 loopLimit = nStr - nPattern;
drhf3139522018-02-09 23:25:14 +00001237 cntExpand = 0;
drh26b6d902007-03-17 13:27:54 +00001238 for(i=j=0; i<=loopLimit; i++){
1239 if( zStr[i]!=zPattern[0] || memcmp(&zStr[i], zPattern, nPattern) ){
1240 zOut[j++] = zStr[i];
1241 }else{
drhf3139522018-02-09 23:25:14 +00001242 if( nRep>nPattern ){
1243 nOut += nRep - nPattern;
drhc86d82f2018-02-10 02:31:30 +00001244 testcase( nOut-1==db->aLimit[SQLITE_LIMIT_LENGTH] );
1245 testcase( nOut-2==db->aLimit[SQLITE_LIMIT_LENGTH] );
drhf3139522018-02-09 23:25:14 +00001246 if( nOut-1>db->aLimit[SQLITE_LIMIT_LENGTH] ){
1247 sqlite3_result_error_toobig(context);
1248 sqlite3_free(zOut);
1249 return;
1250 }
drhf3139522018-02-09 23:25:14 +00001251 cntExpand++;
1252 if( (cntExpand&(cntExpand-1))==0 ){
1253 /* Grow the size of the output buffer only on substitutions
1254 ** whose index is a power of two: 1, 2, 4, 8, 16, 32, ... */
1255 u8 *zOld;
1256 zOld = zOut;
1257 zOut = sqlite3_realloc64(zOut, (int)nOut + (nOut - nStr - 1));
1258 if( zOut==0 ){
1259 sqlite3_result_error_nomem(context);
1260 sqlite3_free(zOld);
1261 return;
1262 }
1263 }
drh2e6400b2007-05-08 15:46:18 +00001264 }
drh26b6d902007-03-17 13:27:54 +00001265 memcpy(&zOut[j], zRep, nRep);
1266 j += nRep;
1267 i += nPattern-1;
1268 }
1269 }
drhf3139522018-02-09 23:25:14 +00001270 assert( j+nStr-i+1<=nOut );
drh26b6d902007-03-17 13:27:54 +00001271 memcpy(&zOut[j], &zStr[i], nStr-i);
1272 j += nStr - i;
1273 assert( j<=nOut );
1274 zOut[j] = 0;
1275 sqlite3_result_text(context, (char*)zOut, j, sqlite3_free);
1276}
1277
drh309b3382007-03-17 17:52:42 +00001278/*
1279** Implementation of the TRIM(), LTRIM(), and RTRIM() functions.
1280** The userdata is 0x1 for left trim, 0x2 for right trim, 0x3 for both.
1281*/
1282static void trimFunc(
1283 sqlite3_context *context,
1284 int argc,
1285 sqlite3_value **argv
1286){
1287 const unsigned char *zIn; /* Input string */
1288 const unsigned char *zCharSet; /* Set of characters to trim */
1289 int nIn; /* Number of bytes in input */
drh7209c692008-04-27 18:40:11 +00001290 int flags; /* 1: trimleft 2: trimright 3: trim */
drhd1e3a612007-04-27 21:59:52 +00001291 int i; /* Loop counter */
drh1bd10f82008-12-10 21:19:56 +00001292 unsigned char *aLen = 0; /* Length of each character in zCharSet */
1293 unsigned char **azChar = 0; /* Individual characters in zCharSet */
drhd1e3a612007-04-27 21:59:52 +00001294 int nChar; /* Number of characters in zCharSet */
1295
drh309b3382007-03-17 17:52:42 +00001296 if( sqlite3_value_type(argv[0])==SQLITE_NULL ){
1297 return;
1298 }
1299 zIn = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001300 if( zIn==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001301 nIn = sqlite3_value_bytes(argv[0]);
1302 assert( zIn==sqlite3_value_text(argv[0]) );
drh309b3382007-03-17 17:52:42 +00001303 if( argc==1 ){
drhd1e3a612007-04-27 21:59:52 +00001304 static const unsigned char lenOne[] = { 1 };
danielk1977a4de4532008-09-02 15:44:08 +00001305 static unsigned char * const azOne[] = { (u8*)" " };
drhd1e3a612007-04-27 21:59:52 +00001306 nChar = 1;
drh8cff3822007-05-02 02:08:28 +00001307 aLen = (u8*)lenOne;
danielk1977bc67da42007-12-11 04:23:19 +00001308 azChar = (unsigned char **)azOne;
drhd1e3a612007-04-27 21:59:52 +00001309 zCharSet = 0;
drh7a521cf2007-04-25 18:23:52 +00001310 }else if( (zCharSet = sqlite3_value_text(argv[1]))==0 ){
drh309b3382007-03-17 17:52:42 +00001311 return;
drhd1e3a612007-04-27 21:59:52 +00001312 }else{
1313 const unsigned char *z;
1314 for(z=zCharSet, nChar=0; *z; nChar++){
drh4a919112007-05-15 11:55:09 +00001315 SQLITE_SKIP_UTF8(z);
drhd1e3a612007-04-27 21:59:52 +00001316 }
1317 if( nChar>0 ){
drhb1a6c3c2008-03-20 16:30:17 +00001318 azChar = contextMalloc(context, ((i64)nChar)*(sizeof(char*)+1));
drhd1e3a612007-04-27 21:59:52 +00001319 if( azChar==0 ){
1320 return;
1321 }
1322 aLen = (unsigned char*)&azChar[nChar];
1323 for(z=zCharSet, nChar=0; *z; nChar++){
danielk1977bc67da42007-12-11 04:23:19 +00001324 azChar[nChar] = (unsigned char *)z;
drh4a919112007-05-15 11:55:09 +00001325 SQLITE_SKIP_UTF8(z);
drh3abbd392008-12-10 23:04:13 +00001326 aLen[nChar] = (u8)(z - azChar[nChar]);
drhd1e3a612007-04-27 21:59:52 +00001327 }
1328 }
drh309b3382007-03-17 17:52:42 +00001329 }
drhd1e3a612007-04-27 21:59:52 +00001330 if( nChar>0 ){
shane1fc41292008-07-08 22:28:48 +00001331 flags = SQLITE_PTR_TO_INT(sqlite3_user_data(context));
drh309b3382007-03-17 17:52:42 +00001332 if( flags & 1 ){
drhd1e3a612007-04-27 21:59:52 +00001333 while( nIn>0 ){
drh1bd10f82008-12-10 21:19:56 +00001334 int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001335 for(i=0; i<nChar; i++){
1336 len = aLen[i];
drh27e62db2009-04-02 10:16:17 +00001337 if( len<=nIn && memcmp(zIn, azChar[i], len)==0 ) break;
drhd1e3a612007-04-27 21:59:52 +00001338 }
1339 if( i>=nChar ) break;
1340 zIn += len;
1341 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001342 }
1343 }
1344 if( flags & 2 ){
drhd1e3a612007-04-27 21:59:52 +00001345 while( nIn>0 ){
drh1bd10f82008-12-10 21:19:56 +00001346 int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001347 for(i=0; i<nChar; i++){
1348 len = aLen[i];
1349 if( len<=nIn && memcmp(&zIn[nIn-len],azChar[i],len)==0 ) break;
1350 }
1351 if( i>=nChar ) break;
1352 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001353 }
1354 }
drhd1e3a612007-04-27 21:59:52 +00001355 if( zCharSet ){
1356 sqlite3_free(azChar);
1357 }
drh309b3382007-03-17 17:52:42 +00001358 }
1359 sqlite3_result_text(context, (char*)zIn, nIn, SQLITE_TRANSIENT);
1360}
drh26b6d902007-03-17 13:27:54 +00001361
danielk1977a4de4532008-09-02 15:44:08 +00001362
drhcc153132016-08-04 12:35:17 +00001363#ifdef SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION
1364/*
1365** The "unknown" function is automatically substituted in place of
1366** any unrecognized function name when doing an EXPLAIN or EXPLAIN QUERY PLAN
1367** when the SQLITE_ENABLE_UNKNOWN_FUNCTION compile-time option is used.
1368** When the "sqlite3" command-line shell is built using this functionality,
1369** that allows an EXPLAIN or EXPLAIN QUERY PLAN for complex queries
1370** involving application-defined functions to be examined in a generic
1371** sqlite3 shell.
1372*/
1373static void unknownFunc(
1374 sqlite3_context *context,
1375 int argc,
1376 sqlite3_value **argv
1377){
1378 /* no-op */
1379}
1380#endif /*SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION*/
1381
1382
drh2ba3ccc2009-12-08 02:06:08 +00001383/* IMP: R-25361-16150 This function is omitted from SQLite by default. It
1384** is only available if the SQLITE_SOUNDEX compile-time option is used
1385** when SQLite is built.
1386*/
drhd24cc422003-03-27 12:51:24 +00001387#ifdef SQLITE_SOUNDEX
1388/*
1389** Compute the soundex encoding of a word.
drh2ba3ccc2009-12-08 02:06:08 +00001390**
1391** IMP: R-59782-00072 The soundex(X) function returns a string that is the
1392** soundex encoding of the string X.
drhd24cc422003-03-27 12:51:24 +00001393*/
drh137c7282007-01-29 17:58:28 +00001394static void soundexFunc(
1395 sqlite3_context *context,
1396 int argc,
1397 sqlite3_value **argv
1398){
drhd24cc422003-03-27 12:51:24 +00001399 char zResult[8];
drh4c755c02004-08-08 20:22:17 +00001400 const u8 *zIn;
drhd24cc422003-03-27 12:51:24 +00001401 int i, j;
1402 static const unsigned char iCode[] = {
1403 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1404 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1405 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1406 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1407 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1408 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1409 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1410 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1411 };
1412 assert( argc==1 );
drh4c755c02004-08-08 20:22:17 +00001413 zIn = (u8*)sqlite3_value_text(argv[0]);
drhbdf67e02006-08-19 11:34:01 +00001414 if( zIn==0 ) zIn = (u8*)"";
drhdc86e2b2009-01-24 11:30:42 +00001415 for(i=0; zIn[i] && !sqlite3Isalpha(zIn[i]); i++){}
drhd24cc422003-03-27 12:51:24 +00001416 if( zIn[i] ){
drhbdf67e02006-08-19 11:34:01 +00001417 u8 prevcode = iCode[zIn[i]&0x7f];
danielk197778ca0e72009-01-20 16:53:39 +00001418 zResult[0] = sqlite3Toupper(zIn[i]);
drhd24cc422003-03-27 12:51:24 +00001419 for(j=1; j<4 && zIn[i]; i++){
1420 int code = iCode[zIn[i]&0x7f];
1421 if( code>0 ){
drhbdf67e02006-08-19 11:34:01 +00001422 if( code!=prevcode ){
1423 prevcode = code;
1424 zResult[j++] = code + '0';
1425 }
1426 }else{
1427 prevcode = 0;
drhd24cc422003-03-27 12:51:24 +00001428 }
1429 }
1430 while( j<4 ){
1431 zResult[j++] = '0';
1432 }
1433 zResult[j] = 0;
danielk1977d8123362004-06-12 09:25:12 +00001434 sqlite3_result_text(context, zResult, 4, SQLITE_TRANSIENT);
drhd24cc422003-03-27 12:51:24 +00001435 }else{
drh2ba3ccc2009-12-08 02:06:08 +00001436 /* IMP: R-64894-50321 The string "?000" is returned if the argument
1437 ** is NULL or contains no ASCII alphabetic characters. */
danielk1977d8123362004-06-12 09:25:12 +00001438 sqlite3_result_text(context, "?000", 4, SQLITE_STATIC);
drhd24cc422003-03-27 12:51:24 +00001439 }
1440}
drh2ba3ccc2009-12-08 02:06:08 +00001441#endif /* SQLITE_SOUNDEX */
drhd24cc422003-03-27 12:51:24 +00001442
drhfdb83b22006-06-17 14:12:47 +00001443#ifndef SQLITE_OMIT_LOAD_EXTENSION
1444/*
1445** A function that loads a shared-library extension then returns NULL.
1446*/
1447static void loadExt(sqlite3_context *context, int argc, sqlite3_value **argv){
danielk197765fd59f2006-06-24 11:51:33 +00001448 const char *zFile = (const char *)sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001449 const char *zProc;
drhfa4a4b92008-03-19 21:45:51 +00001450 sqlite3 *db = sqlite3_context_db_handle(context);
drhfdb83b22006-06-17 14:12:47 +00001451 char *zErrMsg = 0;
1452
drh191dd062016-04-21 01:30:09 +00001453 /* Disallow the load_extension() SQL function unless the SQLITE_LoadExtFunc
drh1a55ded2016-04-20 00:30:05 +00001454 ** flag is set. See the sqlite3_enable_load_extension() API.
1455 */
drhf602a162016-04-21 01:58:21 +00001456 if( (db->flags & SQLITE_LoadExtFunc)==0 ){
1457 sqlite3_result_error(context, "not authorized", -1);
1458 return;
1459 }
drh1a55ded2016-04-20 00:30:05 +00001460
drhfdb83b22006-06-17 14:12:47 +00001461 if( argc==2 ){
danielk197765fd59f2006-06-24 11:51:33 +00001462 zProc = (const char *)sqlite3_value_text(argv[1]);
drh7a521cf2007-04-25 18:23:52 +00001463 }else{
1464 zProc = 0;
drhfdb83b22006-06-17 14:12:47 +00001465 }
drh7a521cf2007-04-25 18:23:52 +00001466 if( zFile && sqlite3_load_extension(db, zFile, zProc, &zErrMsg) ){
drhfdb83b22006-06-17 14:12:47 +00001467 sqlite3_result_error(context, zErrMsg, -1);
1468 sqlite3_free(zErrMsg);
1469 }
1470}
1471#endif
1472
danielk197701427a62005-01-11 13:02:33 +00001473
drh0ac65892002-04-20 14:24:41 +00001474/*
drhd3a149e2002-02-24 17:12:53 +00001475** An instance of the following structure holds the context of a
drhdd5baa92002-02-27 19:50:59 +00001476** sum() or avg() aggregate computation.
1477*/
1478typedef struct SumCtx SumCtx;
1479struct SumCtx {
drh8c08e862006-02-11 17:34:00 +00001480 double rSum; /* Floating point sum */
1481 i64 iSum; /* Integer sum */
1482 i64 cnt; /* Number of elements summed */
1483 u8 overflow; /* True if integer overflow seen */
1484 u8 approx; /* True if non-integer value was input to the sum */
drhdd5baa92002-02-27 19:50:59 +00001485};
1486
1487/*
drha97fdd32006-01-12 22:17:50 +00001488** Routines used to compute the sum, average, and total.
1489**
1490** The SUM() function follows the (broken) SQL standard which means
1491** that it returns NULL if it sums over no inputs. TOTAL returns
1492** 0.0 in that case. In addition, TOTAL always returns a float where
1493** SUM might return an integer if it never encounters a floating point
drhc806d852006-05-11 13:25:39 +00001494** value. TOTAL never fails, but SUM might through an exception if
1495** it overflows an integer.
drhdd5baa92002-02-27 19:50:59 +00001496*/
danielk19770ae8b832004-05-25 12:05:56 +00001497static void sumStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drhdd5baa92002-02-27 19:50:59 +00001498 SumCtx *p;
drh3d1d95e2005-09-08 10:37:01 +00001499 int type;
drh3f219f42005-09-08 19:45:57 +00001500 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001501 UNUSED_PARAMETER(argc);
drh4f26d6c2004-05-26 23:25:30 +00001502 p = sqlite3_aggregate_context(context, sizeof(*p));
drh29d72102006-02-09 22:13:41 +00001503 type = sqlite3_value_numeric_type(argv[0]);
drh3d1d95e2005-09-08 10:37:01 +00001504 if( p && type!=SQLITE_NULL ){
drh739105c2002-05-29 23:22:23 +00001505 p->cnt++;
drh29d72102006-02-09 22:13:41 +00001506 if( type==SQLITE_INTEGER ){
drh8c08e862006-02-11 17:34:00 +00001507 i64 v = sqlite3_value_int64(argv[0]);
1508 p->rSum += v;
drh158b9cb2011-03-05 20:59:46 +00001509 if( (p->approx|p->overflow)==0 && sqlite3AddInt64(&p->iSum, v) ){
drha546ef22018-07-07 20:55:16 +00001510 p->approx = p->overflow = 1;
drh29d72102006-02-09 22:13:41 +00001511 }
1512 }else{
drh8c08e862006-02-11 17:34:00 +00001513 p->rSum += sqlite3_value_double(argv[0]);
drh29d72102006-02-09 22:13:41 +00001514 p->approx = 1;
drh3f219f42005-09-08 19:45:57 +00001515 }
drh739105c2002-05-29 23:22:23 +00001516 }
drhdd5baa92002-02-27 19:50:59 +00001517}
dan67a9b8e2018-06-22 20:51:35 +00001518#ifndef SQLITE_OMIT_WINDOWFUNC
danc3a20c12018-05-23 20:55:37 +00001519static void sumInverse(sqlite3_context *context, int argc, sqlite3_value**argv){
1520 SumCtx *p;
1521 int type;
1522 assert( argc==1 );
1523 UNUSED_PARAMETER(argc);
1524 p = sqlite3_aggregate_context(context, sizeof(*p));
1525 type = sqlite3_value_numeric_type(argv[0]);
drhfd4b7282018-07-07 19:47:21 +00001526 /* p is always non-NULL because sumStep() will have been called first
1527 ** to initialize it */
1528 if( ALWAYS(p) && type!=SQLITE_NULL ){
drha546ef22018-07-07 20:55:16 +00001529 assert( p->cnt>0 );
danc3a20c12018-05-23 20:55:37 +00001530 p->cnt--;
drha546ef22018-07-07 20:55:16 +00001531 assert( type==SQLITE_INTEGER || p->approx );
1532 if( type==SQLITE_INTEGER && p->approx==0 ){
danc3a20c12018-05-23 20:55:37 +00001533 i64 v = sqlite3_value_int64(argv[0]);
1534 p->rSum -= v;
drha546ef22018-07-07 20:55:16 +00001535 p->iSum -= v;
danc3a20c12018-05-23 20:55:37 +00001536 }else{
dand7368292018-07-02 17:45:59 +00001537 p->rSum -= sqlite3_value_double(argv[0]);
danc3a20c12018-05-23 20:55:37 +00001538 }
1539 }
1540}
dan67a9b8e2018-06-22 20:51:35 +00001541#else
1542# define sumInverse 0
1543#endif /* SQLITE_OMIT_WINDOWFUNC */
danielk19770ae8b832004-05-25 12:05:56 +00001544static void sumFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001545 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001546 p = sqlite3_aggregate_context(context, 0);
drhc2bd9132005-09-08 20:37:43 +00001547 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001548 if( p->overflow ){
1549 sqlite3_result_error(context,"integer overflow",-1);
1550 }else if( p->approx ){
1551 sqlite3_result_double(context, p->rSum);
drhc2bd9132005-09-08 20:37:43 +00001552 }else{
drh8c08e862006-02-11 17:34:00 +00001553 sqlite3_result_int64(context, p->iSum);
drhc2bd9132005-09-08 20:37:43 +00001554 }
drh3d1d95e2005-09-08 10:37:01 +00001555 }
drhdd5baa92002-02-27 19:50:59 +00001556}
danielk19770ae8b832004-05-25 12:05:56 +00001557static void avgFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001558 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001559 p = sqlite3_aggregate_context(context, 0);
drh739105c2002-05-29 23:22:23 +00001560 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001561 sqlite3_result_double(context, p->rSum/(double)p->cnt);
drhdd5baa92002-02-27 19:50:59 +00001562 }
1563}
drha97fdd32006-01-12 22:17:50 +00001564static void totalFinalize(sqlite3_context *context){
1565 SumCtx *p;
1566 p = sqlite3_aggregate_context(context, 0);
shanefbd60f82009-02-04 03:59:25 +00001567 /* (double)0 In case of SQLITE_OMIT_FLOATING_POINT... */
1568 sqlite3_result_double(context, p ? p->rSum : (double)0);
drha97fdd32006-01-12 22:17:50 +00001569}
drhdd5baa92002-02-27 19:50:59 +00001570
1571/*
drh0bce8352002-02-28 00:41:10 +00001572** The following structure keeps track of state information for the
1573** count() aggregate function.
1574*/
1575typedef struct CountCtx CountCtx;
1576struct CountCtx {
drhfc6ad392006-02-09 13:38:19 +00001577 i64 n;
dan7262ca92018-07-02 12:07:32 +00001578#ifdef SQLITE_DEBUG
1579 int bInverse; /* True if xInverse() ever called */
1580#endif
drh0bce8352002-02-28 00:41:10 +00001581};
drhdd5baa92002-02-27 19:50:59 +00001582
drh0bce8352002-02-28 00:41:10 +00001583/*
1584** Routines to implement the count() aggregate function.
1585*/
danielk19770ae8b832004-05-25 12:05:56 +00001586static void countStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drh0bce8352002-02-28 00:41:10 +00001587 CountCtx *p;
drh4f26d6c2004-05-26 23:25:30 +00001588 p = sqlite3_aggregate_context(context, sizeof(*p));
drh9c054832004-05-31 18:51:57 +00001589 if( (argc==0 || SQLITE_NULL!=sqlite3_value_type(argv[0])) && p ){
drh0bce8352002-02-28 00:41:10 +00001590 p->n++;
1591 }
drh2e79c3d2009-04-08 23:04:14 +00001592
drhd3264c72009-04-15 13:39:47 +00001593#ifndef SQLITE_OMIT_DEPRECATED
drh2e79c3d2009-04-08 23:04:14 +00001594 /* The sqlite3_aggregate_count() function is deprecated. But just to make
1595 ** sure it still operates correctly, verify that its count agrees with our
1596 ** internal count when using count(*) and when the total count can be
1597 ** expressed as a 32-bit integer. */
dan7262ca92018-07-02 12:07:32 +00001598 assert( argc==1 || p==0 || p->n>0x7fffffff || p->bInverse
drh2e79c3d2009-04-08 23:04:14 +00001599 || p->n==sqlite3_aggregate_count(context) );
drhd3264c72009-04-15 13:39:47 +00001600#endif
drh0bce8352002-02-28 00:41:10 +00001601}
danielk19770ae8b832004-05-25 12:05:56 +00001602static void countFinalize(sqlite3_context *context){
drh0bce8352002-02-28 00:41:10 +00001603 CountCtx *p;
drhabfcea22005-09-06 20:36:48 +00001604 p = sqlite3_aggregate_context(context, 0);
drhfc6ad392006-02-09 13:38:19 +00001605 sqlite3_result_int64(context, p ? p->n : 0);
drh0bce8352002-02-28 00:41:10 +00001606}
dan7262ca92018-07-02 12:07:32 +00001607#ifndef SQLITE_OMIT_WINDOWFUNC
1608static void countInverse(sqlite3_context *ctx, int argc, sqlite3_value **argv){
1609 CountCtx *p;
1610 p = sqlite3_aggregate_context(ctx, sizeof(*p));
drhfd4b7282018-07-07 19:47:21 +00001611 /* p is always non-NULL since countStep() will have been called first */
1612 if( (argc==0 || SQLITE_NULL!=sqlite3_value_type(argv[0])) && ALWAYS(p) ){
dan7262ca92018-07-02 12:07:32 +00001613 p->n--;
1614#ifdef SQLITE_DEBUG
1615 p->bInverse = 1;
1616#endif
1617 }
1618}
dan6b4b8822018-07-02 15:03:50 +00001619#else
1620# define countInverse 0
1621#endif /* SQLITE_OMIT_WINDOWFUNC */
drh0bce8352002-02-28 00:41:10 +00001622
1623/*
drh0bce8352002-02-28 00:41:10 +00001624** Routines to implement min() and max() aggregate functions.
1625*/
danielk197762c14b32008-11-19 09:05:26 +00001626static void minmaxStep(
1627 sqlite3_context *context,
1628 int NotUsed,
1629 sqlite3_value **argv
1630){
danielk197788208052004-05-25 01:13:20 +00001631 Mem *pArg = (Mem *)argv[0];
drh9eb516c2004-07-18 20:52:32 +00001632 Mem *pBest;
danielk197762c14b32008-11-19 09:05:26 +00001633 UNUSED_PARAMETER(NotUsed);
drh9eb516c2004-07-18 20:52:32 +00001634
drh9eb516c2004-07-18 20:52:32 +00001635 pBest = (Mem *)sqlite3_aggregate_context(context, sizeof(*pBest));
danielk19773aeab9e2004-06-24 00:20:04 +00001636 if( !pBest ) return;
drh268380c2004-02-25 13:47:31 +00001637
dan6fb2b542018-06-19 17:13:11 +00001638 if( sqlite3_value_type(pArg)==SQLITE_NULL ){
drh94a6d992012-02-02 18:42:09 +00001639 if( pBest->flags ) sqlite3SkipAccumulatorLoad(context);
1640 }else if( pBest->flags ){
drh9eb516c2004-07-18 20:52:32 +00001641 int max;
1642 int cmp;
danielk1977dc1bdc42004-06-11 10:51:27 +00001643 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk19777e18c252004-05-25 11:47:24 +00001644 /* This step function is used for both the min() and max() aggregates,
1645 ** the only difference between the two being that the sense of the
1646 ** comparison is inverted. For the max() aggregate, the
1647 ** sqlite3_user_data() function returns (void *)-1. For min() it
1648 ** returns (void *)db, where db is the sqlite3* database pointer.
1649 ** Therefore the next statement sets variable 'max' to 1 for the max()
1650 ** aggregate, or 0 for min().
1651 */
drh309b3382007-03-17 17:52:42 +00001652 max = sqlite3_user_data(context)!=0;
danielk1977dc1bdc42004-06-11 10:51:27 +00001653 cmp = sqlite3MemCompare(pBest, pArg, pColl);
danielk197788208052004-05-25 01:13:20 +00001654 if( (max && cmp<0) || (!max && cmp>0) ){
drhb21c8cd2007-08-21 19:33:56 +00001655 sqlite3VdbeMemCopy(pBest, pArg);
drh7a957892012-02-02 17:35:43 +00001656 }else{
1657 sqlite3SkipAccumulatorLoad(context);
danielk197788208052004-05-25 01:13:20 +00001658 }
drh268380c2004-02-25 13:47:31 +00001659 }else{
drh035e5632014-09-16 14:16:31 +00001660 pBest->db = sqlite3_context_db_handle(context);
drhb21c8cd2007-08-21 19:33:56 +00001661 sqlite3VdbeMemCopy(pBest, pArg);
drh0bce8352002-02-28 00:41:10 +00001662 }
1663}
dan6fb2b542018-06-19 17:13:11 +00001664static void minMaxValueFinalize(sqlite3_context *context, int bValue){
danielk197788208052004-05-25 01:13:20 +00001665 sqlite3_value *pRes;
drhabfcea22005-09-06 20:36:48 +00001666 pRes = (sqlite3_value *)sqlite3_aggregate_context(context, 0);
1667 if( pRes ){
drh94a6d992012-02-02 18:42:09 +00001668 if( pRes->flags ){
drhabfcea22005-09-06 20:36:48 +00001669 sqlite3_result_value(context, pRes);
1670 }
dan6fb2b542018-06-19 17:13:11 +00001671 if( bValue==0 ) sqlite3VdbeMemRelease(pRes);
drh0bce8352002-02-28 00:41:10 +00001672 }
1673}
dan67a9b8e2018-06-22 20:51:35 +00001674#ifndef SQLITE_OMIT_WINDOWFUNC
dan6fb2b542018-06-19 17:13:11 +00001675static void minMaxValue(sqlite3_context *context){
drhc7bf5712018-07-09 22:49:01 +00001676 minMaxValueFinalize(context, 1);
dan6fb2b542018-06-19 17:13:11 +00001677}
dan67a9b8e2018-06-22 20:51:35 +00001678#else
1679# define minMaxValue 0
1680#endif /* SQLITE_OMIT_WINDOWFUNC */
dan6fb2b542018-06-19 17:13:11 +00001681static void minMaxFinalize(sqlite3_context *context){
drhc7bf5712018-07-09 22:49:01 +00001682 minMaxValueFinalize(context, 0);
dan6fb2b542018-06-19 17:13:11 +00001683}
drhdd5baa92002-02-27 19:50:59 +00001684
drhb0689692007-11-01 17:38:30 +00001685/*
1686** group_concat(EXPR, ?SEPARATOR?)
1687*/
1688static void groupConcatStep(
1689 sqlite3_context *context,
1690 int argc,
1691 sqlite3_value **argv
1692){
1693 const char *zVal;
drhade86482007-11-28 22:36:40 +00001694 StrAccum *pAccum;
drhb0689692007-11-01 17:38:30 +00001695 const char *zSep;
drh07d31172009-02-02 21:57:05 +00001696 int nVal, nSep;
1697 assert( argc==1 || argc==2 );
1698 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drhade86482007-11-28 22:36:40 +00001699 pAccum = (StrAccum*)sqlite3_aggregate_context(context, sizeof(*pAccum));
1700
1701 if( pAccum ){
drhbb4957f2008-03-20 14:03:29 +00001702 sqlite3 *db = sqlite3_context_db_handle(context);
dane3bf6322018-06-08 20:58:27 +00001703 int firstTerm = pAccum->mxAlloc==0;
drhbb4957f2008-03-20 14:03:29 +00001704 pAccum->mxAlloc = db->aLimit[SQLITE_LIMIT_LENGTH];
drh8bfd7192009-06-19 16:44:41 +00001705 if( !firstTerm ){
drh07d31172009-02-02 21:57:05 +00001706 if( argc==2 ){
1707 zSep = (char*)sqlite3_value_text(argv[1]);
1708 nSep = sqlite3_value_bytes(argv[1]);
drhb0689692007-11-01 17:38:30 +00001709 }else{
1710 zSep = ",";
drhade86482007-11-28 22:36:40 +00001711 nSep = 1;
drhb0689692007-11-01 17:38:30 +00001712 }
drh0cdbe1a2018-05-09 13:46:26 +00001713 if( zSep ) sqlite3_str_append(pAccum, zSep, nSep);
drhb0689692007-11-01 17:38:30 +00001714 }
drh07d31172009-02-02 21:57:05 +00001715 zVal = (char*)sqlite3_value_text(argv[0]);
1716 nVal = sqlite3_value_bytes(argv[0]);
drh0cdbe1a2018-05-09 13:46:26 +00001717 if( zVal ) sqlite3_str_append(pAccum, zVal, nVal);
drhb0689692007-11-01 17:38:30 +00001718 }
1719}
dan67a9b8e2018-06-22 20:51:35 +00001720#ifndef SQLITE_OMIT_WINDOWFUNC
dan03854d22018-06-08 11:45:28 +00001721static void groupConcatInverse(
1722 sqlite3_context *context,
1723 int argc,
1724 sqlite3_value **argv
1725){
1726 int n;
dan03854d22018-06-08 11:45:28 +00001727 StrAccum *pAccum;
drhc7bf5712018-07-09 22:49:01 +00001728 assert( argc==1 || argc==2 );
dan03854d22018-06-08 11:45:28 +00001729 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
1730 pAccum = (StrAccum*)sqlite3_aggregate_context(context, sizeof(*pAccum));
drhfd4b7282018-07-07 19:47:21 +00001731 /* pAccum is always non-NULL since groupConcatStep() will have always
1732 ** run frist to initialize it */
1733 if( ALWAYS(pAccum) ){
dan03854d22018-06-08 11:45:28 +00001734 n = sqlite3_value_bytes(argv[0]);
1735 if( argc==2 ){
1736 n += sqlite3_value_bytes(argv[1]);
dan683b0ff2018-07-05 18:19:29 +00001737 }else{
1738 n++;
dan03854d22018-06-08 11:45:28 +00001739 }
drhc7bf5712018-07-09 22:49:01 +00001740 if( n>=(int)pAccum->nChar ){
dan03854d22018-06-08 11:45:28 +00001741 pAccum->nChar = 0;
1742 }else{
1743 pAccum->nChar -= n;
1744 memmove(pAccum->zText, &pAccum->zText[n], pAccum->nChar);
1745 }
dane3bf6322018-06-08 20:58:27 +00001746 if( pAccum->nChar==0 ) pAccum->mxAlloc = 0;
dan03854d22018-06-08 11:45:28 +00001747 }
1748}
dan67a9b8e2018-06-22 20:51:35 +00001749#else
1750# define groupConcatInverse 0
1751#endif /* SQLITE_OMIT_WINDOWFUNC */
drhb0689692007-11-01 17:38:30 +00001752static void groupConcatFinalize(sqlite3_context *context){
drhade86482007-11-28 22:36:40 +00001753 StrAccum *pAccum;
1754 pAccum = sqlite3_aggregate_context(context, 0);
1755 if( pAccum ){
drh0cdbe1a2018-05-09 13:46:26 +00001756 if( pAccum->accError==SQLITE_TOOBIG ){
drhade86482007-11-28 22:36:40 +00001757 sqlite3_result_error_toobig(context);
drh0cdbe1a2018-05-09 13:46:26 +00001758 }else if( pAccum->accError==SQLITE_NOMEM ){
drhade86482007-11-28 22:36:40 +00001759 sqlite3_result_error_nomem(context);
1760 }else{
1761 sqlite3_result_text(context, sqlite3StrAccumFinish(pAccum), -1,
1762 sqlite3_free);
1763 }
drhb0689692007-11-01 17:38:30 +00001764 }
1765}
dan67a9b8e2018-06-22 20:51:35 +00001766#ifndef SQLITE_OMIT_WINDOWFUNC
dane2f781b2018-05-17 19:24:08 +00001767static void groupConcatValue(sqlite3_context *context){
1768 sqlite3_str *pAccum;
1769 pAccum = (sqlite3_str*)sqlite3_aggregate_context(context, 0);
1770 if( pAccum ){
1771 if( pAccum->accError==SQLITE_TOOBIG ){
1772 sqlite3_result_error_toobig(context);
1773 }else if( pAccum->accError==SQLITE_NOMEM ){
1774 sqlite3_result_error_nomem(context);
1775 }else{
1776 const char *zText = sqlite3_str_value(pAccum);
1777 sqlite3_result_text(context, zText, -1, SQLITE_TRANSIENT);
1778 }
1779 }
1780}
dan67a9b8e2018-06-22 20:51:35 +00001781#else
1782# define groupConcatValue 0
1783#endif /* SQLITE_OMIT_WINDOWFUNC */
drh4e5ffc52004-08-31 00:52:37 +00001784
drhd3a149e2002-02-24 17:12:53 +00001785/*
drha4741842010-04-25 20:58:37 +00001786** This routine does per-connection function registration. Most
1787** of the built-in functions above are part of the global function set.
1788** This routine only deals with those that are not global.
drhdc04c582002-02-24 01:55:15 +00001789*/
drh80738d92016-02-15 00:34:16 +00001790void sqlite3RegisterPerConnectionBuiltinFunctions(sqlite3 *db){
drha4741842010-04-25 20:58:37 +00001791 int rc = sqlite3_overload_function(db, "MATCH", 2);
1792 assert( rc==SQLITE_NOMEM || rc==SQLITE_OK );
1793 if( rc==SQLITE_NOMEM ){
drh4a642b62016-02-05 01:55:27 +00001794 sqlite3OomFault(db);
danielk1977832a58a2007-06-22 15:21:15 +00001795 }
drh55ef4d92005-08-14 01:20:37 +00001796}
1797
1798/*
1799** Set the LIKEOPT flag on the 2-argument function with the given name.
1800*/
drh1bd10f82008-12-10 21:19:56 +00001801static void setLikeOptFlag(sqlite3 *db, const char *zName, u8 flagVal){
drh55ef4d92005-08-14 01:20:37 +00001802 FuncDef *pDef;
drh80738d92016-02-15 00:34:16 +00001803 pDef = sqlite3FindFunction(db, zName, 2, SQLITE_UTF8, 0);
drhd27135a2009-04-02 13:36:37 +00001804 if( ALWAYS(pDef) ){
drhd36e1042013-09-06 13:10:12 +00001805 pDef->funcFlags |= flagVal;
drh55ef4d92005-08-14 01:20:37 +00001806 }
drh8e0f7942019-02-27 23:05:14 +00001807 pDef = sqlite3FindFunction(db, zName, 3, SQLITE_UTF8, 0);
1808 if( pDef ){
1809 pDef->funcFlags |= flagVal;
1810 }
drh55ef4d92005-08-14 01:20:37 +00001811}
1812
1813/*
1814** Register the built-in LIKE and GLOB functions. The caseSensitive
1815** parameter determines whether or not the LIKE operator is case
1816** sensitive. GLOB is always case sensitive.
1817*/
1818void sqlite3RegisterLikeFunctions(sqlite3 *db, int caseSensitive){
1819 struct compareInfo *pInfo;
1820 if( caseSensitive ){
1821 pInfo = (struct compareInfo*)&likeInfoAlt;
1822 }else{
1823 pInfo = (struct compareInfo*)&likeInfoNorm;
1824 }
dan660af932018-06-18 16:55:22 +00001825 sqlite3CreateFunc(db, "like", 2, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0, 0, 0);
1826 sqlite3CreateFunc(db, "like", 3, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0, 0, 0);
drh901e9942010-12-15 18:54:37 +00001827 sqlite3CreateFunc(db, "glob", 2, SQLITE_UTF8,
dan660af932018-06-18 16:55:22 +00001828 (struct compareInfo*)&globInfo, likeFunc, 0, 0, 0, 0, 0);
drhd64fe2f2005-08-28 17:00:23 +00001829 setLikeOptFlag(db, "glob", SQLITE_FUNC_LIKE | SQLITE_FUNC_CASE);
1830 setLikeOptFlag(db, "like",
1831 caseSensitive ? (SQLITE_FUNC_LIKE | SQLITE_FUNC_CASE) : SQLITE_FUNC_LIKE);
drh55ef4d92005-08-14 01:20:37 +00001832}
1833
1834/*
1835** pExpr points to an expression which implements a function. If
1836** it is appropriate to apply the LIKE optimization to that function
drh1d42ea72017-07-27 20:24:29 +00001837** then set aWc[0] through aWc[2] to the wildcard characters and the
1838** escape character and then return TRUE. If the function is not a
1839** LIKE-style function then return FALSE.
1840**
1841** The expression "a LIKE b ESCAPE c" is only considered a valid LIKE
1842** operator if c is a string literal that is exactly one byte in length.
1843** That one byte is stored in aWc[3]. aWc[3] is set to zero if there is
1844** no ESCAPE clause.
drh16897072015-03-07 00:57:37 +00001845**
1846** *pIsNocase is set to true if uppercase and lowercase are equivalent for
1847** the function (default for LIKE). If the function makes the distinction
1848** between uppercase and lowercase (as does GLOB) then *pIsNocase is set to
1849** false.
drh55ef4d92005-08-14 01:20:37 +00001850*/
drhd64fe2f2005-08-28 17:00:23 +00001851int sqlite3IsLikeFunction(sqlite3 *db, Expr *pExpr, int *pIsNocase, char *aWc){
drh55ef4d92005-08-14 01:20:37 +00001852 FuncDef *pDef;
drh1d42ea72017-07-27 20:24:29 +00001853 int nExpr;
1854 if( pExpr->op!=TK_FUNCTION || !pExpr->x.pList ){
drh55ef4d92005-08-14 01:20:37 +00001855 return 0;
1856 }
danielk19776ab3a2e2009-02-19 14:39:25 +00001857 assert( !ExprHasProperty(pExpr, EP_xIsSelect) );
drh1d42ea72017-07-27 20:24:29 +00001858 nExpr = pExpr->x.pList->nExpr;
1859 pDef = sqlite3FindFunction(db, pExpr->u.zToken, nExpr, SQLITE_UTF8, 0);
drhd36e1042013-09-06 13:10:12 +00001860 if( NEVER(pDef==0) || (pDef->funcFlags & SQLITE_FUNC_LIKE)==0 ){
drh55ef4d92005-08-14 01:20:37 +00001861 return 0;
1862 }
drh1d42ea72017-07-27 20:24:29 +00001863 if( nExpr<3 ){
1864 aWc[3] = 0;
1865 }else{
1866 Expr *pEscape = pExpr->x.pList->a[2].pExpr;
1867 char *zEscape;
1868 if( pEscape->op!=TK_STRING ) return 0;
1869 zEscape = pEscape->u.zToken;
1870 if( zEscape[0]==0 || zEscape[1]!=0 ) return 0;
1871 aWc[3] = zEscape[0];
1872 }
drh55ef4d92005-08-14 01:20:37 +00001873
1874 /* The memcpy() statement assumes that the wildcard characters are
1875 ** the first three statements in the compareInfo structure. The
1876 ** asserts() that follow verify that assumption
1877 */
1878 memcpy(aWc, pDef->pUserData, 3);
1879 assert( (char*)&likeInfoAlt == (char*)&likeInfoAlt.matchAll );
1880 assert( &((char*)&likeInfoAlt)[1] == (char*)&likeInfoAlt.matchOne );
1881 assert( &((char*)&likeInfoAlt)[2] == (char*)&likeInfoAlt.matchSet );
drhd36e1042013-09-06 13:10:12 +00001882 *pIsNocase = (pDef->funcFlags & SQLITE_FUNC_CASE)==0;
drh55ef4d92005-08-14 01:20:37 +00001883 return 1;
drhdc04c582002-02-24 01:55:15 +00001884}
danielk19778c0a7912008-08-20 14:49:23 +00001885
drh70a8ca32008-08-21 18:49:27 +00001886/*
peter.d.reid60ec9142014-09-06 16:39:46 +00001887** All of the FuncDef structures in the aBuiltinFunc[] array above
drh777c5382008-08-21 20:21:34 +00001888** to the global function hash table. This occurs at start-time (as
1889** a consequence of calling sqlite3_initialize()).
1890**
1891** After this routine runs
drh70a8ca32008-08-21 18:49:27 +00001892*/
drh80738d92016-02-15 00:34:16 +00001893void sqlite3RegisterBuiltinFunctions(void){
danielk197793ce7412008-09-01 19:14:02 +00001894 /*
1895 ** The following array holds FuncDef structures for all of the functions
1896 ** defined in this file.
1897 **
1898 ** The array cannot be constant since changes are made to the
1899 ** FuncDef.pHash elements at start-time. The elements of this array
1900 ** are read-only after initialization is complete.
drh80738d92016-02-15 00:34:16 +00001901 **
1902 ** For peak efficiency, put the most frequently used function last.
danielk197793ce7412008-09-01 19:14:02 +00001903 */
drh80738d92016-02-15 00:34:16 +00001904 static FuncDef aBuiltinFunc[] = {
1905#ifdef SQLITE_SOUNDEX
1906 FUNCTION(soundex, 1, 0, 0, soundexFunc ),
1907#endif
1908#ifndef SQLITE_OMIT_LOAD_EXTENSION
1909 VFUNCTION(load_extension, 1, 0, 0, loadExt ),
1910 VFUNCTION(load_extension, 2, 0, 0, loadExt ),
1911#endif
1912#if SQLITE_USER_AUTHENTICATION
1913 FUNCTION(sqlite_crypt, 2, 0, 0, sqlite3CryptFunc ),
1914#endif
1915#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
1916 DFUNCTION(sqlite_compileoption_used,1, 0, 0, compileoptionusedFunc ),
1917 DFUNCTION(sqlite_compileoption_get, 1, 0, 0, compileoptiongetFunc ),
1918#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
1919 FUNCTION2(unlikely, 1, 0, 0, noopFunc, SQLITE_FUNC_UNLIKELY),
1920 FUNCTION2(likelihood, 2, 0, 0, noopFunc, SQLITE_FUNC_UNLIKELY),
1921 FUNCTION2(likely, 1, 0, 0, noopFunc, SQLITE_FUNC_UNLIKELY),
drh54240752017-01-03 14:39:30 +00001922#ifdef SQLITE_DEBUG
drha1a523a2016-12-26 00:18:36 +00001923 FUNCTION2(affinity, 1, 0, 0, noopFunc, SQLITE_FUNC_AFFINITY),
drh54240752017-01-03 14:39:30 +00001924#endif
drh092457b2017-12-29 15:04:49 +00001925#ifdef SQLITE_ENABLE_OFFSET_SQL_FUNC
drh35100fb2018-01-04 19:20:37 +00001926 FUNCTION2(sqlite_offset, 1, 0, 0, noopFunc, SQLITE_FUNC_OFFSET|
drh2fc865c2017-12-16 20:20:37 +00001927 SQLITE_FUNC_TYPEOF),
drh092457b2017-12-29 15:04:49 +00001928#endif
danielk197793ce7412008-09-01 19:14:02 +00001929 FUNCTION(ltrim, 1, 1, 0, trimFunc ),
1930 FUNCTION(ltrim, 2, 1, 0, trimFunc ),
1931 FUNCTION(rtrim, 1, 2, 0, trimFunc ),
1932 FUNCTION(rtrim, 2, 2, 0, trimFunc ),
1933 FUNCTION(trim, 1, 3, 0, trimFunc ),
1934 FUNCTION(trim, 2, 3, 0, trimFunc ),
1935 FUNCTION(min, -1, 0, 1, minmaxFunc ),
1936 FUNCTION(min, 0, 0, 1, 0 ),
dan6fb2b542018-06-19 17:13:11 +00001937 WAGGREGATE(min, 1, 0, 1, minmaxStep, minMaxFinalize, minMaxValue, 0,
drh9588ad92014-09-15 14:46:02 +00001938 SQLITE_FUNC_MINMAX ),
danielk197793ce7412008-09-01 19:14:02 +00001939 FUNCTION(max, -1, 1, 1, minmaxFunc ),
1940 FUNCTION(max, 0, 1, 1, 0 ),
dan6fb2b542018-06-19 17:13:11 +00001941 WAGGREGATE(max, 1, 1, 1, minmaxStep, minMaxFinalize, minMaxValue, 0,
drh9588ad92014-09-15 14:46:02 +00001942 SQLITE_FUNC_MINMAX ),
drha748fdc2012-03-28 01:34:47 +00001943 FUNCTION2(typeof, 1, 0, 0, typeofFunc, SQLITE_FUNC_TYPEOF),
1944 FUNCTION2(length, 1, 0, 0, lengthFunc, SQLITE_FUNC_LENGTH),
drhd55e0722012-10-25 03:07:29 +00001945 FUNCTION(instr, 2, 0, 0, instrFunc ),
drha5c14162013-12-17 15:03:06 +00001946 FUNCTION(printf, -1, 0, 0, printfFunc ),
drhd495d8c2013-02-22 19:34:25 +00001947 FUNCTION(unicode, 1, 0, 0, unicodeFunc ),
1948 FUNCTION(char, -1, 0, 0, charFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001949 FUNCTION(abs, 1, 0, 0, absFunc ),
shanefbd60f82009-02-04 03:59:25 +00001950#ifndef SQLITE_OMIT_FLOATING_POINT
danielk197793ce7412008-09-01 19:14:02 +00001951 FUNCTION(round, 1, 0, 0, roundFunc ),
1952 FUNCTION(round, 2, 0, 0, roundFunc ),
shanefbd60f82009-02-04 03:59:25 +00001953#endif
danielk197793ce7412008-09-01 19:14:02 +00001954 FUNCTION(upper, 1, 0, 0, upperFunc ),
1955 FUNCTION(lower, 1, 0, 0, lowerFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001956 FUNCTION(hex, 1, 0, 0, hexFunc ),
drhcca9f3d2013-09-06 15:23:29 +00001957 FUNCTION2(ifnull, 2, 0, 0, noopFunc, SQLITE_FUNC_COALESCE),
drhb1fba282013-11-21 14:33:48 +00001958 VFUNCTION(random, 0, 0, 0, randomFunc ),
1959 VFUNCTION(randomblob, 1, 0, 0, randomBlob ),
danielk197793ce7412008-09-01 19:14:02 +00001960 FUNCTION(nullif, 2, 0, 1, nullifFunc ),
drh03bf26d2015-08-31 21:16:36 +00001961 DFUNCTION(sqlite_version, 0, 0, 0, versionFunc ),
1962 DFUNCTION(sqlite_source_id, 0, 0, 0, sourceidFunc ),
drh840561f2011-04-27 18:08:42 +00001963 FUNCTION(sqlite_log, 2, 0, 0, errlogFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001964 FUNCTION(quote, 1, 0, 0, quoteFunc ),
drhb1fba282013-11-21 14:33:48 +00001965 VFUNCTION(last_insert_rowid, 0, 0, 0, last_insert_rowid),
1966 VFUNCTION(changes, 0, 0, 0, changes ),
1967 VFUNCTION(total_changes, 0, 0, 0, total_changes ),
danielk197793ce7412008-09-01 19:14:02 +00001968 FUNCTION(replace, 3, 0, 0, replaceFunc ),
1969 FUNCTION(zeroblob, 1, 0, 0, zeroblobFunc ),
drh80738d92016-02-15 00:34:16 +00001970 FUNCTION(substr, 2, 0, 0, substrFunc ),
1971 FUNCTION(substr, 3, 0, 0, substrFunc ),
dan6fb2b542018-06-19 17:13:11 +00001972 WAGGREGATE(sum, 1,0,0, sumStep, sumFinalize, sumFinalize, sumInverse, 0),
1973 WAGGREGATE(total, 1,0,0, sumStep,totalFinalize,totalFinalize,sumInverse, 0),
1974 WAGGREGATE(avg, 1,0,0, sumStep, avgFinalize, avgFinalize, sumInverse, 0),
dan7262ca92018-07-02 12:07:32 +00001975 WAGGREGATE(count, 0,0,0, countStep,
1976 countFinalize, countFinalize, countInverse, SQLITE_FUNC_COUNT ),
1977 WAGGREGATE(count, 1,0,0, countStep,
1978 countFinalize, countFinalize, countInverse, 0 ),
dan03854d22018-06-08 11:45:28 +00001979 WAGGREGATE(group_concat, 1, 0, 0, groupConcatStep,
dan6fb2b542018-06-19 17:13:11 +00001980 groupConcatFinalize, groupConcatValue, groupConcatInverse, 0),
dan03854d22018-06-08 11:45:28 +00001981 WAGGREGATE(group_concat, 2, 0, 0, groupConcatStep,
dan6fb2b542018-06-19 17:13:11 +00001982 groupConcatFinalize, groupConcatValue, groupConcatInverse, 0),
danielk197793ce7412008-09-01 19:14:02 +00001983
1984 LIKEFUNC(glob, 2, &globInfo, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
drhcc153132016-08-04 12:35:17 +00001985#ifdef SQLITE_CASE_SENSITIVE_LIKE
danielk197793ce7412008-09-01 19:14:02 +00001986 LIKEFUNC(like, 2, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
1987 LIKEFUNC(like, 3, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
drhcc153132016-08-04 12:35:17 +00001988#else
danielk197793ce7412008-09-01 19:14:02 +00001989 LIKEFUNC(like, 2, &likeInfoNorm, SQLITE_FUNC_LIKE),
1990 LIKEFUNC(like, 3, &likeInfoNorm, SQLITE_FUNC_LIKE),
drhcc153132016-08-04 12:35:17 +00001991#endif
1992#ifdef SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION
1993 FUNCTION(unknown, -1, 0, 0, unknownFunc ),
1994#endif
drh80738d92016-02-15 00:34:16 +00001995 FUNCTION(coalesce, 1, 0, 0, 0 ),
1996 FUNCTION(coalesce, 0, 0, 0, 0 ),
1997 FUNCTION2(coalesce, -1, 0, 0, noopFunc, SQLITE_FUNC_COALESCE),
danielk197793ce7412008-09-01 19:14:02 +00001998 };
drh545f5872010-04-24 14:02:59 +00001999#ifndef SQLITE_OMIT_ALTERTABLE
2000 sqlite3AlterFunctions();
2001#endif
dandfa552f2018-06-02 21:04:28 +00002002 sqlite3WindowFunctions();
dan0106e372013-08-12 16:34:32 +00002003#if defined(SQLITE_ENABLE_STAT3) || defined(SQLITE_ENABLE_STAT4)
2004 sqlite3AnalyzeFunctions();
2005#endif
drh80738d92016-02-15 00:34:16 +00002006 sqlite3RegisterDateTimeFunctions();
2007 sqlite3InsertBuiltinFuncs(aBuiltinFunc, ArraySize(aBuiltinFunc));
2008
2009#if 0 /* Enable to print out how the built-in functions are hashed */
2010 {
2011 int i;
2012 FuncDef *p;
2013 for(i=0; i<SQLITE_FUNC_HASH_SZ; i++){
2014 printf("FUNC-HASH %02d:", i);
2015 for(p=sqlite3BuiltinFunctions.a[i]; p; p=p->u.pHash){
2016 int n = sqlite3Strlen30(p->zName);
2017 int h = p->zName[0] + n;
2018 printf(" %s(%d)", p->zName, h);
2019 }
2020 printf("\n");
2021 }
2022 }
2023#endif
drh70a8ca32008-08-21 18:49:27 +00002024}