wg-metasploit_data_models 4.1.4.01 → 4.1.4.02

data/spec/app/models/metasploit_data_models/search/operation/ip_address_spec.rb

@@ -0,0 +1,180 @@
+RSpec.describe MetasploitDataModels::Search::Operation::IPAddress, type: :model do
+  subject(:operation) {
+    described_class.new(
+        operator: operator,
+        value: formatted_value
+    )
+  }
+
+  let(:operator) {
+    MetasploitDataModels::Search::Operator::IPAddress.new
+  }
+
+  context 'validation' do
+    #
+    # lets
+    #
+
+    let(:blank_error) {
+      I18n.translate!('errors.messages.blank')
+    }
+
+    let(:invalid_error) {
+      I18n.translate!('errors.messages.invalid')
+    }
+
+    #
+    # Callbacks
+    #
+
+    before(:example) do
+      operation.valid?
+    end
+
+    context 'errors on #value' do
+      subject(:value_error) do
+        operation.errors[:value]
+      end
+
+      context 'with IPv4' do
+        context 'with CIDR' do
+          context 'with valid prefix_length' do
+            let(:formatted_value) {
+              '1.2.3.4/8'
+            }
+
+            it { is_expected.to be_empty }
+          end
+
+          context 'without valid prefix_length' do
+            let(:formatted_value) {
+              '1.2.3.4/36'
+            }
+
+            it { is_expected.to include invalid_error }
+          end
+        end
+
+        context 'with Nmap' do
+          context 'with valid segment range', pending: 'MSP-10712' do
+            let(:formatted_value) {
+              '1-2.3.4.5'
+            }
+
+            it { is_expected.to be_empty }
+          end
+
+          context 'without valid segment range' do
+            let(:formatted_value) {
+              '2-1.3.4.5'
+            }
+
+            it { is_expected.to include invalid_error }
+          end
+        end
+
+        context 'with Range' do
+          context 'with ordered range' do
+            let(:formatted_value) {
+              '2.2.2.2-1.1.1.1'
+            }
+
+            it { is_expected.to include invalid_error }
+          end
+
+          context 'without ordered range' do
+            let(:formatted_value) {
+              '1.1.1.1-2.2.2.2'
+            }
+
+            it { is_expected.to be_empty }
+          end
+        end
+
+        context 'with address' do
+          let(:formatted_value) {
+            '1.2.3.4'
+          }
+
+          it { be_empty }
+        end
+      end
+
+      context 'with nil' do
+        let(:formatted_value) {
+          nil
+        }
+
+        it { is_expected.to include blank_error }
+        it { is_expected.not_to include invalid_error }
+      end
+
+      context 'with empty string' do
+        let(:formatted_value) {
+          ''
+        }
+
+        it { is_expected.to include blank_error }
+        it { is_expected.not_to include invalid_error }
+      end
+
+      context 'without matching formatted value' do
+        let(:formatted_value) {
+          'non_matching_value'
+        }
+
+        it { is_expected.to include invalid_error }
+      end
+    end
+  end
+
+  context '#value' do
+    subject(:value) {
+      operation.value
+    }
+
+    context 'with IPv4' do
+      context 'with CIDR' do
+        let(:formatted_value) {
+          '1.2.3.4/8'
+        }
+
+        it { is_expected.to be_a MetasploitDataModels::IPAddress::V4::CIDR }
+      end
+
+      context 'with Nmap', pending: 'MSP-10712' do
+        let(:formatted_value) {
+          '1.2.3.4-5'
+        }
+
+        it { is_expected.to be_a MetasploitDataModels::IPAddress::V4::Nmap }
+      end
+
+      context 'with Range' do
+        let(:formatted_value) {
+          '1.1.1.1-2.2.2.2'
+        }
+
+        it { is_expected.to be_a MetasploitDataModels::IPAddress::V4::Range }
+      end
+
+      context 'with address' do
+        let(:formatted_value) {
+          '1.2.3.4'
+        }
+
+        it { is_expected.to be_a MetasploitDataModels::IPAddress::V4::Single }
+      end
+    end
+
+    context 'without support format' do
+      let(:formatted_value) {
+        'unsupported_formated'
+      }
+
+      it 'is the passed formatted value' do
+        expect(value).to eq(formatted_value)
+      end
+    end
+  end
+end

data/spec/app/models/metasploit_data_models/search/operation/port/number_spec.rb

@@ -0,0 +1,39 @@
+RSpec.describe MetasploitDataModels::Search::Operation::Port::Number, type: :model do
+  context 'CONSTANTS' do
+    context 'BITS' do
+      subject(:bits) {
+        described_class::BITS
+      }
+
+      it { is_expected.to eq(16) }
+    end
+
+    context 'MAXIMUM' do
+      subject(:maxium) {
+        described_class::MAXIMUM
+      }
+
+      it { is_expected.to eq(65535) }
+    end
+
+    context 'MINIMUM' do
+      subject(:minimum) {
+        described_class::MINIMUM
+      }
+
+      it { is_expected.to eq(0) }
+    end
+
+    context 'RANGE' do
+      subject(:range) {
+        described_class::RANGE
+      }
+
+      it { is_expected.to eq((0..65535)) }
+    end
+  end
+
+  context 'validations' do
+    it { is_expected.to validate_inclusion_of(:value).in_range(described_class::RANGE) }
+  end
+end

data/spec/app/models/metasploit_data_models/search/operation/port/range_spec.rb

@@ -0,0 +1,138 @@
+RSpec.describe MetasploitDataModels::Search::Operation::Port::Range, type: :model do
+  subject(:port_range_operation) {
+    described_class.new(
+        value: formatted_value
+    )
+  }
+
+  let(:formatted_value) {
+    '1'
+  }
+
+  it { is_expected.to be_a MetasploitDataModels::Search::Operation::Range }
+
+  context 'validations' do
+    before(:example) do
+      port_range_operation.valid?
+    end
+
+    context 'errors on #value' do
+      subject(:value_errors) {
+        port_range_operation.errors[:value]
+      }
+
+      context 'with Range' do
+        context 'with Integers' do
+          context 'covered by MetasploitDataModels::Search::Operation::Port::Number::RANGE' do
+            let(:formatted_value) {
+              '1-2'
+            }
+
+            it { is_expected.to be_empty }
+          end
+
+          # this can't actually happen because the minimum is 0 and a negative number can't be parsed, but validation
+          # is there in case @value is set directly.
+          context 'without Range#begin covered by MetasploitDataModels::Search::Operation::Port::Number::RANGE' do
+            let(:error) {
+              I18n.translate!(
+                  'metasploit.model.errors.models.metasploit_data_models/search/operation/port/range.attributes.value.port_range_extreme_inclusion',
+                  extreme: :begin,
+                  extreme_value: range_begin,
+                  maximum: MetasploitDataModels::Search::Operation::Port::Number::MAXIMUM,
+                  minimum: MetasploitDataModels::Search::Operation::Port::Number::MINIMUM
+              )
+            }
+
+            let(:formatted_value) {
+              nil
+            }
+
+            let(:port_range_operation) {
+              super().tap { |port_range_operation|
+                port_range_operation.instance_variable_set(:@value, Range.new(range_begin, range_end))
+              }
+            }
+
+            let(:range_begin) {
+              -1
+            }
+
+            let(:range_end) {
+              1
+            }
+
+            it { is_expected.to include error }
+          end
+
+          context 'without Range#begin covered by MetasploitDataModels::Search::Operation::Port::Number::RANGE' do
+            let(:error) {
+              I18n.translate!(
+                  'metasploit.model.errors.models.metasploit_data_models/search/operation/port/range.attributes.value.port_range_extreme_inclusion',
+                  extreme: :end,
+                  extreme_value: range_end,
+                  maximum: MetasploitDataModels::Search::Operation::Port::Number::MAXIMUM,
+                  minimum: MetasploitDataModels::Search::Operation::Port::Number::MINIMUM
+              )
+            }
+
+            let(:formatted_value) {
+              "0-#{range_end}"
+            }
+
+            let(:range_end) {
+              MetasploitDataModels::Search::Operation::Port::Number::MAXIMUM + 1
+            }
+
+            it { is_expected.to include error }
+          end
+        end
+
+        context 'without Integers' do
+          let(:begin_error) {
+            I18n.translate!(
+                  'metasploit.model.errors.models.metasploit_data_models/search/operation/port/range.attributes.value.port_range_extreme_not_an_integer',
+                  extreme: :begin,
+                  extreme_value: range_begin
+              )
+          }
+
+          let(:end_error) {
+            I18n.translate!(
+                'metasploit.model.errors.models.metasploit_data_models/search/operation/port/range.attributes.value.port_range_extreme_not_an_integer',
+                extreme: :end,
+                extreme_value: range_end
+            )
+          }
+
+          let(:formatted_value) {
+            "#{range_begin}-#{range_end}"
+          }
+
+          let(:range_begin) {
+            'a'
+          }
+
+          let(:range_end) {
+            'b'
+          }
+
+          it { is_expected.to include begin_error }
+          it { is_expected.to include end_error }
+        end
+      end
+
+      context 'without Range' do
+        let(:error) {
+          I18n.translate!('metasploit.model.errors.models.metasploit_data_models/search/operation/range.attributes.value.range')
+        }
+
+        let(:formatted_value) {
+          '1'
+        }
+
+        it { is_expected.to include(error) }
+      end
+    end
+  end
+end

data/spec/app/models/metasploit_data_models/search/operation/range_spec.rb

@@ -0,0 +1,233 @@
+RSpec.describe MetasploitDataModels::Search::Operation::Range, type: :model do
+  subject(:range_operation) {
+    described_class.new(attributes)
+  }
+
+  context 'CONSTANTS' do
+    context 'SEPARATOR' do
+      subject(:separator) {
+        described_class::SEPARATOR
+      }
+
+      it { is_expected.to eq('-') }
+    end
+  end
+
+  context 'validations' do
+    #
+    # lets
+    #
+
+    let(:attributes) {
+      {
+          value: value
+      }
+    }
+
+    #
+    # Callbacks
+    #
+
+    before(:example) do
+      range_operation.valid?
+    end
+
+    context 'errors on #value' do
+      subject(:errors) {
+        range_operation.errors[:value]
+      }
+
+      context '#ordered' do
+        let(:error) {
+          I18n.translate!('metasploit.model.errors.models.metasploit_data_models/search/operation/range.attributes.value.order', error_attributes)
+        }
+
+        context 'with Range' do
+          context 'with begin before end' do
+            let(:error_attributes) {
+              {
+                  begin: '"1"',
+                  end: '"2"'
+              }
+            }
+
+            let(:value) {
+              '1-2'
+            }
+
+            it { is_expected.not_to include(error) }
+          end
+
+          context 'with begin same as end' do
+            let(:error_attributes) {
+              {
+                  begin: '"1"',
+                  end: '"1"'
+              }
+            }
+
+            let(:value) {
+              '1-1'
+            }
+
+            it { is_expected.not_to include(error) }
+          end
+
+          context 'with begin after end' do
+            let(:error_attributes) {
+              {
+                  begin: '"2"',
+                  end: '"1"'
+              }
+            }
+
+            let(:value) {
+              '2-1'
+            }
+
+            it { is_expected.to include error }
+          end
+        end
+
+        context 'without Range' do
+          let(:error_attributes) {
+            {
+                begin: '"1"',
+                end: '"2"'
+            }
+          }
+
+          let(:value) {
+            '1..2'
+          }
+
+          it { is_expected.not_to include(error) }
+        end
+      end
+
+      context '#range' do
+        context 'with Range' do
+          let(:value) {
+            '1-2'
+          }
+
+          it { is_expected.to be_empty }
+        end
+
+        context 'without Range' do
+          let(:error) {
+            I18n.translate!('metasploit.model.errors.models.metasploit_data_models/search/operation/range.attributes.value.range')
+          }
+
+          let(:value) {
+            '1..2'
+          }
+
+          it { is_expected.to include error }
+        end
+      end
+    end
+  end
+
+  context '#value' do
+    subject(:value) {
+      range_operation.value
+    }
+
+    #
+    # lets
+    #
+
+    let(:attributes) {
+      {
+          value: formatted_value
+      }
+    }
+
+    context "without '-'" do
+      let(:formatted_value) {
+        'a..b'
+      }
+
+      it 'returns unconvertable value' do
+        expect(value).to eq(formatted_value)
+      end
+    end
+
+    context "with one '-'" do
+      let(:formatted_begin) {
+        'a'
+      }
+
+      let(:formatted_end) {
+        'b'
+      }
+
+      let(:formatted_value) {
+        "#{formatted_begin}-#{formatted_end}"
+      }
+
+      it 'returns Ranges' do
+        expect(value).to be_a Range
+      end
+
+      context '#begin' do
+        subject(:range_begin) {
+          value.begin
+        }
+
+        it "is part before '-'" do
+          expect(range_begin).to eq(formatted_begin)
+        end
+      end
+
+      context '#end' do
+        subject(:range_end) {
+          value.end
+        }
+
+        it "is part after '-'" do
+          expect(range_end).to eq(formatted_end)
+        end
+      end
+    end
+
+    context "with multiple '-'" do
+      let(:formatted_begin) {
+        'a'
+      }
+
+      let(:formatted_end) {
+        'b-c'
+      }
+
+      let(:formatted_value) {
+        "#{formatted_begin}-#{formatted_end}"
+      }
+
+      it 'returns Ranges' do
+        expect(value).to be_a Range
+      end
+
+      context '#begin' do
+        subject(:range_begin) {
+          value.begin
+        }
+
+        it "is part before first '-'" do
+          expect(range_begin).to eq(formatted_begin)
+        end
+      end
+
+      context '#end' do
+        subject(:range_end) {
+          value.end
+        }
+
+        it "is part after first '-'" do
+          expect(range_end).to eq(formatted_end)
+        end
+      end
+    end
+  end
+end

data/spec/app/models/metasploit_data_models/search/operator/ip_address_spec.rb

@@ -0,0 +1,17 @@
+RSpec.describe MetasploitDataModels::Search::Operator::IPAddress, type: :model do
+  subject(:operator) {
+    described_class.new
+  }
+
+  context '#operate_on' do
+    subject(:operate_on) {
+      operator.operate_on(formatted_value)
+    }
+
+    let(:formatted_value) {
+      nil
+    }
+
+    it { is_expected.to be_a MetasploitDataModels::Search::Operation::IPAddress }
+  end
+end