Bart Visscher
13 years ago
parent
ae48e5b496
commit
a6d07cd512
@ -0,0 +1,180 @@ |
||||
<?php |
||||
|
||||
namespace OC\Setup; |
||||
|
||||
class MSSQL { |
||||
public static function setupDatabase($dbhost, $dbuser, $dbpass, $dbname, $dbtableprefix) { |
||||
$l = \OC_Setup::getTrans(); |
||||
|
||||
//check if the database user has admin right |
||||
$masterConnectionInfo = array( "Database" => "master", "UID" => $dbuser, "PWD" => $dbpass); |
||||
|
||||
$masterConnection = @sqlsrv_connect($dbhost, $masterConnectionInfo); |
||||
if(!$masterConnection) { |
||||
$entry = null; |
||||
if( ($errors = sqlsrv_errors() ) != null) { |
||||
$entry='DB Error: "'.print_r(sqlsrv_errors()).'"<br />'; |
||||
} else { |
||||
$entry = ''; |
||||
} |
||||
throw new DatabaseSetupException($l->t('MS SQL username and/or password not valid: %s', array($entry)), |
||||
$l->t('You need to enter either an existing account or the administrator.')); |
||||
} |
||||
|
||||
\OC_Config::setValue('dbuser', $dbuser); |
||||
\OC_Config::setValue('dbpassword', $dbpass); |
||||
|
||||
self::createDBLogin($dbuser, $dbpass, $masterConnection); |
||||
|
||||
self::createDatabase($dbname, $masterConnection); |
||||
|
||||
self::createDBUser($dbuser, $dbname, $masterConnection); |
||||
|
||||
sqlsrv_close($masterConnection); |
||||
|
||||
self::createDatabaseStructure($dbhost, $dbname, $dbuser, $dbpass, $dbtableprefix); |
||||
} |
||||
|
||||
private static function createDBLogin($name, $password, $connection) { |
||||
$query = "SELECT * FROM master.sys.server_principals WHERE name = '".$name."';"; |
||||
$result = sqlsrv_query($connection, $query); |
||||
if ($result === false) { |
||||
if ( ($errors = sqlsrv_errors() ) != null) { |
||||
$entry='DB Error: "'.print_r(sqlsrv_errors()).'"<br />'; |
||||
} else { |
||||
$entry = ''; |
||||
} |
||||
$entry.='Offending command was: '.$query.'<br />'; |
||||
\OC_Log::write('setup.mssql', $entry, \OC_Log::WARN); |
||||
} else { |
||||
$row = sqlsrv_fetch_array($result); |
||||
|
||||
if ($row === false) { |
||||
if ( ($errors = sqlsrv_errors() ) != null) { |
||||
$entry='DB Error: "'.print_r(sqlsrv_errors()).'"<br />'; |
||||
} else { |
||||
$entry = ''; |
||||
} |
||||
$entry.='Offending command was: '.$query.'<br />'; |
||||
\OC_Log::write('setup.mssql', $entry, \OC_Log::WARN); |
||||
} else { |
||||
if ($row == null) { |
||||
$query = "CREATE LOGIN [".$name."] WITH PASSWORD = '".$password."';"; |
||||
$result = sqlsrv_query($connection, $query); |
||||
if (!$result or $result === false) { |
||||
if ( ($errors = sqlsrv_errors() ) != null) { |
||||
$entry='DB Error: "'.print_r(sqlsrv_errors()).'"<br />'; |
||||
} else { |
||||
$entry = ''; |
||||
} |
||||
$entry.='Offending command was: '.$query.'<br />'; |
||||
\OC_Log::write('setup.mssql', $entry, \OC_Log::WARN); |
||||
} |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
private static function createDBUser($name, $dbname, $connection) { |
||||
$query = "SELECT * FROM [".$dbname."].sys.database_principals WHERE name = '".$name."';"; |
||||
$result = sqlsrv_query($connection, $query); |
||||
if ($result === false) { |
||||
if ( ($errors = sqlsrv_errors() ) != null) { |
||||
$entry='DB Error: "'.print_r(sqlsrv_errors()).'"<br />'; |
||||
} else { |
||||
$entry = ''; |
||||
} |
||||
$entry.='Offending command was: '.$query.'<br />'; |
||||
\OC_Log::write('setup.mssql', $entry, \OC_Log::WARN); |
||||
} else { |
||||
$row = sqlsrv_fetch_array($result); |
||||
|
||||
if ($row === false) { |
||||
if ( ($errors = sqlsrv_errors() ) != null) { |
||||
$entry='DB Error: "'.print_r(sqlsrv_errors()).'"<br />'; |
||||
} else { |
||||
$entry = ''; |
||||
} |
||||
$entry.='Offending command was: '.$query.'<br />'; |
||||
\OC_Log::write('setup.mssql', $entry, \OC_Log::WARN); |
||||
} else { |
||||
if ($row == null) { |
||||
$query = "USE [".$dbname."]; CREATE USER [".$name."] FOR LOGIN [".$name."];"; |
||||
$result = sqlsrv_query($connection, $query); |
||||
if (!$result || $result === false) { |
||||
if ( ($errors = sqlsrv_errors() ) != null) { |
||||
$entry = 'DB Error: "'.print_r(sqlsrv_errors()).'"<br />'; |
||||
} else { |
||||
$entry = ''; |
||||
} |
||||
$entry.='Offending command was: '.$query.'<br />'; |
||||
\OC_Log::write('setup.mssql', $entry, \OC_Log::WARN); |
||||
} |
||||
} |
||||
|
||||
$query = "USE [".$dbname."]; EXEC sp_addrolemember 'db_owner', '".$name."';"; |
||||
$result = sqlsrv_query($connection, $query); |
||||
if (!$result || $result === false) { |
||||
if ( ($errors = sqlsrv_errors() ) != null) { |
||||
$entry='DB Error: "'.print_r(sqlsrv_errors()).'"<br />'; |
||||
} else { |
||||
$entry = ''; |
||||
} |
||||
$entry.='Offending command was: '.$query.'<br />'; |
||||
\OC_Log::write('setup.mssql', $entry, \OC_Log::WARN); |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
private static function createDatabase($dbname, $connection) { |
||||
$query = "CREATE DATABASE [".$dbname."];"; |
||||
$result = sqlsrv_query($connection, $query); |
||||
if (!$result || $result === false) { |
||||
if ( ($errors = sqlsrv_errors() ) != null) { |
||||
$entry='DB Error: "'.print_r(sqlsrv_errors()).'"<br />'; |
||||
} else { |
||||
$entry = ''; |
||||
} |
||||
$entry.='Offending command was: '.$query.'<br />'; |
||||
\OC_Log::write('setup.mssql', $entry, \OC_Log::WARN); |
||||
} |
||||
} |
||||
|
||||
private static function createDatabaseStructure($dbhost, $dbname, $dbuser, $dbpass, $dbtableprefix) { |
||||
$connectionInfo = array( "Database" => $dbname, "UID" => $dbuser, "PWD" => $dbpass); |
||||
|
||||
$connection = @sqlsrv_connect($dbhost, $connectionInfo); |
||||
|
||||
//fill the database if needed |
||||
$query = "SELECT * FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = '{$dbname}' AND TABLE_NAME = '{$dbtableprefix}users'"; |
||||
$result = sqlsrv_query($connection, $query); |
||||
if ($result === false) { |
||||
if ( ($errors = sqlsrv_errors() ) != null) { |
||||
$entry='DB Error: "'.print_r(sqlsrv_errors()).'"<br />'; |
||||
} else { |
||||
$entry = ''; |
||||
} |
||||
$entry.='Offending command was: '.$query.'<br />'; |
||||
\OC_Log::write('setup.mssql', $entry, \OC_Log::WARN); |
||||
} else { |
||||
$row = sqlsrv_fetch_array($result); |
||||
|
||||
if ($row === false) { |
||||
if ( ($errors = sqlsrv_errors() ) != null) { |
||||
$entry='DB Error: "'.print_r(sqlsrv_errors()).'"<br />'; |
||||
} else { |
||||
$entry = ''; |
||||
} |
||||
$entry.='Offending command was: '.$query.'<br />'; |
||||
\OC_Log::write('setup.mssql', $entry, \OC_Log::WARN); |
||||
} else { |
||||
if ($row == null) { |
||||
\OC_DB::createDbFromStructure('db_structure.xml'); |
||||
} |
||||
} |
||||
} |
||||
|
||||
sqlsrv_close($connection); |
||||
} |
||||
} |
||||
@ -0,0 +1,92 @@ |
||||
<?php |
||||
|
||||
namespace OC\Setup; |
||||
|
||||
class MySQL { |
||||
public static function setupDatabase($dbhost, $dbuser, $dbpass, $dbname, $dbtableprefix, $username) { |
||||
//check if the database user has admin right |
||||
$l = \OC_Setup::getTrans(); |
||||
$connection = @mysql_connect($dbhost, $dbuser, $dbpass); |
||||
if(!$connection) { |
||||
throw new DatabaseSetupException($l->t('MySQL username and/or password not valid'), |
||||
$l->t('You need to enter either an existing account or the administrator.')); |
||||
} |
||||
$oldUser=\OC_Config::getValue('dbuser', false); |
||||
|
||||
//this should be enough to check for admin rights in mysql |
||||
$query="SELECT user FROM mysql.user WHERE user='$dbuser'"; |
||||
if(mysql_query($query, $connection)) { |
||||
//use the admin login data for the new database user |
||||
|
||||
//add prefix to the mysql user name to prevent collisions |
||||
$dbusername=substr('oc_'.$username, 0, 16); |
||||
if($dbusername!=$oldUser) { |
||||
//hash the password so we don't need to store the admin config in the config file |
||||
$dbpassword=OC_Util::generate_random_bytes(30); |
||||
|
||||
self::createDBUser($dbusername, $dbpassword, $connection); |
||||
|
||||
\OC_Config::setValue('dbuser', $dbusername); |
||||
\OC_Config::setValue('dbpassword', $dbpassword); |
||||
} |
||||
|
||||
//create the database |
||||
self::createDatabase($dbname, $dbusername, $connection); |
||||
} |
||||
else { |
||||
if($dbuser!=$oldUser) { |
||||
\OC_Config::setValue('dbuser', $dbuser); |
||||
\OC_Config::setValue('dbpassword', $dbpass); |
||||
} |
||||
|
||||
//create the database |
||||
self::createDatabase($dbname, $dbuser, $connection); |
||||
} |
||||
|
||||
//fill the database if needed |
||||
$query='select count(*) from information_schema.tables' |
||||
." where table_schema='$dbname' AND table_name = '{$dbtableprefix}users';"; |
||||
$result = mysql_query($query, $connection); |
||||
if($result) { |
||||
$row=mysql_fetch_row($result); |
||||
} |
||||
if(!$result or $row[0]==0) { |
||||
\OC_DB::createDbFromStructure('db_structure.xml'); |
||||
} |
||||
mysql_close($connection); |
||||
} |
||||
|
||||
private static function createDatabase($name, $user, $connection) { |
||||
//we cant use OC_BD functions here because we need to connect as the administrative user. |
||||
$l = \OC_Setup::getTrans(); |
||||
$query = "CREATE DATABASE IF NOT EXISTS `$name`"; |
||||
$result = mysql_query($query, $connection); |
||||
if(!$result) { |
||||
$entry = $l->t('DB Error: "%s"', array(mysql_error($connection))) . '<br />'; |
||||
$entry .= $l->t('Offending command was: "%s"', array($query)) . '<br />'; |
||||
\OC_Log::write('setup.mssql', $entry, \OC_Log::WARN); |
||||
} |
||||
$query="GRANT ALL PRIVILEGES ON `$name` . * TO '$user'"; |
||||
|
||||
//this query will fail if there aren't the right permissions, ignore the error |
||||
mysql_query($query, $connection); |
||||
} |
||||
|
||||
private static function createDBUser($name, $password, $connection) { |
||||
// we need to create 2 accounts, one for global use and one for local user. if we don't specify the local one, |
||||
// the anonymous user would take precedence when there is one. |
||||
$l = \OC_Setup::getTrans(); |
||||
$query = "CREATE USER '$name'@'localhost' IDENTIFIED BY '$password'"; |
||||
$result = mysql_query($query, $connection); |
||||
if (!$result) { |
||||
throw new DatabaseSetupException($l->t("MySQL user '%s'@'localhost' exists already.", array($name)), |
||||
$l->t("Drop this user from MySQL", array($name))); |
||||
} |
||||
$query = "CREATE USER '$name'@'%' IDENTIFIED BY '$password'"; |
||||
$result = mysql_query($query, $connection); |
||||
if (!$result) { |
||||
throw new DatabaseSetupException($l->t("MySQL user '%s'@'%%' already exists", array($name)), |
||||
$l->t("Drop this user from MySQL.")); |
||||
} |
||||
} |
||||
} |
||||
@ -0,0 +1,198 @@ |
||||
<?php |
||||
|
||||
namespace OC\Setup; |
||||
|
||||
class OCI { |
||||
public static function setupDatabase($dbhost, $dbuser, $dbpass, $dbname, $dbtableprefix, $dbtablespace, |
||||
$username) { |
||||
$l = \OC_Setup::getTrans(); |
||||
$e_host = addslashes($dbhost); |
||||
$e_dbname = addslashes($dbname); |
||||
//check if the database user has admin right |
||||
if ($e_host == '') { |
||||
$easy_connect_string = $e_dbname; // use dbname as easy connect name |
||||
} else { |
||||
$easy_connect_string = '//'.$e_host.'/'.$e_dbname; |
||||
} |
||||
\OC_Log::write('setup oracle', 'connect string: ' . $easy_connect_string, \OC_Log::DEBUG); |
||||
$connection = @oci_connect($dbuser, $dbpass, $easy_connect_string); |
||||
if(!$connection) { |
||||
$e = oci_error(); |
||||
if (is_array ($e) && isset ($e['message'])) { |
||||
throw new DatabaseSetupException($l->t('Oracle connection could not be established'), |
||||
$e['message'].' Check environment: ORACLE_HOME='.getenv('ORACLE_HOME') |
||||
.' ORACLE_SID='.getenv('ORACLE_SID') |
||||
.' LD_LIBRARY_PATH='.getenv('LD_LIBRARY_PATH') |
||||
.' NLS_LANG='.getenv('NLS_LANG') |
||||
.' tnsnames.ora is '.(is_readable(getenv('ORACLE_HOME').'/network/admin/tnsnames.ora')?'':'not ').'readable'); |
||||
} |
||||
throw new DatabaseSetupException($l->t($l->t('Oracle username and/or password not valid'), |
||||
'Check environment: ORACLE_HOME='.getenv('ORACLE_HOME') |
||||
.' ORACLE_SID='.getenv('ORACLE_SID') |
||||
.' LD_LIBRARY_PATH='.getenv('LD_LIBRARY_PATH') |
||||
.' NLS_LANG='.getenv('NLS_LANG') |
||||
.' tnsnames.ora is '.(is_readable(getenv('ORACLE_HOME').'/network/admin/tnsnames.ora')?'':'not ').'readable'); |
||||
} |
||||
//check for roles creation rights in oracle |
||||
|
||||
$query='SELECT count(*) FROM user_role_privs, role_sys_privs' |
||||
." WHERE user_role_privs.granted_role = role_sys_privs.role AND privilege = 'CREATE ROLE'"; |
||||
$stmt = oci_parse($connection, $query); |
||||
if (!$stmt) { |
||||
$entry = $l->t('DB Error: "%s"', array(oci_last_error($connection))) . '<br />'; |
||||
$entry .= $l->t('Offending command was: "%s"', array($query)) . '<br />'; |
||||
\OC_Log::write('setup.oci', $entry, \OC_Log::WARN); |
||||
} |
||||
$result = oci_execute($stmt); |
||||
if($result) { |
||||
$row = oci_fetch_row($stmt); |
||||
} |
||||
if($result and $row[0] > 0) { |
||||
//use the admin login data for the new database user |
||||
|
||||
//add prefix to the oracle user name to prevent collisions |
||||
$dbusername='oc_'.$username; |
||||
//create a new password so we don't need to store the admin config in the config file |
||||
$dbpassword=OC_Util::generate_random_bytes(30); |
||||
|
||||
//oracle passwords are treated as identifiers: |
||||
// must start with aphanumeric char |
||||
// needs to be shortened to 30 bytes, as the two " needed to escape the identifier count towards the identifier length. |
||||
$dbpassword=substr($dbpassword, 0, 30); |
||||
|
||||
self::createDBUser($dbusername, $dbpassword, $dbtablespace, $connection); |
||||
|
||||
\OC_Config::setValue('dbuser', $dbusername); |
||||
\OC_Config::setValue('dbname', $dbusername); |
||||
\OC_Config::setValue('dbpassword', $dbpassword); |
||||
|
||||
//create the database not neccessary, oracle implies user = schema |
||||
//self::createDatabase($dbname, $dbusername, $connection); |
||||
} else { |
||||
|
||||
\OC_Config::setValue('dbuser', $dbuser); |
||||
\OC_Config::setValue('dbname', $dbname); |
||||
\OC_Config::setValue('dbpassword', $dbpass); |
||||
|
||||
//create the database not neccessary, oracle implies user = schema |
||||
//self::createDatabase($dbname, $dbuser, $connection); |
||||
} |
||||
|
||||
//FIXME check tablespace exists: select * from user_tablespaces |
||||
|
||||
// the connection to dbname=oracle is not needed anymore |
||||
oci_close($connection); |
||||
|
||||
// connect to the oracle database (schema=$dbuser) an check if the schema needs to be filled |
||||
$dbuser = \OC_Config::getValue('dbuser'); |
||||
//$dbname = \OC_Config::getValue('dbname'); |
||||
$dbpass = \OC_Config::getValue('dbpassword'); |
||||
|
||||
$e_host = addslashes($dbhost); |
||||
$e_dbname = addslashes($dbname); |
||||
|
||||
if ($e_host == '') { |
||||
$easy_connect_string = $e_dbname; // use dbname as easy connect name |
||||
} else { |
||||
$easy_connect_string = '//'.$e_host.'/'.$e_dbname; |
||||
} |
||||
$connection = @oci_connect($dbuser, $dbpass, $easy_connect_string); |
||||
if(!$connection) { |
||||
throw new DatabaseSetupException($l->t('Oracle username and/or password not valid'), |
||||
$l->t('You need to enter either an existing account or the administrator.')); |
||||
} |
||||
$query = "SELECT count(*) FROM user_tables WHERE table_name = :un"; |
||||
$stmt = oci_parse($connection, $query); |
||||
$un = $dbtableprefix.'users'; |
||||
oci_bind_by_name($stmt, ':un', $un); |
||||
if (!$stmt) { |
||||
$entry = $l->t('DB Error: "%s"', array(oci_error($connection))) . '<br />'; |
||||
$entry .= $l->t('Offending command was: "%s"', array($query)) . '<br />'; |
||||
\OC_Log::write('setup.oci', $entry, \OC_Log::WARN); |
||||
} |
||||
$result = oci_execute($stmt); |
||||
|
||||
if($result) { |
||||
$row = oci_fetch_row($stmt); |
||||
} |
||||
if(!$result or $row[0]==0) { |
||||
\OC_DB::createDbFromStructure('db_structure.xml'); |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* |
||||
* @param String $name |
||||
* @param String $password |
||||
* @param String $tablespace |
||||
* @param resource $connection |
||||
*/ |
||||
private static function createDBUser($name, $password, $tablespace, $connection) { |
||||
$l = \OC_Setup::getTrans(); |
||||
$query = "SELECT * FROM all_users WHERE USERNAME = :un"; |
||||
$stmt = oci_parse($connection, $query); |
||||
if (!$stmt) { |
||||
$entry = $l->t('DB Error: "%s"', array(oci_error($connection))) . '<br />'; |
||||
$entry .= $l->t('Offending command was: "%s"', array($query)) . '<br />'; |
||||
\OC_Log::write('setup.oci', $entry, \OC_Log::WARN); |
||||
} |
||||
oci_bind_by_name($stmt, ':un', $name); |
||||
$result = oci_execute($stmt); |
||||
if(!$result) { |
||||
$entry = $l->t('DB Error: "%s"', array(oci_error($connection))) . '<br />'; |
||||
$entry .= $l->t('Offending command was: "%s"', array($query)) . '<br />'; |
||||
\OC_Log::write('setup.oci', $entry, \OC_Log::WARN); |
||||
} |
||||
|
||||
if(! oci_fetch_row($stmt)) { |
||||
//user does not exists let's create it :) |
||||
//password must start with alphabetic character in oracle |
||||
$query = 'CREATE USER '.$name.' IDENTIFIED BY "'.$password.'" DEFAULT TABLESPACE '.$tablespace; //TODO set default tablespace |
||||
$stmt = oci_parse($connection, $query); |
||||
if (!$stmt) { |
||||
$entry = $l->t('DB Error: "%s"', array(oci_error($connection))) . '<br />'; |
||||
$entry .= $l->t('Offending command was: "%s"', array($query)) . '<br />'; |
||||
\OC_Log::write('setup.oci', $entry, \OC_Log::WARN); |
||||
} |
||||
//oci_bind_by_name($stmt, ':un', $name); |
||||
$result = oci_execute($stmt); |
||||
if(!$result) { |
||||
$entry = $l->t('DB Error: "%s"', array(oci_error($connection))) . '<br />'; |
||||
$entry .= $l->t('Offending command was: "%s", name: %s, password: %s', |
||||
array($query, $name, $password)) . '<br />'; |
||||
\OC_Log::write('setup.oci', $entry, \OC_Log::WARN); |
||||
} |
||||
} else { // change password of the existing role |
||||
$query = "ALTER USER :un IDENTIFIED BY :pw"; |
||||
$stmt = oci_parse($connection, $query); |
||||
if (!$stmt) { |
||||
$entry = $l->t('DB Error: "%s"', array(oci_error($connection))) . '<br />'; |
||||
$entry .= $l->t('Offending command was: "%s"', array($query)) . '<br />'; |
||||
\OC_Log::write('setup.oci', $entry, \OC_Log::WARN); |
||||
} |
||||
oci_bind_by_name($stmt, ':un', $name); |
||||
oci_bind_by_name($stmt, ':pw', $password); |
||||
$result = oci_execute($stmt); |
||||
if(!$result) { |
||||
$entry = $l->t('DB Error: "%s"', array(oci_error($connection))) . '<br />'; |
||||
$entry .= $l->t('Offending command was: "%s"', array($query)) . '<br />'; |
||||
\OC_Log::write('setup.oci', $entry, \OC_Log::WARN); |
||||
} |
||||
} |
||||
// grant necessary roles |
||||
$query = 'GRANT CREATE SESSION, CREATE TABLE, CREATE SEQUENCE, CREATE TRIGGER, UNLIMITED TABLESPACE TO '.$name; |
||||
$stmt = oci_parse($connection, $query); |
||||
if (!$stmt) { |
||||
$entry = $l->t('DB Error: "%s"', array(oci_error($connection))) . '<br />'; |
||||
$entry .= $l->t('Offending command was: "%s"', array($query)) . '<br />'; |
||||
\OC_Log::write('setup.oci', $entry, \OC_Log::WARN); |
||||
} |
||||
$result = oci_execute($stmt); |
||||
if(!$result) { |
||||
$entry = $l->t('DB Error: "%s"', array(oci_error($connection))) . '<br />'; |
||||
$entry .= $l->t('Offending command was: "%s", name: %s, password: %s', |
||||
array($query, $name, $password)) . '<br />'; |
||||
\OC_Log::write('setup.oci', $entry, \OC_Log::WARN); |
||||
} |
||||
} |
||||
} |
||||
@ -0,0 +1,142 @@ |
||||
<?php |
||||
|
||||
namespace OC\Setup; |
||||
|
||||
class PostgreSQL { |
||||
public static function setupDatabase($dbhost, $dbuser, $dbpass, $dbname, $dbtableprefix, $username) { |
||||
$e_host = addslashes($dbhost); |
||||
$e_user = addslashes($dbuser); |
||||
$e_password = addslashes($dbpass); |
||||
$l = \OC_Setup::getTrans(); |
||||
|
||||
//check if the database user has admin rights |
||||
$connection_string = "host='$e_host' dbname=postgres user='$e_user' password='$e_password'"; |
||||
$connection = @pg_connect($connection_string); |
||||
if(!$connection) { |
||||
// Try if we can connect to the DB with the specified name |
||||
$e_dbname = addslashes($dbname); |
||||
$connection_string = "host='$e_host' dbname='$e_dbname' user='$e_user' password='$e_password'"; |
||||
$connection = @pg_connect($connection_string); |
||||
|
||||
if(!$connection) |
||||
throw new DatabaseSetupException($l->t('PostgreSQL username and/or password not valid'), |
||||
$l->t('You need to enter either an existing account or the administrator.')); |
||||
} |
||||
$e_user = pg_escape_string($dbuser); |
||||
//check for roles creation rights in postgresql |
||||
$query="SELECT 1 FROM pg_roles WHERE rolcreaterole=TRUE AND rolname='$e_user'"; |
||||
$result = pg_query($connection, $query); |
||||
if($result and pg_num_rows($result) > 0) { |
||||
//use the admin login data for the new database user |
||||
|
||||
//add prefix to the postgresql user name to prevent collisions |
||||
$dbusername='oc_'.$username; |
||||
//create a new password so we don't need to store the admin config in the config file |
||||
$dbpassword=OC_Util::generate_random_bytes(30); |
||||
|
||||
self::createDBUser($dbusername, $dbpassword, $connection); |
||||
|
||||
\OC_Config::setValue('dbuser', $dbusername); |
||||
\OC_Config::setValue('dbpassword', $dbpassword); |
||||
|
||||
//create the database |
||||
self::createDatabase($dbname, $dbusername, $connection); |
||||
} |
||||
else { |
||||
\OC_Config::setValue('dbuser', $dbuser); |
||||
\OC_Config::setValue('dbpassword', $dbpass); |
||||
|
||||
//create the database |
||||
self::createDatabase($dbname, $dbuser, $connection); |
||||
} |
||||
|
||||
// the connection to dbname=postgres is not needed anymore |
||||
pg_close($connection); |
||||
|
||||
// connect to the ownCloud database (dbname=$dbname) and check if it needs to be filled |
||||
$dbuser = \OC_Config::getValue('dbuser'); |
||||
$dbpass = \OC_Config::getValue('dbpassword'); |
||||
|
||||
$e_host = addslashes($dbhost); |
||||
$e_dbname = addslashes($dbname); |
||||
$e_user = addslashes($dbuser); |
||||
$e_password = addslashes($dbpass); |
||||
|
||||
$connection_string = "host='$e_host' dbname='$e_dbname' user='$e_user' password='$e_password'"; |
||||
$connection = @pg_connect($connection_string); |
||||
if(!$connection) { |
||||
throw new DatabaseSetupException($l->t('PostgreSQL username and/or password not valid'), |
||||
$l->t('You need to enter either an existing account or the administrator.')); |
||||
} |
||||
$query = "select count(*) FROM pg_class WHERE relname='{$dbtableprefix}users' limit 1"; |
||||
$result = pg_query($connection, $query); |
||||
if($result) { |
||||
$row = pg_fetch_row($result); |
||||
} |
||||
if(!$result or $row[0]==0) { |
||||
\OC_DB::createDbFromStructure('db_structure.xml'); |
||||
} |
||||
} |
||||
|
||||
private static function createDatabase($name, $user, $connection) { |
||||
|
||||
//we cant use OC_BD functions here because we need to connect as the administrative user. |
||||
$l = \OC_Setup::getTrans(); |
||||
$e_name = pg_escape_string($name); |
||||
$e_user = pg_escape_string($user); |
||||
$query = "select datname from pg_database where datname = '$e_name'"; |
||||
$result = pg_query($connection, $query); |
||||
if(!$result) { |
||||
$entry = $l->t('DB Error: "%s"', array(pg_last_error($connection))) . '<br />'; |
||||
$entry .= $l->t('Offending command was: "%s"', array($query)) . '<br />'; |
||||
\OC_Log::write('setup.pg', $entry, \OC_Log::WARN); |
||||
} |
||||
if(! pg_fetch_row($result)) { |
||||
//The database does not exists... let's create it |
||||
$query = "CREATE DATABASE \"$e_name\" OWNER \"$e_user\""; |
||||
$result = pg_query($connection, $query); |
||||
if(!$result) { |
||||
$entry = $l->t('DB Error: "%s"', array(pg_last_error($connection))) . '<br />'; |
||||
$entry .= $l->t('Offending command was: "%s"', array($query)) . '<br />'; |
||||
\OC_Log::write('setup.pg', $entry, \OC_Log::WARN); |
||||
} |
||||
else { |
||||
$query = "REVOKE ALL PRIVILEGES ON DATABASE \"$e_name\" FROM PUBLIC"; |
||||
pg_query($connection, $query); |
||||
} |
||||
} |
||||
} |
||||
|
||||
private static function createDBUser($name, $password, $connection) { |
||||
$l = \OC_Setup::getTrans(); |
||||
$e_name = pg_escape_string($name); |
||||
$e_password = pg_escape_string($password); |
||||
$query = "select * from pg_roles where rolname='$e_name';"; |
||||
$result = pg_query($connection, $query); |
||||
if(!$result) { |
||||
$entry = $l->t('DB Error: "%s"', array(pg_last_error($connection))) . '<br />'; |
||||
$entry .= $l->t('Offending command was: "%s"', array($query)) . '<br />'; |
||||
\OC_Log::write('setup.pg', $entry, \OC_Log::WARN); |
||||
} |
||||
|
||||
if(! pg_fetch_row($result)) { |
||||
//user does not exists let's create it :) |
||||
$query = "CREATE USER \"$e_name\" CREATEDB PASSWORD '$e_password';"; |
||||
$result = pg_query($connection, $query); |
||||
if(!$result) { |
||||
$entry = $l->t('DB Error: "%s"', array(pg_last_error($connection))) . '<br />'; |
||||
$entry .= $l->t('Offending command was: "%s"', array($query)) . '<br />'; |
||||
\OC_Log::write('setup.pg', $entry, \OC_Log::WARN); |
||||
} |
||||
} |
||||
else { // change password of the existing role |
||||
$query = "ALTER ROLE \"$e_name\" WITH PASSWORD '$e_password';"; |
||||
$result = pg_query($connection, $query); |
||||
if(!$result) { |
||||
$entry = $l->t('DB Error: "%s"', array(pg_last_error($connection))) . '<br />'; |
||||
$entry .= $l->t('Offending command was: "%s"', array($query)) . '<br />'; |
||||
\OC_Log::write('setup.pg', $entry, \OC_Log::WARN); |
||||
} |
||||
} |
||||
} |
||||
} |
||||
Loading…
Reference in new issue