metasploit_data_models 0.18.1 → 0.19.0

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

@@ -0,0 +1,182 @@
+require 'spec_helper'
+
+describe MetasploitDataModels::Search::Operation::IPAddress 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(:each) 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 { should be_empty }
+          end
+
+          context 'without valid prefix_length' do
+            let(:formatted_value) {
+              '1.2.3.4/36'
+            }
+
+            it { should include invalid_error }
+          end
+        end
+
+        context 'with Nmap', pending: 'MSP-10712' do
+          context 'with valid segment range' do
+            let(:formatted_value) {
+              '1-2.3.4.5'
+            }
+
+            it { should be_empty }
+          end
+
+          context 'without valid segment range' do
+            let(:formatted_value) {
+              '2-1.3.4.5'
+            }
+
+            it { should 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 { should include invalid_error }
+          end
+
+          context 'without ordered range' do
+            let(:formatted_value) {
+              '1.1.1.1-2.2.2.2'
+            }
+
+            it { should 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 { should include blank_error }
+        it { should_not include invalid_error }
+      end
+
+      context 'with empty string' do
+        let(:formatted_value) {
+          ''
+        }
+
+        it { should include blank_error }
+        it { should_not include invalid_error }
+      end
+
+      context 'without matching formatted value' do
+        let(:formatted_value) {
+          'non_matching_value'
+        }
+
+        it { should 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 { should be_a MetasploitDataModels::IPAddress::V4::CIDR }
+      end
+
+      context 'with Nmap', pending: 'MSP-10712' do
+        let(:formatted_value) {
+          '1.2.3.4-5'
+        }
+
+        it { should be_a MetasploitDataModels::IPAddress::V4::Nmap }
+      end
+
+      context 'with Range' do
+        let(:formatted_value) {
+          '1.1.1.1-2.2.2.2'
+        }
+
+        it { should be_a MetasploitDataModels::IPAddress::V4::Range }
+      end
+
+      context 'with address' do
+        let(:formatted_value) {
+          '1.2.3.4'
+        }
+
+        it { should 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/operator/ip_address_spec.rb

@@ -0,0 +1,19 @@
+require 'spec_helper'
+
+describe MetasploitDataModels::Search::Operator::IPAddress do
+  subject(:operator) {
+    described_class.new
+  }
+
+  context '#operate_on' do
+    subject(:operate_on) {
+      operator.operate_on(formatted_value)
+    }
+
+    let(:formatted_value) {
+      nil
+    }
+
+    it { should be_a MetasploitDataModels::Search::Operation::IPAddress }
+  end
+end

data/spec/app/models/metasploit_data_models/search/visitor/relation_spec.rb

@@ -158,46 +158,80 @@ describe MetasploitDataModels::Search::Visitor::Relation do
     context 'matching record' do
       context 'Metasploit::Model::Search::Query#klass' do
         context 'with Mdm::Service' do
+          include_context 'Rex::Text'
+
+          #
+          # lets
+          #
+
           let(:klass) {
             Mdm::Service
           }
 
-          let(:matching_ports) {
-            [
-                1,
-                2
-            ]
+          #
+          # Don't use factories to prevent prefix aliasing when sequences go from 1 to 10 or 10 to 100
+          #
+
+          let(:non_matching_info) {
+            'mdm_service_info_c'
           }
 
-          let(:matching_records) {
-            matching_record_by_port.values
+          let(:non_matching_name) {
+            'mdm_service_name_c'
           }
 
           let(:non_matching_port) {
             3
           }
 
+          let(:non_matching_proto) {
+            'udp'
+          }
+
           #
           # let!s
           #
 
-          let!(:matching_record_by_port) {
-            matching_ports.each_with_object({}) { |matching_port, matching_record_by_port|
-              matching_record_by_port[matching_port] = FactoryGirl.create(
-                  :mdm_service,
-                  port: matching_port
-              )
-            }
-          }
-
           let!(:non_matching_record) {
             FactoryGirl.create(
                 :mdm_service,
-                port: non_matching_port
+                info: non_matching_info,
+                name: non_matching_name,
+                port: non_matching_port,
+                proto: non_matching_proto
             )
           }
+          p
 
           context 'with port' do
+            #
+            # lets
+            #
+
+            let(:matching_ports) {
+              [
+                  1,
+                  2
+              ]
+            }
+
+            let(:matching_records) {
+              matching_record_by_port.values
+            }
+
+            #
+            # let!s
+            #
+
+            let!(:matching_record_by_port) {
+              matching_ports.each_with_object({}) { |matching_port, matching_record_by_port|
+                matching_record_by_port[matching_port] = FactoryGirl.create(
+                    :mdm_service,
+                    port: matching_port
+                )
+              }
+            }
+
             context 'with single port number' do
               let(:formatted) {
                 "port:#{matching_port}"
@@ -251,26 +285,77 @@ describe MetasploitDataModels::Search::Visitor::Relation do
             end
           end
 
-          context 'with all operators' do
-            let(:formatted) {
-              %Q{port:#{matching_port}}
+          context 'with single matching record' do
+            #
+            # lets
+            #
+
+            #
+            # Don't use factories to prevent prefix aliasing when sequences go from 1 to 10 or 10 to 100
+            #
+
+            let(:matching_info) {
+              'mdm_service_info_a'
+            }
+
+            let(:matching_name) {
+              'mdm_service_name_a'
+            }
+
+            let(:matching_port) {
+              1
+            }
+
+            let(:matching_proto) {
+              'tcp'
+            }
+
+            #
+            # let!s
+            #
+
+            let!(:matching_record) {
+              FactoryGirl.create(
+                  :mdm_service,
+                  info: matching_info,
+                  name: matching_name,
+                  port: matching_port,
+                  proto: matching_proto
+              )
             }
+
+            it_should_behave_like 'MetasploitDataModels::Search::Visitor::Relation#visit matching record',
+                                  attribute: :info
+
+            it_should_behave_like 'MetasploitDataModels::Search::Visitor::Relation#visit matching record',
+                                  attribute: :name
+
+            it_should_behave_like 'MetasploitDataModels::Search::Visitor::Relation#visit matching record',
+                                  attribute: :proto
+
+            context 'with all operators' do
+              let(:formatted) {
+                %Q{name:#{matching_name} port:#{matching_port} proto:#{matching_proto}}
+              }
+
+              it 'finds only matching record' do
+                expect(visit).to match_array([matching_record])
+              end
+            end
           end
         end
 
         context 'with Mdm::Host' do
           #
           # lets
-          #
-
-          let(:klass) {
-            Mdm::Host
-          }
-
           #
           # Don't use factories to prevent prefix aliasing when sequences go from 1 to 10 or 10 to 100
           #
 
+          let(:matching_record_address) {
+            '1.2.3.4'
+          }
+
           let(:matching_record_os_flavor) {
             'mdm_host_os_flavor_a'
           }
@@ -283,10 +368,6 @@ describe MetasploitDataModels::Search::Visitor::Relation do
             'mdm_host_os_sp_a'
           }
 
-          let(:matching_service_name) {
-            'mdm_service_name_a'
-          }
-
           let(:matching_record_name) {
             'mdm_host_name_a'
           }
@@ -295,6 +376,10 @@ describe MetasploitDataModels::Search::Visitor::Relation do
             'mdm_service_name_a'
           }
 
+          let(:non_matching_record_address) {
+            '5.6.7.8'
+          }
+
           let(:non_matching_record_os_flavor) {
             'mdm_host_os_flavor_b'
           }
@@ -307,10 +392,6 @@ describe MetasploitDataModels::Search::Visitor::Relation do
             'mdm_host_os_sp_b'
           }
 
-          let(:non_matching_service_name) {
-            'mdm_service_name_b'
-          }
-
           let(:non_matching_record_name) {
             'mdm_host_name_b'
           }
@@ -326,6 +407,7 @@ describe MetasploitDataModels::Search::Visitor::Relation do
           let!(:matching_record) do
             FactoryGirl.build(
                 :mdm_host,
+                address: matching_record_address,
                 name: matching_record_name,
                 os_flavor: matching_record_os_flavor,
                 os_name: matching_record_os_name,
@@ -344,6 +426,7 @@ describe MetasploitDataModels::Search::Visitor::Relation do
           let!(:non_matching_record) do
             FactoryGirl.build(
                 :mdm_host,
+                address: non_matching_record_address,
                 name: non_matching_record_name,
                 os_flavor: non_matching_record_os_flavor,
                 os_name: non_matching_record_os_name,
@@ -359,6 +442,42 @@ describe MetasploitDataModels::Search::Visitor::Relation do
             )
           end
 
+          context 'with address operator' do
+            let(:formatted) do
+              "address:#{formatted_address}"
+            end
+
+            context 'with CIDR' do
+              let(:formatted_address) {
+                '1.3.4.5/8'
+              }
+
+              it 'should find only matching record' do
+                expect(visit).to match_array([matching_record])
+              end
+            end
+
+            context 'with Range' do
+              let(:formatted_address) {
+                '1.1.1.1-5.6.7.7'
+              }
+
+              it 'should find only matching record' do
+                expect(visit).to match_array([matching_record])
+              end
+            end
+
+            context 'with single' do
+              let(:formatted_address) {
+                '1.2.3.4'
+              }
+
+              it 'should find only matching record' do
+                expect(visit).to match_array([matching_record])
+              end
+            end
+          end
+
           it_should_behave_like 'MetasploitDataModels::Search::Visitor::Relation#visit matching record',
                                 :attribute => :name
 
@@ -425,7 +544,17 @@ describe MetasploitDataModels::Search::Visitor::Relation do
 
           context 'with all operators' do
             let(:formatted) {
-              %Q{name:"#{matching_record_name}" os:"#{matching_record_os_name} #{matching_record_os_flavor} #{matching_record_os_sp}" os_flavor:"#{matching_record_os_flavor}" os_name:"#{matching_record_os_name}" os_sp:"#{matching_record_os_sp}" services.name:"#{matching_service_name}"}
+              %Q{
+              address:1.3.4.5/8
+              address:1.1.1.1-5.6.7.7
+              address:1.2.3.4
+              name:"#{matching_record_name}"
+              os:"#{matching_record_os_name} #{matching_record_os_flavor} #{matching_record_os_sp}"
+              os_flavor:"#{matching_record_os_flavor}"
+              os_name:"#{matching_record_os_name}"
+              os_sp:"#{matching_record_os_sp}"
+              services.name:"#{matching_service_name}"
+            }
             }
 
             it 'should find only matching record' do
@@ -437,6 +566,70 @@ describe MetasploitDataModels::Search::Visitor::Relation do
             end
           end
         end
+
+
+      end
+
+      context 'with Mdm::Tag' do
+        #
+        # lets
+        #
+
+        let(:klass) {
+          Mdm::Tag
+        }
+
+        let(:matching_desc) {
+          'This is a description'
+        }
+
+        let(:matching_name) {
+          'matching.tag'
+        }
+
+        let(:non_matching_desc) {
+          'This could be a description'
+        }
+
+        let(:non_matching_name) {
+          'tag.does.not.match'
+        }
+
+        #
+        # let!s
+        #
+
+        let!(:matching_record) {
+          FactoryGirl.create(
+            :mdm_tag,
+            desc: matching_desc,
+            name: matching_name
+          )
+        }
+
+        let!(:non_matching_record) {
+          FactoryGirl.create(
+            :mdm_tag,
+            desc: non_matching_desc,
+            name: non_matching_name
+          )
+        }
+
+        it_should_behave_like 'MetasploitDataModels::Search::Visitor::Relation#visit matching record',
+          attribute: :desc
+
+        it_should_behave_like 'MetasploitDataModels::Search::Visitor::Relation#visit matching record',
+          attribute: :name
+
+        context 'with all operators' do
+          let(:formatted) {
+            %Q{desc:"#{matching_desc}" name:"#{matching_name}"}
+          }
+
+          it 'should find only matching record' do
+            expect(visit).to match_array([matching_record])
+          end
+        end
       end
     end
   end
@@ -460,4 +653,4 @@ describe MetasploitDataModels::Search::Visitor::Relation do
     its([:includes]) { should == MetasploitDataModels::Search::Visitor::Includes }
     its([:where]) { should == MetasploitDataModels::Search::Visitor::Where }
   end
-end
+end