1 files changed, 79 insertions, 0 deletions
diff --git a/src/packet/question.c b/src/packet/question.c
new file mode 100644
index 0000000..8159a07
--- /dev/null
+++ b/src/packet/question.c
@@ -0,0 +1,79 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <stdbool.h>
+#include <string.h>
+
+#include "question.h"
+#include "buffer.h"
+#include "record.h"
+
+void create_question(const char* domain, RecordType qtype, Question* question) {
+    uint8_t len = strlen(domain);
+    question->domain = malloc(len + 1);
+    question->qtype = qtype;
+    question->domain[0] = len;
+    question->cls = 1;
+    memcpy(question->domain + 1, domain, len);
+}
+
+bool read_question(PacketBuffer* buffer, Question* question) {
+    buffer_read_qname(buffer, &question->domain);
+    
+    uint16_t qtype_num = buffer_read_short(buffer);
+    record_from_id(qtype_num, &question->qtype);
+    question->cls = buffer_read_short(buffer);
+
+    if (question->qtype == UNKOWN) {
+        free(question->domain);
+        return false;
+    }
+
+    return true;
+}
+
+void write_question(PacketBuffer* buffer, Question* question) {
+    buffer_write_qname(buffer, question->domain);
+    
+    uint16_t id = record_to_id(question->qtype);
+    buffer_write_short(buffer, id);
+
+    buffer_write_short(buffer, question->cls);
+}
+
+void free_question(Question* question) {
+    free(question->domain);
+}
+
+static const char* class_to_str(Question* question) {
+    switch(question->cls) {
+        case 1: return "IN";
+        case 3: return "CH";
+        case 4: return "HS";
+        default: return "??";
+    }
+}
+
+static const char* qtype_to_str(Question* question) {
+    switch(question->qtype) {
+        case A: return "A";
+        case NS: return "NS";
+        case CNAME: return "CNAME";
+        case SOA: return "SOA";
+        case PTR: return "PTR";
+        case MX: return "MX";
+        case TXT: return "TXT";
+        case AAAA: return "AAAA";
+        case SRV: return "SRV";
+        case CAA: return "CAA";
+        default: return "UNKOWN";
+    }
+}
+
+void print_question(Question* question) {
+    printf("%.*s.\t%s %s\n",
+        question->domain[0],
+        question->domain + 1,
+        class_to_str(question),
+        qtype_to_str(question)
+    );
+}