Added MS SQL Server support

1 files changed, 144 insertions, 4 deletions

diff --git a/lib/setup.php b/lib/setup.php
index 4dd190b99fb..d4ea26354ef 100644
--- a/lib/setup.php
+++ b/lib/setup.php
@@ -33,12 +33,14 @@ class OC_Setup {
 			$error[] = 'Specify a data folder.';
 		}
 
-		if($dbtype=='mysql' or $dbtype == 'pgsql' or $dbtype == 'oci') { //mysql and postgresql needs more config options
-			if($dbtype=='mysql')
+		if($dbtype == 'mysql' or $dbtype == 'pgsql' or $dbtype == 'oci' or $dbtype == 'mssql') { //mysql and postgresql needs more config options
+			if($dbtype == 'mysql')
 				$dbprettyname = 'MySQL';
-			else if($dbtype=='pgsql')
+			else if($dbtype == 'pgsql')
 				$dbprettyname = 'PostgreSQL';
-			else
+			else if ($dbtype == 'mssql')
+                $dbprettyname = 'MS SQL Server';
+            else
 				$dbprettyname = 'Oracle';
 
 
@@ -145,6 +147,29 @@ class OC_Setup {
 					return $error;
 				}
 			}
+            elseif ($dbtype == 'mssql') {
+				$dbuser = $options['dbuser'];
+				$dbpass = $options['dbpass'];
+				$dbname = $options['dbname'];
+				$dbhost = $options['dbhost'];
+				$dbtableprefix = isset($options['dbtableprefix']) ? $options['dbtableprefix'] : 'oc_';
+				
+				OC_Config::setValue('dbname', $dbname);
+				OC_Config::setValue('dbhost', $dbhost);
+				OC_Config::setValue('dbuser', $dbuser);
+				OC_Config::setValue('dbpassword', $dbpass);
+				OC_Config::setValue('dbtableprefix', $dbtableprefix);
+
+				try {
+					self::setupMSSQLDatabase($dbhost, $dbuser, $dbpass, $dbname, $dbtableprefix);
+				} catch (Exception $e) {
+					$error[] = array(
+						'error' => 'MS SQL username and/or password not valid',
+						'hint' => 'You need to enter either an existing account or the administrator.'
+					);
+					return $error;
+				}
+            }
 			else {
 				//delete the old sqlite database first, might cause infinte loops otherwise
 				if(file_exists("$datadir/owncloud.db")) {
@@ -563,6 +588,121 @@ class OC_Setup {
 		}
 	}
 
+	private static function setupMSSQLDatabase($dbhost, $dbuser, $dbpass, $dbname, $dbtableprefix) {
+		//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 Exception('MS SQL username and/or password not valid: '.$entry);
+		}
+
+		OC_Config::setValue('dbuser', $dbuser);
+		OC_Config::setValue('dbpassword', $dbpass);
+
+		self::mssql_createDBLogin($dbuser, $dbpass, $masterConnection);
+		
+		self::mssql_createDatabase($dbname, $masterConnection);
+		
+		self::mssql_createDBUser($dbuser, $dbpass, $masterConnection);
+
+		sqlsrv_close($masterConnection);
+
+		$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) {
+			$row=sqlsrv_fetch_array($result);
+		}
+		
+		if(!$result or $row[0] == 0) {
+			OC_DB::createDbFromStructure('db_structure.xml');
+		}
+		
+		sqlsrv_close($connection);
+	}    
+
+    private static function mssql_createDBLogin($name, $password, $connection) {
+		$query = "SELECT * FROM master.sys.server_principals WHERE name = '".$name."';";
+		$result = sqlsrv_query($connection, $query);
+		if ($result) {
+			$row = sqlsrv_fetch_array($result);
+		}
+		
+		if (!$result or $row[0] == 0) {
+			$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 />';
+			    echo($entry);
+			}			
+		}
+	}
+
+	private static function mssql_createDBUser($name, $dbname, $connection) {
+		$query = "SELECT * FROM [".$dbname."].sys.database_principals WHERE name = '".$name."';";
+		$result = sqlsrv_query($connection, $query);
+		if($result) {
+			$row=sqlsrv_fetch_array($result);
+		}		
+		
+		if (!$result or $row[0] == 0) {
+			$query = "USE [".$dbname."]; CREATE USER [".$name."] FOR LOGIN [".$name."];";
+			$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 />';
+			    echo($entry);
+			}			
+		}
+		
+		$query = "USE [".$dbname."]; EXEC sp_addrolemember 'db_owner', '".$name."';";
+		$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 />';
+		    echo($entry);
+		}
+	}
+	
+	private static function mssql_createDatabase($dbname, $connection) {
+		$query = "CREATE DATABASE [".$dbname."];";
+		$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 />';
+		    echo($entry);
+		}
+	}    
+    
+    
 	/**
 	 * create .htaccess files for apache hosts
 	 */