409 lines
15 KiB
C#
409 lines
15 KiB
C#
using AxibugProtobuf;
|
||
using ClientCore.Common;
|
||
using ClientCore.Event;
|
||
using ClientCore.Network;
|
||
using Google.Protobuf;
|
||
using System;
|
||
using System.IO;
|
||
using System.Net;
|
||
using System.Text;
|
||
using System.Threading.Tasks;
|
||
|
||
namespace ClientCore.Manager
|
||
{
|
||
public class P2PFile
|
||
{
|
||
public int cPackSize = 5 * 1024;
|
||
public int WaitSendPackCount = 100;
|
||
//是否自动同意接收
|
||
public bool bAutoRecv = true;
|
||
public P2PFile()
|
||
{
|
||
//发送者协议注册
|
||
NetMsg.Instance.RegNetMsgEvent((int)CommandID.CmdTcpP2PFilepushConfirmResp, RecvTcpP2PFilepushConfirmResp);
|
||
NetMsg.Instance.RegNetMsgEvent((int)CommandID.CmdTcpP2PFilepushResp, RecvTcpP2PFilepushResp);
|
||
//接收者协议注册
|
||
NetMsg.Instance.RegNetMsgEvent((int)CommandID.CmdTcpP2PFilepushConfirm, RecvTcpP2PFilepushConfirm);
|
||
NetMsg.Instance.RegNetMsgEvent((int)CommandID.CmdTcpP2PFilepush, RecvTcpP2PFilepush);
|
||
}
|
||
|
||
|
||
#region 发送者网络消息
|
||
public void SendFilePushConfirm(long UID, FilePushTask taskinfo)
|
||
{
|
||
Protobuf_TcpP2P_FilePushConfirm msg = new Protobuf_TcpP2P_FilePushConfirm()
|
||
{
|
||
TaskID = taskinfo.TaskID,
|
||
PackSize = taskinfo.PackSize,
|
||
ContentType = taskinfo.ContentType,
|
||
FileHash = taskinfo.FileHash,
|
||
FileLenght = taskinfo.FileLenght,
|
||
FileName = taskinfo.FileName,
|
||
PackCount = taskinfo.PackCount,
|
||
};
|
||
App.clientMgr.SendToTargetSocket(UID, (int)CommandID.CmdTcpP2PFilepushConfirm, (int)ErrorCode.ErrorOk, ProtoBufHelper.Serizlize(msg));
|
||
}
|
||
public void SendFilePush(long UID,int TaskID, long Index, byte[] data, int dataLength)
|
||
{
|
||
Protobuf_TcpP2P_FilePush msg = new Protobuf_TcpP2P_FilePush()
|
||
{
|
||
TaskID = TaskID,
|
||
PackIndex = Index,
|
||
PackLenght= dataLength,
|
||
FileData = ByteString.CopyFrom(data)
|
||
};
|
||
App.clientMgr.SendToTargetSocket(UID, (int)CommandID.CmdTcpP2PFilepush, (int)ErrorCode.ErrorOk, ProtoBufHelper.Serizlize(msg));
|
||
}
|
||
void RecvTcpP2PFilepushConfirmResp(long uid, byte[] reqData)
|
||
{
|
||
Protobuf_TcpP2P_FilePushConfirm_RESP msg = ProtoBufHelper.DeSerizlize<Protobuf_TcpP2P_FilePushConfirm_RESP>(reqData);
|
||
// 开始发送
|
||
if (msg.Agree == 1)
|
||
{
|
||
sender_AgreeRecvFile(msg.TaskID);
|
||
}
|
||
else
|
||
{
|
||
//TODO 被拒绝
|
||
}
|
||
}
|
||
void RecvTcpP2PFilepushResp(long uid, byte[] reqData)
|
||
{
|
||
//收到回执信息
|
||
Protobuf_TcpP2P_FilePush_RESP msg = ProtoBufHelper.DeSerizlize<Protobuf_TcpP2P_FilePush_RESP>(reqData);
|
||
sender_StartPushFileThread(msg.TaskID, msg.FinishIndex);
|
||
}
|
||
#endregion
|
||
|
||
#region 发送者逻辑
|
||
Dictionary<int, FilePushTask> DictTaskID2Task_SendReady = new Dictionary<int, FilePushTask>();
|
||
Dictionary<int, FilePushTask> DictTaskID2Task_SendDo = new Dictionary<int, FilePushTask>();
|
||
|
||
/// <summary>
|
||
/// 发送者发送文件请求
|
||
/// </summary>
|
||
/// <param name="UID"></param>
|
||
/// <param name="path"></param>
|
||
public void sender_FilePushConfirmToTarget(long UID, string path)
|
||
{
|
||
FileInfo finfo = new FileInfo(path);
|
||
//向上取整
|
||
long packCount = (long)Math.Ceiling(finfo.Length / (float)cPackSize);
|
||
FilePushTask ftask = new FilePushTask()
|
||
{
|
||
TaskID = 1,//TODO
|
||
fromUID = App.userMgr.MainPlayer.UID,
|
||
targetUID = UID,
|
||
Path = path,
|
||
FileName = finfo.Name,
|
||
PackSize = cPackSize,
|
||
PackCount = packCount,
|
||
FileLenght = finfo.Length,
|
||
FileHash = "",//TODO
|
||
ContentType = "",//TODO
|
||
|
||
CurrIndex = -1,
|
||
ConfirmIndex= -1,
|
||
};
|
||
SendFilePushConfirm(UID, ftask);
|
||
|
||
DictTaskID2Task_SendReady[ftask.TaskID] = ftask;
|
||
}
|
||
public void sender_AgreeRecvFile(int TaskID)
|
||
{
|
||
if (DictTaskID2Task_SendReady.ContainsKey(TaskID))
|
||
{
|
||
FilePushTask task = DictTaskID2Task_SendReady[TaskID];
|
||
DictTaskID2Task_SendDo[TaskID] = task;
|
||
DictTaskID2Task_SendReady.Remove(TaskID);
|
||
sender_StartPushFile(TaskID);
|
||
}
|
||
}
|
||
|
||
//开始发送文件传输
|
||
void sender_StartPushFile(int TaskID)
|
||
{
|
||
if (!DictTaskID2Task_SendDo.ContainsKey(TaskID))
|
||
{
|
||
return;
|
||
}
|
||
|
||
FilePushTask task = DictTaskID2Task_SendDo[TaskID];
|
||
Thread thread = new Thread(sender_StartPushFileThread);
|
||
thread.IsBackground = true;
|
||
thread.Start(TaskID);
|
||
}
|
||
|
||
void sender_StartPushFileThread(int TaskID,long Index)
|
||
{
|
||
if (!DictTaskID2Task_SendDo.ContainsKey(TaskID))
|
||
return;
|
||
DictTaskID2Task_SendDo[TaskID].ConfirmIndex = Index;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 开始发送线程用
|
||
/// </summary>
|
||
/// <param name="obj"></param>
|
||
void sender_StartPushFileThread(object obj)
|
||
{
|
||
int TaskID = (int)obj;
|
||
if (!DictTaskID2Task_SendDo.ContainsKey(TaskID))
|
||
return;
|
||
|
||
FilePushTask task = DictTaskID2Task_SendDo[TaskID];
|
||
|
||
//构造读取文件流对象
|
||
using (FileStream fsRead = new FileStream(task.Path, FileMode.Open)) //打开文件，不能创建新的
|
||
{
|
||
//开辟临时缓存内存
|
||
byte[] byteArrayRead = new byte[cPackSize]; // 1字节*1024 = 1k 1k*1024 = 1M内存
|
||
|
||
App.log.Debug($"[文件发送者]开始发送文件!!");
|
||
//通过死缓存去读文本中的内容
|
||
while (true)
|
||
{
|
||
//若发送下标，大于确认下标一定数量，则等待
|
||
while (task.CurrIndex - task.ConfirmIndex > WaitSendPackCount)
|
||
{
|
||
Thread.Sleep(10);
|
||
}
|
||
|
||
//readCount 这个是保存真正读取到的字节数
|
||
int readCount = fsRead.Read(byteArrayRead, 0, byteArrayRead.Length);
|
||
|
||
//TODO 大小
|
||
//if (readCount == cPackSize)
|
||
//{
|
||
//}
|
||
//else
|
||
//{
|
||
//}
|
||
|
||
task.CurrIndex++;
|
||
SendFilePush(task.targetUID, TaskID, task.CurrIndex, byteArrayRead, readCount);
|
||
//App.log.Debug($"发送文件片段，下标->{task.CurrIndex}");
|
||
|
||
//既然是死循环 那么什么时候我们停止读取文本内容 我们知道文本最后一行的大小肯定是小于缓存内存大小的
|
||
if (readCount < byteArrayRead.Length)
|
||
{
|
||
App.log.Debug($"[文件发送者]文件发送完毕!! task.CurrIndex->{task.CurrIndex}");
|
||
DictTaskID2Task_SendDo.Remove(TaskID);
|
||
fsRead.Close();
|
||
break; //结束循环
|
||
}
|
||
}
|
||
}
|
||
}
|
||
#endregion
|
||
|
||
#region 接收者网络消息
|
||
/// <summary>
|
||
///
|
||
/// </summary>
|
||
/// <param name="UID"></param>
|
||
/// <param name="TaskID"></param>
|
||
/// <param name="bAgree">[1]同意 [0]拒绝</param>
|
||
public void SendFilePushConfirm_Resp(long UID, int TaskID, int bAgree)
|
||
{
|
||
Protobuf_TcpP2P_FilePushConfirm_RESP msg = new Protobuf_TcpP2P_FilePushConfirm_RESP()
|
||
{
|
||
TaskID = TaskID,
|
||
Agree = bAgree
|
||
};
|
||
App.clientMgr.SendToTargetSocket(UID, (int)CommandID.CmdTcpP2PFilepushConfirmResp, (int)ErrorCode.ErrorOk, ProtoBufHelper.Serizlize(msg));
|
||
}
|
||
/// <summary>
|
||
///
|
||
/// </summary>
|
||
/// <param name="UID"></param>
|
||
/// <param name="TaskID"></param>
|
||
/// <param name="Index"></param>
|
||
public void SendFilePush_Resp(long UID, int TaskID, long Index)
|
||
{
|
||
Protobuf_TcpP2P_FilePush_RESP msg = new Protobuf_TcpP2P_FilePush_RESP()
|
||
{
|
||
TaskID = TaskID,
|
||
FinishIndex = Index
|
||
};
|
||
App.clientMgr.SendToTargetSocket(UID, (int)CommandID.CmdTcpP2PFilepushResp, (int)ErrorCode.ErrorOk, ProtoBufHelper.Serizlize(msg));
|
||
}
|
||
void RecvTcpP2PFilepushConfirm(long uid, byte[] reqData)
|
||
{
|
||
Protobuf_TcpP2P_FilePushConfirm msg = ProtoBufHelper.DeSerizlize<Protobuf_TcpP2P_FilePushConfirm>(reqData);
|
||
receiver_AddRecvFileReady(new FilePushTask()
|
||
{
|
||
TaskID = msg.TaskID,
|
||
PackSize = cPackSize,
|
||
ContentType = msg.ContentType,
|
||
FileHash = msg.FileHash,
|
||
FileLenght = msg.FileLenght,
|
||
FileName = msg.FileName,
|
||
PackCount = msg.PackCount,
|
||
|
||
Path = "",//待插入
|
||
fromUID = uid,
|
||
targetUID = App.userMgr.MainPlayer.UID,
|
||
CurrIndex = -1,
|
||
ConfirmIndex= -1,//虽然用不上
|
||
});
|
||
//TODO 事件 EventSystem.Instance.PostEvent(EEvent.OnP2PChatMsg, uid, msg.ChatMsg);
|
||
}
|
||
void RecvTcpP2PFilepush(long uid, byte[] reqData)
|
||
{
|
||
Protobuf_TcpP2P_FilePush msg = ProtoBufHelper.DeSerizlize<Protobuf_TcpP2P_FilePush>(reqData);
|
||
if (!DictTaskID2Task_RecvDataQueue.ContainsKey(msg.TaskID))
|
||
return;
|
||
|
||
if (msg == null)
|
||
{
|
||
}
|
||
|
||
DictTaskID2Task_RecvDataQueue[msg.TaskID].Enqueue(msg);
|
||
}
|
||
|
||
#endregion
|
||
|
||
#region 接受者逻辑
|
||
Dictionary<int, FilePushTask> DictTaskID2Task_RecvReady = new Dictionary<int, FilePushTask>();
|
||
Dictionary<int, FilePushTask> DictTaskID2Task_RecvDo = new Dictionary<int, FilePushTask>();
|
||
Dictionary<int, Queue<Protobuf_TcpP2P_FilePush>> DictTaskID2Task_RecvDataQueue = new Dictionary<int, Queue<Protobuf_TcpP2P_FilePush>>();
|
||
/// <summary>
|
||
/// 接收数据准备
|
||
/// </summary>
|
||
/// <param name="task"></param>
|
||
void receiver_AddRecvFileReady(FilePushTask task)
|
||
{
|
||
DictTaskID2Task_RecvReady[task.TaskID] = task;
|
||
|
||
//自动接收文件
|
||
if (bAutoRecv)
|
||
{
|
||
receiver_AgreeRecvFile(task.TaskID);
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 同意文件传输
|
||
/// </summary>
|
||
/// <param name="TaskID"></param>
|
||
public void receiver_AgreeRecvFile(int TaskID)
|
||
{
|
||
if (DictTaskID2Task_RecvReady.ContainsKey(TaskID))
|
||
{
|
||
FilePushTask task = DictTaskID2Task_RecvReady[TaskID];
|
||
DictTaskID2Task_RecvDo[TaskID] = task;
|
||
DictTaskID2Task_RecvReady.Remove(TaskID);
|
||
receiver_StartRecvFile(TaskID);
|
||
App.log.Debug("[文件接收者]：发送同意接收");
|
||
SendFilePushConfirm_Resp(task.fromUID, task.TaskID, 1);
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 开始接收文件传输
|
||
/// </summary>
|
||
/// <param name="TaskID"></param>
|
||
void receiver_StartRecvFile(int TaskID)
|
||
{
|
||
if (!DictTaskID2Task_RecvDo.ContainsKey(TaskID))
|
||
{
|
||
return;
|
||
}
|
||
|
||
FilePushTask task = DictTaskID2Task_RecvDo[TaskID];
|
||
DictTaskID2Task_RecvDataQueue[TaskID] = new Queue<Protobuf_TcpP2P_FilePush>();
|
||
|
||
Thread thread = new Thread(receiver_StartRecvFileThread);
|
||
thread.IsBackground = true;
|
||
thread.Start(TaskID);
|
||
App.log.Debug("[文件接收者]：准备接收线程");
|
||
}
|
||
|
||
/// <summary>
|
||
/// 接收任务线程
|
||
/// </summary>
|
||
/// <param name="obj"></param>
|
||
void receiver_StartRecvFileThread(object obj)
|
||
{
|
||
int TaskID = (int)obj;
|
||
FilePushTask task = null;
|
||
if (!DictTaskID2Task_RecvDo.ContainsKey(TaskID))
|
||
return;
|
||
|
||
task = DictTaskID2Task_RecvDo[TaskID];
|
||
|
||
string fullpath = System.Environment.CurrentDirectory + "//" + task.FileName;
|
||
FileInfo myFile = new FileInfo(fullpath);
|
||
FileStream fs = myFile.Create();
|
||
//int WriteLenght = 0;
|
||
//TODO 打开文件写入
|
||
while (true)
|
||
{
|
||
Queue<Protobuf_TcpP2P_FilePush> dataqueue = DictTaskID2Task_RecvDataQueue[TaskID];
|
||
bool bflag = false;
|
||
while (dataqueue.Count > 0)
|
||
{
|
||
Protobuf_TcpP2P_FilePush msg = dataqueue.Dequeue();
|
||
if (msg == null)
|
||
continue;
|
||
|
||
bflag = true;
|
||
byte[] bytedata = msg.FileData.ToArray();
|
||
if (msg.PackIndex - task.CurrIndex != 1)
|
||
{
|
||
App.log.Error($"[文件接收者]接收下标错误，msg.PackIndex->{msg.PackIndex} task.CurrIndex=>{task.CurrIndex}");
|
||
}
|
||
//写入文件
|
||
fs.Write(bytedata, 0, msg.PackLenght);
|
||
//WriteLenght += msg.PackLenght;
|
||
task.CurrIndex = msg.PackIndex;
|
||
//todo 写入片段
|
||
//更新进度
|
||
|
||
//App.log.Debug($"[文件接收者]收到文件片段，下标->{task.CurrIndex}");
|
||
}
|
||
|
||
if (bflag)
|
||
{
|
||
//回执收到信息
|
||
SendFilePush_Resp(task.fromUID, TaskID, task.CurrIndex);
|
||
}
|
||
|
||
if (task.CurrIndex + 1 >= task.PackCount)
|
||
{
|
||
//文件接收完毕
|
||
App.log.Debug($"[文件接收者]文件接收完毕!! task.CurrIndex->{task.CurrIndex} task.PackCount->{task.PackCount}");
|
||
//删除进行中字典
|
||
DictTaskID2Task_RecvDo.Remove(TaskID);
|
||
//删除进行中数据队列
|
||
DictTaskID2Task_RecvDataQueue.Remove(TaskID);
|
||
break;
|
||
}
|
||
|
||
Thread.Sleep(10);
|
||
}
|
||
//TODO 结束写入
|
||
fs.Close();
|
||
}
|
||
#endregion
|
||
}
|
||
|
||
public class FilePushTask
|
||
{
|
||
public int TaskID;//传输任务ID
|
||
public long fromUID;//发送者UID
|
||
public long targetUID;
|
||
public string Path;
|
||
public long FileLenght;//文件长度
|
||
public string ContentType;//ContentType
|
||
public string FileName;//文件名
|
||
public int PackSize;//每个包大小
|
||
public long PackCount;//包数量
|
||
public string FileHash;//文件hash
|
||
|
||
public long CurrIndex;
|
||
public long ConfirmIndex;
|
||
}
|
||
}
|