Compare commits

..

8 Commits

Author SHA1 Message Date
z__
0a9f299527 we dont need it anymore 2022-02-09 15:34:13 +07:00
z__
67f25ab20b transform ctor to extension method 2022-02-08 23:36:39 +07:00
z__
ad19a7dc9e better exception for invalid operand type 2022-02-08 23:26:25 +07:00
z__
dd854a159a ok high iq solutions is bad for compatibility 2022-02-04 15:11:49 +07:00
z__
879a373e6a added possibility to call SetConfiguration again if needed 2022-02-04 14:42:37 +07:00
z__
ec1b017946 added nlog custom targets assemblies loading 2022-02-04 14:33:32 +07:00
z__
cf75210304 fixed game analytics logger crash 2022-02-04 13:44:04 +07:00
z__
3696f18714 move nlog config to instance and move default to resources 2022-02-04 13:25:56 +07:00
7 changed files with 205 additions and 165 deletions

View File

@@ -1,5 +1,7 @@
using System;
using System.IO;
using NLog;
using NLog.Config;
using NLog.Targets;
using Torch.Utils;
@@ -12,7 +14,6 @@ namespace Torch.Server
{
var isService = Environment.GetEnvironmentVariable("TORCH_SERVICE")
?.Equals(bool.TrueString, StringComparison.OrdinalIgnoreCase) ?? false;
Target.Register<LogViewerTarget>(nameof(LogViewerTarget));
//Ensures that all the files are downloaded in the Torch directory.
var workingDir = AppContext.BaseDirectory;
var binDir = Path.Combine(Environment.GetEnvironmentVariable("TORCH_GAME_PATH") ?? workingDir, "DedicatedServer64");
@@ -47,6 +48,14 @@ namespace Torch.Server
instancePath = Directory.CreateDirectory(instanceName).FullName;
}
var oldNlog = Path.Combine(workingDir, "NLog.config");
var newNlog = Path.Combine(instancePath, "NLog.config");
if (File.Exists(oldNlog))
File.Move(oldNlog, newNlog, true);
else if (!File.Exists(newNlog))
using (var f = File.Create(newNlog))
typeof(Program).Assembly.GetManifestResourceStream("Torch.Server.NLog.config")!.CopyTo(f);
var oldTorchCfg = Path.Combine(workingDir, "Torch.cfg");
var torchCfg = Path.Combine(instancePath, "Torch.cfg");
@@ -65,42 +74,19 @@ namespace Torch.Server
var handler = new UnhandledExceptionHandler(config.Data, isService);
AppDomain.CurrentDomain.UnhandledException += handler.OnUnhandledException;
Target.Register<LogViewerTarget>(nameof(LogViewerTarget));
TorchLogManager.RegisterTargets(Environment.GetEnvironmentVariable("TORCH_LOG_EXTENSIONS_PATH") ??
Path.Combine(instancePath, "LoggingExtensions"));
TorchLogManager.SetConfiguration(new XmlLoggingConfiguration(newNlog));
var initializer = new Initializer(workingDir, config);
if (!initializer.Initialize(args))
Environment.Exit(1);
TorchLauncher.Launch(workingDir, binDir);
CopyNative(binDir);
initializer.Run(isService, instanceName, instancePath);
}
private static void CopyNative(string binPath)
{
var apiSource = Path.Combine(binPath, "steam_api64.dll");
var apiTarget = Path.Combine(AppContext.BaseDirectory, "steam_api64.dll");
if (!File.Exists(apiTarget))
{
File.Copy(apiSource, apiTarget);
}
else if (File.GetLastWriteTime(apiTarget) < File.GetLastWriteTime(binPath))
{
File.Delete(apiTarget);
File.Copy(apiSource, apiTarget);
}
var havokSource = Path.Combine(binPath, "Havok.dll");
var havokTarget = Path.Combine(AppContext.BaseDirectory, "Havok.dll");
if (!File.Exists(havokTarget))
{
File.Copy(havokSource, havokTarget);
}
else if (File.GetLastWriteTime(havokTarget) < File.GetLastWriteTime(havokSource))
{
File.Delete(havokTarget);
File.Copy(havokSource, havokTarget);
}
}
}
}

View File

@@ -160,6 +160,6 @@
</ItemGroup>
<ItemGroup>
<Page Remove="Views\WorldSelectControl.xaml" />
<None Include="..\NLog.config" Visible="false" CopyToOutputDirectory="PreserveNewest" CopyToPublishDirectory="Always" />
<EmbeddedResource Include="..\NLog.config" Visible="false" />
</ItemGroup>
</Project>

View File

@@ -197,7 +197,7 @@ namespace Torch.Managers.PatchManager
lock (_log)
{
var instructions = context.Body.Instructions
.Select(b => new MsilInstruction(b)).ToList();
.Select(b => b.ToMsilInstruction()).ToList();
LogTarget(PrintModeEnum.Patched, false, "========== Patched method ==========");
MethodTranspiler.IntegrityAnalysis((a, b) => LogTarget(PrintModeEnum.Patched, a, b), instructions, true);
LogTarget(PrintModeEnum.Patched, false, gap);

View File

@@ -0,0 +1,136 @@
using System;
using System.Linq;
using System.Reflection;
using Mono.Cecil;
using Mono.Cecil.Cil;
using MonoMod.Utils;
using Torch.Managers.PatchManager.Transpile;
using OperandType = System.Reflection.Emit.OperandType;
namespace Torch.Managers.PatchManager.MSIL;
internal static class InstructionExtensions
{
public static MsilInstruction ToMsilInstruction(this Instruction instruction)
{
static System.Reflection.Emit.Label CreateLabel(int pos)
{
var instance = Activator.CreateInstance(typeof(System.Reflection.Emit.Label),
BindingFlags.CreateInstance | BindingFlags.Instance | BindingFlags.NonPublic, null,
new object[] {pos}, null);
if (instance == null)
return default;
return (System.Reflection.Emit.Label) instance;
}
var systemOpCode = MethodContext.OpCodeLookup[instruction.OpCode.Value];
var msil = new MsilInstruction(systemOpCode);
if (instruction.Operand is null || instruction.OpCode.OperandType == Mono.Cecil.Cil.OperandType.InlineNone)
return msil;
var opType = systemOpCode.OperandType;
switch (instruction.Operand)
{
case Instruction targetInstruction when opType is OperandType.InlineBrTarget or OperandType.ShortInlineBrTarget:
msil.Operand = new MsilOperandBrTarget(msil)
{
Target = new MsilLabel(CreateLabel(targetInstruction.Offset))
};
break;
case FieldReference reference when opType == OperandType.InlineField:
msil.Operand = new MsilOperandInline.MsilOperandReflected<FieldInfo>(msil)
{
Value = reference.ResolveReflection()
};
break;
case int int32 when opType is OperandType.InlineI or OperandType.ShortInlineI:
msil.Operand = new MsilOperandInline.MsilOperandInt32(msil)
{
Value = int32
};
break;
case long int64 when opType is OperandType.InlineI8:
msil.Operand = new MsilOperandInline.MsilOperandInt64(msil)
{
Value = int64
};
break;
case MethodReference methodReference when opType is OperandType.InlineMethod:
msil.Operand = new MsilOperandInline.MsilOperandReflected<MethodBase>(msil)
{
Value = methodReference.ResolveReflection()
};
break;
case double @double when opType is OperandType.InlineR:
msil.Operand = new MsilOperandInline.MsilOperandDouble(msil)
{
Value = @double
};
break;
case null when opType is OperandType.InlineSig:
throw new NotSupportedException("InlineSignature is not supported by instruction converter");
case string @string when opType == OperandType.InlineString:
msil.Operand = new MsilOperandInline.MsilOperandString(msil)
{
Value = @string
};
break;
case Instruction[] targetInstructions when opType is OperandType.InlineSwitch:
msil.Operand = new MsilOperandSwitch(msil)
{
Labels = targetInstructions.Select(b => new MsilLabel(CreateLabel(b.Offset))).ToArray()
};
break;
case MemberReference memberReference when opType is OperandType.InlineTok:
msil.Operand = new MsilOperandInline.MsilOperandReflected<MemberInfo>(msil)
{
Value = memberReference.ResolveReflection()
};
break;
case TypeReference typeReference when opType is OperandType.InlineType:
msil.Operand = new MsilOperandInline.MsilOperandReflected<Type>(msil)
{
Value = typeReference.ResolveReflection()
};
break;
case VariableDefinition variableDefinition when opType is OperandType.InlineVar or OperandType.ShortInlineVar:
if (systemOpCode.IsLocalStore() || systemOpCode.IsLocalLoad() || systemOpCode.IsLocalLoadByRef())
msil.Operand = new MsilOperandInline.MsilOperandLocal(msil)
{
Value = new MsilLocal(variableDefinition.Index)
};
else
msil.Operand = new MsilOperandInline.MsilOperandArgument(msil)
{
Value = new MsilArgument(variableDefinition.Index)
};
break;
case ParameterDefinition parameterDefinition when opType is OperandType.InlineVar or OperandType.ShortInlineVar:
if (systemOpCode.IsLocalStore() || systemOpCode.IsLocalLoad() || systemOpCode.IsLocalLoadByRef())
msil.Operand = new MsilOperandInline.MsilOperandLocal(msil)
{
Value = new MsilLocal(parameterDefinition.Index)
};
else
msil.Operand = new MsilOperandInline.MsilOperandArgument(msil)
{
Value = new MsilArgument(parameterDefinition.Index)
};
break;
case float @float when opType == OperandType.ShortInlineR:
msil.Operand = new MsilOperandInline.MsilOperandSingle(msil)
{
Value = @float
};
break;
#pragma warning disable 618
case null when opType == OperandType.InlinePhi:
#pragma warning restore 618
default:
throw new ArgumentOutOfRangeException(nameof(instruction.Operand), instruction.Operand, "Invalid operand type");
}
return msil;
}
}

View File

@@ -12,6 +12,7 @@ using Mono.Cecil.Cil;
using MonoMod.Utils;
using Torch.Managers.PatchManager.Transpile;
using Torch.Utils;
using VRage.Game.VisualScripting;
using OpCode = System.Reflection.Emit.OpCode;
using OpCodes = System.Reflection.Emit.OpCodes;
using OperandType = System.Reflection.Emit.OperandType;
@@ -88,132 +89,6 @@ namespace Torch.Managers.PatchManager.MSIL
}
}
public MsilInstruction(Instruction instruction)
{
Label CreateLabel(int pos)
{
var instance = Activator.CreateInstance(typeof(Label),
BindingFlags.CreateInstance | BindingFlags.Instance | BindingFlags.NonPublic, null,
new object[] {pos}, null);
if (instance == null)
return default;
return (Label) instance;
}
if (!MethodContext.OpCodeLookup.TryGetValue(instruction.OpCode.Value, out var opCode))
return;
OpCode = opCode;
var opType = opCode.OperandType;
if (opType == OperandType.InlineNone)
{
Operand = null;
return;
}
switch (instruction.Operand)
{
case OperandType.InlineNone:
break;
case Instruction targetInstruction when opType == OperandType.InlineBrTarget || opType == OperandType.ShortInlineBrTarget:
Operand = new MsilOperandBrTarget(this)
{
Target = new MsilLabel(CreateLabel(targetInstruction.Offset))
};
break;
case FieldReference reference when opType == OperandType.InlineField:
Operand = new MsilOperandInline.MsilOperandReflected<FieldInfo>(this)
{
Value = reference.ResolveReflection()
};
break;
case int int32 when opType == OperandType.InlineI || opType == OperandType.ShortInlineI:
Operand = new MsilOperandInline.MsilOperandInt32(this)
{
Value = int32
};
break;
case long int64 when opType == OperandType.InlineI8:
Operand = new MsilOperandInline.MsilOperandInt64(this)
{
Value = int64
};
break;
case MethodReference methodReference when opType == OperandType.InlineMethod:
Operand = new MsilOperandInline.MsilOperandReflected<MethodBase>(this)
{
Value = methodReference.ResolveReflection()
};
break;
case double @double when opType == OperandType.InlineR:
Operand = new MsilOperandInline.MsilOperandDouble(this)
{
Value = @double
};
break;
case null when opType == OperandType.InlineSig:
throw new NotSupportedException("InlineSignature is not supported by instruction converter");
case string @string when opType == OperandType.InlineString:
Operand = new MsilOperandInline.MsilOperandString(this)
{
Value = @string
};
break;
case Instruction[] targetInstructions when opType == OperandType.InlineSwitch:
Operand = new MsilOperandSwitch(this)
{
Labels = targetInstructions.Select(b => new MsilLabel(CreateLabel(b.Offset))).ToArray()
};
break;
case MemberReference memberReference when opType == OperandType.InlineTok:
Operand = new MsilOperandInline.MsilOperandReflected<MemberInfo>(this)
{
Value = memberReference.ResolveReflection()
};
break;
case TypeReference typeReference when opType == OperandType.InlineType:
Operand = new MsilOperandInline.MsilOperandReflected<Type>(this)
{
Value = typeReference.ResolveReflection()
};
break;
case VariableDefinition variableDefinition when opType == OperandType.InlineVar || opType == OperandType.ShortInlineVar:
if (OpCode.IsLocalStore() || OpCode.IsLocalLoad() || OpCode.IsLocalLoadByRef())
Operand = new MsilOperandInline.MsilOperandLocal(this)
{
Value = new MsilLocal(variableDefinition.Index)
};
else
Operand = new MsilOperandInline.MsilOperandArgument(this)
{
Value = new MsilArgument(variableDefinition.Index)
};
break;
case ParameterDefinition parameterDefinition when opType == OperandType.InlineVar || opType == OperandType.ShortInlineVar:
if (OpCode.IsLocalStore() || OpCode.IsLocalLoad() || OpCode.IsLocalLoadByRef())
Operand = new MsilOperandInline.MsilOperandLocal(this)
{
Value = new MsilLocal(parameterDefinition.Index)
};
else
Operand = new MsilOperandInline.MsilOperandArgument(this)
{
Value = new MsilArgument(parameterDefinition.Index)
};
break;
case float @float when opType == OperandType.ShortInlineR:
Operand = new MsilOperandInline.MsilOperandSingle(this)
{
Value = @float
};
break;
#pragma warning disable 618
case null when opType == OperandType.InlinePhi:
#pragma warning restore 618
default:
throw new ArgumentOutOfRangeException(nameof(instruction.Operand), instruction.Operand, "Invalid operand type");
}
}
/// <summary>
/// Opcode of this instruction
/// </summary>
@@ -227,7 +102,7 @@ namespace Torch.Managers.PatchManager.MSIL
/// <summary>
/// The operand for this instruction, or null.
/// </summary>
public MsilOperand Operand { get; }
public MsilOperand Operand { get; internal set; }
/// <summary>
/// Labels pointing to this instruction.
@@ -284,7 +159,11 @@ namespace Torch.Managers.PatchManager.MSIL
type = type.BaseType;
}
((MsilOperandInline<T>) Operand).Value = o;
if (Operand is not MsilOperandInline<T> operandInline)
throw new InvalidOperationException(
$"Type {typeof(T).FullName} is not valid operand for {Operand?.GetType().Signature()}");
operandInline.Value = o;
return this;
}

View File

@@ -5,6 +5,7 @@ using System.Runtime.CompilerServices;
using NLog;
using NLog.Config;
using Torch.Managers.PatchManager;
using Torch.Utils;
namespace Torch.Patches
{
@@ -36,16 +37,13 @@ namespace Torch.Patches
_log.Warn("GALogger constructor is unknown. Logging may not function.");
return;
}
ctx.GetPattern(ctor).Prefixes.Add(typeof(GameAnalyticsPatch).GetMethod(nameof(PatchLogger),
BindingFlags.Static | BindingFlags.NonPublic | BindingFlags.Public));
ctx.GetPattern(ctor).AddPrefix(nameof(PatchLogger));
}
private static void FixLogging()
{
TorchLogManager.RestoreGlobalConfiguration();
_setLogger(null, LogManager.GetLogger("GameAnalytics"));
if (!(LogManager.Configuration is XmlLoggingConfiguration))
LogManager.Configuration = new XmlLoggingConfiguration(Path.Combine(
Path.GetDirectoryName(Assembly.GetEntryAssembly().Location) ?? Environment.CurrentDirectory, "NLog.config"));
}
private static bool PatchLogger()

View File

@@ -0,0 +1,41 @@
using System;
using System.IO;
using System.Linq;
using System.Reflection;
using System.Runtime.Loader;
using NLog;
using NLog.Config;
using NLog.Targets;
namespace Torch.Utils;
public static class TorchLogManager
{
private static AssemblyLoadContext LoadContext = new("TorchLog");
public static LoggingConfiguration Configuration { get; private set; }
public static void SetConfiguration(LoggingConfiguration configuration)
{
Configuration = configuration;
LogManager.Configuration = configuration;
LogManager.ReconfigExistingLoggers();
}
public static void RegisterTargets(string dir)
{
if (!Directory.Exists(dir)) return;
foreach (var type in Directory.EnumerateFiles(dir, "*.dll").Select(LoadContext.LoadFromAssemblyPath)
.SelectMany(b => b.ExportedTypes)
.Where(b => b.GetCustomAttribute<TargetAttribute>() is { }))
{
Target.Register(type.GetCustomAttribute<TargetAttribute>()!.Name, type);
}
}
public static void RestoreGlobalConfiguration()
{
SetConfiguration(Configuration);
}
}