using System.Collections.Generic;
using System.Collections.ObjectModel;
using System.IO;
+using System.Linq;
using System.Net;
using System.Net.Security;
using System.Net.Sockets;
using System.ServiceModel;
using System.ServiceModel.Description;
+using System.ServiceModel.PeerResolvers;
using System.ServiceModel.Security;
using System.Threading;
+using System.Xml;
namespace System.ServiceModel.Channels
{
internal class PeerDuplexChannel : DuplexChannelBase
{
+ enum RemotePeerStatus
+ {
+ None,
+ Connected,
+ Error,
+ }
+
+ class RemotePeerConnection
+ {
+ public RemotePeerConnection (PeerNodeAddress address)
+ {
+ Address = address;
+ }
+
+ public PeerNodeAddress Address { get; private set; }
+ public RemotePeerStatus Status { get; set; }
+ public LocalPeerReceiver Instance { get; set; }
+ public IPeerConnectorClient Channel { get; set; }
+ public ulong NodeId { get; set; }
+ }
+
+ class LocalPeerReceiver : IPeerConnectorContract
+ {
+ List<PeerNodeAddress> connections = new List<PeerNodeAddress> ();
+ AutoResetEvent connect_handle = new AutoResetEvent (false);
+ public event Action<WelcomeInfo> WelcomeReceived;
+
+ public LocalPeerReceiver (PeerDuplexChannel owner)
+ {
+ this.owner = owner;
+ }
+
+ PeerDuplexChannel owner;
+
+ public void Connect (ConnectInfo connect)
+ {
+ if (connect == null)
+ throw new ArgumentNullException ("connect");
+ var ch = OperationContext.Current.GetCallbackChannel<IPeerConnectorContract> ();
+
+ connections.Add (connect.Address);
+ // FIXME: check and reject if inappropriate. For example, maximum connection exceeded.
+ using (var octx = new OperationContextScope ((IContextChannel) ch)) {
+ OperationContext.Current.OutgoingMessageHeaders.To = new Uri (Constants.WsaAnonymousUri);
+ if (!owner.peers.Any (p => p.Address.EndpointAddress.Equals (connect.Address.EndpointAddress)))
+ owner.peers.Add (new RemotePeerConnection (connect.Address));
+ ch.Welcome (new WelcomeInfo () { NodeId = owner.node.NodeId });
+ }
+ }
+
+ internal void WaitForConnectResponse (TimeSpan timeout)
+ {
+ if (!connect_handle.WaitOne (timeout))
+ throw new TimeoutException ();
+ }
+
+ public void Disconnect (DisconnectInfo disconnect)
+ {
+ if (disconnect == null)
+ throw new ArgumentNullException ("disconnect");
+ // Console.WriteLine ("DisconnectInfo.Reason: " + disconnect.Reason);
+ // FIXME: handle disconnection in practice. So far I see nothing to do.
+ }
+
+ public void Welcome (WelcomeInfo welcome)
+ {
+ if (WelcomeReceived != null)
+ WelcomeReceived (welcome);
+ connect_handle.Set ();
+ }
+
+ public void Refuse (RefuseInfo refuse)
+ {
+ // FIXME: it should not probably actually throw an error.
+ connect_handle.Set ();
+ throw new InvalidOperationException ("Peer connection was refused");
+ }
+
+ public void LinkUtility (LinkUtilityInfo linkUtility)
+ {
+ throw new NotImplementedException ();
+ }
+
+ public void Ping ()
+ {
+ throw new NotImplementedException ();
+ }
+
+ public void SendMessage (Message msg)
+ {
+ int idx = msg.Headers.FindHeader ("PeerTo", Constants.NetPeer);
+ if (idx >= 0)
+ msg.Headers.To = msg.Headers.GetHeader<Uri> (idx);
+ // FIXME: anything to do for PeerVia?
+
+ owner.EnqueueMessage (msg);
+ }
+ }
+
+ interface IPeerConnectorClient : IClientChannel, IPeerConnectorContract
+ {
+ }
+
+ IChannelFactory<IDuplexSessionChannel> client_factory;
PeerTransportBindingElement binding;
- EndpointAddress local_address;
PeerResolver resolver;
PeerNode node;
- TcpListener listener;
+ ServiceHost listener_host;
TcpChannelInfo info;
- List<PeerNodeAddress> peers = new List<PeerNodeAddress> ();
+ List<RemotePeerConnection> peers = new List<RemotePeerConnection> ();
+ PeerNodeAddress local_node_address;
public PeerDuplexChannel (IPeerChannelManager factory, EndpointAddress address, Uri via, PeerResolver resolver)
: base ((ChannelFactoryBase) factory, address, via)
// It could be opened even with empty list of PeerNodeAddresses.
// So, do not create PeerNode per PeerNodeAddress, but do it with PeerNodeAddress[].
- node = new PeerNodeImpl (RemoteAddress, factory.Source.ListenIPAddress, factory.Source.Port);
+ node = new PeerNodeImpl (RemoteAddress.Uri.Host, factory.Source.ListenIPAddress, factory.Source.Port);
}
- // FIXME: receive local_address too
public PeerDuplexChannel (IPeerChannelManager listener)
: base ((ChannelListenerBase) listener)
{
this.resolver = listener.Resolver;
info = new TcpChannelInfo (binding, listener.MessageEncoder, null); // FIXME: fill properties correctly.
- node = new PeerNodeImpl (new EndpointAddress (((ChannelListenerBase) listener).Uri), listener.Source.ListenIPAddress, listener.Source.Port);
- }
-
- public override EndpointAddress LocalAddress {
- get { return local_address; }
+ node = new PeerNodeImpl (((ChannelListenerBase) listener).Uri.Host, listener.Source.ListenIPAddress, listener.Source.Port);
}
public override T GetProperty<T> ()
// DuplexChannelBase
- TcpDuplexSessionChannel CreateInnerChannel (PeerNodeAddress pna)
+ IPeerConnectorClient CreateInnerClient (RemotePeerConnection conn)
{
- var cfb = Manager as ChannelFactoryBase;
- if (cfb != null)
- return new TcpDuplexSessionChannel (cfb, info, pna.EndpointAddress, Via);
- else
- return new TcpDuplexSessionChannel ((ChannelListenerBase) Manager, info, listener.AcceptTcpClient ());
+ conn.Instance = new LocalPeerReceiver (this);
+ conn.Instance.WelcomeReceived += delegate (WelcomeInfo welcome) {
+ conn.NodeId = welcome.NodeId;
+ // FIXME: handle referrals
+ };
+
+ // FIXME: pass more setup parameters
+ var binding = new NetTcpBinding ();
+ binding.Security.Mode = SecurityMode.None;
+ var channel_factory = new DuplexChannelFactory<IPeerConnectorClient> (conn.Instance, binding);
+ channel_factory.Open ();
+
+ var ch = channel_factory.CreateChannel (new EndpointAddress ("net.p2p://" + node.MeshId + "/"), conn.Address.EndpointAddress.Uri);
+ ch.Closed += delegate {
+ channel_factory.Close ();
+ };
+ return ch;
}
public override void Send (Message message, TimeSpan timeout)
DateTime start = DateTime.Now;
- foreach (var pna in peers) {
- var inner = CreateInnerChannel (pna);
- inner.Open (timeout - (DateTime.Now - start));
- inner.Send (message, timeout);
+ // FIXME: give max buffer size
+ var mb = message.CreateBufferedCopy (0x10000);
+
+ for (int i = 0; i < peers.Count; i++) {
+ var pc = peers [i];
+ message = mb.CreateMessage ();
+
+ if (pc.Status == RemotePeerStatus.None) {
+ pc.Status = RemotePeerStatus.Error; // prepare for cases that it resulted in an error in the middle.
+ var inner = CreateInnerClient (pc);
+ pc.Channel = inner;
+ inner.Open (timeout - (DateTime.Now - start));
+ inner.OperationTimeout = timeout - (DateTime.Now - start);
+ inner.Connect (new ConnectInfo () { Address = local_node_address, NodeId = (uint) node.NodeId });
+ pc.Instance.WaitForConnectResponse (timeout - (DateTime.Now - start));
+ pc.Status = RemotePeerStatus.Connected;
+ }
+
+ pc.Channel.OperationTimeout = timeout - (DateTime.Now - start);
+
+ // see [MC-PRCH] 3.2.4.1
+ if (message.Headers.MessageId == null)
+ message.Headers.MessageId = new UniqueId ();
+ message.Headers.Add (MessageHeader.CreateHeader ("PeerTo", Constants.NetPeer, RemoteAddress.Uri));
+ message.Headers.Add (MessageHeader.CreateHeader ("PeerVia", Constants.NetPeer, RemoteAddress.Uri));
+ message.Headers.Add (MessageHeader.CreateHeader ("FloodMessage", Constants.NetPeer, "PeerFlooder"));
+ pc.Channel.SendMessage (message);
}
}
- public override Message Receive (TimeSpan timeout)
+ internal void EnqueueMessage (Message message)
+ {
+ queue.Enqueue (message);
+ receive_handle.Set ();
+ }
+
+ Queue<Message> queue = new Queue<Message> ();
+ AutoResetEvent receive_handle = new AutoResetEvent (false);
+
+ public override bool TryReceive (TimeSpan timeout, out Message message)
{
ThrowIfDisposedOrNotOpen ();
+ DateTime start = DateTime.Now;
- throw new NotImplementedException ();
+ if (queue.Count > 0 || receive_handle.WaitOne (timeout)) {
+ message = queue.Dequeue ();
+ return message == null;
+ } else {
+ message = null;
+ return false;
+ }
}
public override bool WaitForMessage (TimeSpan timeout)
protected override void OnAbort ()
{
+ if (client_factory != null) {
+ client_factory.Abort ();
+ client_factory = null;
+ }
OnClose (TimeSpan.Zero);
}
protected override void OnClose (TimeSpan timeout)
{
DateTime start = DateTime.Now;
+ if (client_factory != null)
+ client_factory.Close (timeout - (DateTime.Now - start));
peers.Clear ();
resolver.Unregister (node.RegisteredId, timeout - (DateTime.Now - start));
node.SetOffline ();
- if (listener != null)
- listener.Stop ();
+ if (listener_host != null)
+ listener_host.Close (timeout - (DateTime.Now - start));
node.RegisteredId = null;
}
+
protected override void OnOpen (TimeSpan timeout)
{
DateTime start = DateTime.Now;
// FIXME: supply maxAddresses
- peers.AddRange (resolver.Resolve (node.MeshId, 3, timeout));
+ foreach (var a in resolver.Resolve (node.MeshId, 3, timeout))
+ peers.Add (new RemotePeerConnection (a));
+
+ // FIXME: pass more configuration
+ var binding = new NetTcpBinding ();
+ binding.Security.Mode = SecurityMode.None;
+
+ int port = 0;
+ var rnd = new Random ();
+ for (int i = 0; i < 1000; i++) {
+ if (DateTime.Now - start > timeout)
+ throw new TimeoutException ();
+ try {
+ port = rnd.Next (50000, 51000);
+ var t = new TcpListener (port);
+ t.Start ();
+ t.Stop ();
+ break;
+ } catch (SocketException) {
+ continue;
+ }
+ }
- listener = node.GetTcpListener ();
- var ep = (IPEndPoint) listener.LocalEndpoint;
string name = Dns.GetHostName ();
- var nid = new Random ().Next (0, int.MaxValue);
- var ea = new EndpointAddress ("net.tcp://" + name + ":" + ep.Port + "/PeerChannelEndpoints/" + Guid.NewGuid ());
- var pna = new PeerNodeAddress (ea, new ReadOnlyCollection<IPAddress> (Dns.GetHostEntry (ep.Address).AddressList));
+ var uri = new Uri ("net.tcp://" + name + ":" + port + "/PeerChannelEndpoints/" + Guid.NewGuid ());
+
+ var peer_receiver = new LocalPeerReceiver (this);
+ listener_host = new ServiceHost (peer_receiver);
+ var sba = listener_host.Description.Behaviors.Find<ServiceBehaviorAttribute> ();
+ sba.InstanceContextMode = InstanceContextMode.Single;
+ sba.IncludeExceptionDetailInFaults = true;
+
+ var se = listener_host.AddServiceEndpoint (typeof (IPeerConnectorContract).FullName, binding, "net.p2p://" + node.MeshId + "/");
+ se.ListenUri = uri;
+
+ // FIXME: remove debugging code
+ listener_host.UnknownMessageReceived += delegate (object obj, UnknownMessageReceivedEventArgs earg) { Console.WriteLine ("%%%%% UNKOWN MESSAGE " + earg.Message); };
+
+ listener_host.Open (timeout - (DateTime.Now - start));
+
+ var nid = (ulong) new Random ().Next (0, int.MaxValue);
+ var ea = new EndpointAddress (uri);
+ var pna = new PeerNodeAddress (ea, new ReadOnlyCollection<IPAddress> (Dns.GetHostEntry (name).AddressList));
+ local_node_address = pna;
node.RegisteredId = resolver.Register (node.MeshId, pna, timeout - (DateTime.Now - start));
node.NodeId = nid;
// Add itself to the local list as well.
// FIXME: it might become unnecessary once it implemented new node registration from peer resolver service.
- peers.Add (pna);
+ peers.Add (new RemotePeerConnection (pna));
node.SetOnline ();
}
projects / mono.git / blobdiff