1
0
mirror of https://github.com/soarqin/DSP_Mods.git synced 2026-03-22 11:53:25 +08:00

Fix potential bugs in UXAssist and CheatEnabler

UXAssist:
- PlanetFunctions: fix infinite loop in constructStats cleanup (i++ -> i--)
- PlanetFunctions: skip entityPool slot 0 (sentinel) in DismantleAll foreach
- PlanetFunctions: fix belt buffer walk infinite loop when buffer[j]==250,
  add cargoPool bounds check
- FactoryPatch: fix UnfixProto guard condition (< 3 -> < PowerPoleIds.Length)
- FactoryPatch: fix Array.Resize(index*2) -> (index+1)*2 to handle index==0
- LogisticsPatch: fix UpdateStorageMax early-exit to check both local and remote
- LogisticsPatch: fix storage max scan to use Math.Max instead of last-write
- LogisticsPatch: add divide-by-zero guard in station storage ratio calculation
- LogisticsPatch: fix station entry right-click delegates using per-instance
  dictionary instead of shared static array to prevent unsubscribe mismatch
- LogisticsPatch: add null checks for GameObject.Find results in InitGUI
- GamePatch: log exception in previously empty catch block
- GameLogic: invoke event handlers individually with try/catch so one failing
  subscriber does not abort subsequent ones
- DysonSpherePatch/LogisticsPatch/PersistPatch/PlayerPatch: replace
  matcher.Labels = null with matcher.Labels = [] (5 sites)
- UIConfigWindow: remove duplicate I18N.Add for 'Outgoing integration count'
- UIConfigWindow: remove two orphaned y += 36f spacing increments

CheatEnabler:
- GamePatch: add null check for GameMain.gameScenario in AbnormalDisabler.OnEnable
- FactoryPatch: add null check for GameMain.mainPlayer in TakeTailItemsPatch
  and GetItemCountPatch
- FactoryPatch: add null check for GameMain.mainPlayer in
  ConstructionSystem_GameTick_Prefix
- FactoryPatch: fix _portalFrom.Remove(beltId) -> Remove(v) (wrong key)
- FactoryPatch: add null guard for GameMain.data before iterating factories
  in WindTurbinesPowerGlobalCoverage
- DysonSphereFunctions: fix shell pool rebuild loop writing all shells to
  slot 1 (id=1); now uses j+1 per iteration
- DysonSphereFunctions: replace GameMain.gameScenario.NotifyOnPlanDysonShell()
  with null-conditional call (?.) at 4 sites
- DysonSpherePatch: fix SkipAbsorbPatch/QuickAbsorbPatch sharing _instantAbsorb:
  OnDisable now recalculates the flag instead of unconditionally clearing it
- PlayerFunctions: add null check for GameMain.galaxy in TeleportToOuterSpace
- UIConfigWindow: add null guard in UpdateButtons before accessing button fields
- PlanetFunctions: add colliderId bounds check in BuryAllVeins
This commit is contained in:
2026-03-16 19:51:16 +08:00
parent e91271e137
commit f69a90d452
16 changed files with 116 additions and 62 deletions

View File

@@ -414,7 +414,7 @@ public class DysonSpherePatch : PatchImpl<DysonSpherePatch>
);
if (jmpTarget == null) return matcher.InstructionEnumeration();
var labels = matcher.Labels;
matcher.Labels = null;
matcher.Labels = [];
matcher.Insert(
new CodeInstruction(OpCodes.Ldarg_0).WithLabels(labels),
new CodeInstruction(OpCodes.Ldfld, AccessTools.Field(typeof(EjectorComponent), nameof(EjectorComponent.runtimeOrbitId))),

View File

@@ -1542,7 +1542,7 @@ public class FactoryPatch : PatchImpl<FactoryPatch>
private static void UnfixProto()
{
if (GetHarmony() == null || OldDragBuild.Count < 3 || DSPGame.IsMenuDemo) return;
if (GetHarmony() == null || OldDragBuild.Count < PowerPoleIds.Length || DSPGame.IsMenuDemo) return;
var i = 0;
foreach (var id in PowerPoleIds)
{
@@ -1860,7 +1860,7 @@ public class FactoryPatch : PatchImpl<FactoryPatch>
if (index < 0) return null;
if (index >= _signalBelts.Length)
{
Array.Resize(ref _signalBelts, index * 2);
Array.Resize(ref _signalBelts, (index + 1) * 2);
}
else
{
@@ -2540,14 +2540,13 @@ public class FactoryPatch : PatchImpl<FactoryPatch>
{
if (buffer[i] >= 246)
{
i += 250 - buffer[i];
var delta = 250 - buffer[i];
if (delta > 0) i += delta;
var index = buffer[i + 1] - 1 + (buffer[i + 2] - 1) * 100 + (buffer[i + 3] - 1) * 10000 + (buffer[i + 4] - 1) * 1000000;
ref var cargo = ref cargoPath.cargoContainer.cargoPool[index];
var item = cargo.item;
var stack = cargo.stack;
var inc = cargo.inc;
takeOutItems[item] = (takeOutItems.TryGetValue(item, out var value) ? value : 0)
+ ((long)stack | ((long)inc << 32));
+ ((long)cargo.stack | ((long)cargo.inc << 32));
Array.Clear(buffer, i - 4, 10);
i += 6;
if (cargoPath.updateLen < i) cargoPath.updateLen = i;

View File

@@ -521,8 +521,9 @@ public class GamePatch : PatchImpl<GamePatch>
__instance.ImportXML(optionPath);
return false;
}
catch
catch (Exception ex)
{
Debug.LogException(ex);
}
}
var gameXMLOptionPath = GameConfig.gameXMLOptionPath;

View File

@@ -481,6 +481,7 @@ public static class LogisticsPatch
var modelIndex = factory.entityPool[stationPool[i].entityId].modelIndex;
var maxCount = LDB.models.Select(modelIndex).prefabDesc.stationMaxItemCount;
var oldMaxCount = maxCount + history.localStationExtraStorage - _valuelf;
if (oldMaxCount < 1.0) continue;
var intOldMaxCount = (int)Math.Round(oldMaxCount);
var intNewMaxCount = maxCount + history.localStationExtraStorage;
var ratio = intNewMaxCount / oldMaxCount;
@@ -506,6 +507,7 @@ public static class LogisticsPatch
var modelIndex = factory.entityPool[stationPool[i].entityId].modelIndex;
var maxCount = LDB.models.Select(modelIndex).prefabDesc.stationMaxItemCount;
var oldMaxCount = maxCount + history.remoteStationExtraStorage - _valuelf;
if (oldMaxCount < 1.0) continue;
var intOldMaxCount = (int)Math.Round(oldMaxCount);
var intNewMaxCount = maxCount + history.remoteStationExtraStorage;
var ratio = intNewMaxCount / oldMaxCount;
@@ -681,7 +683,7 @@ public static class LogisticsPatch
}
var labels = matcher.Labels;
matcher.Labels = null;
matcher.Labels = [];
matcher.Insert(
new CodeInstruction(OpCodes.Ldarg_0).WithLabels(labels),
Transpilers.EmitDelegate((UIGame uiGame) =>
@@ -710,37 +712,37 @@ public static class LogisticsPatch
controlPanelWindow.DetermineFilterResults();
}
private static readonly Action<int>[] OnStationEntryItemIconRightClickActions = new Action<int>[5];
private static readonly Dictionary<UIControlPanelStationEntry, Action<int>[]> OnStationEntryItemIconRightClickActionsMap = new();
[HarmonyPostfix]
[HarmonyPatch(typeof(UIControlPanelStationEntry), nameof(UIControlPanelStationEntry._OnRegEvent))]
private static void UIControlPanelStationEntry__OnRegEvent_Postfix(UIControlPanelStationEntry __instance)
{
OnStationEntryItemIconRightClickActions[0] = _ => OnStationEntryItemIconRightClick(__instance, 0);
OnStationEntryItemIconRightClickActions[1] = _ => OnStationEntryItemIconRightClick(__instance, 1);
OnStationEntryItemIconRightClickActions[2] = _ => OnStationEntryItemIconRightClick(__instance, 2);
OnStationEntryItemIconRightClickActions[3] = _ => OnStationEntryItemIconRightClick(__instance, 3);
OnStationEntryItemIconRightClickActions[4] = _ => OnStationEntryItemIconRightClick(__instance, 4);
__instance.storageItem0.itemButton.onRightClick += OnStationEntryItemIconRightClickActions[0];
__instance.storageItem1.itemButton.onRightClick += OnStationEntryItemIconRightClickActions[1];
__instance.storageItem2.itemButton.onRightClick += OnStationEntryItemIconRightClickActions[2];
__instance.storageItem3.itemButton.onRightClick += OnStationEntryItemIconRightClickActions[3];
__instance.storageItem4.itemButton.onRightClick += OnStationEntryItemIconRightClickActions[4];
var actions = new Action<int>[5];
actions[0] = _ => OnStationEntryItemIconRightClick(__instance, 0);
actions[1] = _ => OnStationEntryItemIconRightClick(__instance, 1);
actions[2] = _ => OnStationEntryItemIconRightClick(__instance, 2);
actions[3] = _ => OnStationEntryItemIconRightClick(__instance, 3);
actions[4] = _ => OnStationEntryItemIconRightClick(__instance, 4);
OnStationEntryItemIconRightClickActionsMap[__instance] = actions;
__instance.storageItem0.itemButton.onRightClick += actions[0];
__instance.storageItem1.itemButton.onRightClick += actions[1];
__instance.storageItem2.itemButton.onRightClick += actions[2];
__instance.storageItem3.itemButton.onRightClick += actions[3];
__instance.storageItem4.itemButton.onRightClick += actions[4];
}
[HarmonyPostfix]
[HarmonyPatch(typeof(UIControlPanelStationEntry), nameof(UIControlPanelStationEntry._OnUnregEvent))]
private static void UIControlPanelStationEntry__OnUnregEvent_Postfix(UIControlPanelStationEntry __instance)
{
__instance.storageItem0.itemButton.onRightClick -= OnStationEntryItemIconRightClickActions[0];
__instance.storageItem1.itemButton.onRightClick -= OnStationEntryItemIconRightClickActions[1];
__instance.storageItem2.itemButton.onRightClick -= OnStationEntryItemIconRightClickActions[2];
__instance.storageItem3.itemButton.onRightClick -= OnStationEntryItemIconRightClickActions[3];
__instance.storageItem4.itemButton.onRightClick -= OnStationEntryItemIconRightClickActions[4];
for (var i = 0; i < 5; i++)
{
OnStationEntryItemIconRightClickActions[i] = null;
}
if (!OnStationEntryItemIconRightClickActionsMap.TryGetValue(__instance, out var actions)) return;
__instance.storageItem0.itemButton.onRightClick -= actions[0];
__instance.storageItem1.itemButton.onRightClick -= actions[1];
__instance.storageItem2.itemButton.onRightClick -= actions[2];
__instance.storageItem3.itemButton.onRightClick -= actions[3];
__instance.storageItem4.itemButton.onRightClick -= actions[4];
OnStationEntryItemIconRightClickActionsMap.Remove(__instance);
}
}
@@ -777,7 +779,7 @@ public static class LogisticsPatch
{
var history = GameMain.history;
if (history == null) return false;
if (_remoteStorageExtra == history.remoteStationExtraStorage) return false;
if (_remoteStorageExtra == history.remoteStationExtraStorage && _localStorageExtra == history.localStationExtraStorage) return false;
_localStorageExtra = history.localStationExtraStorage;
_remoteStorageExtra = history.remoteStationExtraStorage;
_localStorageMaxTotal = _localStorageMax + _localStorageExtra;
@@ -846,14 +848,14 @@ public static class LogisticsPatch
{
if (!prefabDesc.isCollectStation)
{
_remoteStorageMax = prefabDesc.stationMaxItemCount;
_remoteStorageMax = Math.Max(_remoteStorageMax, prefabDesc.stationMaxItemCount);
}
}
else
{
if (!prefabDesc.isVeinCollector)
{
_localStorageMax = prefabDesc.stationMaxItemCount;
_localStorageMax = Math.Max(_localStorageMax, prefabDesc.stationMaxItemCount);
}
}
}
@@ -880,12 +882,36 @@ public static class LogisticsPatch
rtrans.anchoredPosition3D = new Vector3(0, 0, 0f);
var sliderBgPrefab = GameObject.Find("UI Root/Overlay Canvas/In Game/Windows/Station Window/storage-box-0/slider-bg");
if (sliderBgPrefab == null)
{
UXAssist.Logger.LogWarning("RealtimeLogisticsInfoPanel.InitGUI: slider-bg prefab not found, aborting GUI init");
Object.Destroy(_stationTipsRoot);
_stationTipsRoot = null;
return;
}
_tipPrefab = Object.Instantiate(GameObject.Find("UI Root/Overlay Canvas/In Game/Scene UIs/Vein Marks/vein-tips/vein-tip-prefab"), _stationTipsRoot.transform);
var veinTipPrefabGo = GameObject.Find("UI Root/Overlay Canvas/In Game/Scene UIs/Vein Marks/vein-tips/vein-tip-prefab");
if (veinTipPrefabGo == null)
{
UXAssist.Logger.LogWarning("RealtimeLogisticsInfoPanel.InitGUI: vein-tip-prefab not found, aborting GUI init");
Object.Destroy(_stationTipsRoot);
_stationTipsRoot = null;
return;
}
var keyTipPrefabGo = GameObject.Find("UI Root/Overlay Canvas/In Game/Windows/Key Tips/tip-prefab");
if (keyTipPrefabGo == null)
{
UXAssist.Logger.LogWarning("RealtimeLogisticsInfoPanel.InitGUI: key tip-prefab not found, aborting GUI init");
Object.Destroy(_stationTipsRoot);
_stationTipsRoot = null;
return;
}
_tipPrefab = Object.Instantiate(veinTipPrefabGo, _stationTipsRoot.transform);
_tipPrefab.name = "tipPrefab";
Object.Destroy(_tipPrefab.GetComponent<UIVeinDetailNode>());
var image = _tipPrefab.GetComponent<Image>();
image.sprite = GameObject.Find("UI Root/Overlay Canvas/In Game/Windows/Key Tips/tip-prefab").GetComponent<Image>().sprite;
image.sprite = keyTipPrefabGo.GetComponent<Image>().sprite;
image.color = new Color(0, 0, 0, 0.8f);
image.enabled = true;
var rectTrans = (RectTransform)_tipPrefab.transform;

View File

@@ -53,7 +53,7 @@ public class PersistPatch : PatchImpl<PersistPatch>
);
var ldLocOpr = matcher.Operand;
var labels = matcher.Labels;
matcher.Labels = null;
matcher.Labels = [];
matcher.Insert(
new CodeInstruction(OpCodes.Ldloc_S, ldLocOpr).WithLabels(labels),
new CodeInstruction(OpCodes.Callvirt, AccessTools.PropertyGetter(typeof(Component), nameof(Component.transform))),

View File

@@ -266,7 +266,7 @@ public class PlayerPatch : PatchImpl<PlayerPatch>
);
matcher.CreateLabelAt(matcher.Pos + 8, out var jumpPos);
var labels = matcher.Labels;
matcher.Labels = null;
matcher.Labels = [];
matcher.InsertAndAdvance(
new CodeInstruction(OpCodes.Ldsfld, AccessTools.Field(typeof(ShortcutKeysForStarsName), nameof(_showAllStarsNameStatus))).WithLabels(labels),
new CodeInstruction(OpCodes.Ldc_I4_1),
@@ -316,7 +316,7 @@ public class PlayerPatch : PatchImpl<PlayerPatch>
);
matcher.CreateLabelAt(matcher.Pos + 10, out var jumpPos);
var labels = matcher.Labels;
matcher.Labels = null;
matcher.Labels = [];
matcher.InsertAndAdvance(
new CodeInstruction(OpCodes.Ldsfld, AccessTools.Field(typeof(ShortcutKeysForStarsName), nameof(_showAllStarsNameStatus))).WithLabels(labels),
new CodeInstruction(OpCodes.Ldc_I4_1),