Add xdr level tracing

This commit is contained in:
Jakob Borg 2014-05-17 08:43:01 +02:00
parent e788a99c3e
commit 13022817d4
3 changed files with 52 additions and 5 deletions

View File

@ -80,6 +80,7 @@ const (
- "net" (the main packge; connections & network messages) - "net" (the main packge; connections & network messages)
- "scanner" (the scanner package) - "scanner" (the scanner package)
- "upnp" (the upnp package) - "upnp" (the upnp package)
- "xdr" (the xdr package)
- "all" (all of the above) - "all" (all of the above)
STCPUPROFILE Write CPU profile to the specified file.` STCPUPROFILE Write CPU profile to the specified file.`

View File

@ -48,6 +48,9 @@ func (r *Reader) ReadBytesMaxInto(max int, dst []byte) []byte {
if r.err != nil { if r.err != nil {
return nil return nil
} }
if debug {
dl.Debugf("rd bytes len=%d", l)
}
if max > 0 && l > max { if max > 0 && l > max {
r.err = ErrElementSizeExceeded r.err = ErrElementSizeExceeded
return nil return nil
@ -57,8 +60,16 @@ func (r *Reader) ReadBytesMaxInto(max int, dst []byte) []byte {
} else { } else {
dst = dst[:l+pad(l)] dst = dst[:l+pad(l)]
} }
_, r.err = io.ReadFull(r.r, dst) var n int
r.tot += l + pad(l) n, r.err = io.ReadFull(r.r, dst)
r.tot += n
if debug {
if n > maxDebugBytes {
dl.Debugf("rd bytes (%d): %x...", n, dst[:maxDebugBytes])
} else {
dl.Debugf("rd bytes (%d): %x", n, dst)
}
}
return dst[:l] return dst[:l]
} }
@ -68,7 +79,11 @@ func (r *Reader) ReadUint16() uint16 {
} }
_, r.err = io.ReadFull(r.r, r.b[:4]) _, r.err = io.ReadFull(r.r, r.b[:4])
r.tot += 4 r.tot += 4
return uint16(r.b[1]) | uint16(r.b[0])<<8 v := uint16(r.b[1]) | uint16(r.b[0])<<8
if debug {
dl.Debugf("rd uint16=%d", v)
}
return v
} }
func (r *Reader) ReadUint32() uint32 { func (r *Reader) ReadUint32() uint32 {
@ -81,7 +96,11 @@ func (r *Reader) ReadUint32() uint32 {
return 0 return 0
} }
r.tot += n r.tot += n
return uint32(r.b[3]) | uint32(r.b[2])<<8 | uint32(r.b[1])<<16 | uint32(r.b[0])<<24 v := uint32(r.b[3]) | uint32(r.b[2])<<8 | uint32(r.b[1])<<16 | uint32(r.b[0])<<24
if debug {
dl.Debugf("rd uint32=%d", v)
}
return v
} }
func (r *Reader) ReadUint64() uint64 { func (r *Reader) ReadUint64() uint64 {
@ -91,8 +110,12 @@ func (r *Reader) ReadUint64() uint64 {
} }
n, r.err = io.ReadFull(r.r, r.b[:8]) n, r.err = io.ReadFull(r.r, r.b[:8])
r.tot += n r.tot += n
return uint64(r.b[7]) | uint64(r.b[6])<<8 | uint64(r.b[5])<<16 | uint64(r.b[4])<<24 | v := uint64(r.b[7]) | uint64(r.b[6])<<8 | uint64(r.b[5])<<16 | uint64(r.b[4])<<24 |
uint64(r.b[3])<<32 | uint64(r.b[2])<<40 | uint64(r.b[1])<<48 | uint64(r.b[0])<<56 uint64(r.b[3])<<32 | uint64(r.b[2])<<40 | uint64(r.b[1])<<48 | uint64(r.b[0])<<56
if debug {
dl.Debugf("rd uint64=%d", v)
}
return v
} }
func (r *Reader) Tot() int { func (r *Reader) Tot() int {

View File

@ -39,6 +39,14 @@ func (w *Writer) WriteBytes(bs []byte) (int, error) {
return 0, w.err return 0, w.err
} }
if debug {
if len(bs) > maxDebugBytes {
dl.Debugf("wr bytes (%d): %x...", len(bs), bs[:maxDebugBytes])
} else {
dl.Debugf("wr bytes (%d): %x", len(bs), bs)
}
}
var l, n int var l, n int
n, w.err = w.w.Write(bs) n, w.err = w.w.Write(bs)
l += n l += n
@ -56,6 +64,11 @@ func (w *Writer) WriteUint16(v uint16) (int, error) {
if w.err != nil { if w.err != nil {
return 0, w.err return 0, w.err
} }
if debug {
dl.Debugf("wr uint16=%d", v)
}
w.b[0] = byte(v >> 8) w.b[0] = byte(v >> 8)
w.b[1] = byte(v) w.b[1] = byte(v)
w.b[2] = 0 w.b[2] = 0
@ -71,6 +84,11 @@ func (w *Writer) WriteUint32(v uint32) (int, error) {
if w.err != nil { if w.err != nil {
return 0, w.err return 0, w.err
} }
if debug {
dl.Debugf("wr uint32=%d", v)
}
w.b[0] = byte(v >> 24) w.b[0] = byte(v >> 24)
w.b[1] = byte(v >> 16) w.b[1] = byte(v >> 16)
w.b[2] = byte(v >> 8) w.b[2] = byte(v >> 8)
@ -86,6 +104,11 @@ func (w *Writer) WriteUint64(v uint64) (int, error) {
if w.err != nil { if w.err != nil {
return 0, w.err return 0, w.err
} }
if debug {
dl.Debugf("wr uint64=%d", v)
}
w.b[0] = byte(v >> 56) w.b[0] = byte(v >> 56)
w.b[1] = byte(v >> 48) w.b[1] = byte(v >> 48)
w.b[2] = byte(v >> 40) w.b[2] = byte(v >> 40)