73 lines
1.3 KiB
Go
73 lines
1.3 KiB
Go
package recorder
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"io"
|
|
"sync"
|
|
"syscall"
|
|
"time"
|
|
)
|
|
|
|
type ttyRecordHeader struct {
|
|
Sec int32
|
|
USec int32
|
|
Len int32
|
|
}
|
|
|
|
func (h *ttyRecordHeader) Now() {
|
|
var (
|
|
now = time.Now()
|
|
tv = syscall.NsecToTimeval(now.UnixNano())
|
|
)
|
|
h.Sec = int32(tv.Sec)
|
|
h.USec = tv.Usec
|
|
}
|
|
|
|
func (h *ttyRecordHeader) WriteTo(w io.Writer) (int64, error) {
|
|
var b [6]byte
|
|
binary.LittleEndian.PutUint32(b[0:], uint32(h.Sec))
|
|
binary.LittleEndian.PutUint32(b[2:], uint32(h.USec))
|
|
binary.LittleEndian.PutUint32(b[4:], uint32(h.Len))
|
|
n, err := w.Write(b[:])
|
|
return int64(n), err
|
|
}
|
|
|
|
type ttyRecRecorder struct {
|
|
mu sync.Mutex
|
|
wc io.WriteCloser
|
|
header *ttyRecordHeader
|
|
}
|
|
|
|
func newTTYRecRecorder(wc io.WriteCloser) *ttyRecRecorder {
|
|
return &ttyRecRecorder{
|
|
wc: wc,
|
|
header: new(ttyRecordHeader),
|
|
}
|
|
}
|
|
|
|
func (r *ttyRecRecorder) Close() error {
|
|
return r.wc.Close()
|
|
}
|
|
|
|
func (w *ttyRecRecorder) Write(p []byte) (n int, err error) {
|
|
w.mu.Lock()
|
|
defer w.mu.Unlock()
|
|
|
|
w.header.Now()
|
|
if _, err = w.header.WriteTo(w.wc); err != nil {
|
|
return
|
|
}
|
|
if _, err = w.wc.Write(p); err != nil {
|
|
return
|
|
}
|
|
return len(p) + 9, nil
|
|
}
|
|
|
|
func (r *ttyRecRecorder) Reads() io.WriteCloser {
|
|
return r
|
|
}
|
|
|
|
func (r *ttyRecRecorder) Writes() io.WriteCloser {
|
|
return nopCloser{io.Discard}
|
|
}
|