VALUE msgpack_buffer_all_as_string(msgpack_buffer_t* b)
{
if(b->head == &b->tail) {
return _msgpack_buffer_head_chunk_as_string(b);
}
size_t length = msgpack_buffer_all_readable_size(b);
VALUE string = rb_str_new(NULL, length);
char* buffer = RSTRING_PTR(string);
size_t avail = msgpack_buffer_top_readable_size(b);
memcpy(buffer, b->read_buffer, avail);
buffer += avail;
length -= avail;
msgpack_buffer_chunk_t* c = b->head->next;
while(true) {
avail = c->last - c->first;
memcpy(buffer, c->first, avail);
if(length <= avail) {
return string;
}
buffer += avail;
length -= avail;
c = c->next;
}
}
static VALUE Buffer_read(int argc, VALUE* argv, VALUE self)
{
VALUE out = Qnil;
unsigned long n = -1;
bool all = false;
switch(argc) {
case 2:
out = argv[1];
/* pass through */
case 1:
n = FIX2ULONG(argv[0]);
break;
case 0:
all = true;
break;
default:
rb_raise(rb_eArgError, "wrong number of arguments (%d for 0..2)", argc);
}
BUFFER(self, b);
if(out != Qnil) {
CHECK_STRING_TYPE(out);
}
if(all) {
return read_all(b, out);
}
if(n == 0) {
/* do nothing */
MAKE_EMPTY_STRING(out);
return out;
}
#ifndef DISABLE_BUFFER_READ_TO_S_OPTIMIZE
if(!msgpack_buffer_has_io(b) && out == Qnil &&
msgpack_buffer_all_readable_size(b) <= n) {
/* same as to_s && clear; optimize */
VALUE str = msgpack_buffer_all_as_string(b);
msgpack_buffer_clear(b);
if(RSTRING_LEN(str) == 0) {
return Qnil;
} else {
return str;
}
}
#endif
MAKE_EMPTY_STRING(out);
read_until_eof(b, out, n);
if(RSTRING_LEN(out) == 0) {
return Qnil;
} else {
return out;
}
}
static mrb_value Packer_size(mrb_state *mrb, mrb_value self)
{
PACKER(mrb, self, pk);
size_t size = msgpack_buffer_all_readable_size(PACKER_BUFFER_(pk));
//return SIZET2NUM(size);
return mrb_fixnum_value(size);
}
static VALUE Packer_size(VALUE self)
{
PACKER(self, pk);
size_t size = msgpack_buffer_all_readable_size(PACKER_BUFFER_(pk));
return SIZET2NUM(size);
}
static VALUE Buffer_size(VALUE self)
{
BUFFER(self, b);
size_t size = msgpack_buffer_all_readable_size(b);
return SIZET2NUM(size);
}
