lkml.org 
[lkml]   [2009]   [Feb]   [28]   [last100]   RSS Feed
Views: [wrap][no wrap]   [headers]  [forward] 
 
Messages in this thread
Patch in this message
/
Date
From
SubjectRe: [PATCH][RFC] vsprintf: unify the format decoding layer for its 3 users
On Fri, Feb 27, 2009 at 04:39:37PM -0800, Linus Torvalds wrote:
>
>
> On Fri, 27 Feb 2009, Frederic Weisbecker wrote:
> >
> > Ok, here is a try, I've mostly unified the format decoding bits.
> > Most interesting other helpers were already implemented (pointer(),
> > string() and number()
>
> Ok, I don't think this is beautiful, but neither was the original code,
> so..
>
> However, in order to make it at least a _bit_ more palatable, could we
> please agree to at least clean up this:
>
> > +static int format_decode(const char *fmt, enum format_type *type, int *flags,
> > + int *field_width, int *base, int *precision,
> > + int *qualifier)
>
> To
>
> struct printf_spec {
> enum format_type type;
> int flags, field_width, base, precision, qualifier;
> };
>
> and then pass things around like
>
> static int format_decode(const char *fmt, struct printf_spec *spec)
> {
> ..
>
> instead? Wouldn't that be nicer? I suspect it would make the code look
> nicer too (instead of doing "*base = x", you'd see "spec->base = x" and it
> would look less like line noise in the callee, an the caller could just
> do a single "struct format_spec spec = { 0, }" to initialize that thing).
>
> Linus

You're right, that's much proper.
See the V2 below:
---

From 601f73ac457f720f2201ba88a9c741c9a3690119 Mon Sep 17 00:00:00 2001
From: Frederic Weisbecker <fweisbec@gmail.com>
Date: Sat, 28 Feb 2009 08:56:18 +0100
Subject: [PATCH v2] vsprintf: unify the format decoding layer for its 3 users

An new optimization is making its way to ftrace. Its purpose is to
make ftrace_printk() consuming less memory and be faster.

Written by Lai Jiangshan, the approach is to delay the formatting
job from tracing time to output time.
Currently, a call to ftrace_printk will format the whole string and
insert it into the ring buffer. Then you can read it on /debug/tracing/trace
file.

The new implementation stores the address of the format string and
the binary parameters into the ring buffer, making the packet more compact
and faster to insert.
Later, when the user exports the traces, the format string is retrieved
with the binary parameters and the formatting job is eventually done.

Here is the result of a small comparative benchmark while putting the following
ftrace_printk on the timer interrupt. ftrace_printk is the old implementation,
ftrace_bprintk is a the new one:

ftrace_printk("This is the timer interrupt: %llu", jiffies_64);

After some time running on low load (no X, no really active processes):

ftrace_printk: duration average: 2044 ns, avg of bytes stored per entry: 39
ftrace_bprintk: duration average: 1426 ns, avg of bytes stored per entry: 16

Higher load (started X and launched a cat running on an X console looping on
traces printing):

ftrace_printk: duration average: 8812 ns
ftrace_bprintk: duration average: 2611 ns

Which means the new implementation can be 70 % faster on higher load.
And it consumes lesser memory on the ring buffer.

The curent implementation rewrites a lot of format decoding bits from
vsnprintf() function, making now 3 differents functions to maintain
in their duplicated parts of printf format decoding bits.

Suggested by Ingo Molnar, this patch tries to factorize the most
possible common bits from these functions.
The real common part between them is the format decoding. Although
they do somewhat similar jobs, their way to export or import the parameters
is very different. Thus, only the decoding layer is extracted, unless you see
other parts that could be worth factorized.


Changes in V2:

- Address a suggestion from Linus to group the format_decode() parameters inside
a structure.

Signed-off-by: Frederic Weisbecker <fweisbec@gmail.com>
---
lib/vsprintf.c | 828 +++++++++++++++++++++++++++++++-------------------------
1 files changed, 466 insertions(+), 362 deletions(-)

diff --git a/lib/vsprintf.c b/lib/vsprintf.c
index 3543bbe..d03167e 100644
--- a/lib/vsprintf.c
+++ b/lib/vsprintf.c
@@ -700,6 +700,242 @@ static char *pointer(const char *fmt, char *buf, char *end, void *ptr, int field
return number(buf, end, (unsigned long) ptr, 16, field_width, precision, flags);
}

+
+enum format_type {
+ FORMAT_TYPE_NONE, /* Just a string part */
+ FORMAT_TYPE_WITDH,
+ FORMAT_TYPE_PRECISION,
+ FORMAT_TYPE_CHAR,
+ FORMAT_TYPE_STR,
+ FORMAT_TYPE_PTR,
+ FORMAT_TYPE_PERCENT_CHAR,
+ FORMAT_TYPE_INVALID,
+ FORMAT_TYPE_LONG_LONG,
+ FORMAT_TYPE_ULONG,
+ FORMAT_TYPE_LONG,
+ FORMAT_TYPE_USHORT,
+ FORMAT_TYPE_SHORT,
+ FORMAT_TYPE_UINT,
+ FORMAT_TYPE_INT,
+ FORMAT_TYPE_NRCHARS,
+ FORMAT_TYPE_SIZE_T,
+ FORMAT_TYPE_PTRDIFF
+};
+
+struct printf_spec {
+ enum format_type type;
+ int flags; /* flags to number() */
+ int field_width; /* width of output field */
+ int base;
+ /* min. # of digits for integers; max number of chars for from string */
+ int precision;
+ int qualifier;
+};
+
+/*
+ * Helper function to decode printf style format.
+ * Each call decode a token from the format and return the
+ * number of characters read (or likely the delta where it wants
+ * to go on the next call).
+ * The decoded token is returned through the parameters
+ *
+ * 'h', 'l', or 'L' for integer fields
+ * 'z' support added 23/7/1999 S.H.
+ * 'z' changed to 'Z' --davidm 1/25/99
+ * 't' added for ptrdiff_t
+ *
+ * @fmt: the format string
+ * @type of the token returned
+ * @flags: various flags such as +, -, # tokens..
+ * @field_width: overwritten width
+ * @base: base of the number (octal, hex, ...)
+ * @precision: precision of a number
+ * @qualifier: qualifier of a number (long, size_t, ...)
+ */
+static int format_decode(const char *fmt, struct printf_spec *spec)
+{
+ const char *start = fmt;
+ bool sign = false;
+
+ /* we finished early by reading the field width */
+ if (spec->type == FORMAT_TYPE_WITDH) {
+ if (spec->field_width < 0) {
+ spec->field_width = -spec->field_width;
+ spec->flags |= LEFT;
+ }
+ spec->type = FORMAT_TYPE_NONE;
+ goto precision;
+ }
+
+ /* we finished early by reading the precision */
+ if (spec->type == FORMAT_TYPE_PRECISION) {
+ if (spec->precision < 0)
+ spec->precision = 0;
+
+ spec->type = FORMAT_TYPE_NONE;
+ goto qualifier;
+ }
+
+ /* By default */
+ spec->type = FORMAT_TYPE_NONE;
+
+ for (; *fmt ; ++fmt) {
+ if (*fmt == '%')
+ break;
+ }
+
+ /* Return the current non-format string */
+ if (fmt != start || !*fmt)
+ return fmt - start;
+
+ /* Process flags */
+ spec->flags = 0;
+
+ while (1) { /* this also skips first '%' */
+ bool found = true;
+
+ ++fmt;
+
+ switch (*fmt) {
+ case '-':
+ spec->flags |= LEFT;
+ break;
+ case '+':
+ spec->flags |= PLUS;
+ break;
+ case ' ':
+ spec->flags |= SPACE;
+ break;
+ case '#':
+ spec->flags |= SPECIAL;
+ break;
+ case '0':
+ spec->flags |= ZEROPAD;
+ break;
+ default:
+ found = false;
+ }
+
+ if (!found)
+ break;
+ }
+
+ /* get field width */
+ spec->field_width = -1;
+
+ if (isdigit(*fmt))
+ spec->field_width = skip_atoi(&fmt);
+ else if (*fmt == '*') {
+ /* it's the next argument */
+ spec->type = FORMAT_TYPE_WITDH;
+ return ++fmt - start;
+ }
+
+precision:
+ /* get the precision */
+ spec->precision = -1;
+ if (*fmt == '.') {
+ ++fmt;
+ if (isdigit(*fmt)) {
+ spec->precision = skip_atoi(&fmt);
+ if (spec->precision < 0)
+ spec->precision = 0;
+ } else if (*fmt == '*') {
+ /* it's the next argument */
+ spec->type = FORMAT_TYPE_WITDH;
+ return ++fmt - start;
+ }
+ }
+
+qualifier:
+ /* get the conversion qualifier */
+ spec->qualifier = -1;
+ if (*fmt == 'h' || *fmt == 'l' || *fmt == 'L' ||
+ *fmt == 'Z' || *fmt == 'z' || *fmt == 't') {
+ spec->qualifier = *fmt;
+ ++fmt;
+ if (spec->qualifier == 'l' && *fmt == 'l') {
+ spec->qualifier = 'L';
+ ++fmt;
+ }
+ }
+
+ /* default base */
+ spec->base = 10;
+ switch (*fmt) {
+ case 'c':
+ spec->type = FORMAT_TYPE_CHAR;
+ return ++fmt - start;
+
+ case 's':
+ spec->type = FORMAT_TYPE_STR;
+ return ++fmt - start;
+
+ case 'p':
+ spec->type = FORMAT_TYPE_PTR;
+ return fmt - start;
+ /* skip alnum */
+
+ case 'n':
+ /* FIXME:
+ * What does C99 say about the overflow case here? */
+ spec->type = FORMAT_TYPE_NRCHARS;
+ return ++fmt - start;
+
+ case '%':
+ spec->type = FORMAT_TYPE_PERCENT_CHAR;
+ return ++fmt - start;
+
+ /* integer number formats - set up the flags and "break" */
+ case 'o':
+ spec->base = 8;
+ break;
+
+ case 'x':
+ spec->flags |= SMALL;
+
+ case 'X':
+ spec->base = 16;
+ break;
+
+ case 'd':
+ case 'i':
+ sign = true;
+ case 'u':
+ break;
+
+ default:
+ spec->type = FORMAT_TYPE_INVALID;
+ return fmt - start;
+ }
+
+ if (spec->qualifier == 'L')
+ spec->type = FORMAT_TYPE_LONG_LONG;
+ else if (spec->qualifier == 'l') {
+ if (sign)
+ spec->type = FORMAT_TYPE_LONG;
+ else
+ spec->type = FORMAT_TYPE_ULONG;
+ } else if (spec->qualifier == 'Z' || spec->qualifier == 'z') {
+ spec->type = FORMAT_TYPE_SIZE_T;
+ } else if (spec->qualifier == 't') {
+ spec->type = FORMAT_TYPE_PTRDIFF;
+ } else if (spec->qualifier == 'h') {
+ if (sign)
+ spec->type = FORMAT_TYPE_SHORT;
+ else
+ spec->type = FORMAT_TYPE_USHORT;
+ } else {
+ if (sign)
+ spec->type = FORMAT_TYPE_INT;
+ else
+ spec->type = FORMAT_TYPE_UINT;
+ }
+
+ return ++fmt - start;
+}
+
+
/**
* vsnprintf - Format a string and place it in a buffer
* @buf: The buffer to place the result into
@@ -726,18 +962,9 @@ static char *pointer(const char *fmt, char *buf, char *end, void *ptr, int field
int vsnprintf(char *buf, size_t size, const char *fmt, va_list args)
{
unsigned long long num;
- int base;
char *str, *end, c;
-
- int flags; /* flags to number() */
-
- int field_width; /* width of output field */
- int precision; /* min. # of digits for integers; max
- number of chars for from string */
- int qualifier; /* 'h', 'l', or 'L' for integer fields */
- /* 'z' support added 23/7/1999 S.H. */
- /* 'z' changed to 'Z' --davidm 1/25/99 */
- /* 't' added for ptrdiff_t */
+ int read;
+ struct printf_spec spec = {0};

/* Reject out-of-range values early. Large positive sizes are
used for unknown buffer sizes. */
@@ -758,184 +985,141 @@ int vsnprintf(char *buf, size_t size, const char *fmt, va_list args)
size = end - buf;
}

- for (; *fmt ; ++fmt) {
- if (*fmt != '%') {
- if (str < end)
- *str = *fmt;
- ++str;
- continue;
- }
+ while (*fmt) {
+ char *old_fmt = (char *)fmt;

- /* process flags */
- flags = 0;
- repeat:
- ++fmt; /* this also skips first '%' */
- switch (*fmt) {
- case '-': flags |= LEFT; goto repeat;
- case '+': flags |= PLUS; goto repeat;
- case ' ': flags |= SPACE; goto repeat;
- case '#': flags |= SPECIAL; goto repeat;
- case '0': flags |= ZEROPAD; goto repeat;
- }
+ read = format_decode(fmt, &spec);

- /* get field width */
- field_width = -1;
- if (isdigit(*fmt))
- field_width = skip_atoi(&fmt);
- else if (*fmt == '*') {
- ++fmt;
- /* it's the next argument */
- field_width = va_arg(args, int);
- if (field_width < 0) {
- field_width = -field_width;
- flags |= LEFT;
- }
- }
+ fmt += read;

- /* get the precision */
- precision = -1;
- if (*fmt == '.') {
- ++fmt;
- if (isdigit(*fmt))
- precision = skip_atoi(&fmt);
- else if (*fmt == '*') {
- ++fmt;
- /* it's the next argument */
- precision = va_arg(args, int);
+ switch (spec.type) {
+ case FORMAT_TYPE_NONE: {
+ int copy = read;
+ if (str < end) {
+ if (copy > end - str)
+ copy = end - str;
+ memcpy(str, old_fmt, copy);
}
- if (precision < 0)
- precision = 0;
+ str += read;
+ break;
}

- /* get the conversion qualifier */
- qualifier = -1;
- if (*fmt == 'h' || *fmt == 'l' || *fmt == 'L' ||
- *fmt =='Z' || *fmt == 'z' || *fmt == 't') {
- qualifier = *fmt;
- ++fmt;
- if (qualifier == 'l' && *fmt == 'l') {
- qualifier = 'L';
- ++fmt;
- }
- }
+ case FORMAT_TYPE_WITDH:
+ spec.field_width = va_arg(args, int);
+ break;

- /* default base */
- base = 10;
+ case FORMAT_TYPE_PRECISION:
+ spec.precision = va_arg(args, int);
+ break;

- switch (*fmt) {
- case 'c':
- if (!(flags & LEFT)) {
- while (--field_width > 0) {
- if (str < end)
- *str = ' ';
- ++str;
- }
- }
- c = (unsigned char) va_arg(args, int);
- if (str < end)
- *str = c;
- ++str;
- while (--field_width > 0) {
+ case FORMAT_TYPE_CHAR:
+ if (!(spec.flags & LEFT)) {
+ while (--spec.field_width > 0) {
if (str < end)
*str = ' ';
++str;
- }
- continue;
-
- case 's':
- str = string(str, end, va_arg(args, char *), field_width, precision, flags);
- continue;
-
- case 'p':
- str = pointer(fmt+1, str, end,
- va_arg(args, void *),
- field_width, precision, flags);
- /* Skip all alphanumeric pointer suffixes */
- while (isalnum(fmt[1]))
- fmt++;
- continue;
-
- case 'n':
- /* FIXME:
- * What does C99 say about the overflow case here? */
- if (qualifier == 'l') {
- long * ip = va_arg(args, long *);
- *ip = (str - buf);
- } else if (qualifier == 'Z' || qualifier == 'z') {
- size_t * ip = va_arg(args, size_t *);
- *ip = (str - buf);
- } else {
- int * ip = va_arg(args, int *);
- *ip = (str - buf);
- }
- continue;

- case '%':
+ }
+ }
+ c = (unsigned char) va_arg(args, int);
+ if (str < end)
+ *str = c;
+ ++str;
+ while (--spec.field_width > 0) {
if (str < end)
- *str = '%';
+ *str = ' ';
++str;
- continue;
+ }
+ break;

- /* integer number formats - set up the flags and "break" */
- case 'o':
- base = 8;
- break;
+ case FORMAT_TYPE_STR:
+ str = string(str, end, va_arg(args, char *),
+ spec.field_width, spec.precision,
+ spec.flags);
+ break;

- case 'x':
- flags |= SMALL;
- case 'X':
- base = 16;
- break;
+ case FORMAT_TYPE_PTR:
+ str = pointer(fmt+1, str, end, va_arg(args, void *),
+ spec.field_width, spec.precision,
+ spec.flags);
+ while (isalnum(fmt[1]))
+ fmt++;
+ break;

- case 'd':
- case 'i':
- flags |= SIGN;
- case 'u':
- break;
+ case FORMAT_TYPE_PERCENT_CHAR:
+ if (str < end)
+ *str = '%';
+ ++str;
+ break;

- default:
+ case FORMAT_TYPE_INVALID:
+ if (str < end)
+ *str = '%';
+ ++str;
+ if (*fmt) {
if (str < end)
- *str = '%';
+ *str = *fmt;
++str;
- if (*fmt) {
- if (str < end)
- *str = *fmt;
- ++str;
- } else {
- --fmt;
- }
- continue;
+ } else {
+ --fmt;
+ }
+ break;
+
+ case FORMAT_TYPE_NRCHARS: {
+ int qualifier = spec.qualifier;
+
+ if (qualifier == 'l') {
+ long *ip = va_arg(args, long *);
+ *ip = (str - buf);
+ } else if (qualifier == 'Z' ||
+ qualifier == 'z') {
+ size_t *ip = va_arg(args, size_t *);
+ *ip = (str - buf);
+ } else {
+ int *ip = va_arg(args, int *);
+ *ip = (str - buf);
+ }
+ break;
+ }
+
+ default: {
+ enum format_type type = spec.type;
+
+ if (type == FORMAT_TYPE_LONG_LONG)
+ num = va_arg(args, long long);
+ else if (type == FORMAT_TYPE_ULONG)
+ num = va_arg(args, unsigned long);
+ else if (type == FORMAT_TYPE_LONG)
+ num = va_arg(args, unsigned long);
+ else if (type == FORMAT_TYPE_SIZE_T)
+ num = va_arg(args, size_t);
+ else if (type == FORMAT_TYPE_PTRDIFF)
+ num = va_arg(args, ptrdiff_t);
+ else if (type == FORMAT_TYPE_USHORT)
+ num = (unsigned short) va_arg(args, int);
+ else if (type == FORMAT_TYPE_SHORT)
+ num = (short) va_arg(args, int);
+ else if (type == FORMAT_TYPE_UINT)
+ num = va_arg(args, unsigned int);
+ else
+ num = va_arg(args, unsigned int);
+
+ str = number(str, end, num, spec.base,
+ spec.field_width, spec.precision, spec.flags);
}
- if (qualifier == 'L')
- num = va_arg(args, long long);
- else if (qualifier == 'l') {
- num = va_arg(args, unsigned long);
- if (flags & SIGN)
- num = (signed long) num;
- } else if (qualifier == 'Z' || qualifier == 'z') {
- num = va_arg(args, size_t);
- } else if (qualifier == 't') {
- num = va_arg(args, ptrdiff_t);
- } else if (qualifier == 'h') {
- num = (unsigned short) va_arg(args, int);
- if (flags & SIGN)
- num = (signed short) num;
- } else {
- num = va_arg(args, unsigned int);
- if (flags & SIGN)
- num = (signed int) num;
}
- str = number(str, end, num, base,
- field_width, precision, flags);
}
+
if (size > 0) {
if (str < end)
*str = '\0';
else
end[-1] = '\0';
}
+
/* the trailing null byte doesn't count towards the total */
return str-buf;
+
}
EXPORT_SYMBOL(vsnprintf);

@@ -1084,8 +1268,9 @@ EXPORT_SYMBOL(sprintf);
*/
int vbin_printf(u32 *bin_buf, size_t size, const char *fmt, va_list args)
{
+ struct printf_spec spec = {0};
char *str, *end;
- int qualifier;
+ int read;

str = (char *)bin_buf;
end = (char *)(bin_buf + size);
@@ -1110,57 +1295,30 @@ do { \
str += sizeof(type); \
} while (0)

- for (; *fmt ; ++fmt) {
- if (*fmt != '%')
- continue;

-repeat:
- /* parse flags */
- ++fmt; /* this also skips first '%' */
- if (*fmt == '-' || *fmt == '+' || *fmt == ' '
- || *fmt == '#' || *fmt == '0')
- goto repeat;
+ while (*fmt) {
+ read = format_decode(fmt, &spec);

- /* parse field width */
- if (isdigit(*fmt))
- skip_atoi(&fmt);
- else if (*fmt == '*') {
- ++fmt;
- /* it's the next argument */
- save_arg(int);
- }
+ fmt += read;

- /* parse the precision */
- if (*fmt == '.') {
- ++fmt;
- if (isdigit(*fmt))
- skip_atoi(&fmt);
- else if (*fmt == '*') {
- ++fmt;
- /* it's the next argument */
- save_arg(int);
- }
+ switch (spec.type) {
+ case FORMAT_TYPE_NONE: {
+ break;
}

- /* parse the conversion qualifier */
- qualifier = -1;
- if (*fmt == 'h' || *fmt == 'l' || *fmt == 'L' ||
- *fmt == 'Z' || *fmt == 'z' || *fmt == 't') {
- qualifier = *fmt;
- ++fmt;
- if (qualifier == 'l' && *fmt == 'l') {
- qualifier = 'L';
- ++fmt;
- }
- }
+ case FORMAT_TYPE_WITDH:
+ save_arg(int);
+ break;

- /* parse format type */
- switch (*fmt) {
- case 'c':
+ case FORMAT_TYPE_PRECISION:
+ save_arg(int);
+ break;
+
+ case FORMAT_TYPE_CHAR:
save_arg(char);
- continue;
- case 's': {
- /* save the string argument */
+ break;
+
+ case FORMAT_TYPE_STR: {
const char *save_str = va_arg(args, char *);
size_t len;
if ((unsigned long)save_str > (unsigned long)-PAGE_SIZE
@@ -1170,16 +1328,27 @@ repeat:
if (str + len + 1 < end)
memcpy(str, save_str, len + 1);
str += len + 1;
- continue;
+ break;
}
- case 'p':
+
+ case FORMAT_TYPE_PTR:
save_arg(void *);
/* skip all alphanumeric pointer suffixes */
while (isalnum(fmt[1]))
fmt++;
- continue;
- case 'n': {
+ break;
+
+ case FORMAT_TYPE_PERCENT_CHAR:
+ break;
+
+ case FORMAT_TYPE_INVALID:
+ if (!*fmt)
+ --fmt;
+ break;
+
+ case FORMAT_TYPE_NRCHARS: {
/* skip %n 's argument */
+ int qualifier = spec.qualifier;
void *skip_arg;
if (qualifier == 'l')
skip_arg = va_arg(args, long *);
@@ -1187,37 +1356,32 @@ repeat:
skip_arg = va_arg(args, size_t *);
else
skip_arg = va_arg(args, int *);
- continue;
+ break;
}
- case 'o':
- case 'x':
- case 'X':
- case 'd':
- case 'i':
- case 'u':
- /* save arg for case: 'o', 'x', 'X', 'd', 'i', 'u' */
- if (qualifier == 'L')
+
+ default: {
+ enum format_type type = spec.type;
+
+ if (type == FORMAT_TYPE_LONG_LONG)
save_arg(long long);
- else if (qualifier == 'l')
+ else if (type == FORMAT_TYPE_ULONG ||
+ type == FORMAT_TYPE_LONG)
save_arg(unsigned long);
- else if (qualifier == 'Z' || qualifier == 'z')
+ else if (type == FORMAT_TYPE_SIZE_T)
save_arg(size_t);
- else if (qualifier == 't')
+ else if (type == FORMAT_TYPE_PTRDIFF)
save_arg(ptrdiff_t);
- else if (qualifier == 'h')
+ else if (type == FORMAT_TYPE_USHORT ||
+ type == FORMAT_TYPE_SHORT)
save_arg(short);
else
save_arg(int);
- continue;
- default:
- if (!*fmt)
- --fmt;
- continue;
+ }
}
}
-#undef save_arg
-
return (u32 *)(PTR_ALIGN(str, sizeof(u32))) - bin_buf;
+
+#undef save_arg
}
EXPORT_SYMBOL_GPL(vbin_printf);

@@ -1249,14 +1413,10 @@ EXPORT_SYMBOL_GPL(vbin_printf);
int bstr_printf(char *buf, size_t size, const char *fmt, const u32 *bin_buf)
{
unsigned long long num;
- int base;
char *str, *end, c;
const char *args = (const char *)bin_buf;

- int flags;
- int field_width;
- int precision;
- int qualifier;
+ struct printf_spec spec = {0};

if (unlikely((int) size < 0)) {
/* There can be only one.. */
@@ -1290,84 +1450,37 @@ int bstr_printf(char *buf, size_t size, const char *fmt, const u32 *bin_buf)
size = end - buf;
}

- for (; *fmt ; ++fmt) {
- if (*fmt != '%') {
- if (str < end)
- *str = *fmt;
- ++str;
- continue;
- }
+ while (*fmt) {
+ int read;
+ char *old_fmt = (char *)fmt;

- /* process flags */
- flags = 0;
-repeat:
- ++fmt; /* this also skips first '%' */
- switch (*fmt) {
- case '-':
- flags |= LEFT;
- goto repeat;
- case '+':
- flags |= PLUS;
- goto repeat;
- case ' ':
- flags |= SPACE;
- goto repeat;
- case '#':
- flags |= SPECIAL;
- goto repeat;
- case '0':
- flags |= ZEROPAD;
- goto repeat;
- }
+ read = format_decode(fmt, &spec);

- /* get field width */
- field_width = -1;
- if (isdigit(*fmt))
- field_width = skip_atoi(&fmt);
- else if (*fmt == '*') {
- ++fmt;
- /* it's the next argument */
- field_width = get_arg(int);
- if (field_width < 0) {
- field_width = -field_width;
- flags |= LEFT;
- }
- }
+ fmt += read;

- /* get the precision */
- precision = -1;
- if (*fmt == '.') {
- ++fmt;
- if (isdigit(*fmt))
- precision = skip_atoi(&fmt);
- else if (*fmt == '*') {
- ++fmt;
- /* it's the next argument */
- precision = get_arg(int);
+ switch (spec.type) {
+ case FORMAT_TYPE_NONE: {
+ int copy = read;
+ if (str < end) {
+ if (copy > end - str)
+ copy = end - str;
+ memcpy(str, old_fmt, copy);
}
- if (precision < 0)
- precision = 0;
+ str += read;
+ break;
}

- /* get the conversion qualifier */
- qualifier = -1;
- if (*fmt == 'h' || *fmt == 'l' || *fmt == 'L' ||
- *fmt == 'Z' || *fmt == 'z' || *fmt == 't') {
- qualifier = *fmt;
- ++fmt;
- if (qualifier == 'l' && *fmt == 'l') {
- qualifier = 'L';
- ++fmt;
- }
- }
+ case FORMAT_TYPE_WITDH:
+ spec.field_width = get_arg(int);
+ break;

- /* default base */
- base = 10;
+ case FORMAT_TYPE_PRECISION:
+ spec.precision = get_arg(int);
+ break;

- switch (*fmt) {
- case 'c':
- if (!(flags & LEFT)) {
- while (--field_width > 0) {
+ case FORMAT_TYPE_CHAR:
+ if (!(spec.flags & LEFT)) {
+ while (--spec.field_width > 0) {
if (str < end)
*str = ' ';
++str;
@@ -1377,58 +1490,38 @@ repeat:
if (str < end)
*str = c;
++str;
- while (--field_width > 0) {
+ while (--spec.field_width > 0) {
if (str < end)
*str = ' ';
++str;
}
- continue;
+ break;

- case 's':{
+ case FORMAT_TYPE_STR: {
const char *str_arg = args;
size_t len = strlen(str_arg);
args += len + 1;
- str = string(str, end, (char *)str_arg, field_width,
- precision, flags);
- continue;
+ str = string(str, end, (char *)str_arg,
+ spec.field_width,
+ spec.precision, spec.flags);
+ break;
}

- case 'p':
+ case FORMAT_TYPE_PTR:
str = pointer(fmt+1, str, end, get_arg(void *),
- field_width, precision, flags);
- /* Skip all alphanumeric pointer suffixes */
+ spec.field_width, spec.precision,
+ spec.flags);
while (isalnum(fmt[1]))
fmt++;
- continue;
-
- case 'n':
- /* skip %n */
- continue;
+ break;

- case '%':
+ case FORMAT_TYPE_PERCENT_CHAR:
if (str < end)
*str = '%';
++str;
- continue;
-
- /* integer number formats - set up the flags and "break" */
- case 'o':
- base = 8;
- break;
-
- case 'x':
- flags |= SMALL;
- case 'X':
- base = 16;
break;

- case 'd':
- case 'i':
- flags |= SIGN;
- case 'u':
- break;
-
- default:
+ case FORMAT_TYPE_INVALID:
if (str < end)
*str = '%';
++str;
@@ -1439,36 +1532,47 @@ repeat:
} else {
--fmt;
}
- continue;
+ break;
+
+ case FORMAT_TYPE_NRCHARS:
+ /* skip */
+ break;
+
+ default: {
+ enum format_type type = spec.type;
+
+ if (type == FORMAT_TYPE_LONG_LONG)
+ num = get_arg(long long);
+ else if (type == FORMAT_TYPE_ULONG)
+ num = get_arg(unsigned long);
+ else if (type == FORMAT_TYPE_LONG)
+ num = get_arg(unsigned long);
+ else if (type == FORMAT_TYPE_SIZE_T)
+ num = get_arg(size_t);
+ else if (type == FORMAT_TYPE_PTRDIFF)
+ num = get_arg(ptrdiff_t);
+ else if (type == FORMAT_TYPE_USHORT)
+ num = get_arg(unsigned short);
+ else if (type == FORMAT_TYPE_SHORT)
+ num = get_arg(short);
+ else if (type == FORMAT_TYPE_UINT)
+ num = get_arg(unsigned int);
+ else
+ num = get_arg(int);
+
+ str = number(str, end, num, spec.base,
+ spec.field_width, spec.precision, spec.flags);
}
- if (qualifier == 'L')
- num = get_arg(long long);
- else if (qualifier == 'l') {
- num = get_arg(unsigned long);
- if (flags & SIGN)
- num = (signed long) num;
- } else if (qualifier == 'Z' || qualifier == 'z') {
- num = get_arg(size_t);
- } else if (qualifier == 't') {
- num = get_arg(ptrdiff_t);
- } else if (qualifier == 'h') {
- num = (unsigned short) get_arg(short);
- if (flags & SIGN)
- num = (signed short) num;
- } else {
- num = get_arg(unsigned int);
- if (flags & SIGN)
- num = (signed int) num;
}
- str = number(str, end, num, base,
- field_width, precision, flags);
}
+
if (size > 0) {
if (str < end)
*str = '\0';
else
end[-1] = '\0';
}
+
#undef get_arg

/* the trailing null byte doesn't count towards the total */
--
1.6.1





\
 
 \ /
  Last update: 2009-02-28 09:15    [W:0.195 / U:0.624 seconds]
©2003-2020 Jasper Spaans|hosted at Digital Ocean and TransIP|Read the blog|Advertise on this site