/* perform a simple search */
static void *worker(void *arg)
{
MYLDAP_SESSION *session;
MYLDAP_SEARCH *search;
MYLDAP_ENTRY *entry;
const char *attrs[] = { "uid", "cn", "gid", NULL };
struct worker_args *args=(struct worker_args *)arg;
int i;
int rc;
/* initialize session */
session=myldap_create_session();
assert(session!=NULL);
/* perform search */
search=myldap_search(session,nslcd_cfg->ldc_bases[0],
LDAP_SCOPE_SUBTREE,
"(objectclass=posixAccount)",
attrs,NULL);
assert(search!=NULL);
/* go over results */
for (i=0;(entry=myldap_get_entry(search,&rc))!=NULL;i++)
{
if (i<MAXRESULTS)
printf("test_myldap: test_threads(): [worker %d] [%d] DN %s\n",args->id,i,myldap_get_dn(entry));
else if (i==MAXRESULTS)
printf("test_myldap: test_threads(): [worker %d] ...\n",args->id);
}
printf("test_myldap: test_threads(): [worker %d] DONE: %s\n",args->id,ldap_err2string(rc));
assert(rc==LDAP_SUCCESS);
/* clean up */
myldap_session_close(session);
return 0;
}
static MYLDAP_SESSION *get_session(const char *binddn, const char *userdn,
const char *password, int *rcp)
{
MYLDAP_SESSION *session;
char buffer[BUFLEN_DN];
/* set up a new connection */
session = myldap_create_session();
if (session == NULL)
{
*rcp = LDAP_UNAVAILABLE;
return NULL;
}
/* set up credentials for the session */
if (myldap_set_credentials(session, binddn, password))
return NULL;
/* perform search for own object (just to do any kind of search to set
up the connection with fail-over) */
if ((lookup_dn2uid(session, userdn, rcp, buffer, sizeof(buffer)) == NULL) ||
(*rcp != LDAP_SUCCESS))
{
myldap_session_close(session);
return NULL;
}
return session;
}
static void test_get_rdnvalues(void)
{
MYLDAP_SESSION *session;
MYLDAP_SEARCH *search;
MYLDAP_ENTRY *entry;
const char *attrs[] = { "cn", "uid", NULL };
int rc;
char buf[80];
/* initialize session */
printf("test_myldap: test_get_rdnvalues(): getting session...\n");
session=myldap_create_session();
assert(session!=NULL);
/* perform search */
printf("test_myldap: test_get_rdnvalues(): doing search...\n");
search=myldap_search(session,"cn=Aka Ashbach+uid=aashbach,ou=lotsofpeople,dc=test,dc=tld",
LDAP_SCOPE_BASE,
"(objectClass=*)",
attrs,NULL);
assert(search!=NULL);
/* get one entry */
entry=myldap_get_entry(search,&rc);
assert(entry!=NULL);
printf("test_myldap: test_get_rdnvalues(): got DN %s\n",myldap_get_dn(entry));
/* get some values from DN */
printf("test_myldap: test_get_rdnvalues(): DN.uid=%s\n",myldap_get_rdn_value(entry,"uid"));
printf("test_myldap: test_get_rdnvalues(): DN.cn=%s\n",myldap_get_rdn_value(entry,"cn"));
printf("test_myldap: test_get_rdnvalues(): DN.uidNumber=%s\n",myldap_get_rdn_value(entry,"uidNumber"));
/* clean up */
myldap_session_close(session);
/* some tests */
printf("test_myldap: test_get_rdnvalues(): DN.uid=%s\n",myldap_cpy_rdn_value("cn=Aka Ashbach+uid=aashbach,ou=lotsofpeople,dc=test,dc=tld","uid",buf,sizeof(buf)));
printf("test_myldap: test_get_rdnvalues(): DN.cn=%s\n",myldap_cpy_rdn_value("cn=Aka Ashbach+uid=aashbach,ou=lotsofpeople,dc=test,dc=tld","cn",buf,sizeof(buf)));
printf("test_myldap: test_get_rdnvalues(): DN.uidNumber=%s\n",myldap_cpy_rdn_value("cn=Aka Ashbach+uid=aashbach,ou=lotsofpeople,dc=test,dc=tld","uidNumber",buf,sizeof(buf)));
}
/* this method tests to see if we can perform two searches within
one session */
static void test_two_searches(void)
{
MYLDAP_SESSION *session;
MYLDAP_SEARCH *search1,*search2;
MYLDAP_ENTRY *entry;
const char *attrs[] = { "uidNumber", "cn", "gidNumber", "uid", "objectClass", NULL };
const char **vals;
/* initialize session */
printf("test_myldap: test_two_searches(): getting session...\n");
session=myldap_create_session();
assert(session!=NULL);
/* perform search1 */
search1=myldap_search(session,nslcd_cfg->ldc_bases[0],
LDAP_SCOPE_SUBTREE,
"(&(objectClass=posixAccount)(uid=*))",
attrs,NULL);
assert(search1!=NULL);
/* get a result from search1 */
entry=myldap_get_entry(search1,NULL);
assert(entry!=NULL);
printf("test_myldap: test_two_searches(): [search1] DN %s\n",myldap_get_dn(entry));
vals=myldap_get_values(entry,"cn");
assert((vals!=NULL)&&(vals[0]!=NULL));
printf("test_myldap: test_two_searches(): [search1] cn=%s\n",vals[0]);
/* start a second search */
search2=myldap_search(session,nslcd_cfg->ldc_bases[0],
LDAP_SCOPE_SUBTREE,
"(&(objectclass=posixGroup)(gidNumber=*))",
attrs,NULL);
assert(search2!=NULL);
/* get a result from search2 */
entry=myldap_get_entry(search2,NULL);
assert(entry!=NULL);
printf("test_myldap: test_two_searches(): [search2] DN %s\n",myldap_get_dn(entry));
vals=myldap_get_values(entry,"cn");
assert((vals!=NULL)&&(vals[0]!=NULL));
printf("test_myldap: test_two_searches(): [search2] cn=%s\n",vals[0]);
/* get another result from search1 */
entry=myldap_get_entry(search1,NULL);
assert(entry!=NULL);
printf("test_myldap: test_two_searches(): [search1] DN %s\n",myldap_get_dn(entry));
vals=myldap_get_values(entry,"cn");
assert((vals!=NULL)&&(vals[0]!=NULL));
printf("test_myldap: test_two_searches(): [search1] cn=%s\n",vals[0]);
/* stop search1 */
myldap_search_close(search1);
/* get another result from search2 */
entry=myldap_get_entry(search2,NULL);
assert(entry!=NULL);
printf("test_myldap: test_two_searches(): [search2] DN %s\n",myldap_get_dn(entry));
vals=myldap_get_values(entry,"cn");
assert((vals!=NULL)&&(vals[0]!=NULL));
printf("test_myldap: test_two_searches(): [search2] cn=%s\n",vals[0]);
/* clean up */
myldap_session_close(session);
}
/* This search prints a number of attributes from a search */
static void test_get_values(void)
{
MYLDAP_SESSION *session;
MYLDAP_SEARCH *search;
MYLDAP_ENTRY *entry;
const char *attrs[] = { "uidNumber", "cn", "gidNumber", "uid", "objectClass", NULL };
const char **vals;
const char *rdnval;
int i;
/* initialize session */
printf("test_myldap: test_get_values(): getting session...\n");
session=myldap_create_session();
assert(session!=NULL);
/* perform search */
search=myldap_search(session,nslcd_cfg->ldc_bases[0],
LDAP_SCOPE_SUBTREE,
"(&(objectClass=posixAccount)(uid=*))",
attrs,NULL);
assert(search!=NULL);
/* go over results */
for (i=0;(entry=myldap_get_entry(search,NULL))!=NULL;i++)
{
if (i<MAXRESULTS)
printf("test_myldap: test_get_values(): [%d] DN %s\n",i,myldap_get_dn(entry));
else if (i==MAXRESULTS)
printf("test_myldap: test_get_values(): ...\n");
/* try to get uid from attribute */
vals=myldap_get_values(entry,"uidNumber");
assert((vals!=NULL)&&(vals[0]!=NULL));
if (i<MAXRESULTS)
printf("test_myldap: test_get_values(): [%d] uidNumber=%s\n",i,vals[0]);
/* try to get gid from attribute */
vals=myldap_get_values(entry,"gidNumber");
assert((vals!=NULL)&&(vals[0]!=NULL));
if (i<MAXRESULTS)
printf("test_myldap: test_get_values(): [%d] gidNumber=%s\n",i,vals[0]);
/* write LDF_STRING(PASSWD_NAME) */
vals=myldap_get_values(entry,"uid");
assert((vals!=NULL)&&(vals[0]!=NULL));
if (i<MAXRESULTS)
printf("test_myldap: test_get_values(): [%d] uid=%s\n",i,vals[0]);
/* get rdn values */
rdnval=myldap_get_rdn_value(entry,"cn");
if (i<MAXRESULTS)
printf("test_myldap: test_get_values(): [%d] cdrdn=%s\n",i,rdnval==NULL?"NULL":rdnval);
rdnval=myldap_get_rdn_value(entry,"uid");
if (i<MAXRESULTS)
printf("test_myldap: test_get_values(): [%d] uidrdn=%s\n",i,rdnval==NULL?"NULL":rdnval);
/* check objectclass */
assert(myldap_has_objectclass(entry,"posixAccount"));
}
/* clean up */
myldap_session_close(session);
}
/* This is a very basic search test, it performs a test to get certain
entries from the database. It currently just prints out the DNs for
the entries. */
static void test_search(void)
{
MYLDAP_SESSION *session;
MYLDAP_SEARCH *search;
MYLDAP_ENTRY *entry;
const char *attrs[] = { "uid", "cn", "gid", NULL };
int i;
int rc;
/* initialize session */
printf("test_myldap: test_search(): getting session...\n");
session=myldap_create_session();
assert(session!=NULL);
/* perform search */
printf("test_myldap: test_search(): doing search...\n");
search=myldap_search(session,nslcd_cfg->ldc_bases[0],
LDAP_SCOPE_SUBTREE,
"(objectclass=posixAccount)",
attrs,NULL);
assert(search!=NULL);
/* go over results */
printf("test_myldap: test_search(): get results...\n");
for (i=0;(entry=myldap_get_entry(search,&rc))!=NULL;i++)
{
if (i<MAXRESULTS)
printf("test_myldap: test_search(): [%d] DN %s\n",i,myldap_get_dn(entry));
else if (i==MAXRESULTS)
printf("test_myldap: test_search(): ...\n");
}
printf("test_myldap: test_search(): %d entries returned: %s\n",i,ldap_err2string(rc));
assert(rc==LDAP_SUCCESS);
/* perform another search */
printf("test_myldap: test_search(): doing search...\n");
search=myldap_search(session,nslcd_cfg->ldc_bases[0],
LDAP_SCOPE_SUBTREE,
"(objectclass=posixGroup)",
attrs,NULL);
assert(search!=NULL);
/* go over results */
printf("test_myldap: test_search(): get results...\n");
for (i=0;(entry=myldap_get_entry(search,&rc))!=NULL;i++)
{
if (i<MAXRESULTS)
printf("test_myldap: test_search(): [%d] DN %s\n",i,myldap_get_dn(entry));
else if (i==MAXRESULTS)
printf("test_myldap: test_search(): ...\n");
}
printf("test_myldap: test_search(): %d entries returned: %s\n",i,ldap_err2string(rc));
assert(rc==LDAP_SUCCESS);
/* clean up */
myldap_session_close(session);
}
static void test_get(void)
{
MYLDAP_SESSION *session;
MYLDAP_SEARCH *search1,*search2;
MYLDAP_ENTRY *entry;
const char *attrs1[] = { "cn", "userPassword", "memberUid", "gidNumber", "member", NULL };
const char *attrs2[] = { "uid", NULL };
int rc;
/* initialize session */
printf("test_myldap: test_get(): getting session...\n");
session=myldap_create_session();
assert(session!=NULL);
/* perform search */
printf("test_myldap: test_get(): doing search...\n");
search1=myldap_search(session,nslcd_cfg->ldc_bases[0],
LDAP_SCOPE_SUBTREE,
"(&(|(objectClass=posixGroup)(objectClass=groupOfNames))(cn=testgroup2))",
attrs1,NULL);
assert(search1!=NULL);
/* get one entry */
entry=myldap_get_entry(search1,&rc);
assert(entry!=NULL);
printf("test_myldap: test_get(): got DN %s\n",myldap_get_dn(entry));
/* get some attribute values */
(void)myldap_get_values(entry,"gidNumber");
(void)myldap_get_values(entry,"userPassword");
(void)myldap_get_values(entry,"memberUid");
(void)myldap_get_values(entry,"member");
/* perform another search */
printf("test_myldap: test_get(): doing get...\n");
search2=myldap_search(session,"cn=Test User2,ou=people,dc=test,dc=tld",
LDAP_SCOPE_BASE,
"(objectclass=posixAccount)",
attrs2,NULL);
assert(search2!=NULL);
/* get one entry */
entry=myldap_get_entry(search2,&rc);
assert(entry!=NULL);
printf("test_myldap: test_get(): got DN %s\n",myldap_get_dn(entry));
/* test if searches are ok */
assert(myldap_get_entry(search1,&rc)==NULL);
assert(myldap_get_entry(search2,&rc)==NULL);
/* clean up */
myldap_session_close(session);
}
static void test_connections(void)
{
MYLDAP_SESSION *session;
MYLDAP_SEARCH *search;
const char *attrs[] = { "uid", "cn", "gid", NULL };
char *old_uris[NSS_LDAP_CONFIG_URI_MAX+1];
int i;
/* save the old URIs */
for (i=0;i<(NSS_LDAP_CONFIG_URI_MAX+1);i++)
{
old_uris[i]=nslcd_cfg->ldc_uris[i].uri;
nslcd_cfg->ldc_uris[i].uri=NULL;
}
/* set new URIs */
i=0;
nslcd_cfg->ldc_uris[i++].uri="ldapi://%2fdev%2fnull/";
nslcd_cfg->ldc_uris[i++].uri="ldap://10.10.10.10/";
nslcd_cfg->ldc_uris[i++].uri="ldapi://%2fdev%2fnonexistent/";
nslcd_cfg->ldc_uris[i++].uri="ldap://nosuchhost/";
nslcd_cfg->ldc_uris[i++].uri=NULL;
/* initialize session */
printf("test_myldap: test_connections(): getting session...\n");
session=myldap_create_session();
assert(session!=NULL);
/* perform search */
printf("test_myldap: test_connections(): doing search...\n");
search=myldap_search(session,nslcd_cfg->ldc_bases[0],
LDAP_SCOPE_SUBTREE,
"(objectclass=posixAccount)",
attrs,NULL);
assert(search==NULL);
/* clean up */
myldap_session_close(session);
/* restore the old URIs */
for (i=0;i<(NSS_LDAP_CONFIG_URI_MAX+1);i++)
nslcd_cfg->ldc_uris[i].uri=old_uris[i];
}
static void *worker(void UNUSED(*arg))
{
MYLDAP_SESSION *session;
int csock;
int j;
struct sockaddr_storage addr;
socklen_t alen;
fd_set fds;
struct timeval tv;
/* create a new LDAP session */
session = myldap_create_session();
/* clean up the session if we're done */
pthread_cleanup_push(worker_cleanup, session);
/* start waiting for incoming connections */
while (1)
{
/* time out connection to LDAP server if needed */
myldap_session_check(session);
/* set up the set of fds to wait on */
FD_ZERO(&fds);
FD_SET(nslcd_serversocket, &fds);
/* set up our timeout value */
tv.tv_sec = nslcd_cfg->idle_timelimit;
tv.tv_usec = 0;
/* wait for a new connection */
j = select(nslcd_serversocket + 1, &fds, NULL, NULL,
nslcd_cfg->idle_timelimit > 0 ? &tv : NULL);
/* check result of select() */
if (j < 0)
{
if (errno == EINTR)
log_log(LOG_DEBUG, "select() failed (ignored): %s", strerror(errno));
else
log_log(LOG_ERR, "select() failed: %s", strerror(errno));
continue;
}
/* see if our file descriptor is actually ready */
if (!FD_ISSET(nslcd_serversocket, &fds))
continue;
/* wait for a new connection */
alen = (socklen_t)sizeof(struct sockaddr_storage);
csock = accept(nslcd_serversocket, (struct sockaddr *)&addr, &alen);
if (csock < 0)
{
if ((errno == EINTR) || (errno == EAGAIN) || (errno == EWOULDBLOCK))
log_log(LOG_DEBUG, "accept() failed (ignored): %s", strerror(errno));
else
log_log(LOG_ERR, "accept() failed: %s", strerror(errno));
continue;
}
/* make sure O_NONBLOCK is not inherited */
if ((j = fcntl(csock, F_GETFL, 0)) < 0)
{
log_log(LOG_ERR, "fctnl(F_GETFL) failed: %s", strerror(errno));
if (close(csock))
log_log(LOG_WARNING, "problem closing socket: %s", strerror(errno));
continue;
}
if (fcntl(csock, F_SETFL, j & ~O_NONBLOCK) < 0)
{
log_log(LOG_ERR, "fctnl(F_SETFL,~O_NONBLOCK) failed: %s", strerror(errno));
if (close(csock))
log_log(LOG_WARNING, "problem closing socket: %s", strerror(errno));
continue;
}
/* indicate new connection to logging module (generates unique id) */
log_newsession();
/* handle the connection */
handleconnection(csock, session);
/* indicate end of session in log messages */
log_clearsession();
}
pthread_cleanup_pop(1);
return NULL;
}
