mirror of
https://github.com/ruby/ruby.git
synced 2022-11-09 12:17:21 -05:00
17c3d539f0
may not be defined to shut up gcc's -Wundef warnings. [ruby-core:08447] git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/branches/ruby_1_8@10648 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
41 lines
829 B
Ruby
41 lines
829 B
Ruby
# readbytes.rb
|
|
#
|
|
# add IO#readbytes, which reads fixed sized data.
|
|
# it guarantees read data size.
|
|
|
|
class TruncatedDataError<IOError
|
|
def initialize(mesg, data)
|
|
@data = data
|
|
super(mesg)
|
|
end
|
|
attr_reader :data
|
|
end
|
|
|
|
class IO
|
|
# reads exactly n bytes from the IO stream.
|
|
# If the data read is nil, raises EOFError.
|
|
# If the data read is too short, raises TruncatedDataError.
|
|
# The method TruncatedDataError#data may be used to obtain
|
|
# the truncated message.
|
|
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
|