main.cc - android_system_update_engine - Gitiles

 // Copyright (c) 2012 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 <string>
 #include <vector>

 #include <base/at_exit.h>
 #include <base/command_line.h>
 #include <base/file_util.h>
 #include <base/logging.h>
 #include <base/string_util.h>
 #include <base/stringprintf.h>
 #include <gflags/gflags.h>
 #include <glib.h>
 #include <metrics/metrics_library.h>
 #include <sys/types.h>
 #include <sys/stat.h>

 #include "update_engine/certificate_checker.h"
 #include "update_engine/dbus_constants.h"
 #include "update_engine/dbus_interface.h"
 #include "update_engine/dbus_service.h"
 #include "update_engine/real_system_state.h"
 #include "update_engine/subprocess.h"
 #include "update_engine/terminator.h"
 #include "update_engine/update_attempter.h"
 #include "update_engine/update_check_scheduler.h"
 #include "update_engine/utils.h"

 extern "C" {
 #include "update_engine/update_engine.dbusserver.h"
 }

 DEFINE_bool(logtostderr, false,
             "Write logs to stderr instead of to a file in log_dir.");
 DEFINE_bool(foreground, false,
             "Don't daemon()ize; run in foreground.");

 using std::string;
 using std::vector;

 namespace chromeos_update_engine {

 gboolean UpdateBootFlags(void* arg) {
   reinterpret_cast<UpdateAttempter*>(arg)->UpdateBootFlags();
   return FALSE;  // Don't call this callback again
 }

 gboolean BroadcastStatus(void* arg) {
   reinterpret_cast<UpdateAttempter*>(arg)->BroadcastStatus();
   return FALSE;  // Don't call this callback again
 }

 gboolean UpdateEngineStarted(gpointer user_data) {
   reinterpret_cast<UpdateAttempter*>(user_data)->UpdateEngineStarted();
   return FALSE; // Remove idle source (e.g. don't do the callback again)
 }

 namespace {

 void SetupDbusService(UpdateEngineService* service) {
   DBusGConnection *bus;
   DBusGProxy *proxy;
   GError *error = NULL;

   bus = dbus_g_bus_get(DBUS_BUS_SYSTEM, &error);
   LOG_IF(FATAL, !bus) << "Failed to get bus: "
                       << utils::GetAndFreeGError(&error);
   proxy = dbus_g_proxy_new_for_name(bus,
                                     DBUS_SERVICE_DBUS,
                                     DBUS_PATH_DBUS,
                                     DBUS_INTERFACE_DBUS);
   guint32 request_name_ret;
   if (!org_freedesktop_DBus_request_name(proxy,
                                          kUpdateEngineServiceName,
                                          0,
                                          &request_name_ret,
                                          &error)) {
     LOG(FATAL) << "Failed to get name: " << utils::GetAndFreeGError(&error);
   }
   if (request_name_ret != DBUS_REQUEST_NAME_REPLY_PRIMARY_OWNER) {
     g_warning("Got result code %u from requesting name", request_name_ret);
     LOG(FATAL) << "Got result code " << request_name_ret
                << " from requesting name, but expected "
                << DBUS_REQUEST_NAME_REPLY_PRIMARY_OWNER;
   }
   dbus_g_connection_register_g_object(bus,
                                       "/org/chromium/UpdateEngine",
                                       G_OBJECT(service));
 }

 void SetupLogSymlink(const string& symlink_path, const string& log_path) {
   // TODO(petkov): To ensure a smooth transition between non-timestamped and
   // timestamped logs, move an existing log to start the first timestamped
   // one. This code can go away once all clients are switched to this version or
   // we stop caring about the old-style logs.
   if (utils::FileExists(symlink_path.c_str()) &&
       !utils::IsSymlink(symlink_path.c_str())) {
     file_util::ReplaceFile(FilePath(symlink_path), FilePath(log_path));
   }
   file_util::Delete(FilePath(symlink_path), true);
   if (symlink(log_path.c_str(), symlink_path.c_str()) == -1) {
     PLOG(ERROR) << "Unable to create symlink " << symlink_path
                 << " pointing at " << log_path;
   }
 }

 string GetTimeAsString(time_t utime) {
   struct tm tm;
   CHECK(localtime_r(&utime, &tm) == &tm);
   char str[16];
   CHECK(strftime(str, sizeof(str), "%Y%m%d-%H%M%S", &tm) == 15);
   return str;
 }

 string SetupLogFile(const string& kLogsRoot) {
   const string kLogSymlink = kLogsRoot + "/update_engine.log";
   const string kLogsDir = kLogsRoot + "/update_engine";
   const string kLogPath =
       StringPrintf("%s/update_engine.%s",
                    kLogsDir.c_str(),
                    GetTimeAsString(::time(NULL)).c_str());
   mkdir(kLogsDir.c_str(), 0755);
   SetupLogSymlink(kLogSymlink, kLogPath);
   return kLogSymlink;
 }

 void SetupLogging() {
   // Log to stderr initially.
   logging::InitLogging(NULL,
                        logging::LOG_ONLY_TO_SYSTEM_DEBUG_LOG,
                        logging::DONT_LOCK_LOG_FILE,
                        logging::APPEND_TO_OLD_LOG_FILE,
                        logging::DISABLE_DCHECK_FOR_NON_OFFICIAL_RELEASE_BUILDS);
   if (FLAGS_logtostderr) {
     return;
   }
   const string log_file = SetupLogFile("/var/log");
   logging::InitLogging(log_file.c_str(),
                        logging::LOG_ONLY_TO_FILE,
                        logging::DONT_LOCK_LOG_FILE,
                        logging::APPEND_TO_OLD_LOG_FILE,
                        logging::DISABLE_DCHECK_FOR_NON_OFFICIAL_RELEASE_BUILDS);
 }

 }  // namespace {}
 }  // namespace chromeos_update_engine

 int main(int argc, char** argv) {
   ::g_type_init();
   dbus_threads_init_default();
   base::AtExitManager exit_manager;  // Required for base/rand_util.h.
   chromeos_update_engine::Terminator::Init();
   chromeos_update_engine::Subprocess::Init();
   google::ParseCommandLineFlags(&argc, &argv, true);
   CommandLine::Init(argc, argv);
   chromeos_update_engine::SetupLogging();
   if (!FLAGS_foreground)
     PLOG_IF(FATAL, daemon(0, 0) == 1) << "daemon() failed";

   LOG(INFO) << "Chrome OS Update Engine starting";

   // Ensure that all written files have safe permissions.
   // This is a mask, so we _block_ execute for the owner, and ALL
   // permissions for other users.
   // Done _after_ log file creation.
   umask(S_IXUSR | S_IRWXG | S_IRWXO);

   // Create the single GMainLoop
   GMainLoop* loop = g_main_loop_new(g_main_context_default(), FALSE);

   chromeos_update_engine::RealSystemState real_system_state;
   // TODO(garnold) s/false/true/ once we decide to activate actual GPIO-based
   // protocol for testing of MP-signed images (chromium-os:25400).
   LOG_IF(ERROR, !real_system_state.Initialize(false))
       << "Failed to initialize system state.";
   chromeos_update_engine::UpdateAttempter *update_attempter =
       real_system_state.update_attempter();
   CHECK(update_attempter);

   // Sets static members for the certificate checker.
   chromeos_update_engine::CertificateChecker::set_system_state(
       &real_system_state);
   chromeos_update_engine::OpenSSLWrapper openssl_wrapper;
   chromeos_update_engine::CertificateChecker::set_openssl_wrapper(
       &openssl_wrapper);

   // Create the dbus service object:
   dbus_g_object_type_install_info(UPDATE_ENGINE_TYPE_SERVICE,
                                   &dbus_glib_update_engine_service_object_info);
   UpdateEngineService* service =
       UPDATE_ENGINE_SERVICE(g_object_new(UPDATE_ENGINE_TYPE_SERVICE, NULL));
   service->system_state_ = &real_system_state;
   update_attempter->set_dbus_service(service);
   chromeos_update_engine::SetupDbusService(service);

   // Schedule periodic update checks.
   chromeos_update_engine::UpdateCheckScheduler scheduler(update_attempter,
                                                          &real_system_state);
   scheduler.Run();

   // Update boot flags after 45 seconds.
   g_timeout_add_seconds(45,
                         &chromeos_update_engine::UpdateBootFlags,
                         update_attempter);

   // Broadcast the update engine status on startup to ensure consistent system
   // state on crashes.
   g_idle_add(&chromeos_update_engine::BroadcastStatus, update_attempter);

   // Run the UpdateEngineStarted() method on |update_attempter|.
   g_idle_add(&chromeos_update_engine::UpdateEngineStarted, update_attempter);

   // Run the main loop until exit time:
   g_main_loop_run(loop);

   // Cleanup:
   g_main_loop_unref(loop);
   update_attempter->set_dbus_service(NULL);
   g_object_unref(G_OBJECT(service));

   LOG(INFO) << "Chrome OS Update Engine terminating";
   return 0;
 }
	// Copyright (c) 2012 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 <string>
	#include <vector>

	#include <base/at_exit.h>
	#include <base/command_line.h>
	#include <base/file_util.h>
	#include <base/logging.h>
	#include <base/string_util.h>
	#include <base/stringprintf.h>
	#include <gflags/gflags.h>
	#include <glib.h>
	#include <metrics/metrics_library.h>
	#include <sys/types.h>
	#include <sys/stat.h>

	#include "update_engine/certificate_checker.h"
	#include "update_engine/dbus_constants.h"
	#include "update_engine/dbus_interface.h"
	#include "update_engine/dbus_service.h"
	#include "update_engine/real_system_state.h"
	#include "update_engine/subprocess.h"
	#include "update_engine/terminator.h"
	#include "update_engine/update_attempter.h"
	#include "update_engine/update_check_scheduler.h"
	#include "update_engine/utils.h"

	extern "C" {
	#include "update_engine/update_engine.dbusserver.h"
	}

	DEFINE_bool(logtostderr, false,
	"Write logs to stderr instead of to a file in log_dir.");
	DEFINE_bool(foreground, false,
	"Don't daemon()ize; run in foreground.");

	using std::string;
	using std::vector;

	namespace chromeos_update_engine {

	gboolean UpdateBootFlags(void* arg) {
	reinterpret_cast<UpdateAttempter*>(arg)->UpdateBootFlags();
	return FALSE; // Don't call this callback again
	}

	gboolean BroadcastStatus(void* arg) {
	reinterpret_cast<UpdateAttempter*>(arg)->BroadcastStatus();
	return FALSE; // Don't call this callback again
	}

	gboolean UpdateEngineStarted(gpointer user_data) {
	reinterpret_cast<UpdateAttempter*>(user_data)->UpdateEngineStarted();
	return FALSE; // Remove idle source (e.g. don't do the callback again)
	}

	namespace {

	void SetupDbusService(UpdateEngineService* service) {
	DBusGConnection *bus;
	DBusGProxy *proxy;
	GError *error = NULL;

	bus = dbus_g_bus_get(DBUS_BUS_SYSTEM, &error);
	LOG_IF(FATAL, !bus) << "Failed to get bus: "
	<< utils::GetAndFreeGError(&error);
	proxy = dbus_g_proxy_new_for_name(bus,
	DBUS_SERVICE_DBUS,
	DBUS_PATH_DBUS,
	DBUS_INTERFACE_DBUS);
	guint32 request_name_ret;
	if (!org_freedesktop_DBus_request_name(proxy,
	kUpdateEngineServiceName,
	0,
	&request_name_ret,
	&error)) {
	LOG(FATAL) << "Failed to get name: " << utils::GetAndFreeGError(&error);
	}
	if (request_name_ret != DBUS_REQUEST_NAME_REPLY_PRIMARY_OWNER) {
	g_warning("Got result code %u from requesting name", request_name_ret);
	LOG(FATAL) << "Got result code " << request_name_ret
	<< " from requesting name, but expected "
	<< DBUS_REQUEST_NAME_REPLY_PRIMARY_OWNER;
	}
	dbus_g_connection_register_g_object(bus,
	"/org/chromium/UpdateEngine",
	G_OBJECT(service));
	}

	void SetupLogSymlink(const string& symlink_path, const string& log_path) {
	// TODO(petkov): To ensure a smooth transition between non-timestamped and
	// timestamped logs, move an existing log to start the first timestamped
	// one. This code can go away once all clients are switched to this version or
	// we stop caring about the old-style logs.
	if (utils::FileExists(symlink_path.c_str()) &&
	!utils::IsSymlink(symlink_path.c_str())) {
	file_util::ReplaceFile(FilePath(symlink_path), FilePath(log_path));
	}
	file_util::Delete(FilePath(symlink_path), true);
	if (symlink(log_path.c_str(), symlink_path.c_str()) == -1) {
	PLOG(ERROR) << "Unable to create symlink " << symlink_path
	<< " pointing at " << log_path;
	}
	}

	string GetTimeAsString(time_t utime) {
	struct tm tm;
	CHECK(localtime_r(&utime, &tm) == &tm);
	char str[16];
	CHECK(strftime(str, sizeof(str), "%Y%m%d-%H%M%S", &tm) == 15);
	return str;
	}

	string SetupLogFile(const string& kLogsRoot) {
	const string kLogSymlink = kLogsRoot + "/update_engine.log";
	const string kLogsDir = kLogsRoot + "/update_engine";
	const string kLogPath =
	StringPrintf("%s/update_engine.%s",
	kLogsDir.c_str(),
	GetTimeAsString(::time(NULL)).c_str());
	mkdir(kLogsDir.c_str(), 0755);
	SetupLogSymlink(kLogSymlink, kLogPath);
	return kLogSymlink;
	}

	void SetupLogging() {
	// Log to stderr initially.
	logging::InitLogging(NULL,
	logging::LOG_ONLY_TO_SYSTEM_DEBUG_LOG,
	logging::DONT_LOCK_LOG_FILE,
	logging::APPEND_TO_OLD_LOG_FILE,
	logging::DISABLE_DCHECK_FOR_NON_OFFICIAL_RELEASE_BUILDS);
	if (FLAGS_logtostderr) {
	return;
	}
	const string log_file = SetupLogFile("/var/log");
	logging::InitLogging(log_file.c_str(),
	logging::LOG_ONLY_TO_FILE,
	logging::DONT_LOCK_LOG_FILE,
	logging::APPEND_TO_OLD_LOG_FILE,
	logging::DISABLE_DCHECK_FOR_NON_OFFICIAL_RELEASE_BUILDS);
	}

	} // namespace {}
	} // namespace chromeos_update_engine

	int main(int argc, char** argv) {
	::g_type_init();
	dbus_threads_init_default();
	base::AtExitManager exit_manager; // Required for base/rand_util.h.
	chromeos_update_engine::Terminator::Init();
	chromeos_update_engine::Subprocess::Init();
	google::ParseCommandLineFlags(&argc, &argv, true);
	CommandLine::Init(argc, argv);
	chromeos_update_engine::SetupLogging();
	if (!FLAGS_foreground)
	PLOG_IF(FATAL, daemon(0, 0) == 1) << "daemon() failed";

	LOG(INFO) << "Chrome OS Update Engine starting";

	// Ensure that all written files have safe permissions.
	// This is a mask, so we _block_ execute for the owner, and ALL
	// permissions for other users.
	// Done _after_ log file creation.
	umask(S_IXUSR \| S_IRWXG \| S_IRWXO);

	// Create the single GMainLoop
	GMainLoop* loop = g_main_loop_new(g_main_context_default(), FALSE);

	chromeos_update_engine::RealSystemState real_system_state;
	// TODO(garnold) s/false/true/ once we decide to activate actual GPIO-based
	// protocol for testing of MP-signed images (chromium-os:25400).
	LOG_IF(ERROR, !real_system_state.Initialize(false))
	<< "Failed to initialize system state.";
	chromeos_update_engine::UpdateAttempter *update_attempter =
	real_system_state.update_attempter();
	CHECK(update_attempter);

	// Sets static members for the certificate checker.
	chromeos_update_engine::CertificateChecker::set_system_state(
	&real_system_state);
	chromeos_update_engine::OpenSSLWrapper openssl_wrapper;
	chromeos_update_engine::CertificateChecker::set_openssl_wrapper(
	&openssl_wrapper);

	// Create the dbus service object:
	dbus_g_object_type_install_info(UPDATE_ENGINE_TYPE_SERVICE,
	&dbus_glib_update_engine_service_object_info);
	UpdateEngineService* service =
	UPDATE_ENGINE_SERVICE(g_object_new(UPDATE_ENGINE_TYPE_SERVICE, NULL));
	service->system_state_ = &real_system_state;
	update_attempter->set_dbus_service(service);
	chromeos_update_engine::SetupDbusService(service);

	// Schedule periodic update checks.
	chromeos_update_engine::UpdateCheckScheduler scheduler(update_attempter,
	&real_system_state);
	scheduler.Run();

	// Update boot flags after 45 seconds.
	g_timeout_add_seconds(45,
	&chromeos_update_engine::UpdateBootFlags,
	update_attempter);

	// Broadcast the update engine status on startup to ensure consistent system
	// state on crashes.
	g_idle_add(&chromeos_update_engine::BroadcastStatus, update_attempter);

	// Run the UpdateEngineStarted() method on \|update_attempter\|.
	g_idle_add(&chromeos_update_engine::UpdateEngineStarted, update_attempter);

	// Run the main loop until exit time:
	g_main_loop_run(loop);

	// Cleanup:
	g_main_loop_unref(loop);
	update_attempter->set_dbus_service(NULL);
	g_object_unref(G_OBJECT(service));

	LOG(INFO) << "Chrome OS Update Engine terminating";
	return 0;
	}