Support for dynamically addition of DNS records

Signed-off-by: Balaji Vijayakumar <[email protected]>

Author: balajiv113

pkg/client/client.go

@@ -80,3 +80,40 @@ func (c *Client) Unexpose(req *types.UnexposeRequest) error {
 	}
 	return nil
 }
+
+func (c *Client) ListDNS() ([]types.Zone, error) {
+	res, err := c.client.Get(fmt.Sprintf("%s%s", c.base, "/services/dns/all"))
+	if err != nil {
+		return nil, err
+	}
+	defer res.Body.Close()
+	if res.StatusCode != http.StatusOK {
+		return nil, fmt.Errorf("unexpected status: %d", res.StatusCode)
+	}
+	dec := json.NewDecoder(res.Body)
+	var dnsZone []types.Zone
+	if err := dec.Decode(&dnsZone); err != nil {
+		return nil, err
+	}
+	return dnsZone, nil
+}
+
+func (c *Client) AddDNS(req *types.Zone) error {
+	bin, err := json.Marshal(req)
+	if err != nil {
+		return err
+	}
+	res, err := c.client.Post(fmt.Sprintf("%s%s", c.base, "/services/dns/add"), "application/json", bytes.NewReader(bin))
+	if err != nil {
+		return err
+	}
+	defer res.Body.Close()
+	if res.StatusCode != http.StatusOK {
+		err, readErr := io.ReadAll(res.Body)
+		if readErr != nil {
+			return fmt.Errorf("error while reading error message: %v", readErr)
+		}
+		return errors.New(strings.TrimSpace(string(err)))
+	}
+	return nil
+}

pkg/services/dns/dns.go

@@ -2,8 +2,10 @@ package dns
 
 import (
 	"context"
+	"encoding/json"
 	"fmt"
 	"net"
+	"net/http"
 	"strings"
 	"sync"
 
@@ -134,3 +136,30 @@ func (s *Server) Serve() error {
 	}
 	return srv.ActivateAndServe()
 }
+
+func (s *Server) Mux() http.Handler {
+	mux := http.NewServeMux()
+	mux.HandleFunc("/all", func(w http.ResponseWriter, r *http.Request) {
+		s.handler.zonesLock.RLock()
+		_ = json.NewEncoder(w).Encode(s.handler.zones)
+		s.handler.zonesLock.RUnlock()
+	})
+
+	mux.HandleFunc("/add", func(w http.ResponseWriter, r *http.Request) {
+		if r.Method != http.MethodPost {
+			http.Error(w, "post only", http.StatusBadRequest)
+			return
+		}
+		var req types.Zone
+		if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
+			http.Error(w, err.Error(), http.StatusBadRequest)
+			return
+		}
+
+		s.handler.zonesLock.Lock()
+		s.handler.zones = append([]types.Zone{req}, s.handler.zones...)
+		s.handler.zonesLock.Unlock()
+		w.WriteHeader(http.StatusOK)
+	})
+	return mux
+}

pkg/virtualnetwork/services.go

@@ -29,7 +29,8 @@ func addServices(configuration *types.Configuration, s *stack.Stack, ipPool *tap
 	udpForwarder := forwarder.UDP(s, translation, &natLock)
 	s.SetTransportProtocolHandler(udp.ProtocolNumber, udpForwarder.HandlePacket)
 
-	if err := dnsServer(configuration, s); err != nil {
+	dnsMux, err := dnsServer(configuration, s)
+	if err != nil {
 		return nil, err
 	}
 
@@ -45,6 +46,7 @@ func addServices(configuration *types.Configuration, s *stack.Stack, ipPool *tap
 	mux := http.NewServeMux()
 	mux.Handle("/forwarder/", http.StripPrefix("/forwarder", forwarderMux))
 	mux.Handle("/dhcp/", http.StripPrefix("/dhcp", dhcpMux))
+	mux.Handle("/dns/", http.StripPrefix("/dns", dnsMux))
 	return mux, nil
 }
 
@@ -56,24 +58,24 @@ func parseNATTable(configuration *types.Configuration) map[tcpip.Address]tcpip.A
 	return translation
 }
 
-func dnsServer(configuration *types.Configuration, s *stack.Stack) error {
+func dnsServer(configuration *types.Configuration, s *stack.Stack) (http.Handler, error) {
 	udpConn, err := gonet.DialUDP(s, &tcpip.FullAddress{
 		NIC:  1,
 		Addr: tcpip.Address(net.ParseIP(configuration.GatewayIP).To4()),
 		Port: uint16(53),
 	}, nil, ipv4.ProtocolNumber)
 	if err != nil {
-		return err
+		return nil, err
 	}
 	server, err := dns.New(udpConn, configuration.DNS)
 	if err != nil {
-		return err
+		return nil, err
 	}
 
 	go func() {
 		log.Error(server.Serve())
 	}()
-	return nil
+	return server.Mux(), nil
 }
 
 func dhcpServer(configuration *types.Configuration, s *stack.Stack, ipPool *tap.IPPool) (http.Handler, error) {

test/basic_test.go

@@ -1,6 +1,12 @@
 package e2e
 
 import (
+	"context"
+	"net"
+	"net/http"
+
+	gvproxyclient "github.com/containers/gvisor-tap-vsock/pkg/client"
+	"github.com/containers/gvisor-tap-vsock/pkg/types"
 	. "github.com/onsi/ginkgo"
 	. "github.com/onsi/gomega"
 )
@@ -58,4 +64,60 @@ var _ = Describe("dns", func() {
 		Expect(err).ShouldNot(HaveOccurred())
 		Expect(string(out)).To(ContainSubstring("Address: 192.168.127.254"))
 	})
+
+	It("should resolve dynamically added dns entry test.dynamic.internal", func() {
+		client := gvproxyclient.New(&http.Client{
+			Transport: &http.Transport{
+				DialContext: func(ctx context.Context, network, addr string) (net.Conn, error) {
+					return net.Dial("unix", sock)
+				},
+			},
+		}, "http://base")
+		err := client.AddDNS(&types.Zone{
+			Name: "dynamic.internal.",
+			Records: []types.Record{
+				{
+					Name: "test",
+					IP:   net.ParseIP("192.168.127.254"),
+				},
+			},
+		})
+		Expect(err).ShouldNot(HaveOccurred())
+
+		out, err := sshExec("nslookup test.dynamic.internal")
+
+		Expect(err).ShouldNot(HaveOccurred())
+		Expect(string(out)).To(ContainSubstring("Address: 192.168.127.254"))
+	})
+
+	It("should resolve recently added dns entry test.dynamic.internal", func() {
+		client := gvproxyclient.New(&http.Client{
+			Transport: &http.Transport{
+				DialContext: func(ctx context.Context, network, addr string) (net.Conn, error) {
+					return net.Dial("unix", sock)
+				},
+			},
+		}, "http://base")
+		err := client.AddDNS(&types.Zone{
+			Name: "dynamic.internal.",
+			Records: []types.Record{
+				{
+					Name: "test",
+					IP:   net.ParseIP("192.168.127.254"),
+				},
+			},
+		})
+		Expect(err).ShouldNot(HaveOccurred())
+
+		err = client.AddDNS(&types.Zone{
+			Name:      "dynamic.internal.",
+			DefaultIP: net.ParseIP("192.168.127.253"),
+		})
+		Expect(err).ShouldNot(HaveOccurred())
+
+		out, err := sshExec("nslookup test.dynamic.internal")
+
+		Expect(err).ShouldNot(HaveOccurred())
+		Expect(string(out)).To(ContainSubstring("Address: 192.168.127.253"))
+	})
 })