Skip to content

Optimize RootDse search and add cancellation tokens to async search extensions. #173

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 2 commits into
base: master
Choose a base branch
from
Open

Optimize RootDse search and add cancellation tokens to async search extensions. #173

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
67049f2
Get RootDse entry with all attributes in a single request, add cancel…
INEL-AJ Nov 24, 2022
5b07428
RootDse search extensions should now work for all servers, added some…
INEL-AJ Dec 1, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
77 changes: 46 additions & 31 deletions LdapForNet/LdapSearchExtensions.cs
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
using System.Collections.Generic;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using LdapForNet.Utils;
using static LdapForNet.Native.Native;
Expand All @@ -8,50 +9,63 @@ namespace LdapForNet
{
public static class LdapSearchExtensions
{
public static IList<LdapEntry> SearchByCn(this ILdapConnection connection, string @base, string cn,
LdapSearchScope scope = LdapSearchScope.LDAP_SCOPE_SUBTREE)
public static LdapEntry GetRootDse(this ILdapConnection connection)
{
return connection.Search(@base, $"(cn={cn})", scope: scope);
return connection.Search(
null,
"(objectclass=*)",
new[]
{
"*", "+", "serverName", "dNSHostName", "dsServiceName", "currentTime", "defaultNamingContext",
"rootDomainNamingContext", "configurationNamingContext", "schemaNamingContext", "subschemaSubentry",
"namingContexts", "supportedLDAPVersion", "supportedCapabilities", "supportedControl",
"supportedLDAPPolicies", "supportedSaslMechanisms", "supportedExtension", "vendorName", "vendorVersion",
"domainControllerFunctionality", "domainFunctionality", "forestFunctionality", "isSynchronized", "objectClass"
},
LdapSearchScope.LDAP_SCOPE_BASE)
.First();
}

public static LdapEntry GetRootDse(this ILdapConnection connection)
public static async Task<LdapEntry> GetRootDseAsync(this ILdapConnection connection,
CancellationToken token = default)
{
var result = connection.Search(null, "(objectclass=*)",
new[]
{
"namingContexts", "subschemaSubentry", "supportedLDAPVersion", "supportedSASLMechanisms",
"supportedExtension", "supportedControl", "supportedFeatures", "vendorName", "vendorVersion"
}, LdapSearchScope.LDAP_SCOPE_BASE)
.FirstOrDefault();
if (result == null)
{
return null;
}

var rootDse = connection.Search(null, "(objectclass=*)", scope: LdapSearchScope.LDAP_SCOPE_BASE).First();
foreach (var attribute in rootDse.DirectoryAttributes)
{
result.DirectoryAttributes.Remove(attribute.Name);
result.DirectoryAttributes.Add(attribute);
}
return (await connection.SearchAsync(
null,
"(objectclass=*)",
new[]
{
"*", "+", "serverName", "dNSHostName", "dsServiceName", "currentTime", "defaultNamingContext",
"rootDomainNamingContext", "configurationNamingContext", "schemaNamingContext", "subschemaSubentry",
"namingContexts", "supportedLDAPVersion", "supportedCapabilities", "supportedControl",
"supportedLDAPPolicies", "supportedSaslMechanisms", "supportedExtension", "vendorName", "vendorVersion",
"domainControllerFunctionality", "domainFunctionality", "forestFunctionality", "isSynchronized", "objectClass"
},
LdapSearchScope.LDAP_SCOPE_BASE,
token))
.First();
}

return result;
public static IList<LdapEntry> SearchByCn(this ILdapConnection connection, string @base, string cn,
LdapSearchScope scope = LdapSearchScope.LDAP_SCOPE_SUBTREE)
{
return connection.Search(@base, $"(cn={cn})", scope: scope);
}

public static async Task<IList<LdapEntry>> SearchByCnAsync(this ILdapConnection connection, string @base,
string cn, LdapSearchScope scope = LdapSearchScope.LDAP_SCOPE_SUBTREE)
string cn, LdapSearchScope scope = LdapSearchScope.LDAP_SCOPE_SUBTREE, CancellationToken token = default)
{
return await connection.SearchAsync(@base, $"(cn={cn})", scope: scope);
return await connection.SearchAsync(@base, $"(cn={cn})", scope: scope, token: token);
}

public static IList<LdapEntry> SearchByCn(this ILdapConnection connection, string cn)
{
return connection.SearchByCn(LdapUtils.GetDnFromHostname(), cn);
}

public static async Task<IList<LdapEntry>> SearchByCnAsync(this ILdapConnection connection, string cn)
public static async Task<IList<LdapEntry>> SearchByCnAsync(this ILdapConnection connection, string cn,
CancellationToken token = default)
{
return await connection.SearchByCnAsync(LdapUtils.GetDnFromHostname(), cn);
return await connection.SearchByCnAsync(LdapUtils.GetDnFromHostname(), cn, token: token);
}

public static IList<LdapEntry> SearchBySid(this ILdapConnection connection, string @base, string sid,
Expand All @@ -62,20 +76,21 @@ public static IList<LdapEntry> SearchBySid(this ILdapConnection connection, stri
}

public static async Task<IList<LdapEntry>> SearchBySidAsync(this ILdapConnection connection, string @base,
string sid, LdapSearchScope scope = LdapSearchScope.LDAP_SCOPE_SUBTREE)
string sid, LdapSearchScope scope = LdapSearchScope.LDAP_SCOPE_SUBTREE, CancellationToken token = default)
{
var hex = HexEscaper.Escape(LdapSidConverter.ConvertToHex(sid));
return await connection.SearchAsync(@base, $"(objectSID={hex})", scope: scope);
return await connection.SearchAsync(@base, $"(objectSID={hex})", scope: scope, token: token);
}

public static IList<LdapEntry> SearchBySid(this ILdapConnection connection, string sid)
{
return connection.SearchBySid(LdapUtils.GetDnFromHostname(), sid);
}

public static async Task<IList<LdapEntry>> SearchBySidAsync(this ILdapConnection connection, string sid)
public static async Task<IList<LdapEntry>> SearchBySidAsync(this ILdapConnection connection, string sid,
CancellationToken token = default)
{
return await connection.SearchBySidAsync(LdapUtils.GetDnFromHostname(), sid);
return await connection.SearchBySidAsync(LdapUtils.GetDnFromHostname(), sid, token: token);
}
}
}