diff options
Diffstat (limited to 'tools/h5jam/h5unjam.c')
-rw-r--r-- | tools/h5jam/h5unjam.c | 317 |
1 files changed, 317 insertions, 0 deletions
diff --git a/tools/h5jam/h5unjam.c b/tools/h5jam/h5unjam.c new file mode 100644 index 0000000..20378e5 --- /dev/null +++ b/tools/h5jam/h5unjam.c @@ -0,0 +1,317 @@ +/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * + * 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 <sys/types.h> +#include <fcntl.h> +#include <sys/stat.h> + +#ifdef H5_HAVE_UNISTD_H +#include <unistd.h> +#endif + +#include "hdf5.h" +#include "H5private.h" +#include "h5tools_utils.h" + +#define TRUE 1 +#define FALSE 0 + +hsize_t write_pad( int , hsize_t ); +hsize_t compute_pad( hsize_t ); +hsize_t copy_to_file( int , int , ssize_t, ssize_t ); + +const char *progname = "unjam"; +int d_status = EXIT_SUCCESS; +int do_delete = 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 = "hu:i:o:d"; +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 */ + {"delete", no_arg, 'd'}, /* delete ub */ + {"delet", no_arg, 'd'}, + {"dele", no_arg, 'd'}, + {"del", no_arg, 'd'}, + {"de", no_arg, 'd'}, + { 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 -i h5_file -o user_block_file_out -o h5_file_out [-d | --delete]\n", prog); + fprintf(stdout, " Extract user block from 'h5_file' into 'user_block_file'\n"); + fprintf(stdout, " and HDF5 file into 'h5_file_out'\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 'd': + do_delete = TRUE; + break; + case 'h': + usage(progname); + exit(EXIT_SUCCESS); + case '?': + default: + usage(progname); + exit(EXIT_FAILURE); + } + } + + /* check for file name to be processed */ +/* + if (argc <= opt_ind+2) { + error_msg(progname, "missing file name\n"); + usage(progname); + exit(EXIT_FAILURE); + } +*/ +} + +/*------------------------------------------------------------------------- + * Function: main + * + * Purpose: HDF5 user block unjammer + * + * Return: Success: 0 + * Failure: 1 + * + * Programmer: + * + * Modifications: + * + *------------------------------------------------------------------------- + */ +int +main(int argc, const char *argv[]) +{ + int ifid; + int ufid; + int h5fid; + void *edata; + H5E_auto_stack_t func; + hid_t ifile; + off_t fsize; + hsize_t usize; + htri_t testval; + herr_t status; + hid_t plist; + int res; + struct stat sbuf; + + /* Disable error reporting */ + H5Eget_auto_stack(H5E_DEFAULT, &func, &edata); + H5Eset_auto_stack(H5E_DEFAULT, NULL, NULL); + + parse_command_line(argc, argv); + + 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); + } + + if (usize == 0) { + /* no user block to remove: message? */ + error_msg(progname, "\"%s\" has no user block: no change to file\n", input_file); + exit(EXIT_SUCCESS); + + } + + res = stat(input_file, &sbuf); + + if (res < 0) { + error_msg(progname, "Can't stat file \"%s\"\n", input_file); + exit(EXIT_FAILURE); + } + + fsize = sbuf.st_size; + + ifid = HDopen(input_file,O_RDONLY,0); + + if (ifid < 0) { + error_msg(progname, "unable to open input HDF5 file \"%s\"\n", input_file); + exit(EXIT_FAILURE); + } + + if (do_delete && (ub_file != NULL)) { + error_msg(progname, "??\"%s\"\n", ub_file); + exit(EXIT_FAILURE); + } + + if (ub_file == NULL) { + /* write to sdtout */ + ufid = dup(1); + } else { + ufid = HDopen(ub_file,O_WRONLY|O_CREAT|O_TRUNC, 0644 ); + + if (ufid < 0) { + error_msg(progname, "unable to open user block file for output\"%s\"\n", ub_file); + exit(EXIT_FAILURE); + } + } + + if (output_file == NULL) { + h5fid = HDopen(input_file,O_WRONLY, 0); + + if (h5fid < 0) { + error_msg(progname, "unable to open output HDF5 file \"%s\"\n", input_file); + exit(EXIT_FAILURE); + } + } else { + h5fid = open(output_file,O_WRONLY|O_CREAT|O_TRUNC, 0644 ); + + if (h5fid < 0) { + error_msg(progname, "unable to open output HDF5 file \"%s\"\n", output_file); + exit(EXIT_FAILURE); + } + } + + + /* copy from 0 to 'usize - 1' into ufid */ + if (!do_delete) { + copy_to_file( ifid, ufid, 0, (ssize_t) usize); + } + + /* copy from usize to end of file into h5fid, + * starting at end of user block if present + */ + copy_to_file( ifid, h5fid, (ssize_t) usize, (fsize - (ssize_t)usize) ); + + + close(ufid); + close(h5fid); + close(ifid); + + return d_status; +} + +/* + * Copy 'how_much' bytes from the input file to the output file, + * starting at byte 'where' in the input file. + * + * Returns the size of the output file. + */ +hsize_t +copy_to_file( int infid, int ofid, ssize_t where, ssize_t how_much ) { + char buf[1024]; + off_t to; + off_t from; + ssize_t nchars = -1; + + + if (how_much <= 0) { + /* nothing to copy */ + return(where); + } + from = where; + to = 0; + + while( how_much > 0) { + HDlseek(infid,from,SEEK_SET); + if (how_much > 512) { + nchars = HDread(infid,buf,(unsigned)512); + } else { + nchars = HDread(infid,buf,(unsigned)how_much); + } + HDlseek(ofid,to,SEEK_SET); + HDwrite(ofid,buf,(unsigned)nchars); + how_much -= nchars; + from += nchars; + to += nchars; + } + + return (where+how_much); +} |