Colony » Colony Scout

/*-

 * Copyright (c) 2007, Kohsuke Ohtani

 * All rights reserved.

 *

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

 * modification, are permitted provided that the following conditions

 * are met:

 * 1. Redistributions of source code must retain the above copyright

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

 * 2. 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.

 * 3. Neither the name of the author nor the names of any co-contributors

 *    may be used to endorse or promote products derived from this software

 *    without specific prior written permission.

 *

 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``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 OR CONTRIBUTORS 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.

 */

/*

 * fileio.c - file I/O test program

 */

#include <sys/prex.h>

#include <ipc/ipc.h>

#include <sys/syslog.h>

#include <sys/mount.h>

#include <sys/fcntl.h>

#include <unistd.h>

#include <string.h>

#include <stdio.h>

#include <errno.h>

#define IOBUFSZ        512

#define READ_TARGET        "/boot/LICENSE"

#define WRITE_TARGET        "/tmp/test"

static        char iobuf[IOBUFSZ];

static void

test_write(void)

{

        int fd, i;

        if ((fd = open(WRITE_TARGET, O_CREAT|O_RDWR, 0)) < 0)

                panic("can not open file " WRITE_TARGET);

        for (i = 0; i < 50; i++) {

                memset(iobuf, i, IOBUFSZ);

                write(fd, iobuf, IOBUFSZ);

        }

        close(fd);

}

/*

 * Display file contents

 */

static void

cat_file(void)

{

        int rd, fd;

        if ((fd = open(READ_TARGET, O_RDONLY, 0)) < 0)

                panic("can not open file " READ_TARGET);

        while ((rd = read(fd, iobuf, IOBUFSZ)) > 0)

                write(STDOUT_FILENO, iobuf, (size_t)rd);

        close(fd);

}

/*

 * Test invalid request

 */

static void

test_invalid(void)

{

        object_t fsobj;

        struct msg m;

        object_lookup("!fs", &fsobj);

        m.hdr.code = 0x300;

        msg_send(fsobj, &m, sizeof(m));

}

/*

 * Test open/close

 */

static void

test_open(void)

{

        int fd;

        for (;;) {

                if ((fd = open(READ_TARGET, O_RDONLY, 0)) < 0)

                        panic("can not open file " READ_TARGET);

                close(fd);

        }

}

/*

 * Test file read

 */

static void

test_read(void)

{

        int rd, fd;

        if ((fd = open(READ_TARGET, O_RDONLY, 0)) < 0)

                panic("can not open file " READ_TARGET);

        for (;;)

                while ((rd = read(fd, iobuf, IOBUFSZ)) > 0)

                        ;

        close(fd);

}

/*

 * Main routine

 */

int

main(int argc, char *argv[])

{

        char test_str[] = "test stdout...\n\n";

        syslog(LOG_INFO, "\nfileio: fs test program\n");

        /* Wait 1 sec until loading fs server */

        timer_sleep(1000, 0);

        /*

         * Prepare to use a file system.

         */

        fslib_init();

        /*

         * Mount file systems

         */

        mount("", "/", "ramfs", 0, NULL);

        mkdir("/dev", 0);

        mount("", "/dev", "devfs", 0, NULL);                /* device */

        mkdir("/boot", 0);

        mount("/dev/ram0", "/boot", "arfs", 0, NULL);        /* archive */

        mkdir("/tmp", 0);

        /*

         * Prepare stdio

         */

        open("/dev/tty", O_RDWR);        /* stdin */

        dup(0);                                /* stdout */

        dup(0);                                /* stderr */

        /* Test device write */

        write(STDOUT_FILENO, test_str, strlen(test_str));

        test_write();

        cat_file();                /* test read/write */

        test_invalid();                /* test invalid request */

        test_read();                /* test read loop */

        test_open();                /* test open/close loop */

        /*

         * Disconnect from a file system.

         */

        fslib_exit();

        return 0;

}