310 lines
11 KiB
C#
310 lines
11 KiB
C#
using System;
|
||
using System.Collections.Generic;
|
||
using System.IO;
|
||
using System.IO.Compression;
|
||
using System.Linq;
|
||
using System.Text;
|
||
using System.Threading.Tasks;
|
||
using BarrageGrab.ProtoEntity;
|
||
using ProtoBuf;
|
||
using ColorConsole;
|
||
using BarrageGrab.Proxy;
|
||
using BarrageGrab.Proxy.ProxyEventArgs;
|
||
|
||
namespace BarrageGrab
|
||
{
|
||
/// <summary>
|
||
/// 本机Wss弹幕抓取器
|
||
/// </summary>
|
||
public class WssBarrageGrab : IDisposable
|
||
{
|
||
//ISystemProxy proxy = new FiddlerProxy();
|
||
ISystemProxy proxy = new TitaniumProxy();
|
||
Appsetting appsetting = Appsetting.Current;
|
||
ConsoleWriter console = new ConsoleWriter();
|
||
//解包成功的域名缓存
|
||
List<string> succPackHostNames = new List<string>();
|
||
|
||
//已知的弹幕域名服务器
|
||
string[] wsHostNames = Appsetting.Current.HostNameFilter;
|
||
|
||
/// <summary>
|
||
/// 进入直播间
|
||
/// </summary>
|
||
public event EventHandler<MemberMessage> OnMemberMessage;
|
||
|
||
/// <summary>
|
||
/// 关注
|
||
/// </summary>
|
||
public event EventHandler<SocialMessage> OnSocialMessage;
|
||
|
||
/// <summary>
|
||
/// 聊天
|
||
/// </summary>
|
||
public event EventHandler<ChatMessage> OnChatMessage;
|
||
|
||
/// <summary>
|
||
/// 点赞
|
||
/// </summary>
|
||
public event EventHandler<LikeMessage> OnLikeMessage;
|
||
|
||
/// <summary>
|
||
/// 礼物
|
||
/// </summary>
|
||
public event EventHandler<GiftMessage> OnGiftMessage;
|
||
|
||
/// <summary>
|
||
/// 直播间统计
|
||
/// </summary>
|
||
public event EventHandler<RoomUserSeqMessage> OnRoomUserSeqMessage;
|
||
|
||
/// <summary>
|
||
/// 直播间状态变更
|
||
/// </summary>
|
||
public event EventHandler<ControlMessage> OnControlMessage;
|
||
|
||
/// <summary>
|
||
/// 粉丝团消息
|
||
/// </summary>
|
||
public event EventHandler<FansclubMessage> OnFansclubMessage;
|
||
|
||
public WssBarrageGrab()
|
||
{
|
||
proxy.OnWebSocketData += Proxy_OnWebSocketData;
|
||
proxy.HostNameFilter = HostNameChecker;
|
||
}
|
||
|
||
public void Start()
|
||
{
|
||
proxy.Start();
|
||
}
|
||
|
||
public void Dispose()
|
||
{
|
||
proxy.Dispose();
|
||
}
|
||
|
||
//域名过滤器
|
||
private bool HostNameChecker(string hostName)
|
||
{
|
||
if (!Appsetting.Current.FilterHostName) return true;
|
||
if (hostName.StartsWith("webcast")) return true;
|
||
|
||
if (wsHostNames.Any(a => a.ToLower() == hostName.ToLower())) return true;
|
||
|
||
return false;
|
||
}
|
||
|
||
//gzip解压缩
|
||
private byte[] Decompress(byte[] zippedData)
|
||
{
|
||
MemoryStream ms = new MemoryStream(zippedData);
|
||
GZipStream compressedzipStream = new GZipStream(ms, CompressionMode.Decompress);
|
||
MemoryStream outBuffer = new MemoryStream();
|
||
byte[] block = new byte[1024];
|
||
while (true)
|
||
{
|
||
int bytesRead = compressedzipStream.Read(block, 0, block.Length);
|
||
if (bytesRead <= 0)
|
||
break;
|
||
else
|
||
outBuffer.Write(block, 0, bytesRead);
|
||
}
|
||
compressedzipStream.Close();
|
||
return outBuffer.ToArray();
|
||
}
|
||
|
||
//ws数据处理
|
||
private void Proxy_OnWebSocketData(object sender, WsMessageEventArgs e)
|
||
{
|
||
if (!appsetting.ProcessFilter.Contains(e.ProcessName)) return;
|
||
var buff = e.Payload;
|
||
if (buff.Length == 0) return;
|
||
if (buff[0] != 0x08) return;
|
||
|
||
try
|
||
{
|
||
var enty = Serializer.Deserialize<WssResponse>(new ReadOnlyMemory<byte>(buff));
|
||
if (enty == null) return;
|
||
|
||
//检测包格式
|
||
if (!enty.Headers.Any(a => a.Key == "compress_type" && a.Value == "gzip")) return;
|
||
|
||
//解压gzip
|
||
var odata = enty.Payload;
|
||
var decomp = Decompress(odata);
|
||
|
||
var response = Serializer.Deserialize<Response>(new ReadOnlyMemory<byte>(decomp));
|
||
|
||
if (!succPackHostNames.Contains(e.HostName))
|
||
{
|
||
succPackHostNames.Add(e.HostName);
|
||
SaveHostNameCache();
|
||
}
|
||
response.Messages.ForEach(f => DoMessage(f));
|
||
}
|
||
catch (Exception) { }
|
||
}
|
||
|
||
//用于缓存接收过的消息ID，判断是否重复接收
|
||
Dictionary<string,List<long>> msgDic = new Dictionary<string,List<long>>();
|
||
|
||
//发送事件
|
||
private void DoMessage(Message msg)
|
||
{
|
||
List<long> msgIdList;
|
||
if (msgDic.ContainsKey(msg.Method))
|
||
{
|
||
msgIdList = msgDic[msg.Method];
|
||
}
|
||
else
|
||
{
|
||
msgIdList = new List<long>(320);
|
||
msgDic.Add(msg.Method, msgIdList);
|
||
}
|
||
if (msgIdList.Contains(msg.msgId))
|
||
{
|
||
return;
|
||
}
|
||
|
||
msgIdList.Add(msg.msgId);
|
||
if (msgIdList.Count > 300) msgIdList.RemoveAt(0);
|
||
|
||
try
|
||
{
|
||
switch (msg.Method)
|
||
{
|
||
//来了
|
||
case "WebcastMemberMessage":
|
||
{
|
||
var arg = Serializer.Deserialize<MemberMessage>(new ReadOnlyMemory<byte>(msg.Payload));
|
||
this.OnMemberMessage?.Invoke(this, arg);
|
||
break;
|
||
}
|
||
//关注
|
||
case "WebcastSocialMessage":
|
||
{
|
||
var arg = Serializer.Deserialize<SocialMessage>(new ReadOnlyMemory<byte>(msg.Payload));
|
||
this.OnSocialMessage?.Invoke(this, arg);
|
||
break;
|
||
}
|
||
//消息
|
||
case "WebcastChatMessage":
|
||
{
|
||
var arg = Serializer.Deserialize<ChatMessage>(new ReadOnlyMemory<byte>(msg.Payload));
|
||
this.OnChatMessage?.Invoke(this, arg);
|
||
break;
|
||
}
|
||
//点赞
|
||
case "WebcastLikeMessage":
|
||
{
|
||
var arg = Serializer.Deserialize<LikeMessage>(new ReadOnlyMemory<byte>(msg.Payload));
|
||
this.OnLikeMessage?.Invoke(this, arg);
|
||
break;
|
||
}
|
||
//礼物
|
||
case "WebcastGiftMessage":
|
||
{
|
||
var arg = Serializer.Deserialize<GiftMessage>(new ReadOnlyMemory<byte>(msg.Payload));
|
||
|
||
if (arg.Gift == null)
|
||
{
|
||
if (arg.giftId == 685)
|
||
{
|
||
arg.Gift = new ProtoEntity.GiftStruct()
|
||
{
|
||
Id = arg.giftId,
|
||
Name = "粉丝灯牌",
|
||
diamondCount = 1
|
||
};
|
||
}
|
||
else if (arg.giftId == 3389)
|
||
{
|
||
arg.Gift = new GiftStruct()
|
||
{
|
||
Id = arg.giftId,
|
||
Name = "欢乐盲盒",
|
||
diamondCount = 10
|
||
};
|
||
}
|
||
else if (arg.giftId == 4021)
|
||
{
|
||
arg.Gift = new GiftStruct()
|
||
{
|
||
Id = arg.giftId,
|
||
Name = "欢乐拼图",
|
||
diamondCount = 10
|
||
};
|
||
}
|
||
else
|
||
{
|
||
console.WriteLine("未能识别的礼物ID：" + arg.giftId, ConsoleColor.Red);
|
||
break;
|
||
}
|
||
}
|
||
|
||
this.OnGiftMessage?.Invoke(this, arg);
|
||
break;
|
||
}
|
||
//直播间统计
|
||
case "WebcastRoomUserSeqMessage":
|
||
{
|
||
var arg = Serializer.Deserialize<RoomUserSeqMessage>(new ReadOnlyMemory<byte>(msg.Payload));
|
||
this.OnRoomUserSeqMessage?.Invoke(this, arg);
|
||
break;
|
||
}
|
||
//直播间状态变更
|
||
case "WebcastControlMessage":
|
||
{
|
||
var arg = Serializer.Deserialize<ControlMessage>(new ReadOnlyMemory<byte>(msg.Payload));
|
||
this.OnControlMessage?.Invoke(this, arg);
|
||
break;
|
||
}
|
||
//粉丝团消息
|
||
case "WebcastFansclubMessage":
|
||
{
|
||
var arg = Serializer.Deserialize<FansclubMessage>(new ReadOnlyMemory<byte>(msg.Payload));
|
||
this.OnFansclubMessage?.Invoke(this, arg);
|
||
break;
|
||
}
|
||
default:
|
||
break;
|
||
}
|
||
}
|
||
catch (Exception)
|
||
{
|
||
return;
|
||
}
|
||
}
|
||
|
||
//将成功解包的域名缓存到文件
|
||
private void SaveHostNameCache()
|
||
{
|
||
//获取程序运行目录
|
||
var baseDir = Directory.GetCurrentDirectory();
|
||
var fullPath = Path.Combine(baseDir, "成功解包域名缓存.txt");
|
||
|
||
try
|
||
{
|
||
//文件不存在则创建
|
||
if (!File.Exists(fullPath))
|
||
{
|
||
File.Create(fullPath);
|
||
}
|
||
var text = File.ReadAllText(fullPath);
|
||
var currentHosts = text.Split('\n').Where(w => !string.IsNullOrWhiteSpace(w)).Select(s=>s.Trim().Trim('\r')).ToList();
|
||
var newHosts = succPackHostNames.Except(currentHosts).ToList();
|
||
//保存
|
||
if (newHosts.Count > 0)
|
||
{
|
||
File.AppendAllLines(fullPath, newHosts);
|
||
}
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
Console.WriteLine("写入成功解包域名缓存失败：" + ex.Message, ConsoleColor.Red);
|
||
}
|
||
}
|
||
}
|
||
}
|