pass-station 1.2.2 → 1.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7ec0b0404526c7b3184c441897083e2024404e7f22b4ac2851292d81f69cda25
-  data.tar.gz: b27776bfc51827851cc6ef60ce4c378f0fa9ec923f104109abb66733df5ba5d3
+  metadata.gz: dce80d32341aedab22c054f5bba662411df09d1d585e8e7988fcbc6cb7a14a6c
+  data.tar.gz: cc71183091fed6f3766f7d4390973ecc869022a8760ebd8706d837066f240610
 SHA512:
-  metadata.gz: 26891a1b483d324034656b3921dc2684938972ce18f6f39d7746f6aaa3f387bcff6fb4964a45160c87c98f6bdaaf26eb93f0ccc935f8573867cc4fa63f345bad
-  data.tar.gz: 6728d55a675019cd612162b1f5c10f142a0f14008429984681fa3c939422c8ebf0ebdab0afb08d85cb5e3678a2e2e4a743771b24a2645be24c591bf94b0f89f5
+  metadata.gz: 6b48de95b152bcb493551c380390d528de04d61e36123d06d5ec62f2cc3bafd987ec8e411f8b10ef109b1ec116f08375f367da20860ee79813cf92a866d9efad
+  data.tar.gz: b5b30dc1e5885101617d0d893286d6b1086672fc8fd01d2a2767336f37c5d96551d76366897d3193d8e9828ac57fbc7333504895f7b661f30a44210ddfa6d229

data/LICENSE

@@ -1,5 +1,6 @@
 MIT License
 
+Copyright (c) 2021 Alexandre ZANNI
 Copyright (c) 2021 Alexandre ZANNI at SEC-IT
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
@@ -18,4 +19,4 @@ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.
+SOFTWARE.

data/bin/pass-station

@@ -11,11 +11,11 @@ require 'docopt'
 require 'paint'
 
 doc = <<~DOCOPT
-  Pass Station
+  Pass Station v#{PassStation::VERSION}
 
   Usage:
-    pass-station list [--sort <col> --output <format>] [--debug]
-    pass-station search <term> [--field <col> --sort <col> --sensitive --output <format>] [--no-color --debug]
+    pass-station list [--sort <col> --output <format>] [--source <id> --debug]
+    pass-station search <term> [--field <col> --sort <col> --sensitive --output <format>] [--source <id> --no-color --debug]
     pass-station update ([--force] <path> | --check) [--debug]
     pass-station -h | --help
     pass-station --version
@@ -24,10 +24,10 @@ doc = <<~DOCOPT
 
   Output options: can be used with list and search commands
     -o <format>, --output <format>  Output format: JSON, CSV, YAML, table, pretty-table [default: pretty-table]
-    -s <col>, --sort <col>          Sort by column: productvendor | username | password [default: productvendor]
+    -s <col>, --sort <col>          Sort by column (see documentation, columns depends on the database source)
 
   Search options:
-    --field <col>   Search in column: productvendor | username | password | all [default: productvendor]
+    --field <col>   Search in column: column name (see documentation, columns depends on the database source) or all
     --sensitive     Search is case sensitive (case insensitive by default)
 
   Update options: update the password database (replace Pass Station DB with upstream DB, use with care)
@@ -35,10 +35,11 @@ doc = <<~DOCOPT
     -c, --check   Check for possible update
 
   Other options:
-    --no-color  Disable colorized output
-    --debug     Display arguments
-    -h, --help  Show this screen
-    --version   Show version
+    --source <id> Credentials source database: 1 (Default Credentials Cheat Sheet), 2 (Many passwords) [default: 1]
+    --no-color    Disable colorized output
+    --debug       Display arguments
+    -h, --help    Show this screen
+    --version     Show version
 DOCOPT
 
 begin
@@ -47,14 +48,17 @@ begin
   pp args if args['--debug']
   if args['update']
     if args['<path>']
-      opts = {}
-      opts[:sha256] = args['--force'] ? nil : PassStation::DB::UPSTREAM_DATABASE[:HASH]
-      puts '[+] Updating database'
-      path = PassStation::DB.download_upstream(args['<path>'], opts)
-      if path
-        puts "[+] Database updated #{path}"
-      else
-        puts '[+] Database already up to date'
+      PassStation::DB::UPSTREAM_DATABASE[:MAPPING].each do |k, v|
+        opts = {}
+        opts[:sha256] = args['--force'] ? nil : PassStation::DB::UPSTREAM_DATABASE[v][:HASH]
+        puts "[+] Updating database: #{v}"
+        opts[:source_db] = k
+        path = PassStation::DB.download_upstream(args['<path>'], opts)
+        if path
+          puts "[+] Database updated: #{v} (#{path})"
+        else
+          puts "[+] Database #{v} already up to date"
+        end
       end
     elsif args['--check']
       if PassStation::DB.check_for_update
@@ -64,13 +68,16 @@ begin
       end
     end
   elsif args['list']
-    ps = PassStation::DB.new
-    ps.parse(args['--sort'].to_sym)
+    db = args['--source'].nil? ? nil : args['--source'].to_i
+    ps = PassStation::DB.new(db)
+    args['--sort'].nil? ? ps.parse : ps.parse(args['--sort'].to_sym)
     puts ps.output_list(args['--output'])
   elsif args['search']
-    ps = PassStation::DB.new
-    ps.parse(args['--sort'].to_sym)
-    ps.search(args['<term>'], args['--field'].to_sym, sensitive: args['--sensitive'])
+    db = args['--source'].nil? ? nil : args['--source'].to_i
+    ps = PassStation::DB.new(db)
+    args['--sort'].nil? ? ps.parse : ps.parse(args['--sort'].to_sym)
+    field = args['--field'].nil? ? nil : args['--field'].to_sym
+    ps.search(args['<term>'], field, sensitive: args['--sensitive'])
     output = ps.output_search(args['--output'])
     puts '[-] No result' if output.empty?
     puts ps.highlight_found(args['<term>'], output, args['--sensitive'])