source: mainline/uspace/app/blkdump/blkdump.c

Last change on this file was 0460377, checked in by GitHub <noreply@…>, 3 months ago

uspace/app/blkdump: fix argv null deref (#251)

  • Property mode set to 100644
File size: 7.0 KB
RevLine 
[32d3ebf]1/*
[7ae01d5]2 * Copyright (c) 2024 Jiri Svoboda
[32d3ebf]3 * Copyright (c) 2011 Martin Sucha
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 *
10 * - Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * - Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
15 * - The name of the author may not be used to endorse or promote products
16 * derived from this software without specific prior written permission.
17 *
18 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
19 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
20 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
21 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
22 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
23 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
24 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
25 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
27 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28 */
29
[b1834a01]30/** @addtogroup blkdump
[32d3ebf]31 * @{
32 */
33
34/**
35 * @file blockdump.c
36 * @brief Tool for dumping content of block devices
37 *
38 */
39
40#include <stdio.h>
41#include <stdlib.h>
[8d2dd7f2]42#include <stdint.h>
[f73b291]43#include <block.h>
[32d3ebf]44#include <mem.h>
[15f3c3f]45#include <loc.h>
[32d3ebf]46#include <byteorder.h>
[3abf70c7]47#include <scsi/mmc.h>
[7354b5e]48#include <offset.h>
[32d3ebf]49#include <inttypes.h>
50#include <errno.h>
[1d6dd2a]51#include <str.h>
[32d3ebf]52
53#define NAME "blkdump"
54
55static void syntax_print(void);
[7901ac8]56static int print_blocks(aoff64_t block_offset, aoff64_t block_count, size_t block_size);
57static int print_toc(void);
[32d3ebf]58static void print_hex_row(uint8_t *data, size_t length, size_t bytes_per_row);
59
[7901ac8]60static bool relative = false;
61static service_id_t service_id;
62
[32d3ebf]63int main(int argc, char **argv)
64{
65
[b7fd2a0]66 errno_t rc;
[32d3ebf]67 char *dev_path;
68 size_t block_size;
69 char *endptr;
70 aoff64_t block_offset = 0;
71 aoff64_t block_count = 1;
72 aoff64_t dev_nblocks;
[7901ac8]73 bool toc = false;
[a35b458]74
[32d3ebf]75 if (argc < 2) {
76 printf(NAME ": Error, argument missing.\n");
77 syntax_print();
78 return 1;
79 }
80
[1433ecda]81 --argc;
82 ++argv;
[32d3ebf]83
[7901ac8]84 if (str_cmp(*argv, "--toc") == 0) {
[1433ecda]85 --argc;
86 ++argv;
[7901ac8]87 toc = true;
88 goto devname;
89 }
90
[d3842e0]91 if (str_cmp(*argv, "--relative") == 0) {
[1433ecda]92 --argc;
93 ++argv;
[d3842e0]94 relative = true;
95 }
[a35b458]96
[0460377]97 if (*argv && str_cmp(*argv, "--offset") == 0) {
[1433ecda]98 --argc;
99 ++argv;
[32d3ebf]100 if (*argv == NULL) {
101 printf(NAME ": Error, argument missing (offset).\n");
102 syntax_print();
103 return 1;
104 }
105
106 block_offset = strtol(*argv, &endptr, 10);
107 if (*endptr != '\0') {
108 printf(NAME ": Error, invalid argument (offset).\n");
109 syntax_print();
110 return 1;
111 }
112
[1433ecda]113 --argc;
114 ++argv;
[32d3ebf]115 }
[a35b458]116
[0460377]117 if (*argv && str_cmp(*argv, "--count") == 0) {
[1433ecda]118 --argc;
119 ++argv;
[32d3ebf]120 if (*argv == NULL) {
121 printf(NAME ": Error, argument missing (count).\n");
122 syntax_print();
123 return 1;
124 }
125
126 block_count = strtol(*argv, &endptr, 10);
127 if (*endptr != '\0') {
128 printf(NAME ": Error, invalid argument (count).\n");
129 syntax_print();
130 return 1;
131 }
132
[1433ecda]133 --argc;
134 ++argv;
[32d3ebf]135 }
136
[7901ac8]137devname:
[0460377]138 if (*argv == NULL) {
139 printf(NAME ": Error, argument missing (device_name).\n");
140 syntax_print();
141 return 1;
142 }
143
[32d3ebf]144 if (argc != 1) {
145 printf(NAME ": Error, unexpected argument.\n");
146 syntax_print();
147 return 1;
148 }
149
150 dev_path = *argv;
151
[15f3c3f]152 rc = loc_service_get_id(dev_path, &service_id, 0);
[32d3ebf]153 if (rc != EOK) {
154 printf(NAME ": Error resolving device `%s'.\n", dev_path);
155 return 2;
156 }
157
[7ae01d5]158 rc = block_init(service_id);
[32d3ebf]159 if (rc != EOK) {
160 printf(NAME ": Error initializing libblock.\n");
161 return 2;
162 }
163
[15f3c3f]164 rc = block_get_bsize(service_id, &block_size);
[32d3ebf]165 if (rc != EOK) {
166 printf(NAME ": Error determining device block size.\n");
167 return 2;
168 }
169
[15f3c3f]170 rc = block_get_nblocks(service_id, &dev_nblocks);
[32d3ebf]171 if (rc != EOK) {
172 printf(NAME ": Warning, failed to obtain block device size.\n");
173 }
174
[ec534b4]175 printf("Device %s has %" PRIuOFF64 " blocks, %" PRIuOFF64 " bytes each\n", dev_path, dev_nblocks, (aoff64_t) block_size);
[32d3ebf]176
[d5c1051]177 int ret;
[7901ac8]178 if (toc)
[d5c1051]179 ret = print_toc();
[7901ac8]180 else
[d5c1051]181 ret = print_blocks(block_offset, block_count, block_size);
[7901ac8]182
183 block_fini(service_id);
184
[d5c1051]185 return ret;
[7901ac8]186}
187
188static int print_blocks(aoff64_t block_offset, aoff64_t block_count, size_t block_size)
189{
190 uint8_t *data;
191 aoff64_t current;
192 aoff64_t limit;
193 size_t data_offset;
[b7fd2a0]194 errno_t rc;
[7901ac8]195
[32d3ebf]196 data = malloc(block_size);
197 if (data == NULL) {
[ec534b4]198 printf(NAME ": Error allocating data buffer of %" PRIuOFF64 " bytes", (aoff64_t) block_size);
[32d3ebf]199 return 3;
200 }
[7901ac8]201
[32d3ebf]202 limit = block_offset + block_count;
203 for (current = block_offset; current < limit; current++) {
[15f3c3f]204 rc = block_read_direct(service_id, current, 1, data);
[32d3ebf]205 if (rc != EOK) {
206 printf(NAME ": Error reading block at %" PRIuOFF64 " \n", current);
207 free(data);
208 return 3;
209 }
[7901ac8]210
[1433ecda]211 printf("---- Block %" PRIuOFF64 " (at %" PRIuOFF64 ") ----\n", current, current * block_size);
[7901ac8]212
[32d3ebf]213 for (data_offset = 0; data_offset < block_size; data_offset += 16) {
[d3842e0]214 if (relative) {
215 printf("%8" PRIxOFF64 ": ", (aoff64_t) data_offset);
[1433ecda]216 } else {
217 printf("%8" PRIxOFF64 ": ", current * block_size + data_offset);
[d3842e0]218 }
[1433ecda]219 print_hex_row(data + data_offset, block_size - data_offset, 16);
[32d3ebf]220 printf("\n");
221 }
222 printf("\n");
223 }
[7901ac8]224
[32d3ebf]225 free(data);
[7901ac8]226 return 0;
227}
[32d3ebf]228
[7901ac8]229static int print_toc(void)
230{
[3abf70c7]231 scsi_toc_multisess_data_t toc;
[b7fd2a0]232 errno_t rc;
[7901ac8]233
[3abf70c7]234 rc = block_read_toc(service_id, 0, &toc, sizeof(toc));
235 if (rc != EOK)
[7901ac8]236 return 1;
237
[da79df42]238 printf("Multisession Information:\n");
239 printf("\tFirst complete session: %" PRIu8 "\n", toc.first_sess);
240 printf("\tLast complete session: %" PRIu8 "\n", toc.last_sess);
241 printf("\tFirst track of last complete session:\n");
242 printf("\t\tADR / Control: 0x%" PRIx8 "\n", toc.ftrack_lsess.adr_control);
243 printf("\t\tTrack number: %" PRIu8 "\n", toc.ftrack_lsess.track_no);
244 printf("\t\tStart block address: %" PRIu32 "\n", toc.ftrack_lsess.start_addr);
[32d3ebf]245
246 return 0;
247}
248
249/**
250 * Print a row of 16 bytes as commonly seen in hexadecimal dumps
251 */
[1433ecda]252static void print_hex_row(uint8_t *data, size_t length, size_t bytes_per_row)
253{
[32d3ebf]254 size_t pos;
255 uint8_t b;
[a35b458]256
[32d3ebf]257 if (length > bytes_per_row) {
258 length = bytes_per_row;
259 }
[a35b458]260
[1a86c50]261 /* Print hexadecimal values */
[32d3ebf]262 for (pos = 0; pos < length; pos++) {
[1433ecda]263 if (pos == length / 2) {
[32d3ebf]264 printf(" ");
265 }
[cd5816d6]266 printf("%02hhX ", data[pos]);
[32d3ebf]267 }
[a35b458]268
[1a86c50]269 /* Pad with spaces if we have less than 16 bytes */
[32d3ebf]270 for (pos = length; pos < bytes_per_row; pos++) {
[1433ecda]271 if (pos == length / 2) {
[32d3ebf]272 printf(" ");
273 }
[cd5816d6]274 printf(" ");
[32d3ebf]275 }
[a35b458]276
[1a86c50]277 /* Print printable characters */
[32d3ebf]278 for (pos = 0; pos < length; pos++) {
[1433ecda]279 if (pos == length / 2) {
[cd5816d6]280 printf(" ");
281 }
[32d3ebf]282 b = data[pos];
283 if (b >= 32 && b < 128) {
284 putchar(b);
[1433ecda]285 } else {
[32d3ebf]286 putchar('.');
287 }
288 }
289}
290
291static void syntax_print(void)
292{
[da79df42]293 printf("syntax: blkdump [--toc] [--relative] [--offset <num_blocks>] "
294 "[--count <num_blocks>] <device_name>\n");
[32d3ebf]295}
296
297/**
298 * @}
299 */
Note: See TracBrowser for help on using the repository browser.