Hunter0x7c7
2022-08-11 a82f9cb69f63aaeba40c024960deda7d75b9fece
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
// Package blackhole is an outbound handler that blocks all connections.
package blackhole
 
//go:generate go run github.com/v2fly/v2ray-core/v5/common/errors/errorgen
 
import (
    "context"
    "time"
 
    "github.com/v2fly/v2ray-core/v5/common"
    "github.com/v2fly/v2ray-core/v5/transport"
    "github.com/v2fly/v2ray-core/v5/transport/internet"
)
 
// Handler is an outbound connection that silently swallow the entire payload.
type Handler struct {
    response ResponseConfig
}
 
// New creates a new blackhole handler.
func New(ctx context.Context, config *Config) (*Handler, error) {
    response, err := config.GetInternalResponse()
    if err != nil {
        return nil, err
    }
    return &Handler{
        response: response,
    }, nil
}
 
// Process implements OutboundHandler.Dispatch().
func (h *Handler) Process(ctx context.Context, link *transport.Link, dialer internet.Dialer) error {
    nBytes := h.response.WriteTo(link.Writer)
    if nBytes > 0 {
        // Sleep a little here to make sure the response is sent to client.
        time.Sleep(time.Second)
    }
    common.Interrupt(link.Writer)
    return nil
}
 
func init() {
    common.Must(common.RegisterConfig((*Config)(nil), func(ctx context.Context, config interface{}) (interface{}, error) {
        return New(ctx, config.(*Config))
    }))
 
    common.Must(common.RegisterConfig((*SimplifiedConfig)(nil), func(ctx context.Context, config interface{}) (interface{}, error) {
        simplifiedServer := config.(*SimplifiedConfig)
        _ = simplifiedServer
        fullConfig := &Config{}
        return common.CreateObject(ctx, fullConfig)
    }))
}