home *** CD-ROM | disk | FTP | other *** search
/ Chip 2001 January / Chip_2001-01_cd1.bin / tema / mysql / mysql-3.23.28g-win-source.exe / mysys / my_compress.c < prev    next >
C/C++ Source or Header  |  2000-08-31  |  3KB  |  88 lines
  1. /* Copyright (C) 2000 MySQL AB & MySQL Finland AB & TCX DataKonsult AB
  2.    
  3.    This library is free software; you can redistribute it and/or
  4.    modify it under the terms of the GNU Library General Public
  5.    License as published by the Free Software Foundation; either
  6.    version 2 of the License, or (at your option) any later version.
  7.    
  8.    This library is distributed in the hope that it will be useful,
  9.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  10.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11.    Library General Public License for more details.
  12.    
  13.    You should have received a copy of the GNU Library General Public
  14.    License along with this library; if not, write to the Free
  15.    Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
  16.    MA 02111-1307, USA */
  17.  
  18. /* Written by Sinisa Milivojevic <sinisa@coresinc.com> */
  19.  
  20. #include <global.h>
  21. #ifdef HAVE_COMPRESS
  22. #include <my_sys.h>
  23. #include <zlib.h>
  24.  
  25. /*
  26. ** This replaces the packet with a compressed packet
  27. ** Returns 1 on error
  28. ** *complen is 0 if the packet wasn't compressed
  29. */
  30.  
  31. my_bool my_compress(byte *packet, ulong *len, ulong *complen)
  32. {
  33.   if (*len < MIN_COMPRESS_LENGTH)
  34.     *complen=0;
  35.   else
  36.   {
  37.     byte *compbuf=my_compress_alloc(packet,len,complen);
  38.     if (!compbuf)
  39.       return *complen ? 0 : 1;
  40.     memcpy(packet,compbuf,*len);
  41.     my_free(compbuf,MYF(MY_WME));                          }
  42.   return 0;
  43. }
  44.  
  45.  
  46. byte *my_compress_alloc(const byte *packet, ulong *len, ulong *complen)
  47. {
  48.   byte *compbuf;
  49.   *complen =  *len * 120 / 100 + 12;
  50.   if (!(compbuf = (byte *) my_malloc(*complen,MYF(MY_WME))))
  51.     return 0;                    /* Not enough memory */
  52.   if (compress((Bytef*) compbuf,(ulong *) complen, (Bytef*) packet,
  53.            (uLong) *len ) != Z_OK)
  54.   {
  55.     my_free(compbuf,MYF(MY_WME));
  56.     return 0;
  57.   }
  58.   if (*complen >= *len)
  59.   {
  60.     *complen=0;
  61.     my_free(compbuf,MYF(MY_WME));
  62.     return 0;
  63.   }
  64.   swap(ulong,*len,*complen);            /* *len is now packet length */
  65.   return compbuf;
  66. }
  67.  
  68.  
  69. my_bool my_uncompress (byte *packet, ulong *len, ulong *complen)
  70. {
  71.   if (*complen)                    /* If compressed */
  72.   {
  73.     byte *compbuf = (byte *) my_malloc (*complen,MYF(MY_WME));
  74.     if (!compbuf)
  75.       return 1;                    /* Not enough memory */
  76.     if (uncompress((Bytef*) compbuf, complen, (Bytef*) packet, *len) != Z_OK)
  77.     {                        /* Probably wrong packet */
  78.       my_free (compbuf,MYF(MY_WME));
  79.       return 1;
  80.     }
  81.     *len = *complen;
  82.     memcpy(packet,compbuf,*len);
  83.     my_free(compbuf,MYF(MY_WME));
  84.   }
  85.   return 0;
  86. }
  87. #endif /* HAVE_COMPRESS */
  88.