365 lines
6.7 KiB
C
365 lines
6.7 KiB
C
|
// SPDX-License-Identifier: LGPL-3.0-or-later
|
||
|
|
|
||
|
|
#include "sys/sys_local.h"
|
||
|
|
#include "sys/fs.h"
|
||
|
|
|
||
|
|
#include <sys/types.h>
|
||
|
|
#include <sys/stat.h>
|
||
|
|
#include <dirent.h>
|
||
|
|
#include <errno.h>
|
||
|
|
#include <fcntl.h>
|
||
|
|
#include <libgen.h>
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <string.h>
|
||
|
|
#include <unistd.h>
|
||
|
|
|
||
|
|
Fildes Sys_Fopen(const char *path, FopenMode mode, unsigned flags)
|
||
|
|
{
|
||
|
|
Fildes fd;
|
||
|
|
|
||
|
|
errno = 0;
|
||
|
|
|
||
|
|
// Open file
|
||
|
|
switch (mode) {
|
||
|
|
case FM_READ:
|
||
|
|
fd = open(path, O_RDONLY);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case FM_WRITE:
|
||
|
|
fd = open(path, O_WRONLY | O_CREAT | O_TRUNC, 0666);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case FM_APPEND:
|
||
|
|
fd = open(path, O_WRONLY | O_CREAT | O_APPEND, 0666);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case FM_EXCL:
|
||
|
|
fd = open(path, O_WRONLY | O_CREAT | O_EXCL, 0666);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case FM_TEMP: {
|
||
|
|
#ifdef __linux__
|
||
|
|
// Linux supports O_TMPFILE for safer temporary file creation
|
||
|
|
fd = open(path, O_WRONLY | O_EXCL | O_TMPFILE, 0600);
|
||
|
|
if (fd >= 0)
|
||
|
|
break; // success
|
||
|
|
|
||
|
|
if (errno == ENOENT || errno == ENOTDIR || errno == EISDIR)
|
||
|
|
break; // file not found - don't fallback
|
||
|
|
if (errno != EOPNOTSUPP)
|
||
|
|
break; // error - don't fallback
|
||
|
|
|
||
|
|
// Fallback to regular mkstemp()
|
||
|
|
#endif
|
||
|
|
size_t n = strlen(path);
|
||
|
|
char *buf = (char *) alloca(n + 1 + 6 + 1);
|
||
|
|
if (sprintf(buf, "%s/XXXXXX", path) < 0) {
|
||
|
|
Sys_SetErrStat(errno, "sprintf() failed");
|
||
|
|
return FILDES_BAD;
|
||
|
|
}
|
||
|
|
|
||
|
|
fd = mkstemp(buf);
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
default:
|
||
|
|
Sys_SetErrStat(EINVAL, "Bad value for 'mode'");
|
||
|
|
return FILDES_BAD;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (errno == ENOENT || errno == ENOTDIR || errno == EISDIR)
|
||
|
|
errno = 0; // file not found - not an error
|
||
|
|
|
||
|
|
Sys_SetErrStat(errno, "open()/mkstemp()");
|
||
|
|
|
||
|
|
// Apply hints
|
||
|
|
if (fd >= 0) {
|
||
|
|
if (flags & FH_SEQ)
|
||
|
|
posix_fadvise(fd, 0, 0, POSIX_FADV_SEQUENTIAL);
|
||
|
|
if (flags & FH_RAND)
|
||
|
|
posix_fadvise(fd, 0, 0, POSIX_FADV_RANDOM);
|
||
|
|
if (flags & FH_NOREUSE)
|
||
|
|
posix_fadvise(fd, 0, 0, POSIX_FADV_NOREUSE);
|
||
|
|
}
|
||
|
|
return fd;
|
||
|
|
}
|
||
|
|
|
||
|
|
Sint64 Sys_Fread(Fildes fd, void *buf, size_t nbytes)
|
||
|
|
{
|
||
|
|
Sint64 n;
|
||
|
|
|
||
|
|
errno = 0;
|
||
|
|
|
||
|
|
n = read(fd, buf, nbytes);
|
||
|
|
Sys_SetErrStat(errno, "read()");
|
||
|
|
return n;
|
||
|
|
}
|
||
|
|
|
||
|
|
Sint64 Sys_Fwrite(Fildes fd, const void *buf, size_t nbytes)
|
||
|
|
{
|
||
|
|
Sint64 n;
|
||
|
|
|
||
|
|
errno = 0;
|
||
|
|
|
||
|
|
n = write(fd, buf, nbytes);
|
||
|
|
Sys_SetErrStat(errno, "write() failed");
|
||
|
|
return n;
|
||
|
|
}
|
||
|
|
|
||
|
|
Sint64 Sys_Ftell(Fildes fd)
|
||
|
|
{
|
||
|
|
Sint64 pos;
|
||
|
|
|
||
|
|
errno = 0;
|
||
|
|
|
||
|
|
pos = lseek(fd, 0, SEEK_CUR);
|
||
|
|
if (pos < 0 && errno != ESPIPE)
|
||
|
|
errno = 0; // lseek() unsupported, but not a system error
|
||
|
|
|
||
|
|
Sys_SetErrStat(errno, "lseek()");
|
||
|
|
return pos;
|
||
|
|
}
|
||
|
|
|
||
|
|
Sint64 Sys_Fseek(Fildes fd, Sint64 off, SeekMode whence)
|
||
|
|
{
|
||
|
|
Sint64 pos;
|
||
|
|
int mode;
|
||
|
|
|
||
|
|
switch (whence) {
|
||
|
|
case SK_SET: mode = SEEK_SET; break;
|
||
|
|
case SK_CUR: mode = SEEK_CUR; break;
|
||
|
|
case SK_END: mode = SEEK_END; break;
|
||
|
|
default:
|
||
|
|
Sys_SetErrStat(EINVAL, "Bad value for 'whence'");
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
errno = 0;
|
||
|
|
|
||
|
|
pos = lseek(fd, off, mode);
|
||
|
|
if (pos < 0 && errno == ESPIPE)
|
||
|
|
errno = 0; // lseek() unsupported, but not a system error
|
||
|
|
|
||
|
|
Sys_SetErrStat(errno, "lseek()");
|
||
|
|
return pos;
|
||
|
|
}
|
||
|
|
|
||
|
|
Sint64 Sys_FileSize(Fildes fd)
|
||
|
|
{
|
||
|
|
Sint64 size = -1;
|
||
|
|
struct stat buf;
|
||
|
|
|
||
|
|
errno = 0;
|
||
|
|
|
||
|
|
if (fstat(fd, &buf) == 0)
|
||
|
|
size = buf.st_size;
|
||
|
|
|
||
|
|
Sys_SetErrStat(errno, "fstat()");
|
||
|
|
return size;
|
||
|
|
}
|
||
|
|
|
||
|
|
Judgement Sys_SetEof(Fildes fd)
|
||
|
|
{
|
||
|
|
Sint64 pos = Sys_Ftell(fd);
|
||
|
|
if (pos < 0)
|
||
|
|
return NG; // Sys_Ftell() already sets error
|
||
|
|
|
||
|
|
ftruncate(fd, pos);
|
||
|
|
return Sys_SetErrStat(errno, "ftruncate()");
|
||
|
|
}
|
||
|
|
|
||
|
|
Judgement Sys_Fsync(Fildes fd, Boolean fullSync)
|
||
|
|
{
|
||
|
|
errno = 0;
|
||
|
|
|
||
|
|
if (fullSync)
|
||
|
|
fsync(fd);
|
||
|
|
else
|
||
|
|
fdatasync(fd);
|
||
|
|
|
||
|
|
return Sys_SetErrStat(errno, "fsync()/fdatasync()");
|
||
|
|
}
|
||
|
|
|
||
|
|
void Sys_Fclose(Fildes fd)
|
||
|
|
{
|
||
|
|
errno = 0;
|
||
|
|
|
||
|
|
close(fd);
|
||
|
|
Sys_SetErrStat(errno, "close()");
|
||
|
|
}
|
||
|
|
|
||
|
|
typedef struct FileList FileList;
|
||
|
|
struct FileList {
|
||
|
|
FileList *next;
|
||
|
|
size_t len;
|
||
|
|
char name[1]; // dynamically sized, len+1
|
||
|
|
};
|
||
|
|
|
||
|
|
#ifdef _DIRENT_HAVE_D_TYPE
|
||
|
|
// May save some syscalls
|
||
|
|
#define ISUNK(ent) ((ent)->d_type == DT_UNKNOWN)
|
||
|
|
#define ISDIR(ent) ((ent)->d_type == DT_DIR)
|
||
|
|
#define ISLNK(ent) ((ent)->d_type == DT_LNK)
|
||
|
|
#else
|
||
|
|
// ...as if DT_UNKNOWN is always set
|
||
|
|
#define ISUNK(ent) (1)
|
||
|
|
#define ISDIR(ent) (0)
|
||
|
|
#define ISLNK(ent) (0)
|
||
|
|
#endif
|
||
|
|
|
||
|
|
char **Sys_ListFiles(const char *path, unsigned *nfiles, const char *pat)
|
||
|
|
{
|
||
|
|
DIR *dir = opendir(path);
|
||
|
|
if (!dir) {
|
||
|
|
if (errno != ENOENT && errno != ENOTDIR)
|
||
|
|
Sys_SetErrStat(errno, "opendir()");
|
||
|
|
|
||
|
|
return NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!pat)
|
||
|
|
pat = "";
|
||
|
|
|
||
|
|
Boolean dirOnly = (strcmp(pat, "/") == 0);
|
||
|
|
size_t elen = strlen(pat);
|
||
|
|
|
||
|
|
// Scan directory
|
||
|
|
|
||
|
|
FileList *entries = NULL;
|
||
|
|
unsigned count = 0;
|
||
|
|
size_t nchars = 0;
|
||
|
|
|
||
|
|
struct stat st;
|
||
|
|
struct dirent *ent;
|
||
|
|
|
||
|
|
char **files;
|
||
|
|
char *namep;
|
||
|
|
|
||
|
|
errno = 0;
|
||
|
|
|
||
|
|
while ((ent = readdir(dir)) != NULL) {
|
||
|
|
// Skip special entries
|
||
|
|
if (strcmp(ent->d_name, ".") == 0 || strcmp(ent->d_name, "..") == 0)
|
||
|
|
continue;
|
||
|
|
|
||
|
|
size_t len = strlen(ent->d_name);
|
||
|
|
|
||
|
|
if (dirOnly) {
|
||
|
|
// Filter anything other than subdirectories
|
||
|
|
Boolean isDir;
|
||
|
|
if (ISUNK(ent) || ISLNK(ent)) {
|
||
|
|
// Need a full stat() on entry
|
||
|
|
if (fstatat(dirfd(dir), ent->d_name, &st, 0) != 0) {
|
||
|
|
Sys_SetErrStat(errno, "fstatat()");
|
||
|
|
goto fail;
|
||
|
|
}
|
||
|
|
|
||
|
|
isDir = S_ISDIR(st.st_mode);
|
||
|
|
|
||
|
|
} else {
|
||
|
|
// May take advantage of dirent's d_type
|
||
|
|
isDir = ISDIR(ent);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!isDir)
|
||
|
|
continue;
|
||
|
|
|
||
|
|
} else {
|
||
|
|
// Filter by extension pattern
|
||
|
|
if (elen >= len)
|
||
|
|
continue;
|
||
|
|
|
||
|
|
if (strcmp(ent->d_name + len - elen, pat) != 0)
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Update counters
|
||
|
|
count++;
|
||
|
|
nchars += len + 1;
|
||
|
|
|
||
|
|
// ...and remember this entry
|
||
|
|
FileList *e = (FileList *) alloca(sizeof(*e) + len); // includes '\0'
|
||
|
|
|
||
|
|
e->next = entries;
|
||
|
|
e->len = len;
|
||
|
|
memcpy(e->name, ent->d_name, len + 1);
|
||
|
|
|
||
|
|
entries = e;
|
||
|
|
}
|
||
|
|
if (errno != 0) {
|
||
|
|
Sys_SetErrStat(errno, "readdir()");
|
||
|
|
goto fail;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Allocate result, single malloc(), +1 for NULL sentinel
|
||
|
|
files = (char **) malloc(sizeof(*files) * (count + 1) + nchars);
|
||
|
|
if (!files) {
|
||
|
|
Sys_OutOfMemory();
|
||
|
|
goto fail;
|
||
|
|
}
|
||
|
|
|
||
|
|
closedir(dir); // safe, can't fail anymore
|
||
|
|
|
||
|
|
// Collect files in buffer
|
||
|
|
namep = (char *) (files + count + 1);
|
||
|
|
for (unsigned i = 0; i < count; i++) {
|
||
|
|
files[i] = namep;
|
||
|
|
|
||
|
|
size_t n = entries->len + 1;
|
||
|
|
memcpy(namep, entries->name, n);
|
||
|
|
namep += n;
|
||
|
|
|
||
|
|
entries = entries->next;
|
||
|
|
}
|
||
|
|
files[count] = NULL; // NULL-terminate the file list
|
||
|
|
|
||
|
|
if (nfiles)
|
||
|
|
*nfiles = count;
|
||
|
|
|
||
|
|
return files;
|
||
|
|
|
||
|
|
fail:
|
||
|
|
closedir(dir);
|
||
|
|
return NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
Judgement Sys_Mkdir(const char *path)
|
||
|
|
{
|
||
|
|
errno = 0;
|
||
|
|
|
||
|
|
if (mkdir(path, 0755) != 0) {
|
||
|
|
int err = errno; // save errno - avoid overriding by stat()
|
||
|
|
if (err == EEXIST) {
|
||
|
|
// Make sure this is actually a directory
|
||
|
|
struct stat buf;
|
||
|
|
|
||
|
|
if (stat(path, &buf) == 0 && S_ISDIR(buf.st_mode))
|
||
|
|
err = 0; // all good
|
||
|
|
}
|
||
|
|
|
||
|
|
errno = err;
|
||
|
|
}
|
||
|
|
|
||
|
|
return Sys_SetErrStat(errno, "mkdir()");
|
||
|
|
}
|
||
|
|
|
||
|
|
Judgement Sys_Rename(const char *path, const char *new_path)
|
||
|
|
{
|
||
|
|
errno = 0;
|
||
|
|
|
||
|
|
rename(path, new_path);
|
||
|
|
return Sys_SetErrStat(errno, "rename()");
|
||
|
|
}
|
||
|
|
|
||
|
|
Judgement Sys_Remove(const char *path)
|
||
|
|
{
|
||
|
|
errno = 0;
|
||
|
|
|
||
|
|
remove(path);
|
||
|
|
|
||
|
|
// TODO check not found
|
||
|
|
return Sys_SetErrStat(errno, "remove()");
|
||
|
|
}
|
||
|
|
|