1 #include <stdio.h>
2 #include <stdlib.h>
3 #include <string.h>
4 #include <sys/reboot.h>
5 #include <sys/wait.h>
6 #include <unistd.h>
7
8 #include <cutils/android_reboot.h>
9 #include <cutils/partition_utils.h>
10
11 const char *mkfs = "/system/bin/make_ext4fs";
12
setup_fs(const char * blockdev)13 int setup_fs(const char *blockdev)
14 {
15 char buf[256], path[128];
16 pid_t child;
17 int status, n;
18 pid_t pid;
19
20 /* we might be looking at an indirect reference */
21 n = readlink(blockdev, path, sizeof(path) - 1);
22 if (n > 0) {
23 path[n] = 0;
24 if (!memcmp(path, "/dev/block/", 11))
25 blockdev = path + 11;
26 }
27
28 if (strchr(blockdev,'/')) {
29 fprintf(stderr,"not a block device name: %s\n", blockdev);
30 return 0;
31 }
32
33 snprintf(buf, sizeof(buf), "/sys/fs/ext4/%s", blockdev);
34 if (access(buf, F_OK) == 0) {
35 fprintf(stderr,"device %s already has a filesystem\n", blockdev);
36 return 0;
37 }
38 snprintf(buf, sizeof(buf), "/dev/block/%s", blockdev);
39
40 if (!partition_wiped(buf)) {
41 fprintf(stderr,"device %s not wiped, probably encrypted, not wiping\n", blockdev);
42 return 0;
43 }
44
45 fprintf(stderr,"+++\n");
46
47 child = fork();
48 if (child < 0) {
49 fprintf(stderr,"error: setup_fs: fork failed\n");
50 return 0;
51 }
52 if (child == 0) {
53 execl(mkfs, mkfs, buf, NULL);
54 exit(-1);
55 }
56
57 while ((pid=waitpid(-1, &status, 0)) != child) {
58 if (pid == -1) {
59 fprintf(stderr, "error: setup_fs: waitpid failed!\n");
60 return 1;
61 }
62 }
63
64 fprintf(stderr,"---\n");
65 return 1;
66 }
67
68
main(int argc,char ** argv)69 int main(int argc, char **argv)
70 {
71 int need_reboot = 0;
72
73 while (argc > 1) {
74 if (strlen(argv[1]) < 128)
75 need_reboot |= setup_fs(argv[1]);
76 argv++;
77 argc--;
78 }
79
80 if (need_reboot) {
81 fprintf(stderr,"REBOOT!\n");
82 android_reboot(ANDROID_RB_RESTART, 0, 0);
83 exit(-1);
84 }
85 return 0;
86 }
87