2021-07-28 04:42:56 -05:00
|
|
|
// Copyright 2021 The Gitea Authors. All rights reserved.
|
2022-11-27 13:20:29 -05:00
|
|
|
// SPDX-License-Identifier: MIT
|
2021-07-28 04:42:56 -05:00
|
|
|
|
|
|
|
package cmd
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bufio"
|
|
|
|
"bytes"
|
Refactor internal API for git commands, use meaningful messages instead of "Internal Server Error" (#23687)
# Why this PR comes
At first, I'd like to help users like #23636 (there are a lot)
The unclear "Internal Server Error" is quite anonying, scare users,
frustrate contributors, nobody knows what happens.
So, it's always good to provide meaningful messages to end users (of
course, do not leak sensitive information).
When I started working on the "response message to end users", I found
that the related code has a lot of technical debt. A lot of copy&paste
code, unclear fields and usages.
So I think it's good to make everything clear.
# Tech Backgrounds
Gitea has many sub-commands, some are used by admins, some are used by
SSH servers or Git Hooks. Many sub-commands use "internal API" to
communicate with Gitea web server.
Before, Gitea server always use `StatusCode + Json "err" field` to
return messages.
* The CLI sub-commands: they expect to show all error related messages
to site admin
* The Serv/Hook sub-commands (for git clients): they could only show
safe messages to end users, the error log could only be recorded by
"SSHLog" to Gitea web server.
In the old design, it assumes that:
* If the StatusCode is 500 (in some functions), then the "err" field is
error log, shouldn't be exposed to git client.
* If the StatusCode is 40x, then the "err" field could be exposed. And
some functions always read the "err" no matter what the StatusCode is.
The old code is not strict, and it's difficult to distinguish the
messages clearly and then output them correctly.
# This PR
To help to remove duplicate code and make everything clear, this PR
introduces `ResponseExtra` and `requestJSONResp`.
* `ResponseExtra` is a struct which contains "extra" information of a
internal API response, including StatusCode, UserMsg, Error
* `requestJSONResp` is a generic function which can be used for all
cases to help to simplify the calls.
* Remove all `map["err"]`, always use `private.Response{Err}` to
construct error messages.
* User messages and error messages are separated clearly, the `fail` and
`handleCliResponseExtra` will output correct messages.
* Replace all `Internal Server Error` messages with meaningful (still
safe) messages.
This PR saves more than 300 lines, while makes the git client messages
more clear.
Many gitea-serv/git-hook related essential functions are covered by
tests.
---------
Co-authored-by: delvh <dev.lh@web.de>
2023-03-29 01:32:26 -05:00
|
|
|
"context"
|
2024-04-02 18:17:25 -05:00
|
|
|
"io"
|
|
|
|
"net/http"
|
|
|
|
"net/http/httptest"
|
|
|
|
"os"
|
2021-07-28 04:42:56 -05:00
|
|
|
"strings"
|
|
|
|
"testing"
|
2024-04-02 18:17:25 -05:00
|
|
|
"time"
|
|
|
|
|
|
|
|
"code.gitea.io/gitea/modules/setting"
|
|
|
|
"code.gitea.io/gitea/modules/test"
|
2021-07-28 04:42:56 -05:00
|
|
|
|
|
|
|
"github.com/stretchr/testify/assert"
|
2024-04-02 18:17:25 -05:00
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/urfave/cli/v2"
|
2021-07-28 04:42:56 -05:00
|
|
|
)
|
|
|
|
|
2024-04-06 08:43:45 -05:00
|
|
|
// Capture what's being written into a standard file descriptor.
|
|
|
|
func captureOutput(t *testing.T, stdFD *os.File) (finish func() (output string)) {
|
|
|
|
t.Helper()
|
|
|
|
|
|
|
|
r, w, err := os.Pipe()
|
|
|
|
require.NoError(t, err)
|
|
|
|
resetStdout := test.MockVariableValue(stdFD, *w)
|
|
|
|
|
|
|
|
return func() (output string) {
|
|
|
|
w.Close()
|
|
|
|
resetStdout()
|
|
|
|
|
|
|
|
out, err := io.ReadAll(r)
|
|
|
|
require.NoError(t, err)
|
|
|
|
return string(out)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-07-28 04:42:56 -05:00
|
|
|
func TestPktLine(t *testing.T) {
|
Refactor internal API for git commands, use meaningful messages instead of "Internal Server Error" (#23687)
# Why this PR comes
At first, I'd like to help users like #23636 (there are a lot)
The unclear "Internal Server Error" is quite anonying, scare users,
frustrate contributors, nobody knows what happens.
So, it's always good to provide meaningful messages to end users (of
course, do not leak sensitive information).
When I started working on the "response message to end users", I found
that the related code has a lot of technical debt. A lot of copy&paste
code, unclear fields and usages.
So I think it's good to make everything clear.
# Tech Backgrounds
Gitea has many sub-commands, some are used by admins, some are used by
SSH servers or Git Hooks. Many sub-commands use "internal API" to
communicate with Gitea web server.
Before, Gitea server always use `StatusCode + Json "err" field` to
return messages.
* The CLI sub-commands: they expect to show all error related messages
to site admin
* The Serv/Hook sub-commands (for git clients): they could only show
safe messages to end users, the error log could only be recorded by
"SSHLog" to Gitea web server.
In the old design, it assumes that:
* If the StatusCode is 500 (in some functions), then the "err" field is
error log, shouldn't be exposed to git client.
* If the StatusCode is 40x, then the "err" field could be exposed. And
some functions always read the "err" no matter what the StatusCode is.
The old code is not strict, and it's difficult to distinguish the
messages clearly and then output them correctly.
# This PR
To help to remove duplicate code and make everything clear, this PR
introduces `ResponseExtra` and `requestJSONResp`.
* `ResponseExtra` is a struct which contains "extra" information of a
internal API response, including StatusCode, UserMsg, Error
* `requestJSONResp` is a generic function which can be used for all
cases to help to simplify the calls.
* Remove all `map["err"]`, always use `private.Response{Err}` to
construct error messages.
* User messages and error messages are separated clearly, the `fail` and
`handleCliResponseExtra` will output correct messages.
* Replace all `Internal Server Error` messages with meaningful (still
safe) messages.
This PR saves more than 300 lines, while makes the git client messages
more clear.
Many gitea-serv/git-hook related essential functions are covered by
tests.
---------
Co-authored-by: delvh <dev.lh@web.de>
2023-03-29 01:32:26 -05:00
|
|
|
ctx := context.Background()
|
2024-03-28 18:20:21 -05:00
|
|
|
|
|
|
|
t.Run("Read", func(t *testing.T) {
|
|
|
|
s := strings.NewReader("0000")
|
|
|
|
r := bufio.NewReader(s)
|
|
|
|
result, err := readPktLine(ctx, r, pktLineTypeFlush)
|
|
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, pktLineTypeFlush, result.Type)
|
|
|
|
|
|
|
|
s = strings.NewReader("0006a\n")
|
|
|
|
r = bufio.NewReader(s)
|
|
|
|
result, err = readPktLine(ctx, r, pktLineTypeData)
|
|
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, pktLineTypeData, result.Type)
|
|
|
|
assert.Equal(t, []byte("a\n"), result.Data)
|
|
|
|
|
|
|
|
s = strings.NewReader("0004")
|
|
|
|
r = bufio.NewReader(s)
|
|
|
|
result, err = readPktLine(ctx, r, pktLineTypeData)
|
|
|
|
assert.Error(t, err)
|
|
|
|
assert.Nil(t, result)
|
|
|
|
|
|
|
|
data := strings.Repeat("x", 65516)
|
|
|
|
r = bufio.NewReader(strings.NewReader("fff0" + data))
|
|
|
|
result, err = readPktLine(ctx, r, pktLineTypeData)
|
|
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, pktLineTypeData, result.Type)
|
|
|
|
assert.Equal(t, []byte(data), result.Data)
|
|
|
|
|
|
|
|
r = bufio.NewReader(strings.NewReader("fff1a"))
|
|
|
|
result, err = readPktLine(ctx, r, pktLineTypeData)
|
|
|
|
assert.Error(t, err)
|
|
|
|
assert.Nil(t, result)
|
|
|
|
})
|
|
|
|
|
|
|
|
t.Run("Write", func(t *testing.T) {
|
|
|
|
w := bytes.NewBuffer([]byte{})
|
|
|
|
err := writeFlushPktLine(ctx, w)
|
|
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, []byte("0000"), w.Bytes())
|
|
|
|
|
|
|
|
w.Reset()
|
|
|
|
err = writeDataPktLine(ctx, w, []byte("a\nb"))
|
|
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, []byte("0007a\nb"), w.Bytes())
|
|
|
|
|
|
|
|
w.Reset()
|
|
|
|
data := bytes.Repeat([]byte{0x05}, 288)
|
|
|
|
err = writeDataPktLine(ctx, w, data)
|
|
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, append([]byte("0124"), data...), w.Bytes())
|
|
|
|
|
|
|
|
w.Reset()
|
|
|
|
err = writeDataPktLine(ctx, w, nil)
|
|
|
|
assert.Error(t, err)
|
|
|
|
assert.Empty(t, w.Bytes())
|
|
|
|
|
|
|
|
w.Reset()
|
|
|
|
data = bytes.Repeat([]byte{0x64}, 65516)
|
|
|
|
err = writeDataPktLine(ctx, w, data)
|
|
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, append([]byte("fff0"), data...), w.Bytes())
|
|
|
|
|
|
|
|
w.Reset()
|
|
|
|
err = writeDataPktLine(ctx, w, bytes.Repeat([]byte{0x64}, 65516+1))
|
|
|
|
assert.Error(t, err)
|
|
|
|
assert.Empty(t, w.Bytes())
|
|
|
|
})
|
2021-07-28 04:42:56 -05:00
|
|
|
}
|
2024-04-02 18:17:25 -05:00
|
|
|
|
|
|
|
func TestDelayWriter(t *testing.T) {
|
|
|
|
// Setup the environment.
|
|
|
|
defer test.MockVariableValue(&setting.InternalToken, "Random")()
|
|
|
|
defer test.MockVariableValue(&setting.InstallLock, true)()
|
|
|
|
defer test.MockVariableValue(&setting.Git.VerbosePush, true)()
|
|
|
|
require.NoError(t, os.Setenv("SSH_ORIGINAL_COMMAND", "true"))
|
|
|
|
|
|
|
|
// Setup the Stdin.
|
|
|
|
f, err := os.OpenFile(t.TempDir()+"/stdin", os.O_RDWR|os.O_CREATE|os.O_EXCL, 0o666)
|
|
|
|
require.NoError(t, err)
|
|
|
|
_, err = f.Write([]byte("00000000000000000000 00000000000000000001 refs/head/main\n"))
|
|
|
|
require.NoError(t, err)
|
|
|
|
_, err = f.Seek(0, 0)
|
|
|
|
require.NoError(t, err)
|
|
|
|
defer test.MockVariableValue(os.Stdin, *f)()
|
|
|
|
|
|
|
|
// Setup the server that processes the hooks.
|
|
|
|
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
|
|
time.Sleep(time.Millisecond * 600)
|
|
|
|
}))
|
|
|
|
defer ts.Close()
|
|
|
|
defer test.MockVariableValue(&setting.LocalURL, ts.URL+"/")()
|
|
|
|
|
|
|
|
app := cli.NewApp()
|
|
|
|
app.Commands = []*cli.Command{subcmdHookPreReceive}
|
|
|
|
|
|
|
|
t.Run("Should delay", func(t *testing.T) {
|
|
|
|
defer test.MockVariableValue(&setting.Git.VerbosePushDelay, time.Millisecond*500)()
|
2024-04-06 08:43:45 -05:00
|
|
|
finish := captureOutput(t, os.Stdout)
|
2024-04-02 18:17:25 -05:00
|
|
|
|
|
|
|
err = app.Run([]string{"./forgejo", "pre-receive"})
|
|
|
|
require.NoError(t, err)
|
|
|
|
out := finish()
|
|
|
|
|
|
|
|
require.Contains(t, out, "* Checking 1 references")
|
|
|
|
require.Contains(t, out, "Checked 1 references in total")
|
|
|
|
})
|
|
|
|
|
|
|
|
t.Run("Shouldn't delay", func(t *testing.T) {
|
|
|
|
defer test.MockVariableValue(&setting.Git.VerbosePushDelay, time.Second*5)()
|
2024-04-06 08:43:45 -05:00
|
|
|
finish := captureOutput(t, os.Stdout)
|
2024-04-02 18:17:25 -05:00
|
|
|
|
|
|
|
err = app.Run([]string{"./forgejo", "pre-receive"})
|
|
|
|
require.NoError(t, err)
|
|
|
|
out := finish()
|
|
|
|
|
|
|
|
require.NoError(t, err)
|
|
|
|
require.Empty(t, out)
|
|
|
|
})
|
|
|
|
}
|
2024-04-02 19:41:57 -05:00
|
|
|
|
2024-04-06 08:43:45 -05:00
|
|
|
func TestRunHookUpdate(t *testing.T) {
|
|
|
|
app := cli.NewApp()
|
|
|
|
app.Commands = []*cli.Command{subcmdHookUpdate}
|
|
|
|
|
|
|
|
t.Run("Removal of internal reference", func(t *testing.T) {
|
|
|
|
defer test.MockVariableValue(&cli.OsExiter, func(code int) {})()
|
|
|
|
defer test.MockVariableValue(&setting.IsProd, false)()
|
|
|
|
finish := captureOutput(t, os.Stderr)
|
|
|
|
|
|
|
|
err := app.Run([]string{"./forgejo", "update", "refs/pull/1/head", "0a51ae26bc73c47e2f754560c40904cf14ed51a9", "0000000000000000000000000000000000000000"})
|
|
|
|
out := finish()
|
|
|
|
assert.Error(t, err)
|
|
|
|
|
|
|
|
assert.Contains(t, out, "The deletion of refs/pull/1/head is skipped as it's an internal reference.")
|
|
|
|
})
|
|
|
|
|
|
|
|
t.Run("Update of internal reference", func(t *testing.T) {
|
|
|
|
err := app.Run([]string{"./forgejo", "update", "refs/pull/1/head", "0a51ae26bc73c47e2f754560c40904cf14ed51a9", "0000000000000000000000000000000000000001"})
|
|
|
|
assert.NoError(t, err)
|
|
|
|
})
|
|
|
|
|
|
|
|
t.Run("Removal of branch", func(t *testing.T) {
|
|
|
|
err := app.Run([]string{"./forgejo", "update", "refs/head/main", "0a51ae26bc73c47e2f754560c40904cf14ed51a9", "0000000000000000000000000000000000000000"})
|
|
|
|
assert.NoError(t, err)
|
|
|
|
})
|
|
|
|
|
|
|
|
t.Run("Not enough arguments", func(t *testing.T) {
|
|
|
|
err := app.Run([]string{"./forgejo", "update"})
|
|
|
|
assert.NoError(t, err)
|
|
|
|
})
|
|
|
|
}
|