548 lines
16 KiB
C++
548 lines
16 KiB
C++
|
#include "../../Library/ml_pmp/pmp.h"
|
||
|
|
#include "../Winamp/wa_ipc.h"
|
||
|
|
#include <vector>
|
||
|
|
#include "../nu/AutoWide.h"
|
||
|
|
#include "../nu/AutoChar.h"
|
||
|
|
#include "../nu/AutoLock.h"
|
||
|
|
#include "api.h"
|
||
|
|
#include "resource.h"
|
||
|
|
#include "usbdevice.h"
|
||
|
|
#include "deviceprovider.h"
|
||
|
|
|
||
|
|
#include <devguid.h>
|
||
|
|
#include <shlobj.h>
|
||
|
|
#include <shlwapi.h>
|
||
|
|
#include <strsafe.h>
|
||
|
|
|
||
|
|
#define PLUGIN_VERSION L"1.62"
|
||
|
|
static int Init();
|
||
|
|
static void Quit();
|
||
|
|
static bool doRegisterForDevNotification(void);
|
||
|
|
static intptr_t MessageProc(int msg, intptr_t param1, intptr_t param2, intptr_t param3);
|
||
|
|
|
||
|
|
extern PMPDevicePlugin plugin = {PMPHDR_VER,0,Init,Quit,MessageProc};
|
||
|
|
|
||
|
|
bool loading_devices[26] = {0,};
|
||
|
|
|
||
|
|
// start-usb
|
||
|
|
static const wchar_t *winampini;
|
||
|
|
static std::vector<HANDLE> loadingThreads;
|
||
|
|
static std::vector<wchar_t*> blacklist;
|
||
|
|
|
||
|
|
static HDEVNOTIFY hDevNotify;
|
||
|
|
std::vector<USBDevice*> devices;
|
||
|
|
HWND config;
|
||
|
|
|
||
|
|
static DeviceProvider *deviceProvider = NULL;
|
||
|
|
static UINT_PTR rescanTimer = 0;
|
||
|
|
|
||
|
|
static void blacklistLoad() {
|
||
|
|
wchar_t keyname[64] = {0};
|
||
|
|
int l = GetPrivateProfileIntW(L"pmp_usb", L"blacklistnum", 0, winampini);
|
||
|
|
for(int i=l>100?l-100:0; i<l; i++) {
|
||
|
|
wchar_t buf[100] = {0};
|
||
|
|
StringCchPrintfW(keyname, 64, L"blacklist-%d", i);
|
||
|
|
GetPrivateProfileStringW(L"pmp_usb", keyname, L"", buf, 100, winampini);
|
||
|
|
if(buf[0])
|
||
|
|
{
|
||
|
|
blacklist.push_back(_wcsdup(buf));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
static void blacklistSave() {
|
||
|
|
wchar_t buf[64] = {0};
|
||
|
|
StringCchPrintfW(buf, 64, L"%u", blacklist.size());
|
||
|
|
WritePrivateProfileStringW(L"pmp_usb", L"blacklistnum", buf, winampini);
|
||
|
|
for(size_t i=0; i<blacklist.size(); i++)
|
||
|
|
{
|
||
|
|
StringCchPrintfW(buf, 64, L"blacklist-%u", i);
|
||
|
|
WritePrivateProfileStringW(L"pmp_usb", buf, (const wchar_t*)blacklist.at(i), winampini);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
static wchar_t *makeBlacklistString(wchar_t drive) {
|
||
|
|
wchar_t path[4]={drive,L":\\"};
|
||
|
|
wchar_t name[100]=L"";
|
||
|
|
wchar_t buf[FIELD_LENGTH]=L"";
|
||
|
|
DWORD serial=0;
|
||
|
|
UINT olderrmode=SetErrorMode(SEM_NOOPENFILEERRORBOX | SEM_FAILCRITICALERRORS);
|
||
|
|
GetVolumeInformation(path,name,100,&serial,NULL,NULL,NULL,0);
|
||
|
|
|
||
|
|
if(serial)
|
||
|
|
{
|
||
|
|
StringCchPrintf(buf, FIELD_LENGTH, L"s:%d",serial);
|
||
|
|
SetErrorMode(olderrmode);
|
||
|
|
return _wcsdup(buf);
|
||
|
|
}
|
||
|
|
|
||
|
|
{
|
||
|
|
ULARGE_INTEGER tfree={0,}, total={0,}, freeb={0,};
|
||
|
|
GetDiskFreeSpaceEx(path, &tfree, &total, &freeb);
|
||
|
|
StringCchPrintf(buf, FIELD_LENGTH, L"n:%s,%d,%d", name, total.HighPart, total.LowPart);
|
||
|
|
SetErrorMode(olderrmode);
|
||
|
|
return _wcsdup(buf);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
static bool blacklistCheck(wchar_t drive) {
|
||
|
|
wchar_t *s = makeBlacklistString(drive);
|
||
|
|
if (s)
|
||
|
|
{
|
||
|
|
for(size_t i=0; i<blacklist.size(); i++)
|
||
|
|
{
|
||
|
|
if(!wcscmp(s,(wchar_t*)blacklist.at(i)))
|
||
|
|
{
|
||
|
|
free(s);
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
free(s);
|
||
|
|
}
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
// helpers
|
||
|
|
static DWORD WINAPI ThreadFunc_Load(LPVOID lpParam) {
|
||
|
|
wchar_t drive = (wchar_t)(intptr_t)lpParam;
|
||
|
|
pmpDeviceLoading load;
|
||
|
|
Device * d = new USBDevice(drive,&load);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
static Nullsoft::Utility::LockGuard connect_guard;
|
||
|
|
void connectDrive(wchar_t drive, bool checkSize=true, bool checkBlacklist=true)
|
||
|
|
{
|
||
|
|
Nullsoft::Utility::AutoLock connect_lock(connect_guard);
|
||
|
|
// capitalize
|
||
|
|
if (drive >= 'a' && drive <= 'z')
|
||
|
|
drive = drive - 32;
|
||
|
|
|
||
|
|
// reject invalid drive letters
|
||
|
|
if (drive < 'A' || drive > 'Z')
|
||
|
|
return;
|
||
|
|
|
||
|
|
if(checkBlacklist && blacklistCheck(drive)) return;
|
||
|
|
|
||
|
|
// if device is taken already ignore
|
||
|
|
for (std::vector<USBDevice*>::const_iterator e = devices.begin(); e != devices.end(); e++)
|
||
|
|
{
|
||
|
|
if ((*e)->drive == drive)
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (loading_devices[drive-'A'])
|
||
|
|
return;
|
||
|
|
|
||
|
|
loading_devices[drive-'A'] = true;
|
||
|
|
|
||
|
|
wchar_t path[4]=L"x:\\";
|
||
|
|
path[0]=drive;
|
||
|
|
|
||
|
|
if(checkSize)
|
||
|
|
{
|
||
|
|
ULARGE_INTEGER total;
|
||
|
|
|
||
|
|
if (0 == GetDiskFreeSpaceExW(path, NULL, &total, NULL) ||
|
||
|
|
total.HighPart == 0 && total.LowPart == 0)
|
||
|
|
{
|
||
|
|
loading_devices[drive-'A'] = false;
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Ignore iPods...
|
||
|
|
// Check for a "iPod_Control" folder,
|
||
|
|
// if "iPod_Control" is present just bail and
|
||
|
|
// let the iPod plugin handle it
|
||
|
|
const wchar_t iPodDb[] = {drive,L":\\iPod_Control"};
|
||
|
|
WIN32_FIND_DATA ffd={0};
|
||
|
|
HANDLE h = FindFirstFile(iPodDb,&ffd);
|
||
|
|
if (h != INVALID_HANDLE_VALUE)
|
||
|
|
{
|
||
|
|
FindClose(h);
|
||
|
|
|
||
|
|
if (0 != (ffd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY))
|
||
|
|
{
|
||
|
|
loading_devices[drive-'A'] = false;
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Ignore androids too, have a specific pulgin to take care of android...
|
||
|
|
// Check for a "Android" folder,
|
||
|
|
// if "Android" is present just bail and
|
||
|
|
// let the pmp_android plugin handle it
|
||
|
|
|
||
|
|
const wchar_t androidTopLevelDir[] = {drive,L":\\Android"};
|
||
|
|
|
||
|
|
h = FindFirstFile(androidTopLevelDir, &ffd);
|
||
|
|
if (h != INVALID_HANDLE_VALUE)
|
||
|
|
{
|
||
|
|
FindClose(h);
|
||
|
|
|
||
|
|
if (0 != (ffd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY))
|
||
|
|
{
|
||
|
|
loading_devices[drive-'A'] = false;
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
//not an ipod, not an android
|
||
|
|
//do we know about the device already? is there a metadata.dat file ?
|
||
|
|
wchar_t cacheFile[FIELD_LENGTH] = {0};
|
||
|
|
StringCchPrintf(cacheFile, FIELD_LENGTH, L"%c:\\%s", drive, NDE_CACHE_DAT);
|
||
|
|
|
||
|
|
if (!PathFileExists(cacheFile))
|
||
|
|
{
|
||
|
|
// new device, never plugged in before
|
||
|
|
wchar_t drvname[100] = {0}, titleStr[128] = {0};
|
||
|
|
DWORD serial=0;
|
||
|
|
GetVolumeInformation(path,drvname,100,&serial,NULL,NULL,NULL,0);
|
||
|
|
|
||
|
|
wchar_t buf[1024] = {0};
|
||
|
|
StringCchPrintf(buf, ARRAYSIZE(buf), WASABI_API_LNGSTRINGW(IDS_REMOVEABLE_DRIVE_DETECTED),
|
||
|
|
drvname, towupper(drive));
|
||
|
|
|
||
|
|
if(MessageBox(plugin.hwndLibraryParent,buf,
|
||
|
|
WASABI_API_LNGSTRINGW_BUF(IDS_WINAMP_PMP_SUPPORT,titleStr,128),
|
||
|
|
MB_YESNO|MB_SETFOREGROUND|MB_SYSTEMMODAL|MB_TOPMOST) == IDNO)
|
||
|
|
{
|
||
|
|
wchar_t * bstr = makeBlacklistString(drive);
|
||
|
|
loading_devices[drive-'A'] = false;
|
||
|
|
if (bstr)
|
||
|
|
{
|
||
|
|
blacklist.push_back(bstr);
|
||
|
|
blacklistSave();
|
||
|
|
}
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD dwThreadId;
|
||
|
|
HANDLE loadingThread = CreateThread(NULL, 0, ThreadFunc_Load, (LPVOID)(intptr_t)drive, 0, &dwThreadId);
|
||
|
|
if(NULL != loadingThread)
|
||
|
|
loadingThreads.push_back(loadingThread);
|
||
|
|
else
|
||
|
|
loading_devices[drive-'A'] = false;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
static void autoDetectCallback(wchar_t drive,UINT type) {
|
||
|
|
if(type == DRIVE_REMOVABLE)
|
||
|
|
{
|
||
|
|
connectDrive(drive,true,true);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
// end-usb
|
||
|
|
static int Init()
|
||
|
|
{
|
||
|
|
WasabiInit();
|
||
|
|
|
||
|
|
// start-usb
|
||
|
|
winampini = (const wchar_t*)SendMessage(plugin.hwndWinampParent,WM_WA_IPC,0,IPC_GETINIFILEW);
|
||
|
|
// need to have this initialised before we try to do anything with localisation features
|
||
|
|
WASABI_API_START_LANG(plugin.hDllInstance,PmpUSBLangGUID);
|
||
|
|
// end-usb
|
||
|
|
|
||
|
|
static wchar_t szDescription[256];
|
||
|
|
StringCchPrintfW(szDescription, ARRAYSIZE(szDescription),
|
||
|
|
WASABI_API_LNGSTRINGW(IDS_NULLSOFT_USB_DEVICE_PLUGIN), PLUGIN_VERSION);
|
||
|
|
plugin.description = szDescription;
|
||
|
|
|
||
|
|
/** load up the backlist */
|
||
|
|
blacklistLoad();
|
||
|
|
|
||
|
|
if (NULL != AGAVE_API_DEVICEMANAGER &&
|
||
|
|
NULL == deviceProvider)
|
||
|
|
{
|
||
|
|
if (SUCCEEDED(DeviceProvider::CreateInstance(&deviceProvider)) &&
|
||
|
|
FAILED(deviceProvider->Register(AGAVE_API_DEVICEMANAGER)))
|
||
|
|
{
|
||
|
|
deviceProvider->Release();
|
||
|
|
deviceProvider = NULL;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/* Our device shows up as a normal drive */
|
||
|
|
if (NULL == deviceProvider ||
|
||
|
|
FAILED(deviceProvider->BeginDiscovery(AGAVE_API_DEVICEMANAGER)))
|
||
|
|
{
|
||
|
|
SendMessage(plugin.hwndPortablesParent,WM_PMP_IPC,(WPARAM)autoDetectCallback,PMP_IPC_ENUM_ACTIVE_DRIVES);
|
||
|
|
}
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
static void Quit()
|
||
|
|
{
|
||
|
|
if (NULL != deviceProvider)
|
||
|
|
{
|
||
|
|
deviceProvider->Unregister();
|
||
|
|
deviceProvider->Release();
|
||
|
|
deviceProvider = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
WasabiQuit();
|
||
|
|
UnregisterDeviceNotification(hDevNotify);
|
||
|
|
USBDevice::CloseDatabase();
|
||
|
|
}
|
||
|
|
|
||
|
|
static wchar_t FirstDriveFromMask(ULONG *unitmask) {
|
||
|
|
char i;
|
||
|
|
ULONG adj = 0x1, mask = *unitmask;
|
||
|
|
for(i=0; i<26; ++i) {
|
||
|
|
if(mask & 0x1) {
|
||
|
|
*unitmask -= adj;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
adj = adj << 1;
|
||
|
|
mask = mask >> 1;
|
||
|
|
}
|
||
|
|
return (i+L'A');
|
||
|
|
}
|
||
|
|
|
||
|
|
static int GetNumberOfDrivesFromMask(ULONG unitmask) {
|
||
|
|
int count = 0;
|
||
|
|
for(int i=0; i<26; ++i)
|
||
|
|
{
|
||
|
|
if(unitmask & 0x1)
|
||
|
|
count++;
|
||
|
|
|
||
|
|
unitmask = unitmask >> 1;
|
||
|
|
}
|
||
|
|
return count;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
static void CALLBACK RescanOnTimer(HWND hwnd, UINT uMsg, UINT_PTR idEvent, DWORD dwTime)
|
||
|
|
{
|
||
|
|
KillTimer(hwnd, idEvent);
|
||
|
|
if (idEvent == rescanTimer)
|
||
|
|
rescanTimer = 0;
|
||
|
|
|
||
|
|
if (NULL == deviceProvider ||
|
||
|
|
FAILED(deviceProvider->BeginDiscovery(AGAVE_API_DEVICEMANAGER)))
|
||
|
|
{
|
||
|
|
PostMessage(plugin.hwndPortablesParent,WM_PMP_IPC,(WPARAM)autoDetectCallback,PMP_IPC_ENUM_ACTIVE_DRIVES);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
int wmDeviceChange(WPARAM wParam, LPARAM lParam)
|
||
|
|
{
|
||
|
|
UINT olderrmode=SetErrorMode(SEM_NOOPENFILEERRORBOX|SEM_FAILCRITICALERRORS);
|
||
|
|
if(wParam==DBT_DEVICEARRIVAL || wParam==DBT_DEVICEREMOVECOMPLETE)
|
||
|
|
{ // something has been inserted or removed
|
||
|
|
PDEV_BROADCAST_HDR lpdb = (PDEV_BROADCAST_HDR)lParam;
|
||
|
|
if(lpdb->dbch_devicetype == DBT_DEVTYP_VOLUME)
|
||
|
|
{ // its a volume
|
||
|
|
PDEV_BROADCAST_VOLUME lpdbv = (PDEV_BROADCAST_VOLUME)lpdb;
|
||
|
|
if((!(lpdbv->dbcv_flags & DBTF_MEDIA) && !(lpdbv->dbcv_flags & DBTF_NET)))
|
||
|
|
{ // its not a network drive or a CD/floppy, game on!
|
||
|
|
ULONG dbcv_unitmask = lpdbv->dbcv_unitmask;
|
||
|
|
|
||
|
|
// see just how many drives have been flagged on the action
|
||
|
|
// eg one usb drive could have multiple partitions that we handle
|
||
|
|
int count = GetNumberOfDrivesFromMask(dbcv_unitmask);
|
||
|
|
for(int j = 0; j < count; j++)
|
||
|
|
{
|
||
|
|
wchar_t drive = FirstDriveFromMask(&dbcv_unitmask);
|
||
|
|
if((wParam == DBT_DEVICEARRIVAL) && !blacklistCheck(drive))
|
||
|
|
{ // connected
|
||
|
|
connectDrive(drive);
|
||
|
|
//send a message as if the user just selected a drive from the combo box, this way the fields are refreshed to the correct device's settings
|
||
|
|
SendMessage(config, WM_COMMAND,MAKEWPARAM(IDC_DRIVESELECT,CBN_SELCHANGE),0);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{ // removal
|
||
|
|
for(size_t i=0; i < devices.size(); i++) {
|
||
|
|
USBDevice * d = (USBDevice*)devices.at(i);
|
||
|
|
if(d->drive == drive)
|
||
|
|
{
|
||
|
|
devices.erase(devices.begin() + i);
|
||
|
|
if(config) SendMessage(config,WM_USER,0,0); //refresh fields
|
||
|
|
if(config) SendMessage(config,WM_COMMAND, MAKEWPARAM(IDC_DRIVESELECT,CBN_SELCHANGE),0); //update to correct device change as if the user had clicked on the combo box themself
|
||
|
|
SendMessage(plugin.hwndPortablesParent,WM_PMP_IPC,(intptr_t)d,PMP_IPC_DEVICEDISCONNECTED);
|
||
|
|
delete d;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
rescanTimer = SetTimer(NULL, rescanTimer, 10000, RescanOnTimer);
|
||
|
|
}
|
||
|
|
SetErrorMode(olderrmode);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int IsDriveConnectedToPMP(wchar_t drive) {
|
||
|
|
for(size_t i = 0; i < devices.size(); i++)
|
||
|
|
{
|
||
|
|
if(((USBDevice*)devices.at(i))->drive == drive)
|
||
|
|
{
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
static INT_PTR CALLBACK config_dialogProc(HWND hwndDlg, UINT uMsg, WPARAM wParam,LPARAM lParam) {
|
||
|
|
switch(uMsg) {
|
||
|
|
case WM_INITDIALOG:
|
||
|
|
{
|
||
|
|
for(wchar_t d=L'A'; d<='Z'; d++) {
|
||
|
|
wchar_t drive[3] = {d,L':',0}, drv[4] = {d,L':','\\',0};
|
||
|
|
UINT uDriveType = GetDriveType(drv);
|
||
|
|
if(uDriveType == DRIVE_REMOVABLE || uDriveType == DRIVE_CDROM || uDriveType == DRIVE_FIXED) {
|
||
|
|
int position = (int)SendDlgItemMessageW(hwndDlg,IDC_COMBO_MANUALCONNECT,CB_ADDSTRING,0,(LPARAM)drive);
|
||
|
|
SendDlgItemMessage(hwndDlg,IDC_COMBO_MANUALCONNECT,CB_SETITEMDATA,position,d);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
SendDlgItemMessage(hwndDlg,IDC_COMBO_MANUALCONNECT,CB_SETCURSEL,0,0);
|
||
|
|
SendMessage(hwndDlg,WM_COMMAND,MAKEWPARAM(IDC_COMBO_MANUALCONNECT,CBN_SELCHANGE),0);
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
case WM_CLOSE:
|
||
|
|
EndDialog(hwndDlg,0);
|
||
|
|
break;
|
||
|
|
case WM_COMMAND:
|
||
|
|
switch(LOWORD(wParam)) {
|
||
|
|
case IDC_COMBO_MANUALCONNECT:
|
||
|
|
{
|
||
|
|
if(HIWORD(wParam)==CBN_SELCHANGE) {
|
||
|
|
int indx = (int)SendDlgItemMessageW(hwndDlg,IDC_COMBO_MANUALCONNECT,CB_GETCURSEL,0,0);
|
||
|
|
wchar_t drive = (wchar_t)SendDlgItemMessage(hwndDlg,IDC_COMBO_MANUALCONNECT,CB_GETITEMDATA,indx,0);
|
||
|
|
if(indx >= 0)
|
||
|
|
{
|
||
|
|
int connected = IsDriveConnectedToPMP(drive), isblacklisted = blacklistCheck(drive);
|
||
|
|
EnableWindow(GetDlgItem(hwndDlg, IDC_MANUALCONNECT), !connected && !isblacklisted);
|
||
|
|
EnableWindow(GetDlgItem(hwndDlg, IDC_MANUALDISCONNECT), connected);
|
||
|
|
EnableWindow(GetDlgItem(hwndDlg, IDC_MANUALBLACKLIST), TRUE);
|
||
|
|
SetDlgItemText(hwndDlg, IDC_MANUALBLACKLIST, WASABI_API_LNGSTRINGW(isblacklisted ? IDS_UNBLACKLIST_DRIVE : IDS_BLACKLIST_DRIVE));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
case IDC_MANUALCONNECT:
|
||
|
|
{
|
||
|
|
char titleStr[32] = {0};
|
||
|
|
if(MessageBoxA(hwndDlg, WASABI_API_LNGSTRING(IDS_MANUAL_CONNECT_PROMPT),
|
||
|
|
WASABI_API_LNGSTRING_BUF(IDS_WARNING,titleStr,32), MB_YESNO) == IDYES)
|
||
|
|
{
|
||
|
|
int indx = (int)SendDlgItemMessageW(hwndDlg,IDC_COMBO_MANUALCONNECT,CB_GETCURSEL,0,0);
|
||
|
|
wchar_t drive = (wchar_t)SendDlgItemMessage(hwndDlg,IDC_COMBO_MANUALCONNECT,CB_GETITEMDATA,indx,0);
|
||
|
|
if(drive >= L'A' && drive <= L'Z') {
|
||
|
|
wchar_t *bl = makeBlacklistString(drive);
|
||
|
|
if (bl)
|
||
|
|
{
|
||
|
|
for(size_t i=0; i<blacklist.size(); i++)
|
||
|
|
{
|
||
|
|
if(!wcscmp(bl,(wchar_t*)blacklist.at(i)))
|
||
|
|
{
|
||
|
|
free(blacklist.at(i));
|
||
|
|
blacklist.erase(blacklist.begin() + i);
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
free(bl);
|
||
|
|
}
|
||
|
|
connectDrive(drive,false);
|
||
|
|
// should do a better check here incase of failure, etc
|
||
|
|
EnableWindow(GetDlgItem(hwndDlg, IDC_MANUALCONNECT), FALSE);
|
||
|
|
EnableWindow(GetDlgItem(hwndDlg, IDC_MANUALDISCONNECT), TRUE);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
case IDC_MANUALDISCONNECT:
|
||
|
|
{
|
||
|
|
int indx = (int)SendDlgItemMessageW(hwndDlg,IDC_COMBO_MANUALCONNECT,CB_GETCURSEL,0,0);
|
||
|
|
wchar_t drive = (wchar_t)SendDlgItemMessage(hwndDlg,IDC_COMBO_MANUALCONNECT,CB_GETITEMDATA,indx,0);
|
||
|
|
if(drive >= L'A' && drive <= L'Z') {
|
||
|
|
for(size_t i=0; i < devices.size(); i++) {
|
||
|
|
USBDevice * d = (USBDevice*)devices.at(i);
|
||
|
|
if(d->drive == drive)
|
||
|
|
{
|
||
|
|
devices.erase(devices.begin() + i);
|
||
|
|
if(config) SendMessage(config,WM_USER,0,0); //refresh fields
|
||
|
|
if(config) SendMessage(config,WM_COMMAND, MAKEWPARAM(IDC_DRIVESELECT,CBN_SELCHANGE),0); //update to correct device change as if the user had clicked on the combo box themself
|
||
|
|
SendMessage(plugin.hwndPortablesParent,WM_PMP_IPC,(intptr_t)d,PMP_IPC_DEVICEDISCONNECTED);
|
||
|
|
SendMessage(hwndDlg,WM_COMMAND,MAKEWPARAM(IDC_COMBO_MANUALCONNECT,CBN_SELCHANGE),0);
|
||
|
|
delete d;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
case IDC_MANUALBLACKLIST:
|
||
|
|
{
|
||
|
|
int indx = (int)SendDlgItemMessageW(hwndDlg,IDC_COMBO_MANUALCONNECT,CB_GETCURSEL,0,0);
|
||
|
|
wchar_t drive = (wchar_t)SendDlgItemMessage(hwndDlg,IDC_COMBO_MANUALCONNECT,CB_GETITEMDATA,indx,0);
|
||
|
|
if(drive >= L'A' && drive <= L'Z') {
|
||
|
|
wchar_t *bl = makeBlacklistString(drive);
|
||
|
|
if (bl)
|
||
|
|
{
|
||
|
|
if(!blacklistCheck(drive)) {
|
||
|
|
blacklist.push_back(bl);
|
||
|
|
// see if we've got a connected drive and prompt to remove it or wait till restart
|
||
|
|
if(IsDriveConnectedToPMP(drive)) {
|
||
|
|
wchar_t title[96] = {0};
|
||
|
|
GetWindowText(hwndDlg, title, 96);
|
||
|
|
if(MessageBox(hwndDlg,WASABI_API_LNGSTRINGW(IDS_DRIVE_CONNECTED_DISCONNECT_Q),title,MB_YESNO)==IDYES){
|
||
|
|
SendMessage(hwndDlg,WM_COMMAND,MAKEWPARAM(IDC_MANUALDISCONNECT,0),0);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
for(size_t i=0; i < blacklist.size(); i++)
|
||
|
|
{
|
||
|
|
if(!wcscmp(bl,(wchar_t*)blacklist.at(i)))
|
||
|
|
{
|
||
|
|
free(blacklist.at(i));
|
||
|
|
blacklist.erase(blacklist.begin() + i);
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
free(bl);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
SendMessage(hwndDlg,WM_COMMAND,MAKEWPARAM(IDC_COMBO_MANUALCONNECT,CBN_SELCHANGE),0);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
case IDOK:
|
||
|
|
case IDCANCEL:
|
||
|
|
blacklistSave();
|
||
|
|
EndDialog(hwndDlg,0);
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
static intptr_t MessageProc(int msg, intptr_t param1, intptr_t param2, intptr_t param3)
|
||
|
|
{
|
||
|
|
switch(msg) {
|
||
|
|
case PMP_DEVICECHANGE:
|
||
|
|
return wmDeviceChange(param1,param2);
|
||
|
|
case PMP_CONFIG:
|
||
|
|
WASABI_API_DIALOGBOXW(IDD_CONFIG_GLOBAL,(HWND)param1,config_dialogProc);
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
extern "C" __declspec(dllexport) PMPDevicePlugin *winampGetPMPDevicePlugin()
|
||
|
|
{
|
||
|
|
return &plugin;
|
||
|
|
}
|
||
|
|
|