Exemplo n.º 1
0
        // Get implementation
        internal static bool Get
            (string LocalFilename, string RemoteFilename, string Host,
            TFTP.Modes Mode, int BlockSize, int Timeout)            
        {
            int recvLen, remoteFileSize = 0, buffer = BlockSize + 4;
            long bytesReceived = 0;            
            
            BinaryWriter BWriter = new BinaryWriter(File.Open(LocalFilename, FileMode.Create));
            
            TFTP.OpCodes opCode = new TFTP.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 = TFTPPacket.Create.Request
                (TFTP.OpCodes.RRQ, RemoteFilename, 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;

            // Fire connected event
            TFTP.FireOnConnect();
            
            while (true)
            {
                // Read opcode
                opCode = TFTPPacket.Read.OpCode(recvData);                

                // DATA packet
                if (opCode == TFTP.OpCodes.DATA)
                {
                    bytesReceived += recvLen - 4;               


                    
                    // Fire OnTransfer Event and pass along bytesReceived
                    TFTP.FireOnTransfer(bytesReceived, remoteFileSize);

                    for (int h = 4; h < recvLen; h++)
                    {   
                        BWriter.Write(recvData[h]);
                    }
                                        
                    sendData = TFTPPacket.Create.Ack(recvData[2], recvData[3]);

                    // Check if this packet is the last
                    if (recvLen < buffer)
                    {
                        // Send final ACK
                        UDPSock.SendTo(sendData, remoteEP);

                        // Fire OnTransferFinish Event
                        TFTP.FireOnTransferFinish();
                        break;
                    }
                }

                // OACK packet
                else if (opCode == TFTP.OpCodes.OACK)
                {
                    remoteFileSize = TFTPPacket.Read.TSize(recvData);                    
                    sendData = TFTPPacket.Create.Ack(0, 0);                    
                }

                // ERROR packet
                else if (opCode == TFTP.OpCodes.ERROR)
                {
                    TFTPPacket.Read.Error transferError = new TFTPPacket.Read.Error(recvData);
                    TFTP.FireOnTransferError
                        (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();

            // Fire OnDisconnect Event
            TFTP.FireOnDisconnect();
            return true;            
        }        
Exemplo n.º 2
0
        // Put implementation
        internal static bool Put
            (string LocalFilename, string RemoteFilename, string Host,
            TFTP.Modes Mode, int BlockSize, int Timeout)
        {
            int[] block = new int[2];
            int   bufferSize = BlockSize;
            long  fileSize, bytesSent = 0;

            BinaryReader BReader  = new BinaryReader(File.Open(LocalFilename, FileMode.Open));
            FileInfo     sendFile = new FileInfo(LocalFilename);

            TFTP.OpCodes opCode = new TFTP.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 = TFTPPacket.Create.Request
                                  (TFTP.OpCodes.WRQ, RemoteFilename, 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;

            // Fire OnConnect Event
            TFTP.FireOnConnect();

            while (true)
            {
                // Read opcode
                opCode = TFTPPacket.Read.OpCode(recvData);

                // ACK packet
                if (opCode == TFTP.OpCodes.ACK)
                {
                    block = TFTPPacket.Modify.IncrementBock(recvData, block);

                    sendData   = BReader.ReadBytes(bufferSize);
                    bytesSent += sendData.Length;

                    // Fire OnTransfer Event
                    TFTP.FireOnTransfer(bytesSent, fileSize);

                    sendData = TFTPPacket.Create.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 (TFTPPacket.Read.CheckBlock(sendData, recvData))
                            {
                                break;
                            }
                        }

                        // Fire OnTransferFinish Event
                        TFTP.FireOnTransferFinish();
                        break;
                    }
                }

                // OACK packet
                else if (opCode == TFTP.OpCodes.OACK)
                {
                    sendData   = BReader.ReadBytes(bufferSize);
                    sendData   = TFTPPacket.Create.Data(sendData, 0, 1);
                    bytesSent += sendData.Length - 4;

                    // Fire OnTransfer Event
                    TFTP.FireOnTransfer(bytesSent, fileSize);

                    if (fileSize == 0)
                    {
                        // Fire OnTransferFinish Event
                        TFTP.FireOnTransferFinish();
                        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 (TFTPPacket.Read.CheckBlock(sendData, recvData))
                                {
                                    break;
                                }
                            }
                            // Fire OnTransferFinish Event
                            TFTP.FireOnTransferFinish();
                            break;
                        }
                    }
                }
                else if (opCode == TFTP.OpCodes.ERROR)
                {
                    TFTPPacket.Read.Error transferError = new TFTPPacket.Read.Error(recvData);
                    TFTP.FireOnTransferError
                        (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();

            // Fire OnDisconnect Event
            TFTP.FireOnDisconnect();

            return(true);
        }