summaryrefslogtreecommitdiff
path: root/ruby_1_8_6/ext/stringio/stringio.c
diff options
context:
space:
mode:
Diffstat (limited to 'ruby_1_8_6/ext/stringio/stringio.c')
-rw-r--r--ruby_1_8_6/ext/stringio/stringio.c1320
1 files changed, 0 insertions, 1320 deletions
diff --git a/ruby_1_8_6/ext/stringio/stringio.c b/ruby_1_8_6/ext/stringio/stringio.c
deleted file mode 100644
index 86ff56534c..0000000000
--- a/ruby_1_8_6/ext/stringio/stringio.c
+++ /dev/null
@@ -1,1320 +0,0 @@
-/**********************************************************************
-
- stringio.c -
-
- $Author$
- $Date$
- $RoughId: stringio.c,v 1.13 2002/03/14 03:24:18 nobu Exp $
- created at: Tue Feb 19 04:10:38 JST 2002
-
- All the files in this distribution are covered under the Ruby's
- license (see the file COPYING).
-
-**********************************************************************/
-
-#include "ruby.h"
-#include "rubyio.h"
-#if defined(HAVE_FCNTL_H) || defined(_WIN32)
-#include <fcntl.h>
-#elif defined(HAVE_SYS_FCNTL_H)
-#include <sys/fcntl.h>
-#endif
-
-#define STRIO_EOF FMODE_SYNC
-
-struct StringIO {
- VALUE string;
- long pos;
- long lineno;
- int flags;
- int count;
-};
-
-static struct StringIO* strio_alloc _((void));
-static void strio_mark _((struct StringIO *));
-static void strio_free _((struct StringIO *));
-static struct StringIO* check_strio _((VALUE));
-static struct StringIO* get_strio _((VALUE));
-static struct StringIO* readable _((struct StringIO *));
-static struct StringIO* writable _((struct StringIO *));
-static void check_modifiable _((struct StringIO *));
-
-#define IS_STRIO(obj) (RDATA(obj)->dmark == (RUBY_DATA_FUNC)strio_mark)
-#define error_inval(msg) (errno = EINVAL, rb_sys_fail(msg))
-
-static struct StringIO *
-strio_alloc()
-{
- struct StringIO *ptr = ALLOC(struct StringIO);
- ptr->string = Qnil;
- ptr->pos = 0;
- ptr->lineno = 0;
- ptr->flags = 0;
- ptr->count = 1;
- return ptr;
-}
-
-static void
-strio_mark(ptr)
- struct StringIO *ptr;
-{
- if (ptr) {
- rb_gc_mark(ptr->string);
- }
-}
-
-static void
-strio_free(ptr)
- struct StringIO *ptr;
-{
- if (--ptr->count <= 0) {
- xfree(ptr);
- }
-}
-
-static struct StringIO*
-check_strio(self)
- VALUE self;
-{
- Check_Type(self, T_DATA);
- if (!IS_STRIO(self)) {
- rb_raise(rb_eTypeError, "wrong argument type %s (expected StringIO)",
- rb_class2name(CLASS_OF(self)));
- }
- return DATA_PTR(self);
-}
-
-static struct StringIO*
-get_strio(self)
- VALUE self;
-{
- struct StringIO *ptr = check_strio(self);
-
- if (!ptr) {
- rb_raise(rb_eIOError, "uninitialized stream");
- }
- return ptr;
-}
-
-#define StringIO(obj) get_strio(obj)
-
-#define CLOSED(ptr) (!((ptr)->flags & FMODE_READWRITE))
-#define READABLE(ptr) ((ptr)->flags & FMODE_READABLE)
-#define WRITABLE(ptr) ((ptr)->flags & FMODE_WRITABLE)
-
-static struct StringIO*
-readable(ptr)
- struct StringIO *ptr;
-{
- if (!READABLE(ptr)) {
- rb_raise(rb_eIOError, "not opened for reading");
- }
- return ptr;
-}
-
-static struct StringIO*
-writable(ptr)
- struct StringIO *ptr;
-{
- if (!WRITABLE(ptr)) {
- rb_raise(rb_eIOError, "not opened for writing");
- }
- if (!OBJ_TAINTED(ptr->string)) {
- rb_secure(4);
- }
- return ptr;
-}
-
-static void
-check_modifiable(ptr)
- struct StringIO *ptr;
-{
- if (OBJ_FROZEN(ptr->string)) {
- rb_raise(rb_eIOError, "not modifiable string");
- }
-}
-
-static VALUE strio_s_allocate _((VALUE));
-static VALUE strio_s_open _((int, VALUE *, VALUE));
-static void strio_init _((int, VALUE *, struct StringIO *));
-static VALUE strio_initialize _((int, VALUE *, VALUE));
-static VALUE strio_finalize _((VALUE));
-static VALUE strio_self _((VALUE));
-static VALUE strio_false _((VALUE));
-static VALUE strio_nil _((VALUE));
-static VALUE strio_0 _((VALUE));
-static VALUE strio_first _((VALUE, VALUE));
-static VALUE strio_unimpl _((int, VALUE *, VALUE));
-static VALUE strio_get_string _((VALUE));
-static VALUE strio_set_string _((VALUE, VALUE));
-static VALUE strio_close _((VALUE));
-static VALUE strio_close_read _((VALUE));
-static VALUE strio_close_write _((VALUE));
-static VALUE strio_closed _((VALUE));
-static VALUE strio_closed_read _((VALUE));
-static VALUE strio_closed_write _((VALUE));
-static VALUE strio_eof _((VALUE));
-static VALUE strio_get_lineno _((VALUE));
-static VALUE strio_set_lineno _((VALUE, VALUE));
-static VALUE strio_get_pos _((VALUE));
-static VALUE strio_set_pos _((VALUE, VALUE));
-static VALUE strio_rewind _((VALUE));
-static VALUE strio_seek _((int, VALUE *, VALUE));
-static VALUE strio_get_sync _((VALUE));
-static VALUE strio_each_byte _((VALUE));
-static VALUE strio_getc _((VALUE));
-static VALUE strio_ungetc _((VALUE, VALUE));
-static VALUE strio_readchar _((VALUE));
-static VALUE strio_getline _((int, VALUE *, struct StringIO *));
-static VALUE strio_gets _((int, VALUE *, VALUE));
-static VALUE strio_readline _((int, VALUE *, VALUE));
-static VALUE strio_each _((int, VALUE *, VALUE));
-static VALUE strio_readlines _((int, VALUE *, VALUE));
-static VALUE strio_write _((VALUE, VALUE));
-static VALUE strio_putc _((VALUE, VALUE));
-static VALUE strio_read _((int, VALUE *, VALUE));
-static VALUE strio_size _((VALUE));
-static VALUE strio_truncate _((VALUE, VALUE));
-void Init_stringio _((void));
-
-/* Boyer-Moore search: copied from regex.c */
-static void bm_init_skip _((long *, const char *, long));
-static long bm_search _((const char *, long, const char *, long, const long *));
-
-static VALUE
-strio_s_allocate(klass)
- VALUE klass;
-{
- return Data_Wrap_Struct(klass, strio_mark, strio_free, 0);
-}
-
-/*
- * call-seq: StringIO.open(string=""[, mode]) {|strio| ...}
- *
- * Equivalent to StringIO.new except that when it is called with a block, it
- * yields with the new instance and closes it, and returns the result which
- * returned from the block.
- */
-static VALUE
-strio_s_open(argc, argv, klass)
- int argc;
- VALUE *argv;
- VALUE klass;
-{
- VALUE obj = rb_class_new_instance(argc, argv, klass);
- if (!rb_block_given_p()) return obj;
- return rb_ensure(rb_yield, obj, strio_finalize, obj);
-}
-
-/*
- * call-seq: StringIO.new(string=""[, mode])
- *
- * Creates new StringIO instance from with _string_ and _mode_.
- */
-static VALUE
-strio_initialize(argc, argv, self)
- int argc;
- VALUE *argv;
- VALUE self;
-{
- struct StringIO *ptr = check_strio(self);
-
- if (!ptr) {
- DATA_PTR(self) = ptr = strio_alloc();
- }
- rb_call_super(0, 0);
- strio_init(argc, argv, ptr);
- return self;
-}
-
-static void
-strio_init(argc, argv, ptr)
- int argc;
- VALUE *argv;
- struct StringIO *ptr;
-{
- VALUE string, mode;
- int trunc = Qfalse;
-
- switch (rb_scan_args(argc, argv, "02", &string, &mode)) {
- case 2:
- if (FIXNUM_P(mode)) {
- int flags = FIX2INT(mode);
- ptr->flags = rb_io_modenum_flags(flags);
- trunc = flags & O_TRUNC;
- }
- else {
- const char *m = StringValueCStr(mode);
- ptr->flags = rb_io_mode_flags(m);
- trunc = *m == 'w';
- }
- StringValue(string);
- if ((ptr->flags & FMODE_WRITABLE) && OBJ_FROZEN(string)) {
- errno = EACCES;
- rb_sys_fail(0);
- }
- if (trunc) {
- rb_str_resize(string, 0);
- }
- break;
- case 1:
- StringValue(string);
- ptr->flags = OBJ_FROZEN(string) ? FMODE_READABLE : FMODE_READWRITE;
- break;
- case 0:
- string = rb_str_new("", 0);
- ptr->flags = FMODE_READWRITE;
- break;
- }
- ptr->string = string;
-}
-
-static VALUE
-strio_finalize(self)
- VALUE self;
-{
- struct StringIO *ptr = StringIO(self);
- ptr->string = Qnil;
- ptr->flags &= ~FMODE_READWRITE;
- return self;
-}
-
-/*
- * Returns +false+. Just for compatibility to IO.
- */
-static VALUE
-strio_false(self)
- VALUE self;
-{
- StringIO(self);
- return Qfalse;
-}
-
-/*
- * Returns +nil+. Just for compatibility to IO.
- */
-static VALUE
-strio_nil(self)
- VALUE self;
-{
- StringIO(self);
- return Qnil;
-}
-
-/*
- * Returns *strio* itself. Just for compatibility to IO.
- */
-static VALUE
-strio_self(self)
- VALUE self;
-{
- StringIO(self);
- return self;
-}
-
-/*
- * Returns 0. Just for compatibility to IO.
- */
-static VALUE
-strio_0(self)
- VALUE self;
-{
- StringIO(self);
- return INT2FIX(0);
-}
-
-/*
- * Returns the argument unchanged. Just for compatibility to IO.
- */
-static VALUE
-strio_first(self, arg)
- VALUE self, arg;
-{
- StringIO(self);
- return arg;
-}
-
-/*
- * Raises NotImplementedError.
- */
-static VALUE
-strio_unimpl(argc, argv, self)
- int argc;
- VALUE *argv;
- VALUE self;
-{
- StringIO(self);
- rb_notimplement();
- return Qnil; /* not reached */
-}
-
-/*
- * call-seq: strio.string -> string
- *
- * Returns underlying String object, the subject of IO.
- */
-static VALUE
-strio_get_string(self)
- VALUE self;
-{
- return StringIO(self)->string;
-}
-
-/*
- * call-seq:
- * strio.string = string -> string
- *
- * Changes underlying String object, the subject of IO.
- */
-static VALUE
-strio_set_string(self, string)
- VALUE self, string;
-{
- struct StringIO *ptr = StringIO(self);
-
- if (!OBJ_TAINTED(self)) rb_secure(4);
- ptr->flags &= ~FMODE_READWRITE;
- StringValue(string);
- ptr->flags = OBJ_FROZEN(string) ? FMODE_READABLE : FMODE_READWRITE;
- ptr->pos = 0;
- ptr->lineno = 0;
- return ptr->string = string;
-}
-
-/*
- * call-seq:
- * strio.close -> nil
- *
- * Closes strio. The *strio* is unavailable for any further data
- * operations; an +IOError+ is raised if such an attempt is made.
- */
-static VALUE
-strio_close(self)
- VALUE self;
-{
- struct StringIO *ptr = StringIO(self);
- if (CLOSED(ptr)) {
- rb_raise(rb_eIOError, "closed stream");
- }
- ptr->flags &= ~FMODE_READWRITE;
- return Qnil;
-}
-
-/*
- * call-seq:
- * strio.close_read -> nil
- *
- * Closes the read end of a StringIO. Will raise an +IOError+ if the
- * *strio* is not readable.
- */
-static VALUE
-strio_close_read(self)
- VALUE self;
-{
- struct StringIO *ptr = StringIO(self);
- if (!READABLE(ptr)) {
- rb_raise(rb_eIOError, "closing non-duplex IO for reading");
- }
- ptr->flags &= ~FMODE_READABLE;
- return Qnil;
-}
-
-/*
- * call-seq:
- * strio.close_write -> nil
- *
- * Closes the write end of a StringIO. Will raise an +IOError+ if the
- * *strio* is not writeable.
- */
-static VALUE
-strio_close_write(self)
- VALUE self;
-{
- struct StringIO *ptr = StringIO(self);
- if (!WRITABLE(ptr)) {
- rb_raise(rb_eIOError, "closing non-duplex IO for writing");
- }
- ptr->flags &= ~FMODE_WRITABLE;
- return Qnil;
-}
-
-/*
- * call-seq:
- * strio.closed? -> true or false
- *
- * Returns +true+ if *strio* is completely closed, +false+ otherwise.
- */
-static VALUE
-strio_closed(self)
- VALUE self;
-{
- struct StringIO *ptr = StringIO(self);
- if (!CLOSED(ptr)) return Qfalse;
- return Qtrue;
-}
-
-/*
- * call-seq:
- * strio.closed_read? -> true or false
- *
- * Returns +true+ if *strio* is not readable, +false+ otherwise.
- */
-static VALUE
-strio_closed_read(self)
- VALUE self;
-{
- struct StringIO *ptr = StringIO(self);
- if (READABLE(ptr)) return Qfalse;
- return Qtrue;
-}
-
-/*
- * call-seq:
- * strio.closed_write? -> true or false
- *
- * Returns +true+ if *strio* is not writable, +false+ otherwise.
- */
-static VALUE
-strio_closed_write(self)
- VALUE self;
-{
- struct StringIO *ptr = StringIO(self);
- if (WRITABLE(ptr)) return Qfalse;
- return Qtrue;
-}
-
-/*
- * call-seq:
- * strio.eof -> true or false
- * strio.eof? -> true or false
- *
- * Returns true if *strio* is at end of file. The stringio must be
- * opened for reading or an +IOError+ will be raised.
- */
-static VALUE
-strio_eof(self)
- VALUE self;
-{
- struct StringIO *ptr = readable(StringIO(self));
- if (ptr->pos < RSTRING(ptr->string)->len) return Qfalse;
- return Qtrue;
-}
-
-/* :nodoc: */
-static VALUE
-strio_copy(copy, orig)
- VALUE copy, orig;
-{
- struct StringIO *ptr;
-
- orig = rb_convert_type(orig, T_DATA, "StringIO", "to_strio");
- if (copy == orig) return copy;
- ptr = StringIO(orig);
- if (check_strio(copy)) {
- strio_free(DATA_PTR(copy));
- }
- DATA_PTR(copy) = ptr;
- OBJ_INFECT(copy, orig);
- ++ptr->count;
- return copy;
-}
-
-/*
- * call-seq:
- * strio.lineno -> integer
- *
- * Returns the current line number in *strio*. The stringio must be
- * opened for reading. +lineno+ counts the number of times +gets+ is
- * called, rather than the number of newlines encountered. The two
- * values will differ if +gets+ is called with a separator other than
- * newline. See also the <code>$.</code> variable.
- */
-static VALUE
-strio_get_lineno(self)
- VALUE self;
-{
- return LONG2NUM(StringIO(self)->lineno);
-}
-
-/*
- * call-seq:
- * strio.lineno = integer -> integer
- *
- * Manually sets the current line number to the given value.
- * <code>$.</code> is updated only on the next read.
- */
-static VALUE
-strio_set_lineno(self, lineno)
- VALUE self, lineno;
-{
- StringIO(self)->lineno = NUM2LONG(lineno);
- return lineno;
-}
-
-/* call-seq: strio.binmode -> true */
-#define strio_binmode strio_self
-
-/* call-seq: strio.fcntl */
-#define strio_fcntl strio_unimpl
-
-/* call-seq: strio.flush -> strio */
-#define strio_flush strio_self
-
-/* call-seq: strio.fsync -> 0 */
-#define strio_fsync strio_0
-
-/*
- * call-seq:
- * strio.reopen(other_StrIO) -> strio
- * strio.reopen(string, mode) -> strio
- *
- * Reinitializes *strio* with the given <i>other_StrIO</i> or _string_
- * and _mode_ (see StringIO#new).
- */
-static VALUE
-strio_reopen(argc, argv, self)
- int argc;
- VALUE *argv;
- VALUE self;
-{
- if (!OBJ_TAINTED(self)) rb_secure(4);
- if (argc == 1 && TYPE(*argv) != T_STRING) {
- return strio_copy(self, *argv);
- }
- strio_init(argc, argv, StringIO(self));
- return self;
-}
-
-/*
- * call-seq:
- * strio.pos -> integer
- * strio.tell -> integer
- *
- * Returns the current offset (in bytes) of *strio*.
- */
-static VALUE
-strio_get_pos(self)
- VALUE self;
-{
- return LONG2NUM(StringIO(self)->pos);
-}
-
-/*
- * call-seq:
- * strio.pos = integer -> integer
- *
- * Seeks to the given position (in bytes) in *strio*.
- */
-static VALUE
-strio_set_pos(self, pos)
- VALUE self;
- VALUE pos;
-{
- struct StringIO *ptr = StringIO(self);
- long p = NUM2LONG(pos);
- if (p < 0) {
- error_inval(0);
- }
- ptr->pos = p;
- ptr->flags &= ~STRIO_EOF;
- return pos;
-}
-
-/*
- * call-seq:
- * strio.rewind -> 0
- *
- * Positions *strio* to the beginning of input, resetting
- * +lineno+ to zero.
- */
-static VALUE
-strio_rewind(self)
- VALUE self;
-{
- struct StringIO *ptr = StringIO(self);
- ptr->pos = 0;
- ptr->lineno = 0;
- ptr->flags &= ~STRIO_EOF;
- return INT2FIX(0);
-}
-
-/*
- * call-seq:
- * strio.seek(amount, whence=SEEK_SET) -> 0
- *
- * Seeks to a given offset _amount_ in the stream according to
- * the value of _whence_ (see IO#seek).
- */
-static VALUE
-strio_seek(argc, argv, self)
- int argc;
- VALUE *argv;
- VALUE self;
-{
- VALUE whence;
- struct StringIO *ptr = StringIO(self);
- long offset;
-
- rb_scan_args(argc, argv, "11", NULL, &whence);
- offset = NUM2LONG(argv[0]);
- switch (NIL_P(whence) ? 0 : NUM2LONG(whence)) {
- case 0:
- break;
- case 1:
- offset += ptr->pos;
- break;
- case 2:
- offset += RSTRING(ptr->string)->len;
- break;
- default:
- rb_raise(rb_eArgError, "invalid whence %ld", NUM2LONG(whence));
- }
- if (offset < 0) {
- error_inval(0);
- }
- ptr->pos = offset;
- ptr->flags &= ~STRIO_EOF;
- return INT2FIX(0);
-}
-
-/*
- * call-seq:
- * strio.sync -> true
- *
- * Returns +true+ always.
- */
-static VALUE
-strio_get_sync(self)
- VALUE self;
-{
- StringIO(self);
- return Qtrue;
-}
-
-/* call-seq: strio.sync = boolean -> boolean */
-#define strio_set_sync strio_first
-
-#define strio_tell strio_get_pos
-
-/*
- * call-seq:
- * strio.each_byte {|byte| block } -> strio
- *
- * See IO#each_byte.
- */
-static VALUE
-strio_each_byte(self)
- VALUE self;
-{
- struct StringIO *ptr = readable(StringIO(self));
- while (ptr->pos < RSTRING(ptr->string)->len) {
- char c = RSTRING(ptr->string)->ptr[ptr->pos++];
- rb_yield(CHR2FIX(c));
- }
- return Qnil;
-}
-
-/*
- * call-seq:
- * strio.getc -> fixnum or nil
- *
- * See IO#getc.
- */
-static VALUE
-strio_getc(self)
- VALUE self;
-{
- struct StringIO *ptr = readable(StringIO(self));
- int c;
- if (ptr->pos >= RSTRING(ptr->string)->len) {
- ptr->flags |= STRIO_EOF;
- return Qnil;
- }
- c = RSTRING(ptr->string)->ptr[ptr->pos++];
- return CHR2FIX(c);
-}
-
-static void
-strio_extend(ptr, pos, len)
- struct StringIO *ptr;
- long pos, len;
-{
- long olen;
-
- check_modifiable(ptr);
- olen = RSTRING(ptr->string)->len;
- if (pos + len > olen) {
- rb_str_resize(ptr->string, pos + len);
- if (pos > olen)
- MEMZERO(RSTRING(ptr->string)->ptr + olen, char, pos - olen);
- }
- else {
- rb_str_modify(ptr->string);
- }
-}
-
-/*
- * call-seq:
- * strio.ungetc(integer) -> nil
- *
- * Pushes back one character (passed as a parameter) onto *strio*
- * such that a subsequent buffered read will return it. Pushing back
- * behind the beginning of the buffer string is not possible. Nothing
- * will be done if such an attempt is made.
- * In other case, there is no limitation for multiple pushbacks.
- */
-static VALUE
-strio_ungetc(self, ch)
- VALUE self, ch;
-{
- struct StringIO *ptr = readable(StringIO(self));
- int cc = NUM2INT(ch);
- long len, pos = ptr->pos;
-
- if (cc != EOF && pos > 0) {
- if ((len = RSTRING(ptr->string)->len) < pos-- ||
- (unsigned char)RSTRING(ptr->string)->ptr[pos] !=
- (unsigned char)cc) {
- strio_extend(ptr, pos, 1);
- RSTRING(ptr->string)->ptr[pos] = cc;
- OBJ_INFECT(ptr->string, self);
- }
- --ptr->pos;
- ptr->flags &= ~STRIO_EOF;
- }
- return Qnil;
-}
-
-/*
- * call-seq:
- * strio.readchar -> fixnum
- *
- * See IO#readchar.
- */
-static VALUE
-strio_readchar(self)
- VALUE self;
-{
- VALUE c = strio_getc(self);
- if (NIL_P(c)) rb_eof_error();
- return c;
-}
-
-static void
-bm_init_skip(skip, pat, m)
- long *skip;
- const char *pat;
- long m;
-{
- int c;
-
- for (c = 0; c < (1 << CHAR_BIT); c++) {
- skip[c] = m;
- }
- while (--m) {
- skip[(unsigned char)*pat++] = m;
- }
-}
-
-static long
-bm_search(little, llen, big, blen, skip)
- const char *little;
- long llen;
- const char *big;
- long blen;
- const long *skip;
-{
- long i, j, k;
-
- i = llen - 1;
- while (i < blen) {
- k = i;
- j = llen - 1;
- while (j >= 0 && big[k] == little[j]) {
- k--;
- j--;
- }
- if (j < 0) return k + 1;
- i += skip[(unsigned char)big[i]];
- }
- return -1;
-}
-
-static VALUE
-strio_getline(argc, argv, ptr)
- int argc;
- VALUE *argv;
- struct StringIO *ptr;
-{
- const char *s, *e, *p;
- long n;
- VALUE str;
-
- if (argc == 0) {
- str = rb_rs;
- }
- else {
- rb_scan_args(argc, argv, "1", &str);
- if (!NIL_P(str)) StringValue(str);
- }
-
- if (ptr->pos >= (n = RSTRING(ptr->string)->len)) {
- ptr->flags |= STRIO_EOF;
- return Qnil;
- }
- s = RSTRING(ptr->string)->ptr;
- e = s + RSTRING(ptr->string)->len;
- s += ptr->pos;
- if (NIL_P(str)) {
- str = rb_str_substr(ptr->string, ptr->pos, e - s);
- }
- else if ((n = RSTRING(str)->len) == 0) {
- p = s;
- while (*p == '\n') {
- if (++p == e) {
- ptr->flags |= STRIO_EOF;
- return Qnil;
- }
- }
- s = p;
- while ((p = memchr(p, '\n', e - p)) && (p != e)) {
- if (*++p == '\n') {
- e = p;
- break;
- }
- }
- str = rb_str_substr(ptr->string, s - RSTRING(ptr->string)->ptr, e - s);
- }
- else if (n == 1) {
- if ((p = memchr(s, RSTRING(str)->ptr[0], e - s)) != 0) {
- e = p + 1;
- }
- str = rb_str_substr(ptr->string, ptr->pos, e - s);
- }
- else {
- if (n < e - s) {
- if (e - s < 1024) {
- for (p = s; p + n <= e; ++p) {
- if (MEMCMP(p, RSTRING(str)->ptr, char, n) == 0) {
- e = p + n;
- break;
- }
- }
- }
- else {
- long skip[1 << CHAR_BIT], pos;
- p = RSTRING(str)->ptr;
- bm_init_skip(skip, p, n);
- if ((pos = bm_search(p, n, s, e - s, skip)) >= 0) {
- e = s + pos + n;
- }
- }
- }
- str = rb_str_substr(ptr->string, ptr->pos, e - s);
- }
- ptr->pos = e - RSTRING(ptr->string)->ptr;
- ptr->lineno++;
- return str;
-}
-
-/*
- * call-seq:
- * strio.gets(sep_string=$/) -> string or nil
- *
- * See IO#gets.
- */
-static VALUE
-strio_gets(argc, argv, self)
- int argc;
- VALUE *argv;
- VALUE self;
-{
- VALUE str = strio_getline(argc, argv, readable(StringIO(self)));
-
- rb_lastline_set(str);
- return str;
-}
-
-/*
- * call-seq:
- * strio.readline(sep_string=$/) -> string
- *
- * See IO#readline.
- */
-static VALUE
-strio_readline(argc, argv, self)
- int argc;
- VALUE *argv;
- VALUE self;
-{
- VALUE line = strio_gets(argc, argv, self);
- if (NIL_P(line)) rb_eof_error();
- return line;
-}
-
-/*
- * call-seq:
- * strio.each(sep_string=$/) {|line| block } -> strio
- * strio.each_line(sep_string=$/) {|line| block } -> strio
- *
- * See IO#each.
- */
-static VALUE
-strio_each(argc, argv, self)
- int argc;
- VALUE *argv;
- VALUE self;
-{
- struct StringIO *ptr = StringIO(self);
- VALUE line;
-
- while (!NIL_P(line = strio_getline(argc, argv, readable(ptr)))) {
- rb_yield(line);
- }
- return self;
-}
-
-/*
- * call-seq:
- * strio.readlines(sep_string=$/) -> array
- *
- * See IO#readlines.
- */
-static VALUE
-strio_readlines(argc, argv, self)
- int argc;
- VALUE *argv;
- VALUE self;
-{
- struct StringIO *ptr = StringIO(self);
- VALUE ary = rb_ary_new(), line;
- while (!NIL_P(line = strio_getline(argc, argv, readable(ptr)))) {
- rb_ary_push(ary, line);
- }
- return ary;
-}
-
-/*
- * call-seq:
- * strio.write(string) -> integer
- * strio.syswrite(string) -> integer
- *
- * Appends the given string to the underlying buffer string of *strio*.
- * The stream must be opened for writing. If the argument is not a
- * string, it will be converted to a string using <code>to_s</code>.
- * Returns the number of bytes written. See IO#write.
- */
-static VALUE
-strio_write(self, str)
- VALUE self, str;
-{
- struct StringIO *ptr = writable(StringIO(self));
- long len, olen;
-
- if (TYPE(str) != T_STRING)
- str = rb_obj_as_string(str);
- len = RSTRING(str)->len;
- if (!len) return INT2FIX(0);
- check_modifiable(ptr);
- olen = RSTRING(ptr->string)->len;
- if (ptr->flags & FMODE_APPEND) {
- ptr->pos = olen;
- }
- if (ptr->pos == olen) {
- rb_str_cat(ptr->string, RSTRING(str)->ptr, len);
- }
- else {
- strio_extend(ptr, ptr->pos, len);
- rb_str_update(ptr->string, ptr->pos, len, str);
- }
- OBJ_INFECT(ptr->string, self);
- ptr->pos += len;
- return LONG2NUM(len);
-}
-
-/*
- * call-seq:
- * strio << obj -> strio
- *
- * See IO#<<.
- */
-#define strio_addstr rb_io_addstr
-
-/*
- * call-seq:
- * strio.print() -> nil
- * strio.print(obj, ...) -> nil
- *
- * See IO#print.
- */
-#define strio_print rb_io_print
-
-/*
- * call-seq:
- * strio.printf(format_string [, obj, ...] ) -> nil
- *
- * See IO#printf.
- */
-#define strio_printf rb_io_printf
-
-/*
- * call-seq:
- * strio.putc(obj) -> obj
- *
- * See IO#putc.
- */
-static VALUE
-strio_putc(self, ch)
- VALUE self, ch;
-{
- struct StringIO *ptr = writable(StringIO(self));
- int c = NUM2CHR(ch);
- long olen;
-
- check_modifiable(ptr);
- olen = RSTRING(ptr->string)->len;
- if (ptr->flags & FMODE_APPEND) {
- ptr->pos = olen;
- }
- strio_extend(ptr, ptr->pos, 1);
- RSTRING(ptr->string)->ptr[ptr->pos++] = c;
- OBJ_INFECT(ptr->string, self);
- return ch;
-}
-
-/*
- * call-seq:
- * strio.puts(obj, ...) -> nil
- *
- * See IO#puts.
- */
-#define strio_puts rb_io_puts
-
-/*
- * call-seq:
- * strio.read([length [, buffer]]) -> string, buffer, or nil
- *
- * See IO#read.
- */
-static VALUE
-strio_read(argc, argv, self)
- int argc;
- VALUE *argv;
- VALUE self;
-{
- struct StringIO *ptr = readable(StringIO(self));
- VALUE str = Qnil;
- long len, olen;
-
- switch (argc) {
- case 2:
- str = argv[1];
- StringValue(str);
- rb_str_modify(str);
- case 1:
- if (!NIL_P(argv[0])) {
- len = olen = NUM2LONG(argv[0]);
- if (len < 0) {
- rb_raise(rb_eArgError, "negative length %ld given", len);
- }
- if (len > 0 && ptr->pos >= RSTRING(ptr->string)->len) {
- ptr->flags |= STRIO_EOF;
- if (!NIL_P(str)) rb_str_resize(str, 0);
- return Qnil;
- }
- else if (ptr->flags & STRIO_EOF) {
- if (!NIL_P(str)) rb_str_resize(str, 0);
- return Qnil;
- }
- break;
- }
- /* fall through */
- case 0:
- olen = -1;
- len = RSTRING(ptr->string)->len;
- if (len <= ptr->pos) {
- ptr->flags |= STRIO_EOF;
- if (NIL_P(str)) {
- str = rb_str_new(0, 0);
- }
- else {
- rb_str_resize(str, 0);
- }
- return str;
- }
- else {
- len -= ptr->pos;
- }
- break;
- default:
- rb_raise(rb_eArgError, "wrong number of arguments (%d for 0)", argc);
- }
- if (NIL_P(str)) {
- str = rb_str_substr(ptr->string, ptr->pos, len);
- }
- else {
- long rest = RSTRING(ptr->string)->len - ptr->pos;
- if (len > rest) len = rest;
- rb_str_resize(str, len);
- MEMCPY(RSTRING(str)->ptr, RSTRING(ptr->string)->ptr + ptr->pos, char, len);
- }
- if (NIL_P(str)) {
- if (!(ptr->flags & STRIO_EOF)) str = rb_str_new(0, 0);
- len = 0;
- }
- else {
- ptr->pos += len = RSTRING(str)->len;
- }
- if (olen < 0 || olen > len) ptr->flags |= STRIO_EOF;
- return str;
-}
-
-/*
- * call-seq:
- * strio.sysread(integer[, outbuf]) -> string
- *
- * Similar to #read, but raises +EOFError+ at end of string instead of
- * returning +nil+, as well as IO#sysread does.
- */
-static VALUE
-strio_sysread(argc, argv, self)
- int argc;
- VALUE *argv;
- VALUE self;
-{
- VALUE val = strio_read(argc, argv, self);
- if (NIL_P(val) || RSTRING(val)->len == 0) {
- rb_eof_error();
- }
- return val;
-}
-
-#define strio_syswrite strio_write
-
-/* call-seq: strio.path -> nil */
-#define strio_path strio_nil
-
-/*
- * call-seq:
- * strio.isatty -> nil
- * strio.tty? -> nil
- *
- */
-#define strio_isatty strio_false
-
-/* call-seq: strio.pid -> nil */
-#define strio_pid strio_nil
-
-/* call-seq: strio.fileno -> nil */
-#define strio_fileno strio_nil
-
-/*
- * call-seq:
- * strio.size -> integer
- *
- * Returns the size of the buffer string.
- */
-static VALUE
-strio_size(self)
- VALUE self;
-{
- VALUE string = StringIO(self)->string;
- if (NIL_P(string)) {
- rb_raise(rb_eIOError, "not opened");
- }
- return ULONG2NUM(RSTRING(string)->len);
-}
-
-/*
- * call-seq:
- * strio.truncate(integer) -> 0
- *
- * Truncates the buffer string to at most _integer_ bytes. The *strio*
- * must be opened for writing.
- */
-static VALUE
-strio_truncate(self, len)
- VALUE self, len;
-{
- VALUE string = writable(StringIO(self))->string;
- long l = NUM2LONG(len);
- long plen = RSTRING(string)->len;
- if (l < 0) {
- error_inval("negative legnth");
- }
- rb_str_resize(string, l);
- if (plen < l) {
- MEMZERO(RSTRING(string)->ptr + plen, char, l - plen);
- }
- return len;
-}
-
-/*
- * Pseudo I/O on String object.
- */
-void
-Init_stringio()
-{
- VALUE StringIO = rb_define_class("StringIO", rb_cData);
-
- rb_include_module(StringIO, rb_mEnumerable);
- rb_define_alloc_func(StringIO, strio_s_allocate);
- rb_define_singleton_method(StringIO, "open", strio_s_open, -1);
- rb_define_method(StringIO, "initialize", strio_initialize, -1);
- rb_define_method(StringIO, "initialize_copy", strio_copy, 1);
- rb_define_method(StringIO, "reopen", strio_reopen, -1);
-
- rb_define_method(StringIO, "string", strio_get_string, 0);
- rb_define_method(StringIO, "string=", strio_set_string, 1);
- rb_define_method(StringIO, "lineno", strio_get_lineno, 0);
- rb_define_method(StringIO, "lineno=", strio_set_lineno, 1);
-
- rb_define_method(StringIO, "binmode", strio_binmode, 0);
- rb_define_method(StringIO, "close", strio_close, 0);
- rb_define_method(StringIO, "close_read", strio_close_read, 0);
- rb_define_method(StringIO, "close_write", strio_close_write, 0);
- rb_define_method(StringIO, "closed?", strio_closed, 0);
- rb_define_method(StringIO, "closed_read?", strio_closed_read, 0);
- rb_define_method(StringIO, "closed_write?", strio_closed_write, 0);
- rb_define_method(StringIO, "eof", strio_eof, 0);
- rb_define_method(StringIO, "eof?", strio_eof, 0);
- rb_define_method(StringIO, "fcntl", strio_fcntl, -1);
- rb_define_method(StringIO, "flush", strio_flush, 0);
- rb_define_method(StringIO, "fsync", strio_fsync, 0);
- rb_define_method(StringIO, "pos", strio_get_pos, 0);
- rb_define_method(StringIO, "pos=", strio_set_pos, 1);
- rb_define_method(StringIO, "rewind", strio_rewind, 0);
- rb_define_method(StringIO, "seek", strio_seek, -1);
- rb_define_method(StringIO, "sync", strio_get_sync, 0);
- rb_define_method(StringIO, "sync=", strio_set_sync, 1);
- rb_define_method(StringIO, "tell", strio_tell, 0);
- rb_define_method(StringIO, "path", strio_path, 0);
-
- rb_define_method(StringIO, "each", strio_each, -1);
- rb_define_method(StringIO, "each_byte", strio_each_byte, 0);
- rb_define_method(StringIO, "each_line", strio_each, -1);
- rb_define_method(StringIO, "getc", strio_getc, 0);
- rb_define_method(StringIO, "ungetc", strio_ungetc, 1);
- rb_define_method(StringIO, "readchar", strio_readchar, 0);
- rb_define_method(StringIO, "gets", strio_gets, -1);
- rb_define_method(StringIO, "readline", strio_readline, -1);
- rb_define_method(StringIO, "readlines", strio_readlines, -1);
- rb_define_method(StringIO, "read", strio_read, -1);
- rb_define_method(StringIO, "sysread", strio_sysread, -1);
-
- rb_define_method(StringIO, "write", strio_write, 1);
- rb_define_method(StringIO, "<<", strio_addstr, 1);
- rb_define_method(StringIO, "print", strio_print, -1);
- rb_define_method(StringIO, "printf", strio_printf, -1);
- rb_define_method(StringIO, "putc", strio_putc, 1);
- rb_define_method(StringIO, "puts", strio_puts, -1);
- rb_define_method(StringIO, "syswrite", strio_syswrite, 1);
-
- rb_define_method(StringIO, "isatty", strio_isatty, 0);
- rb_define_method(StringIO, "tty?", strio_isatty, 0);
- rb_define_method(StringIO, "pid", strio_pid, 0);
- rb_define_method(StringIO, "fileno", strio_fileno, 0);
- rb_define_method(StringIO, "size", strio_size, 0);
- rb_define_method(StringIO, "length", strio_size, 0);
- rb_define_method(StringIO, "truncate", strio_truncate, 1);
-}
generated by cgit v1.2.3 (git 2.25.1) at 2024-04-30 14:48:47 +0000