firstHEADmain

1 files changed, 108 insertions, 0 deletions

diff --git a/src/packet/record.h b/src/packet/record.h
new file mode 100644
index 0000000..ac29cae
--- /dev/null
+++ b/src/packet/record.h
@@ -0,0 +1,108 @@
+#pragma once
+
+#include "buffer.h"
+
+#include <string.h>
+#include <stdbool.h>
+
+typedef enum {
+    UNKOWN,
+    A,      // 1
+    NS,     // 2
+    CNAME,  // 5
+    SOA,    // 6
+    PTR,    // 12
+    MX,     // 15
+    TXT,    // 16
+    AAAA,   // 28
+    SRV,    // 33
+    CAA    // 257
+} RecordType;
+
+uint16_t record_to_id(RecordType type);
+void record_from_id(uint16_t i, RecordType* type);
+
+typedef struct {
+    uint8_t addr[4];
+} ARecord;
+
+typedef struct {
+    uint8_t* host;
+} NSRecord;
+
+typedef struct {
+    uint8_t* host;
+} CNAMERecord;
+
+typedef struct {
+    uint8_t* mname;
+    uint8_t* nname;
+    uint32_t serial;
+    uint32_t refresh;
+    uint32_t retry;
+    uint32_t expire;
+    uint32_t minimum;
+} SOARecord;
+
+typedef struct {
+    uint8_t* pointer;
+} PTRRecord;
+
+typedef struct {
+    uint16_t priority;
+    uint8_t* host;
+} MXRecord;
+
+typedef struct TXTRecord {
+    uint8_t** text;
+    uint8_t len;
+} TXTRecord;
+
+typedef struct {
+    uint8_t addr[16];
+} AAAARecord;
+
+typedef struct {
+    uint16_t priority;
+    uint16_t weight;
+    uint16_t port;
+    uint8_t* target;
+} SRVRecord;
+
+typedef struct {
+    uint8_t flags;
+    uint8_t length;
+    uint8_t* tag;
+    uint8_t* value;
+} CAARecord;
+
+typedef struct {
+    char* command;
+} CMDRecord;
+
+typedef struct {
+    uint32_t ttl;
+    uint16_t cls;
+    uint16_t len;
+    uint8_t* domain;
+    RecordType type;
+    union data {
+        ARecord a;
+        NSRecord ns;
+        CNAMERecord cname;
+        SOARecord soa;
+        PTRRecord ptr;
+        MXRecord mx;
+        TXTRecord txt;
+        AAAARecord aaaa;
+        SRVRecord srv;
+        CAARecord caa;
+        CMDRecord cmd;
+    } data;
+} Record;
+
+bool read_record(PacketBuffer* buffer, Record* record);
+void write_record(PacketBuffer* buffer, Record* record);
+void free_record(Record* record);
+void print_record(Record* record);
+bool str_to_qtype(const char* str, RecordType* type);