[PKTGEN]: srcmac fix

From: Adit Ranadive <adit.262@gmail.com>

Signed-off-by: Andrew Morton <akpm@linux-foundation.org>
Signed-off-by: David S. Miller <davem@davemloft.net>

1 files changed, 10 insertions, 0 deletions

diff --git a/net/core/pktgen.c b/net/core/pktgen.c
index 36fdea71d74..803d0c8826a 100644
--- a/net/core/pktgen.c
+++ b/net/core/pktgen.c
@@ -111,6 +111,9 @@
  *
  * 802.1Q/Q-in-Q support by Francesco Fondelli (FF) <francesco.fondelli@gmail.com>
  *
+ * Fixed src_mac command to set source mac of packet to value specified in
+ * command by Adit Ranadive <adit.262@gmail.com>
+ *
  */
 #include <linux/sys.h>
 #include <linux/types.h>
@@ -1451,8 +1454,11 @@ static ssize_t pktgen_if_write(struct file *file,
         }
         if (!strcmp(name, "src_mac")) {
                 char *v = valstr;
+                unsigned char old_smac[ETH_ALEN];
                 unsigned char *m = pkt_dev->src_mac;
 
+                memcpy(old_smac, pkt_dev->src_mac, ETH_ALEN);
+
                 len = strn_len(&user_buffer[i], sizeof(valstr) - 1);
                 if (len < 0) {
                         return len;
@@ -1481,6 +1487,10 @@ static ssize_t pktgen_if_write(struct file *file,
                         }
                 }
 
+                /* Set up Src MAC */
+                if (compare_ether_addr(old_smac, pkt_dev->src_mac))
+                        memcpy(&(pkt_dev->hh[6]), pkt_dev->src_mac, ETH_ALEN);
+
                 sprintf(pg_result, "OK: srcmac");
                 return count;
         }