blob: 1baf00d1392dc1ae928d76d89bc833028e7d4a04 [file] [log] [blame]
// Copyright 2019 The Chromium 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 "remoting/test/ftl_signaling_playground.h"
#include <inttypes.h>
#include <string>
#include <utility>
#include <vector>
#include "base/base64.h"
#include "base/bind.h"
#include "base/bind_helpers.h"
#include "base/command_line.h"
#include "base/files/file_path.h"
#include "base/guid.h"
#include "base/logging.h"
#include "base/path_service.h"
#include "base/strings/string_number_conversions.h"
#include "base/strings/stringprintf.h"
#include "base/task/post_task.h"
#include "remoting/base/fake_oauth_token_getter.h"
#include "remoting/base/oauth_token_getter_impl.h"
#include "remoting/signaling/ftl_services.grpc.pb.h"
#include "remoting/test/test_oauth_token_factory.h"
#include "remoting/test/test_token_storage.h"
namespace {
constexpr char kSwitchNameHelp[] = "help";
constexpr char kSwitchNameAuthCode[] = "code";
constexpr char kSwitchNameUsername[] = "username";
constexpr char kSwitchNameStoragePath[] = "storage-path";
constexpr remoting::ftl::FtlCapability::Feature kFtlCapabilities[] = {
remoting::ftl::FtlCapability_Feature_RECEIVE_CALLS_FROM_GAIA,
remoting::ftl::FtlCapability_Feature_GAIA_REACHABLE};
// Reads a newline-terminated string from stdin.
std::string ReadString() {
const int kMaxLen = 1024;
std::string str(kMaxLen, 0);
char* result = fgets(&str[0], kMaxLen, stdin);
if (!result)
return std::string();
size_t newline_index = str.find('\n');
if (newline_index != std::string::npos)
str[newline_index] = '\0';
str.resize(strlen(&str[0]));
return str;
}
// Read the value of |switch_name| from command line if it exists, otherwise
// read from stdin.
std::string ReadStringFromCommandLineOrStdin(const std::string& switch_name,
const std::string& read_prompt) {
base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
if (command_line->HasSwitch(switch_name)) {
return command_line->GetSwitchValueASCII(switch_name);
}
printf("%s", read_prompt.c_str());
return ReadString();
}
// Wait for the user to press enter key on an anonymous thread and calls
// |on_done| once it is done.
void WaitForEnterKey(base::OnceClosure on_done) {
base::PostTaskWithTraitsAndReply(FROM_HERE, {base::MayBlock()},
base::BindOnce([]() { getchar(); }),
std::move(on_done));
}
void PrintGrpcStatusError(const grpc::Status& status) {
DCHECK(!status.ok());
fprintf(stderr, "RPC failed. Code=%d, Message=%s\n", status.error_code(),
status.error_message().c_str());
switch (status.error_code()) {
case grpc::StatusCode::UNAVAILABLE:
fprintf(stderr,
"Set the GRPC_DEFAULT_SSL_ROOTS_FILE_PATH environment variable "
"to third_party/grpc/src/etc/roots.pem if gRPC cannot locate the "
"root certificates.\n");
break;
case grpc::StatusCode::UNAUTHENTICATED:
fprintf(
stderr,
"Request is unauthenticated. Make sure you have run SignInGaia.\n");
break;
default:
break;
}
}
} // namespace
namespace remoting {
FtlSignalingPlayground::FtlSignalingPlayground() : weak_factory_(this) {}
FtlSignalingPlayground::~FtlSignalingPlayground() = default;
bool FtlSignalingPlayground::ShouldPrintHelp() {
return base::CommandLine::ForCurrentProcess()->HasSwitch(kSwitchNameHelp);
}
void FtlSignalingPlayground::PrintHelp() {
printf(
"Usage: %s [--code=<auth-code>] [--storage-path=<storage-path>] "
"[--username=<example@gmail.com>]\n",
base::CommandLine::ForCurrentProcess()
->GetProgram()
.MaybeAsASCII()
.c_str());
}
void FtlSignalingPlayground::StartAndAuthenticate() {
DCHECK(!storage_);
DCHECK(!token_getter_factory_);
DCHECK(!token_getter_);
DCHECK(!ftl_context_);
token_getter_factory_ = std::make_unique<TestOAuthTokenGetterFactory>();
base::CommandLine* cmd_line = base::CommandLine::ForCurrentProcess();
std::string username = cmd_line->GetSwitchValueASCII(kSwitchNameUsername);
base::FilePath storage_path =
cmd_line->GetSwitchValuePath(kSwitchNameStoragePath);
storage_ = test::TestTokenStorage::OnDisk(username, storage_path);
std::string access_token = storage_->FetchAccessToken();
if (access_token.empty()) {
AuthenticateAndResetServices();
} else {
VLOG(0) << "Reusing access token: " << access_token;
token_getter_ = std::make_unique<FakeOAuthTokenGetter>(
OAuthTokenGetter::Status::SUCCESS, "fake_email@gmail.com",
access_token);
ResetServices();
}
StartLoop();
}
void FtlSignalingPlayground::StartLoop() {
while (true) {
printf(
"\nOptions:\n"
" 1. SignInGaia\n"
" 2. GetIceServer\n"
" 3. PullMessages\n"
" 4. ReceiveMessages\n"
" 5. Quit\n\n"
"Your choice [number]: ");
int choice = 0;
base::StringToInt(ReadString(), &choice);
base::RunLoop run_loop;
switch (choice) {
case 1:
SignInGaia(run_loop.QuitClosure());
break;
case 2:
GetIceServer(run_loop.QuitClosure());
break;
case 3:
PullMessages(run_loop.QuitClosure());
break;
case 4:
StartReceivingMessages(run_loop.QuitWhenIdleClosure());
break;
case 5:
return;
default:
fprintf(stderr, "Unknown option\n");
continue;
}
run_loop.Run();
}
}
void FtlSignalingPlayground::ResetServices() {
ftl_context_ = std::make_unique<FtlGrpcContext>(token_getter_.get());
peer_to_peer_stub_ = PeerToPeer::NewStub(ftl_context_->channel());
registration_stub_ = Registration::NewStub(ftl_context_->channel());
message_subscription_.reset();
messaging_client_ = std::make_unique<FtlMessagingClient>(ftl_context_.get());
message_subscription_ = messaging_client_->RegisterMessageCallback(
base::BindRepeating(&FtlSignalingPlayground::OnMessageReceived,
weak_factory_.GetWeakPtr()));
}
void FtlSignalingPlayground::AuthenticateAndResetServices() {
static const std::string read_auth_code_prompt = base::StringPrintf(
"Please authenticate at:\n\n"
" %s\n\n"
"Enter the auth code: ",
TestOAuthTokenGetterFactory::GetAuthorizationCodeUri().c_str());
std::string auth_code = ReadStringFromCommandLineOrStdin(
kSwitchNameAuthCode, read_auth_code_prompt);
// Make sure we don't try to reuse an auth code.
base::CommandLine::ForCurrentProcess()->RemoveSwitch(kSwitchNameAuthCode);
// We can't get back the refresh token since we have first-party scope, so
// we are not trying to store it.
token_getter_ = token_getter_factory_->CreateFromIntermediateCredentials(
auth_code,
base::DoNothing::Repeatedly<const std::string&, const std::string&>());
// Get the access token so that we can reuse it for next time.
base::OnceClosure on_access_token_done = base::DoNothing::Once();
base::RunLoop run_loop;
if (!base::RunLoop::IsRunningOnCurrentThread()) {
on_access_token_done = run_loop.QuitClosure();
}
token_getter_->CallWithToken(base::BindOnce(
&FtlSignalingPlayground::OnAccessToken, weak_factory_.GetWeakPtr(),
std::move(on_access_token_done)));
if (!base::RunLoop::IsRunningOnCurrentThread()) {
run_loop.Run();
}
ResetServices();
}
void FtlSignalingPlayground::OnAccessToken(base::OnceClosure on_done,
OAuthTokenGetter::Status status,
const std::string& user_email,
const std::string& access_token) {
DCHECK(status == OAuthTokenGetter::Status::SUCCESS);
VLOG(0) << "Received access_token: " << access_token;
storage_->StoreAccessToken(access_token);
std::move(on_done).Run();
}
void FtlSignalingPlayground::GetIceServer(base::OnceClosure on_done) {
DCHECK(peer_to_peer_stub_);
VLOG(0) << "Running GetIceServer...";
ftl_context_->ExecuteRpc(
base::BindOnce(&PeerToPeer::Stub::AsyncGetICEServer,
base::Unretained(peer_to_peer_stub_.get())),
ftl::GetICEServerRequest(),
base::BindOnce(&FtlSignalingPlayground::OnGetIceServerResponse,
weak_factory_.GetWeakPtr(), std::move(on_done)));
}
void FtlSignalingPlayground::OnGetIceServerResponse(
base::OnceClosure on_done,
const grpc::Status& status,
const ftl::GetICEServerResponse& response) {
if (status.ok()) {
printf("Ice transport policy: %s\n",
response.ice_config().ice_transport_policy().c_str());
for (const ftl::ICEServerList& server :
response.ice_config().ice_servers()) {
printf(
"ICE server:\n"
" hostname=%s\n"
" username=%s\n"
" credential=%s\n"
" max_rate_kbps=%" PRId64 "\n",
server.hostname().c_str(), server.username().c_str(),
server.credential().c_str(), server.max_rate_kbps());
for (const std::string& url : server.urls()) {
printf(" url=%s\n", url.c_str());
}
}
} else {
PrintGrpcStatusError(status);
}
std::move(on_done).Run();
}
void FtlSignalingPlayground::SignInGaia(base::OnceClosure on_done) {
DCHECK(registration_stub_);
VLOG(0) << "Running SignInGaia...";
// TODO(yuweih): Logic should be cleaned up and moved out of the playground.
std::string device_id = storage_->FetchDeviceId();
if (device_id.empty()) {
device_id = "crd-web-" + base::GenerateGUID();
VLOG(0) << "Generated new device_id: " << device_id;
storage_->StoreDeviceId(device_id);
} else {
VLOG(0) << "Read device_id: " << device_id;
}
VLOG(0) << "Using sign_in_gaia_mode: DEFAULT_CREATE_ACCOUNT";
ftl::SignInGaiaRequest request;
request.set_app(FtlGrpcContext::GetChromotingAppIdentifier());
request.set_mode(ftl::SignInGaiaMode_Value_DEFAULT_CREATE_ACCOUNT);
request.mutable_register_data()->mutable_device_id()->set_id(device_id);
// TODO(yuweih): Consider using different device ID type.
request.mutable_register_data()->mutable_device_id()->set_type(
ftl::DeviceIdType_Type_WEB_UUID);
size_t ftl_capability_count =
sizeof(kFtlCapabilities) / sizeof(ftl::FtlCapability::Feature);
for (size_t i = 0; i < ftl_capability_count; i++) {
request.mutable_register_data()->add_caps(kFtlCapabilities[i]);
}
ftl_context_->ExecuteRpc(
base::BindOnce(&Registration::Stub::AsyncSignInGaia,
base::Unretained(registration_stub_.get())),
request,
base::BindOnce(&FtlSignalingPlayground::OnSignInGaiaResponse,
weak_factory_.GetWeakPtr(), std::move(on_done)));
}
void FtlSignalingPlayground::OnSignInGaiaResponse(
base::OnceClosure on_done,
const grpc::Status& status,
const ftl::SignInGaiaResponse& response) {
if (status.ok()) {
// TODO(yuweih): Allow loading auth token directly from command line.
std::string registration_id_base64;
std::string auth_token_base64;
base::Base64Encode(response.registration_id(), &registration_id_base64);
base::Base64Encode(response.auth_token().payload(), &auth_token_base64);
printf(
"registration_id(base64)=%s\n"
"auth_token.payload(base64)=%s\n"
"auth_token.expires_in=%" PRId64 "\n",
registration_id_base64.c_str(), auth_token_base64.c_str(),
response.auth_token().expires_in());
ftl_context_->SetAuthToken(response.auth_token().payload());
VLOG(0) << "Auth token set on FtlClient";
} else {
if (status.error_code() == grpc::StatusCode::UNAUTHENTICATED) {
VLOG(0) << "Grpc request failed to authenticate. "
<< "Trying to reauthenticate...";
AuthenticateAndResetServices();
} else {
PrintGrpcStatusError(status);
}
}
std::move(on_done).Run();
}
void FtlSignalingPlayground::PullMessages(base::OnceClosure on_done) {
DCHECK(messaging_client_);
VLOG(0) << "Running PullMessages...";
messaging_client_->PullMessages(
base::BindOnce(&FtlSignalingPlayground::OnPullMessagesResponse,
weak_factory_.GetWeakPtr(), std::move(on_done)));
}
void FtlSignalingPlayground::OnPullMessagesResponse(
base::OnceClosure on_done,
const grpc::Status& status) {
if (!status.ok()) {
PrintGrpcStatusError(status);
}
std::move(on_done).Run();
}
void FtlSignalingPlayground::StartReceivingMessages(base::OnceClosure on_done) {
VLOG(0) << "Running StartReceivingMessages...";
messaging_client_->StartReceivingMessages(
base::BindOnce(&FtlSignalingPlayground::OnStartReceivingMessagesDone,
weak_factory_.GetWeakPtr(), std::move(on_done)));
}
void FtlSignalingPlayground::StopReceivingMessages(base::OnceClosure on_done) {
messaging_client_->StopReceivingMessages();
std::move(on_done).Run();
}
void FtlSignalingPlayground::OnMessageReceived(const std::string& sender_id,
const std::string& message) {
printf(
"Received message:\n"
" Sender ID=%s\n"
" Message=%s\n",
sender_id.c_str(), message.c_str());
}
void FtlSignalingPlayground::OnStartReceivingMessagesDone(
base::OnceClosure on_done,
const grpc::Status& status) {
if (status.error_code() == grpc::StatusCode::CANCELLED) {
printf("ReceiveMessages stream canceled by client.\n");
std::move(on_done).Run();
return;
}
if (!status.ok()) {
PrintGrpcStatusError(status);
std::move(on_done).Run();
return;
}
printf("Started receiving messages. Press enter to stop streaming...\n");
WaitForEnterKey(base::BindOnce(&FtlSignalingPlayground::StopReceivingMessages,
weak_factory_.GetWeakPtr(),
std::move(on_done)));
}
} // namespace remoting