include/db.php

   1 <?php
   2
   3 require_once "config.php";
   4
   5 function db_connect($host, $user, $pass, $db) {
   6         if (DB_TYPE == "pgsql") {
   7
   8                 $string = "dbname=$db user=$user";
   9
  10                 if ($pass) {
  11                         $string .= " password=$pass";
  12                 }
  13
  14                 if ($host) {
  15                         $string .= " host=$host";
  16                 }
  17
  18                 if (defined('DB_PORT')) {
  19                         $string = "$string port=" . DB_PORT;
  20                 }
  21
  22                 $link = pg_connect($string);
  23
  24                 if (!$link) {
  25                         die("Unable to connect to database (as $user to $host, database $db):" . pg_last_error());
  26                 }
  27
  28                 return $link;
  29
  30         } else if (DB_TYPE == "mysql") {
  31                 $link = mysql_connect($host, $user, $pass);
  32                 if ($link) {
  33                         $result = mysql_select_db($db, $link);
  34                         if (!$result) {
  35                                 die("Can't select DB: " . mysql_error($link));
  36                         }
  37                         return $link;
  38                 } else {
  39                         die("Unable to connect to database (as $user to $host, database $db): " . mysql_error());
  40                 }
  41         }
  42 }
  43
  44 function db_escape_string($s, $strip_tags = true) {
  45         if ($strip_tags) $s = strip_tags($s);
  46
  47         if (DB_TYPE == "pgsql") {
  48                 return pg_escape_string($s);
  49         } else {
  50                 return mysql_real_escape_string($s);
  51         }
  52 }
  53
  54 function db_query($link, $query, $die_on_error = true) {
  55         if ($_REQUEST["qlog"] || defined('QUERY_LOG')) {
  56                 $query = trim($query);
  57                 error_log($_SESSION["uid"] . ":" . $_REQUEST["op"] . "/" . $_REQUEST["method"] .
  58                                 " $query\n", 3, "/tmp/ttrss-query.log");
  59         }
  60
  61         if (DB_TYPE == "pgsql") {
  62                 $result = pg_query($link, $query);
  63                 if (!$result) {
  64                         $query = htmlspecialchars($query); // just in case
  65                         if ($die_on_error) {
  66                                 die("Query <i>$query</i> failed [$result]: " . ($link ? pg_last_error($link) : "No connection"));
  67                         }
  68                 }
  69                 return $result;
  70         } else if (DB_TYPE == "mysql") {
  71                 $result = mysql_query($query, $link);
  72                 if (!$result) {
  73                         $query = htmlspecialchars($query);
  74                         if ($die_on_error) {
  75                                 die("Query <i>$query</i> failed: " . ($link ? mysql_error($link) : "No connection"));
  76                         }
  77                 }
  78                 return $result;
  79         }
  80 }
  81
  82 function db_fetch_assoc($result) {
  83         if (DB_TYPE == "pgsql") {
  84                 return pg_fetch_assoc($result);
  85         } else if (DB_TYPE == "mysql") {
  86                 return mysql_fetch_assoc($result);
  87         }
  88 }
  89
  90
  91 function db_num_rows($result) {
  92         if (DB_TYPE == "pgsql") {
  93                 return pg_num_rows($result);
  94         } else if (DB_TYPE == "mysql") {
  95                 return mysql_num_rows($result);
  96         }
  97 }
  98
  99 function db_fetch_result($result, $row, $param) {
 100         if (DB_TYPE == "pgsql") {
 101                 return pg_fetch_result($result, $row, $param);
 102         } else if (DB_TYPE == "mysql") {
 103                 // I hate incoherent naming of PHP functions
 104                 return mysql_result($result, $row, $param);
 105         }
 106 }
 107
 108 function db_unescape_string($str) {
 109         $tmp = str_replace("\\\"", "\"", $str);
 110         $tmp = str_replace("\\'", "'", $tmp);
 111         return $tmp;
 112 }
 113
 114 function db_close($link) {
 115         if (DB_TYPE == "pgsql") {
 116
 117                 return pg_close($link);
 118
 119         } else if (DB_TYPE == "mysql") {
 120                 return mysql_close($link);
 121         }
 122 }
 123
 124 function db_affected_rows($link, $result) {
 125         if (DB_TYPE == "pgsql") {
 126                 return pg_affected_rows($result);
 127         } else if (DB_TYPE == "mysql") {
 128                 return mysql_affected_rows($link);
 129         }
 130 }
 131
 132 function db_last_error($link) {
 133         if (DB_TYPE == "pgsql") {
 134                 return pg_last_error($link);
 135         } else if (DB_TYPE == "mysql") {
 136                 return mysql_error($link);
 137         }
 138 }
 139
 140 function db_quote($str){
 141         return("'$str'");
 142 }
 143
 144 ?>