From a4dd0ad63c00f4dee3b86dfd3075d1d61b2b3180 Mon Sep 17 00:00:00 2001 From: sanine Date: Sat, 27 Aug 2022 23:52:56 -0500 Subject: add plibsys --- 3rdparty/plibsys/src/pdir-os2.c | 381 ++++++++++++++++++++++++++++++++++++++++ 1 file changed, 381 insertions(+) create mode 100644 3rdparty/plibsys/src/pdir-os2.c (limited to '3rdparty/plibsys/src/pdir-os2.c') diff --git a/3rdparty/plibsys/src/pdir-os2.c b/3rdparty/plibsys/src/pdir-os2.c new file mode 100644 index 0000000..8869329 --- /dev/null +++ b/3rdparty/plibsys/src/pdir-os2.c @@ -0,0 +1,381 @@ +/* + * The MIT License + * + * Copyright (C) 2017 Alexander Saprykin + * + * Permission is hereby granted, free of charge, to any person obtaining + * a copy of this software and associated documentation files (the + * 'Software'), to deal in the Software without restriction, including + * without limitation the rights to use, copy, modify, merge, publish, + * distribute, sublicense, and/or sell copies of the Software, and to + * permit persons to whom the Software is furnished to do so, subject to + * the following conditions: + * + * The above copyright notice and this permission notice shall be + * included in all copies or substantial portions of the Software. + * + * THE SOFTWARE IS PROVIDED 'AS IS', WITHOUT WARRANTY OF ANY KIND, + * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF + * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. + * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY + * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, + * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE + * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. + */ + +#include "pdir.h" +#include "perror.h" +#include "pmem.h" +#include "pstring.h" +#include "perror-private.h" + +#define INCL_DOSFILEMGR +#define INCL_DOSERRORS +#include + +#include +#include + +struct PDir_ { + FILEFINDBUF3 find_data; + HDIR search_handle; + pboolean cached; + pchar path[CCHMAXPATH]; + pchar *orig_path; +}; + +P_LIB_API PDir * +p_dir_new (const pchar *path, + PError **error) +{ + PDir *ret; + pchar *pathp; + pchar *adj_path; + pint path_len; + APIRET ulrc; + ULONG find_count; + + if (P_UNLIKELY (path == NULL)) { + p_error_set_error_p (error, + (pint) P_ERROR_IO_INVALID_ARGUMENT, + 0, + "Invalid input argument"); + return NULL; + } + + if (P_UNLIKELY ((ret = p_malloc0 (sizeof (PDir))) == NULL)) { + p_error_set_error_p (error, + (pint) P_ERROR_IO_NO_RESOURCES, + 0, + "Failed to allocate memory for directory structure"); + return NULL; + } + + adj_path = NULL; + path_len = strlen (path); + + if (P_UNLIKELY (path[path_len - 1] == '\\' || path[path_len - 1] == '/') && path_len > 1) { + while ((path[path_len - 1] == '\\' || path[path_len - 1] == '/') && path_len > 1) + --path_len; + + if (P_UNLIKELY ((adj_path = p_malloc0 (path_len + 1)) == NULL)) { + p_free (ret); + p_error_set_error_p (error, + (pint) P_ERROR_IO_NO_RESOURCES, + 0, + "Failed to allocate memory for directory path"); + return NULL; + } + + memcpy (adj_path, path, path_len); + + adj_path[path_len] = '\0'; + ret->orig_path = p_strdup (path); + path = (const pchar *) adj_path; + } + + ret->search_handle = HDIR_CREATE; + + ulrc = DosQueryPathInfo ((PSZ) path, FIL_QUERYFULLNAME, ret->path, sizeof (ret->path) - 2); + + if (P_UNLIKELY (ulrc != NO_ERROR)) { + p_error_set_error_p (error, + (pint) p_error_get_io_from_system ((pint) ulrc), + (pint) ulrc, + "Failed to call DosQueryPathInfo() to get directory path"); + + if (P_UNLIKELY (adj_path != NULL)) { + p_free (adj_path); + p_free (ret->orig_path); + } + + p_free (ret); + return NULL; + } + + /* Append the search pattern "\\*\0" to the directory name */ + pathp = strchr (ret->path, '\0'); + + if (ret->path < pathp && *(pathp - 1) != '\\' && *(pathp - 1) != ':') + *pathp++ = '\\'; + + *pathp++ = '*'; + *pathp = '\0'; + + find_count = 1; + + /* Open directory stream and retrieve the first entry */ + ulrc = DosFindFirst (ret->path, + &ret->search_handle, + FILE_NORMAL | FILE_DIRECTORY, + &ret->find_data, + sizeof (ret->find_data), + &find_count, + FIL_STANDARD); + + if (P_UNLIKELY (ulrc != NO_ERROR && ulrc != ERROR_NO_MORE_FILES)) { + p_error_set_error_p (error, + (pint) p_error_get_io_from_system ((pint) ulrc), + (pint) ulrc, + "Failed to call DosFindFirst() to open directory stream"); + + if (P_UNLIKELY (adj_path != NULL)) { + p_free (adj_path); + p_free (ret->orig_path); + } + + p_free (ret); + return NULL; + } + + ret->cached = TRUE; + + if (P_UNLIKELY (adj_path != NULL)) + p_free (adj_path); + else + ret->orig_path = p_strdup (path); + + return ret; +} + +P_LIB_API pboolean +p_dir_create (const pchar *path, + pint mode, + PError **error) +{ + APIRET ulrc; + + P_UNUSED (mode); + + if (P_UNLIKELY (path == NULL)) { + p_error_set_error_p (error, + (pint) P_ERROR_IO_INVALID_ARGUMENT, + 0, + "Invalid input argument"); + return FALSE; + } + + if (p_dir_is_exists (path)) + return TRUE; + + if (P_UNLIKELY ((ulrc = DosCreateDir ((PSZ) path, NULL)) != NO_ERROR)) { + p_error_set_error_p (error, + (pint) p_error_get_io_from_system ((pint) ulrc), + (pint) ulrc, + "Failed to call DosCreateDir() to create directory"); + return FALSE; + } else + return TRUE; +} + +P_LIB_API pboolean +p_dir_remove (const pchar *path, + PError **error) +{ + APIRET ulrc; + + if (P_UNLIKELY (path == NULL)) { + p_error_set_error_p (error, + (pint) P_ERROR_IO_INVALID_ARGUMENT, + 0, + "Invalid input argument"); + return FALSE; + } + + if (!p_dir_is_exists (path)) { + p_error_set_error_p (error, + (pint) P_ERROR_IO_NOT_EXISTS, + 0, + "Specified directory doesn't exist"); + return FALSE; + } + + if (P_UNLIKELY ((ulrc = DosDeleteDir ((PSZ) path)) != NO_ERROR)) { + p_error_set_error_p (error, + (pint) p_error_get_io_from_system ((pint) ulrc), + (pint) ulrc, + "Failed to call DosDeleteDir() to remove directory"); + return FALSE; + } else + return TRUE; +} + +P_LIB_API pboolean +p_dir_is_exists (const pchar *path) +{ + FILESTATUS3 status; + + if (P_UNLIKELY (path == NULL)) + return FALSE; + + if (DosQueryPathInfo ((PSZ) path, FIL_STANDARD, (PVOID) &status, sizeof (status)) != NO_ERROR) + return FALSE; + + return (status.attrFile & FILE_DIRECTORY) != 0; +} + +P_LIB_API pchar * +p_dir_get_path (const PDir *dir) +{ + if (P_UNLIKELY (dir == NULL)) + return NULL; + + return p_strdup (dir->orig_path); +} + +P_LIB_API PDirEntry * +p_dir_get_next_entry (PDir *dir, + PError **error) +{ + PDirEntry *ret; + APIRET ulrc; + ULONG find_count; + + if (P_UNLIKELY (dir == NULL)) { + p_error_set_error_p (error, + (pint) P_ERROR_IO_INVALID_ARGUMENT, + 0, + "Invalid input argument"); + return NULL; + } + + if (dir->cached == TRUE) { + dir->cached = FALSE; + + /* Opened directory is empty */ + if (P_UNLIKELY (dir->search_handle == HDIR_CREATE)) { + p_error_set_error_p (error, + (pint) P_ERROR_IO_NO_MORE, + (pint) ERROR_NO_MORE_FILES, + "Directory is empty to get the next entry"); + return NULL; + } + } else { + if (P_UNLIKELY (dir->search_handle == HDIR_CREATE)) { + p_error_set_error_p (error, + (pint) P_ERROR_IO_INVALID_ARGUMENT, + 0, + "Not a valid (or closed) directory stream"); + return NULL; + } + + find_count = 1; + + ulrc = DosFindNext (dir->search_handle, + (PVOID) &dir->find_data, + sizeof (dir->find_data), + &find_count); + + if (P_UNLIKELY (ulrc != NO_ERROR)) { + p_error_set_error_p (error, + (pint) p_error_get_io_from_system ((pint) ulrc), + (pint) ulrc, + "Failed to call DosFindNext() to read directory stream"); + DosFindClose (dir->search_handle); + dir->search_handle = HDIR_CREATE; + return NULL; + } + } + + if (P_UNLIKELY ((ret = p_malloc0 (sizeof (PDirEntry))) == NULL)) { + p_error_set_error_p (error, + (pint) P_ERROR_IO_NO_RESOURCES, + 0, + "Failed to allocate memory for directory entry"); + return NULL; + } + + ret->name = p_strdup (dir->find_data.achName); + + if ((dir->find_data.attrFile & FILE_DIRECTORY) != 0) + ret->type = P_DIR_ENTRY_TYPE_DIR; + else + ret->type = P_DIR_ENTRY_TYPE_FILE; + + return ret; +} + +P_LIB_API pboolean +p_dir_rewind (PDir *dir, + PError **error) +{ + APIRET ulrc; + ULONG find_count; + + if (P_UNLIKELY (dir == NULL)) { + p_error_set_error_p (error, + (pint) P_ERROR_IO_INVALID_ARGUMENT, + 0, + "Invalid input argument"); + return FALSE; + } + + if (dir->search_handle != HDIR_CREATE) { + if (P_UNLIKELY ((ulrc = DosFindClose (dir->search_handle)) != NO_ERROR)) { + p_error_set_error_p (error, + (pint) p_error_get_io_from_system ((pint) ulrc), + (pint) ulrc, + "Failed to call DosFindClose() to close directory stream"); + return FALSE; + } + + dir->search_handle = HDIR_CREATE; + } + + find_count = 1; + + ulrc = DosFindFirst (dir->path, + &dir->search_handle, + FILE_NORMAL | FILE_DIRECTORY, + &dir->find_data, + sizeof (dir->find_data), + &find_count, + FIL_STANDARD); + + if (P_UNLIKELY (ulrc != NO_ERROR && ulrc != ERROR_NO_MORE_FILES)) { + p_error_set_error_p (error, + (pint) p_error_get_io_from_system ((pint) ulrc), + (pint) ulrc, + "Failed to call DosFindFirst() to open directory stream"); + dir->cached = FALSE; + return FALSE; + } else { + dir->cached = TRUE; + return TRUE; + } +} + +P_LIB_API void +p_dir_free (PDir *dir) +{ + if (dir == NULL) + return; + + if (P_LIKELY (dir->search_handle != HDIR_CREATE)) { + if (P_UNLIKELY (DosFindClose (dir->search_handle) != NO_ERROR)) + P_ERROR ("PDir::p_dir_free: DosFindClose() failed"); + } + + p_free (dir->orig_path); + p_free (dir); +} -- cgit v1.2.1