Home | History | Annotate | Download | only in fastboot 
      1 #include "fs.h"
      2 
      3 #include "fastboot.h"
      4 #include "make_f2fs.h"
      5 
      6 #include <errno.h>
      7 #include <stdio.h>
      8 #include <stdlib.h>
      9 #include <string.h>
     10 #include <sys/stat.h>
     11 #include <sys/types.h>
     12 #include <unistd.h>
     13 
     14 #include <ext4_utils/make_ext4fs.h>
     15 #include <sparse/sparse.h>
     16 
     17 static int generate_ext4_image(int fd, long long partSize, const std::string& initial_dir,
     18                                        unsigned eraseBlkSize, unsigned logicalBlkSize)
     19 {
     20     if (initial_dir.empty()) {
     21         make_ext4fs_sparse_fd_align(fd, partSize, NULL, NULL, eraseBlkSize, logicalBlkSize);
     22     } else {
     23         make_ext4fs_sparse_fd_directory_align(fd, partSize, NULL, NULL, initial_dir.c_str(),
     24                                               eraseBlkSize, logicalBlkSize);
     25     }
     26     return 0;
     27 }
     28 
     29 #ifdef USE_F2FS
     30 static int generate_f2fs_image(int fd, long long partSize, const std::string& initial_dir,
     31                                unsigned /* unused */, unsigned /* unused */)
     32 {
     33     if (!initial_dir.empty()) {
     34         fprintf(stderr, "Unable to set initial directory on F2FS filesystem\n");
     35         return -1;
     36     }
     37     return make_f2fs_sparse_fd(fd, partSize, NULL, NULL);
     38 }
     39 #endif
     40 
     41 static const struct fs_generator {
     42     const char* fs_type;  //must match what fastboot reports for partition type
     43 
     44     //returns 0 or error value
     45     int (*generate)(int fd, long long partSize, const std::string& initial_dir,
     46                     unsigned eraseBlkSize, unsigned logicalBlkSize);
     47 
     48 } generators[] = {
     49     { "ext4", generate_ext4_image},
     50 #ifdef USE_F2FS
     51     { "f2fs", generate_f2fs_image},
     52 #endif
     53 };
     54 
     55 const struct fs_generator* fs_get_generator(const std::string& fs_type) {
     56     for (size_t i = 0; i < sizeof(generators) / sizeof(*generators); i++) {
     57         if (fs_type == generators[i].fs_type) {
     58             return generators + i;
     59         }
     60     }
     61     return nullptr;
     62 }
     63 
     64 int fs_generator_generate(const struct fs_generator* gen, int tmpFileNo, long long partSize,
     65     const std::string& initial_dir, unsigned eraseBlkSize, unsigned logicalBlkSize)
     66 {
     67     return gen->generate(tmpFileNo, partSize, initial_dir, eraseBlkSize, logicalBlkSize);
     68 }
     69