3c4dd19
From c93a27f2e6a323a4985e6fe902eb5c6447fca572 Mon Sep 17 00:00:00 2001
da1991e
From: Matthew Garrett <mjg59@srcf.ucam.org>
da1991e
Date: Mon, 10 Aug 2015 15:27:12 -0700
3c4dd19
Subject: [PATCH 221/229] Measure commands
da1991e
da1991e
Measure each command executed by grub, which includes script execution.
da1991e
---
da1991e
 grub-core/script/execute.c | 25 +++++++++++++++++++++++--
da1991e
 include/grub/tpm.h         |  1 +
da1991e
 2 files changed, 24 insertions(+), 2 deletions(-)
da1991e
da1991e
diff --git a/grub-core/script/execute.c b/grub-core/script/execute.c
da1991e
index cf6cd6601d6..9ae04a05160 100644
da1991e
--- a/grub-core/script/execute.c
da1991e
+++ b/grub-core/script/execute.c
da1991e
@@ -30,6 +30,7 @@
da1991e
 #ifdef GRUB_MACHINE_IEEE1275
da1991e
 #include <grub/ieee1275/ieee1275.h>
da1991e
 #endif
da1991e
+#include <grub/tpm.h>
da1991e
 
da1991e
 /* Max digits for a char is 3 (0xFF is 255), similarly for an int it
da1991e
    is sizeof (int) * 3, and one extra for a possible -ve sign.  */
da1991e
@@ -967,8 +968,9 @@ grub_script_execute_cmdline (struct grub_script_cmd *cmd)
da1991e
   grub_err_t ret = 0;
da1991e
   grub_script_function_t func = 0;
da1991e
   char errnobuf[18];
da1991e
-  char *cmdname;
da1991e
-  int argc;
da1991e
+  char *cmdname, *cmdstring;
da1991e
+  int argc, offset = 0, cmdlen = 0;
da1991e
+  unsigned int i;
da1991e
   char **args;
da1991e
   int invert;
da1991e
   struct grub_script_argv argv = { 0, 0, 0 };
da1991e
@@ -977,6 +979,25 @@ grub_script_execute_cmdline (struct grub_script_cmd *cmd)
da1991e
   if (grub_script_arglist_to_argv (cmdline->arglist, &argv) || ! argv.args[0])
da1991e
     return grub_errno;
da1991e
 
da1991e
+  for (i = 0; i < argv.argc; i++) {
da1991e
+	  cmdlen += grub_strlen (argv.args[i]) + 1;
da1991e
+  }
da1991e
+
da1991e
+  cmdstring = grub_malloc (cmdlen);
da1991e
+  if (!cmdstring)
da1991e
+  {
da1991e
+	  return grub_error (GRUB_ERR_OUT_OF_MEMORY,
da1991e
+			     N_("cannot allocate command buffer"));
da1991e
+  }
da1991e
+
da1991e
+  for (i = 0; i < argv.argc; i++) {
da1991e
+	  offset += grub_snprintf (cmdstring + offset, cmdlen - offset, "%s ",
da1991e
+				   argv.args[i]);
da1991e
+  }
da1991e
+  cmdstring[cmdlen-1]= '\0';
da1991e
+  grub_tpm_measure ((unsigned char *)cmdstring, cmdlen, GRUB_COMMAND_PCR,
da1991e
+		    cmdstring);
da1991e
+  grub_free(cmdstring);
da1991e
   invert = 0;
da1991e
   argc = argv.argc - 1;
da1991e
   args = argv.args + 1;
da1991e
diff --git a/include/grub/tpm.h b/include/grub/tpm.h
da1991e
index 40d3cf65ba6..7fc9d77d277 100644
da1991e
--- a/include/grub/tpm.h
da1991e
+++ b/include/grub/tpm.h
da1991e
@@ -30,6 +30,7 @@
da1991e
 #define GRUB_KERNEL_PCR 10
da1991e
 #define GRUB_INITRD_PCR 11
da1991e
 #define GRUB_CMDLINE_PCR 12
da1991e
+#define GRUB_COMMAND_PCR 13
da1991e
 
da1991e
 #define TPM_TAG_RQU_COMMAND 0x00C1
da1991e
 #define TPM_ORD_Extend 0x14
da1991e
-- 
da1991e
2.15.0
da1991e