allow handlers to handle unrecognized messages

merlinz01 · Apr 13, 2024 · 9a224a7 · 9a224a7
1 parent 6000c6d
commit 9a224a7
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/handler.go b/handler.go
@@ -24,6 +24,8 @@ type Handler interface {
 	ModifyDN(*Conn, *Message, *ModifyDNRequest)
 	// Perform a Search request
 	Search(*Conn, *Message, *SearchRequest)
+	// Handle unrecognized requests
+	Other(*Conn, *Message)
 }
 
 // Basic server functionality.
@@ -104,3 +106,7 @@ func (*BaseHandler) StartTLS(conn *Conn, msg *Message) {
 	}
 	conn.SendResult(msg.MessageID, nil, TypeExtendedResponseOp, &res)
 }
+
+func (*BaseHandler) Other(conn *Conn, msg *Message) {
+	conn.SendResult(msg.MessageID, nil, BerTypeSequence, UnsupportedOperation)
+}
diff --git a/server.go b/server.go
@@ -246,6 +246,6 @@ func (s *LDAPServer) handleMessage(conn *Conn, msg *Message) {
 		conn.Close()
 	default:
 		log.Println("Unknown operation type:", msg.ProtocolOp.Type)
-		conn.SendResult(msg.MessageID, nil, BerTypeSequence, UnsupportedOperation)
+		s.Handler.Other(conn, msg)
 	}
 }