File: snprintf.c

package info (click to toggle)
citadel 7.83-2squeeze2
  • links: PTS, VCS
  • area: main
  • in suites: squeeze
  • size: 4,572 kB
  • ctags: 3,756
  • sloc: ansic: 54,870; sh: 4,298; yacc: 660; makefile: 450; xml: 40
file content (81 lines) | stat: -rw-r--r-- 1,907 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
/*
 * $Id: snprintf.c 5148 2007-05-08 15:40:16Z ajc $
 *
 * Replacements for snprintf() and vsnprintf()
 *
 * modified from Sten Gunterberg's BUGTRAQ post of 22 Jul 1997
 * --nathan bryant <nathan@designtrust.com>
 *
 * Use it only if you have the "spare" cycles needed to effectively
 * do every snprintf operation twice! Why is that? Because everything
 * is first vfprintf()'d to /dev/null to determine the number of bytes.
 * Perhaps a bit slow for demanding applications on slow machines,
 * no problem for a fast machine with some spare cycles.
 *
 * You don't have a /dev/null? Every Linux contains one for free!
 *
 * Because the format string is never even looked at, all current and
 * possible future printf-conversions should be handled just fine.
 *
 * Written July 1997 by Sten Gunterberg (gunterberg@ergon.ch)
 */

#include <stdio.h>
#include <stdlib.h>
#include <stdarg.h>
#include <string.h>

static int
needed (const char *fmt, va_list argp)
{
  static FILE *sink = NULL;

  /* ok, there's a small race here that could result in the sink being
   * opened more than once if we're threaded, but I'd rather ignore it than
   * spend cycles synchronizing :-) */

  if (sink == NULL)
    {
      if ((sink = fopen("/dev/null", "w")) == NULL)
	{
	  perror("/dev/null");
	  exit(1);
	}
    }

  return vfprintf(sink, fmt, argp);
}

int
vsnprintf (char *buf, size_t max, const char *fmt, va_list argp)
{
  char *p;
  int size;

  if ((p = malloc(needed(fmt, argp) + 1)) == NULL)
    {
      fprintf(stderr, "vsnprintf: malloc failed, aborting\n");
      abort();
    }

  if ((size = vsprintf(p, fmt, argp)) >= max)
    size = -1;

  strncpy(buf, p, max);
  buf[max - 1] = 0;
  free(p);
  return size;
}

int
snprintf (char *buf, size_t max, const char *fmt, ...)
{
  va_list argp;
  int bytes;

  va_start(argp, fmt);
  bytes = vsnprintf(buf, max, fmt, argp);
  va_end(argp);

  return bytes;
}