badblock.h File Reference

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
struct	_SBadBlockRange
struct	_SBadBlockListItem

Typedefs
typedef struct _SBadBlockRange	SBadBlockRange
typedef struct _SBadBlockRange *	PSBadBlockRange
typedef struct _SBadBlockListItem	SBadBlockListItem
typedef struct _SBadBlockListItem *	PSBadBlockListItem

Functions
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)

Typedef Documentation

PSBadBlockListItem

typedef struct _SBadBlockListItem * PSBadBlockListItem

PSBadBlockRange

typedef struct _SBadBlockRange * PSBadBlockRange

SBadBlockListItem

typedef struct _SBadBlockListItem SBadBlockListItem

SBadBlockRange

typedef struct _SBadBlockRange SBadBlockRange

Function Documentation

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

Referenced by DriverEntry(), and IssueIdentify().