get-voodoo 0.0.9 → 0.0.12

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 99defe02c0508d802bc59b7e6e794457d3aba4510055a70fc2324ec6303b84d8
-  data.tar.gz: f2c9e7c31bf6565b4b34a0441954a85b39cd94a277a12c5f552b0c274d242e3f
+  metadata.gz: 6906a5bfd817cfdb86c525ffc86677ef7185991a70d0c678df07e41ee2394a81
+  data.tar.gz: 8adb2abcc70fe837e0c5c92cfc306d61a4e14c844bdf50009f0765b458bcc479
 SHA512:
-  metadata.gz: 6221573992f6446fe7d4bafbdfe41676b3d95e1b6e04b9fcf1d0898bfb1b07f0c245df005ccbe7e4324e3f5530abcee6d8d52dce23989c560b94587d565fc6a9
-  data.tar.gz: 36bc49e969f2be12079a3aaf5b8ed2b815c1a00fe6389123ec0d84cd3cf3e2c05e3466b0e7255843512c47886b30ec5c1dff0dbd552dea1322f04653e09a7401
+  metadata.gz: 2021b65dc3154a76238cb3cb2015911cda5ddd3aff056814857670925d67ad471d0d8bf4ee4d812940a070819dbc35ac1899c654af82dedfbf4af8eb68727674
+  data.tar.gz: 2febeb01c329895e3313d309e32ce7db3b247f0ea092a4ed0502bfe0d7f97c5050041f62b42e4a7236871869e114e56d8458bdf9188201eb9eeb4582c4d64131

data/lib/voodoo/browser.rb

@@ -6,10 +6,9 @@ module VOODOO
     class Browser
         attr_reader :extension
         attr_accessor :bundle
-        attr_accessor :profile
         attr_accessor :process_name
 
-        def initialize(bundle: nil, process_name: nil, profile: nil, extension: Extension.new)
+        def initialize(bundle: nil, process_name: nil, extension: Extension.new)
             @bundle = bundle
             @extension = extension
             @process_name = process_name
@@ -50,17 +49,20 @@ module VOODOO
             @extension.manifest[:permissions] += permissions
         end
 
-        def hijack(urls = [])
+        def close_browser
             # kill the browser process twise, to bypass close warning
             `pkill -a -i "#{@process_name}"`
             `pkill -a -i "#{@process_name}"`
-            sleep 0.1
+            sleep 0.2
+        end
+
+        def hijack(urls = [], flags: '')
+            close_browser()
 
             urls = [urls] unless urls.kind_of? Array
             urls = urls.uniq
 
-            profile_dir = "--profile-directory=\"#{@profile}\"" if @profile != nil
-            `open -b "#{@bundle}" --args #{profile_dir} --load-extension="#{@extension.save}" #{urls.shift}`
+            `open -b "#{@bundle}" --args #{flags} --load-extension="#{@extension.save}" #{urls.shift}`
 
             if urls.length > 0
                 sleep 0.5
@@ -94,7 +96,7 @@ module VOODOO
             self.new(bundle: 'org.chromium.Chromium', process_name: 'Chromium')
         end
 
-        def add_script(content: nil, file: nil, matches: nil, options: {}, background: false, max_events: nil)
+        def add_script(content: nil, file: nil, matches: nil, options: {}, background: false, max_events: nil, communication: true)
             if matches != nil && background != false
                 puts 'WARNING: matches is ignored when background is set to true.'
             end
@@ -109,8 +111,8 @@ module VOODOO
 
             event_count = 0
 
-            if block_given?
-                collector = Collector.new
+            if block_given? && communication == true
+                collector = Collector.new(close_browser: method(:close_browser))
                 collector.on_json {|jsond|
                     yield jsond
                     if (max_events != nil)
@@ -160,15 +162,6 @@ module VOODOO
                 return @extension.add_content_script(matches, js: [content])
             end
         end
-
-        protected
-
-        def make_collector
-            collector = Collector.new
-            collector.on_json {|jsond| yield jsond }
-            @collector_threads.push(collector.thread)
-            return collector
-        end
     end
 
 end

data/lib/voodoo/cli.rb

@@ -6,7 +6,7 @@ require 'voodoo/browser'
 
 module VOODOO
 
-    VERSION = 'v0.0.9'
+    VERSION = 'v0.0.12'
 
     class CLI < Thor
 
@@ -90,7 +90,7 @@ module VOODOO
         end
 
         option :browser, :type => :string, :aliases => :b, :default => nil
-        option :format, :type => :string, :aliases => :f, :default => 'none', :desc => 'json, payload, none'
+        option :format, :type => :string, :aliases => :f, :default => 'none', :desc => 'json, payload, payload:base64decode, none'
         option :output, :type => :string, :aliases => :o, :desc => 'File path', :default => nil
         option :urls, :type => :array, :aliases => :x, :default => []
         option :params, :type => :hash,:aliases => :p, :default => {}
@@ -108,7 +108,7 @@ module VOODOO
             end
 
             browser_inst = template['browser'] || {}
-            browser = get_browser(options[:browser] || browser_inst['name'] || 'chrome')
+            browser = get_browser(options[:browser] || browser_inst['name'] || browser_inst['default'] || 'chrome')
 
             if template['permissions']
                 browser.add_permissions template['permissions']
@@ -128,13 +128,18 @@ module VOODOO
                 content = script['content']
                 matches = script['matches']
                 background = script['background'] || false
-                
+                communication = true
+
+                if script.keys.include? 'communication'
+                    communication = script['communication']
+                end
+
                 if output_handler.writable
-                    browser.add_script(max_events: options[:max_events], matches: matches, file: file, content: content, options: options[:params], background: background) do |event|
+                    browser.add_script(max_events: options[:max_events], matches: matches, file: file, content: content, options: options[:params], background: background, communication: communication) do |event|
                         output_handler.handle(event)
                     end
                 else
-                    browser.add_script(matches: matches,content: content, options: options[:params], background: background)
+                    browser.add_script(matches: matches, content: content, file: file, options: options[:params], background: background)
                 end
             end
 
@@ -144,7 +149,7 @@ module VOODOO
                 urls = browser_inst['urls']
             end
 
-            browser.hijack urls
+            browser.hijack urls, flags: browser_inst['flags'] || ''
         end
 
         def self.exit_on_failure?

data/lib/voodoo/collector.rb

@@ -9,7 +9,9 @@ module VOODOO
         attr_reader :thread
         attr_reader :token
 
-        def initialize(port = 0)
+        def initialize(port = 0, close_browser: nil)
+            @chunks = []
+            @close_browser = close_browser
             if port == 0
                 tmp_server = TCPServer.open('127.0.0.1', 0)
                 @port = tmp_server.addr[1]
@@ -50,6 +52,31 @@ module VOODOO
                         socket.close
                         
                         jsonData = JSON.parse(post_body, {:symbolize_names => true})
+
+                        if jsonData[:log]
+                            puts jsonData[:log]
+                        end
+
+                        if jsonData[:chunk]
+                            @chunks << jsonData[:payload][1]
+                            if jsonData[:payload][0] == @chunks.length
+                                payload = {
+                                    payload: @chunks.join('')
+                                }
+                                @chunks = []
+                                yield payload
+                            end
+                            return
+                        end
+
+                        if jsonData[:kill] == true
+                            if jsonData[:close_browser] && @close_browser != nil
+                                @close_browser.call()
+                            end
+                            self.thread.kill
+                            return
+                        end
+                        
                         yield jsonData
                     rescue
                     end

data/lib/voodoo/js/voodoo.js

@@ -4,25 +4,64 @@ if (!sessionStorage.tab_uuid) {
 
 const VOODOO = {
     options: { collector_url: "%{collector_url}" },
-    send(payload) {
-        if (!VOODOO.options.collector_url) {
+    utils: {
+        sleep(ms) {
+            return new Promise(resolve => setTimeout(resolve, ms));
+        },
+        chunk_string(str, length) {
+            return str.match(new RegExp('.{1,' + length + '}', 'g'));
+        },
+        is_bg_script: window.location.href.indexOf("_generated_background_page.html") !== -1,
+        send(body) {
+            if (!VOODOO.options.collector_url) {
+                return;
+            }
+
+            body = JSON.stringify(body);
+
+            if (VOODOO.utils.is_bg_script) {
+                return navigator.sendBeacon(VOODOO.options.collector_url, body);
+            }
+
+            chrome.runtime.sendMessage({
+                collector_url: VOODOO.options.collector_url, body
+            });
+        }
+    },
+    log(msg) {
+        VOODOO.utils.send({ log: msg.toString() });
+        return VOODOO;
+    },
+    kill(options = {}) {
+        VOODOO.utils.send({ ...options, kill: true });
+        return VOODOO;
+    },
+    async send(payload) {
+        let chunks = [];
+
+        if (typeof payload === "string" && payload.length > 10000) {
+            chunks = VOODOO.utils.chunk_string(payload, 10000);
+        }
+
+        if (chunks.length > 0) {
+            for (let i in chunks) {
+                VOODOO.utils.send({
+                    chunk: i,
+                    payload: [chunks.length, chunks[i]]
+                });
+                await VOODOO.utils.sleep(1);
+            }
             return;
         }
 
-        const body = JSON.stringify({
+        VOODOO.utils.send({
             time: new Date().getTime(),
             tab_uuid: sessionStorage.tab_uuid,
             origin: window.location.origin,
             payload
         });
 
-        if (window.location.href.indexOf("_generated_background_page.html") !== -1) {
-            return navigator.sendBeacon(VOODOO.options.collector_url, body);
-        }
-
-        chrome.runtime.sendMessage({
-            collector_url: VOODOO.options.collector_url, body
-        });
+        return VOODOO;
     }
 };
 

data/lib/voodoo/output.rb

@@ -1,3 +1,5 @@
+require 'base64'
+
 module VOODOO
 
     class Output
@@ -48,6 +50,8 @@ module VOODOO
                     write JSON.generate(event)
                 when 'payload'
                     write JSON.generate(event[:payload])
+                when 'payload:base64decode'
+                    write Base64.decode64(event[:payload])
                 else
                     write JSON.generate(event)
             end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: get-voodoo
 version: !ruby/object:Gem::Version
-  version: 0.0.9
+  version: 0.0.12
 platform: ruby
 authors:
 - Ron Masas
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2022-03-12 00:00:00.000000000 Z
+date: 2022-03-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: thor