diff options
| author | Qiu Shumin <shumin.qiu@intel.com> | 2015-06-30 03:18:31 +0000 |
|---|---|---|
| committer | shenshushi <shenshushi@Edk2> | 2015-06-30 03:18:31 +0000 |
| commit | e75390f02971bcd4d67a9696508050bee4936a01 (patch) | |
| tree | 10fc0928560c01484d1952adf3a5990cc2243c26 /ShellPkg/Application/Shell/ShellEnvVar.c | |
| parent | cb9a7ebabcd6b8a49dc0854b2f9592d732b5afbd (diff) | |
| download | edk2-platforms-e75390f02971bcd4d67a9696508050bee4936a01.tar.xz | |
ShellPkg: Use safe string functions to refine code.
Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Qiu Shumin <shumin.qiu@intel.com>
Reviewed-by: Jaben Carsey <jaben.carsey@intel.com>
Reviewed-by: Star Zeng <star.zeng@intel.com>
git-svn-id: https://svn.code.sf.net/p/edk2/code/trunk/edk2@17730 6f19259b-4bc3-4df7-8a09-765794883524
Diffstat (limited to 'ShellPkg/Application/Shell/ShellEnvVar.c')
| -rw-r--r-- | ShellPkg/Application/Shell/ShellEnvVar.c | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/ShellPkg/Application/Shell/ShellEnvVar.c b/ShellPkg/Application/Shell/ShellEnvVar.c index de09174c62..8ed14b34a2 100644 --- a/ShellPkg/Application/Shell/ShellEnvVar.c +++ b/ShellPkg/Application/Shell/ShellEnvVar.c @@ -339,7 +339,10 @@ SetEnvironmentVariables( //
// Copy the string into the Key, leaving the last character allocated as NULL to terminate
//
- StrnCpy(Node->Key, CurrentString, StrStr(CurrentString, L"=") - CurrentString);
+ StrCpyS( Node->Key,
+ StrStr(CurrentString, L"=") - CurrentString + 1,
+ CurrentString
+ );
//
// ValueSize = TotalSize - already removed size - size for '=' + size for terminator (the last 2 items cancel each other)
|