mirror of
https://github.com/rui314/mold.git
synced 2024-11-11 16:58:12 +03:00
145 lines
3.9 KiB
C++
145 lines
3.9 KiB
C++
#include "chibild.h"
|
|
#include "llvm/Option/ArgList.h"
|
|
|
|
#include <iostream>
|
|
|
|
using llvm::file_magic;
|
|
using llvm::object::Archive;
|
|
using llvm::opt::InputArgList;
|
|
|
|
Config config;
|
|
SymbolTable symbol_table;
|
|
|
|
//
|
|
// Command-line option processing
|
|
//
|
|
|
|
// Create enum with OPT_xxx values for each option in Options.td
|
|
enum {
|
|
OPT_INVALID = 0,
|
|
#define OPTION(_1, _2, ID, _4, _5, _6, _7, _8, _9, _10, _11, _12) OPT_##ID,
|
|
#include "options.inc"
|
|
#undef OPTION
|
|
};
|
|
|
|
// Create prefix string literals used in Options.td
|
|
#define PREFIX(NAME, VALUE) const char *const NAME[] = VALUE;
|
|
#include "options.inc"
|
|
#undef PREFIX
|
|
|
|
// Create table mapping all options defined in Options.td
|
|
static const llvm::opt::OptTable::Info opt_info[] = {
|
|
#define OPTION(X1, X2, ID, KIND, GROUP, ALIAS, X7, X8, X9, X10, X11, X12) \
|
|
{X1, X2, X10, X11, OPT_##ID, llvm::opt::Option::KIND##Class, \
|
|
X9, X8, OPT_##GROUP, OPT_##ALIAS, X7, X12},
|
|
#include "options.inc"
|
|
#undef OPTION
|
|
};
|
|
|
|
class MyOptTable : llvm::opt::OptTable {
|
|
public:
|
|
MyOptTable() : OptTable(opt_info) {}
|
|
InputArgList parse(int argc, char **argv);
|
|
};
|
|
|
|
InputArgList MyOptTable::parse(int argc, char **argv) {
|
|
unsigned missingIndex;
|
|
unsigned missingCount;
|
|
SmallVector<const char *, 256> vec(argv, argv + argc);
|
|
|
|
InputArgList args = this->ParseArgs(vec, missingIndex, missingCount);
|
|
if (missingCount)
|
|
error(Twine(args.getArgString(missingIndex)) + ": missing argument");
|
|
|
|
for (auto *arg : args.filtered(OPT_UNKNOWN))
|
|
error("unknown argument '" + arg->getAsString(args) + "'");
|
|
return args;
|
|
}
|
|
|
|
//
|
|
// Main
|
|
//
|
|
|
|
static std::vector<MemoryBufferRef> getArchiveMembers(MemoryBufferRef mb) {
|
|
std::unique_ptr<Archive> file =
|
|
CHECK(Archive::create(mb), mb.getBufferIdentifier() + ": failed to parse archive");
|
|
|
|
std::vector<MemoryBufferRef> vec;
|
|
|
|
Error err = Error::success();
|
|
|
|
for (const Archive::Child &c : file->children(err)) {
|
|
MemoryBufferRef mbref =
|
|
CHECK(c.getMemoryBufferRef(),
|
|
mb.getBufferIdentifier() +
|
|
": could not get the buffer for a child of the archive");
|
|
vec.push_back(mbref);
|
|
}
|
|
|
|
if (err)
|
|
error(mb.getBufferIdentifier() + ": Archive::children failed: " +
|
|
toString(std::move(err)));
|
|
|
|
file.release(); // leak
|
|
return vec;
|
|
}
|
|
|
|
static void add_file(std::vector<ObjectFile *> &files, StringRef path) {
|
|
MemoryBufferRef mb = readFile(path);
|
|
|
|
switch (identify_magic(mb.getBuffer())) {
|
|
case file_magic::archive:
|
|
for (MemoryBufferRef member : getArchiveMembers(mb))
|
|
files.push_back(new ObjectFile(member));
|
|
break;
|
|
case file_magic::elf_relocatable:
|
|
files.push_back(new ObjectFile(mb));
|
|
break;
|
|
default:
|
|
error(path + ": unknown file type");
|
|
}
|
|
}
|
|
|
|
llvm::TimerGroup timers("all", "all");
|
|
|
|
int main(int argc, char **argv) {
|
|
MyOptTable opt_table;
|
|
InputArgList args = opt_table.parse(argc - 1, argv + 1);
|
|
|
|
llvm::Timer add_files_timer("add_files", "add_files", timers);
|
|
llvm::Timer parse_timer("parse", "parse", timers);
|
|
llvm::Timer register_timer("register_defined_symbols", "register_defined_symbols", timers);
|
|
|
|
if (auto *arg = args.getLastArg(OPT_o))
|
|
config.output = arg->getValue();
|
|
else
|
|
error("-o option is missing");
|
|
|
|
std::vector<ObjectFile *> files;
|
|
|
|
add_files_timer.startTimer();
|
|
for (auto *arg : args)
|
|
if (arg->getOption().getID() == OPT_INPUT)
|
|
add_file(files, arg->getValue());
|
|
add_files_timer.stopTimer();
|
|
|
|
parse_timer.startTimer();
|
|
// for (ObjectFile *file : files)
|
|
// file->parse();
|
|
tbb::parallel_for_each(
|
|
files.begin(), files.end(),
|
|
[](ObjectFile *file) { file->parse(); });
|
|
parse_timer.stopTimer();
|
|
|
|
register_timer.startTimer();
|
|
// for (ObjectFile *file : files)
|
|
// file->register_defined_symbols();
|
|
tbb::parallel_for_each(
|
|
files.begin(), files.end(),
|
|
[](ObjectFile *file) { file->register_defined_symbols(); });
|
|
register_timer.stopTimer();
|
|
|
|
write();
|
|
return 0;
|
|
}
|