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: stable/12/usr.sbin/nandtool/nand_erase.c 326276 2017-11-27 15:37:16Z pfg $");
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 
nand_erase(struct cmd_param * params)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 
117