mirror of
https://github.com/cbeuw/Cloak.git
synced 2024-10-21 19:09:30 +00:00
105 lines
2.7 KiB
Go
105 lines
2.7 KiB
Go
|
/*
|
||
|
Copyright (c) 2009 The Go Authors. All rights reserved.
|
||
|
|
||
|
Redistribution and use in source and binary forms, with or without
|
||
|
modification, are permitted provided that the following conditions are
|
||
|
met:
|
||
|
|
||
|
* Redistributions of source code must retain the above copyright
|
||
|
notice, this list of conditions and the following disclaimer.
|
||
|
* Redistributions in binary form must reproduce the above
|
||
|
copyright notice, this list of conditions and the following disclaimer
|
||
|
in the documentation and/or other materials provided with the
|
||
|
distribution.
|
||
|
* Neither the name of Google Inc. nor the names of its
|
||
|
contributors may be used to endorse or promote products derived from
|
||
|
this software without specific prior written permission.
|
||
|
|
||
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||
|
"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||
|
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||
|
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
||
|
OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||
|
SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||
|
LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||
|
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||
|
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||
|
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||
|
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||
|
*/
|
||
|
/*
|
||
|
Forked from https://golang.org/src/io/io.go
|
||
|
*/
|
||
|
package common
|
||
|
|
||
|
import (
|
||
|
"io"
|
||
|
"net"
|
||
|
"time"
|
||
|
)
|
||
|
|
||
|
// copyBuffer is the actual implementation of Copy and CopyBuffer.
|
||
|
// if buf is nil, one is allocated.
|
||
|
func Copy(dst net.Conn, src net.Conn, srcReadTimeout time.Duration) (written int64, err error) {
|
||
|
/*
|
||
|
// If the reader has a WriteTo method, use it to do the copy.
|
||
|
// Avoids an allocation and a copy.
|
||
|
if wt, ok := src.(WriterTo); ok {
|
||
|
return wt.WriteTo(dst)
|
||
|
}
|
||
|
// Similarly, if the writer has a ReadFrom method, use it to do the copy.
|
||
|
if rt, ok := dst.(ReaderFrom); ok {
|
||
|
return rt.ReadFrom(src)
|
||
|
}
|
||
|
|
||
|
*/
|
||
|
//if buf == nil {
|
||
|
size := 32 * 1024
|
||
|
/*
|
||
|
if l, ok := src.(*LimitedReader); ok && int64(size) > l.N {
|
||
|
if l.N < 1 {
|
||
|
size = 1
|
||
|
} else {
|
||
|
size = int(l.N)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
*/
|
||
|
buf := make([]byte, size)
|
||
|
//}
|
||
|
for {
|
||
|
if srcReadTimeout != 0 {
|
||
|
src.SetReadDeadline(time.Now().Add(srcReadTimeout))
|
||
|
/*
|
||
|
err =
|
||
|
if err != nil {
|
||
|
break
|
||
|
}
|
||
|
|
||
|
*/
|
||
|
}
|
||
|
nr, er := src.Read(buf)
|
||
|
if nr > 0 {
|
||
|
var offset int
|
||
|
for offset < nr {
|
||
|
nw, ew := dst.Write(buf[offset:nr])
|
||
|
if nw > 0 {
|
||
|
written += int64(nw)
|
||
|
}
|
||
|
if ew != nil {
|
||
|
err = ew
|
||
|
break
|
||
|
}
|
||
|
offset += nw
|
||
|
}
|
||
|
}
|
||
|
if er != nil {
|
||
|
if er != io.EOF {
|
||
|
err = er
|
||
|
}
|
||
|
break
|
||
|
}
|
||
|
}
|
||
|
return written, err
|
||
|
}
|