diff --git a/src/net-tftp.cpp b/src/net-tftp.cpp index f5c2ec6..d514505 100644 --- a/src/net-tftp.cpp +++ b/src/net-tftp.cpp @@ -114,7 +114,7 @@ namespace Net::Tftp ); } - void HandleTftpDatagram( + void HandlePacket( const EthernetFrameHeader ethernetReqHeader, const Ipv4Header ipv4ReqHeader, const UdpDatagramHeader udpReqHeader, diff --git a/src/net-tftp.h b/src/net-tftp.h index 79add71..fe15fa0 100644 --- a/src/net-tftp.h +++ b/src/net-tftp.h @@ -1,6 +1,12 @@ #pragma once +#include +#include #include +struct EthernetFrameHeader; +struct Ipv4Header; +struct UdpDatagramHeader; + namespace Net::Tftp { const size_t TFTP_BLOCK_SIZE = 512; @@ -69,7 +75,7 @@ namespace Net::Tftp { TftpDataPacket& out, const uint8_t* buffer, size_t length); }; - void HandleTftpDatagram( + void HandlePacket( const EthernetFrameHeader ethernetReqHeader, const Ipv4Header ipv4ReqHeader, const UdpDatagramHeader udpReqHeader, diff --git a/src/net.cpp b/src/net.cpp index 4e9a7f5..b210d8a 100644 --- a/src/net.cpp +++ b/src/net.cpp @@ -63,7 +63,7 @@ void HandleUdpDatagram( } else if (udpHeader.destinationPort == UDP_PORT_TFTP) { - Net::Tftp::HandleTftpDatagram( + Net::Tftp::HandlePacket( ethernetHeader, ipv4Header, udpHeader,