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> 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> 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> 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);