Skip to content

Commit 286d2ae

Browse files
wxiaoguangwxiaoguang
committed
fix
1 parent 31ddbe1 commit 286d2ae

File tree

10 files changed

+70
-21
lines changed

10 files changed

+70
-21
lines changed

‎modules/lfstransfer/backend/backend.go

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -70,14 +70,13 @@ func (g *GiteaBackend) Batch(_ string, pointers []transfer.BatchItem, args trans
7070
g.logger.Log("json marshal error", err)
7171
return nil, err
7272
}
73-
url := g.server.JoinPath("objects/batch").String()
7473
headers := map[string]string{
7574
headerAuthorization: g.authToken,
7675
headerGiteaInternalAuth: g.internalAuth,
7776
headerAccept: mimeGitLFS,
7877
headerContentType: mimeGitLFS,
7978
}
80-
req := newInternalRequestLFS(g.ctx, url, http.MethodPost, headers, bodyBytes)
79+
req := newInternalRequestLFS(g.ctx, g.server.JoinPath("objects/batch").String(), http.MethodPost, headers, bodyBytes)
8180
resp, err := req.Response()
8281
if err != nil {
8382
g.logger.Log("http request error", err)
@@ -179,13 +178,12 @@ func (g *GiteaBackend) Download(oid string, args transfer.Args) (io.ReadCloser,
179178
g.logger.Log("argument id incorrect")
180179
return nil, 0, transfer.ErrCorruptData
181180
}
182-
url := action.Href
183181
headers := map[string]string{
184182
headerAuthorization: g.authToken,
185183
headerGiteaInternalAuth: g.internalAuth,
186184
headerAccept: mimeOctetStream,
187185
}
188-
req := newInternalRequestLFS(g.ctx, url, http.MethodGet, headers, nil)
186+
req := newInternalRequestLFS(g.ctx, toInternalLFSURL(action.Href), http.MethodGet, headers, nil)
189187
resp, err := req.Response()
190188
if err != nil {
191189
return nil, 0, fmt.Errorf("failed to get response: %w", err)
@@ -225,15 +223,14 @@ func (g *GiteaBackend) Upload(oid string, size int64, r io.Reader, args transfer
225223
g.logger.Log("argument id incorrect")
226224
return transfer.ErrCorruptData
227225
}
228-
url := action.Href
229226
headers := map[string]string{
230227
headerAuthorization: g.authToken,
231228
headerGiteaInternalAuth: g.internalAuth,
232229
headerContentType: mimeOctetStream,
233230
headerContentLength: strconv.FormatInt(size, 10),
234231
}
235232

236-
req := newInternalRequestLFS(g.ctx, url, http.MethodPut, headers, nil)
233+
req := newInternalRequestLFS(g.ctx, toInternalLFSURL(action.Href), http.MethodPut, headers, nil)
237234
req.Body(r)
238235
resp, err := req.Response()
239236
if err != nil {
@@ -274,14 +271,13 @@ func (g *GiteaBackend) Verify(oid string, size int64, args transfer.Args) (trans
274271
// the server sent no verify action
275272
return transfer.SuccessStatus(), nil
276273
}
277-
url := action.Href
278274
headers := map[string]string{
279275
headerAuthorization: g.authToken,
280276
headerGiteaInternalAuth: g.internalAuth,
281277
headerAccept: mimeGitLFS,
282278
headerContentType: mimeGitLFS,
283279
}
284-
req := newInternalRequestLFS(g.ctx, url, http.MethodPost, headers, bodyBytes)
280+
req := newInternalRequestLFS(g.ctx, toInternalLFSURL(action.Href), http.MethodPost, headers, bodyBytes)
285281
resp, err := req.Response()
286282
if err != nil {
287283
return transfer.NewStatus(transfer.StatusInternalServerError), err

‎modules/lfstransfer/backend/lock.go

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -43,14 +43,13 @@ func (g *giteaLockBackend) Create(path, refname string) (transfer.Lock, error) {
4343
g.logger.Log("json marshal error", err)
4444
return nil, err
4545
}
46-
url := g.server.String()
4746
headers := map[string]string{
4847
headerAuthorization: g.authToken,
4948
headerGiteaInternalAuth: g.internalAuth,
5049
headerAccept: mimeGitLFS,
5150
headerContentType: mimeGitLFS,
5251
}
53-
req := newInternalRequestLFS(g.ctx, url, http.MethodPost, headers, bodyBytes)
52+
req := newInternalRequestLFS(g.ctx, g.server.String(), http.MethodPost, headers, bodyBytes)
5453
resp, err := req.Response()
5554
if err != nil {
5655
g.logger.Log("http request error", err)
@@ -95,14 +94,13 @@ func (g *giteaLockBackend) Unlock(lock transfer.Lock) error {
9594
g.logger.Log("json marshal error", err)
9695
return err
9796
}
98-
url := g.server.JoinPath(lock.ID(), "unlock").String()
9997
headers := map[string]string{
10098
headerAuthorization: g.authToken,
10199
headerGiteaInternalAuth: g.internalAuth,
102100
headerAccept: mimeGitLFS,
103101
headerContentType: mimeGitLFS,
104102
}
105-
req := newInternalRequestLFS(g.ctx, url, http.MethodPost, headers, bodyBytes)
103+
req := newInternalRequestLFS(g.ctx, g.server.JoinPath(lock.ID(), "unlock").String(), http.MethodPost, headers, bodyBytes)
106104
resp, err := req.Response()
107105
if err != nil {
108106
g.logger.Log("http request error", err)
@@ -176,16 +174,15 @@ func (g *giteaLockBackend) Range(cursor string, limit int, iter func(transfer.Lo
176174
}
177175

178176
func (g *giteaLockBackend) queryLocks(v url.Values) ([]transfer.Lock, string, error) {
179-
urlq := g.server.JoinPath() // get a copy
180-
urlq.RawQuery = v.Encode()
181-
url := urlq.String()
177+
serverURLWithQuery := g.server.JoinPath() // get a copy
178+
serverURLWithQuery.RawQuery = v.Encode()
182179
headers := map[string]string{
183180
headerAuthorization: g.authToken,
184181
headerGiteaInternalAuth: g.internalAuth,
185182
headerAccept: mimeGitLFS,
186183
headerContentType: mimeGitLFS,
187184
}
188-
req := newInternalRequestLFS(g.ctx, url, http.MethodGet, headers, nil)
185+
req := newInternalRequestLFS(g.ctx, serverURLWithQuery.String(), http.MethodGet, headers, nil)
189186
resp, err := req.Response()
190187
if err != nil {
191188
g.logger.Log("http request error", err)

‎modules/lfstransfer/backend/util.go

Lines changed: 18 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -8,9 +8,11 @@ import (
88
"fmt"
99
"io"
1010
"net/http"
11+
"strings"
1112

1213
"code.gitea.io/gitea/modules/httplib"
1314
"code.gitea.io/gitea/modules/private"
15+
"code.gitea.io/gitea/modules/setting"
1416

1517
"github.com/charmbracelet/git-lfs-transfer/transfer"
1618
)
@@ -57,8 +59,7 @@ const (
5759

5860
// Operations enum
5961
const (
60-
opNone = iota
61-
opDownload
62+
opDownload = iota + 1
6263
opUpload
6364
)
6465

@@ -86,8 +87,21 @@ func statusCodeToErr(code int) error {
8687
}
8788
}
8889

89-
func newInternalRequestLFS(ctx context.Context, url, method string, headers map[string]string, body any) *httplib.Request {
90-
req := private.NewInternalRequest(ctx, url, method)
90+
func toInternalLFSURL(s string) string {
91+
pos1 := strings.Index(s, "://")
92+
if pos1 == -1 {
93+
return ""
94+
}
95+
appSubURLWithSlash := setting.AppSubURL + "/"
96+
pos2 := strings.Index(s[pos1+3:], appSubURLWithSlash)
97+
if pos2 == -1 {
98+
return ""
99+
}
100+
return setting.LocalURL + "api/internal/repo/" + s[pos1+3+pos2+len(appSubURLWithSlash):]
101+
}
102+
103+
func newInternalRequestLFS(ctx context.Context, internalURL, method string, headers map[string]string, body any) *httplib.Request {
104+
req := private.NewInternalRequest(ctx, internalURL, method)
91105
for k, v := range headers {
92106
req.Header(k, v)
93107
}

‎modules/lfstransfer/backend/util_test.go

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,29 @@
1+
// Copyright 2025 The Gitea Authors. All rights reserved.
2+
// SPDX-License-Identifier: MIT
3+
4+
package backend
5+
6+
import (
7+
"testing"
8+
9+
"code.gitea.io/gitea/modules/setting"
10+
"code.gitea.io/gitea/modules/test"
11+
12+
"github.com/stretchr/testify/assert"
13+
)
14+
15+
func TestToInternalLFSURL(t *testing.T) {
16+
defer test.MockVariableValue(&setting.LocalURL, "http://localurl/")()
17+
defer test.MockVariableValue(&setting.AppSubURL, "/sub")()
18+
cases := []struct {
19+
url string
20+
expected string
21+
}{
22+
{"http://appurl/any", ""},
23+
{"http://appurl/sub/any", "http://localurl/api/internal/repo/any"},
24+
{"http://localurl/sub/any", "http://localurl/api/internal/repo/any"},
25+
}
26+
for _, c := range cases {
27+
assert.Equal(t, c.expected, toInternalLFSURL(c.url), c.url)
28+
}
29+
}

‎modules/private/internal.go

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,10 @@ func NewInternalRequest(ctx context.Context, url, method string) *httplib.Reques
4040
Ensure you are running in the correct environment or set the correct configuration file with -c.`, setting.CustomConf)
4141
}
4242

43+
if !strings.HasPrefix(url, setting.LocalURL) {
44+
log.Fatal("Invalid internal request URL: %q", url)
45+
}
46+
4347
req := httplib.NewRequest(url, method).
4448
SetContext(ctx).
4549
Header("X-Real-IP", getClientIP()).

‎tests/integration/git_lfs_ssh_test.go

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,9 +54,14 @@ func TestGitLFSSSH(t *testing.T) {
5454
return strings.Contains(s, "POST /api/internal/repo/user2/repo1.git/info/lfs/objects/batch")
5555
})
5656
countUpload := slices.ContainsFunc(routerCalls, func(s string) bool {
57-
return strings.Contains(s, "PUT /user2/repo1.git/info/lfs/objects/")
57+
return strings.Contains(s, "PUT /api/internal/repo/user2/repo1.git/info/lfs/objects/")
58+
})
59+
nonAPIRequests := slices.ContainsFunc(routerCalls, func(s string) bool {
60+
fields := strings.Fields(s)
61+
return !strings.HasPrefix(fields[1], "/api/")
5862
})
5963
assert.NotZero(t, countBatch)
6064
assert.NotZero(t, countUpload)
65+
assert.Zero(t, nonAPIRequests)
6166
})
6267
}

‎tests/mssql.ini.tmpl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@ SIGNING_KEY = none
4545
SSH_DOMAIN = localhost
4646
HTTP_PORT = 3003
4747
ROOT_URL = http://localhost:3003/
48+
LOCAL_ROOT_URL = http://127.0.0.1:3003/
4849
DISABLE_SSH = false
4950
SSH_LISTEN_HOST = localhost
5051
SSH_PORT = 2201

‎tests/mysql.ini.tmpl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,7 @@ SIGNING_KEY = none
4747
SSH_DOMAIN = localhost
4848
HTTP_PORT = 3001
4949
ROOT_URL = http://localhost:3001/
50+
LOCAL_ROOT_URL = http://127.0.0.1:3001/
5051
DISABLE_SSH = false
5152
SSH_LISTEN_HOST = localhost
5253
SSH_PORT = 2201

‎tests/pgsql.ini.tmpl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@ SIGNING_KEY = none
4646
SSH_DOMAIN = localhost
4747
HTTP_PORT = 3002
4848
ROOT_URL = http://localhost:3002/
49+
LOCAL_ROOT_URL = http://127.0.0.1:3002/
4950
DISABLE_SSH = false
5051
SSH_LISTEN_HOST = localhost
5152
SSH_PORT = 2202

‎tests/sqlite.ini.tmpl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,7 @@ SIGNING_KEY = none
4141
SSH_DOMAIN = localhost
4242
HTTP_PORT = 3003
4343
ROOT_URL = http://localhost:3003/
44+
LOCAL_ROOT_URL = http://127.0.0.1:3003/
4445
DISABLE_SSH = false
4546
SSH_LISTEN_HOST = localhost
4647
SSH_PORT = 2203

0 commit comments

Comments
 (0)