blob: 6a279a9d71c73d33a0fdecf47db12999772324ab [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);
drh79b9bc42022-12-21 19:11:56 +00001085 i64 nBlob = sqlite3_value_bytes(pValue);
drhef95d552021-12-10 21:01:24 +00001086 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/*
dana50db432023-01-24 17:19:47 +00001227** Buffer zStr contains nStr bytes of utf-8 encoded text. Return 1 if zStr
1228** contains character ch, or 0 if it does not.
1229*/
1230static int strContainsChar(const u8 *zStr, int nStr, u32 ch){
1231 const u8 *zEnd = &zStr[nStr];
1232 const u8 *z = zStr;
1233 while( z<zEnd ){
1234 u32 tst = Utf8Read(z);
1235 if( tst==ch ) return 1;
1236 }
1237 return 0;
1238}
1239
1240/*
1241** The unhex() function. This function may be invoked with either one or
1242** two arguments. In both cases the first argument is interpreted as text
1243** a text value containing a set of pairs of hexadecimal digits which are
1244** decoded and returned as a blob.
1245**
1246** If there is only a single argument, then it must consist only of an
1247** even number of hexadeximal digits. Otherwise, return NULL.
1248**
1249** Or, if there is a second argument, then any character that appears in
1250** the second argument is also allowed to appear between pairs of hexadecimal
1251** digits in the first argument. If any other character appears in the
1252** first argument, or if one of the allowed characters appears between
1253** two hexadecimal digits that make up a single byte, NULL is returned.
1254**
1255** The following expressions are all true:
1256**
1257** unhex('ABCD') IS x'ABCD'
1258** unhex('AB CD') IS NULL
1259** unhex('AB CD', ' ') IS x'ABCD'
1260** unhex('A BCD', ' ') IS NULL
dane3c11d52023-01-23 14:11:34 +00001261*/
1262static void unhexFunc(
1263 sqlite3_context *pCtx,
1264 int argc,
1265 sqlite3_value **argv
1266){
dana50db432023-01-24 17:19:47 +00001267 const u8 *zPass = (const u8*)"";
1268 int nPass = 0;
dane3c11d52023-01-23 14:11:34 +00001269 const u8 *zHex = sqlite3_value_text(argv[0]);
1270 int nHex = sqlite3_value_bytes(argv[0]);
dana50db432023-01-24 17:19:47 +00001271#ifdef SQLITE_DEBUG
1272 const u8 *zEnd = &zHex[nHex];
1273#endif
1274 u8 *pBlob = 0;
1275 u8 *p = 0;
1276
1277 assert( argc==1 || argc==2 );
1278 if( argc==2 ){
1279 zPass = sqlite3_value_text(argv[1]);
1280 nPass = sqlite3_value_bytes(argv[1]);
1281 }
1282 if( !zHex || !zPass ) return;
1283
1284 p = pBlob = contextMalloc(pCtx, (nHex/2)+1);
1285 if( pBlob ){
1286 u8 c; /* Most significant digit of next byte */
1287 u8 d; /* Least significant digit of next byte */
1288
1289 while( (c = *zHex)!=0x00 ){
1290 while( !sqlite3Isxdigit(c) ){
1291 u32 ch = Utf8Read(zHex);
1292 assert( zHex<=zEnd );
1293 if( !strContainsChar(zPass, nPass, ch) ) goto unhex_null;
1294 c = *zHex;
1295 if( c==0x00 ) goto unhex_done;
dane3c11d52023-01-23 14:11:34 +00001296 }
dana50db432023-01-24 17:19:47 +00001297 zHex++;
1298 assert( *zEnd==0x00 );
1299 assert( zHex<=zEnd );
1300 d = *(zHex++);
1301 if( !sqlite3Isxdigit(d) ) goto unhex_null;
1302 *(p++) = (sqlite3HexToInt(c)<<4) | sqlite3HexToInt(d);
dane3c11d52023-01-23 14:11:34 +00001303 }
1304 }
dana50db432023-01-24 17:19:47 +00001305
1306 unhex_done:
1307 sqlite3_result_blob(pCtx, pBlob, (p - pBlob), sqlite3_free);
1308 return;
1309
1310 unhex_null:
1311 sqlite3_free(pBlob);
1312 return;
dane3c11d52023-01-23 14:11:34 +00001313}
1314
1315
1316/*
drh8cff3822007-05-02 02:08:28 +00001317** The zeroblob(N) function returns a zero-filled blob of size N bytes.
1318*/
1319static void zeroblobFunc(
1320 sqlite3_context *context,
1321 int argc,
1322 sqlite3_value **argv
1323){
drh98640a32007-06-07 19:08:32 +00001324 i64 n;
dana4d5ae82015-07-24 16:24:37 +00001325 int rc;
drh8cff3822007-05-02 02:08:28 +00001326 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001327 UNUSED_PARAMETER(argc);
drh98640a32007-06-07 19:08:32 +00001328 n = sqlite3_value_int64(argv[0]);
dana4d5ae82015-07-24 16:24:37 +00001329 if( n<0 ) n = 0;
1330 rc = sqlite3_result_zeroblob64(context, n); /* IMP: R-00293-64994 */
1331 if( rc ){
1332 sqlite3_result_error_code(context, rc);
drh98640a32007-06-07 19:08:32 +00001333 }
drh8cff3822007-05-02 02:08:28 +00001334}
1335
1336/*
drh26b6d902007-03-17 13:27:54 +00001337** The replace() function. Three arguments are all strings: call
1338** them A, B, and C. The result is also a string which is derived
drhf7b54962013-05-28 12:11:54 +00001339** from A by replacing every occurrence of B with C. The match
drh26b6d902007-03-17 13:27:54 +00001340** must be exact. Collating sequences are not used.
1341*/
1342static void replaceFunc(
1343 sqlite3_context *context,
1344 int argc,
1345 sqlite3_value **argv
1346){
1347 const unsigned char *zStr; /* The input string A */
1348 const unsigned char *zPattern; /* The pattern string B */
1349 const unsigned char *zRep; /* The replacement string C */
1350 unsigned char *zOut; /* The output */
1351 int nStr; /* Size of zStr */
1352 int nPattern; /* Size of zPattern */
1353 int nRep; /* Size of zRep */
drh2e6400b2007-05-08 15:46:18 +00001354 i64 nOut; /* Maximum size of zOut */
drh26b6d902007-03-17 13:27:54 +00001355 int loopLimit; /* Last zStr[] that might match zPattern[] */
1356 int i, j; /* Loop counters */
drhf3139522018-02-09 23:25:14 +00001357 unsigned cntExpand; /* Number zOut expansions */
1358 sqlite3 *db = sqlite3_context_db_handle(context);
drh26b6d902007-03-17 13:27:54 +00001359
1360 assert( argc==3 );
danielk1977f3d3c272008-11-19 16:52:44 +00001361 UNUSED_PARAMETER(argc);
drh26b6d902007-03-17 13:27:54 +00001362 zStr = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001363 if( zStr==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001364 nStr = sqlite3_value_bytes(argv[0]);
1365 assert( zStr==sqlite3_value_text(argv[0]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001366 zPattern = sqlite3_value_text(argv[1]);
drha605fe82009-02-01 18:08:40 +00001367 if( zPattern==0 ){
drh23336062009-02-03 13:19:12 +00001368 assert( sqlite3_value_type(argv[1])==SQLITE_NULL
1369 || sqlite3_context_db_handle(context)->mallocFailed );
drha605fe82009-02-01 18:08:40 +00001370 return;
1371 }
1372 if( zPattern[0]==0 ){
1373 assert( sqlite3_value_type(argv[1])!=SQLITE_NULL );
1374 sqlite3_result_value(context, argv[0]);
1375 return;
1376 }
drh1f0feef2007-05-15 13:27:07 +00001377 nPattern = sqlite3_value_bytes(argv[1]);
1378 assert( zPattern==sqlite3_value_text(argv[1]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001379 zRep = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +00001380 if( zRep==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001381 nRep = sqlite3_value_bytes(argv[2]);
1382 assert( zRep==sqlite3_value_text(argv[2]) );
drh2e6400b2007-05-08 15:46:18 +00001383 nOut = nStr + 1;
1384 assert( nOut<SQLITE_MAX_LENGTH );
drhb1a6c3c2008-03-20 16:30:17 +00001385 zOut = contextMalloc(context, (i64)nOut);
drh2e6400b2007-05-08 15:46:18 +00001386 if( zOut==0 ){
1387 return;
drh26b6d902007-03-17 13:27:54 +00001388 }
drh26b6d902007-03-17 13:27:54 +00001389 loopLimit = nStr - nPattern;
drhf3139522018-02-09 23:25:14 +00001390 cntExpand = 0;
drh26b6d902007-03-17 13:27:54 +00001391 for(i=j=0; i<=loopLimit; i++){
1392 if( zStr[i]!=zPattern[0] || memcmp(&zStr[i], zPattern, nPattern) ){
1393 zOut[j++] = zStr[i];
1394 }else{
drhf3139522018-02-09 23:25:14 +00001395 if( nRep>nPattern ){
1396 nOut += nRep - nPattern;
drhc86d82f2018-02-10 02:31:30 +00001397 testcase( nOut-1==db->aLimit[SQLITE_LIMIT_LENGTH] );
1398 testcase( nOut-2==db->aLimit[SQLITE_LIMIT_LENGTH] );
drhf3139522018-02-09 23:25:14 +00001399 if( nOut-1>db->aLimit[SQLITE_LIMIT_LENGTH] ){
1400 sqlite3_result_error_toobig(context);
1401 sqlite3_free(zOut);
1402 return;
1403 }
drhf3139522018-02-09 23:25:14 +00001404 cntExpand++;
1405 if( (cntExpand&(cntExpand-1))==0 ){
1406 /* Grow the size of the output buffer only on substitutions
1407 ** whose index is a power of two: 1, 2, 4, 8, 16, 32, ... */
1408 u8 *zOld;
1409 zOld = zOut;
drhd924e7b2020-05-17 00:26:44 +00001410 zOut = sqlite3Realloc(zOut, (int)nOut + (nOut - nStr - 1));
drhf3139522018-02-09 23:25:14 +00001411 if( zOut==0 ){
1412 sqlite3_result_error_nomem(context);
1413 sqlite3_free(zOld);
1414 return;
1415 }
1416 }
drh2e6400b2007-05-08 15:46:18 +00001417 }
drh26b6d902007-03-17 13:27:54 +00001418 memcpy(&zOut[j], zRep, nRep);
1419 j += nRep;
1420 i += nPattern-1;
1421 }
1422 }
drhf3139522018-02-09 23:25:14 +00001423 assert( j+nStr-i+1<=nOut );
drh26b6d902007-03-17 13:27:54 +00001424 memcpy(&zOut[j], &zStr[i], nStr-i);
1425 j += nStr - i;
1426 assert( j<=nOut );
1427 zOut[j] = 0;
1428 sqlite3_result_text(context, (char*)zOut, j, sqlite3_free);
1429}
1430
drh309b3382007-03-17 17:52:42 +00001431/*
1432** Implementation of the TRIM(), LTRIM(), and RTRIM() functions.
1433** The userdata is 0x1 for left trim, 0x2 for right trim, 0x3 for both.
1434*/
1435static void trimFunc(
1436 sqlite3_context *context,
1437 int argc,
1438 sqlite3_value **argv
1439){
1440 const unsigned char *zIn; /* Input string */
1441 const unsigned char *zCharSet; /* Set of characters to trim */
drh972da422021-06-15 14:34:21 +00001442 unsigned int nIn; /* Number of bytes in input */
drh7209c692008-04-27 18:40:11 +00001443 int flags; /* 1: trimleft 2: trimright 3: trim */
drhd1e3a612007-04-27 21:59:52 +00001444 int i; /* Loop counter */
drh972da422021-06-15 14:34:21 +00001445 unsigned int *aLen = 0; /* Length of each character in zCharSet */
drh1bd10f82008-12-10 21:19:56 +00001446 unsigned char **azChar = 0; /* Individual characters in zCharSet */
drhd1e3a612007-04-27 21:59:52 +00001447 int nChar; /* Number of characters in zCharSet */
1448
drh309b3382007-03-17 17:52:42 +00001449 if( sqlite3_value_type(argv[0])==SQLITE_NULL ){
1450 return;
1451 }
1452 zIn = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001453 if( zIn==0 ) return;
drh972da422021-06-15 14:34:21 +00001454 nIn = (unsigned)sqlite3_value_bytes(argv[0]);
drh1f0feef2007-05-15 13:27:07 +00001455 assert( zIn==sqlite3_value_text(argv[0]) );
drh309b3382007-03-17 17:52:42 +00001456 if( argc==1 ){
drh972da422021-06-15 14:34:21 +00001457 static const unsigned lenOne[] = { 1 };
danielk1977a4de4532008-09-02 15:44:08 +00001458 static unsigned char * const azOne[] = { (u8*)" " };
drhd1e3a612007-04-27 21:59:52 +00001459 nChar = 1;
drh972da422021-06-15 14:34:21 +00001460 aLen = (unsigned*)lenOne;
danielk1977bc67da42007-12-11 04:23:19 +00001461 azChar = (unsigned char **)azOne;
drhd1e3a612007-04-27 21:59:52 +00001462 zCharSet = 0;
drh7a521cf2007-04-25 18:23:52 +00001463 }else if( (zCharSet = sqlite3_value_text(argv[1]))==0 ){
drh309b3382007-03-17 17:52:42 +00001464 return;
drhd1e3a612007-04-27 21:59:52 +00001465 }else{
1466 const unsigned char *z;
1467 for(z=zCharSet, nChar=0; *z; nChar++){
drh4a919112007-05-15 11:55:09 +00001468 SQLITE_SKIP_UTF8(z);
drhd1e3a612007-04-27 21:59:52 +00001469 }
1470 if( nChar>0 ){
drh972da422021-06-15 14:34:21 +00001471 azChar = contextMalloc(context,
1472 ((i64)nChar)*(sizeof(char*)+sizeof(unsigned)));
drhd1e3a612007-04-27 21:59:52 +00001473 if( azChar==0 ){
1474 return;
1475 }
drh972da422021-06-15 14:34:21 +00001476 aLen = (unsigned*)&azChar[nChar];
drhd1e3a612007-04-27 21:59:52 +00001477 for(z=zCharSet, nChar=0; *z; nChar++){
danielk1977bc67da42007-12-11 04:23:19 +00001478 azChar[nChar] = (unsigned char *)z;
drh4a919112007-05-15 11:55:09 +00001479 SQLITE_SKIP_UTF8(z);
drh972da422021-06-15 14:34:21 +00001480 aLen[nChar] = (unsigned)(z - azChar[nChar]);
drhd1e3a612007-04-27 21:59:52 +00001481 }
1482 }
drh309b3382007-03-17 17:52:42 +00001483 }
drhd1e3a612007-04-27 21:59:52 +00001484 if( nChar>0 ){
shane1fc41292008-07-08 22:28:48 +00001485 flags = SQLITE_PTR_TO_INT(sqlite3_user_data(context));
drh309b3382007-03-17 17:52:42 +00001486 if( flags & 1 ){
drhd1e3a612007-04-27 21:59:52 +00001487 while( nIn>0 ){
drh972da422021-06-15 14:34:21 +00001488 unsigned int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001489 for(i=0; i<nChar; i++){
1490 len = aLen[i];
drh27e62db2009-04-02 10:16:17 +00001491 if( len<=nIn && memcmp(zIn, azChar[i], len)==0 ) break;
drhd1e3a612007-04-27 21:59:52 +00001492 }
1493 if( i>=nChar ) break;
1494 zIn += len;
1495 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001496 }
1497 }
1498 if( flags & 2 ){
drhd1e3a612007-04-27 21:59:52 +00001499 while( nIn>0 ){
drh972da422021-06-15 14:34:21 +00001500 unsigned int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001501 for(i=0; i<nChar; i++){
1502 len = aLen[i];
1503 if( len<=nIn && memcmp(&zIn[nIn-len],azChar[i],len)==0 ) break;
1504 }
1505 if( i>=nChar ) break;
1506 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001507 }
1508 }
drhd1e3a612007-04-27 21:59:52 +00001509 if( zCharSet ){
1510 sqlite3_free(azChar);
1511 }
drh309b3382007-03-17 17:52:42 +00001512 }
1513 sqlite3_result_text(context, (char*)zIn, nIn, SQLITE_TRANSIENT);
1514}
drh26b6d902007-03-17 13:27:54 +00001515
danielk1977a4de4532008-09-02 15:44:08 +00001516
drhcc153132016-08-04 12:35:17 +00001517#ifdef SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION
1518/*
1519** The "unknown" function is automatically substituted in place of
1520** any unrecognized function name when doing an EXPLAIN or EXPLAIN QUERY PLAN
1521** when the SQLITE_ENABLE_UNKNOWN_FUNCTION compile-time option is used.
1522** When the "sqlite3" command-line shell is built using this functionality,
1523** that allows an EXPLAIN or EXPLAIN QUERY PLAN for complex queries
1524** involving application-defined functions to be examined in a generic
1525** sqlite3 shell.
1526*/
1527static void unknownFunc(
1528 sqlite3_context *context,
1529 int argc,
1530 sqlite3_value **argv
1531){
1532 /* no-op */
drh3547e492022-12-23 14:49:24 +00001533 (void)context;
1534 (void)argc;
1535 (void)argv;
drhcc153132016-08-04 12:35:17 +00001536}
1537#endif /*SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION*/
1538
1539
drh2ba3ccc2009-12-08 02:06:08 +00001540/* IMP: R-25361-16150 This function is omitted from SQLite by default. It
1541** is only available if the SQLITE_SOUNDEX compile-time option is used
1542** when SQLite is built.
1543*/
drhd24cc422003-03-27 12:51:24 +00001544#ifdef SQLITE_SOUNDEX
1545/*
1546** Compute the soundex encoding of a word.
drh2ba3ccc2009-12-08 02:06:08 +00001547**
1548** IMP: R-59782-00072 The soundex(X) function returns a string that is the
1549** soundex encoding of the string X.
drhd24cc422003-03-27 12:51:24 +00001550*/
drh137c7282007-01-29 17:58:28 +00001551static void soundexFunc(
1552 sqlite3_context *context,
1553 int argc,
1554 sqlite3_value **argv
1555){
drhd24cc422003-03-27 12:51:24 +00001556 char zResult[8];
drh4c755c02004-08-08 20:22:17 +00001557 const u8 *zIn;
drhd24cc422003-03-27 12:51:24 +00001558 int i, j;
1559 static const unsigned char iCode[] = {
1560 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1561 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1562 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1563 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1564 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1565 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1566 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1567 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1568 };
1569 assert( argc==1 );
drh4c755c02004-08-08 20:22:17 +00001570 zIn = (u8*)sqlite3_value_text(argv[0]);
drhbdf67e02006-08-19 11:34:01 +00001571 if( zIn==0 ) zIn = (u8*)"";
drhdc86e2b2009-01-24 11:30:42 +00001572 for(i=0; zIn[i] && !sqlite3Isalpha(zIn[i]); i++){}
drhd24cc422003-03-27 12:51:24 +00001573 if( zIn[i] ){
drhbdf67e02006-08-19 11:34:01 +00001574 u8 prevcode = iCode[zIn[i]&0x7f];
danielk197778ca0e72009-01-20 16:53:39 +00001575 zResult[0] = sqlite3Toupper(zIn[i]);
drhd24cc422003-03-27 12:51:24 +00001576 for(j=1; j<4 && zIn[i]; i++){
1577 int code = iCode[zIn[i]&0x7f];
1578 if( code>0 ){
drhbdf67e02006-08-19 11:34:01 +00001579 if( code!=prevcode ){
1580 prevcode = code;
1581 zResult[j++] = code + '0';
1582 }
1583 }else{
1584 prevcode = 0;
drhd24cc422003-03-27 12:51:24 +00001585 }
1586 }
1587 while( j<4 ){
1588 zResult[j++] = '0';
1589 }
1590 zResult[j] = 0;
danielk1977d8123362004-06-12 09:25:12 +00001591 sqlite3_result_text(context, zResult, 4, SQLITE_TRANSIENT);
drhd24cc422003-03-27 12:51:24 +00001592 }else{
drh2ba3ccc2009-12-08 02:06:08 +00001593 /* IMP: R-64894-50321 The string "?000" is returned if the argument
1594 ** is NULL or contains no ASCII alphabetic characters. */
danielk1977d8123362004-06-12 09:25:12 +00001595 sqlite3_result_text(context, "?000", 4, SQLITE_STATIC);
drhd24cc422003-03-27 12:51:24 +00001596 }
1597}
drh2ba3ccc2009-12-08 02:06:08 +00001598#endif /* SQLITE_SOUNDEX */
drhd24cc422003-03-27 12:51:24 +00001599
drhfdb83b22006-06-17 14:12:47 +00001600#ifndef SQLITE_OMIT_LOAD_EXTENSION
1601/*
1602** A function that loads a shared-library extension then returns NULL.
1603*/
1604static void loadExt(sqlite3_context *context, int argc, sqlite3_value **argv){
danielk197765fd59f2006-06-24 11:51:33 +00001605 const char *zFile = (const char *)sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001606 const char *zProc;
drhfa4a4b92008-03-19 21:45:51 +00001607 sqlite3 *db = sqlite3_context_db_handle(context);
drhfdb83b22006-06-17 14:12:47 +00001608 char *zErrMsg = 0;
1609
drh191dd062016-04-21 01:30:09 +00001610 /* Disallow the load_extension() SQL function unless the SQLITE_LoadExtFunc
drh1a55ded2016-04-20 00:30:05 +00001611 ** flag is set. See the sqlite3_enable_load_extension() API.
1612 */
drhf602a162016-04-21 01:58:21 +00001613 if( (db->flags & SQLITE_LoadExtFunc)==0 ){
1614 sqlite3_result_error(context, "not authorized", -1);
1615 return;
1616 }
drh1a55ded2016-04-20 00:30:05 +00001617
drhfdb83b22006-06-17 14:12:47 +00001618 if( argc==2 ){
danielk197765fd59f2006-06-24 11:51:33 +00001619 zProc = (const char *)sqlite3_value_text(argv[1]);
drh7a521cf2007-04-25 18:23:52 +00001620 }else{
1621 zProc = 0;
drhfdb83b22006-06-17 14:12:47 +00001622 }
drh7a521cf2007-04-25 18:23:52 +00001623 if( zFile && sqlite3_load_extension(db, zFile, zProc, &zErrMsg) ){
drhfdb83b22006-06-17 14:12:47 +00001624 sqlite3_result_error(context, zErrMsg, -1);
1625 sqlite3_free(zErrMsg);
1626 }
1627}
1628#endif
1629
danielk197701427a62005-01-11 13:02:33 +00001630
drh0ac65892002-04-20 14:24:41 +00001631/*
drhd3a149e2002-02-24 17:12:53 +00001632** An instance of the following structure holds the context of a
drhdd5baa92002-02-27 19:50:59 +00001633** sum() or avg() aggregate computation.
1634*/
1635typedef struct SumCtx SumCtx;
1636struct SumCtx {
drh8c08e862006-02-11 17:34:00 +00001637 double rSum; /* Floating point sum */
1638 i64 iSum; /* Integer sum */
1639 i64 cnt; /* Number of elements summed */
1640 u8 overflow; /* True if integer overflow seen */
1641 u8 approx; /* True if non-integer value was input to the sum */
drhdd5baa92002-02-27 19:50:59 +00001642};
1643
1644/*
drha97fdd32006-01-12 22:17:50 +00001645** Routines used to compute the sum, average, and total.
1646**
1647** The SUM() function follows the (broken) SQL standard which means
1648** that it returns NULL if it sums over no inputs. TOTAL returns
1649** 0.0 in that case. In addition, TOTAL always returns a float where
1650** SUM might return an integer if it never encounters a floating point
drhc806d852006-05-11 13:25:39 +00001651** value. TOTAL never fails, but SUM might through an exception if
1652** it overflows an integer.
drhdd5baa92002-02-27 19:50:59 +00001653*/
danielk19770ae8b832004-05-25 12:05:56 +00001654static void sumStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drhdd5baa92002-02-27 19:50:59 +00001655 SumCtx *p;
drh3d1d95e2005-09-08 10:37:01 +00001656 int type;
drh3f219f42005-09-08 19:45:57 +00001657 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001658 UNUSED_PARAMETER(argc);
drh4f26d6c2004-05-26 23:25:30 +00001659 p = sqlite3_aggregate_context(context, sizeof(*p));
drh29d72102006-02-09 22:13:41 +00001660 type = sqlite3_value_numeric_type(argv[0]);
drh3d1d95e2005-09-08 10:37:01 +00001661 if( p && type!=SQLITE_NULL ){
drh739105c2002-05-29 23:22:23 +00001662 p->cnt++;
drh29d72102006-02-09 22:13:41 +00001663 if( type==SQLITE_INTEGER ){
drh8c08e862006-02-11 17:34:00 +00001664 i64 v = sqlite3_value_int64(argv[0]);
1665 p->rSum += v;
drh158b9cb2011-03-05 20:59:46 +00001666 if( (p->approx|p->overflow)==0 && sqlite3AddInt64(&p->iSum, v) ){
drha546ef22018-07-07 20:55:16 +00001667 p->approx = p->overflow = 1;
drh29d72102006-02-09 22:13:41 +00001668 }
1669 }else{
drh8c08e862006-02-11 17:34:00 +00001670 p->rSum += sqlite3_value_double(argv[0]);
drh29d72102006-02-09 22:13:41 +00001671 p->approx = 1;
drh3f219f42005-09-08 19:45:57 +00001672 }
drh739105c2002-05-29 23:22:23 +00001673 }
drhdd5baa92002-02-27 19:50:59 +00001674}
dan67a9b8e2018-06-22 20:51:35 +00001675#ifndef SQLITE_OMIT_WINDOWFUNC
danc3a20c12018-05-23 20:55:37 +00001676static void sumInverse(sqlite3_context *context, int argc, sqlite3_value**argv){
1677 SumCtx *p;
1678 int type;
1679 assert( argc==1 );
1680 UNUSED_PARAMETER(argc);
1681 p = sqlite3_aggregate_context(context, sizeof(*p));
1682 type = sqlite3_value_numeric_type(argv[0]);
drhfd4b7282018-07-07 19:47:21 +00001683 /* p is always non-NULL because sumStep() will have been called first
1684 ** to initialize it */
1685 if( ALWAYS(p) && type!=SQLITE_NULL ){
drha546ef22018-07-07 20:55:16 +00001686 assert( p->cnt>0 );
danc3a20c12018-05-23 20:55:37 +00001687 p->cnt--;
drha546ef22018-07-07 20:55:16 +00001688 assert( type==SQLITE_INTEGER || p->approx );
1689 if( type==SQLITE_INTEGER && p->approx==0 ){
danc3a20c12018-05-23 20:55:37 +00001690 i64 v = sqlite3_value_int64(argv[0]);
1691 p->rSum -= v;
drha546ef22018-07-07 20:55:16 +00001692 p->iSum -= v;
danc3a20c12018-05-23 20:55:37 +00001693 }else{
dand7368292018-07-02 17:45:59 +00001694 p->rSum -= sqlite3_value_double(argv[0]);
danc3a20c12018-05-23 20:55:37 +00001695 }
1696 }
1697}
dan67a9b8e2018-06-22 20:51:35 +00001698#else
1699# define sumInverse 0
1700#endif /* SQLITE_OMIT_WINDOWFUNC */
danielk19770ae8b832004-05-25 12:05:56 +00001701static void sumFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001702 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001703 p = sqlite3_aggregate_context(context, 0);
drhc2bd9132005-09-08 20:37:43 +00001704 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001705 if( p->overflow ){
1706 sqlite3_result_error(context,"integer overflow",-1);
1707 }else if( p->approx ){
1708 sqlite3_result_double(context, p->rSum);
drhc2bd9132005-09-08 20:37:43 +00001709 }else{
drh8c08e862006-02-11 17:34:00 +00001710 sqlite3_result_int64(context, p->iSum);
drhc2bd9132005-09-08 20:37:43 +00001711 }
drh3d1d95e2005-09-08 10:37:01 +00001712 }
drhdd5baa92002-02-27 19:50:59 +00001713}
danielk19770ae8b832004-05-25 12:05:56 +00001714static void avgFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001715 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001716 p = sqlite3_aggregate_context(context, 0);
drh739105c2002-05-29 23:22:23 +00001717 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001718 sqlite3_result_double(context, p->rSum/(double)p->cnt);
drhdd5baa92002-02-27 19:50:59 +00001719 }
1720}
drha97fdd32006-01-12 22:17:50 +00001721static void totalFinalize(sqlite3_context *context){
1722 SumCtx *p;
1723 p = sqlite3_aggregate_context(context, 0);
shanefbd60f82009-02-04 03:59:25 +00001724 /* (double)0 In case of SQLITE_OMIT_FLOATING_POINT... */
1725 sqlite3_result_double(context, p ? p->rSum : (double)0);
drha97fdd32006-01-12 22:17:50 +00001726}
drhdd5baa92002-02-27 19:50:59 +00001727
1728/*
drh0bce8352002-02-28 00:41:10 +00001729** The following structure keeps track of state information for the
1730** count() aggregate function.
1731*/
1732typedef struct CountCtx CountCtx;
1733struct CountCtx {
drhfc6ad392006-02-09 13:38:19 +00001734 i64 n;
dan7262ca92018-07-02 12:07:32 +00001735#ifdef SQLITE_DEBUG
1736 int bInverse; /* True if xInverse() ever called */
1737#endif
drh0bce8352002-02-28 00:41:10 +00001738};
drhdd5baa92002-02-27 19:50:59 +00001739
drh0bce8352002-02-28 00:41:10 +00001740/*
1741** Routines to implement the count() aggregate function.
1742*/
danielk19770ae8b832004-05-25 12:05:56 +00001743static void countStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drh0bce8352002-02-28 00:41:10 +00001744 CountCtx *p;
drh4f26d6c2004-05-26 23:25:30 +00001745 p = sqlite3_aggregate_context(context, sizeof(*p));
drh9c054832004-05-31 18:51:57 +00001746 if( (argc==0 || SQLITE_NULL!=sqlite3_value_type(argv[0])) && p ){
drh0bce8352002-02-28 00:41:10 +00001747 p->n++;
1748 }
drh2e79c3d2009-04-08 23:04:14 +00001749
drhd3264c72009-04-15 13:39:47 +00001750#ifndef SQLITE_OMIT_DEPRECATED
drh2e79c3d2009-04-08 23:04:14 +00001751 /* The sqlite3_aggregate_count() function is deprecated. But just to make
1752 ** sure it still operates correctly, verify that its count agrees with our
1753 ** internal count when using count(*) and when the total count can be
1754 ** expressed as a 32-bit integer. */
dan7262ca92018-07-02 12:07:32 +00001755 assert( argc==1 || p==0 || p->n>0x7fffffff || p->bInverse
drh2e79c3d2009-04-08 23:04:14 +00001756 || p->n==sqlite3_aggregate_count(context) );
drhd3264c72009-04-15 13:39:47 +00001757#endif
drh0bce8352002-02-28 00:41:10 +00001758}
danielk19770ae8b832004-05-25 12:05:56 +00001759static void countFinalize(sqlite3_context *context){
drh0bce8352002-02-28 00:41:10 +00001760 CountCtx *p;
drhabfcea22005-09-06 20:36:48 +00001761 p = sqlite3_aggregate_context(context, 0);
drhfc6ad392006-02-09 13:38:19 +00001762 sqlite3_result_int64(context, p ? p->n : 0);
drh0bce8352002-02-28 00:41:10 +00001763}
dan7262ca92018-07-02 12:07:32 +00001764#ifndef SQLITE_OMIT_WINDOWFUNC
1765static void countInverse(sqlite3_context *ctx, int argc, sqlite3_value **argv){
1766 CountCtx *p;
1767 p = sqlite3_aggregate_context(ctx, sizeof(*p));
drhfd4b7282018-07-07 19:47:21 +00001768 /* p is always non-NULL since countStep() will have been called first */
1769 if( (argc==0 || SQLITE_NULL!=sqlite3_value_type(argv[0])) && ALWAYS(p) ){
dan7262ca92018-07-02 12:07:32 +00001770 p->n--;
1771#ifdef SQLITE_DEBUG
1772 p->bInverse = 1;
1773#endif
1774 }
1775}
dan6b4b8822018-07-02 15:03:50 +00001776#else
1777# define countInverse 0
1778#endif /* SQLITE_OMIT_WINDOWFUNC */
drh0bce8352002-02-28 00:41:10 +00001779
1780/*
drh0bce8352002-02-28 00:41:10 +00001781** Routines to implement min() and max() aggregate functions.
1782*/
danielk197762c14b32008-11-19 09:05:26 +00001783static void minmaxStep(
1784 sqlite3_context *context,
1785 int NotUsed,
1786 sqlite3_value **argv
1787){
danielk197788208052004-05-25 01:13:20 +00001788 Mem *pArg = (Mem *)argv[0];
drh9eb516c2004-07-18 20:52:32 +00001789 Mem *pBest;
danielk197762c14b32008-11-19 09:05:26 +00001790 UNUSED_PARAMETER(NotUsed);
drh9eb516c2004-07-18 20:52:32 +00001791
drh9eb516c2004-07-18 20:52:32 +00001792 pBest = (Mem *)sqlite3_aggregate_context(context, sizeof(*pBest));
danielk19773aeab9e2004-06-24 00:20:04 +00001793 if( !pBest ) return;
drh268380c2004-02-25 13:47:31 +00001794
dan6fb2b542018-06-19 17:13:11 +00001795 if( sqlite3_value_type(pArg)==SQLITE_NULL ){
drh94a6d992012-02-02 18:42:09 +00001796 if( pBest->flags ) sqlite3SkipAccumulatorLoad(context);
1797 }else if( pBest->flags ){
drh9eb516c2004-07-18 20:52:32 +00001798 int max;
1799 int cmp;
danielk1977dc1bdc42004-06-11 10:51:27 +00001800 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk19777e18c252004-05-25 11:47:24 +00001801 /* This step function is used for both the min() and max() aggregates,
1802 ** the only difference between the two being that the sense of the
1803 ** comparison is inverted. For the max() aggregate, the
1804 ** sqlite3_user_data() function returns (void *)-1. For min() it
1805 ** returns (void *)db, where db is the sqlite3* database pointer.
1806 ** Therefore the next statement sets variable 'max' to 1 for the max()
1807 ** aggregate, or 0 for min().
1808 */
drh309b3382007-03-17 17:52:42 +00001809 max = sqlite3_user_data(context)!=0;
danielk1977dc1bdc42004-06-11 10:51:27 +00001810 cmp = sqlite3MemCompare(pBest, pArg, pColl);
danielk197788208052004-05-25 01:13:20 +00001811 if( (max && cmp<0) || (!max && cmp>0) ){
drhb21c8cd2007-08-21 19:33:56 +00001812 sqlite3VdbeMemCopy(pBest, pArg);
drh7a957892012-02-02 17:35:43 +00001813 }else{
1814 sqlite3SkipAccumulatorLoad(context);
danielk197788208052004-05-25 01:13:20 +00001815 }
drh268380c2004-02-25 13:47:31 +00001816 }else{
drh035e5632014-09-16 14:16:31 +00001817 pBest->db = sqlite3_context_db_handle(context);
drhb21c8cd2007-08-21 19:33:56 +00001818 sqlite3VdbeMemCopy(pBest, pArg);
drh0bce8352002-02-28 00:41:10 +00001819 }
1820}
dan6fb2b542018-06-19 17:13:11 +00001821static void minMaxValueFinalize(sqlite3_context *context, int bValue){
danielk197788208052004-05-25 01:13:20 +00001822 sqlite3_value *pRes;
drhabfcea22005-09-06 20:36:48 +00001823 pRes = (sqlite3_value *)sqlite3_aggregate_context(context, 0);
1824 if( pRes ){
drh94a6d992012-02-02 18:42:09 +00001825 if( pRes->flags ){
drhabfcea22005-09-06 20:36:48 +00001826 sqlite3_result_value(context, pRes);
1827 }
dan6fb2b542018-06-19 17:13:11 +00001828 if( bValue==0 ) sqlite3VdbeMemRelease(pRes);
drh0bce8352002-02-28 00:41:10 +00001829 }
1830}
dan67a9b8e2018-06-22 20:51:35 +00001831#ifndef SQLITE_OMIT_WINDOWFUNC
dan6fb2b542018-06-19 17:13:11 +00001832static void minMaxValue(sqlite3_context *context){
drhc7bf5712018-07-09 22:49:01 +00001833 minMaxValueFinalize(context, 1);
dan6fb2b542018-06-19 17:13:11 +00001834}
dan67a9b8e2018-06-22 20:51:35 +00001835#else
1836# define minMaxValue 0
1837#endif /* SQLITE_OMIT_WINDOWFUNC */
dan6fb2b542018-06-19 17:13:11 +00001838static void minMaxFinalize(sqlite3_context *context){
drhc7bf5712018-07-09 22:49:01 +00001839 minMaxValueFinalize(context, 0);
dan6fb2b542018-06-19 17:13:11 +00001840}
drhdd5baa92002-02-27 19:50:59 +00001841
drhb0689692007-11-01 17:38:30 +00001842/*
1843** group_concat(EXPR, ?SEPARATOR?)
drhf06db3e2021-10-01 00:25:06 +00001844**
1845** The SEPARATOR goes before the EXPR string. This is tragic. The
1846** groupConcatInverse() implementation would have been easier if the
1847** SEPARATOR were appended after EXPR. And the order is undocumented,
1848** so we could change it, in theory. But the old behavior has been
1849** around for so long that we dare not, for fear of breaking something.
drhb0689692007-11-01 17:38:30 +00001850*/
larrybrdde13e62021-09-29 00:32:13 +00001851typedef struct {
1852 StrAccum str; /* The accumulated concatenation */
1853#ifndef SQLITE_OMIT_WINDOWFUNC
1854 int nAccum; /* Number of strings presently concatenated */
drhf06db3e2021-10-01 00:25:06 +00001855 int nFirstSepLength; /* Used to detect separator length change */
larrybrdde13e62021-09-29 00:32:13 +00001856 /* If pnSepLengths!=0, refs an array of inter-string separator lengths,
drhf06db3e2021-10-01 00:25:06 +00001857 ** stored as actually incorporated into presently accumulated result.
1858 ** (Hence, its slots in use number nAccum-1 between method calls.)
1859 ** If pnSepLengths==0, nFirstSepLength is the length used throughout.
1860 */
larrybrdde13e62021-09-29 00:32:13 +00001861 int *pnSepLengths;
1862#endif
1863} GroupConcatCtx;
1864
drhb0689692007-11-01 17:38:30 +00001865static void groupConcatStep(
1866 sqlite3_context *context,
1867 int argc,
1868 sqlite3_value **argv
1869){
1870 const char *zVal;
larrybrdde13e62021-09-29 00:32:13 +00001871 GroupConcatCtx *pGCC;
drhb0689692007-11-01 17:38:30 +00001872 const char *zSep;
drh07d31172009-02-02 21:57:05 +00001873 int nVal, nSep;
1874 assert( argc==1 || argc==2 );
1875 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
larrybrdde13e62021-09-29 00:32:13 +00001876 pGCC = (GroupConcatCtx*)sqlite3_aggregate_context(context, sizeof(*pGCC));
1877 if( pGCC ){
drhbb4957f2008-03-20 14:03:29 +00001878 sqlite3 *db = sqlite3_context_db_handle(context);
larrybrdde13e62021-09-29 00:32:13 +00001879 int firstTerm = pGCC->str.mxAlloc==0;
1880 pGCC->str.mxAlloc = db->aLimit[SQLITE_LIMIT_LENGTH];
drhf06db3e2021-10-01 00:25:06 +00001881 if( argc==1 ){
1882 if( !firstTerm ){
1883 sqlite3_str_appendchar(&pGCC->str, 1, ',');
drhb0689692007-11-01 17:38:30 +00001884 }
larrybrdde13e62021-09-29 00:32:13 +00001885#ifndef SQLITE_OMIT_WINDOWFUNC
drhf06db3e2021-10-01 00:25:06 +00001886 else{
1887 pGCC->nFirstSepLength = 1;
1888 }
1889#endif
1890 }else if( !firstTerm ){
1891 zSep = (char*)sqlite3_value_text(argv[1]);
1892 nSep = sqlite3_value_bytes(argv[1]);
1893 if( zSep ){
1894 sqlite3_str_append(&pGCC->str, zSep, nSep);
1895 }
1896#ifndef SQLITE_OMIT_WINDOWFUNC
1897 else{
1898 nSep = 0;
1899 }
larrybrdde13e62021-09-29 00:32:13 +00001900 if( nSep != pGCC->nFirstSepLength || pGCC->pnSepLengths != 0 ){
drhf06db3e2021-10-01 00:25:06 +00001901 int *pnsl = pGCC->pnSepLengths;
1902 if( pnsl == 0 ){
1903 /* First separator length variation seen, start tracking them. */
1904 pnsl = (int*)sqlite3_malloc64((pGCC->nAccum+1) * sizeof(int));
1905 if( pnsl!=0 ){
1906 int i = 0, nA = pGCC->nAccum-1;
1907 while( i<nA ) pnsl[i++] = pGCC->nFirstSepLength;
1908 }
1909 }else{
1910 pnsl = (int*)sqlite3_realloc64(pnsl, pGCC->nAccum * sizeof(int));
1911 }
1912 if( pnsl!=0 ){
drh3dd01112021-10-01 02:45:48 +00001913 if( ALWAYS(pGCC->nAccum>0) ){
drhf06db3e2021-10-01 00:25:06 +00001914 pnsl[pGCC->nAccum-1] = nSep;
1915 }
1916 pGCC->pnSepLengths = pnsl;
1917 }else{
1918 sqlite3StrAccumSetError(&pGCC->str, SQLITE_NOMEM);
1919 }
larrybrdde13e62021-09-29 00:32:13 +00001920 }
1921#endif
drhb0689692007-11-01 17:38:30 +00001922 }
larrybrdde13e62021-09-29 00:32:13 +00001923#ifndef SQLITE_OMIT_WINDOWFUNC
1924 else{
drh3dd01112021-10-01 02:45:48 +00001925 pGCC->nFirstSepLength = sqlite3_value_bytes(argv[1]);
larrybrdde13e62021-09-29 00:32:13 +00001926 }
1927 pGCC->nAccum += 1;
1928#endif
drh07d31172009-02-02 21:57:05 +00001929 zVal = (char*)sqlite3_value_text(argv[0]);
1930 nVal = sqlite3_value_bytes(argv[0]);
larrybrdde13e62021-09-29 00:32:13 +00001931 if( zVal ) sqlite3_str_append(&pGCC->str, zVal, nVal);
drhb0689692007-11-01 17:38:30 +00001932 }
1933}
larrybrdde13e62021-09-29 00:32:13 +00001934
dan67a9b8e2018-06-22 20:51:35 +00001935#ifndef SQLITE_OMIT_WINDOWFUNC
dan03854d22018-06-08 11:45:28 +00001936static void groupConcatInverse(
1937 sqlite3_context *context,
1938 int argc,
1939 sqlite3_value **argv
1940){
larrybrdde13e62021-09-29 00:32:13 +00001941 GroupConcatCtx *pGCC;
drhc7bf5712018-07-09 22:49:01 +00001942 assert( argc==1 || argc==2 );
drh260ff082021-10-01 22:48:52 +00001943 (void)argc; /* Suppress unused parameter warning */
dan03854d22018-06-08 11:45:28 +00001944 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
larrybrdde13e62021-09-29 00:32:13 +00001945 pGCC = (GroupConcatCtx*)sqlite3_aggregate_context(context, sizeof(*pGCC));
1946 /* pGCC is always non-NULL since groupConcatStep() will have always
drhfd4b7282018-07-07 19:47:21 +00001947 ** run frist to initialize it */
larrybrdde13e62021-09-29 00:32:13 +00001948 if( ALWAYS(pGCC) ){
drh4fc80672021-10-12 22:55:04 +00001949 int nVS;
1950 /* Must call sqlite3_value_text() to convert the argument into text prior
1951 ** to invoking sqlite3_value_bytes(), in case the text encoding is UTF16 */
1952 (void)sqlite3_value_text(argv[0]);
1953 nVS = sqlite3_value_bytes(argv[0]);
larrybrdde13e62021-09-29 00:32:13 +00001954 pGCC->nAccum -= 1;
1955 if( pGCC->pnSepLengths!=0 ){
1956 assert(pGCC->nAccum >= 0);
1957 if( pGCC->nAccum>0 ){
drhf06db3e2021-10-01 00:25:06 +00001958 nVS += *pGCC->pnSepLengths;
1959 memmove(pGCC->pnSepLengths, pGCC->pnSepLengths+1,
1960 (pGCC->nAccum-1)*sizeof(int));
larrybrdde13e62021-09-29 00:32:13 +00001961 }
dan683b0ff2018-07-05 18:19:29 +00001962 }else{
larrybrdde13e62021-09-29 00:32:13 +00001963 /* If removing single accumulated string, harmlessly over-do. */
1964 nVS += pGCC->nFirstSepLength;
dan03854d22018-06-08 11:45:28 +00001965 }
larrybrdde13e62021-09-29 00:32:13 +00001966 if( nVS>=(int)pGCC->str.nChar ){
1967 pGCC->str.nChar = 0;
dan03854d22018-06-08 11:45:28 +00001968 }else{
larrybrdde13e62021-09-29 00:32:13 +00001969 pGCC->str.nChar -= nVS;
1970 memmove(pGCC->str.zText, &pGCC->str.zText[nVS], pGCC->str.nChar);
dan03854d22018-06-08 11:45:28 +00001971 }
larrybrdde13e62021-09-29 00:32:13 +00001972 if( pGCC->str.nChar==0 ){
1973 pGCC->str.mxAlloc = 0;
1974 sqlite3_free(pGCC->pnSepLengths);
1975 pGCC->pnSepLengths = 0;
1976 }
dan03854d22018-06-08 11:45:28 +00001977 }
1978}
dan67a9b8e2018-06-22 20:51:35 +00001979#else
1980# define groupConcatInverse 0
1981#endif /* SQLITE_OMIT_WINDOWFUNC */
drhb0689692007-11-01 17:38:30 +00001982static void groupConcatFinalize(sqlite3_context *context){
larrybrdde13e62021-09-29 00:32:13 +00001983 GroupConcatCtx *pGCC
1984 = (GroupConcatCtx*)sqlite3_aggregate_context(context, 0);
1985 if( pGCC ){
drh5bf47152021-10-03 00:12:43 +00001986 sqlite3ResultStrAccum(context, &pGCC->str);
larrybrdde13e62021-09-29 00:32:13 +00001987#ifndef SQLITE_OMIT_WINDOWFUNC
1988 sqlite3_free(pGCC->pnSepLengths);
1989#endif
drhb0689692007-11-01 17:38:30 +00001990 }
1991}
dan67a9b8e2018-06-22 20:51:35 +00001992#ifndef SQLITE_OMIT_WINDOWFUNC
dane2f781b2018-05-17 19:24:08 +00001993static void groupConcatValue(sqlite3_context *context){
larrybrdde13e62021-09-29 00:32:13 +00001994 GroupConcatCtx *pGCC
1995 = (GroupConcatCtx*)sqlite3_aggregate_context(context, 0);
1996 if( pGCC ){
1997 StrAccum *pAccum = &pGCC->str;
dane2f781b2018-05-17 19:24:08 +00001998 if( pAccum->accError==SQLITE_TOOBIG ){
1999 sqlite3_result_error_toobig(context);
2000 }else if( pAccum->accError==SQLITE_NOMEM ){
2001 sqlite3_result_error_nomem(context);
2002 }else{
2003 const char *zText = sqlite3_str_value(pAccum);
drhf06db3e2021-10-01 00:25:06 +00002004 sqlite3_result_text(context, zText, pAccum->nChar, SQLITE_TRANSIENT);
dane2f781b2018-05-17 19:24:08 +00002005 }
2006 }
2007}
dan67a9b8e2018-06-22 20:51:35 +00002008#else
2009# define groupConcatValue 0
2010#endif /* SQLITE_OMIT_WINDOWFUNC */
drh4e5ffc52004-08-31 00:52:37 +00002011
drhd3a149e2002-02-24 17:12:53 +00002012/*
drha4741842010-04-25 20:58:37 +00002013** This routine does per-connection function registration. Most
2014** of the built-in functions above are part of the global function set.
2015** This routine only deals with those that are not global.
drhdc04c582002-02-24 01:55:15 +00002016*/
drh80738d92016-02-15 00:34:16 +00002017void sqlite3RegisterPerConnectionBuiltinFunctions(sqlite3 *db){
drha4741842010-04-25 20:58:37 +00002018 int rc = sqlite3_overload_function(db, "MATCH", 2);
2019 assert( rc==SQLITE_NOMEM || rc==SQLITE_OK );
2020 if( rc==SQLITE_NOMEM ){
drh4a642b62016-02-05 01:55:27 +00002021 sqlite3OomFault(db);
danielk1977832a58a2007-06-22 15:21:15 +00002022 }
drh55ef4d92005-08-14 01:20:37 +00002023}
2024
2025/*
drh08652b52019-05-08 17:27:18 +00002026** Re-register the built-in LIKE functions. The caseSensitive
drh55ef4d92005-08-14 01:20:37 +00002027** parameter determines whether or not the LIKE operator is case
drh08652b52019-05-08 17:27:18 +00002028** sensitive.
drh55ef4d92005-08-14 01:20:37 +00002029*/
2030void sqlite3RegisterLikeFunctions(sqlite3 *db, int caseSensitive){
2031 struct compareInfo *pInfo;
drhea5c0402019-05-08 19:32:33 +00002032 int flags;
drh55ef4d92005-08-14 01:20:37 +00002033 if( caseSensitive ){
2034 pInfo = (struct compareInfo*)&likeInfoAlt;
drhea5c0402019-05-08 19:32:33 +00002035 flags = SQLITE_FUNC_LIKE | SQLITE_FUNC_CASE;
drh55ef4d92005-08-14 01:20:37 +00002036 }else{
2037 pInfo = (struct compareInfo*)&likeInfoNorm;
drhea5c0402019-05-08 19:32:33 +00002038 flags = SQLITE_FUNC_LIKE;
drh55ef4d92005-08-14 01:20:37 +00002039 }
dan660af932018-06-18 16:55:22 +00002040 sqlite3CreateFunc(db, "like", 2, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0, 0, 0);
2041 sqlite3CreateFunc(db, "like", 3, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0, 0, 0);
drhea5c0402019-05-08 19:32:33 +00002042 sqlite3FindFunction(db, "like", 2, SQLITE_UTF8, 0)->funcFlags |= flags;
2043 sqlite3FindFunction(db, "like", 3, SQLITE_UTF8, 0)->funcFlags |= flags;
drh55ef4d92005-08-14 01:20:37 +00002044}
2045
2046/*
2047** pExpr points to an expression which implements a function. If
2048** it is appropriate to apply the LIKE optimization to that function
drh1d42ea72017-07-27 20:24:29 +00002049** then set aWc[0] through aWc[2] to the wildcard characters and the
2050** escape character and then return TRUE. If the function is not a
2051** LIKE-style function then return FALSE.
2052**
2053** The expression "a LIKE b ESCAPE c" is only considered a valid LIKE
2054** operator if c is a string literal that is exactly one byte in length.
2055** That one byte is stored in aWc[3]. aWc[3] is set to zero if there is
2056** no ESCAPE clause.
drh16897072015-03-07 00:57:37 +00002057**
2058** *pIsNocase is set to true if uppercase and lowercase are equivalent for
2059** the function (default for LIKE). If the function makes the distinction
2060** between uppercase and lowercase (as does GLOB) then *pIsNocase is set to
2061** false.
drh55ef4d92005-08-14 01:20:37 +00002062*/
drhd64fe2f2005-08-28 17:00:23 +00002063int sqlite3IsLikeFunction(sqlite3 *db, Expr *pExpr, int *pIsNocase, char *aWc){
drh55ef4d92005-08-14 01:20:37 +00002064 FuncDef *pDef;
drh1d42ea72017-07-27 20:24:29 +00002065 int nExpr;
drh17988aa2021-01-22 20:28:30 +00002066 assert( pExpr!=0 );
2067 assert( pExpr->op==TK_FUNCTION );
drha4eeccd2021-10-07 17:43:30 +00002068 assert( ExprUseXList(pExpr) );
drh17988aa2021-01-22 20:28:30 +00002069 if( !pExpr->x.pList ){
drh55ef4d92005-08-14 01:20:37 +00002070 return 0;
2071 }
drh1d42ea72017-07-27 20:24:29 +00002072 nExpr = pExpr->x.pList->nExpr;
drhf9751072021-10-07 13:40:29 +00002073 assert( !ExprHasProperty(pExpr, EP_IntValue) );
drh1d42ea72017-07-27 20:24:29 +00002074 pDef = sqlite3FindFunction(db, pExpr->u.zToken, nExpr, SQLITE_UTF8, 0);
drh78b52202020-01-22 23:08:19 +00002075#ifdef SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION
2076 if( pDef==0 ) return 0;
2077#endif
drhd36e1042013-09-06 13:10:12 +00002078 if( NEVER(pDef==0) || (pDef->funcFlags & SQLITE_FUNC_LIKE)==0 ){
drh55ef4d92005-08-14 01:20:37 +00002079 return 0;
2080 }
2081
2082 /* The memcpy() statement assumes that the wildcard characters are
2083 ** the first three statements in the compareInfo structure. The
2084 ** asserts() that follow verify that assumption
2085 */
2086 memcpy(aWc, pDef->pUserData, 3);
2087 assert( (char*)&likeInfoAlt == (char*)&likeInfoAlt.matchAll );
2088 assert( &((char*)&likeInfoAlt)[1] == (char*)&likeInfoAlt.matchOne );
2089 assert( &((char*)&likeInfoAlt)[2] == (char*)&likeInfoAlt.matchSet );
drh589c7872020-03-19 18:13:28 +00002090
2091 if( nExpr<3 ){
2092 aWc[3] = 0;
2093 }else{
2094 Expr *pEscape = pExpr->x.pList->a[2].pExpr;
2095 char *zEscape;
2096 if( pEscape->op!=TK_STRING ) return 0;
drhf9751072021-10-07 13:40:29 +00002097 assert( !ExprHasProperty(pEscape, EP_IntValue) );
drh589c7872020-03-19 18:13:28 +00002098 zEscape = pEscape->u.zToken;
2099 if( zEscape[0]==0 || zEscape[1]!=0 ) return 0;
2100 if( zEscape[0]==aWc[0] ) return 0;
2101 if( zEscape[0]==aWc[1] ) return 0;
2102 aWc[3] = zEscape[0];
2103 }
2104
drhd36e1042013-09-06 13:10:12 +00002105 *pIsNocase = (pDef->funcFlags & SQLITE_FUNC_CASE)==0;
drh55ef4d92005-08-14 01:20:37 +00002106 return 1;
drhdc04c582002-02-24 01:55:15 +00002107}
danielk19778c0a7912008-08-20 14:49:23 +00002108
drh63f8f982020-12-07 21:13:06 +00002109/* Mathematical Constants */
2110#ifndef M_PI
2111# define M_PI 3.141592653589793238462643383279502884
2112#endif
2113#ifndef M_LN10
2114# define M_LN10 2.302585092994045684017991454684364208
2115#endif
2116#ifndef M_LN2
2117# define M_LN2 0.693147180559945309417232121458176568
2118#endif
2119
2120
drhf6e904b2020-12-07 17:15:32 +00002121/* Extra math functions that require linking with -lm
2122*/
2123#ifdef SQLITE_ENABLE_MATH_FUNCTIONS
2124/*
2125** Implementation SQL functions:
2126**
2127** ceil(X)
2128** ceiling(X)
2129** floor(X)
2130**
2131** The sqlite3_user_data() pointer is a pointer to the libm implementation
2132** of the underlying C function.
2133*/
2134static void ceilingFunc(
2135 sqlite3_context *context,
2136 int argc,
2137 sqlite3_value **argv
2138){
2139 assert( argc==1 );
drh63f8f982020-12-07 21:13:06 +00002140 switch( sqlite3_value_numeric_type(argv[0]) ){
2141 case SQLITE_INTEGER: {
drhf6e904b2020-12-07 17:15:32 +00002142 sqlite3_result_int64(context, sqlite3_value_int64(argv[0]));
2143 break;
drh63f8f982020-12-07 21:13:06 +00002144 }
2145 case SQLITE_FLOAT: {
drhf6e904b2020-12-07 17:15:32 +00002146 double (*x)(double) = (double(*)(double))sqlite3_user_data(context);
2147 sqlite3_result_double(context, x(sqlite3_value_double(argv[0])));
2148 break;
2149 }
drh63f8f982020-12-07 21:13:06 +00002150 default: {
2151 break;
2152 }
drhf6e904b2020-12-07 17:15:32 +00002153 }
2154}
2155
2156/*
drh4fd4a7a2021-01-01 01:44:06 +00002157** On some systems, ceil() and floor() are intrinsic function. You are
drhc2dbf352021-01-07 16:10:14 +00002158** unable to take a pointer to these functions. Hence, we here wrap them
drh4fd4a7a2021-01-01 01:44:06 +00002159** in our own actual functions.
2160*/
2161static double xCeil(double x){ return ceil(x); }
2162static double xFloor(double x){ return floor(x); }
2163
2164/*
drhf6e904b2020-12-07 17:15:32 +00002165** Implementation of SQL functions:
2166**
2167** ln(X) - natural logarithm
drh63f8f982020-12-07 21:13:06 +00002168** log(X) - log X base 10
2169** log10(X) - log X base 10
2170** log(B,X) - log X base B
drhf6e904b2020-12-07 17:15:32 +00002171*/
2172static void logFunc(
2173 sqlite3_context *context,
2174 int argc,
2175 sqlite3_value **argv
2176){
drh63f8f982020-12-07 21:13:06 +00002177 double x, b, ans;
drhf6e904b2020-12-07 17:15:32 +00002178 assert( argc==1 || argc==2 );
drh63f8f982020-12-07 21:13:06 +00002179 switch( sqlite3_value_numeric_type(argv[0]) ){
2180 case SQLITE_INTEGER:
2181 case SQLITE_FLOAT:
2182 x = sqlite3_value_double(argv[0]);
drh02d6f9b2021-01-29 16:20:16 +00002183 if( x<=0.0 ) return;
drh63f8f982020-12-07 21:13:06 +00002184 break;
2185 default:
2186 return;
drhf6e904b2020-12-07 17:15:32 +00002187 }
drhf6e904b2020-12-07 17:15:32 +00002188 if( argc==2 ){
drh63f8f982020-12-07 21:13:06 +00002189 switch( sqlite3_value_numeric_type(argv[0]) ){
2190 case SQLITE_INTEGER:
2191 case SQLITE_FLOAT:
drh02d6f9b2021-01-29 16:20:16 +00002192 b = log(x);
2193 if( b<=0.0 ) return;
drh63f8f982020-12-07 21:13:06 +00002194 x = sqlite3_value_double(argv[1]);
drh02d6f9b2021-01-29 16:20:16 +00002195 if( x<=0.0 ) return;
drh63f8f982020-12-07 21:13:06 +00002196 break;
2197 default:
2198 return;
drhf6e904b2020-12-07 17:15:32 +00002199 }
drh02d6f9b2021-01-29 16:20:16 +00002200 ans = log(x)/b;
drh63f8f982020-12-07 21:13:06 +00002201 }else{
drh63f8f982020-12-07 21:13:06 +00002202 switch( SQLITE_PTR_TO_INT(sqlite3_user_data(context)) ){
2203 case 1:
drh3c1572d2022-11-17 14:40:33 +00002204 ans = log10(x);
drh63f8f982020-12-07 21:13:06 +00002205 break;
2206 case 2:
drh3c1572d2022-11-17 14:40:33 +00002207 ans = log2(x);
drh63f8f982020-12-07 21:13:06 +00002208 break;
2209 default:
drh3c1572d2022-11-17 14:40:33 +00002210 ans = log(x);
drh63f8f982020-12-07 21:13:06 +00002211 break;
2212 }
drhf6e904b2020-12-07 17:15:32 +00002213 }
2214 sqlite3_result_double(context, ans);
2215}
drh63f8f982020-12-07 21:13:06 +00002216
2217/*
2218** Functions to converts degrees to radians and radians to degrees.
2219*/
2220static double degToRad(double x){ return x*(M_PI/180.0); }
2221static double radToDeg(double x){ return x*(180.0/M_PI); }
2222
2223/*
2224** Implementation of 1-argument SQL math functions:
2225**
2226** exp(X) - Compute e to the X-th power
2227*/
2228static void math1Func(
2229 sqlite3_context *context,
2230 int argc,
2231 sqlite3_value **argv
2232){
drh63f8f982020-12-07 21:13:06 +00002233 int type0;
2234 double v0, ans;
2235 double (*x)(double);
mistachkind97a4c02020-12-09 23:35:51 +00002236 assert( argc==1 );
drh63f8f982020-12-07 21:13:06 +00002237 type0 = sqlite3_value_numeric_type(argv[0]);
2238 if( type0!=SQLITE_INTEGER && type0!=SQLITE_FLOAT ) return;
2239 v0 = sqlite3_value_double(argv[0]);
2240 x = (double(*)(double))sqlite3_user_data(context);
2241 ans = x(v0);
2242 sqlite3_result_double(context, ans);
2243}
2244
2245/*
2246** Implementation of 2-argument SQL math functions:
2247**
2248** power(X,Y) - Compute X to the Y-th power
2249*/
2250static void math2Func(
2251 sqlite3_context *context,
2252 int argc,
2253 sqlite3_value **argv
2254){
drh63f8f982020-12-07 21:13:06 +00002255 int type0, type1;
2256 double v0, v1, ans;
2257 double (*x)(double,double);
mistachkind97a4c02020-12-09 23:35:51 +00002258 assert( argc==2 );
drh63f8f982020-12-07 21:13:06 +00002259 type0 = sqlite3_value_numeric_type(argv[0]);
2260 if( type0!=SQLITE_INTEGER && type0!=SQLITE_FLOAT ) return;
2261 type1 = sqlite3_value_numeric_type(argv[1]);
2262 if( type1!=SQLITE_INTEGER && type1!=SQLITE_FLOAT ) return;
2263 v0 = sqlite3_value_double(argv[0]);
2264 v1 = sqlite3_value_double(argv[1]);
2265 x = (double(*)(double,double))sqlite3_user_data(context);
2266 ans = x(v0, v1);
2267 sqlite3_result_double(context, ans);
2268}
2269
2270/*
drh81e5a9a2021-04-16 11:05:19 +00002271** Implementation of 0-argument pi() function.
drh63f8f982020-12-07 21:13:06 +00002272*/
2273static void piFunc(
2274 sqlite3_context *context,
2275 int argc,
2276 sqlite3_value **argv
2277){
2278 assert( argc==0 );
drh3547e492022-12-23 14:49:24 +00002279 (void)argv;
drh63f8f982020-12-07 21:13:06 +00002280 sqlite3_result_double(context, M_PI);
2281}
2282
drhf6e904b2020-12-07 17:15:32 +00002283#endif /* SQLITE_ENABLE_MATH_FUNCTIONS */
2284
drh70a8ca32008-08-21 18:49:27 +00002285/*
drh63f8f982020-12-07 21:13:06 +00002286** Implementation of sign(X) function.
2287*/
2288static void signFunc(
2289 sqlite3_context *context,
2290 int argc,
2291 sqlite3_value **argv
2292){
drh63f8f982020-12-07 21:13:06 +00002293 int type0;
2294 double x;
drhe5baf5c2020-12-16 14:20:45 +00002295 UNUSED_PARAMETER(argc);
mistachkind97a4c02020-12-09 23:35:51 +00002296 assert( argc==1 );
drh63f8f982020-12-07 21:13:06 +00002297 type0 = sqlite3_value_numeric_type(argv[0]);
2298 if( type0!=SQLITE_INTEGER && type0!=SQLITE_FLOAT ) return;
2299 x = sqlite3_value_double(argv[0]);
2300 sqlite3_result_int(context, x<0.0 ? -1 : x>0.0 ? +1 : 0);
2301}
2302
2303/*
peter.d.reid60ec9142014-09-06 16:39:46 +00002304** All of the FuncDef structures in the aBuiltinFunc[] array above
drh777c5382008-08-21 20:21:34 +00002305** to the global function hash table. This occurs at start-time (as
2306** a consequence of calling sqlite3_initialize()).
2307**
2308** After this routine runs
drh70a8ca32008-08-21 18:49:27 +00002309*/
drh80738d92016-02-15 00:34:16 +00002310void sqlite3RegisterBuiltinFunctions(void){
danielk197793ce7412008-09-01 19:14:02 +00002311 /*
2312 ** The following array holds FuncDef structures for all of the functions
2313 ** defined in this file.
2314 **
2315 ** The array cannot be constant since changes are made to the
2316 ** FuncDef.pHash elements at start-time. The elements of this array
2317 ** are read-only after initialization is complete.
drh80738d92016-02-15 00:34:16 +00002318 **
2319 ** For peak efficiency, put the most frequently used function last.
danielk197793ce7412008-09-01 19:14:02 +00002320 */
drh80738d92016-02-15 00:34:16 +00002321 static FuncDef aBuiltinFunc[] = {
drh171c50e2020-01-01 15:43:30 +00002322/***** Functions only available with SQLITE_TESTCTRL_INTERNAL_FUNCTIONS *****/
drh3780f9a2021-09-17 13:07:15 +00002323#if !defined(SQLITE_UNTESTABLE)
drh171c50e2020-01-01 15:43:30 +00002324 TEST_FUNC(implies_nonnull_row, 2, INLINEFUNC_implies_nonnull_row, 0),
2325 TEST_FUNC(expr_compare, 2, INLINEFUNC_expr_compare, 0),
2326 TEST_FUNC(expr_implies_expr, 2, INLINEFUNC_expr_implies_expr, 0),
drh3780f9a2021-09-17 13:07:15 +00002327 TEST_FUNC(affinity, 1, INLINEFUNC_affinity, 0),
2328#endif /* !defined(SQLITE_UNTESTABLE) */
drh171c50e2020-01-01 15:43:30 +00002329/***** Regular functions *****/
drh80738d92016-02-15 00:34:16 +00002330#ifdef SQLITE_SOUNDEX
2331 FUNCTION(soundex, 1, 0, 0, soundexFunc ),
2332#endif
2333#ifndef SQLITE_OMIT_LOAD_EXTENSION
drh64de2a52019-12-31 18:39:23 +00002334 SFUNCTION(load_extension, 1, 0, 0, loadExt ),
2335 SFUNCTION(load_extension, 2, 0, 0, loadExt ),
drh80738d92016-02-15 00:34:16 +00002336#endif
2337#if SQLITE_USER_AUTHENTICATION
2338 FUNCTION(sqlite_crypt, 2, 0, 0, sqlite3CryptFunc ),
2339#endif
2340#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
2341 DFUNCTION(sqlite_compileoption_used,1, 0, 0, compileoptionusedFunc ),
2342 DFUNCTION(sqlite_compileoption_get, 1, 0, 0, compileoptiongetFunc ),
2343#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
drh25c42962020-01-01 13:55:08 +00002344 INLINE_FUNC(unlikely, 1, INLINEFUNC_unlikely, SQLITE_FUNC_UNLIKELY),
2345 INLINE_FUNC(likelihood, 2, INLINEFUNC_unlikely, SQLITE_FUNC_UNLIKELY),
2346 INLINE_FUNC(likely, 1, INLINEFUNC_unlikely, SQLITE_FUNC_UNLIKELY),
drh092457b2017-12-29 15:04:49 +00002347#ifdef SQLITE_ENABLE_OFFSET_SQL_FUNC
drh645682a2022-06-01 11:05:59 +00002348 INLINE_FUNC(sqlite_offset, 1, INLINEFUNC_sqlite_offset, 0 ),
drh092457b2017-12-29 15:04:49 +00002349#endif
danielk197793ce7412008-09-01 19:14:02 +00002350 FUNCTION(ltrim, 1, 1, 0, trimFunc ),
2351 FUNCTION(ltrim, 2, 1, 0, trimFunc ),
2352 FUNCTION(rtrim, 1, 2, 0, trimFunc ),
2353 FUNCTION(rtrim, 2, 2, 0, trimFunc ),
2354 FUNCTION(trim, 1, 3, 0, trimFunc ),
2355 FUNCTION(trim, 2, 3, 0, trimFunc ),
2356 FUNCTION(min, -1, 0, 1, minmaxFunc ),
2357 FUNCTION(min, 0, 0, 1, 0 ),
dan6fb2b542018-06-19 17:13:11 +00002358 WAGGREGATE(min, 1, 0, 1, minmaxStep, minMaxFinalize, minMaxValue, 0,
drhbb301232021-07-15 19:29:43 +00002359 SQLITE_FUNC_MINMAX|SQLITE_FUNC_ANYORDER ),
danielk197793ce7412008-09-01 19:14:02 +00002360 FUNCTION(max, -1, 1, 1, minmaxFunc ),
2361 FUNCTION(max, 0, 1, 1, 0 ),
dan6fb2b542018-06-19 17:13:11 +00002362 WAGGREGATE(max, 1, 1, 1, minmaxStep, minMaxFinalize, minMaxValue, 0,
drhbb301232021-07-15 19:29:43 +00002363 SQLITE_FUNC_MINMAX|SQLITE_FUNC_ANYORDER ),
drha748fdc2012-03-28 01:34:47 +00002364 FUNCTION2(typeof, 1, 0, 0, typeofFunc, SQLITE_FUNC_TYPEOF),
drh9d44f182022-01-09 16:54:02 +00002365 FUNCTION2(subtype, 1, 0, 0, subtypeFunc, SQLITE_FUNC_TYPEOF),
drha748fdc2012-03-28 01:34:47 +00002366 FUNCTION2(length, 1, 0, 0, lengthFunc, SQLITE_FUNC_LENGTH),
drhd55e0722012-10-25 03:07:29 +00002367 FUNCTION(instr, 2, 0, 0, instrFunc ),
drha5c14162013-12-17 15:03:06 +00002368 FUNCTION(printf, -1, 0, 0, printfFunc ),
drh6bcd5852022-01-08 21:00:38 +00002369 FUNCTION(format, -1, 0, 0, printfFunc ),
drhd495d8c2013-02-22 19:34:25 +00002370 FUNCTION(unicode, 1, 0, 0, unicodeFunc ),
2371 FUNCTION(char, -1, 0, 0, charFunc ),
danielk197793ce7412008-09-01 19:14:02 +00002372 FUNCTION(abs, 1, 0, 0, absFunc ),
shanefbd60f82009-02-04 03:59:25 +00002373#ifndef SQLITE_OMIT_FLOATING_POINT
danielk197793ce7412008-09-01 19:14:02 +00002374 FUNCTION(round, 1, 0, 0, roundFunc ),
2375 FUNCTION(round, 2, 0, 0, roundFunc ),
shanefbd60f82009-02-04 03:59:25 +00002376#endif
danielk197793ce7412008-09-01 19:14:02 +00002377 FUNCTION(upper, 1, 0, 0, upperFunc ),
2378 FUNCTION(lower, 1, 0, 0, lowerFunc ),
danielk197793ce7412008-09-01 19:14:02 +00002379 FUNCTION(hex, 1, 0, 0, hexFunc ),
dane3c11d52023-01-23 14:11:34 +00002380 FUNCTION(unhex, 1, 0, 0, unhexFunc ),
dana50db432023-01-24 17:19:47 +00002381 FUNCTION(unhex, 2, 0, 0, unhexFunc ),
drhffe421c2020-05-13 17:26:38 +00002382 INLINE_FUNC(ifnull, 2, INLINEFUNC_coalesce, 0 ),
drhb1fba282013-11-21 14:33:48 +00002383 VFUNCTION(random, 0, 0, 0, randomFunc ),
2384 VFUNCTION(randomblob, 1, 0, 0, randomBlob ),
danielk197793ce7412008-09-01 19:14:02 +00002385 FUNCTION(nullif, 2, 0, 1, nullifFunc ),
drh03bf26d2015-08-31 21:16:36 +00002386 DFUNCTION(sqlite_version, 0, 0, 0, versionFunc ),
2387 DFUNCTION(sqlite_source_id, 0, 0, 0, sourceidFunc ),
drh840561f2011-04-27 18:08:42 +00002388 FUNCTION(sqlite_log, 2, 0, 0, errlogFunc ),
danielk197793ce7412008-09-01 19:14:02 +00002389 FUNCTION(quote, 1, 0, 0, quoteFunc ),
drhb1fba282013-11-21 14:33:48 +00002390 VFUNCTION(last_insert_rowid, 0, 0, 0, last_insert_rowid),
2391 VFUNCTION(changes, 0, 0, 0, changes ),
2392 VFUNCTION(total_changes, 0, 0, 0, total_changes ),
danielk197793ce7412008-09-01 19:14:02 +00002393 FUNCTION(replace, 3, 0, 0, replaceFunc ),
2394 FUNCTION(zeroblob, 1, 0, 0, zeroblobFunc ),
drh80738d92016-02-15 00:34:16 +00002395 FUNCTION(substr, 2, 0, 0, substrFunc ),
2396 FUNCTION(substr, 3, 0, 0, substrFunc ),
drh1335ec72020-11-23 14:50:43 +00002397 FUNCTION(substring, 2, 0, 0, substrFunc ),
2398 FUNCTION(substring, 3, 0, 0, substrFunc ),
dan6fb2b542018-06-19 17:13:11 +00002399 WAGGREGATE(sum, 1,0,0, sumStep, sumFinalize, sumFinalize, sumInverse, 0),
2400 WAGGREGATE(total, 1,0,0, sumStep,totalFinalize,totalFinalize,sumInverse, 0),
2401 WAGGREGATE(avg, 1,0,0, sumStep, avgFinalize, avgFinalize, sumInverse, 0),
dan7262ca92018-07-02 12:07:32 +00002402 WAGGREGATE(count, 0,0,0, countStep,
drhbb301232021-07-15 19:29:43 +00002403 countFinalize, countFinalize, countInverse,
2404 SQLITE_FUNC_COUNT|SQLITE_FUNC_ANYORDER ),
dan7262ca92018-07-02 12:07:32 +00002405 WAGGREGATE(count, 1,0,0, countStep,
drhbb301232021-07-15 19:29:43 +00002406 countFinalize, countFinalize, countInverse, SQLITE_FUNC_ANYORDER ),
dan03854d22018-06-08 11:45:28 +00002407 WAGGREGATE(group_concat, 1, 0, 0, groupConcatStep,
dan6fb2b542018-06-19 17:13:11 +00002408 groupConcatFinalize, groupConcatValue, groupConcatInverse, 0),
dan03854d22018-06-08 11:45:28 +00002409 WAGGREGATE(group_concat, 2, 0, 0, groupConcatStep,
dan6fb2b542018-06-19 17:13:11 +00002410 groupConcatFinalize, groupConcatValue, groupConcatInverse, 0),
danielk197793ce7412008-09-01 19:14:02 +00002411
2412 LIKEFUNC(glob, 2, &globInfo, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
drhcc153132016-08-04 12:35:17 +00002413#ifdef SQLITE_CASE_SENSITIVE_LIKE
danielk197793ce7412008-09-01 19:14:02 +00002414 LIKEFUNC(like, 2, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
2415 LIKEFUNC(like, 3, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
drhcc153132016-08-04 12:35:17 +00002416#else
danielk197793ce7412008-09-01 19:14:02 +00002417 LIKEFUNC(like, 2, &likeInfoNorm, SQLITE_FUNC_LIKE),
2418 LIKEFUNC(like, 3, &likeInfoNorm, SQLITE_FUNC_LIKE),
drhcc153132016-08-04 12:35:17 +00002419#endif
2420#ifdef SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION
2421 FUNCTION(unknown, -1, 0, 0, unknownFunc ),
2422#endif
drh80738d92016-02-15 00:34:16 +00002423 FUNCTION(coalesce, 1, 0, 0, 0 ),
2424 FUNCTION(coalesce, 0, 0, 0, 0 ),
drhf6e904b2020-12-07 17:15:32 +00002425#ifdef SQLITE_ENABLE_MATH_FUNCTIONS
drh4fd4a7a2021-01-01 01:44:06 +00002426 MFUNCTION(ceil, 1, xCeil, ceilingFunc ),
2427 MFUNCTION(ceiling, 1, xCeil, ceilingFunc ),
2428 MFUNCTION(floor, 1, xFloor, ceilingFunc ),
mistachkind97a4c02020-12-09 23:35:51 +00002429#if SQLITE_HAVE_C99_MATH_FUNCS
drh63f8f982020-12-07 21:13:06 +00002430 MFUNCTION(trunc, 1, trunc, ceilingFunc ),
mistachkind97a4c02020-12-09 23:35:51 +00002431#endif
drhf6e904b2020-12-07 17:15:32 +00002432 FUNCTION(ln, 1, 0, 0, logFunc ),
2433 FUNCTION(log, 1, 1, 0, logFunc ),
2434 FUNCTION(log10, 1, 1, 0, logFunc ),
drh63f8f982020-12-07 21:13:06 +00002435 FUNCTION(log2, 1, 2, 0, logFunc ),
drhf6e904b2020-12-07 17:15:32 +00002436 FUNCTION(log, 2, 0, 0, logFunc ),
drh63f8f982020-12-07 21:13:06 +00002437 MFUNCTION(exp, 1, exp, math1Func ),
2438 MFUNCTION(pow, 2, pow, math2Func ),
2439 MFUNCTION(power, 2, pow, math2Func ),
2440 MFUNCTION(mod, 2, fmod, math2Func ),
2441 MFUNCTION(acos, 1, acos, math1Func ),
2442 MFUNCTION(asin, 1, asin, math1Func ),
2443 MFUNCTION(atan, 1, atan, math1Func ),
2444 MFUNCTION(atan2, 2, atan2, math2Func ),
2445 MFUNCTION(cos, 1, cos, math1Func ),
2446 MFUNCTION(sin, 1, sin, math1Func ),
2447 MFUNCTION(tan, 1, tan, math1Func ),
2448 MFUNCTION(cosh, 1, cosh, math1Func ),
2449 MFUNCTION(sinh, 1, sinh, math1Func ),
2450 MFUNCTION(tanh, 1, tanh, math1Func ),
mistachkind97a4c02020-12-09 23:35:51 +00002451#if SQLITE_HAVE_C99_MATH_FUNCS
drh63f8f982020-12-07 21:13:06 +00002452 MFUNCTION(acosh, 1, acosh, math1Func ),
2453 MFUNCTION(asinh, 1, asinh, math1Func ),
2454 MFUNCTION(atanh, 1, atanh, math1Func ),
mistachkind97a4c02020-12-09 23:35:51 +00002455#endif
drh63f8f982020-12-07 21:13:06 +00002456 MFUNCTION(sqrt, 1, sqrt, math1Func ),
2457 MFUNCTION(radians, 1, degToRad, math1Func ),
2458 MFUNCTION(degrees, 1, radToDeg, math1Func ),
2459 FUNCTION(pi, 0, 0, 0, piFunc ),
drhf6e904b2020-12-07 17:15:32 +00002460#endif /* SQLITE_ENABLE_MATH_FUNCTIONS */
drh63f8f982020-12-07 21:13:06 +00002461 FUNCTION(sign, 1, 0, 0, signFunc ),
2462 INLINE_FUNC(coalesce, -1, INLINEFUNC_coalesce, 0 ),
2463 INLINE_FUNC(iif, 3, INLINEFUNC_iif, 0 ),
danielk197793ce7412008-09-01 19:14:02 +00002464 };
drh545f5872010-04-24 14:02:59 +00002465#ifndef SQLITE_OMIT_ALTERTABLE
2466 sqlite3AlterFunctions();
2467#endif
dandfa552f2018-06-02 21:04:28 +00002468 sqlite3WindowFunctions();
drh80738d92016-02-15 00:34:16 +00002469 sqlite3RegisterDateTimeFunctions();
drh9dbf96b2022-01-06 01:40:09 +00002470 sqlite3RegisterJsonFunctions();
drh80738d92016-02-15 00:34:16 +00002471 sqlite3InsertBuiltinFuncs(aBuiltinFunc, ArraySize(aBuiltinFunc));
2472
2473#if 0 /* Enable to print out how the built-in functions are hashed */
2474 {
2475 int i;
2476 FuncDef *p;
2477 for(i=0; i<SQLITE_FUNC_HASH_SZ; i++){
2478 printf("FUNC-HASH %02d:", i);
2479 for(p=sqlite3BuiltinFunctions.a[i]; p; p=p->u.pHash){
2480 int n = sqlite3Strlen30(p->zName);
2481 int h = p->zName[0] + n;
drhf9751072021-10-07 13:40:29 +00002482 assert( p->funcFlags & SQLITE_FUNC_BUILTIN );
drh80738d92016-02-15 00:34:16 +00002483 printf(" %s(%d)", p->zName, h);
2484 }
2485 printf("\n");
2486 }
2487 }
2488#endif
drh70a8ca32008-08-21 18:49:27 +00002489}