Added enum to identify PDU type. ARP is being sent but not well implemented

src/ethernetII.cpp

@@ -0,0 +1,115 @@
+/*
+ * libtins is a net packet wrapper library for crafting and
+ * interpreting sniffed packets.
+ *
+ * Copyright (C) 2011 Nasel
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
+ */
+
+#include <cassert>
+#include <cstring>
+
+#ifndef WIN32
+    #include <net/ethernet.h>
+    #include <netpacket/packet.h>
+    #include <netinet/in.h>
+#endif
+
+#include "ethernetII.h"
+#include "utils.h"
+
+#include <iostream>
+
+Tins::EthernetII::EthernetII(const uint8_t* mac_dst, const uint8_t* mac_src, const std::string& iface, PDU* child) throw (std::runtime_error) : PDU(ETHERTYPE_IP, child) {
+
+    this->dst_mac(mac_dst);
+    this->src_mac(mac_src);
+    this->iface(iface);
+    this->header.payload_type = 0;
+
+}
+
+Tins::EthernetII::EthernetII(const uint8_t* mac_dst, const uint8_t* mac_src, uint32_t iface_index, PDU* child) : PDU(ETHERTYPE_IP, child) {
+    this->dst_mac(mac_dst);
+    this->src_mac(mac_src);
+    this->iface(iface_index);
+    this->header.payload_type = 0;
+}
+
+void Tins::EthernetII::dst_mac(const uint8_t* new_dst_mac) {
+    memcpy(this->header.dst_mac, new_dst_mac, 6);
+}
+
+void Tins::EthernetII::src_mac(const uint8_t* new_src_mac) {
+    memcpy(this->header.src_mac, new_src_mac, 6);
+}
+
+void Tins::EthernetII::iface(uint32_t new_iface_index) {
+    this->_iface_index = new_iface_index;
+}
+
+void Tins::EthernetII::iface(const std::string& new_iface) throw (std::runtime_error) {
+    if (!Tins::Utils::interface_id(new_iface, this->_iface_index)) {
+        throw std::runtime_error("Invalid interface name!");
+    }
+}
+
+uint32_t Tins::EthernetII::header_size() const {
+    return sizeof(ethernet_header);
+}
+
+bool Tins::EthernetII::send(PacketSender* sender) {
+
+    struct sockaddr_ll addr;
+
+    memset(&addr, 0, sizeof(struct sockaddr_ll));
+
+    addr.sll_family = Utils::net_to_host_s(PF_PACKET);
+    addr.sll_protocol = Utils::net_to_host_s(ETH_P_ALL);
+    addr.sll_halen = 6;
+    addr.sll_ifindex = this->_iface_index;
+    memcpy(&(addr.sll_addr), this->header.dst_mac, 6);
+
+    return sender->send_l2(this, (struct sockaddr*)&addr, (uint32_t)sizeof(addr));
+
+}
+
+void Tins::EthernetII::write_serialization(uint8_t *buffer, uint32_t total_sz, const PDU *parent) {
+    uint32_t my_sz = header_size();
+    assert(total_sz >= my_sz);
+
+    /* Inner type defaults to IP */
+
+    if ((this->header.payload_type == 0) && this->inner_pdu()) {
+        uint16_t type = ETHERTYPE_IP;
+        switch (this->inner_pdu()->pdu_type()) {
+            case PDU::IP:
+                type = ETHERTYPE_IP;
+                break;
+            case PDU::ARP:
+                type = ETHERTYPE_ARP;
+                break;
+            default:
+                type = 0;
+        }
+        std::cout << std::hex << type << '\n';
+        this->header.payload_type = Utils::net_to_host_s(type);
+    }
+
+
+    memcpy(buffer, &this->header, sizeof(ethernet_header));
+
+}