public byte[] Request(TFTPSession.OpCodes OpCode, string RemoteFileName, TFTPSession.Modes Mode, int BlockSize, long TransferSize, int Timeout) { // Request packet structure // ----------------------------------------------------------------------------- // |OpCode|FileName|0|Mode|0|BlkSize|0|BSVal|0|TSize|0|TSVal|0|Timeout|0|TVal|0| // ----------------------------------------------------------------------------- int len; string packetStr = ""; string mode = Mode.ToString().ToLower(); string blockSize = BlockSize.ToString(); string nullChar = "\0"; byte[] packet; // Create packet as a string switch (OpCode) { case TFTPSession.OpCodes.RRQ: packetStr = nullChar + (char)1; break; case TFTPSession.OpCodes.WRQ: packetStr = nullChar + (char)2; break; } packetStr += RemoteFileName + nullChar + mode + nullChar + "blksize" + nullChar + BlockSize.ToString() + nullChar + "tsize" + nullChar + TransferSize.ToString() + nullChar + "timeout" + nullChar + Timeout.ToString() + nullChar; len = packetStr.Length; packet = new byte[len]; // Encode packet as ASCII bytes packet = System.Text.Encoding.ASCII.GetBytes(packetStr); return(packet); }
public bool Get(string LocalFile, string RemoteFile, string Host, Modes Mode, int BlockSize, int Timeout) { int recvLen, remoteFileSize = 0, buffer = BlockSize + 4; long bytesReceived = 0; BinaryWriter BWriter = new BinaryWriter(File.Open(LocalFile, FileMode.Create)); TFTPSession.OpCodes opCode = new TFTPSession.OpCodes(); IPHostEntry hInfo = Dns.GetHostEntry(Host); IPAddress address = hInfo.AddressList[0]; IPEndPoint remoteEP = new IPEndPoint(address, 69); EndPoint localEP = (remoteEP); Socket UDPSock = new Socket (remoteEP.AddressFamily, SocketType.Dgram, ProtocolType.Udp); // Create initial request and buffer for response byte[] sendData = _packetBuilder.Request(TFTPSession.OpCodes.RRQ, RemoteFile, Mode, BlockSize, 0, Timeout); byte[] recvData = new byte[BlockSize + 4]; UDPSock.ReceiveTimeout = Timeout * 1000; // Send request and wait for response UDPSock.SendTo(sendData, remoteEP); recvLen = UDPSock.ReceiveFrom(recvData, ref localEP); // Get TID remoteEP.Port = ((IPEndPoint)localEP).Port; // Invoke connected event Connected.Invoke(); while (true) { // Read opcode opCode = _packetReader.ReadOpCode(recvData); // DATA packet if (opCode == TFTPSession.OpCodes.DATA) { bytesReceived += recvLen - 4; // Invoke Transferring Event Transferring.Invoke(bytesReceived, remoteFileSize); for (int h = 4; h < recvLen; h++) { BWriter.Write(recvData[h]); } sendData = _packetBuilder.Ack(recvData[2], recvData[3]); // Check if this packet is the last if (recvLen < buffer) { // Send final ACK UDPSock.SendTo(sendData, remoteEP); // Invoked TransferFinished Event TransferFinished.Invoke(); break; } } // OACK packet else if (opCode == TFTPSession.OpCodes.OACK) { remoteFileSize = _packetReader.ReadTransferSize(recvData); sendData = _packetBuilder.Ack(0, 0); } // ERROR packet else if (opCode == TFTPSession.OpCodes.ERROR) { ErrorPacket transferError = _packetReader.ReadError(recvData); TransferFailed.Invoke(transferError.Code, transferError.Message); break; } // Send next packet UDPSock.SendTo(sendData, remoteEP); recvLen = UDPSock.ReceiveFrom(recvData, ref localEP); remoteEP.Port = ((IPEndPoint)localEP).Port; } BWriter.Close(); UDPSock.Close(); // Invoke Disconnected Event Disconnected.Invoke(); return(true); }
public bool Put(string LocalFile, string RemoteFile, string Host, Modes Mode, int BlockSize, int Timeout) { int[] block = new int[2]; int bufferSize = BlockSize; long fileSize, bytesSent = 0; BinaryReader BReader = new BinaryReader(File.Open(LocalFile, FileMode.Open)); FileInfo sendFile = new FileInfo(LocalFile); TFTPSession.OpCodes opCode = new TFTPSession.OpCodes(); IPHostEntry hostInfo = Dns.GetHostEntry(Host); IPAddress address = hostInfo.AddressList[0]; IPEndPoint remoteEP = new IPEndPoint(address, 69); EndPoint localEP = (remoteEP); Socket UDPSock = new Socket (remoteEP.AddressFamily, SocketType.Dgram, ProtocolType.Udp); // Retrieve filesize for tsize option fileSize = sendFile.Length; // Create initial request and buffer for response byte[] sendData = _packetBuilder.Request(TFTPSession.OpCodes.WRQ, RemoteFile, Mode, BlockSize, fileSize, Timeout); byte[] recvData = new byte[bufferSize]; UDPSock.ReceiveTimeout = Timeout * 1000; // Send request and wait for response UDPSock.SendTo(sendData, remoteEP); UDPSock.ReceiveFrom(recvData, ref localEP); //Get TID remoteEP.Port = ((IPEndPoint)localEP).Port; // Invoke Connected Event Connected.Invoke(); while (true) { // Read opcode opCode = _packetReader.ReadOpCode(recvData); // ACK packet if (opCode == TFTPSession.OpCodes.ACK) { block = _packetBuilder.IncrementBock(recvData, block); sendData = BReader.ReadBytes(bufferSize); bytesSent += sendData.Length; // Invoke Transferring Event Transferring.Invoke(bytesSent, fileSize); sendData = _packetBuilder.Data(sendData, block[0], block[1]); // Check if this packet is the last if (sendData.Length < bufferSize + 4) { // Send final data packet and wait for ack while (true) { UDPSock.SendTo(sendData, remoteEP); UDPSock.ReceiveFrom(recvData, ref localEP); remoteEP.Port = ((IPEndPoint)localEP).Port; // Check the blocks and break free if equal if (_packetReader.CompareBlocks(sendData, recvData)) { break; } } // Invoke TransferFinished Event TransferFinished.Invoke(); break; } } // OACK packet else if (opCode == TFTPSession.OpCodes.OACK) { sendData = BReader.ReadBytes(bufferSize); sendData = _packetBuilder.Data(sendData, 0, 1); bytesSent += sendData.Length - 4; // Invoke Transferring Event Transferring.Invoke(bytesSent, fileSize); if (fileSize == 0) { // Invoke TransferFinished Event TransferFinished.Invoke(); break; } else { // Check if this packet is the last if (sendData.Length < bufferSize + 4) { // Send final data packet and wait for ack while (true) { UDPSock.SendTo(sendData, remoteEP); UDPSock.ReceiveFrom(recvData, ref localEP); remoteEP.Port = ((IPEndPoint)localEP).Port; // Check the blocks and break free if equal if (_packetReader.CompareBlocks(sendData, recvData)) { break; } } // Invoke TransferFinished Event TransferFinished.Invoke(); break; } } } else if (opCode == TFTPSession.OpCodes.ERROR) { ErrorPacket transferError = _packetReader.ReadError(recvData); TransferFailed.Invoke(transferError.Code, transferError.Message); break; } // Send next packet UDPSock.SendTo(sendData, remoteEP); UDPSock.ReceiveFrom(recvData, ref localEP); remoteEP.Port = ((IPEndPoint)localEP).Port; } BReader.Close(); UDPSock.Close(); // Invoke Disconnected Event Disconnected.Invoke(); return(true); }