123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560 |
- /*
- * String functions for Mini-XML, a small XML file parsing library.
- *
- * https://www.msweet.org/mxml
- *
- * Copyright © 2003-2019 by Michael R Sweet.
- *
- * Licensed under Apache License v2.0. See the file "LICENSE" for more
- * information.
- */
- /*
- * Include necessary headers...
- */
- #include "config.h"
- /*
- * The va_copy macro is part of C99, but many compilers don't implement it.
- * Provide a "direct assignment" implmentation when va_copy isn't defined...
- */
- #ifndef va_copy
- # ifdef __va_copy
- # define va_copy(dst,src) __va_copy(dst,src)
- # else
- # define va_copy(dst,src) memcpy(&dst, &src, sizeof(va_list))
- # endif /* __va_copy */
- #endif /* va_copy */
- #ifndef HAVE_SNPRINTF
- /*
- * '_mxml_snprintf()' - Format a string.
- */
- int /* O - Number of bytes formatted */
- _mxml_snprintf(char *buffer, /* I - Output buffer */
- size_t bufsize, /* I - Size of output buffer */
- const char *format, /* I - Printf-style format string */
- ...) /* I - Additional arguments as needed */
- {
- va_list ap; /* Argument list */
- int bytes; /* Number of bytes formatted */
- va_start(ap, format);
- bytes = vsnprintf(buffer, bufsize, format, ap);
- va_end(ap);
- return (bytes);
- }
- #endif /* !HAVE_SNPRINTF */
- /*
- * '_mxml_strdup()' - Duplicate a string.
- */
- #ifndef HAVE_STRDUP
- char * /* O - New string pointer */
- _mxml_strdup(const char *s) /* I - String to duplicate */
- {
- char *t; /* New string pointer */
- if (s == NULL)
- return (NULL);
- if ((t = malloc(strlen(s) + 1)) == NULL)
- return (NULL);
- return (strcpy(t, s));
- }
- #endif /* !HAVE_STRDUP */
- /*
- * '_mxml_strdupf()' - Format and duplicate a string.
- */
- char * /* O - New string pointer */
- _mxml_strdupf(const char *format, /* I - Printf-style format string */
- ...) /* I - Additional arguments as needed */
- {
- va_list ap; /* Pointer to additional arguments */
- char *s; /* Pointer to formatted string */
- /*
- * Get a pointer to the additional arguments, format the string,
- * and return it...
- */
- va_start(ap, format);
- #ifdef HAVE_VASPRINTF
- if (vasprintf(&s, format, ap) < 0)
- s = NULL;
- #else
- s = _mxml_vstrdupf(format, ap);
- #endif /* HAVE_VASPRINTF */
- va_end(ap);
- return (s);
- }
- #ifndef HAVE_STRLCAT
- /*
- * '_mxml_strlcat()' - Safely concatenate a string.
- */
- size_t /* O - Number of bytes copied */
- _mxml_strlcat(char *dst, /* I - Destination buffer */
- const char *src, /* I - Source string */
- size_t dstsize) /* I - Size of destinatipon buffer */
- {
- size_t srclen; /* Length of source string */
- size_t dstlen; /* Length of destination string */
- /*
- * Figure out how much room is left...
- */
- dstlen = strlen(dst);
- if (dstsize <= (dstlen + 1))
- return (dstlen); /* No room, return immediately... */
- dstsize -= dstlen + 1;
- /*
- * Figure out how much room is needed...
- */
- srclen = strlen(src);
- /*
- * Copy the appropriate amount...
- */
- if (srclen > dstsize)
- srclen = dstsize;
- memmove(dst + dstlen, src, srclen);
- dst[dstlen + srclen] = '\0';
- return (dstlen + srclen);
- }
- #endif /* !HAVE_STRLCAT */
- #ifndef HAVE_STRLCPY
- /*
- * '_mxml_strlcpy()' - Safely copy a string.
- */
- size_t /* O - Number of bytes copied */
- _mxml_strlcpy(char *dst, /* I - Destination buffer */
- const char *src, /* I - Source string */
- size_t dstsize) /* I - Size of destinatipon buffer */
- {
- size_t srclen; /* Length of source string */
- /*
- * Figure out how much room is needed...
- */
- dstsize --;
- srclen = strlen(src);
- /*
- * Copy the appropriate amount...
- */
- if (srclen > dstsize)
- srclen = dstsize;
- memmove(dst, src, srclen);
- dst[srclen] = '\0';
- return (srclen);
- }
- #endif /* !HAVE_STRLCPY */
- #ifndef HAVE_VSNPRINTF
- /*
- * '_mxml_vsnprintf()' - Format a string into a fixed size buffer.
- */
- int /* O - Number of bytes formatted */
- _mxml_vsnprintf(char *buffer, /* O - Output buffer */
- size_t bufsize, /* O - Size of output buffer */
- const char *format, /* I - Printf-style format string */
- va_list ap) /* I - Pointer to additional arguments */
- {
- char *bufptr, /* Pointer to position in buffer */
- *bufend, /* Pointer to end of buffer */
- sign, /* Sign of format width */
- size, /* Size character (h, l, L) */
- type; /* Format type character */
- int width, /* Width of field */
- prec; /* Number of characters of precision */
- char tformat[100], /* Temporary format string for sprintf() */
- *tptr, /* Pointer into temporary format */
- temp[1024]; /* Buffer for formatted numbers */
- char *s; /* Pointer to string */
- int slen; /* Length of string */
- int bytes; /* Total number of bytes needed */
- /*
- * Loop through the format string, formatting as needed...
- */
- bufptr = buffer;
- bufend = buffer + bufsize - 1;
- bytes = 0;
- while (*format)
- {
- if (*format == '%')
- {
- tptr = tformat;
- *tptr++ = *format++;
- if (*format == '%')
- {
- if (bufptr && bufptr < bufend)
- *bufptr++ = *format;
- bytes ++;
- format ++;
- continue;
- }
- else if (strchr(" -+#\'", *format))
- {
- *tptr++ = *format;
- sign = *format++;
- }
- else
- sign = 0;
- if (*format == '*')
- {
- /*
- * Get width from argument...
- */
- format ++;
- width = va_arg(ap, int);
- snprintf(tptr, sizeof(tformat) - (tptr - tformat), "%d", width);
- tptr += strlen(tptr);
- }
- else
- {
- width = 0;
- while (isdigit(*format & 255))
- {
- if (tptr < (tformat + sizeof(tformat) - 1))
- *tptr++ = *format;
- width = width * 10 + *format++ - '0';
- }
- }
- if (*format == '.')
- {
- if (tptr < (tformat + sizeof(tformat) - 1))
- *tptr++ = *format;
- format ++;
- if (*format == '*')
- {
- /*
- * Get precision from argument...
- */
- format ++;
- prec = va_arg(ap, int);
- snprintf(tptr, sizeof(tformat) - (tptr - tformat), "%d", prec);
- tptr += strlen(tptr);
- }
- else
- {
- prec = 0;
- while (isdigit(*format & 255))
- {
- if (tptr < (tformat + sizeof(tformat) - 1))
- *tptr++ = *format;
- prec = prec * 10 + *format++ - '0';
- }
- }
- }
- else
- prec = -1;
- if (*format == 'l' && format[1] == 'l')
- {
- size = 'L';
- if (tptr < (tformat + sizeof(tformat) - 2))
- {
- *tptr++ = 'l';
- *tptr++ = 'l';
- }
- format += 2;
- }
- else if (*format == 'h' || *format == 'l' || *format == 'L')
- {
- if (tptr < (tformat + sizeof(tformat) - 1))
- *tptr++ = *format;
- size = *format++;
- }
- if (!*format)
- break;
- if (tptr < (tformat + sizeof(tformat) - 1))
- *tptr++ = *format;
- type = *format++;
- *tptr = '\0';
- switch (type)
- {
- case 'E' : /* Floating point formats */
- case 'G' :
- case 'e' :
- case 'f' :
- case 'g' :
- if ((width + 2) > sizeof(temp))
- break;
- sprintf(temp, tformat, va_arg(ap, double));
- bytes += strlen(temp);
- if (bufptr)
- {
- if ((bufptr + strlen(temp)) > bufend)
- {
- strncpy(bufptr, temp, (size_t)(bufend - bufptr));
- bufptr = bufend;
- }
- else
- {
- strcpy(bufptr, temp);
- bufptr += strlen(temp);
- }
- }
- break;
- case 'B' : /* Integer formats */
- case 'X' :
- case 'b' :
- case 'd' :
- case 'i' :
- case 'o' :
- case 'u' :
- case 'x' :
- if ((width + 2) > sizeof(temp))
- break;
- #ifdef HAVE_LONG_LONG
- if (size == 'L')
- sprintf(temp, tformat, va_arg(ap, long long));
- else
- #endif /* HAVE_LONG_LONG */
- sprintf(temp, tformat, va_arg(ap, int));
- bytes += strlen(temp);
- if (bufptr)
- {
- if ((bufptr + strlen(temp)) > bufend)
- {
- strncpy(bufptr, temp, (size_t)(bufend - bufptr));
- bufptr = bufend;
- }
- else
- {
- strcpy(bufptr, temp);
- bufptr += strlen(temp);
- }
- }
- break;
- case 'p' : /* Pointer value */
- if ((width + 2) > sizeof(temp))
- break;
- sprintf(temp, tformat, va_arg(ap, void *));
- bytes += strlen(temp);
- if (bufptr)
- {
- if ((bufptr + strlen(temp)) > bufend)
- {
- strncpy(bufptr, temp, (size_t)(bufend - bufptr));
- bufptr = bufend;
- }
- else
- {
- strcpy(bufptr, temp);
- bufptr += strlen(temp);
- }
- }
- break;
- case 'c' : /* Character or character array */
- bytes += width;
- if (bufptr)
- {
- if (width <= 1)
- *bufptr++ = va_arg(ap, int);
- else
- {
- if ((bufptr + width) > bufend)
- width = bufend - bufptr;
- memcpy(bufptr, va_arg(ap, char *), (size_t)width);
- bufptr += width;
- }
- }
- break;
- case 's' : /* String */
- if ((s = va_arg(ap, char *)) == NULL)
- s = "(null)";
- slen = strlen(s);
- if (slen > width && prec != width)
- width = slen;
- bytes += width;
- if (bufptr)
- {
- if ((bufptr + width) > bufend)
- width = bufend - bufptr;
- if (slen > width)
- slen = width;
- if (sign == '-')
- {
- strncpy(bufptr, s, (size_t)slen);
- memset(bufptr + slen, ' ', (size_t)(width - slen));
- }
- else
- {
- memset(bufptr, ' ', (size_t)(width - slen));
- strncpy(bufptr + width - slen, s, (size_t)slen);
- }
- bufptr += width;
- }
- break;
- case 'n' : /* Output number of chars so far */
- *(va_arg(ap, int *)) = bytes;
- break;
- }
- }
- else
- {
- bytes ++;
- if (bufptr && bufptr < bufend)
- *bufptr++ = *format;
- format ++;
- }
- }
- /*
- * Nul-terminate the string and return the number of characters needed.
- */
- *bufptr = '\0';
- return (bytes);
- }
- #endif /* !HAVE_VSNPRINTF */
- /*
- * '_mxml_vstrdupf()' - Format and duplicate a string.
- */
- char * /* O - New string pointer */
- _mxml_vstrdupf(const char *format, /* I - Printf-style format string */
- va_list ap) /* I - Pointer to additional arguments */
- {
- #ifdef HAVE_VASPRINTF
- char *s; /* String */
- if (vasprintf(&s, format, ap) < 0)
- s = NULL;
- return (s);
- #else
- int bytes; /* Number of bytes required */
- char *buffer, /* String buffer */
- temp[256]; /* Small buffer for first vsnprintf */
- /*
- * First format with a tiny buffer; this will tell us how many bytes are
- * needed...
- */
- # ifdef _WIN32
- bytes = _vscprintf(format, ap);
- # else
- va_list apcopy; /* Copy of argument list */
- va_copy(apcopy, ap);
- if ((bytes = vsnprintf(temp, sizeof(temp), format, apcopy)) < sizeof(temp))
- {
- /*
- * Hey, the formatted string fits in the tiny buffer, so just dup that...
- */
- return (strdup(temp));
- }
- # endif /* _WIN32 */
- /*
- * Allocate memory for the whole thing and reformat to the new buffer...
- */
- if ((buffer = calloc(1, bytes + 1)) != NULL)
- vsnprintf(buffer, bytes + 1, format, ap);
- /*
- * Return the new string...
- */
- return (buffer);
- #endif /* HAVE_VASPRINTF */
- }
|