feat: 为反检测做准备

packages/napcat-core/packet/client/nativeClient.ts

@@ -1,11 +1,8 @@
-import path, { dirname } from 'path';
-import { fileURLToPath } from 'url';
-import fs from 'fs';
-import { constants } from 'node:os';
 import { LogStack } from '@/napcat-core/packet/context/clientContext';
 import { NapCoreContext } from '@/napcat-core/packet/context/napCoreContext';
 import { PacketLogger } from '@/napcat-core/packet/context/loggerContext';
 import { OidbPacket, PacketBuf } from '@/napcat-core/packet/transformer/base';
+import { Napi2NativeLoader } from '@/napcat-core/packet/handler/napi2nativeLoader';
 export interface RecvPacket {
   type: string, // 仅recv
   data: RecvPacketData;
@@ -17,48 +14,36 @@ export interface RecvPacketData {
   data: Buffer;
 }
 
-// 0 send 1 recv
-export interface NativePacketExportType {
-  initHook?: (send: string, recv: string) => boolean;
-}
-
 export class NativePacketClient {
   protected readonly napcore: NapCoreContext;
   protected readonly logger: PacketLogger;
   protected readonly cb = new Map<string, (json: RecvPacketData) => Promise<any> | any>(); // hash-type callback
+  protected readonly napi2nativeLoader: Napi2NativeLoader;
   logStack: LogStack;
   available: boolean = false;
-  private readonly supportedPlatforms = ['win32.x64', 'linux.x64', 'linux.arm64', 'darwin.x64', 'darwin.arm64'];
-  private readonly MoeHooExport: { exports: NativePacketExportType; } = { exports: {} };
 
-  constructor (napCore: NapCoreContext, logger: PacketLogger, logStack: LogStack) {
+  constructor (napCore: NapCoreContext, logger: PacketLogger, logStack: LogStack, napi2nativeLoader: Napi2NativeLoader) {
     this.napcore = napCore;
     this.logger = logger;
     this.logStack = logStack;
+    this.napi2nativeLoader = napi2nativeLoader;
   }
 
   check (): boolean {
-    const platform = process.platform + '.' + process.arch;
-    if (!this.supportedPlatforms.includes(platform)) {
-      this.logStack.pushLogWarn(`NativePacketClient: 不支持的平台: ${platform}`);
-      return false;
-    }
-    const moehoo_path = path.join(dirname(fileURLToPath(import.meta.url)), './native/napi2native/napi2native.' + platform + '.node');
-    if (!fs.existsSync(moehoo_path)) {
-      this.logStack.pushLogWarn(`NativePacketClient: 缺失运行时文件: ${moehoo_path}`);
+    if (!this.napi2nativeLoader.loaded) {
+      this.logStack.pushLogWarn('NativePacketClient: Napi2NativeLoader 未成功加载');
       return false;
     }
     return true;
   }
 
   async init (_pid: number, recv: string, send: string): Promise<void> {
-    const platform = process.platform + '.' + process.arch;
     const isNewQQ = this.napcore.basicInfo.requireMinNTQQBuild('40824');
     if (isNewQQ) {
-      const moehoo_path = path.join(dirname(fileURLToPath(import.meta.url)), './native/napi2native/napi2native.' + platform + '.node');
-      process.dlopen(this.MoeHooExport, moehoo_path, constants.dlopen.RTLD_LAZY);
-      this.MoeHooExport?.exports.initHook?.(send, recv);
-      this.available = true;
+      const success = this.napi2nativeLoader.initHook(send, recv);
+      if (success) {
+        this.available = true;
+      }
     }
   }
 

packages/napcat-core/packet/context/clientContext.ts

@@ -2,6 +2,7 @@ import { NativePacketClient } from '@/napcat-core/packet/client/nativeClient';
 import { OidbPacket } from '@/napcat-core/packet/transformer/base';
 import { PacketLogger } from '@/napcat-core/packet/context/loggerContext';
 import { NapCoreContext } from '@/napcat-core/packet/context/napCoreContext';
+import { Napi2NativeLoader } from '@/napcat-core/packet/handler/napi2nativeLoader';
 
 export class LogStack {
   private stack: string[] = [];
@@ -43,12 +44,14 @@ export class PacketClientContext {
   private readonly napCore: NapCoreContext;
   private readonly logger: PacketLogger;
   private readonly logStack: LogStack;
+  private readonly napi2nativeLoader: Napi2NativeLoader;
   private readonly _client: NativePacketClient;
 
-  constructor (napCore: NapCoreContext, logger: PacketLogger) {
+  constructor (napCore: NapCoreContext, logger: PacketLogger, napi2nativeLoader: Napi2NativeLoader) {
     this.napCore = napCore;
     this.logger = logger;
     this.logStack = new LogStack(logger);
+    this.napi2nativeLoader = napi2nativeLoader;
     this._client = this.newClient();
   }
 
@@ -71,7 +74,7 @@ export class PacketClientContext {
 
   private newClient (): NativePacketClient {
     this.logger.info('使用 NativePacketClient 作为后端');
-    const client = new NativePacketClient(this.napCore, this.logger, this.logStack);
+    const client = new NativePacketClient(this.napCore, this.logger, this.logStack, this.napi2nativeLoader);
     if (!client.check()) {
       throw new Error('[Core] [Packet] NativePacketClient 不可用，NapCat.Packet将不会加载！');
     }

packages/napcat-core/packet/context/napCoreContext.ts

@@ -34,5 +34,9 @@ export class NapCoreContext {
     return this.core.configLoader.configData;
   }
 
+  get napi2nativeLoader () {
+    return this.core.context.napi2nativeLoader;
+  }
+
   sendSsoCmdReqByContend = (cmd: string, data: Buffer) => this.core.context.session.getMsgService().sendSsoCmdReqByContend(cmd, data);
 }

packages/napcat-core/packet/context/packetContext.ts

@@ -18,7 +18,7 @@ export class PacketContext {
     this.msgConverter = new PacketMsgConverter();
     this.napcore = new NapCoreContext(core);
     this.logger = new PacketLogger(this.napcore);
-    this.client = new PacketClientContext(this.napcore, this.logger);
+    this.client = new PacketClientContext(this.napcore, this.logger, this.napcore.napi2nativeLoader);
     this.highway = new PacketHighwayContext(this.napcore, this.logger, this.client);
     this.operation = new PacketOperationContext(this);
   }

packages/napcat-core/packet/handler/napi2nativeLoader.ts

@@ -0,0 +1,79 @@
+import path, { dirname } from 'path';
+import { fileURLToPath } from 'url';
+import fs from 'fs';
+import { constants } from 'node:os';
+import { LogWrapper } from '../../helper/log';
+
+export interface Napi2NativeExportType {
+  initHook?: (send: string, recv: string) => boolean;
+}
+
+export class Napi2NativeLoader {
+  private readonly supportedPlatforms = ['win32.x64', 'linux.x64', 'linux.arm64', 'darwin.x64', 'darwin.arm64'];
+  private readonly exports: { exports: Napi2NativeExportType; } = { exports: {} };
+  protected readonly logger: LogWrapper;
+  private _loaded: boolean = false;
+
+  constructor ({ logger }: { logger: LogWrapper; }) {
+    this.logger = logger;
+    this.load();
+  }
+
+  private load (): void {
+    const platform = process.platform + '.' + process.arch;
+
+    if (!this.supportedPlatforms.includes(platform)) {
+      this.logger.logWarn(`Napi2NativeLoader: 不支持的平台: ${platform}`);
+      this._loaded = false;
+      return;
+    }
+
+    const nativeModulePath = path.join(
+      dirname(fileURLToPath(import.meta.url)),
+      './native/napi2native/napi2native.' + platform + '.node'
+    );
+
+    if (!fs.existsSync(nativeModulePath)) {
+      this.logger.logWarn(`Napi2NativeLoader: 缺失运行时文件: ${nativeModulePath}`);
+      this._loaded = false;
+      return;
+    }
+
+    try {
+      process.dlopen(this.exports, nativeModulePath, constants.dlopen.RTLD_LAZY);
+      this._loaded = true;
+      this.logger.log('[Napi2NativeLoader] 加载成功');
+    } catch (error) {
+      this.logger.logError('Napi2NativeLoader 加载出错:', error);
+      this._loaded = false;
+    }
+  }
+
+  get loaded (): boolean {
+    return this._loaded;
+  }
+
+  get nativeExports (): Napi2NativeExportType {
+    return this.exports.exports;
+  }
+
+  /**
+   * 初始化 Hook
+   * @param send send 偏移地址
+   * @param recv recv 偏移地址
+   * @returns 是否初始化成功
+   */
+  initHook (send: string, recv: string): boolean {
+    if (!this._loaded) {
+      this.logger.logWarn('Napi2NativeLoader 未成功加载，无法初始化 Hook');
+      return false;
+    }
+
+    try {
+      return this.nativeExports.initHook?.(send, recv) ?? false;
+    } catch (error) {
+      this.logger.logError('Napi2NativeLoader initHook 出错:', error);
+      return false;
+    }
+  }
+}