/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*-
 *
 * The contents of this file are subject to the Netscape Public License
 * Version 1.0 (the "NPL"); you may not use this file except in
 * compliance with the NPL.  You may obtain a copy of the NPL at
 * http://www.mozilla.org/NPL/
 *
 * Software distributed under the NPL is distributed on an "AS IS" basis,
 * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the NPL
 * for the specific language governing rights and limitations under the
 * NPL.
 *
 * The Initial Developer of this code under the NPL is Netscape
 * Communications Corporation.  Portions created by Netscape are
 * Copyright (C) 1998 Netscape Communications Corporation.  All Rights
 * Reserved.
 */

/*
 * Copyright (c) 1990 Regents of the University of Michigan.
 * All rights reserved.
 */

/* test.c - lber encoding test program */

#include <stdio.h>
#include <string.h>
#ifdef MACOS
#include <stdlib.h>
#include <unix.h>
#include <fcntl.h>
#include <console.h>
#else /* MACOS */
#include <sys/types.h>
#ifdef _WIN32
#include <windows.h>
#else
#include <sys/socket.h>
#endif /* _WIN32 */
#endif /* MACOS */
#include "lber.h"

int
SSL_Recv( int s, char *b, unsigned l, int dummy )
{
	return( read( s, b, l ) );
}

SSL_Send( int s, char *b, unsigned l, int dummy )
{
	return( write( s, b, l ) );
}

int
getline( char *prompt, char c, char *buf, int bsize )
{
	char	*p;

	if ( prompt != NULL ) {
		fprintf( stderr, "%s: ", prompt );
	} else {
		fprintf( stderr, "enter value for '%c': ", c );
	}
	if ( fgets( buf, bsize, stdin ) == NULL ) {
		return( -1 );
	}
	if ( (p = strchr( buf, '\n' )) != NULL ) {
		*p = '\0';
	}

	return( 0 );
}


static void usage( char *name )
{
	fprintf( stderr, "usage: %s fmtstring\n", name );
}

main( int argc, char **argv )
{
	int		rc, fd;
	char		*s, *p;
	void		*arg1, *arg2;
	Sockbuf		*sb;
	BerElement	*ber;
	char		fmt[2];
	char		buf[BUFSIZ];
	extern int	lber_debug;

	lber_debug = 255;
	if ( argc < 2 ) {
		usage( argv[0] );
		exit( 1 );
	}

	sb = ber_sockbuf_alloc();
	fd = 1;
	ber_sockbuf_set_option( sb, LBER_SOCKBUF_OPT_DESC, &fd );

	if ( (ber = der_alloc()) == NULL ) {
		perror( "ber_alloc" );
		exit( 1 );
	}

	rc = 0;
	fmt[1] = '\0';
	for ( s = argv[1]; *s; s++ ) {
		switch ( *s ) {
		case 'i':	/* int */
		case 'b':	/* boolean */
		case 'e':	/* enumeration */
			getline( NULL, *s, buf, sizeof(buf) );
			arg1 = (void *) atoi( buf );
			break;

		case 'n':	/* null */
			arg1 = NULL;
			break;

		case 'o':	/* octet string (non-null terminated) */
			getline( NULL, *s, buf, sizeof(buf) );
			arg1 = (void *) buf;
			arg2 = (void *) strlen( buf );
			break;

		case 's':	/* string */
			getline( NULL, *s, buf, sizeof(buf) );
			arg1 = (void *) buf;
			break;

		case 'B':	/* bit string */
			getline( NULL, *s, buf, sizeof(buf) );
			arg1 = (void *) buf;
			arg2 = (void *) strlen( buf );
			break;

		case 't':	/* tag for the next element */
			getline( NULL, *s, buf, sizeof(buf) );
			arg1 = (void *) buf;
			break;

		case '{':	/* begin sequence */
		case '}':	/* end sequence */
		case '[':	/* begin set */
		case ']':	/* end set */
			break;

		default:
			fprintf( stderr, "unknown fmt %c\n", *s );
			rc = -1;
			break;
		}

		fmt[0] = *s;
		if ( ber_printf( ber, fmt, arg1, arg2 ) == -1 ) {
			fprintf( stderr, "ber_printf\n" );
			exit( 1 );
		}
	}

	if ( ber_flush( sb, ber, 1 ) != 0 ) {
		perror( "ber_flush" );
		rc = -1;
	}

	return( rc );
}
