Db: clean up methods related to binary PG data
This commit is contained in:
parent
62a31c7858
commit
5de3a6a497
|
@ -3,6 +3,7 @@
|
|||
namespace Icinga\Module\Director\Data\Db;
|
||||
|
||||
use Icinga\Data\Db\DbConnection as IcingaDbConnection;
|
||||
use RuntimeException;
|
||||
use Zend_Db_Expr;
|
||||
|
||||
class DbConnection extends IcingaDbConnection
|
||||
|
@ -19,6 +20,10 @@ class DbConnection extends IcingaDbConnection
|
|||
|
||||
public function quoteBinary($binary)
|
||||
{
|
||||
if ($binary instanceof Zend_Db_Expr) {
|
||||
throw new RuntimeException('Trying to escape binary twice');
|
||||
}
|
||||
|
||||
if ($this->isPgsql()) {
|
||||
return new Zend_Db_Expr("'\\x" . bin2hex($binary) . "'");
|
||||
}
|
||||
|
@ -26,6 +31,15 @@ class DbConnection extends IcingaDbConnection
|
|||
return $binary;
|
||||
}
|
||||
|
||||
public function binaryDbResult($value)
|
||||
{
|
||||
if (is_resource($value)) {
|
||||
return stream_get_contents($value);
|
||||
}
|
||||
|
||||
return $value;
|
||||
}
|
||||
|
||||
public function hasPgExtension($name)
|
||||
{
|
||||
$db = $this->db();
|
||||
|
|
|
@ -760,12 +760,9 @@ abstract class DbObject
|
|||
unset($properties[$this->autoincKeyName]);
|
||||
}
|
||||
}
|
||||
// TODO: Remove this!
|
||||
if ($this->connection->isPgsql()) {
|
||||
foreach ($properties as $key => $value) {
|
||||
if ($this->isBinaryColumn($key)) {
|
||||
$properties[$key] = Util::pgBinEscape($value);
|
||||
}
|
||||
foreach ($properties as $key => $value) {
|
||||
if ($this->isBinaryColumn($key)) {
|
||||
$properties[$key] = $this->getConnection()->quoteBinary($value);
|
||||
}
|
||||
}
|
||||
|
||||
|
|
|
@ -11,7 +11,6 @@ use Icinga\Module\Director\Db\Cache\PrefetchCache;
|
|||
use Icinga\Module\Director\Db;
|
||||
use Icinga\Module\Director\Hook\ShipConfigFilesHook;
|
||||
use Icinga\Module\Director\Objects\IcingaObject;
|
||||
use Icinga\Module\Director\Util;
|
||||
use Icinga\Module\Director\Objects\IcingaHost;
|
||||
use Icinga\Module\Director\Objects\IcingaZone;
|
||||
use InvalidArgumentException;
|
||||
|
@ -270,7 +269,7 @@ class IcingaConfig
|
|||
$checksum = $this->calculateChecksum();
|
||||
$activity = $this->getLastActivityChecksum();
|
||||
|
||||
$lastActivity = $this->binFromDb(
|
||||
$lastActivity = $this->connection->binaryDbResult(
|
||||
$this->db->fetchOne(
|
||||
$this->db->select()->from(
|
||||
self::$table,
|
||||
|
@ -310,20 +309,7 @@ class IcingaConfig
|
|||
|
||||
protected function dbBin($binary)
|
||||
{
|
||||
if ($this->connection->isPgsql()) {
|
||||
return Util::pgBinEscape($binary);
|
||||
} else {
|
||||
return $binary;
|
||||
}
|
||||
}
|
||||
|
||||
protected function binFromDb($value)
|
||||
{
|
||||
if (is_resource($value)) {
|
||||
return stream_get_contents($value);
|
||||
}
|
||||
|
||||
return $value;
|
||||
return $this->connection->quoteBinary($binary);
|
||||
}
|
||||
|
||||
protected function calculateChecksum()
|
||||
|
@ -599,9 +585,9 @@ if (! globals.contains(DirectorOverrideTemplate)) {
|
|||
throw new NotFoundError('Got no config for %s', bin2hex($checksum));
|
||||
}
|
||||
|
||||
$this->checksum = $this->binFromDb($result->checksum);
|
||||
$this->checksum = $this->connection->binaryDbResult($result->checksum);
|
||||
$this->generationTime = $result->duration;
|
||||
$this->lastActivityChecksum = $this->binFromDb($result->last_activity_checksum);
|
||||
$this->lastActivityChecksum = $this->connection->binaryDbResult($result->last_activity_checksum);
|
||||
|
||||
$query = $this->db->select()->from(
|
||||
array('cf' => 'director_generated_config_file'),
|
||||
|
|
|
@ -9,8 +9,6 @@ use Icinga\Exception\NotImplementedError;
|
|||
use Icinga\Exception\ProgrammingError;
|
||||
use ipl\Html\Html;
|
||||
use gipfl\IcingaWeb2\Link;
|
||||
use RuntimeException;
|
||||
use Zend_Db_Expr;
|
||||
|
||||
class Util
|
||||
{
|
||||
|
@ -18,15 +16,6 @@ class Util
|
|||
|
||||
protected static $allowedResources;
|
||||
|
||||
public static function pgBinEscape($binary)
|
||||
{
|
||||
if ($binary instanceof Zend_Db_Expr) {
|
||||
throw new RuntimeException('Trying to escape binary twice');
|
||||
}
|
||||
|
||||
return new Zend_Db_Expr("'\\x" . bin2hex($binary) . "'");
|
||||
}
|
||||
|
||||
/**
|
||||
* PBKDF2 - Password-Based Cryptography Specification (RFC2898)
|
||||
*
|
||||
|
|
Loading…
Reference in New Issue