1 files changed, 95 insertions, 0 deletions
diff --git a/weed/udptransfer/debug.go b/weed/udptransfer/debug.go
new file mode 100644
index 000000000..dc8dbb501
--- /dev/null
+++ b/weed/udptransfer/debug.go
@@ -0,0 +1,95 @@
+package udptransfer
+
+import (
+	"encoding/hex"
+	"fmt"
+	"log"
+	"os"
+	"os/signal"
+	"runtime"
+	"runtime/pprof"
+	"sync/atomic"
+	"syscall"
+)
+
+var enable_pprof bool
+var enable_stacktrace bool
+var debug_inited int32
+
+func set_debug_params(p *Params) {
+	if atomic.CompareAndSwapInt32(&debug_inited, 0, 1) {
+		debug = p.Debug
+		enable_pprof = p.EnablePprof
+		enable_stacktrace = p.Stacktrace
+		if enable_pprof {
+			f, err := os.Create("udptransfer.pprof")
+			if err != nil {
+				log.Fatalln(err)
+			}
+			pprof.StartCPUProfile(f)
+		}
+	}
+}
+
+func (c *Conn) PrintState() {
+	log.Printf("inQ=%d inQReady=%d outQ=%d", c.inQ.size(), len(c.inQReady), c.outQ.size())
+	log.Printf("inMaxCtnSeq=%d lastAck=%d lastReadSeq=%d", c.inQ.maxCtnSeq, c.lastAck, c.lastReadSeq)
+	if c.inPkCnt > 0 {
+		log.Printf("Rx pcnt=%d dups=%d %%d=%f%%", c.inPkCnt, c.inDupCnt, 100*float32(c.inDupCnt)/float32(c.inPkCnt))
+	}
+	if c.outPkCnt > 0 {
+		log.Printf("Tx pcnt=%d dups=%d %%d=%f%%", c.outPkCnt, c.outDupCnt, 100*float32(c.outDupCnt)/float32(c.outPkCnt))
+	}
+	log.Printf("current-rtt=%d FastRetransmit=%d", c.rtt, c.fRCnt)
+	if enable_stacktrace {
+		var buf = make([]byte, 6400)
+		for i := 0; i < 3; i++ {
+			n := runtime.Stack(buf, true)
+			if n >= len(buf) {
+				buf = make([]byte, len(buf)<<1)
+			} else {
+				buf = buf[:n]
+				break
+			}
+		}
+		fmt.Println(string(buf))
+	}
+	if enable_pprof {
+		pprof.StopCPUProfile()
+	}
+}
+
+func (c *Conn) internal_state() {
+	ev := make(chan os.Signal, 10)
+	signal.Notify(ev, syscall.Signal(12), syscall.SIGINT)
+	for v := range ev {
+		c.PrintState()
+		if v == syscall.SIGINT {
+			os.Exit(1)
+		}
+	}
+}
+
+func printBits(b uint64, j, s, d uint32) {
+	fmt.Printf("bits=%064b j=%d seq=%d dis=%d\n", b, j, s, d)
+}
+
+func dumpb(label string, buf []byte) {
+	log.Println(label, "\n", hex.Dump(buf))
+}
+
+func dumpQ(s string, q *linkedMap) {
+	var seqs = make([]uint32, 0, 20)
+	n := q.head
+	for i, m := int32(0), q.size(); i < m && n != nil; i++ {
+		seqs = append(seqs, n.seq)
+		n = n.next
+		if len(seqs) == 20 {
+			log.Printf("%s: Q=%d", s, seqs)
+			seqs = seqs[:0]
+		}
+	}
+	if len(seqs) > 0 {
+		log.Printf("%s: Q=%d", s, seqs)
+	}
+}