setup_fs.c

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