dmapi/src/common/cmd/write_invis.c

   1 /*
   2  * Copyright (c) 2000-2001 Silicon Graphics, Inc.  All Rights Reserved.
   3  *
   4  * This program is free software; you can redistribute it and/or modify it
   5  * under the terms of version 2 of the GNU General Public License as
   6  * published by the Free Software Foundation.
   7  *
   8  * This program is distributed in the hope that it would be useful, but
   9  * WITHOUT ANY WARRANTY; without even the implied warranty of
  10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  11  *
  12  * Further, this software is distributed without any warranty that it is
  13  * free of the rightful claim of any third person regarding infringement
  14  * or the like.  Any license provided herein, whether implied or
  15  * otherwise, applies only to this software file.  Patent licenses, if
  16  * any, provided herein do not apply to combinations of this program with
  17  * other software, or any other product whatsoever.
  18  *
  19  * You should have received a copy of the GNU General Public License along
  20  * with this program; if not, write the Free Software Foundation, Inc., 59
  21  * Temple Place - Suite 330, Boston MA 02111-1307, USA.
  22  *
  23  * Contact information: Silicon Graphics, Inc., 1600 Amphitheatre Pkwy,
  24  * Mountain View, CA  94043, or:
  25  *
  26  * http://www.sgi.com
  27  *
  28  * For further information regarding this notice, see:
  29  *
  30  * http://oss.sgi.com/projects/GenInfo/SGIGPLNoticeExplan/
  31  */
  32
  33 #include <lib/hsm.h>
  34
  35 #include <string.h>
  36 #include <malloc.h>
  37 #include <getopt.h>
  38 #include <sys/types.h>
  39 #include <sys/stat.h>
  40 #include <fcntl.h>
  41
  42 /*---------------------------------------------------------------------------
  43
  44 Test program used to test the DMAPI function dm_write_invis().  The
  45 command line is:
  46
  47         write_invis [-c char] [-o offset] [-l length] [-s sid] \
  48                 [-S storefile] {pathname|handle}
  49
  50 where:
  51 'char' is the character to use as a repeated pattern ('X' is the default),
  52 'offset' is the offset of the start of the write (0 is the default),
  53 'length' is the length of the write in bytes (1 is the default),
  54 'sid' is the session ID whose events you you are interested in.
  55 'pathname' is the name of the file to be written.
  56
  57 DM_WRITE_SYNC is is not supported.
  58
  59 ----------------------------------------------------------------------------*/
  60
  61 #ifndef linux
  62 extern  char    *sys_errlist[];
  63 #endif
  64 extern  int     optind;
  65 extern  char    *optarg;
  66
  67
  68 char    *Progname;
  69
  70
  71 static void
  72 usage(void)
  73 {
  74         fprintf(stderr, "usage:\t%s [-c char] [-o offset] [-l length] "
  75                 "[-s sid] [-S storefile] {pathname|handle}\n", Progname);
  76         exit(1);
  77 }
  78
  79
  80 int
  81 main(
  82         int     argc,
  83         char    **argv)
  84 {
  85         dm_sessid_t     sid = DM_NO_SESSION;
  86         char            *object = NULL;
  87         dm_off_t        offset = 0;
  88         dm_size_t       length = 1;
  89         u_char          ch = 'X';
  90         void            *bufp = NULL;
  91         void            *hanp;
  92         size_t          hlen;
  93         dm_ssize_t      rc;
  94         char            *name;
  95         int             opt;
  96         char            *storefile = NULL;
  97         int             storefd;
  98         int             exit_status = 0;
  99
 100         if (Progname = strrchr(argv[0], '/')) {
 101                 Progname++;
 102         } else {
 103                 Progname = argv[0];
 104         }
 105
 106         /* Crack and validate the command line options. */
 107
 108         while ((opt = getopt(argc, argv, "c:o:l:s:S:")) != EOF) {
 109                 switch (opt) {
 110                 case 'c':
 111                         ch = *optarg;
 112                         break;
 113                 case 'o':
 114                         sscanf(optarg, "%lld", &offset);
 115                         break;
 116                 case 'l':
 117                         sscanf(optarg, "%llu", &length);
 118                         break;
 119                 case 's':
 120                         sid = atol(optarg);
 121                         break;
 122                 case 'S':
 123                         storefile = optarg;
 124                         break;
 125                 case '?':
 126                         usage();
 127                 }
 128         }
 129         if (optind + 1 != argc)
 130                 usage();
 131         object = argv[optind];
 132
 133         if (dm_init_service(&name) == -1)  {
 134                 fprintf(stderr, "Can't initialize the DMAPI\n");
 135                 exit(1);
 136         }
 137         if (sid == DM_NO_SESSION)
 138                 find_test_session(&sid);
 139
 140         /* Get the file's handle. */
 141
 142         if (opaque_to_handle(object, &hanp, &hlen)) {
 143                 fprintf(stderr, "can't get handle for %s\n", object);
 144                 exit(1);
 145         }
 146
 147         if (length > 0) {
 148                 /* In case it is a realtime file, align the buffer on a
 149                    sufficiently big boundary.
 150                 */
 151                 if ((bufp = memalign(4096, length)) == NULL) {
 152                         fprintf(stderr, "malloc of %llu bytes failed\n", length);
 153                         exit(1);
 154                 }
 155                 memset(bufp, ch, length);
 156         }
 157
 158         if (storefile) {
 159                 ssize_t sret;
 160                 off_t   lret;
 161
 162                 if ((storefd = open(storefile, O_RDONLY)) == -1) {
 163                         fprintf(stderr, "unable to open store file for read (%s), errno = %d\n", storefile, errno);
 164                         exit(1);
 165                 }
 166                 lret = lseek(storefd, offset, SEEK_SET);
 167                 if (lret < 0) {
 168                         fprintf(stderr, "unable to lseek(%s) to offset %lld, errno = %d\n",
 169                                 storefile, (long long)lret, errno);
 170                         exit(1);
 171                 }
 172                 sret = read(storefd, bufp, length);
 173                 if (sret < 0) {
 174                         fprintf(stderr, "unable to read store file (%s), errno = %d\n", storefile, errno);
 175                         exit(1);
 176                 }
 177                 else if (sret != length) {
 178                         fprintf(stderr, "read(%s) returned %lld, expected %lld\n",
 179                                 storefile, (long long)sret, (long long)length);
 180                         exit(1);
 181                 }
 182                 close(storefd);
 183         }
 184
 185         rc = dm_write_invis(sid, hanp, hlen, DM_NO_TOKEN, 0, offset, length, bufp);
 186
 187         if (rc < 0) {
 188                 fprintf(stderr, "dm_write_invis failed, %s\n", strerror(errno));
 189                 exit_status++;
 190         } else if (rc != length) {
 191                 fprintf(stderr, "dm_write_invis expected to write %lld bytes, actually "
 192                         "wrote %lld\n", length, rc);
 193                 exit_status++;
 194         }
 195         dm_handle_free(hanp, hlen);
 196         exit(exit_status);
 197 }