diff options
author | shyouhei <shyouhei@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2008-07-07 07:38:25 +0000 |
---|---|---|
committer | shyouhei <shyouhei@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2008-07-07 07:38:25 +0000 |
commit | 9ff1e787f915539b1980654e3d3d2013ff5c81d2 (patch) | |
tree | 8d0fc9ca5b4dbfa9885dc56862292d55091bcaac /ruby_1_8_6/ext/stringio/stringio.c | |
parent | 441546edcfbb1b346c87b69c5f578d1a0e522e06 (diff) |
wrong commit; sorryv1_8_6_269
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/tags/v1_8_6_269@17938 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
Diffstat (limited to 'ruby_1_8_6/ext/stringio/stringio.c')
-rw-r--r-- | ruby_1_8_6/ext/stringio/stringio.c | 1320 |
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); -} |