mirror of
https://github.com/kidfromjupiter/nearby.git
synced 2026-09-14 22:56:12 -04:00
110 lines
3.6 KiB
C++
110 lines
3.6 KiB
C++
// Copyright 2022 Google LLC
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// https://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
#include "presence/implementation/ldt.h"
|
|
|
|
#include <algorithm>
|
|
#include <cstdint>
|
|
#include <string>
|
|
|
|
#include "absl/status/status.h"
|
|
#include "absl/status/statusor.h"
|
|
#include "absl/strings/str_format.h"
|
|
#include "absl/strings/string_view.h"
|
|
#ifdef NEARBY_CHROMIUM
|
|
#include "third_party/nearby/src/presence/implementation/np_ldt.h"
|
|
#else
|
|
#include "np_ldt.h"
|
|
#endif
|
|
|
|
namespace nearby {
|
|
namespace presence {
|
|
|
|
namespace {
|
|
// NP LDT library says that 0 is returned when `NpLdtCreate()` fails.
|
|
constexpr uint64_t kInvalidLdtHandle = 0;
|
|
|
|
template <class T>
|
|
T FromStringView(absl::string_view data) {
|
|
T result{
|
|
.bytes = {0},
|
|
};
|
|
memcpy(result.bytes, data.data(),
|
|
std::min(sizeof(result.bytes), data.size()));
|
|
return result;
|
|
}
|
|
} // namespace
|
|
|
|
LdtEncryptor::LdtEncryptor(LdtEncryptor&& other)
|
|
: ldt_encrypt_handle_(other.ldt_encrypt_handle_),
|
|
ldt_decrypt_handle_(other.ldt_decrypt_handle_) {
|
|
other.ldt_encrypt_handle_.handle = kInvalidLdtHandle;
|
|
other.ldt_decrypt_handle_.handle = kInvalidLdtHandle;
|
|
}
|
|
|
|
LdtEncryptor::~LdtEncryptor() {
|
|
if (ldt_encrypt_handle_.handle != kInvalidLdtHandle) {
|
|
NpLdtEncryptClose(ldt_encrypt_handle_);
|
|
}
|
|
if (ldt_decrypt_handle_.handle != kInvalidLdtHandle) {
|
|
NpLdtDecryptClose(ldt_decrypt_handle_);
|
|
}
|
|
}
|
|
|
|
absl::StatusOr<LdtEncryptor> LdtEncryptor::Create(
|
|
absl::string_view key_seed, absl::string_view known_hmac) {
|
|
NpLdtEncryptHandle encrypt_handle =
|
|
NpLdtEncryptCreate(FromStringView<NpLdtKeySeed>(key_seed));
|
|
NpLdtDecryptHandle decrypt_handle =
|
|
NpLdtDecryptCreate(FromStringView<NpLdtKeySeed>(key_seed),
|
|
FromStringView<NpMetadataKeyHmac>(known_hmac));
|
|
if (encrypt_handle.handle == kInvalidLdtHandle) {
|
|
return absl::UnavailableError("Failed to create LDT encryptor");
|
|
}
|
|
if (decrypt_handle.handle == kInvalidLdtHandle) {
|
|
return absl::UnavailableError("Failed to create LDT decrypter");
|
|
}
|
|
|
|
return LdtEncryptor(encrypt_handle, decrypt_handle);
|
|
}
|
|
|
|
absl::StatusOr<std::string> LdtEncryptor::Encrypt(absl::string_view data,
|
|
absl::string_view salt) {
|
|
std::string encrypted = std::string(data);
|
|
NP_LDT_RESULT result = NpLdtEncrypt(
|
|
ldt_encrypt_handle_, reinterpret_cast<uint8_t*>(encrypted.data()),
|
|
encrypted.size(), FromStringView<NpLdtSalt>(salt));
|
|
if (result == NP_LDT_SUCCESS) {
|
|
return encrypted;
|
|
}
|
|
return absl::InternalError(
|
|
absl::StrFormat("LDT encryption failed, errorcode %d", result));
|
|
}
|
|
|
|
absl::StatusOr<std::string> LdtEncryptor::DecryptAndVerify(
|
|
absl::string_view data, absl::string_view salt) {
|
|
std::string encrypted = std::string(data);
|
|
NP_LDT_RESULT result = NpLdtDecryptAndVerify(
|
|
ldt_decrypt_handle_, reinterpret_cast<uint8_t*>(encrypted.data()),
|
|
encrypted.size(), FromStringView<NpLdtSalt>(salt));
|
|
if (result == NP_LDT_SUCCESS) {
|
|
return encrypted;
|
|
}
|
|
return absl::InternalError(
|
|
absl::StrFormat("LDT encryption failed, errorcode %d", result));
|
|
}
|
|
|
|
} // namespace presence
|
|
} // namespace nearby
|