sweetspot-remote-agent 1.8.6 → 1.8.7

package/mcp-server.js

@@ -273,20 +273,34 @@ if (!isStdio && !noRegister && !regName) {
   process.exit(1);
 }
 
-// ── 토큰 자동 생성/저장 ──
+// ── RSA 키페어 자동 생성/저장 ──
 import fs from "fs";
 
-const tokenFile = `${os.homedir()}/.sweetspot-token`;
+const keypairFile = `${os.homedir()}/.sweetspot-keypair.json`;
 
-function loadOrCreateToken() {
-  if (fs.existsSync(tokenFile)) return fs.readFileSync(tokenFile, "utf-8").trim();
-  const token = crypto.randomUUID();
-  fs.writeFileSync(tokenFile, token, "utf-8");
-  return token;
+function loadOrCreateKeypair() {
+  if (fs.existsSync(keypairFile)) {
+    const { privateKey, publicKey } = JSON.parse(fs.readFileSync(keypairFile, "utf-8"));
+    return { privateKey, publicKey };
+  }
+  const { privateKey, publicKey } = crypto.generateKeyPairSync("rsa", {
+    modulusLength: 2048,
+    publicKeyEncoding:  { type: "spki",  format: "pem" },
+    privateKeyEncoding: { type: "pkcs8", format: "pem" },
+  });
+  fs.writeFileSync(keypairFile, JSON.stringify({ privateKey, publicKey }), { mode: 0o600 });
+  console.error(`[remote-agent] RSA 키페어 생성 완료: ${keypairFile}`);
+  return { privateKey, publicKey };
+}
+
+function signChallenge(privateKey, name, timestamp) {
+  const payload = `${name}:${timestamp}`;
+  return crypto.sign("sha256", Buffer.from(payload), { key: privateKey, padding: crypto.constants.RSA_PKCS1_PSS_PADDING })
+    .toString("base64");
 }
 
 // ── WebSocket 클라이언트 (GCP 역방향 연결) ──
-async function connectWs(name, token) {
+async function connectWs(name, keypair) {
   const { default: WebSocket } = await import("ws");
   const wsUrl = BOT_SERVER.replace(/^http/, "ws") + "/agent";
   let alive = true;
@@ -296,7 +310,15 @@ async function connectWs(name, token) {
     const ws = new WebSocket(wsUrl);
 
     ws.on("open", () => {
-      ws.send(JSON.stringify({ type: "register", name, token }));
+      const timestamp = Date.now();
+      const signature = signChallenge(keypair.privateKey, name, timestamp);
+      ws.send(JSON.stringify({
+        type: "register",
+        name,
+        publicKey: keypair.publicKey,
+        timestamp,
+        signature,
+      }));
       console.error(`[remote-agent] GCP 서버에 연결 중... (${name})`);
     });
 
@@ -432,9 +454,9 @@ async function main() {
     console.error("[remote-agent] MCP 서버 시작 (stdio)");
   } else {
     // WebSocket 모드: GCP 역방향 연결 (ngrok 불필요)
-    const token = loadOrCreateToken();
-    console.error(`[remote-agent] 토큰 파일: ${tokenFile}`);
-    await connectWs(regName, token);
+    const keypair = loadOrCreateKeypair();
+    console.error(`[remote-agent] RSA 키페어 파일: ${keypairFile}`);
+    await connectWs(regName, keypair);
   }
 }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "sweetspot-remote-agent",
-  "version": "1.8.6",
+  "version": "1.8.7",
   "description": "Sweetspot 원격 제어 MCP 서버 — 스크린샷, 마우스/키보드, 앱 제어, 파일 탐색, 셸 실행",
   "type": "module",
   "main": "mcp-server.js",