From e22915818cb7c6a756c4ac34124b14e33621c5aa Mon Sep 17 00:00:00 2001
From: Kyle Spearrin <kyle.spearrin@gmail.com>
Date: Mon, 2 Jul 2018 17:09:45 -0400
Subject: [PATCH] added public key and share key to crypto service

---
 src/abstractions/crypto.service.ts |  2 ++
 src/services/crypto.service.ts     | 42 ++++++++++++++++++++++++++++++
 2 files changed, 44 insertions(+)

diff --git a/src/abstractions/crypto.service.ts b/src/abstractions/crypto.service.ts
index fc55466ee7..c4807087c1 100644
--- a/src/abstractions/crypto.service.ts
+++ b/src/abstractions/crypto.service.ts
@@ -12,6 +12,7 @@ export abstract class CryptoService {
     getKey: () => Promise<SymmetricCryptoKey>;
     getKeyHash: () => Promise<string>;
     getEncKey: () => Promise<SymmetricCryptoKey>;
+    getPublicKey: () => Promise<ArrayBuffer>;
     getPrivateKey: () => Promise<ArrayBuffer>;
     getOrgKeys: () => Promise<Map<string, SymmetricCryptoKey>>;
     getOrgKey: (orgId: string) => Promise<SymmetricCryptoKey>;
@@ -24,6 +25,7 @@ export abstract class CryptoService {
     clearKeys: () => Promise<any>;
     toggleKey: () => Promise<any>;
     makeKey: (password: string, salt: string) => Promise<SymmetricCryptoKey>;
+    makeShareKey: () => Promise<[CipherString, SymmetricCryptoKey]>;
     hashPassword: (password: string, key: SymmetricCryptoKey) => Promise<string>;
     makeEncKey: (key: SymmetricCryptoKey) => Promise<CipherString>;
     encrypt: (plainValue: string | ArrayBuffer, key?: SymmetricCryptoKey) => Promise<CipherString>;
diff --git a/src/services/crypto.service.ts b/src/services/crypto.service.ts
index f01f05b5c0..8c913ad8cc 100644
--- a/src/services/crypto.service.ts
+++ b/src/services/crypto.service.ts
@@ -26,6 +26,7 @@ export class CryptoService implements CryptoServiceAbstraction {
     private encKey: SymmetricCryptoKey;
     private legacyEtmKey: SymmetricCryptoKey;
     private keyHash: string;
+    private publicKey: ArrayBuffer;
     private privateKey: ArrayBuffer;
     private orgKeys: Map<string, SymmetricCryptoKey>;
 
@@ -135,6 +136,20 @@ export class CryptoService implements CryptoServiceAbstraction {
         return this.encKey;
     }
 
+    async getPublicKey(): Promise<ArrayBuffer> {
+        if (this.publicKey != null) {
+            return this.publicKey;
+        }
+
+        const privateKey = await this.getPrivateKey();
+        if (privateKey == null) {
+            return null;
+        }
+
+        this.publicKey = null; // TODO:
+        return this.publicKey;
+    }
+
     async getPrivateKey(): Promise<ArrayBuffer> {
         if (this.privateKey != null) {
             return this.privateKey;
@@ -258,6 +273,14 @@ export class CryptoService implements CryptoServiceAbstraction {
         return new SymmetricCryptoKey(key);
     }
 
+    async makeShareKey(): Promise<[CipherString, SymmetricCryptoKey]> {
+        const shareKey = await this.cryptoFunctionService.randomBytes(64);
+        const publicKey = await this.getPublicKey();
+        const encKey = await this.getEncKey();
+        const encShareKey = await this.rsaEncrypt(shareKey, publicKey, encKey);
+        return [encShareKey, new SymmetricCryptoKey(shareKey)];
+    }
+
     async hashPassword(password: string, key: SymmetricCryptoKey): Promise<string> {
         if (key == null) {
             key = await this.getKey();
@@ -497,6 +520,25 @@ export class CryptoService implements CryptoServiceAbstraction {
         return await this.cryptoFunctionService.aesDecrypt(data, iv, theKey.encKey);
     }
 
+    private async rsaEncrypt(data: ArrayBuffer, publicKey?: ArrayBuffer, key?: SymmetricCryptoKey) {
+        if (publicKey == null) {
+            publicKey = await this.getPublicKey();
+        }
+        if (publicKey == null) {
+            throw new Error('Public key unavailable.');
+        }
+
+        let type = EncryptionType.Rsa2048_OaepSha1_B64;
+        const encBytes = await this.cryptoFunctionService.rsaEncrypt(data, publicKey, 'sha1');
+        let mac: string = null;
+        if (key != null && key.macKey != null) {
+            type = EncryptionType.Rsa2048_OaepSha1_HmacSha256_B64;
+            const macBytes = await this.cryptoFunctionService.hmac(encBytes, key.macKey, 'sha256');
+            mac = Utils.fromBufferToB64(macBytes);
+        }
+        return new CipherString(type, Utils.fromBufferToB64(encBytes), null, mac);
+    }
+
     private async rsaDecrypt(encValue: string): Promise<ArrayBuffer> {
         const headerPieces = encValue.split('.');
         let encType: EncryptionType = null;