Implement lesser? method #862

lib/common/models/wp_item/infos.rb

@@ -10,11 +10,6 @@ class WpItem
       !readme_url.nil?
     end
 
-    # @return [ Boolean ]
-    def outdated?(installed_version, latest_version)
-      installed_version < latest_version
-    end
-
     # @return [ String,nil ] The url to the readme file, nil if not found
     def readme_url
       # See https://github.com/wpscanteam/wpscan/pull/737#issuecomment-66375445

lib/common/models/wp_item/output.rb

@@ -5,14 +5,16 @@ class WpItem
 
     # @return [ Void ]
     def output(verbose = false)
+      outdated = VersionCompare.lesser?(version, latest_version) if latest_version
+
       puts
       puts info("Name: #{self}") #this will also output the version number if detected
-      puts " |  Latest version: #{latest_version} (up to date)" if latest_version && !outdated?(version, latest_version)
+      puts " |  Latest version: #{latest_version} (up to date)" if latest_version && !outdated
       puts " |  Last updated: #{last_updated}" if last_updated
       puts " |  Location: #{url}"
       puts " |  Readme: #{readme_url}" if has_readme?
       puts " |  Changelog: #{changelog_url}" if has_changelog?
-      puts warning("The version is out of date, the latest version is #{latest_version}") if latest_version && outdated?(version, latest_version)
+      puts warning("The version is out of date, the latest version is #{latest_version}") if latest_version && outdated
 
       puts warning("Directory listing is enabled: #{url}") if has_directory_listing?
       puts warning("An error_log file has been found: #{error_log_url}") if has_error_log?

lib/common/version_compare.rb

@@ -11,8 +11,8 @@ class VersionCompare
   # @return [ Boolean ]
   def self.lesser_or_equal?(version1, version2)
     # Prepend a '0' if the version starts with a '.'
-    version1 = "0#{version1}" if version1 && version1[0,1] == '.'
-    version2 = "0#{version2}" if version2 && version2[0,1] == '.'
+    version1 = prepend_zero(version1)
+    version2 = prepend_zero(version2)
 
     return true if (version1 == version2)
     # Both versions must be set
@@ -27,4 +27,36 @@ class VersionCompare
     end
     return false
   end
+
+  # Compares two version strings. Returns true if version1 < version2
+  # and false otherwise
+  #
+  # @param [ String ] version1
+  # @param [ String ] version2
+  #
+  # @return [ Boolean ]
+  def self.lesser?(version1, version2)
+    # Prepend a '0' if the version starts with a '.'
+    version1 = prepend_zero(version1)
+    version2 = prepend_zero(version2)
+
+    return false if (version1 == version2)
+    # Both versions must be set
+    return false unless (version1 and version2)
+    return false if (version1.empty? or version2.empty?)
+    begin
+      return true if (Gem::Version.new(version1) < Gem::Version.new(version2))
+    rescue ArgumentError => e
+      # Example: ArgumentError: Malformed version number string a
+      return false if e.message =~ /Malformed version number string/
+      raise
+    end
+    return false
+  end
+
+  # @return [ String ]
+  def self.prepend_zero(version)
+    return nil if version.nil?
+    version[0,1] == '.' ? "0#{version}" : version
+  end
 end