1
1
mirror of https://github.com/rui314/mold.git synced 2024-12-25 17:34:02 +03:00
mold/output_file.cc

114 lines
3.4 KiB
C++
Raw Normal View History

2021-01-09 11:07:19 +03:00
#include "mold.h"
#include <fcntl.h>
#include <sys/mman.h>
#include <sys/stat.h>
#include <sys/types.h>
static u32 get_umask() {
2021-01-09 17:13:38 +03:00
u32 orig_umask = umask(0);
umask(orig_umask);
return orig_umask;
2021-01-09 11:07:19 +03:00
}
2021-03-29 14:29:57 +03:00
template <typename E>
class MemoryMappedOutputFile : public OutputFile<E> {
2021-01-09 11:07:19 +03:00
public:
2021-03-29 14:29:57 +03:00
MemoryMappedOutputFile(Context<E> &ctx, std::string path, i64 filesize)
: OutputFile<E>(path, filesize) {
2021-04-09 12:59:54 +03:00
std::string dir = path_dirname(path);
2021-04-07 09:38:31 +03:00
this->tmpfile = (char *)save_string(ctx, dir + "/.mold-XXXXXX").data();
2021-03-29 14:29:57 +03:00
i64 fd = mkstemp(this->tmpfile);
2021-01-09 11:07:19 +03:00
if (fd == -1)
2021-03-29 14:29:57 +03:00
Fatal(ctx) << "cannot open " << this->tmpfile << ": " << strerror(errno);
2021-01-09 11:07:19 +03:00
2021-04-09 12:59:54 +03:00
if (rename(path.c_str(), this->tmpfile) == 0) {
::close(fd);
2021-03-29 14:29:57 +03:00
fd = ::open(this->tmpfile, O_RDWR | O_CREAT, 0777);
if (fd == -1) {
if (errno != ETXTBSY)
2021-04-09 12:59:54 +03:00
Fatal(ctx) << "cannot open " << path << ": " << strerror(errno);
2021-03-29 14:29:57 +03:00
unlink(this->tmpfile);
fd = ::open(this->tmpfile, O_RDWR | O_CREAT, 0777);
if (fd == -1)
2021-04-09 12:59:54 +03:00
Fatal(ctx) << "cannot open " << path << ": " << strerror(errno);
}
}
2021-01-09 11:07:19 +03:00
if (ftruncate(fd, filesize))
2021-03-29 10:48:23 +03:00
Fatal(ctx) << "ftruncate failed";
2021-01-09 11:07:19 +03:00
if (fchmod(fd, (0777 & ~get_umask())) == -1)
2021-03-29 10:48:23 +03:00
Fatal(ctx) << "fchmod failed";
2021-03-29 14:29:57 +03:00
this->buf = (u8 *)mmap(nullptr, filesize, PROT_READ | PROT_WRITE,
MAP_SHARED, fd, 0);
if (this->buf == MAP_FAILED)
2021-04-09 12:59:54 +03:00
Fatal(ctx) << path << ": mmap failed: " << strerror(errno);
2021-01-09 11:07:19 +03:00
::close(fd);
}
2021-03-29 14:29:57 +03:00
void close(Context<E> &ctx) override {
2021-04-07 09:38:31 +03:00
Timer t(ctx, "close_file");
2021-03-29 14:29:57 +03:00
munmap(this->buf, this->filesize);
2021-04-09 12:59:54 +03:00
if (rename(this->tmpfile, this->path.c_str()) == -1)
2021-04-09 19:31:45 +03:00
Fatal(ctx) << this->path << ": rename failed: " << strerror(errno);
2021-03-29 14:29:57 +03:00
this->tmpfile = nullptr;
2021-01-09 11:07:19 +03:00
}
};
2021-03-29 14:29:57 +03:00
template <typename E>
class MallocOutputFile : public OutputFile<E> {
2021-01-09 11:07:19 +03:00
public:
2021-03-29 14:29:57 +03:00
MallocOutputFile(Context<E> &ctx, std::string path, u64 filesize)
: OutputFile<E>(path, filesize) {
this->buf = (u8 *)mmap(NULL, filesize, PROT_READ | PROT_WRITE,
MAP_SHARED | MAP_ANONYMOUS, -1, 0);
if (this->buf == MAP_FAILED)
2021-03-29 10:48:23 +03:00
Fatal(ctx) << "mmap failed: " << strerror(errno);
2021-01-09 11:07:19 +03:00
}
2021-03-29 14:29:57 +03:00
void close(Context<E> &ctx) override {
2021-04-07 09:38:31 +03:00
Timer t(ctx, "close_file");
2021-03-29 14:29:57 +03:00
i64 fd = ::open(this->path.c_str(), O_RDWR | O_CREAT, 0777);
2021-01-09 11:07:19 +03:00
if (fd == -1)
2021-04-09 12:59:54 +03:00
Fatal(ctx) << "cannot open " << this->path << ": " << strerror(errno);
2021-01-09 11:07:19 +03:00
FILE *fp = fdopen(fd, "w");
2021-03-29 14:29:57 +03:00
fwrite(this->buf, this->filesize, 1, fp);
2021-01-09 11:07:19 +03:00
fclose(fp);
}
};
2021-03-29 14:29:57 +03:00
template <typename E>
2021-04-06 07:37:01 +03:00
std::unique_ptr<OutputFile<E>>
2021-03-29 14:29:57 +03:00
OutputFile<E>::open(Context<E> &ctx, std::string path, u64 filesize) {
2021-04-07 09:38:31 +03:00
Timer t(ctx, "open_file");
2021-01-09 11:07:19 +03:00
2021-04-09 10:05:52 +03:00
if (path.starts_with('/') && !ctx.arg.chroot.empty())
path = ctx.arg.chroot + "/" + path_clean(path);
bool is_special = false;
struct stat st;
if (stat(path.c_str(), &st) == 0 && (st.st_mode & S_IFMT) != S_IFREG)
is_special = true;
2021-04-06 07:37:01 +03:00
std::unique_ptr<OutputFile<E>> file;
if (is_special)
2021-04-06 07:37:01 +03:00
file = std::make_unique<MallocOutputFile<E>>(ctx, path, filesize);
else
2021-04-06 07:37:01 +03:00
file = std::make_unique<MemoryMappedOutputFile<E>>(ctx, path, filesize);
2021-01-09 11:07:19 +03:00
2021-03-29 07:20:51 +03:00
if (ctx.arg.filler != -1)
memset(file->buf, ctx.arg.filler, filesize);
2021-01-09 11:07:19 +03:00
return file;
}
2021-03-29 14:29:57 +03:00
template
2021-04-06 07:37:01 +03:00
std::unique_ptr<OutputFile<X86_64>>
2021-03-30 08:57:38 +03:00
OutputFile<X86_64>::open(Context<X86_64> &ctx, std::string path, u64 filesize);
2021-03-30 13:27:00 +03:00
template
2021-04-06 07:37:01 +03:00
std::unique_ptr<OutputFile<I386>>
2021-03-30 13:27:00 +03:00
OutputFile<I386>::open(Context<I386> &ctx, std::string path, u64 filesize);