devs/foxbox
1
0
Fork 0
forked from len0rd/rockbox
Code Pull requests Activity

MTP:

Browse source 
* Add Win32 progress callback reporting support


git-svn-id: svn://svn.rockbox.org/rockbox/trunk@18355 a1c6a512-1295-4272-9138-f99709370657
This commit is contained in:
Maurus Cuelenaere 2008-08-27 23:07:08 +00:00
parent e320446f0e
commit 79177edfb2
6 changed files with 178 additions and 95 deletions
Download patch file Download diff file Expand all files Collapse all files

230
utils/MTP/MTP_DLL/sendfirm_win.cpp
View file

@ -11,6 +11,26 @@
#include "sac.h"
#include "scclient.h"
class CProgressHelper :
public IWMDMProgress
{
void (*m_callback)(unsigned int progress, unsigned int max);
DWORD m_max_ticks;
DWORD m_cur_ticks;
DWORD m_counter;
public:
CProgressHelper( void (*callback)(unsigned int progress, unsigned int max) );
~CProgressHelper();
STDMETHOD(Begin)( DWORD dwEstimatedTicks );
STDMETHOD(Progress)( DWORD dwTranspiredTicks );
STDMETHOD(End)();
STDMETHOD(QueryInterface) ( REFIID riid, void __RPC_FAR *__RPC_FAR *ppvObject );
STDMETHOD_(ULONG, AddRef)( void );
STDMETHOD_(ULONG, Release)( void );
};
/*
* Compilation requirements:
*
@ -21,74 +41,66 @@
*
*/
extern "C" {
__declspec(dllexport) bool send_fw(LPWSTR file, int filesize)
__declspec(dllexport) bool send_fw(LPWSTR file, int filesize, void (*callback)(unsigned int progress, unsigned int max))
{
bool return_value = false;
HRESULT hr;
IComponentAuthenticate* pICompAuth;
CSecureChannelClient *m_pSacClient = new CSecureChannelClient;
IWMDeviceManager3* m_pIdvMgr = NULL;
HRESULT hr;
IComponentAuthenticate* pICompAuth;
CSecureChannelClient *m_pSacClient = new CSecureChannelClient;
IWMDeviceManager3* m_pIdvMgr = NULL;
/* these are generic keys */
BYTE abPVK[] = {0x00};
BYTE abCert[] = {0x00};
/* these are generic keys */
BYTE abPVK[] = {0x00};
BYTE abCert[] = {0x00};
CoInitialize(NULL);
CoInitialize(NULL);
/* get an authentication interface */
hr = CoCreateInstance(CLSID_MediaDevMgr, NULL, CLSCTX_ALL ,IID_IComponentAuthenticate, (void **)&pICompAuth);
if SUCCEEDED(hr)
{
/* create a secure channel client certificate */
hr = m_pSacClient->SetCertificate(SAC_CERT_V1, (BYTE*) abCert, sizeof(abCert), (BYTE*) abPVK, sizeof(abPVK));
if SUCCEEDED(hr)
{
/* bind the authentication interface to the secure channel client */
m_pSacClient->SetInterface(pICompAuth);
/* get an authentication interface */
hr = CoCreateInstance(CLSID_MediaDevMgr, NULL, CLSCTX_ALL ,IID_IComponentAuthenticate, (void **)&pICompAuth);
if SUCCEEDED(hr)
{
/* create a secure channel client certificate */
hr = m_pSacClient->SetCertificate(SAC_CERT_V1, (BYTE*) abCert, sizeof(abCert), (BYTE*) abPVK, sizeof(abPVK));
if SUCCEEDED(hr)
{
/* bind the authentication interface to the secure channel client */
m_pSacClient->SetInterface(pICompAuth);
/* trigger communication */
hr = m_pSacClient->Authenticate(SAC_PROTOCOL_V1);
if SUCCEEDED(hr)
{
/* get main interface to media device manager */
hr = pICompAuth->QueryInterface(IID_IWMDeviceManager2, (void**)&m_pIdvMgr);
if SUCCEEDED(hr)
{
/* enumerate devices... */
IWMDMEnumDevice *pIEnumDev;
hr = m_pIdvMgr->EnumDevices2(&pIEnumDev);
if SUCCEEDED(hr)
/* trigger communication */
hr = m_pSacClient->Authenticate(SAC_PROTOCOL_V1);
if SUCCEEDED(hr)
{
/* get main interface to media device manager */
hr = pICompAuth->QueryInterface(IID_IWMDeviceManager2, (void**)&m_pIdvMgr);
if SUCCEEDED(hr)
{
/* enumerate devices... */
IWMDMEnumDevice *pIEnumDev;
hr = m_pIdvMgr->EnumDevices2(&pIEnumDev);
if SUCCEEDED(hr)
{
hr = pIEnumDev->Reset(); /* Next will now return the first device */
if SUCCEEDED(hr)
hr = pIEnumDev->Reset(); /* Next will now return the first device */
if SUCCEEDED(hr)
{
IWMDMDevice3* pIDevice;
unsigned long ulNumFetched;
hr = pIEnumDev->Next(1, (IWMDMDevice **)&pIDevice, &ulNumFetched);
while (SUCCEEDED(hr) && (hr != S_FALSE))
IWMDMDevice3* pIDevice;
unsigned long ulNumFetched;
hr = pIEnumDev->Next(1, (IWMDMDevice **)&pIDevice, &ulNumFetched);
while (SUCCEEDED(hr) && (hr != S_FALSE))
{
#if 0
/* output device name */
wchar_t pwsString[256];
hr = pIDevice->GetName(pwsString, 256);
if SUCCEEDED(hr)
wprintf(L"Found device %s\n", pwsString);
#endif
/* get storage info */
DWORD tempDW;
pIDevice->GetType(&tempDW);
if (tempDW & WMDM_DEVICE_TYPE_STORAGE)
/* get storage info */
DWORD tempDW;
pIDevice->GetType(&tempDW);
if (tempDW & WMDM_DEVICE_TYPE_STORAGE)
{
IWMDMEnumStorage *pIEnumStorage = NULL;
IWMDMStorage *pIStorage = NULL;
IWMDMStorage3 *pIFileStorage = NULL;
hr = pIDevice->EnumStorage(&pIEnumStorage);
if SUCCEEDED(hr)
IWMDMEnumStorage *pIEnumStorage = NULL;
IWMDMStorage *pIStorage = NULL;
IWMDMStorage3 *pIFileStorage = NULL;
hr = pIDevice->EnumStorage(&pIEnumStorage);
if SUCCEEDED(hr)
{
pIEnumStorage->Reset();
hr = pIEnumStorage->Next(1, (IWMDMStorage **)&pIStorage, &ulNumFetched);
while (SUCCEEDED(hr) && (hr != S_FALSE))
pIEnumStorage->Reset();
hr = pIEnumStorage->Next(1, (IWMDMStorage **)&pIStorage, &ulNumFetched);
while (SUCCEEDED(hr) && (hr != S_FALSE))
{
IWMDMStorage3 *pNewStorage;
hr = pIStorage->QueryInterface(IID_IWMDMStorage3, (void **)&pNewStorage);
@ -113,14 +125,15 @@ __declspec(dllexport) bool send_fw(LPWSTR file, int filesize)
if (SUCCEEDED(hr))
{
IWMDMStorage *pNewObject = NULL;
CProgressHelper *progress = new CProgressHelper(callback);
hr = pIWMDMStorageControl->Insert3(
WMDM_MODE_BLOCK | WMDM_CONTENT_FILE,
WMDM_MODE_BLOCK | WMDM_CONTENT_FILE | WMDM_MODE_PROGRESS,
0,
file,
NULL,
NULL,
NULL,
(callback == NULL ? NULL : (IWMDMProgress*)progress),
pIWMDMMetaData,
NULL,
(IWMDMStorage **)&pNewObject);
@ -135,27 +148,88 @@ __declspec(dllexport) bool send_fw(LPWSTR file, int filesize)
}
}
}
}
}
pIEnumStorage->Release();
}
}
}
pIEnumStorage->Release();
}
/* move to next device */
if(!return_value)
hr = pIEnumDev->Next(1, (IWMDMDevice **)&pIDevice, &ulNumFetched);
}
pIEnumDev->Release();
}
m_pIdvMgr->Release();
}
pICompAuth->Release();
}
}
}
}
/* move to next device */
if(!return_value)
hr = pIEnumDev->Next(1, (IWMDMDevice **)&pIDevice, &ulNumFetched);
}
pIEnumDev->Release();
}
m_pIdvMgr->Release();
}
pICompAuth->Release();
}
}
}
}
CoUninitialize();
CoUninitialize();
return return_value;
return return_value;
}
}
CProgressHelper::CProgressHelper( void (*callback)(unsigned int progress, unsigned int max) )
{
m_cur_ticks = 0;
m_max_ticks = 0;
m_counter = 0;
m_callback = callback;
}
CProgressHelper::~CProgressHelper()
{
}
HRESULT CProgressHelper::Begin( DWORD dwEstimatedTicks )
{
m_max_ticks = dwEstimatedTicks;
return S_OK;
}
HRESULT CProgressHelper::Progress( DWORD dwTranspiredTicks )
{
m_cur_ticks = dwTranspiredTicks;
if(m_callback != NULL)
m_callback(m_cur_ticks, max(m_max_ticks, m_cur_ticks));
return S_OK;
}
HRESULT CProgressHelper::End()
{
m_cur_ticks = m_max_ticks;
return S_OK;
}
HRESULT CProgressHelper::QueryInterface( REFIID riid, void __RPC_FAR *__RPC_FAR *ppvObject )
{
if(riid == IID_IWMDMProgress || riid == IID_IUnknown)
{
*ppvObject = this;
return S_OK;
}
else
{
*ppvObject = NULL;
return E_NOINTERFACE;
}
}
ULONG CProgressHelper::AddRef()
{
return m_counter++;
}
ULONG CProgressHelper::Release()
{
return m_counter--;
}