X-Git-Url: http://git.ozlabs.org/?a=blobdiff_plain;f=discover%2Fgrub2%2Fbuiltins.c;h=8bff7328108479f1b95dbd2d3196cdeada238808;hb=e544283809454252c2c2d3148492362ad10643ba;hp=75110760082ec51324f685e5c5b84bebac5e5745;hpb=13e41764a7450302b874668bc1a3fbf6be25b781;p=petitboot

diff --git a/discover/grub2/builtins.c b/discover/grub2/builtins.c
index 7511076..8bff732 100644
--- a/discover/grub2/builtins.c
+++ b/discover/grub2/builtins.c
@@ -124,36 +124,60 @@ static int builtin_search(struct grub2_script *script,
 	return 0;
 }
 
+/* Note that GRUB does not follow symlinks in evaluating all file
+ * tests but -s, unlike below. However, it seems like a bad idea to
+ * emulate GRUB's behavior (e.g., it would take extra work), so we
+ * implement the behavior that coreutils' test binary has. */
 static bool builtin_test_op_file(struct grub2_script *script, char op,
 		const char *file)
 {
 	bool result;
-	int len, rc;
-	char *buf;
+	int rc;
+	struct stat statbuf;
 
-	rc = parser_request_file(script->ctx, script->ctx->device,
-			file, &buf, &len);
+	rc = parser_stat_path(script->ctx, script->ctx->device,
+			file, &statbuf);
 	if (rc)
 		return false;
 
 	switch (op) {
 	case 's':
 		/* -s: return true if file exists and has non-zero size */
-		result = len > 0;
+		result = statbuf.st_size > 0;
 		break;
 	case 'f':
-		/* -f: return true if file exists */
-		result = true;
+		/* -f: return true if file exists and is not a directory. This is
+		 * different than the behavior of "test", but is what GRUB does
+		 * (though note as above that we follow symlinks unlike GRUB). */
+		result = !S_ISDIR(statbuf.st_mode);
 		break;
 	default:
 		result = false;
 
 	}
 
-	talloc_free(buf);
 	return result;
 }
 
+/* See comment at builtin_test_op_file for differences between how
+ * GRUB implements file tests versus Petitboot's GRUB parser. */
+static bool builtin_test_op_dir(struct grub2_script *script, char op,
+		const char *dir)
+{
+	int rc;
+	struct stat statbuf;
+
+	if (op != 'd')
+		return false;
+
+	rc = parser_stat_path(script->ctx, script->ctx->device, dir, &statbuf);
+	if (rc) {
+		return false;
+	}
+
+	return S_ISDIR(statbuf.st_mode);
+}
+
 static bool builtin_test_op(struct grub2_script *script,
 		int argc, char **argv, int *consumed)
 {
@@ -207,6 +231,11 @@ static bool builtin_test_op(struct grub2_script *script,
 			*consumed = 2;
 			return builtin_test_op_file(script, op[1], a1);
 		}
+
+		if (!strcmp(op, "-d")) {
+			*consumed = 2;
+			return builtin_test_op_dir(script, op[1], a1);
+		}
 	}
 
 	op = argv[0];