65 lines
1.7 KiB
C++
65 lines
1.7 KiB
C++
#include "fs.h"
|
|
|
|
#include "fastboot.h"
|
|
#include "make_f2fs.h"
|
|
|
|
#include <errno.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/types.h>
|
|
#include <unistd.h>
|
|
|
|
#include <ext4_utils/make_ext4fs.h>
|
|
#include <sparse/sparse.h>
|
|
|
|
static int generate_ext4_image(int fd, long long partSize, const std::string& initial_dir)
|
|
{
|
|
if (initial_dir.empty()) {
|
|
make_ext4fs_sparse_fd(fd, partSize, NULL, NULL);
|
|
} else {
|
|
make_ext4fs_sparse_fd_directory(fd, partSize, NULL, NULL, initial_dir.c_str());
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
#ifdef USE_F2FS
|
|
static int generate_f2fs_image(int fd, long long partSize, const std::string& initial_dir)
|
|
{
|
|
if (!initial_dir.empty()) {
|
|
fprintf(stderr, "Unable to set initial directory on F2FS filesystem\n");
|
|
return -1;
|
|
}
|
|
return make_f2fs_sparse_fd(fd, partSize, NULL, NULL);
|
|
}
|
|
#endif
|
|
|
|
static const struct fs_generator {
|
|
const char* fs_type; //must match what fastboot reports for partition type
|
|
|
|
//returns 0 or error value
|
|
int (*generate)(int fd, long long partSize, const std::string& initial_dir);
|
|
|
|
} generators[] = {
|
|
{ "ext4", generate_ext4_image},
|
|
#ifdef USE_F2FS
|
|
{ "f2fs", generate_f2fs_image},
|
|
#endif
|
|
};
|
|
|
|
const struct fs_generator* fs_get_generator(const std::string& fs_type) {
|
|
for (size_t i = 0; i < sizeof(generators) / sizeof(*generators); i++) {
|
|
if (fs_type == generators[i].fs_type) {
|
|
return generators + i;
|
|
}
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
int fs_generator_generate(const struct fs_generator* gen, int tmpFileNo, long long partSize,
|
|
const std::string& initial_dir)
|
|
{
|
|
return gen->generate(tmpFileNo, partSize, initial_dir);
|
|
}
|