git.squeep.com Git - akkoma/blob - lib/pleroma/plugs/authentication_plug.ex

   1 defmodule Pleroma.Plugs.AuthenticationPlug do
   2   import Plug.Conn
   3
   4   def init(options) do
   5     options
   6   end
   7
   8   def call(conn, opts) do
   9     with {:ok, username, password} <- decode_header(conn),
  10          {:ok, user} <- opts[:fetcher].(username),
  11          {:ok, verified_user} <- verify(user, password)
  12     do
  13       conn |> assign(:user, verified_user)
  14     else
  15       _ -> conn |> halt_or_continue(opts)
  16     end
  17   end
  18
  19   defp verify(nil, _password) do
  20     Comeonin.Pbkdf2.dummy_checkpw
  21     :error
  22   end
  23
  24   defp verify(user, password) do
  25     if Comeonin.Pbkdf2.checkpw(password, user.password_hash) do
  26       {:ok, user}
  27     else
  28       :error
  29     end
  30   end
  31
  32   defp decode_header(conn) do
  33     with ["Basic " <> header] <- get_req_header(conn, "authorization"),
  34          {:ok, userinfo} <- Base.decode64(header),
  35          [username, password] <- String.split(userinfo, ":")
  36     do
  37       { :ok, username, password }
  38     end
  39   end
  40
  41   defp halt_or_continue(conn, %{optional: true}) do
  42     conn |> assign(:user, nil)
  43   end
  44
  45   defp halt_or_continue(conn, _) do
  46     conn
  47     |> put_resp_content_type("application/json")
  48     |> send_resp(403, Poison.encode!(%{error: "Invalid credentials."}))
  49     |> halt
  50   end
  51 end