Compare commits
1
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
6b14fc847d |
@@ -1006,11 +1006,8 @@ func (e *Engine) deliverLog(
|
|||||||
"webhook event delivered to log target",
|
"webhook event delivered to log target",
|
||||||
"delivery_id", d.ID,
|
"delivery_id", d.ID,
|
||||||
"event_id", d.EventID,
|
"event_id", d.EventID,
|
||||||
"webhook_id", d.Event.WebhookID,
|
|
||||||
"entrypoint_id", d.Event.EntrypointID,
|
|
||||||
"target_id", d.TargetID,
|
"target_id", d.TargetID,
|
||||||
"target_name", d.Target.Name,
|
"target_name", d.Target.Name,
|
||||||
"outcome", database.DeliveryStatusDelivered,
|
|
||||||
"method", d.Event.Method,
|
"method", d.Event.Method,
|
||||||
"content_type", d.Event.ContentType,
|
"content_type", d.Event.ContentType,
|
||||||
"body_length", len(d.Event.Body),
|
"body_length", len(d.Event.Body),
|
||||||
|
|||||||
@@ -1,7 +1,6 @@
|
|||||||
package delivery_test
|
package delivery_test
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
|
||||||
"context"
|
"context"
|
||||||
"database/sql"
|
"database/sql"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
@@ -436,91 +435,6 @@ func TestDeliverLog_ImmediateSuccess(t *testing.T) {
|
|||||||
assert.True(t, result.Success)
|
assert.True(t, result.Success)
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestDeliverLog_StructuredLogFields(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
db := testWebhookDB(t)
|
|
||||||
|
|
||||||
var logBuf bytes.Buffer
|
|
||||||
|
|
||||||
e := delivery.NewTestEngine(
|
|
||||||
slog.New(slog.NewTextHandler(
|
|
||||||
&logBuf,
|
|
||||||
&slog.HandlerOptions{Level: slog.LevelDebug},
|
|
||||||
)),
|
|
||||||
&http.Client{Timeout: 5 * time.Second},
|
|
||||||
1,
|
|
||||||
)
|
|
||||||
|
|
||||||
event := seedEvent(t, db, `{"log":"structured"}`)
|
|
||||||
|
|
||||||
dlv := seedDelivery(
|
|
||||||
t, db, event.ID, uuid.New().String(),
|
|
||||||
database.DeliveryStatusPending,
|
|
||||||
)
|
|
||||||
|
|
||||||
d := &database.Delivery{
|
|
||||||
EventID: event.ID,
|
|
||||||
TargetID: dlv.TargetID,
|
|
||||||
Status: database.DeliveryStatusPending,
|
|
||||||
Event: event,
|
|
||||||
Target: database.Target{
|
|
||||||
Name: "structured-log",
|
|
||||||
Type: database.TargetTypeLog,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
d.ID = dlv.ID
|
|
||||||
|
|
||||||
e.ExportDeliverLog(db, d)
|
|
||||||
|
|
||||||
// The delivery is marked delivered and a success
|
|
||||||
// DeliveryResult with no HTTP status is recorded,
|
|
||||||
// mirroring the other target types' bookkeeping.
|
|
||||||
var updated database.Delivery
|
|
||||||
|
|
||||||
require.NoError(t, db.First(
|
|
||||||
&updated, "id = ?", dlv.ID,
|
|
||||||
).Error)
|
|
||||||
|
|
||||||
assert.Equal(t,
|
|
||||||
database.DeliveryStatusDelivered, updated.Status,
|
|
||||||
"log target should immediately succeed",
|
|
||||||
)
|
|
||||||
|
|
||||||
var result database.DeliveryResult
|
|
||||||
|
|
||||||
require.NoError(t, db.Where(
|
|
||||||
"delivery_id = ?", dlv.ID,
|
|
||||||
).First(&result).Error)
|
|
||||||
|
|
||||||
assert.True(t, result.Success)
|
|
||||||
assert.Equal(t, 0, result.StatusCode,
|
|
||||||
"log target should not have an HTTP status",
|
|
||||||
)
|
|
||||||
|
|
||||||
assertLogFields(t, logBuf.String(), event, "structured-log")
|
|
||||||
}
|
|
||||||
|
|
||||||
// assertLogFields checks that a log target's structured
|
|
||||||
// log line carries the required fields: event id,
|
|
||||||
// webhook/entrypoint, target name, and outcome.
|
|
||||||
func assertLogFields(
|
|
||||||
t *testing.T,
|
|
||||||
logged string,
|
|
||||||
event database.Event,
|
|
||||||
targetName string,
|
|
||||||
) {
|
|
||||||
t.Helper()
|
|
||||||
|
|
||||||
assert.Contains(t, logged, "event_id="+event.ID)
|
|
||||||
assert.Contains(t, logged, "webhook_id="+event.WebhookID)
|
|
||||||
assert.Contains(t,
|
|
||||||
logged, "entrypoint_id="+event.EntrypointID,
|
|
||||||
)
|
|
||||||
assert.Contains(t, logged, "target_name="+targetName)
|
|
||||||
assert.Contains(t, logged, "outcome=delivered")
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestDeliverHTTP_WithRetries_Success(t *testing.T) {
|
func TestDeliverHTTP_WithRetries_Success(t *testing.T) {
|
||||||
t.Parallel()
|
t.Parallel()
|
||||||
|
|
||||||
|
|||||||
@@ -12,13 +12,3 @@ func (s *Handlers) RenderTemplateForTest(
|
|||||||
) {
|
) {
|
||||||
s.renderTemplate(w, r, pageTemplate, data)
|
s.renderTemplate(w, r, pageTemplate, data)
|
||||||
}
|
}
|
||||||
|
|
||||||
// BuildSlackTargetConfigForTest exposes buildSlackTargetConfig
|
|
||||||
// for use in the handlers_test package.
|
|
||||||
func (s *Handlers) BuildSlackTargetConfigForTest(
|
|
||||||
w http.ResponseWriter,
|
|
||||||
r *http.Request,
|
|
||||||
targetURL string,
|
|
||||||
) (string, error) {
|
|
||||||
return s.buildSlackTargetConfig(w, r, targetURL)
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -116,52 +116,6 @@ func TestHandleIndex_Authenticated(t *testing.T) {
|
|||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestBuildSlackTargetConfig_AcceptsPublicURL(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
var h *handlers.Handlers
|
|
||||||
|
|
||||||
app := newTestApp(t, &h)
|
|
||||||
app.RequireStart()
|
|
||||||
|
|
||||||
t.Cleanup(app.RequireStop)
|
|
||||||
|
|
||||||
req := httptest.NewRequestWithContext(
|
|
||||||
context.Background(), http.MethodPost, "/", nil)
|
|
||||||
w := httptest.NewRecorder()
|
|
||||||
|
|
||||||
cfg, err := h.BuildSlackTargetConfigForTest(
|
|
||||||
w, req, "http://93.184.216.34/services/T00/B00/xxx",
|
|
||||||
)
|
|
||||||
|
|
||||||
require.NoError(t, err)
|
|
||||||
assert.Equal(t, http.StatusOK, w.Code)
|
|
||||||
assert.Contains(t, cfg, "webhookUrl")
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestBuildSlackTargetConfig_RejectsReservedURL(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
var h *handlers.Handlers
|
|
||||||
|
|
||||||
app := newTestApp(t, &h)
|
|
||||||
app.RequireStart()
|
|
||||||
|
|
||||||
t.Cleanup(app.RequireStop)
|
|
||||||
|
|
||||||
req := httptest.NewRequestWithContext(
|
|
||||||
context.Background(), http.MethodPost, "/", nil)
|
|
||||||
w := httptest.NewRecorder()
|
|
||||||
|
|
||||||
cfg, err := h.BuildSlackTargetConfigForTest(
|
|
||||||
w, req, "http://169.254.169.254/latest/meta-data/",
|
|
||||||
)
|
|
||||||
|
|
||||||
require.Error(t, err)
|
|
||||||
assert.Empty(t, cfg)
|
|
||||||
assert.Equal(t, http.StatusBadRequest, w.Code)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestRenderTemplate(t *testing.T) {
|
func TestRenderTemplate(t *testing.T) {
|
||||||
t.Parallel()
|
t.Parallel()
|
||||||
|
|
||||||
|
|||||||
@@ -17,13 +17,11 @@ func (h *Handlers) HandleProfile() http.HandlerFunc {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// Get session. RequireAuth middleware guarantees an
|
// Get session
|
||||||
// authenticated session before this handler runs, so we
|
|
||||||
// only need to guard against an unexpected retrieval error.
|
|
||||||
sess, err := h.session.Get(r)
|
sess, err := h.session.Get(r)
|
||||||
if err != nil {
|
if err != nil || !h.session.IsAuthenticated(sess) {
|
||||||
h.log.Error("failed to get session", "error", err)
|
// Redirect to login if not authenticated
|
||||||
http.Error(w, "Internal server error", http.StatusInternalServerError)
|
http.Redirect(w, r, "/pages/login", http.StatusSeeOther)
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,159 +0,0 @@
|
|||||||
package handlers_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"net/http"
|
|
||||||
"net/http/httptest"
|
|
||||||
"testing"
|
|
||||||
|
|
||||||
"github.com/go-chi/chi"
|
|
||||||
"github.com/stretchr/testify/assert"
|
|
||||||
"github.com/stretchr/testify/require"
|
|
||||||
"sneak.berlin/go/webhooker/internal/config"
|
|
||||||
"sneak.berlin/go/webhooker/internal/handlers"
|
|
||||||
"sneak.berlin/go/webhooker/internal/logger"
|
|
||||||
"sneak.berlin/go/webhooker/internal/middleware"
|
|
||||||
"sneak.berlin/go/webhooker/internal/session"
|
|
||||||
)
|
|
||||||
|
|
||||||
// authenticatedCookies creates an authenticated session for the given
|
|
||||||
// user and returns the resulting cookies for use on a later request.
|
|
||||||
func authenticatedCookies(
|
|
||||||
t *testing.T,
|
|
||||||
sess *session.Session,
|
|
||||||
userID, username string,
|
|
||||||
) []*http.Cookie {
|
|
||||||
t.Helper()
|
|
||||||
|
|
||||||
setupReq := httptest.NewRequestWithContext(
|
|
||||||
context.Background(), http.MethodGet, "/setup", nil,
|
|
||||||
)
|
|
||||||
setupW := httptest.NewRecorder()
|
|
||||||
|
|
||||||
s, err := sess.Get(setupReq)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
sess.SetUser(s, userID, username)
|
|
||||||
require.NoError(t, sess.Save(setupReq, setupW, s))
|
|
||||||
|
|
||||||
cookies := setupW.Result().Cookies()
|
|
||||||
require.NotEmpty(t, cookies, "session cookie should be set")
|
|
||||||
|
|
||||||
return cookies
|
|
||||||
}
|
|
||||||
|
|
||||||
// profileRequest builds a GET request for the given profile username,
|
|
||||||
// attaching the supplied cookies and the chi URL parameter that the
|
|
||||||
// handler reads via chi.URLParam.
|
|
||||||
func profileRequest(
|
|
||||||
username string,
|
|
||||||
cookies []*http.Cookie,
|
|
||||||
) *http.Request {
|
|
||||||
req := httptest.NewRequestWithContext(
|
|
||||||
context.Background(), http.MethodGet, "/user/"+username, nil,
|
|
||||||
)
|
|
||||||
|
|
||||||
for _, c := range cookies {
|
|
||||||
req.AddCookie(c)
|
|
||||||
}
|
|
||||||
|
|
||||||
rctx := chi.NewRouteContext()
|
|
||||||
rctx.URLParams.Add("username", username)
|
|
||||||
|
|
||||||
return req.WithContext(
|
|
||||||
context.WithValue(req.Context(), chi.RouteCtxKey, rctx),
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestHandleProfile_OwnProfile_OK(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
var h *handlers.Handlers
|
|
||||||
|
|
||||||
var sess *session.Session
|
|
||||||
|
|
||||||
app := newTestApp(t, &h, &sess)
|
|
||||||
app.RequireStart()
|
|
||||||
|
|
||||||
t.Cleanup(app.RequireStop)
|
|
||||||
|
|
||||||
cookies := authenticatedCookies(t, sess, "test-user-id", "testuser")
|
|
||||||
|
|
||||||
req := profileRequest("testuser", cookies)
|
|
||||||
w := httptest.NewRecorder()
|
|
||||||
|
|
||||||
h.HandleProfile().ServeHTTP(w, req)
|
|
||||||
|
|
||||||
assert.Equal(t, http.StatusOK, w.Code)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestHandleProfile_OtherProfile_Forbidden(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
var h *handlers.Handlers
|
|
||||||
|
|
||||||
var sess *session.Session
|
|
||||||
|
|
||||||
app := newTestApp(t, &h, &sess)
|
|
||||||
app.RequireStart()
|
|
||||||
|
|
||||||
t.Cleanup(app.RequireStop)
|
|
||||||
|
|
||||||
cookies := authenticatedCookies(t, sess, "test-user-id", "testuser")
|
|
||||||
|
|
||||||
req := profileRequest("otheruser", cookies)
|
|
||||||
w := httptest.NewRecorder()
|
|
||||||
|
|
||||||
h.HandleProfile().ServeHTTP(w, req)
|
|
||||||
|
|
||||||
assert.Equal(t, http.StatusForbidden, w.Code)
|
|
||||||
}
|
|
||||||
|
|
||||||
// TestUserRoute_Unauthenticated_RedirectedByMiddleware exercises the
|
|
||||||
// /user/{username} route group's middleware chain (CSRF then
|
|
||||||
// RequireAuth, matching setupUserRoutes) and proves that an
|
|
||||||
// unauthenticated request is redirected to /pages/login at the
|
|
||||||
// middleware layer, never reaching the endpoint handler.
|
|
||||||
func TestUserRoute_Unauthenticated_RedirectedByMiddleware(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
var log *logger.Logger
|
|
||||||
|
|
||||||
var cfg *config.Config
|
|
||||||
|
|
||||||
var sess *session.Session
|
|
||||||
|
|
||||||
app := newTestApp(t, &log, &cfg, &sess)
|
|
||||||
app.RequireStart()
|
|
||||||
|
|
||||||
t.Cleanup(app.RequireStop)
|
|
||||||
|
|
||||||
mw := middleware.NewForTest(log.Get(), cfg, sess)
|
|
||||||
|
|
||||||
var handlerReached bool
|
|
||||||
|
|
||||||
router := chi.NewRouter()
|
|
||||||
router.Route("/user/{username}", func(r chi.Router) {
|
|
||||||
r.Use(mw.CSRF())
|
|
||||||
r.Use(mw.RequireAuth())
|
|
||||||
r.Get("/", func(w http.ResponseWriter, _ *http.Request) {
|
|
||||||
handlerReached = true
|
|
||||||
|
|
||||||
w.WriteHeader(http.StatusOK)
|
|
||||||
})
|
|
||||||
})
|
|
||||||
|
|
||||||
req := httptest.NewRequestWithContext(
|
|
||||||
context.Background(), http.MethodGet, "/user/testuser", nil,
|
|
||||||
)
|
|
||||||
w := httptest.NewRecorder()
|
|
||||||
|
|
||||||
router.ServeHTTP(w, req)
|
|
||||||
|
|
||||||
assert.False(
|
|
||||||
t, handlerReached,
|
|
||||||
"handler must not be reached for unauthenticated request",
|
|
||||||
)
|
|
||||||
assert.Equal(t, http.StatusSeeOther, w.Code)
|
|
||||||
assert.Equal(t, "/pages/login", w.Header().Get("Location"))
|
|
||||||
}
|
|
||||||
@@ -902,7 +902,7 @@ func (h *Handlers) buildTargetConfig(
|
|||||||
case database.TargetTypeHTTP:
|
case database.TargetTypeHTTP:
|
||||||
return h.buildHTTPTargetConfig(w, r, targetURL)
|
return h.buildHTTPTargetConfig(w, r, targetURL)
|
||||||
case database.TargetTypeSlack:
|
case database.TargetTypeSlack:
|
||||||
return h.buildSlackTargetConfig(w, r, targetURL)
|
return h.buildSlackTargetConfig(w, targetURL)
|
||||||
case database.TargetTypeDatabase, database.TargetTypeLog:
|
case database.TargetTypeDatabase, database.TargetTypeLog:
|
||||||
return "", nil
|
return "", nil
|
||||||
default:
|
default:
|
||||||
@@ -967,7 +967,6 @@ func (h *Handlers) buildHTTPTargetConfig(
|
|||||||
// buildSlackTargetConfig builds config JSON for a Slack target.
|
// buildSlackTargetConfig builds config JSON for a Slack target.
|
||||||
func (h *Handlers) buildSlackTargetConfig(
|
func (h *Handlers) buildSlackTargetConfig(
|
||||||
w http.ResponseWriter,
|
w http.ResponseWriter,
|
||||||
r *http.Request,
|
|
||||||
targetURL string,
|
targetURL string,
|
||||||
) (string, error) {
|
) (string, error) {
|
||||||
if targetURL == "" {
|
if targetURL == "" {
|
||||||
@@ -980,24 +979,6 @@ func (h *Handlers) buildSlackTargetConfig(
|
|||||||
return "", errMissingURL
|
return "", errMissingURL
|
||||||
}
|
}
|
||||||
|
|
||||||
err := delivery.ValidateTargetURL(
|
|
||||||
r.Context(), targetURL,
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
h.log.Warn(
|
|
||||||
"target URL blocked by SSRF protection",
|
|
||||||
"url", targetURL,
|
|
||||||
"error", err,
|
|
||||||
)
|
|
||||||
http.Error(
|
|
||||||
w,
|
|
||||||
"Invalid target URL: "+err.Error(),
|
|
||||||
http.StatusBadRequest,
|
|
||||||
)
|
|
||||||
|
|
||||||
return "", err
|
|
||||||
}
|
|
||||||
|
|
||||||
cfg := map[string]any{"webhookUrl": targetURL}
|
cfg := map[string]any{"webhookUrl": targetURL}
|
||||||
|
|
||||||
configBytes, err := json.Marshal(cfg)
|
configBytes, err := json.Marshal(cfg)
|
||||||
|
|||||||
@@ -13,11 +13,8 @@ const (
|
|||||||
httpReadTimeout = 10 * time.Second
|
httpReadTimeout = 10 * time.Second
|
||||||
|
|
||||||
// httpWriteTimeout is the maximum duration before timing out
|
// httpWriteTimeout is the maximum duration before timing out
|
||||||
// writes of the response. It must stay above the router's
|
// writes of the response.
|
||||||
// requestTimeout (60s, in routes.go) so the middleware timeout
|
httpWriteTimeout = 10 * time.Second
|
||||||
// fires first and returns a clean 503, rather than the transport
|
|
||||||
// cutting the connection at the socket write deadline.
|
|
||||||
httpWriteTimeout = 65 * time.Second
|
|
||||||
|
|
||||||
// httpMaxHeaderBytes is the maximum number of bytes the
|
// httpMaxHeaderBytes is the maximum number of bytes the
|
||||||
// server will read parsing the request headers.
|
// server will read parsing the request headers.
|
||||||
|
|||||||
@@ -106,7 +106,6 @@ func (s *Server) setupPageRoutes() {
|
|||||||
func (s *Server) setupUserRoutes() {
|
func (s *Server) setupUserRoutes() {
|
||||||
s.router.Route("/user/{username}", func(r chi.Router) {
|
s.router.Route("/user/{username}", func(r chi.Router) {
|
||||||
r.Use(s.mw.CSRF())
|
r.Use(s.mw.CSRF())
|
||||||
r.Use(s.mw.RequireAuth())
|
|
||||||
r.Get("/", s.h.HandleProfile())
|
r.Get("/", s.h.HandleProfile())
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user