net: Refactor packet length computations

[pandora-u-boot.git] / net / rarp.c

diff --git a/net/rarp.c b/net/rarp.c
index bd073c3..a715c2c 100644 (file)
--- a/net/rarp.c
+++ b/net/rarp.c
@@ -43,10 +43,10 @@ int RarpTry;
  */
 void rarp_receive(struct ip_udp_hdr *ip, unsigned len)
 {
-       ARP_t *arp;
+       struct arp_hdr *arp;
 
        debug("Got RARP\n");
-       arp = (ARP_t *)ip;
+       arp = (struct arp_hdr *)ip;
        if (len < ARP_HDR_SIZE) {
                printf("bad length %d < %d\n", len, ARP_HDR_SIZE);
                return;
@@ -87,14 +87,16 @@ static void RarpTimeout(void)
 void RarpRequest(void)
 {
        uchar *pkt;
-       ARP_t *rarp;
+       struct arp_hdr *rarp;
+       int eth_hdr_size;
 
        printf("RARP broadcast %d\n", ++RarpTry);
        pkt = NetTxPacket;
 
-       pkt += NetSetEther(pkt, NetBcastAddr, PROT_RARP);
+       eth_hdr_size = NetSetEther(pkt, NetBcastAddr, PROT_RARP);
+       pkt += eth_hdr_size;
 
-       rarp = (ARP_t *)pkt;
+       rarp = (struct arp_hdr *)pkt;
 
        rarp->ar_hrd = htons(ARP_ETHER);
        rarp->ar_pro = htons(PROT_IP);
@@ -108,7 +110,7 @@ void RarpRequest(void)
        /* dest IP addr set to broadcast */
        memset(&rarp->ar_data[16], 0xff,        4);
 
-       NetSendPacket(NetTxPacket, (pkt - NetTxPacket) + ARP_HDR_SIZE);
+       NetSendPacket(NetTxPacket, eth_hdr_size + ARP_HDR_SIZE);
 
        NetSetTimeout(TIMEOUT, RarpTimeout);
 }