openmpi/orte/test/system/orte_dfs.c

/* -*- C -*-
 *
 * $HEADER$
 *
 */
#include <stdio.h>
#include <unistd.h>
#include <fcntl.h>

#include "opal/util/output.h"
#include "opal/util/uri.h"
#include "opal/mca/event/event.h"

#include "orte/util/proc_info.h"
#include "orte/util/name_fns.h"
#include "orte/mca/errmgr/errmgr.h"
#include "orte/runtime/orte_globals.h"
#include "orte/runtime/runtime.h"
#include "orte/runtime/orte_wait.h"

#include "orte/mca/dfs/dfs.h"

static bool active;
static bool read_active;
static int numread = 0;

#define READ_SIZE 500
#define OFFSET_VALUE   313

static void dfs_open_cbfunc(int fd, void *cbdata)
{
    int *remote_fd = (int*)cbdata;

    opal_output(0, "GOT FD %d", fd);
    *remote_fd = fd;
    active = false;

}

static void dfs_close_cbfunc(int fd, void *cbdata)
{
    opal_output(0, "CLOSE CONFIRMED");
    active = false;
}

static void dfs_size_cbfunc(long size, void *cbdata)
{
    opal_output(0, "GOT FILE SIZE %ld", size);
    active = false;

}

static void dfs_seek_cbfunc(long offset, void *cbdata)
{
    opal_output(0, "GOT FILE OFFSET %ld vs %d", offset, OFFSET_VALUE);
    active = false;
    if (offset != OFFSET_VALUE) {
        exit(1);
    }

}

static void read_cbfunc(long status, uint8_t *buffer, void *cbdata)
{
    opal_output(0, "GOT READ STATUS %d", (int)status);
    if (status < 0) {
        read_active = false;
        active = false;
        return;
    }
    numread += status;

    if (status < READ_SIZE) {
        read_active = false;
        opal_output(0, "EOF RECEIVED: read total of %d bytes", numread);
        active = false;
        return;
    }
    active = false;
}

int main(int argc, char* argv[])
{
    int rc;
    int fd;
    char *uri, *host;
    uint8_t buffer[READ_SIZE];

    /* user must provide a file to be read - the contents
     * of the file will be output to stdout
     */
    if (1 == argc) {
        fprintf(stderr, "Usage: orte_dfs <input-file> <host [optional]\n");
        exit(1);
    }

    if (0 != (rc = orte_init(&argc, &argv, ORTE_PROC_NON_MPI))) {
        fprintf(stderr, "orte_db: couldn't init orte - error code %d\n", rc);
        return rc;
    }
    
    if (3 == argc) {
        host = strdup(argv[2]);
    } else {
        host = NULL;
    }

    if (NULL == (uri = opal_filename_to_uri(argv[1], host))) {
        return 1;
    }

    active = true;
    orte_dfs.open(uri, dfs_open_cbfunc, &fd);
    ORTE_WAIT_FOR_COMPLETION(active);

    active = true;
    orte_dfs.get_file_size(fd, dfs_size_cbfunc, NULL);
    ORTE_WAIT_FOR_COMPLETION(active);

    active = true;
    read_active = true;
    rc = 0;
    numread = 0;
    while (read_active) {
        opal_output(0, "reading next %d bytes\n", READ_SIZE);
        orte_dfs.read(fd, buffer, READ_SIZE, read_cbfunc, NULL);
        ORTE_WAIT_FOR_COMPLETION(active);
        rc++;
        if (2 == rc) {
            active = true;
            opal_output(0, "execute absolute seek of %d bytes\n", OFFSET_VALUE);
            orte_dfs.seek(fd, OFFSET_VALUE, SEEK_SET, dfs_seek_cbfunc, NULL);
            ORTE_WAIT_FOR_COMPLETION(active);
        }
        if (5 == rc) {
            active = true;
            opal_output(0, "execute relative seek of %d bytes\n", OFFSET_VALUE);
            orte_dfs.seek(fd, OFFSET_VALUE, SEEK_CUR, dfs_seek_cbfunc, NULL);
            ORTE_WAIT_FOR_COMPLETION(active);
        }
        active = true;
    }

    active= true;
    orte_dfs.close(fd, dfs_close_cbfunc, NULL);
    ORTE_WAIT_FOR_COMPLETION(active);

    orte_finalize();
    return 0;
}