Add storefile option to DMAPI invis I/O tests
[xfstests-dev.git] / 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 }