lua-obfuscator 1.0.0 → 1.0.2

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

@@ -1,250 +0,0 @@
--- 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/src/prometheus/src/prometheus/randomLiterals.lua

@@ -1,41 +0,0 @@
--- 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/src/prometheus/src/prometheus/randomStrings.lua

@@ -1,24 +0,0 @@
-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/src/prometheus/src/prometheus/scope.lua

@@ -1,332 +0,0 @@
--- 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;