326 lines
12 KiB
C#
326 lines
12 KiB
C#
|
using CsharpHttpHelper;
|
|||
|
|
using PCRobot.Pack;
|
|||
|
|
using PCRobot.PCWechat;
|
|||
|
|
using SuperSocket.ClientEngine;
|
|||
|
|
using System;
|
|||
|
|
using System.Collections.Generic;
|
|||
|
|
using System.Linq;
|
|||
|
|
using System.Net;
|
|||
|
|
using System.Text;
|
|||
|
|
using System.Threading;
|
|||
|
|
using System.Threading.Tasks;
|
|||
|
|
|
|||
|
|
namespace PCRobot
|
|||
|
|
{
|
|||
|
|
public class EasySoc
|
|||
|
|
{
|
|||
|
|
private static EasySoc _socket;
|
|||
|
|
public static EasySoc GetSocket()
|
|||
|
|
{
|
|||
|
|
if (_socket == null) _socket = new EasySoc();
|
|||
|
|
return _socket;
|
|||
|
|
}
|
|||
|
|
public EasySoc()
|
|||
|
|
{
|
|||
|
|
timer = new System.Threading.Timer(new TimerCallback(timerCall), null, 0, 1000 * 5);
|
|||
|
|
}
|
|||
|
|
public bool IsConnected
|
|||
|
|
{
|
|||
|
|
get
|
|||
|
|
{
|
|||
|
|
if (client != null && client.IsConnected) return true;
|
|||
|
|
return false;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
public bool Stop { get; set; }
|
|||
|
|
AsyncTcpSession client;
|
|||
|
|
//EasyClient client;
|
|||
|
|
|
|||
|
|
private readonly static object session_lock = new object();
|
|||
|
|
|
|||
|
|
public void Close(bool isQuit = false)
|
|||
|
|
{
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
lock (session_lock)
|
|||
|
|
{
|
|||
|
|
if (client != null)
|
|||
|
|
{
|
|||
|
|
if (isQuit)
|
|||
|
|
timer.Dispose();
|
|||
|
|
|
|||
|
|
//client.DataReceived -= Client_DataReceived;
|
|||
|
|
client.Connected -= Client_Connected;
|
|||
|
|
client.Closed -= Client_Closed;
|
|||
|
|
client.Error -= Client_Error;
|
|||
|
|
client.Close();
|
|||
|
|
client = null;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
catch (Exception ex)
|
|||
|
|
{
|
|||
|
|
if (WechatClient.IsLog)
|
|||
|
|
LogHelper.GetSingleObj().Info("Close ERROR:", $"{ex.Message}");
|
|||
|
|
|
|||
|
|
LogHelper.GetSingleObj().Error("Close ERROR:", $"{ex.Message} - {ex.StackTrace}");
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
IPEndPoint endPoint;
|
|||
|
|
private Action<string> ReceiveDataAction;
|
|||
|
|
|
|||
|
|
private Action<string> StateAction;
|
|||
|
|
|
|||
|
|
private Action<Exception> SocketErrorAction;
|
|||
|
|
|
|||
|
|
public string key { get; private set; }
|
|||
|
|
private string ip;
|
|||
|
|
private int port;
|
|||
|
|
private System.Threading.Timer timer = null;
|
|||
|
|
private static bool lock_connection = false;
|
|||
|
|
public void Connection(string ip, int port, Action<string> ReceiveDataAction, Action<Exception> SocketErrorAction, string key, Action<string> StateAction)
|
|||
|
|
{
|
|||
|
|
if (Stop) return;
|
|||
|
|
if (lock_connection) return;
|
|||
|
|
if (key.Length != 16) throw new Exception("秘钥必须 == 16位！");
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
lock_connection = true;
|
|||
|
|
if (this.IsConnected) this.Close();
|
|||
|
|
Console.WriteLine("连接中...");
|
|||
|
|
this.ip = ip;
|
|||
|
|
this.port = port;
|
|||
|
|
this.ReceiveDataAction = ReceiveDataAction;
|
|||
|
|
this.StateAction = StateAction;
|
|||
|
|
this.SocketErrorAction = SocketErrorAction;
|
|||
|
|
this.key = key.Substring(0, 16);
|
|||
|
|
this.endPoint = new IPEndPoint(IPAddress.Parse(ip), port);
|
|||
|
|
lock (session_lock)
|
|||
|
|
{
|
|||
|
|
client = new AsyncTcpSession();
|
|||
|
|
client.ReceiveBufferSize = 10 * 1024;
|
|||
|
|
client.DataReceived += Client_DataReceived;
|
|||
|
|
client.Connected += Client_Connected;
|
|||
|
|
client.Closed += Client_Closed;
|
|||
|
|
client.Error += Client_Error;
|
|||
|
|
client.Connect(endPoint);
|
|||
|
|
Thread.Sleep(500);
|
|||
|
|
if (!client.IsConnected)
|
|||
|
|
{
|
|||
|
|
client.Connect(endPoint);
|
|||
|
|
if (!client.IsConnected)
|
|||
|
|
throw new Exception("连接服务器失败,请检查是否填写正确!");
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
catch (Exception ex)
|
|||
|
|
{
|
|||
|
|
if (WechatClient.IsLog)
|
|||
|
|
LogHelper.GetSingleObj().Info("断开", $"连接服务器失败:{ex.Message}");
|
|||
|
|
StateAction?.Invoke($"连接服务器失败：{ex.Message} - {ex.StackTrace}");
|
|||
|
|
}
|
|||
|
|
finally
|
|||
|
|
{
|
|||
|
|
lock_connection = false;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private void Client_Closed(object sender, EventArgs e)
|
|||
|
|
{
|
|||
|
|
if (WechatClient.IsLog)
|
|||
|
|
LogHelper.GetSingleObj().Info("断开", $"与服务器{ip}连接断开！！");
|
|||
|
|
StateAction?.Invoke($"与服务器{ip}连接断开！");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private void Client_Connected(object sender, EventArgs e)
|
|||
|
|
{
|
|||
|
|
if (WechatClient.IsLog)
|
|||
|
|
LogHelper.GetSingleObj().Info("连接成功", $"与服务器{ip}连接成功！");
|
|||
|
|
StateAction?.Invoke($"与服务器{ip}连接成功！");
|
|||
|
|
}
|
|||
|
|
private long i = 0;
|
|||
|
|
private void timerCall(object state)
|
|||
|
|
{
|
|||
|
|
if (string.IsNullOrEmpty(ip) || string.IsNullOrEmpty(key) || port == 0) return;
|
|||
|
|
//throw new NotImplementedException();
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
if (Stop)
|
|||
|
|
{
|
|||
|
|
if (IsConnected) Close();
|
|||
|
|
}
|
|||
|
|
if (!IsConnected)
|
|||
|
|
{
|
|||
|
|
if (Stop) StateAction?.Invoke($"您已断开与服务器{ip}失联！");
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
StateAction?.Invoke($"与服务器{ip}失联，重连中...");
|
|||
|
|
|
|||
|
|
if (WechatClient.IsLog)
|
|||
|
|
LogHelper.GetSingleObj().Info("失联", $"与服务器{ip}失联，重连中...");
|
|||
|
|
Connection(ip, port, ReceiveDataAction, SocketErrorAction, key, StateAction);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
i++;
|
|||
|
|
//Console.WriteLine($"第{i}次测试,key={this.key}");
|
|||
|
|
|
|||
|
|
if (i == 1 || i % 5 == 0)
|
|||
|
|
{
|
|||
|
|
//Send(new BaseMsg()
|
|||
|
|
//{
|
|||
|
|
// Cmd = PCRobotCMD.heartBreak,
|
|||
|
|
// RobotType = RobotType.客户端微信,
|
|||
|
|
// RobotUsername = "测试",
|
|||
|
|
// RobotUsernick = "测试-可以随时删除"
|
|||
|
|
//});
|
|||
|
|
//Send();
|
|||
|
|
//var wxs = WinForm.nms.ToList();
|
|||
|
|
|
|||
|
|
var robots = PCWechat.WechatClient.Users.Values.ToList();
|
|||
|
|
foreach (var item in robots)
|
|||
|
|
{
|
|||
|
|
//var msg = new BaseMsg()
|
|||
|
|
//{
|
|||
|
|
// Cmd = PCRobotCMD.heartBreak,
|
|||
|
|
// RobotType = RobotType.客户端系统,
|
|||
|
|
// RobotUsername = item.Wxid,
|
|||
|
|
// RobotUsernick = item.Nickname
|
|||
|
|
//};
|
|||
|
|
if (item.HeartBreakTime.AddMinutes(6) > DateTime.Now)
|
|||
|
|
{
|
|||
|
|
var msg = new WechatStatus()
|
|||
|
|
{
|
|||
|
|
Cmd = item.Type == WechatType.Xiaoxie_QY ? PCRobotCMD.offline_workWeChat : PCRobotCMD.offline,
|
|||
|
|
RobotUsername = item.Wxid,
|
|||
|
|
RobotUsernick = item.Nickname,
|
|||
|
|
RobotType = item.Type == WechatType.Xiaoxie_QY ? RobotType.客户端企业微信 : RobotType.客户端微信,
|
|||
|
|
Status = Status.在线,
|
|||
|
|
Uin = item.Uin
|
|||
|
|
};
|
|||
|
|
Send(msg);
|
|||
|
|
Thread.Sleep(50);
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
item.HeartBreakTime = DateTime.Now;
|
|||
|
|
Close();
|
|||
|
|
if (!IsConnected) Connection(ip, port, ReceiveDataAction, SocketErrorAction, key, StateAction);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
StateAction?.Invoke($"与服务器: {ip} 连接成功.");
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
catch (Exception ex)
|
|||
|
|
{
|
|||
|
|
if (WechatClient.IsLog)
|
|||
|
|
LogHelper.GetSingleObj().Info("心跳包异常1", $"{ex.Message}");
|
|||
|
|
StateAction?.Invoke($"心跳包异常: {ex.Message}");
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private void Client_Error(object sender, ErrorEventArgs e)
|
|||
|
|
{
|
|||
|
|
if (SocketErrorAction != null)
|
|||
|
|
{
|
|||
|
|
if (WechatClient.IsLog)
|
|||
|
|
LogHelper.GetSingleObj().Info("Client_Error", $"{e.Exception.Message}");
|
|||
|
|
SocketErrorAction(e.Exception);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private void Client_DataReceived(object sender, DataEventArgs e)
|
|||
|
|
{
|
|||
|
|
byte[] new_data = new byte[e.Length];
|
|||
|
|
Buffer.BlockCopy(e.Data, 0, new_data, 0, e.Length);
|
|||
|
|
|
|||
|
|
var data = Encoding.UTF8.GetString(new_data);
|
|||
|
|
var json = HttpHelper.JsonToObject<WechatStatus>(data) as WechatStatus;
|
|||
|
|
if (json == null)
|
|||
|
|
{
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
//var client = WechatClient.Users.Values.ToList().FirstOrDefault(f => f.Wxid == json.RobotUsername && f.Type == (json.RobotType == RobotType.客户端企业微信 ? WechatType.Xiaoxie_QY : WechatType.Xiaoxie));
|
|||
|
|
var client = WechatClient.Users.Values.ToList().FirstOrDefault(f => f.Wxid == json.RobotUsername);
|
|||
|
|
if (client != null)
|
|||
|
|
client.HeartBreakTime = DateTime.Now;
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
Console.WriteLine(Encoding.UTF8.GetString(new_data));
|
|||
|
|
//ReceiveDataAction.BeginInvoke(Encoding.UTF8.GetString(new_data), null, null);
|
|||
|
|
ReceiveDataAction.BeginInvoke(data, null, null);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int retry = 0;
|
|||
|
|
private Thread threadConnection;
|
|||
|
|
public void Send(BaseMsg msg)
|
|||
|
|
{
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
if (IsConnected)
|
|||
|
|
{
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
//秘钥
|
|||
|
|
//key = string.Empty;
|
|||
|
|
msg.Key = this.key;
|
|||
|
|
//Console.WriteLine("心跳测试:" + i + "," + msg.Key);
|
|||
|
|
var json = HttpHelper.ObjectToJson(msg);
|
|||
|
|
var data = PackTool.CompressString(json);
|
|||
|
|
//
|
|||
|
|
//if (!string.IsNullOrEmpty(key))
|
|||
|
|
//{
|
|||
|
|
// AESCryption aes = new AESCryption();
|
|||
|
|
// data = aes.AesEncrypt(data, key);
|
|||
|
|
//}
|
|||
|
|
|
|||
|
|
//data =HttpHelper.URLEncode(data);
|
|||
|
|
var send = msg.Cmd + " " + data + "\r\n";
|
|||
|
|
var data_bytes = Encoding.UTF8.GetBytes(send);
|
|||
|
|
client.Send(data_bytes,0,data_bytes.Length);
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
catch (Exception ex)
|
|||
|
|
{
|
|||
|
|
StateAction?.Invoke($"发送消息失败: {ex.Message}");
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
StateAction?.Invoke($"发送消息失败: 没有与服务器建立连接!");
|
|||
|
|
if (threadConnection == null || !threadConnection.IsAlive)
|
|||
|
|
{
|
|||
|
|
threadConnection = new Thread(new ThreadStart(delegate
|
|||
|
|
{
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
Thread.Sleep(5000);
|
|||
|
|
if (!IsConnected) Connection(ip, port, ReceiveDataAction, SocketErrorAction, key, StateAction);
|
|||
|
|
}
|
|||
|
|
catch (Exception)
|
|||
|
|
{
|
|||
|
|
}
|
|||
|
|
}));
|
|||
|
|
threadConnection.IsBackground = true;
|
|||
|
|
threadConnection.Start();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
catch (Exception ex)
|
|||
|
|
{
|
|||
|
|
if (WechatClient.IsLog)
|
|||
|
|
LogHelper.GetSingleObj().Info("Send ERROR", $"{ex.Message}");
|
|||
|
|
LogHelper.GetSingleObj().Error("Send ERROR:", $"{ex.Message} - {ex.StackTrace}");
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|