rui314/mold
1
1
Fork 0
mirror of https://github.com/rui314/mold.git synced 2024-09-22 10:27:48 +03:00
Code Issues Projects Releases Wiki Activity

temporary

Browse Source
This commit is contained in:
Rui Ueyama 2020-11-09 12:30:13 +09:00
parent 3c6f0ca9af
commit c505cfebff
2 changed files with 21 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
main.cc
View File

@ -6,6 +6,7 @@
#include <fcntl.h>
#include <iostream>
#include <libgen.h>
#include <sys/mman.h>
#include <sys/stat.h>
#include <sys/types.h>
@ -649,15 +650,20 @@ static void unlink_async(tbb::task_group &tg, StringRef path) {
tg.run([=]() { close(fd); });
}
static FileOutputBuffer *open_output_file(u64 filesize) {
Expected<std::unique_ptr<FileOutputBuffer>> buf_or_err =
FileOutputBuffer::create(config.output, filesize, FileOutputBuffer::F_executable);
static u8 *open_output_file(u64 filesize) {
int fd = open(config.output.str().c_str(), O_RDWR | O_CREAT | O_TRUNC, 0777);
if (fd == -1)
error("cannot open " + config.output + ": " + strerror(errno));
if (!buf_or_err)
error("failed to open " + config.output + ": " +
llvm::toString(buf_or_err.takeError()));
fallocate(fd, 0, 0, filesize);
return std::move(*buf_or_err).release();
void *addr = mmap(nullptr, filesize, PROT_READ | PROT_WRITE,
MAP_SHARED, fd, 0);
if (addr == MAP_FAILED)
error(config.output + ": mmap failed: " + strerror(errno));
close(fd);
return (u8 *)addr;
}
static void write_symtab(u8 *buf, std::vector<ObjectFile *> files) {
@ -931,15 +937,13 @@ int main(int argc, char **argv) {
}
// Create an output file
FileOutputBuffer *output_buffer;
u8 *buf = nullptr;
{
MyTimer t("open_output_file");
output_buffer = open_output_file(filesize);
buf = open_output_file(filesize);
}
u8 *buf = output_buffer->getBufferStart();
// Fill .symtab and .strtab
{
MyTimer t("write_symtab", copy);
@ -968,8 +972,8 @@ int main(int argc, char **argv) {
{
MyTimer t("commit", copy);
if (auto e = output_buffer->commit())
error("failed to write to the output file: " + toString(std::move(e)));
if (munmap(buf, filesize) == -1)
error("munmap failed");
}
total_timer.stopTimer();

4
mold.h
View File

@ -1,5 +1,9 @@
#pragma once
#ifndef _GNU_SOURCE
#define _GNU_SOURCE
#endif
#include "llvm/ADT/ArrayRef.h"
#include "llvm/ADT/SmallVector.h"
#include "llvm/ADT/Twine.h"