germ/germd.cc

// Copyright 2015 The Chromium OS 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 <stdint.h>
#include <sys/prctl.h>

#include <base/bind.h>
#include <base/command_line.h>
#include <base/logging.h>
#include <base/message_loop/message_loop.h>
#include <chromeos/syslog_logging.h>
#include <psyche/psyche_connection.h>
#include <psyche/psyche_daemon.h>

#include "germ/constants.h"
#include "germ/germ_host.h"
#include "germ/germ_zygote.h"
#include "germ/switches.h"

// TODO(usanghi): Find a better way to instantiate PsycheDaemon without
// extending it in each service.
namespace germ {

class GermDaemon : public psyche::PsycheDaemon {
 public:
  explicit GermDaemon(GermZygote* zygote) : host_(zygote) {}
  ~GermDaemon() override {}

 private:
  // Implement PsycheDaemon
  int OnInit() override {
    host_.RegisterWithDaemon(this);

    int return_code = PsycheDaemon::OnInit();
    if (return_code != 0) {
      LOG(ERROR) << "Could not initialize daemon.";
      return return_code;
    }
    if (!psyche_connection()->RegisterService(kGermServiceName, &host_)) {
      LOG(ERROR) << "Could not register with psyche.";
      return 1;
    }

    return 0;
  }

  GermHost host_;

  DISALLOW_COPY_AND_ASSIGN(GermDaemon);
};

}  // namespace germ

int main(int argc, char *argv[]) {
  base::CommandLine::Init(argc, argv);
  base::CommandLine* cmdline = base::CommandLine::ForCurrentProcess();

  int log_flags = chromeos::kLogToSyslog | chromeos::kLogHeader;
  if (cmdline->HasSwitch(germ::kLogToStderr)) {
    log_flags |= chromeos::kLogToStderr;
  }
  chromeos::InitLog(log_flags);

  // Take ownership of children spawned from the zygote.
  PCHECK(prctl(PR_SET_CHILD_SUBREAPER, 1) == 0);

  // The zygote must be started before any binder or daemon initialization.
  germ::GermZygote zygote;
  zygote.Start();

  germ::GermDaemon daemon(&zygote);
  return daemon.Run();
}