+Authenticator::State LdapAuthenticator::init(const QVariantMap &settings)
+{
+ setConnectionProperties(settings);
+
+ bool status = ldapConnect();
+ if (!status)
+ {
+ quInfo() << qPrintable(displayName()) << "Authenticator cannot connect.";
+ return NotAvailable;
+ }
+
+ quInfo() << qPrintable(displayName()) << "Authenticator is ready.";
+ return IsReady;
+}
+
+// Method based on abustany LDAP quassel patch.
+bool LdapAuthenticator::ldapConnect()
+{
+ if (_connection != 0) {
+ ldapDisconnect();
+ }
+
+ int res, v = LDAP_VERSION3;
+
+ QString serverURI;
+ QByteArray serverURIArray;
+
+ // Convert info to hostname:port.
+ serverURI = _hostName + ":" + QString::number(_port);
+ serverURIArray = serverURI.toLocal8Bit();
+ res = ldap_initialize(&_connection, serverURIArray);
+
+ if (res != LDAP_SUCCESS) {
+ qWarning() << "Could not connect to LDAP server:" << ldap_err2string(res);
+ return false;
+ }
+
+ res = ldap_set_option(_connection, LDAP_OPT_PROTOCOL_VERSION, (void*)&v);
+
+ if (res != LDAP_SUCCESS) {
+ qWarning() << "Could not set LDAP protocol version to v3:" << ldap_err2string(res);
+ ldap_unbind_ext(_connection, 0, 0);
+ _connection = 0;
+ return false;
+ }
+
+ return true;
+}
+
+void LdapAuthenticator::ldapDisconnect()
+{
+ if (_connection == 0) {
+ return;
+ }
+
+ ldap_unbind_ext(_connection, 0, 0);
+ _connection = 0;
+}
+
+bool LdapAuthenticator::ldapAuth(const QString &username, const QString &password)
+{
+ if (password.isEmpty()) {
+ return false;
+ }
+
+ int res;
+
+ // Attempt to establish a connection.
+ if (_connection == 0) {
+ if (not ldapConnect()) {
+ return false;
+ }
+ }
+
+ struct berval cred;
+
+ // Convert some things to byte arrays as needed.
+ QByteArray bindPassword = _bindPassword.toLocal8Bit();
+ QByteArray bindDN = _bindDN.toLocal8Bit();
+ QByteArray baseDN = _baseDN.toLocal8Bit();
+ QByteArray uidAttribute = _uidAttribute.toLocal8Bit();
+
+ cred.bv_val = const_cast<char*>(bindPassword.size() > 0 ? bindPassword.constData() : NULL);
+ cred.bv_len = bindPassword.size();
+
+ res = ldap_sasl_bind_s(_connection, bindDN.size() > 0 ? bindDN.constData() : 0, LDAP_SASL_SIMPLE, &cred, 0, 0, 0);
+
+ if (res != LDAP_SUCCESS) {
+ qWarning() << "Refusing connection from" << username << "(LDAP bind failed:" << ldap_err2string(res) << ")";
+ ldapDisconnect();
+ return false;
+ }
+
+ LDAPMessage *msg = NULL, *entry = NULL;
+
+ const QByteArray ldapQuery = "(&(" + uidAttribute + '=' + username.toLocal8Bit() + ")" + _filter.toLocal8Bit() + ")";
+
+ res = ldap_search_ext_s(_connection, baseDN.constData(), LDAP_SCOPE_SUBTREE, ldapQuery.constData(), 0, 0, 0, 0, 0, 0, &msg);
+
+ if (res != LDAP_SUCCESS) {
+ qWarning() << "Refusing connection from" << username << "(LDAP search failed:" << ldap_err2string(res) << ")";
+ return false;
+ }
+
+ if (ldap_count_entries(_connection, msg) > 1) {
+ qWarning() << "Refusing connection from" << username << "(LDAP search returned more than one result)";
+ ldap_msgfree(msg);
+ return false;
+ }
+
+ entry = ldap_first_entry(_connection, msg);
+
+ if (entry == 0) {
+ qWarning() << "Refusing connection from" << username << "(LDAP search returned no results)";
+ ldap_msgfree(msg);
+ return false;
+ }
+
+ const QByteArray passwordArray = password.toLocal8Bit();
+ cred.bv_val = const_cast<char*>(passwordArray.constData());
+ cred.bv_len = password.size();
+
+ char *userDN = ldap_get_dn(_connection, entry);
+
+ res = ldap_sasl_bind_s(_connection, userDN, LDAP_SASL_SIMPLE, &cred, 0, 0, 0);
+
+ if (res != LDAP_SUCCESS) {
+ qWarning() << "Refusing connection from" << username << "(LDAP authentication failed)";
+ ldap_memfree(userDN);
+ ldap_msgfree(msg);
+ return false;
+ }
+
+ // The original implementation had requiredAttributes. I have not included this code
+ // but it would be easy to re-add if someone wants this feature.
+ // Ben Rosser <bjr@acm.jhu.edu> (12/23/15).
+
+ ldap_memfree(userDN);
+ ldap_msgfree(msg);
+ return true;
+}
\ No newline at end of file
projects / quassel.git / commitdiff