1 files changed, 41 insertions, 2 deletions
diff --git a/src/common/Network/Packet/Arp.h b/src/common/Network/Packet/Arp.h
index a16605bd..e23ff139 100644
--- a/src/common/Network/Packet/Arp.h
+++ b/src/common/Network/Packet/Arp.h
@@ -20,8 +20,7 @@ limitations under the License.
 #pragma pack(push, 1)
 class ArpHdr {
  public:
-    enum arp_hdr_enum_e {
-        ARP_HDR_HRD_ETHER = 1,
+    enum arp_hdr_op_e {
         ARP_HDR_OP_REQUEST = 1, /* request to resolve address */
         ARP_HDR_OP_REPLY = 2, /* response to previous request */
         ARP_HDR_OP_REVREQUEST = 3, /* request proto addr given hardware */
@@ -30,6 +29,46 @@ class ArpHdr {
         ARP_HDR_OP_INVREPLY = 6, /* response identifying peer */
     };
 
+    enum arp_hdr_hrd_e {
+        ARP_HDR_HRD_ETHER = 1,
+    };
+    
+    enum arp_hdr_proto_e {
+        ARP_HDR_PROTO_IPV4 = 0x800,
+    };
+    
+    void setOp(uint16_t op) {
+        m_arp_op = PKT_HTONS(op);
+    }
+    
+    uint16_t getOp() const {
+        return PKT_NTOHS(m_arp_op);
+    }
+    
+    uint16_t getHrdType() const {
+        return PKT_NTOHS(m_arp_hrd);
+    }
+    
+    uint16_t getProtocolType() const {
+        return PKT_NTOHS(m_arp_pro);
+    }
+    
+    uint32_t getSip() const {
+        return PKT_NTOHL(m_arp_sip); 
+    }
+    
+    void setSip(uint32_t sip) {
+        m_arp_sip = PKT_HTONL(sip);
+    }
+    
+    uint32_t getTip() const {
+        return PKT_NTOHL(m_arp_tip); 
+    }
+    
+    void setTip(uint32_t tip) {
+        m_arp_tip = PKT_HTONL(tip);
+    }
+    
  public:
 	uint16_t m_arp_hrd;    /* format of hardware address */
 	uint16_t m_arp_pro;    /* format of protocol address */