blob: 289b9964eb084b50b0ccdda93c7181fbd081a9cb [file] [log] [blame]
drhdc04c582002-02-24 01:55:15 +00001/*
2** 2002 February 23
3**
4** The author disclaims copyright to this source code. In place of
5** a legal notice, here is a blessing:
6**
7** May you do good and not evil.
8** May you find forgiveness for yourself and forgive others.
9** May you share freely, never taking more than you give.
10**
11*************************************************************************
peter.d.reid60ec9142014-09-06 16:39:46 +000012** This file contains the C-language implementations for many of the SQL
drhede7ae32014-08-06 11:58:40 +000013** functions of SQLite. (Some function, and in particular the date and
14** time functions, are implemented separately.)
drhdc04c582002-02-24 01:55:15 +000015*/
drhb659e9b2005-01-28 01:29:08 +000016#include "sqliteInt.h"
drhd3a149e2002-02-24 17:12:53 +000017#include <stdlib.h>
drh0bce8352002-02-28 00:41:10 +000018#include <assert.h>
drhef9f7192020-01-17 19:14:08 +000019#ifndef SQLITE_OMIT_FLOATING_POINT
drh05d7bfd2019-05-10 12:06:47 +000020#include <math.h>
drhef9f7192020-01-17 19:14:08 +000021#endif
danielk197788208052004-05-25 01:13:20 +000022#include "vdbeInt.h"
drh0bce8352002-02-28 00:41:10 +000023
drh55ef4d92005-08-14 01:20:37 +000024/*
25** Return the collating function associated with a function.
26*/
danielk1977dc1bdc42004-06-11 10:51:27 +000027static CollSeq *sqlite3GetFuncCollSeq(sqlite3_context *context){
drha9e03b12015-03-12 06:46:52 +000028 VdbeOp *pOp;
29 assert( context->pVdbe!=0 );
30 pOp = &context->pVdbe->aOp[context->iOp-1];
drha15cc472014-09-25 13:17:30 +000031 assert( pOp->opcode==OP_CollSeq );
32 assert( pOp->p4type==P4_COLLSEQ );
33 return pOp->p4.pColl;
danielk1977dc1bdc42004-06-11 10:51:27 +000034}
35
drh0bce8352002-02-28 00:41:10 +000036/*
drh7a957892012-02-02 17:35:43 +000037** Indicate that the accumulator load should be skipped on this
38** iteration of the aggregate loop.
39*/
40static void sqlite3SkipAccumulatorLoad(sqlite3_context *context){
drh21d59782018-01-23 19:24:54 +000041 assert( context->isError<=0 );
42 context->isError = -1;
drh7a957892012-02-02 17:35:43 +000043 context->skipFlag = 1;
44}
45
46/*
drh0bce8352002-02-28 00:41:10 +000047** Implementation of the non-aggregate min() and max() functions
48*/
drhf9b596e2004-05-26 16:54:42 +000049static void minmaxFunc(
50 sqlite3_context *context,
51 int argc,
52 sqlite3_value **argv
53){
drh0bce8352002-02-28 00:41:10 +000054 int i;
drh268380c2004-02-25 13:47:31 +000055 int mask; /* 0 for min() or 0xffffffff for max() */
drhf9b596e2004-05-26 16:54:42 +000056 int iBest;
danielk1977dc1bdc42004-06-11 10:51:27 +000057 CollSeq *pColl;
drh0bce8352002-02-28 00:41:10 +000058
drh65595cd2009-02-02 16:32:55 +000059 assert( argc>1 );
drhc44af712004-09-02 15:53:56 +000060 mask = sqlite3_user_data(context)==0 ? 0 : -1;
danielk1977dc1bdc42004-06-11 10:51:27 +000061 pColl = sqlite3GetFuncCollSeq(context);
62 assert( pColl );
danielk1977c572ef72004-05-27 09:28:41 +000063 assert( mask==-1 || mask==0 );
drhf9b596e2004-05-26 16:54:42 +000064 iBest = 0;
drh9c054832004-05-31 18:51:57 +000065 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drhf9b596e2004-05-26 16:54:42 +000066 for(i=1; i<argc; i++){
drh9c054832004-05-31 18:51:57 +000067 if( sqlite3_value_type(argv[i])==SQLITE_NULL ) return;
danielk1977dc1bdc42004-06-11 10:51:27 +000068 if( (sqlite3MemCompare(argv[iBest], argv[i], pColl)^mask)>=0 ){
drh65595cd2009-02-02 16:32:55 +000069 testcase( mask==0 );
drhf9b596e2004-05-26 16:54:42 +000070 iBest = i;
drh0bce8352002-02-28 00:41:10 +000071 }
72 }
drhf4479502004-05-27 03:12:53 +000073 sqlite3_result_value(context, argv[iBest]);
drh0bce8352002-02-28 00:41:10 +000074}
drh0bce8352002-02-28 00:41:10 +000075
drh268380c2004-02-25 13:47:31 +000076/*
77** Return the type of the argument.
78*/
drhf9b596e2004-05-26 16:54:42 +000079static void typeofFunc(
80 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +000081 int NotUsed,
drhf9b596e2004-05-26 16:54:42 +000082 sqlite3_value **argv
83){
drh9d8e4012017-07-06 13:51:50 +000084 static const char *azType[] = { "integer", "real", "text", "blob", "null" };
85 int i = sqlite3_value_type(argv[0]) - 1;
danielk197762c14b32008-11-19 09:05:26 +000086 UNUSED_PARAMETER(NotUsed);
drh9d8e4012017-07-06 13:51:50 +000087 assert( i>=0 && i<ArraySize(azType) );
88 assert( SQLITE_INTEGER==1 );
89 assert( SQLITE_FLOAT==2 );
90 assert( SQLITE_TEXT==3 );
91 assert( SQLITE_BLOB==4 );
92 assert( SQLITE_NULL==5 );
drh3cef3642017-07-14 19:22:08 +000093 /* EVIDENCE-OF: R-01470-60482 The sqlite3_value_type(V) interface returns
94 ** the datatype code for the initial datatype of the sqlite3_value object
95 ** V. The returned value is one of SQLITE_INTEGER, SQLITE_FLOAT,
96 ** SQLITE_TEXT, SQLITE_BLOB, or SQLITE_NULL. */
drh9d8e4012017-07-06 13:51:50 +000097 sqlite3_result_text(context, azType[i], -1, SQLITE_STATIC);
drh0bce8352002-02-28 00:41:10 +000098}
99
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
260/*
drha5c14162013-12-17 15:03:06 +0000261** 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;
drhc0490572015-05-02 11:45:53 +0000272 sqlite3 *db = sqlite3_context_db_handle(context);
drha5c14162013-12-17 15:03:06 +0000273
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;
drhc0490572015-05-02 11:45:53 +0000278 sqlite3StrAccumInit(&str, db, 0, 0, db->aLimit[SQLITE_LIMIT_LENGTH]);
drh5f4a6862016-01-30 12:50:25 +0000279 str.printfFlags = SQLITE_PRINTF_SQLFUNC;
drh0cdbe1a2018-05-09 13:46:26 +0000280 sqlite3_str_appendf(&str, zFormat, &x);
drha5c14162013-12-17 15:03:06 +0000281 n = str.nChar;
282 sqlite3_result_text(context, sqlite3StrAccumFinish(&str), n,
283 SQLITE_DYNAMIC);
284 }
285}
286
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;
dan9b5c67f2018-11-30 16:26:39 +0000514 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000515 UNUSED_PARAMETER2(NotUsed, NotUsed2);
dan9b5c67f2018-11-30 16:26:39 +0000516 sqlite3FastRandomness(&db->sPrng, sizeof(r), &r);
drh3034e3d2009-04-02 14:05:21 +0000517 if( r<0 ){
518 /* We need to prevent a random number of 0x8000000000000000
519 ** (or -9223372036854775808) since when you do abs() of that
520 ** number of you get the same value back again. To do this
521 ** in a way that is testable, mask the sign bit off of negative
522 ** values, resulting in a positive value. Then take the
523 ** 2s complement of that positive value. The end result can
524 ** therefore be no less than -9223372036854775807.
525 */
drhaf8001b2012-02-11 19:53:24 +0000526 r = -(r & LARGEST_INT64);
drh3034e3d2009-04-02 14:05:21 +0000527 }
drh52fc8492006-02-23 21:43:55 +0000528 sqlite3_result_int64(context, r);
drhf9ffac92002-03-02 19:00:31 +0000529}
530
531/*
drh137c7282007-01-29 17:58:28 +0000532** Implementation of randomblob(N). Return a random blob
533** that is N bytes long.
drh63cf66f2007-01-29 15:50:05 +0000534*/
drh137c7282007-01-29 17:58:28 +0000535static void randomBlob(
drh63cf66f2007-01-29 15:50:05 +0000536 sqlite3_context *context,
537 int argc,
538 sqlite3_value **argv
539){
drh3cb79202019-01-18 14:53:15 +0000540 sqlite3_int64 n;
drh137c7282007-01-29 17:58:28 +0000541 unsigned char *p;
dan9b5c67f2018-11-30 16:26:39 +0000542 sqlite3 *db = sqlite3_context_db_handle(context);
drh63cf66f2007-01-29 15:50:05 +0000543 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000544 UNUSED_PARAMETER(argc);
drh3cb79202019-01-18 14:53:15 +0000545 n = sqlite3_value_int64(argv[0]);
drh023ae032007-05-08 12:12:16 +0000546 if( n<1 ){
547 n = 1;
548 }
danielk1977a1644fd2007-08-29 12:31:25 +0000549 p = contextMalloc(context, n);
drh02d85832007-05-07 19:31:15 +0000550 if( p ){
dan9b5c67f2018-11-30 16:26:39 +0000551 sqlite3FastRandomness(&db->sPrng, n, p);
drh17435752007-08-16 04:30:38 +0000552 sqlite3_result_blob(context, (char*)p, n, sqlite3_free);
drh02d85832007-05-07 19:31:15 +0000553 }
drh63cf66f2007-01-29 15:50:05 +0000554}
555
556/*
drh6ed41ad2002-04-06 14:10:47 +0000557** Implementation of the last_insert_rowid() SQL function. The return
danielk197724b03fd2004-05-10 10:34:34 +0000558** value is the same as the sqlite3_last_insert_rowid() API function.
drh6ed41ad2002-04-06 14:10:47 +0000559*/
danielk197751ad0ec2004-05-24 12:39:02 +0000560static void last_insert_rowid(
danielk19770ae8b832004-05-25 12:05:56 +0000561 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000562 int NotUsed,
563 sqlite3_value **NotUsed2
danielk197751ad0ec2004-05-24 12:39:02 +0000564){
drhfa4a4b92008-03-19 21:45:51 +0000565 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000566 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000567 /* IMP: R-51513-12026 The last_insert_rowid() SQL function is a
568 ** wrapper around the sqlite3_last_insert_rowid() C/C++ interface
569 ** function. */
drhf9b596e2004-05-26 16:54:42 +0000570 sqlite3_result_int64(context, sqlite3_last_insert_rowid(db));
drh6ed41ad2002-04-06 14:10:47 +0000571}
572
rdcf146a772004-02-25 22:51:06 +0000573/*
drhab2f1f92010-01-11 18:26:42 +0000574** Implementation of the changes() SQL function.
575**
576** IMP: R-62073-11209 The changes() SQL function is a wrapper
577** around the sqlite3_changes() C/C++ function and hence follows the same
578** rules for counting changes.
rdcf146a772004-02-25 22:51:06 +0000579*/
danielk1977b28af712004-06-21 06:50:26 +0000580static void changes(
drhf9b596e2004-05-26 16:54:42 +0000581 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000582 int NotUsed,
583 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000584){
drhfa4a4b92008-03-19 21:45:51 +0000585 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000586 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhf4479502004-05-27 03:12:53 +0000587 sqlite3_result_int(context, sqlite3_changes(db));
rdcb0c374f2004-02-20 22:53:38 +0000588}
rdcf146a772004-02-25 22:51:06 +0000589
590/*
danielk1977b28af712004-06-21 06:50:26 +0000591** Implementation of the total_changes() SQL function. The return value is
592** the same as the sqlite3_total_changes() API function.
rdcf146a772004-02-25 22:51:06 +0000593*/
danielk1977b28af712004-06-21 06:50:26 +0000594static void total_changes(
595 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000596 int NotUsed,
597 sqlite3_value **NotUsed2
danielk197751ad0ec2004-05-24 12:39:02 +0000598){
drhfa4a4b92008-03-19 21:45:51 +0000599 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000600 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000601 /* IMP: R-52756-41993 This function is a wrapper around the
602 ** sqlite3_total_changes() C/C++ interface. */
danielk1977b28af712004-06-21 06:50:26 +0000603 sqlite3_result_int(context, sqlite3_total_changes(db));
rdcb0c374f2004-02-20 22:53:38 +0000604}
605
drh6ed41ad2002-04-06 14:10:47 +0000606/*
drh4e5ffc52004-08-31 00:52:37 +0000607** A structure defining how to do GLOB-style comparisons.
danielk1977d02eb1f2004-06-06 09:44:03 +0000608*/
drh4e5ffc52004-08-31 00:52:37 +0000609struct compareInfo {
drh07e83472016-01-11 03:48:18 +0000610 u8 matchAll; /* "*" or "%" */
611 u8 matchOne; /* "?" or "_" */
612 u8 matchSet; /* "[" or 0 */
613 u8 noCase; /* true to ignore case differences */
danielk1977d02eb1f2004-06-06 09:44:03 +0000614};
drh55ef4d92005-08-14 01:20:37 +0000615
drhb9175ae2007-12-07 18:39:04 +0000616/*
617** For LIKE and GLOB matching on EBCDIC machines, assume that every
drhb0870482015-06-17 13:20:54 +0000618** character is exactly one byte in size. Also, provde the Utf8Read()
619** macro for fast reading of the next character in the common case where
620** the next character is ASCII.
drhb9175ae2007-12-07 18:39:04 +0000621*/
622#if defined(SQLITE_EBCDIC)
drh88b33222014-09-25 03:51:37 +0000623# define sqlite3Utf8Read(A) (*((*A)++))
drhb0870482015-06-17 13:20:54 +0000624# define Utf8Read(A) (*(A++))
drhb9175ae2007-12-07 18:39:04 +0000625#else
drhb0870482015-06-17 13:20:54 +0000626# define Utf8Read(A) (A[0]<0x80?*(A++):sqlite3Utf8Read(&A))
drhb9175ae2007-12-07 18:39:04 +0000627#endif
628
drh4e5ffc52004-08-31 00:52:37 +0000629static const struct compareInfo globInfo = { '*', '?', '[', 0 };
drh55ef4d92005-08-14 01:20:37 +0000630/* The correct SQL-92 behavior is for the LIKE operator to ignore
631** case. Thus 'a' LIKE 'A' would be true. */
632static const struct compareInfo likeInfoNorm = { '%', '_', 0, 1 };
633/* If SQLITE_CASE_SENSITIVE_LIKE is defined, then the LIKE operator
634** is case sensitive causing 'a' LIKE 'A' to be false */
635static const struct compareInfo likeInfoAlt = { '%', '_', 0, 0 };
danielk1977d02eb1f2004-06-06 09:44:03 +0000636
637/*
drh698a01c2016-12-01 18:49:40 +0000638** Possible error returns from patternMatch()
639*/
640#define SQLITE_MATCH 0
641#define SQLITE_NOMATCH 1
642#define SQLITE_NOWILDCARDMATCH 2
643
644/*
645** Compare two UTF-8 strings for equality where the first string is
646** a GLOB or LIKE expression. Return values:
647**
648** SQLITE_MATCH: Match
649** SQLITE_NOMATCH: No match
650** SQLITE_NOWILDCARDMATCH: No match in spite of having * or % wildcards.
drh0ac65892002-04-20 14:24:41 +0000651**
drh4e5ffc52004-08-31 00:52:37 +0000652** Globbing rules:
drh0ac65892002-04-20 14:24:41 +0000653**
drh4e5ffc52004-08-31 00:52:37 +0000654** '*' Matches any sequence of zero or more characters.
danielk1977d02eb1f2004-06-06 09:44:03 +0000655**
drh4e5ffc52004-08-31 00:52:37 +0000656** '?' Matches exactly one character.
657**
658** [...] Matches one character from the enclosed list of
659** characters.
660**
661** [^...] Matches one character not in the enclosed list.
662**
663** With the [...] and [^...] matching, a ']' character can be included
664** in the list by making it the first character after '[' or '^'. A
665** range of characters can be specified using '-'. Example:
666** "[a-z]" matches any single lower-case letter. To match a '-', make
667** it the last character in the list.
668**
drh9fdfdc82014-09-25 11:08:57 +0000669** Like matching rules:
670**
671** '%' Matches any sequence of zero or more characters
672**
673*** '_' Matches any one character
674**
675** Ec Where E is the "esc" character and c is any other
676** character, including '%', '_', and esc, match exactly c.
677**
drhb0870482015-06-17 13:20:54 +0000678** The comments within this routine usually assume glob matching.
drh9fdfdc82014-09-25 11:08:57 +0000679**
drh4e5ffc52004-08-31 00:52:37 +0000680** This routine is usually quick, but can be N**2 in the worst case.
drh0ac65892002-04-20 14:24:41 +0000681*/
danielk19777c6303c2004-11-17 16:41:29 +0000682static int patternCompare(
drh4e5ffc52004-08-31 00:52:37 +0000683 const u8 *zPattern, /* The glob pattern */
684 const u8 *zString, /* The string to compare against the glob */
danielk19777c6303c2004-11-17 16:41:29 +0000685 const struct compareInfo *pInfo, /* Information about how to do the compare */
drh698a01c2016-12-01 18:49:40 +0000686 u32 matchOther /* The escape char (LIKE) or '[' (GLOB) */
danielk197751ad0ec2004-05-24 12:39:02 +0000687){
drh9fdfdc82014-09-25 11:08:57 +0000688 u32 c, c2; /* Next pattern and input string chars */
689 u32 matchOne = pInfo->matchOne; /* "?" or "_" */
690 u32 matchAll = pInfo->matchAll; /* "*" or "%" */
drh9fdfdc82014-09-25 11:08:57 +0000691 u8 noCase = pInfo->noCase; /* True if uppercase==lowercase */
692 const u8 *zEscaped = 0; /* One past the last escaped input char */
drh88b33222014-09-25 03:51:37 +0000693
drhb0870482015-06-17 13:20:54 +0000694 while( (c = Utf8Read(zPattern))!=0 ){
drh9fdfdc82014-09-25 11:08:57 +0000695 if( c==matchAll ){ /* Match "*" */
696 /* Skip over multiple "*" characters in the pattern. If there
697 ** are also "?" characters, skip those as well, but consume a
698 ** single character of the input string for each "?" skipped */
drhb0870482015-06-17 13:20:54 +0000699 while( (c=Utf8Read(zPattern)) == matchAll || c == matchOne ){
drh42610962012-09-17 18:56:32 +0000700 if( c==matchOne && sqlite3Utf8Read(&zString)==0 ){
drh698a01c2016-12-01 18:49:40 +0000701 return SQLITE_NOWILDCARDMATCH;
drh4e5ffc52004-08-31 00:52:37 +0000702 }
danielk1977ad7dd422004-06-06 12:41:49 +0000703 }
drh66150952007-07-23 19:12:41 +0000704 if( c==0 ){
drh698a01c2016-12-01 18:49:40 +0000705 return SQLITE_MATCH; /* "*" at the end of the pattern matches */
drh88b33222014-09-25 03:51:37 +0000706 }else if( c==matchOther ){
drh07e83472016-01-11 03:48:18 +0000707 if( pInfo->matchSet==0 ){
drh88b33222014-09-25 03:51:37 +0000708 c = sqlite3Utf8Read(&zPattern);
drh698a01c2016-12-01 18:49:40 +0000709 if( c==0 ) return SQLITE_NOWILDCARDMATCH;
drh88b33222014-09-25 03:51:37 +0000710 }else{
drh9fdfdc82014-09-25 11:08:57 +0000711 /* "[...]" immediately follows the "*". We have to do a slow
712 ** recursive search in this case, but it is an unusual case. */
drh88b33222014-09-25 03:51:37 +0000713 assert( matchOther<0x80 ); /* '[' is a single-byte character */
drh698a01c2016-12-01 18:49:40 +0000714 while( *zString ){
715 int bMatch = patternCompare(&zPattern[-1],zString,pInfo,matchOther);
716 if( bMatch!=SQLITE_NOMATCH ) return bMatch;
drh88b33222014-09-25 03:51:37 +0000717 SQLITE_SKIP_UTF8(zString);
718 }
drh698a01c2016-12-01 18:49:40 +0000719 return SQLITE_NOWILDCARDMATCH;
drh66150952007-07-23 19:12:41 +0000720 }
drh66150952007-07-23 19:12:41 +0000721 }
drh9fdfdc82014-09-25 11:08:57 +0000722
723 /* At this point variable c contains the first character of the
724 ** pattern string past the "*". Search in the input string for the
dan1a4a7372016-12-01 17:34:59 +0000725 ** first matching character and recursively continue the match from
drh9fdfdc82014-09-25 11:08:57 +0000726 ** that point.
727 **
728 ** For a case-insensitive search, set variable cx to be the same as
729 ** c but in the other case and search the input string for either
730 ** c or cx.
731 */
732 if( c<=0x80 ){
drheba21f92017-10-30 18:49:11 +0000733 char zStop[3];
drh698a01c2016-12-01 18:49:40 +0000734 int bMatch;
drh9fdfdc82014-09-25 11:08:57 +0000735 if( noCase ){
drheba21f92017-10-30 18:49:11 +0000736 zStop[0] = sqlite3Toupper(c);
737 zStop[1] = sqlite3Tolower(c);
738 zStop[2] = 0;
drh66150952007-07-23 19:12:41 +0000739 }else{
drheba21f92017-10-30 18:49:11 +0000740 zStop[0] = c;
741 zStop[1] = 0;
drh4e5ffc52004-08-31 00:52:37 +0000742 }
drheba21f92017-10-30 18:49:11 +0000743 while(1){
744 zString += strcspn((const char*)zString, zStop);
745 if( zString[0]==0 ) break;
746 zString++;
drh698a01c2016-12-01 18:49:40 +0000747 bMatch = patternCompare(zPattern,zString,pInfo,matchOther);
748 if( bMatch!=SQLITE_NOMATCH ) return bMatch;
drh9fdfdc82014-09-25 11:08:57 +0000749 }
750 }else{
drh698a01c2016-12-01 18:49:40 +0000751 int bMatch;
drhb0870482015-06-17 13:20:54 +0000752 while( (c2 = Utf8Read(zString))!=0 ){
drh9fdfdc82014-09-25 11:08:57 +0000753 if( c2!=c ) continue;
drh698a01c2016-12-01 18:49:40 +0000754 bMatch = patternCompare(zPattern,zString,pInfo,matchOther);
755 if( bMatch!=SQLITE_NOMATCH ) return bMatch;
drh9fdfdc82014-09-25 11:08:57 +0000756 }
drh66150952007-07-23 19:12:41 +0000757 }
drh698a01c2016-12-01 18:49:40 +0000758 return SQLITE_NOWILDCARDMATCH;
drh88b33222014-09-25 03:51:37 +0000759 }
drh88b33222014-09-25 03:51:37 +0000760 if( c==matchOther ){
drh07e83472016-01-11 03:48:18 +0000761 if( pInfo->matchSet==0 ){
drh88b33222014-09-25 03:51:37 +0000762 c = sqlite3Utf8Read(&zPattern);
drh698a01c2016-12-01 18:49:40 +0000763 if( c==0 ) return SQLITE_NOMATCH;
drh9fdfdc82014-09-25 11:08:57 +0000764 zEscaped = zPattern;
drh88b33222014-09-25 03:51:37 +0000765 }else{
766 u32 prior_c = 0;
drh9fdfdc82014-09-25 11:08:57 +0000767 int seen = 0;
768 int invert = 0;
drh88b33222014-09-25 03:51:37 +0000769 c = sqlite3Utf8Read(&zString);
drh698a01c2016-12-01 18:49:40 +0000770 if( c==0 ) return SQLITE_NOMATCH;
drh88b33222014-09-25 03:51:37 +0000771 c2 = sqlite3Utf8Read(&zPattern);
772 if( c2=='^' ){
773 invert = 1;
774 c2 = sqlite3Utf8Read(&zPattern);
775 }
776 if( c2==']' ){
777 if( c==']' ) seen = 1;
778 c2 = sqlite3Utf8Read(&zPattern);
779 }
780 while( c2 && c2!=']' ){
781 if( c2=='-' && zPattern[0]!=']' && zPattern[0]!=0 && prior_c>0 ){
782 c2 = sqlite3Utf8Read(&zPattern);
783 if( c>=prior_c && c<=c2 ) seen = 1;
784 prior_c = 0;
785 }else{
786 if( c==c2 ){
787 seen = 1;
788 }
789 prior_c = c2;
790 }
791 c2 = sqlite3Utf8Read(&zPattern);
792 }
793 if( c2==0 || (seen ^ invert)==0 ){
drh698a01c2016-12-01 18:49:40 +0000794 return SQLITE_NOMATCH;
drh88b33222014-09-25 03:51:37 +0000795 }
796 continue;
797 }
798 }
drhb0870482015-06-17 13:20:54 +0000799 c2 = Utf8Read(zString);
drh88b33222014-09-25 03:51:37 +0000800 if( c==c2 ) continue;
drhc80937a2016-06-06 01:48:14 +0000801 if( noCase && sqlite3Tolower(c)==sqlite3Tolower(c2) && c<0x80 && c2<0x80 ){
drh9fdfdc82014-09-25 11:08:57 +0000802 continue;
803 }
804 if( c==matchOne && zPattern!=zEscaped && c2!=0 ) continue;
drh698a01c2016-12-01 18:49:40 +0000805 return SQLITE_NOMATCH;
danielk197751ad0ec2004-05-24 12:39:02 +0000806 }
drh698a01c2016-12-01 18:49:40 +0000807 return *zString==0 ? SQLITE_MATCH : SQLITE_NOMATCH;
drh0ac65892002-04-20 14:24:41 +0000808}
drh4e5ffc52004-08-31 00:52:37 +0000809
drh55ef4d92005-08-14 01:20:37 +0000810/*
drh698a01c2016-12-01 18:49:40 +0000811** The sqlite3_strglob() interface. Return 0 on a match (like strcmp()) and
812** non-zero if there is no match.
drh56282a52013-04-10 16:13:38 +0000813*/
814int sqlite3_strglob(const char *zGlobPattern, const char *zString){
drh698a01c2016-12-01 18:49:40 +0000815 return patternCompare((u8*)zGlobPattern, (u8*)zString, &globInfo, '[');
drh56282a52013-04-10 16:13:38 +0000816}
817
818/*
drh698a01c2016-12-01 18:49:40 +0000819** The sqlite3_strlike() interface. Return 0 on a match and non-zero for
820** a miss - like strcmp().
drh8b4a94a2015-11-24 21:23:59 +0000821*/
822int sqlite3_strlike(const char *zPattern, const char *zStr, unsigned int esc){
drh698a01c2016-12-01 18:49:40 +0000823 return patternCompare((u8*)zPattern, (u8*)zStr, &likeInfoNorm, esc);
drh8b4a94a2015-11-24 21:23:59 +0000824}
825
826/*
drh55ef4d92005-08-14 01:20:37 +0000827** Count the number of times that the LIKE operator (or GLOB which is
828** just a variation of LIKE) gets called. This is used for testing
829** only.
830*/
831#ifdef SQLITE_TEST
832int sqlite3_like_count = 0;
833#endif
834
danielk19773f6b0872004-06-17 05:36:44 +0000835
836/*
837** Implementation of the like() SQL function. This function implements
838** the build-in LIKE operator. The first argument to the function is the
839** pattern and the second argument is the string. So, the SQL statements:
840**
841** A LIKE B
842**
843** is implemented as like(B,A).
844**
drh55ef4d92005-08-14 01:20:37 +0000845** This same function (with a different compareInfo structure) computes
846** the GLOB operator.
danielk19773f6b0872004-06-17 05:36:44 +0000847*/
848static void likeFunc(
849 sqlite3_context *context,
850 int argc,
851 sqlite3_value **argv
852){
drhbeb818d2007-05-08 15:34:47 +0000853 const unsigned char *zA, *zB;
drh07e83472016-01-11 03:48:18 +0000854 u32 escape;
drh27e62db2009-04-02 10:16:17 +0000855 int nPat;
drhbb4957f2008-03-20 14:03:29 +0000856 sqlite3 *db = sqlite3_context_db_handle(context);
drh07e83472016-01-11 03:48:18 +0000857 struct compareInfo *pInfo = sqlite3_user_data(context);
drh589c7872020-03-19 18:13:28 +0000858 struct compareInfo backupInfo;
drhbeb818d2007-05-08 15:34:47 +0000859
drh41d2e662015-12-01 21:23:07 +0000860#ifdef SQLITE_LIKE_DOESNT_MATCH_BLOBS
861 if( sqlite3_value_type(argv[0])==SQLITE_BLOB
862 || sqlite3_value_type(argv[1])==SQLITE_BLOB
863 ){
864#ifdef SQLITE_TEST
865 sqlite3_like_count++;
866#endif
867 sqlite3_result_int(context, 0);
868 return;
869 }
870#endif
drh1f0feef2007-05-15 13:27:07 +0000871
drhbeb818d2007-05-08 15:34:47 +0000872 /* Limit the length of the LIKE or GLOB pattern to avoid problems
873 ** of deep recursion and N*N behavior in patternCompare().
874 */
drh27e62db2009-04-02 10:16:17 +0000875 nPat = sqlite3_value_bytes(argv[0]);
876 testcase( nPat==db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH] );
877 testcase( nPat==db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH]+1 );
878 if( nPat > db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH] ){
drhbeb818d2007-05-08 15:34:47 +0000879 sqlite3_result_error(context, "LIKE or GLOB pattern too complex", -1);
880 return;
881 }
danielk19777c6303c2004-11-17 16:41:29 +0000882 if( argc==3 ){
883 /* The escape character string must consist of a single UTF-8 character.
884 ** Otherwise, return an error.
885 */
886 const unsigned char *zEsc = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +0000887 if( zEsc==0 ) return;
drhee858132007-05-08 20:37:38 +0000888 if( sqlite3Utf8CharLen((char*)zEsc, -1)!=1 ){
danielk19777c6303c2004-11-17 16:41:29 +0000889 sqlite3_result_error(context,
890 "ESCAPE expression must be a single character", -1);
891 return;
892 }
drh42610962012-09-17 18:56:32 +0000893 escape = sqlite3Utf8Read(&zEsc);
drh589c7872020-03-19 18:13:28 +0000894 if( escape==pInfo->matchAll || escape==pInfo->matchOne ){
895 memcpy(&backupInfo, pInfo, sizeof(backupInfo));
896 pInfo = &backupInfo;
897 if( escape==pInfo->matchAll ) pInfo->matchAll = 0;
898 if( escape==pInfo->matchOne ) pInfo->matchOne = 0;
899 }
drh07e83472016-01-11 03:48:18 +0000900 }else{
901 escape = pInfo->matchSet;
danielk19777c6303c2004-11-17 16:41:29 +0000902 }
drhcf833232019-04-30 11:54:36 +0000903 zB = sqlite3_value_text(argv[0]);
904 zA = sqlite3_value_text(argv[1]);
danielk19773f6b0872004-06-17 05:36:44 +0000905 if( zA && zB ){
drh55ef4d92005-08-14 01:20:37 +0000906#ifdef SQLITE_TEST
907 sqlite3_like_count++;
908#endif
drhf49759b2017-08-25 19:51:51 +0000909 sqlite3_result_int(context,
910 patternCompare(zB, zA, pInfo, escape)==SQLITE_MATCH);
danielk197751ad0ec2004-05-24 12:39:02 +0000911 }
drh8912d102002-05-26 21:34:58 +0000912}
913
914/*
915** Implementation of the NULLIF(x,y) function. The result is the first
916** argument if the arguments are different. The result is NULL if the
917** arguments are equal to each other.
918*/
drhf9b596e2004-05-26 16:54:42 +0000919static void nullifFunc(
920 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000921 int NotUsed,
drhf9b596e2004-05-26 16:54:42 +0000922 sqlite3_value **argv
923){
danielk1977dc1bdc42004-06-11 10:51:27 +0000924 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk197762c14b32008-11-19 09:05:26 +0000925 UNUSED_PARAMETER(NotUsed);
danielk1977dc1bdc42004-06-11 10:51:27 +0000926 if( sqlite3MemCompare(argv[0], argv[1], pColl)!=0 ){
drhf4479502004-05-27 03:12:53 +0000927 sqlite3_result_value(context, argv[0]);
drh8912d102002-05-26 21:34:58 +0000928 }
drh0ac65892002-04-20 14:24:41 +0000929}
930
drh647cb0e2002-11-04 19:32:25 +0000931/*
drh47baebc2009-08-14 16:01:24 +0000932** Implementation of the sqlite_version() function. The result is the version
drh647cb0e2002-11-04 19:32:25 +0000933** of the SQLite library that is running.
934*/
drhf9b596e2004-05-26 16:54:42 +0000935static void versionFunc(
936 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000937 int NotUsed,
938 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000939){
danielk197762c14b32008-11-19 09:05:26 +0000940 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000941 /* IMP: R-48699-48617 This function is an SQL wrapper around the
942 ** sqlite3_libversion() C-interface. */
943 sqlite3_result_text(context, sqlite3_libversion(), -1, SQLITE_STATIC);
drh647cb0e2002-11-04 19:32:25 +0000944}
945
drh47baebc2009-08-14 16:01:24 +0000946/*
947** Implementation of the sqlite_source_id() function. The result is a string
948** that identifies the particular version of the source code used to build
949** SQLite.
950*/
951static void sourceidFunc(
952 sqlite3_context *context,
953 int NotUsed,
954 sqlite3_value **NotUsed2
955){
956 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000957 /* IMP: R-24470-31136 This function is an SQL wrapper around the
958 ** sqlite3_sourceid() C interface. */
959 sqlite3_result_text(context, sqlite3_sourceid(), -1, SQLITE_STATIC);
drh47baebc2009-08-14 16:01:24 +0000960}
961
shanehbdea6d12010-02-23 04:19:54 +0000962/*
drh3ca84ef2011-04-25 18:03:10 +0000963** Implementation of the sqlite_log() function. This is a wrapper around
964** sqlite3_log(). The return value is NULL. The function exists purely for
965** its side-effects.
966*/
drh840561f2011-04-27 18:08:42 +0000967static void errlogFunc(
drh3ca84ef2011-04-25 18:03:10 +0000968 sqlite3_context *context,
969 int argc,
970 sqlite3_value **argv
971){
972 UNUSED_PARAMETER(argc);
973 UNUSED_PARAMETER(context);
974 sqlite3_log(sqlite3_value_int(argv[0]), "%s", sqlite3_value_text(argv[1]));
975}
976
977/*
shanehdc97a8c2010-02-23 20:08:35 +0000978** Implementation of the sqlite_compileoption_used() function.
979** The result is an integer that identifies if the compiler option
980** was used to build SQLite.
shanehbdea6d12010-02-23 04:19:54 +0000981*/
shanehdc97a8c2010-02-23 20:08:35 +0000982#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
983static void compileoptionusedFunc(
shanehbdea6d12010-02-23 04:19:54 +0000984 sqlite3_context *context,
shanehdc97a8c2010-02-23 20:08:35 +0000985 int argc,
986 sqlite3_value **argv
shanehbdea6d12010-02-23 04:19:54 +0000987){
shanehdc97a8c2010-02-23 20:08:35 +0000988 const char *zOptName;
989 assert( argc==1 );
990 UNUSED_PARAMETER(argc);
drha3e414c2010-08-03 18:06:25 +0000991 /* IMP: R-39564-36305 The sqlite_compileoption_used() SQL
992 ** function is a wrapper around the sqlite3_compileoption_used() C/C++
993 ** function.
994 */
drh264a2d42010-02-25 15:28:41 +0000995 if( (zOptName = (const char*)sqlite3_value_text(argv[0]))!=0 ){
shanehdc97a8c2010-02-23 20:08:35 +0000996 sqlite3_result_int(context, sqlite3_compileoption_used(zOptName));
shanehdc97a8c2010-02-23 20:08:35 +0000997 }
shanehbdea6d12010-02-23 04:19:54 +0000998}
shanehdc97a8c2010-02-23 20:08:35 +0000999#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
1000
1001/*
1002** Implementation of the sqlite_compileoption_get() function.
1003** The result is a string that identifies the compiler options
1004** used to build SQLite.
1005*/
1006#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
1007static void compileoptiongetFunc(
1008 sqlite3_context *context,
1009 int argc,
1010 sqlite3_value **argv
1011){
1012 int n;
1013 assert( argc==1 );
1014 UNUSED_PARAMETER(argc);
drha3e414c2010-08-03 18:06:25 +00001015 /* IMP: R-04922-24076 The sqlite_compileoption_get() SQL function
1016 ** is a wrapper around the sqlite3_compileoption_get() C/C++ function.
1017 */
shanehdc97a8c2010-02-23 20:08:35 +00001018 n = sqlite3_value_int(argv[0]);
1019 sqlite3_result_text(context, sqlite3_compileoption_get(n), -1, SQLITE_STATIC);
1020}
1021#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
shanehbdea6d12010-02-23 04:19:54 +00001022
drh137c7282007-01-29 17:58:28 +00001023/* Array for converting from half-bytes (nybbles) into ASCII hex
1024** digits. */
1025static const char hexdigits[] = {
1026 '0', '1', '2', '3', '4', '5', '6', '7',
1027 '8', '9', 'A', 'B', 'C', 'D', 'E', 'F'
1028};
danielk1977d641d642004-11-18 15:44:29 +00001029
drh47394702003-08-20 01:03:33 +00001030/*
drh47394702003-08-20 01:03:33 +00001031** Implementation of the QUOTE() function. This function takes a single
1032** argument. If the argument is numeric, the return value is the same as
1033** the argument. If the argument is NULL, the return value is the string
1034** "NULL". Otherwise, the argument is enclosed in single quotes with
1035** single-quote escapes.
1036*/
danielk19770ae8b832004-05-25 12:05:56 +00001037static void quoteFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drha0df4cc2009-02-02 17:29:59 +00001038 assert( argc==1 );
drh1d34fde2009-02-03 15:50:33 +00001039 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +00001040 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +00001041 case SQLITE_FLOAT: {
drh72b3fbc2012-06-19 03:11:25 +00001042 double r1, r2;
drh72b3fbc2012-06-19 03:11:25 +00001043 char zBuf[50];
mistachkin2b434a72012-06-19 04:36:48 +00001044 r1 = sqlite3_value_double(argv[0]);
drh72b3fbc2012-06-19 03:11:25 +00001045 sqlite3_snprintf(sizeof(zBuf), zBuf, "%!.15g", r1);
1046 sqlite3AtoF(zBuf, &r2, 20, SQLITE_UTF8);
1047 if( r1!=r2 ){
1048 sqlite3_snprintf(sizeof(zBuf), zBuf, "%!.20e", r1);
1049 }
1050 sqlite3_result_text(context, zBuf, -1, SQLITE_TRANSIENT);
1051 break;
1052 }
1053 case SQLITE_INTEGER: {
drhf4479502004-05-27 03:12:53 +00001054 sqlite3_result_value(context, argv[0]);
drhf9b596e2004-05-26 16:54:42 +00001055 break;
1056 }
danielk19773f41e972004-06-08 00:39:01 +00001057 case SQLITE_BLOB: {
danielk19773f41e972004-06-08 00:39:01 +00001058 char *zText = 0;
danielk19773f41e972004-06-08 00:39:01 +00001059 char const *zBlob = sqlite3_value_blob(argv[0]);
drh1f0feef2007-05-15 13:27:07 +00001060 int nBlob = sqlite3_value_bytes(argv[0]);
1061 assert( zBlob==sqlite3_value_blob(argv[0]) ); /* No encoding change */
drhb1a6c3c2008-03-20 16:30:17 +00001062 zText = (char *)contextMalloc(context, (2*(i64)nBlob)+4);
danielk1977a1644fd2007-08-29 12:31:25 +00001063 if( zText ){
danielk19773f41e972004-06-08 00:39:01 +00001064 int i;
1065 for(i=0; i<nBlob; i++){
1066 zText[(i*2)+2] = hexdigits[(zBlob[i]>>4)&0x0F];
1067 zText[(i*2)+3] = hexdigits[(zBlob[i])&0x0F];
1068 }
1069 zText[(nBlob*2)+2] = '\'';
1070 zText[(nBlob*2)+3] = '\0';
1071 zText[0] = 'X';
1072 zText[1] = '\'';
danielk1977d8123362004-06-12 09:25:12 +00001073 sqlite3_result_text(context, zText, -1, SQLITE_TRANSIENT);
drh17435752007-08-16 04:30:38 +00001074 sqlite3_free(zText);
danielk19773f41e972004-06-08 00:39:01 +00001075 }
1076 break;
1077 }
drh9c054832004-05-31 18:51:57 +00001078 case SQLITE_TEXT: {
drh023ae032007-05-08 12:12:16 +00001079 int i,j;
1080 u64 n;
drh2646da72005-12-09 20:02:05 +00001081 const unsigned char *zArg = sqlite3_value_text(argv[0]);
drhf9b596e2004-05-26 16:54:42 +00001082 char *z;
1083
drh7a521cf2007-04-25 18:23:52 +00001084 if( zArg==0 ) return;
drh023ae032007-05-08 12:12:16 +00001085 for(i=0, n=0; zArg[i]; i++){ if( zArg[i]=='\'' ) n++; }
drhb1a6c3c2008-03-20 16:30:17 +00001086 z = contextMalloc(context, ((i64)i)+((i64)n)+3);
danielk1977a1644fd2007-08-29 12:31:25 +00001087 if( z ){
1088 z[0] = '\'';
1089 for(i=0, j=1; zArg[i]; i++){
1090 z[j++] = zArg[i];
1091 if( zArg[i]=='\'' ){
1092 z[j++] = '\'';
1093 }
drhf9b596e2004-05-26 16:54:42 +00001094 }
danielk1977a1644fd2007-08-29 12:31:25 +00001095 z[j++] = '\'';
1096 z[j] = 0;
1097 sqlite3_result_text(context, z, j, sqlite3_free);
drhf9b596e2004-05-26 16:54:42 +00001098 }
drha0df4cc2009-02-02 17:29:59 +00001099 break;
1100 }
1101 default: {
1102 assert( sqlite3_value_type(argv[0])==SQLITE_NULL );
1103 sqlite3_result_text(context, "NULL", 4, SQLITE_STATIC);
1104 break;
drhf9b596e2004-05-26 16:54:42 +00001105 }
drh47394702003-08-20 01:03:33 +00001106 }
1107}
1108
drh137c7282007-01-29 17:58:28 +00001109/*
drhd495d8c2013-02-22 19:34:25 +00001110** The unicode() function. Return the integer unicode code-point value
1111** for the first character of the input string.
1112*/
1113static void unicodeFunc(
1114 sqlite3_context *context,
1115 int argc,
1116 sqlite3_value **argv
1117){
1118 const unsigned char *z = sqlite3_value_text(argv[0]);
drh1d59d032013-03-01 23:40:26 +00001119 (void)argc;
drhd495d8c2013-02-22 19:34:25 +00001120 if( z && z[0] ) sqlite3_result_int(context, sqlite3Utf8Read(&z));
1121}
1122
1123/*
1124** The char() function takes zero or more arguments, each of which is
1125** an integer. It constructs a string where each character of the string
1126** is the unicode character for the corresponding integer argument.
1127*/
1128static void charFunc(
1129 sqlite3_context *context,
1130 int argc,
1131 sqlite3_value **argv
1132){
1133 unsigned char *z, *zOut;
1134 int i;
drhf3cdcdc2015-04-29 16:50:28 +00001135 zOut = z = sqlite3_malloc64( argc*4+1 );
drhd495d8c2013-02-22 19:34:25 +00001136 if( z==0 ){
1137 sqlite3_result_error_nomem(context);
1138 return;
1139 }
1140 for(i=0; i<argc; i++){
mistachkinc9545442013-02-26 05:42:30 +00001141 sqlite3_int64 x;
drhd495d8c2013-02-22 19:34:25 +00001142 unsigned c;
1143 x = sqlite3_value_int64(argv[i]);
1144 if( x<0 || x>0x10ffff ) x = 0xfffd;
1145 c = (unsigned)(x & 0x1fffff);
drhfe7a5d12013-03-07 14:00:04 +00001146 if( c<0x00080 ){
1147 *zOut++ = (u8)(c&0xFF);
1148 }else if( c<0x00800 ){
1149 *zOut++ = 0xC0 + (u8)((c>>6)&0x1F);
1150 *zOut++ = 0x80 + (u8)(c & 0x3F);
1151 }else if( c<0x10000 ){
1152 *zOut++ = 0xE0 + (u8)((c>>12)&0x0F);
1153 *zOut++ = 0x80 + (u8)((c>>6) & 0x3F);
1154 *zOut++ = 0x80 + (u8)(c & 0x3F);
drhd495d8c2013-02-22 19:34:25 +00001155 }else{
drhfe7a5d12013-03-07 14:00:04 +00001156 *zOut++ = 0xF0 + (u8)((c>>18) & 0x07);
1157 *zOut++ = 0x80 + (u8)((c>>12) & 0x3F);
1158 *zOut++ = 0x80 + (u8)((c>>6) & 0x3F);
1159 *zOut++ = 0x80 + (u8)(c & 0x3F);
1160 } \
drhd495d8c2013-02-22 19:34:25 +00001161 }
drhbbf483f2014-09-09 20:30:24 +00001162 sqlite3_result_text64(context, (char*)z, zOut-z, sqlite3_free, SQLITE_UTF8);
drhd495d8c2013-02-22 19:34:25 +00001163}
1164
1165/*
drh137c7282007-01-29 17:58:28 +00001166** The hex() function. Interpret the argument as a blob. Return
1167** a hexadecimal rendering as text.
1168*/
1169static void hexFunc(
1170 sqlite3_context *context,
1171 int argc,
1172 sqlite3_value **argv
1173){
1174 int i, n;
1175 const unsigned char *pBlob;
1176 char *zHex, *z;
1177 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001178 UNUSED_PARAMETER(argc);
drh1f0feef2007-05-15 13:27:07 +00001179 pBlob = sqlite3_value_blob(argv[0]);
drh137c7282007-01-29 17:58:28 +00001180 n = sqlite3_value_bytes(argv[0]);
drh1f0feef2007-05-15 13:27:07 +00001181 assert( pBlob==sqlite3_value_blob(argv[0]) ); /* No encoding change */
drhb1a6c3c2008-03-20 16:30:17 +00001182 z = zHex = contextMalloc(context, ((i64)n)*2 + 1);
danielk1977a1644fd2007-08-29 12:31:25 +00001183 if( zHex ){
1184 for(i=0; i<n; i++, pBlob++){
1185 unsigned char c = *pBlob;
1186 *(z++) = hexdigits[(c>>4)&0xf];
1187 *(z++) = hexdigits[c&0xf];
1188 }
1189 *z = 0;
1190 sqlite3_result_text(context, zHex, n*2, sqlite3_free);
drh137c7282007-01-29 17:58:28 +00001191 }
drh137c7282007-01-29 17:58:28 +00001192}
1193
drh26b6d902007-03-17 13:27:54 +00001194/*
drh8cff3822007-05-02 02:08:28 +00001195** The zeroblob(N) function returns a zero-filled blob of size N bytes.
1196*/
1197static void zeroblobFunc(
1198 sqlite3_context *context,
1199 int argc,
1200 sqlite3_value **argv
1201){
drh98640a32007-06-07 19:08:32 +00001202 i64 n;
dana4d5ae82015-07-24 16:24:37 +00001203 int rc;
drh8cff3822007-05-02 02:08:28 +00001204 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001205 UNUSED_PARAMETER(argc);
drh98640a32007-06-07 19:08:32 +00001206 n = sqlite3_value_int64(argv[0]);
dana4d5ae82015-07-24 16:24:37 +00001207 if( n<0 ) n = 0;
1208 rc = sqlite3_result_zeroblob64(context, n); /* IMP: R-00293-64994 */
1209 if( rc ){
1210 sqlite3_result_error_code(context, rc);
drh98640a32007-06-07 19:08:32 +00001211 }
drh8cff3822007-05-02 02:08:28 +00001212}
1213
1214/*
drh26b6d902007-03-17 13:27:54 +00001215** The replace() function. Three arguments are all strings: call
1216** them A, B, and C. The result is also a string which is derived
drhf7b54962013-05-28 12:11:54 +00001217** from A by replacing every occurrence of B with C. The match
drh26b6d902007-03-17 13:27:54 +00001218** must be exact. Collating sequences are not used.
1219*/
1220static void replaceFunc(
1221 sqlite3_context *context,
1222 int argc,
1223 sqlite3_value **argv
1224){
1225 const unsigned char *zStr; /* The input string A */
1226 const unsigned char *zPattern; /* The pattern string B */
1227 const unsigned char *zRep; /* The replacement string C */
1228 unsigned char *zOut; /* The output */
1229 int nStr; /* Size of zStr */
1230 int nPattern; /* Size of zPattern */
1231 int nRep; /* Size of zRep */
drh2e6400b2007-05-08 15:46:18 +00001232 i64 nOut; /* Maximum size of zOut */
drh26b6d902007-03-17 13:27:54 +00001233 int loopLimit; /* Last zStr[] that might match zPattern[] */
1234 int i, j; /* Loop counters */
drhf3139522018-02-09 23:25:14 +00001235 unsigned cntExpand; /* Number zOut expansions */
1236 sqlite3 *db = sqlite3_context_db_handle(context);
drh26b6d902007-03-17 13:27:54 +00001237
1238 assert( argc==3 );
danielk1977f3d3c272008-11-19 16:52:44 +00001239 UNUSED_PARAMETER(argc);
drh26b6d902007-03-17 13:27:54 +00001240 zStr = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001241 if( zStr==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001242 nStr = sqlite3_value_bytes(argv[0]);
1243 assert( zStr==sqlite3_value_text(argv[0]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001244 zPattern = sqlite3_value_text(argv[1]);
drha605fe82009-02-01 18:08:40 +00001245 if( zPattern==0 ){
drh23336062009-02-03 13:19:12 +00001246 assert( sqlite3_value_type(argv[1])==SQLITE_NULL
1247 || sqlite3_context_db_handle(context)->mallocFailed );
drha605fe82009-02-01 18:08:40 +00001248 return;
1249 }
1250 if( zPattern[0]==0 ){
1251 assert( sqlite3_value_type(argv[1])!=SQLITE_NULL );
1252 sqlite3_result_value(context, argv[0]);
1253 return;
1254 }
drh1f0feef2007-05-15 13:27:07 +00001255 nPattern = sqlite3_value_bytes(argv[1]);
1256 assert( zPattern==sqlite3_value_text(argv[1]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001257 zRep = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +00001258 if( zRep==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001259 nRep = sqlite3_value_bytes(argv[2]);
1260 assert( zRep==sqlite3_value_text(argv[2]) );
drh2e6400b2007-05-08 15:46:18 +00001261 nOut = nStr + 1;
1262 assert( nOut<SQLITE_MAX_LENGTH );
drhb1a6c3c2008-03-20 16:30:17 +00001263 zOut = contextMalloc(context, (i64)nOut);
drh2e6400b2007-05-08 15:46:18 +00001264 if( zOut==0 ){
1265 return;
drh26b6d902007-03-17 13:27:54 +00001266 }
drh26b6d902007-03-17 13:27:54 +00001267 loopLimit = nStr - nPattern;
drhf3139522018-02-09 23:25:14 +00001268 cntExpand = 0;
drh26b6d902007-03-17 13:27:54 +00001269 for(i=j=0; i<=loopLimit; i++){
1270 if( zStr[i]!=zPattern[0] || memcmp(&zStr[i], zPattern, nPattern) ){
1271 zOut[j++] = zStr[i];
1272 }else{
drhf3139522018-02-09 23:25:14 +00001273 if( nRep>nPattern ){
1274 nOut += nRep - nPattern;
drhc86d82f2018-02-10 02:31:30 +00001275 testcase( nOut-1==db->aLimit[SQLITE_LIMIT_LENGTH] );
1276 testcase( nOut-2==db->aLimit[SQLITE_LIMIT_LENGTH] );
drhf3139522018-02-09 23:25:14 +00001277 if( nOut-1>db->aLimit[SQLITE_LIMIT_LENGTH] ){
1278 sqlite3_result_error_toobig(context);
1279 sqlite3_free(zOut);
1280 return;
1281 }
drhf3139522018-02-09 23:25:14 +00001282 cntExpand++;
1283 if( (cntExpand&(cntExpand-1))==0 ){
1284 /* Grow the size of the output buffer only on substitutions
1285 ** whose index is a power of two: 1, 2, 4, 8, 16, 32, ... */
1286 u8 *zOld;
1287 zOld = zOut;
drhd924e7b2020-05-17 00:26:44 +00001288 zOut = sqlite3Realloc(zOut, (int)nOut + (nOut - nStr - 1));
drhf3139522018-02-09 23:25:14 +00001289 if( zOut==0 ){
1290 sqlite3_result_error_nomem(context);
1291 sqlite3_free(zOld);
1292 return;
1293 }
1294 }
drh2e6400b2007-05-08 15:46:18 +00001295 }
drh26b6d902007-03-17 13:27:54 +00001296 memcpy(&zOut[j], zRep, nRep);
1297 j += nRep;
1298 i += nPattern-1;
1299 }
1300 }
drhf3139522018-02-09 23:25:14 +00001301 assert( j+nStr-i+1<=nOut );
drh26b6d902007-03-17 13:27:54 +00001302 memcpy(&zOut[j], &zStr[i], nStr-i);
1303 j += nStr - i;
1304 assert( j<=nOut );
1305 zOut[j] = 0;
1306 sqlite3_result_text(context, (char*)zOut, j, sqlite3_free);
1307}
1308
drh309b3382007-03-17 17:52:42 +00001309/*
1310** Implementation of the TRIM(), LTRIM(), and RTRIM() functions.
1311** The userdata is 0x1 for left trim, 0x2 for right trim, 0x3 for both.
1312*/
1313static void trimFunc(
1314 sqlite3_context *context,
1315 int argc,
1316 sqlite3_value **argv
1317){
1318 const unsigned char *zIn; /* Input string */
1319 const unsigned char *zCharSet; /* Set of characters to trim */
1320 int nIn; /* Number of bytes in input */
drh7209c692008-04-27 18:40:11 +00001321 int flags; /* 1: trimleft 2: trimright 3: trim */
drhd1e3a612007-04-27 21:59:52 +00001322 int i; /* Loop counter */
drh1bd10f82008-12-10 21:19:56 +00001323 unsigned char *aLen = 0; /* Length of each character in zCharSet */
1324 unsigned char **azChar = 0; /* Individual characters in zCharSet */
drhd1e3a612007-04-27 21:59:52 +00001325 int nChar; /* Number of characters in zCharSet */
1326
drh309b3382007-03-17 17:52:42 +00001327 if( sqlite3_value_type(argv[0])==SQLITE_NULL ){
1328 return;
1329 }
1330 zIn = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001331 if( zIn==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001332 nIn = sqlite3_value_bytes(argv[0]);
1333 assert( zIn==sqlite3_value_text(argv[0]) );
drh309b3382007-03-17 17:52:42 +00001334 if( argc==1 ){
drhd1e3a612007-04-27 21:59:52 +00001335 static const unsigned char lenOne[] = { 1 };
danielk1977a4de4532008-09-02 15:44:08 +00001336 static unsigned char * const azOne[] = { (u8*)" " };
drhd1e3a612007-04-27 21:59:52 +00001337 nChar = 1;
drh8cff3822007-05-02 02:08:28 +00001338 aLen = (u8*)lenOne;
danielk1977bc67da42007-12-11 04:23:19 +00001339 azChar = (unsigned char **)azOne;
drhd1e3a612007-04-27 21:59:52 +00001340 zCharSet = 0;
drh7a521cf2007-04-25 18:23:52 +00001341 }else if( (zCharSet = sqlite3_value_text(argv[1]))==0 ){
drh309b3382007-03-17 17:52:42 +00001342 return;
drhd1e3a612007-04-27 21:59:52 +00001343 }else{
1344 const unsigned char *z;
1345 for(z=zCharSet, nChar=0; *z; nChar++){
drh4a919112007-05-15 11:55:09 +00001346 SQLITE_SKIP_UTF8(z);
drhd1e3a612007-04-27 21:59:52 +00001347 }
1348 if( nChar>0 ){
drhb1a6c3c2008-03-20 16:30:17 +00001349 azChar = contextMalloc(context, ((i64)nChar)*(sizeof(char*)+1));
drhd1e3a612007-04-27 21:59:52 +00001350 if( azChar==0 ){
1351 return;
1352 }
1353 aLen = (unsigned char*)&azChar[nChar];
1354 for(z=zCharSet, nChar=0; *z; nChar++){
danielk1977bc67da42007-12-11 04:23:19 +00001355 azChar[nChar] = (unsigned char *)z;
drh4a919112007-05-15 11:55:09 +00001356 SQLITE_SKIP_UTF8(z);
drh3abbd392008-12-10 23:04:13 +00001357 aLen[nChar] = (u8)(z - azChar[nChar]);
drhd1e3a612007-04-27 21:59:52 +00001358 }
1359 }
drh309b3382007-03-17 17:52:42 +00001360 }
drhd1e3a612007-04-27 21:59:52 +00001361 if( nChar>0 ){
shane1fc41292008-07-08 22:28:48 +00001362 flags = SQLITE_PTR_TO_INT(sqlite3_user_data(context));
drh309b3382007-03-17 17:52:42 +00001363 if( flags & 1 ){
drhd1e3a612007-04-27 21:59:52 +00001364 while( nIn>0 ){
drh1bd10f82008-12-10 21:19:56 +00001365 int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001366 for(i=0; i<nChar; i++){
1367 len = aLen[i];
drh27e62db2009-04-02 10:16:17 +00001368 if( len<=nIn && memcmp(zIn, azChar[i], len)==0 ) break;
drhd1e3a612007-04-27 21:59:52 +00001369 }
1370 if( i>=nChar ) break;
1371 zIn += len;
1372 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001373 }
1374 }
1375 if( flags & 2 ){
drhd1e3a612007-04-27 21:59:52 +00001376 while( nIn>0 ){
drh1bd10f82008-12-10 21:19:56 +00001377 int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001378 for(i=0; i<nChar; i++){
1379 len = aLen[i];
1380 if( len<=nIn && memcmp(&zIn[nIn-len],azChar[i],len)==0 ) break;
1381 }
1382 if( i>=nChar ) break;
1383 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001384 }
1385 }
drhd1e3a612007-04-27 21:59:52 +00001386 if( zCharSet ){
1387 sqlite3_free(azChar);
1388 }
drh309b3382007-03-17 17:52:42 +00001389 }
1390 sqlite3_result_text(context, (char*)zIn, nIn, SQLITE_TRANSIENT);
1391}
drh26b6d902007-03-17 13:27:54 +00001392
danielk1977a4de4532008-09-02 15:44:08 +00001393
drhcc153132016-08-04 12:35:17 +00001394#ifdef SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION
1395/*
1396** The "unknown" function is automatically substituted in place of
1397** any unrecognized function name when doing an EXPLAIN or EXPLAIN QUERY PLAN
1398** when the SQLITE_ENABLE_UNKNOWN_FUNCTION compile-time option is used.
1399** When the "sqlite3" command-line shell is built using this functionality,
1400** that allows an EXPLAIN or EXPLAIN QUERY PLAN for complex queries
1401** involving application-defined functions to be examined in a generic
1402** sqlite3 shell.
1403*/
1404static void unknownFunc(
1405 sqlite3_context *context,
1406 int argc,
1407 sqlite3_value **argv
1408){
1409 /* no-op */
1410}
1411#endif /*SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION*/
1412
1413
drh2ba3ccc2009-12-08 02:06:08 +00001414/* IMP: R-25361-16150 This function is omitted from SQLite by default. It
1415** is only available if the SQLITE_SOUNDEX compile-time option is used
1416** when SQLite is built.
1417*/
drhd24cc422003-03-27 12:51:24 +00001418#ifdef SQLITE_SOUNDEX
1419/*
1420** Compute the soundex encoding of a word.
drh2ba3ccc2009-12-08 02:06:08 +00001421**
1422** IMP: R-59782-00072 The soundex(X) function returns a string that is the
1423** soundex encoding of the string X.
drhd24cc422003-03-27 12:51:24 +00001424*/
drh137c7282007-01-29 17:58:28 +00001425static void soundexFunc(
1426 sqlite3_context *context,
1427 int argc,
1428 sqlite3_value **argv
1429){
drhd24cc422003-03-27 12:51:24 +00001430 char zResult[8];
drh4c755c02004-08-08 20:22:17 +00001431 const u8 *zIn;
drhd24cc422003-03-27 12:51:24 +00001432 int i, j;
1433 static const unsigned char iCode[] = {
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, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1437 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 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 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1441 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1442 };
1443 assert( argc==1 );
drh4c755c02004-08-08 20:22:17 +00001444 zIn = (u8*)sqlite3_value_text(argv[0]);
drhbdf67e02006-08-19 11:34:01 +00001445 if( zIn==0 ) zIn = (u8*)"";
drhdc86e2b2009-01-24 11:30:42 +00001446 for(i=0; zIn[i] && !sqlite3Isalpha(zIn[i]); i++){}
drhd24cc422003-03-27 12:51:24 +00001447 if( zIn[i] ){
drhbdf67e02006-08-19 11:34:01 +00001448 u8 prevcode = iCode[zIn[i]&0x7f];
danielk197778ca0e72009-01-20 16:53:39 +00001449 zResult[0] = sqlite3Toupper(zIn[i]);
drhd24cc422003-03-27 12:51:24 +00001450 for(j=1; j<4 && zIn[i]; i++){
1451 int code = iCode[zIn[i]&0x7f];
1452 if( code>0 ){
drhbdf67e02006-08-19 11:34:01 +00001453 if( code!=prevcode ){
1454 prevcode = code;
1455 zResult[j++] = code + '0';
1456 }
1457 }else{
1458 prevcode = 0;
drhd24cc422003-03-27 12:51:24 +00001459 }
1460 }
1461 while( j<4 ){
1462 zResult[j++] = '0';
1463 }
1464 zResult[j] = 0;
danielk1977d8123362004-06-12 09:25:12 +00001465 sqlite3_result_text(context, zResult, 4, SQLITE_TRANSIENT);
drhd24cc422003-03-27 12:51:24 +00001466 }else{
drh2ba3ccc2009-12-08 02:06:08 +00001467 /* IMP: R-64894-50321 The string "?000" is returned if the argument
1468 ** is NULL or contains no ASCII alphabetic characters. */
danielk1977d8123362004-06-12 09:25:12 +00001469 sqlite3_result_text(context, "?000", 4, SQLITE_STATIC);
drhd24cc422003-03-27 12:51:24 +00001470 }
1471}
drh2ba3ccc2009-12-08 02:06:08 +00001472#endif /* SQLITE_SOUNDEX */
drhd24cc422003-03-27 12:51:24 +00001473
drhfdb83b22006-06-17 14:12:47 +00001474#ifndef SQLITE_OMIT_LOAD_EXTENSION
1475/*
1476** A function that loads a shared-library extension then returns NULL.
1477*/
1478static void loadExt(sqlite3_context *context, int argc, sqlite3_value **argv){
danielk197765fd59f2006-06-24 11:51:33 +00001479 const char *zFile = (const char *)sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001480 const char *zProc;
drhfa4a4b92008-03-19 21:45:51 +00001481 sqlite3 *db = sqlite3_context_db_handle(context);
drhfdb83b22006-06-17 14:12:47 +00001482 char *zErrMsg = 0;
1483
drh191dd062016-04-21 01:30:09 +00001484 /* Disallow the load_extension() SQL function unless the SQLITE_LoadExtFunc
drh1a55ded2016-04-20 00:30:05 +00001485 ** flag is set. See the sqlite3_enable_load_extension() API.
1486 */
drhf602a162016-04-21 01:58:21 +00001487 if( (db->flags & SQLITE_LoadExtFunc)==0 ){
1488 sqlite3_result_error(context, "not authorized", -1);
1489 return;
1490 }
drh1a55ded2016-04-20 00:30:05 +00001491
drhfdb83b22006-06-17 14:12:47 +00001492 if( argc==2 ){
danielk197765fd59f2006-06-24 11:51:33 +00001493 zProc = (const char *)sqlite3_value_text(argv[1]);
drh7a521cf2007-04-25 18:23:52 +00001494 }else{
1495 zProc = 0;
drhfdb83b22006-06-17 14:12:47 +00001496 }
drh7a521cf2007-04-25 18:23:52 +00001497 if( zFile && sqlite3_load_extension(db, zFile, zProc, &zErrMsg) ){
drhfdb83b22006-06-17 14:12:47 +00001498 sqlite3_result_error(context, zErrMsg, -1);
1499 sqlite3_free(zErrMsg);
1500 }
1501}
1502#endif
1503
danielk197701427a62005-01-11 13:02:33 +00001504
drh0ac65892002-04-20 14:24:41 +00001505/*
drhd3a149e2002-02-24 17:12:53 +00001506** An instance of the following structure holds the context of a
drhdd5baa92002-02-27 19:50:59 +00001507** sum() or avg() aggregate computation.
1508*/
1509typedef struct SumCtx SumCtx;
1510struct SumCtx {
drh8c08e862006-02-11 17:34:00 +00001511 double rSum; /* Floating point sum */
1512 i64 iSum; /* Integer sum */
1513 i64 cnt; /* Number of elements summed */
1514 u8 overflow; /* True if integer overflow seen */
1515 u8 approx; /* True if non-integer value was input to the sum */
drhdd5baa92002-02-27 19:50:59 +00001516};
1517
1518/*
drha97fdd32006-01-12 22:17:50 +00001519** Routines used to compute the sum, average, and total.
1520**
1521** The SUM() function follows the (broken) SQL standard which means
1522** that it returns NULL if it sums over no inputs. TOTAL returns
1523** 0.0 in that case. In addition, TOTAL always returns a float where
1524** SUM might return an integer if it never encounters a floating point
drhc806d852006-05-11 13:25:39 +00001525** value. TOTAL never fails, but SUM might through an exception if
1526** it overflows an integer.
drhdd5baa92002-02-27 19:50:59 +00001527*/
danielk19770ae8b832004-05-25 12:05:56 +00001528static void sumStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drhdd5baa92002-02-27 19:50:59 +00001529 SumCtx *p;
drh3d1d95e2005-09-08 10:37:01 +00001530 int type;
drh3f219f42005-09-08 19:45:57 +00001531 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001532 UNUSED_PARAMETER(argc);
drh4f26d6c2004-05-26 23:25:30 +00001533 p = sqlite3_aggregate_context(context, sizeof(*p));
drh29d72102006-02-09 22:13:41 +00001534 type = sqlite3_value_numeric_type(argv[0]);
drh3d1d95e2005-09-08 10:37:01 +00001535 if( p && type!=SQLITE_NULL ){
drh739105c2002-05-29 23:22:23 +00001536 p->cnt++;
drh29d72102006-02-09 22:13:41 +00001537 if( type==SQLITE_INTEGER ){
drh8c08e862006-02-11 17:34:00 +00001538 i64 v = sqlite3_value_int64(argv[0]);
1539 p->rSum += v;
drh158b9cb2011-03-05 20:59:46 +00001540 if( (p->approx|p->overflow)==0 && sqlite3AddInt64(&p->iSum, v) ){
drha546ef22018-07-07 20:55:16 +00001541 p->approx = p->overflow = 1;
drh29d72102006-02-09 22:13:41 +00001542 }
1543 }else{
drh8c08e862006-02-11 17:34:00 +00001544 p->rSum += sqlite3_value_double(argv[0]);
drh29d72102006-02-09 22:13:41 +00001545 p->approx = 1;
drh3f219f42005-09-08 19:45:57 +00001546 }
drh739105c2002-05-29 23:22:23 +00001547 }
drhdd5baa92002-02-27 19:50:59 +00001548}
dan67a9b8e2018-06-22 20:51:35 +00001549#ifndef SQLITE_OMIT_WINDOWFUNC
danc3a20c12018-05-23 20:55:37 +00001550static void sumInverse(sqlite3_context *context, int argc, sqlite3_value**argv){
1551 SumCtx *p;
1552 int type;
1553 assert( argc==1 );
1554 UNUSED_PARAMETER(argc);
1555 p = sqlite3_aggregate_context(context, sizeof(*p));
1556 type = sqlite3_value_numeric_type(argv[0]);
drhfd4b7282018-07-07 19:47:21 +00001557 /* p is always non-NULL because sumStep() will have been called first
1558 ** to initialize it */
1559 if( ALWAYS(p) && type!=SQLITE_NULL ){
drha546ef22018-07-07 20:55:16 +00001560 assert( p->cnt>0 );
danc3a20c12018-05-23 20:55:37 +00001561 p->cnt--;
drha546ef22018-07-07 20:55:16 +00001562 assert( type==SQLITE_INTEGER || p->approx );
1563 if( type==SQLITE_INTEGER && p->approx==0 ){
danc3a20c12018-05-23 20:55:37 +00001564 i64 v = sqlite3_value_int64(argv[0]);
1565 p->rSum -= v;
drha546ef22018-07-07 20:55:16 +00001566 p->iSum -= v;
danc3a20c12018-05-23 20:55:37 +00001567 }else{
dand7368292018-07-02 17:45:59 +00001568 p->rSum -= sqlite3_value_double(argv[0]);
danc3a20c12018-05-23 20:55:37 +00001569 }
1570 }
1571}
dan67a9b8e2018-06-22 20:51:35 +00001572#else
1573# define sumInverse 0
1574#endif /* SQLITE_OMIT_WINDOWFUNC */
danielk19770ae8b832004-05-25 12:05:56 +00001575static void sumFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001576 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001577 p = sqlite3_aggregate_context(context, 0);
drhc2bd9132005-09-08 20:37:43 +00001578 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001579 if( p->overflow ){
1580 sqlite3_result_error(context,"integer overflow",-1);
1581 }else if( p->approx ){
1582 sqlite3_result_double(context, p->rSum);
drhc2bd9132005-09-08 20:37:43 +00001583 }else{
drh8c08e862006-02-11 17:34:00 +00001584 sqlite3_result_int64(context, p->iSum);
drhc2bd9132005-09-08 20:37:43 +00001585 }
drh3d1d95e2005-09-08 10:37:01 +00001586 }
drhdd5baa92002-02-27 19:50:59 +00001587}
danielk19770ae8b832004-05-25 12:05:56 +00001588static void avgFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001589 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001590 p = sqlite3_aggregate_context(context, 0);
drh739105c2002-05-29 23:22:23 +00001591 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001592 sqlite3_result_double(context, p->rSum/(double)p->cnt);
drhdd5baa92002-02-27 19:50:59 +00001593 }
1594}
drha97fdd32006-01-12 22:17:50 +00001595static void totalFinalize(sqlite3_context *context){
1596 SumCtx *p;
1597 p = sqlite3_aggregate_context(context, 0);
shanefbd60f82009-02-04 03:59:25 +00001598 /* (double)0 In case of SQLITE_OMIT_FLOATING_POINT... */
1599 sqlite3_result_double(context, p ? p->rSum : (double)0);
drha97fdd32006-01-12 22:17:50 +00001600}
drhdd5baa92002-02-27 19:50:59 +00001601
1602/*
drh0bce8352002-02-28 00:41:10 +00001603** The following structure keeps track of state information for the
1604** count() aggregate function.
1605*/
1606typedef struct CountCtx CountCtx;
1607struct CountCtx {
drhfc6ad392006-02-09 13:38:19 +00001608 i64 n;
dan7262ca92018-07-02 12:07:32 +00001609#ifdef SQLITE_DEBUG
1610 int bInverse; /* True if xInverse() ever called */
1611#endif
drh0bce8352002-02-28 00:41:10 +00001612};
drhdd5baa92002-02-27 19:50:59 +00001613
drh0bce8352002-02-28 00:41:10 +00001614/*
1615** Routines to implement the count() aggregate function.
1616*/
danielk19770ae8b832004-05-25 12:05:56 +00001617static void countStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drh0bce8352002-02-28 00:41:10 +00001618 CountCtx *p;
drh4f26d6c2004-05-26 23:25:30 +00001619 p = sqlite3_aggregate_context(context, sizeof(*p));
drh9c054832004-05-31 18:51:57 +00001620 if( (argc==0 || SQLITE_NULL!=sqlite3_value_type(argv[0])) && p ){
drh0bce8352002-02-28 00:41:10 +00001621 p->n++;
1622 }
drh2e79c3d2009-04-08 23:04:14 +00001623
drhd3264c72009-04-15 13:39:47 +00001624#ifndef SQLITE_OMIT_DEPRECATED
drh2e79c3d2009-04-08 23:04:14 +00001625 /* The sqlite3_aggregate_count() function is deprecated. But just to make
1626 ** sure it still operates correctly, verify that its count agrees with our
1627 ** internal count when using count(*) and when the total count can be
1628 ** expressed as a 32-bit integer. */
dan7262ca92018-07-02 12:07:32 +00001629 assert( argc==1 || p==0 || p->n>0x7fffffff || p->bInverse
drh2e79c3d2009-04-08 23:04:14 +00001630 || p->n==sqlite3_aggregate_count(context) );
drhd3264c72009-04-15 13:39:47 +00001631#endif
drh0bce8352002-02-28 00:41:10 +00001632}
danielk19770ae8b832004-05-25 12:05:56 +00001633static void countFinalize(sqlite3_context *context){
drh0bce8352002-02-28 00:41:10 +00001634 CountCtx *p;
drhabfcea22005-09-06 20:36:48 +00001635 p = sqlite3_aggregate_context(context, 0);
drhfc6ad392006-02-09 13:38:19 +00001636 sqlite3_result_int64(context, p ? p->n : 0);
drh0bce8352002-02-28 00:41:10 +00001637}
dan7262ca92018-07-02 12:07:32 +00001638#ifndef SQLITE_OMIT_WINDOWFUNC
1639static void countInverse(sqlite3_context *ctx, int argc, sqlite3_value **argv){
1640 CountCtx *p;
1641 p = sqlite3_aggregate_context(ctx, sizeof(*p));
drhfd4b7282018-07-07 19:47:21 +00001642 /* p is always non-NULL since countStep() will have been called first */
1643 if( (argc==0 || SQLITE_NULL!=sqlite3_value_type(argv[0])) && ALWAYS(p) ){
dan7262ca92018-07-02 12:07:32 +00001644 p->n--;
1645#ifdef SQLITE_DEBUG
1646 p->bInverse = 1;
1647#endif
1648 }
1649}
dan6b4b8822018-07-02 15:03:50 +00001650#else
1651# define countInverse 0
1652#endif /* SQLITE_OMIT_WINDOWFUNC */
drh0bce8352002-02-28 00:41:10 +00001653
1654/*
drh0bce8352002-02-28 00:41:10 +00001655** Routines to implement min() and max() aggregate functions.
1656*/
danielk197762c14b32008-11-19 09:05:26 +00001657static void minmaxStep(
1658 sqlite3_context *context,
1659 int NotUsed,
1660 sqlite3_value **argv
1661){
danielk197788208052004-05-25 01:13:20 +00001662 Mem *pArg = (Mem *)argv[0];
drh9eb516c2004-07-18 20:52:32 +00001663 Mem *pBest;
danielk197762c14b32008-11-19 09:05:26 +00001664 UNUSED_PARAMETER(NotUsed);
drh9eb516c2004-07-18 20:52:32 +00001665
drh9eb516c2004-07-18 20:52:32 +00001666 pBest = (Mem *)sqlite3_aggregate_context(context, sizeof(*pBest));
danielk19773aeab9e2004-06-24 00:20:04 +00001667 if( !pBest ) return;
drh268380c2004-02-25 13:47:31 +00001668
dan6fb2b542018-06-19 17:13:11 +00001669 if( sqlite3_value_type(pArg)==SQLITE_NULL ){
drh94a6d992012-02-02 18:42:09 +00001670 if( pBest->flags ) sqlite3SkipAccumulatorLoad(context);
1671 }else if( pBest->flags ){
drh9eb516c2004-07-18 20:52:32 +00001672 int max;
1673 int cmp;
danielk1977dc1bdc42004-06-11 10:51:27 +00001674 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk19777e18c252004-05-25 11:47:24 +00001675 /* This step function is used for both the min() and max() aggregates,
1676 ** the only difference between the two being that the sense of the
1677 ** comparison is inverted. For the max() aggregate, the
1678 ** sqlite3_user_data() function returns (void *)-1. For min() it
1679 ** returns (void *)db, where db is the sqlite3* database pointer.
1680 ** Therefore the next statement sets variable 'max' to 1 for the max()
1681 ** aggregate, or 0 for min().
1682 */
drh309b3382007-03-17 17:52:42 +00001683 max = sqlite3_user_data(context)!=0;
danielk1977dc1bdc42004-06-11 10:51:27 +00001684 cmp = sqlite3MemCompare(pBest, pArg, pColl);
danielk197788208052004-05-25 01:13:20 +00001685 if( (max && cmp<0) || (!max && cmp>0) ){
drhb21c8cd2007-08-21 19:33:56 +00001686 sqlite3VdbeMemCopy(pBest, pArg);
drh7a957892012-02-02 17:35:43 +00001687 }else{
1688 sqlite3SkipAccumulatorLoad(context);
danielk197788208052004-05-25 01:13:20 +00001689 }
drh268380c2004-02-25 13:47:31 +00001690 }else{
drh035e5632014-09-16 14:16:31 +00001691 pBest->db = sqlite3_context_db_handle(context);
drhb21c8cd2007-08-21 19:33:56 +00001692 sqlite3VdbeMemCopy(pBest, pArg);
drh0bce8352002-02-28 00:41:10 +00001693 }
1694}
dan6fb2b542018-06-19 17:13:11 +00001695static void minMaxValueFinalize(sqlite3_context *context, int bValue){
danielk197788208052004-05-25 01:13:20 +00001696 sqlite3_value *pRes;
drhabfcea22005-09-06 20:36:48 +00001697 pRes = (sqlite3_value *)sqlite3_aggregate_context(context, 0);
1698 if( pRes ){
drh94a6d992012-02-02 18:42:09 +00001699 if( pRes->flags ){
drhabfcea22005-09-06 20:36:48 +00001700 sqlite3_result_value(context, pRes);
1701 }
dan6fb2b542018-06-19 17:13:11 +00001702 if( bValue==0 ) sqlite3VdbeMemRelease(pRes);
drh0bce8352002-02-28 00:41:10 +00001703 }
1704}
dan67a9b8e2018-06-22 20:51:35 +00001705#ifndef SQLITE_OMIT_WINDOWFUNC
dan6fb2b542018-06-19 17:13:11 +00001706static void minMaxValue(sqlite3_context *context){
drhc7bf5712018-07-09 22:49:01 +00001707 minMaxValueFinalize(context, 1);
dan6fb2b542018-06-19 17:13:11 +00001708}
dan67a9b8e2018-06-22 20:51:35 +00001709#else
1710# define minMaxValue 0
1711#endif /* SQLITE_OMIT_WINDOWFUNC */
dan6fb2b542018-06-19 17:13:11 +00001712static void minMaxFinalize(sqlite3_context *context){
drhc7bf5712018-07-09 22:49:01 +00001713 minMaxValueFinalize(context, 0);
dan6fb2b542018-06-19 17:13:11 +00001714}
drhdd5baa92002-02-27 19:50:59 +00001715
drhb0689692007-11-01 17:38:30 +00001716/*
1717** group_concat(EXPR, ?SEPARATOR?)
1718*/
1719static void groupConcatStep(
1720 sqlite3_context *context,
1721 int argc,
1722 sqlite3_value **argv
1723){
1724 const char *zVal;
drhade86482007-11-28 22:36:40 +00001725 StrAccum *pAccum;
drhb0689692007-11-01 17:38:30 +00001726 const char *zSep;
drh07d31172009-02-02 21:57:05 +00001727 int nVal, nSep;
1728 assert( argc==1 || argc==2 );
1729 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drhade86482007-11-28 22:36:40 +00001730 pAccum = (StrAccum*)sqlite3_aggregate_context(context, sizeof(*pAccum));
1731
1732 if( pAccum ){
drhbb4957f2008-03-20 14:03:29 +00001733 sqlite3 *db = sqlite3_context_db_handle(context);
dane3bf6322018-06-08 20:58:27 +00001734 int firstTerm = pAccum->mxAlloc==0;
drhbb4957f2008-03-20 14:03:29 +00001735 pAccum->mxAlloc = db->aLimit[SQLITE_LIMIT_LENGTH];
drh8bfd7192009-06-19 16:44:41 +00001736 if( !firstTerm ){
drh07d31172009-02-02 21:57:05 +00001737 if( argc==2 ){
1738 zSep = (char*)sqlite3_value_text(argv[1]);
1739 nSep = sqlite3_value_bytes(argv[1]);
drhb0689692007-11-01 17:38:30 +00001740 }else{
1741 zSep = ",";
drhade86482007-11-28 22:36:40 +00001742 nSep = 1;
drhb0689692007-11-01 17:38:30 +00001743 }
drh0cdbe1a2018-05-09 13:46:26 +00001744 if( zSep ) sqlite3_str_append(pAccum, zSep, nSep);
drhb0689692007-11-01 17:38:30 +00001745 }
drh07d31172009-02-02 21:57:05 +00001746 zVal = (char*)sqlite3_value_text(argv[0]);
1747 nVal = sqlite3_value_bytes(argv[0]);
drh0cdbe1a2018-05-09 13:46:26 +00001748 if( zVal ) sqlite3_str_append(pAccum, zVal, nVal);
drhb0689692007-11-01 17:38:30 +00001749 }
1750}
dan67a9b8e2018-06-22 20:51:35 +00001751#ifndef SQLITE_OMIT_WINDOWFUNC
dan03854d22018-06-08 11:45:28 +00001752static void groupConcatInverse(
1753 sqlite3_context *context,
1754 int argc,
1755 sqlite3_value **argv
1756){
1757 int n;
dan03854d22018-06-08 11:45:28 +00001758 StrAccum *pAccum;
drhc7bf5712018-07-09 22:49:01 +00001759 assert( argc==1 || argc==2 );
dan03854d22018-06-08 11:45:28 +00001760 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
1761 pAccum = (StrAccum*)sqlite3_aggregate_context(context, sizeof(*pAccum));
drhfd4b7282018-07-07 19:47:21 +00001762 /* pAccum is always non-NULL since groupConcatStep() will have always
1763 ** run frist to initialize it */
1764 if( ALWAYS(pAccum) ){
dan03854d22018-06-08 11:45:28 +00001765 n = sqlite3_value_bytes(argv[0]);
1766 if( argc==2 ){
1767 n += sqlite3_value_bytes(argv[1]);
dan683b0ff2018-07-05 18:19:29 +00001768 }else{
1769 n++;
dan03854d22018-06-08 11:45:28 +00001770 }
drhc7bf5712018-07-09 22:49:01 +00001771 if( n>=(int)pAccum->nChar ){
dan03854d22018-06-08 11:45:28 +00001772 pAccum->nChar = 0;
1773 }else{
1774 pAccum->nChar -= n;
1775 memmove(pAccum->zText, &pAccum->zText[n], pAccum->nChar);
1776 }
dane3bf6322018-06-08 20:58:27 +00001777 if( pAccum->nChar==0 ) pAccum->mxAlloc = 0;
dan03854d22018-06-08 11:45:28 +00001778 }
1779}
dan67a9b8e2018-06-22 20:51:35 +00001780#else
1781# define groupConcatInverse 0
1782#endif /* SQLITE_OMIT_WINDOWFUNC */
drhb0689692007-11-01 17:38:30 +00001783static void groupConcatFinalize(sqlite3_context *context){
drhade86482007-11-28 22:36:40 +00001784 StrAccum *pAccum;
1785 pAccum = sqlite3_aggregate_context(context, 0);
1786 if( pAccum ){
drh0cdbe1a2018-05-09 13:46:26 +00001787 if( pAccum->accError==SQLITE_TOOBIG ){
drhade86482007-11-28 22:36:40 +00001788 sqlite3_result_error_toobig(context);
drh0cdbe1a2018-05-09 13:46:26 +00001789 }else if( pAccum->accError==SQLITE_NOMEM ){
drhade86482007-11-28 22:36:40 +00001790 sqlite3_result_error_nomem(context);
1791 }else{
1792 sqlite3_result_text(context, sqlite3StrAccumFinish(pAccum), -1,
1793 sqlite3_free);
1794 }
drhb0689692007-11-01 17:38:30 +00001795 }
1796}
dan67a9b8e2018-06-22 20:51:35 +00001797#ifndef SQLITE_OMIT_WINDOWFUNC
dane2f781b2018-05-17 19:24:08 +00001798static void groupConcatValue(sqlite3_context *context){
1799 sqlite3_str *pAccum;
1800 pAccum = (sqlite3_str*)sqlite3_aggregate_context(context, 0);
1801 if( pAccum ){
1802 if( pAccum->accError==SQLITE_TOOBIG ){
1803 sqlite3_result_error_toobig(context);
1804 }else if( pAccum->accError==SQLITE_NOMEM ){
1805 sqlite3_result_error_nomem(context);
1806 }else{
1807 const char *zText = sqlite3_str_value(pAccum);
1808 sqlite3_result_text(context, zText, -1, SQLITE_TRANSIENT);
1809 }
1810 }
1811}
dan67a9b8e2018-06-22 20:51:35 +00001812#else
1813# define groupConcatValue 0
1814#endif /* SQLITE_OMIT_WINDOWFUNC */
drh4e5ffc52004-08-31 00:52:37 +00001815
drhd3a149e2002-02-24 17:12:53 +00001816/*
drha4741842010-04-25 20:58:37 +00001817** This routine does per-connection function registration. Most
1818** of the built-in functions above are part of the global function set.
1819** This routine only deals with those that are not global.
drhdc04c582002-02-24 01:55:15 +00001820*/
drh80738d92016-02-15 00:34:16 +00001821void sqlite3RegisterPerConnectionBuiltinFunctions(sqlite3 *db){
drha4741842010-04-25 20:58:37 +00001822 int rc = sqlite3_overload_function(db, "MATCH", 2);
1823 assert( rc==SQLITE_NOMEM || rc==SQLITE_OK );
1824 if( rc==SQLITE_NOMEM ){
drh4a642b62016-02-05 01:55:27 +00001825 sqlite3OomFault(db);
danielk1977832a58a2007-06-22 15:21:15 +00001826 }
drh55ef4d92005-08-14 01:20:37 +00001827}
1828
1829/*
drh08652b52019-05-08 17:27:18 +00001830** Re-register the built-in LIKE functions. The caseSensitive
drh55ef4d92005-08-14 01:20:37 +00001831** parameter determines whether or not the LIKE operator is case
drh08652b52019-05-08 17:27:18 +00001832** sensitive.
drh55ef4d92005-08-14 01:20:37 +00001833*/
1834void sqlite3RegisterLikeFunctions(sqlite3 *db, int caseSensitive){
1835 struct compareInfo *pInfo;
drhea5c0402019-05-08 19:32:33 +00001836 int flags;
drh55ef4d92005-08-14 01:20:37 +00001837 if( caseSensitive ){
1838 pInfo = (struct compareInfo*)&likeInfoAlt;
drhea5c0402019-05-08 19:32:33 +00001839 flags = SQLITE_FUNC_LIKE | SQLITE_FUNC_CASE;
drh55ef4d92005-08-14 01:20:37 +00001840 }else{
1841 pInfo = (struct compareInfo*)&likeInfoNorm;
drhea5c0402019-05-08 19:32:33 +00001842 flags = SQLITE_FUNC_LIKE;
drh55ef4d92005-08-14 01:20:37 +00001843 }
dan660af932018-06-18 16:55:22 +00001844 sqlite3CreateFunc(db, "like", 2, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0, 0, 0);
1845 sqlite3CreateFunc(db, "like", 3, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0, 0, 0);
drhea5c0402019-05-08 19:32:33 +00001846 sqlite3FindFunction(db, "like", 2, SQLITE_UTF8, 0)->funcFlags |= flags;
1847 sqlite3FindFunction(db, "like", 3, SQLITE_UTF8, 0)->funcFlags |= flags;
drh55ef4d92005-08-14 01:20:37 +00001848}
1849
1850/*
1851** pExpr points to an expression which implements a function. If
1852** it is appropriate to apply the LIKE optimization to that function
drh1d42ea72017-07-27 20:24:29 +00001853** then set aWc[0] through aWc[2] to the wildcard characters and the
1854** escape character and then return TRUE. If the function is not a
1855** LIKE-style function then return FALSE.
1856**
1857** The expression "a LIKE b ESCAPE c" is only considered a valid LIKE
1858** operator if c is a string literal that is exactly one byte in length.
1859** That one byte is stored in aWc[3]. aWc[3] is set to zero if there is
1860** no ESCAPE clause.
drh16897072015-03-07 00:57:37 +00001861**
1862** *pIsNocase is set to true if uppercase and lowercase are equivalent for
1863** the function (default for LIKE). If the function makes the distinction
1864** between uppercase and lowercase (as does GLOB) then *pIsNocase is set to
1865** false.
drh55ef4d92005-08-14 01:20:37 +00001866*/
drhd64fe2f2005-08-28 17:00:23 +00001867int sqlite3IsLikeFunction(sqlite3 *db, Expr *pExpr, int *pIsNocase, char *aWc){
drh55ef4d92005-08-14 01:20:37 +00001868 FuncDef *pDef;
drh1d42ea72017-07-27 20:24:29 +00001869 int nExpr;
1870 if( pExpr->op!=TK_FUNCTION || !pExpr->x.pList ){
drh55ef4d92005-08-14 01:20:37 +00001871 return 0;
1872 }
danielk19776ab3a2e2009-02-19 14:39:25 +00001873 assert( !ExprHasProperty(pExpr, EP_xIsSelect) );
drh1d42ea72017-07-27 20:24:29 +00001874 nExpr = pExpr->x.pList->nExpr;
1875 pDef = sqlite3FindFunction(db, pExpr->u.zToken, nExpr, SQLITE_UTF8, 0);
drh78b52202020-01-22 23:08:19 +00001876#ifdef SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION
1877 if( pDef==0 ) return 0;
1878#endif
drhd36e1042013-09-06 13:10:12 +00001879 if( NEVER(pDef==0) || (pDef->funcFlags & SQLITE_FUNC_LIKE)==0 ){
drh55ef4d92005-08-14 01:20:37 +00001880 return 0;
1881 }
1882
1883 /* The memcpy() statement assumes that the wildcard characters are
1884 ** the first three statements in the compareInfo structure. The
1885 ** asserts() that follow verify that assumption
1886 */
1887 memcpy(aWc, pDef->pUserData, 3);
1888 assert( (char*)&likeInfoAlt == (char*)&likeInfoAlt.matchAll );
1889 assert( &((char*)&likeInfoAlt)[1] == (char*)&likeInfoAlt.matchOne );
1890 assert( &((char*)&likeInfoAlt)[2] == (char*)&likeInfoAlt.matchSet );
drh589c7872020-03-19 18:13:28 +00001891
1892 if( nExpr<3 ){
1893 aWc[3] = 0;
1894 }else{
1895 Expr *pEscape = pExpr->x.pList->a[2].pExpr;
1896 char *zEscape;
1897 if( pEscape->op!=TK_STRING ) return 0;
1898 zEscape = pEscape->u.zToken;
1899 if( zEscape[0]==0 || zEscape[1]!=0 ) return 0;
1900 if( zEscape[0]==aWc[0] ) return 0;
1901 if( zEscape[0]==aWc[1] ) return 0;
1902 aWc[3] = zEscape[0];
1903 }
1904
drhd36e1042013-09-06 13:10:12 +00001905 *pIsNocase = (pDef->funcFlags & SQLITE_FUNC_CASE)==0;
drh55ef4d92005-08-14 01:20:37 +00001906 return 1;
drhdc04c582002-02-24 01:55:15 +00001907}
danielk19778c0a7912008-08-20 14:49:23 +00001908
drh70a8ca32008-08-21 18:49:27 +00001909/*
peter.d.reid60ec9142014-09-06 16:39:46 +00001910** All of the FuncDef structures in the aBuiltinFunc[] array above
drh777c5382008-08-21 20:21:34 +00001911** to the global function hash table. This occurs at start-time (as
1912** a consequence of calling sqlite3_initialize()).
1913**
1914** After this routine runs
drh70a8ca32008-08-21 18:49:27 +00001915*/
drh80738d92016-02-15 00:34:16 +00001916void sqlite3RegisterBuiltinFunctions(void){
danielk197793ce7412008-09-01 19:14:02 +00001917 /*
1918 ** The following array holds FuncDef structures for all of the functions
1919 ** defined in this file.
1920 **
1921 ** The array cannot be constant since changes are made to the
1922 ** FuncDef.pHash elements at start-time. The elements of this array
1923 ** are read-only after initialization is complete.
drh80738d92016-02-15 00:34:16 +00001924 **
1925 ** For peak efficiency, put the most frequently used function last.
danielk197793ce7412008-09-01 19:14:02 +00001926 */
drh80738d92016-02-15 00:34:16 +00001927 static FuncDef aBuiltinFunc[] = {
drh171c50e2020-01-01 15:43:30 +00001928/***** Functions only available with SQLITE_TESTCTRL_INTERNAL_FUNCTIONS *****/
1929 TEST_FUNC(implies_nonnull_row, 2, INLINEFUNC_implies_nonnull_row, 0),
1930 TEST_FUNC(expr_compare, 2, INLINEFUNC_expr_compare, 0),
1931 TEST_FUNC(expr_implies_expr, 2, INLINEFUNC_expr_implies_expr, 0),
1932#ifdef SQLITE_DEBUG
1933 TEST_FUNC(affinity, 1, INLINEFUNC_affinity, 0),
1934#endif
1935/***** Regular functions *****/
drh80738d92016-02-15 00:34:16 +00001936#ifdef SQLITE_SOUNDEX
1937 FUNCTION(soundex, 1, 0, 0, soundexFunc ),
1938#endif
1939#ifndef SQLITE_OMIT_LOAD_EXTENSION
drh64de2a52019-12-31 18:39:23 +00001940 SFUNCTION(load_extension, 1, 0, 0, loadExt ),
1941 SFUNCTION(load_extension, 2, 0, 0, loadExt ),
drh80738d92016-02-15 00:34:16 +00001942#endif
1943#if SQLITE_USER_AUTHENTICATION
1944 FUNCTION(sqlite_crypt, 2, 0, 0, sqlite3CryptFunc ),
1945#endif
1946#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
1947 DFUNCTION(sqlite_compileoption_used,1, 0, 0, compileoptionusedFunc ),
1948 DFUNCTION(sqlite_compileoption_get, 1, 0, 0, compileoptiongetFunc ),
1949#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
drh25c42962020-01-01 13:55:08 +00001950 INLINE_FUNC(unlikely, 1, INLINEFUNC_unlikely, SQLITE_FUNC_UNLIKELY),
1951 INLINE_FUNC(likelihood, 2, INLINEFUNC_unlikely, SQLITE_FUNC_UNLIKELY),
1952 INLINE_FUNC(likely, 1, INLINEFUNC_unlikely, SQLITE_FUNC_UNLIKELY),
drh092457b2017-12-29 15:04:49 +00001953#ifdef SQLITE_ENABLE_OFFSET_SQL_FUNC
drh35100fb2018-01-04 19:20:37 +00001954 FUNCTION2(sqlite_offset, 1, 0, 0, noopFunc, SQLITE_FUNC_OFFSET|
drh2fc865c2017-12-16 20:20:37 +00001955 SQLITE_FUNC_TYPEOF),
drh092457b2017-12-29 15:04:49 +00001956#endif
danielk197793ce7412008-09-01 19:14:02 +00001957 FUNCTION(ltrim, 1, 1, 0, trimFunc ),
1958 FUNCTION(ltrim, 2, 1, 0, trimFunc ),
1959 FUNCTION(rtrim, 1, 2, 0, trimFunc ),
1960 FUNCTION(rtrim, 2, 2, 0, trimFunc ),
1961 FUNCTION(trim, 1, 3, 0, trimFunc ),
1962 FUNCTION(trim, 2, 3, 0, trimFunc ),
1963 FUNCTION(min, -1, 0, 1, minmaxFunc ),
1964 FUNCTION(min, 0, 0, 1, 0 ),
dan6fb2b542018-06-19 17:13:11 +00001965 WAGGREGATE(min, 1, 0, 1, minmaxStep, minMaxFinalize, minMaxValue, 0,
drh9588ad92014-09-15 14:46:02 +00001966 SQLITE_FUNC_MINMAX ),
danielk197793ce7412008-09-01 19:14:02 +00001967 FUNCTION(max, -1, 1, 1, minmaxFunc ),
1968 FUNCTION(max, 0, 1, 1, 0 ),
dan6fb2b542018-06-19 17:13:11 +00001969 WAGGREGATE(max, 1, 1, 1, minmaxStep, minMaxFinalize, minMaxValue, 0,
drh9588ad92014-09-15 14:46:02 +00001970 SQLITE_FUNC_MINMAX ),
drha748fdc2012-03-28 01:34:47 +00001971 FUNCTION2(typeof, 1, 0, 0, typeofFunc, SQLITE_FUNC_TYPEOF),
1972 FUNCTION2(length, 1, 0, 0, lengthFunc, SQLITE_FUNC_LENGTH),
drhd55e0722012-10-25 03:07:29 +00001973 FUNCTION(instr, 2, 0, 0, instrFunc ),
drha5c14162013-12-17 15:03:06 +00001974 FUNCTION(printf, -1, 0, 0, printfFunc ),
drhd495d8c2013-02-22 19:34:25 +00001975 FUNCTION(unicode, 1, 0, 0, unicodeFunc ),
1976 FUNCTION(char, -1, 0, 0, charFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001977 FUNCTION(abs, 1, 0, 0, absFunc ),
shanefbd60f82009-02-04 03:59:25 +00001978#ifndef SQLITE_OMIT_FLOATING_POINT
danielk197793ce7412008-09-01 19:14:02 +00001979 FUNCTION(round, 1, 0, 0, roundFunc ),
1980 FUNCTION(round, 2, 0, 0, roundFunc ),
shanefbd60f82009-02-04 03:59:25 +00001981#endif
danielk197793ce7412008-09-01 19:14:02 +00001982 FUNCTION(upper, 1, 0, 0, upperFunc ),
1983 FUNCTION(lower, 1, 0, 0, lowerFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001984 FUNCTION(hex, 1, 0, 0, hexFunc ),
drhffe421c2020-05-13 17:26:38 +00001985 INLINE_FUNC(ifnull, 2, INLINEFUNC_coalesce, 0 ),
drhb1fba282013-11-21 14:33:48 +00001986 VFUNCTION(random, 0, 0, 0, randomFunc ),
1987 VFUNCTION(randomblob, 1, 0, 0, randomBlob ),
danielk197793ce7412008-09-01 19:14:02 +00001988 FUNCTION(nullif, 2, 0, 1, nullifFunc ),
drh03bf26d2015-08-31 21:16:36 +00001989 DFUNCTION(sqlite_version, 0, 0, 0, versionFunc ),
1990 DFUNCTION(sqlite_source_id, 0, 0, 0, sourceidFunc ),
drh840561f2011-04-27 18:08:42 +00001991 FUNCTION(sqlite_log, 2, 0, 0, errlogFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001992 FUNCTION(quote, 1, 0, 0, quoteFunc ),
drhb1fba282013-11-21 14:33:48 +00001993 VFUNCTION(last_insert_rowid, 0, 0, 0, last_insert_rowid),
1994 VFUNCTION(changes, 0, 0, 0, changes ),
1995 VFUNCTION(total_changes, 0, 0, 0, total_changes ),
danielk197793ce7412008-09-01 19:14:02 +00001996 FUNCTION(replace, 3, 0, 0, replaceFunc ),
1997 FUNCTION(zeroblob, 1, 0, 0, zeroblobFunc ),
drh80738d92016-02-15 00:34:16 +00001998 FUNCTION(substr, 2, 0, 0, substrFunc ),
1999 FUNCTION(substr, 3, 0, 0, substrFunc ),
dan6fb2b542018-06-19 17:13:11 +00002000 WAGGREGATE(sum, 1,0,0, sumStep, sumFinalize, sumFinalize, sumInverse, 0),
2001 WAGGREGATE(total, 1,0,0, sumStep,totalFinalize,totalFinalize,sumInverse, 0),
2002 WAGGREGATE(avg, 1,0,0, sumStep, avgFinalize, avgFinalize, sumInverse, 0),
dan7262ca92018-07-02 12:07:32 +00002003 WAGGREGATE(count, 0,0,0, countStep,
2004 countFinalize, countFinalize, countInverse, SQLITE_FUNC_COUNT ),
2005 WAGGREGATE(count, 1,0,0, countStep,
2006 countFinalize, countFinalize, countInverse, 0 ),
dan03854d22018-06-08 11:45:28 +00002007 WAGGREGATE(group_concat, 1, 0, 0, groupConcatStep,
dan6fb2b542018-06-19 17:13:11 +00002008 groupConcatFinalize, groupConcatValue, groupConcatInverse, 0),
dan03854d22018-06-08 11:45:28 +00002009 WAGGREGATE(group_concat, 2, 0, 0, groupConcatStep,
dan6fb2b542018-06-19 17:13:11 +00002010 groupConcatFinalize, groupConcatValue, groupConcatInverse, 0),
danielk197793ce7412008-09-01 19:14:02 +00002011
2012 LIKEFUNC(glob, 2, &globInfo, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
drhcc153132016-08-04 12:35:17 +00002013#ifdef SQLITE_CASE_SENSITIVE_LIKE
danielk197793ce7412008-09-01 19:14:02 +00002014 LIKEFUNC(like, 2, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
2015 LIKEFUNC(like, 3, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
drhcc153132016-08-04 12:35:17 +00002016#else
danielk197793ce7412008-09-01 19:14:02 +00002017 LIKEFUNC(like, 2, &likeInfoNorm, SQLITE_FUNC_LIKE),
2018 LIKEFUNC(like, 3, &likeInfoNorm, SQLITE_FUNC_LIKE),
drhcc153132016-08-04 12:35:17 +00002019#endif
2020#ifdef SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION
2021 FUNCTION(unknown, -1, 0, 0, unknownFunc ),
2022#endif
drh80738d92016-02-15 00:34:16 +00002023 FUNCTION(coalesce, 1, 0, 0, 0 ),
2024 FUNCTION(coalesce, 0, 0, 0, 0 ),
drhffe421c2020-05-13 17:26:38 +00002025 INLINE_FUNC(coalesce, -1, INLINEFUNC_coalesce, 0 ),
drh3c0e6062020-05-13 18:03:34 +00002026 INLINE_FUNC(iif, 3, INLINEFUNC_iif, 0 ),
danielk197793ce7412008-09-01 19:14:02 +00002027 };
drh545f5872010-04-24 14:02:59 +00002028#ifndef SQLITE_OMIT_ALTERTABLE
2029 sqlite3AlterFunctions();
2030#endif
dandfa552f2018-06-02 21:04:28 +00002031 sqlite3WindowFunctions();
drh80738d92016-02-15 00:34:16 +00002032 sqlite3RegisterDateTimeFunctions();
2033 sqlite3InsertBuiltinFuncs(aBuiltinFunc, ArraySize(aBuiltinFunc));
2034
2035#if 0 /* Enable to print out how the built-in functions are hashed */
2036 {
2037 int i;
2038 FuncDef *p;
2039 for(i=0; i<SQLITE_FUNC_HASH_SZ; i++){
2040 printf("FUNC-HASH %02d:", i);
2041 for(p=sqlite3BuiltinFunctions.a[i]; p; p=p->u.pHash){
2042 int n = sqlite3Strlen30(p->zName);
2043 int h = p->zName[0] + n;
2044 printf(" %s(%d)", p->zName, h);
2045 }
2046 printf("\n");
2047 }
2048 }
2049#endif
drh70a8ca32008-08-21 18:49:27 +00002050}