mprintf.c 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218
  1. /****************************************************************************
  2. *
  3. * $Id$
  4. *
  5. *************************************************************************
  6. *
  7. * Permission to use, copy, modify, and distribute this software for any
  8. * purpose with or without fee is hereby granted, provided that the above
  9. * copyright notice and this permission notice appear in all copies.
  10. *
  11. * THIS SOFTWARE IS PROVIDED ``AS IS'' AND WITHOUT ANY EXPRESS OR IMPLIED
  12. * WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED WARRANTIES OF
  13. * MERCHANTIBILITY AND FITNESS FOR A PARTICULAR PURPOSE. THE AUTHORS AND
  14. * CONTRIBUTORS ACCEPT NO RESPONSIBILITY IN ANY CONCEIVABLE MANNER.
  15. *
  16. * Purpose:
  17. * A merge of Bjorn Reese's format() function and Daniel's dsprintf()
  18. * 1.0. A full blooded printf() clone with full support for <num>$
  19. * everywhere (parameters, widths and precisions) including variabled
  20. * sized parameters (like doubles, long longs, long doubles and even
  21. * void * in 64-bit architectures).
  22. *
  23. * Current restrictions:
  24. * - Max 128 parameters
  25. * - No 'long double' support.
  26. *
  27. * If you ever want truly portable and good *printf() clones, the project that
  28. * took on from here is named 'Trio' and you find more details on the trio web
  29. * page at http://daniel.haxx.se/trio/
  30. */
  31. #include "setup.h"
  32. #include <sys/types.h>
  33. #include <stdio.h>
  34. #include <stdlib.h>
  35. #include <stdarg.h>
  36. #include <ctype.h>
  37. #include <string.h>
  38. #if defined(DJGPP) && (DJGPP_MINOR < 4)
  39. #undef _MPRINTF_REPLACE /* don't use x_was_used() here */
  40. #endif
  41. #include <curl/mprintf.h>
  42. #ifndef SIZEOF_SIZE_T
  43. /* default to 4 bytes for size_t unless defined in the config.h */
  44. #define SIZEOF_SIZE_T 4
  45. #endif
  46. #ifdef DPRINTF_DEBUG
  47. #define HAVE_LONGLONG
  48. #define LONG_LONG long long
  49. #define ENABLE_64BIT
  50. #endif
  51. #include "memory.h"
  52. /* The last #include file should be: */
  53. #include "memdebug.h"
  54. #define BUFFSIZE 256 /* buffer for long-to-str and float-to-str calcs */
  55. #define MAX_PARAMETERS 128 /* lame static limit */
  56. #undef TRUE
  57. #undef FALSE
  58. #undef BOOL
  59. #ifdef __cplusplus
  60. # define TRUE true
  61. # define FALSE false
  62. # define BOOL bool
  63. #else
  64. # define TRUE ((char)(1 == 1))
  65. # define FALSE ((char)(0 == 1))
  66. # define BOOL char
  67. #endif
  68. #ifdef _AMIGASF
  69. # undef FORMAT_INT
  70. #endif
  71. /* Lower-case digits. */
  72. static const char lower_digits[] = "0123456789abcdefghijklmnopqrstuvwxyz";
  73. /* Upper-case digits. */
  74. static const char upper_digits[] = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ";
  75. #define OUTCHAR(x) \
  76. do{ \
  77. if(stream((unsigned char)(x), (FILE *)data) != -1) \
  78. done++; \
  79. else \
  80. return done; /* return immediately on failure */ \
  81. } while(0)
  82. /* Data type to read from the arglist */
  83. typedef enum {
  84. FORMAT_UNKNOWN = 0,
  85. FORMAT_STRING,
  86. FORMAT_PTR,
  87. FORMAT_INT,
  88. FORMAT_INTPTR,
  89. FORMAT_LONG,
  90. FORMAT_LONGLONG,
  91. FORMAT_DOUBLE,
  92. FORMAT_LONGDOUBLE,
  93. FORMAT_WIDTH /* For internal use */
  94. } FormatType;
  95. /* convertion and display flags */
  96. enum {
  97. FLAGS_NEW = 0,
  98. FLAGS_SPACE = 1<<0,
  99. FLAGS_SHOWSIGN = 1<<1,
  100. FLAGS_LEFT = 1<<2,
  101. FLAGS_ALT = 1<<3,
  102. FLAGS_SHORT = 1<<4,
  103. FLAGS_LONG = 1<<5,
  104. FLAGS_LONGLONG = 1<<6,
  105. FLAGS_LONGDOUBLE = 1<<7,
  106. FLAGS_PAD_NIL = 1<<8,
  107. FLAGS_UNSIGNED = 1<<9,
  108. FLAGS_OCTAL = 1<<10,
  109. FLAGS_HEX = 1<<11,
  110. FLAGS_UPPER = 1<<12,
  111. FLAGS_WIDTH = 1<<13, /* '*' or '*<num>$' used */
  112. FLAGS_WIDTHPARAM = 1<<14, /* width PARAMETER was specified */
  113. FLAGS_PREC = 1<<15, /* precision was specified */
  114. FLAGS_PRECPARAM = 1<<16, /* precision PARAMETER was specified */
  115. FLAGS_CHAR = 1<<17, /* %c story */
  116. FLAGS_FLOATE = 1<<18, /* %e or %E */
  117. FLAGS_FLOATG = 1<<19 /* %g or %G */
  118. };
  119. typedef struct {
  120. FormatType type;
  121. int flags;
  122. long width; /* width OR width parameter number */
  123. long precision; /* precision OR precision parameter number */
  124. union {
  125. char *str;
  126. void *ptr;
  127. long num;
  128. #ifdef ENABLE_64BIT
  129. LONG_LONG lnum;
  130. #endif
  131. double dnum;
  132. } data;
  133. } va_stack_t;
  134. struct nsprintf {
  135. char *buffer;
  136. size_t length;
  137. size_t max;
  138. };
  139. struct asprintf {
  140. char *buffer; /* allocated buffer */
  141. size_t len; /* length of string */
  142. size_t alloc; /* length of alloc */
  143. bool fail; /* TRUE if an alloc has failed and thus the output is not
  144. the complete data */
  145. };
  146. int curl_msprintf(char *buffer, const char *format, ...);
  147. static long dprintf_DollarString(char *input, char **end)
  148. {
  149. int number=0;
  150. while(ISDIGIT(*input)) {
  151. number *= 10;
  152. number += *input-'0';
  153. input++;
  154. }
  155. if(number && ('$'==*input++)) {
  156. *end = input;
  157. return number;
  158. }
  159. return 0;
  160. }
  161. static BOOL dprintf_IsQualifierNoDollar(char c)
  162. {
  163. switch (c) {
  164. case '-': case '+': case ' ': case '#': case '.':
  165. case '0': case '1': case '2': case '3': case '4':
  166. case '5': case '6': case '7': case '8': case '9':
  167. case 'h': case 'l': case 'L': case 'z': case 'q':
  168. case '*': case 'O':
  169. return TRUE;
  170. default:
  171. return FALSE;
  172. }
  173. }
  174. #ifdef DPRINTF_DEBUG2
  175. int dprintf_Pass1Report(va_stack_t *vto, int max)
  176. {
  177. int i;
  178. char buffer[128];
  179. int bit;
  180. int flags;
  181. for(i=0; i<max; i++) {
  182. char *type;
  183. switch(vto[i].type) {
  184. case FORMAT_UNKNOWN:
  185. type = "unknown";
  186. break;
  187. case FORMAT_STRING:
  188. type ="string";
  189. break;
  190. case FORMAT_PTR:
  191. type ="pointer";
  192. break;
  193. case FORMAT_INT:
  194. type = "int";
  195. break;
  196. case FORMAT_LONG:
  197. type = "long";
  198. break;
  199. case FORMAT_LONGLONG:
  200. type = "long long";
  201. break;
  202. case FORMAT_DOUBLE:
  203. type = "double";
  204. break;
  205. case FORMAT_LONGDOUBLE:
  206. type = "long double";
  207. break;
  208. }
  209. buffer[0]=0;
  210. for(bit=0; bit<31; bit++) {
  211. flags = vto[i].flags & (1<<bit);
  212. if(flags & FLAGS_SPACE)
  213. strcat(buffer, "space ");
  214. else if(flags & FLAGS_SHOWSIGN)
  215. strcat(buffer, "plus ");
  216. else if(flags & FLAGS_LEFT)
  217. strcat(buffer, "left ");
  218. else if(flags & FLAGS_ALT)
  219. strcat(buffer, "alt ");
  220. else if(flags & FLAGS_SHORT)
  221. strcat(buffer, "short ");
  222. else if(flags & FLAGS_LONG)
  223. strcat(buffer, "long ");
  224. else if(flags & FLAGS_LONGLONG)
  225. strcat(buffer, "longlong ");
  226. else if(flags & FLAGS_LONGDOUBLE)
  227. strcat(buffer, "longdouble ");
  228. else if(flags & FLAGS_PAD_NIL)
  229. strcat(buffer, "padnil ");
  230. else if(flags & FLAGS_UNSIGNED)
  231. strcat(buffer, "unsigned ");
  232. else if(flags & FLAGS_OCTAL)
  233. strcat(buffer, "octal ");
  234. else if(flags & FLAGS_HEX)
  235. strcat(buffer, "hex ");
  236. else if(flags & FLAGS_UPPER)
  237. strcat(buffer, "upper ");
  238. else if(flags & FLAGS_WIDTH)
  239. strcat(buffer, "width ");
  240. else if(flags & FLAGS_WIDTHPARAM)
  241. strcat(buffer, "widthparam ");
  242. else if(flags & FLAGS_PREC)
  243. strcat(buffer, "precision ");
  244. else if(flags & FLAGS_PRECPARAM)
  245. strcat(buffer, "precparam ");
  246. else if(flags & FLAGS_CHAR)
  247. strcat(buffer, "char ");
  248. else if(flags & FLAGS_FLOATE)
  249. strcat(buffer, "floate ");
  250. else if(flags & FLAGS_FLOATG)
  251. strcat(buffer, "floatg ");
  252. }
  253. printf("REPORT: %d. %s [%s]\n", i, type, buffer);
  254. }
  255. }
  256. #endif
  257. /******************************************************************
  258. *
  259. * Pass 1:
  260. * Create an index with the type of each parameter entry and its
  261. * value (may vary in size)
  262. *
  263. ******************************************************************/
  264. static long dprintf_Pass1(char *format, va_stack_t *vto, char **endpos,
  265. va_list arglist)
  266. {
  267. char *fmt = format;
  268. int param_num = 0;
  269. long this_param;
  270. long width;
  271. long precision;
  272. int flags;
  273. long max_param=0;
  274. long i;
  275. while (*fmt) {
  276. if (*fmt++ == '%') {
  277. if (*fmt == '%') {
  278. fmt++;
  279. continue; /* while */
  280. }
  281. flags = FLAGS_NEW;
  282. /* Handle the positional case (N$) */
  283. param_num++;
  284. this_param = dprintf_DollarString(fmt, &fmt);
  285. if (0 == this_param)
  286. /* we got no positional, get the next counter */
  287. this_param = param_num;
  288. if (this_param > max_param)
  289. max_param = this_param;
  290. /*
  291. * The parameter with number 'i' should be used. Next, we need
  292. * to get SIZE and TYPE of the parameter. Add the information
  293. * to our array.
  294. */
  295. width = 0;
  296. precision = 0;
  297. /* Handle the flags */
  298. while (dprintf_IsQualifierNoDollar(*fmt)) {
  299. switch (*fmt++) {
  300. case ' ':
  301. flags |= FLAGS_SPACE;
  302. break;
  303. case '+':
  304. flags |= FLAGS_SHOWSIGN;
  305. break;
  306. case '-':
  307. flags |= FLAGS_LEFT;
  308. flags &= ~FLAGS_PAD_NIL;
  309. break;
  310. case '#':
  311. flags |= FLAGS_ALT;
  312. break;
  313. case '.':
  314. flags |= FLAGS_PREC;
  315. if ('*' == *fmt) {
  316. /* The precision is picked from a specified parameter */
  317. flags |= FLAGS_PRECPARAM;
  318. fmt++;
  319. param_num++;
  320. i = dprintf_DollarString(fmt, &fmt);
  321. if (i)
  322. precision = i;
  323. else
  324. precision = param_num;
  325. if (precision > max_param)
  326. max_param = precision;
  327. }
  328. else {
  329. flags |= FLAGS_PREC;
  330. precision = strtol(fmt, &fmt, 10);
  331. }
  332. break;
  333. case 'h':
  334. flags |= FLAGS_SHORT;
  335. break;
  336. case 'l':
  337. if (flags & FLAGS_LONG)
  338. flags |= FLAGS_LONGLONG;
  339. else
  340. flags |= FLAGS_LONG;
  341. break;
  342. case 'L':
  343. flags |= FLAGS_LONGDOUBLE;
  344. break;
  345. case 'q':
  346. flags |= FLAGS_LONGLONG;
  347. break;
  348. case 'z':
  349. /* the code below generates a warning if -Wunreachable-code is
  350. used */
  351. #if SIZEOF_SIZE_T>4
  352. flags |= FLAGS_LONGLONG;
  353. #else
  354. flags |= FLAGS_LONG;
  355. #endif
  356. break;
  357. case 'O':
  358. #if SIZEOF_CURL_OFF_T > 4
  359. flags |= FLAGS_LONGLONG;
  360. #else
  361. flags |= FLAGS_LONG;
  362. #endif
  363. break;
  364. case '0':
  365. if (!(flags & FLAGS_LEFT))
  366. flags |= FLAGS_PAD_NIL;
  367. /* FALLTHROUGH */
  368. case '1': case '2': case '3': case '4':
  369. case '5': case '6': case '7': case '8': case '9':
  370. flags |= FLAGS_WIDTH;
  371. width = strtol(fmt-1, &fmt, 10);
  372. break;
  373. case '*': /* Special case */
  374. flags |= FLAGS_WIDTHPARAM;
  375. param_num++;
  376. i = dprintf_DollarString(fmt, &fmt);
  377. if(i)
  378. width = i;
  379. else
  380. width = param_num;
  381. if(width > max_param)
  382. max_param=width;
  383. break;
  384. default:
  385. break;
  386. }
  387. } /* switch */
  388. /* Handle the specifier */
  389. i = this_param - 1;
  390. switch (*fmt) {
  391. case 'S':
  392. flags |= FLAGS_ALT;
  393. /* FALLTHROUGH */
  394. case 's':
  395. vto[i].type = FORMAT_STRING;
  396. break;
  397. case 'n':
  398. vto[i].type = FORMAT_INTPTR;
  399. break;
  400. case 'p':
  401. vto[i].type = FORMAT_PTR;
  402. break;
  403. case 'd': case 'i':
  404. vto[i].type = FORMAT_INT;
  405. break;
  406. case 'u':
  407. vto[i].type = FORMAT_INT;
  408. flags |= FLAGS_UNSIGNED;
  409. break;
  410. case 'o':
  411. vto[i].type = FORMAT_INT;
  412. flags |= FLAGS_OCTAL;
  413. break;
  414. case 'x':
  415. vto[i].type = FORMAT_INT;
  416. flags |= FLAGS_HEX;
  417. break;
  418. case 'X':
  419. vto[i].type = FORMAT_INT;
  420. flags |= FLAGS_HEX|FLAGS_UPPER;
  421. break;
  422. case 'c':
  423. vto[i].type = FORMAT_INT;
  424. flags |= FLAGS_CHAR;
  425. break;
  426. case 'f':
  427. vto[i].type = FORMAT_DOUBLE;
  428. break;
  429. case 'e':
  430. vto[i].type = FORMAT_DOUBLE;
  431. flags |= FLAGS_FLOATE;
  432. break;
  433. case 'E':
  434. vto[i].type = FORMAT_DOUBLE;
  435. flags |= FLAGS_FLOATE|FLAGS_UPPER;
  436. break;
  437. case 'g':
  438. vto[i].type = FORMAT_DOUBLE;
  439. flags |= FLAGS_FLOATG;
  440. break;
  441. case 'G':
  442. vto[i].type = FORMAT_DOUBLE;
  443. flags |= FLAGS_FLOATG|FLAGS_UPPER;
  444. break;
  445. default:
  446. vto[i].type = FORMAT_UNKNOWN;
  447. break;
  448. } /* switch */
  449. vto[i].flags = flags;
  450. vto[i].width = width;
  451. vto[i].precision = precision;
  452. if (flags & FLAGS_WIDTHPARAM) {
  453. /* we have the width specified from a parameter, so we make that
  454. parameter's info setup properly */
  455. vto[i].width = width - 1;
  456. i = width - 1;
  457. vto[i].type = FORMAT_WIDTH;
  458. vto[i].flags = FLAGS_NEW;
  459. vto[i].precision = vto[i].width = 0; /* can't use width or precision
  460. of width! */
  461. }
  462. if (flags & FLAGS_PRECPARAM) {
  463. /* we have the precision specified from a parameter, so we make that
  464. parameter's info setup properly */
  465. vto[i].precision = precision - 1;
  466. i = precision - 1;
  467. vto[i].type = FORMAT_WIDTH;
  468. vto[i].flags = FLAGS_NEW;
  469. vto[i].precision = vto[i].width = 0; /* can't use width or precision
  470. of width! */
  471. }
  472. *endpos++ = fmt + 1; /* end of this sequence */
  473. }
  474. }
  475. #ifdef DPRINTF_DEBUG2
  476. dprintf_Pass1Report(vto, max_param);
  477. #endif
  478. /* Read the arg list parameters into our data list */
  479. for (i=0; i<max_param; i++) {
  480. if ((i + 1 < max_param) && (vto[i + 1].type == FORMAT_WIDTH))
  481. {
  482. /* Width/precision arguments must be read before the main argument
  483. * they are attached to
  484. */
  485. vto[i + 1].data.num = va_arg(arglist, int);
  486. }
  487. switch (vto[i].type)
  488. {
  489. case FORMAT_STRING:
  490. vto[i].data.str = va_arg(arglist, char *);
  491. break;
  492. case FORMAT_INTPTR:
  493. case FORMAT_UNKNOWN:
  494. case FORMAT_PTR:
  495. vto[i].data.ptr = va_arg(arglist, void *);
  496. break;
  497. case FORMAT_INT:
  498. #ifdef ENABLE_64BIT
  499. if(vto[i].flags & FLAGS_LONGLONG)
  500. vto[i].data.lnum = va_arg(arglist, LONG_LONG);
  501. else
  502. #endif
  503. if(vto[i].flags & FLAGS_LONG)
  504. vto[i].data.num = va_arg(arglist, long);
  505. else
  506. vto[i].data.num = va_arg(arglist, int);
  507. break;
  508. case FORMAT_DOUBLE:
  509. vto[i].data.dnum = va_arg(arglist, double);
  510. break;
  511. case FORMAT_WIDTH:
  512. /* Argument has been read. Silently convert it into an integer
  513. * for later use
  514. */
  515. vto[i].type = FORMAT_INT;
  516. break;
  517. default:
  518. break;
  519. }
  520. }
  521. return max_param;
  522. }
  523. static int dprintf_formatf(
  524. void *data, /* untouched by format(), just sent to the stream() function in
  525. the second argument */
  526. /* function pointer called for each output character */
  527. int (*stream)(int, FILE *),
  528. const char *format, /* %-formatted string */
  529. va_list ap_save) /* list of parameters */
  530. {
  531. /* Base-36 digits for numbers. */
  532. const char *digits = lower_digits;
  533. /* Pointer into the format string. */
  534. char *f;
  535. /* Number of characters written. */
  536. int done = 0;
  537. long param; /* current parameter to read */
  538. long param_num=0; /* parameter counter */
  539. va_stack_t vto[MAX_PARAMETERS];
  540. char *endpos[MAX_PARAMETERS];
  541. char **end;
  542. char work[BUFFSIZE];
  543. va_stack_t *p;
  544. /* Do the actual %-code parsing */
  545. dprintf_Pass1((char *)format, vto, endpos, ap_save);
  546. end = &endpos[0]; /* the initial end-position from the list dprintf_Pass1()
  547. created for us */
  548. f = (char *)format;
  549. while (*f != '\0') {
  550. /* Format spec modifiers. */
  551. char alt;
  552. /* Width of a field. */
  553. long width;
  554. /* Precision of a field. */
  555. long prec;
  556. /* Decimal integer is negative. */
  557. char is_neg;
  558. /* Base of a number to be written. */
  559. long base;
  560. /* Integral values to be written. */
  561. #ifdef ENABLE_64BIT
  562. unsigned LONG_LONG num;
  563. #else
  564. unsigned long num;
  565. #endif
  566. long signed_num;
  567. if (*f != '%') {
  568. /* This isn't a format spec, so write everything out until the next one
  569. OR end of string is reached. */
  570. do {
  571. OUTCHAR(*f);
  572. } while(*++f && ('%' != *f));
  573. continue;
  574. }
  575. ++f;
  576. /* Check for "%%". Note that although the ANSI standard lists
  577. '%' as a conversion specifier, it says "The complete format
  578. specification shall be `%%'," so we can avoid all the width
  579. and precision processing. */
  580. if (*f == '%') {
  581. ++f;
  582. OUTCHAR('%');
  583. continue;
  584. }
  585. /* If this is a positional parameter, the position must follow imediately
  586. after the %, thus create a %<num>$ sequence */
  587. param=dprintf_DollarString(f, &f);
  588. if(!param)
  589. param = param_num;
  590. else
  591. --param;
  592. param_num++; /* increase this always to allow "%2$s %1$s %s" and then the
  593. third %s will pick the 3rd argument */
  594. p = &vto[param];
  595. /* pick up the specified width */
  596. if(p->flags & FLAGS_WIDTHPARAM)
  597. width = vto[p->width].data.num;
  598. else
  599. width = p->width;
  600. /* pick up the specified precision */
  601. if(p->flags & FLAGS_PRECPARAM)
  602. prec = vto[p->precision].data.num;
  603. else if(p->flags & FLAGS_PREC)
  604. prec = p->precision;
  605. else
  606. prec = -1;
  607. alt = (p->flags & FLAGS_ALT)?TRUE:FALSE;
  608. switch (p->type) {
  609. case FORMAT_INT:
  610. num = p->data.num;
  611. if(p->flags & FLAGS_CHAR) {
  612. /* Character. */
  613. if (!(p->flags & FLAGS_LEFT))
  614. while (--width > 0)
  615. OUTCHAR(' ');
  616. OUTCHAR((char) num);
  617. if (p->flags & FLAGS_LEFT)
  618. while (--width > 0)
  619. OUTCHAR(' ');
  620. break;
  621. }
  622. if(p->flags & FLAGS_UNSIGNED) {
  623. /* Decimal unsigned integer. */
  624. base = 10;
  625. goto unsigned_number;
  626. }
  627. if(p->flags & FLAGS_OCTAL) {
  628. /* Octal unsigned integer. */
  629. base = 8;
  630. goto unsigned_number;
  631. }
  632. if(p->flags & FLAGS_HEX) {
  633. /* Hexadecimal unsigned integer. */
  634. digits = (p->flags & FLAGS_UPPER)? upper_digits : lower_digits;
  635. base = 16;
  636. goto unsigned_number;
  637. }
  638. /* Decimal integer. */
  639. base = 10;
  640. #ifdef ENABLE_64BIT
  641. if(p->flags & FLAGS_LONGLONG) {
  642. /* long long */
  643. is_neg = p->data.lnum < 0;
  644. num = is_neg ? (- p->data.lnum) : p->data.lnum;
  645. }
  646. else
  647. #endif
  648. {
  649. signed_num = (long) num;
  650. is_neg = signed_num < 0;
  651. num = is_neg ? (- signed_num) : signed_num;
  652. }
  653. goto number;
  654. unsigned_number:
  655. /* Unsigned number of base BASE. */
  656. is_neg = 0;
  657. number:
  658. /* Number of base BASE. */
  659. {
  660. char *workend = &work[sizeof(work) - 1];
  661. char *w;
  662. /* Supply a default precision if none was given. */
  663. if (prec == -1)
  664. prec = 1;
  665. /* Put the number in WORK. */
  666. w = workend;
  667. while (num > 0) {
  668. *w-- = digits[num % base];
  669. num /= base;
  670. }
  671. width -= (long)(workend - w);
  672. prec -= (long)(workend - w);
  673. if (alt && base == 8 && prec <= 0) {
  674. *w-- = '0';
  675. --width;
  676. }
  677. if (prec > 0) {
  678. width -= prec;
  679. while (prec-- > 0)
  680. *w-- = '0';
  681. }
  682. if (alt && base == 16)
  683. width -= 2;
  684. if (is_neg || (p->flags & FLAGS_SHOWSIGN) || (p->flags & FLAGS_SPACE))
  685. --width;
  686. if (!(p->flags & FLAGS_LEFT) && !(p->flags & FLAGS_PAD_NIL))
  687. while (width-- > 0)
  688. OUTCHAR(' ');
  689. if (is_neg)
  690. OUTCHAR('-');
  691. else if (p->flags & FLAGS_SHOWSIGN)
  692. OUTCHAR('+');
  693. else if (p->flags & FLAGS_SPACE)
  694. OUTCHAR(' ');
  695. if (alt && base == 16) {
  696. OUTCHAR('0');
  697. if(p->flags & FLAGS_UPPER)
  698. OUTCHAR('X');
  699. else
  700. OUTCHAR('x');
  701. }
  702. if (!(p->flags & FLAGS_LEFT) && (p->flags & FLAGS_PAD_NIL))
  703. while (width-- > 0)
  704. OUTCHAR('0');
  705. /* Write the number. */
  706. while (++w <= workend) {
  707. OUTCHAR(*w);
  708. }
  709. if (p->flags & FLAGS_LEFT)
  710. while (width-- > 0)
  711. OUTCHAR(' ');
  712. }
  713. break;
  714. case FORMAT_STRING:
  715. /* String. */
  716. {
  717. static const char null[] = "(nil)";
  718. const char *str;
  719. size_t len;
  720. str = (char *) p->data.str;
  721. if ( str == NULL) {
  722. /* Write null[] if there's space. */
  723. if (prec == -1 || prec >= (long) sizeof(null) - 1) {
  724. str = null;
  725. len = sizeof(null) - 1;
  726. /* Disable quotes around (nil) */
  727. p->flags &= (~FLAGS_ALT);
  728. }
  729. else {
  730. str = "";
  731. len = 0;
  732. }
  733. }
  734. else
  735. len = strlen(str);
  736. if (prec != -1 && (size_t) prec < len)
  737. len = prec;
  738. width -= (long)len;
  739. if (p->flags & FLAGS_ALT)
  740. OUTCHAR('"');
  741. if (!(p->flags&FLAGS_LEFT))
  742. while (width-- > 0)
  743. OUTCHAR(' ');
  744. while (len-- > 0)
  745. OUTCHAR(*str++);
  746. if (p->flags&FLAGS_LEFT)
  747. while (width-- > 0)
  748. OUTCHAR(' ');
  749. if (p->flags & FLAGS_ALT)
  750. OUTCHAR('"');
  751. }
  752. break;
  753. case FORMAT_PTR:
  754. /* Generic pointer. */
  755. {
  756. void *ptr;
  757. ptr = (void *) p->data.ptr;
  758. if (ptr != NULL) {
  759. /* If the pointer is not NULL, write it as a %#x spec. */
  760. base = 16;
  761. digits = (p->flags & FLAGS_UPPER)? upper_digits : lower_digits;
  762. alt = 1;
  763. num = (size_t) ptr;
  764. is_neg = 0;
  765. goto number;
  766. }
  767. else {
  768. /* Write "(nil)" for a nil pointer. */
  769. static const char strnil[] = "(nil)";
  770. const char *point;
  771. width -= sizeof(strnil) - 1;
  772. if (p->flags & FLAGS_LEFT)
  773. while (width-- > 0)
  774. OUTCHAR(' ');
  775. for (point = strnil; *point != '\0'; ++point)
  776. OUTCHAR(*point);
  777. if (! (p->flags & FLAGS_LEFT))
  778. while (width-- > 0)
  779. OUTCHAR(' ');
  780. }
  781. }
  782. break;
  783. case FORMAT_DOUBLE:
  784. {
  785. char formatbuf[32]="%";
  786. char *fptr;
  787. size_t left = sizeof(formatbuf)-strlen(formatbuf);
  788. int len;
  789. width = -1;
  790. if (p->flags & FLAGS_WIDTH)
  791. width = p->width;
  792. else if (p->flags & FLAGS_WIDTHPARAM)
  793. width = vto[p->width].data.num;
  794. prec = -1;
  795. if (p->flags & FLAGS_PREC)
  796. prec = p->precision;
  797. else if (p->flags & FLAGS_PRECPARAM)
  798. prec = vto[p->precision].data.num;
  799. if (p->flags & FLAGS_LEFT)
  800. strcat(formatbuf, "-");
  801. if (p->flags & FLAGS_SHOWSIGN)
  802. strcat(formatbuf, "+");
  803. if (p->flags & FLAGS_SPACE)
  804. strcat(formatbuf, " ");
  805. if (p->flags & FLAGS_ALT)
  806. strcat(formatbuf, "#");
  807. fptr=&formatbuf[strlen(formatbuf)];
  808. if(width >= 0) {
  809. /* RECURSIVE USAGE */
  810. len = curl_msnprintf(fptr, left, "%ld", width);
  811. fptr += len;
  812. left -= len;
  813. }
  814. if(prec >= 0) {
  815. /* RECURSIVE USAGE */
  816. len = curl_msnprintf(fptr, left, ".%ld", prec);
  817. fptr += len;
  818. left -= len;
  819. }
  820. if (p->flags & FLAGS_LONG)
  821. *fptr++ = 'l';
  822. if (p->flags & FLAGS_FLOATE)
  823. *fptr++ = p->flags&FLAGS_UPPER ? 'E':'e';
  824. else if (p->flags & FLAGS_FLOATG)
  825. *fptr++ = p->flags & FLAGS_UPPER ? 'G' : 'g';
  826. else
  827. *fptr++ = 'f';
  828. *fptr = 0; /* and a final zero termination */
  829. /* NOTE NOTE NOTE!! Not all sprintf() implementations returns number
  830. of output characters */
  831. (sprintf)(work, formatbuf, p->data.dnum);
  832. for(fptr=work; *fptr; fptr++)
  833. OUTCHAR(*fptr);
  834. }
  835. break;
  836. case FORMAT_INTPTR:
  837. /* Answer the count of characters written. */
  838. #ifdef ENABLE_64BIT
  839. if (p->flags & FLAGS_LONGLONG)
  840. *(LONG_LONG *) p->data.ptr = (LONG_LONG)done;
  841. else
  842. #endif
  843. if (p->flags & FLAGS_LONG)
  844. *(long *) p->data.ptr = (long)done;
  845. else if (!(p->flags & FLAGS_SHORT))
  846. *(int *) p->data.ptr = (int)done;
  847. else
  848. *(short *) p->data.ptr = (short)done;
  849. break;
  850. default:
  851. break;
  852. }
  853. f = *end++; /* goto end of %-code */
  854. }
  855. return done;
  856. }
  857. /* fputc() look-alike */
  858. static int addbyter(int output, FILE *data)
  859. {
  860. struct nsprintf *infop=(struct nsprintf *)data;
  861. unsigned char outc = (unsigned char)output;
  862. if(infop->length < infop->max) {
  863. /* only do this if we haven't reached max length yet */
  864. infop->buffer[0] = outc; /* store */
  865. infop->buffer++; /* increase pointer */
  866. infop->length++; /* we are now one byte larger */
  867. return outc; /* fputc() returns like this on success */
  868. }
  869. return -1;
  870. }
  871. int curl_mvsnprintf(char *buffer, size_t maxlength, const char *format,
  872. va_list ap_save)
  873. {
  874. int retcode;
  875. struct nsprintf info;
  876. info.buffer = buffer;
  877. info.length = 0;
  878. info.max = maxlength;
  879. retcode = dprintf_formatf(&info, addbyter, format, ap_save);
  880. if(info.max) {
  881. /* we terminate this with a zero byte */
  882. if(info.max == info.length)
  883. /* we're at maximum, scrap the last letter */
  884. info.buffer[-1] = 0;
  885. else
  886. info.buffer[0] = 0;
  887. }
  888. return retcode;
  889. }
  890. int curl_msnprintf(char *buffer, size_t maxlength, const char *format, ...)
  891. {
  892. int retcode;
  893. va_list ap_save; /* argument pointer */
  894. va_start(ap_save, format);
  895. retcode = curl_mvsnprintf(buffer, maxlength, format, ap_save);
  896. va_end(ap_save);
  897. return retcode;
  898. }
  899. /* fputc() look-alike */
  900. static int alloc_addbyter(int output, FILE *data)
  901. {
  902. struct asprintf *infop=(struct asprintf *)data;
  903. unsigned char outc = (unsigned char)output;
  904. if(!infop->buffer) {
  905. infop->buffer=(char *)malloc(32);
  906. if(!infop->buffer) {
  907. infop->fail = TRUE;
  908. return -1; /* fail */
  909. }
  910. infop->alloc = 32;
  911. infop->len =0;
  912. }
  913. else if(infop->len+1 >= infop->alloc) {
  914. char *newptr;
  915. newptr = (char *)realloc(infop->buffer, infop->alloc*2);
  916. if(!newptr) {
  917. infop->fail = TRUE;
  918. return -1;
  919. }
  920. infop->buffer = newptr;
  921. infop->alloc *= 2;
  922. }
  923. infop->buffer[ infop->len ] = outc;
  924. infop->len++;
  925. return outc; /* fputc() returns like this on success */
  926. }
  927. char *curl_maprintf(const char *format, ...)
  928. {
  929. va_list ap_save; /* argument pointer */
  930. int retcode;
  931. struct asprintf info;
  932. info.buffer = NULL;
  933. info.len = 0;
  934. info.alloc = 0;
  935. info.fail = FALSE;
  936. va_start(ap_save, format);
  937. retcode = dprintf_formatf(&info, alloc_addbyter, format, ap_save);
  938. va_end(ap_save);
  939. if((-1 == retcode) || info.fail) {
  940. if(info.alloc)
  941. free(info.buffer);
  942. return NULL;
  943. }
  944. if(info.alloc) {
  945. info.buffer[info.len] = 0; /* we terminate this with a zero byte */
  946. return info.buffer;
  947. }
  948. else
  949. return strdup("");
  950. }
  951. char *curl_mvaprintf(const char *format, va_list ap_save)
  952. {
  953. int retcode;
  954. struct asprintf info;
  955. info.buffer = NULL;
  956. info.len = 0;
  957. info.alloc = 0;
  958. info.fail = FALSE;
  959. retcode = dprintf_formatf(&info, alloc_addbyter, format, ap_save);
  960. if((-1 == retcode) || info.fail) {
  961. if(info.alloc)
  962. free(info.buffer);
  963. return NULL;
  964. }
  965. if(info.alloc) {
  966. info.buffer[info.len] = 0; /* we terminate this with a zero byte */
  967. return info.buffer;
  968. }
  969. else
  970. return strdup("");
  971. }
  972. static int storebuffer(int output, FILE *data)
  973. {
  974. char **buffer = (char **)data;
  975. unsigned char outc = (unsigned char)output;
  976. **buffer = outc;
  977. (*buffer)++;
  978. return outc; /* act like fputc() ! */
  979. }
  980. int curl_msprintf(char *buffer, const char *format, ...)
  981. {
  982. va_list ap_save; /* argument pointer */
  983. int retcode;
  984. va_start(ap_save, format);
  985. retcode = dprintf_formatf(&buffer, storebuffer, format, ap_save);
  986. va_end(ap_save);
  987. *buffer=0; /* we terminate this with a zero byte */
  988. return retcode;
  989. }
  990. int curl_mprintf(const char *format, ...)
  991. {
  992. int retcode;
  993. va_list ap_save; /* argument pointer */
  994. va_start(ap_save, format);
  995. retcode = dprintf_formatf(stdout, fputc, format, ap_save);
  996. va_end(ap_save);
  997. return retcode;
  998. }
  999. int curl_mfprintf(FILE *whereto, const char *format, ...)
  1000. {
  1001. int retcode;
  1002. va_list ap_save; /* argument pointer */
  1003. va_start(ap_save, format);
  1004. retcode = dprintf_formatf(whereto, fputc, format, ap_save);
  1005. va_end(ap_save);
  1006. return retcode;
  1007. }
  1008. int curl_mvsprintf(char *buffer, const char *format, va_list ap_save)
  1009. {
  1010. int retcode;
  1011. retcode = dprintf_formatf(&buffer, storebuffer, format, ap_save);
  1012. *buffer=0; /* we terminate this with a zero byte */
  1013. return retcode;
  1014. }
  1015. int curl_mvprintf(const char *format, va_list ap_save)
  1016. {
  1017. return dprintf_formatf(stdout, fputc, format, ap_save);
  1018. }
  1019. int curl_mvfprintf(FILE *whereto, const char *format, va_list ap_save)
  1020. {
  1021. return dprintf_formatf(whereto, fputc, format, ap_save);
  1022. }
  1023. #ifdef DPRINTF_DEBUG
  1024. int main()
  1025. {
  1026. char buffer[129];
  1027. char *ptr;
  1028. #ifdef ENABLE_64BIT
  1029. long long one=99;
  1030. long long two=100;
  1031. long long test = 0x1000000000LL;
  1032. curl_mprintf("%lld %lld %lld\n", one, two, test);
  1033. #endif
  1034. curl_mprintf("%3d %5d\n", 10, 1998);
  1035. ptr=curl_maprintf("test this then baby %s%s%s%s%s%s %d %d %d loser baby get a hit in yer face now!", "", "pretty long string pretty long string pretty long string pretty long string pretty long string", "/", "/", "/", "pretty long string", 1998, 1999, 2001);
  1036. puts(ptr);
  1037. memset(ptr, 55, strlen(ptr)+1);
  1038. free(ptr);
  1039. #if 1
  1040. curl_mprintf(buffer, "%s %s %d", "daniel", "stenberg", 19988);
  1041. puts(buffer);
  1042. curl_mfprintf(stderr, "%s %#08x\n", "dummy", 65);
  1043. printf("%s %#08x\n", "dummy", 65);
  1044. {
  1045. double tryout = 3.14156592;
  1046. curl_mprintf(buffer, "%.2g %G %f %e %E", tryout, tryout, tryout, tryout, tryout);
  1047. puts(buffer);
  1048. printf("%.2g %G %f %e %E\n", tryout, tryout, tryout, tryout, tryout);
  1049. }
  1050. #endif
  1051. return 0;
  1052. }
  1053. #endif