Add Encrypt Strings Step

Author: levno-710

src/prometheus/steps/EncryptStrings.lua

@@ -0,0 +1,217 @@
+-- This Script is Part of the Prometheus Obfuscator by Levno_710
+--
+-- EncryptStrings.lua
+--
+-- This Script provides a Simple Obfuscation Step that encrypts strings
+
+local Step = require("prometheus.step")
+local Ast = require("prometheus.ast")
+local Scope = require("prometheus.scope")
+local RandomStrings = require("prometheus.randomStrings")
+local Parser = require("prometheus.parser")
+local Enums = require("prometheus.enums")
+local logger = require("logger")
+local visitast = require("prometheus.visitast");
+local util     = require("prometheus.util")
+local AstKind = Ast.AstKind;
+
+local EncryptStrings = Step:extend()
+EncryptStrings.Description = "This Step will encrypt strings within your Program."
+EncryptStrings.Name = "Encrypt Strings"
+
+EncryptStrings.SettingsDescriptor = {}
+
+function EncryptStrings:init(settings) end
+
+
+function EncryptStrings:CreateEncrypionService()
+	local usedSeeds = {};
+
+	local secret_key_6 = math.random(0, 63) -- 6-bit  arbitrary integer (0..63)
+	local secret_key_7 = math.random(0, 127) -- 7-bit  arbitrary integer (0..127)
+	local secret_key_44 = math.random(0, 17592186044415) -- 44-bit arbitrary integer (0..17592186044415)
+
+	local floor = math.floor
+
+	local function primitive_root_257(idx)
+		local g, m, d = 1, 128, 2 * idx + 1
+		repeat
+			g, m, d = g * g * (d >= m and 3 or 1) % 257, m / 2, d % m
+		until m < 1
+		return g
+	end
+
+	local param_mul_8 = primitive_root_257(secret_key_7)
+	local param_mul_45 = secret_key_6 * 4 + 1
+	local param_add_45 = secret_key_44 * 2 + 1
+
+	local state_45 = 0
+	local state_8 = 2
+
+	local prev_values = {}
+	local function set_seed(seed_53)
+		state_45 = seed_53 % 35184372088832
+		state_8 = seed_53 % 255 + 2
+		prev_values = {}
+	end
+
+	local function gen_seed()
+		local seed;
+		repeat
+			seed = math.random(0, 35184372088832);
+		until not usedSeeds[seed];
+		return seed;
+	end
+
+	local function get_random_32()
+		state_45 = (state_45 * param_mul_45 + param_add_45) % 35184372088832
+		repeat
+			state_8 = state_8 * param_mul_8 % 257
+		until state_8 ~= 1
+		local r = state_8 % 32
+		local n = floor(state_45 / 2 ^ (13 - (state_8 - r) / 32)) % 2 ^ 32 / 2 ^ r
+		return floor(n % 1 * 2 ^ 32) + floor(n)
+	end
+
+	local function get_next_pseudo_random_byte()
+		if #prev_values == 0 then
+			local rnd = get_random_32() -- value 0..4294967295
+			local low_16 = rnd % 65536
+			local high_16 = (rnd - low_16) / 65536
+			local b1 = low_16 % 256
+			local b2 = (low_16 - b1) / 256
+			local b3 = high_16 % 256
+			local b4 = (high_16 - b3) / 256
+			prev_values = { b1, b2, b3, b4 }
+		end
+		--print(unpack(prev_values))
+		return table.remove(prev_values)
+	end
+
+	local function encrypt(str)
+		local seed = gen_seed();
+		set_seed(seed)
+		local len = string.len(str)
+		local out = {}
+		for i = 1, len do
+			out[i] = string.char((string.byte(str, i) + get_next_pseudo_random_byte()) % 256);
+		end
+		return table.concat(out), seed;
+	end
+
+    local function genCode()
+        local code = [[
+do
+	local floor = math.floor
+	local param_mul_8 = ]] .. tostring(param_mul_8) .. [[
+	local param_mul_45 = ]] .. tostring(param_mul_45) .. [[
+	local param_add_45 = ]] .. tostring(param_add_45) .. [[
+	local state_45 = 0
+	local state_8 = 2
+	local prev_values = {}
+	local function get_next_pseudo_random_byte()
+		if #prev_values == 0 then
+			state_45 = (state_45 * param_mul_45 + param_add_45) % 35184372088832
+			repeat
+				state_8 = state_8 * param_mul_8 % 257
+			until state_8 ~= 1
+			local r = state_8 % 32
+			local n = floor(state_45 / 2 ^ (13 - (state_8 - r) / 32)) % 2 ^ 32 / 2 ^ r
+			local rnd = floor(n % 1 * 2 ^ 32) + floor(n)
+			local low_16 = rnd % 65536
+			local high_16 = (rnd - low_16) / 65536
+			local b1 = low_16 % 256
+			local b2 = (low_16 - b1) / 256
+			local b3 = high_16 % 256
+			local b4 = (high_16 - b3) / 256
+			prev_values = { b1, b2, b3, b4 }
+		end
+		return table.remove(prev_values)
+	end
+
+	local realStrings = {};
+	STRINGS = setmetatable({}, {
+		__index = function(tb, idx)
+			return realStrings[idx];
+		end
+	});
+  	function DECRYPT(str, seed) 
+		if(realStrings[seed]) then
+			return seed;
+		end
+		state_45 = seed % 35184372088832
+		state_8 = seed % 255 + 2
+    	prev_values = {};
+		local len = string.len(str);
+		local out = "";
+		for i=1, len do
+			out = out .. string.char((string.byte(str, i) - get_next_pseudo_random_byte()) % 256);
+		end
+		realStrings[seed] = out;
+		return seed;
+	end
+end]]
+
+		return code;
+    end
+
+    return {
+        encrypt = encrypt,
+        param_mul_45 = param_mul_45,
+        param_mul_8 = param_mul_8,
+        param_add_45 = param_add_45,
+        genCode = genCode,
+    }
+end
+
+function EncryptStrings:apply(ast, pipeline)
+    local Encryptor = self:CreateEncrypionService();
+
+	local code = Encryptor.genCode();
+	local newAst = Parser:new({ LuaVersion = Enums.LuaVersion.Lua51 }):parse(code);
+	local doStat = newAst.body.statements[1];
+
+	local scope = ast.body.scope;
+	local decryptVar = scope:addVariable();
+	local stringsVar = scope:addVariable();
+	
+	doStat.body.scope:setParent(ast.body.scope);
+
+	visitast(newAst, nil, function(node, data)
+		if(node.kind == AstKind.FunctionDeclaration) then
+			if(node.scope:getVariableName(node.id) == "DECRYPT") then
+				data.scope:removeReferenceToHigherScope(node.scope, node.id);
+				data.scope:addReferenceToHigherScope(scope, decryptVar);
+				node.scope = scope;
+				node.id    = decryptVar;
+			end
+		end
+		if(node.kind == AstKind.AssignmentVariable or node.kind == AstKind.VariableExpression) then
+			if(node.scope:getVariableName(node.id) == "STRINGS") then
+				data.scope:removeReferenceToHigherScope(node.scope, node.id);
+				data.scope:addReferenceToHigherScope(scope, stringsVar);
+				node.scope = scope;
+				node.id    = stringsVar;
+			end
+		end
+	end)
+
+	visitast(ast, nil, function(node, data)
+		if(node.kind == AstKind.StringExpression) then
+			data.scope:addReferenceToHigherScope(scope, stringsVar);
+			data.scope:addReferenceToHigherScope(scope, decryptVar);
+			local encrypted, seed = Encryptor.encrypt(node.value);
+			return Ast.IndexExpression(Ast.VariableExpression(scope, stringsVar), Ast.FunctionCallExpression(Ast.VariableExpression(scope, decryptVar), {
+				Ast.StringExpression(encrypted), Ast.NumberExpression(seed),
+			}));
+		end
+	end)
+
+
+	-- Insert to Main Ast
+	table.insert(ast.body.statements, 1, doStat);
+	table.insert(ast.body.statements, 1, Ast.LocalVariableDeclaration(scope, util.shuffle{ decryptVar, stringsVar }, {}));
+	return ast
+end
+
+return EncryptStrings