summaryrefslogtreecommitdiff
path: root/upb/bytestream.h
diff options
context:
space:
mode:
authorJosh Haberman <haberman@google.com>2013-05-11 16:45:38 -0700
committerJosh Haberman <haberman@google.com>2013-05-11 16:45:38 -0700
commitcfdb9907cb87d15eaab72ceefbfa42fd7a4c3127 (patch)
tree63f5d70ad64daeeb4ffc777c2c3afd50e2e281b1 /upb/bytestream.h
parent7d3e2bd2c4cfd1296d1d6f996d7548de26540d41 (diff)
Synced with 3 months of Google-internal development.
Major changes: - Got rid of all bytestream interfaces in favor of using regular handlers. - new Pipeline object represents a upb pipeline, does bump allocation internally to manage memory. - proto2 support now can handle extensions.
Diffstat (limited to 'upb/bytestream.h')
-rw-r--r--upb/bytestream.h627
1 files changed, 24 insertions, 603 deletions
diff --git a/upb/bytestream.h b/upb/bytestream.h
index 41f767a..a0c4110 100644
--- a/upb/bytestream.h
+++ b/upb/bytestream.h
@@ -1,629 +1,50 @@
/*
* upb - a minimalist implementation of protocol buffers.
*
- * Copyright (c) 2011 Google Inc. See LICENSE for details.
+ * Copyright (c) 2013 Google Inc. See LICENSE for details.
* Author: Josh Haberman <jhaberman@gmail.com>
*
- * This file defines three core interfaces:
- * - upb_bytesink: for writing streams of data.
- * - upb_bytesrc: for reading streams of data.
- * - upb_byteregion: for reading from a specific region of a upb_bytesrc;
- * should be used by decoders instead of using upb_bytesrc directly.
+ * This file contains the standard ByteStream msgdef and some useful routines
+ * surrounding it.
*
- * These interfaces are used by streaming encoders and decoders: for example, a
- * protobuf parser gets its input from a upb_byteregion. They are virtual base
- * classes so concrete implementations can get the data from a fd, a FILE*, a
- * string, etc.
+ * This is a mixed C/C++ interface that offers a full API to both languages.
+ * See the top-level README for more information.
*/
-// A upb_byteregion represents a region of data from a bytesrc.
-//
-// Parsers get data from this interface instead of a bytesrc because we often
-// want to parse only a specific region of the input. For example, if we parse
-// a string from our input but know that the string represents a protobuf, we
-// can pass its upb_byteregion to an appropriate protobuf parser.
-//
-// Since the bytes may be coming from a file or network socket, bytes must be
-// fetched before they can be read (though in some cases this fetch may be a
-// no-op). "fetch" is the only operation on a byteregion that could fail or
-// block, because it is the only operation that actually performs I/O.
-//
-// Bytes can be discarded when they are no longer needed. Parsers should
-// always discard bytes they no longer need, both so the buffers can be freed
-// when possible and to give better visibility into what bytes the parser is
-// still using.
-//
-// start discard read fetch end
-// ofs ofs ofs ofs ofs
-// | |--->discard() | |--->fetch() |
-// V V V V V
-// +-------------+-------------------------+-----------------+-----------------+
-// | discarded | | | fetchable |
-// +-------------+-------------------------+-----------------+-----------------+
-// | <------------- loaded ------------------> |
-// | <- available -> |
-// | <---------- remaining ----------> |
-//
-// Note that the start offset may be something other than zero! A byteregion
-// is a view into an underlying bytesrc stream, and the region may start
-// somewhere other than the beginning of that stream.
-//
-// The region can be either delimited or nondelimited. A non-delimited region
-// will keep returning data until the underlying data source returns EOF. A
-// delimited region will return EOF at a predetermined offset.
-//
-// end
-// ofs
-// |
-// V
-// +-----------------------+
-// | delimited region | <-- hard EOF, even if data source has more data.
-// +-----------------------+
-//
-// +------------------------
-// | nondelimited region Z <-- won't return EOF until data source hits EOF.
-// +------------------------
-//
-// TODO: if 64-bit math for stream offsets is a performance issue on
-// non-64-bit machines, we could introduce a upb_off_t typedef that can be
-// defined as a 32-bit type for applications that don't need to handle
-// streams longer than 4GB.
+#ifndef UPB_BYTESTREAM_H_
+#define UPB_BYTESTREAM_H_
+#include "upb/sink.h"
+#include "upb/bytestream.upb.h"
-#ifndef UPB_BYTESTREAM_H
-#define UPB_BYTESTREAM_H
+#define UPB_BYTESTREAM_BYTES &upb_bytestream_fields[0]
-#include "upb.h"
-
-#ifdef __cplusplus
-namespace upb {
-class ByteRegion;
-class StringSource;
-} // namespace upb
-typedef upb::StringSource upb_stringsrc;
-extern "C" {
-#else
-struct upb_stringsrc;
-typedef struct upb_stringsrc upb_stringsrc;
-#endif
-
-typedef enum {
- UPB_BYTE_OK = UPB_OK,
- UPB_BYTE_WOULDBLOCK = UPB_SUSPENDED,
- UPB_BYTE_ERROR = UPB_ERROR,
- UPB_BYTE_EOF
-} upb_bytesuccess_t;
-
-/* upb_bytesrc ****************************************************************/
-
-// A upb_bytesrc allows the consumer of a stream of bytes to obtain buffers as
-// they become available, and to preserve some trailing amount of data before
-// it is discarded. Consumers should not use upb_bytesrc directly, but rather
-// should use a upb_byteregion (which allows access to a region of a bytesrc).
-//
-// upb_bytesrc is a virtual base class with implementations that get data from
-// eg. a string, a cord, a file descriptor, a FILE*, etc.
-
-typedef upb_bytesuccess_t upb_bytesrc_fetch_func(void*, uint64_t, size_t*);
-typedef void upb_bytesrc_discard_func(void*, uint64_t);
-typedef void upb_bytesrc_copy_func(const void*, uint64_t, size_t, char*);
-typedef const char *upb_bytesrc_getptr_func(const void*, uint64_t, size_t*);
-typedef struct _upb_bytesrc_vtbl {
- upb_bytesrc_fetch_func *fetch;
- upb_bytesrc_discard_func *discard;
- upb_bytesrc_copy_func *copy;
- upb_bytesrc_getptr_func *getptr;
-} upb_bytesrc_vtbl;
-
-typedef struct {
- const upb_bytesrc_vtbl *vtbl;
- upb_status status;
-} upb_bytesrc;
-
-INLINE void upb_bytesrc_init(upb_bytesrc *src, const upb_bytesrc_vtbl *vtbl) {
- src->vtbl = vtbl;
- upb_status_init(&src->status);
-}
-
-INLINE void upb_bytesrc_uninit(upb_bytesrc *src) {
- upb_status_uninit(&src->status);
-}
-
-// Fetches at least one byte starting at ofs, returning the success or failure
-// of the operation. If UPB_BYTE_OK is returned, *read indicates the number of
-// of bytes successfully fetched; any error or EOF status will be reflected in
-// upb_bytesrc_status(). It is valid for bytes to be fetched multiple times,
-// as long as the bytes have not been previously discarded.
-INLINE upb_bytesuccess_t upb_bytesrc_fetch(upb_bytesrc *src, uint64_t ofs,
- size_t *read) {
- return src->vtbl->fetch(src, ofs, read);
-}
-
-// Discards all data prior to ofs (except data that is pinned, if pinning
-// support is added -- see TODO below).
-INLINE void upb_bytesrc_discard(upb_bytesrc *src, uint64_t ofs) {
- src->vtbl->discard(src, ofs);
-}
-
-// Copies "len" bytes of data from ofs to "dst", which must be at least "len"
-// bytes long. The given region must not be discarded.
-INLINE void upb_bytesrc_copy(const upb_bytesrc *src, uint64_t ofs, size_t len,
- char *dst) {
- src->vtbl->copy(src, ofs, len, dst);
-}
-
-// Returns a pointer to the bytesrc's internal buffer, storing in *len how much
-// data is available. The given offset must not be discarded. The returned
-// buffer is valid for as long as its bytes are not discarded (in the case that
-// part of the returned buffer is discarded, only the non-discarded bytes
-// remain valid).
-INLINE const char *upb_bytesrc_getptr(const upb_bytesrc *src, uint64_t ofs,
- size_t *len) {
- return src->vtbl->getptr(src, ofs, len);
-}
-
-// TODO: Add if/when there is a demonstrated need:
-//
-// // When the caller pins a region (which must not be already discarded), it
-// // is guaranteed that the region will not be discarded (nor will the bytesrc
-// // be destroyed) until the region is unpinned. However, not all bytesrc's
-// // support pinning; a false return indicates that a pin was not possible.
-// INLINE bool upb_bytesrc_pin(upb_bytesrc *src, uint64_t ofs, size_t len) {
-// return src->vtbl->refregion(src, ofs, len);
-// }
-//
-// // Releases some number of pinned bytes from the beginning of a pinned
-// // region (which may be fewer than the total number of bytes pinned).
-// INLINE void upb_bytesrc_unpin(upb_bytesrc *src, uint64_t ofs, size_t len,
-// size_t bytes_to_release) {
-// src->vtbl->unpin(src, ofs, len);
-// }
-//
-// Adding pinning support would also involve adding a "pin_ofs" parameter to
-// upb_bytesrc_fetch, so that the fetch can extend an already-pinned region.
-
-
-/* upb_byteregion *************************************************************/
-
-#define UPB_NONDELIMITED (0xffffffffffffffffULL)
-
-#ifdef __cplusplus
-} // extern "C"
-
-class upb::ByteRegion {
- public:
- static const uint64_t kNondelimited = UPB_NONDELIMITED;
- typedef upb_bytesuccess_t ByteSuccess;
-
- // Accessors for the regions bounds -- the meaning of these is described in
- // the diagram above.
- uint64_t start_ofs() const;
- uint64_t discard_ofs() const;
- uint64_t fetch_ofs() const;
- uint64_t end_ofs() const;
-
- // Returns how many bytes are fetched and available for reading starting from
- // offset "offset".
- uint64_t BytesAvailable(uint64_t offset) const;
-
- // Returns the total number of bytes remaining after offset "offset", or
- // kNondelimited if the byteregion is non-delimited.
- uint64_t BytesRemaining(uint64_t offset) const;
-
- uint64_t Length() const;
-
- // Sets the value of this byteregion to be a subset of the given byteregion's
- // data. The caller is responsible for releasing this region before the src
- // region is released (unless the region is first pinned, if pinning support
- // is added. see below).
- void Reset(const upb_byteregion *src, uint64_t ofs, uint64_t len);
- void Release();
-
- // Attempts to fetch more data, extending the fetched range of this
- // byteregion. Returns true if the fetched region was extended by at least
- // one byte, false on EOF or error (see *s for details).
- ByteSuccess Fetch();
-
- // Fetches all remaining data, returning false if the operation failed (see
- // *s for details). May only be used on delimited byteregions.
- ByteSuccess FetchAll();
-
- // Discards bytes from the byteregion up until ofs (which must be greater or
- // equal to discard_ofs()). It is valid to discard bytes that have not been
- // fetched (such bytes will never be fetched) but it is an error to discard
- // past the end of a delimited byteregion.
- void Discard(uint64_t ofs);
-
- // Copies "len" bytes of data into "dst", starting at ofs. The specified
- // region must be available.
- void Copy(uint64_t ofs, size_t len, char *dst) const;
-
- // Copies all bytes from the byteregion into dst. Requires that the entire
- // byteregion is fetched and that none has been discarded.
- void CopyAll(char *dst) const;
-
- // Returns a pointer to the internal buffer for the byteregion starting at
- // offset "ofs." Stores the number of bytes available in this buffer in *len.
- // The returned buffer is invalidated when the byteregion is reset or
- // released, or when the bytes are discarded. If the byteregion is not
- // currently pinned, the pointer is only valid for the lifetime of the parent
- // byteregion.
- const char *GetPtr(uint64_t ofs, size_t *len) const;
-
- // Copies the contents of the byteregion into a newly-allocated,
- // NULL-terminated string. Requires that the byteregion is fully fetched.
- char *StrDup() const;
-
- template <typename T> void AssignToString(T* str);
-
-#else
-struct upb_byteregion {
-#endif
- uint64_t start;
- uint64_t discard;
- uint64_t fetch;
- uint64_t end; // UPB_NONDELIMITED if nondelimited.
- upb_bytesrc *bytesrc;
- bool toplevel; // If true, discards hit the underlying bytesrc.
-};
-
-#ifdef __cplusplus
-extern "C" {
-#endif
-
-// Native C API.
-INLINE uint64_t upb_byteregion_startofs(const upb_byteregion *r) {
- return r->start;
-}
-INLINE uint64_t upb_byteregion_discardofs(const upb_byteregion *r) {
- return r->discard;
-}
-INLINE uint64_t upb_byteregion_fetchofs(const upb_byteregion *r) {
- return r->fetch;
-}
-INLINE uint64_t upb_byteregion_endofs(const upb_byteregion *r) {
- return r->end;
-}
-INLINE uint64_t upb_byteregion_available(const upb_byteregion *r, uint64_t o) {
- assert(o >= upb_byteregion_discardofs(r));
- assert(o <= r->fetch); // Could relax this.
- return r->fetch - o;
-}
-INLINE uint64_t upb_byteregion_remaining(const upb_byteregion *r, uint64_t o) {
- return r->end == UPB_NONDELIMITED ? UPB_NONDELIMITED : r->end - o;
-}
-
-INLINE uint64_t upb_byteregion_len(const upb_byteregion *r) {
- return upb_byteregion_remaining(r, r->start);
-}
-void upb_byteregion_reset(upb_byteregion *r, const upb_byteregion *src,
- uint64_t ofs, uint64_t len);
-void upb_byteregion_release(upb_byteregion *r);
-upb_bytesuccess_t upb_byteregion_fetch(upb_byteregion *r);
-INLINE upb_bytesuccess_t upb_byteregion_fetchall(upb_byteregion *r) {
- assert(upb_byteregion_len(r) != UPB_NONDELIMITED);
- upb_bytesuccess_t ret;
- do {
- ret = upb_byteregion_fetch(r);
- } while (ret == UPB_BYTE_OK);
- return ret == UPB_BYTE_EOF ? UPB_BYTE_OK : ret;
-}
-INLINE void upb_byteregion_discard(upb_byteregion *r, uint64_t ofs) {
- assert(ofs >= upb_byteregion_discardofs(r));
- assert(ofs <= upb_byteregion_endofs(r));
- r->discard = ofs;
- if (ofs > r->fetch) r->fetch = ofs;
- if (r->toplevel) upb_bytesrc_discard(r->bytesrc, ofs);
-}
-INLINE void upb_byteregion_copy(const upb_byteregion *r, uint64_t ofs,
- size_t len, char *dst) {
- assert(ofs >= upb_byteregion_discardofs(r));
- assert(len <= upb_byteregion_available(r, ofs));
- upb_bytesrc_copy(r->bytesrc, ofs, len, dst);
-}
-INLINE void upb_byteregion_copyall(const upb_byteregion *r, char *dst) {
- assert(r->start == r->discard && r->end == r->fetch);
- upb_byteregion_copy(r, r->start, upb_byteregion_len(r), dst);
-}
-INLINE const char *upb_byteregion_getptr(const upb_byteregion *r,
- uint64_t ofs, size_t *len) {
- assert(ofs >= upb_byteregion_discardofs(r));
- const char *ret = upb_bytesrc_getptr(r->bytesrc, ofs, len);
- *len = UPB_MIN(*len, upb_byteregion_available(r, ofs));
+// A convenience method that handles the start/end calls and tracks overall
+// success.
+UPB_INLINE bool upb_bytestream_putstr(upb_sink *s, const char *buf, size_t n) {
+ bool ret =
+ upb_sink_startmsg(s) &&
+ upb_sink_startstr(s, UPB_BYTESTREAM_BYTES_STARTSTR, n) &&
+ upb_sink_putstring(s, UPB_BYTESTREAM_BYTES_STRING, buf, n) == n &&
+ upb_sink_endstr(s, UPB_BYTESTREAM_BYTES_ENDSTR);
+ if (ret) upb_sink_endmsg(s);
return ret;
}
-// TODO: add if/when there is a demonstrated need.
-//
-// // Pins this byteregion's bytes in memory, allowing it to outlive its parent
-// // byteregion. Normally a byteregion may only be used while its parent is
-// // still valid, but a pinned byteregion may continue to be used until it is
-// // reset or released. A byteregion must be fully fetched to be pinned
-// // (this implies that the byteregion must be delimited).
-// //
-// // In some cases this operation may cause the input data to be copied.
-// //
-// // void upb_byteregion_pin(upb_byteregion *r);
-
-// Convenience functions for creating and destroying a byteregion with a simple
-// string as its data. These are relatively inefficient compared with creating
-// your own bytesrc (they call malloc() and copy the string data) so should not
-// be used on any critical path.
-//
-// The string data in the returned region is guaranteed to be contiguous and
-// NULL-terminated.
-upb_byteregion *upb_byteregion_new(const void *str);
-upb_byteregion *upb_byteregion_newl(const void *str, size_t len);
-// May *only* be called on a byteregion created with upb_byteregion_new[l]()!
-void upb_byteregion_free(upb_byteregion *r);
-
-// Copies the contents of the byteregion into a newly-allocated, NULL-terminated
-// string. Requires that the byteregion is fully fetched.
-char *upb_byteregion_strdup(const upb_byteregion *r);
-
-
-/* upb_bytesink ***************************************************************/
-
-// A bytesink is an interface that allows the caller to push byte-wise data.
-// It is very simple -- the only special capability is the ability to "rewind"
-// the stream, which is really only a mechanism of having the bytesink ignore
-// some subsequent calls.
-typedef int upb_bytesink_write_func(void*, const void*, int);
-typedef int upb_bytesink_vprintf_func(void*, const char *fmt, va_list args);
-
-typedef struct {
- upb_bytesink_write_func *write;
- upb_bytesink_vprintf_func *vprintf;
-} upb_bytesink_vtbl;
-
-typedef struct {
- upb_bytesink_vtbl *vtbl;
- upb_status status;
- uint64_t offset;
-} upb_bytesink;
-
-// Should be called by derived classes.
-void upb_bytesink_init(upb_bytesink *sink, upb_bytesink_vtbl *vtbl);
-void upb_bytesink_uninit(upb_bytesink *sink);
-
-INLINE int upb_bytesink_write(upb_bytesink *s, const void *buf, int len) {
- return s->vtbl->write(s, buf, len);
-}
-
-#define upb_bytesink_writestr(s, buf) upb_bytesink_write(s, buf, strlen(buf))
-
-// Returns the number of bytes written or -1 on error.
-INLINE int upb_bytesink_printf(upb_bytesink *sink, const char *fmt, ...) {
- va_list args;
- va_start(args, fmt);
- uint32_t ret = sink->vtbl->vprintf(sink, fmt, args);
- va_end(args);
- return ret;
-}
-
-INLINE int upb_bytesink_putc(upb_bytesink *sink, char ch) {
- return upb_bytesink_write(sink, &ch, 1);
-}
-
-INLINE int upb_bytesink_putrepeated(upb_bytesink *sink, char ch, int len) {
- for (int i = 0; i < len; i++)
- if (upb_bytesink_write(sink, &ch, 1) < 0)
- return -1;
- return len;
-}
-
-INLINE uint64_t upb_bytesink_getoffset(upb_bytesink *sink) {
- return sink->offset;
-}
-
-// Rewinds the stream to the given offset. This cannot actually "unput" any
-// data, it is for situations like:
-//
-// // If false is returned (because of error), call again later to resume.
-// bool write_some_data(upb_bytesink *sink, int indent) {
-// uint64_t start_offset = upb_bytesink_getoffset(sink);
-// if (upb_bytesink_writestr(sink, "Some data") < 0) goto err;
-// if (upb_bytesink_putrepeated(sink, ' ', indent) < 0) goto err;
-// return true;
-// err:
-// upb_bytesink_rewind(sink, start_offset);
-// return false;
-// }
-//
-// The subsequent bytesink writes *must* be identical to the writes that were
-// rewinded past.
-INLINE void upb_bytesink_rewind(upb_bytesink *sink, uint64_t offset) {
- // TODO
- (void)sink;
- (void)offset;
-}
-
-// OPT: add getappendbuf()
-// OPT: add writefrombytesrc()
-// TODO: add flush()
-
-
-/* upb_stringsrc **************************************************************/
-
-// bytesrc/bytesink for a simple contiguous string.
-
#ifdef __cplusplus
-} // extern "C"
-
-class upb::StringSource {
- public:
- StringSource();
- template <typename T> explicit StringSource(const T& str);
- StringSource(const char *data, size_t len);
- ~StringSource();
-
- // Resets the stringsrc to a state where it will vend the given string. The
- // string data must be valid until the stringsrc is reset again or destroyed.
- void Reset(const char* data, size_t len);
- template <typename T> void Reset(const T& str);
-
- // Returns the top-level upb_byteregion* for this stringsrc. Invalidated
- // when the stringsrc is reset.
- ByteRegion* AllBytes();
-
- upb_bytesrc* ByteSource();
-
-#else
-struct upb_stringsrc {
-#endif
- upb_bytesrc bytesrc;
- const char *str;
- size_t len;
- upb_byteregion byteregion;
-};
-
-#ifdef __cplusplus
-extern "C" {
-#endif
-
-// Native C API.
-void upb_stringsrc_init(upb_stringsrc *s);
-void upb_stringsrc_uninit(upb_stringsrc *s);
-void upb_stringsrc_reset(upb_stringsrc *s, const char *str, size_t len);
-INLINE upb_bytesrc *upb_stringsrc_bytesrc(upb_stringsrc *s) {
- return &s->bytesrc;
-}
-INLINE upb_byteregion *upb_stringsrc_allbytes(upb_stringsrc *s) {
- return &s->byteregion;
-}
-
-
-/* upb_stringsink *************************************************************/
-
-struct _upb_stringsink {
- upb_bytesink bytesink;
- char *str;
- size_t len, size;
-};
-typedef struct _upb_stringsink upb_stringsink;
-
-// Create/free a stringsrc.
-void upb_stringsink_init(upb_stringsink *s);
-void upb_stringsink_uninit(upb_stringsink *s);
-
-// Resets the sink's string to "str", which the sink takes ownership of.
-// "str" may be NULL, which will make the sink allocate a new string.
-void upb_stringsink_reset(upb_stringsink *s, char *str, size_t len);
-
-// Releases ownership of the returned string (which is "len" bytes long) and
-// resets the internal string to be empty again (as if reset were called with
-// NULL).
-const char *upb_stringsink_release(upb_stringsink *s, size_t *len);
-
-// Returns the upb_bytesink* for this stringsrc. Invalidated by reset above.
-upb_bytesink *upb_stringsink_bytesink(upb_stringsink *s);
-
-#ifdef __cplusplus
-} // extern "C"
namespace upb {
-inline uint64_t ByteRegion::start_ofs() const {
- return upb_byteregion_startofs(this);
-}
-inline uint64_t ByteRegion::discard_ofs() const {
- return upb_byteregion_discardofs(this);
-}
-inline uint64_t ByteRegion::fetch_ofs() const {
- return upb_byteregion_fetchofs(this);
-}
-inline uint64_t ByteRegion::end_ofs() const {
- return upb_byteregion_endofs(this);
-}
-inline uint64_t ByteRegion::BytesAvailable(uint64_t offset) const {
- return upb_byteregion_available(this, offset);
-}
-inline uint64_t ByteRegion::BytesRemaining(uint64_t offset) const {
- return upb_byteregion_remaining(this, offset);
-}
-inline uint64_t ByteRegion::Length() const {
- return upb_byteregion_len(this);
-}
-inline void ByteRegion::Reset(
- const upb_byteregion *src, uint64_t ofs, uint64_t len) {
- upb_byteregion_reset(this, src, ofs, len);
-}
-inline void ByteRegion::Release() {
- upb_byteregion_release(this);
-}
-inline ByteRegion::ByteSuccess ByteRegion::Fetch() {
- return upb_byteregion_fetch(this);
-}
-inline ByteRegion::ByteSuccess ByteRegion::FetchAll() {
- return upb_byteregion_fetchall(this);
-}
-inline void ByteRegion::Discard(uint64_t ofs) {
- upb_byteregion_discard(this, ofs);
-}
-inline void ByteRegion::Copy(uint64_t ofs, size_t len, char *dst) const {
- upb_byteregion_copy(this, ofs, len, dst);
-}
-inline void ByteRegion::CopyAll(char *dst) const {
- upb_byteregion_copyall(this, dst);
-}
-inline const char *ByteRegion::GetPtr(uint64_t ofs, size_t *len) const {
- return upb_byteregion_getptr(this, ofs, len);
-}
-inline char *ByteRegion::StrDup() const {
- return upb_byteregion_strdup(this);
-}
-template <typename T> void ByteRegion::AssignToString(T* str) {
- uint64_t ofs = start_ofs();
- size_t len;
- const char *ptr = GetPtr(ofs, &len);
- // Emperically calling reserve() here is counterproductive and slows down
- // benchmarks. If the parsing is happening in a tight loop that is reusing
- // the string object, there is probably enough data reserved already and
- // the reserve() call is extra overhead.
- str->assign(ptr, len);
- ofs += len;
- while (ofs < end_ofs()) {
- ptr = GetPtr(ofs, &len);
- str->append(ptr, len);
- ofs += len;
- }
-}
-
-template <> inline ByteRegion* GetValue<ByteRegion*>(Value v) {
- return static_cast<ByteRegion*>(upb_value_getbyteregion(v));
-}
-
-template <> inline Value MakeValue<ByteRegion*>(ByteRegion* v) {
- return upb_value_byteregion(v);
+inline bool PutStringToBytestream(Sink* s, const char* buf, size_t n) {
+ return upb_bytestream_putstr(s, buf, n);
}
-inline StringSource::StringSource() { upb_stringsrc_init(this); }
-template <typename T> StringSource::StringSource(const T& str) {
- upb_stringsrc_init(this);
- Reset(str);
-}
-inline StringSource::StringSource(const char *data, size_t len) {
- upb_stringsrc_init(this);
- Reset(data, len);
-}
-inline StringSource::~StringSource() {
- upb_stringsrc_uninit(this);
-}
-inline void StringSource::Reset(const char* data, size_t len) {
- upb_stringsrc_reset(this, data, len);
-}
-template <typename T> void StringSource::Reset(const T& str) {
- upb_stringsrc_reset(this, str.c_str(), str.size());
-}
-inline ByteRegion* StringSource::AllBytes() {
- return upb_stringsrc_allbytes(this);
-}
-inline upb_bytesrc* StringSource::ByteSource() {
- return upb_stringsrc_bytesrc(this);
+template <class T> bool PutStringToBytestream(Sink* s, T str) {
+ return upb_bytestream_putstr(s, str.c_str(), str.size());
}
} // namespace upb
#endif
-#endif
+#endif // UPB_BYTESTREAM_H_
generated by cgit on debian on lair
contact matthew@masot.net with questions or feedback