105 lines
3.4 KiB
TypeScript
105 lines
3.4 KiB
TypeScript
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;
|
|
}
|
|
|
|
export interface RecvPacketData {
|
|
seq: number;
|
|
cmd: string;
|
|
data: Buffer;
|
|
}
|
|
|
|
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;
|
|
|
|
constructor (napCore: NapCoreContext, logger: PacketLogger, logStack: LogStack, napi2nativeLoader: Napi2NativeLoader) {
|
|
this.napcore = napCore;
|
|
this.logger = logger;
|
|
this.logStack = logStack;
|
|
this.napi2nativeLoader = napi2nativeLoader;
|
|
}
|
|
|
|
check (): boolean {
|
|
if (!this.napi2nativeLoader.loaded) {
|
|
this.logStack.pushLogWarn('NativePacketClient: Napi2NativeLoader 未成功加载');
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
async init (_pid: number, recv: string, send: string): Promise<void> {
|
|
this.available = false;
|
|
const isNewQQ = this.napcore.basicInfo.requireMinNTQQBuild('40768');
|
|
if (!isNewQQ) {
|
|
this.logStack.pushLogWarn('[PacketClient] 当前 QQ 版本低于 NativePacketClient 要求，跳过 Hook 初始化');
|
|
return;
|
|
}
|
|
const success = this.napi2nativeLoader.initHook(send, recv);
|
|
if (success) {
|
|
this.available = true;
|
|
this.logStack.pushLogInfo(`[PacketClient] NativePacketClient Hook 初始化成功 platform=${process.platform}.${process.arch}`);
|
|
return;
|
|
}
|
|
this.logStack.pushLogError(`[PacketClient] NativePacketClient Hook 初始化失败 platform=${process.platform}.${process.arch} send=${send} recv=${recv}`);
|
|
}
|
|
|
|
async sendPacket (
|
|
cmd: string,
|
|
data: PacketBuf,
|
|
rsp = false,
|
|
timeout = 5000
|
|
): Promise<RecvPacketData> {
|
|
if (!rsp) {
|
|
this.napcore
|
|
.sendSsoCmdReqByContend(cmd, data)
|
|
.catch((err: any) =>
|
|
this.logger.error(
|
|
`[PacketClient] sendPacket 无响应命令发送失败 cmd=${cmd} err=${err}`
|
|
)
|
|
);
|
|
return { seq: 0, cmd, data: Buffer.alloc(0) };
|
|
}
|
|
|
|
const sendPromise = this.napcore
|
|
.sendSsoCmdReqByContend(cmd, data)
|
|
.then(ret => ({
|
|
seq: 0,
|
|
cmd,
|
|
data: (ret as { rspbuffer: Buffer; }).rspbuffer,
|
|
}));
|
|
|
|
let timeoutHandle: ReturnType<typeof setTimeout>;
|
|
const timeoutPromise = new Promise<RecvPacketData>((_resolve, reject) => {
|
|
timeoutHandle = setTimeout(
|
|
() =>
|
|
reject(
|
|
new Error(
|
|
`[PacketClient] sendPacket 超时 cmd=${cmd} timeout=${timeout}ms`
|
|
)
|
|
),
|
|
timeout
|
|
);
|
|
});
|
|
|
|
try {
|
|
return await Promise.race([sendPromise, timeoutPromise]);
|
|
} finally {
|
|
clearTimeout(timeoutHandle!);
|
|
}
|
|
}
|
|
|
|
async sendOidbPacket (pkt: OidbPacket, rsp = false, timeout = 5000): Promise<RecvPacketData> {
|
|
return await this.sendPacket(pkt.cmd, pkt.data, rsp, timeout);
|
|
}
|
|
}
|