| ... | @@ -11,61 +11,68 @@ pub fn rescanMac(cb: *Bundle, gpa: Allocator) RescanMacError!void { | ... | @@ -11,61 +11,68 @@ pub fn rescanMac(cb: *Bundle, gpa: Allocator) RescanMacError!void { |
| 11 | cb.bytes.clearRetainingCapacity(); | 11 | cb.bytes.clearRetainingCapacity(); |
| 12 | cb.map.clearRetainingCapacity(); | 12 | cb.map.clearRetainingCapacity(); |
| 13 | | 13 | |
| 14 | const file = try fs.openFileAbsolute("/System/Library/Keychains/SystemRootCertificates.keychain", .{}); | 14 | const keychainPaths = [2][]const u8{ |
| 15 | defer file.close(); | 15 | "/System/Library/Keychains/SystemRootCertificates.keychain", |
| | 16 | "/Library/Keychains/System.keychain", |
| | 17 | }; |
| 16 | | 18 | |
| 17 | const bytes = try file.readToEndAlloc(gpa, std.math.maxInt(u32)); | 19 | for (keychainPaths) |keychainPath| { |
| 18 | defer gpa.free(bytes); | 20 | const file = try fs.openFileAbsolute(keychainPath, .{}); |
| | 21 | defer file.close(); |
| 19 | | 22 | |
| 20 | var stream = std.io.fixedBufferStream(bytes); | 23 | const bytes = try file.readToEndAlloc(gpa, std.math.maxInt(u32)); |
| 21 | const reader = stream.reader(); | 24 | defer gpa.free(bytes); |
| 22 | | 25 | |
| 23 | const db_header = try reader.readStructEndian(ApplDbHeader, .big); | 26 | var stream = std.io.fixedBufferStream(bytes); |
| 24 | assert(mem.eql(u8, &db_header.signature, "kych")); | 27 | const reader = stream.reader(); |
| 25 | | 28 | |
| 26 | try stream.seekTo(db_header.schema_offset); | 29 | const db_header = try reader.readStructEndian(ApplDbHeader, .big); |
| | 30 | assert(mem.eql(u8, &db_header.signature, "kych")); |
| 27 | | 31 | |
| 28 | const db_schema = try reader.readStructEndian(ApplDbSchema, .big); | 32 | try stream.seekTo(db_header.schema_offset); |
| 29 | | 33 | |
| 30 | var table_list = try gpa.alloc(u32, db_schema.table_count); | 34 | const db_schema = try reader.readStructEndian(ApplDbSchema, .big); |
| 31 | defer gpa.free(table_list); | | |
| 32 | | 35 | |
| 33 | var table_idx: u32 = 0; | 36 | var table_list = try gpa.alloc(u32, db_schema.table_count); |
| 34 | while (table_idx < table_list.len) : (table_idx += 1) { | 37 | defer gpa.free(table_list); |
| 35 | table_list[table_idx] = try reader.readInt(u32, .big); | | |
| 36 | } | | |
| 37 | | 38 | |
| 38 | const now_sec = std.time.timestamp(); | 39 | var table_idx: u32 = 0; |
| | 40 | while (table_idx < table_list.len) : (table_idx += 1) { |
| | 41 | table_list[table_idx] = try reader.readInt(u32, .big); |
| | 42 | } |
| 39 | | 43 | |
| 40 | for (table_list) |table_offset| { | 44 | const now_sec = std.time.timestamp(); |
| 41 | try stream.seekTo(db_header.schema_offset + table_offset); | | |
| 42 | | 45 | |
| 43 | const table_header = try reader.readStructEndian(TableHeader, .big); | 46 | for (table_list) |table_offset| { |
| | 47 | try stream.seekTo(db_header.schema_offset + table_offset); |
| 44 | | 48 | |
| 45 | if (@as(std.c.DB_RECORDTYPE, @enumFromInt(table_header.table_id)) != .X509_CERTIFICATE) { | 49 | const table_header = try reader.readStructEndian(TableHeader, .big); |
| 46 | continue; | | |
| 47 | } | | |
| 48 | | 50 | |
| 49 | var record_list = try gpa.alloc(u32, table_header.record_count); | 51 | if (@as(std.c.DB_RECORDTYPE, @enumFromInt(table_header.table_id)) != .X509_CERTIFICATE) { |
| 50 | defer gpa.free(record_list); | 52 | continue; |
| | 53 | } |
| 51 | | 54 | |
| 52 | var record_idx: u32 = 0; | 55 | var record_list = try gpa.alloc(u32, table_header.record_count); |
| 53 | while (record_idx < record_list.len) : (record_idx += 1) { | 56 | defer gpa.free(record_list); |
| 54 | record_list[record_idx] = try reader.readInt(u32, .big); | 57 | |
| 55 | } | 58 | var record_idx: u32 = 0; |
| | 59 | while (record_idx < record_list.len) : (record_idx += 1) { |
| | 60 | record_list[record_idx] = try reader.readInt(u32, .big); |
| | 61 | } |
| 56 | | 62 | |
| 57 | for (record_list) |record_offset| { | 63 | for (record_list) |record_offset| { |
| 58 | try stream.seekTo(db_header.schema_offset + table_offset + record_offset); | 64 | try stream.seekTo(db_header.schema_offset + table_offset + record_offset); |
| 59 | | 65 | |
| 60 | const cert_header = try reader.readStructEndian(X509CertHeader, .big); | 66 | const cert_header = try reader.readStructEndian(X509CertHeader, .big); |
| 61 | | 67 | |
| 62 | try cb.bytes.ensureUnusedCapacity(gpa, cert_header.cert_size); | 68 | try cb.bytes.ensureUnusedCapacity(gpa, cert_header.cert_size); |
| 63 | | 69 | |
| 64 | const cert_start = @as(u32, @intCast(cb.bytes.items.len)); | 70 | const cert_start = @as(u32, @intCast(cb.bytes.items.len)); |
| 65 | const dest_buf = cb.bytes.allocatedSlice()[cert_start..]; | 71 | const dest_buf = cb.bytes.allocatedSlice()[cert_start..]; |
| 66 | cb.bytes.items.len += try reader.readAtLeast(dest_buf, cert_header.cert_size); | 72 | cb.bytes.items.len += try reader.readAtLeast(dest_buf, cert_header.cert_size); |
| 67 | | 73 | |
| 68 | try cb.parseCert(gpa, cert_start, now_sec); | 74 | try cb.parseCert(gpa, cert_start, now_sec); |
| | 75 | } |
| 69 | } | 76 | } |
| 70 | } | 77 | } |
| 71 | | 78 | |