blob: 3c5a8523076bbd77bfd9f92c5e70266043f5f23b [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>
drhef9f7192020-01-17 19:14:08 +000019#ifndef SQLITE_OMIT_FLOATING_POINT
drh05d7bfd2019-05-10 12:06:47 +000020#include <math.h>
drhef9f7192020-01-17 19:14:08 +000021#endif
danielk197788208052004-05-25 01:13:20 +000022#include "vdbeInt.h"
drh0bce8352002-02-28 00:41:10 +000023
drh55ef4d92005-08-14 01:20:37 +000024/*
25** Return the collating function associated with a function.
26*/
danielk1977dc1bdc42004-06-11 10:51:27 +000027static CollSeq *sqlite3GetFuncCollSeq(sqlite3_context *context){
drha9e03b12015-03-12 06:46:52 +000028 VdbeOp *pOp;
29 assert( context->pVdbe!=0 );
30 pOp = &context->pVdbe->aOp[context->iOp-1];
drha15cc472014-09-25 13:17:30 +000031 assert( pOp->opcode==OP_CollSeq );
32 assert( pOp->p4type==P4_COLLSEQ );
33 return pOp->p4.pColl;
danielk1977dc1bdc42004-06-11 10:51:27 +000034}
35
drh0bce8352002-02-28 00:41:10 +000036/*
drh7a957892012-02-02 17:35:43 +000037** Indicate that the accumulator load should be skipped on this
38** iteration of the aggregate loop.
39*/
40static void sqlite3SkipAccumulatorLoad(sqlite3_context *context){
drh21d59782018-01-23 19:24:54 +000041 assert( context->isError<=0 );
42 context->isError = -1;
drh7a957892012-02-02 17:35:43 +000043 context->skipFlag = 1;
44}
45
46/*
drh0bce8352002-02-28 00:41:10 +000047** Implementation of the non-aggregate min() and max() functions
48*/
drhf9b596e2004-05-26 16:54:42 +000049static void minmaxFunc(
50 sqlite3_context *context,
51 int argc,
52 sqlite3_value **argv
53){
drh0bce8352002-02-28 00:41:10 +000054 int i;
drh268380c2004-02-25 13:47:31 +000055 int mask; /* 0 for min() or 0xffffffff for max() */
drhf9b596e2004-05-26 16:54:42 +000056 int iBest;
danielk1977dc1bdc42004-06-11 10:51:27 +000057 CollSeq *pColl;
drh0bce8352002-02-28 00:41:10 +000058
drh65595cd2009-02-02 16:32:55 +000059 assert( argc>1 );
drhc44af712004-09-02 15:53:56 +000060 mask = sqlite3_user_data(context)==0 ? 0 : -1;
danielk1977dc1bdc42004-06-11 10:51:27 +000061 pColl = sqlite3GetFuncCollSeq(context);
62 assert( pColl );
danielk1977c572ef72004-05-27 09:28:41 +000063 assert( mask==-1 || mask==0 );
drhf9b596e2004-05-26 16:54:42 +000064 iBest = 0;
drh9c054832004-05-31 18:51:57 +000065 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drhf9b596e2004-05-26 16:54:42 +000066 for(i=1; i<argc; i++){
drh9c054832004-05-31 18:51:57 +000067 if( sqlite3_value_type(argv[i])==SQLITE_NULL ) return;
danielk1977dc1bdc42004-06-11 10:51:27 +000068 if( (sqlite3MemCompare(argv[iBest], argv[i], pColl)^mask)>=0 ){
drh65595cd2009-02-02 16:32:55 +000069 testcase( mask==0 );
drhf9b596e2004-05-26 16:54:42 +000070 iBest = i;
drh0bce8352002-02-28 00:41:10 +000071 }
72 }
drhf4479502004-05-27 03:12:53 +000073 sqlite3_result_value(context, argv[iBest]);
drh0bce8352002-02-28 00:41:10 +000074}
drh0bce8352002-02-28 00:41:10 +000075
drh268380c2004-02-25 13:47:31 +000076/*
77** Return the type of the argument.
78*/
drhf9b596e2004-05-26 16:54:42 +000079static void typeofFunc(
80 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +000081 int NotUsed,
drhf9b596e2004-05-26 16:54:42 +000082 sqlite3_value **argv
83){
drh9d8e4012017-07-06 13:51:50 +000084 static const char *azType[] = { "integer", "real", "text", "blob", "null" };
85 int i = sqlite3_value_type(argv[0]) - 1;
danielk197762c14b32008-11-19 09:05:26 +000086 UNUSED_PARAMETER(NotUsed);
drh9d8e4012017-07-06 13:51:50 +000087 assert( i>=0 && i<ArraySize(azType) );
88 assert( SQLITE_INTEGER==1 );
89 assert( SQLITE_FLOAT==2 );
90 assert( SQLITE_TEXT==3 );
91 assert( SQLITE_BLOB==4 );
92 assert( SQLITE_NULL==5 );
drh3cef3642017-07-14 19:22:08 +000093 /* EVIDENCE-OF: R-01470-60482 The sqlite3_value_type(V) interface returns
94 ** the datatype code for the initial datatype of the sqlite3_value object
95 ** V. The returned value is one of SQLITE_INTEGER, SQLITE_FLOAT,
96 ** SQLITE_TEXT, SQLITE_BLOB, or SQLITE_NULL. */
drh9d8e4012017-07-06 13:51:50 +000097 sqlite3_result_text(context, azType[i], -1, SQLITE_STATIC);
drh0bce8352002-02-28 00:41:10 +000098}
99
drh9d44f182022-01-09 16:54:02 +0000100/* subtype(X)
101**
102** Return the subtype of X
103*/
104static void subtypeFunc(
105 sqlite3_context *context,
106 int argc,
107 sqlite3_value **argv
108){
drh69b0ce32022-02-04 13:15:01 +0000109 UNUSED_PARAMETER(argc);
drh9d44f182022-01-09 16:54:02 +0000110 sqlite3_result_int(context, sqlite3_value_subtype(argv[0]));
111}
drh5708d2d2005-06-22 10:53:59 +0000112
113/*
drh0bce8352002-02-28 00:41:10 +0000114** Implementation of the length() function
115*/
drhf9b596e2004-05-26 16:54:42 +0000116static void lengthFunc(
117 sqlite3_context *context,
118 int argc,
119 sqlite3_value **argv
120){
drh0bce8352002-02-28 00:41:10 +0000121 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000122 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +0000123 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +0000124 case SQLITE_BLOB:
125 case SQLITE_INTEGER:
126 case SQLITE_FLOAT: {
drhf4479502004-05-27 03:12:53 +0000127 sqlite3_result_int(context, sqlite3_value_bytes(argv[0]));
drhf9b596e2004-05-26 16:54:42 +0000128 break;
129 }
drh9c054832004-05-31 18:51:57 +0000130 case SQLITE_TEXT: {
drh2646da72005-12-09 20:02:05 +0000131 const unsigned char *z = sqlite3_value_text(argv[0]);
drh7ea34692018-01-23 04:22:33 +0000132 const unsigned char *z0;
133 unsigned char c;
drh7a521cf2007-04-25 18:23:52 +0000134 if( z==0 ) return;
drh7ea34692018-01-23 04:22:33 +0000135 z0 = z;
136 while( (c = *z)!=0 ){
137 z++;
138 if( c>=0xc0 ){
139 while( (*z & 0xc0)==0x80 ){ z++; z0++; }
140 }
drh4a919112007-05-15 11:55:09 +0000141 }
drh7ea34692018-01-23 04:22:33 +0000142 sqlite3_result_int(context, (int)(z-z0));
drhf9b596e2004-05-26 16:54:42 +0000143 break;
144 }
145 default: {
146 sqlite3_result_null(context);
147 break;
148 }
149 }
drh0bce8352002-02-28 00:41:10 +0000150}
151
152/*
drh2ba3ccc2009-12-08 02:06:08 +0000153** Implementation of the abs() function.
154**
155** IMP: R-23979-26855 The abs(X) function returns the absolute value of
156** the numeric argument X.
drh0bce8352002-02-28 00:41:10 +0000157*/
danielk19770ae8b832004-05-25 12:05:56 +0000158static void absFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drh0bce8352002-02-28 00:41:10 +0000159 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000160 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +0000161 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +0000162 case SQLITE_INTEGER: {
danielk1977f93bbbe2004-05-27 10:30:52 +0000163 i64 iVal = sqlite3_value_int64(argv[0]);
drh52fc8492006-02-23 21:43:55 +0000164 if( iVal<0 ){
drh693e6712014-01-24 22:58:00 +0000165 if( iVal==SMALLEST_INT64 ){
drheb091cd2013-11-09 19:47:15 +0000166 /* IMP: R-31676-45509 If X is the integer -9223372036854775808
167 ** then abs(X) throws an integer overflow error since there is no
drh2ba3ccc2009-12-08 02:06:08 +0000168 ** equivalent positive 64-bit two complement value. */
drh52fc8492006-02-23 21:43:55 +0000169 sqlite3_result_error(context, "integer overflow", -1);
170 return;
171 }
172 iVal = -iVal;
173 }
danielk1977f93bbbe2004-05-27 10:30:52 +0000174 sqlite3_result_int64(context, iVal);
drhf9b596e2004-05-26 16:54:42 +0000175 break;
176 }
drh9c054832004-05-31 18:51:57 +0000177 case SQLITE_NULL: {
drh2ba3ccc2009-12-08 02:06:08 +0000178 /* IMP: R-37434-19929 Abs(X) returns NULL if X is NULL. */
drhf9b596e2004-05-26 16:54:42 +0000179 sqlite3_result_null(context);
180 break;
181 }
182 default: {
drh2ba3ccc2009-12-08 02:06:08 +0000183 /* Because sqlite3_value_double() returns 0.0 if the argument is not
184 ** something that can be converted into a number, we have:
drh643091f2014-11-20 23:21:23 +0000185 ** IMP: R-01992-00519 Abs(X) returns 0.0 if X is a string or blob
186 ** that cannot be converted to a numeric value.
drh2ba3ccc2009-12-08 02:06:08 +0000187 */
danielk1977f93bbbe2004-05-27 10:30:52 +0000188 double rVal = sqlite3_value_double(argv[0]);
drh52fc8492006-02-23 21:43:55 +0000189 if( rVal<0 ) rVal = -rVal;
danielk1977f93bbbe2004-05-27 10:30:52 +0000190 sqlite3_result_double(context, rVal);
drhf9b596e2004-05-26 16:54:42 +0000191 break;
192 }
193 }
drh0bce8352002-02-28 00:41:10 +0000194}
195
196/*
drhd55e0722012-10-25 03:07:29 +0000197** Implementation of the instr() function.
198**
199** instr(haystack,needle) finds the first occurrence of needle
200** in haystack and returns the number of previous characters plus 1,
201** or 0 if needle does not occur within haystack.
202**
203** If both haystack and needle are BLOBs, then the result is one more than
204** the number of bytes in haystack prior to the first occurrence of needle,
205** or 0 if needle never occurs in haystack.
206*/
207static void instrFunc(
208 sqlite3_context *context,
209 int argc,
210 sqlite3_value **argv
211){
212 const unsigned char *zHaystack;
213 const unsigned char *zNeedle;
214 int nHaystack;
215 int nNeedle;
216 int typeHaystack, typeNeedle;
217 int N = 1;
218 int isText;
drhc930b402019-01-08 15:18:24 +0000219 unsigned char firstChar;
drh97b02502019-09-17 03:16:29 +0000220 sqlite3_value *pC1 = 0;
221 sqlite3_value *pC2 = 0;
drhd55e0722012-10-25 03:07:29 +0000222
drh68c804b2012-12-04 11:03:11 +0000223 UNUSED_PARAMETER(argc);
drhd55e0722012-10-25 03:07:29 +0000224 typeHaystack = sqlite3_value_type(argv[0]);
225 typeNeedle = sqlite3_value_type(argv[1]);
226 if( typeHaystack==SQLITE_NULL || typeNeedle==SQLITE_NULL ) return;
227 nHaystack = sqlite3_value_bytes(argv[0]);
228 nNeedle = sqlite3_value_bytes(argv[1]);
dan895decf2016-12-30 14:15:56 +0000229 if( nNeedle>0 ){
230 if( typeHaystack==SQLITE_BLOB && typeNeedle==SQLITE_BLOB ){
231 zHaystack = sqlite3_value_blob(argv[0]);
232 zNeedle = sqlite3_value_blob(argv[1]);
233 isText = 0;
drh97b02502019-09-17 03:16:29 +0000234 }else if( typeHaystack!=SQLITE_BLOB && typeNeedle!=SQLITE_BLOB ){
dan895decf2016-12-30 14:15:56 +0000235 zHaystack = sqlite3_value_text(argv[0]);
236 zNeedle = sqlite3_value_text(argv[1]);
237 isText = 1;
drh97b02502019-09-17 03:16:29 +0000238 }else{
239 pC1 = sqlite3_value_dup(argv[0]);
240 zHaystack = sqlite3_value_text(pC1);
drh9d702842019-09-18 11:16:46 +0000241 if( zHaystack==0 ) goto endInstrOOM;
242 nHaystack = sqlite3_value_bytes(pC1);
drh97b02502019-09-17 03:16:29 +0000243 pC2 = sqlite3_value_dup(argv[1]);
244 zNeedle = sqlite3_value_text(pC2);
drh9d702842019-09-18 11:16:46 +0000245 if( zNeedle==0 ) goto endInstrOOM;
246 nNeedle = sqlite3_value_bytes(pC2);
drh97b02502019-09-17 03:16:29 +0000247 isText = 1;
dan895decf2016-12-30 14:15:56 +0000248 }
drh9d702842019-09-18 11:16:46 +0000249 if( zNeedle==0 || (nHaystack && zHaystack==0) ) goto endInstrOOM;
drhc930b402019-01-08 15:18:24 +0000250 firstChar = zNeedle[0];
251 while( nNeedle<=nHaystack
252 && (zHaystack[0]!=firstChar || memcmp(zHaystack, zNeedle, nNeedle)!=0)
253 ){
dan895decf2016-12-30 14:15:56 +0000254 N++;
255 do{
256 nHaystack--;
257 zHaystack++;
258 }while( isText && (zHaystack[0]&0xc0)==0x80 );
259 }
260 if( nNeedle>nHaystack ) N = 0;
drhd55e0722012-10-25 03:07:29 +0000261 }
drhd55e0722012-10-25 03:07:29 +0000262 sqlite3_result_int(context, N);
drh97b02502019-09-17 03:16:29 +0000263endInstr:
264 sqlite3_value_free(pC1);
265 sqlite3_value_free(pC2);
drh9d702842019-09-18 11:16:46 +0000266 return;
267endInstrOOM:
268 sqlite3_result_error_nomem(context);
269 goto endInstr;
drhd55e0722012-10-25 03:07:29 +0000270}
271
272/*
drh6bcd5852022-01-08 21:00:38 +0000273** Implementation of the printf() (a.k.a. format()) SQL function.
drha5c14162013-12-17 15:03:06 +0000274*/
275static void printfFunc(
276 sqlite3_context *context,
277 int argc,
278 sqlite3_value **argv
279){
280 PrintfArguments x;
281 StrAccum str;
282 const char *zFormat;
283 int n;
drhc0490572015-05-02 11:45:53 +0000284 sqlite3 *db = sqlite3_context_db_handle(context);
drha5c14162013-12-17 15:03:06 +0000285
286 if( argc>=1 && (zFormat = (const char*)sqlite3_value_text(argv[0]))!=0 ){
287 x.nArg = argc-1;
288 x.nUsed = 0;
289 x.apArg = argv+1;
drhc0490572015-05-02 11:45:53 +0000290 sqlite3StrAccumInit(&str, db, 0, 0, db->aLimit[SQLITE_LIMIT_LENGTH]);
drh5f4a6862016-01-30 12:50:25 +0000291 str.printfFlags = SQLITE_PRINTF_SQLFUNC;
drh0cdbe1a2018-05-09 13:46:26 +0000292 sqlite3_str_appendf(&str, zFormat, &x);
drha5c14162013-12-17 15:03:06 +0000293 n = str.nChar;
294 sqlite3_result_text(context, sqlite3StrAccumFinish(&str), n,
295 SQLITE_DYNAMIC);
296 }
297}
298
299/*
drhf764e6f2007-05-15 01:13:47 +0000300** Implementation of the substr() function.
301**
302** substr(x,p1,p2) returns p2 characters of x[] beginning with p1.
303** p1 is 1-indexed. So substr(x,1,1) returns the first character
304** of x. If x is text, then we actually count UTF-8 characters.
305** If x is a blob, then we count bytes.
306**
307** If p1 is negative, then we begin abs(p1) from the end of x[].
shaneh779b8f12009-11-12 05:04:50 +0000308**
drhf7b54962013-05-28 12:11:54 +0000309** If p2 is negative, return the p2 characters preceding p1.
drh0bce8352002-02-28 00:41:10 +0000310*/
drhf9b596e2004-05-26 16:54:42 +0000311static void substrFunc(
312 sqlite3_context *context,
313 int argc,
314 sqlite3_value **argv
315){
drh2646da72005-12-09 20:02:05 +0000316 const unsigned char *z;
317 const unsigned char *z2;
drh023ae032007-05-08 12:12:16 +0000318 int len;
drhf764e6f2007-05-15 01:13:47 +0000319 int p0type;
drh023ae032007-05-08 12:12:16 +0000320 i64 p1, p2;
drh65595cd2009-02-02 16:32:55 +0000321 int negP2 = 0;
drhf9b596e2004-05-26 16:54:42 +0000322
drh64f31512007-10-12 19:11:55 +0000323 assert( argc==3 || argc==2 );
drh8198d252009-02-01 19:42:37 +0000324 if( sqlite3_value_type(argv[1])==SQLITE_NULL
325 || (argc==3 && sqlite3_value_type(argv[2])==SQLITE_NULL)
326 ){
327 return;
328 }
drhf764e6f2007-05-15 01:13:47 +0000329 p0type = sqlite3_value_type(argv[0]);
drh4adc4cb2009-11-11 20:53:31 +0000330 p1 = sqlite3_value_int(argv[1]);
drhf764e6f2007-05-15 01:13:47 +0000331 if( p0type==SQLITE_BLOB ){
332 len = sqlite3_value_bytes(argv[0]);
333 z = sqlite3_value_blob(argv[0]);
334 if( z==0 ) return;
drh1f0feef2007-05-15 13:27:07 +0000335 assert( len==sqlite3_value_bytes(argv[0]) );
drhf764e6f2007-05-15 01:13:47 +0000336 }else{
337 z = sqlite3_value_text(argv[0]);
338 if( z==0 ) return;
drh4a919112007-05-15 11:55:09 +0000339 len = 0;
drh4adc4cb2009-11-11 20:53:31 +0000340 if( p1<0 ){
341 for(z2=z; *z2; len++){
342 SQLITE_SKIP_UTF8(z2);
343 }
drh4a919112007-05-15 11:55:09 +0000344 }
drhf764e6f2007-05-15 01:13:47 +0000345 }
drh883ad042015-02-19 00:29:11 +0000346#ifdef SQLITE_SUBSTR_COMPATIBILITY
347 /* If SUBSTR_COMPATIBILITY is defined then substr(X,0,N) work the same as
348 ** as substr(X,1,N) - it returns the first N characters of X. This
349 ** is essentially a back-out of the bug-fix in check-in [5fc125d362df4b8]
350 ** from 2009-02-02 for compatibility of applications that exploited the
351 ** old buggy behavior. */
352 if( p1==0 ) p1 = 1; /* <rdar://problem/6778339> */
353#endif
drh64f31512007-10-12 19:11:55 +0000354 if( argc==3 ){
355 p2 = sqlite3_value_int(argv[2]);
drh65595cd2009-02-02 16:32:55 +0000356 if( p2<0 ){
357 p2 = -p2;
358 negP2 = 1;
359 }
drh64f31512007-10-12 19:11:55 +0000360 }else{
drhbb4957f2008-03-20 14:03:29 +0000361 p2 = sqlite3_context_db_handle(context)->aLimit[SQLITE_LIMIT_LENGTH];
drh64f31512007-10-12 19:11:55 +0000362 }
drh0bce8352002-02-28 00:41:10 +0000363 if( p1<0 ){
drh89425d52002-02-28 03:04:48 +0000364 p1 += len;
drh653bc752002-02-28 03:31:10 +0000365 if( p1<0 ){
366 p2 += p1;
drh65595cd2009-02-02 16:32:55 +0000367 if( p2<0 ) p2 = 0;
drh653bc752002-02-28 03:31:10 +0000368 p1 = 0;
369 }
drh0bce8352002-02-28 00:41:10 +0000370 }else if( p1>0 ){
371 p1--;
drh65595cd2009-02-02 16:32:55 +0000372 }else if( p2>0 ){
373 p2--;
drh0bce8352002-02-28 00:41:10 +0000374 }
drh65595cd2009-02-02 16:32:55 +0000375 if( negP2 ){
376 p1 -= p2;
drh4e79c592009-02-01 19:23:32 +0000377 if( p1<0 ){
378 p2 += p1;
379 p1 = 0;
380 }
381 }
drh65595cd2009-02-02 16:32:55 +0000382 assert( p1>=0 && p2>=0 );
drhf764e6f2007-05-15 01:13:47 +0000383 if( p0type!=SQLITE_BLOB ){
drh4a919112007-05-15 11:55:09 +0000384 while( *z && p1 ){
385 SQLITE_SKIP_UTF8(z);
386 p1--;
drhf764e6f2007-05-15 01:13:47 +0000387 }
drh4a919112007-05-15 11:55:09 +0000388 for(z2=z; *z2 && p2; p2--){
389 SQLITE_SKIP_UTF8(z2);
drhf764e6f2007-05-15 01:13:47 +0000390 }
drhbbf483f2014-09-09 20:30:24 +0000391 sqlite3_result_text64(context, (char*)z, z2-z, SQLITE_TRANSIENT,
392 SQLITE_UTF8);
drhf764e6f2007-05-15 01:13:47 +0000393 }else{
drh4adc4cb2009-11-11 20:53:31 +0000394 if( p1+p2>len ){
395 p2 = len-p1;
396 if( p2<0 ) p2 = 0;
397 }
drhbbf483f2014-09-09 20:30:24 +0000398 sqlite3_result_blob64(context, (char*)&z[p1], (u64)p2, SQLITE_TRANSIENT);
drh0bce8352002-02-28 00:41:10 +0000399 }
drh0bce8352002-02-28 00:41:10 +0000400}
401
402/*
403** Implementation of the round() function
404*/
shanefbd60f82009-02-04 03:59:25 +0000405#ifndef SQLITE_OMIT_FLOATING_POINT
danielk19770ae8b832004-05-25 12:05:56 +0000406static void roundFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
danielk197751ad0ec2004-05-24 12:39:02 +0000407 int n = 0;
drh0bce8352002-02-28 00:41:10 +0000408 double r;
drh50d654d2009-06-03 01:24:54 +0000409 char *zBuf;
drh0bce8352002-02-28 00:41:10 +0000410 assert( argc==1 || argc==2 );
danielk197751ad0ec2004-05-24 12:39:02 +0000411 if( argc==2 ){
drh9c054832004-05-31 18:51:57 +0000412 if( SQLITE_NULL==sqlite3_value_type(argv[1]) ) return;
danielk197751ad0ec2004-05-24 12:39:02 +0000413 n = sqlite3_value_int(argv[1]);
414 if( n>30 ) n = 30;
415 if( n<0 ) n = 0;
416 }
drhd589a922006-03-02 03:02:48 +0000417 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drh4f26d6c2004-05-26 23:25:30 +0000418 r = sqlite3_value_double(argv[0]);
shaneh147e1762010-02-17 04:19:27 +0000419 /* If Y==0 and X will fit in a 64-bit int,
420 ** handle the rounding directly,
421 ** otherwise use printf.
422 */
drh84422db2019-05-30 13:47:10 +0000423 if( r<-4503599627370496.0 || r>+4503599627370496.0 ){
424 /* The value has no fractional part so there is nothing to round */
425 }else if( n==0 ){
426 r = (double)((sqlite_int64)(r+(r<0?-0.5:+0.5)));
drh50d654d2009-06-03 01:24:54 +0000427 }else{
shaneh147e1762010-02-17 04:19:27 +0000428 zBuf = sqlite3_mprintf("%.*f",n,r);
429 if( zBuf==0 ){
430 sqlite3_result_error_nomem(context);
431 return;
432 }
drh55700bc2019-06-07 22:51:13 +0000433 sqlite3AtoF(zBuf, &r, sqlite3Strlen30(zBuf), SQLITE_UTF8);
drh50d654d2009-06-03 01:24:54 +0000434 sqlite3_free(zBuf);
drh50d654d2009-06-03 01:24:54 +0000435 }
shaneh147e1762010-02-17 04:19:27 +0000436 sqlite3_result_double(context, r);
drh0bce8352002-02-28 00:41:10 +0000437}
shanefbd60f82009-02-04 03:59:25 +0000438#endif
drhdc04c582002-02-24 01:55:15 +0000439
danielk197726783a52007-08-29 14:06:22 +0000440/*
drhf3cdcdc2015-04-29 16:50:28 +0000441** Allocate nByte bytes of space using sqlite3Malloc(). If the
danielk197726783a52007-08-29 14:06:22 +0000442** allocation fails, call sqlite3_result_error_nomem() to notify
drh27e62db2009-04-02 10:16:17 +0000443** the database handle that malloc() has failed and return NULL.
444** If nByte is larger than the maximum string or blob length, then
445** raise an SQLITE_TOOBIG exception and return NULL.
danielk197726783a52007-08-29 14:06:22 +0000446*/
drhb1a6c3c2008-03-20 16:30:17 +0000447static void *contextMalloc(sqlite3_context *context, i64 nByte){
drhbb4957f2008-03-20 14:03:29 +0000448 char *z;
drh27e62db2009-04-02 10:16:17 +0000449 sqlite3 *db = sqlite3_context_db_handle(context);
drhef31c6a2009-04-02 09:07:12 +0000450 assert( nByte>0 );
drh27e62db2009-04-02 10:16:17 +0000451 testcase( nByte==db->aLimit[SQLITE_LIMIT_LENGTH] );
452 testcase( nByte==db->aLimit[SQLITE_LIMIT_LENGTH]+1 );
453 if( nByte>db->aLimit[SQLITE_LIMIT_LENGTH] ){
drhbb4957f2008-03-20 14:03:29 +0000454 sqlite3_result_error_toobig(context);
455 z = 0;
456 }else{
drhda4ca9d2014-09-09 17:27:35 +0000457 z = sqlite3Malloc(nByte);
drhef31c6a2009-04-02 09:07:12 +0000458 if( !z ){
drhbb4957f2008-03-20 14:03:29 +0000459 sqlite3_result_error_nomem(context);
460 }
danielk1977a1644fd2007-08-29 12:31:25 +0000461 }
462 return z;
463}
464
drhdc04c582002-02-24 01:55:15 +0000465/*
466** Implementation of the upper() and lower() SQL functions.
467*/
danielk19770ae8b832004-05-25 12:05:56 +0000468static void upperFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drh7a521cf2007-04-25 18:23:52 +0000469 char *z1;
470 const char *z2;
drh9310ef22007-04-27 17:16:20 +0000471 int i, n;
drh1d34fde2009-02-03 15:50:33 +0000472 UNUSED_PARAMETER(argc);
drh7a521cf2007-04-25 18:23:52 +0000473 z2 = (char*)sqlite3_value_text(argv[0]);
drh1f0feef2007-05-15 13:27:07 +0000474 n = sqlite3_value_bytes(argv[0]);
475 /* Verify that the call to _bytes() does not invalidate the _text() pointer */
476 assert( z2==(char*)sqlite3_value_text(argv[0]) );
drh7a521cf2007-04-25 18:23:52 +0000477 if( z2 ){
drhb1a6c3c2008-03-20 16:30:17 +0000478 z1 = contextMalloc(context, ((i64)n)+1);
drh7a521cf2007-04-25 18:23:52 +0000479 if( z1 ){
drhdf901d32011-10-13 18:00:11 +0000480 for(i=0; i<n; i++){
481 z1[i] = (char)sqlite3Toupper(z2[i]);
drh7a521cf2007-04-25 18:23:52 +0000482 }
drhdf901d32011-10-13 18:00:11 +0000483 sqlite3_result_text(context, z1, n, sqlite3_free);
drh7a521cf2007-04-25 18:23:52 +0000484 }
drhdc04c582002-02-24 01:55:15 +0000485 }
486}
danielk19770ae8b832004-05-25 12:05:56 +0000487static void lowerFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drhdf901d32011-10-13 18:00:11 +0000488 char *z1;
drh7a521cf2007-04-25 18:23:52 +0000489 const char *z2;
drh9310ef22007-04-27 17:16:20 +0000490 int i, n;
drh1d34fde2009-02-03 15:50:33 +0000491 UNUSED_PARAMETER(argc);
drh7a521cf2007-04-25 18:23:52 +0000492 z2 = (char*)sqlite3_value_text(argv[0]);
drh1f0feef2007-05-15 13:27:07 +0000493 n = sqlite3_value_bytes(argv[0]);
494 /* Verify that the call to _bytes() does not invalidate the _text() pointer */
495 assert( z2==(char*)sqlite3_value_text(argv[0]) );
drh7a521cf2007-04-25 18:23:52 +0000496 if( z2 ){
drhb1a6c3c2008-03-20 16:30:17 +0000497 z1 = contextMalloc(context, ((i64)n)+1);
drh7a521cf2007-04-25 18:23:52 +0000498 if( z1 ){
drhdf901d32011-10-13 18:00:11 +0000499 for(i=0; i<n; i++){
500 z1[i] = sqlite3Tolower(z2[i]);
drh7a521cf2007-04-25 18:23:52 +0000501 }
drhdf901d32011-10-13 18:00:11 +0000502 sqlite3_result_text(context, z1, n, sqlite3_free);
drh7a521cf2007-04-25 18:23:52 +0000503 }
drhdc04c582002-02-24 01:55:15 +0000504 }
505}
506
drhae6bb952009-11-11 00:24:31 +0000507/*
drhcca9f3d2013-09-06 15:23:29 +0000508** Some functions like COALESCE() and IFNULL() and UNLIKELY() are implemented
509** as VDBE code so that unused argument values do not have to be computed.
510** However, we still need some kind of function implementation for this
511** routines in the function table. The noopFunc macro provides this.
512** noopFunc will never be called so it doesn't matter what the implementation
513** is. We might as well use the "version()" function as a substitute.
drhae6bb952009-11-11 00:24:31 +0000514*/
drhcca9f3d2013-09-06 15:23:29 +0000515#define noopFunc versionFunc /* Substitute function - never called */
drh3212e182002-02-28 00:46:26 +0000516
517/*
drhf9ffac92002-03-02 19:00:31 +0000518** Implementation of random(). Return a random integer.
519*/
drhf9b596e2004-05-26 16:54:42 +0000520static void randomFunc(
521 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000522 int NotUsed,
523 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000524){
drh52fc8492006-02-23 21:43:55 +0000525 sqlite_int64 r;
danielk197762c14b32008-11-19 09:05:26 +0000526 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drh2fa18682008-03-19 14:15:34 +0000527 sqlite3_randomness(sizeof(r), &r);
drh3034e3d2009-04-02 14:05:21 +0000528 if( r<0 ){
529 /* We need to prevent a random number of 0x8000000000000000
530 ** (or -9223372036854775808) since when you do abs() of that
531 ** number of you get the same value back again. To do this
532 ** in a way that is testable, mask the sign bit off of negative
533 ** values, resulting in a positive value. Then take the
534 ** 2s complement of that positive value. The end result can
535 ** therefore be no less than -9223372036854775807.
536 */
drhaf8001b2012-02-11 19:53:24 +0000537 r = -(r & LARGEST_INT64);
drh3034e3d2009-04-02 14:05:21 +0000538 }
drh52fc8492006-02-23 21:43:55 +0000539 sqlite3_result_int64(context, r);
drhf9ffac92002-03-02 19:00:31 +0000540}
541
542/*
drh137c7282007-01-29 17:58:28 +0000543** Implementation of randomblob(N). Return a random blob
544** that is N bytes long.
drh63cf66f2007-01-29 15:50:05 +0000545*/
drh137c7282007-01-29 17:58:28 +0000546static void randomBlob(
drh63cf66f2007-01-29 15:50:05 +0000547 sqlite3_context *context,
548 int argc,
549 sqlite3_value **argv
550){
drh3cb79202019-01-18 14:53:15 +0000551 sqlite3_int64 n;
drh137c7282007-01-29 17:58:28 +0000552 unsigned char *p;
drh63cf66f2007-01-29 15:50:05 +0000553 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000554 UNUSED_PARAMETER(argc);
drh3cb79202019-01-18 14:53:15 +0000555 n = sqlite3_value_int64(argv[0]);
drh023ae032007-05-08 12:12:16 +0000556 if( n<1 ){
557 n = 1;
558 }
danielk1977a1644fd2007-08-29 12:31:25 +0000559 p = contextMalloc(context, n);
drh02d85832007-05-07 19:31:15 +0000560 if( p ){
drh2fa18682008-03-19 14:15:34 +0000561 sqlite3_randomness(n, p);
drh17435752007-08-16 04:30:38 +0000562 sqlite3_result_blob(context, (char*)p, n, sqlite3_free);
drh02d85832007-05-07 19:31:15 +0000563 }
drh63cf66f2007-01-29 15:50:05 +0000564}
565
566/*
drh6ed41ad2002-04-06 14:10:47 +0000567** Implementation of the last_insert_rowid() SQL function. The return
danielk197724b03fd2004-05-10 10:34:34 +0000568** value is the same as the sqlite3_last_insert_rowid() API function.
drh6ed41ad2002-04-06 14:10:47 +0000569*/
danielk197751ad0ec2004-05-24 12:39:02 +0000570static void last_insert_rowid(
danielk19770ae8b832004-05-25 12:05:56 +0000571 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-51513-12026 The last_insert_rowid() SQL function is a
578 ** wrapper around the sqlite3_last_insert_rowid() C/C++ interface
579 ** function. */
drhf9b596e2004-05-26 16:54:42 +0000580 sqlite3_result_int64(context, sqlite3_last_insert_rowid(db));
drh6ed41ad2002-04-06 14:10:47 +0000581}
582
rdcf146a772004-02-25 22:51:06 +0000583/*
drhab2f1f92010-01-11 18:26:42 +0000584** Implementation of the changes() SQL function.
585**
drhc0bd26a2021-09-14 18:57:30 +0000586** IMP: R-32760-32347 The changes() SQL function is a wrapper
587** around the sqlite3_changes64() C/C++ function and hence follows the
588** same rules for counting changes.
rdcf146a772004-02-25 22:51:06 +0000589*/
danielk1977b28af712004-06-21 06:50:26 +0000590static void changes(
drhf9b596e2004-05-26 16:54:42 +0000591 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000592 int NotUsed,
593 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000594){
drhfa4a4b92008-03-19 21:45:51 +0000595 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000596 UNUSED_PARAMETER2(NotUsed, NotUsed2);
dan2c718872021-06-22 18:32:05 +0000597 sqlite3_result_int64(context, sqlite3_changes64(db));
rdcb0c374f2004-02-20 22:53:38 +0000598}
rdcf146a772004-02-25 22:51:06 +0000599
600/*
danielk1977b28af712004-06-21 06:50:26 +0000601** Implementation of the total_changes() SQL function. The return value is
larrybr10496f72021-06-23 16:07:20 +0000602** the same as the sqlite3_total_changes64() API function.
rdcf146a772004-02-25 22:51:06 +0000603*/
danielk1977b28af712004-06-21 06:50:26 +0000604static void total_changes(
605 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000606 int NotUsed,
607 sqlite3_value **NotUsed2
danielk197751ad0ec2004-05-24 12:39:02 +0000608){
drhfa4a4b92008-03-19 21:45:51 +0000609 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000610 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhc0bd26a2021-09-14 18:57:30 +0000611 /* IMP: R-11217-42568 This function is a wrapper around the
612 ** sqlite3_total_changes64() C/C++ interface. */
larrybr10496f72021-06-23 16:07:20 +0000613 sqlite3_result_int64(context, sqlite3_total_changes64(db));
rdcb0c374f2004-02-20 22:53:38 +0000614}
615
drh6ed41ad2002-04-06 14:10:47 +0000616/*
drh4e5ffc52004-08-31 00:52:37 +0000617** A structure defining how to do GLOB-style comparisons.
danielk1977d02eb1f2004-06-06 09:44:03 +0000618*/
drh4e5ffc52004-08-31 00:52:37 +0000619struct compareInfo {
drh07e83472016-01-11 03:48:18 +0000620 u8 matchAll; /* "*" or "%" */
621 u8 matchOne; /* "?" or "_" */
622 u8 matchSet; /* "[" or 0 */
623 u8 noCase; /* true to ignore case differences */
danielk1977d02eb1f2004-06-06 09:44:03 +0000624};
drh55ef4d92005-08-14 01:20:37 +0000625
drhb9175ae2007-12-07 18:39:04 +0000626/*
627** For LIKE and GLOB matching on EBCDIC machines, assume that every
drhb0870482015-06-17 13:20:54 +0000628** character is exactly one byte in size. Also, provde the Utf8Read()
629** macro for fast reading of the next character in the common case where
630** the next character is ASCII.
drhb9175ae2007-12-07 18:39:04 +0000631*/
632#if defined(SQLITE_EBCDIC)
drh88b33222014-09-25 03:51:37 +0000633# define sqlite3Utf8Read(A) (*((*A)++))
drhb0870482015-06-17 13:20:54 +0000634# define Utf8Read(A) (*(A++))
drhb9175ae2007-12-07 18:39:04 +0000635#else
drhb0870482015-06-17 13:20:54 +0000636# define Utf8Read(A) (A[0]<0x80?*(A++):sqlite3Utf8Read(&A))
drhb9175ae2007-12-07 18:39:04 +0000637#endif
638
drh4e5ffc52004-08-31 00:52:37 +0000639static const struct compareInfo globInfo = { '*', '?', '[', 0 };
drh55ef4d92005-08-14 01:20:37 +0000640/* The correct SQL-92 behavior is for the LIKE operator to ignore
641** case. Thus 'a' LIKE 'A' would be true. */
642static const struct compareInfo likeInfoNorm = { '%', '_', 0, 1 };
643/* If SQLITE_CASE_SENSITIVE_LIKE is defined, then the LIKE operator
644** is case sensitive causing 'a' LIKE 'A' to be false */
645static const struct compareInfo likeInfoAlt = { '%', '_', 0, 0 };
danielk1977d02eb1f2004-06-06 09:44:03 +0000646
647/*
drh698a01c2016-12-01 18:49:40 +0000648** Possible error returns from patternMatch()
649*/
650#define SQLITE_MATCH 0
651#define SQLITE_NOMATCH 1
652#define SQLITE_NOWILDCARDMATCH 2
653
654/*
655** Compare two UTF-8 strings for equality where the first string is
656** a GLOB or LIKE expression. Return values:
657**
658** SQLITE_MATCH: Match
659** SQLITE_NOMATCH: No match
660** SQLITE_NOWILDCARDMATCH: No match in spite of having * or % wildcards.
drh0ac65892002-04-20 14:24:41 +0000661**
drh4e5ffc52004-08-31 00:52:37 +0000662** Globbing rules:
drh0ac65892002-04-20 14:24:41 +0000663**
drh4e5ffc52004-08-31 00:52:37 +0000664** '*' Matches any sequence of zero or more characters.
danielk1977d02eb1f2004-06-06 09:44:03 +0000665**
drh4e5ffc52004-08-31 00:52:37 +0000666** '?' Matches exactly one character.
667**
668** [...] Matches one character from the enclosed list of
669** characters.
670**
671** [^...] Matches one character not in the enclosed list.
672**
673** With the [...] and [^...] matching, a ']' character can be included
674** in the list by making it the first character after '[' or '^'. A
675** range of characters can be specified using '-'. Example:
676** "[a-z]" matches any single lower-case letter. To match a '-', make
677** it the last character in the list.
678**
drh9fdfdc82014-09-25 11:08:57 +0000679** Like matching rules:
680**
681** '%' Matches any sequence of zero or more characters
682**
683*** '_' Matches any one character
684**
685** Ec Where E is the "esc" character and c is any other
686** character, including '%', '_', and esc, match exactly c.
687**
drhb0870482015-06-17 13:20:54 +0000688** The comments within this routine usually assume glob matching.
drh9fdfdc82014-09-25 11:08:57 +0000689**
drh4e5ffc52004-08-31 00:52:37 +0000690** This routine is usually quick, but can be N**2 in the worst case.
drh0ac65892002-04-20 14:24:41 +0000691*/
danielk19777c6303c2004-11-17 16:41:29 +0000692static int patternCompare(
drh4e5ffc52004-08-31 00:52:37 +0000693 const u8 *zPattern, /* The glob pattern */
694 const u8 *zString, /* The string to compare against the glob */
danielk19777c6303c2004-11-17 16:41:29 +0000695 const struct compareInfo *pInfo, /* Information about how to do the compare */
drh698a01c2016-12-01 18:49:40 +0000696 u32 matchOther /* The escape char (LIKE) or '[' (GLOB) */
danielk197751ad0ec2004-05-24 12:39:02 +0000697){
drh9fdfdc82014-09-25 11:08:57 +0000698 u32 c, c2; /* Next pattern and input string chars */
699 u32 matchOne = pInfo->matchOne; /* "?" or "_" */
700 u32 matchAll = pInfo->matchAll; /* "*" or "%" */
drh9fdfdc82014-09-25 11:08:57 +0000701 u8 noCase = pInfo->noCase; /* True if uppercase==lowercase */
702 const u8 *zEscaped = 0; /* One past the last escaped input char */
drh88b33222014-09-25 03:51:37 +0000703
drhb0870482015-06-17 13:20:54 +0000704 while( (c = Utf8Read(zPattern))!=0 ){
drh9fdfdc82014-09-25 11:08:57 +0000705 if( c==matchAll ){ /* Match "*" */
706 /* Skip over multiple "*" characters in the pattern. If there
707 ** are also "?" characters, skip those as well, but consume a
708 ** single character of the input string for each "?" skipped */
drh33941692021-02-15 17:02:01 +0000709 while( (c=Utf8Read(zPattern)) == matchAll
710 || (c == matchOne && matchOne!=0) ){
drh42610962012-09-17 18:56:32 +0000711 if( c==matchOne && sqlite3Utf8Read(&zString)==0 ){
drh698a01c2016-12-01 18:49:40 +0000712 return SQLITE_NOWILDCARDMATCH;
drh4e5ffc52004-08-31 00:52:37 +0000713 }
danielk1977ad7dd422004-06-06 12:41:49 +0000714 }
drh66150952007-07-23 19:12:41 +0000715 if( c==0 ){
drh698a01c2016-12-01 18:49:40 +0000716 return SQLITE_MATCH; /* "*" at the end of the pattern matches */
drh88b33222014-09-25 03:51:37 +0000717 }else if( c==matchOther ){
drh07e83472016-01-11 03:48:18 +0000718 if( pInfo->matchSet==0 ){
drh88b33222014-09-25 03:51:37 +0000719 c = sqlite3Utf8Read(&zPattern);
drh698a01c2016-12-01 18:49:40 +0000720 if( c==0 ) return SQLITE_NOWILDCARDMATCH;
drh88b33222014-09-25 03:51:37 +0000721 }else{
drh9fdfdc82014-09-25 11:08:57 +0000722 /* "[...]" immediately follows the "*". We have to do a slow
723 ** recursive search in this case, but it is an unusual case. */
drh88b33222014-09-25 03:51:37 +0000724 assert( matchOther<0x80 ); /* '[' is a single-byte character */
drh698a01c2016-12-01 18:49:40 +0000725 while( *zString ){
726 int bMatch = patternCompare(&zPattern[-1],zString,pInfo,matchOther);
727 if( bMatch!=SQLITE_NOMATCH ) return bMatch;
drh88b33222014-09-25 03:51:37 +0000728 SQLITE_SKIP_UTF8(zString);
729 }
drh698a01c2016-12-01 18:49:40 +0000730 return SQLITE_NOWILDCARDMATCH;
drh66150952007-07-23 19:12:41 +0000731 }
drh66150952007-07-23 19:12:41 +0000732 }
drh9fdfdc82014-09-25 11:08:57 +0000733
734 /* At this point variable c contains the first character of the
735 ** pattern string past the "*". Search in the input string for the
dan1a4a7372016-12-01 17:34:59 +0000736 ** first matching character and recursively continue the match from
drh9fdfdc82014-09-25 11:08:57 +0000737 ** that point.
738 **
739 ** For a case-insensitive search, set variable cx to be the same as
740 ** c but in the other case and search the input string for either
741 ** c or cx.
742 */
dan2bc4a6c2022-10-14 15:10:36 +0000743 if( c<0x80 ){
drheba21f92017-10-30 18:49:11 +0000744 char zStop[3];
drh698a01c2016-12-01 18:49:40 +0000745 int bMatch;
drh9fdfdc82014-09-25 11:08:57 +0000746 if( noCase ){
drheba21f92017-10-30 18:49:11 +0000747 zStop[0] = sqlite3Toupper(c);
748 zStop[1] = sqlite3Tolower(c);
749 zStop[2] = 0;
drh66150952007-07-23 19:12:41 +0000750 }else{
drheba21f92017-10-30 18:49:11 +0000751 zStop[0] = c;
752 zStop[1] = 0;
drh4e5ffc52004-08-31 00:52:37 +0000753 }
drheba21f92017-10-30 18:49:11 +0000754 while(1){
755 zString += strcspn((const char*)zString, zStop);
756 if( zString[0]==0 ) break;
757 zString++;
drh698a01c2016-12-01 18:49:40 +0000758 bMatch = patternCompare(zPattern,zString,pInfo,matchOther);
759 if( bMatch!=SQLITE_NOMATCH ) return bMatch;
drh9fdfdc82014-09-25 11:08:57 +0000760 }
761 }else{
drh698a01c2016-12-01 18:49:40 +0000762 int bMatch;
drhb0870482015-06-17 13:20:54 +0000763 while( (c2 = Utf8Read(zString))!=0 ){
drh9fdfdc82014-09-25 11:08:57 +0000764 if( c2!=c ) continue;
drh698a01c2016-12-01 18:49:40 +0000765 bMatch = patternCompare(zPattern,zString,pInfo,matchOther);
766 if( bMatch!=SQLITE_NOMATCH ) return bMatch;
drh9fdfdc82014-09-25 11:08:57 +0000767 }
drh66150952007-07-23 19:12:41 +0000768 }
drh698a01c2016-12-01 18:49:40 +0000769 return SQLITE_NOWILDCARDMATCH;
drh88b33222014-09-25 03:51:37 +0000770 }
drh88b33222014-09-25 03:51:37 +0000771 if( c==matchOther ){
drh07e83472016-01-11 03:48:18 +0000772 if( pInfo->matchSet==0 ){
drh88b33222014-09-25 03:51:37 +0000773 c = sqlite3Utf8Read(&zPattern);
drh698a01c2016-12-01 18:49:40 +0000774 if( c==0 ) return SQLITE_NOMATCH;
drh9fdfdc82014-09-25 11:08:57 +0000775 zEscaped = zPattern;
drh88b33222014-09-25 03:51:37 +0000776 }else{
777 u32 prior_c = 0;
drh9fdfdc82014-09-25 11:08:57 +0000778 int seen = 0;
779 int invert = 0;
drh88b33222014-09-25 03:51:37 +0000780 c = sqlite3Utf8Read(&zString);
drh698a01c2016-12-01 18:49:40 +0000781 if( c==0 ) return SQLITE_NOMATCH;
drh88b33222014-09-25 03:51:37 +0000782 c2 = sqlite3Utf8Read(&zPattern);
783 if( c2=='^' ){
784 invert = 1;
785 c2 = sqlite3Utf8Read(&zPattern);
786 }
787 if( c2==']' ){
788 if( c==']' ) seen = 1;
789 c2 = sqlite3Utf8Read(&zPattern);
790 }
791 while( c2 && c2!=']' ){
792 if( c2=='-' && zPattern[0]!=']' && zPattern[0]!=0 && prior_c>0 ){
793 c2 = sqlite3Utf8Read(&zPattern);
794 if( c>=prior_c && c<=c2 ) seen = 1;
795 prior_c = 0;
796 }else{
797 if( c==c2 ){
798 seen = 1;
799 }
800 prior_c = c2;
801 }
802 c2 = sqlite3Utf8Read(&zPattern);
803 }
804 if( c2==0 || (seen ^ invert)==0 ){
drh698a01c2016-12-01 18:49:40 +0000805 return SQLITE_NOMATCH;
drh88b33222014-09-25 03:51:37 +0000806 }
807 continue;
808 }
809 }
drhb0870482015-06-17 13:20:54 +0000810 c2 = Utf8Read(zString);
drh88b33222014-09-25 03:51:37 +0000811 if( c==c2 ) continue;
drhc80937a2016-06-06 01:48:14 +0000812 if( noCase && sqlite3Tolower(c)==sqlite3Tolower(c2) && c<0x80 && c2<0x80 ){
drh9fdfdc82014-09-25 11:08:57 +0000813 continue;
814 }
815 if( c==matchOne && zPattern!=zEscaped && c2!=0 ) continue;
drh698a01c2016-12-01 18:49:40 +0000816 return SQLITE_NOMATCH;
danielk197751ad0ec2004-05-24 12:39:02 +0000817 }
drh698a01c2016-12-01 18:49:40 +0000818 return *zString==0 ? SQLITE_MATCH : SQLITE_NOMATCH;
drh0ac65892002-04-20 14:24:41 +0000819}
drh4e5ffc52004-08-31 00:52:37 +0000820
drh55ef4d92005-08-14 01:20:37 +0000821/*
drh698a01c2016-12-01 18:49:40 +0000822** The sqlite3_strglob() interface. Return 0 on a match (like strcmp()) and
823** non-zero if there is no match.
drh56282a52013-04-10 16:13:38 +0000824*/
825int sqlite3_strglob(const char *zGlobPattern, const char *zString){
drh42dddb92022-10-25 13:44:18 +0000826 if( zString==0 ){
827 return zGlobPattern!=0;
828 }else if( zGlobPattern==0 ){
829 return 1;
830 }else {
831 return patternCompare((u8*)zGlobPattern, (u8*)zString, &globInfo, '[');
832 }
drh56282a52013-04-10 16:13:38 +0000833}
834
835/*
drh698a01c2016-12-01 18:49:40 +0000836** The sqlite3_strlike() interface. Return 0 on a match and non-zero for
837** a miss - like strcmp().
drh8b4a94a2015-11-24 21:23:59 +0000838*/
839int sqlite3_strlike(const char *zPattern, const char *zStr, unsigned int esc){
drh42dddb92022-10-25 13:44:18 +0000840 if( zStr==0 ){
841 return zPattern!=0;
842 }else if( zPattern==0 ){
843 return 1;
844 }else{
845 return patternCompare((u8*)zPattern, (u8*)zStr, &likeInfoNorm, esc);
846 }
drh8b4a94a2015-11-24 21:23:59 +0000847}
848
849/*
drh55ef4d92005-08-14 01:20:37 +0000850** Count the number of times that the LIKE operator (or GLOB which is
851** just a variation of LIKE) gets called. This is used for testing
852** only.
853*/
854#ifdef SQLITE_TEST
855int sqlite3_like_count = 0;
856#endif
857
danielk19773f6b0872004-06-17 05:36:44 +0000858
859/*
860** Implementation of the like() SQL function. This function implements
861** the build-in LIKE operator. The first argument to the function is the
862** pattern and the second argument is the string. So, the SQL statements:
863**
864** A LIKE B
865**
866** is implemented as like(B,A).
867**
drh55ef4d92005-08-14 01:20:37 +0000868** This same function (with a different compareInfo structure) computes
869** the GLOB operator.
danielk19773f6b0872004-06-17 05:36:44 +0000870*/
871static void likeFunc(
872 sqlite3_context *context,
873 int argc,
874 sqlite3_value **argv
875){
drhbeb818d2007-05-08 15:34:47 +0000876 const unsigned char *zA, *zB;
drh07e83472016-01-11 03:48:18 +0000877 u32 escape;
drh27e62db2009-04-02 10:16:17 +0000878 int nPat;
drhbb4957f2008-03-20 14:03:29 +0000879 sqlite3 *db = sqlite3_context_db_handle(context);
drh07e83472016-01-11 03:48:18 +0000880 struct compareInfo *pInfo = sqlite3_user_data(context);
drh589c7872020-03-19 18:13:28 +0000881 struct compareInfo backupInfo;
drhbeb818d2007-05-08 15:34:47 +0000882
drh41d2e662015-12-01 21:23:07 +0000883#ifdef SQLITE_LIKE_DOESNT_MATCH_BLOBS
884 if( sqlite3_value_type(argv[0])==SQLITE_BLOB
885 || sqlite3_value_type(argv[1])==SQLITE_BLOB
886 ){
887#ifdef SQLITE_TEST
888 sqlite3_like_count++;
889#endif
890 sqlite3_result_int(context, 0);
891 return;
892 }
893#endif
drh1f0feef2007-05-15 13:27:07 +0000894
drhbeb818d2007-05-08 15:34:47 +0000895 /* Limit the length of the LIKE or GLOB pattern to avoid problems
896 ** of deep recursion and N*N behavior in patternCompare().
897 */
drh27e62db2009-04-02 10:16:17 +0000898 nPat = sqlite3_value_bytes(argv[0]);
899 testcase( nPat==db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH] );
900 testcase( nPat==db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH]+1 );
901 if( nPat > db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH] ){
drhbeb818d2007-05-08 15:34:47 +0000902 sqlite3_result_error(context, "LIKE or GLOB pattern too complex", -1);
903 return;
904 }
danielk19777c6303c2004-11-17 16:41:29 +0000905 if( argc==3 ){
906 /* The escape character string must consist of a single UTF-8 character.
907 ** Otherwise, return an error.
908 */
909 const unsigned char *zEsc = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +0000910 if( zEsc==0 ) return;
drhee858132007-05-08 20:37:38 +0000911 if( sqlite3Utf8CharLen((char*)zEsc, -1)!=1 ){
danielk19777c6303c2004-11-17 16:41:29 +0000912 sqlite3_result_error(context,
913 "ESCAPE expression must be a single character", -1);
914 return;
915 }
drh42610962012-09-17 18:56:32 +0000916 escape = sqlite3Utf8Read(&zEsc);
drh589c7872020-03-19 18:13:28 +0000917 if( escape==pInfo->matchAll || escape==pInfo->matchOne ){
918 memcpy(&backupInfo, pInfo, sizeof(backupInfo));
919 pInfo = &backupInfo;
920 if( escape==pInfo->matchAll ) pInfo->matchAll = 0;
921 if( escape==pInfo->matchOne ) pInfo->matchOne = 0;
922 }
drh07e83472016-01-11 03:48:18 +0000923 }else{
924 escape = pInfo->matchSet;
danielk19777c6303c2004-11-17 16:41:29 +0000925 }
drhcf833232019-04-30 11:54:36 +0000926 zB = sqlite3_value_text(argv[0]);
927 zA = sqlite3_value_text(argv[1]);
danielk19773f6b0872004-06-17 05:36:44 +0000928 if( zA && zB ){
drh55ef4d92005-08-14 01:20:37 +0000929#ifdef SQLITE_TEST
930 sqlite3_like_count++;
931#endif
drhf49759b2017-08-25 19:51:51 +0000932 sqlite3_result_int(context,
933 patternCompare(zB, zA, pInfo, escape)==SQLITE_MATCH);
danielk197751ad0ec2004-05-24 12:39:02 +0000934 }
drh8912d102002-05-26 21:34:58 +0000935}
936
937/*
938** Implementation of the NULLIF(x,y) function. The result is the first
939** argument if the arguments are different. The result is NULL if the
940** arguments are equal to each other.
941*/
drhf9b596e2004-05-26 16:54:42 +0000942static void nullifFunc(
943 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000944 int NotUsed,
drhf9b596e2004-05-26 16:54:42 +0000945 sqlite3_value **argv
946){
danielk1977dc1bdc42004-06-11 10:51:27 +0000947 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk197762c14b32008-11-19 09:05:26 +0000948 UNUSED_PARAMETER(NotUsed);
danielk1977dc1bdc42004-06-11 10:51:27 +0000949 if( sqlite3MemCompare(argv[0], argv[1], pColl)!=0 ){
drhf4479502004-05-27 03:12:53 +0000950 sqlite3_result_value(context, argv[0]);
drh8912d102002-05-26 21:34:58 +0000951 }
drh0ac65892002-04-20 14:24:41 +0000952}
953
drh647cb0e2002-11-04 19:32:25 +0000954/*
drh47baebc2009-08-14 16:01:24 +0000955** Implementation of the sqlite_version() function. The result is the version
drh647cb0e2002-11-04 19:32:25 +0000956** of the SQLite library that is running.
957*/
drhf9b596e2004-05-26 16:54:42 +0000958static void versionFunc(
959 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000960 int NotUsed,
961 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000962){
danielk197762c14b32008-11-19 09:05:26 +0000963 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000964 /* IMP: R-48699-48617 This function is an SQL wrapper around the
965 ** sqlite3_libversion() C-interface. */
966 sqlite3_result_text(context, sqlite3_libversion(), -1, SQLITE_STATIC);
drh647cb0e2002-11-04 19:32:25 +0000967}
968
drh47baebc2009-08-14 16:01:24 +0000969/*
970** Implementation of the sqlite_source_id() function. The result is a string
971** that identifies the particular version of the source code used to build
972** SQLite.
973*/
974static void sourceidFunc(
975 sqlite3_context *context,
976 int NotUsed,
977 sqlite3_value **NotUsed2
978){
979 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000980 /* IMP: R-24470-31136 This function is an SQL wrapper around the
981 ** sqlite3_sourceid() C interface. */
982 sqlite3_result_text(context, sqlite3_sourceid(), -1, SQLITE_STATIC);
drh47baebc2009-08-14 16:01:24 +0000983}
984
shanehbdea6d12010-02-23 04:19:54 +0000985/*
drh3ca84ef2011-04-25 18:03:10 +0000986** Implementation of the sqlite_log() function. This is a wrapper around
987** sqlite3_log(). The return value is NULL. The function exists purely for
988** its side-effects.
989*/
drh840561f2011-04-27 18:08:42 +0000990static void errlogFunc(
drh3ca84ef2011-04-25 18:03:10 +0000991 sqlite3_context *context,
992 int argc,
993 sqlite3_value **argv
994){
995 UNUSED_PARAMETER(argc);
996 UNUSED_PARAMETER(context);
997 sqlite3_log(sqlite3_value_int(argv[0]), "%s", sqlite3_value_text(argv[1]));
998}
999
1000/*
shanehdc97a8c2010-02-23 20:08:35 +00001001** Implementation of the sqlite_compileoption_used() function.
1002** The result is an integer that identifies if the compiler option
1003** was used to build SQLite.
shanehbdea6d12010-02-23 04:19:54 +00001004*/
shanehdc97a8c2010-02-23 20:08:35 +00001005#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
1006static void compileoptionusedFunc(
shanehbdea6d12010-02-23 04:19:54 +00001007 sqlite3_context *context,
shanehdc97a8c2010-02-23 20:08:35 +00001008 int argc,
1009 sqlite3_value **argv
shanehbdea6d12010-02-23 04:19:54 +00001010){
shanehdc97a8c2010-02-23 20:08:35 +00001011 const char *zOptName;
1012 assert( argc==1 );
1013 UNUSED_PARAMETER(argc);
drha3e414c2010-08-03 18:06:25 +00001014 /* IMP: R-39564-36305 The sqlite_compileoption_used() SQL
1015 ** function is a wrapper around the sqlite3_compileoption_used() C/C++
1016 ** function.
1017 */
drh264a2d42010-02-25 15:28:41 +00001018 if( (zOptName = (const char*)sqlite3_value_text(argv[0]))!=0 ){
shanehdc97a8c2010-02-23 20:08:35 +00001019 sqlite3_result_int(context, sqlite3_compileoption_used(zOptName));
shanehdc97a8c2010-02-23 20:08:35 +00001020 }
shanehbdea6d12010-02-23 04:19:54 +00001021}
shanehdc97a8c2010-02-23 20:08:35 +00001022#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
1023
1024/*
1025** Implementation of the sqlite_compileoption_get() function.
1026** The result is a string that identifies the compiler options
1027** used to build SQLite.
1028*/
1029#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
1030static void compileoptiongetFunc(
1031 sqlite3_context *context,
1032 int argc,
1033 sqlite3_value **argv
1034){
1035 int n;
1036 assert( argc==1 );
1037 UNUSED_PARAMETER(argc);
drha3e414c2010-08-03 18:06:25 +00001038 /* IMP: R-04922-24076 The sqlite_compileoption_get() SQL function
1039 ** is a wrapper around the sqlite3_compileoption_get() C/C++ function.
1040 */
shanehdc97a8c2010-02-23 20:08:35 +00001041 n = sqlite3_value_int(argv[0]);
1042 sqlite3_result_text(context, sqlite3_compileoption_get(n), -1, SQLITE_STATIC);
1043}
1044#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
shanehbdea6d12010-02-23 04:19:54 +00001045
drh137c7282007-01-29 17:58:28 +00001046/* Array for converting from half-bytes (nybbles) into ASCII hex
1047** digits. */
1048static const char hexdigits[] = {
1049 '0', '1', '2', '3', '4', '5', '6', '7',
1050 '8', '9', 'A', 'B', 'C', 'D', 'E', 'F'
1051};
danielk1977d641d642004-11-18 15:44:29 +00001052
drh47394702003-08-20 01:03:33 +00001053/*
drhef95d552021-12-10 21:01:24 +00001054** Append to pStr text that is the SQL literal representation of the
1055** value contained in pValue.
drh47394702003-08-20 01:03:33 +00001056*/
drhef95d552021-12-10 21:01:24 +00001057void sqlite3QuoteValue(StrAccum *pStr, sqlite3_value *pValue){
1058 /* As currently implemented, the string must be initially empty.
1059 ** we might relax this requirement in the future, but that will
1060 ** require enhancements to the implementation. */
1061 assert( pStr!=0 && pStr->nChar==0 );
1062
1063 switch( sqlite3_value_type(pValue) ){
drh9c054832004-05-31 18:51:57 +00001064 case SQLITE_FLOAT: {
drh72b3fbc2012-06-19 03:11:25 +00001065 double r1, r2;
drhef95d552021-12-10 21:01:24 +00001066 const char *zVal;
1067 r1 = sqlite3_value_double(pValue);
1068 sqlite3_str_appendf(pStr, "%!.15g", r1);
1069 zVal = sqlite3_str_value(pStr);
1070 if( zVal ){
1071 sqlite3AtoF(zVal, &r2, pStr->nChar, SQLITE_UTF8);
1072 if( r1!=r2 ){
1073 sqlite3_str_reset(pStr);
1074 sqlite3_str_appendf(pStr, "%!.20e", r1);
1075 }
drh72b3fbc2012-06-19 03:11:25 +00001076 }
drh72b3fbc2012-06-19 03:11:25 +00001077 break;
1078 }
1079 case SQLITE_INTEGER: {
drhef95d552021-12-10 21:01:24 +00001080 sqlite3_str_appendf(pStr, "%lld", sqlite3_value_int64(pValue));
drhf9b596e2004-05-26 16:54:42 +00001081 break;
1082 }
danielk19773f41e972004-06-08 00:39:01 +00001083 case SQLITE_BLOB: {
drhef95d552021-12-10 21:01:24 +00001084 char const *zBlob = sqlite3_value_blob(pValue);
1085 int nBlob = sqlite3_value_bytes(pValue);
1086 assert( zBlob==sqlite3_value_blob(pValue) ); /* No encoding change */
1087 sqlite3StrAccumEnlarge(pStr, nBlob*2 + 4);
1088 if( pStr->accError==0 ){
1089 char *zText = pStr->zText;
danielk19773f41e972004-06-08 00:39:01 +00001090 int i;
1091 for(i=0; i<nBlob; i++){
1092 zText[(i*2)+2] = hexdigits[(zBlob[i]>>4)&0x0F];
1093 zText[(i*2)+3] = hexdigits[(zBlob[i])&0x0F];
1094 }
1095 zText[(nBlob*2)+2] = '\'';
1096 zText[(nBlob*2)+3] = '\0';
1097 zText[0] = 'X';
1098 zText[1] = '\'';
drhef95d552021-12-10 21:01:24 +00001099 pStr->nChar = nBlob*2 + 3;
danielk19773f41e972004-06-08 00:39:01 +00001100 }
1101 break;
1102 }
drh9c054832004-05-31 18:51:57 +00001103 case SQLITE_TEXT: {
drhef95d552021-12-10 21:01:24 +00001104 const unsigned char *zArg = sqlite3_value_text(pValue);
1105 sqlite3_str_appendf(pStr, "%Q", zArg);
drha0df4cc2009-02-02 17:29:59 +00001106 break;
1107 }
1108 default: {
drhef95d552021-12-10 21:01:24 +00001109 assert( sqlite3_value_type(pValue)==SQLITE_NULL );
1110 sqlite3_str_append(pStr, "NULL", 4);
drha0df4cc2009-02-02 17:29:59 +00001111 break;
drhf9b596e2004-05-26 16:54:42 +00001112 }
drh47394702003-08-20 01:03:33 +00001113 }
1114}
1115
drh137c7282007-01-29 17:58:28 +00001116/*
drhef95d552021-12-10 21:01:24 +00001117** Implementation of the QUOTE() function.
1118**
1119** The quote(X) function returns the text of an SQL literal which is the
1120** value of its argument suitable for inclusion into an SQL statement.
1121** Strings are surrounded by single-quotes with escapes on interior quotes
1122** as needed. BLOBs are encoded as hexadecimal literals. Strings with
1123** embedded NUL characters cannot be represented as string literals in SQL
1124** and hence the returned string literal is truncated prior to the first NUL.
1125*/
1126static void quoteFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
1127 sqlite3_str str;
1128 sqlite3 *db = sqlite3_context_db_handle(context);
1129 assert( argc==1 );
1130 UNUSED_PARAMETER(argc);
1131 sqlite3StrAccumInit(&str, db, 0, 0, db->aLimit[SQLITE_LIMIT_LENGTH]);
1132 sqlite3QuoteValue(&str,argv[0]);
1133 sqlite3_result_text(context, sqlite3StrAccumFinish(&str), str.nChar,
1134 SQLITE_DYNAMIC);
danaf7b8dc2022-02-12 13:37:27 +00001135 if( str.accError!=SQLITE_OK ){
1136 sqlite3_result_null(context);
1137 sqlite3_result_error_code(context, str.accError);
drhef95d552021-12-10 21:01:24 +00001138 }
1139}
1140
1141/*
drhd495d8c2013-02-22 19:34:25 +00001142** The unicode() function. Return the integer unicode code-point value
1143** for the first character of the input string.
1144*/
1145static void unicodeFunc(
1146 sqlite3_context *context,
1147 int argc,
1148 sqlite3_value **argv
1149){
1150 const unsigned char *z = sqlite3_value_text(argv[0]);
drh1d59d032013-03-01 23:40:26 +00001151 (void)argc;
drhd495d8c2013-02-22 19:34:25 +00001152 if( z && z[0] ) sqlite3_result_int(context, sqlite3Utf8Read(&z));
1153}
1154
1155/*
1156** The char() function takes zero or more arguments, each of which is
1157** an integer. It constructs a string where each character of the string
1158** is the unicode character for the corresponding integer argument.
1159*/
1160static void charFunc(
1161 sqlite3_context *context,
1162 int argc,
1163 sqlite3_value **argv
1164){
1165 unsigned char *z, *zOut;
1166 int i;
drhf3cdcdc2015-04-29 16:50:28 +00001167 zOut = z = sqlite3_malloc64( argc*4+1 );
drhd495d8c2013-02-22 19:34:25 +00001168 if( z==0 ){
1169 sqlite3_result_error_nomem(context);
1170 return;
1171 }
1172 for(i=0; i<argc; i++){
mistachkinc9545442013-02-26 05:42:30 +00001173 sqlite3_int64 x;
drhd495d8c2013-02-22 19:34:25 +00001174 unsigned c;
1175 x = sqlite3_value_int64(argv[i]);
1176 if( x<0 || x>0x10ffff ) x = 0xfffd;
1177 c = (unsigned)(x & 0x1fffff);
drhfe7a5d12013-03-07 14:00:04 +00001178 if( c<0x00080 ){
1179 *zOut++ = (u8)(c&0xFF);
1180 }else if( c<0x00800 ){
1181 *zOut++ = 0xC0 + (u8)((c>>6)&0x1F);
1182 *zOut++ = 0x80 + (u8)(c & 0x3F);
1183 }else if( c<0x10000 ){
1184 *zOut++ = 0xE0 + (u8)((c>>12)&0x0F);
1185 *zOut++ = 0x80 + (u8)((c>>6) & 0x3F);
1186 *zOut++ = 0x80 + (u8)(c & 0x3F);
drhd495d8c2013-02-22 19:34:25 +00001187 }else{
drhfe7a5d12013-03-07 14:00:04 +00001188 *zOut++ = 0xF0 + (u8)((c>>18) & 0x07);
1189 *zOut++ = 0x80 + (u8)((c>>12) & 0x3F);
1190 *zOut++ = 0x80 + (u8)((c>>6) & 0x3F);
1191 *zOut++ = 0x80 + (u8)(c & 0x3F);
1192 } \
drhd495d8c2013-02-22 19:34:25 +00001193 }
drhbbf483f2014-09-09 20:30:24 +00001194 sqlite3_result_text64(context, (char*)z, zOut-z, sqlite3_free, SQLITE_UTF8);
drhd495d8c2013-02-22 19:34:25 +00001195}
1196
1197/*
drh137c7282007-01-29 17:58:28 +00001198** The hex() function. Interpret the argument as a blob. Return
1199** a hexadecimal rendering as text.
1200*/
1201static void hexFunc(
1202 sqlite3_context *context,
1203 int argc,
1204 sqlite3_value **argv
1205){
1206 int i, n;
1207 const unsigned char *pBlob;
1208 char *zHex, *z;
1209 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001210 UNUSED_PARAMETER(argc);
drh1f0feef2007-05-15 13:27:07 +00001211 pBlob = sqlite3_value_blob(argv[0]);
drh137c7282007-01-29 17:58:28 +00001212 n = sqlite3_value_bytes(argv[0]);
drh1f0feef2007-05-15 13:27:07 +00001213 assert( pBlob==sqlite3_value_blob(argv[0]) ); /* No encoding change */
drhb1a6c3c2008-03-20 16:30:17 +00001214 z = zHex = contextMalloc(context, ((i64)n)*2 + 1);
danielk1977a1644fd2007-08-29 12:31:25 +00001215 if( zHex ){
1216 for(i=0; i<n; i++, pBlob++){
1217 unsigned char c = *pBlob;
1218 *(z++) = hexdigits[(c>>4)&0xf];
1219 *(z++) = hexdigits[c&0xf];
1220 }
1221 *z = 0;
1222 sqlite3_result_text(context, zHex, n*2, sqlite3_free);
drh137c7282007-01-29 17:58:28 +00001223 }
drh137c7282007-01-29 17:58:28 +00001224}
1225
drh26b6d902007-03-17 13:27:54 +00001226/*
drh8cff3822007-05-02 02:08:28 +00001227** The zeroblob(N) function returns a zero-filled blob of size N bytes.
1228*/
1229static void zeroblobFunc(
1230 sqlite3_context *context,
1231 int argc,
1232 sqlite3_value **argv
1233){
drh98640a32007-06-07 19:08:32 +00001234 i64 n;
dana4d5ae82015-07-24 16:24:37 +00001235 int rc;
drh8cff3822007-05-02 02:08:28 +00001236 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001237 UNUSED_PARAMETER(argc);
drh98640a32007-06-07 19:08:32 +00001238 n = sqlite3_value_int64(argv[0]);
dana4d5ae82015-07-24 16:24:37 +00001239 if( n<0 ) n = 0;
1240 rc = sqlite3_result_zeroblob64(context, n); /* IMP: R-00293-64994 */
1241 if( rc ){
1242 sqlite3_result_error_code(context, rc);
drh98640a32007-06-07 19:08:32 +00001243 }
drh8cff3822007-05-02 02:08:28 +00001244}
1245
1246/*
drh26b6d902007-03-17 13:27:54 +00001247** The replace() function. Three arguments are all strings: call
1248** them A, B, and C. The result is also a string which is derived
drhf7b54962013-05-28 12:11:54 +00001249** from A by replacing every occurrence of B with C. The match
drh26b6d902007-03-17 13:27:54 +00001250** must be exact. Collating sequences are not used.
1251*/
1252static void replaceFunc(
1253 sqlite3_context *context,
1254 int argc,
1255 sqlite3_value **argv
1256){
1257 const unsigned char *zStr; /* The input string A */
1258 const unsigned char *zPattern; /* The pattern string B */
1259 const unsigned char *zRep; /* The replacement string C */
1260 unsigned char *zOut; /* The output */
1261 int nStr; /* Size of zStr */
1262 int nPattern; /* Size of zPattern */
1263 int nRep; /* Size of zRep */
drh2e6400b2007-05-08 15:46:18 +00001264 i64 nOut; /* Maximum size of zOut */
drh26b6d902007-03-17 13:27:54 +00001265 int loopLimit; /* Last zStr[] that might match zPattern[] */
1266 int i, j; /* Loop counters */
drhf3139522018-02-09 23:25:14 +00001267 unsigned cntExpand; /* Number zOut expansions */
1268 sqlite3 *db = sqlite3_context_db_handle(context);
drh26b6d902007-03-17 13:27:54 +00001269
1270 assert( argc==3 );
danielk1977f3d3c272008-11-19 16:52:44 +00001271 UNUSED_PARAMETER(argc);
drh26b6d902007-03-17 13:27:54 +00001272 zStr = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001273 if( zStr==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001274 nStr = sqlite3_value_bytes(argv[0]);
1275 assert( zStr==sqlite3_value_text(argv[0]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001276 zPattern = sqlite3_value_text(argv[1]);
drha605fe82009-02-01 18:08:40 +00001277 if( zPattern==0 ){
drh23336062009-02-03 13:19:12 +00001278 assert( sqlite3_value_type(argv[1])==SQLITE_NULL
1279 || sqlite3_context_db_handle(context)->mallocFailed );
drha605fe82009-02-01 18:08:40 +00001280 return;
1281 }
1282 if( zPattern[0]==0 ){
1283 assert( sqlite3_value_type(argv[1])!=SQLITE_NULL );
1284 sqlite3_result_value(context, argv[0]);
1285 return;
1286 }
drh1f0feef2007-05-15 13:27:07 +00001287 nPattern = sqlite3_value_bytes(argv[1]);
1288 assert( zPattern==sqlite3_value_text(argv[1]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001289 zRep = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +00001290 if( zRep==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001291 nRep = sqlite3_value_bytes(argv[2]);
1292 assert( zRep==sqlite3_value_text(argv[2]) );
drh2e6400b2007-05-08 15:46:18 +00001293 nOut = nStr + 1;
1294 assert( nOut<SQLITE_MAX_LENGTH );
drhb1a6c3c2008-03-20 16:30:17 +00001295 zOut = contextMalloc(context, (i64)nOut);
drh2e6400b2007-05-08 15:46:18 +00001296 if( zOut==0 ){
1297 return;
drh26b6d902007-03-17 13:27:54 +00001298 }
drh26b6d902007-03-17 13:27:54 +00001299 loopLimit = nStr - nPattern;
drhf3139522018-02-09 23:25:14 +00001300 cntExpand = 0;
drh26b6d902007-03-17 13:27:54 +00001301 for(i=j=0; i<=loopLimit; i++){
1302 if( zStr[i]!=zPattern[0] || memcmp(&zStr[i], zPattern, nPattern) ){
1303 zOut[j++] = zStr[i];
1304 }else{
drhf3139522018-02-09 23:25:14 +00001305 if( nRep>nPattern ){
1306 nOut += nRep - nPattern;
drhc86d82f2018-02-10 02:31:30 +00001307 testcase( nOut-1==db->aLimit[SQLITE_LIMIT_LENGTH] );
1308 testcase( nOut-2==db->aLimit[SQLITE_LIMIT_LENGTH] );
drhf3139522018-02-09 23:25:14 +00001309 if( nOut-1>db->aLimit[SQLITE_LIMIT_LENGTH] ){
1310 sqlite3_result_error_toobig(context);
1311 sqlite3_free(zOut);
1312 return;
1313 }
drhf3139522018-02-09 23:25:14 +00001314 cntExpand++;
1315 if( (cntExpand&(cntExpand-1))==0 ){
1316 /* Grow the size of the output buffer only on substitutions
1317 ** whose index is a power of two: 1, 2, 4, 8, 16, 32, ... */
1318 u8 *zOld;
1319 zOld = zOut;
drhd924e7b2020-05-17 00:26:44 +00001320 zOut = sqlite3Realloc(zOut, (int)nOut + (nOut - nStr - 1));
drhf3139522018-02-09 23:25:14 +00001321 if( zOut==0 ){
1322 sqlite3_result_error_nomem(context);
1323 sqlite3_free(zOld);
1324 return;
1325 }
1326 }
drh2e6400b2007-05-08 15:46:18 +00001327 }
drh26b6d902007-03-17 13:27:54 +00001328 memcpy(&zOut[j], zRep, nRep);
1329 j += nRep;
1330 i += nPattern-1;
1331 }
1332 }
drhf3139522018-02-09 23:25:14 +00001333 assert( j+nStr-i+1<=nOut );
drh26b6d902007-03-17 13:27:54 +00001334 memcpy(&zOut[j], &zStr[i], nStr-i);
1335 j += nStr - i;
1336 assert( j<=nOut );
1337 zOut[j] = 0;
1338 sqlite3_result_text(context, (char*)zOut, j, sqlite3_free);
1339}
1340
drh309b3382007-03-17 17:52:42 +00001341/*
1342** Implementation of the TRIM(), LTRIM(), and RTRIM() functions.
1343** The userdata is 0x1 for left trim, 0x2 for right trim, 0x3 for both.
1344*/
1345static void trimFunc(
1346 sqlite3_context *context,
1347 int argc,
1348 sqlite3_value **argv
1349){
1350 const unsigned char *zIn; /* Input string */
1351 const unsigned char *zCharSet; /* Set of characters to trim */
drh972da422021-06-15 14:34:21 +00001352 unsigned int nIn; /* Number of bytes in input */
drh7209c692008-04-27 18:40:11 +00001353 int flags; /* 1: trimleft 2: trimright 3: trim */
drhd1e3a612007-04-27 21:59:52 +00001354 int i; /* Loop counter */
drh972da422021-06-15 14:34:21 +00001355 unsigned int *aLen = 0; /* Length of each character in zCharSet */
drh1bd10f82008-12-10 21:19:56 +00001356 unsigned char **azChar = 0; /* Individual characters in zCharSet */
drhd1e3a612007-04-27 21:59:52 +00001357 int nChar; /* Number of characters in zCharSet */
1358
drh309b3382007-03-17 17:52:42 +00001359 if( sqlite3_value_type(argv[0])==SQLITE_NULL ){
1360 return;
1361 }
1362 zIn = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001363 if( zIn==0 ) return;
drh972da422021-06-15 14:34:21 +00001364 nIn = (unsigned)sqlite3_value_bytes(argv[0]);
drh1f0feef2007-05-15 13:27:07 +00001365 assert( zIn==sqlite3_value_text(argv[0]) );
drh309b3382007-03-17 17:52:42 +00001366 if( argc==1 ){
drh972da422021-06-15 14:34:21 +00001367 static const unsigned lenOne[] = { 1 };
danielk1977a4de4532008-09-02 15:44:08 +00001368 static unsigned char * const azOne[] = { (u8*)" " };
drhd1e3a612007-04-27 21:59:52 +00001369 nChar = 1;
drh972da422021-06-15 14:34:21 +00001370 aLen = (unsigned*)lenOne;
danielk1977bc67da42007-12-11 04:23:19 +00001371 azChar = (unsigned char **)azOne;
drhd1e3a612007-04-27 21:59:52 +00001372 zCharSet = 0;
drh7a521cf2007-04-25 18:23:52 +00001373 }else if( (zCharSet = sqlite3_value_text(argv[1]))==0 ){
drh309b3382007-03-17 17:52:42 +00001374 return;
drhd1e3a612007-04-27 21:59:52 +00001375 }else{
1376 const unsigned char *z;
1377 for(z=zCharSet, nChar=0; *z; nChar++){
drh4a919112007-05-15 11:55:09 +00001378 SQLITE_SKIP_UTF8(z);
drhd1e3a612007-04-27 21:59:52 +00001379 }
1380 if( nChar>0 ){
drh972da422021-06-15 14:34:21 +00001381 azChar = contextMalloc(context,
1382 ((i64)nChar)*(sizeof(char*)+sizeof(unsigned)));
drhd1e3a612007-04-27 21:59:52 +00001383 if( azChar==0 ){
1384 return;
1385 }
drh972da422021-06-15 14:34:21 +00001386 aLen = (unsigned*)&azChar[nChar];
drhd1e3a612007-04-27 21:59:52 +00001387 for(z=zCharSet, nChar=0; *z; nChar++){
danielk1977bc67da42007-12-11 04:23:19 +00001388 azChar[nChar] = (unsigned char *)z;
drh4a919112007-05-15 11:55:09 +00001389 SQLITE_SKIP_UTF8(z);
drh972da422021-06-15 14:34:21 +00001390 aLen[nChar] = (unsigned)(z - azChar[nChar]);
drhd1e3a612007-04-27 21:59:52 +00001391 }
1392 }
drh309b3382007-03-17 17:52:42 +00001393 }
drhd1e3a612007-04-27 21:59:52 +00001394 if( nChar>0 ){
shane1fc41292008-07-08 22:28:48 +00001395 flags = SQLITE_PTR_TO_INT(sqlite3_user_data(context));
drh309b3382007-03-17 17:52:42 +00001396 if( flags & 1 ){
drhd1e3a612007-04-27 21:59:52 +00001397 while( nIn>0 ){
drh972da422021-06-15 14:34:21 +00001398 unsigned int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001399 for(i=0; i<nChar; i++){
1400 len = aLen[i];
drh27e62db2009-04-02 10:16:17 +00001401 if( len<=nIn && memcmp(zIn, azChar[i], len)==0 ) break;
drhd1e3a612007-04-27 21:59:52 +00001402 }
1403 if( i>=nChar ) break;
1404 zIn += len;
1405 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001406 }
1407 }
1408 if( flags & 2 ){
drhd1e3a612007-04-27 21:59:52 +00001409 while( nIn>0 ){
drh972da422021-06-15 14:34:21 +00001410 unsigned int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001411 for(i=0; i<nChar; i++){
1412 len = aLen[i];
1413 if( len<=nIn && memcmp(&zIn[nIn-len],azChar[i],len)==0 ) break;
1414 }
1415 if( i>=nChar ) break;
1416 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001417 }
1418 }
drhd1e3a612007-04-27 21:59:52 +00001419 if( zCharSet ){
1420 sqlite3_free(azChar);
1421 }
drh309b3382007-03-17 17:52:42 +00001422 }
1423 sqlite3_result_text(context, (char*)zIn, nIn, SQLITE_TRANSIENT);
1424}
drh26b6d902007-03-17 13:27:54 +00001425
danielk1977a4de4532008-09-02 15:44:08 +00001426
drhcc153132016-08-04 12:35:17 +00001427#ifdef SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION
1428/*
1429** The "unknown" function is automatically substituted in place of
1430** any unrecognized function name when doing an EXPLAIN or EXPLAIN QUERY PLAN
1431** when the SQLITE_ENABLE_UNKNOWN_FUNCTION compile-time option is used.
1432** When the "sqlite3" command-line shell is built using this functionality,
1433** that allows an EXPLAIN or EXPLAIN QUERY PLAN for complex queries
1434** involving application-defined functions to be examined in a generic
1435** sqlite3 shell.
1436*/
1437static void unknownFunc(
1438 sqlite3_context *context,
1439 int argc,
1440 sqlite3_value **argv
1441){
1442 /* no-op */
1443}
1444#endif /*SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION*/
1445
1446
drh2ba3ccc2009-12-08 02:06:08 +00001447/* IMP: R-25361-16150 This function is omitted from SQLite by default. It
1448** is only available if the SQLITE_SOUNDEX compile-time option is used
1449** when SQLite is built.
1450*/
drhd24cc422003-03-27 12:51:24 +00001451#ifdef SQLITE_SOUNDEX
1452/*
1453** Compute the soundex encoding of a word.
drh2ba3ccc2009-12-08 02:06:08 +00001454**
1455** IMP: R-59782-00072 The soundex(X) function returns a string that is the
1456** soundex encoding of the string X.
drhd24cc422003-03-27 12:51:24 +00001457*/
drh137c7282007-01-29 17:58:28 +00001458static void soundexFunc(
1459 sqlite3_context *context,
1460 int argc,
1461 sqlite3_value **argv
1462){
drhd24cc422003-03-27 12:51:24 +00001463 char zResult[8];
drh4c755c02004-08-08 20:22:17 +00001464 const u8 *zIn;
drhd24cc422003-03-27 12:51:24 +00001465 int i, j;
1466 static const unsigned char iCode[] = {
1467 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1468 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1469 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1470 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1471 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1472 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1473 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1474 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1475 };
1476 assert( argc==1 );
drh4c755c02004-08-08 20:22:17 +00001477 zIn = (u8*)sqlite3_value_text(argv[0]);
drhbdf67e02006-08-19 11:34:01 +00001478 if( zIn==0 ) zIn = (u8*)"";
drhdc86e2b2009-01-24 11:30:42 +00001479 for(i=0; zIn[i] && !sqlite3Isalpha(zIn[i]); i++){}
drhd24cc422003-03-27 12:51:24 +00001480 if( zIn[i] ){
drhbdf67e02006-08-19 11:34:01 +00001481 u8 prevcode = iCode[zIn[i]&0x7f];
danielk197778ca0e72009-01-20 16:53:39 +00001482 zResult[0] = sqlite3Toupper(zIn[i]);
drhd24cc422003-03-27 12:51:24 +00001483 for(j=1; j<4 && zIn[i]; i++){
1484 int code = iCode[zIn[i]&0x7f];
1485 if( code>0 ){
drhbdf67e02006-08-19 11:34:01 +00001486 if( code!=prevcode ){
1487 prevcode = code;
1488 zResult[j++] = code + '0';
1489 }
1490 }else{
1491 prevcode = 0;
drhd24cc422003-03-27 12:51:24 +00001492 }
1493 }
1494 while( j<4 ){
1495 zResult[j++] = '0';
1496 }
1497 zResult[j] = 0;
danielk1977d8123362004-06-12 09:25:12 +00001498 sqlite3_result_text(context, zResult, 4, SQLITE_TRANSIENT);
drhd24cc422003-03-27 12:51:24 +00001499 }else{
drh2ba3ccc2009-12-08 02:06:08 +00001500 /* IMP: R-64894-50321 The string "?000" is returned if the argument
1501 ** is NULL or contains no ASCII alphabetic characters. */
danielk1977d8123362004-06-12 09:25:12 +00001502 sqlite3_result_text(context, "?000", 4, SQLITE_STATIC);
drhd24cc422003-03-27 12:51:24 +00001503 }
1504}
drh2ba3ccc2009-12-08 02:06:08 +00001505#endif /* SQLITE_SOUNDEX */
drhd24cc422003-03-27 12:51:24 +00001506
drhfdb83b22006-06-17 14:12:47 +00001507#ifndef SQLITE_OMIT_LOAD_EXTENSION
1508/*
1509** A function that loads a shared-library extension then returns NULL.
1510*/
1511static void loadExt(sqlite3_context *context, int argc, sqlite3_value **argv){
danielk197765fd59f2006-06-24 11:51:33 +00001512 const char *zFile = (const char *)sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001513 const char *zProc;
drhfa4a4b92008-03-19 21:45:51 +00001514 sqlite3 *db = sqlite3_context_db_handle(context);
drhfdb83b22006-06-17 14:12:47 +00001515 char *zErrMsg = 0;
1516
drh191dd062016-04-21 01:30:09 +00001517 /* Disallow the load_extension() SQL function unless the SQLITE_LoadExtFunc
drh1a55ded2016-04-20 00:30:05 +00001518 ** flag is set. See the sqlite3_enable_load_extension() API.
1519 */
drhf602a162016-04-21 01:58:21 +00001520 if( (db->flags & SQLITE_LoadExtFunc)==0 ){
1521 sqlite3_result_error(context, "not authorized", -1);
1522 return;
1523 }
drh1a55ded2016-04-20 00:30:05 +00001524
drhfdb83b22006-06-17 14:12:47 +00001525 if( argc==2 ){
danielk197765fd59f2006-06-24 11:51:33 +00001526 zProc = (const char *)sqlite3_value_text(argv[1]);
drh7a521cf2007-04-25 18:23:52 +00001527 }else{
1528 zProc = 0;
drhfdb83b22006-06-17 14:12:47 +00001529 }
drh7a521cf2007-04-25 18:23:52 +00001530 if( zFile && sqlite3_load_extension(db, zFile, zProc, &zErrMsg) ){
drhfdb83b22006-06-17 14:12:47 +00001531 sqlite3_result_error(context, zErrMsg, -1);
1532 sqlite3_free(zErrMsg);
1533 }
1534}
1535#endif
1536
danielk197701427a62005-01-11 13:02:33 +00001537
drh0ac65892002-04-20 14:24:41 +00001538/*
drhd3a149e2002-02-24 17:12:53 +00001539** An instance of the following structure holds the context of a
drhdd5baa92002-02-27 19:50:59 +00001540** sum() or avg() aggregate computation.
1541*/
1542typedef struct SumCtx SumCtx;
1543struct SumCtx {
drh8c08e862006-02-11 17:34:00 +00001544 double rSum; /* Floating point sum */
1545 i64 iSum; /* Integer sum */
1546 i64 cnt; /* Number of elements summed */
1547 u8 overflow; /* True if integer overflow seen */
1548 u8 approx; /* True if non-integer value was input to the sum */
drhdd5baa92002-02-27 19:50:59 +00001549};
1550
1551/*
drha97fdd32006-01-12 22:17:50 +00001552** Routines used to compute the sum, average, and total.
1553**
1554** The SUM() function follows the (broken) SQL standard which means
1555** that it returns NULL if it sums over no inputs. TOTAL returns
1556** 0.0 in that case. In addition, TOTAL always returns a float where
1557** SUM might return an integer if it never encounters a floating point
drhc806d852006-05-11 13:25:39 +00001558** value. TOTAL never fails, but SUM might through an exception if
1559** it overflows an integer.
drhdd5baa92002-02-27 19:50:59 +00001560*/
danielk19770ae8b832004-05-25 12:05:56 +00001561static void sumStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drhdd5baa92002-02-27 19:50:59 +00001562 SumCtx *p;
drh3d1d95e2005-09-08 10:37:01 +00001563 int type;
drh3f219f42005-09-08 19:45:57 +00001564 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001565 UNUSED_PARAMETER(argc);
drh4f26d6c2004-05-26 23:25:30 +00001566 p = sqlite3_aggregate_context(context, sizeof(*p));
drh29d72102006-02-09 22:13:41 +00001567 type = sqlite3_value_numeric_type(argv[0]);
drh3d1d95e2005-09-08 10:37:01 +00001568 if( p && type!=SQLITE_NULL ){
drh739105c2002-05-29 23:22:23 +00001569 p->cnt++;
drh29d72102006-02-09 22:13:41 +00001570 if( type==SQLITE_INTEGER ){
drh8c08e862006-02-11 17:34:00 +00001571 i64 v = sqlite3_value_int64(argv[0]);
1572 p->rSum += v;
drh158b9cb2011-03-05 20:59:46 +00001573 if( (p->approx|p->overflow)==0 && sqlite3AddInt64(&p->iSum, v) ){
drha546ef22018-07-07 20:55:16 +00001574 p->approx = p->overflow = 1;
drh29d72102006-02-09 22:13:41 +00001575 }
1576 }else{
drh8c08e862006-02-11 17:34:00 +00001577 p->rSum += sqlite3_value_double(argv[0]);
drh29d72102006-02-09 22:13:41 +00001578 p->approx = 1;
drh3f219f42005-09-08 19:45:57 +00001579 }
drh739105c2002-05-29 23:22:23 +00001580 }
drhdd5baa92002-02-27 19:50:59 +00001581}
dan67a9b8e2018-06-22 20:51:35 +00001582#ifndef SQLITE_OMIT_WINDOWFUNC
danc3a20c12018-05-23 20:55:37 +00001583static void sumInverse(sqlite3_context *context, int argc, sqlite3_value**argv){
1584 SumCtx *p;
1585 int type;
1586 assert( argc==1 );
1587 UNUSED_PARAMETER(argc);
1588 p = sqlite3_aggregate_context(context, sizeof(*p));
1589 type = sqlite3_value_numeric_type(argv[0]);
drhfd4b7282018-07-07 19:47:21 +00001590 /* p is always non-NULL because sumStep() will have been called first
1591 ** to initialize it */
1592 if( ALWAYS(p) && type!=SQLITE_NULL ){
drha546ef22018-07-07 20:55:16 +00001593 assert( p->cnt>0 );
danc3a20c12018-05-23 20:55:37 +00001594 p->cnt--;
drha546ef22018-07-07 20:55:16 +00001595 assert( type==SQLITE_INTEGER || p->approx );
1596 if( type==SQLITE_INTEGER && p->approx==0 ){
danc3a20c12018-05-23 20:55:37 +00001597 i64 v = sqlite3_value_int64(argv[0]);
1598 p->rSum -= v;
drha546ef22018-07-07 20:55:16 +00001599 p->iSum -= v;
danc3a20c12018-05-23 20:55:37 +00001600 }else{
dand7368292018-07-02 17:45:59 +00001601 p->rSum -= sqlite3_value_double(argv[0]);
danc3a20c12018-05-23 20:55:37 +00001602 }
1603 }
1604}
dan67a9b8e2018-06-22 20:51:35 +00001605#else
1606# define sumInverse 0
1607#endif /* SQLITE_OMIT_WINDOWFUNC */
danielk19770ae8b832004-05-25 12:05:56 +00001608static void sumFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001609 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001610 p = sqlite3_aggregate_context(context, 0);
drhc2bd9132005-09-08 20:37:43 +00001611 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001612 if( p->overflow ){
1613 sqlite3_result_error(context,"integer overflow",-1);
1614 }else if( p->approx ){
1615 sqlite3_result_double(context, p->rSum);
drhc2bd9132005-09-08 20:37:43 +00001616 }else{
drh8c08e862006-02-11 17:34:00 +00001617 sqlite3_result_int64(context, p->iSum);
drhc2bd9132005-09-08 20:37:43 +00001618 }
drh3d1d95e2005-09-08 10:37:01 +00001619 }
drhdd5baa92002-02-27 19:50:59 +00001620}
danielk19770ae8b832004-05-25 12:05:56 +00001621static void avgFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001622 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001623 p = sqlite3_aggregate_context(context, 0);
drh739105c2002-05-29 23:22:23 +00001624 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001625 sqlite3_result_double(context, p->rSum/(double)p->cnt);
drhdd5baa92002-02-27 19:50:59 +00001626 }
1627}
drha97fdd32006-01-12 22:17:50 +00001628static void totalFinalize(sqlite3_context *context){
1629 SumCtx *p;
1630 p = sqlite3_aggregate_context(context, 0);
shanefbd60f82009-02-04 03:59:25 +00001631 /* (double)0 In case of SQLITE_OMIT_FLOATING_POINT... */
1632 sqlite3_result_double(context, p ? p->rSum : (double)0);
drha97fdd32006-01-12 22:17:50 +00001633}
drhdd5baa92002-02-27 19:50:59 +00001634
1635/*
drh0bce8352002-02-28 00:41:10 +00001636** The following structure keeps track of state information for the
1637** count() aggregate function.
1638*/
1639typedef struct CountCtx CountCtx;
1640struct CountCtx {
drhfc6ad392006-02-09 13:38:19 +00001641 i64 n;
dan7262ca92018-07-02 12:07:32 +00001642#ifdef SQLITE_DEBUG
1643 int bInverse; /* True if xInverse() ever called */
1644#endif
drh0bce8352002-02-28 00:41:10 +00001645};
drhdd5baa92002-02-27 19:50:59 +00001646
drh0bce8352002-02-28 00:41:10 +00001647/*
1648** Routines to implement the count() aggregate function.
1649*/
danielk19770ae8b832004-05-25 12:05:56 +00001650static void countStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drh0bce8352002-02-28 00:41:10 +00001651 CountCtx *p;
drh4f26d6c2004-05-26 23:25:30 +00001652 p = sqlite3_aggregate_context(context, sizeof(*p));
drh9c054832004-05-31 18:51:57 +00001653 if( (argc==0 || SQLITE_NULL!=sqlite3_value_type(argv[0])) && p ){
drh0bce8352002-02-28 00:41:10 +00001654 p->n++;
1655 }
drh2e79c3d2009-04-08 23:04:14 +00001656
drhd3264c72009-04-15 13:39:47 +00001657#ifndef SQLITE_OMIT_DEPRECATED
drh2e79c3d2009-04-08 23:04:14 +00001658 /* The sqlite3_aggregate_count() function is deprecated. But just to make
1659 ** sure it still operates correctly, verify that its count agrees with our
1660 ** internal count when using count(*) and when the total count can be
1661 ** expressed as a 32-bit integer. */
dan7262ca92018-07-02 12:07:32 +00001662 assert( argc==1 || p==0 || p->n>0x7fffffff || p->bInverse
drh2e79c3d2009-04-08 23:04:14 +00001663 || p->n==sqlite3_aggregate_count(context) );
drhd3264c72009-04-15 13:39:47 +00001664#endif
drh0bce8352002-02-28 00:41:10 +00001665}
danielk19770ae8b832004-05-25 12:05:56 +00001666static void countFinalize(sqlite3_context *context){
drh0bce8352002-02-28 00:41:10 +00001667 CountCtx *p;
drhabfcea22005-09-06 20:36:48 +00001668 p = sqlite3_aggregate_context(context, 0);
drhfc6ad392006-02-09 13:38:19 +00001669 sqlite3_result_int64(context, p ? p->n : 0);
drh0bce8352002-02-28 00:41:10 +00001670}
dan7262ca92018-07-02 12:07:32 +00001671#ifndef SQLITE_OMIT_WINDOWFUNC
1672static void countInverse(sqlite3_context *ctx, int argc, sqlite3_value **argv){
1673 CountCtx *p;
1674 p = sqlite3_aggregate_context(ctx, sizeof(*p));
drhfd4b7282018-07-07 19:47:21 +00001675 /* p is always non-NULL since countStep() will have been called first */
1676 if( (argc==0 || SQLITE_NULL!=sqlite3_value_type(argv[0])) && ALWAYS(p) ){
dan7262ca92018-07-02 12:07:32 +00001677 p->n--;
1678#ifdef SQLITE_DEBUG
1679 p->bInverse = 1;
1680#endif
1681 }
1682}
dan6b4b8822018-07-02 15:03:50 +00001683#else
1684# define countInverse 0
1685#endif /* SQLITE_OMIT_WINDOWFUNC */
drh0bce8352002-02-28 00:41:10 +00001686
1687/*
drh0bce8352002-02-28 00:41:10 +00001688** Routines to implement min() and max() aggregate functions.
1689*/
danielk197762c14b32008-11-19 09:05:26 +00001690static void minmaxStep(
1691 sqlite3_context *context,
1692 int NotUsed,
1693 sqlite3_value **argv
1694){
danielk197788208052004-05-25 01:13:20 +00001695 Mem *pArg = (Mem *)argv[0];
drh9eb516c2004-07-18 20:52:32 +00001696 Mem *pBest;
danielk197762c14b32008-11-19 09:05:26 +00001697 UNUSED_PARAMETER(NotUsed);
drh9eb516c2004-07-18 20:52:32 +00001698
drh9eb516c2004-07-18 20:52:32 +00001699 pBest = (Mem *)sqlite3_aggregate_context(context, sizeof(*pBest));
danielk19773aeab9e2004-06-24 00:20:04 +00001700 if( !pBest ) return;
drh268380c2004-02-25 13:47:31 +00001701
dan6fb2b542018-06-19 17:13:11 +00001702 if( sqlite3_value_type(pArg)==SQLITE_NULL ){
drh94a6d992012-02-02 18:42:09 +00001703 if( pBest->flags ) sqlite3SkipAccumulatorLoad(context);
1704 }else if( pBest->flags ){
drh9eb516c2004-07-18 20:52:32 +00001705 int max;
1706 int cmp;
danielk1977dc1bdc42004-06-11 10:51:27 +00001707 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk19777e18c252004-05-25 11:47:24 +00001708 /* This step function is used for both the min() and max() aggregates,
1709 ** the only difference between the two being that the sense of the
1710 ** comparison is inverted. For the max() aggregate, the
1711 ** sqlite3_user_data() function returns (void *)-1. For min() it
1712 ** returns (void *)db, where db is the sqlite3* database pointer.
1713 ** Therefore the next statement sets variable 'max' to 1 for the max()
1714 ** aggregate, or 0 for min().
1715 */
drh309b3382007-03-17 17:52:42 +00001716 max = sqlite3_user_data(context)!=0;
danielk1977dc1bdc42004-06-11 10:51:27 +00001717 cmp = sqlite3MemCompare(pBest, pArg, pColl);
danielk197788208052004-05-25 01:13:20 +00001718 if( (max && cmp<0) || (!max && cmp>0) ){
drhb21c8cd2007-08-21 19:33:56 +00001719 sqlite3VdbeMemCopy(pBest, pArg);
drh7a957892012-02-02 17:35:43 +00001720 }else{
1721 sqlite3SkipAccumulatorLoad(context);
danielk197788208052004-05-25 01:13:20 +00001722 }
drh268380c2004-02-25 13:47:31 +00001723 }else{
drh035e5632014-09-16 14:16:31 +00001724 pBest->db = sqlite3_context_db_handle(context);
drhb21c8cd2007-08-21 19:33:56 +00001725 sqlite3VdbeMemCopy(pBest, pArg);
drh0bce8352002-02-28 00:41:10 +00001726 }
1727}
dan6fb2b542018-06-19 17:13:11 +00001728static void minMaxValueFinalize(sqlite3_context *context, int bValue){
danielk197788208052004-05-25 01:13:20 +00001729 sqlite3_value *pRes;
drhabfcea22005-09-06 20:36:48 +00001730 pRes = (sqlite3_value *)sqlite3_aggregate_context(context, 0);
1731 if( pRes ){
drh94a6d992012-02-02 18:42:09 +00001732 if( pRes->flags ){
drhabfcea22005-09-06 20:36:48 +00001733 sqlite3_result_value(context, pRes);
1734 }
dan6fb2b542018-06-19 17:13:11 +00001735 if( bValue==0 ) sqlite3VdbeMemRelease(pRes);
drh0bce8352002-02-28 00:41:10 +00001736 }
1737}
dan67a9b8e2018-06-22 20:51:35 +00001738#ifndef SQLITE_OMIT_WINDOWFUNC
dan6fb2b542018-06-19 17:13:11 +00001739static void minMaxValue(sqlite3_context *context){
drhc7bf5712018-07-09 22:49:01 +00001740 minMaxValueFinalize(context, 1);
dan6fb2b542018-06-19 17:13:11 +00001741}
dan67a9b8e2018-06-22 20:51:35 +00001742#else
1743# define minMaxValue 0
1744#endif /* SQLITE_OMIT_WINDOWFUNC */
dan6fb2b542018-06-19 17:13:11 +00001745static void minMaxFinalize(sqlite3_context *context){
drhc7bf5712018-07-09 22:49:01 +00001746 minMaxValueFinalize(context, 0);
dan6fb2b542018-06-19 17:13:11 +00001747}
drhdd5baa92002-02-27 19:50:59 +00001748
drhb0689692007-11-01 17:38:30 +00001749/*
1750** group_concat(EXPR, ?SEPARATOR?)
drhf06db3e2021-10-01 00:25:06 +00001751**
1752** The SEPARATOR goes before the EXPR string. This is tragic. The
1753** groupConcatInverse() implementation would have been easier if the
1754** SEPARATOR were appended after EXPR. And the order is undocumented,
1755** so we could change it, in theory. But the old behavior has been
1756** around for so long that we dare not, for fear of breaking something.
drhb0689692007-11-01 17:38:30 +00001757*/
larrybrdde13e62021-09-29 00:32:13 +00001758typedef struct {
1759 StrAccum str; /* The accumulated concatenation */
1760#ifndef SQLITE_OMIT_WINDOWFUNC
1761 int nAccum; /* Number of strings presently concatenated */
drhf06db3e2021-10-01 00:25:06 +00001762 int nFirstSepLength; /* Used to detect separator length change */
larrybrdde13e62021-09-29 00:32:13 +00001763 /* If pnSepLengths!=0, refs an array of inter-string separator lengths,
drhf06db3e2021-10-01 00:25:06 +00001764 ** stored as actually incorporated into presently accumulated result.
1765 ** (Hence, its slots in use number nAccum-1 between method calls.)
1766 ** If pnSepLengths==0, nFirstSepLength is the length used throughout.
1767 */
larrybrdde13e62021-09-29 00:32:13 +00001768 int *pnSepLengths;
1769#endif
1770} GroupConcatCtx;
1771
drhb0689692007-11-01 17:38:30 +00001772static void groupConcatStep(
1773 sqlite3_context *context,
1774 int argc,
1775 sqlite3_value **argv
1776){
1777 const char *zVal;
larrybrdde13e62021-09-29 00:32:13 +00001778 GroupConcatCtx *pGCC;
drhb0689692007-11-01 17:38:30 +00001779 const char *zSep;
drh07d31172009-02-02 21:57:05 +00001780 int nVal, nSep;
1781 assert( argc==1 || argc==2 );
1782 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
larrybrdde13e62021-09-29 00:32:13 +00001783 pGCC = (GroupConcatCtx*)sqlite3_aggregate_context(context, sizeof(*pGCC));
1784 if( pGCC ){
drhbb4957f2008-03-20 14:03:29 +00001785 sqlite3 *db = sqlite3_context_db_handle(context);
larrybrdde13e62021-09-29 00:32:13 +00001786 int firstTerm = pGCC->str.mxAlloc==0;
1787 pGCC->str.mxAlloc = db->aLimit[SQLITE_LIMIT_LENGTH];
drhf06db3e2021-10-01 00:25:06 +00001788 if( argc==1 ){
1789 if( !firstTerm ){
1790 sqlite3_str_appendchar(&pGCC->str, 1, ',');
drhb0689692007-11-01 17:38:30 +00001791 }
larrybrdde13e62021-09-29 00:32:13 +00001792#ifndef SQLITE_OMIT_WINDOWFUNC
drhf06db3e2021-10-01 00:25:06 +00001793 else{
1794 pGCC->nFirstSepLength = 1;
1795 }
1796#endif
1797 }else if( !firstTerm ){
1798 zSep = (char*)sqlite3_value_text(argv[1]);
1799 nSep = sqlite3_value_bytes(argv[1]);
1800 if( zSep ){
1801 sqlite3_str_append(&pGCC->str, zSep, nSep);
1802 }
1803#ifndef SQLITE_OMIT_WINDOWFUNC
1804 else{
1805 nSep = 0;
1806 }
larrybrdde13e62021-09-29 00:32:13 +00001807 if( nSep != pGCC->nFirstSepLength || pGCC->pnSepLengths != 0 ){
drhf06db3e2021-10-01 00:25:06 +00001808 int *pnsl = pGCC->pnSepLengths;
1809 if( pnsl == 0 ){
1810 /* First separator length variation seen, start tracking them. */
1811 pnsl = (int*)sqlite3_malloc64((pGCC->nAccum+1) * sizeof(int));
1812 if( pnsl!=0 ){
1813 int i = 0, nA = pGCC->nAccum-1;
1814 while( i<nA ) pnsl[i++] = pGCC->nFirstSepLength;
1815 }
1816 }else{
1817 pnsl = (int*)sqlite3_realloc64(pnsl, pGCC->nAccum * sizeof(int));
1818 }
1819 if( pnsl!=0 ){
drh3dd01112021-10-01 02:45:48 +00001820 if( ALWAYS(pGCC->nAccum>0) ){
drhf06db3e2021-10-01 00:25:06 +00001821 pnsl[pGCC->nAccum-1] = nSep;
1822 }
1823 pGCC->pnSepLengths = pnsl;
1824 }else{
1825 sqlite3StrAccumSetError(&pGCC->str, SQLITE_NOMEM);
1826 }
larrybrdde13e62021-09-29 00:32:13 +00001827 }
1828#endif
drhb0689692007-11-01 17:38:30 +00001829 }
larrybrdde13e62021-09-29 00:32:13 +00001830#ifndef SQLITE_OMIT_WINDOWFUNC
1831 else{
drh3dd01112021-10-01 02:45:48 +00001832 pGCC->nFirstSepLength = sqlite3_value_bytes(argv[1]);
larrybrdde13e62021-09-29 00:32:13 +00001833 }
1834 pGCC->nAccum += 1;
1835#endif
drh07d31172009-02-02 21:57:05 +00001836 zVal = (char*)sqlite3_value_text(argv[0]);
1837 nVal = sqlite3_value_bytes(argv[0]);
larrybrdde13e62021-09-29 00:32:13 +00001838 if( zVal ) sqlite3_str_append(&pGCC->str, zVal, nVal);
drhb0689692007-11-01 17:38:30 +00001839 }
1840}
larrybrdde13e62021-09-29 00:32:13 +00001841
dan67a9b8e2018-06-22 20:51:35 +00001842#ifndef SQLITE_OMIT_WINDOWFUNC
dan03854d22018-06-08 11:45:28 +00001843static void groupConcatInverse(
1844 sqlite3_context *context,
1845 int argc,
1846 sqlite3_value **argv
1847){
larrybrdde13e62021-09-29 00:32:13 +00001848 GroupConcatCtx *pGCC;
drhc7bf5712018-07-09 22:49:01 +00001849 assert( argc==1 || argc==2 );
drh260ff082021-10-01 22:48:52 +00001850 (void)argc; /* Suppress unused parameter warning */
dan03854d22018-06-08 11:45:28 +00001851 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
larrybrdde13e62021-09-29 00:32:13 +00001852 pGCC = (GroupConcatCtx*)sqlite3_aggregate_context(context, sizeof(*pGCC));
1853 /* pGCC is always non-NULL since groupConcatStep() will have always
drhfd4b7282018-07-07 19:47:21 +00001854 ** run frist to initialize it */
larrybrdde13e62021-09-29 00:32:13 +00001855 if( ALWAYS(pGCC) ){
drh4fc80672021-10-12 22:55:04 +00001856 int nVS;
1857 /* Must call sqlite3_value_text() to convert the argument into text prior
1858 ** to invoking sqlite3_value_bytes(), in case the text encoding is UTF16 */
1859 (void)sqlite3_value_text(argv[0]);
1860 nVS = sqlite3_value_bytes(argv[0]);
larrybrdde13e62021-09-29 00:32:13 +00001861 pGCC->nAccum -= 1;
1862 if( pGCC->pnSepLengths!=0 ){
1863 assert(pGCC->nAccum >= 0);
1864 if( pGCC->nAccum>0 ){
drhf06db3e2021-10-01 00:25:06 +00001865 nVS += *pGCC->pnSepLengths;
1866 memmove(pGCC->pnSepLengths, pGCC->pnSepLengths+1,
1867 (pGCC->nAccum-1)*sizeof(int));
larrybrdde13e62021-09-29 00:32:13 +00001868 }
dan683b0ff2018-07-05 18:19:29 +00001869 }else{
larrybrdde13e62021-09-29 00:32:13 +00001870 /* If removing single accumulated string, harmlessly over-do. */
1871 nVS += pGCC->nFirstSepLength;
dan03854d22018-06-08 11:45:28 +00001872 }
larrybrdde13e62021-09-29 00:32:13 +00001873 if( nVS>=(int)pGCC->str.nChar ){
1874 pGCC->str.nChar = 0;
dan03854d22018-06-08 11:45:28 +00001875 }else{
larrybrdde13e62021-09-29 00:32:13 +00001876 pGCC->str.nChar -= nVS;
1877 memmove(pGCC->str.zText, &pGCC->str.zText[nVS], pGCC->str.nChar);
dan03854d22018-06-08 11:45:28 +00001878 }
larrybrdde13e62021-09-29 00:32:13 +00001879 if( pGCC->str.nChar==0 ){
1880 pGCC->str.mxAlloc = 0;
1881 sqlite3_free(pGCC->pnSepLengths);
1882 pGCC->pnSepLengths = 0;
1883 }
dan03854d22018-06-08 11:45:28 +00001884 }
1885}
dan67a9b8e2018-06-22 20:51:35 +00001886#else
1887# define groupConcatInverse 0
1888#endif /* SQLITE_OMIT_WINDOWFUNC */
drhb0689692007-11-01 17:38:30 +00001889static void groupConcatFinalize(sqlite3_context *context){
larrybrdde13e62021-09-29 00:32:13 +00001890 GroupConcatCtx *pGCC
1891 = (GroupConcatCtx*)sqlite3_aggregate_context(context, 0);
1892 if( pGCC ){
drh5bf47152021-10-03 00:12:43 +00001893 sqlite3ResultStrAccum(context, &pGCC->str);
larrybrdde13e62021-09-29 00:32:13 +00001894#ifndef SQLITE_OMIT_WINDOWFUNC
1895 sqlite3_free(pGCC->pnSepLengths);
1896#endif
drhb0689692007-11-01 17:38:30 +00001897 }
1898}
dan67a9b8e2018-06-22 20:51:35 +00001899#ifndef SQLITE_OMIT_WINDOWFUNC
dane2f781b2018-05-17 19:24:08 +00001900static void groupConcatValue(sqlite3_context *context){
larrybrdde13e62021-09-29 00:32:13 +00001901 GroupConcatCtx *pGCC
1902 = (GroupConcatCtx*)sqlite3_aggregate_context(context, 0);
1903 if( pGCC ){
1904 StrAccum *pAccum = &pGCC->str;
dane2f781b2018-05-17 19:24:08 +00001905 if( pAccum->accError==SQLITE_TOOBIG ){
1906 sqlite3_result_error_toobig(context);
1907 }else if( pAccum->accError==SQLITE_NOMEM ){
1908 sqlite3_result_error_nomem(context);
1909 }else{
1910 const char *zText = sqlite3_str_value(pAccum);
drhf06db3e2021-10-01 00:25:06 +00001911 sqlite3_result_text(context, zText, pAccum->nChar, SQLITE_TRANSIENT);
dane2f781b2018-05-17 19:24:08 +00001912 }
1913 }
1914}
dan67a9b8e2018-06-22 20:51:35 +00001915#else
1916# define groupConcatValue 0
1917#endif /* SQLITE_OMIT_WINDOWFUNC */
drh4e5ffc52004-08-31 00:52:37 +00001918
drhd3a149e2002-02-24 17:12:53 +00001919/*
drha4741842010-04-25 20:58:37 +00001920** This routine does per-connection function registration. Most
1921** of the built-in functions above are part of the global function set.
1922** This routine only deals with those that are not global.
drhdc04c582002-02-24 01:55:15 +00001923*/
drh80738d92016-02-15 00:34:16 +00001924void sqlite3RegisterPerConnectionBuiltinFunctions(sqlite3 *db){
drha4741842010-04-25 20:58:37 +00001925 int rc = sqlite3_overload_function(db, "MATCH", 2);
1926 assert( rc==SQLITE_NOMEM || rc==SQLITE_OK );
1927 if( rc==SQLITE_NOMEM ){
drh4a642b62016-02-05 01:55:27 +00001928 sqlite3OomFault(db);
danielk1977832a58a2007-06-22 15:21:15 +00001929 }
drh55ef4d92005-08-14 01:20:37 +00001930}
1931
1932/*
drh08652b52019-05-08 17:27:18 +00001933** Re-register the built-in LIKE functions. The caseSensitive
drh55ef4d92005-08-14 01:20:37 +00001934** parameter determines whether or not the LIKE operator is case
drh08652b52019-05-08 17:27:18 +00001935** sensitive.
drh55ef4d92005-08-14 01:20:37 +00001936*/
1937void sqlite3RegisterLikeFunctions(sqlite3 *db, int caseSensitive){
1938 struct compareInfo *pInfo;
drhea5c0402019-05-08 19:32:33 +00001939 int flags;
drh55ef4d92005-08-14 01:20:37 +00001940 if( caseSensitive ){
1941 pInfo = (struct compareInfo*)&likeInfoAlt;
drhea5c0402019-05-08 19:32:33 +00001942 flags = SQLITE_FUNC_LIKE | SQLITE_FUNC_CASE;
drh55ef4d92005-08-14 01:20:37 +00001943 }else{
1944 pInfo = (struct compareInfo*)&likeInfoNorm;
drhea5c0402019-05-08 19:32:33 +00001945 flags = SQLITE_FUNC_LIKE;
drh55ef4d92005-08-14 01:20:37 +00001946 }
dan660af932018-06-18 16:55:22 +00001947 sqlite3CreateFunc(db, "like", 2, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0, 0, 0);
1948 sqlite3CreateFunc(db, "like", 3, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0, 0, 0);
drhea5c0402019-05-08 19:32:33 +00001949 sqlite3FindFunction(db, "like", 2, SQLITE_UTF8, 0)->funcFlags |= flags;
1950 sqlite3FindFunction(db, "like", 3, SQLITE_UTF8, 0)->funcFlags |= flags;
drh55ef4d92005-08-14 01:20:37 +00001951}
1952
1953/*
1954** pExpr points to an expression which implements a function. If
1955** it is appropriate to apply the LIKE optimization to that function
drh1d42ea72017-07-27 20:24:29 +00001956** then set aWc[0] through aWc[2] to the wildcard characters and the
1957** escape character and then return TRUE. If the function is not a
1958** LIKE-style function then return FALSE.
1959**
1960** The expression "a LIKE b ESCAPE c" is only considered a valid LIKE
1961** operator if c is a string literal that is exactly one byte in length.
1962** That one byte is stored in aWc[3]. aWc[3] is set to zero if there is
1963** no ESCAPE clause.
drh16897072015-03-07 00:57:37 +00001964**
1965** *pIsNocase is set to true if uppercase and lowercase are equivalent for
1966** the function (default for LIKE). If the function makes the distinction
1967** between uppercase and lowercase (as does GLOB) then *pIsNocase is set to
1968** false.
drh55ef4d92005-08-14 01:20:37 +00001969*/
drhd64fe2f2005-08-28 17:00:23 +00001970int sqlite3IsLikeFunction(sqlite3 *db, Expr *pExpr, int *pIsNocase, char *aWc){
drh55ef4d92005-08-14 01:20:37 +00001971 FuncDef *pDef;
drh1d42ea72017-07-27 20:24:29 +00001972 int nExpr;
drh17988aa2021-01-22 20:28:30 +00001973 assert( pExpr!=0 );
1974 assert( pExpr->op==TK_FUNCTION );
drha4eeccd2021-10-07 17:43:30 +00001975 assert( ExprUseXList(pExpr) );
drh17988aa2021-01-22 20:28:30 +00001976 if( !pExpr->x.pList ){
drh55ef4d92005-08-14 01:20:37 +00001977 return 0;
1978 }
drh1d42ea72017-07-27 20:24:29 +00001979 nExpr = pExpr->x.pList->nExpr;
drhf9751072021-10-07 13:40:29 +00001980 assert( !ExprHasProperty(pExpr, EP_IntValue) );
drh1d42ea72017-07-27 20:24:29 +00001981 pDef = sqlite3FindFunction(db, pExpr->u.zToken, nExpr, SQLITE_UTF8, 0);
drh78b52202020-01-22 23:08:19 +00001982#ifdef SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION
1983 if( pDef==0 ) return 0;
1984#endif
drhd36e1042013-09-06 13:10:12 +00001985 if( NEVER(pDef==0) || (pDef->funcFlags & SQLITE_FUNC_LIKE)==0 ){
drh55ef4d92005-08-14 01:20:37 +00001986 return 0;
1987 }
1988
1989 /* The memcpy() statement assumes that the wildcard characters are
1990 ** the first three statements in the compareInfo structure. The
1991 ** asserts() that follow verify that assumption
1992 */
1993 memcpy(aWc, pDef->pUserData, 3);
1994 assert( (char*)&likeInfoAlt == (char*)&likeInfoAlt.matchAll );
1995 assert( &((char*)&likeInfoAlt)[1] == (char*)&likeInfoAlt.matchOne );
1996 assert( &((char*)&likeInfoAlt)[2] == (char*)&likeInfoAlt.matchSet );
drh589c7872020-03-19 18:13:28 +00001997
1998 if( nExpr<3 ){
1999 aWc[3] = 0;
2000 }else{
2001 Expr *pEscape = pExpr->x.pList->a[2].pExpr;
2002 char *zEscape;
2003 if( pEscape->op!=TK_STRING ) return 0;
drhf9751072021-10-07 13:40:29 +00002004 assert( !ExprHasProperty(pEscape, EP_IntValue) );
drh589c7872020-03-19 18:13:28 +00002005 zEscape = pEscape->u.zToken;
2006 if( zEscape[0]==0 || zEscape[1]!=0 ) return 0;
2007 if( zEscape[0]==aWc[0] ) return 0;
2008 if( zEscape[0]==aWc[1] ) return 0;
2009 aWc[3] = zEscape[0];
2010 }
2011
drhd36e1042013-09-06 13:10:12 +00002012 *pIsNocase = (pDef->funcFlags & SQLITE_FUNC_CASE)==0;
drh55ef4d92005-08-14 01:20:37 +00002013 return 1;
drhdc04c582002-02-24 01:55:15 +00002014}
danielk19778c0a7912008-08-20 14:49:23 +00002015
drh63f8f982020-12-07 21:13:06 +00002016/* Mathematical Constants */
2017#ifndef M_PI
2018# define M_PI 3.141592653589793238462643383279502884
2019#endif
2020#ifndef M_LN10
2021# define M_LN10 2.302585092994045684017991454684364208
2022#endif
2023#ifndef M_LN2
2024# define M_LN2 0.693147180559945309417232121458176568
2025#endif
2026
2027
drhf6e904b2020-12-07 17:15:32 +00002028/* Extra math functions that require linking with -lm
2029*/
2030#ifdef SQLITE_ENABLE_MATH_FUNCTIONS
2031/*
2032** Implementation SQL functions:
2033**
2034** ceil(X)
2035** ceiling(X)
2036** floor(X)
2037**
2038** The sqlite3_user_data() pointer is a pointer to the libm implementation
2039** of the underlying C function.
2040*/
2041static void ceilingFunc(
2042 sqlite3_context *context,
2043 int argc,
2044 sqlite3_value **argv
2045){
2046 assert( argc==1 );
drh63f8f982020-12-07 21:13:06 +00002047 switch( sqlite3_value_numeric_type(argv[0]) ){
2048 case SQLITE_INTEGER: {
drhf6e904b2020-12-07 17:15:32 +00002049 sqlite3_result_int64(context, sqlite3_value_int64(argv[0]));
2050 break;
drh63f8f982020-12-07 21:13:06 +00002051 }
2052 case SQLITE_FLOAT: {
drhf6e904b2020-12-07 17:15:32 +00002053 double (*x)(double) = (double(*)(double))sqlite3_user_data(context);
2054 sqlite3_result_double(context, x(sqlite3_value_double(argv[0])));
2055 break;
2056 }
drh63f8f982020-12-07 21:13:06 +00002057 default: {
2058 break;
2059 }
drhf6e904b2020-12-07 17:15:32 +00002060 }
2061}
2062
2063/*
drh4fd4a7a2021-01-01 01:44:06 +00002064** On some systems, ceil() and floor() are intrinsic function. You are
drhc2dbf352021-01-07 16:10:14 +00002065** unable to take a pointer to these functions. Hence, we here wrap them
drh4fd4a7a2021-01-01 01:44:06 +00002066** in our own actual functions.
2067*/
2068static double xCeil(double x){ return ceil(x); }
2069static double xFloor(double x){ return floor(x); }
2070
2071/*
drhf6e904b2020-12-07 17:15:32 +00002072** Implementation of SQL functions:
2073**
2074** ln(X) - natural logarithm
drh63f8f982020-12-07 21:13:06 +00002075** log(X) - log X base 10
2076** log10(X) - log X base 10
2077** log(B,X) - log X base B
drhf6e904b2020-12-07 17:15:32 +00002078*/
2079static void logFunc(
2080 sqlite3_context *context,
2081 int argc,
2082 sqlite3_value **argv
2083){
drh63f8f982020-12-07 21:13:06 +00002084 double x, b, ans;
drhf6e904b2020-12-07 17:15:32 +00002085 assert( argc==1 || argc==2 );
drh63f8f982020-12-07 21:13:06 +00002086 switch( sqlite3_value_numeric_type(argv[0]) ){
2087 case SQLITE_INTEGER:
2088 case SQLITE_FLOAT:
2089 x = sqlite3_value_double(argv[0]);
drh02d6f9b2021-01-29 16:20:16 +00002090 if( x<=0.0 ) return;
drh63f8f982020-12-07 21:13:06 +00002091 break;
2092 default:
2093 return;
drhf6e904b2020-12-07 17:15:32 +00002094 }
drhf6e904b2020-12-07 17:15:32 +00002095 if( argc==2 ){
drh63f8f982020-12-07 21:13:06 +00002096 switch( sqlite3_value_numeric_type(argv[0]) ){
2097 case SQLITE_INTEGER:
2098 case SQLITE_FLOAT:
drh02d6f9b2021-01-29 16:20:16 +00002099 b = log(x);
2100 if( b<=0.0 ) return;
drh63f8f982020-12-07 21:13:06 +00002101 x = sqlite3_value_double(argv[1]);
drh02d6f9b2021-01-29 16:20:16 +00002102 if( x<=0.0 ) return;
drh63f8f982020-12-07 21:13:06 +00002103 break;
2104 default:
2105 return;
drhf6e904b2020-12-07 17:15:32 +00002106 }
drh02d6f9b2021-01-29 16:20:16 +00002107 ans = log(x)/b;
drh63f8f982020-12-07 21:13:06 +00002108 }else{
drh63f8f982020-12-07 21:13:06 +00002109 switch( SQLITE_PTR_TO_INT(sqlite3_user_data(context)) ){
2110 case 1:
drh3c1572d2022-11-17 14:40:33 +00002111 ans = log10(x);
drh63f8f982020-12-07 21:13:06 +00002112 break;
2113 case 2:
drh3c1572d2022-11-17 14:40:33 +00002114 ans = log2(x);
drh63f8f982020-12-07 21:13:06 +00002115 break;
2116 default:
drh3c1572d2022-11-17 14:40:33 +00002117 ans = log(x);
drh63f8f982020-12-07 21:13:06 +00002118 break;
2119 }
drhf6e904b2020-12-07 17:15:32 +00002120 }
2121 sqlite3_result_double(context, ans);
2122}
drh63f8f982020-12-07 21:13:06 +00002123
2124/*
2125** Functions to converts degrees to radians and radians to degrees.
2126*/
2127static double degToRad(double x){ return x*(M_PI/180.0); }
2128static double radToDeg(double x){ return x*(180.0/M_PI); }
2129
2130/*
2131** Implementation of 1-argument SQL math functions:
2132**
2133** exp(X) - Compute e to the X-th power
2134*/
2135static void math1Func(
2136 sqlite3_context *context,
2137 int argc,
2138 sqlite3_value **argv
2139){
drh63f8f982020-12-07 21:13:06 +00002140 int type0;
2141 double v0, ans;
2142 double (*x)(double);
mistachkind97a4c02020-12-09 23:35:51 +00002143 assert( argc==1 );
drh63f8f982020-12-07 21:13:06 +00002144 type0 = sqlite3_value_numeric_type(argv[0]);
2145 if( type0!=SQLITE_INTEGER && type0!=SQLITE_FLOAT ) return;
2146 v0 = sqlite3_value_double(argv[0]);
2147 x = (double(*)(double))sqlite3_user_data(context);
2148 ans = x(v0);
2149 sqlite3_result_double(context, ans);
2150}
2151
2152/*
2153** Implementation of 2-argument SQL math functions:
2154**
2155** power(X,Y) - Compute X to the Y-th power
2156*/
2157static void math2Func(
2158 sqlite3_context *context,
2159 int argc,
2160 sqlite3_value **argv
2161){
drh63f8f982020-12-07 21:13:06 +00002162 int type0, type1;
2163 double v0, v1, ans;
2164 double (*x)(double,double);
mistachkind97a4c02020-12-09 23:35:51 +00002165 assert( argc==2 );
drh63f8f982020-12-07 21:13:06 +00002166 type0 = sqlite3_value_numeric_type(argv[0]);
2167 if( type0!=SQLITE_INTEGER && type0!=SQLITE_FLOAT ) return;
2168 type1 = sqlite3_value_numeric_type(argv[1]);
2169 if( type1!=SQLITE_INTEGER && type1!=SQLITE_FLOAT ) return;
2170 v0 = sqlite3_value_double(argv[0]);
2171 v1 = sqlite3_value_double(argv[1]);
2172 x = (double(*)(double,double))sqlite3_user_data(context);
2173 ans = x(v0, v1);
2174 sqlite3_result_double(context, ans);
2175}
2176
2177/*
drh81e5a9a2021-04-16 11:05:19 +00002178** Implementation of 0-argument pi() function.
drh63f8f982020-12-07 21:13:06 +00002179*/
2180static void piFunc(
2181 sqlite3_context *context,
2182 int argc,
2183 sqlite3_value **argv
2184){
2185 assert( argc==0 );
2186 sqlite3_result_double(context, M_PI);
2187}
2188
drhf6e904b2020-12-07 17:15:32 +00002189#endif /* SQLITE_ENABLE_MATH_FUNCTIONS */
2190
drh70a8ca32008-08-21 18:49:27 +00002191/*
drh63f8f982020-12-07 21:13:06 +00002192** Implementation of sign(X) function.
2193*/
2194static void signFunc(
2195 sqlite3_context *context,
2196 int argc,
2197 sqlite3_value **argv
2198){
drh63f8f982020-12-07 21:13:06 +00002199 int type0;
2200 double x;
drhe5baf5c2020-12-16 14:20:45 +00002201 UNUSED_PARAMETER(argc);
mistachkind97a4c02020-12-09 23:35:51 +00002202 assert( argc==1 );
drh63f8f982020-12-07 21:13:06 +00002203 type0 = sqlite3_value_numeric_type(argv[0]);
2204 if( type0!=SQLITE_INTEGER && type0!=SQLITE_FLOAT ) return;
2205 x = sqlite3_value_double(argv[0]);
2206 sqlite3_result_int(context, x<0.0 ? -1 : x>0.0 ? +1 : 0);
2207}
2208
2209/*
peter.d.reid60ec9142014-09-06 16:39:46 +00002210** All of the FuncDef structures in the aBuiltinFunc[] array above
drh777c5382008-08-21 20:21:34 +00002211** to the global function hash table. This occurs at start-time (as
2212** a consequence of calling sqlite3_initialize()).
2213**
2214** After this routine runs
drh70a8ca32008-08-21 18:49:27 +00002215*/
drh80738d92016-02-15 00:34:16 +00002216void sqlite3RegisterBuiltinFunctions(void){
danielk197793ce7412008-09-01 19:14:02 +00002217 /*
2218 ** The following array holds FuncDef structures for all of the functions
2219 ** defined in this file.
2220 **
2221 ** The array cannot be constant since changes are made to the
2222 ** FuncDef.pHash elements at start-time. The elements of this array
2223 ** are read-only after initialization is complete.
drh80738d92016-02-15 00:34:16 +00002224 **
2225 ** For peak efficiency, put the most frequently used function last.
danielk197793ce7412008-09-01 19:14:02 +00002226 */
drh80738d92016-02-15 00:34:16 +00002227 static FuncDef aBuiltinFunc[] = {
drh171c50e2020-01-01 15:43:30 +00002228/***** Functions only available with SQLITE_TESTCTRL_INTERNAL_FUNCTIONS *****/
drh3780f9a2021-09-17 13:07:15 +00002229#if !defined(SQLITE_UNTESTABLE)
drh171c50e2020-01-01 15:43:30 +00002230 TEST_FUNC(implies_nonnull_row, 2, INLINEFUNC_implies_nonnull_row, 0),
2231 TEST_FUNC(expr_compare, 2, INLINEFUNC_expr_compare, 0),
2232 TEST_FUNC(expr_implies_expr, 2, INLINEFUNC_expr_implies_expr, 0),
drh3780f9a2021-09-17 13:07:15 +00002233 TEST_FUNC(affinity, 1, INLINEFUNC_affinity, 0),
2234#endif /* !defined(SQLITE_UNTESTABLE) */
drh171c50e2020-01-01 15:43:30 +00002235/***** Regular functions *****/
drh80738d92016-02-15 00:34:16 +00002236#ifdef SQLITE_SOUNDEX
2237 FUNCTION(soundex, 1, 0, 0, soundexFunc ),
2238#endif
2239#ifndef SQLITE_OMIT_LOAD_EXTENSION
drh64de2a52019-12-31 18:39:23 +00002240 SFUNCTION(load_extension, 1, 0, 0, loadExt ),
2241 SFUNCTION(load_extension, 2, 0, 0, loadExt ),
drh80738d92016-02-15 00:34:16 +00002242#endif
2243#if SQLITE_USER_AUTHENTICATION
2244 FUNCTION(sqlite_crypt, 2, 0, 0, sqlite3CryptFunc ),
2245#endif
2246#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
2247 DFUNCTION(sqlite_compileoption_used,1, 0, 0, compileoptionusedFunc ),
2248 DFUNCTION(sqlite_compileoption_get, 1, 0, 0, compileoptiongetFunc ),
2249#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
drh25c42962020-01-01 13:55:08 +00002250 INLINE_FUNC(unlikely, 1, INLINEFUNC_unlikely, SQLITE_FUNC_UNLIKELY),
2251 INLINE_FUNC(likelihood, 2, INLINEFUNC_unlikely, SQLITE_FUNC_UNLIKELY),
2252 INLINE_FUNC(likely, 1, INLINEFUNC_unlikely, SQLITE_FUNC_UNLIKELY),
drh092457b2017-12-29 15:04:49 +00002253#ifdef SQLITE_ENABLE_OFFSET_SQL_FUNC
drh645682a2022-06-01 11:05:59 +00002254 INLINE_FUNC(sqlite_offset, 1, INLINEFUNC_sqlite_offset, 0 ),
drh092457b2017-12-29 15:04:49 +00002255#endif
danielk197793ce7412008-09-01 19:14:02 +00002256 FUNCTION(ltrim, 1, 1, 0, trimFunc ),
2257 FUNCTION(ltrim, 2, 1, 0, trimFunc ),
2258 FUNCTION(rtrim, 1, 2, 0, trimFunc ),
2259 FUNCTION(rtrim, 2, 2, 0, trimFunc ),
2260 FUNCTION(trim, 1, 3, 0, trimFunc ),
2261 FUNCTION(trim, 2, 3, 0, trimFunc ),
2262 FUNCTION(min, -1, 0, 1, minmaxFunc ),
2263 FUNCTION(min, 0, 0, 1, 0 ),
dan6fb2b542018-06-19 17:13:11 +00002264 WAGGREGATE(min, 1, 0, 1, minmaxStep, minMaxFinalize, minMaxValue, 0,
drhbb301232021-07-15 19:29:43 +00002265 SQLITE_FUNC_MINMAX|SQLITE_FUNC_ANYORDER ),
danielk197793ce7412008-09-01 19:14:02 +00002266 FUNCTION(max, -1, 1, 1, minmaxFunc ),
2267 FUNCTION(max, 0, 1, 1, 0 ),
dan6fb2b542018-06-19 17:13:11 +00002268 WAGGREGATE(max, 1, 1, 1, minmaxStep, minMaxFinalize, minMaxValue, 0,
drhbb301232021-07-15 19:29:43 +00002269 SQLITE_FUNC_MINMAX|SQLITE_FUNC_ANYORDER ),
drha748fdc2012-03-28 01:34:47 +00002270 FUNCTION2(typeof, 1, 0, 0, typeofFunc, SQLITE_FUNC_TYPEOF),
drh9d44f182022-01-09 16:54:02 +00002271 FUNCTION2(subtype, 1, 0, 0, subtypeFunc, SQLITE_FUNC_TYPEOF),
drha748fdc2012-03-28 01:34:47 +00002272 FUNCTION2(length, 1, 0, 0, lengthFunc, SQLITE_FUNC_LENGTH),
drhd55e0722012-10-25 03:07:29 +00002273 FUNCTION(instr, 2, 0, 0, instrFunc ),
drha5c14162013-12-17 15:03:06 +00002274 FUNCTION(printf, -1, 0, 0, printfFunc ),
drh6bcd5852022-01-08 21:00:38 +00002275 FUNCTION(format, -1, 0, 0, printfFunc ),
drhd495d8c2013-02-22 19:34:25 +00002276 FUNCTION(unicode, 1, 0, 0, unicodeFunc ),
2277 FUNCTION(char, -1, 0, 0, charFunc ),
danielk197793ce7412008-09-01 19:14:02 +00002278 FUNCTION(abs, 1, 0, 0, absFunc ),
shanefbd60f82009-02-04 03:59:25 +00002279#ifndef SQLITE_OMIT_FLOATING_POINT
danielk197793ce7412008-09-01 19:14:02 +00002280 FUNCTION(round, 1, 0, 0, roundFunc ),
2281 FUNCTION(round, 2, 0, 0, roundFunc ),
shanefbd60f82009-02-04 03:59:25 +00002282#endif
danielk197793ce7412008-09-01 19:14:02 +00002283 FUNCTION(upper, 1, 0, 0, upperFunc ),
2284 FUNCTION(lower, 1, 0, 0, lowerFunc ),
danielk197793ce7412008-09-01 19:14:02 +00002285 FUNCTION(hex, 1, 0, 0, hexFunc ),
drhffe421c2020-05-13 17:26:38 +00002286 INLINE_FUNC(ifnull, 2, INLINEFUNC_coalesce, 0 ),
drhb1fba282013-11-21 14:33:48 +00002287 VFUNCTION(random, 0, 0, 0, randomFunc ),
2288 VFUNCTION(randomblob, 1, 0, 0, randomBlob ),
danielk197793ce7412008-09-01 19:14:02 +00002289 FUNCTION(nullif, 2, 0, 1, nullifFunc ),
drh03bf26d2015-08-31 21:16:36 +00002290 DFUNCTION(sqlite_version, 0, 0, 0, versionFunc ),
2291 DFUNCTION(sqlite_source_id, 0, 0, 0, sourceidFunc ),
drh840561f2011-04-27 18:08:42 +00002292 FUNCTION(sqlite_log, 2, 0, 0, errlogFunc ),
danielk197793ce7412008-09-01 19:14:02 +00002293 FUNCTION(quote, 1, 0, 0, quoteFunc ),
drhb1fba282013-11-21 14:33:48 +00002294 VFUNCTION(last_insert_rowid, 0, 0, 0, last_insert_rowid),
2295 VFUNCTION(changes, 0, 0, 0, changes ),
2296 VFUNCTION(total_changes, 0, 0, 0, total_changes ),
danielk197793ce7412008-09-01 19:14:02 +00002297 FUNCTION(replace, 3, 0, 0, replaceFunc ),
2298 FUNCTION(zeroblob, 1, 0, 0, zeroblobFunc ),
drh80738d92016-02-15 00:34:16 +00002299 FUNCTION(substr, 2, 0, 0, substrFunc ),
2300 FUNCTION(substr, 3, 0, 0, substrFunc ),
drh1335ec72020-11-23 14:50:43 +00002301 FUNCTION(substring, 2, 0, 0, substrFunc ),
2302 FUNCTION(substring, 3, 0, 0, substrFunc ),
dan6fb2b542018-06-19 17:13:11 +00002303 WAGGREGATE(sum, 1,0,0, sumStep, sumFinalize, sumFinalize, sumInverse, 0),
2304 WAGGREGATE(total, 1,0,0, sumStep,totalFinalize,totalFinalize,sumInverse, 0),
2305 WAGGREGATE(avg, 1,0,0, sumStep, avgFinalize, avgFinalize, sumInverse, 0),
dan7262ca92018-07-02 12:07:32 +00002306 WAGGREGATE(count, 0,0,0, countStep,
drhbb301232021-07-15 19:29:43 +00002307 countFinalize, countFinalize, countInverse,
2308 SQLITE_FUNC_COUNT|SQLITE_FUNC_ANYORDER ),
dan7262ca92018-07-02 12:07:32 +00002309 WAGGREGATE(count, 1,0,0, countStep,
drhbb301232021-07-15 19:29:43 +00002310 countFinalize, countFinalize, countInverse, SQLITE_FUNC_ANYORDER ),
dan03854d22018-06-08 11:45:28 +00002311 WAGGREGATE(group_concat, 1, 0, 0, groupConcatStep,
dan6fb2b542018-06-19 17:13:11 +00002312 groupConcatFinalize, groupConcatValue, groupConcatInverse, 0),
dan03854d22018-06-08 11:45:28 +00002313 WAGGREGATE(group_concat, 2, 0, 0, groupConcatStep,
dan6fb2b542018-06-19 17:13:11 +00002314 groupConcatFinalize, groupConcatValue, groupConcatInverse, 0),
danielk197793ce7412008-09-01 19:14:02 +00002315
2316 LIKEFUNC(glob, 2, &globInfo, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
drhcc153132016-08-04 12:35:17 +00002317#ifdef SQLITE_CASE_SENSITIVE_LIKE
danielk197793ce7412008-09-01 19:14:02 +00002318 LIKEFUNC(like, 2, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
2319 LIKEFUNC(like, 3, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
drhcc153132016-08-04 12:35:17 +00002320#else
danielk197793ce7412008-09-01 19:14:02 +00002321 LIKEFUNC(like, 2, &likeInfoNorm, SQLITE_FUNC_LIKE),
2322 LIKEFUNC(like, 3, &likeInfoNorm, SQLITE_FUNC_LIKE),
drhcc153132016-08-04 12:35:17 +00002323#endif
2324#ifdef SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION
2325 FUNCTION(unknown, -1, 0, 0, unknownFunc ),
2326#endif
drh80738d92016-02-15 00:34:16 +00002327 FUNCTION(coalesce, 1, 0, 0, 0 ),
2328 FUNCTION(coalesce, 0, 0, 0, 0 ),
drhf6e904b2020-12-07 17:15:32 +00002329#ifdef SQLITE_ENABLE_MATH_FUNCTIONS
drh4fd4a7a2021-01-01 01:44:06 +00002330 MFUNCTION(ceil, 1, xCeil, ceilingFunc ),
2331 MFUNCTION(ceiling, 1, xCeil, ceilingFunc ),
2332 MFUNCTION(floor, 1, xFloor, ceilingFunc ),
mistachkind97a4c02020-12-09 23:35:51 +00002333#if SQLITE_HAVE_C99_MATH_FUNCS
drh63f8f982020-12-07 21:13:06 +00002334 MFUNCTION(trunc, 1, trunc, ceilingFunc ),
mistachkind97a4c02020-12-09 23:35:51 +00002335#endif
drhf6e904b2020-12-07 17:15:32 +00002336 FUNCTION(ln, 1, 0, 0, logFunc ),
2337 FUNCTION(log, 1, 1, 0, logFunc ),
2338 FUNCTION(log10, 1, 1, 0, logFunc ),
drh63f8f982020-12-07 21:13:06 +00002339 FUNCTION(log2, 1, 2, 0, logFunc ),
drhf6e904b2020-12-07 17:15:32 +00002340 FUNCTION(log, 2, 0, 0, logFunc ),
drh63f8f982020-12-07 21:13:06 +00002341 MFUNCTION(exp, 1, exp, math1Func ),
2342 MFUNCTION(pow, 2, pow, math2Func ),
2343 MFUNCTION(power, 2, pow, math2Func ),
2344 MFUNCTION(mod, 2, fmod, math2Func ),
2345 MFUNCTION(acos, 1, acos, math1Func ),
2346 MFUNCTION(asin, 1, asin, math1Func ),
2347 MFUNCTION(atan, 1, atan, math1Func ),
2348 MFUNCTION(atan2, 2, atan2, math2Func ),
2349 MFUNCTION(cos, 1, cos, math1Func ),
2350 MFUNCTION(sin, 1, sin, math1Func ),
2351 MFUNCTION(tan, 1, tan, math1Func ),
2352 MFUNCTION(cosh, 1, cosh, math1Func ),
2353 MFUNCTION(sinh, 1, sinh, math1Func ),
2354 MFUNCTION(tanh, 1, tanh, math1Func ),
mistachkind97a4c02020-12-09 23:35:51 +00002355#if SQLITE_HAVE_C99_MATH_FUNCS
drh63f8f982020-12-07 21:13:06 +00002356 MFUNCTION(acosh, 1, acosh, math1Func ),
2357 MFUNCTION(asinh, 1, asinh, math1Func ),
2358 MFUNCTION(atanh, 1, atanh, math1Func ),
mistachkind97a4c02020-12-09 23:35:51 +00002359#endif
drh63f8f982020-12-07 21:13:06 +00002360 MFUNCTION(sqrt, 1, sqrt, math1Func ),
2361 MFUNCTION(radians, 1, degToRad, math1Func ),
2362 MFUNCTION(degrees, 1, radToDeg, math1Func ),
2363 FUNCTION(pi, 0, 0, 0, piFunc ),
drhf6e904b2020-12-07 17:15:32 +00002364#endif /* SQLITE_ENABLE_MATH_FUNCTIONS */
drh63f8f982020-12-07 21:13:06 +00002365 FUNCTION(sign, 1, 0, 0, signFunc ),
2366 INLINE_FUNC(coalesce, -1, INLINEFUNC_coalesce, 0 ),
2367 INLINE_FUNC(iif, 3, INLINEFUNC_iif, 0 ),
danielk197793ce7412008-09-01 19:14:02 +00002368 };
drh545f5872010-04-24 14:02:59 +00002369#ifndef SQLITE_OMIT_ALTERTABLE
2370 sqlite3AlterFunctions();
2371#endif
dandfa552f2018-06-02 21:04:28 +00002372 sqlite3WindowFunctions();
drh80738d92016-02-15 00:34:16 +00002373 sqlite3RegisterDateTimeFunctions();
drh9dbf96b2022-01-06 01:40:09 +00002374 sqlite3RegisterJsonFunctions();
drh80738d92016-02-15 00:34:16 +00002375 sqlite3InsertBuiltinFuncs(aBuiltinFunc, ArraySize(aBuiltinFunc));
2376
2377#if 0 /* Enable to print out how the built-in functions are hashed */
2378 {
2379 int i;
2380 FuncDef *p;
2381 for(i=0; i<SQLITE_FUNC_HASH_SZ; i++){
2382 printf("FUNC-HASH %02d:", i);
2383 for(p=sqlite3BuiltinFunctions.a[i]; p; p=p->u.pHash){
2384 int n = sqlite3Strlen30(p->zName);
2385 int h = p->zName[0] + n;
drhf9751072021-10-07 13:40:29 +00002386 assert( p->funcFlags & SQLITE_FUNC_BUILTIN );
drh80738d92016-02-15 00:34:16 +00002387 printf(" %s(%d)", p->zName, h);
2388 }
2389 printf("\n");
2390 }
2391 }
2392#endif
drh70a8ca32008-08-21 18:49:27 +00002393}