dlltest.c

   1 /*
   2    minibz2
   3       libbz2.dll test program.
   4       by Yoshioka Tsuneo (tsuneo@rr.iij4u.or.jp)
   5       This file is Public Domain.  Welcome any email to me.
   6
   7    usage: minibz2 [-d] [-{1,2,..9}] [[srcfilename] destfilename]
   8 */
   9
  10 #define BZ_IMPORT
  11 #include <stdio.h>
  12 #include <stdlib.h>
  13 #include "bzlib.h"
  14 #ifdef _WIN32
  15 #include <io.h>
  16 #endif
  17
  18
  19 #ifdef _WIN32
  20
  21 #define BZ2_LIBNAME "libbz2-1.0.2.DLL"
  22
  23 #include <windows.h>
  24 static int BZ2DLLLoaded = 0;
  25 static HINSTANCE BZ2DLLhLib;
  26 int BZ2DLLLoadLibrary(void)
  27 {
  28    HINSTANCE hLib;
  29
  30    if(BZ2DLLLoaded==1){return 0;}
  31    hLib=LoadLibrary(BZ2_LIBNAME);
  32    if(hLib == NULL){
  33       fprintf(stderr,"Can't load %s\n",BZ2_LIBNAME);
  34       return -1;
  35    }
  36    BZ2_bzlibVersion=GetProcAddress(hLib,"BZ2_bzlibVersion");
  37    BZ2_bzopen=GetProcAddress(hLib,"BZ2_bzopen");
  38    BZ2_bzdopen=GetProcAddress(hLib,"BZ2_bzdopen");
  39    BZ2_bzread=GetProcAddress(hLib,"BZ2_bzread");
  40    BZ2_bzwrite=GetProcAddress(hLib,"BZ2_bzwrite");
  41    BZ2_bzflush=GetProcAddress(hLib,"BZ2_bzflush");
  42    BZ2_bzclose=GetProcAddress(hLib,"BZ2_bzclose");
  43    BZ2_bzerror=GetProcAddress(hLib,"BZ2_bzerror");
  44
  45    if (!BZ2_bzlibVersion || !BZ2_bzopen || !BZ2_bzdopen
  46        || !BZ2_bzread || !BZ2_bzwrite || !BZ2_bzflush
  47        || !BZ2_bzclose || !BZ2_bzerror) {
  48       fprintf(stderr,"GetProcAddress failed.\n");
  49       return -1;
  50    }
  51    BZ2DLLLoaded=1;
  52    BZ2DLLhLib=hLib;
  53    return 0;
  54
  55 }
  56 int BZ2DLLFreeLibrary(void)
  57 {
  58    if(BZ2DLLLoaded==0){return 0;}
  59    FreeLibrary(BZ2DLLhLib);
  60    BZ2DLLLoaded=0;
  61 }
  62 #endif /* WIN32 */
  63
  64 void usage(void)
  65 {
  66    puts("usage: minibz2 [-d] [-{1,2,..9}] [[srcfilename] destfilename]");
  67 }
  68
  69 int main(int argc,char *argv[])
  70 {
  71    int decompress = 0;
  72    int level = 9;
  73    char *fn_r = NULL;
  74    char *fn_w = NULL;
  75
  76 #ifdef _WIN32
  77    if(BZ2DLLLoadLibrary()<0){
  78       fprintf(stderr,"Loading of %s failed.  Giving up.\n", BZ2_LIBNAME);
  79       exit(1);
  80    }
  81    printf("Loading of %s succeeded.  Library version is %s.\n",
  82           BZ2_LIBNAME, BZ2_bzlibVersion() );
  83 #endif
  84    while(++argv,--argc){
  85       if(**argv =='-' || **argv=='/'){
  86          char *p;
  87
  88          for(p=*argv+1;*p;p++){
  89             if(*p=='d'){
  90                decompress = 1;
  91             }else if('1'<=*p && *p<='9'){
  92                level = *p - '0';
  93             }else{
  94                usage();
  95                exit(1);
  96             }
  97          }
  98       }else{
  99          break;
 100       }
 101    }
 102    if(argc>=1){
 103       fn_r = *argv;
 104       argc--;argv++;
 105    }else{
 106       fn_r = NULL;
 107    }
 108    if(argc>=1){
 109       fn_w = *argv;
 110       argc--;argv++;
 111    }else{
 112       fn_w = NULL;
 113    }
 114    {
 115       int len;
 116       char buff[0x1000];
 117       char mode[10];
 118
 119       if(decompress){
 120          BZFILE *BZ2fp_r = NULL;
 121          FILE *fp_w = NULL;
 122
 123          if(fn_w){
 124             if((fp_w = fopen(fn_w,"wb"))==NULL){
 125                printf("can't open [%s]\n",fn_w);
 126                perror("reason:");
 127                exit(1);
 128             }
 129          }else{
 130             fp_w = stdout;
 131          }
 132          if((fn_r == NULL && (BZ2fp_r = BZ2_bzdopen(fileno(stdin),"rb"))==NULL)
 133             || (fn_r != NULL && (BZ2fp_r = BZ2_bzopen(fn_r,"rb"))==NULL)){
 134             printf("can't bz2openstream\n");
 135             exit(1);
 136          }
 137          while((len=BZ2_bzread(BZ2fp_r,buff,0x1000))>0){
 138             fwrite(buff,1,len,fp_w);
 139          }
 140          BZ2_bzclose(BZ2fp_r);
 141          if(fp_w != stdout) fclose(fp_w);
 142       }else{
 143          BZFILE *BZ2fp_w = NULL;
 144          FILE *fp_r = NULL;
 145
 146          if(fn_r){
 147             if((fp_r = fopen(fn_r,"rb"))==NULL){
 148                printf("can't open [%s]\n",fn_r);
 149                perror("reason:");
 150                exit(1);
 151             }
 152          }else{
 153             fp_r = stdin;
 154          }
 155          mode[0]='w';
 156          mode[1] = '0' + level;
 157          mode[2] = '\0';
 158
 159          if((fn_w == NULL && (BZ2fp_w = BZ2_bzdopen(fileno(stdout),mode))==NULL)
 160             || (fn_w !=NULL && (BZ2fp_w = BZ2_bzopen(fn_w,mode))==NULL)){
 161             printf("can't bz2openstream\n");
 162             exit(1);
 163          }
 164          while((len=fread(buff,1,0x1000,fp_r))>0){
 165             BZ2_bzwrite(BZ2fp_w,buff,len);
 166          }
 167          BZ2_bzclose(BZ2fp_w);
 168          if(fp_r!=stdin)fclose(fp_r);
 169       }
 170    }
 171 #ifdef _WIN32
 172    BZ2DLLFreeLibrary();
 173 #endif
 174    return 0;
 175 }