get-voodoo 0.0.11 → 0.0.12

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 965c3f91b6855d9b381e18ced1bfee172e5273d3180d50ef7a2892a66adbd30d
-  data.tar.gz: e20ea428ba96e9c5f82ab31a5bab21761c189cd451d05b31bbc38ec0f857b56d
+  metadata.gz: 6906a5bfd817cfdb86c525ffc86677ef7185991a70d0c678df07e41ee2394a81
+  data.tar.gz: 8adb2abcc70fe837e0c5c92cfc306d61a4e14c844bdf50009f0765b458bcc479
 SHA512:
-  metadata.gz: a8a34ddd9b063f44353195203f633f16e9fcdb872cbab1b52662b2e9582a9af5ba4ced78c708cafebd9408196a5dfd7034296b141acc11b13bdc9a05b27920b9
-  data.tar.gz: 03c5e0884a62321e2053b93a3f8d606252bb127aded3748a997f33002d56070720e9a63a835ab96abbf2e70c1c99b8d0fb4550e11348c4ccd2cdc6921cfe4d2e
+  metadata.gz: 2021b65dc3154a76238cb3cb2015911cda5ddd3aff056814857670925d67ad471d0d8bf4ee4d812940a070819dbc35ac1899c654af82dedfbf4af8eb68727674
+  data.tar.gz: 2febeb01c329895e3313d309e32ce7db3b247f0ea092a4ed0502bfe0d7f97c5050041f62b42e4a7236871869e114e56d8458bdf9188201eb9eeb4582c4d64131

data/lib/voodoo/browser.rb

@@ -49,11 +49,15 @@ module VOODOO
             @extension.manifest[:permissions] += permissions
         end
 
-        def hijack(urls = [], flags: '')
+        def close_browser
             # kill the browser process twise, to bypass close warning
             `pkill -a -i "#{@process_name}"`
             `pkill -a -i "#{@process_name}"`
             sleep 0.2
+        end
+
+        def hijack(urls = [], flags: '')
+            close_browser()
 
             urls = [urls] unless urls.kind_of? Array
             urls = urls.uniq
@@ -92,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
@@ -107,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)
@@ -158,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.11'
+    VERSION = 'v0.0.12'
 
     class CLI < Thor
 
@@ -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,9 +128,14 @@ 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

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;
     }
 };
 

metadata

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