Prevents leaking SQL errors in API response
This commit is contained in:
@ -1,4 +1,5 @@
|
|||||||
<?php
|
<?php
|
||||||
|
|
||||||
namespace MailPoet\API\JSON;
|
namespace MailPoet\API\JSON;
|
||||||
|
|
||||||
if(!defined('ABSPATH')) exit;
|
if(!defined('ABSPATH')) exit;
|
||||||
@ -12,23 +13,19 @@ class ErrorResponse extends Response {
|
|||||||
}
|
}
|
||||||
|
|
||||||
function getData() {
|
function getData() {
|
||||||
if(empty($this->errors)) {
|
return (empty($this->errors)) ? null : array('errors' => $this->errors);
|
||||||
return null;
|
|
||||||
} else {
|
|
||||||
return array(
|
|
||||||
'errors' => $this->errors
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
function formatErrors($errors = array()) {
|
function formatErrors($errors = array()) {
|
||||||
$formatted_errors = array();
|
return array_map(function($error, $message) {
|
||||||
foreach($errors as $error => $message) {
|
// sanitize SQL error
|
||||||
$formatted_errors[] = array(
|
if(preg_match('/^SQLSTATE/i', $message)) {
|
||||||
|
$message = __('An unknown error occurred.', 'mailpoet');
|
||||||
|
}
|
||||||
|
return array(
|
||||||
'error' => $error,
|
'error' => $error,
|
||||||
'message' => $message
|
'message' => $message
|
||||||
);
|
);
|
||||||
}
|
}, array_keys($errors), array_values($errors));
|
||||||
return $formatted_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