mirror of
https://github.com/postgres/postgres.git
synced 2025-04-24 10:47:04 +03:00
Try this snprintf() implementation, used in sendmail...
This commit is contained in:
Marc G. Fournier
parent
370d6cdb64
commit
edd626cb3e
@ -1,10 +1,8 @@
|
|||||||
/*-
|
/*
|
||||||
* Copyright (c) 1990, 1993
|
* Copyright (c) 1983, 1995, 1996 Eric P. Allman
|
||||||
|
* Copyright (c) 1988, 1993
|
||||||
* The Regents of the University of California. All rights reserved.
|
* The Regents of the University of California. All rights reserved.
|
||||||
*
|
*
|
||||||
* This code is derived from software contributed to Berkeley by
|
|
||||||
* Chris Torek.
|
|
||||||
*
|
|
||||||
* Redistribution and use in source and binary forms, with or without
|
* Redistribution and use in source and binary forms, with or without
|
||||||
* modification, are permitted provided that the following conditions
|
* modification, are permitted provided that the following conditions
|
||||||
* are met:
|
* are met:
|
||||||
@ -34,59 +32,312 @@
|
|||||||
* SUCH DAMAGE.
|
* SUCH DAMAGE.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#if defined(LIBC_SCCS) && !defined(lint)
|
# include "sendmail.h"
|
||||||
#if 0
|
# include "pathnames.h"
|
||||||
static char sccsid[] = "@(#)snprintf.c 8.1 (Berkeley) 6/4/93";
|
# include <sys/ioctl.h>
|
||||||
|
# include <sys/param.h>
|
||||||
|
|
||||||
#endif
|
/*
|
||||||
static const char rcsid[] =
|
** SNPRINTF, VSNPRINT -- counted versions of printf
|
||||||
"$Id: snprintf.c,v 1.3 1998/09/01 04:30:44 momjian Exp $";
|
**
|
||||||
|
** These versions have been grabbed off the net. They have been
|
||||||
|
** cleaned up to compile properly and support for .precision and
|
||||||
|
** %lx has been added.
|
||||||
|
*/
|
||||||
|
|
||||||
#endif /* LIBC_SCCS and not lint */
|
/**************************************************************
|
||||||
|
* Original:
|
||||||
|
* Patrick Powell Tue Apr 11 09:48:21 PDT 1995
|
||||||
|
* A bombproof version of doprnt (dopr) included.
|
||||||
|
* Sigh. This sort of thing is always nasty do deal with. Note that
|
||||||
|
* the version here does not include floating point...
|
||||||
|
*
|
||||||
|
* snprintf() is used instead of sprintf() as it does limit checks
|
||||||
|
* for string length. This covers a nasty loophole.
|
||||||
|
*
|
||||||
|
* The other functions are there to prevent NULL pointers from
|
||||||
|
* causing nast effects.
|
||||||
|
**************************************************************/
|
||||||
|
|
||||||
#include <limits.h>
|
/*static char _id[] = "$Id: snprintf.c,v 1.4 1998/09/04 14:34:23 scrappy Exp $";*/
|
||||||
#include <stdio.h>
|
static void dopr();
|
||||||
#if __STDC__
|
static char *end;
|
||||||
#include <stdarg.h>
|
static int SnprfOverflow;
|
||||||
#else
|
|
||||||
#include <varargs.h>
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#if __STDC__
|
/* VARARGS3 */
|
||||||
int
|
int
|
||||||
snprintf(char *str, size_t n, char const * fmt,...)
|
# ifdef __STDC__
|
||||||
#else
|
snprintf(char *str, size_t count, const char *fmt, ...)
|
||||||
int
|
# else
|
||||||
snprintf(str, n, fmt, va_alist)
|
snprintf(str, count, fmt, va_alist)
|
||||||
char *str;
|
char *str;
|
||||||
size_t n;
|
size_t count;
|
||||||
char *fmt;
|
const char *fmt;
|
||||||
|
va_dcl
|
||||||
va_dcl
|
|
||||||
#endif
|
#endif
|
||||||
{
|
{
|
||||||
size_t on;
|
int len;
|
||||||
int ret;
|
VA_LOCAL_DECL
|
||||||
va_list ap;
|
|
||||||
FILE f;
|
|
||||||
|
|
||||||
on = n;
|
VA_START(fmt);
|
||||||
if (n != 0)
|
len = vsnprintf(str, count, fmt, ap);
|
||||||
n--;
|
VA_END;
|
||||||
if (n > INT_MAX)
|
return len;
|
||||||
n = INT_MAX;
|
|
||||||
#if __STDC__
|
|
||||||
va_start(ap, fmt);
|
|
||||||
#else
|
|
||||||
va_start(ap);
|
|
||||||
#endif
|
|
||||||
f._file = -1;
|
|
||||||
f._flags = __SWR | __SSTR;
|
|
||||||
f._bf._base = f._p = (unsigned char *) str;
|
|
||||||
f._bf._size = f._w = n;
|
|
||||||
ret = vfprintf(&f, fmt, ap);
|
|
||||||
if (on > 0)
|
|
||||||
*f._p = '\0';
|
|
||||||
va_end(ap);
|
|
||||||
return ret;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
# ifndef luna2
|
||||||
|
int
|
||||||
|
vsnprintf(str, count, fmt, args)
|
||||||
|
char *str;
|
||||||
|
size_t count;
|
||||||
|
const char *fmt;
|
||||||
|
va_list args;
|
||||||
|
{
|
||||||
|
str[0] = 0;
|
||||||
|
end = str + count - 1;
|
||||||
|
SnprfOverflow = 0;
|
||||||
|
dopr( str, fmt, args );
|
||||||
|
if (count > 0)
|
||||||
|
end[0] = 0;
|
||||||
|
if (SnprfOverflow && tTd(57, 2))
|
||||||
|
printf("\nvsnprintf overflow, len = %d, str = %s",
|
||||||
|
count, shortenstring(str, 203));
|
||||||
|
return strlen(str);
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* dopr(): poor man's version of doprintf
|
||||||
|
*/
|
||||||
|
|
||||||
|
static void fmtstr __P((char *value, int ljust, int len, int zpad, int maxwidth));
|
||||||
|
static void fmtnum __P((long value, int base, int dosign, int ljust, int len, int zpad));
|
||||||
|
static void dostr __P(( char * , int ));
|
||||||
|
static char *output;
|
||||||
|
static void dopr_outch __P(( int c ));
|
||||||
|
|
||||||
|
static void
|
||||||
|
dopr( buffer, format, args )
|
||||||
|
char *buffer;
|
||||||
|
const char *format;
|
||||||
|
va_list args;
|
||||||
|
{
|
||||||
|
int ch;
|
||||||
|
long value;
|
||||||
|
int longflag = 0;
|
||||||
|
int pointflag = 0;
|
||||||
|
int maxwidth = 0;
|
||||||
|
char *strvalue;
|
||||||
|
int ljust;
|
||||||
|
int len;
|
||||||
|
int zpad;
|
||||||
|
|
||||||
|
output = buffer;
|
||||||
|
while( (ch = *format++) ){
|
||||||
|
switch( ch ){
|
||||||
|
case '%':
|
||||||
|
ljust = len = zpad = maxwidth = 0;
|
||||||
|
longflag = pointflag = 0;
|
||||||
|
nextch:
|
||||||
|
ch = *format++;
|
||||||
|
switch( ch ){
|
||||||
|
case 0:
|
||||||
|
dostr( "**end of format**" , 0);
|
||||||
|
return;
|
||||||
|
case '-': ljust = 1; goto nextch;
|
||||||
|
case '0': /* set zero padding if len not set */
|
||||||
|
if(len==0 && !pointflag) zpad = '0';
|
||||||
|
case '1': case '2': case '3':
|
||||||
|
case '4': case '5': case '6':
|
||||||
|
case '7': case '8': case '9':
|
||||||
|
if (pointflag)
|
||||||
|
maxwidth = maxwidth*10 + ch - '0';
|
||||||
|
else
|
||||||
|
len = len*10 + ch - '0';
|
||||||
|
goto nextch;
|
||||||
|
case '*':
|
||||||
|
if (pointflag)
|
||||||
|
maxwidth = va_arg( args, int );
|
||||||
|
else
|
||||||
|
len = va_arg( args, int );
|
||||||
|
goto nextch;
|
||||||
|
case '.': pointflag = 1; goto nextch;
|
||||||
|
case 'l': longflag = 1; goto nextch;
|
||||||
|
case 'u': case 'U':
|
||||||
|
/*fmtnum(value,base,dosign,ljust,len,zpad) */
|
||||||
|
if( longflag ){
|
||||||
|
value = va_arg( args, long );
|
||||||
|
} else {
|
||||||
|
value = va_arg( args, int );
|
||||||
|
}
|
||||||
|
fmtnum( value, 10,0, ljust, len, zpad ); break;
|
||||||
|
case 'o': case 'O':
|
||||||
|
/*fmtnum(value,base,dosign,ljust,len,zpad) */
|
||||||
|
if( longflag ){
|
||||||
|
value = va_arg( args, long );
|
||||||
|
} else {
|
||||||
|
value = va_arg( args, int );
|
||||||
|
}
|
||||||
|
fmtnum( value, 8,0, ljust, len, zpad ); break;
|
||||||
|
case 'd': case 'D':
|
||||||
|
if( longflag ){
|
||||||
|
value = va_arg( args, long );
|
||||||
|
} else {
|
||||||
|
value = va_arg( args, int );
|
||||||
|
}
|
||||||
|
fmtnum( value, 10,1, ljust, len, zpad ); break;
|
||||||
|
case 'x':
|
||||||
|
if( longflag ){
|
||||||
|
value = va_arg( args, long );
|
||||||
|
} else {
|
||||||
|
value = va_arg( args, int );
|
||||||
|
}
|
||||||
|
fmtnum( value, 16,0, ljust, len, zpad ); break;
|
||||||
|
case 'X':
|
||||||
|
if( longflag ){
|
||||||
|
value = va_arg( args, long );
|
||||||
|
} else {
|
||||||
|
value = va_arg( args, int );
|
||||||
|
}
|
||||||
|
fmtnum( value,-16,0, ljust, len, zpad ); break;
|
||||||
|
case 's':
|
||||||
|
strvalue = va_arg( args, char *);
|
||||||
|
if (maxwidth > 0 || !pointflag) {
|
||||||
|
if (pointflag && len > maxwidth)
|
||||||
|
len = maxwidth; /* Adjust padding */
|
||||||
|
fmtstr( strvalue,ljust,len,zpad, maxwidth);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case 'c':
|
||||||
|
ch = va_arg( args, int );
|
||||||
|
dopr_outch( ch ); break;
|
||||||
|
case '%': dopr_outch( ch ); continue;
|
||||||
|
default:
|
||||||
|
dostr( "???????" , 0);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
dopr_outch( ch );
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
*output = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
fmtstr( value, ljust, len, zpad, maxwidth )
|
||||||
|
char *value;
|
||||||
|
int ljust, len, zpad, maxwidth;
|
||||||
|
{
|
||||||
|
int padlen, strlen; /* amount to pad */
|
||||||
|
|
||||||
|
if( value == 0 ){
|
||||||
|
value = "<NULL>";
|
||||||
|
}
|
||||||
|
for( strlen = 0; value[strlen]; ++ strlen ); /* strlen */
|
||||||
|
if (strlen > maxwidth && maxwidth)
|
||||||
|
strlen = maxwidth;
|
||||||
|
padlen = len - strlen;
|
||||||
|
if( padlen < 0 ) padlen = 0;
|
||||||
|
if( ljust ) padlen = -padlen;
|
||||||
|
while( padlen > 0 ) {
|
||||||
|
dopr_outch( ' ' );
|
||||||
|
--padlen;
|
||||||
|
}
|
||||||
|
dostr( value, maxwidth );
|
||||||
|
while( padlen < 0 ) {
|
||||||
|
dopr_outch( ' ' );
|
||||||
|
++padlen;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
fmtnum( value, base, dosign, ljust, len, zpad )
|
||||||
|
long value;
|
||||||
|
int base, dosign, ljust, len, zpad;
|
||||||
|
{
|
||||||
|
int signvalue = 0;
|
||||||
|
unsigned long uvalue;
|
||||||
|
char convert[20];
|
||||||
|
int place = 0;
|
||||||
|
int padlen = 0; /* amount to pad */
|
||||||
|
int caps = 0;
|
||||||
|
|
||||||
|
/* DEBUGP(("value 0x%x, base %d, dosign %d, ljust %d, len %d, zpad %d\n",
|
||||||
|
value, base, dosign, ljust, len, zpad )); */
|
||||||
|
uvalue = value;
|
||||||
|
if( dosign ){
|
||||||
|
if( value < 0 ) {
|
||||||
|
signvalue = '-';
|
||||||
|
uvalue = -value;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if( base < 0 ){
|
||||||
|
caps = 1;
|
||||||
|
base = -base;
|
||||||
|
}
|
||||||
|
do{
|
||||||
|
convert[place++] =
|
||||||
|
(caps? "0123456789ABCDEF":"0123456789abcdef")
|
||||||
|
[uvalue % (unsigned)base ];
|
||||||
|
uvalue = (uvalue / (unsigned)base );
|
||||||
|
}while(uvalue);
|
||||||
|
convert[place] = 0;
|
||||||
|
padlen = len - place;
|
||||||
|
if( padlen < 0 ) padlen = 0;
|
||||||
|
if( ljust ) padlen = -padlen;
|
||||||
|
/* DEBUGP(( "str '%s', place %d, sign %c, padlen %d\n",
|
||||||
|
convert,place,signvalue,padlen)); */
|
||||||
|
if( zpad && padlen > 0 ){
|
||||||
|
if( signvalue ){
|
||||||
|
dopr_outch( signvalue );
|
||||||
|
--padlen;
|
||||||
|
signvalue = 0;
|
||||||
|
}
|
||||||
|
while( padlen > 0 ){
|
||||||
|
dopr_outch( zpad );
|
||||||
|
--padlen;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
while( padlen > 0 ) {
|
||||||
|
dopr_outch( ' ' );
|
||||||
|
--padlen;
|
||||||
|
}
|
||||||
|
if( signvalue ) dopr_outch( signvalue );
|
||||||
|
while( place > 0 ) dopr_outch( convert[--place] );
|
||||||
|
while( padlen < 0 ){
|
||||||
|
dopr_outch( ' ' );
|
||||||
|
++padlen;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
dostr( str , cut)
|
||||||
|
char *str;
|
||||||
|
int cut;
|
||||||
|
{
|
||||||
|
if (cut) {
|
||||||
|
while(*str && cut-- > 0) dopr_outch(*str++);
|
||||||
|
} else {
|
||||||
|
while(*str) dopr_outch(*str++);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
dopr_outch( c )
|
||||||
|
int c;
|
||||||
|
{
|
||||||
|
#if 0
|
||||||
|
if( iscntrl(c) && c != '\n' && c != '\t' ){
|
||||||
|
c = '@' + (c & 0x1F);
|
||||||
|
if( end == 0 || output < end )
|
||||||
|
*output++ = '^';
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
if( end == 0 || output < end )
|
||||||
|
*output++ = c;
|
||||||
|
else
|
||||||
|
SnprfOverflow++;
|
||||||
|
}
|
||||||
|
|
||||||
|
# endif /* !luna2 */
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user