root
/
telebit
mirror of https://github.com/therootcompany/telebit.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

debug cleanup

This commit is contained in:
AJ ONeal 2020-07-21 00:35:45 -06:00
parent b57a61831e
commit 2b5ddb0407
10 changed files with 106 additions and 56 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
connwrap.go
View File

@ -4,6 +4,7 @@ import (
"bufio" "bufio"
"fmt" "fmt"
"net" "net"
"os"
"time" "time"
"git.rootprojects.org/root/telebit/dbg" "git.rootprojects.org/root/telebit/dbg"
@ -139,7 +140,7 @@ func (c *ConnWrap) isEncrypted() bool {
n := 6 n := 6
b, err := c.Peek(n) b, err := c.Peek(n)
if dbg.Debug { if dbg.Debug {
fmt.Printf("[debug] [wrap] Peek(%d): %s %v\n", n, string(b), err) fmt.Fprintf(os.Stderr, "[debug] [wrap] Peek(%d): %s %v\n", n, string(b), err)
} }
if nil != err { if nil != err {
// TODO return error on error? // TODO return error on error?

10
decoder.go
View File

@ -3,7 +3,7 @@ package telebit
import ( import (
"fmt" "fmt"
"io" "io"
"log" "os"
"git.rootprojects.org/root/telebit/dbg" "git.rootprojects.org/root/telebit/dbg"
) )
@ -35,13 +35,13 @@ func (d *Decoder) Decode(out Router) error {
b := make([]byte, d.bufferSize) b := make([]byte, d.bufferSize)
n, err := d.in.Read(b) n, err := d.in.Read(b)
if dbg.Debug { if dbg.Debug {
log.Println("[debug] [decoder] [srv] Tunnel read", n, dbg.Trunc(b, n)) fmt.Fprintf(os.Stderr, "[debug] [decoder] [srv] Tunnel read %d %s\n", n, dbg.Trunc(b, n))
} }
if n > 0 { if n > 0 {
rx <- b[:n] rx <- b[:n]
} }
if nil != err { if nil != err {
fmt.Println("[debug] [decoder] [srv] Tunnel read err", err) fmt.Fprintf(os.Stderr, "[decoder] [srv] Tunnel read err: %s\n", err)
rxErr <- err rxErr <- err
return return
} }
@ -53,11 +53,11 @@ func (d *Decoder) Decode(out Router) error {
case b := <-rx: case b := <-rx:
n, err := p.Write(b) n, err := p.Write(b)
if dbg.Debug { if dbg.Debug {
fmt.Println("[debug] [decoder] [srv] Tunnel write", n, len(b), dbg.Trunc(b, len(b))) fmt.Fprintf(os.Stderr, "[debug] [decoder] [srv] Tunnel write %d %d %s\n", n, len(b), dbg.Trunc(b, len(b)))
} }
// TODO BUG: handle when 'n' bytes written is less than len(b) // TODO BUG: handle when 'n' bytes written is less than len(b)
if nil != err { if nil != err {
fmt.Println("[debug] [decoder] [srv] Tunnel write error") fmt.Fprintf(os.Stderr, "[decoder] [srv] Tunnel write err: %s\n", err)
// an error to write represents an unrecoverable error, // an error to write represents an unrecoverable error,
// not just a downstream client error // not just a downstream client error
//d.in.Close() //d.in.Close()

27
encoder.go
View File

@ -5,6 +5,7 @@ import (
"errors" "errors"
"fmt" "fmt"
"io" "io"
"os"
"strings" "strings"
"sync" "sync"
@ -68,7 +69,7 @@ func (enc *Encoder) Encode(rin io.Reader, src, dst Addr) error {
rx := make(chan []byte) rx := make(chan []byte)
rxErr := make(chan error) rxErr := make(chan error)
//fmt.Println("[debug] what's the source to encode?", src) //fmt.Fprintf(os.Stderr, "[debug] what's the source to encode? %v\n", src)
go func() { go func() {
for { for {
@ -76,13 +77,15 @@ func (enc *Encoder) Encode(rin io.Reader, src, dst Addr) error {
//fmt.Println("loopers gonna loop") //fmt.Println("loopers gonna loop")
n, err := rin.Read(b) n, err := rin.Read(b)
if dbg.Debug { if dbg.Debug {
fmt.Println("[debug] [encoder] [srv] Browser read", n, dbg.Trunc(b, n)) fmt.Fprintf(os.Stderr, "[debug] [encoder] [srv] Browser read %d %s\n", n, dbg.Trunc(b, n))
} }
if n > 0 { if n > 0 {
rx <- b[:n] rx <- b[:n]
} }
if nil != err { if nil != err {
fmt.Println("[debug] [encoder] [srv] Browser read error", err) if io.EOF != err {
fmt.Fprintf(os.Stderr, "[encoder] [srv] Browser read err: %s\n", err)
}
rxErr <- err rxErr <- err
return return
} }
@ -99,32 +102,32 @@ func (enc *Encoder) Encode(rin io.Reader, src, dst Addr) error {
// TODO: verify that closing the reader will cause the goroutine to be released // TODO: verify that closing the reader will cause the goroutine to be released
//rin.Close() //rin.Close()
if dbg.Debug { if dbg.Debug {
fmt.Println("[debug] [encoder] [srv] Browser ctx.Done()") fmt.Fprintf(os.Stderr, "[debug] [encoder] [srv] Browser ctx.Done()\n")
} }
return errors.New("cancelled by encoder read or parent context") return errors.New("cancelled by encoder read or parent context")
/* /*
case <-enc.subctx.Done(): case <-enc.subctx.Done():
//rin.Close() //rin.Close()
fmt.Println("[debug] [encoder] [srv] Browser subctx.Done()") fmt.Fprintf(os.Stderr, "[debug] [encoder] [srv] Browser subctx.Done()\n")
return errors.New("cancelled by encoder write context") return errors.New("cancelled by encoder write context")
*/ */
case b := <-rx: case b := <-rx:
header, _, err := Encode(b, src, Addr{scheme: src.scheme, addr: dst.Hostname(), port: dst.Port()}) header, _, err := Encode(b, src, Addr{scheme: src.scheme, addr: dst.Hostname(), port: dst.Port()})
if nil != err { if nil != err {
//rin.Close() //rin.Close()
fmt.Println("[debug] [encoder] [srv] Browser Encode err", err) fmt.Fprintf(os.Stderr, "[encoder] [srv] Browser Encode err: %s\n", err)
return err return err
} }
//fmt.Println("[debug] encode header:", string(header)) //fmt.Fprintf(os.Stderr, "[debug] encode header: %s\n", string(header))
//fmt.Println("[debug] encode payload:", string(b)) //fmt.Fprintf(os.Stderr, "[debug] encode payload: %s\n", string(b))
_, err = enc.write(header, b) _, err = enc.write(header, b)
if dbg.Debug { if dbg.Debug {
fmt.Println("[debug] [encoder] [srv] Browser-to-tun write", len(header), strings.TrimSpace(string(header))) fmt.Fprintf(os.Stderr, "[debug] [encoder] [srv] Browser-to-tun write %d %s\n", len(header), strings.TrimSpace(string(header)))
fmt.Println("[debug] [encoder] [srv]", len(b), dbg.Trunc(b, len(b))) fmt.Fprintf(os.Stderr, "[debug] [encoder] [srv] %d %s\n", len(b), dbg.Trunc(b, len(b)))
} }
if nil != err { if nil != err {
fmt.Println("[debug] [encoder] [srv] Browser-to-tun write err", err) fmt.Fprintf(os.Stderr, "[encoder] [srv] Browser-to-tun write err: %s\n", err)
//rin.Close() //rin.Close()
return err return err
} }
@ -133,7 +136,7 @@ func (enc *Encoder) Encode(rin io.Reader, src, dst Addr) error {
//rin.Close() //rin.Close()
if io.EOF == err { if io.EOF == err {
header, _, _ := Encode(nil, src, Addr{scheme: "end"}) header, _, _ := Encode(nil, src, Addr{scheme: "end"})
//fmt.Println("[debug] encode end: ", string(header)) //fmt.Fprintf(os.Stderr, "[debug] encode end: %s\n", string(header))
// ignore err, which may have already closed // ignore err, which may have already closed
_, _ = enc.write(header, nil) _, _ = enc.write(header, nil)
return nil return nil

13
listener.go
View File

@ -6,7 +6,10 @@ import (
"io" "io"
"net" "net"
"net/http" "net/http"
"os"
"strings" "strings"
"git.rootprojects.org/root/telebit/dbg"
) )
// A Listener transforms a multiplexed websocket connection into individual net.Conn-like connections. // A Listener transforms a multiplexed websocket connection into individual net.Conn-like connections.
@ -83,7 +86,7 @@ func Serve(listener net.Listener, mux Handler) error {
if io.EOF != err && io.ErrClosedPipe != err && !strings.Contains(err.Error(), errNetClosing) { if io.EOF != err && io.ErrClosedPipe != err && !strings.Contains(err.Error(), errNetClosing) {
fmt.Printf("client could not be served: %q\n", err.Error()) fmt.Printf("client could not be served: %q\n", err.Error())
} }
fmt.Println("[debug] closing original client", err) fmt.Printf("closing original client: %s\n", err)
client.Close() client.Close()
} }
}() }()
@ -142,7 +145,9 @@ func (l *Listener) RouteBytes(srcAddr, dstAddr Addr, b []byte) {
} }
// EOF, if needed // EOF, if needed
if "end" == string(dst.scheme) { if "end" == string(dst.scheme) {
fmt.Println("[debug] end") if dbg.Debug {
fmt.Fprintf(os.Stderr, "[debug] end\n")
}
pipe.Close() pipe.Close()
delete(l.conns, src.String()) delete(l.conns, src.String())
} }
@ -179,9 +184,9 @@ func (l *Listener) getPipe(src, dst *Addr, count int) net.Conn {
newconn.Close() newconn.Close()
pipe.Close() pipe.Close()
if nil != err { if nil != err {
fmt.Printf("[debug] [ln-pipe] encode stream ended:\n%+v\n%+v\n%q\n", *src, *dst, err) fmt.Fprintf(os.Stderr, "[debug] [ln-pipe] encode stream ended:\n%+v\n%+v\n%q\n", *src, *dst, err)
} else { } else {
fmt.Printf("[debug] [ln-pipe] encode stream ended gracefully:\n%+v\n%+v\n", *src, *dst) fmt.Fprintf(os.Stderr, "[debug] [ln-pipe] encode stream ended gracefully:\n%+v\n%+v\n", *src, *dst)
} }
}() }()

5
mgmt/auth.go
View File

@ -5,10 +5,11 @@ import (
"encoding/json" "encoding/json"
"fmt" "fmt"
"io/ioutil" "io/ioutil"
"os"
telebit "git.rootprojects.org/root/telebit"
"git.rootprojects.org/root/telebit/dbg" "git.rootprojects.org/root/telebit/dbg"
"git.rootprojects.org/root/telebit/mgmt/authstore" "git.rootprojects.org/root/telebit/mgmt/authstore"
telebit "git.rootprojects.org/root/telebit"
) )
type SuccessResponse struct { type SuccessResponse struct {
@ -40,7 +41,7 @@ func Register(authURL, secret, ppid string) (kid string, err error) {
jsonb := bytes.NewBuffer([]byte(jsons)) jsonb := bytes.NewBuffer([]byte(jsons))
fullURL := authURL + "/register-device/" + secret fullURL := authURL + "/register-device/" + secret
if dbg.Debug { if dbg.Debug {
fmt.Println("[debug] authURL, secret, ppid", fullURL, secret, jsons) fmt.Fprintf(os.Stderr, "[debug] authURL=%s, secret=%s, ppid=%s\n", fullURL, secret, jsons)
} }
msg, err := telebit.Request("POST", fullURL, "", jsonb) msg, err := telebit.Request("POST", fullURL, "", jsonb)
if nil != err { if nil != err {

5
parser.go
View File

@ -3,6 +3,7 @@ package telebit
import ( import (
"errors" "errors"
"fmt" "fmt"
"os"
"git.rootprojects.org/root/telebit/dbg" "git.rootprojects.org/root/telebit/dbg"
) )
@ -72,7 +73,7 @@ func (p *Parser) Write(b []byte) (int, error) {
switch p.parseState { switch p.parseState {
case VersionState: case VersionState:
if dbg.Debug { if dbg.Debug {
fmt.Println("[debug] MPLEXY version byte", b[0], string(b)) fmt.Fprintf(os.Stderr, "[debug] MPLEXY version byte %v %s", b[0], string(b))
} }
p.state.version = b[0] p.state.version = b[0]
b = b[1:] b = b[1:]
@ -85,7 +86,7 @@ func (p *Parser) Write(b []byte) (int, error) {
switch p.state.version { switch p.state.version {
case V1: case V1:
if dbg.Debug { if dbg.Debug {
fmt.Println("[debug] MPLEXY packet is of type v1") fmt.Fprintf(os.Stderr, "[debug] MPLEXY packet is of type v1\n")
} }
return p.unpackV1(b) return p.unpackV1(b)
default: default:

17
routemux.go
View File

@ -4,9 +4,12 @@ import (
"errors" "errors"
"fmt" "fmt"
"net" "net"
"os"
"strconv" "strconv"
"strings" "strings"
"time" "time"
"git.rootprojects.org/root/telebit/dbg"
) )
// A RouteMux is a net.Conn multiplexer. // A RouteMux is a net.Conn multiplexer.
@ -38,7 +41,9 @@ func NewRouteMux() *RouteMux {
// Serve dispatches the connection to the handler whose selectors matches the attributes. // Serve dispatches the connection to the handler whose selectors matches the attributes.
func (m *RouteMux) Serve(client net.Conn) error { func (m *RouteMux) Serve(client net.Conn) error {
fmt.Println("\n\n[debug] mux.Serve(client)") if dbg.Debug {
fmt.Fprintf(os.Stderr, "\n\n[debug] mux.Serve(client)\n")
}
var wconn *ConnWrap var wconn *ConnWrap
switch conn := client.(type) { switch conn := client.(type) {
@ -85,7 +90,7 @@ func (m *RouteMux) Serve(client net.Conn) error {
} }
fmt.Println("\nMeta:", meta.comment, "meta.addr="+meta.addr, "servername="+servername) fmt.Println("\nMeta:", meta.comment, "meta.addr="+meta.addr, "servername="+servername)
if servername == meta.addr || "*" == meta.addr || port == meta.addr { if servername == meta.addr || "*" == meta.addr || port == meta.addr {
//fmt.Println("[debug] test of route:", meta) //fmt.Fprintf(os.Stderr, "[debug] test of route: %v\n", meta)
// Only keep trying handlers if ErrNotHandled was returned // Only keep trying handlers if ErrNotHandled was returned
if err := meta.handler.Serve(wconn); ErrNotHandled != err { if err := meta.handler.Serve(wconn); ErrNotHandled != err {
return err return err
@ -159,12 +164,16 @@ func (m *RouteMux) HandleTLS(servername string, acme *ACME, next Handler, commen
} }
if !wconn.isEncrypted() { if !wconn.isEncrypted() {
fmt.Println("[debug] HandleTLS: conn is not encrypted") if dbg.Debug {
fmt.Fprintf(os.Stderr, "[debug] HandleTLS: conn is not encrypted\n")
}
// TODO handle underlying Peek() timeout error // TODO handle underlying Peek() timeout error
return ErrNotHandled return ErrNotHandled
} }
fmt.Println("[debug] HandleTLS: decrypted connection, recursing") if dbg.Debug {
fmt.Fprintf(os.Stderr, "[debug] HandleTLS: decrypted connection, recursing\n")
}
//NewTerminator(acme, handler)(client) //NewTerminator(acme, handler)(client)
//return handler.Serve(client) //return handler.Serve(client)

53
table/table.go
View File

@ -3,14 +3,15 @@ package table
import ( import (
"fmt" "fmt"
"net" "net"
"os"
"sync" "sync"
"io" "io"
"strconv" "strconv"
"strings" "strings"
"git.rootprojects.org/root/telebit/dbg"
telebit "git.rootprojects.org/root/telebit" telebit "git.rootprojects.org/root/telebit"
"git.rootprojects.org/root/telebit/dbg"
"github.com/gorilla/websocket" "github.com/gorilla/websocket"
) )
@ -53,13 +54,17 @@ func Add(server *SubscriberConn) {
func RemoveServer(server *SubscriberConn) bool { func RemoveServer(server *SubscriberConn) bool {
// TODO remove by RemoteAddr // TODO remove by RemoteAddr
//return false //return false
fmt.Printf("[warn] RemoveServer() still calls Remove(subject) instead of removing by RemoteAddr\n") fmt.Fprintf(
os.Stderr,
"[warn] RemoveServer() still calls Remove(subject) instead of removing by RemoteAddr\n",
)
return Remove(server.Grants.Subject) return Remove(server.Grants.Subject)
} }
func Remove(subject string) bool { func Remove(subject string) bool {
srvMapX, ok := Servers.Load(subject) srvMapX, ok := Servers.Load(subject)
fmt.Printf("[debug] has server for %s? %t\n", subject, ok) fmt.Printf("Remove(%s): exists? %t\n", subject, ok)
if !ok { if !ok {
return false return false
} }
@ -115,20 +120,24 @@ type SubscriberConn struct {
func (s *SubscriberConn) RouteBytes(src, dst telebit.Addr, payload []byte) { func (s *SubscriberConn) RouteBytes(src, dst telebit.Addr, payload []byte) {
id := fmt.Sprintf("%s:%d", src.Hostname(), src.Port()) id := fmt.Sprintf("%s:%d", src.Hostname(), src.Port())
if dbg.Debug { if dbg.Debug {
fmt.Println("[debug] Routing some more bytes:", dbg.Trunc(payload, len(payload))) fmt.Fprintf(
os.Stderr,
"[debug] Routing some more bytes: %s\n",
dbg.Trunc(payload, len(payload)),
)
fmt.Printf("\tid %s\nsrc %+v\n", id, src)
fmt.Printf("\tdst %s %+v\n", dst.Scheme(), dst)
} }
fmt.Printf("id %s\nsrc %+v\n", id, src)
fmt.Printf("dst %s %+v\n", dst.Scheme(), dst)
clientX, ok := s.Clients.Load(id) clientX, ok := s.Clients.Load(id)
if !ok { if !ok {
// TODO send back closed client error // TODO send back closed client error
fmt.Println("[debug] no client found for", id) fmt.Printf("RouteBytes({ %s }, %v, ...) [debug] no client found for %s\n", id, dst)
return return
} }
client, _ := clientX.(net.Conn) client, _ := clientX.(net.Conn)
if "end" == dst.Scheme() { if "end" == dst.Scheme() {
fmt.Println("[debug] closing client", id) fmt.Printf("RouteBytes: { %s }.Close(): %v\n", id, dst)
_ = client.Close() _ = client.Close()
return return
} }
@ -136,7 +145,7 @@ func (s *SubscriberConn) RouteBytes(src, dst telebit.Addr, payload []byte) {
for { for {
n, err := client.Write(payload) n, err := client.Write(payload)
if dbg.Debug { if dbg.Debug {
fmt.Println("[debug] table Write", dbg.Trunc(payload, len(payload))) fmt.Fprintf(os.Stderr, "[debug] table Write %s\n", dbg.Trunc(payload, len(payload)))
} }
if nil == err || io.EOF == err { if nil == err || io.EOF == err {
break break
@ -162,10 +171,12 @@ func (s *SubscriberConn) Serve(client net.Conn) error {
} }
id := client.RemoteAddr().String() id := client.RemoteAddr().String()
fmt.Printf("[DEBUG] NEW ID (ip:port) %s\n", id) if dbg.Debug {
fmt.Fprintf(os.Stderr, "[debug] NEW ID (ip:port) %s\n", id)
}
s.Clients.Store(id, client) s.Clients.Store(id, client)
//fmt.Println("[debug] immediately cancel client to simplify testing / debugging") //fmt.Fprintf(os.Stderr, "[debug] immediately cancel client to simplify testing / debugging\n")
//_ = client.Close() //_ = client.Close()
// TODO // TODO
@ -176,12 +187,16 @@ func (s *SubscriberConn) Serve(client net.Conn) error {
srcParts := strings.Split(client.RemoteAddr().String(), ":") srcParts := strings.Split(client.RemoteAddr().String(), ":")
srcAddr := srcParts[0] srcAddr := srcParts[0]
srcPort, _ := strconv.Atoi(srcParts[1]) srcPort, _ := strconv.Atoi(srcParts[1])
fmt.Println("[debug] srcParts", srcParts)
dstParts := strings.Split(client.LocalAddr().String(), ":") dstParts := strings.Split(client.LocalAddr().String(), ":")
dstAddr := dstParts[0] dstAddr := dstParts[0]
dstPort, _ := strconv.Atoi(dstParts[1]) dstPort, _ := strconv.Atoi(dstParts[1])
fmt.Println("[debug] dstParts", dstParts)
if dbg.Debug {
fmt.Fprintf(os.Stderr, "[debug] srcParts %v\n", srcParts)
fmt.Fprintf(os.Stderr, "[debug] dstParts %v\n", dstParts)
}
servername := wconn.Servername() servername := wconn.Servername()
termination := telebit.Unknown termination := telebit.Unknown
@ -210,12 +225,18 @@ func (s *SubscriberConn) Serve(client net.Conn) error {
dstAddr, dstAddr,
dstPort, dstPort,
) )
fmt.Printf("[debug] NewAddr src %+v\n", src)
fmt.Printf("[debug] NewAddr dst %+v\n", dst) if dbg.Debug {
fmt.Fprintf(os.Stderr, "[debug] NewAddr src %+v\n", src)
fmt.Fprintf(os.Stderr, "[debug] NewAddr dst %+v\n", dst)
}
err := s.MultiEncoder.Encode(wconn, *src, *dst) err := s.MultiEncoder.Encode(wconn, *src, *dst)
_ = wconn.Close() _ = wconn.Close()
fmt.Printf("[debug] Encoder Complete %+v %+v\n", id, err)
if dbg.Debug {
fmt.Fprintf(os.Stderr, "[debug] Encoder Complete %+v %+v\n", id, err)
}
s.Clients.Delete(id) s.Clients.Delete(id)
return err return err
} }

2
telebit.go
View File

@ -116,7 +116,7 @@ func Forward(client net.Conn, target net.Conn, timeout time.Duration) error {
}() }()
fmt.Println( fmt.Println(
"[debug] forwarding tcp connection", "Forwarding TCP connection",
client.LocalAddr(), client.LocalAddr(),
client.RemoteAddr(), client.RemoteAddr(),
target.LocalAddr(), target.LocalAddr(),

27
v1.go
View File

@ -3,6 +3,7 @@ package telebit
import ( import (
"errors" "errors"
"fmt" "fmt"
"os"
"strconv" "strconv"
"strings" "strings"
@ -52,7 +53,9 @@ func (p *Parser) unpackV1(b []byte) (int, error) {
z++ z++
n := len(b) n := len(b)
if n < 1 { if n < 1 {
fmt.Printf("[debug] v1 message unpacked (%d loops) (%d bytes left)\n", z, n) if dbg.Debug {
fmt.Fprintf(os.Stderr, "[debug] v1 message unpacked (%d loops) (%d bytes left)\n", z, n)
}
break break
} }
@ -90,7 +93,7 @@ func (p *Parser) unpackV1(b []byte) (int, error) {
return consumed, err return consumed, err
} }
default: default:
fmt.Println("[debug] v1 unknown state") fmt.Fprintf(os.Stderr, "[debug] v1 unknown state\n")
// do nothing // do nothing
consumed := p.consumed consumed := p.consumed
p.consumed = 0 p.consumed = 0
@ -142,9 +145,13 @@ func (p *Parser) unpackV1Header(b []byte, n int) ([]byte, error) {
port, _ := strconv.Atoi(parts[PortIndex]) port, _ := strconv.Atoi(parts[PortIndex])
service := parts[ServiceIndex] service := parts[ServiceIndex]
fmt.Printf("[debug] parts: %s\n", strings.Join(parts, " | ")) if dbg.Debug {
fmt.Fprintf(os.Stderr, "[debug] parts: %s\n", strings.Join(parts, " | "))
}
if "control" == service { if "control" == service {
fmt.Printf("[debug] control: %s\n", service) if dbg.Debug {
fmt.Fprintf(os.Stderr, "[debug] control: %s\n", service)
}
return nil, errors.New("'control' messages not implemented") return nil, errors.New("'control' messages not implemented")
} }
@ -185,7 +192,7 @@ func (p *Parser) unpackV1Header(b []byte, n int) ([]byte, error) {
*/ */
p.parseState++ p.parseState++
if dbg.Debug { if dbg.Debug {
fmt.Printf("[debug] unpackV1 parse state: %v\n", p.parseState) fmt.Fprintf(os.Stderr, "[debug] unpackV1 parse state: %v\n", p.parseState)
} }
if "end" == service { if "end" == service {
@ -197,7 +204,7 @@ func (p *Parser) unpackV1Header(b []byte, n int) ([]byte, error) {
func (p *Parser) unpackV1Payload(b []byte, n int) ([]byte, error) { func (p *Parser) unpackV1Payload(b []byte, n int) ([]byte, error) {
if dbg.Debug { if dbg.Debug {
fmt.Printf("[debug] unpackV1 payload state: %+v\n", p.state) fmt.Fprintf(os.Stderr, "[debug] unpackV1 payload state: %+v\n", p.state)
} }
// Handle "connect" and "end" // Handle "connect" and "end"
if 0 == p.state.payloadLen { if 0 == p.state.payloadLen {
@ -216,8 +223,10 @@ func (p *Parser) unpackV1Payload(b []byte, n int) ([]byte, error) {
return b, nil return b, nil
*/ */
//fmt.Printf("[debug] [2] payload written: %d | payload length: %d\n", p.state.payloadWritten, p.state.payloadLen) if dbg.Debug {
fmt.Printf("[debug] unpackV1 RouteBytes: %#v %#v %s\n", p.state.srcAddr, p.state.dstAddr, p.state.dstAddr.scheme) //fmt.Fprintf(os.Stderr, "[debug] [2] payload written: %d | payload length: %d\n", p.state.payloadWritten, p.state.payloadLen)
fmt.Fprintf(os.Stderr, "[debug] unpackV1 RouteBytes: %#v %#v %s\n", p.state.srcAddr, p.state.dstAddr, p.state.dstAddr.scheme)
}
p.handler.RouteBytes(p.state.srcAddr, p.state.dstAddr, []byte{}) p.handler.RouteBytes(p.state.srcAddr, p.state.dstAddr, []byte{})
return b, nil return b, nil
} }
@ -242,7 +251,7 @@ func (p *Parser) unpackV1Payload(b []byte, n int) ([]byte, error) {
p.consumed += k p.consumed += k
p.state.payloadWritten += k p.state.payloadWritten += k
//fmt.Printf("[debug] [1] payload written: %d | payload length: %d\n", p.state.payloadWritten, p.state.payloadLen) //fmt.Fprintf(os.Stderr, "[debug] [1] payload written: %d | payload length: %d\n", p.state.payloadWritten, p.state.payloadLen)
// if this payload is complete, reset all state // if this payload is complete, reset all state
if p.state.payloadWritten == p.state.payloadLen { if p.state.payloadWritten == p.state.payloadLen {
p.state = ParserState{} p.state = ParserState{}