mirror of https://github.com/v2ray/v2ray-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.
31 lines
565 B
31 lines
565 B
9 years ago
|
package net
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
emptyTime time.Time
|
||
|
|
)
|
||
|
|
|
||
|
|
type TimeOutReader struct {
|
||
|
|
timeout int
|
||
|
|
connection net.Conn
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewTimeOutReader(timeout int, connection net.Conn) *TimeOutReader {
|
||
|
|
return &TimeOutReader{
|
||
|
|
timeout: timeout,
|
||
|
|
connection: connection,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (reader *TimeOutReader) Read(p []byte) (n int, err error) {
|
||
|
|
deadline := time.Duration(reader.timeout) * time.Second
|
||
|
|
reader.connection.SetReadDeadline(time.Now().Add(deadline))
|
||
|
|
n, err = reader.connection.Read(p)
|
||
|
|
reader.connection.SetReadDeadline(emptyTime)
|
||
|
|
return
|
||
|
|
}
|