Contents

##
# This class represents any CVSS vector. Do not instantiate this class!

class Cvss

  ##
  # Metric of a CVSS vector.

  attr_reader :base, :temporal, :environmental

  ##
  # Returns version of current CVSS vector.

  attr_reader :version

  ##
  # Returns the vector itself.

  attr_reader :vector

  ##
  # Creates a new CVSS vector by a +vector+ and a +version+.
  #
  # Raises an exception if it is called on Cvss class.

  def initialize(vector, version)
    raise 'Do not instantiate this class!' if self.class == Cvss
    @version = version
    @vector = vector
    @properties = []
    extract_metrics
    init_metrics
  end

  ##
  # Returns if CVSS vector is valid.

  def valid?
    if @amount_of_properties == required_amount_of_properties
        base = @base.valid?
        temporal = @base.valid? && @temporal.valid?
        environmental = @base.valid? && @environmental.valid?
        full = @base.valid? && @temporal.valid? && @environmental.valid?
        base || temporal || environmental || full
    else
      false
    end
  end

  ##
  # Returns the Overall Score of the CVSS vector.

  def overall_score
    check_valid
    return temporal_score if @temporal.valid? && !@environmental.valid?
    return environmental_score if @environmental.valid?
    base_score
  end

  private

  def extract_metrics
    properties = prepared_vector.split('/')
    @amount_of_properties = properties.size
    properties.each_with_index do |property, index|
      property = property.split(':')
      @properties.push({ name: property[0], selected: property[1], position: index })
    end
  end

  def check_valid
    raise 'Vector is not valid!' unless valid?
  end

  def prepared_vector
    start_of_vector = @vector.index('AV')
    @vector[start_of_vector..-1]
  end

  def required_amount_of_properties
    total = @base.count if @base.valid?
    total += @temporal.count if @temporal.valid?
    total += @environmental.count if @environmental.valid?
    total ||= 0
  end

end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
cvss-suite-1.0.6	lib/cvss_suite/cvss.rb
cvss-suite-1.0.5	lib/cvss_suite/cvss.rb