mcs/class/System/System.Net/EndPointListener.cs

   1 //
   2 // System.Net.EndPointListener
   3 //
   4 // Author:
   5 //      Gonzalo Paniagua Javier (gonzalo@novell.com)
   6 //
   7 // Copyright (c) 2005 Novell, Inc. (http://www.novell.com)
   8 //
   9 // Permission is hereby granted, free of charge, to any person obtaining
  10 // a copy of this software and associated documentation files (the
  11 // "Software"), to deal in the Software without restriction, including
  12 // without limitation the rights to use, copy, modify, merge, publish,
  13 // distribute, sublicense, and/or sell copies of the Software, and to
  14 // permit persons to whom the Software is furnished to do so, subject to
  15 // the following conditions:
  16 //
  17 // The above copyright notice and this permission notice shall be
  18 // included in all copies or substantial portions of the Software.
  19 //
  20 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  21 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  22 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  23 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  24 // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  25 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  26 // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  27 //
  28
  29 #if NET_2_0 && SECURITY_DEP
  30
  31 using System.IO;
  32 using System.Net.Sockets;
  33 using System.Collections.Generic;
  34 using System.Security.Cryptography;
  35 using System.Security.Cryptography.X509Certificates;
  36 using Mono.Security.Authenticode;
  37
  38 namespace System.Net {
  39         sealed class EndPointListener
  40         {
  41                 IPEndPoint endpoint;
  42                 Socket sock;
  43                 Dictionary<ListenerPrefix, HttpListener> prefixes;
  44                 List<ListenerPrefix> unhandled; // host = '*'
  45                 List<ListenerPrefix> all; // host = '+'
  46                 X509Certificate2 cert;
  47                 AsymmetricAlgorithm key;
  48                 bool secure;
  49
  50                 public EndPointListener (IPAddress addr, int port, bool secure)
  51                 {
  52                         if (secure) {
  53                                 this.secure = secure;
  54                                 LoadCertificateAndKey (addr, port);
  55                         }
  56
  57                         endpoint = new IPEndPoint (addr, port);
  58                         sock = new Socket (addr.AddressFamily, SocketType.Stream, ProtocolType.Tcp);
  59                         sock.Bind (endpoint);
  60                         sock.Listen (500);
  61                         sock.BeginAccept (OnAccept, this);
  62                         prefixes = new Dictionary<ListenerPrefix, HttpListener> ();
  63                 }
  64
  65                 void LoadCertificateAndKey (IPAddress addr, int port)
  66                 {
  67                         // Actually load the certificate
  68                         try {
  69                                 string dirname = Environment.GetFolderPath (Environment.SpecialFolder.ApplicationData);
  70                                 string path = Path.Combine (dirname, ".mono");
  71                                 path = Path.Combine (path, "httplistener");
  72                                 string cert_file = Path.Combine (path, String.Format ("{0}.cer", port));
  73                                 string pvk_file = Path.Combine (path, String.Format ("{0}.pvk", port));
  74                                 cert = new X509Certificate2 (cert_file);
  75                                 key = PrivateKey.CreateFromFile (pvk_file).RSA;
  76                         } catch {
  77                                 // ignore errors
  78                         }
  79                 }
  80
  81                 static void OnAccept (IAsyncResult ares)
  82                 {
  83                         EndPointListener epl = (EndPointListener) ares.AsyncState;
  84                         Socket accepted = null;
  85                         try {
  86                                 accepted = epl.sock.EndAccept (ares);
  87                         } catch {
  88                                 // Anything to do here?
  89                         } finally {
  90                                 try {
  91                                         epl.sock.BeginAccept (OnAccept, epl);
  92                                 } catch {
  93                                         if (accepted != null) {
  94                                                 try {
  95                                                         accepted.Close ();
  96                                                 } catch {}
  97                                                 accepted = null;
  98                                         }
  99                                 }
 100                         }
 101
 102                         if (accepted == null)
 103                                 return;
 104
 105                         if (epl.secure && (epl.cert == null || epl.key == null)) {
 106                                 accepted.Close ();
 107                                 return;
 108                         }
 109                         HttpConnection conn = new HttpConnection (accepted, epl, epl.secure, epl.cert, epl.key);
 110                         conn.BeginReadRequest ();
 111                 }
 112
 113                 public bool BindContext (HttpListenerContext context)
 114                 {
 115                         HttpListenerRequest req = context.Request;
 116                         ListenerPrefix prefix;
 117                         HttpListener listener = SearchListener (req.UserHostName, req.RawUrl, out prefix);
 118                         if (listener == null)
 119                                 return false;
 120
 121                         context.Listener = listener;
 122                         context.Connection.Prefix = prefix;
 123                         listener.RegisterContext (context);
 124                         return true;
 125                 }
 126
 127                 public void UnbindContext (HttpListenerContext context)
 128                 {
 129                         if (context == null || context.Request == null)
 130                                 return;
 131
 132                         HttpListenerRequest req = context.Request;
 133                         ListenerPrefix prefix;
 134                         HttpListener listener = SearchListener (req.UserHostName, req.RawUrl, out prefix);
 135                         if (listener != null)
 136                                 listener.UnregisterContext (context);
 137                 }
 138
 139                 HttpListener SearchListener (string host, string raw_url, out ListenerPrefix prefix)
 140                 {
 141                         prefix = null;
 142                         if (raw_url == null)
 143                                 return null;
 144
 145                         //TODO: We should use a ReaderWriterLock between this and the add/remove operations.
 146                         if (host != null) {
 147                                 int colon = host.IndexOf (':');
 148                                 if (colon >= 0)
 149                                         host = host.Substring (0, colon);
 150                         }
 151
 152                         string path = HttpUtility.UrlDecode (raw_url);
 153                         string path_slash = path [path.Length - 1] == '/' ? path : path + "/";
 154
 155                         HttpListener best_match = null;
 156                         int best_length = -1;
 157
 158                         lock (prefixes) {
 159                                 if (host != null && host != "") {
 160                                         foreach (ListenerPrefix p in prefixes.Keys) {
 161                                                 string ppath = p.Path;
 162                                                 if (ppath.Length < best_length)
 163                                                         continue;
 164
 165                                                 if (p.Host == host && (path.StartsWith (ppath) || path_slash.StartsWith (ppath))) {
 166                                                         best_length = ppath.Length;
 167                                                         best_match = prefixes [p];
 168                                                         prefix = p;
 169                                                 }
 170                                         }
 171                                         if (best_length != -1)
 172                                                 return best_match;
 173                                 }
 174
 175                                 best_match = MatchFromList (host, path, unhandled, out prefix);
 176                                 if (best_match != null)
 177                                         return best_match;
 178
 179                                 best_match = MatchFromList (host, path, all, out prefix);
 180                                 if (best_match != null)
 181                                         return best_match;
 182                         }
 183                         return null;
 184                 }
 185
 186                 HttpListener MatchFromList (string host, string path, List<ListenerPrefix> list, out ListenerPrefix prefix)
 187                 {
 188                         prefix = null;
 189                         if (list == null)
 190                                 return null;
 191
 192                         HttpListener best_match = null;
 193                         int best_length = -1;
 194
 195                         foreach (ListenerPrefix p in list) {
 196                                 string ppath = p.Path;
 197                                 if (ppath.Length < best_length)
 198                                         continue;
 199
 200                                 if (path.StartsWith (ppath)) {
 201                                         best_length = ppath.Length;
 202                                         best_match = p.Listener;
 203                                         prefix = p;
 204                                 }
 205                         }
 206
 207                         return best_match;
 208                 }
 209
 210                 void AddSpecial (List<ListenerPrefix> coll, ListenerPrefix prefix)
 211                 {
 212                         if (coll == null)
 213                                 return;
 214
 215                         foreach (ListenerPrefix p in coll) {
 216                                 if (p.Path == prefix.Path) //TODO: code
 217                                         throw new HttpListenerException (400, "Prefix already in use.");
 218                         }
 219
 220                         coll.Add (prefix);
 221                 }
 222
 223                 void RemoveSpecial (List<ListenerPrefix> coll, ListenerPrefix prefix)
 224                 {
 225                         if (coll == null)
 226                                 return;
 227
 228                         int c = coll.Count;
 229                         for (int i = 0; i < c; i++) {
 230                                 ListenerPrefix p = coll [i];
 231                                 if (p.Path == prefix.Path) {
 232                                         coll.RemoveAt (i);
 233                                         CheckIfRemove ();
 234                                         return;
 235                                 }
 236                         }
 237                 }
 238
 239                 void CheckIfRemove ()
 240                 {
 241                         if (prefixes.Count > 0)
 242                                 return;
 243
 244                         if (unhandled != null && unhandled.Count > 0)
 245                                 return;
 246
 247                         if (all != null && all.Count > 0)
 248                                 return;
 249
 250                         EndPointManager.RemoveEndPoint (this, endpoint);
 251                 }
 252
 253                 public void Close ()
 254                 {
 255                         sock.Close ();
 256                 }
 257
 258                 public void AddPrefix (ListenerPrefix prefix, HttpListener listener)
 259                 {
 260                         lock (prefixes) {
 261                                 if (prefix.Host == "*") {
 262                                         if (unhandled == null)
 263                                                 unhandled = new List<ListenerPrefix> ();
 264
 265                                         prefix.Listener = listener;
 266                                         AddSpecial (unhandled, prefix);
 267                                         return;
 268                                 }
 269
 270                                 if (prefix.Host == "+") {
 271                                         if (all == null)
 272                                                 all = new List<ListenerPrefix> ();
 273                                         prefix.Listener = listener;
 274                                         AddSpecial (all, prefix);
 275                                         return;
 276                                 }
 277
 278                                 if (prefixes.ContainsKey (prefix)) {
 279                                         HttpListener other = prefixes [prefix];
 280                                         if (other != listener) // TODO: code.
 281                                                 throw new HttpListenerException (400, "There's another listener for " + prefix);
 282                                         return;
 283                                 }
 284
 285                                 prefixes [prefix] = listener;
 286                         }
 287                 }
 288
 289                 public void RemovePrefix (ListenerPrefix prefix, HttpListener listener)
 290                 {
 291                         lock (prefixes) {
 292                                 if (prefix.Host == "*") {
 293                                         RemoveSpecial (unhandled, prefix);
 294                                         return;
 295                                 }
 296
 297                                 if (prefix.Host == "+") {
 298                                         RemoveSpecial (all, prefix);
 299                                         return;
 300                                 }
 301
 302                                 if (prefixes.ContainsKey (prefix)) {
 303                                         prefixes.Remove (prefix);
 304                                 }
 305                         }
 306                 }
 307         }
 308 }
 309 #endif
 310