summaryrefslogtreecommitdiff
path: root/BaseTools/Source/C/EfiRom
diff options
context:
space:
mode:
authorHao Wu <hao.a.wu@intel.com>2016-10-11 11:08:41 +0800
committerHao Wu <hao.a.wu@intel.com>2016-11-08 16:37:09 +0800
commit47affb48e9baf3966842919acc0c419129c65392 (patch)
tree2c69e7359a4c7a15f2ef66d111d78bcd13a8d051 /BaseTools/Source/C/EfiRom
parentbab5ad2fd14bf8d1e9e688327a11136c8bfb523e (diff)
downloadedk2-47affb48e9baf3966842919acc0c419129c65392.zip
edk2-47affb48e9baf3966842919acc0c419129c65392.tar.gz
edk2-47affb48e9baf3966842919acc0c419129c65392.tar.bz2
BaseTools/EfiRom: Add checks for user/file inputs
Cc: Liming Gao <liming.gao@intel.com> Cc: Yonghong Zhu <yonghong.zhu@intel.com> Contributed-under: TianoCore Contribution Agreement 1.0 Signed-off-by: Hao Wu <hao.a.wu@intel.com> Reviewed-by: Liming Gao <liming.gao@intel.com>
Diffstat (limited to 'BaseTools/Source/C/EfiRom')
-rw-r--r--BaseTools/Source/C/EfiRom/EfiRom.c7
1 files changed, 6 insertions, 1 deletions
diff --git a/BaseTools/Source/C/EfiRom/EfiRom.c b/BaseTools/Source/C/EfiRom/EfiRom.c
index 622a12f..d95864a 100644
--- a/BaseTools/Source/C/EfiRom/EfiRom.c
+++ b/BaseTools/Source/C/EfiRom/EfiRom.c
@@ -979,7 +979,12 @@ Returns:
Error (NULL, 0, 2000, "Invalid parameter", "Missing output file name with %s option!", Argv[0]);
return STATUS_ERROR;
}
- strcpy (Options->OutFileName, Argv[1]);
+ if (strlen (Argv[1]) > MAX_PATH - 1) {
+ Error (NULL, 0, 2000, "Invalid parameter", "Output file name %s is too long!", Argv[1]);
+ return STATUS_ERROR;
+ }
+ strncpy (Options->OutFileName, Argv[1], MAX_PATH - 1);
+ Options->OutFileName[MAX_PATH - 1] = 0;
Argv++;
Argc--;