p-testpipe.cc

#include "u-lib.hh"

void process_main() {
    console_printf("creating pipes...\n");

    int pfd[2] = {-1, -1};
    int x = sys_pipe(pfd);
    assert_eq(x, 0);
    assert(pfd[0] > 2 && pfd[1] > 2);
    assert(pfd[0] != pfd[1]);

    int qfd[2] = {-1, -1};
    x = sys_pipe(qfd);
    assert_eq(x, 0);
    assert(qfd[0] > 2 && qfd[1] > 2);
    assert(qfd[0] != qfd[1]);
    assert(pfd[0] != qfd[0] && pfd[0] != qfd[1]);
    assert(pfd[1] != qfd[0] && pfd[1] != qfd[1]);

    console_printf("pfd %d,%d, qfd %d,%d\n", pfd[0], pfd[1], qfd[0], qfd[1]);


    // simple tests
    console_printf("simple tests...\n");

    ssize_t n = sys_write(pfd[1], "pfd1", 4);
    assert_eq(n, 4);

    n = sys_write(qfd[1], "qfd1", 4);
    assert_eq(n, 4);

    char buf[400];
    n = sys_read(pfd[0], buf, 2);
    assert_eq(n, 2);
    assert_memeq(buf, "pf", 2);

    n = sys_read(pfd[0], buf, 8);
    assert_eq(n, 2);
    assert_memeq(buf, "d1", 2);

    n = sys_read(qfd[0], buf, 100);
    assert_eq(n, 4);
    assert_memeq(buf, "qfd1", 4);


    // interleaving tests
    console_printf("interleaving tests...\n");

    n = sys_write(pfd[1], "W1", 2);
    assert_eq(n, 2);

    n = sys_write(qfd[1], "w1", 2);
    assert_eq(n, 2);

    n = sys_write(qfd[1], "w2", 2);
    assert_eq(n, 2);

    n = sys_write(pfd[1], "W2", 2);
    assert_eq(n, 2);

    n = sys_write(pfd[1], "W3!", 3);
    assert_eq(n, 3);

    n = sys_write(qfd[1], "w3!", 3);
    assert_eq(n, 3);

    n = sys_write(pfd[1], "W4!!!", 5);
    assert_eq(n, 5);

    n = sys_write(qfd[1], "w4!!!", 5);
    assert_eq(n, 5);

    n = sys_read(pfd[0], buf, 100);
    assert_eq(n, 12);
    assert_memeq(buf, "W1W2W3!W4!!!", 12);

    n = sys_read(qfd[0], buf, 100);
    assert_eq(n, 12);
    assert_memeq(buf, "w1w2w3!w4!!!", 12);


    // can't read from write end or write to read end
    console_printf("read/write error tests...\n");

    n = sys_read(pfd[1], buf, 8);
    assert_eq(n, E_BADF);

    n = sys_write(qfd[0], buf, 1);
    assert_eq(n, E_BADF);


    // inheritance tests
    console_printf("child tests...\n");

    pid_t p = sys_fork();
    assert_ge(p, 0);

    if (p == 0) {
        x = sys_close(pfd[1]);
        assert_eq(x, 0);

        x = sys_close(qfd[0]);
        assert_eq(x, 0);

        x = sys_close(pfd[1]);
        assert_eq(x, E_BADF);

        n = sys_write(qfd[1], "hello mom", 9);
        assert_eq(n, 9);

        n = sys_read(pfd[0], buf, 100);
        assert_eq(n, 5);
        assert_memeq(buf, "hello", 5);

        n = sys_read(pfd[0], buf, 100);
        assert_eq(n, 5);
        assert_memeq(buf, " babe", 5);

        x = sys_msleep(300);
        assert_eq(x, 0);

        x = sys_close(pfd[0]);
        assert_eq(x, 0);

        x = sys_close(qfd[1]);
        assert_eq(x, 0);

        sys_exit(0);
    }

    x = sys_close(pfd[0]);
    assert_eq(x, 0);

    x = sys_close(qfd[1]);
    assert_eq(x, 0);

    n = sys_read(qfd[0], buf, 100);
    assert_eq(n, 9);
    assert_memeq(buf, "hello mom", 9);

    n = sys_write(pfd[1], "hello", 5);
    assert_eq(n, 5);

    sys_msleep(300);

    n = sys_write(pfd[1], " babe", 5);
    assert_eq(n, 5);

    n = sys_read(qfd[0], buf, 100);
    assert_eq(n, 0);

    n = sys_write(pfd[1], "wharg", 5);
    assert_eq(n, E_PIPE);

    x = sys_close(qfd[0]);
    assert_eq(x, 0);

    x = sys_close(pfd[1]);
    assert_eq(x, 0);

    sys_msleep(100); // try to ensure first child exits


    // inheritance tests with close-on-exit
    console_printf("close-on-exit tests...\n");

    x = sys_pipe(pfd);
    assert(x == 0 && pfd[0] > 2 && pfd[1] > 2 && pfd[0] != pfd[1]);

    n = sys_write(pfd[1], "hello", 5);
    assert_eq(n, 5);

    p = sys_fork();
    assert_ge(p, 0);

    if (p == 0) {
        n = sys_read(pfd[0], buf, 100);
        assert_eq(n, 5);
        assert_memeq(buf, "hello", 5);

        x = sys_msleep(400);
        assert_eq(x, 0);

        sys_exit(0);
    }

    x = sys_msleep(100);
    assert_eq(x, 0);

    x = sys_close(pfd[1]);
    assert_eq(x, 0);

    n = sys_read(pfd[0], buf, 100);
    assert_eq(n, 0);


    console_printf("testpipe succeeded.\n");
    sys_exit(0);
}