Commit 21b69085 authored by Mitchell Hashimoto's avatar Mitchell Hashimoto

fmt

parent c74b3758
...@@ -416,9 +416,9 @@ func (m *MuxConn) loop() { ...@@ -416,9 +416,9 @@ func (m *MuxConn) loop() {
stream.mu.Unlock() stream.mu.Unlock()
} }
} }
} }
func (m *MuxConn) write(from muxPacketFrom, id uint32, dataType muxPacketType, p []byte) (int, error) { func (m *MuxConn) write(from muxPacketFrom, id uint32, dataType muxPacketType, p []byte) (int, error) {
m.wlock.Lock() m.wlock.Lock()
defer m.wlock.Unlock() defer m.wlock.Unlock()
...@@ -438,11 +438,11 @@ func (m *MuxConn) loop() { ...@@ -438,11 +438,11 @@ func (m *MuxConn) loop() {
return 0, nil return 0, nil
} }
return m.rwc.Write(p) return m.rwc.Write(p)
} }
// Stream is a single stream of data and implements io.ReadWriteCloser. // Stream is a single stream of data and implements io.ReadWriteCloser.
// A Stream is full-duplex so you can write data as well as read data. // A Stream is full-duplex so you can write data as well as read data.
type Stream struct { type Stream struct {
from muxPacketFrom from muxPacketFrom
id uint32 id uint32
mux *MuxConn mux *MuxConn
...@@ -452,11 +452,11 @@ func (m *MuxConn) loop() { ...@@ -452,11 +452,11 @@ func (m *MuxConn) loop() {
stateUpdated time.Time stateUpdated time.Time
mu sync.Mutex mu sync.Mutex
writeCh chan<- []byte writeCh chan<- []byte
} }
type streamState byte type streamState byte
const ( const (
streamStateClosed streamState = iota streamStateClosed streamState = iota
streamStateListen streamStateListen
streamStateSynRecv streamStateSynRecv
...@@ -467,9 +467,9 @@ func (m *MuxConn) loop() { ...@@ -467,9 +467,9 @@ func (m *MuxConn) loop() {
streamStateCloseWait streamStateCloseWait
streamStateClosing streamStateClosing
streamStateLastAck streamStateLastAck
) )
func newStream(from muxPacketFrom, id uint32, m *MuxConn) *Stream { func newStream(from muxPacketFrom, id uint32, m *MuxConn) *Stream {
// Create the stream object and channel where data will be sent to // Create the stream object and channel where data will be sent to
dataR, dataW := io.Pipe() dataR, dataW := io.Pipe()
writeCh := make(chan []byte, 4096) writeCh := make(chan []byte, 4096)
...@@ -505,9 +505,9 @@ func (m *MuxConn) loop() { ...@@ -505,9 +505,9 @@ func (m *MuxConn) loop() {
}() }()
return stream return stream
} }
func (s *Stream) Close() error { func (s *Stream) Close() error {
s.mu.Lock() s.mu.Lock()
defer s.mu.Unlock() defer s.mu.Unlock()
...@@ -523,13 +523,13 @@ func (m *MuxConn) loop() { ...@@ -523,13 +523,13 @@ func (m *MuxConn) loop() {
s.write(muxPacketFin, nil) s.write(muxPacketFin, nil)
return nil return nil
} }
func (s *Stream) Read(p []byte) (int, error) { func (s *Stream) Read(p []byte) (int, error) {
return s.reader.Read(p) return s.reader.Read(p)
} }
func (s *Stream) Write(p []byte) (int, error) { func (s *Stream) Write(p []byte) (int, error) {
s.mu.Lock() s.mu.Lock()
state := s.state state := s.state
s.mu.Unlock() s.mu.Unlock()
...@@ -539,13 +539,13 @@ func (m *MuxConn) loop() { ...@@ -539,13 +539,13 @@ func (m *MuxConn) loop() {
} }
return s.write(muxPacketData, p) return s.write(muxPacketData, p)
} }
func (s *Stream) closeWriter() { func (s *Stream) closeWriter() {
s.writeCh <- nil s.writeCh <- nil
} }
func (s *Stream) setState(state streamState) { func (s *Stream) setState(state streamState) {
//log.Printf("[TRACE] %p: Stream %d (%s) went to state %d", s.mux, s.id, s.from, state) //log.Printf("[TRACE] %p: Stream %d (%s) went to state %d", s.mux, s.id, s.from, state)
s.state = state s.state = state
s.stateUpdated = time.Now().UTC() s.stateUpdated = time.Now().UTC()
...@@ -555,9 +555,9 @@ func (m *MuxConn) loop() { ...@@ -555,9 +555,9 @@ func (m *MuxConn) loop() {
default: default:
} }
} }
} }
func (s *Stream) waitState(target streamState) error { func (s *Stream) waitState(target streamState) error {
// Register a state change listener to wait for changes // Register a state change listener to wait for changes
stateCh := make(chan streamState, 10) stateCh := make(chan streamState, 10)
s.stateChange[stateCh] = struct{}{} s.stateChange[stateCh] = struct{}{}
...@@ -574,8 +574,8 @@ func (m *MuxConn) loop() { ...@@ -574,8 +574,8 @@ func (m *MuxConn) loop() {
} else { } else {
return fmt.Errorf("Stream %d went to bad state: %d", s.id, state) return fmt.Errorf("Stream %d went to bad state: %d", s.id, state)
} }
} }
func (s *Stream) write(dataType muxPacketType, p []byte) (int, error) { func (s *Stream) write(dataType muxPacketType, p []byte) (int, error) {
return s.mux.write(s.from, s.id, dataType, p) return s.mux.write(s.from, s.id, dataType, p)
} }
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment