bufconn.go 2.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106
  1. package net
  2. import (
  3. "bufio"
  4. "net"
  5. "github.com/metacubex/mihomo/common/buf"
  6. )
  7. var _ ExtendedConn = (*BufferedConn)(nil)
  8. type BufferedConn struct {
  9. r *bufio.Reader
  10. ExtendedConn
  11. peeked bool
  12. }
  13. func NewBufferedConn(c net.Conn) *BufferedConn {
  14. if bc, ok := c.(*BufferedConn); ok {
  15. return bc
  16. }
  17. return &BufferedConn{bufio.NewReader(c), NewExtendedConn(c), false}
  18. }
  19. func WarpConnWithBioReader(c net.Conn, br *bufio.Reader) net.Conn {
  20. if br != nil && br.Buffered() > 0 {
  21. if bc, ok := c.(*BufferedConn); ok && bc.r == br {
  22. return bc
  23. }
  24. return &BufferedConn{br, NewExtendedConn(c), true}
  25. }
  26. return c
  27. }
  28. // Reader returns the internal bufio.Reader.
  29. func (c *BufferedConn) Reader() *bufio.Reader {
  30. return c.r
  31. }
  32. func (c *BufferedConn) ResetPeeked() {
  33. c.peeked = false
  34. }
  35. func (c *BufferedConn) Peeked() bool {
  36. return c.peeked
  37. }
  38. // Peek returns the next n bytes without advancing the reader.
  39. func (c *BufferedConn) Peek(n int) ([]byte, error) {
  40. c.peeked = true
  41. return c.r.Peek(n)
  42. }
  43. func (c *BufferedConn) Discard(n int) (discarded int, err error) {
  44. return c.r.Discard(n)
  45. }
  46. func (c *BufferedConn) Read(p []byte) (int, error) {
  47. return c.r.Read(p)
  48. }
  49. func (c *BufferedConn) ReadByte() (byte, error) {
  50. return c.r.ReadByte()
  51. }
  52. func (c *BufferedConn) UnreadByte() error {
  53. return c.r.UnreadByte()
  54. }
  55. func (c *BufferedConn) Buffered() int {
  56. return c.r.Buffered()
  57. }
  58. func (c *BufferedConn) ReadBuffer(buffer *buf.Buffer) (err error) {
  59. if c.r != nil && c.r.Buffered() > 0 {
  60. _, err = buffer.ReadOnceFrom(c.r)
  61. return
  62. }
  63. return c.ExtendedConn.ReadBuffer(buffer)
  64. }
  65. func (c *BufferedConn) ReadCached() *buf.Buffer { // call in sing/common/bufio.Copy
  66. if c.r != nil && c.r.Buffered() > 0 {
  67. length := c.r.Buffered()
  68. b, _ := c.r.Peek(length)
  69. _, _ = c.r.Discard(length)
  70. return buf.As(b)
  71. }
  72. c.r = nil // drop bufio.Reader to let gc can clean up its internal buf
  73. return nil
  74. }
  75. func (c *BufferedConn) Upstream() any {
  76. return c.ExtendedConn
  77. }
  78. func (c *BufferedConn) ReaderReplaceable() bool {
  79. if c.r != nil && c.r.Buffered() > 0 {
  80. return false
  81. }
  82. return true
  83. }
  84. func (c *BufferedConn) WriterReplaceable() bool {
  85. return true
  86. }