/*++ Copyright (c) 1997 Microsoft Corporation Module Name: infparse.c Abstract: The code in this file read in an INF file, organizing it into a data structure that can be manipulated. The entry points are: OpenInfFile - Parses the INF associated with the STF file. InfParse_WriteInfToDisk - Writes the INF memory structure to disk AddInfSectionToTable - Adds a new section to the INF memory structure AddInfLineToTable - Adds a new line to a section's memory structure FindInfSectionInTable - Performs a sequential search for a specific section name FindLineInInfSection - Locates a line given a specific key DeleteLineInInfSection - Removes a line from an INF section DeleteSectionInInfFile - Removes a complete section from the INF memory structure GetInfSectionLineCount - Returns the number of lines in a section GetFirstLineInSectionStruct - Begins a line enumeration given an INF section ptr GetFirstLineInSectionStr - Begins a line enumeration given an INF section string GetNextLineInSection - Continues a line enumeration Author: Jim Schmidt (jimschm) 20-Sept-1997 Revision History: --*/ #include "pch.h" #include "migutilp.h" // // Globals to manage INF file reading // static PBYTE g_Buf1, g_Buf2; static DWORD g_Buf1Start, g_Buf2Start; static DWORD g_Buf1End, g_Buf2End; #define INF_BUFFER_SIZE 32768 #define ASSERT_VALID_INF(handle) MYASSERT((handle) != INVALID_HANDLE_VALUE && (handle) != NULL) WCHAR pGetInfFileWchar ( IN HANDLE File, IN DWORD Pos, OUT PBOOL Error ); PCWSTR pGetNextInfLine ( IN HANDLE File, IN PGROWBUFFER LineBuf, IN OUT PDWORD Pos, IN BOOL UnicodeMode ); typedef struct { HANDLE SourceInfFile; HANDLE DestInfFile; POOLHANDLE InfPool; // A pool for appended INF data PINFSECTION FirstInfSection; // The first section of the parsed INF PINFSECTION LastInfSection; // The last section of the parsed INF BOOL InfIsUnicode; } INFFILE, *PINFFILE; BOOL pReadInfIntoTable ( IN OUT PINFFILE InfFile, IN PWSTR SectionList, IN BOOL KeepComments ) /*++ Routine Description: Reads the specified file into memory, parsing the lines according to basic INF structure. Arguments: InfFile - Specifies the structure initilized with the INF file handle. Receives the complete INF structure. Return Value: TRUE if parsing was successful, or FALSE if parsing failed. --*/ { WCHAR ch; BOOL Error; GROWBUFFER LineBuf = GROWBUF_INIT; PCWSTR Text; DWORD Pos; PCWSTR Key, Data; PWSTR p, q; DWORD i; PINFSECTION Section = NULL; DWORD LineFlags; BOOL Result = FALSE; HASHTABLE ht = NULL; BOOL neededSection = FALSE; PWSTR list; Section = AddInfSectionToTableW (InfFile, L""); if (!Section) { LOG ((LOG_ERROR, "Read Inf Into Table: Could not add comment section")); return FALSE; } // // If we have a list of sections to fill, add them to a ht, for faster retrieval. // if (SectionList) { list = PoolMemDuplicateStringW (InfFile->InfPool, SectionList); MYASSERT(list); ht = HtAllocW (); if (ht) { while (list) { p = wcschr (list, L','); if (p) { *p = 0; } HtAddStringW (ht, SkipSpaceW(list)); if (p) { *p = L','; list = p + 1; } else { list = p; } } } else { LOG ((LOG_ERROR, "Read Inf Into Table: Could not allocate section hash table.")); return FALSE; } } g_Buf1Start = 0; g_Buf2Start = 0; g_Buf1End = 0; g_Buf2End = 0; g_Buf1 = (PBYTE) MemAlloc (g_hHeap, 0, INF_BUFFER_SIZE); g_Buf2 = (PBYTE) MemAlloc (g_hHeap, 0, INF_BUFFER_SIZE); MYASSERT(g_Buf1 && g_Buf2); __try { // // Determine if this file is UNICODE // ch = pGetInfFileWchar (InfFile->SourceInfFile, 0, &Error); InfFile->InfIsUnicode = (ch == 0xfeff) && !Error; // // Parse each line. // Pos = 0; for (;;) { // // Get the line // Text = pGetNextInfLine ( InfFile->SourceInfFile, &LineBuf, &Pos, InfFile->InfIsUnicode ); if (!Text) { break; } // // If a comment line or blank line, skip it // p = (PWSTR) SkipSpaceW (Text); if (!p[0] || p[0] == L';') { if (KeepComments && !AddInfLineToTableW (InfFile, Section, NULL, Text, LINEFLAG_ALL_COMMENTS)) { LOG ((LOG_ERROR, "Read Inf Into Table: Can't add line comments to table", Text)); __leave; } continue; } // // If a section line, start the new section // if (p[0] == L'[') { p++; q = wcschr (p, L']'); if (!q) { q = GetEndOfStringW (p); } else { *q = 0; } if (!ht || HtFindStringW (ht, p)) { Section = AddInfSectionToTableW (InfFile, p); neededSection = TRUE; if (!Section) { LOG ((LOG_ERROR, "Read Inf Into Table: Could not add section %s", p)); __leave; } } else { // // We must not care about this section. Make sure we don't add any lines. // neededSection = FALSE; } } // // Otherwise it must be a valid line // else { if (!Section) { DEBUGMSG ((DBG_WARNING, "InfParse_ReadInfIntoTable: Ignoring unrecognized line %s", p)); continue; } if (!neededSection) { continue; } // // Split key and line: Skip key that is surrounded by quotes, then // find the first // LineFlags = 0; q = p; Key = NULL; Data = Text; while (q[0] == L'\"') { q = wcschr (q + 1, L'\"'); if (!q) { q = p; break; } else { q++; } } i = (DWORD)wcscspn (q, L"\"="); if (q[i] == L'=') { q += i; Data = SkipSpaceW (q + 1); *q = 0; q = (PWSTR) SkipSpaceRW (Text, q); if (q && *q) { q++; *q = 0; } Key = p; if (Key[0] == L'\"') { LineFlags |= LINEFLAG_KEY_QUOTED; Key++; p = GetEndOfStringW (Key); p = (PWSTR) SkipSpaceRW (Key, p); if (p && *p) { if (p[0] != L'\"') { p++; } *p = 0; } } } if (!AddInfLineToTableW (InfFile, Section, Key, Data, LineFlags)) { LOG ((LOG_ERROR, "Read Inf Into Table: Can't add line %s to table", Text)); __leave; } } } if (Pos != GetFileSize (InfFile->SourceInfFile, NULL)) { LOG ((LOG_ERROR, "Read Inf Into Table: Could not read entire INF")); __leave; } Result = TRUE; } __finally { MemFree (g_hHeap, 0, g_Buf1); MemFree (g_hHeap, 0, g_Buf2); FreeGrowBuffer (&LineBuf); if (ht) { HtFree (ht); } } return Result; } VOID CloseInfFile ( HINF InfFile ) { PINFFILE inf = (PINFFILE) InfFile; ASSERT_VALID_INF(InfFile); PoolMemEmptyPool (inf->InfPool); PoolMemDestroyPool (inf->InfPool); MemFree (g_hHeap, 0, inf); } HINF OpenInfFileExA ( IN PCSTR InfFilePath, IN PSTR SectionList, IN BOOL KeepComments ) { PINFFILE InfFile; BOOL b = TRUE; PWSTR wSectionList = NULL; if(!InfFilePath){ DEBUGMSG((DBG_ERROR, "OpenInfFileExW: InfFilePath is NULL.")); return (HINF)INVALID_HANDLE_VALUE; } if (SectionList) { wSectionList = (PWSTR) ConvertAtoW (SectionList); } InfFile = MemAlloc (g_hHeap, HEAP_ZERO_MEMORY, sizeof (INFFILE)); InfFile->SourceInfFile = CreateFileA ( InfFilePath, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL ); if (InfFile->SourceInfFile == INVALID_HANDLE_VALUE) { b = FALSE; } else { InfFile->InfPool = PoolMemInitNamedPool ("INF File"); b = pReadInfIntoTable (InfFile, wSectionList, KeepComments); CloseHandle (InfFile->SourceInfFile); } if (wSectionList) { FreeConvertedStr (wSectionList); } if (!b) { if (InfFile->InfPool) { PoolMemDestroyPool (InfFile->InfPool); } MemFree (g_hHeap, 0, InfFile); return INVALID_HANDLE_VALUE; } InfFile->SourceInfFile = INVALID_HANDLE_VALUE; return (HINF) InfFile; } HINF OpenInfFileExW ( IN PCWSTR InfFilePath, IN PWSTR SectionList, IN BOOL KeepComments ) { PINFFILE InfFile; BOOL b = TRUE; if(!InfFilePath){ DEBUGMSG((DBG_ERROR, "OpenInfFileExW: InfFilePath is NULL.")); return (HINF)INVALID_HANDLE_VALUE; } InfFile = MemAlloc (g_hHeap, HEAP_ZERO_MEMORY, sizeof (INFFILE)); InfFile->SourceInfFile = CreateFileW ( InfFilePath, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL ); if (InfFile->SourceInfFile == INVALID_HANDLE_VALUE) { b = FALSE; } else { InfFile->InfPool = PoolMemInitNamedPool ("INF File"); b = pReadInfIntoTable (InfFile, SectionList, KeepComments); CloseHandle (InfFile->SourceInfFile); } if (!b) { if (InfFile->InfPool) { PoolMemDestroyPool (InfFile->InfPool); } MemFree (g_hHeap, 0, InfFile); return INVALID_HANDLE_VALUE; } InfFile->SourceInfFile = INVALID_HANDLE_VALUE; return (HINF) InfFile; } BOOL pWriteFileStringBufferedA ( IN PGROWBUFFER Buffer, IN HANDLE File, IN PCSTR String ) { UINT stringBytes; PBYTE byteBuf; DWORD bytesWritten; MYASSERT(Buffer); if (!String) { if (Buffer->End) { if (!WriteFile (File, Buffer->Buf, Buffer->End, &bytesWritten, NULL)) { return FALSE; } if (bytesWritten != Buffer->End) { return FALSE; } Buffer->End = 0; } return TRUE; } stringBytes = ByteCountA (String); if (!stringBytes) { return TRUE; } if (stringBytes + Buffer->End > Buffer->Size) { // // Flush buffer by calling ourselves with a NULL String // if (!pWriteFileStringBufferedA (Buffer, File, NULL)) { return FALSE; } // // If string is huge, just write it directly // if (stringBytes > Buffer->Size) { if (!WriteFile (File, String, stringBytes, &bytesWritten, NULL)) { return FALSE; } if (bytesWritten != stringBytes) { return FALSE; } return TRUE; } } // // Put string in buffer // byteBuf = GrowBuffer (Buffer, stringBytes); MYASSERT(byteBuf); CopyMemory (byteBuf, String, stringBytes); return TRUE; } BOOL pWriteFileStringBufferedW ( IN PGROWBUFFER Buffer, IN HANDLE File, IN PCWSTR String ) { UINT stringBytes; PBYTE byteBuf; DWORD bytesWritten; MYASSERT(Buffer); if (!String) { if (Buffer->End) { if (!WriteFile (File, Buffer->Buf, Buffer->End, &bytesWritten, NULL)) { return FALSE; } if (bytesWritten != Buffer->End) { return FALSE; } Buffer->End = 0; } return TRUE; } stringBytes = ByteCountW (String); if (stringBytes + Buffer->End > Buffer->Size) { // // Flush buffer by calling ourselves with a NULL String // if (!pWriteFileStringBufferedW (Buffer, File, NULL)) { return FALSE; } // // If string is huge, just write it directly // if (stringBytes > Buffer->Size) { if (!WriteFile (File, String, stringBytes, &bytesWritten, NULL)) { return FALSE; } if (bytesWritten != stringBytes) { return FALSE; } return TRUE; } } // // Put string in buffer // byteBuf = GrowBuffer (Buffer, stringBytes); MYASSERT(byteBuf); CopyMemory (byteBuf, String, stringBytes); return TRUE; } BOOL pSaveInfToFile ( IN PINFFILE InfFile ) /*++ Routine Description: pSaveInfToFile writes the INF represented by the given memory image to disk. This is done by enumerating the INF data structures in the INF. Arguments: InfFile - Specifies the table to process Return Value: TRUE if successful, FALSE if not. --*/ { PINFSECTION Section; PINFLINE Line; BYTE UnicodeHeader[] = { 0xff, 0xfe }; DWORD DontCare; BOOL b = FALSE; GROWBUFFER outputBuf = GROWBUF_INIT; GROWBUFFER conversionBuf = GROWBUF_INIT; UINT maxBytes; ASSERT_VALID_INF(InfFile); MYASSERT (InfFile->SourceInfFile == INVALID_HANDLE_VALUE); MYASSERT (InfFile->DestInfFile != INVALID_HANDLE_VALUE); // // Write the INF as we have it in memory // __try { // // Write the unicode indicator. We rely on the fact that this is first-- // the output buffer is not in use yet. // if (InfFile->InfIsUnicode) { if (!WriteFile (InfFile->DestInfFile, UnicodeHeader, sizeof (UnicodeHeader), &DontCare, NULL)) { __leave; } } // // Initialize the output buffer. It will never grow larger than what // we specify here. // GrowBuffer (&outputBuf, 16384); outputBuf.End = 0; // // Loop through all the sections // Section = InfFile->FirstInfSection; while (Section) { // // If a section name exists, write it in brackets. Section names // can be empty when comments appear at the top of the file. // if (Section->Name[0]) { if (InfFile->InfIsUnicode) { if (!pWriteFileStringBufferedW (&outputBuf, InfFile->DestInfFile, L"[") || !pWriteFileStringBufferedW (&outputBuf, InfFile->DestInfFile, Section->Name) || !pWriteFileStringBufferedW (&outputBuf, InfFile->DestInfFile, L"]\r\n") ) { __leave; } } else { maxBytes = SizeOfStringW (Section->Name); conversionBuf.End = 0; GrowBuffer (&conversionBuf, maxBytes); DirectUnicodeToDbcsN ((PSTR) conversionBuf.Buf, Section->Name, maxBytes); if (!pWriteFileStringBufferedA (&outputBuf, InfFile->DestInfFile, "[") || !pWriteFileStringBufferedA (&outputBuf, InfFile->DestInfFile, (PCSTR) conversionBuf.Buf) || !pWriteFileStringBufferedA (&outputBuf, InfFile->DestInfFile, "]\r\n") ) { __leave; } } } // // Write all the lines within the section // Line = Section->FirstLine; while (Line) { // // Write the key if it is present. Quote it if necessary. // if (Line->Key) { if (Line->LineFlags & LINEFLAG_KEY_QUOTED) { if (InfFile->InfIsUnicode) { if (!pWriteFileStringBufferedW (&outputBuf, InfFile->DestInfFile, L"\"")) { __leave; } } else { if (!pWriteFileStringBufferedA (&outputBuf, InfFile->DestInfFile, "\"")) { __leave; } } } if (InfFile->InfIsUnicode) { if (!pWriteFileStringBufferedW (&outputBuf, InfFile->DestInfFile, Line->Key)) { __leave; } } else { maxBytes = SizeOfStringW (Line->Key); conversionBuf.End = 0; GrowBuffer (&conversionBuf, maxBytes); DirectUnicodeToDbcsN ((PSTR) conversionBuf.Buf, Line->Key, maxBytes); if (!pWriteFileStringBufferedA (&outputBuf, InfFile->DestInfFile, (PCSTR) conversionBuf.Buf)) { __leave; } } if (Line->LineFlags & LINEFLAG_KEY_QUOTED) { if (InfFile->InfIsUnicode) { if (!pWriteFileStringBufferedW (&outputBuf, InfFile->DestInfFile, L"\"")) { __leave; } } else { if (!pWriteFileStringBufferedA (&outputBuf, InfFile->DestInfFile, "\"")) { __leave; } } } // // Note that when we write equals, we might add some // space. Since space is trimmed by the INF parser, we say // this is acceptable because it improves readability. // if (InfFile->InfIsUnicode) { if (!pWriteFileStringBufferedW (&outputBuf, InfFile->DestInfFile, L" = ")) { __leave; } } else { if (!pWriteFileStringBufferedA (&outputBuf, InfFile->DestInfFile, " = ")) { __leave; } } } // // Write the rest of the line // if (InfFile->InfIsUnicode) { if (!pWriteFileStringBufferedW (&outputBuf, InfFile->DestInfFile, Line->Data) || !pWriteFileStringBufferedW (&outputBuf, InfFile->DestInfFile, L"\r\n") ) { __leave; } } else { maxBytes = SizeOfStringW (Line->Data); conversionBuf.End = 0; GrowBuffer (&conversionBuf, maxBytes); DirectUnicodeToDbcsN ((PSTR) conversionBuf.Buf, Line->Data, maxBytes); if (!pWriteFileStringBufferedA (&outputBuf, InfFile->DestInfFile, (PCSTR) conversionBuf.Buf) || !pWriteFileStringBufferedA (&outputBuf, InfFile->DestInfFile, "\r\n") ) { __leave; } } Line = Line->Next; } Section = Section->Next; } // // Flush output buffer // if (InfFile->InfIsUnicode) { if (!pWriteFileStringBufferedW (&outputBuf, InfFile->DestInfFile, NULL)) { __leave; } } else { if (!pWriteFileStringBufferedA (&outputBuf, InfFile->DestInfFile, NULL)) { __leave; } } b = TRUE; } __finally { FreeGrowBuffer (&outputBuf); FreeGrowBuffer (&conversionBuf); DEBUGMSG_IF((!b, DBG_ERROR, "Write Inf To Disk: Cannot write INF")); } return b; } BOOL SaveInfFileA ( IN HINF Inf, IN PCSTR SaveToFileSpec ) { PINFFILE InfFile = (PINFFILE) Inf; BOOL b; ASSERT_VALID_INF(Inf); if(!SaveToFileSpec){ DEBUGMSG((DBG_ERROR, "SaveInfFileA: SaveToFileSpec is NULL.")); return FALSE; } if (Inf == INVALID_HANDLE_VALUE) { return FALSE; } InfFile->DestInfFile = CreateFileA ( SaveToFileSpec, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL ); if (InfFile->DestInfFile == INVALID_HANDLE_VALUE) { return FALSE; } b = pSaveInfToFile (InfFile); CloseHandle (InfFile->DestInfFile); InfFile->DestInfFile = INVALID_HANDLE_VALUE; if (!b) { DeleteFileA (SaveToFileSpec); } return b; } BOOL SaveInfFileW ( IN HINF Inf, IN PCWSTR SaveToFileSpec ) { PINFFILE InfFile = (PINFFILE) Inf; BOOL b; ASSERT_VALID_INF(Inf); if(!SaveToFileSpec){ DEBUGMSG((DBG_ERROR, "SaveInfFileW: SaveToFileSpec is NULL.")); return FALSE; } if (Inf == INVALID_HANDLE_VALUE) { return FALSE; } InfFile->DestInfFile = CreateFileW ( SaveToFileSpec, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL ); if (InfFile->DestInfFile == INVALID_HANDLE_VALUE) { return FALSE; } b = pSaveInfToFile (InfFile); CloseHandle (InfFile->DestInfFile); InfFile->DestInfFile = INVALID_HANDLE_VALUE; if (!b) { DeleteFileW (SaveToFileSpec); } return b; } PINFSECTION AddInfSectionToTableA ( IN HINF Inf, IN PCSTR SectionName ) { PINFSECTION SectionPtr; PCWSTR UnicodeSectionName; ASSERT_VALID_INF(Inf); if(!SectionName){ DEBUGMSG((DBG_ERROR, "AddInfSectionToTableA: SectionName is NULL")); return NULL; } UnicodeSectionName = ConvertAtoW (SectionName); SectionPtr = AddInfSectionToTableW (Inf, UnicodeSectionName); FreeConvertedStr (UnicodeSectionName); return SectionPtr; } PINFSECTION AddInfSectionToTableW ( IN HINF Inf, IN PCWSTR SectionName ) /*++ Routine Description: Creates a new section in our linked list structure if necessary. The return structure can be used to add lines to the section. Arguments: Inf - Specifies the INF to add the section to SectionName - Specifies the name of the new section Return Value: A pointer to the new INF section struct, or NULL if an error occurred. --*/ { PINFSECTION NewSection; PINFFILE InfFile = (PINFFILE) Inf; ASSERT_VALID_INF(Inf); // // Return early if this section already exists // MYASSERT(SectionName); NewSection = FindInfSectionInTableW (InfFile, SectionName); if (NewSection) { return NewSection; } // // Allocate a section struct // NewSection = (PINFSECTION) PoolMemGetAlignedMemory ( InfFile->InfPool, sizeof (INFSECTION) ); if (!NewSection) { return NULL; } // // Fill in members of the struct and link // ZeroMemory (NewSection, sizeof (INFSECTION)); NewSection->Name = PoolMemDuplicateStringW ( InfFile->InfPool, SectionName ); if (!NewSection->Name) { return NULL; } NewSection->Prev = InfFile->LastInfSection; if (NewSection->Prev) { NewSection->Prev->Next = NewSection; } else { InfFile->FirstInfSection = NewSection; } // // Add a blank line to LastInfSection to make things tidy // if (InfFile->LastInfSection) { AddInfLineToTableW (Inf, InfFile->LastInfSection, NULL, L"", 0); } // // Finalize linkage // InfFile->LastInfSection = NewSection; return NewSection; } PINFLINE AddInfLineToTableA ( IN HINF Inf, IN PINFSECTION SectionPtr, IN PCSTR Key, OPTIONAL IN PCSTR Data, IN DWORD LineFlags ) { PCWSTR UnicodeKey; PCWSTR UnicodeData; PINFLINE Line; ASSERT_VALID_INF(Inf); if (Key) { UnicodeKey = ConvertAtoW (Key); } else { UnicodeKey = NULL; } MYASSERT(Data); UnicodeData = ConvertAtoW (Data); Line = AddInfLineToTableW (Inf, SectionPtr, UnicodeKey, UnicodeData, LineFlags); if (Key) { FreeConvertedStr (UnicodeKey); } FreeConvertedStr (UnicodeData); return Line; } PINFLINE AddInfLineToTableW ( IN HINF Inf, IN PINFSECTION SectionPtr, IN PCWSTR Key, OPTIONAL IN PCWSTR Data, IN DWORD LineFlags ) /*++ Routine Description: Adds a line to the specified section. The caller specifies the full formatted data, and an optional key. The caller does NOT supply the equals sign between the key and data. Arguments: InfFile - Specifies the table to add the INF line to SectionName - Specifies the name of the section to add the line to Key - If specified, supplies the left-hand side of the equals line Data - Specifies the text for the line, or the right-hand side of the key = value expression. LineFlags - Specifies the flags for the INF line (see LINEFLAG_*) Return Value: TRUE if the line was added to the structure, or FALSE if not. --*/ { PINFLINE NewLine; PINFFILE InfFile = (PINFFILE) Inf; ASSERT_VALID_INF(Inf); MYASSERT(SectionPtr); // // Allocate line struct // NewLine = (PINFLINE) PoolMemGetAlignedMemory ( InfFile->InfPool, sizeof (INFLINE) ); if (!NewLine) { return NULL; } // // Fill in members of the struct and link. We insert the line at the end // of the section but before all blank space. // ZeroMemory (NewLine, sizeof (INFLINE)); if (Key) { NewLine->Key = PoolMemDuplicateStringW ( InfFile->InfPool, Key ); if (!NewLine->Key) { return NULL; } } NewLine->Data = PoolMemDuplicateStringW ( InfFile->InfPool, Data ); if (!NewLine->Data) { return NULL; } NewLine->Next = NULL; NewLine->Prev = SectionPtr->LastLine; NewLine->Section = SectionPtr; NewLine->LineFlags = LineFlags; while (NewLine->Prev) { if (NewLine->Prev->Key || *NewLine->Prev->Data) { break; } NewLine->Next = NewLine->Prev; NewLine->Prev = NewLine->Prev->Prev; } if (NewLine->Prev) { NewLine->Prev->Next = NewLine; } else { SectionPtr->FirstLine = NewLine; } if (NewLine->Next) { NewLine->Next->Prev = NewLine; } else { SectionPtr->LastLine = NewLine; } SectionPtr->LineCount++; return NewLine; } PINFSECTION FindInfSectionInTableA ( IN HINF Inf, IN PCSTR SectionName ) { PINFSECTION InfSectionPtr; PCWSTR UnicodeSectionName; ASSERT_VALID_INF(Inf); if(!SectionName){ MYASSERT(SectionName); return NULL; } UnicodeSectionName = ConvertAtoW (SectionName); InfSectionPtr = FindInfSectionInTableW (Inf, UnicodeSectionName); FreeConvertedStr (UnicodeSectionName); return InfSectionPtr; } PINFSECTION FindInfSectionInTableW ( IN HINF Inf, IN PCWSTR SectionName ) /*++ Routine Description: Scans the INF for a specific section. This routine scans the INF structures sequentially and does a case-insensitive comparison. Arguments: Inf - Specifies the INF to search SectionName - Specifies the name of the section to find Return Value: A pointer to the matching INF section struct, or NULL if the section was not found. --*/ { PINFSECTION Section; PINFFILE InfFile = (PINFFILE) Inf; ASSERT_VALID_INF(Inf); if(!SectionName){ MYASSERT(SectionName); return NULL; } Section = InfFile->FirstInfSection; while (Section) { if (StringIMatchW (Section->Name, SectionName)) { return Section; } Section = Section->Next; } return NULL; } PINFSECTION GetFirstInfSectionInTable ( IN HINF Inf ) { PINFFILE InfFile = (PINFFILE) Inf; ASSERT_VALID_INF(Inf); if (InfFile && InfFile != INVALID_HANDLE_VALUE) { return InfFile->FirstInfSection; } return NULL; } PINFSECTION GetNextInfSectionInTable ( IN PINFSECTION Section ) { if (Section) { return Section->Next; } return NULL; } PINFLINE FindLineInInfSectionA ( IN HINF Inf, IN PINFSECTION Section, IN PCSTR Key ) { PCWSTR UnicodeKey; PINFLINE LinePtr; ASSERT_VALID_INF(Inf); if(!Key){ MYASSERT(Key); return NULL; } UnicodeKey = ConvertAtoW (Key); LinePtr = FindLineInInfSectionW (Inf, Section, UnicodeKey); FreeConvertedStr (UnicodeKey); return LinePtr; } PINFLINE FindLineInInfSectionW ( IN HINF Inf, IN PINFSECTION Section, IN PCWSTR Key ) /*++ Routine Description: Scans the specified INF section for a specific key. This routine scans the INF line structures sequentially and does a case-insensitive comparison. Arguments: Inf - Specifies the INF to search Section - Specifies the section to search Key - Specifies the key to find Return Value: A pointer to the matching INF line struct, or NULL if the section was not found. --*/ { PINFLINE Line; ASSERT_VALID_INF(Inf); if(!Key){ MYASSERT(Key); return NULL; } Line = Section->FirstLine; while (Line) { if (Line->Key && StringIMatchW (Line->Key, Key)) { return Line; } Line = Line->Next; } return NULL; } PINFLINE GetFirstLineInSectionStruct ( IN PINFSECTION Section ) /*++ Routine Description: GetFirstLineInSectionStruct returns the first INFLINE pointer for the section, or NULL if no lines exist. Call GetNextLineInSection to continue enumeration. This routine does not return lines consisting only of comments. Arguments: Section - Specifies the section structure to enumerate lines frmo Return Value: A pointer to the first INFLINE struct, or NULL if no lines exist. --*/ { if (!Section || !Section->FirstLine) { return NULL; } if (Section->FirstLine->LineFlags & LINEFLAG_ALL_COMMENTS) { return GetNextLineInSection (Section->FirstLine); } return Section->FirstLine; } PINFLINE GetNextLineInSection ( IN PINFLINE PrevLine ) /*++ Routine Description: GetNextLineInSection returns the next INFLINE pointer for the section, based on the previous line, or NULL if no lines exist. This routine does not return lines with comments. Arguments: PrevLine - Specifies previous line (returned from GetFirstLineInSectionStruct or GetFirstLineInSectionStr). Return Value: This routine does not return lines consisting only of comments. --*/ { while (PrevLine) { PrevLine = PrevLine->Next; if (!PrevLine || !(PrevLine->LineFlags & LINEFLAG_ALL_COMMENTS)) { break; } } return PrevLine; } PINFLINE GetFirstLineInSectionStrA ( IN HINF Inf, IN PCSTR Section ) /*++ Routine Description: GetFirstLineInSectionStruct returns the first INFLINE pointer for the section, or NULL if no lines exist. Call GetNextLineInSection to continue enumeration. Arguments: Inf - Specifies the INF that has the section Section - Specifies the name of the section in the INF Return Value: A pointer to the first INFLINE struct, or NULL if no lines exist. --*/ { PCWSTR UnicodeSection; PINFLINE LinePtr; ASSERT_VALID_INF(Inf); if(!Section){ MYASSERT(Section); return NULL; } UnicodeSection = ConvertAtoW (Section); LinePtr = GetFirstLineInSectionStrW (Inf, UnicodeSection); FreeConvertedStr (UnicodeSection); return LinePtr; } PINFLINE GetFirstLineInSectionStrW ( IN HINF Inf, IN PCWSTR Section ) /*++ Routine Description: GetFirstLineInSectionStruct returns the first INFLINE pointer for the section, or NULL if no lines exist. Call GetNextLineInSection to continue enumeration. Arguments: Inf - Specifies the INF that has the section Section - Specifies the name of the section in the INF Return Value: A pointer to the first INFLINE struct, or NULL if no lines exist. --*/ { PINFSECTION SectionPtr; PINFFILE Table = (PINFFILE) Inf; ASSERT_VALID_INF(Inf); if(!Section){ MYASSERT(Section); return NULL; } SectionPtr = FindInfSectionInTableW (Table, Section); if (!SectionPtr) { return NULL; } return GetFirstLineInSectionStruct (SectionPtr); } INT pGetInfFileByte ( IN HANDLE File, IN DWORD Pos ) /*++ Routine Description: Returns the byte at the specified position, or -1 if the file could not be read at that position. Two buffers are used to allow fast relative access. Memory-mapped files were NOT used because problems were introduced when the swap file started filling up during GUI mode. Arguments: File - Specifies the file to read Pos - Specifies the 32-bit file offset to read (zero-based, in bytes) Return Value: The byte at the specified position, or -1 if an error was encountered. (Errors are usually caused by reading past the end of the file.) --*/ { DWORD Read; PBYTE BufSwap; MYASSERT(File != INVALID_HANDLE_VALUE); // // If we read the buffer previously, then return data in our buffer // if (Pos >= g_Buf1Start && Pos < g_Buf1End) { return g_Buf1[Pos - g_Buf1Start]; } if (Pos >= g_Buf2Start && Pos < g_Buf2End) { return g_Buf2[Pos - g_Buf2Start]; } // // Buffer not available; move buffer 2 to buffer 1, then read buffer 2 // g_Buf1Start = g_Buf2Start; g_Buf1End = g_Buf2End; BufSwap = g_Buf1; g_Buf1 = g_Buf2; g_Buf2 = BufSwap; g_Buf2Start = Pos - (Pos % 256); SetFilePointer (File, (LONG)g_Buf2Start, NULL, FILE_BEGIN); if (!ReadFile (File, g_Buf2, INF_BUFFER_SIZE, &Read, NULL)) { g_Buf2End = g_Buf2Start; return -1; } g_Buf2End = g_Buf2Start + Read; if (Pos >= g_Buf2Start && Pos < g_Buf2End) { return g_Buf2[Pos - g_Buf2Start]; } return -1; } WCHAR pGetInfFileWchar ( IN HANDLE File, IN DWORD Pos, OUT PBOOL Error ) /*++ Routine Description: Returns the WCHAR at the specified position, or 0 if the file could not be read at that position. Two buffers are used to allow fast relative access. Memory-mapped files were NOT used because problems were introduced when the swap file started filling up during GUI mode. Arguments: File - Specifies the file to read Pos - Specifies the 32-bit file offset to read (zero-based, in bytes) Error - Receives TRUE if an error was encountered, or FALSE if an error was not encountered. Return Value: The WCHAR at the specified position, or 0 if an error was encountered. (Errors are usually caused by reading past the end of the file.) If an error was encountered, the Error variable is also set to TRUE. --*/ { INT c; WCHAR ch; MYASSERT(File != INVALID_HANDLE_VALUE); c = pGetInfFileByte (File, Pos); if (c == -1 || c == 26) { if(Error){ *Error = TRUE; } return (WORD) c; } ch = (WORD) c; c = pGetInfFileByte (File, Pos + 1); if (c == -1 || c == 26) { if(Error){ *Error = TRUE; } return 0; } // pGetInfFileByte return a byte value or -1. // Since we checked for -1 the next cast is valid. ch += (WORD)(c * 256); if(Error){ *Error = FALSE; } return ch; } PCSTR pGetInfLineA ( IN HANDLE File, IN DWORD StartPos, OUT PDWORD EndPosPtr, OPTIONAL IN OUT PGROWBUFFER LineBuf ) /*++ Routine Description: Returns a DBCS string supplying the line. This string can be any length and is nul-terminated. It does not include the \r or \n characters. If supplied, the EndPosPtr is updated to point to the start of the next line. Arguments: File - Specifies the file to read StartPos - Specifies the 32-bit file offset to read (zero-based, in bytes) EndPosPtr - If specified, receives the 32-bit file offset of the next line, or equal to the file size for the last line. LineBuf - Specifies a reused GROWBUFFER that the caller initializes and pGetInfLineA uses for line allocation. The caller is responsible for cleanup. Return Value: A pointer to the DBCS string supplying the full line (with the \r, \n or \r\n sequence stripped), or NULL if an error occurs. --*/ { DWORD EndPos; INT c; PBYTE Data; DWORD Pos; DWORD ByteLen = 0; MYASSERT(File != INVALID_HANDLE_VALUE); MYASSERT(LineBuf); EndPos = StartPos; for (;;) { c = pGetInfFileByte (File, EndPos); if (c == -1 || c == 26) { break; } if (IsDBCSLeadByte ((BYTE) c)) { EndPos++; c = pGetInfFileByte (File, EndPos); if (c == -1 || c == 26) { break; } ByteLen++; } else { if (c == '\r' || c == '\n') { EndPos++; if (c == '\r') { c = pGetInfFileByte (File, EndPos); if (c == '\n') { EndPos++; } } break; } } EndPos++; ByteLen++; } // // NOTE: If you make a change here, make one below in W version // // Ctrl+Z ends the file if (c == 26) { EndPos = GetFileSize (File, NULL); } // Allocate buffer, caller frees LineBuf->End = 0; Data = GrowBuffer (LineBuf, ByteLen + 2); if (!Data) { return NULL; } // We've been successful -- copy end pos to caller's variable if (EndPosPtr) { *EndPosPtr = EndPos; } // End of file condition: zero-length, but not a blank line if (!ByteLen && c != '\r' && c != '\n') { return NULL; } // Copy line to buffer for (Pos = 0 ; Pos < ByteLen ; Pos++) { Data[Pos] = (BYTE) pGetInfFileByte (File, StartPos); StartPos++; } Data[Pos] = 0; Data[Pos + 1] = 0; return (PCSTR) Data; } PCWSTR pGetInfLineW ( IN HANDLE File, IN DWORD StartPos, OUT PDWORD EndPosPtr, OPTIONAL IN OUT PGROWBUFFER LineBuf ) /*++ Routine Description: Returns a UNICODE string supplying the line. This string can be any length and is nul-terminated. It does not include the \r or \n characters. If supplied, the EndPosPtr is updated to point to the start of the next line. Arguments: File - Specifies the file to read StartPos - Specifies the 32-bit file offset to read (zero-based, in bytes) EndPosPtr - If specified, receives the 32-bit file offset of the next line, or equal to the file size for the last line. LineBuf - Specifies a reused GROWBUFFER that the caller initializes and pGetInfLineA uses for line allocation. The caller is responsible for cleanup. Return Value: A pointer to the UNICODE string supplying the full line (with the \r, \n or \r\n sequence stripped), or NULL if an error occurs. --*/ { DWORD EndPos; PBYTE Data; DWORD Pos; DWORD ByteLen = 0; WCHAR ch; BOOL Error; MYASSERT(File != INVALID_HANDLE_VALUE); MYASSERT(LineBuf); EndPos = StartPos; for (;;) { ch = pGetInfFileWchar (File, EndPos, &Error); if (Error) { break; } if (ch == L'\r' || ch == L'\n') { EndPos += 2; if (ch == L'\r') { ch = pGetInfFileWchar (File, EndPos, &Error); if (ch == '\n') { EndPos += 2; } } break; } EndPos += 2; ByteLen += 2; } // // NOTE: If you make a change here, make one above in A version // // Ctrl+Z ends the file if (ch == 26) { EndPos = GetFileSize (File, NULL); } // Allocate buffer LineBuf->End = 0; Data = GrowBuffer (LineBuf, ByteLen + 2); if (!Data) { return NULL; } // We've been successful -- copy end pos to caller's variable if (EndPosPtr) { *EndPosPtr = EndPos; } // End of file condition: zero-length, but not a blank line if (!ByteLen && ch != L'\r' && ch != L'\n') { return NULL; } // Copy to buffer for (Pos = 0 ; Pos < ByteLen ; Pos++) { Data[Pos] = (BYTE) pGetInfFileByte (File, StartPos); StartPos++; } Data[Pos] = 0; Data[Pos + 1] = 0; if (EndPosPtr) { *EndPosPtr = EndPos; } return (PCWSTR) Data; } PCWSTR pGetNextInfLine ( IN HANDLE File, IN PGROWBUFFER LineBuf, IN OUT PDWORD Pos, IN BOOL UnicodeMode ) /*++ Routine Description: Returns a string supplying the line. This string can be any length and is nul-terminated. It does not include the \r or \n characters. Arguments: File - Specifies the file to read LineBuf - Specifies a reused GROWBUFFER that the caller initializes and pGetInfLineA uses for line allocation. The caller is responsible for cleanup. Pos - Specifies the byte offset to the start of the line. Receives the byte offset to the next line. UnicodeMode - Specifies TRUE if the file being read is a UNICODE file, or FALSE if the file being read is a DBCS file. Return Value: A pointer to the string supplying the full line (with the \r, \n or \r\n sequence stripped), or NULL if an error occurs. --*/ { PCSTR AnsiStr = NULL; PCWSTR UnicodeStr = NULL; PCWSTR FinalStr; BOOL Converted = FALSE; MYASSERT(File != INVALID_HANDLE_VALUE); MYASSERT(LineBuf); // // Obtain the text from the file // if (UnicodeMode) { UnicodeStr = pGetInfLineW (File, *Pos, Pos, LineBuf); if (!UnicodeStr) { return NULL; } } else { AnsiStr = pGetInfLineA (File, *Pos, Pos, LineBuf); if (!AnsiStr) { return NULL; } } if (AnsiStr) { UnicodeStr = ConvertAtoW (AnsiStr); if (!UnicodeStr) { return NULL; } Converted = TRUE; } FinalStr = UnicodeStr; // // Copy converted string into line buffer // if (Converted) { LineBuf->End = 0; Converted = MultiSzAppendW (LineBuf, FinalStr); FreeConvertedStr (FinalStr); if (!Converted) { return NULL; } } return (PCWSTR) LineBuf->Buf; } BOOL DeleteLineInInfSection ( IN HINF Inf, IN PINFLINE InfLine ) /*++ Routine Description: DeleteLineInInfSection removes the specified InfLine from its section, cleaning up memory used by the line. Arguments: Inf - Specifies the INF to modify InfLine - Specifies the line to delete Return Value: TRUE if the line was deleted successfully, or FALSE if an error occurred. --*/ { PINFFILE InfFile = (PINFFILE) Inf; ASSERT_VALID_INF(Inf); MYASSERT(InfLine); if (InfLine->Prev) { InfLine->Prev->Next = InfLine->Next; } else { InfLine->Section->FirstLine = InfLine->Next; } if (InfLine->Next) { InfLine->Next->Prev = InfLine->Prev; } else { InfLine->Section->LastLine = InfLine->Prev; } if (InfLine->Key) { PoolMemReleaseMemory (InfFile->InfPool, (PVOID) InfLine->Key); } if (InfLine->Data) { PoolMemReleaseMemory (InfFile->InfPool, (PVOID) InfLine->Data); } InfLine->Section->LineCount--; PoolMemReleaseMemory (InfFile->InfPool, (PVOID) InfLine); return TRUE; } BOOL DeleteSectionInInfFile ( IN HINF Inf, IN PINFSECTION Section ) /*++ Routine Description: DeleteSectionInInfFile removes the specified section from the INF data structure, removing all lines cleaning up memory used by the section. Arguments: InfFile - Specifies the table owning the INF line Section - Specifies the section to delete Return Value: TRUE if the section was deleted successfully, or FALSE if an error occurred. --*/ { PINFLINE InfLine; PINFLINE DelInfLine; PINFFILE InfFile = (PINFFILE) Inf; ASSERT_VALID_INF(Inf); MYASSERT(Section); InfLine = Section->FirstLine; while (InfLine) { DelInfLine = InfLine; InfLine = InfLine->Next; if (!DeleteLineInInfSection (InfFile, DelInfLine)) { return FALSE; } } if (Section->Prev) { Section->Prev->Next = Section->Next; } else { InfFile->FirstInfSection = Section->Next; } if (Section->Next) { Section->Next->Prev = Section->Prev; } else { InfFile->LastInfSection = Section->Prev; } PoolMemReleaseMemory (InfFile->InfPool, (PVOID) Section->Name); PoolMemReleaseMemory (InfFile->InfPool, (PVOID) Section); return TRUE; } UINT GetInfSectionLineCount ( IN PINFSECTION Section ) /*++ Routine Description: GetInfSectionLineCount returns the number of lines in the specified INF section. Arguments: Section - Specifies the section to query Return Value: The number of lines, or zero if the section has no lines. --*/ { MYASSERT(Section); return Section->LineCount; }