Rename id helper method.

[akkoma] / lib / pleroma / plugs / http_signature.ex

diff --git a/lib/pleroma/plugs/http_signature.ex b/lib/pleroma/plugs/http_signature.ex
index 17030cdbfcd594339bb87da602eb34ee30601262..38bcd3a782d688d3748e710bd3de912206cff64d 100644 (file)
--- a/lib/pleroma/plugs/http_signature.ex
+++ b/lib/pleroma/plugs/http_signature.ex
@@ -1,19 +1,40 @@
 defmodule Pleroma.Web.Plugs.HTTPSignaturePlug do
   alias Pleroma.Web.HTTPSignatures
+  alias Pleroma.Web.ActivityPub.Utils
   import Plug.Conn
+  require Logger
 
   def init(options) do
     options
   end
 
-  def call(conn, opts) do
-    if get_req_header(conn, "signature") do
-      conn = conn
-      |> put_req_header("(request-target)", String.downcase("#{conn.method} #{conn.request_path}"))
+  def call(%{assigns: %{valid_signature: true}} = conn, _opts) do
+    conn
+  end
+
+  def call(conn, _opts) do
+    user = Utils.get_ap_id(conn.params["actor"])
+    Logger.debug("Checking sig for #{user}")
+    [signature | _] = get_req_header(conn, "signature")
+
+    cond do
+      signature && String.contains?(signature, user) ->
+        conn =
+          conn
+          |> put_req_header(
+            "(request-target)",
+            String.downcase("#{conn.method}") <> " #{conn.request_path}"
+          )
+
+        assign(conn, :valid_signature, HTTPSignatures.validate_conn(conn))
+
+      signature ->
+        Logger.debug("Signature not from actor")
+        assign(conn, :valid_signature, false)
 
-      assign(conn, :valid_signature, HTTPSignatures.validate_conn(conn))
-    else
-      conn
+      true ->
+        Logger.debug("No signature header!")
+        conn
     end
   end
 end