host/commands/metrics/host_receiver.cc

//
// 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