mirror of
https://github.com/rui314/mold.git
synced 2024-11-11 05:46:58 +03:00
104 lines
2.7 KiB
C++
104 lines
2.7 KiB
C++
#include "mold.h"
|
|
|
|
namespace mold {
|
|
|
|
// A tar file consists of one or more Ustar header followed by data.
|
|
// Each Ustar header represents a single file in an archive.
|
|
//
|
|
// tar is an old file format, and its `name` field is only 100 bytes long.
|
|
// If `name` is longer than 100 bytes, we can emit a PAX header before a
|
|
// Ustar header to store a long filename.
|
|
//
|
|
// For simplicity, we always emit a PAX header even for a short filename.
|
|
struct UstarHeader {
|
|
void flush() {
|
|
memset(checksum, ' ', sizeof(checksum));
|
|
memcpy(magic, "ustar", 5);
|
|
memcpy(version, "00", 2);
|
|
|
|
// Compute checksum
|
|
int sum = 0;
|
|
for (i64 i = 0; i < sizeof(*this); i++)
|
|
sum += ((u8 *)this)[i];
|
|
sprintf(checksum, "%06o", sum);
|
|
}
|
|
|
|
char name[100];
|
|
char mode[8];
|
|
char uid[8];
|
|
char gid[8];
|
|
char size[12];
|
|
char mtime[12];
|
|
char checksum[8];
|
|
char typeflag[1];
|
|
char linkname[100];
|
|
char magic[6];
|
|
char version[2];
|
|
char uname[32];
|
|
char gname[32];
|
|
char devmajor[8];
|
|
char devminor[8];
|
|
char prefix[155];
|
|
char pad[12];
|
|
};
|
|
|
|
std::string TarFile::encode_path(std::string path) {
|
|
path = path_clean(basedir + "/" + path);
|
|
|
|
// Construct a string which contains something like
|
|
// "16 path=foo/bar\n" where 16 is the size of the string
|
|
// including the size string itself.
|
|
i64 len = std::string(" path=\n").size() + path.size();
|
|
i64 total = std::to_string(len).size() + len;
|
|
total = std::to_string(total).size() + len;
|
|
return std::to_string(total) + " path=" + path + "\n";
|
|
}
|
|
|
|
void TarFile::append(std::string path, std::string_view data) {
|
|
contents.push_back({path, data});
|
|
|
|
size_ += BLOCK_SIZE * 2;
|
|
size_ += align_to(encode_path(path).size(), BLOCK_SIZE);
|
|
size_ += align_to(data.size(), BLOCK_SIZE);
|
|
}
|
|
|
|
void TarFile::write_to(u8 *buf) {
|
|
u8 *start = buf;
|
|
memset(buf, 0, size_);
|
|
|
|
for (i64 i = 0; i < contents.size(); i++) {
|
|
assert(buf - start <= size_);
|
|
|
|
const std::string &path = contents[i].first;
|
|
std::string_view data = contents[i].second;
|
|
|
|
// Write PAX header
|
|
static_assert(sizeof(UstarHeader) == BLOCK_SIZE);
|
|
UstarHeader &pax = *(UstarHeader *)buf;
|
|
buf += BLOCK_SIZE;
|
|
|
|
std::string attr = encode_path(path);
|
|
sprintf(pax.size, "%011zo", attr.size());
|
|
pax.typeflag[0] = 'x';
|
|
pax.flush();
|
|
|
|
// Write pathname
|
|
memcpy(buf, attr.data(), attr.size());
|
|
buf += align_to(attr.size(), BLOCK_SIZE);
|
|
|
|
// Write Ustar header
|
|
UstarHeader &ustar = *(UstarHeader *)buf;
|
|
buf += BLOCK_SIZE;
|
|
|
|
memcpy(ustar.mode, "0000664", 8);
|
|
sprintf(ustar.size, "%011zo", data.size());
|
|
ustar.flush();
|
|
|
|
// Write file contents
|
|
memcpy(buf, data.data(), data.size());
|
|
buf += align_to(data.size(), BLOCK_SIZE);
|
|
}
|
|
}
|
|
|
|
} // namespace mold
|