net/: Augment CONFIG_COMMANDS tests with defined(CONFIG_CMD_*).
[platform/kernel/u-boot.git] / net / rarp.c
index b8cc542..39cabb3 100644 (file)
@@ -29,7 +29,7 @@
 #include "rarp.h"
 #include "tftp.h"
 
-#if (CONFIG_COMMANDS & CFG_CMD_NET)
+#if (CONFIG_COMMANDS & CFG_CMD_NET) || defined(CONFIG_CMD_NET)
 
 #define TIMEOUT                5               /* Seconds before trying BOOTP again */
 #ifndef        CONFIG_NET_RETRY_COUNT
@@ -49,15 +49,22 @@ RarpHandler(uchar * dummi0, unsigned dummi1, unsigned dummi2, unsigned dummi3)
 {
        char *s;
 #ifdef DEBUG
-       printf("Got good RARP\n");
+       puts ("Got good RARP\n");
 #endif
-       if (((s = getenv("autoload")) != NULL) && (*s == 'n')) {
-               NetState = NETLOOP_SUCCESS;
-               return;
-       }
-       else if ((s != NULL) && !strcmp(s, "NFS")) {
-               NfsStart();
-               return;
+       if ((s = getenv("autoload")) != NULL) {
+               if (*s == 'n') {
+                       /*
+                        * Just use RARP to configure system;
+                        * Do not use TFTP/NFS to to load the bootfile.
+                        */
+                       NetState = NETLOOP_SUCCESS;
+                       return;
+#if (CONFIG_COMMANDS & CFG_CMD_NFS) || defined(CONFIG_CMD_NFS)
+               } else if ((s != NULL) && !strcmp(s, "NFS")) {
+                       NfsStart();
+                       return;
+#endif
+               }
        }
        TftpStart ();
 }
@@ -89,8 +96,7 @@ RarpRequest (void)
        printf("RARP broadcast %d\n", ++RarpTry);
        pkt = NetTxPacket;
 
-       NetSetEther(pkt, NetBcastAddr, PROT_RARP);
-       pkt += ETHER_HDR_SIZE;
+       pkt += NetSetEther(pkt, NetBcastAddr, PROT_RARP);
 
        rarp = (ARP_t *)pkt;
 
@@ -107,7 +113,7 @@ RarpRequest (void)
                rarp->ar_data[16 + i] = 0xff;
        }
 
-       NetSendPacket(NetTxPacket, ETHER_HDR_SIZE + ARP_HDR_SIZE);
+       NetSendPacket(NetTxPacket, (pkt - NetTxPacket) + ARP_HDR_SIZE);
 
        NetSetTimeout(TIMEOUT * CFG_HZ, RarpTimeout);
        NetSetHandler(RarpHandler);