mirror of https://github.com/XTLS/Xray-core
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
47 lines
1.1 KiB
47 lines
1.1 KiB
package blackhole |
|
|
|
import ( |
|
"github.com/xtls/xray-core/common" |
|
"github.com/xtls/xray-core/common/buf" |
|
) |
|
|
|
const ( |
|
http403response = `HTTP/1.1 403 Forbidden |
|
Connection: close |
|
Cache-Control: max-age=3600, public |
|
Content-Length: 0 |
|
|
|
|
|
` |
|
) |
|
|
|
// ResponseConfig is the configuration for blackhole responses. |
|
type ResponseConfig interface { |
|
// WriteTo writes predefined response to the give buffer. |
|
WriteTo(buf.Writer) int32 |
|
} |
|
|
|
// WriteTo implements ResponseConfig.WriteTo(). |
|
func (*NoneResponse) WriteTo(buf.Writer) int32 { return 0 } |
|
|
|
// WriteTo implements ResponseConfig.WriteTo(). |
|
func (*HTTPResponse) WriteTo(writer buf.Writer) int32 { |
|
b := buf.New() |
|
common.Must2(b.WriteString(http403response)) |
|
n := b.Len() |
|
writer.WriteMultiBuffer(buf.MultiBuffer{b}) |
|
return n |
|
} |
|
|
|
// GetInternalResponse converts response settings from proto to internal data structure. |
|
func (c *Config) GetInternalResponse() (ResponseConfig, error) { |
|
if c.GetResponse() == nil { |
|
return new(NoneResponse), nil |
|
} |
|
|
|
config, err := c.GetResponse().GetInstance() |
|
if err != nil { |
|
return nil, err |
|
} |
|
return config.(ResponseConfig), nil |
|
}
|
|
|