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:
parent
192def75a1
commit
ad3f450bce
@ -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)
|
||||||
}
|
}
|
||||||
|
@ -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()
|
||||||
|
@ -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)
|
||||||
|
@ -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)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -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)
|
||||||
|
@ -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
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user