Cosmos/source/Cosmos.VS.Windows/ToolWindows/StackUC.xaml.cs

using System;
using System.Collections;
using System.Collections.Generic;
using System.ComponentModel;
using System.ComponentModel.Design;
using System.Data;
using System.Linq;
using System.Runtime.InteropServices;
using System.Text;
using System.Threading;
using System.Windows;
using System.Windows.Controls;
using System.Windows.Data;
using System.Windows.Documents;
using System.Windows.Input;
using System.Windows.Media;
using System.Windows.Media.Imaging;
using System.Windows.Navigation;
using System.Windows.Shapes;
using System.Windows.Threading;
using Cosmos.Debug.Common;
using Microsoft.VisualStudio.Shell;
using Microsoft.VisualStudio.Shell.Interop;

namespace Cosmos.VS.Windows
{
    [Guid("A64D0FCC-8DCC-439A-9B16-3C43128AAD51")]
    public class StackTW : ToolWindowPane2
    {
        public StackTW()
        {
            Caption = "Cosmos Stack";
            BitmapResourceID = 301;
            BitmapIndex = 1;

            mUserControl = new StackUC();
            Content = mUserControl;
        }
    }

    public partial class StackUC : DebuggerUC
    {
        protected byte[] stackData = new byte[0];


        public StackUC()
        {
            InitializeComponent();
        }


        public override void Update(string aTag, byte[] aData)
        {
            if (aTag == "FRAME")
            {
                mData = aData;
            }
            else
            {
                stackData = aData;
            }
            DoUpdate(aTag);
        }

        protected override void DoUpdate(string aTag)
        {
            if (aTag == "STACK")
            {
                UpdateStack(stackData);
            }
            else if (aTag == "FRAME")
            {
                UpdateFrame(mData);
            }
        }

        public void UpdateFrame(byte[] aData)
        {
            System.Windows.Application.Current.Dispatcher.Invoke(DispatcherPriority.Normal,
                (Action)delegate()
                {
                    if (aData == null)
                    {
                        memvEBP.Clear();
                    }
                    else
                    {
                        try
                        {
                            var xValues = MemoryViewUC.Split(aData);
                            int xCount = xValues.Count;
                            memvEBP.Clear();
                            for (int i = 0; i < xCount; i++)
                            {
                                // We start at EBP + 8, because lower is not transmitted
                                // [EBP] is old EBP - not needed
                                // [EBP + 4] is saved EIP - not needed
                                memvEBP.Add("[EBP + " + (i * 4 + 8) + "]", xValues[i]);
                            }
                        }
                        catch
                        {
                            memvEBP.Clear();
                        }
                    }
                }
            );
        }

        public void UpdateStack(byte[] aData)
        {
            System.Windows.Application.Current.Dispatcher.Invoke(DispatcherPriority.Normal,
                (Action)delegate()
                {
                    if (aData == null)
                    {
                        memvESP.Clear();
                    }
                    else
                    {
                        try
                        {
                            var xValues = MemoryViewUC.Split(aData);
                            int xCount = xValues.Count;
                            memvESP.Clear();
                            for (int i = 0; i < xCount; i++)
                            {
                                memvESP.Add(("[EBP - " + ((xCount - i) * 4) + "]").PadRight(10) + " [ESP + " + (i * 4) + "]", xValues[i]);
                            }
                        }
                        catch
                        {
                            memvESP.Clear();
                        }
                    }
                }
            );
        }

        public override byte[] GetCurrentState()
        {
            byte[] aFrameData = mData == null ? new byte[0] : mData;
            byte[] aStackData = stackData == null ? new byte[0] : stackData;
            return BitConverter.GetBytes(aFrameData.Length).Concat(aFrameData.Concat(aStackData)).ToArray();
        }
        public override void SetCurrentState(byte[] aData)
        {
            if (aData == null)
            {
                Update("FRAME", null);
                Update("STACK", null);
            }
            else
            {
                int mDataLength = BitConverter.ToInt32(aData, 0);
                byte[] aFrameData = new byte[mDataLength];
                byte[] aStackData = new byte[aData.Length - mDataLength - 4];
                Array.Copy(aData, 4, aFrameData, 0, aFrameData.Length);
                Array.Copy(aData, 4 + mDataLength, aStackData, 0, aStackData.Length);
                Update("FRAME", aFrameData);
                Update("STACK", aStackData);
            }
        }
    }
}