@@ -17,7 +17,8 @@ import (
1717
1818// Config contains configurations about log4shell server.
1919type Config struct {
20- LogOut io.Writer
20+ // Logger is used to set server logger writer.
21+ Logger io.Writer
2122
2223 // Hostname can be set IP address or domain name,
2324 // If enable AutoCert, must set domain name.
@@ -70,8 +71,8 @@ type Server struct {
7071// New is used to create a new log4shell server.
7172func New (cfg * Config ) (* Server , error ) {
7273 // check configuration
73- if cfg .LogOut == nil {
74- panic ("log4shell: Config.LogOut can not be nil" )
74+ if cfg .Logger == nil {
75+ panic ("log4shell: Config.Logger can not be nil" )
7576 }
7677 if cfg .Hostname == "" {
7778 return nil , errors .New ("empty host name" )
@@ -84,8 +85,8 @@ func New(cfg *Config) (*Server, error) {
8485 return nil , errors .Errorf ("\" %s\" is not a directory" , cfg .PayloadDir )
8586 }
8687
87- // set logger
88- logger := log .New (cfg .LogOut , "" , log .LstdFlags )
88+ // set server logger
89+ logger := log .New (cfg .Logger , "" , log .LstdFlags )
8990 ldapserver .Logger = logger
9091
9192 // initial tls config
@@ -146,10 +147,10 @@ func New(cfg *Config) (*Server, error) {
146147 return nil , errors .WithStack (err )
147148 }
148149 addr := net .JoinHostPort (cfg .Hostname , port )
149- url := fmt .Sprintf ("%s://%s/%s/" , scheme , addr , secret )
150+ codeBase := fmt .Sprintf ("%s://%s/%s/" , scheme , addr , secret )
150151 ldapHandler := ldapHandler {
151- logger : logger ,
152- url : url ,
152+ logger : logger ,
153+ codeBase : codeBase ,
153154 }
154155 ldapRoute := ldapserver .NewRouteMux ()
155156 ldapRoute .Bind (ldapHandler .handleBind )
0 commit comments