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