95 lines
2.9 KiB
C++
95 lines
2.9 KiB
C++
|
//
|
||
|
|
// Copyright (C) 2022 The Android Open Source Project
|
||
|
|
//
|
||
|
|
// 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 <android-base/logging.h>
|
||
|
|
#include <android-base/strings.h>
|
||
|
|
#include <curl/curl.h>
|
||
|
|
#include <gflags/gflags.h>
|
||
|
|
#include <json/json.h>
|
||
|
|
#include <net/if.h>
|
||
|
|
#include <netinet/in.h>
|
||
|
|
#include <string.h>
|
||
|
|
#include <sys/ioctl.h>
|
||
|
|
#include <sys/time.h>
|
||
|
|
#include <sys/utsname.h>
|
||
|
|
#include <chrono>
|
||
|
|
#include <ctime>
|
||
|
|
#include <iostream>
|
||
|
|
|
||
|
|
#include "common/libs/utils/tee_logging.h"
|
||
|
|
#include "host/commands/metrics/events.h"
|
||
|
|
#include "host/commands/metrics/host_receiver.h"
|
||
|
|
#include "host/commands/metrics/metrics_defs.h"
|
||
|
|
#include "host/commands/metrics/proto/cf_metrics_proto.h"
|
||
|
|
#include "host/libs/config/cuttlefish_config.h"
|
||
|
|
#include "host/libs/metrics/metrics_receiver.h"
|
||
|
|
#include "host/libs/msg_queue/msg_queue.h"
|
||
|
|
#include "host/libs/vm_manager/crosvm_manager.h"
|
||
|
|
#include "host/libs/vm_manager/qemu_manager.h"
|
||
|
|
|
||
|
|
using cuttlefish::MetricsExitCodes;
|
||
|
|
|
||
|
|
namespace cuttlefish {
|
||
|
|
|
||
|
|
MetricsHostReceiver::MetricsHostReceiver(
|
||
|
|
const cuttlefish::CuttlefishConfig& config)
|
||
|
|
: config_(config) {}
|
||
|
|
|
||
|
|
MetricsHostReceiver::~MetricsHostReceiver() {}
|
||
|
|
|
||
|
|
void MetricsHostReceiver::ServerLoop() {
|
||
|
|
auto msg_queue = cuttlefish::SysVMessageQueue::Create("cuttlefish_ipc", 'a');
|
||
|
|
if (msg_queue == NULL) {
|
||
|
|
LOG(FATAL) << "create: failed to create cuttlefish_ipc";
|
||
|
|
}
|
||
|
|
|
||
|
|
struct msg_buffer msg = {0, {0}};
|
||
|
|
while (1) {
|
||
|
|
int rc = msg_queue->Receive(&msg, MAX_MSG_SIZE, 1, true);
|
||
|
|
if (rc == -1) {
|
||
|
|
LOG(FATAL) << "receive: failed to receive any messages";
|
||
|
|
}
|
||
|
|
std::string text(msg.mesg_text);
|
||
|
|
LOG(INFO) << "Metrics host received: " << text;
|
||
|
|
auto hostDev = cuttlefish::CuttlefishLogEvent::CUTTLEFISH_DEVICE_TYPE_HOST;
|
||
|
|
if (text == "VMStart") {
|
||
|
|
rc = Clearcut::SendVMStart(hostDev);
|
||
|
|
} else if (text == "VMStop") {
|
||
|
|
rc = Clearcut::SendVMStop(hostDev);
|
||
|
|
} else if (text == "DeviceBoot") {
|
||
|
|
rc = Clearcut::SendDeviceBoot(hostDev);
|
||
|
|
} else if (text == "LockScreen") {
|
||
|
|
rc = Clearcut::SendLockScreen(hostDev);
|
||
|
|
}
|
||
|
|
if (rc != MetricsExitCodes::kSuccess) {
|
||
|
|
LOG(ERROR) << "Message failed to send to ClearCut: " << text;
|
||
|
|
}
|
||
|
|
sleep(1);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void MetricsHostReceiver::Join() { thread_.join(); }
|
||
|
|
|
||
|
|
bool MetricsHostReceiver::Initialize() {
|
||
|
|
if (!config_.enable_metrics()) {
|
||
|
|
LOG(ERROR) << "init: metrics not enabled";
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
thread_ = std::thread(&MetricsHostReceiver::ServerLoop, this);
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace cuttlefish
|