blob: 3d5a059a9f1fa4ee5def89bed9bd791a207ec210 [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>
danielk197788208052004-05-25 01:13:20 +000019#include "vdbeInt.h"
drh0bce8352002-02-28 00:41:10 +000020
drh55ef4d92005-08-14 01:20:37 +000021/*
22** Return the collating function associated with a function.
23*/
danielk1977dc1bdc42004-06-11 10:51:27 +000024static CollSeq *sqlite3GetFuncCollSeq(sqlite3_context *context){
drha9e03b12015-03-12 06:46:52 +000025 VdbeOp *pOp;
26 assert( context->pVdbe!=0 );
27 pOp = &context->pVdbe->aOp[context->iOp-1];
drha15cc472014-09-25 13:17:30 +000028 assert( pOp->opcode==OP_CollSeq );
29 assert( pOp->p4type==P4_COLLSEQ );
30 return pOp->p4.pColl;
danielk1977dc1bdc42004-06-11 10:51:27 +000031}
32
drh0bce8352002-02-28 00:41:10 +000033/*
drh7a957892012-02-02 17:35:43 +000034** Indicate that the accumulator load should be skipped on this
35** iteration of the aggregate loop.
36*/
37static void sqlite3SkipAccumulatorLoad(sqlite3_context *context){
38 context->skipFlag = 1;
39}
40
41/*
drh0bce8352002-02-28 00:41:10 +000042** Implementation of the non-aggregate min() and max() functions
43*/
drhf9b596e2004-05-26 16:54:42 +000044static void minmaxFunc(
45 sqlite3_context *context,
46 int argc,
47 sqlite3_value **argv
48){
drh0bce8352002-02-28 00:41:10 +000049 int i;
drh268380c2004-02-25 13:47:31 +000050 int mask; /* 0 for min() or 0xffffffff for max() */
drhf9b596e2004-05-26 16:54:42 +000051 int iBest;
danielk1977dc1bdc42004-06-11 10:51:27 +000052 CollSeq *pColl;
drh0bce8352002-02-28 00:41:10 +000053
drh65595cd2009-02-02 16:32:55 +000054 assert( argc>1 );
drhc44af712004-09-02 15:53:56 +000055 mask = sqlite3_user_data(context)==0 ? 0 : -1;
danielk1977dc1bdc42004-06-11 10:51:27 +000056 pColl = sqlite3GetFuncCollSeq(context);
57 assert( pColl );
danielk1977c572ef72004-05-27 09:28:41 +000058 assert( mask==-1 || mask==0 );
drhf9b596e2004-05-26 16:54:42 +000059 iBest = 0;
drh9c054832004-05-31 18:51:57 +000060 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drhf9b596e2004-05-26 16:54:42 +000061 for(i=1; i<argc; i++){
drh9c054832004-05-31 18:51:57 +000062 if( sqlite3_value_type(argv[i])==SQLITE_NULL ) return;
danielk1977dc1bdc42004-06-11 10:51:27 +000063 if( (sqlite3MemCompare(argv[iBest], argv[i], pColl)^mask)>=0 ){
drh65595cd2009-02-02 16:32:55 +000064 testcase( mask==0 );
drhf9b596e2004-05-26 16:54:42 +000065 iBest = i;
drh0bce8352002-02-28 00:41:10 +000066 }
67 }
drhf4479502004-05-27 03:12:53 +000068 sqlite3_result_value(context, argv[iBest]);
drh0bce8352002-02-28 00:41:10 +000069}
drh0bce8352002-02-28 00:41:10 +000070
drh268380c2004-02-25 13:47:31 +000071/*
72** Return the type of the argument.
73*/
drhf9b596e2004-05-26 16:54:42 +000074static void typeofFunc(
75 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +000076 int NotUsed,
drhf9b596e2004-05-26 16:54:42 +000077 sqlite3_value **argv
78){
danielk197735bb9d02004-05-24 12:55:54 +000079 const char *z = 0;
danielk197762c14b32008-11-19 09:05:26 +000080 UNUSED_PARAMETER(NotUsed);
danielk197735bb9d02004-05-24 12:55:54 +000081 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +000082 case SQLITE_INTEGER: z = "integer"; break;
83 case SQLITE_TEXT: z = "text"; break;
84 case SQLITE_FLOAT: z = "real"; break;
85 case SQLITE_BLOB: z = "blob"; break;
drh65595cd2009-02-02 16:32:55 +000086 default: z = "null"; break;
danielk197735bb9d02004-05-24 12:55:54 +000087 }
danielk1977d8123362004-06-12 09:25:12 +000088 sqlite3_result_text(context, z, -1, SQLITE_STATIC);
drh0bce8352002-02-28 00:41:10 +000089}
90
drh5708d2d2005-06-22 10:53:59 +000091
92/*
drh0bce8352002-02-28 00:41:10 +000093** Implementation of the length() function
94*/
drhf9b596e2004-05-26 16:54:42 +000095static void lengthFunc(
96 sqlite3_context *context,
97 int argc,
98 sqlite3_value **argv
99){
drh0bce8352002-02-28 00:41:10 +0000100 int len;
101
102 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000103 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +0000104 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +0000105 case SQLITE_BLOB:
106 case SQLITE_INTEGER:
107 case SQLITE_FLOAT: {
drhf4479502004-05-27 03:12:53 +0000108 sqlite3_result_int(context, sqlite3_value_bytes(argv[0]));
drhf9b596e2004-05-26 16:54:42 +0000109 break;
110 }
drh9c054832004-05-31 18:51:57 +0000111 case SQLITE_TEXT: {
drh2646da72005-12-09 20:02:05 +0000112 const unsigned char *z = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +0000113 if( z==0 ) return;
drh4a919112007-05-15 11:55:09 +0000114 len = 0;
115 while( *z ){
116 len++;
117 SQLITE_SKIP_UTF8(z);
118 }
drhf4479502004-05-27 03:12:53 +0000119 sqlite3_result_int(context, len);
drhf9b596e2004-05-26 16:54:42 +0000120 break;
121 }
122 default: {
123 sqlite3_result_null(context);
124 break;
125 }
126 }
drh0bce8352002-02-28 00:41:10 +0000127}
128
129/*
drh2ba3ccc2009-12-08 02:06:08 +0000130** Implementation of the abs() function.
131**
132** IMP: R-23979-26855 The abs(X) function returns the absolute value of
133** the numeric argument X.
drh0bce8352002-02-28 00:41:10 +0000134*/
danielk19770ae8b832004-05-25 12:05:56 +0000135static void absFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drh0bce8352002-02-28 00:41:10 +0000136 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000137 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +0000138 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +0000139 case SQLITE_INTEGER: {
danielk1977f93bbbe2004-05-27 10:30:52 +0000140 i64 iVal = sqlite3_value_int64(argv[0]);
drh52fc8492006-02-23 21:43:55 +0000141 if( iVal<0 ){
drh693e6712014-01-24 22:58:00 +0000142 if( iVal==SMALLEST_INT64 ){
drheb091cd2013-11-09 19:47:15 +0000143 /* IMP: R-31676-45509 If X is the integer -9223372036854775808
144 ** then abs(X) throws an integer overflow error since there is no
drh2ba3ccc2009-12-08 02:06:08 +0000145 ** equivalent positive 64-bit two complement value. */
drh52fc8492006-02-23 21:43:55 +0000146 sqlite3_result_error(context, "integer overflow", -1);
147 return;
148 }
149 iVal = -iVal;
150 }
danielk1977f93bbbe2004-05-27 10:30:52 +0000151 sqlite3_result_int64(context, iVal);
drhf9b596e2004-05-26 16:54:42 +0000152 break;
153 }
drh9c054832004-05-31 18:51:57 +0000154 case SQLITE_NULL: {
drh2ba3ccc2009-12-08 02:06:08 +0000155 /* IMP: R-37434-19929 Abs(X) returns NULL if X is NULL. */
drhf9b596e2004-05-26 16:54:42 +0000156 sqlite3_result_null(context);
157 break;
158 }
159 default: {
drh2ba3ccc2009-12-08 02:06:08 +0000160 /* Because sqlite3_value_double() returns 0.0 if the argument is not
161 ** something that can be converted into a number, we have:
drh643091f2014-11-20 23:21:23 +0000162 ** IMP: R-01992-00519 Abs(X) returns 0.0 if X is a string or blob
163 ** that cannot be converted to a numeric value.
drh2ba3ccc2009-12-08 02:06:08 +0000164 */
danielk1977f93bbbe2004-05-27 10:30:52 +0000165 double rVal = sqlite3_value_double(argv[0]);
drh52fc8492006-02-23 21:43:55 +0000166 if( rVal<0 ) rVal = -rVal;
danielk1977f93bbbe2004-05-27 10:30:52 +0000167 sqlite3_result_double(context, rVal);
drhf9b596e2004-05-26 16:54:42 +0000168 break;
169 }
170 }
drh0bce8352002-02-28 00:41:10 +0000171}
172
173/*
drhd55e0722012-10-25 03:07:29 +0000174** Implementation of the instr() function.
175**
176** instr(haystack,needle) finds the first occurrence of needle
177** in haystack and returns the number of previous characters plus 1,
178** or 0 if needle does not occur within haystack.
179**
180** If both haystack and needle are BLOBs, then the result is one more than
181** the number of bytes in haystack prior to the first occurrence of needle,
182** or 0 if needle never occurs in haystack.
183*/
184static void instrFunc(
185 sqlite3_context *context,
186 int argc,
187 sqlite3_value **argv
188){
189 const unsigned char *zHaystack;
190 const unsigned char *zNeedle;
191 int nHaystack;
192 int nNeedle;
193 int typeHaystack, typeNeedle;
194 int N = 1;
195 int isText;
196
drh68c804b2012-12-04 11:03:11 +0000197 UNUSED_PARAMETER(argc);
drhd55e0722012-10-25 03:07:29 +0000198 typeHaystack = sqlite3_value_type(argv[0]);
199 typeNeedle = sqlite3_value_type(argv[1]);
200 if( typeHaystack==SQLITE_NULL || typeNeedle==SQLITE_NULL ) return;
201 nHaystack = sqlite3_value_bytes(argv[0]);
202 nNeedle = sqlite3_value_bytes(argv[1]);
dan895decf2016-12-30 14:15:56 +0000203 if( nNeedle>0 ){
204 if( typeHaystack==SQLITE_BLOB && typeNeedle==SQLITE_BLOB ){
205 zHaystack = sqlite3_value_blob(argv[0]);
206 zNeedle = sqlite3_value_blob(argv[1]);
207 isText = 0;
208 }else{
209 zHaystack = sqlite3_value_text(argv[0]);
210 zNeedle = sqlite3_value_text(argv[1]);
211 isText = 1;
dan895decf2016-12-30 14:15:56 +0000212 }
drhb30574b2017-03-16 14:28:52 +0000213 if( zNeedle==0 || (nHaystack && zHaystack==0) ) return;
dan895decf2016-12-30 14:15:56 +0000214 while( nNeedle<=nHaystack && memcmp(zHaystack, zNeedle, nNeedle)!=0 ){
215 N++;
216 do{
217 nHaystack--;
218 zHaystack++;
219 }while( isText && (zHaystack[0]&0xc0)==0x80 );
220 }
221 if( nNeedle>nHaystack ) N = 0;
drhd55e0722012-10-25 03:07:29 +0000222 }
drhd55e0722012-10-25 03:07:29 +0000223 sqlite3_result_int(context, N);
224}
225
226/*
drha5c14162013-12-17 15:03:06 +0000227** Implementation of the printf() function.
228*/
229static void printfFunc(
230 sqlite3_context *context,
231 int argc,
232 sqlite3_value **argv
233){
234 PrintfArguments x;
235 StrAccum str;
236 const char *zFormat;
237 int n;
drhc0490572015-05-02 11:45:53 +0000238 sqlite3 *db = sqlite3_context_db_handle(context);
drha5c14162013-12-17 15:03:06 +0000239
240 if( argc>=1 && (zFormat = (const char*)sqlite3_value_text(argv[0]))!=0 ){
241 x.nArg = argc-1;
242 x.nUsed = 0;
243 x.apArg = argv+1;
drhc0490572015-05-02 11:45:53 +0000244 sqlite3StrAccumInit(&str, db, 0, 0, db->aLimit[SQLITE_LIMIT_LENGTH]);
drh5f4a6862016-01-30 12:50:25 +0000245 str.printfFlags = SQLITE_PRINTF_SQLFUNC;
246 sqlite3XPrintf(&str, zFormat, &x);
drha5c14162013-12-17 15:03:06 +0000247 n = str.nChar;
248 sqlite3_result_text(context, sqlite3StrAccumFinish(&str), n,
249 SQLITE_DYNAMIC);
250 }
251}
252
253/*
drhf764e6f2007-05-15 01:13:47 +0000254** Implementation of the substr() function.
255**
256** substr(x,p1,p2) returns p2 characters of x[] beginning with p1.
257** p1 is 1-indexed. So substr(x,1,1) returns the first character
258** of x. If x is text, then we actually count UTF-8 characters.
259** If x is a blob, then we count bytes.
260**
261** If p1 is negative, then we begin abs(p1) from the end of x[].
shaneh779b8f12009-11-12 05:04:50 +0000262**
drhf7b54962013-05-28 12:11:54 +0000263** If p2 is negative, return the p2 characters preceding p1.
drh0bce8352002-02-28 00:41:10 +0000264*/
drhf9b596e2004-05-26 16:54:42 +0000265static void substrFunc(
266 sqlite3_context *context,
267 int argc,
268 sqlite3_value **argv
269){
drh2646da72005-12-09 20:02:05 +0000270 const unsigned char *z;
271 const unsigned char *z2;
drh023ae032007-05-08 12:12:16 +0000272 int len;
drhf764e6f2007-05-15 01:13:47 +0000273 int p0type;
drh023ae032007-05-08 12:12:16 +0000274 i64 p1, p2;
drh65595cd2009-02-02 16:32:55 +0000275 int negP2 = 0;
drhf9b596e2004-05-26 16:54:42 +0000276
drh64f31512007-10-12 19:11:55 +0000277 assert( argc==3 || argc==2 );
drh8198d252009-02-01 19:42:37 +0000278 if( sqlite3_value_type(argv[1])==SQLITE_NULL
279 || (argc==3 && sqlite3_value_type(argv[2])==SQLITE_NULL)
280 ){
281 return;
282 }
drhf764e6f2007-05-15 01:13:47 +0000283 p0type = sqlite3_value_type(argv[0]);
drh4adc4cb2009-11-11 20:53:31 +0000284 p1 = sqlite3_value_int(argv[1]);
drhf764e6f2007-05-15 01:13:47 +0000285 if( p0type==SQLITE_BLOB ){
286 len = sqlite3_value_bytes(argv[0]);
287 z = sqlite3_value_blob(argv[0]);
288 if( z==0 ) return;
drh1f0feef2007-05-15 13:27:07 +0000289 assert( len==sqlite3_value_bytes(argv[0]) );
drhf764e6f2007-05-15 01:13:47 +0000290 }else{
291 z = sqlite3_value_text(argv[0]);
292 if( z==0 ) return;
drh4a919112007-05-15 11:55:09 +0000293 len = 0;
drh4adc4cb2009-11-11 20:53:31 +0000294 if( p1<0 ){
295 for(z2=z; *z2; len++){
296 SQLITE_SKIP_UTF8(z2);
297 }
drh4a919112007-05-15 11:55:09 +0000298 }
drhf764e6f2007-05-15 01:13:47 +0000299 }
drh883ad042015-02-19 00:29:11 +0000300#ifdef SQLITE_SUBSTR_COMPATIBILITY
301 /* If SUBSTR_COMPATIBILITY is defined then substr(X,0,N) work the same as
302 ** as substr(X,1,N) - it returns the first N characters of X. This
303 ** is essentially a back-out of the bug-fix in check-in [5fc125d362df4b8]
304 ** from 2009-02-02 for compatibility of applications that exploited the
305 ** old buggy behavior. */
306 if( p1==0 ) p1 = 1; /* <rdar://problem/6778339> */
307#endif
drh64f31512007-10-12 19:11:55 +0000308 if( argc==3 ){
309 p2 = sqlite3_value_int(argv[2]);
drh65595cd2009-02-02 16:32:55 +0000310 if( p2<0 ){
311 p2 = -p2;
312 negP2 = 1;
313 }
drh64f31512007-10-12 19:11:55 +0000314 }else{
drhbb4957f2008-03-20 14:03:29 +0000315 p2 = sqlite3_context_db_handle(context)->aLimit[SQLITE_LIMIT_LENGTH];
drh64f31512007-10-12 19:11:55 +0000316 }
drh0bce8352002-02-28 00:41:10 +0000317 if( p1<0 ){
drh89425d52002-02-28 03:04:48 +0000318 p1 += len;
drh653bc752002-02-28 03:31:10 +0000319 if( p1<0 ){
320 p2 += p1;
drh65595cd2009-02-02 16:32:55 +0000321 if( p2<0 ) p2 = 0;
drh653bc752002-02-28 03:31:10 +0000322 p1 = 0;
323 }
drh0bce8352002-02-28 00:41:10 +0000324 }else if( p1>0 ){
325 p1--;
drh65595cd2009-02-02 16:32:55 +0000326 }else if( p2>0 ){
327 p2--;
drh0bce8352002-02-28 00:41:10 +0000328 }
drh65595cd2009-02-02 16:32:55 +0000329 if( negP2 ){
330 p1 -= p2;
drh4e79c592009-02-01 19:23:32 +0000331 if( p1<0 ){
332 p2 += p1;
333 p1 = 0;
334 }
335 }
drh65595cd2009-02-02 16:32:55 +0000336 assert( p1>=0 && p2>=0 );
drhf764e6f2007-05-15 01:13:47 +0000337 if( p0type!=SQLITE_BLOB ){
drh4a919112007-05-15 11:55:09 +0000338 while( *z && p1 ){
339 SQLITE_SKIP_UTF8(z);
340 p1--;
drhf764e6f2007-05-15 01:13:47 +0000341 }
drh4a919112007-05-15 11:55:09 +0000342 for(z2=z; *z2 && p2; p2--){
343 SQLITE_SKIP_UTF8(z2);
drhf764e6f2007-05-15 01:13:47 +0000344 }
drhbbf483f2014-09-09 20:30:24 +0000345 sqlite3_result_text64(context, (char*)z, z2-z, SQLITE_TRANSIENT,
346 SQLITE_UTF8);
drhf764e6f2007-05-15 01:13:47 +0000347 }else{
drh4adc4cb2009-11-11 20:53:31 +0000348 if( p1+p2>len ){
349 p2 = len-p1;
350 if( p2<0 ) p2 = 0;
351 }
drhbbf483f2014-09-09 20:30:24 +0000352 sqlite3_result_blob64(context, (char*)&z[p1], (u64)p2, SQLITE_TRANSIENT);
drh0bce8352002-02-28 00:41:10 +0000353 }
drh0bce8352002-02-28 00:41:10 +0000354}
355
356/*
357** Implementation of the round() function
358*/
shanefbd60f82009-02-04 03:59:25 +0000359#ifndef SQLITE_OMIT_FLOATING_POINT
danielk19770ae8b832004-05-25 12:05:56 +0000360static void roundFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
danielk197751ad0ec2004-05-24 12:39:02 +0000361 int n = 0;
drh0bce8352002-02-28 00:41:10 +0000362 double r;
drh50d654d2009-06-03 01:24:54 +0000363 char *zBuf;
drh0bce8352002-02-28 00:41:10 +0000364 assert( argc==1 || argc==2 );
danielk197751ad0ec2004-05-24 12:39:02 +0000365 if( argc==2 ){
drh9c054832004-05-31 18:51:57 +0000366 if( SQLITE_NULL==sqlite3_value_type(argv[1]) ) return;
danielk197751ad0ec2004-05-24 12:39:02 +0000367 n = sqlite3_value_int(argv[1]);
368 if( n>30 ) n = 30;
369 if( n<0 ) n = 0;
370 }
drhd589a922006-03-02 03:02:48 +0000371 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drh4f26d6c2004-05-26 23:25:30 +0000372 r = sqlite3_value_double(argv[0]);
shaneh147e1762010-02-17 04:19:27 +0000373 /* If Y==0 and X will fit in a 64-bit int,
374 ** handle the rounding directly,
375 ** otherwise use printf.
376 */
377 if( n==0 && r>=0 && r<LARGEST_INT64-1 ){
378 r = (double)((sqlite_int64)(r+0.5));
379 }else if( n==0 && r<0 && (-r)<LARGEST_INT64-1 ){
380 r = -(double)((sqlite_int64)((-r)+0.5));
drh50d654d2009-06-03 01:24:54 +0000381 }else{
shaneh147e1762010-02-17 04:19:27 +0000382 zBuf = sqlite3_mprintf("%.*f",n,r);
383 if( zBuf==0 ){
384 sqlite3_result_error_nomem(context);
385 return;
386 }
drh9339da12010-09-30 00:50:49 +0000387 sqlite3AtoF(zBuf, &r, sqlite3Strlen30(zBuf), SQLITE_UTF8);
drh50d654d2009-06-03 01:24:54 +0000388 sqlite3_free(zBuf);
drh50d654d2009-06-03 01:24:54 +0000389 }
shaneh147e1762010-02-17 04:19:27 +0000390 sqlite3_result_double(context, r);
drh0bce8352002-02-28 00:41:10 +0000391}
shanefbd60f82009-02-04 03:59:25 +0000392#endif
drhdc04c582002-02-24 01:55:15 +0000393
danielk197726783a52007-08-29 14:06:22 +0000394/*
drhf3cdcdc2015-04-29 16:50:28 +0000395** Allocate nByte bytes of space using sqlite3Malloc(). If the
danielk197726783a52007-08-29 14:06:22 +0000396** allocation fails, call sqlite3_result_error_nomem() to notify
drh27e62db2009-04-02 10:16:17 +0000397** the database handle that malloc() has failed and return NULL.
398** If nByte is larger than the maximum string or blob length, then
399** raise an SQLITE_TOOBIG exception and return NULL.
danielk197726783a52007-08-29 14:06:22 +0000400*/
drhb1a6c3c2008-03-20 16:30:17 +0000401static void *contextMalloc(sqlite3_context *context, i64 nByte){
drhbb4957f2008-03-20 14:03:29 +0000402 char *z;
drh27e62db2009-04-02 10:16:17 +0000403 sqlite3 *db = sqlite3_context_db_handle(context);
drhef31c6a2009-04-02 09:07:12 +0000404 assert( nByte>0 );
drh27e62db2009-04-02 10:16:17 +0000405 testcase( nByte==db->aLimit[SQLITE_LIMIT_LENGTH] );
406 testcase( nByte==db->aLimit[SQLITE_LIMIT_LENGTH]+1 );
407 if( nByte>db->aLimit[SQLITE_LIMIT_LENGTH] ){
drhbb4957f2008-03-20 14:03:29 +0000408 sqlite3_result_error_toobig(context);
409 z = 0;
410 }else{
drhda4ca9d2014-09-09 17:27:35 +0000411 z = sqlite3Malloc(nByte);
drhef31c6a2009-04-02 09:07:12 +0000412 if( !z ){
drhbb4957f2008-03-20 14:03:29 +0000413 sqlite3_result_error_nomem(context);
414 }
danielk1977a1644fd2007-08-29 12:31:25 +0000415 }
416 return z;
417}
418
drhdc04c582002-02-24 01:55:15 +0000419/*
420** Implementation of the upper() and lower() SQL functions.
421*/
danielk19770ae8b832004-05-25 12:05:56 +0000422static void upperFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drh7a521cf2007-04-25 18:23:52 +0000423 char *z1;
424 const char *z2;
drh9310ef22007-04-27 17:16:20 +0000425 int i, n;
drh1d34fde2009-02-03 15:50:33 +0000426 UNUSED_PARAMETER(argc);
drh7a521cf2007-04-25 18:23:52 +0000427 z2 = (char*)sqlite3_value_text(argv[0]);
drh1f0feef2007-05-15 13:27:07 +0000428 n = sqlite3_value_bytes(argv[0]);
429 /* Verify that the call to _bytes() does not invalidate the _text() pointer */
430 assert( z2==(char*)sqlite3_value_text(argv[0]) );
drh7a521cf2007-04-25 18:23:52 +0000431 if( z2 ){
drhb1a6c3c2008-03-20 16:30:17 +0000432 z1 = contextMalloc(context, ((i64)n)+1);
drh7a521cf2007-04-25 18:23:52 +0000433 if( z1 ){
drhdf901d32011-10-13 18:00:11 +0000434 for(i=0; i<n; i++){
435 z1[i] = (char)sqlite3Toupper(z2[i]);
drh7a521cf2007-04-25 18:23:52 +0000436 }
drhdf901d32011-10-13 18:00:11 +0000437 sqlite3_result_text(context, z1, n, sqlite3_free);
drh7a521cf2007-04-25 18:23:52 +0000438 }
drhdc04c582002-02-24 01:55:15 +0000439 }
440}
danielk19770ae8b832004-05-25 12:05:56 +0000441static void lowerFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drhdf901d32011-10-13 18:00:11 +0000442 char *z1;
drh7a521cf2007-04-25 18:23:52 +0000443 const char *z2;
drh9310ef22007-04-27 17:16:20 +0000444 int i, n;
drh1d34fde2009-02-03 15:50:33 +0000445 UNUSED_PARAMETER(argc);
drh7a521cf2007-04-25 18:23:52 +0000446 z2 = (char*)sqlite3_value_text(argv[0]);
drh1f0feef2007-05-15 13:27:07 +0000447 n = sqlite3_value_bytes(argv[0]);
448 /* Verify that the call to _bytes() does not invalidate the _text() pointer */
449 assert( z2==(char*)sqlite3_value_text(argv[0]) );
drh7a521cf2007-04-25 18:23:52 +0000450 if( z2 ){
drhb1a6c3c2008-03-20 16:30:17 +0000451 z1 = contextMalloc(context, ((i64)n)+1);
drh7a521cf2007-04-25 18:23:52 +0000452 if( z1 ){
drhdf901d32011-10-13 18:00:11 +0000453 for(i=0; i<n; i++){
454 z1[i] = sqlite3Tolower(z2[i]);
drh7a521cf2007-04-25 18:23:52 +0000455 }
drhdf901d32011-10-13 18:00:11 +0000456 sqlite3_result_text(context, z1, n, sqlite3_free);
drh7a521cf2007-04-25 18:23:52 +0000457 }
drhdc04c582002-02-24 01:55:15 +0000458 }
459}
460
drhae6bb952009-11-11 00:24:31 +0000461/*
drhcca9f3d2013-09-06 15:23:29 +0000462** Some functions like COALESCE() and IFNULL() and UNLIKELY() are implemented
463** as VDBE code so that unused argument values do not have to be computed.
464** However, we still need some kind of function implementation for this
465** routines in the function table. The noopFunc macro provides this.
466** noopFunc will never be called so it doesn't matter what the implementation
467** is. We might as well use the "version()" function as a substitute.
drhae6bb952009-11-11 00:24:31 +0000468*/
drhcca9f3d2013-09-06 15:23:29 +0000469#define noopFunc versionFunc /* Substitute function - never called */
drh3212e182002-02-28 00:46:26 +0000470
471/*
drhf9ffac92002-03-02 19:00:31 +0000472** Implementation of random(). Return a random integer.
473*/
drhf9b596e2004-05-26 16:54:42 +0000474static void randomFunc(
475 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000476 int NotUsed,
477 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000478){
drh52fc8492006-02-23 21:43:55 +0000479 sqlite_int64 r;
danielk197762c14b32008-11-19 09:05:26 +0000480 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drh2fa18682008-03-19 14:15:34 +0000481 sqlite3_randomness(sizeof(r), &r);
drh3034e3d2009-04-02 14:05:21 +0000482 if( r<0 ){
483 /* We need to prevent a random number of 0x8000000000000000
484 ** (or -9223372036854775808) since when you do abs() of that
485 ** number of you get the same value back again. To do this
486 ** in a way that is testable, mask the sign bit off of negative
487 ** values, resulting in a positive value. Then take the
488 ** 2s complement of that positive value. The end result can
489 ** therefore be no less than -9223372036854775807.
490 */
drhaf8001b2012-02-11 19:53:24 +0000491 r = -(r & LARGEST_INT64);
drh3034e3d2009-04-02 14:05:21 +0000492 }
drh52fc8492006-02-23 21:43:55 +0000493 sqlite3_result_int64(context, r);
drhf9ffac92002-03-02 19:00:31 +0000494}
495
496/*
drh137c7282007-01-29 17:58:28 +0000497** Implementation of randomblob(N). Return a random blob
498** that is N bytes long.
drh63cf66f2007-01-29 15:50:05 +0000499*/
drh137c7282007-01-29 17:58:28 +0000500static void randomBlob(
drh63cf66f2007-01-29 15:50:05 +0000501 sqlite3_context *context,
502 int argc,
503 sqlite3_value **argv
504){
drh137c7282007-01-29 17:58:28 +0000505 int n;
506 unsigned char *p;
drh63cf66f2007-01-29 15:50:05 +0000507 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000508 UNUSED_PARAMETER(argc);
drh63cf66f2007-01-29 15:50:05 +0000509 n = sqlite3_value_int(argv[0]);
drh023ae032007-05-08 12:12:16 +0000510 if( n<1 ){
511 n = 1;
512 }
danielk1977a1644fd2007-08-29 12:31:25 +0000513 p = contextMalloc(context, n);
drh02d85832007-05-07 19:31:15 +0000514 if( p ){
drh2fa18682008-03-19 14:15:34 +0000515 sqlite3_randomness(n, p);
drh17435752007-08-16 04:30:38 +0000516 sqlite3_result_blob(context, (char*)p, n, sqlite3_free);
drh02d85832007-05-07 19:31:15 +0000517 }
drh63cf66f2007-01-29 15:50:05 +0000518}
519
520/*
drh6ed41ad2002-04-06 14:10:47 +0000521** Implementation of the last_insert_rowid() SQL function. The return
danielk197724b03fd2004-05-10 10:34:34 +0000522** value is the same as the sqlite3_last_insert_rowid() API function.
drh6ed41ad2002-04-06 14:10:47 +0000523*/
danielk197751ad0ec2004-05-24 12:39:02 +0000524static void last_insert_rowid(
danielk19770ae8b832004-05-25 12:05:56 +0000525 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000526 int NotUsed,
527 sqlite3_value **NotUsed2
danielk197751ad0ec2004-05-24 12:39:02 +0000528){
drhfa4a4b92008-03-19 21:45:51 +0000529 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000530 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000531 /* IMP: R-51513-12026 The last_insert_rowid() SQL function is a
532 ** wrapper around the sqlite3_last_insert_rowid() C/C++ interface
533 ** function. */
drhf9b596e2004-05-26 16:54:42 +0000534 sqlite3_result_int64(context, sqlite3_last_insert_rowid(db));
drh6ed41ad2002-04-06 14:10:47 +0000535}
536
rdcf146a772004-02-25 22:51:06 +0000537/*
drhab2f1f92010-01-11 18:26:42 +0000538** Implementation of the changes() SQL function.
539**
540** IMP: R-62073-11209 The changes() SQL function is a wrapper
541** around the sqlite3_changes() C/C++ function and hence follows the same
542** rules for counting changes.
rdcf146a772004-02-25 22:51:06 +0000543*/
danielk1977b28af712004-06-21 06:50:26 +0000544static void changes(
drhf9b596e2004-05-26 16:54:42 +0000545 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000546 int NotUsed,
547 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000548){
drhfa4a4b92008-03-19 21:45:51 +0000549 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000550 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhf4479502004-05-27 03:12:53 +0000551 sqlite3_result_int(context, sqlite3_changes(db));
rdcb0c374f2004-02-20 22:53:38 +0000552}
rdcf146a772004-02-25 22:51:06 +0000553
554/*
danielk1977b28af712004-06-21 06:50:26 +0000555** Implementation of the total_changes() SQL function. The return value is
556** the same as the sqlite3_total_changes() API function.
rdcf146a772004-02-25 22:51:06 +0000557*/
danielk1977b28af712004-06-21 06:50:26 +0000558static void total_changes(
559 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-52756-41993 This function is a wrapper around the
566 ** sqlite3_total_changes() C/C++ interface. */
danielk1977b28af712004-06-21 06:50:26 +0000567 sqlite3_result_int(context, sqlite3_total_changes(db));
rdcb0c374f2004-02-20 22:53:38 +0000568}
569
drh6ed41ad2002-04-06 14:10:47 +0000570/*
drh4e5ffc52004-08-31 00:52:37 +0000571** A structure defining how to do GLOB-style comparisons.
danielk1977d02eb1f2004-06-06 09:44:03 +0000572*/
drh4e5ffc52004-08-31 00:52:37 +0000573struct compareInfo {
drh07e83472016-01-11 03:48:18 +0000574 u8 matchAll; /* "*" or "%" */
575 u8 matchOne; /* "?" or "_" */
576 u8 matchSet; /* "[" or 0 */
577 u8 noCase; /* true to ignore case differences */
danielk1977d02eb1f2004-06-06 09:44:03 +0000578};
drh55ef4d92005-08-14 01:20:37 +0000579
drhb9175ae2007-12-07 18:39:04 +0000580/*
581** For LIKE and GLOB matching on EBCDIC machines, assume that every
drhb0870482015-06-17 13:20:54 +0000582** character is exactly one byte in size. Also, provde the Utf8Read()
583** macro for fast reading of the next character in the common case where
584** the next character is ASCII.
drhb9175ae2007-12-07 18:39:04 +0000585*/
586#if defined(SQLITE_EBCDIC)
drh88b33222014-09-25 03:51:37 +0000587# define sqlite3Utf8Read(A) (*((*A)++))
drhb0870482015-06-17 13:20:54 +0000588# define Utf8Read(A) (*(A++))
drhb9175ae2007-12-07 18:39:04 +0000589#else
drhb0870482015-06-17 13:20:54 +0000590# define Utf8Read(A) (A[0]<0x80?*(A++):sqlite3Utf8Read(&A))
drhb9175ae2007-12-07 18:39:04 +0000591#endif
592
drh4e5ffc52004-08-31 00:52:37 +0000593static const struct compareInfo globInfo = { '*', '?', '[', 0 };
drh55ef4d92005-08-14 01:20:37 +0000594/* The correct SQL-92 behavior is for the LIKE operator to ignore
595** case. Thus 'a' LIKE 'A' would be true. */
596static const struct compareInfo likeInfoNorm = { '%', '_', 0, 1 };
597/* If SQLITE_CASE_SENSITIVE_LIKE is defined, then the LIKE operator
598** is case sensitive causing 'a' LIKE 'A' to be false */
599static const struct compareInfo likeInfoAlt = { '%', '_', 0, 0 };
danielk1977d02eb1f2004-06-06 09:44:03 +0000600
601/*
drh698a01c2016-12-01 18:49:40 +0000602** Possible error returns from patternMatch()
603*/
604#define SQLITE_MATCH 0
605#define SQLITE_NOMATCH 1
606#define SQLITE_NOWILDCARDMATCH 2
607
608/*
609** Compare two UTF-8 strings for equality where the first string is
610** a GLOB or LIKE expression. Return values:
611**
612** SQLITE_MATCH: Match
613** SQLITE_NOMATCH: No match
614** SQLITE_NOWILDCARDMATCH: No match in spite of having * or % wildcards.
drh0ac65892002-04-20 14:24:41 +0000615**
drh4e5ffc52004-08-31 00:52:37 +0000616** Globbing rules:
drh0ac65892002-04-20 14:24:41 +0000617**
drh4e5ffc52004-08-31 00:52:37 +0000618** '*' Matches any sequence of zero or more characters.
danielk1977d02eb1f2004-06-06 09:44:03 +0000619**
drh4e5ffc52004-08-31 00:52:37 +0000620** '?' Matches exactly one character.
621**
622** [...] Matches one character from the enclosed list of
623** characters.
624**
625** [^...] Matches one character not in the enclosed list.
626**
627** With the [...] and [^...] matching, a ']' character can be included
628** in the list by making it the first character after '[' or '^'. A
629** range of characters can be specified using '-'. Example:
630** "[a-z]" matches any single lower-case letter. To match a '-', make
631** it the last character in the list.
632**
drh9fdfdc82014-09-25 11:08:57 +0000633** Like matching rules:
634**
635** '%' Matches any sequence of zero or more characters
636**
637*** '_' Matches any one character
638**
639** Ec Where E is the "esc" character and c is any other
640** character, including '%', '_', and esc, match exactly c.
641**
drhb0870482015-06-17 13:20:54 +0000642** The comments within this routine usually assume glob matching.
drh9fdfdc82014-09-25 11:08:57 +0000643**
drh4e5ffc52004-08-31 00:52:37 +0000644** This routine is usually quick, but can be N**2 in the worst case.
drh0ac65892002-04-20 14:24:41 +0000645*/
danielk19777c6303c2004-11-17 16:41:29 +0000646static int patternCompare(
drh4e5ffc52004-08-31 00:52:37 +0000647 const u8 *zPattern, /* The glob pattern */
648 const u8 *zString, /* The string to compare against the glob */
danielk19777c6303c2004-11-17 16:41:29 +0000649 const struct compareInfo *pInfo, /* Information about how to do the compare */
drh698a01c2016-12-01 18:49:40 +0000650 u32 matchOther /* The escape char (LIKE) or '[' (GLOB) */
danielk197751ad0ec2004-05-24 12:39:02 +0000651){
drh9fdfdc82014-09-25 11:08:57 +0000652 u32 c, c2; /* Next pattern and input string chars */
653 u32 matchOne = pInfo->matchOne; /* "?" or "_" */
654 u32 matchAll = pInfo->matchAll; /* "*" or "%" */
drh9fdfdc82014-09-25 11:08:57 +0000655 u8 noCase = pInfo->noCase; /* True if uppercase==lowercase */
656 const u8 *zEscaped = 0; /* One past the last escaped input char */
drh88b33222014-09-25 03:51:37 +0000657
drhb0870482015-06-17 13:20:54 +0000658 while( (c = Utf8Read(zPattern))!=0 ){
drh9fdfdc82014-09-25 11:08:57 +0000659 if( c==matchAll ){ /* Match "*" */
660 /* Skip over multiple "*" characters in the pattern. If there
661 ** are also "?" characters, skip those as well, but consume a
662 ** single character of the input string for each "?" skipped */
drhb0870482015-06-17 13:20:54 +0000663 while( (c=Utf8Read(zPattern)) == matchAll || c == matchOne ){
drh42610962012-09-17 18:56:32 +0000664 if( c==matchOne && sqlite3Utf8Read(&zString)==0 ){
drh698a01c2016-12-01 18:49:40 +0000665 return SQLITE_NOWILDCARDMATCH;
drh4e5ffc52004-08-31 00:52:37 +0000666 }
danielk1977ad7dd422004-06-06 12:41:49 +0000667 }
drh66150952007-07-23 19:12:41 +0000668 if( c==0 ){
drh698a01c2016-12-01 18:49:40 +0000669 return SQLITE_MATCH; /* "*" at the end of the pattern matches */
drh88b33222014-09-25 03:51:37 +0000670 }else if( c==matchOther ){
drh07e83472016-01-11 03:48:18 +0000671 if( pInfo->matchSet==0 ){
drh88b33222014-09-25 03:51:37 +0000672 c = sqlite3Utf8Read(&zPattern);
drh698a01c2016-12-01 18:49:40 +0000673 if( c==0 ) return SQLITE_NOWILDCARDMATCH;
drh88b33222014-09-25 03:51:37 +0000674 }else{
drh9fdfdc82014-09-25 11:08:57 +0000675 /* "[...]" immediately follows the "*". We have to do a slow
676 ** recursive search in this case, but it is an unusual case. */
drh88b33222014-09-25 03:51:37 +0000677 assert( matchOther<0x80 ); /* '[' is a single-byte character */
drh698a01c2016-12-01 18:49:40 +0000678 while( *zString ){
679 int bMatch = patternCompare(&zPattern[-1],zString,pInfo,matchOther);
680 if( bMatch!=SQLITE_NOMATCH ) return bMatch;
drh88b33222014-09-25 03:51:37 +0000681 SQLITE_SKIP_UTF8(zString);
682 }
drh698a01c2016-12-01 18:49:40 +0000683 return SQLITE_NOWILDCARDMATCH;
drh66150952007-07-23 19:12:41 +0000684 }
drh66150952007-07-23 19:12:41 +0000685 }
drh9fdfdc82014-09-25 11:08:57 +0000686
687 /* At this point variable c contains the first character of the
688 ** pattern string past the "*". Search in the input string for the
dan1a4a7372016-12-01 17:34:59 +0000689 ** first matching character and recursively continue the match from
drh9fdfdc82014-09-25 11:08:57 +0000690 ** that point.
691 **
692 ** For a case-insensitive search, set variable cx to be the same as
693 ** c but in the other case and search the input string for either
694 ** c or cx.
695 */
696 if( c<=0x80 ){
697 u32 cx;
drh698a01c2016-12-01 18:49:40 +0000698 int bMatch;
drh9fdfdc82014-09-25 11:08:57 +0000699 if( noCase ){
700 cx = sqlite3Toupper(c);
701 c = sqlite3Tolower(c);
drh66150952007-07-23 19:12:41 +0000702 }else{
drh9fdfdc82014-09-25 11:08:57 +0000703 cx = c;
drh4e5ffc52004-08-31 00:52:37 +0000704 }
drh9fdfdc82014-09-25 11:08:57 +0000705 while( (c2 = *(zString++))!=0 ){
706 if( c2!=c && c2!=cx ) continue;
drh698a01c2016-12-01 18:49:40 +0000707 bMatch = patternCompare(zPattern,zString,pInfo,matchOther);
708 if( bMatch!=SQLITE_NOMATCH ) return bMatch;
drh9fdfdc82014-09-25 11:08:57 +0000709 }
710 }else{
drh698a01c2016-12-01 18:49:40 +0000711 int bMatch;
drhb0870482015-06-17 13:20:54 +0000712 while( (c2 = Utf8Read(zString))!=0 ){
drh9fdfdc82014-09-25 11:08:57 +0000713 if( c2!=c ) continue;
drh698a01c2016-12-01 18:49:40 +0000714 bMatch = patternCompare(zPattern,zString,pInfo,matchOther);
715 if( bMatch!=SQLITE_NOMATCH ) return bMatch;
drh9fdfdc82014-09-25 11:08:57 +0000716 }
drh66150952007-07-23 19:12:41 +0000717 }
drh698a01c2016-12-01 18:49:40 +0000718 return SQLITE_NOWILDCARDMATCH;
drh88b33222014-09-25 03:51:37 +0000719 }
drh88b33222014-09-25 03:51:37 +0000720 if( c==matchOther ){
drh07e83472016-01-11 03:48:18 +0000721 if( pInfo->matchSet==0 ){
drh88b33222014-09-25 03:51:37 +0000722 c = sqlite3Utf8Read(&zPattern);
drh698a01c2016-12-01 18:49:40 +0000723 if( c==0 ) return SQLITE_NOMATCH;
drh9fdfdc82014-09-25 11:08:57 +0000724 zEscaped = zPattern;
drh88b33222014-09-25 03:51:37 +0000725 }else{
726 u32 prior_c = 0;
drh9fdfdc82014-09-25 11:08:57 +0000727 int seen = 0;
728 int invert = 0;
drh88b33222014-09-25 03:51:37 +0000729 c = sqlite3Utf8Read(&zString);
drh698a01c2016-12-01 18:49:40 +0000730 if( c==0 ) return SQLITE_NOMATCH;
drh88b33222014-09-25 03:51:37 +0000731 c2 = sqlite3Utf8Read(&zPattern);
732 if( c2=='^' ){
733 invert = 1;
734 c2 = sqlite3Utf8Read(&zPattern);
735 }
736 if( c2==']' ){
737 if( c==']' ) seen = 1;
738 c2 = sqlite3Utf8Read(&zPattern);
739 }
740 while( c2 && c2!=']' ){
741 if( c2=='-' && zPattern[0]!=']' && zPattern[0]!=0 && prior_c>0 ){
742 c2 = sqlite3Utf8Read(&zPattern);
743 if( c>=prior_c && c<=c2 ) seen = 1;
744 prior_c = 0;
745 }else{
746 if( c==c2 ){
747 seen = 1;
748 }
749 prior_c = c2;
750 }
751 c2 = sqlite3Utf8Read(&zPattern);
752 }
753 if( c2==0 || (seen ^ invert)==0 ){
drh698a01c2016-12-01 18:49:40 +0000754 return SQLITE_NOMATCH;
drh88b33222014-09-25 03:51:37 +0000755 }
756 continue;
757 }
758 }
drhb0870482015-06-17 13:20:54 +0000759 c2 = Utf8Read(zString);
drh88b33222014-09-25 03:51:37 +0000760 if( c==c2 ) continue;
drhc80937a2016-06-06 01:48:14 +0000761 if( noCase && sqlite3Tolower(c)==sqlite3Tolower(c2) && c<0x80 && c2<0x80 ){
drh9fdfdc82014-09-25 11:08:57 +0000762 continue;
763 }
764 if( c==matchOne && zPattern!=zEscaped && c2!=0 ) continue;
drh698a01c2016-12-01 18:49:40 +0000765 return SQLITE_NOMATCH;
danielk197751ad0ec2004-05-24 12:39:02 +0000766 }
drh698a01c2016-12-01 18:49:40 +0000767 return *zString==0 ? SQLITE_MATCH : SQLITE_NOMATCH;
drh0ac65892002-04-20 14:24:41 +0000768}
drh4e5ffc52004-08-31 00:52:37 +0000769
drh55ef4d92005-08-14 01:20:37 +0000770/*
drh698a01c2016-12-01 18:49:40 +0000771** The sqlite3_strglob() interface. Return 0 on a match (like strcmp()) and
772** non-zero if there is no match.
drh56282a52013-04-10 16:13:38 +0000773*/
774int sqlite3_strglob(const char *zGlobPattern, const char *zString){
drh698a01c2016-12-01 18:49:40 +0000775 return patternCompare((u8*)zGlobPattern, (u8*)zString, &globInfo, '[');
drh56282a52013-04-10 16:13:38 +0000776}
777
778/*
drh698a01c2016-12-01 18:49:40 +0000779** The sqlite3_strlike() interface. Return 0 on a match and non-zero for
780** a miss - like strcmp().
drh8b4a94a2015-11-24 21:23:59 +0000781*/
782int sqlite3_strlike(const char *zPattern, const char *zStr, unsigned int esc){
drh698a01c2016-12-01 18:49:40 +0000783 return patternCompare((u8*)zPattern, (u8*)zStr, &likeInfoNorm, esc);
drh8b4a94a2015-11-24 21:23:59 +0000784}
785
786/*
drh55ef4d92005-08-14 01:20:37 +0000787** Count the number of times that the LIKE operator (or GLOB which is
788** just a variation of LIKE) gets called. This is used for testing
789** only.
790*/
791#ifdef SQLITE_TEST
792int sqlite3_like_count = 0;
793#endif
794
danielk19773f6b0872004-06-17 05:36:44 +0000795
796/*
797** Implementation of the like() SQL function. This function implements
798** the build-in LIKE operator. The first argument to the function is the
799** pattern and the second argument is the string. So, the SQL statements:
800**
801** A LIKE B
802**
803** is implemented as like(B,A).
804**
drh55ef4d92005-08-14 01:20:37 +0000805** This same function (with a different compareInfo structure) computes
806** the GLOB operator.
danielk19773f6b0872004-06-17 05:36:44 +0000807*/
808static void likeFunc(
809 sqlite3_context *context,
810 int argc,
811 sqlite3_value **argv
812){
drhbeb818d2007-05-08 15:34:47 +0000813 const unsigned char *zA, *zB;
drh07e83472016-01-11 03:48:18 +0000814 u32 escape;
drh27e62db2009-04-02 10:16:17 +0000815 int nPat;
drhbb4957f2008-03-20 14:03:29 +0000816 sqlite3 *db = sqlite3_context_db_handle(context);
drh07e83472016-01-11 03:48:18 +0000817 struct compareInfo *pInfo = sqlite3_user_data(context);
drhbeb818d2007-05-08 15:34:47 +0000818
drh41d2e662015-12-01 21:23:07 +0000819#ifdef SQLITE_LIKE_DOESNT_MATCH_BLOBS
820 if( sqlite3_value_type(argv[0])==SQLITE_BLOB
821 || sqlite3_value_type(argv[1])==SQLITE_BLOB
822 ){
823#ifdef SQLITE_TEST
824 sqlite3_like_count++;
825#endif
826 sqlite3_result_int(context, 0);
827 return;
828 }
829#endif
drh1f0feef2007-05-15 13:27:07 +0000830 zB = sqlite3_value_text(argv[0]);
831 zA = sqlite3_value_text(argv[1]);
832
drhbeb818d2007-05-08 15:34:47 +0000833 /* Limit the length of the LIKE or GLOB pattern to avoid problems
834 ** of deep recursion and N*N behavior in patternCompare().
835 */
drh27e62db2009-04-02 10:16:17 +0000836 nPat = sqlite3_value_bytes(argv[0]);
837 testcase( nPat==db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH] );
838 testcase( nPat==db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH]+1 );
839 if( nPat > db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH] ){
drhbeb818d2007-05-08 15:34:47 +0000840 sqlite3_result_error(context, "LIKE or GLOB pattern too complex", -1);
841 return;
842 }
drh1f0feef2007-05-15 13:27:07 +0000843 assert( zB==sqlite3_value_text(argv[0]) ); /* Encoding did not change */
drhbeb818d2007-05-08 15:34:47 +0000844
danielk19777c6303c2004-11-17 16:41:29 +0000845 if( argc==3 ){
846 /* The escape character string must consist of a single UTF-8 character.
847 ** Otherwise, return an error.
848 */
849 const unsigned char *zEsc = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +0000850 if( zEsc==0 ) return;
drhee858132007-05-08 20:37:38 +0000851 if( sqlite3Utf8CharLen((char*)zEsc, -1)!=1 ){
danielk19777c6303c2004-11-17 16:41:29 +0000852 sqlite3_result_error(context,
853 "ESCAPE expression must be a single character", -1);
854 return;
855 }
drh42610962012-09-17 18:56:32 +0000856 escape = sqlite3Utf8Read(&zEsc);
drh07e83472016-01-11 03:48:18 +0000857 }else{
858 escape = pInfo->matchSet;
danielk19777c6303c2004-11-17 16:41:29 +0000859 }
danielk19773f6b0872004-06-17 05:36:44 +0000860 if( zA && zB ){
drh55ef4d92005-08-14 01:20:37 +0000861#ifdef SQLITE_TEST
862 sqlite3_like_count++;
863#endif
drh698a01c2016-12-01 18:49:40 +0000864 sqlite3_result_int(context, patternCompare(zB, zA, pInfo, escape)==SQLITE_MATCH);
danielk197751ad0ec2004-05-24 12:39:02 +0000865 }
drh8912d102002-05-26 21:34:58 +0000866}
867
868/*
869** Implementation of the NULLIF(x,y) function. The result is the first
870** argument if the arguments are different. The result is NULL if the
871** arguments are equal to each other.
872*/
drhf9b596e2004-05-26 16:54:42 +0000873static void nullifFunc(
874 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000875 int NotUsed,
drhf9b596e2004-05-26 16:54:42 +0000876 sqlite3_value **argv
877){
danielk1977dc1bdc42004-06-11 10:51:27 +0000878 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk197762c14b32008-11-19 09:05:26 +0000879 UNUSED_PARAMETER(NotUsed);
danielk1977dc1bdc42004-06-11 10:51:27 +0000880 if( sqlite3MemCompare(argv[0], argv[1], pColl)!=0 ){
drhf4479502004-05-27 03:12:53 +0000881 sqlite3_result_value(context, argv[0]);
drh8912d102002-05-26 21:34:58 +0000882 }
drh0ac65892002-04-20 14:24:41 +0000883}
884
drh647cb0e2002-11-04 19:32:25 +0000885/*
drh47baebc2009-08-14 16:01:24 +0000886** Implementation of the sqlite_version() function. The result is the version
drh647cb0e2002-11-04 19:32:25 +0000887** of the SQLite library that is running.
888*/
drhf9b596e2004-05-26 16:54:42 +0000889static void versionFunc(
890 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000891 int NotUsed,
892 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000893){
danielk197762c14b32008-11-19 09:05:26 +0000894 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000895 /* IMP: R-48699-48617 This function is an SQL wrapper around the
896 ** sqlite3_libversion() C-interface. */
897 sqlite3_result_text(context, sqlite3_libversion(), -1, SQLITE_STATIC);
drh647cb0e2002-11-04 19:32:25 +0000898}
899
drh47baebc2009-08-14 16:01:24 +0000900/*
901** Implementation of the sqlite_source_id() function. The result is a string
902** that identifies the particular version of the source code used to build
903** SQLite.
904*/
905static void sourceidFunc(
906 sqlite3_context *context,
907 int NotUsed,
908 sqlite3_value **NotUsed2
909){
910 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000911 /* IMP: R-24470-31136 This function is an SQL wrapper around the
912 ** sqlite3_sourceid() C interface. */
913 sqlite3_result_text(context, sqlite3_sourceid(), -1, SQLITE_STATIC);
drh47baebc2009-08-14 16:01:24 +0000914}
915
shanehbdea6d12010-02-23 04:19:54 +0000916/*
drh3ca84ef2011-04-25 18:03:10 +0000917** Implementation of the sqlite_log() function. This is a wrapper around
918** sqlite3_log(). The return value is NULL. The function exists purely for
919** its side-effects.
920*/
drh840561f2011-04-27 18:08:42 +0000921static void errlogFunc(
drh3ca84ef2011-04-25 18:03:10 +0000922 sqlite3_context *context,
923 int argc,
924 sqlite3_value **argv
925){
926 UNUSED_PARAMETER(argc);
927 UNUSED_PARAMETER(context);
928 sqlite3_log(sqlite3_value_int(argv[0]), "%s", sqlite3_value_text(argv[1]));
929}
930
931/*
shanehdc97a8c2010-02-23 20:08:35 +0000932** Implementation of the sqlite_compileoption_used() function.
933** The result is an integer that identifies if the compiler option
934** was used to build SQLite.
shanehbdea6d12010-02-23 04:19:54 +0000935*/
shanehdc97a8c2010-02-23 20:08:35 +0000936#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
937static void compileoptionusedFunc(
shanehbdea6d12010-02-23 04:19:54 +0000938 sqlite3_context *context,
shanehdc97a8c2010-02-23 20:08:35 +0000939 int argc,
940 sqlite3_value **argv
shanehbdea6d12010-02-23 04:19:54 +0000941){
shanehdc97a8c2010-02-23 20:08:35 +0000942 const char *zOptName;
943 assert( argc==1 );
944 UNUSED_PARAMETER(argc);
drha3e414c2010-08-03 18:06:25 +0000945 /* IMP: R-39564-36305 The sqlite_compileoption_used() SQL
946 ** function is a wrapper around the sqlite3_compileoption_used() C/C++
947 ** function.
948 */
drh264a2d42010-02-25 15:28:41 +0000949 if( (zOptName = (const char*)sqlite3_value_text(argv[0]))!=0 ){
shanehdc97a8c2010-02-23 20:08:35 +0000950 sqlite3_result_int(context, sqlite3_compileoption_used(zOptName));
shanehdc97a8c2010-02-23 20:08:35 +0000951 }
shanehbdea6d12010-02-23 04:19:54 +0000952}
shanehdc97a8c2010-02-23 20:08:35 +0000953#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
954
955/*
956** Implementation of the sqlite_compileoption_get() function.
957** The result is a string that identifies the compiler options
958** used to build SQLite.
959*/
960#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
961static void compileoptiongetFunc(
962 sqlite3_context *context,
963 int argc,
964 sqlite3_value **argv
965){
966 int n;
967 assert( argc==1 );
968 UNUSED_PARAMETER(argc);
drha3e414c2010-08-03 18:06:25 +0000969 /* IMP: R-04922-24076 The sqlite_compileoption_get() SQL function
970 ** is a wrapper around the sqlite3_compileoption_get() C/C++ function.
971 */
shanehdc97a8c2010-02-23 20:08:35 +0000972 n = sqlite3_value_int(argv[0]);
973 sqlite3_result_text(context, sqlite3_compileoption_get(n), -1, SQLITE_STATIC);
974}
975#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
shanehbdea6d12010-02-23 04:19:54 +0000976
drh137c7282007-01-29 17:58:28 +0000977/* Array for converting from half-bytes (nybbles) into ASCII hex
978** digits. */
979static const char hexdigits[] = {
980 '0', '1', '2', '3', '4', '5', '6', '7',
981 '8', '9', 'A', 'B', 'C', 'D', 'E', 'F'
982};
danielk1977d641d642004-11-18 15:44:29 +0000983
drh47394702003-08-20 01:03:33 +0000984/*
drh47394702003-08-20 01:03:33 +0000985** Implementation of the QUOTE() function. This function takes a single
986** argument. If the argument is numeric, the return value is the same as
987** the argument. If the argument is NULL, the return value is the string
988** "NULL". Otherwise, the argument is enclosed in single quotes with
989** single-quote escapes.
990*/
danielk19770ae8b832004-05-25 12:05:56 +0000991static void quoteFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drha0df4cc2009-02-02 17:29:59 +0000992 assert( argc==1 );
drh1d34fde2009-02-03 15:50:33 +0000993 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +0000994 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +0000995 case SQLITE_FLOAT: {
drh72b3fbc2012-06-19 03:11:25 +0000996 double r1, r2;
drh72b3fbc2012-06-19 03:11:25 +0000997 char zBuf[50];
mistachkin2b434a72012-06-19 04:36:48 +0000998 r1 = sqlite3_value_double(argv[0]);
drh72b3fbc2012-06-19 03:11:25 +0000999 sqlite3_snprintf(sizeof(zBuf), zBuf, "%!.15g", r1);
1000 sqlite3AtoF(zBuf, &r2, 20, SQLITE_UTF8);
1001 if( r1!=r2 ){
1002 sqlite3_snprintf(sizeof(zBuf), zBuf, "%!.20e", r1);
1003 }
1004 sqlite3_result_text(context, zBuf, -1, SQLITE_TRANSIENT);
1005 break;
1006 }
1007 case SQLITE_INTEGER: {
drhf4479502004-05-27 03:12:53 +00001008 sqlite3_result_value(context, argv[0]);
drhf9b596e2004-05-26 16:54:42 +00001009 break;
1010 }
danielk19773f41e972004-06-08 00:39:01 +00001011 case SQLITE_BLOB: {
danielk19773f41e972004-06-08 00:39:01 +00001012 char *zText = 0;
danielk19773f41e972004-06-08 00:39:01 +00001013 char const *zBlob = sqlite3_value_blob(argv[0]);
drh1f0feef2007-05-15 13:27:07 +00001014 int nBlob = sqlite3_value_bytes(argv[0]);
1015 assert( zBlob==sqlite3_value_blob(argv[0]) ); /* No encoding change */
drhb1a6c3c2008-03-20 16:30:17 +00001016 zText = (char *)contextMalloc(context, (2*(i64)nBlob)+4);
danielk1977a1644fd2007-08-29 12:31:25 +00001017 if( zText ){
danielk19773f41e972004-06-08 00:39:01 +00001018 int i;
1019 for(i=0; i<nBlob; i++){
1020 zText[(i*2)+2] = hexdigits[(zBlob[i]>>4)&0x0F];
1021 zText[(i*2)+3] = hexdigits[(zBlob[i])&0x0F];
1022 }
1023 zText[(nBlob*2)+2] = '\'';
1024 zText[(nBlob*2)+3] = '\0';
1025 zText[0] = 'X';
1026 zText[1] = '\'';
danielk1977d8123362004-06-12 09:25:12 +00001027 sqlite3_result_text(context, zText, -1, SQLITE_TRANSIENT);
drh17435752007-08-16 04:30:38 +00001028 sqlite3_free(zText);
danielk19773f41e972004-06-08 00:39:01 +00001029 }
1030 break;
1031 }
drh9c054832004-05-31 18:51:57 +00001032 case SQLITE_TEXT: {
drh023ae032007-05-08 12:12:16 +00001033 int i,j;
1034 u64 n;
drh2646da72005-12-09 20:02:05 +00001035 const unsigned char *zArg = sqlite3_value_text(argv[0]);
drhf9b596e2004-05-26 16:54:42 +00001036 char *z;
1037
drh7a521cf2007-04-25 18:23:52 +00001038 if( zArg==0 ) return;
drh023ae032007-05-08 12:12:16 +00001039 for(i=0, n=0; zArg[i]; i++){ if( zArg[i]=='\'' ) n++; }
drhb1a6c3c2008-03-20 16:30:17 +00001040 z = contextMalloc(context, ((i64)i)+((i64)n)+3);
danielk1977a1644fd2007-08-29 12:31:25 +00001041 if( z ){
1042 z[0] = '\'';
1043 for(i=0, j=1; zArg[i]; i++){
1044 z[j++] = zArg[i];
1045 if( zArg[i]=='\'' ){
1046 z[j++] = '\'';
1047 }
drhf9b596e2004-05-26 16:54:42 +00001048 }
danielk1977a1644fd2007-08-29 12:31:25 +00001049 z[j++] = '\'';
1050 z[j] = 0;
1051 sqlite3_result_text(context, z, j, sqlite3_free);
drhf9b596e2004-05-26 16:54:42 +00001052 }
drha0df4cc2009-02-02 17:29:59 +00001053 break;
1054 }
1055 default: {
1056 assert( sqlite3_value_type(argv[0])==SQLITE_NULL );
1057 sqlite3_result_text(context, "NULL", 4, SQLITE_STATIC);
1058 break;
drhf9b596e2004-05-26 16:54:42 +00001059 }
drh47394702003-08-20 01:03:33 +00001060 }
1061}
1062
drh137c7282007-01-29 17:58:28 +00001063/*
drhd495d8c2013-02-22 19:34:25 +00001064** The unicode() function. Return the integer unicode code-point value
1065** for the first character of the input string.
1066*/
1067static void unicodeFunc(
1068 sqlite3_context *context,
1069 int argc,
1070 sqlite3_value **argv
1071){
1072 const unsigned char *z = sqlite3_value_text(argv[0]);
drh1d59d032013-03-01 23:40:26 +00001073 (void)argc;
drhd495d8c2013-02-22 19:34:25 +00001074 if( z && z[0] ) sqlite3_result_int(context, sqlite3Utf8Read(&z));
1075}
1076
1077/*
1078** The char() function takes zero or more arguments, each of which is
1079** an integer. It constructs a string where each character of the string
1080** is the unicode character for the corresponding integer argument.
1081*/
1082static void charFunc(
1083 sqlite3_context *context,
1084 int argc,
1085 sqlite3_value **argv
1086){
1087 unsigned char *z, *zOut;
1088 int i;
drhf3cdcdc2015-04-29 16:50:28 +00001089 zOut = z = sqlite3_malloc64( argc*4+1 );
drhd495d8c2013-02-22 19:34:25 +00001090 if( z==0 ){
1091 sqlite3_result_error_nomem(context);
1092 return;
1093 }
1094 for(i=0; i<argc; i++){
mistachkinc9545442013-02-26 05:42:30 +00001095 sqlite3_int64 x;
drhd495d8c2013-02-22 19:34:25 +00001096 unsigned c;
1097 x = sqlite3_value_int64(argv[i]);
1098 if( x<0 || x>0x10ffff ) x = 0xfffd;
1099 c = (unsigned)(x & 0x1fffff);
drhfe7a5d12013-03-07 14:00:04 +00001100 if( c<0x00080 ){
1101 *zOut++ = (u8)(c&0xFF);
1102 }else if( c<0x00800 ){
1103 *zOut++ = 0xC0 + (u8)((c>>6)&0x1F);
1104 *zOut++ = 0x80 + (u8)(c & 0x3F);
1105 }else if( c<0x10000 ){
1106 *zOut++ = 0xE0 + (u8)((c>>12)&0x0F);
1107 *zOut++ = 0x80 + (u8)((c>>6) & 0x3F);
1108 *zOut++ = 0x80 + (u8)(c & 0x3F);
drhd495d8c2013-02-22 19:34:25 +00001109 }else{
drhfe7a5d12013-03-07 14:00:04 +00001110 *zOut++ = 0xF0 + (u8)((c>>18) & 0x07);
1111 *zOut++ = 0x80 + (u8)((c>>12) & 0x3F);
1112 *zOut++ = 0x80 + (u8)((c>>6) & 0x3F);
1113 *zOut++ = 0x80 + (u8)(c & 0x3F);
1114 } \
drhd495d8c2013-02-22 19:34:25 +00001115 }
drhbbf483f2014-09-09 20:30:24 +00001116 sqlite3_result_text64(context, (char*)z, zOut-z, sqlite3_free, SQLITE_UTF8);
drhd495d8c2013-02-22 19:34:25 +00001117}
1118
1119/*
drh137c7282007-01-29 17:58:28 +00001120** The hex() function. Interpret the argument as a blob. Return
1121** a hexadecimal rendering as text.
1122*/
1123static void hexFunc(
1124 sqlite3_context *context,
1125 int argc,
1126 sqlite3_value **argv
1127){
1128 int i, n;
1129 const unsigned char *pBlob;
1130 char *zHex, *z;
1131 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001132 UNUSED_PARAMETER(argc);
drh1f0feef2007-05-15 13:27:07 +00001133 pBlob = sqlite3_value_blob(argv[0]);
drh137c7282007-01-29 17:58:28 +00001134 n = sqlite3_value_bytes(argv[0]);
drh1f0feef2007-05-15 13:27:07 +00001135 assert( pBlob==sqlite3_value_blob(argv[0]) ); /* No encoding change */
drhb1a6c3c2008-03-20 16:30:17 +00001136 z = zHex = contextMalloc(context, ((i64)n)*2 + 1);
danielk1977a1644fd2007-08-29 12:31:25 +00001137 if( zHex ){
1138 for(i=0; i<n; i++, pBlob++){
1139 unsigned char c = *pBlob;
1140 *(z++) = hexdigits[(c>>4)&0xf];
1141 *(z++) = hexdigits[c&0xf];
1142 }
1143 *z = 0;
1144 sqlite3_result_text(context, zHex, n*2, sqlite3_free);
drh137c7282007-01-29 17:58:28 +00001145 }
drh137c7282007-01-29 17:58:28 +00001146}
1147
drh26b6d902007-03-17 13:27:54 +00001148/*
drh8cff3822007-05-02 02:08:28 +00001149** The zeroblob(N) function returns a zero-filled blob of size N bytes.
1150*/
1151static void zeroblobFunc(
1152 sqlite3_context *context,
1153 int argc,
1154 sqlite3_value **argv
1155){
drh98640a32007-06-07 19:08:32 +00001156 i64 n;
dana4d5ae82015-07-24 16:24:37 +00001157 int rc;
drh8cff3822007-05-02 02:08:28 +00001158 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001159 UNUSED_PARAMETER(argc);
drh98640a32007-06-07 19:08:32 +00001160 n = sqlite3_value_int64(argv[0]);
dana4d5ae82015-07-24 16:24:37 +00001161 if( n<0 ) n = 0;
1162 rc = sqlite3_result_zeroblob64(context, n); /* IMP: R-00293-64994 */
1163 if( rc ){
1164 sqlite3_result_error_code(context, rc);
drh98640a32007-06-07 19:08:32 +00001165 }
drh8cff3822007-05-02 02:08:28 +00001166}
1167
1168/*
drh26b6d902007-03-17 13:27:54 +00001169** The replace() function. Three arguments are all strings: call
1170** them A, B, and C. The result is also a string which is derived
drhf7b54962013-05-28 12:11:54 +00001171** from A by replacing every occurrence of B with C. The match
drh26b6d902007-03-17 13:27:54 +00001172** must be exact. Collating sequences are not used.
1173*/
1174static void replaceFunc(
1175 sqlite3_context *context,
1176 int argc,
1177 sqlite3_value **argv
1178){
1179 const unsigned char *zStr; /* The input string A */
1180 const unsigned char *zPattern; /* The pattern string B */
1181 const unsigned char *zRep; /* The replacement string C */
1182 unsigned char *zOut; /* The output */
1183 int nStr; /* Size of zStr */
1184 int nPattern; /* Size of zPattern */
1185 int nRep; /* Size of zRep */
drh2e6400b2007-05-08 15:46:18 +00001186 i64 nOut; /* Maximum size of zOut */
drh26b6d902007-03-17 13:27:54 +00001187 int loopLimit; /* Last zStr[] that might match zPattern[] */
1188 int i, j; /* Loop counters */
1189
1190 assert( argc==3 );
danielk1977f3d3c272008-11-19 16:52:44 +00001191 UNUSED_PARAMETER(argc);
drh26b6d902007-03-17 13:27:54 +00001192 zStr = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001193 if( zStr==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001194 nStr = sqlite3_value_bytes(argv[0]);
1195 assert( zStr==sqlite3_value_text(argv[0]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001196 zPattern = sqlite3_value_text(argv[1]);
drha605fe82009-02-01 18:08:40 +00001197 if( zPattern==0 ){
drh23336062009-02-03 13:19:12 +00001198 assert( sqlite3_value_type(argv[1])==SQLITE_NULL
1199 || sqlite3_context_db_handle(context)->mallocFailed );
drha605fe82009-02-01 18:08:40 +00001200 return;
1201 }
1202 if( zPattern[0]==0 ){
1203 assert( sqlite3_value_type(argv[1])!=SQLITE_NULL );
1204 sqlite3_result_value(context, argv[0]);
1205 return;
1206 }
drh1f0feef2007-05-15 13:27:07 +00001207 nPattern = sqlite3_value_bytes(argv[1]);
1208 assert( zPattern==sqlite3_value_text(argv[1]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001209 zRep = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +00001210 if( zRep==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001211 nRep = sqlite3_value_bytes(argv[2]);
1212 assert( zRep==sqlite3_value_text(argv[2]) );
drh2e6400b2007-05-08 15:46:18 +00001213 nOut = nStr + 1;
1214 assert( nOut<SQLITE_MAX_LENGTH );
drhb1a6c3c2008-03-20 16:30:17 +00001215 zOut = contextMalloc(context, (i64)nOut);
drh2e6400b2007-05-08 15:46:18 +00001216 if( zOut==0 ){
1217 return;
drh26b6d902007-03-17 13:27:54 +00001218 }
drh26b6d902007-03-17 13:27:54 +00001219 loopLimit = nStr - nPattern;
1220 for(i=j=0; i<=loopLimit; i++){
1221 if( zStr[i]!=zPattern[0] || memcmp(&zStr[i], zPattern, nPattern) ){
1222 zOut[j++] = zStr[i];
1223 }else{
drh4a50aac2007-08-23 02:47:53 +00001224 u8 *zOld;
drhbb4957f2008-03-20 14:03:29 +00001225 sqlite3 *db = sqlite3_context_db_handle(context);
drh2e6400b2007-05-08 15:46:18 +00001226 nOut += nRep - nPattern;
drh27e62db2009-04-02 10:16:17 +00001227 testcase( nOut-1==db->aLimit[SQLITE_LIMIT_LENGTH] );
1228 testcase( nOut-2==db->aLimit[SQLITE_LIMIT_LENGTH] );
1229 if( nOut-1>db->aLimit[SQLITE_LIMIT_LENGTH] ){
drha0206bc2007-05-08 15:15:02 +00001230 sqlite3_result_error_toobig(context);
drhb9755982010-07-24 16:34:37 +00001231 sqlite3_free(zOut);
danielk197717374e82007-05-08 14:39:04 +00001232 return;
1233 }
drh4a50aac2007-08-23 02:47:53 +00001234 zOld = zOut;
drhf3cdcdc2015-04-29 16:50:28 +00001235 zOut = sqlite3_realloc64(zOut, (int)nOut);
drh2e6400b2007-05-08 15:46:18 +00001236 if( zOut==0 ){
danielk1977a1644fd2007-08-29 12:31:25 +00001237 sqlite3_result_error_nomem(context);
drhb9755982010-07-24 16:34:37 +00001238 sqlite3_free(zOld);
drh2e6400b2007-05-08 15:46:18 +00001239 return;
1240 }
drh26b6d902007-03-17 13:27:54 +00001241 memcpy(&zOut[j], zRep, nRep);
1242 j += nRep;
1243 i += nPattern-1;
1244 }
1245 }
drh2e6400b2007-05-08 15:46:18 +00001246 assert( j+nStr-i+1==nOut );
drh26b6d902007-03-17 13:27:54 +00001247 memcpy(&zOut[j], &zStr[i], nStr-i);
1248 j += nStr - i;
1249 assert( j<=nOut );
1250 zOut[j] = 0;
1251 sqlite3_result_text(context, (char*)zOut, j, sqlite3_free);
1252}
1253
drh309b3382007-03-17 17:52:42 +00001254/*
1255** Implementation of the TRIM(), LTRIM(), and RTRIM() functions.
1256** The userdata is 0x1 for left trim, 0x2 for right trim, 0x3 for both.
1257*/
1258static void trimFunc(
1259 sqlite3_context *context,
1260 int argc,
1261 sqlite3_value **argv
1262){
1263 const unsigned char *zIn; /* Input string */
1264 const unsigned char *zCharSet; /* Set of characters to trim */
1265 int nIn; /* Number of bytes in input */
drh7209c692008-04-27 18:40:11 +00001266 int flags; /* 1: trimleft 2: trimright 3: trim */
drhd1e3a612007-04-27 21:59:52 +00001267 int i; /* Loop counter */
drh1bd10f82008-12-10 21:19:56 +00001268 unsigned char *aLen = 0; /* Length of each character in zCharSet */
1269 unsigned char **azChar = 0; /* Individual characters in zCharSet */
drhd1e3a612007-04-27 21:59:52 +00001270 int nChar; /* Number of characters in zCharSet */
1271
drh309b3382007-03-17 17:52:42 +00001272 if( sqlite3_value_type(argv[0])==SQLITE_NULL ){
1273 return;
1274 }
1275 zIn = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001276 if( zIn==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001277 nIn = sqlite3_value_bytes(argv[0]);
1278 assert( zIn==sqlite3_value_text(argv[0]) );
drh309b3382007-03-17 17:52:42 +00001279 if( argc==1 ){
drhd1e3a612007-04-27 21:59:52 +00001280 static const unsigned char lenOne[] = { 1 };
danielk1977a4de4532008-09-02 15:44:08 +00001281 static unsigned char * const azOne[] = { (u8*)" " };
drhd1e3a612007-04-27 21:59:52 +00001282 nChar = 1;
drh8cff3822007-05-02 02:08:28 +00001283 aLen = (u8*)lenOne;
danielk1977bc67da42007-12-11 04:23:19 +00001284 azChar = (unsigned char **)azOne;
drhd1e3a612007-04-27 21:59:52 +00001285 zCharSet = 0;
drh7a521cf2007-04-25 18:23:52 +00001286 }else if( (zCharSet = sqlite3_value_text(argv[1]))==0 ){
drh309b3382007-03-17 17:52:42 +00001287 return;
drhd1e3a612007-04-27 21:59:52 +00001288 }else{
1289 const unsigned char *z;
1290 for(z=zCharSet, nChar=0; *z; nChar++){
drh4a919112007-05-15 11:55:09 +00001291 SQLITE_SKIP_UTF8(z);
drhd1e3a612007-04-27 21:59:52 +00001292 }
1293 if( nChar>0 ){
drhb1a6c3c2008-03-20 16:30:17 +00001294 azChar = contextMalloc(context, ((i64)nChar)*(sizeof(char*)+1));
drhd1e3a612007-04-27 21:59:52 +00001295 if( azChar==0 ){
1296 return;
1297 }
1298 aLen = (unsigned char*)&azChar[nChar];
1299 for(z=zCharSet, nChar=0; *z; nChar++){
danielk1977bc67da42007-12-11 04:23:19 +00001300 azChar[nChar] = (unsigned char *)z;
drh4a919112007-05-15 11:55:09 +00001301 SQLITE_SKIP_UTF8(z);
drh3abbd392008-12-10 23:04:13 +00001302 aLen[nChar] = (u8)(z - azChar[nChar]);
drhd1e3a612007-04-27 21:59:52 +00001303 }
1304 }
drh309b3382007-03-17 17:52:42 +00001305 }
drhd1e3a612007-04-27 21:59:52 +00001306 if( nChar>0 ){
shane1fc41292008-07-08 22:28:48 +00001307 flags = SQLITE_PTR_TO_INT(sqlite3_user_data(context));
drh309b3382007-03-17 17:52:42 +00001308 if( flags & 1 ){
drhd1e3a612007-04-27 21:59:52 +00001309 while( nIn>0 ){
drh1bd10f82008-12-10 21:19:56 +00001310 int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001311 for(i=0; i<nChar; i++){
1312 len = aLen[i];
drh27e62db2009-04-02 10:16:17 +00001313 if( len<=nIn && memcmp(zIn, azChar[i], len)==0 ) break;
drhd1e3a612007-04-27 21:59:52 +00001314 }
1315 if( i>=nChar ) break;
1316 zIn += len;
1317 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001318 }
1319 }
1320 if( flags & 2 ){
drhd1e3a612007-04-27 21:59:52 +00001321 while( nIn>0 ){
drh1bd10f82008-12-10 21:19:56 +00001322 int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001323 for(i=0; i<nChar; i++){
1324 len = aLen[i];
1325 if( len<=nIn && memcmp(&zIn[nIn-len],azChar[i],len)==0 ) break;
1326 }
1327 if( i>=nChar ) break;
1328 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001329 }
1330 }
drhd1e3a612007-04-27 21:59:52 +00001331 if( zCharSet ){
1332 sqlite3_free(azChar);
1333 }
drh309b3382007-03-17 17:52:42 +00001334 }
1335 sqlite3_result_text(context, (char*)zIn, nIn, SQLITE_TRANSIENT);
1336}
drh26b6d902007-03-17 13:27:54 +00001337
danielk1977a4de4532008-09-02 15:44:08 +00001338
drhcc153132016-08-04 12:35:17 +00001339#ifdef SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION
1340/*
1341** The "unknown" function is automatically substituted in place of
1342** any unrecognized function name when doing an EXPLAIN or EXPLAIN QUERY PLAN
1343** when the SQLITE_ENABLE_UNKNOWN_FUNCTION compile-time option is used.
1344** When the "sqlite3" command-line shell is built using this functionality,
1345** that allows an EXPLAIN or EXPLAIN QUERY PLAN for complex queries
1346** involving application-defined functions to be examined in a generic
1347** sqlite3 shell.
1348*/
1349static void unknownFunc(
1350 sqlite3_context *context,
1351 int argc,
1352 sqlite3_value **argv
1353){
1354 /* no-op */
1355}
1356#endif /*SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION*/
1357
1358
drh2ba3ccc2009-12-08 02:06:08 +00001359/* IMP: R-25361-16150 This function is omitted from SQLite by default. It
1360** is only available if the SQLITE_SOUNDEX compile-time option is used
1361** when SQLite is built.
1362*/
drhd24cc422003-03-27 12:51:24 +00001363#ifdef SQLITE_SOUNDEX
1364/*
1365** Compute the soundex encoding of a word.
drh2ba3ccc2009-12-08 02:06:08 +00001366**
1367** IMP: R-59782-00072 The soundex(X) function returns a string that is the
1368** soundex encoding of the string X.
drhd24cc422003-03-27 12:51:24 +00001369*/
drh137c7282007-01-29 17:58:28 +00001370static void soundexFunc(
1371 sqlite3_context *context,
1372 int argc,
1373 sqlite3_value **argv
1374){
drhd24cc422003-03-27 12:51:24 +00001375 char zResult[8];
drh4c755c02004-08-08 20:22:17 +00001376 const u8 *zIn;
drhd24cc422003-03-27 12:51:24 +00001377 int i, j;
1378 static const unsigned char iCode[] = {
1379 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1380 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1381 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1382 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1383 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1384 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1385 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1386 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1387 };
1388 assert( argc==1 );
drh4c755c02004-08-08 20:22:17 +00001389 zIn = (u8*)sqlite3_value_text(argv[0]);
drhbdf67e02006-08-19 11:34:01 +00001390 if( zIn==0 ) zIn = (u8*)"";
drhdc86e2b2009-01-24 11:30:42 +00001391 for(i=0; zIn[i] && !sqlite3Isalpha(zIn[i]); i++){}
drhd24cc422003-03-27 12:51:24 +00001392 if( zIn[i] ){
drhbdf67e02006-08-19 11:34:01 +00001393 u8 prevcode = iCode[zIn[i]&0x7f];
danielk197778ca0e72009-01-20 16:53:39 +00001394 zResult[0] = sqlite3Toupper(zIn[i]);
drhd24cc422003-03-27 12:51:24 +00001395 for(j=1; j<4 && zIn[i]; i++){
1396 int code = iCode[zIn[i]&0x7f];
1397 if( code>0 ){
drhbdf67e02006-08-19 11:34:01 +00001398 if( code!=prevcode ){
1399 prevcode = code;
1400 zResult[j++] = code + '0';
1401 }
1402 }else{
1403 prevcode = 0;
drhd24cc422003-03-27 12:51:24 +00001404 }
1405 }
1406 while( j<4 ){
1407 zResult[j++] = '0';
1408 }
1409 zResult[j] = 0;
danielk1977d8123362004-06-12 09:25:12 +00001410 sqlite3_result_text(context, zResult, 4, SQLITE_TRANSIENT);
drhd24cc422003-03-27 12:51:24 +00001411 }else{
drh2ba3ccc2009-12-08 02:06:08 +00001412 /* IMP: R-64894-50321 The string "?000" is returned if the argument
1413 ** is NULL or contains no ASCII alphabetic characters. */
danielk1977d8123362004-06-12 09:25:12 +00001414 sqlite3_result_text(context, "?000", 4, SQLITE_STATIC);
drhd24cc422003-03-27 12:51:24 +00001415 }
1416}
drh2ba3ccc2009-12-08 02:06:08 +00001417#endif /* SQLITE_SOUNDEX */
drhd24cc422003-03-27 12:51:24 +00001418
drhfdb83b22006-06-17 14:12:47 +00001419#ifndef SQLITE_OMIT_LOAD_EXTENSION
1420/*
1421** A function that loads a shared-library extension then returns NULL.
1422*/
1423static void loadExt(sqlite3_context *context, int argc, sqlite3_value **argv){
danielk197765fd59f2006-06-24 11:51:33 +00001424 const char *zFile = (const char *)sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001425 const char *zProc;
drhfa4a4b92008-03-19 21:45:51 +00001426 sqlite3 *db = sqlite3_context_db_handle(context);
drhfdb83b22006-06-17 14:12:47 +00001427 char *zErrMsg = 0;
1428
drh191dd062016-04-21 01:30:09 +00001429 /* Disallow the load_extension() SQL function unless the SQLITE_LoadExtFunc
drh1a55ded2016-04-20 00:30:05 +00001430 ** flag is set. See the sqlite3_enable_load_extension() API.
1431 */
drhf602a162016-04-21 01:58:21 +00001432 if( (db->flags & SQLITE_LoadExtFunc)==0 ){
1433 sqlite3_result_error(context, "not authorized", -1);
1434 return;
1435 }
drh1a55ded2016-04-20 00:30:05 +00001436
drhfdb83b22006-06-17 14:12:47 +00001437 if( argc==2 ){
danielk197765fd59f2006-06-24 11:51:33 +00001438 zProc = (const char *)sqlite3_value_text(argv[1]);
drh7a521cf2007-04-25 18:23:52 +00001439 }else{
1440 zProc = 0;
drhfdb83b22006-06-17 14:12:47 +00001441 }
drh7a521cf2007-04-25 18:23:52 +00001442 if( zFile && sqlite3_load_extension(db, zFile, zProc, &zErrMsg) ){
drhfdb83b22006-06-17 14:12:47 +00001443 sqlite3_result_error(context, zErrMsg, -1);
1444 sqlite3_free(zErrMsg);
1445 }
1446}
1447#endif
1448
danielk197701427a62005-01-11 13:02:33 +00001449
drh0ac65892002-04-20 14:24:41 +00001450/*
drhd3a149e2002-02-24 17:12:53 +00001451** An instance of the following structure holds the context of a
drhdd5baa92002-02-27 19:50:59 +00001452** sum() or avg() aggregate computation.
1453*/
1454typedef struct SumCtx SumCtx;
1455struct SumCtx {
drh8c08e862006-02-11 17:34:00 +00001456 double rSum; /* Floating point sum */
1457 i64 iSum; /* Integer sum */
1458 i64 cnt; /* Number of elements summed */
1459 u8 overflow; /* True if integer overflow seen */
1460 u8 approx; /* True if non-integer value was input to the sum */
drhdd5baa92002-02-27 19:50:59 +00001461};
1462
1463/*
drha97fdd32006-01-12 22:17:50 +00001464** Routines used to compute the sum, average, and total.
1465**
1466** The SUM() function follows the (broken) SQL standard which means
1467** that it returns NULL if it sums over no inputs. TOTAL returns
1468** 0.0 in that case. In addition, TOTAL always returns a float where
1469** SUM might return an integer if it never encounters a floating point
drhc806d852006-05-11 13:25:39 +00001470** value. TOTAL never fails, but SUM might through an exception if
1471** it overflows an integer.
drhdd5baa92002-02-27 19:50:59 +00001472*/
danielk19770ae8b832004-05-25 12:05:56 +00001473static void sumStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drhdd5baa92002-02-27 19:50:59 +00001474 SumCtx *p;
drh3d1d95e2005-09-08 10:37:01 +00001475 int type;
drh3f219f42005-09-08 19:45:57 +00001476 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001477 UNUSED_PARAMETER(argc);
drh4f26d6c2004-05-26 23:25:30 +00001478 p = sqlite3_aggregate_context(context, sizeof(*p));
drh29d72102006-02-09 22:13:41 +00001479 type = sqlite3_value_numeric_type(argv[0]);
drh3d1d95e2005-09-08 10:37:01 +00001480 if( p && type!=SQLITE_NULL ){
drh739105c2002-05-29 23:22:23 +00001481 p->cnt++;
drh29d72102006-02-09 22:13:41 +00001482 if( type==SQLITE_INTEGER ){
drh8c08e862006-02-11 17:34:00 +00001483 i64 v = sqlite3_value_int64(argv[0]);
1484 p->rSum += v;
drh158b9cb2011-03-05 20:59:46 +00001485 if( (p->approx|p->overflow)==0 && sqlite3AddInt64(&p->iSum, v) ){
1486 p->overflow = 1;
drh29d72102006-02-09 22:13:41 +00001487 }
1488 }else{
drh8c08e862006-02-11 17:34:00 +00001489 p->rSum += sqlite3_value_double(argv[0]);
drh29d72102006-02-09 22:13:41 +00001490 p->approx = 1;
drh3f219f42005-09-08 19:45:57 +00001491 }
drh739105c2002-05-29 23:22:23 +00001492 }
drhdd5baa92002-02-27 19:50:59 +00001493}
danielk19770ae8b832004-05-25 12:05:56 +00001494static void sumFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001495 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001496 p = sqlite3_aggregate_context(context, 0);
drhc2bd9132005-09-08 20:37:43 +00001497 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001498 if( p->overflow ){
1499 sqlite3_result_error(context,"integer overflow",-1);
1500 }else if( p->approx ){
1501 sqlite3_result_double(context, p->rSum);
drhc2bd9132005-09-08 20:37:43 +00001502 }else{
drh8c08e862006-02-11 17:34:00 +00001503 sqlite3_result_int64(context, p->iSum);
drhc2bd9132005-09-08 20:37:43 +00001504 }
drh3d1d95e2005-09-08 10:37:01 +00001505 }
drhdd5baa92002-02-27 19:50:59 +00001506}
danielk19770ae8b832004-05-25 12:05:56 +00001507static void avgFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001508 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001509 p = sqlite3_aggregate_context(context, 0);
drh739105c2002-05-29 23:22:23 +00001510 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001511 sqlite3_result_double(context, p->rSum/(double)p->cnt);
drhdd5baa92002-02-27 19:50:59 +00001512 }
1513}
drha97fdd32006-01-12 22:17:50 +00001514static void totalFinalize(sqlite3_context *context){
1515 SumCtx *p;
1516 p = sqlite3_aggregate_context(context, 0);
shanefbd60f82009-02-04 03:59:25 +00001517 /* (double)0 In case of SQLITE_OMIT_FLOATING_POINT... */
1518 sqlite3_result_double(context, p ? p->rSum : (double)0);
drha97fdd32006-01-12 22:17:50 +00001519}
drhdd5baa92002-02-27 19:50:59 +00001520
1521/*
drh0bce8352002-02-28 00:41:10 +00001522** The following structure keeps track of state information for the
1523** count() aggregate function.
1524*/
1525typedef struct CountCtx CountCtx;
1526struct CountCtx {
drhfc6ad392006-02-09 13:38:19 +00001527 i64 n;
drh0bce8352002-02-28 00:41:10 +00001528};
drhdd5baa92002-02-27 19:50:59 +00001529
drh0bce8352002-02-28 00:41:10 +00001530/*
1531** Routines to implement the count() aggregate function.
1532*/
danielk19770ae8b832004-05-25 12:05:56 +00001533static void countStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drh0bce8352002-02-28 00:41:10 +00001534 CountCtx *p;
drh4f26d6c2004-05-26 23:25:30 +00001535 p = sqlite3_aggregate_context(context, sizeof(*p));
drh9c054832004-05-31 18:51:57 +00001536 if( (argc==0 || SQLITE_NULL!=sqlite3_value_type(argv[0])) && p ){
drh0bce8352002-02-28 00:41:10 +00001537 p->n++;
1538 }
drh2e79c3d2009-04-08 23:04:14 +00001539
drhd3264c72009-04-15 13:39:47 +00001540#ifndef SQLITE_OMIT_DEPRECATED
drh2e79c3d2009-04-08 23:04:14 +00001541 /* The sqlite3_aggregate_count() function is deprecated. But just to make
1542 ** sure it still operates correctly, verify that its count agrees with our
1543 ** internal count when using count(*) and when the total count can be
1544 ** expressed as a 32-bit integer. */
1545 assert( argc==1 || p==0 || p->n>0x7fffffff
1546 || p->n==sqlite3_aggregate_count(context) );
drhd3264c72009-04-15 13:39:47 +00001547#endif
drh0bce8352002-02-28 00:41:10 +00001548}
danielk19770ae8b832004-05-25 12:05:56 +00001549static void countFinalize(sqlite3_context *context){
drh0bce8352002-02-28 00:41:10 +00001550 CountCtx *p;
drhabfcea22005-09-06 20:36:48 +00001551 p = sqlite3_aggregate_context(context, 0);
drhfc6ad392006-02-09 13:38:19 +00001552 sqlite3_result_int64(context, p ? p->n : 0);
drh0bce8352002-02-28 00:41:10 +00001553}
1554
1555/*
drh0bce8352002-02-28 00:41:10 +00001556** Routines to implement min() and max() aggregate functions.
1557*/
danielk197762c14b32008-11-19 09:05:26 +00001558static void minmaxStep(
1559 sqlite3_context *context,
1560 int NotUsed,
1561 sqlite3_value **argv
1562){
danielk197788208052004-05-25 01:13:20 +00001563 Mem *pArg = (Mem *)argv[0];
drh9eb516c2004-07-18 20:52:32 +00001564 Mem *pBest;
danielk197762c14b32008-11-19 09:05:26 +00001565 UNUSED_PARAMETER(NotUsed);
drh9eb516c2004-07-18 20:52:32 +00001566
drh9eb516c2004-07-18 20:52:32 +00001567 pBest = (Mem *)sqlite3_aggregate_context(context, sizeof(*pBest));
danielk19773aeab9e2004-06-24 00:20:04 +00001568 if( !pBest ) return;
drh268380c2004-02-25 13:47:31 +00001569
drh94a6d992012-02-02 18:42:09 +00001570 if( sqlite3_value_type(argv[0])==SQLITE_NULL ){
1571 if( pBest->flags ) sqlite3SkipAccumulatorLoad(context);
1572 }else if( pBest->flags ){
drh9eb516c2004-07-18 20:52:32 +00001573 int max;
1574 int cmp;
danielk1977dc1bdc42004-06-11 10:51:27 +00001575 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk19777e18c252004-05-25 11:47:24 +00001576 /* This step function is used for both the min() and max() aggregates,
1577 ** the only difference between the two being that the sense of the
1578 ** comparison is inverted. For the max() aggregate, the
1579 ** sqlite3_user_data() function returns (void *)-1. For min() it
1580 ** returns (void *)db, where db is the sqlite3* database pointer.
1581 ** Therefore the next statement sets variable 'max' to 1 for the max()
1582 ** aggregate, or 0 for min().
1583 */
drh309b3382007-03-17 17:52:42 +00001584 max = sqlite3_user_data(context)!=0;
danielk1977dc1bdc42004-06-11 10:51:27 +00001585 cmp = sqlite3MemCompare(pBest, pArg, pColl);
danielk197788208052004-05-25 01:13:20 +00001586 if( (max && cmp<0) || (!max && cmp>0) ){
drhb21c8cd2007-08-21 19:33:56 +00001587 sqlite3VdbeMemCopy(pBest, pArg);
drh7a957892012-02-02 17:35:43 +00001588 }else{
1589 sqlite3SkipAccumulatorLoad(context);
danielk197788208052004-05-25 01:13:20 +00001590 }
drh268380c2004-02-25 13:47:31 +00001591 }else{
drh035e5632014-09-16 14:16:31 +00001592 pBest->db = sqlite3_context_db_handle(context);
drhb21c8cd2007-08-21 19:33:56 +00001593 sqlite3VdbeMemCopy(pBest, pArg);
drh0bce8352002-02-28 00:41:10 +00001594 }
1595}
danielk19770ae8b832004-05-25 12:05:56 +00001596static void minMaxFinalize(sqlite3_context *context){
danielk197788208052004-05-25 01:13:20 +00001597 sqlite3_value *pRes;
drhabfcea22005-09-06 20:36:48 +00001598 pRes = (sqlite3_value *)sqlite3_aggregate_context(context, 0);
1599 if( pRes ){
drh94a6d992012-02-02 18:42:09 +00001600 if( pRes->flags ){
drhabfcea22005-09-06 20:36:48 +00001601 sqlite3_result_value(context, pRes);
1602 }
1603 sqlite3VdbeMemRelease(pRes);
drh0bce8352002-02-28 00:41:10 +00001604 }
1605}
drhdd5baa92002-02-27 19:50:59 +00001606
drhb0689692007-11-01 17:38:30 +00001607/*
1608** group_concat(EXPR, ?SEPARATOR?)
1609*/
1610static void groupConcatStep(
1611 sqlite3_context *context,
1612 int argc,
1613 sqlite3_value **argv
1614){
1615 const char *zVal;
drhade86482007-11-28 22:36:40 +00001616 StrAccum *pAccum;
drhb0689692007-11-01 17:38:30 +00001617 const char *zSep;
drh07d31172009-02-02 21:57:05 +00001618 int nVal, nSep;
1619 assert( argc==1 || argc==2 );
1620 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drhade86482007-11-28 22:36:40 +00001621 pAccum = (StrAccum*)sqlite3_aggregate_context(context, sizeof(*pAccum));
1622
1623 if( pAccum ){
drhbb4957f2008-03-20 14:03:29 +00001624 sqlite3 *db = sqlite3_context_db_handle(context);
drhc0490572015-05-02 11:45:53 +00001625 int firstTerm = pAccum->mxAlloc==0;
drhbb4957f2008-03-20 14:03:29 +00001626 pAccum->mxAlloc = db->aLimit[SQLITE_LIMIT_LENGTH];
drh8bfd7192009-06-19 16:44:41 +00001627 if( !firstTerm ){
drh07d31172009-02-02 21:57:05 +00001628 if( argc==2 ){
1629 zSep = (char*)sqlite3_value_text(argv[1]);
1630 nSep = sqlite3_value_bytes(argv[1]);
drhb0689692007-11-01 17:38:30 +00001631 }else{
1632 zSep = ",";
drhade86482007-11-28 22:36:40 +00001633 nSep = 1;
drhb0689692007-11-01 17:38:30 +00001634 }
dan3043b532016-12-30 17:40:14 +00001635 if( zSep ) sqlite3StrAccumAppend(pAccum, zSep, nSep);
drhb0689692007-11-01 17:38:30 +00001636 }
drh07d31172009-02-02 21:57:05 +00001637 zVal = (char*)sqlite3_value_text(argv[0]);
1638 nVal = sqlite3_value_bytes(argv[0]);
drh8009c9b2014-05-07 20:24:00 +00001639 if( zVal ) sqlite3StrAccumAppend(pAccum, zVal, nVal);
drhb0689692007-11-01 17:38:30 +00001640 }
1641}
1642static void groupConcatFinalize(sqlite3_context *context){
drhade86482007-11-28 22:36:40 +00001643 StrAccum *pAccum;
1644 pAccum = sqlite3_aggregate_context(context, 0);
1645 if( pAccum ){
drhb49bc862013-08-21 21:12:10 +00001646 if( pAccum->accError==STRACCUM_TOOBIG ){
drhade86482007-11-28 22:36:40 +00001647 sqlite3_result_error_toobig(context);
drhb49bc862013-08-21 21:12:10 +00001648 }else if( pAccum->accError==STRACCUM_NOMEM ){
drhade86482007-11-28 22:36:40 +00001649 sqlite3_result_error_nomem(context);
1650 }else{
1651 sqlite3_result_text(context, sqlite3StrAccumFinish(pAccum), -1,
1652 sqlite3_free);
1653 }
drhb0689692007-11-01 17:38:30 +00001654 }
1655}
drh4e5ffc52004-08-31 00:52:37 +00001656
drhd3a149e2002-02-24 17:12:53 +00001657/*
drha4741842010-04-25 20:58:37 +00001658** This routine does per-connection function registration. Most
1659** of the built-in functions above are part of the global function set.
1660** This routine only deals with those that are not global.
drhdc04c582002-02-24 01:55:15 +00001661*/
drh80738d92016-02-15 00:34:16 +00001662void sqlite3RegisterPerConnectionBuiltinFunctions(sqlite3 *db){
drha4741842010-04-25 20:58:37 +00001663 int rc = sqlite3_overload_function(db, "MATCH", 2);
1664 assert( rc==SQLITE_NOMEM || rc==SQLITE_OK );
1665 if( rc==SQLITE_NOMEM ){
drh4a642b62016-02-05 01:55:27 +00001666 sqlite3OomFault(db);
danielk1977832a58a2007-06-22 15:21:15 +00001667 }
drh55ef4d92005-08-14 01:20:37 +00001668}
1669
1670/*
1671** Set the LIKEOPT flag on the 2-argument function with the given name.
1672*/
drh1bd10f82008-12-10 21:19:56 +00001673static void setLikeOptFlag(sqlite3 *db, const char *zName, u8 flagVal){
drh55ef4d92005-08-14 01:20:37 +00001674 FuncDef *pDef;
drh80738d92016-02-15 00:34:16 +00001675 pDef = sqlite3FindFunction(db, zName, 2, SQLITE_UTF8, 0);
drhd27135a2009-04-02 13:36:37 +00001676 if( ALWAYS(pDef) ){
drhd36e1042013-09-06 13:10:12 +00001677 pDef->funcFlags |= flagVal;
drh55ef4d92005-08-14 01:20:37 +00001678 }
1679}
1680
1681/*
1682** Register the built-in LIKE and GLOB functions. The caseSensitive
1683** parameter determines whether or not the LIKE operator is case
1684** sensitive. GLOB is always case sensitive.
1685*/
1686void sqlite3RegisterLikeFunctions(sqlite3 *db, int caseSensitive){
1687 struct compareInfo *pInfo;
1688 if( caseSensitive ){
1689 pInfo = (struct compareInfo*)&likeInfoAlt;
1690 }else{
1691 pInfo = (struct compareInfo*)&likeInfoNorm;
1692 }
drh901e9942010-12-15 18:54:37 +00001693 sqlite3CreateFunc(db, "like", 2, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0);
1694 sqlite3CreateFunc(db, "like", 3, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0);
1695 sqlite3CreateFunc(db, "glob", 2, SQLITE_UTF8,
dand2199f02010-08-27 17:48:52 +00001696 (struct compareInfo*)&globInfo, likeFunc, 0, 0, 0);
drhd64fe2f2005-08-28 17:00:23 +00001697 setLikeOptFlag(db, "glob", SQLITE_FUNC_LIKE | SQLITE_FUNC_CASE);
1698 setLikeOptFlag(db, "like",
1699 caseSensitive ? (SQLITE_FUNC_LIKE | SQLITE_FUNC_CASE) : SQLITE_FUNC_LIKE);
drh55ef4d92005-08-14 01:20:37 +00001700}
1701
1702/*
1703** pExpr points to an expression which implements a function. If
1704** it is appropriate to apply the LIKE optimization to that function
1705** then set aWc[0] through aWc[2] to the wildcard characters and
1706** return TRUE. If the function is not a LIKE-style function then
1707** return FALSE.
drh16897072015-03-07 00:57:37 +00001708**
1709** *pIsNocase is set to true if uppercase and lowercase are equivalent for
1710** the function (default for LIKE). If the function makes the distinction
1711** between uppercase and lowercase (as does GLOB) then *pIsNocase is set to
1712** false.
drh55ef4d92005-08-14 01:20:37 +00001713*/
drhd64fe2f2005-08-28 17:00:23 +00001714int sqlite3IsLikeFunction(sqlite3 *db, Expr *pExpr, int *pIsNocase, char *aWc){
drh55ef4d92005-08-14 01:20:37 +00001715 FuncDef *pDef;
danielk19776ab3a2e2009-02-19 14:39:25 +00001716 if( pExpr->op!=TK_FUNCTION
1717 || !pExpr->x.pList
1718 || pExpr->x.pList->nExpr!=2
1719 ){
drh55ef4d92005-08-14 01:20:37 +00001720 return 0;
1721 }
danielk19776ab3a2e2009-02-19 14:39:25 +00001722 assert( !ExprHasProperty(pExpr, EP_xIsSelect) );
drh80738d92016-02-15 00:34:16 +00001723 pDef = sqlite3FindFunction(db, pExpr->u.zToken, 2, SQLITE_UTF8, 0);
drhd36e1042013-09-06 13:10:12 +00001724 if( NEVER(pDef==0) || (pDef->funcFlags & SQLITE_FUNC_LIKE)==0 ){
drh55ef4d92005-08-14 01:20:37 +00001725 return 0;
1726 }
1727
1728 /* The memcpy() statement assumes that the wildcard characters are
1729 ** the first three statements in the compareInfo structure. The
1730 ** asserts() that follow verify that assumption
1731 */
1732 memcpy(aWc, pDef->pUserData, 3);
1733 assert( (char*)&likeInfoAlt == (char*)&likeInfoAlt.matchAll );
1734 assert( &((char*)&likeInfoAlt)[1] == (char*)&likeInfoAlt.matchOne );
1735 assert( &((char*)&likeInfoAlt)[2] == (char*)&likeInfoAlt.matchSet );
drhd36e1042013-09-06 13:10:12 +00001736 *pIsNocase = (pDef->funcFlags & SQLITE_FUNC_CASE)==0;
drh55ef4d92005-08-14 01:20:37 +00001737 return 1;
drhdc04c582002-02-24 01:55:15 +00001738}
danielk19778c0a7912008-08-20 14:49:23 +00001739
drh70a8ca32008-08-21 18:49:27 +00001740/*
peter.d.reid60ec9142014-09-06 16:39:46 +00001741** All of the FuncDef structures in the aBuiltinFunc[] array above
drh777c5382008-08-21 20:21:34 +00001742** to the global function hash table. This occurs at start-time (as
1743** a consequence of calling sqlite3_initialize()).
1744**
1745** After this routine runs
drh70a8ca32008-08-21 18:49:27 +00001746*/
drh80738d92016-02-15 00:34:16 +00001747void sqlite3RegisterBuiltinFunctions(void){
danielk197793ce7412008-09-01 19:14:02 +00001748 /*
1749 ** The following array holds FuncDef structures for all of the functions
1750 ** defined in this file.
1751 **
1752 ** The array cannot be constant since changes are made to the
1753 ** FuncDef.pHash elements at start-time. The elements of this array
1754 ** are read-only after initialization is complete.
drh80738d92016-02-15 00:34:16 +00001755 **
1756 ** For peak efficiency, put the most frequently used function last.
danielk197793ce7412008-09-01 19:14:02 +00001757 */
drh80738d92016-02-15 00:34:16 +00001758 static FuncDef aBuiltinFunc[] = {
1759#ifdef SQLITE_SOUNDEX
1760 FUNCTION(soundex, 1, 0, 0, soundexFunc ),
1761#endif
1762#ifndef SQLITE_OMIT_LOAD_EXTENSION
1763 VFUNCTION(load_extension, 1, 0, 0, loadExt ),
1764 VFUNCTION(load_extension, 2, 0, 0, loadExt ),
1765#endif
1766#if SQLITE_USER_AUTHENTICATION
1767 FUNCTION(sqlite_crypt, 2, 0, 0, sqlite3CryptFunc ),
1768#endif
1769#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
1770 DFUNCTION(sqlite_compileoption_used,1, 0, 0, compileoptionusedFunc ),
1771 DFUNCTION(sqlite_compileoption_get, 1, 0, 0, compileoptiongetFunc ),
1772#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
1773 FUNCTION2(unlikely, 1, 0, 0, noopFunc, SQLITE_FUNC_UNLIKELY),
1774 FUNCTION2(likelihood, 2, 0, 0, noopFunc, SQLITE_FUNC_UNLIKELY),
1775 FUNCTION2(likely, 1, 0, 0, noopFunc, SQLITE_FUNC_UNLIKELY),
drh54240752017-01-03 14:39:30 +00001776#ifdef SQLITE_DEBUG
drha1a523a2016-12-26 00:18:36 +00001777 FUNCTION2(affinity, 1, 0, 0, noopFunc, SQLITE_FUNC_AFFINITY),
drh54240752017-01-03 14:39:30 +00001778#endif
danielk197793ce7412008-09-01 19:14:02 +00001779 FUNCTION(ltrim, 1, 1, 0, trimFunc ),
1780 FUNCTION(ltrim, 2, 1, 0, trimFunc ),
1781 FUNCTION(rtrim, 1, 2, 0, trimFunc ),
1782 FUNCTION(rtrim, 2, 2, 0, trimFunc ),
1783 FUNCTION(trim, 1, 3, 0, trimFunc ),
1784 FUNCTION(trim, 2, 3, 0, trimFunc ),
1785 FUNCTION(min, -1, 0, 1, minmaxFunc ),
1786 FUNCTION(min, 0, 0, 1, 0 ),
drh9588ad92014-09-15 14:46:02 +00001787 AGGREGATE2(min, 1, 0, 1, minmaxStep, minMaxFinalize,
1788 SQLITE_FUNC_MINMAX ),
danielk197793ce7412008-09-01 19:14:02 +00001789 FUNCTION(max, -1, 1, 1, minmaxFunc ),
1790 FUNCTION(max, 0, 1, 1, 0 ),
drh9588ad92014-09-15 14:46:02 +00001791 AGGREGATE2(max, 1, 1, 1, minmaxStep, minMaxFinalize,
1792 SQLITE_FUNC_MINMAX ),
drha748fdc2012-03-28 01:34:47 +00001793 FUNCTION2(typeof, 1, 0, 0, typeofFunc, SQLITE_FUNC_TYPEOF),
1794 FUNCTION2(length, 1, 0, 0, lengthFunc, SQLITE_FUNC_LENGTH),
drhd55e0722012-10-25 03:07:29 +00001795 FUNCTION(instr, 2, 0, 0, instrFunc ),
drha5c14162013-12-17 15:03:06 +00001796 FUNCTION(printf, -1, 0, 0, printfFunc ),
drhd495d8c2013-02-22 19:34:25 +00001797 FUNCTION(unicode, 1, 0, 0, unicodeFunc ),
1798 FUNCTION(char, -1, 0, 0, charFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001799 FUNCTION(abs, 1, 0, 0, absFunc ),
shanefbd60f82009-02-04 03:59:25 +00001800#ifndef SQLITE_OMIT_FLOATING_POINT
danielk197793ce7412008-09-01 19:14:02 +00001801 FUNCTION(round, 1, 0, 0, roundFunc ),
1802 FUNCTION(round, 2, 0, 0, roundFunc ),
shanefbd60f82009-02-04 03:59:25 +00001803#endif
danielk197793ce7412008-09-01 19:14:02 +00001804 FUNCTION(upper, 1, 0, 0, upperFunc ),
1805 FUNCTION(lower, 1, 0, 0, lowerFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001806 FUNCTION(hex, 1, 0, 0, hexFunc ),
drhcca9f3d2013-09-06 15:23:29 +00001807 FUNCTION2(ifnull, 2, 0, 0, noopFunc, SQLITE_FUNC_COALESCE),
drhb1fba282013-11-21 14:33:48 +00001808 VFUNCTION(random, 0, 0, 0, randomFunc ),
1809 VFUNCTION(randomblob, 1, 0, 0, randomBlob ),
danielk197793ce7412008-09-01 19:14:02 +00001810 FUNCTION(nullif, 2, 0, 1, nullifFunc ),
drh03bf26d2015-08-31 21:16:36 +00001811 DFUNCTION(sqlite_version, 0, 0, 0, versionFunc ),
1812 DFUNCTION(sqlite_source_id, 0, 0, 0, sourceidFunc ),
drh840561f2011-04-27 18:08:42 +00001813 FUNCTION(sqlite_log, 2, 0, 0, errlogFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001814 FUNCTION(quote, 1, 0, 0, quoteFunc ),
drhb1fba282013-11-21 14:33:48 +00001815 VFUNCTION(last_insert_rowid, 0, 0, 0, last_insert_rowid),
1816 VFUNCTION(changes, 0, 0, 0, changes ),
1817 VFUNCTION(total_changes, 0, 0, 0, total_changes ),
danielk197793ce7412008-09-01 19:14:02 +00001818 FUNCTION(replace, 3, 0, 0, replaceFunc ),
1819 FUNCTION(zeroblob, 1, 0, 0, zeroblobFunc ),
drh80738d92016-02-15 00:34:16 +00001820 FUNCTION(substr, 2, 0, 0, substrFunc ),
1821 FUNCTION(substr, 3, 0, 0, substrFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001822 AGGREGATE(sum, 1, 0, 0, sumStep, sumFinalize ),
1823 AGGREGATE(total, 1, 0, 0, sumStep, totalFinalize ),
1824 AGGREGATE(avg, 1, 0, 0, sumStep, avgFinalize ),
drh9588ad92014-09-15 14:46:02 +00001825 AGGREGATE2(count, 0, 0, 0, countStep, countFinalize,
1826 SQLITE_FUNC_COUNT ),
danielk197793ce7412008-09-01 19:14:02 +00001827 AGGREGATE(count, 1, 0, 0, countStep, countFinalize ),
drh07d31172009-02-02 21:57:05 +00001828 AGGREGATE(group_concat, 1, 0, 0, groupConcatStep, groupConcatFinalize),
1829 AGGREGATE(group_concat, 2, 0, 0, groupConcatStep, groupConcatFinalize),
danielk197793ce7412008-09-01 19:14:02 +00001830
1831 LIKEFUNC(glob, 2, &globInfo, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
drhcc153132016-08-04 12:35:17 +00001832#ifdef SQLITE_CASE_SENSITIVE_LIKE
danielk197793ce7412008-09-01 19:14:02 +00001833 LIKEFUNC(like, 2, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
1834 LIKEFUNC(like, 3, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
drhcc153132016-08-04 12:35:17 +00001835#else
danielk197793ce7412008-09-01 19:14:02 +00001836 LIKEFUNC(like, 2, &likeInfoNorm, SQLITE_FUNC_LIKE),
1837 LIKEFUNC(like, 3, &likeInfoNorm, SQLITE_FUNC_LIKE),
drhcc153132016-08-04 12:35:17 +00001838#endif
1839#ifdef SQLITE_ENABLE_UNKNOWN_SQL_FUNCTION
1840 FUNCTION(unknown, -1, 0, 0, unknownFunc ),
1841#endif
drh80738d92016-02-15 00:34:16 +00001842 FUNCTION(coalesce, 1, 0, 0, 0 ),
1843 FUNCTION(coalesce, 0, 0, 0, 0 ),
1844 FUNCTION2(coalesce, -1, 0, 0, noopFunc, SQLITE_FUNC_COALESCE),
danielk197793ce7412008-09-01 19:14:02 +00001845 };
drh545f5872010-04-24 14:02:59 +00001846#ifndef SQLITE_OMIT_ALTERTABLE
1847 sqlite3AlterFunctions();
1848#endif
dan0106e372013-08-12 16:34:32 +00001849#if defined(SQLITE_ENABLE_STAT3) || defined(SQLITE_ENABLE_STAT4)
1850 sqlite3AnalyzeFunctions();
1851#endif
drh80738d92016-02-15 00:34:16 +00001852 sqlite3RegisterDateTimeFunctions();
1853 sqlite3InsertBuiltinFuncs(aBuiltinFunc, ArraySize(aBuiltinFunc));
1854
1855#if 0 /* Enable to print out how the built-in functions are hashed */
1856 {
1857 int i;
1858 FuncDef *p;
1859 for(i=0; i<SQLITE_FUNC_HASH_SZ; i++){
1860 printf("FUNC-HASH %02d:", i);
1861 for(p=sqlite3BuiltinFunctions.a[i]; p; p=p->u.pHash){
1862 int n = sqlite3Strlen30(p->zName);
1863 int h = p->zName[0] + n;
1864 printf(" %s(%d)", p->zName, h);
1865 }
1866 printf("\n");
1867 }
1868 }
1869#endif
drh70a8ca32008-08-21 18:49:27 +00001870}