add related domains to neighbors

Author: lanrat

graph/graph.go

@@ -69,24 +69,16 @@ func (graph *CertGraph) GetDomain(domain string) (*DomainNode, bool) {
 // cdn will include CDN certs as well
 func (graph *CertGraph) GetDomainNeighbors(domain string, cdn bool) []string {
 	neighbors := make(map[string]bool)
+	// TODO return
 
 	//domain = directDomain(domain)
 	node, ok := graph.domains.Load(domain)
 	if ok {
 		domainNode := node.(*DomainNode)
-		// visited cert neighbors
-		/*node, ok := graph.certs.Load(domainNode.VisitedCert)
-		if ok {
-			certNode := node.(*CertNode)
-			if !cdn && certNode.CDNCert() {
-				//v(domain, "-> CDN CERT")
-			} else {
-				for _, neighbor := range certNode.Domains {
-					neighbors[neighbor] = true
-					//v(domain, "- CERT ->", neighbor)
-				}
-			}
-		}*/
+		// related cert neighbors
+		for relatedDomain := range domainNode.RelatedDomains {
+			neighbors[relatedDomain] = true
+		}
 
 		// Cert neighbors
 		for _, fp := range domainNode.Certs {

graph/misc.go

@@ -1,12 +1,10 @@
 package graph
 
+import (
+	"strings"
+)
+
 // given a domain returns the non-wildcard version of that domain
 func directDomain(domain string) string {
-	if len(domain) < 3 {
-		return domain
-	}
-	if domain[0:2] == "*." {
-		domain = domain[2:]
-	}
-	return domain
+	return strings.TrimPrefix(domain, "*.")
 }