test_zlib.rar_The Program_linux zlib

/* * Copyright (C) 2008 The Android Open Source Project * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions * are met: * * Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * * Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in * the documentation and/or other materials provided with the * distribution. * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF * SUCH DAMAGE. */ /* this small program is used to measure the performance of zlib's inflate * algorithm... */ /* most code lifted from the public-domain http://www.zlib.net/zpipe.c */ #include <zlib.h> #include <time.h> #include <stdio.h> #include <stdlib.h> #include <string.h> #include <errno.h> #include <unistd.h> #include <sys/time.h> #define CHUNK 32768 int def(FILE *source, FILE *dest, int level) { int ret, flush; unsigned have; z_stream strm; unsigned char in[CHUNK]; unsigned char out[CHUNK]; /* allocate deflate state */ strm.zalloc = Z_NULL; strm.zfree = Z_NULL; strm.opaque = Z_NULL; ret = deflateInit(&strm, level); if (ret != Z_OK) return ret; /* compress until end of file */ do { strm.avail_in = fread(in, 1, CHUNK, source); if (ferror(source)) { (void)deflateEnd(&strm); return Z_ERRNO; } flush = feof(source) ? Z_FINISH : Z_NO_FLUSH; strm.next_in = in; /* run deflate() on input until output buffer not full, finish compression if all of source has been read in */ do { strm.avail_out = CHUNK; strm.next_out = out; ret = deflate(&strm, flush); /* no bad return value */ have = CHUNK - strm.avail_out; if (fwrite(out, 1, have, dest) != have || ferror(dest)) { (void)deflateEnd(&strm); return Z_ERRNO; } } while (strm.avail_out == 0); /* done when last data in file processed */ } while (flush != Z_FINISH); /* clean up and return */ (void)deflateEnd(&strm); return Z_OK; } int inf(FILE *source) { int ret; unsigned have; z_stream strm; static unsigned char in[CHUNK]; static unsigned char out[CHUNK]; /* allocate inflate state */ strm.zalloc = Z_NULL; strm.zfree = Z_NULL; strm.opaque = Z_NULL; strm.avail_in = 0; strm.next_in = Z_NULL; ret = inflateInit(&strm); if (ret != Z_OK) return ret; /* decompress until deflate stream ends or end of file */ do { strm.avail_in = fread(in, 1, CHUNK, source); if (ferror(source)) { (void)inflateEnd(&strm); return Z_ERRNO; } if (strm.avail_in == 0) break; strm.next_in = in; /* run inflate() on input until output buffer not full */ do { strm.avail_out = CHUNK; strm.next_out = out; ret = inflate(&strm, Z_NO_FLUSH); switch (ret) { case Z_NEED_DICT: ret = Z_DATA_ERROR; /* and fall through */ case Z_DATA_ERROR: case Z_MEM_ERROR: (void)inflateEnd(&strm); return ret; } } while (strm.avail_out == 0); /* done when inflate() says it's done */ } while (ret != Z_STREAM_END); /* clean up and return */ (void)inflateEnd(&strm); return ret == Z_STREAM_END ? Z_OK : Z_DATA_ERROR; } #define DEFAULT_REPEAT 10 #define DEFAULT_LEVEL 9 static void usage(void) { fprintf(stderr, "usage: test_zlib [options] filename [filename2 ...]\n" ); fprintf(stderr, "options: -r NN repeat count (default %d)\n", DEFAULT_REPEAT ); fprintf(stderr, " -N set compression level (default %d)\n", DEFAULT_LEVEL ); exit(1); } static double get_time_usec( void ) { #ifdef HAVE_ANDROID_OS struct timespec ts; if ( clock_gettime( CLOCK_MONOTONIC, &ts ) < 0 ) fprintf(stderr, "clock_gettime: %s\n", strerror(errno) ); return ts.tv_sec*1e6 + ts.tv_nsec*1e-3; #else struct timeval tv; if (gettimeofday( &tv, NULL ) < 0) fprintf(stderr, "gettimeofday: %s\n", strerror(errno) ); return tv.tv_sec*1000000. + tv.tv_usec*1.0; #endif } int main( int argc, char** argv ) { FILE* f; char tempfile[256]; int repeat_count = DEFAULT_REPEAT; int compression_level = DEFAULT_LEVEL; double usec0, usec1; if (argc < 2) usage(); for ( ; argc > 1 && argv[1][0] == '-'; argc--, argv++) { const char* arg = &argv[1][1]; switch (arg[0]) { case 'r': if (arg[1] == 0) { if (argc < 3) usage(); arg = argv[2]; argc--; argv++; } else arg += 1; repeat_count = strtol(arg, NULL, 10); if (repeat_count <= 0) repeat_count = 1; break; case '0': case '1': case '2': case '3': case '4': case '5': case '6': case '7': case '8': case '9': compression_level = arg[0] - '0'; break; default: usage(); } } sprintf(tempfile, "/tmp/ztest.%d", getpid() ); for ( ; argc > 1; argc--, argv++ ) { /* first, compress the file into a temporary storage */ FILE* f = fopen(argv[1], "rb"); FILE* out = NULL; long fsize; int ret, rr; if (f == NULL) { fprintf(stderr, "could not open '%s': %s\n", argv[1], strerror(errno) ); continue; } printf( "testing %s\n", argv[1] ); fseek( f, 0, SEEK_END ); fsize = ftell(f); fseek( f, 0, SEEK_SET ); out = fopen( tempfile, "wb" ); if (out == NULL) { fprintf(stderr, "could not create '%s': %s\n", tempfile, strerror(errno)); fclose(f); continue; } usec0 = get_time_usec(); ret = def( f, out, compression_level ); usec1 = get_time_usec() - usec0; printf( "compression took: %10.3f ms (%.2f KB/s)\n", usec1/1e3, fsize*(1e6/1024)/usec1 ); fclose( out ); fclose(f); usec0 = get_time_usec(); f = fopen( tempfile, "rb" ); for ( rr = repeat_count; rr > 0; rr -- ) { fseek( f, 0, SEEK_SET ); inf(f); } fclose( f ); usec1 = get_time_usec() - usec0; printf( "decompression took: %10.3f ms (%.2f KB/s, %d passes)\n", usec1/1e3, fsize*(1e6/1024)*repeat_count/usec1, repeat_count ); } unlink(tempfile); return 0; }