moul/sshportal
1
1
Fork 0
mirror of https://github.com/moul/sshportal.git synced 2024-09-20 15:06:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

add fatal error when record file cannot be opened.

Browse source
This commit is contained in:
Manuel 2018-01-04 11:43:44 +01:00
parent 9db4b92d4e
commit 9e1c395810
1 changed files with 10 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
pkg/bastionsession/bastionsession.go
View file

@ -56,18 +56,16 @@ func pipe(lreqs, rreqs <-chan *gossh.Request, lch, rch gossh.Channel, logsLocati
file_name := strings.Join([]string{logsLocation, "/", user, "-", time.Now().Format(time.RFC3339)}, "") // get user file_name := strings.Join([]string{logsLocation, "/", user, "-", time.Now().Format(time.RFC3339)}, "") // get user
f, err := os.OpenFile(file_name, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0640) f, err := os.OpenFile(file_name, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0640)
if err != nil { if err != nil {
go func() { log.Fatalf("error: %v", err)
_, _ = io.Copy(lch, rch) }
errch <- errors.New("lch closed the connection")
}()
} else {
log.Printf("Session is recorded in %v", file_name) log.Printf("Session is recorded in %v", file_name)
wrappedlch := logchannel.New(lch, f) wrappedlch := logchannel.New(lch, f)
go func() { go func() {
_, _ = io.Copy(wrappedlch, rch) _, _ = io.Copy(wrappedlch, rch)
errch <- errors.New("lch closed the connection") errch <- errors.New("lch closed the connection")
}() }()
}
defer f.Close() defer f.Close()
go func() { go func() {