blob: 48b5f5f1984724c24520367eb8f38e0d17d61d02 [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
shearer3a8847a2020-09-25 14:26:49 +000013** functions of SQLite. (Some function, and in particular the date and
drhede7ae32014-08-06 11:58:40 +000014** 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
drh5708d2d2005-06-22 10:53:59 +0000100
101/*
drh0bce8352002-02-28 00:41:10 +0000102** Implementation of the length() function
103*/
drhf9b596e2004-05-26 16:54:42 +0000104static void lengthFunc(
105 sqlite3_context *context,
106 int argc,
107 sqlite3_value **argv
108){
drh0bce8352002-02-28 00:41:10 +0000109 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000110 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +0000111 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +0000112 case SQLITE_BLOB:
113 case SQLITE_INTEGER:
114 case SQLITE_FLOAT: {
drhf4479502004-05-27 03:12:53 +0000115 sqlite3_result_int(context, sqlite3_value_bytes(argv[0]));
drhf9b596e2004-05-26 16:54:42 +0000116 break;
117 }
drh9c054832004-05-31 18:51:57 +0000118 case SQLITE_TEXT: {
drh2646da72005-12-09 20:02:05 +0000119 const unsigned char *z = sqlite3_value_text(argv[0]);
drh7ea34692018-01-23 04:22:33 +0000120 const unsigned char *z0;
121 unsigned char c;
drh7a521cf2007-04-25 18:23:52 +0000122 if( z==0 ) return;
drh7ea34692018-01-23 04:22:33 +0000123 z0 = z;
124 while( (c = *z)!=0 ){
125 z++;
126 if( c>=0xc0 ){
127 while( (*z & 0xc0)==0x80 ){ z++; z0++; }
128 }
drh4a919112007-05-15 11:55:09 +0000129 }
drh7ea34692018-01-23 04:22:33 +0000130 sqlite3_result_int(context, (int)(z-z0));
drhf9b596e2004-05-26 16:54:42 +0000131 break;
132 }
133 default: {
134 sqlite3_result_null(context);
135 break;
136 }
137 }
drh0bce8352002-02-28 00:41:10 +0000138}
139
140/*
drh2ba3ccc2009-12-08 02:06:08 +0000141** Implementation of the abs() function.
142**
143** IMP: R-23979-26855 The abs(X) function returns the absolute value of
144** the numeric argument X.
drh0bce8352002-02-28 00:41:10 +0000145*/
danielk19770ae8b832004-05-25 12:05:56 +0000146static void absFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drh0bce8352002-02-28 00:41:10 +0000147 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000148 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +0000149 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +0000150 case SQLITE_INTEGER: {
danielk1977f93bbbe2004-05-27 10:30:52 +0000151 i64 iVal = sqlite3_value_int64(argv[0]);
drh52fc8492006-02-23 21:43:55 +0000152 if( iVal<0 ){
drh693e6712014-01-24 22:58:00 +0000153 if( iVal==SMALLEST_INT64 ){
drheb091cd2013-11-09 19:47:15 +0000154 /* IMP: R-31676-45509 If X is the integer -9223372036854775808
155 ** then abs(X) throws an integer overflow error since there is no
drh2ba3ccc2009-12-08 02:06:08 +0000156 ** equivalent positive 64-bit two complement value. */
drh52fc8492006-02-23 21:43:55 +0000157 sqlite3_result_error(context, "integer overflow", -1);
158 return;
159 }
160 iVal = -iVal;
161 }
danielk1977f93bbbe2004-05-27 10:30:52 +0000162 sqlite3_result_int64(context, iVal);
drhf9b596e2004-05-26 16:54:42 +0000163 break;
164 }
drh9c054832004-05-31 18:51:57 +0000165 case SQLITE_NULL: {
drh2ba3ccc2009-12-08 02:06:08 +0000166 /* IMP: R-37434-19929 Abs(X) returns NULL if X is NULL. */
drhf9b596e2004-05-26 16:54:42 +0000167 sqlite3_result_null(context);
168 break;
169 }
170 default: {
drh2ba3ccc2009-12-08 02:06:08 +0000171 /* Because sqlite3_value_double() returns 0.0 if the argument is not
172 ** something that can be converted into a number, we have:
drh643091f2014-11-20 23:21:23 +0000173 ** IMP: R-01992-00519 Abs(X) returns 0.0 if X is a string or blob
174 ** that cannot be converted to a numeric value.
drh2ba3ccc2009-12-08 02:06:08 +0000175 */
danielk1977f93bbbe2004-05-27 10:30:52 +0000176 double rVal = sqlite3_value_double(argv[0]);
drh52fc8492006-02-23 21:43:55 +0000177 if( rVal<0 ) rVal = -rVal;
danielk1977f93bbbe2004-05-27 10:30:52 +0000178 sqlite3_result_double(context, rVal);
drhf9b596e2004-05-26 16:54:42 +0000179 break;
180 }
181 }
drh0bce8352002-02-28 00:41:10 +0000182}
183
184/*
drhd55e0722012-10-25 03:07:29 +0000185** Implementation of the instr() function.
186**
187** instr(haystack,needle) finds the first occurrence of needle
188** in haystack and returns the number of previous characters plus 1,
189** or 0 if needle does not occur within haystack.
190**
191** If both haystack and needle are BLOBs, then the result is one more than
192** the number of bytes in haystack prior to the first occurrence of needle,
193** or 0 if needle never occurs in haystack.
194*/
195static void instrFunc(
196 sqlite3_context *context,
197 int argc,
198 sqlite3_value **argv
199){
200 const unsigned char *zHaystack;
201 const unsigned char *zNeedle;
202 int nHaystack;
203 int nNeedle;
204 int typeHaystack, typeNeedle;
205 int N = 1;
206 int isText;
drhc930b402019-01-08 15:18:24 +0000207 unsigned char firstChar;
drh97b02502019-09-17 03:16:29 +0000208 sqlite3_value *pC1 = 0;
209 sqlite3_value *pC2 = 0;
drhd55e0722012-10-25 03:07:29 +0000210
drh68c804b2012-12-04 11:03:11 +0000211 UNUSED_PARAMETER(argc);
drhd55e0722012-10-25 03:07:29 +0000212 typeHaystack = sqlite3_value_type(argv[0]);
213 typeNeedle = sqlite3_value_type(argv[1]);
214 if( typeHaystack==SQLITE_NULL || typeNeedle==SQLITE_NULL ) return;
215 nHaystack = sqlite3_value_bytes(argv[0]);
216 nNeedle = sqlite3_value_bytes(argv[1]);
dan895decf2016-12-30 14:15:56 +0000217 if( nNeedle>0 ){
218 if( typeHaystack==SQLITE_BLOB && typeNeedle==SQLITE_BLOB ){
219 zHaystack = sqlite3_value_blob(argv[0]);
220 zNeedle = sqlite3_value_blob(argv[1]);
221 isText = 0;
drh97b02502019-09-17 03:16:29 +0000222 }else if( typeHaystack!=SQLITE_BLOB && typeNeedle!=SQLITE_BLOB ){
dan895decf2016-12-30 14:15:56 +0000223 zHaystack = sqlite3_value_text(argv[0]);
224 zNeedle = sqlite3_value_text(argv[1]);
225 isText = 1;
drh97b02502019-09-17 03:16:29 +0000226 }else{
227 pC1 = sqlite3_value_dup(argv[0]);
228 zHaystack = sqlite3_value_text(pC1);
drh9d702842019-09-18 11:16:46 +0000229 if( zHaystack==0 ) goto endInstrOOM;
230 nHaystack = sqlite3_value_bytes(pC1);
drh97b02502019-09-17 03:16:29 +0000231 pC2 = sqlite3_value_dup(argv[1]);
232 zNeedle = sqlite3_value_text(pC2);
drh9d702842019-09-18 11:16:46 +0000233 if( zNeedle==0 ) goto endInstrOOM;
234 nNeedle = sqlite3_value_bytes(pC2);
drh97b02502019-09-17 03:16:29 +0000235 isText = 1;
dan895decf2016-12-30 14:15:56 +0000236 }
drh9d702842019-09-18 11:16:46 +0000237 if( zNeedle==0 || (nHaystack && zHaystack==0) ) goto endInstrOOM;
drhc930b402019-01-08 15:18:24 +0000238 firstChar = zNeedle[0];
239 while( nNeedle<=nHaystack
240 && (zHaystack[0]!=firstChar || memcmp(zHaystack, zNeedle, nNeedle)!=0)
241 ){
dan895decf2016-12-30 14:15:56 +0000242 N++;
243 do{
244 nHaystack--;
245 zHaystack++;
246 }while( isText && (zHaystack[0]&0xc0)==0x80 );
247 }
248 if( nNeedle>nHaystack ) N = 0;
drhd55e0722012-10-25 03:07:29 +0000249 }
drhd55e0722012-10-25 03:07:29 +0000250 sqlite3_result_int(context, N);
drh97b02502019-09-17 03:16:29 +0000251endInstr:
252 sqlite3_value_free(pC1);
253 sqlite3_value_free(pC2);
drh9d702842019-09-18 11:16:46 +0000254 return;
255endInstrOOM:
256 sqlite3_result_error_nomem(context);
257 goto endInstr;
drhd55e0722012-10-25 03:07:29 +0000258}
259
shearer3a8847a2020-09-25 14:26:49 +0000260/*
261** Implementation of the printf() function.
262*/
263static void printfFunc(
264 sqlite3_context *context,
265 int argc,
266 sqlite3_value **argv
267){
268 PrintfArguments x;
269 StrAccum str;
270 const char *zFormat;
271 int n;
272 sqlite3 *db = sqlite3_context_db_handle(context);
273
274 if( argc>=1 && (zFormat = (const char*)sqlite3_value_text(argv[0]))!=0 ){
275 x.nArg = argc-1;
276 x.nUsed = 0;
277 x.apArg = argv+1;
278 sqlite3StrAccumInit(&str, db, 0, 0, db->aLimit[SQLITE_LIMIT_LENGTH]);
279 str.printfFlags = SQLITE_PRINTF_SQLFUNC;
280 sqlite3_str_appendf(&str, zFormat, &x);
281 n = str.nChar;
282 sqlite3_result_text(context, sqlite3StrAccumFinish(&str), n,
283 SQLITE_DYNAMIC);
284 }
285}
drha5c14162013-12-17 15:03:06 +0000286
287/*
drhf764e6f2007-05-15 01:13:47 +0000288** Implementation of the substr() function.
289**
290** substr(x,p1,p2) returns p2 characters of x[] beginning with p1.
291** p1 is 1-indexed. So substr(x,1,1) returns the first character
292** of x. If x is text, then we actually count UTF-8 characters.
293** If x is a blob, then we count bytes.
294**
295** If p1 is negative, then we begin abs(p1) from the end of x[].
shaneh779b8f12009-11-12 05:04:50 +0000296**
drhf7b54962013-05-28 12:11:54 +0000297** If p2 is negative, return the p2 characters preceding p1.
drh0bce8352002-02-28 00:41:10 +0000298*/
drhf9b596e2004-05-26 16:54:42 +0000299static void substrFunc(
300 sqlite3_context *context,
301 int argc,
302 sqlite3_value **argv
303){
drh2646da72005-12-09 20:02:05 +0000304 const unsigned char *z;
305 const unsigned char *z2;
drh023ae032007-05-08 12:12:16 +0000306 int len;
drhf764e6f2007-05-15 01:13:47 +0000307 int p0type;
drh023ae032007-05-08 12:12:16 +0000308 i64 p1, p2;
drh65595cd2009-02-02 16:32:55 +0000309 int negP2 = 0;
drhf9b596e2004-05-26 16:54:42 +0000310
drh64f31512007-10-12 19:11:55 +0000311 assert( argc==3 || argc==2 );
drh8198d252009-02-01 19:42:37 +0000312 if( sqlite3_value_type(argv[1])==SQLITE_NULL
313 || (argc==3 && sqlite3_value_type(argv[2])==SQLITE_NULL)
314 ){
315 return;
316 }
drhf764e6f2007-05-15 01:13:47 +0000317 p0type = sqlite3_value_type(argv[0]);
drh4adc4cb2009-11-11 20:53:31 +0000318 p1 = sqlite3_value_int(argv[1]);
drhf764e6f2007-05-15 01:13:47 +0000319 if( p0type==SQLITE_BLOB ){
320 len = sqlite3_value_bytes(argv[0]);
321 z = sqlite3_value_blob(argv[0]);
322 if( z==0 ) return;
drh1f0feef2007-05-15 13:27:07 +0000323 assert( len==sqlite3_value_bytes(argv[0]) );
drhf764e6f2007-05-15 01:13:47 +0000324 }else{
325 z = sqlite3_value_text(argv[0]);
326 if( z==0 ) return;
drh4a919112007-05-15 11:55:09 +0000327 len = 0;
drh4adc4cb2009-11-11 20:53:31 +0000328 if( p1<0 ){
329 for(z2=z; *z2; len++){
330 SQLITE_SKIP_UTF8(z2);
331 }
drh4a919112007-05-15 11:55:09 +0000332 }
drhf764e6f2007-05-15 01:13:47 +0000333 }
drh883ad042015-02-19 00:29:11 +0000334#ifdef SQLITE_SUBSTR_COMPATIBILITY
335 /* If SUBSTR_COMPATIBILITY is defined then substr(X,0,N) work the same as
336 ** as substr(X,1,N) - it returns the first N characters of X. This
337 ** is essentially a back-out of the bug-fix in check-in [5fc125d362df4b8]
338 ** from 2009-02-02 for compatibility of applications that exploited the
339 ** old buggy behavior. */
340 if( p1==0 ) p1 = 1; /* <rdar://problem/6778339> */
341#endif
drh64f31512007-10-12 19:11:55 +0000342 if( argc==3 ){
343 p2 = sqlite3_value_int(argv[2]);
drh65595cd2009-02-02 16:32:55 +0000344 if( p2<0 ){
345 p2 = -p2;
346 negP2 = 1;
347 }
drh64f31512007-10-12 19:11:55 +0000348 }else{
drhbb4957f2008-03-20 14:03:29 +0000349 p2 = sqlite3_context_db_handle(context)->aLimit[SQLITE_LIMIT_LENGTH];
drh64f31512007-10-12 19:11:55 +0000350 }
drh0bce8352002-02-28 00:41:10 +0000351 if( p1<0 ){
drh89425d52002-02-28 03:04:48 +0000352 p1 += len;
drh653bc752002-02-28 03:31:10 +0000353 if( p1<0 ){
354 p2 += p1;
drh65595cd2009-02-02 16:32:55 +0000355 if( p2<0 ) p2 = 0;
drh653bc752002-02-28 03:31:10 +0000356 p1 = 0;
357 }
drh0bce8352002-02-28 00:41:10 +0000358 }else if( p1>0 ){
359 p1--;
drh65595cd2009-02-02 16:32:55 +0000360 }else if( p2>0 ){
361 p2--;
drh0bce8352002-02-28 00:41:10 +0000362 }
drh65595cd2009-02-02 16:32:55 +0000363 if( negP2 ){
364 p1 -= p2;
drh4e79c592009-02-01 19:23:32 +0000365 if( p1<0 ){
366 p2 += p1;
367 p1 = 0;
368 }
369 }
drh65595cd2009-02-02 16:32:55 +0000370 assert( p1>=0 && p2>=0 );
drhf764e6f2007-05-15 01:13:47 +0000371 if( p0type!=SQLITE_BLOB ){
drh4a919112007-05-15 11:55:09 +0000372 while( *z && p1 ){
373 SQLITE_SKIP_UTF8(z);
374 p1--;
drhf764e6f2007-05-15 01:13:47 +0000375 }
drh4a919112007-05-15 11:55:09 +0000376 for(z2=z; *z2 && p2; p2--){
377 SQLITE_SKIP_UTF8(z2);
drhf764e6f2007-05-15 01:13:47 +0000378 }
drhbbf483f2014-09-09 20:30:24 +0000379 sqlite3_result_text64(context, (char*)z, z2-z, SQLITE_TRANSIENT,
380 SQLITE_UTF8);
drhf764e6f2007-05-15 01:13:47 +0000381 }else{
drh4adc4cb2009-11-11 20:53:31 +0000382 if( p1+p2>len ){
383 p2 = len-p1;
384 if( p2<0 ) p2 = 0;
385 }
drhbbf483f2014-09-09 20:30:24 +0000386 sqlite3_result_blob64(context, (char*)&z[p1], (u64)p2, SQLITE_TRANSIENT);
drh0bce8352002-02-28 00:41:10 +0000387 }
drh0bce8352002-02-28 00:41:10 +0000388}
389
390/*
391** Implementation of the round() function
392*/
shanefbd60f82009-02-04 03:59:25 +0000393#ifndef SQLITE_OMIT_FLOATING_POINT
danielk19770ae8b832004-05-25 12:05:56 +0000394static void roundFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
danielk197751ad0ec2004-05-24 12:39:02 +0000395 int n = 0;
drh0bce8352002-02-28 00:41:10 +0000396 double r;
drh50d654d2009-06-03 01:24:54 +0000397 char *zBuf;
drh0bce8352002-02-28 00:41:10 +0000398 assert( argc==1 || argc==2 );
danielk197751ad0ec2004-05-24 12:39:02 +0000399 if( argc==2 ){
drh9c054832004-05-31 18:51:57 +0000400 if( SQLITE_NULL==sqlite3_value_type(argv[1]) ) return;
danielk197751ad0ec2004-05-24 12:39:02 +0000401 n = sqlite3_value_int(argv[1]);
402 if( n>30 ) n = 30;
403 if( n<0 ) n = 0;
404 }
drhd589a922006-03-02 03:02:48 +0000405 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drh4f26d6c2004-05-26 23:25:30 +0000406 r = sqlite3_value_double(argv[0]);
shaneh147e1762010-02-17 04:19:27 +0000407 /* If Y==0 and X will fit in a 64-bit int,
408 ** handle the rounding directly,
409 ** otherwise use printf.
410 */
drh84422db2019-05-30 13:47:10 +0000411 if( r<-4503599627370496.0 || r>+4503599627370496.0 ){
412 /* The value has no fractional part so there is nothing to round */
413 }else if( n==0 ){
414 r = (double)((sqlite_int64)(r+(r<0?-0.5:+0.5)));
drh50d654d2009-06-03 01:24:54 +0000415 }else{
shaneh147e1762010-02-17 04:19:27 +0000416 zBuf = sqlite3_mprintf("%.*f",n,r);
417 if( zBuf==0 ){
418 sqlite3_result_error_nomem(context);
419 return;
420 }
drh55700bc2019-06-07 22:51:13 +0000421 sqlite3AtoF(zBuf, &r, sqlite3Strlen30(zBuf), SQLITE_UTF8);
drh50d654d2009-06-03 01:24:54 +0000422 sqlite3_free(zBuf);
drh50d654d2009-06-03 01:24:54 +0000423 }
shaneh147e1762010-02-17 04:19:27 +0000424 sqlite3_result_double(context, r);
drh0bce8352002-02-28 00:41:10 +0000425}
shanefbd60f82009-02-04 03:59:25 +0000426#endif
drhdc04c582002-02-24 01:55:15 +0000427
danielk197726783a52007-08-29 14:06:22 +0000428/*
drhf3cdcdc2015-04-29 16:50:28 +0000429** Allocate nByte bytes of space using sqlite3Malloc(). If the
danielk197726783a52007-08-29 14:06:22 +0000430** allocation fails, call sqlite3_result_error_nomem() to notify
drh27e62db2009-04-02 10:16:17 +0000431** the database handle that malloc() has failed and return NULL.
432** If nByte is larger than the maximum string or blob length, then
433** raise an SQLITE_TOOBIG exception and return NULL.
danielk197726783a52007-08-29 14:06:22 +0000434*/
drhb1a6c3c2008-03-20 16:30:17 +0000435static void *contextMalloc(sqlite3_context *context, i64 nByte){
drhbb4957f2008-03-20 14:03:29 +0000436 char *z;
drh27e62db2009-04-02 10:16:17 +0000437 sqlite3 *db = sqlite3_context_db_handle(context);
drhef31c6a2009-04-02 09:07:12 +0000438 assert( nByte>0 );
drh27e62db2009-04-02 10:16:17 +0000439 testcase( nByte==db->aLimit[SQLITE_LIMIT_LENGTH] );
440 testcase( nByte==db->aLimit[SQLITE_LIMIT_LENGTH]+1 );
441 if( nByte>db->aLimit[SQLITE_LIMIT_LENGTH] ){
drhbb4957f2008-03-20 14:03:29 +0000442 sqlite3_result_error_toobig(context);
443 z = 0;
444 }else{
drhda4ca9d2014-09-09 17:27:35 +0000445 z = sqlite3Malloc(nByte);
drhef31c6a2009-04-02 09:07:12 +0000446 if( !z ){
drhbb4957f2008-03-20 14:03:29 +0000447 sqlite3_result_error_nomem(context);
448 }
danielk1977a1644fd2007-08-29 12:31:25 +0000449 }
450 return z;
451}
452
drhdc04c582002-02-24 01:55:15 +0000453/*
454** Implementation of the upper() and lower() SQL functions.
455*/
danielk19770ae8b832004-05-25 12:05:56 +0000456static void upperFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drh7a521cf2007-04-25 18:23:52 +0000457 char *z1;
458 const char *z2;
drh9310ef22007-04-27 17:16:20 +0000459 int i, n;
drh1d34fde2009-02-03 15:50:33 +0000460 UNUSED_PARAMETER(argc);
drh7a521cf2007-04-25 18:23:52 +0000461 z2 = (char*)sqlite3_value_text(argv[0]);
drh1f0feef2007-05-15 13:27:07 +0000462 n = sqlite3_value_bytes(argv[0]);
463 /* Verify that the call to _bytes() does not invalidate the _text() pointer */
464 assert( z2==(char*)sqlite3_value_text(argv[0]) );
drh7a521cf2007-04-25 18:23:52 +0000465 if( z2 ){
drhb1a6c3c2008-03-20 16:30:17 +0000466 z1 = contextMalloc(context, ((i64)n)+1);
drh7a521cf2007-04-25 18:23:52 +0000467 if( z1 ){
drhdf901d32011-10-13 18:00:11 +0000468 for(i=0; i<n; i++){
469 z1[i] = (char)sqlite3Toupper(z2[i]);
drh7a521cf2007-04-25 18:23:52 +0000470 }
drhdf901d32011-10-13 18:00:11 +0000471 sqlite3_result_text(context, z1, n, sqlite3_free);
drh7a521cf2007-04-25 18:23:52 +0000472 }
drhdc04c582002-02-24 01:55:15 +0000473 }
474}
danielk19770ae8b832004-05-25 12:05:56 +0000475static void lowerFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drhdf901d32011-10-13 18:00:11 +0000476 char *z1;
drh7a521cf2007-04-25 18:23:52 +0000477 const char *z2;
drh9310ef22007-04-27 17:16:20 +0000478 int i, n;
drh1d34fde2009-02-03 15:50:33 +0000479 UNUSED_PARAMETER(argc);
drh7a521cf2007-04-25 18:23:52 +0000480 z2 = (char*)sqlite3_value_text(argv[0]);
drh1f0feef2007-05-15 13:27:07 +0000481 n = sqlite3_value_bytes(argv[0]);
482 /* Verify that the call to _bytes() does not invalidate the _text() pointer */
483 assert( z2==(char*)sqlite3_value_text(argv[0]) );
drh7a521cf2007-04-25 18:23:52 +0000484 if( z2 ){
drhb1a6c3c2008-03-20 16:30:17 +0000485 z1 = contextMalloc(context, ((i64)n)+1);
drh7a521cf2007-04-25 18:23:52 +0000486 if( z1 ){
drhdf901d32011-10-13 18:00:11 +0000487 for(i=0; i<n; i++){
488 z1[i] = sqlite3Tolower(z2[i]);
drh7a521cf2007-04-25 18:23:52 +0000489 }
drhdf901d32011-10-13 18:00:11 +0000490 sqlite3_result_text(context, z1, n, sqlite3_free);
drh7a521cf2007-04-25 18:23:52 +0000491 }
drhdc04c582002-02-24 01:55:15 +0000492 }
493}
494
drhae6bb952009-11-11 00:24:31 +0000495/*
drhcca9f3d2013-09-06 15:23:29 +0000496** Some functions like COALESCE() and IFNULL() and UNLIKELY() are implemented
497** as VDBE code so that unused argument values do not have to be computed.
498** However, we still need some kind of function implementation for this
499** routines in the function table. The noopFunc macro provides this.
500** noopFunc will never be called so it doesn't matter what the implementation
501** is. We might as well use the "version()" function as a substitute.
drhae6bb952009-11-11 00:24:31 +0000502*/
drhcca9f3d2013-09-06 15:23:29 +0000503#define noopFunc versionFunc /* Substitute function - never called */
drh3212e182002-02-28 00:46:26 +0000504
505/*
drhf9ffac92002-03-02 19:00:31 +0000506** Implementation of random(). Return a random integer.
507*/
drhf9b596e2004-05-26 16:54:42 +0000508static void randomFunc(
509 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000510 int NotUsed,
511 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000512){
drh52fc8492006-02-23 21:43:55 +0000513 sqlite_int64 r;
danielk197762c14b32008-11-19 09:05:26 +0000514 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drh2fa18682008-03-19 14:15:34 +0000515 sqlite3_randomness(sizeof(r), &r);
drh3034e3d2009-04-02 14:05:21 +0000516 if( r<0 ){
517 /* We need to prevent a random number of 0x8000000000000000
518 ** (or -9223372036854775808) since when you do abs() of that
519 ** number of you get the same value back again. To do this
520 ** in a way that is testable, mask the sign bit off of negative
521 ** values, resulting in a positive value. Then take the
522 ** 2s complement of that positive value. The end result can
523 ** therefore be no less than -9223372036854775807.
524 */
drhaf8001b2012-02-11 19:53:24 +0000525 r = -(r & LARGEST_INT64);
drh3034e3d2009-04-02 14:05:21 +0000526 }
drh52fc8492006-02-23 21:43:55 +0000527 sqlite3_result_int64(context, r);
drhf9ffac92002-03-02 19:00:31 +0000528}
529
530/*
drh137c7282007-01-29 17:58:28 +0000531** Implementation of randomblob(N). Return a random blob
532** that is N bytes long.
drh63cf66f2007-01-29 15:50:05 +0000533*/
drh137c7282007-01-29 17:58:28 +0000534static void randomBlob(
drh63cf66f2007-01-29 15:50:05 +0000535 sqlite3_context *context,
536 int argc,
537 sqlite3_value **argv
538){
drh3cb79202019-01-18 14:53:15 +0000539 sqlite3_int64 n;
drh137c7282007-01-29 17:58:28 +0000540 unsigned char *p;
drh63cf66f2007-01-29 15:50:05 +0000541 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000542 UNUSED_PARAMETER(argc);
drh3cb79202019-01-18 14:53:15 +0000543 n = sqlite3_value_int64(argv[0]);
drh023ae032007-05-08 12:12:16 +0000544 if( n<1 ){
545 n = 1;
546 }
danielk1977a1644fd2007-08-29 12:31:25 +0000547 p = contextMalloc(context, n);
drh02d85832007-05-07 19:31:15 +0000548 if( p ){
drh2fa18682008-03-19 14:15:34 +0000549 sqlite3_randomness(n, p);
drh17435752007-08-16 04:30:38 +0000550 sqlite3_result_blob(context, (char*)p, n, sqlite3_free);
drh02d85832007-05-07 19:31:15 +0000551 }
drh63cf66f2007-01-29 15:50:05 +0000552}
553
554/*
drh6ed41ad2002-04-06 14:10:47 +0000555** Implementation of the last_insert_rowid() SQL function. The return
danielk197724b03fd2004-05-10 10:34:34 +0000556** value is the same as the sqlite3_last_insert_rowid() API function.
drh6ed41ad2002-04-06 14:10:47 +0000557*/
danielk197751ad0ec2004-05-24 12:39:02 +0000558static void last_insert_rowid(
danielk19770ae8b832004-05-25 12:05:56 +0000559 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000560 int NotUsed,
561 sqlite3_value **NotUsed2
danielk197751ad0ec2004-05-24 12:39:02 +0000562){
drhfa4a4b92008-03-19 21:45:51 +0000563 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000564 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000565 /* IMP: R-51513-12026 The last_insert_rowid() SQL function is a
566 ** wrapper around the sqlite3_last_insert_rowid() C/C++ interface
567 ** function. */
drhf9b596e2004-05-26 16:54:42 +0000568 sqlite3_result_int64(context, sqlite3_last_insert_rowid(db));
drh6ed41ad2002-04-06 14:10:47 +0000569}
570
rdcf146a772004-02-25 22:51:06 +0000571/*
drhab2f1f92010-01-11 18:26:42 +0000572** Implementation of the changes() SQL function.
573**
574** IMP: R-62073-11209 The changes() SQL function is a wrapper
575** around the sqlite3_changes() C/C++ function and hence follows the same
576** rules for counting changes.
rdcf146a772004-02-25 22:51:06 +0000577*/
danielk1977b28af712004-06-21 06:50:26 +0000578static void changes(
drhf9b596e2004-05-26 16:54:42 +0000579 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000580 int NotUsed,
581 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000582){
drhfa4a4b92008-03-19 21:45:51 +0000583 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000584 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhf4479502004-05-27 03:12:53 +0000585 sqlite3_result_int(context, sqlite3_changes(db));
rdcb0c374f2004-02-20 22:53:38 +0000586}
rdcf146a772004-02-25 22:51:06 +0000587
588/*
danielk1977b28af712004-06-21 06:50:26 +0000589** Implementation of the total_changes() SQL function. The return value is
590** the same as the sqlite3_total_changes() API function.
rdcf146a772004-02-25 22:51:06 +0000591*/
danielk1977b28af712004-06-21 06:50:26 +0000592static void total_changes(
593 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000594 int NotUsed,
595 sqlite3_value **NotUsed2
danielk197751ad0ec2004-05-24 12:39:02 +0000596){
drhfa4a4b92008-03-19 21:45:51 +0000597 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000598 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000599 /* IMP: R-52756-41993 This function is a wrapper around the
600 ** sqlite3_total_changes() C/C++ interface. */
danielk1977b28af712004-06-21 06:50:26 +0000601 sqlite3_result_int(context, sqlite3_total_changes(db));
rdcb0c374f2004-02-20 22:53:38 +0000602}
603
drh6ed41ad2002-04-06 14:10:47 +0000604/*
drh4e5ffc52004-08-31 00:52:37 +0000605** A structure defining how to do GLOB-style comparisons.
danielk1977d02eb1f2004-06-06 09:44:03 +0000606*/
drh4e5ffc52004-08-31 00:52:37 +0000607struct compareInfo {
drh07e83472016-01-11 03:48:18 +0000608 u8 matchAll; /* "*" or "%" */
609 u8 matchOne; /* "?" or "_" */
610 u8 matchSet; /* "[" or 0 */
611 u8 noCase; /* true to ignore case differences */
danielk1977d02eb1f2004-06-06 09:44:03 +0000612};
drh55ef4d92005-08-14 01:20:37 +0000613
drhb9175ae2007-12-07 18:39:04 +0000614/*
615** For LIKE and GLOB matching on EBCDIC machines, assume that every
drhb0870482015-06-17 13:20:54 +0000616** character is exactly one byte in size. Also, provde the Utf8Read()
617** macro for fast reading of the next character in the common case where
618** the next character is ASCII.
drhb9175ae2007-12-07 18:39:04 +0000619*/
620#if defined(SQLITE_EBCDIC)
drh88b33222014-09-25 03:51:37 +0000621# define sqlite3Utf8Read(A) (*((*A)++))
drhb0870482015-06-17 13:20:54 +0000622# define Utf8Read(A) (*(A++))
drhb9175ae2007-12-07 18:39:04 +0000623#else
drhb0870482015-06-17 13:20:54 +0000624# define Utf8Read(A) (A[0]<0x80?*(A++):sqlite3Utf8Read(&A))
drhb9175ae2007-12-07 18:39:04 +0000625#endif
626
drh4e5ffc52004-08-31 00:52:37 +0000627static const struct compareInfo globInfo = { '*', '?', '[', 0 };
drh55ef4d92005-08-14 01:20:37 +0000628/* The correct SQL-92 behavior is for the LIKE operator to ignore
629** case. Thus 'a' LIKE 'A' would be true. */
630static const struct compareInfo likeInfoNorm = { '%', '_', 0, 1 };
631/* If SQLITE_CASE_SENSITIVE_LIKE is defined, then the LIKE operator
632** is case sensitive causing 'a' LIKE 'A' to be false */
633static const struct compareInfo likeInfoAlt = { '%', '_', 0, 0 };
danielk1977d02eb1f2004-06-06 09:44:03 +0000634
635/*
drh698a01c2016-12-01 18:49:40 +0000636** Possible error returns from patternMatch()
637*/
638#define SQLITE_MATCH 0
639#define SQLITE_NOMATCH 1
640#define SQLITE_NOWILDCARDMATCH 2
641
642/*
643** Compare two UTF-8 strings for equality where the first string is
644** a GLOB or LIKE expression. Return values:
645**
646** SQLITE_MATCH: Match
647** SQLITE_NOMATCH: No match
648** SQLITE_NOWILDCARDMATCH: No match in spite of having * or % wildcards.
drh0ac65892002-04-20 14:24:41 +0000649**
drh4e5ffc52004-08-31 00:52:37 +0000650** Globbing rules:
drh0ac65892002-04-20 14:24:41 +0000651**
drh4e5ffc52004-08-31 00:52:37 +0000652** '*' Matches any sequence of zero or more characters.
danielk1977d02eb1f2004-06-06 09:44:03 +0000653**
drh4e5ffc52004-08-31 00:52:37 +0000654** '?' Matches exactly one character.
655**
656** [...] Matches one character from the enclosed list of
657** characters.
658**
659** [^...] Matches one character not in the enclosed list.
660**
661** With the [...] and [^...] matching, a ']' character can be included
662** in the list by making it the first character after '[' or '^'. A
663** range of characters can be specified using '-'. Example:
664** "[a-z]" matches any single lower-case letter. To match a '-', make
665** it the last character in the list.
666**
drh9fdfdc82014-09-25 11:08:57 +0000667** Like matching rules:
668**
669** '%' Matches any sequence of zero or more characters
670**
671*** '_' Matches any one character
672**
673** Ec Where E is the "esc" character and c is any other
674** character, including '%', '_', and esc, match exactly c.
675**
drhb0870482015-06-17 13:20:54 +0000676** The comments within this routine usually assume glob matching.
drh9fdfdc82014-09-25 11:08:57 +0000677**
drh4e5ffc52004-08-31 00:52:37 +0000678** This routine is usually quick, but can be N**2 in the worst case.
drh0ac65892002-04-20 14:24:41 +0000679*/
danielk19777c6303c2004-11-17 16:41:29 +0000680static int patternCompare(
drh4e5ffc52004-08-31 00:52:37 +0000681 const u8 *zPattern, /* The glob pattern */
682 const u8 *zString, /* The string to compare against the glob */
danielk19777c6303c2004-11-17 16:41:29 +0000683 const struct compareInfo *pInfo, /* Information about how to do the compare */
drh698a01c2016-12-01 18:49:40 +0000684 u32 matchOther /* The escape char (LIKE) or '[' (GLOB) */
danielk197751ad0ec2004-05-24 12:39:02 +0000685){
drh9fdfdc82014-09-25 11:08:57 +0000686 u32 c, c2; /* Next pattern and input string chars */
687 u32 matchOne = pInfo->matchOne; /* "?" or "_" */
688 u32 matchAll = pInfo->matchAll; /* "*" or "%" */
drh9fdfdc82014-09-25 11:08:57 +0000689 u8 noCase = pInfo->noCase; /* True if uppercase==lowercase */
690 const u8 *zEscaped = 0; /* One past the last escaped input char */
drh88b33222014-09-25 03:51:37 +0000691
drhb0870482015-06-17 13:20:54 +0000692 while( (c = Utf8Read(zPattern))!=0 ){
drh9fdfdc82014-09-25 11:08:57 +0000693 if( c==matchAll ){ /* Match "*" */
694 /* Skip over multiple "*" characters in the pattern. If there
695 ** are also "?" characters, skip those as well, but consume a
696 ** single character of the input string for each "?" skipped */
drhb0870482015-06-17 13:20:54 +0000697 while( (c=Utf8Read(zPattern)) == matchAll || c == matchOne ){
drh42610962012-09-17 18:56:32 +0000698 if( c==matchOne && sqlite3Utf8Read(&zString)==0 ){
drh698a01c2016-12-01 18:49:40 +0000699 return SQLITE_NOWILDCARDMATCH;
drh4e5ffc52004-08-31 00:52:37 +0000700 }
danielk1977ad7dd422004-06-06 12:41:49 +0000701 }
drh66150952007-07-23 19:12:41 +0000702 if( c==0 ){
drh698a01c2016-12-01 18:49:40 +0000703 return SQLITE_MATCH; /* "*" at the end of the pattern matches */
drh88b33222014-09-25 03:51:37 +0000704 }else if( c==matchOther ){
drh07e83472016-01-11 03:48:18 +0000705 if( pInfo->matchSet==0 ){
drh88b33222014-09-25 03:51:37 +0000706 c = sqlite3Utf8Read(&zPattern);
drh698a01c2016-12-01 18:49:40 +0000707 if( c==0 ) return SQLITE_NOWILDCARDMATCH;
drh88b33222014-09-25 03:51:37 +0000708 }else{
drh9fdfdc82014-09-25 11:08:57 +0000709 /* "[...]" immediately follows the "*". We have to do a slow
710 ** recursive search in this case, but it is an unusual case. */
drh88b33222014-09-25 03:51:37 +0000711 assert( matchOther<0x80 ); /* '[' is a single-byte character */
drh698a01c2016-12-01 18:49:40 +0000712 while( *zString ){
713 int bMatch = patternCompare(&zPattern[-1],zString,pInfo,matchOther);
714 if( bMatch!=SQLITE_NOMATCH ) return bMatch;
drh88b33222014-09-25 03:51:37 +0000715 SQLITE_SKIP_UTF8(zString);
716 }
drh698a01c2016-12-01 18:49:40 +0000717 return SQLITE_NOWILDCARDMATCH;
drh66150952007-07-23 19:12:41 +0000718 }
drh66150952007-07-23 19:12:41 +0000719 }
drh9fdfdc82014-09-25 11:08:57 +0000720
721 /* At this point variable c contains the first character of the
722 ** pattern string past the "*". Search in the input string for the
dan1a4a7372016-12-01 17:34:59 +0000723 ** first matching character and recursively continue the match from
drh9fdfdc82014-09-25 11:08:57 +0000724 ** that point.
725 **
726 ** For a case-insensitive search, set variable cx to be the same as
727 ** c but in the other case and search the input string for either
728 ** c or cx.
729 */
730 if( c<=0x80 ){
drheba21f92017-10-30 18:49:11 +0000731 char zStop[3];
drh698a01c2016-12-01 18:49:40 +0000732 int bMatch;
drh9fdfdc82014-09-25 11:08:57 +0000733 if( noCase ){
drheba21f92017-10-30 18:49:11 +0000734 zStop[0] = sqlite3Toupper(c);
735 zStop[1] = sqlite3Tolower(c);
736 zStop[2] = 0;
drh66150952007-07-23 19:12:41 +0000737 }else{
drheba21f92017-10-30 18:49:11 +0000738 zStop[0] = c;
739 zStop[1] = 0;
drh4e5ffc52004-08-31 00:52:37 +0000740 }
drheba21f92017-10-30 18:49:11 +0000741 while(1){
742 zString += strcspn((const char*)zString, zStop);
743 if( zString[0]==0 ) break;
744 zString++;
drh698a01c2016-12-01 18:49:40 +0000745 bMatch = patternCompare(zPattern,zString,pInfo,matchOther);
746 if( bMatch!=SQLITE_NOMATCH ) return bMatch;
drh9fdfdc82014-09-25 11:08:57 +0000747 }
748 }else{
drh698a01c2016-12-01 18:49:40 +0000749 int bMatch;
drhb0870482015-06-17 13:20:54 +0000750 while( (c2 = Utf8Read(zString))!=0 ){
drh9fdfdc82014-09-25 11:08:57 +0000751 if( c2!=c ) continue;
drh698a01c2016-12-01 18:49:40 +0000752 bMatch = patternCompare(zPattern,zString,pInfo,matchOther);
753 if( bMatch!=SQLITE_NOMATCH ) return bMatch;
drh9fdfdc82014-09-25 11:08:57 +0000754 }
drh66150952007-07-23 19:12:41 +0000755 }
drh698a01c2016-12-01 18:49:40 +0000756 return SQLITE_NOWILDCARDMATCH;
drh88b33222014-09-25 03:51:37 +0000757 }
drh88b33222014-09-25 03:51:37 +0000758 if( c==matchOther ){
drh07e83472016-01-11 03:48:18 +0000759 if( pInfo->matchSet==0 ){
drh88b33222014-09-25 03:51:37 +0000760 c = sqlite3Utf8Read(&zPattern);
drh698a01c2016-12-01 18:49:40 +0000761 if( c==0 ) return SQLITE_NOMATCH;
drh9fdfdc82014-09-25 11:08:57 +0000762 zEscaped = zPattern;
drh88b33222014-09-25 03:51:37 +0000763 }else{
764 u32 prior_c = 0;
drh9fdfdc82014-09-25 11:08:57 +0000765 int seen = 0;
766 int invert = 0;
drh88b33222014-09-25 03:51:37 +0000767 c = sqlite3Utf8Read(&zString);
drh698a01c2016-12-01 18:49:40 +0000768 if( c==0 ) return SQLITE_NOMATCH;
drh88b33222014-09-25 03:51:37 +0000769 c2 = sqlite3Utf8Read(&zPattern);
770 if( c2=='^' ){
771 invert = 1;
772 c2 = sqlite3Utf8Read(&zPattern);
773 }
774 if( c2==']' ){
775 if( c==']' ) seen = 1;
776 c2 = sqlite3Utf8Read(&zPattern);
777 }
778 while( c2 && c2!=']' ){
779 if( c2=='-' && zPattern[0]!=']' && zPattern[0]!=0 && prior_c>0 ){
780 c2 = sqlite3Utf8Read(&zPattern);
781 if( c>=prior_c && c<=c2 ) seen = 1;
782 prior_c = 0;
783 }else{
784 if( c==c2 ){
785 seen = 1;
786 }
787 prior_c = c2;
788 }
789 c2 = sqlite3Utf8Read(&zPattern);
790 }
791 if( c2==0 || (seen ^ invert)==0 ){
drh698a01c2016-12-01 18:49:40 +0000792 return SQLITE_NOMATCH;
drh88b33222014-09-25 03:51:37 +0000793 }
794 continue;
795 }
796 }
drhb0870482015-06-17 13:20:54 +0000797 c2 = Utf8Read(zString);
drh88b33222014-09-25 03:51:37 +0000798 if( c==c2 ) continue;
drhc80937a2016-06-06 01:48:14 +0000799 if( noCase && sqlite3Tolower(c)==sqlite3Tolower(c2) && c<0x80 && c2<0x80 ){
drh9fdfdc82014-09-25 11:08:57 +0000800 continue;
801 }
802 if( c==matchOne && zPattern!=zEscaped && c2!=0 ) continue;
drh698a01c2016-12-01 18:49:40 +0000803 return SQLITE_NOMATCH;
danielk197751ad0ec2004-05-24 12:39:02 +0000804 }
drh698a01c2016-12-01 18:49:40 +0000805 return *zString==0 ? SQLITE_MATCH : SQLITE_NOMATCH;
drh0ac65892002-04-20 14:24:41 +0000806}
drh4e5ffc52004-08-31 00:52:37 +0000807
drh55ef4d92005-08-14 01:20:37 +0000808/*
drh698a01c2016-12-01 18:49:40 +0000809** The sqlite3_strglob() interface. Return 0 on a match (like strcmp()) and
810** non-zero if there is no match.
drh56282a52013-04-10 16:13:38 +0000811*/
812int sqlite3_strglob(const char *zGlobPattern, const char *zString){
drh698a01c2016-12-01 18:49:40 +0000813 return patternCompare((u8*)zGlobPattern, (u8*)zString, &globInfo, '[');
drh56282a52013-04-10 16:13:38 +0000814}
815
816/*
drh698a01c2016-12-01 18:49:40 +0000817** The sqlite3_strlike() interface. Return 0 on a match and non-zero for
818** a miss - like strcmp().
drh8b4a94a2015-11-24 21:23:59 +0000819*/
820int sqlite3_strlike(const char *zPattern, const char *zStr, unsigned int esc){
drh698a01c2016-12-01 18:49:40 +0000821 return patternCompare((u8*)zPattern, (u8*)zStr, &likeInfoNorm, esc);
drh8b4a94a2015-11-24 21:23:59 +0000822}
823
824/*
drh55ef4d92005-08-14 01:20:37 +0000825** Count the number of times that the LIKE operator (or GLOB which is
826** just a variation of LIKE) gets called. This is used for testing
827** only.
828*/
829#ifdef SQLITE_TEST
830int sqlite3_like_count = 0;
831#endif
832
danielk19773f6b0872004-06-17 05:36:44 +0000833
834/*
835** Implementation of the like() SQL function. This function implements
836** the build-in LIKE operator. The first argument to the function is the
837** pattern and the second argument is the string. So, the SQL statements:
838**
839** A LIKE B
840**
841** is implemented as like(B,A).
842**
drh55ef4d92005-08-14 01:20:37 +0000843** This same function (with a different compareInfo structure) computes
844** the GLOB operator.
danielk19773f6b0872004-06-17 05:36:44 +0000845*/
846static void likeFunc(
847 sqlite3_context *context,
848 int argc,
849 sqlite3_value **argv
850){
drhbeb818d2007-05-08 15:34:47 +0000851 const unsigned char *zA, *zB;
drh07e83472016-01-11 03:48:18 +0000852 u32 escape;
drh27e62db2009-04-02 10:16:17 +0000853 int nPat;
drhbb4957f2008-03-20 14:03:29 +0000854 sqlite3 *db = sqlite3_context_db_handle(context);
drh07e83472016-01-11 03:48:18 +0000855 struct compareInfo *pInfo = sqlite3_user_data(context);
drh589c7872020-03-19 18:13:28 +0000856 struct compareInfo backupInfo;
drhbeb818d2007-05-08 15:34:47 +0000857
drh41d2e662015-12-01 21:23:07 +0000858#ifdef SQLITE_LIKE_DOESNT_MATCH_BLOBS
859 if( sqlite3_value_type(argv[0])==SQLITE_BLOB
860 || sqlite3_value_type(argv[1])==SQLITE_BLOB
861 ){
862#ifdef SQLITE_TEST
863 sqlite3_like_count++;
864#endif
865 sqlite3_result_int(context, 0);
866 return;
867 }
868#endif
drh1f0feef2007-05-15 13:27:07 +0000869
drhbeb818d2007-05-08 15:34:47 +0000870 /* Limit the length of the LIKE or GLOB pattern to avoid problems
871 ** of deep recursion and N*N behavior in patternCompare().
872 */
drh27e62db2009-04-02 10:16:17 +0000873 nPat = sqlite3_value_bytes(argv[0]);
874 testcase( nPat==db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH] );
875 testcase( nPat==db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH]+1 );
876 if( nPat > db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH] ){
drhbeb818d2007-05-08 15:34:47 +0000877 sqlite3_result_error(context, "LIKE or GLOB pattern too complex", -1);
878 return;
879 }
danielk19777c6303c2004-11-17 16:41:29 +0000880 if( argc==3 ){
881 /* The escape character string must consist of a single UTF-8 character.
882 ** Otherwise, return an error.
883 */
884 const unsigned char *zEsc = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +0000885 if( zEsc==0 ) return;
drhee858132007-05-08 20:37:38 +0000886 if( sqlite3Utf8CharLen((char*)zEsc, -1)!=1 ){
danielk19777c6303c2004-11-17 16:41:29 +0000887 sqlite3_result_error(context,
888 "ESCAPE expression must be a single character", -1);
889 return;
890 }
drh42610962012-09-17 18:56:32 +0000891 escape = sqlite3Utf8Read(&zEsc);
drh589c7872020-03-19 18:13:28 +0000892 if( escape==pInfo->matchAll || escape==pInfo->matchOne ){
893 memcpy(&backupInfo, pInfo, sizeof(backupInfo));
894 pInfo = &backupInfo;
895 if( escape==pInfo->matchAll ) pInfo->matchAll = 0;
896 if( escape==pInfo->matchOne ) pInfo->matchOne = 0;
897 }
drh07e83472016-01-11 03:48:18 +0000898 }else{
899 escape = pInfo->matchSet;
danielk19777c6303c2004-11-17 16:41:29 +0000900 }
drhcf833232019-04-30 11:54:36 +0000901 zB = sqlite3_value_text(argv[0]);
902 zA = sqlite3_value_text(argv[1]);
danielk19773f6b0872004-06-17 05:36:44 +0000903 if( zA && zB ){
drh55ef4d92005-08-14 01:20:37 +0000904#ifdef SQLITE_TEST
905 sqlite3_like_count++;
906#endif
drhf49759b2017-08-25 19:51:51 +0000907 sqlite3_result_int(context,
908 patternCompare(zB, zA, pInfo, escape)==SQLITE_MATCH);
danielk197751ad0ec2004-05-24 12:39:02 +0000909 }
drh8912d102002-05-26 21:34:58 +0000910}
911
912/*
913** Implementation of the NULLIF(x,y) function. The result is the first
914** argument if the arguments are different. The result is NULL if the
915** arguments are equal to each other.
916*/
drhf9b596e2004-05-26 16:54:42 +0000917static void nullifFunc(
918 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000919 int NotUsed,
drhf9b596e2004-05-26 16:54:42 +0000920 sqlite3_value **argv
921){
danielk1977dc1bdc42004-06-11 10:51:27 +0000922 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk197762c14b32008-11-19 09:05:26 +0000923 UNUSED_PARAMETER(NotUsed);
danielk1977dc1bdc42004-06-11 10:51:27 +0000924 if( sqlite3MemCompare(argv[0], argv[1], pColl)!=0 ){
drhf4479502004-05-27 03:12:53 +0000925 sqlite3_result_value(context, argv[0]);
drh8912d102002-05-26 21:34:58 +0000926 }
drh0ac65892002-04-20 14:24:41 +0000927}
928
drh647cb0e2002-11-04 19:32:25 +0000929/*
drh47baebc2009-08-14 16:01:24 +0000930** Implementation of the sqlite_version() function. The result is the version
drh647cb0e2002-11-04 19:32:25 +0000931** of the SQLite library that is running.
932*/
drhf9b596e2004-05-26 16:54:42 +0000933static void versionFunc(
934 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000935 int NotUsed,
936 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000937){
danielk197762c14b32008-11-19 09:05:26 +0000938 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000939 /* IMP: R-48699-48617 This function is an SQL wrapper around the
940 ** sqlite3_libversion() C-interface. */
941 sqlite3_result_text(context, sqlite3_libversion(), -1, SQLITE_STATIC);
drh647cb0e2002-11-04 19:32:25 +0000942}
943
drh47baebc2009-08-14 16:01:24 +0000944/*
945** Implementation of the sqlite_source_id() function. The result is a string
946** that identifies the particular version of the source code used to build
947** SQLite.
948*/
949static void sourceidFunc(
950 sqlite3_context *context,
951 int NotUsed,
952 sqlite3_value **NotUsed2
953){
954 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000955 /* IMP: R-24470-31136 This function is an SQL wrapper around the
956 ** sqlite3_sourceid() C interface. */
957 sqlite3_result_text(context, sqlite3_sourceid(), -1, SQLITE_STATIC);
drh47baebc2009-08-14 16:01:24 +0000958}
959
shanehbdea6d12010-02-23 04:19:54 +0000960/*
drh3ca84ef2011-04-25 18:03:10 +0000961** Implementation of the sqlite_log() function. This is a wrapper around
962** sqlite3_log(). The return value is NULL. The function exists purely for
963** its side-effects.
964*/
drh840561f2011-04-27 18:08:42 +0000965static void errlogFunc(
drh3ca84ef2011-04-25 18:03:10 +0000966 sqlite3_context *context,
967 int argc,
968 sqlite3_value **argv
969){
970 UNUSED_PARAMETER(argc);
971 UNUSED_PARAMETER(context);
972 sqlite3_log(sqlite3_value_int(argv[0]), "%s", sqlite3_value_text(argv[1]));
973}
974
975/*
shanehdc97a8c2010-02-23 20:08:35 +0000976** Implementation of the sqlite_compileoption_used() function.
977** The result is an integer that identifies if the compiler option
978** was used to build SQLite.
shanehbdea6d12010-02-23 04:19:54 +0000979*/
shanehdc97a8c2010-02-23 20:08:35 +0000980#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
981static void compileoptionusedFunc(
shanehbdea6d12010-02-23 04:19:54 +0000982 sqlite3_context *context,
shanehdc97a8c2010-02-23 20:08:35 +0000983 int argc,
984 sqlite3_value **argv
shanehbdea6d12010-02-23 04:19:54 +0000985){
shanehdc97a8c2010-02-23 20:08:35 +0000986 const char *zOptName;
987 assert( argc==1 );
988 UNUSED_PARAMETER(argc);
drha3e414c2010-08-03 18:06:25 +0000989 /* IMP: R-39564-36305 The sqlite_compileoption_used() SQL
990 ** function is a wrapper around the sqlite3_compileoption_used() C/C++
991 ** function.
992 */
drh264a2d42010-02-25 15:28:41 +0000993 if( (zOptName = (const char*)sqlite3_value_text(argv[0]))!=0 ){
shanehdc97a8c2010-02-23 20:08:35 +0000994 sqlite3_result_int(context, sqlite3_compileoption_used(zOptName));
shanehdc97a8c2010-02-23 20:08:35 +0000995 }
shanehbdea6d12010-02-23 04:19:54 +0000996}
shanehdc97a8c2010-02-23 20:08:35 +0000997#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
998
999/*
1000** Implementation of the sqlite_compileoption_get() function.
1001** The result is a string that identifies the compiler options
1002** used to build SQLite.
1003*/
1004#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
1005static void compileoptiongetFunc(
1006 sqlite3_context *context,
1007 int argc,
1008 sqlite3_value **argv
1009){
1010 int n;
1011 assert( argc==1 );
1012 UNUSED_PARAMETER(argc);
drha3e414c2010-08-03 18:06:25 +00001013 /* IMP: R-04922-24076 The sqlite_compileoption_get() SQL function
1014 ** is a wrapper around the sqlite3_compileoption_get() C/C++ function.
1015 */
shanehdc97a8c2010-02-23 20:08:35 +00001016 n = sqlite3_value_int(argv[0]);
1017 sqlite3_result_text(context, sqlite3_compileoption_get(n), -1, SQLITE_STATIC);
1018}
1019#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
shanehbdea6d12010-02-23 04:19:54 +00001020
drh137c7282007-01-29 17:58:28 +00001021/* Array for converting from half-bytes (nybbles) into ASCII hex
1022** digits. */
1023static const char hexdigits[] = {
1024 '0', '1', '2', '3', '4', '5', '6', '7',
1025 '8', '9', 'A', 'B', 'C', 'D', 'E', 'F'
1026};
danielk1977d641d642004-11-18 15:44:29 +00001027
drh47394702003-08-20 01:03:33 +00001028/*
drh47394702003-08-20 01:03:33 +00001029** Implementation of the QUOTE() function. This function takes a single
1030** argument. If the argument is numeric, the return value is the same as
1031** the argument. If the argument is NULL, the return value is the string
1032** "NULL". Otherwise, the argument is enclosed in single quotes with
1033** single-quote escapes.
1034*/
danielk19770ae8b832004-05-25 12:05:56 +00001035static void quoteFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drha0df4cc2009-02-02 17:29:59 +00001036 assert( argc==1 );
drh1d34fde2009-02-03 15:50:33 +00001037 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +00001038 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +00001039 case SQLITE_FLOAT: {
drh72b3fbc2012-06-19 03:11:25 +00001040 double r1, r2;
drh72b3fbc2012-06-19 03:11:25 +00001041 char zBuf[50];
mistachkin2b434a72012-06-19 04:36:48 +00001042 r1 = sqlite3_value_double(argv[0]);
drh72b3fbc2012-06-19 03:11:25 +00001043 sqlite3_snprintf(sizeof(zBuf), zBuf, "%!.15g", r1);
1044 sqlite3AtoF(zBuf, &r2, 20, SQLITE_UTF8);
1045 if( r1!=r2 ){
1046 sqlite3_snprintf(sizeof(zBuf), zBuf, "%!.20e", r1);
1047 }
1048 sqlite3_result_text(context, zBuf, -1, SQLITE_TRANSIENT);
1049 break;
1050 }
1051 case SQLITE_INTEGER: {
drhf4479502004-05-27 03:12:53 +00001052 sqlite3_result_value(context, argv[0]);
drhf9b596e2004-05-26 16:54:42 +00001053 break;
1054 }
danielk19773f41e972004-06-08 00:39:01 +00001055 case SQLITE_BLOB: {
danielk19773f41e972004-06-08 00:39:01 +00001056 char *zText = 0;
danielk19773f41e972004-06-08 00:39:01 +00001057 char const *zBlob = sqlite3_value_blob(argv[0]);
drh1f0feef2007-05-15 13:27:07 +00001058 int nBlob = sqlite3_value_bytes(argv[0]);
1059 assert( zBlob==sqlite3_value_blob(argv[0]) ); /* No encoding change */
drhb1a6c3c2008-03-20 16:30:17 +00001060 zText = (char *)contextMalloc(context, (2*(i64)nBlob)+4);
danielk1977a1644fd2007-08-29 12:31:25 +00001061 if( zText ){
danielk19773f41e972004-06-08 00:39:01 +00001062 int i;
1063 for(i=0; i<nBlob; i++){
1064 zText[(i*2)+2] = hexdigits[(zBlob[i]>>4)&0x0F];
1065 zText[(i*2)+3] = hexdigits[(zBlob[i])&0x0F];
1066 }
1067 zText[(nBlob*2)+2] = '\'';
1068 zText[(nBlob*2)+3] = '\0';
1069 zText[0] = 'X';
1070 zText[1] = '\'';
danielk1977d8123362004-06-12 09:25:12 +00001071 sqlite3_result_text(context, zText, -1, SQLITE_TRANSIENT);
drh17435752007-08-16 04:30:38 +00001072 sqlite3_free(zText);
danielk19773f41e972004-06-08 00:39:01 +00001073 }
1074 break;
1075 }
drh9c054832004-05-31 18:51:57 +00001076 case SQLITE_TEXT: {
drh023ae032007-05-08 12:12:16 +00001077 int i,j;
1078 u64 n;
drh2646da72005-12-09 20:02:05 +00001079 const unsigned char *zArg = sqlite3_value_text(argv[0]);
drhf9b596e2004-05-26 16:54:42 +00001080 char *z;
1081
drh7a521cf2007-04-25 18:23:52 +00001082 if( zArg==0 ) return;
drh023ae032007-05-08 12:12:16 +00001083 for(i=0, n=0; zArg[i]; i++){ if( zArg[i]=='\'' ) n++; }
drhb1a6c3c2008-03-20 16:30:17 +00001084 z = contextMalloc(context, ((i64)i)+((i64)n)+3);
danielk1977a1644fd2007-08-29 12:31:25 +00001085 if( z ){
1086 z[0] = '\'';
1087 for(i=0, j=1; zArg[i]; i++){
1088 z[j++] = zArg[i];
1089 if( zArg[i]=='\'' ){
1090 z[j++] = '\'';
1091 }
drhf9b596e2004-05-26 16:54:42 +00001092 }
danielk1977a1644fd2007-08-29 12:31:25 +00001093 z[j++] = '\'';
1094 z[j] = 0;
1095 sqlite3_result_text(context, z, j, sqlite3_free);
drhf9b596e2004-05-26 16:54:42 +00001096 }
drha0df4cc2009-02-02 17:29:59 +00001097 break;
1098 }
1099 default: {
1100 assert( sqlite3_value_type(argv[0])==SQLITE_NULL );
1101 sqlite3_result_text(context, "NULL", 4, SQLITE_STATIC);
1102 break;
drhf9b596e2004-05-26 16:54:42 +00001103 }
drh47394702003-08-20 01:03:33 +00001104 }
1105}
1106
drh137c7282007-01-29 17:58:28 +00001107/*
drhd495d8c2013-02-22 19:34:25 +00001108** The unicode() function. Return the integer unicode code-point value
1109** for the first character of the input string.
1110*/
1111static void unicodeFunc(
1112 sqlite3_context *context,
1113 int argc,
1114 sqlite3_value **argv
1115){
1116 const unsigned char *z = sqlite3_value_text(argv[0]);
drh1d59d032013-03-01 23:40:26 +00001117 (void)argc;
drhd495d8c2013-02-22 19:34:25 +00001118 if( z && z[0] ) sqlite3_result_int(context, sqlite3Utf8Read(&z));
1119}
1120
1121/*
1122** The char() function takes zero or more arguments, each of which is
1123** an integer. It constructs a string where each character of the string
1124** is the unicode character for the corresponding integer argument.
1125*/
1126static void charFunc(
1127 sqlite3_context *context,
1128 int argc,
1129 sqlite3_value **argv
1130){
1131 unsigned char *z, *zOut;
1132 int i;
drhf3cdcdc2015-04-29 16:50:28 +00001133 zOut = z = sqlite3_malloc64( argc*4+1 );
drhd495d8c2013-02-22 19:34:25 +00001134 if( z==0 ){
1135 sqlite3_result_error_nomem(context);
1136 return;
1137 }
1138 for(i=0; i<argc; i++){
mistachkinc9545442013-02-26 05:42:30 +00001139 sqlite3_int64 x;
drhd495d8c2013-02-22 19:34:25 +00001140 unsigned c;
1141 x = sqlite3_value_int64(argv[i]);
1142 if( x<0 || x>0x10ffff ) x = 0xfffd;
1143 c = (unsigned)(x & 0x1fffff);
drhfe7a5d12013-03-07 14:00:04 +00001144 if( c<0x00080 ){
1145 *zOut++ = (u8)(c&0xFF);
1146 }else if( c<0x00800 ){
1147 *zOut++ = 0xC0 + (u8)((c>>6)&0x1F);
1148 *zOut++ = 0x80 + (u8)(c & 0x3F);
1149 }else if( c<0x10000 ){
1150 *zOut++ = 0xE0 + (u8)((c>>12)&0x0F);
1151 *zOut++ = 0x80 + (u8)((c>>6) & 0x3F);
1152 *zOut++ = 0x80 + (u8)(c & 0x3F);
drhd495d8c2013-02-22 19:34:25 +00001153 }else{
drhfe7a5d12013-03-07 14:00:04 +00001154 *zOut++ = 0xF0 + (u8)((c>>18) & 0x07);
1155 *zOut++ = 0x80 + (u8)((c>>12) & 0x3F);
1156 *zOut++ = 0x80 + (u8)((c>>6) & 0x3F);
1157 *zOut++ = 0x80 + (u8)(c & 0x3F);
1158 } \
drhd495d8c2013-02-22 19:34:25 +00001159 }
drhbbf483f2014-09-09 20:30:24 +00001160 sqlite3_result_text64(context, (char*)z, zOut-z, sqlite3_free, SQLITE_UTF8);
drhd495d8c2013-02-22 19:34:25 +00001161}
1162
1163/*
drh137c7282007-01-29 17:58:28 +00001164** The hex() function. Interpret the argument as a blob. Return
1165** a hexadecimal rendering as text.
1166*/
1167static void hexFunc(
1168 sqlite3_context *context,
1169 int argc,
1170 sqlite3_value **argv
1171){
1172 int i, n;
1173 const unsigned char *pBlob;
1174 char *zHex, *z;
1175 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001176 UNUSED_PARAMETER(argc);
drh1f0feef2007-05-15 13:27:07 +00001177 pBlob = sqlite3_value_blob(argv[0]);
drh137c7282007-01-29 17:58:28 +00001178 n = sqlite3_value_bytes(argv[0]);
drh1f0feef2007-05-15 13:27:07 +00001179 assert( pBlob==sqlite3_value_blob(argv[0]) ); /* No encoding change */
drhb1a6c3c2008-03-20 16:30:17 +00001180 z = zHex = contextMalloc(context, ((i64)n)*2 + 1);
danielk1977a1644fd2007-08-29 12:31:25 +00001181 if( zHex ){
1182 for(i=0; i<n; i++, pBlob++){
1183 unsigned char c = *pBlob;
1184 *(z++) = hexdigits[(c>>4)&0xf];
1185 *(z++) = hexdigits[c&0xf];
1186 }
1187 *z = 0;
1188 sqlite3_result_text(context, zHex, n*2, sqlite3_free);
drh137c7282007-01-29 17:58:28 +00001189 }
drh137c7282007-01-29 17:58:28 +00001190}
1191
drh26b6d902007-03-17 13:27:54 +00001192/*
drh8cff3822007-05-02 02:08:28 +00001193** The zeroblob(N) function returns a zero-filled blob of size N bytes.
1194*/
1195static void zeroblobFunc(
1196 sqlite3_context *context,
1197 int argc,
1198 sqlite3_value **argv
1199){
drh98640a32007-06-07 19:08:32 +00001200 i64 n;
dana4d5ae82015-07-24 16:24:37 +00001201 int rc;
drh8cff3822007-05-02 02:08:28 +00001202 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001203 UNUSED_PARAMETER(argc);
drh98640a32007-06-07 19:08:32 +00001204 n = sqlite3_value_int64(argv[0]);
dana4d5ae82015-07-24 16:24:37 +00001205 if( n<0 ) n = 0;
1206 rc = sqlite3_result_zeroblob64(context, n); /* IMP: R-00293-64994 */
1207 if( rc ){
1208 sqlite3_result_error_code(context, rc);
drh98640a32007-06-07 19:08:32 +00001209 }
drh8cff3822007-05-02 02:08:28 +00001210}
1211
1212/*
drh26b6d902007-03-17 13:27:54 +00001213** The replace() function. Three arguments are all strings: call
1214** them A, B, and C. The result is also a string which is derived
drhf7b54962013-05-28 12:11:54 +00001215** from A by replacing every occurrence of B with C. The match
drh26b6d902007-03-17 13:27:54 +00001216** must be exact. Collating sequences are not used.
1217*/
1218static void replaceFunc(
1219 sqlite3_context *context,
1220 int argc,
1221 sqlite3_value **argv
1222){
1223 const unsigned char *zStr; /* The input string A */
1224 const unsigned char *zPattern; /* The pattern string B */
1225 const unsigned char *zRep; /* The replacement string C */
1226 unsigned char *zOut; /* The output */
1227 int nStr; /* Size of zStr */
1228 int nPattern; /* Size of zPattern */
1229 int nRep; /* Size of zRep */
drh2e6400b2007-05-08 15:46:18 +00001230 i64 nOut; /* Maximum size of zOut */
drh26b6d902007-03-17 13:27:54 +00001231 int loopLimit; /* Last zStr[] that might match zPattern[] */
1232 int i, j; /* Loop counters */
drhf3139522018-02-09 23:25:14 +00001233 unsigned cntExpand; /* Number zOut expansions */
1234 sqlite3 *db = sqlite3_context_db_handle(context);
drh26b6d902007-03-17 13:27:54 +00001235
1236 assert( argc==3 );
danielk1977f3d3c272008-11-19 16:52:44 +00001237 UNUSED_PARAMETER(argc);
drh26b6d902007-03-17 13:27:54 +00001238 zStr = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001239 if( zStr==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001240 nStr = sqlite3_value_bytes(argv[0]);
1241 assert( zStr==sqlite3_value_text(argv[0]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001242 zPattern = sqlite3_value_text(argv[1]);
drha605fe82009-02-01 18:08:40 +00001243 if( zPattern==0 ){
drh23336062009-02-03 13:19:12 +00001244 assert( sqlite3_value_type(argv[1])==SQLITE_NULL
1245 || sqlite3_context_db_handle(context)->mallocFailed );
drha605fe82009-02-01 18:08:40 +00001246 return;
1247 }
1248 if( zPattern[0]==0 ){
1249 assert( sqlite3_value_type(argv[1])!=SQLITE_NULL );
1250 sqlite3_result_value(context, argv[0]);
1251 return;
1252 }
drh1f0feef2007-05-15 13:27:07 +00001253 nPattern = sqlite3_value_bytes(argv[1]);
1254 assert( zPattern==sqlite3_value_text(argv[1]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001255 zRep = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +00001256 if( zRep==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001257 nRep = sqlite3_value_bytes(argv[2]);
1258 assert( zRep==sqlite3_value_text(argv[2]) );
drh2e6400b2007-05-08 15:46:18 +00001259 nOut = nStr + 1;
1260 assert( nOut<SQLITE_MAX_LENGTH );
drhb1a6c3c2008-03-20 16:30:17 +00001261 zOut = contextMalloc(context, (i64)nOut);
drh2e6400b2007-05-08 15:46:18 +00001262 if( zOut==0 ){
1263 return;
drh26b6d902007-03-17 13:27:54 +00001264 }
drh26b6d902007-03-17 13:27:54 +00001265 loopLimit = nStr - nPattern;
drhf3139522018-02-09 23:25:14 +00001266 cntExpand = 0;
drh26b6d902007-03-17 13:27:54 +00001267 for(i=j=0; i<=loopLimit; i++){
1268 if( zStr[i]!=zPattern[0] || memcmp(&zStr[i], zPattern, nPattern) ){
1269 zOut[j++] = zStr[i];
1270 }else{
drhf3139522018-02-09 23:25:14 +00001271 if( nRep>nPattern ){
1272 nOut += nRep - nPattern;
drhc86d82f2018-02-10 02:31:30 +00001273 testcase( nOut-1==db->aLimit[SQLITE_LIMIT_LENGTH] );
1274 testcase( nOut-2==db->aLimit[SQLITE_LIMIT_LENGTH] );
drhf3139522018-02-09 23:25:14 +00001275 if( nOut-1>db->aLimit[SQLITE_LIMIT_LENGTH] ){
1276 sqlite3_result_error_toobig(context);
1277 sqlite3_free(zOut);
1278 return;
1279 }
drhf3139522018-02-09 23:25:14 +00001280 cntExpand++;
1281 if( (cntExpand&(cntExpand-1))==0 ){
1282 /* Grow the size of the output buffer only on substitutions
1283 ** whose index is a power of two: 1, 2, 4, 8, 16, 32, ... */
1284 u8 *zOld;
1285 zOld = zOut;
drhd924e7b2020-05-17 00:26:44 +00001286 zOut = sqlite3Realloc(zOut, (int)nOut + (nOut - nStr - 1));
drhf3139522018-02-09 23:25:14 +00001287 if( zOut==0 ){
1288 sqlite3_result_error_nomem(context);
1289 sqlite3_free(zOld);
1290 return;
1291 }
1292 }
drh2e6400b2007-05-08 15:46:18 +00001293 }
drh26b6d902007-03-17 13:27:54 +00001294 memcpy(&zOut[j], zRep, nRep);
1295 j += nRep;
1296 i += nPattern-1;
1297 }
1298 }
drhf3139522018-02-09 23:25:14 +00001299 assert( j+nStr-i+1<=nOut );
drh26b6d902007-03-17 13:27:54 +00001300 memcpy(&zOut[j], &zStr[i], nStr-i);
1301 j += nStr - i;
1302 assert( j<=nOut );
1303 zOut[j] = 0;
1304 sqlite3_result_text(context, (char*)zOut, j, sqlite3_free);
1305}
1306
drh309b3382007-03-17 17:52:42 +00001307/*
1308** Implementation of the TRIM(), LTRIM(), and RTRIM() functions.
1309** The userdata is 0x1 for left trim, 0x2 for right trim, 0x3 for both.
1310*/
1311static void trimFunc(
1312 sqlite3_context *context,
1313 int argc,
1314 sqlite3_value **argv
1315){
1316 const unsigned char *zIn; /* Input string */
1317 const unsigned char *zCharSet; /* Set of characters to trim */
1318 int nIn; /* Number of bytes in input */
drh7209c692008-04-27 18:40:11 +00001319 int flags; /* 1: trimleft 2: trimright 3: trim */
drhd1e3a612007-04-27 21:59:52 +00001320 int i; /* Loop counter */
drh1bd10f82008-12-10 21:19:56 +00001321 unsigned char *aLen = 0; /* Length of each character in zCharSet */
1322 unsigned char **azChar = 0; /* Individual characters in zCharSet */
drhd1e3a612007-04-27 21:59:52 +00001323 int nChar; /* Number of characters in zCharSet */
1324
drh309b3382007-03-17 17:52:42 +00001325 if( sqlite3_value_type(argv[0])==SQLITE_NULL ){
1326 return;
1327 }
1328 zIn = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001329 if( zIn==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001330 nIn = sqlite3_value_bytes(argv[0]);
1331 assert( zIn==sqlite3_value_text(argv[0]) );
drh309b3382007-03-17 17:52:42 +00001332 if( argc==1 ){
drhd1e3a612007-04-27 21:59:52 +00001333 static const unsigned char lenOne[] = { 1 };
danielk1977a4de4532008-09-02 15:44:08 +00001334 static unsigned char * const azOne[] = { (u8*)" " };
drhd1e3a612007-04-27 21:59:52 +00001335 nChar = 1;
drh8cff3822007-05-02 02:08:28 +00001336 aLen = (u8*)lenOne;
danielk1977bc67da42007-12-11 04:23:19 +00001337 azChar = (unsigned char **)azOne;
drhd1e3a612007-04-27 21:59:52 +00001338 zCharSet = 0;
drh7a521cf2007-04-25 18:23:52 +00001339 }else if( (zCharSet = sqlite3_value_text(argv[1]))==0 ){
drh309b3382007-03-17 17:52:42 +00001340 return;
drhd1e3a612007-04-27 21:59:52 +00001341 }else{
1342 const unsigned char *z;
1343 for(z=zCharSet, nChar=0; *z; nChar++){
drh4a919112007-05-15 11:55:09 +00001344 SQLITE_SKIP_UTF8(z);
drhd1e3a612007-04-27 21:59:52 +00001345 }
1346 if( nChar>0 ){
drhb1a6c3c2008-03-20 16:30:17 +00001347 azChar = contextMalloc(context, ((i64)nChar)*(sizeof(char*)+1));
drhd1e3a612007-04-27 21:59:52 +00001348 if( azChar==0 ){
1349 return;
1350 }
1351 aLen = (unsigned char*)&azChar[nChar];
1352 for(z=zCharSet, nChar=0; *z; nChar++){
danielk1977bc67da42007-12-11 04:23:19 +00001353 azChar[nChar] = (unsigned char *)z;
drh4a919112007-05-15 11:55:09 +00001354 SQLITE_SKIP_UTF8(z);
drh3abbd392008-12-10 23:04:13 +00001355 aLen[nChar] = (u8)(z - azChar[nChar]);
drhd1e3a612007-04-27 21:59:52 +00001356 }
1357 }
drh309b3382007-03-17 17:52:42 +00001358 }
drhd1e3a612007-04-27 21:59:52 +00001359 if( nChar>0 ){
shane1fc41292008-07-08 22:28:48 +00001360 flags = SQLITE_PTR_TO_INT(sqlite3_user_data(context));
drh309b3382007-03-17 17:52:42 +00001361 if( flags & 1 ){
drhd1e3a612007-04-27 21:59:52 +00001362 while( nIn>0 ){
drh1bd10f82008-12-10 21:19:56 +00001363 int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001364 for(i=0; i<nChar; i++){
1365 len = aLen[i];
drh27e62db2009-04-02 10:16:17 +00001366 if( len<=nIn && memcmp(zIn, azChar[i], len)==0 ) break;
drhd1e3a612007-04-27 21:59:52 +00001367 }
1368 if( i>=nChar ) break;
1369 zIn += len;
1370 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001371 }
1372 }
1373 if( flags & 2 ){
drhd1e3a612007-04-27 21:59:52 +00001374 while( nIn>0 ){
drh1bd10f82008-12-10 21:19:56 +00001375 int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001376 for(i=0; i<nChar; i++){
1377 len = aLen[i];
1378 if( len<=nIn && memcmp(&zIn[nIn-len],azChar[i],len)==0 ) break;
1379 }
1380 if( i>=nChar ) break;
1381 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001382 }
1383 }
drhd1e3a612007-04-27 21:59:52 +00001384 if( zCharSet ){
1385 sqlite3_free(azChar);
1386 }
drh309b3382007-03-17 17:52:42 +00001387 }
1388 sqlite3_result_text(context, (char*)zIn, nIn, SQLITE_TRANSIENT);
1389}
drh26b6d902007-03-17 13:27:54 +00001390
danielk1977a4de4532008-09-02 15:44:08 +00001391
drhcc153132016-08-04 12:35:17 +00001392#ifdef SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION
1393/*
1394** The "unknown" function is automatically substituted in place of
1395** any unrecognized function name when doing an EXPLAIN or EXPLAIN QUERY PLAN
1396** when the SQLITE_ENABLE_UNKNOWN_FUNCTION compile-time option is used.
1397** When the "sqlite3" command-line shell is built using this functionality,
1398** that allows an EXPLAIN or EXPLAIN QUERY PLAN for complex queries
1399** involving application-defined functions to be examined in a generic
1400** sqlite3 shell.
1401*/
1402static void unknownFunc(
1403 sqlite3_context *context,
1404 int argc,
1405 sqlite3_value **argv
1406){
1407 /* no-op */
1408}
1409#endif /*SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION*/
1410
1411
drh2ba3ccc2009-12-08 02:06:08 +00001412/* IMP: R-25361-16150 This function is omitted from SQLite by default. It
1413** is only available if the SQLITE_SOUNDEX compile-time option is used
1414** when SQLite is built.
1415*/
drhd24cc422003-03-27 12:51:24 +00001416#ifdef SQLITE_SOUNDEX
1417/*
1418** Compute the soundex encoding of a word.
drh2ba3ccc2009-12-08 02:06:08 +00001419**
1420** IMP: R-59782-00072 The soundex(X) function returns a string that is the
1421** soundex encoding of the string X.
drhd24cc422003-03-27 12:51:24 +00001422*/
drh137c7282007-01-29 17:58:28 +00001423static void soundexFunc(
1424 sqlite3_context *context,
1425 int argc,
1426 sqlite3_value **argv
1427){
drhd24cc422003-03-27 12:51:24 +00001428 char zResult[8];
drh4c755c02004-08-08 20:22:17 +00001429 const u8 *zIn;
drhd24cc422003-03-27 12:51:24 +00001430 int i, j;
1431 static const unsigned char iCode[] = {
1432 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1433 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1434 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1435 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1436 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1437 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1438 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1439 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1440 };
1441 assert( argc==1 );
drh4c755c02004-08-08 20:22:17 +00001442 zIn = (u8*)sqlite3_value_text(argv[0]);
drhbdf67e02006-08-19 11:34:01 +00001443 if( zIn==0 ) zIn = (u8*)"";
drhdc86e2b2009-01-24 11:30:42 +00001444 for(i=0; zIn[i] && !sqlite3Isalpha(zIn[i]); i++){}
drhd24cc422003-03-27 12:51:24 +00001445 if( zIn[i] ){
drhbdf67e02006-08-19 11:34:01 +00001446 u8 prevcode = iCode[zIn[i]&0x7f];
danielk197778ca0e72009-01-20 16:53:39 +00001447 zResult[0] = sqlite3Toupper(zIn[i]);
drhd24cc422003-03-27 12:51:24 +00001448 for(j=1; j<4 && zIn[i]; i++){
1449 int code = iCode[zIn[i]&0x7f];
1450 if( code>0 ){
drhbdf67e02006-08-19 11:34:01 +00001451 if( code!=prevcode ){
1452 prevcode = code;
1453 zResult[j++] = code + '0';
1454 }
1455 }else{
1456 prevcode = 0;
drhd24cc422003-03-27 12:51:24 +00001457 }
1458 }
1459 while( j<4 ){
1460 zResult[j++] = '0';
1461 }
1462 zResult[j] = 0;
danielk1977d8123362004-06-12 09:25:12 +00001463 sqlite3_result_text(context, zResult, 4, SQLITE_TRANSIENT);
drhd24cc422003-03-27 12:51:24 +00001464 }else{
drh2ba3ccc2009-12-08 02:06:08 +00001465 /* IMP: R-64894-50321 The string "?000" is returned if the argument
1466 ** is NULL or contains no ASCII alphabetic characters. */
danielk1977d8123362004-06-12 09:25:12 +00001467 sqlite3_result_text(context, "?000", 4, SQLITE_STATIC);
drhd24cc422003-03-27 12:51:24 +00001468 }
1469}
drh2ba3ccc2009-12-08 02:06:08 +00001470#endif /* SQLITE_SOUNDEX */
drhd24cc422003-03-27 12:51:24 +00001471
drhfdb83b22006-06-17 14:12:47 +00001472#ifndef SQLITE_OMIT_LOAD_EXTENSION
1473/*
1474** A function that loads a shared-library extension then returns NULL.
1475*/
1476static void loadExt(sqlite3_context *context, int argc, sqlite3_value **argv){
danielk197765fd59f2006-06-24 11:51:33 +00001477 const char *zFile = (const char *)sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001478 const char *zProc;
drhfa4a4b92008-03-19 21:45:51 +00001479 sqlite3 *db = sqlite3_context_db_handle(context);
drhfdb83b22006-06-17 14:12:47 +00001480 char *zErrMsg = 0;
1481
drh191dd062016-04-21 01:30:09 +00001482 /* Disallow the load_extension() SQL function unless the SQLITE_LoadExtFunc
drh1a55ded2016-04-20 00:30:05 +00001483 ** flag is set. See the sqlite3_enable_load_extension() API.
1484 */
drhf602a162016-04-21 01:58:21 +00001485 if( (db->flags & SQLITE_LoadExtFunc)==0 ){
1486 sqlite3_result_error(context, "not authorized", -1);
1487 return;
1488 }
drh1a55ded2016-04-20 00:30:05 +00001489
drhfdb83b22006-06-17 14:12:47 +00001490 if( argc==2 ){
danielk197765fd59f2006-06-24 11:51:33 +00001491 zProc = (const char *)sqlite3_value_text(argv[1]);
drh7a521cf2007-04-25 18:23:52 +00001492 }else{
1493 zProc = 0;
drhfdb83b22006-06-17 14:12:47 +00001494 }
drh7a521cf2007-04-25 18:23:52 +00001495 if( zFile && sqlite3_load_extension(db, zFile, zProc, &zErrMsg) ){
drhfdb83b22006-06-17 14:12:47 +00001496 sqlite3_result_error(context, zErrMsg, -1);
1497 sqlite3_free(zErrMsg);
1498 }
1499}
1500#endif
1501
danielk197701427a62005-01-11 13:02:33 +00001502
drh0ac65892002-04-20 14:24:41 +00001503/*
drhd3a149e2002-02-24 17:12:53 +00001504** An instance of the following structure holds the context of a
drhdd5baa92002-02-27 19:50:59 +00001505** sum() or avg() aggregate computation.
1506*/
1507typedef struct SumCtx SumCtx;
1508struct SumCtx {
drh8c08e862006-02-11 17:34:00 +00001509 double rSum; /* Floating point sum */
1510 i64 iSum; /* Integer sum */
1511 i64 cnt; /* Number of elements summed */
1512 u8 overflow; /* True if integer overflow seen */
1513 u8 approx; /* True if non-integer value was input to the sum */
drhdd5baa92002-02-27 19:50:59 +00001514};
1515
1516/*
drha97fdd32006-01-12 22:17:50 +00001517** Routines used to compute the sum, average, and total.
1518**
1519** The SUM() function follows the (broken) SQL standard which means
1520** that it returns NULL if it sums over no inputs. TOTAL returns
1521** 0.0 in that case. In addition, TOTAL always returns a float where
1522** SUM might return an integer if it never encounters a floating point
drhc806d852006-05-11 13:25:39 +00001523** value. TOTAL never fails, but SUM might through an exception if
1524** it overflows an integer.
drhdd5baa92002-02-27 19:50:59 +00001525*/
danielk19770ae8b832004-05-25 12:05:56 +00001526static void sumStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drhdd5baa92002-02-27 19:50:59 +00001527 SumCtx *p;
drh3d1d95e2005-09-08 10:37:01 +00001528 int type;
drh3f219f42005-09-08 19:45:57 +00001529 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001530 UNUSED_PARAMETER(argc);
drh4f26d6c2004-05-26 23:25:30 +00001531 p = sqlite3_aggregate_context(context, sizeof(*p));
drh29d72102006-02-09 22:13:41 +00001532 type = sqlite3_value_numeric_type(argv[0]);
drh3d1d95e2005-09-08 10:37:01 +00001533 if( p && type!=SQLITE_NULL ){
drh739105c2002-05-29 23:22:23 +00001534 p->cnt++;
drh29d72102006-02-09 22:13:41 +00001535 if( type==SQLITE_INTEGER ){
drh8c08e862006-02-11 17:34:00 +00001536 i64 v = sqlite3_value_int64(argv[0]);
1537 p->rSum += v;
drh158b9cb2011-03-05 20:59:46 +00001538 if( (p->approx|p->overflow)==0 && sqlite3AddInt64(&p->iSum, v) ){
drha546ef22018-07-07 20:55:16 +00001539 p->approx = p->overflow = 1;
drh29d72102006-02-09 22:13:41 +00001540 }
1541 }else{
drh8c08e862006-02-11 17:34:00 +00001542 p->rSum += sqlite3_value_double(argv[0]);
drh29d72102006-02-09 22:13:41 +00001543 p->approx = 1;
drh3f219f42005-09-08 19:45:57 +00001544 }
drh739105c2002-05-29 23:22:23 +00001545 }
drhdd5baa92002-02-27 19:50:59 +00001546}
dan67a9b8e2018-06-22 20:51:35 +00001547#ifndef SQLITE_OMIT_WINDOWFUNC
danc3a20c12018-05-23 20:55:37 +00001548static void sumInverse(sqlite3_context *context, int argc, sqlite3_value**argv){
1549 SumCtx *p;
1550 int type;
1551 assert( argc==1 );
1552 UNUSED_PARAMETER(argc);
1553 p = sqlite3_aggregate_context(context, sizeof(*p));
1554 type = sqlite3_value_numeric_type(argv[0]);
drhfd4b7282018-07-07 19:47:21 +00001555 /* p is always non-NULL because sumStep() will have been called first
1556 ** to initialize it */
1557 if( ALWAYS(p) && type!=SQLITE_NULL ){
drha546ef22018-07-07 20:55:16 +00001558 assert( p->cnt>0 );
danc3a20c12018-05-23 20:55:37 +00001559 p->cnt--;
drha546ef22018-07-07 20:55:16 +00001560 assert( type==SQLITE_INTEGER || p->approx );
1561 if( type==SQLITE_INTEGER && p->approx==0 ){
danc3a20c12018-05-23 20:55:37 +00001562 i64 v = sqlite3_value_int64(argv[0]);
1563 p->rSum -= v;
drha546ef22018-07-07 20:55:16 +00001564 p->iSum -= v;
danc3a20c12018-05-23 20:55:37 +00001565 }else{
dand7368292018-07-02 17:45:59 +00001566 p->rSum -= sqlite3_value_double(argv[0]);
danc3a20c12018-05-23 20:55:37 +00001567 }
1568 }
1569}
dan67a9b8e2018-06-22 20:51:35 +00001570#else
1571# define sumInverse 0
1572#endif /* SQLITE_OMIT_WINDOWFUNC */
danielk19770ae8b832004-05-25 12:05:56 +00001573static void sumFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001574 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001575 p = sqlite3_aggregate_context(context, 0);
drhc2bd9132005-09-08 20:37:43 +00001576 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001577 if( p->overflow ){
1578 sqlite3_result_error(context,"integer overflow",-1);
1579 }else if( p->approx ){
1580 sqlite3_result_double(context, p->rSum);
drhc2bd9132005-09-08 20:37:43 +00001581 }else{
drh8c08e862006-02-11 17:34:00 +00001582 sqlite3_result_int64(context, p->iSum);
drhc2bd9132005-09-08 20:37:43 +00001583 }
drh3d1d95e2005-09-08 10:37:01 +00001584 }
drhdd5baa92002-02-27 19:50:59 +00001585}
danielk19770ae8b832004-05-25 12:05:56 +00001586static void avgFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001587 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001588 p = sqlite3_aggregate_context(context, 0);
drh739105c2002-05-29 23:22:23 +00001589 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001590 sqlite3_result_double(context, p->rSum/(double)p->cnt);
drhdd5baa92002-02-27 19:50:59 +00001591 }
1592}
drha97fdd32006-01-12 22:17:50 +00001593static void totalFinalize(sqlite3_context *context){
1594 SumCtx *p;
1595 p = sqlite3_aggregate_context(context, 0);
shanefbd60f82009-02-04 03:59:25 +00001596 /* (double)0 In case of SQLITE_OMIT_FLOATING_POINT... */
1597 sqlite3_result_double(context, p ? p->rSum : (double)0);
drha97fdd32006-01-12 22:17:50 +00001598}
drhdd5baa92002-02-27 19:50:59 +00001599
1600/*
drh0bce8352002-02-28 00:41:10 +00001601** The following structure keeps track of state information for the
1602** count() aggregate function.
1603*/
1604typedef struct CountCtx CountCtx;
1605struct CountCtx {
drhfc6ad392006-02-09 13:38:19 +00001606 i64 n;
dan7262ca92018-07-02 12:07:32 +00001607#ifdef SQLITE_DEBUG
1608 int bInverse; /* True if xInverse() ever called */
1609#endif
drh0bce8352002-02-28 00:41:10 +00001610};
drhdd5baa92002-02-27 19:50:59 +00001611
drh0bce8352002-02-28 00:41:10 +00001612/*
1613** Routines to implement the count() aggregate function.
1614*/
danielk19770ae8b832004-05-25 12:05:56 +00001615static void countStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drh0bce8352002-02-28 00:41:10 +00001616 CountCtx *p;
drh4f26d6c2004-05-26 23:25:30 +00001617 p = sqlite3_aggregate_context(context, sizeof(*p));
drh9c054832004-05-31 18:51:57 +00001618 if( (argc==0 || SQLITE_NULL!=sqlite3_value_type(argv[0])) && p ){
drh0bce8352002-02-28 00:41:10 +00001619 p->n++;
1620 }
drh2e79c3d2009-04-08 23:04:14 +00001621
drhd3264c72009-04-15 13:39:47 +00001622#ifndef SQLITE_OMIT_DEPRECATED
drh2e79c3d2009-04-08 23:04:14 +00001623 /* The sqlite3_aggregate_count() function is deprecated. But just to make
1624 ** sure it still operates correctly, verify that its count agrees with our
1625 ** internal count when using count(*) and when the total count can be
1626 ** expressed as a 32-bit integer. */
dan7262ca92018-07-02 12:07:32 +00001627 assert( argc==1 || p==0 || p->n>0x7fffffff || p->bInverse
drh2e79c3d2009-04-08 23:04:14 +00001628 || p->n==sqlite3_aggregate_count(context) );
drhd3264c72009-04-15 13:39:47 +00001629#endif
drh0bce8352002-02-28 00:41:10 +00001630}
danielk19770ae8b832004-05-25 12:05:56 +00001631static void countFinalize(sqlite3_context *context){
drh0bce8352002-02-28 00:41:10 +00001632 CountCtx *p;
drhabfcea22005-09-06 20:36:48 +00001633 p = sqlite3_aggregate_context(context, 0);
drhfc6ad392006-02-09 13:38:19 +00001634 sqlite3_result_int64(context, p ? p->n : 0);
drh0bce8352002-02-28 00:41:10 +00001635}
dan7262ca92018-07-02 12:07:32 +00001636#ifndef SQLITE_OMIT_WINDOWFUNC
1637static void countInverse(sqlite3_context *ctx, int argc, sqlite3_value **argv){
1638 CountCtx *p;
1639 p = sqlite3_aggregate_context(ctx, sizeof(*p));
drhfd4b7282018-07-07 19:47:21 +00001640 /* p is always non-NULL since countStep() will have been called first */
1641 if( (argc==0 || SQLITE_NULL!=sqlite3_value_type(argv[0])) && ALWAYS(p) ){
dan7262ca92018-07-02 12:07:32 +00001642 p->n--;
1643#ifdef SQLITE_DEBUG
1644 p->bInverse = 1;
1645#endif
1646 }
1647}
dan6b4b8822018-07-02 15:03:50 +00001648#else
1649# define countInverse 0
1650#endif /* SQLITE_OMIT_WINDOWFUNC */
drh0bce8352002-02-28 00:41:10 +00001651
1652/*
drh0bce8352002-02-28 00:41:10 +00001653** Routines to implement min() and max() aggregate functions.
1654*/
danielk197762c14b32008-11-19 09:05:26 +00001655static void minmaxStep(
1656 sqlite3_context *context,
1657 int NotUsed,
1658 sqlite3_value **argv
1659){
danielk197788208052004-05-25 01:13:20 +00001660 Mem *pArg = (Mem *)argv[0];
drh9eb516c2004-07-18 20:52:32 +00001661 Mem *pBest;
danielk197762c14b32008-11-19 09:05:26 +00001662 UNUSED_PARAMETER(NotUsed);
drh9eb516c2004-07-18 20:52:32 +00001663
drh9eb516c2004-07-18 20:52:32 +00001664 pBest = (Mem *)sqlite3_aggregate_context(context, sizeof(*pBest));
danielk19773aeab9e2004-06-24 00:20:04 +00001665 if( !pBest ) return;
drh268380c2004-02-25 13:47:31 +00001666
dan6fb2b542018-06-19 17:13:11 +00001667 if( sqlite3_value_type(pArg)==SQLITE_NULL ){
drh94a6d992012-02-02 18:42:09 +00001668 if( pBest->flags ) sqlite3SkipAccumulatorLoad(context);
1669 }else if( pBest->flags ){
drh9eb516c2004-07-18 20:52:32 +00001670 int max;
1671 int cmp;
danielk1977dc1bdc42004-06-11 10:51:27 +00001672 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk19777e18c252004-05-25 11:47:24 +00001673 /* This step function is used for both the min() and max() aggregates,
1674 ** the only difference between the two being that the sense of the
1675 ** comparison is inverted. For the max() aggregate, the
1676 ** sqlite3_user_data() function returns (void *)-1. For min() it
1677 ** returns (void *)db, where db is the sqlite3* database pointer.
1678 ** Therefore the next statement sets variable 'max' to 1 for the max()
1679 ** aggregate, or 0 for min().
1680 */
drh309b3382007-03-17 17:52:42 +00001681 max = sqlite3_user_data(context)!=0;
danielk1977dc1bdc42004-06-11 10:51:27 +00001682 cmp = sqlite3MemCompare(pBest, pArg, pColl);
danielk197788208052004-05-25 01:13:20 +00001683 if( (max && cmp<0) || (!max && cmp>0) ){
drhb21c8cd2007-08-21 19:33:56 +00001684 sqlite3VdbeMemCopy(pBest, pArg);
drh7a957892012-02-02 17:35:43 +00001685 }else{
1686 sqlite3SkipAccumulatorLoad(context);
danielk197788208052004-05-25 01:13:20 +00001687 }
drh268380c2004-02-25 13:47:31 +00001688 }else{
drh035e5632014-09-16 14:16:31 +00001689 pBest->db = sqlite3_context_db_handle(context);
drhb21c8cd2007-08-21 19:33:56 +00001690 sqlite3VdbeMemCopy(pBest, pArg);
drh0bce8352002-02-28 00:41:10 +00001691 }
1692}
dan6fb2b542018-06-19 17:13:11 +00001693static void minMaxValueFinalize(sqlite3_context *context, int bValue){
danielk197788208052004-05-25 01:13:20 +00001694 sqlite3_value *pRes;
drhabfcea22005-09-06 20:36:48 +00001695 pRes = (sqlite3_value *)sqlite3_aggregate_context(context, 0);
1696 if( pRes ){
drh94a6d992012-02-02 18:42:09 +00001697 if( pRes->flags ){
drhabfcea22005-09-06 20:36:48 +00001698 sqlite3_result_value(context, pRes);
1699 }
dan6fb2b542018-06-19 17:13:11 +00001700 if( bValue==0 ) sqlite3VdbeMemRelease(pRes);
drh0bce8352002-02-28 00:41:10 +00001701 }
1702}
dan67a9b8e2018-06-22 20:51:35 +00001703#ifndef SQLITE_OMIT_WINDOWFUNC
dan6fb2b542018-06-19 17:13:11 +00001704static void minMaxValue(sqlite3_context *context){
drhc7bf5712018-07-09 22:49:01 +00001705 minMaxValueFinalize(context, 1);
dan6fb2b542018-06-19 17:13:11 +00001706}
dan67a9b8e2018-06-22 20:51:35 +00001707#else
1708# define minMaxValue 0
1709#endif /* SQLITE_OMIT_WINDOWFUNC */
dan6fb2b542018-06-19 17:13:11 +00001710static void minMaxFinalize(sqlite3_context *context){
drhc7bf5712018-07-09 22:49:01 +00001711 minMaxValueFinalize(context, 0);
dan6fb2b542018-06-19 17:13:11 +00001712}
drhdd5baa92002-02-27 19:50:59 +00001713
drhb0689692007-11-01 17:38:30 +00001714/*
1715** group_concat(EXPR, ?SEPARATOR?)
1716*/
1717static void groupConcatStep(
1718 sqlite3_context *context,
1719 int argc,
1720 sqlite3_value **argv
1721){
1722 const char *zVal;
drhade86482007-11-28 22:36:40 +00001723 StrAccum *pAccum;
drhb0689692007-11-01 17:38:30 +00001724 const char *zSep;
drh07d31172009-02-02 21:57:05 +00001725 int nVal, nSep;
1726 assert( argc==1 || argc==2 );
1727 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drhade86482007-11-28 22:36:40 +00001728 pAccum = (StrAccum*)sqlite3_aggregate_context(context, sizeof(*pAccum));
1729
1730 if( pAccum ){
drhbb4957f2008-03-20 14:03:29 +00001731 sqlite3 *db = sqlite3_context_db_handle(context);
dane3bf6322018-06-08 20:58:27 +00001732 int firstTerm = pAccum->mxAlloc==0;
drhbb4957f2008-03-20 14:03:29 +00001733 pAccum->mxAlloc = db->aLimit[SQLITE_LIMIT_LENGTH];
drh8bfd7192009-06-19 16:44:41 +00001734 if( !firstTerm ){
drh07d31172009-02-02 21:57:05 +00001735 if( argc==2 ){
1736 zSep = (char*)sqlite3_value_text(argv[1]);
1737 nSep = sqlite3_value_bytes(argv[1]);
drhb0689692007-11-01 17:38:30 +00001738 }else{
1739 zSep = ",";
drhade86482007-11-28 22:36:40 +00001740 nSep = 1;
drhb0689692007-11-01 17:38:30 +00001741 }
drh0cdbe1a2018-05-09 13:46:26 +00001742 if( zSep ) sqlite3_str_append(pAccum, zSep, nSep);
drhb0689692007-11-01 17:38:30 +00001743 }
drh07d31172009-02-02 21:57:05 +00001744 zVal = (char*)sqlite3_value_text(argv[0]);
1745 nVal = sqlite3_value_bytes(argv[0]);
drh0cdbe1a2018-05-09 13:46:26 +00001746 if( zVal ) sqlite3_str_append(pAccum, zVal, nVal);
drhb0689692007-11-01 17:38:30 +00001747 }
1748}
dan67a9b8e2018-06-22 20:51:35 +00001749#ifndef SQLITE_OMIT_WINDOWFUNC
dan03854d22018-06-08 11:45:28 +00001750static void groupConcatInverse(
1751 sqlite3_context *context,
1752 int argc,
1753 sqlite3_value **argv
1754){
1755 int n;
dan03854d22018-06-08 11:45:28 +00001756 StrAccum *pAccum;
drhc7bf5712018-07-09 22:49:01 +00001757 assert( argc==1 || argc==2 );
dan03854d22018-06-08 11:45:28 +00001758 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
1759 pAccum = (StrAccum*)sqlite3_aggregate_context(context, sizeof(*pAccum));
drhfd4b7282018-07-07 19:47:21 +00001760 /* pAccum is always non-NULL since groupConcatStep() will have always
1761 ** run frist to initialize it */
1762 if( ALWAYS(pAccum) ){
dan03854d22018-06-08 11:45:28 +00001763 n = sqlite3_value_bytes(argv[0]);
1764 if( argc==2 ){
1765 n += sqlite3_value_bytes(argv[1]);
dan683b0ff2018-07-05 18:19:29 +00001766 }else{
1767 n++;
dan03854d22018-06-08 11:45:28 +00001768 }
drhc7bf5712018-07-09 22:49:01 +00001769 if( n>=(int)pAccum->nChar ){
dan03854d22018-06-08 11:45:28 +00001770 pAccum->nChar = 0;
1771 }else{
1772 pAccum->nChar -= n;
1773 memmove(pAccum->zText, &pAccum->zText[n], pAccum->nChar);
1774 }
dane3bf6322018-06-08 20:58:27 +00001775 if( pAccum->nChar==0 ) pAccum->mxAlloc = 0;
dan03854d22018-06-08 11:45:28 +00001776 }
1777}
dan67a9b8e2018-06-22 20:51:35 +00001778#else
1779# define groupConcatInverse 0
1780#endif /* SQLITE_OMIT_WINDOWFUNC */
drhb0689692007-11-01 17:38:30 +00001781static void groupConcatFinalize(sqlite3_context *context){
drhade86482007-11-28 22:36:40 +00001782 StrAccum *pAccum;
1783 pAccum = sqlite3_aggregate_context(context, 0);
1784 if( pAccum ){
drh0cdbe1a2018-05-09 13:46:26 +00001785 if( pAccum->accError==SQLITE_TOOBIG ){
drhade86482007-11-28 22:36:40 +00001786 sqlite3_result_error_toobig(context);
drh0cdbe1a2018-05-09 13:46:26 +00001787 }else if( pAccum->accError==SQLITE_NOMEM ){
drhade86482007-11-28 22:36:40 +00001788 sqlite3_result_error_nomem(context);
1789 }else{
1790 sqlite3_result_text(context, sqlite3StrAccumFinish(pAccum), -1,
1791 sqlite3_free);
1792 }
drhb0689692007-11-01 17:38:30 +00001793 }
1794}
dan67a9b8e2018-06-22 20:51:35 +00001795#ifndef SQLITE_OMIT_WINDOWFUNC
dane2f781b2018-05-17 19:24:08 +00001796static void groupConcatValue(sqlite3_context *context){
1797 sqlite3_str *pAccum;
1798 pAccum = (sqlite3_str*)sqlite3_aggregate_context(context, 0);
1799 if( pAccum ){
1800 if( pAccum->accError==SQLITE_TOOBIG ){
1801 sqlite3_result_error_toobig(context);
1802 }else if( pAccum->accError==SQLITE_NOMEM ){
1803 sqlite3_result_error_nomem(context);
1804 }else{
1805 const char *zText = sqlite3_str_value(pAccum);
1806 sqlite3_result_text(context, zText, -1, SQLITE_TRANSIENT);
1807 }
1808 }
1809}
dan67a9b8e2018-06-22 20:51:35 +00001810#else
1811# define groupConcatValue 0
1812#endif /* SQLITE_OMIT_WINDOWFUNC */
drh4e5ffc52004-08-31 00:52:37 +00001813
drhd3a149e2002-02-24 17:12:53 +00001814/*
drha4741842010-04-25 20:58:37 +00001815** This routine does per-connection function registration. Most
1816** of the built-in functions above are part of the global function set.
1817** This routine only deals with those that are not global.
drhdc04c582002-02-24 01:55:15 +00001818*/
drh80738d92016-02-15 00:34:16 +00001819void sqlite3RegisterPerConnectionBuiltinFunctions(sqlite3 *db){
drha4741842010-04-25 20:58:37 +00001820 int rc = sqlite3_overload_function(db, "MATCH", 2);
1821 assert( rc==SQLITE_NOMEM || rc==SQLITE_OK );
1822 if( rc==SQLITE_NOMEM ){
drh4a642b62016-02-05 01:55:27 +00001823 sqlite3OomFault(db);
danielk1977832a58a2007-06-22 15:21:15 +00001824 }
drh55ef4d92005-08-14 01:20:37 +00001825}
1826
1827/*
drh08652b52019-05-08 17:27:18 +00001828** Re-register the built-in LIKE functions. The caseSensitive
drh55ef4d92005-08-14 01:20:37 +00001829** parameter determines whether or not the LIKE operator is case
drh08652b52019-05-08 17:27:18 +00001830** sensitive.
drh55ef4d92005-08-14 01:20:37 +00001831*/
1832void sqlite3RegisterLikeFunctions(sqlite3 *db, int caseSensitive){
1833 struct compareInfo *pInfo;
drhea5c0402019-05-08 19:32:33 +00001834 int flags;
drh55ef4d92005-08-14 01:20:37 +00001835 if( caseSensitive ){
1836 pInfo = (struct compareInfo*)&likeInfoAlt;
drhea5c0402019-05-08 19:32:33 +00001837 flags = SQLITE_FUNC_LIKE | SQLITE_FUNC_CASE;
drh55ef4d92005-08-14 01:20:37 +00001838 }else{
1839 pInfo = (struct compareInfo*)&likeInfoNorm;
drhea5c0402019-05-08 19:32:33 +00001840 flags = SQLITE_FUNC_LIKE;
drh55ef4d92005-08-14 01:20:37 +00001841 }
dan660af932018-06-18 16:55:22 +00001842 sqlite3CreateFunc(db, "like", 2, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0, 0, 0);
1843 sqlite3CreateFunc(db, "like", 3, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0, 0, 0);
drhea5c0402019-05-08 19:32:33 +00001844 sqlite3FindFunction(db, "like", 2, SQLITE_UTF8, 0)->funcFlags |= flags;
1845 sqlite3FindFunction(db, "like", 3, SQLITE_UTF8, 0)->funcFlags |= flags;
drh55ef4d92005-08-14 01:20:37 +00001846}
1847
1848/*
1849** pExpr points to an expression which implements a function. If
1850** it is appropriate to apply the LIKE optimization to that function
drh1d42ea72017-07-27 20:24:29 +00001851** then set aWc[0] through aWc[2] to the wildcard characters and the
1852** escape character and then return TRUE. If the function is not a
1853** LIKE-style function then return FALSE.
1854**
1855** The expression "a LIKE b ESCAPE c" is only considered a valid LIKE
1856** operator if c is a string literal that is exactly one byte in length.
1857** That one byte is stored in aWc[3]. aWc[3] is set to zero if there is
1858** no ESCAPE clause.
drh16897072015-03-07 00:57:37 +00001859**
1860** *pIsNocase is set to true if uppercase and lowercase are equivalent for
1861** the function (default for LIKE). If the function makes the distinction
1862** between uppercase and lowercase (as does GLOB) then *pIsNocase is set to
1863** false.
drh55ef4d92005-08-14 01:20:37 +00001864*/
drhd64fe2f2005-08-28 17:00:23 +00001865int sqlite3IsLikeFunction(sqlite3 *db, Expr *pExpr, int *pIsNocase, char *aWc){
drh55ef4d92005-08-14 01:20:37 +00001866 FuncDef *pDef;
drh1d42ea72017-07-27 20:24:29 +00001867 int nExpr;
1868 if( pExpr->op!=TK_FUNCTION || !pExpr->x.pList ){
drh55ef4d92005-08-14 01:20:37 +00001869 return 0;
1870 }
danielk19776ab3a2e2009-02-19 14:39:25 +00001871 assert( !ExprHasProperty(pExpr, EP_xIsSelect) );
drh1d42ea72017-07-27 20:24:29 +00001872 nExpr = pExpr->x.pList->nExpr;
1873 pDef = sqlite3FindFunction(db, pExpr->u.zToken, nExpr, SQLITE_UTF8, 0);
drh78b52202020-01-22 23:08:19 +00001874#ifdef SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION
1875 if( pDef==0 ) return 0;
1876#endif
drhd36e1042013-09-06 13:10:12 +00001877 if( NEVER(pDef==0) || (pDef->funcFlags & SQLITE_FUNC_LIKE)==0 ){
drh55ef4d92005-08-14 01:20:37 +00001878 return 0;
1879 }
1880
1881 /* The memcpy() statement assumes that the wildcard characters are
1882 ** the first three statements in the compareInfo structure. The
1883 ** asserts() that follow verify that assumption
1884 */
1885 memcpy(aWc, pDef->pUserData, 3);
1886 assert( (char*)&likeInfoAlt == (char*)&likeInfoAlt.matchAll );
1887 assert( &((char*)&likeInfoAlt)[1] == (char*)&likeInfoAlt.matchOne );
1888 assert( &((char*)&likeInfoAlt)[2] == (char*)&likeInfoAlt.matchSet );
drh589c7872020-03-19 18:13:28 +00001889
1890 if( nExpr<3 ){
1891 aWc[3] = 0;
1892 }else{
1893 Expr *pEscape = pExpr->x.pList->a[2].pExpr;
1894 char *zEscape;
1895 if( pEscape->op!=TK_STRING ) return 0;
1896 zEscape = pEscape->u.zToken;
1897 if( zEscape[0]==0 || zEscape[1]!=0 ) return 0;
1898 if( zEscape[0]==aWc[0] ) return 0;
1899 if( zEscape[0]==aWc[1] ) return 0;
1900 aWc[3] = zEscape[0];
1901 }
1902
drhd36e1042013-09-06 13:10:12 +00001903 *pIsNocase = (pDef->funcFlags & SQLITE_FUNC_CASE)==0;
drh55ef4d92005-08-14 01:20:37 +00001904 return 1;
drhdc04c582002-02-24 01:55:15 +00001905}
danielk19778c0a7912008-08-20 14:49:23 +00001906
drh70a8ca32008-08-21 18:49:27 +00001907/*
peter.d.reid60ec9142014-09-06 16:39:46 +00001908** All of the FuncDef structures in the aBuiltinFunc[] array above
drh777c5382008-08-21 20:21:34 +00001909** to the global function hash table. This occurs at start-time (as
1910** a consequence of calling sqlite3_initialize()).
1911**
1912** After this routine runs
drh70a8ca32008-08-21 18:49:27 +00001913*/
drh80738d92016-02-15 00:34:16 +00001914void sqlite3RegisterBuiltinFunctions(void){
danielk197793ce7412008-09-01 19:14:02 +00001915 /*
1916 ** The following array holds FuncDef structures for all of the functions
1917 ** defined in this file.
1918 **
1919 ** The array cannot be constant since changes are made to the
1920 ** FuncDef.pHash elements at start-time. The elements of this array
1921 ** are read-only after initialization is complete.
drh80738d92016-02-15 00:34:16 +00001922 **
1923 ** For peak efficiency, put the most frequently used function last.
danielk197793ce7412008-09-01 19:14:02 +00001924 */
drh80738d92016-02-15 00:34:16 +00001925 static FuncDef aBuiltinFunc[] = {
drh171c50e2020-01-01 15:43:30 +00001926/***** Functions only available with SQLITE_TESTCTRL_INTERNAL_FUNCTIONS *****/
1927 TEST_FUNC(implies_nonnull_row, 2, INLINEFUNC_implies_nonnull_row, 0),
1928 TEST_FUNC(expr_compare, 2, INLINEFUNC_expr_compare, 0),
1929 TEST_FUNC(expr_implies_expr, 2, INLINEFUNC_expr_implies_expr, 0),
1930#ifdef SQLITE_DEBUG
1931 TEST_FUNC(affinity, 1, INLINEFUNC_affinity, 0),
1932#endif
1933/***** Regular functions *****/
drh80738d92016-02-15 00:34:16 +00001934#ifdef SQLITE_SOUNDEX
1935 FUNCTION(soundex, 1, 0, 0, soundexFunc ),
1936#endif
1937#ifndef SQLITE_OMIT_LOAD_EXTENSION
drh64de2a52019-12-31 18:39:23 +00001938 SFUNCTION(load_extension, 1, 0, 0, loadExt ),
1939 SFUNCTION(load_extension, 2, 0, 0, loadExt ),
drh80738d92016-02-15 00:34:16 +00001940#endif
1941#if SQLITE_USER_AUTHENTICATION
1942 FUNCTION(sqlite_crypt, 2, 0, 0, sqlite3CryptFunc ),
1943#endif
1944#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
1945 DFUNCTION(sqlite_compileoption_used,1, 0, 0, compileoptionusedFunc ),
1946 DFUNCTION(sqlite_compileoption_get, 1, 0, 0, compileoptiongetFunc ),
1947#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
drh25c42962020-01-01 13:55:08 +00001948 INLINE_FUNC(unlikely, 1, INLINEFUNC_unlikely, SQLITE_FUNC_UNLIKELY),
1949 INLINE_FUNC(likelihood, 2, INLINEFUNC_unlikely, SQLITE_FUNC_UNLIKELY),
1950 INLINE_FUNC(likely, 1, INLINEFUNC_unlikely, SQLITE_FUNC_UNLIKELY),
drh092457b2017-12-29 15:04:49 +00001951#ifdef SQLITE_ENABLE_OFFSET_SQL_FUNC
drh35100fb2018-01-04 19:20:37 +00001952 FUNCTION2(sqlite_offset, 1, 0, 0, noopFunc, SQLITE_FUNC_OFFSET|
drh2fc865c2017-12-16 20:20:37 +00001953 SQLITE_FUNC_TYPEOF),
drh092457b2017-12-29 15:04:49 +00001954#endif
danielk197793ce7412008-09-01 19:14:02 +00001955 FUNCTION(ltrim, 1, 1, 0, trimFunc ),
1956 FUNCTION(ltrim, 2, 1, 0, trimFunc ),
1957 FUNCTION(rtrim, 1, 2, 0, trimFunc ),
1958 FUNCTION(rtrim, 2, 2, 0, trimFunc ),
1959 FUNCTION(trim, 1, 3, 0, trimFunc ),
1960 FUNCTION(trim, 2, 3, 0, trimFunc ),
1961 FUNCTION(min, -1, 0, 1, minmaxFunc ),
1962 FUNCTION(min, 0, 0, 1, 0 ),
dan6fb2b542018-06-19 17:13:11 +00001963 WAGGREGATE(min, 1, 0, 1, minmaxStep, minMaxFinalize, minMaxValue, 0,
drh9588ad92014-09-15 14:46:02 +00001964 SQLITE_FUNC_MINMAX ),
danielk197793ce7412008-09-01 19:14:02 +00001965 FUNCTION(max, -1, 1, 1, minmaxFunc ),
1966 FUNCTION(max, 0, 1, 1, 0 ),
dan6fb2b542018-06-19 17:13:11 +00001967 WAGGREGATE(max, 1, 1, 1, minmaxStep, minMaxFinalize, minMaxValue, 0,
drh9588ad92014-09-15 14:46:02 +00001968 SQLITE_FUNC_MINMAX ),
drha748fdc2012-03-28 01:34:47 +00001969 FUNCTION2(typeof, 1, 0, 0, typeofFunc, SQLITE_FUNC_TYPEOF),
1970 FUNCTION2(length, 1, 0, 0, lengthFunc, SQLITE_FUNC_LENGTH),
drhd55e0722012-10-25 03:07:29 +00001971 FUNCTION(instr, 2, 0, 0, instrFunc ),
shearer3a8847a2020-09-25 14:26:49 +00001972 FUNCTION(printf, -1, 0, 0, printfFunc ),
drhd495d8c2013-02-22 19:34:25 +00001973 FUNCTION(unicode, 1, 0, 0, unicodeFunc ),
1974 FUNCTION(char, -1, 0, 0, charFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001975 FUNCTION(abs, 1, 0, 0, absFunc ),
shanefbd60f82009-02-04 03:59:25 +00001976#ifndef SQLITE_OMIT_FLOATING_POINT
danielk197793ce7412008-09-01 19:14:02 +00001977 FUNCTION(round, 1, 0, 0, roundFunc ),
1978 FUNCTION(round, 2, 0, 0, roundFunc ),
shanefbd60f82009-02-04 03:59:25 +00001979#endif
danielk197793ce7412008-09-01 19:14:02 +00001980 FUNCTION(upper, 1, 0, 0, upperFunc ),
1981 FUNCTION(lower, 1, 0, 0, lowerFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001982 FUNCTION(hex, 1, 0, 0, hexFunc ),
drhffe421c2020-05-13 17:26:38 +00001983 INLINE_FUNC(ifnull, 2, INLINEFUNC_coalesce, 0 ),
drhb1fba282013-11-21 14:33:48 +00001984 VFUNCTION(random, 0, 0, 0, randomFunc ),
1985 VFUNCTION(randomblob, 1, 0, 0, randomBlob ),
danielk197793ce7412008-09-01 19:14:02 +00001986 FUNCTION(nullif, 2, 0, 1, nullifFunc ),
drh03bf26d2015-08-31 21:16:36 +00001987 DFUNCTION(sqlite_version, 0, 0, 0, versionFunc ),
1988 DFUNCTION(sqlite_source_id, 0, 0, 0, sourceidFunc ),
drh840561f2011-04-27 18:08:42 +00001989 FUNCTION(sqlite_log, 2, 0, 0, errlogFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001990 FUNCTION(quote, 1, 0, 0, quoteFunc ),
drhb1fba282013-11-21 14:33:48 +00001991 VFUNCTION(last_insert_rowid, 0, 0, 0, last_insert_rowid),
1992 VFUNCTION(changes, 0, 0, 0, changes ),
1993 VFUNCTION(total_changes, 0, 0, 0, total_changes ),
danielk197793ce7412008-09-01 19:14:02 +00001994 FUNCTION(replace, 3, 0, 0, replaceFunc ),
1995 FUNCTION(zeroblob, 1, 0, 0, zeroblobFunc ),
drh80738d92016-02-15 00:34:16 +00001996 FUNCTION(substr, 2, 0, 0, substrFunc ),
1997 FUNCTION(substr, 3, 0, 0, substrFunc ),
dan6fb2b542018-06-19 17:13:11 +00001998 WAGGREGATE(sum, 1,0,0, sumStep, sumFinalize, sumFinalize, sumInverse, 0),
1999 WAGGREGATE(total, 1,0,0, sumStep,totalFinalize,totalFinalize,sumInverse, 0),
2000 WAGGREGATE(avg, 1,0,0, sumStep, avgFinalize, avgFinalize, sumInverse, 0),
dan7262ca92018-07-02 12:07:32 +00002001 WAGGREGATE(count, 0,0,0, countStep,
2002 countFinalize, countFinalize, countInverse, SQLITE_FUNC_COUNT ),
2003 WAGGREGATE(count, 1,0,0, countStep,
2004 countFinalize, countFinalize, countInverse, 0 ),
dan03854d22018-06-08 11:45:28 +00002005 WAGGREGATE(group_concat, 1, 0, 0, groupConcatStep,
dan6fb2b542018-06-19 17:13:11 +00002006 groupConcatFinalize, groupConcatValue, groupConcatInverse, 0),
dan03854d22018-06-08 11:45:28 +00002007 WAGGREGATE(group_concat, 2, 0, 0, groupConcatStep,
dan6fb2b542018-06-19 17:13:11 +00002008 groupConcatFinalize, groupConcatValue, groupConcatInverse, 0),
danielk197793ce7412008-09-01 19:14:02 +00002009
2010 LIKEFUNC(glob, 2, &globInfo, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
drhcc153132016-08-04 12:35:17 +00002011#ifdef SQLITE_CASE_SENSITIVE_LIKE
danielk197793ce7412008-09-01 19:14:02 +00002012 LIKEFUNC(like, 2, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
2013 LIKEFUNC(like, 3, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
drhcc153132016-08-04 12:35:17 +00002014#else
danielk197793ce7412008-09-01 19:14:02 +00002015 LIKEFUNC(like, 2, &likeInfoNorm, SQLITE_FUNC_LIKE),
2016 LIKEFUNC(like, 3, &likeInfoNorm, SQLITE_FUNC_LIKE),
drhcc153132016-08-04 12:35:17 +00002017#endif
2018#ifdef SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION
2019 FUNCTION(unknown, -1, 0, 0, unknownFunc ),
2020#endif
drh80738d92016-02-15 00:34:16 +00002021 FUNCTION(coalesce, 1, 0, 0, 0 ),
2022 FUNCTION(coalesce, 0, 0, 0, 0 ),
drhffe421c2020-05-13 17:26:38 +00002023 INLINE_FUNC(coalesce, -1, INLINEFUNC_coalesce, 0 ),
drh3c0e6062020-05-13 18:03:34 +00002024 INLINE_FUNC(iif, 3, INLINEFUNC_iif, 0 ),
danielk197793ce7412008-09-01 19:14:02 +00002025 };
drh545f5872010-04-24 14:02:59 +00002026#ifndef SQLITE_OMIT_ALTERTABLE
2027 sqlite3AlterFunctions();
2028#endif
dandfa552f2018-06-02 21:04:28 +00002029 sqlite3WindowFunctions();
drh80738d92016-02-15 00:34:16 +00002030 sqlite3RegisterDateTimeFunctions();
2031 sqlite3InsertBuiltinFuncs(aBuiltinFunc, ArraySize(aBuiltinFunc));
2032
2033#if 0 /* Enable to print out how the built-in functions are hashed */
2034 {
2035 int i;
2036 FuncDef *p;
2037 for(i=0; i<SQLITE_FUNC_HASH_SZ; i++){
2038 printf("FUNC-HASH %02d:", i);
2039 for(p=sqlite3BuiltinFunctions.a[i]; p; p=p->u.pHash){
2040 int n = sqlite3Strlen30(p->zName);
2041 int h = p->zName[0] + n;
2042 printf(" %s(%d)", p->zName, h);
2043 }
2044 printf("\n");
2045 }
2046 }
2047#endif
drh70a8ca32008-08-21 18:49:27 +00002048}