//Sample code file: var/ndk/webBuildengine/tmp/viewable_samples/64ef4950-494f-459b-8642-30a48c1a59a8/extensions/getcount.c

/* $Novell: getcount.c,v 1.12 2003/05/12 13:15:35 $ */
/**************************************************************************
*  Novell Software Developer Kit
*
*  Copyright (C) 2002-2003 Novell, Inc. All Rights Reserved.
*
*  THIS WORK IS SUBJECT TO U.S. AND INTERNATIONAL COPYRIGHT LAWS AND TREATIES.
*  USE AND REDISTRIBUTION OF THIS WORK IS SUBJECT TO THE LICENSE AGREEMENT
*  ACCOMPANYING THE SOFTWARE DEVELOPER KIT (SDK) THAT CONTAINS THIS WORK.
*  PURSUANT TO THE SDK LICENSE AGREEMENT, NOVELL HEREBY GRANTS TO DEVELOPER A
*  ROYALTY-FREE, NON-EXCLUSIVE LICENSE TO INCLUDE NOVELL'S SAMPLE CODE IN ITS
*  PRODUCT. NOVELL GRANTS DEVELOPER WORLDWIDE DISTRIBUTION RIGHTS TO MARKET,
*  DISTRIBUTE, OR SELL NOVELL'S SAMPLE CODE AS A COMPONENT OF DEVELOPER'S
*  PRODUCTS. NOVELL SHALL HAVE NO OBLIGATIONS TO DEVELOPER OR DEVELOPER'S
*  CUSTOMERS WITH RESPECT TO THIS CODE.
*
***************************************************************************
   getcount.c
***************************************************************************
   Description: The getcount.c sample counts the number of entries in the 
                specified partition.
***************************************************************************/


#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <ldap.h>
#include <ldapx.h>
#if defined(N_PLAT_NLM) && defined(LIBC)
#include <screen.h>
#endif

static char usage[] =
"\n Usage:   getcount <host name> <port number> <login dn>" 
"\n\t  <password> <partition dn>\n"
"\n Example: getcount acme.com 389 cn=admin,o=acme secret"
"\n\t  ou=sales,o=acme\n";

int main(int argc, char **argv)
{  
    int            rc, version, ldapPort;
    unsigned long  count;    
    char           *ldapHost;         
    char           *loginDN; 
    char           *password; 
    char           *partitionDN;
    char           *msg;
    LDAP           *ld;
    struct timeval timeOut = {10,0};   /* 10 second connection timeout */
   
    #if defined(N_PLAT_NLM) && defined(LIBC)
    setscreenmode(SCR_NO_MODE);              /* Don't clear screen on exit */
    #endif
    
    if (argc != 6)
    {
        printf("%s", usage);
        return (1);
    }
        
         
    ldapHost    =  argv[1];
    ldapPort    =  atoi(argv[2]);
    loginDN     =  argv[3];
    password    =  argv[4];
    partitionDN =  argv[5];

    /* Set LDAP version to 3 and set connection timeout. */
    version = LDAP_VERSION3;
    ldap_set_option( NULL, LDAP_OPT_PROTOCOL_VERSION, &version);
    ldap_set_option( NULL, LDAP_OPT_NETWORK_TIMEOUT, &timeOut);
   
    /* Initialize the LDAP session */
    printf( "    Initializing the LDAP session handle ...\n" );
    if ( (ld = ldap_init( ldapHost, ldapPort )) == NULL )
    {
        perror( "ldap_init" );
        return( 1 );  
    }
    
    /* Authenticate to the server */
    printf( "    Binding to the directory...\n" );
    rc = ldap_simple_bind_s( ld, loginDN, password );
    if ( rc != LDAP_SUCCESS )
    {
        printf( "ldap_simple_bind_s: %s\n", ldap_err2string(rc));
        ldap_unbind_s( ld );
        return( 1 );
    }

    /* 
     * Novell extended operation to get the number of entries 
     * in a partition.
     */
    printf( "    Calling ldap_partition_entry_count...\n" );
    rc = ldap_partition_entry_count(
                               ld,           /* LDAP session handle */
                               partitionDN,  /* dn of the partition root */
                               &count );     /* returned entry count */

    if ( rc != LDAP_SUCCESS )
    {
        /* print out LDAP server message */
        if(ldap_get_option (ld, LDAP_OPT_ERROR_STRING, &msg ) == 
                                                      LDAP_OPT_SUCCESS ) {
            
            if ( (msg != NULL) && (strlen(msg) != 0) )
             {
                printf("\n    LDAP server message: %s", msg);
           ldap_memfree(msg);
             }
        }
        /* print out LDAP error message */
        printf("    LDAP error  message: ldap_partition_entry"
                             "_count: %s (%d)\n", ldap_err2string( rc ), rc);
        
        printf("\n    ldap_partition_entry_count failed\n\n");
        ldap_unbind_s( ld );
        return( 1 );
    }
             
    printf("\n    ldap_partition_entry_count succeeded\n");     
    printf("\n        This partition has %d entries\n\n", count);

    ldap_unbind_s( ld );
    return 0;
}