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