blob: 5b7056b401048c513be3fdde34c6d9c728a93b79 [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){
25 return context->pColl;
26}
27
drh0bce8352002-02-28 00:41:10 +000028/*
drh7a957892012-02-02 17:35:43 +000029** Indicate that the accumulator load should be skipped on this
30** iteration of the aggregate loop.
31*/
32static void sqlite3SkipAccumulatorLoad(sqlite3_context *context){
33 context->skipFlag = 1;
34}
35
36/*
drh0bce8352002-02-28 00:41:10 +000037** Implementation of the non-aggregate min() and max() functions
38*/
drhf9b596e2004-05-26 16:54:42 +000039static void minmaxFunc(
40 sqlite3_context *context,
41 int argc,
42 sqlite3_value **argv
43){
drh0bce8352002-02-28 00:41:10 +000044 int i;
drh268380c2004-02-25 13:47:31 +000045 int mask; /* 0 for min() or 0xffffffff for max() */
drhf9b596e2004-05-26 16:54:42 +000046 int iBest;
danielk1977dc1bdc42004-06-11 10:51:27 +000047 CollSeq *pColl;
drh0bce8352002-02-28 00:41:10 +000048
drh65595cd2009-02-02 16:32:55 +000049 assert( argc>1 );
drhc44af712004-09-02 15:53:56 +000050 mask = sqlite3_user_data(context)==0 ? 0 : -1;
danielk1977dc1bdc42004-06-11 10:51:27 +000051 pColl = sqlite3GetFuncCollSeq(context);
52 assert( pColl );
danielk1977c572ef72004-05-27 09:28:41 +000053 assert( mask==-1 || mask==0 );
drhf9b596e2004-05-26 16:54:42 +000054 iBest = 0;
drh9c054832004-05-31 18:51:57 +000055 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drhf9b596e2004-05-26 16:54:42 +000056 for(i=1; i<argc; i++){
drh9c054832004-05-31 18:51:57 +000057 if( sqlite3_value_type(argv[i])==SQLITE_NULL ) return;
danielk1977dc1bdc42004-06-11 10:51:27 +000058 if( (sqlite3MemCompare(argv[iBest], argv[i], pColl)^mask)>=0 ){
drh65595cd2009-02-02 16:32:55 +000059 testcase( mask==0 );
drhf9b596e2004-05-26 16:54:42 +000060 iBest = i;
drh0bce8352002-02-28 00:41:10 +000061 }
62 }
drhf4479502004-05-27 03:12:53 +000063 sqlite3_result_value(context, argv[iBest]);
drh0bce8352002-02-28 00:41:10 +000064}
drh0bce8352002-02-28 00:41:10 +000065
drh268380c2004-02-25 13:47:31 +000066/*
67** Return the type of the argument.
68*/
drhf9b596e2004-05-26 16:54:42 +000069static void typeofFunc(
70 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +000071 int NotUsed,
drhf9b596e2004-05-26 16:54:42 +000072 sqlite3_value **argv
73){
danielk197735bb9d02004-05-24 12:55:54 +000074 const char *z = 0;
danielk197762c14b32008-11-19 09:05:26 +000075 UNUSED_PARAMETER(NotUsed);
danielk197735bb9d02004-05-24 12:55:54 +000076 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +000077 case SQLITE_INTEGER: z = "integer"; break;
78 case SQLITE_TEXT: z = "text"; break;
79 case SQLITE_FLOAT: z = "real"; break;
80 case SQLITE_BLOB: z = "blob"; break;
drh65595cd2009-02-02 16:32:55 +000081 default: z = "null"; break;
danielk197735bb9d02004-05-24 12:55:54 +000082 }
danielk1977d8123362004-06-12 09:25:12 +000083 sqlite3_result_text(context, z, -1, SQLITE_STATIC);
drh0bce8352002-02-28 00:41:10 +000084}
85
drh5708d2d2005-06-22 10:53:59 +000086
87/*
drh0bce8352002-02-28 00:41:10 +000088** Implementation of the length() function
89*/
drhf9b596e2004-05-26 16:54:42 +000090static void lengthFunc(
91 sqlite3_context *context,
92 int argc,
93 sqlite3_value **argv
94){
drh0bce8352002-02-28 00:41:10 +000095 int len;
96
97 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +000098 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +000099 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +0000100 case SQLITE_BLOB:
101 case SQLITE_INTEGER:
102 case SQLITE_FLOAT: {
drhf4479502004-05-27 03:12:53 +0000103 sqlite3_result_int(context, sqlite3_value_bytes(argv[0]));
drhf9b596e2004-05-26 16:54:42 +0000104 break;
105 }
drh9c054832004-05-31 18:51:57 +0000106 case SQLITE_TEXT: {
drh2646da72005-12-09 20:02:05 +0000107 const unsigned char *z = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +0000108 if( z==0 ) return;
drh4a919112007-05-15 11:55:09 +0000109 len = 0;
110 while( *z ){
111 len++;
112 SQLITE_SKIP_UTF8(z);
113 }
drhf4479502004-05-27 03:12:53 +0000114 sqlite3_result_int(context, len);
drhf9b596e2004-05-26 16:54:42 +0000115 break;
116 }
117 default: {
118 sqlite3_result_null(context);
119 break;
120 }
121 }
drh0bce8352002-02-28 00:41:10 +0000122}
123
124/*
drh2ba3ccc2009-12-08 02:06:08 +0000125** Implementation of the abs() function.
126**
127** IMP: R-23979-26855 The abs(X) function returns the absolute value of
128** the numeric argument X.
drh0bce8352002-02-28 00:41:10 +0000129*/
danielk19770ae8b832004-05-25 12:05:56 +0000130static void absFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drh0bce8352002-02-28 00:41:10 +0000131 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000132 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +0000133 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +0000134 case SQLITE_INTEGER: {
danielk1977f93bbbe2004-05-27 10:30:52 +0000135 i64 iVal = sqlite3_value_int64(argv[0]);
drh52fc8492006-02-23 21:43:55 +0000136 if( iVal<0 ){
drh693e6712014-01-24 22:58:00 +0000137 if( iVal==SMALLEST_INT64 ){
drheb091cd2013-11-09 19:47:15 +0000138 /* IMP: R-31676-45509 If X is the integer -9223372036854775808
139 ** then abs(X) throws an integer overflow error since there is no
drh2ba3ccc2009-12-08 02:06:08 +0000140 ** equivalent positive 64-bit two complement value. */
drh52fc8492006-02-23 21:43:55 +0000141 sqlite3_result_error(context, "integer overflow", -1);
142 return;
143 }
144 iVal = -iVal;
145 }
danielk1977f93bbbe2004-05-27 10:30:52 +0000146 sqlite3_result_int64(context, iVal);
drhf9b596e2004-05-26 16:54:42 +0000147 break;
148 }
drh9c054832004-05-31 18:51:57 +0000149 case SQLITE_NULL: {
drh2ba3ccc2009-12-08 02:06:08 +0000150 /* IMP: R-37434-19929 Abs(X) returns NULL if X is NULL. */
drhf9b596e2004-05-26 16:54:42 +0000151 sqlite3_result_null(context);
152 break;
153 }
154 default: {
drh2ba3ccc2009-12-08 02:06:08 +0000155 /* Because sqlite3_value_double() returns 0.0 if the argument is not
156 ** something that can be converted into a number, we have:
157 ** IMP: R-57326-31541 Abs(X) return 0.0 if X is a string or blob that
158 ** cannot be converted to a numeric value.
159 */
danielk1977f93bbbe2004-05-27 10:30:52 +0000160 double rVal = sqlite3_value_double(argv[0]);
drh52fc8492006-02-23 21:43:55 +0000161 if( rVal<0 ) rVal = -rVal;
danielk1977f93bbbe2004-05-27 10:30:52 +0000162 sqlite3_result_double(context, rVal);
drhf9b596e2004-05-26 16:54:42 +0000163 break;
164 }
165 }
drh0bce8352002-02-28 00:41:10 +0000166}
167
168/*
drhd55e0722012-10-25 03:07:29 +0000169** Implementation of the instr() function.
170**
171** instr(haystack,needle) finds the first occurrence of needle
172** in haystack and returns the number of previous characters plus 1,
173** or 0 if needle does not occur within haystack.
174**
175** If both haystack and needle are BLOBs, then the result is one more than
176** the number of bytes in haystack prior to the first occurrence of needle,
177** or 0 if needle never occurs in haystack.
178*/
179static void instrFunc(
180 sqlite3_context *context,
181 int argc,
182 sqlite3_value **argv
183){
184 const unsigned char *zHaystack;
185 const unsigned char *zNeedle;
186 int nHaystack;
187 int nNeedle;
188 int typeHaystack, typeNeedle;
189 int N = 1;
190 int isText;
191
drh68c804b2012-12-04 11:03:11 +0000192 UNUSED_PARAMETER(argc);
drhd55e0722012-10-25 03:07:29 +0000193 typeHaystack = sqlite3_value_type(argv[0]);
194 typeNeedle = sqlite3_value_type(argv[1]);
195 if( typeHaystack==SQLITE_NULL || typeNeedle==SQLITE_NULL ) return;
196 nHaystack = sqlite3_value_bytes(argv[0]);
197 nNeedle = sqlite3_value_bytes(argv[1]);
198 if( typeHaystack==SQLITE_BLOB && typeNeedle==SQLITE_BLOB ){
199 zHaystack = sqlite3_value_blob(argv[0]);
200 zNeedle = sqlite3_value_blob(argv[1]);
201 isText = 0;
202 }else{
203 zHaystack = sqlite3_value_text(argv[0]);
204 zNeedle = sqlite3_value_text(argv[1]);
205 isText = 1;
206 }
207 while( nNeedle<=nHaystack && memcmp(zHaystack, zNeedle, nNeedle)!=0 ){
208 N++;
209 do{
210 nHaystack--;
211 zHaystack++;
212 }while( isText && (zHaystack[0]&0xc0)==0x80 );
213 }
214 if( nNeedle>nHaystack ) N = 0;
215 sqlite3_result_int(context, N);
216}
217
218/*
drha5c14162013-12-17 15:03:06 +0000219** Implementation of the printf() function.
220*/
221static void printfFunc(
222 sqlite3_context *context,
223 int argc,
224 sqlite3_value **argv
225){
226 PrintfArguments x;
227 StrAccum str;
228 const char *zFormat;
229 int n;
230
231 if( argc>=1 && (zFormat = (const char*)sqlite3_value_text(argv[0]))!=0 ){
232 x.nArg = argc-1;
233 x.nUsed = 0;
234 x.apArg = argv+1;
235 sqlite3StrAccumInit(&str, 0, 0, SQLITE_MAX_LENGTH);
236 str.db = sqlite3_context_db_handle(context);
237 sqlite3XPrintf(&str, SQLITE_PRINTF_SQLFUNC, zFormat, &x);
238 n = str.nChar;
239 sqlite3_result_text(context, sqlite3StrAccumFinish(&str), n,
240 SQLITE_DYNAMIC);
241 }
242}
243
244/*
drhf764e6f2007-05-15 01:13:47 +0000245** Implementation of the substr() function.
246**
247** substr(x,p1,p2) returns p2 characters of x[] beginning with p1.
248** p1 is 1-indexed. So substr(x,1,1) returns the first character
249** of x. If x is text, then we actually count UTF-8 characters.
250** If x is a blob, then we count bytes.
251**
252** If p1 is negative, then we begin abs(p1) from the end of x[].
shaneh779b8f12009-11-12 05:04:50 +0000253**
drhf7b54962013-05-28 12:11:54 +0000254** If p2 is negative, return the p2 characters preceding p1.
drh0bce8352002-02-28 00:41:10 +0000255*/
drhf9b596e2004-05-26 16:54:42 +0000256static void substrFunc(
257 sqlite3_context *context,
258 int argc,
259 sqlite3_value **argv
260){
drh2646da72005-12-09 20:02:05 +0000261 const unsigned char *z;
262 const unsigned char *z2;
drh023ae032007-05-08 12:12:16 +0000263 int len;
drhf764e6f2007-05-15 01:13:47 +0000264 int p0type;
drh023ae032007-05-08 12:12:16 +0000265 i64 p1, p2;
drh65595cd2009-02-02 16:32:55 +0000266 int negP2 = 0;
drhf9b596e2004-05-26 16:54:42 +0000267
drh64f31512007-10-12 19:11:55 +0000268 assert( argc==3 || argc==2 );
drh8198d252009-02-01 19:42:37 +0000269 if( sqlite3_value_type(argv[1])==SQLITE_NULL
270 || (argc==3 && sqlite3_value_type(argv[2])==SQLITE_NULL)
271 ){
272 return;
273 }
drhf764e6f2007-05-15 01:13:47 +0000274 p0type = sqlite3_value_type(argv[0]);
drh4adc4cb2009-11-11 20:53:31 +0000275 p1 = sqlite3_value_int(argv[1]);
drhf764e6f2007-05-15 01:13:47 +0000276 if( p0type==SQLITE_BLOB ){
277 len = sqlite3_value_bytes(argv[0]);
278 z = sqlite3_value_blob(argv[0]);
279 if( z==0 ) return;
drh1f0feef2007-05-15 13:27:07 +0000280 assert( len==sqlite3_value_bytes(argv[0]) );
drhf764e6f2007-05-15 01:13:47 +0000281 }else{
282 z = sqlite3_value_text(argv[0]);
283 if( z==0 ) return;
drh4a919112007-05-15 11:55:09 +0000284 len = 0;
drh4adc4cb2009-11-11 20:53:31 +0000285 if( p1<0 ){
286 for(z2=z; *z2; len++){
287 SQLITE_SKIP_UTF8(z2);
288 }
drh4a919112007-05-15 11:55:09 +0000289 }
drhf764e6f2007-05-15 01:13:47 +0000290 }
drh64f31512007-10-12 19:11:55 +0000291 if( argc==3 ){
292 p2 = sqlite3_value_int(argv[2]);
drh65595cd2009-02-02 16:32:55 +0000293 if( p2<0 ){
294 p2 = -p2;
295 negP2 = 1;
296 }
drh64f31512007-10-12 19:11:55 +0000297 }else{
drhbb4957f2008-03-20 14:03:29 +0000298 p2 = sqlite3_context_db_handle(context)->aLimit[SQLITE_LIMIT_LENGTH];
drh64f31512007-10-12 19:11:55 +0000299 }
drh0bce8352002-02-28 00:41:10 +0000300 if( p1<0 ){
drh89425d52002-02-28 03:04:48 +0000301 p1 += len;
drh653bc752002-02-28 03:31:10 +0000302 if( p1<0 ){
303 p2 += p1;
drh65595cd2009-02-02 16:32:55 +0000304 if( p2<0 ) p2 = 0;
drh653bc752002-02-28 03:31:10 +0000305 p1 = 0;
306 }
drh0bce8352002-02-28 00:41:10 +0000307 }else if( p1>0 ){
308 p1--;
drh65595cd2009-02-02 16:32:55 +0000309 }else if( p2>0 ){
310 p2--;
drh0bce8352002-02-28 00:41:10 +0000311 }
drh65595cd2009-02-02 16:32:55 +0000312 if( negP2 ){
313 p1 -= p2;
drh4e79c592009-02-01 19:23:32 +0000314 if( p1<0 ){
315 p2 += p1;
316 p1 = 0;
317 }
318 }
drh65595cd2009-02-02 16:32:55 +0000319 assert( p1>=0 && p2>=0 );
drhf764e6f2007-05-15 01:13:47 +0000320 if( p0type!=SQLITE_BLOB ){
drh4a919112007-05-15 11:55:09 +0000321 while( *z && p1 ){
322 SQLITE_SKIP_UTF8(z);
323 p1--;
drhf764e6f2007-05-15 01:13:47 +0000324 }
drh4a919112007-05-15 11:55:09 +0000325 for(z2=z; *z2 && p2; p2--){
326 SQLITE_SKIP_UTF8(z2);
drhf764e6f2007-05-15 01:13:47 +0000327 }
drhbbf483f2014-09-09 20:30:24 +0000328 sqlite3_result_text64(context, (char*)z, z2-z, SQLITE_TRANSIENT,
329 SQLITE_UTF8);
drhf764e6f2007-05-15 01:13:47 +0000330 }else{
drh4adc4cb2009-11-11 20:53:31 +0000331 if( p1+p2>len ){
332 p2 = len-p1;
333 if( p2<0 ) p2 = 0;
334 }
drhbbf483f2014-09-09 20:30:24 +0000335 sqlite3_result_blob64(context, (char*)&z[p1], (u64)p2, SQLITE_TRANSIENT);
drh0bce8352002-02-28 00:41:10 +0000336 }
drh0bce8352002-02-28 00:41:10 +0000337}
338
339/*
340** Implementation of the round() function
341*/
shanefbd60f82009-02-04 03:59:25 +0000342#ifndef SQLITE_OMIT_FLOATING_POINT
danielk19770ae8b832004-05-25 12:05:56 +0000343static void roundFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
danielk197751ad0ec2004-05-24 12:39:02 +0000344 int n = 0;
drh0bce8352002-02-28 00:41:10 +0000345 double r;
drh50d654d2009-06-03 01:24:54 +0000346 char *zBuf;
drh0bce8352002-02-28 00:41:10 +0000347 assert( argc==1 || argc==2 );
danielk197751ad0ec2004-05-24 12:39:02 +0000348 if( argc==2 ){
drh9c054832004-05-31 18:51:57 +0000349 if( SQLITE_NULL==sqlite3_value_type(argv[1]) ) return;
danielk197751ad0ec2004-05-24 12:39:02 +0000350 n = sqlite3_value_int(argv[1]);
351 if( n>30 ) n = 30;
352 if( n<0 ) n = 0;
353 }
drhd589a922006-03-02 03:02:48 +0000354 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drh4f26d6c2004-05-26 23:25:30 +0000355 r = sqlite3_value_double(argv[0]);
shaneh147e1762010-02-17 04:19:27 +0000356 /* If Y==0 and X will fit in a 64-bit int,
357 ** handle the rounding directly,
358 ** otherwise use printf.
359 */
360 if( n==0 && r>=0 && r<LARGEST_INT64-1 ){
361 r = (double)((sqlite_int64)(r+0.5));
362 }else if( n==0 && r<0 && (-r)<LARGEST_INT64-1 ){
363 r = -(double)((sqlite_int64)((-r)+0.5));
drh50d654d2009-06-03 01:24:54 +0000364 }else{
shaneh147e1762010-02-17 04:19:27 +0000365 zBuf = sqlite3_mprintf("%.*f",n,r);
366 if( zBuf==0 ){
367 sqlite3_result_error_nomem(context);
368 return;
369 }
drh9339da12010-09-30 00:50:49 +0000370 sqlite3AtoF(zBuf, &r, sqlite3Strlen30(zBuf), SQLITE_UTF8);
drh50d654d2009-06-03 01:24:54 +0000371 sqlite3_free(zBuf);
drh50d654d2009-06-03 01:24:54 +0000372 }
shaneh147e1762010-02-17 04:19:27 +0000373 sqlite3_result_double(context, r);
drh0bce8352002-02-28 00:41:10 +0000374}
shanefbd60f82009-02-04 03:59:25 +0000375#endif
drhdc04c582002-02-24 01:55:15 +0000376
danielk197726783a52007-08-29 14:06:22 +0000377/*
378** Allocate nByte bytes of space using sqlite3_malloc(). If the
379** allocation fails, call sqlite3_result_error_nomem() to notify
drh27e62db2009-04-02 10:16:17 +0000380** the database handle that malloc() has failed and return NULL.
381** If nByte is larger than the maximum string or blob length, then
382** raise an SQLITE_TOOBIG exception and return NULL.
danielk197726783a52007-08-29 14:06:22 +0000383*/
drhb1a6c3c2008-03-20 16:30:17 +0000384static void *contextMalloc(sqlite3_context *context, i64 nByte){
drhbb4957f2008-03-20 14:03:29 +0000385 char *z;
drh27e62db2009-04-02 10:16:17 +0000386 sqlite3 *db = sqlite3_context_db_handle(context);
drhef31c6a2009-04-02 09:07:12 +0000387 assert( nByte>0 );
drh27e62db2009-04-02 10:16:17 +0000388 testcase( nByte==db->aLimit[SQLITE_LIMIT_LENGTH] );
389 testcase( nByte==db->aLimit[SQLITE_LIMIT_LENGTH]+1 );
390 if( nByte>db->aLimit[SQLITE_LIMIT_LENGTH] ){
drhbb4957f2008-03-20 14:03:29 +0000391 sqlite3_result_error_toobig(context);
392 z = 0;
393 }else{
drhda4ca9d2014-09-09 17:27:35 +0000394 z = sqlite3Malloc(nByte);
drhef31c6a2009-04-02 09:07:12 +0000395 if( !z ){
drhbb4957f2008-03-20 14:03:29 +0000396 sqlite3_result_error_nomem(context);
397 }
danielk1977a1644fd2007-08-29 12:31:25 +0000398 }
399 return z;
400}
401
drhdc04c582002-02-24 01:55:15 +0000402/*
403** Implementation of the upper() and lower() SQL functions.
404*/
danielk19770ae8b832004-05-25 12:05:56 +0000405static void upperFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drh7a521cf2007-04-25 18:23:52 +0000406 char *z1;
407 const char *z2;
drh9310ef22007-04-27 17:16:20 +0000408 int i, n;
drh1d34fde2009-02-03 15:50:33 +0000409 UNUSED_PARAMETER(argc);
drh7a521cf2007-04-25 18:23:52 +0000410 z2 = (char*)sqlite3_value_text(argv[0]);
drh1f0feef2007-05-15 13:27:07 +0000411 n = sqlite3_value_bytes(argv[0]);
412 /* Verify that the call to _bytes() does not invalidate the _text() pointer */
413 assert( z2==(char*)sqlite3_value_text(argv[0]) );
drh7a521cf2007-04-25 18:23:52 +0000414 if( z2 ){
drhb1a6c3c2008-03-20 16:30:17 +0000415 z1 = contextMalloc(context, ((i64)n)+1);
drh7a521cf2007-04-25 18:23:52 +0000416 if( z1 ){
drhdf901d32011-10-13 18:00:11 +0000417 for(i=0; i<n; i++){
418 z1[i] = (char)sqlite3Toupper(z2[i]);
drh7a521cf2007-04-25 18:23:52 +0000419 }
drhdf901d32011-10-13 18:00:11 +0000420 sqlite3_result_text(context, z1, n, sqlite3_free);
drh7a521cf2007-04-25 18:23:52 +0000421 }
drhdc04c582002-02-24 01:55:15 +0000422 }
423}
danielk19770ae8b832004-05-25 12:05:56 +0000424static void lowerFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drhdf901d32011-10-13 18:00:11 +0000425 char *z1;
drh7a521cf2007-04-25 18:23:52 +0000426 const char *z2;
drh9310ef22007-04-27 17:16:20 +0000427 int i, n;
drh1d34fde2009-02-03 15:50:33 +0000428 UNUSED_PARAMETER(argc);
drh7a521cf2007-04-25 18:23:52 +0000429 z2 = (char*)sqlite3_value_text(argv[0]);
drh1f0feef2007-05-15 13:27:07 +0000430 n = sqlite3_value_bytes(argv[0]);
431 /* Verify that the call to _bytes() does not invalidate the _text() pointer */
432 assert( z2==(char*)sqlite3_value_text(argv[0]) );
drh7a521cf2007-04-25 18:23:52 +0000433 if( z2 ){
drhb1a6c3c2008-03-20 16:30:17 +0000434 z1 = contextMalloc(context, ((i64)n)+1);
drh7a521cf2007-04-25 18:23:52 +0000435 if( z1 ){
drhdf901d32011-10-13 18:00:11 +0000436 for(i=0; i<n; i++){
437 z1[i] = sqlite3Tolower(z2[i]);
drh7a521cf2007-04-25 18:23:52 +0000438 }
drhdf901d32011-10-13 18:00:11 +0000439 sqlite3_result_text(context, z1, n, sqlite3_free);
drh7a521cf2007-04-25 18:23:52 +0000440 }
drhdc04c582002-02-24 01:55:15 +0000441 }
442}
443
drhae6bb952009-11-11 00:24:31 +0000444/*
drhcca9f3d2013-09-06 15:23:29 +0000445** Some functions like COALESCE() and IFNULL() and UNLIKELY() are implemented
446** as VDBE code so that unused argument values do not have to be computed.
447** However, we still need some kind of function implementation for this
448** routines in the function table. The noopFunc macro provides this.
449** noopFunc will never be called so it doesn't matter what the implementation
450** is. We might as well use the "version()" function as a substitute.
drhae6bb952009-11-11 00:24:31 +0000451*/
drhcca9f3d2013-09-06 15:23:29 +0000452#define noopFunc versionFunc /* Substitute function - never called */
drh3212e182002-02-28 00:46:26 +0000453
454/*
drhf9ffac92002-03-02 19:00:31 +0000455** Implementation of random(). Return a random integer.
456*/
drhf9b596e2004-05-26 16:54:42 +0000457static void randomFunc(
458 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000459 int NotUsed,
460 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000461){
drh52fc8492006-02-23 21:43:55 +0000462 sqlite_int64 r;
danielk197762c14b32008-11-19 09:05:26 +0000463 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drh2fa18682008-03-19 14:15:34 +0000464 sqlite3_randomness(sizeof(r), &r);
drh3034e3d2009-04-02 14:05:21 +0000465 if( r<0 ){
466 /* We need to prevent a random number of 0x8000000000000000
467 ** (or -9223372036854775808) since when you do abs() of that
468 ** number of you get the same value back again. To do this
469 ** in a way that is testable, mask the sign bit off of negative
470 ** values, resulting in a positive value. Then take the
471 ** 2s complement of that positive value. The end result can
472 ** therefore be no less than -9223372036854775807.
473 */
drhaf8001b2012-02-11 19:53:24 +0000474 r = -(r & LARGEST_INT64);
drh3034e3d2009-04-02 14:05:21 +0000475 }
drh52fc8492006-02-23 21:43:55 +0000476 sqlite3_result_int64(context, r);
drhf9ffac92002-03-02 19:00:31 +0000477}
478
479/*
drh137c7282007-01-29 17:58:28 +0000480** Implementation of randomblob(N). Return a random blob
481** that is N bytes long.
drh63cf66f2007-01-29 15:50:05 +0000482*/
drh137c7282007-01-29 17:58:28 +0000483static void randomBlob(
drh63cf66f2007-01-29 15:50:05 +0000484 sqlite3_context *context,
485 int argc,
486 sqlite3_value **argv
487){
drh137c7282007-01-29 17:58:28 +0000488 int n;
489 unsigned char *p;
drh63cf66f2007-01-29 15:50:05 +0000490 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000491 UNUSED_PARAMETER(argc);
drh63cf66f2007-01-29 15:50:05 +0000492 n = sqlite3_value_int(argv[0]);
drh023ae032007-05-08 12:12:16 +0000493 if( n<1 ){
494 n = 1;
495 }
danielk1977a1644fd2007-08-29 12:31:25 +0000496 p = contextMalloc(context, n);
drh02d85832007-05-07 19:31:15 +0000497 if( p ){
drh2fa18682008-03-19 14:15:34 +0000498 sqlite3_randomness(n, p);
drh17435752007-08-16 04:30:38 +0000499 sqlite3_result_blob(context, (char*)p, n, sqlite3_free);
drh02d85832007-05-07 19:31:15 +0000500 }
drh63cf66f2007-01-29 15:50:05 +0000501}
502
503/*
drh6ed41ad2002-04-06 14:10:47 +0000504** Implementation of the last_insert_rowid() SQL function. The return
danielk197724b03fd2004-05-10 10:34:34 +0000505** value is the same as the sqlite3_last_insert_rowid() API function.
drh6ed41ad2002-04-06 14:10:47 +0000506*/
danielk197751ad0ec2004-05-24 12:39:02 +0000507static void last_insert_rowid(
danielk19770ae8b832004-05-25 12:05:56 +0000508 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000509 int NotUsed,
510 sqlite3_value **NotUsed2
danielk197751ad0ec2004-05-24 12:39:02 +0000511){
drhfa4a4b92008-03-19 21:45:51 +0000512 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000513 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000514 /* IMP: R-51513-12026 The last_insert_rowid() SQL function is a
515 ** wrapper around the sqlite3_last_insert_rowid() C/C++ interface
516 ** function. */
drhf9b596e2004-05-26 16:54:42 +0000517 sqlite3_result_int64(context, sqlite3_last_insert_rowid(db));
drh6ed41ad2002-04-06 14:10:47 +0000518}
519
rdcf146a772004-02-25 22:51:06 +0000520/*
drhab2f1f92010-01-11 18:26:42 +0000521** Implementation of the changes() SQL function.
522**
523** IMP: R-62073-11209 The changes() SQL function is a wrapper
524** around the sqlite3_changes() C/C++ function and hence follows the same
525** rules for counting changes.
rdcf146a772004-02-25 22:51:06 +0000526*/
danielk1977b28af712004-06-21 06:50:26 +0000527static void changes(
drhf9b596e2004-05-26 16:54:42 +0000528 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000529 int NotUsed,
530 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000531){
drhfa4a4b92008-03-19 21:45:51 +0000532 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000533 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhf4479502004-05-27 03:12:53 +0000534 sqlite3_result_int(context, sqlite3_changes(db));
rdcb0c374f2004-02-20 22:53:38 +0000535}
rdcf146a772004-02-25 22:51:06 +0000536
537/*
danielk1977b28af712004-06-21 06:50:26 +0000538** Implementation of the total_changes() SQL function. The return value is
539** the same as the sqlite3_total_changes() API function.
rdcf146a772004-02-25 22:51:06 +0000540*/
danielk1977b28af712004-06-21 06:50:26 +0000541static void total_changes(
542 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000543 int NotUsed,
544 sqlite3_value **NotUsed2
danielk197751ad0ec2004-05-24 12:39:02 +0000545){
drhfa4a4b92008-03-19 21:45:51 +0000546 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000547 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000548 /* IMP: R-52756-41993 This function is a wrapper around the
549 ** sqlite3_total_changes() C/C++ interface. */
danielk1977b28af712004-06-21 06:50:26 +0000550 sqlite3_result_int(context, sqlite3_total_changes(db));
rdcb0c374f2004-02-20 22:53:38 +0000551}
552
drh6ed41ad2002-04-06 14:10:47 +0000553/*
drh4e5ffc52004-08-31 00:52:37 +0000554** A structure defining how to do GLOB-style comparisons.
danielk1977d02eb1f2004-06-06 09:44:03 +0000555*/
drh4e5ffc52004-08-31 00:52:37 +0000556struct compareInfo {
557 u8 matchAll;
558 u8 matchOne;
559 u8 matchSet;
560 u8 noCase;
danielk1977d02eb1f2004-06-06 09:44:03 +0000561};
drh55ef4d92005-08-14 01:20:37 +0000562
drhb9175ae2007-12-07 18:39:04 +0000563/*
564** For LIKE and GLOB matching on EBCDIC machines, assume that every
565** character is exactly one byte in size. Also, all characters are
566** able to participate in upper-case-to-lower-case mappings in EBCDIC
567** whereas only characters less than 0x80 do in ASCII.
568*/
569#if defined(SQLITE_EBCDIC)
drh88b33222014-09-25 03:51:37 +0000570# define sqlite3Utf8Read(A) (*((*A)++))
571# define GlobUpperToLower(A) A = sqlite3UpperToLower[A]
572# define GlobUpperToLowerAscii(A) A = sqlite3UpperToLower[A]
drhb9175ae2007-12-07 18:39:04 +0000573#else
drh88b33222014-09-25 03:51:37 +0000574# define GlobUpperToLower(A) if( A<=0x7f ){ A = sqlite3UpperToLower[A]; }
575# define GlobUpperToLowerAscii(A) A = sqlite3UpperToLower[A]
drhb9175ae2007-12-07 18:39:04 +0000576#endif
577
drh4e5ffc52004-08-31 00:52:37 +0000578static const struct compareInfo globInfo = { '*', '?', '[', 0 };
drh55ef4d92005-08-14 01:20:37 +0000579/* The correct SQL-92 behavior is for the LIKE operator to ignore
580** case. Thus 'a' LIKE 'A' would be true. */
581static const struct compareInfo likeInfoNorm = { '%', '_', 0, 1 };
582/* If SQLITE_CASE_SENSITIVE_LIKE is defined, then the LIKE operator
583** is case sensitive causing 'a' LIKE 'A' to be false */
584static const struct compareInfo likeInfoAlt = { '%', '_', 0, 0 };
danielk1977d02eb1f2004-06-06 09:44:03 +0000585
586/*
drh4e5ffc52004-08-31 00:52:37 +0000587** Compare two UTF-8 strings for equality where the first string can
drh9fdfdc82014-09-25 11:08:57 +0000588** potentially be a "glob" or "like" expression. Return true (1) if they
drh4e5ffc52004-08-31 00:52:37 +0000589** are the same and false (0) if they are different.
drh0ac65892002-04-20 14:24:41 +0000590**
drh4e5ffc52004-08-31 00:52:37 +0000591** Globbing rules:
drh0ac65892002-04-20 14:24:41 +0000592**
drh4e5ffc52004-08-31 00:52:37 +0000593** '*' Matches any sequence of zero or more characters.
danielk1977d02eb1f2004-06-06 09:44:03 +0000594**
drh4e5ffc52004-08-31 00:52:37 +0000595** '?' Matches exactly one character.
596**
597** [...] Matches one character from the enclosed list of
598** characters.
599**
600** [^...] Matches one character not in the enclosed list.
601**
602** With the [...] and [^...] matching, a ']' character can be included
603** in the list by making it the first character after '[' or '^'. A
604** range of characters can be specified using '-'. Example:
605** "[a-z]" matches any single lower-case letter. To match a '-', make
606** it the last character in the list.
607**
drh9fdfdc82014-09-25 11:08:57 +0000608** Like matching rules:
609**
610** '%' Matches any sequence of zero or more characters
611**
612*** '_' Matches any one character
613**
614** Ec Where E is the "esc" character and c is any other
615** character, including '%', '_', and esc, match exactly c.
616**
617** The comments through this routine usually assume glob matching.
618**
drh4e5ffc52004-08-31 00:52:37 +0000619** This routine is usually quick, but can be N**2 in the worst case.
drh0ac65892002-04-20 14:24:41 +0000620*/
danielk19777c6303c2004-11-17 16:41:29 +0000621static int patternCompare(
drh4e5ffc52004-08-31 00:52:37 +0000622 const u8 *zPattern, /* The glob pattern */
623 const u8 *zString, /* The string to compare against the glob */
danielk19777c6303c2004-11-17 16:41:29 +0000624 const struct compareInfo *pInfo, /* Information about how to do the compare */
drh0a32fa62011-06-13 12:19:21 +0000625 u32 esc /* The escape character */
danielk197751ad0ec2004-05-24 12:39:02 +0000626){
drh9fdfdc82014-09-25 11:08:57 +0000627 u32 c, c2; /* Next pattern and input string chars */
628 u32 matchOne = pInfo->matchOne; /* "?" or "_" */
629 u32 matchAll = pInfo->matchAll; /* "*" or "%" */
630 u32 matchOther; /* "[" or the escape character */
631 u8 noCase = pInfo->noCase; /* True if uppercase==lowercase */
632 const u8 *zEscaped = 0; /* One past the last escaped input char */
drh88b33222014-09-25 03:51:37 +0000633
634 /* The GLOB operator does not have an ESCAPE clause. And LIKE does not
635 ** have the matchSet operator. So we either have to look for one or
636 ** the other, never both. Hence the single variable matchOther is used
637 ** to store the one we have to look for.
638 */
639 matchOther = esc ? esc : pInfo->matchSet;
danielk1977d02eb1f2004-06-06 09:44:03 +0000640
drh42610962012-09-17 18:56:32 +0000641 while( (c = sqlite3Utf8Read(&zPattern))!=0 ){
drh9fdfdc82014-09-25 11:08:57 +0000642 if( c==matchAll ){ /* Match "*" */
643 /* Skip over multiple "*" characters in the pattern. If there
644 ** are also "?" characters, skip those as well, but consume a
645 ** single character of the input string for each "?" skipped */
drh42610962012-09-17 18:56:32 +0000646 while( (c=sqlite3Utf8Read(&zPattern)) == matchAll
drh66150952007-07-23 19:12:41 +0000647 || c == matchOne ){
drh42610962012-09-17 18:56:32 +0000648 if( c==matchOne && sqlite3Utf8Read(&zString)==0 ){
drh66150952007-07-23 19:12:41 +0000649 return 0;
drh4e5ffc52004-08-31 00:52:37 +0000650 }
danielk1977ad7dd422004-06-06 12:41:49 +0000651 }
drh66150952007-07-23 19:12:41 +0000652 if( c==0 ){
drh9fdfdc82014-09-25 11:08:57 +0000653 return 1; /* "*" at the end of the pattern matches */
drh88b33222014-09-25 03:51:37 +0000654 }else if( c==matchOther ){
655 if( esc ){
656 c = sqlite3Utf8Read(&zPattern);
657 if( c==0 ) return 0;
658 }else{
drh9fdfdc82014-09-25 11:08:57 +0000659 /* "[...]" immediately follows the "*". We have to do a slow
660 ** recursive search in this case, but it is an unusual case. */
drh88b33222014-09-25 03:51:37 +0000661 assert( matchOther<0x80 ); /* '[' is a single-byte character */
662 while( *zString
663 && patternCompare(&zPattern[-1],zString,pInfo,esc)==0 ){
664 SQLITE_SKIP_UTF8(zString);
665 }
666 return *zString!=0;
drh66150952007-07-23 19:12:41 +0000667 }
drh66150952007-07-23 19:12:41 +0000668 }
drh9fdfdc82014-09-25 11:08:57 +0000669
670 /* At this point variable c contains the first character of the
671 ** pattern string past the "*". Search in the input string for the
672 ** first matching character and recursively contine the match from
673 ** that point.
674 **
675 ** For a case-insensitive search, set variable cx to be the same as
676 ** c but in the other case and search the input string for either
677 ** c or cx.
678 */
679 if( c<=0x80 ){
680 u32 cx;
681 if( noCase ){
682 cx = sqlite3Toupper(c);
683 c = sqlite3Tolower(c);
drh66150952007-07-23 19:12:41 +0000684 }else{
drh9fdfdc82014-09-25 11:08:57 +0000685 cx = c;
drh4e5ffc52004-08-31 00:52:37 +0000686 }
drh9fdfdc82014-09-25 11:08:57 +0000687 while( (c2 = *(zString++))!=0 ){
688 if( c2!=c && c2!=cx ) continue;
689 if( patternCompare(zPattern,zString,pInfo,esc) ) return 1;
690 }
691 }else{
692 while( (c2 = sqlite3Utf8Read(&zString))!=0 ){
693 if( c2!=c ) continue;
694 if( patternCompare(zPattern,zString,pInfo,esc) ) return 1;
695 }
drh66150952007-07-23 19:12:41 +0000696 }
697 return 0;
drh88b33222014-09-25 03:51:37 +0000698 }
drh88b33222014-09-25 03:51:37 +0000699 if( c==matchOther ){
700 if( esc ){
701 c = sqlite3Utf8Read(&zPattern);
702 if( c==0 ) return 0;
drh9fdfdc82014-09-25 11:08:57 +0000703 zEscaped = zPattern;
drh88b33222014-09-25 03:51:37 +0000704 }else{
705 u32 prior_c = 0;
drh9fdfdc82014-09-25 11:08:57 +0000706 int seen = 0;
707 int invert = 0;
drh88b33222014-09-25 03:51:37 +0000708 c = sqlite3Utf8Read(&zString);
709 if( c==0 ) return 0;
710 c2 = sqlite3Utf8Read(&zPattern);
711 if( c2=='^' ){
712 invert = 1;
713 c2 = sqlite3Utf8Read(&zPattern);
714 }
715 if( c2==']' ){
716 if( c==']' ) seen = 1;
717 c2 = sqlite3Utf8Read(&zPattern);
718 }
719 while( c2 && c2!=']' ){
720 if( c2=='-' && zPattern[0]!=']' && zPattern[0]!=0 && prior_c>0 ){
721 c2 = sqlite3Utf8Read(&zPattern);
722 if( c>=prior_c && c<=c2 ) seen = 1;
723 prior_c = 0;
724 }else{
725 if( c==c2 ){
726 seen = 1;
727 }
728 prior_c = c2;
729 }
730 c2 = sqlite3Utf8Read(&zPattern);
731 }
732 if( c2==0 || (seen ^ invert)==0 ){
733 return 0;
734 }
735 continue;
736 }
737 }
738 c2 = sqlite3Utf8Read(&zString);
739 if( c==c2 ) continue;
drh9fdfdc82014-09-25 11:08:57 +0000740 if( noCase && c<0x80 && c2<0x80 && sqlite3Tolower(c)==sqlite3Tolower(c2) ){
741 continue;
742 }
743 if( c==matchOne && zPattern!=zEscaped && c2!=0 ) continue;
744 return 0;
danielk197751ad0ec2004-05-24 12:39:02 +0000745 }
drh4e5ffc52004-08-31 00:52:37 +0000746 return *zString==0;
drh0ac65892002-04-20 14:24:41 +0000747}
drh4e5ffc52004-08-31 00:52:37 +0000748
drh55ef4d92005-08-14 01:20:37 +0000749/*
drh56282a52013-04-10 16:13:38 +0000750** The sqlite3_strglob() interface.
751*/
752int sqlite3_strglob(const char *zGlobPattern, const char *zString){
753 return patternCompare((u8*)zGlobPattern, (u8*)zString, &globInfo, 0)==0;
754}
755
756/*
drh55ef4d92005-08-14 01:20:37 +0000757** Count the number of times that the LIKE operator (or GLOB which is
758** just a variation of LIKE) gets called. This is used for testing
759** only.
760*/
761#ifdef SQLITE_TEST
762int sqlite3_like_count = 0;
763#endif
764
danielk19773f6b0872004-06-17 05:36:44 +0000765
766/*
767** Implementation of the like() SQL function. This function implements
768** the build-in LIKE operator. The first argument to the function is the
769** pattern and the second argument is the string. So, the SQL statements:
770**
771** A LIKE B
772**
773** is implemented as like(B,A).
774**
drh55ef4d92005-08-14 01:20:37 +0000775** This same function (with a different compareInfo structure) computes
776** the GLOB operator.
danielk19773f6b0872004-06-17 05:36:44 +0000777*/
778static void likeFunc(
779 sqlite3_context *context,
780 int argc,
781 sqlite3_value **argv
782){
drhbeb818d2007-05-08 15:34:47 +0000783 const unsigned char *zA, *zB;
drh0a32fa62011-06-13 12:19:21 +0000784 u32 escape = 0;
drh27e62db2009-04-02 10:16:17 +0000785 int nPat;
drhbb4957f2008-03-20 14:03:29 +0000786 sqlite3 *db = sqlite3_context_db_handle(context);
drhbeb818d2007-05-08 15:34:47 +0000787
drh1f0feef2007-05-15 13:27:07 +0000788 zB = sqlite3_value_text(argv[0]);
789 zA = sqlite3_value_text(argv[1]);
790
drhbeb818d2007-05-08 15:34:47 +0000791 /* Limit the length of the LIKE or GLOB pattern to avoid problems
792 ** of deep recursion and N*N behavior in patternCompare().
793 */
drh27e62db2009-04-02 10:16:17 +0000794 nPat = sqlite3_value_bytes(argv[0]);
795 testcase( nPat==db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH] );
796 testcase( nPat==db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH]+1 );
797 if( nPat > db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH] ){
drhbeb818d2007-05-08 15:34:47 +0000798 sqlite3_result_error(context, "LIKE or GLOB pattern too complex", -1);
799 return;
800 }
drh1f0feef2007-05-15 13:27:07 +0000801 assert( zB==sqlite3_value_text(argv[0]) ); /* Encoding did not change */
drhbeb818d2007-05-08 15:34:47 +0000802
danielk19777c6303c2004-11-17 16:41:29 +0000803 if( argc==3 ){
804 /* The escape character string must consist of a single UTF-8 character.
805 ** Otherwise, return an error.
806 */
807 const unsigned char *zEsc = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +0000808 if( zEsc==0 ) return;
drhee858132007-05-08 20:37:38 +0000809 if( sqlite3Utf8CharLen((char*)zEsc, -1)!=1 ){
danielk19777c6303c2004-11-17 16:41:29 +0000810 sqlite3_result_error(context,
811 "ESCAPE expression must be a single character", -1);
812 return;
813 }
drh42610962012-09-17 18:56:32 +0000814 escape = sqlite3Utf8Read(&zEsc);
danielk19777c6303c2004-11-17 16:41:29 +0000815 }
danielk19773f6b0872004-06-17 05:36:44 +0000816 if( zA && zB ){
drh55ef4d92005-08-14 01:20:37 +0000817 struct compareInfo *pInfo = sqlite3_user_data(context);
818#ifdef SQLITE_TEST
819 sqlite3_like_count++;
820#endif
drhbeb818d2007-05-08 15:34:47 +0000821
danielk1977b56fe1f2007-05-09 08:24:44 +0000822 sqlite3_result_int(context, patternCompare(zB, zA, pInfo, escape));
danielk197751ad0ec2004-05-24 12:39:02 +0000823 }
drh8912d102002-05-26 21:34:58 +0000824}
825
826/*
827** Implementation of the NULLIF(x,y) function. The result is the first
828** argument if the arguments are different. The result is NULL if the
829** arguments are equal to each other.
830*/
drhf9b596e2004-05-26 16:54:42 +0000831static void nullifFunc(
832 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000833 int NotUsed,
drhf9b596e2004-05-26 16:54:42 +0000834 sqlite3_value **argv
835){
danielk1977dc1bdc42004-06-11 10:51:27 +0000836 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk197762c14b32008-11-19 09:05:26 +0000837 UNUSED_PARAMETER(NotUsed);
danielk1977dc1bdc42004-06-11 10:51:27 +0000838 if( sqlite3MemCompare(argv[0], argv[1], pColl)!=0 ){
drhf4479502004-05-27 03:12:53 +0000839 sqlite3_result_value(context, argv[0]);
drh8912d102002-05-26 21:34:58 +0000840 }
drh0ac65892002-04-20 14:24:41 +0000841}
842
drh647cb0e2002-11-04 19:32:25 +0000843/*
drh47baebc2009-08-14 16:01:24 +0000844** Implementation of the sqlite_version() function. The result is the version
drh647cb0e2002-11-04 19:32:25 +0000845** of the SQLite library that is running.
846*/
drhf9b596e2004-05-26 16:54:42 +0000847static void versionFunc(
848 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000849 int NotUsed,
850 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000851){
danielk197762c14b32008-11-19 09:05:26 +0000852 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000853 /* IMP: R-48699-48617 This function is an SQL wrapper around the
854 ** sqlite3_libversion() C-interface. */
855 sqlite3_result_text(context, sqlite3_libversion(), -1, SQLITE_STATIC);
drh647cb0e2002-11-04 19:32:25 +0000856}
857
drh47baebc2009-08-14 16:01:24 +0000858/*
859** Implementation of the sqlite_source_id() function. The result is a string
860** that identifies the particular version of the source code used to build
861** SQLite.
862*/
863static void sourceidFunc(
864 sqlite3_context *context,
865 int NotUsed,
866 sqlite3_value **NotUsed2
867){
868 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000869 /* IMP: R-24470-31136 This function is an SQL wrapper around the
870 ** sqlite3_sourceid() C interface. */
871 sqlite3_result_text(context, sqlite3_sourceid(), -1, SQLITE_STATIC);
drh47baebc2009-08-14 16:01:24 +0000872}
873
shanehbdea6d12010-02-23 04:19:54 +0000874/*
drh3ca84ef2011-04-25 18:03:10 +0000875** Implementation of the sqlite_log() function. This is a wrapper around
876** sqlite3_log(). The return value is NULL. The function exists purely for
877** its side-effects.
878*/
drh840561f2011-04-27 18:08:42 +0000879static void errlogFunc(
drh3ca84ef2011-04-25 18:03:10 +0000880 sqlite3_context *context,
881 int argc,
882 sqlite3_value **argv
883){
884 UNUSED_PARAMETER(argc);
885 UNUSED_PARAMETER(context);
886 sqlite3_log(sqlite3_value_int(argv[0]), "%s", sqlite3_value_text(argv[1]));
887}
888
889/*
shanehdc97a8c2010-02-23 20:08:35 +0000890** Implementation of the sqlite_compileoption_used() function.
891** The result is an integer that identifies if the compiler option
892** was used to build SQLite.
shanehbdea6d12010-02-23 04:19:54 +0000893*/
shanehdc97a8c2010-02-23 20:08:35 +0000894#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
895static void compileoptionusedFunc(
shanehbdea6d12010-02-23 04:19:54 +0000896 sqlite3_context *context,
shanehdc97a8c2010-02-23 20:08:35 +0000897 int argc,
898 sqlite3_value **argv
shanehbdea6d12010-02-23 04:19:54 +0000899){
shanehdc97a8c2010-02-23 20:08:35 +0000900 const char *zOptName;
901 assert( argc==1 );
902 UNUSED_PARAMETER(argc);
drha3e414c2010-08-03 18:06:25 +0000903 /* IMP: R-39564-36305 The sqlite_compileoption_used() SQL
904 ** function is a wrapper around the sqlite3_compileoption_used() C/C++
905 ** function.
906 */
drh264a2d42010-02-25 15:28:41 +0000907 if( (zOptName = (const char*)sqlite3_value_text(argv[0]))!=0 ){
shanehdc97a8c2010-02-23 20:08:35 +0000908 sqlite3_result_int(context, sqlite3_compileoption_used(zOptName));
shanehdc97a8c2010-02-23 20:08:35 +0000909 }
shanehbdea6d12010-02-23 04:19:54 +0000910}
shanehdc97a8c2010-02-23 20:08:35 +0000911#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
912
913/*
914** Implementation of the sqlite_compileoption_get() function.
915** The result is a string that identifies the compiler options
916** used to build SQLite.
917*/
918#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
919static void compileoptiongetFunc(
920 sqlite3_context *context,
921 int argc,
922 sqlite3_value **argv
923){
924 int n;
925 assert( argc==1 );
926 UNUSED_PARAMETER(argc);
drha3e414c2010-08-03 18:06:25 +0000927 /* IMP: R-04922-24076 The sqlite_compileoption_get() SQL function
928 ** is a wrapper around the sqlite3_compileoption_get() C/C++ function.
929 */
shanehdc97a8c2010-02-23 20:08:35 +0000930 n = sqlite3_value_int(argv[0]);
931 sqlite3_result_text(context, sqlite3_compileoption_get(n), -1, SQLITE_STATIC);
932}
933#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
shanehbdea6d12010-02-23 04:19:54 +0000934
drh137c7282007-01-29 17:58:28 +0000935/* Array for converting from half-bytes (nybbles) into ASCII hex
936** digits. */
937static const char hexdigits[] = {
938 '0', '1', '2', '3', '4', '5', '6', '7',
939 '8', '9', 'A', 'B', 'C', 'D', 'E', 'F'
940};
danielk1977d641d642004-11-18 15:44:29 +0000941
drh47394702003-08-20 01:03:33 +0000942/*
drh47394702003-08-20 01:03:33 +0000943** Implementation of the QUOTE() function. This function takes a single
944** argument. If the argument is numeric, the return value is the same as
945** the argument. If the argument is NULL, the return value is the string
946** "NULL". Otherwise, the argument is enclosed in single quotes with
947** single-quote escapes.
948*/
danielk19770ae8b832004-05-25 12:05:56 +0000949static void quoteFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drha0df4cc2009-02-02 17:29:59 +0000950 assert( argc==1 );
drh1d34fde2009-02-03 15:50:33 +0000951 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +0000952 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +0000953 case SQLITE_FLOAT: {
drh72b3fbc2012-06-19 03:11:25 +0000954 double r1, r2;
drh72b3fbc2012-06-19 03:11:25 +0000955 char zBuf[50];
mistachkin2b434a72012-06-19 04:36:48 +0000956 r1 = sqlite3_value_double(argv[0]);
drh72b3fbc2012-06-19 03:11:25 +0000957 sqlite3_snprintf(sizeof(zBuf), zBuf, "%!.15g", r1);
958 sqlite3AtoF(zBuf, &r2, 20, SQLITE_UTF8);
959 if( r1!=r2 ){
960 sqlite3_snprintf(sizeof(zBuf), zBuf, "%!.20e", r1);
961 }
962 sqlite3_result_text(context, zBuf, -1, SQLITE_TRANSIENT);
963 break;
964 }
965 case SQLITE_INTEGER: {
drhf4479502004-05-27 03:12:53 +0000966 sqlite3_result_value(context, argv[0]);
drhf9b596e2004-05-26 16:54:42 +0000967 break;
968 }
danielk19773f41e972004-06-08 00:39:01 +0000969 case SQLITE_BLOB: {
danielk19773f41e972004-06-08 00:39:01 +0000970 char *zText = 0;
danielk19773f41e972004-06-08 00:39:01 +0000971 char const *zBlob = sqlite3_value_blob(argv[0]);
drh1f0feef2007-05-15 13:27:07 +0000972 int nBlob = sqlite3_value_bytes(argv[0]);
973 assert( zBlob==sqlite3_value_blob(argv[0]) ); /* No encoding change */
drhb1a6c3c2008-03-20 16:30:17 +0000974 zText = (char *)contextMalloc(context, (2*(i64)nBlob)+4);
danielk1977a1644fd2007-08-29 12:31:25 +0000975 if( zText ){
danielk19773f41e972004-06-08 00:39:01 +0000976 int i;
977 for(i=0; i<nBlob; i++){
978 zText[(i*2)+2] = hexdigits[(zBlob[i]>>4)&0x0F];
979 zText[(i*2)+3] = hexdigits[(zBlob[i])&0x0F];
980 }
981 zText[(nBlob*2)+2] = '\'';
982 zText[(nBlob*2)+3] = '\0';
983 zText[0] = 'X';
984 zText[1] = '\'';
danielk1977d8123362004-06-12 09:25:12 +0000985 sqlite3_result_text(context, zText, -1, SQLITE_TRANSIENT);
drh17435752007-08-16 04:30:38 +0000986 sqlite3_free(zText);
danielk19773f41e972004-06-08 00:39:01 +0000987 }
988 break;
989 }
drh9c054832004-05-31 18:51:57 +0000990 case SQLITE_TEXT: {
drh023ae032007-05-08 12:12:16 +0000991 int i,j;
992 u64 n;
drh2646da72005-12-09 20:02:05 +0000993 const unsigned char *zArg = sqlite3_value_text(argv[0]);
drhf9b596e2004-05-26 16:54:42 +0000994 char *z;
995
drh7a521cf2007-04-25 18:23:52 +0000996 if( zArg==0 ) return;
drh023ae032007-05-08 12:12:16 +0000997 for(i=0, n=0; zArg[i]; i++){ if( zArg[i]=='\'' ) n++; }
drhb1a6c3c2008-03-20 16:30:17 +0000998 z = contextMalloc(context, ((i64)i)+((i64)n)+3);
danielk1977a1644fd2007-08-29 12:31:25 +0000999 if( z ){
1000 z[0] = '\'';
1001 for(i=0, j=1; zArg[i]; i++){
1002 z[j++] = zArg[i];
1003 if( zArg[i]=='\'' ){
1004 z[j++] = '\'';
1005 }
drhf9b596e2004-05-26 16:54:42 +00001006 }
danielk1977a1644fd2007-08-29 12:31:25 +00001007 z[j++] = '\'';
1008 z[j] = 0;
1009 sqlite3_result_text(context, z, j, sqlite3_free);
drhf9b596e2004-05-26 16:54:42 +00001010 }
drha0df4cc2009-02-02 17:29:59 +00001011 break;
1012 }
1013 default: {
1014 assert( sqlite3_value_type(argv[0])==SQLITE_NULL );
1015 sqlite3_result_text(context, "NULL", 4, SQLITE_STATIC);
1016 break;
drhf9b596e2004-05-26 16:54:42 +00001017 }
drh47394702003-08-20 01:03:33 +00001018 }
1019}
1020
drh137c7282007-01-29 17:58:28 +00001021/*
drhd495d8c2013-02-22 19:34:25 +00001022** The unicode() function. Return the integer unicode code-point value
1023** for the first character of the input string.
1024*/
1025static void unicodeFunc(
1026 sqlite3_context *context,
1027 int argc,
1028 sqlite3_value **argv
1029){
1030 const unsigned char *z = sqlite3_value_text(argv[0]);
drh1d59d032013-03-01 23:40:26 +00001031 (void)argc;
drhd495d8c2013-02-22 19:34:25 +00001032 if( z && z[0] ) sqlite3_result_int(context, sqlite3Utf8Read(&z));
1033}
1034
1035/*
1036** The char() function takes zero or more arguments, each of which is
1037** an integer. It constructs a string where each character of the string
1038** is the unicode character for the corresponding integer argument.
1039*/
1040static void charFunc(
1041 sqlite3_context *context,
1042 int argc,
1043 sqlite3_value **argv
1044){
1045 unsigned char *z, *zOut;
1046 int i;
danb72cad12014-03-08 19:07:03 +00001047 zOut = z = sqlite3_malloc( argc*4+1 );
drhd495d8c2013-02-22 19:34:25 +00001048 if( z==0 ){
1049 sqlite3_result_error_nomem(context);
1050 return;
1051 }
1052 for(i=0; i<argc; i++){
mistachkinc9545442013-02-26 05:42:30 +00001053 sqlite3_int64 x;
drhd495d8c2013-02-22 19:34:25 +00001054 unsigned c;
1055 x = sqlite3_value_int64(argv[i]);
1056 if( x<0 || x>0x10ffff ) x = 0xfffd;
1057 c = (unsigned)(x & 0x1fffff);
drhfe7a5d12013-03-07 14:00:04 +00001058 if( c<0x00080 ){
1059 *zOut++ = (u8)(c&0xFF);
1060 }else if( c<0x00800 ){
1061 *zOut++ = 0xC0 + (u8)((c>>6)&0x1F);
1062 *zOut++ = 0x80 + (u8)(c & 0x3F);
1063 }else if( c<0x10000 ){
1064 *zOut++ = 0xE0 + (u8)((c>>12)&0x0F);
1065 *zOut++ = 0x80 + (u8)((c>>6) & 0x3F);
1066 *zOut++ = 0x80 + (u8)(c & 0x3F);
drhd495d8c2013-02-22 19:34:25 +00001067 }else{
drhfe7a5d12013-03-07 14:00:04 +00001068 *zOut++ = 0xF0 + (u8)((c>>18) & 0x07);
1069 *zOut++ = 0x80 + (u8)((c>>12) & 0x3F);
1070 *zOut++ = 0x80 + (u8)((c>>6) & 0x3F);
1071 *zOut++ = 0x80 + (u8)(c & 0x3F);
1072 } \
drhd495d8c2013-02-22 19:34:25 +00001073 }
drhbbf483f2014-09-09 20:30:24 +00001074 sqlite3_result_text64(context, (char*)z, zOut-z, sqlite3_free, SQLITE_UTF8);
drhd495d8c2013-02-22 19:34:25 +00001075}
1076
1077/*
drh137c7282007-01-29 17:58:28 +00001078** The hex() function. Interpret the argument as a blob. Return
1079** a hexadecimal rendering as text.
1080*/
1081static void hexFunc(
1082 sqlite3_context *context,
1083 int argc,
1084 sqlite3_value **argv
1085){
1086 int i, n;
1087 const unsigned char *pBlob;
1088 char *zHex, *z;
1089 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001090 UNUSED_PARAMETER(argc);
drh1f0feef2007-05-15 13:27:07 +00001091 pBlob = sqlite3_value_blob(argv[0]);
drh137c7282007-01-29 17:58:28 +00001092 n = sqlite3_value_bytes(argv[0]);
drh1f0feef2007-05-15 13:27:07 +00001093 assert( pBlob==sqlite3_value_blob(argv[0]) ); /* No encoding change */
drhb1a6c3c2008-03-20 16:30:17 +00001094 z = zHex = contextMalloc(context, ((i64)n)*2 + 1);
danielk1977a1644fd2007-08-29 12:31:25 +00001095 if( zHex ){
1096 for(i=0; i<n; i++, pBlob++){
1097 unsigned char c = *pBlob;
1098 *(z++) = hexdigits[(c>>4)&0xf];
1099 *(z++) = hexdigits[c&0xf];
1100 }
1101 *z = 0;
1102 sqlite3_result_text(context, zHex, n*2, sqlite3_free);
drh137c7282007-01-29 17:58:28 +00001103 }
drh137c7282007-01-29 17:58:28 +00001104}
1105
drh26b6d902007-03-17 13:27:54 +00001106/*
drh8cff3822007-05-02 02:08:28 +00001107** The zeroblob(N) function returns a zero-filled blob of size N bytes.
1108*/
1109static void zeroblobFunc(
1110 sqlite3_context *context,
1111 int argc,
1112 sqlite3_value **argv
1113){
drh98640a32007-06-07 19:08:32 +00001114 i64 n;
drh27e62db2009-04-02 10:16:17 +00001115 sqlite3 *db = sqlite3_context_db_handle(context);
drh8cff3822007-05-02 02:08:28 +00001116 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001117 UNUSED_PARAMETER(argc);
drh98640a32007-06-07 19:08:32 +00001118 n = sqlite3_value_int64(argv[0]);
drh27e62db2009-04-02 10:16:17 +00001119 testcase( n==db->aLimit[SQLITE_LIMIT_LENGTH] );
1120 testcase( n==db->aLimit[SQLITE_LIMIT_LENGTH]+1 );
1121 if( n>db->aLimit[SQLITE_LIMIT_LENGTH] ){
drh98640a32007-06-07 19:08:32 +00001122 sqlite3_result_error_toobig(context);
1123 }else{
drhab2f1f92010-01-11 18:26:42 +00001124 sqlite3_result_zeroblob(context, (int)n); /* IMP: R-00293-64994 */
drh98640a32007-06-07 19:08:32 +00001125 }
drh8cff3822007-05-02 02:08:28 +00001126}
1127
1128/*
drh26b6d902007-03-17 13:27:54 +00001129** The replace() function. Three arguments are all strings: call
1130** them A, B, and C. The result is also a string which is derived
drhf7b54962013-05-28 12:11:54 +00001131** from A by replacing every occurrence of B with C. The match
drh26b6d902007-03-17 13:27:54 +00001132** must be exact. Collating sequences are not used.
1133*/
1134static void replaceFunc(
1135 sqlite3_context *context,
1136 int argc,
1137 sqlite3_value **argv
1138){
1139 const unsigned char *zStr; /* The input string A */
1140 const unsigned char *zPattern; /* The pattern string B */
1141 const unsigned char *zRep; /* The replacement string C */
1142 unsigned char *zOut; /* The output */
1143 int nStr; /* Size of zStr */
1144 int nPattern; /* Size of zPattern */
1145 int nRep; /* Size of zRep */
drh2e6400b2007-05-08 15:46:18 +00001146 i64 nOut; /* Maximum size of zOut */
drh26b6d902007-03-17 13:27:54 +00001147 int loopLimit; /* Last zStr[] that might match zPattern[] */
1148 int i, j; /* Loop counters */
1149
1150 assert( argc==3 );
danielk1977f3d3c272008-11-19 16:52:44 +00001151 UNUSED_PARAMETER(argc);
drh26b6d902007-03-17 13:27:54 +00001152 zStr = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001153 if( zStr==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001154 nStr = sqlite3_value_bytes(argv[0]);
1155 assert( zStr==sqlite3_value_text(argv[0]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001156 zPattern = sqlite3_value_text(argv[1]);
drha605fe82009-02-01 18:08:40 +00001157 if( zPattern==0 ){
drh23336062009-02-03 13:19:12 +00001158 assert( sqlite3_value_type(argv[1])==SQLITE_NULL
1159 || sqlite3_context_db_handle(context)->mallocFailed );
drha605fe82009-02-01 18:08:40 +00001160 return;
1161 }
1162 if( zPattern[0]==0 ){
1163 assert( sqlite3_value_type(argv[1])!=SQLITE_NULL );
1164 sqlite3_result_value(context, argv[0]);
1165 return;
1166 }
drh1f0feef2007-05-15 13:27:07 +00001167 nPattern = sqlite3_value_bytes(argv[1]);
1168 assert( zPattern==sqlite3_value_text(argv[1]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001169 zRep = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +00001170 if( zRep==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001171 nRep = sqlite3_value_bytes(argv[2]);
1172 assert( zRep==sqlite3_value_text(argv[2]) );
drh2e6400b2007-05-08 15:46:18 +00001173 nOut = nStr + 1;
1174 assert( nOut<SQLITE_MAX_LENGTH );
drhb1a6c3c2008-03-20 16:30:17 +00001175 zOut = contextMalloc(context, (i64)nOut);
drh2e6400b2007-05-08 15:46:18 +00001176 if( zOut==0 ){
1177 return;
drh26b6d902007-03-17 13:27:54 +00001178 }
drh26b6d902007-03-17 13:27:54 +00001179 loopLimit = nStr - nPattern;
1180 for(i=j=0; i<=loopLimit; i++){
1181 if( zStr[i]!=zPattern[0] || memcmp(&zStr[i], zPattern, nPattern) ){
1182 zOut[j++] = zStr[i];
1183 }else{
drh4a50aac2007-08-23 02:47:53 +00001184 u8 *zOld;
drhbb4957f2008-03-20 14:03:29 +00001185 sqlite3 *db = sqlite3_context_db_handle(context);
drh2e6400b2007-05-08 15:46:18 +00001186 nOut += nRep - nPattern;
drh27e62db2009-04-02 10:16:17 +00001187 testcase( nOut-1==db->aLimit[SQLITE_LIMIT_LENGTH] );
1188 testcase( nOut-2==db->aLimit[SQLITE_LIMIT_LENGTH] );
1189 if( nOut-1>db->aLimit[SQLITE_LIMIT_LENGTH] ){
drha0206bc2007-05-08 15:15:02 +00001190 sqlite3_result_error_toobig(context);
drhb9755982010-07-24 16:34:37 +00001191 sqlite3_free(zOut);
danielk197717374e82007-05-08 14:39:04 +00001192 return;
1193 }
drh4a50aac2007-08-23 02:47:53 +00001194 zOld = zOut;
drh2e6400b2007-05-08 15:46:18 +00001195 zOut = sqlite3_realloc(zOut, (int)nOut);
1196 if( zOut==0 ){
danielk1977a1644fd2007-08-29 12:31:25 +00001197 sqlite3_result_error_nomem(context);
drhb9755982010-07-24 16:34:37 +00001198 sqlite3_free(zOld);
drh2e6400b2007-05-08 15:46:18 +00001199 return;
1200 }
drh26b6d902007-03-17 13:27:54 +00001201 memcpy(&zOut[j], zRep, nRep);
1202 j += nRep;
1203 i += nPattern-1;
1204 }
1205 }
drh2e6400b2007-05-08 15:46:18 +00001206 assert( j+nStr-i+1==nOut );
drh26b6d902007-03-17 13:27:54 +00001207 memcpy(&zOut[j], &zStr[i], nStr-i);
1208 j += nStr - i;
1209 assert( j<=nOut );
1210 zOut[j] = 0;
1211 sqlite3_result_text(context, (char*)zOut, j, sqlite3_free);
1212}
1213
drh309b3382007-03-17 17:52:42 +00001214/*
1215** Implementation of the TRIM(), LTRIM(), and RTRIM() functions.
1216** The userdata is 0x1 for left trim, 0x2 for right trim, 0x3 for both.
1217*/
1218static void trimFunc(
1219 sqlite3_context *context,
1220 int argc,
1221 sqlite3_value **argv
1222){
1223 const unsigned char *zIn; /* Input string */
1224 const unsigned char *zCharSet; /* Set of characters to trim */
1225 int nIn; /* Number of bytes in input */
drh7209c692008-04-27 18:40:11 +00001226 int flags; /* 1: trimleft 2: trimright 3: trim */
drhd1e3a612007-04-27 21:59:52 +00001227 int i; /* Loop counter */
drh1bd10f82008-12-10 21:19:56 +00001228 unsigned char *aLen = 0; /* Length of each character in zCharSet */
1229 unsigned char **azChar = 0; /* Individual characters in zCharSet */
drhd1e3a612007-04-27 21:59:52 +00001230 int nChar; /* Number of characters in zCharSet */
1231
drh309b3382007-03-17 17:52:42 +00001232 if( sqlite3_value_type(argv[0])==SQLITE_NULL ){
1233 return;
1234 }
1235 zIn = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001236 if( zIn==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001237 nIn = sqlite3_value_bytes(argv[0]);
1238 assert( zIn==sqlite3_value_text(argv[0]) );
drh309b3382007-03-17 17:52:42 +00001239 if( argc==1 ){
drhd1e3a612007-04-27 21:59:52 +00001240 static const unsigned char lenOne[] = { 1 };
danielk1977a4de4532008-09-02 15:44:08 +00001241 static unsigned char * const azOne[] = { (u8*)" " };
drhd1e3a612007-04-27 21:59:52 +00001242 nChar = 1;
drh8cff3822007-05-02 02:08:28 +00001243 aLen = (u8*)lenOne;
danielk1977bc67da42007-12-11 04:23:19 +00001244 azChar = (unsigned char **)azOne;
drhd1e3a612007-04-27 21:59:52 +00001245 zCharSet = 0;
drh7a521cf2007-04-25 18:23:52 +00001246 }else if( (zCharSet = sqlite3_value_text(argv[1]))==0 ){
drh309b3382007-03-17 17:52:42 +00001247 return;
drhd1e3a612007-04-27 21:59:52 +00001248 }else{
1249 const unsigned char *z;
1250 for(z=zCharSet, nChar=0; *z; nChar++){
drh4a919112007-05-15 11:55:09 +00001251 SQLITE_SKIP_UTF8(z);
drhd1e3a612007-04-27 21:59:52 +00001252 }
1253 if( nChar>0 ){
drhb1a6c3c2008-03-20 16:30:17 +00001254 azChar = contextMalloc(context, ((i64)nChar)*(sizeof(char*)+1));
drhd1e3a612007-04-27 21:59:52 +00001255 if( azChar==0 ){
1256 return;
1257 }
1258 aLen = (unsigned char*)&azChar[nChar];
1259 for(z=zCharSet, nChar=0; *z; nChar++){
danielk1977bc67da42007-12-11 04:23:19 +00001260 azChar[nChar] = (unsigned char *)z;
drh4a919112007-05-15 11:55:09 +00001261 SQLITE_SKIP_UTF8(z);
drh3abbd392008-12-10 23:04:13 +00001262 aLen[nChar] = (u8)(z - azChar[nChar]);
drhd1e3a612007-04-27 21:59:52 +00001263 }
1264 }
drh309b3382007-03-17 17:52:42 +00001265 }
drhd1e3a612007-04-27 21:59:52 +00001266 if( nChar>0 ){
shane1fc41292008-07-08 22:28:48 +00001267 flags = SQLITE_PTR_TO_INT(sqlite3_user_data(context));
drh309b3382007-03-17 17:52:42 +00001268 if( flags & 1 ){
drhd1e3a612007-04-27 21:59:52 +00001269 while( nIn>0 ){
drh1bd10f82008-12-10 21:19:56 +00001270 int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001271 for(i=0; i<nChar; i++){
1272 len = aLen[i];
drh27e62db2009-04-02 10:16:17 +00001273 if( len<=nIn && memcmp(zIn, azChar[i], len)==0 ) break;
drhd1e3a612007-04-27 21:59:52 +00001274 }
1275 if( i>=nChar ) break;
1276 zIn += len;
1277 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001278 }
1279 }
1280 if( flags & 2 ){
drhd1e3a612007-04-27 21:59:52 +00001281 while( nIn>0 ){
drh1bd10f82008-12-10 21:19:56 +00001282 int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001283 for(i=0; i<nChar; i++){
1284 len = aLen[i];
1285 if( len<=nIn && memcmp(&zIn[nIn-len],azChar[i],len)==0 ) break;
1286 }
1287 if( i>=nChar ) break;
1288 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001289 }
1290 }
drhd1e3a612007-04-27 21:59:52 +00001291 if( zCharSet ){
1292 sqlite3_free(azChar);
1293 }
drh309b3382007-03-17 17:52:42 +00001294 }
1295 sqlite3_result_text(context, (char*)zIn, nIn, SQLITE_TRANSIENT);
1296}
drh26b6d902007-03-17 13:27:54 +00001297
danielk1977a4de4532008-09-02 15:44:08 +00001298
drh2ba3ccc2009-12-08 02:06:08 +00001299/* IMP: R-25361-16150 This function is omitted from SQLite by default. It
1300** is only available if the SQLITE_SOUNDEX compile-time option is used
1301** when SQLite is built.
1302*/
drhd24cc422003-03-27 12:51:24 +00001303#ifdef SQLITE_SOUNDEX
1304/*
1305** Compute the soundex encoding of a word.
drh2ba3ccc2009-12-08 02:06:08 +00001306**
1307** IMP: R-59782-00072 The soundex(X) function returns a string that is the
1308** soundex encoding of the string X.
drhd24cc422003-03-27 12:51:24 +00001309*/
drh137c7282007-01-29 17:58:28 +00001310static void soundexFunc(
1311 sqlite3_context *context,
1312 int argc,
1313 sqlite3_value **argv
1314){
drhd24cc422003-03-27 12:51:24 +00001315 char zResult[8];
drh4c755c02004-08-08 20:22:17 +00001316 const u8 *zIn;
drhd24cc422003-03-27 12:51:24 +00001317 int i, j;
1318 static const unsigned char iCode[] = {
1319 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1320 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1321 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1322 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1323 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1324 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1325 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1326 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1327 };
1328 assert( argc==1 );
drh4c755c02004-08-08 20:22:17 +00001329 zIn = (u8*)sqlite3_value_text(argv[0]);
drhbdf67e02006-08-19 11:34:01 +00001330 if( zIn==0 ) zIn = (u8*)"";
drhdc86e2b2009-01-24 11:30:42 +00001331 for(i=0; zIn[i] && !sqlite3Isalpha(zIn[i]); i++){}
drhd24cc422003-03-27 12:51:24 +00001332 if( zIn[i] ){
drhbdf67e02006-08-19 11:34:01 +00001333 u8 prevcode = iCode[zIn[i]&0x7f];
danielk197778ca0e72009-01-20 16:53:39 +00001334 zResult[0] = sqlite3Toupper(zIn[i]);
drhd24cc422003-03-27 12:51:24 +00001335 for(j=1; j<4 && zIn[i]; i++){
1336 int code = iCode[zIn[i]&0x7f];
1337 if( code>0 ){
drhbdf67e02006-08-19 11:34:01 +00001338 if( code!=prevcode ){
1339 prevcode = code;
1340 zResult[j++] = code + '0';
1341 }
1342 }else{
1343 prevcode = 0;
drhd24cc422003-03-27 12:51:24 +00001344 }
1345 }
1346 while( j<4 ){
1347 zResult[j++] = '0';
1348 }
1349 zResult[j] = 0;
danielk1977d8123362004-06-12 09:25:12 +00001350 sqlite3_result_text(context, zResult, 4, SQLITE_TRANSIENT);
drhd24cc422003-03-27 12:51:24 +00001351 }else{
drh2ba3ccc2009-12-08 02:06:08 +00001352 /* IMP: R-64894-50321 The string "?000" is returned if the argument
1353 ** is NULL or contains no ASCII alphabetic characters. */
danielk1977d8123362004-06-12 09:25:12 +00001354 sqlite3_result_text(context, "?000", 4, SQLITE_STATIC);
drhd24cc422003-03-27 12:51:24 +00001355 }
1356}
drh2ba3ccc2009-12-08 02:06:08 +00001357#endif /* SQLITE_SOUNDEX */
drhd24cc422003-03-27 12:51:24 +00001358
drhfdb83b22006-06-17 14:12:47 +00001359#ifndef SQLITE_OMIT_LOAD_EXTENSION
1360/*
1361** A function that loads a shared-library extension then returns NULL.
1362*/
1363static void loadExt(sqlite3_context *context, int argc, sqlite3_value **argv){
danielk197765fd59f2006-06-24 11:51:33 +00001364 const char *zFile = (const char *)sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001365 const char *zProc;
drhfa4a4b92008-03-19 21:45:51 +00001366 sqlite3 *db = sqlite3_context_db_handle(context);
drhfdb83b22006-06-17 14:12:47 +00001367 char *zErrMsg = 0;
1368
1369 if( argc==2 ){
danielk197765fd59f2006-06-24 11:51:33 +00001370 zProc = (const char *)sqlite3_value_text(argv[1]);
drh7a521cf2007-04-25 18:23:52 +00001371 }else{
1372 zProc = 0;
drhfdb83b22006-06-17 14:12:47 +00001373 }
drh7a521cf2007-04-25 18:23:52 +00001374 if( zFile && sqlite3_load_extension(db, zFile, zProc, &zErrMsg) ){
drhfdb83b22006-06-17 14:12:47 +00001375 sqlite3_result_error(context, zErrMsg, -1);
1376 sqlite3_free(zErrMsg);
1377 }
1378}
1379#endif
1380
danielk197701427a62005-01-11 13:02:33 +00001381
drh0ac65892002-04-20 14:24:41 +00001382/*
drhd3a149e2002-02-24 17:12:53 +00001383** An instance of the following structure holds the context of a
drhdd5baa92002-02-27 19:50:59 +00001384** sum() or avg() aggregate computation.
1385*/
1386typedef struct SumCtx SumCtx;
1387struct SumCtx {
drh8c08e862006-02-11 17:34:00 +00001388 double rSum; /* Floating point sum */
1389 i64 iSum; /* Integer sum */
1390 i64 cnt; /* Number of elements summed */
1391 u8 overflow; /* True if integer overflow seen */
1392 u8 approx; /* True if non-integer value was input to the sum */
drhdd5baa92002-02-27 19:50:59 +00001393};
1394
1395/*
drha97fdd32006-01-12 22:17:50 +00001396** Routines used to compute the sum, average, and total.
1397**
1398** The SUM() function follows the (broken) SQL standard which means
1399** that it returns NULL if it sums over no inputs. TOTAL returns
1400** 0.0 in that case. In addition, TOTAL always returns a float where
1401** SUM might return an integer if it never encounters a floating point
drhc806d852006-05-11 13:25:39 +00001402** value. TOTAL never fails, but SUM might through an exception if
1403** it overflows an integer.
drhdd5baa92002-02-27 19:50:59 +00001404*/
danielk19770ae8b832004-05-25 12:05:56 +00001405static void sumStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drhdd5baa92002-02-27 19:50:59 +00001406 SumCtx *p;
drh3d1d95e2005-09-08 10:37:01 +00001407 int type;
drh3f219f42005-09-08 19:45:57 +00001408 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001409 UNUSED_PARAMETER(argc);
drh4f26d6c2004-05-26 23:25:30 +00001410 p = sqlite3_aggregate_context(context, sizeof(*p));
drh29d72102006-02-09 22:13:41 +00001411 type = sqlite3_value_numeric_type(argv[0]);
drh3d1d95e2005-09-08 10:37:01 +00001412 if( p && type!=SQLITE_NULL ){
drh739105c2002-05-29 23:22:23 +00001413 p->cnt++;
drh29d72102006-02-09 22:13:41 +00001414 if( type==SQLITE_INTEGER ){
drh8c08e862006-02-11 17:34:00 +00001415 i64 v = sqlite3_value_int64(argv[0]);
1416 p->rSum += v;
drh158b9cb2011-03-05 20:59:46 +00001417 if( (p->approx|p->overflow)==0 && sqlite3AddInt64(&p->iSum, v) ){
1418 p->overflow = 1;
drh29d72102006-02-09 22:13:41 +00001419 }
1420 }else{
drh8c08e862006-02-11 17:34:00 +00001421 p->rSum += sqlite3_value_double(argv[0]);
drh29d72102006-02-09 22:13:41 +00001422 p->approx = 1;
drh3f219f42005-09-08 19:45:57 +00001423 }
drh739105c2002-05-29 23:22:23 +00001424 }
drhdd5baa92002-02-27 19:50:59 +00001425}
danielk19770ae8b832004-05-25 12:05:56 +00001426static void sumFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001427 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001428 p = sqlite3_aggregate_context(context, 0);
drhc2bd9132005-09-08 20:37:43 +00001429 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001430 if( p->overflow ){
1431 sqlite3_result_error(context,"integer overflow",-1);
1432 }else if( p->approx ){
1433 sqlite3_result_double(context, p->rSum);
drhc2bd9132005-09-08 20:37:43 +00001434 }else{
drh8c08e862006-02-11 17:34:00 +00001435 sqlite3_result_int64(context, p->iSum);
drhc2bd9132005-09-08 20:37:43 +00001436 }
drh3d1d95e2005-09-08 10:37:01 +00001437 }
drhdd5baa92002-02-27 19:50:59 +00001438}
danielk19770ae8b832004-05-25 12:05:56 +00001439static void avgFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001440 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001441 p = sqlite3_aggregate_context(context, 0);
drh739105c2002-05-29 23:22:23 +00001442 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001443 sqlite3_result_double(context, p->rSum/(double)p->cnt);
drhdd5baa92002-02-27 19:50:59 +00001444 }
1445}
drha97fdd32006-01-12 22:17:50 +00001446static void totalFinalize(sqlite3_context *context){
1447 SumCtx *p;
1448 p = sqlite3_aggregate_context(context, 0);
shanefbd60f82009-02-04 03:59:25 +00001449 /* (double)0 In case of SQLITE_OMIT_FLOATING_POINT... */
1450 sqlite3_result_double(context, p ? p->rSum : (double)0);
drha97fdd32006-01-12 22:17:50 +00001451}
drhdd5baa92002-02-27 19:50:59 +00001452
1453/*
drh0bce8352002-02-28 00:41:10 +00001454** The following structure keeps track of state information for the
1455** count() aggregate function.
1456*/
1457typedef struct CountCtx CountCtx;
1458struct CountCtx {
drhfc6ad392006-02-09 13:38:19 +00001459 i64 n;
drh0bce8352002-02-28 00:41:10 +00001460};
drhdd5baa92002-02-27 19:50:59 +00001461
drh0bce8352002-02-28 00:41:10 +00001462/*
1463** Routines to implement the count() aggregate function.
1464*/
danielk19770ae8b832004-05-25 12:05:56 +00001465static void countStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drh0bce8352002-02-28 00:41:10 +00001466 CountCtx *p;
drh4f26d6c2004-05-26 23:25:30 +00001467 p = sqlite3_aggregate_context(context, sizeof(*p));
drh9c054832004-05-31 18:51:57 +00001468 if( (argc==0 || SQLITE_NULL!=sqlite3_value_type(argv[0])) && p ){
drh0bce8352002-02-28 00:41:10 +00001469 p->n++;
1470 }
drh2e79c3d2009-04-08 23:04:14 +00001471
drhd3264c72009-04-15 13:39:47 +00001472#ifndef SQLITE_OMIT_DEPRECATED
drh2e79c3d2009-04-08 23:04:14 +00001473 /* The sqlite3_aggregate_count() function is deprecated. But just to make
1474 ** sure it still operates correctly, verify that its count agrees with our
1475 ** internal count when using count(*) and when the total count can be
1476 ** expressed as a 32-bit integer. */
1477 assert( argc==1 || p==0 || p->n>0x7fffffff
1478 || p->n==sqlite3_aggregate_count(context) );
drhd3264c72009-04-15 13:39:47 +00001479#endif
drh0bce8352002-02-28 00:41:10 +00001480}
danielk19770ae8b832004-05-25 12:05:56 +00001481static void countFinalize(sqlite3_context *context){
drh0bce8352002-02-28 00:41:10 +00001482 CountCtx *p;
drhabfcea22005-09-06 20:36:48 +00001483 p = sqlite3_aggregate_context(context, 0);
drhfc6ad392006-02-09 13:38:19 +00001484 sqlite3_result_int64(context, p ? p->n : 0);
drh0bce8352002-02-28 00:41:10 +00001485}
1486
1487/*
drh0bce8352002-02-28 00:41:10 +00001488** Routines to implement min() and max() aggregate functions.
1489*/
danielk197762c14b32008-11-19 09:05:26 +00001490static void minmaxStep(
1491 sqlite3_context *context,
1492 int NotUsed,
1493 sqlite3_value **argv
1494){
danielk197788208052004-05-25 01:13:20 +00001495 Mem *pArg = (Mem *)argv[0];
drh9eb516c2004-07-18 20:52:32 +00001496 Mem *pBest;
danielk197762c14b32008-11-19 09:05:26 +00001497 UNUSED_PARAMETER(NotUsed);
drh9eb516c2004-07-18 20:52:32 +00001498
drh9eb516c2004-07-18 20:52:32 +00001499 pBest = (Mem *)sqlite3_aggregate_context(context, sizeof(*pBest));
danielk19773aeab9e2004-06-24 00:20:04 +00001500 if( !pBest ) return;
drh268380c2004-02-25 13:47:31 +00001501
drh94a6d992012-02-02 18:42:09 +00001502 if( sqlite3_value_type(argv[0])==SQLITE_NULL ){
1503 if( pBest->flags ) sqlite3SkipAccumulatorLoad(context);
1504 }else if( pBest->flags ){
drh9eb516c2004-07-18 20:52:32 +00001505 int max;
1506 int cmp;
danielk1977dc1bdc42004-06-11 10:51:27 +00001507 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk19777e18c252004-05-25 11:47:24 +00001508 /* This step function is used for both the min() and max() aggregates,
1509 ** the only difference between the two being that the sense of the
1510 ** comparison is inverted. For the max() aggregate, the
1511 ** sqlite3_user_data() function returns (void *)-1. For min() it
1512 ** returns (void *)db, where db is the sqlite3* database pointer.
1513 ** Therefore the next statement sets variable 'max' to 1 for the max()
1514 ** aggregate, or 0 for min().
1515 */
drh309b3382007-03-17 17:52:42 +00001516 max = sqlite3_user_data(context)!=0;
danielk1977dc1bdc42004-06-11 10:51:27 +00001517 cmp = sqlite3MemCompare(pBest, pArg, pColl);
danielk197788208052004-05-25 01:13:20 +00001518 if( (max && cmp<0) || (!max && cmp>0) ){
drhb21c8cd2007-08-21 19:33:56 +00001519 sqlite3VdbeMemCopy(pBest, pArg);
drh7a957892012-02-02 17:35:43 +00001520 }else{
1521 sqlite3SkipAccumulatorLoad(context);
danielk197788208052004-05-25 01:13:20 +00001522 }
drh268380c2004-02-25 13:47:31 +00001523 }else{
drh035e5632014-09-16 14:16:31 +00001524 pBest->db = sqlite3_context_db_handle(context);
drhb21c8cd2007-08-21 19:33:56 +00001525 sqlite3VdbeMemCopy(pBest, pArg);
drh0bce8352002-02-28 00:41:10 +00001526 }
1527}
danielk19770ae8b832004-05-25 12:05:56 +00001528static void minMaxFinalize(sqlite3_context *context){
danielk197788208052004-05-25 01:13:20 +00001529 sqlite3_value *pRes;
drhabfcea22005-09-06 20:36:48 +00001530 pRes = (sqlite3_value *)sqlite3_aggregate_context(context, 0);
1531 if( pRes ){
drh94a6d992012-02-02 18:42:09 +00001532 if( pRes->flags ){
drhabfcea22005-09-06 20:36:48 +00001533 sqlite3_result_value(context, pRes);
1534 }
1535 sqlite3VdbeMemRelease(pRes);
drh0bce8352002-02-28 00:41:10 +00001536 }
1537}
drhdd5baa92002-02-27 19:50:59 +00001538
drhb0689692007-11-01 17:38:30 +00001539/*
1540** group_concat(EXPR, ?SEPARATOR?)
1541*/
1542static void groupConcatStep(
1543 sqlite3_context *context,
1544 int argc,
1545 sqlite3_value **argv
1546){
1547 const char *zVal;
drhade86482007-11-28 22:36:40 +00001548 StrAccum *pAccum;
drhb0689692007-11-01 17:38:30 +00001549 const char *zSep;
drh07d31172009-02-02 21:57:05 +00001550 int nVal, nSep;
1551 assert( argc==1 || argc==2 );
1552 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drhade86482007-11-28 22:36:40 +00001553 pAccum = (StrAccum*)sqlite3_aggregate_context(context, sizeof(*pAccum));
1554
1555 if( pAccum ){
drhbb4957f2008-03-20 14:03:29 +00001556 sqlite3 *db = sqlite3_context_db_handle(context);
drh8bfd7192009-06-19 16:44:41 +00001557 int firstTerm = pAccum->useMalloc==0;
drhb9755982010-07-24 16:34:37 +00001558 pAccum->useMalloc = 2;
drhbb4957f2008-03-20 14:03:29 +00001559 pAccum->mxAlloc = db->aLimit[SQLITE_LIMIT_LENGTH];
drh8bfd7192009-06-19 16:44:41 +00001560 if( !firstTerm ){
drh07d31172009-02-02 21:57:05 +00001561 if( argc==2 ){
1562 zSep = (char*)sqlite3_value_text(argv[1]);
1563 nSep = sqlite3_value_bytes(argv[1]);
drhb0689692007-11-01 17:38:30 +00001564 }else{
1565 zSep = ",";
drhade86482007-11-28 22:36:40 +00001566 nSep = 1;
drhb0689692007-11-01 17:38:30 +00001567 }
drha9ab4812013-12-11 11:00:44 +00001568 if( nSep ) sqlite3StrAccumAppend(pAccum, zSep, nSep);
drhb0689692007-11-01 17:38:30 +00001569 }
drh07d31172009-02-02 21:57:05 +00001570 zVal = (char*)sqlite3_value_text(argv[0]);
1571 nVal = sqlite3_value_bytes(argv[0]);
drh8009c9b2014-05-07 20:24:00 +00001572 if( zVal ) sqlite3StrAccumAppend(pAccum, zVal, nVal);
drhb0689692007-11-01 17:38:30 +00001573 }
1574}
1575static void groupConcatFinalize(sqlite3_context *context){
drhade86482007-11-28 22:36:40 +00001576 StrAccum *pAccum;
1577 pAccum = sqlite3_aggregate_context(context, 0);
1578 if( pAccum ){
drhb49bc862013-08-21 21:12:10 +00001579 if( pAccum->accError==STRACCUM_TOOBIG ){
drhade86482007-11-28 22:36:40 +00001580 sqlite3_result_error_toobig(context);
drhb49bc862013-08-21 21:12:10 +00001581 }else if( pAccum->accError==STRACCUM_NOMEM ){
drhade86482007-11-28 22:36:40 +00001582 sqlite3_result_error_nomem(context);
1583 }else{
1584 sqlite3_result_text(context, sqlite3StrAccumFinish(pAccum), -1,
1585 sqlite3_free);
1586 }
drhb0689692007-11-01 17:38:30 +00001587 }
1588}
drh4e5ffc52004-08-31 00:52:37 +00001589
drhd3a149e2002-02-24 17:12:53 +00001590/*
drha4741842010-04-25 20:58:37 +00001591** This routine does per-connection function registration. Most
1592** of the built-in functions above are part of the global function set.
1593** This routine only deals with those that are not global.
drhdc04c582002-02-24 01:55:15 +00001594*/
drh9bb575f2004-09-06 17:24:11 +00001595void sqlite3RegisterBuiltinFunctions(sqlite3 *db){
drha4741842010-04-25 20:58:37 +00001596 int rc = sqlite3_overload_function(db, "MATCH", 2);
1597 assert( rc==SQLITE_NOMEM || rc==SQLITE_OK );
1598 if( rc==SQLITE_NOMEM ){
1599 db->mallocFailed = 1;
danielk1977832a58a2007-06-22 15:21:15 +00001600 }
drh55ef4d92005-08-14 01:20:37 +00001601}
1602
1603/*
1604** Set the LIKEOPT flag on the 2-argument function with the given name.
1605*/
drh1bd10f82008-12-10 21:19:56 +00001606static void setLikeOptFlag(sqlite3 *db, const char *zName, u8 flagVal){
drh55ef4d92005-08-14 01:20:37 +00001607 FuncDef *pDef;
drhea678832008-12-10 19:26:22 +00001608 pDef = sqlite3FindFunction(db, zName, sqlite3Strlen30(zName),
1609 2, SQLITE_UTF8, 0);
drhd27135a2009-04-02 13:36:37 +00001610 if( ALWAYS(pDef) ){
drhd36e1042013-09-06 13:10:12 +00001611 pDef->funcFlags |= flagVal;
drh55ef4d92005-08-14 01:20:37 +00001612 }
1613}
1614
1615/*
1616** Register the built-in LIKE and GLOB functions. The caseSensitive
1617** parameter determines whether or not the LIKE operator is case
1618** sensitive. GLOB is always case sensitive.
1619*/
1620void sqlite3RegisterLikeFunctions(sqlite3 *db, int caseSensitive){
1621 struct compareInfo *pInfo;
1622 if( caseSensitive ){
1623 pInfo = (struct compareInfo*)&likeInfoAlt;
1624 }else{
1625 pInfo = (struct compareInfo*)&likeInfoNorm;
1626 }
drh901e9942010-12-15 18:54:37 +00001627 sqlite3CreateFunc(db, "like", 2, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0);
1628 sqlite3CreateFunc(db, "like", 3, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0);
1629 sqlite3CreateFunc(db, "glob", 2, SQLITE_UTF8,
dand2199f02010-08-27 17:48:52 +00001630 (struct compareInfo*)&globInfo, likeFunc, 0, 0, 0);
drhd64fe2f2005-08-28 17:00:23 +00001631 setLikeOptFlag(db, "glob", SQLITE_FUNC_LIKE | SQLITE_FUNC_CASE);
1632 setLikeOptFlag(db, "like",
1633 caseSensitive ? (SQLITE_FUNC_LIKE | SQLITE_FUNC_CASE) : SQLITE_FUNC_LIKE);
drh55ef4d92005-08-14 01:20:37 +00001634}
1635
1636/*
1637** pExpr points to an expression which implements a function. If
1638** it is appropriate to apply the LIKE optimization to that function
1639** then set aWc[0] through aWc[2] to the wildcard characters and
1640** return TRUE. If the function is not a LIKE-style function then
1641** return FALSE.
1642*/
drhd64fe2f2005-08-28 17:00:23 +00001643int sqlite3IsLikeFunction(sqlite3 *db, Expr *pExpr, int *pIsNocase, char *aWc){
drh55ef4d92005-08-14 01:20:37 +00001644 FuncDef *pDef;
danielk19776ab3a2e2009-02-19 14:39:25 +00001645 if( pExpr->op!=TK_FUNCTION
1646 || !pExpr->x.pList
1647 || pExpr->x.pList->nExpr!=2
1648 ){
drh55ef4d92005-08-14 01:20:37 +00001649 return 0;
1650 }
danielk19776ab3a2e2009-02-19 14:39:25 +00001651 assert( !ExprHasProperty(pExpr, EP_xIsSelect) );
drh33e619f2009-05-28 01:00:55 +00001652 pDef = sqlite3FindFunction(db, pExpr->u.zToken,
1653 sqlite3Strlen30(pExpr->u.zToken),
drhb7916a72009-05-27 10:31:29 +00001654 2, SQLITE_UTF8, 0);
drhd36e1042013-09-06 13:10:12 +00001655 if( NEVER(pDef==0) || (pDef->funcFlags & SQLITE_FUNC_LIKE)==0 ){
drh55ef4d92005-08-14 01:20:37 +00001656 return 0;
1657 }
1658
1659 /* The memcpy() statement assumes that the wildcard characters are
1660 ** the first three statements in the compareInfo structure. The
1661 ** asserts() that follow verify that assumption
1662 */
1663 memcpy(aWc, pDef->pUserData, 3);
1664 assert( (char*)&likeInfoAlt == (char*)&likeInfoAlt.matchAll );
1665 assert( &((char*)&likeInfoAlt)[1] == (char*)&likeInfoAlt.matchOne );
1666 assert( &((char*)&likeInfoAlt)[2] == (char*)&likeInfoAlt.matchSet );
drhd36e1042013-09-06 13:10:12 +00001667 *pIsNocase = (pDef->funcFlags & SQLITE_FUNC_CASE)==0;
drh55ef4d92005-08-14 01:20:37 +00001668 return 1;
drhdc04c582002-02-24 01:55:15 +00001669}
danielk19778c0a7912008-08-20 14:49:23 +00001670
drh70a8ca32008-08-21 18:49:27 +00001671/*
peter.d.reid60ec9142014-09-06 16:39:46 +00001672** All of the FuncDef structures in the aBuiltinFunc[] array above
drh777c5382008-08-21 20:21:34 +00001673** to the global function hash table. This occurs at start-time (as
1674** a consequence of calling sqlite3_initialize()).
1675**
1676** After this routine runs
drh70a8ca32008-08-21 18:49:27 +00001677*/
1678void sqlite3RegisterGlobalFunctions(void){
danielk197793ce7412008-09-01 19:14:02 +00001679 /*
1680 ** The following array holds FuncDef structures for all of the functions
1681 ** defined in this file.
1682 **
1683 ** The array cannot be constant since changes are made to the
1684 ** FuncDef.pHash elements at start-time. The elements of this array
1685 ** are read-only after initialization is complete.
1686 */
1687 static SQLITE_WSD FuncDef aBuiltinFunc[] = {
1688 FUNCTION(ltrim, 1, 1, 0, trimFunc ),
1689 FUNCTION(ltrim, 2, 1, 0, trimFunc ),
1690 FUNCTION(rtrim, 1, 2, 0, trimFunc ),
1691 FUNCTION(rtrim, 2, 2, 0, trimFunc ),
1692 FUNCTION(trim, 1, 3, 0, trimFunc ),
1693 FUNCTION(trim, 2, 3, 0, trimFunc ),
1694 FUNCTION(min, -1, 0, 1, minmaxFunc ),
1695 FUNCTION(min, 0, 0, 1, 0 ),
drh9588ad92014-09-15 14:46:02 +00001696 AGGREGATE2(min, 1, 0, 1, minmaxStep, minMaxFinalize,
1697 SQLITE_FUNC_MINMAX ),
danielk197793ce7412008-09-01 19:14:02 +00001698 FUNCTION(max, -1, 1, 1, minmaxFunc ),
1699 FUNCTION(max, 0, 1, 1, 0 ),
drh9588ad92014-09-15 14:46:02 +00001700 AGGREGATE2(max, 1, 1, 1, minmaxStep, minMaxFinalize,
1701 SQLITE_FUNC_MINMAX ),
drha748fdc2012-03-28 01:34:47 +00001702 FUNCTION2(typeof, 1, 0, 0, typeofFunc, SQLITE_FUNC_TYPEOF),
1703 FUNCTION2(length, 1, 0, 0, lengthFunc, SQLITE_FUNC_LENGTH),
drhd55e0722012-10-25 03:07:29 +00001704 FUNCTION(instr, 2, 0, 0, instrFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001705 FUNCTION(substr, 2, 0, 0, substrFunc ),
1706 FUNCTION(substr, 3, 0, 0, substrFunc ),
drha5c14162013-12-17 15:03:06 +00001707 FUNCTION(printf, -1, 0, 0, printfFunc ),
drhd495d8c2013-02-22 19:34:25 +00001708 FUNCTION(unicode, 1, 0, 0, unicodeFunc ),
1709 FUNCTION(char, -1, 0, 0, charFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001710 FUNCTION(abs, 1, 0, 0, absFunc ),
shanefbd60f82009-02-04 03:59:25 +00001711#ifndef SQLITE_OMIT_FLOATING_POINT
danielk197793ce7412008-09-01 19:14:02 +00001712 FUNCTION(round, 1, 0, 0, roundFunc ),
1713 FUNCTION(round, 2, 0, 0, roundFunc ),
shanefbd60f82009-02-04 03:59:25 +00001714#endif
danielk197793ce7412008-09-01 19:14:02 +00001715 FUNCTION(upper, 1, 0, 0, upperFunc ),
1716 FUNCTION(lower, 1, 0, 0, lowerFunc ),
1717 FUNCTION(coalesce, 1, 0, 0, 0 ),
danielk197793ce7412008-09-01 19:14:02 +00001718 FUNCTION(coalesce, 0, 0, 0, 0 ),
drhcca9f3d2013-09-06 15:23:29 +00001719 FUNCTION2(coalesce, -1, 0, 0, noopFunc, SQLITE_FUNC_COALESCE),
danielk197793ce7412008-09-01 19:14:02 +00001720 FUNCTION(hex, 1, 0, 0, hexFunc ),
drhcca9f3d2013-09-06 15:23:29 +00001721 FUNCTION2(ifnull, 2, 0, 0, noopFunc, SQLITE_FUNC_COALESCE),
1722 FUNCTION2(unlikely, 1, 0, 0, noopFunc, SQLITE_FUNC_UNLIKELY),
drhaae0f9e2013-09-11 11:38:58 +00001723 FUNCTION2(likelihood, 2, 0, 0, noopFunc, SQLITE_FUNC_UNLIKELY),
drh03202a92014-06-17 16:11:28 +00001724 FUNCTION2(likely, 1, 0, 0, noopFunc, SQLITE_FUNC_UNLIKELY),
drhb1fba282013-11-21 14:33:48 +00001725 VFUNCTION(random, 0, 0, 0, randomFunc ),
1726 VFUNCTION(randomblob, 1, 0, 0, randomBlob ),
danielk197793ce7412008-09-01 19:14:02 +00001727 FUNCTION(nullif, 2, 0, 1, nullifFunc ),
1728 FUNCTION(sqlite_version, 0, 0, 0, versionFunc ),
drh47baebc2009-08-14 16:01:24 +00001729 FUNCTION(sqlite_source_id, 0, 0, 0, sourceidFunc ),
drh840561f2011-04-27 18:08:42 +00001730 FUNCTION(sqlite_log, 2, 0, 0, errlogFunc ),
drhf442e332014-09-10 19:01:14 +00001731#if SQLITE_USER_AUTHENTICATION
1732 FUNCTION(sqlite_crypt, 2, 0, 0, sqlite3CryptFunc ),
1733#endif
shanehdc97a8c2010-02-23 20:08:35 +00001734#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
drh8bb76d32010-02-26 16:37:47 +00001735 FUNCTION(sqlite_compileoption_used,1, 0, 0, compileoptionusedFunc ),
1736 FUNCTION(sqlite_compileoption_get, 1, 0, 0, compileoptiongetFunc ),
shanehdc97a8c2010-02-23 20:08:35 +00001737#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
danielk197793ce7412008-09-01 19:14:02 +00001738 FUNCTION(quote, 1, 0, 0, quoteFunc ),
drhb1fba282013-11-21 14:33:48 +00001739 VFUNCTION(last_insert_rowid, 0, 0, 0, last_insert_rowid),
1740 VFUNCTION(changes, 0, 0, 0, changes ),
1741 VFUNCTION(total_changes, 0, 0, 0, total_changes ),
danielk197793ce7412008-09-01 19:14:02 +00001742 FUNCTION(replace, 3, 0, 0, replaceFunc ),
1743 FUNCTION(zeroblob, 1, 0, 0, zeroblobFunc ),
1744 #ifdef SQLITE_SOUNDEX
1745 FUNCTION(soundex, 1, 0, 0, soundexFunc ),
1746 #endif
1747 #ifndef SQLITE_OMIT_LOAD_EXTENSION
1748 FUNCTION(load_extension, 1, 0, 0, loadExt ),
1749 FUNCTION(load_extension, 2, 0, 0, loadExt ),
1750 #endif
1751 AGGREGATE(sum, 1, 0, 0, sumStep, sumFinalize ),
1752 AGGREGATE(total, 1, 0, 0, sumStep, totalFinalize ),
1753 AGGREGATE(avg, 1, 0, 0, sumStep, avgFinalize ),
drh9588ad92014-09-15 14:46:02 +00001754 AGGREGATE2(count, 0, 0, 0, countStep, countFinalize,
1755 SQLITE_FUNC_COUNT ),
danielk197793ce7412008-09-01 19:14:02 +00001756 AGGREGATE(count, 1, 0, 0, countStep, countFinalize ),
drh07d31172009-02-02 21:57:05 +00001757 AGGREGATE(group_concat, 1, 0, 0, groupConcatStep, groupConcatFinalize),
1758 AGGREGATE(group_concat, 2, 0, 0, groupConcatStep, groupConcatFinalize),
danielk197793ce7412008-09-01 19:14:02 +00001759
1760 LIKEFUNC(glob, 2, &globInfo, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
1761 #ifdef SQLITE_CASE_SENSITIVE_LIKE
1762 LIKEFUNC(like, 2, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
1763 LIKEFUNC(like, 3, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
1764 #else
1765 LIKEFUNC(like, 2, &likeInfoNorm, SQLITE_FUNC_LIKE),
1766 LIKEFUNC(like, 3, &likeInfoNorm, SQLITE_FUNC_LIKE),
1767 #endif
1768 };
1769
drh70a8ca32008-08-21 18:49:27 +00001770 int i;
danielk197793ce7412008-09-01 19:14:02 +00001771 FuncDefHash *pHash = &GLOBAL(FuncDefHash, sqlite3GlobalFunctions);
drh106cee52008-09-03 17:11:16 +00001772 FuncDef *aFunc = (FuncDef*)&GLOBAL(FuncDef, aBuiltinFunc);
danielk197793ce7412008-09-01 19:14:02 +00001773
drh70a8ca32008-08-21 18:49:27 +00001774 for(i=0; i<ArraySize(aBuiltinFunc); i++){
danielk197793ce7412008-09-01 19:14:02 +00001775 sqlite3FuncDefInsert(pHash, &aFunc[i]);
drh70a8ca32008-08-21 18:49:27 +00001776 }
drh777c5382008-08-21 20:21:34 +00001777 sqlite3RegisterDateTimeFunctions();
drh545f5872010-04-24 14:02:59 +00001778#ifndef SQLITE_OMIT_ALTERTABLE
1779 sqlite3AlterFunctions();
1780#endif
dan0106e372013-08-12 16:34:32 +00001781#if defined(SQLITE_ENABLE_STAT3) || defined(SQLITE_ENABLE_STAT4)
1782 sqlite3AnalyzeFunctions();
1783#endif
drh70a8ca32008-08-21 18:49:27 +00001784}