blob: ca8d26e4f917afa6e082c18f3c9546b476cceccf [file] [log] [blame]
drha18c5682000-10-08 22:20:57 +00001/*
2** The "printf" code that follows dates from the 1980's. It is in
drh38b41492015-06-08 15:08:15 +00003** the public domain.
drhe84a3062004-02-02 12:29:25 +00004**
5**************************************************************************
drha18c5682000-10-08 22:20:57 +00006**
drhed1fddf2011-10-12 18:52:59 +00007** This file contains code for a set of "printf"-like routines. These
8** routines format strings much like the printf() from the standard C
9** library, though the implementation here has enhancements to support
mistachkinf5b5f9b2015-06-08 17:42:57 +000010** SQLite.
drha18c5682000-10-08 22:20:57 +000011*/
12#include "sqliteInt.h"
drh7c68d602000-10-11 19:28:51 +000013
drha18c5682000-10-08 22:20:57 +000014/*
drha18c5682000-10-08 22:20:57 +000015** Conversion types fall into various categories as defined by the
16** following enumeration.
17*/
drh2c338a92017-02-10 19:38:36 +000018#define etRADIX 0 /* non-decimal integer types. %x %o */
drhad5a9d72016-05-05 11:53:12 +000019#define etFLOAT 1 /* Floating point. %f */
20#define etEXP 2 /* Exponentional notation. %e and %E */
21#define etGENERIC 3 /* Floating or exponential, depending on exponent. %g */
22#define etSIZE 4 /* Return number of characters processed so far. %n */
23#define etSTRING 5 /* Strings. %s */
24#define etDYNSTRING 6 /* Dynamically allocated strings. %z */
25#define etPERCENT 7 /* Percent symbol. %% */
26#define etCHARX 8 /* Characters. %c */
drha18c5682000-10-08 22:20:57 +000027/* The rest are extensions, not normally found in printf() */
drhad5a9d72016-05-05 11:53:12 +000028#define etSQLESCAPE 9 /* Strings with '\'' doubled. %q */
29#define etSQLESCAPE2 10 /* Strings with '\'' doubled and enclosed in '',
chw0cfcf3f2002-06-16 04:55:48 +000030 NULL pointers replaced by SQL NULL. %Q */
drhad5a9d72016-05-05 11:53:12 +000031#define etTOKEN 11 /* a pointer to a Token structure */
32#define etSRCLIST 12 /* a pointer to a SrcList */
33#define etPOINTER 13 /* The %p conversion */
34#define etSQLESCAPE3 14 /* %w -> Strings with '\"' doubled */
35#define etORDINAL 15 /* %r -> 1st, 2nd, 3rd, 4th, etc. English only */
drh2c338a92017-02-10 19:38:36 +000036#define etDECIMAL 16 /* %d or %u, but not %x, %o */
drhe84a3062004-02-02 12:29:25 +000037
drh2c338a92017-02-10 19:38:36 +000038#define etINVALID 17 /* Any unrecognized conversion type */
drh874ba042009-04-08 16:10:04 +000039
drhe84a3062004-02-02 12:29:25 +000040
41/*
42** An "etByte" is an 8-bit unsigned value.
43*/
44typedef unsigned char etByte;
drha18c5682000-10-08 22:20:57 +000045
46/*
47** Each builtin conversion character (ex: the 'd' in "%d") is described
48** by an instance of the following structure
49*/
50typedef struct et_info { /* Information about each format field */
drhe84a3062004-02-02 12:29:25 +000051 char fmttype; /* The format field code letter */
52 etByte base; /* The base for radix conversion */
53 etByte flags; /* One or more of FLAG_ constants below */
54 etByte type; /* Conversion paradigm */
drh76ff3a02004-09-24 22:32:30 +000055 etByte charset; /* Offset into aDigits[] of the digits string */
56 etByte prefix; /* Offset into aPrefix[] of the prefix string */
drha18c5682000-10-08 22:20:57 +000057} et_info;
58
59/*
drhe84a3062004-02-02 12:29:25 +000060** Allowed values for et_info.flags
61*/
drh2c338a92017-02-10 19:38:36 +000062#define FLAG_SIGNED 1 /* True if the value to convert is signed */
63#define FLAG_STRING 4 /* Allow infinite precision */
drhe84a3062004-02-02 12:29:25 +000064
65
66/*
drha18c5682000-10-08 22:20:57 +000067** The following table is searched linearly, so it is good to put the
68** most frequently used conversion types first.
69*/
drh76ff3a02004-09-24 22:32:30 +000070static const char aDigits[] = "0123456789ABCDEF0123456789abcdef";
71static const char aPrefix[] = "-x0\000X0";
drh57196282004-10-06 15:41:16 +000072static const et_info fmtinfo[] = {
drh2c338a92017-02-10 19:38:36 +000073 { 'd', 10, 1, etDECIMAL, 0, 0 },
drh4794f732004-11-05 17:17:50 +000074 { 's', 0, 4, etSTRING, 0, 0 },
drh557cc602005-08-13 12:59:14 +000075 { 'g', 0, 1, etGENERIC, 30, 0 },
drh153c62c2007-08-24 03:51:33 +000076 { 'z', 0, 4, etDYNSTRING, 0, 0 },
drh4794f732004-11-05 17:17:50 +000077 { 'q', 0, 4, etSQLESCAPE, 0, 0 },
78 { 'Q', 0, 4, etSQLESCAPE2, 0, 0 },
danielk1977f3b863e2007-06-24 06:32:17 +000079 { 'w', 0, 4, etSQLESCAPE3, 0, 0 },
drh76ff3a02004-09-24 22:32:30 +000080 { 'c', 0, 0, etCHARX, 0, 0 },
81 { 'o', 8, 0, etRADIX, 0, 2 },
drh2c338a92017-02-10 19:38:36 +000082 { 'u', 10, 0, etDECIMAL, 0, 0 },
drh76ff3a02004-09-24 22:32:30 +000083 { 'x', 16, 0, etRADIX, 16, 1 },
84 { 'X', 16, 0, etRADIX, 0, 4 },
drhb37df7b2005-10-13 02:09:49 +000085#ifndef SQLITE_OMIT_FLOATING_POINT
drh76ff3a02004-09-24 22:32:30 +000086 { 'f', 0, 1, etFLOAT, 0, 0 },
87 { 'e', 0, 1, etEXP, 30, 0 },
88 { 'E', 0, 1, etEXP, 14, 0 },
drh76ff3a02004-09-24 22:32:30 +000089 { 'G', 0, 1, etGENERIC, 14, 0 },
drhb37df7b2005-10-13 02:09:49 +000090#endif
drh2c338a92017-02-10 19:38:36 +000091 { 'i', 10, 1, etDECIMAL, 0, 0 },
drh76ff3a02004-09-24 22:32:30 +000092 { 'n', 0, 0, etSIZE, 0, 0 },
93 { '%', 0, 0, etPERCENT, 0, 0 },
94 { 'p', 16, 0, etPOINTER, 0, 1 },
drh7e3ff5d2009-04-08 11:49:42 +000095
drh8236f682017-01-04 00:26:28 +000096 /* All the rest are undocumented and are for internal use only */
97 { 'T', 0, 0, etTOKEN, 0, 0 },
98 { 'S', 0, 0, etSRCLIST, 0, 0 },
99 { 'r', 10, 1, etORDINAL, 0, 0 },
drha18c5682000-10-08 22:20:57 +0000100};
drha18c5682000-10-08 22:20:57 +0000101
102/*
drhb37df7b2005-10-13 02:09:49 +0000103** If SQLITE_OMIT_FLOATING_POINT is defined, then none of the floating point
drha18c5682000-10-08 22:20:57 +0000104** conversions will work.
105*/
drhb37df7b2005-10-13 02:09:49 +0000106#ifndef SQLITE_OMIT_FLOATING_POINT
drha18c5682000-10-08 22:20:57 +0000107/*
108** "*val" is a double such that 0.1 <= *val < 10.0
109** Return the ascii code for the leading digit of *val, then
110** multiply "*val" by 10.0 to renormalize.
111**
112** Example:
113** input: *val = 3.14159
114** output: *val = 1.4159 function return = '3'
115**
116** The counter *cnt is incremented each time. After counter exceeds
117** 16 (the number of significant digits in a 64-bit float) '0' is
118** always returned.
119*/
drhea678832008-12-10 19:26:22 +0000120static char et_getdigit(LONGDOUBLE_TYPE *val, int *cnt){
drha18c5682000-10-08 22:20:57 +0000121 int digit;
drh384eef32004-01-07 03:04:27 +0000122 LONGDOUBLE_TYPE d;
drh72b3fbc2012-06-19 03:11:25 +0000123 if( (*cnt)<=0 ) return '0';
124 (*cnt)--;
drha18c5682000-10-08 22:20:57 +0000125 digit = (int)*val;
126 d = digit;
127 digit += '0';
128 *val = (*val - d)*10.0;
drhea678832008-12-10 19:26:22 +0000129 return (char)digit;
drha18c5682000-10-08 22:20:57 +0000130}
drhb37df7b2005-10-13 02:09:49 +0000131#endif /* SQLITE_OMIT_FLOATING_POINT */
drha18c5682000-10-08 22:20:57 +0000132
drh79158e12005-09-06 21:40:45 +0000133/*
drha6353a32013-12-09 19:03:26 +0000134** Set the StrAccum object to an error mode.
135*/
drha5c14162013-12-17 15:03:06 +0000136static void setStrAccumError(StrAccum *p, u8 eError){
drhc0490572015-05-02 11:45:53 +0000137 assert( eError==STRACCUM_NOMEM || eError==STRACCUM_TOOBIG );
drha6353a32013-12-09 19:03:26 +0000138 p->accError = eError;
139 p->nAlloc = 0;
140}
141
142/*
drha5c14162013-12-17 15:03:06 +0000143** Extra argument values from a PrintfArguments object
144*/
145static sqlite3_int64 getIntArg(PrintfArguments *p){
146 if( p->nArg<=p->nUsed ) return 0;
147 return sqlite3_value_int64(p->apArg[p->nUsed++]);
148}
149static double getDoubleArg(PrintfArguments *p){
150 if( p->nArg<=p->nUsed ) return 0.0;
151 return sqlite3_value_double(p->apArg[p->nUsed++]);
152}
153static char *getTextArg(PrintfArguments *p){
154 if( p->nArg<=p->nUsed ) return 0;
155 return (char*)sqlite3_value_text(p->apArg[p->nUsed++]);
156}
157
158
159/*
drh79158e12005-09-06 21:40:45 +0000160** On machines with a small stack size, you can redefine the
drhed1fddf2011-10-12 18:52:59 +0000161** SQLITE_PRINT_BUF_SIZE to be something smaller, if desired.
drh79158e12005-09-06 21:40:45 +0000162*/
163#ifndef SQLITE_PRINT_BUF_SIZE
drh59eedf72011-10-11 17:54:54 +0000164# define SQLITE_PRINT_BUF_SIZE 70
drh79158e12005-09-06 21:40:45 +0000165#endif
166#define etBUFSIZE SQLITE_PRINT_BUF_SIZE /* Size of the output buffer */
drha18c5682000-10-08 22:20:57 +0000167
168/*
drhed1fddf2011-10-12 18:52:59 +0000169** Render a string given by "fmt" into the StrAccum object.
drha18c5682000-10-08 22:20:57 +0000170*/
drhf089aa42008-07-08 19:34:06 +0000171void sqlite3VXPrintf(
drha5c14162013-12-17 15:03:06 +0000172 StrAccum *pAccum, /* Accumulate results here */
drha5c14162013-12-17 15:03:06 +0000173 const char *fmt, /* Format string */
174 va_list ap /* arguments */
drha18c5682000-10-08 22:20:57 +0000175){
drhe84a3062004-02-02 12:29:25 +0000176 int c; /* Next character in the format string */
177 char *bufpt; /* Pointer to the conversion buffer */
178 int precision; /* Precision of the current field */
179 int length; /* Length of the field */
180 int idx; /* A general purpose loop counter */
drhe84a3062004-02-02 12:29:25 +0000181 int width; /* Width of the current field */
182 etByte flag_leftjustify; /* True if "-" flag is present */
drh2c338a92017-02-10 19:38:36 +0000183 etByte flag_prefix; /* '+' or ' ' or 0 for prefix */
drhe84a3062004-02-02 12:29:25 +0000184 etByte flag_alternateform; /* True if "#" flag is present */
drh531fe872005-08-13 13:40:42 +0000185 etByte flag_altform2; /* True if "!" flag is present */
drhe84a3062004-02-02 12:29:25 +0000186 etByte flag_zeropad; /* True if field width constant starts with zero */
drh2c338a92017-02-10 19:38:36 +0000187 etByte flag_long; /* 1 for the "l" flag, 2 for "ll", 0 by default */
drh3e9aeec2005-08-13 13:39:02 +0000188 etByte done; /* Loop termination flag */
drh2c338a92017-02-10 19:38:36 +0000189 etByte cThousand; /* Thousands separator for %d and %u */
drhad5a9d72016-05-05 11:53:12 +0000190 etByte xtype = etINVALID; /* Conversion paradigm */
drha5c14162013-12-17 15:03:06 +0000191 u8 bArgList; /* True for SQLITE_PRINTF_SQLFUNC */
drhed1fddf2011-10-12 18:52:59 +0000192 char prefix; /* Prefix character. "+" or "-" or " " or '\0'. */
drh27436af2006-03-28 23:57:17 +0000193 sqlite_uint64 longvalue; /* Value for integer types */
drh384eef32004-01-07 03:04:27 +0000194 LONGDOUBLE_TYPE realvalue; /* Value for real types */
drh57196282004-10-06 15:41:16 +0000195 const et_info *infop; /* Pointer to the appropriate info structure */
drh59eedf72011-10-11 17:54:54 +0000196 char *zOut; /* Rendering buffer */
197 int nOut; /* Size of the rendering buffer */
drhaf8f5132014-10-29 18:20:18 +0000198 char *zExtra = 0; /* Malloced memory used by some conversion */
drhb37df7b2005-10-13 02:09:49 +0000199#ifndef SQLITE_OMIT_FLOATING_POINT
drh557cc602005-08-13 12:59:14 +0000200 int exp, e2; /* exponent of real numbers */
drhed1fddf2011-10-12 18:52:59 +0000201 int nsd; /* Number of significant digits returned */
drhe84a3062004-02-02 12:29:25 +0000202 double rounder; /* Used for rounding floating point values */
203 etByte flag_dp; /* True if decimal point should be shown */
204 etByte flag_rtz; /* True if trailing zeros should be removed */
drha18c5682000-10-08 22:20:57 +0000205#endif
drha5c14162013-12-17 15:03:06 +0000206 PrintfArguments *pArgList = 0; /* Arguments for SQLITE_PRINTF_SQLFUNC */
drhed1fddf2011-10-12 18:52:59 +0000207 char buf[etBUFSIZE]; /* Conversion buffer */
drha18c5682000-10-08 22:20:57 +0000208
drha18c5682000-10-08 22:20:57 +0000209 bufpt = 0;
drh8236f682017-01-04 00:26:28 +0000210 if( (pAccum->printfFlags & SQLITE_PRINTF_SQLFUNC)!=0 ){
211 pArgList = va_arg(ap, PrintfArguments*);
212 bArgList = 1;
drha5c14162013-12-17 15:03:06 +0000213 }else{
drh8236f682017-01-04 00:26:28 +0000214 bArgList = 0;
drha5c14162013-12-17 15:03:06 +0000215 }
drha18c5682000-10-08 22:20:57 +0000216 for(; (c=(*fmt))!=0; ++fmt){
217 if( c!='%' ){
drha18c5682000-10-08 22:20:57 +0000218 bufpt = (char *)fmt;
drh760b1592014-09-18 01:50:09 +0000219#if HAVE_STRCHRNUL
220 fmt = strchrnul(fmt, '%');
221#else
222 do{ fmt++; }while( *fmt && *fmt != '%' );
223#endif
drha70a0732014-03-17 14:24:27 +0000224 sqlite3StrAccumAppend(pAccum, bufpt, (int)(fmt - bufpt));
drh760b1592014-09-18 01:50:09 +0000225 if( *fmt==0 ) break;
drha18c5682000-10-08 22:20:57 +0000226 }
227 if( (c=(*++fmt))==0 ){
drhade86482007-11-28 22:36:40 +0000228 sqlite3StrAccumAppend(pAccum, "%", 1);
drha18c5682000-10-08 22:20:57 +0000229 break;
230 }
231 /* Find out what flags are present */
drh2c338a92017-02-10 19:38:36 +0000232 flag_leftjustify = flag_prefix = cThousand =
drh557cc602005-08-13 12:59:14 +0000233 flag_alternateform = flag_altform2 = flag_zeropad = 0;
drh3e9aeec2005-08-13 13:39:02 +0000234 done = 0;
drha18c5682000-10-08 22:20:57 +0000235 do{
236 switch( c ){
drh3e9aeec2005-08-13 13:39:02 +0000237 case '-': flag_leftjustify = 1; break;
drh2c338a92017-02-10 19:38:36 +0000238 case '+': flag_prefix = '+'; break;
239 case ' ': flag_prefix = ' '; break;
drh3e9aeec2005-08-13 13:39:02 +0000240 case '#': flag_alternateform = 1; break;
241 case '!': flag_altform2 = 1; break;
242 case '0': flag_zeropad = 1; break;
drh2c338a92017-02-10 19:38:36 +0000243 case ',': cThousand = ','; break;
drh3e9aeec2005-08-13 13:39:02 +0000244 default: done = 1; break;
drha18c5682000-10-08 22:20:57 +0000245 }
drh3e9aeec2005-08-13 13:39:02 +0000246 }while( !done && (c=(*++fmt))!=0 );
drha18c5682000-10-08 22:20:57 +0000247 /* Get the field width */
drha18c5682000-10-08 22:20:57 +0000248 if( c=='*' ){
drha5c14162013-12-17 15:03:06 +0000249 if( bArgList ){
250 width = (int)getIntArg(pArgList);
251 }else{
252 width = va_arg(ap,int);
253 }
drha18c5682000-10-08 22:20:57 +0000254 if( width<0 ){
255 flag_leftjustify = 1;
drhb6f47de2015-04-07 15:39:29 +0000256 width = width >= -2147483647 ? -width : 0;
drha18c5682000-10-08 22:20:57 +0000257 }
258 c = *++fmt;
259 }else{
drhb6f47de2015-04-07 15:39:29 +0000260 unsigned wx = 0;
drh17a68932001-01-31 13:28:08 +0000261 while( c>='0' && c<='9' ){
drhb6f47de2015-04-07 15:39:29 +0000262 wx = wx*10 + c - '0';
drha18c5682000-10-08 22:20:57 +0000263 c = *++fmt;
264 }
drhb6f47de2015-04-07 15:39:29 +0000265 testcase( wx>0x7fffffff );
266 width = wx & 0x7fffffff;
drha18c5682000-10-08 22:20:57 +0000267 }
drhc386ef42015-11-30 22:52:14 +0000268 assert( width>=0 );
269#ifdef SQLITE_PRINTF_PRECISION_LIMIT
270 if( width>SQLITE_PRINTF_PRECISION_LIMIT ){
271 width = SQLITE_PRINTF_PRECISION_LIMIT;
272 }
273#endif
dan8c069142015-04-07 14:38:57 +0000274
drha18c5682000-10-08 22:20:57 +0000275 /* Get the precision */
276 if( c=='.' ){
drha18c5682000-10-08 22:20:57 +0000277 c = *++fmt;
278 if( c=='*' ){
drha5c14162013-12-17 15:03:06 +0000279 if( bArgList ){
280 precision = (int)getIntArg(pArgList);
281 }else{
282 precision = va_arg(ap,int);
283 }
drha18c5682000-10-08 22:20:57 +0000284 c = *++fmt;
drhb6f47de2015-04-07 15:39:29 +0000285 if( precision<0 ){
286 precision = precision >= -2147483647 ? -precision : -1;
287 }
drha18c5682000-10-08 22:20:57 +0000288 }else{
drhb6f47de2015-04-07 15:39:29 +0000289 unsigned px = 0;
drh17a68932001-01-31 13:28:08 +0000290 while( c>='0' && c<='9' ){
drhb6f47de2015-04-07 15:39:29 +0000291 px = px*10 + c - '0';
drha18c5682000-10-08 22:20:57 +0000292 c = *++fmt;
293 }
drhb6f47de2015-04-07 15:39:29 +0000294 testcase( px>0x7fffffff );
295 precision = px & 0x7fffffff;
drha18c5682000-10-08 22:20:57 +0000296 }
drha18c5682000-10-08 22:20:57 +0000297 }else{
298 precision = -1;
299 }
drhc386ef42015-11-30 22:52:14 +0000300 assert( precision>=(-1) );
301#ifdef SQLITE_PRINTF_PRECISION_LIMIT
302 if( precision>SQLITE_PRINTF_PRECISION_LIMIT ){
303 precision = SQLITE_PRINTF_PRECISION_LIMIT;
304 }
305#endif
306
307
drha18c5682000-10-08 22:20:57 +0000308 /* Get the conversion type modifier */
309 if( c=='l' ){
310 flag_long = 1;
311 c = *++fmt;
drha34b6762004-05-07 13:30:42 +0000312 if( c=='l' ){
drh2c338a92017-02-10 19:38:36 +0000313 flag_long = 2;
drha34b6762004-05-07 13:30:42 +0000314 c = *++fmt;
drha34b6762004-05-07 13:30:42 +0000315 }
drha18c5682000-10-08 22:20:57 +0000316 }else{
drh2c338a92017-02-10 19:38:36 +0000317 flag_long = 0;
drha18c5682000-10-08 22:20:57 +0000318 }
319 /* Fetch the info entry for the field */
drh874ba042009-04-08 16:10:04 +0000320 infop = &fmtinfo[0];
321 xtype = etINVALID;
danielk197700e13612008-11-17 19:18:54 +0000322 for(idx=0; idx<ArraySize(fmtinfo); idx++){
drha18c5682000-10-08 22:20:57 +0000323 if( c==fmtinfo[idx].fmttype ){
324 infop = &fmtinfo[idx];
drh8236f682017-01-04 00:26:28 +0000325 xtype = infop->type;
drha18c5682000-10-08 22:20:57 +0000326 break;
327 }
328 }
drh43617e92006-03-06 20:55:46 +0000329
drha18c5682000-10-08 22:20:57 +0000330 /*
331 ** At this point, variables are initialized as follows:
332 **
333 ** flag_alternateform TRUE if a '#' is present.
drh3e9aeec2005-08-13 13:39:02 +0000334 ** flag_altform2 TRUE if a '!' is present.
drh2c338a92017-02-10 19:38:36 +0000335 ** flag_prefix '+' or ' ' or zero
drha18c5682000-10-08 22:20:57 +0000336 ** flag_leftjustify TRUE if a '-' is present or if the
337 ** field width was negative.
338 ** flag_zeropad TRUE if the width began with 0.
drh2c338a92017-02-10 19:38:36 +0000339 ** flag_long 1 for "l", 2 for "ll"
drha18c5682000-10-08 22:20:57 +0000340 ** width The specified field width. This is
341 ** always non-negative. Zero is the default.
342 ** precision The specified precision. The default
343 ** is -1.
344 ** xtype The class of the conversion.
345 ** infop Pointer to the appropriate info struct.
346 */
347 switch( xtype ){
drhfe63d1c2004-09-08 20:13:04 +0000348 case etPOINTER:
drh2c338a92017-02-10 19:38:36 +0000349 flag_long = sizeof(char*)==sizeof(i64) ? 2 :
350 sizeof(char*)==sizeof(long int) ? 1 : 0;
drhfe63d1c2004-09-08 20:13:04 +0000351 /* Fall through into the next case */
drh9a993342007-12-13 02:45:31 +0000352 case etORDINAL:
drh2c338a92017-02-10 19:38:36 +0000353 case etRADIX:
354 cThousand = 0;
355 /* Fall through into the next case */
356 case etDECIMAL:
drhe84a3062004-02-02 12:29:25 +0000357 if( infop->flags & FLAG_SIGNED ){
drhe9707672004-06-25 01:10:48 +0000358 i64 v;
drha5c14162013-12-17 15:03:06 +0000359 if( bArgList ){
360 v = getIntArg(pArgList);
drheeb23a42009-05-04 20:20:16 +0000361 }else if( flag_long ){
drh2c338a92017-02-10 19:38:36 +0000362 if( flag_long==2 ){
363 v = va_arg(ap,i64) ;
364 }else{
365 v = va_arg(ap,long int);
366 }
drheeb23a42009-05-04 20:20:16 +0000367 }else{
368 v = va_arg(ap,int);
369 }
drhe9707672004-06-25 01:10:48 +0000370 if( v<0 ){
drh158b9cb2011-03-05 20:59:46 +0000371 if( v==SMALLEST_INT64 ){
372 longvalue = ((u64)1)<<63;
373 }else{
374 longvalue = -v;
375 }
drhe9707672004-06-25 01:10:48 +0000376 prefix = '-';
danielk1977cfcdaef2004-05-12 07:33:33 +0000377 }else{
drhe9707672004-06-25 01:10:48 +0000378 longvalue = v;
drh2c338a92017-02-10 19:38:36 +0000379 prefix = flag_prefix;
danielk1977cfcdaef2004-05-12 07:33:33 +0000380 }
drhe9707672004-06-25 01:10:48 +0000381 }else{
drha5c14162013-12-17 15:03:06 +0000382 if( bArgList ){
383 longvalue = (u64)getIntArg(pArgList);
drheeb23a42009-05-04 20:20:16 +0000384 }else if( flag_long ){
drh2c338a92017-02-10 19:38:36 +0000385 if( flag_long==2 ){
386 longvalue = va_arg(ap,u64);
387 }else{
388 longvalue = va_arg(ap,unsigned long int);
389 }
drheeb23a42009-05-04 20:20:16 +0000390 }else{
391 longvalue = va_arg(ap,unsigned int);
392 }
drhe9707672004-06-25 01:10:48 +0000393 prefix = 0;
394 }
395 if( longvalue==0 ) flag_alternateform = 0;
drha18c5682000-10-08 22:20:57 +0000396 if( flag_zeropad && precision<width-(prefix!=0) ){
397 precision = width-(prefix!=0);
drh9adf9ac2002-05-15 11:44:13 +0000398 }
drh2c338a92017-02-10 19:38:36 +0000399 if( precision<etBUFSIZE-10-etBUFSIZE/3 ){
drh59eedf72011-10-11 17:54:54 +0000400 nOut = etBUFSIZE;
401 zOut = buf;
402 }else{
drh5f429952017-03-20 16:34:18 +0000403 u64 n = (u64)precision + 10 + precision/3;
404 zOut = zExtra = sqlite3Malloc( n );
drh59eedf72011-10-11 17:54:54 +0000405 if( zOut==0 ){
drha6353a32013-12-09 19:03:26 +0000406 setStrAccumError(pAccum, STRACCUM_NOMEM);
drh59eedf72011-10-11 17:54:54 +0000407 return;
408 }
drh5f429952017-03-20 16:34:18 +0000409 nOut = (int)n;
drh59eedf72011-10-11 17:54:54 +0000410 }
411 bufpt = &zOut[nOut-1];
drh9a993342007-12-13 02:45:31 +0000412 if( xtype==etORDINAL ){
drh43f6e062007-12-13 17:50:22 +0000413 static const char zOrd[] = "thstndrd";
drhea678832008-12-10 19:26:22 +0000414 int x = (int)(longvalue % 10);
drh43f6e062007-12-13 17:50:22 +0000415 if( x>=4 || (longvalue/10)%10==1 ){
416 x = 0;
417 }
drh59eedf72011-10-11 17:54:54 +0000418 *(--bufpt) = zOrd[x*2+1];
419 *(--bufpt) = zOrd[x*2];
drh9a993342007-12-13 02:45:31 +0000420 }
drha18c5682000-10-08 22:20:57 +0000421 {
drh0e682092014-03-17 15:06:57 +0000422 const char *cset = &aDigits[infop->charset];
423 u8 base = infop->base;
drha18c5682000-10-08 22:20:57 +0000424 do{ /* Convert to ascii */
425 *(--bufpt) = cset[longvalue%base];
426 longvalue = longvalue/base;
427 }while( longvalue>0 );
drh9adf9ac2002-05-15 11:44:13 +0000428 }
drh59eedf72011-10-11 17:54:54 +0000429 length = (int)(&zOut[nOut-1]-bufpt);
drh2c338a92017-02-10 19:38:36 +0000430 while( precision>length ){
drha18c5682000-10-08 22:20:57 +0000431 *(--bufpt) = '0'; /* Zero pad */
drh2c338a92017-02-10 19:38:36 +0000432 length++;
433 }
434 if( cThousand ){
435 int nn = (length - 1)/3; /* Number of "," to insert */
436 int ix = (length - 1)%3 + 1;
437 bufpt -= nn;
438 for(idx=0; nn>0; idx++){
439 bufpt[idx] = bufpt[idx+nn];
440 ix--;
441 if( ix==0 ){
442 bufpt[++idx] = cThousand;
443 nn--;
444 ix = 3;
445 }
446 }
drh9adf9ac2002-05-15 11:44:13 +0000447 }
drha18c5682000-10-08 22:20:57 +0000448 if( prefix ) *(--bufpt) = prefix; /* Add sign */
449 if( flag_alternateform && infop->prefix ){ /* Add "0" or "0x" */
drh76ff3a02004-09-24 22:32:30 +0000450 const char *pre;
451 char x;
452 pre = &aPrefix[infop->prefix];
drhaf005fb2008-07-09 16:51:51 +0000453 for(; (x=(*pre))!=0; pre++) *(--bufpt) = x;
drha18c5682000-10-08 22:20:57 +0000454 }
drh59eedf72011-10-11 17:54:54 +0000455 length = (int)(&zOut[nOut-1]-bufpt);
drha18c5682000-10-08 22:20:57 +0000456 break;
457 case etFLOAT:
458 case etEXP:
459 case etGENERIC:
drha5c14162013-12-17 15:03:06 +0000460 if( bArgList ){
461 realvalue = getDoubleArg(pArgList);
462 }else{
463 realvalue = va_arg(ap,double);
464 }
drh69ef7032010-03-04 17:11:31 +0000465#ifdef SQLITE_OMIT_FLOATING_POINT
466 length = 0;
467#else
drha18c5682000-10-08 22:20:57 +0000468 if( precision<0 ) precision = 6; /* Set default precision */
drha18c5682000-10-08 22:20:57 +0000469 if( realvalue<0.0 ){
470 realvalue = -realvalue;
471 prefix = '-';
drh9adf9ac2002-05-15 11:44:13 +0000472 }else{
drh2c338a92017-02-10 19:38:36 +0000473 prefix = flag_prefix;
drh9adf9ac2002-05-15 11:44:13 +0000474 }
drh3e9aeec2005-08-13 13:39:02 +0000475 if( xtype==etGENERIC && precision>0 ) precision--;
drha30d22a2015-04-07 13:28:41 +0000476 testcase( precision>0xfff );
drh74b42272015-04-07 12:41:17 +0000477 for(idx=precision&0xfff, rounder=0.5; idx>0; idx--, rounder*=0.1){}
drh3e9aeec2005-08-13 13:39:02 +0000478 if( xtype==etFLOAT ) realvalue += rounder;
drha18c5682000-10-08 22:20:57 +0000479 /* Normalize realvalue to within 10.0 > realvalue >= 1.0 */
480 exp = 0;
drhea678832008-12-10 19:26:22 +0000481 if( sqlite3IsNaN((double)realvalue) ){
drh53c14022007-05-10 17:23:11 +0000482 bufpt = "NaN";
483 length = 3;
484 break;
485 }
drha18c5682000-10-08 22:20:57 +0000486 if( realvalue>0.0 ){
drh72b3fbc2012-06-19 03:11:25 +0000487 LONGDOUBLE_TYPE scale = 1.0;
drh4ef94132012-06-19 03:35:05 +0000488 while( realvalue>=1e100*scale && exp<=350 ){ scale *= 1e100;exp+=100;}
drh2a8f6712015-09-02 21:00:48 +0000489 while( realvalue>=1e10*scale && exp<=350 ){ scale *= 1e10; exp+=10; }
drh72b3fbc2012-06-19 03:11:25 +0000490 while( realvalue>=10.0*scale && exp<=350 ){ scale *= 10.0; exp++; }
491 realvalue /= scale;
drh93a960a2008-07-10 00:32:42 +0000492 while( realvalue<1e-8 ){ realvalue *= 1e8; exp-=8; }
493 while( realvalue<1.0 ){ realvalue *= 10.0; exp--; }
drhaf005fb2008-07-09 16:51:51 +0000494 if( exp>350 ){
drh2a8f6712015-09-02 21:00:48 +0000495 bufpt = buf;
496 buf[0] = prefix;
497 memcpy(buf+(prefix!=0),"Inf",4);
498 length = 3+(prefix!=0);
drha18c5682000-10-08 22:20:57 +0000499 break;
500 }
drh9adf9ac2002-05-15 11:44:13 +0000501 }
drha18c5682000-10-08 22:20:57 +0000502 bufpt = buf;
503 /*
504 ** If the field type is etGENERIC, then convert to either etEXP
505 ** or etFLOAT, as appropriate.
506 */
drha18c5682000-10-08 22:20:57 +0000507 if( xtype!=etFLOAT ){
508 realvalue += rounder;
509 if( realvalue>=10.0 ){ realvalue *= 0.1; exp++; }
510 }
511 if( xtype==etGENERIC ){
512 flag_rtz = !flag_alternateform;
513 if( exp<-4 || exp>precision ){
514 xtype = etEXP;
515 }else{
516 precision = precision - exp;
517 xtype = etFLOAT;
518 }
drh9adf9ac2002-05-15 11:44:13 +0000519 }else{
drh72b3fbc2012-06-19 03:11:25 +0000520 flag_rtz = flag_altform2;
drh9adf9ac2002-05-15 11:44:13 +0000521 }
drh557cc602005-08-13 12:59:14 +0000522 if( xtype==etEXP ){
523 e2 = 0;
524 }else{
525 e2 = exp;
526 }
drh74b42272015-04-07 12:41:17 +0000527 if( MAX(e2,0)+(i64)precision+(i64)width > etBUFSIZE - 15 ){
528 bufpt = zExtra
529 = sqlite3Malloc( MAX(e2,0)+(i64)precision+(i64)width+15 );
drh59eedf72011-10-11 17:54:54 +0000530 if( bufpt==0 ){
drha6353a32013-12-09 19:03:26 +0000531 setStrAccumError(pAccum, STRACCUM_NOMEM);
drh59eedf72011-10-11 17:54:54 +0000532 return;
533 }
534 }
535 zOut = bufpt;
drh72b3fbc2012-06-19 03:11:25 +0000536 nsd = 16 + flag_altform2*10;
drhea678832008-12-10 19:26:22 +0000537 flag_dp = (precision>0 ?1:0) | flag_alternateform | flag_altform2;
drh557cc602005-08-13 12:59:14 +0000538 /* The sign in front of the number */
539 if( prefix ){
540 *(bufpt++) = prefix;
541 }
542 /* Digits prior to the decimal point */
543 if( e2<0 ){
544 *(bufpt++) = '0';
545 }else{
546 for(; e2>=0; e2--){
547 *(bufpt++) = et_getdigit(&realvalue,&nsd);
drha18c5682000-10-08 22:20:57 +0000548 }
drh9adf9ac2002-05-15 11:44:13 +0000549 }
drh557cc602005-08-13 12:59:14 +0000550 /* The decimal point */
551 if( flag_dp ){
552 *(bufpt++) = '.';
553 }
554 /* "0" digits after the decimal point but before the first
555 ** significant digit of the number */
drhaf005fb2008-07-09 16:51:51 +0000556 for(e2++; e2<0; precision--, e2++){
557 assert( precision>0 );
drh557cc602005-08-13 12:59:14 +0000558 *(bufpt++) = '0';
559 }
560 /* Significant digits after the decimal point */
561 while( (precision--)>0 ){
562 *(bufpt++) = et_getdigit(&realvalue,&nsd);
563 }
564 /* Remove trailing zeros and the "." if no digits follow the "." */
565 if( flag_rtz && flag_dp ){
drh3e9aeec2005-08-13 13:39:02 +0000566 while( bufpt[-1]=='0' ) *(--bufpt) = 0;
drh59eedf72011-10-11 17:54:54 +0000567 assert( bufpt>zOut );
drh3e9aeec2005-08-13 13:39:02 +0000568 if( bufpt[-1]=='.' ){
drh557cc602005-08-13 12:59:14 +0000569 if( flag_altform2 ){
570 *(bufpt++) = '0';
571 }else{
572 *(--bufpt) = 0;
573 }
574 }
575 }
576 /* Add the "eNNN" suffix */
drh59eedf72011-10-11 17:54:54 +0000577 if( xtype==etEXP ){
drh557cc602005-08-13 12:59:14 +0000578 *(bufpt++) = aDigits[infop->charset];
579 if( exp<0 ){
580 *(bufpt++) = '-'; exp = -exp;
581 }else{
582 *(bufpt++) = '+';
583 }
584 if( exp>=100 ){
drhea678832008-12-10 19:26:22 +0000585 *(bufpt++) = (char)((exp/100)+'0'); /* 100's digit */
drh557cc602005-08-13 12:59:14 +0000586 exp %= 100;
587 }
drhea678832008-12-10 19:26:22 +0000588 *(bufpt++) = (char)(exp/10+'0'); /* 10's digit */
589 *(bufpt++) = (char)(exp%10+'0'); /* 1's digit */
drh557cc602005-08-13 12:59:14 +0000590 }
591 *bufpt = 0;
592
drha18c5682000-10-08 22:20:57 +0000593 /* The converted number is in buf[] and zero terminated. Output it.
594 ** Note that the number is in the usual order, not reversed as with
595 ** integer conversions. */
drh59eedf72011-10-11 17:54:54 +0000596 length = (int)(bufpt-zOut);
597 bufpt = zOut;
drha18c5682000-10-08 22:20:57 +0000598
599 /* Special case: Add leading zeros if the flag_zeropad flag is
600 ** set and we are not left justified */
601 if( flag_zeropad && !flag_leftjustify && length < width){
602 int i;
603 int nPad = width - length;
604 for(i=width; i>=nPad; i--){
605 bufpt[i] = bufpt[i-nPad];
606 }
607 i = prefix!=0;
608 while( nPad-- ) bufpt[i++] = '0';
609 length = width;
610 }
drh69ef7032010-03-04 17:11:31 +0000611#endif /* !defined(SQLITE_OMIT_FLOATING_POINT) */
drha18c5682000-10-08 22:20:57 +0000612 break;
613 case etSIZE:
drhfc6ee9d2013-12-17 15:58:42 +0000614 if( !bArgList ){
615 *(va_arg(ap,int*)) = pAccum->nChar;
616 }
drha18c5682000-10-08 22:20:57 +0000617 length = width = 0;
618 break;
619 case etPERCENT:
620 buf[0] = '%';
621 bufpt = buf;
622 length = 1;
623 break;
drha18c5682000-10-08 22:20:57 +0000624 case etCHARX:
drha5c14162013-12-17 15:03:06 +0000625 if( bArgList ){
drhfc6ee9d2013-12-17 15:58:42 +0000626 bufpt = getTextArg(pArgList);
627 c = bufpt ? bufpt[0] : 0;
drha5c14162013-12-17 15:03:06 +0000628 }else{
629 c = va_arg(ap,int);
630 }
drhaf8f5132014-10-29 18:20:18 +0000631 if( precision>1 ){
632 width -= precision-1;
633 if( width>1 && !flag_leftjustify ){
634 sqlite3AppendChar(pAccum, width-1, ' ');
635 width = 0;
636 }
637 sqlite3AppendChar(pAccum, precision-1, c);
drh9adf9ac2002-05-15 11:44:13 +0000638 }
drhaf8f5132014-10-29 18:20:18 +0000639 length = 1;
640 buf[0] = c;
drha18c5682000-10-08 22:20:57 +0000641 bufpt = buf;
642 break;
643 case etSTRING:
drhd93d8a82003-06-16 03:08:18 +0000644 case etDYNSTRING:
drha5c14162013-12-17 15:03:06 +0000645 if( bArgList ){
646 bufpt = getTextArg(pArgList);
drh2a8f6712015-09-02 21:00:48 +0000647 xtype = etSTRING;
drha5c14162013-12-17 15:03:06 +0000648 }else{
649 bufpt = va_arg(ap,char*);
650 }
drhd93d8a82003-06-16 03:08:18 +0000651 if( bufpt==0 ){
652 bufpt = "";
drh2a8f6712015-09-02 21:00:48 +0000653 }else if( xtype==etDYNSTRING ){
drhd93d8a82003-06-16 03:08:18 +0000654 zExtra = bufpt;
655 }
drhe5090942008-04-29 15:22:27 +0000656 if( precision>=0 ){
657 for(length=0; length<precision && bufpt[length]; length++){}
658 }else{
drhc84ddf12017-08-19 20:38:18 +0000659 length = 0x7fffffff & (int)strlen(bufpt);
drhe5090942008-04-29 15:22:27 +0000660 }
drha18c5682000-10-08 22:20:57 +0000661 break;
drh2a8f6712015-09-02 21:00:48 +0000662 case etSQLESCAPE: /* Escape ' characters */
663 case etSQLESCAPE2: /* Escape ' and enclose in '...' */
664 case etSQLESCAPE3: { /* Escape " characters */
drh8965b502009-11-25 16:53:37 +0000665 int i, j, k, n, isnull;
drh5eba8c02005-08-19 02:26:27 +0000666 int needQuote;
drhea678832008-12-10 19:26:22 +0000667 char ch;
danielk1977f3b863e2007-06-24 06:32:17 +0000668 char q = ((xtype==etSQLESCAPE3)?'"':'\''); /* Quote character */
drha5c14162013-12-17 15:03:06 +0000669 char *escarg;
670
671 if( bArgList ){
672 escarg = getTextArg(pArgList);
673 }else{
674 escarg = va_arg(ap,char*);
675 }
danielk1977f0113002006-01-24 12:09:17 +0000676 isnull = escarg==0;
677 if( isnull ) escarg = (xtype==etSQLESCAPE2 ? "NULL" : "(NULL)");
drh8965b502009-11-25 16:53:37 +0000678 k = precision;
dan60d4a302010-03-04 17:58:45 +0000679 for(i=n=0; k!=0 && (ch=escarg[i])!=0; i++, k--){
danielk1977f3b863e2007-06-24 06:32:17 +0000680 if( ch==q ) n++;
drha18c5682000-10-08 22:20:57 +0000681 }
drh5eba8c02005-08-19 02:26:27 +0000682 needQuote = !isnull && xtype==etSQLESCAPE2;
drh2a8f6712015-09-02 21:00:48 +0000683 n += i + 3;
drh5eba8c02005-08-19 02:26:27 +0000684 if( n>etBUFSIZE ){
drhe5ae5732008-06-15 02:51:47 +0000685 bufpt = zExtra = sqlite3Malloc( n );
drhd164fd32008-11-20 18:20:28 +0000686 if( bufpt==0 ){
drha6353a32013-12-09 19:03:26 +0000687 setStrAccumError(pAccum, STRACCUM_NOMEM);
drhd164fd32008-11-20 18:20:28 +0000688 return;
689 }
drh5eba8c02005-08-19 02:26:27 +0000690 }else{
691 bufpt = buf;
692 }
693 j = 0;
danielk1977f3b863e2007-06-24 06:32:17 +0000694 if( needQuote ) bufpt[j++] = q;
drh8965b502009-11-25 16:53:37 +0000695 k = i;
696 for(i=0; i<k; i++){
697 bufpt[j++] = ch = escarg[i];
danielk1977f3b863e2007-06-24 06:32:17 +0000698 if( ch==q ) bufpt[j++] = ch;
drh5eba8c02005-08-19 02:26:27 +0000699 }
danielk1977f3b863e2007-06-24 06:32:17 +0000700 if( needQuote ) bufpt[j++] = q;
drh5eba8c02005-08-19 02:26:27 +0000701 bufpt[j] = 0;
702 length = j;
drh8965b502009-11-25 16:53:37 +0000703 /* The precision in %q and %Q means how many input characters to
704 ** consume, not the length of the output...
705 ** if( precision>=0 && precision<length ) length = precision; */
drha18c5682000-10-08 22:20:57 +0000706 break;
drh5eba8c02005-08-19 02:26:27 +0000707 }
drh5f968432004-02-21 19:02:30 +0000708 case etTOKEN: {
drh8236f682017-01-04 00:26:28 +0000709 Token *pToken;
710 if( (pAccum->printfFlags & SQLITE_PRINTF_INTERNAL)==0 ) return;
711 pToken = va_arg(ap, Token*);
drha5c14162013-12-17 15:03:06 +0000712 assert( bArgList==0 );
drha9ab4812013-12-11 11:00:44 +0000713 if( pToken && pToken->n ){
drhade86482007-11-28 22:36:40 +0000714 sqlite3StrAccumAppend(pAccum, (const char*)pToken->z, pToken->n);
drhad6d9462004-09-19 02:15:24 +0000715 }
drh5f968432004-02-21 19:02:30 +0000716 length = width = 0;
717 break;
718 }
719 case etSRCLIST: {
drh8236f682017-01-04 00:26:28 +0000720 SrcList *pSrc;
721 int k;
722 struct SrcList_item *pItem;
723 if( (pAccum->printfFlags & SQLITE_PRINTF_INTERNAL)==0 ) return;
724 pSrc = va_arg(ap, SrcList*);
725 k = va_arg(ap, int);
726 pItem = &pSrc->a[k];
drha5c14162013-12-17 15:03:06 +0000727 assert( bArgList==0 );
drh5f968432004-02-21 19:02:30 +0000728 assert( k>=0 && k<pSrc->nSrc );
drh93a960a2008-07-10 00:32:42 +0000729 if( pItem->zDatabase ){
drha6353a32013-12-09 19:03:26 +0000730 sqlite3StrAccumAppendAll(pAccum, pItem->zDatabase);
drhade86482007-11-28 22:36:40 +0000731 sqlite3StrAccumAppend(pAccum, ".", 1);
drh5f968432004-02-21 19:02:30 +0000732 }
drha6353a32013-12-09 19:03:26 +0000733 sqlite3StrAccumAppendAll(pAccum, pItem->zName);
drh5f968432004-02-21 19:02:30 +0000734 length = width = 0;
735 break;
736 }
drh874ba042009-04-08 16:10:04 +0000737 default: {
738 assert( xtype==etINVALID );
739 return;
740 }
drha18c5682000-10-08 22:20:57 +0000741 }/* End switch over the format type */
742 /*
743 ** The text of the conversion is pointed to by "bufpt" and is
744 ** "length" characters long. The field width is "width". Do
745 ** the output.
746 */
drha70a0732014-03-17 14:24:27 +0000747 width -= length;
drh8236f682017-01-04 00:26:28 +0000748 if( width>0 ){
749 if( !flag_leftjustify ) sqlite3AppendChar(pAccum, width, ' ');
750 sqlite3StrAccumAppend(pAccum, bufpt, length);
751 if( flag_leftjustify ) sqlite3AppendChar(pAccum, width, ' ');
752 }else{
753 sqlite3StrAccumAppend(pAccum, bufpt, length);
754 }
drha70a0732014-03-17 14:24:27 +0000755
drhaf8f5132014-10-29 18:20:18 +0000756 if( zExtra ){
drh96ceaf82015-11-14 22:04:22 +0000757 sqlite3DbFree(pAccum->db, zExtra);
drhaf8f5132014-10-29 18:20:18 +0000758 zExtra = 0;
759 }
drha18c5682000-10-08 22:20:57 +0000760 }/* End for loop over the format string */
drha18c5682000-10-08 22:20:57 +0000761} /* End of function */
762
drhade86482007-11-28 22:36:40 +0000763/*
drha70a0732014-03-17 14:24:27 +0000764** Enlarge the memory allocation on a StrAccum object so that it is
765** able to accept at least N more bytes of text.
766**
767** Return the number of bytes of text that StrAccum is able to accept
768** after the attempted enlargement. The value returned might be zero.
769*/
770static int sqlite3StrAccumEnlarge(StrAccum *p, int N){
771 char *zNew;
drha30d22a2015-04-07 13:28:41 +0000772 assert( p->nChar+(i64)N >= p->nAlloc ); /* Only called if really needed */
drha70a0732014-03-17 14:24:27 +0000773 if( p->accError ){
774 testcase(p->accError==STRACCUM_TOOBIG);
775 testcase(p->accError==STRACCUM_NOMEM);
776 return 0;
777 }
drhc0490572015-05-02 11:45:53 +0000778 if( p->mxAlloc==0 ){
drha70a0732014-03-17 14:24:27 +0000779 N = p->nAlloc - p->nChar - 1;
780 setStrAccumError(p, STRACCUM_TOOBIG);
781 return N;
782 }else{
drh5f4a6862016-01-30 12:50:25 +0000783 char *zOld = isMalloced(p) ? p->zText : 0;
drha70a0732014-03-17 14:24:27 +0000784 i64 szNew = p->nChar;
785 szNew += N + 1;
drh7b4d7802014-11-03 14:46:29 +0000786 if( szNew+p->nChar<=p->mxAlloc ){
787 /* Force exponential buffer size growth as long as it does not overflow,
788 ** to avoid having to call this routine too often */
789 szNew += p->nChar;
790 }
drha70a0732014-03-17 14:24:27 +0000791 if( szNew > p->mxAlloc ){
792 sqlite3StrAccumReset(p);
793 setStrAccumError(p, STRACCUM_TOOBIG);
794 return 0;
795 }else{
796 p->nAlloc = (int)szNew;
797 }
drhc0490572015-05-02 11:45:53 +0000798 if( p->db ){
drha70a0732014-03-17 14:24:27 +0000799 zNew = sqlite3DbRealloc(p->db, zOld, p->nAlloc);
800 }else{
drhf3cdcdc2015-04-29 16:50:28 +0000801 zNew = sqlite3_realloc64(zOld, p->nAlloc);
drha70a0732014-03-17 14:24:27 +0000802 }
803 if( zNew ){
drh7ef4d1c2014-05-31 15:39:53 +0000804 assert( p->zText!=0 || p->nChar==0 );
drh5f4a6862016-01-30 12:50:25 +0000805 if( !isMalloced(p) && p->nChar>0 ) memcpy(zNew, p->zText, p->nChar);
drha70a0732014-03-17 14:24:27 +0000806 p->zText = zNew;
drh7f5a7ec2014-11-03 13:24:12 +0000807 p->nAlloc = sqlite3DbMallocSize(p->db, zNew);
drh5f4a6862016-01-30 12:50:25 +0000808 p->printfFlags |= SQLITE_PRINTF_MALLOCED;
drha70a0732014-03-17 14:24:27 +0000809 }else{
810 sqlite3StrAccumReset(p);
811 setStrAccumError(p, STRACCUM_NOMEM);
812 return 0;
813 }
814 }
815 return N;
816}
817
818/*
drhaf8f5132014-10-29 18:20:18 +0000819** Append N copies of character c to the given string buffer.
drha70a0732014-03-17 14:24:27 +0000820*/
drhaf8f5132014-10-29 18:20:18 +0000821void sqlite3AppendChar(StrAccum *p, int N, char c){
drha30d22a2015-04-07 13:28:41 +0000822 testcase( p->nChar + (i64)N > 0x7fffffff );
823 if( p->nChar+(i64)N >= p->nAlloc && (N = sqlite3StrAccumEnlarge(p, N))<=0 ){
824 return;
825 }
drhaf8f5132014-10-29 18:20:18 +0000826 while( (N--)>0 ) p->zText[p->nChar++] = c;
drha70a0732014-03-17 14:24:27 +0000827}
828
829/*
830** The StrAccum "p" is not large enough to accept N new bytes of z[].
831** So enlarge if first, then do the append.
832**
833** This is a helper routine to sqlite3StrAccumAppend() that does special-case
834** work (enlarging the buffer) using tail recursion, so that the
835** sqlite3StrAccumAppend() routine can use fast calling semantics.
836*/
drh172087f2014-08-22 15:40:20 +0000837static void SQLITE_NOINLINE enlargeAndAppend(StrAccum *p, const char *z, int N){
drha70a0732014-03-17 14:24:27 +0000838 N = sqlite3StrAccumEnlarge(p, N);
839 if( N>0 ){
840 memcpy(&p->zText[p->nChar], z, N);
841 p->nChar += N;
842 }
843}
844
845/*
846** Append N bytes of text from z to the StrAccum object. Increase the
847** size of the memory allocation for StrAccum if necessary.
drha18c5682000-10-08 22:20:57 +0000848*/
drhade86482007-11-28 22:36:40 +0000849void sqlite3StrAccumAppend(StrAccum *p, const char *z, int N){
drh34573382015-04-15 05:38:35 +0000850 assert( z!=0 || N==0 );
drha6353a32013-12-09 19:03:26 +0000851 assert( p->zText!=0 || p->nChar==0 || p->accError );
852 assert( N>=0 );
853 assert( p->accError==0 || p->nAlloc==0 );
drhade86482007-11-28 22:36:40 +0000854 if( p->nChar+N >= p->nAlloc ){
drha70a0732014-03-17 14:24:27 +0000855 enlargeAndAppend(p,z,N);
dan895decf2016-12-30 14:15:56 +0000856 }else if( N ){
drh172087f2014-08-22 15:40:20 +0000857 assert( p->zText );
858 p->nChar += N;
859 memcpy(&p->zText[p->nChar-N], z, N);
drh5f968432004-02-21 19:02:30 +0000860 }
drh483750b2003-01-29 18:46:51 +0000861}
862
863/*
drha6353a32013-12-09 19:03:26 +0000864** Append the complete text of zero-terminated string z[] to the p string.
865*/
866void sqlite3StrAccumAppendAll(StrAccum *p, const char *z){
mistachkin53cd9642013-12-11 02:21:19 +0000867 sqlite3StrAccumAppend(p, z, sqlite3Strlen30(z));
drha6353a32013-12-09 19:03:26 +0000868}
869
870
871/*
drhade86482007-11-28 22:36:40 +0000872** Finish off a string by making sure it is zero-terminated.
873** Return a pointer to the resulting string. Return a NULL
874** pointer if any kind of error was encountered.
drh5f968432004-02-21 19:02:30 +0000875*/
drh043e5862016-11-25 15:11:26 +0000876static SQLITE_NOINLINE char *strAccumFinishRealloc(StrAccum *p){
drh3f18e6d2017-08-12 02:01:55 +0000877 char *zText;
drh043e5862016-11-25 15:11:26 +0000878 assert( p->mxAlloc>0 && !isMalloced(p) );
drh3f18e6d2017-08-12 02:01:55 +0000879 zText = sqlite3DbMallocRaw(p->db, p->nChar+1 );
880 if( zText ){
881 memcpy(zText, p->zText, p->nChar+1);
drh043e5862016-11-25 15:11:26 +0000882 p->printfFlags |= SQLITE_PRINTF_MALLOCED;
883 }else{
884 setStrAccumError(p, STRACCUM_NOMEM);
885 }
drh3f18e6d2017-08-12 02:01:55 +0000886 p->zText = zText;
887 return zText;
drh043e5862016-11-25 15:11:26 +0000888}
drhade86482007-11-28 22:36:40 +0000889char *sqlite3StrAccumFinish(StrAccum *p){
890 if( p->zText ){
891 p->zText[p->nChar] = 0;
drh5f4a6862016-01-30 12:50:25 +0000892 if( p->mxAlloc>0 && !isMalloced(p) ){
drh043e5862016-11-25 15:11:26 +0000893 return strAccumFinishRealloc(p);
drhade86482007-11-28 22:36:40 +0000894 }
895 }
896 return p->zText;
897}
898
899/*
900** Reset an StrAccum string. Reclaim all malloced memory.
901*/
902void sqlite3StrAccumReset(StrAccum *p){
drh5f4a6862016-01-30 12:50:25 +0000903 if( isMalloced(p) ){
drhc0490572015-05-02 11:45:53 +0000904 sqlite3DbFree(p->db, p->zText);
drh5f4a6862016-01-30 12:50:25 +0000905 p->printfFlags &= ~SQLITE_PRINTF_MALLOCED;
drhade86482007-11-28 22:36:40 +0000906 }
drhf089aa42008-07-08 19:34:06 +0000907 p->zText = 0;
drhade86482007-11-28 22:36:40 +0000908}
909
910/*
drhc0490572015-05-02 11:45:53 +0000911** Initialize a string accumulator.
912**
913** p: The accumulator to be initialized.
914** db: Pointer to a database connection. May be NULL. Lookaside
915** memory is used if not NULL. db->mallocFailed is set appropriately
916** when not NULL.
917** zBase: An initial buffer. May be NULL in which case the initial buffer
918** is malloced.
919** n: Size of zBase in bytes. If total space requirements never exceed
920** n then no memory allocations ever occur.
921** mx: Maximum number of bytes to accumulate. If mx==0 then no memory
922** allocations will ever occur.
drhade86482007-11-28 22:36:40 +0000923*/
drhc0490572015-05-02 11:45:53 +0000924void sqlite3StrAccumInit(StrAccum *p, sqlite3 *db, char *zBase, int n, int mx){
drh3f18e6d2017-08-12 02:01:55 +0000925 p->zText = zBase;
drhc0490572015-05-02 11:45:53 +0000926 p->db = db;
drhade86482007-11-28 22:36:40 +0000927 p->nAlloc = n;
drhbb4957f2008-03-20 14:03:29 +0000928 p->mxAlloc = mx;
drh3f18e6d2017-08-12 02:01:55 +0000929 p->nChar = 0;
drhb49bc862013-08-21 21:12:10 +0000930 p->accError = 0;
drh5f4a6862016-01-30 12:50:25 +0000931 p->printfFlags = 0;
drh5f968432004-02-21 19:02:30 +0000932}
933
934/*
935** Print into memory obtained from sqliteMalloc(). Use the internal
936** %-conversion extensions.
937*/
drh17435752007-08-16 04:30:38 +0000938char *sqlite3VMPrintf(sqlite3 *db, const char *zFormat, va_list ap){
939 char *z;
drh79158e12005-09-06 21:40:45 +0000940 char zBase[SQLITE_PRINT_BUF_SIZE];
drhade86482007-11-28 22:36:40 +0000941 StrAccum acc;
drhbc6160b2009-04-08 15:45:31 +0000942 assert( db!=0 );
drhc0490572015-05-02 11:45:53 +0000943 sqlite3StrAccumInit(&acc, db, zBase, sizeof(zBase),
drhbc6160b2009-04-08 15:45:31 +0000944 db->aLimit[SQLITE_LIMIT_LENGTH]);
drh5f4a6862016-01-30 12:50:25 +0000945 acc.printfFlags = SQLITE_PRINTF_INTERNAL;
946 sqlite3VXPrintf(&acc, zFormat, ap);
drhade86482007-11-28 22:36:40 +0000947 z = sqlite3StrAccumFinish(&acc);
drhb49bc862013-08-21 21:12:10 +0000948 if( acc.accError==STRACCUM_NOMEM ){
drh4a642b62016-02-05 01:55:27 +0000949 sqlite3OomFault(db);
drh17435752007-08-16 04:30:38 +0000950 }
951 return z;
drh5f968432004-02-21 19:02:30 +0000952}
953
954/*
955** Print into memory obtained from sqliteMalloc(). Use the internal
956** %-conversion extensions.
957*/
drh17435752007-08-16 04:30:38 +0000958char *sqlite3MPrintf(sqlite3 *db, const char *zFormat, ...){
drh5f968432004-02-21 19:02:30 +0000959 va_list ap;
960 char *z;
drh5f968432004-02-21 19:02:30 +0000961 va_start(ap, zFormat);
drhade86482007-11-28 22:36:40 +0000962 z = sqlite3VMPrintf(db, zFormat, ap);
drh5f968432004-02-21 19:02:30 +0000963 va_end(ap);
964 return z;
965}
966
967/*
drh28dd4792006-06-26 21:35:44 +0000968** Print into memory obtained from sqlite3_malloc(). Omit the internal
969** %-conversion extensions.
970*/
971char *sqlite3_vmprintf(const char *zFormat, va_list ap){
drhade86482007-11-28 22:36:40 +0000972 char *z;
drh28dd4792006-06-26 21:35:44 +0000973 char zBase[SQLITE_PRINT_BUF_SIZE];
drhade86482007-11-28 22:36:40 +0000974 StrAccum acc;
drh9ca95732014-10-24 00:35:58 +0000975
976#ifdef SQLITE_ENABLE_API_ARMOR
977 if( zFormat==0 ){
978 (void)SQLITE_MISUSE_BKPT;
979 return 0;
980 }
981#endif
drhff1590e2008-07-14 12:52:53 +0000982#ifndef SQLITE_OMIT_AUTOINIT
983 if( sqlite3_initialize() ) return 0;
984#endif
drhc0490572015-05-02 11:45:53 +0000985 sqlite3StrAccumInit(&acc, 0, zBase, sizeof(zBase), SQLITE_MAX_LENGTH);
drh5f4a6862016-01-30 12:50:25 +0000986 sqlite3VXPrintf(&acc, zFormat, ap);
drhade86482007-11-28 22:36:40 +0000987 z = sqlite3StrAccumFinish(&acc);
988 return z;
drh28dd4792006-06-26 21:35:44 +0000989}
990
991/*
992** Print into memory obtained from sqlite3_malloc()(). Omit the internal
993** %-conversion extensions.
drh483750b2003-01-29 18:46:51 +0000994*/
danielk19776f8a5032004-05-10 10:34:51 +0000995char *sqlite3_mprintf(const char *zFormat, ...){
drha18c5682000-10-08 22:20:57 +0000996 va_list ap;
drh5f968432004-02-21 19:02:30 +0000997 char *z;
drhff1590e2008-07-14 12:52:53 +0000998#ifndef SQLITE_OMIT_AUTOINIT
999 if( sqlite3_initialize() ) return 0;
1000#endif
drh28dd4792006-06-26 21:35:44 +00001001 va_start(ap, zFormat);
drhb3738b62007-03-31 15:02:49 +00001002 z = sqlite3_vmprintf(zFormat, ap);
drha18c5682000-10-08 22:20:57 +00001003 va_end(ap);
drh5f968432004-02-21 19:02:30 +00001004 return z;
drha18c5682000-10-08 22:20:57 +00001005}
1006
drh93a5c6b2003-12-23 02:17:35 +00001007/*
danielk19776f8a5032004-05-10 10:34:51 +00001008** sqlite3_snprintf() works like snprintf() except that it ignores the
drh93a5c6b2003-12-23 02:17:35 +00001009** current locale settings. This is important for SQLite because we
1010** are not able to use a "," as the decimal point in place of "." as
1011** specified by some locales.
drhdb26d4c2011-01-05 12:20:09 +00001012**
1013** Oops: The first two arguments of sqlite3_snprintf() are backwards
1014** from the snprintf() standard. Unfortunately, it is too late to change
1015** this without breaking compatibility, so we just have to live with the
1016** mistake.
1017**
1018** sqlite3_vsnprintf() is the varargs version.
drh93a5c6b2003-12-23 02:17:35 +00001019*/
drhdb26d4c2011-01-05 12:20:09 +00001020char *sqlite3_vsnprintf(int n, char *zBuf, const char *zFormat, va_list ap){
1021 StrAccum acc;
1022 if( n<=0 ) return zBuf;
drh9ca95732014-10-24 00:35:58 +00001023#ifdef SQLITE_ENABLE_API_ARMOR
1024 if( zBuf==0 || zFormat==0 ) {
1025 (void)SQLITE_MISUSE_BKPT;
drh96c707a2015-02-13 16:36:14 +00001026 if( zBuf ) zBuf[0] = 0;
drh9ca95732014-10-24 00:35:58 +00001027 return zBuf;
1028 }
1029#endif
drhc0490572015-05-02 11:45:53 +00001030 sqlite3StrAccumInit(&acc, 0, zBuf, n, 0);
drh5f4a6862016-01-30 12:50:25 +00001031 sqlite3VXPrintf(&acc, zFormat, ap);
drhe9bb5662016-11-25 15:47:53 +00001032 zBuf[acc.nChar] = 0;
1033 return zBuf;
drhdb26d4c2011-01-05 12:20:09 +00001034}
danielk19776f8a5032004-05-10 10:34:51 +00001035char *sqlite3_snprintf(int n, char *zBuf, const char *zFormat, ...){
drh5f968432004-02-21 19:02:30 +00001036 char *z;
drh93a5c6b2003-12-23 02:17:35 +00001037 va_list ap;
drh93a5c6b2003-12-23 02:17:35 +00001038 va_start(ap,zFormat);
drhdb26d4c2011-01-05 12:20:09 +00001039 z = sqlite3_vsnprintf(n, zBuf, zFormat, ap);
drh93a5c6b2003-12-23 02:17:35 +00001040 va_end(ap);
drh5f968432004-02-21 19:02:30 +00001041 return z;
drh93a5c6b2003-12-23 02:17:35 +00001042}
1043
drh3f280702010-02-18 18:45:09 +00001044/*
drh7c0c4602010-03-03 22:25:18 +00001045** This is the routine that actually formats the sqlite3_log() message.
1046** We house it in a separate routine from sqlite3_log() to avoid using
1047** stack space on small-stack systems when logging is disabled.
1048**
1049** sqlite3_log() must render into a static buffer. It cannot dynamically
1050** allocate memory because it might be called while the memory allocator
1051** mutex is held.
drha8dbd522015-07-14 22:43:37 +00001052**
1053** sqlite3VXPrintf() might ask for *temporary* memory allocations for
1054** certain format characters (%q) or for very large precisions or widths.
1055** Care must be taken that any sqlite3_log() calls that occur while the
1056** memory mutex is held do not use these mechanisms.
drh7c0c4602010-03-03 22:25:18 +00001057*/
1058static void renderLogMsg(int iErrCode, const char *zFormat, va_list ap){
drha64fa912010-03-04 00:53:32 +00001059 StrAccum acc; /* String accumulator */
1060 char zMsg[SQLITE_PRINT_BUF_SIZE*3]; /* Complete log message */
drh7c0c4602010-03-03 22:25:18 +00001061
drhc0490572015-05-02 11:45:53 +00001062 sqlite3StrAccumInit(&acc, 0, zMsg, sizeof(zMsg), 0);
drh5f4a6862016-01-30 12:50:25 +00001063 sqlite3VXPrintf(&acc, zFormat, ap);
drh7c0c4602010-03-03 22:25:18 +00001064 sqlite3GlobalConfig.xLog(sqlite3GlobalConfig.pLogArg, iErrCode,
1065 sqlite3StrAccumFinish(&acc));
1066}
1067
1068/*
drh3f280702010-02-18 18:45:09 +00001069** Format and write a message to the log if logging is enabled.
1070*/
drha7564662010-02-22 19:32:31 +00001071void sqlite3_log(int iErrCode, const char *zFormat, ...){
drh3f280702010-02-18 18:45:09 +00001072 va_list ap; /* Vararg list */
drh7c0c4602010-03-03 22:25:18 +00001073 if( sqlite3GlobalConfig.xLog ){
drh3f280702010-02-18 18:45:09 +00001074 va_start(ap, zFormat);
drh7c0c4602010-03-03 22:25:18 +00001075 renderLogMsg(iErrCode, zFormat, ap);
drh3f280702010-02-18 18:45:09 +00001076 va_end(ap);
drh3f280702010-02-18 18:45:09 +00001077 }
1078}
1079
mistachkin02b0e262015-04-16 03:37:19 +00001080#if defined(SQLITE_DEBUG) || defined(SQLITE_HAVE_OS_TRACE)
drhe54ca3f2004-06-07 01:52:14 +00001081/*
1082** A version of printf() that understands %lld. Used for debugging.
1083** The printf() built into some versions of windows does not understand %lld
1084** and segfaults if you give it a long long int.
1085*/
1086void sqlite3DebugPrintf(const char *zFormat, ...){
1087 va_list ap;
drhade86482007-11-28 22:36:40 +00001088 StrAccum acc;
drhe54ca3f2004-06-07 01:52:14 +00001089 char zBuf[500];
drhc0490572015-05-02 11:45:53 +00001090 sqlite3StrAccumInit(&acc, 0, zBuf, sizeof(zBuf), 0);
drhade86482007-11-28 22:36:40 +00001091 va_start(ap,zFormat);
drh5f4a6862016-01-30 12:50:25 +00001092 sqlite3VXPrintf(&acc, zFormat, ap);
drhe54ca3f2004-06-07 01:52:14 +00001093 va_end(ap);
drhbed8e7e2007-12-08 17:55:35 +00001094 sqlite3StrAccumFinish(&acc);
mistachkin4a9ff912017-11-09 17:29:04 +00001095#ifdef SQLITE_OS_TRACE_PROC
1096 {
1097 extern void SQLITE_OS_TRACE_PROC(const char *zBuf, int nBuf);
1098 SQLITE_OS_TRACE_PROC(zBuf, sizeof(zBuf));
1099 }
1100#else
drh485f0032007-01-26 19:23:33 +00001101 fprintf(stdout,"%s", zBuf);
drh2ac3ee92004-06-07 16:27:46 +00001102 fflush(stdout);
mistachkin4a9ff912017-11-09 17:29:04 +00001103#endif
drhe54ca3f2004-06-07 01:52:14 +00001104}
1105#endif
drhc7bc4fd2009-11-25 18:03:42 +00001106
drh4fa4a542014-09-30 12:33:33 +00001107
drhc7bc4fd2009-11-25 18:03:42 +00001108/*
drhd37bea52015-09-02 15:37:50 +00001109** variable-argument wrapper around sqlite3VXPrintf(). The bFlags argument
1110** can contain the bit SQLITE_PRINTF_INTERNAL enable internal formats.
drhc7bc4fd2009-11-25 18:03:42 +00001111*/
drh5f4a6862016-01-30 12:50:25 +00001112void sqlite3XPrintf(StrAccum *p, const char *zFormat, ...){
drhc7bc4fd2009-11-25 18:03:42 +00001113 va_list ap;
1114 va_start(ap,zFormat);
drh5f4a6862016-01-30 12:50:25 +00001115 sqlite3VXPrintf(p, zFormat, ap);
drhc7bc4fd2009-11-25 18:03:42 +00001116 va_end(ap);
1117}