111 lines
4.0 KiB
C#
111 lines
4.0 KiB
C#
// Copyright 2025 Code Philosophy
|
||
//
|
||
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||
// of this software and associated documentation files (the "Software"), to deal
|
||
// in the Software without restriction, including without limitation the rights
|
||
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||
// copies of the Software, and to permit persons to whom the Software is
|
||
// furnished to do so, subject to the following conditions:
|
||
//
|
||
// The above copyright notice and this permission notice shall be included in all
|
||
// copies or substantial portions of the Software.
|
||
//
|
||
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||
// SOFTWARE.
|
||
|
||
using Obfuz.Utils;
|
||
using System.Collections.Generic;
|
||
using UnityEngine.Assertions;
|
||
|
||
namespace Obfuz.EncryptionVM
|
||
{
|
||
|
||
public class VirtualMachineSimulator : EncryptorBase
|
||
{
|
||
private readonly EncryptionInstructionWithOpCode[] _opCodes;
|
||
private readonly int[] _secretKey;
|
||
|
||
public override int OpCodeCount => _opCodes.Length;
|
||
|
||
public VirtualMachineSimulator(VirtualMachine vm, byte[] byteSecretKey)
|
||
{
|
||
_opCodes = vm.opCodes;
|
||
_secretKey = KeyGenerator.ConvertToIntKey(byteSecretKey);
|
||
|
||
VerifyInstructions();
|
||
}
|
||
|
||
private void VerifyInstructions()
|
||
{
|
||
int value = 0x11223344;
|
||
for (int i = 0; i < _opCodes.Length; i++)
|
||
{
|
||
int encryptedValue = _opCodes[i].Encrypt(value, _secretKey, i);
|
||
int decryptedValue = _opCodes[i].Decrypt(encryptedValue, _secretKey, i);
|
||
//Debug.Log($"instruction type:{_opCodes[i].function.GetType()}");
|
||
Assert.AreEqual(value, decryptedValue);
|
||
}
|
||
|
||
int ops = 11223344;
|
||
int salt = 789;
|
||
Assert.AreEqual(1, Decrypt(Encrypt(1, ops, salt), ops, salt));
|
||
Assert.AreEqual(1L, Decrypt(Encrypt(1L, ops, salt), ops, salt));
|
||
Assert.AreEqual(1.0f, Decrypt(Encrypt(1.0f, ops, salt), ops, salt));
|
||
Assert.AreEqual(1.0, Decrypt(Encrypt(1.0, ops, salt), ops, salt));
|
||
|
||
byte[] strBytes = Encrypt("abcdef", ops, salt);
|
||
Assert.AreEqual("abcdef", DecryptString(strBytes, 0, strBytes.Length, ops, salt));
|
||
var arr = new byte[100];
|
||
for (int i = 0; i < arr.Length; i++)
|
||
{
|
||
arr[i] = (byte)i;
|
||
}
|
||
EncryptBlock(arr, ops, salt);
|
||
DecryptBlock(arr, ops, salt);
|
||
for (int i = 0; i < arr.Length; i++)
|
||
{
|
||
Assert.AreEqual(i, arr[i]);
|
||
}
|
||
}
|
||
|
||
private List<uint> DecodeOps(uint ops)
|
||
{
|
||
var codes = new List<uint>();
|
||
while (ops != 0)
|
||
{
|
||
uint code = ops % (uint)_opCodes.Length;
|
||
codes.Add(code);
|
||
ops /= (uint)_opCodes.Length;
|
||
}
|
||
return codes;
|
||
}
|
||
|
||
public override int Encrypt(int value, int ops, int salt)
|
||
{
|
||
var codes = DecodeOps((uint)ops);
|
||
for (int i = codes.Count - 1; i >= 0; i--)
|
||
{
|
||
var opCode = _opCodes[codes[i]];
|
||
value = opCode.Encrypt(value, _secretKey, salt);
|
||
}
|
||
return value;
|
||
}
|
||
|
||
public override int Decrypt(int value, int ops, int salt)
|
||
{
|
||
var codes = DecodeOps((uint)ops);
|
||
foreach (var code in codes)
|
||
{
|
||
var opCode = _opCodes[code];
|
||
value = opCode.Decrypt(value, _secretKey, salt);
|
||
}
|
||
return value;
|
||
}
|
||
}
|
||
}
|