git.squeep.com Git - akkoma/blob - test/pleroma/web/plugs/http_security_plug_test.exs

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.Web.Plugs.HTTPSecurityPlugTest do
   6   use Pleroma.Web.ConnCase
   7
   8   alias Pleroma.Config
   9   alias Plug.Conn
  10
  11   describe "http security enabled" do
  12     setup do: clear_config([:http_security, :enabled], true)
  13
  14     test "it sends CSP headers when enabled", %{conn: conn} do
  15       conn = get(conn, "/api/v1/instance")
  16
  17       refute Conn.get_resp_header(conn, "x-xss-protection") == []
  18       refute Conn.get_resp_header(conn, "x-permitted-cross-domain-policies") == []
  19       refute Conn.get_resp_header(conn, "x-frame-options") == []
  20       refute Conn.get_resp_header(conn, "x-content-type-options") == []
  21       refute Conn.get_resp_header(conn, "x-download-options") == []
  22       refute Conn.get_resp_header(conn, "referrer-policy") == []
  23       refute Conn.get_resp_header(conn, "content-security-policy") == []
  24     end
  25
  26     test "it sends STS headers when enabled", %{conn: conn} do
  27       clear_config([:http_security, :sts], true)
  28
  29       conn = get(conn, "/api/v1/instance")
  30
  31       refute Conn.get_resp_header(conn, "strict-transport-security") == []
  32       refute Conn.get_resp_header(conn, "expect-ct") == []
  33     end
  34
  35     test "it does not send STS headers when disabled", %{conn: conn} do
  36       clear_config([:http_security, :sts], false)
  37
  38       conn = get(conn, "/api/v1/instance")
  39
  40       assert Conn.get_resp_header(conn, "strict-transport-security") == []
  41       assert Conn.get_resp_header(conn, "expect-ct") == []
  42     end
  43
  44     test "referrer-policy header reflects configured value", %{conn: conn} do
  45       resp = get(conn, "/api/v1/instance")
  46
  47       assert Conn.get_resp_header(resp, "referrer-policy") == ["same-origin"]
  48
  49       clear_config([:http_security, :referrer_policy], "no-referrer")
  50
  51       resp = get(conn, "/api/v1/instance")
  52
  53       assert Conn.get_resp_header(resp, "referrer-policy") == ["no-referrer"]
  54     end
  55
  56     test "it sends `report-to` & `report-uri` CSP response headers", %{conn: conn} do
  57       conn = get(conn, "/api/v1/instance")
  58
  59       [csp] = Conn.get_resp_header(conn, "content-security-policy")
  60
  61       assert csp =~ ~r|report-uri https://endpoint.com;report-to csp-endpoint;|
  62
  63       [reply_to] = Conn.get_resp_header(conn, "reply-to")
  64
  65       assert reply_to ==
  66                "{\"endpoints\":[{\"url\":\"https://endpoint.com\"}],\"group\":\"csp-endpoint\",\"max-age\":10886400}"
  67     end
  68
  69     test "default values for img-src and media-src with disabled media proxy", %{conn: conn} do
  70       conn = get(conn, "/api/v1/instance")
  71
  72       [csp] = Conn.get_resp_header(conn, "content-security-policy")
  73       assert csp =~ "media-src 'self' https:;"
  74       assert csp =~ "img-src 'self' data: blob: https:;"
  75     end
  76   end
  77
  78   describe "img-src and media-src" do
  79     setup do
  80       clear_config([:http_security, :enabled], true)
  81       clear_config([:media_proxy, :enabled], true)
  82       clear_config([:media_proxy, :proxy_opts, :redirect_on_failure], false)
  83     end
  84
  85     test "media_proxy with base_url", %{conn: conn} do
  86       url = "https://example.com"
  87       clear_config([:media_proxy, :base_url], url)
  88       assert_media_img_src(conn, url)
  89     end
  90
  91     test "upload with base url", %{conn: conn} do
  92       url = "https://example2.com"
  93       clear_config([Pleroma.Upload, :base_url], url)
  94       assert_media_img_src(conn, url)
  95     end
  96
  97     test "with S3 public endpoint", %{conn: conn} do
  98       url = "https://example3.com"
  99       clear_config([Pleroma.Uploaders.S3, :public_endpoint], url)
 100       assert_media_img_src(conn, url)
 101     end
 102
 103     test "with captcha endpoint", %{conn: conn} do
 104       clear_config([Pleroma.Captcha.Mock, :endpoint], "https://captcha.com")
 105       assert_media_img_src(conn, "https://captcha.com")
 106     end
 107
 108     test "with media_proxy whitelist", %{conn: conn} do
 109       clear_config([:media_proxy, :whitelist], ["https://example6.com", "https://example7.com"])
 110       assert_media_img_src(conn, "https://example7.com https://example6.com")
 111     end
 112
 113     # TODO: delete after removing support bare domains for media proxy whitelist
 114     test "with media_proxy bare domains whitelist (deprecated)", %{conn: conn} do
 115       clear_config([:media_proxy, :whitelist], ["example4.com", "example5.com"])
 116       assert_media_img_src(conn, "example5.com example4.com")
 117     end
 118   end
 119
 120   defp assert_media_img_src(conn, url) do
 121     conn = get(conn, "/api/v1/instance")
 122     [csp] = Conn.get_resp_header(conn, "content-security-policy")
 123     assert csp =~ "media-src 'self' #{url};"
 124     assert csp =~ "img-src 'self' data: blob: #{url};"
 125   end
 126
 127   test "it does not send CSP headers when disabled", %{conn: conn} do
 128     clear_config([:http_security, :enabled], false)
 129
 130     conn = get(conn, "/api/v1/instance")
 131
 132     assert Conn.get_resp_header(conn, "x-xss-protection") == []
 133     assert Conn.get_resp_header(conn, "x-permitted-cross-domain-policies") == []
 134     assert Conn.get_resp_header(conn, "x-frame-options") == []
 135     assert Conn.get_resp_header(conn, "x-content-type-options") == []
 136     assert Conn.get_resp_header(conn, "x-download-options") == []
 137     assert Conn.get_resp_header(conn, "referrer-policy") == []
 138     assert Conn.get_resp_header(conn, "content-security-policy") == []
 139   end
 140 end