private DynObject Transfer(ushort serverPort, byte[] buffer, int count) { //发送数据 Send(serverPort, buffer); ZmqMessage message = _ftpSocket.ReceiveMessage(new TimeSpan(0, 1, 30)); if (message.FrameCount == 0) { _isWorking = false; ReStart(); throw new TimeoutException("发送超时,请检查你的网络是否有问题,稍后重新上传"); } else { //反序列化,取出服务端响应数据 byte[] respData = message[message.FrameCount - 1].Buffer; TSerializer serializer = new TBinarySerializer(); serializer.FromBytes(respData); DynObject fileFragment = DynSerialize.ReadDynObject(serializer); serializer.Flush(); //105:发生异常 if ((byte)fileFragment["State"] == 105) { _isWorking = false; throw new ApplicationException(fileFragment["ExcepMsg"] as string); } return(fileFragment); } }
/// <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(); } }