diff options
| -rw-r--r-- | test/stream_test.c | 145 |
1 files changed, 116 insertions, 29 deletions
diff --git a/test/stream_test.c b/test/stream_test.c index a62c5cb..aecf656 100644 --- a/test/stream_test.c +++ b/test/stream_test.c @@ -17,15 +17,20 @@ /* * This program tests the functionality of the Stream Virtual File Driver. * 1. It spawns two new processes, a sender and a receiver. - * 2. The sender opens an HDF5 file for writing and writes - * a sample dataset to it. + * 2. The sender opens an HDF5 file for writing using the Stream driver. + * It will use a reserved port which should fail to be bound. + * Then it will try a couple of successive ports until bind succeeds. + * This final "hostname:port" information is written into a temporary + * file as a single line of text. + * The sender then writes a sample dataset to the HDF5 file. * On closing the file the Stream VFD would send the file * contents to any connected client. * 3. The receiver serves as a client attempting to open an * HDF5 file for reading. On opening the file the Stream VFD * would establish a socket connection to the sender process, - * identified by its hostname (which is localhost in this example) - * and a port number, and read the file contents via this socket. + * identified by its hostname and a port number (which is obtained + * from the temporary text file the sender should have created), + * and read the file contents via this socket. * Aftwerwards the dataset is read from the file into memory * and verified. * 4. The main program waits for termination of its two child @@ -60,11 +65,15 @@ int main (void) #include <sys/wait.h> -#define DELAY 10 /* sleeping time in seconds */ -#define RANK 2 /* sample dataset rank */ -#define DIMS 50 /* sample dataset dimensions */ -#define DATASETNAME "IntArray" /* sample dataset name */ -#define FILENAME "localhost:10007" /* filename argument */ +#define SLEEPTIME 10 /* sleeping time in seconds */ +#define RANK 2 /* sample dataset rank */ +#define DIMS 50 /* sample dataset dimensions */ +#define DATASETNAME "IntArray" /* sample dataset name */ +#define HOSTNAME "localhost" /* hostname of this machine */ +#define PORT "5678" /* default port to use */ +#define MAXHUNT 500 /* max number of ports to hunt */ +#define HDF5_FILENAME HOSTNAME ":" PORT /* name of the streamed file */ +#define TEMPFILENAME "stream_test.tmp" /* temporary filename */ static int sender (void) @@ -75,6 +84,8 @@ static int sender (void) herr_t status; hid_t fapl, file; hid_t dataspace, dataset; + H5FD_stream_fapl_t stream_fapl; + FILE *tempfile; /* @@ -87,7 +98,26 @@ static int sender (void) return (-1); } - status = H5Pset_fapl_stream (fapl, NULL); + /* + * Setup file access property list and select Stream VFD. + * + * - block increment for realloc() should be chosen by the driver + * - no external socket is provided (should be created internally) + * - do I/O on this processor on this socket + * - only one client is allowed to connect at a time + * - no READ broadcast function is provided (since we only send data) + * - if bind to default port (given in the filename argument) fails + * do port hunting on the following MAXHUNT ports + */ + stream_fapl.increment = 0; + stream_fapl.socket = H5FD_STREAM_INVALID_SOCKET; + stream_fapl.do_socket_io = 1; + stream_fapl.backlog = 1; + stream_fapl.broadcast_fn = NULL; + stream_fapl.broadcast_arg = NULL; + stream_fapl.maxhunt = MAXHUNT; + + status = H5Pset_fapl_stream (fapl, &stream_fapl); if (status < 0) { fprintf (stderr, "sender: couldn't set file access property list " @@ -134,11 +164,13 @@ static int sender (void) * default file creation properties, and STREAM file * access properties. */ - printf (" sender: opening file '%s' for writing...\n", FILENAME); - file = H5Fcreate (FILENAME, H5F_ACC_TRUNC, H5P_DEFAULT, fapl); + printf (" sender: opening file on host '%s' port %s for writing...\n", + HOSTNAME, PORT); + file = H5Fcreate (HDF5_FILENAME, H5F_ACC_TRUNC, H5P_DEFAULT, fapl); if (file < 0) { - fprintf (stderr, "sender: couldn't create file for '%s'\n", FILENAME); + fprintf (stderr, "sender: couldn't create file on '%s' using port %s and " + "following %d\n", HOSTNAME, PORT, MAXHUNT); free (data); H5Sclose (dataspace); H5Pclose (fapl); @@ -146,6 +178,38 @@ static int sender (void) } /* + * Get the file access property list to find out what port is actually used. + */ + status = H5Pget_fapl_stream (fapl, &stream_fapl); + if (status < 0) + { + fprintf (stderr, "sender: couldn't get file access property list " + "for Stream VFD\n"); + free (data); + H5Sclose (dataspace); + H5Pclose (fapl); + return (-6); + } + printf (" sender: using port %d...\n", (int) stream_fapl.port); + + /* + * Write the "hostname:port" information to a temporary file + * which can be read by the receiver process. + */ + tempfile = fopen (TEMPFILENAME, "w"); + if (tempfile == NULL) + { + fprintf (stderr, "sender: couldn't open temporary file to write " + "\"hostname:port\" information\n"); + free (data); + H5Sclose (dataspace); + H5Pclose (fapl); + return (-7); + } + fprintf (tempfile, "%s:%d", HOSTNAME, (int) stream_fapl.port); + fclose (tempfile); + + /* * Create a new dataset within the file using defined dataspace and * default dataset creation properties. */ @@ -158,14 +222,14 @@ static int sender (void) H5Fclose (file); H5Sclose (dataspace); H5Pclose (fapl); - return (-6); + return (-8); } /* * Write the data to the dataset using default transfer properties. */ - printf (" sender: writing dataset '%s' of type INTEGER to file '%s'...\n", - DATASETNAME, FILENAME); + printf (" sender: writing dataset '%s' of type INTEGER to file '%s:%d'...\n", + DATASETNAME, HOSTNAME, (int) stream_fapl.port); status = H5Dwrite (dataset, H5T_NATIVE_INT, H5S_ALL, H5S_ALL, H5P_DEFAULT, data); if (status < 0) @@ -176,16 +240,16 @@ static int sender (void) H5Sclose (dataspace); H5Pclose (fapl); fprintf (stderr, "sender: couldn't write dataset\n"); - return (-7); + return (-9); } /* * Now give the receiver some time to connect before closing the file * and releasing resources. */ - printf (" sender: sleeping for %d seconds...\n", DELAY); - sleep (DELAY); - printf (" sender: closing file '%s'\n", FILENAME); + printf (" sender: sleeping for %d seconds...\n", SLEEPTIME); + sleep (SLEEPTIME); + printf (" sender: closing file '%s:%d'\n", HOSTNAME, (int) stream_fapl.port); H5Sclose (dataspace); H5Dclose (dataset); H5Fclose (file); @@ -210,6 +274,8 @@ static int receiver (void) int *data; /* read buffer */ int nerrors; /* total number of errors during verify */ int status; /* return code of HDF5 routines */ + char filename[50]; /* filename of the streamed HDF5 file */ + FILE *tempfile; /* descriptor for temporary file */ /* @@ -227,21 +293,42 @@ static int receiver (void) { fprintf (stderr, "receiver: couldn't set file access property list " "for Stream VFD\n"); + H5Pclose (fapl); return (-2); } /* * Now give the sender some time to open the file and accepting connections. */ - printf (" receiver: sleeping for %d seconds...\n", DELAY / 2); - sleep (DELAY / 2); - printf (" receiver: opening file '%s' for reading...\n", FILENAME); - file = H5Fopen (FILENAME, H5F_ACC_RDONLY, fapl); + printf (" receiver: sleeping for %d seconds...\n", SLEEPTIME / 2); + sleep (SLEEPTIME / 2); + + /* + * Read the "hostname:port" information from the temporary file + * the sender should have created. + */ + tempfile = fopen (TEMPFILENAME, "r"); + if (tempfile == NULL) + { + fprintf (stderr, "receiver: couldn't open temporary file to read " + "\"hostname:port\" information\n"); + H5Pclose (fapl); + return (-3); + } + fgets (filename, sizeof (filename) - 1, tempfile); + fclose (tempfile); + unlink (TEMPFILENAME); + + /* + * Open the streamed HDF5 file for reading. + */ + printf (" receiver: opening file '%s' for reading...\n", filename); + file = H5Fopen (filename, H5F_ACC_RDONLY, fapl); H5Pclose (fapl); if (file < 0) { - fprintf (stderr, "receiver: couldn't open file from '%s'\n", FILENAME); - return (-3); + fprintf (stderr, "receiver: couldn't open file from '%s'\n", filename); + return (-4); } /* @@ -252,7 +339,7 @@ static int receiver (void) if (dataset < 0) { fprintf (stderr, "receiver: couldn't open dataset '%s'\n", DATASETNAME); - return (-4); + return (-5); } /* @@ -291,7 +378,7 @@ static int receiver (void) /* * Read dataset from file into memory. */ - data = (int *) malloc ((size_t)(nelems * sizeof (int))); + data = (int *) malloc ((size_t) nelems * sizeof (int)); status = H5Dread (dataset, H5T_NATIVE_INT, H5S_ALL, H5S_ALL, H5P_DEFAULT, data); H5Dclose (dataset); @@ -299,7 +386,7 @@ static int receiver (void) /* * Close the file. */ - printf (" receiver: closing file '%s'...\n", FILENAME); + printf (" receiver: closing file '%s'...\n", filename); H5Fclose (file); /* |