| 110 | } |
| 111 | |
| 112 | func ReadFromSession(session *tunnel.Session, sessionsOut chan<- *tunnel.Session, log logpkg.Logger) { |
| 113 | log.Debugf("started reading conn %s", session.ID) |
| 114 | defer log.Debugf("finished reading conn %s", session.ID) |
| 115 | |
| 116 | conn := session.Conn |
| 117 | buff := make([]byte, tunnel.BufferSize) |
| 118 | |
| 119 | loop: |
| 120 | for { |
| 121 | br, err := conn.Read(buff) |
| 122 | select { |
| 123 | case <-session.Context.Done(): |
| 124 | return |
| 125 | default: |
| 126 | if err != nil { |
| 127 | if err != io.EOF { |
| 128 | log.Errorf("%s: failed reading from socket, exiting: %v", session.ID.String(), err) |
| 129 | } else { |
| 130 | log.Debugf("read EOF from conn") |
| 131 | } |
| 132 | session.Open = false |
| 133 | sessionsOut <- session |
| 134 | break loop |
| 135 | } |
| 136 | |
| 137 | log.Debugf("read %d bytes from conn", br) |
| 138 | if br > 0 { |
| 139 | session.Lock() |
| 140 | _, err = session.Buf.Write(buff[0:br]) |
| 141 | session.Unlock() |
| 142 | log.Debugf("wrote %d bytes to session", br) |
| 143 | } |
| 144 | if err != nil { |
| 145 | log.Errorf("%s: failed writing to session buffer: %v", session.ID, err) |
| 146 | break loop |
| 147 | } |
| 148 | |
| 149 | sessionsOut <- session |
| 150 | } |
| 151 | } |
| 152 | } |
| 153 | |
| 154 | func SendData(stream remote.Tunnel_InitTunnelClient, sessions <-chan *tunnel.Session, closeChan <-chan bool, log logpkg.Logger) error { |
| 155 | for { |