beagled/NetworkService.cs

   1 //
   2 // NetworkService.cs
   3 //
   4 // Copyright (C) 2004 Novell, Inc.
   5 //
   6 // Authors:
   7 //   Fredrik Hedberg (fredrik.hedberg@avafan.com)
   8 //
   9
  10 //
  11 // Permission is hereby granted, free of charge, to any person obtaining a
  12 // copy of this software and associated documentation files (the "Software"),
  13 // to deal in the Software without restriction, including without limitation
  14 // the rights to use, copy, modify, merge, publish, distribute, sublicense,
  15 // and/or sell copies of the Software, and to permit persons to whom the
  16 // Software is furnished to do so, subject to the following conditions:
  17 //
  18 // The above copyright notice and this permission notice shall be included in
  19 // all copies or substantial portions of the Software.
  20 //
  21 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  22 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  23 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  24 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  25 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  26 // FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  27 // DEALINGS IN THE SOFTWARE.
  28 //
  29
  30 using System;
  31 using System.IO;
  32 using System.Xml;
  33 using System.Text;
  34 using System.Threading;
  35
  36 using System.Net;
  37 using System.Net.Sockets;
  38
  39 using System.Reflection;
  40 using System.Collections;
  41
  42 using System.Runtime.Remoting;
  43 using System.Runtime.Remoting.Channels;
  44 using System.Runtime.Remoting.Channels.Tcp;
  45
  46 #if ENABLE_RENDEZVOUS
  47 using Mono.P2p.mDnsResponder;
  48 using Mono.P2p.mDnsResponderApi;
  49 #endif
  50
  51 using Beagle;
  52 using Beagle.Util;
  53
  54 namespace Beagle.Daemon
  55 {
  56         public class NetworkService
  57         {
  58                 private static Logger log = Logger.Get ("Network");
  59
  60                 QueryDriver queryDriver;
  61                 ArrayList authenticators;
  62
  63                 TcpListener server;
  64                 Thread serverThread;
  65
  66                 int localPort = 8505;
  67                 string localAddress = "EVO";
  68                 static string serviceName = "beagle._tcp.local";
  69
  70                 public NetworkService (QueryDriver queryDriver, int localPort)
  71                 {
  72                         this.queryDriver = queryDriver;
  73
  74                         if (localPort != 0)
  75                                 this.localPort = localPort;
  76
  77                         authenticators = new ArrayList();
  78                 }
  79
  80                 // Start the show
  81
  82                 public void Start ()
  83                 {
  84                         log.Debug ("Network service starting");
  85
  86                         ScanForAuthenticators ();
  87
  88                         try {
  89                                 server = new TcpListener (localPort);
  90                                 server.Start();
  91                         } catch (Exception e) {
  92                                 log.Error("Network service could not be started: {0}",e);
  93                                 return;
  94                         }
  95
  96                         serverThread = new Thread (new ThreadStart (Listen));
  97                         serverThread.Start ();
  98
  99                         RegisterWithRendezvous ();
 100
 101                         log.Debug ("Network service started");
 102                 }
 103
 104                 // Shutdown everything
 105
 106                 public void Stop ()
 107                 {
 108                         log.Debug ("Network service stopping");
 109
 110                         serverThread.Suspend ();
 111
 112                         log.Debug ("Network service stopped");
 113                 }
 114
 115                 // Service handling loop
 116
 117                 void Listen ()
 118                 {
 119                         while (true) {
 120                                 TcpClient client = server.AcceptTcpClient ();
 121                                 ServerNetworkHandler handler = new ServerNetworkHandler (client, queryDriver, this);
 122                                 handler.Start ();
 123                         }
 124                 }
 125
 126                 // Register with rendezvous
 127
 128                 void RegisterWithRendezvous ()
 129                 {
 130 #if ENABLE_RENDEZVOUS
 131                         log.Debug ("Starting rendezvous service");
 132
 133                         // Should we start the mDnsResponder service if it's not running?
 134
 135                         try {
 136                                 IRemoteFactory factory = (IRemoteFactory) Activator.GetObject (typeof (IRemoteFactory),
 137                                                "tcp://localhost:8091/mDnsRemoteFactory.tcp");
 138
 139                                 IResourceRegistration rr = factory.GetRegistrationInstance ();
 140
 141                                 rr.RegisterServiceLocation (localAddress,serviceName, localPort, 0, 0);
 142                         }
 143                         catch (RemotingException e)
 144                         {
 145                                 log.Debug ("No multicast service found, not advertising via rendezvous");
 146                         }
 147                         finally
 148                         {
 149                                 log.Debug ("Done registering rendezvous service");
 150                         }
 151 #endif
 152                 }
 153
 154                 // Scan assembly for authenticators
 155
 156                 void ScanForAuthenticators ()
 157                 {
 158                         log.Debug ("Scanning for network authenticators");
 159
 160                         authenticators.Clear ();
 161
 162                         foreach (Type type in Assembly.GetCallingAssembly ().GetTypes ()) {
 163
 164                                 if ( !TypeImplementsInterface (type, typeof (IHitAuthenticator)))
 165                                    continue;
 166
 167                                 log.Debug ("Found network authenticator: {0}", type.ToString ());
 168
 169                                 IHitAuthenticator authenticator = (IHitAuthenticator) Activator.CreateInstance (type);
 170
 171                                 authenticators.Add (authenticator);
 172                         }
 173
 174                         log.Debug ("Done scanning for network authenticators");
 175                 }
 176
 177                 // From QueryDriver.cs
 178
 179                 static bool ThisApiSoVeryIsBroken (Type m, object criteria)
 180                 {
 181                         return m == (Type) criteria;
 182                 }
 183
 184                 static bool TypeImplementsInterface (Type t, Type iface)
 185                 {
 186                         Type[] impls = t.FindInterfaces (new TypeFilter (ThisApiSoVeryIsBroken),
 187                                                          iface);
 188                         return impls.Length > 0;
 189                 }
 190
 191                 // Authenticate a hit
 192
 193                 public ArrayList AuthenticateHit (Hit hit, QueryBody query)
 194                 {
 195                         ArrayList result = new ArrayList ();
 196
 197                         foreach (IHitAuthenticator authenticator in authenticators) {
 198                                 try {
 199                                         result.AddRange (authenticator.Authenticate (hit, query));
 200                                 } catch {
 201                                         // Do nothing, hit not authenticated
 202                                 }
 203                         }
 204                         return result;
 205                 }
 206         }
 207
 208         public interface IHitAuthenticator
 209         {
 210                 ICollection Authenticate (Hit hit, QueryBody query);
 211         }
 212 }