╭─────────────╮ │scroll to top│ ╰─────────────╯ ╭ serve_ssh.zig ────────────────────────────────────────────────╮ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ │ ╰───────────────────────────────────────────────────────────────╯
╭ find file ─ ─ ─ ─ ─ ─ ─ ─ ─ ─╮ ╭────╮ ╭ clone ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ╮ │ │ │http│ ssh │ │ ╰ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─╯ ╰────╯ ╰ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ─ ╯
544 try push.receiveFork(repo.self_repo_opts, io, allocator, repo, &target_repo, handler.server, &route.id, author, timestamp, reader, writer, handler.max_repo_size, handler.err, sess); 545 }, 546 } 547 } 548 } 549 550 fn parseGitCommand(allocator: std.mem.Allocator, command: []const u8) !ParsedGitCommand { 551 var tokens = try std.process.Args.IteratorGeneral(.{ .single_quotes = true }).init(allocator, command); 552 defer tokens.deinit(); 553 554 const service_token = tokens.next() orelse return error.InvalidCommand; 555 const dir_token = tokens.next() orelse return error.InvalidCommand; 556 557 const service: GitService = 558 if (std.mem.eql(u8, service_token, "git-upload-pack") or std.mem.eql(u8, service_token, "upload-pack")) 559 .upload_pack 560 else if (std.mem.eql(u8, service_token, "git-receive-pack") or std.mem.eql(u8, service_token, "receive-pack")) 561 .receive_pack 562 else 563 return error.UnsupportedService; 564 565 return .{ .service = service, .dir = try allocator.dupe(u8, dir_token) }; 566 } 567 568 const RepoAuthorization = union(enum) { allowed: Allowed, create: RepoCreator, denied, not_found }; 569 570 // what a key may do in an existing repo 571 const Allowed = union(enum) { 572 fetch, 573 push: Pusher, 574 }; 575 576 // the key holder a push acts as 577 const Pusher = struct { 578 user_id: [evt.event_id_size]u8, 579 author: evt.CommitAuthor, 580 protected_branches: []const []const u8, 581 locked_branches: []const []const u8, 582 // only owners may clear the undo history 583 owner: bool, 584 }; 585 586 // the owner of a repo a push may create 587 const RepoCreator = struct { 588 owner_id: [evt.event_id_size]u8, 589 owner: evt.CommitAuthor, 590 }; 591 592 fn authorizeRepoKey( 593 io: std.Io, 594 arena: *std.heap.ArenaAllocator, 595 admin_repo_path: []const u8, 596 users_dir: []const u8, 597 owner_name: []const u8, 598 repo_name: []const u8, 599 service: GitService, 600 fingerprint: *const [ssh.fingerprint_len]u8, 601 ) !RepoAuthorization { 602 const allocator = arena.child_allocator; 603 var admin = rp.Repo(.xit, evt.admin_repo_opts).open(io, allocator, .{ .path = admin_repo_path }) catch |err| switch (err) { 604 error.RepoNotFound => return .not_found, 605 else => |e| return e, 606 }; 607 defer admin.deinit(io, allocator); 608 609 const moment = try evt.currentMoment(evt.admin_repo_opts, &admin); 610 // no viewer yet, so this carries the base role 611 const repo = (try evt.readRepoByOwnerAndName(io, allocator, arena, moment, users_dir, owner_name, repo_name, null)) orelse { 612 if (service == .upload_pack) return .not_found; 613 const owner_id = (try evt.User.readIdByName(evt.AdminDB, evt.admin_repo_opts.hash, moment, owner_name)) orelse return .not_found; 614 if (!try evt.User.holdsKey(evt.AdminDB, evt.admin_repo_opts.hash, moment, arena, &owner_id, fingerprint)) return .denied; 615 return .{ .create = .{ .owner_id = owner_id, .owner = try commitAuthor(moment, arena, &owner_id) } }; 616 }; 617 618 const min_role: evt.Repo.Role = switch (service) { 619 .upload_pack => .read, 620 .receive_pack => .write, 621 }; 622 // any key may fetch a public repo; a push always needs a key holder 623 if (service == .upload_pack and repo.role != null) return .{ .allowed = .fetch }; 624 625 // a key can be shared, so the push acts as the holder with the highest role 626 const owner_id = try evt.parseEventId(&repo.repo.event.user_id); 627 var best: ?struct { id: [evt.event_id_size]u8, role: evt.Repo.Role } = null; 628 for (try evt.User.readIdsByFingerprint(evt.AdminDB, evt.admin_repo_opts.hash, moment, arena, fingerprint)) |holder_id| { 629 const seen = (try evt.readRepoById(io, allocator, arena, users_dir, &owner_id, &repo.event_id, holder_id)) orelse return .not_found; 630 const role = seen.role orelse continue; 631 if (best) |held| if (held.role.atLeast(role)) continue; 632 best = .{ .id = holder_id, .role = role }; 633 } 634 const holder = best orelse return .denied; 635 if (!holder.role.atLeast(min_role)) return .denied; 636 return .{ 637 .allowed = switch (service) { 638 .upload_pack => .fetch, 639 .receive_pack => blk: { 640 const names = try repo.repo.event.protectedBranchNames(arena.allocator()); 641 // members change protected branches only by merging a patch 642 const owner = holder.role == .owner; 643 break :blk .{ .push = .{ 644 .user_id = holder.id, 645 .author = try commitAuthor(moment, arena, &holder.id), 646 .protected_branches = if (owner) names else &.{}, 647 .locked_branches = if (owner) &.{} else names, 648 .owner = owner, 649 } }; 650 }, 651 }, 652 }; 653 } 654 655 // an active user as a commit author 656 fn commitAuthor( 657 moment: evt.AdminDB.HashMap(.read_only), 658 arena: *std.heap.ArenaAllocator, 659 user_id: *const [evt.event_id_size]u8, 660 ) !evt.CommitAuthor { 661 const user = (try evt.User.readById(evt.AdminDB, evt.admin_repo_opts.hash, moment, arena, user_id)) orelse unreachable; 662 return .{ .name = user.event.name, .email = user.event.email }; 663 } 664 665 // the authorized_keys line of a pubkey blob: its type, then the blob in base64 666 fn authorizedKeyLine(allocator: std.mem.Allocator, blob: []const u8) ![]const u8 { 667 var reader = std.Io.Reader.fixed(blob); 668 const key_type = try ssh.takeString(&reader, 64); 669 const encoder = std.base64.standard.Encoder; 670 const line = try allocator.alloc(u8, key_type.len + 1 + encoder.calcSize(blob.len)); 671 @memcpy(line[0..key_type.len], key_type); 672 line[key_type.len] = ' '; 673 _ = encoder.encode(line[key_type.len + 1 ..], blob); 674 return line; 675 } 676 677 fn writeError(sess: *ssh.SessionCtx, comptime msg: []const u8) !void { 678 const body = "ERR haxy ssh: " ++ msg; 679 var packet: [body.len + 4]u8 = undefined; 680 const line = try std.fmt.bufPrint(&packet, "{x:0>4}{s}", .{ packet.len, body }); 681 try sess.writeBytes(line); 682 try sess.exit(1); 683 } 684 685 // a refusal naming the key the client presented. a key on no account gets its 686 // own message, since clients send whichever key they offer first. 687 fn writeKeyError(handler: *const SessionHandler, sess: *ssh.SessionCtx, comptime msg: []const u8) !void { 688 const io = sess.conn.io; 689 const allocator = sess.conn.allocator; 690 const known = known: { 691 var arena = std.heap.ArenaAllocator.init(allocator); 692 defer arena.deinit(); 693 var admin = try rp.Repo(.xit, evt.admin_repo_opts).open(io, allocator, .{ .path = handler.admin_repo_path }); 694 defer admin.deinit(io, allocator); 695 const moment = try evt.currentMoment(evt.admin_repo_opts, &admin); 696 break :known (try evt.User.readIdsByFingerprint(evt.AdminDB, evt.admin_repo_opts.hash, moment, &arena, &sess.fingerprint)).len > 0; 697 }; 698 if (known) return writeFingerprintError(sess, msg); 699 return writeFingerprintError(sess, "unauthorized: this SSH key isn't added to any haxy account. if you have several keys, your SSH client may be offering a different one first"); 700 } 701 702 // an error line ending with the presented key's fingerprint 703 fn writeFingerprintError(sess: *ssh.SessionCtx, comptime msg: []const u8) !void { 704 const prefix = "ERR haxy ssh: " ++ msg ++ " (key "; 705 var packet: [4 + prefix.len + ssh.fingerprint_len + 1]u8 = undefined; 706 const line = try std.fmt.bufPrint(&packet, "{x:0>4}" ++ prefix ++ "{s})", .{ packet.len, &sess.fingerprint }); 707 try sess.writeBytes(line); 708 try sess.exit(1); 709 } 710 711 test "parseGitCommand rejects malformed input" { 712 const allocator = std.testing.allocator; 713 try std.testing.expectError(error.InvalidCommand, parseGitCommand(allocator, "git-upload-pack")); 714 try std.testing.expectError(error.UnsupportedService, parseGitCommand(allocator, "ls -la")); 715 try std.testing.expectError(error.UnsupportedService, parseGitCommand(allocator, "git-fake-pack 'repo'")); 716 } 717 718 test "parseGitCommand happy paths" { 719 const allocator = std.testing.allocator; 720 721 { 722 const parsed = try parseGitCommand(allocator, "git-upload-pack 'some-repo'"); 723 defer parsed.deinit(allocator); 724 try std.testing.expectEqual(GitService.upload_pack, parsed.service); 725 try std.testing.expectEqualStrings("some-repo", parsed.dir); 726 } 727 { 728 const parsed = try parseGitCommand(allocator, "git-receive-pack 'user/proj'"); 729 defer parsed.deinit(allocator); 730 try std.testing.expectEqual(GitService.receive_pack, parsed.service); 731 try std.testing.expectEqualStrings("user/proj", parsed.dir); 732 } 733 { 734 const parsed = try parseGitCommand(allocator, "git-upload-pack repo"); 735 defer parsed.deinit(allocator); 736 try std.testing.expectEqual(GitService.upload_pack, parsed.service); 737 try std.testing.expectEqualStrings("repo", parsed.dir); 738 } 739 } 740