390 lines
12 KiB
C#
390 lines
12 KiB
C#
using System;
|
||
using System.Collections.Generic;
|
||
using System.Threading;
|
||
using DotNetty.Buffers;
|
||
using DotNetty.Common.Utilities;
|
||
using base_kcp;
|
||
|
||
namespace fec
|
||
{
|
||
public class LatencySimulator : KcpOutput
|
||
{
|
||
private static long long2Uint(long n)
|
||
{
|
||
return n & 0x00000000FFFFFFFFL;
|
||
}
|
||
|
||
private long current;
|
||
|
||
/**
|
||
* 丢包率
|
||
**/
|
||
private int lostrate;
|
||
private int rttmin;
|
||
private int rttmax;
|
||
private LinkedList<DelayPacket> p12 = new LinkedList<DelayPacket>();
|
||
private LinkedList<DelayPacket> p21 = new LinkedList<DelayPacket>();
|
||
private Random r12 = new Random();
|
||
private Random r21 = new Random();
|
||
|
||
private Random _random = new Random();
|
||
|
||
|
||
// lostrate: 往返一周丢包率的百分比，默认 10%
|
||
// rttmin：rtt最小值，默认 60
|
||
// rttmax：rtt最大值，默认 125
|
||
//func (p *LatencySimulator)Init(int lostrate = 10, int rttmin = 60, int rttmax = 125, int nmax = 1000):
|
||
public void init(int lostrate, int rttmin, int rttmax)
|
||
{
|
||
this.current = DateTime.Now.Ticks/10000;
|
||
this.lostrate = lostrate / 2; // 上面数据是往返丢包率，单程除以2
|
||
this.rttmin = rttmin / 2;
|
||
this.rttmax = rttmax / 2;
|
||
}
|
||
|
||
|
||
// 发送数据
|
||
// peer - 端点0/1，从0发送，从1接收；从1发送从0接收
|
||
public int send(int peer, IByteBuffer data)
|
||
{
|
||
int rnd;
|
||
if (peer == 0)
|
||
{
|
||
rnd = r12.Next(100);
|
||
}
|
||
else
|
||
{
|
||
rnd = r21.Next(100);
|
||
}
|
||
|
||
//println("!!!!!!!!!!!!!!!!!!!!", rnd, p.lostrate, peer)
|
||
if (rnd < lostrate)
|
||
{
|
||
return 0;
|
||
}
|
||
|
||
DelayPacket pkt = new DelayPacket();
|
||
pkt.init(data);
|
||
current = DateTime.Now.Ticks/10000;
|
||
int delay = rttmin;
|
||
if (rttmax > rttmin)
|
||
{
|
||
delay += _random.Next(10000) % (rttmax - rttmin);
|
||
}
|
||
|
||
pkt.setTs(current + delay);
|
||
if (peer == 0)
|
||
{
|
||
p12.AddLast(pkt);
|
||
}
|
||
else
|
||
{
|
||
p21.AddLast(pkt);
|
||
}
|
||
|
||
return 1;
|
||
}
|
||
|
||
// 接收数据
|
||
public int recv(int peer, IByteBuffer data)
|
||
{
|
||
DelayPacket pkt;
|
||
if (peer == 0)
|
||
{
|
||
if (p21.Count == 0)
|
||
{
|
||
return -1;
|
||
}
|
||
|
||
pkt = p21.First.Value;
|
||
}
|
||
else
|
||
{
|
||
if (p12.Count == 0)
|
||
{
|
||
return -1;
|
||
}
|
||
|
||
pkt = p12.First.Value;
|
||
}
|
||
|
||
current = DateTime.Now.Ticks/10000;
|
||
|
||
if (current < pkt.getTs())
|
||
{
|
||
return -2;
|
||
}
|
||
|
||
if (peer == 0)
|
||
{
|
||
p21.RemoveFirst();
|
||
}
|
||
else
|
||
{
|
||
p12.RemoveFirst();
|
||
}
|
||
|
||
int maxsize = pkt.getPtr().ReadableBytes;
|
||
// IByteBuffer data1 = data;
|
||
// IByteBuffer data2 = pkt.getPtr();
|
||
// Console.WriteLine(data1.AddressOfPinnedMemory().ToString());
|
||
// Console.WriteLine(data2.AddressOfPinnedMemory().ToString());
|
||
|
||
data.WriteBytes(pkt.getPtr());
|
||
// data2.Release();
|
||
pkt.Release();
|
||
// Console.WriteLine(data.ReferenceCount);
|
||
return maxsize;
|
||
}
|
||
|
||
|
||
public static void Main(String[] args)
|
||
{
|
||
// LatencySimulator latencySimulator = new LatencySimulator();
|
||
// try
|
||
// {
|
||
//// //latencySimulator.test(0);
|
||
//// //latencySimulator.test(1);
|
||
// latencySimulator.test(2);
|
||
// }
|
||
// catch (Exception e)
|
||
// {
|
||
// Console.WriteLine(e);
|
||
// }
|
||
|
||
// latencySimulator.BenchmarkFlush();
|
||
}
|
||
|
||
|
||
//测试flush性能
|
||
public void BenchmarkFlush()
|
||
{
|
||
Kcp kcp = new Kcp(1, new LatencySimulator());
|
||
for (int i = 0; i < 1000; i++)
|
||
{
|
||
Segment segment = Segment.createSegment(null);
|
||
kcp.sndBuf.AddLast(segment);
|
||
}
|
||
|
||
foreach (var seg in kcp.sndBuf)
|
||
{
|
||
seg.Xmit = 1;
|
||
seg.Resendts = kcp.currentMs() + 10000;
|
||
}
|
||
|
||
//预热
|
||
for (int i = 0; i < 1000000; i++)
|
||
{
|
||
kcp.flush(false, kcp.currentMs());
|
||
}
|
||
|
||
long start = kcp.currentMs();
|
||
for (int i = 0; i < 200000; i++)
|
||
{
|
||
kcp.flush(false, kcp.currentMs());
|
||
}
|
||
|
||
Console.WriteLine((kcp.currentMs() - start) / 200000);
|
||
}
|
||
|
||
|
||
|
||
class TestOutPut:KcpOutput
|
||
{
|
||
private LatencySimulator vnet;
|
||
private int id;
|
||
|
||
public TestOutPut(LatencySimulator vnet,int id)
|
||
{
|
||
this.vnet = vnet;
|
||
this.id = id;
|
||
}
|
||
|
||
public void outPut(IByteBuffer data, Kcp kcp)
|
||
{
|
||
vnet.send(id, data);
|
||
data.Release();
|
||
}
|
||
}
|
||
|
||
|
||
|
||
|
||
public void test(int mode)
|
||
{
|
||
LatencySimulator vnet = new LatencySimulator();
|
||
vnet.init(20, 600, 600);
|
||
TestOutPut output1 = new TestOutPut(vnet, 0);
|
||
TestOutPut output2 = new TestOutPut(vnet, 1);
|
||
|
||
|
||
Kcp kcp1 = new Kcp(0x11223344, output1);
|
||
Kcp kcp2 = new Kcp(0x11223344, output2);
|
||
//kcp1.setAckMaskSize(8);
|
||
//kcp2.setAckMaskSize(8);
|
||
|
||
current = long2Uint(kcp1.currentMs());
|
||
long slap = current + 20;
|
||
int index = 0;
|
||
int next = 0;
|
||
long sumrtt = 0;
|
||
int count = 0;
|
||
int maxrtt = 0;
|
||
kcp1.RcvWnd = 512;
|
||
kcp1.SndWnd = 512;
|
||
kcp2.RcvWnd = 512;
|
||
kcp2.SndWnd = 512;
|
||
|
||
// 判断测试用例的模式
|
||
if (mode == 0)
|
||
{
|
||
// 默认模式
|
||
kcp1.initNodelay(false, 10, 0, false);
|
||
kcp2.initNodelay(false, 10, 0, false);
|
||
}
|
||
else if (mode == 1)
|
||
{
|
||
// 普通模式，关闭流控等
|
||
kcp1.initNodelay(false, 10, 0, true);
|
||
kcp2.initNodelay(false, 10, 0, true);
|
||
}
|
||
else
|
||
{
|
||
// 启动快速模式
|
||
// 第二个参数 nodelay-启用以后若干常规加速将启动
|
||
// 第三个参数 interval为内部处理时钟，默认设置为 10ms
|
||
// 第四个参数 resend为快速重传指标，设置为2
|
||
// 第五个参数 为是否禁用常规流控，这里禁止
|
||
kcp1.initNodelay(true, 10, 2, true);
|
||
kcp2.initNodelay(true, 10, 2, true);
|
||
kcp1.RxMinrto = 10;
|
||
kcp1.Fastresend = 1;
|
||
}
|
||
|
||
int hr;
|
||
long ts1 = kcp1.currentMs();
|
||
|
||
//写数据 定时更新
|
||
for (;;)
|
||
{
|
||
current = long2Uint(kcp1.currentMs());
|
||
Thread.Sleep(1);
|
||
long now = kcp1.currentMs();
|
||
kcp1.update(now);
|
||
kcp2.update(now);
|
||
|
||
|
||
//每隔 20ms，kcp1发送数据
|
||
for (; current >= slap; slap += 20)
|
||
{
|
||
IByteBuffer buf = PooledByteBufferAllocator.Default.Buffer();
|
||
buf.WriteIntLE(index);
|
||
index++;
|
||
buf.WriteIntLE((int) current);
|
||
kcp1.send(buf);
|
||
buf.Release();
|
||
}
|
||
|
||
//处理虚拟网络：检测是否有udp包从p1->p2
|
||
for (;;)
|
||
{
|
||
IByteBuffer buffer = PooledByteBufferAllocator.Default.DirectBuffer(2000);
|
||
// Console.WriteLine("buffer:" +buffer.AddressOfPinnedMemory().ToString());
|
||
try
|
||
{
|
||
hr = vnet.recv(1, buffer);
|
||
if (hr < 0)
|
||
{
|
||
break;
|
||
}
|
||
|
||
kcp2.input(buffer, true, kcp1.currentMs());
|
||
}
|
||
finally
|
||
{
|
||
buffer.Release();
|
||
}
|
||
}
|
||
|
||
// 处理虚拟网络：检测是否有udp包从p2->p1
|
||
for (;;)
|
||
{
|
||
IByteBuffer buffer = PooledByteBufferAllocator.Default.Buffer(2000);
|
||
try
|
||
{
|
||
hr = vnet.recv(0, buffer);
|
||
if (hr < 0)
|
||
{
|
||
break;
|
||
}
|
||
|
||
// 如果 p1收到udp，则作为下层协议输入到kcp1
|
||
kcp1.input(buffer, true, kcp1.currentMs());
|
||
}
|
||
finally
|
||
{
|
||
buffer.Release();
|
||
}
|
||
}
|
||
|
||
// kcp2接收到任何包都返回回去
|
||
List<IByteBuffer> bufList = new List<IByteBuffer>();
|
||
kcp2.recv(bufList);
|
||
foreach (var byteBuf in bufList)
|
||
{
|
||
kcp2.send(byteBuf);
|
||
byteBuf.Release();
|
||
}
|
||
|
||
// kcp1收到kcp2的回射数据
|
||
bufList = new List<IByteBuffer>();
|
||
kcp1.recv(bufList);
|
||
foreach (var byteBuf in bufList){
|
||
long sn = byteBuf.ReadIntLE();
|
||
long ts = byteBuf.ReadUnsignedIntLE();
|
||
long rtt = 0;
|
||
rtt = current - ts;
|
||
Console.WriteLine("rtt :" + rtt);
|
||
|
||
|
||
if (sn != next)
|
||
{
|
||
// 如果收到的包不连续
|
||
//for i:=0;i<8 ;i++ {
|
||
//println("---", i, buffer[i])
|
||
//}
|
||
Console.WriteLine("ERROR sn " + count + "<->" + next + sn);
|
||
return;
|
||
}
|
||
|
||
next++;
|
||
sumrtt += rtt;
|
||
count++;
|
||
if (rtt > maxrtt)
|
||
{
|
||
maxrtt = (int) rtt;
|
||
}
|
||
|
||
byteBuf.Release();
|
||
}
|
||
if (next > 1000)
|
||
{
|
||
break;
|
||
}
|
||
}
|
||
|
||
ts1 = kcp1.currentMs() - ts1;
|
||
String[] names = new String[] {"default", "normal", "fast"};
|
||
Console.WriteLine(names[mode]+" mode result :"+ts1+" \n");
|
||
Console.WriteLine("avgrtt="+(sumrtt / count)+" maxrtt="+maxrtt+" \n");
|
||
Console.WriteLine("lost percent: " + (Snmp.snmp.RetransSegs)+"\n");
|
||
Console.WriteLine("snmp: " + (Snmp.snmp.ToString()));
|
||
}
|
||
|
||
|
||
public void outPut(IByteBuffer data, Kcp kcp)
|
||
{
|
||
throw new NotImplementedException();
|
||
}
|
||
}
|
||
} |