internal RioConnectionlessSocket(RioSocketPool pool, RioFixedBufferPool sendBufferPool, RioFixedBufferPool receiveBufferPool,
uint maxOutstandingReceive, uint maxOutstandingSend, IntPtr SendCompletionQueue, IntPtr ReceiveCompletionQueue)
: base(sendBufferPool, receiveBufferPool, maxOutstandingReceive, maxOutstandingSend,
SendCompletionQueue, ReceiveCompletionQueue,
ADDRESS_FAMILIES.AF_INET, SOCKET_TYPE.SOCK_DGRAM, PROTOCOL.IPPROTO_UDP)
{
}
static void Main(string[] args)
{
pipeLineDeph = int.Parse(args.FirstOrDefault(f => f.StartsWith("-p"))?.Substring(2) ?? "16");
int connections = int.Parse(args.FirstOrDefault(f => f.StartsWith("-c"))?.Substring(2) ?? "1024");
Console.WriteLine("RioSharp http server");
Console.WriteLine("Optimizing for " + connections + " connections");
Console.WriteLine("Optimizing for pipeline depth of: " + pipeLineDeph);
sendPool = new RioFixedBufferPool(10 * connections, 256 * pipeLineDeph);
recivePool = new RioFixedBufferPool(10 * connections, 256 * pipeLineDeph);
listener = new RioTcpListener(sendPool, recivePool, (uint)connections);
currentSegment = listener.PreAllocateWrite(GetResponse());
responseBytes = GetResponse();
//Task.Run(async () =>
//{
// while (true)
// {
// UpdateResponse();
// await Task.Delay(60000);
// }
//});
listener.OnAccepted = new Action<RioSocket>(s => ThreadPool.QueueUserWorkItem(o => Servebuff((RioSocket)o), s));
listener.Listen(new IPEndPoint(new IPAddress(new byte[] { 0, 0, 0, 0 }), 5000), 1024 * connections);
Console.WriteLine("Listening on : http://localhost:5000");
Console.WriteLine("Press enter to exit");
Console.ReadLine();
listener.Dispose();
}
public RioTcpClientPool(RioFixedBufferPool sendPool, RioFixedBufferPool revicePool, uint socketCount,
uint maxOutstandingReceive = 1024, uint maxOutstandingSend = 1024)
: base(sendPool, revicePool, socketCount, ADDRESS_FAMILIES.AF_INET, SOCKET_TYPE.SOCK_STREAM, PROTOCOL.IPPROTO_TCP, maxOutstandingReceive, maxOutstandingSend, (maxOutstandingReceive + maxOutstandingSend) * socketCount)
{
foreach (var s in allSockets)
{
_freeSockets.Enqueue(s);
in_addr inAddress = new in_addr();
inAddress.s_b1 = 0;
inAddress.s_b2 = 0;
inAddress.s_b3 = 0;
inAddress.s_b4 = 0;
sockaddr_in sa = new sockaddr_in();
sa.sin_family = adressFam;
sa.sin_port = 0;
//Imports.ThrowLastWSAError();
sa.sin_addr = inAddress;
unsafe
{
if (WinSock.bind(s.Socket, ref sa, sizeof(sockaddr_in)) == WinSock.SOCKET_ERROR)
{
WinSock.ThrowLastWSAError();
}
}
}
}
internal RioSocket(RioFixedBufferPool sendBufferPool, RioFixedBufferPool receiveBufferPool,
uint maxOutstandingReceive, uint maxOutstandingSend, IntPtr SendCompletionQueue, IntPtr ReceiveCompletionQueue,
ADDRESS_FAMILIES adressFam, SOCKET_TYPE sockType, PROTOCOL protocol)
{
if ((Socket = WinSock.WSASocket(adressFam, sockType, protocol, IntPtr.Zero, 0, SOCKET_FLAGS.REGISTERED_IO | SOCKET_FLAGS.WSA_FLAG_OVERLAPPED)) == IntPtr.Zero)
{
WinSock.ThrowLastWSAError();
}
SendBufferPool = sendBufferPool;
ReceiveBufferPool = receiveBufferPool;
_requestQueue = RioStatic.CreateRequestQueue(Socket, maxOutstandingReceive, 1, maxOutstandingSend, 1, ReceiveCompletionQueue, SendCompletionQueue, GetHashCode());
WinSock.ThrowLastWSAError();
onIncommingSegmentWrapper = (socket, segment) =>
{
onIncommingSegmentSafe(segment);
if (segment.CurrentContentLength > 0)
{
socket.BeginReceive();
}
else
{
socket.Dispose();
}
segment.Dispose();
};
}
public unsafe RioConnectionOrientedSocketPool(RioFixedBufferPool sendPool, RioFixedBufferPool revicePool, uint socketCount, ADDRESS_FAMILIES adressFam, SOCKET_TYPE sockType, PROTOCOL protocol,
uint maxOutstandingReceive = 1024, uint maxOutstandingSend = 1024, uint maxConnections = 1024)
: base(sendPool, revicePool, adressFam, sockType, protocol, maxOutstandingReceive, maxOutstandingSend, maxConnections)
{
var adrSize = (sizeof(sockaddr_in) + 16) * 2;
var overlapped = Marshal.AllocHGlobal(new IntPtr(socketCount * Marshal.SizeOf <RioNativeOverlapped>()));
var adressBuffer = Marshal.AllocHGlobal(new IntPtr(socketCount * adrSize));
allSockets = new RioConnectionOrientedSocket[socketCount];
for (int i = 0; i < socketCount; i++)
{
allSockets[i] = new RioConnectionOrientedSocket(overlapped + (i * Marshal.SizeOf <RioNativeOverlapped>()), adressBuffer + (i * adrSize), this, SendBufferPool, ReceiveBufferPool, maxOutstandingReceive, maxOutstandingSend, SendCompletionQueue, ReceiveCompletionQueue, adressFam, sockType, protocol);
allSockets[i]._overlapped->SocketIndex = i;
}
if ((socketIocp = Kernel32.CreateIoCompletionPort((IntPtr)(-1), IntPtr.Zero, 0, 1)) == IntPtr.Zero)
{
Kernel32.ThrowLastError();
}
foreach (var s in allSockets)
{
if ((Kernel32.CreateIoCompletionPort(s.Socket, socketIocp, 0, 1)) == IntPtr.Zero)
{
Kernel32.ThrowLastError();
}
}
Thread SocketIocpThread = new Thread(SocketIocpComplete);
SocketIocpThread.IsBackground = true;
SocketIocpThread.Start();
}
public unsafe RioTcpListener(RioFixedBufferPool sendPool, RioFixedBufferPool revicePool, uint socketCount, uint maxOutstandingReceive = 2048, uint maxOutstandingSend = 2048)
: base(sendPool, revicePool, socketCount, ADDRESS_FAMILIES.AF_INET, SOCKET_TYPE.SOCK_STREAM, PROTOCOL.IPPROTO_TCP, maxOutstandingReceive, maxOutstandingSend)
{
if ((_listenerSocket = WinSock.WSASocket(adressFam, sockType, protocol, IntPtr.Zero, 0, SOCKET_FLAGS.REGISTERED_IO | SOCKET_FLAGS.WSA_FLAG_OVERLAPPED)) == IntPtr.Zero)
{
WinSock.ThrowLastWSAError();
}
int True = 1;
uint dwBytes = 0;
if (WinSock.WSAIoctlGeneral2(_listenerSocket, WinSock.SIO_LOOPBACK_FAST_PATH, &True, sizeof(int), (void *)0, 0, out dwBytes, IntPtr.Zero, IntPtr.Zero) != 0)
{
WinSock.ThrowLastWSAError();
}
if (WinSock.setsockopt(_listenerSocket, WinSock.IPPROTO_TCP, WinSock.TCP_NODELAY, &True, 4) != 0)
{
WinSock.ThrowLastWSAError();
}
if ((_listenIocp = Kernel32.CreateIoCompletionPort(_listenerSocket, _listenIocp, 0, 1)) == IntPtr.Zero)
{
Kernel32.ThrowLastError();
}
Thread AcceptIocpThread = new Thread(AcceptIocpComplete);
AcceptIocpThread.IsBackground = true;
AcceptIocpThread.Start();
}
public static unsafe void Main(string[] args)
{
var sendPool = new RioFixedBufferPool(10, 256);
var recivePool = new RioFixedBufferPool(10, 256);
var pool = new RioConnectionlessSocketPool(sendPool, recivePool, ADDRESS_FAMILIES.AF_INET, SOCKET_TYPE.SOCK_DGRAM, PROTOCOL.IPPROTO_UDP);
RioConnectionlessSocket sock = null;
var multicastAdress = IPAddress.Parse("238.0.3.15");
try
{
sock = pool.Bind(new IPEndPoint(new IPAddress(new byte[] { 0, 0, 0, 0 }), 3000));
}
catch (Exception)
{
sock = pool.Bind();
}
var nics = NetworkInterface.GetAllNetworkInterfaces()
.Where(n => n.Supports(NetworkInterfaceComponent.IPv4))
.Select(n => new { n.GetIPProperties().GetIPv4Properties().Index });
sock.JoinMulticastGroup(multicastAdress, 0);
RioSegmentReader r = new RioSegmentReader(sock);
r.OnIncommingSegment = segment => Console.WriteLine(Encoding.ASCII.GetString(segment.DataPointer, segment.CurrentContentLength));
r.Start();
while (true)
{
sock.Send(Encoding.ASCII.GetBytes("Hello, i'm process " + Process.GetCurrentProcess().Id), new IPEndPoint(multicastAdress, 3000));
Thread.Sleep(1000);
}
}
public RioTcpClientPool(RioFixedBufferPool sendPool, RioFixedBufferPool revicePool, uint socketCount,
uint maxOutstandingReceive = 1024, uint maxOutstandingSend = 1024)
: base(sendPool, revicePool, socketCount, maxOutstandingReceive, maxOutstandingSend, (maxOutstandingReceive + maxOutstandingSend) * socketCount)
{
foreach (var s in allSockets)
{
_freeSockets.Enqueue(s);
in_addr inAddress = new in_addr();
inAddress.s_b1 = 0;
inAddress.s_b2 = 0;
inAddress.s_b3 = 0;
inAddress.s_b4 = 0;
sockaddr_in sa = new sockaddr_in();
sa.sin_family = ADDRESS_FAMILIES.AF_INET;
sa.sin_port = 0;
//Imports.ThrowLastWSAError();
sa.sin_addr = inAddress;
unsafe
{
if (WinSock.bind(s.Socket, ref sa, sizeof(sockaddr_in)) == WinSock.SOCKET_ERROR)
WinSock.ThrowLastWSAError();
}
}
}
public unsafe RioConnectionOrientedSocketPool(RioFixedBufferPool sendPool, RioFixedBufferPool revicePool, uint socketCount, ADDRESS_FAMILIES adressFam, SOCKET_TYPE sockType, PROTOCOL protocol,
uint maxOutstandingReceive = 1024, uint maxOutstandingSend = 1024, uint maxConnections = 1024)
: base(sendPool, revicePool, adressFam, sockType, protocol, maxOutstandingReceive, maxOutstandingSend, maxConnections)
{
var adrSize = (sizeof(sockaddr_in) + 16) * 2;
var overlapped = Marshal.AllocHGlobal(new IntPtr(socketCount * Marshal.SizeOf<RioNativeOverlapped>()));
var adressBuffer = Marshal.AllocHGlobal(new IntPtr(socketCount * adrSize));
allSockets = new RioConnectionOrientedSocket[socketCount];
for (int i = 0; i < socketCount; i++)
{
allSockets[i] = new RioConnectionOrientedSocket(overlapped + (i * Marshal.SizeOf<RioNativeOverlapped>()), adressBuffer + (i * adrSize), this, SendBufferPool, ReceiveBufferPool, maxOutstandingReceive, maxOutstandingSend, SendCompletionQueue, ReceiveCompletionQueue, adressFam, sockType, protocol);
allSockets[i]._overlapped->SocketIndex = i;
}
if ((socketIocp = Kernel32.CreateIoCompletionPort((IntPtr)(-1), IntPtr.Zero, 0, 1)) == IntPtr.Zero)
Kernel32.ThrowLastError();
foreach (var s in allSockets)
{
if ((Kernel32.CreateIoCompletionPort(s.Socket, socketIocp, 0, 1)) == IntPtr.Zero)
Kernel32.ThrowLastError();
}
Thread SocketIocpThread = new Thread(SocketIocpComplete);
SocketIocpThread.IsBackground = true;
SocketIocpThread.Start();
}
//ADDRESS_FAMILIES.AF_INET, SOCKET_TYPE.SOCK_DGRAM, PROTOCOL.IPPROTO_UDP
internal RioConnectionlessSocket(RioSocketPool pool, RioFixedBufferPool sendBufferPool, RioFixedBufferPool receiveBufferPool, RioFixedBufferPool adressBufferPool,
uint maxOutstandingReceive, uint maxOutstandingSend, IntPtr SendCompletionQueue, IntPtr ReceiveCompletionQueue,
ADDRESS_FAMILIES adressFam, SOCKET_TYPE sockType, PROTOCOL protocol)
: base(sendBufferPool, receiveBufferPool, adressBufferPool, maxOutstandingReceive, maxOutstandingSend,
SendCompletionQueue, ReceiveCompletionQueue,
adressFam, sockType, protocol)
{
}
internal RioConnectionlessSocket(RioSocketPool pool, RioFixedBufferPool sendBufferPool, RioFixedBufferPool receiveBufferPool,
uint maxOutstandingReceive, uint maxOutstandingSend, IntPtr SendCompletionQueue, IntPtr ReceiveCompletionQueue,
ADDRESS_FAMILIES adressFam, SOCKET_TYPE sockType, PROTOCOL protocol) :
base(sendBufferPool, receiveBufferPool, maxOutstandingReceive, maxOutstandingSend,
SendCompletionQueue, ReceiveCompletionQueue,
adressFam, sockType, protocol) //ADDRESS_FAMILIES.AF_INET, SOCKET_TYPE.SOCK_DGRAM, PROTOCOL.IPPROTO_UDP
{
}
public RioTcpClientPool(RioFixedBufferPool sendPool, RioFixedBufferPool revicePool, uint socketCount,
uint maxOutstandingReceive = 1024, uint maxOutstandingSend = 1024)
: base(sendPool, revicePool, socketCount, ADDRESS_FAMILIES.AF_INET, SOCKET_TYPE.SOCK_STREAM, PROTOCOL.IPPROTO_TCP, maxOutstandingReceive, maxOutstandingSend)
{
foreach (var s in allSockets)
{
InitializeSocket(s);
_freeSockets.Enqueue(s);
}
}
public RioTcpClientPool(RioFixedBufferPool sendPool, RioFixedBufferPool revicePool, uint socketCount,
uint maxOutstandingReceive = 1024, uint maxOutstandingSend = 1024)
: base(sendPool, revicePool, socketCount, ADDRESS_FAMILIES.AF_INET, SOCKET_TYPE.SOCK_STREAM, PROTOCOL.IPPROTO_TCP, maxOutstandingReceive, maxOutstandingSend)
{
foreach (var s in allSockets)
{
InitializeSocket(s);
_freeSockets.Enqueue(s);
}
}
internal RioSocketBase(RioFixedBufferPool sendBufferPool, RioFixedBufferPool receiveBufferPool,
uint maxOutstandingReceive, uint maxOutstandingSend, IntPtr SendCompletionQueue, IntPtr ReceiveCompletionQueue,
ADDRESS_FAMILIES adressFam, SOCKET_TYPE sockType, PROTOCOL protocol)
{
if ((Socket = WinSock.WSASocket(adressFam, sockType, protocol, IntPtr.Zero, 0, SOCKET_FLAGS.REGISTERED_IO | SOCKET_FLAGS.WSA_FLAG_OVERLAPPED)) == IntPtr.Zero)
WinSock.ThrowLastWSAError();
SendBufferPool = sendBufferPool;
ReceiveBufferPool = receiveBufferPool;
_requestQueue = RioStatic.CreateRequestQueue(Socket, maxOutstandingReceive, 1, maxOutstandingSend, 1, ReceiveCompletionQueue, SendCompletionQueue, GetHashCode());
WinSock.ThrowLastWSAError();
}
internal RioConnectionOrientedSocket(IntPtr overlapped, IntPtr adressBuffer, RioConnectionOrientedSocketPool pool, RioFixedBufferPool sendBufferPool, RioFixedBufferPool receiveBufferPool,
uint maxOutstandingReceive, uint maxOutstandingSend, IntPtr SendCompletionQueue, IntPtr ReceiveCompletionQueue)
: base(sendBufferPool, receiveBufferPool, maxOutstandingReceive, maxOutstandingSend, SendCompletionQueue, ReceiveCompletionQueue,
ADDRESS_FAMILIES.AF_INET, SOCKET_TYPE.SOCK_STREAM, PROTOCOL.IPPROTO_TCP)
{
_overlapped = (RioNativeOverlapped*)overlapped.ToPointer();
_eventHandle = Kernel32.CreateEvent(IntPtr.Zero, false, false, null);
_adressBuffer = adressBuffer;
_pool = pool;
unsafe
{
var n = (NativeOverlapped*)overlapped.ToPointer();
n->EventHandle = _eventHandle;
}
}
internal RioSocket(RioFixedBufferPool sendBufferPool, RioFixedBufferPool receiveBufferPool,
uint maxOutstandingReceive, uint maxOutstandingSend, IntPtr SendCompletionQueue, IntPtr ReceiveCompletionQueue,
ADDRESS_FAMILIES adressFam, SOCKET_TYPE sockType, PROTOCOL protocol)
{
if ((Socket = WinSock.WSASocket(adressFam, sockType, protocol, IntPtr.Zero, 0, SOCKET_FLAGS.REGISTERED_IO | SOCKET_FLAGS.WSA_FLAG_OVERLAPPED)) == IntPtr.Zero)
{
WinSock.ThrowLastWSAError();
}
SendBufferPool = sendBufferPool;
ReceiveBufferPool = receiveBufferPool;
_requestQueue = RioStatic.CreateRequestQueue(Socket, maxOutstandingReceive, 1, maxOutstandingSend, 1, ReceiveCompletionQueue, SendCompletionQueue, GetHashCode());
WinSock.ThrowLastWSAError();
}
internal RioBufferSegment(RioFixedBufferPool pool, IntPtr bufferStartPointer, IntPtr segmentStartPointer, int index, int Length)
{
Index = index;
TotalLength = Length;
_pool = pool;
AutoFree = true;
var offset = index * Length;
RawPointer = (byte*)(bufferStartPointer + offset).ToPointer();
SegmentPointer = (RIO_BUFSEGMENT*)(segmentStartPointer + index * Marshal.SizeOf<RIO_BUFSEGMENT>()).ToPointer();
SegmentPointer->BufferId = IntPtr.Zero;
SegmentPointer->Offset = offset;
SegmentPointer->Length = TotalLength;
}
internal RioBufferSegment(RioFixedBufferPool pool, IntPtr bufferStartPointer, IntPtr segmentStartPointer, int index, int Length)
{
Index = index;
TotalLength = Length;
_pool = pool;
AutoFree = true;
var offset = index * Length;
RawPointer = (byte *)(bufferStartPointer + offset).ToPointer();
SegmentPointer = (RIO_BUFSEGMENT *)(segmentStartPointer + index * Marshal.SizeOf <RIO_BUFSEGMENT>()).ToPointer();
SegmentPointer->BufferId = IntPtr.Zero;
SegmentPointer->Offset = offset;
SegmentPointer->Length = TotalLength;
}
internal RioBufferSegment(RioFixedBufferPool pool, IntPtr bufferStartPointer, IntPtr segmentStartPointer, int index, int Length)
{
Index = index;
TotalLength = Length;
_pool = pool;
var offset = index * Length;
RawPointer = (byte*)(bufferStartPointer + offset).ToPointer();
SegmentPointer = (RIO_BUFSEGMENT*)(segmentStartPointer + index * Marshal.SizeOf<RIO_BUFSEGMENT>()).ToPointer();
SegmentPointer->BufferId = IntPtr.Zero;
SegmentPointer->Offset = offset;
SegmentPointer->Length = TotalLength;
_continuationWrapperDelegate = o => ((Action)o)();
}
internal RioBufferSegment(RioFixedBufferPool pool, IntPtr bufferStartPointer, IntPtr segmentStartPointer, int index, int Length)
{
Index = index;
TotalLength = Length;
_pool = pool;
var offset = index * Length;
RawPointer = (byte *)(bufferStartPointer + offset).ToPointer();
SegmentPointer = (RIO_BUFSEGMENT *)(segmentStartPointer + index * Marshal.SizeOf <RIO_BUFSEGMENT>()).ToPointer();
SegmentPointer->BufferId = IntPtr.Zero;
SegmentPointer->Offset = offset;
SegmentPointer->Length = TotalLength;
_continuationWrapperDelegate = o => ((Action)o)();
}
internal RioSocket(RioFixedBufferPool sendBufferPool, RioFixedBufferPool receiveBufferPool, RioFixedBufferPool adressBufferPool,
uint maxOutstandingReceive, uint maxOutstandingSend, IntPtr SendCompletionQueue, IntPtr ReceiveCompletionQueue,
ADDRESS_FAMILIES adressFam, SOCKET_TYPE sockType, PROTOCOL protocol)
{
this.adressFam = adressFam;
this.sockType = sockType;
this.protocol = protocol;
this.maxOutstandingReceive = maxOutstandingReceive;
this.maxOutstandingSend = maxOutstandingSend;
this.ReceiveCompletionQueue = ReceiveCompletionQueue;
this.SendCompletionQueue = SendCompletionQueue;
SendBufferPool = sendBufferPool;
ReceiveBufferPool = receiveBufferPool;
AdressPool = adressBufferPool;
ResetSocket();
}
internal RioSocket(RioFixedBufferPool sendBufferPool, RioFixedBufferPool receiveBufferPool, RioFixedBufferPool adressBufferPool,
uint maxOutstandingReceive, uint maxOutstandingSend, IntPtr SendCompletionQueue, IntPtr ReceiveCompletionQueue,
ADDRESS_FAMILIES adressFam, SOCKET_TYPE sockType, PROTOCOL protocol)
{
this.adressFam = adressFam;
this.sockType = sockType;
this.protocol = protocol;
this.maxOutstandingReceive = maxOutstandingReceive;
this.maxOutstandingSend = maxOutstandingSend;
this.ReceiveCompletionQueue = ReceiveCompletionQueue;
this.SendCompletionQueue = SendCompletionQueue;
SendBufferPool = sendBufferPool;
ReceiveBufferPool = receiveBufferPool;
AdressPool = adressBufferPool;
ResetSocket();
}
public unsafe RioTcpListener(RioFixedBufferPool sendPool, RioFixedBufferPool revicePool, uint socketCount, uint maxOutstandingReceive = 2048, uint maxOutstandingSend = 2048)
: base(sendPool, revicePool, socketCount, ADDRESS_FAMILIES.AF_INET, SOCKET_TYPE.SOCK_STREAM, PROTOCOL.IPPROTO_TCP, maxOutstandingReceive, maxOutstandingSend, (maxOutstandingReceive + maxOutstandingSend) * socketCount)
{
if ((_listenerSocket = WinSock.WSASocket(adressFam, sockType, protocol, IntPtr.Zero, 0, SOCKET_FLAGS.REGISTERED_IO | SOCKET_FLAGS.WSA_FLAG_OVERLAPPED)) == IntPtr.Zero)
WinSock.ThrowLastWSAError();
int True = -1;
UInt32 dwBytes = 0;
WinSock.setsockopt(_listenerSocket, WinSock.IPPROTO_TCP, WinSock.TCP_NODELAY, (char*)&True, 4);
WinSock.WSAIoctlGeneral(_listenerSocket, WinSock.SIO_LOOPBACK_FAST_PATH, &True, 4, null, 0, out dwBytes, IntPtr.Zero, IntPtr.Zero);
if ((_listenIocp = Kernel32.CreateIoCompletionPort(_listenerSocket, _listenIocp, 0, 1)) == IntPtr.Zero)
Kernel32.ThrowLastError();
Thread ListenIocpThread = new Thread(ListenIocpComplete);
ListenIocpThread.IsBackground = true;
ListenIocpThread.Start();
}
internal RioSocket(RioFixedBufferPool sendBufferPool, RioFixedBufferPool receiveBufferPool,
uint maxOutstandingReceive, uint maxOutstandingSend, IntPtr SendCompletionQueue, IntPtr ReceiveCompletionQueue,
ADDRESS_FAMILIES adressFam, SOCKET_TYPE sockType, PROTOCOL protocol)
{
if ((Socket = WinSock.WSASocket(adressFam, sockType, protocol, IntPtr.Zero, 0, SOCKET_FLAGS.REGISTERED_IO | SOCKET_FLAGS.WSA_FLAG_OVERLAPPED)) == IntPtr.Zero)
WinSock.ThrowLastWSAError();
SendBufferPool = sendBufferPool;
ReceiveBufferPool = receiveBufferPool;
_requestQueue = RioStatic.CreateRequestQueue(Socket, maxOutstandingReceive, 1, maxOutstandingSend, 1, ReceiveCompletionQueue, SendCompletionQueue, GetHashCode());
WinSock.ThrowLastWSAError();
onIncommingSegmentWrapper = (socket, segment) =>
{
onIncommingSegmentSafe(segment);
if (segment.CurrentContentLength > 0)
socket.BeginReceive();
else
socket.Dispose();
segment.Dispose();
};
}
static void Main(string[] args)
{
pipeLineDeph = int.Parse(args.FirstOrDefault(f => f.StartsWith("-p"))?.Substring(2) ?? "16");
int connections = int.Parse(args.FirstOrDefault(f => f.StartsWith("-c"))?.Substring(2) ?? "1024");
sendPool = new RioFixedBufferPool(10 * connections, 140 * pipeLineDeph);
recivePool = new RioFixedBufferPool(10 * connections, 128 * pipeLineDeph);
listener = new RioTcpListener(sendPool, recivePool, 1024);
currentSegment = listener.PreAllocateWrite(GetResponse());
responseBytes = GetResponse();
//Task.Run(async () =>
//{
// while (true)
// {
// UpdateResponse();
// await Task.Delay(60000);
// }
//});
listener.OnAccepted = new Action<RioConnectionOrientedSocket>(s => ThreadPool.QueueUserWorkItem(o => Servebuff((RioConnectionOrientedSocket)o), s));
listener.Listen(new IPEndPoint(new IPAddress(new byte[] { 0, 0, 0, 0 }), 5000), 1024 * connections);
Console.ReadLine();
}
public unsafe RioSocketPool(RioFixedBufferPool sendPool, RioFixedBufferPool receivePool, ADDRESS_FAMILIES adressFam, SOCKET_TYPE sockType, PROTOCOL protocol,
uint maxOutstandingReceive = 1024, uint maxOutstandingSend = 1024, uint maxOutsandingCompletions = 2048)
{
MaxOutstandingReceive = maxOutstandingReceive;
MaxOutstandingSend = maxOutstandingSend;
MaxOutsandingCompletions = maxOutsandingCompletions;
SendBufferPool = sendPool;
ReceiveBufferPool = receivePool;
this.adressFam = adressFam;
this.sockType = sockType;
this.protocol = protocol;
var version = new Version(2, 2);
WSAData data;
var result = WinSock.WSAStartup((short)version.Raw, out data);
if (result != 0)
{
WinSock.ThrowLastWSAError();
}
RioStatic.Initalize();
if ((ReceiveCompletionPort = Kernel32.CreateIoCompletionPort((IntPtr)(-1), IntPtr.Zero, 0, 1)) == IntPtr.Zero)
{
Kernel32.ThrowLastError();
}
if ((SendCompletionPort = Kernel32.CreateIoCompletionPort((IntPtr)(-1), IntPtr.Zero, 0, 1)) == IntPtr.Zero)
{
Kernel32.ThrowLastError();
}
_sendBufferId = RioStatic.RegisterBuffer(SendBufferPool.BufferPointer, (uint)SendBufferPool.TotalLength);
WinSock.ThrowLastWSAError();
SendBufferPool.SetBufferId(_sendBufferId);
_reciveBufferId = RioStatic.RegisterBuffer(ReceiveBufferPool.BufferPointer, (uint)ReceiveBufferPool.TotalLength);
WinSock.ThrowLastWSAError();
ReceiveBufferPool.SetBufferId(_reciveBufferId);
var sendCompletionMethod = new RIO_NOTIFICATION_COMPLETION()
{
Type = RIO_NOTIFICATION_COMPLETION_TYPE.IOCP_COMPLETION,
Iocp = new RIO_NOTIFICATION_COMPLETION_IOCP()
{
IocpHandle = SendCompletionPort,
QueueCorrelation = 0,
Overlapped = (NativeOverlapped *)-1
}
};
if ((SendCompletionQueue = RioStatic.CreateCompletionQueue(MaxOutsandingCompletions, sendCompletionMethod)) == IntPtr.Zero)
{
WinSock.ThrowLastWSAError();
}
var receiveCompletionMethod = new RIO_NOTIFICATION_COMPLETION()
{
Type = RIO_NOTIFICATION_COMPLETION_TYPE.IOCP_COMPLETION,
Iocp = new RIO_NOTIFICATION_COMPLETION_IOCP()
{
IocpHandle = ReceiveCompletionPort,
QueueCorrelation = 0,
Overlapped = (NativeOverlapped *)-1
}
};
if ((ReceiveCompletionQueue = RioStatic.CreateCompletionQueue(MaxOutsandingCompletions, receiveCompletionMethod)) == IntPtr.Zero)
{
WinSock.ThrowLastWSAError();
}
Thread receiveThread = new Thread(ProcessReceiveCompletes);
receiveThread.IsBackground = true;
receiveThread.Start();
Thread sendThread = new Thread(ProcessSendCompletes);
sendThread.IsBackground = true;
sendThread.Start();
}
public RioConnectionlessSocketPool(RioFixedBufferPool sendPool, RioFixedBufferPool revicePool, ADDRESS_FAMILIES adressFam, SOCKET_TYPE sockType, PROTOCOL protocol,
uint maxOutstandingReceive = 1024, uint maxOutstandingSend = 1024, uint maxSockets = 128, int adressBufferSize = 1024)
: base(sendPool, revicePool, adressFam, sockType, protocol, maxOutstandingReceive, maxOutstandingSend, maxSockets, adressBufferSize)
{
}
internal RioConnectionOrientedSocket(IntPtr overlapped, IntPtr adressBuffer, RioConnectionOrientedSocketPool pool, RioFixedBufferPool sendBufferPool, RioFixedBufferPool receiveBufferPool,
uint maxOutstandingReceive, uint maxOutstandingSend, IntPtr SendCompletionQueue, IntPtr ReceiveCompletionQueue,
ADDRESS_FAMILIES adressFam, SOCKET_TYPE sockType, PROTOCOL protocol) :
base(sendBufferPool, receiveBufferPool, maxOutstandingReceive, maxOutstandingSend, SendCompletionQueue, ReceiveCompletionQueue,
adressFam, sockType, protocol) //ADDRESS_FAMILIES.AF_INET, SOCKET_TYPE.SOCK_STREAM, PROTOCOL.IPPROTO_TCP
{
_overlapped = (RioNativeOverlapped *)overlapped.ToPointer();
_eventHandle = Kernel32.CreateEvent(IntPtr.Zero, false, false, null);
_adressBuffer = adressBuffer;
_pool = pool;
unsafe
{
var n = (NativeOverlapped *)overlapped.ToPointer();
n->EventHandle = _eventHandle;
}
}
internal RioConnectionOrientedSocket(ulong socketid, IntPtr overlapped, IntPtr adressBuffer, RioConnectionOrientedSocketPool pool, RioFixedBufferPool sendBufferPool, RioFixedBufferPool receiveBufferPool, RioFixedBufferPool adressBufferPool,
uint maxOutstandingReceive, uint maxOutstandingSend, IntPtr SendCompletionQueue, IntPtr ReceiveCompletionQueue,
ADDRESS_FAMILIES adressFam, SOCKET_TYPE sockType, PROTOCOL protocol) :
base(sendBufferPool, receiveBufferPool, adressBufferPool, maxOutstandingReceive, maxOutstandingSend, SendCompletionQueue, ReceiveCompletionQueue,
adressFam, sockType, protocol)
{
_overlapped = (RioNativeOverlapped *)overlapped.ToPointer();
_adressBuffer = adressBuffer;
_pool = pool;
onreadCompletion = id => { if (id == currentId)
{
Interlocked.Decrement(ref pendingRecives);
}
};
onSendCompletion = id => { if (id == currentId)
{
Interlocked.Decrement(ref pendingSends);
}
};
sendTimeout = Stopwatch.Frequency * 5;
receiveTimeout = Stopwatch.Frequency * 5;
currentId = socketid;
}
public RioConnectionlessSocketPool(RioFixedBufferPool sendPool, RioFixedBufferPool revicePool, int socketCount,
uint maxOutstandingReceive = 1024, uint maxOutstandingSend = 1024, uint maxOutsandingCompletions = 1024)
: base(sendPool, revicePool, maxOutstandingReceive, maxOutstandingSend, maxOutsandingCompletions)
{
}
public RioConnectionlessSocketPool(RioFixedBufferPool sendPool, RioFixedBufferPool revicePool, ADDRESS_FAMILIES adressFam, SOCKET_TYPE sockType, PROTOCOL protocol,
uint maxOutstandingReceive = 1024, uint maxOutstandingSend = 1024, uint maxOutsandingCompletions = 1024)
: base(sendPool, revicePool, adressFam, sockType, protocol, maxOutstandingReceive, maxOutstandingSend, maxOutsandingCompletions)
{
}
public unsafe RioSocketPool(RioFixedBufferPool sendPool, RioFixedBufferPool receivePool, ADDRESS_FAMILIES adressFam, SOCKET_TYPE sockType, PROTOCOL protocol,
uint maxOutstandingReceive = 1024, uint maxOutstandingSend = 1024, uint maxOutsandingCompletions = 2048)
{
MaxOutstandingReceive = maxOutstandingReceive;
MaxOutstandingSend = maxOutstandingSend;
MaxOutsandingCompletions = maxOutsandingCompletions;
SendBufferPool = sendPool;
ReceiveBufferPool = receivePool;
this.adressFam = adressFam;
this.sockType = sockType;
this.protocol = protocol;
var version = new Version(2, 2);
WSAData data;
var result = WinSock.WSAStartup((short)version.Raw, out data);
if (result != 0)
WinSock.ThrowLastWSAError();
RioStatic.Initalize();
if ((ReceiveCompletionPort = Kernel32.CreateIoCompletionPort((IntPtr)(-1), IntPtr.Zero, 0, 1)) == IntPtr.Zero)
Kernel32.ThrowLastError();
if ((SendCompletionPort = Kernel32.CreateIoCompletionPort((IntPtr)(-1), IntPtr.Zero, 0, 1)) == IntPtr.Zero)
Kernel32.ThrowLastError();
_sendBufferId = RioStatic.RegisterBuffer(SendBufferPool.BufferPointer, (uint)SendBufferPool.TotalLength);
WinSock.ThrowLastWSAError();
SendBufferPool.SetBufferId(_sendBufferId);
_reciveBufferId = RioStatic.RegisterBuffer(ReceiveBufferPool.BufferPointer, (uint)ReceiveBufferPool.TotalLength);
WinSock.ThrowLastWSAError();
ReceiveBufferPool.SetBufferId(_reciveBufferId);
var sendCompletionMethod = new RIO_NOTIFICATION_COMPLETION()
{
Type = RIO_NOTIFICATION_COMPLETION_TYPE.IOCP_COMPLETION,
Iocp = new RIO_NOTIFICATION_COMPLETION_IOCP()
{
IocpHandle = SendCompletionPort,
QueueCorrelation = 0,
Overlapped = (NativeOverlapped*)-1
}
};
if ((SendCompletionQueue = RioStatic.CreateCompletionQueue(MaxOutsandingCompletions, sendCompletionMethod)) == IntPtr.Zero)
WinSock.ThrowLastWSAError();
var receiveCompletionMethod = new RIO_NOTIFICATION_COMPLETION()
{
Type = RIO_NOTIFICATION_COMPLETION_TYPE.IOCP_COMPLETION,
Iocp = new RIO_NOTIFICATION_COMPLETION_IOCP()
{
IocpHandle = ReceiveCompletionPort,
QueueCorrelation = 0,
Overlapped = (NativeOverlapped*)-1
}
};
if ((ReceiveCompletionQueue = RioStatic.CreateCompletionQueue(MaxOutsandingCompletions, receiveCompletionMethod)) == IntPtr.Zero)
WinSock.ThrowLastWSAError();
Thread receiveThread = new Thread(ProcessReceiveCompletes);
receiveThread.IsBackground = true;
receiveThread.Start();
Thread sendThread = new Thread(ProcessSendCompletes);
sendThread.IsBackground = true;
sendThread.Start();
}
