blob: dec205df4301ef03be57ed7c0737e3aea3cc52d5 [file] [log] [blame]
drhc81c11f2009-11-10 01:30:52 +00001/*
2** 2001 September 15
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** Utility functions used throughout sqlite.
13**
14** This file contains functions for allocating memory, comparing
15** strings, and stuff like that.
16**
17*/
18#include "sqliteInt.h"
19#include <stdarg.h>
drhef9f7192020-01-17 19:14:08 +000020#ifndef SQLITE_OMIT_FLOATING_POINT
drh7e6dc5d2019-05-10 12:14:51 +000021#include <math.h>
drhef9f7192020-01-17 19:14:08 +000022#endif
drhc81c11f2009-11-10 01:30:52 +000023
24/*
drhce059e52019-04-05 17:22:50 +000025** Calls to sqlite3FaultSim() are used to simulate a failure during testing,
26** or to bypass normal error detection during testing in order to let
27** execute proceed futher downstream.
drhc007f612014-05-16 14:17:01 +000028**
drhce059e52019-04-05 17:22:50 +000029** In deployment, sqlite3FaultSim() *always* return SQLITE_OK (0). The
30** sqlite3FaultSim() function only returns non-zero during testing.
drhc007f612014-05-16 14:17:01 +000031**
drhce059e52019-04-05 17:22:50 +000032** During testing, if the test harness has set a fault-sim callback using
33** a call to sqlite3_test_control(SQLITE_TESTCTRL_FAULT_INSTALL), then
34** each call to sqlite3FaultSim() is relayed to that application-supplied
35** callback and the integer return value form the application-supplied
36** callback is returned by sqlite3FaultSim().
37**
38** The integer argument to sqlite3FaultSim() is a code to identify which
39** sqlite3FaultSim() instance is being invoked. Each call to sqlite3FaultSim()
40** should have a unique code. To prevent legacy testing applications from
41** breaking, the codes should not be changed or reused.
drhc007f612014-05-16 14:17:01 +000042*/
drhd12602a2016-12-07 15:49:02 +000043#ifndef SQLITE_UNTESTABLE
drhc007f612014-05-16 14:17:01 +000044int sqlite3FaultSim(int iTest){
45 int (*xCallback)(int) = sqlite3GlobalConfig.xTestCallback;
46 return xCallback ? xCallback(iTest) : SQLITE_OK;
47}
48#endif
49
drh85c8f292010-01-13 17:39:53 +000050#ifndef SQLITE_OMIT_FLOATING_POINT
drhc81c11f2009-11-10 01:30:52 +000051/*
52** Return true if the floating point value is Not a Number (NaN).
drhe534c7b2021-09-06 11:44:19 +000053**
54** Use the math library isnan() function if compiled with SQLITE_HAVE_ISNAN.
55** Otherwise, we have our own implementation that works on most systems.
drhc81c11f2009-11-10 01:30:52 +000056*/
57int sqlite3IsNaN(double x){
drhe534c7b2021-09-06 11:44:19 +000058 int rc; /* The value return */
59#if !SQLITE_HAVE_ISNAN && !HAVE_ISNAN
drh05921222019-05-30 00:46:37 +000060 u64 y;
61 memcpy(&y,&x,sizeof(y));
drhe534c7b2021-09-06 11:44:19 +000062 rc = IsNaN(y);
63#else
64 rc = isnan(x);
65#endif /* HAVE_ISNAN */
66 testcase( rc );
67 return rc;
drhc81c11f2009-11-10 01:30:52 +000068}
drh85c8f292010-01-13 17:39:53 +000069#endif /* SQLITE_OMIT_FLOATING_POINT */
drhc81c11f2009-11-10 01:30:52 +000070
71/*
72** Compute a string length that is limited to what can be stored in
73** lower 30 bits of a 32-bit signed integer.
74**
75** The value returned will never be negative. Nor will it ever be greater
76** than the actual length of the string. For very long strings (greater
77** than 1GiB) the value returned might be less than the true string length.
78*/
79int sqlite3Strlen30(const char *z){
drhc81c11f2009-11-10 01:30:52 +000080 if( z==0 ) return 0;
drh1116bf12015-06-30 03:18:33 +000081 return 0x3fffffff & (int)strlen(z);
drhc81c11f2009-11-10 01:30:52 +000082}
83
84/*
drhd7564862016-03-22 20:05:09 +000085** Return the declared type of a column. Or return zDflt if the column
86** has no declared type.
87**
88** The column type is an extra string stored after the zero-terminator on
89** the column name if and only if the COLFLAG_HASTYPE flag is set.
drh94eaafa2016-02-29 15:53:11 +000090*/
drhd7564862016-03-22 20:05:09 +000091char *sqlite3ColumnType(Column *pCol, char *zDflt){
drh77441fa2021-07-30 18:39:59 +000092 if( pCol->colFlags & COLFLAG_HASTYPE ){
drhcf9d36d2021-08-02 18:03:43 +000093 return pCol->zCnName + strlen(pCol->zCnName) + 1;
drhb70f2ea2021-08-18 12:05:22 +000094 }else if( pCol->eCType ){
95 assert( pCol->eCType<=SQLITE_N_STDTYPE );
96 return (char*)sqlite3StdType[pCol->eCType-1];
drh77441fa2021-07-30 18:39:59 +000097 }else{
98 return zDflt;
99 }
drh94eaafa2016-02-29 15:53:11 +0000100}
101
102/*
drh80fbee02016-03-21 11:57:13 +0000103** Helper function for sqlite3Error() - called rarely. Broken out into
104** a separate routine to avoid unnecessary register saves on entry to
105** sqlite3Error().
drh13f40da2014-08-22 18:00:11 +0000106*/
drh8d2f41c2016-03-21 11:38:01 +0000107static SQLITE_NOINLINE void sqlite3ErrorFinish(sqlite3 *db, int err_code){
108 if( db->pErr ) sqlite3ValueSetNull(db->pErr);
109 sqlite3SystemError(db, err_code);
110}
drh80fbee02016-03-21 11:57:13 +0000111
112/*
113** Set the current error code to err_code and clear any prior error message.
114** Also set iSysErrno (by calling sqlite3System) if the err_code indicates
115** that would be appropriate.
116*/
drh13f40da2014-08-22 18:00:11 +0000117void sqlite3Error(sqlite3 *db, int err_code){
118 assert( db!=0 );
119 db->errCode = err_code;
drhf62641e2021-12-24 20:22:13 +0000120 if( err_code || db->pErr ){
121 sqlite3ErrorFinish(db, err_code);
122 }else{
123 db->errByteOffset = -1;
124 }
drh13f40da2014-08-22 18:00:11 +0000125}
126
127/*
drh88efc792021-01-01 18:23:56 +0000128** The equivalent of sqlite3Error(db, SQLITE_OK). Clear the error state
129** and error message.
130*/
131void sqlite3ErrorClear(sqlite3 *db){
132 assert( db!=0 );
133 db->errCode = SQLITE_OK;
drhf62641e2021-12-24 20:22:13 +0000134 db->errByteOffset = -1;
drh88efc792021-01-01 18:23:56 +0000135 if( db->pErr ) sqlite3ValueSetNull(db->pErr);
136}
137
138/*
drh1b9f2142016-03-17 16:01:23 +0000139** Load the sqlite3.iSysErrno field if that is an appropriate thing
140** to do based on the SQLite error code in rc.
141*/
142void sqlite3SystemError(sqlite3 *db, int rc){
143 if( rc==SQLITE_IOERR_NOMEM ) return;
144 rc &= 0xff;
145 if( rc==SQLITE_CANTOPEN || rc==SQLITE_IOERR ){
146 db->iSysErrno = sqlite3OsGetLastError(db->pVfs);
147 }
148}
149
150/*
drhc81c11f2009-11-10 01:30:52 +0000151** Set the most recent error code and error string for the sqlite
152** handle "db". The error code is set to "err_code".
153**
154** If it is not NULL, string zFormat specifies the format of the
drhf62641e2021-12-24 20:22:13 +0000155** error string. zFormat and any string tokens that follow it are
156** assumed to be encoded in UTF-8.
drhc81c11f2009-11-10 01:30:52 +0000157**
158** To clear the most recent error for sqlite handle "db", sqlite3Error
159** should be called with err_code set to SQLITE_OK and zFormat set
160** to NULL.
161*/
drh13f40da2014-08-22 18:00:11 +0000162void sqlite3ErrorWithMsg(sqlite3 *db, int err_code, const char *zFormat, ...){
drha3cc0072013-12-13 16:23:55 +0000163 assert( db!=0 );
164 db->errCode = err_code;
drh8d2f41c2016-03-21 11:38:01 +0000165 sqlite3SystemError(db, err_code);
drh13f40da2014-08-22 18:00:11 +0000166 if( zFormat==0 ){
167 sqlite3Error(db, err_code);
168 }else if( db->pErr || (db->pErr = sqlite3ValueNew(db))!=0 ){
drha3cc0072013-12-13 16:23:55 +0000169 char *z;
170 va_list ap;
171 va_start(ap, zFormat);
172 z = sqlite3VMPrintf(db, zFormat, ap);
173 va_end(ap);
174 sqlite3ValueSetStr(db->pErr, -1, z, SQLITE_UTF8, SQLITE_DYNAMIC);
drhc81c11f2009-11-10 01:30:52 +0000175 }
176}
177
178/*
179** Add an error message to pParse->zErrMsg and increment pParse->nErr.
drhc81c11f2009-11-10 01:30:52 +0000180**
drh13f40da2014-08-22 18:00:11 +0000181** This function should be used to report any error that occurs while
drhc81c11f2009-11-10 01:30:52 +0000182** compiling an SQL statement (i.e. within sqlite3_prepare()). The
183** last thing the sqlite3_prepare() function does is copy the error
184** stored by this function into the database handle using sqlite3Error().
drh13f40da2014-08-22 18:00:11 +0000185** Functions sqlite3Error() or sqlite3ErrorWithMsg() should be used
186** during statement execution (sqlite3_step() etc.).
drhc81c11f2009-11-10 01:30:52 +0000187*/
188void sqlite3ErrorMsg(Parse *pParse, const char *zFormat, ...){
drha7564662010-02-22 19:32:31 +0000189 char *zMsg;
drhc81c11f2009-11-10 01:30:52 +0000190 va_list ap;
191 sqlite3 *db = pParse->db;
drhf62641e2021-12-24 20:22:13 +0000192 db->errByteOffset = -2;
drhc81c11f2009-11-10 01:30:52 +0000193 va_start(ap, zFormat);
drha7564662010-02-22 19:32:31 +0000194 zMsg = sqlite3VMPrintf(db, zFormat, ap);
drhc81c11f2009-11-10 01:30:52 +0000195 va_end(ap);
drhf62641e2021-12-24 20:22:13 +0000196 if( db->errByteOffset<-1 ) db->errByteOffset = -1;
drha7564662010-02-22 19:32:31 +0000197 if( db->suppressErr ){
198 sqlite3DbFree(db, zMsg);
199 }else{
200 pParse->nErr++;
201 sqlite3DbFree(db, pParse->zErrMsg);
202 pParse->zErrMsg = zMsg;
203 pParse->rc = SQLITE_ERROR;
drh46a31cd2019-11-09 14:38:58 +0000204 pParse->pWith = 0;
drha7564662010-02-22 19:32:31 +0000205 }
drhc81c11f2009-11-10 01:30:52 +0000206}
207
208/*
drhc3dcdba2019-04-09 21:32:46 +0000209** If database connection db is currently parsing SQL, then transfer
210** error code errCode to that parser if the parser has not already
211** encountered some other kind of error.
212*/
213int sqlite3ErrorToParser(sqlite3 *db, int errCode){
214 Parse *pParse;
215 if( db==0 || (pParse = db->pParse)==0 ) return errCode;
216 pParse->rc = errCode;
217 pParse->nErr++;
218 return errCode;
219}
220
221/*
drhc81c11f2009-11-10 01:30:52 +0000222** Convert an SQL-style quoted string into a normal string by removing
223** the quote characters. The conversion is done in-place. If the
224** input does not begin with a quote character, then this routine
225** is a no-op.
226**
227** The input string must be zero-terminated. A new zero-terminator
228** is added to the dequoted string.
229**
230** The return value is -1 if no dequoting occurs or the length of the
231** dequoted string, exclusive of the zero terminator, if dequoting does
232** occur.
233**
drh51d35b02019-01-11 13:32:23 +0000234** 2002-02-14: This routine is extended to remove MS-Access style
peter.d.reid60ec9142014-09-06 16:39:46 +0000235** brackets from around identifiers. For example: "[a-b-c]" becomes
drhc81c11f2009-11-10 01:30:52 +0000236** "a-b-c".
237*/
drh244b9d62016-04-11 19:01:08 +0000238void sqlite3Dequote(char *z){
drhc81c11f2009-11-10 01:30:52 +0000239 char quote;
240 int i, j;
drh244b9d62016-04-11 19:01:08 +0000241 if( z==0 ) return;
drhc81c11f2009-11-10 01:30:52 +0000242 quote = z[0];
drh244b9d62016-04-11 19:01:08 +0000243 if( !sqlite3Isquote(quote) ) return;
244 if( quote=='[' ) quote = ']';
drh9ccd8652013-09-13 16:36:46 +0000245 for(i=1, j=0;; i++){
246 assert( z[i] );
drhc81c11f2009-11-10 01:30:52 +0000247 if( z[i]==quote ){
248 if( z[i+1]==quote ){
249 z[j++] = quote;
250 i++;
251 }else{
252 break;
253 }
254 }else{
255 z[j++] = z[i];
256 }
257 }
258 z[j] = 0;
drhc81c11f2009-11-10 01:30:52 +0000259}
drh51d35b02019-01-11 13:32:23 +0000260void sqlite3DequoteExpr(Expr *p){
drhf9751072021-10-07 13:40:29 +0000261 assert( !ExprHasProperty(p, EP_IntValue) );
drh51d35b02019-01-11 13:32:23 +0000262 assert( sqlite3Isquote(p->u.zToken[0]) );
263 p->flags |= p->u.zToken[0]=='"' ? EP_Quoted|EP_DblQuoted : EP_Quoted;
264 sqlite3Dequote(p->u.zToken);
265}
drhc81c11f2009-11-10 01:30:52 +0000266
drh40aced52016-01-22 17:48:09 +0000267/*
drh77441fa2021-07-30 18:39:59 +0000268** If the input token p is quoted, try to adjust the token to remove
269** the quotes. This is not always possible:
270**
271** "abc" -> abc
272** "ab""cd" -> (not possible because of the interior "")
273**
274** Remove the quotes if possible. This is a optimization. The overall
275** system should still return the correct answer even if this routine
276** is always a no-op.
277*/
278void sqlite3DequoteToken(Token *p){
drh15482bc2021-08-06 15:26:01 +0000279 unsigned int i;
drh77441fa2021-07-30 18:39:59 +0000280 if( p->n<2 ) return;
281 if( !sqlite3Isquote(p->z[0]) ) return;
282 for(i=1; i<p->n-1; i++){
283 if( sqlite3Isquote(p->z[i]) ) return;
284 }
285 p->n -= 2;
286 p->z++;
287}
288
289/*
drh40aced52016-01-22 17:48:09 +0000290** Generate a Token object from a string
291*/
292void sqlite3TokenInit(Token *p, char *z){
293 p->z = z;
294 p->n = sqlite3Strlen30(z);
295}
296
drhc81c11f2009-11-10 01:30:52 +0000297/* Convenient short-hand */
298#define UpperToLower sqlite3UpperToLower
299
300/*
301** Some systems have stricmp(). Others have strcasecmp(). Because
302** there is no consistency, we will define our own.
drh9f129f42010-08-31 15:27:32 +0000303**
drh0299b402012-03-19 17:42:46 +0000304** IMPLEMENTATION-OF: R-30243-02494 The sqlite3_stricmp() and
305** sqlite3_strnicmp() APIs allow applications and extensions to compare
306** the contents of two buffers containing UTF-8 strings in a
307** case-independent fashion, using the same definition of "case
308** independence" that SQLite uses internally when comparing identifiers.
drhc81c11f2009-11-10 01:30:52 +0000309*/
drh3fa97302012-02-22 16:58:36 +0000310int sqlite3_stricmp(const char *zLeft, const char *zRight){
drh9ca95732014-10-24 00:35:58 +0000311 if( zLeft==0 ){
312 return zRight ? -1 : 0;
313 }else if( zRight==0 ){
314 return 1;
315 }
drh80738d92016-02-15 00:34:16 +0000316 return sqlite3StrICmp(zLeft, zRight);
317}
318int sqlite3StrICmp(const char *zLeft, const char *zRight){
319 unsigned char *a, *b;
drh7e427332019-04-17 11:34:44 +0000320 int c, x;
drhc81c11f2009-11-10 01:30:52 +0000321 a = (unsigned char *)zLeft;
322 b = (unsigned char *)zRight;
drh80738d92016-02-15 00:34:16 +0000323 for(;;){
drh7e427332019-04-17 11:34:44 +0000324 c = *a;
325 x = *b;
326 if( c==x ){
327 if( c==0 ) break;
328 }else{
329 c = (int)UpperToLower[c] - (int)UpperToLower[x];
330 if( c ) break;
331 }
drh80738d92016-02-15 00:34:16 +0000332 a++;
333 b++;
334 }
335 return c;
drhc81c11f2009-11-10 01:30:52 +0000336}
337int sqlite3_strnicmp(const char *zLeft, const char *zRight, int N){
338 register unsigned char *a, *b;
drh9ca95732014-10-24 00:35:58 +0000339 if( zLeft==0 ){
340 return zRight ? -1 : 0;
341 }else if( zRight==0 ){
342 return 1;
343 }
drhc81c11f2009-11-10 01:30:52 +0000344 a = (unsigned char *)zLeft;
345 b = (unsigned char *)zRight;
346 while( N-- > 0 && *a!=0 && UpperToLower[*a]==UpperToLower[*b]){ a++; b++; }
347 return N<0 ? 0 : UpperToLower[*a] - UpperToLower[*b];
348}
349
350/*
drhd44390c2020-04-06 18:16:31 +0000351** Compute an 8-bit hash on a string that is insensitive to case differences
352*/
353u8 sqlite3StrIHash(const char *z){
354 u8 h = 0;
355 if( z==0 ) return 0;
356 while( z[0] ){
357 h += UpperToLower[(unsigned char)z[0]];
358 z++;
359 }
360 return h;
361}
362
363/*
drh02a43f62017-12-26 14:46:20 +0000364** Compute 10 to the E-th power. Examples: E==1 results in 10.
365** E==2 results in 100. E==50 results in 1.0e50.
366**
367** This routine only works for values of E between 1 and 341.
368*/
369static LONGDOUBLE_TYPE sqlite3Pow10(int E){
drh3dc97272018-01-17 21:14:17 +0000370#if defined(_MSC_VER)
371 static const LONGDOUBLE_TYPE x[] = {
drh38a59af2019-05-25 17:41:07 +0000372 1.0e+001L,
373 1.0e+002L,
374 1.0e+004L,
375 1.0e+008L,
376 1.0e+016L,
377 1.0e+032L,
378 1.0e+064L,
379 1.0e+128L,
380 1.0e+256L
drh3dc97272018-01-17 21:14:17 +0000381 };
382 LONGDOUBLE_TYPE r = 1.0;
383 int i;
384 assert( E>=0 && E<=307 );
385 for(i=0; E!=0; i++, E >>=1){
386 if( E & 1 ) r *= x[i];
387 }
388 return r;
389#else
drh02a43f62017-12-26 14:46:20 +0000390 LONGDOUBLE_TYPE x = 10.0;
391 LONGDOUBLE_TYPE r = 1.0;
392 while(1){
393 if( E & 1 ) r *= x;
394 E >>= 1;
395 if( E==0 ) break;
396 x *= x;
397 }
398 return r;
drh3dc97272018-01-17 21:14:17 +0000399#endif
drh02a43f62017-12-26 14:46:20 +0000400}
401
402/*
drh9339da12010-09-30 00:50:49 +0000403** The string z[] is an text representation of a real number.
drh025586a2010-09-30 17:33:11 +0000404** Convert this string to a double and write it into *pResult.
drhc81c11f2009-11-10 01:30:52 +0000405**
drh9339da12010-09-30 00:50:49 +0000406** The string z[] is length bytes in length (bytes, not characters) and
407** uses the encoding enc. The string is not necessarily zero-terminated.
drhc81c11f2009-11-10 01:30:52 +0000408**
drh9339da12010-09-30 00:50:49 +0000409** Return TRUE if the result is a valid real number (or integer) and FALSE
drh8a3884e2019-05-29 21:18:27 +0000410** if the string is empty or contains extraneous text. More specifically
411** return
412** 1 => The input string is a pure integer
413** 2 or more => The input has a decimal point or eNNN clause
drh9a278222019-06-07 22:26:08 +0000414** 0 or less => The input string is not a valid number
415** -1 => Not a valid number, but has a valid prefix which
416** includes a decimal point and/or an eNNN clause
drh8a3884e2019-05-29 21:18:27 +0000417**
418** Valid numbers are in one of these formats:
drh025586a2010-09-30 17:33:11 +0000419**
420** [+-]digits[E[+-]digits]
421** [+-]digits.[digits][E[+-]digits]
422** [+-].digits[E[+-]digits]
423**
424** Leading and trailing whitespace is ignored for the purpose of determining
425** validity.
426**
427** If some prefix of the input string is a valid number, this routine
428** returns FALSE but it still converts the prefix and writes the result
429** into *pResult.
drhc81c11f2009-11-10 01:30:52 +0000430*/
mistachkin6dcf9a42019-10-10 23:58:16 +0000431#if defined(_MSC_VER)
432#pragma warning(disable : 4756)
433#endif
drh9339da12010-09-30 00:50:49 +0000434int sqlite3AtoF(const char *z, double *pResult, int length, u8 enc){
drhc81c11f2009-11-10 01:30:52 +0000435#ifndef SQLITE_OMIT_FLOATING_POINT
drh0e5fba72013-03-20 12:04:29 +0000436 int incr;
drhe3a4f2c2019-12-13 23:38:57 +0000437 const char *zEnd;
drhc81c11f2009-11-10 01:30:52 +0000438 /* sign * significand * (10 ^ (esign * exponent)) */
drh025586a2010-09-30 17:33:11 +0000439 int sign = 1; /* sign of significand */
440 i64 s = 0; /* significand */
441 int d = 0; /* adjust exponent for shifting decimal point */
442 int esign = 1; /* sign of exponent */
443 int e = 0; /* exponent */
444 int eValid = 1; /* True exponent is either not used or is well-formed */
drhc81c11f2009-11-10 01:30:52 +0000445 double result;
drhc2b893a2019-05-25 18:17:53 +0000446 int nDigit = 0; /* Number of digits processed */
drh8a3884e2019-05-29 21:18:27 +0000447 int eType = 1; /* 1: pure integer, 2+: fractional -1 or less: bad UTF16 */
drhc81c11f2009-11-10 01:30:52 +0000448
drh0e5fba72013-03-20 12:04:29 +0000449 assert( enc==SQLITE_UTF8 || enc==SQLITE_UTF16LE || enc==SQLITE_UTF16BE );
drh025586a2010-09-30 17:33:11 +0000450 *pResult = 0.0; /* Default return value, in case of an error */
drhe3a4f2c2019-12-13 23:38:57 +0000451 if( length==0 ) return 0;
drh025586a2010-09-30 17:33:11 +0000452
drh0e5fba72013-03-20 12:04:29 +0000453 if( enc==SQLITE_UTF8 ){
454 incr = 1;
drhe3a4f2c2019-12-13 23:38:57 +0000455 zEnd = z + length;
drh0e5fba72013-03-20 12:04:29 +0000456 }else{
457 int i;
458 incr = 2;
drh87969b22020-01-08 12:17:46 +0000459 length &= ~1;
drh0e5fba72013-03-20 12:04:29 +0000460 assert( SQLITE_UTF16LE==2 && SQLITE_UTF16BE==3 );
drh84422db2019-05-30 13:47:10 +0000461 testcase( enc==SQLITE_UTF16LE );
462 testcase( enc==SQLITE_UTF16BE );
drh0e5fba72013-03-20 12:04:29 +0000463 for(i=3-enc; i<length && z[i]==0; i+=2){}
drh8a3884e2019-05-29 21:18:27 +0000464 if( i<length ) eType = -100;
drhad975d52016-04-27 15:24:13 +0000465 zEnd = &z[i^1];
drh0e5fba72013-03-20 12:04:29 +0000466 z += (enc&1);
467 }
drh9339da12010-09-30 00:50:49 +0000468
drhc81c11f2009-11-10 01:30:52 +0000469 /* skip leading spaces */
drh9339da12010-09-30 00:50:49 +0000470 while( z<zEnd && sqlite3Isspace(*z) ) z+=incr;
drh025586a2010-09-30 17:33:11 +0000471 if( z>=zEnd ) return 0;
drh9339da12010-09-30 00:50:49 +0000472
drhc81c11f2009-11-10 01:30:52 +0000473 /* get sign of significand */
474 if( *z=='-' ){
475 sign = -1;
drh9339da12010-09-30 00:50:49 +0000476 z+=incr;
drhc81c11f2009-11-10 01:30:52 +0000477 }else if( *z=='+' ){
drh9339da12010-09-30 00:50:49 +0000478 z+=incr;
drhc81c11f2009-11-10 01:30:52 +0000479 }
drh9339da12010-09-30 00:50:49 +0000480
drhc81c11f2009-11-10 01:30:52 +0000481 /* copy max significant digits to significand */
drhc2b893a2019-05-25 18:17:53 +0000482 while( z<zEnd && sqlite3Isdigit(*z) ){
drhc81c11f2009-11-10 01:30:52 +0000483 s = s*10 + (*z - '0');
drhc2b893a2019-05-25 18:17:53 +0000484 z+=incr; nDigit++;
485 if( s>=((LARGEST_INT64-9)/10) ){
486 /* skip non-significant significand digits
487 ** (increase exponent by d to shift decimal left) */
488 while( z<zEnd && sqlite3Isdigit(*z) ){ z+=incr; d++; }
489 }
drhc81c11f2009-11-10 01:30:52 +0000490 }
drh9339da12010-09-30 00:50:49 +0000491 if( z>=zEnd ) goto do_atof_calc;
drhc81c11f2009-11-10 01:30:52 +0000492
493 /* if decimal point is present */
494 if( *z=='.' ){
drh9339da12010-09-30 00:50:49 +0000495 z+=incr;
drh8a3884e2019-05-29 21:18:27 +0000496 eType++;
drhc81c11f2009-11-10 01:30:52 +0000497 /* copy digits from after decimal to significand
498 ** (decrease exponent by d to shift decimal right) */
drh15af62a2016-04-26 23:14:45 +0000499 while( z<zEnd && sqlite3Isdigit(*z) ){
500 if( s<((LARGEST_INT64-9)/10) ){
501 s = s*10 + (*z - '0');
502 d--;
drhc2b893a2019-05-25 18:17:53 +0000503 nDigit++;
drh15af62a2016-04-26 23:14:45 +0000504 }
drhc2b893a2019-05-25 18:17:53 +0000505 z+=incr;
drhc81c11f2009-11-10 01:30:52 +0000506 }
drhc81c11f2009-11-10 01:30:52 +0000507 }
drh9339da12010-09-30 00:50:49 +0000508 if( z>=zEnd ) goto do_atof_calc;
drhc81c11f2009-11-10 01:30:52 +0000509
510 /* if exponent is present */
511 if( *z=='e' || *z=='E' ){
drh9339da12010-09-30 00:50:49 +0000512 z+=incr;
drh025586a2010-09-30 17:33:11 +0000513 eValid = 0;
drh8a3884e2019-05-29 21:18:27 +0000514 eType++;
drhad975d52016-04-27 15:24:13 +0000515
516 /* This branch is needed to avoid a (harmless) buffer overread. The
517 ** special comment alerts the mutation tester that the correct answer
518 ** is obtained even if the branch is omitted */
519 if( z>=zEnd ) goto do_atof_calc; /*PREVENTS-HARMLESS-OVERREAD*/
520
drhc81c11f2009-11-10 01:30:52 +0000521 /* get sign of exponent */
522 if( *z=='-' ){
523 esign = -1;
drh9339da12010-09-30 00:50:49 +0000524 z+=incr;
drhc81c11f2009-11-10 01:30:52 +0000525 }else if( *z=='+' ){
drh9339da12010-09-30 00:50:49 +0000526 z+=incr;
drhc81c11f2009-11-10 01:30:52 +0000527 }
528 /* copy digits to exponent */
drh9339da12010-09-30 00:50:49 +0000529 while( z<zEnd && sqlite3Isdigit(*z) ){
drh57db4a72011-10-17 20:41:46 +0000530 e = e<10000 ? (e*10 + (*z - '0')) : 10000;
drh9339da12010-09-30 00:50:49 +0000531 z+=incr;
drh025586a2010-09-30 17:33:11 +0000532 eValid = 1;
drhc81c11f2009-11-10 01:30:52 +0000533 }
534 }
535
drh025586a2010-09-30 17:33:11 +0000536 /* skip trailing spaces */
drhc6daa012016-04-27 02:35:03 +0000537 while( z<zEnd && sqlite3Isspace(*z) ) z+=incr;
drh025586a2010-09-30 17:33:11 +0000538
drh9339da12010-09-30 00:50:49 +0000539do_atof_calc:
drhc81c11f2009-11-10 01:30:52 +0000540 /* adjust exponent by d, and update sign */
541 e = (e*esign) + d;
542 if( e<0 ) {
543 esign = -1;
544 e *= -1;
545 } else {
546 esign = 1;
547 }
548
drhad975d52016-04-27 15:24:13 +0000549 if( s==0 ) {
550 /* In the IEEE 754 standard, zero is signed. */
drhc6daa012016-04-27 02:35:03 +0000551 result = sign<0 ? -(double)0 : (double)0;
drhc81c11f2009-11-10 01:30:52 +0000552 } else {
drhad975d52016-04-27 15:24:13 +0000553 /* Attempt to reduce exponent.
554 **
555 ** Branches that are not required for the correct answer but which only
556 ** help to obtain the correct answer faster are marked with special
557 ** comments, as a hint to the mutation tester.
558 */
559 while( e>0 ){ /*OPTIMIZATION-IF-TRUE*/
560 if( esign>0 ){
561 if( s>=(LARGEST_INT64/10) ) break; /*OPTIMIZATION-IF-FALSE*/
562 s *= 10;
563 }else{
564 if( s%10!=0 ) break; /*OPTIMIZATION-IF-FALSE*/
565 s /= 10;
566 }
567 e--;
drhc81c11f2009-11-10 01:30:52 +0000568 }
569
570 /* adjust the sign of significand */
571 s = sign<0 ? -s : s;
572
drhad975d52016-04-27 15:24:13 +0000573 if( e==0 ){ /*OPTIMIZATION-IF-TRUE*/
574 result = (double)s;
575 }else{
drhc81c11f2009-11-10 01:30:52 +0000576 /* attempt to handle extremely small/large numbers better */
drhad975d52016-04-27 15:24:13 +0000577 if( e>307 ){ /*OPTIMIZATION-IF-TRUE*/
578 if( e<342 ){ /*OPTIMIZATION-IF-TRUE*/
drh02a43f62017-12-26 14:46:20 +0000579 LONGDOUBLE_TYPE scale = sqlite3Pow10(e-308);
drhad975d52016-04-27 15:24:13 +0000580 if( esign<0 ){
581 result = s / scale;
582 result /= 1.0e+308;
583 }else{
584 result = s * scale;
585 result *= 1.0e+308;
586 }
587 }else{ assert( e>=342 );
588 if( esign<0 ){
589 result = 0.0*s;
590 }else{
drhb9772e72017-09-12 13:27:43 +0000591#ifdef INFINITY
drh3ba18ad2017-09-12 15:05:34 +0000592 result = INFINITY*s;
drhb9772e72017-09-12 13:27:43 +0000593#else
drhad975d52016-04-27 15:24:13 +0000594 result = 1e308*1e308*s; /* Infinity */
drhb9772e72017-09-12 13:27:43 +0000595#endif
drhad975d52016-04-27 15:24:13 +0000596 }
drh2458a2e2011-10-17 12:14:26 +0000597 }
drhc81c11f2009-11-10 01:30:52 +0000598 }else{
drh02a43f62017-12-26 14:46:20 +0000599 LONGDOUBLE_TYPE scale = sqlite3Pow10(e);
drhc81c11f2009-11-10 01:30:52 +0000600 if( esign<0 ){
601 result = s / scale;
602 }else{
603 result = s * scale;
604 }
605 }
drhc81c11f2009-11-10 01:30:52 +0000606 }
607 }
608
609 /* store the result */
610 *pResult = result;
611
drh025586a2010-09-30 17:33:11 +0000612 /* return true if number and no extra non-whitespace chracters after */
drh9a278222019-06-07 22:26:08 +0000613 if( z==zEnd && nDigit>0 && eValid && eType>0 ){
614 return eType;
drh378a7d32019-06-10 23:45:10 +0000615 }else if( eType>=2 && (eType==3 || eValid) && nDigit>0 ){
drh9a278222019-06-07 22:26:08 +0000616 return -1;
617 }else{
618 return 0;
619 }
drhc81c11f2009-11-10 01:30:52 +0000620#else
shaneh5f1d6b62010-09-30 16:51:25 +0000621 return !sqlite3Atoi64(z, pResult, length, enc);
drhc81c11f2009-11-10 01:30:52 +0000622#endif /* SQLITE_OMIT_FLOATING_POINT */
623}
mistachkin6dcf9a42019-10-10 23:58:16 +0000624#if defined(_MSC_VER)
625#pragma warning(default : 4756)
626#endif
drhc81c11f2009-11-10 01:30:52 +0000627
628/*
drh82b0f102020-07-21 18:25:19 +0000629** Render an signed 64-bit integer as text. Store the result in zOut[].
630**
631** The caller must ensure that zOut[] is at least 21 bytes in size.
632*/
633void sqlite3Int64ToText(i64 v, char *zOut){
634 int i;
635 u64 x;
636 char zTemp[22];
637 if( v<0 ){
drh8deae5a2020-07-29 12:23:20 +0000638 x = (v==SMALLEST_INT64) ? ((u64)1)<<63 : (u64)-v;
drh82b0f102020-07-21 18:25:19 +0000639 }else{
640 x = v;
641 }
642 i = sizeof(zTemp)-2;
643 zTemp[sizeof(zTemp)-1] = 0;
644 do{
645 zTemp[i--] = (x%10) + '0';
646 x = x/10;
647 }while( x );
648 if( v<0 ) zTemp[i--] = '-';
649 memcpy(zOut, &zTemp[i+1], sizeof(zTemp)-1-i);
650}
651
652/*
drhc81c11f2009-11-10 01:30:52 +0000653** Compare the 19-character string zNum against the text representation
654** value 2^63: 9223372036854775808. Return negative, zero, or positive
655** if zNum is less than, equal to, or greater than the string.
shaneh5f1d6b62010-09-30 16:51:25 +0000656** Note that zNum must contain exactly 19 characters.
drhc81c11f2009-11-10 01:30:52 +0000657**
658** Unlike memcmp() this routine is guaranteed to return the difference
659** in the values of the last digit if the only difference is in the
660** last digit. So, for example,
661**
drh9339da12010-09-30 00:50:49 +0000662** compare2pow63("9223372036854775800", 1)
drhc81c11f2009-11-10 01:30:52 +0000663**
664** will return -8.
665*/
drh9339da12010-09-30 00:50:49 +0000666static int compare2pow63(const char *zNum, int incr){
667 int c = 0;
668 int i;
669 /* 012345678901234567 */
670 const char *pow63 = "922337203685477580";
671 for(i=0; c==0 && i<18; i++){
672 c = (zNum[i*incr]-pow63[i])*10;
673 }
drhc81c11f2009-11-10 01:30:52 +0000674 if( c==0 ){
drh9339da12010-09-30 00:50:49 +0000675 c = zNum[18*incr] - '8';
drh44dbca82010-01-13 04:22:20 +0000676 testcase( c==(-1) );
677 testcase( c==0 );
678 testcase( c==(+1) );
drhc81c11f2009-11-10 01:30:52 +0000679 }
680 return c;
681}
682
drhc81c11f2009-11-10 01:30:52 +0000683/*
drh9296c182014-07-23 13:40:49 +0000684** Convert zNum to a 64-bit signed integer. zNum must be decimal. This
685** routine does *not* accept hexadecimal notation.
drh158b9cb2011-03-05 20:59:46 +0000686**
drh84d4f1a2017-09-20 10:47:10 +0000687** Returns:
drh158b9cb2011-03-05 20:59:46 +0000688**
drh9a278222019-06-07 22:26:08 +0000689** -1 Not even a prefix of the input text looks like an integer
drh84d4f1a2017-09-20 10:47:10 +0000690** 0 Successful transformation. Fits in a 64-bit signed integer.
drh4eb57ce2018-01-26 18:37:34 +0000691** 1 Excess non-space text after the integer value
drh84d4f1a2017-09-20 10:47:10 +0000692** 2 Integer too large for a 64-bit signed integer or is malformed
693** 3 Special case of 9223372036854775808
drhc81c11f2009-11-10 01:30:52 +0000694**
drh9339da12010-09-30 00:50:49 +0000695** length is the number of bytes in the string (bytes, not characters).
696** The string is not necessarily zero-terminated. The encoding is
697** given by enc.
drhc81c11f2009-11-10 01:30:52 +0000698*/
drh9339da12010-09-30 00:50:49 +0000699int sqlite3Atoi64(const char *zNum, i64 *pNum, int length, u8 enc){
drh0e5fba72013-03-20 12:04:29 +0000700 int incr;
drh158b9cb2011-03-05 20:59:46 +0000701 u64 u = 0;
shaneh5f1d6b62010-09-30 16:51:25 +0000702 int neg = 0; /* assume positive */
drh9339da12010-09-30 00:50:49 +0000703 int i;
704 int c = 0;
drh609d5842016-04-28 00:32:16 +0000705 int nonNum = 0; /* True if input contains UTF16 with high byte non-zero */
drh84d4f1a2017-09-20 10:47:10 +0000706 int rc; /* Baseline return code */
drhc81c11f2009-11-10 01:30:52 +0000707 const char *zStart;
drh9339da12010-09-30 00:50:49 +0000708 const char *zEnd = zNum + length;
drh0e5fba72013-03-20 12:04:29 +0000709 assert( enc==SQLITE_UTF8 || enc==SQLITE_UTF16LE || enc==SQLITE_UTF16BE );
710 if( enc==SQLITE_UTF8 ){
711 incr = 1;
712 }else{
713 incr = 2;
drh359941b2020-08-27 16:28:30 +0000714 length &= ~1;
drh0e5fba72013-03-20 12:04:29 +0000715 assert( SQLITE_UTF16LE==2 && SQLITE_UTF16BE==3 );
716 for(i=3-enc; i<length && zNum[i]==0; i+=2){}
717 nonNum = i<length;
drh609d5842016-04-28 00:32:16 +0000718 zEnd = &zNum[i^1];
drh0e5fba72013-03-20 12:04:29 +0000719 zNum += (enc&1);
720 }
drh9339da12010-09-30 00:50:49 +0000721 while( zNum<zEnd && sqlite3Isspace(*zNum) ) zNum+=incr;
drh158b9cb2011-03-05 20:59:46 +0000722 if( zNum<zEnd ){
723 if( *zNum=='-' ){
724 neg = 1;
725 zNum+=incr;
726 }else if( *zNum=='+' ){
727 zNum+=incr;
728 }
drhc81c11f2009-11-10 01:30:52 +0000729 }
730 zStart = zNum;
drh9339da12010-09-30 00:50:49 +0000731 while( zNum<zEnd && zNum[0]=='0' ){ zNum+=incr; } /* Skip leading zeros. */
732 for(i=0; &zNum[i]<zEnd && (c=zNum[i])>='0' && c<='9'; i+=incr){
drh158b9cb2011-03-05 20:59:46 +0000733 u = u*10 + c - '0';
drhc81c11f2009-11-10 01:30:52 +0000734 }
drh4eb57ce2018-01-26 18:37:34 +0000735 testcase( i==18*incr );
736 testcase( i==19*incr );
737 testcase( i==20*incr );
drh1822ebf2018-01-27 14:25:27 +0000738 if( u>LARGEST_INT64 ){
739 /* This test and assignment is needed only to suppress UB warnings
740 ** from clang and -fsanitize=undefined. This test and assignment make
741 ** the code a little larger and slower, and no harm comes from omitting
742 ** them, but we must appaise the undefined-behavior pharisees. */
743 *pNum = neg ? SMALLEST_INT64 : LARGEST_INT64;
744 }else if( neg ){
drh158b9cb2011-03-05 20:59:46 +0000745 *pNum = -(i64)u;
746 }else{
747 *pNum = (i64)u;
748 }
drh4eb57ce2018-01-26 18:37:34 +0000749 rc = 0;
drh9a278222019-06-07 22:26:08 +0000750 if( i==0 && zStart==zNum ){ /* No digits */
751 rc = -1;
752 }else if( nonNum ){ /* UTF16 with high-order bytes non-zero */
drh84d4f1a2017-09-20 10:47:10 +0000753 rc = 1;
drh4eb57ce2018-01-26 18:37:34 +0000754 }else if( &zNum[i]<zEnd ){ /* Extra bytes at the end */
755 int jj = i;
756 do{
757 if( !sqlite3Isspace(zNum[jj]) ){
758 rc = 1; /* Extra non-space text after the integer */
759 break;
760 }
761 jj += incr;
762 }while( &zNum[jj]<zEnd );
drh84d4f1a2017-09-20 10:47:10 +0000763 }
drh4eb57ce2018-01-26 18:37:34 +0000764 if( i<19*incr ){
drhc81c11f2009-11-10 01:30:52 +0000765 /* Less than 19 digits, so we know that it fits in 64 bits */
drh158b9cb2011-03-05 20:59:46 +0000766 assert( u<=LARGEST_INT64 );
drh84d4f1a2017-09-20 10:47:10 +0000767 return rc;
drhc81c11f2009-11-10 01:30:52 +0000768 }else{
drh158b9cb2011-03-05 20:59:46 +0000769 /* zNum is a 19-digit numbers. Compare it against 9223372036854775808. */
drh4eb57ce2018-01-26 18:37:34 +0000770 c = i>19*incr ? 1 : compare2pow63(zNum, incr);
drh158b9cb2011-03-05 20:59:46 +0000771 if( c<0 ){
772 /* zNum is less than 9223372036854775808 so it fits */
773 assert( u<=LARGEST_INT64 );
drh84d4f1a2017-09-20 10:47:10 +0000774 return rc;
drh158b9cb2011-03-05 20:59:46 +0000775 }else{
drh4eb57ce2018-01-26 18:37:34 +0000776 *pNum = neg ? SMALLEST_INT64 : LARGEST_INT64;
777 if( c>0 ){
778 /* zNum is greater than 9223372036854775808 so it overflows */
779 return 2;
780 }else{
781 /* zNum is exactly 9223372036854775808. Fits if negative. The
782 ** special case 2 overflow if positive */
783 assert( u-1==LARGEST_INT64 );
784 return neg ? rc : 3;
785 }
drh158b9cb2011-03-05 20:59:46 +0000786 }
drhc81c11f2009-11-10 01:30:52 +0000787 }
788}
789
790/*
drh9296c182014-07-23 13:40:49 +0000791** Transform a UTF-8 integer literal, in either decimal or hexadecimal,
792** into a 64-bit signed integer. This routine accepts hexadecimal literals,
793** whereas sqlite3Atoi64() does not.
794**
795** Returns:
796**
797** 0 Successful transformation. Fits in a 64-bit signed integer.
drh84d4f1a2017-09-20 10:47:10 +0000798** 1 Excess text after the integer value
799** 2 Integer too large for a 64-bit signed integer or is malformed
800** 3 Special case of 9223372036854775808
drh9296c182014-07-23 13:40:49 +0000801*/
802int sqlite3DecOrHexToI64(const char *z, i64 *pOut){
803#ifndef SQLITE_OMIT_HEX_INTEGER
804 if( z[0]=='0'
805 && (z[1]=='x' || z[1]=='X')
drh9296c182014-07-23 13:40:49 +0000806 ){
807 u64 u = 0;
808 int i, k;
809 for(i=2; z[i]=='0'; i++){}
810 for(k=i; sqlite3Isxdigit(z[k]); k++){
811 u = u*16 + sqlite3HexToInt(z[k]);
812 }
813 memcpy(pOut, &u, 8);
drh84d4f1a2017-09-20 10:47:10 +0000814 return (z[k]==0 && k-i<=16) ? 0 : 2;
drh9296c182014-07-23 13:40:49 +0000815 }else
816#endif /* SQLITE_OMIT_HEX_INTEGER */
817 {
818 return sqlite3Atoi64(z, pOut, sqlite3Strlen30(z), SQLITE_UTF8);
819 }
820}
821
822/*
drhc81c11f2009-11-10 01:30:52 +0000823** If zNum represents an integer that will fit in 32-bits, then set
824** *pValue to that integer and return true. Otherwise return false.
825**
drh9296c182014-07-23 13:40:49 +0000826** This routine accepts both decimal and hexadecimal notation for integers.
827**
drhc81c11f2009-11-10 01:30:52 +0000828** Any non-numeric characters that following zNum are ignored.
829** This is different from sqlite3Atoi64() which requires the
830** input number to be zero-terminated.
831*/
832int sqlite3GetInt32(const char *zNum, int *pValue){
833 sqlite_int64 v = 0;
834 int i, c;
835 int neg = 0;
836 if( zNum[0]=='-' ){
837 neg = 1;
838 zNum++;
839 }else if( zNum[0]=='+' ){
840 zNum++;
841 }
drh28e048c2014-07-23 01:26:51 +0000842#ifndef SQLITE_OMIT_HEX_INTEGER
843 else if( zNum[0]=='0'
844 && (zNum[1]=='x' || zNum[1]=='X')
845 && sqlite3Isxdigit(zNum[2])
846 ){
847 u32 u = 0;
848 zNum += 2;
849 while( zNum[0]=='0' ) zNum++;
850 for(i=0; sqlite3Isxdigit(zNum[i]) && i<8; i++){
851 u = u*16 + sqlite3HexToInt(zNum[i]);
852 }
853 if( (u&0x80000000)==0 && sqlite3Isxdigit(zNum[i])==0 ){
854 memcpy(pValue, &u, 4);
855 return 1;
856 }else{
857 return 0;
858 }
859 }
860#endif
drh313e6fd2017-05-03 17:44:28 +0000861 if( !sqlite3Isdigit(zNum[0]) ) return 0;
drh935f2e72015-04-18 04:45:00 +0000862 while( zNum[0]=='0' ) zNum++;
drhc81c11f2009-11-10 01:30:52 +0000863 for(i=0; i<11 && (c = zNum[i] - '0')>=0 && c<=9; i++){
864 v = v*10 + c;
865 }
866
867 /* The longest decimal representation of a 32 bit integer is 10 digits:
868 **
869 ** 1234567890
870 ** 2^31 -> 2147483648
871 */
drh44dbca82010-01-13 04:22:20 +0000872 testcase( i==10 );
drhc81c11f2009-11-10 01:30:52 +0000873 if( i>10 ){
874 return 0;
875 }
drh44dbca82010-01-13 04:22:20 +0000876 testcase( v-neg==2147483647 );
drhc81c11f2009-11-10 01:30:52 +0000877 if( v-neg>2147483647 ){
878 return 0;
879 }
880 if( neg ){
881 v = -v;
882 }
883 *pValue = (int)v;
884 return 1;
885}
886
887/*
drh60ac3f42010-11-23 18:59:27 +0000888** Return a 32-bit integer value extracted from a string. If the
889** string is not an integer, just return 0.
890*/
891int sqlite3Atoi(const char *z){
892 int x = 0;
drh48bf2d72020-07-30 17:14:55 +0000893 sqlite3GetInt32(z, &x);
drh60ac3f42010-11-23 18:59:27 +0000894 return x;
895}
896
897/*
drhabc38152020-07-22 13:38:04 +0000898** Try to convert z into an unsigned 32-bit integer. Return true on
899** success and false if there is an error.
900**
901** Only decimal notation is accepted.
902*/
903int sqlite3GetUInt32(const char *z, u32 *pI){
904 u64 v = 0;
905 int i;
906 for(i=0; sqlite3Isdigit(z[i]); i++){
907 v = v*10 + z[i] - '0';
drh69306bf2020-07-22 20:12:10 +0000908 if( v>4294967296LL ){ *pI = 0; return 0; }
drhabc38152020-07-22 13:38:04 +0000909 }
drh69306bf2020-07-22 20:12:10 +0000910 if( i==0 || z[i]!=0 ){ *pI = 0; return 0; }
drhabc38152020-07-22 13:38:04 +0000911 *pI = (u32)v;
912 return 1;
913}
914
915/*
drhc81c11f2009-11-10 01:30:52 +0000916** The variable-length integer encoding is as follows:
917**
918** KEY:
919** A = 0xxxxxxx 7 bits of data and one flag bit
920** B = 1xxxxxxx 7 bits of data and one flag bit
921** C = xxxxxxxx 8 bits of data
922**
923** 7 bits - A
924** 14 bits - BA
925** 21 bits - BBA
926** 28 bits - BBBA
927** 35 bits - BBBBA
928** 42 bits - BBBBBA
929** 49 bits - BBBBBBA
930** 56 bits - BBBBBBBA
931** 64 bits - BBBBBBBBC
932*/
933
934/*
935** Write a 64-bit variable-length integer to memory starting at p[0].
936** The length of data write will be between 1 and 9 bytes. The number
937** of bytes written is returned.
938**
939** A variable-length integer consists of the lower 7 bits of each byte
940** for all bytes that have the 8th bit set and one byte with the 8th
941** bit clear. Except, if we get to the 9th byte, it stores the full
942** 8 bits and is the last byte.
943*/
drh2f2b2b82014-08-22 18:48:25 +0000944static int SQLITE_NOINLINE putVarint64(unsigned char *p, u64 v){
drhc81c11f2009-11-10 01:30:52 +0000945 int i, j, n;
946 u8 buf[10];
947 if( v & (((u64)0xff000000)<<32) ){
948 p[8] = (u8)v;
949 v >>= 8;
950 for(i=7; i>=0; i--){
951 p[i] = (u8)((v & 0x7f) | 0x80);
952 v >>= 7;
953 }
954 return 9;
955 }
956 n = 0;
957 do{
958 buf[n++] = (u8)((v & 0x7f) | 0x80);
959 v >>= 7;
960 }while( v!=0 );
961 buf[0] &= 0x7f;
962 assert( n<=9 );
963 for(i=0, j=n-1; j>=0; j--, i++){
964 p[i] = buf[j];
965 }
966 return n;
967}
drh2f2b2b82014-08-22 18:48:25 +0000968int sqlite3PutVarint(unsigned char *p, u64 v){
969 if( v<=0x7f ){
970 p[0] = v&0x7f;
drhc81c11f2009-11-10 01:30:52 +0000971 return 1;
972 }
drh2f2b2b82014-08-22 18:48:25 +0000973 if( v<=0x3fff ){
974 p[0] = ((v>>7)&0x7f)|0x80;
975 p[1] = v&0x7f;
drhc81c11f2009-11-10 01:30:52 +0000976 return 2;
977 }
drh2f2b2b82014-08-22 18:48:25 +0000978 return putVarint64(p,v);
drhc81c11f2009-11-10 01:30:52 +0000979}
980
981/*
drh0b2864c2010-03-03 15:18:38 +0000982** Bitmasks used by sqlite3GetVarint(). These precomputed constants
983** are defined here rather than simply putting the constant expressions
984** inline in order to work around bugs in the RVT compiler.
985**
986** SLOT_2_0 A mask for (0x7f<<14) | 0x7f
987**
988** SLOT_4_2_0 A mask for (0x7f<<28) | SLOT_2_0
989*/
990#define SLOT_2_0 0x001fc07f
991#define SLOT_4_2_0 0xf01fc07f
992
993
994/*
drhc81c11f2009-11-10 01:30:52 +0000995** Read a 64-bit variable-length integer from memory starting at p[0].
996** Return the number of bytes read. The value is stored in *v.
997*/
998u8 sqlite3GetVarint(const unsigned char *p, u64 *v){
999 u32 a,b,s;
1000
drh698c86f2019-04-17 12:07:08 +00001001 if( ((signed char*)p)[0]>=0 ){
1002 *v = *p;
drhc81c11f2009-11-10 01:30:52 +00001003 return 1;
1004 }
drh698c86f2019-04-17 12:07:08 +00001005 if( ((signed char*)p)[1]>=0 ){
1006 *v = ((u32)(p[0]&0x7f)<<7) | p[1];
drhc81c11f2009-11-10 01:30:52 +00001007 return 2;
1008 }
1009
drh0b2864c2010-03-03 15:18:38 +00001010 /* Verify that constants are precomputed correctly */
1011 assert( SLOT_2_0 == ((0x7f<<14) | (0x7f)) );
shaneh1da207e2010-03-09 14:41:12 +00001012 assert( SLOT_4_2_0 == ((0xfU<<28) | (0x7f<<14) | (0x7f)) );
drh0b2864c2010-03-03 15:18:38 +00001013
drh698c86f2019-04-17 12:07:08 +00001014 a = ((u32)p[0])<<14;
1015 b = p[1];
1016 p += 2;
drhc81c11f2009-11-10 01:30:52 +00001017 a |= *p;
1018 /* a: p0<<14 | p2 (unmasked) */
1019 if (!(a&0x80))
1020 {
drh0b2864c2010-03-03 15:18:38 +00001021 a &= SLOT_2_0;
drhc81c11f2009-11-10 01:30:52 +00001022 b &= 0x7f;
1023 b = b<<7;
1024 a |= b;
1025 *v = a;
1026 return 3;
1027 }
1028
1029 /* CSE1 from below */
drh0b2864c2010-03-03 15:18:38 +00001030 a &= SLOT_2_0;
drhc81c11f2009-11-10 01:30:52 +00001031 p++;
1032 b = b<<14;
1033 b |= *p;
1034 /* b: p1<<14 | p3 (unmasked) */
1035 if (!(b&0x80))
1036 {
drh0b2864c2010-03-03 15:18:38 +00001037 b &= SLOT_2_0;
drhc81c11f2009-11-10 01:30:52 +00001038 /* moved CSE1 up */
1039 /* a &= (0x7f<<14)|(0x7f); */
1040 a = a<<7;
1041 a |= b;
1042 *v = a;
1043 return 4;
1044 }
1045
1046 /* a: p0<<14 | p2 (masked) */
1047 /* b: p1<<14 | p3 (unmasked) */
1048 /* 1:save off p0<<21 | p1<<14 | p2<<7 | p3 (masked) */
1049 /* moved CSE1 up */
1050 /* a &= (0x7f<<14)|(0x7f); */
drh0b2864c2010-03-03 15:18:38 +00001051 b &= SLOT_2_0;
drhc81c11f2009-11-10 01:30:52 +00001052 s = a;
1053 /* s: p0<<14 | p2 (masked) */
1054
1055 p++;
1056 a = a<<14;
1057 a |= *p;
1058 /* a: p0<<28 | p2<<14 | p4 (unmasked) */
1059 if (!(a&0x80))
1060 {
drh62aaa6c2015-11-21 17:27:42 +00001061 /* we can skip these cause they were (effectively) done above
1062 ** while calculating s */
drhc81c11f2009-11-10 01:30:52 +00001063 /* a &= (0x7f<<28)|(0x7f<<14)|(0x7f); */
1064 /* b &= (0x7f<<14)|(0x7f); */
1065 b = b<<7;
1066 a |= b;
1067 s = s>>18;
1068 *v = ((u64)s)<<32 | a;
1069 return 5;
1070 }
1071
1072 /* 2:save off p0<<21 | p1<<14 | p2<<7 | p3 (masked) */
1073 s = s<<7;
1074 s |= b;
1075 /* s: p0<<21 | p1<<14 | p2<<7 | p3 (masked) */
1076
1077 p++;
1078 b = b<<14;
1079 b |= *p;
1080 /* b: p1<<28 | p3<<14 | p5 (unmasked) */
1081 if (!(b&0x80))
1082 {
1083 /* we can skip this cause it was (effectively) done above in calc'ing s */
1084 /* b &= (0x7f<<28)|(0x7f<<14)|(0x7f); */
drh0b2864c2010-03-03 15:18:38 +00001085 a &= SLOT_2_0;
drhc81c11f2009-11-10 01:30:52 +00001086 a = a<<7;
1087 a |= b;
1088 s = s>>18;
1089 *v = ((u64)s)<<32 | a;
1090 return 6;
1091 }
1092
1093 p++;
1094 a = a<<14;
1095 a |= *p;
1096 /* a: p2<<28 | p4<<14 | p6 (unmasked) */
1097 if (!(a&0x80))
1098 {
drh0b2864c2010-03-03 15:18:38 +00001099 a &= SLOT_4_2_0;
1100 b &= SLOT_2_0;
drhc81c11f2009-11-10 01:30:52 +00001101 b = b<<7;
1102 a |= b;
1103 s = s>>11;
1104 *v = ((u64)s)<<32 | a;
1105 return 7;
1106 }
1107
1108 /* CSE2 from below */
drh0b2864c2010-03-03 15:18:38 +00001109 a &= SLOT_2_0;
drhc81c11f2009-11-10 01:30:52 +00001110 p++;
1111 b = b<<14;
1112 b |= *p;
1113 /* b: p3<<28 | p5<<14 | p7 (unmasked) */
1114 if (!(b&0x80))
1115 {
drh0b2864c2010-03-03 15:18:38 +00001116 b &= SLOT_4_2_0;
drhc81c11f2009-11-10 01:30:52 +00001117 /* moved CSE2 up */
1118 /* a &= (0x7f<<14)|(0x7f); */
1119 a = a<<7;
1120 a |= b;
1121 s = s>>4;
1122 *v = ((u64)s)<<32 | a;
1123 return 8;
1124 }
1125
1126 p++;
1127 a = a<<15;
1128 a |= *p;
1129 /* a: p4<<29 | p6<<15 | p8 (unmasked) */
1130
1131 /* moved CSE2 up */
1132 /* a &= (0x7f<<29)|(0x7f<<15)|(0xff); */
drh0b2864c2010-03-03 15:18:38 +00001133 b &= SLOT_2_0;
drhc81c11f2009-11-10 01:30:52 +00001134 b = b<<8;
1135 a |= b;
1136
1137 s = s<<4;
1138 b = p[-4];
1139 b &= 0x7f;
1140 b = b>>3;
1141 s |= b;
1142
1143 *v = ((u64)s)<<32 | a;
1144
1145 return 9;
1146}
1147
1148/*
1149** Read a 32-bit variable-length integer from memory starting at p[0].
1150** Return the number of bytes read. The value is stored in *v.
1151**
1152** If the varint stored in p[0] is larger than can fit in a 32-bit unsigned
1153** integer, then set *v to 0xffffffff.
1154**
1155** A MACRO version, getVarint32, is provided which inlines the
1156** single-byte case. All code should use the MACRO version as
1157** this function assumes the single-byte case has already been handled.
1158*/
1159u8 sqlite3GetVarint32(const unsigned char *p, u32 *v){
1160 u32 a,b;
1161
1162 /* The 1-byte case. Overwhelmingly the most common. Handled inline
1163 ** by the getVarin32() macro */
1164 a = *p;
1165 /* a: p0 (unmasked) */
1166#ifndef getVarint32
1167 if (!(a&0x80))
1168 {
1169 /* Values between 0 and 127 */
1170 *v = a;
1171 return 1;
1172 }
1173#endif
1174
1175 /* The 2-byte case */
1176 p++;
1177 b = *p;
1178 /* b: p1 (unmasked) */
1179 if (!(b&0x80))
1180 {
1181 /* Values between 128 and 16383 */
1182 a &= 0x7f;
1183 a = a<<7;
1184 *v = a | b;
1185 return 2;
1186 }
1187
1188 /* The 3-byte case */
1189 p++;
1190 a = a<<14;
1191 a |= *p;
1192 /* a: p0<<14 | p2 (unmasked) */
1193 if (!(a&0x80))
1194 {
1195 /* Values between 16384 and 2097151 */
1196 a &= (0x7f<<14)|(0x7f);
1197 b &= 0x7f;
1198 b = b<<7;
1199 *v = a | b;
1200 return 3;
1201 }
1202
1203 /* A 32-bit varint is used to store size information in btrees.
1204 ** Objects are rarely larger than 2MiB limit of a 3-byte varint.
1205 ** A 3-byte varint is sufficient, for example, to record the size
1206 ** of a 1048569-byte BLOB or string.
1207 **
1208 ** We only unroll the first 1-, 2-, and 3- byte cases. The very
1209 ** rare larger cases can be handled by the slower 64-bit varint
1210 ** routine.
1211 */
1212#if 1
1213 {
1214 u64 v64;
1215 u8 n;
1216
drh15cedda2020-07-02 17:05:11 +00001217 n = sqlite3GetVarint(p-2, &v64);
drhc81c11f2009-11-10 01:30:52 +00001218 assert( n>3 && n<=9 );
1219 if( (v64 & SQLITE_MAX_U32)!=v64 ){
1220 *v = 0xffffffff;
1221 }else{
1222 *v = (u32)v64;
1223 }
1224 return n;
1225 }
1226
1227#else
1228 /* For following code (kept for historical record only) shows an
1229 ** unrolling for the 3- and 4-byte varint cases. This code is
1230 ** slightly faster, but it is also larger and much harder to test.
1231 */
1232 p++;
1233 b = b<<14;
1234 b |= *p;
1235 /* b: p1<<14 | p3 (unmasked) */
1236 if (!(b&0x80))
1237 {
1238 /* Values between 2097152 and 268435455 */
1239 b &= (0x7f<<14)|(0x7f);
1240 a &= (0x7f<<14)|(0x7f);
1241 a = a<<7;
1242 *v = a | b;
1243 return 4;
1244 }
1245
1246 p++;
1247 a = a<<14;
1248 a |= *p;
1249 /* a: p0<<28 | p2<<14 | p4 (unmasked) */
1250 if (!(a&0x80))
1251 {
dan3bbe7612010-03-03 16:02:05 +00001252 /* Values between 268435456 and 34359738367 */
1253 a &= SLOT_4_2_0;
1254 b &= SLOT_4_2_0;
drhc81c11f2009-11-10 01:30:52 +00001255 b = b<<7;
1256 *v = a | b;
1257 return 5;
1258 }
1259
1260 /* We can only reach this point when reading a corrupt database
1261 ** file. In that case we are not in any hurry. Use the (relatively
1262 ** slow) general-purpose sqlite3GetVarint() routine to extract the
1263 ** value. */
1264 {
1265 u64 v64;
1266 u8 n;
1267
1268 p -= 4;
1269 n = sqlite3GetVarint(p, &v64);
1270 assert( n>5 && n<=9 );
1271 *v = (u32)v64;
1272 return n;
1273 }
1274#endif
1275}
1276
1277/*
1278** Return the number of bytes that will be needed to store the given
1279** 64-bit integer.
1280*/
1281int sqlite3VarintLen(u64 v){
drh59a53642015-09-01 22:29:07 +00001282 int i;
drh6f17c092016-03-04 21:18:09 +00001283 for(i=1; (v >>= 7)!=0; i++){ assert( i<10 ); }
drhc81c11f2009-11-10 01:30:52 +00001284 return i;
1285}
1286
1287
1288/*
1289** Read or write a four-byte big-endian integer value.
1290*/
1291u32 sqlite3Get4byte(const u8 *p){
drh5372e4d2015-06-30 12:47:09 +00001292#if SQLITE_BYTEORDER==4321
1293 u32 x;
1294 memcpy(&x,p,4);
1295 return x;
drhdc5ece82017-02-15 15:09:09 +00001296#elif SQLITE_BYTEORDER==1234 && GCC_VERSION>=4003000
drh5372e4d2015-06-30 12:47:09 +00001297 u32 x;
1298 memcpy(&x,p,4);
1299 return __builtin_bswap32(x);
drha39284b2017-02-09 17:12:22 +00001300#elif SQLITE_BYTEORDER==1234 && MSVC_VERSION>=1300
mistachkin647ca462015-06-30 17:28:40 +00001301 u32 x;
1302 memcpy(&x,p,4);
1303 return _byteswap_ulong(x);
drh5372e4d2015-06-30 12:47:09 +00001304#else
drh693e6712014-01-24 22:58:00 +00001305 testcase( p[0]&0x80 );
1306 return ((unsigned)p[0]<<24) | (p[1]<<16) | (p[2]<<8) | p[3];
drh5372e4d2015-06-30 12:47:09 +00001307#endif
drhc81c11f2009-11-10 01:30:52 +00001308}
1309void sqlite3Put4byte(unsigned char *p, u32 v){
drh5372e4d2015-06-30 12:47:09 +00001310#if SQLITE_BYTEORDER==4321
1311 memcpy(p,&v,4);
drhdc5ece82017-02-15 15:09:09 +00001312#elif SQLITE_BYTEORDER==1234 && GCC_VERSION>=4003000
drh5372e4d2015-06-30 12:47:09 +00001313 u32 x = __builtin_bswap32(v);
1314 memcpy(p,&x,4);
drha39284b2017-02-09 17:12:22 +00001315#elif SQLITE_BYTEORDER==1234 && MSVC_VERSION>=1300
mistachkin647ca462015-06-30 17:28:40 +00001316 u32 x = _byteswap_ulong(v);
1317 memcpy(p,&x,4);
drh5372e4d2015-06-30 12:47:09 +00001318#else
drhc81c11f2009-11-10 01:30:52 +00001319 p[0] = (u8)(v>>24);
1320 p[1] = (u8)(v>>16);
1321 p[2] = (u8)(v>>8);
1322 p[3] = (u8)v;
drh5372e4d2015-06-30 12:47:09 +00001323#endif
drhc81c11f2009-11-10 01:30:52 +00001324}
1325
drh9296c182014-07-23 13:40:49 +00001326
1327
1328/*
1329** Translate a single byte of Hex into an integer.
1330** This routine only works if h really is a valid hexadecimal
1331** character: 0..9a..fA..F
1332*/
1333u8 sqlite3HexToInt(int h){
1334 assert( (h>='0' && h<='9') || (h>='a' && h<='f') || (h>='A' && h<='F') );
1335#ifdef SQLITE_ASCII
1336 h += 9*(1&(h>>6));
1337#endif
1338#ifdef SQLITE_EBCDIC
1339 h += 9*(1&~(h>>4));
1340#endif
1341 return (u8)(h & 0xf);
1342}
1343
drhb48c0d52020-02-07 01:12:53 +00001344#if !defined(SQLITE_OMIT_BLOB_LITERAL)
drhc81c11f2009-11-10 01:30:52 +00001345/*
1346** Convert a BLOB literal of the form "x'hhhhhh'" into its binary
1347** value. Return a pointer to its binary value. Space to hold the
1348** binary value has been obtained from malloc and must be freed by
1349** the calling routine.
1350*/
1351void *sqlite3HexToBlob(sqlite3 *db, const char *z, int n){
1352 char *zBlob;
1353 int i;
1354
drh575fad62016-02-05 13:38:36 +00001355 zBlob = (char *)sqlite3DbMallocRawNN(db, n/2 + 1);
drhc81c11f2009-11-10 01:30:52 +00001356 n--;
1357 if( zBlob ){
1358 for(i=0; i<n; i+=2){
dancd74b612011-04-22 19:37:32 +00001359 zBlob[i/2] = (sqlite3HexToInt(z[i])<<4) | sqlite3HexToInt(z[i+1]);
drhc81c11f2009-11-10 01:30:52 +00001360 }
1361 zBlob[i/2] = 0;
1362 }
1363 return zBlob;
1364}
drhb48c0d52020-02-07 01:12:53 +00001365#endif /* !SQLITE_OMIT_BLOB_LITERAL */
drhc81c11f2009-11-10 01:30:52 +00001366
drh413c3d32010-02-23 20:11:56 +00001367/*
1368** Log an error that is an API call on a connection pointer that should
1369** not have been used. The "type" of connection pointer is given as the
1370** argument. The zType is a word like "NULL" or "closed" or "invalid".
1371*/
1372static void logBadConnection(const char *zType){
1373 sqlite3_log(SQLITE_MISUSE,
1374 "API call with %s database connection pointer",
1375 zType
1376 );
1377}
drhc81c11f2009-11-10 01:30:52 +00001378
1379/*
drhc81c11f2009-11-10 01:30:52 +00001380** Check to make sure we have a valid db pointer. This test is not
1381** foolproof but it does provide some measure of protection against
1382** misuse of the interface such as passing in db pointers that are
1383** NULL or which have been previously closed. If this routine returns
1384** 1 it means that the db pointer is valid and 0 if it should not be
1385** dereferenced for any reason. The calling function should invoke
1386** SQLITE_MISUSE immediately.
1387**
1388** sqlite3SafetyCheckOk() requires that the db pointer be valid for
1389** use. sqlite3SafetyCheckSickOrOk() allows a db pointer that failed to
1390** open properly and is not fit for general use but which can be
1391** used as an argument to sqlite3_errmsg() or sqlite3_close().
1392*/
1393int sqlite3SafetyCheckOk(sqlite3 *db){
drh5f9de6e2021-08-07 23:16:52 +00001394 u8 eOpenState;
drh413c3d32010-02-23 20:11:56 +00001395 if( db==0 ){
1396 logBadConnection("NULL");
1397 return 0;
1398 }
drh5f9de6e2021-08-07 23:16:52 +00001399 eOpenState = db->eOpenState;
1400 if( eOpenState!=SQLITE_STATE_OPEN ){
drhe294da02010-02-25 23:44:15 +00001401 if( sqlite3SafetyCheckSickOrOk(db) ){
1402 testcase( sqlite3GlobalConfig.xLog!=0 );
drh413c3d32010-02-23 20:11:56 +00001403 logBadConnection("unopened");
1404 }
drhc81c11f2009-11-10 01:30:52 +00001405 return 0;
1406 }else{
1407 return 1;
1408 }
1409}
1410int sqlite3SafetyCheckSickOrOk(sqlite3 *db){
drh5f9de6e2021-08-07 23:16:52 +00001411 u8 eOpenState;
1412 eOpenState = db->eOpenState;
1413 if( eOpenState!=SQLITE_STATE_SICK &&
1414 eOpenState!=SQLITE_STATE_OPEN &&
1415 eOpenState!=SQLITE_STATE_BUSY ){
drhe294da02010-02-25 23:44:15 +00001416 testcase( sqlite3GlobalConfig.xLog!=0 );
drhaf46dc12010-02-24 21:44:07 +00001417 logBadConnection("invalid");
drh413c3d32010-02-23 20:11:56 +00001418 return 0;
1419 }else{
1420 return 1;
1421 }
drhc81c11f2009-11-10 01:30:52 +00001422}
drh158b9cb2011-03-05 20:59:46 +00001423
1424/*
1425** Attempt to add, substract, or multiply the 64-bit signed value iB against
1426** the other 64-bit signed integer at *pA and store the result in *pA.
1427** Return 0 on success. Or if the operation would have resulted in an
1428** overflow, leave *pA unchanged and return 1.
1429*/
1430int sqlite3AddInt64(i64 *pA, i64 iB){
drhb9772e72017-09-12 13:27:43 +00001431#if GCC_VERSION>=5004000 && !defined(__INTEL_COMPILER)
drh4a477612017-01-03 17:33:43 +00001432 return __builtin_add_overflow(*pA, iB, pA);
1433#else
drh158b9cb2011-03-05 20:59:46 +00001434 i64 iA = *pA;
1435 testcase( iA==0 ); testcase( iA==1 );
1436 testcase( iB==-1 ); testcase( iB==0 );
1437 if( iB>=0 ){
1438 testcase( iA>0 && LARGEST_INT64 - iA == iB );
1439 testcase( iA>0 && LARGEST_INT64 - iA == iB - 1 );
1440 if( iA>0 && LARGEST_INT64 - iA < iB ) return 1;
drh158b9cb2011-03-05 20:59:46 +00001441 }else{
1442 testcase( iA<0 && -(iA + LARGEST_INT64) == iB + 1 );
1443 testcase( iA<0 && -(iA + LARGEST_INT64) == iB + 2 );
1444 if( iA<0 && -(iA + LARGEST_INT64) > iB + 1 ) return 1;
drh158b9cb2011-03-05 20:59:46 +00001445 }
drh53a6eb32014-02-10 12:59:15 +00001446 *pA += iB;
drh158b9cb2011-03-05 20:59:46 +00001447 return 0;
drh4a477612017-01-03 17:33:43 +00001448#endif
drh158b9cb2011-03-05 20:59:46 +00001449}
1450int sqlite3SubInt64(i64 *pA, i64 iB){
drhb9772e72017-09-12 13:27:43 +00001451#if GCC_VERSION>=5004000 && !defined(__INTEL_COMPILER)
drh4a477612017-01-03 17:33:43 +00001452 return __builtin_sub_overflow(*pA, iB, pA);
1453#else
drh158b9cb2011-03-05 20:59:46 +00001454 testcase( iB==SMALLEST_INT64+1 );
1455 if( iB==SMALLEST_INT64 ){
1456 testcase( (*pA)==(-1) ); testcase( (*pA)==0 );
1457 if( (*pA)>=0 ) return 1;
1458 *pA -= iB;
1459 return 0;
1460 }else{
1461 return sqlite3AddInt64(pA, -iB);
1462 }
drh4a477612017-01-03 17:33:43 +00001463#endif
drh158b9cb2011-03-05 20:59:46 +00001464}
drh158b9cb2011-03-05 20:59:46 +00001465int sqlite3MulInt64(i64 *pA, i64 iB){
drhb9772e72017-09-12 13:27:43 +00001466#if GCC_VERSION>=5004000 && !defined(__INTEL_COMPILER)
drh4a477612017-01-03 17:33:43 +00001467 return __builtin_mul_overflow(*pA, iB, pA);
1468#else
drh158b9cb2011-03-05 20:59:46 +00001469 i64 iA = *pA;
drh09952c62016-09-20 22:04:05 +00001470 if( iB>0 ){
1471 if( iA>LARGEST_INT64/iB ) return 1;
1472 if( iA<SMALLEST_INT64/iB ) return 1;
1473 }else if( iB<0 ){
1474 if( iA>0 ){
1475 if( iB<SMALLEST_INT64/iA ) return 1;
1476 }else if( iA<0 ){
1477 if( iB==SMALLEST_INT64 ) return 1;
1478 if( iA==SMALLEST_INT64 ) return 1;
1479 if( -iA>LARGEST_INT64/-iB ) return 1;
drh53a6eb32014-02-10 12:59:15 +00001480 }
drh53a6eb32014-02-10 12:59:15 +00001481 }
drh09952c62016-09-20 22:04:05 +00001482 *pA = iA*iB;
drh158b9cb2011-03-05 20:59:46 +00001483 return 0;
drh4a477612017-01-03 17:33:43 +00001484#endif
drh158b9cb2011-03-05 20:59:46 +00001485}
drhd50ffc42011-03-08 02:38:28 +00001486
1487/*
1488** Compute the absolute value of a 32-bit signed integer, of possible. Or
1489** if the integer has a value of -2147483648, return +2147483647
1490*/
1491int sqlite3AbsInt32(int x){
1492 if( x>=0 ) return x;
drh87e79ae2011-03-08 13:06:41 +00001493 if( x==(int)0x80000000 ) return 0x7fffffff;
drhd50ffc42011-03-08 02:38:28 +00001494 return -x;
1495}
drh81cc5162011-05-17 20:36:21 +00001496
1497#ifdef SQLITE_ENABLE_8_3_NAMES
1498/*
drhb51bf432011-07-21 21:29:35 +00001499** If SQLITE_ENABLE_8_3_NAMES is set at compile-time and if the database
drh81cc5162011-05-17 20:36:21 +00001500** filename in zBaseFilename is a URI with the "8_3_names=1" parameter and
1501** if filename in z[] has a suffix (a.k.a. "extension") that is longer than
1502** three characters, then shorten the suffix on z[] to be the last three
1503** characters of the original suffix.
1504**
drhb51bf432011-07-21 21:29:35 +00001505** If SQLITE_ENABLE_8_3_NAMES is set to 2 at compile-time, then always
1506** do the suffix shortening regardless of URI parameter.
1507**
drh81cc5162011-05-17 20:36:21 +00001508** Examples:
1509**
1510** test.db-journal => test.nal
1511** test.db-wal => test.wal
1512** test.db-shm => test.shm
drhf5808602011-12-16 00:33:04 +00001513** test.db-mj7f3319fa => test.9fa
drh81cc5162011-05-17 20:36:21 +00001514*/
1515void sqlite3FileSuffix3(const char *zBaseFilename, char *z){
drhb51bf432011-07-21 21:29:35 +00001516#if SQLITE_ENABLE_8_3_NAMES<2
drh7d39e172012-01-02 12:41:53 +00001517 if( sqlite3_uri_boolean(zBaseFilename, "8_3_names", 0) )
drhb51bf432011-07-21 21:29:35 +00001518#endif
1519 {
drh81cc5162011-05-17 20:36:21 +00001520 int i, sz;
1521 sz = sqlite3Strlen30(z);
drhc83f2d42011-05-18 02:41:10 +00001522 for(i=sz-1; i>0 && z[i]!='/' && z[i]!='.'; i--){}
drhc02a43a2012-01-10 23:18:38 +00001523 if( z[i]=='.' && ALWAYS(sz>i+4) ) memmove(&z[i+1], &z[sz-3], 4);
drh81cc5162011-05-17 20:36:21 +00001524 }
1525}
1526#endif
drhbf539c42013-10-05 18:16:02 +00001527
1528/*
1529** Find (an approximate) sum of two LogEst values. This computation is
1530** not a simple "+" operator because LogEst is stored as a logarithmic
1531** value.
1532**
1533*/
1534LogEst sqlite3LogEstAdd(LogEst a, LogEst b){
1535 static const unsigned char x[] = {
1536 10, 10, /* 0,1 */
1537 9, 9, /* 2,3 */
1538 8, 8, /* 4,5 */
1539 7, 7, 7, /* 6,7,8 */
1540 6, 6, 6, /* 9,10,11 */
1541 5, 5, 5, /* 12-14 */
1542 4, 4, 4, 4, /* 15-18 */
1543 3, 3, 3, 3, 3, 3, /* 19-24 */
1544 2, 2, 2, 2, 2, 2, 2, /* 25-31 */
1545 };
1546 if( a>=b ){
1547 if( a>b+49 ) return a;
1548 if( a>b+31 ) return a+1;
1549 return a+x[a-b];
1550 }else{
1551 if( b>a+49 ) return b;
1552 if( b>a+31 ) return b+1;
1553 return b+x[b-a];
1554 }
1555}
1556
1557/*
drh224155d2014-04-30 13:19:09 +00001558** Convert an integer into a LogEst. In other words, compute an
1559** approximation for 10*log2(x).
drhbf539c42013-10-05 18:16:02 +00001560*/
1561LogEst sqlite3LogEst(u64 x){
1562 static LogEst a[] = { 0, 2, 3, 5, 6, 7, 8, 9 };
1563 LogEst y = 40;
1564 if( x<8 ){
1565 if( x<2 ) return 0;
1566 while( x<8 ){ y -= 10; x <<= 1; }
1567 }else{
drhceb4b1d2017-08-17 20:53:07 +00001568#if GCC_VERSION>=5004000
1569 int i = 60 - __builtin_clzll(x);
1570 y += i*10;
1571 x >>= i;
1572#else
drh75ab50c2016-04-28 14:15:12 +00001573 while( x>255 ){ y += 40; x >>= 4; } /*OPTIMIZATION-IF-TRUE*/
drhbf539c42013-10-05 18:16:02 +00001574 while( x>15 ){ y += 10; x >>= 1; }
drhceb4b1d2017-08-17 20:53:07 +00001575#endif
drhbf539c42013-10-05 18:16:02 +00001576 }
1577 return a[x&7] + y - 10;
1578}
1579
drhbf539c42013-10-05 18:16:02 +00001580/*
1581** Convert a double into a LogEst
1582** In other words, compute an approximation for 10*log2(x).
1583*/
1584LogEst sqlite3LogEstFromDouble(double x){
1585 u64 a;
1586 LogEst e;
1587 assert( sizeof(x)==8 && sizeof(a)==8 );
1588 if( x<=1 ) return 0;
1589 if( x<=2000000000 ) return sqlite3LogEst((u64)x);
1590 memcpy(&a, &x, 8);
1591 e = (a>>52) - 1022;
1592 return e*10;
1593}
drhbf539c42013-10-05 18:16:02 +00001594
1595/*
1596** Convert a LogEst into an integer.
1597*/
1598u64 sqlite3LogEstToInt(LogEst x){
1599 u64 n;
drhbf539c42013-10-05 18:16:02 +00001600 n = x%10;
1601 x /= 10;
1602 if( n>=5 ) n -= 2;
1603 else if( n>=1 ) n -= 1;
drhecdf20d2016-03-10 14:28:24 +00001604 if( x>60 ) return (u64)LARGEST_INT64;
drhecdf20d2016-03-10 14:28:24 +00001605 return x>=3 ? (n+8)<<(x-3) : (n+8)>>(3-x);
drhbf539c42013-10-05 18:16:02 +00001606}
drh9bf755c2016-12-23 03:59:31 +00001607
1608/*
1609** Add a new name/number pair to a VList. This might require that the
1610** VList object be reallocated, so return the new VList. If an OOM
drhce1bbe52016-12-23 13:52:45 +00001611** error occurs, the original VList returned and the
drh9bf755c2016-12-23 03:59:31 +00001612** db->mallocFailed flag is set.
1613**
1614** A VList is really just an array of integers. To destroy a VList,
1615** simply pass it to sqlite3DbFree().
1616**
1617** The first integer is the number of integers allocated for the whole
1618** VList. The second integer is the number of integers actually used.
1619** Each name/number pair is encoded by subsequent groups of 3 or more
1620** integers.
1621**
drhce1bbe52016-12-23 13:52:45 +00001622** Each name/number pair starts with two integers which are the numeric
drh9bf755c2016-12-23 03:59:31 +00001623** value for the pair and the size of the name/number pair, respectively.
1624** The text name overlays one or more following integers. The text name
1625** is always zero-terminated.
drhce1bbe52016-12-23 13:52:45 +00001626**
1627** Conceptually:
1628**
1629** struct VList {
1630** int nAlloc; // Number of allocated slots
1631** int nUsed; // Number of used slots
1632** struct VListEntry {
1633** int iValue; // Value for this entry
1634** int nSlot; // Slots used by this entry
1635** // ... variable name goes here
1636** } a[0];
1637** }
1638**
1639** During code generation, pointers to the variable names within the
1640** VList are taken. When that happens, nAlloc is set to zero as an
1641** indication that the VList may never again be enlarged, since the
1642** accompanying realloc() would invalidate the pointers.
drh9bf755c2016-12-23 03:59:31 +00001643*/
1644VList *sqlite3VListAdd(
1645 sqlite3 *db, /* The database connection used for malloc() */
1646 VList *pIn, /* The input VList. Might be NULL */
1647 const char *zName, /* Name of symbol to add */
1648 int nName, /* Bytes of text in zName */
1649 int iVal /* Value to associate with zName */
1650){
1651 int nInt; /* number of sizeof(int) objects needed for zName */
drhce1bbe52016-12-23 13:52:45 +00001652 char *z; /* Pointer to where zName will be stored */
1653 int i; /* Index in pIn[] where zName is stored */
drh9bf755c2016-12-23 03:59:31 +00001654
1655 nInt = nName/4 + 3;
drhce1bbe52016-12-23 13:52:45 +00001656 assert( pIn==0 || pIn[0]>=3 ); /* Verify ok to add new elements */
drh9bf755c2016-12-23 03:59:31 +00001657 if( pIn==0 || pIn[1]+nInt > pIn[0] ){
1658 /* Enlarge the allocation */
drh0aa32312019-04-13 04:01:12 +00001659 sqlite3_int64 nAlloc = (pIn ? 2*(sqlite3_int64)pIn[0] : 10) + nInt;
drh9bf755c2016-12-23 03:59:31 +00001660 VList *pOut = sqlite3DbRealloc(db, pIn, nAlloc*sizeof(int));
drhce1bbe52016-12-23 13:52:45 +00001661 if( pOut==0 ) return pIn;
drh9bf755c2016-12-23 03:59:31 +00001662 if( pIn==0 ) pOut[1] = 2;
1663 pIn = pOut;
1664 pIn[0] = nAlloc;
1665 }
1666 i = pIn[1];
1667 pIn[i] = iVal;
1668 pIn[i+1] = nInt;
1669 z = (char*)&pIn[i+2];
1670 pIn[1] = i+nInt;
1671 assert( pIn[1]<=pIn[0] );
1672 memcpy(z, zName, nName);
1673 z[nName] = 0;
1674 return pIn;
1675}
1676
1677/*
1678** Return a pointer to the name of a variable in the given VList that
1679** has the value iVal. Or return a NULL if there is no such variable in
1680** the list
1681*/
1682const char *sqlite3VListNumToName(VList *pIn, int iVal){
1683 int i, mx;
1684 if( pIn==0 ) return 0;
1685 mx = pIn[1];
1686 i = 2;
1687 do{
1688 if( pIn[i]==iVal ) return (char*)&pIn[i+2];
1689 i += pIn[i+1];
1690 }while( i<mx );
1691 return 0;
1692}
1693
1694/*
1695** Return the number of the variable named zName, if it is in VList.
1696** or return 0 if there is no such variable.
1697*/
1698int sqlite3VListNameToNum(VList *pIn, const char *zName, int nName){
1699 int i, mx;
1700 if( pIn==0 ) return 0;
1701 mx = pIn[1];
1702 i = 2;
1703 do{
1704 const char *z = (const char*)&pIn[i+2];
1705 if( strncmp(z,zName,nName)==0 && z[nName]==0 ) return pIn[i];
1706 i += pIn[i+1];
1707 }while( i<mx );
1708 return 0;
1709}