prevent using net.Conn.WriteTo()

pull/787/head
Darien Raymond 2017-11-24 23:24:12 +01:00
parent 2fdccb2424
commit 74327ea8ae
No known key found for this signature in database
GPG Key ID: 7251FFA14BB18169
1 changed files with 5 additions and 1 deletions

View File

@ -86,8 +86,12 @@ func parseBasicAuth(auth string) (username, password string, ok bool) {
return cs[:s], cs[s+1:], true return cs[:s], cs[s+1:], true
} }
type readerOnly struct {
io.Reader
}
func (s *Server) Process(ctx context.Context, network net.Network, conn internet.Connection, dispatcher dispatcher.Interface) error { func (s *Server) Process(ctx context.Context, network net.Network, conn internet.Connection, dispatcher dispatcher.Interface) error {
reader := bufio.NewReaderSize(conn, buf.Size) reader := bufio.NewReaderSize(readerOnly{conn}, buf.Size)
Start: Start:
conn.SetReadDeadline(time.Now().Add(time.Second * 16)) conn.SetReadDeadline(time.Now().Add(time.Second * 16))