mirror of
https://github.com/daeuniverse/dae.git
synced 2025-01-08 14:34:08 +07:00
50 lines
860 B
Go
50 lines
860 B
Go
/*
|
|
* SPDX-License-Identifier: AGPL-3.0-only
|
|
* Copyright (c) 2022-2023, daeuniverse Organization <dae@v2raya.org>
|
|
*/
|
|
|
|
package sniffing
|
|
|
|
import (
|
|
"errors"
|
|
"net"
|
|
"strings"
|
|
"sync"
|
|
)
|
|
|
|
type ConnSniffer struct {
|
|
net.Conn
|
|
*Sniffer
|
|
|
|
mu sync.Mutex
|
|
}
|
|
|
|
func NewConnSniffer(conn net.Conn, snifferBufSize int) *ConnSniffer {
|
|
s := &ConnSniffer{
|
|
Conn: conn,
|
|
Sniffer: NewStreamSniffer(conn, snifferBufSize),
|
|
}
|
|
return s
|
|
}
|
|
|
|
func (s *ConnSniffer) Read(p []byte) (n int, err error) {
|
|
s.mu.Lock()
|
|
n, err = s.Sniffer.Read(p)
|
|
s.mu.Unlock()
|
|
return n, err
|
|
}
|
|
|
|
func (s *ConnSniffer) Close() (err error) {
|
|
var errs []string
|
|
if err = s.Sniffer.Close(); err != nil {
|
|
errs = append(errs, err.Error())
|
|
}
|
|
if err = s.Conn.Close(); err != nil {
|
|
errs = append(errs, err.Error())
|
|
}
|
|
if len(errs) > 0 {
|
|
return errors.New(strings.Join(errs, "; "))
|
|
}
|
|
return nil
|
|
}
|