Prevents leaking SQL errors in API response
This commit is contained in:
@ -1,4 +1,5 @@
|
||||
<?php
|
||||
|
||||
namespace MailPoet\API\JSON;
|
||||
|
||||
if(!defined('ABSPATH')) exit;
|
||||
@ -12,23 +13,19 @@ class ErrorResponse extends Response {
|
||||
}
|
||||
|
||||
function getData() {
|
||||
if(empty($this->errors)) {
|
||||
return null;
|
||||
} else {
|
||||
return array(
|
||||
'errors' => $this->errors
|
||||
);
|
||||
}
|
||||
return (empty($this->errors)) ? null : array('errors' => $this->errors);
|
||||
}
|
||||
|
||||
function formatErrors($errors = array()) {
|
||||
$formatted_errors = array();
|
||||
foreach($errors as $error => $message) {
|
||||
$formatted_errors[] = array(
|
||||
return array_map(function($error, $message) {
|
||||
// sanitize SQL error
|
||||
if(preg_match('/^SQLSTATE/i', $message)) {
|
||||
$message = __('An unknown error occurred.', 'mailpoet');
|
||||
}
|
||||
return array(
|
||||
'error' => $error,
|
||||
'message' => $message
|
||||
);
|
||||
}
|
||||
return $formatted_errors;
|
||||
}, array_keys($errors), array_values($errors));
|
||||
}
|
||||
}
|
34
tests/unit/API/JSON/ErrorResponseTest.php
Normal file
34
tests/unit/API/JSON/ErrorResponseTest.php
Normal file
@ -0,0 +1,34 @@
|
||||
<?php
|
||||
|
||||
namespace MailPoet\Test\API\JSON;
|
||||
|
||||
use MailPoet\API\JSON\ErrorResponse;
|
||||
|
||||
class ErrorResponseTest extends \MailPoetTest {
|
||||
function testItSanitizesSqlErrorsWhenReturningResponse() {
|
||||
$errors = array(
|
||||
'valid error',
|
||||
'SQLSTATE[22001]: Some SQL error',
|
||||
'another valid error'
|
||||
);
|
||||
$error_response = new ErrorResponse($errors);
|
||||
expect($error_response->getData())->equals(
|
||||
array(
|
||||
'errors' => array(
|
||||
array(
|
||||
'error' => 0,
|
||||
'message' => 'valid error'
|
||||
),
|
||||
array(
|
||||
'error' => 1,
|
||||
'message' => 'An unknown error occurred.'
|
||||
),
|
||||
array(
|
||||
'error' => 2,
|
||||
'message' => 'another valid error'
|
||||
)
|
||||
)
|
||||
)
|
||||
);
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user