mirror-processhacker/trunk/ProcessHacker/Components/StructViewer.cs

using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Drawing;
using System.Data;
using System.Text;
using System.Windows.Forms;
using Aga.Controls.Tree;
using ProcessHacker.Structs;

namespace ProcessHacker
{
    public partial class StructViewer : UserControl
    {
        private StructModel _model = new StructModel();
        int _pid;
        int _address;
        StructDef _struct;

        public StructViewer(int pid, int address, StructDef struc)
        {
            InitializeComponent();

            _pid = pid;
            _address = address;
            _struct = struc;
            treeStruct.Model = _model;
            treeStruct.ContextMenu = menuStruct;

            GenericViewMenu.AddMenuItems(copyMenuItem.MenuItems, treeStruct);

            try
            {
                FieldValue[] values;

                _struct.Offset = address;
                _struct.IOProvider = new ProcessMemoryIO(pid);
                _struct.Structs = Program.Structs;
                values = _struct.Read();

                _model.Nodes.Add(new StructNode(new FieldValue() 
                { Name = "Struct", FieldType = FieldType.StringUTF16, Value = "" }));

                foreach (FieldValue val in values)
                    this.AddNode(_model.Nodes[0], val);

                treeStruct.Root.Children[0].IsExpanded = true;
            }
            catch (Exception ex)
            {
                MessageBox.Show(ex.Message, "Process Hacker", MessageBoxButtons.OK, MessageBoxIcon.Error);
                this.Error = true;
            }
        }

        private void AddNode(Node node, FieldValue value)
        {
            StructNode newNode = new StructNode(value);

            if (value.Value is FieldValue[])
            {
                foreach (FieldValue val in (FieldValue[])value.Value)
                    AddNode(newNode, val);
            }

            node.Nodes.Add(newNode);
        }

        public bool Error { get; private set; }

        private void menuStruct_Popup(object sender, EventArgs e)
        {
            decMenuItem.Checked = false;
            hexMenuItem.Checked = false;

            if (_model.IntegerDisplayBase == IntegerDisplayBase.Decimal)
                decMenuItem.Checked = true;
            else if (_model.IntegerDisplayBase == IntegerDisplayBase.Hexadecimal)
                hexMenuItem.Checked = true;
        }

        private void decMenuItem_Click(object sender, EventArgs e)
        {
            _model.IntegerDisplayBase = IntegerDisplayBase.Decimal;
            _model.OnStructureChanged(new TreePathEventArgs(new TreePath()));
        }

        private void hexMenuItem_Click(object sender, EventArgs e)
        {
            _model.IntegerDisplayBase = IntegerDisplayBase.Hexadecimal;
            _model.OnStructureChanged(new TreePathEventArgs(new TreePath()));
        }
    }

    public enum IntegerDisplayBase
    {
        Decimal,
        Hexadecimal
    }

    public class StructModel : TreeModel
    {
        public IntegerDisplayBase IntegerDisplayBase { get; set; }
    }

    public class StructNode : Node
    {
        private FieldValue _value;

        public StructNode(FieldValue value)
        {
            _value = value;
        }

        public string Name
        {
            get { return _value.Name; }
        }

        public string Value
        {
            get
            {
                FieldType type = _value.FieldType & (~FieldType.Array) & (~FieldType.Pointer);

                if ((_value.FieldType & FieldType.Array) != 0)
                {
                    int memberCount = ((FieldValue[])_value.Value).Length;

                    if (_value.StructName != null)
                        return _value.StructName + "[" + memberCount.ToString() +  "]";
                    else
                        return type.ToString() + "[" + memberCount.ToString() + "]";
                }

                if (_value.StructName != null)
                    return _value.StructName;

                if (_value.Value == null)
                    return "null";

                IntegerDisplayBase b = (this.FindModel() as StructModel).IntegerDisplayBase;
                string formatStr = "{0:d}";

                switch (b)
                {
                    case IntegerDisplayBase.Decimal:
                        formatStr = "{0:d}";
                        break;
                    case IntegerDisplayBase.Hexadecimal:
                        formatStr = "0x{0:x}";
                        break;
                }

                switch (type)
                {
                    case FieldType.Bool32:
                    case FieldType.Bool8:
                        return ((bool)_value.Value) ? "True" : "False";
                    case FieldType.CharASCII:
                    case FieldType.CharUTF16:
                        return ((char)_value.Value).ToString();
                    case FieldType.Int16:
                    case FieldType.Int32:
                    case FieldType.Int64:
                    case FieldType.Int8:
                    case FieldType.UInt16:
                    case FieldType.UInt32:
                    case FieldType.UInt64:
                    case FieldType.UInt8:
                        return string.Format(formatStr, long.Parse(_value.Value.ToString()));
                    case FieldType.PVoid:
                        return "0x" + int.Parse(_value.Value.ToString()).ToString("x8");
                    case FieldType.StringASCII:
                    case FieldType.StringUTF16:
                        return _value.Value.ToString();
                    default:
                        return _value.Value.ToString();
                }
            }
        }
    }
}