lua-obfuscator 1.0.0

package/dist/prometheus/src/prometheus/pipeline.lua

@@ -0,0 +1,250 @@
+-- This Script is Part of the Prometheus Obfuscator by Levno_710
+--
+-- pipeline.lua
+--
+-- This Script Provides a Configurable Obfuscation Pipeline that can obfuscate code using different Modules
+-- These Modules can simply be added to the pipeline
+
+local config = require("config");
+local Ast    = require("prometheus.ast");
+local Enums  = require("prometheus.enums");
+local util = require("prometheus.util");
+local Parser = require("prometheus.parser");
+local Unparser = require("prometheus.unparser");
+local logger = require("logger");
+
+local NameGenerators = require("prometheus.namegenerators");
+
+local Steps = require("prometheus.steps");
+
+local lookupify = util.lookupify;
+local LuaVersion = Enums.LuaVersion;
+local AstKind = Ast.AstKind;
+
+-- On Windows os.clock can be used. On other Systems os.time must be used for benchmarking
+local isWindows = package and package.config and type(package.config) == "string" and package.config:sub(1,1) == "\\";
+local function gettime()
+	if isWindows then
+		return os.clock();
+	else
+		return os.time();
+	end
+end
+
+local Pipeline = {
+	NameGenerators = NameGenerators;
+	Steps = Steps;
+	DefaultSettings = {
+		LuaVersion = LuaVersion.LuaU; -- The Lua Version to use for the Tokenizer, Parser and Unparser
+		PrettyPrint = false; -- Note that Pretty Print is currently not producing Pretty results
+		Seed = 0; -- The Seed. 0 or below uses the current time as a seed
+		VarNamePrefix = ""; -- The Prefix that every variable will start with
+	}
+}
+
+
+function Pipeline:new(settings)
+	local luaVersion = settings.luaVersion or settings.LuaVersion or Pipeline.DefaultSettings.LuaVersion;
+	local conventions = Enums.Conventions[luaVersion];
+	if(not conventions) then
+		logger:error("The Lua Version \"" .. luaVersion 
+			.. "\" is not recognised by the Tokenizer! Please use one of the following: \"" .. table.concat(util.keys(Enums.Conventions), "\",\"") .. "\"");
+	end
+	
+	local prettyPrint = settings.PrettyPrint or Pipeline.DefaultSettings.PrettyPrint;
+	local prefix = settings.VarNamePrefix or Pipeline.DefaultSettings.VarNamePrefix;
+	local seed = settings.Seed or 0;
+	
+	local pipeline = {
+		LuaVersion = luaVersion;
+		PrettyPrint = prettyPrint;
+		VarNamePrefix = prefix;
+		Seed = seed;
+		parser = Parser:new({
+			LuaVersion = luaVersion;
+		});
+		unparser = Unparser:new({
+			LuaVersion = luaVersion;
+			PrettyPrint = prettyPrint;
+			Highlight = settings.Highlight;
+		});
+		namegenerator = Pipeline.NameGenerators.MangledShuffled;
+		conventions = conventions;
+		steps = {};
+	}
+	
+	setmetatable(pipeline, self);
+	self.__index = self;
+	
+	return pipeline;
+end
+
+function Pipeline:fromConfig(config)
+	config = config or {};
+	local pipeline = Pipeline:new({
+		LuaVersion    = config.LuaVersion or LuaVersion.Lua51;
+		PrettyPrint   = config.PrettyPrint or false;
+		VarNamePrefix = config.VarNamePrefix or "";
+		Seed          = config.Seed or 0;
+	});
+
+	pipeline:setNameGenerator(config.NameGenerator or "MangledShuffled")
+
+	-- Add all Steps defined in Config
+	local steps = config.Steps or {};
+	for i, step in ipairs(steps) do
+		if type(step.Name) ~= "string" then
+			logger:error("Step.Name must be a String");
+		end
+		local constructor = pipeline.Steps[step.Name];
+		if not constructor then
+			logger:error(string.format("The Step \"%s\" was not found!", step.Name));
+		end
+		pipeline:addStep(constructor:new(step.Settings or {}));
+	end
+
+	return pipeline;
+end
+
+function Pipeline:addStep(step)
+	table.insert(self.steps, step);
+end
+
+function Pipeline:resetSteps(step)
+	self.steps = {};
+end
+
+function Pipeline:getSteps()
+	return self.steps;
+end
+
+function Pipeline:setOption(name, value)
+	assert(false, "TODO");
+	if(Pipeline.DefaultSettings[name] ~= nil) then
+		
+	else
+		logger:error(string.format("\"%s\" is not a valid setting"));
+	end
+end
+
+function Pipeline:setLuaVersion(luaVersion)
+	local conventions = Enums.Conventions[luaVersion];
+	if(not conventions) then
+		logger:error("The Lua Version \"" .. luaVersion 
+			.. "\" is not recognised by the Tokenizer! Please use one of the following: \"" .. table.concat(util.keys(Enums.Conventions), "\",\"") .. "\"");
+	end
+	
+	self.parser = Parser:new({
+		luaVersion = luaVersion;
+	});
+	self.unparser = Unparser:new({
+		luaVersion = luaVersion;
+	});
+	self.conventions = conventions;
+end
+
+function Pipeline:getLuaVersion()
+	return self.luaVersion;
+end
+
+function Pipeline:setNameGenerator(nameGenerator)
+	if(type(nameGenerator) == "string") then
+		nameGenerator = Pipeline.NameGenerators[nameGenerator];
+	end
+	
+	if(type(nameGenerator) == "function" or type(nameGenerator) == "table") then
+		self.namegenerator = nameGenerator;
+		return;
+	else
+		logger:error("The Argument to Pipeline:setNameGenerator must be a valid NameGenerator function or function name e.g: \"mangled\"")
+	end
+end
+
+function Pipeline:apply(code, filename)
+	local startTime = gettime();
+	filename = filename or "Anonymus Script";
+	logger:info(string.format("Applying Obfuscation Pipeline to %s ...", filename));
+	-- Seed the Random Generator
+	if(self.Seed > 0) then
+		math.randomseed(self.Seed);
+	else
+		math.randomseed(os.time())
+	end
+	
+	logger:info("Parsing ...");
+	local parserStartTime = gettime();
+
+	local sourceLen = string.len(code);
+	local ast = self.parser:parse(code);
+
+	local parserTimeDiff = gettime() - parserStartTime;
+	logger:info(string.format("Parsing Done in %.2f seconds", parserTimeDiff));
+	
+	-- User Defined Steps
+	for i, step in ipairs(self.steps) do
+		local stepStartTime = gettime();
+		logger:info(string.format("Applying Step \"%s\" ...", step.Name or "Unnamed"));
+		local newAst = step:apply(ast, self);
+		if type(newAst) == "table" then
+			ast = newAst;
+		end
+		logger:info(string.format("Step \"%s\" Done in %.2f seconds", step.Name or "Unnamed", gettime() - stepStartTime));
+	end
+	
+	-- Rename Variables Step
+	self:renameVariables(ast);
+	
+	code = self:unparse(ast);
+	
+	local timeDiff = gettime() - startTime;
+	logger:info(string.format("Obfuscation Done in %.2f seconds", timeDiff));
+	
+	logger:info(string.format("Generated Code size is %.2f%% of the Source Code size", (string.len(code) / sourceLen)*100))
+	
+	return code;
+end
+
+function Pipeline:unparse(ast)
+	local startTime = gettime();
+	logger:info("Generating Code ...");
+	
+	local unparsed = self.unparser:unparse(ast);
+	
+	local timeDiff = gettime() - startTime;
+	logger:info(string.format("Code Generation Done in %.2f seconds", timeDiff));
+	
+	return unparsed;
+end
+
+function Pipeline:renameVariables(ast)
+	local startTime = gettime();
+	logger:info("Renaming Variables ...");
+	
+	
+	local generatorFunction = self.namegenerator or Pipeline.NameGenerators.mangled;
+	if(type(generatorFunction) == "table") then
+		if (type(generatorFunction.prepare) == "function") then
+			generatorFunction.prepare(ast);
+		end
+		generatorFunction = generatorFunction.generateName;
+	end
+	
+	if not self.unparser:isValidIdentifier(self.VarNamePrefix) and #self.VarNamePrefix ~= 0 then
+		logger:error(string.format("The Prefix \"%s\" is not a valid Identifier in %s", self.VarNamePrefix, self.LuaVersion));
+	end
+
+	local globalScope = ast.globalScope;
+	globalScope:renameVariables({
+		Keywords = self.conventions.Keywords;
+		generateName = generatorFunction;
+		prefix = self.VarNamePrefix;
+	});
+	
+	local timeDiff = gettime() - startTime;
+	logger:info(string.format("Renaming Done in %.2f seconds", timeDiff));
+end
+
+
+
+
+return Pipeline;

package/dist/prometheus/src/prometheus/randomLiterals.lua

@@ -0,0 +1,41 @@
+-- This Script is Part of the Prometheus Obfuscator by Levno_710
+--
+-- Library for Creating Random Literals
+
+local Ast = require("prometheus.ast");
+local RandomStrings = require("prometheus.randomStrings");
+
+local RandomLiterals = {};
+
+local function callNameGenerator(generatorFunction, ...)
+	if(type(generatorFunction) == "table") then
+		generatorFunction = generatorFunction.generateName;
+	end
+	return generatorFunction(...);
+end
+
+function RandomLiterals.String(pipeline)
+    return Ast.StringExpression(callNameGenerator(pipeline.namegenerator, math.random(1, 4096)));
+end
+
+function RandomLiterals.Dictionary()
+    return RandomStrings.randomStringNode(true);
+end
+
+function RandomLiterals.Number()
+    return Ast.NumberExpression(math.random(-8388608, 8388607));
+end
+
+function RandomLiterals.Any(pipeline)
+    local type = math.random(1, 3);
+    if type == 1 then
+        return RandomLiterals.String(pipeline);
+    elseif type == 2 then
+        return RandomLiterals.Number();
+    elseif type == 3 then
+        return RandomLiterals.Dictionary();
+    end
+end
+
+
+return RandomLiterals;

package/dist/prometheus/src/prometheus/randomStrings.lua

@@ -0,0 +1,24 @@
+local Ast, utils = require("prometheus.ast"), require("prometheus.util");
+local charset = utils.chararray("qwertyuiopasdfghjklzxcvbnmQWERTYUIOPASDFGHJKLZXCVBNM1234567890")
+
+local function randomString(wordsOrLen)
+	if type(wordsOrLen) == "table" then
+		return wordsOrLen[math.random(1, #wordsOrLen)];
+	end
+
+	wordsOrLen = wordsOrLen or math.random(2, 15);
+	if wordsOrLen > 0 then
+		return randomString(wordsOrLen - 1) .. charset[math.random(1, #charset)]
+	else
+		return ""
+	end
+end
+
+local function randomStringNode(wordsOrLen)
+	return Ast.StringExpression(randomString(wordsOrLen))
+end
+
+return {
+	randomString = randomString,
+	randomStringNode = randomStringNode,
+}

package/dist/prometheus/src/prometheus/scope.lua

@@ -0,0 +1,332 @@
+-- This Script is Part of the Prometheus Obfuscator by Levno_710
+--
+-- scope.lua
+
+local logger = require("logger");
+local config = require("config");
+
+local Scope = {};
+
+local scopeI = 0;
+local function nextName()
+	scopeI = scopeI + 1;
+	return "local_scope_" .. tostring(scopeI);
+end
+
+local function generateWarning(token, message)
+	return "Warning at Position " .. tostring(token.line) .. ":" .. tostring(token.linePos) .. ", " .. message;
+end
+
+-- Create a new Local Scope
+function Scope:new(parentScope, name)
+	local scope = {
+		isGlobal = false,
+		parentScope = parentScope,
+		variables = {},
+		referenceCounts = {};
+		variablesLookup = {},
+		variablesFromHigherScopes = {},
+		skipIdLookup = {};
+		name = name or nextName(),
+		children = {},
+		level = parentScope.level and (parentScope.level + 1) or 1;
+	}
+	
+	setmetatable(scope, self);
+	self.__index = self;
+	parentScope:addChild(scope);
+	return scope;
+end
+
+-- Create a new Global Scope
+function Scope:newGlobal()
+	local scope = {
+		isGlobal = true,
+		parentScope = nil,
+		variables = {},
+		variablesLookup = {};
+		referenceCounts = {};
+		skipIdLookup = {};
+		name = "global_scope",
+		children = {},
+		level = 0,
+	};
+	
+	setmetatable(scope, self);
+	self.__index = self;
+	
+	return scope;
+end
+
+-- Returns the Parent Scope
+function Scope:getParent(parentScope)
+	return self.parentScope;
+end
+
+function Scope:setParent(parentScope)
+	self.parentScope:removeChild(self);
+	parentScope:addChild(self);
+	self.parentScope = parentScope;
+	self.level = parentScope.level + 1;
+end
+
+local next_name_i = 1;
+-- Adds a Variable to the scope and returns the variable id, if no name is passed then a name is generated
+function Scope:addVariable(name, token)
+	if (not name) then
+		name = string.format("%s%i", config.IdentPrefix, next_name_i);
+		next_name_i = next_name_i + 1;
+	end
+	
+	if self.variablesLookup[name] ~= nil then
+		if(token) then
+			logger:warn(generateWarning(token, "the variable \"" .. name .. "\" is already defined in that scope"));
+		else
+			logger:error(string.format("A variable with the name \"%s\" was already defined, you should have no variables starting with \"%s\"", name, config.IdentPrefix));
+		end
+		
+		--return self.variablesLookup[name];
+	end
+	
+	table.insert(self.variables, name);
+	local id = #self.variables;
+	self.variablesLookup[name] = id;
+	return id;
+end
+
+function Scope:enableVariable(id)
+	local name = self.variables[id];
+	self.variablesLookup[name] = id;
+end
+
+function Scope:addDisabledVariable(name, token)
+	if (not name) then
+		name = string.format("%s%i", config.IdentPrefix, next_name_i);
+		next_name_i = next_name_i + 1;
+	end
+	
+	if self.variablesLookup[name] ~= nil then
+		if(token) then
+			logger:warn(generateWarning(token, "the variable \"" .. name .. "\" is already defined in that scope"));
+		else
+			logger:warn(string.format("a variable with the name \"%s\" was already defined", name));
+		end
+		
+		--return self.variablesLookup[name];
+	end
+	
+	table.insert(self.variables, name);
+	local id = #self.variables;
+	return id;
+end
+
+function Scope:addIfNotExists(id)
+	if(not self.variables[id]) then
+		local name = string.format("%s%i", config.IdentPrefix, next_name_i);
+		next_name_i = next_name_i + 1;
+		self.variables[id] = name;
+		self.variablesLookup[name] = id;
+	end
+	return id;
+end
+
+-- Returns wether the variable is defined in this Scope
+function Scope:hasVariable(name)
+	if(self.isGlobal) then
+		if self.variablesLookup[name] == nil then
+			self:addVariable(name);
+		end
+		return true;
+	end
+	return self.variablesLookup[name] ~= nil;
+end
+
+-- Get List of all Variables defined in this Scope
+function Scope:getVariables()
+	return self.variables;
+end
+
+function Scope:resetReferences(id)
+	self.referenceCounts[id] = 0;
+end
+
+function Scope:getReferences(id)
+	return self.referenceCounts[id] or 0;
+end
+
+function Scope:removeReference(id)
+	self.referenceCounts[id] = (self.referenceCounts[id] or 0) - 1;
+end
+
+function Scope:addReference(id)
+	self.referenceCounts[id] = (self.referenceCounts[id] or 0) + 1;
+end
+
+-- Resolve the scope of a variable by name
+function Scope:resolve(name)
+	if(self:hasVariable(name)) then
+		return self, self.variablesLookup[name];
+	end
+	assert(self.parentScope, "No Global Variable Scope was Created! This should not be Possible!");
+	local scope, id = self.parentScope:resolve(name);
+	self:addReferenceToHigherScope(scope, id, nil, true);
+	return scope, id;
+end
+
+function Scope:resolveGlobal(name)
+	if(self.isGlobal and self:hasVariable(name)) then
+		return self, self.variablesLookup[name];
+	end
+	assert(self.parentScope, "No Global Variable Scope was Created! This should not be Possible!");
+	local scope, id = self.parentScope:resolveGlobal(name);
+	self:addReferenceToHigherScope(scope, id, nil, true);
+	return scope, id;
+end
+
+-- Returns the name of an Variable by id - this is used for unparsing
+function Scope:getVariableName(id)
+	return self.variables[id];
+end
+
+-- Remove A Variable from this Scope
+function Scope:removeVariable(id)
+	local name = self.variables[id];
+	self.variables[id] = nil;
+	self.variablesLookup[name] = nil;
+	self.skipIdLookup[id] = true;
+end
+
+-- Add a Children Scope
+function Scope:addChild(scope)
+	-- This will add all References from that Scope to higher Scopes. Note that the higher scopes may only be global
+	for scope, ids in pairs(scope.variablesFromHigherScopes) do
+		for id, count in pairs(ids) do
+			if count and count > 0 then
+				self:addReferenceToHigherScope(scope, id, count);
+			end
+		end
+	end
+	table.insert(self.children, scope);
+end
+
+function Scope:clearReferences()
+	self.referenceCounts = {};
+	self.variablesFromHigherScopes = {};
+end
+
+function Scope:removeChild(child)
+	for i, v in ipairs(self.children) do
+		if(v == child) then
+			-- This will add all References from that Scope to higher Scopes. Note that the higher scopes may only be global
+			for scope, ids in pairs(v.variablesFromHigherScopes) do
+				for id, count in pairs(ids) do
+					if count and count > 0 then
+						self:removeReferenceToHigherScope(scope, id, count);
+					end
+				end
+			end
+			return table.remove(self.children, i);
+		end
+	end
+end
+
+function Scope:getMaxId()
+	return #self.variables;
+end
+
+function Scope:addReferenceToHigherScope(scope, id, n, b)
+	n = n or 1;
+	if self.isGlobal then
+		if not scope.isGlobal then
+			logger:error(string.format("Could not resolve Scope \"%s\"", scope.name))
+		end
+		return
+	end
+	if scope == self then
+		self.referenceCounts[id] = (self.referenceCounts[id] or 0) + n;
+		return
+	end
+	if not self.variablesFromHigherScopes[scope] then
+		self.variablesFromHigherScopes[scope] = {};
+	end
+	local scopeReferences = self.variablesFromHigherScopes[scope];
+	if scopeReferences[id] then
+		scopeReferences[id]  = scopeReferences[id] + n;
+	else
+		scopeReferences[id] = n;
+	end
+	if not b then
+		self.parentScope:addReferenceToHigherScope(scope, id, n);
+	end
+end
+
+function Scope:removeReferenceToHigherScope(scope, id, n, b)
+	n = n or 1;
+	if self.isGlobal then
+		return
+	end
+	if scope == self then
+		self.referenceCounts[id] = (self.referenceCounts[id] or 0) - n;
+		return
+	end
+	if not self.variablesFromHigherScopes[scope] then
+		self.variablesFromHigherScopes[scope] = {};
+	end
+	local scopeReferences = self.variablesFromHigherScopes[scope];
+	if scopeReferences[id] then
+		scopeReferences[id]  = scopeReferences[id] - n;
+	else
+		scopeReferences[id] = 0;
+	end
+	if not b then
+		self.parentScope:removeReferenceToHigherScope(scope, id, n);
+	end
+end
+
+-- Rename Variables from that scope downwards
+-- this function needs a settings object with the following properties
+-- Keywords => forbidden Variable Names
+-- generateName(id, scope, originalName) => function to generate unique variable name based on the id and scope
+function Scope:renameVariables(settings)
+	if(not self.isGlobal) then
+		local prefix = settings.prefix or "";
+		local forbiddenNamesLookup = {};
+		for _, keyword in pairs(settings.Keywords) do
+			forbiddenNamesLookup[keyword] = true;
+		end
+		
+		for scope, ids in pairs(self.variablesFromHigherScopes) do
+			for id, count in pairs(ids) do
+				if count and count > 0 then
+					local name = scope:getVariableName(id);
+					forbiddenNamesLookup[name] = true;
+				end
+			end
+		end
+		
+		self.variablesLookup = {};
+		
+		local i = 0;
+		for id, originalName in pairs(self.variables) do
+			if(not self.skipIdLookup[id] and (self.referenceCounts[id] or 0) >= 0) then
+				local name;
+				repeat
+					name = prefix .. settings.generateName(i, self, originalName);
+					if name == nil then
+						name = originalName;
+					end
+					i = i + 1;
+				until not forbiddenNamesLookup[name];
+
+				self.variables[id] = name;
+				self.variablesLookup[name] = id;
+			end
+		end
+	end
+	
+	for _, scope in pairs(self.children) do
+		scope:renameVariables(settings);
+	end
+end
+
+return Scope;