Bläddra i källkod

Kernel: Use NetworkOrdered<T> in ARPPacket.

Andreas Kling 6 år sedan
förälder
incheckning
63f029ef9b
1 ändrade filer med 10 tillägg och 9 borttagningar
  1. 10 9
      Kernel/Net/ARP.h

+ 10 - 9
Kernel/Net/ARP.h

@@ -1,5 +1,6 @@
 #pragma once
 
+#include <AK/NetworkOrdered.h>
 #include <Kernel/Net/EtherType.h>
 #include <Kernel/Net/IPv4.h>
 #include <Kernel/Net/MACAddress.h>
@@ -20,11 +21,11 @@ struct ARPHardwareType {
 class [[gnu::packed]] ARPPacket
 {
 public:
-    word hardware_type() const { return ntohs(m_hardware_type); }
-    void set_hardware_type(word w) { m_hardware_type = htons(w); }
+    word hardware_type() const { return m_hardware_type; }
+    void set_hardware_type(word w) { m_hardware_type = w; }
 
-    word protocol_type() const { return ntohs(m_protocol_type); }
-    void set_protocol_type(word w) { m_protocol_type = htons(w); }
+    word protocol_type() const { return m_protocol_type; }
+    void set_protocol_type(word w) { m_protocol_type = w; }
 
     byte hardware_address_length() const { return m_hardware_address_length; }
     void set_hardware_address_length(byte b) { m_hardware_address_length = b; }
@@ -32,8 +33,8 @@ public:
     byte protocol_address_length() const { return m_protocol_address_length; }
     void set_protocol_address_length(byte b) { m_protocol_address_length = b; }
 
-    word operation() const { return ntohs(m_operation); }
-    void set_operation(word w) { m_operation = htons(w); }
+    word operation() const { return m_operation; }
+    void set_operation(word w) { m_operation = w; }
 
     const MACAddress& sender_hardware_address() const { return m_sender_hardware_address; }
     void set_sender_hardware_address(const MACAddress& address) { m_sender_hardware_address = address; }
@@ -48,11 +49,11 @@ public:
     void set_target_protocol_address(const IPv4Address& address) { m_target_protocol_address = address; }
 
 private:
-    word m_hardware_type { 0x0100 };
-    word m_protocol_type { 0x0008 };
+    NetworkOrdered<word> m_hardware_type { ARPHardwareType::Ethernet };
+    NetworkOrdered<word> m_protocol_type { EtherType::IPv4 };
     byte m_hardware_address_length { sizeof(MACAddress) };
     byte m_protocol_address_length { sizeof(IPv4Address) };
-    word m_operation { 0 };
+    NetworkOrdered<word> m_operation;
     MACAddress m_sender_hardware_address;
     IPv4Address m_sender_protocol_address;
     MACAddress m_target_hardware_address;