/* 

 * usage : lzocompress.exe <size></size>

 * 

 * lzocompress reads <size> bytes from stdin, compress them and write the result to stdout</size>

 * errors are logged to stderr

 * return == 0 iff no errors

 */

#include "lzo/lzoconf.h"

#include "lzo/lzo1x.h"

/* portability layer */

#define WANT_LZO_MALLOC 1

#include "lzo/lzoutil.h"

#include <fcntl.h></fcntl.h>

#include <stdio.h></stdio.h>

#include <stdlib.h></stdlib.h>

#if defined(_WIN32)

#include <io.h></io.h>

#endif

int main(int argc, char *argv[])

{

  lzo_uint src_len, dst_len, max_dst_len;

  lzo_bytep src;

  lzo_bytep dst;

  lzo_bytep wrkmem;

  int r;

#if defined(_WIN32)

  /* stdin/stdout must be binary streams */

  r = _setmode(_fileno(stdout), _O_BINARY);

  r = _setmode(_fileno(stdin), _O_BINARY);

#endif

  if(argc != 2) 

  {

    fprintf(stderr, "Usage : %s <size>\n    <size> is the input buffer size in bytes\n", argv[0]);</size></size>

	return -1; 

  }

  src_len = atoi(argv[1]); 

  if (lzo_init() != LZO_E_OK) 

  {

    fprintf(stderr, "Couldn't initialize lzo\n");

	return -2;

  }

  /* read input buffer */

  src = (lzo_bytep) lzo_malloc(src_len);    

  fread(src, 1, src_len, stdin);

  /* allocate the buffer to store the compressed data */

  max_dst_len = src_len + src_len/64 + 16 + 3;  

  dst = (lzo_bytep) lzo_malloc(max_dst_len);

  /* allocate the working memory for lzo */  

  wrkmem = (lzo_bytep) lzo_malloc(LZO1X_1_MEM_COMPRESS);

  /* compress data */  

  dst_len = 0;

  r = lzo1x_1_compress(src, src_len, dst, &dst_len, wrkmem);

  if (r == LZO_E_OK)

  {	

    /* write compressed data */

	fwrite(dst, 1, dst_len, stdout); 	

  }

  else

  {

	fprintf(stderr, "Compression failed : code = %d\n", r);

  }

  /* free memory */  

  lzo_free(wrkmem);

  lzo_free(dst);

  lzo_free(src);

  /* return final result */

  if (r == LZO_E_OK)

    return 0;

  else 

    return -3;

}