using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Data;
using System.Drawing;
using System.Linq;
using System.Text;
using System.Windows.Forms;
using System.Threading;
using System.IO;
namespace HScan
{
    public partial class Form1 : Form
    {

        int _currentThreads = 0;
        int _maxThreads = 100;
        Thread main = null;
        Thread mt = null;
        List<Thread> threads = new List<Thread>();
        public Form1()
        {
            InitializeComponent();
            Control.CheckForIllegalCrossThreadCalls = false;
        }

        private void btnStart_Click(object sender, EventArgs e)
        {
            btnStart.Enabled = false;
            if (txtStart.Text.Trim() == "")
            {
                MessageBox.Show("起始IP不能為空.");
                return;
            }
            if (txtEnd.Text.Trim() == "")
            {
                MessageBox.Show("結(jié)束IP不能為空.");
                return;
            }
            int ts = Convert.ToInt32(txtThreads.Text);
            _maxThreads = ts;
            string startIp = txtStart.Text;
            string endIp = txtEnd.Text;
            TParameter tp=new TParameter();
            tp.StartIp=startIp;
            tp.EndIp=endIp;
            tp.ThreadCount=ts;
            main = new Thread(new ParameterizedThreadStart(StartMe));

            main.Start(tp);

           

        }
        protected void ThreadManage()
        {
            Thread c=null;
            while (true)
            {
                System.Object lockThis = new System.Object();
                lock (lockThis)
                {
                    for (int i = 0; i < threads.Count; i  )
                    {
                        if (threads[i] != null && !threads[i].IsAlive)
                        {
                            c = threads[i];
                            break;
                        }
                    }


                    if (c != null)
                    {
                        threads.Remove(c);
                    }
                }
            }
        }


        protected void StartMe(object ob)
        {

            mt = new Thread(new ThreadStart(ThreadManage));
            mt.Start();

            TParameter p = ob as TParameter;
            string curIp = p.StartIp;

            while (true)
            {

                for (int i = 0; i < _maxThreads; i  )
                {
                    if (curIp != "")
                    {
                        if (_currentThreads >= _maxThreads)
                            break;
                        System.Object lockThis = new System.Object();
                        lock (lockThis)
                        {
                            _currentThreads  ;
                            if (_currentThreads > _maxThreads)
                                _currentThreads = _maxThreads;
                            string tip = curIp;
                            Thread t = new Thread(new ParameterizedThreadStart(Run));
                            t.Start(tip);
                            threads.Add(t);
                            curIp = IPUtility.getLastIp(curIp, p.EndIp, 1);

                        }
                    }
                    else
                    {
                        break;
                    }
                }
            }     
        }
        protected void Run(object ob)
        {
            string ip = ob.ToString();
            SocketGetHead h = new SocketGetHead();
            
            string ret = h.GetHtml(ip, 80);
            if (ret.IndexOf("DVRDVS-Webs") > 0)
            {
                ListViewItem item = new ListViewItem();

                item.SubItems[0].Text = (listView1.Items.Count   1).ToString();


                ListViewItem.ListViewSubItem lvSubItem = new ListViewItem.ListViewSubItem();
                lvSubItem.Text = ip;
                item.SubItems.Add(lvSubItem);


                lvSubItem = new ListViewItem.ListViewSubItem();
                lvSubItem.Text = "DVRDVS-Webs";
                item.SubItems.Add(lvSubItem);

                listView1.Items.Add(item);
            

            }

            System.Object lockThis = new System.Object();
            lock(lockThis)
            {
                lblCurIp.Text = ip;

                _currentThreads--;
                if (_currentThreads < 0)
                    _currentThreads = 0;
            }
            

        }

        private void tsmCopy_Click(object sender, EventArgs e)
        {
            if (listView1.SelectedItems.Count > 0)
            {
                string ip = listView1.SelectedItems[0].SubItems[1].Text;
                Clipboard.SetText(ip);
               
            }
        }

        private void tsmExport_Click(object sender, EventArgs e)
        {
            StreamWriter writer = new StreamWriter(AppDomain.CurrentDomain.BaseDirectory "\\export.txt",true);
            foreach (ListViewItem item in listView1.Items)
            { 
                string ip=item.SubItems[1].Text;
                writer.WriteLine(ip);
                writer.Flush();
            }
            writer.Flush();
            writer.Close();
            MessageBox.Show("導(dǎo)出成功!");
        }

        private void Form1_FormClosing(object sender, FormClosingEventArgs e)
        {
            try
            {
                if (mt != null)
                {
                    mt.Interrupt();
                    mt.Abort();
                }

                foreach (Thread t in threads)
                {
                    t.Interrupt();
                    t.Abort();
                }
                if (main != null)
                {
                    main.Interrupt();
                    main.Abort();
                }
                
            }
            catch { }
            Thread.Sleep(5000);
        }

        private void btnStop_Click(object sender, EventArgs e)
        {
            try
            {
                if (mt != null)
                {
                    mt.Interrupt();
                    mt.Abort();
                }
                foreach (Thread t in threads)
                {
                    t.Interrupt();
                    t.Abort();
                }
                if (main != null)
                {
                    main.Interrupt();
                    main.Abort();
                }
                
            }
            catch { }
            btnStart.Enabled = true;
        }
    }
}