diff options
author | shyouhei <shyouhei@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2007-08-22 01:53:51 +0000 |
---|---|---|
committer | shyouhei <shyouhei@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2007-08-22 01:53:51 +0000 |
commit | 1e760c0be3ed35874204114e7454509f740c0fe2 (patch) | |
tree | a75eb2b1eea073830902d1fa49c568c4525c8b57 /ruby_1_8_6/lib/readbytes.rb | |
parent | a2055d63b41a6678dc7aeb17d0bece314e700c5a (diff) |
add tag v1_8_6_71v1_8_5_71
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/tags/v1_8_5_71@13189 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
Diffstat (limited to 'ruby_1_8_6/lib/readbytes.rb')
-rw-r--r-- | ruby_1_8_6/lib/readbytes.rb | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/ruby_1_8_6/lib/readbytes.rb b/ruby_1_8_6/lib/readbytes.rb new file mode 100644 index 0000000000..ebb5fc0bd4 --- /dev/null +++ b/ruby_1_8_6/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 |