From d1424a3e175ed0c07f3af33a089548f5bc1e3ce9 Mon Sep 17 00:00:00 2001
From: bit4 <bit4woo@163.com>
Date: Sun, 30 Sep 2018 10:35:19 +0800
Subject: [PATCH] update

---
 src/burp/BurpExtender.java | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/burp/BurpExtender.java b/src/burp/BurpExtender.java
index bf2cae4..981725e 100644
--- a/src/burp/BurpExtender.java
+++ b/src/burp/BurpExtender.java
@@ -115,7 +115,7 @@ public Map<String, Set<String>> search(String subdomainof, String domainlike){
 					//stdout.println(shortURL);
 					//stdout.println(Host);
 					
-					if (Host.endsWith("."+subdomainof)){
+					if (Host.endsWith("."+subdomainof)||Host.equalsIgnoreCase(subdomainof)){
 						subdomainofset.add(Host);
 						//stdout.println(subdomainofset);
 						//get SANs info to get related domain, only when the [subdomain] is using https.
@@ -126,7 +126,7 @@ public Map<String, Set<String>> search(String subdomainof, String domainlike){
 					}
 					
 					
-					if (!domainlike.equals("") && Host.contains(domainlike) && !Host.equalsIgnoreCase(subdomainof)){
+					if (!domainlike.equals("") && Host.contains(domainlike)){
 						domainlikeset.add(Host);
 						if(protocol.equalsIgnoreCase("https")) {
 							httpsURLs.add(shortURL);
@@ -197,9 +197,9 @@ public Map<String, Set<String>> search(String subdomainof, String domainlike){
 			    //对 SANs的结果再做一次分类。
 			    for (String item:tmpRelatedDomainSet) {
 			    	if (item.contains(".")&&!item.endsWith(".")&&!item.startsWith(".")) {
-				    	if (item.endsWith("."+subdomainof)){
+				    	if (item.endsWith("."+subdomainof) || item.equalsIgnoreCase(subdomainof)){
 							subdomainofset.add(item);
-						}else if (!domainlike.equals("") && item.contains(domainlike) && !item.equalsIgnoreCase(subdomainof)){
+						}else if (!domainlike.equals("") && item.contains(domainlike)){
 							domainlikeset.add(item);
 						}else {
 							relatedDomainSet.add(item);