mirror of
https://codeberg.org/ziglang/zig.git
synced 2025-12-06 13:54:21 +00:00
Move fetch-them-macos-headers tools to ziglang/zig.
This commit is contained in:
parent
09a8fa2120
commit
ee47094a33
4 changed files with 1044 additions and 0 deletions
|
|
@ -31,6 +31,8 @@ pub fn build(b: *std.Build) void {
|
|||
const tools_target = b.resolveTargetQuery(.{});
|
||||
for ([_][]const u8{
|
||||
// Alphabetically sorted. No need to build `tools/spirv/grammar.zig`.
|
||||
"../../tools/fetch_them_macos_headers.zig",
|
||||
"../../tools/gen_macos_headers_c.zig",
|
||||
"../../tools/gen_outline_atomics.zig",
|
||||
"../../tools/gen_spirv_spec.zig",
|
||||
"../../tools/gen_stubs.zig",
|
||||
|
|
|
|||
738
tools/fetch_them_macos_headers.zig
Normal file
738
tools/fetch_them_macos_headers.zig
Normal file
|
|
@ -0,0 +1,738 @@
|
|||
const std = @import("std");
|
||||
const fs = std.fs;
|
||||
const io = std.io;
|
||||
const mem = std.mem;
|
||||
const process = std.process;
|
||||
const assert = std.debug.assert;
|
||||
const tmpDir = std.testing.tmpDir;
|
||||
|
||||
const Allocator = mem.Allocator;
|
||||
const Blake3 = std.crypto.hash.Blake3;
|
||||
const OsTag = std.Target.Os.Tag;
|
||||
|
||||
var general_purpose_allocator = std.heap.GeneralPurposeAllocator(.{}){};
|
||||
const gpa = general_purpose_allocator.allocator();
|
||||
|
||||
const Arch = enum {
|
||||
any,
|
||||
aarch64,
|
||||
x86_64,
|
||||
};
|
||||
|
||||
const Abi = enum { any, none };
|
||||
|
||||
const OsVer = enum(u32) {
|
||||
any = 0,
|
||||
catalina = 10,
|
||||
big_sur = 11,
|
||||
monterey = 12,
|
||||
ventura = 13,
|
||||
sonoma = 14,
|
||||
sequoia = 15,
|
||||
};
|
||||
|
||||
const Target = struct {
|
||||
arch: Arch,
|
||||
os: OsTag = .macos,
|
||||
os_ver: OsVer,
|
||||
abi: Abi = .none,
|
||||
|
||||
fn hash(a: Target) u32 {
|
||||
var hasher = std.hash.Wyhash.init(0);
|
||||
std.hash.autoHash(&hasher, a.arch);
|
||||
std.hash.autoHash(&hasher, a.os);
|
||||
std.hash.autoHash(&hasher, a.os_ver);
|
||||
std.hash.autoHash(&hasher, a.abi);
|
||||
return @as(u32, @truncate(hasher.final()));
|
||||
}
|
||||
|
||||
fn eql(a: Target, b: Target) bool {
|
||||
return a.arch == b.arch and
|
||||
a.os == b.os and
|
||||
a.os_ver == b.os_ver and
|
||||
a.abi == b.abi;
|
||||
}
|
||||
|
||||
fn name(self: Target, allocator: Allocator) ![]const u8 {
|
||||
return std.fmt.allocPrint(allocator, "{s}-{s}-{s}", .{
|
||||
@tagName(self.arch),
|
||||
@tagName(self.os),
|
||||
@tagName(self.abi),
|
||||
});
|
||||
}
|
||||
|
||||
fn fullName(self: Target, allocator: Allocator) ![]const u8 {
|
||||
if (self.os_ver == .any) return self.name(allocator);
|
||||
return std.fmt.allocPrint(allocator, "{s}-{s}.{d}-{s}", .{
|
||||
@tagName(self.arch),
|
||||
@tagName(self.os),
|
||||
@intFromEnum(self.os_ver),
|
||||
@tagName(self.abi),
|
||||
});
|
||||
}
|
||||
};
|
||||
|
||||
const targets = [_]Target{
|
||||
Target{
|
||||
.arch = .any,
|
||||
.abi = .any,
|
||||
.os_ver = .any,
|
||||
},
|
||||
Target{
|
||||
.arch = .aarch64,
|
||||
.os_ver = .any,
|
||||
},
|
||||
Target{
|
||||
.arch = .x86_64,
|
||||
.os_ver = .any,
|
||||
},
|
||||
Target{
|
||||
.arch = .x86_64,
|
||||
.os_ver = .catalina,
|
||||
},
|
||||
Target{
|
||||
.arch = .x86_64,
|
||||
.os_ver = .big_sur,
|
||||
},
|
||||
Target{
|
||||
.arch = .x86_64,
|
||||
.os_ver = .monterey,
|
||||
},
|
||||
Target{
|
||||
.arch = .x86_64,
|
||||
.os_ver = .ventura,
|
||||
},
|
||||
Target{
|
||||
.arch = .x86_64,
|
||||
.os_ver = .sonoma,
|
||||
},
|
||||
Target{
|
||||
.arch = .x86_64,
|
||||
.os_ver = .sequoia,
|
||||
},
|
||||
Target{
|
||||
.arch = .aarch64,
|
||||
.os_ver = .big_sur,
|
||||
},
|
||||
Target{
|
||||
.arch = .aarch64,
|
||||
.os_ver = .monterey,
|
||||
},
|
||||
Target{
|
||||
.arch = .aarch64,
|
||||
.os_ver = .ventura,
|
||||
},
|
||||
Target{
|
||||
.arch = .aarch64,
|
||||
.os_ver = .sonoma,
|
||||
},
|
||||
Target{
|
||||
.arch = .aarch64,
|
||||
.os_ver = .sequoia,
|
||||
},
|
||||
};
|
||||
|
||||
const headers_source_prefix: []const u8 = "headers";
|
||||
|
||||
const Contents = struct {
|
||||
bytes: []const u8,
|
||||
hit_count: usize,
|
||||
hash: []const u8,
|
||||
is_generic: bool,
|
||||
|
||||
fn hitCountLessThan(context: void, lhs: *const Contents, rhs: *const Contents) bool {
|
||||
_ = context;
|
||||
return lhs.hit_count < rhs.hit_count;
|
||||
}
|
||||
};
|
||||
|
||||
const TargetToHashContext = struct {
|
||||
pub fn hash(self: @This(), target: Target) u32 {
|
||||
_ = self;
|
||||
return target.hash();
|
||||
}
|
||||
pub fn eql(self: @This(), a: Target, b: Target, b_index: usize) bool {
|
||||
_ = self;
|
||||
_ = b_index;
|
||||
return a.eql(b);
|
||||
}
|
||||
};
|
||||
const TargetToHash = std.ArrayHashMap(Target, []const u8, TargetToHashContext, true);
|
||||
|
||||
const HashToContents = std.StringHashMap(Contents);
|
||||
const PathTable = std.StringHashMap(*TargetToHash);
|
||||
|
||||
/// The don't-dedup-list contains file paths with known problematic headers
|
||||
/// which while contain the same contents between architectures, should not be
|
||||
/// deduped since they contain includes, etc. which are relative and thus cannot be separated
|
||||
/// into a shared include dir such as `any-macos-any`.
|
||||
const dont_dedup_list = &[_][]const u8{
|
||||
"libkern/OSAtomic.h",
|
||||
"libkern/OSAtomicDeprecated.h",
|
||||
"libkern/OSSpinLockDeprecated.h",
|
||||
"libkern/OSAtomicQueue.h",
|
||||
};
|
||||
|
||||
fn generateDontDedupMap(arena: Allocator) !std.StringHashMap(void) {
|
||||
var map = std.StringHashMap(void).init(arena);
|
||||
try map.ensureTotalCapacity(dont_dedup_list.len);
|
||||
for (dont_dedup_list) |path| {
|
||||
map.putAssumeCapacityNoClobber(path, {});
|
||||
}
|
||||
return map;
|
||||
}
|
||||
|
||||
const usage =
|
||||
\\fetch_them_macos_headers fetch
|
||||
\\fetch_them_macos_headers dedup
|
||||
\\
|
||||
\\Commands:
|
||||
\\ fetch Fetch libc headers into headers/<arch>-macos.<os_ver> dir
|
||||
\\ dedup Generate deduplicated dirs into a given <destination> path
|
||||
\\
|
||||
\\General Options:
|
||||
\\-h, --help Print this help and exit
|
||||
;
|
||||
|
||||
pub fn main() anyerror!void {
|
||||
var arena = std.heap.ArenaAllocator.init(gpa);
|
||||
defer arena.deinit();
|
||||
|
||||
const all_args = try std.process.argsAlloc(arena.allocator());
|
||||
const args = all_args[1..];
|
||||
if (args.len == 0) fatal("no command or option specified", .{});
|
||||
|
||||
const cmd = args[0];
|
||||
if (mem.eql(u8, cmd, "--help") or mem.eql(u8, cmd, "-h")) {
|
||||
return info(usage, .{});
|
||||
} else if (mem.eql(u8, cmd, "dedup")) {
|
||||
return dedup(arena.allocator(), args[1..]);
|
||||
} else if (mem.eql(u8, cmd, "fetch")) {
|
||||
return fetch(arena.allocator(), args[1..]);
|
||||
} else fatal("unknown command or option: {s}", .{cmd});
|
||||
}
|
||||
|
||||
const ArgsIterator = struct {
|
||||
args: []const []const u8,
|
||||
i: usize = 0,
|
||||
|
||||
fn next(it: *@This()) ?[]const u8 {
|
||||
if (it.i >= it.args.len) {
|
||||
return null;
|
||||
}
|
||||
defer it.i += 1;
|
||||
return it.args[it.i];
|
||||
}
|
||||
|
||||
fn nextOrFatal(it: *@This()) []const u8 {
|
||||
const arg = it.next() orelse fatal("expected parameter after '{s}'", .{it.args[it.i - 1]});
|
||||
return arg;
|
||||
}
|
||||
};
|
||||
|
||||
fn info(comptime format: []const u8, args: anytype) void {
|
||||
const msg = std.fmt.allocPrint(gpa, "info: " ++ format ++ "\n", args) catch return;
|
||||
std.io.getStdOut().writeAll(msg) catch {};
|
||||
}
|
||||
|
||||
fn fatal(comptime format: []const u8, args: anytype) noreturn {
|
||||
ret: {
|
||||
const msg = std.fmt.allocPrint(gpa, "fatal: " ++ format ++ "\n", args) catch break :ret;
|
||||
std.io.getStdErr().writeAll(msg) catch {};
|
||||
}
|
||||
std.process.exit(1);
|
||||
}
|
||||
|
||||
const fetch_usage =
|
||||
\\fetch_them_macos_headers fetch
|
||||
\\
|
||||
\\Options:
|
||||
\\ --sysroot Path to macOS SDK
|
||||
\\
|
||||
\\General Options:
|
||||
\\-h, --help Print this help and exit
|
||||
;
|
||||
|
||||
fn fetch(arena: Allocator, args: []const []const u8) !void {
|
||||
var argv = std.ArrayList([]const u8).init(arena);
|
||||
var sysroot: ?[]const u8 = null;
|
||||
|
||||
var args_iter = ArgsIterator{ .args = args };
|
||||
while (args_iter.next()) |arg| {
|
||||
if (mem.eql(u8, arg, "--help") or mem.eql(u8, arg, "-h")) {
|
||||
return info(fetch_usage, .{});
|
||||
} else if (mem.eql(u8, arg, "--sysroot")) {
|
||||
sysroot = args_iter.nextOrFatal();
|
||||
} else try argv.append(arg);
|
||||
}
|
||||
|
||||
const sysroot_path = sysroot orelse blk: {
|
||||
const target = try std.zig.system.resolveTargetQuery(.{});
|
||||
break :blk std.zig.system.darwin.getSdk(arena, target) orelse
|
||||
fatal("no SDK found; you can provide one explicitly with '--sysroot' flag", .{});
|
||||
};
|
||||
|
||||
var sdk_dir = try std.fs.cwd().openDir(sysroot_path, .{});
|
||||
defer sdk_dir.close();
|
||||
const sdk_info = try sdk_dir.readFileAlloc(arena, "SDKSettings.json", std.math.maxInt(u32));
|
||||
|
||||
const parsed_json = try std.json.parseFromSlice(struct {
|
||||
DefaultProperties: struct { MACOSX_DEPLOYMENT_TARGET: []const u8 },
|
||||
}, arena, sdk_info, .{ .ignore_unknown_fields = true });
|
||||
|
||||
const version = Version.parse(parsed_json.value.DefaultProperties.MACOSX_DEPLOYMENT_TARGET) orelse
|
||||
fatal("don't know how to parse SDK version: {s}", .{
|
||||
parsed_json.value.DefaultProperties.MACOSX_DEPLOYMENT_TARGET,
|
||||
});
|
||||
const os_ver: OsVer = switch (version.major) {
|
||||
10 => .catalina,
|
||||
11 => .big_sur,
|
||||
12 => .monterey,
|
||||
13 => .ventura,
|
||||
14 => .sonoma,
|
||||
15 => .sequoia,
|
||||
else => unreachable,
|
||||
};
|
||||
info("found SDK deployment target macOS {} aka '{s}'", .{ version, @tagName(os_ver) });
|
||||
|
||||
var tmp = tmpDir(.{});
|
||||
defer tmp.cleanup();
|
||||
|
||||
for (&[_]Arch{ .aarch64, .x86_64 }) |arch| {
|
||||
const target: Target = .{
|
||||
.arch = arch,
|
||||
.os_ver = os_ver,
|
||||
};
|
||||
try fetchTarget(arena, argv.items, sysroot_path, target, version, tmp);
|
||||
}
|
||||
}
|
||||
|
||||
fn fetchTarget(
|
||||
arena: Allocator,
|
||||
args: []const []const u8,
|
||||
sysroot: []const u8,
|
||||
target: Target,
|
||||
ver: Version,
|
||||
tmp: std.testing.TmpDir,
|
||||
) !void {
|
||||
const tmp_filename = "headers";
|
||||
const headers_list_filename = "headers.o.d";
|
||||
const tmp_path = try tmp.dir.realpathAlloc(arena, ".");
|
||||
const tmp_file_path = try fs.path.join(arena, &[_][]const u8{ tmp_path, tmp_filename });
|
||||
const headers_list_path = try fs.path.join(arena, &[_][]const u8{ tmp_path, headers_list_filename });
|
||||
|
||||
const macos_version = try std.fmt.allocPrint(arena, "-mmacosx-version-min={d}.{d}", .{
|
||||
ver.major,
|
||||
ver.minor,
|
||||
});
|
||||
|
||||
var cc_argv = std.ArrayList([]const u8).init(arena);
|
||||
try cc_argv.appendSlice(&[_][]const u8{
|
||||
"cc",
|
||||
"-arch",
|
||||
switch (target.arch) {
|
||||
.x86_64 => "x86_64",
|
||||
.aarch64 => "arm64",
|
||||
else => unreachable,
|
||||
},
|
||||
macos_version,
|
||||
"-isysroot",
|
||||
sysroot,
|
||||
"-iwithsysroot",
|
||||
"/usr/include",
|
||||
"-o",
|
||||
tmp_file_path,
|
||||
"macos-headers.c",
|
||||
"-MD",
|
||||
"-MV",
|
||||
"-MF",
|
||||
headers_list_path,
|
||||
});
|
||||
try cc_argv.appendSlice(args);
|
||||
|
||||
// TODO instead of calling `cc` as a child process here,
|
||||
// hook in directly to `zig cc` API.
|
||||
const res = try std.process.Child.run(.{
|
||||
.allocator = arena,
|
||||
.argv = cc_argv.items,
|
||||
});
|
||||
|
||||
if (res.stderr.len != 0) {
|
||||
std.log.err("{s}", .{res.stderr});
|
||||
}
|
||||
|
||||
// Read in the contents of `upgrade.o.d`
|
||||
const headers_list_file = try tmp.dir.openFile(headers_list_filename, .{});
|
||||
defer headers_list_file.close();
|
||||
|
||||
var headers_dir = fs.cwd().openDir(headers_source_prefix, .{}) catch |err| switch (err) {
|
||||
error.FileNotFound,
|
||||
error.NotDir,
|
||||
=> fatal("path '{s}' not found or not a directory. Did you accidentally delete it?", .{
|
||||
headers_source_prefix,
|
||||
}),
|
||||
else => return err,
|
||||
};
|
||||
defer headers_dir.close();
|
||||
|
||||
const dest_path = try target.fullName(arena);
|
||||
try headers_dir.deleteTree(dest_path);
|
||||
|
||||
var dest_dir = try headers_dir.makeOpenPath(dest_path, .{});
|
||||
var dirs = std.StringHashMap(fs.Dir).init(arena);
|
||||
try dirs.putNoClobber(".", dest_dir);
|
||||
|
||||
const headers_list_str = try headers_list_file.reader().readAllAlloc(arena, std.math.maxInt(usize));
|
||||
const prefix = "/usr/include";
|
||||
|
||||
var it = mem.splitScalar(u8, headers_list_str, '\n');
|
||||
while (it.next()) |line| {
|
||||
if (mem.lastIndexOf(u8, line, "clang") != null) continue;
|
||||
if (mem.lastIndexOf(u8, line, prefix[0..])) |idx| {
|
||||
const out_rel_path = line[idx + prefix.len + 1 ..];
|
||||
const out_rel_path_stripped = mem.trim(u8, out_rel_path, " \\");
|
||||
const dirname = fs.path.dirname(out_rel_path_stripped) orelse ".";
|
||||
const maybe_dir = try dirs.getOrPut(dirname);
|
||||
if (!maybe_dir.found_existing) {
|
||||
maybe_dir.value_ptr.* = try dest_dir.makeOpenPath(dirname, .{});
|
||||
}
|
||||
const basename = fs.path.basename(out_rel_path_stripped);
|
||||
|
||||
const line_stripped = mem.trim(u8, line, " \\");
|
||||
const abs_dirname = fs.path.dirname(line_stripped).?;
|
||||
var orig_subdir = try fs.cwd().openDir(abs_dirname, .{});
|
||||
defer orig_subdir.close();
|
||||
|
||||
try orig_subdir.copyFile(basename, maybe_dir.value_ptr.*, basename, .{});
|
||||
}
|
||||
}
|
||||
|
||||
var dir_it = dirs.iterator();
|
||||
while (dir_it.next()) |entry| {
|
||||
entry.value_ptr.close();
|
||||
}
|
||||
}
|
||||
|
||||
const dedup_usage =
|
||||
\\fetch_them_macos_headers dedup [path]
|
||||
\\
|
||||
\\General Options:
|
||||
\\-h, --help Print this help and exit
|
||||
;
|
||||
|
||||
/// Dedups libs headers assuming the following layered structure:
|
||||
/// layer 1: x86_64-macos.10 x86_64-macos.11 x86_64-macos.12 aarch64-macos.11 aarch64-macos.12
|
||||
/// layer 2: any-macos.10 any-macos.11 any-macos.12
|
||||
/// layer 3: any-macos
|
||||
///
|
||||
/// The first layer consists of headers specific to a CPU architecture AND macOS version. The second
|
||||
/// layer consists of headers common to a macOS version across CPU architectures, and the final
|
||||
/// layer consists of headers common to all libc headers.
|
||||
fn dedup(arena: Allocator, args: []const []const u8) !void {
|
||||
var path: ?[]const u8 = null;
|
||||
var args_iter = ArgsIterator{ .args = args };
|
||||
while (args_iter.next()) |arg| {
|
||||
if (mem.eql(u8, arg, "--help") or mem.eql(u8, arg, "-h")) {
|
||||
return info(dedup_usage, .{});
|
||||
} else {
|
||||
if (path != null) fatal("too many arguments", .{});
|
||||
path = arg;
|
||||
}
|
||||
}
|
||||
|
||||
const dest_path = path orelse fatal("no destination path specified", .{});
|
||||
var dest_dir = fs.cwd().makeOpenPath(dest_path, .{}) catch |err| switch (err) {
|
||||
error.NotDir => fatal("path '{s}' not a directory", .{dest_path}),
|
||||
else => return err,
|
||||
};
|
||||
defer dest_dir.close();
|
||||
|
||||
var dont_dedup_map = try generateDontDedupMap(arena);
|
||||
var layer_2_targets = std.ArrayList(TargetWithPrefix).init(arena);
|
||||
|
||||
for (&[_]OsVer{ .catalina, .big_sur, .monterey, .ventura, .sonoma, .sequoia }) |os_ver| {
|
||||
var layer_1_targets = std.ArrayList(TargetWithPrefix).init(arena);
|
||||
|
||||
for (targets) |target| {
|
||||
if (target.os_ver != os_ver) continue;
|
||||
try layer_1_targets.append(.{
|
||||
.prefix = headers_source_prefix,
|
||||
.target = target,
|
||||
});
|
||||
}
|
||||
|
||||
if (layer_1_targets.items.len < 2) {
|
||||
try layer_2_targets.appendSlice(layer_1_targets.items);
|
||||
continue;
|
||||
}
|
||||
|
||||
const layer_2_target = try dedupDirs(arena, .{
|
||||
.os_ver = os_ver,
|
||||
.dest_path = dest_path,
|
||||
.dest_dir = dest_dir,
|
||||
.targets = layer_1_targets.items,
|
||||
.dont_dedup_map = &dont_dedup_map,
|
||||
});
|
||||
try layer_2_targets.append(layer_2_target);
|
||||
}
|
||||
|
||||
const layer_3_target = try dedupDirs(arena, .{
|
||||
.os_ver = .any,
|
||||
.dest_path = dest_path,
|
||||
.dest_dir = dest_dir,
|
||||
.targets = layer_2_targets.items,
|
||||
.dont_dedup_map = &dont_dedup_map,
|
||||
});
|
||||
assert(layer_3_target.target.eql(targets[0]));
|
||||
}
|
||||
|
||||
const TargetWithPrefix = struct {
|
||||
prefix: []const u8,
|
||||
target: Target,
|
||||
};
|
||||
|
||||
const DedupDirsArgs = struct {
|
||||
os_ver: OsVer,
|
||||
dest_path: []const u8,
|
||||
dest_dir: fs.Dir,
|
||||
targets: []const TargetWithPrefix,
|
||||
dont_dedup_map: *const std.StringHashMap(void),
|
||||
};
|
||||
|
||||
fn dedupDirs(arena: Allocator, args: DedupDirsArgs) !TargetWithPrefix {
|
||||
var tmp = tmpDir(.{ .iterate = true });
|
||||
defer tmp.cleanup();
|
||||
|
||||
var path_table = PathTable.init(arena);
|
||||
var hash_to_contents = HashToContents.init(arena);
|
||||
|
||||
var savings = FindResult{};
|
||||
for (args.targets) |target| {
|
||||
const res = try findDuplicates(target.target, arena, target.prefix, &path_table, &hash_to_contents);
|
||||
savings.max_bytes_saved += res.max_bytes_saved;
|
||||
savings.total_bytes += res.total_bytes;
|
||||
}
|
||||
|
||||
info("summary: {} could be reduced to {}", .{
|
||||
std.fmt.fmtIntSizeBin(savings.total_bytes),
|
||||
std.fmt.fmtIntSizeBin(savings.total_bytes - savings.max_bytes_saved),
|
||||
});
|
||||
|
||||
const output_target = Target{
|
||||
.arch = .any,
|
||||
.abi = .any,
|
||||
.os_ver = args.os_ver,
|
||||
};
|
||||
const common_name = try output_target.fullName(arena);
|
||||
|
||||
var missed_opportunity_bytes: usize = 0;
|
||||
// Iterate path_table. For each path, put all the hashes into a list. Sort by hit_count.
|
||||
// The hash with the highest hit_count gets to be the "generic" one. Everybody else
|
||||
// gets their header in a separate arch directory.
|
||||
var path_it = path_table.iterator();
|
||||
while (path_it.next()) |path_kv| {
|
||||
if (!args.dont_dedup_map.contains(path_kv.key_ptr.*)) {
|
||||
var contents_list = std.ArrayList(*Contents).init(arena);
|
||||
{
|
||||
var hash_it = path_kv.value_ptr.*.iterator();
|
||||
while (hash_it.next()) |hash_kv| {
|
||||
const contents = &hash_to_contents.getEntry(hash_kv.value_ptr.*).?.value_ptr.*;
|
||||
try contents_list.append(contents);
|
||||
}
|
||||
}
|
||||
std.mem.sort(*Contents, contents_list.items, {}, Contents.hitCountLessThan);
|
||||
const best_contents = contents_list.popOrNull().?;
|
||||
if (best_contents.hit_count > 1) {
|
||||
// Put it in `any-macos-none`.
|
||||
const full_path = try fs.path.join(arena, &[_][]const u8{ common_name, path_kv.key_ptr.* });
|
||||
try tmp.dir.makePath(fs.path.dirname(full_path).?);
|
||||
try tmp.dir.writeFile(.{ .sub_path = full_path, .data = best_contents.bytes });
|
||||
best_contents.is_generic = true;
|
||||
while (contents_list.popOrNull()) |contender| {
|
||||
if (contender.hit_count > 1) {
|
||||
const this_missed_bytes = contender.hit_count * contender.bytes.len;
|
||||
missed_opportunity_bytes += this_missed_bytes;
|
||||
info("Missed opportunity ({}): {s}", .{
|
||||
std.fmt.fmtIntSizeBin(this_missed_bytes),
|
||||
path_kv.key_ptr.*,
|
||||
});
|
||||
} else break;
|
||||
}
|
||||
}
|
||||
}
|
||||
var hash_it = path_kv.value_ptr.*.iterator();
|
||||
while (hash_it.next()) |hash_kv| {
|
||||
const contents = &hash_to_contents.getEntry(hash_kv.value_ptr.*).?.value_ptr.*;
|
||||
if (contents.is_generic) continue;
|
||||
|
||||
const target = hash_kv.key_ptr.*;
|
||||
const target_name = try target.fullName(arena);
|
||||
const full_path = try fs.path.join(arena, &[_][]const u8{ target_name, path_kv.key_ptr.* });
|
||||
try tmp.dir.makePath(fs.path.dirname(full_path).?);
|
||||
try tmp.dir.writeFile(.{ .sub_path = full_path, .data = contents.bytes });
|
||||
}
|
||||
}
|
||||
|
||||
for (args.targets) |target| {
|
||||
const target_name = try target.target.fullName(arena);
|
||||
try args.dest_dir.deleteTree(target_name);
|
||||
}
|
||||
try args.dest_dir.deleteTree(common_name);
|
||||
|
||||
var tmp_it = tmp.dir.iterate();
|
||||
while (try tmp_it.next()) |entry| {
|
||||
switch (entry.kind) {
|
||||
.directory => {
|
||||
const sub_dir = try tmp.dir.openDir(entry.name, .{ .iterate = true });
|
||||
const dest_sub_dir = try args.dest_dir.makeOpenPath(entry.name, .{});
|
||||
try copyDirAll(sub_dir, dest_sub_dir);
|
||||
},
|
||||
else => info("unexpected file format: not a directory: '{s}'", .{entry.name}),
|
||||
}
|
||||
}
|
||||
|
||||
return TargetWithPrefix{
|
||||
.prefix = args.dest_path,
|
||||
.target = output_target,
|
||||
};
|
||||
}
|
||||
|
||||
const FindResult = struct {
|
||||
max_bytes_saved: usize = 0,
|
||||
total_bytes: usize = 0,
|
||||
};
|
||||
|
||||
fn findDuplicates(
|
||||
target: Target,
|
||||
arena: Allocator,
|
||||
dest_path: []const u8,
|
||||
path_table: *PathTable,
|
||||
hash_to_contents: *HashToContents,
|
||||
) !FindResult {
|
||||
var result = FindResult{};
|
||||
|
||||
const target_name = try target.fullName(arena);
|
||||
const target_include_dir = try fs.path.join(arena, &[_][]const u8{ dest_path, target_name });
|
||||
var dir_stack = std.ArrayList([]const u8).init(arena);
|
||||
try dir_stack.append(target_include_dir);
|
||||
|
||||
while (dir_stack.popOrNull()) |full_dir_name| {
|
||||
var dir = fs.cwd().openDir(full_dir_name, .{ .iterate = true }) catch |err| switch (err) {
|
||||
error.FileNotFound => break,
|
||||
error.AccessDenied => break,
|
||||
else => return err,
|
||||
};
|
||||
defer dir.close();
|
||||
|
||||
var dir_it = dir.iterate();
|
||||
|
||||
while (try dir_it.next()) |entry| {
|
||||
const full_path = try fs.path.join(arena, &[_][]const u8{ full_dir_name, entry.name });
|
||||
switch (entry.kind) {
|
||||
.directory => try dir_stack.append(full_path),
|
||||
.file => {
|
||||
const rel_path = try fs.path.relative(arena, target_include_dir, full_path);
|
||||
const max_size = 2 * 1024 * 1024 * 1024;
|
||||
const raw_bytes = try fs.cwd().readFileAlloc(arena, full_path, max_size);
|
||||
const trimmed = mem.trim(u8, raw_bytes, " \r\n\t");
|
||||
result.total_bytes += raw_bytes.len;
|
||||
const hash = try arena.alloc(u8, 32);
|
||||
var hasher = Blake3.init(.{});
|
||||
hasher.update(rel_path);
|
||||
hasher.update(trimmed);
|
||||
hasher.final(hash);
|
||||
const gop = try hash_to_contents.getOrPut(hash);
|
||||
if (gop.found_existing) {
|
||||
result.max_bytes_saved += raw_bytes.len;
|
||||
gop.value_ptr.hit_count += 1;
|
||||
info("duplicate: {s} {s} ({})", .{
|
||||
target_name,
|
||||
rel_path,
|
||||
std.fmt.fmtIntSizeBin(raw_bytes.len),
|
||||
});
|
||||
} else {
|
||||
gop.value_ptr.* = Contents{
|
||||
.bytes = trimmed,
|
||||
.hit_count = 1,
|
||||
.hash = hash,
|
||||
.is_generic = false,
|
||||
};
|
||||
}
|
||||
const path_gop = try path_table.getOrPut(rel_path);
|
||||
const target_to_hash = if (path_gop.found_existing) path_gop.value_ptr.* else blk: {
|
||||
const ptr = try arena.create(TargetToHash);
|
||||
ptr.* = TargetToHash.init(arena);
|
||||
path_gop.value_ptr.* = ptr;
|
||||
break :blk ptr;
|
||||
};
|
||||
try target_to_hash.putNoClobber(target, hash);
|
||||
},
|
||||
else => info("unexpected file: {s}", .{full_path}),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
fn copyDirAll(source: fs.Dir, dest: fs.Dir) anyerror!void {
|
||||
var it = source.iterate();
|
||||
while (try it.next()) |next| {
|
||||
switch (next.kind) {
|
||||
.directory => {
|
||||
var sub_dir = try dest.makeOpenPath(next.name, .{});
|
||||
var sub_source = try source.openDir(next.name, .{ .iterate = true });
|
||||
defer {
|
||||
sub_dir.close();
|
||||
sub_source.close();
|
||||
}
|
||||
try copyDirAll(sub_source, sub_dir);
|
||||
},
|
||||
.file => {
|
||||
var source_file = try source.openFile(next.name, .{});
|
||||
var dest_file = try dest.createFile(next.name, .{});
|
||||
defer {
|
||||
source_file.close();
|
||||
dest_file.close();
|
||||
}
|
||||
const stat = try source_file.stat();
|
||||
const ncopied = try source_file.copyRangeAll(0, dest_file, 0, stat.size);
|
||||
assert(ncopied == stat.size);
|
||||
},
|
||||
else => |kind| info("unexpected file kind '{s}' will be ignored", .{@tagName(kind)}),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
const Version = struct {
|
||||
major: u16,
|
||||
minor: u8,
|
||||
patch: u8,
|
||||
|
||||
fn parse(raw: []const u8) ?Version {
|
||||
var parsed: [3]u16 = [_]u16{0} ** 3;
|
||||
var count: usize = 0;
|
||||
var it = std.mem.splitAny(u8, raw, ".");
|
||||
while (it.next()) |comp| {
|
||||
if (count >= 3) return null;
|
||||
parsed[count] = std.fmt.parseInt(u16, comp, 10) catch return null;
|
||||
count += 1;
|
||||
}
|
||||
if (count == 0) return null;
|
||||
const major = parsed[0];
|
||||
const minor = std.math.cast(u8, parsed[1]) orelse return null;
|
||||
const patch = std.math.cast(u8, parsed[2]) orelse return null;
|
||||
return .{ .major = major, .minor = minor, .patch = patch };
|
||||
}
|
||||
|
||||
pub fn format(
|
||||
v: Version,
|
||||
comptime unused_fmt_string: []const u8,
|
||||
options: std.fmt.FormatOptions,
|
||||
writer: anytype,
|
||||
) !void {
|
||||
_ = unused_fmt_string;
|
||||
_ = options;
|
||||
try writer.print("{d}.{d}.{d}", .{ v.major, v.minor, v.patch });
|
||||
}
|
||||
};
|
||||
97
tools/gen_macos_headers_c.zig
Normal file
97
tools/gen_macos_headers_c.zig
Normal file
|
|
@ -0,0 +1,97 @@
|
|||
const std = @import("std");
|
||||
const assert = std.debug.assert;
|
||||
|
||||
const Allocator = std.mem.Allocator;
|
||||
|
||||
var general_purpose_allocator = std.heap.GeneralPurposeAllocator(.{}){};
|
||||
const gpa = general_purpose_allocator.allocator();
|
||||
|
||||
const usage =
|
||||
\\gen_macos_headers_c [dir]
|
||||
\\
|
||||
\\General Options:
|
||||
\\-h, --help Print this help and exit
|
||||
;
|
||||
|
||||
fn info(comptime format: []const u8, args: anytype) void {
|
||||
const msg = std.fmt.allocPrint(gpa, "info: " ++ format ++ "\n", args) catch return;
|
||||
std.io.getStdOut().writeAll(msg) catch {};
|
||||
}
|
||||
|
||||
fn fatal(comptime format: []const u8, args: anytype) noreturn {
|
||||
ret: {
|
||||
const msg = std.fmt.allocPrint(gpa, "fatal: " ++ format ++ "\n", args) catch break :ret;
|
||||
std.io.getStdErr().writeAll(msg) catch {};
|
||||
}
|
||||
std.process.exit(1);
|
||||
}
|
||||
|
||||
pub fn main() anyerror!void {
|
||||
var arena_allocator = std.heap.ArenaAllocator.init(gpa);
|
||||
defer arena_allocator.deinit();
|
||||
const arena = arena_allocator.allocator();
|
||||
|
||||
const args = try std.process.argsAlloc(arena);
|
||||
if (args.len == 1) fatal("no command or option specified", .{});
|
||||
|
||||
var positionals = std.ArrayList([]const u8).init(arena);
|
||||
|
||||
for (args[1..]) |arg| {
|
||||
if (std.mem.eql(u8, arg, "--help") or std.mem.eql(u8, arg, "-h")) {
|
||||
return info(usage, .{});
|
||||
} else try positionals.append(arg);
|
||||
}
|
||||
|
||||
if (positionals.items.len != 1) fatal("expected one positional argument: [dir]", .{});
|
||||
|
||||
var dir = try std.fs.cwd().openDir(positionals.items[0], .{ .no_follow = true });
|
||||
defer dir.close();
|
||||
var paths = std.ArrayList([]const u8).init(arena);
|
||||
try findHeaders(arena, dir, "", &paths);
|
||||
|
||||
const SortFn = struct {
|
||||
pub fn lessThan(ctx: void, lhs: []const u8, rhs: []const u8) bool {
|
||||
_ = ctx;
|
||||
return std.mem.lessThan(u8, lhs, rhs);
|
||||
}
|
||||
};
|
||||
|
||||
std.mem.sort([]const u8, paths.items, {}, SortFn.lessThan);
|
||||
|
||||
const stdout = std.io.getStdOut().writer();
|
||||
try stdout.writeAll("#define _XOPEN_SOURCE\n");
|
||||
for (paths.items) |path| {
|
||||
try stdout.print("#include <{s}>\n", .{path});
|
||||
}
|
||||
try stdout.writeAll(
|
||||
\\int main(int argc, char **argv) {
|
||||
\\ return 0;
|
||||
\\}
|
||||
);
|
||||
}
|
||||
|
||||
fn findHeaders(
|
||||
arena: Allocator,
|
||||
dir: std.fs.Dir,
|
||||
prefix: []const u8,
|
||||
paths: *std.ArrayList([]const u8),
|
||||
) anyerror!void {
|
||||
var it = dir.iterate();
|
||||
while (try it.next()) |entry| {
|
||||
switch (entry.kind) {
|
||||
.directory => {
|
||||
const path = try std.fs.path.join(arena, &.{ prefix, entry.name });
|
||||
var subdir = try dir.openDir(entry.name, .{ .no_follow = true });
|
||||
defer subdir.close();
|
||||
try findHeaders(arena, subdir, path, paths);
|
||||
},
|
||||
.file, .sym_link => {
|
||||
const ext = std.fs.path.extension(entry.name);
|
||||
if (!std.mem.eql(u8, ext, ".h")) continue;
|
||||
const path = try std.fs.path.join(arena, &.{ prefix, entry.name });
|
||||
try paths.append(path);
|
||||
},
|
||||
else => {},
|
||||
}
|
||||
}
|
||||
}
|
||||
207
tools/macos-headers.c
Normal file
207
tools/macos-headers.c
Normal file
|
|
@ -0,0 +1,207 @@
|
|||
// Source: https://en.wikipedia.org/wiki/C_standard_library#Header_files
|
||||
#include <assert.h>
|
||||
#include <complex.h>
|
||||
#include <ctype.h>
|
||||
#include <errno.h>
|
||||
#include <fenv.h>
|
||||
#include <float.h>
|
||||
#include <getopt.h>
|
||||
#include <inttypes.h>
|
||||
#include <iso646.h>
|
||||
#include <limits.h>
|
||||
#include <locale.h>
|
||||
#include <math.h>
|
||||
#include <setjmp.h>
|
||||
#include <signal.h>
|
||||
#include <stdalign.h>
|
||||
#include <stdarg.h>
|
||||
#include <stdatomic.h>
|
||||
#include <stdbool.h>
|
||||
#include <stddef.h>
|
||||
#include <stdint.h>
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <stdnoreturn.h>
|
||||
#include <string.h>
|
||||
#include <tgmath.h>
|
||||
#include <time.h>
|
||||
#include <wchar.h>
|
||||
#include <wctype.h>
|
||||
|
||||
// Source: https://en.wikipedia.org/wiki/C_standard_library#BSD_libc
|
||||
#include <fts.h>
|
||||
#include <db.h>
|
||||
#include <err.h>
|
||||
#include <vis.h>
|
||||
|
||||
// Source: https://en.wikipedia.org/wiki/C_POSIX_library
|
||||
#include <aio.h>
|
||||
#include <arpa/inet.h>
|
||||
#include <assert.h>
|
||||
#include <complex.h>
|
||||
#include <cpio.h>
|
||||
#include <ctype.h>
|
||||
#include <dirent.h>
|
||||
#include <dlfcn.h>
|
||||
#include <errno.h>
|
||||
#include <fcntl.h>
|
||||
#include <fenv.h>
|
||||
#include <float.h>
|
||||
#include <fmtmsg.h>
|
||||
#include <fnmatch.h>
|
||||
#include <ftw.h>
|
||||
#include <glob.h>
|
||||
#include <grp.h>
|
||||
#include <iconv.h>
|
||||
#include <inttypes.h>
|
||||
#include <iso646.h>
|
||||
#include <langinfo.h>
|
||||
#include <libgen.h>
|
||||
#include <limits.h>
|
||||
#include <locale.h>
|
||||
#include <math.h>
|
||||
#include <monetary.h>
|
||||
/* #include <mqueue.h> - not found on macos catalina */
|
||||
#include <ndbm.h>
|
||||
#include <net/if.h>
|
||||
#include <netdb.h>
|
||||
#include <netinet/in.h>
|
||||
#include <netinet/ip.h>
|
||||
#include <netinet/tcp.h>
|
||||
#include <netinet/udp.h>
|
||||
#include <nl_types.h>
|
||||
#include <poll.h>
|
||||
#include <pthread.h>
|
||||
#include <pwd.h>
|
||||
#include <regex.h>
|
||||
#include <sched.h>
|
||||
#include <search.h>
|
||||
#include <semaphore.h>
|
||||
#include <setjmp.h>
|
||||
#include <signal.h>
|
||||
#include <spawn.h>
|
||||
#include <stdarg.h>
|
||||
#include <stdbool.h>
|
||||
#include <stddef.h>
|
||||
#include <stdint.h>
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <strings.h>
|
||||
/* #include <stropts.h> - not found on macos catalina */
|
||||
#include <sys/ipc.h>
|
||||
#include <sys/mman.h>
|
||||
#include <sys/msg.h>
|
||||
#include <sys/random.h>
|
||||
#include <sys/resource.h>
|
||||
#include <sys/select.h>
|
||||
#include <sys/sem.h>
|
||||
#include <sys/shm.h>
|
||||
#include <sys/socket.h>
|
||||
#include <sys/stat.h>
|
||||
#include <sys/statvfs.h>
|
||||
#include <sys/time.h>
|
||||
#include <sys/times.h>
|
||||
#include <sys/timex.h>
|
||||
#include <sys/types.h>
|
||||
#include <sys/uio.h>
|
||||
#include <sys/un.h>
|
||||
#include <sys/utsname.h>
|
||||
#include <sys/wait.h>
|
||||
#include <syslog.h>
|
||||
#include <tar.h>
|
||||
#include <termios.h>
|
||||
#include <tgmath.h>
|
||||
#include <time.h>
|
||||
/* #include <trace.h> - not found on macos catalina */
|
||||
#include <ulimit.h>
|
||||
#include <unistd.h>
|
||||
#include <utime.h>
|
||||
#include <utmpx.h>
|
||||
#include <wchar.h>
|
||||
#include <wctype.h>
|
||||
#include <wordexp.h>
|
||||
|
||||
// macOS system headers
|
||||
#include <mach/clock.h>
|
||||
#include <mach/mach.h>
|
||||
#include <mach/mach_time.h>
|
||||
#include <mach/thread_state.h>
|
||||
#include <mach/vm_param.h>
|
||||
#include <sys/acl.h>
|
||||
#include <sys/attr.h>
|
||||
#include <sys/ioctl.h>
|
||||
#include <sys/mount.h>
|
||||
#include <sys/param.h>
|
||||
#include <sys/sysctl.h>
|
||||
#include <sys/clonefile.h>
|
||||
#include <libproc.h>
|
||||
|
||||
// Depended on by libcxx
|
||||
#include <Block.h>
|
||||
#include <xlocale.h>
|
||||
#include <copyfile.h>
|
||||
#include <mach-o/dyld.h>
|
||||
#include <mach-o/fat.h>
|
||||
#include <mach-o/nlist.h>
|
||||
#include <mach-o/reloc.h>
|
||||
#include <mach-o/arch.h>
|
||||
#include <mach-o/stab.h>
|
||||
#include <mach-o/ranlib.h>
|
||||
#include <mach-o/compact_unwind_encoding.h>
|
||||
#include <mach-o/arm64/reloc.h>
|
||||
#include <mach-o/x86_64/reloc.h>
|
||||
#include <ar.h>
|
||||
|
||||
// Depended on by LLVM
|
||||
#include <sysexits.h>
|
||||
#include <crt_externs.h>
|
||||
#include <execinfo.h>
|
||||
|
||||
// Depended on by several frameworks
|
||||
#include <AssertMacros.h>
|
||||
#include <device/device_types.h>
|
||||
#include <dispatch/dispatch.h>
|
||||
#include <hfs/hfs_format.h>
|
||||
#include <hfs/hfs_unistr.h>
|
||||
#include <libkern/OSAtomic.h>
|
||||
#include <libkern/OSAtomicQueue.h>
|
||||
#include <libkern/OSByteOrder.h>
|
||||
#include <libkern/OSCacheControl.h>
|
||||
#include <libkern/OSDebug.h>
|
||||
#include <libkern/OSKextLib.h>
|
||||
#include <libkern/OSReturn.h>
|
||||
#include <libkern/OSThermalNotification.h>
|
||||
#include <libkern/OSTypes.h>
|
||||
#include <MacTypes.h>
|
||||
#include <os/lock.h>
|
||||
#include <simd/simd.h>
|
||||
#include <xpc/xpc.h>
|
||||
#include <CommonCrypto/CommonDigest.h>
|
||||
|
||||
#include <objc/message.h>
|
||||
#include <objc/NSObject.h>
|
||||
#include <objc/NSObjCRuntime.h>
|
||||
#include <objc/objc.h>
|
||||
#include <objc/objc-runtime.h>
|
||||
|
||||
// Depended on by libuv
|
||||
#include <paths.h>
|
||||
#include <ifaddrs.h>
|
||||
#include <net/if_dl.h>
|
||||
#include <sys/paths.h>
|
||||
|
||||
// Depended on by sqlite-amalgamation
|
||||
#include <sys/file.h>
|
||||
#include <malloc/malloc.h>
|
||||
|
||||
// Provided by macOS LibC
|
||||
#include <memory.h>
|
||||
#include <zlib.h>
|
||||
|
||||
#define _XOPEN_SOURCE
|
||||
#include <ucontext.h>
|
||||
|
||||
int main(int argc, char **argv) {
|
||||
return 0;
|
||||
}
|
||||
Loading…
Add table
Reference in a new issue