XNSim/XNSimHtml/utils/systemMonitor.js

const ffi = require('ffi-napi');
const ref = require('ref-napi');
const path = require('path');
const os = require('os');

// 获取XNCore环境变量
const xnCorePath = process.env.XNCore || '';

// 根据操作系统类型确定动态库扩展名和路径
const isWindows = os.platform() === 'win32';
const libExtension = isWindows ? '.dll' : '.so';
const libPrefix = isWindows ? '' : 'lib';

// Login库配置
const loginLibName = `${libPrefix}Login${libExtension}`;
const loginLibPath = path.join(xnCorePath, 'lib', loginLibName);

// MonitorServer库配置
const monitorLibName = `${libPrefix}XNMonitorServer${libExtension}`;
const monitorLibPath = path.join(xnCorePath, 'lib', monitorLibName);

// 定义Buffer类型
const BufferType = ref.refType(ref.types.void);
const StringType = ref.types.CString;
const IntType = ref.types.int;

// 定义动态库函数接口
let loginLib;
let monitorLib;

try {
  loginLib = ffi.Library(loginLibPath, {
    'validateUser': ['int', [BufferType, 'size_t', BufferType, 'size_t']],
    'getUserInfo': [StringType, ['int']],
    'cleanup': ['void', []],
    'registerUser': ['int', [BufferType, 'size_t', BufferType, 'size_t', BufferType, 'size_t']]
  });
} catch (error) {
  console.error(`加载 ${loginLibName} 失败:`, error);
}

try {
  monitorLib = ffi.Library(monitorLibPath, {
    'XN_Initialize': ['int', [StringType, 'int', StringType, 'int']],
    'XN_Cleanup': ['void', []],
    'XN_StartMonitorSystemInfo': ['int', [StringType, 'int']],
    'XN_GetSystemInfo': ['int', [StringType, 'int']],
    'XN_GetAllThreadInfo': ['int', [StringType, 'int']],
    'XN_StopMonitorSystemInfo': ['void', []],
    'XN_StartMonitorModelInfo': ['int', [StringType, 'int']],
    'XN_GetModelInfo': ['int', [StringType, 'int']],
    'XN_StopMonitorModelInfo': ['void', []],
    'XN_InitializeEngineControl': ['int', [StringType, 'int']],
    'XN_PauseEngine': ['void', [StringType, 'int']],
    'XN_ResumeEngine': ['void', [StringType, 'int']],
    'XN_StopEngine': ['void', [StringType, 'int']],
    'XN_StartDataMonitor': ['int', [StringType, 'int', StringType, 'int']],
    'XN_StopDataMonitor': ['void', [StringType, 'int', StringType, 'int']],
    'XN_GetDataMonitorInfo': ['int', [StringType, 'int', StringType, 'int', StringType, 'int', StringType, 'int']],
    'XN_InjectDataInterface': ['int', [StringType, 'int', StringType, 'int', StringType, 'int']],
    'XN_StartInjectContinuous': ['int', [StringType, 'int', StringType, 'int', 'double', StringType, 'int']],
    'XN_StopInjectContinuous': ['int', [StringType, 'int', StringType, 'int']]
  });
} catch (error) {
  console.error(`加载 ${monitorLibName} 失败:`, error);
}

// 注册进程退出时的清理函数
function performCleanup() {
  console.log('正在执行清理操作...');
  if (loginLib) {
    try {
      loginLib.cleanup();
      console.log('清理操作完成');
    } catch (error) {
      console.error('清理操作失败:', error);
    }
  }
}

// 转换字符串为Buffer的函数
function stringToBuffer(str) {
  try {
    if (!str) {
      return { buffer: Buffer.alloc(0), length: 0 };
    }
    const buffer = Buffer.from(str, 'utf8');
    return { buffer: buffer, length: buffer.length };
  } catch (error) {
    console.error('字符串转Buffer失败:', error);
    return { buffer: Buffer.alloc(0), length: 0 };
  }
}

// 初始化监控服务器
function initializeMonitor(domainId) {
  if (!monitorLib) {
    return '监控服务器库未加载';
  }
  try {
    // 创建错误消息缓冲区
    const errorMsg = Buffer.alloc(1024);
    const result = monitorLib.XN_Initialize(domainId, domainId.length, errorMsg, errorMsg.length);
    
    if (result !== 0) {
      return `初始化失败: ${errorMsg.toString('utf8').replace(/\0/g, '')}`;
    }
    return '初始化成功';
  } catch (error) {
    return `初始化失败: ${error.message}`;
  }
}

// 清理监控服务器资源
function cleanupMonitor() {
  if (!monitorLib) {
    return;
  }
  try {
    monitorLib.XN_Cleanup();
  } catch (error) {
    return;
  }
}

// 启动监控系统信息
function startMonitorSystemInfo() {
  if (!monitorLib) {
    return '监控服务器库未加载';
  }
  try {
    const errorMsg = Buffer.alloc(1024);
    const result = monitorLib.XN_StartMonitorSystemInfo(errorMsg, errorMsg.length);
    if (result !== 0) {
      return `启动失败: ${errorMsg.toString('utf8').replace(/\0/g, '')}`;
    }
    return '启动成功';
  } catch (error) {
    return `启动失败: ${error.message}`;
  }
}

// 获取系统信息
function getSystemInfo() {
  if (!monitorLib) {
    return '监控服务器库未加载';
  }
  try {
    const infoMsg = Buffer.alloc(8192);
    const result = monitorLib.XN_GetSystemInfo(infoMsg, infoMsg.length);
    
    if (result !== 0) {
      return `获取失败: ${infoMsg.toString('utf8').replace(/\0/g, '')}`;
    }
    return infoMsg.toString('utf8').replace(/\0/g, '');
  } catch (error) {
    return `获取失败: ${error.message}`;
  }
}

// 获取所有线程信息
function getAllThreadInfo() {
  if (!monitorLib) {
    return '监控服务器库未加载';
  }
  try {
    const infoMsg = Buffer.alloc(8192);
    const result = monitorLib.XN_GetAllThreadInfo(infoMsg, infoMsg.length);
    
    if (result !== 0) {
      return `获取失败: ${infoMsg.toString('utf8').replace(/\0/g, '')}`;
    }
    return infoMsg.toString('utf8').replace(/\0/g, '');
  } catch (error) {
    return `获取失败: ${error.message}`;
  }
}

// 停止监控系统信息
function stopMonitorSystemInfo() {
  if (!monitorLib) {
    return;
  }
  try {
    monitorLib.XN_StopMonitorSystemInfo();
  } catch (error) {
    return;
  }
}

// 启动监控模型信息
function startMonitorModelInfo() {
  if (!monitorLib) {
    return '监控服务器库未加载';
  }
  try {
    const errorMsg = Buffer.alloc(1024);
    const result = monitorLib.XN_StartMonitorModelInfo(errorMsg, errorMsg.length);
    if (result !== 0) {
      return `启动失败: ${errorMsg.toString('utf8').replace(/\0/g, '')}`;
    }
    return '启动成功';
  } catch (error) {
    return `启动失败: ${error.message}`;
  }
}

// 获取模型信息
function getModelInfo() {
  if (!monitorLib) {
    return '监控服务器库未加载';
  }
  try {
    const infoMsg = Buffer.alloc(16384);
    const result = monitorLib.XN_GetModelInfo(infoMsg, infoMsg.length);
    
    if (result !== 0) {
      return `获取失败: ${infoMsg.toString('utf8').replace(/\0/g, '')}`;
    }
    return infoMsg.toString('utf8').replace(/\0/g, '');
  } catch (error) {
    return `获取失败: ${error.message}`;
  }
}

// 停止监控模型信息
function stopMonitorModelInfo() {
  if (!monitorLib) {
    return;
  }
  try {
    monitorLib.XN_StopMonitorModelInfo();
  } catch (error) {
    return;
  }
}

// 初始化引擎控制
function initializeEngineControl() {
  if (!monitorLib) {
    return '监控服务器库未加载';
  }
  try {
    const errorMsg = Buffer.alloc(1024);
    const result = monitorLib.XN_InitializeEngineControl(errorMsg, errorMsg.length);
    if (result !== 0) {
      return `初始化引擎控制失败: ${errorMsg.toString('utf8').replace(/\0/g, '')}`;
    }
    return '初始化引擎控制成功';
  } catch (error) {
    return `初始化引擎控制失败: ${error.message}`;
  }
}

// 暂停引擎
function pauseEngine() {
  if (!monitorLib) {
    return '监控服务器库未加载';
  }
  try {
    const errorMsg = Buffer.alloc(1024);
    monitorLib.XN_PauseEngine(errorMsg, errorMsg.length);
    const error = errorMsg.toString('utf8').replace(/\0/g, '');
    return error ? `暂停引擎失败: ${error}` : '暂停引擎成功';
  } catch (error) {
    return `暂停引擎失败: ${error.message}`;
  }
}

// 恢复引擎
function resumeEngine() {
  if (!monitorLib) {
    return '监控服务器库未加载';
  }
  try {
    const errorMsg = Buffer.alloc(1024);
    monitorLib.XN_ResumeEngine(errorMsg, errorMsg.length);
    const error = errorMsg.toString('utf8').replace(/\0/g, '');
    return error ? `恢复引擎失败: ${error}` : '恢复引擎成功';
  } catch (error) {
    return `恢复引擎失败: ${error.message}`;
  }
}

// 停止引擎
function stopEngine() {
  if (!monitorLib) {
    return '监控服务器库未加载';
  }
  try {
    const errorMsg = Buffer.alloc(1024);
    monitorLib.XN_StopEngine(errorMsg, errorMsg.length);
    const error = errorMsg.toString('utf8').replace(/\0/g, '');
    return error ? `停止引擎失败: ${error}` : '停止引擎成功';
  } catch (error) {
    return `停止引擎失败: ${error.message}`;
  }
}

// 启动数据监控
function startDataMonitor(structName) {
  if (!monitorLib) {
    return '监控服务器库未加载';
  }
  try {
    const errorMsg = Buffer.alloc(1024);
    const result = monitorLib.XN_StartDataMonitor(structName, structName.length, errorMsg, errorMsg.length);
    if (result !== 0) {
      return `启动数据监控失败: ${errorMsg.toString('utf8').replace(/\0/g, '')}`;
    }
    return '启动数据监控成功';
  } catch (error) {
    return `启动数据监控失败: ${error.message}`;
  }
}

// 停止数据监控
function stopDataMonitor(structName) {
  if (!monitorLib) {
    return '监控服务器库未加载';
  }
  try {
    const errorMsg = Buffer.alloc(1024);
    monitorLib.XN_StopDataMonitor(structName, structName.length, errorMsg, errorMsg.length);
    const error = errorMsg.toString('utf8').replace(/\0/g, '');
    return error ? `停止数据监控失败: ${error}` : '停止数据监控成功';
  } catch (error) {
    return `停止数据监控失败: ${error.message}`;
  }
}

// 获取数据监控信息
function getDataMonitorInfo(structName, interfaceName, bufferSize = 8192) {
  if (!monitorLib) {
    return '监控服务器库未加载';
  }
  try {
    const infoMsg = Buffer.alloc(1024);
    const dataBuffer = Buffer.alloc(bufferSize);  // 使用指定的缓冲区大小
    const result = monitorLib.XN_GetDataMonitorInfo(
      structName, structName.length,
      interfaceName, interfaceName.length,
      dataBuffer, dataBuffer.length,
      infoMsg, infoMsg.length
    );
    
    if (result !== 0) {
      return `获取数据监控信息失败: ${infoMsg.toString('utf8').replace(/\0/g, '')}`;
    }
    return {
      data: dataBuffer.toString('utf8').replace(/\0/g, ''),
      info: infoMsg.toString('utf8').replace(/\0/g, '')
    };
  } catch (error) {
    return `获取数据监控信息失败: ${error.message}`;
  }
}

// 注入数据接口
function injectDataInterface(structName, interfaceNameAndData) {
  if (!monitorLib) {
    return '监控服务器库未加载';
  }
  try {
    const infoMsg = Buffer.alloc(1024);
    const result = monitorLib.XN_InjectDataInterface(
      structName, structName.length,
      interfaceNameAndData, interfaceNameAndData.length,
      infoMsg, infoMsg.length
    );
    
    if (result !== 0) {
      return `注入数据接口失败: ${infoMsg.toString('utf8').replace(/\0/g, '')}`;
    }
    return '注入数据接口成功';
  } catch (error) {
    return `注入数据接口失败: ${error.message}`;
  }
}

// 持续注入数据接口
function startInjectContinuous(structName, interfaceNameAndData, frequency) {
  if (!monitorLib) {
    return '监控服务器库未加载';
  }
  try {
    const infoMsg = Buffer.alloc(1024);
    const result = monitorLib.XN_StartInjectContinuous(
      structName, structName.length,
      interfaceNameAndData, interfaceNameAndData.length,
      frequency,
      infoMsg, infoMsg.length
    );
    
    if (result !== 0) {
      return `启动持续注入数据失败: ${infoMsg.toString('utf8').replace(/\0/g, '')}`;
    }
    return '启动持续注入数据成功';
  } catch (error) {
    return `启动持续注入数据失败: ${error.message}`;
  }
}

// 停止持续注入数据接口
function stopInjectContinuous(structName) {
  if (!monitorLib) {
    return '监控服务器库未加载';
  }
  try {
    const infoMsg = Buffer.alloc(1024);
    const result = monitorLib.XN_StopInjectContinuous(
      structName, structName.length,
      infoMsg, infoMsg.length
    );
    
    if (result !== 0) {
      return `停止持续注入数据失败: ${infoMsg.toString('utf8').replace(/\0/g, '')}`;
    }
    return '停止持续注入数据成功';
  } catch (error) {
    return `停止持续注入数据失败: ${error.message}`;
  }
}

module.exports = {
  loginLib,
  monitorLib,
  performCleanup,
  stringToBuffer,
  initializeMonitor,
  cleanupMonitor,
  startMonitorSystemInfo,
  getSystemInfo,
  getAllThreadInfo,
  stopMonitorSystemInfo,
  startMonitorModelInfo,
  getModelInfo,
  stopMonitorModelInfo,
  initializeEngineControl,
  pauseEngine,
  resumeEngine,
  stopEngine,
  startDataMonitor,
  stopDataMonitor,
  getDataMonitorInfo,
  injectDataInterface,
  startInjectContinuous,
  stopInjectContinuous
};