/
proc.go
72 lines (61 loc) · 1.36 KB
/
proc.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
package main
import (
"bytes"
"log"
"strings"
"github.com/lunixbochs/vtclean"
)
type proc struct {
Pid int
Login []string
Shell bool
Top bool
Logger *log.Logger
buf bytes.Buffer
lastStdin [][]byte
}
func (p *proc) LoginData(data []byte) {
p.Login = append(p.Login, string(data))
}
func (p *proc) Stdin(data []byte) {
p.lastStdin = append(p.lastStdin, bytes.Trim(data, "\r\n"))
}
func (p *proc) Stdout(data []byte) {
if len(p.lastStdin) > 1 {
for _, v := range p.lastStdin {
p.buf.Write(v)
}
}
p.lastStdin = nil
p.buf.Write(data)
if bytes.Contains(p.buf.Bytes(), []byte("\r\n")) {
split := bytes.Split(p.buf.Bytes(), []byte("\r\n"))
for _, v := range split[:len(split)-1] {
p.logline(v)
}
p.buf.Reset()
p.buf.Write(split[len(split)-1])
}
}
func (p *proc) Flush() {
line := p.buf.Bytes()
if len(line) > 0 {
p.logline(p.buf.Bytes())
}
}
func (p *proc) Exit() {
p.Flush()
p.Logger.Printf("[%d]-<logout>", p.Pid)
}
func (p *proc) LogLogin(login []string, success bool) {
if success {
p.Logger.Printf("[%d]-<login as: %s>", p.Pid, strings.Join(login, ", "))
} else {
p.Logger.Printf("[%d]-<login attempt: %s>", p.Pid, strings.Join(login, ", "))
}
}
func (p *proc) logline(line []byte) {
sline := vtclean.Clean(string(line), true)
sline = strings.Replace(sline, "\n", "\\n", -1)
p.Logger.Printf("[%d] %s", p.Pid, sline)
}