Codebase list ruby-maxmind-db / f1313f2 lib / maxmind / db / file_reader.rb
f1313f2

Tree @f1313f2 (Download .tar.gz)

file_reader.rb @f1313f2raw · history · blame

# frozen_string_literal: true

require 'maxmind/db/errors'

module MaxMind
  class DB
    # @!visibility private
    class FileReader
      def initialize(filename)
        @fh = File.new(filename, 'rb')
        @size = @fh.size
        @mutex = Mutex.new
      end

      attr_reader :size

      def close
        @fh.close
      end

      def read(offset, size)
        return ''.b if size == 0

        # When we support only Ruby 2.5+, remove this and require pread.
        if @fh.respond_to?(:pread)
          buf = @fh.pread(size, offset)
        else
          @mutex.synchronize do
            @fh.seek(offset, IO::SEEK_SET)
            buf = @fh.read(size)
          end
        end

        raise InvalidDatabaseError, 'The MaxMind DB file contains bad data' if buf.nil? || buf.length != size

        buf
      end
    end
  end
end