diff options
| author | Shulhan <ms@kilabit.info> | 2025-01-17 00:18:36 +0700 |
|---|---|---|
| committer | Shulhan <ms@kilabit.info> | 2025-01-17 00:18:36 +0700 |
| commit | 73a9b318ea8afe535050c1a007e427b6cd8be453 (patch) | |
| tree | 22a9b914fb4899b27600888924da7575afc502b6 | |
| parent | 20101c63d8cb9d5d673bbd2b93f9dfd8344f06a2 (diff) | |
| download | pakakeh.go-73a9b318ea8afe535050c1a007e427b6cd8be453.tar.xz | |
lib/dns: improve the logging prefix on serveTCPClient
The serveTCPClient is used to serve TCP and DoT clients.
Previously, the error returned from this method is prefixed
based on the kind, for example
serveTCPClient TCP: ...
serveTCPClient DoT: ...
This changes pass the log prefix to the method so now it
become
serveTCPClient: ...
serveDoTClient: ...
| -rw-r--r-- | lib/dns/server.go | 23 |
1 files changed, 9 insertions, 14 deletions
diff --git a/lib/dns/server.go b/lib/dns/server.go index 96c4f0fc..8d2c7f4e 100644 --- a/lib/dns/server.go +++ b/lib/dns/server.go @@ -305,7 +305,7 @@ func (srv *Server) serveDoT() { conn: conn, } - go srv.serveTCPClient(cl, connTypeDoT) + go srv.serveTCPClient(`serveDoTClient`, cl, connTypeDoT) } } } @@ -338,7 +338,7 @@ func (srv *Server) serveTCP() { conn: conn, } - go srv.serveTCPClient(cl, connTypeTCP) + go srv.serveTCPClient(`serveTCPClient`, cl, connTypeTCP) } } @@ -510,20 +510,15 @@ func (srv *Server) incForwarder() { srv.fwLocker.Unlock() } -func (srv *Server) serveTCPClient(cl *TCPClient, kind connType) { - var ( - logp = `serveTCPClient` - - req *request - err error - ) +func (srv *Server) serveTCPClient(logp string, cl *TCPClient, kind connType) { for { - req = newRequest() + var err error + var req = newRequest() req.message.packet, err = cl.recv() if err != nil { if !errors.Is(err, io.EOF) { - log.Printf(`%s %s: %s`, logp, connTypeNames[kind], err) + log.Printf(`%s: %s`, logp, err) } break } @@ -533,7 +528,7 @@ func (srv *Server) serveTCPClient(cl *TCPClient, kind connType) { err = req.message.UnpackHeaderQuestion() if err != nil { - log.Printf(`%s %s: %s`, logp, connTypeNames[kind], err) + log.Printf(`%s: %s`, logp, err) req.error(RCodeErrServer) continue } @@ -541,9 +536,9 @@ func (srv *Server) serveTCPClient(cl *TCPClient, kind connType) { srv.requestq <- req } - err = cl.conn.Close() + var err = cl.conn.Close() if err != nil { - log.Printf(`%s %s: connection closed: %s`, logp, connTypeNames[kind], err) + log.Printf(`%s: connection closed: %s`, logp, err) } } |
