Subversion Repositories HelenOS

/*

 * Copyright (c) 2008 Jakub Jermar

 * All rights reserved.

 *

 * Redistribution and use in source and binary forms, with or without

 * modification, are permitted provided that the following conditions

 * are met:

 *

 * - Redistributions of source code must retain the above copyright

 *   notice, this list of conditions and the following disclaimer.

 * - Redistributions in binary form must reproduce the above copyright

 *   notice, this list of conditions and the following disclaimer in the

 *   documentation and/or other materials provided with the distribution.

 * - The name of the author may not be used to endorse or promote products

 *   derived from this software without specific prior written permission.

 *

 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR

 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES

 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.

 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,

 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT

 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,

 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY

 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF

 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

 */

/** @addtogroup libc

 * @{

 */

/** @file

 */

#include <vfs/vfs.h>

#include <vfs/canonify.h>

#include <stdlib.h>

#include <unistd.h>

#include <dirent.h>

#include <fcntl.h>

#include <sys/stat.h>

#include <sys/types.h>

#include <ipc/ipc.h>

#include <ipc/services.h>

#include <async.h>

#include <atomic.h>

#include <futex.h>

#include <errno.h>

#include <string.h>

#include "../../srv/vfs/vfs.h"

int vfs_phone = -1;

futex_t vfs_phone_futex = FUTEX_INITIALIZER;

futex_t cwd_futex = FUTEX_INITIALIZER;

DIR *cwd_dir = NULL;

char *cwd_path = NULL;

size_t cwd_len = 0;

static char *absolutize(const char *path)

{

    char *ncwd_path;

    futex_down(&cwd_futex);

    size_t len = strlen(path);

    if (*path != '/') {

        if (!cwd_path) {

            futex_up(&cwd_futex);

            return NULL;

        }

        ncwd_path = malloc(len + cwd_len + 1);

        if (!ncwd_path) {

            futex_up(&cwd_futex);

            return NULL;

        }

        strcpy(ncwd_path, cwd_path);

        ncwd_path[cwd_len] = '/';

        ncwd_path[cwd_len + 1] = '\0';

    } else {

        ncwd_path = malloc(len + 1);

        if (!ncwd_path) {

            futex_up(&cwd_futex);

            return NULL;

        }

        ncwd_path[0] = '\0';

    }

    strcat(ncwd_path, path);

    futex_up(&cwd_futex);

    return ncwd_path;

}

static int vfs_connect(void)

{

    if (vfs_phone < 0)

        vfs_phone = ipc_connect_me_to(PHONE_NS, SERVICE_VFS, 0, 0);

    return vfs_phone;

}

int mount(const char *fs_name, const char *mp, const char *dev)

{

    int res;

    ipcarg_t rc;

    aid_t req;

    int dev_handle = 0; /* TODO */

    char *mpa = absolutize(mp);

    if (!mpa)

        return ENOMEM;

    size_t mpc_len;

    char *mpc = canonify(mpa, &mpc_len);

    if (!mpc) {

        free(mpa);

        return EINVAL;

    }

    futex_down(&vfs_phone_futex);

    async_serialize_start();

    if (vfs_phone < 0) {

        res = vfs_connect();

        if (res < 0) {

            async_serialize_end();

            futex_up(&vfs_phone_futex);

            free(mpa);

            return res;

        }

    }

    req = async_send_1(vfs_phone, VFS_MOUNT, dev_handle, NULL);

    rc = ipc_data_write_start(vfs_phone, (void *)fs_name, strlen(fs_name));

    if (rc != EOK) {

        async_wait_for(req, NULL);

        async_serialize_end();

        futex_up(&vfs_phone_futex);

        free(mpa);

        return (int) rc;

    }

    rc = ipc_data_write_start(vfs_phone, (void *)mpc, mpc_len);

    if (rc != EOK) {

        async_wait_for(req, NULL);

        async_serialize_end();

        futex_up(&vfs_phone_futex);

        free(mpa);

        return (int) rc;

    }

    async_wait_for(req, &rc);

    async_serialize_end();

    futex_up(&vfs_phone_futex);

    free(mpa);

    return (int) rc;

}

static int _open(const char *path, int lflag, int oflag, ...)

{

    int res;

    ipcarg_t rc;

    ipc_call_t answer;

    aid_t req;

    char *pa = absolutize(path);

    if (!pa)

        return ENOMEM;

    size_t pc_len;

    char *pc = canonify(pa, &pc_len);

    if (!pc) {

        free(pa);

        return EINVAL;

    }

    futex_down(&vfs_phone_futex);

    async_serialize_start();

    if (vfs_phone < 0) {

        res = vfs_connect();

        if (res < 0) {

            async_serialize_end();

            futex_up(&vfs_phone_futex);

            free(pa);

            return res;

        }

    }

    req = async_send_3(vfs_phone, VFS_OPEN, lflag, oflag, 0, &answer);

    rc = ipc_data_write_start(vfs_phone, pc, pc_len);

    if (rc != EOK) {

        async_wait_for(req, NULL);

        async_serialize_end();

        futex_up(&vfs_phone_futex);

        free(pa);

        return (int) rc;

    }

    async_wait_for(req, &rc);

    async_serialize_end();

    futex_up(&vfs_phone_futex);

    free(pa);

    return (int) IPC_GET_ARG1(answer);

}

int open(const char *path, int oflag, ...)

{

    return _open(path, L_FILE, oflag);

}

int close(int fildes)

{

    int res;

    ipcarg_t rc;

    futex_down(&vfs_phone_futex);

    async_serialize_start();

    if (vfs_phone < 0) {

        res = vfs_connect();

        if (res < 0) {

            async_serialize_end();

            futex_up(&vfs_phone_futex);

            return res;

        }

    }

    rc = async_req_1_0(vfs_phone, VFS_CLOSE, fildes);

    async_serialize_end();

    futex_up(&vfs_phone_futex);

    return (int)rc;

}

ssize_t read(int fildes, void *buf, size_t nbyte)

{

    int res;

    ipcarg_t rc;

    ipc_call_t answer;

    aid_t req;

    futex_down(&vfs_phone_futex);

    async_serialize_start();

    if (vfs_phone < 0) {

        res = vfs_connect();

        if (res < 0) {

            async_serialize_end();

            futex_up(&vfs_phone_futex);

            return res;

        }

    }

    req = async_send_1(vfs_phone, VFS_READ, fildes, &answer);

    rc = ipc_data_read_start(vfs_phone, (void *)buf, nbyte);

    if (rc != EOK) {

        async_wait_for(req, NULL);

        async_serialize_end();

        futex_up(&vfs_phone_futex);

        return (ssize_t) rc;

    }

    async_wait_for(req, &rc);

    async_serialize_end();

    futex_up(&vfs_phone_futex);

    if (rc == EOK)

        return (ssize_t) IPC_GET_ARG1(answer);

    else

        return -1;

}

ssize_t write(int fildes, const void *buf, size_t nbyte)

{

    int res;

    ipcarg_t rc;

    ipc_call_t answer;

    aid_t req;

    futex_down(&vfs_phone_futex);

    async_serialize_start();

    if (vfs_phone < 0) {

        res = vfs_connect();

        if (res < 0) {

            async_serialize_end();

            futex_up(&vfs_phone_futex);

            return res;

        }

    }

    req = async_send_1(vfs_phone, VFS_WRITE, fildes, &answer);

    rc = ipc_data_write_start(vfs_phone, (void *)buf, nbyte);

    if (rc != EOK) {

        async_wait_for(req, NULL);

        async_serialize_end();

        futex_up(&vfs_phone_futex);

        return (ssize_t) rc;

    }

    async_wait_for(req, &rc);

    async_serialize_end();

    futex_up(&vfs_phone_futex);

    if (rc == EOK)

        return (ssize_t) IPC_GET_ARG1(answer);

    else

        return -1;

}

off_t lseek(int fildes, off_t offset, int whence)

{

    int res;

    ipcarg_t rc;

    futex_down(&vfs_phone_futex);

    async_serialize_start();

    if (vfs_phone < 0) {

        res = vfs_connect();

        if (res < 0) {

            async_serialize_end();

            futex_up(&vfs_phone_futex);

            return res;

        }

    }

    off_t newoffs;

    rc = async_req_3_1(vfs_phone, VFS_SEEK, fildes, offset, whence,

        (ipcarg_t)&newoffs);

    async_serialize_end();

    futex_up(&vfs_phone_futex);

    if (rc != EOK)

        return (off_t) -1;

    return newoffs;

}

int ftruncate(int fildes, off_t length)

{

    int res;

    ipcarg_t rc;

    futex_down(&vfs_phone_futex);

    async_serialize_start();

    if (vfs_phone < 0) {

        res = vfs_connect();

        if (res < 0) {

            async_serialize_end();

            futex_up(&vfs_phone_futex);

            return res;

        }

    }

    rc = async_req_2_0(vfs_phone, VFS_TRUNCATE, fildes, length);

    async_serialize_end();

    futex_up(&vfs_phone_futex);

    return (int) rc;

}

DIR *opendir(const char *dirname)

{

    DIR *dirp = malloc(sizeof(DIR));

    if (!dirp)

        return NULL;

    dirp->fd = _open(dirname, L_DIRECTORY, 0);

    if (dirp->fd < 0) {

        free(dirp);

        return NULL;

    }

    return dirp;

}

struct dirent *readdir(DIR *dirp)

{

    ssize_t len = read(dirp->fd, &dirp->res.d_name[0], NAME_MAX + 1);

    if (len <= 0)

        return NULL;

    return &dirp->res;

}

void rewinddir(DIR *dirp)

{

    (void) lseek(dirp->fd, 0, SEEK_SET);

}

int closedir(DIR *dirp)

{

    (void) close(dirp->fd);

    free(dirp);

    return 0;

}

int mkdir(const char *path, mode_t mode)

{

    int res;

    ipcarg_t rc;

    aid_t req;

    char *pa = absolutize(path);

    if (!pa)

        return ENOMEM;

    size_t pc_len;

    char *pc = canonify(pa, &pc_len);

    if (!pc) {

        free(pa);

        return EINVAL;

    }

    futex_down(&vfs_phone_futex);

    async_serialize_start();

    if (vfs_phone < 0) {

        res = vfs_connect();

        if (res < 0) {

            async_serialize_end();

            futex_up(&vfs_phone_futex);

            free(pa);

            return res;

        }

    }

    req = async_send_1(vfs_phone, VFS_MKDIR, mode, NULL);

    rc = ipc_data_write_start(vfs_phone, pc, pc_len);

    if (rc != EOK) {

        async_wait_for(req, NULL);

        async_serialize_end();

        futex_up(&vfs_phone_futex);

        free(pa);

        return (int) rc;

    }

    async_wait_for(req, &rc);

    async_serialize_end();

    futex_up(&vfs_phone_futex);

    free(pa);

    return rc;

}

static int _unlink(const char *path, int lflag)

{

    int res;

    ipcarg_t rc;

    aid_t req;

    char *pa = absolutize(path);

    if (!pa)

        return ENOMEM;

    size_t pc_len;

    char *pc = canonify(pa, &pc_len);

    if (!pc) {

        free(pa);

        return EINVAL;

    }

    futex_down(&vfs_phone_futex);

    async_serialize_start();

    if (vfs_phone < 0) {

        res = vfs_connect();

        if (res < 0) {

            async_serialize_end();

            futex_up(&vfs_phone_futex);

            free(pa);

            return res;

        }

    }

    req = async_send_0(vfs_phone, VFS_UNLINK, NULL);

    rc = ipc_data_write_start(vfs_phone, pc, pc_len);

    if (rc != EOK) {

        async_wait_for(req, NULL);

        async_serialize_end();

        futex_up(&vfs_phone_futex);

        free(pa);

        return (int) rc;

    }

    async_wait_for(req, &rc);

    async_serialize_end();

    futex_up(&vfs_phone_futex);

    free(pa);

    return rc;

}

int unlink(const char *path)

{

    return _unlink(path, L_NONE);

}

int rmdir(const char *path)

{

    return _unlink(path, L_DIRECTORY);

}

int chdir(const char *path)

{

    char *pa = absolutize(path);

    if (!pa)

        return ENOMEM;

    size_t pc_len;

    char *pc = canonify(pa, &pc_len);

    if (!pc) {

        free(pa);

        return ENOENT;

    }

    DIR *d = opendir(pc);

    if (!d) {

        free(pa);

        return ENOENT;

    }

    futex_down(&cwd_futex);

    if (cwd_dir) {

        closedir(cwd_dir);

        cwd_dir = NULL;

        free(cwd_path);

        cwd_path = NULL;

        cwd_len = 0;

    }

    cwd_dir = d;

    cwd_path = pc;

    cwd_len = pc_len;

    futex_up(&cwd_futex);

}

char *getcwd(char *buf, size_t size)

{

    if (!size)

        return NULL;

    futex_down(&cwd_futex);

    if (size < cwd_len + 1) {

        futex_up(&cwd_futex);

        return NULL;

    }

    strcpy(buf, cwd_path);

    futex_up(&cwd_futex);

    return buf;

}

/** @}

 */