diff options
Diffstat (limited to 'pkg/server/hook')
| -rw-r--r-- | pkg/server/hook/dump.go (renamed from pkg/server/hook/tee.go) | 36 |
1 files changed, 18 insertions, 18 deletions
diff --git a/pkg/server/hook/tee.go b/pkg/server/hook/dump.go index 34b2eca..d6bcd90 100644 --- a/pkg/server/hook/tee.go +++ b/pkg/server/hook/dump.go @@ -13,19 +13,19 @@ import ( "tunnel/pkg/server/queue" ) -const teeDefaultFile = "/tmp/tunnel/dump" +const dumpDefaultFile = "/tmp/tunnel/dump" -type tee struct { +type dump struct { f *os.File - h *teeHook + h *dumpHook } -type teeHook struct { +type dumpHook struct { file string time bool } -func (t *tee) dump(s string, p []byte) error { +func (t *dump) write(s string, p []byte) error { var out bytes.Buffer if t.h.time { @@ -46,41 +46,41 @@ func (t *tee) dump(s string, p []byte) error { return nil } -func (t *tee) Send(rq, wq queue.Q) error { +func (t *dump) Send(rq, wq queue.Q) error { for b := range rq { - t.dump(">", b) + t.write(">", b) wq <- b } return nil } -func (t *tee) Recv(rq, wq queue.Q) error { +func (t *dump) Recv(rq, wq queue.Q) error { for b := range rq { - t.dump("<", b) + t.write("<", b) wq <- b } return nil } -func (t *tee) Close() { +func (t *dump) Close() { t.f.Close() } -func (h *teeHook) where(env env.Env) string { +func (h *dumpHook) where(env env.Env) string { if h.file != "" { return h.file } - if v := env.GetLocal("tee.file"); v != "" { + if v := env.GetLocal("dump.file"); v != "" { return v } - return teeDefaultFile + return dumpDefaultFile } -func (h *teeHook) Open(env env.Env) (interface{}, error) { +func (h *dumpHook) Open(env env.Env) (interface{}, error) { file := h.where(env) dir := path.Dir(file) @@ -91,7 +91,7 @@ func (h *teeHook) Open(env env.Env) (interface{}, error) { tid, sid := env.Get("tunnel"), env.Get("stream") name := fmt.Sprintf("%s.%s.%s", file, tid, sid) - t := &tee{h: h} + t := &dump{h: h} if f, err := os.Create(name); err != nil { return nil, err @@ -102,8 +102,8 @@ func (h *teeHook) Open(env env.Env) (interface{}, error) { return t, nil } -func newTeeHook(opts opts.Opts) (hook, error) { - h := &teeHook{ +func newDumpHook(opts opts.Opts) (hook, error) { + h := &dumpHook{ file: opts["file"], time: opts.Bool("time"), } @@ -111,5 +111,5 @@ func newTeeHook(opts opts.Opts) (hook, error) { } func init() { - register("tee", newTeeHook) + register("dump", newDumpHook) } |
