[mirror_edk2.git] / BaseTools / Source / C / VfrCompile / VfrCompiler.cpp

/** @file\r
\r
  VfrCompiler main class and main function.\r
\r
Copyright (c) 2004 - 2019, Intel Corporation. All rights reserved.<BR>\r
SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include "stdio.h"\r
#include "stdlib.h"\r
#include "string.h"\r
#include "VfrCompiler.h"\r
#include "CommonLib.h"\r
#include "EfiUtilityMsgs.h"\r
\r
PACKAGE_DATA  gCBuffer;\r
PACKAGE_DATA  gRBuffer;\r
CVfrStringDB  gCVfrStringDB;\r
\r
VOID\r
CVfrCompiler::DebugError (\r
  IN CHAR8         *FileName,\r
  IN UINT32        LineNumber,\r
  IN UINT32        MessageCode,\r
  IN CONST CHAR8   *Text,\r
  IN CONST CHAR8   *MsgFmt,\r
  ...\r
  )\r
{\r
  va_list List;\r
  va_start (List, MsgFmt);\r
  PrintMessage ((CHAR8 *) "ERROR", FileName, LineNumber, MessageCode, (CHAR8 *) Text, (CHAR8 *) MsgFmt, List);\r
  va_end (List);\r
}\r
\r
VOID\r
CVfrCompiler::SET_RUN_STATUS (\r
  IN COMPILER_RUN_STATUS Status\r
  )\r
{\r
  mRunStatus = Status;\r
}\r
\r
BOOLEAN\r
CVfrCompiler::IS_RUN_STATUS (\r
  IN COMPILER_RUN_STATUS Status\r
  )\r
{\r
  return mRunStatus == Status;\r
}\r
\r
VOID\r
CVfrCompiler::OptionInitialization (\r
  IN INT32      Argc,\r
  IN CHAR8      **Argv\r
  )\r
{\r
  INT32         Index;\r
  EFI_STATUS    Status;\r
\r
  Status = EFI_SUCCESS;\r
  SetUtilityName ((CHAR8*) PROGRAM_NAME);\r
\r
  mOptions.VfrFileName                   = NULL;\r
  mOptions.RecordListFile                = NULL;\r
  mOptions.CreateRecordListFile          = FALSE;\r
  mOptions.CreateIfrPkgFile              = FALSE;\r
  mOptions.PkgOutputFileName             = NULL;\r
  mOptions.COutputFileName               = NULL;\r
  mOptions.OutputDirectory               = NULL;\r
  mOptions.PreprocessorOutputFileName    = NULL;\r
  mOptions.VfrBaseFileName               = NULL;\r
  mOptions.IncludePaths                  = NULL;\r
  mOptions.SkipCPreprocessor             = TRUE;\r
  mOptions.CPreprocessorOptions          = NULL;\r
  mOptions.HasOverrideClassGuid          = FALSE;\r
  mOptions.WarningAsError                = FALSE;\r
  mOptions.AutoDefault                   = FALSE;\r
  mOptions.CheckDefault                  = FALSE;\r
  memset (&mOptions.OverrideClassGuid, 0, sizeof (EFI_GUID));\r
\r
  if (Argc == 1) {\r
    Usage ();\r
    SET_RUN_STATUS (STATUS_DEAD);\r
    return;\r
  }\r
\r
  for (Index = 1; (Index < Argc) && (Argv[Index][0] == '-'); Index++) {\r
    if ((stricmp(Argv[Index], "-h") == 0) || (stricmp(Argv[Index], "--help") == 0)) {\r
      Usage ();\r
      SET_RUN_STATUS (STATUS_DEAD);\r
      return;\r
    } else if (stricmp(Argv[Index], "--version") == 0) {\r
      Version ();\r
      SET_RUN_STATUS (STATUS_DEAD);\r
      return;\r
    } else if (stricmp(Argv[Index], "-l") == 0) {\r
      mOptions.CreateRecordListFile = TRUE;\r
      gCIfrRecordInfoDB.TurnOn ();\r
    } else if (stricmp(Argv[Index], "-i") == 0) {\r
      Index++;\r
      if ((Index >= Argc) || (Argv[Index][0] == '-')) {\r
        DebugError (NULL, 0, 1001, "Missing option", "-i missing path argument");\r
        goto Fail;\r
      }\r
\r
      AppendIncludePath(Argv[Index]);\r
    } else if (stricmp(Argv[Index], "-o") == 0 || stricmp(Argv[Index], "--output-directory") == 0 || stricmp(Argv[Index], "-od") == 0) {\r
      Index++;\r
      if ((Index >= Argc) || (Argv[Index][0] == '-')) {\r
        DebugError (NULL, 0, 1001, "Missing option", "-o missing output directory name");\r
        goto Fail;\r
      }\r
\r
      mOptions.OutputDirectory = (CHAR8 *) malloc (strlen (Argv[Index]) + strlen ("\\") + 1);\r
      if (mOptions.OutputDirectory == NULL) {\r
        DebugError (NULL, 0, 4001, "Resource: memory can't be allocated", NULL);\r
        goto Fail;\r
      }\r
      strcpy (mOptions.OutputDirectory, Argv[Index]);\r
\r
      CHAR8 lastChar = mOptions.OutputDirectory[strlen(mOptions.OutputDirectory) - 1];\r
      if ((lastChar != '/') && (lastChar != '\\')) {\r
        if (strchr(mOptions.OutputDirectory, '/') != NULL) {\r
          strcat (mOptions.OutputDirectory, "/");\r
        } else {\r
          strcat (mOptions.OutputDirectory, "\\");\r
        }\r
      }\r
      DebugMsg (NULL, 0, 9, (CHAR8 *) "Output Directory", (CHAR8 *) "%s", mOptions.OutputDirectory);\r
    } else if (stricmp(Argv[Index], "-b") == 0 || stricmp(Argv[Index], "--create-ifr-package") == 0 || stricmp(Argv[Index], "-ibin") == 0) {\r
      mOptions.CreateIfrPkgFile = TRUE;\r
    } else if (stricmp(Argv[Index], "-n") == 0 || stricmp(Argv[Index], "--no-pre-processing") == 0 || stricmp(Argv[Index], "-nopp") == 0) {\r
      mOptions.SkipCPreprocessor = TRUE;\r
    } else if (stricmp(Argv[Index], "-f") == 0 || stricmp(Argv[Index], "--pre-processing-flag") == 0 || stricmp(Argv[Index], "-ppflag") == 0) {\r
      Index++;\r
      if ((Index >= Argc) || (Argv[Index][0] == '-')) {\r
        DebugError (NULL, 0, 1001, "Missing option", "-od - missing C-preprocessor argument");\r
        goto Fail;\r
      }\r
\r
      AppendCPreprocessorOptions (Argv[Index]);\r
    } else if (stricmp(Argv[Index], "-s") == 0|| stricmp(Argv[Index], "--string-db") == 0) {\r
      Index++;\r
      if ((Index >= Argc) || (Argv[Index][0] == '-')) {\r
        DebugError (NULL, 0, 1001, "Missing option", "-s missing input string file name");\r
        goto Fail;\r
      }\r
      gCVfrStringDB.SetStringFileName(Argv[Index]);\r
      DebugMsg (NULL, 0, 9, (CHAR8 *) "Input string file path", (CHAR8 *) "%s", Argv[Index]);\r
    } else if ((stricmp (Argv[Index], "-g") == 0) || (stricmp (Argv[Index], "--guid") == 0)) {\r
      Index++;\r
      Status = StringToGuid (Argv[Index], &mOptions.OverrideClassGuid);\r
      if (EFI_ERROR (Status)) {\r
        DebugError (NULL, 0, 1000, "Invalid format:", "%s", Argv[Index]);\r
        goto Fail;\r
      }\r
      mOptions.HasOverrideClassGuid = TRUE;\r
    } else if (stricmp(Argv[Index], "-w") == 0 || stricmp(Argv[Index], "--warning-as-error") == 0) {\r
      mOptions.WarningAsError = TRUE;\r
    } else if (stricmp(Argv[Index], "-a") == 0 ||stricmp(Argv[Index], "--autodefault") == 0) {\r
      mOptions.AutoDefault = TRUE;\r
    } else if (stricmp(Argv[Index], "-d") == 0 ||stricmp(Argv[Index], "--checkdefault") == 0) {\r
      mOptions.CheckDefault = TRUE;\r
    } else {\r
      DebugError (NULL, 0, 1000, "Unknown option", "unrecognized option %s", Argv[Index]);\r
      goto Fail;\r
    }\r
  }\r
\r
  if (Index != Argc - 1) {\r
    DebugError (NULL, 0, 1001, "Missing option", "VFR file name is not specified.");\r
    goto Fail;\r
  } else {\r
    mOptions.VfrFileName = (CHAR8 *) malloc (strlen (Argv[Index]) + 1);\r
    if (mOptions.VfrFileName == NULL) {\r
      DebugError (NULL, 0, 4001, "Resource: memory can't be allocated", NULL);\r
      goto Fail;\r
    }\r
    strcpy (mOptions.VfrFileName, Argv[Index]);\r
\r
    if (mOptions.OutputDirectory == NULL) {\r
      mOptions.OutputDirectory = (CHAR8 *) malloc (1);\r
      if (mOptions.OutputDirectory == NULL) {\r
        DebugError (NULL, 0, 4001, "Resource: memory can't be allocated", NULL);\r
        goto Fail;\r
      }\r
      mOptions.OutputDirectory[0] = '\0';\r
    }\r
  }\r
\r
  if (SetBaseFileName() != 0) {\r
    goto Fail;\r
  }\r
  if (SetPkgOutputFileName () != 0) {\r
    goto Fail;\r
  }\r
  if (SetCOutputFileName() != 0) {\r
    goto Fail;\r
  }\r
  if (SetPreprocessorOutputFileName () != 0) {\r
    goto Fail;\r
  }\r
  if (SetRecordListFileName () != 0) {\r
    goto Fail;\r
  }\r
  return;\r
\r
Fail:\r
  SET_RUN_STATUS (STATUS_DEAD);\r
\r
  mOptions.CreateRecordListFile          = FALSE;\r
  mOptions.CreateIfrPkgFile              = FALSE;\r
\r
  if (mOptions.VfrFileName != NULL) {\r
    free (mOptions.VfrFileName);\r
    mOptions.VfrFileName                 = NULL;\r
  }\r
  if (mOptions.VfrBaseFileName != NULL) {\r
    free (mOptions.VfrBaseFileName);\r
    mOptions.VfrBaseFileName             = NULL;\r
  }\r
  if (mOptions.OutputDirectory != NULL) {\r
    free (mOptions.OutputDirectory);\r
    mOptions.OutputDirectory             = NULL;\r
  }\r
  if (mOptions.PkgOutputFileName != NULL) {\r
    free (mOptions.PkgOutputFileName);\r
    mOptions.PkgOutputFileName           = NULL;\r
  }\r
  if (mOptions.COutputFileName != NULL) {\r
    free (mOptions.COutputFileName);\r
    mOptions.COutputFileName             = NULL;\r
  }\r
  if (mOptions.PreprocessorOutputFileName != NULL) {\r
    free (mOptions.PreprocessorOutputFileName);\r
    mOptions.PreprocessorOutputFileName  = NULL;\r
  }\r
  if (mOptions.RecordListFile != NULL) {\r
    free (mOptions.RecordListFile);\r
    mOptions.RecordListFile              = NULL;\r
  }\r
  if (mOptions.IncludePaths != NULL) {\r
    delete mOptions.IncludePaths;\r
    mOptions.IncludePaths                = NULL;\r
  }\r
  if (mOptions.CPreprocessorOptions != NULL) {\r
    delete mOptions.CPreprocessorOptions;\r
    mOptions.CPreprocessorOptions        = NULL;\r
  }\r
}\r
\r
VOID\r
CVfrCompiler::AppendIncludePath (\r
  IN CHAR8      *PathStr\r
  )\r
{\r
  UINT32  Len           = 0;\r
  CHAR8   *IncludePaths = NULL;\r
\r
  Len = strlen (" -I ") + strlen (PathStr) + 1;\r
  if (mOptions.IncludePaths != NULL) {\r
    Len += strlen (mOptions.IncludePaths);\r
  }\r
  IncludePaths = new CHAR8[Len];\r
  if (IncludePaths == NULL) {\r
    DebugError (NULL, 0, 4001, "Resource: memory can't be allocated", NULL);\r
    return;\r
  }\r
  IncludePaths[0] = '\0';\r
  if (mOptions.IncludePaths != NULL) {\r
    strcat (IncludePaths, mOptions.IncludePaths);\r
  }\r
  strcat (IncludePaths, " -I ");\r
  strcat (IncludePaths, PathStr);\r
  if (mOptions.IncludePaths != NULL) {\r
    delete[] mOptions.IncludePaths;\r
  }\r
  mOptions.IncludePaths = IncludePaths;\r
}\r
\r
VOID\r
CVfrCompiler::AppendCPreprocessorOptions (\r
  IN CHAR8      *Options\r
  )\r
{\r
  UINT32  Len           = 0;\r
  CHAR8   *Opt          = NULL;\r
\r
  Len = strlen (Options) + strlen (" ") + 1;\r
  if (mOptions.CPreprocessorOptions != NULL) {\r
    Len += strlen (mOptions.CPreprocessorOptions);\r
  }\r
  Opt = new CHAR8[Len];\r
  if (Opt == NULL) {\r
    DebugError (NULL, 0, 4001, "Resource: memory can't be allocated", NULL);\r
    return;\r
  }\r
  Opt[0] = 0;\r
  if (mOptions.CPreprocessorOptions != NULL) {\r
    strcat (Opt, mOptions.CPreprocessorOptions);\r
  }\r
  strcat (Opt, " ");\r
  strcat (Opt, Options);\r
  if (mOptions.CPreprocessorOptions != NULL) {\r
    delete[] mOptions.CPreprocessorOptions;\r
  }\r
  mOptions.CPreprocessorOptions = Opt;\r
}\r
\r
INT8\r
CVfrCompiler::SetBaseFileName (\r
  VOID\r
  )\r
{\r
  CHAR8         *pFileName, *pPath, *pExt;\r
\r
  if (mOptions.VfrFileName == NULL) {\r
    return -1;\r
  }\r
\r
  pFileName = mOptions.VfrFileName;\r
  while (\r
    ((pPath = strchr (pFileName, '\\')) != NULL) ||\r
    ((pPath = strchr (pFileName, '/')) != NULL)\r
    )\r
  {\r
    pFileName = pPath + 1;\r
  }\r
\r
  if (pFileName == NULL) {\r
    return -1;\r
  }\r
\r
  if ((pExt = strchr (pFileName, '.')) == NULL) {\r
    return -1;\r
  }\r
\r
  *pExt = '\0';\r
\r
  mOptions.VfrBaseFileName = (CHAR8 *) malloc (strlen (pFileName) + 1);\r
  if (mOptions.VfrBaseFileName == NULL) {\r
    *pExt = '.';\r
    return -1;\r
  }\r
\r
  strcpy (mOptions.VfrBaseFileName, pFileName);\r
  *pExt = '.';\r
\r
  return 0;\r
}\r
\r
INT8\r
CVfrCompiler::SetPkgOutputFileName (\r
  VOID\r
  )\r
{\r
  INTN Length;\r
\r
  if (mOptions.VfrBaseFileName == NULL) {\r
    return -1;\r
  }\r
\r
  Length = strlen (mOptions.OutputDirectory) +\r
           strlen (mOptions.VfrBaseFileName) +\r
           strlen (VFR_PACKAGE_FILENAME_EXTENSION) +\r
           1;\r
\r
  mOptions.PkgOutputFileName = (CHAR8 *) malloc (Length);\r
  if (mOptions.PkgOutputFileName == NULL) {\r
    return -1;\r
  }\r
\r
  strcpy (mOptions.PkgOutputFileName, mOptions.OutputDirectory);\r
  strcat (mOptions.PkgOutputFileName, mOptions.VfrBaseFileName);\r
  strcat (mOptions.PkgOutputFileName, VFR_PACKAGE_FILENAME_EXTENSION);\r
\r
  return 0;\r
}\r
\r
INT8\r
CVfrCompiler::SetCOutputFileName (\r
  VOID\r
  )\r
{\r
  INTN Length;\r
\r
  if (mOptions.VfrBaseFileName == NULL) {\r
    return -1;\r
  }\r
\r
  Length = strlen (mOptions.OutputDirectory) +\r
           strlen (mOptions.VfrBaseFileName) +\r
           strlen (".c") +\r
           1;\r
\r
  mOptions.COutputFileName = (CHAR8 *) malloc (Length);\r
  if (mOptions.COutputFileName == NULL) {\r
    return -1;\r
  }\r
\r
  strcpy (mOptions.COutputFileName, mOptions.OutputDirectory);\r
  strcat (mOptions.COutputFileName, mOptions.VfrBaseFileName);\r
  strcat (mOptions.COutputFileName, ".c");\r
\r
  return 0;\r
}\r
\r
INT8\r
CVfrCompiler::SetPreprocessorOutputFileName (\r
  VOID\r
  )\r
{\r
  INTN Length;\r
\r
  if (mOptions.VfrBaseFileName == NULL) {\r
    return -1;\r
  }\r
\r
  Length = strlen (mOptions.OutputDirectory) +\r
           strlen (mOptions.VfrBaseFileName) +\r
           strlen (VFR_PREPROCESS_FILENAME_EXTENSION) +\r
           1;\r
\r
  mOptions.PreprocessorOutputFileName = (CHAR8 *) malloc (Length);\r
  if (mOptions.PreprocessorOutputFileName == NULL) {\r
    return -1;\r
  }\r
\r
  strcpy (mOptions.PreprocessorOutputFileName, mOptions.OutputDirectory);\r
  strcat (mOptions.PreprocessorOutputFileName, mOptions.VfrBaseFileName);\r
  strcat (mOptions.PreprocessorOutputFileName, VFR_PREPROCESS_FILENAME_EXTENSION);\r
\r
  return 0;\r
}\r
\r
INT8\r
CVfrCompiler::SetRecordListFileName (\r
  VOID\r
  )\r
{\r
  INTN Length;\r
\r
  if (mOptions.VfrBaseFileName == NULL) {\r
    return -1;\r
  }\r
\r
  Length = strlen (mOptions.OutputDirectory) +\r
           strlen (mOptions.VfrBaseFileName) +\r
           strlen (VFR_RECORDLIST_FILENAME_EXTENSION) +\r
           1;\r
\r
  mOptions.RecordListFile = (CHAR8 *) malloc (Length);\r
  if (mOptions.RecordListFile == NULL) {\r
    return -1;\r
  }\r
\r
  strcpy (mOptions.RecordListFile, mOptions.OutputDirectory);\r
  strcat (mOptions.RecordListFile, mOptions.VfrBaseFileName);\r
  strcat (mOptions.RecordListFile, VFR_RECORDLIST_FILENAME_EXTENSION);\r
\r
  return 0;\r
}\r
\r
CVfrCompiler::CVfrCompiler (\r
  IN INT32      Argc,\r
  IN CHAR8      **Argv\r
  )\r
{\r
  mPreProcessCmd = (CHAR8 *) PREPROCESSOR_COMMAND;\r
  mPreProcessOpt = (CHAR8 *) PREPROCESSOR_OPTIONS;\r
\r
  SET_RUN_STATUS (STATUS_STARTED);\r
\r
  OptionInitialization(Argc, Argv);\r
\r
  if ((IS_RUN_STATUS(STATUS_FAILED)) || (IS_RUN_STATUS(STATUS_DEAD))) {\r
    return;\r
  }\r
\r
  SET_RUN_STATUS(STATUS_INITIALIZED);\r
}\r
\r
CVfrCompiler::~CVfrCompiler (\r
  VOID\r
  )\r
{\r
  if (mOptions.VfrFileName != NULL) {\r
    free (mOptions.VfrFileName);\r
    mOptions.VfrFileName = NULL;\r
  }\r
\r
  if (mOptions.VfrBaseFileName != NULL) {\r
    free (mOptions.VfrBaseFileName);\r
    mOptions.VfrBaseFileName = NULL;\r
  }\r
\r
  if (mOptions.OutputDirectory != NULL) {\r
    free (mOptions.OutputDirectory);\r
    mOptions.OutputDirectory = NULL;\r
  }\r
\r
  if (mOptions.PkgOutputFileName != NULL) {\r
    free (mOptions.PkgOutputFileName);\r
    mOptions.PkgOutputFileName = NULL;\r
  }\r
\r
  if (mOptions.COutputFileName != NULL) {\r
    free (mOptions.COutputFileName);\r
    mOptions.COutputFileName = NULL;\r
  }\r
\r
  if (mOptions.PreprocessorOutputFileName != NULL) {\r
    free (mOptions.PreprocessorOutputFileName);\r
    mOptions.PreprocessorOutputFileName = NULL;\r
  }\r
\r
  if (mOptions.RecordListFile != NULL) {\r
    free (mOptions.RecordListFile);\r
    mOptions.RecordListFile = NULL;\r
  }\r
\r
  if (mOptions.IncludePaths != NULL) {\r
    delete[] mOptions.IncludePaths;\r
    mOptions.IncludePaths = NULL;\r
  }\r
\r
  if (mOptions.CPreprocessorOptions != NULL) {\r
    delete[] mOptions.CPreprocessorOptions;\r
    mOptions.CPreprocessorOptions = NULL;\r
  }\r
\r
  SET_RUN_STATUS(STATUS_DEAD);\r
}\r
\r
VOID\r
CVfrCompiler::Usage (\r
  VOID\r
  )\r
{\r
  UINT32 Index;\r
  CONST  CHAR8 *Help[] = {\r
    " ",\r
    "VfrCompile version " VFR_COMPILER_VERSION "Build " __BUILD_VERSION,\r
    "Copyright (c) 2004-2016 Intel Corporation. All rights reserved.",\r
    " ",\r
    "Usage: VfrCompile [options] VfrFile",\r
    " ",\r
    "Options:",\r
    "  -h, --help     prints this help",\r
    "  --version      prints version info",\r
    "  -l             create an output IFR listing file",\r
    "  -o DIR, --output-directory DIR",\r
    "                 deposit all output files to directory OutputDir",\r
    "                 default is current directory",\r
    "  -b, --create-ifr-package",\r
    "                 create an IFR HII pack file",\r
    "  -n, --no-pre-processing",\r
    "                 do not preprocessing input file",\r
    "  -s, --string-db",\r
    "                 input uni string package file",\r
    "  -g, --guid",\r
    "                 override class guid input",\r
    "                 format is xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx",\r
    "  -w  --warning-as-error",\r
    "                 treat warning as an error",\r
    "  -a  --autodefaut    generate default value for question opcode if some default is missing",\r
    "  -d  --checkdefault  check the default information in a question opcode",\r
    NULL\r
    };\r
  for (Index = 0; Help[Index] != NULL; Index++) {\r
    fprintf (stdout, "%s\n", Help[Index]);\r
  }\r
}\r
\r
VOID\r
CVfrCompiler::Version (\r
  VOID\r
  )\r
{\r
  UINT32 Index;\r
  CONST  CHAR8 *Help[] = {\r
    "VfrCompile version " VFR_COMPILER_VERSION "Build " __BUILD_VERSION,\r
    NULL\r
    };\r
  for (Index = 0; Help[Index] != NULL; Index++) {\r
    fprintf (stdout, "%s\n", Help[Index]);\r
  }\r
}\r
\r
VOID\r
CVfrCompiler::PreProcess (\r
  VOID\r
  )\r
{\r
  FILE    *pVfrFile      = NULL;\r
  UINT32  CmdLen         = 0;\r
  CHAR8   *PreProcessCmd = NULL;\r
\r
  if (!IS_RUN_STATUS(STATUS_INITIALIZED)) {\r
    goto Fail;\r
  }\r
\r
  if (mOptions.SkipCPreprocessor == TRUE) {\r
    goto Out;\r
  }\r
\r
  if ((pVfrFile = fopen (LongFilePath (mOptions.VfrFileName), "r")) == NULL) {\r
    DebugError (NULL, 0, 0001, "Error opening the input VFR file", "%s", mOptions.VfrFileName);\r
    goto Fail;\r
  }\r
  fclose (pVfrFile);\r
\r
  CmdLen = strlen (mPreProcessCmd) + strlen (mPreProcessOpt) +\r
           strlen (mOptions.VfrFileName) + strlen (mOptions.PreprocessorOutputFileName);\r
  if (mOptions.CPreprocessorOptions != NULL) {\r
    CmdLen += strlen (mOptions.CPreprocessorOptions);\r
  }\r
  if (mOptions.IncludePaths != NULL) {\r
    CmdLen += strlen (mOptions.IncludePaths);\r
  }\r
\r
  PreProcessCmd = new CHAR8[CmdLen + 10];\r
  if (PreProcessCmd == NULL) {\r
    DebugError (NULL, 0, 4001, "Resource: memory can't be allocated", NULL);\r
    goto Fail;\r
  }\r
  strcpy (PreProcessCmd, mPreProcessCmd), strcat (PreProcessCmd, " ");\r
  strcat (PreProcessCmd, mPreProcessOpt), strcat (PreProcessCmd, " ");\r
  if (mOptions.IncludePaths != NULL) {\r
    strcat (PreProcessCmd, mOptions.IncludePaths), strcat (PreProcessCmd, " ");\r
  }\r
  if (mOptions.CPreprocessorOptions != NULL) {\r
    strcat (PreProcessCmd, mOptions.CPreprocessorOptions), strcat (PreProcessCmd, " ");\r
  }\r
  strcat (PreProcessCmd, mOptions.VfrFileName), strcat (PreProcessCmd, " > ");\r
  strcat (PreProcessCmd, mOptions.PreprocessorOutputFileName);\r
\r
  if (system (PreProcessCmd) != 0) {\r
    DebugError (NULL, 0, 0003, "Error parsing file", "failed to spawn C preprocessor on VFR file %s\n", PreProcessCmd);\r
    goto Fail;\r
  }\r
\r
  delete[] PreProcessCmd;\r
\r
Out:\r
  SET_RUN_STATUS (STATUS_PREPROCESSED);\r
  return;\r
\r
Fail:\r
  if (!IS_RUN_STATUS(STATUS_DEAD)) {\r
    SET_RUN_STATUS (STATUS_FAILED);\r
  }\r
  delete[] PreProcessCmd;\r
}\r
\r
extern UINT8 VfrParserStart (IN FILE *, IN INPUT_INFO_TO_SYNTAX *);\r
\r
VOID\r
CVfrCompiler::Compile (\r
  VOID\r
  )\r
{\r
  FILE  *pInFile    = NULL;\r
  CHAR8 *InFileName = NULL;\r
  INPUT_INFO_TO_SYNTAX InputInfo;\r
\r
  if (!IS_RUN_STATUS(STATUS_PREPROCESSED)) {\r
    goto Fail;\r
  }\r
\r
  InFileName = (mOptions.SkipCPreprocessor == TRUE) ? mOptions.VfrFileName : mOptions.PreprocessorOutputFileName;\r
\r
  gCVfrErrorHandle.SetInputFile (InFileName);\r
  gCVfrErrorHandle.SetWarningAsError(mOptions.WarningAsError);\r
\r
  if ((pInFile = fopen (LongFilePath (InFileName), "r")) == NULL) {\r
    DebugError (NULL, 0, 0001, "Error opening the input file", "%s", InFileName);\r
    goto Fail;\r
  }\r
\r
  if (mOptions.HasOverrideClassGuid) {\r
    InputInfo.OverrideClassGuid = &mOptions.OverrideClassGuid;\r
  } else {\r
    InputInfo.OverrideClassGuid = NULL;\r
  }\r
\r
  if (VfrParserStart (pInFile, &InputInfo) != 0) {\r
    goto Fail;\r
  }\r
\r
  fclose (pInFile);\r
  pInFile = NULL;\r
\r
  if (gCFormPkg.HavePendingUnassigned () == TRUE) {\r
    gCFormPkg.PendingAssignPrintAll ();\r
    goto Fail;\r
  }\r
\r
  SET_RUN_STATUS (STATUS_COMPILEED);\r
  return;\r
\r
Fail:\r
  if (!IS_RUN_STATUS(STATUS_DEAD)) {\r
    DebugError (NULL, 0, 0003, "Error parsing", "compile error in file %s", InFileName);\r
    SET_RUN_STATUS (STATUS_FAILED);\r
  }\r
  if (pInFile != NULL) {\r
    fclose (pInFile);\r
  }\r
}\r
\r
VOID\r
CVfrCompiler::AdjustBin (\r
  VOID\r
  )\r
{\r
  EFI_VFR_RETURN_CODE Status;\r
\r
  if (!IS_RUN_STATUS(STATUS_COMPILEED)) {\r
    return;\r
  }\r
\r
  if (gNeedAdjustOpcode) {\r
    //\r
    // When parsing the Vfr, has created some opcodes, now need to update the record info.\r
    //\r
    gCIfrRecordInfoDB.IfrUpdateRecordInfoForDynamicOpcode (FALSE);\r
  }\r
\r
  //\r
  // Check whether need to check default info for question or auto add default for question.\r
  //\r
  if (mOptions.AutoDefault || mOptions.CheckDefault) {\r
    gCIfrRecordInfoDB.IfrCheckAddDefaultRecord (mOptions.AutoDefault, mOptions.CheckDefault);\r
  }\r
\r
  //\r
  // Check Binary Code consistent between Form and IfrRecord\r
  //\r
\r
  //\r
  // Get Package Data and IfrRecord Data\r
  //\r
  gCFormPkg.BuildPkg (gCBuffer);\r
  gCIfrRecordInfoDB.IfrRecordOutput (gRBuffer);\r
\r
  //\r
  // Compare Form and Record data\r
  //\r
  if (gCBuffer.Buffer != NULL && gRBuffer.Buffer != NULL) {\r
    UINT32 Index;\r
    if (gCBuffer.Size != gRBuffer.Size) {\r
      DebugError (NULL, 0, 0001, "Error parsing vfr file", " %s. FormBinary Size 0x%X is not same to RecordBuffer Size 0x%X", mOptions.VfrFileName, gCBuffer.Size, gRBuffer.Size);\r
    }\r
    for (Index = 0; Index < gCBuffer.Size; Index ++) {\r
      if (gCBuffer.Buffer[Index] != gRBuffer.Buffer[Index]) {\r
        break;\r
      }\r
    }\r
    if (Index != gCBuffer.Size) {\r
      DebugError (NULL, 0, 0001, "Error parsing vfr file", " %s. the 0x%X byte is different between Form and Record", mOptions.VfrFileName, Index);\r
    }\r
    DebugMsg (NULL, 0, 9, (CHAR8 *) "IFR Buffer", (CHAR8 *) "Form Buffer same to Record Buffer and Size is 0x%X", Index);\r
  } else if (gCBuffer.Buffer == NULL && gRBuffer.Buffer == NULL) {\r
    //ok\r
  } else {\r
    DebugError (NULL, 0, 0001, "Error parsing vfr file", " %s.Buffer not allocated.", mOptions.VfrFileName);\r
  }\r
\r
  return;\r
}\r
\r
VOID\r
CVfrCompiler::GenBinary (\r
  VOID\r
  )\r
{\r
  FILE                    *pFile = NULL;\r
\r
  if (!IS_RUN_STATUS(STATUS_COMPILEED)) {\r
    goto Fail;\r
  }\r
\r
  if (mOptions.CreateIfrPkgFile == TRUE) {\r
    if ((pFile = fopen (LongFilePath (mOptions.PkgOutputFileName), "wb")) == NULL) {\r
      DebugError (NULL, 0, 0001, "Error opening file", "%s", mOptions.PkgOutputFileName);\r
      goto Fail;\r
    }\r
    if (gCFormPkg.BuildPkg (pFile, &gRBuffer) != VFR_RETURN_SUCCESS) {\r
      fclose (pFile);\r
      goto Fail;\r
    }\r
    fclose (pFile);\r
  }\r
\r
  SET_RUN_STATUS (STATUS_GENBINARY);\r
\r
  return;\r
\r
Fail:\r
  if (!IS_RUN_STATUS(STATUS_DEAD)) {\r
    SET_RUN_STATUS (STATUS_FAILED);\r
  }\r
}\r
\r
static const char *gSourceFileHeader[] = {\r
  "//",\r
  "//  DO NOT EDIT -- auto-generated file",\r
  "//",\r
  "//  This file is generated by the vfrcompiler utility",\r
  "//",\r
  NULL\r
};\r
\r
VOID\r
CVfrCompiler::GenCFile (\r
  VOID\r
  )\r
{\r
  FILE                    *pFile;\r
  UINT32                  Index;\r
\r
  if (!IS_RUN_STATUS(STATUS_GENBINARY)) {\r
    goto Fail;\r
  }\r
\r
  if (!mOptions.CreateIfrPkgFile) {\r
    if ((pFile = fopen (LongFilePath (mOptions.COutputFileName), "w")) == NULL) {\r
      DebugError (NULL, 0, 0001, "Error opening output C file", "%s", mOptions.COutputFileName);\r
      goto Fail;\r
    }\r
\r
    for (Index = 0; gSourceFileHeader[Index] != NULL; Index++) {\r
      fprintf (pFile, "%s\n", gSourceFileHeader[Index]);\r
    }\r
\r
    if (gCFormPkg.GenCFile (mOptions.VfrBaseFileName, pFile, &gRBuffer) != VFR_RETURN_SUCCESS) {\r
      fclose (pFile);\r
      goto Fail;\r
    }\r
    fclose (pFile);\r
  }\r
\r
  SET_RUN_STATUS (STATUS_FINISHED);\r
  return;\r
\r
Fail:\r
  if (!IS_RUN_STATUS(STATUS_DEAD)) {\r
    SET_RUN_STATUS (STATUS_FAILED);\r
  }\r
}\r
\r
VOID\r
CVfrCompiler::GenRecordListFile (\r
  VOID\r
  )\r
{\r
  CHAR8  *InFileName = NULL;\r
  FILE   *pInFile    = NULL;\r
  FILE   *pOutFile   = NULL;\r
  CHAR8  LineBuf[MAX_VFR_LINE_LEN];\r
  UINT32 LineNo;\r
\r
  InFileName = (mOptions.SkipCPreprocessor == TRUE) ? mOptions.VfrFileName : mOptions.PreprocessorOutputFileName;\r
\r
  if (mOptions.CreateRecordListFile == TRUE && InFileName != NULL && mOptions.RecordListFile != NULL) {\r
    if ((InFileName[0] == '\0') || (mOptions.RecordListFile[0] == '\0')) {\r
      return;\r
    }\r
\r
    if ((pInFile = fopen (LongFilePath (InFileName), "r")) == NULL) {\r
      DebugError (NULL, 0, 0001, "Error opening the input VFR preprocessor output file", "%s", InFileName);\r
      return;\r
    }\r
\r
    if ((pOutFile = fopen (LongFilePath (mOptions.RecordListFile), "w")) == NULL) {\r
      DebugError (NULL, 0, 0001, "Error opening the record list file", "%s", mOptions.RecordListFile);\r
      goto Err1;\r
    }\r
\r
    fprintf (pOutFile, "//\n//  VFR compiler version " VFR_COMPILER_VERSION __BUILD_VERSION "\n//\n");\r
    LineNo = 0;\r
    while (!feof (pInFile)) {\r
      if (fgets (LineBuf, MAX_VFR_LINE_LEN, pInFile) != NULL) {\r
        fprintf (pOutFile, "%s", LineBuf);\r
        LineNo++;\r
        gCIfrRecordInfoDB.IfrRecordOutput (pOutFile, LineNo);\r
      }\r
    }\r
\r
    fprintf (pOutFile, "\n//\n// All Opcode Record List \n//\n");\r
    gCIfrRecordInfoDB.IfrRecordOutput (pOutFile, 0);\r
    gCVfrVarDataTypeDB.Dump(pOutFile);\r
\r
    fclose (pOutFile);\r
    fclose (pInFile);\r
  }\r
\r
  return;\r
\r
Err1:\r
  fclose (pInFile);\r
}\r
\r
int\r
main (\r
  IN int             Argc,\r
  IN char            **Argv\r
  )\r
{\r
  COMPILER_RUN_STATUS  Status;\r
\r
  SetPrintLevel(WARNING_LOG_LEVEL);\r
  CVfrCompiler         Compiler(Argc, Argv);\r
\r
  Compiler.PreProcess();\r
  Compiler.Compile();\r
  Compiler.AdjustBin();\r
  Compiler.GenBinary();\r
  Compiler.GenCFile();\r
  Compiler.GenRecordListFile ();\r
\r
  Status = Compiler.RunStatus ();\r
  if ((Status == STATUS_DEAD) || (Status == STATUS_FAILED)) {\r
    return 2;\r
  }\r
\r
  if (gCBuffer.Buffer != NULL) {\r
    delete[] gCBuffer.Buffer;\r
  }\r
\r
  if (gRBuffer.Buffer != NULL) {\r
    delete[] gRBuffer.Buffer;\r
  }\r
\r
  return GetUtilityStatus ();\r
}\r
\r
\r