public SimConn(SimSocket socket, SimEndpoint remoteAddress, SimProc proc, uint seq, uint ackNumber)
{
_socket = socket;
RemoteAddress = remoteAddress;
_proc = proc;
_sequenceNumber = seq;
_ackNumber = ackNumber;
}
public SimPacket(SimEndpoint source, SimEndpoint destination, object payload,
uint seqNumber,
uint ackNumber,
SimFlag flag)
{
Source = source;
Destination = destination;
Payload = payload;
SeqNumber = seqNumber;
AckNumber = ackNumber;
Flag = flag;
}
public async Task <IConn> Connect(SimProc process, SimEndpoint destination)
{
SimRoute route;
if (!Cluster.TryGetRoute(Name, destination.Machine, out route))
{
throw new IOException($"Route not found");
}
// todo: allocate port
// todo: allow Azure SNAT delay scenario
var socketId = NextSocketID();
var source = new SimEndpoint(Name, socketId);
var clientSocket = new SimSocket(process, source, Cluster);
try {
_sockets.Add(socketId, clientSocket);
var conn = new SimConn(clientSocket, destination, process, 0, 0);
clientSocket._connections.Add(destination, conn);
// handshake
await conn.Write(null, SimFlag.Syn);
var response = await conn.Read(5.Sec());
if (response.Flag != (SimFlag.Ack | SimFlag.Syn))
{
Debug(LogType.Warning, $"Bad handshake: {response.BodyString()}");
await conn.Write(null, SimFlag.Reset);
clientSocket._connections.Remove(destination);
throw new IOException($"Failed to connect (got {response.BodyString()})");
}
await conn.Write(null, SimFlag.Ack);
return(new ClientConn(conn));
} catch (IOException) {
throw;
}
catch (Exception ex) {
_sockets.Remove(socketId);
clientSocket.Dispose();
throw new IOException($"Failed to connect: {ex.Message}", ex);
}
}
public async Task <ISocket> Bind(SimProc proc, ushort port, TimeSpan timeout)
{
// socket is bound to the owner
var endpoint = new SimEndpoint(proc.Id.Machine, port);
if (_sockets.ContainsKey(port))
{
throw new IOException($"Address {endpoint} in use");
}
var socket = new SimSocket(proc, endpoint, Cluster);
_sockets.Add(port, socket);
proc.RegisterSocket(socket);
return(socket);
}
public SimSocket(SimProc proc, SimEndpoint endpoint, SimCluster net)
{
_proc = proc;
Endpoint = endpoint;
_net = net;
}
public async Task <IConn> Connect(SimEndpoint endpoint)
{
return(await Machine.Connect(this, endpoint));
}
