From 8a401ee15687059836c93dda62ee5ea3fbf800cf Mon Sep 17 00:00:00 2001
From: Bjoern Schiessle <bjoern@schiessle.org>
Date: Wed, 23 Nov 2016 12:07:04 +0100
Subject: [PATCH] check if session is initialized

Signed-off-by: Bjoern Schiessle <bjoern@schiessle.org>
---
 apps/encryption/lib/Crypto/Encryption.php     |  8 ++++++
 apps/encryption/lib/KeyManager.php            |  3 ---
 apps/encryption/lib/Session.php               | 10 +++++++
 .../tests/Crypto/EncryptionTest.php           | 15 +++++++++++
 apps/encryption/tests/SessionTest.php         | 26 +++++++++++++++++++
 5 files changed, 59 insertions(+), 3 deletions(-)

diff --git a/apps/encryption/lib/Crypto/Encryption.php b/apps/encryption/lib/Crypto/Encryption.php
index c71e25b6442..fdcbd41a09c 100644
--- a/apps/encryption/lib/Crypto/Encryption.php
+++ b/apps/encryption/lib/Crypto/Encryption.php
@@ -177,6 +177,14 @@ class Encryption implements IEncryptionModule {
 		$this->isWriteOperation = false;
 		$this->writeCache = '';
 
+		if($this->session->isReady() === false) {
+			// if the master key is enabled we can initialize encryption
+			// with a empty password and user name
+			if ($this->util->isMasterKeyEnabled()) {
+				$this->keyManager->init('', '');
+			}
+		}
+
 		if ($this->session->decryptAllModeActivated()) {
 			$encryptedFileKey = $this->keyManager->getEncryptedFileKey($this->path);
 			$shareKey = $this->keyManager->getShareKey($this->path, $this->session->getDecryptAllUid());
diff --git a/apps/encryption/lib/KeyManager.php b/apps/encryption/lib/KeyManager.php
index 96264d01363..26f023ed8f9 100644
--- a/apps/encryption/lib/KeyManager.php
+++ b/apps/encryption/lib/KeyManager.php
@@ -406,9 +406,6 @@ class KeyManager {
 		}
 
 		if ($this->util->isMasterKeyEnabled()) {
-			if ($this->session->getStatus() === Session::NOT_INITIALIZED)
-                        	$this->init('', '');
-			
 			$uid = $this->getMasterKeyId();
 		}
 
diff --git a/apps/encryption/lib/Session.php b/apps/encryption/lib/Session.php
index 92132d6080c..a61ee25fadb 100644
--- a/apps/encryption/lib/Session.php
+++ b/apps/encryption/lib/Session.php
@@ -67,6 +67,16 @@ class Session {
 		return $status;
 	}
 
+	/**
+	 * check if encryption was initialized successfully
+	 *
+	 * @return bool
+	 */
+	public function isReady() {
+		$status = $this->getStatus();
+		return $status === self::INIT_SUCCESSFUL;
+	}
+
 	/**
 	 * Gets user or public share private key from session
 	 *
diff --git a/apps/encryption/tests/Crypto/EncryptionTest.php b/apps/encryption/tests/Crypto/EncryptionTest.php
index 658f6275bb4..3525d2d4aec 100644
--- a/apps/encryption/tests/Crypto/EncryptionTest.php
+++ b/apps/encryption/tests/Crypto/EncryptionTest.php
@@ -279,6 +279,21 @@ class EncryptionTest extends TestCase {
 		);
 	}
 
+	/**
+	 * test begin() if encryption is not initialized but the master key is enabled
+	 * in this case we can initialize the encryption without a username/password
+	 * and continue
+	 */
+	public function testBeginInitMasterKey() {
+
+		$this->sessionMock->expects($this->once())->method('isReady')->willReturn(false);
+		$this->utilMock->expects($this->once())->method('isMasterKeyEnabled')
+			->willReturn(true);
+		$this->keyManagerMock->expects($this->once())->method('init')->with('', '');
+
+		$this->instance->begin('/user/files/welcome.txt', 'user', 'r', [], []);
+	}
+
 	/**
 	 * @dataProvider dataTestUpdate
 	 *
diff --git a/apps/encryption/tests/SessionTest.php b/apps/encryption/tests/SessionTest.php
index 099acddbca1..3000fedf2c3 100644
--- a/apps/encryption/tests/SessionTest.php
+++ b/apps/encryption/tests/SessionTest.php
@@ -133,6 +133,32 @@ class SessionTest extends TestCase {
 		$this->assertEquals(2, $this->instance->getStatus());
 	}
 
+	/**
+	 * @dataProvider dataTestIsReady
+	 *
+	 * @param int $status
+	 * @param bool $expected
+	 */
+	public function testIsReady($status, $expected) {
+		/** @var Session | \PHPUnit_Framework_MockObject_MockObject $instance */
+		$instance = $this->getMockBuilder(Session::class)
+			->setConstructorArgs([$this->sessionMock])
+			->setMethods(['getStatus'])->getMock();
+
+		$instance->expects($this->once())->method('getStatus')
+			->willReturn($status);
+
+		$this->assertSame($expected, $instance->isReady());
+	}
+
+	public function dataTestIsReady() {
+		return [
+			[Session::INIT_SUCCESSFUL, true],
+			[Session::INIT_EXECUTED, false],
+			[Session::NOT_INITIALIZED, false],
+		];
+	}
+
 	/**
 	 * @param $key
 	 * @param $value
-- 
2.39.5