From 1b527fd494e8b2ad713ea0afcf9af485d452d61c Mon Sep 17 00:00:00 2001
From: wwqgtxx <wwqgtxx@gmail.com>
Date: Fri, 8 Dec 2023 08:55:45 +0800
Subject: [PATCH] chore: windows process will return DOS format instead of NT
 format

---
 component/process/process_windows.go | 32 ++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/component/process/process_windows.go b/component/process/process_windows.go
index d43c78c6..820493ec 100644
--- a/component/process/process_windows.go
+++ b/component/process/process_windows.go
@@ -3,6 +3,8 @@ package process
 import (
 	"fmt"
 	"net/netip"
+	"path/filepath"
+	"strings"
 	"sync"
 	"syscall"
 	"unsafe"
@@ -101,6 +103,10 @@ func findProcessName(network string, ip netip.Addr, srcPort int) (uint32, string
 		return 0, "", err
 	}
 	pp, err := getExecPathFromPID(pid)
+	if err != nil {
+		return 0, "", err
+	}
+	pp, err = convertDOSPath(pp)
 	return 0, pp, err
 }
 
@@ -227,3 +233,29 @@ func getExecPathFromPID(pid uint32) (string, error) {
 	}
 	return syscall.UTF16ToString(buf[:size]), nil
 }
+
+// modify from https://github.com/shirou/gopsutil/blob/9deadc99147d80f732af3a59e624af73d0143891/internal/common/common_windows.go#L220-L241
+// Convert paths using native DOS format like:
+//
+//	"\Device\HarddiskVolume1\Windows\systemew\file.txt"
+//
+// into:
+//
+//	"C:\Windows\systemew\file.txt"
+func convertDOSPath(p string) (string, error) {
+	rawDrive := strings.Join(strings.Split(p, `\`)[:3], `\`)
+
+	for d := 'A'; d <= 'Z'; d++ {
+		szDeviceName := string(d) + ":"
+		deviceName, err := syscall.UTF16PtrFromString(szDeviceName)
+		if err != nil {
+			return "", err
+		}
+		szTarget := make([]uint16, 512)
+		n, err := windows.QueryDosDevice(deviceName, &szTarget[0], uint32(len(szTarget)))
+		if err == nil && windows.UTF16ToString(szTarget[:n]) == rawDrive {
+			return filepath.Join(szDeviceName, p[len(rawDrive):]), nil
+		}
+	}
+	return p, nil
+}