mirror of
https://github.com/samba-team/samba.git
synced 2024-12-25 23:21:54 +03:00
4a8cfe1650
Signed-off-by: Volker Lendecke <vl@samba.org> Reviewed-by: Jeremy Allison <jra@samba.org> Autobuild-User(master): Jeremy Allison <jra@samba.org> Autobuild-Date(master): Fri Jun 23 18:29:40 UTC 2023 on atb-devel-224
268 lines
6.7 KiB
C
268 lines
6.7 KiB
C
/*
|
|
* VFS module to provide a sorted directory list.
|
|
*
|
|
* Copyright (C) Andy Kelk (andy@mopoke.co.uk), 2009
|
|
*
|
|
*
|
|
* This program 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 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program 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 this program; if not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "includes.h"
|
|
#include "smbd/smbd.h"
|
|
#include "system/filesys.h"
|
|
|
|
static int compare_dirent (const struct dirent *da, const struct dirent *db)
|
|
{
|
|
return strcasecmp_m(da->d_name, db->d_name);
|
|
}
|
|
|
|
struct dirsort_privates {
|
|
struct dirsort_privates *prev, *next;
|
|
long pos;
|
|
struct dirent *directory_list;
|
|
unsigned int number_of_entries;
|
|
struct timespec mtime;
|
|
DIR *source_directory;
|
|
files_struct *fsp; /* If open via FDOPENDIR. */
|
|
struct smb_filename *smb_fname; /* If open via OPENDIR */
|
|
};
|
|
|
|
static bool get_sorted_dir_mtime(vfs_handle_struct *handle,
|
|
struct dirsort_privates *data,
|
|
struct timespec *ret_mtime)
|
|
{
|
|
int ret;
|
|
struct timespec mtime;
|
|
NTSTATUS status;
|
|
|
|
if (data->fsp) {
|
|
status = vfs_stat_fsp(data->fsp);
|
|
if (!NT_STATUS_IS_OK(status)) {
|
|
return false;
|
|
}
|
|
mtime = data->fsp->fsp_name->st.st_ex_mtime;
|
|
} else {
|
|
ret = SMB_VFS_STAT(handle->conn, data->smb_fname);
|
|
if (ret == -1) {
|
|
return false;
|
|
}
|
|
mtime = data->smb_fname->st.st_ex_mtime;
|
|
}
|
|
|
|
*ret_mtime = mtime;
|
|
|
|
return true;
|
|
}
|
|
|
|
static bool open_and_sort_dir(vfs_handle_struct *handle,
|
|
struct dirsort_privates *data)
|
|
{
|
|
uint32_t total_count = 0;
|
|
/* This should be enough for most use cases */
|
|
uint32_t dirent_allocated = 64;
|
|
struct dirent *dp;
|
|
|
|
data->number_of_entries = 0;
|
|
|
|
if (get_sorted_dir_mtime(handle, data, &data->mtime) == false) {
|
|
return false;
|
|
}
|
|
|
|
dp = SMB_VFS_NEXT_READDIR(handle, data->fsp, data->source_directory);
|
|
if (dp == NULL) {
|
|
return false;
|
|
}
|
|
|
|
/* Set up an array and read the directory entries into it */
|
|
TALLOC_FREE(data->directory_list); /* destroy previous cache if needed */
|
|
data->directory_list = talloc_zero_array(data,
|
|
struct dirent,
|
|
dirent_allocated);
|
|
if (data->directory_list == NULL) {
|
|
return false;
|
|
}
|
|
|
|
do {
|
|
if (total_count >= dirent_allocated) {
|
|
struct dirent *dlist;
|
|
|
|
/*
|
|
* Be memory friendly.
|
|
*
|
|
* We should not double the amount of memory. With a lot
|
|
* of files we reach easily 50MB, and doubling will
|
|
* get much bigger just for a few files more.
|
|
*
|
|
* For 200k files this means 50 memory reallocations.
|
|
*/
|
|
dirent_allocated += 4096;
|
|
|
|
dlist = talloc_realloc(data,
|
|
data->directory_list,
|
|
struct dirent,
|
|
dirent_allocated);
|
|
if (dlist == NULL) {
|
|
break;
|
|
}
|
|
data->directory_list = dlist;
|
|
}
|
|
data->directory_list[total_count] = *dp;
|
|
|
|
total_count++;
|
|
dp = SMB_VFS_NEXT_READDIR(handle,
|
|
data->fsp,
|
|
data->source_directory);
|
|
} while (dp != NULL);
|
|
|
|
data->number_of_entries = total_count;
|
|
|
|
/* Sort the directory entries by name */
|
|
TYPESAFE_QSORT(data->directory_list, data->number_of_entries, compare_dirent);
|
|
return true;
|
|
}
|
|
|
|
static DIR *dirsort_fdopendir(vfs_handle_struct *handle,
|
|
files_struct *fsp,
|
|
const char *mask,
|
|
uint32_t attr)
|
|
{
|
|
struct dirsort_privates *list_head = NULL;
|
|
struct dirsort_privates *data = NULL;
|
|
|
|
if (SMB_VFS_HANDLE_TEST_DATA(handle)) {
|
|
/* Find the list head of all open directories. */
|
|
SMB_VFS_HANDLE_GET_DATA(handle, list_head, struct dirsort_privates,
|
|
return NULL);
|
|
}
|
|
|
|
/* set up our private data about this directory */
|
|
data = talloc_zero(handle->conn, struct dirsort_privates);
|
|
if (!data) {
|
|
return NULL;
|
|
}
|
|
|
|
data->fsp = fsp;
|
|
|
|
/* Open the underlying directory and count the number of entries */
|
|
data->source_directory = SMB_VFS_NEXT_FDOPENDIR(handle, fsp, mask,
|
|
attr);
|
|
|
|
if (data->source_directory == NULL) {
|
|
TALLOC_FREE(data);
|
|
return NULL;
|
|
}
|
|
|
|
if (!open_and_sort_dir(handle, data)) {
|
|
SMB_VFS_NEXT_CLOSEDIR(handle,data->source_directory);
|
|
TALLOC_FREE(data);
|
|
/* fd is now closed. */
|
|
fsp_set_fd(fsp, -1);
|
|
return NULL;
|
|
}
|
|
|
|
/* Add to the private list of all open directories. */
|
|
DLIST_ADD(list_head, data);
|
|
SMB_VFS_HANDLE_SET_DATA(handle, list_head, NULL,
|
|
struct dirsort_privates, return NULL);
|
|
|
|
return data->source_directory;
|
|
}
|
|
|
|
static struct dirent *dirsort_readdir(vfs_handle_struct *handle,
|
|
struct files_struct *dirfsp,
|
|
DIR *dirp)
|
|
{
|
|
struct dirsort_privates *data = NULL;
|
|
struct timespec current_mtime;
|
|
|
|
SMB_VFS_HANDLE_GET_DATA(handle, data, struct dirsort_privates,
|
|
return NULL);
|
|
|
|
while(data && (data->source_directory != dirp)) {
|
|
data = data->next;
|
|
}
|
|
if (data == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
if (get_sorted_dir_mtime(handle, data, ¤t_mtime) == false) {
|
|
return NULL;
|
|
}
|
|
|
|
/* throw away cache and re-read the directory if we've changed */
|
|
if (timespec_compare(¤t_mtime, &data->mtime)) {
|
|
SMB_VFS_NEXT_REWINDDIR(handle, data->source_directory);
|
|
open_and_sort_dir(handle, data);
|
|
}
|
|
|
|
if (data->pos >= data->number_of_entries) {
|
|
return NULL;
|
|
}
|
|
|
|
return &data->directory_list[data->pos++];
|
|
}
|
|
|
|
static void dirsort_rewinddir(vfs_handle_struct *handle, DIR *dirp)
|
|
{
|
|
struct dirsort_privates *data = NULL;
|
|
SMB_VFS_HANDLE_GET_DATA(handle, data, struct dirsort_privates, return);
|
|
|
|
/* Find the entry holding dirp. */
|
|
while(data && (data->source_directory != dirp)) {
|
|
data = data->next;
|
|
}
|
|
if (data == NULL) {
|
|
return;
|
|
}
|
|
data->pos = 0;
|
|
}
|
|
|
|
static int dirsort_closedir(vfs_handle_struct *handle, DIR *dirp)
|
|
{
|
|
struct dirsort_privates *list_head = NULL;
|
|
struct dirsort_privates *data = NULL;
|
|
int ret;
|
|
|
|
SMB_VFS_HANDLE_GET_DATA(handle, list_head, struct dirsort_privates, return -1);
|
|
/* Find the entry holding dirp. */
|
|
for(data = list_head; data && (data->source_directory != dirp); data = data->next) {
|
|
;
|
|
}
|
|
if (data == NULL) {
|
|
return -1;
|
|
}
|
|
/* Remove from the list and re-store the list head. */
|
|
DLIST_REMOVE(list_head, data);
|
|
SMB_VFS_HANDLE_SET_DATA(handle, list_head, NULL,
|
|
struct dirsort_privates, return -1);
|
|
|
|
ret = SMB_VFS_NEXT_CLOSEDIR(handle, dirp);
|
|
TALLOC_FREE(data);
|
|
return ret;
|
|
}
|
|
|
|
static struct vfs_fn_pointers vfs_dirsort_fns = {
|
|
.fdopendir_fn = dirsort_fdopendir,
|
|
.readdir_fn = dirsort_readdir,
|
|
.rewind_dir_fn = dirsort_rewinddir,
|
|
.closedir_fn = dirsort_closedir,
|
|
};
|
|
|
|
static_decl_vfs;
|
|
NTSTATUS vfs_dirsort_init(TALLOC_CTX *ctx)
|
|
{
|
|
return smb_register_vfs(SMB_VFS_INTERFACE_VERSION, "dirsort",
|
|
&vfs_dirsort_fns);
|
|
}
|