1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
/* Please see the LICENSE file for copyright and distribution information */
#include "ruby_libxml.h"
#include <ruby/io.h>
static ID READ_METHOD;
static ID WRITE_METHOD;
/* This method is called by libxml when it wants to read
more data from a stream. We go with the duck typing
solution to support StringIO objects. */
int rxml_read_callback(void *context, char *buffer, int len)
{
VALUE io = (VALUE) context;
VALUE string = rb_funcall(io, READ_METHOD, 1, INT2NUM(len));
size_t size;
if (string == Qnil)
return 0;
size = RSTRING_LEN(string);
memcpy(buffer, StringValuePtr(string), size);
return (int)size;
}
int rxml_write_callback(VALUE io, const char *buffer, int len)
{
if (rb_io_check_io(io) == Qnil)
{
// Could be StringIO
VALUE written, string;
string = rb_external_str_new_with_enc(buffer, (long)strlen(buffer), rb_enc_get(io));
written = rb_funcall(io, WRITE_METHOD, 1, string);
return NUM2INT(written);
}
else
{
return (int)rb_io_bufwrite(io, buffer, (size_t)len);
}
}
void rxml_init_io(void)
{
READ_METHOD = rb_intern("read");
WRITE_METHOD = rb_intern("write");
}
|
