1 /* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
2 /*
3 * Copyright (c) 2005 INRIA
4 *
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License version 2 as
7 * published by the Free Software Foundation;
8 *
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17 *
18 * Author: Mathieu Lacage <mathieu.lacage@sophia.inria.fr>
19 */
20
21 #include "ns3/assert.h"
22 #include "ns3/address-utils.h"
23 #include "arp-header.h"
24 #include "ns3/log.h"
25
26 namespace ns3 {
27
28 NS_LOG_COMPONENT_DEFINE ("ArpHeader");
29
30 NS_OBJECT_ENSURE_REGISTERED (ArpHeader);
31
32 void
SetRequest(Address sourceHardwareAddress,Ipv4Address sourceProtocolAddress,Address destinationHardwareAddress,Ipv4Address destinationProtocolAddress)33 ArpHeader::SetRequest (Address sourceHardwareAddress,
34 Ipv4Address sourceProtocolAddress,
35 Address destinationHardwareAddress,
36 Ipv4Address destinationProtocolAddress)
37 {
38 NS_LOG_FUNCTION (this << sourceHardwareAddress << sourceProtocolAddress << destinationHardwareAddress << destinationProtocolAddress);
39 m_type = ARP_TYPE_REQUEST;
40 m_macSource = sourceHardwareAddress;
41 m_macDest = destinationHardwareAddress;
42 m_ipv4Source = sourceProtocolAddress;
43 m_ipv4Dest = destinationProtocolAddress;
44 }
45 void
SetReply(Address sourceHardwareAddress,Ipv4Address sourceProtocolAddress,Address destinationHardwareAddress,Ipv4Address destinationProtocolAddress)46 ArpHeader::SetReply (Address sourceHardwareAddress,
47 Ipv4Address sourceProtocolAddress,
48 Address destinationHardwareAddress,
49 Ipv4Address destinationProtocolAddress)
50 {
51 NS_LOG_FUNCTION (this << sourceHardwareAddress << sourceProtocolAddress << destinationHardwareAddress << destinationProtocolAddress);
52 m_type = ARP_TYPE_REPLY;
53 m_macSource = sourceHardwareAddress;
54 m_macDest = destinationHardwareAddress;
55 m_ipv4Source = sourceProtocolAddress;
56 m_ipv4Dest = destinationProtocolAddress;
57 }
58 bool
IsRequest(void) const59 ArpHeader::IsRequest (void) const
60 {
61 NS_LOG_FUNCTION (this);
62 return (m_type == ARP_TYPE_REQUEST) ? true : false;
63 }
64 bool
IsReply(void) const65 ArpHeader::IsReply (void) const
66 {
67 NS_LOG_FUNCTION (this);
68 return (m_type == ARP_TYPE_REPLY) ? true : false;
69 }
70 Address
GetSourceHardwareAddress(void) const71 ArpHeader::GetSourceHardwareAddress (void) const
72 {
73 NS_LOG_FUNCTION (this);
74 return m_macSource;
75 }
76 Address
GetDestinationHardwareAddress(void) const77 ArpHeader::GetDestinationHardwareAddress (void) const
78 {
79 NS_LOG_FUNCTION (this);
80 return m_macDest;
81 }
82 Ipv4Address
GetSourceIpv4Address(void) const83 ArpHeader::GetSourceIpv4Address (void) const
84 {
85 NS_LOG_FUNCTION (this);
86 return m_ipv4Source;
87 }
88 Ipv4Address
GetDestinationIpv4Address(void) const89 ArpHeader::GetDestinationIpv4Address (void) const
90 {
91 NS_LOG_FUNCTION (this);
92 return m_ipv4Dest;
93 }
94
95
96 TypeId
GetTypeId(void)97 ArpHeader::GetTypeId (void)
98 {
99 static TypeId tid = TypeId ("ns3::ArpHeader")
100 .SetParent<Header> ()
101 .SetGroupName ("Internet")
102 .AddConstructor<ArpHeader> ()
103 ;
104 return tid;
105 }
106 TypeId
GetInstanceTypeId(void) const107 ArpHeader::GetInstanceTypeId (void) const
108 {
109 NS_LOG_FUNCTION (this);
110 return GetTypeId ();
111 }
112 void
Print(std::ostream & os) const113 ArpHeader::Print (std::ostream &os) const
114 {
115 NS_LOG_FUNCTION (this << &os);
116 if (IsRequest ())
117 {
118 os << "request "
119 << "source mac: " << m_macSource << " "
120 << "source ipv4: " << m_ipv4Source << " "
121 << "dest ipv4: " << m_ipv4Dest
122 ;
123 }
124 else
125 {
126 NS_ASSERT (IsReply ());
127 os << "reply "
128 << "source mac: " << m_macSource << " "
129 << "source ipv4: " << m_ipv4Source << " "
130 << "dest mac: " << m_macDest << " "
131 << "dest ipv4: " <<m_ipv4Dest
132 ;
133 }
134 }
135 uint32_t
GetSerializedSize(void) const136 ArpHeader::GetSerializedSize (void) const
137 {
138 NS_LOG_FUNCTION (this);
139 NS_ASSERT ((m_macSource.GetLength () == 6) || (m_macSource.GetLength () == 8) || (m_macSource.GetLength () == 1));
140 NS_ASSERT (m_macSource.GetLength () == m_macDest.GetLength ());
141
142 uint32_t length = 16; // Length minus two hardware addresses
143 length += m_macSource.GetLength () * 2;
144
145 return length;
146 }
147
148 void
Serialize(Buffer::Iterator start) const149 ArpHeader::Serialize (Buffer::Iterator start) const
150 {
151 NS_LOG_FUNCTION (this << &start);
152 Buffer::Iterator i = start;
153 NS_ASSERT (m_macSource.GetLength () == m_macDest.GetLength ());
154
155 /* ethernet */
156 i.WriteHtonU16 (0x0001);
157 /* ipv4 */
158 i.WriteHtonU16 (0x0800);
159 i.WriteU8 (m_macSource.GetLength ());
160 i.WriteU8 (4);
161 i.WriteHtonU16 (m_type);
162 WriteTo (i, m_macSource);
163 WriteTo (i, m_ipv4Source);
164 WriteTo (i, m_macDest);
165 WriteTo (i, m_ipv4Dest);
166 }
167
168 uint32_t
Deserialize(Buffer::Iterator start)169 ArpHeader::Deserialize (Buffer::Iterator start)
170 {
171 NS_LOG_FUNCTION (this << &start);
172 Buffer::Iterator i = start;
173 i.Next (2); // Skip HRD
174 uint32_t protocolType = i.ReadNtohU16 (); // Read PRO
175 uint32_t hardwareAddressLen = i.ReadU8 (); // Read HLN
176 uint32_t protocolAddressLen = i.ReadU8 (); // Read PLN
177
178 //
179 // It is implicit here that we have a protocol type of 0x800 (IP).
180 // It is also implicit here that we are using Ipv4 (PLN == 4).
181 // If this isn't the case, we need to return an error since we don't want to
182 // be too fragile if we get connected to real networks.
183 //
184 if (protocolType != 0x800 || protocolAddressLen != 4)
185 {
186 return 0;
187 }
188
189 m_type = i.ReadNtohU16 (); // Read OP
190 ReadFrom (i, m_macSource, hardwareAddressLen); // Read SHA (size HLN)
191 ReadFrom (i, m_ipv4Source); // Read SPA (size PLN == 4)
192 ReadFrom (i, m_macDest, hardwareAddressLen); // Read THA (size HLN)
193 ReadFrom (i, m_ipv4Dest); // Read TPA (size PLN == 4)
194 return GetSerializedSize ();
195 }
196
197 } // namespace ns3
198