JNIEXPORT jobject JNICALL Java_jmtp_implWin32_PortableDeviceKeyCollectionImplWin32_getAt
(JNIEnv* env, jobject obj, jlong jlPosition)
{
HRESULT hr;
IPortableDeviceKeyCollection* pKeyCollection;
PROPERTYKEY key;
DWORD dwCount;
pKeyCollection = GetPortableDeviceKeyCollection(env, obj);
hr = pKeyCollection->GetCount(&dwCount);
if(FAILED(hr))
{
ThrowCOMException(env, L"Failed to count the collection", hr);
return NULL;
}
if(jlPosition < dwCount && jlPosition >= 0)
{
hr = pKeyCollection->GetAt(static_cast<DWORD>(jlPosition), &key);
if(FAILED(hr))
{
ThrowCOMException(env, L"Failed to retrieve the specified element of the collection", hr);
return NULL;
}
}
else
{
env->ThrowNew(env->FindClass("java/lang/IndexOutOfBoundsException"), "Invalid index");
return NULL;
}
return ConvertPropertyKeyToJava(env, key);
}
JNIEXPORT void JNICALL Java_jmtp_implWin32_PortableDeviceKeyCollectionImplWin32_removeAt
(JNIEnv* env, jobject obj, jlong jlPosition)
{
HRESULT hr;
IPortableDeviceKeyCollection* pKeyCollection;
DWORD dwCount;
pKeyCollection = GetPortableDeviceKeyCollection(env, obj);
hr = pKeyCollection->GetCount(&dwCount);
if(FAILED(hr))
{
ThrowCOMException(env, L"Failed to count the collection", hr);
return;
}
if(jlPosition < dwCount && jlPosition >= 0)
{
hr = pKeyCollection->RemoveAt(static_cast<DWORD>(jlPosition));
if(FAILED(hr))
{
ThrowCOMException(env, L"Failed to remove the specified element from the collection", hr);
return;
}
}
else
{
env->ThrowNew(env->FindClass("java/lang/IndexOutOfBoundsException"), "Invalid index");
}
}
JNIEXPORT void JNICALL Java_jmtp_implWin32_PortableDeviceKeyCollectionImplWin32_clear
(JNIEnv* env, jobject obj)
{
HRESULT hr;
IPortableDeviceKeyCollection* pKeyCollection;
pKeyCollection = GetPortableDeviceKeyCollection(env, obj);
hr = pKeyCollection->Clear();
if(FAILED(hr))
{
ThrowCOMException(env, L"Failed to clear the collection", hr);
return;
}
}
JNIEXPORT jlong JNICALL Java_jmtp_implWin32_PortableDeviceKeyCollectionImplWin32_count
(JNIEnv* env, jobject obj)
{
HRESULT hr;
IPortableDeviceKeyCollection* pKeyCollection;
DWORD dwCount;
pKeyCollection = GetPortableDeviceKeyCollection(env, obj);
hr = pKeyCollection->GetCount(&dwCount);
if(FAILED(hr))
{
ThrowCOMException(env, L"Failed to count the collection", hr);
return -1;
}
return dwCount;
}
JNIEXPORT void JNICALL Java_jmtp_implWin32_PortableDeviceKeyCollectionImplWin32_add
(JNIEnv* env, jobject obj, jobject jobjKey)
{
HRESULT hr;
IPortableDeviceKeyCollection* pKeyCollection;
if(jobjKey == NULL)
{
env->ThrowNew(env->FindClass("java/lang/IllegalArgumentException"), "key can't be null");
return;
}
else
{
pKeyCollection = GetPortableDeviceKeyCollection(env, obj);
hr = pKeyCollection->Add(ConvertJavaToPropertyKey(env, jobjKey));
if(FAILED(hr))
{
ThrowCOMException(env, L"Failed to add the key to the collection", hr);
return;
}
}
}
PyObject* get_device_information(IPortableDevice *device, IPortableDevicePropertiesBulk **pb) { // {{{
IPortableDeviceContent *content = NULL;
IPortableDeviceProperties *properties = NULL;
IPortableDevicePropertiesBulk *properties_bulk = NULL;
IPortableDeviceKeyCollection *keys = NULL;
IPortableDeviceValues *values = NULL;
IPortableDeviceCapabilities *capabilities = NULL;
IPortableDevicePropVariantCollection *categories = NULL;
HRESULT hr;
DWORD num_of_categories, i;
LPWSTR temp;
ULONG ti;
PyObject *t, *ans = NULL, *storage = NULL;
const char *type = NULL;
Py_BEGIN_ALLOW_THREADS;
hr = CoCreateInstance(CLSID_PortableDeviceKeyCollection, NULL,
CLSCTX_INPROC_SERVER, IID_PPV_ARGS(&keys));
Py_END_ALLOW_THREADS;
if (FAILED(hr)) {hresult_set_exc("Failed to create IPortableDeviceKeyCollection", hr); goto end;}
Py_BEGIN_ALLOW_THREADS;
hr = keys->Add(WPD_DEVICE_PROTOCOL);
// Despite the MSDN documentation, this does not exist in PortableDevice.h
// hr = keys->Add(WPD_DEVICE_TRANSPORT);
hr = keys->Add(WPD_DEVICE_FRIENDLY_NAME);
hr = keys->Add(WPD_DEVICE_MANUFACTURER);
hr = keys->Add(WPD_DEVICE_MODEL);
hr = keys->Add(WPD_DEVICE_SERIAL_NUMBER);
hr = keys->Add(WPD_DEVICE_FIRMWARE_VERSION);
hr = keys->Add(WPD_DEVICE_TYPE);
Py_END_ALLOW_THREADS;
if (FAILED(hr)) {hresult_set_exc("Failed to add keys to IPortableDeviceKeyCollection", hr); goto end;}
Py_BEGIN_ALLOW_THREADS;
hr = device->Content(&content);
Py_END_ALLOW_THREADS;
if (FAILED(hr)) {hresult_set_exc("Failed to get IPortableDeviceContent", hr); goto end; }
Py_BEGIN_ALLOW_THREADS;
hr = content->Properties(&properties);
Py_END_ALLOW_THREADS;
if (FAILED(hr)) {hresult_set_exc("Failed to get IPortableDeviceProperties", hr); goto end; }
Py_BEGIN_ALLOW_THREADS;
hr = properties->GetValues(WPD_DEVICE_OBJECT_ID, keys, &values);
Py_END_ALLOW_THREADS;
if(FAILED(hr)) {hresult_set_exc("Failed to get device info", hr); goto end; }
Py_BEGIN_ALLOW_THREADS;
hr = device->Capabilities(&capabilities);
Py_END_ALLOW_THREADS;
if(FAILED(hr)) {hresult_set_exc("Failed to get device capabilities", hr); goto end; }
Py_BEGIN_ALLOW_THREADS;
hr = capabilities->GetFunctionalCategories(&categories);
Py_END_ALLOW_THREADS;
if(FAILED(hr)) {hresult_set_exc("Failed to get device functional categories", hr); goto end; }
Py_BEGIN_ALLOW_THREADS;
hr = categories->GetCount(&num_of_categories);
Py_END_ALLOW_THREADS;
if(FAILED(hr)) {hresult_set_exc("Failed to get device functional categories number", hr); goto end; }
ans = PyDict_New();
if (ans == NULL) {PyErr_NoMemory(); goto end;}
if (SUCCEEDED(values->GetStringValue(WPD_DEVICE_PROTOCOL, &temp))) {
t = PyUnicode_FromWideChar(temp, wcslen(temp));
if (t != NULL) {PyDict_SetItemString(ans, "protocol", t); Py_DECREF(t);}
CoTaskMemFree(temp);
}
// if (SUCCEEDED(values->GetUnsignedIntegerValue(WPD_DEVICE_TRANSPORT, &ti))) {
// PyDict_SetItemString(ans, "isusb", (ti == WPD_DEVICE_TRANSPORT_USB) ? Py_True : Py_False);
// t = PyLong_FromUnsignedLong(ti);
// }
if (SUCCEEDED(values->GetUnsignedIntegerValue(WPD_DEVICE_TYPE, &ti))) {
switch (ti) {
case WPD_DEVICE_TYPE_CAMERA:
type = "camera"; break;
case WPD_DEVICE_TYPE_MEDIA_PLAYER:
type = "media player"; break;
case WPD_DEVICE_TYPE_PHONE:
type = "phone"; break;
case WPD_DEVICE_TYPE_VIDEO:
type = "video"; break;
case WPD_DEVICE_TYPE_PERSONAL_INFORMATION_MANAGER:
type = "personal information manager"; break;
case WPD_DEVICE_TYPE_AUDIO_RECORDER:
type = "audio recorder"; break;
default:
type = "unknown";
}
#if PY_MAJOR_VERSION >= 3
t = PyUnicode_FromString(type);
#else
t = PyString_FromString(type);
#endif
if (t != NULL) {
PyDict_SetItemString(ans, "type", t); Py_DECREF(t);
}
}
if (SUCCEEDED(values->GetStringValue(WPD_DEVICE_FRIENDLY_NAME, &temp))) {
t = PyUnicode_FromWideChar(temp, wcslen(temp));
if (t != NULL) {PyDict_SetItemString(ans, "friendly_name", t); Py_DECREF(t);}
CoTaskMemFree(temp);
}
if (SUCCEEDED(values->GetStringValue(WPD_DEVICE_MANUFACTURER, &temp))) {
t = PyUnicode_FromWideChar(temp, wcslen(temp));
if (t != NULL) {PyDict_SetItemString(ans, "manufacturer_name", t); Py_DECREF(t);}
CoTaskMemFree(temp);
}
if (SUCCEEDED(values->GetStringValue(WPD_DEVICE_MODEL, &temp))) {
t = PyUnicode_FromWideChar(temp, wcslen(temp));
if (t != NULL) {PyDict_SetItemString(ans, "model_name", t); Py_DECREF(t);}
CoTaskMemFree(temp);
}
if (SUCCEEDED(values->GetStringValue(WPD_DEVICE_SERIAL_NUMBER, &temp))) {
t = PyUnicode_FromWideChar(temp, wcslen(temp));
if (t != NULL) {PyDict_SetItemString(ans, "serial_number", t); Py_DECREF(t);}
CoTaskMemFree(temp);
}
if (SUCCEEDED(values->GetStringValue(WPD_DEVICE_FIRMWARE_VERSION, &temp))) {
t = PyUnicode_FromWideChar(temp, wcslen(temp));
if (t != NULL) {PyDict_SetItemString(ans, "device_version", t); Py_DECREF(t);}
CoTaskMemFree(temp);
}
t = Py_False;
for (i = 0; i < num_of_categories; i++) {
PROPVARIANT pv;
PropVariantInit(&pv);
if (SUCCEEDED(categories->GetAt(i, &pv)) && pv.puuid != NULL) {
if (IsEqualGUID(WPD_FUNCTIONAL_CATEGORY_STORAGE, *pv.puuid)) {
t = Py_True;
}
}
PropVariantClear(&pv);
if (t == Py_True) break;
}
PyDict_SetItemString(ans, "has_storage", t);
if (t == Py_True) {
storage = get_storage_info(device);
if (storage == NULL) {
PyObject *exc_type, *exc_value, *exc_tb;
PyErr_Fetch(&exc_type, &exc_value, &exc_tb);
if (exc_type != NULL && exc_value != NULL) {
PyErr_NormalizeException(&exc_type, &exc_value, &exc_tb);
PyDict_SetItemString(ans, "storage_error", exc_value);
Py_DECREF(exc_value); exc_value = NULL;
}
Py_XDECREF(exc_type); Py_XDECREF(exc_value); Py_XDECREF(exc_tb);
goto end;
}
PyDict_SetItemString(ans, "storage", storage);
}
Py_BEGIN_ALLOW_THREADS;
hr = properties->QueryInterface(IID_PPV_ARGS(&properties_bulk));
Py_END_ALLOW_THREADS;
PyDict_SetItemString(ans, "has_bulk_properties", (FAILED(hr)) ? Py_False: Py_True);
if (pb != NULL) *pb = (SUCCEEDED(hr)) ? properties_bulk : NULL;
end:
if (keys != NULL) keys->Release();
if (values != NULL) values->Release();
if (properties != NULL) properties->Release();
if (properties_bulk != NULL && pb == NULL) properties_bulk->Release();
if (content != NULL) content->Release();
if (capabilities != NULL) capabilities->Release();
if (categories != NULL) categories->Release();
return ans;
} // }}}
void Parse_MTP::mtp_get_metadata(PCWSTR object_id, IPortableDeviceContent *pdc, bool clear)
{
IPortableDeviceValues *pdv;
IPortableDeviceKeyCollection *pdkc;
IPortableDeviceProperties *pdp = NULL;
HRESULT hr;
scrob_entry new_entry;
LPWSTR album = NULL;
LPWSTR artist = NULL;
LPWSTR name = NULL;
ULONG usecount;
ULONG skipcount;
ULONG tracknum;
ULONGLONG duration;
PROPVARIANT date;
new_entry.length = 0;
hr = CoCreateInstance(CLSID_PortableDeviceValues, NULL, CLSCTX_INPROC_SERVER,
IID_IPortableDeviceValues, (VOID**)&pdv);
if (FAILED(hr))
{
emit add_log(LOG_ERROR, QString("mtp_get_metadata: Failed to create IPortableDeviceValues: %1")
.arg(hr, 0, 16));
return;
}
hr = CoCreateInstance(CLSID_PortableDeviceKeyCollection, NULL, CLSCTX_INPROC_SERVER,
IID_IPortableDeviceKeyCollection, (VOID**)&pdkc);
if (FAILED(hr))
{
emit add_log(LOG_ERROR, QString("mtp_get_metadata: Failed to create IPortableDeviceKeyCollection: %1")
.arg(hr, 0, 16));
return;
}
hr = pdc->Properties(&pdp);
if (FAILED(hr))
{
emit add_log(LOG_ERROR, QString("mtp_get_metadata: Failed to get properties from IPortableDeviceContent: %1")
.arg(hr, 0, 16));
goto mtp_get_metadata_cleanup;
}
hr = pdkc->Add(WPD_MEDIA_USE_COUNT); //VT_UI4
hr = pdkc->Add(WPD_MEDIA_SKIP_COUNT); //VT_UI4
hr = pdkc->Add(WPD_MEDIA_LAST_ACCESSED_TIME); //VT_DATE
hr = pdkc->Add(WPD_MUSIC_ALBUM); //VT_LPWSTR
hr = pdkc->Add(WPD_MUSIC_TRACK); //VT_UI4
hr = pdkc->Add(WPD_MEDIA_DURATION); //VT_UI8, in milliseconds
hr = pdkc->Add(WPD_OBJECT_NAME); //VT_LPWSTR
hr = pdkc->Add(WPD_MEDIA_ARTIST); //VT_LPWSTR
if (FAILED(hr))
{
emit add_log(LOG_ERROR, QString("mtp_get_metadata: Failed to add to IPortableDeviceKeyCollection: %1")
.arg(hr, 0, 16));
goto mtp_get_metadata_cleanup;
}
hr = pdp->GetValues(object_id, pdkc, &pdv);
if (FAILED(hr))
{
emit add_log(LOG_ERROR, QString("mtp_get_metadata: Failed to get values from IPortableDeviceProperties: %1")
.arg(hr, 0, 16));
goto mtp_get_metadata_cleanup;
}
hr = pdv->GetStringValue(WPD_MUSIC_ALBUM, &album);
if (SUCCEEDED(hr))
new_entry.album = QString::fromStdWString(album);
hr = pdv->GetStringValue(WPD_MEDIA_ARTIST, &artist);
if (SUCCEEDED(hr))
new_entry.artist = QString::fromStdWString(artist);
hr = pdv->GetStringValue(WPD_OBJECT_NAME, &name);
if (SUCCEEDED(hr))
new_entry.title = QString::fromStdWString(name);
hr = pdv->GetUnsignedIntegerValue(WPD_MEDIA_USE_COUNT, &usecount);
hr = pdv->GetUnsignedIntegerValue(WPD_MEDIA_SKIP_COUNT, &skipcount);
hr = pdv->GetUnsignedIntegerValue(WPD_MUSIC_TRACK, &tracknum);
if (SUCCEEDED(hr))
new_entry.tracknum = tracknum;
hr = pdv->GetUnsignedLargeIntegerValue(WPD_MEDIA_DURATION, &duration);
if (SUCCEEDED(hr))
new_entry.length = duration/1000;
hr = pdv->GetValue(WPD_MEDIA_LAST_ACCESSED_TIME, &date);
if (SUCCEEDED(hr))
{
SYSTEMTIME st;
VariantTimeToSystemTime(date.date, &st);
PropVariantClear(&date);
QDate d = QDate(st.wYear, st.wMonth, st.wDay);
QTime t = QTime(st.wHour, st.wMinute, st.wSecond);
new_entry.when = QDateTime(d, t).toTime_t() - tz_offset;
// Not all MTP devices provide a valid result for
// WPD_MEDIA_LAST_ACCESSED_TIME, so sanity check it
// are we within (+/-) 30 days?
long offset = new_entry.when - QDateTime::currentDateTime().toTime_t();
if (abs(offset) > 2592000)
new_entry.when = 0;
}
emit add_log(LOG_TRACE,
QString("%1 : %2 : %3 : %4 : %5 : %6 : %7")
.arg(new_entry.artist)
.arg(new_entry.title)
.arg(new_entry.album)
.arg(new_entry.tracknum)
.arg(new_entry.length)
.arg(usecount)
.arg(skipcount)
);
tracks++;
if (usecount > 0
&& new_entry.artist.length()
&& new_entry.title.length())
{
if (clear)
{
IPortableDeviceValues *pdv_clear = NULL;
hr = pdv->Clear();
hr = pdv->SetUnsignedIntegerValue(WPD_MEDIA_USE_COUNT, 0);
hr = pdp->SetValues(object_id, pdv, &pdv_clear);
if (SUCCEEDED(hr))
{
if (pdv_clear != NULL)
pdv_clear->Clear();
pdv_clear = NULL;
}
}
else
{
playcounts++;
new_entry.played = 'L';
for (ULONG i = 0; i < usecount; i++)
{
entries++;
// Only use the last playcount for one instance
// we'll recalc the rest afterwards
// via Scrobble::check_timestamps()
if (i > 1)
new_entry.when = 0;
emit entry(new_entry);
}
}
}
mtp_get_metadata_cleanup:
if (album)
CoTaskMemFree(album);
if (artist)
CoTaskMemFree(artist);
if (name)
CoTaskMemFree(name);
if (pdp)
{
pdp->Release();
pdp = NULL;
}
if (pdv)
{
pdv->Release();
pdv = NULL;
}
}
bool Parse_MTP::is_device_mtp(DWORD index, LPCWSTR device_id)
{
bool is_mtp = false;
IPortableDevice *pd;
IPortableDeviceValues *pdv;
IPortableDeviceKeyCollection *pdkc;
IPortableDeviceProperties *pdp = NULL;
IPortableDeviceContent *pdc = NULL;
HRESULT hr;
LPWSTR dev_protocol = NULL;
QString mtp;
hr = CoCreateInstance(CLSID_PortableDevice, NULL, CLSCTX_INPROC_SERVER,
IID_IPortableDevice, (VOID**)&pd);
if (FAILED(hr))
{
emit add_log(LOG_ERROR, QString("is_device_mtp: Failed to create IPortableDevice: %1")
.arg(hr, 0, 16));
return false;
}
hr = CoCreateInstance(CLSID_PortableDeviceValues, NULL, CLSCTX_INPROC_SERVER,
IID_IPortableDeviceValues, (VOID**)&pdv);
if (FAILED(hr))
{
emit add_log(LOG_ERROR, QString("is_device_mtp: Failed to create IPortableDeviceValues: %1")
.arg(hr, 0, 16));
return false;
}
hr = CoCreateInstance(CLSID_PortableDeviceKeyCollection, NULL, CLSCTX_INPROC_SERVER,
IID_IPortableDeviceKeyCollection, (VOID**)&pdkc);
if (FAILED(hr))
{
emit add_log(LOG_ERROR, QString("is_device_mtp: Failed to create IPortableDeviceKeyCollection: %1")
.arg(hr, 0, 16));
return false;
}
// after this we're creating objects which need to be cleaned up.
hr = pd->Open(device_id, pdv);
if (FAILED(hr))
{
emit add_log(LOG_ERROR, QString("is_device_mtp: Failed to open IPortableDevice: %1")
.arg(hr, 0, 16));
goto is_mtp_cleanup;
}
hr = pd->Content(&pdc);
if (FAILED(hr))
{
emit add_log(LOG_ERROR, QString("is_device_mtp: Failed to retrieve content from IPortableDevice: %1")
.arg(hr, 0, 16));
goto is_mtp_cleanup;
}
hr = pdc->Properties(&pdp);
if (FAILED(hr))
{
emit add_log(LOG_ERROR, QString("is_device_mtp: Failed to get properties from IPortableDeviceContent: %1")
.arg(hr, 0, 16));
goto is_mtp_cleanup;
}
hr = pdkc->Add(WPD_DEVICE_PROTOCOL);
if (FAILED(hr))
{
emit add_log(LOG_ERROR, QString("is_device_mtp: Failed to add to IPortableDeviceKeyCollection: %1")
.arg(hr, 0, 16));
goto is_mtp_cleanup;
}
// WPD_DEVICE_OBJECT_ID is the top level object
hr = pdp->GetValues(WPD_DEVICE_OBJECT_ID, pdkc, &pdv);
if (FAILED(hr))
{
emit add_log(LOG_ERROR, QString("is_device_mtp: Failed to get values from IPortableDeviceProperties: %1")
.arg(hr, 0, 16));
goto is_mtp_cleanup;
}
hr = pdv->GetStringValue(WPD_DEVICE_PROTOCOL, &dev_protocol);
if (FAILED(hr))
{
emit add_log(LOG_ERROR, QString("is_device_mtp: Failed to GetStringValue: %1")
.arg(hr, 0, 16));
goto is_mtp_cleanup;
}
mtp = QString::fromStdWString(dev_protocol);
emit add_log(LOG_INFO, QString("Device %1: %2").arg(index).arg(mtp));
if (mtp.startsWith("MTP"))
{
is_mtp = true;
emit add_log(LOG_INFO, QString("Device %1: Is a MTP device").arg(index));
}
is_mtp_cleanup:
if (dev_protocol)
CoTaskMemFree(dev_protocol);
if (pdc)
{
pdc->Release();
pdc = NULL;
}
if (pdp)
{
pdp->Release();
pdp = NULL;
}
if (pdv)
{
pdv->Release();
pdv = NULL;
}
return is_mtp;
}
