jsobfu 0.1.2 → 0.1.3

checksums.yaml

@@ -1,15 +1,15 @@
 ---
 !binary "U0hBMQ==":
   metadata.gz: !binary |-
-    MzdlMDRlNzdmNTVjM2Y2NTk0NGIyNjZmOGFjMGU4YWZlOGNhNmU5OA==
+    M2RiZjA5MGU2NDVkNDk1NmI5NDc4ZWM2YTkyM2JhMWU3YzNjN2Q1Mg==
   data.tar.gz: !binary |-
-    MTJjMzU2ZDJjMGU2ODZkZjhkNWNhYjBhYmQ5MWMyNDUwNzY2MWM3Ng==
+    MmEzZjk4NjY5MWExOWY3ODNjNTNhMGI3N2QwNDhiZWM3NjViM2VjNA==
 SHA512:
   metadata.gz: !binary |-
-    ZmQ0MjBiZWMyZDhkMWJkOWY1MmU3MmRjNmQwNDhhODYyMzFmMDU2ZTY3MzNi
-    YmQ2Y2E1MjhlNTI1ZTBhYTZlY2YzYWVlZTUwZmRmZDljNWEwM2M2ZTI0YTRk
-    ZGM4MjQwMzhhOTZlMzc0Y2YzYmNmNDcxNmRjYzU1YzY1ZDM1ZmU=
+    NzUwYzZhNzIzZTQxYzcxMGE1NTkyMTYwZjMzNGJhZDViYmZmYjM0NzY5ODY4
+    NGFiODQxYWNkYzViMDI4NjYyYTg3OTAwNTg3YTY1NjRjNzQ2ZTE1ZDViMDdj
+    ZmMyMGVhZjljODlkZTU0ZTYzZTljM2M2OWE2MmU1YjQzZjQ4NDE=
   data.tar.gz: !binary |-
-    ZmE4Yjg0YWE4OGNkMTdjYzc0ODg2MmFjZDJlMGEzZDM0YmZkNTQ2ZTAzZGMy
-    MDE5YThmNTNlYTg2Y2Q5YjM4MmE1ZDI4Y2NiNTgxMWE4ZjhhZTBmMTYwNzVi
-    MTE5NDczMjE0MzcyMGRhZDEyNGZlNmFkZTIxNjE2MGYwNDZkZTI=
+    YzhjMDk4NDY3MGU2OWQ1NzM4MGJhOWQ0NjQ2ZDk1ZGE3ZDUyOThlYjA4NmM4
+    NmRjOWEyMTUyMjc2YmNhYzAyYzE4Y2Q5MzExYzM1MzY1MjE2MGUwNjNmMTdk
+    YzQ5ZTQ2MzZjZTc4NjJlMzEzMWJmNDc2NzAzOTRiOTVlODc1Njc=

data/lib/jsobfu.rb

@@ -14,12 +14,9 @@ class JSObfu
 
   # Saves +code+ for later obfuscation with #obfuscate
   # @param code [#to_s] the code to obfuscate
-  # @param opts [Hash] the options hash
-  # @option opts [Integer] number of times to run the obfuscator on this code (1)
-  def initialize(code, opts={})
+  def initialize(code)
     @code = code.to_s
     @scope = Scope.new
-    @iterations = opts.fetch(:iterations, 1).to_i
   end
 
   # Add +str+ to the un-obfuscated code.
@@ -41,22 +38,38 @@ class JSObfu
   # Parse and obfuscate
   #
   # @param opts [Hash] the options hash
-  # @option opts [Boolean] :strip_whitespace allow whitespace in the output code
-  #
+  # @option opts [Boolean] :strip_whitespace removes unnecessary whitespace from
+  #   the output code (true)
+  # @option opts [Integer] :iterations number of times to run the
+  #   obfuscator on this code (1)
   # @return [String] if successful
   def obfuscate(opts={})
-    @iterations.times do |i|
-      @obfuscator = JSObfu::Obfuscator.new(scope: @scope)
-      @code = @obfuscator.accept(ast).to_s
-      if opts.fetch(:strip_whitespace, true)
+    iterations = opts.fetch(:iterations, 1).to_i
+    strip_whitespace = opts.fetch(:strip_whitespace, true)
+
+    iterations.times do |i|
+      obfuscator = JSObfu::Obfuscator.new(scope: @scope)
+      @code = obfuscator.accept(ast).to_s
+      if strip_whitespace
         @code.gsub!(/(^\s+|\s+$)/, '')
         @code.delete!("\n")
         @code.delete!("\r")
       end
 
-      unless i == @iterations-1
+      new_renames = obfuscator.renames.dup
+      if @renames
+        # "patch up" the renames after each iteration
+        @renames.each do |key, prev_rename|
+          @renames[key] = new_renames[prev_rename]
+        end
+      else
+        # on first iteration, take the renames as-is
+        @renames = new_renames
+      end
+
+      unless i == iterations-1
         @scope = Scope.new
-        @ast = nil
+        @ast = nil # force a re-parse
       end
     end
 
@@ -68,8 +81,8 @@ class JSObfu
   # @param [String] sym the name of the variable or function 
   # @return [String] the obfuscated name
   def sym(sym)
-    raise RuntimeError, "Must obfuscate before calling #sym" if @obfuscator.nil?
-    @obfuscator.renames[sym.to_s]
+    raise RuntimeError, "Must obfuscate before calling #sym" if @renames.nil?
+    @renames[sym.to_s]
   end
 
 protected

data/spec/jsobfu_spec.rb

@@ -5,9 +5,13 @@ describe JSObfu do
   TEST_STRING = 'var x; function y() {};'
 
   subject(:jsobfu) do
-    instance = described_class.new(TEST_STRING)
-    instance.obfuscate
-    instance
+    described_class.new(TEST_STRING)
+  end
+
+  let(:iterations) { 1 }
+
+  before do
+    jsobfu.obfuscate(iterations: iterations)
   end
 
   describe '#sym' do
@@ -22,6 +26,22 @@ describe JSObfu do
         expect(jsobfu.sym('YOLOSWAG')).to be_nil
       end
     end
+
+    context 'when iterations: 2 is passed to obfuscate()' do
+      let(:iterations) { 2 }
+
+      context 'when given the string "x"' do
+        it 'returns some string' do
+          expect(jsobfu.sym('x')).not_to be_nil
+        end
+      end
+
+      context 'when given the string "YOLOSWAG"' do
+        it 'returns nil' do
+          expect(jsobfu.sym('YOLOSWAG')).to be_nil
+        end
+      end
+    end
   end
 
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: jsobfu
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - James Lee