mirror of
https://github.com/kidfromjupiter/nearby.git
synced 2026-09-15 07:06:11 -04:00
91 lines
2.4 KiB
C++
91 lines
2.4 KiB
C++
// Copyright 2020 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 "connections/implementation/mediums/utils.h"
|
|
|
|
#include <cstddef>
|
|
#include <cstdint>
|
|
#include <string>
|
|
|
|
#include "internal/platform/base64_utils.h"
|
|
#include "internal/platform/byte_array.h"
|
|
#include "internal/platform/crypto.h" //NOLINT
|
|
#include "internal/platform/prng.h"
|
|
|
|
namespace nearby {
|
|
namespace connections {
|
|
|
|
namespace {
|
|
constexpr int kDefaultSaltLength = 16;
|
|
} // namespace
|
|
|
|
using ::location::nearby::connections::LocationHint;
|
|
using ::location::nearby::connections::LocationStandard;
|
|
|
|
ByteArray Utils::GenerateRandomBytes(size_t length) {
|
|
Prng prng;
|
|
std::string data;
|
|
data.reserve(length);
|
|
|
|
// Adds 4 random bytes per iteration.
|
|
while (length > 0) {
|
|
std::uint32_t val = prng.NextUint32();
|
|
for (int i = 0; i < 4; i++) {
|
|
data += val & 0xFF;
|
|
val >>= 8;
|
|
length--;
|
|
|
|
if (!length) break;
|
|
}
|
|
}
|
|
|
|
return ByteArray(data);
|
|
}
|
|
|
|
ByteArray Utils::Sha256Hash(const ByteArray& source, size_t length) {
|
|
return Utils::Sha256Hash(std::string(source), length);
|
|
}
|
|
|
|
ByteArray Utils::Sha256Hash(const std::string& source, size_t length) {
|
|
ByteArray full_hash(length);
|
|
full_hash.CopyAt(0, Crypto::Sha256(source));
|
|
return full_hash;
|
|
}
|
|
|
|
LocationHint Utils::BuildLocationHint(const std::string& location) {
|
|
LocationHint location_hint;
|
|
location_hint.set_format(LocationStandard::UNKNOWN);
|
|
|
|
if (!location.empty()) {
|
|
location_hint.set_location(location);
|
|
if (location.at(0) == '+') {
|
|
location_hint.set_format(LocationStandard::E164_CALLING);
|
|
} else {
|
|
location_hint.set_format(LocationStandard::ISO_3166_1_ALPHA_2);
|
|
}
|
|
}
|
|
return location_hint;
|
|
}
|
|
|
|
// Generates salts.
|
|
std::string Utils::GenerateSalt() { return GenerateSalt(kDefaultSaltLength); }
|
|
|
|
std::string Utils::GenerateSalt(size_t length) {
|
|
ByteArray salt = GenerateRandomBytes(length);
|
|
return Base64Utils::Encode(salt);
|
|
}
|
|
|
|
} // namespace connections
|
|
} // namespace nearby
|