Merge remote-tracking branch 'upstream/pull/5391'

[rails.git] / lib / gpx.rb

diff --git a/lib/gpx.rb b/lib/gpx.rb
index d0607233a2d7f4f108ce35f0b86122c71956a0c3..45a4dcf5fc0bd62335ea0339409e2c1fbfd49e6d 100644 (file)
--- a/lib/gpx.rb
+++ b/lib/gpx.rb
@@ -6,8 +6,9 @@ module GPX
 
     attr_reader :possible_points, :actual_points, :tracksegs
 
 
     attr_reader :possible_points, :actual_points, :tracksegs
 

-    def initialize(file)
+    def initialize(file, options = {})

       @file = file
       @file = file

+      @maximum_points = options[:maximum_points] || Float::INFINITY

     end
 
     def parse_file(reader)
     end
 
     def parse_file(reader)


@@ -19,13 +20,14 @@ module GPX
           if reader.name == "trkpt"
             point = TrkPt.new(@tracksegs, reader["lat"].to_f, reader["lon"].to_f)
             @possible_points += 1
           if reader.name == "trkpt"
             point = TrkPt.new(@tracksegs, reader["lat"].to_f, reader["lon"].to_f)
             @possible_points += 1

+            raise FileTooBigError if @possible_points > @maximum_points

           elsif reader.name == "ele" && point
             point.altitude = reader.read_string.to_f
           elsif reader.name == "time" && point
           elsif reader.name == "ele" && point
             point.altitude = reader.read_string.to_f
           elsif reader.name == "time" && point

-            point.timestamp = Time.parse(reader.read_string)
+            point.timestamp = Time.parse(reader.read_string).utc

           end
         when XML::Reader::TYPE_END_ELEMENT
           end
         when XML::Reader::TYPE_END_ELEMENT

-          if reader.name == "trkpt" && point && point.valid?
+          if reader.name == "trkpt" && point&.valid?

             point.altitude ||= 0
             yield point
             @actual_points += 1
             point.altitude ||= 0
             yield point
             @actual_points += 1


@@ -44,18 +46,18 @@ module GPX
       @tracksegs = 0
 
       begin
       @tracksegs = 0
 
       begin

-        Archive::Reader.open_filename(@file).each_entry_with_data do |_entry, data|
-          parse_file(XML::Reader.string(data), &block)
+        Archive::Reader.open_filename(@file).each_entry_with_data do |entry, data|
+          parse_file(XML::Reader.string(data), &block) if entry.regular?

         end
       rescue Archive::Error
         io = ::File.open(@file)
 
         end
       rescue Archive::Error
         io = ::File.open(@file)
 

-        case MimeMagic.by_magic(io)&.type
+        case Marcel::MimeType.for(io)

         when "application/gzip" then io = Zlib::GzipReader.open(@file)
         when "application/x-bzip" then io = Bzip2::FFI::Reader.open(@file)
         end
 
         when "application/gzip" then io = Zlib::GzipReader.open(@file)
         when "application/x-bzip" then io = Bzip2::FFI::Reader.open(@file)
         end
 

-        parse_file(XML::Reader.io(io), &block)
+        parse_file(XML::Reader.io(io, :options => XML::Parser::Options::NOERROR), &block)

       end
     end
 
       end
     end
 


@@ -121,7 +123,7 @@ module GPX
 
       output = StringIO.new
       image.export(output)
 
       output = StringIO.new
       image.export(output)

-      output.read
+      output

     end
 
     def icon(min_lat, min_lon, max_lat, max_lon)
     end
 
     def icon(min_lat, min_lon, max_lat, max_lon)


@@ -161,7 +163,7 @@ module GPX
         end
       end
 
         end
       end
 

-      image.gif
+      StringIO.new(image.gif)

     end
   end
 
     end
   end
 


@@ -172,4 +174,10 @@ module GPX
         longitude >= -180 && longitude <= 180
     end
   end
         longitude >= -180 && longitude <= 180
     end
   end

+
+  class FileTooBigError < RuntimeError
+    def initialise
+      super("GPX File contains too many points")
+    end
+  end

 end
 end