del: log

yaklang · Nov 22, 2024 · 50b43d9 · 50b43d9
1 parent 6ec049c
commit 50b43d9
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/app/renderer/src/main/src/components/HTTPFlowTable/HTTPFlowTable.tsx b/app/renderer/src/main/src/components/HTTPFlowTable/HTTPFlowTable.tsx
@@ -1141,7 +1141,6 @@ export const HTTPFlowTable = React.memo<HTTPFlowTableProp>((props) => {
         }
 
         updateQueryParams(realQuery)
-        // console.log('realQuery', realQuery);
 
         ipcRenderer
             .invoke("QueryHTTPFlows", realQuery)

diff --git a/app/renderer/src/main/src/pages/mitm/MITMServerHijacking/MITMLog.tsx b/app/renderer/src/main/src/pages/mitm/MITMServerHijacking/MITMLog.tsx
@@ -104,7 +104,22 @@ export const MITMLogHeardExtra: React.FC<MITMLogHeardExtraProps> = React.memo((p
                             .filter((name: string) => name)
                             .map((name: string) => {
                                 const lowerName = name.toLocaleLowerCase()
-                                const icon = Object.keys(iconProcessMap).find((key) => lowerName.startsWith(key))
+                                const lastDotIndex = lowerName.lastIndexOf(".")
+                                const icon = Object.keys(iconProcessMap).find((key) => {
+                                    if (key.startsWith("docker") && lowerName.startsWith("docker")) {
+                                        return true
+                                    } else if (key.startsWith("jdk") && lowerName.startsWith("jdk")) {
+                                        return true
+                                    } else if (key.startsWith("java") && lowerName.startsWith("java")) {
+                                        return true
+                                    } else if (key.startsWith("vmware") && lowerName.startsWith("vmware")) {
+                                        return true
+                                    } else if (lastDotIndex !== -1) {
+                                        return key.toLocaleLowerCase() === lowerName.slice(0, lastDotIndex)
+                                    } else {
+                                        return false
+                                    }
+                                })
                                 return {process: name, icon: icon ? iconProcessMap[icon] : undefined}
                             })
                         setProcessList(processArr)