Browse Source

Fix ThreadResolveWorker getting queued with invalid URLs (#9628)

Eugen Rochko 5 years ago
parent
commit
aa9a20cde0
3 changed files with 14 additions and 2 deletions
  1. 1 1
      app/lib/activitypub/activity/create.rb
  2. 1 1
      app/lib/ostatus/activity/creation.rb
  3. 12 0
      app/lib/request.rb

+ 1 - 1
app/lib/activitypub/activity/create.rb

@@ -210,7 +210,7 @@ class ActivityPub::Activity::Create < ActivityPub::Activity
   end
 
   def resolve_thread(status)
-    return unless status.reply? && status.thread.nil?
+    return unless status.reply? && status.thread.nil? && Request.valid_url?(in_reply_to_uri)
     ThreadResolveWorker.perform_async(status.id, in_reply_to_uri)
   end
 

+ 1 - 1
app/lib/ostatus/activity/creation.rb

@@ -57,7 +57,7 @@ class OStatus::Activity::Creation < OStatus::Activity::Base
       save_emojis(status)
     end
 
-    if thread? && status.thread.nil?
+    if thread? && status.thread.nil? && Request.valid_url?(thread.second)
       Rails.logger.debug "Trying to attach #{status.id} (#{id}) to #{thread.first}"
       ThreadResolveWorker.perform_async(status.id, thread.second)
     end

+ 12 - 0
app/lib/request.rb

@@ -66,6 +66,18 @@ class Request
     (@account ? @headers.merge('Signature' => signature) : @headers).without(REQUEST_TARGET)
   end
 
+  class << self
+    def valid_url?(url)
+      begin
+        parsed_url = Addressable::URI.parse(url)
+      rescue Addressable::URI::InvalidURIError
+        return false
+      end
+
+      %w(http https).include?(parsed_url.scheme) && parsed_url.host.present?
+    end
+  end
+
   private
 
   def set_common_headers!