using DarkControls;
using KsDumper11.Driver;
using Newtonsoft.Json;
using System;
using System.IO;
using System.Collections.Generic;
using System.ComponentModel;
using System.Data;
using System.Diagnostics;
using System.Drawing;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Windows.Forms;
using static System.Windows.Forms.VisualStyles.VisualStyleElement;
using static System.Windows.Forms.VisualStyles.VisualStyleElement.Button;

namespace KsDumper11
{
    public partial class ProviderSelector : Form
    {
        protected override CreateParams CreateParams
        {
            get
            {
                // Activate double buffering at the form level.  All child controls will be double buffered as well.
                CreateParams cp = base.CreateParams;
                cp.ExStyle |= 0x02000000;  // Turn on WS_EX_COMPOSITED
                return cp;
            }
        }

        KduWrapper wrapper;

        JsonSettingsManager settingsManager;
        LabelDrawer labelDrawer;

        public ProviderSelector()
        {
            InitializeComponent();

            settingsManager = new JsonSettingsManager();

            this.FormBorderStyle = FormBorderStyle.None;
            this.Region = Region.FromHrgn(Utils.CreateRoundRectRgn(0, 0, Width, Height, 10, 10));
            this.closeBtn.Region = Region.FromHrgn(Utils.CreateRoundRectRgn(0, 0, closeBtn.Width, closeBtn.Height, 10, 10));

            this.appIcon1.DragForm = this;

            KduSelfExtract.Extract();

            wrapper = new KduWrapper(KduSelfExtract.KduPath);
            wrapper.DriverLoaded += Wrapper_DriverLoaded;
            wrapper.ProvidersLoaded += Wrapper_ProvidersLoaded;
            //wrapper.IsDirtyChanged += Wrapper_IsDirtyChanged;

            wrapper.LoadProviders();
        }

        private void setdriverLoadedLblVisible(bool visible)
        {
            for (int i = 0; i < labelDrawer.labelInfos.Length; i++)
            {
                if (labelDrawer.labelInfos[i].Name == "driverLoadedLbl")
                {
                    labelDrawer.labelInfos[i].Visible = visible;
                }
            }

            this.Invalidate();
        }

        private void Wrapper_IsDirtyChanged(object sender, EventArgs e)
        {
            
        }

        private void Wrapper_ProvidersLoaded(object sender, EventArgs e)
        {
            foreach (KduProvider p in wrapper.providers)
            {
                ListViewItem item = new ListViewItem(p.ProviderIndex.ToString());
                item.SubItems.Add(p.ProviderName);

                if (p.ProviderName.Contains("[NOT WORKING]"))
                {
                    item.ForeColor = Color.Red;
                }

                if (p.ProviderName.Contains("[WORKING]"))
                {
                    item.ForeColor = Color.Green;
                }

                providerList.Items.Add(item);
            }

            if (wrapper.DefaultProvider != -1)
            {
                providerList.SelectedIndices.Add(wrapper.DefaultProvider);
            }
            else
            {
                providerList.SelectedIndices.Add(0);
            }
        }

        private void Wrapper_DriverLoaded(object sender, object[] e)
        {
            if (this.InvokeRequired)
            {
                this.Invoke(new EventHandler<object[]>(Wrapper_DriverLoaded), new object[] { sender, e });
            }
            else
            {
                bool res = (bool)e[0];

                int idx = (int)e[1];

                ListViewItem item = providerList.Items[idx];

                for (int i = 0; i < wrapper.providers.Count; i++)
                {
                    string non_W = "[NOT WORKING] ";
                    string W_ = "[WORKING] ";

                    if (wrapper.providers[i].ProviderName == item.SubItems[1].Text)
                    {
                        if (res)
                        {
                            wrapper.providers[i].ProviderName = W_ + wrapper.providers[i].ProviderName;

                            if (wrapper.IsDirty == false)
                            {
                                wrapper.IsDirty = true;
                            }
                        }
                        else
                        {
                            wrapper.providers[i].ProviderName = non_W + wrapper.providers[i].ProviderName;

                            if (wrapper.IsDirty == false)
                            {
                                wrapper.IsDirty = true;
                            }
                        }

                        
                        break;
                    }
                }
                this.wipeSettingsBtn.Enabled = wrapper.IsDirty;
                if (res)
                {
                    driverLoadedLbl.ForeColor = Color.Green;
                    driverLoadedLbl.Text = "Driver Loaded!";

                    item.SubItems[1].Text = "[WORKING] " + item.SubItems[1].Text;

                    if (providerList.SelectedIndices.Count > 0 && providerList.SelectedIndices[0] == idx)
                    {
                        setDefaultProviderBtn.Enabled = true;
                    }

                    item.ForeColor = Color.Green;
                }
                else
                {
                    driverLoadedLbl.ForeColor = Color.Red;
                    driverLoadedLbl.Text = "Driver failed to load!";

                    item.SubItems[1].Text = "[NOT WORKING] " + item.SubItems[1].Text;

                    if (providerList.SelectedIndices.Count > 0 && providerList.SelectedIndices[0] == idx)
                    {
                        setDefaultProviderBtn.Enabled = false;
                    }

                    item.ForeColor = Color.Red;
                }

                if (settingsManager.JsonSettings.enableAntiAntiDebuggerTools)
                {
                    setdriverLoadedLblVisible(true);
                }
                else
                {
                    driverLoadedLbl.Visible = true;
                }
                driverLoadedLblTimer.Start();
            }
        }

        protected override void WndProc(ref Message m)
        {
            base.WndProc(ref m);
            if (m.Msg == Utils.WM_NCHITTEST)
                m.Result = (IntPtr)(Utils.HT_CAPTION);
        }

        private void providerList_SelectedIndexChanged(object sender, EventArgs e)
        {
            if (providerList.SelectedIndices.Count > 0)
            {
                int idx = providerList.SelectedIndices[0];

                KduProvider p = wrapper.providers[idx];

                if (p.ProviderName.Contains("[NOT WORKING]") || p.ProviderName.Contains("[WORKING]"))
                {
                    testProviderBtn.Enabled = false;

                }
                else
                {
                    testProviderBtn.Enabled = true;
                }

                if (p.ProviderName.Contains("[NOT WORKING]"))
                {
                    setDefaultProviderBtn.Enabled = false;
                }
                else if (p.ProviderName.Contains("[WORKING]"))
                {
                    setDefaultProviderBtn.Enabled = true;
                }
                else
                {
                    setDefaultProviderBtn.Enabled = false;
                }

                driverNameBox.Text = p.DriverName;
                deviceNameBox.Text = p.DeviceName;
                signerNameBox.Text = p.SignerName;
                minWinBuildBox.Text = p.MinWindowsBuild;
                maxWinBuildBox.Text = p.MaxWindowsBuild;
                driverWhqlSignedBox.Checked = p.IsWHQL_Signed;
                shellcodeMaskBox.Text = p.ShellcodeSupportMask;

                defaultProviderIDBox.Text = wrapper.DefaultProvider.ToString();

                if (p.ExtraInfo.Length > 0)
                {
                    providerExtraInfoBox.Clear();
                    for (int i = 0; i < p.ExtraInfo.Length; i++)
                    {
                        providerExtraInfoBox.AppendText(p.ExtraInfo[i] + Environment.NewLine);
                    }
                }
            }
        }

        private void testProviderBtn_Click(object sender, EventArgs e)
        {
            if (providerList.SelectedIndices.Count > 0)
            {
                testProviderBtn.Enabled = false;

                int idx = providerList.SelectedIndices[0];

                KduProvider p = wrapper.providers[idx];

                wrapper.tryLoad(p.ProviderIndex);
            }
        }

        private void driverLoadedLblTimer_Tick(object sender, EventArgs e)
        {
            testProviderBtn.Enabled = true;

            if (settingsManager.JsonSettings.enableAntiAntiDebuggerTools)
            {
                setdriverLoadedLblVisible(false);
            }
            else
            {
                driverLoadedLbl.Visible = false;
            }
                
            driverLoadedLblTimer.Stop();
        }

        private void setDefaultProviderBtn_Click(object sender, EventArgs e)
        {
            if (providerList.SelectedIndices.Count > 0)
            {
                testProviderBtn.Enabled = false;

                int idx = providerList.SelectedIndices[0];

                wrapper.SetDefaultProvider(idx);

                defaultProviderIDBox.Text = wrapper.DefaultProvider.ToString();

                wrapper.Start();

                Program.ProviderIsClosing = true;

                this.Close();
            }
        }

        private void wipeSettingsBtn_Click(object sender, EventArgs e)
        {
            wrapper.ResetProviders();

            providerList.Items.Clear();

            Wrapper_ProvidersLoaded(sender, e);

            this.wipeSettingsBtn.Enabled = false;
        }

        private void ProviderSelector_Load(object sender, EventArgs e)
        {
            this.wipeSettingsBtn.Enabled = wrapper.IsDirty;

            if (settingsManager.JsonSettings.enableAntiAntiDebuggerTools)
            {
                labelDrawer = new LabelDrawer(this);
                setdriverLoadedLblVisible(false);

                SnifferBypass.SelfTitle(this.Handle);

                foreach (Control ctrl in this.Controls)
                {
                    if (ctrl is System.Windows.Forms.TextBox) continue;
                    SnifferBypass.SelfTitle(ctrl.Handle);
                }

                this.Text = SnifferBypass.GenerateRandomString(this.Text.Length);
            }
        }

        private void closeBtn_Click(object sender, EventArgs e)
        {
            DialogResult = DialogResult.OK;
        }
    }
}