MessageViewer.Loading.cs

#pragma warning disable 

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Windows.Forms;
using System.Drawing;
using System.IO;
using Mooege.Net.GS.Message.Definitions.Player;

namespace GameMessageViewer
{
    public partial class MessageViewer
    {
        private byte[] String_To_Bytes(string strInput)
        {
            // i variable used to hold position in string  
            int i = 0;
            // x variable used to hold byte array element position  
            int x = 0;
            // allocate byte array based on half of string length  
            byte[] bytes = new byte[(strInput.Length) / 2];
            // loop through the string - 2 bytes at a time converting  
            //  it to decimal equivalent and store in byte array  
            while (strInput.Length > i + 1)
            {
                long lngDecimal = Convert.ToInt32(strInput.Substring(i, 2), 16);
                bytes[x] = Convert.ToByte(lngDecimal);
                i = i + 2;
                ++x;
            }
            // return the finished byte array of decimal values  
            return bytes;
        }


        private void LoadWiresharkHex(string text)
        {
            if (text.Contains(" "))
            {
                String[] rows = text.Split('\n');
                String currentBuffer = "";
                text = "";

                for (int i = 0; i < rows.Length; i++)
                {
                    if (i > 0 && (rows[i].StartsWith(" ") ^ rows[i - 1].StartsWith(" ")))
                    {
                        Buffer buffer = new Buffer(String_To_Bytes(currentBuffer));
                        BufferNode newNode = new BufferNode(buffer, actors, questTree, "1");
                        newNode.Start = text.Length;
                        newNode.BackColor = rows[i].StartsWith(" ") ? newNode.BackColor = Color.LightCoral : Color.LightBlue;
                        tree.Nodes.Add(newNode);
                        text += currentBuffer;
                        currentBuffer = "";
                    }

                    currentBuffer += (rows[i].StartsWith(" ") ? rows[i].Substring(14, 3 * 16) : rows[i].Substring(10, 3 * 16)).Trim().Replace(" ", "");
                }
            }

            else
            {
                Buffer buffer = new Buffer(String_To_Bytes(text));
                BufferNode newNode = new BufferNode(buffer, actors, questTree, "1");
                newNode.Parse();
                tree.Nodes.Add(newNode);
            }

            input.Text = text;

            ApplyFilter();
        }


        /// <summary>
        /// Pcap loading extracts the largest session from a pcap, transforms it to dump format (in a memory stream)
        /// which can be read by LoadDump afterwards
        /// </summary>
        /// <param name="path"></param>
        private void LoadPcap(string path)
        {
            try
            {
                // This ugly thing returns a list of MemoryStreams. One for each session in the cap
                var streams = pCapReader.ReconSingleFileSharpPcap(path);

                // Take the largest session (yeah if its no working for you, this may cause it)
                var ordered = streams.OrderBy(stream => stream.Length);
                ordered.Last().Seek(0, SeekOrigin.Begin);
                LoadDump(new StreamReader(ordered.Last()).ReadToEnd());
            }
            catch (SharpPcap.PcapException) {
                MessageBox.Show("The file could not be read. Only lipcap cap files can be loaded. If you want to load a NetMon cap the README tells you how to", "Loading error", MessageBoxButtons.OK, MessageBoxIcon.Error);
            }
        }


        private void LoadPreparsed(string text)
        {
            String[] rows = text.Split('\n');
            String currentBuffer = "";
            text = "";
            string currentDirection = "";
            progressBar.Maximum = rows.Length;
            progressBar.Value = 0;
            progressBar.Visible = true;

            for (int i = 0; i < rows.Length; i++)
            {
                progressBar.Value = i;
                Application.DoEvents();
                if (this.Visible == false)
                    break;

            }

        }


        private void LoadDump(string text)
        {
            String[] rows = text.Split('\n');
            String currentBuffer = "";
            text = "";
            string currentDirection = "";
            progressBar.Maximum = rows.Length;
            progressBar.Value = 0;
            progressBar.Visible = true;
            allNodes = new List<BufferNode>();

            Color[][] trafficColors = new Color[][]
            { new Color[] { Color.LightCoral , Color.LightBlue },
              new Color[] { Color.Tomato, Color.Blue },
              new Color[] { Color.Red, Color.BlueViolet },
              new Color[] { Color.PaleVioletRed, Color.CadetBlue } 
            };


            List<string> clients = new List<string>();
            Dictionary<string, Color[]> colors = new Dictionary<string,Color[]>();
        
            // to read mooege dumps, some leading info must be removed
            // the amount of chars is fixed so its calculated once
            int removeChars = rows[0].IndexOf("Inc:");
            if(removeChars < 0)
                removeChars = rows[0].IndexOf("Out:");
            string clientHash = "";
            for (int i = 0; i < rows.Length; i++)
            {
                if (rows[i].Length > removeChars)
                {
                    // Skip anything til the Inc/Out part (for mooege dumps), note client hash
                    rows[i] = rows[i].Substring(removeChars);
                    clientHash = rows[i].Substring(4, 8);
                    if (clients.Contains(clientHash) == false)
                    {
                        clients.Add(clientHash);
                        colors[clientHash] = trafficColors[clients.Count - 1];
                    }

                    progressBar.Value = i;

                    //causes bugs
                    //Application.DoEvents();
                    //if (this.Visible == false)
                    //    break;

                    if (rows[i].Length > 3)
                    {
                        // Everytime the direction of data changes, the buffer is parsed and emptied
                        // this is for pcap dumps where the data stream is sent in smaller packets
                        // in mooege, data is dumped in whole
                        if (i > 0 && rows[i].Substring(0, 1) != currentDirection)
                        {
                            Buffer buffer = new Buffer(String_To_Bytes(currentBuffer));
                            BufferNode newNode = new BufferNode(buffer, actors, questTree, clientHash);
                            newNode.Start = text.Length;
                            newNode.BackColor = currentDirection == "I" ? colors[clientHash][0] : colors[clientHash][1];
                            allNodes.Add(newNode);
                            //tree.Nodes.Add(newNode);
                            newNode.ApplyFilter(filterWindow.Filter);
                            text += currentBuffer;
                            currentBuffer = "";
                            currentDirection = rows[i].Substring(0, 1);
                        }

                        if (currentDirection == "") currentDirection = rows[i].Substring(0, 1);
                        currentBuffer += (rows[i].Substring(13)).Replace("\r", "");
                    }
                }
            }


            if (currentBuffer.Length > 10)
            {
                Buffer buffer = new Buffer(String_To_Bytes(currentBuffer));
                BufferNode newNode = new BufferNode(buffer, actors, questTree, clientHash);
                newNode.Start = text.Length;
                newNode.BackColor = currentDirection == "I" ? colors[clientHash][0] : colors[clientHash][1];
                allNodes.Add(newNode);
                newNode.ApplyFilter(filterWindow.Filter);
                text += currentBuffer;
            }




            // Create a filter menu entry for every client in the stream.
            filterPlayersToolStripMenuItem.DropDownItems.Clear();
            foreach (string client in clients)
            {
                ToolStripMenuItem m = new ToolStripMenuItem(client);
                m.Tag = client;
                m.Checked = true;
                m.Click += new EventHandler(FilterPlayerClick);

                // find toon name for client hash
                foreach(BufferNode bn in allNodes)
                    if(bn.clientHash.Equals(m.Tag.ToString()))
                        foreach(MessageNode mn in bn.Nodes)
                            if(mn.gameMessage is NewPlayerMessage)
                            {
                                m.Text = (mn.gameMessage as NewPlayerMessage).ToonName;
                                goto hell;
                            }

            hell:
                filterPlayersToolStripMenuItem.DropDownItems.Add(m);
            }


            tree.Nodes.AddRange(allNodes.ToArray());
            input.Text = text;
            progressBar.Visible = false;
        }

        void FilterPlayerClick(object sender, EventArgs e)
        {
            Dictionary<string, bool> filter = new Dictionary<string, bool>();
            ((ToolStripMenuItem)sender).Checked = !((ToolStripMenuItem)sender).Checked;

            foreach (ToolStripMenuItem m in filterPlayersToolStripMenuItem.DropDownItems)
                filter.Add(m.Tag.ToString(), m.Checked);

            tree.Nodes.Clear();
            foreach (BufferNode bn in allNodes)
                if (filter[bn.clientHash])
                    tree.Nodes.Add(bn);

        }
    }
}