1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-12-21 17:46:58 -05:00

fix usage of io.EOF

This commit is contained in:
Darien Raymond 2016-11-17 23:21:44 +01:00
parent 192def75a1
commit ad3f450bce
No known key found for this signature in database
GPG Key ID: 7251FFA14BB18169
6 changed files with 12 additions and 19 deletions

View File

@ -92,7 +92,7 @@ func (this *ProxyConnection) Read(b []byte) (int, error) {
func (this *ProxyConnection) Write(b []byte) (int, error) { func (this *ProxyConnection) Write(b []byte) (int, error) {
if this.closed { if this.closed {
return 0, io.EOF return 0, errors.New("Proxy|Outbound: Writing into closed connection.")
} }
return this.writer.Write(b) return this.writer.Write(b)
} }

View File

@ -27,7 +27,7 @@ func (this *BufferedWriter) ReadFrom(reader io.Reader) (int64, error) {
defer this.Unlock() defer this.Unlock()
if this.writer == nil { if this.writer == nil {
return 0, io.EOF return 0, io.ErrClosedPipe
} }
totalBytes := int64(0) totalBytes := int64(0)
@ -49,7 +49,7 @@ func (this *BufferedWriter) Write(b []byte) (int, error) {
defer this.Unlock() defer this.Unlock()
if this.writer == nil { if this.writer == nil {
return 0, io.EOF return 0, io.ErrClosedPipe
} }
if !this.cached { if !this.cached {
@ -67,7 +67,7 @@ func (this *BufferedWriter) Flush() error {
defer this.Unlock() defer this.Unlock()
if this.writer == nil { if this.writer == nil {
return io.EOF return io.ErrClosedPipe
} }
return this.FlushWithoutLock() return this.FlushWithoutLock()

View File

@ -20,7 +20,7 @@ func NewChainWriter(writer Writer) *ChainWriter {
func (this *ChainWriter) Write(payload []byte) (int, error) { func (this *ChainWriter) Write(payload []byte) (int, error) {
if this.writer == nil { if this.writer == nil {
return 0, io.EOF return 0, io.ErrClosedPipe
} }
size := len(payload) size := len(payload)
@ -30,7 +30,7 @@ func (this *ChainWriter) Write(payload []byte) (int, error) {
this.Lock() this.Lock()
defer this.Unlock() defer this.Unlock()
if this.writer == nil { if this.writer == nil {
return 0, io.EOF return 0, io.ErrClosedPipe
} }
err := this.writer.Write(buffer) err := this.writer.Write(buffer)

View File

@ -102,6 +102,7 @@ func (this *VMessOutboundHandler) handleRequest(session *encoding.ClientSession,
} }
if !payload.IsEmpty() { if !payload.IsEmpty() {
if err := streamWriter.Write(payload); err != nil { if err := streamWriter.Write(payload); err != nil {
log.Info("VMess|Outbound: Failed to write payload. Disabling connection reuse.")
conn.SetReusable(false) conn.SetReusable(false)
} }
} }

View File

@ -30,9 +30,7 @@ func (ws *wsconn) Read(b []byte) (n int, err error) {
} }
func (ws *wsconn) read(b []byte) (n int, err error) { func (ws *wsconn) read(b []byte) (n int, err error) {
if ws.connClosing { if ws.connClosing {
return 0, io.EOF return 0, io.EOF
} }
@ -79,9 +77,8 @@ func (ws *wsconn) readNext(b []byte) (n int, err error) {
func (ws *wsconn) Write(b []byte) (n int, err error) { func (ws *wsconn) Write(b []byte) (n int, err error) {
ws.wlock.Lock() ws.wlock.Lock()
if ws.connClosing { if ws.connClosing {
return 0, io.EOF return 0, io.ErrClosedPipe
} }
n, err = ws.write(b) n, err = ws.write(b)

View File

@ -1,7 +1,6 @@
package ray package ray
import ( import (
"errors"
"io" "io"
"sync" "sync"
"time" "time"
@ -13,10 +12,6 @@ const (
bufferSize = 128 bufferSize = 128
) )
var (
ErrIOTimeout = errors.New("IO Timeout")
)
// NewRay creates a new Ray for direct traffic transport. // NewRay creates a new Ray for direct traffic transport.
func NewRay() Ray { func NewRay() Ray {
return &directRay{ return &directRay{
@ -79,24 +74,24 @@ func (this *Stream) Read() (*alloc.Buffer, error) {
func (this *Stream) Write(data *alloc.Buffer) error { func (this *Stream) Write(data *alloc.Buffer) error {
for !this.closed { for !this.closed {
err := this.TryWriteOnce(data) err := this.TryWriteOnce(data)
if err != ErrIOTimeout { if err != io.ErrNoProgress {
return err return err
} }
} }
return io.EOF return io.ErrClosedPipe
} }
func (this *Stream) TryWriteOnce(data *alloc.Buffer) error { func (this *Stream) TryWriteOnce(data *alloc.Buffer) error {
this.access.RLock() this.access.RLock()
defer this.access.RUnlock() defer this.access.RUnlock()
if this.closed { if this.closed {
return io.EOF return io.ErrClosedPipe
} }
select { select {
case this.buffer <- data: case this.buffer <- data:
return nil return nil
case <-time.After(2 * time.Second): case <-time.After(2 * time.Second):
return ErrIOTimeout return io.ErrNoProgress
} }
} }