root
/
sftp
mirror of https://github.com/pkg/sftp.git
1
0
Fork 0
Code Issues Actions Packages Projects Releases Wiki Activity

request-server to use new packet ordering code 

The 2 servers now use all the same packet managing code to ensure
ordered processing and reply packets.
This commit is contained in:
John Eikenberry 2017-04-23 14:27:25 -07:00
parent 5024cb048c
commit ced55b1bf0
1 changed files with 18 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
request-server.go
View File

@ -28,7 +28,6 @@ type Handlers struct {
type RequestServer struct {
serverConn
Handlers Handlers
pktChan chan requestPacket
pktMgr packetManager
openRequests map[string]Request
openRequestLock sync.RWMutex
@ -47,7 +46,6 @@ func NewRequestServer(rwc io.ReadWriteCloser, h Handlers) *RequestServer {
return &RequestServer{
serverConn: svrConn,
Handlers: h,
pktChan: make(chan requestPacket, sftpServerWorkerCount),
pktMgr: newPktMgr(&svrConn),
openRequests: make(map[string]Request),
}
@ -84,15 +82,15 @@ func (rs *RequestServer) Close() error { return rs.conn.Close() }
// Serve requests for user session
func (rs *RequestServer) Serve() error {
var wg sync.WaitGroup
wg.Add(sftpServerWorkerCount)
for i := 0; i < sftpServerWorkerCount; i++ {
go func() {
wg.Add(1)
workerFunc := func(ch requestChan) {
wg.Add(1)
defer wg.Done()
if err := rs.packetWorker(); err != nil {
if err := rs.packetWorker(ch); err != nil {
rs.conn.Close() // shuts down recvPacket
}
}()
}
pktChan := rs.pktMgr.workerChan(workerFunc)
var err error
var pkt requestPacket
@ -103,24 +101,26 @@ func (rs *RequestServer) Serve() error {
if err != nil {
break
}
pkt, err = makePacket(rxPacket{fxp(pktType), pktBytes})
if err != nil {
debug("makePacket err: %v", err)
rs.conn.Close() // shuts down recvPacket
break
}
rs.pktMgr.incomingPacket(pkt)
rs.pktChan <- pkt
}
close(rs.pktChan) // shuts down sftpServerWorkers
pktChan <- pkt
}
wg.Done()
close(pktChan) // shuts down sftpServerWorkers
wg.Wait() // wait for all workers to exit
rs.pktMgr.close() // shuts down packetManager
return err
}
func (rs *RequestServer) packetWorker() error {
for pkt := range rs.pktChan {
func (rs *RequestServer) packetWorker(pktChan chan requestPacket) error {
for pkt := range pktChan {
var rpkt responsePacket
switch pkt := pkt.(type) {
case *sshFxInitPacket: