XmlRpcBasicDOSProtector.cs 3.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091
  1. /*
  2. * Copyright (c) Contributors, http://opensimulator.org/
  3. * See CONTRIBUTORS.TXT for a full list of copyright holders.
  4. *
  5. * Redistribution and use in source and binary forms, with or without
  6. * modification, are permitted provided that the following conditions are met:
  7. * * Redistributions of source code must retain the above copyright
  8. * notice, this list of conditions and the following disclaimer.
  9. * * Redistributions in binary form must reproduce the above copyright
  10. * notice, this list of conditions and the following disclaimer in the
  11. * documentation and/or other materials provided with the distribution.
  12. * * Neither the name of the OpenSimulator Project nor the
  13. * names of its contributors may be used to endorse or promote products
  14. * derived from this software without specific prior written permission.
  15. *
  16. * THIS SOFTWARE IS PROVIDED BY THE DEVELOPERS ``AS IS'' AND ANY
  17. * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  18. * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  19. * DISCLAIMED. IN NO EVENT SHALL THE CONTRIBUTORS BE LIABLE FOR ANY
  20. * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  21. * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  22. * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  23. * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  24. * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  25. * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  26. */
  27. using System.Net;
  28. using Nwc.XmlRpc;
  29. using OpenSim.Framework;
  30. namespace OpenSim.Framework.Servers.HttpServer
  31. {
  32. public class XmlRpcBasicDOSProtector
  33. {
  34. private readonly XmlRpcMethod _normalMethod;
  35. private readonly XmlRpcMethod _throttledMethod;
  36. private readonly BasicDosProtectorOptions _options;
  37. private readonly BasicDOSProtector _dosProtector;
  38. public XmlRpcBasicDOSProtector(XmlRpcMethod normalMethod, XmlRpcMethod throttledMethod,BasicDosProtectorOptions options)
  39. {
  40. _normalMethod = normalMethod;
  41. _throttledMethod = throttledMethod;
  42. _options = options;
  43. _dosProtector = new BasicDOSProtector(_options);
  44. }
  45. public XmlRpcResponse Process(XmlRpcRequest request, IPEndPoint client)
  46. {
  47. XmlRpcResponse resp = null;
  48. string clientstring = GetClientString(request, client);
  49. string endpoint = GetEndPoint(request, client);
  50. if (_dosProtector.Process(clientstring, endpoint))
  51. resp = _normalMethod(request, client);
  52. else
  53. resp = _throttledMethod(request, client);
  54. if (_options.MaxConcurrentSessions > 0)
  55. _dosProtector.ProcessEnd(clientstring, endpoint);
  56. return resp;
  57. }
  58. private string GetClientString(XmlRpcRequest request, IPEndPoint client)
  59. {
  60. string clientstring;
  61. if (_options.AllowXForwardedFor && request.Params.Count > 3)
  62. {
  63. object headerstr = request.Params[3];
  64. if (headerstr != null && !string.IsNullOrEmpty(headerstr.ToString()))
  65. clientstring = request.Params[3].ToString();
  66. else
  67. clientstring = client.Address.ToString();
  68. }
  69. else
  70. clientstring = client.Address.ToString();
  71. return clientstring;
  72. }
  73. private string GetEndPoint(XmlRpcRequest request, IPEndPoint client)
  74. {
  75. return client.Address.ToString();
  76. }
  77. }
  78. }