2 * Copyright (c) 2003, Index Data
3 * See the file LICENSE for details.
5 * $Id: bertorture.c,v 1.1 2003-03-11 11:06:35 adam Exp $
16 #include <yaz/yaz-util.h>
17 #include <yaz/proto.h>
18 #include <yaz/comstack.h>
20 #define PACKET_SIZE 64
24 static int send_packet(const char *host)
26 char buf[PACKET_SIZE];
31 COMSTACK cs = cs_create_host(host, 1, &add);
36 if (cs_connect(cs, add) < 0)
39 for (i = 0; i<sizeof(buf); i++)
42 buf[i] = rand() & 0xff;
44 cs_put(cs, buf, sizeof(buf));
50 static void test_file(const char *fname)
53 ODR odr = odr_createmem(ODR_DECODE);
54 char buf[PACKET_SIZE];
56 int fd =open(fname, O_RDONLY, 0666);
59 yaz_log(LOG_ERRNO|LOG_FATAL, "open %s", fname);
62 while (off < sizeof(buf))
65 rd = read(fd, buf+off, sizeof(buf)-off);
67 yaz_log(LOG_ERRNO|LOG_FATAL, "read %s", fname);
76 yaz_log(LOG_ERRNO|LOG_FATAL, "close %s", fname);
79 odr_setbuf(odr, buf, off, 0);
80 z_GDU(odr, &req, 0, 0);
85 static void test_random(int run, const char *fname, const char *fname2,
89 char buf[PACKET_SIZE];
94 if (!strcmp(fname2, "-"))
97 dumpfile = fopen(fname2, "w");
100 for (i = 0; i<sizeof(buf); i++)
101 buf[i] = rand() & 0xff;
103 for (j = 0; j<sizeof(buf)-1; j++)
110 odr = odr_createmem(ODR_DECODE);
114 int fd =open(fname, O_TRUNC|O_CREAT|O_WRONLY, 0666);
117 yaz_log(LOG_ERRNO|LOG_FATAL, "open %s", fname);
120 while (sizeof(buf)-j-off > 0)
123 wrote = write(fd, buf+off+j, sizeof(buf)-j-off);
125 yaz_log(LOG_ERRNO|LOG_FATAL, "write %s", fname);
132 yaz_log(LOG_ERRNO|LOG_FATAL, "close %s", fname);
136 mbuf = malloc(sizeof(buf)-j);
137 memcpy(mbuf, buf+j, sizeof(buf)-j);
138 odr_setbuf(odr, mbuf, sizeof(buf)-j, 0);
139 if (z_GDU(odr, &req, 0, 0))
144 odr_geterrorx(odr, &ex);
148 odr_dumpBER(dumpfile, buf+j, sizeof(buf)-j);
154 if (dumpfile && dumpfile != stdout)
158 void sigint_handler(int x)
163 int main(int argc, char **argv)
165 int start = 0, end = 10000000, ret, i, estat[100];
168 char *packet_fname = 0;
170 signal(SIGINT, sigint_handler);
171 signal(SIGTERM, sigint_handler);
172 for (i = 0; i<sizeof(estat)/sizeof(*estat); i++)
175 while ((ret = options("s:e:b:p:", argv, argc, &arg)) != -2)
192 if (!strcmp(arg, "random"))
195 while(!stop && (end == 0 || i < end))
199 printf ("\r[%d]", i); fflush(stdout);
200 test_random(i, packet_fname, ber_fname, estat);
206 fprintf (stderr, "usage\n");
207 fprintf (stderr, " [-s start] [-e end] [-b berdump] [-p packetdump] random\n");
212 for (i = 0; i < sizeof(estat)/sizeof(*estat); i++)
214 printf ("%3d %9d\n", i, estat[i]);