/*
 * Copyright (c) 2001 by Sun Microsystems, Inc.
 * All rights reserved.
 */

#include <stdio.h>
#include <string.h>

#ifdef MACOS
#include "macos.h"
#endif /* MACOS */

#if !defined( MACOS ) && !defined( DOS )
#include <sys/types.h>
#include <sys/socket.h>
#endif

#include "lber.h"
#include "ldap.h"
#include "ldap-int.h"

int ldap_create_page_control(LDAP *ld, unsigned int pagesize, struct berval *cookie, char isCritical, LDAPControl **output)
{
	BerElement *ber;
	int rc;

	if (NULL == ld || NULL == output)
		return (LDAP_PARAM_ERROR);

	if ((ber = ber_alloc_t(LBER_USE_DER)) == NULLBER){
		return (LDAP_NO_MEMORY);
	}

	if (ber_printf(ber, "{io}", pagesize,
			(cookie && cookie->bv_val) ? cookie->bv_val : "",
			(cookie && cookie->bv_val) ? cookie->bv_len : 0)
				 == LBER_ERROR) {
		ber_free(ber, 1);
		return (LDAP_ENCODING_ERROR);
	}

	rc = nsldapi_build_control(LDAP_CONTROL_SIMPLE_PAGE, ber, 1, isCritical,
		output);

	ld->ld_errno = rc;
	return (rc);
}

int ldap_parse_page_control(LDAP *ld, LDAPControl **controls, unsigned int *totalcount, struct berval **cookie)
{
	int i, rc;
	BerElement *theBer;
	LDAPControl *listCtrlp;

	for (i = 0; controls[i] != NULL; i++){
		if (strcmp(controls[i]->ldctl_oid, "1.2.840.113556.1.4.319") == 0) {
			listCtrlp = controls[i];
			if ((theBer = ber_init(&listCtrlp->ldctl_value)) == NULLBER){
				return (LDAP_NO_MEMORY);
			}
			if ((rc = ber_scanf(theBer, "{iO}", totalcount, cookie)) == LBER_ERROR){
				ber_free(theBer, 1);
				return (LDAP_DECODING_ERROR);
			}
			ber_free(theBer, 1);
			return (LDAP_SUCCESS);
		}
	}
	return (LDAP_CONTROL_NOT_FOUND);
}