File: ruby_xml_io.c

package info (click to toggle)
ruby-libxml 2.7.0-3
  • links: PTS, VCS
  • area: main
  • in suites: jessie, jessie-kfreebsd
  • size: 2,024 kB
  • ctags: 2,251
  • sloc: xml: 9,628; ansic: 8,409; ruby: 7,806; makefile: 3
file content (50 lines) | stat: -rw-r--r-- 1,228 bytes parent folder | download
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
48
49
50
/* Please see the LICENSE file for copyright and distribution information */

#include "extconf.h"
#include "ruby_libxml.h"

static ID READ_METHOD;
#ifndef HAVE_RB_IO_BUFWRITE
static ID WRITE_METHOD;
#endif /* !HAVE_RB_IO_BUFWRITE */

/* 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));
  int size;

  if (string == Qnil)
    return 0;

  size = RSTRING_LEN(string);
  memcpy(buffer, StringValuePtr(string), size);

  return size;
}

int rxml_write_callback(void *context, const char *buffer, int len)
{
#ifndef HAVE_RB_IO_BUFWRITE
  VALUE io, written, string;

  io = (VALUE) context;
  string = rb_str_new(buffer, len);
  written = rb_funcall(io, WRITE_METHOD, 1, string);

  return NUM2INT(written);
#else
  return rb_io_bufwrite((VALUE) context, buffer, (size_t) len);
#endif /* !HAVE_RB_IO_BUFWRITE */
}

void rxml_init_io(void)
{
  READ_METHOD = rb_intern("read");
#ifndef HAVE_RB_IO_BUFWRITE
  WRITE_METHOD = rb_intern("write");
#endif /* !HAVE_RB_IO_BUFWRITE */
}