Renamed IEEE802_3 to Dot3.

src/dot3.cpp

@@ -0,0 +1,166 @@
+/*
+ * Copyright (c) 2012, Nasel
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are
+ * met:
+ * 
+ * * Redistributions of source code must retain the above copyright
+ *   notice, this list of conditions and the following disclaimer.
+ * * Redistributions in binary form must reproduce the above
+ *   copyright notice, this list of conditions and the following disclaimer
+ *   in the documentation and/or other materials provided with the
+ *   distribution.
+ * 
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+ * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ *
+ */
+
+#include <cassert>
+#include <cstring>
+#include <stdexcept>
+#include <algorithm>
+#include "macros.h"
+#ifndef WIN32
+    #ifdef BSD
+        #include <net/if_dl.h>
+    #else
+        #include <netpacket/packet.h>
+    #endif
+    #include <net/ethernet.h>
+    #include <netinet/in.h>
+#endif
+#include "dot3.h"
+#include "packet_sender.h"
+#include "llc.h"
+
+namespace Tins {
+const Dot3::address_type Dot3::BROADCAST("ff:ff:ff:ff:ff:ff");
+
+Dot3::Dot3(const NetworkInterface& iface, 
+  const address_type &dst_hw_addr, const address_type &src_hw_addr, 
+  PDU* child)
+: PDU(child) 
+{
+    memset(&_eth, 0, sizeof(ethhdr));
+    this->dst_addr(dst_hw_addr);
+    this->src_addr(src_hw_addr);
+    this->iface(iface);
+    this->_eth.length = 0;
+
+}
+
+Dot3::Dot3(const uint8_t *buffer, uint32_t total_sz) 
+{
+    if(total_sz < sizeof(ethhdr))
+        throw std::runtime_error("Not enough size for an ethernetII header in the buffer.");
+    memcpy(&_eth, buffer, sizeof(ethhdr));
+    buffer += sizeof(ethhdr);
+    total_sz -= sizeof(ethhdr);
+    if(total_sz)
+        inner_pdu(new Tins::LLC(buffer, total_sz));
+}
+
+void Dot3::dst_addr(const address_type &new_dst_mac) {
+    std::copy(new_dst_mac.begin(), new_dst_mac.end(), _eth.dst_mac);
+}
+
+void Dot3::src_addr(const address_type &new_src_mac) {
+    std::copy(new_src_mac.begin(), new_src_mac.end(), _eth.src_mac);
+}
+
+void Dot3::iface(const NetworkInterface &new_iface) {
+    _iface = new_iface;
+}
+
+void Dot3::length(uint16_t new_length) {
+    this->_eth.length = Endian::host_to_be(new_length);
+}
+
+uint32_t Dot3::header_size() const {
+    return sizeof(ethhdr);
+}
+
+#ifndef WIN32
+void Dot3::send(PacketSender &sender) {
+    if(!_iface)
+        throw std::runtime_error("Interface has not been set");
+        
+    #ifndef BSD
+        struct sockaddr_ll addr;
+
+        memset(&addr, 0, sizeof(struct sockaddr_ll));
+
+        addr.sll_family = Endian::host_to_be<uint16_t>(PF_PACKET);
+        addr.sll_protocol = Endian::host_to_be<uint16_t>(ETH_P_ALL);
+        addr.sll_halen = address_type::address_size;
+        addr.sll_ifindex = _iface.id();
+        memcpy(&(addr.sll_addr), _eth.dst_mac, sizeof(_eth.dst_mac));
+
+        sender.send_l2(*this, (struct sockaddr*)&addr, (uint32_t)sizeof(addr));
+    #else
+        sender.send_l2(*this, 0, 0, _iface);
+    #endif
+}
+#endif // WIN32
+
+bool Dot3::matches_response(uint8_t *ptr, uint32_t total_sz) {
+    if(total_sz < sizeof(ethhdr))
+        return false;
+    ethhdr *eth_ptr = (ethhdr*)ptr;
+    if(!memcmp(eth_ptr->dst_mac, _eth.src_mac, sizeof(_eth.src_mac))) {
+        return true;
+    }
+    return false;
+}
+
+void Dot3::write_serialization(uint8_t *buffer, uint32_t total_sz, const PDU *parent) {
+    uint32_t my_sz = header_size();
+    bool set_length = _eth.length == 0;
+    assert(total_sz >= my_sz);
+
+    if (set_length)
+    	_eth.length = Endian::host_to_be(size() - sizeof(_eth));
+
+    memcpy(buffer, &_eth, sizeof(ethhdr));
+
+    if (set_length)
+    	_eth.length = 0;
+}
+
+#ifndef WIN32
+PDU *Dot3::recv_response(PacketSender &sender) {
+    if(!_iface)
+        throw std::runtime_error("Interface has not been set");
+    #ifndef BSD
+        struct sockaddr_ll addr;
+        memset(&addr, 0, sizeof(struct sockaddr_ll));
+
+        addr.sll_family = Endian::host_to_be<uint16_t>(PF_PACKET);
+        addr.sll_protocol = Endian::host_to_be<uint16_t>(ETH_P_802_3);
+        addr.sll_halen = address_type::address_size;
+        addr.sll_ifindex = _iface.id();
+        memcpy(&(addr.sll_addr), _eth.dst_mac, sizeof(_eth.dst_mac));
+
+        return sender.recv_l2(*this, (struct sockaddr*)&addr, (uint32_t)sizeof(addr));
+    #else
+        return sender.recv_l2(*this, 0, 0, _iface);
+    #endif
+}
+#endif // WIN32
+
+PDU *Dot3::clone_packet(const uint8_t *ptr, uint32_t total_sz) {
+    return new Dot3(ptr, total_sz);
+}
+}