wifimap 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: bf34d96b423042eeb6528fc637a1d3b622d1d14881b495882b9be47212d31b38
-  data.tar.gz: 04c7fde8424f6885c1a932a7dffe0492465f0ab36fe11d37b37c012088ef614e
+  metadata.gz: 93a98307084f4ecda1bb218964bece248af40a57b95ad2b423dac9654154dd89
+  data.tar.gz: 897d5c06126dadf291ab26207cce1abb75a36a17a2ef8e2697c7c89edcd0b4a3
 SHA512:
-  metadata.gz: 1152bf26fe94bdc03d45aac4d678d4c494912eac9a5977d81bccbdef58c412a27f48ce11c930876502e47350592552f0aa856c383b2d17d5a773d0a4d4a38a21
-  data.tar.gz: 6e8dfc0f49d28ee6b540f066548f80c77befbe74b9d9fecef0988e46c30e825718f313741468a0a91aaf7863cf8367a22cffbf431c598d9271d087aed40437c2
+  metadata.gz: dcbd8568e6634cf0a59d0048db5d43145ed61dfbca93c85dd3ab2d9bc8166c96bcaafa7a707a50a8ecff1eb3451385968e38535ef60ae65e9bad56aa1cd1eb81
+  data.tar.gz: d54d442bda06642a18a10c51a64b3488f7bb963a3f4b3d81e43ccf1e8a75d8e02f1465279ff189c8722d352530d5ee45ea89efb4529d6815a5dc8b9d8f7c2a88

data/.gitignore

@@ -6,6 +6,7 @@
 /pkg/
 /spec/reports/
 /tmp/
+.DS_Store
 
 # rspec failure tracking
 .rspec_status

data/CHANGELOG.md

@@ -0,0 +1,4 @@
+# v0.2.0
+- The parse method is now on the top-level namespace (`Wifimap.parse`).
+- Added `Wifimap.parse_file(filename)`.
+- Added support for [sniff-probes](https://github.com/brannondorsey/sniff-probes).

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    wifimap (0.1.0)
+    wifimap (0.2.0)
       louis
 
 GEM

data/README.md

@@ -22,12 +22,14 @@ Or install it yourself as:
 
 ## Usage
 
-Require the parser and pass the content of a dump file to the `parse` method:
+Require wifimap and pass the content of a dump file to the `parse` method:
 ```rb
-require 'wifimap/parser'
+require 'wifimap'
 
 dump = File.read('airodump.csv')
-parsed = Wifimap::Parser.parse(dump)
+parsed = Wifimap.parse(dump)
+# Alternative:
+# parsed = Wifimap.parse_file('airodump.csv')
 
 parsed.access_points  # returns an array of AccessPoint objects
 parsed.stations       # returns an array of Station objects
@@ -55,6 +57,7 @@ station.manufacturer  # ex.: 'Apple, Inc.'
 
 ### Supported dump formats
 - Airodump CSV ([airodump-ng](https://www.aircrack-ng.org/doku.php?id=airodump-ng))
+- Sniff Probes ([sniff-probes](https://github.com/brannondorsey/sniff-probes))
 
 ## Development
 

data/lib/wifimap/parser/sniff_probes.rb

@@ -0,0 +1,43 @@
+# frozen_string_literal: true
+
+require 'wifimap/station'
+
+module Wifimap
+  module Parser
+    # Parse the content of a sniff-probes file.
+    class SniffProbes
+      attr_reader :access_points
+
+      def initialize(dump)
+        @dump = dump
+        @access_points = []
+      end
+
+      # Get the list of stations from the dump.
+      def stations
+        unique_macs.map do |mac|
+          station = Station.new(mac: mac)
+          rows.each do |row|
+            fields = row.split('"')
+            unless station.probes.include?(fields[1])
+              station.probes << fields[1] if fields[0].include?(mac)
+            end
+          end
+          station
+        end
+      end
+
+      private
+
+      # Return an array of lines from the dump content.
+      def rows
+        @dump.split("\n")
+      end
+
+      # Return all unique mac addresses.
+      def unique_macs
+        rows.map { |row| row.split[2] }.uniq
+      end
+    end
+  end
+end

data/lib/wifimap/parser.rb

@@ -1,22 +1,29 @@
 # frozen_string_literal: true
 
-require 'wifimap/parser/airodump'
+require 'wifimap/mac'
 
 module Wifimap
   module Parser
-    # Parse the content of a dump file and return a hash of APs, stations and probes.
-    def self.parse(file_content)
-      case dump_format(file_content)
-      when :airodump
-        Wifimap::Parser::Airodump.new(file_content)
-      else
-        raise Error, 'Unsupported dump format'
-      end
+    # Check the file content and identify the correct dump format.
+    def self.dump_format(file_content)
+      return :airodump if airodump_format?(file_content)
+      return :sniff_probes if sniff_probes_format?(file_content)
     end
 
-    # Check the file content and identify the dump format.
-    def self.dump_format(file_content)
-      airodump_header = 'BSSID, First time seen, Last time seen, channel, Speed, Privacy, Cipher, Authentication, Power'
-      :airodump if file_content.include?(airodump_header)    end
+    class << self
+      private
+
+      def airodump_format?(file_content)
+        airodump_header = 'BSSID, First time seen, Last time seen, channel, Speed, Privacy, Cipher, Authentication, Power'
+        file_content.include?(airodump_header)
+      end
+
+      def sniff_probes_format?(file_content)
+        file_content.split("\n").all? do |row|
+          fields = row.split
+          fields[1].include?('dBm') && Wifimap::Mac.valid?(fields[2])
+        end
+      end
+    end
   end
 end

data/lib/wifimap/version.rb

@@ -1,3 +1,3 @@
 module Wifimap
-  VERSION = "0.1.0"
+  VERSION = "0.2.0"
 end

data/lib/wifimap.rb

@@ -1,7 +1,27 @@
 # frozen_string_literal: true
 
 require 'wifimap/version'
+require 'wifimap/parser'
+require 'wifimap/parser/airodump'
 
 module Wifimap
   class Error < StandardError; end
+
+  # Parse the content of a dump file and return a hash of APs, stations and probes.
+  def self.parse(file_content)
+    case Wifimap::Parser.dump_format(file_content)
+    when :airodump
+      Wifimap::Parser::Airodump.new(file_content)
+    when :sniff_probes
+      Wifimap::Parser::SniffProbes.new(file_content)
+    else
+      raise Error, 'Unsupported dump format'
+    end
+  end
+
+  # A wrapper on .parse to read directly from a file.
+  def self.parse_file(filename)
+    file_content = File.read(filename)
+    parse(file_content)
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: wifimap
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Gianpiero Addis
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-04-17 00:00:00.000000000 Z
+date: 2020-04-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: louis
@@ -48,6 +48,7 @@ files:
 - lib/wifimap/mac.rb
 - lib/wifimap/parser.rb
 - lib/wifimap/parser/airodump.rb
+- lib/wifimap/parser/sniff_probes.rb
 - lib/wifimap/station.rb
 - lib/wifimap/version.rb
 - wifimap.gemspec