mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
650 lines
13 KiB
650 lines
13 KiB
/*++
|
|
|
|
© 1998 Seagate Software, Inc. All rights reserved
|
|
|
|
Module Name:
|
|
|
|
RmsDvice.cpp
|
|
|
|
Abstract:
|
|
|
|
Implementation of CRmsDevice
|
|
|
|
Author:
|
|
|
|
Brian Dodd [brian] 15-Nov-1996
|
|
|
|
Revision History:
|
|
|
|
--*/
|
|
|
|
#include "stdafx.h"
|
|
|
|
#include "RmsDvice.h"
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// IRmsDevice implementation
|
|
//
|
|
|
|
|
|
CRmsDevice::CRmsDevice(
|
|
void
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
CRmsDevice constructor
|
|
|
|
Arguments:
|
|
|
|
None
|
|
|
|
Return Value:
|
|
|
|
None
|
|
|
|
--*/
|
|
{
|
|
|
|
// Initialize values
|
|
m_deviceName = RMS_UNDEFINED_STRING;
|
|
|
|
m_deviceType = RmsDeviceUnknown;
|
|
|
|
m_sizeofDeviceInfo = 0;
|
|
|
|
memset(m_deviceInfo, 0, MaxInfo);
|
|
|
|
m_port = 0xff;
|
|
|
|
m_bus = 0xff;
|
|
|
|
m_targetId = 0xff;
|
|
|
|
m_lun = 0xff;
|
|
|
|
}
|
|
|
|
|
|
HRESULT
|
|
CRmsDevice::CompareTo(
|
|
IN IUnknown *pCollectable,
|
|
OUT SHORT *pResult
|
|
)
|
|
/*++
|
|
|
|
Implements:
|
|
|
|
CRmsDevice::CompareTo
|
|
|
|
--*/
|
|
{
|
|
HRESULT hr = E_FAIL;
|
|
SHORT result = 1;
|
|
|
|
WsbTraceIn( OLESTR("CRmsDevice::CompareTo"), OLESTR("") );
|
|
|
|
try {
|
|
|
|
// Validate arguments - Okay if pResult is NULL
|
|
WsbAssertPointer( pCollectable );
|
|
|
|
CComQIPtr<IRmsDevice, &IID_IRmsDevice> pDevice = pCollectable;
|
|
WsbAssertPointer( pDevice );
|
|
|
|
CComQIPtr<IRmsComObject, &IID_IRmsComObject> pObject = pCollectable;
|
|
WsbAssertPointer( pObject );
|
|
|
|
switch ( m_findBy ) {
|
|
|
|
case RmsFindByDeviceInfo:
|
|
{
|
|
//
|
|
// What we really want to do here is compare by
|
|
// a unique device identifier like a serial number.
|
|
//
|
|
// However, since we don't have a serial number for
|
|
// all devices, we'll compare using a best effort
|
|
// strategy using for following criteria:
|
|
//
|
|
// 1) DeviceName, this is unique for fixed drives and
|
|
// floppy drives, and SCSI devices, but may not
|
|
// survive accross reboot, and may not be the same
|
|
// for a device if it's SCSI address is changed.
|
|
//
|
|
// NOTE: We can't tell if the address was changed
|
|
// a device.
|
|
//
|
|
// NOTE: We can't tell if the drive letter was
|
|
// changed for a fixed drive.
|
|
//
|
|
// TODO: Add serial number support.
|
|
//
|
|
|
|
CWsbBstrPtr name;
|
|
|
|
// Get the target device name
|
|
pDevice->GetDeviceName( &name );
|
|
|
|
// Compare the names
|
|
result = (SHORT)wcscmp( m_deviceName, name );
|
|
hr = ( 0 == result ) ? S_OK : S_FALSE;
|
|
|
|
}
|
|
break;
|
|
|
|
case RmsFindByDeviceAddress:
|
|
{
|
|
|
|
BYTE port, bus, targetId, lun;
|
|
|
|
// Get the target device address
|
|
pDevice->GetDeviceAddress(&port, &bus, &targetId, &lun);
|
|
|
|
if( (m_port == port) &&
|
|
(m_bus == bus) &&
|
|
(m_targetId == targetId) &&
|
|
(m_lun == lun) ) {
|
|
|
|
// Device addresses match
|
|
hr = S_OK;
|
|
result = 0;
|
|
|
|
}
|
|
else {
|
|
hr = S_FALSE;
|
|
result = 1;
|
|
}
|
|
|
|
}
|
|
break;
|
|
|
|
case RmsFindByDeviceName:
|
|
{
|
|
|
|
CWsbBstrPtr name;
|
|
|
|
// Get the target device name
|
|
pDevice->GetDeviceName( &name );
|
|
|
|
// Compare the names
|
|
result = (SHORT)wcscmp( m_deviceName, name );
|
|
hr = ( 0 == result ) ? S_OK : S_FALSE;
|
|
|
|
}
|
|
break;
|
|
|
|
case RmsFindByDeviceType:
|
|
{
|
|
|
|
RmsDevice type;
|
|
|
|
// Get the target device name
|
|
pDevice->GetDeviceType( (LONG *) &type );
|
|
|
|
if ( m_deviceType == type ) {
|
|
|
|
// Device types match
|
|
hr = S_OK;
|
|
result = 0;
|
|
|
|
}
|
|
else {
|
|
hr = S_FALSE;
|
|
result = 1;
|
|
}
|
|
|
|
}
|
|
break;
|
|
|
|
default:
|
|
|
|
//
|
|
// Since devices aren't CWsbCollectables, we should
|
|
// never come here. CRmsDrive, or CRmsChanger will
|
|
// handle the default case.
|
|
//
|
|
|
|
WsbAssertHr( E_UNEXPECTED );
|
|
break;
|
|
}
|
|
|
|
}
|
|
WsbCatch(hr);
|
|
|
|
if (0 != pResult) {
|
|
*pResult = result;
|
|
}
|
|
|
|
WsbTraceOut(OLESTR("CRmsDevice::CompareTo"), OLESTR("hr = <%ls>, result = <%ls>"), WsbHrAsString(hr), WsbPtrToShortAsString(pResult));
|
|
|
|
return(hr);
|
|
}
|
|
|
|
|
|
HRESULT
|
|
CRmsDevice::GetSizeMax(
|
|
OUT ULARGE_INTEGER* pcbSize
|
|
)
|
|
/*++
|
|
|
|
Implements:
|
|
|
|
IPersistStream::GetSizeMax
|
|
|
|
--*/
|
|
{
|
|
HRESULT hr = E_NOTIMPL;
|
|
|
|
// ULONG deviceNameLen;
|
|
|
|
WsbTraceIn(OLESTR("CRmsDevice::GetSizeMax"), OLESTR(""));
|
|
|
|
// try {
|
|
// WsbAssert(0 != pcbSize, E_POINTER);
|
|
|
|
// deviceNameLen = SysStringByteLen(m_deviceName);
|
|
|
|
// // Get max size
|
|
// pcbSize->QuadPart = WsbPersistSizeOf(LONG) + // length of m_deviceName
|
|
// deviceNameLen + // m_deviceName
|
|
// WsbPersistSizeOf(LONG) + // m_deviceType
|
|
// WsbPersistSizeOf(SHORT) + // m_sizeOfDeviceInfo
|
|
// MaxInfo + // m_deviceInfo
|
|
// WsbPersistSizeOf(BYTE) + // m_port
|
|
// WsbPersistSizeOf(BYTE) + // m_bus
|
|
// WsbPersistSizeOf(BYTE) + // m_targetId
|
|
// WsbPersistSizeOf(BYTE); // m_lun
|
|
|
|
// } WsbCatch(hr);
|
|
|
|
WsbTraceOut(OLESTR("CRmsDevice::GetSizeMax"), OLESTR("hr = <%ls>, Size = <%ls>"), WsbHrAsString(hr), WsbPtrToUliAsString(pcbSize));
|
|
|
|
return(hr);
|
|
}
|
|
|
|
|
|
HRESULT
|
|
CRmsDevice::Load(
|
|
IN IStream* pStream
|
|
)
|
|
/*++
|
|
|
|
Implements:
|
|
|
|
IPersistStream::Load
|
|
|
|
--*/
|
|
{
|
|
HRESULT hr = S_OK;
|
|
ULONG ulBytes = 0;
|
|
|
|
WsbTraceIn(OLESTR("CRmsDevice::Load"), OLESTR(""));
|
|
|
|
try {
|
|
ULONG temp;
|
|
|
|
WsbAssert(0 != pStream, E_POINTER);
|
|
|
|
WsbAffirmHr(CRmsChangerElement::Load(pStream));
|
|
|
|
// Read value
|
|
m_deviceName.Free();
|
|
WsbAffirmHr(WsbBstrFromStream(pStream, &m_deviceName));
|
|
|
|
WsbAffirmHr(WsbLoadFromStream(pStream, &temp));
|
|
m_deviceType = (RmsDevice)temp;
|
|
|
|
WsbAffirmHr(WsbLoadFromStream(pStream, &m_sizeofDeviceInfo));
|
|
|
|
WsbAffirmHr(WsbLoadFromStream(pStream, &(m_deviceInfo [0]), MaxInfo));
|
|
|
|
WsbAffirmHr(WsbLoadFromStream(pStream, &m_port));
|
|
|
|
WsbAffirmHr(WsbLoadFromStream(pStream, &m_bus));
|
|
|
|
WsbAffirmHr(WsbLoadFromStream(pStream, &m_targetId));
|
|
|
|
WsbAffirmHr(WsbLoadFromStream(pStream, &m_lun));
|
|
|
|
} WsbCatch(hr);
|
|
|
|
WsbTraceOut(OLESTR("CRmsDevice::Load"), OLESTR("hr = <%ls>"), WsbHrAsString(hr));
|
|
|
|
return(hr);
|
|
}
|
|
|
|
|
|
HRESULT
|
|
CRmsDevice::Save(
|
|
IN IStream* pStream,
|
|
IN BOOL clearDirty
|
|
)
|
|
/*++
|
|
|
|
Implements:
|
|
|
|
IPersistStream::Save
|
|
|
|
--*/
|
|
{
|
|
HRESULT hr = S_OK;
|
|
ULONG ulBytes = 0;
|
|
|
|
WsbTraceIn(OLESTR("CRmsDevice::Save"), OLESTR("clearDirty = <%ls>"), WsbBoolAsString(clearDirty));
|
|
|
|
try {
|
|
WsbAssert(0 != pStream, E_POINTER);
|
|
|
|
WsbAffirmHr(CRmsChangerElement::Save(pStream, clearDirty));
|
|
|
|
// Write value
|
|
WsbAffirmHr(WsbBstrToStream(pStream, m_deviceName));
|
|
|
|
WsbAffirmHr(WsbSaveToStream(pStream, (ULONG) m_deviceType));
|
|
|
|
WsbAffirmHr(WsbSaveToStream(pStream, m_sizeofDeviceInfo));
|
|
|
|
WsbAffirmHr(WsbSaveToStream(pStream, &(m_deviceInfo [0]), MaxInfo));
|
|
|
|
WsbAffirmHr(WsbSaveToStream(pStream, m_port));
|
|
|
|
WsbAffirmHr(WsbSaveToStream(pStream, m_bus));
|
|
|
|
WsbAffirmHr(WsbSaveToStream(pStream, m_targetId));
|
|
|
|
WsbAffirmHr(WsbSaveToStream(pStream, m_lun));
|
|
|
|
} WsbCatch(hr);
|
|
|
|
WsbTraceOut(OLESTR("CRmsDevice::Save"), OLESTR("hr = <%ls>"), WsbHrAsString(hr));
|
|
|
|
return(hr);
|
|
}
|
|
|
|
|
|
HRESULT
|
|
CRmsDevice::Test(
|
|
OUT USHORT *pPassed,
|
|
OUT USHORT *pFailed
|
|
)
|
|
/*++
|
|
|
|
Implements:
|
|
|
|
IWsbTestable::Test
|
|
|
|
--*/
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
CComPtr<IRmsMediaSet> pMediaSet1;
|
|
CComPtr<IRmsMediaSet> pMediaSet2;
|
|
|
|
CComPtr<IPersistFile> pFile1;
|
|
CComPtr<IPersistFile> pFile2;
|
|
|
|
LONG i;
|
|
|
|
LONG passfail = TRUE;
|
|
|
|
CWsbBstrPtr bstrVal1 = OLESTR("5A5A5A");
|
|
CWsbBstrPtr bstrWork1;
|
|
CWsbBstrPtr bstrWork2;
|
|
|
|
SHORT ucharLenVal1 = 10;
|
|
UCHAR ucharVal1[MaxInfo] = {1,2,3,4,5,6,7,8,9,10};
|
|
|
|
SHORT ucharLenWork1;
|
|
UCHAR ucharWork1[MaxInfo];
|
|
|
|
BYTE byteVal1 = 1;
|
|
BYTE byteVal2 = 2;
|
|
BYTE byteVal3 = 3;
|
|
BYTE byteVal4 = 4;
|
|
|
|
BYTE byteWork1;
|
|
BYTE byteWork2;
|
|
BYTE byteWork3;
|
|
BYTE byteWork4;
|
|
|
|
|
|
WsbTraceIn(OLESTR("CRmsDevice::Test"), OLESTR(""));
|
|
|
|
try {
|
|
// Get the MediaSet interface.
|
|
hr = S_OK;
|
|
try {
|
|
WsbAssertHr(((IUnknown*) (IRmsMediaSet*) this)->QueryInterface(IID_IRmsMediaSet, (void**) &pMediaSet1));
|
|
|
|
// Test SetDeviceName & GetDeviceName interface
|
|
bstrWork1 = bstrVal1;
|
|
|
|
SetDeviceName(bstrWork1);
|
|
|
|
GetDeviceName(&bstrWork2);
|
|
|
|
if (bstrWork1 == bstrWork2){
|
|
(*pPassed)++;
|
|
} else {
|
|
(*pFailed)++;
|
|
}
|
|
|
|
// Test SetDeviceInfo & GetDeviceInfo interface
|
|
SetDeviceInfo(ucharVal1, ucharLenVal1);
|
|
|
|
GetDeviceInfo(ucharWork1, &ucharLenWork1);
|
|
|
|
if (ucharLenVal1 == ucharLenWork1){
|
|
(*pPassed)++;
|
|
} else {
|
|
(*pFailed)++;
|
|
}
|
|
|
|
passfail = TRUE;
|
|
|
|
for(i = 0; i < ucharLenVal1; i++){
|
|
if(ucharVal1[i] != ucharWork1[i]){
|
|
passfail = FALSE;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (passfail == TRUE){
|
|
(*pPassed)++;
|
|
} else {
|
|
(*pFailed)++;
|
|
}
|
|
|
|
// Test SetDeviceAddress & GetDeviceAddress
|
|
SetDeviceAddress(byteVal1, byteVal2, byteVal3, byteVal4);
|
|
|
|
GetDeviceAddress(&byteWork1, &byteWork2, &byteWork3, &byteWork4);
|
|
|
|
if ((byteVal1 == byteWork1) &&
|
|
(byteVal2 == byteWork2) &&
|
|
(byteVal3 == byteWork3) &&
|
|
(byteVal4 == byteWork4)){
|
|
(*pPassed)++;
|
|
} else {
|
|
(*pFailed)++;
|
|
}
|
|
|
|
} WsbCatch(hr);
|
|
|
|
// Tally up the results
|
|
|
|
hr = S_OK;
|
|
|
|
if (*pFailed) {
|
|
hr = S_FALSE;
|
|
}
|
|
|
|
} WsbCatch(hr);
|
|
|
|
WsbTraceOut(OLESTR("CRmsDevice::Test"), OLESTR("hr = <%ls>"), WsbHrAsString(hr));
|
|
|
|
return(hr);
|
|
}
|
|
|
|
|
|
STDMETHODIMP
|
|
CRmsDevice::GetDeviceName(
|
|
BSTR *pName
|
|
)
|
|
/*++
|
|
|
|
Implements:
|
|
|
|
IRmsDevice::GetDeviceName
|
|
|
|
--*/
|
|
{
|
|
WsbAssertPointer (pName);
|
|
|
|
m_deviceName. CopyToBstr (pName);
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
STDMETHODIMP
|
|
CRmsDevice::SetDeviceName(
|
|
BSTR name
|
|
)
|
|
/*++
|
|
|
|
Implements:
|
|
|
|
IRmsDevice::SetDeviceName
|
|
|
|
--*/
|
|
{
|
|
m_deviceName = name;
|
|
// m_isDirty = TRUE;
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
STDMETHODIMP
|
|
CRmsDevice::GetDeviceType(
|
|
LONG *pType
|
|
)
|
|
/*++
|
|
|
|
Implements:
|
|
|
|
IRmsDevice::GetDeviceType
|
|
|
|
--*/
|
|
{
|
|
*pType = m_deviceType;
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
STDMETHODIMP
|
|
CRmsDevice::SetDeviceType(
|
|
LONG type
|
|
)
|
|
/*++
|
|
|
|
Implements:
|
|
|
|
IRmsDevice::SetDeviceType
|
|
|
|
--*/
|
|
{
|
|
m_deviceType = (RmsDevice)type;
|
|
// m_isDirty = TRUE;
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
STDMETHODIMP
|
|
CRmsDevice::GetDeviceInfo(
|
|
UCHAR *pId,
|
|
SHORT *pSize
|
|
)
|
|
/*++
|
|
|
|
Implements:
|
|
|
|
IRmsDevice::GetDeviceInfo
|
|
|
|
--*/
|
|
{
|
|
memmove (pId, m_deviceInfo, m_sizeofDeviceInfo);
|
|
*pSize = m_sizeofDeviceInfo;
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
STDMETHODIMP
|
|
CRmsDevice::SetDeviceInfo(
|
|
UCHAR *pId,
|
|
SHORT size
|
|
)
|
|
/*++
|
|
|
|
Implements:
|
|
|
|
IRmsDevice::SetDeviceInfo
|
|
|
|
--*/
|
|
{
|
|
memmove (m_deviceInfo, pId, size);
|
|
m_sizeofDeviceInfo = size;
|
|
// m_isDirty = TRUE;
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
STDMETHODIMP
|
|
CRmsDevice::GetDeviceAddress(
|
|
LPBYTE pPort,
|
|
LPBYTE pBus,
|
|
LPBYTE pId,
|
|
LPBYTE pLun
|
|
)
|
|
/*++
|
|
|
|
Implements:
|
|
|
|
IRmsDevice::GetDeviceAddress
|
|
|
|
--*/
|
|
{
|
|
*pPort = m_port;
|
|
*pBus = m_bus;
|
|
*pId = m_targetId;
|
|
*pLun = m_lun;
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
STDMETHODIMP
|
|
CRmsDevice::SetDeviceAddress(
|
|
BYTE port,
|
|
BYTE bus,
|
|
BYTE id,
|
|
BYTE lun
|
|
)
|
|
/*++
|
|
|
|
Implements:
|
|
|
|
IRmsDevice::SetDeviceAddress
|
|
|
|
--*/
|
|
{
|
|
m_port = port;
|
|
m_bus = bus;
|
|
m_targetId = id;
|
|
m_lun = lun;
|
|
|
|
// m_isDirty = TRUE;
|
|
return S_OK;
|
|
}
|
|
|