fileio.c

Go to the documentation of this file.

/*
# _____     ___ ____     ___ ____
#  ____|   |    ____|   |        | |____|
# |     ___|   |____ ___|    ____| |    \    PS2DEV Open Source Project.
#-----------------------------------------------------------------------
# (C)2001, Gustavo Scotti (gustavo@scotti.com)
# (c) 2003 Marcus R. Brown (mrbrown@0xd6.org)
# Licenced under Academic Free License version 2.0
# Review ps2sdk README & LICENSE files for further details.
*/

 
#include <tamtypes.h>
#include <ps2lib_err.h>
#include <kernel.h>
#include <sifrpc.h>
#include "sifarg.h"
#define NEWLIB_PORT_AWARE
#include <fileio.h>
#include <string.h>
#include <iopcontrol.h>
#include <fileio-common.h>
 
#define D(fmt, args...) printf("(%s:%s:%i):" #fmt, __FILE__, __FUNCTION__, __LINE__, ##args)
 
extern SifRpcClientData_t _fio_cd;
extern int _fio_block_mode;
extern int _fio_io_sema;
extern int _fio_completion_sema;
extern int _fio_recv_data[512];
extern int _fio_intr_data[32];
 
void _fio_read_intr(struct _fio_read_data *);
void _fio_intr();
 
SifRpcClientData_t _fio_cd;
int _fio_recv_data[512] __attribute__((aligned(64)));
int _fio_intr_data[32] __attribute__((aligned(64)));
int _fio_block_mode;
int _fio_io_sema;
int _fio_completion_sema;
 
int fioInit(void)
{
    int res;
    ee_sema_t sema;
    if (HasIopRebootedSinceLastCall())
        fioExit();
 
    if (_fio_cd.server)
        return 0;
 
    sceSifInitRpc(0);
 
    while (((res = sceSifBindRpc(&_fio_cd, 0x80000001, 0)) >= 0) &&
           (_fio_cd.server == NULL))
        nopdelay();
 
    if (res < 0)
        return res;
 
    sema.init_count      = 1;
    sema.max_count       = 1;
    sema.option          = 0;
    _fio_completion_sema = CreateSema(&sema);
    if (_fio_completion_sema < 0)
    {
        _fio_completion_sema = 0;
        return -E_LIB_SEMA_CREATE;
    }
 
    // Unofficial: create a locking semaphore to prevent a thread from overwriting another thread's return status.
    sema.init_count = 1;
    sema.max_count  = 1;
    sema.option     = 0;
    _fio_io_sema    = CreateSema(&sema);
    if (_fio_io_sema < 0)
    {
        _fio_io_sema = 0;
        return -E_LIB_SEMA_CREATE;
    }
 
    _fio_block_mode = FIO_WAIT;
 
    return 0;
}
 
void _fio_intr(void)
{
    iSignalSema(_fio_completion_sema);
}
 
int fioSync(int mode, int *retVal)
{
    int res;
 
    if ((res = fioInit()) < 0)
        return res;
 
    if (_fio_block_mode != FIO_NOWAIT)
        return -E_LIB_UNSUPPORTED;
 
    switch (mode) {
        case FIO_WAIT:
 
            WaitSema(_fio_completion_sema);
            SignalSema(_fio_completion_sema);
 
            if (retVal != NULL)
                *retVal = *(int *)UNCACHED_SEG(&_fio_recv_data[0]);
 
            return FIO_COMPLETE;
 
        case FIO_NOWAIT:
 
            if (PollSema(_fio_completion_sema) < 0)
                return FIO_INCOMPLETE;
 
            SignalSema(_fio_completion_sema);
 
            if (retVal != NULL)
                *retVal = *(int *)UNCACHED_SEG(&_fio_recv_data[0]);
 
            return FIO_COMPLETE;
 
        default:
            return -E_LIB_UNSUPPORTED;
    }
}
 
void fioSetBlockMode(int blocking)
{
    _fio_block_mode = blocking;
}
 
void fioExit(void)
{
    if (_fio_cd.server) {
        memset(&_fio_cd, 0, sizeof _fio_cd);
        if (_fio_completion_sema > 0) {
            DeleteSema(_fio_completion_sema);
            _fio_completion_sema = 0;
        }
        if (_fio_io_sema > 0) {
            DeleteSema(_fio_io_sema);
            _fio_io_sema = 0;
        }
    }
}
 
int fioOpen(const char *name, int mode)
{
    SIF_ARG(struct _fio_open_arg, arg);
    int res, result;
 
    if ((res = fioInit()) < 0)
        return res;
 
    WaitSema(_fio_io_sema);
    WaitSema(_fio_completion_sema);
 
    arg->mode = mode;
    strncpy(arg->name, name, sizeof(arg->name));
    arg->name[sizeof(arg->name) - 1] = 0;
 
    if ((res = sceSifCallRpc(&_fio_cd, FIO_F_OPEN, _fio_block_mode, arg, sizeof *arg,
                          _fio_recv_data, 4, (void *)_fio_intr, NULL)) >= 0) {
        result = (_fio_block_mode == FIO_NOWAIT) ? 0 : _fio_recv_data[0];
    } else { // Signal semaphore to avoid a deadlock if sceSifCallRpc fails.
        SignalSema(_fio_completion_sema);
        result = res;
    }
 
    SignalSema(_fio_io_sema);
 
    return result;
}
 
int fioClose(int fd)
{
    typedef union
    {
        int fd;
        int result;
    } arg_t;
    SIF_ARG(arg_t, arg);
    int res, result;
 
    if ((res = fioInit()) < 0)
        return res;
 
    WaitSema(_fio_io_sema);
    WaitSema(_fio_completion_sema);
 
    arg->fd = fd;
 
    if ((res = sceSifCallRpc(&_fio_cd, FIO_F_CLOSE, 0, arg, 4, arg, 4,
                          (void *)_fio_intr, NULL)) >= 0) {
        result = arg->result;
    } else { // Signal semaphore to avoid a deadlock if sceSifCallRpc fails.
        SignalSema(_fio_completion_sema);
        result = res;
    }
 
    SignalSema(_fio_io_sema);
 
    return result;
}
 
void _fio_read_intr(struct _fio_read_data *data)
{
    struct _fio_read_data *uncached = UNCACHED_SEG(data);
 
    if (uncached->size1 && uncached->dest1) {
        memcpy(uncached->dest1, uncached->buf1, uncached->size1);
    }
 
    if (uncached->size2 && uncached->dest2) {
        memcpy(uncached->dest2, uncached->buf2, uncached->size2);
    }
 
    iSignalSema(_fio_completion_sema);
}
 
int fioRead(int fd, void *ptr, int size)
{
    SIF_ARG(struct _fio_read_arg, arg);
    int res, result;
 
    if ((res = fioInit()) < 0)
        return res;
 
    WaitSema(_fio_io_sema);
    WaitSema(_fio_completion_sema);
 
    arg->fd        = fd;
    arg->ptr       = ptr;
    arg->size      = size;
    arg->read_data = (struct _fio_read_data *)_fio_intr_data;
 
    if (!IS_UNCACHED_SEG(ptr))
        sceSifWriteBackDCache(ptr, size);
 
    if ((res = sceSifCallRpc(&_fio_cd, FIO_F_READ, _fio_block_mode, arg, sizeof *arg,
                          _fio_recv_data, 4, (void *)_fio_read_intr, _fio_intr_data)) >= 0) {
        result = (_fio_block_mode == FIO_NOWAIT) ? 0 : _fio_recv_data[0];
    } else { // Signal semaphore to avoid a deadlock if sceSifCallRpc fails.
        SignalSema(_fio_completion_sema);
        result = res;
    }
 
    SignalSema(_fio_io_sema);
 
    return result;
}
 
int fioWrite(int fd, const void *ptr, int size)
{
    SIF_ARG(struct _fio_write_arg, arg);
    int mis, res, result;
 
    if ((res = fioInit()) < 0)
        return res;
 
    WaitSema(_fio_io_sema);
    WaitSema(_fio_completion_sema);
 
    arg->fd   = fd;
    arg->ptr  = ptr;
    arg->size = size;
 
    /* Copy the unaligned (16-byte) portion into the argument */
    mis = 0;
    if ((u32)ptr & 0xf) {
        mis = 16 - ((u32)ptr & 0xf);
        if (mis > size)
            mis = size;
    }
    arg->mis = mis;
 
    if (mis)
        memcpy(arg->aligned, ptr, mis);
 
    if (!IS_UNCACHED_SEG(ptr))
        sceSifWriteBackDCache((void *)ptr, size);
 
    if ((res = sceSifCallRpc(&_fio_cd, FIO_F_WRITE, _fio_block_mode, arg, sizeof *arg,
                          _fio_recv_data, 4, (void *)_fio_intr, NULL)) >= 0) {
        result = (_fio_block_mode == FIO_NOWAIT) ? 0 : _fio_recv_data[0];
    } else { // Signal semaphore to avoid a deadlock if sceSifCallRpc fails.
        SignalSema(_fio_completion_sema);
        result = res;
    }
 
    SignalSema(_fio_io_sema);
 
    return result;
}
 
int fioLseek(int fd, int offset, int whence)
{
    SIF_ARG(struct _fio_lseek_arg, arg);
    int res, result;
 
    if ((res = fioInit()) < 0)
        return res;
 
    WaitSema(_fio_io_sema);
    WaitSema(_fio_completion_sema);
 
    arg->p.fd   = fd;
    arg->offset = offset;
    arg->whence = whence;
 
    if ((res = sceSifCallRpc(&_fio_cd, FIO_F_LSEEK, 0, arg, sizeof *arg,
                          arg, 4, (void *)_fio_intr, NULL)) >= 0) {
        result = arg->p.result;
    } else { // Signal semaphore to avoid a deadlock if sceSifCallRpc fails.
        SignalSema(_fio_completion_sema);
        result = res;
    }
 
    SignalSema(_fio_io_sema);
 
    return result;
}
 
int fioIoctl(int fd, int request, void *data)
{
    SIF_ARG(struct _fio_ioctl_arg, arg);
    int res, result;
 
    if ((res = fioInit()) < 0)
        return res;
 
    WaitSema(_fio_io_sema);
    WaitSema(_fio_completion_sema);
 
    arg->p.fd    = fd;
    arg->request = request;
 
    if (data != NULL)
        memcpy(arg->data, data, 1024);
 
    if ((res = sceSifCallRpc(&_fio_cd, FIO_F_IOCTL, 0, arg, sizeof *arg,
                          arg, 4, (void *)_fio_intr, NULL)) >= 0) {
        result = arg->p.result;
    } else { // Signal semaphore to avoid a deadlock if sceSifCallRpc fails.
        SignalSema(_fio_completion_sema);
        result = res;
    }
 
    SignalSema(_fio_io_sema);
 
    return result;
}
 
int fioRemove(const char *name)
{
    typedef union
    {
        char path[FIO_PATH_MAX];
        int result;
    } arg_t;
    SIF_ARG(arg_t, arg);
    int res, result;
 
    if ((res = fioInit()) < 0)
        return res;
 
    WaitSema(_fio_io_sema);
    WaitSema(_fio_completion_sema);
 
    strncpy(arg->path, name, sizeof(arg->path));
    arg->path[sizeof(arg->path) - 1] = 0;
 
    if ((res = sceSifCallRpc(&_fio_cd, FIO_F_REMOVE, 0, arg, sizeof *arg,
                          arg, 4, (void *)_fio_intr, NULL)) >= 0) {
        result = arg->result;
    } else { // Signal semaphore to avoid a deadlock if sceSifCallRpc fails.
        SignalSema(_fio_completion_sema);
        result = res;
    }
 
    SignalSema(_fio_io_sema);
 
    return result;
}
 
int fioMkdir(const char *path)
{
    typedef union
    {
        char path[FIO_PATH_MAX];
        int result;
    } arg_t;
    SIF_ARG(arg_t, arg);
    int res, result;
 
    if ((res = fioInit()) < 0)
        return res;
 
    WaitSema(_fio_io_sema);
    WaitSema(_fio_completion_sema);
 
    strncpy(arg->path, path, sizeof(arg->path));
    arg->path[sizeof(arg->path) - 1] = 0;
 
    if ((res = sceSifCallRpc(&_fio_cd, FIO_F_MKDIR, 0, arg, sizeof *arg,
                          arg, 4, (void *)_fio_intr, NULL)) >= 0) {
        result = arg->result;
    } else { // Signal semaphore to avoid a deadlock if sceSifCallRpc fails.
        SignalSema(_fio_completion_sema);
        result = res;
    }
 
    SignalSema(_fio_io_sema);
 
    return result;
}
 
int fioRmdir(const char *dirname)
{
    typedef union
    {
        char path[FIO_PATH_MAX];
        int result;
    } arg_t;
    SIF_ARG(arg_t, arg);
    int res, result;
 
    if ((res = fioInit()) < 0)
        return res;
 
    WaitSema(_fio_io_sema);
    WaitSema(_fio_completion_sema);
 
    strncpy(arg->path, dirname, sizeof(arg->path));
    arg->path[sizeof(arg->path) - 1] = 0;
 
    if ((res = sceSifCallRpc(&_fio_cd, FIO_F_RMDIR, 0, arg, sizeof *arg,
                          arg, 4, (void *)_fio_intr, NULL)) >= 0) {
        result = arg->result;
    } else { // Signal semaphore to avoid a deadlock if sceSifCallRpc fails.
        SignalSema(_fio_completion_sema);
        result = res;
    }
 
    SignalSema(_fio_io_sema);
 
    return result;
}
 
int fioPutc(int fd, int c)
{
    return fioWrite(fd, &c, 1);
}
 
int fioGetc(int fd)
{
    int c;
 
    fioRead(fd, &c, 1);
    return c;
}
 
int fioGets(int fd, char *buff, int n)
{
    // Rather than doing a slow byte-at-a-time read
    // read upto the max langth, then re-position file afterwards
    int read, i;
 
    read = fioRead(fd, buff, n);
 
    for (i = 0; i < (read - 1); i++) {
        switch (buff[i]) {
            case '\n':
                fioLseek(fd, (i + 1) - read, SEEK_CUR);
                buff[i] = 0; // terminate after newline
                return i;
 
            case 0:
                fioLseek(fd, i - read, SEEK_CUR);
                return i;
        }
    }
 
    // if we reached here, then we havent found the end of a string
    return i;
}
 
int fioDopen(const char *name)
{
    typedef union
    {
        char name[FIO_PATH_MAX];
        int result;
    } arg_t;
    SIF_ARG(arg_t, arg);
    int res, result;
 
    if ((res = fioInit()) < 0)
        return res;
 
    WaitSema(_fio_io_sema);
    WaitSema(_fio_completion_sema);
 
    strncpy(arg->name, name, sizeof(arg->name));
    arg->name[sizeof(arg->name) - 1] = 0;
 
    if ((res = sceSifCallRpc(&_fio_cd, FIO_F_DOPEN, 0, arg, sizeof *arg,
                          arg, 4, (void *)_fio_intr, NULL)) >= 0) {
        result = arg->result;
    } else { // Signal semaphore to avoid a deadlock if sceSifCallRpc fails.
        SignalSema(_fio_completion_sema);
        result = res;
    }
 
    SignalSema(_fio_io_sema);
 
    return result;
}
 
int fioDclose(int fd)
{
    typedef union
    {
        int fd;
        int result;
    } arg_t;
    SIF_ARG(arg_t, arg);
    int res, result;
 
    if ((res = fioInit()) < 0)
        return res;
 
    WaitSema(_fio_io_sema);
    WaitSema(_fio_completion_sema);
 
    arg->fd = fd;
 
    if ((res = sceSifCallRpc(&_fio_cd, FIO_F_DCLOSE, 0, arg, sizeof *arg,
                          arg, 4, (void *)_fio_intr, NULL)) >= 0) {
        result = arg->result;
    } else { // Signal semaphore to avoid a deadlock if sceSifCallRpc fails.
        SignalSema(_fio_completion_sema);
        result = res;
    }
 
    SignalSema(_fio_io_sema);
 
    return result;
}
 
int fioDread(int fd, io_dirent_t *buf)
{
    SIF_ARG(struct _fio_dread_arg, arg);
    int res, result;
 
    if ((res = fioInit()) < 0)
        return res;
 
    WaitSema(_fio_io_sema);
    WaitSema(_fio_completion_sema);
 
    arg->p.fd = fd;
    arg->buf  = buf;
 
    if (!IS_UNCACHED_SEG(buf))
        sceSifWriteBackDCache(buf, sizeof(io_dirent_t));
 
    if ((res = sceSifCallRpc(&_fio_cd, FIO_F_DREAD, 0, arg, sizeof *arg,
                          arg, 4, (void *)_fio_intr, NULL)) >= 0) {
        result = arg->p.result;
    } else { // Signal semaphore to avoid a deadlock if sceSifCallRpc fails.
        SignalSema(_fio_completion_sema);
        result = res;
    }
 
    SignalSema(_fio_io_sema);
 
    return result;
}
 
int fioGetstat(const char *name, io_stat_t *buf)
{
    SIF_ARG(struct _fio_getstat_arg, arg);
    int res, result;
 
    if ((res = fioInit()) < 0)
        return res;
 
    WaitSema(_fio_io_sema);
    WaitSema(_fio_completion_sema);
 
    arg->p.buf = buf;
    strncpy(arg->name, name, sizeof(arg->name));
    arg->name[sizeof(arg->name) - 1] = 0;
 
    if (!IS_UNCACHED_SEG(buf))
        sceSifWriteBackDCache(buf, sizeof(io_stat_t));
 
    if ((res = sceSifCallRpc(&_fio_cd, FIO_F_GETSTAT, 0, arg, sizeof *arg,
                          arg, 4, (void *)_fio_intr, NULL)) >= 0) {
        result = arg->p.result;
    } else { // Signal semaphore to avoid a deadlock if sceSifCallRpc fails.
        SignalSema(_fio_completion_sema);
        result = res;
    }
 
    SignalSema(_fio_io_sema);
 
    return result;
}
 
int fioChstat(const char *name, io_stat_t *buf, u32 cbit)
{
    SIF_ARG(struct _fio_chstat_arg, arg);
    int res, result;
 
    if ((res = fioInit()) < 0)
        return res;
 
    WaitSema(_fio_io_sema);
    WaitSema(_fio_completion_sema);
 
    arg->p.cbit = cbit;
    memcpy(&arg->stat, buf, sizeof(io_stat_t));
    strncpy(arg->name, name, sizeof(arg->name));
    arg->name[sizeof(arg->name) - 1] = 0;
 
    if ((res = sceSifCallRpc(&_fio_cd, FIO_F_CHSTAT, 0, arg, sizeof *arg,
                          arg, 4, (void *)_fio_intr, NULL)) >= 0) {
        result = arg->p.result;
    } else { // Signal semaphore to avoid a deadlock if sceSifCallRpc fails.
        SignalSema(_fio_completion_sema);
        result = res;
    }
 
    SignalSema(_fio_io_sema);
 
    return result;
}
 
int fioFormat(const char *name)
{
    typedef union
    {
        char path[FIO_PATH_MAX];
        int result;
    } arg_t;
    SIF_ARG(arg_t, arg);
    int res, result;
 
    if ((res = fioInit()) < 0)
        return res;
 
    WaitSema(_fio_io_sema);
    WaitSema(_fio_completion_sema);
 
    strncpy(arg->path, name, sizeof(arg->path));
    arg->path[sizeof(arg->path) - 1] = 0;
 
    if ((res = sceSifCallRpc(&_fio_cd, FIO_F_FORMAT, 0, arg, sizeof *arg,
                          arg, 4, (void *)_fio_intr, NULL)) >= 0) {
        result = arg->result;
    } else { // Signal semaphore to avoid a deadlock if sceSifCallRpc fails.
        SignalSema(_fio_completion_sema);
        result = res;
    }
 
    SignalSema(_fio_io_sema);
 
    return result;
}