blob: edf2210e0a5ae49abf6a50d98ed55a0754a9f707 [file] [log] [blame]
drha18c5682000-10-08 22:20:57 +00001/*
2** The "printf" code that follows dates from the 1980's. It is in
3** the public domain. The original comments are included here for
drhe84a3062004-02-02 12:29:25 +00004** completeness. They are very out-of-date but might be useful as
5** an historical reference. Most of the "enhancements" have been backed
6** out so that the functionality is now the same as standard printf().
7**
8**************************************************************************
drha18c5682000-10-08 22:20:57 +00009**
drhed1fddf2011-10-12 18:52:59 +000010** This file contains code for a set of "printf"-like routines. These
11** routines format strings much like the printf() from the standard C
12** library, though the implementation here has enhancements to support
13** SQLlite.
drha18c5682000-10-08 22:20:57 +000014*/
15#include "sqliteInt.h"
drh7c68d602000-10-11 19:28:51 +000016
drha18c5682000-10-08 22:20:57 +000017/*
drha18c5682000-10-08 22:20:57 +000018** Conversion types fall into various categories as defined by the
19** following enumeration.
20*/
drhe84a3062004-02-02 12:29:25 +000021#define etRADIX 1 /* Integer types. %d, %x, %o, and so forth */
22#define etFLOAT 2 /* Floating point. %f */
23#define etEXP 3 /* Exponentional notation. %e and %E */
24#define etGENERIC 4 /* Floating or exponential, depending on exponent. %g */
25#define etSIZE 5 /* Return number of characters processed so far. %n */
26#define etSTRING 6 /* Strings. %s */
27#define etDYNSTRING 7 /* Dynamically allocated strings. %z */
28#define etPERCENT 8 /* Percent symbol. %% */
29#define etCHARX 9 /* Characters. %c */
drha18c5682000-10-08 22:20:57 +000030/* The rest are extensions, not normally found in printf() */
drhaf005fb2008-07-09 16:51:51 +000031#define etSQLESCAPE 10 /* Strings with '\'' doubled. %q */
32#define etSQLESCAPE2 11 /* Strings with '\'' doubled and enclosed in '',
chw0cfcf3f2002-06-16 04:55:48 +000033 NULL pointers replaced by SQL NULL. %Q */
drhaf005fb2008-07-09 16:51:51 +000034#define etTOKEN 12 /* a pointer to a Token structure */
35#define etSRCLIST 13 /* a pointer to a SrcList */
36#define etPOINTER 14 /* The %p conversion */
37#define etSQLESCAPE3 15 /* %w -> Strings with '\"' doubled */
38#define etORDINAL 16 /* %r -> 1st, 2nd, 3rd, 4th, etc. English only */
drhe84a3062004-02-02 12:29:25 +000039
drh874ba042009-04-08 16:10:04 +000040#define etINVALID 0 /* Any unrecognized conversion type */
41
drhe84a3062004-02-02 12:29:25 +000042
43/*
44** An "etByte" is an 8-bit unsigned value.
45*/
46typedef unsigned char etByte;
drha18c5682000-10-08 22:20:57 +000047
48/*
49** Each builtin conversion character (ex: the 'd' in "%d") is described
50** by an instance of the following structure
51*/
52typedef struct et_info { /* Information about each format field */
drhe84a3062004-02-02 12:29:25 +000053 char fmttype; /* The format field code letter */
54 etByte base; /* The base for radix conversion */
55 etByte flags; /* One or more of FLAG_ constants below */
56 etByte type; /* Conversion paradigm */
drh76ff3a02004-09-24 22:32:30 +000057 etByte charset; /* Offset into aDigits[] of the digits string */
58 etByte prefix; /* Offset into aPrefix[] of the prefix string */
drha18c5682000-10-08 22:20:57 +000059} et_info;
60
61/*
drhe84a3062004-02-02 12:29:25 +000062** Allowed values for et_info.flags
63*/
64#define FLAG_SIGNED 1 /* True if the value to convert is signed */
65#define FLAG_INTERN 2 /* True if for internal use only */
drh4794f732004-11-05 17:17:50 +000066#define FLAG_STRING 4 /* Allow infinity precision */
drhe84a3062004-02-02 12:29:25 +000067
68
69/*
drha18c5682000-10-08 22:20:57 +000070** The following table is searched linearly, so it is good to put the
71** most frequently used conversion types first.
72*/
drh76ff3a02004-09-24 22:32:30 +000073static const char aDigits[] = "0123456789ABCDEF0123456789abcdef";
74static const char aPrefix[] = "-x0\000X0";
drh57196282004-10-06 15:41:16 +000075static const et_info fmtinfo[] = {
drh76ff3a02004-09-24 22:32:30 +000076 { 'd', 10, 1, etRADIX, 0, 0 },
drh4794f732004-11-05 17:17:50 +000077 { 's', 0, 4, etSTRING, 0, 0 },
drh557cc602005-08-13 12:59:14 +000078 { 'g', 0, 1, etGENERIC, 30, 0 },
drh153c62c2007-08-24 03:51:33 +000079 { 'z', 0, 4, etDYNSTRING, 0, 0 },
drh4794f732004-11-05 17:17:50 +000080 { 'q', 0, 4, etSQLESCAPE, 0, 0 },
81 { 'Q', 0, 4, etSQLESCAPE2, 0, 0 },
danielk1977f3b863e2007-06-24 06:32:17 +000082 { 'w', 0, 4, etSQLESCAPE3, 0, 0 },
drh76ff3a02004-09-24 22:32:30 +000083 { 'c', 0, 0, etCHARX, 0, 0 },
84 { 'o', 8, 0, etRADIX, 0, 2 },
85 { 'u', 10, 0, etRADIX, 0, 0 },
86 { 'x', 16, 0, etRADIX, 16, 1 },
87 { 'X', 16, 0, etRADIX, 0, 4 },
drhb37df7b2005-10-13 02:09:49 +000088#ifndef SQLITE_OMIT_FLOATING_POINT
drh76ff3a02004-09-24 22:32:30 +000089 { 'f', 0, 1, etFLOAT, 0, 0 },
90 { 'e', 0, 1, etEXP, 30, 0 },
91 { 'E', 0, 1, etEXP, 14, 0 },
drh76ff3a02004-09-24 22:32:30 +000092 { 'G', 0, 1, etGENERIC, 14, 0 },
drhb37df7b2005-10-13 02:09:49 +000093#endif
drh76ff3a02004-09-24 22:32:30 +000094 { 'i', 10, 1, etRADIX, 0, 0 },
95 { 'n', 0, 0, etSIZE, 0, 0 },
96 { '%', 0, 0, etPERCENT, 0, 0 },
97 { 'p', 16, 0, etPOINTER, 0, 1 },
drh7e3ff5d2009-04-08 11:49:42 +000098
99/* All the rest have the FLAG_INTERN bit set and are thus for internal
100** use only */
drh76ff3a02004-09-24 22:32:30 +0000101 { 'T', 0, 2, etTOKEN, 0, 0 },
102 { 'S', 0, 2, etSRCLIST, 0, 0 },
drh9a993342007-12-13 02:45:31 +0000103 { 'r', 10, 3, etORDINAL, 0, 0 },
drha18c5682000-10-08 22:20:57 +0000104};
drha18c5682000-10-08 22:20:57 +0000105
106/*
drhb37df7b2005-10-13 02:09:49 +0000107** If SQLITE_OMIT_FLOATING_POINT is defined, then none of the floating point
drha18c5682000-10-08 22:20:57 +0000108** conversions will work.
109*/
drhb37df7b2005-10-13 02:09:49 +0000110#ifndef SQLITE_OMIT_FLOATING_POINT
drha18c5682000-10-08 22:20:57 +0000111/*
112** "*val" is a double such that 0.1 <= *val < 10.0
113** Return the ascii code for the leading digit of *val, then
114** multiply "*val" by 10.0 to renormalize.
115**
116** Example:
117** input: *val = 3.14159
118** output: *val = 1.4159 function return = '3'
119**
120** The counter *cnt is incremented each time. After counter exceeds
121** 16 (the number of significant digits in a 64-bit float) '0' is
122** always returned.
123*/
drhea678832008-12-10 19:26:22 +0000124static char et_getdigit(LONGDOUBLE_TYPE *val, int *cnt){
drha18c5682000-10-08 22:20:57 +0000125 int digit;
drh384eef32004-01-07 03:04:27 +0000126 LONGDOUBLE_TYPE d;
drh72b3fbc2012-06-19 03:11:25 +0000127 if( (*cnt)<=0 ) return '0';
128 (*cnt)--;
drha18c5682000-10-08 22:20:57 +0000129 digit = (int)*val;
130 d = digit;
131 digit += '0';
132 *val = (*val - d)*10.0;
drhea678832008-12-10 19:26:22 +0000133 return (char)digit;
drha18c5682000-10-08 22:20:57 +0000134}
drhb37df7b2005-10-13 02:09:49 +0000135#endif /* SQLITE_OMIT_FLOATING_POINT */
drha18c5682000-10-08 22:20:57 +0000136
drh79158e12005-09-06 21:40:45 +0000137/*
drha6353a32013-12-09 19:03:26 +0000138** Set the StrAccum object to an error mode.
139*/
drha5c14162013-12-17 15:03:06 +0000140static void setStrAccumError(StrAccum *p, u8 eError){
drhc0490572015-05-02 11:45:53 +0000141 assert( eError==STRACCUM_NOMEM || eError==STRACCUM_TOOBIG );
drha6353a32013-12-09 19:03:26 +0000142 p->accError = eError;
143 p->nAlloc = 0;
144}
145
146/*
drha5c14162013-12-17 15:03:06 +0000147** Extra argument values from a PrintfArguments object
148*/
149static sqlite3_int64 getIntArg(PrintfArguments *p){
150 if( p->nArg<=p->nUsed ) return 0;
151 return sqlite3_value_int64(p->apArg[p->nUsed++]);
152}
153static double getDoubleArg(PrintfArguments *p){
154 if( p->nArg<=p->nUsed ) return 0.0;
155 return sqlite3_value_double(p->apArg[p->nUsed++]);
156}
157static char *getTextArg(PrintfArguments *p){
158 if( p->nArg<=p->nUsed ) return 0;
159 return (char*)sqlite3_value_text(p->apArg[p->nUsed++]);
160}
161
162
163/*
drh79158e12005-09-06 21:40:45 +0000164** On machines with a small stack size, you can redefine the
drhed1fddf2011-10-12 18:52:59 +0000165** SQLITE_PRINT_BUF_SIZE to be something smaller, if desired.
drh79158e12005-09-06 21:40:45 +0000166*/
167#ifndef SQLITE_PRINT_BUF_SIZE
drh59eedf72011-10-11 17:54:54 +0000168# define SQLITE_PRINT_BUF_SIZE 70
drh79158e12005-09-06 21:40:45 +0000169#endif
170#define etBUFSIZE SQLITE_PRINT_BUF_SIZE /* Size of the output buffer */
drha18c5682000-10-08 22:20:57 +0000171
172/*
drhed1fddf2011-10-12 18:52:59 +0000173** Render a string given by "fmt" into the StrAccum object.
drha18c5682000-10-08 22:20:57 +0000174*/
drhf089aa42008-07-08 19:34:06 +0000175void sqlite3VXPrintf(
drha5c14162013-12-17 15:03:06 +0000176 StrAccum *pAccum, /* Accumulate results here */
177 u32 bFlags, /* SQLITE_PRINTF_* flags */
178 const char *fmt, /* Format string */
179 va_list ap /* arguments */
drha18c5682000-10-08 22:20:57 +0000180){
drhe84a3062004-02-02 12:29:25 +0000181 int c; /* Next character in the format string */
182 char *bufpt; /* Pointer to the conversion buffer */
183 int precision; /* Precision of the current field */
184 int length; /* Length of the field */
185 int idx; /* A general purpose loop counter */
drhe84a3062004-02-02 12:29:25 +0000186 int width; /* Width of the current field */
187 etByte flag_leftjustify; /* True if "-" flag is present */
188 etByte flag_plussign; /* True if "+" flag is present */
189 etByte flag_blanksign; /* True if " " flag is present */
190 etByte flag_alternateform; /* True if "#" flag is present */
drh531fe872005-08-13 13:40:42 +0000191 etByte flag_altform2; /* True if "!" flag is present */
drhe84a3062004-02-02 12:29:25 +0000192 etByte flag_zeropad; /* True if field width constant starts with zero */
193 etByte flag_long; /* True if "l" flag is present */
drha34b6762004-05-07 13:30:42 +0000194 etByte flag_longlong; /* True if the "ll" flag is present */
drh3e9aeec2005-08-13 13:39:02 +0000195 etByte done; /* Loop termination flag */
drhed1fddf2011-10-12 18:52:59 +0000196 etByte xtype = 0; /* Conversion paradigm */
drha5c14162013-12-17 15:03:06 +0000197 u8 bArgList; /* True for SQLITE_PRINTF_SQLFUNC */
198 u8 useIntern; /* Ok to use internal conversions (ex: %T) */
drhed1fddf2011-10-12 18:52:59 +0000199 char prefix; /* Prefix character. "+" or "-" or " " or '\0'. */
drh27436af2006-03-28 23:57:17 +0000200 sqlite_uint64 longvalue; /* Value for integer types */
drh384eef32004-01-07 03:04:27 +0000201 LONGDOUBLE_TYPE realvalue; /* Value for real types */
drh57196282004-10-06 15:41:16 +0000202 const et_info *infop; /* Pointer to the appropriate info structure */
drh59eedf72011-10-11 17:54:54 +0000203 char *zOut; /* Rendering buffer */
204 int nOut; /* Size of the rendering buffer */
drhaf8f5132014-10-29 18:20:18 +0000205 char *zExtra = 0; /* Malloced memory used by some conversion */
drhb37df7b2005-10-13 02:09:49 +0000206#ifndef SQLITE_OMIT_FLOATING_POINT
drh557cc602005-08-13 12:59:14 +0000207 int exp, e2; /* exponent of real numbers */
drhed1fddf2011-10-12 18:52:59 +0000208 int nsd; /* Number of significant digits returned */
drhe84a3062004-02-02 12:29:25 +0000209 double rounder; /* Used for rounding floating point values */
210 etByte flag_dp; /* True if decimal point should be shown */
211 etByte flag_rtz; /* True if trailing zeros should be removed */
drha18c5682000-10-08 22:20:57 +0000212#endif
drha5c14162013-12-17 15:03:06 +0000213 PrintfArguments *pArgList = 0; /* Arguments for SQLITE_PRINTF_SQLFUNC */
drhed1fddf2011-10-12 18:52:59 +0000214 char buf[etBUFSIZE]; /* Conversion buffer */
drha18c5682000-10-08 22:20:57 +0000215
drha18c5682000-10-08 22:20:57 +0000216 bufpt = 0;
drha5c14162013-12-17 15:03:06 +0000217 if( bFlags ){
218 if( (bArgList = (bFlags & SQLITE_PRINTF_SQLFUNC))!=0 ){
219 pArgList = va_arg(ap, PrintfArguments*);
220 }
221 useIntern = bFlags & SQLITE_PRINTF_INTERNAL;
222 }else{
223 bArgList = useIntern = 0;
224 }
drha18c5682000-10-08 22:20:57 +0000225 for(; (c=(*fmt))!=0; ++fmt){
226 if( c!='%' ){
drha18c5682000-10-08 22:20:57 +0000227 bufpt = (char *)fmt;
drh760b1592014-09-18 01:50:09 +0000228#if HAVE_STRCHRNUL
229 fmt = strchrnul(fmt, '%');
230#else
231 do{ fmt++; }while( *fmt && *fmt != '%' );
232#endif
drha70a0732014-03-17 14:24:27 +0000233 sqlite3StrAccumAppend(pAccum, bufpt, (int)(fmt - bufpt));
drh760b1592014-09-18 01:50:09 +0000234 if( *fmt==0 ) break;
drha18c5682000-10-08 22:20:57 +0000235 }
236 if( (c=(*++fmt))==0 ){
drhade86482007-11-28 22:36:40 +0000237 sqlite3StrAccumAppend(pAccum, "%", 1);
drha18c5682000-10-08 22:20:57 +0000238 break;
239 }
240 /* Find out what flags are present */
241 flag_leftjustify = flag_plussign = flag_blanksign =
drh557cc602005-08-13 12:59:14 +0000242 flag_alternateform = flag_altform2 = flag_zeropad = 0;
drh3e9aeec2005-08-13 13:39:02 +0000243 done = 0;
drha18c5682000-10-08 22:20:57 +0000244 do{
245 switch( c ){
drh3e9aeec2005-08-13 13:39:02 +0000246 case '-': flag_leftjustify = 1; break;
247 case '+': flag_plussign = 1; break;
248 case ' ': flag_blanksign = 1; break;
249 case '#': flag_alternateform = 1; break;
250 case '!': flag_altform2 = 1; break;
251 case '0': flag_zeropad = 1; break;
252 default: done = 1; break;
drha18c5682000-10-08 22:20:57 +0000253 }
drh3e9aeec2005-08-13 13:39:02 +0000254 }while( !done && (c=(*++fmt))!=0 );
drha18c5682000-10-08 22:20:57 +0000255 /* Get the field width */
drha18c5682000-10-08 22:20:57 +0000256 if( c=='*' ){
drha5c14162013-12-17 15:03:06 +0000257 if( bArgList ){
258 width = (int)getIntArg(pArgList);
259 }else{
260 width = va_arg(ap,int);
261 }
drha18c5682000-10-08 22:20:57 +0000262 if( width<0 ){
263 flag_leftjustify = 1;
drhb6f47de2015-04-07 15:39:29 +0000264 width = width >= -2147483647 ? -width : 0;
drha18c5682000-10-08 22:20:57 +0000265 }
266 c = *++fmt;
267 }else{
drhb6f47de2015-04-07 15:39:29 +0000268 unsigned wx = 0;
drh17a68932001-01-31 13:28:08 +0000269 while( c>='0' && c<='9' ){
drhb6f47de2015-04-07 15:39:29 +0000270 wx = wx*10 + c - '0';
drha18c5682000-10-08 22:20:57 +0000271 c = *++fmt;
272 }
drhb6f47de2015-04-07 15:39:29 +0000273 testcase( wx>0x7fffffff );
274 width = wx & 0x7fffffff;
drha18c5682000-10-08 22:20:57 +0000275 }
dan8c069142015-04-07 14:38:57 +0000276
drha18c5682000-10-08 22:20:57 +0000277 /* Get the precision */
278 if( c=='.' ){
drha18c5682000-10-08 22:20:57 +0000279 c = *++fmt;
280 if( c=='*' ){
drha5c14162013-12-17 15:03:06 +0000281 if( bArgList ){
282 precision = (int)getIntArg(pArgList);
283 }else{
284 precision = va_arg(ap,int);
285 }
drha18c5682000-10-08 22:20:57 +0000286 c = *++fmt;
drhb6f47de2015-04-07 15:39:29 +0000287 if( precision<0 ){
288 precision = precision >= -2147483647 ? -precision : -1;
289 }
drha18c5682000-10-08 22:20:57 +0000290 }else{
drhb6f47de2015-04-07 15:39:29 +0000291 unsigned px = 0;
drh17a68932001-01-31 13:28:08 +0000292 while( c>='0' && c<='9' ){
drhb6f47de2015-04-07 15:39:29 +0000293 px = px*10 + c - '0';
drha18c5682000-10-08 22:20:57 +0000294 c = *++fmt;
295 }
drhb6f47de2015-04-07 15:39:29 +0000296 testcase( px>0x7fffffff );
297 precision = px & 0x7fffffff;
drha18c5682000-10-08 22:20:57 +0000298 }
drha18c5682000-10-08 22:20:57 +0000299 }else{
300 precision = -1;
301 }
302 /* Get the conversion type modifier */
303 if( c=='l' ){
304 flag_long = 1;
305 c = *++fmt;
drha34b6762004-05-07 13:30:42 +0000306 if( c=='l' ){
307 flag_longlong = 1;
308 c = *++fmt;
309 }else{
310 flag_longlong = 0;
311 }
drha18c5682000-10-08 22:20:57 +0000312 }else{
drha34b6762004-05-07 13:30:42 +0000313 flag_long = flag_longlong = 0;
drha18c5682000-10-08 22:20:57 +0000314 }
315 /* Fetch the info entry for the field */
drh874ba042009-04-08 16:10:04 +0000316 infop = &fmtinfo[0];
317 xtype = etINVALID;
danielk197700e13612008-11-17 19:18:54 +0000318 for(idx=0; idx<ArraySize(fmtinfo); idx++){
drha18c5682000-10-08 22:20:57 +0000319 if( c==fmtinfo[idx].fmttype ){
320 infop = &fmtinfo[idx];
drha5c14162013-12-17 15:03:06 +0000321 if( useIntern || (infop->flags & FLAG_INTERN)==0 ){
drh5f968432004-02-21 19:02:30 +0000322 xtype = infop->type;
drhc4413562006-05-22 22:04:00 +0000323 }else{
drhade86482007-11-28 22:36:40 +0000324 return;
drh5f968432004-02-21 19:02:30 +0000325 }
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.
drha18c5682000-10-08 22:20:57 +0000335 ** flag_plussign TRUE if a '+' is present.
336 ** 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.
339 ** flag_long TRUE if the letter 'l' (ell) prefixed
340 ** the conversion character.
drha34b6762004-05-07 13:30:42 +0000341 ** flag_longlong TRUE if the letter 'll' (ell ell) prefixed
342 ** the conversion character.
drha18c5682000-10-08 22:20:57 +0000343 ** flag_blanksign TRUE if a ' ' is present.
344 ** width The specified field width. This is
345 ** always non-negative. Zero is the default.
346 ** precision The specified precision. The default
347 ** is -1.
348 ** xtype The class of the conversion.
349 ** infop Pointer to the appropriate info struct.
350 */
351 switch( xtype ){
drhfe63d1c2004-09-08 20:13:04 +0000352 case etPOINTER:
353 flag_longlong = sizeof(char*)==sizeof(i64);
354 flag_long = sizeof(char*)==sizeof(long int);
355 /* Fall through into the next case */
drh9a993342007-12-13 02:45:31 +0000356 case etORDINAL:
drha18c5682000-10-08 22:20:57 +0000357 case etRADIX:
drhe84a3062004-02-02 12:29:25 +0000358 if( infop->flags & FLAG_SIGNED ){
drhe9707672004-06-25 01:10:48 +0000359 i64 v;
drha5c14162013-12-17 15:03:06 +0000360 if( bArgList ){
361 v = getIntArg(pArgList);
362 }else if( flag_longlong ){
drheeb23a42009-05-04 20:20:16 +0000363 v = va_arg(ap,i64);
364 }else if( flag_long ){
365 v = va_arg(ap,long int);
366 }else{
367 v = va_arg(ap,int);
368 }
drhe9707672004-06-25 01:10:48 +0000369 if( v<0 ){
drh158b9cb2011-03-05 20:59:46 +0000370 if( v==SMALLEST_INT64 ){
371 longvalue = ((u64)1)<<63;
372 }else{
373 longvalue = -v;
374 }
drhe9707672004-06-25 01:10:48 +0000375 prefix = '-';
danielk1977cfcdaef2004-05-12 07:33:33 +0000376 }else{
drhe9707672004-06-25 01:10:48 +0000377 longvalue = v;
378 if( flag_plussign ) prefix = '+';
danielk1977cfcdaef2004-05-12 07:33:33 +0000379 else if( flag_blanksign ) prefix = ' ';
380 else prefix = 0;
381 }
drhe9707672004-06-25 01:10:48 +0000382 }else{
drha5c14162013-12-17 15:03:06 +0000383 if( bArgList ){
384 longvalue = (u64)getIntArg(pArgList);
385 }else if( flag_longlong ){
drheeb23a42009-05-04 20:20:16 +0000386 longvalue = va_arg(ap,u64);
387 }else if( flag_long ){
388 longvalue = va_arg(ap,unsigned long int);
389 }else{
390 longvalue = va_arg(ap,unsigned int);
391 }
drhe9707672004-06-25 01:10:48 +0000392 prefix = 0;
393 }
394 if( longvalue==0 ) flag_alternateform = 0;
drha18c5682000-10-08 22:20:57 +0000395 if( flag_zeropad && precision<width-(prefix!=0) ){
396 precision = width-(prefix!=0);
drh9adf9ac2002-05-15 11:44:13 +0000397 }
drh59eedf72011-10-11 17:54:54 +0000398 if( precision<etBUFSIZE-10 ){
399 nOut = etBUFSIZE;
400 zOut = buf;
401 }else{
402 nOut = precision + 10;
403 zOut = zExtra = sqlite3Malloc( nOut );
404 if( zOut==0 ){
drha6353a32013-12-09 19:03:26 +0000405 setStrAccumError(pAccum, STRACCUM_NOMEM);
drh59eedf72011-10-11 17:54:54 +0000406 return;
407 }
408 }
409 bufpt = &zOut[nOut-1];
drh9a993342007-12-13 02:45:31 +0000410 if( xtype==etORDINAL ){
drh43f6e062007-12-13 17:50:22 +0000411 static const char zOrd[] = "thstndrd";
drhea678832008-12-10 19:26:22 +0000412 int x = (int)(longvalue % 10);
drh43f6e062007-12-13 17:50:22 +0000413 if( x>=4 || (longvalue/10)%10==1 ){
414 x = 0;
415 }
drh59eedf72011-10-11 17:54:54 +0000416 *(--bufpt) = zOrd[x*2+1];
417 *(--bufpt) = zOrd[x*2];
drh9a993342007-12-13 02:45:31 +0000418 }
drha18c5682000-10-08 22:20:57 +0000419 {
drh0e682092014-03-17 15:06:57 +0000420 const char *cset = &aDigits[infop->charset];
421 u8 base = infop->base;
drha18c5682000-10-08 22:20:57 +0000422 do{ /* Convert to ascii */
423 *(--bufpt) = cset[longvalue%base];
424 longvalue = longvalue/base;
425 }while( longvalue>0 );
drh9adf9ac2002-05-15 11:44:13 +0000426 }
drh59eedf72011-10-11 17:54:54 +0000427 length = (int)(&zOut[nOut-1]-bufpt);
drha18c5682000-10-08 22:20:57 +0000428 for(idx=precision-length; idx>0; idx--){
429 *(--bufpt) = '0'; /* Zero pad */
drh9adf9ac2002-05-15 11:44:13 +0000430 }
drha18c5682000-10-08 22:20:57 +0000431 if( prefix ) *(--bufpt) = prefix; /* Add sign */
432 if( flag_alternateform && infop->prefix ){ /* Add "0" or "0x" */
drh76ff3a02004-09-24 22:32:30 +0000433 const char *pre;
434 char x;
435 pre = &aPrefix[infop->prefix];
drhaf005fb2008-07-09 16:51:51 +0000436 for(; (x=(*pre))!=0; pre++) *(--bufpt) = x;
drha18c5682000-10-08 22:20:57 +0000437 }
drh59eedf72011-10-11 17:54:54 +0000438 length = (int)(&zOut[nOut-1]-bufpt);
drha18c5682000-10-08 22:20:57 +0000439 break;
440 case etFLOAT:
441 case etEXP:
442 case etGENERIC:
drha5c14162013-12-17 15:03:06 +0000443 if( bArgList ){
444 realvalue = getDoubleArg(pArgList);
445 }else{
446 realvalue = va_arg(ap,double);
447 }
drh69ef7032010-03-04 17:11:31 +0000448#ifdef SQLITE_OMIT_FLOATING_POINT
449 length = 0;
450#else
drha18c5682000-10-08 22:20:57 +0000451 if( precision<0 ) precision = 6; /* Set default precision */
drha18c5682000-10-08 22:20:57 +0000452 if( realvalue<0.0 ){
453 realvalue = -realvalue;
454 prefix = '-';
drh9adf9ac2002-05-15 11:44:13 +0000455 }else{
drha18c5682000-10-08 22:20:57 +0000456 if( flag_plussign ) prefix = '+';
457 else if( flag_blanksign ) prefix = ' ';
458 else prefix = 0;
drh9adf9ac2002-05-15 11:44:13 +0000459 }
drh3e9aeec2005-08-13 13:39:02 +0000460 if( xtype==etGENERIC && precision>0 ) precision--;
drha30d22a2015-04-07 13:28:41 +0000461 testcase( precision>0xfff );
drh74b42272015-04-07 12:41:17 +0000462 for(idx=precision&0xfff, rounder=0.5; idx>0; idx--, rounder*=0.1){}
drh3e9aeec2005-08-13 13:39:02 +0000463 if( xtype==etFLOAT ) realvalue += rounder;
drha18c5682000-10-08 22:20:57 +0000464 /* Normalize realvalue to within 10.0 > realvalue >= 1.0 */
465 exp = 0;
drhea678832008-12-10 19:26:22 +0000466 if( sqlite3IsNaN((double)realvalue) ){
drh53c14022007-05-10 17:23:11 +0000467 bufpt = "NaN";
468 length = 3;
469 break;
470 }
drha18c5682000-10-08 22:20:57 +0000471 if( realvalue>0.0 ){
drh72b3fbc2012-06-19 03:11:25 +0000472 LONGDOUBLE_TYPE scale = 1.0;
drh4ef94132012-06-19 03:35:05 +0000473 while( realvalue>=1e100*scale && exp<=350 ){ scale *= 1e100;exp+=100;}
drh72b3fbc2012-06-19 03:11:25 +0000474 while( realvalue>=1e64*scale && exp<=350 ){ scale *= 1e64; exp+=64; }
475 while( realvalue>=1e8*scale && exp<=350 ){ scale *= 1e8; exp+=8; }
476 while( realvalue>=10.0*scale && exp<=350 ){ scale *= 10.0; exp++; }
477 realvalue /= scale;
drh93a960a2008-07-10 00:32:42 +0000478 while( realvalue<1e-8 ){ realvalue *= 1e8; exp-=8; }
479 while( realvalue<1.0 ){ realvalue *= 10.0; exp--; }
drhaf005fb2008-07-09 16:51:51 +0000480 if( exp>350 ){
drh53c14022007-05-10 17:23:11 +0000481 if( prefix=='-' ){
482 bufpt = "-Inf";
483 }else if( prefix=='+' ){
484 bufpt = "+Inf";
485 }else{
486 bufpt = "Inf";
487 }
drhea678832008-12-10 19:26:22 +0000488 length = sqlite3Strlen30(bufpt);
drha18c5682000-10-08 22:20:57 +0000489 break;
490 }
drh9adf9ac2002-05-15 11:44:13 +0000491 }
drha18c5682000-10-08 22:20:57 +0000492 bufpt = buf;
493 /*
494 ** If the field type is etGENERIC, then convert to either etEXP
495 ** or etFLOAT, as appropriate.
496 */
drha18c5682000-10-08 22:20:57 +0000497 if( xtype!=etFLOAT ){
498 realvalue += rounder;
499 if( realvalue>=10.0 ){ realvalue *= 0.1; exp++; }
500 }
501 if( xtype==etGENERIC ){
502 flag_rtz = !flag_alternateform;
503 if( exp<-4 || exp>precision ){
504 xtype = etEXP;
505 }else{
506 precision = precision - exp;
507 xtype = etFLOAT;
508 }
drh9adf9ac2002-05-15 11:44:13 +0000509 }else{
drh72b3fbc2012-06-19 03:11:25 +0000510 flag_rtz = flag_altform2;
drh9adf9ac2002-05-15 11:44:13 +0000511 }
drh557cc602005-08-13 12:59:14 +0000512 if( xtype==etEXP ){
513 e2 = 0;
514 }else{
515 e2 = exp;
516 }
drh74b42272015-04-07 12:41:17 +0000517 if( MAX(e2,0)+(i64)precision+(i64)width > etBUFSIZE - 15 ){
518 bufpt = zExtra
519 = sqlite3Malloc( MAX(e2,0)+(i64)precision+(i64)width+15 );
drh59eedf72011-10-11 17:54:54 +0000520 if( bufpt==0 ){
drha6353a32013-12-09 19:03:26 +0000521 setStrAccumError(pAccum, STRACCUM_NOMEM);
drh59eedf72011-10-11 17:54:54 +0000522 return;
523 }
524 }
525 zOut = bufpt;
drh72b3fbc2012-06-19 03:11:25 +0000526 nsd = 16 + flag_altform2*10;
drhea678832008-12-10 19:26:22 +0000527 flag_dp = (precision>0 ?1:0) | flag_alternateform | flag_altform2;
drh557cc602005-08-13 12:59:14 +0000528 /* The sign in front of the number */
529 if( prefix ){
530 *(bufpt++) = prefix;
531 }
532 /* Digits prior to the decimal point */
533 if( e2<0 ){
534 *(bufpt++) = '0';
535 }else{
536 for(; e2>=0; e2--){
537 *(bufpt++) = et_getdigit(&realvalue,&nsd);
drha18c5682000-10-08 22:20:57 +0000538 }
drh9adf9ac2002-05-15 11:44:13 +0000539 }
drh557cc602005-08-13 12:59:14 +0000540 /* The decimal point */
541 if( flag_dp ){
542 *(bufpt++) = '.';
543 }
544 /* "0" digits after the decimal point but before the first
545 ** significant digit of the number */
drhaf005fb2008-07-09 16:51:51 +0000546 for(e2++; e2<0; precision--, e2++){
547 assert( precision>0 );
drh557cc602005-08-13 12:59:14 +0000548 *(bufpt++) = '0';
549 }
550 /* Significant digits after the decimal point */
551 while( (precision--)>0 ){
552 *(bufpt++) = et_getdigit(&realvalue,&nsd);
553 }
554 /* Remove trailing zeros and the "." if no digits follow the "." */
555 if( flag_rtz && flag_dp ){
drh3e9aeec2005-08-13 13:39:02 +0000556 while( bufpt[-1]=='0' ) *(--bufpt) = 0;
drh59eedf72011-10-11 17:54:54 +0000557 assert( bufpt>zOut );
drh3e9aeec2005-08-13 13:39:02 +0000558 if( bufpt[-1]=='.' ){
drh557cc602005-08-13 12:59:14 +0000559 if( flag_altform2 ){
560 *(bufpt++) = '0';
561 }else{
562 *(--bufpt) = 0;
563 }
564 }
565 }
566 /* Add the "eNNN" suffix */
drh59eedf72011-10-11 17:54:54 +0000567 if( xtype==etEXP ){
drh557cc602005-08-13 12:59:14 +0000568 *(bufpt++) = aDigits[infop->charset];
569 if( exp<0 ){
570 *(bufpt++) = '-'; exp = -exp;
571 }else{
572 *(bufpt++) = '+';
573 }
574 if( exp>=100 ){
drhea678832008-12-10 19:26:22 +0000575 *(bufpt++) = (char)((exp/100)+'0'); /* 100's digit */
drh557cc602005-08-13 12:59:14 +0000576 exp %= 100;
577 }
drhea678832008-12-10 19:26:22 +0000578 *(bufpt++) = (char)(exp/10+'0'); /* 10's digit */
579 *(bufpt++) = (char)(exp%10+'0'); /* 1's digit */
drh557cc602005-08-13 12:59:14 +0000580 }
581 *bufpt = 0;
582
drha18c5682000-10-08 22:20:57 +0000583 /* The converted number is in buf[] and zero terminated. Output it.
584 ** Note that the number is in the usual order, not reversed as with
585 ** integer conversions. */
drh59eedf72011-10-11 17:54:54 +0000586 length = (int)(bufpt-zOut);
587 bufpt = zOut;
drha18c5682000-10-08 22:20:57 +0000588
589 /* Special case: Add leading zeros if the flag_zeropad flag is
590 ** set and we are not left justified */
591 if( flag_zeropad && !flag_leftjustify && length < width){
592 int i;
593 int nPad = width - length;
594 for(i=width; i>=nPad; i--){
595 bufpt[i] = bufpt[i-nPad];
596 }
597 i = prefix!=0;
598 while( nPad-- ) bufpt[i++] = '0';
599 length = width;
600 }
drh69ef7032010-03-04 17:11:31 +0000601#endif /* !defined(SQLITE_OMIT_FLOATING_POINT) */
drha18c5682000-10-08 22:20:57 +0000602 break;
603 case etSIZE:
drhfc6ee9d2013-12-17 15:58:42 +0000604 if( !bArgList ){
605 *(va_arg(ap,int*)) = pAccum->nChar;
606 }
drha18c5682000-10-08 22:20:57 +0000607 length = width = 0;
608 break;
609 case etPERCENT:
610 buf[0] = '%';
611 bufpt = buf;
612 length = 1;
613 break;
drha18c5682000-10-08 22:20:57 +0000614 case etCHARX:
drha5c14162013-12-17 15:03:06 +0000615 if( bArgList ){
drhfc6ee9d2013-12-17 15:58:42 +0000616 bufpt = getTextArg(pArgList);
617 c = bufpt ? bufpt[0] : 0;
drha5c14162013-12-17 15:03:06 +0000618 }else{
619 c = va_arg(ap,int);
620 }
drhaf8f5132014-10-29 18:20:18 +0000621 if( precision>1 ){
622 width -= precision-1;
623 if( width>1 && !flag_leftjustify ){
624 sqlite3AppendChar(pAccum, width-1, ' ');
625 width = 0;
626 }
627 sqlite3AppendChar(pAccum, precision-1, c);
drh9adf9ac2002-05-15 11:44:13 +0000628 }
drhaf8f5132014-10-29 18:20:18 +0000629 length = 1;
630 buf[0] = c;
drha18c5682000-10-08 22:20:57 +0000631 bufpt = buf;
632 break;
633 case etSTRING:
drhd93d8a82003-06-16 03:08:18 +0000634 case etDYNSTRING:
drha5c14162013-12-17 15:03:06 +0000635 if( bArgList ){
636 bufpt = getTextArg(pArgList);
637 }else{
638 bufpt = va_arg(ap,char*);
639 }
drhd93d8a82003-06-16 03:08:18 +0000640 if( bufpt==0 ){
641 bufpt = "";
drha5c14162013-12-17 15:03:06 +0000642 }else if( xtype==etDYNSTRING && !bArgList ){
drhd93d8a82003-06-16 03:08:18 +0000643 zExtra = bufpt;
644 }
drhe5090942008-04-29 15:22:27 +0000645 if( precision>=0 ){
646 for(length=0; length<precision && bufpt[length]; length++){}
647 }else{
drhea678832008-12-10 19:26:22 +0000648 length = sqlite3Strlen30(bufpt);
drhe5090942008-04-29 15:22:27 +0000649 }
drha18c5682000-10-08 22:20:57 +0000650 break;
651 case etSQLESCAPE:
danielk1977f3b863e2007-06-24 06:32:17 +0000652 case etSQLESCAPE2:
653 case etSQLESCAPE3: {
drh8965b502009-11-25 16:53:37 +0000654 int i, j, k, n, isnull;
drh5eba8c02005-08-19 02:26:27 +0000655 int needQuote;
drhea678832008-12-10 19:26:22 +0000656 char ch;
danielk1977f3b863e2007-06-24 06:32:17 +0000657 char q = ((xtype==etSQLESCAPE3)?'"':'\''); /* Quote character */
drha5c14162013-12-17 15:03:06 +0000658 char *escarg;
659
660 if( bArgList ){
661 escarg = getTextArg(pArgList);
662 }else{
663 escarg = va_arg(ap,char*);
664 }
danielk1977f0113002006-01-24 12:09:17 +0000665 isnull = escarg==0;
666 if( isnull ) escarg = (xtype==etSQLESCAPE2 ? "NULL" : "(NULL)");
drh8965b502009-11-25 16:53:37 +0000667 k = precision;
dan60d4a302010-03-04 17:58:45 +0000668 for(i=n=0; k!=0 && (ch=escarg[i])!=0; i++, k--){
danielk1977f3b863e2007-06-24 06:32:17 +0000669 if( ch==q ) n++;
drha18c5682000-10-08 22:20:57 +0000670 }
drh5eba8c02005-08-19 02:26:27 +0000671 needQuote = !isnull && xtype==etSQLESCAPE2;
672 n += i + 1 + needQuote*2;
673 if( n>etBUFSIZE ){
drhe5ae5732008-06-15 02:51:47 +0000674 bufpt = zExtra = sqlite3Malloc( n );
drhd164fd32008-11-20 18:20:28 +0000675 if( bufpt==0 ){
drha6353a32013-12-09 19:03:26 +0000676 setStrAccumError(pAccum, STRACCUM_NOMEM);
drhd164fd32008-11-20 18:20:28 +0000677 return;
678 }
drh5eba8c02005-08-19 02:26:27 +0000679 }else{
680 bufpt = buf;
681 }
682 j = 0;
danielk1977f3b863e2007-06-24 06:32:17 +0000683 if( needQuote ) bufpt[j++] = q;
drh8965b502009-11-25 16:53:37 +0000684 k = i;
685 for(i=0; i<k; i++){
686 bufpt[j++] = ch = escarg[i];
danielk1977f3b863e2007-06-24 06:32:17 +0000687 if( ch==q ) bufpt[j++] = ch;
drh5eba8c02005-08-19 02:26:27 +0000688 }
danielk1977f3b863e2007-06-24 06:32:17 +0000689 if( needQuote ) bufpt[j++] = q;
drh5eba8c02005-08-19 02:26:27 +0000690 bufpt[j] = 0;
691 length = j;
drh8965b502009-11-25 16:53:37 +0000692 /* The precision in %q and %Q means how many input characters to
693 ** consume, not the length of the output...
694 ** if( precision>=0 && precision<length ) length = precision; */
drha18c5682000-10-08 22:20:57 +0000695 break;
drh5eba8c02005-08-19 02:26:27 +0000696 }
drh5f968432004-02-21 19:02:30 +0000697 case etTOKEN: {
698 Token *pToken = va_arg(ap, Token*);
drha5c14162013-12-17 15:03:06 +0000699 assert( bArgList==0 );
drha9ab4812013-12-11 11:00:44 +0000700 if( pToken && pToken->n ){
drhade86482007-11-28 22:36:40 +0000701 sqlite3StrAccumAppend(pAccum, (const char*)pToken->z, pToken->n);
drhad6d9462004-09-19 02:15:24 +0000702 }
drh5f968432004-02-21 19:02:30 +0000703 length = width = 0;
704 break;
705 }
706 case etSRCLIST: {
707 SrcList *pSrc = va_arg(ap, SrcList*);
708 int k = va_arg(ap, int);
709 struct SrcList_item *pItem = &pSrc->a[k];
drha5c14162013-12-17 15:03:06 +0000710 assert( bArgList==0 );
drh5f968432004-02-21 19:02:30 +0000711 assert( k>=0 && k<pSrc->nSrc );
drh93a960a2008-07-10 00:32:42 +0000712 if( pItem->zDatabase ){
drha6353a32013-12-09 19:03:26 +0000713 sqlite3StrAccumAppendAll(pAccum, pItem->zDatabase);
drhade86482007-11-28 22:36:40 +0000714 sqlite3StrAccumAppend(pAccum, ".", 1);
drh5f968432004-02-21 19:02:30 +0000715 }
drha6353a32013-12-09 19:03:26 +0000716 sqlite3StrAccumAppendAll(pAccum, pItem->zName);
drh5f968432004-02-21 19:02:30 +0000717 length = width = 0;
718 break;
719 }
drh874ba042009-04-08 16:10:04 +0000720 default: {
721 assert( xtype==etINVALID );
722 return;
723 }
drha18c5682000-10-08 22:20:57 +0000724 }/* End switch over the format type */
725 /*
726 ** The text of the conversion is pointed to by "bufpt" and is
727 ** "length" characters long. The field width is "width". Do
728 ** the output.
729 */
drha70a0732014-03-17 14:24:27 +0000730 width -= length;
drhaf8f5132014-10-29 18:20:18 +0000731 if( width>0 && !flag_leftjustify ) sqlite3AppendChar(pAccum, width, ' ');
drha70a0732014-03-17 14:24:27 +0000732 sqlite3StrAccumAppend(pAccum, bufpt, length);
drhaf8f5132014-10-29 18:20:18 +0000733 if( width>0 && flag_leftjustify ) sqlite3AppendChar(pAccum, width, ' ');
drha70a0732014-03-17 14:24:27 +0000734
drhaf8f5132014-10-29 18:20:18 +0000735 if( zExtra ){
736 sqlite3_free(zExtra);
737 zExtra = 0;
738 }
drha18c5682000-10-08 22:20:57 +0000739 }/* End for loop over the format string */
drha18c5682000-10-08 22:20:57 +0000740} /* End of function */
741
drhade86482007-11-28 22:36:40 +0000742/*
drha70a0732014-03-17 14:24:27 +0000743** Enlarge the memory allocation on a StrAccum object so that it is
744** able to accept at least N more bytes of text.
745**
746** Return the number of bytes of text that StrAccum is able to accept
747** after the attempted enlargement. The value returned might be zero.
748*/
749static int sqlite3StrAccumEnlarge(StrAccum *p, int N){
750 char *zNew;
drha30d22a2015-04-07 13:28:41 +0000751 assert( p->nChar+(i64)N >= p->nAlloc ); /* Only called if really needed */
drha70a0732014-03-17 14:24:27 +0000752 if( p->accError ){
753 testcase(p->accError==STRACCUM_TOOBIG);
754 testcase(p->accError==STRACCUM_NOMEM);
755 return 0;
756 }
drhc0490572015-05-02 11:45:53 +0000757 if( p->mxAlloc==0 ){
drha70a0732014-03-17 14:24:27 +0000758 N = p->nAlloc - p->nChar - 1;
759 setStrAccumError(p, STRACCUM_TOOBIG);
760 return N;
761 }else{
762 char *zOld = (p->zText==p->zBase ? 0 : p->zText);
763 i64 szNew = p->nChar;
764 szNew += N + 1;
drh7b4d7802014-11-03 14:46:29 +0000765 if( szNew+p->nChar<=p->mxAlloc ){
766 /* Force exponential buffer size growth as long as it does not overflow,
767 ** to avoid having to call this routine too often */
768 szNew += p->nChar;
769 }
drha70a0732014-03-17 14:24:27 +0000770 if( szNew > p->mxAlloc ){
771 sqlite3StrAccumReset(p);
772 setStrAccumError(p, STRACCUM_TOOBIG);
773 return 0;
774 }else{
775 p->nAlloc = (int)szNew;
776 }
drhc0490572015-05-02 11:45:53 +0000777 if( p->db ){
drha70a0732014-03-17 14:24:27 +0000778 zNew = sqlite3DbRealloc(p->db, zOld, p->nAlloc);
779 }else{
drhf3cdcdc2015-04-29 16:50:28 +0000780 zNew = sqlite3_realloc64(zOld, p->nAlloc);
drha70a0732014-03-17 14:24:27 +0000781 }
782 if( zNew ){
drh7ef4d1c2014-05-31 15:39:53 +0000783 assert( p->zText!=0 || p->nChar==0 );
drha70a0732014-03-17 14:24:27 +0000784 if( zOld==0 && p->nChar>0 ) memcpy(zNew, p->zText, p->nChar);
785 p->zText = zNew;
drh7f5a7ec2014-11-03 13:24:12 +0000786 p->nAlloc = sqlite3DbMallocSize(p->db, zNew);
drha70a0732014-03-17 14:24:27 +0000787 }else{
788 sqlite3StrAccumReset(p);
789 setStrAccumError(p, STRACCUM_NOMEM);
790 return 0;
791 }
792 }
793 return N;
794}
795
796/*
drhaf8f5132014-10-29 18:20:18 +0000797** Append N copies of character c to the given string buffer.
drha70a0732014-03-17 14:24:27 +0000798*/
drhaf8f5132014-10-29 18:20:18 +0000799void sqlite3AppendChar(StrAccum *p, int N, char c){
drha30d22a2015-04-07 13:28:41 +0000800 testcase( p->nChar + (i64)N > 0x7fffffff );
801 if( p->nChar+(i64)N >= p->nAlloc && (N = sqlite3StrAccumEnlarge(p, N))<=0 ){
802 return;
803 }
drhaf8f5132014-10-29 18:20:18 +0000804 while( (N--)>0 ) p->zText[p->nChar++] = c;
drha70a0732014-03-17 14:24:27 +0000805}
806
807/*
808** The StrAccum "p" is not large enough to accept N new bytes of z[].
809** So enlarge if first, then do the append.
810**
811** This is a helper routine to sqlite3StrAccumAppend() that does special-case
812** work (enlarging the buffer) using tail recursion, so that the
813** sqlite3StrAccumAppend() routine can use fast calling semantics.
814*/
drh172087f2014-08-22 15:40:20 +0000815static void SQLITE_NOINLINE enlargeAndAppend(StrAccum *p, const char *z, int N){
drha70a0732014-03-17 14:24:27 +0000816 N = sqlite3StrAccumEnlarge(p, N);
817 if( N>0 ){
818 memcpy(&p->zText[p->nChar], z, N);
819 p->nChar += N;
820 }
821}
822
823/*
824** Append N bytes of text from z to the StrAccum object. Increase the
825** size of the memory allocation for StrAccum if necessary.
drha18c5682000-10-08 22:20:57 +0000826*/
drhade86482007-11-28 22:36:40 +0000827void sqlite3StrAccumAppend(StrAccum *p, const char *z, int N){
drh34573382015-04-15 05:38:35 +0000828 assert( z!=0 || N==0 );
drha6353a32013-12-09 19:03:26 +0000829 assert( p->zText!=0 || p->nChar==0 || p->accError );
830 assert( N>=0 );
831 assert( p->accError==0 || p->nAlloc==0 );
drhade86482007-11-28 22:36:40 +0000832 if( p->nChar+N >= p->nAlloc ){
drha70a0732014-03-17 14:24:27 +0000833 enlargeAndAppend(p,z,N);
drh172087f2014-08-22 15:40:20 +0000834 }else{
835 assert( p->zText );
836 p->nChar += N;
837 memcpy(&p->zText[p->nChar-N], z, N);
drh5f968432004-02-21 19:02:30 +0000838 }
drh483750b2003-01-29 18:46:51 +0000839}
840
841/*
drha6353a32013-12-09 19:03:26 +0000842** Append the complete text of zero-terminated string z[] to the p string.
843*/
844void sqlite3StrAccumAppendAll(StrAccum *p, const char *z){
mistachkin53cd9642013-12-11 02:21:19 +0000845 sqlite3StrAccumAppend(p, z, sqlite3Strlen30(z));
drha6353a32013-12-09 19:03:26 +0000846}
847
848
849/*
drhade86482007-11-28 22:36:40 +0000850** Finish off a string by making sure it is zero-terminated.
851** Return a pointer to the resulting string. Return a NULL
852** pointer if any kind of error was encountered.
drh5f968432004-02-21 19:02:30 +0000853*/
drhade86482007-11-28 22:36:40 +0000854char *sqlite3StrAccumFinish(StrAccum *p){
855 if( p->zText ){
856 p->zText[p->nChar] = 0;
drhc0490572015-05-02 11:45:53 +0000857 if( p->mxAlloc>0 && p->zText==p->zBase ){
858 p->zText = sqlite3DbMallocRaw(p->db, p->nChar+1 );
drhade86482007-11-28 22:36:40 +0000859 if( p->zText ){
860 memcpy(p->zText, p->zBase, p->nChar+1);
861 }else{
drha6353a32013-12-09 19:03:26 +0000862 setStrAccumError(p, STRACCUM_NOMEM);
drhade86482007-11-28 22:36:40 +0000863 }
864 }
865 }
866 return p->zText;
867}
868
869/*
870** Reset an StrAccum string. Reclaim all malloced memory.
871*/
872void sqlite3StrAccumReset(StrAccum *p){
873 if( p->zText!=p->zBase ){
drhc0490572015-05-02 11:45:53 +0000874 sqlite3DbFree(p->db, p->zText);
drhade86482007-11-28 22:36:40 +0000875 }
drhf089aa42008-07-08 19:34:06 +0000876 p->zText = 0;
drhade86482007-11-28 22:36:40 +0000877}
878
879/*
drhc0490572015-05-02 11:45:53 +0000880** Initialize a string accumulator.
881**
882** p: The accumulator to be initialized.
883** db: Pointer to a database connection. May be NULL. Lookaside
884** memory is used if not NULL. db->mallocFailed is set appropriately
885** when not NULL.
886** zBase: An initial buffer. May be NULL in which case the initial buffer
887** is malloced.
888** n: Size of zBase in bytes. If total space requirements never exceed
889** n then no memory allocations ever occur.
890** mx: Maximum number of bytes to accumulate. If mx==0 then no memory
891** allocations will ever occur.
drhade86482007-11-28 22:36:40 +0000892*/
drhc0490572015-05-02 11:45:53 +0000893void sqlite3StrAccumInit(StrAccum *p, sqlite3 *db, char *zBase, int n, int mx){
drhade86482007-11-28 22:36:40 +0000894 p->zText = p->zBase = zBase;
drhc0490572015-05-02 11:45:53 +0000895 p->db = db;
drhade86482007-11-28 22:36:40 +0000896 p->nChar = 0;
897 p->nAlloc = n;
drhbb4957f2008-03-20 14:03:29 +0000898 p->mxAlloc = mx;
drhb49bc862013-08-21 21:12:10 +0000899 p->accError = 0;
drh5f968432004-02-21 19:02:30 +0000900}
901
902/*
903** Print into memory obtained from sqliteMalloc(). Use the internal
904** %-conversion extensions.
905*/
drh17435752007-08-16 04:30:38 +0000906char *sqlite3VMPrintf(sqlite3 *db, const char *zFormat, va_list ap){
907 char *z;
drh79158e12005-09-06 21:40:45 +0000908 char zBase[SQLITE_PRINT_BUF_SIZE];
drhade86482007-11-28 22:36:40 +0000909 StrAccum acc;
drhbc6160b2009-04-08 15:45:31 +0000910 assert( db!=0 );
drhc0490572015-05-02 11:45:53 +0000911 sqlite3StrAccumInit(&acc, db, zBase, sizeof(zBase),
drhbc6160b2009-04-08 15:45:31 +0000912 db->aLimit[SQLITE_LIMIT_LENGTH]);
drha5c14162013-12-17 15:03:06 +0000913 sqlite3VXPrintf(&acc, SQLITE_PRINTF_INTERNAL, zFormat, ap);
drhade86482007-11-28 22:36:40 +0000914 z = sqlite3StrAccumFinish(&acc);
drhb49bc862013-08-21 21:12:10 +0000915 if( acc.accError==STRACCUM_NOMEM ){
drh17435752007-08-16 04:30:38 +0000916 db->mallocFailed = 1;
917 }
918 return z;
drh5f968432004-02-21 19:02:30 +0000919}
920
921/*
922** Print into memory obtained from sqliteMalloc(). Use the internal
923** %-conversion extensions.
924*/
drh17435752007-08-16 04:30:38 +0000925char *sqlite3MPrintf(sqlite3 *db, const char *zFormat, ...){
drh5f968432004-02-21 19:02:30 +0000926 va_list ap;
927 char *z;
drh5f968432004-02-21 19:02:30 +0000928 va_start(ap, zFormat);
drhade86482007-11-28 22:36:40 +0000929 z = sqlite3VMPrintf(db, zFormat, ap);
drh5f968432004-02-21 19:02:30 +0000930 va_end(ap);
931 return z;
932}
933
934/*
drh28dd4792006-06-26 21:35:44 +0000935** Print into memory obtained from sqlite3_malloc(). Omit the internal
936** %-conversion extensions.
937*/
938char *sqlite3_vmprintf(const char *zFormat, va_list ap){
drhade86482007-11-28 22:36:40 +0000939 char *z;
drh28dd4792006-06-26 21:35:44 +0000940 char zBase[SQLITE_PRINT_BUF_SIZE];
drhade86482007-11-28 22:36:40 +0000941 StrAccum acc;
drh9ca95732014-10-24 00:35:58 +0000942
943#ifdef SQLITE_ENABLE_API_ARMOR
944 if( zFormat==0 ){
945 (void)SQLITE_MISUSE_BKPT;
946 return 0;
947 }
948#endif
drhff1590e2008-07-14 12:52:53 +0000949#ifndef SQLITE_OMIT_AUTOINIT
950 if( sqlite3_initialize() ) return 0;
951#endif
drhc0490572015-05-02 11:45:53 +0000952 sqlite3StrAccumInit(&acc, 0, zBase, sizeof(zBase), SQLITE_MAX_LENGTH);
drhf089aa42008-07-08 19:34:06 +0000953 sqlite3VXPrintf(&acc, 0, zFormat, ap);
drhade86482007-11-28 22:36:40 +0000954 z = sqlite3StrAccumFinish(&acc);
955 return z;
drh28dd4792006-06-26 21:35:44 +0000956}
957
958/*
959** Print into memory obtained from sqlite3_malloc()(). Omit the internal
960** %-conversion extensions.
drh483750b2003-01-29 18:46:51 +0000961*/
danielk19776f8a5032004-05-10 10:34:51 +0000962char *sqlite3_mprintf(const char *zFormat, ...){
drha18c5682000-10-08 22:20:57 +0000963 va_list ap;
drh5f968432004-02-21 19:02:30 +0000964 char *z;
drhff1590e2008-07-14 12:52:53 +0000965#ifndef SQLITE_OMIT_AUTOINIT
966 if( sqlite3_initialize() ) return 0;
967#endif
drh28dd4792006-06-26 21:35:44 +0000968 va_start(ap, zFormat);
drhb3738b62007-03-31 15:02:49 +0000969 z = sqlite3_vmprintf(zFormat, ap);
drha18c5682000-10-08 22:20:57 +0000970 va_end(ap);
drh5f968432004-02-21 19:02:30 +0000971 return z;
drha18c5682000-10-08 22:20:57 +0000972}
973
drh93a5c6b2003-12-23 02:17:35 +0000974/*
danielk19776f8a5032004-05-10 10:34:51 +0000975** sqlite3_snprintf() works like snprintf() except that it ignores the
drh93a5c6b2003-12-23 02:17:35 +0000976** current locale settings. This is important for SQLite because we
977** are not able to use a "," as the decimal point in place of "." as
978** specified by some locales.
drhdb26d4c2011-01-05 12:20:09 +0000979**
980** Oops: The first two arguments of sqlite3_snprintf() are backwards
981** from the snprintf() standard. Unfortunately, it is too late to change
982** this without breaking compatibility, so we just have to live with the
983** mistake.
984**
985** sqlite3_vsnprintf() is the varargs version.
drh93a5c6b2003-12-23 02:17:35 +0000986*/
drhdb26d4c2011-01-05 12:20:09 +0000987char *sqlite3_vsnprintf(int n, char *zBuf, const char *zFormat, va_list ap){
988 StrAccum acc;
989 if( n<=0 ) return zBuf;
drh9ca95732014-10-24 00:35:58 +0000990#ifdef SQLITE_ENABLE_API_ARMOR
991 if( zBuf==0 || zFormat==0 ) {
992 (void)SQLITE_MISUSE_BKPT;
drh96c707a2015-02-13 16:36:14 +0000993 if( zBuf ) zBuf[0] = 0;
drh9ca95732014-10-24 00:35:58 +0000994 return zBuf;
995 }
996#endif
drhc0490572015-05-02 11:45:53 +0000997 sqlite3StrAccumInit(&acc, 0, zBuf, n, 0);
drhdb26d4c2011-01-05 12:20:09 +0000998 sqlite3VXPrintf(&acc, 0, zFormat, ap);
999 return sqlite3StrAccumFinish(&acc);
1000}
danielk19776f8a5032004-05-10 10:34:51 +00001001char *sqlite3_snprintf(int n, char *zBuf, const char *zFormat, ...){
drh5f968432004-02-21 19:02:30 +00001002 char *z;
drh93a5c6b2003-12-23 02:17:35 +00001003 va_list ap;
drh93a5c6b2003-12-23 02:17:35 +00001004 va_start(ap,zFormat);
drhdb26d4c2011-01-05 12:20:09 +00001005 z = sqlite3_vsnprintf(n, zBuf, zFormat, ap);
drh93a5c6b2003-12-23 02:17:35 +00001006 va_end(ap);
drh5f968432004-02-21 19:02:30 +00001007 return z;
drh93a5c6b2003-12-23 02:17:35 +00001008}
1009
drh3f280702010-02-18 18:45:09 +00001010/*
drh7c0c4602010-03-03 22:25:18 +00001011** This is the routine that actually formats the sqlite3_log() message.
1012** We house it in a separate routine from sqlite3_log() to avoid using
1013** stack space on small-stack systems when logging is disabled.
1014**
1015** sqlite3_log() must render into a static buffer. It cannot dynamically
1016** allocate memory because it might be called while the memory allocator
1017** mutex is held.
1018*/
1019static void renderLogMsg(int iErrCode, const char *zFormat, va_list ap){
drha64fa912010-03-04 00:53:32 +00001020 StrAccum acc; /* String accumulator */
1021 char zMsg[SQLITE_PRINT_BUF_SIZE*3]; /* Complete log message */
drh7c0c4602010-03-03 22:25:18 +00001022
drhc0490572015-05-02 11:45:53 +00001023 sqlite3StrAccumInit(&acc, 0, zMsg, sizeof(zMsg), 0);
drh7c0c4602010-03-03 22:25:18 +00001024 sqlite3VXPrintf(&acc, 0, zFormat, ap);
1025 sqlite3GlobalConfig.xLog(sqlite3GlobalConfig.pLogArg, iErrCode,
1026 sqlite3StrAccumFinish(&acc));
1027}
1028
1029/*
drh3f280702010-02-18 18:45:09 +00001030** Format and write a message to the log if logging is enabled.
1031*/
drha7564662010-02-22 19:32:31 +00001032void sqlite3_log(int iErrCode, const char *zFormat, ...){
drh3f280702010-02-18 18:45:09 +00001033 va_list ap; /* Vararg list */
drh7c0c4602010-03-03 22:25:18 +00001034 if( sqlite3GlobalConfig.xLog ){
drh3f280702010-02-18 18:45:09 +00001035 va_start(ap, zFormat);
drh7c0c4602010-03-03 22:25:18 +00001036 renderLogMsg(iErrCode, zFormat, ap);
drh3f280702010-02-18 18:45:09 +00001037 va_end(ap);
drh3f280702010-02-18 18:45:09 +00001038 }
1039}
1040
mistachkin02b0e262015-04-16 03:37:19 +00001041#if defined(SQLITE_DEBUG) || defined(SQLITE_HAVE_OS_TRACE)
drhe54ca3f2004-06-07 01:52:14 +00001042/*
1043** A version of printf() that understands %lld. Used for debugging.
1044** The printf() built into some versions of windows does not understand %lld
1045** and segfaults if you give it a long long int.
1046*/
1047void sqlite3DebugPrintf(const char *zFormat, ...){
1048 va_list ap;
drhade86482007-11-28 22:36:40 +00001049 StrAccum acc;
drhe54ca3f2004-06-07 01:52:14 +00001050 char zBuf[500];
drhc0490572015-05-02 11:45:53 +00001051 sqlite3StrAccumInit(&acc, 0, zBuf, sizeof(zBuf), 0);
drhade86482007-11-28 22:36:40 +00001052 va_start(ap,zFormat);
drhf089aa42008-07-08 19:34:06 +00001053 sqlite3VXPrintf(&acc, 0, zFormat, ap);
drhe54ca3f2004-06-07 01:52:14 +00001054 va_end(ap);
drhbed8e7e2007-12-08 17:55:35 +00001055 sqlite3StrAccumFinish(&acc);
drh485f0032007-01-26 19:23:33 +00001056 fprintf(stdout,"%s", zBuf);
drh2ac3ee92004-06-07 16:27:46 +00001057 fflush(stdout);
drhe54ca3f2004-06-07 01:52:14 +00001058}
1059#endif
drhc7bc4fd2009-11-25 18:03:42 +00001060
drh4fa4a542014-09-30 12:33:33 +00001061#ifdef SQLITE_DEBUG
1062/*************************************************************************
1063** Routines for implementing the "TreeView" display of hierarchical
1064** data structures for debugging.
1065**
1066** The main entry points (coded elsewhere) are:
1067** sqlite3TreeViewExpr(0, pExpr, 0);
1068** sqlite3TreeViewExprList(0, pList, 0, 0);
1069** sqlite3TreeViewSelect(0, pSelect, 0);
1070** Insert calls to those routines while debugging in order to display
1071** a diagram of Expr, ExprList, and Select objects.
1072**
1073*/
1074/* Add a new subitem to the tree. The moreToFollow flag indicates that this
1075** is not the last item in the tree. */
1076TreeView *sqlite3TreeViewPush(TreeView *p, u8 moreToFollow){
1077 if( p==0 ){
drhf3cdcdc2015-04-29 16:50:28 +00001078 p = sqlite3_malloc64( sizeof(*p) );
drh4fa4a542014-09-30 12:33:33 +00001079 if( p==0 ) return 0;
1080 memset(p, 0, sizeof(*p));
1081 }else{
1082 p->iLevel++;
1083 }
1084 assert( moreToFollow==0 || moreToFollow==1 );
drhb08cd3f2014-09-30 19:04:41 +00001085 if( p->iLevel<sizeof(p->bLine) ) p->bLine[p->iLevel] = moreToFollow;
drh4fa4a542014-09-30 12:33:33 +00001086 return p;
1087}
1088/* Finished with one layer of the tree */
1089void sqlite3TreeViewPop(TreeView *p){
1090 if( p==0 ) return;
1091 p->iLevel--;
1092 if( p->iLevel<0 ) sqlite3_free(p);
1093}
1094/* Generate a single line of output for the tree, with a prefix that contains
1095** all the appropriate tree lines */
1096void sqlite3TreeViewLine(TreeView *p, const char *zFormat, ...){
1097 va_list ap;
1098 int i;
1099 StrAccum acc;
1100 char zBuf[500];
drhc0490572015-05-02 11:45:53 +00001101 sqlite3StrAccumInit(&acc, 0, zBuf, sizeof(zBuf), 0);
drh4fa4a542014-09-30 12:33:33 +00001102 if( p ){
drhb08cd3f2014-09-30 19:04:41 +00001103 for(i=0; i<p->iLevel && i<sizeof(p->bLine)-1; i++){
1104 sqlite3StrAccumAppend(&acc, p->bLine[i] ? "| " : " ", 4);
drh4fa4a542014-09-30 12:33:33 +00001105 }
drhb08cd3f2014-09-30 19:04:41 +00001106 sqlite3StrAccumAppend(&acc, p->bLine[i] ? "|-- " : "'-- ", 4);
drh4fa4a542014-09-30 12:33:33 +00001107 }
1108 va_start(ap, zFormat);
1109 sqlite3VXPrintf(&acc, 0, zFormat, ap);
1110 va_end(ap);
1111 if( zBuf[acc.nChar-1]!='\n' ) sqlite3StrAccumAppend(&acc, "\n", 1);
1112 sqlite3StrAccumFinish(&acc);
1113 fprintf(stdout,"%s", zBuf);
1114 fflush(stdout);
1115}
1116/* Shorthand for starting a new tree item that consists of a single label */
1117void sqlite3TreeViewItem(TreeView *p, const char *zLabel, u8 moreToFollow){
1118 p = sqlite3TreeViewPush(p, moreToFollow);
1119 sqlite3TreeViewLine(p, "%s", zLabel);
1120}
1121#endif /* SQLITE_DEBUG */
1122
drhc7bc4fd2009-11-25 18:03:42 +00001123/*
1124** variable-argument wrapper around sqlite3VXPrintf().
1125*/
drha5c14162013-12-17 15:03:06 +00001126void sqlite3XPrintf(StrAccum *p, u32 bFlags, const char *zFormat, ...){
drhc7bc4fd2009-11-25 18:03:42 +00001127 va_list ap;
1128 va_start(ap,zFormat);
drha5c14162013-12-17 15:03:06 +00001129 sqlite3VXPrintf(p, bFlags, zFormat, ap);
drhc7bc4fd2009-11-25 18:03:42 +00001130 va_end(ap);
1131}