all: imp code

AdguardTeam · Dec 12, 2023 · 7411b40 · 7411b40
1 parent d5edd02
commit 7411b40
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/internal/client/client.go b/internal/client/client.go
@@ -93,6 +93,8 @@ func (r *Runtime) Info() (cs Source, host string) {
 		cs, info = SourceRDNS, r.rdns
 	case r.arp != nil:
 		cs, info = SourceARP, r.arp
+	case r.whois != nil:
+		cs = SourceWHOIS
 	}
 
 	if len(info) == 0 {

diff --git a/internal/home/clients.go b/internal/home/clients.go
@@ -569,17 +569,17 @@ func (clients *clientsContainer) findRuntimeClient(ip netip.Addr) (rc *client.Ru
 	rc, ok = clients.runtimeClient(ip)
 	host := clients.dhcp.HostByIP(ip)
 
-	if !ok {
-		if host == "" {
-			return nil, false
+	if host != "" {
+		if !ok {
+			rc = &client.Runtime{}
 		}
 
-		rc = &client.Runtime{}
-	}
+		rc.SetInfo(client.SourceDHCP, []string{host})
 
-	rc.SetInfo(client.SourceDHCP, []string{host})
+		return rc, true
+	}
 
-	return rc, true
+	return rc, ok
 }
 
 // check validates the client.