Please disclose if any significant portion of your mod was created using AI tools by adding the 'AI Generated' category. Failing to do so may result in the mod being removed from Thunderstore.
Decompiled source of ForgingPotential v1.0.0
ForgingPotential.dll
Decompiled 4 days agousing System; using System.Collections; using System.Collections.Generic; using System.Diagnostics; using System.Globalization; using System.IO; using System.IO.Compression; using System.Linq; using System.Reflection; using System.Runtime.CompilerServices; using System.Runtime.InteropServices; using System.Runtime.Serialization; using System.Runtime.Versioning; using BepInEx; using BepInEx.Configuration; using HarmonyLib; using JetBrains.Annotations; using Microsoft.CodeAnalysis; using ServerSync; using TMPro; using UnityEngine; using UnityEngine.UI; [assembly: CompilationRelaxations(8)] [assembly: RuntimeCompatibility(WrapNonExceptionThrows = true)] [assembly: Debuggable(DebuggableAttribute.DebuggingModes.Default | DebuggableAttribute.DebuggingModes.DisableOptimizations | DebuggableAttribute.DebuggingModes.IgnoreSymbolStoreSequencePoints | DebuggableAttribute.DebuggingModes.EnableEditAndContinue)] [assembly: AssemblyTitle("ForgingPotential")] [assembly: AssemblyDescription("")] [assembly: AssemblyConfiguration("")] [assembly: AssemblyCompany("")] [assembly: AssemblyProduct("ForgingPotential")] [assembly: AssemblyCopyright("Copyright © 2026")] [assembly: AssemblyTrademark("")] [assembly: ComVisible(false)] [assembly: Guid("21413daf-9aad-4755-9a5d-5369e071cadc")] [assembly: AssemblyFileVersion("1.0.0.0")] [assembly: TargetFramework(".NETFramework,Version=v4.8", FrameworkDisplayName = ".NET Framework 4.8")] [assembly: AssemblyVersion("1.0.0.0")] [module: RefSafetyRules(11)] namespace Microsoft.CodeAnalysis { [CompilerGenerated] [Embedded] internal sealed class EmbeddedAttribute : Attribute { } } namespace System.Runtime.CompilerServices { [CompilerGenerated] [Embedded] [AttributeUsage(AttributeTargets.Class | AttributeTargets.Property | AttributeTargets.Field | AttributeTargets.Event | AttributeTargets.Parameter | AttributeTargets.ReturnValue | AttributeTargets.GenericParameter, AllowMultiple = false, Inherited = false)] internal sealed class NullableAttribute : Attribute { public readonly byte[] NullableFlags; public NullableAttribute(byte P_0) { NullableFlags = new byte[1] { P_0 }; } public NullableAttribute(byte[] P_0) { NullableFlags = P_0; } } [CompilerGenerated] [Embedded] [AttributeUsage(AttributeTargets.Class | AttributeTargets.Struct | AttributeTargets.Method | AttributeTargets.Interface | AttributeTargets.Delegate, AllowMultiple = false, Inherited = false)] internal sealed class NullableContextAttribute : Attribute { public readonly byte Flag; public NullableContextAttribute(byte P_0) { Flag = P_0; } } [CompilerGenerated] [Embedded] [AttributeUsage(AttributeTargets.Module, AllowMultiple = false, Inherited = false)] internal sealed class RefSafetyRulesAttribute : Attribute { public readonly int Version; public RefSafetyRulesAttribute(int P_0) { Version = P_0; } } } namespace ForgingPotential { internal static class ForgePatches { private struct MobIdol { public readonly int Tier; public readonly bool IsBattle; public MobIdol(int tier, bool isBattle) { Tier = tier; IsBattle = isBattle; } } private const string ForgeCountKey = "fp_forge_n"; private const string ForgeTotalKey = "fp_forge_total"; private static readonly string[] BattlePrefabs = new string[8] { "Upgrader0Weapon", "Upgrader1Weapon", "Upgrader2Weapon", "Upgrader3Weapon", "Upgrader4Weapon", "Upgrader5Weapon", "Upgrader6Weapon", "Upgrader7Weapon" }; private static readonly string[] ArmorPrefabs = new string[8] { "Upgrader0Armor", "Upgrader1Armor", "Upgrader2Armor", "Upgrader3Armor", "Upgrader4Armor", "Upgrader5Armor", "Upgrader6Armor", "Upgrader7Armor" }; private static readonly Dictionary<string, int> BossTier = new Dictionary<string, int>(StringComparer.OrdinalIgnoreCase) { { "Eikthyr", 0 }, { "gd_king", 1 }, { "Bonemass", 2 }, { "Dragon", 3 }, { "GoblinKing", 4 }, { "SeekerQueen", 5 }, { "Fader", 6 }, { "FrozenKing_p3", 7 } }; private static readonly HashSet<string> InstantDespawnPrefabs = new HashSet<string>(StringComparer.OrdinalIgnoreCase) { "Bonemass", "Dragon", "SeekerQueen", "FrozenKing_p3", "Gjall", "Writhan", "StoneGolem", "SeekerBrute", "FallenValkyrie", "Morgen", "Barka", "JotunWitch" }; private static readonly Dictionary<string, float> WoodenIdolChances = new Dictionary<string, float>(StringComparer.OrdinalIgnoreCase) { { "Greyling", 1f }, { "Greydwarf", 2f } }; private static readonly Dictionary<string, MobIdol> MobIdolTable = new Dictionary<string, MobIdol>(StringComparer.OrdinalIgnoreCase) { { "Troll", new MobIdol(1, isBattle: true) }, { "Bjorn", new MobIdol(1, isBattle: false) }, { "Abomination", new MobIdol(2, isBattle: true) }, { "Writhan", new MobIdol(2, isBattle: false) }, { "Fenring", new MobIdol(3, isBattle: true) }, { "StoneGolem", new MobIdol(3, isBattle: false) }, { "GoblinBrute", new MobIdol(4, isBattle: true) }, { "GoblinShaman", new MobIdol(4, isBattle: false) }, { "Gjall", new MobIdol(5, isBattle: true) }, { "SeekerBrute", new MobIdol(5, isBattle: false) }, { "FallenValkyrie", new MobIdol(6, isBattle: true) }, { "Morgen", new MobIdol(6, isBattle: false) }, { "Barka", new MobIdol(7, isBattle: true) }, { "JotunWitch", new MobIdol(7, isBattle: false) } }; private const float MobDropChancePercent = 10f; private static ItemData _pendingForgeItem; private static string _pendingItemName; private static int _pendingAttempt; private static int _pendingQualityBefore; private static int _pendingTotal; private static bool _pendingIsForge; private static bool _repairMode; private static object _craftBtnTextObj; private static int _lastAltToggleFrame = -1; private const string BtnRefine = "Attempt Refinement\n<color=white>(Press [ALT] to toggle)</color>"; private const string BtnRepair = "Repair\n<color=white>(Press [ALT] to toggle)</color>"; private static bool _idolLogOnce; private static FieldInfo _craftTimerField; private static bool _craftTimerFieldSearched; private static bool IsAltKeyDown() { try { return Input.GetKeyDown((KeyCode)308) || Input.GetKeyDown((KeyCode)307); } catch { return false; } } [HarmonyPrefix] [HarmonyPatch(typeof(InventoryGui), "DoCrafting")] private static bool DoCrafting_Prefix(InventoryGui __instance) { _pendingIsForge = false; _pendingForgeItem = null; _pendingItemName = null; _pendingAttempt = 0; _pendingQualityBefore = 0; _pendingTotal = 0; try { if (!ForgingPotentialPlugin.IsProgressiveForgeEnabled()) { return true; } if (!IsForgeOfPotentialStation()) { return true; } ItemData val = ResolveCraftItem(__instance); if (val == null) { return true; } if (_repairMode) { TryRepairOdds(__instance, val); return false; } _pendingIsForge = true; _pendingForgeItem = val; _pendingItemName = ((val.m_shared != null) ? val.m_shared.m_name : null); _pendingQualityBefore = val.m_quality; _pendingAttempt = GetForgeCount(val); _pendingTotal = GetForgeTotalStored(val); float successPercent = ForgingPotentialPlugin.GetSuccessPercent(_pendingAttempt); SetIdolChancesFromPercent(successPercent); ForgingPotentialPlugin.LogAlways($"FoP attempt #{_pendingAttempt + 1} '{_pendingItemName}' q={_pendingQualityBefore} idol success={successPercent:0}%"); } catch (Exception ex) { ForgingPotentialPlugin.LogAlways("DoCrafting prefix: " + ex.Message); } return true; } [HarmonyPostfix] [HarmonyPatch(typeof(InventoryGui), "DoCrafting")] private static void DoCrafting_Postfix(InventoryGui __instance) { try { if (!_pendingIsForge || !ForgingPotentialPlugin.IsProgressiveForgeEnabled()) { return; } int pendingAttempt = _pendingAttempt; int pendingQualityBefore = _pendingQualityBefore; Player localPlayer = Player.m_localPlayer; ItemData val = FindLiveItem(_pendingItemName, _pendingForgeItem, pendingQualityBefore + 1); if (val == null) { val = FindLiveItem(_pendingItemName, _pendingForgeItem); } if (val != null && val.m_quality > pendingQualityBefore) { int num = pendingAttempt + 1; int num2 = _pendingTotal + 1; SetForgeCount(val, num); SetForgeTotal(val, num2); if (_pendingForgeItem != null && _pendingForgeItem != val) { SetForgeCount(_pendingForgeItem, num); SetForgeTotal(_pendingForgeItem, num2); } ForgingPotentialPlugin.LogAlways($"FoP total {_pendingTotal}→{num2}, risk→{num} on '{_pendingItemName}'"); if ((Object)(object)localPlayer != (Object)null) { ((Character)localPlayer).Message((MessageType)2, $"Upgraded to Tier {val.m_quality}!", 0, (Sprite)null, false); } } } catch (Exception ex) { ForgingPotentialPlugin.LogAlways("DoCrafting postfix: " + ex.Message); } finally { _pendingIsForge = false; _pendingForgeItem = null; _pendingItemName = null; } } private static void TryRepairOdds(InventoryGui gui, ItemData item) { Player localPlayer = Player.m_localPlayer; if (!((Object)(object)localPlayer == (Object)null) && item != null) { int forgeCount = GetForgeCount(item); if (forgeCount <= 0) { ((Character)localPlayer).Message((MessageType)2, "Odds already at maximum (100%)", 0, (Sprite)null, false); return; } if (!TryConsumeOneIdol(gui, localPlayer)) { ((Character)localPlayer).Message((MessageType)2, "Requires a matching idol to repair", 0, (Sprite)null, false); return; } int num = forgeCount - 1; SetForgeCount(item, num); float successPercent = ForgingPotentialPlugin.GetSuccessPercent(num); ((Character)localPlayer).Message((MessageType)2, $"Repaired to {successPercent:0}%", 0, (Sprite)null, false); ForgingPotentialPlugin.LogAlways($"FoP REPAIR '{item.m_shared?.m_name}' attempts {forgeCount}→{num} ({successPercent:0}%)"); } } private static bool TryConsumeOneIdol(InventoryGui gui, Player pl) { try { Inventory inventory = ((Humanoid)pl).GetInventory(); if (inventory == null) { return false; } string text = null; try { object value = Traverse.Create((object)gui).Field("m_craftRecipe").GetValue(); if (value != null) { object obj = (value.GetType().GetField("m_resources", BindingFlags.Instance | BindingFlags.Public | BindingFlags.NonPublic) ?? value.GetType().GetField("m_requirements", BindingFlags.Instance | BindingFlags.Public | BindingFlags.NonPublic))?.GetValue(value); if (obj is Array array) { foreach (object item in array) { if (item == null) { continue; } object? obj2 = (item.GetType().GetField("m_resItem", BindingFlags.Instance | BindingFlags.Public | BindingFlags.NonPublic) ?? item.GetType().GetField("m_item", BindingFlags.Instance | BindingFlags.Public | BindingFlags.NonPublic))?.GetValue(item); ItemDrop val = (ItemDrop)((obj2 is ItemDrop) ? obj2 : null); if ((Object)(object)val == (Object)null) { continue; } string text2 = ((Object)val).name ?? ""; if (text2.IndexOf("Upgrader", StringComparison.OrdinalIgnoreCase) < 0) { continue; } if (val.m_itemData?.m_shared != null) { text = val.m_itemData.m_shared.m_name; } break; } } } } catch { } ItemData val2 = null; foreach (ItemData allItem in inventory.GetAllItems()) { if (allItem?.m_shared == null) { continue; } string text3 = allItem.m_shared.m_name ?? ""; if (text3.IndexOf("upgrader", StringComparison.OrdinalIgnoreCase) >= 0 || text3.IndexOf("idol", StringComparison.OrdinalIgnoreCase) >= 0) { if (string.IsNullOrEmpty(text)) { val2 = allItem; break; } if (text3 == text) { val2 = allItem; break; } } } if (val2 == null) { return false; } try { inventory.RemoveItem(val2, 1); } catch { try { inventory.RemoveItem(val2); } catch { return false; } } return true; } catch (Exception ex) { ForgingPotentialPlugin.LogAlways("TryConsumeOneIdol: " + ex.Message); return false; } } [HarmonyPostfix] [HarmonyPatch(typeof(InventoryGui), "Update")] private static void InventoryGui_Update_FoPRepair_Postfix(InventoryGui __instance) { FoPUiTick(__instance, handleAlt: true); } [HarmonyPostfix] [HarmonyPatch(typeof(InventoryGui), "UpdateRecipe")] private static void InventoryGui_UpdateRecipe_FoPUi_Postfix(InventoryGui __instance, Player player, float dt) { FoPUiTick(__instance, handleAlt: false); } private static void FoPUiTick(InventoryGui gui, bool handleAlt) { if ((Object)(object)gui == (Object)null || !ForgingPotentialPlugin.IsProgressiveForgeEnabled()) { return; } try { if (!((Behaviour)gui).isActiveAndEnabled || !IsForgeOfPotentialStation()) { if (_repairMode) { _repairMode = false; } _craftBtnTextObj = null; return; } if (handleAlt) { bool flag = IsCraftingInProgress(gui); int frameCount = Time.frameCount; if (!flag && IsAltKeyDown() && frameCount != _lastAltToggleFrame) { _lastAltToggleFrame = frameCount; _repairMode = !_repairMode; Player localPlayer = Player.m_localPlayer; if ((Object)(object)localPlayer != (Object)null) { ((Character)localPlayer).Message((MessageType)2, _repairMode ? "Repair active" : "Refinement active", 0, (Sprite)null, false); } } } ApplyCraftButtonLabel(gui, _repairMode ? "Repair\n<color=white>(Press [ALT] to toggle)</color>" : "Attempt Refinement\n<color=white>(Press [ALT] to toggle)</color>"); } catch { } } private static bool IsOurCraftButtonText(string cur) { if (string.IsNullOrEmpty(cur)) { return false; } if (cur.IndexOf("Attempt Refinement", StringComparison.OrdinalIgnoreCase) >= 0) { return true; } if (cur.IndexOf("Repair", StringComparison.OrdinalIgnoreCase) >= 0 && cur.IndexOf("Press", StringComparison.OrdinalIgnoreCase) >= 0) { return true; } if (cur.IndexOf("Press [ALT]", StringComparison.OrdinalIgnoreCase) >= 0) { return true; } return false; } private static void ApplyCraftButtonLabel(InventoryGui gui, string label) { try { object craftBtnTextObj = _craftBtnTextObj; Text val = (Text)((craftBtnTextObj is Text) ? craftBtnTextObj : null); if (val != null && (Object)(object)val != (Object)null) { if (IsOurCraftButtonText(val.text) || val.text == label) { if (val.text != label) { val.text = label; } return; } _craftBtnTextObj = null; } } catch { _craftBtnTextObj = null; } try { Text[] componentsInChildren = ((Component)gui).GetComponentsInChildren<Text>(true); if (componentsInChildren == null) { return; } Text[] array = componentsInChildren; foreach (Text val2 in array) { if (!((Object)(object)val2 == (Object)null) && !string.IsNullOrEmpty(val2.text) && IsOurCraftButtonText(val2.text)) { _craftBtnTextObj = val2; val2.text = label; return; } } } catch { } try { Component[] componentsInChildren2 = ((Component)gui).GetComponentsInChildren<Component>(true); foreach (Component val3 in componentsInChildren2) { if ((Object)(object)val3 == (Object)null || val3 is Text || ((object)val3).GetType().Name.IndexOf("Text", StringComparison.OrdinalIgnoreCase) < 0) { continue; } PropertyInfo property = ((object)val3).GetType().GetProperty("text"); if (!(property == null) && !(property.PropertyType != typeof(string)) && property.CanWrite) { string cur = property.GetValue(val3, null) as string; if (IsOurCraftButtonText(cur)) { _craftBtnTextObj = val3; property.SetValue(val3, label, null); break; } } } } catch { } } private static ItemData FindLiveItem(string sharedName, ItemData fallback, int expectedQuality = -1) { try { Player localPlayer = Player.m_localPlayer; Inventory val = ((localPlayer != null) ? ((Humanoid)localPlayer).GetInventory() : null); if (val != null && !string.IsNullOrEmpty(sharedName)) { ItemData val2 = null; foreach (ItemData allItem in val.GetAllItems()) { if (allItem?.m_shared != null && !(allItem.m_shared.m_name != sharedName)) { if (fallback != null && allItem == fallback) { return allItem; } if (expectedQuality > 0 && allItem.m_quality == expectedQuality) { return allItem; } val2 = allItem; } } if (val2 != null) { return val2; } } } catch { } return fallback; } private static void SetIdolChancesFromPercent(float pct) { float upgrade = Mathf.Clamp01(pct / 100f); float breakChance = ((upgrade >= 1f) ? 0f : 1f); ObjectDB instance = ObjectDB.instance; if ((Object)(object)instance == (Object)null) { return; } if (instance.m_items != null) { foreach (GameObject item in instance.m_items) { PatchGo(item); } } for (int i = 0; i <= 7; i++) { PatchGo(instance.GetItemPrefab($"Upgrader{i}Weapon")); PatchGo(instance.GetItemPrefab($"Upgrader{i}Armor")); } void PatchGo(GameObject go) { if (!((Object)(object)go == (Object)null)) { string text = ((Object)go).name ?? ""; if (text.IndexOf("Upgrader", StringComparison.OrdinalIgnoreCase) >= 0) { ItemDrop component = go.GetComponent<ItemDrop>(); if (component?.m_itemData?.m_shared != null) { SharedData shared = component.m_itemData.m_shared; Type type = ((object)shared).GetType(); FieldInfo fieldInfo = AccessTools.Field(type, "m_upgradeChance"); if (fieldInfo != null && fieldInfo.FieldType == typeof(float)) { fieldInfo.SetValue(shared, upgrade); } FieldInfo fieldInfo2 = AccessTools.Field(type, "m_breakChance"); if (fieldInfo2 != null && fieldInfo2.FieldType == typeof(float)) { fieldInfo2.SetValue(shared, breakChance); } } } } } } public static void ItemData_GetTooltip_Postfix(ItemData __instance, ref string __result) { try { if (ForgingPotentialPlugin.IsProgressiveForgeEnabled() && ForgingPotentialPlugin.ShowForgeTooltip != null && ForgingPotentialPlugin.ShowForgeTooltip.Value && __instance != null && !string.IsNullOrEmpty(__result)) { SharedData shared = __instance.m_shared; if (shared != null && shared.m_maxQuality > 1 && __result.IndexOf("Forge refinement chance:", StringComparison.Ordinal) < 0) { int forgeCount = GetForgeCount(__instance); float successPercent = ForgingPotentialPlugin.GetSuccessPercent(forgeCount); __result = __result + "\nForge refinement chance: <color=orange>" + successPercent.ToString("0") + "%</color>"; } } } catch { } } [HarmonyPostfix] [HarmonyPatch(typeof(ObjectDB), "Awake")] private static void ObjectDB_Awake_Postfix(ObjectDB __instance) { if (!ForgingPotentialPlugin.IsProgressiveForgeEnabled()) { return; } try { EnsureIdolChances(__instance); } catch { } } [HarmonyPostfix] [HarmonyPatch(typeof(ObjectDB), "CopyOtherDB")] private static void ObjectDB_CopyOtherDB_Postfix(ObjectDB __instance) { if (!ForgingPotentialPlugin.IsProgressiveForgeEnabled()) { return; } try { EnsureIdolChances(__instance); } catch { } } private static void EnsureIdolChances(ObjectDB db) { if ((Object)(object)db == (Object)null) { return; } int patched = 0; if (db.m_items != null) { foreach (GameObject item in db.m_items) { PatchGo(item); } } for (int i = 0; i <= 7; i++) { PatchGo(db.GetItemPrefab($"Upgrader{i}Weapon")); PatchGo(db.GetItemPrefab($"Upgrader{i}Armor")); } if (!_idolLogOnce) { ForgingPotentialPlugin.LogAlways($"Idol chance fields initialized on {patched} items."); _idolLogOnce = true; } void PatchGo(GameObject go) { if (!((Object)(object)go == (Object)null)) { string text = ((Object)go).name ?? ""; if (text.IndexOf("Upgrader", StringComparison.OrdinalIgnoreCase) >= 0) { ItemDrop component = go.GetComponent<ItemDrop>(); if (component?.m_itemData?.m_shared != null) { SharedData shared = component.m_itemData.m_shared; Type type = ((object)shared).GetType(); bool flag = false; FieldInfo fieldInfo = AccessTools.Field(type, "m_upgradeChance"); if (fieldInfo != null && fieldInfo.FieldType == typeof(float)) { fieldInfo.SetValue(shared, 1f); flag = true; } FieldInfo fieldInfo2 = AccessTools.Field(type, "m_breakChance"); if (fieldInfo2 != null && fieldInfo2.FieldType == typeof(float)) { fieldInfo2.SetValue(shared, 0f); flag = true; } if (flag) { patched++; } } } } } } private static bool IsForgeOfPotentialStation() { Player localPlayer = Player.m_localPlayer; if ((Object)(object)localPlayer == (Object)null) { return false; } CraftingStation currentCraftingStation = localPlayer.GetCurrentCraftingStation(); if ((Object)(object)currentCraftingStation == (Object)null) { return false; } string text = (((Object)(object)((Component)currentCraftingStation).gameObject != (Object)null) ? (((Object)((Component)currentCraftingStation).gameObject).name ?? "") : ""); int num = text.IndexOf("(Clone)", StringComparison.OrdinalIgnoreCase); if (num > 0) { text = text.Substring(0, num).Trim(); } if (text.IndexOf("upgradestation", StringComparison.OrdinalIgnoreCase) >= 0) { return true; } if (text.IndexOf("AncientUpgradeStation", StringComparison.OrdinalIgnoreCase) >= 0) { return true; } string text2 = currentCraftingStation.m_name ?? ""; if (text2.IndexOf("$piece_upgradestation", StringComparison.OrdinalIgnoreCase) >= 0) { return true; } if (text2.IndexOf("potential", StringComparison.OrdinalIgnoreCase) >= 0) { return true; } return false; } private static bool IsCraftingInProgress(InventoryGui gui) { if ((Object)(object)gui == (Object)null) { return false; } try { if (!_craftTimerFieldSearched) { _craftTimerFieldSearched = true; _craftTimerField = typeof(InventoryGui).GetField("m_craftTimer", BindingFlags.Instance | BindingFlags.Public | BindingFlags.NonPublic); } if (_craftTimerField == null || _craftTimerField.FieldType != typeof(float)) { return false; } float num = (float)_craftTimerField.GetValue(gui); return num > 0f; } catch { return false; } } private static ItemData ResolveCraftItem(InventoryGui gui) { try { object value = Traverse.Create((object)gui).Field("m_craftUpgradeItem").GetValue(); ItemData val = (ItemData)((value is ItemData) ? value : null); if (val != null) { return val; } } catch { } try { object value2 = Traverse.Create((object)gui).Field("m_craftItem").GetValue(); ItemData val2 = (ItemData)((value2 is ItemData) ? value2 : null); if (val2 != null) { return val2; } } catch { } try { object value3 = Traverse.Create((object)gui).Field("m_dragItem").GetValue(); ItemData val3 = (ItemData)((value3 is ItemData) ? value3 : null); if (val3 != null) { return val3; } } catch { } try { object value4 = Traverse.Create((object)gui).Field("m_craftRecipe").GetValue(); if (value4 != null) { object? obj4 = (AccessTools.Field(value4.GetType(), "m_item") ?? AccessTools.Field(value4.GetType(), "m_itemData"))?.GetValue(value4); ItemData val4 = (ItemData)((obj4 is ItemData) ? obj4 : null); if (val4 != null) { return val4; } } } catch { } return null; } private static int GetForgeCount(ItemData item) { if (item == null) { return 0; } try { Dictionary<string, string> customData = GetCustomData(item); if (customData != null && customData.TryGetValue("fp_forge_n", out var value) && int.TryParse(value, out var result) && result >= 0) { return result; } } catch { } return 0; } private static void SetForgeCount(ItemData item, int count) { if (item == null || count < 0) { return; } try { Dictionary<string, string> customData = GetCustomData(item); if (customData != null) { customData["fp_forge_n"] = count.ToString(); } } catch { } } private static int GetForgeTotalStored(ItemData item) { if (item == null) { return 0; } try { Dictionary<string, string> customData = GetCustomData(item); if (customData != null && customData.TryGetValue("fp_forge_total", out var value) && int.TryParse(value, out var result) && result >= 0) { return result; } } catch { } return 0; } private static void SetForgeTotal(ItemData item, int count) { if (item == null || count < 0) { return; } try { Dictionary<string, string> customData = GetCustomData(item); if (customData != null) { customData["fp_forge_total"] = count.ToString(); } } catch { } } private static Dictionary<string, string> GetCustomData(ItemData item) { if (item == null) { return null; } try { FieldInfo fieldInfo = typeof(ItemData).GetField("m_customData", BindingFlags.Instance | BindingFlags.Public | BindingFlags.NonPublic); if (fieldInfo == null) { fieldInfo = AccessTools.Field(typeof(ItemData), "m_customData"); } if (fieldInfo == null) { return null; } object value = fieldInfo.GetValue(item); if (value is Dictionary<string, string> result) { return result; } Dictionary<string, string> dictionary = new Dictionary<string, string>(); fieldInfo.SetValue(item, dictionary); return dictionary; } catch { return null; } } [HarmonyPostfix] [HarmonyPatch(typeof(Character), "OnDeath")] private static void Character_OnDeath_IdolSchedule_Postfix(Character __instance) { //IL_004b: Unknown result type (might be due to invalid IL or missing references) //IL_0050: Unknown result type (might be due to invalid IL or missing references) //IL_00a8: Unknown result type (might be due to invalid IL or missing references) //IL_0140: Unknown result type (might be due to invalid IL or missing references) //IL_01b3: Unknown result type (might be due to invalid IL or missing references) //IL_01a0: Unknown result type (might be due to invalid IL or missing references) try { if ((Object)(object)__instance == (Object)null || !IsDeathDropAuthority(__instance)) { return; } string prefabName = Utils.GetPrefabName(((Component)__instance).gameObject); if (string.IsNullOrEmpty(prefabName)) { return; } Vector3 position = ((Component)__instance).transform.position; bool flag = false; try { flag = __instance.IsBoss(); } catch { } int num = ResolveBossTier(prefabName); if (ForgingPotentialPlugin.EnableBossIdolDrops.Value && num >= 0 && (flag || BossTier.ContainsKey(prefabName))) { int num2 = Random.Range(1, 3); int num3 = Random.Range(1, 3); ScheduleIdolDrop(prefabName, position, num, num2, num3); ForgingPotentialPlugin.LogAlways($"Scheduled boss idol drop '{prefabName}' tier={num} battle={num2} protection={num3}"); } else { if (!ForgingPotentialPlugin.EnableMobIdolDrops.Value) { return; } MobIdol info; if (TryResolveWoodenChance(prefabName, out var chance)) { if (Random.Range(0f, 100f) < chance) { bool flag2 = Random.value < 0.5f; ScheduleIdolDrop(prefabName, position, 0, flag2 ? 1 : 0, (!flag2) ? 1 : 0); } } else if (TryResolveMob(prefabName, out info) && !(Random.Range(0f, 100f) >= 10f)) { if (info.IsBattle) { ScheduleIdolDrop(prefabName, position, info.Tier, 1, 0); } else { ScheduleIdolDrop(prefabName, position, info.Tier, 0, 1); } } } } catch (Exception ex) { ForgingPotentialPlugin.LogAlways("OnDeath schedule: " + ex.Message); } } private static bool IsInstantDespawnPrefab(string prefab) { if (string.IsNullOrEmpty(prefab)) { return false; } return InstantDespawnPrefabs.Contains(prefab); } private static void ScheduleIdolDrop(string prefab, Vector3 deathPos, int tier, int nBattle, int nArmor) { //IL_0013: Unknown result type (might be due to invalid IL or missing references) //IL_0014: Unknown result type (might be due to invalid IL or missing references) //IL_0027: Unknown result type (might be due to invalid IL or missing references) //IL_0028: Unknown result type (might be due to invalid IL or missing references) //IL_0090: Unknown result type (might be due to invalid IL or missing references) //IL_007e: Unknown result type (might be due to invalid IL or missing references) if (IsInstantDespawnPrefab(prefab)) { SpawnIdols(BattlePrefabs, tier, nBattle, deathPos, Quaternion.identity); SpawnIdols(ArmorPrefabs, tier, nArmor, deathPos, Quaternion.identity); ForgingPotentialPlugin.LogAlways($"Instant idol drop '{prefab}' tier={tier} battle={nBattle} protection={nArmor}"); return; } Player localPlayer = Player.m_localPlayer; if ((Object)(object)localPlayer != (Object)null) { ((MonoBehaviour)localPlayer).StartCoroutine(AttachToRagdollNextFrame(deathPos, tier, nBattle, nArmor)); return; } Ragdoll val = FindNearestRagdoll(deathPos, 8f); if ((Object)(object)val != (Object)null) { AttachMarker(val, tier, nBattle, nArmor); } else { ForgingPotentialPlugin.LogAlways("Idol drop skipped '" + prefab + "' — no ragdoll"); } } private static IEnumerator AttachToRagdollNextFrame(Vector3 deathPos, int tier, int nBattle, int nArmor) { //IL_0007: Unknown result type (might be due to invalid IL or missing references) //IL_0008: Unknown result type (might be due to invalid IL or missing references) yield return null; Ragdoll ragdoll = FindNearestRagdoll(deathPos, 8f); if ((Object)(object)ragdoll != (Object)null) { AttachMarker(ragdoll, tier, nBattle, nArmor); ForgingPotentialPlugin.LogAlways("Idol drop attached to ragdoll (spawns on body despawn)"); } else { ForgingPotentialPlugin.LogAlways("Idol drop skipped — no ragdoll found after death"); } } private static void AttachMarker(Ragdoll ragdoll, int tier, int nBattle, int nArmor) { if (!((Object)(object)ragdoll == (Object)null)) { FoPRagdollIdolDrop foPRagdollIdolDrop = ((Component)ragdoll).gameObject.GetComponent<FoPRagdollIdolDrop>(); if ((Object)(object)foPRagdollIdolDrop == (Object)null) { foPRagdollIdolDrop = ((Component)ragdoll).gameObject.AddComponent<FoPRagdollIdolDrop>(); } foPRagdollIdolDrop.Tier = tier; foPRagdollIdolDrop.BattleCount = nBattle; foPRagdollIdolDrop.ArmorCount = nArmor; } } private static Ragdoll FindNearestRagdoll(Vector3 pos, float maxDist) { //IL_0060: Unknown result type (might be due to invalid IL or missing references) //IL_0065: Unknown result type (might be due to invalid IL or missing references) try { Ragdoll result = null; float num = maxDist; Ragdoll[] array = null; try { array = Object.FindObjectsByType<Ragdoll>((FindObjectsSortMode)0); } catch { try { array = Object.FindObjectsByType<Ragdoll>((FindObjectsInactive)0, (FindObjectsSortMode)0); } catch { return null; } } if (array == null) { return null; } Ragdoll[] array2 = array; foreach (Ragdoll val in array2) { if (!((Object)(object)val == (Object)null)) { float num2 = Vector3.Distance(((Component)val).transform.position, pos); if (num2 < num) { num = num2; result = val; } } } return result; } catch { return null; } } private static bool IsDeathDropAuthority(Character character) { try { ZNetView component = ((Component)character).GetComponent<ZNetView>(); if ((Object)(object)component != (Object)null && component.IsValid() && component.IsOwner()) { return true; } } catch { } try { if ((Object)(object)ZNet.instance != (Object)null && ZNet.instance.IsServer()) { return true; } } catch { } return false; } private static void SpawnIdols(string[] prefabs, int tier, int amount, Vector3 pos, Quaternion rot) { //IL_00ea: Unknown result type (might be due to invalid IL or missing references) //IL_00eb: Unknown result type (might be due to invalid IL or missing references) //IL_00ed: Unknown result type (might be due to invalid IL or missing references) //IL_00f2: Unknown result type (might be due to invalid IL or missing references) if (amount <= 0 || tier < 0 || prefabs == null || tier >= prefabs.Length) { return; } string text = prefabs[tier]; GameObject val = null; try { if ((Object)(object)ObjectDB.instance != (Object)null) { val = ObjectDB.instance.GetItemPrefab(text); } } catch { } try { if ((Object)(object)val == (Object)null && (Object)(object)ZNetScene.instance != (Object)null) { val = ZNetScene.instance.GetPrefab(text); } } catch { } if ((Object)(object)val == (Object)null) { ForgingPotentialPlugin.LogAlways("SpawnIdols: prefab not found '" + text + "'"); return; } Vector3 val2 = default(Vector3); for (int i = 0; i < amount; i++) { ((Vector3)(ref val2))..ctor(Random.Range(-0.5f, 0.5f), 0.25f, Random.Range(-0.5f, 0.5f)); try { GameObject val3 = Object.Instantiate<GameObject>(val, pos + val2, rot); if (!((Object)(object)val3 == (Object)null)) { ItemDrop component = val3.GetComponent<ItemDrop>(); if ((Object)(object)component != (Object)null && component.m_itemData != null) { component.m_itemData.m_stack = 1; } } } catch (Exception ex) { ForgingPotentialPlugin.LogAlways("SpawnIdols: " + text + " failed: " + ex.Message); } } } internal static void SpawnIdolsFromRagdoll(Vector3 pos, int tier, int nBattle, int nArmor) { //IL_0008: Unknown result type (might be due to invalid IL or missing references) //IL_0009: Unknown result type (might be due to invalid IL or missing references) //IL_001b: Unknown result type (might be due to invalid IL or missing references) //IL_001c: Unknown result type (might be due to invalid IL or missing references) SpawnIdols(BattlePrefabs, tier, nBattle, pos, Quaternion.identity); SpawnIdols(ArmorPrefabs, tier, nArmor, pos, Quaternion.identity); ForgingPotentialPlugin.LogAlways($"Ragdoll despawn idol drop tier={tier} battle={nBattle} protection={nArmor}"); } private static int ResolveBossTier(string prefab) { if (string.IsNullOrEmpty(prefab)) { return -1; } if (BossTier.TryGetValue(prefab, out var value)) { return value; } foreach (KeyValuePair<string, int> item in BossTier) { if (item.Key.IndexOf("FrozenKing", StringComparison.OrdinalIgnoreCase) >= 0 || prefab.IndexOf(item.Key, StringComparison.OrdinalIgnoreCase) < 0) { continue; } return item.Value; } return -1; } private static bool TryResolveWoodenChance(string prefab, out float chance) { chance = 0f; if (string.IsNullOrEmpty(prefab)) { return false; } if (WoodenIdolChances.TryGetValue(prefab, out chance)) { return true; } foreach (KeyValuePair<string, float> woodenIdolChance in WoodenIdolChances) { if (prefab.IndexOf(woodenIdolChance.Key, StringComparison.OrdinalIgnoreCase) >= 0) { chance = woodenIdolChance.Value; return true; } } return false; } private static bool TryResolveMob(string prefab, out MobIdol info) { info = default(MobIdol); if (string.IsNullOrEmpty(prefab)) { return false; } if (MobIdolTable.TryGetValue(prefab, out info)) { return true; } foreach (KeyValuePair<string, MobIdol> item in MobIdolTable) { if (prefab.IndexOf(item.Key, StringComparison.OrdinalIgnoreCase) >= 0) { info = item.Value; return true; } } return false; } } internal class FoPRagdollIdolDrop : MonoBehaviour { public int Tier; public int BattleCount; public int ArmorCount; private bool _spawned; private void OnDestroy() { //IL_001b: Unknown result type (might be due to invalid IL or missing references) //IL_0020: Unknown result type (might be due to invalid IL or missing references) //IL_002a: Unknown result type (might be due to invalid IL or missing references) //IL_002f: Unknown result type (might be due to invalid IL or missing references) //IL_0034: Unknown result type (might be due to invalid IL or missing references) //IL_0035: Unknown result type (might be due to invalid IL or missing references) if (_spawned) { return; } _spawned = true; try { Vector3 pos = ((Component)this).transform.position + Vector3.up * 0.25f; ForgePatches.SpawnIdolsFromRagdoll(pos, Tier, BattleCount, ArmorCount); } catch { } } } [BepInPlugin("devon752.ForgingPotential", "Forging Potential", "1.0.0")] [BepInProcess("valheim.exe")] [BepInProcess("valheim_server.exe")] public class ForgingPotentialPlugin : BaseUnityPlugin { public const string ModGUID = "devon752.ForgingPotential"; public const string ModName = "Forging Potential"; public const string ModVersion = "1.0.0"; internal static ForgingPotentialPlugin Instance; private ConfigSync _configSync; private readonly Harmony _harmony = new Harmony("devon752.ForgingPotential"); private ConfigReloadWatcher _watcher; public static ConfigEntry<bool> LockConfig; public static ConfigEntry<bool> EnableProgressiveForge; public static ConfigEntry<float> ForgeSuccessStep; public static ConfigEntry<float> ForgeSuccessFloor; public static ConfigEntry<bool> EnableBossIdolDrops; public static ConfigEntry<bool> EnableMobIdolDrops; public static ConfigEntry<bool> ShowForgeTooltip; public static ConfigEntry<bool> EnableModLogs; private void Awake() { //IL_01e1: Unknown result type (might be due to invalid IL or missing references) //IL_01e7: Expected O, but got Unknown Instance = this; try { _configSync = new ConfigSync("devon752.ForgingPotential") { DisplayName = "Forging Potential", CurrentVersion = "1.0.0", MinimumRequiredVersion = "1.0.0", ModRequired = false }; LockConfig = BindSynced(_configSync, "1 - General", "Lock configuration", value: true, "If enabled, the host/server decides values for everyone. Clients receive these values and cannot override them."); _configSync.AddLockingConfigEntry<bool>(LockConfig); EnableModLogs = BindSynced(_configSync, "1 - General", "Enable mod logs (Client-side)", value: false, "If enabled, writes the mod's debug/info lines to the BepInEx log.", synchronizedSetting: false); EnableProgressiveForge = BindSynced(_configSync, "2 - Features", "Forging Potential", value: true, "If enabled, uses the Forging Potential system instead of vanilla.\n- every item starts with 100% chance of success on the Forge of Potential\n- every refinement attempt degrades the chance of success by 5% (minimum 65%) and every repair restores 5%\n- failing to refine still results in the items destruction\n"); ForgeSuccessStep = BindSynced(_configSync, "2 - Features", "Forge degradation/repair amount", 5f, "[Only used when Forging Potential is ON]\nIf set to 5, the item loses 5% success chance when refined and restores 5% success chance when repaired (every item starts with 100% chance of success)."); ForgeSuccessFloor = BindSynced(_configSync, "2 - Features", "Forge chance minimum", 65f, "[Only used when Forging Potential is ON]\nIf set to 65, the item cannot degrade below 65% chance of success."); EnableBossIdolDrops = BindSynced(_configSync, "2 - Features", "Boss idol drops", value: true, "If enabled, bosses drop idols corresponding to their biome."); EnableMobIdolDrops = BindSynced(_configSync, "2 - Features", "Mob idol drops", value: true, "If enabled, each biomes toughest monsters have a chance to drop idols."); ShowForgeTooltip = BindSynced(_configSync, "2 - Features", "Show tooltips (Client-side)", value: true, "If enabled, shows forge success chance on items.", synchronizedSetting: false); _harmony.PatchAll(typeof(ForgePatches)); try { foreach (MethodInfo declaredMethod in AccessTools.GetDeclaredMethods(typeof(ItemData))) { if (!(declaredMethod.Name != "GetTooltip") && !(declaredMethod.ReturnType != typeof(string))) { HarmonyMethod val = new HarmonyMethod(typeof(ForgePatches), "ItemData_GetTooltip_Postfix", (Type[])null); _harmony.Patch((MethodBase)declaredMethod, (HarmonyMethod)null, val, (HarmonyMethod)null, (HarmonyMethod)null, (HarmonyMethod)null); LogAlways($"Patched ItemData.GetTooltip ({declaredMethod.GetParameters().Length} args)"); } } } catch (Exception ex) { ((BaseUnityPlugin)this).Logger.LogWarning((object)("[Forging Potential] GetTooltip patch: " + ex.Message)); } _watcher = ((Component)this).gameObject.AddComponent<ConfigReloadWatcher>(); _watcher.Initialize(((BaseUnityPlugin)this).Config, _configSync); ((BaseUnityPlugin)this).Logger.LogInfo((object)"[Forging Potential] Loaded v1.0.0"); } catch (Exception arg) { ((BaseUnityPlugin)this).Logger.LogError((object)string.Format("[{0}] Awake failed: {1}", "Forging Potential", arg)); } } private void OnDestroy() { _harmony.UnpatchSelf(); } public static bool IsProgressiveForgeEnabled() { return EnableProgressiveForge != null && EnableProgressiveForge.Value; } public static float GetSuccessPercent(int attemptIndex) { if (!IsProgressiveForgeEnabled()) { return 65f; } float num = ((ForgeSuccessStep != null) ? ForgeSuccessStep.Value : 5f); float num2 = ((ForgeSuccessFloor != null) ? ForgeSuccessFloor.Value : 65f); if (num < 0f) { num = 0f; } if (num2 < 0f) { num2 = 0f; } if (num2 > 100f) { num2 = 100f; } if (attemptIndex < 0) { attemptIndex = 0; } float num3 = 100f - (float)attemptIndex * num; if (num3 < num2) { num3 = num2; } if (num3 > 100f) { num3 = 100f; } return num3; } internal ConfigEntry<T> BindSynced<T>(ConfigSync configSync, string group, string name, T value, string description, bool synchronizedSetting = true) { //IL_000f: Unknown result type (might be due to invalid IL or missing references) //IL_001b: Expected O, but got Unknown return BindSynced(configSync, group, name, value, new ConfigDescription(description, (AcceptableValueBase)null, Array.Empty<object>()), synchronizedSetting); } internal ConfigEntry<T> BindSynced<T>(ConfigSync configSync, string group, string name, T value, ConfigDescription description, bool synchronizedSetting = true) { ConfigEntry<T> val = ((BaseUnityPlugin)this).Config.Bind<T>(group, name, value, description); SyncedConfigEntry<T> syncedConfigEntry = configSync.AddConfigEntry<T>(val); syncedConfigEntry.SynchronizedConfig = synchronizedSetting; return val; } internal static void LogAlways(string msg) { if (!((Object)(object)Instance == (Object)null) && EnableModLogs != null && EnableModLogs.Value) { ((BaseUnityPlugin)Instance).Logger.LogInfo((object)("[Forging Potential] " + msg)); } } } internal sealed class ConfigReloadWatcher : MonoBehaviour { private const float PollIntervalSeconds = 60f; private ConfigFile _config; private ConfigSync _configSync; private string _path; private DateTime _lastWriteUtc; private float _pollElapsed; internal void Initialize(ConfigFile config, ConfigSync configSync) { _config = config; _configSync = configSync; _path = config.ConfigFilePath; _lastWriteUtc = ReadWriteTime(); _pollElapsed = 0f; } private void Update() { _pollElapsed += Time.unscaledDeltaTime; if (!(_pollElapsed < 60f)) { _pollElapsed = 0f; PollFile(); } } private void PollFile() { if (_config == null || string.IsNullOrEmpty(_path) || !File.Exists(_path)) { return; } DateTime dateTime = ReadWriteTime(); if (!(dateTime <= _lastWriteUtc)) { _lastWriteUtc = dateTime; if (_configSync == null || _configSync.IsSourceOfTruth) { _config.Reload(); ForgingPotentialPlugin.LogAlways("Config file changed on disk — reloaded."); } } } private DateTime ReadWriteTime() { try { return File.GetLastWriteTimeUtc(_path); } catch { return DateTime.MinValue; } } } } namespace ServerSync { [PublicAPI] public abstract class OwnConfigEntryBase { public object? LocalBaseValue; public bool SynchronizedConfig = true; public abstract ConfigEntryBase BaseConfig { get; } } [PublicAPI] public class SyncedConfigEntry<T> : OwnConfigEntryBase { public readonly ConfigEntry<T> SourceConfig; public override ConfigEntryBase BaseConfig => (ConfigEntryBase)(object)SourceConfig; public T Value { get { return SourceConfig.Value; } set { SourceConfig.Value = value; } } public SyncedConfigEntry(ConfigEntry<T> sourceConfig) { SourceConfig = sourceConfig; } public void AssignLocalValue(T value) { if (LocalBaseValue == null) { Value = value; } else { LocalBaseValue = value; } } } public abstract class CustomSyncedValueBase { public object? LocalBaseValue; public readonly string Identifier; public readonly Type Type; private object? boxedValue; protected bool localIsOwner; public readonly int Priority; public object? BoxedValue { get { return boxedValue; } set { boxedValue = value; this.ValueChanged?.Invoke(); } } public event Action? ValueChanged; protected CustomSyncedValueBase(ConfigSync configSync, string identifier, Type type, int priority) { Priority = priority; Identifier = identifier; Type = type; configSync.AddCustomValue(this); localIsOwner = configSync.IsSourceOfTruth; configSync.SourceOfTruthChanged += delegate(bool truth) { localIsOwner = truth; }; } } [PublicAPI] public sealed class CustomSyncedValue<T> : CustomSyncedValueBase { public T Value { get { return (T)base.BoxedValue; } set { base.BoxedValue = value; } } public CustomSyncedValue(ConfigSync configSync, string identifier, T value = default(T), int priority = 0) : base(configSync, identifier, typeof(T), priority) { Value = value; } public void AssignLocalValue(T value) { if (localIsOwner) { Value = value; } else { LocalBaseValue = value; } } } internal class ConfigurationManagerAttributes { [UsedImplicitly] public bool? ReadOnly = false; } [PublicAPI] public class ConfigSync { [HarmonyPatch(typeof(ZRpc), "HandlePackage")] private static class SnatchCurrentlyHandlingRPC { public static ZRpc? currentRpc; [HarmonyPrefix] private static void Prefix(ZRpc __instance) { currentRpc = __instance; } } [HarmonyPatch(typeof(ZNet), "Awake")] internal static class RegisterRPCPatch { [HarmonyPostfix] private static void Postfix(ZNet __instance) { isServer = __instance.IsServer(); foreach (ConfigSync configSync2 in configSyncs) { ZRoutedRpc.instance.Register<ZPackage>(configSync2.Name + " ConfigSync", (Action<long, ZPackage>)configSync2.RPC_FromOtherClientConfigSync); if (isServer) { configSync2.InitialSyncDone = true; Debug.Log((object)("Registered '" + configSync2.Name + " ConfigSync' RPC - waiting for incoming connections")); } } if (isServer) { ((MonoBehaviour)__instance).StartCoroutine(WatchAdminListChanges()); } static void SendAdmin(List<ZNetPeer> peers, bool isAdmin) { ZPackage package = ConfigsToPackage(null, null, new PackageEntry[1] { new PackageEntry { section = "Internal", key = "lockexempt", type = typeof(bool), value = isAdmin } }); ConfigSync configSync = configSyncs.First(); if (configSync != null) { ((MonoBehaviour)ZNet.instance).StartCoroutine(configSync.sendZPackage(peers, package)); } } static IEnumerator WatchAdminListChanges() { MethodInfo listContainsId = AccessTools.DeclaredMethod(typeof(ZNet), "ListContainsId", (Type[])null, (Type[])null); SyncedList adminList = (SyncedList)AccessTools.DeclaredField(typeof(ZNet), "m_adminList").GetValue(ZNet.instance); List<string> CurrentList = new List<string>(adminList.GetList()); while (true) { yield return (object)new WaitForSeconds(30f); if (!adminList.GetList().SequenceEqual(CurrentList)) { CurrentList = new List<string>(adminList.GetList()); List<ZNetPeer> adminPeer = ZNet.instance.GetPeers().Where(delegate(ZNetPeer p) { string hostName = p.m_rpc.GetSocket().GetHostName(); return ((object)listContainsId == null) ? adminList.Contains(hostName) : ((bool)listContainsId.Invoke(ZNet.instance, new object[2] { adminList, hostName })); }).ToList(); List<ZNetPeer> nonAdminPeer = ZNet.instance.GetPeers().Except(adminPeer).ToList(); SendAdmin(nonAdminPeer, isAdmin: false); SendAdmin(adminPeer, isAdmin: true); } } } } } [HarmonyPatch(typeof(ZNet), "OnNewConnection")] private static class RegisterClientRPCPatch { [HarmonyPostfix] private static void Postfix(ZNet __instance, ZNetPeer peer) { if (__instance.IsServer()) { return; } foreach (ConfigSync configSync in configSyncs) { peer.m_rpc.Register<ZPackage>(configSync.Name + " ConfigSync", (Action<ZRpc, ZPackage>)configSync.RPC_FromServerConfigSync); } } } private class ParsedConfigs { public readonly Dictionary<OwnConfigEntryBase, object?> configValues = new Dictionary<OwnConfigEntryBase, object>(); public readonly Dictionary<CustomSyncedValueBase, object?> customValues = new Dictionary<CustomSyncedValueBase, object>(); } [HarmonyPatch(typeof(ZNet), "Shutdown")] private class ResetConfigsOnShutdown { [HarmonyPostfix] private static void Postfix() { ProcessingServerUpdate = true; foreach (ConfigSync configSync in configSyncs) { configSync.resetConfigsFromServer(); configSync.IsSourceOfTruth = true; configSync.InitialSyncDone = false; } ProcessingServerUpdate = false; } } [HarmonyPatch(typeof(ZNet), "RPC_PeerInfo")] private class SendConfigsAfterLogin { private class BufferingSocket : ZPlayFabSocket, ISocket { public volatile bool finished = false; public volatile int versionMatchQueued = -1; public readonly List<ZPackage> Package = new List<ZPackage>(); public readonly ISocket Original; public BufferingSocket(ISocket original) { Original = original; } public bool IsConnected() { return Original.IsConnected(); } public ZPackage Recv() { return Original.Recv(); } public int GetSendQueueSize() { return Original.GetSendQueueSize(); } public int GetCurrentSendRate() { return Original.GetCurrentSendRate(); } public bool IsHost() { return Original.IsHost(); } public void Dispose() { Original.Dispose(); } public bool GotNewData() { return Original.GotNewData(); } public void Close() { Original.Close(); } public string GetEndPointString() { return Original.GetEndPointString(); } public void GetAndResetStats(out int totalSent, out int totalRecv) { Original.GetAndResetStats(ref totalSent, ref totalRecv); } public void GetConnectionQuality(out float localQuality, out float remoteQuality, out int ping, out float outByteSec, out float inByteSec) { Original.GetConnectionQuality(ref localQuality, ref remoteQuality, ref ping, ref outByteSec, ref inByteSec); } public ISocket Accept() { return Original.Accept(); } public int GetHostPort() { return Original.GetHostPort(); } public bool Flush() { return Original.Flush(); } public string GetHostName() { return Original.GetHostName(); } public void VersionMatch() { if (finished) { Original.VersionMatch(); } else { versionMatchQueued = Package.Count; } } public void Send(ZPackage pkg) { //IL_0057: Unknown result type (might be due to invalid IL or missing references) //IL_005d: Expected O, but got Unknown int pos = pkg.GetPos(); pkg.SetPos(0); int num = pkg.ReadInt(); if ((num == StringExtensionMethods.GetStableHashCode("PeerInfo") || num == StringExtensionMethods.GetStableHashCode("RoutedRPC") || num == StringExtensionMethods.GetStableHashCode("ZDOData")) && !finished) { ZPackage val = new ZPackage(pkg.GetArray()); val.SetPos(pos); Package.Add(val); } else { pkg.SetPos(pos); Original.Send(pkg); } } } [HarmonyPriority(800)] [HarmonyPrefix] private static void Prefix(ref Dictionary<Assembly, BufferingSocket>? __state, ZNet __instance, ZRpc rpc) { //IL_0078: Unknown result type (might be due to invalid IL or missing references) //IL_007e: Invalid comparison between Unknown and I4 if (!__instance.IsServer()) { return; } BufferingSocket bufferingSocket = new BufferingSocket(rpc.GetSocket()); AccessTools.DeclaredField(typeof(ZRpc), "m_socket").SetValue(rpc, bufferingSocket); object? obj = AccessTools.DeclaredMethod(typeof(ZNet), "GetPeer", new Type[1] { typeof(ZRpc) }, (Type[])null).Invoke(__instance, new object[1] { rpc }); ZNetPeer val = (ZNetPeer)((obj is ZNetPeer) ? obj : null); if (val != null && (int)ZNet.m_onlineBackend > 0) { FieldInfo fieldInfo = AccessTools.DeclaredField(typeof(ZNetPeer), "m_socket"); object? value = fieldInfo.GetValue(val); ZPlayFabSocket val2 = (ZPlayFabSocket)((value is ZPlayFabSocket) ? value : null); if (val2 != null) { typeof(ZPlayFabSocket).GetField("m_remotePlayerId").SetValue(bufferingSocket, val2.m_remotePlayerId); } fieldInfo.SetValue(val, bufferingSocket); } if (__state == null) { __state = new Dictionary<Assembly, BufferingSocket>(); } __state[Assembly.GetExecutingAssembly()] = bufferingSocket; } [HarmonyPostfix] private static void Postfix(Dictionary<Assembly, BufferingSocket> __state, ZNet __instance, ZRpc rpc) { ZNetPeer peer; if (__instance.IsServer()) { object obj = AccessTools.DeclaredMethod(typeof(ZNet), "GetPeer", new Type[1] { typeof(ZRpc) }, (Type[])null).Invoke(__instance, new object[1] { rpc }); peer = (ZNetPeer)((obj is ZNetPeer) ? obj : null); if (peer == null) { SendBufferedData(); } else { ((MonoBehaviour)__instance).StartCoroutine(sendAsync()); } } void SendBufferedData() { if (rpc.GetSocket() is BufferingSocket bufferingSocket) { AccessTools.DeclaredField(typeof(ZRpc), "m_socket").SetValue(rpc, bufferingSocket.Original); object? obj2 = AccessTools.DeclaredMethod(typeof(ZNet), "GetPeer", new Type[1] { typeof(ZRpc) }, (Type[])null).Invoke(__instance, new object[1] { rpc }); ZNetPeer val = (ZNetPeer)((obj2 is ZNetPeer) ? obj2 : null); if (val != null) { AccessTools.DeclaredField(typeof(ZNetPeer), "m_socket").SetValue(val, bufferingSocket.Original); } } BufferingSocket bufferingSocket2 = __state[Assembly.GetExecutingAssembly()]; bufferingSocket2.finished = true; for (int i = 0; i < bufferingSocket2.Package.Count; i++) { if (i == bufferingSocket2.versionMatchQueued) { bufferingSocket2.Original.VersionMatch(); } bufferingSocket2.Original.Send(bufferingSocket2.Package[i]); } if (bufferingSocket2.Package.Count == bufferingSocket2.versionMatchQueued) { bufferingSocket2.Original.VersionMatch(); } } IEnumerator sendAsync() { foreach (ConfigSync configSync in configSyncs) { List<PackageEntry> entries = new List<PackageEntry>(); if (configSync.CurrentVersion != null) { entries.Add(new PackageEntry { section = "Internal", key = "serverversion", type = typeof(string), value = configSync.CurrentVersion }); } MethodInfo listContainsId = AccessTools.DeclaredMethod(typeof(ZNet), "ListContainsId", (Type[])null, (Type[])null); SyncedList adminList = (SyncedList)AccessTools.DeclaredField(typeof(ZNet), "m_adminList").GetValue(ZNet.instance); entries.Add(new PackageEntry { section = "Internal", key = "lockexempt", type = typeof(bool), value = (((object)listContainsId == null) ? ((object)adminList.Contains(rpc.GetSocket().GetHostName())) : listContainsId.Invoke(ZNet.instance, new object[2] { adminList, rpc.GetSocket().GetHostName() })) }); ZPackage package = ConfigsToPackage(configSync.allConfigs.Select((OwnConfigEntryBase c) => c.BaseConfig), configSync.allCustomValues, entries, partial: false); yield return ((MonoBehaviour)__instance).StartCoroutine(configSync.sendZPackage(new List<ZNetPeer> { peer }, package)); } SendBufferedData(); } } } private class PackageEntry { public string section = null; public string key = null; public Type type = null; public object? value; } [HarmonyPatch(typeof(ConfigEntryBase), "GetSerializedValue")] private static class PreventSavingServerInfo { [HarmonyPrefix] private static bool Prefix(ConfigEntryBase __instance, ref string __result) { OwnConfigEntryBase ownConfigEntryBase = configData(__instance); if (ownConfigEntryBase == null || isWritableConfig(ownConfigEntryBase)) { return true; } __result = TomlTypeConverter.ConvertToString(ownConfigEntryBase.LocalBaseValue, __instance.SettingType); return false; } } [HarmonyPatch(typeof(ConfigEntryBase), "SetSerializedValue")] private static class PreventConfigRereadChangingValues { [HarmonyPrefix] private static bool Prefix(ConfigEntryBase __instance, string value) { OwnConfigEntryBase ownConfigEntryBase = configData(__instance); if (ownConfigEntryBase == null || ownConfigEntryBase.LocalBaseValue == null) { return true; } try { ownConfigEntryBase.LocalBaseValue = TomlTypeConverter.ConvertToValue(value, __instance.SettingType); } catch (Exception ex) { Debug.LogWarning((object)$"Config value of setting \"{__instance.Definition}\" could not be parsed and will be ignored. Reason: {ex.Message}; Value: {value}"); } return false; } } private class InvalidDeserializationTypeException : Exception { public string expected = null; public string received = null; public string field = ""; } public static bool ProcessingServerUpdate; public readonly string Name; public string? DisplayName; public string? CurrentVersion; public string? MinimumRequiredVersion; public bool ModRequired = false; private bool? forceConfigLocking; private bool isSourceOfTruth = true; private static readonly HashSet<ConfigSync> configSyncs; private readonly HashSet<OwnConfigEntryBase> allConfigs = new HashSet<OwnConfigEntryBase>(); private HashSet<CustomSyncedValueBase> allCustomValues = new HashSet<CustomSyncedValueBase>(); private static bool isServer; private static bool lockExempt; private OwnConfigEntryBase? lockedConfig = null; private const byte PARTIAL_CONFIGS = 1; private const byte FRAGMENTED_CONFIG = 2; private const byte COMPRESSED_CONFIG = 4; private readonly Dictionary<string, SortedDictionary<int, byte[]>> configValueCache = new Dictionary<string, SortedDictionary<int, byte[]>>(); private readonly List<KeyValuePair<long, string>> cacheExpirations = new List<KeyValuePair<long, string>>(); private static long packageCounter; public bool IsLocked { get { bool? flag = forceConfigLocking; bool num; if (!flag.HasValue) { if (lockedConfig == null) { goto IL_0052; } num = ((IConvertible)lockedConfig.BaseConfig.BoxedValue).ToInt32(CultureInfo.InvariantCulture) != 0; } else { num = flag == true; } if (!num) { goto IL_0052; } int result = ((!lockExempt) ? 1 : 0); goto IL_0053; IL_0052: result = 0; goto IL_0053; IL_0053: return (byte)result != 0; } set { forceConfigLocking = value; } } public bool IsAdmin => lockExempt || isSourceOfTruth; public bool IsSourceOfTruth { get { return isSourceOfTruth; } private set { if (value != isSourceOfTruth) { isSourceOfTruth = value; this.SourceOfTruthChanged?.Invoke(value); } } } public bool InitialSyncDone { get; private set; } = false; public event Action<bool>? SourceOfTruthChanged; private event Action? lockedConfigChanged; static ConfigSync() { ProcessingServerUpdate = false; configSyncs = new HashSet<ConfigSync>(); lockExempt = false; packageCounter = 0L; RuntimeHelpers.RunClassConstructor(typeof(VersionCheck).TypeHandle); } public ConfigSync(string name) { Name = name; configSyncs.Add(this); new VersionCheck(this); } public SyncedConfigEntry<T> AddConfigEntry<T>(ConfigEntry<T> configEntry) { OwnConfigEntryBase ownConfigEntryBase = configData((ConfigEntryBase)(object)configEntry); SyncedConfigEntry<T> syncedEntry = ownConfigEntryBase as SyncedConfigEntry<T>; if (syncedEntry == null) { syncedEntry = new SyncedConfigEntry<T>(configEntry); AccessTools.DeclaredField(typeof(ConfigDescription), "<Tags>k__BackingField").SetValue(((ConfigEntryBase)configEntry).Description, new object[1] { new ConfigurationManagerAttributes() }.Concat(((ConfigEntryBase)configEntry).Description.Tags ?? Array.Empty<object>()).Concat(new SyncedConfigEntry<T>[1] { syncedEntry }).ToArray()); configEntry.SettingChanged += delegate { if (!ProcessingServerUpdate && syncedEntry.SynchronizedConfig) { Broadcast(0L, (ConfigEntryBase)configEntry); } }; allConfigs.Add(syncedEntry); } return syncedEntry; } public SyncedConfigEntry<T> AddLockingConfigEntry<T>(ConfigEntry<T> lockingConfig) where T : IConvertible { if (lockedConfig != null) { throw new Exception("Cannot initialize locking ConfigEntry twice"); } lockedConfig = AddConfigEntry<T>(lockingConfig); lockingConfig.SettingChanged += delegate { this.lockedConfigChanged?.Invoke(); }; return (SyncedConfigEntry<T>)lockedConfig; } internal void AddCustomValue(CustomSyncedValueBase customValue) { if (allCustomValues.Select((CustomSyncedValueBase v) => v.Identifier).Concat(new string[1] { "serverversion" }).Contains(customValue.Identifier)) { throw new Exception("Cannot have multiple settings with the same name or with a reserved name (serverversion)"); } allCustomValues.Add(customValue); allCustomValues = new HashSet<CustomSyncedValueBase>(allCustomValues.OrderByDescending((CustomSyncedValueBase v) => v.Priority)); customValue.ValueChanged += delegate { if (!ProcessingServerUpdate) { Broadcast(0L, customValue); } }; } private void RPC_FromServerConfigSync(ZRpc rpc, ZPackage package) { lockedConfigChanged += serverLockedSettingChanged; IsSourceOfTruth = false; if (HandleConfigSyncRPC(0L, package, clientUpdate: false)) { InitialSyncDone = true; } } private void RPC_FromOtherClientConfigSync(long sender, ZPackage package) { HandleConfigSyncRPC(sender, package, clientUpdate: true); } private bool HandleConfigSyncRPC(long sender, ZPackage package, bool clientUpdate) { //IL_0076: Unknown result type (might be due to invalid IL or missing references) //IL_007d: Expected O, but got Unknown //IL_0250: Unknown result type (might be due to invalid IL or missing references) //IL_0257: Expected O, but got Unknown //IL_01ea: Unknown result type (might be due to invalid IL or missing references) //IL_01f1: Expected O, but got Unknown try { if (isServer && IsLocked) { ZRpc? currentRpc = SnatchCurrentlyHandlingRPC.currentRpc; object obj; if (currentRpc == null) { obj = null; } else { ISocket socket = currentRpc.GetSocket(); obj = ((socket != null) ? socket.GetHostName() : null); } string text = (string)obj; if (text != null) { MethodInfo methodInfo = AccessTools.DeclaredMethod(typeof(ZNet), "ListContainsId", (Type[])null, (Type[])null); SyncedList val = (SyncedList)AccessTools.DeclaredField(typeof(ZNet), "m_adminList").GetValue(ZNet.instance); if (!(((object)methodInfo == null) ? val.Contains(text) : ((bool)methodInfo.Invoke(ZNet.instance, new object[2] { val, text })))) { return false; } } } cacheExpirations.RemoveAll(delegate(KeyValuePair<long, string> kv) { if (kv.Key < DateTimeOffset.Now.Ticks) { configValueCache.Remove(kv.Value); return true; } return false; }); byte b = package.ReadByte(); if ((b & 2) != 0) { long num = package.ReadLong(); string text2 = sender.ToString() + num; if (!configValueCache.TryGetValue(text2, out SortedDictionary<int, byte[]> value)) { value = new SortedDictionary<int, byte[]>(); configValueCache[text2] = value; cacheExpirations.Add(new KeyValuePair<long, string>(DateTimeOffset.Now.AddSeconds(60.0).Ticks, text2)); } int key = package.ReadInt(); int num2 = package.ReadInt(); value.Add(key, package.ReadByteArray()); if (value.Count < num2) { return false; } configValueCache.Remove(text2); package = new ZPackage(value.Values.SelectMany((byte[] a) => a).ToArray()); b = package.ReadByte(); } ProcessingServerUpdate = true; if ((b & 4) != 0) { byte[] buffer = package.ReadByteArray(); MemoryStream stream = new MemoryStream(buffer); MemoryStream memoryStream = new MemoryStream(); using (DeflateStream deflateStream = new DeflateStream(stream, CompressionMode.Decompress)) { deflateStream.CopyTo(memoryStream); } package = new ZPackage(memoryStream.ToArray()); b = package.ReadByte(); } if ((b & 1) == 0) { resetConfigsFromServer(); } ParsedConfigs parsedConfigs = ReadConfigsFromPackage(package); ConfigFile val2 = null; bool saveOnConfigSet = false; foreach (KeyValuePair<OwnConfigEntryBase, object> configValue in parsedConfigs.configValues) { if (!isServer && configValue.Key.LocalBaseValue == null) { configValue.Key.LocalBaseValue = configValue.Key.BaseConfig.BoxedValue; } if (val2 == null) { val2 = configValue.Key.BaseConfig.ConfigFile; saveOnConfigSet = val2.SaveOnConfigSet; val2.SaveOnConfigSet = false; } configValue.Key.BaseConfig.BoxedValue = configValue.Value; } if (val2 != null) { val2.SaveOnConfigSet = saveOnConfigSet; val2.Save(); } foreach (KeyValuePair<CustomSyncedValueBase, object> customValue in parsedConfigs.customValues) { if (!isServer) { CustomSyncedValueBase key2 = customValue.Key; if (key2.LocalBaseValue == null) { key2.LocalBaseValue = customValue.Key.BoxedValue; } } customValue.Key.BoxedValue = customValue.Value; } Debug.Log((object)string.Format("Received {0} configs and {1} custom values from {2} for mod {3}", parsedConfigs.configValues.Count, parsedConfigs.customValues.Count, (isServer || clientUpdate) ? $"client {sender}" : "the server", DisplayName ?? Name)); if (!isServer) { serverLockedSettingChanged(); } return true; } finally { ProcessingServerUpdate = false; } } private ParsedConfigs ReadConfigsFromPackage(ZPackage package) { ParsedConfigs parsedConfigs = new ParsedConfigs(); Dictionary<string, OwnConfigEntryBase> dictionary = allConfigs.Where((OwnConfigEntryBase c) => c.SynchronizedConfig).ToDictionary((OwnConfigEntryBase c) => c.BaseConfig.Definition.Section + "_" + c.BaseConfig.Definition.Key, (OwnConfigEntryBase c) => c); Dictionary<string, CustomSyncedValueBase> dictionary2 = allCustomValues.ToDictionary((CustomSyncedValueBase c) => c.Identifier, (CustomSyncedValueBase c) => c); int num = package.ReadInt(); for (int num2 = 0; num2 < num; num2++) { string text = package.ReadString(); string text2 = package.ReadString(); string text3 = package.ReadString(); Type type = Type.GetType(text3); if (text3 == "" || type != null) { object obj; try { obj = ((text3 == "") ? null : ReadValueWithTypeFromZPackage(package, type)); } catch (InvalidDeserializationTypeException ex) { Debug.LogWarning((object)("Got unexpected struct internal type " + ex.received + " for field " + ex.field + " struct " + text3 + " for " + text2 + " in section " + text + " for mod " + (DisplayName ?? Name) + ", expecting " + ex.expected)); continue; } OwnConfigEntryBase value2; if (text == "Internal") { CustomSyncedValueBase value; if (text2 == "serverversion") { if (obj?.ToString() != CurrentVersion) { Debug.LogWarning((object)("Received server version is not equal: server version = " + (obj?.ToString() ?? "null") + "; local version = " + (CurrentVersion ?? "unknown"))); } } else if (text2 == "lockexempt") { if (obj is bool flag) { lockExempt = flag; } } else if (dictionary2.TryGetValue(text2, out value)) { if ((text3 == "" && (!value.Type.IsValueType || Nullable.GetUnderlyingType(value.Type) != null)) || GetZPackageTypeString(value.Type) == text3) { parsedConfigs.customValues[value] = obj; continue; } Debug.LogWarning((object)("Got unexpected type " + text3 + " for internal value " + text2 + " for mod " + (DisplayName ?? Name) + ", expecting " + value.Type.AssemblyQualifiedName)); } } else if (dictionary.TryGetValue(text + "_" + text2, out value2)) { Type type2 = configType(value2.BaseConfig); if ((text3 == "" && (!type2.IsValueType || Nullable.GetUnderlyingType(type2) != null)) || GetZPackageTypeString(type2) == text3) { parsedConfigs.configValues[value2] = obj; continue; } Debug.LogWarning((object)("Got unexpected type " + text3 + " for " + text2 + " in section " + text + " for mod " + (DisplayName ?? Name) + ", expecting " + type2.AssemblyQualifiedName)); } else { Debug.LogWarning((object)("Received unknown config entry " + text2 + " in section " + text + " for mod " + (DisplayName ?? Name) + ". This may happen if client and server versions of the mod do not match.")); } continue; } Debug.LogWarning((object)("Got invalid type " + text3 + ", abort reading of received configs")); return new ParsedConfigs(); } return parsedConfigs; } private static bool isWritableConfig(OwnConfigEntryBase config) { ConfigSync configSync = configSyncs.FirstOrDefault((ConfigSync cs) => cs.allConfigs.Contains(config)); if (configSync == null) { return true; } return configSync.IsSourceOfTruth || !config.SynchronizedConfig || config.LocalBaseValue == null || (!configSync.IsLocked && (config != configSync.lockedConfig || lockExempt)); } private void serverLockedSettingChanged() { foreach (OwnConfigEntryBase allConfig in allConfigs) { configAttribute<ConfigurationManagerAttributes>(allConfig.BaseConfig).ReadOnly = !isWritableConfig(allConfig); } } private void resetConfigsFromServer() { ConfigFile val = null; bool saveOnConfigSet = false; foreach (OwnConfigEntryBase item in allConfigs.Where((OwnConfigEntryBase config) => config.LocalBaseValue != null)) { if (val == null) { val = item.BaseConfig.ConfigFile; saveOnConfigSet = val.SaveOnConfigSet; val.SaveOnConfigSet = false; } item.BaseConfig.BoxedValue = item.LocalBaseValue; item.LocalBaseValue = null; } if (val != null) { val.SaveOnConfigSet = saveOnConfigSet; } foreach (CustomSyncedValueBase item2 in allCustomValues.Where((CustomSyncedValueBase config) => config.LocalBaseValue != null)) { item2.BoxedValue = item2.LocalBaseValue; item2.LocalBaseValue = null; } lockedConfigChanged -= serverLockedSettingChanged; serverLockedSettingChanged(); } private IEnumerator<bool> distributeConfigToPeers(ZNetPeer peer, ZPackage package) { ZRoutedRpc rpc = ZRoutedRpc.instance; if (rpc == null) { yield break; } byte[] data = package.GetArray(); if (data != null && data.LongLength > 250000) { int fragments = (int)(1 + (data.LongLength - 1) / 250000); long packageIdentifier = ++packageCounter; int fragment = 0; while (fragment < fragments) { foreach (bool item in waitForQueue()) { yield return item; } if (peer.m_socket.IsConnected()) { ZPackage fragmentedPackage = new ZPackage(); fragmentedPackage.Write((byte)2); fragmentedPackage.Write(packageIdentifier); fragmentedPackage.Write(fragment); fragmentedPackage.Write(fragments); fragmentedPackage.Write(data.Skip(250000 * fragment).Take(250000).ToArray()); SendPackage(fragmentedPackage); if (fragment != fragments - 1) { yield return true; } int num = fragment + 1; fragment = num; continue; } break; } yield break; } foreach (bool item2 in waitForQueue()) { yield return item2; } SendPackage(package); void SendPackage(ZPackage pkg) { string text = Name + " ConfigSync"; if (isServer) { peer.m_rpc.Invoke(text, new object[1] { pkg }); } else { rpc.InvokeRoutedRPC(peer.m_server ? 0 : peer.m_uid, text, new object[1] { pkg }); } } IEnumerable<bool> waitForQueue() { float timeout = Time.time + 30f; while (peer.m_socket.GetSendQueueSize() > 20000) { if (Time.time > timeout) { Debug.Log((object)$"Disconnecting {peer.m_uid} after 30 seconds config sending timeout"); peer.m_rpc.Invoke("Error", new object[1] { (object)(ConnectionStatus)5 }); ZNet.instance.Disconnect(peer); break; } yield return false; } } } private IEnumerator sendZPackage(long target, ZPackage package) { if (!Object.op_Implicit((Object)(object)ZNet.instance)) { return Enumerable.Empty<object>().GetEnumerator(); } List<ZNetPeer> list = (List<ZNetPeer>)AccessTools.DeclaredField(typeof(ZRoutedRpc), "m_peers").GetValue(ZRoutedRpc.instance); if (target != 0) { list = list.Where((ZNetPeer p) => p.m_uid == target).ToList(); } return sendZPackage(list, package); } private IEnumerator sendZPackage(List<ZNetPeer> peers, ZPackage package) { if (!Object.op_Implicit((Object)(object)ZNet.instance)) { yield break; } byte[] rawData = package.GetArray(); if (rawData != null && rawData.LongLength > 10000) { ZPackage compressedPackage = new ZPackage(); compressedPackage.Write((byte)4); MemoryStream output = new MemoryStream(); using (DeflateStream deflateStream = new DeflateStream(output, CompressionLevel.Optimal)) { deflateStream.Write(rawData, 0, rawData.Length); } compressedPackage.Write(output.ToArray()); package = compressedPackage; } List<IEnumerator<bool>> writers = (from p in peers where p.IsReady() select distributeConfigToPeers(p, package)).ToList(); writers.RemoveAll((IEnumerator<bool> writer) => !writer.MoveNext()); while (writers.Count > 0) { yield return null; writers.RemoveAll((IEnumerator<bool> writer) => !writer.MoveNext()); } } private void Broadcast(long target, params ConfigEntryBase[] configs) { if (!IsLocked || isServer) { ZPackage package = ConfigsToPackage(configs); ZNet instance = ZNet.instance; if (instance != null) { ((MonoBehaviour)instance).StartCoroutine(sendZPackage(target, package)); } } } private void Broadcast(long target, params CustomSyncedValueBase[] customValues) { if (!IsLocked || isServer) { ZPackage package = ConfigsToPackage(null, customValues); ZNet instance = ZNet.instance; if (instance != null) { ((MonoBehaviour)instance).StartCoroutine(sendZPackage(target, package)); } } } private static OwnConfigEntryBase? configData(ConfigEntryBase config) { return config.Description.Tags?.OfType<OwnConfigEntryBase>().SingleOrDefault(); } public static SyncedConfigEntry<T>? ConfigData<T>(ConfigEntry<T> config) { return ((ConfigEntryBase)config).Description.Tags?.OfType<SyncedConfigEntry<T>>().SingleOrDefault(); } private static T configAttribute<T>(ConfigEntryBase config) { return config.Description.Tags.OfType<T>().First(); } private static Type configType(ConfigEntryBase config) { return configType(config.SettingType); } private static Type configType(Type type) { return type.IsEnum ? Enum.GetUnderlyingType(type) : type; } private static ZPackage ConfigsToPackage(IEnumerable<ConfigEntryBase>? configs = null, IEnumerable<CustomSyncedValueBase>? customValues = null, IEnumerable<PackageEntry>? packageEntries = null, bool partial = true) { //IL_0051: Unknown result type (might be due to invalid IL or missing references) //IL_0057: Expected O, but got Unknown List<ConfigEntryBase> list = configs?.Where((ConfigEntryBase config) => configData(config).SynchronizedConfig).ToList() ?? new List<ConfigEntryBase>(); List<CustomSyncedValueBase> list2 = customValues?.ToList() ?? new List<CustomSyncedValueBase>(); ZPackage val = new ZPackage(); val.Write((byte)(partial ? 1 : 0)); val.Write(list.Count + list2.Count + (packageEntries?.Count() ?? 0)); foreach (PackageEntry item in packageEntries ?? Array.Empty<PackageEntry>()) { AddEntryToPackage(val, item); } foreach (CustomSyncedValueBase item2 in list2) { AddEntryToPackage(val, new PackageEntry { section = "Internal", key = item2.Identifier, type = item2.Type, value = item2.BoxedValue }); } foreach (ConfigEntryBase item3 in list) { AddEntryToPackage(val, new PackageEntry { section = item3.Definition.Section, key = item3.Definition.Key, type = configType(item3), value = item3.BoxedValue }); } return val; } private static void AddEntryToPackage(ZPackage package, PackageEntry entry) { package.Write(entry.section); package.Write(entry.key); package.Write((entry.value == null) ? "" : GetZPackageTypeString(entry.type)); AddValueToZPackage(package, entry.value); } private static string GetZPackageTypeString(Type type) { return type.AssemblyQualifiedName; } private static void AddValueToZPackage(ZPackage package, object? value) { Type type = value?.GetType(); if (value is Enum) { value = ((IConvertible)value).ToType(Enum.GetUnderlyingType(value.GetType()), CultureInfo.InvariantCulture); } else { if (value is ICollection collection) { package.Write(collection.Count); { foreach (object item in collection) { AddValueToZPackage(package, item); } return; } } if ((object)type != null && type.IsValueType && !type.IsPrimitive) { FieldInfo[] fields = type.GetFields(BindingFlags.Instance | BindingFlags.Public | BindingFlags.NonPublic); package.Write(fields.Length); FieldInfo[] array = fields; foreach (FieldInfo fieldInfo in array) { package.Write(GetZPackageTypeString(fieldInfo.FieldType)); AddValueToZPackage(package, fieldInfo.GetValue(value)); } return; } } ZRpc.Serialize(new object[1] { value }, ref package); } private static object ReadValueWithTypeFromZPackage(ZPackage package, Type type) { if ((object)type != null && type.IsValueType && !type.IsPrimitive && !type.IsEnum) { FieldInfo[] fields = type.GetFields(BindingFlags.Instance | BindingFlags.Public | BindingFlags.NonPublic); int num = package.ReadInt(); if (num != fields.Length) { throw new InvalidDeserializationTypeException { received = $"(field count: {num})", expected = $"(field count: {fields.Length})" }; } object uninitializedObject = FormatterServices.GetUninitializedObject(type); FieldInfo[] array = fields; foreach (FieldInfo fieldInfo in array) { string text = package.ReadString(); if (text != GetZPackageTypeString(fieldInfo.FieldType)) { throw new InvalidDeserializationTypeException { received = text, expected = GetZPackageTypeString(fieldInfo.FieldType), field = fieldInfo.Name }; } fieldInfo.SetValue(uninitializedObject, ReadValueWithTypeFromZPackage(package, fieldInfo.FieldType)); } return uninitializedObject; } if (type.IsGenericType && type.GetGenericTypeDefinition() == typeof(Dictionary<, >)) { int num2 = package.ReadInt(); IDictionary dictionary = (IDictionary)Activator.CreateInstance(type); Type type2 = typeof(KeyValuePair<, >).MakeGenericType(type.GenericTypeArguments); FieldInfo field = type2.GetField("key", BindingFlags.Instance | BindingFlags.NonPublic); FieldInfo field2 = type2.GetField("value", BindingFlags.Instance | BindingFlags.NonPublic); for (int j = 0; j < num2; j++) { object obj = ReadValueWithTypeFromZPackage(package, type2); dictionary.Add(field.GetValue(obj), field2.GetValue(obj)); } return dictionary; } if (type != typeof(List<string>) && type.IsGenericType) { Type type3 = typeof(ICollection<>).MakeGenericType(type.GenericTypeArguments[0]); if ((object)type3 != null && type3.IsAssignableFrom(type)) { int num3 = package.ReadInt(); object obj2 = Activator.CreateInstance(type); MethodInfo method = type3.GetMethod("Add"); for (int k = 0; k < num3; k++) { method.Invoke(obj2, new object[1] { ReadValueWithTypeFromZPackage(package, type.GenericTypeArguments[0]) }); } return obj2; } } ParameterInfo parameterInfo = (ParameterInfo)FormatterServices.GetUninitializedObject(typeof(ParameterInfo)); AccessTools.DeclaredField(typeof(ParameterInfo), "ClassImpl").SetValue(parameterInfo, type); List<object> source = new List<object>(); ZRpc.Deserialize(new ParameterInfo[2] { null, parameterInfo }, package, ref source); return source.First(); } } [PublicAPI] [HarmonyPatch] public class VersionCheck { private static readonly HashSet<VersionCheck> versionChecks; private static readonly Dictionary<string, string> notProcessedNames; public string Name; private string? displayName; private string? currentVersion; private string? minimumRequiredVersion; public bool ModRequired = true; private string? ReceivedCurrentVersion; private string? ReceivedMinimumRequiredVersion; private readonly List<ZRpc> ValidatedClients = new List<ZRpc>(); private ConfigSync? ConfigSync; public string DisplayName { get { return displayName ?? Name; } set { displayName = value; } } public string CurrentVersion { get { return currentVersion ?? "0.0.0"; } set { currentVersion = value; } } public string MinimumRequiredVersion { get { return minimumRequiredVersion ?? (ModRequired ? CurrentVersion : "0.0.0"); } set { minimumRequiredVersion = value; } } private static void PatchServerSync() { //IL_005e: Unknown result type (might be due to invalid IL or missing references) //IL_0064: Expected O, but got Unknown Patches patchInfo = PatchProcessor.GetPatchInfo((MethodBase)AccessTools.DeclaredMethod(typeof(ZNet), "Awake", (Type[])null, (Type[])null)); if (patchInfo != null && patchInfo.Postfixes.Count((Patch p) => p.PatchMethod.DeclaringType == typeof(ConfigSync.RegisterRPCPatch)) > 0) { return; } Harmony val = new Harmony("org.bepinex.helpers.ServerSync"); foreach (Type item in from t in typeof(ConfigSync).GetNestedTypes(BindingFlags.NonPublic).Concat(new Type[1] { typeof(VersionCheck) }) where t.IsClass select t) { val.PatchAll(item); } } static VersionCheck() { versionChecks = new HashSet<VersionCheck>(); notProcessedNames = new Dictionary<string, string>(); typeof(ThreadingHelper).GetMethod("StartSyncInvoke").Invoke(ThreadingHelper.Instance, new object[1] { new Action(PatchServerSync) }); } public VersionCheck(string name) { Name = name; ModRequired = true; versionChecks.Add(this); } public VersionCheck(ConfigSync configSync) { ConfigSync = configSync; Name = ConfigSync.Name; versionChecks.Add(this); } public void Initialize() { ReceivedCurrentVersion = null; ReceivedMinimumRequiredVersion = null; if (ConfigSync != null) { Name = ConfigSync.Name; DisplayName = ConfigSync.DisplayName; CurrentVersion = ConfigSync.CurrentVersion; MinimumRequiredVersion = ConfigSync.MinimumRequiredVersion; ModRequired = ConfigSync.ModRequired; } } private bool IsVersionOk() { if (ReceivedMinimumRequiredVersion == null || ReceivedCurrentVersion == null) { return !ModRequired; } bool flag = new Version(CurrentVersion) >= new Version(ReceivedMinimumRequiredVersion); bool flag2 = new Version(ReceivedCurrentVersion) >= new Version(MinimumRequiredVersion); return flag && flag2; } private string ErrorClient() { if (ReceivedMinimumRequiredVersion == null) { return DisplayName + " is not installed on the server."; } return (new Version(CurrentVersion) >= new Version(ReceivedMinimumRequiredVersion)) ? (DisplayName + " may not be higher than version " + ReceivedCurrentVersion + ". You have version " + CurrentVersion + ".") : (DisplayName + " needs to be at least version " + ReceivedMinimumRequiredVersion + ". You have version " + CurrentVersion + "."); } private string ErrorServer(ZRpc rpc) { return "Disconnect: The client (" + rpc.GetSocket().GetHostName() + ") doesn't have the correct " + DisplayName + " version " + MinimumRequiredVersion; } private string Error(ZRpc? rpc = null) { return (rpc == null) ? ErrorClient() : ErrorServer(rpc); } private static VersionCheck[] GetFailedClient() { return versionChecks.Where((VersionCheck check) => !check.IsVersionOk()).ToArray(); } private static VersionCheck[] GetFailedServer(ZRpc rpc) { return versionChecks.Where((VersionCheck check) => check.ModRequired && !check.ValidatedClients.Contains(rpc)).ToArray(); } private static void Logout() { Game.instance.Logout(true, true); AccessTools.DeclaredField(typeof(ZNet), "m_connectionStatus").SetValue(null, (object)(ConnectionStatus)3); } private static void DisconnectClient(ZRpc rpc) { rpc.Invoke("Error", new object[1] { 3 }); } private static void CheckVersion(ZRpc rpc, ZPackage pkg) { CheckVersion(rpc, pkg, null); } private static void CheckVersion(ZRpc rpc, ZPackage pkg, Action<ZRpc, ZPackage>? original) { string text = pkg.ReadString(); string text2 = pkg.ReadString(); string text3 = pkg.ReadString(); bool flag = false; foreach (VersionCheck versionCheck in versionChecks) { if (!(text != versionCheck.Name)) { Debug.Log((object)("Received " + versionCheck.DisplayName + " version " + text3 + " and minimum version " + text2 + " from the " + (ZNet.instance.IsServer() ? "client" : "server") + ".")); versionCheck.ReceivedMinimumRequiredVersion = text2; versionCheck.ReceivedCurrentVersion = text3; if (ZNet.instance.IsServer() && versionCheck.IsVersionOk()) { versionCheck.ValidatedClients.Add(rpc); } flag = true; } } if (flag) { return; } pkg.SetPos(0); if (original != null) { original(rpc, pkg); if (pkg.GetPos() == 0) { notProcessedNames.Add(text, text3); } } } [HarmonyPatch(typeof(ZNet), "RPC_PeerInfo")] [HarmonyPrefix] private static bool RPC_PeerInfo(ZRpc rpc, ZNet __instance) { VersionCheck[] array = (__instance.IsServer() ? GetFailedServer(rpc) : GetFailedClient()); if (array.Length == 0) { return true; } VersionCheck[] array2 = array; foreach (VersionCheck versionCheck in array2) { Debug.LogWarning((object)versionCheck.Error(rpc)); } if (__instance.IsServer()) { DisconnectClient(rpc); } else { Logout(); } return false; } [HarmonyPatch(typeof(ZNet), "OnNewConnection")] [HarmonyPrefix] private static void RegisterAndCheckVersion(ZNetPeer peer, ZNet __instance) { //IL_018e: Unknown result type (might be due to invalid IL or missing references) //IL_0195: Expected O, but got Unknown notProcessedNames.Clear(); IDictionary dictionary = (IDictionary)typeof(ZRpc).GetField("m_functions", BindingFlags.Instance | BindingFlags.NonPublic).GetValue(peer.m_rpc); if (dictionary.Contains(StringExtensionMethods.GetStableHashCode("ServerSync VersionCheck"))) { object obj = dictionary[StringExtensionMethods.GetStableHashCode("ServerSync VersionCheck")]; Action<ZRpc, ZPackage> action = (Action<ZRpc, ZPackage>)obj.GetType().GetField("m_action", BindingFlags.Instance | BindingFlags.NonPublic).GetValue(obj); peer.m_rpc.Register<ZPackage>("ServerSync VersionCheck", (Action<ZRpc, ZPackage>)delegate(ZRpc rpc, ZPackage pkg) { CheckVersion(rpc, pkg, action); }); } else { peer.m_rpc.Register<ZPackage>("ServerSync VersionCheck", (Action<ZRpc, ZPackage>)CheckVersion); } foreach (VersionCheck versionCheck in versionChecks) { versionCheck.Initialize(); if (versionCheck.ModRequired || __instance.IsServer()) { Debug.Log((object)("Sending " + versionCheck.DisplayName + " version " + versionCheck.CurrentVersion + " and minimum version " + versionCheck.MinimumRequiredVersion + " to the " + (__instance.IsServer() ? "client" : "server") + ".")); ZPackage val = new ZPackage(); val.Write(versionCheck.Name); val.Write(versionCheck.MinimumRequiredVersion); val.Write(versionCheck.CurrentVersion); peer.m_rpc.Invoke("ServerSync VersionCheck", new object[1] { val }); } } } [HarmonyPatch(typeof(ZNet), "Disconnect")] [HarmonyPrefix] private static void RemoveDisconnected(ZNetPeer peer, ZNet __instance) { if (!__instance.IsServer()) { return; } foreach (VersionCheck versionCheck in versionChecks) { versionCheck.ValidatedClients.Remove(peer.m_rpc); } } [HarmonyPatch(typeof(FejdStartup), "ShowConnectError")] [HarmonyPostfix] private static void ShowConnectionError(FejdStartup __instance) { //IL_000e: Unknown result type (might be due to invalid IL or missing references) //IL_0014: Invalid comparison between Unknown and I4 //IL_018b: Unknown result type (might be due to invalid IL or missing references) //IL_0195: Unknown result type (might be due to invalid IL or missing references) //IL_01dd: Unknown result type (might be due to invalid IL or missing references) //IL_01e9: Unknown result type (might be due to invalid IL or missing references) //IL_01f7: Unknown result type (might be due to invalid IL or missing references) //IL_0209: Unknown result type (might be due to invalid IL or missing references) //IL_0218: Unknown result type (might be due to invalid IL or missing references) //IL_0224: Unknown result type (might be due to invalid IL or missing references) if (!__instance.m_connectionFailedPanel.activeSelf || (int)ZNet.GetConnectionStatus() != 3) { return; } bool flag = false; VersionCheck[] failedClient = GetFailedClient(); if (failedClient.Length != 0) { string text = string.Join("\n", failedClient.Select((VersionCheck check) => check.Error())); TMP_Text connectionFailedError = __instance.m_connectionFailedError; connectionFailedError.text = connectionFailedError.text + "\n" + text; flag = true; } foreach (KeyValuePair<string, string> item in notProcessedNames.OrderBy<KeyValuePair<string, string>, string>((KeyValuePair<string, string> kv) => kv.Key)) { if (!__instance.m_connectionFailedError.text.Contains(item.Key)) { TMP_Text connectionFailedError2 = __instance.m_connectionFailedError; connectionFailedError2.text = connectionFailedError2.text + "\nServer expects you to have " + item.Key + " (Version: " + item.Value + ") installed."; flag = true; } } if (flag) { RectTransform component = ((Component)__instance.m_connectionFailedPanel.transform.Find("Image")).GetComponent<RectTransform>(); component.sizeDelta = new Vector2(675f, component.sizeDelta.y); __instance.m_connectionFailedError.ForceMeshUpdate(false, false); float num = __instance.m_connectionFailedError.renderedHeight + 105f; RectTransform component2 = ((Component)((Component)component).transform.Find("ButtonOk")).GetComponent<RectTransform>(); component2.anchoredPosition = new Vector2(component2.anchoredPosition.x, component2.anchoredPosition.y - (num - component.sizeDelta.y) / 2f); component.sizeDelta = new Vector2(component.sizeDelta.x, num); } } } }