Diff: dnsname.h

Revision:

0:61831b843b44

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/dnsname.h	Sat Apr 14 17:21:11 2012 +0000
@@ -0,0 +1,51 @@
+// dnsname.h 2012/4/13
+#ifndef DNSNAME_H
+#define DNSNAME_H
+class dnsname {
+public:
+    uint8_t *buf;
+    string str;
+    dnsname(uint8_t *s) {
+        buf = s;
+    }
+    int decode(int pos) {
+        while(1) {
+            int len = buf[pos++];
+            if (len == 0x00) {
+                break;
+            }
+            if ((len&0xc0) == 0xc0) { //compress
+                int offset = (len&0x3f)<<8|buf[pos];
+                decode(offset);
+                return pos+1;
+            }
+            if (!str.empty()) {
+                str.append(".");
+            }
+            str.append((const char*)(buf+pos), len);
+            pos += len;
+        }
+        return pos;
+    }
+
+    int encode(int pos, char* s) {
+        while(*s) {  
+            char *f = strchr(s, '.');
+            if (f == NULL) {
+                int len = strlen(s);
+                buf[pos++] = len;
+                memcpy(buf+pos, s, len);
+                pos += len;
+                break;
+            }
+            int len = f - s;
+            buf[pos++] = len;
+            memcpy(buf+pos, s, len);
+            s = f+1;
+            pos += len;
+        }
+        buf[pos++] = 0x00;
+        return pos;
+    }
+};
+#endif //DNSNAME_H