| 
									
										
										
										
											2016-08-19 17:14:17 -07:00
										 |  |  | package dnsserver
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | import (
 | 
					
						
							|  |  |  | 	"fmt"
 | 
					
						
							|  |  |  | 	"net"
 | 
					
						
							|  |  |  | 	"strings"
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	"github.com/miekg/dns"
 | 
					
						
							|  |  |  | )
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | type zoneAddr struct {
 | 
					
						
							|  |  |  | 	Zone string
 | 
					
						
							|  |  |  | 	Port string
 | 
					
						
							|  |  |  | }
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | // String return z.Zone + ":" + z.Port as a string.
 | 
					
						
							|  |  |  | func (z zoneAddr) String() string { return z.Zone + ":" + z.Port }
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | // normalizeZone parses an zone string into a structured format with separate
 | 
					
						
							|  |  |  | // host, and port portions, as well as the original input string.
 | 
					
						
							| 
									
										
										
										
											2016-09-19 11:26:00 +01:00
										 |  |  | //
 | 
					
						
							|  |  |  | // TODO(miek): possibly move this to middleware/normalize.go
 | 
					
						
							| 
									
										
										
										
											2016-08-19 17:14:17 -07:00
										 |  |  | func normalizeZone(str string) (zoneAddr, error) {
 | 
					
						
							|  |  |  | 	var err error
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// separate host and port
 | 
					
						
							|  |  |  | 	host, port, err := net.SplitHostPort(str)
 | 
					
						
							|  |  |  | 	if err != nil {
 | 
					
						
							|  |  |  | 		host, port, err = net.SplitHostPort(str + ":")
 | 
					
						
							|  |  |  | 		// no error check here; return err at end of function
 | 
					
						
							|  |  |  | 	}
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	if len(host) > 255 {
 | 
					
						
							|  |  |  | 		return zoneAddr{}, fmt.Errorf("specified zone is too long: %d > 255", len(host))
 | 
					
						
							|  |  |  | 	}
 | 
					
						
							|  |  |  | 	_, d := dns.IsDomainName(host)
 | 
					
						
							|  |  |  | 	if !d {
 | 
					
						
							|  |  |  | 		return zoneAddr{}, fmt.Errorf("zone is not a valid domain name: %s", host)
 | 
					
						
							|  |  |  | 	}
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	if port == "" {
 | 
					
						
							|  |  |  | 		port = "53"
 | 
					
						
							|  |  |  | 	}
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	return zoneAddr{Zone: strings.ToLower(dns.Fqdn(host)), Port: port}, err
 | 
					
						
							|  |  |  | }
 |