mirror of
				https://gitee.com/gitea/gitea
				synced 2025-11-04 16:40:24 +08:00 
			
		
		
		
	Link mentioned user in markdown only if they are visible to viewer (#21554)
We need to make sure a user can't confirm the existence of a user with private visibility * Follow up on #21533 ### Before #### User  #### Admin  ### After #### User  #### Admin  Signed-off-by: Yarden Shoham <hrsi88@gmail.com> Co-authored-by: wxiaoguang <wxiaoguang@gmail.com> Co-authored-by: Lunny Xiao <xiaolunwen@gmail.com>
This commit is contained in:
		@@ -8,22 +8,26 @@ import (
 | 
			
		||||
	"context"
 | 
			
		||||
 | 
			
		||||
	"code.gitea.io/gitea/models/user"
 | 
			
		||||
	"code.gitea.io/gitea/modules/log"
 | 
			
		||||
	gitea_context "code.gitea.io/gitea/modules/context"
 | 
			
		||||
	"code.gitea.io/gitea/modules/markup"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func ProcessorHelper() *markup.ProcessorHelper {
 | 
			
		||||
	return &markup.ProcessorHelper{
 | 
			
		||||
		IsUsernameMentionable: func(ctx context.Context, username string) bool {
 | 
			
		||||
			// TODO: cast ctx to modules/context.Context and use IsUserVisibleToViewer
 | 
			
		||||
 | 
			
		||||
			// Only link if the user actually exists
 | 
			
		||||
			userExists, err := user.IsUserExist(ctx, 0, username)
 | 
			
		||||
			mentionedUser, err := user.GetUserByName(ctx, username)
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				log.Error("Failed to validate user in mention %q exists, assuming it does", username)
 | 
			
		||||
				userExists = true
 | 
			
		||||
				return false
 | 
			
		||||
			}
 | 
			
		||||
			return userExists
 | 
			
		||||
 | 
			
		||||
			giteaCtx, ok := ctx.(*gitea_context.Context)
 | 
			
		||||
			if !ok {
 | 
			
		||||
				// when using general context, use user's visibility to check
 | 
			
		||||
				return mentionedUser.Visibility.IsPublic()
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			// when using gitea context (web context), use user's visibility and user's permission to check
 | 
			
		||||
			return user.IsUserVisibleToViewer(giteaCtx, mentionedUser, giteaCtx.Doer)
 | 
			
		||||
		},
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -6,15 +6,48 @@ package markup
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"testing"
 | 
			
		||||
 | 
			
		||||
	"code.gitea.io/gitea/models/db"
 | 
			
		||||
	"code.gitea.io/gitea/models/unittest"
 | 
			
		||||
	"code.gitea.io/gitea/models/user"
 | 
			
		||||
	gitea_context "code.gitea.io/gitea/modules/context"
 | 
			
		||||
 | 
			
		||||
	"github.com/stretchr/testify/assert"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func TestProcessorHelper(t *testing.T) {
 | 
			
		||||
	assert.NoError(t, unittest.PrepareTestDatabase())
 | 
			
		||||
	assert.True(t, ProcessorHelper().IsUsernameMentionable(context.Background(), "user10"))
 | 
			
		||||
	assert.False(t, ProcessorHelper().IsUsernameMentionable(context.Background(), "no-such-user"))
 | 
			
		||||
 | 
			
		||||
	userPublic := "user1"
 | 
			
		||||
	userPrivate := "user31"
 | 
			
		||||
	userLimited := "user33"
 | 
			
		||||
	userNoSuch := "no-such-user"
 | 
			
		||||
 | 
			
		||||
	unittest.AssertCount(t, &user.User{Name: userPublic}, 1)
 | 
			
		||||
	unittest.AssertCount(t, &user.User{Name: userPrivate}, 1)
 | 
			
		||||
	unittest.AssertCount(t, &user.User{Name: userLimited}, 1)
 | 
			
		||||
	unittest.AssertCount(t, &user.User{Name: userNoSuch}, 0)
 | 
			
		||||
 | 
			
		||||
	// when using general context, use user's visibility to check
 | 
			
		||||
	assert.True(t, ProcessorHelper().IsUsernameMentionable(context.Background(), userPublic))
 | 
			
		||||
	assert.False(t, ProcessorHelper().IsUsernameMentionable(context.Background(), userLimited))
 | 
			
		||||
	assert.False(t, ProcessorHelper().IsUsernameMentionable(context.Background(), userPrivate))
 | 
			
		||||
	assert.False(t, ProcessorHelper().IsUsernameMentionable(context.Background(), userNoSuch))
 | 
			
		||||
 | 
			
		||||
	// when using web context, use user.IsUserVisibleToViewer to check
 | 
			
		||||
	var err error
 | 
			
		||||
	giteaCtx := &gitea_context.Context{}
 | 
			
		||||
	giteaCtx.Req, err = http.NewRequest("GET", "/", nil)
 | 
			
		||||
	assert.NoError(t, err)
 | 
			
		||||
 | 
			
		||||
	giteaCtx.Doer = nil
 | 
			
		||||
	assert.True(t, ProcessorHelper().IsUsernameMentionable(giteaCtx, userPublic))
 | 
			
		||||
	assert.False(t, ProcessorHelper().IsUsernameMentionable(giteaCtx, userPrivate))
 | 
			
		||||
 | 
			
		||||
	giteaCtx.Doer, err = user.GetUserByName(db.DefaultContext, userPrivate)
 | 
			
		||||
	assert.NoError(t, err)
 | 
			
		||||
	assert.True(t, ProcessorHelper().IsUsernameMentionable(giteaCtx, userPublic))
 | 
			
		||||
	assert.True(t, ProcessorHelper().IsUsernameMentionable(giteaCtx, userPrivate))
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user