--- /dev/null
+/*++\r
+\r
+Copyright (c) 2008 - 2010, Intel Corporation. All rights reserved.<BR>\r
+This program and the accompanying materials \r
+are licensed and made available under the terms and conditions of the BSD License \r
+which accompanies this distribution. The full text of the license may be found at \r
+http://opensource.org/licenses/bsd-license.php \r
+ \r
+THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS, \r
+WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED. \r
+\r
+Module Name:\r
+\r
+ MultiThread.c\r
+\r
+Abstract:\r
+\r
+ This module is used to add multi-thread build support to ProcessDsc utility \r
+ to improve the build performance. \r
+\r
+--*/\r
+\r
+#include <windows.h>\r
+#include <stdio.h>\r
+#include <string.h>\r
+#include <stdlib.h>\r
+#include <direct.h>\r
+#include "Common.h"\r
+#include "MultiThread.h"\r
+\r
+BUILD_ITEM *\r
+AddBuildItem (\r
+ BUILD_ITEM **BuildList,\r
+ INT8 *BaseName,\r
+ INT8 *Processor,\r
+ INT8 *Makefile\r
+ )\r
+/*++\r
+\r
+Routine Description:\r
+ \r
+ Add a build item to a specified build list\r
+\r
+Arguments:\r
+ \r
+ BuildList - build list where the new build item will be added\r
+ BaseName - base name of the new module\r
+ Processor - processor type of the new module\r
+ Makefile - makefile name of the new module\r
+\r
+Returns:\r
+\r
+ Pointer to the newly added build item\r
+\r
+--*/\r
+{\r
+ BUILD_ITEM *NewBuildItem;\r
+ \r
+ //\r
+ // Create a new build item\r
+ //\r
+ NewBuildItem = malloc (sizeof (BUILD_ITEM));\r
+ if (NewBuildItem == NULL) {\r
+ return NULL;\r
+ }\r
+ memset (NewBuildItem, 0, sizeof (BUILD_ITEM));\r
+ NewBuildItem->BaseName = _strdup (BaseName);\r
+ NewBuildItem->Processor = _strdup (Processor);\r
+ NewBuildItem->Makefile = _strdup (Makefile);\r
+ \r
+ //\r
+ // Add the build item to the head of the build list\r
+ //\r
+ NewBuildItem->Next = *BuildList;\r
+ *BuildList = NewBuildItem;\r
+ \r
+ return NewBuildItem;\r
+}\r
+\r
+SOURCE_FILE_ITEM *\r
+AddSourceFile (\r
+ BUILD_ITEM *BuildItem, \r
+ INT8 *FileName\r
+ )\r
+/*++\r
+\r
+Routine Description:\r
+ \r
+ Add a source file for a build item\r
+\r
+Arguments:\r
+ \r
+ BuildItem - build item to add the source file\r
+ FileName - source file name to be added\r
+\r
+Returns:\r
+\r
+ Pointer to the newly added source file item\r
+\r
+--*/\r
+{\r
+ SOURCE_FILE_ITEM *NewSourceFile;\r
+ \r
+ //\r
+ // Create a new source file item\r
+ //\r
+ NewSourceFile = malloc (sizeof (SOURCE_FILE_ITEM));\r
+ if (NewSourceFile == NULL) {\r
+ return NULL;\r
+ }\r
+ memset (NewSourceFile, 0, sizeof (SOURCE_FILE_ITEM));\r
+ NewSourceFile->FileName = _strdup (FileName);\r
+ \r
+ //\r
+ // Add the source file item to the head of the source file list\r
+ //\r
+ NewSourceFile->Next = BuildItem->SourceFileList;\r
+ BuildItem->SourceFileList = NewSourceFile;\r
+ \r
+ return NewSourceFile; \r
+}\r
+\r
+DEPENDENCY_ITEM *\r
+AddDependency (\r
+ BUILD_ITEM *BuildList, \r
+ BUILD_ITEM *BuildItem, \r
+ INT8 *BaseName,\r
+ INT8 AdjustIndex\r
+ )\r
+/*++\r
+\r
+Routine Description:\r
+ \r
+ Add a build dependency for a build item in the specified build list\r
+\r
+Arguments:\r
+ \r
+ BuildList - build list where to search the dependency\r
+ BuildItem - build item to add the dependency\r
+ BaseName - dependency module base name\r
+ AdjustIndex - Adjust BuildItem->Index when non-zero\r
+\r
+Returns:\r
+\r
+ Pointer to the newly added build dependency\r
+\r
+--*/\r
+{\r
+ BUILD_ITEM *TempBuildItem;\r
+ DEPENDENCY_ITEM *NewDependency;\r
+ \r
+ //\r
+ // Search the dependency in the build list\r
+ //\r
+ TempBuildItem = BuildList;\r
+ while (TempBuildItem != NULL) {\r
+ if ((_stricmp (TempBuildItem->BaseName, BaseName) == 0) &&\r
+ (_stricmp (TempBuildItem->Processor, BuildItem->Processor) == 0) &&\r
+ (TempBuildItem != BuildItem)) {\r
+ break;\r
+ }\r
+ TempBuildItem = TempBuildItem->Next;\r
+ }\r
+ if (TempBuildItem == NULL) {\r
+ return NULL;\r
+ }\r
+ \r
+ //\r
+ // This index is used to isolate two modules with same base name and processor.\r
+ // (ProcessDsc allows duplicate base name libraries.)\r
+ //\r
+ if (AdjustIndex) {\r
+ BuildItem->Index = TempBuildItem->Index + 1;\r
+ }\r
+ \r
+ //\r
+ // Create a new build dependency item\r
+ //\r
+ NewDependency = malloc (sizeof (DEPENDENCY_ITEM));\r
+ if (NewDependency == NULL) {\r
+ return NULL;\r
+ }\r
+ memset (NewDependency, 0, sizeof (DEPENDENCY_ITEM));\r
+ NewDependency->Dependency = TempBuildItem;\r
+ \r
+ //\r
+ // Add the build dependency item to the head of the dependency list\r
+ //\r
+ NewDependency->Next = BuildItem->DependencyList;\r
+ BuildItem->DependencyList = NewDependency;\r
+ \r
+ return NewDependency; \r
+}\r
+\r
+void\r
+FreeBuildList (\r
+ BUILD_ITEM *BuildList\r
+ )\r
+/*++\r
+\r
+Routine Description:\r
+ \r
+ Free a build list\r
+\r
+Arguments:\r
+ \r
+ BuildList - build list to be freed\r
+\r
+Returns:\r
+\r
+--*/\r
+{\r
+ BUILD_ITEM *TempBuildItem;\r
+ BUILD_ITEM *FreeBuildItem;\r
+ SOURCE_FILE_ITEM *TempSourceFile;\r
+ SOURCE_FILE_ITEM *FreeSourceFile;\r
+ DEPENDENCY_ITEM *TempDependency;\r
+ DEPENDENCY_ITEM *FreeDependency;\r
+ \r
+ TempBuildItem = BuildList;\r
+ while (TempBuildItem != NULL) {\r
+ free (TempBuildItem->BaseName);\r
+ free (TempBuildItem->Processor);\r
+ free (TempBuildItem->Makefile);\r
+\r
+ //\r
+ // Free source file list\r
+ //\r
+ TempSourceFile = TempBuildItem->SourceFileList;\r
+ while (TempSourceFile != NULL) {\r
+ FreeSourceFile = TempSourceFile;\r
+ TempSourceFile = TempSourceFile->Next;\r
+ free (FreeSourceFile);\r
+ }\r
+\r
+ //\r
+ // Free dependency list\r
+ //\r
+ TempDependency = TempBuildItem->DependencyList;\r
+ while (TempDependency != NULL) {\r
+ FreeDependency = TempDependency;\r
+ TempDependency = TempDependency->Next;\r
+ free (FreeDependency);\r
+ }\r
+\r
+ FreeBuildItem = TempBuildItem;\r
+ TempBuildItem = TempBuildItem->Next;\r
+ free (FreeBuildItem);\r
+ }\r
+}\r
+\r
+COMPONENTS_ITEM *\r
+AddComponentsItem (\r
+ COMPONENTS_ITEM **ComponentsList\r
+ )\r
+/*++\r
+\r
+Routine Description:\r
+ \r
+ Add a new components item to a specified components list\r
+\r
+Arguments:\r
+ \r
+ ComponentsList - components list where the new components item will be added\r
+\r
+Returns:\r
+\r
+ Pointer to the newly added components item\r
+\r
+--*/\r
+{\r
+ COMPONENTS_ITEM *NewComponents;\r
+ COMPONENTS_ITEM *TempComponents;\r
+ \r
+ //\r
+ // Create a new components item\r
+ //\r
+ NewComponents = malloc (sizeof (COMPONENTS_ITEM));\r
+ if (NewComponents == NULL) {\r
+ return NULL;\r
+ }\r
+ memset (NewComponents, 0, sizeof (COMPONENTS_ITEM));\r
+ \r
+ //\r
+ // Add the components item to the tail of the components list\r
+ //\r
+ TempComponents = *ComponentsList;\r
+ if (TempComponents == NULL) {\r
+ *ComponentsList = NewComponents;\r
+ } else {\r
+ while (TempComponents->Next != NULL) {\r
+ TempComponents = TempComponents->Next;\r
+ }\r
+ TempComponents->Next = NewComponents;\r
+ }\r
+ \r
+ return NewComponents;\r
+}\r
+\r
+void\r
+FreeComponentsList (\r
+ COMPONENTS_ITEM *ComponentsList\r
+ )\r
+/*++\r
+\r
+Routine Description:\r
+ \r
+ Free a components list\r
+\r
+Arguments:\r
+ \r
+ ComponentsList - components list to be freed\r
+\r
+Returns:\r
+\r
+--*/\r
+{\r
+ COMPONENTS_ITEM *TempComponents;\r
+ COMPONENTS_ITEM *FreeComponents;\r
+ \r
+ TempComponents = ComponentsList;\r
+ while (TempComponents != NULL) {\r
+ FreeBuildList (TempComponents->BuildList);\r
+ FreeComponents = TempComponents;\r
+ TempComponents = TempComponents->Next;\r
+ free (FreeComponents);\r
+ }\r
+}\r
+\r
+//\r
+// Module globals for multi-thread build\r
+//\r
+static INT8 mError; // non-zero means error occurred\r
+static INT8 mDone; // non-zero means no more build items available for build\r
+static UINT32 mThreadNumber; // thread number\r
+static INT8 *mBuildDir; // build directory\r
+static INT8 mLogDir[MAX_PATH]; // build item log dir\r
+static CRITICAL_SECTION mCriticalSection; // critical section object\r
+static HANDLE mSemaphoreHandle; // semaphore for "ready for build" items in mWaitingList\r
+static HANDLE mEventHandle; // event signaled when one build item is finished\r
+static BUILD_ITEM *mPendingList; // build list for build items which are not ready for build\r
+static BUILD_ITEM *mWaitingList; // build list for build items which are ready for build\r
+static BUILD_ITEM *mBuildingList; // build list for build items which are buiding\r
+static BUILD_ITEM *mDoneList; // build list for build items which already finish the build\r
+\r
+//\r
+// Restore the BuildList (not care about the sequence of the build items)\r
+//\r
+static void\r
+RestoreBuildList (\r
+ BUILD_ITEM **BuildList\r
+ )\r
+{\r
+ BUILD_ITEM *TempBuildItem;\r
+ \r
+ if (mPendingList != NULL) {\r
+ //\r
+ // Add the mPendingList to the header of *BuildList\r
+ //\r
+ TempBuildItem = mPendingList;\r
+ while (TempBuildItem->Next != NULL) {\r
+ TempBuildItem = TempBuildItem->Next;\r
+ }\r
+ TempBuildItem->Next = *BuildList;\r
+ *BuildList = mPendingList;\r
+ }\r
+\r
+ if (mWaitingList != NULL) {\r
+ //\r
+ // Add the mWaitingList to the header of *BuildList\r
+ //\r
+ TempBuildItem = mWaitingList;\r
+ while (TempBuildItem->Next != NULL) {\r
+ TempBuildItem = TempBuildItem->Next;\r
+ }\r
+ TempBuildItem->Next = *BuildList;\r
+ *BuildList = mWaitingList;\r
+ }\r
+\r
+ if (mBuildingList != NULL) {\r
+ //\r
+ // Add the mBuildingList to the header of *BuildList\r
+ //\r
+ TempBuildItem = mBuildingList;\r
+ while (TempBuildItem->Next != NULL) {\r
+ TempBuildItem = TempBuildItem->Next;\r
+ }\r
+ TempBuildItem->Next = *BuildList;\r
+ *BuildList = mBuildingList;\r
+ }\r
+\r
+ if (mDoneList != NULL) {\r
+ //\r
+ // Add the mDoneList to the header of *BuildList\r
+ //\r
+ TempBuildItem = mDoneList;\r
+ while (TempBuildItem->Next != NULL) {\r
+ TempBuildItem = TempBuildItem->Next;\r
+ }\r
+ TempBuildItem->Next = *BuildList;\r
+ *BuildList = mDoneList;\r
+ }\r
+}\r
+\r
+//\r
+// Return non-zero when no source file build conflict\r
+//\r
+static INT8\r
+CheckSourceFile (\r
+ SOURCE_FILE_ITEM *SourceFileList\r
+ )\r
+{\r
+ BUILD_ITEM *TempBuildItem;\r
+ SOURCE_FILE_ITEM *TempSourceFile;\r
+ \r
+ while (SourceFileList != NULL) {\r
+ TempBuildItem = mBuildingList;\r
+ while (TempBuildItem != NULL) {\r
+ TempSourceFile = TempBuildItem->SourceFileList;\r
+ while (TempSourceFile != NULL) {\r
+ if (_stricmp (SourceFileList->FileName, TempSourceFile->FileName) == 0) {\r
+ return 0;\r
+ }\r
+ TempSourceFile = TempSourceFile->Next;\r
+ }\r
+ TempBuildItem = TempBuildItem->Next;\r
+ }\r
+ SourceFileList = SourceFileList->Next;\r
+ }\r
+ \r
+ return 1;\r
+}\r
+\r
+//\r
+// Return non-zero when all the dependency build items has been built\r
+//\r
+static INT8\r
+CheckDependency (\r
+ DEPENDENCY_ITEM *DependencyList\r
+ )\r
+{\r
+ while (DependencyList != NULL) {\r
+ if (!(DependencyList->Dependency->CompleteFlag)) {\r
+ return 0;\r
+ }\r
+ DependencyList = DependencyList->Next;\r
+ }\r
+ \r
+ return 1;\r
+}\r
+\r
+//\r
+// Run the build task. The system() function call will cause stdout conflict \r
+// in multi-thread envroment, so implement this through CreateProcess().\r
+//\r
+static INT8\r
+RunBuildTask (\r
+ INT8 *WorkingDir, \r
+ INT8 *LogFile, \r
+ INT8 *BuildCmd\r
+ )\r
+{\r
+ HANDLE FileHandle;\r
+ SECURITY_ATTRIBUTES SecAttr;\r
+ PROCESS_INFORMATION ProcInfo; \r
+ STARTUPINFO StartInfo;\r
+ BOOL FuncRetn;\r
+ DWORD ExitCode;\r
+ \r
+ //\r
+ // Init SecAttr\r
+ //\r
+ SecAttr.nLength = sizeof (SECURITY_ATTRIBUTES); \r
+ SecAttr.bInheritHandle = TRUE; \r
+ SecAttr.lpSecurityDescriptor = NULL;\r
+ \r
+ //\r
+ // Create the log file\r
+ //\r
+ FileHandle = CreateFile (\r
+ LogFile, // file to create\r
+ GENERIC_WRITE, // open for writing\r
+ 0, // do not share\r
+ &SecAttr, // can be inherited by child processes\r
+ CREATE_ALWAYS, // overwrite existing\r
+ FILE_ATTRIBUTE_NORMAL, // normal file\r
+ NULL // no attr. template\r
+ );\r
+\r
+ if (FileHandle == INVALID_HANDLE_VALUE) { \r
+ EnterCriticalSection (&mCriticalSection);\r
+ Error (NULL, 0, 0, NULL, "could not open file %s", LogFile);\r
+ LeaveCriticalSection (&mCriticalSection);\r
+ return 1;\r
+ }\r
+ \r
+ //\r
+ // Init ProcInfo and StartInfo\r
+ //\r
+ ZeroMemory (&ProcInfo, sizeof (PROCESS_INFORMATION));\r
+ ZeroMemory (&StartInfo, sizeof (STARTUPINFO));\r
+ StartInfo.cb = sizeof (STARTUPINFO); \r
+ StartInfo.hStdError = FileHandle;\r
+ StartInfo.hStdOutput = FileHandle;\r
+ StartInfo.hStdInput = GetStdHandle (STD_INPUT_HANDLE);\r
+ StartInfo.dwFlags = STARTF_USESTDHANDLES;\r
+\r
+ //\r
+ // Create the child process\r
+ //\r
+ FuncRetn = CreateProcess (\r
+ NULL, // no application name\r
+ BuildCmd, // command line \r
+ NULL, // process security attributes \r
+ NULL, // primary thread security attributes \r
+ TRUE, // handles are inherited \r
+ 0, // creation flags \r
+ NULL, // use parent's environment \r
+ WorkingDir, // set current directory \r
+ &StartInfo, // STARTUPINFO pointer \r
+ &ProcInfo // receives PROCESS_INFORMATION \r
+ );\r
+ \r
+ if (FuncRetn == FALSE) {\r
+ EnterCriticalSection (&mCriticalSection);\r
+ Error (NULL, 0, 0, NULL, "could not create child process");\r
+ LeaveCriticalSection (&mCriticalSection);\r
+ CloseHandle (FileHandle);\r
+ return 1;\r
+ } \r
+ \r
+ //\r
+ // Wait until child process exits\r
+ //\r
+ WaitForSingleObject (ProcInfo.hProcess, INFINITE);\r
+ GetExitCodeProcess (ProcInfo.hProcess, &ExitCode);\r
+ CloseHandle (ProcInfo.hProcess);\r
+ CloseHandle (ProcInfo.hThread);\r
+ CloseHandle (FileHandle);\r
+ \r
+ if (ExitCode != 0) {\r
+ return 1;\r
+ } else {\r
+ return 0;\r
+ }\r
+}\r
+\r
+//\r
+// Thread function\r
+//\r
+static DWORD WINAPI\r
+ThreadProc (\r
+ LPVOID lpParam\r
+ )\r
+{\r
+ UINT32 ThreadId;\r
+ BUILD_ITEM *PreviousBuildItem;\r
+ BUILD_ITEM *CurrentBuildItem;\r
+ BUILD_ITEM *NextBuildItem;\r
+ INT8 WorkingDir[MAX_PATH]; \r
+ INT8 LogFile[MAX_PATH];\r
+ INT8 BuildCmd[MAX_PATH];\r
+ \r
+ ThreadId = (UINT32)lpParam;\r
+ //\r
+ // Loop until error occurred or no more build items available for build\r
+ //\r
+ for (;;) {\r
+ WaitForSingleObject (mSemaphoreHandle, INFINITE);\r
+ if (mError || mDone) {\r
+ return 0;\r
+ }\r
+ \r
+ //\r
+ // When code runs here, there must have one build item available for this \r
+ // thread. Loop until error occurred or get one build item for build.\r
+ //\r
+ for (;;) {\r
+ EnterCriticalSection (&mCriticalSection);\r
+ PreviousBuildItem = NULL;\r
+ CurrentBuildItem = mWaitingList;\r
+ while (CurrentBuildItem != NULL) {\r
+ NextBuildItem = CurrentBuildItem->Next;\r
+ //\r
+ // CheckSourceFile() is to avoid concurrently build the same source file\r
+ // which may cause the muti-thread build failure\r
+ //\r
+ if (CheckSourceFile (CurrentBuildItem->SourceFileList)) {\r
+ //\r
+ // Move the current build item from mWaitingList\r
+ //\r
+ if (PreviousBuildItem != NULL) {\r
+ PreviousBuildItem->Next = NextBuildItem;\r
+ } else {\r
+ mWaitingList = NextBuildItem;\r
+ }\r
+ //\r
+ // Add the current build item to the head of mBuildingList\r
+ //\r
+ CurrentBuildItem->Next = mBuildingList;\r
+ mBuildingList = CurrentBuildItem;\r
+ //\r
+ // If no more build items is pending or waiting for build,\r
+ // wake up every child thread for exit.\r
+ //\r
+ if ((mPendingList == NULL) && (mWaitingList == NULL)) {\r
+ mDone = 1;\r
+ //\r
+ // Make sure to wake up every child thread for exit\r
+ // \r
+ ReleaseSemaphore (mSemaphoreHandle, mThreadNumber, NULL);\r
+ }\r
+ break;\r
+ }\r
+ PreviousBuildItem = CurrentBuildItem;\r
+ CurrentBuildItem = NextBuildItem;\r
+ }\r
+ if (CurrentBuildItem != NULL) {\r
+ //\r
+ // Display build item info\r
+ //\r
+ printf ("\t[Thread_%d] nmake -nologo -f %s all\n", ThreadId, CurrentBuildItem->Makefile);\r
+ //\r
+ // Prepare build task\r
+ //\r
+ sprintf (WorkingDir, "%s\\%s", mBuildDir, CurrentBuildItem->Processor);\r
+ sprintf (LogFile, "%s\\%s_%s_%d.txt", mLogDir, CurrentBuildItem->BaseName, \r
+ CurrentBuildItem->Processor, CurrentBuildItem->Index);\r
+ sprintf (BuildCmd, "nmake -nologo -f %s all", CurrentBuildItem->Makefile);\r
+ LeaveCriticalSection (&mCriticalSection);\r
+ break;\r
+ } else {\r
+ LeaveCriticalSection (&mCriticalSection);\r
+ //\r
+ // All the build items in mWaitingList have source file conflict with \r
+ // mBuildingList. This rarely hapeens. Need wait for the build items in\r
+ // mBuildingList to be finished by other child threads.\r
+ //\r
+ Sleep (1000);\r
+ if (mError) {\r
+ return 0;\r
+ }\r
+ }\r
+ }\r
+ \r
+ //\r
+ // Start to build the CurrentBuildItem\r
+ //\r
+ if (RunBuildTask (WorkingDir, LogFile, BuildCmd)) {\r
+ //\r
+ // Build failure\r
+ //\r
+ mError = 1;\r
+ //\r
+ // Make sure to wake up every child thread for exit\r
+ //\r
+ ReleaseSemaphore (mSemaphoreHandle, mThreadNumber, NULL);\r
+ SetEvent(mEventHandle);\r
+\r
+ return mError;\r
+ } else {\r
+ //\r
+ // Build success\r
+ //\r
+ CurrentBuildItem->CompleteFlag = 1;\r
+ \r
+ EnterCriticalSection (&mCriticalSection);\r
+ //\r
+ // Move this build item from mBuildingList\r
+ //\r
+ if (mBuildingList == CurrentBuildItem) {\r
+ mBuildingList = mBuildingList->Next;\r
+ } else {\r
+ NextBuildItem = mBuildingList;\r
+ while (NextBuildItem->Next != CurrentBuildItem) {\r
+ NextBuildItem = NextBuildItem->Next;\r
+ }\r
+ NextBuildItem->Next = CurrentBuildItem->Next;\r
+ }\r
+ //\r
+ // Add this build item to mDoneList\r
+ //\r
+ CurrentBuildItem->Next = mDoneList;\r
+ mDoneList = CurrentBuildItem;\r
+ LeaveCriticalSection (&mCriticalSection);\r
+ \r
+ SetEvent(mEventHandle);\r
+ }\r
+ }\r
+}\r
+\r
+INT8\r
+StartMultiThreadBuild (\r
+ BUILD_ITEM **BuildList,\r
+ UINT32 ThreadNumber,\r
+ INT8 *BuildDir\r
+ )\r
+/*++\r
+\r
+Routine Description:\r
+ \r
+ Start multi-thread build for a specified build list\r
+\r
+Arguments:\r
+ \r
+ BuildList - build list for multi-thread build\r
+ ThreadNumber - thread number for multi-thread build\r
+ BuildDir - build dir\r
+\r
+Returns:\r
+\r
+ 0 - Successfully finished the multi-thread build\r
+ other value - Build failure\r
+\r
+--*/\r
+{\r
+ UINT32 Index;\r
+ UINT32 Count;\r
+ BUILD_ITEM *PreviousBuildItem;\r
+ BUILD_ITEM *CurrentBuildItem;\r
+ BUILD_ITEM *NextBuildItem;\r
+ HANDLE *ThreadHandle;\r
+ INT8 Cmd[MAX_PATH];\r
+ \r
+ mError = 0;\r
+ mDone = 0;\r
+ mThreadNumber = ThreadNumber;\r
+ mBuildDir = BuildDir;\r
+ mPendingList = *BuildList;\r
+ *BuildList = NULL;\r
+ mWaitingList = NULL;\r
+ mBuildingList = NULL;\r
+ mDoneList = NULL;\r
+ \r
+ //\r
+ // Do nothing when mPendingList is empty\r
+ //\r
+ if (mPendingList == NULL) {\r
+ return 0;\r
+ }\r
+ \r
+ //\r
+ // Get build item count of mPendingList\r
+ //\r
+ Count = 0;\r
+ CurrentBuildItem = mPendingList;\r
+ while (CurrentBuildItem != NULL) {\r
+ Count++;\r
+ CurrentBuildItem = CurrentBuildItem->Next;\r
+ }\r
+ \r
+ //\r
+ // The semaphore is also used to wake up child threads for exit,\r
+ // so need to make sure "maximum count" >= "thread number".\r
+ //\r
+ if (Count < ThreadNumber) {\r
+ Count = ThreadNumber;\r
+ }\r
+ \r
+ //\r
+ // Init mSemaphoreHandle\r
+ //\r
+ mSemaphoreHandle = CreateSemaphore (\r
+ NULL, // default security attributes\r
+ 0, // initial count\r
+ Count, // maximum count\r
+ NULL // unnamed semaphore\r
+ );\r
+ if (mSemaphoreHandle == NULL) {\r
+ Error (NULL, 0, 0, NULL, "failed to create semaphore");\r
+ RestoreBuildList (BuildList);\r
+ return 1;\r
+ } \r
+\r
+ //\r
+ // Init mEventHandle\r
+ //\r
+ mEventHandle = CreateEvent( \r
+ NULL, // default security attributes\r
+ FALSE, // auto-reset event\r
+ TRUE, // initial state is signaled\r
+ NULL // object not named\r
+ ); \r
+ if (mEventHandle == NULL) { \r
+ Error (NULL, 0, 0, NULL, "failed to create event");\r
+ CloseHandle (mSemaphoreHandle);\r
+ RestoreBuildList (BuildList);\r
+ return 1;\r
+ }\r
+ \r
+ //\r
+ // Init mCriticalSection\r
+ //\r
+ InitializeCriticalSection (&mCriticalSection);\r
+ \r
+ //\r
+ // Create build item log dir\r
+ //\r
+ sprintf (mLogDir, "%s\\Log", mBuildDir);\r
+ _mkdir (mLogDir);\r
+ \r
+ //\r
+ // Create child threads for muti-thread build\r
+ //\r
+ ThreadHandle = malloc (ThreadNumber * sizeof (HANDLE));\r
+ if (ThreadHandle == NULL) {\r
+ Error (NULL, 0, 0, NULL, "failed to allocate memory");\r
+ CloseHandle (mSemaphoreHandle);\r
+ CloseHandle (mEventHandle);\r
+ RestoreBuildList (BuildList);\r
+ return 1;\r
+ }\r
+ for (Index = 0; Index < ThreadNumber; Index++) {\r
+ ThreadHandle[Index] = CreateThread (\r
+ NULL, // default security attributes\r
+ 0, // use default stack size\r
+ ThreadProc, // thread function\r
+ (LPVOID)Index, // argument to thread function: use Index as thread id\r
+ 0, // use default creation flags\r
+ NULL // thread identifier not needed\r
+ );\r
+ if (ThreadHandle[Index] == NULL) {\r
+ Error (NULL, 0, 0, NULL, "failed to create Thread_%d", Index);\r
+ mError = 1;\r
+ ThreadNumber = Index;\r
+ //\r
+ // Make sure to wake up every child thread for exit\r
+ //\r
+ ReleaseSemaphore (mSemaphoreHandle, ThreadNumber, NULL);\r
+ break;\r
+ }\r
+ }\r
+ \r
+ //\r
+ // Loop until error occurred or no more build items pending for build\r
+ //\r
+ for (;;) {\r
+ WaitForSingleObject (mEventHandle, INFINITE);\r
+ if (mError) {\r
+ break;\r
+ }\r
+ Count = 0;\r
+ \r
+ EnterCriticalSection (&mCriticalSection);\r
+ PreviousBuildItem = NULL;\r
+ CurrentBuildItem = mPendingList;\r
+ while (CurrentBuildItem != NULL) {\r
+ NextBuildItem = CurrentBuildItem->Next;\r
+ if (CheckDependency (CurrentBuildItem->DependencyList)) {\r
+ //\r
+ // Move the current build item from mPendingList\r
+ //\r
+ if (PreviousBuildItem != NULL) {\r
+ PreviousBuildItem->Next = NextBuildItem;\r
+ } else {\r
+ mPendingList = NextBuildItem;\r
+ }\r
+ //\r
+ // Add the current build item to the head of mWaitingList\r
+ //\r
+ CurrentBuildItem->Next = mWaitingList;\r
+ mWaitingList = CurrentBuildItem;\r
+ Count++;\r
+ } else {\r
+ PreviousBuildItem = CurrentBuildItem;\r
+ }\r
+ CurrentBuildItem = NextBuildItem;\r
+ }\r
+ LeaveCriticalSection (&mCriticalSection);\r
+ \r
+ ReleaseSemaphore (mSemaphoreHandle, Count, NULL);\r
+ if (mPendingList == NULL) {\r
+ break;\r
+ }\r
+ }\r
+\r
+ //\r
+ // Wait until all threads have terminated\r
+ //\r
+ WaitForMultipleObjects (ThreadNumber, ThreadHandle, TRUE, INFINITE);\r
+ \r
+ if (mError && (mBuildingList != NULL)) {\r
+ //\r
+ // Dump build failure log of the first build item which doesn't finish the build\r
+ //\r
+ printf ("\tnmake -nologo -f %s all\n", mBuildingList->Makefile);\r
+ sprintf (Cmd, "type %s\\%s_%s_%d.txt 2>NUL", mLogDir, mBuildingList->BaseName,\r
+ mBuildingList->Processor, mBuildingList->Index);\r
+ _flushall ();\r
+ if (system (Cmd)) {\r
+ Error (NULL, 0, 0, NULL, "failed to run \"%s\"", Cmd);\r
+ }\r
+ }\r
+\r
+ DeleteCriticalSection (&mCriticalSection);\r
+ for (Index = 0; Index < ThreadNumber; Index++) {\r
+ CloseHandle (ThreadHandle[Index]);\r
+ }\r
+ free (ThreadHandle);\r
+ CloseHandle (mSemaphoreHandle);\r
+ CloseHandle (mEventHandle);\r
+ RestoreBuildList (BuildList);\r
+\r
+ return mError;\r
+}\r
projects / mirror_edk2.git / blobdiff