summaryrefslogtreecommitdiffstats
path: root/tools/h5jam/h5jam.c
diff options
context:
space:
mode:
authorRobert E. McGrath <mcgrath@ncsa.uiuc.edu>2004-08-05 15:10:39 (GMT)
committerRobert E. McGrath <mcgrath@ncsa.uiuc.edu>2004-08-05 15:10:39 (GMT)
commitf53299c0b33a767f58c9e4fb50b4d7b3f2f0c5c2 (patch)
tree6f904201828a301e670785bada08ad2ea19ad687 /tools/h5jam/h5jam.c
parent5390a63a77772ef0f91a0fd286128be9d5a84e8c (diff)
downloadhdf5-f53299c0b33a767f58c9e4fb50b4d7b3f2f0c5c2.zip
hdf5-f53299c0b33a767f58c9e4fb50b4d7b3f2f0c5c2.tar.gz
hdf5-f53299c0b33a767f58c9e4fb50b4d7b3f2f0c5c2.tar.bz2
[svn-r9019] Purpose:
Adding new 'jam' utility Description: New utility, plus changes to makefiles Solution: See http://hdf.ncsa.uiuc.edu/RFC/Jam Platforms tested: verbena (fortran,C++), arabica, hirdls (SGI Irix64) Misc. update: Manifest will be done in next checkin.
Diffstat (limited to 'tools/h5jam/h5jam.c')
-rw-r--r--tools/h5jam/h5jam.c534
1 files changed, 534 insertions, 0 deletions
diff --git a/tools/h5jam/h5jam.c b/tools/h5jam/h5jam.c
new file mode 100644
index 0000000..9528b0a
--- /dev/null
+++ b/tools/h5jam/h5jam.c
@@ -0,0 +1,534 @@
+/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
+ * Copyright by the Board of Trustees of the University of Illinois. *
+ * All rights reserved. *
+ * *
+ * This file is part of HDF5. The full HDF5 copyright notice, including *
+ * terms governing use, modification, and redistribution, is contained in *
+ * the files COPYING and Copyright.html. COPYING can be found at the root *
+ * of the source code distribution tree; Copyright.html can be found at the *
+ * root level of an installed copy of the electronic HDF5 document set and *
+ * is linked from the top-level documents page. It can also be found at *
+ * http://hdf.ncsa.uiuc.edu/HDF5/doc/Copyright.html. If you do not have *
+ * access to either file, you may request a copy from hdfhelp@ncsa.uiuc.edu. *
+ * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <fcntl.h>
+#include <sys/stat.h>
+#include <unistd.h>
+
+#include "hdf5.h"
+#include "h5tools_utils.h"
+
+#define TRUE 1
+#define FALSE 0
+
+hsize_t write_pad (int, hsize_t);
+hsize_t compute_user_block_size (hsize_t);
+hsize_t copy_some_to_file (int, int, hsize_t, hsize_t, ssize_t);
+void parse_command_line (int, const char *[]);
+
+const char *progname = "jam";
+int d_status = EXIT_SUCCESS;
+int do_clobber = FALSE;
+char *output_file = NULL;
+char *input_file = NULL;
+char *ub_file = NULL;
+
+/*
+ * Command-line options: The user can specify short or long-named
+ * parameters. The long-named ones can be partially spelled. When
+ * adding more, make sure that they don't clash with each other.
+ */
+static const char *s_opts = "hi:u:o:c"; /* add more later ? */
+static struct long_options l_opts[] = {
+ {"help", no_arg, 'h'},
+ {"hel", no_arg, 'h'},
+ {"i", require_arg, 'i'}, /* input file */
+ {"u", require_arg, 'u'}, /* user block file */
+ {"o", require_arg, 'o'}, /* output file */
+ {"clobber", no_arg, 'c'}, /* clobber existing UB */
+ {"clobbe", no_arg, 'c'},
+ {"clobb", no_arg, 'c'},
+ {"clob", no_arg, 'c'},
+ {"clo", no_arg, 'c'},
+ {"cl", no_arg, 'c'},
+ {NULL, 0, '\0'}
+};
+
+/*-------------------------------------------------------------------------
+ * Function: usage
+ *
+ * Purpose: Print the usage message
+ *
+ * Return: void
+ *
+ * Programmer:
+ *
+ * Modifications:
+ *
+ *-------------------------------------------------------------------------
+ */
+static void
+usage (const char *prog)
+{
+ fflush (stdout);
+ fprintf (stdout,
+ "usage: %s -u user_block_file -i h5_file [-o ofile | --clobber] \n",
+ prog);
+ fprintf (stdout, " Add 'user_block_file' to front of \n");
+ fprintf (stdout,
+ " 'h5_file', pad so 'h5_file' starts on proper\n");
+ fprintf (stdout, " byte.\n");
+ fprintf (stdout, "\n");
+ fprintf (stdout, " %s -h \n", prog);
+ fprintf (stdout, " Print a usage message and exit\n");
+}
+
+/*-------------------------------------------------------------------------
+ * Function: parse_command_line
+ *
+ * Purpose: Parse the command line for the h5dumper.
+ *
+ * Return: Success:
+ *
+ * Failure: Exits program with EXIT_FAILURE value.
+ *
+ * Programmer:
+ *
+ * Modifications:
+ *
+ *-------------------------------------------------------------------------
+ */
+
+void
+parse_command_line (int argc, const char *argv[])
+{
+ int opt = FALSE;
+
+ /* parse command line options */
+ while ((opt = get_option (argc, argv, s_opts, l_opts)) != EOF)
+ {
+ switch ((char) opt)
+ {
+ case 'o':
+ output_file = strdup (opt_arg);
+ break;
+ case 'i':
+ input_file = strdup (opt_arg);
+ break;
+ case 'u':
+ ub_file = strdup (opt_arg);
+ break;
+ case 'c':
+ do_clobber = TRUE;
+ break;
+ case 'h':
+ usage (progname);
+ exit (EXIT_SUCCESS);
+ case '?':
+ default:
+ usage (progname);
+ exit (EXIT_FAILURE);
+ }
+ }
+}
+
+/*-------------------------------------------------------------------------
+ * Function: main
+ *
+ * Purpose: HDF5 user block jammer
+ *
+ * Return: Success: 0
+ * Failure: 1
+ *
+ * Programmer:
+ *
+ * Modifications:
+ *
+ *-------------------------------------------------------------------------
+ */
+int
+main (int argc, const char *argv[])
+{
+ int ufid;
+ int h5fid;
+ int ofid;
+ void *edata;
+ hid_t (*func) (void *);
+ hid_t ifile;
+ hid_t plist;
+ herr_t status;
+ htri_t testval;
+ hsize_t usize;
+ hsize_t h5fsize;
+ hsize_t startub;
+ hsize_t where;
+ hsize_t newubsize;
+ off_t fsize;
+ struct stat sbuf;
+ struct stat sbuf2;
+ int res;
+
+ /* Disable error reporting */
+ H5Eget_auto (H5E_DEFAULT, &func, &edata);
+ H5Eset_auto (H5E_DEFAULT, NULL, NULL);
+
+ parse_command_line (argc, argv);
+
+ if (ub_file == NULL)
+ {
+ /* no user block */
+ error_msg (progname, "no user block file name\n");
+ usage (progname);
+ exit (EXIT_FAILURE);
+ }
+
+ if (input_file == NULL)
+ {
+ /* no user block */
+ error_msg (progname, "no HDF5 file\n");
+ usage (progname);
+ exit (EXIT_FAILURE);
+ }
+
+ testval = H5Fis_hdf5 (input_file);
+
+ if (testval <= 0)
+ {
+ error_msg (progname, "Input HDF5 file is not HDF \"%s\"\n", input_file);
+ exit (EXIT_FAILURE);
+ }
+
+ ifile = H5Fopen (input_file, H5F_ACC_RDONLY, H5P_DEFAULT);
+
+ if (ifile < 0)
+ {
+ error_msg (progname, "Can't open input HDF5 file \"%s\"\n", input_file);
+ exit (EXIT_FAILURE);
+ }
+
+ plist = H5Fget_create_plist (ifile);
+ if (plist < 0)
+ {
+ error_msg (progname, "Can't get file creation plist for file \"%s\"\n",
+ input_file);
+ exit (EXIT_FAILURE);
+ }
+
+ status = H5Pget_userblock (plist, &usize);
+ if (status < 0)
+ {
+ error_msg (progname, "Can't get user block for file \"%s\"\n",
+ input_file);
+ exit (EXIT_FAILURE);
+ }
+
+ H5Pclose (plist);
+ H5Fclose (ifile);
+
+ ufid = open (ub_file, O_RDONLY);
+
+ if (ufid < 0)
+ {
+ error_msg (progname, "unable to open user block file \"%s\"\n",
+ ub_file);
+ exit (EXIT_FAILURE);
+ }
+
+ res = stat (ub_file, &sbuf);
+
+ if (res < 0)
+ {
+ error_msg (progname, "Can't stat file \"%s\"\n", ub_file);
+ exit (EXIT_FAILURE);
+ }
+
+ fsize = sbuf.st_size;
+
+ h5fid = open (input_file, O_RDONLY);
+
+ if (h5fid < 0)
+ {
+ error_msg (progname, "unable to open HDF5 file for read \"%s\"\n",
+ input_file);
+ exit (EXIT_FAILURE);
+ }
+
+ res = stat (input_file, &sbuf2);
+
+ if (res < 0)
+ {
+ error_msg (progname, "Can't stat file \"%s\"\n", input_file);
+ exit (EXIT_FAILURE);
+ }
+
+ h5fsize = sbuf2.st_size;
+
+ if (output_file == NULL)
+ {
+ ofid = open (input_file, O_WRONLY);
+
+ if (ofid < 0)
+ {
+ error_msg (progname, "unable to open output file \"%s\"\n",
+ output_file);
+ exit (EXIT_FAILURE);
+ }
+ }
+ else
+ {
+ ofid = open (output_file, O_WRONLY | O_CREAT | O_TRUNC, 0644);
+
+ if (ofid < 0)
+ {
+ error_msg (progname, "unable to create output file \"%s\"\n",
+ output_file);
+ exit (EXIT_FAILURE);
+ }
+ }
+
+ newubsize = compute_user_block_size ((hsize_t) fsize);
+
+ startub = usize;
+
+ if (usize > 0)
+ {
+ if (do_clobber == TRUE)
+ {
+ /* where is max of the current size or the new UB */
+ if (usize > newubsize)
+ {
+ newubsize = usize;
+ }
+ startub = 0; /*blast the old */
+ }
+ else
+ {
+ /* add new ub to current ublock, pad to new offset */
+ newubsize += usize;
+ newubsize = compute_user_block_size ((hsize_t) newubsize);
+ }
+ }
+
+ /* copy the HDF5 from starting at usize to starting at newubsize:
+ * makes room at 'from' for new ub */
+ /* if no current ub, usize is 0 */
+ copy_some_to_file (h5fid, ofid, usize, newubsize,
+ (ssize_t) (h5fsize - usize));
+
+ /* copy the old ub to the beginning of the new file */
+ if (!do_clobber)
+ {
+ where =
+ copy_some_to_file (h5fid, ofid, (hsize_t) 0, (hsize_t) 0,
+ (ssize_t) usize);
+ }
+
+ /* copy the new ub to the end of the ub */
+ where = copy_some_to_file (ufid, ofid, (hsize_t) 0, startub, (ssize_t) - 1);
+
+ /* pad the ub */
+ where = write_pad (ofid, where);
+
+
+ close (ufid);
+ close (h5fid);
+ close (ofid);
+
+ return d_status;
+}
+
+/*-------------------------------------------------------------------------
+ * Function: copy_some_to_file
+ *
+ * Purpose: Copy part of the input file to output.
+ * infid: fd of file to read
+ * outfid: fd of file to write
+ * startin: offset of where to read from infid
+ * startout: offset of where to write to outfid
+ * limit: bytes to read/write
+ *
+ * If limit is < 0, the entire input file is copied.
+ *
+ * Note: this routine can be used to copy within
+ * the same file, i.e., infid and outfid can be the
+ * same file.
+ *
+ * Return: Success: last byte written in the output.
+ * Failure: Exits program with EXIT_FAILURE value.
+ *
+ * Programmer:
+ *
+ * Modifications:
+ *
+ *-------------------------------------------------------------------------
+ */
+hsize_t
+copy_some_to_file (int infid, int outfid, hsize_t startin, hsize_t startout,
+ ssize_t limit)
+{
+ char buf[1024];
+ struct stat sbuf;
+ int res;
+ ssize_t tot = 0;
+ ssize_t howmuch = 0;
+ ssize_t nchars = -1;
+/* used in assertion check
+ ssize_t ncw = -1;
+*/
+ ssize_t to;
+ ssize_t from;
+ ssize_t toend;
+ ssize_t fromend;
+
+ if (startin > startout)
+ {
+ /* this case is prohibited */
+ error_msg (progname, "copy_some_to_file: panic: startin > startout?\n");
+ exit (EXIT_FAILURE);
+ }
+
+ if (limit < 0)
+ {
+ res = fstat (infid, &sbuf);
+
+ if (res < 0)
+ {
+ error_msg (progname, "Can't stat file \n");
+ exit (EXIT_FAILURE);
+ }
+
+ howmuch = sbuf.st_size;
+ }
+ else
+ {
+ howmuch = limit;
+ }
+
+ if (howmuch == 0)
+ {
+ return 0;
+ }
+
+ /* assert (howmuch > 0) */
+
+ toend = (ssize_t) startout + howmuch;
+ fromend = (ssize_t) startin + howmuch;
+
+ if (howmuch > 512)
+ {
+ to = toend - 512;
+ from = fromend - 512;
+ }
+ else
+ {
+ to = toend - howmuch;
+ from = fromend - howmuch;
+ }
+
+ while (howmuch > 0)
+ {
+ lseek (outfid, (off_t) to, SEEK_SET);
+ lseek (infid, (off_t) from, SEEK_SET);
+
+ if (howmuch > 512)
+ {
+ nchars = read (infid, buf, (unsigned) 512);
+ }
+ else
+ {
+ nchars = read (infid, buf, howmuch);
+ }
+ if (nchars <= 0)
+ {
+ printf ("huh? \n");
+ exit (1);
+ }
+ /*ncw = */ write (outfid, buf, (unsigned) nchars);
+
+ /* assert (ncw == nchars) */
+
+ tot += nchars;
+ howmuch -= nchars;
+ if (howmuch > 512)
+ {
+ to -= nchars;
+ from -= nchars;
+ }
+ else
+ {
+ to -= howmuch;
+ from -= howmuch;
+ }
+ }
+
+ /* assert howmuch == 0 */
+ /* assert tot == limit */
+
+ return ((hsize_t) tot + (hsize_t) startout);
+}
+
+
+/*-------------------------------------------------------------------------
+ * Function: compute_user_block_size
+ *
+ * Purpose: Find the offset of the HDF5 header after the user block:
+ * align at 0, 512, 1024, etc.
+ * ublock_size: the size of the user block (bytes).
+ *
+ * Return: Success: the location of the header == the size of the
+ * padded user block.
+ * Failure: none
+ *
+ * Return: Success: last byte written in the output.
+ * Failure: Exits program with EXIT_FAILURE value.
+ *
+ * Programmer:
+ *
+ * Modifications:
+ *
+ *-------------------------------------------------------------------------
+ */
+hsize_t
+compute_user_block_size (hsize_t ublock_size)
+{
+ ssize_t where = 512;
+
+ if (ublock_size == 0)
+ return 0;
+
+ while (where < ublock_size)
+ {
+ where *= 2;
+ }
+
+ return (where);
+}
+
+/*
+ * Write zeroes to fill the file from 'where' to 512, 1024, etc. bytes.
+ *
+ * Returns the size of the padded file.
+ */
+hsize_t
+write_pad (int ofile, hsize_t where)
+{
+ unsigned int i;
+ char buf[1];
+ hsize_t psize;
+
+ buf[0] = '\0';
+
+ lseek (ofile, (off_t) where, SEEK_SET);
+
+ psize = compute_user_block_size (where);
+ psize -= where;
+
+ for (i = 0; i < psize; i++)
+ {
+ write (ofile, buf, 1);
+ }
+ return (where + psize); /* the new size of the file. */
+}