@@ -50,6 +50,7 @@ type Config struct {
5050// check and exploit Apache Log4j2 vulnerability easily.
5151type Server struct {
5252 logger * log.Logger
53+ hostname string
5354 enableTLS bool
5455
5556 secret string
@@ -162,6 +163,7 @@ func New(cfg *Config) (*Server, error) {
162163 // create log4shell server
163164 server := Server {
164165 logger : logger ,
166+ hostname : cfg .Hostname ,
165167 enableTLS : enableTLS ,
166168 secret : secret ,
167169 httpListener : httpListener ,
@@ -213,6 +215,8 @@ func (srv *Server) Start() error {
213215 case <- time .After (250 * time .Millisecond ):
214216 }
215217
218+ srv .logger .Println ("[info]" , "current hostname:" , srv .hostname )
219+ srv .logger .Println ("[info]" , "secret about http handler:" , srv .secret )
216220 if srv .enableTLS {
217221 srv .logger .Println ("[info]" , "start https server" , srv .httpListener .Addr ())
218222 srv .logger .Println ("[info]" , "start ldaps server" , srv .ldapListener .Addr ())
@@ -231,14 +235,22 @@ func (srv *Server) Stop() error {
231235
232236 // close ldap server
233237 srv .ldapServer .Stop ()
234- srv .logger .Println ("[info]" , "ldap server is stopped" )
238+ if srv .enableTLS {
239+ srv .logger .Println ("[info]" , "ldaps server is stopped" )
240+ } else {
241+ srv .logger .Println ("[info]" , "ldap server is stopped" )
242+ }
235243
236244 // close http server
237245 err := srv .httpServer .Close ()
238246 if err != nil {
239247 return errors .Wrap (err , "failed to close http server" )
240248 }
241- srv .logger .Println ("[info]" , "http server is stopped" )
249+ if srv .enableTLS {
250+ srv .logger .Println ("[info]" , "https server is stopped" )
251+ } else {
252+ srv .logger .Println ("[info]" , "http server is stopped" )
253+ }
242254
243255 srv .wg .Wait ()
244256 srv .logger .Println ("[info]" , "log4shell server is stopped" )
0 commit comments