1 /*
2 * Copyright (C) 2011, 2012 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 #include <stdlib.h>
18 #include <stdio.h>
19 #include <sys/types.h>
20 #include <sys/stat.h>
21 #include <fcntl.h>
22 #include <unistd.h>
23 #include <sys/ioctl.h>
24 #include <linux/fs.h>
25 #include <errno.h>
26
27 typedef unsigned long long u64;
28
29 #ifndef BLKDISCARD
30 #define BLKDISCARD _IO(0x12,119)
31 #endif
32
33 #ifndef BLKSECDISCARD
34 #define BLKSECDISCARD _IO(0x12,125)
35 #endif
36
get_block_device_size(int fd)37 static u64 get_block_device_size(int fd)
38 {
39 u64 size = 0;
40 int ret;
41
42 ret = ioctl(fd, BLKGETSIZE64, &size);
43
44 if (ret)
45 return 0;
46
47 return size;
48 }
49
wipe_block_device(int fd,u64 len,int secure)50 static int wipe_block_device(int fd, u64 len, int secure)
51 {
52 u64 range[2];
53 int ret;
54 int req;
55
56 range[0] = 0;
57 range[1] = len;
58 if (secure) {
59 req = BLKSECDISCARD;
60 } else {
61 req = BLKDISCARD;
62 }
63
64 ret = ioctl(fd, req, &range);
65 if (ret < 0) {
66 fprintf(stderr, "%s discard failed, errno = %d\n",
67 secure ? "Secure" : "Nonsecure", errno);
68 }
69
70 return ret;
71 }
72
usage(void)73 static void usage(void)
74 {
75 fprintf(stderr, "Usage: wipe_blkdev [-s] <partition>\n");
76 exit(1);
77 }
78
main(int argc,char * argv[])79 int main(int argc, char *argv[])
80 {
81 int secure = 0;
82 char *devname;
83 int fd;
84 u64 len;
85 struct stat statbuf;
86 int ret;
87
88 if ((argc != 2) && (argc != 3)) {
89 usage();
90 }
91
92 if (argc == 3) {
93 if (!strcmp(argv[1], "-s")) {
94 secure = 1;
95 devname = argv[2];
96 } else {
97 usage();
98 }
99 } else {
100 devname = argv[1];
101 }
102
103 fd = open(devname, O_RDWR);
104 if (fd < 0) {
105 fprintf(stderr, "Cannot open device %s\n", devname);
106 exit(1);
107 }
108
109 if (fstat(fd, &statbuf) < 0) {
110 fprintf(stderr, "Cannot stat %s\n", devname);
111 exit(1);
112 }
113
114 if (!S_ISBLK(statbuf.st_mode)) {
115 fprintf(stderr, "%s is not a block device\n", devname);
116 exit(1);
117 }
118
119 len = get_block_device_size(fd);
120
121 if (! len) {
122 fprintf(stderr, "Cannot get size of block device %s\n", devname);
123 exit(1);
124 }
125
126 ret = wipe_block_device(fd, len, secure);
127
128 close(fd);
129
130 return ret;
131 }
132