blob: aedbda6f362ace0f52deff7362b277331e814ef2 [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;
danielk197762c14b32008-11-19 09:05:26 +0000514 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drh2fa18682008-03-19 14:15:34 +0000515 sqlite3_randomness(sizeof(r), &r);
drh3034e3d2009-04-02 14:05:21 +0000516 if( r<0 ){
517 /* We need to prevent a random number of 0x8000000000000000
518 ** (or -9223372036854775808) since when you do abs() of that
519 ** number of you get the same value back again. To do this
520 ** in a way that is testable, mask the sign bit off of negative
521 ** values, resulting in a positive value. Then take the
522 ** 2s complement of that positive value. The end result can
523 ** therefore be no less than -9223372036854775807.
524 */
drhaf8001b2012-02-11 19:53:24 +0000525 r = -(r & LARGEST_INT64);
drh3034e3d2009-04-02 14:05:21 +0000526 }
drh52fc8492006-02-23 21:43:55 +0000527 sqlite3_result_int64(context, r);
drhf9ffac92002-03-02 19:00:31 +0000528}
529
530/*
drh137c7282007-01-29 17:58:28 +0000531** Implementation of randomblob(N). Return a random blob
532** that is N bytes long.
drh63cf66f2007-01-29 15:50:05 +0000533*/
drh137c7282007-01-29 17:58:28 +0000534static void randomBlob(
drh63cf66f2007-01-29 15:50:05 +0000535 sqlite3_context *context,
536 int argc,
537 sqlite3_value **argv
538){
drh3cb79202019-01-18 14:53:15 +0000539 sqlite3_int64 n;
drh137c7282007-01-29 17:58:28 +0000540 unsigned char *p;
drh63cf66f2007-01-29 15:50:05 +0000541 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000542 UNUSED_PARAMETER(argc);
drh3cb79202019-01-18 14:53:15 +0000543 n = sqlite3_value_int64(argv[0]);
drh023ae032007-05-08 12:12:16 +0000544 if( n<1 ){
545 n = 1;
546 }
danielk1977a1644fd2007-08-29 12:31:25 +0000547 p = contextMalloc(context, n);
drh02d85832007-05-07 19:31:15 +0000548 if( p ){
drh2fa18682008-03-19 14:15:34 +0000549 sqlite3_randomness(n, p);
drh17435752007-08-16 04:30:38 +0000550 sqlite3_result_blob(context, (char*)p, n, sqlite3_free);
drh02d85832007-05-07 19:31:15 +0000551 }
drh63cf66f2007-01-29 15:50:05 +0000552}
553
554/*
drh6ed41ad2002-04-06 14:10:47 +0000555** Implementation of the last_insert_rowid() SQL function. The return
danielk197724b03fd2004-05-10 10:34:34 +0000556** value is the same as the sqlite3_last_insert_rowid() API function.
drh6ed41ad2002-04-06 14:10:47 +0000557*/
danielk197751ad0ec2004-05-24 12:39:02 +0000558static void last_insert_rowid(
danielk19770ae8b832004-05-25 12:05:56 +0000559 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000560 int NotUsed,
561 sqlite3_value **NotUsed2
danielk197751ad0ec2004-05-24 12:39:02 +0000562){
drhfa4a4b92008-03-19 21:45:51 +0000563 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000564 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000565 /* IMP: R-51513-12026 The last_insert_rowid() SQL function is a
566 ** wrapper around the sqlite3_last_insert_rowid() C/C++ interface
567 ** function. */
drhf9b596e2004-05-26 16:54:42 +0000568 sqlite3_result_int64(context, sqlite3_last_insert_rowid(db));
drh6ed41ad2002-04-06 14:10:47 +0000569}
570
rdcf146a772004-02-25 22:51:06 +0000571/*
drhab2f1f92010-01-11 18:26:42 +0000572** Implementation of the changes() SQL function.
573**
574** IMP: R-62073-11209 The changes() SQL function is a wrapper
575** around the sqlite3_changes() C/C++ function and hence follows the same
576** rules for counting changes.
rdcf146a772004-02-25 22:51:06 +0000577*/
danielk1977b28af712004-06-21 06:50:26 +0000578static void changes(
drhf9b596e2004-05-26 16:54:42 +0000579 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000580 int NotUsed,
581 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000582){
drhfa4a4b92008-03-19 21:45:51 +0000583 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000584 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhf4479502004-05-27 03:12:53 +0000585 sqlite3_result_int(context, sqlite3_changes(db));
rdcb0c374f2004-02-20 22:53:38 +0000586}
rdcf146a772004-02-25 22:51:06 +0000587
588/*
danielk1977b28af712004-06-21 06:50:26 +0000589** Implementation of the total_changes() SQL function. The return value is
590** the same as the sqlite3_total_changes() API function.
rdcf146a772004-02-25 22:51:06 +0000591*/
danielk1977b28af712004-06-21 06:50:26 +0000592static void total_changes(
593 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000594 int NotUsed,
595 sqlite3_value **NotUsed2
danielk197751ad0ec2004-05-24 12:39:02 +0000596){
drhfa4a4b92008-03-19 21:45:51 +0000597 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000598 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000599 /* IMP: R-52756-41993 This function is a wrapper around the
600 ** sqlite3_total_changes() C/C++ interface. */
danielk1977b28af712004-06-21 06:50:26 +0000601 sqlite3_result_int(context, sqlite3_total_changes(db));
rdcb0c374f2004-02-20 22:53:38 +0000602}
603
drh6ed41ad2002-04-06 14:10:47 +0000604/*
drh4e5ffc52004-08-31 00:52:37 +0000605** A structure defining how to do GLOB-style comparisons.
danielk1977d02eb1f2004-06-06 09:44:03 +0000606*/
drh4e5ffc52004-08-31 00:52:37 +0000607struct compareInfo {
drh07e83472016-01-11 03:48:18 +0000608 u8 matchAll; /* "*" or "%" */
609 u8 matchOne; /* "?" or "_" */
610 u8 matchSet; /* "[" or 0 */
611 u8 noCase; /* true to ignore case differences */
danielk1977d02eb1f2004-06-06 09:44:03 +0000612};
drh55ef4d92005-08-14 01:20:37 +0000613
drhb9175ae2007-12-07 18:39:04 +0000614/*
615** For LIKE and GLOB matching on EBCDIC machines, assume that every
drhb0870482015-06-17 13:20:54 +0000616** character is exactly one byte in size. Also, provde the Utf8Read()
617** macro for fast reading of the next character in the common case where
618** the next character is ASCII.
drhb9175ae2007-12-07 18:39:04 +0000619*/
620#if defined(SQLITE_EBCDIC)
drh88b33222014-09-25 03:51:37 +0000621# define sqlite3Utf8Read(A) (*((*A)++))
drhb0870482015-06-17 13:20:54 +0000622# define Utf8Read(A) (*(A++))
drhb9175ae2007-12-07 18:39:04 +0000623#else
drhb0870482015-06-17 13:20:54 +0000624# define Utf8Read(A) (A[0]<0x80?*(A++):sqlite3Utf8Read(&A))
drhb9175ae2007-12-07 18:39:04 +0000625#endif
626
drh4e5ffc52004-08-31 00:52:37 +0000627static const struct compareInfo globInfo = { '*', '?', '[', 0 };
drh55ef4d92005-08-14 01:20:37 +0000628/* The correct SQL-92 behavior is for the LIKE operator to ignore
629** case. Thus 'a' LIKE 'A' would be true. */
630static const struct compareInfo likeInfoNorm = { '%', '_', 0, 1 };
631/* If SQLITE_CASE_SENSITIVE_LIKE is defined, then the LIKE operator
632** is case sensitive causing 'a' LIKE 'A' to be false */
633static const struct compareInfo likeInfoAlt = { '%', '_', 0, 0 };
danielk1977d02eb1f2004-06-06 09:44:03 +0000634
635/*
drh698a01c2016-12-01 18:49:40 +0000636** Possible error returns from patternMatch()
637*/
638#define SQLITE_MATCH 0
639#define SQLITE_NOMATCH 1
640#define SQLITE_NOWILDCARDMATCH 2
641
642/*
643** Compare two UTF-8 strings for equality where the first string is
644** a GLOB or LIKE expression. Return values:
645**
646** SQLITE_MATCH: Match
647** SQLITE_NOMATCH: No match
648** SQLITE_NOWILDCARDMATCH: No match in spite of having * or % wildcards.
drh0ac65892002-04-20 14:24:41 +0000649**
drh4e5ffc52004-08-31 00:52:37 +0000650** Globbing rules:
drh0ac65892002-04-20 14:24:41 +0000651**
drh4e5ffc52004-08-31 00:52:37 +0000652** '*' Matches any sequence of zero or more characters.
danielk1977d02eb1f2004-06-06 09:44:03 +0000653**
drh4e5ffc52004-08-31 00:52:37 +0000654** '?' Matches exactly one character.
655**
656** [...] Matches one character from the enclosed list of
657** characters.
658**
659** [^...] Matches one character not in the enclosed list.
660**
661** With the [...] and [^...] matching, a ']' character can be included
662** in the list by making it the first character after '[' or '^'. A
663** range of characters can be specified using '-'. Example:
664** "[a-z]" matches any single lower-case letter. To match a '-', make
665** it the last character in the list.
666**
drh9fdfdc82014-09-25 11:08:57 +0000667** Like matching rules:
668**
669** '%' Matches any sequence of zero or more characters
670**
671*** '_' Matches any one character
672**
673** Ec Where E is the "esc" character and c is any other
674** character, including '%', '_', and esc, match exactly c.
675**
drhb0870482015-06-17 13:20:54 +0000676** The comments within this routine usually assume glob matching.
drh9fdfdc82014-09-25 11:08:57 +0000677**
drh4e5ffc52004-08-31 00:52:37 +0000678** This routine is usually quick, but can be N**2 in the worst case.
drh0ac65892002-04-20 14:24:41 +0000679*/
danielk19777c6303c2004-11-17 16:41:29 +0000680static int patternCompare(
drh4e5ffc52004-08-31 00:52:37 +0000681 const u8 *zPattern, /* The glob pattern */
682 const u8 *zString, /* The string to compare against the glob */
danielk19777c6303c2004-11-17 16:41:29 +0000683 const struct compareInfo *pInfo, /* Information about how to do the compare */
drh698a01c2016-12-01 18:49:40 +0000684 u32 matchOther /* The escape char (LIKE) or '[' (GLOB) */
danielk197751ad0ec2004-05-24 12:39:02 +0000685){
drh9fdfdc82014-09-25 11:08:57 +0000686 u32 c, c2; /* Next pattern and input string chars */
687 u32 matchOne = pInfo->matchOne; /* "?" or "_" */
688 u32 matchAll = pInfo->matchAll; /* "*" or "%" */
drh9fdfdc82014-09-25 11:08:57 +0000689 u8 noCase = pInfo->noCase; /* True if uppercase==lowercase */
690 const u8 *zEscaped = 0; /* One past the last escaped input char */
drh88b33222014-09-25 03:51:37 +0000691
drhb0870482015-06-17 13:20:54 +0000692 while( (c = Utf8Read(zPattern))!=0 ){
drh9fdfdc82014-09-25 11:08:57 +0000693 if( c==matchAll ){ /* Match "*" */
694 /* Skip over multiple "*" characters in the pattern. If there
695 ** are also "?" characters, skip those as well, but consume a
696 ** single character of the input string for each "?" skipped */
drh33941692021-02-15 17:02:01 +0000697 while( (c=Utf8Read(zPattern)) == matchAll
698 || (c == matchOne && matchOne!=0) ){
drh42610962012-09-17 18:56:32 +0000699 if( c==matchOne && sqlite3Utf8Read(&zString)==0 ){
drh698a01c2016-12-01 18:49:40 +0000700 return SQLITE_NOWILDCARDMATCH;
drh4e5ffc52004-08-31 00:52:37 +0000701 }
danielk1977ad7dd422004-06-06 12:41:49 +0000702 }
drh66150952007-07-23 19:12:41 +0000703 if( c==0 ){
drh698a01c2016-12-01 18:49:40 +0000704 return SQLITE_MATCH; /* "*" at the end of the pattern matches */
drh88b33222014-09-25 03:51:37 +0000705 }else if( c==matchOther ){
drh07e83472016-01-11 03:48:18 +0000706 if( pInfo->matchSet==0 ){
drh88b33222014-09-25 03:51:37 +0000707 c = sqlite3Utf8Read(&zPattern);
drh698a01c2016-12-01 18:49:40 +0000708 if( c==0 ) return SQLITE_NOWILDCARDMATCH;
drh88b33222014-09-25 03:51:37 +0000709 }else{
drh9fdfdc82014-09-25 11:08:57 +0000710 /* "[...]" immediately follows the "*". We have to do a slow
711 ** recursive search in this case, but it is an unusual case. */
drh88b33222014-09-25 03:51:37 +0000712 assert( matchOther<0x80 ); /* '[' is a single-byte character */
drh698a01c2016-12-01 18:49:40 +0000713 while( *zString ){
714 int bMatch = patternCompare(&zPattern[-1],zString,pInfo,matchOther);
715 if( bMatch!=SQLITE_NOMATCH ) return bMatch;
drh88b33222014-09-25 03:51:37 +0000716 SQLITE_SKIP_UTF8(zString);
717 }
drh698a01c2016-12-01 18:49:40 +0000718 return SQLITE_NOWILDCARDMATCH;
drh66150952007-07-23 19:12:41 +0000719 }
drh66150952007-07-23 19:12:41 +0000720 }
drh9fdfdc82014-09-25 11:08:57 +0000721
722 /* At this point variable c contains the first character of the
723 ** pattern string past the "*". Search in the input string for the
dan1a4a7372016-12-01 17:34:59 +0000724 ** first matching character and recursively continue the match from
drh9fdfdc82014-09-25 11:08:57 +0000725 ** that point.
726 **
727 ** For a case-insensitive search, set variable cx to be the same as
728 ** c but in the other case and search the input string for either
729 ** c or cx.
730 */
731 if( c<=0x80 ){
drheba21f92017-10-30 18:49:11 +0000732 char zStop[3];
drh698a01c2016-12-01 18:49:40 +0000733 int bMatch;
drh9fdfdc82014-09-25 11:08:57 +0000734 if( noCase ){
drheba21f92017-10-30 18:49:11 +0000735 zStop[0] = sqlite3Toupper(c);
736 zStop[1] = sqlite3Tolower(c);
737 zStop[2] = 0;
drh66150952007-07-23 19:12:41 +0000738 }else{
drheba21f92017-10-30 18:49:11 +0000739 zStop[0] = c;
740 zStop[1] = 0;
drh4e5ffc52004-08-31 00:52:37 +0000741 }
drheba21f92017-10-30 18:49:11 +0000742 while(1){
743 zString += strcspn((const char*)zString, zStop);
744 if( zString[0]==0 ) break;
745 zString++;
drh698a01c2016-12-01 18:49:40 +0000746 bMatch = patternCompare(zPattern,zString,pInfo,matchOther);
747 if( bMatch!=SQLITE_NOMATCH ) return bMatch;
drh9fdfdc82014-09-25 11:08:57 +0000748 }
749 }else{
drh698a01c2016-12-01 18:49:40 +0000750 int bMatch;
drhb0870482015-06-17 13:20:54 +0000751 while( (c2 = Utf8Read(zString))!=0 ){
drh9fdfdc82014-09-25 11:08:57 +0000752 if( c2!=c ) continue;
drh698a01c2016-12-01 18:49:40 +0000753 bMatch = patternCompare(zPattern,zString,pInfo,matchOther);
754 if( bMatch!=SQLITE_NOMATCH ) return bMatch;
drh9fdfdc82014-09-25 11:08:57 +0000755 }
drh66150952007-07-23 19:12:41 +0000756 }
drh698a01c2016-12-01 18:49:40 +0000757 return SQLITE_NOWILDCARDMATCH;
drh88b33222014-09-25 03:51:37 +0000758 }
drh88b33222014-09-25 03:51:37 +0000759 if( c==matchOther ){
drh07e83472016-01-11 03:48:18 +0000760 if( pInfo->matchSet==0 ){
drh88b33222014-09-25 03:51:37 +0000761 c = sqlite3Utf8Read(&zPattern);
drh698a01c2016-12-01 18:49:40 +0000762 if( c==0 ) return SQLITE_NOMATCH;
drh9fdfdc82014-09-25 11:08:57 +0000763 zEscaped = zPattern;
drh88b33222014-09-25 03:51:37 +0000764 }else{
765 u32 prior_c = 0;
drh9fdfdc82014-09-25 11:08:57 +0000766 int seen = 0;
767 int invert = 0;
drh88b33222014-09-25 03:51:37 +0000768 c = sqlite3Utf8Read(&zString);
drh698a01c2016-12-01 18:49:40 +0000769 if( c==0 ) return SQLITE_NOMATCH;
drh88b33222014-09-25 03:51:37 +0000770 c2 = sqlite3Utf8Read(&zPattern);
771 if( c2=='^' ){
772 invert = 1;
773 c2 = sqlite3Utf8Read(&zPattern);
774 }
775 if( c2==']' ){
776 if( c==']' ) seen = 1;
777 c2 = sqlite3Utf8Read(&zPattern);
778 }
779 while( c2 && c2!=']' ){
780 if( c2=='-' && zPattern[0]!=']' && zPattern[0]!=0 && prior_c>0 ){
781 c2 = sqlite3Utf8Read(&zPattern);
782 if( c>=prior_c && c<=c2 ) seen = 1;
783 prior_c = 0;
784 }else{
785 if( c==c2 ){
786 seen = 1;
787 }
788 prior_c = c2;
789 }
790 c2 = sqlite3Utf8Read(&zPattern);
791 }
792 if( c2==0 || (seen ^ invert)==0 ){
drh698a01c2016-12-01 18:49:40 +0000793 return SQLITE_NOMATCH;
drh88b33222014-09-25 03:51:37 +0000794 }
795 continue;
796 }
797 }
drhb0870482015-06-17 13:20:54 +0000798 c2 = Utf8Read(zString);
drh88b33222014-09-25 03:51:37 +0000799 if( c==c2 ) continue;
drhc80937a2016-06-06 01:48:14 +0000800 if( noCase && sqlite3Tolower(c)==sqlite3Tolower(c2) && c<0x80 && c2<0x80 ){
drh9fdfdc82014-09-25 11:08:57 +0000801 continue;
802 }
803 if( c==matchOne && zPattern!=zEscaped && c2!=0 ) continue;
drh698a01c2016-12-01 18:49:40 +0000804 return SQLITE_NOMATCH;
danielk197751ad0ec2004-05-24 12:39:02 +0000805 }
drh698a01c2016-12-01 18:49:40 +0000806 return *zString==0 ? SQLITE_MATCH : SQLITE_NOMATCH;
drh0ac65892002-04-20 14:24:41 +0000807}
drh4e5ffc52004-08-31 00:52:37 +0000808
drh55ef4d92005-08-14 01:20:37 +0000809/*
drh698a01c2016-12-01 18:49:40 +0000810** The sqlite3_strglob() interface. Return 0 on a match (like strcmp()) and
811** non-zero if there is no match.
drh56282a52013-04-10 16:13:38 +0000812*/
813int sqlite3_strglob(const char *zGlobPattern, const char *zString){
drh698a01c2016-12-01 18:49:40 +0000814 return patternCompare((u8*)zGlobPattern, (u8*)zString, &globInfo, '[');
drh56282a52013-04-10 16:13:38 +0000815}
816
817/*
drh698a01c2016-12-01 18:49:40 +0000818** The sqlite3_strlike() interface. Return 0 on a match and non-zero for
819** a miss - like strcmp().
drh8b4a94a2015-11-24 21:23:59 +0000820*/
821int sqlite3_strlike(const char *zPattern, const char *zStr, unsigned int esc){
drh698a01c2016-12-01 18:49:40 +0000822 return patternCompare((u8*)zPattern, (u8*)zStr, &likeInfoNorm, esc);
drh8b4a94a2015-11-24 21:23:59 +0000823}
824
825/*
drh55ef4d92005-08-14 01:20:37 +0000826** Count the number of times that the LIKE operator (or GLOB which is
827** just a variation of LIKE) gets called. This is used for testing
828** only.
829*/
830#ifdef SQLITE_TEST
831int sqlite3_like_count = 0;
832#endif
833
danielk19773f6b0872004-06-17 05:36:44 +0000834
835/*
836** Implementation of the like() SQL function. This function implements
837** the build-in LIKE operator. The first argument to the function is the
838** pattern and the second argument is the string. So, the SQL statements:
839**
840** A LIKE B
841**
842** is implemented as like(B,A).
843**
drh55ef4d92005-08-14 01:20:37 +0000844** This same function (with a different compareInfo structure) computes
845** the GLOB operator.
danielk19773f6b0872004-06-17 05:36:44 +0000846*/
847static void likeFunc(
848 sqlite3_context *context,
849 int argc,
850 sqlite3_value **argv
851){
drhbeb818d2007-05-08 15:34:47 +0000852 const unsigned char *zA, *zB;
drh07e83472016-01-11 03:48:18 +0000853 u32 escape;
drh27e62db2009-04-02 10:16:17 +0000854 int nPat;
drhbb4957f2008-03-20 14:03:29 +0000855 sqlite3 *db = sqlite3_context_db_handle(context);
drh07e83472016-01-11 03:48:18 +0000856 struct compareInfo *pInfo = sqlite3_user_data(context);
drh589c7872020-03-19 18:13:28 +0000857 struct compareInfo backupInfo;
drhbeb818d2007-05-08 15:34:47 +0000858
drh41d2e662015-12-01 21:23:07 +0000859#ifdef SQLITE_LIKE_DOESNT_MATCH_BLOBS
860 if( sqlite3_value_type(argv[0])==SQLITE_BLOB
861 || sqlite3_value_type(argv[1])==SQLITE_BLOB
862 ){
863#ifdef SQLITE_TEST
864 sqlite3_like_count++;
865#endif
866 sqlite3_result_int(context, 0);
867 return;
868 }
869#endif
drh1f0feef2007-05-15 13:27:07 +0000870
drhbeb818d2007-05-08 15:34:47 +0000871 /* Limit the length of the LIKE or GLOB pattern to avoid problems
872 ** of deep recursion and N*N behavior in patternCompare().
873 */
drh27e62db2009-04-02 10:16:17 +0000874 nPat = sqlite3_value_bytes(argv[0]);
875 testcase( nPat==db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH] );
876 testcase( nPat==db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH]+1 );
877 if( nPat > db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH] ){
drhbeb818d2007-05-08 15:34:47 +0000878 sqlite3_result_error(context, "LIKE or GLOB pattern too complex", -1);
879 return;
880 }
danielk19777c6303c2004-11-17 16:41:29 +0000881 if( argc==3 ){
882 /* The escape character string must consist of a single UTF-8 character.
883 ** Otherwise, return an error.
884 */
885 const unsigned char *zEsc = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +0000886 if( zEsc==0 ) return;
drhee858132007-05-08 20:37:38 +0000887 if( sqlite3Utf8CharLen((char*)zEsc, -1)!=1 ){
danielk19777c6303c2004-11-17 16:41:29 +0000888 sqlite3_result_error(context,
889 "ESCAPE expression must be a single character", -1);
890 return;
891 }
drh42610962012-09-17 18:56:32 +0000892 escape = sqlite3Utf8Read(&zEsc);
drh589c7872020-03-19 18:13:28 +0000893 if( escape==pInfo->matchAll || escape==pInfo->matchOne ){
894 memcpy(&backupInfo, pInfo, sizeof(backupInfo));
895 pInfo = &backupInfo;
896 if( escape==pInfo->matchAll ) pInfo->matchAll = 0;
897 if( escape==pInfo->matchOne ) pInfo->matchOne = 0;
898 }
drh07e83472016-01-11 03:48:18 +0000899 }else{
900 escape = pInfo->matchSet;
danielk19777c6303c2004-11-17 16:41:29 +0000901 }
drhcf833232019-04-30 11:54:36 +0000902 zB = sqlite3_value_text(argv[0]);
903 zA = sqlite3_value_text(argv[1]);
danielk19773f6b0872004-06-17 05:36:44 +0000904 if( zA && zB ){
drh55ef4d92005-08-14 01:20:37 +0000905#ifdef SQLITE_TEST
906 sqlite3_like_count++;
907#endif
drhf49759b2017-08-25 19:51:51 +0000908 sqlite3_result_int(context,
909 patternCompare(zB, zA, pInfo, escape)==SQLITE_MATCH);
danielk197751ad0ec2004-05-24 12:39:02 +0000910 }
drh8912d102002-05-26 21:34:58 +0000911}
912
913/*
914** Implementation of the NULLIF(x,y) function. The result is the first
915** argument if the arguments are different. The result is NULL if the
916** arguments are equal to each other.
917*/
drhf9b596e2004-05-26 16:54:42 +0000918static void nullifFunc(
919 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000920 int NotUsed,
drhf9b596e2004-05-26 16:54:42 +0000921 sqlite3_value **argv
922){
danielk1977dc1bdc42004-06-11 10:51:27 +0000923 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk197762c14b32008-11-19 09:05:26 +0000924 UNUSED_PARAMETER(NotUsed);
danielk1977dc1bdc42004-06-11 10:51:27 +0000925 if( sqlite3MemCompare(argv[0], argv[1], pColl)!=0 ){
drhf4479502004-05-27 03:12:53 +0000926 sqlite3_result_value(context, argv[0]);
drh8912d102002-05-26 21:34:58 +0000927 }
drh0ac65892002-04-20 14:24:41 +0000928}
929
drh647cb0e2002-11-04 19:32:25 +0000930/*
drh47baebc2009-08-14 16:01:24 +0000931** Implementation of the sqlite_version() function. The result is the version
drh647cb0e2002-11-04 19:32:25 +0000932** of the SQLite library that is running.
933*/
drhf9b596e2004-05-26 16:54:42 +0000934static void versionFunc(
935 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000936 int NotUsed,
937 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000938){
danielk197762c14b32008-11-19 09:05:26 +0000939 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000940 /* IMP: R-48699-48617 This function is an SQL wrapper around the
941 ** sqlite3_libversion() C-interface. */
942 sqlite3_result_text(context, sqlite3_libversion(), -1, SQLITE_STATIC);
drh647cb0e2002-11-04 19:32:25 +0000943}
944
drh47baebc2009-08-14 16:01:24 +0000945/*
946** Implementation of the sqlite_source_id() function. The result is a string
947** that identifies the particular version of the source code used to build
948** SQLite.
949*/
950static void sourceidFunc(
951 sqlite3_context *context,
952 int NotUsed,
953 sqlite3_value **NotUsed2
954){
955 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000956 /* IMP: R-24470-31136 This function is an SQL wrapper around the
957 ** sqlite3_sourceid() C interface. */
958 sqlite3_result_text(context, sqlite3_sourceid(), -1, SQLITE_STATIC);
drh47baebc2009-08-14 16:01:24 +0000959}
960
shanehbdea6d12010-02-23 04:19:54 +0000961/*
drh3ca84ef2011-04-25 18:03:10 +0000962** Implementation of the sqlite_log() function. This is a wrapper around
963** sqlite3_log(). The return value is NULL. The function exists purely for
964** its side-effects.
965*/
drh840561f2011-04-27 18:08:42 +0000966static void errlogFunc(
drh3ca84ef2011-04-25 18:03:10 +0000967 sqlite3_context *context,
968 int argc,
969 sqlite3_value **argv
970){
971 UNUSED_PARAMETER(argc);
972 UNUSED_PARAMETER(context);
973 sqlite3_log(sqlite3_value_int(argv[0]), "%s", sqlite3_value_text(argv[1]));
974}
975
976/*
shanehdc97a8c2010-02-23 20:08:35 +0000977** Implementation of the sqlite_compileoption_used() function.
978** The result is an integer that identifies if the compiler option
979** was used to build SQLite.
shanehbdea6d12010-02-23 04:19:54 +0000980*/
shanehdc97a8c2010-02-23 20:08:35 +0000981#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
982static void compileoptionusedFunc(
shanehbdea6d12010-02-23 04:19:54 +0000983 sqlite3_context *context,
shanehdc97a8c2010-02-23 20:08:35 +0000984 int argc,
985 sqlite3_value **argv
shanehbdea6d12010-02-23 04:19:54 +0000986){
shanehdc97a8c2010-02-23 20:08:35 +0000987 const char *zOptName;
988 assert( argc==1 );
989 UNUSED_PARAMETER(argc);
drha3e414c2010-08-03 18:06:25 +0000990 /* IMP: R-39564-36305 The sqlite_compileoption_used() SQL
991 ** function is a wrapper around the sqlite3_compileoption_used() C/C++
992 ** function.
993 */
drh264a2d42010-02-25 15:28:41 +0000994 if( (zOptName = (const char*)sqlite3_value_text(argv[0]))!=0 ){
shanehdc97a8c2010-02-23 20:08:35 +0000995 sqlite3_result_int(context, sqlite3_compileoption_used(zOptName));
shanehdc97a8c2010-02-23 20:08:35 +0000996 }
shanehbdea6d12010-02-23 04:19:54 +0000997}
shanehdc97a8c2010-02-23 20:08:35 +0000998#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
999
1000/*
1001** Implementation of the sqlite_compileoption_get() function.
1002** The result is a string that identifies the compiler options
1003** used to build SQLite.
1004*/
1005#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
1006static void compileoptiongetFunc(
1007 sqlite3_context *context,
1008 int argc,
1009 sqlite3_value **argv
1010){
1011 int n;
1012 assert( argc==1 );
1013 UNUSED_PARAMETER(argc);
drha3e414c2010-08-03 18:06:25 +00001014 /* IMP: R-04922-24076 The sqlite_compileoption_get() SQL function
1015 ** is a wrapper around the sqlite3_compileoption_get() C/C++ function.
1016 */
shanehdc97a8c2010-02-23 20:08:35 +00001017 n = sqlite3_value_int(argv[0]);
1018 sqlite3_result_text(context, sqlite3_compileoption_get(n), -1, SQLITE_STATIC);
1019}
1020#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
shanehbdea6d12010-02-23 04:19:54 +00001021
drh137c7282007-01-29 17:58:28 +00001022/* Array for converting from half-bytes (nybbles) into ASCII hex
1023** digits. */
1024static const char hexdigits[] = {
1025 '0', '1', '2', '3', '4', '5', '6', '7',
1026 '8', '9', 'A', 'B', 'C', 'D', 'E', 'F'
1027};
danielk1977d641d642004-11-18 15:44:29 +00001028
drh47394702003-08-20 01:03:33 +00001029/*
drh47394702003-08-20 01:03:33 +00001030** Implementation of the QUOTE() function. This function takes a single
1031** argument. If the argument is numeric, the return value is the same as
1032** the argument. If the argument is NULL, the return value is the string
1033** "NULL". Otherwise, the argument is enclosed in single quotes with
1034** single-quote escapes.
1035*/
danielk19770ae8b832004-05-25 12:05:56 +00001036static void quoteFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drha0df4cc2009-02-02 17:29:59 +00001037 assert( argc==1 );
drh1d34fde2009-02-03 15:50:33 +00001038 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +00001039 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +00001040 case SQLITE_FLOAT: {
drh72b3fbc2012-06-19 03:11:25 +00001041 double r1, r2;
drh72b3fbc2012-06-19 03:11:25 +00001042 char zBuf[50];
mistachkin2b434a72012-06-19 04:36:48 +00001043 r1 = sqlite3_value_double(argv[0]);
drh72b3fbc2012-06-19 03:11:25 +00001044 sqlite3_snprintf(sizeof(zBuf), zBuf, "%!.15g", r1);
1045 sqlite3AtoF(zBuf, &r2, 20, SQLITE_UTF8);
1046 if( r1!=r2 ){
1047 sqlite3_snprintf(sizeof(zBuf), zBuf, "%!.20e", r1);
1048 }
1049 sqlite3_result_text(context, zBuf, -1, SQLITE_TRANSIENT);
1050 break;
1051 }
1052 case SQLITE_INTEGER: {
drhf4479502004-05-27 03:12:53 +00001053 sqlite3_result_value(context, argv[0]);
drhf9b596e2004-05-26 16:54:42 +00001054 break;
1055 }
danielk19773f41e972004-06-08 00:39:01 +00001056 case SQLITE_BLOB: {
danielk19773f41e972004-06-08 00:39:01 +00001057 char *zText = 0;
danielk19773f41e972004-06-08 00:39:01 +00001058 char const *zBlob = sqlite3_value_blob(argv[0]);
drh1f0feef2007-05-15 13:27:07 +00001059 int nBlob = sqlite3_value_bytes(argv[0]);
1060 assert( zBlob==sqlite3_value_blob(argv[0]) ); /* No encoding change */
drhb1a6c3c2008-03-20 16:30:17 +00001061 zText = (char *)contextMalloc(context, (2*(i64)nBlob)+4);
danielk1977a1644fd2007-08-29 12:31:25 +00001062 if( zText ){
danielk19773f41e972004-06-08 00:39:01 +00001063 int i;
1064 for(i=0; i<nBlob; i++){
1065 zText[(i*2)+2] = hexdigits[(zBlob[i]>>4)&0x0F];
1066 zText[(i*2)+3] = hexdigits[(zBlob[i])&0x0F];
1067 }
1068 zText[(nBlob*2)+2] = '\'';
1069 zText[(nBlob*2)+3] = '\0';
1070 zText[0] = 'X';
1071 zText[1] = '\'';
danielk1977d8123362004-06-12 09:25:12 +00001072 sqlite3_result_text(context, zText, -1, SQLITE_TRANSIENT);
drh17435752007-08-16 04:30:38 +00001073 sqlite3_free(zText);
danielk19773f41e972004-06-08 00:39:01 +00001074 }
1075 break;
1076 }
drh9c054832004-05-31 18:51:57 +00001077 case SQLITE_TEXT: {
drh023ae032007-05-08 12:12:16 +00001078 int i,j;
1079 u64 n;
drh2646da72005-12-09 20:02:05 +00001080 const unsigned char *zArg = sqlite3_value_text(argv[0]);
drhf9b596e2004-05-26 16:54:42 +00001081 char *z;
1082
drh7a521cf2007-04-25 18:23:52 +00001083 if( zArg==0 ) return;
drh023ae032007-05-08 12:12:16 +00001084 for(i=0, n=0; zArg[i]; i++){ if( zArg[i]=='\'' ) n++; }
drhb1a6c3c2008-03-20 16:30:17 +00001085 z = contextMalloc(context, ((i64)i)+((i64)n)+3);
danielk1977a1644fd2007-08-29 12:31:25 +00001086 if( z ){
1087 z[0] = '\'';
1088 for(i=0, j=1; zArg[i]; i++){
1089 z[j++] = zArg[i];
1090 if( zArg[i]=='\'' ){
1091 z[j++] = '\'';
1092 }
drhf9b596e2004-05-26 16:54:42 +00001093 }
danielk1977a1644fd2007-08-29 12:31:25 +00001094 z[j++] = '\'';
1095 z[j] = 0;
1096 sqlite3_result_text(context, z, j, sqlite3_free);
drhf9b596e2004-05-26 16:54:42 +00001097 }
drha0df4cc2009-02-02 17:29:59 +00001098 break;
1099 }
1100 default: {
1101 assert( sqlite3_value_type(argv[0])==SQLITE_NULL );
1102 sqlite3_result_text(context, "NULL", 4, SQLITE_STATIC);
1103 break;
drhf9b596e2004-05-26 16:54:42 +00001104 }
drh47394702003-08-20 01:03:33 +00001105 }
1106}
1107
drh137c7282007-01-29 17:58:28 +00001108/*
drhd495d8c2013-02-22 19:34:25 +00001109** The unicode() function. Return the integer unicode code-point value
1110** for the first character of the input string.
1111*/
1112static void unicodeFunc(
1113 sqlite3_context *context,
1114 int argc,
1115 sqlite3_value **argv
1116){
1117 const unsigned char *z = sqlite3_value_text(argv[0]);
drh1d59d032013-03-01 23:40:26 +00001118 (void)argc;
drhd495d8c2013-02-22 19:34:25 +00001119 if( z && z[0] ) sqlite3_result_int(context, sqlite3Utf8Read(&z));
1120}
1121
1122/*
1123** The char() function takes zero or more arguments, each of which is
1124** an integer. It constructs a string where each character of the string
1125** is the unicode character for the corresponding integer argument.
1126*/
1127static void charFunc(
1128 sqlite3_context *context,
1129 int argc,
1130 sqlite3_value **argv
1131){
1132 unsigned char *z, *zOut;
1133 int i;
drhf3cdcdc2015-04-29 16:50:28 +00001134 zOut = z = sqlite3_malloc64( argc*4+1 );
drhd495d8c2013-02-22 19:34:25 +00001135 if( z==0 ){
1136 sqlite3_result_error_nomem(context);
1137 return;
1138 }
1139 for(i=0; i<argc; i++){
mistachkinc9545442013-02-26 05:42:30 +00001140 sqlite3_int64 x;
drhd495d8c2013-02-22 19:34:25 +00001141 unsigned c;
1142 x = sqlite3_value_int64(argv[i]);
1143 if( x<0 || x>0x10ffff ) x = 0xfffd;
1144 c = (unsigned)(x & 0x1fffff);
drhfe7a5d12013-03-07 14:00:04 +00001145 if( c<0x00080 ){
1146 *zOut++ = (u8)(c&0xFF);
1147 }else if( c<0x00800 ){
1148 *zOut++ = 0xC0 + (u8)((c>>6)&0x1F);
1149 *zOut++ = 0x80 + (u8)(c & 0x3F);
1150 }else if( c<0x10000 ){
1151 *zOut++ = 0xE0 + (u8)((c>>12)&0x0F);
1152 *zOut++ = 0x80 + (u8)((c>>6) & 0x3F);
1153 *zOut++ = 0x80 + (u8)(c & 0x3F);
drhd495d8c2013-02-22 19:34:25 +00001154 }else{
drhfe7a5d12013-03-07 14:00:04 +00001155 *zOut++ = 0xF0 + (u8)((c>>18) & 0x07);
1156 *zOut++ = 0x80 + (u8)((c>>12) & 0x3F);
1157 *zOut++ = 0x80 + (u8)((c>>6) & 0x3F);
1158 *zOut++ = 0x80 + (u8)(c & 0x3F);
1159 } \
drhd495d8c2013-02-22 19:34:25 +00001160 }
drhbbf483f2014-09-09 20:30:24 +00001161 sqlite3_result_text64(context, (char*)z, zOut-z, sqlite3_free, SQLITE_UTF8);
drhd495d8c2013-02-22 19:34:25 +00001162}
1163
1164/*
drh137c7282007-01-29 17:58:28 +00001165** The hex() function. Interpret the argument as a blob. Return
1166** a hexadecimal rendering as text.
1167*/
1168static void hexFunc(
1169 sqlite3_context *context,
1170 int argc,
1171 sqlite3_value **argv
1172){
1173 int i, n;
1174 const unsigned char *pBlob;
1175 char *zHex, *z;
1176 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001177 UNUSED_PARAMETER(argc);
drh1f0feef2007-05-15 13:27:07 +00001178 pBlob = sqlite3_value_blob(argv[0]);
drh137c7282007-01-29 17:58:28 +00001179 n = sqlite3_value_bytes(argv[0]);
drh1f0feef2007-05-15 13:27:07 +00001180 assert( pBlob==sqlite3_value_blob(argv[0]) ); /* No encoding change */
drhb1a6c3c2008-03-20 16:30:17 +00001181 z = zHex = contextMalloc(context, ((i64)n)*2 + 1);
danielk1977a1644fd2007-08-29 12:31:25 +00001182 if( zHex ){
1183 for(i=0; i<n; i++, pBlob++){
1184 unsigned char c = *pBlob;
1185 *(z++) = hexdigits[(c>>4)&0xf];
1186 *(z++) = hexdigits[c&0xf];
1187 }
1188 *z = 0;
1189 sqlite3_result_text(context, zHex, n*2, sqlite3_free);
drh137c7282007-01-29 17:58:28 +00001190 }
drh137c7282007-01-29 17:58:28 +00001191}
1192
drh26b6d902007-03-17 13:27:54 +00001193/*
drh8cff3822007-05-02 02:08:28 +00001194** The zeroblob(N) function returns a zero-filled blob of size N bytes.
1195*/
1196static void zeroblobFunc(
1197 sqlite3_context *context,
1198 int argc,
1199 sqlite3_value **argv
1200){
drh98640a32007-06-07 19:08:32 +00001201 i64 n;
dana4d5ae82015-07-24 16:24:37 +00001202 int rc;
drh8cff3822007-05-02 02:08:28 +00001203 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001204 UNUSED_PARAMETER(argc);
drh98640a32007-06-07 19:08:32 +00001205 n = sqlite3_value_int64(argv[0]);
dana4d5ae82015-07-24 16:24:37 +00001206 if( n<0 ) n = 0;
1207 rc = sqlite3_result_zeroblob64(context, n); /* IMP: R-00293-64994 */
1208 if( rc ){
1209 sqlite3_result_error_code(context, rc);
drh98640a32007-06-07 19:08:32 +00001210 }
drh8cff3822007-05-02 02:08:28 +00001211}
1212
1213/*
drh26b6d902007-03-17 13:27:54 +00001214** The replace() function. Three arguments are all strings: call
1215** them A, B, and C. The result is also a string which is derived
drhf7b54962013-05-28 12:11:54 +00001216** from A by replacing every occurrence of B with C. The match
drh26b6d902007-03-17 13:27:54 +00001217** must be exact. Collating sequences are not used.
1218*/
1219static void replaceFunc(
1220 sqlite3_context *context,
1221 int argc,
1222 sqlite3_value **argv
1223){
1224 const unsigned char *zStr; /* The input string A */
1225 const unsigned char *zPattern; /* The pattern string B */
1226 const unsigned char *zRep; /* The replacement string C */
1227 unsigned char *zOut; /* The output */
1228 int nStr; /* Size of zStr */
1229 int nPattern; /* Size of zPattern */
1230 int nRep; /* Size of zRep */
drh2e6400b2007-05-08 15:46:18 +00001231 i64 nOut; /* Maximum size of zOut */
drh26b6d902007-03-17 13:27:54 +00001232 int loopLimit; /* Last zStr[] that might match zPattern[] */
1233 int i, j; /* Loop counters */
drhf3139522018-02-09 23:25:14 +00001234 unsigned cntExpand; /* Number zOut expansions */
1235 sqlite3 *db = sqlite3_context_db_handle(context);
drh26b6d902007-03-17 13:27:54 +00001236
1237 assert( argc==3 );
danielk1977f3d3c272008-11-19 16:52:44 +00001238 UNUSED_PARAMETER(argc);
drh26b6d902007-03-17 13:27:54 +00001239 zStr = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001240 if( zStr==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001241 nStr = sqlite3_value_bytes(argv[0]);
1242 assert( zStr==sqlite3_value_text(argv[0]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001243 zPattern = sqlite3_value_text(argv[1]);
drha605fe82009-02-01 18:08:40 +00001244 if( zPattern==0 ){
drh23336062009-02-03 13:19:12 +00001245 assert( sqlite3_value_type(argv[1])==SQLITE_NULL
1246 || sqlite3_context_db_handle(context)->mallocFailed );
drha605fe82009-02-01 18:08:40 +00001247 return;
1248 }
1249 if( zPattern[0]==0 ){
1250 assert( sqlite3_value_type(argv[1])!=SQLITE_NULL );
1251 sqlite3_result_value(context, argv[0]);
1252 return;
1253 }
drh1f0feef2007-05-15 13:27:07 +00001254 nPattern = sqlite3_value_bytes(argv[1]);
1255 assert( zPattern==sqlite3_value_text(argv[1]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001256 zRep = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +00001257 if( zRep==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001258 nRep = sqlite3_value_bytes(argv[2]);
1259 assert( zRep==sqlite3_value_text(argv[2]) );
drh2e6400b2007-05-08 15:46:18 +00001260 nOut = nStr + 1;
1261 assert( nOut<SQLITE_MAX_LENGTH );
drhb1a6c3c2008-03-20 16:30:17 +00001262 zOut = contextMalloc(context, (i64)nOut);
drh2e6400b2007-05-08 15:46:18 +00001263 if( zOut==0 ){
1264 return;
drh26b6d902007-03-17 13:27:54 +00001265 }
drh26b6d902007-03-17 13:27:54 +00001266 loopLimit = nStr - nPattern;
drhf3139522018-02-09 23:25:14 +00001267 cntExpand = 0;
drh26b6d902007-03-17 13:27:54 +00001268 for(i=j=0; i<=loopLimit; i++){
1269 if( zStr[i]!=zPattern[0] || memcmp(&zStr[i], zPattern, nPattern) ){
1270 zOut[j++] = zStr[i];
1271 }else{
drhf3139522018-02-09 23:25:14 +00001272 if( nRep>nPattern ){
1273 nOut += nRep - nPattern;
drhc86d82f2018-02-10 02:31:30 +00001274 testcase( nOut-1==db->aLimit[SQLITE_LIMIT_LENGTH] );
1275 testcase( nOut-2==db->aLimit[SQLITE_LIMIT_LENGTH] );
drhf3139522018-02-09 23:25:14 +00001276 if( nOut-1>db->aLimit[SQLITE_LIMIT_LENGTH] ){
1277 sqlite3_result_error_toobig(context);
1278 sqlite3_free(zOut);
1279 return;
1280 }
drhf3139522018-02-09 23:25:14 +00001281 cntExpand++;
1282 if( (cntExpand&(cntExpand-1))==0 ){
1283 /* Grow the size of the output buffer only on substitutions
1284 ** whose index is a power of two: 1, 2, 4, 8, 16, 32, ... */
1285 u8 *zOld;
1286 zOld = zOut;
drhd924e7b2020-05-17 00:26:44 +00001287 zOut = sqlite3Realloc(zOut, (int)nOut + (nOut - nStr - 1));
drhf3139522018-02-09 23:25:14 +00001288 if( zOut==0 ){
1289 sqlite3_result_error_nomem(context);
1290 sqlite3_free(zOld);
1291 return;
1292 }
1293 }
drh2e6400b2007-05-08 15:46:18 +00001294 }
drh26b6d902007-03-17 13:27:54 +00001295 memcpy(&zOut[j], zRep, nRep);
1296 j += nRep;
1297 i += nPattern-1;
1298 }
1299 }
drhf3139522018-02-09 23:25:14 +00001300 assert( j+nStr-i+1<=nOut );
drh26b6d902007-03-17 13:27:54 +00001301 memcpy(&zOut[j], &zStr[i], nStr-i);
1302 j += nStr - i;
1303 assert( j<=nOut );
1304 zOut[j] = 0;
1305 sqlite3_result_text(context, (char*)zOut, j, sqlite3_free);
1306}
1307
drh309b3382007-03-17 17:52:42 +00001308/*
1309** Implementation of the TRIM(), LTRIM(), and RTRIM() functions.
1310** The userdata is 0x1 for left trim, 0x2 for right trim, 0x3 for both.
1311*/
1312static void trimFunc(
1313 sqlite3_context *context,
1314 int argc,
1315 sqlite3_value **argv
1316){
1317 const unsigned char *zIn; /* Input string */
1318 const unsigned char *zCharSet; /* Set of characters to trim */
1319 int nIn; /* Number of bytes in input */
drh7209c692008-04-27 18:40:11 +00001320 int flags; /* 1: trimleft 2: trimright 3: trim */
drhd1e3a612007-04-27 21:59:52 +00001321 int i; /* Loop counter */
drh1bd10f82008-12-10 21:19:56 +00001322 unsigned char *aLen = 0; /* Length of each character in zCharSet */
1323 unsigned char **azChar = 0; /* Individual characters in zCharSet */
drhd1e3a612007-04-27 21:59:52 +00001324 int nChar; /* Number of characters in zCharSet */
1325
drh309b3382007-03-17 17:52:42 +00001326 if( sqlite3_value_type(argv[0])==SQLITE_NULL ){
1327 return;
1328 }
1329 zIn = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001330 if( zIn==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001331 nIn = sqlite3_value_bytes(argv[0]);
1332 assert( zIn==sqlite3_value_text(argv[0]) );
drh309b3382007-03-17 17:52:42 +00001333 if( argc==1 ){
drhd1e3a612007-04-27 21:59:52 +00001334 static const unsigned char lenOne[] = { 1 };
danielk1977a4de4532008-09-02 15:44:08 +00001335 static unsigned char * const azOne[] = { (u8*)" " };
drhd1e3a612007-04-27 21:59:52 +00001336 nChar = 1;
drh8cff3822007-05-02 02:08:28 +00001337 aLen = (u8*)lenOne;
danielk1977bc67da42007-12-11 04:23:19 +00001338 azChar = (unsigned char **)azOne;
drhd1e3a612007-04-27 21:59:52 +00001339 zCharSet = 0;
drh7a521cf2007-04-25 18:23:52 +00001340 }else if( (zCharSet = sqlite3_value_text(argv[1]))==0 ){
drh309b3382007-03-17 17:52:42 +00001341 return;
drhd1e3a612007-04-27 21:59:52 +00001342 }else{
1343 const unsigned char *z;
1344 for(z=zCharSet, nChar=0; *z; nChar++){
drh4a919112007-05-15 11:55:09 +00001345 SQLITE_SKIP_UTF8(z);
drhd1e3a612007-04-27 21:59:52 +00001346 }
1347 if( nChar>0 ){
drhb1a6c3c2008-03-20 16:30:17 +00001348 azChar = contextMalloc(context, ((i64)nChar)*(sizeof(char*)+1));
drhd1e3a612007-04-27 21:59:52 +00001349 if( azChar==0 ){
1350 return;
1351 }
1352 aLen = (unsigned char*)&azChar[nChar];
1353 for(z=zCharSet, nChar=0; *z; nChar++){
danielk1977bc67da42007-12-11 04:23:19 +00001354 azChar[nChar] = (unsigned char *)z;
drh4a919112007-05-15 11:55:09 +00001355 SQLITE_SKIP_UTF8(z);
drh3abbd392008-12-10 23:04:13 +00001356 aLen[nChar] = (u8)(z - azChar[nChar]);
drhd1e3a612007-04-27 21:59:52 +00001357 }
1358 }
drh309b3382007-03-17 17:52:42 +00001359 }
drhd1e3a612007-04-27 21:59:52 +00001360 if( nChar>0 ){
shane1fc41292008-07-08 22:28:48 +00001361 flags = SQLITE_PTR_TO_INT(sqlite3_user_data(context));
drh309b3382007-03-17 17:52:42 +00001362 if( flags & 1 ){
drhd1e3a612007-04-27 21:59:52 +00001363 while( nIn>0 ){
drh1bd10f82008-12-10 21:19:56 +00001364 int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001365 for(i=0; i<nChar; i++){
1366 len = aLen[i];
drh27e62db2009-04-02 10:16:17 +00001367 if( len<=nIn && memcmp(zIn, azChar[i], len)==0 ) break;
drhd1e3a612007-04-27 21:59:52 +00001368 }
1369 if( i>=nChar ) break;
1370 zIn += len;
1371 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001372 }
1373 }
1374 if( flags & 2 ){
drhd1e3a612007-04-27 21:59:52 +00001375 while( nIn>0 ){
drh1bd10f82008-12-10 21:19:56 +00001376 int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001377 for(i=0; i<nChar; i++){
1378 len = aLen[i];
1379 if( len<=nIn && memcmp(&zIn[nIn-len],azChar[i],len)==0 ) break;
1380 }
1381 if( i>=nChar ) break;
1382 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001383 }
1384 }
drhd1e3a612007-04-27 21:59:52 +00001385 if( zCharSet ){
1386 sqlite3_free(azChar);
1387 }
drh309b3382007-03-17 17:52:42 +00001388 }
1389 sqlite3_result_text(context, (char*)zIn, nIn, SQLITE_TRANSIENT);
1390}
drh26b6d902007-03-17 13:27:54 +00001391
danielk1977a4de4532008-09-02 15:44:08 +00001392
drhcc153132016-08-04 12:35:17 +00001393#ifdef SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION
1394/*
1395** The "unknown" function is automatically substituted in place of
1396** any unrecognized function name when doing an EXPLAIN or EXPLAIN QUERY PLAN
1397** when the SQLITE_ENABLE_UNKNOWN_FUNCTION compile-time option is used.
1398** When the "sqlite3" command-line shell is built using this functionality,
1399** that allows an EXPLAIN or EXPLAIN QUERY PLAN for complex queries
1400** involving application-defined functions to be examined in a generic
1401** sqlite3 shell.
1402*/
1403static void unknownFunc(
1404 sqlite3_context *context,
1405 int argc,
1406 sqlite3_value **argv
1407){
1408 /* no-op */
1409}
1410#endif /*SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION*/
1411
1412
drh2ba3ccc2009-12-08 02:06:08 +00001413/* IMP: R-25361-16150 This function is omitted from SQLite by default. It
1414** is only available if the SQLITE_SOUNDEX compile-time option is used
1415** when SQLite is built.
1416*/
drhd24cc422003-03-27 12:51:24 +00001417#ifdef SQLITE_SOUNDEX
1418/*
1419** Compute the soundex encoding of a word.
drh2ba3ccc2009-12-08 02:06:08 +00001420**
1421** IMP: R-59782-00072 The soundex(X) function returns a string that is the
1422** soundex encoding of the string X.
drhd24cc422003-03-27 12:51:24 +00001423*/
drh137c7282007-01-29 17:58:28 +00001424static void soundexFunc(
1425 sqlite3_context *context,
1426 int argc,
1427 sqlite3_value **argv
1428){
drhd24cc422003-03-27 12:51:24 +00001429 char zResult[8];
drh4c755c02004-08-08 20:22:17 +00001430 const u8 *zIn;
drhd24cc422003-03-27 12:51:24 +00001431 int i, j;
1432 static const unsigned char iCode[] = {
1433 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1434 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1435 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1436 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1437 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1438 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1439 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1440 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1441 };
1442 assert( argc==1 );
drh4c755c02004-08-08 20:22:17 +00001443 zIn = (u8*)sqlite3_value_text(argv[0]);
drhbdf67e02006-08-19 11:34:01 +00001444 if( zIn==0 ) zIn = (u8*)"";
drhdc86e2b2009-01-24 11:30:42 +00001445 for(i=0; zIn[i] && !sqlite3Isalpha(zIn[i]); i++){}
drhd24cc422003-03-27 12:51:24 +00001446 if( zIn[i] ){
drhbdf67e02006-08-19 11:34:01 +00001447 u8 prevcode = iCode[zIn[i]&0x7f];
danielk197778ca0e72009-01-20 16:53:39 +00001448 zResult[0] = sqlite3Toupper(zIn[i]);
drhd24cc422003-03-27 12:51:24 +00001449 for(j=1; j<4 && zIn[i]; i++){
1450 int code = iCode[zIn[i]&0x7f];
1451 if( code>0 ){
drhbdf67e02006-08-19 11:34:01 +00001452 if( code!=prevcode ){
1453 prevcode = code;
1454 zResult[j++] = code + '0';
1455 }
1456 }else{
1457 prevcode = 0;
drhd24cc422003-03-27 12:51:24 +00001458 }
1459 }
1460 while( j<4 ){
1461 zResult[j++] = '0';
1462 }
1463 zResult[j] = 0;
danielk1977d8123362004-06-12 09:25:12 +00001464 sqlite3_result_text(context, zResult, 4, SQLITE_TRANSIENT);
drhd24cc422003-03-27 12:51:24 +00001465 }else{
drh2ba3ccc2009-12-08 02:06:08 +00001466 /* IMP: R-64894-50321 The string "?000" is returned if the argument
1467 ** is NULL or contains no ASCII alphabetic characters. */
danielk1977d8123362004-06-12 09:25:12 +00001468 sqlite3_result_text(context, "?000", 4, SQLITE_STATIC);
drhd24cc422003-03-27 12:51:24 +00001469 }
1470}
drh2ba3ccc2009-12-08 02:06:08 +00001471#endif /* SQLITE_SOUNDEX */
drhd24cc422003-03-27 12:51:24 +00001472
drhfdb83b22006-06-17 14:12:47 +00001473#ifndef SQLITE_OMIT_LOAD_EXTENSION
1474/*
1475** A function that loads a shared-library extension then returns NULL.
1476*/
1477static void loadExt(sqlite3_context *context, int argc, sqlite3_value **argv){
danielk197765fd59f2006-06-24 11:51:33 +00001478 const char *zFile = (const char *)sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001479 const char *zProc;
drhfa4a4b92008-03-19 21:45:51 +00001480 sqlite3 *db = sqlite3_context_db_handle(context);
drhfdb83b22006-06-17 14:12:47 +00001481 char *zErrMsg = 0;
1482
drh191dd062016-04-21 01:30:09 +00001483 /* Disallow the load_extension() SQL function unless the SQLITE_LoadExtFunc
drh1a55ded2016-04-20 00:30:05 +00001484 ** flag is set. See the sqlite3_enable_load_extension() API.
1485 */
drhf602a162016-04-21 01:58:21 +00001486 if( (db->flags & SQLITE_LoadExtFunc)==0 ){
1487 sqlite3_result_error(context, "not authorized", -1);
1488 return;
1489 }
drh1a55ded2016-04-20 00:30:05 +00001490
drhfdb83b22006-06-17 14:12:47 +00001491 if( argc==2 ){
danielk197765fd59f2006-06-24 11:51:33 +00001492 zProc = (const char *)sqlite3_value_text(argv[1]);
drh7a521cf2007-04-25 18:23:52 +00001493 }else{
1494 zProc = 0;
drhfdb83b22006-06-17 14:12:47 +00001495 }
drh7a521cf2007-04-25 18:23:52 +00001496 if( zFile && sqlite3_load_extension(db, zFile, zProc, &zErrMsg) ){
drhfdb83b22006-06-17 14:12:47 +00001497 sqlite3_result_error(context, zErrMsg, -1);
1498 sqlite3_free(zErrMsg);
1499 }
1500}
1501#endif
1502
danielk197701427a62005-01-11 13:02:33 +00001503
drh0ac65892002-04-20 14:24:41 +00001504/*
drhd3a149e2002-02-24 17:12:53 +00001505** An instance of the following structure holds the context of a
drhdd5baa92002-02-27 19:50:59 +00001506** sum() or avg() aggregate computation.
1507*/
1508typedef struct SumCtx SumCtx;
1509struct SumCtx {
drh8c08e862006-02-11 17:34:00 +00001510 double rSum; /* Floating point sum */
1511 i64 iSum; /* Integer sum */
1512 i64 cnt; /* Number of elements summed */
1513 u8 overflow; /* True if integer overflow seen */
1514 u8 approx; /* True if non-integer value was input to the sum */
drhdd5baa92002-02-27 19:50:59 +00001515};
1516
1517/*
drha97fdd32006-01-12 22:17:50 +00001518** Routines used to compute the sum, average, and total.
1519**
1520** The SUM() function follows the (broken) SQL standard which means
1521** that it returns NULL if it sums over no inputs. TOTAL returns
1522** 0.0 in that case. In addition, TOTAL always returns a float where
1523** SUM might return an integer if it never encounters a floating point
drhc806d852006-05-11 13:25:39 +00001524** value. TOTAL never fails, but SUM might through an exception if
1525** it overflows an integer.
drhdd5baa92002-02-27 19:50:59 +00001526*/
danielk19770ae8b832004-05-25 12:05:56 +00001527static void sumStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drhdd5baa92002-02-27 19:50:59 +00001528 SumCtx *p;
drh3d1d95e2005-09-08 10:37:01 +00001529 int type;
drh3f219f42005-09-08 19:45:57 +00001530 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001531 UNUSED_PARAMETER(argc);
drh4f26d6c2004-05-26 23:25:30 +00001532 p = sqlite3_aggregate_context(context, sizeof(*p));
drh29d72102006-02-09 22:13:41 +00001533 type = sqlite3_value_numeric_type(argv[0]);
drh3d1d95e2005-09-08 10:37:01 +00001534 if( p && type!=SQLITE_NULL ){
drh739105c2002-05-29 23:22:23 +00001535 p->cnt++;
drh29d72102006-02-09 22:13:41 +00001536 if( type==SQLITE_INTEGER ){
drh8c08e862006-02-11 17:34:00 +00001537 i64 v = sqlite3_value_int64(argv[0]);
1538 p->rSum += v;
drh158b9cb2011-03-05 20:59:46 +00001539 if( (p->approx|p->overflow)==0 && sqlite3AddInt64(&p->iSum, v) ){
drha546ef22018-07-07 20:55:16 +00001540 p->approx = p->overflow = 1;
drh29d72102006-02-09 22:13:41 +00001541 }
1542 }else{
drh8c08e862006-02-11 17:34:00 +00001543 p->rSum += sqlite3_value_double(argv[0]);
drh29d72102006-02-09 22:13:41 +00001544 p->approx = 1;
drh3f219f42005-09-08 19:45:57 +00001545 }
drh739105c2002-05-29 23:22:23 +00001546 }
drhdd5baa92002-02-27 19:50:59 +00001547}
dan67a9b8e2018-06-22 20:51:35 +00001548#ifndef SQLITE_OMIT_WINDOWFUNC
danc3a20c12018-05-23 20:55:37 +00001549static void sumInverse(sqlite3_context *context, int argc, sqlite3_value**argv){
1550 SumCtx *p;
1551 int type;
1552 assert( argc==1 );
1553 UNUSED_PARAMETER(argc);
1554 p = sqlite3_aggregate_context(context, sizeof(*p));
1555 type = sqlite3_value_numeric_type(argv[0]);
drhfd4b7282018-07-07 19:47:21 +00001556 /* p is always non-NULL because sumStep() will have been called first
1557 ** to initialize it */
1558 if( ALWAYS(p) && type!=SQLITE_NULL ){
drha546ef22018-07-07 20:55:16 +00001559 assert( p->cnt>0 );
danc3a20c12018-05-23 20:55:37 +00001560 p->cnt--;
drha546ef22018-07-07 20:55:16 +00001561 assert( type==SQLITE_INTEGER || p->approx );
1562 if( type==SQLITE_INTEGER && p->approx==0 ){
danc3a20c12018-05-23 20:55:37 +00001563 i64 v = sqlite3_value_int64(argv[0]);
1564 p->rSum -= v;
drha546ef22018-07-07 20:55:16 +00001565 p->iSum -= v;
danc3a20c12018-05-23 20:55:37 +00001566 }else{
dand7368292018-07-02 17:45:59 +00001567 p->rSum -= sqlite3_value_double(argv[0]);
danc3a20c12018-05-23 20:55:37 +00001568 }
1569 }
1570}
dan67a9b8e2018-06-22 20:51:35 +00001571#else
1572# define sumInverse 0
1573#endif /* SQLITE_OMIT_WINDOWFUNC */
danielk19770ae8b832004-05-25 12:05:56 +00001574static void sumFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001575 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001576 p = sqlite3_aggregate_context(context, 0);
drhc2bd9132005-09-08 20:37:43 +00001577 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001578 if( p->overflow ){
1579 sqlite3_result_error(context,"integer overflow",-1);
1580 }else if( p->approx ){
1581 sqlite3_result_double(context, p->rSum);
drhc2bd9132005-09-08 20:37:43 +00001582 }else{
drh8c08e862006-02-11 17:34:00 +00001583 sqlite3_result_int64(context, p->iSum);
drhc2bd9132005-09-08 20:37:43 +00001584 }
drh3d1d95e2005-09-08 10:37:01 +00001585 }
drhdd5baa92002-02-27 19:50:59 +00001586}
danielk19770ae8b832004-05-25 12:05:56 +00001587static void avgFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001588 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001589 p = sqlite3_aggregate_context(context, 0);
drh739105c2002-05-29 23:22:23 +00001590 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001591 sqlite3_result_double(context, p->rSum/(double)p->cnt);
drhdd5baa92002-02-27 19:50:59 +00001592 }
1593}
drha97fdd32006-01-12 22:17:50 +00001594static void totalFinalize(sqlite3_context *context){
1595 SumCtx *p;
1596 p = sqlite3_aggregate_context(context, 0);
shanefbd60f82009-02-04 03:59:25 +00001597 /* (double)0 In case of SQLITE_OMIT_FLOATING_POINT... */
1598 sqlite3_result_double(context, p ? p->rSum : (double)0);
drha97fdd32006-01-12 22:17:50 +00001599}
drhdd5baa92002-02-27 19:50:59 +00001600
1601/*
drh0bce8352002-02-28 00:41:10 +00001602** The following structure keeps track of state information for the
1603** count() aggregate function.
1604*/
1605typedef struct CountCtx CountCtx;
1606struct CountCtx {
drhfc6ad392006-02-09 13:38:19 +00001607 i64 n;
dan7262ca92018-07-02 12:07:32 +00001608#ifdef SQLITE_DEBUG
1609 int bInverse; /* True if xInverse() ever called */
1610#endif
drh0bce8352002-02-28 00:41:10 +00001611};
drhdd5baa92002-02-27 19:50:59 +00001612
drh0bce8352002-02-28 00:41:10 +00001613/*
1614** Routines to implement the count() aggregate function.
1615*/
danielk19770ae8b832004-05-25 12:05:56 +00001616static void countStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drh0bce8352002-02-28 00:41:10 +00001617 CountCtx *p;
drh4f26d6c2004-05-26 23:25:30 +00001618 p = sqlite3_aggregate_context(context, sizeof(*p));
drh9c054832004-05-31 18:51:57 +00001619 if( (argc==0 || SQLITE_NULL!=sqlite3_value_type(argv[0])) && p ){
drh0bce8352002-02-28 00:41:10 +00001620 p->n++;
1621 }
drh2e79c3d2009-04-08 23:04:14 +00001622
drhd3264c72009-04-15 13:39:47 +00001623#ifndef SQLITE_OMIT_DEPRECATED
drh2e79c3d2009-04-08 23:04:14 +00001624 /* The sqlite3_aggregate_count() function is deprecated. But just to make
1625 ** sure it still operates correctly, verify that its count agrees with our
1626 ** internal count when using count(*) and when the total count can be
1627 ** expressed as a 32-bit integer. */
dan7262ca92018-07-02 12:07:32 +00001628 assert( argc==1 || p==0 || p->n>0x7fffffff || p->bInverse
drh2e79c3d2009-04-08 23:04:14 +00001629 || p->n==sqlite3_aggregate_count(context) );
drhd3264c72009-04-15 13:39:47 +00001630#endif
drh0bce8352002-02-28 00:41:10 +00001631}
danielk19770ae8b832004-05-25 12:05:56 +00001632static void countFinalize(sqlite3_context *context){
drh0bce8352002-02-28 00:41:10 +00001633 CountCtx *p;
drhabfcea22005-09-06 20:36:48 +00001634 p = sqlite3_aggregate_context(context, 0);
drhfc6ad392006-02-09 13:38:19 +00001635 sqlite3_result_int64(context, p ? p->n : 0);
drh0bce8352002-02-28 00:41:10 +00001636}
dan7262ca92018-07-02 12:07:32 +00001637#ifndef SQLITE_OMIT_WINDOWFUNC
1638static void countInverse(sqlite3_context *ctx, int argc, sqlite3_value **argv){
1639 CountCtx *p;
1640 p = sqlite3_aggregate_context(ctx, sizeof(*p));
drhfd4b7282018-07-07 19:47:21 +00001641 /* p is always non-NULL since countStep() will have been called first */
1642 if( (argc==0 || SQLITE_NULL!=sqlite3_value_type(argv[0])) && ALWAYS(p) ){
dan7262ca92018-07-02 12:07:32 +00001643 p->n--;
1644#ifdef SQLITE_DEBUG
1645 p->bInverse = 1;
1646#endif
1647 }
1648}
dan6b4b8822018-07-02 15:03:50 +00001649#else
1650# define countInverse 0
1651#endif /* SQLITE_OMIT_WINDOWFUNC */
drh0bce8352002-02-28 00:41:10 +00001652
1653/*
drh0bce8352002-02-28 00:41:10 +00001654** Routines to implement min() and max() aggregate functions.
1655*/
danielk197762c14b32008-11-19 09:05:26 +00001656static void minmaxStep(
1657 sqlite3_context *context,
1658 int NotUsed,
1659 sqlite3_value **argv
1660){
danielk197788208052004-05-25 01:13:20 +00001661 Mem *pArg = (Mem *)argv[0];
drh9eb516c2004-07-18 20:52:32 +00001662 Mem *pBest;
danielk197762c14b32008-11-19 09:05:26 +00001663 UNUSED_PARAMETER(NotUsed);
drh9eb516c2004-07-18 20:52:32 +00001664
drh9eb516c2004-07-18 20:52:32 +00001665 pBest = (Mem *)sqlite3_aggregate_context(context, sizeof(*pBest));
danielk19773aeab9e2004-06-24 00:20:04 +00001666 if( !pBest ) return;
drh268380c2004-02-25 13:47:31 +00001667
dan6fb2b542018-06-19 17:13:11 +00001668 if( sqlite3_value_type(pArg)==SQLITE_NULL ){
drh94a6d992012-02-02 18:42:09 +00001669 if( pBest->flags ) sqlite3SkipAccumulatorLoad(context);
1670 }else if( pBest->flags ){
drh9eb516c2004-07-18 20:52:32 +00001671 int max;
1672 int cmp;
danielk1977dc1bdc42004-06-11 10:51:27 +00001673 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk19777e18c252004-05-25 11:47:24 +00001674 /* This step function is used for both the min() and max() aggregates,
1675 ** the only difference between the two being that the sense of the
1676 ** comparison is inverted. For the max() aggregate, the
1677 ** sqlite3_user_data() function returns (void *)-1. For min() it
1678 ** returns (void *)db, where db is the sqlite3* database pointer.
1679 ** Therefore the next statement sets variable 'max' to 1 for the max()
1680 ** aggregate, or 0 for min().
1681 */
drh309b3382007-03-17 17:52:42 +00001682 max = sqlite3_user_data(context)!=0;
danielk1977dc1bdc42004-06-11 10:51:27 +00001683 cmp = sqlite3MemCompare(pBest, pArg, pColl);
danielk197788208052004-05-25 01:13:20 +00001684 if( (max && cmp<0) || (!max && cmp>0) ){
drhb21c8cd2007-08-21 19:33:56 +00001685 sqlite3VdbeMemCopy(pBest, pArg);
drh7a957892012-02-02 17:35:43 +00001686 }else{
1687 sqlite3SkipAccumulatorLoad(context);
danielk197788208052004-05-25 01:13:20 +00001688 }
drh268380c2004-02-25 13:47:31 +00001689 }else{
drh035e5632014-09-16 14:16:31 +00001690 pBest->db = sqlite3_context_db_handle(context);
drhb21c8cd2007-08-21 19:33:56 +00001691 sqlite3VdbeMemCopy(pBest, pArg);
drh0bce8352002-02-28 00:41:10 +00001692 }
1693}
dan6fb2b542018-06-19 17:13:11 +00001694static void minMaxValueFinalize(sqlite3_context *context, int bValue){
danielk197788208052004-05-25 01:13:20 +00001695 sqlite3_value *pRes;
drhabfcea22005-09-06 20:36:48 +00001696 pRes = (sqlite3_value *)sqlite3_aggregate_context(context, 0);
1697 if( pRes ){
drh94a6d992012-02-02 18:42:09 +00001698 if( pRes->flags ){
drhabfcea22005-09-06 20:36:48 +00001699 sqlite3_result_value(context, pRes);
1700 }
dan6fb2b542018-06-19 17:13:11 +00001701 if( bValue==0 ) sqlite3VdbeMemRelease(pRes);
drh0bce8352002-02-28 00:41:10 +00001702 }
1703}
dan67a9b8e2018-06-22 20:51:35 +00001704#ifndef SQLITE_OMIT_WINDOWFUNC
dan6fb2b542018-06-19 17:13:11 +00001705static void minMaxValue(sqlite3_context *context){
drhc7bf5712018-07-09 22:49:01 +00001706 minMaxValueFinalize(context, 1);
dan6fb2b542018-06-19 17:13:11 +00001707}
dan67a9b8e2018-06-22 20:51:35 +00001708#else
1709# define minMaxValue 0
1710#endif /* SQLITE_OMIT_WINDOWFUNC */
dan6fb2b542018-06-19 17:13:11 +00001711static void minMaxFinalize(sqlite3_context *context){
drhc7bf5712018-07-09 22:49:01 +00001712 minMaxValueFinalize(context, 0);
dan6fb2b542018-06-19 17:13:11 +00001713}
drhdd5baa92002-02-27 19:50:59 +00001714
drhb0689692007-11-01 17:38:30 +00001715/*
1716** group_concat(EXPR, ?SEPARATOR?)
1717*/
1718static void groupConcatStep(
1719 sqlite3_context *context,
1720 int argc,
1721 sqlite3_value **argv
1722){
1723 const char *zVal;
drhade86482007-11-28 22:36:40 +00001724 StrAccum *pAccum;
drhb0689692007-11-01 17:38:30 +00001725 const char *zSep;
drh07d31172009-02-02 21:57:05 +00001726 int nVal, nSep;
1727 assert( argc==1 || argc==2 );
1728 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drhade86482007-11-28 22:36:40 +00001729 pAccum = (StrAccum*)sqlite3_aggregate_context(context, sizeof(*pAccum));
1730
1731 if( pAccum ){
drhbb4957f2008-03-20 14:03:29 +00001732 sqlite3 *db = sqlite3_context_db_handle(context);
dane3bf6322018-06-08 20:58:27 +00001733 int firstTerm = pAccum->mxAlloc==0;
drhbb4957f2008-03-20 14:03:29 +00001734 pAccum->mxAlloc = db->aLimit[SQLITE_LIMIT_LENGTH];
drh8bfd7192009-06-19 16:44:41 +00001735 if( !firstTerm ){
drh07d31172009-02-02 21:57:05 +00001736 if( argc==2 ){
1737 zSep = (char*)sqlite3_value_text(argv[1]);
1738 nSep = sqlite3_value_bytes(argv[1]);
drhb0689692007-11-01 17:38:30 +00001739 }else{
1740 zSep = ",";
drhade86482007-11-28 22:36:40 +00001741 nSep = 1;
drhb0689692007-11-01 17:38:30 +00001742 }
drh0cdbe1a2018-05-09 13:46:26 +00001743 if( zSep ) sqlite3_str_append(pAccum, zSep, nSep);
drhb0689692007-11-01 17:38:30 +00001744 }
drh07d31172009-02-02 21:57:05 +00001745 zVal = (char*)sqlite3_value_text(argv[0]);
1746 nVal = sqlite3_value_bytes(argv[0]);
drh0cdbe1a2018-05-09 13:46:26 +00001747 if( zVal ) sqlite3_str_append(pAccum, zVal, nVal);
drhb0689692007-11-01 17:38:30 +00001748 }
1749}
dan67a9b8e2018-06-22 20:51:35 +00001750#ifndef SQLITE_OMIT_WINDOWFUNC
dan03854d22018-06-08 11:45:28 +00001751static void groupConcatInverse(
1752 sqlite3_context *context,
1753 int argc,
1754 sqlite3_value **argv
1755){
1756 int n;
dan03854d22018-06-08 11:45:28 +00001757 StrAccum *pAccum;
drhc7bf5712018-07-09 22:49:01 +00001758 assert( argc==1 || argc==2 );
dan03854d22018-06-08 11:45:28 +00001759 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
1760 pAccum = (StrAccum*)sqlite3_aggregate_context(context, sizeof(*pAccum));
drhfd4b7282018-07-07 19:47:21 +00001761 /* pAccum is always non-NULL since groupConcatStep() will have always
1762 ** run frist to initialize it */
1763 if( ALWAYS(pAccum) ){
dan03854d22018-06-08 11:45:28 +00001764 n = sqlite3_value_bytes(argv[0]);
1765 if( argc==2 ){
1766 n += sqlite3_value_bytes(argv[1]);
dan683b0ff2018-07-05 18:19:29 +00001767 }else{
1768 n++;
dan03854d22018-06-08 11:45:28 +00001769 }
drhc7bf5712018-07-09 22:49:01 +00001770 if( n>=(int)pAccum->nChar ){
dan03854d22018-06-08 11:45:28 +00001771 pAccum->nChar = 0;
1772 }else{
1773 pAccum->nChar -= n;
1774 memmove(pAccum->zText, &pAccum->zText[n], pAccum->nChar);
1775 }
dane3bf6322018-06-08 20:58:27 +00001776 if( pAccum->nChar==0 ) pAccum->mxAlloc = 0;
dan03854d22018-06-08 11:45:28 +00001777 }
1778}
dan67a9b8e2018-06-22 20:51:35 +00001779#else
1780# define groupConcatInverse 0
1781#endif /* SQLITE_OMIT_WINDOWFUNC */
drhb0689692007-11-01 17:38:30 +00001782static void groupConcatFinalize(sqlite3_context *context){
drhade86482007-11-28 22:36:40 +00001783 StrAccum *pAccum;
1784 pAccum = sqlite3_aggregate_context(context, 0);
1785 if( pAccum ){
drh0cdbe1a2018-05-09 13:46:26 +00001786 if( pAccum->accError==SQLITE_TOOBIG ){
drhade86482007-11-28 22:36:40 +00001787 sqlite3_result_error_toobig(context);
drh0cdbe1a2018-05-09 13:46:26 +00001788 }else if( pAccum->accError==SQLITE_NOMEM ){
drhade86482007-11-28 22:36:40 +00001789 sqlite3_result_error_nomem(context);
1790 }else{
1791 sqlite3_result_text(context, sqlite3StrAccumFinish(pAccum), -1,
1792 sqlite3_free);
1793 }
drhb0689692007-11-01 17:38:30 +00001794 }
1795}
dan67a9b8e2018-06-22 20:51:35 +00001796#ifndef SQLITE_OMIT_WINDOWFUNC
dane2f781b2018-05-17 19:24:08 +00001797static void groupConcatValue(sqlite3_context *context){
1798 sqlite3_str *pAccum;
1799 pAccum = (sqlite3_str*)sqlite3_aggregate_context(context, 0);
1800 if( pAccum ){
1801 if( pAccum->accError==SQLITE_TOOBIG ){
1802 sqlite3_result_error_toobig(context);
1803 }else if( pAccum->accError==SQLITE_NOMEM ){
1804 sqlite3_result_error_nomem(context);
1805 }else{
1806 const char *zText = sqlite3_str_value(pAccum);
1807 sqlite3_result_text(context, zText, -1, SQLITE_TRANSIENT);
1808 }
1809 }
1810}
dan67a9b8e2018-06-22 20:51:35 +00001811#else
1812# define groupConcatValue 0
1813#endif /* SQLITE_OMIT_WINDOWFUNC */
drh4e5ffc52004-08-31 00:52:37 +00001814
drhd3a149e2002-02-24 17:12:53 +00001815/*
drha4741842010-04-25 20:58:37 +00001816** This routine does per-connection function registration. Most
1817** of the built-in functions above are part of the global function set.
1818** This routine only deals with those that are not global.
drhdc04c582002-02-24 01:55:15 +00001819*/
drh80738d92016-02-15 00:34:16 +00001820void sqlite3RegisterPerConnectionBuiltinFunctions(sqlite3 *db){
drha4741842010-04-25 20:58:37 +00001821 int rc = sqlite3_overload_function(db, "MATCH", 2);
1822 assert( rc==SQLITE_NOMEM || rc==SQLITE_OK );
1823 if( rc==SQLITE_NOMEM ){
drh4a642b62016-02-05 01:55:27 +00001824 sqlite3OomFault(db);
danielk1977832a58a2007-06-22 15:21:15 +00001825 }
drh55ef4d92005-08-14 01:20:37 +00001826}
1827
1828/*
drh08652b52019-05-08 17:27:18 +00001829** Re-register the built-in LIKE functions. The caseSensitive
drh55ef4d92005-08-14 01:20:37 +00001830** parameter determines whether or not the LIKE operator is case
drh08652b52019-05-08 17:27:18 +00001831** sensitive.
drh55ef4d92005-08-14 01:20:37 +00001832*/
1833void sqlite3RegisterLikeFunctions(sqlite3 *db, int caseSensitive){
1834 struct compareInfo *pInfo;
drhea5c0402019-05-08 19:32:33 +00001835 int flags;
drh55ef4d92005-08-14 01:20:37 +00001836 if( caseSensitive ){
1837 pInfo = (struct compareInfo*)&likeInfoAlt;
drhea5c0402019-05-08 19:32:33 +00001838 flags = SQLITE_FUNC_LIKE | SQLITE_FUNC_CASE;
drh55ef4d92005-08-14 01:20:37 +00001839 }else{
1840 pInfo = (struct compareInfo*)&likeInfoNorm;
drhea5c0402019-05-08 19:32:33 +00001841 flags = SQLITE_FUNC_LIKE;
drh55ef4d92005-08-14 01:20:37 +00001842 }
dan660af932018-06-18 16:55:22 +00001843 sqlite3CreateFunc(db, "like", 2, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0, 0, 0);
1844 sqlite3CreateFunc(db, "like", 3, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0, 0, 0);
drhea5c0402019-05-08 19:32:33 +00001845 sqlite3FindFunction(db, "like", 2, SQLITE_UTF8, 0)->funcFlags |= flags;
1846 sqlite3FindFunction(db, "like", 3, SQLITE_UTF8, 0)->funcFlags |= flags;
drh55ef4d92005-08-14 01:20:37 +00001847}
1848
1849/*
1850** pExpr points to an expression which implements a function. If
1851** it is appropriate to apply the LIKE optimization to that function
drh1d42ea72017-07-27 20:24:29 +00001852** then set aWc[0] through aWc[2] to the wildcard characters and the
1853** escape character and then return TRUE. If the function is not a
1854** LIKE-style function then return FALSE.
1855**
1856** The expression "a LIKE b ESCAPE c" is only considered a valid LIKE
1857** operator if c is a string literal that is exactly one byte in length.
1858** That one byte is stored in aWc[3]. aWc[3] is set to zero if there is
1859** no ESCAPE clause.
drh16897072015-03-07 00:57:37 +00001860**
1861** *pIsNocase is set to true if uppercase and lowercase are equivalent for
1862** the function (default for LIKE). If the function makes the distinction
1863** between uppercase and lowercase (as does GLOB) then *pIsNocase is set to
1864** false.
drh55ef4d92005-08-14 01:20:37 +00001865*/
drhd64fe2f2005-08-28 17:00:23 +00001866int sqlite3IsLikeFunction(sqlite3 *db, Expr *pExpr, int *pIsNocase, char *aWc){
drh55ef4d92005-08-14 01:20:37 +00001867 FuncDef *pDef;
drh1d42ea72017-07-27 20:24:29 +00001868 int nExpr;
drh17988aa2021-01-22 20:28:30 +00001869 assert( pExpr!=0 );
1870 assert( pExpr->op==TK_FUNCTION );
1871 if( !pExpr->x.pList ){
drh55ef4d92005-08-14 01:20:37 +00001872 return 0;
1873 }
danielk19776ab3a2e2009-02-19 14:39:25 +00001874 assert( !ExprHasProperty(pExpr, EP_xIsSelect) );
drh1d42ea72017-07-27 20:24:29 +00001875 nExpr = pExpr->x.pList->nExpr;
1876 pDef = sqlite3FindFunction(db, pExpr->u.zToken, nExpr, SQLITE_UTF8, 0);
drh78b52202020-01-22 23:08:19 +00001877#ifdef SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION
1878 if( pDef==0 ) return 0;
1879#endif
drhd36e1042013-09-06 13:10:12 +00001880 if( NEVER(pDef==0) || (pDef->funcFlags & SQLITE_FUNC_LIKE)==0 ){
drh55ef4d92005-08-14 01:20:37 +00001881 return 0;
1882 }
1883
1884 /* The memcpy() statement assumes that the wildcard characters are
1885 ** the first three statements in the compareInfo structure. The
1886 ** asserts() that follow verify that assumption
1887 */
1888 memcpy(aWc, pDef->pUserData, 3);
1889 assert( (char*)&likeInfoAlt == (char*)&likeInfoAlt.matchAll );
1890 assert( &((char*)&likeInfoAlt)[1] == (char*)&likeInfoAlt.matchOne );
1891 assert( &((char*)&likeInfoAlt)[2] == (char*)&likeInfoAlt.matchSet );
drh589c7872020-03-19 18:13:28 +00001892
1893 if( nExpr<3 ){
1894 aWc[3] = 0;
1895 }else{
1896 Expr *pEscape = pExpr->x.pList->a[2].pExpr;
1897 char *zEscape;
1898 if( pEscape->op!=TK_STRING ) return 0;
1899 zEscape = pEscape->u.zToken;
1900 if( zEscape[0]==0 || zEscape[1]!=0 ) return 0;
1901 if( zEscape[0]==aWc[0] ) return 0;
1902 if( zEscape[0]==aWc[1] ) return 0;
1903 aWc[3] = zEscape[0];
1904 }
1905
drhd36e1042013-09-06 13:10:12 +00001906 *pIsNocase = (pDef->funcFlags & SQLITE_FUNC_CASE)==0;
drh55ef4d92005-08-14 01:20:37 +00001907 return 1;
drhdc04c582002-02-24 01:55:15 +00001908}
danielk19778c0a7912008-08-20 14:49:23 +00001909
drh63f8f982020-12-07 21:13:06 +00001910/* Mathematical Constants */
1911#ifndef M_PI
1912# define M_PI 3.141592653589793238462643383279502884
1913#endif
1914#ifndef M_LN10
1915# define M_LN10 2.302585092994045684017991454684364208
1916#endif
1917#ifndef M_LN2
1918# define M_LN2 0.693147180559945309417232121458176568
1919#endif
1920
1921
drhf6e904b2020-12-07 17:15:32 +00001922/* Extra math functions that require linking with -lm
1923*/
1924#ifdef SQLITE_ENABLE_MATH_FUNCTIONS
1925/*
1926** Implementation SQL functions:
1927**
1928** ceil(X)
1929** ceiling(X)
1930** floor(X)
1931**
1932** The sqlite3_user_data() pointer is a pointer to the libm implementation
1933** of the underlying C function.
1934*/
1935static void ceilingFunc(
1936 sqlite3_context *context,
1937 int argc,
1938 sqlite3_value **argv
1939){
1940 assert( argc==1 );
drh63f8f982020-12-07 21:13:06 +00001941 switch( sqlite3_value_numeric_type(argv[0]) ){
1942 case SQLITE_INTEGER: {
drhf6e904b2020-12-07 17:15:32 +00001943 sqlite3_result_int64(context, sqlite3_value_int64(argv[0]));
1944 break;
drh63f8f982020-12-07 21:13:06 +00001945 }
1946 case SQLITE_FLOAT: {
drhf6e904b2020-12-07 17:15:32 +00001947 double (*x)(double) = (double(*)(double))sqlite3_user_data(context);
1948 sqlite3_result_double(context, x(sqlite3_value_double(argv[0])));
1949 break;
1950 }
drh63f8f982020-12-07 21:13:06 +00001951 default: {
1952 break;
1953 }
drhf6e904b2020-12-07 17:15:32 +00001954 }
1955}
1956
1957/*
drh4fd4a7a2021-01-01 01:44:06 +00001958** On some systems, ceil() and floor() are intrinsic function. You are
drhc2dbf352021-01-07 16:10:14 +00001959** unable to take a pointer to these functions. Hence, we here wrap them
drh4fd4a7a2021-01-01 01:44:06 +00001960** in our own actual functions.
1961*/
1962static double xCeil(double x){ return ceil(x); }
1963static double xFloor(double x){ return floor(x); }
1964
1965/*
drhf6e904b2020-12-07 17:15:32 +00001966** Implementation of SQL functions:
1967**
1968** ln(X) - natural logarithm
drh63f8f982020-12-07 21:13:06 +00001969** log(X) - log X base 10
1970** log10(X) - log X base 10
1971** log(B,X) - log X base B
drhf6e904b2020-12-07 17:15:32 +00001972*/
1973static void logFunc(
1974 sqlite3_context *context,
1975 int argc,
1976 sqlite3_value **argv
1977){
drh63f8f982020-12-07 21:13:06 +00001978 double x, b, ans;
drhf6e904b2020-12-07 17:15:32 +00001979 assert( argc==1 || argc==2 );
drh63f8f982020-12-07 21:13:06 +00001980 switch( sqlite3_value_numeric_type(argv[0]) ){
1981 case SQLITE_INTEGER:
1982 case SQLITE_FLOAT:
1983 x = sqlite3_value_double(argv[0]);
drh02d6f9b2021-01-29 16:20:16 +00001984 if( x<=0.0 ) return;
drh63f8f982020-12-07 21:13:06 +00001985 break;
1986 default:
1987 return;
drhf6e904b2020-12-07 17:15:32 +00001988 }
drhf6e904b2020-12-07 17:15:32 +00001989 if( argc==2 ){
drh63f8f982020-12-07 21:13:06 +00001990 switch( sqlite3_value_numeric_type(argv[0]) ){
1991 case SQLITE_INTEGER:
1992 case SQLITE_FLOAT:
drh02d6f9b2021-01-29 16:20:16 +00001993 b = log(x);
1994 if( b<=0.0 ) return;
drh63f8f982020-12-07 21:13:06 +00001995 x = sqlite3_value_double(argv[1]);
drh02d6f9b2021-01-29 16:20:16 +00001996 if( x<=0.0 ) return;
drh63f8f982020-12-07 21:13:06 +00001997 break;
1998 default:
1999 return;
drhf6e904b2020-12-07 17:15:32 +00002000 }
drh02d6f9b2021-01-29 16:20:16 +00002001 ans = log(x)/b;
drh63f8f982020-12-07 21:13:06 +00002002 }else{
2003 ans = log(x);
2004 switch( SQLITE_PTR_TO_INT(sqlite3_user_data(context)) ){
2005 case 1:
2006 /* Convert from natural logarithm to log base 10 */
2007 ans *= 1.0/M_LN10;
2008 break;
2009 case 2:
2010 /* Convert from natural logarithm to log base 2 */
2011 ans *= 1.0/M_LN2;
2012 break;
2013 default:
2014 break;
2015 }
drhf6e904b2020-12-07 17:15:32 +00002016 }
2017 sqlite3_result_double(context, ans);
2018}
drh63f8f982020-12-07 21:13:06 +00002019
2020/*
2021** Functions to converts degrees to radians and radians to degrees.
2022*/
2023static double degToRad(double x){ return x*(M_PI/180.0); }
2024static double radToDeg(double x){ return x*(180.0/M_PI); }
2025
2026/*
2027** Implementation of 1-argument SQL math functions:
2028**
2029** exp(X) - Compute e to the X-th power
2030*/
2031static void math1Func(
2032 sqlite3_context *context,
2033 int argc,
2034 sqlite3_value **argv
2035){
drh63f8f982020-12-07 21:13:06 +00002036 int type0;
2037 double v0, ans;
2038 double (*x)(double);
mistachkind97a4c02020-12-09 23:35:51 +00002039 assert( argc==1 );
drh63f8f982020-12-07 21:13:06 +00002040 type0 = sqlite3_value_numeric_type(argv[0]);
2041 if( type0!=SQLITE_INTEGER && type0!=SQLITE_FLOAT ) return;
2042 v0 = sqlite3_value_double(argv[0]);
2043 x = (double(*)(double))sqlite3_user_data(context);
2044 ans = x(v0);
2045 sqlite3_result_double(context, ans);
2046}
2047
2048/*
2049** Implementation of 2-argument SQL math functions:
2050**
2051** power(X,Y) - Compute X to the Y-th power
2052*/
2053static void math2Func(
2054 sqlite3_context *context,
2055 int argc,
2056 sqlite3_value **argv
2057){
drh63f8f982020-12-07 21:13:06 +00002058 int type0, type1;
2059 double v0, v1, ans;
2060 double (*x)(double,double);
mistachkind97a4c02020-12-09 23:35:51 +00002061 assert( argc==2 );
drh63f8f982020-12-07 21:13:06 +00002062 type0 = sqlite3_value_numeric_type(argv[0]);
2063 if( type0!=SQLITE_INTEGER && type0!=SQLITE_FLOAT ) return;
2064 type1 = sqlite3_value_numeric_type(argv[1]);
2065 if( type1!=SQLITE_INTEGER && type1!=SQLITE_FLOAT ) return;
2066 v0 = sqlite3_value_double(argv[0]);
2067 v1 = sqlite3_value_double(argv[1]);
2068 x = (double(*)(double,double))sqlite3_user_data(context);
2069 ans = x(v0, v1);
2070 sqlite3_result_double(context, ans);
2071}
2072
2073/*
2074** Implementation of 2-argument SQL math functions:
2075**
2076** power(X,Y) - Compute X to the Y-th power
2077*/
2078static void piFunc(
2079 sqlite3_context *context,
2080 int argc,
2081 sqlite3_value **argv
2082){
2083 assert( argc==0 );
2084 sqlite3_result_double(context, M_PI);
2085}
2086
drhf6e904b2020-12-07 17:15:32 +00002087#endif /* SQLITE_ENABLE_MATH_FUNCTIONS */
2088
drh70a8ca32008-08-21 18:49:27 +00002089/*
drh63f8f982020-12-07 21:13:06 +00002090** Implementation of sign(X) function.
2091*/
2092static void signFunc(
2093 sqlite3_context *context,
2094 int argc,
2095 sqlite3_value **argv
2096){
drh63f8f982020-12-07 21:13:06 +00002097 int type0;
2098 double x;
drhe5baf5c2020-12-16 14:20:45 +00002099 UNUSED_PARAMETER(argc);
mistachkind97a4c02020-12-09 23:35:51 +00002100 assert( argc==1 );
drh63f8f982020-12-07 21:13:06 +00002101 type0 = sqlite3_value_numeric_type(argv[0]);
2102 if( type0!=SQLITE_INTEGER && type0!=SQLITE_FLOAT ) return;
2103 x = sqlite3_value_double(argv[0]);
2104 sqlite3_result_int(context, x<0.0 ? -1 : x>0.0 ? +1 : 0);
2105}
2106
2107/*
peter.d.reid60ec9142014-09-06 16:39:46 +00002108** All of the FuncDef structures in the aBuiltinFunc[] array above
drh777c5382008-08-21 20:21:34 +00002109** to the global function hash table. This occurs at start-time (as
2110** a consequence of calling sqlite3_initialize()).
2111**
2112** After this routine runs
drh70a8ca32008-08-21 18:49:27 +00002113*/
drh80738d92016-02-15 00:34:16 +00002114void sqlite3RegisterBuiltinFunctions(void){
danielk197793ce7412008-09-01 19:14:02 +00002115 /*
2116 ** The following array holds FuncDef structures for all of the functions
2117 ** defined in this file.
2118 **
2119 ** The array cannot be constant since changes are made to the
2120 ** FuncDef.pHash elements at start-time. The elements of this array
2121 ** are read-only after initialization is complete.
drh80738d92016-02-15 00:34:16 +00002122 **
2123 ** For peak efficiency, put the most frequently used function last.
danielk197793ce7412008-09-01 19:14:02 +00002124 */
drh80738d92016-02-15 00:34:16 +00002125 static FuncDef aBuiltinFunc[] = {
drh171c50e2020-01-01 15:43:30 +00002126/***** Functions only available with SQLITE_TESTCTRL_INTERNAL_FUNCTIONS *****/
2127 TEST_FUNC(implies_nonnull_row, 2, INLINEFUNC_implies_nonnull_row, 0),
2128 TEST_FUNC(expr_compare, 2, INLINEFUNC_expr_compare, 0),
2129 TEST_FUNC(expr_implies_expr, 2, INLINEFUNC_expr_implies_expr, 0),
2130#ifdef SQLITE_DEBUG
2131 TEST_FUNC(affinity, 1, INLINEFUNC_affinity, 0),
2132#endif
2133/***** Regular functions *****/
drh80738d92016-02-15 00:34:16 +00002134#ifdef SQLITE_SOUNDEX
2135 FUNCTION(soundex, 1, 0, 0, soundexFunc ),
2136#endif
2137#ifndef SQLITE_OMIT_LOAD_EXTENSION
drh64de2a52019-12-31 18:39:23 +00002138 SFUNCTION(load_extension, 1, 0, 0, loadExt ),
2139 SFUNCTION(load_extension, 2, 0, 0, loadExt ),
drh80738d92016-02-15 00:34:16 +00002140#endif
2141#if SQLITE_USER_AUTHENTICATION
2142 FUNCTION(sqlite_crypt, 2, 0, 0, sqlite3CryptFunc ),
2143#endif
2144#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
2145 DFUNCTION(sqlite_compileoption_used,1, 0, 0, compileoptionusedFunc ),
2146 DFUNCTION(sqlite_compileoption_get, 1, 0, 0, compileoptiongetFunc ),
2147#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
drh25c42962020-01-01 13:55:08 +00002148 INLINE_FUNC(unlikely, 1, INLINEFUNC_unlikely, SQLITE_FUNC_UNLIKELY),
2149 INLINE_FUNC(likelihood, 2, INLINEFUNC_unlikely, SQLITE_FUNC_UNLIKELY),
2150 INLINE_FUNC(likely, 1, INLINEFUNC_unlikely, SQLITE_FUNC_UNLIKELY),
drh092457b2017-12-29 15:04:49 +00002151#ifdef SQLITE_ENABLE_OFFSET_SQL_FUNC
drh35100fb2018-01-04 19:20:37 +00002152 FUNCTION2(sqlite_offset, 1, 0, 0, noopFunc, SQLITE_FUNC_OFFSET|
drh2fc865c2017-12-16 20:20:37 +00002153 SQLITE_FUNC_TYPEOF),
drh092457b2017-12-29 15:04:49 +00002154#endif
danielk197793ce7412008-09-01 19:14:02 +00002155 FUNCTION(ltrim, 1, 1, 0, trimFunc ),
2156 FUNCTION(ltrim, 2, 1, 0, trimFunc ),
2157 FUNCTION(rtrim, 1, 2, 0, trimFunc ),
2158 FUNCTION(rtrim, 2, 2, 0, trimFunc ),
2159 FUNCTION(trim, 1, 3, 0, trimFunc ),
2160 FUNCTION(trim, 2, 3, 0, trimFunc ),
2161 FUNCTION(min, -1, 0, 1, minmaxFunc ),
2162 FUNCTION(min, 0, 0, 1, 0 ),
dan6fb2b542018-06-19 17:13:11 +00002163 WAGGREGATE(min, 1, 0, 1, minmaxStep, minMaxFinalize, minMaxValue, 0,
drh9588ad92014-09-15 14:46:02 +00002164 SQLITE_FUNC_MINMAX ),
danielk197793ce7412008-09-01 19:14:02 +00002165 FUNCTION(max, -1, 1, 1, minmaxFunc ),
2166 FUNCTION(max, 0, 1, 1, 0 ),
dan6fb2b542018-06-19 17:13:11 +00002167 WAGGREGATE(max, 1, 1, 1, minmaxStep, minMaxFinalize, minMaxValue, 0,
drh9588ad92014-09-15 14:46:02 +00002168 SQLITE_FUNC_MINMAX ),
drha748fdc2012-03-28 01:34:47 +00002169 FUNCTION2(typeof, 1, 0, 0, typeofFunc, SQLITE_FUNC_TYPEOF),
2170 FUNCTION2(length, 1, 0, 0, lengthFunc, SQLITE_FUNC_LENGTH),
drhd55e0722012-10-25 03:07:29 +00002171 FUNCTION(instr, 2, 0, 0, instrFunc ),
drha5c14162013-12-17 15:03:06 +00002172 FUNCTION(printf, -1, 0, 0, printfFunc ),
drhd495d8c2013-02-22 19:34:25 +00002173 FUNCTION(unicode, 1, 0, 0, unicodeFunc ),
2174 FUNCTION(char, -1, 0, 0, charFunc ),
danielk197793ce7412008-09-01 19:14:02 +00002175 FUNCTION(abs, 1, 0, 0, absFunc ),
shanefbd60f82009-02-04 03:59:25 +00002176#ifndef SQLITE_OMIT_FLOATING_POINT
danielk197793ce7412008-09-01 19:14:02 +00002177 FUNCTION(round, 1, 0, 0, roundFunc ),
2178 FUNCTION(round, 2, 0, 0, roundFunc ),
shanefbd60f82009-02-04 03:59:25 +00002179#endif
danielk197793ce7412008-09-01 19:14:02 +00002180 FUNCTION(upper, 1, 0, 0, upperFunc ),
2181 FUNCTION(lower, 1, 0, 0, lowerFunc ),
danielk197793ce7412008-09-01 19:14:02 +00002182 FUNCTION(hex, 1, 0, 0, hexFunc ),
drhffe421c2020-05-13 17:26:38 +00002183 INLINE_FUNC(ifnull, 2, INLINEFUNC_coalesce, 0 ),
drhb1fba282013-11-21 14:33:48 +00002184 VFUNCTION(random, 0, 0, 0, randomFunc ),
2185 VFUNCTION(randomblob, 1, 0, 0, randomBlob ),
danielk197793ce7412008-09-01 19:14:02 +00002186 FUNCTION(nullif, 2, 0, 1, nullifFunc ),
drh03bf26d2015-08-31 21:16:36 +00002187 DFUNCTION(sqlite_version, 0, 0, 0, versionFunc ),
2188 DFUNCTION(sqlite_source_id, 0, 0, 0, sourceidFunc ),
drh840561f2011-04-27 18:08:42 +00002189 FUNCTION(sqlite_log, 2, 0, 0, errlogFunc ),
danielk197793ce7412008-09-01 19:14:02 +00002190 FUNCTION(quote, 1, 0, 0, quoteFunc ),
drhb1fba282013-11-21 14:33:48 +00002191 VFUNCTION(last_insert_rowid, 0, 0, 0, last_insert_rowid),
2192 VFUNCTION(changes, 0, 0, 0, changes ),
2193 VFUNCTION(total_changes, 0, 0, 0, total_changes ),
danielk197793ce7412008-09-01 19:14:02 +00002194 FUNCTION(replace, 3, 0, 0, replaceFunc ),
2195 FUNCTION(zeroblob, 1, 0, 0, zeroblobFunc ),
drh80738d92016-02-15 00:34:16 +00002196 FUNCTION(substr, 2, 0, 0, substrFunc ),
2197 FUNCTION(substr, 3, 0, 0, substrFunc ),
drh1335ec72020-11-23 14:50:43 +00002198 FUNCTION(substring, 2, 0, 0, substrFunc ),
2199 FUNCTION(substring, 3, 0, 0, substrFunc ),
dan6fb2b542018-06-19 17:13:11 +00002200 WAGGREGATE(sum, 1,0,0, sumStep, sumFinalize, sumFinalize, sumInverse, 0),
2201 WAGGREGATE(total, 1,0,0, sumStep,totalFinalize,totalFinalize,sumInverse, 0),
2202 WAGGREGATE(avg, 1,0,0, sumStep, avgFinalize, avgFinalize, sumInverse, 0),
dan7262ca92018-07-02 12:07:32 +00002203 WAGGREGATE(count, 0,0,0, countStep,
2204 countFinalize, countFinalize, countInverse, SQLITE_FUNC_COUNT ),
2205 WAGGREGATE(count, 1,0,0, countStep,
2206 countFinalize, countFinalize, countInverse, 0 ),
dan03854d22018-06-08 11:45:28 +00002207 WAGGREGATE(group_concat, 1, 0, 0, groupConcatStep,
dan6fb2b542018-06-19 17:13:11 +00002208 groupConcatFinalize, groupConcatValue, groupConcatInverse, 0),
dan03854d22018-06-08 11:45:28 +00002209 WAGGREGATE(group_concat, 2, 0, 0, groupConcatStep,
dan6fb2b542018-06-19 17:13:11 +00002210 groupConcatFinalize, groupConcatValue, groupConcatInverse, 0),
danielk197793ce7412008-09-01 19:14:02 +00002211
2212 LIKEFUNC(glob, 2, &globInfo, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
drhcc153132016-08-04 12:35:17 +00002213#ifdef SQLITE_CASE_SENSITIVE_LIKE
danielk197793ce7412008-09-01 19:14:02 +00002214 LIKEFUNC(like, 2, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
2215 LIKEFUNC(like, 3, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
drhcc153132016-08-04 12:35:17 +00002216#else
danielk197793ce7412008-09-01 19:14:02 +00002217 LIKEFUNC(like, 2, &likeInfoNorm, SQLITE_FUNC_LIKE),
2218 LIKEFUNC(like, 3, &likeInfoNorm, SQLITE_FUNC_LIKE),
drhcc153132016-08-04 12:35:17 +00002219#endif
2220#ifdef SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION
2221 FUNCTION(unknown, -1, 0, 0, unknownFunc ),
2222#endif
drh80738d92016-02-15 00:34:16 +00002223 FUNCTION(coalesce, 1, 0, 0, 0 ),
2224 FUNCTION(coalesce, 0, 0, 0, 0 ),
drhf6e904b2020-12-07 17:15:32 +00002225#ifdef SQLITE_ENABLE_MATH_FUNCTIONS
drh4fd4a7a2021-01-01 01:44:06 +00002226 MFUNCTION(ceil, 1, xCeil, ceilingFunc ),
2227 MFUNCTION(ceiling, 1, xCeil, ceilingFunc ),
2228 MFUNCTION(floor, 1, xFloor, ceilingFunc ),
mistachkind97a4c02020-12-09 23:35:51 +00002229#if SQLITE_HAVE_C99_MATH_FUNCS
drh63f8f982020-12-07 21:13:06 +00002230 MFUNCTION(trunc, 1, trunc, ceilingFunc ),
mistachkind97a4c02020-12-09 23:35:51 +00002231#endif
drhf6e904b2020-12-07 17:15:32 +00002232 FUNCTION(ln, 1, 0, 0, logFunc ),
2233 FUNCTION(log, 1, 1, 0, logFunc ),
2234 FUNCTION(log10, 1, 1, 0, logFunc ),
drh63f8f982020-12-07 21:13:06 +00002235 FUNCTION(log2, 1, 2, 0, logFunc ),
drhf6e904b2020-12-07 17:15:32 +00002236 FUNCTION(log, 2, 0, 0, logFunc ),
drh63f8f982020-12-07 21:13:06 +00002237 MFUNCTION(exp, 1, exp, math1Func ),
2238 MFUNCTION(pow, 2, pow, math2Func ),
2239 MFUNCTION(power, 2, pow, math2Func ),
2240 MFUNCTION(mod, 2, fmod, math2Func ),
2241 MFUNCTION(acos, 1, acos, math1Func ),
2242 MFUNCTION(asin, 1, asin, math1Func ),
2243 MFUNCTION(atan, 1, atan, math1Func ),
2244 MFUNCTION(atan2, 2, atan2, math2Func ),
2245 MFUNCTION(cos, 1, cos, math1Func ),
2246 MFUNCTION(sin, 1, sin, math1Func ),
2247 MFUNCTION(tan, 1, tan, math1Func ),
2248 MFUNCTION(cosh, 1, cosh, math1Func ),
2249 MFUNCTION(sinh, 1, sinh, math1Func ),
2250 MFUNCTION(tanh, 1, tanh, math1Func ),
mistachkind97a4c02020-12-09 23:35:51 +00002251#if SQLITE_HAVE_C99_MATH_FUNCS
drh63f8f982020-12-07 21:13:06 +00002252 MFUNCTION(acosh, 1, acosh, math1Func ),
2253 MFUNCTION(asinh, 1, asinh, math1Func ),
2254 MFUNCTION(atanh, 1, atanh, math1Func ),
mistachkind97a4c02020-12-09 23:35:51 +00002255#endif
drh63f8f982020-12-07 21:13:06 +00002256 MFUNCTION(sqrt, 1, sqrt, math1Func ),
2257 MFUNCTION(radians, 1, degToRad, math1Func ),
2258 MFUNCTION(degrees, 1, radToDeg, math1Func ),
2259 FUNCTION(pi, 0, 0, 0, piFunc ),
drhf6e904b2020-12-07 17:15:32 +00002260#endif /* SQLITE_ENABLE_MATH_FUNCTIONS */
drh63f8f982020-12-07 21:13:06 +00002261 FUNCTION(sign, 1, 0, 0, signFunc ),
2262 INLINE_FUNC(coalesce, -1, INLINEFUNC_coalesce, 0 ),
2263 INLINE_FUNC(iif, 3, INLINEFUNC_iif, 0 ),
danielk197793ce7412008-09-01 19:14:02 +00002264 };
drh545f5872010-04-24 14:02:59 +00002265#ifndef SQLITE_OMIT_ALTERTABLE
2266 sqlite3AlterFunctions();
2267#endif
dandfa552f2018-06-02 21:04:28 +00002268 sqlite3WindowFunctions();
drh80738d92016-02-15 00:34:16 +00002269 sqlite3RegisterDateTimeFunctions();
2270 sqlite3InsertBuiltinFuncs(aBuiltinFunc, ArraySize(aBuiltinFunc));
2271
2272#if 0 /* Enable to print out how the built-in functions are hashed */
2273 {
2274 int i;
2275 FuncDef *p;
2276 for(i=0; i<SQLITE_FUNC_HASH_SZ; i++){
2277 printf("FUNC-HASH %02d:", i);
2278 for(p=sqlite3BuiltinFunctions.a[i]; p; p=p->u.pHash){
2279 int n = sqlite3Strlen30(p->zName);
2280 int h = p->zName[0] + n;
2281 printf(" %s(%d)", p->zName, h);
2282 }
2283 printf("\n");
2284 }
2285 }
2286#endif
drh70a8ca32008-08-21 18:49:27 +00002287}