Contents

#
# Copyright (C) 2002-2004 Satoru Takabayashi <[e-mail]> 
# Copyright (C) 2003-2006 Kouichirou Eto
#     All rights reserved.
#     This is free software with ABSOLUTELY NO WARRANTY.
#
# You can redistribute it and/or modify it under the terms of 
# the GNU General Public License version 2.
#

$LOAD_PATH.unshift '..' unless $LOAD_PATH.include? '..'

#class TooLongLine < StandardError; end
class TooLongLine < Exception; end	# FIXME: should be StandardError.

module SafeGetsModule
  def safe_gets (max_length = 1024)
    s = ''
    while ! self.eof?
      c = self.read(1)		# FIXME: This code may be slow.
      s << c

      if max_length < s.length
	raise TooLongLine
      end

      if c == "\n"
	return s
      end
    end

    res = if s.empty? then nil else s end
    return res
  end
end

class IO
  include SafeGetsModule
end

if $0 == __FILE__
  require 'test/unit'
  require 'stringio'
  $test = true
end

if defined?($test) && $test
  class StringIO
    include SafeGetsModule
  end

  class TestUtilSafe < Test::Unit::TestCase
    def test_safe_gets
      # under max
      line = 'a' * 1024
      assert_equal line, StringIO.new(line).gets
      assert_equal line, StringIO.new(line).safe_gets

      # over max
      line = 'a' * 1025
      assert_equal line, StringIO.new(line).gets
      assert_raise(TooLongLine) {
	assert_equal line, StringIO.new(line).safe_gets
      }
    end
  end
end

Version data entries

7 entries across 7 versions & 2 rubygems

Version	Path
qwik2md-2.0.1	vendor/qwik/lib/qwik/util-safe.rb
qwik2md-2.0.0	vendor/qwik/lib/qwik/util-safe.rb
qwik2md-1.0.2	vendor/qwik/lib/qwik/util-safe.rb
qwik2md-1.0.1	vendor/qwik/lib/qwik/util-safe.rb
qwik2md-1.0.0	vendor/qwik/lib/qwik/util-safe.rb
qwik2md-0.1.0	vendor/qwik/lib/qwik/util-safe.rb
qwikdoc-0.0.1	vendor/qwik/util-safe.rb