summaryrefslogtreecommitdiff
path: root/pkg
diff options
context:
space:
mode:
authorMikhail Osipov <mike.osipov@gmail.com>2020-05-27 17:29:59 +0300
committerMikhail Osipov <mike.osipov@gmail.com>2020-05-27 17:29:59 +0300
commite00995cf2a5968ae94d4dc0abfe21a66dae5cd54 (patch)
treeefc7c389818b9713d92dde5f231a71fde0185699 /pkg
parenta29c4614431c0f7b705ee793f0a98348cdd29595 (diff)
rename hook tee to dump
Diffstat (limited to 'pkg')
-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)
}