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, $link = NULL) {
  45         if ($strip_tags) $s = strip_tags($s);
  46
  47         if (DB_TYPE == "pgsql") {
  48                 if ($link) {
  49                         return pg_escape_string($link, $s);
  50                 } else {
  51                         return pg_escape_string($s);
  52                 }
  53         } else {
  54                 if ($link) {
  55                         return mysql_real_escape_string($s, $link);
  56                 } else {
  57                         return mysql_real_escape_string($s);
  58                 }
  59         }
  60 }
  61
  62 function db_query($link, $query, $die_on_error = true) {
  63         if (DB_TYPE == "pgsql") {
  64                 $result = pg_query($link, $query);
  65                 if (!$result) {
  66                         $query = htmlspecialchars($query); // just in case
  67                         if ($die_on_error) {
  68                                 die("Query <i>$query</i> failed [$result]: " . ($link ? pg_last_error($link) : "No connection"));
  69                         }
  70                 }
  71                 return $result;
  72         } else if (DB_TYPE == "mysql") {
  73                 $result = mysql_query($query, $link);
  74                 if (!$result) {
  75                         $query = htmlspecialchars($query);
  76                         if ($die_on_error) {
  77                                 die("Query <i>$query</i> failed: " . ($link ? mysql_error($link) : "No connection"));
  78                         }
  79                 }
  80                 return $result;
  81         }
  82 }
  83
  84 function db_fetch_assoc($result) {
  85         if (DB_TYPE == "pgsql") {
  86                 return pg_fetch_assoc($result);
  87         } else if (DB_TYPE == "mysql") {
  88                 return mysql_fetch_assoc($result);
  89         }
  90 }
  91
  92
  93 function db_num_rows($result) {
  94         if (DB_TYPE == "pgsql") {
  95                 return pg_num_rows($result);
  96         } else if (DB_TYPE == "mysql") {
  97                 return mysql_num_rows($result);
  98         }
  99 }
 100
 101 function db_fetch_result($result, $row, $param) {
 102         if (DB_TYPE == "pgsql") {
 103                 return pg_fetch_result($result, $row, $param);
 104         } else if (DB_TYPE == "mysql") {
 105                 // I hate incoherent naming of PHP functions
 106                 return mysql_result($result, $row, $param);
 107         }
 108 }
 109
 110 function db_unescape_string($str) {
 111         $tmp = str_replace("\\\"", "\"", $str);
 112         $tmp = str_replace("\\'", "'", $tmp);
 113         return $tmp;
 114 }
 115
 116 function db_close($link) {
 117         if (DB_TYPE == "pgsql") {
 118
 119                 return pg_close($link);
 120
 121         } else if (DB_TYPE == "mysql") {
 122                 return mysql_close($link);
 123         }
 124 }
 125
 126 function db_affected_rows($link, $result) {
 127         if (DB_TYPE == "pgsql") {
 128                 return pg_affected_rows($result);
 129         } else if (DB_TYPE == "mysql") {
 130                 return mysql_affected_rows($link);
 131         }
 132 }
 133
 134 function db_last_error($link) {
 135         if (DB_TYPE == "pgsql") {
 136                 return pg_last_error($link);
 137         } else if (DB_TYPE == "mysql") {
 138                 return mysql_error($link);
 139         }
 140 }
 141
 142 function db_quote($str){
 143         return("'$str'");
 144 }
 145
 146 ?>