Format beacon frames

1 files changed, 97 insertions, 0 deletions

@@ -0,0 +1,97 @@
+#include <fcntl.h>
+#include <libwifi.h>
+#include <net/if.h>
+#include <netinet/ip.h>
+#include <pcap/pcap.h>
+#include <stdio.h>
+#include <sys/ioctl.h>
+#include <unistd.h>
+
+#include "beacon_data.h"
+
+int get_interface_mac_address(int sock, const char *interface, unsigned char *output)
+{
+  struct ifreq ifr = {0};
+  strcpy(ifr.ifr_name, interface);
+
+  if (ioctl(sock, SIOCGIFHWADDR, &ifr) < 0)
+  {
+    return -1;
+  }
+
+  memcpy(output, ifr.ifr_hwaddr.sa_data, 6);
+  return 0;
+}
+
+int main(int argc, char **argv)
+{
+  int sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_IP);
+  if (sock == -1)
+  {
+    printf("Could not open socket.\n");
+    return 1;
+  }
+
+  unsigned char tx_addr[6];
+  if (get_interface_mac_address(sock, "wlo1", tx_addr))
+  {
+    printf("Could not get hardware address.\n");
+    return 2;
+  }
+
+  static unsigned char bcast[] = "\xFF\xFF\xFF\xFF\xFF\xFF";
+  struct libwifi_beacon beacon = {0};
+
+  if (libwifi_create_beacon(&beacon, bcast, tx_addr, tx_addr, "ballze", 7))
+  {
+    printf("Could not create beacon frame.\n");
+    return 3;
+  }
+
+  libwifi_remove_tag(&beacon.tags, TAG_SSID);
+
+  unsigned char payload_data[BEACON_PAYLOAD_LENGTH + 8];
+  payload_data[0] = 0x00;
+  payload_data[1] = 0x09;
+  payload_data[2] = 0xBF;
+  payload_data[3] = 0x00;
+  payload_data[4] = 0xFF;
+  payload_data[5] = 0xFF;
+  payload_data[6] = 0x00;
+  payload_data[7] = 0x00;
+  memcpy(payload_data + 8, kBeaconPayloads[0], BEACON_PAYLOAD_LENGTH);
+
+  if (libwifi_quick_add_tag(&beacon.tags, TAG_VENDOR_SPECIFIC, payload_data, BEACON_PAYLOAD_LENGTH + 8))
+  {
+    printf("Could not add beacon data tag.\n");
+    return 6;
+  }
+
+  size_t beacon_size = libwifi_get_beacon_length(&beacon);
+  unsigned char *beacon_output = (unsigned char *)malloc(beacon_size);
+  if (beacon_output == NULL)
+  {
+    printf("Could not allocate beacon output.\n");
+    return 4;
+  }
+
+  if (libwifi_dump_beacon(&beacon, beacon_output, beacon_size) < 0)
+  {
+    printf("Could not dump beacon.\n");
+    return 5;
+  }
+
+  libwifi_free_beacon(&beacon);
+
+  int fd = open("output.dat", O_WRONLY | O_CREAT, 00600);
+
+  static unsigned char radiotap[] = "\x00\x00\x08\x00\x00\x00\x00\x00";
+  write(fd, radiotap, 8);
+
+  write(fd, beacon_output, beacon_size);
+  close(fd);
+
+  free(beacon_output);
+
+  return 0;
+}