1 /* $NetBSD: etest.c,v 1.1.1.3 2010/12/12 15:21:28 adam Exp $ */ 2 3 /* etest.c - lber encoding test program */ 4 /* OpenLDAP: pkg/ldap/libraries/liblber/etest.c,v 1.35.2.6 2010/04/13 20:22:54 kurt Exp */ 5 /* This work is part of OpenLDAP Software <http://www.openldap.org/>. 6 * 7 * Copyright 1998-2010 The OpenLDAP Foundation. 8 * All rights reserved. 9 * 10 * Redistribution and use in source and binary forms, with or without 11 * modification, are permitted only as authorized by the OpenLDAP 12 * Public License. 13 * 14 * A copy of this license is available in the file LICENSE in the 15 * top-level directory of the distribution or, alternatively, at 16 * <http://www.OpenLDAP.org/license.html>. 17 */ 18 /* Portions Copyright (c) 1990 Regents of the University of Michigan. 19 * All rights reserved. 20 * 21 * Redistribution and use in source and binary forms are permitted 22 * provided that this notice is preserved and that due credit is given 23 * to the University of Michigan at Ann Arbor. The name of the University 24 * may not be used to endorse or promote products derived from this 25 * software without specific prior written permission. This software 26 * is provided ``as is'' without express or implied warranty. 27 */ 28 /* ACKNOWLEDGEMENTS: 29 * This work was originally developed by the University of Michigan 30 * (as part of U-MICH LDAP). 31 */ 32 33 #include "portable.h" 34 35 #include <stdio.h> 36 37 #include <ac/stdlib.h> 38 39 #include <ac/socket.h> 40 #include <ac/string.h> 41 #include <ac/unistd.h> 42 43 #ifdef HAVE_CONSOLE_H 44 #include <console.h> 45 #endif /* HAVE_CONSOLE_H */ 46 47 #include "lber.h" 48 49 static void usage( const char *name ) 50 { 51 fprintf( stderr, "usage: %s fmtstring\n", name ); 52 } 53 54 static char* getbuf( void ) { 55 char *p; 56 static char buf[1024]; 57 58 if ( fgets( buf, sizeof(buf), stdin ) == NULL ) return NULL; 59 60 if ( (p = strchr( buf, '\n' )) != NULL ) *p = '\0'; 61 62 return buf; 63 } 64 65 int 66 main( int argc, char **argv ) 67 { 68 char *s; 69 int tag; 70 71 int fd, rc; 72 BerElement *ber; 73 Sockbuf *sb; 74 75 /* enable debugging */ 76 int ival = -1; 77 ber_set_option( NULL, LBER_OPT_DEBUG_LEVEL, &ival ); 78 79 if ( argc < 2 ) { 80 usage( argv[0] ); 81 return( EXIT_FAILURE ); 82 } 83 84 #ifdef HAVE_CONSOLE_H 85 ccommand( &argv ); 86 cshow( stdout ); 87 88 if (( fd = open( "lber-test", O_WRONLY|O_CREAT|O_TRUNC|O_BINARY )) 89 < 0 ) { 90 perror( "open" ); 91 return( EXIT_FAILURE ); 92 } 93 94 #else 95 fd = fileno(stdout); 96 #endif 97 98 sb = ber_sockbuf_alloc(); 99 ber_sockbuf_add_io( sb, &ber_sockbuf_io_fd, LBER_SBIOD_LEVEL_PROVIDER, 100 (void *)&fd ); 101 102 if( sb == NULL ) { 103 perror( "ber_sockbuf_alloc_fd" ); 104 return( EXIT_FAILURE ); 105 } 106 107 if ( (ber = ber_alloc_t( LBER_USE_DER )) == NULL ) { 108 perror( "ber_alloc" ); 109 return( EXIT_FAILURE ); 110 } 111 112 fprintf(stderr, "encode: start\n" ); 113 if( ber_printf( ber, "{" /*}*/ ) ) { 114 perror( "ber_printf {" /*}*/ ); 115 return( EXIT_FAILURE ); 116 } 117 118 for ( s = argv[1]; *s; s++ ) { 119 char *buf; 120 char fmt[2]; 121 122 fmt[0] = *s; 123 fmt[1] = '\0'; 124 125 fprintf(stderr, "encode: %s\n", fmt ); 126 switch ( *s ) { 127 case 'i': /* int */ 128 case 'b': /* boolean */ 129 case 'e': /* enumeration */ 130 buf = getbuf(); 131 rc = ber_printf( ber, fmt, atoi(buf) ); 132 break; 133 134 case 'n': /* null */ 135 case '{': /* begin sequence */ 136 case '}': /* end sequence */ 137 case '[': /* begin set */ 138 case ']': /* end set */ 139 rc = ber_printf( ber, fmt ); 140 break; 141 142 case 'o': /* octet string (non-null terminated) */ 143 case 'B': /* bit string */ 144 buf = getbuf(); 145 rc = ber_printf( ber, fmt, buf, strlen(buf) ); 146 break; 147 148 case 's': /* string */ 149 buf = getbuf(); 150 rc = ber_printf( ber, fmt, buf ); 151 break; 152 case 't': /* tag for the next element */ 153 buf = getbuf(); 154 tag = atoi(buf); 155 rc = ber_printf( ber, fmt, tag ); 156 break; 157 158 default: 159 fprintf( stderr, "encode: unknown fmt %c\n", *fmt ); 160 rc = -1; 161 break; 162 } 163 164 if( rc == -1 ) { 165 perror( "ber_printf" ); 166 return( EXIT_FAILURE ); 167 } 168 } 169 170 fprintf(stderr, "encode: end\n" ); 171 if( ber_printf( ber, /*{*/ "N}" ) == -1 ) { 172 perror( /*{*/ "ber_printf }" ); 173 return( EXIT_FAILURE ); 174 } 175 176 if ( ber_flush2( sb, ber, LBER_FLUSH_FREE_ALWAYS ) == -1 ) { 177 perror( "ber_flush2" ); 178 return( EXIT_FAILURE ); 179 } 180 181 ber_sockbuf_free( sb ); 182 return( EXIT_SUCCESS ); 183 } 184