mirror of https://github.com/cbeuw/Cloak
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
98 lines
1.9 KiB
Go
98 lines
1.9 KiB
Go
// This is base on https://github.com/golang/go/blob/0436b162397018c45068b47ca1b5924a3eafdee0/src/net/net_fake.go#L173
|
|
|
|
package multiplex
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"io"
|
|
"sync"
|
|
"sync/atomic"
|
|
"time"
|
|
)
|
|
|
|
const BUF_SIZE_LIMIT = 1 << 20 * 500
|
|
|
|
var ErrTimeout = errors.New("deadline exceeded")
|
|
|
|
// The point of a bufferedPipe is that Read() will block until data is available
|
|
type bufferedPipe struct {
|
|
// only alloc when on first Read or Write
|
|
buf *bytes.Buffer
|
|
|
|
closed uint32
|
|
rwCond *sync.Cond
|
|
rDeadline time.Time
|
|
}
|
|
|
|
func NewBufferedPipe() *bufferedPipe {
|
|
p := &bufferedPipe{
|
|
rwCond: sync.NewCond(&sync.Mutex{}),
|
|
}
|
|
return p
|
|
}
|
|
|
|
func (p *bufferedPipe) Read(target []byte) (int, error) {
|
|
p.rwCond.L.Lock()
|
|
defer p.rwCond.L.Unlock()
|
|
if p.buf == nil {
|
|
p.buf = new(bytes.Buffer)
|
|
}
|
|
for {
|
|
if atomic.LoadUint32(&p.closed) == 1 && p.buf.Len() == 0 {
|
|
return 0, io.EOF
|
|
}
|
|
if !p.rDeadline.IsZero() {
|
|
d := time.Until(p.rDeadline)
|
|
if d <= 0 {
|
|
return 0, ErrTimeout
|
|
}
|
|
time.AfterFunc(d, p.rwCond.Broadcast)
|
|
}
|
|
if p.buf.Len() > 0 {
|
|
break
|
|
}
|
|
p.rwCond.Wait()
|
|
}
|
|
n, err := p.buf.Read(target)
|
|
// err will always be nil because we have already verified that buf.Len() != 0
|
|
p.rwCond.Broadcast()
|
|
return n, err
|
|
}
|
|
|
|
func (p *bufferedPipe) Write(input []byte) (int, error) {
|
|
p.rwCond.L.Lock()
|
|
defer p.rwCond.L.Unlock()
|
|
if p.buf == nil {
|
|
p.buf = new(bytes.Buffer)
|
|
}
|
|
for {
|
|
if atomic.LoadUint32(&p.closed) == 1 {
|
|
return 0, io.ErrClosedPipe
|
|
}
|
|
if p.buf.Len() <= BUF_SIZE_LIMIT {
|
|
// if p.buf gets too large, write() will panic. We don't want this to happen
|
|
break
|
|
}
|
|
p.rwCond.Wait()
|
|
}
|
|
n, err := p.buf.Write(input)
|
|
// err will always be nil
|
|
p.rwCond.Broadcast()
|
|
return n, err
|
|
}
|
|
|
|
func (p *bufferedPipe) Close() error {
|
|
atomic.StoreUint32(&p.closed, 1)
|
|
p.rwCond.Broadcast()
|
|
return nil
|
|
}
|
|
|
|
func (p *bufferedPipe) SetReadDeadline(t time.Time) {
|
|
p.rwCond.L.Lock()
|
|
defer p.rwCond.L.Unlock()
|
|
|
|
p.rDeadline = t
|
|
p.rwCond.Broadcast()
|
|
}
|