lzhr/v2fly
1
0
Fork 0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-06-27 09:55:22 +00:00
Code Issues Releases Wiki Activity

refactor

Browse Source
This commit is contained in:
Darien Raymond 2017-04-02 13:43:24 +02:00
parent 73c4696e00
commit 3161c56aba
No known key found for this signature in database
GPG Key ID: 7251FFA14BB18169
4 changed files with 13 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
app/proxyman/mux/mux.go
View File

@ -113,7 +113,7 @@ func (m *Client) Closed() bool {
func (m *Client) fetchInput(ctx context.Context, s *session) {
dest, _ := proxy.TargetFromContext(ctx)
writer := &MuxWriter{
writer := &Writer{
dest: dest,
id: s.id,
writer: m.inboundRay.InboundInput(),

2
app/proxyman/mux/mux_test.go
View File

@ -20,7 +20,7 @@ func TestReaderWriter(t *testing.T) {
payload := buf.New()
payload.AppendBytes('a', 'b', 'c', 'd')
writer.Write(payload)
assert.Error(writer.Write(payload)).IsNil()
writer.Close()

10
app/proxyman/mux/reader.go
View File

@ -7,20 +7,20 @@ import (
"v2ray.com/core/common/serial"
)
type MuxReader struct {
type Reader struct {
reader io.Reader
remainingLength int
buffer *buf.Buffer
}
func NewReader(reader buf.Reader) *MuxReader {
return &MuxReader{
func NewReader(reader buf.Reader) *Reader {
return &Reader{
reader: buf.ToBytesReader(reader),
buffer: buf.NewLocal(1024),
}
}
func (r *MuxReader) ReadMetadata() (*FrameMetadata, error) {
func (r *Reader) ReadMetadata() (*FrameMetadata, error) {
b := r.buffer
b.Clear()
@ -35,7 +35,7 @@ func (r *MuxReader) ReadMetadata() (*FrameMetadata, error) {
return ReadFrameFrom(b.Bytes())
}
func (r *MuxReader) Read() (*buf.Buffer, bool, error) {
func (r *Reader) Read() (*buf.Buffer, bool, error) {
b := buf.New()
var dataLen int
if r.remainingLength > 0 {

12
app/proxyman/mux/writer.go
View File

@ -6,22 +6,22 @@ import (
"v2ray.com/core/common/serial"
)
type MuxWriter struct {
type Writer struct {
id uint16
dest net.Destination
writer buf.Writer
followup bool
}
func NewWriter(id uint16, dest net.Destination, writer buf.Writer) *MuxWriter {
return &MuxWriter{
func NewWriter(id uint16, dest net.Destination, writer buf.Writer) *Writer {
return &Writer{
id: id,
dest: dest,
writer: writer,
}
}
func (w *MuxWriter) writeInternal(b *buf.Buffer) error {
func (w *Writer) writeInternal(b *buf.Buffer) error {
meta := FrameMetadata{
SessionID: w.id,
Target: w.dest,
@ -57,7 +57,7 @@ func (w *MuxWriter) writeInternal(b *buf.Buffer) error {
return w.writer.Write(frame)
}
func (w *MuxWriter) Write(b *buf.Buffer) error {
func (w *Writer) Write(b *buf.Buffer) error {
defer b.Release()
if err := w.writeInternal(b); err != nil {
@ -71,7 +71,7 @@ func (w *MuxWriter) Write(b *buf.Buffer) error {
return nil
}
func (w *MuxWriter) Close() {
func (w *Writer) Close() {
meta := FrameMetadata{
SessionID: w.id,
Target: w.dest,