update

Author: bit4woo

src/base/FindUrlAction.java

@@ -29,8 +29,6 @@
 import burp.IHttpRequestResponse;
 import burp.IHttpService;
 import burp.threadRequester;
-import org.apache.tika.mime.MediaType;
-import org.apache.tika.mime.MimeTypes;
 
 public class FindUrlAction implements ActionListener {
 	private IContextMenuInvocation invocation;

src/messageTab/Info/InfoTable.java

@@ -197,7 +197,9 @@ public List<String> getAllUrlsOfTarget() {
 	public String choseBaseUrlToRequest(List<String> allUrlsOfTarget) {
 		String referUrl = getReferUrl();
 		String currentUrl = getFullUrl();
-		return FindUrlAction.choseAndEditBaseURL(allUrlsOfTarget,referUrl,currentUrl);
+		String baseUrl = FindUrlAction.choseAndEditBaseURL(allUrlsOfTarget,referUrl,currentUrl);
+		((InfoPanelHeadPanel)(InfoPanel.getHeadPanel())).setBaseUrl(baseUrl.trim());
+		return baseUrl;
 	}
 	/**
 	 * 从已有记录中直接获取【构建URL所需要的基准URL（BaseURL）】，或者从数据包中查找并选择 
@@ -213,6 +215,7 @@ public String getOrFindBaseUrl() {
 				FindUrlAction.httpServiceBaseUrlMap.put(originUrl, baseurl);
 			}
 		}
+		((InfoPanelHeadPanel)(InfoPanel.getHeadPanel())).setBaseUrl(baseurl.trim());
 		return baseurl;
 	}
 

src/messageTab/Info/InfoTableMenu.java

@@ -45,7 +45,6 @@ public void actionPerformed(ActionEvent actionEvent) {
 
 				if (StringUtils.isNotEmpty(originUrl) && StringUtils.isNotEmpty(baseurl)) {
 					FindUrlAction.httpServiceBaseUrlMap.put(originUrl, baseurl);
-					((InfoPanelHeadPanel)(InfoPanel.getHeadPanel())).setBaseUrl(baseurl);
 				}
 			}
 		});
@@ -61,9 +60,7 @@ public void actionPerformed(ActionEvent actionEvent) {
 
 				if (StringUtils.isNotEmpty(originUrl) && StringUtils.isNotEmpty(baseurl)) {
 					FindUrlAction.httpServiceBaseUrlMap.put(originUrl, baseurl);
-					((InfoPanelHeadPanel)(InfoPanel.getHeadPanel())).setBaseUrl(baseurl);
 				}
-				
 			}
 		});