#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()