Ensure parent folders exist before extract

This commit is contained in:
topjohnwu
2022-08-07 04:06:18 -07:00
parent 07b45f39df
commit 1b4a6850b8
3 changed files with 17 additions and 10 deletions
+10 -5
View File
@@ -1,5 +1,6 @@
#include <unistd.h>
#include <fcntl.h>
#include <libgen.h>
#include <algorithm>
#include <base.hpp>
@@ -79,17 +80,21 @@ void cpio::extract_entry(const entry_map::value_type &e, const char *file) {
fprintf(stderr, "Extract [%s] to [%s]\n", e.first.data(), file);
unlink(file);
rmdir(file);
// Make sure parent folders exist
char *parent = dirname(file);
xmkdirs(parent, 0755);
if (S_ISDIR(e.second->mode)) {
::mkdir(file, e.second->mode & 0777);
xmkdir(file, e.second->mode & 0777);
} else if (S_ISREG(e.second->mode)) {
int fd = creat(file, e.second->mode & 0777);
int fd = xopen(file, O_CREAT | O_WRONLY | O_TRUNC, e.second->mode & 0777);
xwrite(fd, e.second->data, e.second->filesize);
fchown(fd, e.second->uid, e.second->gid);
close(fd);
} else if (S_ISLNK(e.second->mode)) {
auto target = strndup((char *) e.second->data, e.second->filesize);
} else if (S_ISLNK(e.second->mode) && e.second->filesize < 4096) {
char target[4096];
memcpy(target, e.second->data, e.second->filesize);
target[e.second->filesize] = '\0';
symlink(target, file);
free(target);
}
}