]> CyberLeo.Net >> Repos - FreeBSD/FreeBSD.git/blob - usr.sbin/nandtool/nand_erase.c
Followup to r347996
[FreeBSD/FreeBSD.git] / usr.sbin / nandtool / nand_erase.c
1 /*-
2  * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
3  *
4  * Copyright (c) 2010-2012 Semihalf.
5  * All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  * 2. 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  *
16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26  * SUCH DAMAGE.
27  */
28
29 #include <sys/cdefs.h>
30 __FBSDID("$FreeBSD$");
31
32 #include <stdio.h>
33 #include <stdlib.h>
34 #include <unistd.h>
35 #include <sys/types.h>
36 #include <sys/disk.h>
37 #include <libgeom.h>
38 #include <dev/nand/nand_dev.h>
39 #include "nandtool.h"
40
41 int nand_erase(struct cmd_param *params)
42 {
43         struct chip_param_io chip_params;
44         char *dev;
45         int fd = -1, ret = 0;
46         off_t pos, count;
47         off_t start, nblocks, i;
48         int block_size, mult;
49
50         if (!(dev = param_get_string(params, "dev"))) {
51                 fprintf(stderr, "Please supply valid 'dev' parameter.\n");
52                 return (1);
53         }
54
55         if (param_has_value(params, "count"))
56                 count = param_get_intx(params, "count");
57         else
58                 count = 1;
59
60         if ((fd = g_open(dev, 1)) < 0) {
61                 perrorf("Cannot open %s", dev);
62                 return (1);
63         }
64
65         if (ioctl(fd, NAND_IO_GET_CHIP_PARAM, &chip_params) == -1) {
66                 perrorf("Cannot ioctl(NAND_IO_GET_CHIP_PARAM)");
67                 ret = 1;
68                 goto out;
69         }
70
71         block_size = chip_params.page_size * chip_params.pages_per_block;
72
73         if (param_has_value(params, "page")) {
74                 pos = chip_params.page_size * param_get_intx(params, "page");
75                 mult = chip_params.page_size;
76         } else if (param_has_value(params, "block")) {
77                 pos = block_size * param_get_intx(params, "block");
78                 mult = block_size;
79         } else if (param_has_value(params, "pos")) {
80                 pos = param_get_intx(params, "pos");
81                 mult = 1;
82         } else {
83                 /* Erase whole chip */
84                 if (ioctl(fd, DIOCGMEDIASIZE, &count) == -1) {
85                         ret = 1;
86                         goto out;
87                 }
88
89                 pos = 0;
90                 mult = 1;
91         }
92
93         if (pos % block_size) {
94                 fprintf(stderr, "Position must be block-size aligned!\n");
95                 ret = 1;
96                 goto out;
97         }
98
99         count *= mult;
100         start = pos / block_size;
101         nblocks = count / block_size;
102
103         for (i = 0; i < nblocks; i++) {
104                 if (g_delete(fd, (start + i) * block_size, block_size) == -1) {
105                         perrorf("Cannot erase block %d - probably a bad block",
106                             start + i);
107                         ret = 1;
108                 }
109         }
110
111 out:
112         g_close(fd);
113
114         return (ret);
115 }
116