| 141 | } |
| 142 | |
| 143 | func readConn(ctx context.Context, session *Session, sessions chan<- *Session) { |
| 144 | for { |
| 145 | buff := make([]byte, BufferSize) |
| 146 | br, err := session.Conn.Read(buff) |
| 147 | |
| 148 | select { |
| 149 | case <-ctx.Done(): |
| 150 | stderrlog.Debugf("closing connection") |
| 151 | session.Close() |
| 152 | return |
| 153 | default: |
| 154 | session.Lock() |
| 155 | if err != nil { |
| 156 | if err != io.EOF { |
| 157 | stderrlog.Errorf("failed to read from conn: %v", err) |
| 158 | } |
| 159 | |
| 160 | // setting Open to false triggers SendData() to |
| 161 | // send ShouldClose |
| 162 | session.Open = false |
| 163 | } |
| 164 | |
| 165 | // write the data to the session buffer, if we have data |
| 166 | if br > 0 { |
| 167 | session.Buf.Write(buff[0:br]) |
| 168 | } |
| 169 | session.Unlock() |
| 170 | |
| 171 | sessions <- session |
| 172 | if !session.Open { |
| 173 | return |
| 174 | } |
| 175 | } |
| 176 | } |
| 177 | } |
| 178 | |
| 179 | // Ping returns empty |
| 180 | func (t *tunnelServer) Ping(context.Context, *remote.Empty) (*remote.Empty, error) { |