Passed
Pull Request — master (#6)
by
unknown
09:42
created

isCSPReportingAvailable()   B

Complexity

Conditions 7
Paths 7

Size

Total Lines 39
Code Lines 19

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
cc 7
eloc 19
c 0
b 0
f 0
nc 7
nop 0
dl 0
loc 39
rs 8.8333
1
<?php
2
3
namespace Signify\Middleware;
4
5
use Signify\Extensions\SecurityHeaderSiteconfigExtension;
6
use SilverStripe\Control\Director;
7
use SilverStripe\Control\HTTPRequest;
8
use SilverStripe\Control\Middleware\HTTPMiddleware;
9
use SilverStripe\Core\ClassInfo;
10
use SilverStripe\Core\Extensible;
11
use SilverStripe\Core\Config\Configurable;
12
use SilverStripe\Dev\TestOnly;
13
use SilverStripe\ORM\DB;
14
use SilverStripe\ORM\DataObject;
15
use SilverStripe\SiteConfig\SiteConfig;
16
17
class SecurityHeaderMiddleware implements HTTPMiddleware
18
{
19
    use Configurable;
20
    use Extensible;
21
22
    /**
23
     * An array of HTTP headers.
24
     * @config
25
     * @var array
26
     */
27
    private static $headers = [
0 ignored issues
show
introduced by
The private property $headers is not used, and could be removed.
Loading history...
28
        'global' => array(),
29
    ];
30
31
    /**
32
     * Whether to automatically add the CMS report endpoint to the CSP config.
33
     * @config
34
     * @var string
35
     */
36
    private static $enable_reporting = true;
0 ignored issues
show
introduced by
The private property $enable_reporting is not used, and could be removed.
Loading history...
37
38
    /**
39
     * The URI to report CSP violations to.
40
     * See routes.yml
41
     * @config
42
     * @var string
43
     */
44
    private static $report_uri = 'cspviolations/report';
0 ignored issues
show
introduced by
The private property $report_uri is not used, and could be removed.
Loading history...
45
46
    /**
47
     * Whether to use the report-to header and CSP directive.
48
     * @config
49
     * @var string
50
     */
51
    private static $use_report_to = false;
0 ignored issues
show
introduced by
The private property $use_report_to is not used, and could be removed.
Loading history...
52
53
    /**
54
     * Whether subdomains should report to the same endpoint.
55
     * @config
56
     * @var string
57
     */
58
    private static $report_to_subdomains = false;
0 ignored issues
show
introduced by
The private property $report_to_subdomains is not used, and could be removed.
Loading history...
59
60
    /**
61
     * The group name for the report-to CSP directive.
62
     * @config
63
     * @var string
64
     */
65
    private static $report_to_group = 'signify-csp-violation';
0 ignored issues
show
introduced by
The private property $report_to_group is not used, and could be removed.
Loading history...
66
67
    /**
68
     * Can isCSPReportingOnly be used safely.
69
     *
70
     * This is not a config option.
71
     *
72
     * @var boolean
73
     */
74
    private static $is_csp_reporting_safe = false;
75
76
77
    public function process(HTTPRequest $request, callable $delegate)
78
    {
79
        $response = $delegate($request);
80
81
        $headersConfig = (array) $this->config()->get('headers');
82
        if (empty($headersConfig['global'])) {
83
            return $response;
84
        }
85
86
        // Disable CSP
87
        if ($this->disableCSP()) {
88
            return $response;
89
        }
90
91
        $headersToSend = $headersConfig['global'];
92
93
        if (!$this->disableReporting() && $this->config()->get('use_report_to')) {
94
            $this->addReportToHeader($headersToSend);
95
        }
96
97
        // Update CSP header.
98
        if (array_key_exists('Content-Security-Policy', $headersToSend)) {
99
            $header = 'Content-Security-Policy';
100
            $headerValue = $headersToSend['Content-Security-Policy'];
101
            // Set report only mode if appropriate.
102
            if ($this->isCSPReportingOnly()) {
103
                unset($headersToSend['Content-Security-Policy']);
104
                $header = 'Content-Security-Policy-Report-Only';
105
            }
106
            // Update CSP header value.
107
            $headersToSend[$header] = $this->updateCspHeader($headerValue);
108
        }
109
        $this->extend('updateHeaders', $headersToSend, $request);
110
111
        // Add headers to response.
112
        foreach ($headersToSend as $header => $value) {
113
            if (empty($value)) {
114
                continue;
115
            }
116
            $value = preg_replace('/\v/', '', $value);
117
            $this->extend('updateHeader', $header, $value, $request);
118
            if ($value) {
119
                $response->addHeader($header, $value);
120
            }
121
        }
122
123
        return $response;
124
    }
125
126
    /**
127
     * Return true if the Disable CSP is checked
128
     *
129
     * @return boolean
130
     */
131
    public function disableCSP()
132
    {
133
        if (self::isCSPReportingAvailable() && SiteConfig::current_site_config()->CSPReportingOnly == SecurityHeaderSiteconfigExtension::CSP_DISABLE) {
134
            return true;
135
        }
136
137
        return false;
138
    }
139
140
    /**
141
     * Return true if the Disable reporting is checked
142
     *
143
     * The CMS setting can disable reporting even if the 'enable_reporting' is true
144
     *
145
     * @return boolean
146
     */
147
    public function disableReporting()
148
    {
149
        if (self::isCSPReportingAvailable()) {
150
            return SiteConfig::current_site_config()->CSPReportingOnly == SecurityHeaderSiteconfigExtension::CSP_WITHOUT_REPORTING || !$this->config()->get('enable_reporting');
151
        }
152
153
        return false;
154
    }
155
156
    /**
157
     * Returns true if the Content-Security-Policy-Report-Only header should be used.
158
     *
159
     * @return boolean
160
     */
161
    public function isCSPReportingOnly()
162
    {
163
        if (self::isCSPReportingAvailable() && SiteConfig::current_site_config()->CSPReportingOnly == SecurityHeaderSiteconfigExtension::CSP_REPORTING_ONLY) {
164
            return true;
165
        }
166
167
        return false;
168
    }
169
170
    protected function getReportURI()
171
    {
172
        return Director::absoluteURL($this->config()->get('report_uri'));
173
    }
174
175
    protected function getIncludeSubdomains()
176
    {
177
        return $this->config()->get('report_to_subdomains');
178
    }
179
180
    protected function getReportToGroup()
181
    {
182
        return $this->config()->get('report_to_group');
183
    }
184
185
    protected function getReportURIDirective()
186
    {
187
        return "report-uri {$this->getReportURI()}";
188
    }
189
190
    protected function getReportToDirective()
191
    {
192
        return "report-to {$this->getReportToGroup()}";
193
    }
194
195
    protected function addReportToHeader(&$headers)
196
    {
197
        if (array_key_exists('Report-To', $headers)) {
198
            $headers['Report-To'] = $headers['Report-To'] . ',' . $this->getReportToHeader();
199
        } else {
200
            $headers['Report-To'] = $this->getReportToHeader();
201
        }
202
    }
203
204
    protected function getReportToHeader()
205
    {
206
        $header = [
207
            'group' => $this->getReportToGroup(),
208
            'max_age' => 1800,
209
            'endpoints' => [[
210
                'url' => $this->getReportURI(),
211
            ],],
212
            'include_subdomains' => $this->getIncludeSubdomains(),
213
        ];
214
        return json_encode($header);
215
    }
216
217
    protected function updateCspHeader($cspHeader)
218
    {
219
        if (!$this->disableReporting()) {
220
            // Add or update report-uri directive.
221
            if (strpos($cspHeader, 'report-uri')) {
222
                $cspHeader = str_replace('report-uri', $this->getReportURIDirective(), $cspHeader);
223
            } else {
224
                $cspHeader = rtrim($cspHeader, ';') . "; {$this->getReportURIDirective()};";
225
            }
226
227
            // Add report-to directive.
228
            // Note that unlike report-uri, only the first endpoint is used if multiple are declared.
229
            if ($this->config()->get('use_report_to')) {
230
                if (strpos($cspHeader, 'report-to') === false) {
231
                    $cspHeader = rtrim($cspHeader, ';') . "; {$this->getReportToDirective()};";
232
                }
233
            }
234
        }
235
236
        return $cspHeader;
237
    }
238
239
    /**
240
     * Is the CSPReportingOnly field safe to read.
241
     *
242
     * If the module is installed and the codebase is flushed before the database has been built,
243
     * accessing SiteConfig causes an error.
244
     *
245
     * @return boolean
246
     */
247
    private static function isCSPReportingAvailable()
248
    {
249
        // Cached true value.
250
        if (self::$is_csp_reporting_safe) {
251
            return self::$is_csp_reporting_safe;
252
        }
253
254
        // Check if all tables and fields required for the class exist in the database.
255
        $requiredClasses = ClassInfo::dataClassesFor(SiteConfig::class);
256
        $schema = DataObject::getSchema();
257
        foreach (array_unique($requiredClasses) as $required) {
258
            // Skip test classes, as not all test classes are scaffolded at once
259
            if (is_a($required, TestOnly::class, true)) {
260
                continue;
261
            }
262
263
            // if any of the tables aren't created in the database
264
            $table = $schema->tableName($required);
265
            if (!ClassInfo::hasTable($table)) {
266
                return false;
267
            }
268
269
            // if any of the tables don't have any fields mapped as table columns
270
            $dbFields = DB::field_list($table);
271
            if (!$dbFields) {
0 ignored issues
show
Bug Best Practice introduced by
The expression $dbFields of type array is implicitly converted to a boolean; are you sure this is intended? If so, consider using empty($expr) instead to make it clear that you intend to check for an array without elements.

This check marks implicit conversions of arrays to boolean values in a comparison. While in PHP an empty array is considered to be equal (but not identical) to false, this is not always apparent.

Consider making the comparison explicit by using empty(..) or ! empty(...) instead.

Loading history...
272
                return false;
273
            }
274
275
            // if any of the tables are missing fields mapped as table columns
276
            $objFields = $schema->databaseFields($required, false);
277
            $missingFields = array_diff_key($objFields, $dbFields);
278
            if ($missingFields) {
0 ignored issues
show
Bug Best Practice introduced by
The expression $missingFields of type array is implicitly converted to a boolean; are you sure this is intended? If so, consider using ! empty($expr) instead to make it clear that you intend to check for an array without elements.

This check marks implicit conversions of arrays to boolean values in a comparison. While in PHP an empty array is considered to be equal (but not identical) to false, this is not always apparent.

Consider making the comparison explicit by using empty(..) or ! empty(...) instead.

Loading history...
279
                return false;
280
            }
281
        }
282
283
        self::$is_csp_reporting_safe = true;
284
285
        return true;
286
    }
287
}
288