1*44704f69SBart Van Assche /* A utility program for copying files. Specialised for "files" that
2*44704f69SBart Van Assche * represent devices that understand the SCSI command set.
3*44704f69SBart Van Assche *
4*44704f69SBart Van Assche * Copyright (C) 1999 - 2022 D. Gilbert and P. Allworth
5*44704f69SBart Van Assche * This program is free software; you can redistribute it and/or modify
6*44704f69SBart Van Assche * it under the terms of the GNU General Public License as published by
7*44704f69SBart Van Assche * the Free Software Foundation; either version 2, or (at your option)
8*44704f69SBart Van Assche * any later version.
9*44704f69SBart Van Assche *
10*44704f69SBart Van Assche * SPDX-License-Identifier: GPL-2.0-or-later
11*44704f69SBart Van Assche *
12*44704f69SBart Van Assche * This program is a specialisation of the Unix "dd" command in which
13*44704f69SBart Van Assche * either the input or the output file is a scsi generic device, raw
14*44704f69SBart Van Assche * device, a block device or a normal file. The logical block size ('bs')
15*44704f69SBart Van Assche * is assumed to be 512 if not given. This program complains if 'ibs' or
16*44704f69SBart Van Assche * 'obs' are given with a value that differs from 'bs' (or the default 512).
17*44704f69SBart Van Assche * If 'if' is not given or 'if=-' then stdin is assumed. If 'of' is
18*44704f69SBart Van Assche * not given or 'of=-' then stdout assumed.
19*44704f69SBart Van Assche *
20*44704f69SBart Van Assche * A non-standard argument "bpt" (blocks per transfer) is added to control
21*44704f69SBart Van Assche * the maximum number of blocks in each transfer. The default value is 128.
22*44704f69SBart Van Assche * For example if "bs=512" and "bpt=32" then a maximum of 32 blocks (16 KiB
23*44704f69SBart Van Assche * in this case) is transferred to or from the sg device in a single SCSI
24*44704f69SBart Van Assche * command. The actual size of the SCSI READ or WRITE command block can be
25*44704f69SBart Van Assche * selected with the "cdbsz" argument.
26*44704f69SBart Van Assche *
27*44704f69SBart Van Assche * This version is designed for the Linux kernel 2, 3, 4 and 5 series.
28*44704f69SBart Van Assche */
29*44704f69SBart Van Assche
30*44704f69SBart Van Assche #define _XOPEN_SOURCE 600
31*44704f69SBart Van Assche #ifndef _GNU_SOURCE
32*44704f69SBart Van Assche #define _GNU_SOURCE 1
33*44704f69SBart Van Assche #endif
34*44704f69SBart Van Assche
35*44704f69SBart Van Assche #include <unistd.h>
36*44704f69SBart Van Assche #include <fcntl.h>
37*44704f69SBart Van Assche #include <stdio.h>
38*44704f69SBart Van Assche #include <stdlib.h>
39*44704f69SBart Van Assche #include <stdarg.h>
40*44704f69SBart Van Assche #include <stdbool.h>
41*44704f69SBart Van Assche #include <string.h>
42*44704f69SBart Van Assche #include <signal.h>
43*44704f69SBart Van Assche #include <ctype.h>
44*44704f69SBart Van Assche #include <errno.h>
45*44704f69SBart Van Assche #include <time.h>
46*44704f69SBart Van Assche #include <limits.h>
47*44704f69SBart Van Assche #define __STDC_FORMAT_MACROS 1
48*44704f69SBart Van Assche #include <inttypes.h>
49*44704f69SBart Van Assche #include <sys/ioctl.h>
50*44704f69SBart Van Assche #include <sys/stat.h>
51*44704f69SBart Van Assche #include <sys/time.h>
52*44704f69SBart Van Assche #include <sys/file.h>
53*44704f69SBart Van Assche #include <sys/sysmacros.h>
54*44704f69SBart Van Assche #ifndef major
55*44704f69SBart Van Assche #include <sys/types.h>
56*44704f69SBart Van Assche #endif
57*44704f69SBart Van Assche #include <linux/major.h> /* for MEM_MAJOR, SCSI_GENERIC_MAJOR, etc */
58*44704f69SBart Van Assche #include <linux/fs.h> /* for BLKSSZGET and friends */
59*44704f69SBart Van Assche
60*44704f69SBart Van Assche #ifdef HAVE_CONFIG_H
61*44704f69SBart Van Assche #include "config.h"
62*44704f69SBart Van Assche #endif
63*44704f69SBart Van Assche #ifdef HAVE_GETRANDOM
64*44704f69SBart Van Assche #include <sys/random.h> /* for getrandom() system call */
65*44704f69SBart Van Assche #endif
66*44704f69SBart Van Assche #include "sg_lib.h"
67*44704f69SBart Van Assche #include "sg_cmds_basic.h"
68*44704f69SBart Van Assche #include "sg_cmds_extra.h"
69*44704f69SBart Van Assche #include "sg_io_linux.h"
70*44704f69SBart Van Assche #include "sg_unaligned.h"
71*44704f69SBart Van Assche #include "sg_pr2serr.h"
72*44704f69SBart Van Assche
73*44704f69SBart Van Assche static const char * version_str = "6.35 20220826";
74*44704f69SBart Van Assche
75*44704f69SBart Van Assche
76*44704f69SBart Van Assche #define ME "sg_dd: "
77*44704f69SBart Van Assche
78*44704f69SBart Van Assche
79*44704f69SBart Van Assche #define STR_SZ 1024
80*44704f69SBart Van Assche #define INOUTF_SZ 512
81*44704f69SBart Van Assche #define EBUFF_SZ 768
82*44704f69SBart Van Assche
83*44704f69SBart Van Assche #define DEF_BLOCK_SIZE 512
84*44704f69SBart Van Assche #define DEF_BLOCKS_PER_TRANSFER 128
85*44704f69SBart Van Assche #define DEF_BLOCKS_PER_2048TRANSFER 32
86*44704f69SBart Van Assche #define DEF_SCSI_CDBSZ 10
87*44704f69SBart Van Assche #define MAX_SCSI_CDBSZ 16
88*44704f69SBart Van Assche #define MAX_BPT_VALUE (1 << 24) /* used for maximum bs as well */
89*44704f69SBart Van Assche #define MAX_COUNT_SKIP_SEEK (1LL << 48) /* coverity wants upper bound */
90*44704f69SBart Van Assche
91*44704f69SBart Van Assche #define DEF_MODE_CDB_SZ 10
92*44704f69SBart Van Assche #define DEF_MODE_RESP_LEN 252
93*44704f69SBart Van Assche #define RW_ERR_RECOVERY_MP 1
94*44704f69SBart Van Assche #define CACHING_MP 8
95*44704f69SBart Van Assche #define CONTROL_MP 0xa
96*44704f69SBart Van Assche
97*44704f69SBart Van Assche #define SENSE_BUFF_LEN 64 /* Arbitrary, could be larger */
98*44704f69SBart Van Assche #define READ_CAP_REPLY_LEN 8
99*44704f69SBart Van Assche #define RCAP16_REPLY_LEN 32
100*44704f69SBart Van Assche #define READ_LONG_OPCODE 0x3E
101*44704f69SBart Van Assche #define READ_LONG_CMD_LEN 10
102*44704f69SBart Van Assche #define READ_LONG_DEF_BLK_INC 8
103*44704f69SBart Van Assche #define VERIFY10 0x2f
104*44704f69SBart Van Assche #define VERIFY12 0xaf
105*44704f69SBart Van Assche #define VERIFY16 0x8f
106*44704f69SBart Van Assche
107*44704f69SBart Van Assche #define DEF_TIMEOUT 60000 /* 60,000 millisecs == 60 seconds */
108*44704f69SBart Van Assche
109*44704f69SBart Van Assche #ifndef RAW_MAJOR
110*44704f69SBart Van Assche #define RAW_MAJOR 255 /*unlikely value */
111*44704f69SBart Van Assche #endif
112*44704f69SBart Van Assche
113*44704f69SBart Van Assche #define SG_LIB_FLOCK_ERR 90
114*44704f69SBart Van Assche
115*44704f69SBart Van Assche #define FT_OTHER 1 /* filetype is probably normal */
116*44704f69SBart Van Assche #define FT_SG 2 /* filetype is sg char device or supports
117*44704f69SBart Van Assche SG_IO ioctl */
118*44704f69SBart Van Assche #define FT_RAW 4 /* filetype is raw char device */
119*44704f69SBart Van Assche #define FT_DEV_NULL 8 /* either "/dev/null" or "." as filename */
120*44704f69SBart Van Assche #define FT_ST 16 /* filetype is st char device (tape) */
121*44704f69SBart Van Assche #define FT_BLOCK 32 /* filetype is block device */
122*44704f69SBart Van Assche #define FT_FIFO 64 /* filetype is a fifo (name pipe) */
123*44704f69SBart Van Assche #define FT_NVME 128 /* NVMe char device (e.g. /dev/nvme2) */
124*44704f69SBart Van Assche #define FT_RANDOM_0_FF 256 /* iflag=00, iflag=ff and iflag=random
125*44704f69SBart Van Assche overriding if=IFILE */
126*44704f69SBart Van Assche #define FT_ERROR 512 /* couldn't "stat" file */
127*44704f69SBart Van Assche
128*44704f69SBart Van Assche #define DEV_NULL_MINOR_NUM 3
129*44704f69SBart Van Assche
130*44704f69SBart Van Assche #define SG_DD_BYPASS 999 /* failed but coe set */
131*44704f69SBart Van Assche
132*44704f69SBart Van Assche /* If platform does not support O_DIRECT then define it harmlessly */
133*44704f69SBart Van Assche #ifndef O_DIRECT
134*44704f69SBart Van Assche #define O_DIRECT 0
135*44704f69SBart Van Assche #endif
136*44704f69SBart Van Assche
137*44704f69SBart Van Assche #define MIN_RESERVED_SIZE 8192
138*44704f69SBart Van Assche
139*44704f69SBart Van Assche #define MAX_UNIT_ATTENTIONS 10
140*44704f69SBart Van Assche #define MAX_ABORTED_CMDS 256
141*44704f69SBart Van Assche
142*44704f69SBart Van Assche #define PROGRESS_TRIGGER_MS 120000 /* milliseconds: 2 minutes */
143*44704f69SBart Van Assche #define PROGRESS2_TRIGGER_MS 60000 /* milliseconds: 1 minute */
144*44704f69SBart Van Assche #define PROGRESS3_TRIGGER_MS 30000 /* milliseconds: 30 seconds */
145*44704f69SBart Van Assche
146*44704f69SBart Van Assche static int sum_of_resids = 0;
147*44704f69SBart Van Assche
148*44704f69SBart Van Assche static int64_t dd_count = -1;
149*44704f69SBart Van Assche static int64_t req_count = 0;
150*44704f69SBart Van Assche static int64_t in_full = 0;
151*44704f69SBart Van Assche static int in_partial = 0;
152*44704f69SBart Van Assche static int64_t out_full = 0;
153*44704f69SBart Van Assche static int out_partial = 0;
154*44704f69SBart Van Assche static int64_t out_sparse_num = 0;
155*44704f69SBart Van Assche static int recovered_errs = 0;
156*44704f69SBart Van Assche static int unrecovered_errs = 0;
157*44704f69SBart Van Assche static int miscompare_errs = 0;
158*44704f69SBart Van Assche static int read_longs = 0;
159*44704f69SBart Van Assche static int num_retries = 0;
160*44704f69SBart Van Assche static int progress = 0;
161*44704f69SBart Van Assche static int dry_run = 0;
162*44704f69SBart Van Assche
163*44704f69SBart Van Assche static bool do_time = false;
164*44704f69SBart Van Assche static bool start_tm_valid = false;
165*44704f69SBart Van Assche static bool do_verify = false; /* when false: do copy */
166*44704f69SBart Van Assche static int verbose = 0;
167*44704f69SBart Van Assche static int blk_sz = 0;
168*44704f69SBart Van Assche static int max_uas = MAX_UNIT_ATTENTIONS;
169*44704f69SBart Van Assche static int max_aborted = MAX_ABORTED_CMDS;
170*44704f69SBart Van Assche static int coe_limit = 0;
171*44704f69SBart Van Assche static int coe_count = 0;
172*44704f69SBart Van Assche static int cmd_timeout = DEF_TIMEOUT; /* in milliseconds */
173*44704f69SBart Van Assche static uint32_t glob_pack_id = 0; /* pre-increment */
174*44704f69SBart Van Assche static struct timeval start_tm;
175*44704f69SBart Van Assche
176*44704f69SBart Van Assche static uint8_t * zeros_buff = NULL;
177*44704f69SBart Van Assche static uint8_t * free_zeros_buff = NULL;
178*44704f69SBart Van Assche static int read_long_blk_inc = READ_LONG_DEF_BLK_INC;
179*44704f69SBart Van Assche
180*44704f69SBart Van Assche static long seed;
181*44704f69SBart Van Assche #ifdef HAVE_SRAND48_R /* gcc extension. N.B. non-reentrant version slower */
182*44704f69SBart Van Assche static struct drand48_data drand;/* opaque, used by srand48_r and mrand48_r */
183*44704f69SBart Van Assche #endif
184*44704f69SBart Van Assche
185*44704f69SBart Van Assche static const char * sg_allow_dio = "/sys/module/sg/parameters/allow_dio";
186*44704f69SBart Van Assche
187*44704f69SBart Van Assche struct flags_t {
188*44704f69SBart Van Assche bool append;
189*44704f69SBart Van Assche bool dio;
190*44704f69SBart Van Assche bool direct;
191*44704f69SBart Van Assche bool dpo;
192*44704f69SBart Van Assche bool dsync;
193*44704f69SBart Van Assche bool excl;
194*44704f69SBart Van Assche bool flock;
195*44704f69SBart Van Assche bool ff;
196*44704f69SBart Van Assche bool fua;
197*44704f69SBart Van Assche bool nocreat;
198*44704f69SBart Van Assche bool random;
199*44704f69SBart Van Assche bool sgio;
200*44704f69SBart Van Assche bool sparse;
201*44704f69SBart Van Assche bool zero;
202*44704f69SBart Van Assche int cdbsz;
203*44704f69SBart Van Assche int cdl;
204*44704f69SBart Van Assche int coe;
205*44704f69SBart Van Assche int nocache;
206*44704f69SBart Van Assche int pdt;
207*44704f69SBart Van Assche int retries;
208*44704f69SBart Van Assche };
209*44704f69SBart Van Assche
210*44704f69SBart Van Assche static struct flags_t iflag;
211*44704f69SBart Van Assche static struct flags_t oflag;
212*44704f69SBart Van Assche
213*44704f69SBart Van Assche static void calc_duration_throughput(bool contin);
214*44704f69SBart Van Assche
215*44704f69SBart Van Assche
216*44704f69SBart Van Assche static void
install_handler(int sig_num,void (* sig_handler)(int sig))217*44704f69SBart Van Assche install_handler(int sig_num, void (*sig_handler)(int sig))
218*44704f69SBart Van Assche {
219*44704f69SBart Van Assche struct sigaction sigact;
220*44704f69SBart Van Assche
221*44704f69SBart Van Assche sigaction(sig_num, NULL, &sigact);
222*44704f69SBart Van Assche if (sigact.sa_handler != SIG_IGN) {
223*44704f69SBart Van Assche sigact.sa_handler = sig_handler;
224*44704f69SBart Van Assche sigemptyset(&sigact.sa_mask);
225*44704f69SBart Van Assche sigact.sa_flags = 0;
226*44704f69SBart Van Assche sigaction(sig_num, &sigact, NULL);
227*44704f69SBart Van Assche }
228*44704f69SBart Van Assche }
229*44704f69SBart Van Assche
230*44704f69SBart Van Assche
231*44704f69SBart Van Assche static void
print_stats(const char * str)232*44704f69SBart Van Assche print_stats(const char * str)
233*44704f69SBart Van Assche {
234*44704f69SBart Van Assche if (0 != dd_count)
235*44704f69SBart Van Assche pr2serr(" remaining block count=%" PRId64 "\n", dd_count);
236*44704f69SBart Van Assche pr2serr("%s%" PRId64 "+%d records in\n", str, in_full - in_partial,
237*44704f69SBart Van Assche in_partial);
238*44704f69SBart Van Assche pr2serr("%s%" PRId64 "+%d records %s\n", str, out_full - out_partial,
239*44704f69SBart Van Assche out_partial, (do_verify ? "verified" : "out"));
240*44704f69SBart Van Assche if (oflag.sparse)
241*44704f69SBart Van Assche pr2serr("%s%" PRId64 " bypassed records out\n", str, out_sparse_num);
242*44704f69SBart Van Assche if (recovered_errs > 0)
243*44704f69SBart Van Assche pr2serr("%s%d recovered errors\n", str, recovered_errs);
244*44704f69SBart Van Assche if (num_retries > 0)
245*44704f69SBart Van Assche pr2serr("%s%d retries attempted\n", str, num_retries);
246*44704f69SBart Van Assche if (unrecovered_errs > 0) {
247*44704f69SBart Van Assche pr2serr("%s%d unrecovered error(s)\n", str, unrecovered_errs);
248*44704f69SBart Van Assche if (iflag.coe || oflag.coe)
249*44704f69SBart Van Assche pr2serr("%s%d read_longs fetched part of unrecovered read "
250*44704f69SBart Van Assche "errors\n", str, read_longs);
251*44704f69SBart Van Assche }
252*44704f69SBart Van Assche if (miscompare_errs > 0)
253*44704f69SBart Van Assche pr2serr("%s%d miscompare error(s)\n", str, miscompare_errs);
254*44704f69SBart Van Assche }
255*44704f69SBart Van Assche
256*44704f69SBart Van Assche
257*44704f69SBart Van Assche static void
interrupt_handler(int sig)258*44704f69SBart Van Assche interrupt_handler(int sig)
259*44704f69SBart Van Assche {
260*44704f69SBart Van Assche struct sigaction sigact;
261*44704f69SBart Van Assche
262*44704f69SBart Van Assche sigact.sa_handler = SIG_DFL;
263*44704f69SBart Van Assche sigemptyset(&sigact.sa_mask);
264*44704f69SBart Van Assche sigact.sa_flags = 0;
265*44704f69SBart Van Assche sigaction(sig, &sigact, NULL);
266*44704f69SBart Van Assche pr2serr("Interrupted by signal,");
267*44704f69SBart Van Assche if (do_time)
268*44704f69SBart Van Assche calc_duration_throughput(false);
269*44704f69SBart Van Assche print_stats("");
270*44704f69SBart Van Assche kill(getpid (), sig);
271*44704f69SBart Van Assche }
272*44704f69SBart Van Assche
273*44704f69SBart Van Assche
274*44704f69SBart Van Assche static void
siginfo_handler(int sig)275*44704f69SBart Van Assche siginfo_handler(int sig)
276*44704f69SBart Van Assche {
277*44704f69SBart Van Assche if (sig) { ; } /* unused, dummy to suppress warning */
278*44704f69SBart Van Assche pr2serr("Progress report, continuing ...\n");
279*44704f69SBart Van Assche if (do_time)
280*44704f69SBart Van Assche calc_duration_throughput(true);
281*44704f69SBart Van Assche print_stats(" ");
282*44704f69SBart Van Assche }
283*44704f69SBart Van Assche
284*44704f69SBart Van Assche static bool bsg_major_checked = false;
285*44704f69SBart Van Assche static int bsg_major = 0;
286*44704f69SBart Van Assche
287*44704f69SBart Van Assche static void
find_bsg_major(void)288*44704f69SBart Van Assche find_bsg_major(void)
289*44704f69SBart Van Assche {
290*44704f69SBart Van Assche int n;
291*44704f69SBart Van Assche char *cp;
292*44704f69SBart Van Assche FILE *fp;
293*44704f69SBart Van Assche const char *proc_devices = "/proc/devices";
294*44704f69SBart Van Assche char a[128];
295*44704f69SBart Van Assche char b[128];
296*44704f69SBart Van Assche
297*44704f69SBart Van Assche if (NULL == (fp = fopen(proc_devices, "r"))) {
298*44704f69SBart Van Assche if (verbose)
299*44704f69SBart Van Assche pr2serr("fopen %s failed: %s\n", proc_devices, strerror(errno));
300*44704f69SBart Van Assche return;
301*44704f69SBart Van Assche }
302*44704f69SBart Van Assche while ((cp = fgets(b, sizeof(b), fp))) {
303*44704f69SBart Van Assche if ((1 == sscanf(b, "%126s", a)) &&
304*44704f69SBart Van Assche (0 == memcmp(a, "Character", 9)))
305*44704f69SBart Van Assche break;
306*44704f69SBart Van Assche }
307*44704f69SBart Van Assche while (cp && (cp = fgets(b, sizeof(b), fp))) {
308*44704f69SBart Van Assche if (2 == sscanf(b, "%d %126s", &n, a)) {
309*44704f69SBart Van Assche if (0 == strcmp("bsg", a)) {
310*44704f69SBart Van Assche bsg_major = n;
311*44704f69SBart Van Assche break;
312*44704f69SBart Van Assche }
313*44704f69SBart Van Assche } else
314*44704f69SBart Van Assche break;
315*44704f69SBart Van Assche }
316*44704f69SBart Van Assche if (verbose > 5) {
317*44704f69SBart Van Assche if (cp)
318*44704f69SBart Van Assche pr2serr("found bsg_major=%d\n", bsg_major);
319*44704f69SBart Van Assche else
320*44704f69SBart Van Assche pr2serr("found no bsg char device in %s\n", proc_devices);
321*44704f69SBart Van Assche }
322*44704f69SBart Van Assche fclose(fp);
323*44704f69SBart Van Assche }
324*44704f69SBart Van Assche
325*44704f69SBart Van Assche static bool nvme_major_checked = false;
326*44704f69SBart Van Assche static int nvme_major = 0;
327*44704f69SBart Van Assche
328*44704f69SBart Van Assche static void
find_nvme_major(void)329*44704f69SBart Van Assche find_nvme_major(void)
330*44704f69SBart Van Assche {
331*44704f69SBart Van Assche int n;
332*44704f69SBart Van Assche char *cp;
333*44704f69SBart Van Assche FILE *fp;
334*44704f69SBart Van Assche const char *proc_devices = "/proc/devices";
335*44704f69SBart Van Assche char a[128];
336*44704f69SBart Van Assche char b[128];
337*44704f69SBart Van Assche
338*44704f69SBart Van Assche if (NULL == (fp = fopen(proc_devices, "r"))) {
339*44704f69SBart Van Assche if (verbose)
340*44704f69SBart Van Assche pr2serr("fopen %s failed: %s\n", proc_devices, strerror(errno));
341*44704f69SBart Van Assche return;
342*44704f69SBart Van Assche }
343*44704f69SBart Van Assche while ((cp = fgets(b, sizeof(b), fp))) {
344*44704f69SBart Van Assche if ((1 == sscanf(b, "%126s", a)) &&
345*44704f69SBart Van Assche (0 == memcmp(a, "Character", 9)))
346*44704f69SBart Van Assche break;
347*44704f69SBart Van Assche }
348*44704f69SBart Van Assche while (cp && (cp = fgets(b, sizeof(b), fp))) {
349*44704f69SBart Van Assche if (2 == sscanf(b, "%d %126s", &n, a)) {
350*44704f69SBart Van Assche if (0 == strcmp("nvme", a)) {
351*44704f69SBart Van Assche nvme_major = n;
352*44704f69SBart Van Assche break;
353*44704f69SBart Van Assche }
354*44704f69SBart Van Assche } else
355*44704f69SBart Van Assche break;
356*44704f69SBart Van Assche }
357*44704f69SBart Van Assche if (verbose > 5) {
358*44704f69SBart Van Assche if (cp)
359*44704f69SBart Van Assche pr2serr("found nvme_major=%d\n", bsg_major);
360*44704f69SBart Van Assche else
361*44704f69SBart Van Assche pr2serr("found no nvme char device in %s\n", proc_devices);
362*44704f69SBart Van Assche }
363*44704f69SBart Van Assche fclose(fp);
364*44704f69SBart Van Assche }
365*44704f69SBart Van Assche
366*44704f69SBart Van Assche
367*44704f69SBart Van Assche static int
dd_filetype(const char * filename)368*44704f69SBart Van Assche dd_filetype(const char * filename)
369*44704f69SBart Van Assche {
370*44704f69SBart Van Assche size_t len = strlen(filename);
371*44704f69SBart Van Assche struct stat st;
372*44704f69SBart Van Assche
373*44704f69SBart Van Assche if ((1 == len) && ('.' == filename[0]))
374*44704f69SBart Van Assche return FT_DEV_NULL;
375*44704f69SBart Van Assche if (stat(filename, &st) < 0)
376*44704f69SBart Van Assche return FT_ERROR;
377*44704f69SBart Van Assche if (S_ISCHR(st.st_mode)) {
378*44704f69SBart Van Assche /* major() and minor() defined in sys/sysmacros.h */
379*44704f69SBart Van Assche if ((MEM_MAJOR == major(st.st_rdev)) &&
380*44704f69SBart Van Assche (DEV_NULL_MINOR_NUM == minor(st.st_rdev)))
381*44704f69SBart Van Assche return FT_DEV_NULL;
382*44704f69SBart Van Assche if (RAW_MAJOR == major(st.st_rdev))
383*44704f69SBart Van Assche return FT_RAW;
384*44704f69SBart Van Assche if (SCSI_GENERIC_MAJOR == major(st.st_rdev))
385*44704f69SBart Van Assche return FT_SG;
386*44704f69SBart Van Assche if (SCSI_TAPE_MAJOR == major(st.st_rdev))
387*44704f69SBart Van Assche return FT_ST;
388*44704f69SBart Van Assche if (! bsg_major_checked) {
389*44704f69SBart Van Assche bsg_major_checked = true;
390*44704f69SBart Van Assche find_bsg_major();
391*44704f69SBart Van Assche }
392*44704f69SBart Van Assche if (bsg_major == (int)major(st.st_rdev))
393*44704f69SBart Van Assche return FT_SG;
394*44704f69SBart Van Assche if (! nvme_major_checked) {
395*44704f69SBart Van Assche nvme_major_checked = true;
396*44704f69SBart Van Assche find_nvme_major();
397*44704f69SBart Van Assche }
398*44704f69SBart Van Assche if (nvme_major == (int)major(st.st_rdev))
399*44704f69SBart Van Assche return FT_NVME; /* treat as sg device */
400*44704f69SBart Van Assche } else if (S_ISBLK(st.st_mode))
401*44704f69SBart Van Assche return FT_BLOCK;
402*44704f69SBart Van Assche else if (S_ISFIFO(st.st_mode))
403*44704f69SBart Van Assche return FT_FIFO;
404*44704f69SBart Van Assche return FT_OTHER;
405*44704f69SBart Van Assche }
406*44704f69SBart Van Assche
407*44704f69SBart Van Assche
408*44704f69SBart Van Assche static char *
dd_filetype_str(int ft,char * buff)409*44704f69SBart Van Assche dd_filetype_str(int ft, char * buff)
410*44704f69SBart Van Assche {
411*44704f69SBart Van Assche int off = 0;
412*44704f69SBart Van Assche
413*44704f69SBart Van Assche if (FT_DEV_NULL & ft)
414*44704f69SBart Van Assche off += sg_scnpr(buff + off, 32, "null device ");
415*44704f69SBart Van Assche if (FT_SG & ft)
416*44704f69SBart Van Assche off += sg_scnpr(buff + off, 32, "SCSI generic (sg) device ");
417*44704f69SBart Van Assche if (FT_BLOCK & ft)
418*44704f69SBart Van Assche off += sg_scnpr(buff + off, 32, "block device ");
419*44704f69SBart Van Assche if (FT_FIFO & ft)
420*44704f69SBart Van Assche off += sg_scnpr(buff + off, 32, "fifo (named pipe) ");
421*44704f69SBart Van Assche if (FT_ST & ft)
422*44704f69SBart Van Assche off += sg_scnpr(buff + off, 32, "SCSI tape device ");
423*44704f69SBart Van Assche if (FT_RAW & ft)
424*44704f69SBart Van Assche off += sg_scnpr(buff + off, 32, "raw device ");
425*44704f69SBart Van Assche if (FT_NVME & ft)
426*44704f69SBart Van Assche off += sg_scnpr(buff + off, 32, "NVMe char device ");
427*44704f69SBart Van Assche if (FT_OTHER & ft)
428*44704f69SBart Van Assche off += sg_scnpr(buff + off, 32, "other (perhaps ordinary file) ");
429*44704f69SBart Van Assche if (FT_ERROR & ft)
430*44704f69SBart Van Assche sg_scnpr(buff + off, 32, "unable to 'stat' file ");
431*44704f69SBart Van Assche return buff;
432*44704f69SBart Van Assche }
433*44704f69SBart Van Assche
434*44704f69SBart Van Assche
435*44704f69SBart Van Assche static void
usage()436*44704f69SBart Van Assche usage()
437*44704f69SBart Van Assche {
438*44704f69SBart Van Assche pr2serr("Usage: sg_dd [bs=BS] [conv=CONV] [count=COUNT] [ibs=BS] "
439*44704f69SBart Van Assche "[if=IFILE]\n"
440*44704f69SBart Van Assche " [iflag=FLAGS] [obs=BS] [of=OFILE] [oflag=FLAGS] "
441*44704f69SBart Van Assche "[seek=SEEK]\n"
442*44704f69SBart Van Assche " [skip=SKIP] [--dry-run] [--help] [--verbose] "
443*44704f69SBart Van Assche "[--version]\n\n"
444*44704f69SBart Van Assche " [blk_sgio=0|1] [bpt=BPT] [cdbsz=6|10|12|16] "
445*44704f69SBart Van Assche "[cdl=CDL]\n"
446*44704f69SBart Van Assche " [coe=0|1|2|3] [coe_limit=CL] [dio=0|1] "
447*44704f69SBart Van Assche "[odir=0|1]\n"
448*44704f69SBart Van Assche " [of2=OFILE2] [retries=RETR] [sync=0|1] "
449*44704f69SBart Van Assche "[time=0|1[,TO]]\n"
450*44704f69SBart Van Assche " [verbose=VERB] [--progress] [--verify]\n"
451*44704f69SBart Van Assche " where:\n"
452*44704f69SBart Van Assche " blk_sgio 0->block device use normal I/O(def), 1->use "
453*44704f69SBart Van Assche "SG_IO\n"
454*44704f69SBart Van Assche " bpt is blocks_per_transfer (default is 128 or 32 "
455*44704f69SBart Van Assche "when BS>=2048)\n"
456*44704f69SBart Van Assche " bs logical block size (default is 512)\n");
457*44704f69SBart Van Assche pr2serr(" cdbsz size of SCSI READ or WRITE cdb (default is "
458*44704f69SBart Van Assche "10)\n"
459*44704f69SBart Van Assche " cdl command duration limits value 0 to 7 (def: "
460*44704f69SBart Van Assche "0 (no cdl))\n"
461*44704f69SBart Van Assche " coe 0->exit on error (def), 1->continue on sg "
462*44704f69SBart Van Assche "error (zero\n"
463*44704f69SBart Van Assche " fill), 2->also try read_long on unrecovered "
464*44704f69SBart Van Assche "reads,\n"
465*44704f69SBart Van Assche " 3->and set the CORRCT bit on the read long\n"
466*44704f69SBart Van Assche " coe_limit limit consecutive 'bad' blocks on reads to CL "
467*44704f69SBart Van Assche "times\n"
468*44704f69SBart Van Assche " when COE>1 (default: 0 which is no limit)\n"
469*44704f69SBart Van Assche " conv comma separated list from: [nocreat,noerror,"
470*44704f69SBart Van Assche "notrunc,\n"
471*44704f69SBart Van Assche " null,sparse,sync]\n"
472*44704f69SBart Van Assche " count number of blocks to copy (def: device size)\n"
473*44704f69SBart Van Assche " dio for direct IO, 1->attempt, 0->indirect IO "
474*44704f69SBart Van Assche "(def)\n"
475*44704f69SBart Van Assche " ibs input logical block size (if given must be same "
476*44704f69SBart Van Assche "as 'bs=')\n"
477*44704f69SBart Van Assche " if file or device to read from (def: stdin)\n"
478*44704f69SBart Van Assche " iflag comma separated list from: [00,coe,dio,direct,"
479*44704f69SBart Van Assche "dpo,dsync,\n"
480*44704f69SBart Van Assche " excl,ff,flock,fua,nocache,null,random,sgio]\n"
481*44704f69SBart Van Assche " obs output logical block size (if given must be "
482*44704f69SBart Van Assche "same as 'bs=')\n"
483*44704f69SBart Van Assche " odir 1->use O_DIRECT when opening block dev, "
484*44704f69SBart Van Assche "0->don't(def)\n"
485*44704f69SBart Van Assche " of file or device to write to (def: stdout), "
486*44704f69SBart Van Assche "OFILE of '.'\n");
487*44704f69SBart Van Assche pr2serr(" treated as /dev/null\n"
488*44704f69SBart Van Assche " of2 additional output file (def: /dev/null), "
489*44704f69SBart Van Assche "OFILE2 should be\n"
490*44704f69SBart Van Assche " normal file or pipe\n"
491*44704f69SBart Van Assche " oflag comma separated list from: [append,coe,dio,"
492*44704f69SBart Van Assche "direct,dpo,\n"
493*44704f69SBart Van Assche " dsync,excl,flock,fua,nocache,nocreat,null,sgio,"
494*44704f69SBart Van Assche "sparse]\n"
495*44704f69SBart Van Assche " retries retry sgio errors RETR times (def: 0)\n"
496*44704f69SBart Van Assche " seek block position to start writing to OFILE\n"
497*44704f69SBart Van Assche " skip block position to start reading from IFILE\n"
498*44704f69SBart Van Assche " sync 0->no sync(def), 1->SYNCHRONIZE CACHE on "
499*44704f69SBart Van Assche "OFILE after copy\n"
500*44704f69SBart Van Assche " time 0->no timing(def), 1->time plus calculate "
501*44704f69SBart Van Assche "throughput;\n"
502*44704f69SBart Van Assche " TO is command timeout in seconds (def: 60)\n"
503*44704f69SBart Van Assche " verbose 0->quiet(def), 1->some noise, 2->more noise, "
504*44704f69SBart Van Assche "etc\n"
505*44704f69SBart Van Assche " --dry-run do preparation but bypass copy (or read)\n"
506*44704f69SBart Van Assche " --help|-h print out this usage message then exit\n"
507*44704f69SBart Van Assche " --progress|-p print progress report every 2 minutes\n"
508*44704f69SBart Van Assche " --verbose|-v same as 'verbose=1', can be used multiple "
509*44704f69SBart Van Assche "times\n"
510*44704f69SBart Van Assche " --verify|-x do verify/compare rather than copy "
511*44704f69SBart Van Assche "(OFILE must\n"
512*44704f69SBart Van Assche " be a sg device)\n"
513*44704f69SBart Van Assche " --version|-V print version information then exit\n\n"
514*44704f69SBart Van Assche "Copy from IFILE to OFILE, similar to dd command; specialized "
515*44704f69SBart Van Assche "for SCSI\ndevices. If the --verify option is given then IFILE "
516*44704f69SBart Van Assche "is read and that data\nis used to compare with OFILE using "
517*44704f69SBart Van Assche "the VERIFY(n) SCSI command (with\nBYTCHK=1).\n");
518*44704f69SBart Van Assche }
519*44704f69SBart Van Assche
520*44704f69SBart Van Assche
521*44704f69SBart Van Assche /* Return of 0 -> success, see sg_ll_read_capacity*() otherwise */
522*44704f69SBart Van Assche static int
scsi_read_capacity(int sg_fd,int64_t * num_sect,int * sect_sz)523*44704f69SBart Van Assche scsi_read_capacity(int sg_fd, int64_t * num_sect, int * sect_sz)
524*44704f69SBart Van Assche {
525*44704f69SBart Van Assche int res, verb;
526*44704f69SBart Van Assche unsigned int ui;
527*44704f69SBart Van Assche uint8_t rcBuff[RCAP16_REPLY_LEN];
528*44704f69SBart Van Assche
529*44704f69SBart Van Assche verb = (verbose ? verbose - 1: 0);
530*44704f69SBart Van Assche res = sg_ll_readcap_10(sg_fd, false, 0, rcBuff, READ_CAP_REPLY_LEN, true,
531*44704f69SBart Van Assche verb);
532*44704f69SBart Van Assche if (0 != res)
533*44704f69SBart Van Assche return res;
534*44704f69SBart Van Assche
535*44704f69SBart Van Assche if ((0xff == rcBuff[0]) && (0xff == rcBuff[1]) && (0xff == rcBuff[2]) &&
536*44704f69SBart Van Assche (0xff == rcBuff[3])) {
537*44704f69SBart Van Assche int64_t ls;
538*44704f69SBart Van Assche
539*44704f69SBart Van Assche res = sg_ll_readcap_16(sg_fd, false, 0, rcBuff, RCAP16_REPLY_LEN,
540*44704f69SBart Van Assche true, verb);
541*44704f69SBart Van Assche if (0 != res)
542*44704f69SBart Van Assche return res;
543*44704f69SBart Van Assche ls = (int64_t)sg_get_unaligned_be64(rcBuff);
544*44704f69SBart Van Assche *num_sect = ls + 1;
545*44704f69SBart Van Assche *sect_sz = (int)sg_get_unaligned_be32(rcBuff + 8);
546*44704f69SBart Van Assche } else {
547*44704f69SBart Van Assche ui = sg_get_unaligned_be32(rcBuff);
548*44704f69SBart Van Assche /* take care not to sign extend values > 0x7fffffff */
549*44704f69SBart Van Assche *num_sect = (int64_t)ui + 1;
550*44704f69SBart Van Assche *sect_sz = (int)sg_get_unaligned_be32(rcBuff + 4);
551*44704f69SBart Van Assche }
552*44704f69SBart Van Assche if (verbose)
553*44704f69SBart Van Assche pr2serr(" number of blocks=%" PRId64 " [0x%" PRIx64 "], "
554*44704f69SBart Van Assche "logical block size=%d\n", *num_sect, *num_sect, *sect_sz);
555*44704f69SBart Van Assche return 0;
556*44704f69SBart Van Assche }
557*44704f69SBart Van Assche
558*44704f69SBart Van Assche
559*44704f69SBart Van Assche /* Return of 0 -> success, -1 -> failure. BLKGETSIZE64, BLKGETSIZE and */
560*44704f69SBart Van Assche /* BLKSSZGET macros problematic (from <linux/fs.h> or <sys/mount.h>). */
561*44704f69SBart Van Assche static int
read_blkdev_capacity(int sg_fd,int64_t * num_sect,int * sect_sz)562*44704f69SBart Van Assche read_blkdev_capacity(int sg_fd, int64_t * num_sect, int * sect_sz)
563*44704f69SBart Van Assche {
564*44704f69SBart Van Assche #ifdef BLKSSZGET
565*44704f69SBart Van Assche if ((ioctl(sg_fd, BLKSSZGET, sect_sz) < 0) && (*sect_sz > 0)) {
566*44704f69SBart Van Assche perror("BLKSSZGET ioctl error");
567*44704f69SBart Van Assche return -1;
568*44704f69SBart Van Assche } else {
569*44704f69SBart Van Assche #ifdef BLKGETSIZE64
570*44704f69SBart Van Assche uint64_t ull;
571*44704f69SBart Van Assche
572*44704f69SBart Van Assche if (ioctl(sg_fd, BLKGETSIZE64, &ull) < 0) {
573*44704f69SBart Van Assche
574*44704f69SBart Van Assche perror("BLKGETSIZE64 ioctl error");
575*44704f69SBart Van Assche return -1;
576*44704f69SBart Van Assche }
577*44704f69SBart Van Assche *num_sect = ((int64_t)ull / (int64_t)*sect_sz);
578*44704f69SBart Van Assche if (verbose)
579*44704f69SBart Van Assche pr2serr(" [bgs64] number of blocks=%" PRId64 " [0x%" PRIx64
580*44704f69SBart Van Assche "], logical block size=%d\n", *num_sect, *num_sect,
581*44704f69SBart Van Assche *sect_sz);
582*44704f69SBart Van Assche #else
583*44704f69SBart Van Assche unsigned long ul;
584*44704f69SBart Van Assche
585*44704f69SBart Van Assche if (ioctl(sg_fd, BLKGETSIZE, &ul) < 0) {
586*44704f69SBart Van Assche perror("BLKGETSIZE ioctl error");
587*44704f69SBart Van Assche return -1;
588*44704f69SBart Van Assche }
589*44704f69SBart Van Assche *num_sect = (int64_t)ul;
590*44704f69SBart Van Assche if (verbose)
591*44704f69SBart Van Assche pr2serr(" [bgs] number of blocks=%" PRId64 " [0x%" PRIx64
592*44704f69SBart Van Assche "], logical block size=%d\n", *num_sect, *num_sect,
593*44704f69SBart Van Assche *sect_sz);
594*44704f69SBart Van Assche #endif
595*44704f69SBart Van Assche }
596*44704f69SBart Van Assche return 0;
597*44704f69SBart Van Assche #else
598*44704f69SBart Van Assche if (verbose)
599*44704f69SBart Van Assche pr2serr(" BLKSSZGET+BLKGETSIZE ioctl not available\n");
600*44704f69SBart Van Assche *num_sect = 0;
601*44704f69SBart Van Assche *sect_sz = 0;
602*44704f69SBart Van Assche return -1;
603*44704f69SBart Van Assche #endif
604*44704f69SBart Van Assche }
605*44704f69SBart Van Assche
606*44704f69SBart Van Assche
607*44704f69SBart Van Assche static int
sg_build_scsi_cdb(uint8_t * cdbp,int cdb_sz,unsigned int blocks,int64_t start_block,bool is_verify,bool write_true,bool fua,bool dpo,int cdl)608*44704f69SBart Van Assche sg_build_scsi_cdb(uint8_t * cdbp, int cdb_sz, unsigned int blocks,
609*44704f69SBart Van Assche int64_t start_block, bool is_verify, bool write_true,
610*44704f69SBart Van Assche bool fua, bool dpo, int cdl)
611*44704f69SBart Van Assche {
612*44704f69SBart Van Assche int sz_ind;
613*44704f69SBart Van Assche int rd_opcode[] = {0x8, 0x28, 0xa8, 0x88};
614*44704f69SBart Van Assche int ve_opcode[] = {0xff /* no VERIFY(6) */, VERIFY10, VERIFY12, VERIFY16};
615*44704f69SBart Van Assche int wr_opcode[] = {0xa, 0x2a, 0xaa, 0x8a};
616*44704f69SBart Van Assche
617*44704f69SBart Van Assche memset(cdbp, 0, cdb_sz);
618*44704f69SBart Van Assche if (is_verify)
619*44704f69SBart Van Assche cdbp[1] = 0x2; /* (BYTCHK=1) << 1 */
620*44704f69SBart Van Assche else {
621*44704f69SBart Van Assche if (dpo)
622*44704f69SBart Van Assche cdbp[1] |= 0x10;
623*44704f69SBart Van Assche if (fua)
624*44704f69SBart Van Assche cdbp[1] |= 0x8;
625*44704f69SBart Van Assche }
626*44704f69SBart Van Assche switch (cdb_sz) {
627*44704f69SBart Van Assche case 6:
628*44704f69SBart Van Assche sz_ind = 0;
629*44704f69SBart Van Assche if (is_verify && write_true) {
630*44704f69SBart Van Assche pr2serr(ME "there is no VERIFY(6), choose a larger cdbsz\n");
631*44704f69SBart Van Assche return 1;
632*44704f69SBart Van Assche }
633*44704f69SBart Van Assche cdbp[0] = (uint8_t)(write_true ? wr_opcode[sz_ind] :
634*44704f69SBart Van Assche rd_opcode[sz_ind]);
635*44704f69SBart Van Assche sg_put_unaligned_be24(0x1fffff & start_block, cdbp + 1);
636*44704f69SBart Van Assche cdbp[4] = (256 == blocks) ? 0 : (uint8_t)blocks;
637*44704f69SBart Van Assche if (blocks > 256) {
638*44704f69SBart Van Assche pr2serr(ME "for 6 byte commands, maximum number of blocks is "
639*44704f69SBart Van Assche "256\n");
640*44704f69SBart Van Assche return 1;
641*44704f69SBart Van Assche }
642*44704f69SBart Van Assche if ((start_block + blocks - 1) & (~0x1fffff)) {
643*44704f69SBart Van Assche pr2serr(ME "for 6 byte commands, can't address blocks beyond "
644*44704f69SBart Van Assche "%d\n", 0x1fffff);
645*44704f69SBart Van Assche return 1;
646*44704f69SBart Van Assche }
647*44704f69SBart Van Assche if (dpo || fua) {
648*44704f69SBart Van Assche pr2serr(ME "for 6 byte commands, neither dpo nor fua bits "
649*44704f69SBart Van Assche "supported\n");
650*44704f69SBart Van Assche return 1;
651*44704f69SBart Van Assche }
652*44704f69SBart Van Assche break;
653*44704f69SBart Van Assche case 10:
654*44704f69SBart Van Assche sz_ind = 1;
655*44704f69SBart Van Assche if (is_verify && write_true)
656*44704f69SBart Van Assche cdbp[0] = ve_opcode[sz_ind];
657*44704f69SBart Van Assche else
658*44704f69SBart Van Assche cdbp[0] = (uint8_t)(write_true ? wr_opcode[sz_ind] :
659*44704f69SBart Van Assche rd_opcode[sz_ind]);
660*44704f69SBart Van Assche sg_put_unaligned_be32(start_block, cdbp + 2);
661*44704f69SBart Van Assche sg_put_unaligned_be16(blocks, cdbp + 7);
662*44704f69SBart Van Assche if (blocks & (~0xffff)) {
663*44704f69SBart Van Assche pr2serr(ME "for 10 byte commands, maximum number of blocks is "
664*44704f69SBart Van Assche "%d\n", 0xffff);
665*44704f69SBart Van Assche return 1;
666*44704f69SBart Van Assche }
667*44704f69SBart Van Assche break;
668*44704f69SBart Van Assche case 12:
669*44704f69SBart Van Assche sz_ind = 2;
670*44704f69SBart Van Assche if (is_verify && write_true)
671*44704f69SBart Van Assche cdbp[0] = ve_opcode[sz_ind];
672*44704f69SBart Van Assche else
673*44704f69SBart Van Assche cdbp[0] = (uint8_t)(write_true ? wr_opcode[sz_ind] :
674*44704f69SBart Van Assche rd_opcode[sz_ind]);
675*44704f69SBart Van Assche sg_put_unaligned_be32(start_block, cdbp + 2);
676*44704f69SBart Van Assche sg_put_unaligned_be32(blocks, cdbp + 6);
677*44704f69SBart Van Assche break;
678*44704f69SBart Van Assche case 16:
679*44704f69SBart Van Assche sz_ind = 3;
680*44704f69SBart Van Assche if (is_verify && write_true)
681*44704f69SBart Van Assche cdbp[0] = ve_opcode[sz_ind];
682*44704f69SBart Van Assche else
683*44704f69SBart Van Assche cdbp[0] = (uint8_t)(write_true ? wr_opcode[sz_ind] :
684*44704f69SBart Van Assche rd_opcode[sz_ind]);
685*44704f69SBart Van Assche if ((! is_verify) && (cdl > 0)) {
686*44704f69SBart Van Assche if (cdl & 0x4)
687*44704f69SBart Van Assche cdbp[1] |= 0x1;
688*44704f69SBart Van Assche if (cdl & 0x3)
689*44704f69SBart Van Assche cdbp[14] |= ((cdl & 0x3) << 6);
690*44704f69SBart Van Assche }
691*44704f69SBart Van Assche sg_put_unaligned_be64(start_block, cdbp + 2);
692*44704f69SBart Van Assche sg_put_unaligned_be32(blocks, cdbp + 10);
693*44704f69SBart Van Assche break;
694*44704f69SBart Van Assche default:
695*44704f69SBart Van Assche pr2serr(ME "expected cdb size of 6, 10, 12, or 16 but got %d\n",
696*44704f69SBart Van Assche cdb_sz);
697*44704f69SBart Van Assche return 1;
698*44704f69SBart Van Assche }
699*44704f69SBart Van Assche return 0;
700*44704f69SBart Van Assche }
701*44704f69SBart Van Assche
702*44704f69SBart Van Assche
703*44704f69SBart Van Assche /* Does SCSI READ on IFILE. Returns 0 -> successful,
704*44704f69SBart Van Assche * SG_LIB_SYNTAX_ERROR -> unable to build cdb,
705*44704f69SBart Van Assche * SG_LIB_CAT_UNIT_ATTENTION -> try again,
706*44704f69SBart Van Assche * SG_LIB_CAT_MEDIUM_HARD_WITH_INFO -> 'io_addrp' written to,
707*44704f69SBart Van Assche * SG_LIB_CAT_MEDIUM_HARD -> no info field,
708*44704f69SBart Van Assche * SG_LIB_CAT_NOT_READY, SG_LIB_CAT_ABORTED_COMMAND,
709*44704f69SBart Van Assche * -2 -> ENOMEM, -1 other errors */
710*44704f69SBart Van Assche static int
sg_read_low(int sg_fd,uint8_t * buff,int blocks,int64_t from_block,int bs,const struct flags_t * ifp,bool * diop,uint64_t * io_addrp)711*44704f69SBart Van Assche sg_read_low(int sg_fd, uint8_t * buff, int blocks, int64_t from_block,
712*44704f69SBart Van Assche int bs, const struct flags_t * ifp, bool * diop,
713*44704f69SBart Van Assche uint64_t * io_addrp)
714*44704f69SBart Van Assche {
715*44704f69SBart Van Assche bool info_valid;
716*44704f69SBart Van Assche bool print_cdb_after = false;
717*44704f69SBart Van Assche int res, slen;
718*44704f69SBart Van Assche const uint8_t * sbp;
719*44704f69SBart Van Assche uint8_t rdCmd[MAX_SCSI_CDBSZ];
720*44704f69SBart Van Assche uint8_t senseBuff[SENSE_BUFF_LEN] SG_C_CPP_ZERO_INIT;
721*44704f69SBart Van Assche struct sg_io_hdr io_hdr;
722*44704f69SBart Van Assche
723*44704f69SBart Van Assche if (sg_build_scsi_cdb(rdCmd, ifp->cdbsz, blocks, from_block, do_verify,
724*44704f69SBart Van Assche false, ifp->fua, ifp->dpo, ifp->cdl)) {
725*44704f69SBart Van Assche pr2serr(ME "bad rd cdb build, from_block=%" PRId64 ", blocks=%d\n",
726*44704f69SBart Van Assche from_block, blocks);
727*44704f69SBart Van Assche return SG_LIB_SYNTAX_ERROR;
728*44704f69SBart Van Assche }
729*44704f69SBart Van Assche
730*44704f69SBart Van Assche memset(&io_hdr, 0, sizeof(struct sg_io_hdr));
731*44704f69SBart Van Assche io_hdr.interface_id = 'S';
732*44704f69SBart Van Assche io_hdr.cmd_len = ifp->cdbsz;
733*44704f69SBart Van Assche io_hdr.cmdp = rdCmd;
734*44704f69SBart Van Assche io_hdr.dxfer_direction = SG_DXFER_FROM_DEV;
735*44704f69SBart Van Assche io_hdr.dxfer_len = bs * blocks;
736*44704f69SBart Van Assche io_hdr.dxferp = buff;
737*44704f69SBart Van Assche io_hdr.mx_sb_len = SENSE_BUFF_LEN;
738*44704f69SBart Van Assche io_hdr.sbp = senseBuff;
739*44704f69SBart Van Assche io_hdr.timeout = cmd_timeout;
740*44704f69SBart Van Assche io_hdr.pack_id = (int)++glob_pack_id;
741*44704f69SBart Van Assche if (diop && *diop)
742*44704f69SBart Van Assche io_hdr.flags |= SG_FLAG_DIRECT_IO;
743*44704f69SBart Van Assche
744*44704f69SBart Van Assche if (verbose > 2)
745*44704f69SBart Van Assche sg_print_command_len(rdCmd, ifp->cdbsz);
746*44704f69SBart Van Assche
747*44704f69SBart Van Assche while (((res = ioctl(sg_fd, SG_IO, &io_hdr)) < 0) &&
748*44704f69SBart Van Assche ((EINTR == errno) || (EAGAIN == errno) || (EBUSY == errno)))
749*44704f69SBart Van Assche ;
750*44704f69SBart Van Assche if (res < 0) {
751*44704f69SBart Van Assche if (ENOMEM == errno)
752*44704f69SBart Van Assche return -2;
753*44704f69SBart Van Assche perror("reading (SG_IO) on sg device, error");
754*44704f69SBart Van Assche return -1;
755*44704f69SBart Van Assche }
756*44704f69SBart Van Assche if (verbose > 2)
757*44704f69SBart Van Assche pr2serr(" duration=%u ms\n", io_hdr.duration);
758*44704f69SBart Van Assche res = sg_err_category3(&io_hdr);
759*44704f69SBart Van Assche sbp = io_hdr.sbp;
760*44704f69SBart Van Assche slen = io_hdr.sb_len_wr;
761*44704f69SBart Van Assche switch (res) {
762*44704f69SBart Van Assche case SG_LIB_CAT_CLEAN:
763*44704f69SBart Van Assche case SG_LIB_CAT_CONDITION_MET:
764*44704f69SBart Van Assche break;
765*44704f69SBart Van Assche case SG_LIB_CAT_RECOVERED:
766*44704f69SBart Van Assche ++recovered_errs;
767*44704f69SBart Van Assche info_valid = sg_get_sense_info_fld(sbp, slen, io_addrp);
768*44704f69SBart Van Assche if (info_valid) {
769*44704f69SBart Van Assche pr2serr(" lba of last recovered error in this READ=0x%" PRIx64
770*44704f69SBart Van Assche "\n", *io_addrp);
771*44704f69SBart Van Assche if (verbose > 1)
772*44704f69SBart Van Assche sg_chk_n_print3("reading", &io_hdr, true);
773*44704f69SBart Van Assche } else {
774*44704f69SBart Van Assche pr2serr("Recovered error: [no info] reading from block=0x%" PRIx64
775*44704f69SBart Van Assche ", num=%d\n", from_block, blocks);
776*44704f69SBart Van Assche sg_chk_n_print3("reading", &io_hdr, verbose > 1);
777*44704f69SBart Van Assche }
778*44704f69SBart Van Assche break;
779*44704f69SBart Van Assche case SG_LIB_CAT_ABORTED_COMMAND:
780*44704f69SBart Van Assche case SG_LIB_CAT_UNIT_ATTENTION:
781*44704f69SBart Van Assche sg_chk_n_print3("reading", &io_hdr, verbose > 1);
782*44704f69SBart Van Assche return res;
783*44704f69SBart Van Assche case SG_LIB_CAT_MEDIUM_HARD:
784*44704f69SBart Van Assche if (verbose > 1)
785*44704f69SBart Van Assche sg_chk_n_print3("reading", &io_hdr, verbose > 1);
786*44704f69SBart Van Assche ++unrecovered_errs;
787*44704f69SBart Van Assche info_valid = sg_get_sense_info_fld(sbp, slen, io_addrp);
788*44704f69SBart Van Assche /* MMC devices don't necessarily set VALID bit */
789*44704f69SBart Van Assche if (info_valid || ((5 == ifp->pdt) && (*io_addrp > 0)))
790*44704f69SBart Van Assche return SG_LIB_CAT_MEDIUM_HARD_WITH_INFO;
791*44704f69SBart Van Assche else {
792*44704f69SBart Van Assche pr2serr("Medium, hardware or blank check error but no lba of "
793*44704f69SBart Van Assche "failure in sense\n");
794*44704f69SBart Van Assche return res;
795*44704f69SBart Van Assche }
796*44704f69SBart Van Assche break;
797*44704f69SBart Van Assche case SG_LIB_CAT_NOT_READY:
798*44704f69SBart Van Assche ++unrecovered_errs;
799*44704f69SBart Van Assche if (verbose > 0)
800*44704f69SBart Van Assche sg_chk_n_print3("reading", &io_hdr, verbose > 1);
801*44704f69SBart Van Assche return res;
802*44704f69SBart Van Assche case SG_LIB_CAT_ILLEGAL_REQ:
803*44704f69SBart Van Assche if (5 == ifp->pdt) { /* MMC READs can go down this path */
804*44704f69SBart Van Assche bool ili;
805*44704f69SBart Van Assche struct sg_scsi_sense_hdr ssh;
806*44704f69SBart Van Assche
807*44704f69SBart Van Assche if (verbose > 1)
808*44704f69SBart Van Assche sg_chk_n_print3("reading", &io_hdr, verbose > 1);
809*44704f69SBart Van Assche if (sg_scsi_normalize_sense(sbp, slen, &ssh) &&
810*44704f69SBart Van Assche (0x64 == ssh.asc) && (0x0 == ssh.ascq)) {
811*44704f69SBart Van Assche if (sg_get_sense_filemark_eom_ili(sbp, slen, NULL, NULL,
812*44704f69SBart Van Assche &ili) && ili) {
813*44704f69SBart Van Assche sg_get_sense_info_fld(sbp, slen, io_addrp);
814*44704f69SBart Van Assche if (*io_addrp > 0) {
815*44704f69SBart Van Assche ++unrecovered_errs;
816*44704f69SBart Van Assche return SG_LIB_CAT_MEDIUM_HARD_WITH_INFO;
817*44704f69SBart Van Assche } else
818*44704f69SBart Van Assche pr2serr("MMC READ gave 'illegal mode for this track' "
819*44704f69SBart Van Assche "and ILI but no LBA of failure\n");
820*44704f69SBart Van Assche }
821*44704f69SBart Van Assche ++unrecovered_errs;
822*44704f69SBart Van Assche return SG_LIB_CAT_MEDIUM_HARD;
823*44704f69SBart Van Assche }
824*44704f69SBart Van Assche }
825*44704f69SBart Van Assche if (verbose > 0)
826*44704f69SBart Van Assche print_cdb_after = true;
827*44704f69SBart Van Assche #if defined(__GNUC__)
828*44704f69SBart Van Assche #if (__GNUC__ >= 7)
829*44704f69SBart Van Assche __attribute__((fallthrough));
830*44704f69SBart Van Assche /* FALL THROUGH */
831*44704f69SBart Van Assche #endif
832*44704f69SBart Van Assche #endif
833*44704f69SBart Van Assche default:
834*44704f69SBart Van Assche ++unrecovered_errs;
835*44704f69SBart Van Assche if (verbose > 0)
836*44704f69SBart Van Assche sg_chk_n_print3("reading", &io_hdr, verbose > 1);
837*44704f69SBart Van Assche if (print_cdb_after)
838*44704f69SBart Van Assche sg_print_command_len(rdCmd, ifp->cdbsz);
839*44704f69SBart Van Assche return res;
840*44704f69SBart Van Assche }
841*44704f69SBart Van Assche if (diop && *diop &&
842*44704f69SBart Van Assche ((io_hdr.info & SG_INFO_DIRECT_IO_MASK) != SG_INFO_DIRECT_IO))
843*44704f69SBart Van Assche *diop = false; /* flag that dio not done (completely) */
844*44704f69SBart Van Assche sum_of_resids += io_hdr.resid;
845*44704f69SBart Van Assche return 0;
846*44704f69SBart Van Assche }
847*44704f69SBart Van Assche
848*44704f69SBart Van Assche
849*44704f69SBart Van Assche /* Does repeats associated with a SCSI READ on IFILE. Returns 0 -> successful,
850*44704f69SBart Van Assche * SG_LIB_SYNTAX_ERROR -> unable to build cdb, SG_LIB_CAT_UNIT_ATTENTION ->
851*44704f69SBart Van Assche * try again, SG_LIB_CAT_NOT_READY, SG_LIB_CAT_MEDIUM_HARD,
852*44704f69SBart Van Assche * SG_LIB_CAT_ABORTED_COMMAND, -2 -> ENOMEM, -1 other errors */
853*44704f69SBart Van Assche static int
sg_read(int sg_fd,uint8_t * buff,int blocks,int64_t from_block,int bs,struct flags_t * ifp,bool * diop,int * blks_readp)854*44704f69SBart Van Assche sg_read(int sg_fd, uint8_t * buff, int blocks, int64_t from_block,
855*44704f69SBart Van Assche int bs, struct flags_t * ifp, bool * diop, int * blks_readp)
856*44704f69SBart Van Assche {
857*44704f69SBart Van Assche bool may_coe = false;
858*44704f69SBart Van Assche bool repeat;
859*44704f69SBart Van Assche int res, blks, xferred;
860*44704f69SBart Van Assche int ret = 0;
861*44704f69SBart Van Assche int retries_tmp;
862*44704f69SBart Van Assche uint64_t io_addr;
863*44704f69SBart Van Assche int64_t lba;
864*44704f69SBart Van Assche uint8_t * bp;
865*44704f69SBart Van Assche
866*44704f69SBart Van Assche retries_tmp = ifp->retries;
867*44704f69SBart Van Assche for (xferred = 0, blks = blocks, lba = from_block, bp = buff;
868*44704f69SBart Van Assche blks > 0; blks = blocks - xferred) {
869*44704f69SBart Van Assche io_addr = 0;
870*44704f69SBart Van Assche repeat = false;
871*44704f69SBart Van Assche may_coe = false;
872*44704f69SBart Van Assche res = sg_read_low(sg_fd, bp, blks, lba, bs, ifp, diop, &io_addr);
873*44704f69SBart Van Assche switch (res) {
874*44704f69SBart Van Assche case 0:
875*44704f69SBart Van Assche if (blks_readp)
876*44704f69SBart Van Assche *blks_readp = xferred + blks;
877*44704f69SBart Van Assche if (coe_limit > 0)
878*44704f69SBart Van Assche coe_count = 0; /* good read clears coe_count */
879*44704f69SBart Van Assche return 0;
880*44704f69SBart Van Assche case -2: /* ENOMEM */
881*44704f69SBart Van Assche return res;
882*44704f69SBart Van Assche case SG_LIB_CAT_NOT_READY:
883*44704f69SBart Van Assche pr2serr("Device (r) not ready\n");
884*44704f69SBart Van Assche return res;
885*44704f69SBart Van Assche case SG_LIB_CAT_ABORTED_COMMAND:
886*44704f69SBart Van Assche if (--max_aborted > 0) {
887*44704f69SBart Van Assche pr2serr("Aborted command, continuing (r)\n");
888*44704f69SBart Van Assche repeat = true;
889*44704f69SBart Van Assche } else {
890*44704f69SBart Van Assche pr2serr("Aborted command, too many (r)\n");
891*44704f69SBart Van Assche return res;
892*44704f69SBart Van Assche }
893*44704f69SBart Van Assche break;
894*44704f69SBart Van Assche case SG_LIB_CAT_UNIT_ATTENTION:
895*44704f69SBart Van Assche if (--max_uas > 0) {
896*44704f69SBart Van Assche pr2serr("Unit attention, continuing (r)\n");
897*44704f69SBart Van Assche repeat = true;
898*44704f69SBart Van Assche } else {
899*44704f69SBart Van Assche pr2serr("Unit attention, too many (r)\n");
900*44704f69SBart Van Assche return res;
901*44704f69SBart Van Assche }
902*44704f69SBart Van Assche break;
903*44704f69SBart Van Assche case SG_LIB_CAT_MEDIUM_HARD_WITH_INFO:
904*44704f69SBart Van Assche if (retries_tmp > 0) {
905*44704f69SBart Van Assche pr2serr(">>> retrying a sgio read, lba=0x%" PRIx64 "\n",
906*44704f69SBart Van Assche (uint64_t)lba);
907*44704f69SBart Van Assche --retries_tmp;
908*44704f69SBart Van Assche ++num_retries;
909*44704f69SBart Van Assche if (unrecovered_errs > 0)
910*44704f69SBart Van Assche --unrecovered_errs;
911*44704f69SBart Van Assche repeat = true;
912*44704f69SBart Van Assche }
913*44704f69SBart Van Assche ret = SG_LIB_CAT_MEDIUM_HARD;
914*44704f69SBart Van Assche break; /* unrecovered read error at lba=io_addr */
915*44704f69SBart Van Assche case SG_LIB_SYNTAX_ERROR:
916*44704f69SBart Van Assche ifp->coe = 0;
917*44704f69SBart Van Assche ret = res;
918*44704f69SBart Van Assche goto err_out;
919*44704f69SBart Van Assche case -1:
920*44704f69SBart Van Assche ret = res;
921*44704f69SBart Van Assche goto err_out;
922*44704f69SBart Van Assche case SG_LIB_CAT_MEDIUM_HARD:
923*44704f69SBart Van Assche may_coe = true;
924*44704f69SBart Van Assche #if defined(__GNUC__)
925*44704f69SBart Van Assche #if (__GNUC__ >= 7)
926*44704f69SBart Van Assche __attribute__((fallthrough));
927*44704f69SBart Van Assche /* FALL THROUGH */
928*44704f69SBart Van Assche #endif
929*44704f69SBart Van Assche #endif
930*44704f69SBart Van Assche default:
931*44704f69SBart Van Assche if (retries_tmp > 0) {
932*44704f69SBart Van Assche pr2serr(">>> retrying a sgio read, lba=0x%" PRIx64 "\n",
933*44704f69SBart Van Assche (uint64_t)lba);
934*44704f69SBart Van Assche --retries_tmp;
935*44704f69SBart Van Assche ++num_retries;
936*44704f69SBart Van Assche if (unrecovered_errs > 0)
937*44704f69SBart Van Assche --unrecovered_errs;
938*44704f69SBart Van Assche repeat = true;
939*44704f69SBart Van Assche break;
940*44704f69SBart Van Assche }
941*44704f69SBart Van Assche ret = res;
942*44704f69SBart Van Assche goto err_out;
943*44704f69SBart Van Assche }
944*44704f69SBart Van Assche if (repeat)
945*44704f69SBart Van Assche continue;
946*44704f69SBart Van Assche if ((io_addr < (uint64_t)lba) ||
947*44704f69SBart Van Assche (io_addr >= (uint64_t)(lba + blks))) {
948*44704f69SBart Van Assche pr2serr(" Unrecovered error lba 0x%" PRIx64 " not in "
949*44704f69SBart Van Assche "correct range:\n\t[0x%" PRIx64 ",0x%" PRIx64 "]\n",
950*44704f69SBart Van Assche io_addr, (uint64_t)lba,
951*44704f69SBart Van Assche (uint64_t)(lba + blks - 1));
952*44704f69SBart Van Assche may_coe = true;
953*44704f69SBart Van Assche goto err_out;
954*44704f69SBart Van Assche }
955*44704f69SBart Van Assche blks = (int)(io_addr - (uint64_t)lba);
956*44704f69SBart Van Assche if (blks > 0) {
957*44704f69SBart Van Assche if (verbose)
958*44704f69SBart Van Assche pr2serr(" partial read of %d blocks prior to medium error\n",
959*44704f69SBart Van Assche blks);
960*44704f69SBart Van Assche res = sg_read_low(sg_fd, bp, blks, lba, bs, ifp, diop, &io_addr);
961*44704f69SBart Van Assche switch (res) {
962*44704f69SBart Van Assche case 0:
963*44704f69SBart Van Assche break;
964*44704f69SBart Van Assche case -1:
965*44704f69SBart Van Assche ifp->coe = 0;
966*44704f69SBart Van Assche ret = res;
967*44704f69SBart Van Assche goto err_out;
968*44704f69SBart Van Assche case -2:
969*44704f69SBart Van Assche pr2serr("ENOMEM again, unexpected (r)\n");
970*44704f69SBart Van Assche return -1;
971*44704f69SBart Van Assche case SG_LIB_CAT_NOT_READY:
972*44704f69SBart Van Assche pr2serr("device (r) not ready\n");
973*44704f69SBart Van Assche return res;
974*44704f69SBart Van Assche case SG_LIB_CAT_UNIT_ATTENTION:
975*44704f69SBart Van Assche pr2serr("Unit attention, unexpected (r)\n");
976*44704f69SBart Van Assche return res;
977*44704f69SBart Van Assche case SG_LIB_CAT_ABORTED_COMMAND:
978*44704f69SBart Van Assche pr2serr("Aborted command, unexpected (r)\n");
979*44704f69SBart Van Assche return res;
980*44704f69SBart Van Assche case SG_LIB_CAT_MEDIUM_HARD_WITH_INFO:
981*44704f69SBart Van Assche case SG_LIB_CAT_MEDIUM_HARD:
982*44704f69SBart Van Assche ret = SG_LIB_CAT_MEDIUM_HARD;
983*44704f69SBart Van Assche goto err_out;
984*44704f69SBart Van Assche case SG_LIB_SYNTAX_ERROR:
985*44704f69SBart Van Assche default:
986*44704f69SBart Van Assche pr2serr(">> unexpected result=%d from sg_read_low() 2\n",
987*44704f69SBart Van Assche res);
988*44704f69SBart Van Assche ret = res;
989*44704f69SBart Van Assche goto err_out;
990*44704f69SBart Van Assche }
991*44704f69SBart Van Assche }
992*44704f69SBart Van Assche xferred += blks;
993*44704f69SBart Van Assche if (0 == ifp->coe) {
994*44704f69SBart Van Assche /* give up at block before problem unless 'coe' */
995*44704f69SBart Van Assche if (blks_readp)
996*44704f69SBart Van Assche *blks_readp = xferred;
997*44704f69SBart Van Assche return ret;
998*44704f69SBart Van Assche }
999*44704f69SBart Van Assche if (bs < 32) {
1000*44704f69SBart Van Assche pr2serr(">> bs=%d too small for read_long\n", bs);
1001*44704f69SBart Van Assche return -1; /* nah, block size can't be that small */
1002*44704f69SBart Van Assche }
1003*44704f69SBart Van Assche bp += (blks * bs);
1004*44704f69SBart Van Assche lba += blks;
1005*44704f69SBart Van Assche if ((0 != ifp->pdt) || (ifp->coe < 2)) {
1006*44704f69SBart Van Assche pr2serr(">> unrecovered read error at blk=%" PRId64 ", pdt=%d, "
1007*44704f69SBart Van Assche "use zeros\n", lba, ifp->pdt);
1008*44704f69SBart Van Assche memset(bp, 0, bs);
1009*44704f69SBart Van Assche } else if (io_addr < UINT_MAX) {
1010*44704f69SBart Van Assche bool corrct, ok;
1011*44704f69SBart Van Assche int offset, nl, r;
1012*44704f69SBart Van Assche uint8_t * buffp;
1013*44704f69SBart Van Assche uint8_t * free_buffp;
1014*44704f69SBart Van Assche
1015*44704f69SBart Van Assche buffp = sg_memalign(bs * 2, 0, &free_buffp, false);
1016*44704f69SBart Van Assche if (NULL == buffp) {
1017*44704f69SBart Van Assche pr2serr(">> heap problems\n");
1018*44704f69SBart Van Assche return -1;
1019*44704f69SBart Van Assche }
1020*44704f69SBart Van Assche corrct = (ifp->coe > 2);
1021*44704f69SBart Van Assche res = sg_ll_read_long10(sg_fd, /* pblock */false, corrct, lba,
1022*44704f69SBart Van Assche buffp, bs + read_long_blk_inc, &offset,
1023*44704f69SBart Van Assche true, verbose);
1024*44704f69SBart Van Assche ok = false;
1025*44704f69SBart Van Assche switch (res) {
1026*44704f69SBart Van Assche case 0:
1027*44704f69SBart Van Assche ok = true;
1028*44704f69SBart Van Assche ++read_longs;
1029*44704f69SBart Van Assche break;
1030*44704f69SBart Van Assche case SG_LIB_CAT_ILLEGAL_REQ_WITH_INFO:
1031*44704f69SBart Van Assche nl = bs + read_long_blk_inc - offset;
1032*44704f69SBart Van Assche if ((nl < 32) || (nl > (bs * 2))) {
1033*44704f69SBart Van Assche pr2serr(">> read_long(10) len=%d unexpected\n", nl);
1034*44704f69SBart Van Assche break;
1035*44704f69SBart Van Assche }
1036*44704f69SBart Van Assche /* remember for next read_long attempt, if required */
1037*44704f69SBart Van Assche read_long_blk_inc = nl - bs;
1038*44704f69SBart Van Assche
1039*44704f69SBart Van Assche if (verbose)
1040*44704f69SBart Van Assche pr2serr("read_long(10): adjusted len=%d\n", nl);
1041*44704f69SBart Van Assche r = sg_ll_read_long10(sg_fd, false, corrct, lba, buffp, nl,
1042*44704f69SBart Van Assche &offset, true, verbose);
1043*44704f69SBart Van Assche if (0 == r) {
1044*44704f69SBart Van Assche ok = true;
1045*44704f69SBart Van Assche ++read_longs;
1046*44704f69SBart Van Assche break;
1047*44704f69SBart Van Assche } else
1048*44704f69SBart Van Assche pr2serr(">> unexpected result=%d on second "
1049*44704f69SBart Van Assche "read_long(10)\n", r);
1050*44704f69SBart Van Assche break;
1051*44704f69SBart Van Assche case SG_LIB_CAT_INVALID_OP:
1052*44704f69SBart Van Assche pr2serr(">> read_long(10); not supported\n");
1053*44704f69SBart Van Assche break;
1054*44704f69SBart Van Assche case SG_LIB_CAT_ILLEGAL_REQ:
1055*44704f69SBart Van Assche pr2serr(">> read_long(10): bad cdb field\n");
1056*44704f69SBart Van Assche break;
1057*44704f69SBart Van Assche case SG_LIB_CAT_NOT_READY:
1058*44704f69SBart Van Assche pr2serr(">> read_long(10): device not ready\n");
1059*44704f69SBart Van Assche break;
1060*44704f69SBart Van Assche case SG_LIB_CAT_UNIT_ATTENTION:
1061*44704f69SBart Van Assche pr2serr(">> read_long(10): unit attention\n");
1062*44704f69SBart Van Assche break;
1063*44704f69SBart Van Assche case SG_LIB_CAT_ABORTED_COMMAND:
1064*44704f69SBart Van Assche pr2serr(">> read_long(10): aborted command\n");
1065*44704f69SBart Van Assche break;
1066*44704f69SBart Van Assche default:
1067*44704f69SBart Van Assche pr2serr(">> read_long(10): problem (%d)\n", res);
1068*44704f69SBart Van Assche break;
1069*44704f69SBart Van Assche }
1070*44704f69SBart Van Assche if (ok)
1071*44704f69SBart Van Assche memcpy(bp, buffp, bs);
1072*44704f69SBart Van Assche else
1073*44704f69SBart Van Assche memset(bp, 0, bs);
1074*44704f69SBart Van Assche free(free_buffp);
1075*44704f69SBart Van Assche } else {
1076*44704f69SBart Van Assche pr2serr(">> read_long(10) cannot handle blk=%" PRId64 ", use "
1077*44704f69SBart Van Assche "zeros\n", lba);
1078*44704f69SBart Van Assche memset(bp, 0, bs);
1079*44704f69SBart Van Assche }
1080*44704f69SBart Van Assche ++xferred;
1081*44704f69SBart Van Assche bp += bs;
1082*44704f69SBart Van Assche ++lba;
1083*44704f69SBart Van Assche if ((coe_limit > 0) && (++coe_count > coe_limit)) {
1084*44704f69SBart Van Assche if (blks_readp)
1085*44704f69SBart Van Assche *blks_readp = xferred + blks;
1086*44704f69SBart Van Assche pr2serr(">> coe_limit on consecutive reads exceeded\n");
1087*44704f69SBart Van Assche return SG_LIB_CAT_MEDIUM_HARD;
1088*44704f69SBart Van Assche }
1089*44704f69SBart Van Assche }
1090*44704f69SBart Van Assche if (blks_readp)
1091*44704f69SBart Van Assche *blks_readp = xferred;
1092*44704f69SBart Van Assche return 0;
1093*44704f69SBart Van Assche
1094*44704f69SBart Van Assche err_out:
1095*44704f69SBart Van Assche if (ifp->coe) {
1096*44704f69SBart Van Assche memset(bp, 0, bs * blks);
1097*44704f69SBart Van Assche pr2serr(">> unable to read at blk=%" PRId64 " for %d bytes, use "
1098*44704f69SBart Van Assche "zeros\n", lba, bs * blks);
1099*44704f69SBart Van Assche if (blks > 1)
1100*44704f69SBart Van Assche pr2serr(">> try reducing bpt to limit number of zeros written "
1101*44704f69SBart Van Assche "near bad block(s)\n");
1102*44704f69SBart Van Assche /* fudge success */
1103*44704f69SBart Van Assche if (blks_readp)
1104*44704f69SBart Van Assche *blks_readp = xferred + blks;
1105*44704f69SBart Van Assche if ((coe_limit > 0) && (++coe_count > coe_limit)) {
1106*44704f69SBart Van Assche pr2serr(">> coe_limit on consecutive reads exceeded\n");
1107*44704f69SBart Van Assche return ret;
1108*44704f69SBart Van Assche }
1109*44704f69SBart Van Assche return may_coe ? 0 : ret;
1110*44704f69SBart Van Assche } else
1111*44704f69SBart Van Assche return ret;
1112*44704f69SBart Van Assche }
1113*44704f69SBart Van Assche
1114*44704f69SBart Van Assche
1115*44704f69SBart Van Assche /* Does a SCSI WRITE or VERIFY (if do_verify set) on OFILE. Returns:
1116*44704f69SBart Van Assche * 0 -> successful, SG_LIB_SYNTAX_ERROR -> unable to build cdb,
1117*44704f69SBart Van Assche * SG_LIB_CAT_NOT_READY, SG_LIB_CAT_UNIT_ATTENTION, SG_LIB_CAT_MEDIUM_HARD,
1118*44704f69SBart Van Assche * SG_LIB_CAT_ABORTED_COMMAND, -2 -> recoverable (ENOMEM),
1119*44704f69SBart Van Assche * -1 -> unrecoverable error + others. SG_DD_BYPASS -> failed but coe set. */
1120*44704f69SBart Van Assche static int
sg_write(int sg_fd,uint8_t * buff,int blocks,int64_t to_block,int bs,const struct flags_t * ofp,bool * diop)1121*44704f69SBart Van Assche sg_write(int sg_fd, uint8_t * buff, int blocks, int64_t to_block,
1122*44704f69SBart Van Assche int bs, const struct flags_t * ofp, bool * diop)
1123*44704f69SBart Van Assche {
1124*44704f69SBart Van Assche bool info_valid;
1125*44704f69SBart Van Assche int res;
1126*44704f69SBart Van Assche uint64_t io_addr = 0;
1127*44704f69SBart Van Assche const char * op_str = do_verify ? "verifying" : "writing";
1128*44704f69SBart Van Assche uint8_t wrCmd[MAX_SCSI_CDBSZ];
1129*44704f69SBart Van Assche uint8_t senseBuff[SENSE_BUFF_LEN] SG_C_CPP_ZERO_INIT;
1130*44704f69SBart Van Assche struct sg_io_hdr io_hdr;
1131*44704f69SBart Van Assche
1132*44704f69SBart Van Assche if (sg_build_scsi_cdb(wrCmd, ofp->cdbsz, blocks, to_block, do_verify,
1133*44704f69SBart Van Assche true, ofp->fua, ofp->dpo, ofp->cdl)) {
1134*44704f69SBart Van Assche pr2serr(ME "bad wr cdb build, to_block=%" PRId64 ", blocks=%d\n",
1135*44704f69SBart Van Assche to_block, blocks);
1136*44704f69SBart Van Assche return SG_LIB_SYNTAX_ERROR;
1137*44704f69SBart Van Assche }
1138*44704f69SBart Van Assche
1139*44704f69SBart Van Assche memset(&io_hdr, 0, sizeof(struct sg_io_hdr));
1140*44704f69SBart Van Assche io_hdr.interface_id = 'S';
1141*44704f69SBart Van Assche io_hdr.cmd_len = ofp->cdbsz;
1142*44704f69SBart Van Assche io_hdr.cmdp = wrCmd;
1143*44704f69SBart Van Assche io_hdr.dxfer_direction = SG_DXFER_TO_DEV;
1144*44704f69SBart Van Assche io_hdr.dxfer_len = bs * blocks;
1145*44704f69SBart Van Assche io_hdr.dxferp = buff;
1146*44704f69SBart Van Assche io_hdr.mx_sb_len = SENSE_BUFF_LEN;
1147*44704f69SBart Van Assche io_hdr.sbp = senseBuff;
1148*44704f69SBart Van Assche io_hdr.timeout = cmd_timeout;
1149*44704f69SBart Van Assche io_hdr.pack_id = (int)++glob_pack_id;
1150*44704f69SBart Van Assche if (diop && *diop)
1151*44704f69SBart Van Assche io_hdr.flags |= SG_FLAG_DIRECT_IO;
1152*44704f69SBart Van Assche
1153*44704f69SBart Van Assche if (verbose > 2)
1154*44704f69SBart Van Assche sg_print_command_len(wrCmd, ofp->cdbsz);
1155*44704f69SBart Van Assche
1156*44704f69SBart Van Assche while (((res = ioctl(sg_fd, SG_IO, &io_hdr)) < 0) &&
1157*44704f69SBart Van Assche ((EINTR == errno) || (EAGAIN == errno) || (EBUSY == errno)))
1158*44704f69SBart Van Assche ;
1159*44704f69SBart Van Assche if (res < 0) {
1160*44704f69SBart Van Assche if (ENOMEM == errno)
1161*44704f69SBart Van Assche return -2;
1162*44704f69SBart Van Assche if (do_verify)
1163*44704f69SBart Van Assche perror("verifying (SG_IO) on sg device, error");
1164*44704f69SBart Van Assche else
1165*44704f69SBart Van Assche perror("writing (SG_IO) on sg device, error");
1166*44704f69SBart Van Assche return -1;
1167*44704f69SBart Van Assche }
1168*44704f69SBart Van Assche
1169*44704f69SBart Van Assche if (verbose > 2)
1170*44704f69SBart Van Assche pr2serr(" duration=%u ms\n", io_hdr.duration);
1171*44704f69SBart Van Assche res = sg_err_category3(&io_hdr);
1172*44704f69SBart Van Assche switch (res) {
1173*44704f69SBart Van Assche case SG_LIB_CAT_CLEAN:
1174*44704f69SBart Van Assche case SG_LIB_CAT_CONDITION_MET:
1175*44704f69SBart Van Assche break;
1176*44704f69SBart Van Assche case SG_LIB_CAT_RECOVERED:
1177*44704f69SBart Van Assche ++recovered_errs;
1178*44704f69SBart Van Assche info_valid = sg_get_sense_info_fld(io_hdr.sbp, io_hdr.sb_len_wr,
1179*44704f69SBart Van Assche &io_addr);
1180*44704f69SBart Van Assche if (info_valid) {
1181*44704f69SBart Van Assche pr2serr(" lba of last recovered error in this WRITE=0x%" PRIx64
1182*44704f69SBart Van Assche "\n", io_addr);
1183*44704f69SBart Van Assche if (verbose > 1)
1184*44704f69SBart Van Assche sg_chk_n_print3(op_str, &io_hdr, true);
1185*44704f69SBart Van Assche } else {
1186*44704f69SBart Van Assche pr2serr("Recovered error: [no info] %s to block=0x%" PRIx64
1187*44704f69SBart Van Assche ", num=%d\n", op_str, to_block, blocks);
1188*44704f69SBart Van Assche sg_chk_n_print3(op_str, &io_hdr, verbose > 1);
1189*44704f69SBart Van Assche }
1190*44704f69SBart Van Assche break;
1191*44704f69SBart Van Assche case SG_LIB_CAT_ABORTED_COMMAND:
1192*44704f69SBart Van Assche case SG_LIB_CAT_UNIT_ATTENTION:
1193*44704f69SBart Van Assche sg_chk_n_print3(op_str, &io_hdr, verbose > 1);
1194*44704f69SBart Van Assche return res;
1195*44704f69SBart Van Assche case SG_LIB_CAT_MISCOMPARE: /* must be VERIFY cpommand */
1196*44704f69SBart Van Assche ++miscompare_errs;
1197*44704f69SBart Van Assche if (ofp->coe) {
1198*44704f69SBart Van Assche if (verbose > 1)
1199*44704f69SBart Van Assche pr2serr(">> bypass due to miscompare: out blk=%" PRId64
1200*44704f69SBart Van Assche " for %d blocks\n", to_block, blocks);
1201*44704f69SBart Van Assche return SG_DD_BYPASS; /* fudge success */
1202*44704f69SBart Van Assche } else {
1203*44704f69SBart Van Assche pr2serr("VERIFY reports miscompare\n");
1204*44704f69SBart Van Assche return res;
1205*44704f69SBart Van Assche }
1206*44704f69SBart Van Assche case SG_LIB_CAT_NOT_READY:
1207*44704f69SBart Van Assche ++unrecovered_errs;
1208*44704f69SBart Van Assche pr2serr("device not ready (w)\n");
1209*44704f69SBart Van Assche return res;
1210*44704f69SBart Van Assche case SG_LIB_CAT_MEDIUM_HARD:
1211*44704f69SBart Van Assche default:
1212*44704f69SBart Van Assche sg_chk_n_print3(op_str, &io_hdr, verbose > 1);
1213*44704f69SBart Van Assche if ((SG_LIB_CAT_ILLEGAL_REQ == res) && verbose)
1214*44704f69SBart Van Assche sg_print_command_len(wrCmd, ofp->cdbsz);
1215*44704f69SBart Van Assche ++unrecovered_errs;
1216*44704f69SBart Van Assche if (ofp->coe) {
1217*44704f69SBart Van Assche if (verbose > 1)
1218*44704f69SBart Van Assche pr2serr(">> ignored errors for out blk=%" PRId64 " for %d "
1219*44704f69SBart Van Assche "bytes\n", to_block, bs * blocks);
1220*44704f69SBart Van Assche return SG_DD_BYPASS; /* fudge success */
1221*44704f69SBart Van Assche } else
1222*44704f69SBart Van Assche return res;
1223*44704f69SBart Van Assche }
1224*44704f69SBart Van Assche if (diop && *diop &&
1225*44704f69SBart Van Assche ((io_hdr.info & SG_INFO_DIRECT_IO_MASK) != SG_INFO_DIRECT_IO))
1226*44704f69SBart Van Assche *diop = false; /* flag that dio not done (completely) */
1227*44704f69SBart Van Assche return 0;
1228*44704f69SBart Van Assche }
1229*44704f69SBart Van Assche
1230*44704f69SBart Van Assche
1231*44704f69SBart Van Assche static void
calc_duration_throughput(bool contin)1232*44704f69SBart Van Assche calc_duration_throughput(bool contin)
1233*44704f69SBart Van Assche {
1234*44704f69SBart Van Assche struct timeval end_tm, res_tm;
1235*44704f69SBart Van Assche double a, b;
1236*44704f69SBart Van Assche int64_t blks;
1237*44704f69SBart Van Assche
1238*44704f69SBart Van Assche if (start_tm_valid && (start_tm.tv_sec || start_tm.tv_usec)) {
1239*44704f69SBart Van Assche blks = (in_full > out_full) ? in_full : out_full;
1240*44704f69SBart Van Assche gettimeofday(&end_tm, NULL);
1241*44704f69SBart Van Assche res_tm.tv_sec = end_tm.tv_sec - start_tm.tv_sec;
1242*44704f69SBart Van Assche res_tm.tv_usec = end_tm.tv_usec - start_tm.tv_usec;
1243*44704f69SBart Van Assche if (res_tm.tv_usec < 0) {
1244*44704f69SBart Van Assche --res_tm.tv_sec;
1245*44704f69SBart Van Assche res_tm.tv_usec += 1000000;
1246*44704f69SBart Van Assche }
1247*44704f69SBart Van Assche a = res_tm.tv_sec;
1248*44704f69SBart Van Assche a += (0.000001 * res_tm.tv_usec);
1249*44704f69SBart Van Assche b = (double)blk_sz * blks;
1250*44704f69SBart Van Assche pr2serr("time to %s data%s: %d.%06d secs",
1251*44704f69SBart Van Assche (do_verify ? "verify" : "copy"), (contin ? " so far" : ""),
1252*44704f69SBart Van Assche (int)res_tm.tv_sec, (int)res_tm.tv_usec);
1253*44704f69SBart Van Assche if ((a > 0.00001) && (b > 511))
1254*44704f69SBart Van Assche pr2serr(" at %.2f MB/sec\n", b / (a * 1000000.0));
1255*44704f69SBart Van Assche else
1256*44704f69SBart Van Assche pr2serr("\n");
1257*44704f69SBart Van Assche }
1258*44704f69SBart Van Assche }
1259*44704f69SBart Van Assche
1260*44704f69SBart Van Assche /* Process arguments given to 'iflag=" or 'oflag=" options. Returns 0
1261*44704f69SBart Van Assche * on success, 1 on error. */
1262*44704f69SBart Van Assche static int
process_flags(const char * arg,struct flags_t * fp)1263*44704f69SBart Van Assche process_flags(const char * arg, struct flags_t * fp)
1264*44704f69SBart Van Assche {
1265*44704f69SBart Van Assche char buff[256];
1266*44704f69SBart Van Assche char * cp;
1267*44704f69SBart Van Assche char * np;
1268*44704f69SBart Van Assche
1269*44704f69SBart Van Assche strncpy(buff, arg, sizeof(buff));
1270*44704f69SBart Van Assche buff[sizeof(buff) - 1] = '\0';
1271*44704f69SBart Van Assche if ('\0' == buff[0]) {
1272*44704f69SBart Van Assche pr2serr("no flag found\n");
1273*44704f69SBart Van Assche return 1;
1274*44704f69SBart Van Assche }
1275*44704f69SBart Van Assche cp = buff;
1276*44704f69SBart Van Assche do {
1277*44704f69SBart Van Assche np = strchr(cp, ',');
1278*44704f69SBart Van Assche if (np)
1279*44704f69SBart Van Assche *np++ = '\0';
1280*44704f69SBart Van Assche if (0 == strcmp(cp, "00"))
1281*44704f69SBart Van Assche fp->zero = true;
1282*44704f69SBart Van Assche else if (0 == strcmp(cp, "append"))
1283*44704f69SBart Van Assche fp->append = true;
1284*44704f69SBart Van Assche else if (0 == strcmp(cp, "coe"))
1285*44704f69SBart Van Assche ++fp->coe;
1286*44704f69SBart Van Assche else if (0 == strcmp(cp, "dio"))
1287*44704f69SBart Van Assche fp->dio = true;
1288*44704f69SBart Van Assche else if (0 == strcmp(cp, "direct"))
1289*44704f69SBart Van Assche fp->direct = true;
1290*44704f69SBart Van Assche else if (0 == strcmp(cp, "dpo"))
1291*44704f69SBart Van Assche fp->dpo = true;
1292*44704f69SBart Van Assche else if (0 == strcmp(cp, "dsync"))
1293*44704f69SBart Van Assche fp->dsync = true;
1294*44704f69SBart Van Assche else if (0 == strcmp(cp, "excl"))
1295*44704f69SBart Van Assche fp->excl = true;
1296*44704f69SBart Van Assche else if (0 == strcmp(cp, "flock"))
1297*44704f69SBart Van Assche fp->flock = true;
1298*44704f69SBart Van Assche else if (0 == strcmp(cp, "ff"))
1299*44704f69SBart Van Assche fp->ff = true;
1300*44704f69SBart Van Assche else if (0 == strcmp(cp, "fua"))
1301*44704f69SBart Van Assche fp->fua = true;
1302*44704f69SBart Van Assche else if (0 == strcmp(cp, "nocache"))
1303*44704f69SBart Van Assche ++fp->nocache;
1304*44704f69SBart Van Assche else if (0 == strcmp(cp, "nocreat"))
1305*44704f69SBart Van Assche fp->nocreat = true;
1306*44704f69SBart Van Assche else if (0 == strcmp(cp, "null"))
1307*44704f69SBart Van Assche ;
1308*44704f69SBart Van Assche else if (0 == strcmp(cp, "random"))
1309*44704f69SBart Van Assche fp->random = true;
1310*44704f69SBart Van Assche else if (0 == strcmp(cp, "sgio"))
1311*44704f69SBart Van Assche fp->sgio = true;
1312*44704f69SBart Van Assche else if (0 == strcmp(cp, "sparse"))
1313*44704f69SBart Van Assche fp->sparse = true;
1314*44704f69SBart Van Assche else {
1315*44704f69SBart Van Assche pr2serr("unrecognised flag: %s\n", cp);
1316*44704f69SBart Van Assche return 1;
1317*44704f69SBart Van Assche }
1318*44704f69SBart Van Assche cp = np;
1319*44704f69SBart Van Assche } while (cp);
1320*44704f69SBart Van Assche return 0;
1321*44704f69SBart Van Assche }
1322*44704f69SBart Van Assche
1323*44704f69SBart Van Assche /* Process arguments given to 'conv=" option. Returns 0 on success,
1324*44704f69SBart Van Assche * 1 on error. */
1325*44704f69SBart Van Assche static int
process_conv(const char * arg,struct flags_t * ifp,struct flags_t * ofp)1326*44704f69SBart Van Assche process_conv(const char * arg, struct flags_t * ifp, struct flags_t * ofp)
1327*44704f69SBart Van Assche {
1328*44704f69SBart Van Assche char buff[256];
1329*44704f69SBart Van Assche char * cp;
1330*44704f69SBart Van Assche char * np;
1331*44704f69SBart Van Assche
1332*44704f69SBart Van Assche strncpy(buff, arg, sizeof(buff));
1333*44704f69SBart Van Assche buff[sizeof(buff) - 1] = '\0';
1334*44704f69SBart Van Assche if ('\0' == buff[0]) {
1335*44704f69SBart Van Assche pr2serr("no conversions found\n");
1336*44704f69SBart Van Assche return 1;
1337*44704f69SBart Van Assche }
1338*44704f69SBart Van Assche cp = buff;
1339*44704f69SBart Van Assche do {
1340*44704f69SBart Van Assche np = strchr(cp, ',');
1341*44704f69SBart Van Assche if (np)
1342*44704f69SBart Van Assche *np++ = '\0';
1343*44704f69SBart Van Assche if (0 == strcmp(cp, "nocreat"))
1344*44704f69SBart Van Assche ofp->nocreat = true;
1345*44704f69SBart Van Assche else if (0 == strcmp(cp, "noerror"))
1346*44704f69SBart Van Assche ++ifp->coe; /* will still fail on write error */
1347*44704f69SBart Van Assche else if (0 == strcmp(cp, "notrunc"))
1348*44704f69SBart Van Assche ; /* this is the default action of sg_dd so ignore */
1349*44704f69SBart Van Assche else if (0 == strcmp(cp, "null"))
1350*44704f69SBart Van Assche ;
1351*44704f69SBart Van Assche else if (0 == strcmp(cp, "sparse"))
1352*44704f69SBart Van Assche ofp->sparse = true;
1353*44704f69SBart Van Assche else if (0 == strcmp(cp, "sync"))
1354*44704f69SBart Van Assche ; /* dd(susv4): pad errored block(s) with zeros but sg_dd does
1355*44704f69SBart Van Assche * that by default. Typical dd use: 'conv=noerror,sync' */
1356*44704f69SBart Van Assche else {
1357*44704f69SBart Van Assche pr2serr("unrecognised flag: %s\n", cp);
1358*44704f69SBart Van Assche return 1;
1359*44704f69SBart Van Assche }
1360*44704f69SBart Van Assche cp = np;
1361*44704f69SBart Van Assche } while (cp);
1362*44704f69SBart Van Assche return 0;
1363*44704f69SBart Van Assche }
1364*44704f69SBart Van Assche
1365*44704f69SBart Van Assche /* Returns open input file descriptor (>= 0) or a negative value
1366*44704f69SBart Van Assche * (-SG_LIB_FILE_ERROR or -SG_LIB_CAT_OTHER) if error.
1367*44704f69SBart Van Assche */
1368*44704f69SBart Van Assche static int
open_if(const char * inf,int64_t skip,int bpt,struct flags_t * ifp,int * in_typep,int vb)1369*44704f69SBart Van Assche open_if(const char * inf, int64_t skip, int bpt, struct flags_t * ifp,
1370*44704f69SBart Van Assche int * in_typep, int vb)
1371*44704f69SBart Van Assche {
1372*44704f69SBart Van Assche int infd = -1;
1373*44704f69SBart Van Assche int flags, fl, t, res;
1374*44704f69SBart Van Assche char ebuff[EBUFF_SZ];
1375*44704f69SBart Van Assche struct sg_simple_inquiry_resp sir;
1376*44704f69SBart Van Assche
1377*44704f69SBart Van Assche *in_typep = dd_filetype(inf);
1378*44704f69SBart Van Assche if (vb)
1379*44704f69SBart Van Assche pr2serr(" >> Input file type: %s\n",
1380*44704f69SBart Van Assche dd_filetype_str(*in_typep, ebuff));
1381*44704f69SBart Van Assche if (FT_ERROR & *in_typep) {
1382*44704f69SBart Van Assche pr2serr(ME "unable access %s\n", inf);
1383*44704f69SBart Van Assche goto file_err;
1384*44704f69SBart Van Assche } else if ((FT_BLOCK & *in_typep) && ifp->sgio)
1385*44704f69SBart Van Assche *in_typep |= FT_SG;
1386*44704f69SBart Van Assche
1387*44704f69SBart Van Assche if (FT_ST & *in_typep) {
1388*44704f69SBart Van Assche pr2serr(ME "unable to use scsi tape device %s\n", inf);
1389*44704f69SBart Van Assche goto file_err;
1390*44704f69SBart Van Assche } else if (FT_SG & *in_typep) {
1391*44704f69SBart Van Assche flags = O_NONBLOCK;
1392*44704f69SBart Van Assche if (ifp->direct)
1393*44704f69SBart Van Assche flags |= O_DIRECT;
1394*44704f69SBart Van Assche if (ifp->excl)
1395*44704f69SBart Van Assche flags |= O_EXCL;
1396*44704f69SBart Van Assche if (ifp->dsync)
1397*44704f69SBart Van Assche flags |= O_SYNC;
1398*44704f69SBart Van Assche fl = O_RDWR;
1399*44704f69SBart Van Assche if ((infd = open(inf, fl | flags)) < 0) {
1400*44704f69SBart Van Assche fl = O_RDONLY;
1401*44704f69SBart Van Assche if ((infd = open(inf, fl | flags)) < 0) {
1402*44704f69SBart Van Assche snprintf(ebuff, EBUFF_SZ,
1403*44704f69SBart Van Assche ME "could not open %s for sg reading", inf);
1404*44704f69SBart Van Assche perror(ebuff);
1405*44704f69SBart Van Assche goto file_err;
1406*44704f69SBart Van Assche }
1407*44704f69SBart Van Assche }
1408*44704f69SBart Van Assche if (vb)
1409*44704f69SBart Van Assche pr2serr(" open input(sg_io), flags=0x%x\n", fl | flags);
1410*44704f69SBart Van Assche if (sg_simple_inquiry(infd, &sir, false, (vb ? (vb - 1) : 0))) {
1411*44704f69SBart Van Assche pr2serr("INQUIRY failed on %s\n", inf);
1412*44704f69SBart Van Assche goto other_err;
1413*44704f69SBart Van Assche }
1414*44704f69SBart Van Assche ifp->pdt = sir.peripheral_type;
1415*44704f69SBart Van Assche if (vb)
1416*44704f69SBart Van Assche pr2serr(" %s: %.8s %.16s %.4s [pdt=%d]\n", inf, sir.vendor,
1417*44704f69SBart Van Assche sir.product, sir.revision, ifp->pdt);
1418*44704f69SBart Van Assche if (! (FT_BLOCK & *in_typep)) {
1419*44704f69SBart Van Assche t = blk_sz * bpt;
1420*44704f69SBart Van Assche res = ioctl(infd, SG_SET_RESERVED_SIZE, &t);
1421*44704f69SBart Van Assche if (res < 0)
1422*44704f69SBart Van Assche perror(ME "SG_SET_RESERVED_SIZE error");
1423*44704f69SBart Van Assche res = ioctl(infd, SG_GET_VERSION_NUM, &t);
1424*44704f69SBart Van Assche if ((res < 0) || (t < 30000)) {
1425*44704f69SBart Van Assche if (FT_BLOCK & *in_typep)
1426*44704f69SBart Van Assche pr2serr(ME "SG_IO unsupported on this block device\n");
1427*44704f69SBart Van Assche else
1428*44704f69SBart Van Assche pr2serr(ME "sg driver prior to 3.x.y\n");
1429*44704f69SBart Van Assche goto file_err;
1430*44704f69SBart Van Assche }
1431*44704f69SBart Van Assche }
1432*44704f69SBart Van Assche } else if (FT_NVME & *in_typep) {
1433*44704f69SBart Van Assche pr2serr("Don't support NVMe char devices as IFILE\n");
1434*44704f69SBart Van Assche goto file_err;
1435*44704f69SBart Van Assche } else {
1436*44704f69SBart Van Assche flags = O_RDONLY;
1437*44704f69SBart Van Assche if (ifp->direct)
1438*44704f69SBart Van Assche flags |= O_DIRECT;
1439*44704f69SBart Van Assche if (ifp->excl)
1440*44704f69SBart Van Assche flags |= O_EXCL;
1441*44704f69SBart Van Assche if (ifp->dsync)
1442*44704f69SBart Van Assche flags |= O_SYNC;
1443*44704f69SBart Van Assche infd = open(inf, flags);
1444*44704f69SBart Van Assche if (infd < 0) {
1445*44704f69SBart Van Assche snprintf(ebuff, EBUFF_SZ,
1446*44704f69SBart Van Assche ME "could not open %s for reading", inf);
1447*44704f69SBart Van Assche perror(ebuff);
1448*44704f69SBart Van Assche goto file_err;
1449*44704f69SBart Van Assche } else {
1450*44704f69SBart Van Assche if (vb)
1451*44704f69SBart Van Assche pr2serr(" open input, flags=0x%x\n", flags);
1452*44704f69SBart Van Assche if (skip > 0) {
1453*44704f69SBart Van Assche off64_t offset = skip;
1454*44704f69SBart Van Assche
1455*44704f69SBart Van Assche offset *= blk_sz; /* could exceed 32 bits here! */
1456*44704f69SBart Van Assche if (lseek64(infd, offset, SEEK_SET) < 0) {
1457*44704f69SBart Van Assche snprintf(ebuff, EBUFF_SZ, ME "couldn't skip to "
1458*44704f69SBart Van Assche "required position on %s", inf);
1459*44704f69SBart Van Assche perror(ebuff);
1460*44704f69SBart Van Assche goto file_err;
1461*44704f69SBart Van Assche }
1462*44704f69SBart Van Assche if (vb)
1463*44704f69SBart Van Assche pr2serr(" >> skip: lseek64 SEEK_SET, byte offset=0x%"
1464*44704f69SBart Van Assche PRIx64 "\n", (uint64_t)offset);
1465*44704f69SBart Van Assche }
1466*44704f69SBart Van Assche #ifdef HAVE_POSIX_FADVISE
1467*44704f69SBart Van Assche if (ifp->nocache) {
1468*44704f69SBart Van Assche int rt;
1469*44704f69SBart Van Assche
1470*44704f69SBart Van Assche rt = posix_fadvise(infd, 0, 0, POSIX_FADV_SEQUENTIAL);
1471*44704f69SBart Van Assche if (rt)
1472*44704f69SBart Van Assche pr2serr("open_if: posix_fadvise(SEQUENTIAL), err=%d\n",
1473*44704f69SBart Van Assche rt);
1474*44704f69SBart Van Assche }
1475*44704f69SBart Van Assche #endif
1476*44704f69SBart Van Assche }
1477*44704f69SBart Van Assche }
1478*44704f69SBart Van Assche if (ifp->flock && (infd >= 0)) {
1479*44704f69SBart Van Assche res = flock(infd, LOCK_EX | LOCK_NB);
1480*44704f69SBart Van Assche if (res < 0) {
1481*44704f69SBart Van Assche close(infd);
1482*44704f69SBart Van Assche snprintf(ebuff, EBUFF_SZ, ME "flock(LOCK_EX | LOCK_NB) on %s "
1483*44704f69SBart Van Assche "failed", inf);
1484*44704f69SBart Van Assche perror(ebuff);
1485*44704f69SBart Van Assche return -SG_LIB_FLOCK_ERR;
1486*44704f69SBart Van Assche }
1487*44704f69SBart Van Assche }
1488*44704f69SBart Van Assche return infd;
1489*44704f69SBart Van Assche
1490*44704f69SBart Van Assche file_err:
1491*44704f69SBart Van Assche if (infd >= 0)
1492*44704f69SBart Van Assche close(infd);
1493*44704f69SBart Van Assche return -SG_LIB_FILE_ERROR;
1494*44704f69SBart Van Assche other_err:
1495*44704f69SBart Van Assche if (infd >= 0)
1496*44704f69SBart Van Assche close(infd);
1497*44704f69SBart Van Assche return -SG_LIB_CAT_OTHER;
1498*44704f69SBart Van Assche }
1499*44704f69SBart Van Assche
1500*44704f69SBart Van Assche /* Returns open output file descriptor (>= 0), -1 for don't
1501*44704f69SBart Van Assche * bother opening (e.g. /dev/null), or a more negative value
1502*44704f69SBart Van Assche * (-SG_LIB_FILE_ERROR or -SG_LIB_CAT_OTHER) if error.
1503*44704f69SBart Van Assche */
1504*44704f69SBart Van Assche static int
open_of(const char * outf,int64_t seek,int bpt,struct flags_t * ofp,int * out_typep,int vb)1505*44704f69SBart Van Assche open_of(const char * outf, int64_t seek, int bpt, struct flags_t * ofp,
1506*44704f69SBart Van Assche int * out_typep, int vb)
1507*44704f69SBart Van Assche {
1508*44704f69SBart Van Assche bool not_found;
1509*44704f69SBart Van Assche int outfd = -1;
1510*44704f69SBart Van Assche int flags, t, res;
1511*44704f69SBart Van Assche char ebuff[EBUFF_SZ];
1512*44704f69SBart Van Assche struct sg_simple_inquiry_resp sir;
1513*44704f69SBart Van Assche
1514*44704f69SBart Van Assche *out_typep = dd_filetype(outf);
1515*44704f69SBart Van Assche if (vb)
1516*44704f69SBart Van Assche pr2serr(" >> Output file type: %s\n",
1517*44704f69SBart Van Assche dd_filetype_str(*out_typep, ebuff));
1518*44704f69SBart Van Assche not_found = (FT_ERROR == *out_typep); /* assume error was not found */
1519*44704f69SBart Van Assche
1520*44704f69SBart Van Assche if ((FT_BLOCK & *out_typep) && ofp->sgio)
1521*44704f69SBart Van Assche *out_typep |= FT_SG;
1522*44704f69SBart Van Assche
1523*44704f69SBart Van Assche if (FT_ST & *out_typep) {
1524*44704f69SBart Van Assche pr2serr(ME "unable to use scsi tape device %s\n", outf);
1525*44704f69SBart Van Assche goto file_err;
1526*44704f69SBart Van Assche } else if (FT_SG & *out_typep) {
1527*44704f69SBart Van Assche flags = O_RDWR | O_NONBLOCK;
1528*44704f69SBart Van Assche if (ofp->direct)
1529*44704f69SBart Van Assche flags |= O_DIRECT;
1530*44704f69SBart Van Assche if (ofp->excl)
1531*44704f69SBart Van Assche flags |= O_EXCL;
1532*44704f69SBart Van Assche if (ofp->dsync)
1533*44704f69SBart Van Assche flags |= O_SYNC;
1534*44704f69SBart Van Assche if ((outfd = open(outf, flags)) < 0) {
1535*44704f69SBart Van Assche snprintf(ebuff, EBUFF_SZ,
1536*44704f69SBart Van Assche ME "could not open %s for sg writing", outf);
1537*44704f69SBart Van Assche perror(ebuff);
1538*44704f69SBart Van Assche goto file_err;
1539*44704f69SBart Van Assche }
1540*44704f69SBart Van Assche if (vb)
1541*44704f69SBart Van Assche pr2serr(" open output(sg_io), flags=0x%x\n", flags);
1542*44704f69SBart Van Assche if (sg_simple_inquiry(outfd, &sir, false, (vb ? (vb - 1) : 0))) {
1543*44704f69SBart Van Assche pr2serr("INQUIRY failed on %s\n", outf);
1544*44704f69SBart Van Assche goto other_err;
1545*44704f69SBart Van Assche }
1546*44704f69SBart Van Assche ofp->pdt = sir.peripheral_type;
1547*44704f69SBart Van Assche if (vb)
1548*44704f69SBart Van Assche pr2serr(" %s: %.8s %.16s %.4s [pdt=%d]\n", outf, sir.vendor,
1549*44704f69SBart Van Assche sir.product, sir.revision, ofp->pdt);
1550*44704f69SBart Van Assche if (! (FT_BLOCK & *out_typep)) {
1551*44704f69SBart Van Assche t = blk_sz * bpt;
1552*44704f69SBart Van Assche res = ioctl(outfd, SG_SET_RESERVED_SIZE, &t);
1553*44704f69SBart Van Assche if (res < 0)
1554*44704f69SBart Van Assche perror(ME "SG_SET_RESERVED_SIZE error");
1555*44704f69SBart Van Assche res = ioctl(outfd, SG_GET_VERSION_NUM, &t);
1556*44704f69SBart Van Assche if ((res < 0) || (t < 30000)) {
1557*44704f69SBart Van Assche pr2serr(ME "sg driver prior to 3.x.y\n");
1558*44704f69SBart Van Assche goto file_err;
1559*44704f69SBart Van Assche }
1560*44704f69SBart Van Assche }
1561*44704f69SBart Van Assche } else if (FT_NVME & *out_typep) {
1562*44704f69SBart Van Assche pr2serr("Don't support NVMe char devices as OFILE\n");
1563*44704f69SBart Van Assche goto file_err;
1564*44704f69SBart Van Assche } else if (FT_DEV_NULL & *out_typep)
1565*44704f69SBart Van Assche outfd = -1; /* don't bother opening */
1566*44704f69SBart Van Assche else if (FT_RAW & *out_typep) {
1567*44704f69SBart Van Assche flags = O_WRONLY;
1568*44704f69SBart Van Assche if (ofp->direct)
1569*44704f69SBart Van Assche flags |= O_DIRECT;
1570*44704f69SBart Van Assche if (ofp->excl)
1571*44704f69SBart Van Assche flags |= O_EXCL;
1572*44704f69SBart Van Assche if (ofp->dsync)
1573*44704f69SBart Van Assche flags |= O_SYNC;
1574*44704f69SBart Van Assche if ((outfd = open(outf, flags)) < 0) {
1575*44704f69SBart Van Assche snprintf(ebuff, EBUFF_SZ,
1576*44704f69SBart Van Assche ME "could not open %s for raw writing", outf);
1577*44704f69SBart Van Assche perror(ebuff);
1578*44704f69SBart Van Assche goto file_err;
1579*44704f69SBart Van Assche }
1580*44704f69SBart Van Assche } else { /* FT_OTHER or FT_ERROR (not found so create) */
1581*44704f69SBart Van Assche flags = O_WRONLY;
1582*44704f69SBart Van Assche if (! ofp->nocreat)
1583*44704f69SBart Van Assche flags |= O_CREAT;
1584*44704f69SBart Van Assche if (ofp->direct)
1585*44704f69SBart Van Assche flags |= O_DIRECT;
1586*44704f69SBart Van Assche if (ofp->excl)
1587*44704f69SBart Van Assche flags |= O_EXCL;
1588*44704f69SBart Van Assche if (ofp->dsync)
1589*44704f69SBart Van Assche flags |= O_SYNC;
1590*44704f69SBart Van Assche if (ofp->append)
1591*44704f69SBart Van Assche flags |= O_APPEND;
1592*44704f69SBart Van Assche if ((outfd = open(outf, flags, 0666)) < 0) {
1593*44704f69SBart Van Assche snprintf(ebuff, EBUFF_SZ,
1594*44704f69SBart Van Assche ME "could not open %s for writing", outf);
1595*44704f69SBart Van Assche perror(ebuff);
1596*44704f69SBart Van Assche goto file_err;
1597*44704f69SBart Van Assche }
1598*44704f69SBart Van Assche if (vb)
1599*44704f69SBart Van Assche pr2serr(" %s output, flags=0x%x\n",
1600*44704f69SBart Van Assche (not_found ? "create" : "open"), flags);
1601*44704f69SBart Van Assche if (seek > 0) {
1602*44704f69SBart Van Assche off64_t offset = seek;
1603*44704f69SBart Van Assche
1604*44704f69SBart Van Assche offset *= blk_sz; /* could exceed 32 bits here! */
1605*44704f69SBart Van Assche if (lseek64(outfd, offset, SEEK_SET) < 0) {
1606*44704f69SBart Van Assche snprintf(ebuff, EBUFF_SZ,
1607*44704f69SBart Van Assche ME "couldn't seek to required position on %s", outf);
1608*44704f69SBart Van Assche perror(ebuff);
1609*44704f69SBart Van Assche goto file_err;
1610*44704f69SBart Van Assche }
1611*44704f69SBart Van Assche if (vb)
1612*44704f69SBart Van Assche pr2serr(" >> seek: lseek64 SEEK_SET, byte offset=0x%" PRIx64
1613*44704f69SBart Van Assche "\n", (uint64_t)offset);
1614*44704f69SBart Van Assche }
1615*44704f69SBart Van Assche }
1616*44704f69SBart Van Assche if (ofp->flock && (outfd >= 0)) {
1617*44704f69SBart Van Assche res = flock(outfd, LOCK_EX | LOCK_NB);
1618*44704f69SBart Van Assche if (res < 0) {
1619*44704f69SBart Van Assche snprintf(ebuff, EBUFF_SZ, ME "flock(LOCK_EX | LOCK_NB) on %s "
1620*44704f69SBart Van Assche "failed", outf);
1621*44704f69SBart Van Assche perror(ebuff);
1622*44704f69SBart Van Assche close(outfd);
1623*44704f69SBart Van Assche return -SG_LIB_FLOCK_ERR;
1624*44704f69SBart Van Assche }
1625*44704f69SBart Van Assche }
1626*44704f69SBart Van Assche return outfd;
1627*44704f69SBart Van Assche
1628*44704f69SBart Van Assche file_err:
1629*44704f69SBart Van Assche if (outfd >= 0)
1630*44704f69SBart Van Assche close(outfd);
1631*44704f69SBart Van Assche return -SG_LIB_FILE_ERROR;
1632*44704f69SBart Van Assche other_err:
1633*44704f69SBart Van Assche if (outfd >= 0)
1634*44704f69SBart Van Assche close(outfd);
1635*44704f69SBart Van Assche return -SG_LIB_CAT_OTHER;
1636*44704f69SBart Van Assche }
1637*44704f69SBart Van Assche
1638*44704f69SBart Van Assche /* Returns the number of times 'ch' is found in string 's' given the
1639*44704f69SBart Van Assche * string's length. */
1640*44704f69SBart Van Assche static int
num_chs_in_str(const char * s,int slen,int ch)1641*44704f69SBart Van Assche num_chs_in_str(const char * s, int slen, int ch)
1642*44704f69SBart Van Assche {
1643*44704f69SBart Van Assche int res = 0;
1644*44704f69SBart Van Assche
1645*44704f69SBart Van Assche while (--slen >= 0) {
1646*44704f69SBart Van Assche if (ch == s[slen])
1647*44704f69SBart Van Assche ++res;
1648*44704f69SBart Van Assche }
1649*44704f69SBart Van Assche return res;
1650*44704f69SBart Van Assche }
1651*44704f69SBart Van Assche
1652*44704f69SBart Van Assche /* Returns true when it time to output a progress report; else false. */
1653*44704f69SBart Van Assche static bool
check_progress(void)1654*44704f69SBart Van Assche check_progress(void)
1655*44704f69SBart Van Assche {
1656*44704f69SBart Van Assche #if defined(HAVE_CLOCK_GETTIME) && defined(CLOCK_MONOTONIC)
1657*44704f69SBart Van Assche static bool have_prev, measure;
1658*44704f69SBart Van Assche static struct timespec prev_true_tm;
1659*44704f69SBart Van Assche static int count, threshold;
1660*44704f69SBart Van Assche bool res = false;
1661*44704f69SBart Van Assche uint32_t elapsed_ms, ms;
1662*44704f69SBart Van Assche struct timespec now_tm, res_tm;
1663*44704f69SBart Van Assche
1664*44704f69SBart Van Assche if (progress) {
1665*44704f69SBart Van Assche if (! have_prev) {
1666*44704f69SBart Van Assche have_prev = true;
1667*44704f69SBart Van Assche measure = true;
1668*44704f69SBart Van Assche clock_gettime(CLOCK_MONOTONIC, &prev_true_tm);
1669*44704f69SBart Van Assche return false; /* starting reference */
1670*44704f69SBart Van Assche }
1671*44704f69SBart Van Assche if (! measure) {
1672*44704f69SBart Van Assche if (++count >= threshold)
1673*44704f69SBart Van Assche count = 0;
1674*44704f69SBart Van Assche else
1675*44704f69SBart Van Assche return false;
1676*44704f69SBart Van Assche }
1677*44704f69SBart Van Assche clock_gettime(CLOCK_MONOTONIC, &now_tm);
1678*44704f69SBart Van Assche res_tm.tv_sec = now_tm.tv_sec - prev_true_tm.tv_sec;
1679*44704f69SBart Van Assche res_tm.tv_nsec = now_tm.tv_nsec - prev_true_tm.tv_nsec;
1680*44704f69SBart Van Assche if (res_tm.tv_nsec < 0) {
1681*44704f69SBart Van Assche --res_tm.tv_sec;
1682*44704f69SBart Van Assche res_tm.tv_nsec += 1000000000;
1683*44704f69SBart Van Assche }
1684*44704f69SBart Van Assche elapsed_ms = (1000 * res_tm.tv_sec) + (res_tm.tv_nsec / 1000000);
1685*44704f69SBart Van Assche if (measure) {
1686*44704f69SBart Van Assche ++threshold;
1687*44704f69SBart Van Assche if (elapsed_ms > 80) /* 80 milliseconds */
1688*44704f69SBart Van Assche measure = false;
1689*44704f69SBart Van Assche }
1690*44704f69SBart Van Assche if (elapsed_ms >= PROGRESS3_TRIGGER_MS) {
1691*44704f69SBart Van Assche if (elapsed_ms >= PROGRESS2_TRIGGER_MS) {
1692*44704f69SBart Van Assche if (elapsed_ms >= PROGRESS_TRIGGER_MS) {
1693*44704f69SBart Van Assche ms = PROGRESS_TRIGGER_MS;
1694*44704f69SBart Van Assche res = true;
1695*44704f69SBart Van Assche } else if (progress > 1) {
1696*44704f69SBart Van Assche ms = PROGRESS2_TRIGGER_MS;
1697*44704f69SBart Van Assche res = true;
1698*44704f69SBart Van Assche }
1699*44704f69SBart Van Assche } else if (progress > 2) {
1700*44704f69SBart Van Assche ms = PROGRESS3_TRIGGER_MS;
1701*44704f69SBart Van Assche res = true;
1702*44704f69SBart Van Assche }
1703*44704f69SBart Van Assche }
1704*44704f69SBart Van Assche if (res) {
1705*44704f69SBart Van Assche prev_true_tm.tv_sec += (ms / 1000);
1706*44704f69SBart Van Assche prev_true_tm.tv_nsec += (ms % 1000) * 1000000;
1707*44704f69SBart Van Assche if (prev_true_tm.tv_nsec >= 1000000000) {
1708*44704f69SBart Van Assche ++prev_true_tm.tv_sec;
1709*44704f69SBart Van Assche prev_true_tm.tv_nsec -= 1000000000;
1710*44704f69SBart Van Assche }
1711*44704f69SBart Van Assche }
1712*44704f69SBart Van Assche }
1713*44704f69SBart Van Assche return res;
1714*44704f69SBart Van Assche
1715*44704f69SBart Van Assche #elif defined(HAVE_GETTIMEOFDAY)
1716*44704f69SBart Van Assche static bool have_prev, measure;
1717*44704f69SBart Van Assche static struct timeval prev_true_tm;
1718*44704f69SBart Van Assche static int count, threshold;
1719*44704f69SBart Van Assche bool res = false;
1720*44704f69SBart Van Assche uint32_t elapsed_ms, ms;
1721*44704f69SBart Van Assche struct timeval now_tm, res_tm;
1722*44704f69SBart Van Assche
1723*44704f69SBart Van Assche if (progress) {
1724*44704f69SBart Van Assche if (! have_prev) {
1725*44704f69SBart Van Assche have_prev = true;
1726*44704f69SBart Van Assche gettimeofday(&prev_true_tm, NULL);
1727*44704f69SBart Van Assche return false; /* starting reference */
1728*44704f69SBart Van Assche }
1729*44704f69SBart Van Assche if (! measure) {
1730*44704f69SBart Van Assche if (++count >= threshold)
1731*44704f69SBart Van Assche count = 0;
1732*44704f69SBart Van Assche else
1733*44704f69SBart Van Assche return false;
1734*44704f69SBart Van Assche }
1735*44704f69SBart Van Assche gettimeofday(&now_tm, NULL);
1736*44704f69SBart Van Assche res_tm.tv_sec = now_tm.tv_sec - prev_true_tm.tv_sec;
1737*44704f69SBart Van Assche res_tm.tv_usec = now_tm.tv_usec - prev_true_tm.tv_usec;
1738*44704f69SBart Van Assche if (res_tm.tv_usec < 0) {
1739*44704f69SBart Van Assche --res_tm.tv_sec;
1740*44704f69SBart Van Assche res_tm.tv_usec += 1000000;
1741*44704f69SBart Van Assche }
1742*44704f69SBart Van Assche elapsed_ms = (1000 * res_tm.tv_sec) + (res_tm.tv_usec / 1000);
1743*44704f69SBart Van Assche if (measure) {
1744*44704f69SBart Van Assche ++threshold;
1745*44704f69SBart Van Assche if (elapsed_ms > 80) /* 80 milliseconds */
1746*44704f69SBart Van Assche measure = false;
1747*44704f69SBart Van Assche }
1748*44704f69SBart Van Assche if (elapsed_ms >= PROGRESS3_TRIGGER_MS) {
1749*44704f69SBart Van Assche if (elapsed_ms >= PROGRESS2_TRIGGER_MS) {
1750*44704f69SBart Van Assche if (elapsed_ms >= PROGRESS_TRIGGER_MS) {
1751*44704f69SBart Van Assche ms = PROGRESS_TRIGGER_MS;
1752*44704f69SBart Van Assche res = true;
1753*44704f69SBart Van Assche } else if (progress > 1) {
1754*44704f69SBart Van Assche ms = PROGRESS2_TRIGGER_MS;
1755*44704f69SBart Van Assche res = true;
1756*44704f69SBart Van Assche }
1757*44704f69SBart Van Assche } else if (progress > 2) {
1758*44704f69SBart Van Assche ms = PROGRESS3_TRIGGER_MS;
1759*44704f69SBart Van Assche res = true;
1760*44704f69SBart Van Assche }
1761*44704f69SBart Van Assche }
1762*44704f69SBart Van Assche if (res) {
1763*44704f69SBart Van Assche prev_true_tm.tv_sec += (ms / 1000);
1764*44704f69SBart Van Assche prev_true_tm.tv_usec += (ms % 1000) * 1000;
1765*44704f69SBart Van Assche if (prev_true_tm.tv_usec >= 1000000) {
1766*44704f69SBart Van Assche ++prev_true_tm.tv_sec;
1767*44704f69SBart Van Assche prev_true_tm.tv_usec -= 1000000;
1768*44704f69SBart Van Assche }
1769*44704f69SBart Van Assche }
1770*44704f69SBart Van Assche }
1771*44704f69SBart Van Assche return res;
1772*44704f69SBart Van Assche
1773*44704f69SBart Van Assche #else /* no clock reading functions available */
1774*44704f69SBart Van Assche return false;
1775*44704f69SBart Van Assche #endif
1776*44704f69SBart Van Assche }
1777*44704f69SBart Van Assche
1778*44704f69SBart Van Assche
1779*44704f69SBart Van Assche int
main(int argc,char * argv[])1780*44704f69SBart Van Assche main(int argc, char * argv[])
1781*44704f69SBart Van Assche {
1782*44704f69SBart Van Assche bool bpt_given = false;
1783*44704f69SBart Van Assche bool cdbsz_given = false;
1784*44704f69SBart Van Assche bool cdl_given = false;
1785*44704f69SBart Van Assche bool dio_tmp, first;
1786*44704f69SBart Van Assche bool do_sync = false;
1787*44704f69SBart Van Assche bool penult_sparse_skip = false;
1788*44704f69SBart Van Assche bool sparse_skip = false;
1789*44704f69SBart Van Assche bool verbose_given = false;
1790*44704f69SBart Van Assche bool version_given = false;
1791*44704f69SBart Van Assche int res, k, n, t, buf_sz, blocks_per, infd, outfd, out2fd, keylen;
1792*44704f69SBart Van Assche int retries_tmp, blks_read, bytes_read, bytes_of2, bytes_of;
1793*44704f69SBart Van Assche int in_sect_sz, out_sect_sz;
1794*44704f69SBart Van Assche int blocks = 0;
1795*44704f69SBart Van Assche int bpt = DEF_BLOCKS_PER_TRANSFER;
1796*44704f69SBart Van Assche int dio_incomplete_count = 0;
1797*44704f69SBart Van Assche int ibs = 0;
1798*44704f69SBart Van Assche int in_type = FT_OTHER;
1799*44704f69SBart Van Assche int obs = 0;
1800*44704f69SBart Van Assche int out_type = FT_OTHER;
1801*44704f69SBart Van Assche int out2_type = FT_OTHER;
1802*44704f69SBart Van Assche int penult_blocks = 0;
1803*44704f69SBart Van Assche int ret = 0;
1804*44704f69SBart Van Assche int64_t skip = 0;
1805*44704f69SBart Van Assche int64_t seek = 0;
1806*44704f69SBart Van Assche int64_t in_num_sect = -1;
1807*44704f69SBart Van Assche int64_t out_num_sect = -1;
1808*44704f69SBart Van Assche char * key;
1809*44704f69SBart Van Assche char * buf;
1810*44704f69SBart Van Assche const char * ccp = NULL;
1811*44704f69SBart Van Assche const char * cc2p;
1812*44704f69SBart Van Assche uint8_t * wrkBuff = NULL;
1813*44704f69SBart Van Assche uint8_t * wrkPos;
1814*44704f69SBart Van Assche char inf[INOUTF_SZ];
1815*44704f69SBart Van Assche char outf[INOUTF_SZ];
1816*44704f69SBart Van Assche char out2f[INOUTF_SZ];
1817*44704f69SBart Van Assche char str[STR_SZ];
1818*44704f69SBart Van Assche char ebuff[EBUFF_SZ];
1819*44704f69SBart Van Assche
1820*44704f69SBart Van Assche inf[0] = '\0';
1821*44704f69SBart Van Assche outf[0] = '\0';
1822*44704f69SBart Van Assche out2f[0] = '\0';
1823*44704f69SBart Van Assche iflag.cdbsz = DEF_SCSI_CDBSZ;
1824*44704f69SBart Van Assche oflag.cdbsz = DEF_SCSI_CDBSZ;
1825*44704f69SBart Van Assche
1826*44704f69SBart Van Assche for (k = 1; k < argc; k++) {
1827*44704f69SBart Van Assche if (argv[k]) {
1828*44704f69SBart Van Assche strncpy(str, argv[k], STR_SZ);
1829*44704f69SBart Van Assche str[STR_SZ - 1] = '\0';
1830*44704f69SBart Van Assche } else
1831*44704f69SBart Van Assche continue;
1832*44704f69SBart Van Assche for (key = str, buf = key; *buf && *buf != '=';)
1833*44704f69SBart Van Assche buf++;
1834*44704f69SBart Van Assche if (*buf)
1835*44704f69SBart Van Assche *buf++ = '\0';
1836*44704f69SBart Van Assche keylen = strlen(key);
1837*44704f69SBart Van Assche if (0 == strncmp(key, "app", 3)) {
1838*44704f69SBart Van Assche iflag.append = !! sg_get_num(buf);
1839*44704f69SBart Van Assche oflag.append = iflag.append;
1840*44704f69SBart Van Assche } else if (0 == strcmp(key, "blk_sgio")) {
1841*44704f69SBart Van Assche iflag.sgio = !! sg_get_num(buf);
1842*44704f69SBart Van Assche oflag.sgio = iflag.sgio;
1843*44704f69SBart Van Assche } else if (0 == strcmp(key, "bpt")) {
1844*44704f69SBart Van Assche bpt = sg_get_num(buf);
1845*44704f69SBart Van Assche if (-1 == bpt) {
1846*44704f69SBart Van Assche pr2serr(ME "bad argument to 'bpt='\n");
1847*44704f69SBart Van Assche return SG_LIB_SYNTAX_ERROR;
1848*44704f69SBart Van Assche }
1849*44704f69SBart Van Assche bpt_given = true;
1850*44704f69SBart Van Assche } else if (0 == strcmp(key, "bs")) {
1851*44704f69SBart Van Assche blk_sz = sg_get_num(buf);
1852*44704f69SBart Van Assche if ((blk_sz < 0) || (blk_sz > MAX_BPT_VALUE)) {
1853*44704f69SBart Van Assche pr2serr(ME "bad argument to 'bs='\n");
1854*44704f69SBart Van Assche return SG_LIB_SYNTAX_ERROR;
1855*44704f69SBart Van Assche }
1856*44704f69SBart Van Assche } else if (0 == strcmp(key, "cdbsz")) {
1857*44704f69SBart Van Assche iflag.cdbsz = sg_get_num(buf);
1858*44704f69SBart Van Assche if ((iflag.cdbsz < 6) || (iflag.cdbsz > 32)) {
1859*44704f69SBart Van Assche pr2serr(ME "'cdbsz' expects 6, 10, 12, 16 or 32\n");
1860*44704f69SBart Van Assche return SG_LIB_SYNTAX_ERROR;
1861*44704f69SBart Van Assche }
1862*44704f69SBart Van Assche oflag.cdbsz = iflag.cdbsz;
1863*44704f69SBart Van Assche cdbsz_given = true;
1864*44704f69SBart Van Assche } else if (0 == strcmp(key, "cdl")) {
1865*44704f69SBart Van Assche const char * cp = strchr(buf, ',');
1866*44704f69SBart Van Assche
1867*44704f69SBart Van Assche iflag.cdl = sg_get_num(buf);
1868*44704f69SBart Van Assche if ((iflag.cdl < 0) || (iflag.cdl > 7)) {
1869*44704f69SBart Van Assche pr2serr(ME "bad argument to 'cdl=', expect 0 to 7\n");
1870*44704f69SBart Van Assche return SG_LIB_SYNTAX_ERROR;
1871*44704f69SBart Van Assche }
1872*44704f69SBart Van Assche if (cp) {
1873*44704f69SBart Van Assche oflag.cdl = sg_get_num(cp + 1);
1874*44704f69SBart Van Assche if ((oflag.cdl < 0) || (oflag.cdl > 7)) {
1875*44704f69SBart Van Assche pr2serr(ME "bad argument to 'cdl=ICDL,OCDL', expect OCDL "
1876*44704f69SBart Van Assche "to be 0 to 7\n");
1877*44704f69SBart Van Assche return SG_LIB_SYNTAX_ERROR;
1878*44704f69SBart Van Assche }
1879*44704f69SBart Van Assche } else
1880*44704f69SBart Van Assche oflag.cdl = iflag.cdl;
1881*44704f69SBart Van Assche cdl_given = true;
1882*44704f69SBart Van Assche } else if (0 == strcmp(key, "coe")) {
1883*44704f69SBart Van Assche iflag.coe = sg_get_num(buf);
1884*44704f69SBart Van Assche oflag.coe = iflag.coe;
1885*44704f69SBart Van Assche } else if (0 == strcmp(key, "coe_limit")) {
1886*44704f69SBart Van Assche coe_limit = sg_get_num(buf);
1887*44704f69SBart Van Assche if (-1 == coe_limit) {
1888*44704f69SBart Van Assche pr2serr(ME "bad argument to 'coe_limit='\n");
1889*44704f69SBart Van Assche return SG_LIB_SYNTAX_ERROR;
1890*44704f69SBart Van Assche }
1891*44704f69SBart Van Assche } else if (0 == strcmp(key, "conv")) {
1892*44704f69SBart Van Assche if (process_conv(buf, &iflag, &oflag)) {
1893*44704f69SBart Van Assche pr2serr(ME "bad argument to 'conv='\n");
1894*44704f69SBart Van Assche return SG_LIB_SYNTAX_ERROR;
1895*44704f69SBart Van Assche }
1896*44704f69SBart Van Assche } else if (0 == strcmp(key, "count")) {
1897*44704f69SBart Van Assche if (0 != strcmp("-1", buf)) {
1898*44704f69SBart Van Assche dd_count = sg_get_llnum(buf);
1899*44704f69SBart Van Assche if ((dd_count < 0) || (dd_count > MAX_COUNT_SKIP_SEEK)) {
1900*44704f69SBart Van Assche pr2serr(ME "bad argument to 'count='\n");
1901*44704f69SBart Van Assche return SG_LIB_SYNTAX_ERROR;
1902*44704f69SBart Van Assche }
1903*44704f69SBart Van Assche } /* treat 'count=-1' as calculate count (same as not given) */
1904*44704f69SBart Van Assche } else if (0 == strcmp(key, "dio")) {
1905*44704f69SBart Van Assche oflag.dio = !! sg_get_num(buf);
1906*44704f69SBart Van Assche iflag.dio = oflag.dio;
1907*44704f69SBart Van Assche } else if (0 == strcmp(key, "fua")) {
1908*44704f69SBart Van Assche t = sg_get_num(buf);
1909*44704f69SBart Van Assche oflag.fua = !! (t & 1);
1910*44704f69SBart Van Assche iflag.fua = !! (t & 2);
1911*44704f69SBart Van Assche } else if (0 == strcmp(key, "ibs")) {
1912*44704f69SBart Van Assche ibs = sg_get_num(buf);
1913*44704f69SBart Van Assche if ((ibs < 0) || (ibs > MAX_BPT_VALUE)) {
1914*44704f69SBart Van Assche pr2serr(ME "bad argument to 'ibs='\n");
1915*44704f69SBart Van Assche return SG_LIB_SYNTAX_ERROR;
1916*44704f69SBart Van Assche }
1917*44704f69SBart Van Assche } else if (strcmp(key, "if") == 0) {
1918*44704f69SBart Van Assche if ('\0' != inf[0]) {
1919*44704f69SBart Van Assche pr2serr("Second IFILE argument??\n");
1920*44704f69SBart Van Assche return SG_LIB_SYNTAX_ERROR;
1921*44704f69SBart Van Assche } else {
1922*44704f69SBart Van Assche memcpy(inf, buf, INOUTF_SZ - 1);
1923*44704f69SBart Van Assche inf[INOUTF_SZ - 1] = '\0';
1924*44704f69SBart Van Assche }
1925*44704f69SBart Van Assche } else if (0 == strcmp(key, "iflag")) {
1926*44704f69SBart Van Assche if (process_flags(buf, &iflag)) {
1927*44704f69SBart Van Assche pr2serr(ME "bad argument to 'iflag='\n");
1928*44704f69SBart Van Assche return SG_LIB_SYNTAX_ERROR;
1929*44704f69SBart Van Assche }
1930*44704f69SBart Van Assche } else if (0 == strcmp(key, "obs")) {
1931*44704f69SBart Van Assche obs = sg_get_num(buf);
1932*44704f69SBart Van Assche if ((obs < 0) || (obs > MAX_BPT_VALUE)) {
1933*44704f69SBart Van Assche pr2serr(ME "bad argument to 'obs='\n");
1934*44704f69SBart Van Assche return SG_LIB_SYNTAX_ERROR;
1935*44704f69SBart Van Assche }
1936*44704f69SBart Van Assche } else if (0 == strcmp(key, "odir")) {
1937*44704f69SBart Van Assche iflag.direct = !! sg_get_num(buf);
1938*44704f69SBart Van Assche oflag.direct = iflag.direct;
1939*44704f69SBart Van Assche } else if (strcmp(key, "of") == 0) {
1940*44704f69SBart Van Assche if ('\0' != outf[0]) {
1941*44704f69SBart Van Assche pr2serr("Second OFILE argument??\n");
1942*44704f69SBart Van Assche return SG_LIB_CONTRADICT;
1943*44704f69SBart Van Assche } else {
1944*44704f69SBart Van Assche memcpy(outf, buf, INOUTF_SZ - 1);
1945*44704f69SBart Van Assche outf[INOUTF_SZ - 1] = '\0';
1946*44704f69SBart Van Assche }
1947*44704f69SBart Van Assche } else if (strcmp(key, "of2") == 0) {
1948*44704f69SBart Van Assche if ('\0' != out2f[0]) {
1949*44704f69SBart Van Assche pr2serr("Second OFILE2 argument??\n");
1950*44704f69SBart Van Assche return SG_LIB_CONTRADICT;
1951*44704f69SBart Van Assche } else {
1952*44704f69SBart Van Assche memcpy(out2f, buf, INOUTF_SZ - 1);
1953*44704f69SBart Van Assche out2f[INOUTF_SZ - 1] = '\0';
1954*44704f69SBart Van Assche }
1955*44704f69SBart Van Assche } else if (0 == strcmp(key, "oflag")) {
1956*44704f69SBart Van Assche if (process_flags(buf, &oflag)) {
1957*44704f69SBart Van Assche pr2serr(ME "bad argument to 'oflag='\n");
1958*44704f69SBart Van Assche return SG_LIB_SYNTAX_ERROR;
1959*44704f69SBart Van Assche }
1960*44704f69SBart Van Assche } else if (0 == strcmp(key, "retries")) {
1961*44704f69SBart Van Assche iflag.retries = sg_get_num(buf);
1962*44704f69SBart Van Assche oflag.retries = iflag.retries;
1963*44704f69SBart Van Assche if (-1 == iflag.retries) {
1964*44704f69SBart Van Assche pr2serr(ME "bad argument to 'retries='\n");
1965*44704f69SBart Van Assche return SG_LIB_SYNTAX_ERROR;
1966*44704f69SBart Van Assche }
1967*44704f69SBart Van Assche } else if (0 == strcmp(key, "seek")) {
1968*44704f69SBart Van Assche seek = sg_get_llnum(buf);
1969*44704f69SBart Van Assche if ((seek < 0) || (seek > MAX_COUNT_SKIP_SEEK)) {
1970*44704f69SBart Van Assche pr2serr(ME "bad argument to 'seek='\n");
1971*44704f69SBart Van Assche return SG_LIB_SYNTAX_ERROR;
1972*44704f69SBart Van Assche }
1973*44704f69SBart Van Assche } else if (0 == strcmp(key, "skip")) {
1974*44704f69SBart Van Assche skip = sg_get_llnum(buf);
1975*44704f69SBart Van Assche if ((skip < 0) || (skip > MAX_COUNT_SKIP_SEEK)) {
1976*44704f69SBart Van Assche pr2serr(ME "bad argument to 'skip='\n");
1977*44704f69SBart Van Assche return SG_LIB_SYNTAX_ERROR;
1978*44704f69SBart Van Assche }
1979*44704f69SBart Van Assche } else if (0 == strcmp(key, "sync"))
1980*44704f69SBart Van Assche do_sync = !! sg_get_num(buf);
1981*44704f69SBart Van Assche else if (0 == strcmp(key, "time")) {
1982*44704f69SBart Van Assche const char * cp = strchr(buf, ',');
1983*44704f69SBart Van Assche
1984*44704f69SBart Van Assche do_time = !! sg_get_num(buf);
1985*44704f69SBart Van Assche if (cp) {
1986*44704f69SBart Van Assche n = sg_get_num(cp + 1);
1987*44704f69SBart Van Assche if (n < 0) {
1988*44704f69SBart Van Assche pr2serr(ME "bad argument to 'time=0|1,TO'\n");
1989*44704f69SBart Van Assche return SG_LIB_SYNTAX_ERROR;
1990*44704f69SBart Van Assche }
1991*44704f69SBart Van Assche cmd_timeout = n ? (n * 1000) : DEF_TIMEOUT;
1992*44704f69SBart Van Assche }
1993*44704f69SBart Van Assche } else if (0 == strncmp(key, "verb", 4))
1994*44704f69SBart Van Assche verbose = sg_get_num(buf);
1995*44704f69SBart Van Assche else if ((keylen > 1) && ('-' == key[0]) && ('-' != key[1])) {
1996*44704f69SBart Van Assche res = 0;
1997*44704f69SBart Van Assche n = num_chs_in_str(key + 1, keylen - 1, 'd');
1998*44704f69SBart Van Assche dry_run += n;
1999*44704f69SBart Van Assche res += n;
2000*44704f69SBart Van Assche n = num_chs_in_str(key + 1, keylen - 1, 'h');
2001*44704f69SBart Van Assche if (n > 0) {
2002*44704f69SBart Van Assche usage();
2003*44704f69SBart Van Assche return 0;
2004*44704f69SBart Van Assche }
2005*44704f69SBart Van Assche n = num_chs_in_str(key + 1, keylen - 1, 'p');
2006*44704f69SBart Van Assche progress += n;
2007*44704f69SBart Van Assche res += n;
2008*44704f69SBart Van Assche n = num_chs_in_str(key + 1, keylen - 1, 'v');
2009*44704f69SBart Van Assche if (n > 0)
2010*44704f69SBart Van Assche verbose_given = true;
2011*44704f69SBart Van Assche verbose += n;
2012*44704f69SBart Van Assche res += n;
2013*44704f69SBart Van Assche n = num_chs_in_str(key + 1, keylen - 1, 'V');
2014*44704f69SBart Van Assche if (n > 0)
2015*44704f69SBart Van Assche version_given = true;
2016*44704f69SBart Van Assche res += n;
2017*44704f69SBart Van Assche n = num_chs_in_str(key + 1, keylen - 1, 'x');
2018*44704f69SBart Van Assche if (n > 0)
2019*44704f69SBart Van Assche do_verify = true;
2020*44704f69SBart Van Assche res += n;
2021*44704f69SBart Van Assche if (res < (keylen - 1)) {
2022*44704f69SBart Van Assche pr2serr("Unrecognised short option in '%s', try '--help'\n",
2023*44704f69SBart Van Assche key);
2024*44704f69SBart Van Assche return SG_LIB_SYNTAX_ERROR;
2025*44704f69SBart Van Assche }
2026*44704f69SBart Van Assche } else if ((0 == strncmp(key, "--dry-run", 9)) ||
2027*44704f69SBart Van Assche (0 == strncmp(key, "--dry_run", 9)))
2028*44704f69SBart Van Assche ++dry_run;
2029*44704f69SBart Van Assche else if ((0 == strncmp(key, "--help", 6)) ||
2030*44704f69SBart Van Assche (0 == strcmp(key, "-?"))) {
2031*44704f69SBart Van Assche usage();
2032*44704f69SBart Van Assche return 0;
2033*44704f69SBart Van Assche } else if (0 == strncmp(key, "--progress", 10))
2034*44704f69SBart Van Assche ++progress;
2035*44704f69SBart Van Assche else if (0 == strncmp(key, "--verb", 6)) {
2036*44704f69SBart Van Assche verbose_given = true;
2037*44704f69SBart Van Assche ++verbose;
2038*44704f69SBart Van Assche } else if (0 == strncmp(key, "--veri", 6))
2039*44704f69SBart Van Assche do_verify = true;
2040*44704f69SBart Van Assche else if (0 == strncmp(key, "--vers", 6))
2041*44704f69SBart Van Assche version_given = true;
2042*44704f69SBart Van Assche else {
2043*44704f69SBart Van Assche pr2serr("Unrecognized option '%s'\n", key);
2044*44704f69SBart Van Assche pr2serr("For more information use '--help'\n");
2045*44704f69SBart Van Assche return SG_LIB_SYNTAX_ERROR;
2046*44704f69SBart Van Assche }
2047*44704f69SBart Van Assche }
2048*44704f69SBart Van Assche #ifdef DEBUG
2049*44704f69SBart Van Assche pr2serr("In DEBUG mode, ");
2050*44704f69SBart Van Assche if (verbose_given && version_given) {
2051*44704f69SBart Van Assche pr2serr("but override: '-vV' given, zero verbose and continue\n");
2052*44704f69SBart Van Assche verbose_given = false;
2053*44704f69SBart Van Assche version_given = false;
2054*44704f69SBart Van Assche verbose = 0;
2055*44704f69SBart Van Assche } else if (! verbose_given) {
2056*44704f69SBart Van Assche pr2serr("set '-vv'\n");
2057*44704f69SBart Van Assche verbose = 2;
2058*44704f69SBart Van Assche } else
2059*44704f69SBart Van Assche pr2serr("keep verbose=%d\n", verbose);
2060*44704f69SBart Van Assche #else
2061*44704f69SBart Van Assche if (verbose_given && version_given)
2062*44704f69SBart Van Assche pr2serr("Not in DEBUG mode, so '-vV' has no special action\n");
2063*44704f69SBart Van Assche #endif
2064*44704f69SBart Van Assche if (version_given) {
2065*44704f69SBart Van Assche pr2serr(ME "version: %s\n", version_str);
2066*44704f69SBart Van Assche return 0;
2067*44704f69SBart Van Assche }
2068*44704f69SBart Van Assche if (progress > 0 && !do_time)
2069*44704f69SBart Van Assche do_time = true;
2070*44704f69SBart Van Assche if (argc < 2) {
2071*44704f69SBart Van Assche pr2serr("Won't default both IFILE to stdin _and_ OFILE to stdout\n");
2072*44704f69SBart Van Assche pr2serr("For more information use '--help'\n");
2073*44704f69SBart Van Assche return SG_LIB_CONTRADICT;
2074*44704f69SBart Van Assche }
2075*44704f69SBart Van Assche if (blk_sz <= 0) {
2076*44704f69SBart Van Assche blk_sz = DEF_BLOCK_SIZE;
2077*44704f69SBart Van Assche pr2serr("Assume default 'bs' ((logical) block size) of %d bytes\n",
2078*44704f69SBart Van Assche blk_sz);
2079*44704f69SBart Van Assche }
2080*44704f69SBart Van Assche if ((ibs && (ibs != blk_sz)) || (obs && (obs != blk_sz))) {
2081*44704f69SBart Van Assche pr2serr("If 'ibs' or 'obs' given must be same as 'bs'\n");
2082*44704f69SBart Van Assche pr2serr("For more information use '--help'\n");
2083*44704f69SBart Van Assche return SG_LIB_CONTRADICT;
2084*44704f69SBart Van Assche }
2085*44704f69SBart Van Assche if ((skip < 0) || (seek < 0)) {
2086*44704f69SBart Van Assche pr2serr("skip and seek cannot be negative\n");
2087*44704f69SBart Van Assche return SG_LIB_CONTRADICT;
2088*44704f69SBart Van Assche }
2089*44704f69SBart Van Assche if (oflag.append && (seek > 0)) {
2090*44704f69SBart Van Assche pr2serr("Can't use both append and seek switches\n");
2091*44704f69SBart Van Assche return SG_LIB_CONTRADICT;
2092*44704f69SBart Van Assche }
2093*44704f69SBart Van Assche if ((bpt < 1) || (bpt > MAX_BPT_VALUE)) {
2094*44704f69SBart Van Assche pr2serr("bpt must be > 0 and <= %d\n", MAX_BPT_VALUE);
2095*44704f69SBart Van Assche return SG_LIB_SYNTAX_ERROR;
2096*44704f69SBart Van Assche }
2097*44704f69SBart Van Assche if (iflag.sparse)
2098*44704f69SBart Van Assche pr2serr("sparse flag ignored for iflag\n");
2099*44704f69SBart Van Assche
2100*44704f69SBart Van Assche /* defaulting transfer size to 128*2048 for CD/DVDs is too large
2101*44704f69SBart Van Assche for the block layer in lk 2.6 and results in an EIO on the
2102*44704f69SBart Van Assche SG_IO ioctl. So reduce it in that case. */
2103*44704f69SBart Van Assche if ((blk_sz >= 2048) && (! bpt_given))
2104*44704f69SBart Van Assche bpt = DEF_BLOCKS_PER_2048TRANSFER;
2105*44704f69SBart Van Assche #ifdef DEBUG
2106*44704f69SBart Van Assche pr2serr(ME "if=%s skip=%" PRId64 " of=%s seek=%" PRId64 " count=%" PRId64
2107*44704f69SBart Van Assche "\n", inf, skip, outf, seek, dd_count);
2108*44704f69SBart Van Assche #endif
2109*44704f69SBart Van Assche install_handler(SIGINT, interrupt_handler);
2110*44704f69SBart Van Assche install_handler(SIGQUIT, interrupt_handler);
2111*44704f69SBart Van Assche install_handler(SIGPIPE, interrupt_handler);
2112*44704f69SBart Van Assche install_handler(SIGUSR1, siginfo_handler);
2113*44704f69SBart Van Assche
2114*44704f69SBart Van Assche infd = STDIN_FILENO;
2115*44704f69SBart Van Assche outfd = STDOUT_FILENO;
2116*44704f69SBart Van Assche iflag.pdt = -1;
2117*44704f69SBart Van Assche oflag.pdt = -1;
2118*44704f69SBart Van Assche if (iflag.zero && iflag.ff) {
2119*44704f69SBart Van Assche ccp = "<addr_as_data>";
2120*44704f69SBart Van Assche cc2p = "addr_as_data";
2121*44704f69SBart Van Assche } else if (iflag.ff) {
2122*44704f69SBart Van Assche ccp = "<0xff bytes>";
2123*44704f69SBart Van Assche cc2p = "ff";
2124*44704f69SBart Van Assche } else if (iflag.random) {
2125*44704f69SBart Van Assche ccp = "<random>";
2126*44704f69SBart Van Assche cc2p = "random";
2127*44704f69SBart Van Assche #ifdef HAVE_GETRANDOM
2128*44704f69SBart Van Assche {
2129*44704f69SBart Van Assche ssize_t ssz = getrandom(&seed, sizeof(seed), GRND_NONBLOCK);
2130*44704f69SBart Van Assche
2131*44704f69SBart Van Assche if (ssz < (ssize_t)sizeof(seed)) {
2132*44704f69SBart Van Assche pr2serr("getrandom() failed, ret=%d\n", (int)ssz);
2133*44704f69SBart Van Assche seed = (long)time(NULL);
2134*44704f69SBart Van Assche }
2135*44704f69SBart Van Assche }
2136*44704f69SBart Van Assche #else
2137*44704f69SBart Van Assche seed = (long)time(NULL); /* use seconds since epoch as proxy */
2138*44704f69SBart Van Assche #endif
2139*44704f69SBart Van Assche if (verbose > 1)
2140*44704f69SBart Van Assche pr2serr("seed=%ld\n", seed);
2141*44704f69SBart Van Assche #ifdef HAVE_SRAND48_R
2142*44704f69SBart Van Assche srand48_r(seed, &drand);
2143*44704f69SBart Van Assche #else
2144*44704f69SBart Van Assche srand48(seed);
2145*44704f69SBart Van Assche #endif
2146*44704f69SBart Van Assche } else if (iflag.zero) {
2147*44704f69SBart Van Assche ccp = "<zero bytes>";
2148*44704f69SBart Van Assche cc2p = "00";
2149*44704f69SBart Van Assche }
2150*44704f69SBart Van Assche if (ccp) {
2151*44704f69SBart Van Assche if (inf[0]) {
2152*44704f69SBart Van Assche pr2serr("iflag=%s and if=%s contradict\n", cc2p, inf);
2153*44704f69SBart Van Assche return SG_LIB_CONTRADICT;
2154*44704f69SBart Van Assche }
2155*44704f69SBart Van Assche in_type = FT_RANDOM_0_FF;
2156*44704f69SBart Van Assche strcpy(inf, ccp);
2157*44704f69SBart Van Assche infd = -1;
2158*44704f69SBart Van Assche } else if (inf[0] && ('-' != inf[0])) {
2159*44704f69SBart Van Assche infd = open_if(inf, skip, bpt, &iflag, &in_type, verbose);
2160*44704f69SBart Van Assche if (infd < 0)
2161*44704f69SBart Van Assche return -infd;
2162*44704f69SBart Van Assche }
2163*44704f69SBart Van Assche
2164*44704f69SBart Van Assche if (outf[0] && ('-' != outf[0])) {
2165*44704f69SBart Van Assche outfd = open_of(outf, seek, bpt, &oflag, &out_type, verbose);
2166*44704f69SBart Van Assche if (outfd < -1)
2167*44704f69SBart Van Assche return -outfd;
2168*44704f69SBart Van Assche }
2169*44704f69SBart Van Assche if (do_verify) {
2170*44704f69SBart Van Assche if (! (FT_SG & out_type)) {
2171*44704f69SBart Van Assche pr2serr("--verify only supported when OFILE is a sg device or "
2172*44704f69SBart Van Assche "oflag=sgio\n");
2173*44704f69SBart Van Assche ret = SG_LIB_CONTRADICT;
2174*44704f69SBart Van Assche goto bypass_copy;
2175*44704f69SBart Van Assche }
2176*44704f69SBart Van Assche if (oflag.sparse) {
2177*44704f69SBart Van Assche pr2serr("--verify cannot be used with oflag=sparse\n");
2178*44704f69SBart Van Assche ret = SG_LIB_CONTRADICT;
2179*44704f69SBart Van Assche goto bypass_copy;
2180*44704f69SBart Van Assche }
2181*44704f69SBart Van Assche }
2182*44704f69SBart Van Assche if (cdl_given && (! cdbsz_given)) {
2183*44704f69SBart Van Assche bool changed = false;
2184*44704f69SBart Van Assche
2185*44704f69SBart Van Assche if ((iflag.cdbsz < 16) && (iflag.cdl > 0)) {
2186*44704f69SBart Van Assche iflag.cdbsz = 16;
2187*44704f69SBart Van Assche changed = true;
2188*44704f69SBart Van Assche }
2189*44704f69SBart Van Assche if ((oflag.cdbsz < 16) && (! do_verify) && (oflag.cdl > 0)) {
2190*44704f69SBart Van Assche oflag.cdbsz = 16;
2191*44704f69SBart Van Assche changed = true;
2192*44704f69SBart Van Assche }
2193*44704f69SBart Van Assche if (changed)
2194*44704f69SBart Van Assche pr2serr(">> increasing cdbsz to 16 due to cdl > 0\n");
2195*44704f69SBart Van Assche }
2196*44704f69SBart Van Assche if (out2f[0]) {
2197*44704f69SBart Van Assche out2_type = dd_filetype(out2f);
2198*44704f69SBart Van Assche if ((out2fd = open(out2f, O_WRONLY | O_CREAT, 0666)) < 0) {
2199*44704f69SBart Van Assche res = errno;
2200*44704f69SBart Van Assche snprintf(ebuff, EBUFF_SZ,
2201*44704f69SBart Van Assche ME "could not open %s for writing", out2f);
2202*44704f69SBart Van Assche perror(ebuff);
2203*44704f69SBart Van Assche return res;
2204*44704f69SBart Van Assche }
2205*44704f69SBart Van Assche } else
2206*44704f69SBart Van Assche out2fd = -1;
2207*44704f69SBart Van Assche
2208*44704f69SBart Van Assche if ((STDIN_FILENO == infd) && (STDOUT_FILENO == outfd)) {
2209*44704f69SBart Van Assche pr2serr("Can't have both 'if' as stdin _and_ 'of' as stdout\n");
2210*44704f69SBart Van Assche pr2serr("For more information use '--help'\n");
2211*44704f69SBart Van Assche return SG_LIB_CONTRADICT;
2212*44704f69SBart Van Assche }
2213*44704f69SBart Van Assche if (oflag.sparse) {
2214*44704f69SBart Van Assche if (STDOUT_FILENO == outfd) {
2215*44704f69SBart Van Assche pr2serr("oflag=sparse needs seekable output file\n");
2216*44704f69SBart Van Assche return SG_LIB_CONTRADICT;
2217*44704f69SBart Van Assche }
2218*44704f69SBart Van Assche }
2219*44704f69SBart Van Assche
2220*44704f69SBart Van Assche if ((dd_count < 0) || ((verbose > 0) && (0 == dd_count))) {
2221*44704f69SBart Van Assche in_num_sect = -1;
2222*44704f69SBart Van Assche in_sect_sz = -1;
2223*44704f69SBart Van Assche if (FT_SG & in_type) {
2224*44704f69SBart Van Assche res = scsi_read_capacity(infd, &in_num_sect, &in_sect_sz);
2225*44704f69SBart Van Assche if (SG_LIB_CAT_UNIT_ATTENTION == res) {
2226*44704f69SBart Van Assche pr2serr("Unit attention (readcap in), continuing\n");
2227*44704f69SBart Van Assche res = scsi_read_capacity(infd, &in_num_sect, &in_sect_sz);
2228*44704f69SBart Van Assche } else if (SG_LIB_CAT_ABORTED_COMMAND == res) {
2229*44704f69SBart Van Assche pr2serr("Aborted command (readcap in), continuing\n");
2230*44704f69SBart Van Assche res = scsi_read_capacity(infd, &in_num_sect, &in_sect_sz);
2231*44704f69SBart Van Assche }
2232*44704f69SBart Van Assche if (0 != res) {
2233*44704f69SBart Van Assche if (res == SG_LIB_CAT_INVALID_OP)
2234*44704f69SBart Van Assche pr2serr("read capacity not supported on %s\n", inf);
2235*44704f69SBart Van Assche else if (res == SG_LIB_CAT_NOT_READY)
2236*44704f69SBart Van Assche pr2serr("read capacity failed on %s - not ready\n", inf);
2237*44704f69SBart Van Assche else
2238*44704f69SBart Van Assche pr2serr("Unable to read capacity on %s\n", inf);
2239*44704f69SBart Van Assche in_num_sect = -1;
2240*44704f69SBart Van Assche } else if (in_sect_sz != blk_sz)
2241*44704f69SBart Van Assche pr2serr(">> warning: logical block size on %s confusion: "
2242*44704f69SBart Van Assche "bs=%d, device claims=%d\n", inf, blk_sz, in_sect_sz);
2243*44704f69SBart Van Assche } else if (FT_BLOCK & in_type) {
2244*44704f69SBart Van Assche if (0 != read_blkdev_capacity(infd, &in_num_sect, &in_sect_sz)) {
2245*44704f69SBart Van Assche pr2serr("Unable to read block capacity on %s\n", inf);
2246*44704f69SBart Van Assche in_num_sect = -1;
2247*44704f69SBart Van Assche }
2248*44704f69SBart Van Assche if (blk_sz != in_sect_sz) {
2249*44704f69SBart Van Assche pr2serr("logical block size on %s confusion: bs=%d, device "
2250*44704f69SBart Van Assche "claims=%d\n", inf, blk_sz, in_sect_sz);
2251*44704f69SBart Van Assche in_num_sect = -1;
2252*44704f69SBart Van Assche }
2253*44704f69SBart Van Assche }
2254*44704f69SBart Van Assche if (in_num_sect > skip)
2255*44704f69SBart Van Assche in_num_sect -= skip;
2256*44704f69SBart Van Assche
2257*44704f69SBart Van Assche out_num_sect = -1;
2258*44704f69SBart Van Assche out_sect_sz = -1;
2259*44704f69SBart Van Assche if (FT_SG & out_type) {
2260*44704f69SBart Van Assche res = scsi_read_capacity(outfd, &out_num_sect, &out_sect_sz);
2261*44704f69SBart Van Assche if (SG_LIB_CAT_UNIT_ATTENTION == res) {
2262*44704f69SBart Van Assche pr2serr("Unit attention (readcap out), continuing\n");
2263*44704f69SBart Van Assche res = scsi_read_capacity(outfd, &out_num_sect, &out_sect_sz);
2264*44704f69SBart Van Assche } else if (SG_LIB_CAT_ABORTED_COMMAND == res) {
2265*44704f69SBart Van Assche pr2serr("Aborted command (readcap out), continuing\n");
2266*44704f69SBart Van Assche res = scsi_read_capacity(outfd, &out_num_sect, &out_sect_sz);
2267*44704f69SBart Van Assche }
2268*44704f69SBart Van Assche if (0 != res) {
2269*44704f69SBart Van Assche if (res == SG_LIB_CAT_INVALID_OP)
2270*44704f69SBart Van Assche pr2serr("read capacity not supported on %s\n", outf);
2271*44704f69SBart Van Assche else
2272*44704f69SBart Van Assche pr2serr("Unable to read capacity on %s\n", outf);
2273*44704f69SBart Van Assche out_num_sect = -1;
2274*44704f69SBart Van Assche } else if (blk_sz != out_sect_sz)
2275*44704f69SBart Van Assche pr2serr(">> warning: logical block size on %s confusion: "
2276*44704f69SBart Van Assche "bs=%d, device claims=%d\n", outf, blk_sz,
2277*44704f69SBart Van Assche out_sect_sz);
2278*44704f69SBart Van Assche } else if (FT_BLOCK & out_type) {
2279*44704f69SBart Van Assche if (0 != read_blkdev_capacity(outfd, &out_num_sect,
2280*44704f69SBart Van Assche &out_sect_sz)) {
2281*44704f69SBart Van Assche pr2serr("Unable to read block capacity on %s\n", outf);
2282*44704f69SBart Van Assche out_num_sect = -1;
2283*44704f69SBart Van Assche } else if (blk_sz != out_sect_sz) {
2284*44704f69SBart Van Assche pr2serr("logical block size on %s confusion: bs=%d, device "
2285*44704f69SBart Van Assche "claims=%d\n", outf, blk_sz, out_sect_sz);
2286*44704f69SBart Van Assche out_num_sect = -1;
2287*44704f69SBart Van Assche }
2288*44704f69SBart Van Assche }
2289*44704f69SBart Van Assche if (out_num_sect > seek)
2290*44704f69SBart Van Assche out_num_sect -= seek;
2291*44704f69SBart Van Assche #ifdef DEBUG
2292*44704f69SBart Van Assche pr2serr("Start of loop, count=%" PRId64 ", in_num_sect=%" PRId64
2293*44704f69SBart Van Assche ", out_num_sect=%" PRId64 "\n", dd_count, in_num_sect,
2294*44704f69SBart Van Assche out_num_sect);
2295*44704f69SBart Van Assche #endif
2296*44704f69SBart Van Assche if (dd_count < 0) {
2297*44704f69SBart Van Assche if (in_num_sect > 0) {
2298*44704f69SBart Van Assche if (out_num_sect > 0)
2299*44704f69SBart Van Assche dd_count = (in_num_sect > out_num_sect) ? out_num_sect :
2300*44704f69SBart Van Assche in_num_sect;
2301*44704f69SBart Van Assche else
2302*44704f69SBart Van Assche dd_count = in_num_sect;
2303*44704f69SBart Van Assche } else
2304*44704f69SBart Van Assche dd_count = out_num_sect;
2305*44704f69SBart Van Assche }
2306*44704f69SBart Van Assche }
2307*44704f69SBart Van Assche
2308*44704f69SBart Van Assche if (dd_count < 0) {
2309*44704f69SBart Van Assche pr2serr("Couldn't calculate count, please give one\n");
2310*44704f69SBart Van Assche return SG_LIB_CAT_OTHER;
2311*44704f69SBart Van Assche }
2312*44704f69SBart Van Assche if (! cdbsz_given) {
2313*44704f69SBart Van Assche if ((FT_SG & in_type) && (MAX_SCSI_CDBSZ != iflag.cdbsz) &&
2314*44704f69SBart Van Assche (((dd_count + skip) > UINT_MAX) || (bpt > USHRT_MAX))) {
2315*44704f69SBart Van Assche pr2serr("Note: SCSI command size increased to 16 bytes (for "
2316*44704f69SBart Van Assche "'if')\n");
2317*44704f69SBart Van Assche iflag.cdbsz = MAX_SCSI_CDBSZ;
2318*44704f69SBart Van Assche }
2319*44704f69SBart Van Assche if ((FT_SG & out_type) && (MAX_SCSI_CDBSZ != oflag.cdbsz) &&
2320*44704f69SBart Van Assche (((dd_count + seek) > UINT_MAX) || (bpt > USHRT_MAX))) {
2321*44704f69SBart Van Assche pr2serr("Note: SCSI command size increased to 16 bytes (for "
2322*44704f69SBart Van Assche "'of')\n");
2323*44704f69SBart Van Assche oflag.cdbsz = MAX_SCSI_CDBSZ;
2324*44704f69SBart Van Assche }
2325*44704f69SBart Van Assche }
2326*44704f69SBart Van Assche
2327*44704f69SBart Van Assche if (iflag.dio || iflag.direct || oflag.direct || (FT_RAW & in_type) ||
2328*44704f69SBart Van Assche (FT_RAW & out_type)) { /* want heap buffer aligned to page_size */
2329*44704f69SBart Van Assche
2330*44704f69SBart Van Assche wrkPos = sg_memalign(blk_sz * bpt, 0, &wrkBuff, false);
2331*44704f69SBart Van Assche if (NULL == wrkPos) {
2332*44704f69SBart Van Assche pr2serr("sg_memalign: error, out of memory?\n");
2333*44704f69SBart Van Assche return sg_convert_errno(ENOMEM);
2334*44704f69SBart Van Assche }
2335*44704f69SBart Van Assche } else {
2336*44704f69SBart Van Assche wrkPos = sg_memalign(blk_sz * bpt, 0, &wrkBuff, false);
2337*44704f69SBart Van Assche if (0 == wrkPos) {
2338*44704f69SBart Van Assche pr2serr("Not enough user memory\n");
2339*44704f69SBart Van Assche return sg_convert_errno(ENOMEM);
2340*44704f69SBart Van Assche }
2341*44704f69SBart Van Assche }
2342*44704f69SBart Van Assche
2343*44704f69SBart Van Assche blocks_per = bpt;
2344*44704f69SBart Van Assche #ifdef DEBUG
2345*44704f69SBart Van Assche pr2serr("Start of loop, count=%" PRId64 ", blocks_per=%d\n", dd_count,
2346*44704f69SBart Van Assche blocks_per);
2347*44704f69SBart Van Assche #endif
2348*44704f69SBart Van Assche if (do_time) {
2349*44704f69SBart Van Assche start_tm.tv_sec = 0;
2350*44704f69SBart Van Assche start_tm.tv_usec = 0;
2351*44704f69SBart Van Assche gettimeofday(&start_tm, NULL);
2352*44704f69SBart Van Assche start_tm_valid = true;
2353*44704f69SBart Van Assche }
2354*44704f69SBart Van Assche req_count = dd_count;
2355*44704f69SBart Van Assche
2356*44704f69SBart Van Assche if (dry_run > 0) {
2357*44704f69SBart Van Assche pr2serr("Since --dry-run option given, bypassing copy\n");
2358*44704f69SBart Van Assche goto bypass_copy;
2359*44704f69SBart Van Assche }
2360*44704f69SBart Van Assche
2361*44704f69SBart Van Assche /* <<< main loop that does the copy >>> */
2362*44704f69SBart Van Assche while (dd_count > 0) {
2363*44704f69SBart Van Assche bytes_read = 0;
2364*44704f69SBart Van Assche bytes_of = 0;
2365*44704f69SBart Van Assche bytes_of2 = 0;
2366*44704f69SBart Van Assche penult_sparse_skip = sparse_skip;
2367*44704f69SBart Van Assche penult_blocks = penult_sparse_skip ? blocks : 0;
2368*44704f69SBart Van Assche sparse_skip = false;
2369*44704f69SBart Van Assche blocks = (dd_count > blocks_per) ? blocks_per : dd_count;
2370*44704f69SBart Van Assche if (FT_SG & in_type) {
2371*44704f69SBart Van Assche dio_tmp = iflag.dio;
2372*44704f69SBart Van Assche res = sg_read(infd, wrkPos, blocks, skip, blk_sz, &iflag,
2373*44704f69SBart Van Assche &dio_tmp, &blks_read);
2374*44704f69SBart Van Assche if (-2 == res) { /* ENOMEM, find what's available+try that */
2375*44704f69SBart Van Assche if (ioctl(infd, SG_GET_RESERVED_SIZE, &buf_sz) < 0) {
2376*44704f69SBart Van Assche perror("RESERVED_SIZE ioctls failed");
2377*44704f69SBart Van Assche ret = res;
2378*44704f69SBart Van Assche break;
2379*44704f69SBart Van Assche }
2380*44704f69SBart Van Assche if (buf_sz < MIN_RESERVED_SIZE)
2381*44704f69SBart Van Assche buf_sz = MIN_RESERVED_SIZE;
2382*44704f69SBart Van Assche blocks_per = (buf_sz + blk_sz - 1) / blk_sz;
2383*44704f69SBart Van Assche if (blocks_per < blocks) {
2384*44704f69SBart Van Assche blocks = blocks_per;
2385*44704f69SBart Van Assche pr2serr("Reducing read to %d blocks per loop\n",
2386*44704f69SBart Van Assche blocks_per);
2387*44704f69SBart Van Assche res = sg_read(infd, wrkPos, blocks, skip, blk_sz,
2388*44704f69SBart Van Assche &iflag, &dio_tmp, &blks_read);
2389*44704f69SBart Van Assche }
2390*44704f69SBart Van Assche }
2391*44704f69SBart Van Assche if (res) {
2392*44704f69SBart Van Assche pr2serr("sg_read failed,%s at or after lba=%" PRId64 " [0x%"
2393*44704f69SBart Van Assche PRIx64 "]\n", ((-2 == res) ?
2394*44704f69SBart Van Assche " try reducing bpt," : ""), skip, skip);
2395*44704f69SBart Van Assche ret = res;
2396*44704f69SBart Van Assche break;
2397*44704f69SBart Van Assche } else {
2398*44704f69SBart Van Assche if (blks_read < blocks) {
2399*44704f69SBart Van Assche dd_count = 0; /* force exit after write */
2400*44704f69SBart Van Assche blocks = blks_read;
2401*44704f69SBart Van Assche }
2402*44704f69SBart Van Assche in_full += blocks;
2403*44704f69SBart Van Assche if (iflag.dio && (! dio_tmp))
2404*44704f69SBart Van Assche dio_incomplete_count++;
2405*44704f69SBart Van Assche }
2406*44704f69SBart Van Assche } else if (FT_RANDOM_0_FF == in_type) {
2407*44704f69SBart Van Assche int j;
2408*44704f69SBart Van Assche
2409*44704f69SBart Van Assche res = blocks * blk_sz;
2410*44704f69SBart Van Assche if (iflag.zero && iflag.ff && (blk_sz >= 4)) {
2411*44704f69SBart Van Assche uint32_t pos = (uint32_t)skip;
2412*44704f69SBart Van Assche uint32_t off;
2413*44704f69SBart Van Assche
2414*44704f69SBart Van Assche for (k = 0, off = 0; k < blocks; ++k, off += blk_sz, ++pos) {
2415*44704f69SBart Van Assche for (j = 0; j < (blk_sz - 3); j += 4)
2416*44704f69SBart Van Assche sg_put_unaligned_be32(pos, wrkPos + off + j);
2417*44704f69SBart Van Assche }
2418*44704f69SBart Van Assche } else if (iflag.zero)
2419*44704f69SBart Van Assche memset(wrkPos, 0, res);
2420*44704f69SBart Van Assche else if (iflag.ff)
2421*44704f69SBart Van Assche memset(wrkPos, 0xff, res);
2422*44704f69SBart Van Assche else {
2423*44704f69SBart Van Assche int kk, jj;
2424*44704f69SBart Van Assche const int jbump = sizeof(uint32_t);
2425*44704f69SBart Van Assche long rn;
2426*44704f69SBart Van Assche uint8_t * bp;
2427*44704f69SBart Van Assche
2428*44704f69SBart Van Assche bp = wrkPos;
2429*44704f69SBart Van Assche for (kk = 0; kk < blocks; ++kk, bp += blk_sz) {
2430*44704f69SBart Van Assche for (jj = 0; jj < blk_sz; jj += jbump) {
2431*44704f69SBart Van Assche /* mrand48 takes uniformly from [-2^31, 2^31) */
2432*44704f69SBart Van Assche #ifdef HAVE_SRAND48_R
2433*44704f69SBart Van Assche mrand48_r(&drand, &rn);
2434*44704f69SBart Van Assche #else
2435*44704f69SBart Van Assche rn = mrand48();
2436*44704f69SBart Van Assche #endif
2437*44704f69SBart Van Assche *((uint32_t *)(bp + jj)) = (uint32_t)rn;
2438*44704f69SBart Van Assche }
2439*44704f69SBart Van Assche }
2440*44704f69SBart Van Assche }
2441*44704f69SBart Van Assche bytes_read = res;
2442*44704f69SBart Van Assche in_full += blocks;
2443*44704f69SBart Van Assche } else {
2444*44704f69SBart Van Assche while (((res = read(infd, wrkPos, blocks * blk_sz)) < 0) &&
2445*44704f69SBart Van Assche ((EINTR == errno) || (EAGAIN == errno) ||
2446*44704f69SBart Van Assche (EBUSY == errno)))
2447*44704f69SBart Van Assche ;
2448*44704f69SBart Van Assche if (verbose > 2)
2449*44704f69SBart Van Assche pr2serr("read(unix): count=%d, res=%d\n", blocks * blk_sz,
2450*44704f69SBart Van Assche res);
2451*44704f69SBart Van Assche if (res < 0) {
2452*44704f69SBart Van Assche snprintf(ebuff, EBUFF_SZ, ME "reading, skip=%" PRId64 " ",
2453*44704f69SBart Van Assche skip);
2454*44704f69SBart Van Assche perror(ebuff);
2455*44704f69SBart Van Assche ret = -1;
2456*44704f69SBart Van Assche break;
2457*44704f69SBart Van Assche } else if (res < blocks * blk_sz) {
2458*44704f69SBart Van Assche dd_count = 0;
2459*44704f69SBart Van Assche blocks = res / blk_sz;
2460*44704f69SBart Van Assche if ((res % blk_sz) > 0) {
2461*44704f69SBart Van Assche blocks++;
2462*44704f69SBart Van Assche in_partial++;
2463*44704f69SBart Van Assche }
2464*44704f69SBart Van Assche }
2465*44704f69SBart Van Assche bytes_read = res;
2466*44704f69SBart Van Assche in_full += blocks;
2467*44704f69SBart Van Assche }
2468*44704f69SBart Van Assche
2469*44704f69SBart Van Assche if (0 == blocks)
2470*44704f69SBart Van Assche break; /* nothing read so leave loop */
2471*44704f69SBart Van Assche
2472*44704f69SBart Van Assche if (out2f[0]) {
2473*44704f69SBart Van Assche while (((res = write(out2fd, wrkPos, blocks * blk_sz)) < 0) &&
2474*44704f69SBart Van Assche ((EINTR == errno) || (EAGAIN == errno) ||
2475*44704f69SBart Van Assche (EBUSY == errno)))
2476*44704f69SBart Van Assche ;
2477*44704f69SBart Van Assche if (verbose > 2)
2478*44704f69SBart Van Assche pr2serr("write to of2: count=%d, res=%d\n", blocks * blk_sz,
2479*44704f69SBart Van Assche res);
2480*44704f69SBart Van Assche if (res < 0) {
2481*44704f69SBart Van Assche snprintf(ebuff, EBUFF_SZ, ME "writing to of2, seek=%" PRId64
2482*44704f69SBart Van Assche " ", seek);
2483*44704f69SBart Van Assche perror(ebuff);
2484*44704f69SBart Van Assche ret = -1;
2485*44704f69SBart Van Assche break;
2486*44704f69SBart Van Assche }
2487*44704f69SBart Van Assche bytes_of2 = res;
2488*44704f69SBart Van Assche }
2489*44704f69SBart Van Assche
2490*44704f69SBart Van Assche if (oflag.sparse && (dd_count > blocks) &&
2491*44704f69SBart Van Assche (! (FT_DEV_NULL & out_type))) {
2492*44704f69SBart Van Assche if (NULL == zeros_buff) {
2493*44704f69SBart Van Assche zeros_buff = sg_memalign(blocks * blk_sz, 0, &free_zeros_buff,
2494*44704f69SBart Van Assche false);
2495*44704f69SBart Van Assche if (NULL == zeros_buff) {
2496*44704f69SBart Van Assche pr2serr("zeros_buff sg_memalign failed\n");
2497*44704f69SBart Van Assche ret = -1;
2498*44704f69SBart Van Assche break;
2499*44704f69SBart Van Assche }
2500*44704f69SBart Van Assche }
2501*44704f69SBart Van Assche if (0 == memcmp(wrkPos, zeros_buff, blocks * blk_sz))
2502*44704f69SBart Van Assche sparse_skip = true;
2503*44704f69SBart Van Assche }
2504*44704f69SBart Van Assche if (sparse_skip) {
2505*44704f69SBart Van Assche if (FT_SG & out_type) {
2506*44704f69SBart Van Assche out_sparse_num += blocks;
2507*44704f69SBart Van Assche if (verbose > 2)
2508*44704f69SBart Van Assche pr2serr("sparse bypassing sg_write: seek blk=%" PRId64
2509*44704f69SBart Van Assche ", offset blks=%d\n", seek, blocks);
2510*44704f69SBart Van Assche } else if (FT_DEV_NULL & out_type)
2511*44704f69SBart Van Assche ;
2512*44704f69SBart Van Assche else {
2513*44704f69SBart Van Assche off64_t offset = (off64_t)blocks * blk_sz;
2514*44704f69SBart Van Assche off64_t off_res;
2515*44704f69SBart Van Assche
2516*44704f69SBart Van Assche if (verbose > 2)
2517*44704f69SBart Van Assche pr2serr("sparse bypassing write: seek=%" PRId64 ", rel "
2518*44704f69SBart Van Assche "offset=%" PRId64 "\n", (seek * blk_sz),
2519*44704f69SBart Van Assche (int64_t)offset);
2520*44704f69SBart Van Assche off_res = lseek64(outfd, offset, SEEK_CUR);
2521*44704f69SBart Van Assche if (off_res < 0) {
2522*44704f69SBart Van Assche pr2serr("sparse tried to bypass write: seek=%" PRId64
2523*44704f69SBart Van Assche ", rel offset=%" PRId64 " but ...\n",
2524*44704f69SBart Van Assche (seek * blk_sz), (int64_t)offset);
2525*44704f69SBart Van Assche perror("lseek64 on output");
2526*44704f69SBart Van Assche ret = SG_LIB_FILE_ERROR;
2527*44704f69SBart Van Assche break;
2528*44704f69SBart Van Assche } else if (verbose > 4)
2529*44704f69SBart Van Assche pr2serr("oflag=sparse lseek64 result=%" PRId64 "\n",
2530*44704f69SBart Van Assche (int64_t)off_res);
2531*44704f69SBart Van Assche out_sparse_num += blocks;
2532*44704f69SBart Van Assche }
2533*44704f69SBart Van Assche } else if (FT_SG & out_type) {
2534*44704f69SBart Van Assche dio_tmp = oflag.dio;
2535*44704f69SBart Van Assche retries_tmp = oflag.retries;
2536*44704f69SBart Van Assche first = true;
2537*44704f69SBart Van Assche while (1) {
2538*44704f69SBart Van Assche ret = sg_write(outfd, wrkPos, blocks, seek, blk_sz, &oflag,
2539*44704f69SBart Van Assche &dio_tmp);
2540*44704f69SBart Van Assche if ((0 == ret) || (SG_DD_BYPASS == ret))
2541*44704f69SBart Van Assche break;
2542*44704f69SBart Van Assche if ((SG_LIB_CAT_NOT_READY == ret) ||
2543*44704f69SBart Van Assche (SG_LIB_SYNTAX_ERROR == ret))
2544*44704f69SBart Van Assche break;
2545*44704f69SBart Van Assche else if ((-2 == ret) && first) {
2546*44704f69SBart Van Assche /* ENOMEM: find what's available and try that */
2547*44704f69SBart Van Assche if (ioctl(outfd, SG_GET_RESERVED_SIZE, &buf_sz) < 0) {
2548*44704f69SBart Van Assche perror("RESERVED_SIZE ioctls failed");
2549*44704f69SBart Van Assche break;
2550*44704f69SBart Van Assche }
2551*44704f69SBart Van Assche if (buf_sz < MIN_RESERVED_SIZE)
2552*44704f69SBart Van Assche buf_sz = MIN_RESERVED_SIZE;
2553*44704f69SBart Van Assche blocks_per = (buf_sz + blk_sz - 1) / blk_sz;
2554*44704f69SBart Van Assche if (blocks_per < blocks) {
2555*44704f69SBart Van Assche blocks = blocks_per;
2556*44704f69SBart Van Assche pr2serr("Reducing %s to %d blocks per loop\n",
2557*44704f69SBart Van Assche (do_verify ? "verify" : "write"), blocks);
2558*44704f69SBart Van Assche } else
2559*44704f69SBart Van Assche break;
2560*44704f69SBart Van Assche } else if ((SG_LIB_CAT_UNIT_ATTENTION == ret) && first) {
2561*44704f69SBart Van Assche if (--max_uas > 0)
2562*44704f69SBart Van Assche pr2serr("Unit attention, continuing (w)\n");
2563*44704f69SBart Van Assche else {
2564*44704f69SBart Van Assche pr2serr("Unit attention, too many (w)\n");
2565*44704f69SBart Van Assche break;
2566*44704f69SBart Van Assche }
2567*44704f69SBart Van Assche } else if ((SG_LIB_CAT_ABORTED_COMMAND == ret) && first) {
2568*44704f69SBart Van Assche if (--max_aborted > 0)
2569*44704f69SBart Van Assche pr2serr("Aborted command, continuing (w)\n");
2570*44704f69SBart Van Assche else {
2571*44704f69SBart Van Assche pr2serr("Aborted command, too many (w)\n");
2572*44704f69SBart Van Assche break;
2573*44704f69SBart Van Assche }
2574*44704f69SBart Van Assche } else if (ret < 0)
2575*44704f69SBart Van Assche break;
2576*44704f69SBart Van Assche else if (retries_tmp > 0) {
2577*44704f69SBart Van Assche pr2serr(">>> retrying a sgio %s, lba=0x%" PRIx64 "\n",
2578*44704f69SBart Van Assche (do_verify ? "verify" : "write"), (uint64_t)seek);
2579*44704f69SBart Van Assche --retries_tmp;
2580*44704f69SBart Van Assche ++num_retries;
2581*44704f69SBart Van Assche if (unrecovered_errs > 0)
2582*44704f69SBart Van Assche --unrecovered_errs;
2583*44704f69SBart Van Assche } else
2584*44704f69SBart Van Assche break;
2585*44704f69SBart Van Assche first = false;
2586*44704f69SBart Van Assche }
2587*44704f69SBart Van Assche if (SG_DD_BYPASS == ret)
2588*44704f69SBart Van Assche ret = 0; /* not bumping out_full */
2589*44704f69SBart Van Assche else if (0 != ret) {
2590*44704f69SBart Van Assche pr2serr("sg_write failed,%s seek=%" PRId64 "\n",
2591*44704f69SBart Van Assche ((-2 == ret) ? " try reducing bpt," : ""), seek);
2592*44704f69SBart Van Assche break;
2593*44704f69SBart Van Assche } else {
2594*44704f69SBart Van Assche out_full += blocks;
2595*44704f69SBart Van Assche if (oflag.dio && (! dio_tmp))
2596*44704f69SBart Van Assche dio_incomplete_count++;
2597*44704f69SBart Van Assche }
2598*44704f69SBart Van Assche } else if (FT_DEV_NULL & out_type)
2599*44704f69SBart Van Assche out_full += blocks; /* act as if written out without error */
2600*44704f69SBart Van Assche else {
2601*44704f69SBart Van Assche while (((res = write(outfd, wrkPos, blocks * blk_sz)) < 0) &&
2602*44704f69SBart Van Assche ((EINTR == errno) || (EAGAIN == errno) ||
2603*44704f69SBart Van Assche (EBUSY == errno)))
2604*44704f69SBart Van Assche ;
2605*44704f69SBart Van Assche if (verbose > 2)
2606*44704f69SBart Van Assche pr2serr("write(unix): count=%d, res=%d\n", blocks * blk_sz,
2607*44704f69SBart Van Assche res);
2608*44704f69SBart Van Assche if (res < 0) {
2609*44704f69SBart Van Assche snprintf(ebuff, EBUFF_SZ, ME "writing, seek=%" PRId64 " ",
2610*44704f69SBart Van Assche seek);
2611*44704f69SBart Van Assche perror(ebuff);
2612*44704f69SBart Van Assche ret = -1;
2613*44704f69SBart Van Assche break;
2614*44704f69SBart Van Assche } else if (res < blocks * blk_sz) {
2615*44704f69SBart Van Assche pr2serr("output file probably full, seek=%" PRId64 " ", seek);
2616*44704f69SBart Van Assche blocks = res / blk_sz;
2617*44704f69SBart Van Assche out_full += blocks;
2618*44704f69SBart Van Assche if ((res % blk_sz) > 0)
2619*44704f69SBart Van Assche out_partial++;
2620*44704f69SBart Van Assche ret = -1;
2621*44704f69SBart Van Assche break;
2622*44704f69SBart Van Assche } else {
2623*44704f69SBart Van Assche out_full += blocks;
2624*44704f69SBart Van Assche bytes_of = res;
2625*44704f69SBart Van Assche }
2626*44704f69SBart Van Assche }
2627*44704f69SBart Van Assche #ifdef HAVE_POSIX_FADVISE
2628*44704f69SBart Van Assche {
2629*44704f69SBart Van Assche int rt, in_valid, out2_valid, out_valid;
2630*44704f69SBart Van Assche
2631*44704f69SBart Van Assche in_valid = ((FT_OTHER == in_type) || (FT_BLOCK == in_type));
2632*44704f69SBart Van Assche out2_valid = ((FT_OTHER == out2_type) || (FT_BLOCK == out2_type));
2633*44704f69SBart Van Assche out_valid = ((FT_OTHER == out_type) || (FT_BLOCK == out_type));
2634*44704f69SBart Van Assche if (iflag.nocache && (bytes_read > 0) && in_valid) {
2635*44704f69SBart Van Assche rt = posix_fadvise(infd, 0, (skip * blk_sz) + bytes_read,
2636*44704f69SBart Van Assche POSIX_FADV_DONTNEED);
2637*44704f69SBart Van Assche // rt = posix_fadvise(infd, (skip * blk_sz), bytes_read,
2638*44704f69SBart Van Assche // POSIX_FADV_DONTNEED);
2639*44704f69SBart Van Assche // rt = posix_fadvise(infd, 0, 0, POSIX_FADV_DONTNEED);
2640*44704f69SBart Van Assche if (rt) /* returns error as result */
2641*44704f69SBart Van Assche pr2serr("posix_fadvise on read, skip=%" PRId64
2642*44704f69SBart Van Assche " ,err=%d\n", skip, rt);
2643*44704f69SBart Van Assche }
2644*44704f69SBart Van Assche if ((oflag.nocache & 2) && (bytes_of2 > 0) && out2_valid) {
2645*44704f69SBart Van Assche rt = posix_fadvise(out2fd, 0, 0, POSIX_FADV_DONTNEED);
2646*44704f69SBart Van Assche if (rt)
2647*44704f69SBart Van Assche pr2serr("posix_fadvise on of2, seek=%" PRId64
2648*44704f69SBart Van Assche " ,err=%d\n", seek, rt);
2649*44704f69SBart Van Assche }
2650*44704f69SBart Van Assche if ((oflag.nocache & 1) && (bytes_of > 0) && out_valid) {
2651*44704f69SBart Van Assche rt = posix_fadvise(outfd, 0, 0, POSIX_FADV_DONTNEED);
2652*44704f69SBart Van Assche if (rt)
2653*44704f69SBart Van Assche pr2serr("posix_fadvise on output, seek=%" PRId64
2654*44704f69SBart Van Assche " ,err=%d\n", seek, rt);
2655*44704f69SBart Van Assche }
2656*44704f69SBart Van Assche }
2657*44704f69SBart Van Assche #endif
2658*44704f69SBart Van Assche if (dd_count > 0)
2659*44704f69SBart Van Assche dd_count -= blocks;
2660*44704f69SBart Van Assche skip += blocks;
2661*44704f69SBart Van Assche seek += blocks;
2662*44704f69SBart Van Assche if (progress > 0) {
2663*44704f69SBart Van Assche if (check_progress()) {
2664*44704f69SBart Van Assche calc_duration_throughput(true);
2665*44704f69SBart Van Assche print_stats("");
2666*44704f69SBart Van Assche }
2667*44704f69SBart Van Assche }
2668*44704f69SBart Van Assche } /* end of main loop that does the copy ... */
2669*44704f69SBart Van Assche
2670*44704f69SBart Van Assche if (ret && penult_sparse_skip && (penult_blocks > 0)) {
2671*44704f69SBart Van Assche /* if error and skipped last output due to sparse ... */
2672*44704f69SBart Van Assche if ((FT_SG & out_type) || (FT_DEV_NULL & out_type))
2673*44704f69SBart Van Assche ;
2674*44704f69SBart Van Assche else {
2675*44704f69SBart Van Assche /* ... try writing to extend ofile to length prior to error */
2676*44704f69SBart Van Assche while (((res = write(outfd, zeros_buff, penult_blocks * blk_sz))
2677*44704f69SBart Van Assche < 0) && ((EINTR == errno) || (EAGAIN == errno) ||
2678*44704f69SBart Van Assche (EBUSY == errno)))
2679*44704f69SBart Van Assche ;
2680*44704f69SBart Van Assche if (verbose > 2)
2681*44704f69SBart Van Assche pr2serr("write(unix, sparse after error): count=%d, res=%d\n",
2682*44704f69SBart Van Assche penult_blocks * blk_sz, res);
2683*44704f69SBart Van Assche if (res < 0) {
2684*44704f69SBart Van Assche snprintf(ebuff, EBUFF_SZ, ME "writing(sparse after error), "
2685*44704f69SBart Van Assche "seek=%" PRId64 " ", seek);
2686*44704f69SBart Van Assche perror(ebuff);
2687*44704f69SBart Van Assche }
2688*44704f69SBart Van Assche }
2689*44704f69SBart Van Assche }
2690*44704f69SBart Van Assche
2691*44704f69SBart Van Assche if (do_sync) {
2692*44704f69SBart Van Assche if (FT_SG & out_type) {
2693*44704f69SBart Van Assche pr2serr(">> Synchronizing cache on %s\n", outf);
2694*44704f69SBart Van Assche res = sg_ll_sync_cache_10(outfd, false, false, 0, 0, 0, true, 0);
2695*44704f69SBart Van Assche if (SG_LIB_CAT_UNIT_ATTENTION == res) {
2696*44704f69SBart Van Assche pr2serr("Unit attention (out, sync cache), continuing\n");
2697*44704f69SBart Van Assche res = sg_ll_sync_cache_10(outfd, false, false, 0, 0, 0,
2698*44704f69SBart Van Assche false, 0);
2699*44704f69SBart Van Assche }
2700*44704f69SBart Van Assche if (0 != res)
2701*44704f69SBart Van Assche pr2serr("Unable to synchronize cache\n");
2702*44704f69SBart Van Assche }
2703*44704f69SBart Van Assche }
2704*44704f69SBart Van Assche
2705*44704f69SBart Van Assche bypass_copy:
2706*44704f69SBart Van Assche if (do_time)
2707*44704f69SBart Van Assche calc_duration_throughput(false);
2708*44704f69SBart Van Assche if (progress > 0)
2709*44704f69SBart Van Assche pr2serr("\nCompleted:\n");
2710*44704f69SBart Van Assche
2711*44704f69SBart Van Assche if (wrkBuff)
2712*44704f69SBart Van Assche free(wrkBuff);
2713*44704f69SBart Van Assche if (free_zeros_buff)
2714*44704f69SBart Van Assche free(free_zeros_buff);
2715*44704f69SBart Van Assche if ((STDIN_FILENO != infd) && (infd >= 0))
2716*44704f69SBart Van Assche close(infd);
2717*44704f69SBart Van Assche if (! ((STDOUT_FILENO == outfd) || (FT_DEV_NULL & out_type))) {
2718*44704f69SBart Van Assche if (outfd >= 0)
2719*44704f69SBart Van Assche close(outfd);
2720*44704f69SBart Van Assche }
2721*44704f69SBart Van Assche if (dry_run > 0)
2722*44704f69SBart Van Assche goto bypass2;
2723*44704f69SBart Van Assche
2724*44704f69SBart Van Assche if (0 != dd_count) {
2725*44704f69SBart Van Assche pr2serr("Some error occurred,");
2726*44704f69SBart Van Assche if (0 == ret)
2727*44704f69SBart Van Assche ret = SG_LIB_CAT_OTHER;
2728*44704f69SBart Van Assche }
2729*44704f69SBart Van Assche print_stats("");
2730*44704f69SBart Van Assche if (dio_incomplete_count) {
2731*44704f69SBart Van Assche int fd;
2732*44704f69SBart Van Assche char c;
2733*44704f69SBart Van Assche
2734*44704f69SBart Van Assche pr2serr(">> Direct IO requested but incomplete %d times\n",
2735*44704f69SBart Van Assche dio_incomplete_count);
2736*44704f69SBart Van Assche if ((fd = open(sg_allow_dio, O_RDONLY)) >= 0) {
2737*44704f69SBart Van Assche if (1 == read(fd, &c, 1)) {
2738*44704f69SBart Van Assche if ('0' == c)
2739*44704f69SBart Van Assche pr2serr(">>> %s set to '0' but should be set to '1' for "
2740*44704f69SBart Van Assche "direct IO\n", sg_allow_dio);
2741*44704f69SBart Van Assche }
2742*44704f69SBart Van Assche close(fd);
2743*44704f69SBart Van Assche }
2744*44704f69SBart Van Assche }
2745*44704f69SBart Van Assche if (sum_of_resids)
2746*44704f69SBart Van Assche pr2serr(">> Non-zero sum of residual counts=%d\n", sum_of_resids);
2747*44704f69SBart Van Assche
2748*44704f69SBart Van Assche bypass2:
2749*44704f69SBart Van Assche return (ret >= 0) ? ret : SG_LIB_CAT_OTHER;
2750*44704f69SBart Van Assche }
2751