diff options
Diffstat (limited to 'Src/Winamp/ExtendedInfo.cpp')
| -rw-r--r-- | Src/Winamp/ExtendedInfo.cpp | 363 |
1 files changed, 363 insertions, 0 deletions
diff --git a/Src/Winamp/ExtendedInfo.cpp b/Src/Winamp/ExtendedInfo.cpp new file mode 100644 index 00000000..4c743583 --- /dev/null +++ b/Src/Winamp/ExtendedInfo.cpp @@ -0,0 +1,363 @@ +#include "main.h" +#include "api.h" +#include "../nu/AutoWide.h" +#include "../nu/AutoWideFn.h" +#include "../nu/AutoChar.h" +#include "../nu/AutoCharFn.h" +#include "../nu/AutoLock.h" +#include "../nu/ns_wc.h" +#include <malloc.h> + +using namespace Nullsoft::Utility; +/* Get Extended File Info */ + +typedef int (__cdecl *GetANSIInfo)(const char *fn, const char *data, char *dest, size_t destlen); +typedef int (__cdecl *GetUnicodeInfo)(const wchar_t *fn, const char *data, wchar_t *dest, size_t destlen); + +int ConvertGetExtendedFileInfo(GetUnicodeInfo getter, const char *fn, const char *data, char *dest, size_t destlen) +{ + wchar_t *destW = 0; + + if (dest && destlen) + { + destW = (wchar_t *)_malloca(destlen * sizeof(wchar_t)); + memset(destW, 0, destlen * sizeof(wchar_t)); + } + + int retVal = getter(AutoWideFn(fn), data, destW, destlen); + if (retVal && dest && destlen) + WideCharToMultiByteSZ(CP_ACP, 0, destW, -1, dest, (int)destlen, 0, 0); + + if (destW) _freea(destW); + return retVal; +} + +int ConvertGetExtendedFileInfo(GetANSIInfo getter, const char *fn, const char *data, wchar_t *dest, size_t destlen) +{ + char *destA = 0; + + if (dest && destlen) + { + destA = (char *)_malloca(destlen * sizeof(char)); + memset(destA, 0, destlen * sizeof(char)); + } + + int retVal = getter(fn, data, destA, destlen); + if (retVal && dest && destlen) + MultiByteToWideCharSZ(CP_ACP, 0, destA, -1, dest, (int)destlen); + + if (destA) _freea(destA); + return retVal; +} + +static GetANSIInfo gefi; +static GetUnicodeInfo gefiW; +static char cachedExt[16]; +LockGuard getMetadataGuard; + +int in_get_extended_fileinfo(const char *fn, const char *metadata, char *dest, size_t destlen) +{ + AutoLock lock(getMetadataGuard); + + In_Module *i=0; + char ext[16] = {0}; + int a = 0; + + // do some extra checks on the params as quite a few clients crash when accessing here + try { + if (destlen > 65536 || destlen == 0 || !fn || (unsigned int)(ULONG_PTR)fn < 65536 || + fn && !*fn || !metadata || (unsigned int)(ULONG_PTR)metadata < 65536 || + metadata && !*metadata || !dest || (unsigned int)(ULONG_PTR)dest < 65536) + return 0; + } catch (...) { + return 0; + } + + if (dest) + memset(dest, 0, destlen); + + extension_ex(fn, ext, sizeof(ext)); + + if (!_stricmp(cachedExt, ext) && *ext) + { + if (gefi) + return gefi(fn, metadata, dest, destlen); + else if (gefiW) // should always be true if we got this far + return ConvertGetExtendedFileInfo(gefiW, fn, metadata, dest, destlen); + else + return 0; + } + + while (a >= 0) + { + i = in_setmod_noplay(AutoWideFn(fn), &a); + if (!i) + break; + + if (a >= 0) a++; + + lstrcpynA(cachedExt, ext, sizeof(cachedExt)/sizeof(*cachedExt)); + + gefi = (GetANSIInfo)GetProcAddress(i->hDllInstance, "winampGetExtendedFileInfo"); + gefiW = (GetUnicodeInfo)GetProcAddress(i->hDllInstance, "winampGetExtendedFileInfoW"); + if (gefi || gefiW) + break; + } + + // if no one claimed it, then check if it's the currently playing track + // benski> TODO: there is a race condition here. In theory, in_mod could change between the lstrcmpiW and the assignment + if (!i && !lstrcmpiW(FileName, AutoWide(fn))) + { + i=in_mod; + if (i) + { + cachedExt[0]=0; + gefi = (GetANSIInfo)GetProcAddress(i->hDllInstance, "winampGetExtendedFileInfo"); + gefiW = (GetUnicodeInfo)GetProcAddress(i->hDllInstance, "winampGetExtendedFileInfoW"); + } + } + + if (!i) + return 0; + + if (gefi) + return gefi(fn, metadata, dest, destlen); + else if (gefiW) + return ConvertGetExtendedFileInfo(gefiW, fn, metadata, dest, destlen); + else + return 0; +} + +int in_get_extended_fileinfoW(const wchar_t *fn, const wchar_t *metadata, wchar_t *dest, size_t destlen) +{ + AutoLock lock(getMetadataGuard); + + In_Module *i=0; + char ext[16] = {0}; + int a = 0; + + // do some extra checks on the params as quite a few clients crash when accessing here + try { + if (destlen > 65536 || destlen == 0 || !fn || (unsigned int)(ULONG_PTR)fn < 65536 || + fn && !*fn || !metadata || (unsigned int)(ULONG_PTR)metadata < 65536 || + metadata && !*metadata || !dest || (unsigned int)(ULONG_PTR)dest < 65536) + return 0; + } catch (...) { + return 0; + } + + if (dest) + memset(dest, 0, destlen); + + AutoCharFn charFn(fn); + extension_ex(charFn, ext, sizeof(ext)); + + if (!_stricmp(cachedExt, ext) && *ext) + { + if (gefiW) // should always be true if we got this far + return gefiW(fn, AutoChar(metadata), dest, destlen); + if (gefi) + return ConvertGetExtendedFileInfo(gefi, charFn, AutoChar(metadata), dest, destlen); + else + return 0; + } + + while (a >= 0) + { + i = in_setmod_noplay(fn, &a); + if (!i) + break; + + if (a >= 0) a++; + + lstrcpynA(cachedExt, ext, sizeof(cachedExt)/sizeof(*cachedExt)); + + gefi = (GetANSIInfo)GetProcAddress(i->hDllInstance, "winampGetExtendedFileInfo"); + gefiW = (GetUnicodeInfo)GetProcAddress(i->hDllInstance, "winampGetExtendedFileInfoW"); + if (gefi || gefiW) + break; + } + + // if no one claimed it, then check if it's the currently playing track + // benski> TODO: there is a race condition here. In theory, in_mod could change between the lstrcmpiW and the assignment + if (!i && !lstrcmpiW(FileName, fn)) // currently playing file? + { + i=in_mod; + if (i) + { + cachedExt[0]=0; + gefi = (GetANSIInfo)GetProcAddress(i->hDllInstance, "winampGetExtendedFileInfo"); + gefiW = (GetUnicodeInfo)GetProcAddress(i->hDllInstance, "winampGetExtendedFileInfoW"); + } + } + + if (!i) + return 0; + + if (gefiW) + return gefiW(fn, AutoChar(metadata), dest, destlen); + else if (gefi) + return ConvertGetExtendedFileInfo(gefi, charFn, AutoChar(metadata), dest, destlen); + else + return 0; +} + +/* Set Extended File Info */ + +typedef int (__cdecl *SetANSIInfo)(const char *fn, const char *metadata, const char *data); +typedef int (__cdecl *SetUnicodeInfo)(const wchar_t *fn, const char *metadata, const wchar_t *data); + +static SetANSIInfo sefi; +static SetUnicodeInfo sefiW; +static int (*wefi)(); +static char cachedExtSet[16]; + +static int ConvertSetExtendedFileInfo(SetUnicodeInfo setter, const char *fn, const char *metadata, const char *data) +{ + return setter(AutoWide(fn), metadata, AutoWide(data)); +} + +static int ConvertSetExtendedFileInfo(SetANSIInfo setter, const char *fn, const char *metadata, const wchar_t *data) +{ + return setter(fn, metadata, AutoChar(data)); +} + +int in_set_extended_fileinfo(const char *fn, const char *metadata, char *data) +{ + AutoLock lock(getMetadataGuard); + char ext[16] = {0}; + int a = 0; + + extension_ex(fn, ext, sizeof(ext)); + if (!_stricmp(cachedExtSet, ext) && *ext) + { + if (sefi) + return sefi(fn, metadata, data); + else if (sefiW) + return ConvertSetExtendedFileInfo(sefiW, fn, metadata, data); + else + return 0; + } + + while (a >= 0) + { + wefi = NULL; + sefi = NULL; // if we fail finding an input plugin, dont let in_write_extended_fileinfo work :) + sefiW = NULL; + In_Module *i = in_setmod_noplay(AutoWideFn(fn), &a); + if (a >= 0) a++; + cachedExtSet[0] = 0; + if (!i) return 0; + lstrcpynA(cachedExtSet, ext, 16); + + sefi = (SetANSIInfo)GetProcAddress(i->hDllInstance, "winampSetExtendedFileInfo"); + sefiW = (SetUnicodeInfo)GetProcAddress(i->hDllInstance, "winampSetExtendedFileInfoW"); + wefi = (int (__cdecl *)())GetProcAddress(i->hDllInstance, "winampWriteExtendedFileInfo"); + if (sefi || sefiW) break; + } + + if (sefi) + return sefi(fn, metadata, data); + else if (sefiW) + return ConvertSetExtendedFileInfo(sefiW, fn, metadata, data); + else + return 0; +} + +int in_set_extended_fileinfoW(const wchar_t *fn, const wchar_t *metadata, wchar_t *data) +{ + AutoLock lock(getMetadataGuard); + + char ext[16] = {0}; + int a = 0; + + AutoCharFn charFn(fn); + extension_ex(charFn, ext, sizeof(ext)); + if (!_stricmp(cachedExtSet, ext) && *ext) + { + if (sefiW) + return sefiW(fn, AutoChar(metadata), data); + else if (sefi) + return ConvertSetExtendedFileInfo(sefi, charFn, AutoChar(metadata), data); + else + return 0; + } + + while (a >= 0) + { + wefi = NULL; + sefi = NULL; // if we fail finding an input plugin, dont let in_write_extended_fileinfo work :) + sefiW = NULL; + In_Module *i = in_setmod_noplay(fn, &a); + if (a >= 0) a++; + cachedExtSet[0] = 0; + if (!i) return 0; + lstrcpynA(cachedExtSet, ext, 16); + + sefi = (SetANSIInfo)GetProcAddress(i->hDllInstance, "winampSetExtendedFileInfo"); + sefiW = (SetUnicodeInfo)GetProcAddress(i->hDllInstance, "winampSetExtendedFileInfoW"); + wefi = (int (__cdecl *)())GetProcAddress(i->hDllInstance, "winampWriteExtendedFileInfo"); + if (sefi || sefiW) break; + } + if (sefiW) + return sefiW(fn, AutoChar(metadata), data); + else if (sefi) + return ConvertSetExtendedFileInfo(sefi, charFn, AutoChar(metadata), data); + else + return 0; +} + +int in_write_extended_fileinfo() +{ + AutoLock lock(getMetadataGuard); + if (!wefi) + return 0; + return + wefi(); +} + +inline void COPY_METADATA(const wchar_t *src, const wchar_t *dest, const wchar_t *item, wchar_t *buf, size_t buflen) +{ + if (NULL != src && NULL != item && NULL != buf) + { + buf[0]=0; + extendedFileInfoStructW efis= + { + src, + item, + buf, + buflen, + }; + + if (SendMessageW(hMainWindow,WM_WA_IPC,(WPARAM)&efis,IPC_GET_EXTENDED_FILE_INFOW_HOOKABLE)) + in_set_extended_fileinfoW(dest, item, buf); + } + +} + + +//#define COPY_METADATA(src, dest, item, buf, buflen) { buf[0]=0; if (in_get_extended_fileinfoW(src, item, buf, buflen)) in_set_extended_fileinfoW(dest, item, buf); } +void CopyExtendedFileInfo(const wchar_t *source, const wchar_t *destination) +{ + wchar_t bigData[32768] = {0}; // hopefully big enough for all reasonable metadata + + COPY_METADATA(source, destination, L"title", bigData, 32768); + COPY_METADATA(source, destination, L"artist", bigData, 32768); + COPY_METADATA(source, destination, L"albumartist", bigData, 32768); + COPY_METADATA(source, destination, L"album", bigData, 32768); + COPY_METADATA(source, destination, L"genre", bigData, 32768); + COPY_METADATA(source, destination, L"year", bigData, 32768); + COPY_METADATA(source, destination, L"disc", bigData, 32768); + COPY_METADATA(source, destination, L"publisher", bigData, 32768); + COPY_METADATA(source, destination, L"comment", bigData, 32768); + COPY_METADATA(source, destination, L"track", bigData, 32768); + COPY_METADATA(source, destination, L"tool", bigData, 32768); + COPY_METADATA(source, destination, L"composer", bigData, 32768); + COPY_METADATA(source, destination, L"conductor", bigData, 32768); + COPY_METADATA(source, destination, L"bpm", bigData, 32768); + COPY_METADATA(source, destination, L"GracenoteFileID", bigData, 32768); + COPY_METADATA(source, destination, L"GracenoteExtData", bigData, 32768); + in_write_extended_fileinfo(); + + //albumArt->CopyAlbumArt(source, destination); +}
\ No newline at end of file |