git.sur5r.net Git - openldap/blob - libraries/liblber/etest.c

   1 /* test.c - lber encoding test program */
   2 /* $OpenLDAP$ */
   3 /*
   4  * Copyright 1998-2002 The OpenLDAP Foundation, All Rights Reserved.
   5  * COPYING RESTRICTIONS APPLY, see COPYRIGHT file
   6  */
   7 /* Portions
   8  * Copyright (c) 1990 Regents of the University of Michigan.
   9  * All rights reserved.
  10  */
  11
  12 #include "portable.h"
  13
  14 #include <stdio.h>
  15
  16 #include <ac/stdlib.h>
  17
  18 #include <ac/socket.h>
  19 #include <ac/string.h>
  20 #include <ac/unistd.h>
  21
  22 #ifdef HAVE_CONSOLE_H
  23 #include <console.h>
  24 #endif /* HAVE_CONSOLE_H */
  25
  26 #include "lber.h"
  27
  28 static void usage( const char *name )
  29 {
  30         fprintf( stderr, "usage: %s fmtstring\n", name );
  31 }
  32
  33 static char* getbuf( void ) {
  34         char *p;
  35         static char buf[128];
  36
  37         if ( fgets( buf, sizeof(buf), stdin ) == NULL )
  38                 return NULL;
  39
  40         if ( (p = strchr( buf, '\n' )) != NULL )
  41                 *p = '\0';
  42
  43         return buf;
  44 }
  45
  46 int
  47 main( int argc, char **argv )
  48 {
  49         char    *s;
  50
  51         int                     fd, rc;
  52         BerElement      *ber;
  53         Sockbuf         *sb;
  54
  55         /* enable debugging */
  56         int ival = -1;
  57         ber_set_option( NULL, LBER_OPT_DEBUG_LEVEL, &ival );
  58
  59         if ( argc < 2 ) {
  60                 usage( argv[0] );
  61                 return( EXIT_FAILURE );
  62         }
  63
  64 #ifdef HAVE_CONSOLE_H
  65         ccommand( &argv );
  66         cshow( stdout );
  67
  68         if (( fd = open( "lber-test", O_WRONLY|O_CREAT|O_TRUNC|O_BINARY ))
  69                 < 0 ) {
  70             perror( "open" );
  71             return( EXIT_FAILURE );
  72         }
  73
  74 #else
  75         fd = fileno(stdout);
  76 #endif
  77
  78         sb = ber_sockbuf_alloc();
  79         ber_sockbuf_add_io( sb, &ber_sockbuf_io_fd, LBER_SBIOD_LEVEL_PROVIDER,
  80                 (void *)&fd );
  81
  82         if( sb == NULL ) {
  83                 perror( "ber_sockbuf_alloc_fd" );
  84                 return( EXIT_FAILURE );
  85         }
  86
  87         if ( (ber = ber_alloc_t( LBER_USE_DER )) == NULL ) {
  88                 perror( "ber_alloc" );
  89                 return( EXIT_FAILURE );
  90         }
  91
  92         fprintf(stderr, "encode: start\n" );
  93         if( ber_printf( ber, "{" /*}*/ ) ) {
  94                 perror( "ber_printf {" /*}*/ );
  95                 return( EXIT_FAILURE );
  96         }
  97
  98         for ( s = argv[1]; *s; s++ ) {
  99                 char *buf;
 100                 char fmt[2];
 101
 102                 fmt[0] = *s;
 103                 fmt[1] = '\0';
 104
 105                 fprintf(stderr, "encode: %s\n", fmt );
 106                 switch ( *s ) {
 107                 case 'i':       /* int */
 108                 case 'b':       /* boolean */
 109                 case 'e':       /* enumeration */
 110                         buf = getbuf();
 111                         rc = ber_printf( ber, fmt, atoi(buf) );
 112                         break;
 113
 114                 case 'n':       /* null */
 115                 case '{':       /* begin sequence */
 116                 case '}':       /* end sequence */
 117                 case '[':       /* begin set */
 118                 case ']':       /* end set */
 119                         rc = ber_printf( ber, fmt );
 120                         break;
 121
 122                 case 'o':       /* octet string (non-null terminated) */
 123                 case 'B':       /* bit string */
 124                         buf = getbuf();
 125                         rc = ber_printf( ber, fmt, buf, strlen(buf) );
 126                         break;
 127
 128                 case 's':       /* string */
 129                 case 't':       /* tag for the next element */
 130                         buf = getbuf();
 131                         rc = ber_printf( ber, fmt, buf );
 132                         break;
 133
 134                 default:
 135                         fprintf( stderr, "encode: unknown fmt %c\n", *fmt );
 136                         rc = -1;
 137                         break;
 138                 }
 139
 140                 if( rc == -1 ) {
 141                         perror( "ber_printf" );
 142                         return( EXIT_FAILURE );
 143                 }
 144         }
 145
 146         fprintf(stderr, "encode: end\n" );
 147         if( ber_printf( ber, /*{*/ "N}" ) == -1 ) {
 148                 perror( /*{*/ "ber_printf }" );
 149                 return( EXIT_FAILURE );
 150         }
 151
 152         if ( ber_flush( sb, ber, 1 ) == -1 ) {
 153                 perror( "ber_flush" );
 154                 return( EXIT_FAILURE );
 155         }
 156
 157         ber_sockbuf_free( sb );
 158         return( EXIT_SUCCESS );
 159 }