blob: 137ed36114e2b8ebd7066d7aad537f6d3a546cda [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*************************************************************************
12** This file contains the C functions that implement various SQL
13** functions of SQLite.
14**
15** There is only one exported symbol in this file - the function
16** sqliteRegisterBuildinFunctions() found at the bottom of the file.
17** All other code has file scope.
drhdc04c582002-02-24 01:55:15 +000018*/
drhb659e9b2005-01-28 01:29:08 +000019#include "sqliteInt.h"
drhd3a149e2002-02-24 17:12:53 +000020#include <stdlib.h>
drh0bce8352002-02-28 00:41:10 +000021#include <assert.h>
danielk197788208052004-05-25 01:13:20 +000022#include "vdbeInt.h"
drh0bce8352002-02-28 00:41:10 +000023
drh55ef4d92005-08-14 01:20:37 +000024/*
25** Return the collating function associated with a function.
26*/
danielk1977dc1bdc42004-06-11 10:51:27 +000027static CollSeq *sqlite3GetFuncCollSeq(sqlite3_context *context){
28 return context->pColl;
29}
30
drh0bce8352002-02-28 00:41:10 +000031/*
drh7a957892012-02-02 17:35:43 +000032** Indicate that the accumulator load should be skipped on this
33** iteration of the aggregate loop.
34*/
35static void sqlite3SkipAccumulatorLoad(sqlite3_context *context){
36 context->skipFlag = 1;
37}
38
39/*
drh0bce8352002-02-28 00:41:10 +000040** Implementation of the non-aggregate min() and max() functions
41*/
drhf9b596e2004-05-26 16:54:42 +000042static void minmaxFunc(
43 sqlite3_context *context,
44 int argc,
45 sqlite3_value **argv
46){
drh0bce8352002-02-28 00:41:10 +000047 int i;
drh268380c2004-02-25 13:47:31 +000048 int mask; /* 0 for min() or 0xffffffff for max() */
drhf9b596e2004-05-26 16:54:42 +000049 int iBest;
danielk1977dc1bdc42004-06-11 10:51:27 +000050 CollSeq *pColl;
drh0bce8352002-02-28 00:41:10 +000051
drh65595cd2009-02-02 16:32:55 +000052 assert( argc>1 );
drhc44af712004-09-02 15:53:56 +000053 mask = sqlite3_user_data(context)==0 ? 0 : -1;
danielk1977dc1bdc42004-06-11 10:51:27 +000054 pColl = sqlite3GetFuncCollSeq(context);
55 assert( pColl );
danielk1977c572ef72004-05-27 09:28:41 +000056 assert( mask==-1 || mask==0 );
drhf9b596e2004-05-26 16:54:42 +000057 iBest = 0;
drh9c054832004-05-31 18:51:57 +000058 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drhf9b596e2004-05-26 16:54:42 +000059 for(i=1; i<argc; i++){
drh9c054832004-05-31 18:51:57 +000060 if( sqlite3_value_type(argv[i])==SQLITE_NULL ) return;
danielk1977dc1bdc42004-06-11 10:51:27 +000061 if( (sqlite3MemCompare(argv[iBest], argv[i], pColl)^mask)>=0 ){
drh65595cd2009-02-02 16:32:55 +000062 testcase( mask==0 );
drhf9b596e2004-05-26 16:54:42 +000063 iBest = i;
drh0bce8352002-02-28 00:41:10 +000064 }
65 }
drhf4479502004-05-27 03:12:53 +000066 sqlite3_result_value(context, argv[iBest]);
drh0bce8352002-02-28 00:41:10 +000067}
drh0bce8352002-02-28 00:41:10 +000068
drh268380c2004-02-25 13:47:31 +000069/*
70** Return the type of the argument.
71*/
drhf9b596e2004-05-26 16:54:42 +000072static void typeofFunc(
73 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +000074 int NotUsed,
drhf9b596e2004-05-26 16:54:42 +000075 sqlite3_value **argv
76){
danielk197735bb9d02004-05-24 12:55:54 +000077 const char *z = 0;
danielk197762c14b32008-11-19 09:05:26 +000078 UNUSED_PARAMETER(NotUsed);
danielk197735bb9d02004-05-24 12:55:54 +000079 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +000080 case SQLITE_INTEGER: z = "integer"; break;
81 case SQLITE_TEXT: z = "text"; break;
82 case SQLITE_FLOAT: z = "real"; break;
83 case SQLITE_BLOB: z = "blob"; break;
drh65595cd2009-02-02 16:32:55 +000084 default: z = "null"; break;
danielk197735bb9d02004-05-24 12:55:54 +000085 }
danielk1977d8123362004-06-12 09:25:12 +000086 sqlite3_result_text(context, z, -1, SQLITE_STATIC);
drh0bce8352002-02-28 00:41:10 +000087}
88
drh5708d2d2005-06-22 10:53:59 +000089
90/*
drh0bce8352002-02-28 00:41:10 +000091** Implementation of the length() function
92*/
drhf9b596e2004-05-26 16:54:42 +000093static void lengthFunc(
94 sqlite3_context *context,
95 int argc,
96 sqlite3_value **argv
97){
drh0bce8352002-02-28 00:41:10 +000098 int len;
99
100 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000101 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +0000102 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +0000103 case SQLITE_BLOB:
104 case SQLITE_INTEGER:
105 case SQLITE_FLOAT: {
drhf4479502004-05-27 03:12:53 +0000106 sqlite3_result_int(context, sqlite3_value_bytes(argv[0]));
drhf9b596e2004-05-26 16:54:42 +0000107 break;
108 }
drh9c054832004-05-31 18:51:57 +0000109 case SQLITE_TEXT: {
drh2646da72005-12-09 20:02:05 +0000110 const unsigned char *z = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +0000111 if( z==0 ) return;
drh4a919112007-05-15 11:55:09 +0000112 len = 0;
113 while( *z ){
114 len++;
115 SQLITE_SKIP_UTF8(z);
116 }
drhf4479502004-05-27 03:12:53 +0000117 sqlite3_result_int(context, len);
drhf9b596e2004-05-26 16:54:42 +0000118 break;
119 }
120 default: {
121 sqlite3_result_null(context);
122 break;
123 }
124 }
drh0bce8352002-02-28 00:41:10 +0000125}
126
127/*
drh2ba3ccc2009-12-08 02:06:08 +0000128** Implementation of the abs() function.
129**
130** IMP: R-23979-26855 The abs(X) function returns the absolute value of
131** the numeric argument X.
drh0bce8352002-02-28 00:41:10 +0000132*/
danielk19770ae8b832004-05-25 12:05:56 +0000133static void absFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drh0bce8352002-02-28 00:41:10 +0000134 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000135 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +0000136 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +0000137 case SQLITE_INTEGER: {
danielk1977f93bbbe2004-05-27 10:30:52 +0000138 i64 iVal = sqlite3_value_int64(argv[0]);
drh52fc8492006-02-23 21:43:55 +0000139 if( iVal<0 ){
140 if( (iVal<<1)==0 ){
drh2ba3ccc2009-12-08 02:06:08 +0000141 /* IMP: R-35460-15084 If X is the integer -9223372036854775807 then
142 ** abs(X) throws an integer overflow error since there is no
143 ** equivalent positive 64-bit two complement value. */
drh52fc8492006-02-23 21:43:55 +0000144 sqlite3_result_error(context, "integer overflow", -1);
145 return;
146 }
147 iVal = -iVal;
148 }
danielk1977f93bbbe2004-05-27 10:30:52 +0000149 sqlite3_result_int64(context, iVal);
drhf9b596e2004-05-26 16:54:42 +0000150 break;
151 }
drh9c054832004-05-31 18:51:57 +0000152 case SQLITE_NULL: {
drh2ba3ccc2009-12-08 02:06:08 +0000153 /* IMP: R-37434-19929 Abs(X) returns NULL if X is NULL. */
drhf9b596e2004-05-26 16:54:42 +0000154 sqlite3_result_null(context);
155 break;
156 }
157 default: {
drh2ba3ccc2009-12-08 02:06:08 +0000158 /* Because sqlite3_value_double() returns 0.0 if the argument is not
159 ** something that can be converted into a number, we have:
160 ** IMP: R-57326-31541 Abs(X) return 0.0 if X is a string or blob that
161 ** cannot be converted to a numeric value.
162 */
danielk1977f93bbbe2004-05-27 10:30:52 +0000163 double rVal = sqlite3_value_double(argv[0]);
drh52fc8492006-02-23 21:43:55 +0000164 if( rVal<0 ) rVal = -rVal;
danielk1977f93bbbe2004-05-27 10:30:52 +0000165 sqlite3_result_double(context, rVal);
drhf9b596e2004-05-26 16:54:42 +0000166 break;
167 }
168 }
drh0bce8352002-02-28 00:41:10 +0000169}
170
171/*
drhf764e6f2007-05-15 01:13:47 +0000172** Implementation of the substr() function.
173**
174** substr(x,p1,p2) returns p2 characters of x[] beginning with p1.
175** p1 is 1-indexed. So substr(x,1,1) returns the first character
176** of x. If x is text, then we actually count UTF-8 characters.
177** If x is a blob, then we count bytes.
178**
179** If p1 is negative, then we begin abs(p1) from the end of x[].
shaneh779b8f12009-11-12 05:04:50 +0000180**
181** If p2 is negative, return the p2 characters preceeding p1.
drh0bce8352002-02-28 00:41:10 +0000182*/
drhf9b596e2004-05-26 16:54:42 +0000183static void substrFunc(
184 sqlite3_context *context,
185 int argc,
186 sqlite3_value **argv
187){
drh2646da72005-12-09 20:02:05 +0000188 const unsigned char *z;
189 const unsigned char *z2;
drh023ae032007-05-08 12:12:16 +0000190 int len;
drhf764e6f2007-05-15 01:13:47 +0000191 int p0type;
drh023ae032007-05-08 12:12:16 +0000192 i64 p1, p2;
drh65595cd2009-02-02 16:32:55 +0000193 int negP2 = 0;
drhf9b596e2004-05-26 16:54:42 +0000194
drh64f31512007-10-12 19:11:55 +0000195 assert( argc==3 || argc==2 );
drh8198d252009-02-01 19:42:37 +0000196 if( sqlite3_value_type(argv[1])==SQLITE_NULL
197 || (argc==3 && sqlite3_value_type(argv[2])==SQLITE_NULL)
198 ){
199 return;
200 }
drhf764e6f2007-05-15 01:13:47 +0000201 p0type = sqlite3_value_type(argv[0]);
drh4adc4cb2009-11-11 20:53:31 +0000202 p1 = sqlite3_value_int(argv[1]);
drhf764e6f2007-05-15 01:13:47 +0000203 if( p0type==SQLITE_BLOB ){
204 len = sqlite3_value_bytes(argv[0]);
205 z = sqlite3_value_blob(argv[0]);
206 if( z==0 ) return;
drh1f0feef2007-05-15 13:27:07 +0000207 assert( len==sqlite3_value_bytes(argv[0]) );
drhf764e6f2007-05-15 01:13:47 +0000208 }else{
209 z = sqlite3_value_text(argv[0]);
210 if( z==0 ) return;
drh4a919112007-05-15 11:55:09 +0000211 len = 0;
drh4adc4cb2009-11-11 20:53:31 +0000212 if( p1<0 ){
213 for(z2=z; *z2; len++){
214 SQLITE_SKIP_UTF8(z2);
215 }
drh4a919112007-05-15 11:55:09 +0000216 }
drhf764e6f2007-05-15 01:13:47 +0000217 }
drh64f31512007-10-12 19:11:55 +0000218 if( argc==3 ){
219 p2 = sqlite3_value_int(argv[2]);
drh65595cd2009-02-02 16:32:55 +0000220 if( p2<0 ){
221 p2 = -p2;
222 negP2 = 1;
223 }
drh64f31512007-10-12 19:11:55 +0000224 }else{
drhbb4957f2008-03-20 14:03:29 +0000225 p2 = sqlite3_context_db_handle(context)->aLimit[SQLITE_LIMIT_LENGTH];
drh64f31512007-10-12 19:11:55 +0000226 }
drh0bce8352002-02-28 00:41:10 +0000227 if( p1<0 ){
drh89425d52002-02-28 03:04:48 +0000228 p1 += len;
drh653bc752002-02-28 03:31:10 +0000229 if( p1<0 ){
230 p2 += p1;
drh65595cd2009-02-02 16:32:55 +0000231 if( p2<0 ) p2 = 0;
drh653bc752002-02-28 03:31:10 +0000232 p1 = 0;
233 }
drh0bce8352002-02-28 00:41:10 +0000234 }else if( p1>0 ){
235 p1--;
drh65595cd2009-02-02 16:32:55 +0000236 }else if( p2>0 ){
237 p2--;
drh0bce8352002-02-28 00:41:10 +0000238 }
drh65595cd2009-02-02 16:32:55 +0000239 if( negP2 ){
240 p1 -= p2;
drh4e79c592009-02-01 19:23:32 +0000241 if( p1<0 ){
242 p2 += p1;
243 p1 = 0;
244 }
245 }
drh65595cd2009-02-02 16:32:55 +0000246 assert( p1>=0 && p2>=0 );
drhf764e6f2007-05-15 01:13:47 +0000247 if( p0type!=SQLITE_BLOB ){
drh4a919112007-05-15 11:55:09 +0000248 while( *z && p1 ){
249 SQLITE_SKIP_UTF8(z);
250 p1--;
drhf764e6f2007-05-15 01:13:47 +0000251 }
drh4a919112007-05-15 11:55:09 +0000252 for(z2=z; *z2 && p2; p2--){
253 SQLITE_SKIP_UTF8(z2);
drhf764e6f2007-05-15 01:13:47 +0000254 }
drh1bd10f82008-12-10 21:19:56 +0000255 sqlite3_result_text(context, (char*)z, (int)(z2-z), SQLITE_TRANSIENT);
drhf764e6f2007-05-15 01:13:47 +0000256 }else{
drh4adc4cb2009-11-11 20:53:31 +0000257 if( p1+p2>len ){
258 p2 = len-p1;
259 if( p2<0 ) p2 = 0;
260 }
drh1bd10f82008-12-10 21:19:56 +0000261 sqlite3_result_blob(context, (char*)&z[p1], (int)p2, SQLITE_TRANSIENT);
drh0bce8352002-02-28 00:41:10 +0000262 }
drh0bce8352002-02-28 00:41:10 +0000263}
264
265/*
266** Implementation of the round() function
267*/
shanefbd60f82009-02-04 03:59:25 +0000268#ifndef SQLITE_OMIT_FLOATING_POINT
danielk19770ae8b832004-05-25 12:05:56 +0000269static void roundFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
danielk197751ad0ec2004-05-24 12:39:02 +0000270 int n = 0;
drh0bce8352002-02-28 00:41:10 +0000271 double r;
drh50d654d2009-06-03 01:24:54 +0000272 char *zBuf;
drh0bce8352002-02-28 00:41:10 +0000273 assert( argc==1 || argc==2 );
danielk197751ad0ec2004-05-24 12:39:02 +0000274 if( argc==2 ){
drh9c054832004-05-31 18:51:57 +0000275 if( SQLITE_NULL==sqlite3_value_type(argv[1]) ) return;
danielk197751ad0ec2004-05-24 12:39:02 +0000276 n = sqlite3_value_int(argv[1]);
277 if( n>30 ) n = 30;
278 if( n<0 ) n = 0;
279 }
drhd589a922006-03-02 03:02:48 +0000280 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drh4f26d6c2004-05-26 23:25:30 +0000281 r = sqlite3_value_double(argv[0]);
shaneh147e1762010-02-17 04:19:27 +0000282 /* If Y==0 and X will fit in a 64-bit int,
283 ** handle the rounding directly,
284 ** otherwise use printf.
285 */
286 if( n==0 && r>=0 && r<LARGEST_INT64-1 ){
287 r = (double)((sqlite_int64)(r+0.5));
288 }else if( n==0 && r<0 && (-r)<LARGEST_INT64-1 ){
289 r = -(double)((sqlite_int64)((-r)+0.5));
drh50d654d2009-06-03 01:24:54 +0000290 }else{
shaneh147e1762010-02-17 04:19:27 +0000291 zBuf = sqlite3_mprintf("%.*f",n,r);
292 if( zBuf==0 ){
293 sqlite3_result_error_nomem(context);
294 return;
295 }
drh9339da12010-09-30 00:50:49 +0000296 sqlite3AtoF(zBuf, &r, sqlite3Strlen30(zBuf), SQLITE_UTF8);
drh50d654d2009-06-03 01:24:54 +0000297 sqlite3_free(zBuf);
drh50d654d2009-06-03 01:24:54 +0000298 }
shaneh147e1762010-02-17 04:19:27 +0000299 sqlite3_result_double(context, r);
drh0bce8352002-02-28 00:41:10 +0000300}
shanefbd60f82009-02-04 03:59:25 +0000301#endif
drhdc04c582002-02-24 01:55:15 +0000302
danielk197726783a52007-08-29 14:06:22 +0000303/*
304** Allocate nByte bytes of space using sqlite3_malloc(). If the
305** allocation fails, call sqlite3_result_error_nomem() to notify
drh27e62db2009-04-02 10:16:17 +0000306** the database handle that malloc() has failed and return NULL.
307** If nByte is larger than the maximum string or blob length, then
308** raise an SQLITE_TOOBIG exception and return NULL.
danielk197726783a52007-08-29 14:06:22 +0000309*/
drhb1a6c3c2008-03-20 16:30:17 +0000310static void *contextMalloc(sqlite3_context *context, i64 nByte){
drhbb4957f2008-03-20 14:03:29 +0000311 char *z;
drh27e62db2009-04-02 10:16:17 +0000312 sqlite3 *db = sqlite3_context_db_handle(context);
drhef31c6a2009-04-02 09:07:12 +0000313 assert( nByte>0 );
drh27e62db2009-04-02 10:16:17 +0000314 testcase( nByte==db->aLimit[SQLITE_LIMIT_LENGTH] );
315 testcase( nByte==db->aLimit[SQLITE_LIMIT_LENGTH]+1 );
316 if( nByte>db->aLimit[SQLITE_LIMIT_LENGTH] ){
drhbb4957f2008-03-20 14:03:29 +0000317 sqlite3_result_error_toobig(context);
318 z = 0;
319 }else{
drh1bd10f82008-12-10 21:19:56 +0000320 z = sqlite3Malloc((int)nByte);
drhef31c6a2009-04-02 09:07:12 +0000321 if( !z ){
drhbb4957f2008-03-20 14:03:29 +0000322 sqlite3_result_error_nomem(context);
323 }
danielk1977a1644fd2007-08-29 12:31:25 +0000324 }
325 return z;
326}
327
drhdc04c582002-02-24 01:55:15 +0000328/*
329** Implementation of the upper() and lower() SQL functions.
330*/
danielk19770ae8b832004-05-25 12:05:56 +0000331static void upperFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drh7a521cf2007-04-25 18:23:52 +0000332 char *z1;
333 const char *z2;
drh9310ef22007-04-27 17:16:20 +0000334 int i, n;
drh1d34fde2009-02-03 15:50:33 +0000335 UNUSED_PARAMETER(argc);
drh7a521cf2007-04-25 18:23:52 +0000336 z2 = (char*)sqlite3_value_text(argv[0]);
drh1f0feef2007-05-15 13:27:07 +0000337 n = sqlite3_value_bytes(argv[0]);
338 /* Verify that the call to _bytes() does not invalidate the _text() pointer */
339 assert( z2==(char*)sqlite3_value_text(argv[0]) );
drh7a521cf2007-04-25 18:23:52 +0000340 if( z2 ){
drhb1a6c3c2008-03-20 16:30:17 +0000341 z1 = contextMalloc(context, ((i64)n)+1);
drh7a521cf2007-04-25 18:23:52 +0000342 if( z1 ){
drhdf901d32011-10-13 18:00:11 +0000343 for(i=0; i<n; i++){
344 z1[i] = (char)sqlite3Toupper(z2[i]);
drh7a521cf2007-04-25 18:23:52 +0000345 }
drhdf901d32011-10-13 18:00:11 +0000346 sqlite3_result_text(context, z1, n, sqlite3_free);
drh7a521cf2007-04-25 18:23:52 +0000347 }
drhdc04c582002-02-24 01:55:15 +0000348 }
349}
danielk19770ae8b832004-05-25 12:05:56 +0000350static void lowerFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drhdf901d32011-10-13 18:00:11 +0000351 char *z1;
drh7a521cf2007-04-25 18:23:52 +0000352 const char *z2;
drh9310ef22007-04-27 17:16:20 +0000353 int i, n;
drh1d34fde2009-02-03 15:50:33 +0000354 UNUSED_PARAMETER(argc);
drh7a521cf2007-04-25 18:23:52 +0000355 z2 = (char*)sqlite3_value_text(argv[0]);
drh1f0feef2007-05-15 13:27:07 +0000356 n = sqlite3_value_bytes(argv[0]);
357 /* Verify that the call to _bytes() does not invalidate the _text() pointer */
358 assert( z2==(char*)sqlite3_value_text(argv[0]) );
drh7a521cf2007-04-25 18:23:52 +0000359 if( z2 ){
drhb1a6c3c2008-03-20 16:30:17 +0000360 z1 = contextMalloc(context, ((i64)n)+1);
drh7a521cf2007-04-25 18:23:52 +0000361 if( z1 ){
drhdf901d32011-10-13 18:00:11 +0000362 for(i=0; i<n; i++){
363 z1[i] = sqlite3Tolower(z2[i]);
drh7a521cf2007-04-25 18:23:52 +0000364 }
drhdf901d32011-10-13 18:00:11 +0000365 sqlite3_result_text(context, z1, n, sqlite3_free);
drh7a521cf2007-04-25 18:23:52 +0000366 }
drhdc04c582002-02-24 01:55:15 +0000367 }
368}
369
drhae6bb952009-11-11 00:24:31 +0000370
371#if 0 /* This function is never used. */
372/*
373** The COALESCE() and IFNULL() functions used to be implemented as shown
374** here. But now they are implemented as VDBE code so that unused arguments
375** do not have to be computed. This legacy implementation is retained as
376** comment.
377*/
drhdc04c582002-02-24 01:55:15 +0000378/*
drhfbc99082002-02-28 03:14:18 +0000379** Implementation of the IFNULL(), NVL(), and COALESCE() functions.
jplyonb6c9e6e2004-01-19 04:53:24 +0000380** All three do the same thing. They return the first non-NULL
381** argument.
drh3212e182002-02-28 00:46:26 +0000382*/
drhf9b596e2004-05-26 16:54:42 +0000383static void ifnullFunc(
384 sqlite3_context *context,
385 int argc,
386 sqlite3_value **argv
387){
drhfbc99082002-02-28 03:14:18 +0000388 int i;
389 for(i=0; i<argc; i++){
drh9c054832004-05-31 18:51:57 +0000390 if( SQLITE_NULL!=sqlite3_value_type(argv[i]) ){
drhf4479502004-05-27 03:12:53 +0000391 sqlite3_result_value(context, argv[i]);
drhfbc99082002-02-28 03:14:18 +0000392 break;
393 }
394 }
drh3212e182002-02-28 00:46:26 +0000395}
drhae6bb952009-11-11 00:24:31 +0000396#endif /* NOT USED */
397#define ifnullFunc versionFunc /* Substitute function - never called */
drh3212e182002-02-28 00:46:26 +0000398
399/*
drhf9ffac92002-03-02 19:00:31 +0000400** Implementation of random(). Return a random integer.
401*/
drhf9b596e2004-05-26 16:54:42 +0000402static void randomFunc(
403 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000404 int NotUsed,
405 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000406){
drh52fc8492006-02-23 21:43:55 +0000407 sqlite_int64 r;
danielk197762c14b32008-11-19 09:05:26 +0000408 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drh2fa18682008-03-19 14:15:34 +0000409 sqlite3_randomness(sizeof(r), &r);
drh3034e3d2009-04-02 14:05:21 +0000410 if( r<0 ){
411 /* We need to prevent a random number of 0x8000000000000000
412 ** (or -9223372036854775808) since when you do abs() of that
413 ** number of you get the same value back again. To do this
414 ** in a way that is testable, mask the sign bit off of negative
415 ** values, resulting in a positive value. Then take the
416 ** 2s complement of that positive value. The end result can
417 ** therefore be no less than -9223372036854775807.
418 */
drhaf8001b2012-02-11 19:53:24 +0000419 r = -(r & LARGEST_INT64);
drh3034e3d2009-04-02 14:05:21 +0000420 }
drh52fc8492006-02-23 21:43:55 +0000421 sqlite3_result_int64(context, r);
drhf9ffac92002-03-02 19:00:31 +0000422}
423
424/*
drh137c7282007-01-29 17:58:28 +0000425** Implementation of randomblob(N). Return a random blob
426** that is N bytes long.
drh63cf66f2007-01-29 15:50:05 +0000427*/
drh137c7282007-01-29 17:58:28 +0000428static void randomBlob(
drh63cf66f2007-01-29 15:50:05 +0000429 sqlite3_context *context,
430 int argc,
431 sqlite3_value **argv
432){
drh137c7282007-01-29 17:58:28 +0000433 int n;
434 unsigned char *p;
drh63cf66f2007-01-29 15:50:05 +0000435 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000436 UNUSED_PARAMETER(argc);
drh63cf66f2007-01-29 15:50:05 +0000437 n = sqlite3_value_int(argv[0]);
drh023ae032007-05-08 12:12:16 +0000438 if( n<1 ){
439 n = 1;
440 }
danielk1977a1644fd2007-08-29 12:31:25 +0000441 p = contextMalloc(context, n);
drh02d85832007-05-07 19:31:15 +0000442 if( p ){
drh2fa18682008-03-19 14:15:34 +0000443 sqlite3_randomness(n, p);
drh17435752007-08-16 04:30:38 +0000444 sqlite3_result_blob(context, (char*)p, n, sqlite3_free);
drh02d85832007-05-07 19:31:15 +0000445 }
drh63cf66f2007-01-29 15:50:05 +0000446}
447
448/*
drh6ed41ad2002-04-06 14:10:47 +0000449** Implementation of the last_insert_rowid() SQL function. The return
danielk197724b03fd2004-05-10 10:34:34 +0000450** value is the same as the sqlite3_last_insert_rowid() API function.
drh6ed41ad2002-04-06 14:10:47 +0000451*/
danielk197751ad0ec2004-05-24 12:39:02 +0000452static void last_insert_rowid(
danielk19770ae8b832004-05-25 12:05:56 +0000453 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000454 int NotUsed,
455 sqlite3_value **NotUsed2
danielk197751ad0ec2004-05-24 12:39:02 +0000456){
drhfa4a4b92008-03-19 21:45:51 +0000457 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000458 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000459 /* IMP: R-51513-12026 The last_insert_rowid() SQL function is a
460 ** wrapper around the sqlite3_last_insert_rowid() C/C++ interface
461 ** function. */
drhf9b596e2004-05-26 16:54:42 +0000462 sqlite3_result_int64(context, sqlite3_last_insert_rowid(db));
drh6ed41ad2002-04-06 14:10:47 +0000463}
464
rdcf146a772004-02-25 22:51:06 +0000465/*
drhab2f1f92010-01-11 18:26:42 +0000466** Implementation of the changes() SQL function.
467**
468** IMP: R-62073-11209 The changes() SQL function is a wrapper
469** around the sqlite3_changes() C/C++ function and hence follows the same
470** rules for counting changes.
rdcf146a772004-02-25 22:51:06 +0000471*/
danielk1977b28af712004-06-21 06:50:26 +0000472static void changes(
drhf9b596e2004-05-26 16:54:42 +0000473 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000474 int NotUsed,
475 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000476){
drhfa4a4b92008-03-19 21:45:51 +0000477 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000478 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhf4479502004-05-27 03:12:53 +0000479 sqlite3_result_int(context, sqlite3_changes(db));
rdcb0c374f2004-02-20 22:53:38 +0000480}
rdcf146a772004-02-25 22:51:06 +0000481
482/*
danielk1977b28af712004-06-21 06:50:26 +0000483** Implementation of the total_changes() SQL function. The return value is
484** the same as the sqlite3_total_changes() API function.
rdcf146a772004-02-25 22:51:06 +0000485*/
danielk1977b28af712004-06-21 06:50:26 +0000486static void total_changes(
487 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000488 int NotUsed,
489 sqlite3_value **NotUsed2
danielk197751ad0ec2004-05-24 12:39:02 +0000490){
drhfa4a4b92008-03-19 21:45:51 +0000491 sqlite3 *db = sqlite3_context_db_handle(context);
danielk197762c14b32008-11-19 09:05:26 +0000492 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000493 /* IMP: R-52756-41993 This function is a wrapper around the
494 ** sqlite3_total_changes() C/C++ interface. */
danielk1977b28af712004-06-21 06:50:26 +0000495 sqlite3_result_int(context, sqlite3_total_changes(db));
rdcb0c374f2004-02-20 22:53:38 +0000496}
497
drh6ed41ad2002-04-06 14:10:47 +0000498/*
drh4e5ffc52004-08-31 00:52:37 +0000499** A structure defining how to do GLOB-style comparisons.
danielk1977d02eb1f2004-06-06 09:44:03 +0000500*/
drh4e5ffc52004-08-31 00:52:37 +0000501struct compareInfo {
502 u8 matchAll;
503 u8 matchOne;
504 u8 matchSet;
505 u8 noCase;
danielk1977d02eb1f2004-06-06 09:44:03 +0000506};
drh55ef4d92005-08-14 01:20:37 +0000507
drhb9175ae2007-12-07 18:39:04 +0000508/*
509** For LIKE and GLOB matching on EBCDIC machines, assume that every
510** character is exactly one byte in size. Also, all characters are
511** able to participate in upper-case-to-lower-case mappings in EBCDIC
512** whereas only characters less than 0x80 do in ASCII.
513*/
514#if defined(SQLITE_EBCDIC)
drh0a32fa62011-06-13 12:19:21 +0000515# define sqlite3Utf8Read(A,C) (*(A++))
516# define GlogUpperToLower(A) A = sqlite3UpperToLower[A]
drhb9175ae2007-12-07 18:39:04 +0000517#else
drh0a32fa62011-06-13 12:19:21 +0000518# define GlogUpperToLower(A) if( !((A)&~0x7f) ){ A = sqlite3UpperToLower[A]; }
drhb9175ae2007-12-07 18:39:04 +0000519#endif
520
drh4e5ffc52004-08-31 00:52:37 +0000521static const struct compareInfo globInfo = { '*', '?', '[', 0 };
drh55ef4d92005-08-14 01:20:37 +0000522/* The correct SQL-92 behavior is for the LIKE operator to ignore
523** case. Thus 'a' LIKE 'A' would be true. */
524static const struct compareInfo likeInfoNorm = { '%', '_', 0, 1 };
525/* If SQLITE_CASE_SENSITIVE_LIKE is defined, then the LIKE operator
526** is case sensitive causing 'a' LIKE 'A' to be false */
527static const struct compareInfo likeInfoAlt = { '%', '_', 0, 0 };
danielk1977d02eb1f2004-06-06 09:44:03 +0000528
529/*
drh4e5ffc52004-08-31 00:52:37 +0000530** Compare two UTF-8 strings for equality where the first string can
531** potentially be a "glob" expression. Return true (1) if they
532** are the same and false (0) if they are different.
drh0ac65892002-04-20 14:24:41 +0000533**
drh4e5ffc52004-08-31 00:52:37 +0000534** Globbing rules:
drh0ac65892002-04-20 14:24:41 +0000535**
drh4e5ffc52004-08-31 00:52:37 +0000536** '*' Matches any sequence of zero or more characters.
danielk1977d02eb1f2004-06-06 09:44:03 +0000537**
drh4e5ffc52004-08-31 00:52:37 +0000538** '?' Matches exactly one character.
539**
540** [...] Matches one character from the enclosed list of
541** characters.
542**
543** [^...] Matches one character not in the enclosed list.
544**
545** With the [...] and [^...] matching, a ']' character can be included
546** in the list by making it the first character after '[' or '^'. A
547** range of characters can be specified using '-'. Example:
548** "[a-z]" matches any single lower-case letter. To match a '-', make
549** it the last character in the list.
550**
551** This routine is usually quick, but can be N**2 in the worst case.
552**
553** Hints: to match '*' or '?', put them in "[]". Like this:
554**
555** abc[*]xyz Matches "abc*xyz" only
drh0ac65892002-04-20 14:24:41 +0000556*/
danielk19777c6303c2004-11-17 16:41:29 +0000557static int patternCompare(
drh4e5ffc52004-08-31 00:52:37 +0000558 const u8 *zPattern, /* The glob pattern */
559 const u8 *zString, /* The string to compare against the glob */
danielk19777c6303c2004-11-17 16:41:29 +0000560 const struct compareInfo *pInfo, /* Information about how to do the compare */
drh0a32fa62011-06-13 12:19:21 +0000561 u32 esc /* The escape character */
danielk197751ad0ec2004-05-24 12:39:02 +0000562){
drh0a32fa62011-06-13 12:19:21 +0000563 u32 c, c2;
drh4e5ffc52004-08-31 00:52:37 +0000564 int invert;
565 int seen;
drh4e5ffc52004-08-31 00:52:37 +0000566 u8 matchOne = pInfo->matchOne;
567 u8 matchAll = pInfo->matchAll;
568 u8 matchSet = pInfo->matchSet;
569 u8 noCase = pInfo->noCase;
danielk19777c6303c2004-11-17 16:41:29 +0000570 int prevEscape = 0; /* True if the previous character was 'escape' */
danielk1977d02eb1f2004-06-06 09:44:03 +0000571
drh769e97e2009-04-01 16:33:37 +0000572 while( (c = sqlite3Utf8Read(zPattern,&zPattern))!=0 ){
danielk19777c6303c2004-11-17 16:41:29 +0000573 if( !prevEscape && c==matchAll ){
drh769e97e2009-04-01 16:33:37 +0000574 while( (c=sqlite3Utf8Read(zPattern,&zPattern)) == matchAll
drh66150952007-07-23 19:12:41 +0000575 || c == matchOne ){
drh769e97e2009-04-01 16:33:37 +0000576 if( c==matchOne && sqlite3Utf8Read(zString, &zString)==0 ){
drh66150952007-07-23 19:12:41 +0000577 return 0;
drh4e5ffc52004-08-31 00:52:37 +0000578 }
danielk1977ad7dd422004-06-06 12:41:49 +0000579 }
drh66150952007-07-23 19:12:41 +0000580 if( c==0 ){
581 return 1;
582 }else if( c==esc ){
drh769e97e2009-04-01 16:33:37 +0000583 c = sqlite3Utf8Read(zPattern, &zPattern);
drh66150952007-07-23 19:12:41 +0000584 if( c==0 ){
585 return 0;
586 }
587 }else if( c==matchSet ){
588 assert( esc==0 ); /* This is GLOB, not LIKE */
589 assert( matchSet<0x80 ); /* '[' is a single-byte character */
590 while( *zString && patternCompare(&zPattern[-1],zString,pInfo,esc)==0 ){
drh4a919112007-05-15 11:55:09 +0000591 SQLITE_SKIP_UTF8(zString);
drh4e5ffc52004-08-31 00:52:37 +0000592 }
593 return *zString!=0;
drh66150952007-07-23 19:12:41 +0000594 }
drh769e97e2009-04-01 16:33:37 +0000595 while( (c2 = sqlite3Utf8Read(zString,&zString))!=0 ){
drh66150952007-07-23 19:12:41 +0000596 if( noCase ){
drh6ed4b782007-12-10 18:07:20 +0000597 GlogUpperToLower(c2);
598 GlogUpperToLower(c);
drh66150952007-07-23 19:12:41 +0000599 while( c2 != 0 && c2 != c ){
drh769e97e2009-04-01 16:33:37 +0000600 c2 = sqlite3Utf8Read(zString, &zString);
drh6ed4b782007-12-10 18:07:20 +0000601 GlogUpperToLower(c2);
drh4e5ffc52004-08-31 00:52:37 +0000602 }
drh66150952007-07-23 19:12:41 +0000603 }else{
604 while( c2 != 0 && c2 != c ){
drh769e97e2009-04-01 16:33:37 +0000605 c2 = sqlite3Utf8Read(zString, &zString);
drh66150952007-07-23 19:12:41 +0000606 }
drh4e5ffc52004-08-31 00:52:37 +0000607 }
drh66150952007-07-23 19:12:41 +0000608 if( c2==0 ) return 0;
609 if( patternCompare(zPattern,zString,pInfo,esc) ) return 1;
610 }
611 return 0;
612 }else if( !prevEscape && c==matchOne ){
drh769e97e2009-04-01 16:33:37 +0000613 if( sqlite3Utf8Read(zString, &zString)==0 ){
drh4e5ffc52004-08-31 00:52:37 +0000614 return 0;
danielk1977d02eb1f2004-06-06 09:44:03 +0000615 }
drh4e5ffc52004-08-31 00:52:37 +0000616 }else if( c==matchSet ){
drh1aa4f3e2011-06-15 12:43:36 +0000617 u32 prior_c = 0;
danielk19777c6303c2004-11-17 16:41:29 +0000618 assert( esc==0 ); /* This only occurs for GLOB, not LIKE */
drh4e5ffc52004-08-31 00:52:37 +0000619 seen = 0;
620 invert = 0;
drh769e97e2009-04-01 16:33:37 +0000621 c = sqlite3Utf8Read(zString, &zString);
drh4e5ffc52004-08-31 00:52:37 +0000622 if( c==0 ) return 0;
drh769e97e2009-04-01 16:33:37 +0000623 c2 = sqlite3Utf8Read(zPattern, &zPattern);
drh66150952007-07-23 19:12:41 +0000624 if( c2=='^' ){
625 invert = 1;
drh769e97e2009-04-01 16:33:37 +0000626 c2 = sqlite3Utf8Read(zPattern, &zPattern);
drh66150952007-07-23 19:12:41 +0000627 }
drh4e5ffc52004-08-31 00:52:37 +0000628 if( c2==']' ){
629 if( c==']' ) seen = 1;
drh769e97e2009-04-01 16:33:37 +0000630 c2 = sqlite3Utf8Read(zPattern, &zPattern);
drh4e5ffc52004-08-31 00:52:37 +0000631 }
drh66150952007-07-23 19:12:41 +0000632 while( c2 && c2!=']' ){
633 if( c2=='-' && zPattern[0]!=']' && zPattern[0]!=0 && prior_c>0 ){
drh769e97e2009-04-01 16:33:37 +0000634 c2 = sqlite3Utf8Read(zPattern, &zPattern);
drh4e5ffc52004-08-31 00:52:37 +0000635 if( c>=prior_c && c<=c2 ) seen = 1;
636 prior_c = 0;
drh4e5ffc52004-08-31 00:52:37 +0000637 }else{
drh66150952007-07-23 19:12:41 +0000638 if( c==c2 ){
639 seen = 1;
640 }
drh4e5ffc52004-08-31 00:52:37 +0000641 prior_c = c2;
642 }
drh769e97e2009-04-01 16:33:37 +0000643 c2 = sqlite3Utf8Read(zPattern, &zPattern);
drh4e5ffc52004-08-31 00:52:37 +0000644 }
drh66150952007-07-23 19:12:41 +0000645 if( c2==0 || (seen ^ invert)==0 ){
646 return 0;
647 }
648 }else if( esc==c && !prevEscape ){
danielk19777c6303c2004-11-17 16:41:29 +0000649 prevEscape = 1;
drh4e5ffc52004-08-31 00:52:37 +0000650 }else{
drh769e97e2009-04-01 16:33:37 +0000651 c2 = sqlite3Utf8Read(zString, &zString);
drh4e5ffc52004-08-31 00:52:37 +0000652 if( noCase ){
drh6ed4b782007-12-10 18:07:20 +0000653 GlogUpperToLower(c);
654 GlogUpperToLower(c2);
drh4e5ffc52004-08-31 00:52:37 +0000655 }
drh66150952007-07-23 19:12:41 +0000656 if( c!=c2 ){
657 return 0;
658 }
danielk19777c6303c2004-11-17 16:41:29 +0000659 prevEscape = 0;
danielk1977d02eb1f2004-06-06 09:44:03 +0000660 }
danielk197751ad0ec2004-05-24 12:39:02 +0000661 }
drh4e5ffc52004-08-31 00:52:37 +0000662 return *zString==0;
drh0ac65892002-04-20 14:24:41 +0000663}
drh4e5ffc52004-08-31 00:52:37 +0000664
drh55ef4d92005-08-14 01:20:37 +0000665/*
666** Count the number of times that the LIKE operator (or GLOB which is
667** just a variation of LIKE) gets called. This is used for testing
668** only.
669*/
670#ifdef SQLITE_TEST
671int sqlite3_like_count = 0;
672#endif
673
danielk19773f6b0872004-06-17 05:36:44 +0000674
675/*
676** Implementation of the like() SQL function. This function implements
677** the build-in LIKE operator. The first argument to the function is the
678** pattern and the second argument is the string. So, the SQL statements:
679**
680** A LIKE B
681**
682** is implemented as like(B,A).
683**
drh55ef4d92005-08-14 01:20:37 +0000684** This same function (with a different compareInfo structure) computes
685** the GLOB operator.
danielk19773f6b0872004-06-17 05:36:44 +0000686*/
687static void likeFunc(
688 sqlite3_context *context,
689 int argc,
690 sqlite3_value **argv
691){
drhbeb818d2007-05-08 15:34:47 +0000692 const unsigned char *zA, *zB;
drh0a32fa62011-06-13 12:19:21 +0000693 u32 escape = 0;
drh27e62db2009-04-02 10:16:17 +0000694 int nPat;
drhbb4957f2008-03-20 14:03:29 +0000695 sqlite3 *db = sqlite3_context_db_handle(context);
drhbeb818d2007-05-08 15:34:47 +0000696
drh1f0feef2007-05-15 13:27:07 +0000697 zB = sqlite3_value_text(argv[0]);
698 zA = sqlite3_value_text(argv[1]);
699
drhbeb818d2007-05-08 15:34:47 +0000700 /* Limit the length of the LIKE or GLOB pattern to avoid problems
701 ** of deep recursion and N*N behavior in patternCompare().
702 */
drh27e62db2009-04-02 10:16:17 +0000703 nPat = sqlite3_value_bytes(argv[0]);
704 testcase( nPat==db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH] );
705 testcase( nPat==db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH]+1 );
706 if( nPat > db->aLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH] ){
drhbeb818d2007-05-08 15:34:47 +0000707 sqlite3_result_error(context, "LIKE or GLOB pattern too complex", -1);
708 return;
709 }
drh1f0feef2007-05-15 13:27:07 +0000710 assert( zB==sqlite3_value_text(argv[0]) ); /* Encoding did not change */
drhbeb818d2007-05-08 15:34:47 +0000711
danielk19777c6303c2004-11-17 16:41:29 +0000712 if( argc==3 ){
713 /* The escape character string must consist of a single UTF-8 character.
714 ** Otherwise, return an error.
715 */
716 const unsigned char *zEsc = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +0000717 if( zEsc==0 ) return;
drhee858132007-05-08 20:37:38 +0000718 if( sqlite3Utf8CharLen((char*)zEsc, -1)!=1 ){
danielk19777c6303c2004-11-17 16:41:29 +0000719 sqlite3_result_error(context,
720 "ESCAPE expression must be a single character", -1);
721 return;
722 }
drh769e97e2009-04-01 16:33:37 +0000723 escape = sqlite3Utf8Read(zEsc, &zEsc);
danielk19777c6303c2004-11-17 16:41:29 +0000724 }
danielk19773f6b0872004-06-17 05:36:44 +0000725 if( zA && zB ){
drh55ef4d92005-08-14 01:20:37 +0000726 struct compareInfo *pInfo = sqlite3_user_data(context);
727#ifdef SQLITE_TEST
728 sqlite3_like_count++;
729#endif
drhbeb818d2007-05-08 15:34:47 +0000730
danielk1977b56fe1f2007-05-09 08:24:44 +0000731 sqlite3_result_int(context, patternCompare(zB, zA, pInfo, escape));
danielk197751ad0ec2004-05-24 12:39:02 +0000732 }
drh8912d102002-05-26 21:34:58 +0000733}
734
735/*
736** Implementation of the NULLIF(x,y) function. The result is the first
737** argument if the arguments are different. The result is NULL if the
738** arguments are equal to each other.
739*/
drhf9b596e2004-05-26 16:54:42 +0000740static void nullifFunc(
741 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000742 int NotUsed,
drhf9b596e2004-05-26 16:54:42 +0000743 sqlite3_value **argv
744){
danielk1977dc1bdc42004-06-11 10:51:27 +0000745 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk197762c14b32008-11-19 09:05:26 +0000746 UNUSED_PARAMETER(NotUsed);
danielk1977dc1bdc42004-06-11 10:51:27 +0000747 if( sqlite3MemCompare(argv[0], argv[1], pColl)!=0 ){
drhf4479502004-05-27 03:12:53 +0000748 sqlite3_result_value(context, argv[0]);
drh8912d102002-05-26 21:34:58 +0000749 }
drh0ac65892002-04-20 14:24:41 +0000750}
751
drh647cb0e2002-11-04 19:32:25 +0000752/*
drh47baebc2009-08-14 16:01:24 +0000753** Implementation of the sqlite_version() function. The result is the version
drh647cb0e2002-11-04 19:32:25 +0000754** of the SQLite library that is running.
755*/
drhf9b596e2004-05-26 16:54:42 +0000756static void versionFunc(
757 sqlite3_context *context,
danielk197762c14b32008-11-19 09:05:26 +0000758 int NotUsed,
759 sqlite3_value **NotUsed2
drhf9b596e2004-05-26 16:54:42 +0000760){
danielk197762c14b32008-11-19 09:05:26 +0000761 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000762 /* IMP: R-48699-48617 This function is an SQL wrapper around the
763 ** sqlite3_libversion() C-interface. */
764 sqlite3_result_text(context, sqlite3_libversion(), -1, SQLITE_STATIC);
drh647cb0e2002-11-04 19:32:25 +0000765}
766
drh47baebc2009-08-14 16:01:24 +0000767/*
768** Implementation of the sqlite_source_id() function. The result is a string
769** that identifies the particular version of the source code used to build
770** SQLite.
771*/
772static void sourceidFunc(
773 sqlite3_context *context,
774 int NotUsed,
775 sqlite3_value **NotUsed2
776){
777 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhab2f1f92010-01-11 18:26:42 +0000778 /* IMP: R-24470-31136 This function is an SQL wrapper around the
779 ** sqlite3_sourceid() C interface. */
780 sqlite3_result_text(context, sqlite3_sourceid(), -1, SQLITE_STATIC);
drh47baebc2009-08-14 16:01:24 +0000781}
782
shanehbdea6d12010-02-23 04:19:54 +0000783/*
drh3ca84ef2011-04-25 18:03:10 +0000784** Implementation of the sqlite_log() function. This is a wrapper around
785** sqlite3_log(). The return value is NULL. The function exists purely for
786** its side-effects.
787*/
drh840561f2011-04-27 18:08:42 +0000788static void errlogFunc(
drh3ca84ef2011-04-25 18:03:10 +0000789 sqlite3_context *context,
790 int argc,
791 sqlite3_value **argv
792){
793 UNUSED_PARAMETER(argc);
794 UNUSED_PARAMETER(context);
795 sqlite3_log(sqlite3_value_int(argv[0]), "%s", sqlite3_value_text(argv[1]));
796}
797
798/*
shanehdc97a8c2010-02-23 20:08:35 +0000799** Implementation of the sqlite_compileoption_used() function.
800** The result is an integer that identifies if the compiler option
801** was used to build SQLite.
shanehbdea6d12010-02-23 04:19:54 +0000802*/
shanehdc97a8c2010-02-23 20:08:35 +0000803#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
804static void compileoptionusedFunc(
shanehbdea6d12010-02-23 04:19:54 +0000805 sqlite3_context *context,
shanehdc97a8c2010-02-23 20:08:35 +0000806 int argc,
807 sqlite3_value **argv
shanehbdea6d12010-02-23 04:19:54 +0000808){
shanehdc97a8c2010-02-23 20:08:35 +0000809 const char *zOptName;
810 assert( argc==1 );
811 UNUSED_PARAMETER(argc);
drha3e414c2010-08-03 18:06:25 +0000812 /* IMP: R-39564-36305 The sqlite_compileoption_used() SQL
813 ** function is a wrapper around the sqlite3_compileoption_used() C/C++
814 ** function.
815 */
drh264a2d42010-02-25 15:28:41 +0000816 if( (zOptName = (const char*)sqlite3_value_text(argv[0]))!=0 ){
shanehdc97a8c2010-02-23 20:08:35 +0000817 sqlite3_result_int(context, sqlite3_compileoption_used(zOptName));
shanehdc97a8c2010-02-23 20:08:35 +0000818 }
shanehbdea6d12010-02-23 04:19:54 +0000819}
shanehdc97a8c2010-02-23 20:08:35 +0000820#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
821
822/*
823** Implementation of the sqlite_compileoption_get() function.
824** The result is a string that identifies the compiler options
825** used to build SQLite.
826*/
827#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
828static void compileoptiongetFunc(
829 sqlite3_context *context,
830 int argc,
831 sqlite3_value **argv
832){
833 int n;
834 assert( argc==1 );
835 UNUSED_PARAMETER(argc);
drha3e414c2010-08-03 18:06:25 +0000836 /* IMP: R-04922-24076 The sqlite_compileoption_get() SQL function
837 ** is a wrapper around the sqlite3_compileoption_get() C/C++ function.
838 */
shanehdc97a8c2010-02-23 20:08:35 +0000839 n = sqlite3_value_int(argv[0]);
840 sqlite3_result_text(context, sqlite3_compileoption_get(n), -1, SQLITE_STATIC);
841}
842#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
shanehbdea6d12010-02-23 04:19:54 +0000843
drh61d2f912012-06-18 14:23:06 +0000844/*
845** 64-bit byte swap on little-endian platforms. No-op on bigenidan.
846*/
847static void byteSwap64(unsigned char *x){
848 if( !SQLITE_BIGENDIAN ){
849 unsigned char c;
850 int i;
851 for(i=0; i<4; i++){
852 c = x[i];
853 x[i] = x[7-i];
854 x[7-i] = c;
855 }
856 }
857}
858
859/*
860** The ieee754() function converts a blob into a REAL. The blob must be
861** an big-endian ieee754 floating point number.
862**
863** Any argument other than an 8-byte blob returns NULL.
864*/
865static void ieee754Func(
866 sqlite3_context *context,
867 int argc,
868 sqlite3_value **argv
869){
870 assert( argc==1 );
871 UNUSED_PARAMETER(argc);
872 if( sqlite3_value_type(argv[0])==SQLITE_BLOB
873 && sqlite3_value_bytes(argv[0])==8
874 ){
875 double r;
876 memcpy(&r, sqlite3_value_blob(argv[0]), 8);
877 byteSwap64((unsigned char*)&r);
878 sqlite3_result_double(context, r);
879 }
880}
881
drh137c7282007-01-29 17:58:28 +0000882/* Array for converting from half-bytes (nybbles) into ASCII hex
883** digits. */
884static const char hexdigits[] = {
885 '0', '1', '2', '3', '4', '5', '6', '7',
886 '8', '9', 'A', 'B', 'C', 'D', 'E', 'F'
887};
danielk1977d641d642004-11-18 15:44:29 +0000888
drh47394702003-08-20 01:03:33 +0000889/*
890** EXPERIMENTAL - This is not an official function. The interface may
891** change. This function may disappear. Do not write code that depends
892** on this function.
893**
894** Implementation of the QUOTE() function. This function takes a single
895** argument. If the argument is numeric, the return value is the same as
896** the argument. If the argument is NULL, the return value is the string
897** "NULL". Otherwise, the argument is enclosed in single quotes with
898** single-quote escapes.
899*/
danielk19770ae8b832004-05-25 12:05:56 +0000900static void quoteFunc(sqlite3_context *context, int argc, sqlite3_value **argv){
drha0df4cc2009-02-02 17:29:59 +0000901 assert( argc==1 );
drh1d34fde2009-02-03 15:50:33 +0000902 UNUSED_PARAMETER(argc);
drhf9b596e2004-05-26 16:54:42 +0000903 switch( sqlite3_value_type(argv[0]) ){
drh9c054832004-05-31 18:51:57 +0000904 case SQLITE_FLOAT: {
drh61d2f912012-06-18 14:23:06 +0000905 union {
906 double r;
907 unsigned char x[8];
908 } v;
909 sqlite3_int64 vi;
910 int i;
911 char zAns[30];
912 v.r = sqlite3_value_double(argv[0]);
913 vi = (sqlite3_int64)v.r;
914 if( v.r==(double)vi ){
915 sqlite3_result_value(context, argv[0]);
916 }else{
917 byteSwap64(v.x);
918 /* 0123456789 123456789 12345678 */
919 memcpy(zAns, "ieee754(X'----------------')", 29);
920 for(i=0; i<8; i++){
921 zAns[10+i*2] = hexdigits[v.x[i]>>4];
922 zAns[11+i*2] = hexdigits[v.x[i]&0xf];
923 }
924 sqlite3_result_text(context, zAns, 28, SQLITE_TRANSIENT);
925 }
926 break;
927 }
928 case SQLITE_INTEGER: {
drhf4479502004-05-27 03:12:53 +0000929 sqlite3_result_value(context, argv[0]);
drhf9b596e2004-05-26 16:54:42 +0000930 break;
931 }
danielk19773f41e972004-06-08 00:39:01 +0000932 case SQLITE_BLOB: {
danielk19773f41e972004-06-08 00:39:01 +0000933 char *zText = 0;
danielk19773f41e972004-06-08 00:39:01 +0000934 char const *zBlob = sqlite3_value_blob(argv[0]);
drh1f0feef2007-05-15 13:27:07 +0000935 int nBlob = sqlite3_value_bytes(argv[0]);
936 assert( zBlob==sqlite3_value_blob(argv[0]) ); /* No encoding change */
drhb1a6c3c2008-03-20 16:30:17 +0000937 zText = (char *)contextMalloc(context, (2*(i64)nBlob)+4);
danielk1977a1644fd2007-08-29 12:31:25 +0000938 if( zText ){
danielk19773f41e972004-06-08 00:39:01 +0000939 int i;
940 for(i=0; i<nBlob; i++){
941 zText[(i*2)+2] = hexdigits[(zBlob[i]>>4)&0x0F];
942 zText[(i*2)+3] = hexdigits[(zBlob[i])&0x0F];
943 }
944 zText[(nBlob*2)+2] = '\'';
945 zText[(nBlob*2)+3] = '\0';
946 zText[0] = 'X';
947 zText[1] = '\'';
danielk1977d8123362004-06-12 09:25:12 +0000948 sqlite3_result_text(context, zText, -1, SQLITE_TRANSIENT);
drh17435752007-08-16 04:30:38 +0000949 sqlite3_free(zText);
danielk19773f41e972004-06-08 00:39:01 +0000950 }
951 break;
952 }
drh9c054832004-05-31 18:51:57 +0000953 case SQLITE_TEXT: {
drh023ae032007-05-08 12:12:16 +0000954 int i,j;
955 u64 n;
drh2646da72005-12-09 20:02:05 +0000956 const unsigned char *zArg = sqlite3_value_text(argv[0]);
drhf9b596e2004-05-26 16:54:42 +0000957 char *z;
958
drh7a521cf2007-04-25 18:23:52 +0000959 if( zArg==0 ) return;
drh023ae032007-05-08 12:12:16 +0000960 for(i=0, n=0; zArg[i]; i++){ if( zArg[i]=='\'' ) n++; }
drhb1a6c3c2008-03-20 16:30:17 +0000961 z = contextMalloc(context, ((i64)i)+((i64)n)+3);
danielk1977a1644fd2007-08-29 12:31:25 +0000962 if( z ){
963 z[0] = '\'';
964 for(i=0, j=1; zArg[i]; i++){
965 z[j++] = zArg[i];
966 if( zArg[i]=='\'' ){
967 z[j++] = '\'';
968 }
drhf9b596e2004-05-26 16:54:42 +0000969 }
danielk1977a1644fd2007-08-29 12:31:25 +0000970 z[j++] = '\'';
971 z[j] = 0;
972 sqlite3_result_text(context, z, j, sqlite3_free);
drhf9b596e2004-05-26 16:54:42 +0000973 }
drha0df4cc2009-02-02 17:29:59 +0000974 break;
975 }
976 default: {
977 assert( sqlite3_value_type(argv[0])==SQLITE_NULL );
978 sqlite3_result_text(context, "NULL", 4, SQLITE_STATIC);
979 break;
drhf9b596e2004-05-26 16:54:42 +0000980 }
drh47394702003-08-20 01:03:33 +0000981 }
982}
983
drh137c7282007-01-29 17:58:28 +0000984/*
985** The hex() function. Interpret the argument as a blob. Return
986** a hexadecimal rendering as text.
987*/
988static void hexFunc(
989 sqlite3_context *context,
990 int argc,
991 sqlite3_value **argv
992){
993 int i, n;
994 const unsigned char *pBlob;
995 char *zHex, *z;
996 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +0000997 UNUSED_PARAMETER(argc);
drh1f0feef2007-05-15 13:27:07 +0000998 pBlob = sqlite3_value_blob(argv[0]);
drh137c7282007-01-29 17:58:28 +0000999 n = sqlite3_value_bytes(argv[0]);
drh1f0feef2007-05-15 13:27:07 +00001000 assert( pBlob==sqlite3_value_blob(argv[0]) ); /* No encoding change */
drhb1a6c3c2008-03-20 16:30:17 +00001001 z = zHex = contextMalloc(context, ((i64)n)*2 + 1);
danielk1977a1644fd2007-08-29 12:31:25 +00001002 if( zHex ){
1003 for(i=0; i<n; i++, pBlob++){
1004 unsigned char c = *pBlob;
1005 *(z++) = hexdigits[(c>>4)&0xf];
1006 *(z++) = hexdigits[c&0xf];
1007 }
1008 *z = 0;
1009 sqlite3_result_text(context, zHex, n*2, sqlite3_free);
drh137c7282007-01-29 17:58:28 +00001010 }
drh137c7282007-01-29 17:58:28 +00001011}
1012
drh26b6d902007-03-17 13:27:54 +00001013/*
drh8cff3822007-05-02 02:08:28 +00001014** The zeroblob(N) function returns a zero-filled blob of size N bytes.
1015*/
1016static void zeroblobFunc(
1017 sqlite3_context *context,
1018 int argc,
1019 sqlite3_value **argv
1020){
drh98640a32007-06-07 19:08:32 +00001021 i64 n;
drh27e62db2009-04-02 10:16:17 +00001022 sqlite3 *db = sqlite3_context_db_handle(context);
drh8cff3822007-05-02 02:08:28 +00001023 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001024 UNUSED_PARAMETER(argc);
drh98640a32007-06-07 19:08:32 +00001025 n = sqlite3_value_int64(argv[0]);
drh27e62db2009-04-02 10:16:17 +00001026 testcase( n==db->aLimit[SQLITE_LIMIT_LENGTH] );
1027 testcase( n==db->aLimit[SQLITE_LIMIT_LENGTH]+1 );
1028 if( n>db->aLimit[SQLITE_LIMIT_LENGTH] ){
drh98640a32007-06-07 19:08:32 +00001029 sqlite3_result_error_toobig(context);
1030 }else{
drhab2f1f92010-01-11 18:26:42 +00001031 sqlite3_result_zeroblob(context, (int)n); /* IMP: R-00293-64994 */
drh98640a32007-06-07 19:08:32 +00001032 }
drh8cff3822007-05-02 02:08:28 +00001033}
1034
1035/*
drh26b6d902007-03-17 13:27:54 +00001036** The replace() function. Three arguments are all strings: call
1037** them A, B, and C. The result is also a string which is derived
1038** from A by replacing every occurance of B with C. The match
1039** must be exact. Collating sequences are not used.
1040*/
1041static void replaceFunc(
1042 sqlite3_context *context,
1043 int argc,
1044 sqlite3_value **argv
1045){
1046 const unsigned char *zStr; /* The input string A */
1047 const unsigned char *zPattern; /* The pattern string B */
1048 const unsigned char *zRep; /* The replacement string C */
1049 unsigned char *zOut; /* The output */
1050 int nStr; /* Size of zStr */
1051 int nPattern; /* Size of zPattern */
1052 int nRep; /* Size of zRep */
drh2e6400b2007-05-08 15:46:18 +00001053 i64 nOut; /* Maximum size of zOut */
drh26b6d902007-03-17 13:27:54 +00001054 int loopLimit; /* Last zStr[] that might match zPattern[] */
1055 int i, j; /* Loop counters */
1056
1057 assert( argc==3 );
danielk1977f3d3c272008-11-19 16:52:44 +00001058 UNUSED_PARAMETER(argc);
drh26b6d902007-03-17 13:27:54 +00001059 zStr = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001060 if( zStr==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001061 nStr = sqlite3_value_bytes(argv[0]);
1062 assert( zStr==sqlite3_value_text(argv[0]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001063 zPattern = sqlite3_value_text(argv[1]);
drha605fe82009-02-01 18:08:40 +00001064 if( zPattern==0 ){
drh23336062009-02-03 13:19:12 +00001065 assert( sqlite3_value_type(argv[1])==SQLITE_NULL
1066 || sqlite3_context_db_handle(context)->mallocFailed );
drha605fe82009-02-01 18:08:40 +00001067 return;
1068 }
1069 if( zPattern[0]==0 ){
1070 assert( sqlite3_value_type(argv[1])!=SQLITE_NULL );
1071 sqlite3_result_value(context, argv[0]);
1072 return;
1073 }
drh1f0feef2007-05-15 13:27:07 +00001074 nPattern = sqlite3_value_bytes(argv[1]);
1075 assert( zPattern==sqlite3_value_text(argv[1]) ); /* No encoding change */
drh26b6d902007-03-17 13:27:54 +00001076 zRep = sqlite3_value_text(argv[2]);
drh7a521cf2007-04-25 18:23:52 +00001077 if( zRep==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001078 nRep = sqlite3_value_bytes(argv[2]);
1079 assert( zRep==sqlite3_value_text(argv[2]) );
drh2e6400b2007-05-08 15:46:18 +00001080 nOut = nStr + 1;
1081 assert( nOut<SQLITE_MAX_LENGTH );
drhb1a6c3c2008-03-20 16:30:17 +00001082 zOut = contextMalloc(context, (i64)nOut);
drh2e6400b2007-05-08 15:46:18 +00001083 if( zOut==0 ){
1084 return;
drh26b6d902007-03-17 13:27:54 +00001085 }
drh26b6d902007-03-17 13:27:54 +00001086 loopLimit = nStr - nPattern;
1087 for(i=j=0; i<=loopLimit; i++){
1088 if( zStr[i]!=zPattern[0] || memcmp(&zStr[i], zPattern, nPattern) ){
1089 zOut[j++] = zStr[i];
1090 }else{
drh4a50aac2007-08-23 02:47:53 +00001091 u8 *zOld;
drhbb4957f2008-03-20 14:03:29 +00001092 sqlite3 *db = sqlite3_context_db_handle(context);
drh2e6400b2007-05-08 15:46:18 +00001093 nOut += nRep - nPattern;
drh27e62db2009-04-02 10:16:17 +00001094 testcase( nOut-1==db->aLimit[SQLITE_LIMIT_LENGTH] );
1095 testcase( nOut-2==db->aLimit[SQLITE_LIMIT_LENGTH] );
1096 if( nOut-1>db->aLimit[SQLITE_LIMIT_LENGTH] ){
drha0206bc2007-05-08 15:15:02 +00001097 sqlite3_result_error_toobig(context);
drhb9755982010-07-24 16:34:37 +00001098 sqlite3_free(zOut);
danielk197717374e82007-05-08 14:39:04 +00001099 return;
1100 }
drh4a50aac2007-08-23 02:47:53 +00001101 zOld = zOut;
drh2e6400b2007-05-08 15:46:18 +00001102 zOut = sqlite3_realloc(zOut, (int)nOut);
1103 if( zOut==0 ){
danielk1977a1644fd2007-08-29 12:31:25 +00001104 sqlite3_result_error_nomem(context);
drhb9755982010-07-24 16:34:37 +00001105 sqlite3_free(zOld);
drh2e6400b2007-05-08 15:46:18 +00001106 return;
1107 }
drh26b6d902007-03-17 13:27:54 +00001108 memcpy(&zOut[j], zRep, nRep);
1109 j += nRep;
1110 i += nPattern-1;
1111 }
1112 }
drh2e6400b2007-05-08 15:46:18 +00001113 assert( j+nStr-i+1==nOut );
drh26b6d902007-03-17 13:27:54 +00001114 memcpy(&zOut[j], &zStr[i], nStr-i);
1115 j += nStr - i;
1116 assert( j<=nOut );
1117 zOut[j] = 0;
1118 sqlite3_result_text(context, (char*)zOut, j, sqlite3_free);
1119}
1120
drh309b3382007-03-17 17:52:42 +00001121/*
1122** Implementation of the TRIM(), LTRIM(), and RTRIM() functions.
1123** The userdata is 0x1 for left trim, 0x2 for right trim, 0x3 for both.
1124*/
1125static void trimFunc(
1126 sqlite3_context *context,
1127 int argc,
1128 sqlite3_value **argv
1129){
1130 const unsigned char *zIn; /* Input string */
1131 const unsigned char *zCharSet; /* Set of characters to trim */
1132 int nIn; /* Number of bytes in input */
drh7209c692008-04-27 18:40:11 +00001133 int flags; /* 1: trimleft 2: trimright 3: trim */
drhd1e3a612007-04-27 21:59:52 +00001134 int i; /* Loop counter */
drh1bd10f82008-12-10 21:19:56 +00001135 unsigned char *aLen = 0; /* Length of each character in zCharSet */
1136 unsigned char **azChar = 0; /* Individual characters in zCharSet */
drhd1e3a612007-04-27 21:59:52 +00001137 int nChar; /* Number of characters in zCharSet */
1138
drh309b3382007-03-17 17:52:42 +00001139 if( sqlite3_value_type(argv[0])==SQLITE_NULL ){
1140 return;
1141 }
1142 zIn = sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001143 if( zIn==0 ) return;
drh1f0feef2007-05-15 13:27:07 +00001144 nIn = sqlite3_value_bytes(argv[0]);
1145 assert( zIn==sqlite3_value_text(argv[0]) );
drh309b3382007-03-17 17:52:42 +00001146 if( argc==1 ){
drhd1e3a612007-04-27 21:59:52 +00001147 static const unsigned char lenOne[] = { 1 };
danielk1977a4de4532008-09-02 15:44:08 +00001148 static unsigned char * const azOne[] = { (u8*)" " };
drhd1e3a612007-04-27 21:59:52 +00001149 nChar = 1;
drh8cff3822007-05-02 02:08:28 +00001150 aLen = (u8*)lenOne;
danielk1977bc67da42007-12-11 04:23:19 +00001151 azChar = (unsigned char **)azOne;
drhd1e3a612007-04-27 21:59:52 +00001152 zCharSet = 0;
drh7a521cf2007-04-25 18:23:52 +00001153 }else if( (zCharSet = sqlite3_value_text(argv[1]))==0 ){
drh309b3382007-03-17 17:52:42 +00001154 return;
drhd1e3a612007-04-27 21:59:52 +00001155 }else{
1156 const unsigned char *z;
1157 for(z=zCharSet, nChar=0; *z; nChar++){
drh4a919112007-05-15 11:55:09 +00001158 SQLITE_SKIP_UTF8(z);
drhd1e3a612007-04-27 21:59:52 +00001159 }
1160 if( nChar>0 ){
drhb1a6c3c2008-03-20 16:30:17 +00001161 azChar = contextMalloc(context, ((i64)nChar)*(sizeof(char*)+1));
drhd1e3a612007-04-27 21:59:52 +00001162 if( azChar==0 ){
1163 return;
1164 }
1165 aLen = (unsigned char*)&azChar[nChar];
1166 for(z=zCharSet, nChar=0; *z; nChar++){
danielk1977bc67da42007-12-11 04:23:19 +00001167 azChar[nChar] = (unsigned char *)z;
drh4a919112007-05-15 11:55:09 +00001168 SQLITE_SKIP_UTF8(z);
drh3abbd392008-12-10 23:04:13 +00001169 aLen[nChar] = (u8)(z - azChar[nChar]);
drhd1e3a612007-04-27 21:59:52 +00001170 }
1171 }
drh309b3382007-03-17 17:52:42 +00001172 }
drhd1e3a612007-04-27 21:59:52 +00001173 if( nChar>0 ){
shane1fc41292008-07-08 22:28:48 +00001174 flags = SQLITE_PTR_TO_INT(sqlite3_user_data(context));
drh309b3382007-03-17 17:52:42 +00001175 if( flags & 1 ){
drhd1e3a612007-04-27 21:59:52 +00001176 while( nIn>0 ){
drh1bd10f82008-12-10 21:19:56 +00001177 int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001178 for(i=0; i<nChar; i++){
1179 len = aLen[i];
drh27e62db2009-04-02 10:16:17 +00001180 if( len<=nIn && memcmp(zIn, azChar[i], len)==0 ) break;
drhd1e3a612007-04-27 21:59:52 +00001181 }
1182 if( i>=nChar ) break;
1183 zIn += len;
1184 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001185 }
1186 }
1187 if( flags & 2 ){
drhd1e3a612007-04-27 21:59:52 +00001188 while( nIn>0 ){
drh1bd10f82008-12-10 21:19:56 +00001189 int len = 0;
drhd1e3a612007-04-27 21:59:52 +00001190 for(i=0; i<nChar; i++){
1191 len = aLen[i];
1192 if( len<=nIn && memcmp(&zIn[nIn-len],azChar[i],len)==0 ) break;
1193 }
1194 if( i>=nChar ) break;
1195 nIn -= len;
drh309b3382007-03-17 17:52:42 +00001196 }
1197 }
drhd1e3a612007-04-27 21:59:52 +00001198 if( zCharSet ){
1199 sqlite3_free(azChar);
1200 }
drh309b3382007-03-17 17:52:42 +00001201 }
1202 sqlite3_result_text(context, (char*)zIn, nIn, SQLITE_TRANSIENT);
1203}
drh26b6d902007-03-17 13:27:54 +00001204
danielk1977a4de4532008-09-02 15:44:08 +00001205
drh2ba3ccc2009-12-08 02:06:08 +00001206/* IMP: R-25361-16150 This function is omitted from SQLite by default. It
1207** is only available if the SQLITE_SOUNDEX compile-time option is used
1208** when SQLite is built.
1209*/
drhd24cc422003-03-27 12:51:24 +00001210#ifdef SQLITE_SOUNDEX
1211/*
1212** Compute the soundex encoding of a word.
drh2ba3ccc2009-12-08 02:06:08 +00001213**
1214** IMP: R-59782-00072 The soundex(X) function returns a string that is the
1215** soundex encoding of the string X.
drhd24cc422003-03-27 12:51:24 +00001216*/
drh137c7282007-01-29 17:58:28 +00001217static void soundexFunc(
1218 sqlite3_context *context,
1219 int argc,
1220 sqlite3_value **argv
1221){
drhd24cc422003-03-27 12:51:24 +00001222 char zResult[8];
drh4c755c02004-08-08 20:22:17 +00001223 const u8 *zIn;
drhd24cc422003-03-27 12:51:24 +00001224 int i, j;
1225 static const unsigned char iCode[] = {
1226 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1227 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1228 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1229 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1230 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1231 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1232 0, 0, 1, 2, 3, 0, 1, 2, 0, 0, 2, 2, 4, 5, 5, 0,
1233 1, 2, 6, 2, 3, 0, 1, 0, 2, 0, 2, 0, 0, 0, 0, 0,
1234 };
1235 assert( argc==1 );
drh4c755c02004-08-08 20:22:17 +00001236 zIn = (u8*)sqlite3_value_text(argv[0]);
drhbdf67e02006-08-19 11:34:01 +00001237 if( zIn==0 ) zIn = (u8*)"";
drhdc86e2b2009-01-24 11:30:42 +00001238 for(i=0; zIn[i] && !sqlite3Isalpha(zIn[i]); i++){}
drhd24cc422003-03-27 12:51:24 +00001239 if( zIn[i] ){
drhbdf67e02006-08-19 11:34:01 +00001240 u8 prevcode = iCode[zIn[i]&0x7f];
danielk197778ca0e72009-01-20 16:53:39 +00001241 zResult[0] = sqlite3Toupper(zIn[i]);
drhd24cc422003-03-27 12:51:24 +00001242 for(j=1; j<4 && zIn[i]; i++){
1243 int code = iCode[zIn[i]&0x7f];
1244 if( code>0 ){
drhbdf67e02006-08-19 11:34:01 +00001245 if( code!=prevcode ){
1246 prevcode = code;
1247 zResult[j++] = code + '0';
1248 }
1249 }else{
1250 prevcode = 0;
drhd24cc422003-03-27 12:51:24 +00001251 }
1252 }
1253 while( j<4 ){
1254 zResult[j++] = '0';
1255 }
1256 zResult[j] = 0;
danielk1977d8123362004-06-12 09:25:12 +00001257 sqlite3_result_text(context, zResult, 4, SQLITE_TRANSIENT);
drhd24cc422003-03-27 12:51:24 +00001258 }else{
drh2ba3ccc2009-12-08 02:06:08 +00001259 /* IMP: R-64894-50321 The string "?000" is returned if the argument
1260 ** is NULL or contains no ASCII alphabetic characters. */
danielk1977d8123362004-06-12 09:25:12 +00001261 sqlite3_result_text(context, "?000", 4, SQLITE_STATIC);
drhd24cc422003-03-27 12:51:24 +00001262 }
1263}
drh2ba3ccc2009-12-08 02:06:08 +00001264#endif /* SQLITE_SOUNDEX */
drhd24cc422003-03-27 12:51:24 +00001265
drhfdb83b22006-06-17 14:12:47 +00001266#ifndef SQLITE_OMIT_LOAD_EXTENSION
1267/*
1268** A function that loads a shared-library extension then returns NULL.
1269*/
1270static void loadExt(sqlite3_context *context, int argc, sqlite3_value **argv){
danielk197765fd59f2006-06-24 11:51:33 +00001271 const char *zFile = (const char *)sqlite3_value_text(argv[0]);
drh7a521cf2007-04-25 18:23:52 +00001272 const char *zProc;
drhfa4a4b92008-03-19 21:45:51 +00001273 sqlite3 *db = sqlite3_context_db_handle(context);
drhfdb83b22006-06-17 14:12:47 +00001274 char *zErrMsg = 0;
1275
1276 if( argc==2 ){
danielk197765fd59f2006-06-24 11:51:33 +00001277 zProc = (const char *)sqlite3_value_text(argv[1]);
drh7a521cf2007-04-25 18:23:52 +00001278 }else{
1279 zProc = 0;
drhfdb83b22006-06-17 14:12:47 +00001280 }
drh7a521cf2007-04-25 18:23:52 +00001281 if( zFile && sqlite3_load_extension(db, zFile, zProc, &zErrMsg) ){
drhfdb83b22006-06-17 14:12:47 +00001282 sqlite3_result_error(context, zErrMsg, -1);
1283 sqlite3_free(zErrMsg);
1284 }
1285}
1286#endif
1287
danielk197701427a62005-01-11 13:02:33 +00001288
drh0ac65892002-04-20 14:24:41 +00001289/*
drhd3a149e2002-02-24 17:12:53 +00001290** An instance of the following structure holds the context of a
drhdd5baa92002-02-27 19:50:59 +00001291** sum() or avg() aggregate computation.
1292*/
1293typedef struct SumCtx SumCtx;
1294struct SumCtx {
drh8c08e862006-02-11 17:34:00 +00001295 double rSum; /* Floating point sum */
1296 i64 iSum; /* Integer sum */
1297 i64 cnt; /* Number of elements summed */
1298 u8 overflow; /* True if integer overflow seen */
1299 u8 approx; /* True if non-integer value was input to the sum */
drhdd5baa92002-02-27 19:50:59 +00001300};
1301
1302/*
drha97fdd32006-01-12 22:17:50 +00001303** Routines used to compute the sum, average, and total.
1304**
1305** The SUM() function follows the (broken) SQL standard which means
1306** that it returns NULL if it sums over no inputs. TOTAL returns
1307** 0.0 in that case. In addition, TOTAL always returns a float where
1308** SUM might return an integer if it never encounters a floating point
drhc806d852006-05-11 13:25:39 +00001309** value. TOTAL never fails, but SUM might through an exception if
1310** it overflows an integer.
drhdd5baa92002-02-27 19:50:59 +00001311*/
danielk19770ae8b832004-05-25 12:05:56 +00001312static void sumStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drhdd5baa92002-02-27 19:50:59 +00001313 SumCtx *p;
drh3d1d95e2005-09-08 10:37:01 +00001314 int type;
drh3f219f42005-09-08 19:45:57 +00001315 assert( argc==1 );
danielk1977f3d3c272008-11-19 16:52:44 +00001316 UNUSED_PARAMETER(argc);
drh4f26d6c2004-05-26 23:25:30 +00001317 p = sqlite3_aggregate_context(context, sizeof(*p));
drh29d72102006-02-09 22:13:41 +00001318 type = sqlite3_value_numeric_type(argv[0]);
drh3d1d95e2005-09-08 10:37:01 +00001319 if( p && type!=SQLITE_NULL ){
drh739105c2002-05-29 23:22:23 +00001320 p->cnt++;
drh29d72102006-02-09 22:13:41 +00001321 if( type==SQLITE_INTEGER ){
drh8c08e862006-02-11 17:34:00 +00001322 i64 v = sqlite3_value_int64(argv[0]);
1323 p->rSum += v;
drh158b9cb2011-03-05 20:59:46 +00001324 if( (p->approx|p->overflow)==0 && sqlite3AddInt64(&p->iSum, v) ){
1325 p->overflow = 1;
drh29d72102006-02-09 22:13:41 +00001326 }
1327 }else{
drh8c08e862006-02-11 17:34:00 +00001328 p->rSum += sqlite3_value_double(argv[0]);
drh29d72102006-02-09 22:13:41 +00001329 p->approx = 1;
drh3f219f42005-09-08 19:45:57 +00001330 }
drh739105c2002-05-29 23:22:23 +00001331 }
drhdd5baa92002-02-27 19:50:59 +00001332}
danielk19770ae8b832004-05-25 12:05:56 +00001333static void sumFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001334 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001335 p = sqlite3_aggregate_context(context, 0);
drhc2bd9132005-09-08 20:37:43 +00001336 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001337 if( p->overflow ){
1338 sqlite3_result_error(context,"integer overflow",-1);
1339 }else if( p->approx ){
1340 sqlite3_result_double(context, p->rSum);
drhc2bd9132005-09-08 20:37:43 +00001341 }else{
drh8c08e862006-02-11 17:34:00 +00001342 sqlite3_result_int64(context, p->iSum);
drhc2bd9132005-09-08 20:37:43 +00001343 }
drh3d1d95e2005-09-08 10:37:01 +00001344 }
drhdd5baa92002-02-27 19:50:59 +00001345}
danielk19770ae8b832004-05-25 12:05:56 +00001346static void avgFinalize(sqlite3_context *context){
drhdd5baa92002-02-27 19:50:59 +00001347 SumCtx *p;
drhabfcea22005-09-06 20:36:48 +00001348 p = sqlite3_aggregate_context(context, 0);
drh739105c2002-05-29 23:22:23 +00001349 if( p && p->cnt>0 ){
drh8c08e862006-02-11 17:34:00 +00001350 sqlite3_result_double(context, p->rSum/(double)p->cnt);
drhdd5baa92002-02-27 19:50:59 +00001351 }
1352}
drha97fdd32006-01-12 22:17:50 +00001353static void totalFinalize(sqlite3_context *context){
1354 SumCtx *p;
1355 p = sqlite3_aggregate_context(context, 0);
shanefbd60f82009-02-04 03:59:25 +00001356 /* (double)0 In case of SQLITE_OMIT_FLOATING_POINT... */
1357 sqlite3_result_double(context, p ? p->rSum : (double)0);
drha97fdd32006-01-12 22:17:50 +00001358}
drhdd5baa92002-02-27 19:50:59 +00001359
1360/*
drh0bce8352002-02-28 00:41:10 +00001361** The following structure keeps track of state information for the
1362** count() aggregate function.
1363*/
1364typedef struct CountCtx CountCtx;
1365struct CountCtx {
drhfc6ad392006-02-09 13:38:19 +00001366 i64 n;
drh0bce8352002-02-28 00:41:10 +00001367};
drhdd5baa92002-02-27 19:50:59 +00001368
drh0bce8352002-02-28 00:41:10 +00001369/*
1370** Routines to implement the count() aggregate function.
1371*/
danielk19770ae8b832004-05-25 12:05:56 +00001372static void countStep(sqlite3_context *context, int argc, sqlite3_value **argv){
drh0bce8352002-02-28 00:41:10 +00001373 CountCtx *p;
drh4f26d6c2004-05-26 23:25:30 +00001374 p = sqlite3_aggregate_context(context, sizeof(*p));
drh9c054832004-05-31 18:51:57 +00001375 if( (argc==0 || SQLITE_NULL!=sqlite3_value_type(argv[0])) && p ){
drh0bce8352002-02-28 00:41:10 +00001376 p->n++;
1377 }
drh2e79c3d2009-04-08 23:04:14 +00001378
drhd3264c72009-04-15 13:39:47 +00001379#ifndef SQLITE_OMIT_DEPRECATED
drh2e79c3d2009-04-08 23:04:14 +00001380 /* The sqlite3_aggregate_count() function is deprecated. But just to make
1381 ** sure it still operates correctly, verify that its count agrees with our
1382 ** internal count when using count(*) and when the total count can be
1383 ** expressed as a 32-bit integer. */
1384 assert( argc==1 || p==0 || p->n>0x7fffffff
1385 || p->n==sqlite3_aggregate_count(context) );
drhd3264c72009-04-15 13:39:47 +00001386#endif
drh0bce8352002-02-28 00:41:10 +00001387}
danielk19770ae8b832004-05-25 12:05:56 +00001388static void countFinalize(sqlite3_context *context){
drh0bce8352002-02-28 00:41:10 +00001389 CountCtx *p;
drhabfcea22005-09-06 20:36:48 +00001390 p = sqlite3_aggregate_context(context, 0);
drhfc6ad392006-02-09 13:38:19 +00001391 sqlite3_result_int64(context, p ? p->n : 0);
drh0bce8352002-02-28 00:41:10 +00001392}
1393
1394/*
drh0bce8352002-02-28 00:41:10 +00001395** Routines to implement min() and max() aggregate functions.
1396*/
danielk197762c14b32008-11-19 09:05:26 +00001397static void minmaxStep(
1398 sqlite3_context *context,
1399 int NotUsed,
1400 sqlite3_value **argv
1401){
danielk197788208052004-05-25 01:13:20 +00001402 Mem *pArg = (Mem *)argv[0];
drh9eb516c2004-07-18 20:52:32 +00001403 Mem *pBest;
danielk197762c14b32008-11-19 09:05:26 +00001404 UNUSED_PARAMETER(NotUsed);
drh9eb516c2004-07-18 20:52:32 +00001405
drh9eb516c2004-07-18 20:52:32 +00001406 pBest = (Mem *)sqlite3_aggregate_context(context, sizeof(*pBest));
danielk19773aeab9e2004-06-24 00:20:04 +00001407 if( !pBest ) return;
drh268380c2004-02-25 13:47:31 +00001408
drh94a6d992012-02-02 18:42:09 +00001409 if( sqlite3_value_type(argv[0])==SQLITE_NULL ){
1410 if( pBest->flags ) sqlite3SkipAccumulatorLoad(context);
1411 }else if( pBest->flags ){
drh9eb516c2004-07-18 20:52:32 +00001412 int max;
1413 int cmp;
danielk1977dc1bdc42004-06-11 10:51:27 +00001414 CollSeq *pColl = sqlite3GetFuncCollSeq(context);
danielk19777e18c252004-05-25 11:47:24 +00001415 /* This step function is used for both the min() and max() aggregates,
1416 ** the only difference between the two being that the sense of the
1417 ** comparison is inverted. For the max() aggregate, the
1418 ** sqlite3_user_data() function returns (void *)-1. For min() it
1419 ** returns (void *)db, where db is the sqlite3* database pointer.
1420 ** Therefore the next statement sets variable 'max' to 1 for the max()
1421 ** aggregate, or 0 for min().
1422 */
drh309b3382007-03-17 17:52:42 +00001423 max = sqlite3_user_data(context)!=0;
danielk1977dc1bdc42004-06-11 10:51:27 +00001424 cmp = sqlite3MemCompare(pBest, pArg, pColl);
danielk197788208052004-05-25 01:13:20 +00001425 if( (max && cmp<0) || (!max && cmp>0) ){
drhb21c8cd2007-08-21 19:33:56 +00001426 sqlite3VdbeMemCopy(pBest, pArg);
drh7a957892012-02-02 17:35:43 +00001427 }else{
1428 sqlite3SkipAccumulatorLoad(context);
danielk197788208052004-05-25 01:13:20 +00001429 }
drh268380c2004-02-25 13:47:31 +00001430 }else{
drhb21c8cd2007-08-21 19:33:56 +00001431 sqlite3VdbeMemCopy(pBest, pArg);
drh0bce8352002-02-28 00:41:10 +00001432 }
1433}
danielk19770ae8b832004-05-25 12:05:56 +00001434static void minMaxFinalize(sqlite3_context *context){
danielk197788208052004-05-25 01:13:20 +00001435 sqlite3_value *pRes;
drhabfcea22005-09-06 20:36:48 +00001436 pRes = (sqlite3_value *)sqlite3_aggregate_context(context, 0);
1437 if( pRes ){
drh94a6d992012-02-02 18:42:09 +00001438 if( pRes->flags ){
drhabfcea22005-09-06 20:36:48 +00001439 sqlite3_result_value(context, pRes);
1440 }
1441 sqlite3VdbeMemRelease(pRes);
drh0bce8352002-02-28 00:41:10 +00001442 }
1443}
drhdd5baa92002-02-27 19:50:59 +00001444
drhb0689692007-11-01 17:38:30 +00001445/*
1446** group_concat(EXPR, ?SEPARATOR?)
1447*/
1448static void groupConcatStep(
1449 sqlite3_context *context,
1450 int argc,
1451 sqlite3_value **argv
1452){
1453 const char *zVal;
drhade86482007-11-28 22:36:40 +00001454 StrAccum *pAccum;
drhb0689692007-11-01 17:38:30 +00001455 const char *zSep;
drh07d31172009-02-02 21:57:05 +00001456 int nVal, nSep;
1457 assert( argc==1 || argc==2 );
1458 if( sqlite3_value_type(argv[0])==SQLITE_NULL ) return;
drhade86482007-11-28 22:36:40 +00001459 pAccum = (StrAccum*)sqlite3_aggregate_context(context, sizeof(*pAccum));
1460
1461 if( pAccum ){
drhbb4957f2008-03-20 14:03:29 +00001462 sqlite3 *db = sqlite3_context_db_handle(context);
drh8bfd7192009-06-19 16:44:41 +00001463 int firstTerm = pAccum->useMalloc==0;
drhb9755982010-07-24 16:34:37 +00001464 pAccum->useMalloc = 2;
drhbb4957f2008-03-20 14:03:29 +00001465 pAccum->mxAlloc = db->aLimit[SQLITE_LIMIT_LENGTH];
drh8bfd7192009-06-19 16:44:41 +00001466 if( !firstTerm ){
drh07d31172009-02-02 21:57:05 +00001467 if( argc==2 ){
1468 zSep = (char*)sqlite3_value_text(argv[1]);
1469 nSep = sqlite3_value_bytes(argv[1]);
drhb0689692007-11-01 17:38:30 +00001470 }else{
1471 zSep = ",";
drhade86482007-11-28 22:36:40 +00001472 nSep = 1;
drhb0689692007-11-01 17:38:30 +00001473 }
drhade86482007-11-28 22:36:40 +00001474 sqlite3StrAccumAppend(pAccum, zSep, nSep);
drhb0689692007-11-01 17:38:30 +00001475 }
drh07d31172009-02-02 21:57:05 +00001476 zVal = (char*)sqlite3_value_text(argv[0]);
1477 nVal = sqlite3_value_bytes(argv[0]);
1478 sqlite3StrAccumAppend(pAccum, zVal, nVal);
drhb0689692007-11-01 17:38:30 +00001479 }
1480}
1481static void groupConcatFinalize(sqlite3_context *context){
drhade86482007-11-28 22:36:40 +00001482 StrAccum *pAccum;
1483 pAccum = sqlite3_aggregate_context(context, 0);
1484 if( pAccum ){
1485 if( pAccum->tooBig ){
1486 sqlite3_result_error_toobig(context);
1487 }else if( pAccum->mallocFailed ){
1488 sqlite3_result_error_nomem(context);
1489 }else{
1490 sqlite3_result_text(context, sqlite3StrAccumFinish(pAccum), -1,
1491 sqlite3_free);
1492 }
drhb0689692007-11-01 17:38:30 +00001493 }
1494}
drh4e5ffc52004-08-31 00:52:37 +00001495
drhd3a149e2002-02-24 17:12:53 +00001496/*
drha4741842010-04-25 20:58:37 +00001497** This routine does per-connection function registration. Most
1498** of the built-in functions above are part of the global function set.
1499** This routine only deals with those that are not global.
drhdc04c582002-02-24 01:55:15 +00001500*/
drh9bb575f2004-09-06 17:24:11 +00001501void sqlite3RegisterBuiltinFunctions(sqlite3 *db){
drha4741842010-04-25 20:58:37 +00001502 int rc = sqlite3_overload_function(db, "MATCH", 2);
1503 assert( rc==SQLITE_NOMEM || rc==SQLITE_OK );
1504 if( rc==SQLITE_NOMEM ){
1505 db->mallocFailed = 1;
danielk1977832a58a2007-06-22 15:21:15 +00001506 }
drh55ef4d92005-08-14 01:20:37 +00001507}
1508
1509/*
1510** Set the LIKEOPT flag on the 2-argument function with the given name.
1511*/
drh1bd10f82008-12-10 21:19:56 +00001512static void setLikeOptFlag(sqlite3 *db, const char *zName, u8 flagVal){
drh55ef4d92005-08-14 01:20:37 +00001513 FuncDef *pDef;
drhea678832008-12-10 19:26:22 +00001514 pDef = sqlite3FindFunction(db, zName, sqlite3Strlen30(zName),
1515 2, SQLITE_UTF8, 0);
drhd27135a2009-04-02 13:36:37 +00001516 if( ALWAYS(pDef) ){
drhd64fe2f2005-08-28 17:00:23 +00001517 pDef->flags = flagVal;
drh55ef4d92005-08-14 01:20:37 +00001518 }
1519}
1520
1521/*
1522** Register the built-in LIKE and GLOB functions. The caseSensitive
1523** parameter determines whether or not the LIKE operator is case
1524** sensitive. GLOB is always case sensitive.
1525*/
1526void sqlite3RegisterLikeFunctions(sqlite3 *db, int caseSensitive){
1527 struct compareInfo *pInfo;
1528 if( caseSensitive ){
1529 pInfo = (struct compareInfo*)&likeInfoAlt;
1530 }else{
1531 pInfo = (struct compareInfo*)&likeInfoNorm;
1532 }
drh901e9942010-12-15 18:54:37 +00001533 sqlite3CreateFunc(db, "like", 2, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0);
1534 sqlite3CreateFunc(db, "like", 3, SQLITE_UTF8, pInfo, likeFunc, 0, 0, 0);
1535 sqlite3CreateFunc(db, "glob", 2, SQLITE_UTF8,
dand2199f02010-08-27 17:48:52 +00001536 (struct compareInfo*)&globInfo, likeFunc, 0, 0, 0);
drhd64fe2f2005-08-28 17:00:23 +00001537 setLikeOptFlag(db, "glob", SQLITE_FUNC_LIKE | SQLITE_FUNC_CASE);
1538 setLikeOptFlag(db, "like",
1539 caseSensitive ? (SQLITE_FUNC_LIKE | SQLITE_FUNC_CASE) : SQLITE_FUNC_LIKE);
drh55ef4d92005-08-14 01:20:37 +00001540}
1541
1542/*
1543** pExpr points to an expression which implements a function. If
1544** it is appropriate to apply the LIKE optimization to that function
1545** then set aWc[0] through aWc[2] to the wildcard characters and
1546** return TRUE. If the function is not a LIKE-style function then
1547** return FALSE.
1548*/
drhd64fe2f2005-08-28 17:00:23 +00001549int sqlite3IsLikeFunction(sqlite3 *db, Expr *pExpr, int *pIsNocase, char *aWc){
drh55ef4d92005-08-14 01:20:37 +00001550 FuncDef *pDef;
danielk19776ab3a2e2009-02-19 14:39:25 +00001551 if( pExpr->op!=TK_FUNCTION
1552 || !pExpr->x.pList
1553 || pExpr->x.pList->nExpr!=2
1554 ){
drh55ef4d92005-08-14 01:20:37 +00001555 return 0;
1556 }
danielk19776ab3a2e2009-02-19 14:39:25 +00001557 assert( !ExprHasProperty(pExpr, EP_xIsSelect) );
drh33e619f2009-05-28 01:00:55 +00001558 pDef = sqlite3FindFunction(db, pExpr->u.zToken,
1559 sqlite3Strlen30(pExpr->u.zToken),
drhb7916a72009-05-27 10:31:29 +00001560 2, SQLITE_UTF8, 0);
drhd27135a2009-04-02 13:36:37 +00001561 if( NEVER(pDef==0) || (pDef->flags & SQLITE_FUNC_LIKE)==0 ){
drh55ef4d92005-08-14 01:20:37 +00001562 return 0;
1563 }
1564
1565 /* The memcpy() statement assumes that the wildcard characters are
1566 ** the first three statements in the compareInfo structure. The
1567 ** asserts() that follow verify that assumption
1568 */
1569 memcpy(aWc, pDef->pUserData, 3);
1570 assert( (char*)&likeInfoAlt == (char*)&likeInfoAlt.matchAll );
1571 assert( &((char*)&likeInfoAlt)[1] == (char*)&likeInfoAlt.matchOne );
1572 assert( &((char*)&likeInfoAlt)[2] == (char*)&likeInfoAlt.matchSet );
drhd64fe2f2005-08-28 17:00:23 +00001573 *pIsNocase = (pDef->flags & SQLITE_FUNC_CASE)==0;
drh55ef4d92005-08-14 01:20:37 +00001574 return 1;
drhdc04c582002-02-24 01:55:15 +00001575}
danielk19778c0a7912008-08-20 14:49:23 +00001576
drh70a8ca32008-08-21 18:49:27 +00001577/*
drh777c5382008-08-21 20:21:34 +00001578** All all of the FuncDef structures in the aBuiltinFunc[] array above
1579** to the global function hash table. This occurs at start-time (as
1580** a consequence of calling sqlite3_initialize()).
1581**
1582** After this routine runs
drh70a8ca32008-08-21 18:49:27 +00001583*/
1584void sqlite3RegisterGlobalFunctions(void){
danielk197793ce7412008-09-01 19:14:02 +00001585 /*
1586 ** The following array holds FuncDef structures for all of the functions
1587 ** defined in this file.
1588 **
1589 ** The array cannot be constant since changes are made to the
1590 ** FuncDef.pHash elements at start-time. The elements of this array
1591 ** are read-only after initialization is complete.
1592 */
1593 static SQLITE_WSD FuncDef aBuiltinFunc[] = {
1594 FUNCTION(ltrim, 1, 1, 0, trimFunc ),
1595 FUNCTION(ltrim, 2, 1, 0, trimFunc ),
1596 FUNCTION(rtrim, 1, 2, 0, trimFunc ),
1597 FUNCTION(rtrim, 2, 2, 0, trimFunc ),
1598 FUNCTION(trim, 1, 3, 0, trimFunc ),
1599 FUNCTION(trim, 2, 3, 0, trimFunc ),
1600 FUNCTION(min, -1, 0, 1, minmaxFunc ),
1601 FUNCTION(min, 0, 0, 1, 0 ),
1602 AGGREGATE(min, 1, 0, 1, minmaxStep, minMaxFinalize ),
1603 FUNCTION(max, -1, 1, 1, minmaxFunc ),
1604 FUNCTION(max, 0, 1, 1, 0 ),
1605 AGGREGATE(max, 1, 1, 1, minmaxStep, minMaxFinalize ),
drha748fdc2012-03-28 01:34:47 +00001606 FUNCTION2(typeof, 1, 0, 0, typeofFunc, SQLITE_FUNC_TYPEOF),
1607 FUNCTION2(length, 1, 0, 0, lengthFunc, SQLITE_FUNC_LENGTH),
danielk197793ce7412008-09-01 19:14:02 +00001608 FUNCTION(substr, 2, 0, 0, substrFunc ),
1609 FUNCTION(substr, 3, 0, 0, substrFunc ),
1610 FUNCTION(abs, 1, 0, 0, absFunc ),
shanefbd60f82009-02-04 03:59:25 +00001611#ifndef SQLITE_OMIT_FLOATING_POINT
danielk197793ce7412008-09-01 19:14:02 +00001612 FUNCTION(round, 1, 0, 0, roundFunc ),
1613 FUNCTION(round, 2, 0, 0, roundFunc ),
shanefbd60f82009-02-04 03:59:25 +00001614#endif
danielk197793ce7412008-09-01 19:14:02 +00001615 FUNCTION(upper, 1, 0, 0, upperFunc ),
1616 FUNCTION(lower, 1, 0, 0, lowerFunc ),
1617 FUNCTION(coalesce, 1, 0, 0, 0 ),
danielk197793ce7412008-09-01 19:14:02 +00001618 FUNCTION(coalesce, 0, 0, 0, 0 ),
drha748fdc2012-03-28 01:34:47 +00001619 FUNCTION2(coalesce, -1, 0, 0, ifnullFunc, SQLITE_FUNC_COALESCE),
danielk197793ce7412008-09-01 19:14:02 +00001620 FUNCTION(hex, 1, 0, 0, hexFunc ),
drha748fdc2012-03-28 01:34:47 +00001621 FUNCTION2(ifnull, 2, 0, 0, ifnullFunc, SQLITE_FUNC_COALESCE),
drh9373b012009-02-02 01:50:39 +00001622 FUNCTION(random, 0, 0, 0, randomFunc ),
danielk197793ce7412008-09-01 19:14:02 +00001623 FUNCTION(randomblob, 1, 0, 0, randomBlob ),
1624 FUNCTION(nullif, 2, 0, 1, nullifFunc ),
1625 FUNCTION(sqlite_version, 0, 0, 0, versionFunc ),
drh47baebc2009-08-14 16:01:24 +00001626 FUNCTION(sqlite_source_id, 0, 0, 0, sourceidFunc ),
drh840561f2011-04-27 18:08:42 +00001627 FUNCTION(sqlite_log, 2, 0, 0, errlogFunc ),
shanehdc97a8c2010-02-23 20:08:35 +00001628#ifndef SQLITE_OMIT_COMPILEOPTION_DIAGS
drh8bb76d32010-02-26 16:37:47 +00001629 FUNCTION(sqlite_compileoption_used,1, 0, 0, compileoptionusedFunc ),
1630 FUNCTION(sqlite_compileoption_get, 1, 0, 0, compileoptiongetFunc ),
shanehdc97a8c2010-02-23 20:08:35 +00001631#endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */
drh61d2f912012-06-18 14:23:06 +00001632#ifndef SQLITE_OMIT_FLOATING_POINT
1633 FUNCTION(ieee754, 1, 0, 0, ieee754Func ),
1634#endif
danielk197793ce7412008-09-01 19:14:02 +00001635 FUNCTION(quote, 1, 0, 0, quoteFunc ),
1636 FUNCTION(last_insert_rowid, 0, 0, 0, last_insert_rowid),
1637 FUNCTION(changes, 0, 0, 0, changes ),
1638 FUNCTION(total_changes, 0, 0, 0, total_changes ),
1639 FUNCTION(replace, 3, 0, 0, replaceFunc ),
1640 FUNCTION(zeroblob, 1, 0, 0, zeroblobFunc ),
1641 #ifdef SQLITE_SOUNDEX
1642 FUNCTION(soundex, 1, 0, 0, soundexFunc ),
1643 #endif
1644 #ifndef SQLITE_OMIT_LOAD_EXTENSION
1645 FUNCTION(load_extension, 1, 0, 0, loadExt ),
1646 FUNCTION(load_extension, 2, 0, 0, loadExt ),
1647 #endif
1648 AGGREGATE(sum, 1, 0, 0, sumStep, sumFinalize ),
1649 AGGREGATE(total, 1, 0, 0, sumStep, totalFinalize ),
1650 AGGREGATE(avg, 1, 0, 0, sumStep, avgFinalize ),
danielk1977a5533162009-02-24 10:01:51 +00001651 /* AGGREGATE(count, 0, 0, 0, countStep, countFinalize ), */
drh6ac78a02010-09-28 14:26:36 +00001652 {0,SQLITE_UTF8,SQLITE_FUNC_COUNT,0,0,0,countStep,countFinalize,"count",0,0},
danielk197793ce7412008-09-01 19:14:02 +00001653 AGGREGATE(count, 1, 0, 0, countStep, countFinalize ),
drh07d31172009-02-02 21:57:05 +00001654 AGGREGATE(group_concat, 1, 0, 0, groupConcatStep, groupConcatFinalize),
1655 AGGREGATE(group_concat, 2, 0, 0, groupConcatStep, groupConcatFinalize),
danielk197793ce7412008-09-01 19:14:02 +00001656
1657 LIKEFUNC(glob, 2, &globInfo, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
1658 #ifdef SQLITE_CASE_SENSITIVE_LIKE
1659 LIKEFUNC(like, 2, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
1660 LIKEFUNC(like, 3, &likeInfoAlt, SQLITE_FUNC_LIKE|SQLITE_FUNC_CASE),
1661 #else
1662 LIKEFUNC(like, 2, &likeInfoNorm, SQLITE_FUNC_LIKE),
1663 LIKEFUNC(like, 3, &likeInfoNorm, SQLITE_FUNC_LIKE),
1664 #endif
1665 };
1666
drh70a8ca32008-08-21 18:49:27 +00001667 int i;
danielk197793ce7412008-09-01 19:14:02 +00001668 FuncDefHash *pHash = &GLOBAL(FuncDefHash, sqlite3GlobalFunctions);
drh106cee52008-09-03 17:11:16 +00001669 FuncDef *aFunc = (FuncDef*)&GLOBAL(FuncDef, aBuiltinFunc);
danielk197793ce7412008-09-01 19:14:02 +00001670
drh70a8ca32008-08-21 18:49:27 +00001671 for(i=0; i<ArraySize(aBuiltinFunc); i++){
danielk197793ce7412008-09-01 19:14:02 +00001672 sqlite3FuncDefInsert(pHash, &aFunc[i]);
drh70a8ca32008-08-21 18:49:27 +00001673 }
drh777c5382008-08-21 20:21:34 +00001674 sqlite3RegisterDateTimeFunctions();
drh545f5872010-04-24 14:02:59 +00001675#ifndef SQLITE_OMIT_ALTERTABLE
1676 sqlite3AlterFunctions();
1677#endif
drh70a8ca32008-08-21 18:49:27 +00001678}