using System;

using System.Collections.Generic;

using System.Diagnostics.Contracts;

using ReClassNET.Memory;

using ReClassNET.UI;

using ReClassNET.Util;

namespace ReClassNET.Nodes

{

public abstract class BaseFunctionPtrNode : BaseNode

{

private IntPtr address = IntPtr.Zero;

private readonly List<string> assembledCode = new List<string>();

/// <summary>Size of the node in bytes.</summary>

public override int MemorySize => IntPtr.Size;

public override string GetToolTipText(HotSpot spot, MemoryBuffer memory)

{

var ptr = memory.ReadObject<IntPtr>(Offset);

DisassembleRemoteCode(memory, ptr);

return string.Join("\n", assembledCode);

}

protected int Draw(ViewInfo view, int x, int y, string type, string name)

{

Contract.Requires(view != null);

Contract.Requires(type != null);

Contract.Requires(name != null);

if (IsHidden)

{

return DrawHidden(view, x, y);

}

AddSelection(view, x, y, view.Font.Height);

AddDelete(view, x, y);

AddTypeDrop(view, x, y);

x += TextPadding;

x = AddIcon(view, x, y, Icons.Function, -1, HotSpotType.None);

x = AddAddressOffset(view, x, y);

var tx = x;

x = AddText(view, x, y, Program.Settings.TypeColor, HotSpot.NoneId, type) + view.Font.Width;

x = AddText(view, x, y, Program.Settings.NameColor, HotSpot.NameId, name) + view.Font.Width;

x = AddOpenClose(view, x, y) + view.Font.Width;

x = AddComment(view, x, y);

if (levelsOpen[view.Level])

{

var ptr = view.Memory.ReadObject<IntPtr>(Offset);

DisassembleRemoteCode(view.Memory, ptr);

foreach (var line in assembledCode)

{

y += view.Font.Height;

AddText(view, tx, y, Program.Settings.NameColor, -1, line);

}

}

return y + view.Font.Height;

}

public override int CalculateHeight(ViewInfo view)

{

if (IsHidden)

{

return HiddenHeight;

}

var h = view.Font.Height;

if (levelsOpen[view.Level])

{

h += assembledCode.Count * view.Font.Height;

}

return h;

}

private void DisassembleRemoteCode(MemoryBuffer memory, IntPtr address)

{

Contract.Requires(memory != null);

if (this.address != address)

{

assembledCode.Clear();

this.address = address;

if (!address.IsNull() && memory.Process.IsValid)

{

memory.Process.NativeHelper.DisassembleRemoteCode(

memory.Process.Process.Handle,

address,

200,

#if WIN64

(a, l, i) => assembledCode.Add($"{a.ToString("X08")} {i}")

#else

(a, l, i) => assembledCode.Add($"{a.ToString("X04")} {i}")

#endif

);

}

}

}

}

}