C# EasyConnect 断线自动重连的实现 V2.0

507 阅读2分钟

C# EasyConnect 断线自动重连的实现 V1.0 juejin.cn/post/705256…

image.png

image.png

image.png

using System;
using System.Collections.Generic;
using System.Text;
using System.Windows.Forms;
using System.Runtime.InteropServices;
using System.Threading;
using System.Diagnostics;

namespace WinFormsApp
{
    public partial class Form1 : Form
    {

        private static System.Threading.Timer timer;
        private int i = 0;
        private int type = 1;
        private List<IntPtr> ListTitleHandle = new List<IntPtr>();
        private const uint BM_CLICK = 0xF5;
        private const uint WM_SETTEXT = 0x000C;
        private string username;
        private string password;
        public Form1()
        {
            InitializeComponent();
        }
        /// <summary>
        /// 获取窗体的句柄函数
        /// </summary>
        /// <param name="lpClassName">窗口类名</param>
        /// <param name="lpWindowName">窗口标题名</param>
        /// <returns>返回句柄</returns>
        [DllImport("user32.dll", EntryPoint = "FindWindow", SetLastError = true)]
        public static extern IntPtr FindWindow(string lpClassName, string lpWindowName);
        //
        //
        //
        //
        /// <summary>
        /// 发送消息到窗体函数
        /// </summary>
        /// <param name="hwnd">窗体句柄</param>
        /// <param name="wMsg">消息类型</param>
        /// <param name="wParam">附带消息</param>
        /// <param name="lParam"></param>
        /// <returns></returns>
        [DllImport("user32.dll", EntryPoint = "SendMessage", SetLastError = true, CharSet = CharSet.Auto)]
        public static extern IntPtr SendMessage(IntPtr hwnd, uint wMsg, int wParam, int lParam);
        [DllImport("user32.dll", EntryPoint = "SendMessage", SetLastError = true, CharSet = CharSet.Auto)]
        public static extern IntPtr SendMessageString(IntPtr hWnd, uint Msg, int wParam, string lParam);

        //函数指针类型
        delegate bool deleWindowsProc(IntPtr hWnd, string lParam);

        [DllImport("user32.dll")]
        static extern int EnumWindows(deleWindowsProc hWnd, string lParam);

        public delegate bool Callback(IntPtr hwnd, IntPtr lParam);
        [DllImport("user32")]
        [return: MarshalAs(UnmanagedType.Bool)]
        public static extern bool EnumChildWindows(IntPtr parentHandle, Callback callback, IntPtr lParam);

        [DllImport("user32.dll")]
        public static extern int GetWindowText(IntPtr hWnd, StringBuilder lpString, int nMaxCount);

        [DllImport("user32.dll", EntryPoint = "GetClassName")]
        public static extern int GetClassName(IntPtr hWnd, StringBuilder lpString, int nMaxCont);


        [DllImport("user32.dll", EntryPoint = "FindWindowEx", SetLastError = true)]
        private static extern IntPtr FindWindowEx(
            IntPtr hwndParent, uint hwndChildAfter, string lpszClass, string lpszWindow
         );
        private void btn_exit_Click(object sender, EventArgs e)
        {
            System.Environment.Exit(0);
        }
        private void login() {
            var r = EnumWindows(PrintWindow, "EasyConnect");
            if (type == 1)
            {
                Callback callback = new Callback(InitEnumProc);
                foreach (IntPtr i in ListTitleHandle)
                {
                    lock (callback) { 
                        EnumChildWindows(i, callback, IntPtr.Zero);
                    }
                }
            }
            else
            {
                Callback callback = new Callback(EnumProc);
                foreach (IntPtr i in ListTitleHandle)
                {
                    lock (callback)
                    {
                        EnumChildWindows(i, callback, IntPtr.Zero);
                    }
                }
            }
            if (i > 133)
            {
                if (i == 134)
                {
                    i = 0;
                    ListTitleHandle.Clear();
                    Debug.WriteLine("正在初始化");
                    Thread.Sleep(1000);
                    login();
                }
                else
                {
                    type = 2;
                    i = 0;
                    ListTitleHandle.Clear();
                    Debug.WriteLine("初始化完成");
                    Thread.Sleep(1000);
                    login();
                }
            }
            i = 0;
            type = 1;
            ListTitleHandle.Clear();

        }
        
        private void start(object source)
        {
            Debug.WriteLine("开始监听");
            IntPtr window = FindWindow(null, "EasyConnect");
            if (window != IntPtr.Zero)
            {
                Debug.WriteLine("找到:EasyConnect");
                IntPtr buttonObj = FindWindowEx(window, 0, "Button", "重新登录");
                if (buttonObj != IntPtr.Zero)
                {
                    Debug.WriteLine("找到:重新登录");
                    //这里不知道为什么要两次才生效
                    SendMessage(buttonObj, BM_CLICK, 0, 0);
                    SendMessage(buttonObj, BM_CLICK, 0, 0);
                    Thread.Sleep(2000);
                    login();
                }
            }
            Debug.WriteLine("结束监听");
        }

        private void btnCloseText_Click(object sender, EventArgs e)
        {
            username = textBox1.Text;
            password = textBox2.Text;
            if (string.IsNullOrEmpty(username))
            {
                MessageBox.Show("用户名不能为空");
                return;
            }
            if (string.IsNullOrEmpty(password))
            {
                MessageBox.Show("密码不能为空");
                return;
            }
            // 第一个参数是:回调方法,表示要定时执行的方法,
            // 第二个参数是:回调方法要使用的信息的对象,或者为空引用,
            // 第三个参数是:调用 callback 之前延迟的时间量(以毫秒为单位),指定 Timeout.Infinite 以防止计时器开始计时。指定零 (0) 以立即启动计时器。
            // 第四个参数是:定时的时间时隔,以毫秒为单位
            timer = new System.Threading.Timer(start, null, 0, 10000);
        }

        public Boolean InitEnumProc(IntPtr hWnd, IntPtr lParam)
        {
            StringBuilder sb = new StringBuilder(256);
            GetWindowText(hWnd, sb, sb.Capacity);
            if (i == 77 && "正在初始化...".Equals(sb.ToString()))
            {
                return false;
            }
            i++;
            return true;
        }
        public Boolean EnumProc(IntPtr hWnd, IntPtr lParam)
        {

            //由于新手,没办法知道控件的唯一标识,只能用下标了,有没有谁有更好的办法
            if (i == 98)
            {
                SendMessageString(hWnd, WM_SETTEXT, 0, username);
            }
            if (i == 100)
            {
                SendMessageString(hWnd, WM_SETTEXT, 0, password);
            }
            if (i == 126)
            {
                SendMessage(hWnd, BM_CLICK, 0, 0);
                return false;
            } 
            i++;
            return true;
        }


        //回调函数
        bool PrintWindow(IntPtr hWnd, string lParam)
        {
            //分配空间
            var sb = new StringBuilder(50);
            GetWindowText(hWnd, sb, sb.Capacity);
            //注意某些窗口没有标题
            if (sb.ToString() != String.Empty && sb.ToString().Equals(lParam))
            {
                GetClassName(hWnd, sb, sb.Capacity);
                ListTitleHandle.Add(hWnd);
            }
            //返回True继续往下回调,返回False终止
            return true;
        }

    }
}