/// <summary>
/// 向服务端发送下载文件的请求
/// </summary>
/// <param name="destVPort">目的端口</param>
/// <param name="downloadPath">要下载的文件在服务端的路径</param>
/// <returns>将要下载文件的基本信息</returns>
private DynObject RequestDownload(ushort destVPort, string downloadPath)
{
DynObject fileFragment = new DynObject("FileFragment");
fileFragment["Path"] = Path.GetDirectoryName(downloadPath);
fileFragment["FileName"] = Path.GetFileName(downloadPath);
fileFragment["Extension"] = Path.GetExtension(downloadPath);
fileFragment["State"] = (byte)100;
fileFragment["MsgID"] = -1;
fileFragment["DataLength"] = 0;
fileFragment["Data"] = null;
//序列化要发送的数据
TSerializer serializer = new TBinarySerializer();
DynSerialize.WriteDynObject(serializer, fileFragment);
byte[] buffer = serializer.ToBytes();
serializer.Flush();
Send(_serverPort, buffer);
ZmqMessage msg = _ftpSocket.ReceiveMessage(new TimeSpan(0, 1, 30));
if (msg.FrameCount == 0)
{
_isWorking = false;
ReStart();
throw new TimeoutException("发送超时，请检查你的网络是否有问题,稍后重新下载");
}
//反序列化，服务器返回的数据
serializer = new TBinarySerializer();
serializer.FromBytes(msg[msg.FrameCount - 1].Buffer);
DynObject respFragment = DynSerialize.ReadDynObject(serializer);
serializer.Flush();
//发生异常
if ((byte)respFragment["State"] == 105)
{
_isWorking = false;
throw new FtpException(respFragment["ExcepMsg"] as string);
}
return(respFragment);
}
/// <summary>
/// 处理接收到的数据
/// </summary>
/// <param name="msgData">接收到的消息</param>
private byte[] DealMessage(byte[] msgData)
{
//反序列化，取出数据
TSerializer serializer = new TBinarySerializer();
serializer.FromBytes(msgData);
DynObject reqFragment = DynSerialize.ReadDynObject(serializer);
serializer.Flush();
try
{
DynObject repFragment = OnFileDataReceived(reqFragment);
//序列化并把数据返回给客户端
serializer = new TBinarySerializer();
DynSerialize.WriteDynObject(serializer, repFragment);
byte[] repData = serializer.ToBytes();
serializer.Flush();
return(repData);
}
catch (Exception ex)
{
//如果服务端需要记录日志，则由本事件把异常信息发出
if (OnException != null)
{
OnException(ex);
}
_isRunning = false;
DynObject repFragment = new DynObject("FileFragment");
repFragment["State"] = (byte)105;
repFragment["ExcepMsg"] = "FtpServer端异常，" + ex.Message;
//序列化并把异常数据返回给客户端
serializer = new TBinarySerializer();
DynSerialize.WriteDynObject(serializer, repFragment);
byte[] repData = serializer.ToBytes();
serializer.Flush();
return(repData);
}
}
/// <summary>
/// 文件下载
/// </summary>
/// <param name="recvAddress">服务端机器的地址</param>
/// <param name="savePath">文件在本地的保存路径</param>
/// <param name="downloadPath">要下载的文件在服务端的路径</param>
public void FileDownload(string recvAddress, string savePath, string downloadPath)
{
#region 前置检查
if (_isWorking)
{
throw new FtpException("一个FtpClient不能同时执行多个任务");
}
//验证本地是否已有同名文件
if (File.Exists(Path.Combine(savePath, Path.GetFileName(downloadPath))))
{
throw new FtpException(savePath + "目录中已存在同名文件");
}
#endregion 前置检查
#region 获取文件的基本信息
_remoteAddress = new VirtuaIP(recvAddress);
_serverPort = GetServerPort("FileDownload");
_isWorking = true;
//发送下载文件的请求
long fileSize = 0; //文件大小（字节）
long remaining = 0; //剩余未下载的文件大小
DynObject fileFragment = RequestDownload(_serverPort, downloadPath);
fileSize = remaining = (long)fileFragment["FileLength"];
#endregion 获取文件的基本信息
FileStream writeFileStream = null;
_isCancelDownload = false;
while (remaining >= 0 && _isWorking)
{
if (_isCancelDownload)
{
fileFragment["State"] = (byte)103;
}
switch ((byte)fileFragment["State"])
{
case 100:
string tempFile = Path.Combine(savePath, fileFragment["MD5"] + ".temp");
//如果存在该文件的临时文件，则进行断点续传,否则以MD5为名字创建临时文件开始传送数据
if (File.Exists(tempFile))
{
writeFileStream = new FileStream(tempFile, FileMode.Open);
fileFragment["MsgID"] = (int)(writeFileStream.Length / _bufferLength);
fileFragment["State"] = (byte)101;
fileFragment["Data"] = null;
}
else
{
writeFileStream = new FileStream(tempFile, FileMode.Create);
fileFragment["MsgID"] = 0;
fileFragment["State"] = (byte)101;
fileFragment["Data"] = null;
}
remaining = fileSize - (int)fileFragment["MsgID"] * _bufferLength;
break;
case 101: //文件传送中
writeFileStream.Seek((int)fileFragment["MsgID"] * _bufferLength, SeekOrigin.Begin);
writeFileStream.Write(fileFragment["Data"] as byte[], 0, (int)fileFragment["DataLength"]);
remaining = fileSize - ((int)fileFragment["MsgID"] * _bufferLength + (int)fileFragment["DataLength"]);
if (remaining <= 0)
{
//文件下载结束
fileFragment["State"] = (byte)102;
writeFileStream.Flush();
writeFileStream.Close();
writeFileStream = null;
_isWorking = false;
//校验文件传送是否正确
string path = Path.Combine(savePath, fileFragment["MD5"] as string + ".temp");
string tempFileMD5 = GetMD5HashFromFile(path);
if (tempFileMD5 == fileFragment["MD5"] as string)
{
File.Move(path, Path.Combine(savePath, fileFragment["FileName"] as string));
}
else
{
File.Delete(path);
throw new FtpException("文件未能成功下载");
}
}
fileFragment["MsgID"] = (int)fileFragment["MsgID"] + 1;
fileFragment["DataLength"] = 0;
fileFragment["Data"] = null;
break;
case 103: //中断文件传送
fileFragment["DataLength"] = 0;
fileFragment["Data"] = null;
writeFileStream.Flush();
writeFileStream.Close();
writeFileStream = null;
_isWorking = false;
break;
case 105: //服务端发生异常
writeFileStream.Flush();
writeFileStream.Close();
writeFileStream = null;
_isWorking = false;
throw new FtpException(fileFragment["ExcepMsg"] as string);
}
//向外界通知文件床送进度
if (ProgressEvent != null)
{
ProgressEvent(fileSize, fileSize - remaining);
}
//发送请求下一个包数据（发生异常告诉对方关闭文件流）
TSerializer serialize = new TBinarySerializer();
DynSerialize.WriteDynObject(serialize, fileFragment);
byte[] buffer = serialize.ToBytes();
serialize.Flush();
Send(_serverPort, buffer);
ZmqMessage message = _ftpSocket.ReceiveMessage(new TimeSpan(0, 1, 30));
if (message.FrameCount == 0)
{
writeFileStream.Close();
writeFileStream = null;
_isWorking = false;
ReStart();
throw new TimeoutException("发送超时，请检查你的网络是否有问题,稍后重新下载");
}
serialize = new TBinarySerializer();
serialize.FromBytes(message[message.FrameCount - 1].Buffer);
fileFragment = DynSerialize.ReadDynObject(serialize);
serialize.Flush();
}
}
/// <summary>
/// 文件上传
/// </summary>
/// <param name="recvAddress">服务端机器的地址</param>
/// <param name="filePath">准备上传的文件</param>
/// <param name="savePath">文件在服务端的保存路径</param>
public void FileUpload(string recvAddress, string filePath, string savePath)
{
if (_isWorking)
{
throw new FtpException("一个FtpClient不能同时执行多个任务");
}
CheckPostfix(filePath);//检查上传的文件类型是否合法
string md5 = GetMD5HashFromFile(filePath);
_remoteAddress = new VirtuaIP(recvAddress);
_serverPort = GetServerPort("FileUpload");
using (FileStream fStream = File.OpenRead(filePath))
{
long fileSize = fStream.Length; //文件数据长度
long remaining = fileSize; // 剩余没上传的数据长度
byte[] buffer = new byte[_bufferLength];
//构建文件片段
DynObject fileFragment = new DynObject("FileFragment");
fileFragment["MD5"] = md5;
fileFragment["FileName"] = Path.GetFileName(filePath);
fileFragment["Extension"] = Path.GetExtension(filePath);
fileFragment["Path"] = savePath;
fileFragment["DataLength"] = 0;
fileFragment["MsgID"] = -1;
fileFragment["State"] = (byte)0;
_isCancelUpload = false;
_isWorking = true;
int offset = 0;
while (remaining >= 0 && _isWorking)
{
if (_isCancelUpload)
{
fileFragment["State"] = (byte)3;
}
switch ((byte)fileFragment["State"])
{
case 0: //文件传送开始
//第一次发送文件,发送一些基本信息（如文件名等）
fileFragment["DataLength"] = 0;
fileFragment["Data"] = null;
break;
case 1: //文件传送中
fStream.Seek((int)fileFragment["MsgID"] * _bufferLength, SeekOrigin.Begin);
fileFragment["DataLength"] = fStream.Read(buffer, offset, buffer.Length - offset);
fileFragment["Data"] = buffer;
//重置剩余数据长度
remaining = fileSize - ((int)fileFragment["MsgID"] * _bufferLength + (int)fileFragment["DataLength"]);
if (remaining == 0)
{
_isWorking = false;
remaining = -1;
fileFragment["State"] = (byte)2;
}
break;
case 3: //中断文件传送
fileFragment["DataLength"] = 0;
fileFragment["Data"] = null;
_isWorking = false;
break;
}
//序列化
TSerializer serializer = new TBinarySerializer();
DynSerialize.WriteDynObject(serializer, fileFragment);
byte[] data = serializer.ToBytes();
serializer.Flush();
DynObject repFragment = Transfer(_serverPort, data, 0);
fileFragment["State"] = repFragment["State"];
fileFragment["MsgID"] = repFragment["MsgID"];
//向外界通知文件传送进度
if (ProgressEvent != null)
{
ProgressEvent(fileSize, fileSize - remaining);
}
}
}
}
