| // Copyright 2021 The Chromium OS Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "missive/encryption/encryption_module.h" |
| |
| #include <atomic> |
| #include <string> |
| #include <utility> |
| |
| #include <base/bind.h> |
| #include <base/callback.h> |
| #include <base/strings/string_piece.h> |
| #include <base/task/thread_pool.h> |
| #include <base/time/time.h> |
| |
| #include "missive/encryption/encryption_module_interface.h" |
| #include "missive/proto/record.pb.h" |
| #include "missive/util/status.h" |
| #include "missive/util/statusor.h" |
| |
| namespace reporting { |
| |
| namespace { |
| |
| // Helper function for asynchronous encryption. |
| void AddToRecord(base::StringPiece record, |
| Encryptor::Handle* handle, |
| base::OnceCallback<void(StatusOr<EncryptedRecord>)> cb) { |
| handle->AddToRecord( |
| record, |
| base::BindOnce( |
| [](Encryptor::Handle* handle, |
| base::OnceCallback<void(StatusOr<EncryptedRecord>)> cb, |
| Status status) { |
| if (!status.ok()) { |
| std::move(cb).Run(status); |
| return; |
| } |
| base::ThreadPool::PostTask( |
| FROM_HERE, |
| base::BindOnce(&Encryptor::Handle::CloseRecord, |
| base::Unretained(handle), std::move(cb))); |
| }, |
| base::Unretained(handle), std::move(cb))); |
| } |
| |
| } // namespace |
| |
| EncryptionModule::EncryptionModule(base::TimeDelta renew_encryption_key_period) |
| : EncryptionModuleInterface(renew_encryption_key_period) { |
| static_assert(std::is_same<PublicKeyId, Encryptor::PublicKeyId>::value, |
| "Public key id types must match"); |
| auto encryptor_result = Encryptor::Create(); |
| DCHECK(encryptor_result.ok()); |
| encryptor_ = std::move(encryptor_result.ValueOrDie()); |
| } |
| |
| EncryptionModule::~EncryptionModule() = default; |
| |
| void EncryptionModule::EncryptRecordImpl( |
| base::StringPiece record, |
| base::OnceCallback<void(StatusOr<EncryptedRecord>)> cb) const { |
| // Encryption key is available, encrypt. |
| encryptor_->OpenRecord(base::BindOnce( |
| [](base::StringPiece record, |
| base::OnceCallback<void(StatusOr<EncryptedRecord>)> cb, |
| StatusOr<Encryptor::Handle*> handle_result) { |
| if (!handle_result.ok()) { |
| std::move(cb).Run(handle_result.status()); |
| return; |
| } |
| base::ThreadPool::PostTask( |
| FROM_HERE, |
| base::BindOnce(&AddToRecord, std::string(record), |
| base::Unretained(handle_result.ValueOrDie()), |
| std::move(cb))); |
| }, |
| std::string(record), std::move(cb))); |
| } |
| |
| void EncryptionModule::UpdateAsymmetricKeyImpl( |
| base::StringPiece new_public_key, |
| PublicKeyId new_public_key_id, |
| base::OnceCallback<void(Status)> response_cb) { |
| encryptor_->UpdateAsymmetricKey(new_public_key, new_public_key_id, |
| std::move(response_cb)); |
| } |
| |
| // static |
| scoped_refptr<EncryptionModuleInterface> EncryptionModule::Create( |
| base::TimeDelta renew_encryption_key_period) { |
| return base::WrapRefCounted( |
| new EncryptionModule(renew_encryption_key_period)); |
| } |
| |
| } // namespace reporting |