Refine code to make it more safely.

Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Eric Dong <eric.dong@intel.com>
Reviewed-by: Jeff Fan <jeff.fan@intel.com>

git-svn-id: svn://svn.code.sf.net/p/edk2/code/trunk/edk2/IntelFrameworkModulePkg@15594 6f19259b-4bc3-4df7-8a09-765794883524