mirror of
https://github.com/mpv-player/mpv
synced 2024-12-25 16:33:02 +00:00
0cef033d48
glob-win.c wasn't big, so it was easier to rewrite it. The new version supports Unicode, handles directories properly, sorts the output and puts all its allocations in the same talloc context to simplify the implementation of globfree. Notably, the old glob had error checking code, but didn't do anything with the errors since the error reporting code was commented out. The new glob doesn't copy this behaviour. It just treats errors as if there were no more matching files, which shouldn't matter for mpv, since it ignores glob errors too. To match the other Windows I/O helper functions, the definition is moved to osdep/io.h.
156 lines
5.1 KiB
C
156 lines
5.1 KiB
C
/*
|
|
* This file is part of mpv.
|
|
*
|
|
* mpv is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* mpv is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along
|
|
* with mpv. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <windows.h>
|
|
#include <stdbool.h>
|
|
#include <string.h>
|
|
#include "osdep/io.h"
|
|
#include "talloc.h"
|
|
|
|
static wchar_t *talloc_wcsdup(void *ctx, const wchar_t *wcs)
|
|
{
|
|
size_t len = (wcslen(wcs) + 1) * sizeof(wchar_t);
|
|
return talloc_memdup(ctx, (void*)wcs, len);
|
|
}
|
|
|
|
static int compare_wcscoll(const void *v1, const void *v2)
|
|
{
|
|
wchar_t * const* p1 = v1;
|
|
wchar_t * const* p2 = v2;
|
|
return wcscoll(*p1, *p2);
|
|
}
|
|
|
|
static bool exists(const char *filename)
|
|
{
|
|
wchar_t *wfilename = mp_from_utf8(NULL, filename);
|
|
bool result = GetFileAttributesW(wfilename) != INVALID_FILE_ATTRIBUTES;
|
|
talloc_free(wfilename);
|
|
return result;
|
|
}
|
|
|
|
int mp_glob(const char *restrict pattern, int flags,
|
|
int (*errfunc)(const char*, int), mp_glob_t *restrict pglob)
|
|
{
|
|
// This glob implementation never calls errfunc and doesn't understand any
|
|
// flags. These features are currently unused in mpv, however if new code
|
|
// were to use these them, it would probably break on Windows.
|
|
|
|
unsigned dirlen = 0;
|
|
bool wildcards = false;
|
|
|
|
// Check for drive relative paths eg. "C:*.flac"
|
|
if (pattern[0] != '\0' && pattern[1] == ':')
|
|
dirlen = 2;
|
|
|
|
// Split the directory and filename. All files returned by FindFirstFile
|
|
// will be in this directory. Also check the filename for wildcards.
|
|
for (unsigned i = 0; pattern[i]; i ++) {
|
|
if (pattern[i] == '?' || pattern[i] == '*')
|
|
wildcards = true;
|
|
|
|
if (pattern[i] == '\\' || pattern[i] == '/') {
|
|
dirlen = i + 1;
|
|
wildcards = false;
|
|
}
|
|
}
|
|
|
|
// FindFirstFile is unreliable with certain input (it returns weird results
|
|
// with paths like "." and "..", and presumably others.) If there are no
|
|
// wildcards in the filename, don't call it, just check if the file exists.
|
|
// The CRT globbing code does this too.
|
|
if (!wildcards) {
|
|
if (!exists(pattern)) {
|
|
pglob->gl_pathc = 0;
|
|
return GLOB_NOMATCH;
|
|
}
|
|
|
|
pglob->ctx = talloc_new(NULL);
|
|
pglob->gl_pathc = 1;
|
|
pglob->gl_pathv = talloc_array_ptrtype(pglob->ctx, pglob->gl_pathv, 2);
|
|
pglob->gl_pathv[0] = talloc_strdup(pglob->ctx, pattern);
|
|
pglob->gl_pathv[1] = NULL;
|
|
return 0;
|
|
}
|
|
|
|
wchar_t *wpattern = mp_from_utf8(NULL, pattern);
|
|
WIN32_FIND_DATAW data;
|
|
HANDLE find = FindFirstFileW(wpattern, &data);
|
|
talloc_free(wpattern);
|
|
|
|
// Assume an error means there were no matches. mpv doesn't check for
|
|
// glob() errors, so this should be fine for now.
|
|
if (find == INVALID_HANDLE_VALUE) {
|
|
pglob->gl_pathc = 0;
|
|
return GLOB_NOMATCH;
|
|
}
|
|
|
|
size_t pathc = 0;
|
|
void *tmp = talloc_new(NULL);
|
|
wchar_t **wnamev = NULL;
|
|
|
|
// Read a list of filenames. Unlike glob(), FindFirstFile doesn't return
|
|
// the full path, since all files are relative to the directory specified
|
|
// in the pattern.
|
|
do {
|
|
if (!wcscmp(data.cFileName, L".") || !wcscmp(data.cFileName, L".."))
|
|
continue;
|
|
|
|
wchar_t *wname = talloc_wcsdup(tmp, data.cFileName);
|
|
MP_TARRAY_APPEND(tmp, wnamev, pathc, wname);
|
|
} while (FindNextFileW(find, &data));
|
|
FindClose(find);
|
|
|
|
if (!wnamev) {
|
|
talloc_free(tmp);
|
|
pglob->gl_pathc = 0;
|
|
return GLOB_NOMATCH;
|
|
}
|
|
|
|
// POSIX glob() is supposed to sort paths according to LC_COLLATE.
|
|
// FindFirstFile just returns paths in the order they are read from the
|
|
// directory, so sort them manually with wcscoll.
|
|
qsort(wnamev, pathc, sizeof(wchar_t*), compare_wcscoll);
|
|
|
|
pglob->ctx = talloc_new(NULL);
|
|
pglob->gl_pathc = pathc;
|
|
pglob->gl_pathv = talloc_array_ptrtype(pglob->ctx, pglob->gl_pathv,
|
|
pathc + 1);
|
|
|
|
// Now convert all filenames to UTF-8 (they had to be in UTF-16 for
|
|
// sorting) and prepend the directory
|
|
for (unsigned i = 0; i < pathc; i ++) {
|
|
int namelen = WideCharToMultiByte(CP_UTF8, 0, wnamev[i], -1, NULL, 0,
|
|
NULL, NULL);
|
|
char *path = talloc_array(pglob->ctx, char, namelen + dirlen);
|
|
|
|
memcpy(path, pattern, dirlen);
|
|
WideCharToMultiByte(CP_UTF8, 0, wnamev[i], -1, path + dirlen,
|
|
namelen, NULL, NULL);
|
|
pglob->gl_pathv[i] = path;
|
|
}
|
|
|
|
// gl_pathv must be null terminated
|
|
pglob->gl_pathv[pathc] = NULL;
|
|
talloc_free(tmp);
|
|
return 0;
|
|
}
|
|
|
|
void mp_globfree(mp_glob_t *pglob)
|
|
{
|
|
talloc_free(pglob->ctx);
|
|
}
|