xref: /aosp_15_r20/external/sg3_utils/testing/sg_tst_async.cpp (revision 44704f698541f6367e81f991ef8bb54ccbf3fc18)
1*44704f69SBart Van Assche /*
2*44704f69SBart Van Assche  * Copyright (c) 2014-2022 Douglas Gilbert.
3*44704f69SBart Van Assche  * All rights reserved.
4*44704f69SBart Van Assche  *
5*44704f69SBart Van Assche  * Redistribution and use in source and binary forms, with or without
6*44704f69SBart Van Assche  * modification, are permitted provided that the following conditions
7*44704f69SBart Van Assche  * are met:
8*44704f69SBart Van Assche  * 1. Redistributions of source code must retain the above copyright
9*44704f69SBart Van Assche  *    notice, this list of conditions and the following disclaimer.
10*44704f69SBart Van Assche  * 2. Redistributions in binary form must reproduce the above copyright
11*44704f69SBart Van Assche  *    notice, this list of conditions and the following disclaimer in the
12*44704f69SBart Van Assche  *    documentation and/or other materials provided with the distribution.
13*44704f69SBart Van Assche  *
14*44704f69SBart Van Assche  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15*44704f69SBart Van Assche  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16*44704f69SBart Van Assche  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17*44704f69SBart Van Assche  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18*44704f69SBart Van Assche  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19*44704f69SBart Van Assche  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20*44704f69SBart Van Assche  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21*44704f69SBart Van Assche  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22*44704f69SBart Van Assche  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23*44704f69SBart Van Assche  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24*44704f69SBart Van Assche  * SUCH DAMAGE.
25*44704f69SBart Van Assche  *
26*44704f69SBart Van Assche  * SPDX-License-Identifier: BSD-2-Clause
27*44704f69SBart Van Assche  */
28*44704f69SBart Van Assche 
29*44704f69SBart Van Assche #include <iostream>
30*44704f69SBart Van Assche #include <vector>
31*44704f69SBart Van Assche #include <map>
32*44704f69SBart Van Assche #include <list>
33*44704f69SBart Van Assche #include <system_error>
34*44704f69SBart Van Assche #include <thread>
35*44704f69SBart Van Assche #include <mutex>
36*44704f69SBart Van Assche #include <chrono>
37*44704f69SBart Van Assche #include <atomic>
38*44704f69SBart Van Assche #include <random>
39*44704f69SBart Van Assche 
40*44704f69SBart Van Assche #include <unistd.h>
41*44704f69SBart Van Assche #include <fcntl.h>
42*44704f69SBart Van Assche #include <stdio.h>
43*44704f69SBart Van Assche #include <stdlib.h>
44*44704f69SBart Van Assche #include <stdarg.h>
45*44704f69SBart Van Assche #include <string.h>
46*44704f69SBart Van Assche #include <poll.h>
47*44704f69SBart Van Assche #include <errno.h>
48*44704f69SBart Van Assche #include <ctype.h>
49*44704f69SBart Van Assche #include <time.h>
50*44704f69SBart Van Assche #include <limits.h>
51*44704f69SBart Van Assche #include <getopt.h>
52*44704f69SBart Van Assche #define __STDC_FORMAT_MACROS 1
53*44704f69SBart Van Assche #include <inttypes.h>
54*44704f69SBart Van Assche #include <sys/ioctl.h>
55*44704f69SBart Van Assche #include <sys/types.h>
56*44704f69SBart Van Assche #include <sys/stat.h>
57*44704f69SBart Van Assche #include <sys/mman.h>
58*44704f69SBart Van Assche #include <sys/resource.h>       /* getrusage */
59*44704f69SBart Van Assche 
60*44704f69SBart Van Assche 
61*44704f69SBart Van Assche #ifdef HAVE_CONFIG_H
62*44704f69SBart Van Assche #include "config.h"
63*44704f69SBart Van Assche #endif
64*44704f69SBart Van Assche 
65*44704f69SBart Van Assche #ifndef HAVE_LINUX_SG_V4_HDR
66*44704f69SBart Van Assche 
67*44704f69SBart Van Assche /* Kernel uapi header contain __user decorations on user space pointers
68*44704f69SBart Van Assche  * to indicate they are unsafe in the kernel space. However glibc takes
69*44704f69SBart Van Assche  * all those __user decorations out from headers in /usr/include/linux .
70*44704f69SBart Van Assche  * So to stop compile errors when directly importing include/uapi/scsi/sg.h
71*44704f69SBart Van Assche  * undef __user before doing that include. */
72*44704f69SBart Van Assche #define __user
73*44704f69SBart Van Assche 
74*44704f69SBart Van Assche /* Want to block the original sg.h header from also being included. That
75*44704f69SBart Van Assche  * causes lots of multiple definition errors. This will only work if this
76*44704f69SBart Van Assche  * header is included _before_ the original sg.h header.  */
77*44704f69SBart Van Assche #define _SCSI_GENERIC_H         /* original kernel header guard */
78*44704f69SBart Van Assche #define _SCSI_SG_H              /* glibc header guard */
79*44704f69SBart Van Assche 
80*44704f69SBart Van Assche #include "uapi_sg.h"    /* local copy of include/uapi/scsi/sg.h */
81*44704f69SBart Van Assche 
82*44704f69SBart Van Assche #else
83*44704f69SBart Van Assche #define __user
84*44704f69SBart Van Assche #endif  /* end of: ifndef HAVE_LINUX_SG_V4_HDR */
85*44704f69SBart Van Assche 
86*44704f69SBart Van Assche #include "sg_lib.h"
87*44704f69SBart Van Assche #include "sg_io_linux.h"
88*44704f69SBart Van Assche #include "sg_unaligned.h"
89*44704f69SBart Van Assche #include "sg_pt.h"
90*44704f69SBart Van Assche #include "sg_cmds.h"
91*44704f69SBart Van Assche 
92*44704f69SBart Van Assche static const char * version_str = "1.42 20220425";
93*44704f69SBart Van Assche static const char * util_name = "sg_tst_async";
94*44704f69SBart Van Assche 
95*44704f69SBart Van Assche /* This is a test program for checking the async usage of the Linux sg
96*44704f69SBart Van Assche  * driver. Each thread opens 1 file descriptor to the next sg device (1
97*44704f69SBart Van Assche  * or more can be given on the command line) and then starts up to
98*44704f69SBart Van Assche  * num_per_thread commands or more while checking with the poll command (or
99*44704f69SBart Van Assche  * ioctl(SG_GET_NUM_WAITING) ) for the completion of those commands. Each
100*44704f69SBart Van Assche  * command has a unique "pack_id" which is a sequence starting at 1.
101*44704f69SBart Van Assche  * Either TEST UNIT UNIT, READ(16) or WRITE(16) commands are issued.
102*44704f69SBart Van Assche  *
103*44704f69SBart Van Assche  * This is C++ code with some things from C++11 (e.g. threads) and was
104*44704f69SBart Van Assche  * only just able to compile (when some things were reverted) with gcc/g++
105*44704f69SBart Van Assche  * version 4.7.3 found in Ubuntu 13.04 . C++11 "feature complete" support
106*44704f69SBart Van Assche  * was not available until g++ version 4.8.1 . It should build okay on
107*44704f69SBart Van Assche  * recent distributions.
108*44704f69SBart Van Assche  *
109*44704f69SBart Van Assche  * The build uses various object files from the <sg3_utils>/lib directory
110*44704f69SBart Van Assche  * which is assumed to be a sibling of this examples directory. Those
111*44704f69SBart Van Assche  * object files in the lib directory can be built with:
112*44704f69SBart Van Assche  *   cd <sg3_utils_package_root> ; ./configure ; cd lib; make
113*44704f69SBart Van Assche  *   cd ../testing
114*44704f69SBart Van Assche  *   make sg_tst_async
115*44704f69SBart Van Assche  *
116*44704f69SBart Van Assche  * Currently this utility is Linux only and uses the sg driver. The bsg
117*44704f69SBart Van Assche  * driver is known to be broken (it doesn't match responses to the
118*44704f69SBart Van Assche  * correct file descriptor that requested them). Around Linux kernel 4.15
119*44704f69SBart Van Assche  * the async capability of the bsg driver was removed. So this test code
120*44704f69SBart Van Assche  * no longer appiles to the bsg driver.
121*44704f69SBart Van Assche  *
122*44704f69SBart Van Assche  * BEWARE: >>> This utility will modify a logical block (default LBA 1000)
123*44704f69SBart Van Assche  * on the given device _when_ the '-W' option is given.
124*44704f69SBart Van Assche  *
125*44704f69SBart Van Assche  */
126*44704f69SBart Van Assche 
127*44704f69SBart Van Assche using namespace std;
128*44704f69SBart Van Assche using namespace std::chrono;
129*44704f69SBart Van Assche 
130*44704f69SBart Van Assche #define DEF_NUM_PER_THREAD 1000
131*44704f69SBart Van Assche #define DEF_NUM_THREADS 4
132*44704f69SBart Van Assche #define DEF_WAIT_MS 10          /* 0: yield or no wait */
133*44704f69SBart Van Assche #define DEF_NANOSEC_WAIT 25000  /* 25 microsecs */
134*44704f69SBart Van Assche #define DEF_TIMEOUT_MS 20000    /* 20 seconds */
135*44704f69SBart Van Assche #define DEF_LB_SZ 512
136*44704f69SBart Van Assche #define DEF_BLOCKING 0
137*44704f69SBart Van Assche #define DEF_DIRECT false        /* true: direct_io */
138*44704f69SBart Van Assche #define DEF_MMAP_IO false       /* true: mmap-ed IO with sg */
139*44704f69SBart Van Assche #define DEF_NO_XFER 0
140*44704f69SBart Van Assche #define DEF_LBA 1000U
141*44704f69SBart Van Assche 
142*44704f69SBart Van Assche #define MAX_Q_PER_FD 16383      /* sg driver per file descriptor limit */
143*44704f69SBart Van Assche #define MAX_CONSEC_NOMEMS 4     /* was 16 */
144*44704f69SBart Van Assche #define URANDOM_DEV "/dev/urandom"
145*44704f69SBart Van Assche 
146*44704f69SBart Van Assche #ifndef SG_FLAG_Q_AT_TAIL
147*44704f69SBart Van Assche #define SG_FLAG_Q_AT_TAIL 0x10
148*44704f69SBart Van Assche #endif
149*44704f69SBart Van Assche #ifndef SG_FLAG_Q_AT_HEAD
150*44704f69SBart Van Assche #define SG_FLAG_Q_AT_HEAD 0x20
151*44704f69SBart Van Assche #endif
152*44704f69SBart Van Assche 
153*44704f69SBart Van Assche 
154*44704f69SBart Van Assche #define DEF_PT_TIMEOUT 60       /* 60 seconds */
155*44704f69SBart Van Assche 
156*44704f69SBart Van Assche #define EBUFF_SZ 256
157*44704f69SBart Van Assche 
158*44704f69SBart Van Assche static mutex console_mutex;
159*44704f69SBart Van Assche static mutex rand_lba_mutex;
160*44704f69SBart Van Assche static atomic<int> async_starts(0);
161*44704f69SBart Van Assche static atomic<int> sync_starts(0);
162*44704f69SBart Van Assche static atomic<int> async_finishes(0);
163*44704f69SBart Van Assche static atomic<int> start_ebusy_count(0);
164*44704f69SBart Van Assche static atomic<int> start_e2big_count(0);
165*44704f69SBart Van Assche static atomic<int> start_eagain_count(0);
166*44704f69SBart Van Assche static atomic<int> fin_eagain_count(0);
167*44704f69SBart Van Assche static atomic<int> fin_ebusy_count(0);
168*44704f69SBart Van Assche static atomic<int> start_edom_count(0);
169*44704f69SBart Van Assche static atomic<int> enomem_count(0);
170*44704f69SBart Van Assche static atomic<int> uniq_pack_id(1);
171*44704f69SBart Van Assche // static atomic<int> generic_errs(0);
172*44704f69SBart Van Assche 
173*44704f69SBart Van Assche static int page_size = 4096;   /* rough guess, will ask sysconf() */
174*44704f69SBart Van Assche 
175*44704f69SBart Van Assche enum command2execute {SCSI_TUR, SCSI_READ16, SCSI_WRITE16};
176*44704f69SBart Van Assche /* Linux Block layer queue disciplines: */
177*44704f69SBart Van Assche enum blkLQDiscipline {BLQ_DEFAULT, BLQ_AT_HEAD, BLQ_AT_TAIL};
178*44704f69SBart Van Assche /* Queue disciplines of this utility. When both completions and
179*44704f69SBart Van Assche  * queuing a new command are both possible: */
180*44704f69SBart Van Assche enum myQDiscipline {MYQD_LOW,   /* favour completions over new cmds */
181*44704f69SBart Van Assche                     MYQD_MEDIUM,
182*44704f69SBart Van Assche                     MYQD_HIGH}; /* favour new cmds over completions */
183*44704f69SBart Van Assche 
184*44704f69SBart Van Assche struct opts_t {
185*44704f69SBart Van Assche     vector<const char *> dev_names;
186*44704f69SBart Van Assche     vector<int> blk_szs;
187*44704f69SBart Van Assche     bool block;
188*44704f69SBart Van Assche     bool cmd_time;
189*44704f69SBart Van Assche     bool direct;
190*44704f69SBart Van Assche     bool excl;
191*44704f69SBart Van Assche     bool generic_sync;
192*44704f69SBart Van Assche     bool masync;
193*44704f69SBart Van Assche     bool mmap_io;
194*44704f69SBart Van Assche     bool no_xfer;
195*44704f69SBart Van Assche     bool pack_id_force;
196*44704f69SBart Van Assche     bool sg_vn_ge_40000;
197*44704f69SBart Van Assche     bool sg_vn_ge_40030;
198*44704f69SBart Van Assche     bool submit;
199*44704f69SBart Van Assche     bool verbose_given;
200*44704f69SBart Van Assche     bool v3;
201*44704f69SBart Van Assche     bool v3_given;
202*44704f69SBart Van Assche     bool v4;
203*44704f69SBart Van Assche     bool v4_given;
204*44704f69SBart Van Assche     bool version_given;
205*44704f69SBart Van Assche     int maxq_per_thread;
206*44704f69SBart Van Assche     int num_per_thread;
207*44704f69SBart Van Assche     uint64_t lba;
208*44704f69SBart Van Assche     unsigned int hi_lba;        /* last one, inclusive range */
209*44704f69SBart Van Assche     vector<unsigned int> hi_lbas; /* only used when hi_lba=-1 */
210*44704f69SBart Van Assche     int lb_sz;
211*44704f69SBart Van Assche     int num_lbs;
212*44704f69SBart Van Assche     int ovn;            /* override number for submission */
213*44704f69SBart Van Assche     int stats;
214*44704f69SBart Van Assche     int verbose;
215*44704f69SBart Van Assche     int wait_ms;
216*44704f69SBart Van Assche     command2execute c2e;
217*44704f69SBart Van Assche     blkLQDiscipline blqd;       /* --qat= 0|1 -> at_head|at_tail */
218*44704f69SBart Van Assche     myQDiscipline myqd;         /* --qfav= value (def: 2 --> MYQD_HIGH) */
219*44704f69SBart Van Assche };
220*44704f69SBart Van Assche 
221*44704f69SBart Van Assche static struct opts_t a_opts;    /* Expect zero fill on simple types */
222*44704f69SBart Van Assche 
223*44704f69SBart Van Assche static int pr_rusage(int id);
224*44704f69SBart Van Assche 
225*44704f69SBart Van Assche #if 0
226*44704f69SBart Van Assche class Rand_uint {
227*44704f69SBart Van Assche public:
228*44704f69SBart Van Assche     Rand_uint(unsigned int lo, unsigned int hi) : p{lo, hi} {}
229*44704f69SBart Van Assche     unsigned int operator()() const { return r(); }
230*44704f69SBart Van Assche private:
231*44704f69SBart Van Assche     uniform_int_distribution<unsigned int>::param_type p;
232*44704f69SBart Van Assche     auto r = bind(uniform_int_distribution<unsigned int>{p},
233*44704f69SBart Van Assche                   default_random_engine());
234*44704f69SBart Van Assche     /* compiler thinks auto should be a static, bs again? */
235*44704f69SBart Van Assche };
236*44704f69SBart Van Assche #endif
237*44704f69SBart Van Assche 
238*44704f69SBart Van Assche #if 0
239*44704f69SBart Van Assche class Rand_uint {
240*44704f69SBart Van Assche public:
241*44704f69SBart Van Assche     Rand_uint(unsigned int lo, unsigned int hi, unsigned int my_seed)
242*44704f69SBart Van Assche         : r(bind(uniform_int_distribution<unsigned int>{lo, hi},
243*44704f69SBart Van Assche                  default_random_engine())) { r.seed(myseed); }
244*44704f69SBart Van Assche     unsigned int operator()() const { return r(); }
245*44704f69SBart Van Assche private:
246*44704f69SBart Van Assche     function<unsigned int()> r;
247*44704f69SBart Van Assche };
248*44704f69SBart Van Assche #endif
249*44704f69SBart Van Assche 
250*44704f69SBart Van Assche /* Use this class to wrap C++11 <random> features to produce uniform random
251*44704f69SBart Van Assche  * unsigned ints in the range [lo, hi] (inclusive) given a_seed */
252*44704f69SBart Van Assche class Rand_uint {
253*44704f69SBart Van Assche public:
Rand_uint(unsigned int lo,unsigned int hi,unsigned int a_seed)254*44704f69SBart Van Assche     Rand_uint(unsigned int lo, unsigned int hi, unsigned int a_seed)
255*44704f69SBart Van Assche         : uid(lo, hi), dre(a_seed) { }
256*44704f69SBart Van Assche     /* uid ctor takes inclusive range when integral type */
257*44704f69SBart Van Assche 
get()258*44704f69SBart Van Assche     unsigned int get() { return uid(dre); }
259*44704f69SBart Van Assche 
260*44704f69SBart Van Assche private:
261*44704f69SBart Van Assche     uniform_int_distribution<unsigned int> uid;
262*44704f69SBart Van Assche     default_random_engine dre;
263*44704f69SBart Van Assche };
264*44704f69SBart Van Assche 
265*44704f69SBart Van Assche static struct option long_options[] = {
266*44704f69SBart Van Assche         {"v3", no_argument, 0, '3'},
267*44704f69SBart Van Assche         {"v4", no_argument, 0, '4'},
268*44704f69SBart Van Assche         {"more-async", no_argument, 0, 'a'},
269*44704f69SBart Van Assche         {"more_async", no_argument, 0, 'a'},
270*44704f69SBart Van Assche         {"masync", no_argument, 0, 'a'},
271*44704f69SBart Van Assche         {"cmd-time", no_argument, 0, 'c'},
272*44704f69SBart Van Assche         {"cmd_time", no_argument, 0, 'c'},
273*44704f69SBart Van Assche         {"direct", no_argument, 0, 'd'},
274*44704f69SBart Van Assche         {"excl", no_argument, 0, 'e'},
275*44704f69SBart Van Assche         {"force", no_argument, 0, 'f'},
276*44704f69SBart Van Assche         {"generic-sync", no_argument, 0, 'g'},
277*44704f69SBart Van Assche         {"generic_sync", no_argument, 0, 'g'},
278*44704f69SBart Van Assche         {"help", no_argument, 0, 'h'},
279*44704f69SBart Van Assche         {"lba", required_argument, 0, 'l'},
280*44704f69SBart Van Assche         {"lbsz", required_argument, 0, 'L'},
281*44704f69SBart Van Assche         {"maxqpt", required_argument, 0, 'M'},
282*44704f69SBart Van Assche         {"mmap-io", no_argument, 0, 'm'},
283*44704f69SBart Van Assche         {"mmap_io", no_argument, 0, 'm'},
284*44704f69SBart Van Assche         {"numpt", required_argument, 0, 'n'},
285*44704f69SBart Van Assche         {"num-pt", required_argument, 0, 'n'},
286*44704f69SBart Van Assche         {"num_pt", required_argument, 0, 'n'},
287*44704f69SBart Van Assche         {"noxfer", no_argument, 0, 'N'},
288*44704f69SBart Van Assche         {"override", required_argument, 0, 'O'},
289*44704f69SBart Van Assche         {"pack-id", no_argument, 0, 'p'},
290*44704f69SBart Van Assche         {"pack_id", no_argument, 0, 'p'},
291*44704f69SBart Van Assche         {"qat", required_argument, 0, 'q'},
292*44704f69SBart Van Assche         {"qfav", required_argument, 0, 'Q'},
293*44704f69SBart Van Assche         {"read", no_argument, 0, 'R'},
294*44704f69SBart Van Assche         {"stats", no_argument, 0, 'S'},
295*44704f69SBart Van Assche         {"submit", no_argument, 0, 'u'},
296*44704f69SBart Van Assche         {"szlb", required_argument, 0, 's'},
297*44704f69SBart Van Assche         {"tnum", required_argument, 0, 't'},
298*44704f69SBart Van Assche         {"tur", no_argument, 0, 'T'},
299*44704f69SBart Van Assche         {"verbose", no_argument, 0, 'v'},
300*44704f69SBart Van Assche         {"version", no_argument, 0, 'V'},
301*44704f69SBart Van Assche         {"wait", required_argument, 0, 'w'},
302*44704f69SBart Van Assche         {"write", no_argument, 0, 'W'},
303*44704f69SBart Van Assche         {0, 0, 0, 0},
304*44704f69SBart Van Assche };
305*44704f69SBart Van Assche 
306*44704f69SBart Van Assche 
307*44704f69SBart Van Assche static void
usage(void)308*44704f69SBart Van Assche usage(void)
309*44704f69SBart Van Assche {
310*44704f69SBart Van Assche     printf("Usage: %s [--cmd-time] [--direct] [--excl] [--force]\n"
311*44704f69SBart Van Assche            "                    [--generic-sync] [--help] [--lba=LBA+] "
312*44704f69SBart Van Assche            "[--lbsz=LBSZ]\n"
313*44704f69SBart Van Assche            "                    [--masync] [--maxqpt=QPT] [--mmap-io] "
314*44704f69SBart Van Assche            "[--no-waitq]\n"
315*44704f69SBart Van Assche            "                    [--noxfer] [--numpt=NPT] [--override=OVN] "
316*44704f69SBart Van Assche            "[--pack-id]\n"
317*44704f69SBart Van Assche            "                    [--qat=AT] [-qfav=FAV] [--read] [--stats] "
318*44704f69SBart Van Assche            "[--submit]\n"
319*44704f69SBart Van Assche            "                    [--szlb=LB[,NLBS]] [--tnum=NT] [--tur] "
320*44704f69SBart Van Assche            "[--v3] [--v4]\n"
321*44704f69SBart Van Assche            "                    [--verbose] [--version] [--wait=MS] "
322*44704f69SBart Van Assche            "[--write]\n"
323*44704f69SBart Van Assche            "                    <sg_disk_device>*\n",
324*44704f69SBart Van Assche            util_name);
325*44704f69SBart Van Assche     printf("  where\n");
326*44704f69SBart Van Assche     printf("    --cmd-time|-c    calculate per command average time (ns)\n");
327*44704f69SBart Van Assche     printf("    --direct|-d     do direct_io (def: indirect)\n");
328*44704f69SBart Van Assche     printf("    --excl|-e       do wait_exclusive calls\n");
329*44704f69SBart Van Assche     printf("    --force|-f      force: any sg device (def: only scsi_debug "
330*44704f69SBart Van Assche            "owned)\n");
331*44704f69SBart Van Assche     printf("                    WARNING: <lba> written to if '-W' given\n");
332*44704f69SBart Van Assche     printf("    --generic-sync|-g    use generic synchronous SG_IO ioctl "
333*44704f69SBart Van Assche            "instead\n");
334*44704f69SBart Van Assche     printf("                       of Linux sg driver assuming /dev/sg* "
335*44704f69SBart Van Assche            "(def)\n");
336*44704f69SBart Van Assche     printf("    --help|-h       print this usage message then exit\n");
337*44704f69SBart Van Assche     printf("    --lba=LBA|-l LBA    logical block to access (def: %u)\n",
338*44704f69SBart Van Assche            DEF_LBA);
339*44704f69SBart Van Assche     printf("    --lba=LBA,HI_LBA|-l LBA,HI_LBA    logical block range "
340*44704f69SBart Van Assche            "(inclusive)\n"
341*44704f69SBart Van Assche            "                          if hi_lba=-1 assume last block on "
342*44704f69SBart Van Assche            "device\n");
343*44704f69SBart Van Assche     printf("    --lbsz=LBSZ|-L LBSZ    logical block size in bytes (def: "
344*44704f69SBart Van Assche            "512)\n"
345*44704f69SBart Van Assche            "                           should be power of 2 (0 --> 512)\n");
346*44704f69SBart Van Assche     printf("    --masync|-a     set 'more async' flag on devices\n");
347*44704f69SBart Van Assche     printf("    --maxqpt=QPT|-M QPT    maximum commands queued per thread "
348*44704f69SBart Van Assche            "(def:%d)\n", MAX_Q_PER_FD);
349*44704f69SBart Van Assche     printf("    --mmap-io|-m    mmap-ed IO (1 cmd outstanding per thread)\n");
350*44704f69SBart Van Assche     printf("    --noxfer|-N          no data xfer (def: xfer on READ and "
351*44704f69SBart Van Assche            "WRITE)\n");
352*44704f69SBart Van Assche     printf("    --numpt=NPT|-n NPT    number of commands per thread "
353*44704f69SBart Van Assche            "(def: %d)\n", DEF_NUM_PER_THREAD);
354*44704f69SBart Van Assche     printf("    --override OVN|-O OVN    override FAV=2 when OVN queue "
355*44704f69SBart Van Assche            "depth\n"
356*44704f69SBart Van Assche            "                             reached (def: 0 -> no override)\n");
357*44704f69SBart Van Assche     printf("    --pack-id|-p    set FORCE_PACK_ID, pack-id input to "
358*44704f69SBart Van Assche            "read/finish\n");
359*44704f69SBart Van Assche     printf("    --qat=AT|-q AT       AT=0: q_at_head; AT=1: q_at_tail (def: "
360*44704f69SBart Van Assche            "(drv): head)\n");
361*44704f69SBart Van Assche     printf("    --qfav=FAV|-Q FAV    FAV=0: favour completions (smaller q),\n"
362*44704f69SBart Van Assche            "                         FAV=1: medium,\n"
363*44704f69SBart Van Assche            "                         FAV=2: favour submissions (larger q, "
364*44704f69SBart Van Assche            "default)\n");
365*44704f69SBart Van Assche     printf("    --read|-R       do READs (def: TUR)\n");
366*44704f69SBart Van Assche     printf("    --stats|-S      show more statistics on completion\n");
367*44704f69SBart Van Assche     printf("    --submit|-u     use SG_IOSUBMIT+SG_IORECEIVE instead of "
368*44704f69SBart Van Assche            "write+read\n");
369*44704f69SBart Van Assche     printf("    --szlb=LB[,NLBS]|    LB is logical block size (def: 512)\n");
370*44704f69SBart Van Assche     printf("         -s LB[,NLBS]    NLBS is number of logical blocks (def: "
371*44704f69SBart Van Assche            "1)\n");
372*44704f69SBart Van Assche     printf("    --tnum=NT|-t NT    number of threads (def: %d)\n",
373*44704f69SBart Van Assche            DEF_NUM_THREADS);
374*44704f69SBart Van Assche     printf("    --tur|-T        do TEST UNIT READYs (default is TURs)\n");
375*44704f69SBart Van Assche     printf("    --v3|-3         use sg v3 interface (def: v3 if driver < "
376*44704f69SBart Van Assche            "3.9)\n");
377*44704f69SBart Van Assche     printf("    --v4|-4         use sg v4 interface (def if v4 driver). Sets "
378*44704f69SBart Van Assche            "--submit\n");
379*44704f69SBart Van Assche     printf("    --verbose|-v    increase verbosity\n");
380*44704f69SBart Van Assche     printf("    --version|-V    print version number then exit\n");
381*44704f69SBart Van Assche     printf("    --wait=MS|-w MS    >0: poll(<wait_ms>); =0: poll(0); (def: "
382*44704f69SBart Van Assche            "%d)\n", DEF_WAIT_MS);
383*44704f69SBart Van Assche     printf("    --write|-W      do WRITEs (def: TUR)\n\n");
384*44704f69SBart Van Assche     printf("Multiple threads send READ(16), WRITE(16) or TEST UNIT READY "
385*44704f69SBart Van Assche            "(TUR) SCSI\ncommands. There can be 1 or more <sg_disk_device>s "
386*44704f69SBart Van Assche            "and each thread takes\nthe next in a round robin fashion. "
387*44704f69SBart Van Assche            "Each thread queues up to NT commands.\nOne block is transferred "
388*44704f69SBart Van Assche            "by each READ and WRITE; zeros are written. If a\nlogical block "
389*44704f69SBart Van Assche            "range is given, a uniform distribution generates a pseudo\n"
390*44704f69SBart Van Assche            "random sequence of LBAs. Set environment variable\n"
391*44704f69SBart Van Assche            "SG3_UTILS_LINUX_NANO to get command timings in nanoseconds\n");
392*44704f69SBart Van Assche }
393*44704f69SBart Van Assche 
394*44704f69SBart Van Assche #ifdef __GNUC__
395*44704f69SBart Van Assche static int pr2serr_lk(const char * fmt, ...)
396*44704f69SBart Van Assche         __attribute__ ((format (printf, 1, 2)));
397*44704f69SBart Van Assche static void pr_errno_lk(int e_no, const char * fmt, ...)
398*44704f69SBart Van Assche         __attribute__ ((format (printf, 2, 3)));
399*44704f69SBart Van Assche #else
400*44704f69SBart Van Assche static int pr2serr_lk(const char * fmt, ...);
401*44704f69SBart Van Assche static void pr_errno_lk(int e_no, const char * fmt, ...);
402*44704f69SBart Van Assche #endif
403*44704f69SBart Van Assche 
404*44704f69SBart Van Assche 
405*44704f69SBart Van Assche static int
pr2serr_lk(const char * fmt,...)406*44704f69SBart Van Assche pr2serr_lk(const char * fmt, ...)
407*44704f69SBart Van Assche {
408*44704f69SBart Van Assche     int n;
409*44704f69SBart Van Assche     va_list args;
410*44704f69SBart Van Assche     lock_guard<mutex> lg(console_mutex);
411*44704f69SBart Van Assche 
412*44704f69SBart Van Assche     va_start(args, fmt);
413*44704f69SBart Van Assche     n = vfprintf(stderr, fmt, args);
414*44704f69SBart Van Assche     va_end(args);
415*44704f69SBart Van Assche     return n;
416*44704f69SBart Van Assche }
417*44704f69SBart Van Assche 
418*44704f69SBart Van Assche static void
pr_errno_lk(int e_no,const char * fmt,...)419*44704f69SBart Van Assche pr_errno_lk(int e_no, const char * fmt, ...)
420*44704f69SBart Van Assche {
421*44704f69SBart Van Assche     char b[160];
422*44704f69SBart Van Assche     va_list args;
423*44704f69SBart Van Assche     lock_guard<mutex> lg(console_mutex);
424*44704f69SBart Van Assche 
425*44704f69SBart Van Assche     va_start(args, fmt);
426*44704f69SBart Van Assche     vsnprintf(b, sizeof(b), fmt, args);
427*44704f69SBart Van Assche     fprintf(stderr, "%s: %s\n", b, strerror(e_no));
428*44704f69SBart Van Assche     va_end(args);
429*44704f69SBart Van Assche }
430*44704f69SBart Van Assche 
431*44704f69SBart Van Assche static unsigned int
get_urandom_uint(void)432*44704f69SBart Van Assche get_urandom_uint(void)
433*44704f69SBart Van Assche {
434*44704f69SBart Van Assche     unsigned int res = 0;
435*44704f69SBart Van Assche     lock_guard<mutex> lg(rand_lba_mutex);
436*44704f69SBart Van Assche 
437*44704f69SBart Van Assche     int fd = open(URANDOM_DEV, O_RDONLY);
438*44704f69SBart Van Assche     if (fd >= 0) {
439*44704f69SBart Van Assche         uint8_t b[sizeof(unsigned int)];
440*44704f69SBart Van Assche         int n = read(fd, b, sizeof(unsigned int));
441*44704f69SBart Van Assche 
442*44704f69SBart Van Assche         if (sizeof(unsigned int) == n)
443*44704f69SBart Van Assche             memcpy(&res, b, sizeof(unsigned int));
444*44704f69SBart Van Assche         close(fd);
445*44704f69SBart Van Assche     }
446*44704f69SBart Van Assche     return res;
447*44704f69SBart Van Assche }
448*44704f69SBart Van Assche 
449*44704f69SBart Van Assche #define TUR_CMD_LEN 6
450*44704f69SBart Van Assche #define READ16_CMD_LEN 16
451*44704f69SBart Van Assche #define READ16_REPLY_LEN 4096
452*44704f69SBart Van Assche #define WRITE16_REPLY_LEN 4096
453*44704f69SBart Van Assche #define WRITE16_CMD_LEN 16
454*44704f69SBart Van Assche 
455*44704f69SBart Van Assche /* Returns 0 if command injected okay, return -1 for error and 2 for
456*44704f69SBart Van Assche  * not done due to queue data size limit struck. */
457*44704f69SBart Van Assche static int
start_sg3_cmd(int sg_fd,command2execute cmd2exe,int pack_id,uint64_t lba,uint8_t * lbp,int xfer_bytes,int flags,bool submit,unsigned int & enomem,unsigned int & eagains,unsigned int & ebusy,unsigned int & e2big,unsigned int & edom)458*44704f69SBart Van Assche start_sg3_cmd(int sg_fd, command2execute cmd2exe, int pack_id, uint64_t lba,
459*44704f69SBart Van Assche               uint8_t * lbp, int xfer_bytes, int flags, bool submit,
460*44704f69SBart Van Assche               unsigned int & enomem, unsigned int & eagains,
461*44704f69SBart Van Assche               unsigned int & ebusy, unsigned int & e2big, unsigned int & edom)
462*44704f69SBart Van Assche {
463*44704f69SBart Van Assche     struct sg_io_hdr pt;
464*44704f69SBart Van Assche     struct sg_io_v4 p4t;
465*44704f69SBart Van Assche     uint8_t turCmdBlk[TUR_CMD_LEN] = {0, 0, 0, 0, 0, 0};
466*44704f69SBart Van Assche     uint8_t r16CmdBlk[READ16_CMD_LEN] =
467*44704f69SBart Van Assche                 {0x88, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0};
468*44704f69SBart Van Assche     uint8_t w16CmdBlk[WRITE16_CMD_LEN] =
469*44704f69SBart Van Assche                 {0x8a, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0};
470*44704f69SBart Van Assche     uint8_t sense_buffer[64] SG_C_CPP_ZERO_INIT;
471*44704f69SBart Van Assche     const char * np = NULL;
472*44704f69SBart Van Assche     struct sg_io_hdr * ptp;
473*44704f69SBart Van Assche 
474*44704f69SBart Van Assche     if (submit) {       /* nest a v3 interface inside a store for v4 */
475*44704f69SBart Van Assche         memset(&p4t, 0, sizeof(p4t));
476*44704f69SBart Van Assche         ptp = (struct sg_io_hdr *)&p4t; /* p4t is larger than pt */
477*44704f69SBart Van Assche     } else {
478*44704f69SBart Van Assche         ptp = &pt;
479*44704f69SBart Van Assche         memset(ptp, 0, sizeof(*ptp));
480*44704f69SBart Van Assche     }
481*44704f69SBart Van Assche     switch (cmd2exe) {
482*44704f69SBart Van Assche     case SCSI_TUR:
483*44704f69SBart Van Assche         np = "TEST UNIT READY";
484*44704f69SBart Van Assche         ptp->cmdp = turCmdBlk;
485*44704f69SBart Van Assche         ptp->cmd_len = sizeof(turCmdBlk);
486*44704f69SBart Van Assche         ptp->dxfer_direction = SG_DXFER_NONE;
487*44704f69SBart Van Assche         break;
488*44704f69SBart Van Assche     case SCSI_READ16:
489*44704f69SBart Van Assche         np = "READ(16)";
490*44704f69SBart Van Assche         if (lba > 0xffffffff)
491*44704f69SBart Van Assche             sg_put_unaligned_be32(lba >> 32, &r16CmdBlk[2]);
492*44704f69SBart Van Assche         sg_put_unaligned_be32(lba & 0xffffffff, &r16CmdBlk[6]);
493*44704f69SBart Van Assche         ptp->cmdp = r16CmdBlk;
494*44704f69SBart Van Assche         ptp->cmd_len = sizeof(r16CmdBlk);
495*44704f69SBart Van Assche         ptp->dxfer_direction = SG_DXFER_FROM_DEV;
496*44704f69SBart Van Assche         ptp->dxferp = lbp;
497*44704f69SBart Van Assche         ptp->dxfer_len = xfer_bytes;
498*44704f69SBart Van Assche         break;
499*44704f69SBart Van Assche     case SCSI_WRITE16:
500*44704f69SBart Van Assche         np = "WRITE(16)";
501*44704f69SBart Van Assche         if (lba > 0xffffffff)
502*44704f69SBart Van Assche             sg_put_unaligned_be32(lba >> 32, &w16CmdBlk[2]);
503*44704f69SBart Van Assche         sg_put_unaligned_be32(lba & 0xffffffff, &w16CmdBlk[6]);
504*44704f69SBart Van Assche         ptp->cmdp = w16CmdBlk;
505*44704f69SBart Van Assche         ptp->cmd_len = sizeof(w16CmdBlk);
506*44704f69SBart Van Assche         ptp->dxfer_direction = SG_DXFER_TO_DEV;
507*44704f69SBart Van Assche         ptp->dxferp = lbp;
508*44704f69SBart Van Assche         ptp->dxfer_len = xfer_bytes;
509*44704f69SBart Van Assche         break;
510*44704f69SBart Van Assche     }
511*44704f69SBart Van Assche     ptp->interface_id = 'S';
512*44704f69SBart Van Assche     ptp->mx_sb_len = sizeof(sense_buffer);
513*44704f69SBart Van Assche     ptp->sbp = sense_buffer;      /* ignored .... */
514*44704f69SBart Van Assche     ptp->timeout = DEF_TIMEOUT_MS;
515*44704f69SBart Van Assche     ptp->pack_id = pack_id;
516*44704f69SBart Van Assche     ptp->flags = flags;
517*44704f69SBart Van Assche 
518*44704f69SBart Van Assche     for (int k = 0;
519*44704f69SBart Van Assche          (submit ? ioctl(sg_fd, SG_IOSUBMIT_V3, ptp) :
520*44704f69SBart Van Assche                    write(sg_fd, ptp, sizeof(*ptp)) < 0);
521*44704f69SBart Van Assche          ++k) {
522*44704f69SBart Van Assche         if ((ENOMEM == errno) && (k < MAX_CONSEC_NOMEMS)) {
523*44704f69SBart Van Assche             ++enomem;
524*44704f69SBart Van Assche             this_thread::yield();
525*44704f69SBart Van Assche             continue;
526*44704f69SBart Van Assche         } else if (EAGAIN == errno) {
527*44704f69SBart Van Assche             ++eagains;
528*44704f69SBart Van Assche             this_thread::yield();
529*44704f69SBart Van Assche             continue;
530*44704f69SBart Van Assche         } else if (EBUSY == errno) {
531*44704f69SBart Van Assche             ++ebusy;
532*44704f69SBart Van Assche             this_thread::yield();
533*44704f69SBart Van Assche             continue;
534*44704f69SBart Van Assche         } else if (E2BIG == errno) {
535*44704f69SBart Van Assche             ++e2big;
536*44704f69SBart Van Assche             return 2;
537*44704f69SBart Van Assche         } else if (EDOM == errno)
538*44704f69SBart Van Assche             ++edom;
539*44704f69SBart Van Assche         else if (ENOMEM == errno)
540*44704f69SBart Van Assche             pr_rusage(-1);
541*44704f69SBart Van Assche         pr_errno_lk(errno, "%s: %s, pack_id=%d", __func__, np, pack_id);
542*44704f69SBart Van Assche         return -1;
543*44704f69SBart Van Assche     }
544*44704f69SBart Van Assche     return 0;
545*44704f69SBart Van Assche }
546*44704f69SBart Van Assche 
547*44704f69SBart Van Assche static int
finish_sg3_cmd(int sg_fd,command2execute cmd2exe,int & pack_id,bool receive,int wait_ms,unsigned int & enomem,unsigned int & eagains,unsigned int & ebusys,unsigned int & nanosecs)548*44704f69SBart Van Assche finish_sg3_cmd(int sg_fd, command2execute cmd2exe, int & pack_id,
549*44704f69SBart Van Assche                bool receive, int wait_ms, unsigned int & enomem,
550*44704f69SBart Van Assche                unsigned int & eagains, unsigned int & ebusys,
551*44704f69SBart Van Assche                unsigned int & nanosecs)
552*44704f69SBart Van Assche {
553*44704f69SBart Van Assche     bool ok;
554*44704f69SBart Van Assche     int res, k;
555*44704f69SBart Van Assche     uint8_t sense_buffer[64] SG_C_CPP_ZERO_INIT;
556*44704f69SBart Van Assche     const char * np = NULL;
557*44704f69SBart Van Assche     struct sg_io_hdr pt;
558*44704f69SBart Van Assche     struct sg_io_hdr * ptp;
559*44704f69SBart Van Assche     struct sg_io_v4 p4t;
560*44704f69SBart Van Assche 
561*44704f69SBart Van Assche     if (receive) {      /* nest a v3 interface inside a store for v4 */
562*44704f69SBart Van Assche         memset(&p4t, 0, sizeof(p4t));
563*44704f69SBart Van Assche         ptp = (struct sg_io_hdr *)&p4t; /* p4t is larger than pt */
564*44704f69SBart Van Assche     } else {
565*44704f69SBart Van Assche         ptp = &pt;
566*44704f69SBart Van Assche         memset(ptp, 0, sizeof(*ptp));
567*44704f69SBart Van Assche     }
568*44704f69SBart Van Assche     switch (cmd2exe) {
569*44704f69SBart Van Assche     case SCSI_TUR:
570*44704f69SBart Van Assche         np = "TEST UNIT READY";
571*44704f69SBart Van Assche         ptp->dxfer_direction = SG_DXFER_NONE;
572*44704f69SBart Van Assche         break;
573*44704f69SBart Van Assche     case SCSI_READ16:
574*44704f69SBart Van Assche         np = "READ(16)";
575*44704f69SBart Van Assche         ptp->dxfer_direction = SG_DXFER_FROM_DEV;
576*44704f69SBart Van Assche         break;
577*44704f69SBart Van Assche     case SCSI_WRITE16:
578*44704f69SBart Van Assche         np = "WRITE(16)";
579*44704f69SBart Van Assche         ptp->dxfer_direction = SG_DXFER_TO_DEV;
580*44704f69SBart Van Assche         break;
581*44704f69SBart Van Assche     }
582*44704f69SBart Van Assche     ptp->interface_id = 'S';
583*44704f69SBart Van Assche     ptp->mx_sb_len = sizeof(sense_buffer);
584*44704f69SBart Van Assche     ptp->sbp = sense_buffer;
585*44704f69SBart Van Assche     ptp->timeout = DEF_TIMEOUT_MS;
586*44704f69SBart Van Assche     /* if SG_SET_FORCE_PACK_ID, then need to set ptp->dxfer_direction */
587*44704f69SBart Van Assche     ptp->pack_id = pack_id;
588*44704f69SBart Van Assche 
589*44704f69SBart Van Assche     k = 0;
590*44704f69SBart Van Assche     while ((((res = receive ? ioctl(sg_fd, SG_IORECEIVE_V3, ptp) :
591*44704f69SBart Van Assche                               read(sg_fd, ptp, sizeof(*ptp)))) < 0) &&
592*44704f69SBart Van Assche            ((EAGAIN == errno) || (EBUSY == errno) || (ENOMEM == errno))) {
593*44704f69SBart Van Assche         if (ENOMEM == errno)
594*44704f69SBart Van Assche             ++enomem;
595*44704f69SBart Van Assche         else if (EAGAIN == errno)
596*44704f69SBart Van Assche             ++eagains;
597*44704f69SBart Van Assche         else
598*44704f69SBart Van Assche             ++ebusys;
599*44704f69SBart Van Assche         ++k;
600*44704f69SBart Van Assche         if (k > 10000) {
601*44704f69SBart Van Assche             pr2serr_lk("%s: sg_fd=%d: after %d EAGAINs, unable to find "
602*44704f69SBart Van Assche                        "pack_id=%d\n", __func__, sg_fd, k, pack_id);
603*44704f69SBart Van Assche             return -1;      /* crash out */
604*44704f69SBart Van Assche         }
605*44704f69SBart Van Assche         if (wait_ms > 0)
606*44704f69SBart Van Assche             this_thread::sleep_for(milliseconds{wait_ms});
607*44704f69SBart Van Assche         else if (0 == wait_ms)
608*44704f69SBart Van Assche             this_thread::yield();
609*44704f69SBart Van Assche         else if (-2 == wait_ms)
610*44704f69SBart Van Assche             sleep(0);                   // process yield ??
611*44704f69SBart Van Assche     }
612*44704f69SBart Van Assche     if (res < 0) {
613*44704f69SBart Van Assche         if (ENOMEM == errno)
614*44704f69SBart Van Assche             pr_rusage(-1);
615*44704f69SBart Van Assche         pr_errno_lk(errno, "%s: %s", __func__, np);
616*44704f69SBart Van Assche         return -1;
617*44704f69SBart Van Assche     }
618*44704f69SBart Van Assche     /* now for the error processing */
619*44704f69SBart Van Assche     pack_id = ptp->pack_id;
620*44704f69SBart Van Assche     ok = false;
621*44704f69SBart Van Assche     switch (sg_err_category3(ptp)) {
622*44704f69SBart Van Assche     case SG_LIB_CAT_CLEAN:
623*44704f69SBart Van Assche         ok = true;
624*44704f69SBart Van Assche         break;
625*44704f69SBart Van Assche     case SG_LIB_CAT_RECOVERED:
626*44704f69SBart Van Assche         pr2serr_lk("%s: Recovered error on %s, continuing\n", __func__, np);
627*44704f69SBart Van Assche         ok = true;
628*44704f69SBart Van Assche         break;
629*44704f69SBart Van Assche     default: /* won't bother decoding other categories */
630*44704f69SBart Van Assche         {
631*44704f69SBart Van Assche             lock_guard<mutex> lg(console_mutex);
632*44704f69SBart Van Assche             sg_chk_n_print3(np, ptp, 1);
633*44704f69SBart Van Assche         }
634*44704f69SBart Van Assche         break;
635*44704f69SBart Van Assche     }
636*44704f69SBart Van Assche     if (ok)
637*44704f69SBart Van Assche         nanosecs = ptp->duration;
638*44704f69SBart Van Assche     return ok ? 0 : -1;
639*44704f69SBart Van Assche }
640*44704f69SBart Van Assche 
641*44704f69SBart Van Assche /* Returns 0 if command injected okay, return -1 for error and 2 for
642*44704f69SBart Van Assche  * not done due to queue data size limit struck. */
643*44704f69SBart Van Assche static int
start_sg4_cmd(int sg_fd,command2execute cmd2exe,int pack_id,uint64_t lba,uint8_t * lbp,int xfer_bytes,int flags,bool submit,unsigned int & enomem,unsigned int & eagains,unsigned int & ebusy,unsigned int & e2big,unsigned int & edom)644*44704f69SBart Van Assche start_sg4_cmd(int sg_fd, command2execute cmd2exe, int pack_id, uint64_t lba,
645*44704f69SBart Van Assche               uint8_t * lbp, int xfer_bytes, int flags, bool submit,
646*44704f69SBart Van Assche               unsigned int & enomem, unsigned int & eagains,
647*44704f69SBart Van Assche               unsigned int & ebusy, unsigned int & e2big, unsigned int & edom)
648*44704f69SBart Van Assche {
649*44704f69SBart Van Assche     struct sg_io_v4 p4t;
650*44704f69SBart Van Assche     uint8_t turCmdBlk[TUR_CMD_LEN] = {0, 0, 0, 0, 0, 0};
651*44704f69SBart Van Assche     uint8_t r16CmdBlk[READ16_CMD_LEN] =
652*44704f69SBart Van Assche                 {0x88, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0};
653*44704f69SBart Van Assche     uint8_t w16CmdBlk[WRITE16_CMD_LEN] =
654*44704f69SBart Van Assche                 {0x8a, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0};
655*44704f69SBart Van Assche     uint8_t sense_buffer[64] SG_C_CPP_ZERO_INIT;
656*44704f69SBart Van Assche     const char * np = NULL;
657*44704f69SBart Van Assche     struct sg_io_v4 * ptp;
658*44704f69SBart Van Assche 
659*44704f69SBart Van Assche     if (! submit) {
660*44704f69SBart Van Assche         pr2serr_lk("%s: logic error, submit must be true, isn't\n", __func__);
661*44704f69SBart Van Assche         return -1;
662*44704f69SBart Van Assche     }
663*44704f69SBart Van Assche     ptp = &p4t;
664*44704f69SBart Van Assche     memset(ptp, 0, sizeof(*ptp));
665*44704f69SBart Van Assche     switch (cmd2exe) {
666*44704f69SBart Van Assche     case SCSI_TUR:
667*44704f69SBart Van Assche         np = "TEST UNIT READY";
668*44704f69SBart Van Assche         ptp->request = (uint64_t)turCmdBlk;
669*44704f69SBart Van Assche         ptp->request_len = sizeof(turCmdBlk);
670*44704f69SBart Van Assche         break;
671*44704f69SBart Van Assche     case SCSI_READ16:
672*44704f69SBart Van Assche         np = "READ(16)";
673*44704f69SBart Van Assche         if (lba > 0xffffffff)
674*44704f69SBart Van Assche             sg_put_unaligned_be32(lba >> 32, &r16CmdBlk[2]);
675*44704f69SBart Van Assche         sg_put_unaligned_be32(lba & 0xffffffff, &r16CmdBlk[6]);
676*44704f69SBart Van Assche         ptp->request = (uint64_t)r16CmdBlk;
677*44704f69SBart Van Assche         ptp->request_len = sizeof(r16CmdBlk);
678*44704f69SBart Van Assche         ptp->din_xferp = (uint64_t)lbp;
679*44704f69SBart Van Assche         ptp->din_xfer_len = xfer_bytes;
680*44704f69SBart Van Assche         break;
681*44704f69SBart Van Assche     case SCSI_WRITE16:
682*44704f69SBart Van Assche         np = "WRITE(16)";
683*44704f69SBart Van Assche         if (lba > 0xffffffff)
684*44704f69SBart Van Assche             sg_put_unaligned_be32(lba >> 32, &w16CmdBlk[2]);
685*44704f69SBart Van Assche         sg_put_unaligned_be32(lba & 0xffffffff, &w16CmdBlk[6]);
686*44704f69SBart Van Assche         ptp->request = (uint64_t)w16CmdBlk;
687*44704f69SBart Van Assche         ptp->request_len = sizeof(w16CmdBlk);
688*44704f69SBart Van Assche         ptp->dout_xferp = (uint64_t)lbp;
689*44704f69SBart Van Assche         ptp->dout_xfer_len = xfer_bytes;
690*44704f69SBart Van Assche         break;
691*44704f69SBart Van Assche     }
692*44704f69SBart Van Assche     ptp->guard = 'Q';
693*44704f69SBart Van Assche     ptp->max_response_len = sizeof(sense_buffer);
694*44704f69SBart Van Assche     ptp->response = (uint64_t)sense_buffer;      /* ignored .... */
695*44704f69SBart Van Assche     ptp->timeout = DEF_TIMEOUT_MS;
696*44704f69SBart Van Assche     ptp->request_extra = pack_id;
697*44704f69SBart Van Assche     ptp->flags = flags;
698*44704f69SBart Van Assche 
699*44704f69SBart Van Assche     for (int k = 0; ioctl(sg_fd, SG_IOSUBMIT, ptp) < 0; ++k) {
700*44704f69SBart Van Assche         if ((ENOMEM == errno) && (k < MAX_CONSEC_NOMEMS)) {
701*44704f69SBart Van Assche             ++enomem;
702*44704f69SBart Van Assche             this_thread::yield();
703*44704f69SBart Van Assche             continue;
704*44704f69SBart Van Assche         } else if (EAGAIN == errno) {
705*44704f69SBart Van Assche             ++eagains;
706*44704f69SBart Van Assche             this_thread::yield();
707*44704f69SBart Van Assche             continue;
708*44704f69SBart Van Assche         } else if (EBUSY == errno) {
709*44704f69SBart Van Assche             ++ebusy;
710*44704f69SBart Van Assche             this_thread::yield();
711*44704f69SBart Van Assche             continue;
712*44704f69SBart Van Assche         } else if (E2BIG == errno) {
713*44704f69SBart Van Assche             ++e2big;
714*44704f69SBart Van Assche             return 2;
715*44704f69SBart Van Assche         } else if (EDOM == errno)
716*44704f69SBart Van Assche             ++edom;
717*44704f69SBart Van Assche         else if (ENOMEM == errno)
718*44704f69SBart Van Assche             pr_rusage(-1);
719*44704f69SBart Van Assche         pr_errno_lk(errno, "%s: %s, pack_id=%d", __func__, np, pack_id);
720*44704f69SBart Van Assche         return -1;
721*44704f69SBart Van Assche     }
722*44704f69SBart Van Assche     return 0;
723*44704f69SBart Van Assche }
724*44704f69SBart Van Assche 
725*44704f69SBart Van Assche static int
finish_sg4_cmd(int sg_fd,command2execute cmd2exe,int & pack_id,bool receive,int wait_ms,unsigned int & enomem,unsigned int & eagains,unsigned int & ebusys,unsigned int & nanosecs)726*44704f69SBart Van Assche finish_sg4_cmd(int sg_fd, command2execute cmd2exe, int & pack_id,
727*44704f69SBart Van Assche                bool receive, int wait_ms, unsigned int & enomem,
728*44704f69SBart Van Assche                unsigned int & eagains, unsigned int & ebusys,
729*44704f69SBart Van Assche                unsigned int & nanosecs)
730*44704f69SBart Van Assche {
731*44704f69SBart Van Assche     bool ok;
732*44704f69SBart Van Assche     int res, k;
733*44704f69SBart Van Assche     uint8_t sense_buffer[64] SG_C_CPP_ZERO_INIT;
734*44704f69SBart Van Assche     const char * np = NULL;
735*44704f69SBart Van Assche     struct sg_io_v4 * ptp;
736*44704f69SBart Van Assche     struct sg_io_v4 p4t;
737*44704f69SBart Van Assche 
738*44704f69SBart Van Assche     if (! receive) {
739*44704f69SBart Van Assche         pr2serr_lk("%s: logic error, receive must be true, isn't\n",
740*44704f69SBart Van Assche                    __func__);
741*44704f69SBart Van Assche         return -1;
742*44704f69SBart Van Assche     }
743*44704f69SBart Van Assche     ptp = &p4t;
744*44704f69SBart Van Assche     memset(ptp, 0, sizeof(*ptp));
745*44704f69SBart Van Assche     switch (cmd2exe) {
746*44704f69SBart Van Assche     case SCSI_TUR:
747*44704f69SBart Van Assche         np = "TEST UNIT READY";
748*44704f69SBart Van Assche         break;
749*44704f69SBart Van Assche     case SCSI_READ16:
750*44704f69SBart Van Assche         np = "READ(16)";
751*44704f69SBart Van Assche         break;
752*44704f69SBart Van Assche     case SCSI_WRITE16:
753*44704f69SBart Van Assche         np = "WRITE(16)";
754*44704f69SBart Van Assche         break;
755*44704f69SBart Van Assche     }
756*44704f69SBart Van Assche     ptp->guard = 'Q';
757*44704f69SBart Van Assche     ptp->max_response_len = sizeof(sense_buffer);
758*44704f69SBart Van Assche     ptp->response = (uint64_t)sense_buffer;
759*44704f69SBart Van Assche     ptp->timeout = DEF_TIMEOUT_MS;
760*44704f69SBart Van Assche     /* if SG_SET_FORCE_PACK_ID, then need to set ptp->dxfer_direction */
761*44704f69SBart Van Assche     ptp->request_extra = pack_id;
762*44704f69SBart Van Assche 
763*44704f69SBart Van Assche     k = 0;
764*44704f69SBart Van Assche     while ((((res = ioctl(sg_fd, SG_IORECEIVE, ptp))) < 0) &&
765*44704f69SBart Van Assche            ((EAGAIN == errno) || (EBUSY == errno))) {
766*44704f69SBart Van Assche         if (EAGAIN == errno)
767*44704f69SBart Van Assche             ++eagains;
768*44704f69SBart Van Assche         else
769*44704f69SBart Van Assche             ++ebusys;
770*44704f69SBart Van Assche         ++k;
771*44704f69SBart Van Assche         if (k > 10000) {
772*44704f69SBart Van Assche             pr2serr_lk("%s: sg_fd=%d: after %d EAGAINs, unable to find "
773*44704f69SBart Van Assche                        "pack_id=%d\n", __func__, sg_fd, k, pack_id);
774*44704f69SBart Van Assche             return -1;      /* crash out */
775*44704f69SBart Van Assche         }
776*44704f69SBart Van Assche         if (wait_ms > 0)
777*44704f69SBart Van Assche             this_thread::sleep_for(milliseconds{wait_ms});
778*44704f69SBart Van Assche         else if (0 == wait_ms)
779*44704f69SBart Van Assche             this_thread::yield();
780*44704f69SBart Van Assche         else if (-2 == wait_ms)
781*44704f69SBart Van Assche             sleep(0);                   // process yield ??
782*44704f69SBart Van Assche     }
783*44704f69SBart Van Assche     if (res < 0) {
784*44704f69SBart Van Assche         if (ENOMEM == errno) {
785*44704f69SBart Van Assche             ++enomem;
786*44704f69SBart Van Assche             pr_rusage(-1);
787*44704f69SBart Van Assche         }
788*44704f69SBart Van Assche         pr_errno_lk(errno, "%s: %s", __func__, np);
789*44704f69SBart Van Assche         return -1;
790*44704f69SBart Van Assche     }
791*44704f69SBart Van Assche     /* now for the error processing */
792*44704f69SBart Van Assche     pack_id = ptp->request_extra;
793*44704f69SBart Van Assche     ok = false;
794*44704f69SBart Van Assche     res = sg_err_category_new(ptp->device_status, ptp->transport_status,
795*44704f69SBart Van Assche                               ptp->driver_status,
796*44704f69SBart Van Assche                               (const uint8_t *)ptp->response,
797*44704f69SBart Van Assche                               ptp->response_len);
798*44704f69SBart Van Assche     switch (res) {
799*44704f69SBart Van Assche     case SG_LIB_CAT_CLEAN:
800*44704f69SBart Van Assche         ok = true;
801*44704f69SBart Van Assche         break;
802*44704f69SBart Van Assche     case SG_LIB_CAT_RECOVERED:
803*44704f69SBart Van Assche         pr2serr_lk("%s: Recovered error on %s, continuing\n", __func__, np);
804*44704f69SBart Van Assche         ok = true;
805*44704f69SBart Van Assche         break;
806*44704f69SBart Van Assche     default: /* won't bother decoding other categories */
807*44704f69SBart Van Assche         {
808*44704f69SBart Van Assche             lock_guard<mutex> lg(console_mutex);
809*44704f69SBart Van Assche 
810*44704f69SBart Van Assche             sg_linux_sense_print(np, ptp->device_status,
811*44704f69SBart Van Assche                                  ptp->transport_status,
812*44704f69SBart Van Assche                                  ptp->driver_status,
813*44704f69SBart Van Assche                                  (const uint8_t *)ptp->response,
814*44704f69SBart Van Assche                                  ptp->response_len, true);
815*44704f69SBart Van Assche         }
816*44704f69SBart Van Assche         break;
817*44704f69SBart Van Assche     }
818*44704f69SBart Van Assche     if (ok)
819*44704f69SBart Van Assche         nanosecs = ptp->duration;
820*44704f69SBart Van Assche     return ok ? 0 : -1;
821*44704f69SBart Van Assche }
822*44704f69SBart Van Assche 
823*44704f69SBart Van Assche static int
num_submitted(int sg_fd)824*44704f69SBart Van Assche num_submitted(int sg_fd)
825*44704f69SBart Van Assche {
826*44704f69SBart Van Assche     uint32_t num_subm_wait = 0;
827*44704f69SBart Van Assche     struct sg_extended_info sei;
828*44704f69SBart Van Assche     struct sg_extended_info *seip = &sei;
829*44704f69SBart Van Assche     const char * err = NULL;
830*44704f69SBart Van Assche 
831*44704f69SBart Van Assche     memset(seip, 0, sizeof(*seip));
832*44704f69SBart Van Assche     seip->sei_wr_mask |= SG_SEIM_READ_VAL;
833*44704f69SBart Van Assche     seip->sei_rd_mask |= SG_SEIM_READ_VAL;
834*44704f69SBart Van Assche     seip->read_value = SG_SEIRV_SUBMITTED;
835*44704f69SBart Van Assche     if (ioctl(sg_fd, SG_SET_GET_EXTENDED, seip) < 0)
836*44704f69SBart Van Assche         err = "ioctl(SG_SET_GET_EXTENDED) failed\n";
837*44704f69SBart Van Assche     else
838*44704f69SBart Van Assche         num_subm_wait = seip->read_value;
839*44704f69SBart Van Assche     if (err)
840*44704f69SBart Van Assche         pr2serr_lk("%s: %s, errno=%d\n", __func__, err, errno);
841*44704f69SBart Van Assche     return err ? -1 : (int)num_subm_wait;
842*44704f69SBart Van Assche }
843*44704f69SBart Van Assche 
844*44704f69SBart Van Assche static int
pr_rusage(int id)845*44704f69SBart Van Assche pr_rusage(int id)
846*44704f69SBart Van Assche {
847*44704f69SBart Van Assche     int res;
848*44704f69SBart Van Assche     struct rusage ru;
849*44704f69SBart Van Assche 
850*44704f69SBart Van Assche     res = getrusage(RUSAGE_SELF /* RUSAGE_THREAD */, &ru);
851*44704f69SBart Van Assche     if (res < 0) {
852*44704f69SBart Van Assche         pr2serr_lk("%d->id: %s: getrusage() failed, errno=%d\n", id,
853*44704f69SBart Van Assche                    __func__, errno);
854*44704f69SBart Van Assche         return res;
855*44704f69SBart Van Assche     }
856*44704f69SBart Van Assche     pr2serr_lk("%d->id: maxrss=%ldKB  nvcsw=%ld nivcsw=%ld  majflt=%ld\n", id,
857*44704f69SBart Van Assche                ru.ru_maxrss, ru.ru_nvcsw, ru.ru_nivcsw, ru.ru_majflt);
858*44704f69SBart Van Assche     return 0;
859*44704f69SBart Van Assche }
860*44704f69SBart Van Assche 
861*44704f69SBart Van Assche static void
work_sync_thread(int id,const char * dev_name,unsigned int,struct opts_t * op)862*44704f69SBart Van Assche work_sync_thread(int id, const char * dev_name, unsigned int /* hi_lba */,
863*44704f69SBart Van Assche                  struct opts_t * op)
864*44704f69SBart Van Assche {
865*44704f69SBart Van Assche     bool is_rw = (SCSI_TUR != op->c2e);
866*44704f69SBart Van Assche     int k, sg_fd, err, rs, n, sense_cat, ret;
867*44704f69SBart Van Assche     int vb = op->verbose;
868*44704f69SBart Van Assche     int num_errs = 0;
869*44704f69SBart Van Assche     int thr_sync_starts = 0;
870*44704f69SBart Van Assche     struct sg_pt_base * ptp = NULL;
871*44704f69SBart Van Assche     uint8_t cdb[6];
872*44704f69SBart Van Assche     uint8_t sense_b[32] SG_C_CPP_ZERO_INIT;
873*44704f69SBart Van Assche     char b[120];
874*44704f69SBart Van Assche 
875*44704f69SBart Van Assche     if (is_rw) {
876*44704f69SBart Van Assche         pr2serr_lk("id=%d: only support TUR here for now\n", id);
877*44704f69SBart Van Assche         goto err_out;
878*44704f69SBart Van Assche     }
879*44704f69SBart Van Assche     if (op->verbose)
880*44704f69SBart Van Assche         pr2serr_lk("id=%d: using libsgutils generic sync passthrough\n", id);
881*44704f69SBart Van Assche 
882*44704f69SBart Van Assche     if ((sg_fd = sg_cmds_open_device(dev_name, false /* ro */, vb)) < 0) {
883*44704f69SBart Van Assche         pr2serr_lk("id=%d: error opening file: %s: %s\n", id, dev_name,
884*44704f69SBart Van Assche                    safe_strerror(-sg_fd));
885*44704f69SBart Van Assche         if (ENOMEM == -sg_fd)
886*44704f69SBart Van Assche             pr_rusage(id);
887*44704f69SBart Van Assche         goto err_out;
888*44704f69SBart Van Assche     }
889*44704f69SBart Van Assche     if (vb > 2)
890*44704f69SBart Van Assche         pr2serr_lk(">>>> id=%d: open(%s) --> fd=%d\n", id, dev_name, sg_fd);
891*44704f69SBart Van Assche 
892*44704f69SBart Van Assche     ptp = construct_scsi_pt_obj_with_fd(sg_fd, vb);
893*44704f69SBart Van Assche     err = 0;
894*44704f69SBart Van Assche     if ((NULL == ptp) || ((err = get_scsi_pt_os_err(ptp)))) {
895*44704f69SBart Van Assche         ret = sg_convert_errno(err ? err : ENOMEM);
896*44704f69SBart Van Assche         sg_exit2str(ret, true, sizeof(b), b);
897*44704f69SBart Van Assche         pr2serr_lk("id=%d: construct_scsi_pt_obj_with_fd: %s\n", id, b);
898*44704f69SBart Van Assche         goto err_out;
899*44704f69SBart Van Assche     }
900*44704f69SBart Van Assche     for (k = 0; k < op->num_per_thread; ++k) {
901*44704f69SBart Van Assche         /* Might get Unit Attention on first invocation */
902*44704f69SBart Van Assche         memset(cdb, 0, sizeof(cdb));    /* TUR's cdb is 6 zeros */
903*44704f69SBart Van Assche         set_scsi_pt_cdb(ptp, cdb, sizeof(cdb));
904*44704f69SBart Van Assche         set_scsi_pt_sense(ptp, sense_b, sizeof(sense_b));
905*44704f69SBart Van Assche         set_scsi_pt_packet_id(ptp, uniq_pack_id.fetch_add(1));
906*44704f69SBart Van Assche         ++thr_sync_starts;
907*44704f69SBart Van Assche         rs = do_scsi_pt(ptp, -1, DEF_PT_TIMEOUT, vb);
908*44704f69SBart Van Assche         n = sg_cmds_process_resp(ptp, "Test unit ready", rs,
909*44704f69SBart Van Assche                                  (0 == k), vb, &sense_cat);
910*44704f69SBart Van Assche         if (-1 == n) {
911*44704f69SBart Van Assche             ret = sg_convert_errno(get_scsi_pt_os_err(ptp));
912*44704f69SBart Van Assche             sg_exit2str(ret, true, sizeof(b), b);
913*44704f69SBart Van Assche             pr2serr_lk("id=%d: do_scsi_pt: %s\n", id, b);
914*44704f69SBart Van Assche             goto err_out;
915*44704f69SBart Van Assche         } else if (-2 == n) {
916*44704f69SBart Van Assche             switch (sense_cat) {
917*44704f69SBart Van Assche             case SG_LIB_CAT_RECOVERED:
918*44704f69SBart Van Assche             case SG_LIB_CAT_NO_SENSE:
919*44704f69SBart Van Assche                 break;
920*44704f69SBart Van Assche             case SG_LIB_CAT_NOT_READY:
921*44704f69SBart Van Assche                 ++num_errs;
922*44704f69SBart Van Assche                 if (1 ==  op->num_per_thread) {
923*44704f69SBart Van Assche                     pr2serr_lk("id=%d: device not ready\n", id);
924*44704f69SBart Van Assche                 }
925*44704f69SBart Van Assche                 break;
926*44704f69SBart Van Assche             case SG_LIB_CAT_UNIT_ATTENTION:
927*44704f69SBart Van Assche                 ++num_errs;
928*44704f69SBart Van Assche                 if (vb)
929*44704f69SBart Van Assche                     pr2serr_lk("Ignoring Unit attention (sense key)\n");
930*44704f69SBart Van Assche                 break;
931*44704f69SBart Van Assche             default:
932*44704f69SBart Van Assche                 ++num_errs;
933*44704f69SBart Van Assche                 if (1 == op->num_per_thread) {
934*44704f69SBart Van Assche                     sg_get_category_sense_str(sense_cat, sizeof(b), b, vb);
935*44704f69SBart Van Assche                     pr2serr_lk("%s\n", b);
936*44704f69SBart Van Assche                     goto err_out;
937*44704f69SBart Van Assche                 }
938*44704f69SBart Van Assche                 break;
939*44704f69SBart Van Assche             }
940*44704f69SBart Van Assche         }
941*44704f69SBart Van Assche         clear_scsi_pt_obj(ptp);
942*44704f69SBart Van Assche     }
943*44704f69SBart Van Assche err_out:
944*44704f69SBart Van Assche     if (ptp)
945*44704f69SBart Van Assche         destruct_scsi_pt_obj(ptp);
946*44704f69SBart Van Assche     if (num_errs > 0)
947*44704f69SBart Van Assche         pr2serr_lk("id=%d: number of errors: %d\n", id, num_errs);
948*44704f69SBart Van Assche     sync_starts += thr_sync_starts;
949*44704f69SBart Van Assche }
950*44704f69SBart Van Assche 
951*44704f69SBart Van Assche static void
work_thread(int id,struct opts_t * op)952*44704f69SBart Van Assche work_thread(int id, struct opts_t * op)
953*44704f69SBart Van Assche {
954*44704f69SBart Van Assche     bool is_rw = (SCSI_TUR != op->c2e);
955*44704f69SBart Van Assche     bool need_finish, repeat;
956*44704f69SBart Van Assche     bool once = false;
957*44704f69SBart Van Assche     bool once1000 = false;
958*44704f69SBart Van Assche     bool once_2000 = false;
959*44704f69SBart Van Assche     bool once_4000 = false;
960*44704f69SBart Van Assche     bool once5000 = false;
961*44704f69SBart Van Assche     bool once_6000 = false;
962*44704f69SBart Van Assche     bool once_7000 = false;
963*44704f69SBart Van Assche     bool once10_000 = false;
964*44704f69SBart Van Assche     bool once20_000 = false;
965*44704f69SBart Van Assche     int open_flags = O_RDWR;
966*44704f69SBart Van Assche     int thr_async_starts = 0;
967*44704f69SBart Van Assche     int thr_async_finishes = 0;
968*44704f69SBart Van Assche     int vb = op->verbose;
969*44704f69SBart Van Assche     int k, n, res, sg_fd, num_outstanding, do_inc, npt, pack_id, sg_flags;
970*44704f69SBart Van Assche     int num_waiting_read, sz, encore_pack_id, ask, j, m, o;
971*44704f69SBart Van Assche     int prev_pack_id, blk_sz;
972*44704f69SBart Van Assche     unsigned int thr_enomem_count = 0;
973*44704f69SBart Van Assche     unsigned int thr_start_eagain_count = 0;
974*44704f69SBart Van Assche     unsigned int thr_start_ebusy_count = 0;
975*44704f69SBart Van Assche     unsigned int thr_start_e2big_count = 0;
976*44704f69SBart Van Assche     unsigned int thr_fin_eagain_count = 0;
977*44704f69SBart Van Assche     unsigned int thr_fin_ebusy_count = 0;
978*44704f69SBart Van Assche     unsigned int thr_start_edom_count = 0;
979*44704f69SBart Van Assche     int needed_sz = op->lb_sz * op->num_lbs;
980*44704f69SBart Van Assche     unsigned int nanosecs;
981*44704f69SBart Van Assche     unsigned int hi_lba;
982*44704f69SBart Van Assche     uint64_t lba;
983*44704f69SBart Van Assche     uint64_t sum_nanosecs = 0;
984*44704f69SBart Van Assche     uint8_t * lbp;
985*44704f69SBart Van Assche     uint8_t * free_lbp = NULL;
986*44704f69SBart Van Assche     uint8_t * wrkMmap = NULL;
987*44704f69SBart Van Assche     const char * dev_name;
988*44704f69SBart Van Assche     const char * err = NULL;
989*44704f69SBart Van Assche     Rand_uint * ruip = NULL;
990*44704f69SBart Van Assche     char ebuff[EBUFF_SZ];
991*44704f69SBart Van Assche     struct pollfd  pfd[1];
992*44704f69SBart Van Assche     list<pair<uint8_t *, uint8_t *> > free_lst;   /* of aligned lb buffers */
993*44704f69SBart Van Assche     map<int, pair<uint8_t *, uint8_t *> > pi2buff;/* pack_id -> lb buffer */
994*44704f69SBart Van Assche     map<int, uint64_t> pi_2_lba;            /* pack_id -> LBA */
995*44704f69SBart Van Assche     pair<uint8_t *, uint8_t *> encore_lbps;
996*44704f69SBart Van Assche 
997*44704f69SBart Van Assche     /* device name and hi_lba may depend on id */
998*44704f69SBart Van Assche     n = op->dev_names.size();
999*44704f69SBart Van Assche     dev_name = op->dev_names[id % n];
1000*44704f69SBart Van Assche     if (op->blk_szs.size() >= (unsigned)n)
1001*44704f69SBart Van Assche         blk_sz = op->blk_szs[id % n];
1002*44704f69SBart Van Assche     else
1003*44704f69SBart Van Assche         blk_sz = DEF_LB_SZ;
1004*44704f69SBart Van Assche     if ((UINT_MAX == op->hi_lba) && (n == (int)op->hi_lbas.size()))
1005*44704f69SBart Van Assche         hi_lba = op->hi_lbas[id % n];
1006*44704f69SBart Van Assche     else
1007*44704f69SBart Van Assche         hi_lba = op->hi_lba;
1008*44704f69SBart Van Assche 
1009*44704f69SBart Van Assche     if (vb) {
1010*44704f69SBart Van Assche         if ((vb > 1) && hi_lba)
1011*44704f69SBart Van Assche             pr2serr_lk("Enter work_t_id=%d using %s\n"
1012*44704f69SBart Van Assche                        "    LBA range: 0x%x to 0x%x (inclusive)\n",
1013*44704f69SBart Van Assche                        id, dev_name, (unsigned int)op->lba, hi_lba);
1014*44704f69SBart Van Assche         else
1015*44704f69SBart Van Assche             pr2serr_lk("Enter work_t_id=%d using %s\n", id, dev_name);
1016*44704f69SBart Van Assche     }
1017*44704f69SBart Van Assche     if (op->generic_sync) {
1018*44704f69SBart Van Assche         work_sync_thread(id, dev_name, hi_lba, op);
1019*44704f69SBart Van Assche         return;
1020*44704f69SBart Van Assche     }
1021*44704f69SBart Van Assche     if (! op->block)
1022*44704f69SBart Van Assche         open_flags |= O_NONBLOCK;
1023*44704f69SBart Van Assche 
1024*44704f69SBart Van Assche     sg_fd = open(dev_name, open_flags);
1025*44704f69SBart Van Assche     if (sg_fd < 0) {
1026*44704f69SBart Van Assche         pr_errno_lk(errno, "%s: id=%d, error opening file: %s", __func__, id,
1027*44704f69SBart Van Assche                     dev_name);
1028*44704f69SBart Van Assche         if (ENOMEM == -sg_fd)
1029*44704f69SBart Van Assche             pr_rusage(id);
1030*44704f69SBart Van Assche         return;
1031*44704f69SBart Van Assche     }
1032*44704f69SBart Van Assche     if (vb > 2)
1033*44704f69SBart Van Assche         pr2serr_lk(">>>> id=%d: open(%s) --> fd=%d\n", id, dev_name, sg_fd);
1034*44704f69SBart Van Assche     if (op->pack_id_force) {
1035*44704f69SBart Van Assche         k = 1;
1036*44704f69SBart Van Assche         if (ioctl(sg_fd, SG_SET_FORCE_PACK_ID, &k) < 0)
1037*44704f69SBart Van Assche             pr2serr_lk("ioctl(SG_SET_FORCE_PACK_ID) failed, errno=%d %s\n",
1038*44704f69SBart Van Assche                        errno, strerror(errno));
1039*44704f69SBart Van Assche     }
1040*44704f69SBart Van Assche     if (op->sg_vn_ge_40000) {
1041*44704f69SBart Van Assche         if (ioctl(sg_fd, SG_GET_RESERVED_SIZE, &k) >= 0) {
1042*44704f69SBart Van Assche             if (needed_sz > k)
1043*44704f69SBart Van Assche                 ioctl(sg_fd, SG_SET_RESERVED_SIZE, &needed_sz);
1044*44704f69SBart Van Assche         }
1045*44704f69SBart Van Assche         if (op->sg_vn_ge_40030 && (op->cmd_time || op->masync)) {
1046*44704f69SBart Van Assche             struct sg_extended_info sei;
1047*44704f69SBart Van Assche             struct sg_extended_info * seip;
1048*44704f69SBart Van Assche 
1049*44704f69SBart Van Assche             seip = &sei;
1050*44704f69SBart Van Assche             memset(seip, 0, sizeof(*seip));
1051*44704f69SBart Van Assche             seip->sei_wr_mask |= SG_SEIM_CTL_FLAGS;
1052*44704f69SBart Van Assche             seip->sei_rd_mask |= SG_SEIM_CTL_FLAGS;
1053*44704f69SBart Van Assche             if (op->cmd_time) {
1054*44704f69SBart Van Assche                 seip->ctl_flags_wr_mask |= SG_CTL_FLAGM_TIME_IN_NS;
1055*44704f69SBart Van Assche                 seip->ctl_flags_rd_mask |= SG_CTL_FLAGM_TIME_IN_NS;
1056*44704f69SBart Van Assche                 seip->ctl_flags |= SG_CTL_FLAGM_TIME_IN_NS;
1057*44704f69SBart Van Assche             }
1058*44704f69SBart Van Assche             if (op->masync) {
1059*44704f69SBart Van Assche                 seip->ctl_flags_wr_mask |= SG_CTL_FLAGM_MORE_ASYNC;
1060*44704f69SBart Van Assche                 seip->ctl_flags |= SG_CTL_FLAGM_MORE_ASYNC;
1061*44704f69SBart Van Assche             }
1062*44704f69SBart Van Assche             if (op->excl) {
1063*44704f69SBart Van Assche                 seip->ctl_flags_wr_mask |= SG_CTL_FLAGM_EXCL_WAITQ;
1064*44704f69SBart Van Assche                 seip->ctl_flags |= SG_CTL_FLAGM_EXCL_WAITQ;
1065*44704f69SBart Van Assche             }
1066*44704f69SBart Van Assche             if (ioctl(sg_fd, SG_SET_GET_EXTENDED, seip) < 0) {
1067*44704f69SBart Van Assche                 pr2serr_lk("ioctl(EXTENDED(TIME_IN_NS)) failed, errno=%d %s\n",
1068*44704f69SBart Van Assche                            errno, strerror(errno));
1069*44704f69SBart Van Assche             }
1070*44704f69SBart Van Assche             if (op->cmd_time &&
1071*44704f69SBart Van Assche                 (! (SG_CTL_FLAGM_TIME_IN_NS & seip->ctl_flags))) {
1072*44704f69SBart Van Assche                 memset(seip, 0, sizeof(*seip));
1073*44704f69SBart Van Assche                 seip->sei_rd_mask |= SG_SEIM_CTL_FLAGS;
1074*44704f69SBart Van Assche                 seip->sei_wr_mask |= SG_SEIM_CTL_FLAGS;
1075*44704f69SBart Van Assche                 seip->ctl_flags_wr_mask |= SG_CTL_FLAGM_TIME_IN_NS;
1076*44704f69SBart Van Assche                 seip->ctl_flags |= SG_CTL_FLAGM_TIME_IN_NS;
1077*44704f69SBart Van Assche                 if (ioctl(sg_fd, SG_SET_GET_EXTENDED, seip) < 0)
1078*44704f69SBart Van Assche                     pr2serr_lk("ioctl(EXTENDED(TIME_IN_NS)) failed, "
1079*44704f69SBart Van Assche                                "errno=%d %s\n", errno, strerror(errno));
1080*44704f69SBart Van Assche                 else if (vb > 1)
1081*44704f69SBart Van Assche                     pr2serr_lk("t_id: %d: set TIME_IN_NS flag\n", id);
1082*44704f69SBart Van Assche             }
1083*44704f69SBart Van Assche         }
1084*44704f69SBart Van Assche     }
1085*44704f69SBart Van Assche     if (is_rw && op->mmap_io) {
1086*44704f69SBart Van Assche 
1087*44704f69SBart Van Assche         if (ioctl(sg_fd, SG_GET_RESERVED_SIZE, &sz) < 0) {
1088*44704f69SBart Van Assche             pr2serr_lk("t_id=%d: ioctl(SG_GET_RESERVED_SIZE) errno=%d\n",
1089*44704f69SBart Van Assche                        id, errno);
1090*44704f69SBart Van Assche             return;
1091*44704f69SBart Van Assche         }
1092*44704f69SBart Van Assche         if (sz < needed_sz) {
1093*44704f69SBart Van Assche             sz = needed_sz;
1094*44704f69SBart Van Assche             if (ioctl(sg_fd, SG_SET_RESERVED_SIZE, &sz) < 0) {
1095*44704f69SBart Van Assche                 pr2serr_lk("t_id=%d: ioctl(SG_SET_RESERVED_SIZE) errno=%d\n",
1096*44704f69SBart Van Assche                            id, errno);
1097*44704f69SBart Van Assche                 return;
1098*44704f69SBart Van Assche             }
1099*44704f69SBart Van Assche             if (ioctl(sg_fd, SG_GET_RESERVED_SIZE, &sz) < 0) {
1100*44704f69SBart Van Assche                 pr2serr_lk("t_id=%d: ioctl(SG_GET_RESERVED_SIZE) errno=%d\n",
1101*44704f69SBart Van Assche                            id, errno);
1102*44704f69SBart Van Assche                 return;
1103*44704f69SBart Van Assche             }
1104*44704f69SBart Van Assche             if (sz < needed_sz) {
1105*44704f69SBart Van Assche                 pr2serr_lk("t_id=%d: unable to grow reserve buffer to %d "
1106*44704f69SBart Van Assche                            "bytes\n", id, needed_sz);
1107*44704f69SBart Van Assche                 return;
1108*44704f69SBart Van Assche             }
1109*44704f69SBart Van Assche         }
1110*44704f69SBart Van Assche         wrkMmap = (uint8_t *)mmap(NULL, needed_sz, PROT_READ | PROT_WRITE,
1111*44704f69SBart Van Assche                                   MAP_SHARED, sg_fd, 0);
1112*44704f69SBart Van Assche         if (MAP_FAILED == wrkMmap) {
1113*44704f69SBart Van Assche             int ern = errno;
1114*44704f69SBart Van Assche 
1115*44704f69SBart Van Assche             pr2serr_lk("t_id=%d: mmap() failed, errno=%d\n", id, ern);
1116*44704f69SBart Van Assche             return;
1117*44704f69SBart Van Assche         }
1118*44704f69SBart Van Assche     }
1119*44704f69SBart Van Assche     pfd[0].fd = sg_fd;
1120*44704f69SBart Van Assche     pfd[0].events = POLLIN;
1121*44704f69SBart Van Assche     if (is_rw && hi_lba) {
1122*44704f69SBart Van Assche         unsigned int seed = get_urandom_uint();
1123*44704f69SBart Van Assche 
1124*44704f69SBart Van Assche         if (vb > 1)
1125*44704f69SBart Van Assche             pr2serr_lk("  id=%d, /dev/urandom seed=0x%x\n", id, seed);
1126*44704f69SBart Van Assche         ruip = new Rand_uint((unsigned int)op->lba, hi_lba, seed);
1127*44704f69SBart Van Assche     }
1128*44704f69SBart Van Assche 
1129*44704f69SBart Van Assche     sg_flags = 0;
1130*44704f69SBart Van Assche     if (BLQ_AT_TAIL == op->blqd)
1131*44704f69SBart Van Assche         sg_flags |= SG_FLAG_Q_AT_TAIL;
1132*44704f69SBart Van Assche     else if (BLQ_AT_HEAD == op->blqd)
1133*44704f69SBart Van Assche         sg_flags |= SG_FLAG_Q_AT_HEAD;
1134*44704f69SBart Van Assche     if (op->direct)
1135*44704f69SBart Van Assche         sg_flags |= SG_FLAG_DIRECT_IO;
1136*44704f69SBart Van Assche     if (op->mmap_io)
1137*44704f69SBart Van Assche         sg_flags |= SG_FLAG_MMAP_IO;
1138*44704f69SBart Van Assche     if (op->no_xfer)
1139*44704f69SBart Van Assche         sg_flags |= SG_FLAG_NO_DXFER;
1140*44704f69SBart Van Assche     if (vb > 1)
1141*44704f69SBart Van Assche         pr2serr_lk("  id=%d, sg_flags=0x%x, %s cmds\n", id, sg_flags,
1142*44704f69SBart Van Assche                    ((SCSI_TUR == op->c2e) ? "TUR":
1143*44704f69SBart Van Assche                     ((SCSI_READ16 == op->c2e) ? "READ" : "WRITE")));
1144*44704f69SBart Van Assche 
1145*44704f69SBart Van Assche     npt = op->num_per_thread;
1146*44704f69SBart Van Assche     need_finish = false;
1147*44704f69SBart Van Assche     lba = 0;
1148*44704f69SBart Van Assche     pack_id = 0;
1149*44704f69SBart Van Assche     prev_pack_id = 0;
1150*44704f69SBart Van Assche     encore_pack_id = 0;
1151*44704f69SBart Van Assche     do_inc = 0;
1152*44704f69SBart Van Assche     /* main loop, continues until num_per_thread exhausted and there are
1153*44704f69SBart Van Assche      * no more outstanding responses */
1154*44704f69SBart Van Assche     for (k = 0, m = 0, o=0, num_outstanding = 0; (k < npt) || num_outstanding;
1155*44704f69SBart Van Assche          k = do_inc ? k + 1 : k, ++o) {
1156*44704f69SBart Van Assche         int num_to_read = 0;
1157*44704f69SBart Van Assche 
1158*44704f69SBart Van Assche         if (do_inc)
1159*44704f69SBart Van Assche             m = 0;
1160*44704f69SBart Van Assche         else {
1161*44704f69SBart Van Assche             ++m;
1162*44704f69SBart Van Assche             if (m > 100) {
1163*44704f69SBart Van Assche                 if (vb)
1164*44704f69SBart Van Assche                     pr2serr_lk("%d->id: no main loop inc =%d times\n", id, m);
1165*44704f69SBart Van Assche                 m = 0;
1166*44704f69SBart Van Assche             }
1167*44704f69SBart Van Assche         }
1168*44704f69SBart Van Assche         if (vb && (! once1000) && (num_outstanding >= 1000)) {
1169*44704f69SBart Van Assche             int num_waiting;
1170*44704f69SBart Van Assche             int num_subm = (op->sg_vn_ge_40030) ? num_submitted(sg_fd) :
1171*44704f69SBart Van Assche                                                   pi2buff.size();
1172*44704f69SBart Van Assche 
1173*44704f69SBart Van Assche             once1000 = true;
1174*44704f69SBart Van Assche             if (ioctl(sg_fd, SG_GET_NUM_WAITING, &num_waiting) < 0) {
1175*44704f69SBart Van Assche                 err = "ioctl(SG_GET_NUM_WAITING) failed";
1176*44704f69SBart Van Assche                 break;
1177*44704f69SBart Van Assche             }
1178*44704f69SBart Van Assche             pr2serr_lk("%d->id: once 1000: k=%d, submitted=%d waiting=%d; "
1179*44704f69SBart Van Assche                        "pi2buff.sz=%u\n", id, k, num_subm, num_waiting,
1180*44704f69SBart Van Assche                        (uint32_t)pi2buff.size());
1181*44704f69SBart Van Assche             pr_rusage(id);
1182*44704f69SBart Van Assche         }
1183*44704f69SBart Van Assche         if (vb && ! once5000 && num_outstanding >= 5000) {
1184*44704f69SBart Van Assche             int num_waiting;
1185*44704f69SBart Van Assche             int num_subm = (op->sg_vn_ge_40030) ? num_submitted(sg_fd) :
1186*44704f69SBart Van Assche                                                   pi2buff.size();
1187*44704f69SBart Van Assche 
1188*44704f69SBart Van Assche             once5000 = true;
1189*44704f69SBart Van Assche             if (ioctl(sg_fd, SG_GET_NUM_WAITING, &num_waiting) < 0) {
1190*44704f69SBart Van Assche                 err = "ioctl(SG_GET_NUM_WAITING) failed";
1191*44704f69SBart Van Assche                 break;
1192*44704f69SBart Van Assche             }
1193*44704f69SBart Van Assche             pr2serr_lk("%d->id: once 5000: k=%d, submitted=%d waiting=%d\n",
1194*44704f69SBart Van Assche                        id, k, num_subm, num_waiting);
1195*44704f69SBart Van Assche             pr_rusage(id);
1196*44704f69SBart Van Assche         }
1197*44704f69SBart Van Assche         if (vb && ! once_7000 && num_outstanding >= 7000) {
1198*44704f69SBart Van Assche             int num_waiting;
1199*44704f69SBart Van Assche             int num_subm = (op->sg_vn_ge_40030) ? num_submitted(sg_fd) :
1200*44704f69SBart Van Assche                                                   pi2buff.size();
1201*44704f69SBart Van Assche 
1202*44704f69SBart Van Assche             once_7000 = true;
1203*44704f69SBart Van Assche             if (ioctl(sg_fd, SG_GET_NUM_WAITING, &num_waiting) < 0) {
1204*44704f69SBart Van Assche                 err = "ioctl(SG_GET_NUM_WAITING) failed";
1205*44704f69SBart Van Assche                 break;
1206*44704f69SBart Van Assche             }
1207*44704f69SBart Van Assche             pr2serr_lk("%d->id: once 7000: k=%d, submitted=%d waiting=%d\n",
1208*44704f69SBart Van Assche                        id, k, num_subm, num_waiting);
1209*44704f69SBart Van Assche             pr_rusage(id);
1210*44704f69SBart Van Assche         }
1211*44704f69SBart Van Assche         if (vb && ! once10_000 && num_outstanding >= 10000) {
1212*44704f69SBart Van Assche             int num_waiting;
1213*44704f69SBart Van Assche             int num_subm = (op->sg_vn_ge_40030) ? num_submitted(sg_fd) :
1214*44704f69SBart Van Assche                                                   pi2buff.size();
1215*44704f69SBart Van Assche 
1216*44704f69SBart Van Assche             once10_000 = true;
1217*44704f69SBart Van Assche             if (ioctl(sg_fd, SG_GET_NUM_WAITING, &num_waiting) < 0) {
1218*44704f69SBart Van Assche                 err = "ioctl(SG_GET_NUM_WAITING) failed";
1219*44704f69SBart Van Assche                 break;
1220*44704f69SBart Van Assche             }
1221*44704f69SBart Van Assche             pr2serr_lk("%d->id: once 10^4: k=%d, submitted=%d waiting=%d\n",
1222*44704f69SBart Van Assche                        id, k, num_subm, num_waiting);
1223*44704f69SBart Van Assche             pr_rusage(id);
1224*44704f69SBart Van Assche         }
1225*44704f69SBart Van Assche         if (vb && ! once20_000 && num_outstanding >= 20000) {
1226*44704f69SBart Van Assche             int num_waiting;
1227*44704f69SBart Van Assche             int num_subm = (op->sg_vn_ge_40030) ? num_submitted(sg_fd) :
1228*44704f69SBart Van Assche                                                   pi2buff.size();
1229*44704f69SBart Van Assche 
1230*44704f69SBart Van Assche             once20_000 = true;
1231*44704f69SBart Van Assche             if (ioctl(sg_fd, SG_GET_NUM_WAITING, &num_waiting) < 0) {
1232*44704f69SBart Van Assche                 err = "ioctl(SG_GET_NUM_WAITING) failed";
1233*44704f69SBart Van Assche                 break;
1234*44704f69SBart Van Assche             }
1235*44704f69SBart Van Assche             pr2serr_lk("%d->id: once 20000: k=%d, submitted=%d waiting=%d\n",
1236*44704f69SBart Van Assche                        id, k, num_subm, num_waiting);
1237*44704f69SBart Van Assche             pr_rusage(id);
1238*44704f69SBart Van Assche         }
1239*44704f69SBart Van Assche         do_inc = 0;
1240*44704f69SBart Van Assche         if ((num_outstanding < op->maxq_per_thread) && (k < npt)) {
1241*44704f69SBart Van Assche             do_inc = 1;
1242*44704f69SBart Van Assche             if (need_finish) {
1243*44704f69SBart Van Assche                 pack_id = encore_pack_id;
1244*44704f69SBart Van Assche                 need_finish = false;
1245*44704f69SBart Van Assche                 repeat = true;
1246*44704f69SBart Van Assche             } else {
1247*44704f69SBart Van Assche                 prev_pack_id = pack_id;
1248*44704f69SBart Van Assche                 pack_id = uniq_pack_id.fetch_add(1);
1249*44704f69SBart Van Assche                 repeat = false;
1250*44704f69SBart Van Assche             }
1251*44704f69SBart Van Assche             if (is_rw) {    /* get new lb buffer or one from free list */
1252*44704f69SBart Van Assche                 if (free_lst.empty()) {
1253*44704f69SBart Van Assche                     lbp = sg_memalign(op->lb_sz * op->num_lbs, 0, &free_lbp,
1254*44704f69SBart Van Assche                                       false);
1255*44704f69SBart Van Assche                     if (NULL == lbp) {
1256*44704f69SBart Van Assche                         err = "out of memory";
1257*44704f69SBart Van Assche                         break;
1258*44704f69SBart Van Assche                     }
1259*44704f69SBart Van Assche                 } else if (! repeat) {
1260*44704f69SBart Van Assche                     lbp = free_lst.back().first;
1261*44704f69SBart Van Assche                     free_lbp = free_lst.back().second;
1262*44704f69SBart Van Assche                     free_lst.pop_back();
1263*44704f69SBart Van Assche                 } else {
1264*44704f69SBart Van Assche                     lbp = encore_lbps.first;
1265*44704f69SBart Van Assche                     free_lbp = encore_lbps.second;
1266*44704f69SBart Van Assche                     if (vb && !once && free_lst.size() > 1000) {
1267*44704f69SBart Van Assche                         once = true;
1268*44704f69SBart Van Assche                         pr2serr_lk("%d->id: free_lst.size() over 1000\n", id);
1269*44704f69SBart Van Assche                     }
1270*44704f69SBart Van Assche                     if (vb && !once_2000 && free_lst.size() > 2000) {
1271*44704f69SBart Van Assche                         once_2000 = true;
1272*44704f69SBart Van Assche                         pr2serr_lk("%d->id: free_lst.size() over 2000\n", id);
1273*44704f69SBart Van Assche                     }
1274*44704f69SBart Van Assche                     if (vb && !once_6000 && free_lst.size() > 6000) {
1275*44704f69SBart Van Assche                         once_2000 = true;
1276*44704f69SBart Van Assche                         pr2serr_lk("%d->id: free_lst.size() over 6000\n", id);
1277*44704f69SBart Van Assche                     }
1278*44704f69SBart Van Assche                 }
1279*44704f69SBart Van Assche             } else
1280*44704f69SBart Van Assche                 lbp = NULL;
1281*44704f69SBart Van Assche             if (is_rw) {
1282*44704f69SBart Van Assche                 if (ruip) {
1283*44704f69SBart Van Assche                     if (! repeat) {
1284*44704f69SBart Van Assche                         lba = ruip->get();  /* fetch a random LBA */
1285*44704f69SBart Van Assche                         if (vb > 3)
1286*44704f69SBart Van Assche                             pr2serr_lk("  id=%d: start IO at lba=0x%" PRIx64
1287*44704f69SBart Van Assche                                        "\n", id, lba);
1288*44704f69SBart Van Assche                     }
1289*44704f69SBart Van Assche                 } else
1290*44704f69SBart Van Assche                     lba = op->lba;
1291*44704f69SBart Van Assche             } else
1292*44704f69SBart Van Assche                 lba = 0;
1293*44704f69SBart Van Assche             if (vb > 4)
1294*44704f69SBart Van Assche                 pr2serr_lk("t_id=%d: starting pack_id=%d\n", id, pack_id);
1295*44704f69SBart Van Assche             res = (op->v4) ?
1296*44704f69SBart Van Assche                 start_sg4_cmd(sg_fd, op->c2e, pack_id, lba, lbp,
1297*44704f69SBart Van Assche                               blk_sz * op->num_lbs, sg_flags, op->submit,
1298*44704f69SBart Van Assche                               thr_enomem_count, thr_start_eagain_count,
1299*44704f69SBart Van Assche                               thr_start_ebusy_count, thr_start_e2big_count,
1300*44704f69SBart Van Assche                               thr_start_edom_count)  :
1301*44704f69SBart Van Assche                 start_sg3_cmd(sg_fd, op->c2e, pack_id, lba, lbp,
1302*44704f69SBart Van Assche                               blk_sz * op->num_lbs, sg_flags, op->submit,
1303*44704f69SBart Van Assche                               thr_enomem_count, thr_start_eagain_count,
1304*44704f69SBart Van Assche                               thr_start_ebusy_count, thr_start_e2big_count,
1305*44704f69SBart Van Assche                               thr_start_edom_count);
1306*44704f69SBart Van Assche             if (res) {
1307*44704f69SBart Van Assche                 if (res > 1) { /* here if E2BIG, start not done, try finish */
1308*44704f69SBart Van Assche                     do_inc = 0;
1309*44704f69SBart Van Assche                     need_finish = true;
1310*44704f69SBart Van Assche                     encore_pack_id = pack_id;
1311*44704f69SBart Van Assche                     pack_id = prev_pack_id;
1312*44704f69SBart Van Assche                     encore_lbps = make_pair(lbp, free_lbp);
1313*44704f69SBart Van Assche                     if (vb > 2)
1314*44704f69SBart Van Assche                         pr2serr_lk("t_id=%d: E2BIG hit, prev_pack_id=%d, "
1315*44704f69SBart Van Assche                                    "encore_pack_id=%d\n", id, prev_pack_id,
1316*44704f69SBart Van Assche                                    encore_pack_id);
1317*44704f69SBart Van Assche                 } else {
1318*44704f69SBart Van Assche                     err = "start_sg3_cmd()";
1319*44704f69SBart Van Assche                     break;
1320*44704f69SBart Van Assche                 }
1321*44704f69SBart Van Assche             } else {    /* no error */
1322*44704f69SBart Van Assche                 ++thr_async_starts;
1323*44704f69SBart Van Assche                 ++num_outstanding;
1324*44704f69SBart Van Assche                 pi2buff[pack_id] = make_pair(lbp, free_lbp);
1325*44704f69SBart Van Assche                 if (ruip)
1326*44704f69SBart Van Assche                     pi_2_lba[pack_id] = lba;
1327*44704f69SBart Van Assche             }
1328*44704f69SBart Van Assche             if (vb && !once && (pi2buff.size() > 1000)) {
1329*44704f69SBart Van Assche                 once = true;
1330*44704f69SBart Van Assche                 pr2serr_lk("%d->id: pi2buff.size() over 1000 (b)\n", id);
1331*44704f69SBart Van Assche             }
1332*44704f69SBart Van Assche             if (vb && !once_2000 && free_lst.size() > 2000) {
1333*44704f69SBart Van Assche                 once_2000 = true;
1334*44704f69SBart Van Assche                 pr2serr_lk("%d->id: free_lst.size() over 2000 (b)\n", id);
1335*44704f69SBart Van Assche             }
1336*44704f69SBart Van Assche             if (vb && !once_6000 && free_lst.size() > 6000) {
1337*44704f69SBart Van Assche                 once_2000 = true;
1338*44704f69SBart Van Assche                 pr2serr_lk("%d->id: free_lst.size() over 6000 (b)\n", id);
1339*44704f69SBart Van Assche             }
1340*44704f69SBart Van Assche         }
1341*44704f69SBart Van Assche         if (need_finish) {
1342*44704f69SBart Van Assche             num_waiting_read = 0;
1343*44704f69SBart Van Assche             if (ioctl(sg_fd, SG_GET_NUM_WAITING, &num_waiting_read) < 0) {
1344*44704f69SBart Van Assche                 err = "ioctl(SG_GET_NUM_WAITING) failed";
1345*44704f69SBart Van Assche                 break;
1346*44704f69SBart Van Assche             } else if (vb > 3)
1347*44704f69SBart Van Assche                 pr2serr_lk("t_id=%d: num_waiting_read=%d\n", id,
1348*44704f69SBart Van Assche                            num_waiting_read);
1349*44704f69SBart Van Assche             if (num_waiting_read > 0)
1350*44704f69SBart Van Assche                 num_to_read = num_waiting_read;
1351*44704f69SBart Van Assche             else {
1352*44704f69SBart Van Assche                 struct timespec tspec = {0, 100000 /* 100 usecs */};
1353*44704f69SBart Van Assche 
1354*44704f69SBart Van Assche                 nanosleep(&tspec, NULL);
1355*44704f69SBart Van Assche                 if (vb > 3)
1356*44704f69SBart Van Assche                     pr2serr_lk("t_id=%d: E2BIG, 100 usecs sleep\n", id);
1357*44704f69SBart Van Assche                 // err = "strange, E2BIG but nothing to read";
1358*44704f69SBart Van Assche                 // break;
1359*44704f69SBart Van Assche             }
1360*44704f69SBart Van Assche         } else if ((num_outstanding >= op->maxq_per_thread) || (k >= npt)) {
1361*44704f69SBart Van Assche             /* full queue or finished injecting */
1362*44704f69SBart Van Assche             num_waiting_read = 0;
1363*44704f69SBart Van Assche             if (ioctl(sg_fd, SG_GET_NUM_WAITING, &num_waiting_read) < 0) {
1364*44704f69SBart Van Assche                 err = "ioctl(SG_GET_NUM_WAITING) failed";
1365*44704f69SBart Van Assche                 break;
1366*44704f69SBart Van Assche             }
1367*44704f69SBart Van Assche             if (1 == num_waiting_read)
1368*44704f69SBart Van Assche                 num_to_read = num_waiting_read;
1369*44704f69SBart Van Assche             else if (num_waiting_read > 0) {
1370*44704f69SBart Van Assche                 if (k >= npt)
1371*44704f69SBart Van Assche                     num_to_read = num_waiting_read;
1372*44704f69SBart Van Assche                 else {
1373*44704f69SBart Van Assche                     switch (op->myqd) {
1374*44704f69SBart Van Assche                     case MYQD_LOW:
1375*44704f69SBart Van Assche                         num_to_read = num_waiting_read;
1376*44704f69SBart Van Assche                         break;
1377*44704f69SBart Van Assche                     case MYQD_MEDIUM:
1378*44704f69SBart Van Assche                         num_to_read = num_waiting_read / 2;
1379*44704f69SBart Van Assche                         break;
1380*44704f69SBart Van Assche                     case MYQD_HIGH:
1381*44704f69SBart Van Assche                     default:
1382*44704f69SBart Van Assche                         if (op->ovn > 0) {
1383*44704f69SBart Van Assche                             if (op->sg_vn_ge_40030) {
1384*44704f69SBart Van Assche                                 int num_subm = num_submitted(sg_fd);
1385*44704f69SBart Van Assche 
1386*44704f69SBart Van Assche                                 if (num_subm > op->ovn) {
1387*44704f69SBart Van Assche                                     num_to_read = num_waiting_read > 0 ?
1388*44704f69SBart Van Assche                                                     num_waiting_read : 1;
1389*44704f69SBart Van Assche                                     break;
1390*44704f69SBart Van Assche                                 }
1391*44704f69SBart Van Assche                             } else {
1392*44704f69SBart Van Assche                                 if (num_waiting_read > (op->ovn / 2)) {
1393*44704f69SBart Van Assche                                     num_to_read = num_waiting_read / 2;
1394*44704f69SBart Van Assche                                     break;
1395*44704f69SBart Van Assche                                 }
1396*44704f69SBart Van Assche                             }
1397*44704f69SBart Van Assche                         }
1398*44704f69SBart Van Assche                         num_to_read = 1;
1399*44704f69SBart Van Assche                         break;
1400*44704f69SBart Van Assche                     }
1401*44704f69SBart Van Assche                 }
1402*44704f69SBart Van Assche             } else {    /* nothing waiting to be read */
1403*44704f69SBart Van Assche                 if (op->sg_vn_ge_40030) {
1404*44704f69SBart Van Assche                     int val = num_submitted(sg_fd);
1405*44704f69SBart Van Assche 
1406*44704f69SBart Van Assche                     if (0 == val) {
1407*44704f69SBart Van Assche                         err = "nothing submitted now ??";
1408*44704f69SBart Van Assche                         break;
1409*44704f69SBart Van Assche                     } else if (val < 0) {
1410*44704f69SBart Van Assche                         err = "num_submitted failed";
1411*44704f69SBart Van Assche                         break;
1412*44704f69SBart Van Assche                     }
1413*44704f69SBart Van Assche                 }
1414*44704f69SBart Van Assche                 n = (op->wait_ms > 0) ? op->wait_ms : 0;
1415*44704f69SBart Van Assche                 if (n > 0) {
1416*44704f69SBart Van Assche                     for (j = 0; (j < 1000000) &&
1417*44704f69SBart Van Assche                          (0 == (res = poll(pfd, 1, n)));
1418*44704f69SBart Van Assche                          ++j)
1419*44704f69SBart Van Assche                         ;
1420*44704f69SBart Van Assche                     if (j >= 1000000) {
1421*44704f69SBart Van Assche                         err = "poll() looped 1 million times";
1422*44704f69SBart Van Assche                         break;
1423*44704f69SBart Van Assche                     }
1424*44704f69SBart Van Assche                     if (res < 0) {
1425*44704f69SBart Van Assche                         err = "poll(wait_ms) failed";
1426*44704f69SBart Van Assche                         break;
1427*44704f69SBart Van Assche                     }
1428*44704f69SBart Van Assche                 } else {
1429*44704f69SBart Van Assche                     struct timespec ts;
1430*44704f69SBart Van Assche 
1431*44704f69SBart Van Assche                     ts.tv_sec = 0;
1432*44704f69SBart Van Assche                     ts.tv_nsec = DEF_NANOSEC_WAIT;
1433*44704f69SBart Van Assche                     if (nanosleep(&ts, NULL) < 0) {
1434*44704f69SBart Van Assche                         err = "nanosleep() failed";
1435*44704f69SBart Van Assche                         break;
1436*44704f69SBart Van Assche                     }
1437*44704f69SBart Van Assche                 }
1438*44704f69SBart Van Assche             }
1439*44704f69SBart Van Assche         } else {        /* not full, not finished injecting */
1440*44704f69SBart Van Assche             if (MYQD_HIGH == op->myqd) {
1441*44704f69SBart Van Assche                 num_to_read = 0;
1442*44704f69SBart Van Assche                 if (op->ovn) {
1443*44704f69SBart Van Assche                     if (op->sg_vn_ge_40030) {
1444*44704f69SBart Van Assche                         int num_subm = num_submitted(sg_fd);
1445*44704f69SBart Van Assche 
1446*44704f69SBart Van Assche                         if (num_subm > op->ovn)
1447*44704f69SBart Van Assche                             num_to_read = num_waiting_read > 0 ?
1448*44704f69SBart Van Assche                                             num_waiting_read : 1;
1449*44704f69SBart Van Assche                     } else {
1450*44704f69SBart Van Assche                         num_waiting_read = 0;
1451*44704f69SBart Van Assche                         if (ioctl(sg_fd, SG_GET_NUM_WAITING,
1452*44704f69SBart Van Assche                                   &num_waiting_read) < 0) {
1453*44704f69SBart Van Assche                             err = "ioctl(SG_GET_NUM_WAITING) failed";
1454*44704f69SBart Van Assche                             break;
1455*44704f69SBart Van Assche                         }
1456*44704f69SBart Van Assche                         if (num_waiting_read > (op->ovn / 2))
1457*44704f69SBart Van Assche                             num_to_read = num_waiting_read / 2;
1458*44704f69SBart Van Assche                     }
1459*44704f69SBart Van Assche                 }
1460*44704f69SBart Van Assche             } else {
1461*44704f69SBart Van Assche                 num_waiting_read = 0;
1462*44704f69SBart Van Assche                 if (ioctl(sg_fd, SG_GET_NUM_WAITING, &num_waiting_read) < 0) {
1463*44704f69SBart Van Assche                     err = "ioctl(SG_GET_NUM_WAITING) failed";
1464*44704f69SBart Van Assche                     break;
1465*44704f69SBart Van Assche                 }
1466*44704f69SBart Van Assche                 if (num_waiting_read > 0)
1467*44704f69SBart Van Assche                     num_to_read = num_waiting_read /
1468*44704f69SBart Van Assche                                   ((MYQD_LOW == op->myqd) ? 1 : 2);
1469*44704f69SBart Van Assche                 else
1470*44704f69SBart Van Assche                     num_to_read = 0;
1471*44704f69SBart Van Assche             }
1472*44704f69SBart Van Assche         }
1473*44704f69SBart Van Assche 
1474*44704f69SBart Van Assche         if (vb && !once_4000 && (num_to_read > 4000)) {
1475*44704f69SBart Van Assche             once_4000 = true;
1476*44704f69SBart Van Assche             pr2serr_lk("%d->id: num_to_read=%d\n", id, num_to_read);
1477*44704f69SBart Van Assche         }
1478*44704f69SBart Van Assche         while (num_to_read > 0) {
1479*44704f69SBart Van Assche             --num_to_read;
1480*44704f69SBart Van Assche             if (op->pack_id_force) {
1481*44704f69SBart Van Assche                 j = pi2buff.size();
1482*44704f69SBart Van Assche                 if (j > 0)
1483*44704f69SBart Van Assche                     pack_id = pi2buff.begin()->first;
1484*44704f69SBart Van Assche                 else
1485*44704f69SBart Van Assche                     pack_id = -1;
1486*44704f69SBart Van Assche             } else
1487*44704f69SBart Van Assche                 pack_id = -1;
1488*44704f69SBart Van Assche             ask = pack_id;
1489*44704f69SBart Van Assche             res = (op->v4) ?
1490*44704f69SBart Van Assche                     finish_sg4_cmd(sg_fd, op->c2e, pack_id, op->submit,
1491*44704f69SBart Van Assche                                    op->wait_ms, thr_enomem_count,
1492*44704f69SBart Van Assche                                    thr_fin_eagain_count, thr_fin_ebusy_count,
1493*44704f69SBart Van Assche                                    nanosecs)           :
1494*44704f69SBart Van Assche                     finish_sg3_cmd(sg_fd, op->c2e, pack_id, op->submit,
1495*44704f69SBart Van Assche                                    op->wait_ms, thr_enomem_count,
1496*44704f69SBart Van Assche                                    thr_fin_eagain_count, thr_fin_ebusy_count,
1497*44704f69SBart Van Assche                                    nanosecs);
1498*44704f69SBart Van Assche             if (res) {
1499*44704f69SBart Van Assche                 err = "finish_sg3_cmd()";
1500*44704f69SBart Van Assche                 if (ruip && (pack_id > 0)) {
1501*44704f69SBart Van Assche                     auto q = pi_2_lba.find(pack_id);
1502*44704f69SBart Van Assche 
1503*44704f69SBart Van Assche                     if (q != pi_2_lba.end()) {
1504*44704f69SBart Van Assche                         snprintf(ebuff, sizeof(ebuff), "%s: lba=0x%" PRIx64 ,
1505*44704f69SBart Van Assche                                  err, q->second);
1506*44704f69SBart Van Assche                         err = ebuff;
1507*44704f69SBart Van Assche                     }
1508*44704f69SBart Van Assche                 }
1509*44704f69SBart Van Assche                 break;
1510*44704f69SBart Van Assche             }
1511*44704f69SBart Van Assche             if (op->cmd_time && op->sg_vn_ge_40030)
1512*44704f69SBart Van Assche                 sum_nanosecs += nanosecs;
1513*44704f69SBart Van Assche             ++thr_async_finishes;
1514*44704f69SBart Van Assche             --num_outstanding;
1515*44704f69SBart Van Assche             if (vb > 4)
1516*44704f69SBart Van Assche                 pr2serr_lk("t_id=%d: finishing pack_id ask=%d, got=%d, "
1517*44704f69SBart Van Assche                            "outstanding=%d\n", id, ask, pack_id,
1518*44704f69SBart Van Assche                            num_outstanding);
1519*44704f69SBart Van Assche             auto p = pi2buff.find(pack_id);
1520*44704f69SBart Van Assche 
1521*44704f69SBart Van Assche             if (p == pi2buff.end()) {
1522*44704f69SBart Van Assche                 snprintf(ebuff, sizeof(ebuff), "pack_id=%d from "
1523*44704f69SBart Van Assche                          "finish_sg3_cmd() not found\n", pack_id);
1524*44704f69SBart Van Assche                 if (! err)
1525*44704f69SBart Van Assche                     err = ebuff;
1526*44704f69SBart Van Assche             } else {
1527*44704f69SBart Van Assche                 lbp = p->second.first;
1528*44704f69SBart Van Assche                 free_lbp = p->second.second;
1529*44704f69SBart Van Assche                 pi2buff.erase(p);
1530*44704f69SBart Van Assche                 if (lbp)
1531*44704f69SBart Van Assche                     free_lst.push_front(make_pair(lbp, free_lbp));
1532*44704f69SBart Van Assche             }
1533*44704f69SBart Van Assche             if (ruip && (pack_id > 0)) {
1534*44704f69SBart Van Assche                 auto q = pi_2_lba.find(pack_id);
1535*44704f69SBart Van Assche 
1536*44704f69SBart Van Assche                 if (q != pi_2_lba.end()) {
1537*44704f69SBart Van Assche                     if (vb > 3)
1538*44704f69SBart Van Assche                         pr2serr_lk("    id=%d: finish IO at lba=0x%" PRIx64
1539*44704f69SBart Van Assche                                    "\n", id, q->second);
1540*44704f69SBart Van Assche                     pi_2_lba.erase(q);
1541*44704f69SBart Van Assche                 }
1542*44704f69SBart Van Assche             }
1543*44704f69SBart Van Assche             if (err)
1544*44704f69SBart Van Assche                 break;
1545*44704f69SBart Van Assche         }       /* end of while loop counting down num_to_read */
1546*44704f69SBart Van Assche         if (err)
1547*44704f69SBart Van Assche             break;
1548*44704f69SBart Van Assche     }           /* end of for loop over npt (number per thread) */
1549*44704f69SBart Van Assche     if (vb)
1550*44704f69SBart Van Assche         pr2serr_lk("%d->id: leaving main thread loop; k=%d, o=%d\n", id, k,
1551*44704f69SBart Van Assche                    o);
1552*44704f69SBart Van Assche     close(sg_fd);       // sg driver will handle any commands "in flight"
1553*44704f69SBart Van Assche     if (ruip)
1554*44704f69SBart Van Assche         delete ruip;
1555*44704f69SBart Van Assche 
1556*44704f69SBart Van Assche     if (err || (k < npt)) {
1557*44704f69SBart Van Assche         if (k < npt)
1558*44704f69SBart Van Assche             pr2serr_lk("t_id=%d FAILed at iteration %d%s%s\n", id, k,
1559*44704f69SBart Van Assche                        (err ? ", Reason: " : ""), (err ? err : ""));
1560*44704f69SBart Van Assche         else
1561*44704f69SBart Van Assche             pr2serr_lk("t_id=%d FAILed on last%s%s\n", id,
1562*44704f69SBart Van Assche                        (err ? ", Reason: " : ""), (err ? err : ""));
1563*44704f69SBart Van Assche     }
1564*44704f69SBart Van Assche     n = pi2buff.size();
1565*44704f69SBart Van Assche     if (n > 0)
1566*44704f69SBart Van Assche         pr2serr_lk("t_id=%d Still %d elements in pi2buff map on "
1567*44704f69SBart Van Assche                    "exit\n", id, n);
1568*44704f69SBart Van Assche     for (k = 0; ! free_lst.empty(); ++k) {
1569*44704f69SBart Van Assche         lbp = free_lst.back().first;
1570*44704f69SBart Van Assche         free_lbp = free_lst.back().second;
1571*44704f69SBart Van Assche         free_lst.back().second = NULL;
1572*44704f69SBart Van Assche         free_lst.pop_back();
1573*44704f69SBart Van Assche         if (vb > 6)
1574*44704f69SBart Van Assche             pr2serr_lk("t_id=%d freeing %p (free_ %p)\n", id, lbp, free_lbp);
1575*44704f69SBart Van Assche         if (free_lbp) {
1576*44704f69SBart Van Assche             free(free_lbp);
1577*44704f69SBart Van Assche             free_lbp = NULL;
1578*44704f69SBart Van Assche         }
1579*44704f69SBart Van Assche     }
1580*44704f69SBart Van Assche     if ((vb > 2) && (k > 0))
1581*44704f69SBart Van Assche         pr2serr_lk("%d->id: Maximum number of READ/WRITEs queued: %d\n",
1582*44704f69SBart Van Assche                    id, k);
1583*44704f69SBart Van Assche     async_starts += thr_async_starts;
1584*44704f69SBart Van Assche     async_finishes += thr_async_finishes;
1585*44704f69SBart Van Assche     start_eagain_count += thr_start_eagain_count;
1586*44704f69SBart Van Assche     start_ebusy_count += thr_start_ebusy_count;
1587*44704f69SBart Van Assche     start_e2big_count += thr_start_e2big_count;
1588*44704f69SBart Van Assche     fin_eagain_count += thr_fin_eagain_count;
1589*44704f69SBart Van Assche     fin_ebusy_count += thr_fin_ebusy_count;
1590*44704f69SBart Van Assche     enomem_count += thr_enomem_count;
1591*44704f69SBart Van Assche     start_edom_count += thr_start_edom_count;
1592*44704f69SBart Van Assche     if (op->cmd_time && op->sg_vn_ge_40030 && (npt > 0)) {
1593*44704f69SBart Van Assche         pr2serr_lk("t_id=%d average nanosecs per cmd: %" PRId64
1594*44704f69SBart Van Assche                    "\n", id, sum_nanosecs / npt);
1595*44704f69SBart Van Assche     }
1596*44704f69SBart Van Assche }
1597*44704f69SBart Van Assche 
1598*44704f69SBart Van Assche #define INQ_REPLY_LEN 96
1599*44704f69SBart Van Assche #define INQ_CMD_LEN 6
1600*44704f69SBart Van Assche 
1601*44704f69SBart Van Assche /* Send INQUIRY and fetches response. If okay puts PRODUCT ID field
1602*44704f69SBart Van Assche  * in b (up to m_blen bytes). Does not use O_EXCL flag. Returns 0 on success,
1603*44704f69SBart Van Assche  * else -1 . */
1604*44704f69SBart Van Assche static int
do_inquiry_prod_id(const char * dev_name,int block,int & sg_ver_num,char * b,int b_mlen)1605*44704f69SBart Van Assche do_inquiry_prod_id(const char * dev_name, int block, int & sg_ver_num,
1606*44704f69SBart Van Assche                    char * b, int b_mlen)
1607*44704f69SBart Van Assche {
1608*44704f69SBart Van Assche     int sg_fd, ok, ret;
1609*44704f69SBart Van Assche     struct sg_io_hdr pt;
1610*44704f69SBart Van Assche     uint8_t inqCmdBlk [INQ_CMD_LEN] =
1611*44704f69SBart Van Assche                                 {0x12, 0, 0, 0, INQ_REPLY_LEN, 0};
1612*44704f69SBart Van Assche     uint8_t inqBuff[INQ_REPLY_LEN];
1613*44704f69SBart Van Assche     uint8_t sense_buffer[64] SG_C_CPP_ZERO_INIT;
1614*44704f69SBart Van Assche     int open_flags = O_RDWR;    /* O_EXCL | O_RDONLY fails with EPERM */
1615*44704f69SBart Van Assche 
1616*44704f69SBart Van Assche     if (! block)
1617*44704f69SBart Van Assche         open_flags |= O_NONBLOCK;
1618*44704f69SBart Van Assche     sg_fd = open(dev_name, open_flags);
1619*44704f69SBart Van Assche     if (sg_fd < 0) {
1620*44704f69SBart Van Assche         pr_errno_lk(errno, "%s: error opening file: %s", __func__, dev_name);
1621*44704f69SBart Van Assche         return -1;
1622*44704f69SBart Van Assche     }
1623*44704f69SBart Van Assche     if (ioctl(sg_fd, SG_GET_VERSION_NUM, &sg_ver_num) < 0)
1624*44704f69SBart Van Assche         sg_ver_num = 0;
1625*44704f69SBart Van Assche     /* Prepare INQUIRY command */
1626*44704f69SBart Van Assche     memset(&pt, 0, sizeof(pt));
1627*44704f69SBart Van Assche     pt.interface_id = 'S';
1628*44704f69SBart Van Assche     pt.cmd_len = sizeof(inqCmdBlk);
1629*44704f69SBart Van Assche     /* pt.iovec_count = 0; */  /* memset takes care of this */
1630*44704f69SBart Van Assche     pt.mx_sb_len = sizeof(sense_buffer);
1631*44704f69SBart Van Assche     pt.dxfer_direction = SG_DXFER_FROM_DEV;
1632*44704f69SBart Van Assche     pt.dxfer_len = INQ_REPLY_LEN;
1633*44704f69SBart Van Assche     pt.dxferp = inqBuff;
1634*44704f69SBart Van Assche     pt.cmdp = inqCmdBlk;
1635*44704f69SBart Van Assche     pt.sbp = sense_buffer;
1636*44704f69SBart Van Assche     pt.timeout = 20000;     /* 20000 millisecs == 20 seconds */
1637*44704f69SBart Van Assche     /* pt.flags = 0; */     /* take defaults: indirect IO, etc */
1638*44704f69SBart Van Assche     /* pt.pack_id = 0; */
1639*44704f69SBart Van Assche     /* pt.usr_ptr = NULL; */
1640*44704f69SBart Van Assche 
1641*44704f69SBart Van Assche     if (ioctl(sg_fd, SG_IO, &pt) < 0) {
1642*44704f69SBart Van Assche         pr_errno_lk(errno, "%s: Inquiry SG_IO ioctl error", __func__);
1643*44704f69SBart Van Assche         close(sg_fd);
1644*44704f69SBart Van Assche         return -1;
1645*44704f69SBart Van Assche     }
1646*44704f69SBart Van Assche 
1647*44704f69SBart Van Assche     /* now for the error processing */
1648*44704f69SBart Van Assche     ok = 0;
1649*44704f69SBart Van Assche     switch (sg_err_category3(&pt)) {
1650*44704f69SBart Van Assche     case SG_LIB_CAT_CLEAN:
1651*44704f69SBart Van Assche         ok = 1;
1652*44704f69SBart Van Assche         break;
1653*44704f69SBart Van Assche     case SG_LIB_CAT_RECOVERED:
1654*44704f69SBart Van Assche         pr2serr_lk("Recovered error on INQUIRY, continuing\n");
1655*44704f69SBart Van Assche         ok = 1;
1656*44704f69SBart Van Assche         break;
1657*44704f69SBart Van Assche     default: /* won't bother decoding other categories */
1658*44704f69SBart Van Assche         {
1659*44704f69SBart Van Assche             lock_guard<mutex> lg(console_mutex);
1660*44704f69SBart Van Assche             sg_chk_n_print3("INQUIRY command error", &pt, 1);
1661*44704f69SBart Van Assche         }
1662*44704f69SBart Van Assche         break;
1663*44704f69SBart Van Assche     }
1664*44704f69SBart Van Assche     if (ok) {
1665*44704f69SBart Van Assche         /* Good, so fetch Product ID from response, copy to 'b' */
1666*44704f69SBart Van Assche         if (b_mlen > 0) {
1667*44704f69SBart Van Assche             if (b_mlen > 16) {
1668*44704f69SBart Van Assche                 memcpy(b, inqBuff + 16, 16);
1669*44704f69SBart Van Assche                 b[16] = '\0';
1670*44704f69SBart Van Assche             } else {
1671*44704f69SBart Van Assche                 memcpy(b, inqBuff + 16, b_mlen - 1);
1672*44704f69SBart Van Assche                 b[b_mlen - 1] = '\0';
1673*44704f69SBart Van Assche             }
1674*44704f69SBart Van Assche         }
1675*44704f69SBart Van Assche         ret = 0;
1676*44704f69SBart Van Assche     } else
1677*44704f69SBart Van Assche         ret = -1;
1678*44704f69SBart Van Assche 
1679*44704f69SBart Van Assche     close(sg_fd);
1680*44704f69SBart Van Assche     return ret;
1681*44704f69SBart Van Assche }
1682*44704f69SBart Van Assche 
1683*44704f69SBart Van Assche /* Only allow ranges up to 2**32-1 upper limit, so READ CAPACITY(10)
1684*44704f69SBart Van Assche  * sufficient. Return of 0 -> success, -1 -> failure, 2 -> try again */
1685*44704f69SBart Van Assche static int
do_read_capacity(const char * dev_name,int block,unsigned int * last_lba,unsigned int * blk_sz)1686*44704f69SBart Van Assche do_read_capacity(const char * dev_name, int block, unsigned int * last_lba,
1687*44704f69SBart Van Assche                  unsigned int * blk_sz)
1688*44704f69SBart Van Assche {
1689*44704f69SBart Van Assche     int res, sg_fd;
1690*44704f69SBart Van Assche     uint8_t rcCmdBlk [10] = {0x25, 0, 0, 0, 0, 0, 0, 0, 0, 0};
1691*44704f69SBart Van Assche     uint8_t rcBuff[64];
1692*44704f69SBart Van Assche     uint8_t sense_b[64] SG_C_CPP_ZERO_INIT;
1693*44704f69SBart Van Assche     sg_io_hdr_t io_hdr SG_C_CPP_ZERO_INIT;
1694*44704f69SBart Van Assche     int open_flags = O_RDWR;    /* O_EXCL | O_RDONLY fails with EPERM */
1695*44704f69SBart Van Assche 
1696*44704f69SBart Van Assche     if (! block)
1697*44704f69SBart Van Assche         open_flags |= O_NONBLOCK;
1698*44704f69SBart Van Assche     sg_fd = open(dev_name, open_flags);
1699*44704f69SBart Van Assche     if (sg_fd < 0) {
1700*44704f69SBart Van Assche         pr_errno_lk(errno, "%s: error opening file: %s", __func__, dev_name);
1701*44704f69SBart Van Assche         return -1;
1702*44704f69SBart Van Assche     }
1703*44704f69SBart Van Assche     /* Prepare READ CAPACITY(10) command */
1704*44704f69SBart Van Assche     io_hdr.interface_id = 'S';
1705*44704f69SBart Van Assche     io_hdr.cmd_len = sizeof(rcCmdBlk);
1706*44704f69SBart Van Assche     io_hdr.mx_sb_len = sizeof(sense_b);
1707*44704f69SBart Van Assche     io_hdr.dxfer_direction = SG_DXFER_FROM_DEV;
1708*44704f69SBart Van Assche     io_hdr.dxfer_len = sizeof(rcBuff);
1709*44704f69SBart Van Assche     io_hdr.dxferp = rcBuff;
1710*44704f69SBart Van Assche     io_hdr.cmdp = rcCmdBlk;
1711*44704f69SBart Van Assche     io_hdr.sbp = sense_b;
1712*44704f69SBart Van Assche     io_hdr.timeout = 20000;     /* 20000 millisecs == 20 seconds */;
1713*44704f69SBart Van Assche 
1714*44704f69SBart Van Assche     if (ioctl(sg_fd, SG_IO, &io_hdr) < 0) {
1715*44704f69SBart Van Assche         pr_errno_lk(errno, "%s (SG_IO) error", __func__);
1716*44704f69SBart Van Assche         close(sg_fd);
1717*44704f69SBart Van Assche         return -1;
1718*44704f69SBart Van Assche     }
1719*44704f69SBart Van Assche     res = sg_err_category3(&io_hdr);
1720*44704f69SBart Van Assche     if (SG_LIB_CAT_UNIT_ATTENTION == res) {
1721*44704f69SBart Van Assche         lock_guard<mutex> lg(console_mutex);
1722*44704f69SBart Van Assche         sg_chk_n_print3("read capacity", &io_hdr, 1);
1723*44704f69SBart Van Assche         close(sg_fd);
1724*44704f69SBart Van Assche         return 2; /* probably have another go ... */
1725*44704f69SBart Van Assche     } else if (SG_LIB_CAT_CLEAN != res) {
1726*44704f69SBart Van Assche         lock_guard<mutex> lg(console_mutex);
1727*44704f69SBart Van Assche         sg_chk_n_print3("read capacity", &io_hdr, 1);
1728*44704f69SBart Van Assche         close(sg_fd);
1729*44704f69SBart Van Assche         return -1;
1730*44704f69SBart Van Assche     }
1731*44704f69SBart Van Assche     *last_lba = sg_get_unaligned_be32(&rcBuff[0]);
1732*44704f69SBart Van Assche     *blk_sz = sg_get_unaligned_be32(&rcBuff[4]);
1733*44704f69SBart Van Assche     close(sg_fd);
1734*44704f69SBart Van Assche     return 0;
1735*44704f69SBart Van Assche }
1736*44704f69SBart Van Assche 
1737*44704f69SBart Van Assche 
1738*44704f69SBart Van Assche int
main(int argc,char * argv[])1739*44704f69SBart Van Assche main(int argc, char * argv[])
1740*44704f69SBart Van Assche {
1741*44704f69SBart Van Assche     bool maxq_per_thread_given = false;
1742*44704f69SBart Van Assche     int n;
1743*44704f69SBart Van Assche     int force = 0;
1744*44704f69SBart Van Assche     int64_t ll;
1745*44704f69SBart Van Assche     int num_threads = DEF_NUM_THREADS;
1746*44704f69SBart Van Assche     struct timespec start_tm, end_tm;
1747*44704f69SBart Van Assche     struct opts_t * op;
1748*44704f69SBart Van Assche     const char * cp;
1749*44704f69SBart Van Assche 
1750*44704f69SBart Van Assche     op = &a_opts;
1751*44704f69SBart Van Assche #if 0
1752*44704f69SBart Van Assche     memset(op, 0, sizeof(*op));         // C++ doesn't like this
1753*44704f69SBart Van Assche #endif
1754*44704f69SBart Van Assche     op->direct = DEF_DIRECT;
1755*44704f69SBart Van Assche     op->lba = DEF_LBA;
1756*44704f69SBart Van Assche     op->hi_lba = 0;
1757*44704f69SBart Van Assche     op->lb_sz = DEF_LB_SZ;
1758*44704f69SBart Van Assche     op->maxq_per_thread = MAX_Q_PER_FD;
1759*44704f69SBart Van Assche     op->mmap_io = DEF_MMAP_IO;
1760*44704f69SBart Van Assche     op->num_per_thread = DEF_NUM_PER_THREAD;
1761*44704f69SBart Van Assche     op->num_lbs = 1;
1762*44704f69SBart Van Assche     op->no_xfer = !! DEF_NO_XFER;
1763*44704f69SBart Van Assche     op->verbose = 0;
1764*44704f69SBart Van Assche     op->wait_ms = DEF_WAIT_MS;
1765*44704f69SBart Van Assche     op->c2e = SCSI_TUR;
1766*44704f69SBart Van Assche     op->blqd = BLQ_DEFAULT;
1767*44704f69SBart Van Assche     op->block = !! DEF_BLOCKING;
1768*44704f69SBart Van Assche     op->myqd = MYQD_HIGH;
1769*44704f69SBart Van Assche     page_size = sysconf(_SC_PAGESIZE);
1770*44704f69SBart Van Assche 
1771*44704f69SBart Van Assche     while (1) {
1772*44704f69SBart Van Assche         int option_index = 0;
1773*44704f69SBart Van Assche         int c;
1774*44704f69SBart Van Assche 
1775*44704f69SBart Van Assche         c = getopt_long(argc, argv,
1776*44704f69SBart Van Assche                         "34acdefghl:L:mM:n:NO:pq:Q:Rs:St:TuvVw:W",
1777*44704f69SBart Van Assche                         long_options, &option_index);
1778*44704f69SBart Van Assche         if (c == -1)
1779*44704f69SBart Van Assche             break;
1780*44704f69SBart Van Assche 
1781*44704f69SBart Van Assche         switch (c) {
1782*44704f69SBart Van Assche         case '3':
1783*44704f69SBart Van Assche             op->v3 = true;
1784*44704f69SBart Van Assche             op->v3_given = true;
1785*44704f69SBart Van Assche             op->v4 = false;     /* if '-4 -3' take latter */
1786*44704f69SBart Van Assche             op->v4_given = false;
1787*44704f69SBart Van Assche             break;
1788*44704f69SBart Van Assche         case '4':
1789*44704f69SBart Van Assche             op->v4 = true;
1790*44704f69SBart Van Assche             op->v4_given = true;
1791*44704f69SBart Van Assche             op->v3 = false;
1792*44704f69SBart Van Assche             op->v3_given = false;
1793*44704f69SBart Van Assche             break;
1794*44704f69SBart Van Assche         case 'a':
1795*44704f69SBart Van Assche             op->masync = true;
1796*44704f69SBart Van Assche             break;
1797*44704f69SBart Van Assche         case 'c':
1798*44704f69SBart Van Assche             op->cmd_time = true;
1799*44704f69SBart Van Assche             break;
1800*44704f69SBart Van Assche         case 'd':
1801*44704f69SBart Van Assche             op->direct = true;
1802*44704f69SBart Van Assche             break;
1803*44704f69SBart Van Assche         case 'e':
1804*44704f69SBart Van Assche             op->excl = true;
1805*44704f69SBart Van Assche             break;
1806*44704f69SBart Van Assche         case 'f':
1807*44704f69SBart Van Assche             force = true;
1808*44704f69SBart Van Assche             break;
1809*44704f69SBart Van Assche         case 'g':
1810*44704f69SBart Van Assche             op->generic_sync = true;
1811*44704f69SBart Van Assche             break;
1812*44704f69SBart Van Assche         case 'h':
1813*44704f69SBart Van Assche         case '?':
1814*44704f69SBart Van Assche             usage();
1815*44704f69SBart Van Assche             return 0;
1816*44704f69SBart Van Assche         case 'l':
1817*44704f69SBart Van Assche             if (isdigit(*optarg)) {
1818*44704f69SBart Van Assche                 ll = sg_get_llnum(optarg);
1819*44704f69SBart Van Assche                 if (-1 == ll) {
1820*44704f69SBart Van Assche                     pr2serr_lk("could not decode lba\n");
1821*44704f69SBart Van Assche                     return 1;
1822*44704f69SBart Van Assche                 } else
1823*44704f69SBart Van Assche                     op->lba = (uint64_t)ll;
1824*44704f69SBart Van Assche                 cp = strchr(optarg, ',');
1825*44704f69SBart Van Assche                 if (cp) {
1826*44704f69SBart Van Assche                     if (0 == strcmp("-1", cp + 1))
1827*44704f69SBart Van Assche                         op->hi_lba = UINT_MAX;
1828*44704f69SBart Van Assche                     else {
1829*44704f69SBart Van Assche                         ll = sg_get_llnum(cp + 1);
1830*44704f69SBart Van Assche                         if ((-1 == ll) || (ll > UINT_MAX)) {
1831*44704f69SBart Van Assche                             pr2serr_lk("could not decode hi_lba, or > "
1832*44704f69SBart Van Assche                                        "UINT_MAX\n");
1833*44704f69SBart Van Assche                             return 1;
1834*44704f69SBart Van Assche                         } else
1835*44704f69SBart Van Assche                             op->hi_lba = (unsigned int)ll;
1836*44704f69SBart Van Assche                     }
1837*44704f69SBart Van Assche                 }
1838*44704f69SBart Van Assche             } else {
1839*44704f69SBart Van Assche                 pr2serr_lk("--lba= expects a number\n");
1840*44704f69SBart Van Assche                 return 1;
1841*44704f69SBart Van Assche             }
1842*44704f69SBart Van Assche             break;
1843*44704f69SBart Van Assche         case 'L':
1844*44704f69SBart Van Assche             op->lb_sz = sg_get_num(optarg);
1845*44704f69SBart Van Assche             if (op->lb_sz < 0) {
1846*44704f69SBart Van Assche                 pr2serr_lk("--lbsz= expects power of 2\n");
1847*44704f69SBart Van Assche                 return 1;
1848*44704f69SBart Van Assche             }
1849*44704f69SBart Van Assche             if (0 == op->lb_sz)
1850*44704f69SBart Van Assche                 op->lb_sz = DEF_LB_SZ;
1851*44704f69SBart Van Assche             break;
1852*44704f69SBart Van Assche         case 'm':
1853*44704f69SBart Van Assche             op->mmap_io = true;
1854*44704f69SBart Van Assche             break;
1855*44704f69SBart Van Assche         case 'M':
1856*44704f69SBart Van Assche             if (isdigit(*optarg)) {
1857*44704f69SBart Van Assche                 n = atoi(optarg);
1858*44704f69SBart Van Assche                 if ((n < 1) || (n > MAX_Q_PER_FD)) {
1859*44704f69SBart Van Assche                     pr2serr_lk("-M expects a value from 1 to %d\n",
1860*44704f69SBart Van Assche                                MAX_Q_PER_FD);
1861*44704f69SBart Van Assche                     return 1;
1862*44704f69SBart Van Assche                 }
1863*44704f69SBart Van Assche                 maxq_per_thread_given = true;
1864*44704f69SBart Van Assche                 op->maxq_per_thread = n;
1865*44704f69SBart Van Assche             } else {
1866*44704f69SBart Van Assche                 pr2serr_lk("--maxqpt= expects a number\n");
1867*44704f69SBart Van Assche                 return 1;
1868*44704f69SBart Van Assche             }
1869*44704f69SBart Van Assche             break;
1870*44704f69SBart Van Assche         case 'n':
1871*44704f69SBart Van Assche             if (isdigit(*optarg))
1872*44704f69SBart Van Assche                 op->num_per_thread = sg_get_num(optarg);
1873*44704f69SBart Van Assche             else {
1874*44704f69SBart Van Assche                 pr2serr_lk("--numpt= expects a number\n");
1875*44704f69SBart Van Assche                 return 1;
1876*44704f69SBart Van Assche             }
1877*44704f69SBart Van Assche             break;
1878*44704f69SBart Van Assche         case 'N':
1879*44704f69SBart Van Assche             op->no_xfer = true;
1880*44704f69SBart Van Assche             break;
1881*44704f69SBart Van Assche         case 'O':
1882*44704f69SBart Van Assche             if (isdigit(*optarg))
1883*44704f69SBart Van Assche                 op->ovn = sg_get_num(optarg);
1884*44704f69SBart Van Assche             else {
1885*44704f69SBart Van Assche                 pr2serr_lk("--override= expects a number\n");
1886*44704f69SBart Van Assche                 return 1;
1887*44704f69SBart Van Assche             }
1888*44704f69SBart Van Assche             if (op->ovn < 0) {
1889*44704f69SBart Van Assche                 pr2serr_lk("--override= bad number\n");
1890*44704f69SBart Van Assche                 return 1;
1891*44704f69SBart Van Assche             }
1892*44704f69SBart Van Assche             break;
1893*44704f69SBart Van Assche         case 'p':
1894*44704f69SBart Van Assche             op->pack_id_force = true;
1895*44704f69SBart Van Assche             break;
1896*44704f69SBart Van Assche         case 'q':
1897*44704f69SBart Van Assche             if (isdigit(*optarg)) {
1898*44704f69SBart Van Assche                 n = atoi(optarg);
1899*44704f69SBart Van Assche                 if (0 == n)
1900*44704f69SBart Van Assche                     op->blqd = BLQ_AT_HEAD;
1901*44704f69SBart Van Assche                 else if (1 == n)
1902*44704f69SBart Van Assche                     op->blqd = BLQ_AT_TAIL;
1903*44704f69SBart Van Assche             } else {
1904*44704f69SBart Van Assche                 pr2serr_lk("--qat= expects a number: 0 or 1\n");
1905*44704f69SBart Van Assche                 return 1;
1906*44704f69SBart Van Assche             }
1907*44704f69SBart Van Assche             break;
1908*44704f69SBart Van Assche         case 'Q':
1909*44704f69SBart Van Assche             if (isdigit(*optarg)) {
1910*44704f69SBart Van Assche                 n = atoi(optarg);
1911*44704f69SBart Van Assche                 if (0 == n)
1912*44704f69SBart Van Assche                     op->myqd = MYQD_LOW;
1913*44704f69SBart Van Assche                 else if (1 == n)
1914*44704f69SBart Van Assche                     op->myqd = MYQD_MEDIUM;
1915*44704f69SBart Van Assche                 else if (2 == n)
1916*44704f69SBart Van Assche                     op->myqd = MYQD_HIGH;
1917*44704f69SBart Van Assche             } else {
1918*44704f69SBart Van Assche                 pr2serr_lk("--qfav= expects a number: 0, 1 or 2\n");
1919*44704f69SBart Van Assche                 return 1;
1920*44704f69SBart Van Assche             }
1921*44704f69SBart Van Assche             break;
1922*44704f69SBart Van Assche         case 'R':
1923*44704f69SBart Van Assche             op->c2e = SCSI_READ16;
1924*44704f69SBart Van Assche             break;
1925*44704f69SBart Van Assche         case 's':
1926*44704f69SBart Van Assche             if (isdigit(*optarg)) {
1927*44704f69SBart Van Assche                 op->lb_sz = atoi(optarg);
1928*44704f69SBart Van Assche                 if (op->lb_sz < 256) {
1929*44704f69SBart Van Assche                     cerr << "Strange lb_sz, using 256" << endl;
1930*44704f69SBart Van Assche                     op->lb_sz = 256;
1931*44704f69SBart Van Assche                 }
1932*44704f69SBart Van Assche             } else {
1933*44704f69SBart Van Assche                 pr2serr_lk("--szlb= expects a number\n");
1934*44704f69SBart Van Assche                 return 1;
1935*44704f69SBart Van Assche             }
1936*44704f69SBart Van Assche             if ((cp = strchr(optarg, ','))) {
1937*44704f69SBart Van Assche                 n = sg_get_num(cp + 1);
1938*44704f69SBart Van Assche                 if (n < 1) {
1939*44704f69SBart Van Assche                     pr2serr_lk("could not decode 2nd part of "
1940*44704f69SBart Van Assche                                "--szlb=LBS,NLBS\n");
1941*44704f69SBart Van Assche                     return 1;
1942*44704f69SBart Van Assche                 }
1943*44704f69SBart Van Assche                 op->num_lbs = n;
1944*44704f69SBart Van Assche             }
1945*44704f69SBart Van Assche             break;
1946*44704f69SBart Van Assche         case 'S':
1947*44704f69SBart Van Assche             ++op->stats;
1948*44704f69SBart Van Assche             break;
1949*44704f69SBart Van Assche         case 't':
1950*44704f69SBart Van Assche             if (isdigit(*optarg))
1951*44704f69SBart Van Assche                 num_threads = atoi(optarg);
1952*44704f69SBart Van Assche             else {
1953*44704f69SBart Van Assche                 pr2serr_lk("--tnum= expects a number\n");
1954*44704f69SBart Van Assche                 return 1;
1955*44704f69SBart Van Assche             }
1956*44704f69SBart Van Assche             break;
1957*44704f69SBart Van Assche         case 'T':
1958*44704f69SBart Van Assche             op->c2e = SCSI_TUR;
1959*44704f69SBart Van Assche             break;
1960*44704f69SBart Van Assche         case 'u':
1961*44704f69SBart Van Assche             op->submit = true;
1962*44704f69SBart Van Assche             break;
1963*44704f69SBart Van Assche         case 'v':
1964*44704f69SBart Van Assche             op->verbose_given = true;
1965*44704f69SBart Van Assche             ++op->verbose;
1966*44704f69SBart Van Assche             break;
1967*44704f69SBart Van Assche         case 'V':
1968*44704f69SBart Van Assche             op->version_given = true;
1969*44704f69SBart Van Assche             break;
1970*44704f69SBart Van Assche         case 'w':
1971*44704f69SBart Van Assche             if ((isdigit(*optarg) || ('-' == *optarg))) {
1972*44704f69SBart Van Assche                 if ('-' == *optarg)
1973*44704f69SBart Van Assche                     op->wait_ms = - atoi(optarg + 1);
1974*44704f69SBart Van Assche                 else
1975*44704f69SBart Van Assche                     op->wait_ms = atoi(optarg);
1976*44704f69SBart Van Assche             } else {
1977*44704f69SBart Van Assche                 pr2serr_lk("--wait= expects a number\n");
1978*44704f69SBart Van Assche                 return 1;
1979*44704f69SBart Van Assche             }
1980*44704f69SBart Van Assche             break;
1981*44704f69SBart Van Assche         case 'W':
1982*44704f69SBart Van Assche             op->c2e = SCSI_WRITE16;
1983*44704f69SBart Van Assche             break;
1984*44704f69SBart Van Assche         default:
1985*44704f69SBart Van Assche             pr2serr_lk("unrecognised option code 0x%x ??\n", c);
1986*44704f69SBart Van Assche             usage();
1987*44704f69SBart Van Assche             return 1;
1988*44704f69SBart Van Assche         }
1989*44704f69SBart Van Assche     }
1990*44704f69SBart Van Assche     if (optind < argc) {
1991*44704f69SBart Van Assche         for (; optind < argc; ++optind)
1992*44704f69SBart Van Assche             op->dev_names.push_back(argv[optind]);
1993*44704f69SBart Van Assche     }
1994*44704f69SBart Van Assche #ifdef DEBUG
1995*44704f69SBart Van Assche     pr2serr_lk("In DEBUG mode, ");
1996*44704f69SBart Van Assche     if (op->verbose_given && op->version_given) {
1997*44704f69SBart Van Assche         pr2serr_lk("but override: '-vV' given, zero verbose and continue\n");
1998*44704f69SBart Van Assche         op->verbose_given = false;
1999*44704f69SBart Van Assche         op->version_given = false;
2000*44704f69SBart Van Assche         op->verbose = 0;
2001*44704f69SBart Van Assche     } else if (! op->verbose_given) {
2002*44704f69SBart Van Assche         pr2serr_lk("set '-vv'\n");
2003*44704f69SBart Van Assche         op->verbose = 2;
2004*44704f69SBart Van Assche     } else
2005*44704f69SBart Van Assche         pr2serr_lk("keep verbose=%d\n", op->verbose);
2006*44704f69SBart Van Assche #else
2007*44704f69SBart Van Assche     if (op->verbose_given && op->version_given)
2008*44704f69SBart Van Assche         pr2serr_lk("Not in DEBUG mode, so '-vV' has no special action\n");
2009*44704f69SBart Van Assche #endif
2010*44704f69SBart Van Assche     if (op->version_given) {
2011*44704f69SBart Van Assche         pr2serr_lk("version: %s\n", version_str);
2012*44704f69SBart Van Assche         return 0;
2013*44704f69SBart Van Assche     }
2014*44704f69SBart Van Assche     if (op->mmap_io) {
2015*44704f69SBart Van Assche         if (maxq_per_thread_given && (op->maxq_per_thread > 1)) {
2016*44704f69SBart Van Assche             pr2serr_lk("With mmap_io selected, QPT cannot exceed 1\n");
2017*44704f69SBart Van Assche             return 1;
2018*44704f69SBart Van Assche         } else if (op->direct) {
2019*44704f69SBart Van Assche             pr2serr_lk("direct IO and mmap-ed IO cannot both be selected\n");
2020*44704f69SBart Van Assche             return 1;
2021*44704f69SBart Van Assche         } else if (op->generic_sync) {
2022*44704f69SBart Van Assche             pr2serr_lk("--generic-sync and and mmap-ed IO are compatible\n");
2023*44704f69SBart Van Assche             return 1;
2024*44704f69SBart Van Assche         } else
2025*44704f69SBart Van Assche             op->maxq_per_thread = 1;
2026*44704f69SBart Van Assche     }
2027*44704f69SBart Van Assche     if (! op->cmd_time && getenv("SG3_UTILS_LINUX_NANO")) {
2028*44704f69SBart Van Assche         op->cmd_time = true;
2029*44704f69SBart Van Assche         if (op->verbose)
2030*44704f69SBart Van Assche             fprintf(stderr, "setting nanosecond timing due to environment "
2031*44704f69SBart Van Assche                     "variable: SG3_UTILS_LINUX_NANO\n");
2032*44704f69SBart Van Assche     }
2033*44704f69SBart Van Assche     if (0 == op->dev_names.size()) {
2034*44704f69SBart Van Assche         fprintf(stderr, "No sg_disk_device-s given\n\n");
2035*44704f69SBart Van Assche         usage();
2036*44704f69SBart Van Assche         return 1;
2037*44704f69SBart Van Assche     }
2038*44704f69SBart Van Assche     if (op->hi_lba && (op->lba > op->hi_lba)) {
2039*44704f69SBart Van Assche         cerr << "lba,hi_lba range is illegal" << endl;
2040*44704f69SBart Van Assche         return 1;
2041*44704f69SBart Van Assche     }
2042*44704f69SBart Van Assche     if (op->v4) {
2043*44704f69SBart Van Assche         if (! op->submit) {
2044*44704f69SBart Van Assche             op->submit = true;
2045*44704f69SBart Van Assche             if (op->verbose > 1)
2046*44704f69SBart Van Assche                 cerr << "when --v4 is given, --submit will be set" << endl;
2047*44704f69SBart Van Assche         }
2048*44704f69SBart Van Assche     }
2049*44704f69SBart Van Assche 
2050*44704f69SBart Van Assche     try {
2051*44704f69SBart Van Assche         int k, sg_ver_num;
2052*44704f69SBart Van Assche         unsigned int last_lba;
2053*44704f69SBart Van Assche         unsigned int blk_sz;
2054*44704f69SBart Van Assche         struct stat a_stat;
2055*44704f69SBart Van Assche 
2056*44704f69SBart Van Assche         for (k = 0; k < (int)op->dev_names.size(); ++k) {
2057*44704f69SBart Van Assche             int res;
2058*44704f69SBart Van Assche             const char * dev_name;
2059*44704f69SBart Van Assche             char b[128];
2060*44704f69SBart Van Assche 
2061*44704f69SBart Van Assche             dev_name = op->dev_names[k];
2062*44704f69SBart Van Assche             if (stat(dev_name, &a_stat) < 0) {
2063*44704f69SBart Van Assche                 snprintf(b, sizeof(b), "could not stat() %s", dev_name);
2064*44704f69SBart Van Assche                 perror(b);
2065*44704f69SBart Van Assche                 return 1;
2066*44704f69SBart Van Assche             }
2067*44704f69SBart Van Assche             if (! S_ISCHR(a_stat.st_mode)) {
2068*44704f69SBart Van Assche                 pr2serr_lk("%s should be a sg device which is a char "
2069*44704f69SBart Van Assche                            "device. %s\n", dev_name, dev_name);
2070*44704f69SBart Van Assche                 pr2serr_lk("is not a char device and damage could be done "
2071*44704f69SBart Van Assche                            "if it is a BLOCK\ndevice, exiting ...\n");
2072*44704f69SBart Van Assche                 return 1;
2073*44704f69SBart Van Assche             }
2074*44704f69SBart Van Assche             res = do_inquiry_prod_id(dev_name, op->block, sg_ver_num,
2075*44704f69SBart Van Assche                                      b, sizeof(b));
2076*44704f69SBart Van Assche             if (! force) {
2077*44704f69SBart Van Assche                 if (res) {
2078*44704f69SBart Van Assche                     pr2serr_lk("INQUIRY failed on %s\n", dev_name);
2079*44704f69SBart Van Assche                     return 1;
2080*44704f69SBart Van Assche                 }
2081*44704f69SBart Van Assche                 // For safety, since <lba> written to, only permit scsi_debug
2082*44704f69SBart Van Assche                 // devices. Bypass this with '-f' option.
2083*44704f69SBart Van Assche                 if (0 != memcmp("scsi_debug", b, 10)) {
2084*44704f69SBart Van Assche                     pr2serr_lk("Since this utility may write to LBAs, "
2085*44704f69SBart Van Assche                                "only devices with the\n"
2086*44704f69SBart Van Assche                                "product ID 'scsi_debug' accepted. Use '-f' "
2087*44704f69SBart Van Assche                                "to override.\n");
2088*44704f69SBart Van Assche                     return 2;
2089*44704f69SBart Van Assche                 }
2090*44704f69SBart Van Assche             }
2091*44704f69SBart Van Assche             if (sg_ver_num < 30000) {
2092*44704f69SBart Van Assche                 pr2serr_lk("%s either not sg device or too old\n", dev_name);
2093*44704f69SBart Van Assche                 return 2;
2094*44704f69SBart Van Assche             } else if (sg_ver_num >= 40030) {
2095*44704f69SBart Van Assche                 op->sg_vn_ge_40030 = true;
2096*44704f69SBart Van Assche                 op->sg_vn_ge_40000 = true;
2097*44704f69SBart Van Assche                 if (! (op->v3_given || op->v4_given)) {
2098*44704f69SBart Van Assche                     op->v4 = true;
2099*44704f69SBart Van Assche                     op->v3 = false;
2100*44704f69SBart Van Assche                     op->submit = true;
2101*44704f69SBart Van Assche                 }
2102*44704f69SBart Van Assche             } else if (sg_ver_num >= 40000) {
2103*44704f69SBart Van Assche                 op->sg_vn_ge_40030 = false;
2104*44704f69SBart Van Assche                 op->sg_vn_ge_40000 = true;
2105*44704f69SBart Van Assche                 if (! (op->v3_given || op->v4_given)) {
2106*44704f69SBart Van Assche                     op->v4 = true;
2107*44704f69SBart Van Assche                     op->v3 = false;
2108*44704f69SBart Van Assche                     op->submit = true;
2109*44704f69SBart Van Assche                 }
2110*44704f69SBart Van Assche             } else {
2111*44704f69SBart Van Assche                 if (! (op->v3_given || op->v4_given)) {
2112*44704f69SBart Van Assche                     op->v4 = false;
2113*44704f69SBart Van Assche                     op->v3 = true;
2114*44704f69SBart Van Assche                     op->submit = false;
2115*44704f69SBart Van Assche                 }
2116*44704f69SBart Van Assche             }
2117*44704f69SBart Van Assche 
2118*44704f69SBart Van Assche             if ((SCSI_WRITE16 == op->c2e) || (SCSI_READ16 == op->c2e)) {
2119*44704f69SBart Van Assche                 res = do_read_capacity(dev_name, op->block, &last_lba,
2120*44704f69SBart Van Assche                                        &blk_sz);
2121*44704f69SBart Van Assche                 if (2 == res)
2122*44704f69SBart Van Assche                     res = do_read_capacity(dev_name, op->block, &last_lba,
2123*44704f69SBart Van Assche                                            &blk_sz);
2124*44704f69SBart Van Assche                 if (res) {
2125*44704f69SBart Van Assche                     pr2serr_lk("READ CAPACITY(10) failed on %s\n", dev_name);
2126*44704f69SBart Van Assche                     return 1;
2127*44704f69SBart Van Assche                 }
2128*44704f69SBart Van Assche                 if (blk_sz != (unsigned int)op->lb_sz) {
2129*44704f69SBart Van Assche                     pr2serr_lk(">>> Logical block size (%d) of %s\n"
2130*44704f69SBart Van Assche                                "    differs from command line option (or "
2131*44704f69SBart Van Assche                                "default)\n", blk_sz, dev_name);
2132*44704f69SBart Van Assche                    pr2serr_lk("... continue anyway\n");
2133*44704f69SBart Van Assche                 }
2134*44704f69SBart Van Assche                 op->blk_szs.push_back(blk_sz);
2135*44704f69SBart Van Assche                 if (UINT_MAX == op->hi_lba)
2136*44704f69SBart Van Assche                     op->hi_lbas.push_back(last_lba);
2137*44704f69SBart Van Assche             }
2138*44704f69SBart Van Assche         }
2139*44704f69SBart Van Assche 
2140*44704f69SBart Van Assche         start_tm.tv_sec = 0;
2141*44704f69SBart Van Assche         start_tm.tv_nsec = 0;
2142*44704f69SBart Van Assche         if (clock_gettime(CLOCK_MONOTONIC, &start_tm) < 0)
2143*44704f69SBart Van Assche             perror("clock_gettime failed");
2144*44704f69SBart Van Assche 
2145*44704f69SBart Van Assche         vector<thread *> vt;
2146*44704f69SBart Van Assche 
2147*44704f69SBart Van Assche         /* start multi-threaded section */
2148*44704f69SBart Van Assche         for (k = 0; k < num_threads; ++k) {
2149*44704f69SBart Van Assche             thread * tp = new thread {work_thread, k, op};
2150*44704f69SBart Van Assche             vt.push_back(tp);
2151*44704f69SBart Van Assche         }
2152*44704f69SBart Van Assche 
2153*44704f69SBart Van Assche         // g++ 4.7.3 didn't like range-for loop here
2154*44704f69SBart Van Assche         for (k = 0; k < (int)vt.size(); ++k)
2155*44704f69SBart Van Assche             vt[k]->join();
2156*44704f69SBart Van Assche         /* end multi-threaded section, just this main thread left */
2157*44704f69SBart Van Assche 
2158*44704f69SBart Van Assche         for (k = 0; k < (int)vt.size(); ++k)
2159*44704f69SBart Van Assche             delete vt[k];
2160*44704f69SBart Van Assche 
2161*44704f69SBart Van Assche         n = uniq_pack_id.load() - 1;
2162*44704f69SBart Van Assche         if (((n > 0) || op->generic_sync) &&
2163*44704f69SBart Van Assche             (0 == clock_gettime(CLOCK_MONOTONIC, &end_tm))) {
2164*44704f69SBart Van Assche             struct timespec res_tm;
2165*44704f69SBart Van Assche             double a, b;
2166*44704f69SBart Van Assche 
2167*44704f69SBart Van Assche             if (op->generic_sync)
2168*44704f69SBart Van Assche                 n = op->num_per_thread * num_threads;
2169*44704f69SBart Van Assche             res_tm.tv_sec = end_tm.tv_sec - start_tm.tv_sec;
2170*44704f69SBart Van Assche             res_tm.tv_nsec = end_tm.tv_nsec - start_tm.tv_nsec;
2171*44704f69SBart Van Assche             if (res_tm.tv_nsec < 0) {
2172*44704f69SBart Van Assche                 --res_tm.tv_sec;
2173*44704f69SBart Van Assche                 res_tm.tv_nsec += 1000000000;
2174*44704f69SBart Van Assche             }
2175*44704f69SBart Van Assche             a = res_tm.tv_sec;
2176*44704f69SBart Van Assche             a += (0.000001 * (res_tm.tv_nsec / 1000));
2177*44704f69SBart Van Assche             b = (double)n;
2178*44704f69SBart Van Assche             if (a > 0.000001) {
2179*44704f69SBart Van Assche                 printf("Time to complete %d commands was %d.%06d seconds\n",
2180*44704f69SBart Van Assche                        n, (int)res_tm.tv_sec, (int)(res_tm.tv_nsec / 1000));
2181*44704f69SBart Van Assche                 printf("Implies %.0f IOPS\n", (b / a));
2182*44704f69SBart Van Assche             }
2183*44704f69SBart Van Assche         }
2184*44704f69SBart Van Assche 
2185*44704f69SBart Van Assche         if (op->verbose || op->stats) {
2186*44704f69SBart Van Assche             cout << "Number of sync_starts: " << sync_starts.load() << endl;
2187*44704f69SBart Van Assche             cout << "Number of async_starts: " << async_starts.load() << endl;
2188*44704f69SBart Van Assche             cout << "Number of async_finishes: " << async_finishes.load() <<
2189*44704f69SBart Van Assche                     endl;
2190*44704f69SBart Van Assche             cout << "Last pack_id: " << n << endl;
2191*44704f69SBart Van Assche         }
2192*44704f69SBart Van Assche         n = start_ebusy_count.load();
2193*44704f69SBart Van Assche         if (op->verbose || op->stats || (n > 0))
2194*44704f69SBart Van Assche             cout << "Number of start EBUSYs: " << n << endl;
2195*44704f69SBart Van Assche         n = fin_ebusy_count.load();
2196*44704f69SBart Van Assche         if (op->verbose || op->stats || (n > 0))
2197*44704f69SBart Van Assche             cout << "Number of finish EBUSYs: " << n << endl;
2198*44704f69SBart Van Assche         n = start_eagain_count.load();
2199*44704f69SBart Van Assche         if (op->verbose || op->stats || (n > 0))
2200*44704f69SBart Van Assche             cout << "Number of start EAGAINs: " << n << endl;
2201*44704f69SBart Van Assche         n = fin_eagain_count.load();
2202*44704f69SBart Van Assche         if (op->verbose || op->stats || (n > 0))
2203*44704f69SBart Van Assche             cout << "Number of finish EAGAINs: " << n << endl;
2204*44704f69SBart Van Assche         n = start_e2big_count.load();
2205*44704f69SBart Van Assche         if (op->verbose || op->stats || (n > 0))
2206*44704f69SBart Van Assche             cout << "Number of E2BIGs: " << n << endl;
2207*44704f69SBart Van Assche         n = start_edom_count.load();
2208*44704f69SBart Van Assche         if (op->verbose || op->stats || (n > 0))
2209*44704f69SBart Van Assche             cout << "Number of EDOMs: " << n << endl;
2210*44704f69SBart Van Assche         n = enomem_count.load();
2211*44704f69SBart Van Assche         if (op->verbose || op->stats || (n > 0))
2212*44704f69SBart Van Assche             cout << "Number of ENOMEMs: " << n << endl;
2213*44704f69SBart Van Assche     }
2214*44704f69SBart Van Assche     catch(system_error& e)  {
2215*44704f69SBart Van Assche         cerr << "got a system_error exception: " << e.what() << '\n';
2216*44704f69SBart Van Assche         auto ec = e.code();
2217*44704f69SBart Van Assche         cerr << "category: " << ec.category().name() << '\n';
2218*44704f69SBart Van Assche         cerr << "value: " << ec.value() << '\n';
2219*44704f69SBart Van Assche         cerr << "message: " << ec.message() << '\n';
2220*44704f69SBart Van Assche         cerr << "\nNote: if g++ may need '-pthread' or similar in "
2221*44704f69SBart Van Assche                 "compile/link line" << '\n';
2222*44704f69SBart Van Assche     }
2223*44704f69SBart Van Assche     catch(...) {
2224*44704f69SBart Van Assche         cerr << "got another exception: " << '\n';
2225*44704f69SBart Van Assche     }
2226*44704f69SBart Van Assche     return 0;
2227*44704f69SBart Van Assche }
2228