Add Mario support for fwupdate_tries

host/arch/x86/lib/crossystem_arch.c

 /* Copyright (c) 2011 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 <stdio.h>
 #include <string.h>
 #include <sys/types.h>
 #include <sys/stat.h>
 #include <unistd.h>
@@ skipping 66 matching lines @@
 #define ACPI_VBNV_PATH ACPI_BASE_PATH "/VBNV"
 #define ACPI_VDAT_PATH ACPI_BASE_PATH "/VDAT"
 
 /* Base name for GPIO files */
 #define GPIO_BASE_PATH "/sys/class/gpio"
 #define GPIO_EXPORT_PATH GPIO_BASE_PATH "/export"
 
 /* Filename for NVRAM file */
 #define NVRAM_PATH "/dev/nvram"
 
+/* Filename for legacy firmware update tries */
+#define NEED_FWUPDATE_PATH "/mnt/stateful_partition/.need_firmware_update"
+
 
 int VbReadNvStorage(VbNvContext* vnc) {
   FILE* f;
   int offs;
 
   /* Get the byte offset from VBNV */
   offs = ReadFileInt(ACPI_VBNV_PATH ".0");
   if (offs == -1)
     return -1;
   if (VBNV_BLOCK_SIZE > ReadFileInt(ACPI_VBNV_PATH ".1"))
@@ skipping 418 matching lines @@
       value = VbGetCmosRebootField(CMOSRF_RECOVERY);
   } else if (!strcasecmp(name,"dbg_reset")) {
     value = VbGetNvStorage(VBNV_DEBUG_RESET_MODE);
     if (-1 == value)
       value = VbGetCmosRebootField(CMOSRF_DEBUG_RESET);
   } else if (!strcasecmp(name,"fwb_tries")) {
     value = VbGetNvStorage(VBNV_TRY_B_COUNT);
     if (-1 == value)
       value = VbGetCmosRebootField(CMOSRF_TRY_B);
   }
+  /* Firmware update tries is now stored in the kernel field.  On
+   * older systems where it's not, it was stored in a file in the
+   * stateful partition. */
+  else if (!strcasecmp(name,"fwupdate_tries")) {
+    if (-1 != VbGetNvStorage(VBNV_KERNEL_FIELD))
+      return -1;  /* NvStorage supported; fail through arch-specific
+                   * implementation to normal implementation. */
+
+    /* Read value from file; missing file means value=0. */
+    value = ReadFileInt(NEED_FWUPDATE_PATH);
+    if (-1 == value)
+      value = 0;
+  }
 
   return value;
 }
 
 
 const char* VbGetArchPropertyString(const char* name, char* dest, int size) {
 
   if (!strcasecmp(name,"arch")) {
     return StrCopy(dest, "x86", size);
   } else if (!strcasecmp(name,"hwid")) {
@@ skipping 39 matching lines @@
     return VbSetCmosRebootField(CMOSRF_RECOVERY, value);
   } else if (!strcasecmp(name,"dbg_reset")) {
     if (0 == VbSetNvStorage(VBNV_DEBUG_RESET_MODE, value))
       return 0;
     return  VbSetCmosRebootField(CMOSRF_DEBUG_RESET, value);
   } else if (!strcasecmp(name,"fwb_tries")) {
     if (0 == VbSetNvStorage(VBNV_TRY_B_COUNT, value))
       return 0;
     return VbSetCmosRebootField(CMOSRF_TRY_B, value);
   }
+  /* Firmware update tries is now stored in the kernel field.  On
+   * older systems where it's not, it was stored in a file in the
+   * stateful partition. */
+  else if (!strcasecmp(name,"fwupdate_tries")) {
+    if (-1 != VbGetNvStorage(VBNV_KERNEL_FIELD))
+      return -1;  /* NvStorage supported; fail through arch-specific
+                   * implementation to normal implementation */
+
+    if (value) {
+      char buf[32];
+      snprintf(buf, sizeof(buf), "%d", value);
+      return WriteFile(NEED_FWUPDATE_PATH, buf, strlen(buf));
+    } else {
+      /* No update tries, so remove file if it exists. */
+      unlink(NEED_FWUPDATE_PATH);
+      return 0;
+    }
+  }
 
   return -1;
 }
 
 
 int VbSetArchPropertyString(const char* name, const char* value) {
   /* If there were settable architecture-dependent string properties,
    * they'd be here. */
   return -1;
 }