diff options
| -rw-r--r-- | util/cbfstool/common.h | 3 | ||||
| -rw-r--r-- | util/cbfstool/xdr.c | 9 |
2 files changed, 10 insertions, 2 deletions
diff --git a/util/cbfstool/common.h b/util/cbfstool/common.h index 16e75f9cf9..d0069c3d58 100644 --- a/util/cbfstool/common.h +++ b/util/cbfstool/common.h @@ -112,6 +112,7 @@ struct xdr { /* xdr.c */ extern struct xdr xdr_le, xdr_be; -int bgets(struct buffer *input, void *output, size_t len); +size_t bgets(struct buffer *input, void *output, size_t len); +size_t bputs(struct buffer *b, const void *data, size_t len); #endif diff --git a/util/cbfstool/xdr.c b/util/cbfstool/xdr.c index e2cafd14b4..df2c4ba4a0 100644 --- a/util/cbfstool/xdr.c +++ b/util/cbfstool/xdr.c @@ -25,7 +25,7 @@ #include <stdint.h> #include "common.h" -int bgets(struct buffer *input, void *output, size_t len) +size_t bgets(struct buffer *input, void *output, size_t len) { len = input->size < len ? input->size : len; memmove(output, input->data, len); @@ -34,6 +34,13 @@ int bgets(struct buffer *input, void *output, size_t len) return len; } +size_t bputs(struct buffer *b, const void *data, size_t len) +{ + memmove(&b->data[b->size], data, len); + b->size += len; + return len; +} + /* The assumption in all this code is that we're given a pointer to enough data. * Hence, we do not check for underflow. */ |