rt-thread-official/bsp/simulator/drivers/dfs_win32.c

584 lines
14 KiB
C

/*
* Copyright (c) 2006-2021, RT-Thread Development Team
*
* SPDX-License-Identifier: Apache-2.0
*
* Change Logs:
* Date Author Notes
* 2012-11-27 prife the first version
* 2013-03-03 aozima add dfs_win32_stat st_mtime support.
* 2017-10-20 urey support rt-thread 3.0
*/
#include <rtthread.h>
#include <dfs_fs.h>
#include <dfs_file.h>
#include <rtdevice.h>
#include <io.h>
#include <fcntl.h>
#include <string.h>
#include <stdio.h>
#include <string.h>
#include <WinError.h>
#include <windows.h>
/*
* RT-Thread DFS Interface for win-directory as an disk device
*/
#define FILE_PATH_MAX 256 /* the longest file path */
#define WIN32_DIRDISK_ROOT "./disk"
typedef struct {
HANDLE handle;
char *start;
char *end;
char *curr;
struct _finddata_t finddata;
} WINDIR;
/* There are so many error codes in windows, you'd better google for details.
* google "System Error Codes (Windows)"
* http://msdn.microsoft.com/ZH-CN/library/windows/desktop/ms681381(v=vs.85).aspx
*/
struct _errcode_map
{
rt_uint16_t dfserr;
rt_uint16_t win32err;
};
static const struct _errcode_map errcode_table[] =
{
{ENOENT, ERROR_FILE_NOT_FOUND },
{ENOENT, ERROR_PATH_NOT_FOUND },
{EEXIST, ERROR_FILE_EXISTS },
{EEXIST, ERROR_ALREADY_EXISTS },
{ENOTEMPTY, ERROR_DIR_NOT_EMPTY },
{EBUSY, ERROR_PATH_BUSY },
{EINVAL, ERROR_ACCESS_DENIED },
#if 0 /* TODO: MORE NEED BE ADDED */
{DFS_STATUS_EISDIR, ERROR_FILE_EXISTS },
{DFS_STATUS_ENOTDIR, ERROR_FILE_EXISTS },
{DFS_STATUS_EBADF, ERROR_FILE_EXISTS },
{DFS_STATUS_EBUSY, ERROR_FILE_EXISTS },
{DFS_STATUS_ENOMEM, ERROR_FILE_EXISTS },
{DFS_STATUS_ENOSPC, ERROR_FILE_EXISTS },
#endif
};
static int win32_result_to_dfs(DWORD res)
{
int i;
int err = 0;
for (i = 0; i < sizeof(errcode_table) / sizeof(struct _errcode_map); i++)
{
if (errcode_table[i].win32err == (res & 0xFFFF))
{
err = -errcode_table[i].dfserr;
return err;
}
}
/* unknown error */
rt_kprintf("dfs win32 error not supported yet: %d\n", res);
return -1;
}
static int dfs_win32_mount(
struct dfs_filesystem *fs,
unsigned long rwflag,
const void *data)
{
return 0;
}
static int dfs_win32_unmount(struct dfs_filesystem *fs)
{
return 0;
}
static int dfs_win32_mkfs(rt_device_t devid)
{
return -ENOSYS;
}
static int dfs_win32_statfs(struct dfs_filesystem *fs,
struct statfs *buf)
{
return -ENOSYS;
}
static char *winpath_dirdup(char *des, const char *src)
{
char *path;
int i = 0;
path = rt_malloc(FILE_PATH_MAX);
if (path == RT_NULL)
return RT_NULL;
strcpy(path, des);
strcat(path, src);
while (1)
{
if (path[i] == 0)
break;
if (path[i] == '/')
path[i] = '\\';
i++;
}
return path;
}
/* This function can convert the path in rt-thread/dfs to the path in windows */
char * dfs_win32_dirdup(char * path)
{
char * file_path;
file_path = winpath_dirdup(WIN32_DIRDISK_ROOT, path);
return file_path;
}
static int dfs_win32_open(struct dfs_file *file)
{
int fd;
uint32_t oflag, mode;
char *file_path;
int res;
oflag = file->flags;
if (oflag & O_DIRECTORY) /* operations about dir */
{
WINDIR *wdirp;
HANDLE handle;
int len;
file_path = winpath_dirdup(WIN32_DIRDISK_ROOT, file->vnode->path);
if (oflag & O_CREAT) /* create a dir*/
{
res = CreateDirectory(file_path, NULL);
if (res == 0)
{
rt_free(file_path);
return win32_result_to_dfs(GetLastError());
}
}
len = strlen(file_path);
if (file_path[len - 1] != '\\')
{
file_path[len] = '\\';
file_path[len + 1] = 0;
}
strcat(file_path, "*.*");
/* _findfirst will get '.' */
/* save this pointer,will used by dfs_win32_getdents*/
wdirp = rt_malloc(sizeof(WINDIR));
RT_ASSERT(wdirp != NULL);
if ((handle = _findfirst(file_path, &wdirp->finddata)) == -1)
{
rt_free(wdirp);
rt_free(file_path);
goto __err;
}
len = strlen(wdirp->finddata.name) + 1;
wdirp->handle = handle;
//wdirp->nfiles = 1;
wdirp->start = (char *)malloc(len); //not rt_malloc!
wdirp->end = wdirp->curr = wdirp->start;
wdirp->end += len;
rt_strncpy(wdirp->curr, wdirp->finddata.name, len);
file->vnode->data = (void *)wdirp;
rt_free(file_path);
return 0;
}
/* regular file operations */
mode = O_BINARY;
if (oflag & O_RDONLY) mode |= O_RDONLY;
if (oflag & O_WRONLY) mode |= O_WRONLY;
if (oflag & O_RDWR) mode |= O_RDWR;
/* Opens the file, if it is existing. If not, a new file is created. */
if (oflag & O_CREAT) mode |= O_CREAT;
/* Creates a new file. If the file is existing, it is truncated and overwritten. */
if (oflag & O_TRUNC) mode |= O_TRUNC;
/* Creates a new file. The function fails if the file is already existing. */
if (oflag & O_EXCL) mode |= O_EXCL;
file_path = winpath_dirdup(WIN32_DIRDISK_ROOT, file->vnode->path);
fd = _open(file_path, mode, 0x0100 | 0x0080); /* _S_IREAD | _S_IWRITE */
rt_free(file_path);
if (fd < 0)
goto __err;
/* save this pointer, it will be used when calling read(), write(),
* flush(), seek(), and will be free when calling close()*/
file->data = (void *)fd;
file->pos = 0;
file->vnode->size = _lseek(fd, 0, SEEK_END);
if (oflag & O_APPEND)
{
file->pos = file->vnode->size;
}
else
_lseek(fd, 0, SEEK_SET);
return 0;
__err:
res = GetLastError();
return win32_result_to_dfs(res);
}
static int dfs_win32_close(struct dfs_file *file)
{
if (file->flags & O_DIRECTORY)
{
WINDIR *wdirp = (WINDIR*)(file->vnode->data);
RT_ASSERT(wdirp != RT_NULL);
if (_findclose((intptr_t)wdirp->handle) == 0) {
free(wdirp->start); /* NOTE: here we don't use rt_free! */
rt_free(wdirp);
return 0;
}
}
else /* regular file operations */
{
if (_close((int)(file->data)) == 0)
return 0;
}
return win32_result_to_dfs(GetLastError());
}
static int dfs_win32_ioctl(struct dfs_file *file, int cmd, void *args)
{
return -ENOSYS;
}
static int dfs_win32_read(struct dfs_file *file, void *buf, size_t len)
{
int fd;
int char_read;
fd = (int)(file->data);
char_read = _read(fd, buf, len);
if (char_read < 0)
return win32_result_to_dfs(GetLastError());
/* update position */
file->pos = _lseek(fd, 0, SEEK_CUR);
return char_read;
}
static int dfs_win32_write(struct dfs_file *file, const void *buf, size_t len)
{
int fd;
int char_write;
fd = (int)(file->data);
char_write = _write(fd, buf, len);
if (char_write < 0)
return win32_result_to_dfs(GetLastError());
/* update position */
file->pos = _lseek(fd, 0, SEEK_CUR);
return char_write;
}
static int dfs_win32_flush(struct dfs_file *file)
{
return 0;
}
static int dfs_win32_seek(struct dfs_file *file,
rt_off_t offset)
{
int result;
/* set offset as current offset */
if (file->vnode->type == FT_DIRECTORY)
{
WINDIR* wdirp = (WINDIR*)(file->vnode->data);
RT_ASSERT(wdirp != RT_NULL);
wdirp->curr = wdirp->start + offset;
return offset;
}
else //file->type == FT_REGULAR)
{
result = _lseek((int)(file->data), offset, SEEK_SET);
if (result >= 0)
return offset;
else
return win32_result_to_dfs(GetLastError());
}
}
/* return the size of struct dirent*/
static int dfs_win32_getdents(struct dfs_file *file, struct dirent *dirp, rt_uint32_t count)
{
WINDIR *wdirp;
struct dirent *d = dirp;
int result;
/* make integer count */
if (count / sizeof(struct dirent) != 1)
return -EINVAL;
wdirp = (WINDIR*)(file->vnode->data);
RT_ASSERT(wdirp != RT_NULL);
if (wdirp->curr == NULL) //no more entries in this directory
return 0;
/* get the current entry */
if (wdirp->finddata.attrib & _A_SUBDIR)
d->d_type = DT_DIR;
else
d->d_type = DT_REG;
d->d_namlen = (rt_uint8_t)strlen(wdirp->curr);
strncpy(d->d_name, wdirp->curr, DIRENT_NAME_MAX);
d->d_reclen = (rt_uint16_t)sizeof(struct dirent);
wdirp->curr += (strlen(wdirp->curr) + 1);
file->pos = wdirp->curr - wdirp->start + sizeof(struct dirent);//NOTE!
/* now set up for the next call to readdir */
if (wdirp->curr >= wdirp->end)
{
if (_findnext(wdirp->handle, &wdirp->finddata) == 0)
{
char* old_start = wdirp->start;
long name_len = strlen(wdirp->finddata.name) + 1;
wdirp->start = realloc(wdirp->start, wdirp->end - wdirp->start + name_len);
wdirp->curr = wdirp->start + (wdirp->curr - old_start);
wdirp->end = wdirp->curr + name_len;
rt_strcpy(wdirp->curr, wdirp->finddata.name);
}
else
{
if ((result = GetLastError()) == ERROR_NO_MORE_FILES)
wdirp->curr = NULL;
else
return win32_result_to_dfs(result);
}
}
return sizeof(struct dirent);
}
static int dfs_win32_unlink(struct dfs_filesystem *fs, const char *path)
{
int result;
char *fp;
fp = winpath_dirdup(WIN32_DIRDISK_ROOT, path);
if (fp == RT_NULL)
{
rt_kprintf("out of memory.\n");
return -ENOMEM;
}
result = GetFileAttributes(fp);
if (result == INVALID_FILE_ATTRIBUTES)
goto __err;
if (result & FILE_ATTRIBUTE_DIRECTORY)//winnt.h
{
if (RemoveDirectory(fp) == RT_FALSE)
goto __err;
}
else //(result & FILE_ATTRIBUTE_NORMAL)
{
if (_unlink(fp) < 0)
goto __err;
}
rt_free(fp);
return 0;
__err:
rt_free(fp);
return win32_result_to_dfs(GetLastError());
}
static int dfs_win32_rename(
struct dfs_filesystem *fs,
const char *oldpath,
const char *newpath)
{
int result;
char *op, *np;
op = winpath_dirdup(WIN32_DIRDISK_ROOT, oldpath);
np = winpath_dirdup(WIN32_DIRDISK_ROOT, newpath);
if (op == RT_NULL || np == RT_NULL)
{
rt_kprintf("out of memory.\n");
return -ENOMEM;
}
/* If the function fails, the return value is zero. */
result = MoveFile(op, np);
rt_free(op);
rt_free(np);
if (result == 0)
return win32_result_to_dfs(GetLastError());
return 0;
}
static int dfs_win32_stat(struct dfs_filesystem *fs, const char *path, struct stat *st)
{
WIN32_FIND_DATA fileInfo;
HANDLE hFind;
char *fp;
fp = winpath_dirdup(WIN32_DIRDISK_ROOT, path);
if (fp == RT_NULL)
{
rt_kprintf("out of memory.\n");
return -ENOMEM;
}
hFind = FindFirstFile(fp, &fileInfo);
rt_free(fp);
if (hFind == INVALID_HANDLE_VALUE)
goto __err;
st->st_mode = S_IFREG | S_IRUSR | S_IRGRP | S_IROTH |
S_IWUSR | S_IWGRP | S_IWOTH;
/* convert file info to dfs stat structure */
if (fileInfo.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
{
st->st_mode &= ~S_IFREG;
st->st_mode |= S_IFDIR | S_IXUSR | S_IXGRP | S_IXOTH;
}
st->st_dev = 0;
st->st_size = fileInfo.nFileSizeLow;
/* get st_mtime. */
{
LARGE_INTEGER time_tmp;
time_tmp.LowPart = fileInfo.ftLastWriteTime.dwLowDateTime;
time_tmp.HighPart = fileInfo.ftLastWriteTime.dwHighDateTime;
/* removes the diff between 1970 and 1601. */
time_tmp.QuadPart -= 11644473600000 * 10000;
/* converts back from 100-nanoseconds to seconds. */
time_tmp.QuadPart /= 10UL * 1000 * 1000;
st->st_mtime = time_tmp.QuadPart;
}
FindClose(hFind);
return 0;
__err:
return win32_result_to_dfs(GetLastError());
}
static const struct dfs_file_ops dfs_win32_file_ops =
{
dfs_win32_open,
dfs_win32_close,
dfs_win32_ioctl,
dfs_win32_read,
dfs_win32_write,
dfs_win32_flush,
dfs_win32_seek,
dfs_win32_getdents,
};
static const struct dfs_filesystem_ops dfs_win32_ops =
{
"wdir", /* file system type: dir */
DFS_FS_FLAG_DEFAULT,
&dfs_win32_file_ops,
dfs_win32_mount,
dfs_win32_unmount,
dfs_win32_mkfs,
dfs_win32_statfs,
dfs_win32_unlink,
dfs_win32_stat,
dfs_win32_rename,
};
int dfs_win32_init(void)
{
/* register uffs file system */
dfs_register(&dfs_win32_ops);
return 0;
}
static rt_err_t nop_init(rt_device_t dev)
{
return RT_EOK;
}
static rt_err_t nop_open(rt_device_t dev, rt_uint16_t oflag)
{
return RT_EOK;
}
static rt_err_t nop_close(rt_device_t dev)
{
return RT_EOK;
}
static rt_ssize_t nop_read(rt_device_t dev,
rt_off_t pos,
void *buffer,
rt_size_t size)
{
return size;
}
static rt_ssize_t nop_write(rt_device_t dev,
rt_off_t pos,
const void *buffer,
rt_size_t size)
{
return size;
}
static rt_err_t nop_control(rt_device_t dev, int cmd, void *args)
{
return RT_EOK;
}
static struct rt_device win_sharedir_dev;
rt_err_t rt_win_sharedir_init(const char *name)
{
rt_device_t dev;
dev = &win_sharedir_dev;
RT_ASSERT(dev != RT_NULL);
/* set device class and generic device interface */
dev->type = RT_Device_Class_Block;
dev->init = nop_init;
dev->open = nop_open;
dev->read = nop_read;
dev->write = nop_write;
dev->close = nop_close;
dev->control = nop_control;
dev->rx_indicate = RT_NULL;
dev->tx_complete = RT_NULL;
/* register to RT-Thread device system */
return rt_device_register(dev, name, RT_DEVICE_FLAG_RDWR | RT_DEVICE_FLAG_STANDALONE);
}