5 files changed, 34 insertions, 37 deletions
diff --git a/src/vendorcode/google/chromeos/Makefile.inc b/src/vendorcode/google/chromeos/Makefile.inc
index 253c30e326..dffbcd5b23 100644
--- a/src/vendorcode/google/chromeos/Makefile.inc
+++ b/src/vendorcode/google/chromeos/Makefile.inc
@@ -24,7 +24,7 @@ ramstage-$(CONFIG_CHROMEOS_RAMOOPS) += ramoops.c
 romstage-y += vpd_decode.c
 ramstage-y += vpd_decode.c cros_vpd.c vpd_mac.c vpd_serialno.c vpd_calibration.c
 ramstage-$(CONFIG_HAVE_REGULATORY_DOMAIN) += wrdd.c
-ifeq ($(CONFIG_ARCH_X86)$(CONFIG_ARCH_MIPS),)
+ifeq ($(CONFIG_ARCH_MIPS),)
 bootblock-y += watchdog.c
 ramstage-y += watchdog.c
 endif
diff --git a/src/vendorcode/google/chromeos/chromeos.h b/src/vendorcode/google/chromeos/chromeos.h
index 92c943cbd2..0a93ccd307 100644
--- a/src/vendorcode/google/chromeos/chromeos.h
+++ b/src/vendorcode/google/chromeos/chromeos.h
@@ -26,16 +26,10 @@
 void save_chromeos_gpios(void);
 
 #if CONFIG_CHROMEOS
-/* functions implemented in elog.c */
-void elog_add_boot_reason(void);
-
 /* functions implemented in watchdog.c */
-void elog_add_watchdog_reset(void);
 void mark_watchdog_tombstone(void);
 void reboot_from_watchdog(void);
 #else
-static inline void elog_add_boot_reason(void) { return; }
-static inline void elog_add_watchdog_reset(void) { return; }
 static inline void mark_watchdog_tombstone(void) { return; }
 static inline void reboot_from_watchdog(void) { return; }
 #endif /* CONFIG_CHROMEOS */
diff --git a/src/vendorcode/google/chromeos/elog.c b/src/vendorcode/google/chromeos/elog.c
index 206f9d9d7b..061c1776dd 100644
--- a/src/vendorcode/google/chromeos/elog.c
+++ b/src/vendorcode/google/chromeos/elog.c
@@ -13,6 +13,7 @@
  * GNU General Public License for more details.
  */
 
+#include <bootstate.h>
 #include <cbmem.h>
 #include <console/console.h>
 #include <elog.h>
@@ -21,19 +22,26 @@
 #include <vboot/vboot_common.h>
 #include <vboot_struct.h>
 
-void elog_add_boot_reason(void)
+static void elog_add_boot_reason(void *unused)
 {
-	if (vboot_developer_mode_enabled()) {
+	int rec = vboot_recovery_mode_enabled();
+	int dev = vboot_developer_mode_enabled();
+
+	if (!rec && !dev)
+		return;
+
+	if (rec) {
+		u8 reason = vboot_check_recovery_request();
+		elog_add_event_byte(ELOG_TYPE_CROS_RECOVERY_MODE, reason);
+		printk(BIOS_DEBUG, "%s: Logged recovery mode boot%s, "
+		       "reason: 0x%02x\n", __func__,
+		       dev ? " (Dev-switch on)" : "", reason);
+	}
+
+	if (dev) {
 		elog_add_event(ELOG_TYPE_CROS_DEVELOPER_MODE);
 		printk(BIOS_DEBUG, "%s: Logged dev mode boot\n", __func__);
-	} else if (vboot_recovery_mode_enabled()) {
-		u8 reason = vboot_check_recovery_request();
-		elog_add_event_byte(ELOG_TYPE_CROS_RECOVERY_MODE,
-			reason ? reason : ELOG_CROS_RECOVERY_MODE_BUTTON);
-		printk(BIOS_DEBUG, "%s: Logged recovery mode boot, "
-				"reason: 0x%02x\n", __func__, reason);
-	} else {
-		printk(BIOS_DEBUG, "%s: Normal mode boot, nothing "
-					"interesting to log\n", __func__);
 	}
 }
+
+BOOT_STATE_INIT_ENTRY(BS_POST_DEVICE, BS_ON_ENTRY, elog_add_boot_reason, NULL);
diff --git a/src/vendorcode/google/chromeos/gnvs.c b/src/vendorcode/google/chromeos/gnvs.c
index 5e14849004..1f13d1c195 100644
--- a/src/vendorcode/google/chromeos/gnvs.c
+++ b/src/vendorcode/google/chromeos/gnvs.c
@@ -52,24 +52,6 @@ void chromeos_init_vboot(chromeos_acpi_t *chromeos)
 		       ARRAY_SIZE(chromeos->vdat));
 #endif
 
-#if CONFIG_ELOG
-	if (vboot_developer_mode_enabled() ||
-	    (vboot_wants_oprom() && !vboot_recovery_mode_enabled()))
-		elog_add_event(ELOG_TYPE_CROS_DEVELOPER_MODE);
-	if (vboot_recovery_mode_enabled()) {
-		int reason = get_recovery_mode_from_vbnv();
-#if CONFIG_VBOOT
-		if (vboot_handoff && !reason) {
-			VbSharedDataHeader *sd = (VbSharedDataHeader *)
-				vboot_handoff->shared_data;
-			reason = sd->recovery_reason;
-		}
-#endif
-		elog_add_event_byte(ELOG_TYPE_CROS_RECOVERY_MODE,
-			    reason ? reason : ELOG_CROS_RECOVERY_MODE_BUTTON);
-	}
-#endif
-
 	chromeos_ram_oops_init(chromeos);
 }
 
diff --git a/src/vendorcode/google/chromeos/watchdog.c b/src/vendorcode/google/chromeos/watchdog.c
index a2b18b75e5..fdaa17761a 100644
--- a/src/vendorcode/google/chromeos/watchdog.c
+++ b/src/vendorcode/google/chromeos/watchdog.c
@@ -14,24 +14,37 @@
  */
 
 #include <arch/io.h>
+#include <assert.h>
+#include <bootstate.h>
 #include <console/console.h>
 #include <elog.h>
 #include <reset.h>
+#include <symbols.h>
 
 #include "chromeos.h"
 #include "symbols.h"
 
 #define WATCHDOG_TOMBSTONE_MAGIC	0x9d2f41a7
 
-void elog_add_watchdog_reset(void)
+DECLARE_OPTIONAL_REGION(watchdog_tombstone);
+
+static void elog_handle_watchdog_tombstone(void *unused)
 {
+	if (!_watchdog_tombstone_size)
+		return;
+
 	if (read32(_watchdog_tombstone) == WATCHDOG_TOMBSTONE_MAGIC)
 		elog_add_event(ELOG_TYPE_ASYNC_HW_TIMER_EXPIRED);
+
 	write32(_watchdog_tombstone, 0);
 }
 
+BOOT_STATE_INIT_ENTRY(BS_POST_DEVICE, BS_ON_ENTRY,
+		      elog_handle_watchdog_tombstone, NULL);
+
 void mark_watchdog_tombstone(void)
 {
+	assert(_watchdog_tombstone_size);
 	write32(_watchdog_tombstone, WATCHDOG_TOMBSTONE_MAGIC);
 }