aboutsummaryrefslogtreecommitdiff
path: root/cmd/receivepack9418/conn.go
blob: 755cf0225cc2932262f5f6cbaafc0487a8851d08 (about) (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
package main

import (
	"bufio"
	"context"
	"fmt"
	"log"
	"net"
	"os"
	"strings"

	"codeberg.org/lindenii/furgit/network/receivepack"
	objectdual "codeberg.org/lindenii/furgit/object/store/dual"
	objectloose "codeberg.org/lindenii/furgit/object/store/loose"
	objectpacked "codeberg.org/lindenii/furgit/object/store/packed"
)

func (srv *server) handleConn(conn net.Conn) {
	defer func() { _ = conn.Close() }()

	reader := bufio.NewReader(conn)
	writer := bufio.NewWriter(conn)

	req, err := readGitProtoRequest(reader)
	if err != nil {
		writeErrPkt(writer, fmt.Sprintf("invalid initial request: %v", err))
		_ = writer.Flush()

		log.Printf("receivepack9418: %s: invalid initial request: %v", conn.RemoteAddr(), err)

		return
	}

	if req.Command != "git-receive-pack" {
		writeErrPkt(writer, fmt.Sprintf("unsupported command %q", req.Command))
		_ = writer.Flush()

		log.Printf("receivepack9418: %s: unsupported command %q", conn.RemoteAddr(), req.Command)

		return
	}

	gitProtocol := strings.Join(req.ExtraParameters, ":")

	objectIngress, cleanupObjectIngress, err := srv.openObjectIngress()
	if err != nil {
		writeErrPkt(writer, fmt.Sprintf("object ingress unavailable: %v", err))
		_ = writer.Flush()

		log.Printf("receivepack9418: %s: object ingress unavailable: %v", conn.RemoteAddr(), err)

		return
	}

	defer cleanupObjectIngress()

	opts := receivepack.Options{
		GitProtocol:     gitProtocol,
		Algorithm:       srv.repo.Algorithm(),
		Refs:            srv.repo.Refs(),
		ExistingObjects: srv.repo.Objects(),
		ObjectIngress:   objectIngress,
	}

	err = receivepack.ReceivePack(context.Background(), writer, reader, opts)
	if err != nil {
		_ = writer.Flush()

		log.Printf(
			"receivepack9418: %s: receive-pack failed (path=%q host=%q extras=%v): %v",
			conn.RemoteAddr(),
			req.Pathname,
			req.Host,
			req.ExtraParameters,
			err,
		)

		return
	}

	err = writer.Flush()
	if err != nil {
		log.Printf("receivepack9418: %s: flush failed: %v", conn.RemoteAddr(), err)

		return
	}
}

func (srv *server) openObjectIngress() (*objectdual.Dual, func(), error) {
	err := srv.objectsRoot.Mkdir("pack", 0o755)
	if err != nil && !os.IsExist(err) {
		return nil, nil, err
	}

	packRoot, err := srv.objectsRoot.OpenRoot("pack")
	if err != nil {
		return nil, nil, err
	}

	looseStore, err := objectloose.New(srv.objectsRoot, srv.repo.Algorithm())
	if err != nil {
		_ = packRoot.Close()

		return nil, nil, err
	}

	packedStore, err := objectpacked.New(packRoot, srv.repo.Algorithm(), objectpacked.Options{WriteRev: true})
	if err != nil {
		_ = looseStore.Close()
		_ = packRoot.Close()

		return nil, nil, err
	}

	cleanup := func() {
		_ = packedStore.Close()
		_ = looseStore.Close()
		_ = packRoot.Close()
	}

	return objectdual.New(looseStore, packedStore), cleanup, nil
}