blob: 5592d0cf9f4eddd55df148c4a189f09dca11e1be [file] [log] [blame] [edit]
// Copyright 2019 The Cobalt Authors. All Rights Reserved.
//
// 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
//
// http://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 "cobalt/browser/device_authentication.h"
#include "base/base64.h"
#include "base/base64url.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace cobalt {
namespace browser {
#if SB_API_VERSION >= 11
constexpr size_t kSHA256DigestSize = 32;
namespace {
std::string ToBase64Message(const std::string& cert_scope,
const std::string& start_time) {
std::string base64_message;
CHECK(base::Base64Encode(browser::ComputeMessage(cert_scope, start_time),
&base64_message));
return base64_message;
}
std::string ComputeBase64URLSignatureFromBase64Message(
const std::string& base64_message, const std::string& base64_secret_key) {
uint8_t signature[kSHA256DigestSize];
std::string message;
CHECK(base::Base64Decode(base64_message, &message));
ComputeHMACSHA256SignatureWithProvidedKey(message, base64_secret_key,
signature, kSHA256DigestSize);
std::string base_64_url_signature;
base::Base64UrlEncode(std::string(signature, signature + kSHA256DigestSize),
base::Base64UrlEncodePolicy::OMIT_PADDING,
&base_64_url_signature);
return base_64_url_signature;
}
std::string CreateSignedURLQueryString(const std::string& cert_scope,
const std::string& start_time,
const std::string& base64_secret_key) {
return GetDeviceAuthenticationSignedURLQueryStringFromComponents(
cert_scope, start_time,
ComputeBase64URLSignatureFromBase64Message(
ToBase64Message(cert_scope, start_time), base64_secret_key));
}
} // namespace
TEST(DeviceAuthenticationTest, ComputeMessageTest) {
EXPECT_EQ(
"AAAACmNlcnRfc2NvcGUAAAANbXktY2VydC1zY29wZQAAAApzdGFydF90aW1lAAAACjE1NjUx"
"NjE1NTY=",
ToBase64Message("my-cert-scope", "1565161556"));
EXPECT_EQ(
"AAAACmNlcnRfc2NvcGUAAAATbXktb3RoZXItY2VydC1zY29wZQAAAApzdGFydF90aW1lAAAA"
"CjE1NjUxNjMwNjU=",
ToBase64Message("my-other-cert-scope", "1565163065"));
EXPECT_EQ(
"AAAACmNlcnRfc2NvcGUAAAATbXktb3RoZXItY2VydC1zY29wZQAAAApzdGFydF90aW1lAAAA"
"CjE1NjUxNjMyNDU=",
ToBase64Message("my-other-cert-scope", "1565163245"));
EXPECT_EQ(
"AAAACmNlcnRfc2NvcGUAAAATbXktb3RoZXItY2VydC1zY29wZQAAAApzdGFydF90aW1lAAAA"
"CjI3OTAzMjY2Mzk=",
ToBase64Message("my-other-cert-scope", "2790326639"));
EXPECT_EQ(
"AAAACmNlcnRfc2NvcGUAAAAEeWFjcwAAAApzdGFydF90aW1lAAAACjEzNDAwMDAzMTI=",
ToBase64Message("yacs", "1340000312"));
}
TEST(DeviceAuthenticationTest, ComputeSignatureWithProvidedSecretTest) {
EXPECT_EQ("5duoLo5TELzyMQ6Fz-nmtLH3-nR4GrYfJ5RqTWU33LY",
ComputeBase64URLSignatureFromBase64Message(
"AAAACmNlcnRfc2NvcGUAAAANbXktY2VydC1zY29wZQAAAApzdGFydF90aW1lAA"
"AACjE1NjUxNjE1NTY=",
"abcdefghijklmnop1234567890abcdefghijklmnopqr"));
EXPECT_EQ("zrHJxDjsg60vF-fnGD9J7QaK2fw2QEZD7PIIfHMYmUg",
ComputeBase64URLSignatureFromBase64Message(
"AAAACmNlcnRfc2NvcGUAAAATbXktb3RoZXItY2VydC1zY29wZQAAAApzdGFydF"
"90aW1lAAAACjE1NjUxNjMwNjU=",
"abcdefghijklmnop1234567890abcdefghijklmnopqr"));
EXPECT_EQ("FnycnpFnmzgcNBIdJoymQvrS0_1uet_bmCtpuAmQR2s",
ComputeBase64URLSignatureFromBase64Message(
"AAAACmNlcnRfc2NvcGUAAAATbXktb3RoZXItY2VydC1zY29wZQAAAApzdGFydF"
"90aW1lAAAACjE1NjUxNjMyNDU=",
"abcdefghi5555nop1234567890abcdef5555klmn6666"));
EXPECT_EQ("N-nDY11_8HWFKEZPiQenQrC3SXuuPahs-er-n7Xh6ig",
ComputeBase64URLSignatureFromBase64Message(
"AAAACmNlcnRfc2NvcGUAAAATbXktb3RoZXItY2VydC1zY29wZQAAAApzdGFydF"
"90aW1lAAAACjI3OTAzMjY2Mzk=",
"abcdefghi5555nop1234567890abcdef5555klmn6666"));
EXPECT_EQ("dX0apAEsG2yWm9da6qTded4qd-mqgExeuJam99z-AHk",
ComputeBase64URLSignatureFromBase64Message(
"AAAACmNlcnRfc2NvcGUAAAAEeWFjcwAAAApzdGFydF90aW1lAAAACjEzNDAwMD"
"AzMTI=",
"11111111111111111111111111111111111111111111"));
}
// This is the main end-to-end test for device authentication.
TEST(DeviceAuthenticationTest,
GetDeviceAuthenticationSignedURLQueryStringFromComponentsTest) {
EXPECT_EQ(
"cert_scope=my-cert-scope&sig=5duoLo5TELzyMQ6Fz-nmtLH3-"
"nR4GrYfJ5RqTWU33LY&start_time=1565161556",
CreateSignedURLQueryString(
"my-cert-scope", "1565161556",
"abcdefghijklmnop1234567890abcdefghijklmnopqr"));
EXPECT_EQ(
"cert_scope=my-other-cert-scope&sig="
"Lf2zunrdhjH8q3ehdUy0tneTtamWigcyTgQl7zxWgnQ&start_time=123456789",
CreateSignedURLQueryString(
"my-other-cert-scope", "123456789",
"abcdefghijklmnop1234567890abcdefghijklmnopqr"));
EXPECT_EQ(
"cert_scope=my-other-cert-scope&sig="
"c5YZB0Rtv8Nf8gLSbE052ZvCUEpouP28nUq77FEgg88&start_time=11111111",
CreateSignedURLQueryString(
"my-other-cert-scope", "11111111",
"11111111111111111111111111111111111111111111"));
EXPECT_EQ(
"cert_scope=yacs&sig=YKjLEzSl2_ub-05Ajaww0HOOPElxEPUc4SEiQnGAfaE&start_"
"time=11111111",
CreateSignedURLQueryString(
"yacs", "11111111", "11111111111111111111111111111111111111111111"));
}
TEST(DeviceAuthenticationTest, NoCertSignatureImpliesNoQueryParameters) {
EXPECT_EQ("", GetDeviceAuthenticationSignedURLQueryStringFromComponents(
"my_cert_scope", "1234", ""));
}
#endif // SB_API_VERSION >= 11
} // namespace browser
} // namespace cobalt