Classe DB

Introdução da classe DB (que utiliza mysqli e evita injeção de SQL), criação do arquivo correspondente, e adaptação dos arquivos inc/database.php para utilizar mysqli a partir dessa classe.

Classe DB
Introdução da classe DB (que utiliza mysqli e evita injeção de SQL), criação do arquivo correspondente, e adaptação dos arquivos inc/database.php para utilizar mysqli a partir dessa classe.
Leandro Arndt
1 parent a06dd354
Showing 3 changed files with 59 additions and 22 deletions Show diff stats
class/db.class.php
inc/database.php
restrito/inc/database.php
@@ -0,0 +1,43 @@
+<?php
+/**********************************************************************************
+ Sistema e-SIC Livre: sistema de acesso a informação baseado na lei de acesso.
+ 
+ Copyright (C) 2014 Prefeitura Municipal do Natal
+ 
+ Este programa é software livre; você pode redistribuí-lo e/ou
+ modificá-lo sob os termos da Licença GPL2.
+***********************************************************************************/
+
+class DB {
+	private static $mysqli;
+	private function __construct() {}
+	
+	static function conn() {
+		if (!self::$mysqli) {
+			self::$mysqli = new mysqli('p:'.DBHOST, DBUSER, DBPASS, DBNAME);
+		}
+		return self::$mysqli;
+	}
+	
+	public function execQuery($query) {
+		// Resolve o problema da injeção de SQL:
+		$rs = mysqli_query(DB::conn(), mysqli_real_escape_string(DB::CONN(), $query));
+		return $rs;
+	}
+	
+	public function rs_to_array($result, $numass=MYSQL_BOTH) {
+		$got = array();
+	
+		if(mysqli_num_rows($result) == 0)
+		return $got;
+	
+		mysqli_data_seek($result, 0);
+	
+		while ($row = mysqli_fetch_array($result, $numass)) {
+			array_push($got, $row);
+		}
+	
+		return $got;
+	}
+}
+?>
 \ No newline at end of file
@@ -9,6 +9,7 @@
 ***********************************************************************************/
 require_once("config.php");
+require_once("../class/db.class.php");
 function db_open() {
@@ -25,28 +26,24 @@ function db_close($conn) {
 //retorna objeto de conexao com o banco para transações
 function db_open_trans()
 {
-	//conecta ao mysqli
-	$mysqli = new mysqli(DBHOST, DBUSER, DBPASS, DBNAME);
-	
+	$mysqli = DB::conn();
 	/* check connection */
 	if (mysqli_connect_errno()) {
 		die("Falha na conexao: ". mysqli_connect_error());
 	}		
 	$mysqli->autocommit(false);
-	
-	return $mysqli;
-
-}
+	return $mysqli; // mantido para facilitar a transição do código para a global
+}
 function execQuery($query) {
-    $conn = db_open();
-
-    $rs = mysql_query($query, $conn); // or die (mysql_error());
+	$conn = db_open();
-    db_close($conn);
+	$rs = mysql_query($query, $conn); // or die (mysql_error());
+	
+	db_close($conn);
     return $rs;
 }
@@ -9,6 +9,7 @@
 ***********************************************************************************/
 require_once("config.php");
+include_once(DIR_CLASSES_LEIACESSO."/db.class.php");
 function db_open() {
@@ -25,28 +26,24 @@ function db_close($conn) {
 //retorna objeto de conexao com o banco para transações
 function db_open_trans()
 {
-	//conecta ao mysqli
-	$mysqli = new mysqli(DBHOST, DBUSER, DBPASS, DBNAME);
-	
+	$mysqli = DB::conn();
 	/* check connection */
 	if (mysqli_connect_errno()) {
 		die("Falha na conexao: ". mysqli_connect_error());
 	}		
 	$mysqli->autocommit(false);
-	
-	return $mysqli;
-
-}
+	return $mysqli; // mantido para facilitar a transição do código para a global
+}
 function execQuery($query) {
-    $conn = db_open();
-
-    $rs = mysql_query($query, $conn); // or die (mysql_error());
+	$conn = db_open();
-    db_close($conn);
+	$rs = mysql_query($query, $conn); // or die (mysql_error());
+	
+	db_close($conn);
     return $rs;
 }
	@@ -0,0 +1,43 @@		@@ -0,0 +1,43 @@
		1	+<?php
		2	+/**********************************************************************************
		3	+ Sistema e-SIC Livre: sistema de acesso a informação baseado na lei de acesso.
		4	+
		5	+ Copyright (C) 2014 Prefeitura Municipal do Natal
		6	+
		7	+ Este programa é software livre; você pode redistribuí-lo e/ou
		8	+ modificá-lo sob os termos da Licença GPL2.
		9	+***********************************************************************************/
		10	+
		11	+class DB {
		12	+ private static $mysqli;
		13	+ private function __construct() {}
		14	+
		15	+ static function conn() {
		16	+ if (!self::$mysqli) {
		17	+ self::$mysqli = new mysqli('p:'.DBHOST, DBUSER, DBPASS, DBNAME);
		18	+ }
		19	+ return self::$mysqli;
		20	+ }
		21	+
		22	+ public function execQuery($query) {
		23	+ // Resolve o problema da injeção de SQL:
		24	+ $rs = mysqli_query(DB::conn(), mysqli_real_escape_string(DB::CONN(), $query));
		25	+ return $rs;
		26	+ }
		27	+
		28	+ public function rs_to_array($result, $numass=MYSQL_BOTH) {
		29	+ $got = array();
		30	+
		31	+ if(mysqli_num_rows($result) == 0)
		32	+ return $got;
		33	+
		34	+ mysqli_data_seek($result, 0);
		35	+
		36	+ while ($row = mysqli_fetch_array($result, $numass)) {
		37	+ array_push($got, $row);
		38	+ }
		39	+
		40	+ return $got;
		41	+ }
		42	+}
		43	+?>
0	\ No newline at end of file	44	\ No newline at end of file