Skip to content

Commit 386730d

Browse files
committed
order imports
Signed-off-by: Eliott Bouhana <[email protected]>
1 parent af120e7 commit 386730d

File tree

6 files changed

+21
-11
lines changed

6 files changed

+21
-11
lines changed

Diff for: appsec/events/block.go

+9
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,12 @@
77
// It allows finer-grained integrations of appsec into your Go errors' management logic.
88
package events
99

10+
import "errors"
11+
1012
var _ error = (*BlockingSecurityEvent)(nil)
1113

14+
var securityError = &BlockingSecurityEvent{}
15+
1216
// BlockingSecurityEvent is the error type returned by function calls blocked by appsec.
1317
// Even though appsec takes care of responding automatically to the blocked requests, it
1418
// is your duty to abort the request handlers that are calling functions blocked by appsec.
@@ -22,3 +26,8 @@ type BlockingSecurityEvent struct{}
2226
func (*BlockingSecurityEvent) Error() string {
2327
return "request blocked by WAF"
2428
}
29+
30+
// IsSecurityError returns true if the error is a security event.
31+
func IsSecurityError(err error) bool {
32+
return errors.Is(err, securityError)
33+
}

Diff for: contrib/google.golang.org/grpc/appsec.go

+1
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ package grpc
77

88
import (
99
"context"
10+
1011
"gopkg.in/DataDog/dd-trace-go.v1/ddtrace"
1112
"gopkg.in/DataDog/dd-trace-go.v1/internal/appsec/dyngo"
1213
"gopkg.in/DataDog/dd-trace-go.v1/internal/appsec/emitter/grpcsec"

Diff for: contrib/labstack/echo.v4/appsec.go

+3-2
Original file line numberDiff line numberDiff line change
@@ -6,12 +6,13 @@
66
package echo
77

88
import (
9-
"gopkg.in/DataDog/dd-trace-go.v1/appsec/events"
109
"net/http"
1110

12-
"github.com/labstack/echo/v4"
11+
"gopkg.in/DataDog/dd-trace-go.v1/appsec/events"
1312
"gopkg.in/DataDog/dd-trace-go.v1/ddtrace/tracer"
1413
"gopkg.in/DataDog/dd-trace-go.v1/internal/appsec/emitter/httpsec"
14+
15+
"github.com/labstack/echo/v4"
1516
)
1617

1718
func withAppSec(next echo.HandlerFunc, span tracer.Span) echo.HandlerFunc {

Diff for: contrib/net/http/roundtripper.go

+2-5
Original file line numberDiff line numberDiff line change
@@ -6,14 +6,13 @@
66
package http
77

88
import (
9-
"errors"
109
"fmt"
10+
"gopkg.in/DataDog/dd-trace-go.v1/appsec/events"
1111
"math"
1212
"net/http"
1313
"os"
1414
"strconv"
1515

16-
"gopkg.in/DataDog/dd-trace-go.v1/appsec/events"
1716
"gopkg.in/DataDog/dd-trace-go.v1/ddtrace"
1817
"gopkg.in/DataDog/dd-trace-go.v1/ddtrace/ext"
1918
"gopkg.in/DataDog/dd-trace-go.v1/ddtrace/tracer"
@@ -26,8 +25,6 @@ type roundTripper struct {
2625
cfg *roundTripperConfig
2726
}
2827

29-
var securityError = &events.BlockingSecurityEvent{}
30-
3128
func (rt *roundTripper) RoundTrip(req *http.Request) (res *http.Response, err error) {
3229
if rt.cfg.ignoreRequest(req) {
3330
return rt.base.RoundTrip(req)
@@ -63,7 +60,7 @@ func (rt *roundTripper) RoundTrip(req *http.Request) (res *http.Response, err er
6360
if rt.cfg.after != nil {
6461
rt.cfg.after(res, span)
6562
}
66-
if !errors.Is(err, securityError) && (rt.cfg.errCheck == nil || rt.cfg.errCheck(err)) {
63+
if !events.IsSecurityError(err) && (rt.cfg.errCheck == nil || rt.cfg.errCheck(err)) {
6764
span.Finish(tracer.WithError(err))
6865
} else {
6966
span.Finish()

Diff for: internal/appsec/listener/sharedsec/shared.go

+5-3
Original file line numberDiff line numberDiff line change
@@ -8,14 +8,16 @@ package sharedsec
88
import (
99
"encoding/json"
1010
"errors"
11-
"github.com/DataDog/appsec-internal-go/limiter"
12-
waf "github.com/DataDog/go-libddwaf/v3"
13-
wafErrors "github.com/DataDog/go-libddwaf/v3/errors"
11+
1412
"gopkg.in/DataDog/dd-trace-go.v1/appsec/events"
1513
"gopkg.in/DataDog/dd-trace-go.v1/internal/appsec/dyngo"
1614
"gopkg.in/DataDog/dd-trace-go.v1/internal/appsec/emitter/sharedsec"
1715
"gopkg.in/DataDog/dd-trace-go.v1/internal/appsec/trace"
1816
"gopkg.in/DataDog/dd-trace-go.v1/internal/log"
17+
18+
"github.com/DataDog/appsec-internal-go/limiter"
19+
waf "github.com/DataDog/go-libddwaf/v3"
20+
wafErrors "github.com/DataDog/go-libddwaf/v3/errors"
1921
)
2022

2123
const (

Diff for: internal/stacktrace/event_test.go

+1-1
Original file line numberDiff line numberDiff line change
@@ -6,14 +6,14 @@
66
package stacktrace
77

88
import (
9-
"github.com/tinylib/msgp/msgp"
109
"testing"
1110

1211
"gopkg.in/DataDog/dd-trace-go.v1/ddtrace/mocktracer"
1312
ddtracer "gopkg.in/DataDog/dd-trace-go.v1/ddtrace/tracer"
1413
"gopkg.in/DataDog/dd-trace-go.v1/internal"
1514

1615
"github.com/stretchr/testify/require"
16+
"github.com/tinylib/msgp/msgp"
1717
)
1818

1919
func TestNewEvent(t *testing.T) {

0 commit comments

Comments
 (0)