New: Added CAPTCHA support to Rarbg.

2026-04-27 22:57:09 -04:00 · 2016-08-10 20:45:48 +02:00
parent e9eab0ae48
commit 73fb216e6f
36 changed files with 458 additions and 130 deletions
@@ -0,0 +1,56 @@
+using System.Net;
+using System.Text.RegularExpressions;
+using NLog;
+using NzbDrone.Common.Exceptions;
+using NzbDrone.Common.Http;
+
+namespace NzbDrone.Core.Http.CloudFlare
+{
+    public class CloudFlareHttpInterceptor : IHttpRequestInterceptor
+    {
+        private readonly Logger _logger;
+
+        private const string _cloudFlareChallengeScript = "cdn-cgi/scripts/cf.challenge.js";
+        private static readonly Regex _cloudFlareRegex = new Regex(@"data-ray=""(?<Ray>[\w-_]+)"".*?data-sitekey=""(?<SiteKey>[\w-_]+)"".*?data-stoken=""(?<SecretToken>[\w-_]+)""", RegexOptions.Compiled);
+
+        public CloudFlareHttpInterceptor(Logger logger)
+        {
+            _logger = logger;
+        }
+
+        public HttpRequest PreRequest(HttpRequest request)
+        {
+            return request;
+        }
+
+        public HttpResponse PostResponse(HttpResponse response)
+        {
+            if (response.StatusCode == HttpStatusCode.Forbidden && response.Content.Contains(_cloudFlareChallengeScript))
+            {
+                _logger.Error("CloudFlare CAPTCHA block on {0}", response.Request.Url);
+                throw new CloudFlareCaptchaException(response, CreateCaptchaRequest(response));
+            }
+
+            return response;
+        }
+
+        private CloudFlareCaptchaRequest CreateCaptchaRequest(HttpResponse response)
+        {
+            var match = _cloudFlareRegex.Match(response.Content);
+
+            if (!match.Success)
+            {
+                return null;
+            }
+
+            return new CloudFlareCaptchaRequest
+            {
+                Host = response.Request.Url.Host,
+                SiteKey = match.Groups["SiteKey"].Value,
+                Ray = match.Groups["Ray"].Value,
+                SecretToken = match.Groups["SecretToken"].Value,
+                ResponseUrl = response.Request.Url + new HttpUri("/cdn-cgi/l/chk_captcha")
+            };
+        }
+    }
+}