public bool AcquireEchoDataArg(out byte[] buffer, out int offset, out int length, out int echoint)
{
mutex.WaitOne();
buffer = null;
offset = -1;
length = -1;
echoint = -1;
int echoIndex = -1;
if (ConfigManager.Instance.DataProtocol == "fakes5echo")
{
echoIndex = 97;
if (occupiedIndexPool.IndexOf(echoIndex) != -1)
{
Logger.Instance.LogError("已经有正在进行的97号请求");
mutex.ReleaseMutex();
return(false);
}
}
else
{
if (!echoIndexPool.TryDequeue(out echoIndex))
{
Logger.Instance.LogError("发送池枯竭");
mutex.ReleaseMutex();
return(false);
}
}
occupiedIndexPool.Add(echoIndex);
mutex.ReleaseMutex();
buffer = new byte[poolUnitSize];
offset = 0;
length = poolUnitSize;
Buffer.BlockCopy(echoDataPool, echoIndex * poolUnitSize, buffer, 0, poolUnitSize);
if (BitConverter.ToInt32(buffer) != echoIndex)
{
Logger.Instance.LogError($"数据校验出错: {BitConverter.ToInt32(buffer)} != {echoIndex}");
return(false);
}
// Logger.Instance.LogInfo($"[{echoIndex}] sent");
echoint = echoIndex;
ProtocolWrapperFactory.GetProtocolWrapper("default").PackProtocol(ref buffer, ref offset, ref length);
return(true);
}
public void OnReceive(byte[] data, int offset, int length)
{
long latency = 0;
ProtocolWrapperFactory.GetProtocolWrapper("default").UnpackProtocol(ref data, ref offset, ref length);
for (int i = offset; i < length + offset; i += poolUnitSize)
{
if (RecycleEchoDataArg(data, i, poolUnitSize, out latency))
{
// for debug
// Logger.Instance.LogInfo($"[{BitConverter.ToInt32(data, i)}]: {latency}ms");
latencyMeters.ToList().ForEach(x => x.IncreaseLatency(latency));
}
lossMeters.ToList().ForEach(x => x.IncreaseRecv());
}
}
public bool Init(string configFileName)
{
if (!ConfigManager.Instance.Init(configFileName))
{
Logger.Instance.LogError("读取配置文件失败");
return(false);
}
Logger.Instance.LogTrace("配置文件读取成功");
protocolWrapper = ProtocolWrapperFactory.GetProtocolWrapper("");
metricManager = new MetricManager();
if (!metricManager.Init(ConfigManager.Instance.SendPoolSize))
{
Logger.Instance.LogError("性能计数器初始化失败");
return(false);
}
Logger.Instance.LogTrace("性能计数器初始化成功");
netClients = new NetClients(ConfigManager.Instance.Protocol, ConfigManager.Instance.ClientCount,
ConfigManager.Instance.ThreadsPerClient, ConfigManager.Instance.BufferSize);
netClients.OnSend += this.OnSend;
netClients.OnSend += metricManager.OnSend;
netClients.OnReceive += this.OnReceive;
netClients.OnReceive += metricManager.OnReceive;
if (!netClients.Init(ConfigManager.Instance.IP, ConfigManager.Instance.Port))
{
Logger.Instance.LogError("客户端连接初始化失败");
return(false);
}
Logger.Instance.LogTrace("客户端连接初始化成功");
stopWatch = new Stopwatch();
Console.WriteLine($"压测目标: [{ConfigManager.Instance.Protocol.ToUpper()}] {ConfigManager.Instance.Target.ToString()}");
Console.WriteLine($"客户端连接数: {ConfigManager.Instance.ClientCount}, 发送池大小: {ConfigManager.Instance.SendPoolSize}");
if (ConfigManager.Instance.ControlMode == "count")
{
Console.WriteLine($"预计发包数量: {ConfigManager.Instance.ControlParam}");
}
else if (ConfigManager.Instance.ControlMode == "time")
{
Console.WriteLine($"预计发包时间: {ConfigManager.Instance.ControlParam / 1000}s");
}
else if (ConfigManager.Instance.ControlMode == "manual")
{
Console.WriteLine("手动停止发包(按任意键退出程序)");
}
else
{
Console.WriteLine($"未知控制模式: {ConfigManager.Instance.ControlMode}");
}
if (ConfigManager.Instance.DataProtocol == "raw")
{
Console.WriteLine($"数据协议: 普通回显服务");
}
else if (ConfigManager.Instance.DataProtocol == "fakes5")
{
Console.WriteLine($"数据协议: 经典LSP协议/回显, 真实目标: {ConfigManager.Instance.TargetAcc}");
}
else if (ConfigManager.Instance.DataProtocol == "fakes5echo")
{
Console.WriteLine($"数据协议: 经典LspUdping, 真实目标: 海外S5节点");
}
else
{
Console.WriteLine($"数据协议: 未知, 不可思议的事情发生了");
}
if (ConfigManager.Instance.RelayMode == "interval")
{
Console.WriteLine($"发包模式: 连续发送, 间隔时间: {ConfigManager.Instance.RelayInterval}ms");
}
else if (ConfigManager.Instance.RelayMode == "onsend")
{
Console.WriteLine($"发包模式: 固定速度发包, 发包速度: {ConfigManager.Instance.RelayMaxPps}pps");
}
else if (ConfigManager.Instance.RelayMode == "onrecv")
{
Console.WriteLine($"发包模式: 接力发送, 热身包量: {ConfigManager.Instance.WarmupCount}, 热身包间隔: {ConfigManager.Instance.WarmupInterval}ms");
}
else
{
Console.WriteLine($"未知发包模式: {ConfigManager.Instance.RelayMode}");
}
Console.WriteLine("");
Logger.Instance.LogInfo("压测程序初始化成功\r\n");
return(true);
}
