#include "stdafx.h"

Include dependency graph for id_badblock.cpp:

Functions
static NTSTATUS __stdcall	BadBlockQueryRoutine (IN PWSTR ValueName, IN ULONG ValueType, IN PVOID ValueData, IN ULONG ValueLength, IN PVOID Context, IN PVOID EntryContext)

void NTAPI	InitBadBlocks (IN PHW_LU_EXTENSION LunExt)

void NTAPI	ForgetBadBlocks (IN PHW_LU_EXTENSION LunExt)

bool NTAPI	CheckIfBadBlock (IN PHW_LU_EXTENSION LunExt, IN ULONGLONG lba, IN ULONG count)

Variables
LIST_ENTRY	BBList

BOOLEAN	BBListInited = FALSE

Function Documentation

◆ BadBlockQueryRoutine()

static NTSTATUS __stdcall BadBlockQueryRoutine	(	IN PWSTR	ValueName,
		IN ULONG	ValueType,
		IN PVOID	ValueData,
		IN ULONG	ValueLength,
		IN PVOID	Context,
		IN PVOID	EntryContext
	)

static

Definition at line 52 of file id_badblock.cpp.

 {
     PSBadBlockListItem cur;
     PLIST_ENTRY      link;
     ULONG i;
     // The ValueType should be REG_SZ
     // The ValueData is UNICODE string of the following format:
     //  "badblocks_start_from_lba  badblocks_end_at_lba"
 
     KdPrint(( "BadBlockQueryRoutine: S/N:%S\n type %#x, len %#x\n", ValueName, ValueType, ValueLength));
 
     if(!BBListInited)
         return STATUS_SUCCESS;
 
     if((ValueType == REG_BINARY)  &&            // STRING
         ValueLength &&            // At least "0 0 0"
         !(ValueLength % sizeof(SBadBlockRange)))    // There is free space for the record
     {
         cur = NULL;
         link = BBList.Flink;
         while(link != &BBList) {
             cur = CONTAINING_RECORD( link, SBadBlockListItem, List);
             link = link->Flink;
             if(!wcscmp(cur->SerNumStr, ValueName)) {
                 KdPrint(( "already loaded\n"));
                 if(cur->LunExt) {
                     cur->LunExt->nBadBlocks   = 0;
                     cur->LunExt->arrBadBlocks = NULL;
                     cur->LunExt->bbListDescr  = NULL;
                     cur->LunExt = NULL;
                 }
                 break;
             }
         }
 
         if(!cur) {
             cur = (PSBadBlockListItem)ExAllocatePool(NonPagedPool, sizeof(SBadBlockListItem));
             if(!cur)
                 return STATUS_SUCCESS;
         } else {
             if(cur->arrBadBlocks) {
                 ExFreePool(cur->arrBadBlocks);
                 cur->arrBadBlocks = NULL;
             }
         }
         cur->arrBadBlocks = (SBadBlockRange*)ExAllocatePool(NonPagedPool, ValueLength);
         if(!cur->arrBadBlocks) {
             ExFreePool(cur);
             return STATUS_SUCCESS;
         }
         RtlCopyMemory(cur->arrBadBlocks, ValueData, ValueLength);
         wcsncpy(cur->SerNumStr, ValueName, 127);
         cur->SerNumStr[127] = 0;
         cur->nBadBlocks = ValueLength/sizeof(SBadBlockRange);
         cur->LunExt = NULL;
         InitializeListHead(&cur->List);
         InsertTailList(&BBList, &(cur->List));
         for(i=0; i<cur->nBadBlocks; i++) {
             KdPrint(( "BB: %I64x - %I64x\n", cur->arrBadBlocks[i].m_lbaStart, cur->arrBadBlocks[i].m_lbaEnd-1));
         }
     }
     return STATUS_SUCCESS;
 } // end BadBlockQueryRoutine()

Referenced by InitBadBlocks().

◆ CheckIfBadBlock()

bool NTAPI CheckIfBadBlock	(	IN PHW_LU_EXTENSION	LunExt,
		IN ULONGLONG	lba,
		IN ULONG	count
	)

Definition at line 251 of file id_badblock.cpp.

 {
     if (LunExt->nBadBlocks == 0)
         return false;
 /*
     // this is checked by caller
     if(!(AtaCommandFlags[command] & ATA_CMD_FLAG_LBAsupp)) {
         return false;
 */
     ULONG nBadBlocks = LunExt->nBadBlocks;
     SBadBlockRange*  arrBadBlocks = LunExt->arrBadBlocks;
 
     // back transform for possibly CHS'ed LBA
     lba = UniAtaCalculateLBARegsBack(LunExt, lba);
 
     for (ULONG i = 0; i < nBadBlocks; i++)
     {
         if (lba + count > arrBadBlocks->m_lbaStart  &&
                     lba < arrBadBlocks->m_lbaEnd) {
             KdPrint(( "listed BB @ %I64x\n", lba));
             return true;
         }
         arrBadBlocks++;
     }
 
     return false;
 
 } // end CheckIfBadBlock()

Referenced by AtaCommand48(), and UniataAhciSetupFIS_H2D().

◆ ForgetBadBlocks()

void NTAPI ForgetBadBlocks ( IN PHW_LU_EXTENSION LunExt )

Definition at line 237 of file id_badblock.cpp.

 {
     if(LunExt->bbListDescr) {
         LunExt->bbListDescr->LunExt = NULL;
         LunExt->nBadBlocks   = 0;
         LunExt->arrBadBlocks = NULL;
         LunExt->bbListDescr  = NULL;
     }
 } // end ForgetBadBlocks()

Referenced by AtapiStartIo__().

◆ InitBadBlocks()

void NTAPI InitBadBlocks ( IN PHW_LU_EXTENSION LunExt )

Definition at line 126 of file id_badblock.cpp.

 {
     RTL_QUERY_REGISTRY_TABLE QueryTable[2];    // Main record and zero filled end of array marker
     WCHAR DevSerial[128];
 #ifdef _DEBUG
     UCHAR cDevSerial[128];
     ULONG i;
 #endif
     ULONG Length;
     PLIST_ENTRY      link;
     PSBadBlockListItem   cur;
     // Read from the registry necessary badblock pairs and fill in arrBadBlocks array
     // HKEY_LOCAL_MACHINE\System\CurrentControlSet\Services\UniATA\Parameters\BadBlocks
 
     if(!LunExt) {
         // init
         KdPrint(( "InitBadBlocks general\n"));
         if(!BBListInited) {
             InitializeListHead(&BBList);
             BBListInited = TRUE;
         }
 
         QueryTable[0].QueryRoutine    = BadBlockQueryRoutine;
         QueryTable[0].Flags           = RTL_QUERY_REGISTRY_REQUIRED;
         QueryTable[0].Name            = NULL;   // If Name is NULL, the QueryRoutine function
                                                 //  specified for this table entry is called
                                                 //  for all values associated with the current
                                                 //  registry key.
         QueryTable[0].EntryContext    = NULL;
         QueryTable[0].DefaultType     = REG_NONE;
         QueryTable[0].DefaultData     = 0;
         QueryTable[0].DefaultLength   = 0;
 
         RtlZeroMemory(QueryTable + 1, sizeof(RTL_QUERY_REGISTRY_TABLE));    // EOF
 
         NTSTATUS status = RtlQueryRegistryValues(RTL_REGISTRY_SERVICES,
                                                 L"UniATA\\Parameters\\BadBlocks",
                                                 QueryTable, 0, 0);
 
 #ifdef _DEBUG
         KdPrint(( "InitBadBlocks returned: %#x\n", status));
 #else
         UNREFERENCED_PARAMETER(status);
 #endif
     } else {
 
         KdPrint(( "InitBadBlocks local\n"));
         Length = EncodeVendorStr(DevSerial, (PUCHAR)LunExt->IdentifyData.ModelNumber, sizeof(LunExt->IdentifyData.ModelNumber));
         DevSerial[Length] = '-';
         Length++;
         Length += EncodeVendorStr(DevSerial+Length, LunExt->IdentifyData.SerialNumber, sizeof(LunExt->IdentifyData.SerialNumber));
 
 #ifdef _DEBUG
         KdPrint(( "LunExt %#x\n", LunExt));
         for(i=0; i<Length; i++) {
             cDevSerial[i] = (UCHAR)(DevSerial[i]);
         }
         cDevSerial[i] = 0;
         KdPrint(( "S/N:%s\n", cDevSerial));
 #endif
 
         LunExt->nBadBlocks = 0;
         LunExt->arrBadBlocks = NULL;
 
         link = BBList.Flink;
         while(link != &BBList) {
             cur = CONTAINING_RECORD( link, SBadBlockListItem, List);
             link = link->Flink;
             if(cur->LunExt == LunExt) {
                 KdPrint(( "  deassociate BB list (by LunExt)\n"));
                 cur->LunExt->nBadBlocks   = 0;
                 cur->LunExt->arrBadBlocks = NULL;
                 cur->LunExt->bbListDescr  = NULL;
                 cur->LunExt = NULL;
             } else
             if(!wcscmp(cur->SerNumStr, DevSerial)) {
                 KdPrint(( "  deassociate BB list (by Serial)\n"));
                 if(cur->LunExt) {
                     cur->LunExt->nBadBlocks   = 0;
                     cur->LunExt->arrBadBlocks = NULL;
                     cur->LunExt->bbListDescr  = NULL;
                     cur->LunExt = NULL;
                 }
             }
         }
 
         if(!(LunExt->DeviceFlags & DFLAGS_ATAPI_DEVICE)) {
 
             link = BBList.Flink;
             while(link != &BBList) {
                 cur = CONTAINING_RECORD( link, SBadBlockListItem, List);
                 link = link->Flink;
                 if(!wcscmp(cur->SerNumStr, DevSerial)) {
                     KdPrint(( "found BB:List\n"));
                     cur->LunExt = LunExt;
                     LunExt->arrBadBlocks = cur->arrBadBlocks;
                     LunExt->nBadBlocks   = cur->nBadBlocks;
                     LunExt->bbListDescr  = cur;
                     return;
                 }
             }
         }
     }
     return;
 } // end InitBadBlocks()