diff options
author | shyouhei <shyouhei@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2007-08-15 19:08:43 +0000 |
---|---|---|
committer | shyouhei <shyouhei@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2007-08-15 19:08:43 +0000 |
commit | d464704f111d211c1f1ff9ef23ef1d755054be00 (patch) | |
tree | b58b17b645dc463322e5fca57fe282360db659c9 /ruby_1_8_5/lib/readbytes.rb | |
parent | e4f06b3f2dec4b5d6334c5e9907e1cecbf649fc4 (diff) |
add tag v1_8_5_54
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/tags/v1_8_5_54@12952 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
Diffstat (limited to 'ruby_1_8_5/lib/readbytes.rb')
-rw-r--r-- | ruby_1_8_5/lib/readbytes.rb | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/ruby_1_8_5/lib/readbytes.rb b/ruby_1_8_5/lib/readbytes.rb new file mode 100644 index 0000000000..ebb5fc0bd4 --- /dev/null +++ b/ruby_1_8_5/lib/readbytes.rb @@ -0,0 +1,41 @@ +# TruncatedDataError is raised when IO#readbytes fails to read enough data. + +class TruncatedDataError<IOError + def initialize(mesg, data) # :nodoc: + @data = data + super(mesg) + end + + # The read portion of an IO#readbytes attempt. + attr_reader :data +end + +class IO + # Reads exactly +n+ bytes. + # + # If the data read is nil an EOFError is raised. + # + # If the data read is too short a TruncatedDataError is raised and the read + # data is obtainable via its #data method. + def readbytes(n) + str = read(n) + if str == nil + raise EOFError, "End of file reached" + end + if str.size < n + raise TruncatedDataError.new("data truncated", str) + end + str + end +end + +if __FILE__ == $0 + begin + loop do + print STDIN.readbytes(6) + end + rescue TruncatedDataError + p $!.data + raise + end +end |