1 files changed, 68 insertions, 0 deletions
diff --git a/utils/other_utils.go b/utils/other_utils.go
new file mode 100644
index 0000000..a25dcdc
--- /dev/null
+++ b/utils/other_utils.go
@@ -0,0 +1,68 @@
+package utils
+
+import (
+	"encoding/json"
+	"fmt"
+	"net"
+	"net/http"
+	"strconv"
+	"strings"
+)
+
+type ResolverBehavior struct {
+	Dnssec        bool  `json:"dnssec"`
+	QnameEncode   bool  `json:"0x20"`
+	QueryMerge    int   `json:"merge_dup"`
+	MaxNsDepth    int   `json:"max_ns_depth"`
+	MaxCnameDepth int   `json:"max_cname_depth"`
+	RetryLimit    int   `json:"retry_limit"`
+	FetchLimit    int   `json:"fetch_limit"`
+	TimeoutStart  int64 `json:"timeout_start"`
+	TimeoutEnd    int64 `json:"timeout_end"`
+}
+
+func IsValidIP(ip string) bool {
+	res := net.ParseIP(ip)
+	return res != nil
+}
+
+func IPv4ToInt(ip string) (uint32, error) {
+	if !IsValidIP(ip) {
+		return 0, fmt.Errorf("invalid IP address: %s", ip)
+	}
+	labels := strings.Split(ip, ".")
+	var result uint32
+	for _, label := range labels {
+		label_val, _ := strconv.Atoi(label)
+		result = (result << 8) | uint32(label_val)
+	}
+	return result, nil
+}
+
+func GetResult(addr string, token int) (*ResolverBehavior, error) {
+	data := new(ResolverBehavior)
+	url := "http://" + addr + "/results/" + strconv.Itoa(token)
+	response, err := http.Get(url)
+	if err != nil {
+		return data, err
+	}
+	defer response.Body.Close()
+
+	if response.StatusCode != http.StatusOK {
+		return data, fmt.Errorf("wrong HTTP status code : %v", response.StatusCode)
+	}
+
+	err = json.NewDecoder(response.Body).Decode(data)
+	if err != nil {
+		return data, err
+	}
+	return data, nil
+}
+
+func OutputJson(data interface{}) (string, error) {
+	json_byte, err := json.Marshal(data)
+	if err != nil {
+		return "", err
+	}
+	return string(json_byte), nil
+}