2 Commits

+22 -8
View File
@@ -26,7 +26,7 @@ import (
) )
const ( const (
version = "1.0.1" version = "1.0.3"
) )
var ( var (
@@ -176,6 +176,8 @@ func handleConnection(connClt net.Conn, conLdap *ldap.Conn) {
// Dont notice if client closed connection // Dont notice if client closed connection
if err.Error() != "EOF" && !strings.HasSuffix(err.Error(), "i/o timeout") { if err.Error() != "EOF" && !strings.HasSuffix(err.Error(), "i/o timeout") {
logstream.Errorf("Error reading connection: %v\n", err.Error()) logstream.Errorf("Error reading connection: %v\n", err.Error())
} else {
logstream.Debug("Error reading on connection : %s", err.Error())
} }
return return
} }
@@ -276,12 +278,14 @@ func sendResponse(con net.Conn, respMsg string, respCode int) error {
return nil return nil
} }
// attempt is useless now. It was introduced to break after N attempts, but this is not implemented. Should it be?
func searchLdap(searchReq *ldap.SearchRequest, attempt int) (*ldap.SearchResult, error) { func searchLdap(searchReq *ldap.SearchRequest, attempt int) (*ldap.SearchResult, error) {
mutex.Lock() mutex.Lock()
result, err := conLdap.Search(searchReq) result, err := conLdap.Search(searchReq)
mutex.Unlock() mutex.Unlock()
// Let's just manage connection errors here // Let's just manage connection errors here
if err != nil && strings.HasSuffix(err.Error(), "ldap: connection closed") { if err != nil {
if strings.HasSuffix(err.Error(), "ldap: connection closed") || err == ldap.ErrNilConnection {
logstream.Error("LDAP connection closed, retrying") logstream.Error("LDAP connection closed, retrying")
mutex.Lock() mutex.Lock()
// 16/01/2023: panic: runtime error: invalid memory address or nil pointer dereference // 16/01/2023: panic: runtime error: invalid memory address or nil pointer dereference
@@ -298,6 +302,7 @@ func searchLdap(searchReq *ldap.SearchRequest, attempt int) (*ldap.SearchResult,
return searchLdap(searchReq, attempt) return searchLdap(searchReq, attempt)
} }
} }
}
return result, err return result, err
} }
@@ -316,13 +321,22 @@ func connectLdap() (*ldap.Conn, error) {
return conLdap, err return conLdap, err
} }
// TODO : buildNetCache should have its own LDAP connection func periodicallyUpdateCache() {
func periodicallyUpdateCache(conLdap *ldap.Conn) { myLdap, err := connectLdap()
if err != nil {
return
}
// On initialise immediatement le cache // On initialise immediatement le cache
buildNetCacheFromIPNetwork(conLdap) buildNetCacheFromIPNetwork(myLdap)
myLdap.Close()
for range time.Tick(time.Second * time.Duration(*refreshInterval)) { for range time.Tick(time.Second * time.Duration(*refreshInterval)) {
buildNetCacheFromIPNetwork(conLdap) myLdap, err := connectLdap()
if err != nil {
return
}
buildNetCacheFromIPNetwork(myLdap)
myLdap.Close()
} }
} }
@@ -336,7 +350,7 @@ func run() {
return return
} }
go periodicallyUpdateCache(conLdap) go periodicallyUpdateCache()
// Spawn a go routine for incoming connection // Spawn a go routine for incoming connection
for { for {
@@ -375,7 +389,7 @@ func main() {
return return
} }
fmt.Printf("MyNetTCPTable v.%s\n", version) fmt.Printf("%s: MyNetTCPTable v.%s listening on %s\n", time.Now().Format(time.RFC3339), version, *listen)
logstream = logrus.New() logstream = logrus.New()
level, err := logrus.ParseLevel(*logLevel) level, err := logrus.ParseLevel(*logLevel)