Ed Tanous | d4b6c66 | 2021-03-10 13:29:30 -0800 | [diff] [blame] | 1 | #pragma once |
Ed Tanous | 3ccb3ad | 2023-01-13 17:40:03 -0800 | [diff] [blame] | 2 | #include "http_request.hpp" |
| 3 | #include "http_response.hpp" |
| 4 | #include "http_utility.hpp" |
Ed Tanous | d4b6c66 | 2021-03-10 13:29:30 -0800 | [diff] [blame] | 5 | |
| 6 | namespace forward_unauthorized |
| 7 | { |
| 8 | |
Ed Tanous | cf9e417 | 2022-12-21 09:30:16 -0800 | [diff] [blame] | 9 | // NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables) |
Ed Tanous | 4f48d5f | 2021-06-21 08:27:45 -0700 | [diff] [blame] | 10 | static bool hasWebuiRoute = false; |
Ed Tanous | d4b6c66 | 2021-03-10 13:29:30 -0800 | [diff] [blame] | 11 | |
Ed Tanous | c127a0f | 2022-05-11 15:23:59 -0700 | [diff] [blame] | 12 | inline void sendUnauthorized(std::string_view url, |
| 13 | std::string_view xRequestedWith, |
John Edward Broadbent | 59b98b2 | 2021-07-13 15:36:32 -0700 | [diff] [blame] | 14 | std::string_view accept, crow::Response& res) |
Ed Tanous | d4b6c66 | 2021-03-10 13:29:30 -0800 | [diff] [blame] | 15 | { |
| 16 | // If it's a browser connecting, don't send the HTTP authenticate |
| 17 | // header, to avoid possible CSRF attacks with basic auth |
Ed Tanous | 4a0e1a0 | 2022-09-21 15:28:04 -0700 | [diff] [blame] | 18 | if (http_helpers::isContentTypeAllowed( |
| 19 | accept, http_helpers::ContentType::HTML, false /*allowWildcard*/)) |
Ed Tanous | d4b6c66 | 2021-03-10 13:29:30 -0800 | [diff] [blame] | 20 | { |
| 21 | // If we have a webui installed, redirect to that login page |
| 22 | if (hasWebuiRoute) |
| 23 | { |
| 24 | res.result(boost::beast::http::status::temporary_redirect); |
Ed Tanous | d9f6c62 | 2022-03-17 09:12:17 -0700 | [diff] [blame] | 25 | res.addHeader(boost::beast::http::field::location, |
John Edward Broadbent | 59b98b2 | 2021-07-13 15:36:32 -0700 | [diff] [blame] | 26 | "/#/login?next=" + http_helpers::urlEncode(url)); |
Ed Tanous | 347d1a1 | 2022-06-22 15:49:37 -0700 | [diff] [blame] | 27 | return; |
Ed Tanous | d4b6c66 | 2021-03-10 13:29:30 -0800 | [diff] [blame] | 28 | } |
Ed Tanous | 347d1a1 | 2022-06-22 15:49:37 -0700 | [diff] [blame] | 29 | // If we don't have a webui installed, just return an unauthorized |
| 30 | // body |
Ed Tanous | d4b6c66 | 2021-03-10 13:29:30 -0800 | [diff] [blame] | 31 | res.result(boost::beast::http::status::unauthorized); |
Ed Tanous | 347d1a1 | 2022-06-22 15:49:37 -0700 | [diff] [blame] | 32 | res.body() = "Unauthorized"; |
| 33 | return; |
Ed Tanous | d4b6c66 | 2021-03-10 13:29:30 -0800 | [diff] [blame] | 34 | } |
Ed Tanous | 347d1a1 | 2022-06-22 15:49:37 -0700 | [diff] [blame] | 35 | |
| 36 | res.result(boost::beast::http::status::unauthorized); |
| 37 | |
| 38 | // XHR requests from a browser will set the X-Requested-With header when |
| 39 | // doing their requests, even though they might not be requesting html. |
| 40 | if (!xRequestedWith.empty()) |
| 41 | { |
| 42 | return; |
| 43 | } |
| 44 | // if basic auth is disabled, don't propose it. |
| 45 | if (!persistent_data::SessionStore::getInstance() |
| 46 | .getAuthMethodsConfig() |
| 47 | .basic) |
| 48 | { |
| 49 | return; |
| 50 | } |
Ed Tanous | d9f6c62 | 2022-03-17 09:12:17 -0700 | [diff] [blame] | 51 | res.addHeader(boost::beast::http::field::www_authenticate, "Basic"); |
Ed Tanous | d4b6c66 | 2021-03-10 13:29:30 -0800 | [diff] [blame] | 52 | } |
| 53 | } // namespace forward_unauthorized |