hook and socket args check fix, tests

1 files changed, 5 insertions, 14 deletions

diff --git a/pkg/server/hook/zip.go b/pkg/server/hook/zip.go
index bde4957..615b50d 100644
--- a/pkg/server/hook/zip.go
+++ b/pkg/server/hook/zip.go
@@ -3,14 +3,13 @@ package hook
 import (
 	"compress/flate"
 	"io"
-	"tunnel/pkg/server/env"
-	"tunnel/pkg/server/opts"
+
 	"tunnel/pkg/server/queue"
 )
 
-type zipHook struct{}
+type zipPipe struct{}
 
-func (zipHook) Send(rq, wq queue.Q) error {
+func (zipPipe) Send(rq, wq queue.Q) error {
 	w, err := flate.NewWriter(wq.Writer(), flate.BestCompression)
 	if err != nil {
 		return err
@@ -28,7 +27,7 @@ func (zipHook) Send(rq, wq queue.Q) error {
 	return w.Close()
 }
 
-func (zipHook) Recv(rq, wq queue.Q) error {
+func (zipPipe) Recv(rq, wq queue.Q) error {
 	r := flate.NewReader(rq.Reader())
 
 	// FIXME: not received ending due to ultimate conn.Close
@@ -42,14 +41,6 @@ func (zipHook) Recv(rq, wq queue.Q) error {
 	return r.Close()
 }
 
-func (h zipHook) Open(env.Env) (interface{}, error) {
-	return h, nil
-}
-
-func newZipHook(opts.Opts) (hook, error) {
-	return zipHook{}, nil
-}
-
 func init() {
-	register("zip", newZipHook)
+	registerPipe("zip", zipPipe{})
 }