Cleanups to RNDIS device demos. Fix issue in RNDIS demos where the memory would becom...

[pub/lufa.git] / Demos / Device / ClassDriver / RNDISEthernet / Lib / UDP.c

diff --git a/Demos/Device/ClassDriver/RNDISEthernet/Lib/UDP.c b/Demos/Device/ClassDriver/RNDISEthernet/Lib/UDP.c
index dbd33d1..ea67951 100644 (file)
--- a/Demos/Device/ClassDriver/RNDISEthernet/Lib/UDP.c
+++ b/Demos/Device/ClassDriver/RNDISEthernet/Lib/UDP.c
@@ -55,12 +55,13 @@ int16_t UDP_ProcessUDPPacket(void* IPHeaderInStart, void* UDPHeaderInStart, void
        \r
        DecodeUDPHeader(UDPHeaderInStart);\r
        \r
-       /* Check to see if the UDP packet is a DHCP packet */\r
-       if (SwapEndian_16(UDPHeaderIN->DestinationPort) == UDP_PORT_DHCP_REQUEST)\r
+       switch (SwapEndian_16(UDPHeaderIN->DestinationPort))\r
        {\r
-               RetSize = DHCP_ProcessDHCPPacket(IPHeaderInStart,\r
-                                                &((uint8_t*)UDPHeaderInStart)[sizeof(UDP_Header_t)],\r
-                                            &((uint8_t*)UDPHeaderOutStart)[sizeof(UDP_Header_t)]);\r
+               case UDP_PORT_DHCP_REQUEST:\r
+                       RetSize = DHCP_ProcessDHCPPacket(IPHeaderInStart,\r
+                                                        &((uint8_t*)UDPHeaderInStart)[sizeof(UDP_Header_t)],\r
+                                                    &((uint8_t*)UDPHeaderOutStart)[sizeof(UDP_Header_t)]);\r
+                       break;\r
        }\r
        \r
        /* Check to see if the protocol processing routine has filled out a response */\r