summaryrefslogtreecommitdiff
path: root/internal/resolver/hints.go
blob: c7b23b4223862d04d6fa8d8f98f559eda5dde14e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
package resolver

import (
	"bufio"
	"fmt"
	"os"
	"strings"
)

type NS struct {
	Name string
	Addr string
}

func LoadHints(path string) ([]NS, error) {
	f, err := os.Open(path)
	if err != nil {
		return nil, fmt.Errorf("hints file %s not found — download from https://www.internic.net/domain/named.root: %w", path, err)
	}
	defer f.Close()

	var nsNames []string
	glue := make(map[string]string)
	sc := bufio.NewScanner(f)
	for sc.Scan() {
		fields := strings.Fields(sc.Text())
		if len(fields) < 4 || fields[0] == ";" {
			continue
		}
		if fields[0] == "." && fields[2] == "NS" {
			nsNames = append(nsNames, fields[3])
		}
		if fields[2] == "A" {
			glue[fields[0]] = fields[3]
		}
	}
	if err := sc.Err(); err != nil {
		return nil, err
	}

	var hints []NS
	for _, name := range nsNames {
		hints = append(hints, NS{Name: name, Addr: glue[name]})
	}
	if len(hints) == 0 {
		return nil, fmt.Errorf("no root hints parsed from %s", path)
	}
	return hints, nil
}