@memoraone/mcp 0.1.66 → 0.1.68

package/dist/pluginPairing.cjs

@@ -1000,12 +1000,18 @@ async function readLinuxDeviceId() {
   throw new Error("[memoraone-mcp] Unable to read Linux machine-id");
 }
 async function readWindowsDeviceId() {
-  const { stdout } = await execFileAsync("reg", [
-    "query",
-    "HKLM\\SOFTWARE\\Microsoft\\Cryptography",
-    "/v",
-    "MachineGuid"
-  ]);
+  const { stdout } = await execFileAsync(
+    "reg",
+    [
+      "query",
+      "HKLM\\SOFTWARE\\Microsoft\\Cryptography",
+      "/v",
+      "MachineGuid"
+    ],
+    {
+      windowsHide: true
+    }
+  );
   const match = stdout.match(/MachineGuid\s+REG_SZ\s+(.+)/i);
   if (!match?.[1]) {
     throw new Error("[memoraone-mcp] Unable to read Windows MachineGuid");
@@ -2310,35 +2316,74 @@ var WINDOWS_LIST_PROCESSES_PS = [
   "  [Console]::Error.WriteLine([string]$_)",
   "  exit 1",
   "}",
-  "$currentIdentity = [string][System.Security.Principal.WindowsIdentity]::GetCurrent().Name",
-  "if ([string]::IsNullOrWhiteSpace($currentIdentity)) {",
-  '  throw "Unable to determine the current Windows identity."',
-  "}",
-  "$systemDirectory = [System.Environment]::GetFolderPath([System.Environment+SpecialFolder]::System)",
-  "if ([string]::IsNullOrWhiteSpace($systemDirectory)) {",
-  '  throw "Unable to determine the Windows system directory."',
+  "$currentIdentity = [System.Security.Principal.WindowsIdentity]::GetCurrent()",
+  "$currentUserSid = [string]$currentIdentity.User.Value",
+  "if ([string]::IsNullOrWhiteSpace($currentUserSid)) {",
+  '  throw "Unable to determine the current Windows identity SID."',
   "}",
-  '$tasklistPath = [System.IO.Path]::Combine($systemDirectory, "tasklist.exe")',
-  '$tasklistArgs = @("/FI", "USERNAME eq $currentIdentity", "/FO", "CSV", "/NH")',
-  "$tasklistRows = @(& $tasklistPath $tasklistArgs)",
-  "$tasklistExitCode = $LASTEXITCODE",
-  "if ($tasklistExitCode -ne 0) {",
-  '  throw "tasklist.exe failed with exit code $tasklistExitCode."',
+  "$ownerLookupSource = @'",
+  "using System;",
+  "using System.Collections.Generic;",
+  "using System.Runtime.InteropServices;",
+  "using System.Security.Principal;",
+  "",
+  "public static class MemoraOneProcessOwner {",
+  "  private const uint ProcessQueryLimitedInformation = 0x1000;",
+  "  private const uint TokenQuery = 0x0008;",
+  "  private const int TokenUser = 1;",
+  "",
+  '  [DllImport("kernel32.dll", SetLastError = true)]',
+  "  private static extern IntPtr OpenProcess(uint access, bool inheritHandle, int processId);",
+  '  [DllImport("advapi32.dll", SetLastError = true)]',
+  "  private static extern bool OpenProcessToken(IntPtr processHandle, uint desiredAccess, out IntPtr tokenHandle);",
+  '  [DllImport("advapi32.dll", SetLastError = true)]',
+  "  private static extern bool GetTokenInformation(IntPtr tokenHandle, int tokenInformationClass, IntPtr tokenInformation, int tokenInformationLength, out int returnLength);",
+  '  [DllImport("kernel32.dll")]',
+  "  private static extern bool CloseHandle(IntPtr handle);",
+  "",
+  "  public static int[] OwnedProcessIds(int[] processIds, string currentSid) {",
+  "    var expectedSid = new SecurityIdentifier(currentSid);",
+  "    var owned = new List<int>();",
+  "    foreach (int processId in processIds) {",
+  "      IntPtr processHandle = OpenProcess(ProcessQueryLimitedInformation, false, processId);",
+  "      if (processHandle == IntPtr.Zero) continue;",
+  "      try {",
+  "        IntPtr tokenHandle;",
+  "        if (!OpenProcessToken(processHandle, TokenQuery, out tokenHandle)) continue;",
+  "        try {",
+  "          int tokenLength;",
+  "          GetTokenInformation(tokenHandle, TokenUser, IntPtr.Zero, 0, out tokenLength);",
+  "          if (tokenLength <= 0) continue;",
+  "          IntPtr tokenBuffer = Marshal.AllocHGlobal(tokenLength);",
+  "          try {",
+  "            if (!GetTokenInformation(tokenHandle, TokenUser, tokenBuffer, tokenLength, out tokenLength)) continue;",
+  "            IntPtr sidPointer = Marshal.ReadIntPtr(tokenBuffer);",
+  "            if (expectedSid.Equals(new SecurityIdentifier(sidPointer))) owned.Add(processId);",
+  "          } finally {",
+  "            Marshal.FreeHGlobal(tokenBuffer);",
+  "          }",
+  "        } finally {",
+  "          CloseHandle(tokenHandle);",
+  "        }",
+  "      } finally {",
+  "        CloseHandle(processHandle);",
+  "      }",
+  "    }",
+  "    return owned.ToArray();",
+  "  }",
   "}",
+  "'@",
+  "Add-Type -TypeDefinition $ownerLookupSource -Language CSharp",
+  "$processes = @(Get-CimInstance -ClassName Win32_Process -Property ProcessId,CommandLine)",
+  "$processIds = [int[]]@($processes | ForEach-Object { [int]$_.ProcessId })",
+  "$ownedProcessIds = [MemoraOneProcessOwner]::OwnedProcessIds($processIds, $currentUserSid)",
   '$currentUserPids = New-Object "System.Collections.Generic.HashSet[int]"',
-  "foreach ($tasklistRow in $tasklistRows) {",
-  "  if ([string]::IsNullOrWhiteSpace([string]$tasklistRow)) { continue }",
-  '  $task = $tasklistRow | ConvertFrom-Csv -Header "ImageName","ProcessId","SessionName","SessionNumber","MemoryUsage"',
-  "  $taskPid = 0",
-  "  if ($null -eq $task -or -not [int]::TryParse([string]$task.ProcessId, [ref]$taskPid) -or $taskPid -le 0) {",
-  '    throw "tasklist.exe returned an invalid process row."',
-  "  }",
-  "  $null = $currentUserPids.Add($taskPid)",
+  "foreach ($ownedProcessId in $ownedProcessIds) {",
+  "  $null = $currentUserPids.Add([int]$ownedProcessId)",
   "}",
   "if (-not $currentUserPids.Contains([int]$PID)) {",
-  '  throw "tasklist.exe results did not include the running PowerShell process."',
+  '  throw "Process token results did not include the running PowerShell process."',
   "}",
-  "$processes = @(Get-CimInstance -ClassName Win32_Process -Property ProcessId,CommandLine)",
   "foreach ($proc in $processes) {",
   "  if ($null -eq $proc.ProcessId) { continue }",
   "  $procId = [int]$proc.ProcessId",
@@ -2425,7 +2470,8 @@ async function listProcessCommandLineRows(deps = {}) {
     const { file, args } = windowsProcessListInvocation();
     const { stdout: stdout2 } = await execFileImpl(file, args, {
       maxBuffer: PROCESS_LIST_MAX_BUFFER,
-      timeout: WINDOWS_PROCESS_LIST_TIMEOUT_MS
+      timeout: WINDOWS_PROCESS_LIST_TIMEOUT_MS,
+      windowsHide: true
     });
     return parseWindowsProcessJsonLines(stdoutToString(stdout2));
   }
@@ -6431,7 +6477,6 @@ function buildMemoraoneAntigravityMcpServer(options) {
       environment,
       apiUrl: options.apiUrl ?? (environment === "local" ? DEV_API_URL : void 0)
     }),
-    MEMORAONE_IDE_TYPE: "antigravity",
     [MEMORAONE_WORKSPACE_ROOT_ENV]: path35.resolve(options.workspaceRoot)
   };
   if (environment === "local" || options.devMode) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@memoraone/mcp",
-  "version": "0.1.66",
+  "version": "0.1.68",
   "type": "module",
   "main": "dist/index.cjs",
   "exports": {