winamp/Src/Plugins/Portable/pmp_usb/main.cpp
2024-09-24 14:54:57 +02:00

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;
}