x

internal/wechat/key/darwin/glance/vmmap.go

@@ -0,0 +1,158 @@
+package glance
+
+import (
+	"bufio"
+	"fmt"
+	"os/exec"
+	"regexp"
+	"strconv"
+	"strings"
+)
+
+const (
+	FilterRegionType = "MALLOC_NANO"
+	FilterSHRMOD     = "SM=PRV"
+	CommandVmmap     = "vmmap"
+)
+
+type MemRegion struct {
+	RegionType   string
+	Start        uint64
+	End          uint64
+	VSize        uint64 // Size in bytes
+	RSDNT        uint64 // Resident memory size in bytes (new field)
+	SHRMOD       string
+	Permissions  string
+	RegionDetail string
+}
+
+func GetVmmap(pid uint32) ([]MemRegion, error) {
+	// Execute vmmap command
+	cmd := exec.Command(CommandVmmap, "-wide", fmt.Sprintf("%d", pid))
+	output, err := cmd.CombinedOutput()
+	if err != nil {
+		return nil, fmt.Errorf("error executing vmmap command: %w", err)
+	}
+
+	// Parse the output using the existing LoadVmmap function
+	return LoadVmmap(string(output))
+}
+
+func LoadVmmap(output string) ([]MemRegion, error) {
+	var regions []MemRegion
+
+	scanner := bufio.NewScanner(strings.NewReader(output))
+
+	// Skip lines until we find the header
+	foundHeader := false
+	for scanner.Scan() {
+		line := scanner.Text()
+		if strings.HasPrefix(line, "==== Writable regions for") {
+			foundHeader = true
+			// Skip the column headers line
+			scanner.Scan()
+			break
+		}
+	}
+
+	if !foundHeader {
+		return nil, nil // No vmmap data found
+	}
+
+	// Regular expression to parse the vmmap output lines
+	// Format: REGION TYPE                    START - END         [ VSIZE  RSDNT  DIRTY   SWAP] PRT/MAX SHRMOD PURGE    REGION DETAIL
+	// Updated regex to capture RSDNT value (second value in brackets)
+	re := regexp.MustCompile(`^(\S+)\s+([0-9a-f]+)-([0-9a-f]+)\s+\[\s*(\S+)\s+(\S+)(?:\s+\S+){2}\]\s+(\S+)\s+(\S+)(?:\s+\S+)?\s+(.*)$`)
+
+	// Parse each line
+	for scanner.Scan() {
+		line := scanner.Text()
+		if line == "" {
+			continue
+		}
+
+		matches := re.FindStringSubmatch(line)
+		if len(matches) >= 9 { // Updated to check for at least 9 matches
+
+			// Parse start and end addresses
+			start, _ := strconv.ParseUint(matches[2], 16, 64)
+			end, _ := strconv.ParseUint(matches[3], 16, 64)
+
+			// Parse VSize as numeric value
+			vsize := parseSize(matches[4])
+
+			// Parse RSDNT as numeric value (new)
+			rsdnt := parseSize(matches[5])
+
+			region := MemRegion{
+				RegionType:   strings.TrimSpace(matches[1]),
+				Start:        start,
+				End:          end,
+				VSize:        vsize,
+				RSDNT:        rsdnt,                         // Add the new RSDNT field
+				Permissions:  matches[6],                    // Shifted index
+				SHRMOD:       matches[7],                    // Shifted index
+				RegionDetail: strings.TrimSpace(matches[8]), // Shifted index
+			}
+
+			regions = append(regions, region)
+		}
+	}
+
+	return regions, nil
+}
+
+func MemRegionsFilter(regions []MemRegion) []MemRegion {
+	var filteredRegions []MemRegion
+	for _, region := range regions {
+		if region.RegionType == FilterRegionType {
+			filteredRegions = append(filteredRegions, region)
+		}
+	}
+	return filteredRegions
+}
+
+// parseSize converts size strings like "5616K" or "128.0M" to bytes (uint64)
+func parseSize(sizeStr string) uint64 {
+	// Remove any whitespace
+	sizeStr = strings.TrimSpace(sizeStr)
+
+	// Define multipliers for different units
+	multipliers := map[string]uint64{
+		"B":  1,
+		"K":  1024,
+		"KB": 1024,
+		"M":  1024 * 1024,
+		"MB": 1024 * 1024,
+		"G":  1024 * 1024 * 1024,
+		"GB": 1024 * 1024 * 1024,
+	}
+
+	// Regular expression to match numbers with optional decimal point and unit
+	// This will match formats like: "5616K", "128.0M", "1.5G", etc.
+	re := regexp.MustCompile(`^(\d+(?:\.\d+)?)([KMGB]+)?$`)
+	matches := re.FindStringSubmatch(sizeStr)
+
+	if len(matches) < 2 {
+		return 0 // No match found
+	}
+
+	// Parse the numeric part (which may include a decimal point)
+	numStr := matches[1]
+	numVal, err := strconv.ParseFloat(numStr, 64)
+	if err != nil {
+		return 0
+	}
+
+	// Determine the multiplier based on the unit
+	multiplier := uint64(1) // Default if no unit specified
+	if len(matches) >= 3 && matches[2] != "" {
+		unit := matches[2]
+		if m, ok := multipliers[unit]; ok {
+			multiplier = m
+		}
+	}
+
+	// Calculate final size in bytes (rounding to nearest integer)
+	return uint64(numVal*float64(multiplier) + 0.5)
+}