1
1
mirror of https://github.com/rui314/mold.git synced 2024-10-06 01:28:12 +03:00
mold/main.cc
2022-01-03 18:15:51 +09:00

62 lines
1.5 KiB
C++

#include "elf/mold.h"
#include "macho/mold.h"
#include <cstring>
#include <signal.h>
namespace mold {
std::string_view errno_string() {
static thread_local char buf[200];
// There are two incompatible strerror_r implementations as follows.
//
// GNU: char *strerror_r(int, char *, size_t)
// POSIX: int strerror_r(int, char *, size_t)
//
// GNU version may write an error message to a buffer other than the
// given one and returns a pointer to the error message. POSIX version
// always write an error message to a given buffer.
if (std::is_same<decltype(strerror_r(errno, buf, sizeof(buf))), char *>::value)
return reinterpret_cast<char *>(strerror_r(errno, buf, sizeof(buf)));
strerror_r(errno, buf, sizeof(buf));
return buf;
}
#ifdef GIT_HASH
const std::string mold_version =
"mold " MOLD_VERSION " (" GIT_HASH "; compatible with GNU ld and GNU gold)";
#else
const std::string mold_version =
"mold " MOLD_VERSION " (compatible with GNU ld and GNU gold)";
#endif
void cleanup() {
if (output_tmpfile)
unlink(output_tmpfile);
if (socket_tmpfile)
unlink(socket_tmpfile);
}
static void signal_handler(int) {
cleanup();
_exit(1);
}
void install_signal_handler() {
signal(SIGINT, signal_handler);
signal(SIGTERM, signal_handler);
}
} // namespace mold
int main(int argc, char **argv) {
std::string cmd = mold::filepath(argv[0]).filename();
if (cmd == "ld64" || cmd == "ld64.mold")
return mold::macho::main(argc, argv);
return mold::elf::main(argc, argv);
}