1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-06-10 18:00:43 +00:00
v2fly/proxy/blackhole/blackhole.go

48 lines
1.1 KiB
Go
Raw Normal View History

2016-12-22 12:33:28 +00:00
// Package blackhole is an outbound handler that blocks all connections.
2015-10-28 16:41:14 +00:00
package blackhole
2018-09-30 21:08:41 +00:00
//go:generate errorgen
2017-04-08 23:43:25 +00:00
2015-10-28 16:41:14 +00:00
import (
"context"
2017-01-10 14:10:12 +00:00
"time"
"v2ray.com/core/common"
2018-11-03 11:36:29 +00:00
"v2ray.com/core/transport"
2018-10-22 20:12:50 +00:00
"v2ray.com/core/transport/internet"
2018-04-16 22:31:10 +00:00
"v2ray.com/core/transport/pipe"
2015-10-28 16:41:14 +00:00
)
2017-02-10 15:50:45 +00:00
// Handler is an outbound connection that silently swallow the entire payload.
2016-12-21 20:43:00 +00:00
type Handler struct {
2016-09-22 10:01:36 +00:00
response ResponseConfig
2015-10-28 16:41:14 +00:00
}
2016-12-21 20:43:00 +00:00
// New creates a new blackhole handler.
func New(ctx context.Context, config *Config) (*Handler, error) {
2016-10-16 12:22:21 +00:00
response, err := config.GetInternalResponse()
2016-09-22 10:01:36 +00:00
if err != nil {
return nil, err
}
2016-12-21 20:43:00 +00:00
return &Handler{
2016-09-22 10:01:36 +00:00
response: response,
}, nil
2015-10-28 16:41:14 +00:00
}
2017-02-10 23:41:14 +00:00
// Process implements OutboundHandler.Dispatch().
2018-11-03 11:36:29 +00:00
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)
}
2018-04-16 22:31:10 +00:00
pipe.CloseError(link.Writer)
2017-02-10 10:41:50 +00:00
return nil
2015-10-28 16:41:14 +00:00
}
func init() {
common.Must(common.RegisterConfig((*Config)(nil), func(ctx context.Context, config interface{}) (interface{}, error) {
return New(ctx, config.(*Config))
}))
2016-06-14 20:54:08 +00:00
}