]> git.sur5r.net Git - openldap/blob - libraries/liblber/etest.c
More contrib cleanout
[openldap] / libraries / liblber / etest.c
1 /* test.c - lber encoding test program */
2 /* $OpenLDAP$ */
3 /*
4  * Copyright 1998-2000 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( char *name )
29 {
30         fprintf( stderr, "usage: %s fmtstring\n", name );
31 }
32
33 static char* getbuf() {
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 }