functions/db_prefs.php

   1 <?php
   2
   3 /**
   4  * db_prefs.php
   5  *
   6  * This contains functions for manipulating user preferences
   7  * stored in a database, accessed though the Pear DB layer.
   8  *
   9  * Database:
  10  *
  11  * The preferences table should have three columns:
  12  *    user       char  \  primary
  13  *    prefkey    char  /  key
  14  *    prefval    blob
  15  *
  16  *   CREATE TABLE userprefs (user CHAR(128) NOT NULL DEFAULT '',
  17  *                           prefkey CHAR(64) NOT NULL DEFAULT '',
  18  *                           prefval BLOB NOT NULL DEFAULT '',
  19  *                           primary key (user,prefkey));
  20  *
  21  * Configuration of databasename, username and password is done
  22  * by using conf.pl or the administrator plugin
  23  *
  24  * @copyright &copy; 1999-2007 The SquirrelMail Project Team
  25  * @license http://opensource.org/licenses/gpl-license.php GNU Public License
  26  * @version $Id$
  27  * @package squirrelmail
  28  * @subpackage prefs
  29  * @since 1.1.3
  30  */
  31
  32 /** @ignore */
  33 if (!defined('SM_PATH')) define('SM_PATH','../');
  34
  35 /** Unknown database */
  36 define('SMDB_UNKNOWN', 0);
  37 /** MySQL */
  38 define('SMDB_MYSQL', 1);
  39 /** PostgreSQL */
  40 define('SMDB_PGSQL', 2);
  41
  42 /**
  43  * don't display errors (no code execution in functions/*.php).
  44  * will handle error in dbPrefs class.
  45  */
  46 @include_once('DB.php');
  47
  48 global $prefs_are_cached, $prefs_cache;
  49
  50 /**
  51  * @ignore
  52  */
  53 function cachePrefValues($username) {
  54     global $prefs_are_cached, $prefs_cache;
  55
  56     sqgetGlobalVar('prefs_are_cached', $prefs_are_cached, SQ_SESSION );
  57     if ($prefs_are_cached) {
  58         sqgetGlobalVar('prefs_cache', $prefs_cache, SQ_SESSION );
  59         return;
  60     }
  61
  62     sqsession_unregister('prefs_cache');
  63     sqsession_unregister('prefs_are_cached');
  64
  65     $db = new dbPrefs;
  66     if(isset($db->error)) {
  67         printf( _("Preference database error (%s). Exiting abnormally"),
  68               $db->error);
  69         exit;
  70     }
  71
  72     $db->fillPrefsCache($username);
  73     if (isset($db->error)) {
  74         printf( _("Preference database error (%s). Exiting abnormally"),
  75               $db->error);
  76         exit;
  77     }
  78
  79     $prefs_are_cached = true;
  80
  81     sqsession_register($prefs_cache, 'prefs_cache');
  82     sqsession_register($prefs_are_cached, 'prefs_are_cached');
  83 }
  84
  85 /**
  86  * Class used to handle connections to prefs database and operations with preferences
  87  *
  88  * @package squirrelmail
  89  * @subpackage prefs
  90  * @since 1.1.3
  91  *
  92  */
  93 class dbPrefs {
  94     /**
  95      * Table used to store preferences
  96      * @var string
  97      */
  98     var $table = 'userprefs';
  99
 100     /**
 101      * Field used to store owner of preference
 102      * @var string
 103      */
 104     var $user_field = 'user';
 105
 106     /**
 107      * Field used to store preference name
 108      * @var string
 109      */
 110     var $key_field = 'prefkey';
 111
 112     /**
 113      * Field used to store preference value
 114      * @var string
 115      */
 116     var $val_field = 'prefval';
 117
 118     /**
 119      * Database connection object
 120      * @var object
 121      */
 122     var $dbh   = NULL;
 123
 124     /**
 125      * Error messages
 126      * @var string
 127      */
 128     var $error = NULL;
 129
 130     /**
 131      * Database type (SMDB_* constants)
 132      * Is used in setKey().
 133      * @var integer
 134      */
 135     var $db_type = SMDB_UNKNOWN;
 136
 137     /**
 138      * Default preferences
 139      * @var array
 140      */
 141     var $default = Array('theme_default' => 0,
 142                          'show_html_default' => '0');
 143
 144     /**
 145      * Preference owner field size
 146      * @var integer
 147      * @since 1.5.1
 148      */
 149     var $user_size = 128;
 150
 151     /**
 152      * Preference key field size
 153      * @var integer
 154      * @since 1.5.1
 155      */
 156     var $key_size = 64;
 157
 158     /**
 159      * Preference value field size
 160      * @var integer
 161      * @since 1.5.1
 162      */
 163     var $val_size = 65536;
 164
 165
 166
 167     /**
 168      * initialize the default preferences array.
 169      *
 170      */
 171     function dbPrefs() {
 172         // Try and read the default preferences file.
 173         $default_pref = SM_PATH . 'config/default_pref';
 174         if (@file_exists($default_pref)) {
 175             if ($file = @fopen($default_pref, 'r')) {
 176                 while (!feof($file)) {
 177                     $pref = fgets($file, 1024);
 178                     $i = strpos($pref, '=');
 179                     if ($i > 0) {
 180                         $this->default[trim(substr($pref, 0, $i))] = trim(substr($pref, $i + 1));
 181                     }
 182                 }
 183                 fclose($file);
 184             }
 185         }
 186     }
 187
 188     /**
 189      * initialize DB connection object
 190      *
 191      * @return boolean true, if object is initialized
 192      *
 193      */
 194     function open() {
 195         global $prefs_dsn, $prefs_table;
 196         global $prefs_user_field, $prefs_key_field, $prefs_val_field;
 197         global $prefs_user_size, $prefs_key_size, $prefs_val_size;
 198
 199         /* test if Pear DB class is available and freak out if it is not */
 200         if (! class_exists('DB')) {
 201             // same error also in abook_database.php
 202             $this->error  = _("Could not include PEAR database functions required for the database backend.") . "\n";
 203             $this->error .= sprintf(_("Is PEAR installed, and is the include path set correctly to find %s?"),
 204                               'DB.php') . "\n";
 205             $this->error .= _("Please contact your system administrator and report this error.");
 206             return false;
 207         }
 208
 209         if(isset($this->dbh)) {
 210             return true;
 211         }
 212
 213         if (preg_match('/^mysql/', $prefs_dsn)) {
 214             $this->db_type = SMDB_MYSQL;
 215         } elseif (preg_match('/^pgsql/', $prefs_dsn)) {
 216             $this->db_type = SMDB_PGSQL;
 217         }
 218
 219         if (!empty($prefs_table)) {
 220             $this->table = $prefs_table;
 221         }
 222         if (!empty($prefs_user_field)) {
 223             $this->user_field = $prefs_user_field;
 224         }
 225         if (!empty($prefs_key_field)) {
 226             $this->key_field = $prefs_key_field;
 227         }
 228         if (!empty($prefs_val_field)) {
 229             $this->val_field = $prefs_val_field;
 230         }
 231         if (!empty($prefs_user_size)) {
 232             $this->user_size = (int) $prefs_user_size;
 233         }
 234         if (!empty($prefs_key_size)) {
 235             $this->key_size = (int) $prefs_key_size;
 236         }
 237         if (!empty($prefs_val_size)) {
 238             $this->val_size = (int) $prefs_val_size;
 239         }
 240         $dbh = DB::connect($prefs_dsn, true);
 241
 242         if(DB::isError($dbh)) {
 243             $this->error = DB::errorMessage($dbh);
 244             return false;
 245         }
 246
 247         $this->dbh = $dbh;
 248         return true;
 249     }
 250
 251     /**
 252      * Function used to handle database connection errors
 253      *
 254      * @param object PEAR Error object
 255      *
 256      */
 257     function failQuery($res = NULL) {
 258         if($res == NULL) {
 259             printf(_("Preference database error (%s). Exiting abnormally"),
 260                   $this->error);
 261         } else {
 262             printf(_("Preference database error (%s). Exiting abnormally"),
 263                   DB::errorMessage($res));
 264         }
 265         exit;
 266     }
 267
 268     /**
 269      * Get user's prefs setting
 270      *
 271      * @param string $user user name
 272      * @param string $key preference name
 273      * @param mixed $default (since 1.2.5) default value
 274      *
 275      * @return mixed preference value
 276      *
 277      */
 278     function getKey($user, $key, $default = '') {
 279         global $prefs_cache;
 280
 281         cachePrefValues($user);
 282
 283         if (isset($prefs_cache[$key])) {
 284             return $prefs_cache[$key];
 285         } else {
 286             if (isset($this->default[$key])) {
 287                 return $this->default[$key];
 288             } else {
 289                 return $default;
 290             }
 291         }
 292     }
 293
 294     /**
 295      * Delete user's prefs setting
 296      *
 297      * @param string $user user name
 298      * @param string $key  preference name
 299      *
 300      * @return boolean
 301      *
 302      */
 303     function deleteKey($user, $key) {
 304         global $prefs_cache;
 305
 306         if (!$this->open()) {
 307             return false;
 308         }
 309         $query = sprintf("DELETE FROM %s WHERE %s='%s' AND %s='%s'",
 310                          $this->table,
 311                          $this->user_field,
 312                          $this->dbh->quoteString($user),
 313                          $this->key_field,
 314                          $this->dbh->quoteString($key));
 315
 316         $res = $this->dbh->simpleQuery($query);
 317         if(DB::isError($res)) {
 318             $this->failQuery($res);
 319         }
 320
 321         unset($prefs_cache[$key]);
 322
 323         return true;
 324     }
 325
 326     /**
 327      * Set user's preference
 328      *
 329      * @param string $user  user name
 330      * @param string $key   preference name
 331      * @param mixed  $value preference value
 332      *
 333      * @return boolean
 334      *
 335      */
 336     function setKey($user, $key, $value) {
 337         if (!$this->open()) {
 338             return false;
 339         }
 340
 341         /**
 342          * Check if username fits into db field
 343          */
 344         if (strlen($user) > $this->user_size) {
 345             $this->error = "Oversized username value."
 346                 ." Your preferences can't be saved."
 347                 ." See the administrator's manual or contact your system administrator.";
 348
 349             /**
 350              * Debugging function. Can be used to log all issues that trigger
 351              * oversized field errors. Function should be enabled in all three
 352              * strlen checks. See http://www.php.net/error-log
 353              */
 354             // error_log($user.'|'.$key.'|'.$value."\n",3,'/tmp/oversized_log');
 355
 356             // error is fatal
 357             $this->failQuery(null);
 358         }
 359         /**
 360          * Check if preference key fits into db field
 361          */
 362         if (strlen($key) > $this->key_size) {
 363             $err_msg = "Oversized user's preference key."
 364                 ." Some preferences were not saved."
 365                 ." See the administrator's manual or contact your system administrator.";
 366             // error is not fatal. Only some preference is not saved.
 367             trigger_error($err_msg,E_USER_WARNING);
 368             return false;
 369         }
 370         /**
 371          * Check if preference value fits into db field
 372          */
 373         if (strlen($value) > $this->val_size) {
 374             $err_msg = "Oversized user's preference value."
 375                 ." Some preferences were not saved."
 376                 ." See the administrator's manual or contact your system administrator.";
 377             // error is not fatal. Only some preference is not saved.
 378             trigger_error($err_msg,E_USER_WARNING);
 379             return false;
 380         }
 381
 382
 383         if ($this->db_type == SMDB_MYSQL) {
 384             $query = sprintf("REPLACE INTO %s (%s, %s, %s) ".
 385                              "VALUES('%s','%s','%s')",
 386                              $this->table,
 387                              $this->user_field,
 388                              $this->key_field,
 389                              $this->val_field,
 390                              $this->dbh->quoteString($user),
 391                              $this->dbh->quoteString($key),
 392                              $this->dbh->quoteString($value));
 393
 394             $res = $this->dbh->simpleQuery($query);
 395             if(DB::isError($res)) {
 396                 $this->failQuery($res);
 397             }
 398         } elseif ($this->db_type == SMDB_PGSQL) {
 399             $this->dbh->simpleQuery("BEGIN TRANSACTION");
 400             $query = sprintf("DELETE FROM %s WHERE %s='%s' AND %s='%s'",
 401                              $this->table,
 402                              $this->user_field,
 403                              $this->dbh->quoteString($user),
 404                              $this->key_field,
 405                              $this->dbh->quoteString($key));
 406             $res = $this->dbh->simpleQuery($query);
 407             if (DB::isError($res)) {
 408                 $this->dbh->simpleQuery("ROLLBACK TRANSACTION");
 409                 $this->failQuery($res);
 410             }
 411             $query = sprintf("INSERT INTO %s (%s, %s, %s) VALUES ('%s', '%s', '%s')",
 412                              $this->table,
 413                              $this->user_field,
 414                              $this->key_field,
 415                              $this->val_field,
 416                              $this->dbh->quoteString($user),
 417                              $this->dbh->quoteString($key),
 418                              $this->dbh->quoteString($value));
 419             $res = $this->dbh->simpleQuery($query);
 420             if (DB::isError($res)) {
 421                 $this->dbh->simpleQuery("ROLLBACK TRANSACTION");
 422                 $this->failQuery($res);
 423             }
 424             $this->dbh->simpleQuery("COMMIT TRANSACTION");
 425         } else {
 426             $query = sprintf("DELETE FROM %s WHERE %s='%s' AND %s='%s'",
 427                              $this->table,
 428                              $this->user_field,
 429                              $this->dbh->quoteString($user),
 430                              $this->key_field,
 431                              $this->dbh->quoteString($key));
 432             $res = $this->dbh->simpleQuery($query);
 433             if (DB::isError($res)) {
 434                 $this->failQuery($res);
 435             }
 436             $query = sprintf("INSERT INTO %s (%s, %s, %s) VALUES ('%s', '%s', '%s')",
 437                              $this->table,
 438                              $this->user_field,
 439                              $this->key_field,
 440                              $this->val_field,
 441                              $this->dbh->quoteString($user),
 442                              $this->dbh->quoteString($key),
 443                              $this->dbh->quoteString($value));
 444             $res = $this->dbh->simpleQuery($query);
 445             if (DB::isError($res)) {
 446                 $this->failQuery($res);
 447             }
 448         }
 449
 450         return true;
 451     }
 452
 453     /**
 454      * Fill preference cache array
 455      *
 456      * @param string $user user name
 457      *
 458      * @since 1.2.3
 459      *
 460      */
 461     function fillPrefsCache($user) {
 462         global $prefs_cache;
 463
 464         if (!$this->open()) {
 465             return;
 466         }
 467
 468         $prefs_cache = array();
 469         $query = sprintf("SELECT %s as prefkey, %s as prefval FROM %s ".
 470                          "WHERE %s = '%s'",
 471                          $this->key_field,
 472                          $this->val_field,
 473                          $this->table,
 474                          $this->user_field,
 475                          $this->dbh->quoteString($user));
 476         $res = $this->dbh->query($query);
 477         if (DB::isError($res)) {
 478             $this->failQuery($res);
 479         }
 480
 481         while ($row = $res->fetchRow(DB_FETCHMODE_ASSOC)) {
 482             $prefs_cache[$row['prefkey']] = $row['prefval'];
 483         }
 484     }
 485
 486 } /* end class dbPrefs */
 487
 488
 489 /**
 490  * Returns the value for the requested preference
 491  * @ignore
 492  */
 493 function getPref($data_dir, $username, $pref_name, $default = '') {
 494     $db = new dbPrefs;
 495     if(isset($db->error)) {
 496         printf( _("Preference database error (%s). Exiting abnormally"),
 497               $db->error);
 498         exit;
 499     }
 500
 501     return $db->getKey($username, $pref_name, $default);
 502 }
 503
 504 /**
 505  * Remove the desired preference setting ($pref_name)
 506  * @ignore
 507  */
 508 function removePref($data_dir, $username, $pref_name) {
 509     global $prefs_cache;
 510     $db = new dbPrefs;
 511     if(isset($db->error)) {
 512         $db->failQuery();
 513     }
 514
 515     $db->deleteKey($username, $pref_name);
 516
 517     if (isset($prefs_cache[$pref_name])) {
 518         unset($prefs_cache[$pref_name]);
 519     }
 520
 521     sqsession_register($prefs_cache , 'prefs_cache');
 522     return;
 523 }
 524
 525 /**
 526  * Sets the desired preference setting ($pref_name) to whatever is in $value
 527  * @ignore
 528  */
 529 function setPref($data_dir, $username, $pref_name, $value) {
 530     global $prefs_cache;
 531
 532     if (isset($prefs_cache[$pref_name]) && ($prefs_cache[$pref_name] == $value)) {
 533         return;
 534     }
 535
 536     if ($value === '') {
 537         removePref($data_dir, $username, $pref_name);
 538         return;
 539     }
 540
 541     $db = new dbPrefs;
 542     if(isset($db->error)) {
 543         $db->failQuery();
 544     }
 545
 546     $db->setKey($username, $pref_name, $value);
 547     $prefs_cache[$pref_name] = $value;
 548     assert_options(ASSERT_ACTIVE, 1);
 549     assert_options(ASSERT_BAIL, 1);
 550     assert ('$value == $prefs_cache[$pref_name]');
 551     sqsession_register($prefs_cache , 'prefs_cache');
 552     return;
 553 }
 554
 555 /**
 556  * This checks if the prefs are available
 557  * @ignore
 558  */
 559 function checkForPrefs($data_dir, $username) {
 560     $db = new dbPrefs;
 561     if(isset($db->error)) {
 562         $db->failQuery();
 563     }
 564 }
 565
 566 /**
 567  * Writes the Signature
 568  * @ignore
 569  */
 570 function setSig($data_dir, $username, $number, $value) {
 571     if ($number == "g") {
 572         $key = '___signature___';
 573     } else {
 574         $key = sprintf('___sig%s___', $number);
 575     }
 576     setPref($data_dir, $username, $key, $value);
 577     return;
 578 }
 579
 580 /**
 581  * Gets the signature
 582  * @ignore
 583  */
 584 function getSig($data_dir, $username, $number) {
 585     if ($number == "g") {
 586         $key = '___signature___';
 587     } else {
 588         $key = sprintf('___sig%d___', $number);
 589     }
 590     return getPref($data_dir, $username, $key);
 591 }