Added exit() function if database connection fails.

opencart · Mar 21, 2015 · c1a6532 · c1a6532 · quincywu · Jul 8, 2015
1 parent 22d6bdc
commit c1a6532
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 0 deletions.
diff --git a/upload/system/library/db/mpdo.php b/upload/system/library/db/mpdo.php
@@ -9,6 +9,7 @@ public function __construct($hostname, $username, $password, $database, $port =
 			$this->pdo = new \PDO("mysql:host=" . $hostname . ";port=" . $port . ";dbname=" . $database, $username, $password, array(\PDO::ATTR_PERSISTENT => true));
 		} catch(\PDOException $e) {
 			trigger_error('Error: Could not make a database link ( ' . $e->getMessage() . '). Error Code : ' . $e->getCode() . ' <br />');
+			exit();
 		}
 
 		$this->pdo->exec("SET NAMES 'utf8'");

diff --git a/upload/system/library/db/mysql.php b/upload/system/library/db/mysql.php
@@ -6,10 +6,12 @@ final class MySQL {
 	public function __construct($hostname, $username, $password, $database) {
 		if (!$this->link = mysql_connect($hostname, $username, $password)) {
 			trigger_error('Error: Could not make a database link using ' . $username . '@' . $hostname);
+			exit();
 		}
 
 		if (!mysql_select_db($database, $this->link)) {
 			trigger_error('Error: Could not connect to database ' . $database);
+			exit();
 		}
 
 		mysql_query("SET NAMES 'utf8'", $this->link);

diff --git a/upload/system/library/db/mysqli.php b/upload/system/library/db/mysqli.php
@@ -8,6 +8,7 @@ public function __construct($hostname, $username, $password, $database) {
 
 		if ($this->link->connect_error) {
 			trigger_error('Error: Could not make a database link (' . $this->link->connect_errno . ') ' . $this->link->connect_error);
+			exit();
 		}
 
 		$this->link->set_charset("utf8");

diff --git a/upload/system/library/db/postgre.php b/upload/system/library/db/postgre.php
@@ -6,10 +6,12 @@ final class Postgre {
 	public function __construct($hostname, $username, $password, $database) {
 		if (!$this->link = pg_connect('hostname=' . $hostname . ' username=' . $username . ' password='	. $password . ' database=' . $database)) {
 			trigger_error('Error: Could not make a database link using ' . $username . '@' . $hostname);
+			exit();
 		}
 
 		if (!mysql_select_db($database, $this->link)) {
 			trigger_error('Error: Could not connect to database ' . $database);
+			exit();
 		}
 
 		pg_query($this->link, "SET CLIENT_ENCODING TO 'UTF8'");