Merge pull request #104 from oxince/master

Author: levno-710

src/presets.lua

@@ -34,6 +34,12 @@ return {
         Seed = 0;
         -- Obfuscation steps
         Steps = {
+            {
+                Name = "WatermarkCheck";
+                Settings = {
+                    Content = "This Script was obfuscated using the Weak Preset of Prometheus Obfuscator by Levno_710";
+                }
+            },
             {
                 Name = "Vmify";
                 Settings = {
@@ -68,6 +74,12 @@ return {
         Seed = 0;
         -- Obfuscation steps
         Steps = {
+            {
+                Name = "WatermarkCheck";
+                Settings = {
+                    Content = "This Script was obfuscated using the Medium Preset of Prometheus Obfuscator by Levno_710";
+                }
+            },
             {
                 Name = "EncryptStrings";
                 Settings = {
@@ -108,12 +120,6 @@ return {
 
                 }
             },
-            {
-                Name = "Watermark";
-                Settings = {
-                    Content = "This Script was obfuscated using the Medium Preset of Prometheus Obfuscator by Levno_710"
-                }
-            }
         }
     };
     ["Strong"] = {
@@ -129,6 +135,12 @@ return {
         Seed = 0;
         -- Obfuscation steps
         Steps = {
+            {
+                Name = "WatermarkCheck";
+                Settings = {
+                    Content = "This Script was obfuscated using the Strong Preset of Prometheus Obfuscator by Levno_710";
+                }
+            },
             {
                 Name = "Vmify";
                 Settings = {

src/prometheus/steps.lua

@@ -8,5 +8,5 @@ return {
 	EncryptStrings 			 = require("prometheus.steps.EncryptStrings");
 	NumbersToExpressions = require("prometheus.steps.NumbersToExpressions");
 	AddVararg 					 = require("prometheus.steps.AddVararg");
-	Watermark 	         = require("prometheus.steps.Watermark");
+	WatermarkCheck		   = require("prometheus.steps.WatermarkCheck");
 }

src/prometheus/steps/Watermark.lua

@@ -19,6 +19,12 @@ Watermark.SettingsDescriptor = {
     type = "string",
     default = "This Script is Part of the Prometheus Obfuscator by Levno_710",
   },
+  CustomVariable = {
+    name = "Custom Variable",
+    description = "The Variable that will be used for the Watermark",
+    type = "string",
+    default = "_WATERMARK",
+  }
 }
 
 function Watermark:init(settings)
@@ -28,8 +34,30 @@ end
 function Watermark:apply(ast)
   local body = ast.body;
   if string.len(self.Content) > 0 then
-    local watermark = body.scope:addVariable();
-    table.insert(body.statements, 1, Ast.LocalVariableDeclaration(body.scope, {watermark}, {Ast.StringExpression(self.Content)}));
+    local scope, variable = ast.globalScope:resolve(self.CustomVariable);
+    local watermark = Ast.AssignmentVariable(ast.globalScope, variable);
+
+    -- body.scope:addReferenceToHigherScope(ast.globalScope, variable);
+    -- table.insert(body.statements, 1, Ast.AssignmentStatement({watermark}, {Ast.StringExpression(self.Content)}));
+  
+    local functionScope = Scope:new(body.scope);
+    functionScope:addReferenceToHigherScope(ast.globalScope, variable);
+    
+    local arg = functionScope:addVariable();
+    local statement = Ast.PassSelfFunctionCallStatement(Ast.StringExpression(self.Content), "gsub", {
+      Ast.StringExpression(".+"),
+      Ast.FunctionLiteralExpression({
+        Ast.VariableExpression(functionScope, arg)
+      }, Ast.Block({
+        Ast.AssignmentStatement({
+          watermark
+        }, {
+          Ast.VariableExpression(functionScope, arg)
+        })
+      }, functionScope))
+    });
+
+    table.insert(ast.body.statements, 1, statement)
   end
 end
 

src/prometheus/steps/WatermarkCheck.lua

@@ -0,0 +1,50 @@
+-- This Script is Part of the Prometheus Obfuscator by Levno_710
+--
+-- WatermarkCheck.lua
+--
+-- This Script provides a Step that will add a watermark to the script
+
+local Step = require("prometheus.step");
+local Ast = require("prometheus.ast");
+local Scope = require("prometheus.scope");
+local Watermark = require("prometheus.steps.Watermark");
+
+local WatermarkCheck = Step:extend();
+WatermarkCheck.Description = "This Step will add a watermark to the script";
+WatermarkCheck.Name = "WatermarkCheck";
+
+WatermarkCheck.SettingsDescriptor = {
+  Content = {
+    name = "Content",
+    description = "The Content of the WatermarkCheck",
+    type = "string",
+    default = "This Script is Part of the Prometheus Obfuscator by Levno_710",
+  },
+}
+
+local function callNameGenerator(generatorFunction, ...)
+	if(type(generatorFunction) == "table") then
+		generatorFunction = generatorFunction.generateName;
+	end
+	return generatorFunction(...);
+end
+
+function WatermarkCheck:init(settings)
+
+end
+
+function WatermarkCheck:apply(ast, pipeline)
+  self.CustomVariable = "_" .. callNameGenerator(pipeline.namegenerator, math.random(10000000000, 100000000000));
+  pipeline:addStep(Watermark:new(self));
+
+  local body = ast.body;
+  local watermarkExpression = Ast.StringExpression(self.Content);
+  local scope, variable = ast.globalScope:resolve(self.CustomVariable);
+  local watermark = Ast.VariableExpression(ast.globalScope, variable);
+  local notEqualsExpression = Ast.NotEqualsExpression(watermark, watermarkExpression);
+  local ifBody = Ast.Block({Ast.ReturnStatement({})}, Scope:new(ast.body.scope));
+
+  table.insert(body.statements, 1, Ast.IfStatement(notEqualsExpression, ifBody, {}, nil));
+end
+
+return WatermarkCheck;