lzexpand.c File Reference

#include "config.h"
#include <string.h>
#include <ctype.h>
#include <sys/types.h>
#include <stdarg.h>
#include <stdio.h>
#include "windef.h"
#include "winbase.h"
#include "lzexpand.h"
#include "wine/unicode.h"
#include "wine/debug.h"

Include dependency graph for lzexpand.c:

Go to the source code of this file.

Classes
struct	lzfileheader
struct	lzstate

Macros
#define	GETLEN   2048
#define	LZ_MAGIC_LEN   8
#define	LZ_HEADER_LEN   14
#define	LZ_TABLE_SIZE   0x1000
#define	MAX_LZSTATES   16
#define	LZ_MIN_HANDLE   0x400
#define	IS_LZ_HANDLE(h)   (((h) >= LZ_MIN_HANDLE) && ((h) < LZ_MIN_HANDLE+MAX_LZSTATES))
#define	GET_LZ_STATE(h)   (IS_LZ_HANDLE(h) ? lzstates[(h)-LZ_MIN_HANDLE] : NULL)
#define	GET(lzs, b)   _lzget(lzs,&b)
#define	GET_FLUSH(lzs)   lzs->getcur=lzs->getlen;
#define	DECOMPRESS_ONE_BYTE
#define	BUFLEN   1000

Functions
	WINE_DEFAULT_DEBUG_CHANNEL (file)
static int	_lzget (struct lzstate *lzs, BYTE *b)
static INT	read_header (HFILE fd, struct lzfileheader *head)
INT WINAPI	LZStart (void)
HFILE WINAPI	LZInit (HFILE hfSrc)
void WINAPI	LZDone (void)
INT WINAPI	GetExpandedNameA (LPSTR in, LPSTR out)
INT WINAPI	GetExpandedNameW (LPWSTR in, LPWSTR out)
INT WINAPI	LZRead (HFILE fd, LPSTR vbuf, INT toread)
LONG WINAPI	LZSeek (HFILE fd, LONG off, INT type)
LONG WINAPI	LZCopy (HFILE src, HFILE dest)
static LPSTR	LZEXPAND_MangleName (LPCSTR fn)
HFILE WINAPI	LZOpenFileA (LPSTR fn, LPOFSTRUCT ofs, WORD mode)
HFILE WINAPI	LZOpenFileW (LPWSTR fn, LPOFSTRUCT ofs, WORD mode)
void WINAPI	LZClose (HFILE fd)

Variables
static const BYTE	LZMagic [LZ_MAGIC_LEN] ={'S','Z','D','D',0x88,0xf0,0x27,0x33}
static struct lzstate *	lzstates [MAX_LZSTATES]

Macro Definition Documentation

BUFLEN

#define BUFLEN   1000

DECOMPRESS_ONE_BYTE

#define DECOMPRESS_ONE_BYTE

GET

#define GET	(		lzs,
			b
	)		_lzget(lzs,&b)

Definition at line 123 of file lzexpand.c.

GET_FLUSH

#define GET_FLUSH

(

lzs

)

lzs->getcur=lzs->getlen;

Definition at line 124 of file lzexpand.c.

GET_LZ_STATE

#define GET_LZ_STATE

(

h

)

(IS_LZ_HANDLE(h) ? lzstates[(h)-LZ_MIN_HANDLE] : NULL)

Definition at line 120 of file lzexpand.c.

GETLEN

#define GETLEN   2048

Definition at line 77 of file lzexpand.c.

IS_LZ_HANDLE

#define IS_LZ_HANDLE

(

h

)

(((h) >= LZ_MIN_HANDLE) && ((h) < LZ_MIN_HANDLE+MAX_LZSTATES))

Definition at line 119 of file lzexpand.c.

LZ_HEADER_LEN

#define LZ_HEADER_LEN   14

Definition at line 80 of file lzexpand.c.

LZ_MAGIC_LEN

#define LZ_MAGIC_LEN   8

Definition at line 79 of file lzexpand.c.

LZ_MIN_HANDLE

#define LZ_MIN_HANDLE   0x400

Definition at line 118 of file lzexpand.c.

LZ_TABLE_SIZE

#define LZ_TABLE_SIZE   0x1000

Definition at line 91 of file lzexpand.c.

MAX_LZSTATES

#define MAX_LZSTATES   16

Definition at line 115 of file lzexpand.c.

Function Documentation

_lzget()

static int _lzget ( struct lzstate *  lzs, BYTE *  b  )

static

Definition at line 127 of file lzexpand.c.

                                    {
    if (lzs->getcur<lzs->getlen) {
        *b      = lzs->get[lzs->getcur++];
        return      1;
    } else {
        int ret = _lread(lzs->realfd,lzs->get,GETLEN);
        if (ret==HFILE_ERROR)
            return HFILE_ERROR;
        if (ret==0)
            return 0;
        lzs->getlen = ret;
        lzs->getcur = 1;
        *b      = *(lzs->get);
        return 1;
    }
}

GetExpandedNameA()

INT WINAPI GetExpandedNameA	(	LPSTR	in,
		LPSTR	out
	)

Definition at line 257 of file lzexpand.c.

{
    struct lzfileheader head;
    HFILE       fd;
    OFSTRUCT    ofs;
    INT     fnislowercased,ret,len;
    LPSTR       s,t;
 
    TRACE("(%s)\n",in);
    fd=OpenFile(in,&ofs,OF_READ);
    if (fd==HFILE_ERROR)
        return (INT)(INT16)LZERROR_BADINHANDLE;
    strcpy(out,in);
    ret=read_header(fd,&head);
    if (ret<=0) {
        /* not a LZ compressed file, so the expanded name is the same
         * as the input name */
        _lclose(fd);
        return 1;
    }
 
 
    /* look for directory prefix and skip it. */
    s=out;
    while (NULL!=(t=strpbrk(s,"/\\:")))
        s=t+1;
 
    /* now mangle the basename */
    if (!*s) {
        /* FIXME: hmm. shouldn't happen? */
        WARN("Specified a directory or what? (%s)\n",in);
        _lclose(fd);
        return 1;
    }
    /* see if we should use lowercase or uppercase on the last char */
    fnislowercased=1;
    t=s+strlen(s)-1;
    while (t>=out) {
        if (!isalpha(*t)) {
            t--;
            continue;
        }
        fnislowercased=islower(*t);
        break;
    }
    if (isalpha(head.lastchar)) {
        if (fnislowercased)
            head.lastchar=tolower(head.lastchar);
        else
            head.lastchar=toupper(head.lastchar);
    }
 
    /* now look where to replace the last character */
    if (NULL!=(t=strchr(s,'.'))) {
        if (t[1]=='\0') {
            t[0]='\0';
        } else {
            len=strlen(t)-1;
            if (t[len]=='_')
                t[len]=head.lastchar;
        }
    } /* else no modification necessary */
    _lclose(fd);
    return 1;
}

Referenced by GetExpandedNameW(), and main().

GetExpandedNameW()

INT WINAPI GetExpandedNameW	(	LPWSTR	in,
		LPWSTR	out
	)

Definition at line 327 of file lzexpand.c.

{
    INT ret;
    DWORD len = WideCharToMultiByte( CP_ACP, 0, in, -1, NULL, 0, NULL, NULL );
    char *xin = HeapAlloc( GetProcessHeap(), 0, len );
    char *xout = HeapAlloc( GetProcessHeap(), 0, len+3 );
    WideCharToMultiByte( CP_ACP, 0, in, -1, xin, len, NULL, NULL );
    if ((ret = GetExpandedNameA( xin, xout )) > 0)
        MultiByteToWideChar( CP_ACP, 0, xout, -1, out, strlenW(in)+4 );
    HeapFree( GetProcessHeap(), 0, xin );
    HeapFree( GetProcessHeap(), 0, xout );
    return ret;
}

LZClose()

void WINAPI LZClose

(

HFILE

fd

)

Definition at line 595 of file lzexpand.c.

{
    struct lzstate *lzs;
 
    TRACE("(%d)\n",fd);
        if (!(lzs = GET_LZ_STATE(fd))) _lclose(fd);
        else
        {
            HeapFree( GetProcessHeap(), 0, lzs->get );

Referenced by create_file(), create_fileW(), decompress_file_lz(), delete_file(), delete_fileW(), do_file_copyW(), get_file_sizes_lz(), GetFileVersionInfoExW(), GetFileVersionInfoSizeExW(), Imm32CopyImeFile(), LZCopy(), main(), test_LZCopy(), test_LZOpenFileA(), test_LZOpenFileA_existing_compressed(), test_LZOpenFileW(), test_LZOpenFileW_existing_compressed(), test_LZRead(), TLB_NEFile_Open(), and VerInstallFileA().

LZCopy()

LONG WINAPI LZCopy	(	HFILE	src,
		HFILE	dest
	)

Definition at line 467 of file lzexpand.c.

{
    int usedlzinit = 0, ret, wret;
    LONG    len;
    HFILE   oldsrc = src, srcfd;
    FILETIME filetime;
    struct  lzstate *lzs;
#define BUFLEN  1000
    CHAR    buf[BUFLEN];
    /* we need that weird typedef, for i can't seem to get function pointer
     * casts right. (Or they probably just do not like WINAPI in general)
     */
    typedef UINT    (WINAPI *_readfun)(HFILE,LPVOID,UINT);
 
    _readfun    xread;
 
    TRACE("(%d,%d)\n",src,dest);
    if (!IS_LZ_HANDLE(src)) {
        src = LZInit(src);
                if ((INT)src <= 0) return 0;
        if (src != oldsrc) usedlzinit=1;
    }
 
    /* not compressed? just copy */
        if (!IS_LZ_HANDLE(src))
#ifdef __REACTOS__
        xread=(_readfun)_hread; // ROSHACK
#else
        xread=_lread;
#endif
    else
        xread=(_readfun)LZRead;
    len=0;
    while (1) {
        ret=xread(src,buf,BUFLEN);
        if (ret<=0) {
            if (ret==0)
                break;
            if (ret==-1)
                return LZERROR_READ;
            return ret;
        }
        len    += ret;
        wret    = _lwrite(dest,buf,ret);
        if (wret!=ret)
            return LZERROR_WRITE;
    }
 
    /* Maintain the timestamp of source file to destination file */
    srcfd = (!(lzs = GET_LZ_STATE(src))) ? src : lzs->realfd;
    GetFileTime( LongToHandle(srcfd), NULL, NULL, &filetime );
    SetFileTime( LongToHandle(dest), NULL, NULL, &filetime );
 
    /* close handle */

Referenced by decompress_file_lz(), do_file_copyW(), Imm32CopyImeFile(), main(), test_LZCopy(), and VerInstallFileA().

LZDone()

void WINAPI LZDone

(

void

)

Definition at line 240 of file lzexpand.c.

{
    TRACE("(void)\n");
}

LZEXPAND_MangleName()

static LPSTR LZEXPAND_MangleName ( LPCSTR  fn )

static

Definition at line 528 of file lzexpand.c.

{
    char *p;
    char *mfn = HeapAlloc( GetProcessHeap(), 0, strlen(fn) + 3 ); /* "._" and \0 */
    if(mfn == NULL) return NULL;
    strcpy( mfn, fn );
    if (!(p = strrchr( mfn, '\\' ))) p = mfn;
    if ((p = strchr( p, '.' )))
    {
        p++;
        if (strlen(p) < 3) strcat( p, "_" );  /* append '_' */

Referenced by LZOpenFileA().

LZInit()

HFILE WINAPI LZInit

(

HFILE

hfSrc

)

Definition at line 197 of file lzexpand.c.

{
 
    struct  lzfileheader    head;
    struct  lzstate     *lzs;
    int i, ret;
 
    TRACE("(%d)\n",hfSrc);
    ret=read_header(hfSrc,&head);
    if (ret<=0) {
        _llseek(hfSrc,0,SEEK_SET);
        return ret?ret:hfSrc;
    }
        for (i = 0; i < MAX_LZSTATES; i++) if (!lzstates[i]) break;
        if (i == MAX_LZSTATES) return LZERROR_GLOBALLOC;
    lzstates[i] = lzs = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*lzs) );
    if(lzs == NULL) return LZERROR_GLOBALLOC;
 
    lzs->realfd = hfSrc;
    lzs->lastchar   = head.lastchar;
    lzs->reallength = head.reallength;
 
    lzs->get    = HeapAlloc( GetProcessHeap(), 0, GETLEN );
    lzs->getlen = 0;
    lzs->getcur = 0;
 
    if(lzs->get == NULL) {
        HeapFree(GetProcessHeap(), 0, lzs);
        lzstates[i] = NULL;
        return LZERROR_GLOBALLOC;
    }
 
    /* Yes, preinitialize with spaces */
    memset(lzs->table,' ',LZ_TABLE_SIZE);
    /* Yes, start 16 byte from the END of the table */
    lzs->curtabent  = 0xff0;
    return LZ_MIN_HANDLE + i;
}

Referenced by LZCopy(), and LZOpenFileA().

LZOpenFileA()

HFILE WINAPI LZOpenFileA	(	LPSTR	fn,
		LPOFSTRUCT	ofs,
		WORD	mode
	)

Definition at line 551 of file lzexpand.c.

{
    HFILE   fd,cfd;
    BYTE    ofs_cBytes = ofs->cBytes;
 
    TRACE("(%s,%p,%d)\n",fn,ofs,mode);
    /* 0x70 represents all OF_SHARE_* flags, ignore them for the check */
    fd=OpenFile(fn,ofs,mode);
    if (fd==HFILE_ERROR)
        {
            LPSTR mfn = LZEXPAND_MangleName(fn);
            fd = OpenFile(mfn,ofs,mode);
            HeapFree( GetProcessHeap(), 0, mfn );
    }
    if (fd==HFILE_ERROR)
        ofs->cBytes = ofs_cBytes;
    if ((mode&~0x70)!=OF_READ)
        return fd;
    if (fd==HFILE_ERROR)

Referenced by create_file(), delete_file(), LZOpenFileW(), main(), test_LZCopy(), test_LZOpenFileA(), test_LZOpenFileA_existing_compressed(), test_LZOpenFileA_nonexisting_compressed(), test_LZRead(), and VerInstallFileA().

LZOpenFileW()

HFILE WINAPI LZOpenFileW	(	LPWSTR	fn,
		LPOFSTRUCT	ofs,
		WORD	mode
	)

Definition at line 580 of file lzexpand.c.

{
    HFILE ret;
    DWORD len = WideCharToMultiByte( CP_ACP, 0, fn, -1, NULL, 0, NULL, NULL );
    LPSTR xfn = HeapAlloc( GetProcessHeap(), 0, len );

Referenced by create_fileW(), decompress_file_lz(), delete_fileW(), do_file_copyW(), get_file_sizes_lz(), GetFileVersionInfoExW(), GetFileVersionInfoSizeExW(), Imm32CopyImeFile(), test_LZOpenFileW(), test_LZOpenFileW_existing_compressed(), test_LZOpenFileW_nonexisting_compressed(), and TLB_NEFile_Open().

LZRead()

INT WINAPI LZRead	(	HFILE	fd,
		LPSTR	vbuf,
		INT	toread
	)

Definition at line 345 of file lzexpand.c.

{
    int howmuch;
    BYTE    b,*buf;
    struct  lzstate *lzs;
 
    buf=(LPBYTE)vbuf;
    TRACE("(%d,%p,%d)\n",fd,buf,toread);
    howmuch=toread;
    if (!(lzs = GET_LZ_STATE(fd))) return _lread(fd,buf,toread);
 
/* The decompressor itself is in a define, cause we need it twice
 * in this function. (the decompressed byte will be in b)
 */
#define DECOMPRESS_ONE_BYTE                         \
        if (lzs->stringlen) {                   \
            b       = lzs->table[lzs->stringpos];   \
            lzs->stringpos  = (lzs->stringpos+1)&0xFFF; \
            lzs->stringlen--;               \
        } else {                        \
            if (!(lzs->bytetype&0x100)) {           \
                if (1!=GET(lzs,b))          \
                    return toread-howmuch;      \
                lzs->bytetype = b|0xFF00;       \
            }                       \
            if (lzs->bytetype & 1) {            \
                if (1!=GET(lzs,b))          \
                    return toread-howmuch;      \
            } else {                    \
                BYTE    b1,b2;              \
                                    \
                if (1!=GET(lzs,b1))         \
                    return toread-howmuch;      \
                if (1!=GET(lzs,b2))         \
                    return toread-howmuch;      \
                /* Format:              \
                 * b1 b2                \
                 * AB CD                \
                 * where CAB is the stringoffset in the table\
                 * and D+3 is the len of the string \
                 */                 \
                lzs->stringpos  = b1|((b2&0xf0)<<4);    \
                lzs->stringlen  = (b2&0xf)+2;       \
                /* 3, but we use a  byte already below ... */\
                b       = lzs->table[lzs->stringpos];\
                lzs->stringpos  = (lzs->stringpos+1)&0xFFF;\
            }                       \
            lzs->bytetype>>=1;              \
        }                           \
        /* store b in table */                  \
        lzs->table[lzs->curtabent++]= b;            \
        lzs->curtabent  &= 0xFFF;               \
        lzs->realcurrent++;
 
    /* if someone has seeked, we have to bring the decompressor
     * to that position
     */
    if (lzs->realcurrent!=lzs->realwanted) {
        /* if the wanted position is before the current position
         * I see no easy way to unroll ... We have to restart at
         * the beginning. *sigh*
         */
        if (lzs->realcurrent>lzs->realwanted) {
            /* flush decompressor state */
            _llseek(lzs->realfd,LZ_HEADER_LEN,SEEK_SET);
            GET_FLUSH(lzs);
            lzs->realcurrent= 0;
            lzs->bytetype   = 0;
            lzs->stringlen  = 0;
            memset(lzs->table,' ',LZ_TABLE_SIZE);
            lzs->curtabent  = 0xFF0;
        }
        while (lzs->realcurrent<lzs->realwanted) {
            DECOMPRESS_ONE_BYTE;
        }
    }
 
    while (howmuch) {
        DECOMPRESS_ONE_BYTE;
        lzs->realwanted++;
        *buf++      = b;

Referenced by find_ne_resource(), find_pe_resource(), GetFileVersionInfoExW(), LZCopy(), read_xx_header(), test_LZRead(), and TLB_NEFile_Open().

LZSeek()

LONG WINAPI LZSeek	(	HFILE	fd,
		LONG	off,
		INT	type
	)

Definition at line 431 of file lzexpand.c.

{
    struct  lzstate *lzs;
    LONG    newwanted;
 
    TRACE("(%d,%d,%d)\n",fd,off,type);
    /* not compressed? just use normal _llseek() */
        if (!(lzs = GET_LZ_STATE(fd))) return _llseek(fd,off,type);
    newwanted = lzs->realwanted;
    switch (type) {
    case 1: /* SEEK_CUR */
        newwanted      += off;
        break;
    case 2: /* SEEK_END */
        newwanted   = lzs->reallength-off;
        break;
    default:/* SEEK_SET */
        newwanted   = off;
        break;
    }
    if (newwanted>lzs->reallength)
        return LZERROR_BADVALUE;

Referenced by find_ne_resource(), find_pe_resource(), get_file_sizes_lz(), GetFileVersionInfoExW(), read_xx_header(), and TLB_NEFile_Open().

LZStart()

INT WINAPI LZStart

(

void

)

Definition at line 179 of file lzexpand.c.

{
    TRACE("(void)\n");
    return 1;
}

read_header()

static INT read_header ( HFILE  fd, struct lzfileheader *  head  )

static

Definition at line 149 of file lzexpand.c.

{
    BYTE    buf[LZ_HEADER_LEN];
 
    if (_llseek(fd,0,SEEK_SET)==-1)
        return LZERROR_BADINHANDLE;
 
    /* We can't directly read the lzfileheader struct due to
     * structure element alignment
     */
    if (_lread(fd,buf,LZ_HEADER_LEN)<LZ_HEADER_LEN)
        return 0;
    memcpy(head->magic,buf,LZ_MAGIC_LEN);
    memcpy(&(head->compressiontype),buf+LZ_MAGIC_LEN,1);
    memcpy(&(head->lastchar),buf+LZ_MAGIC_LEN+1,1);
 
    /* FIXME: consider endianness on non-intel architectures */
    memcpy(&(head->reallength),buf+LZ_MAGIC_LEN+2,4);
 
    if (memcmp(head->magic,LZMagic,LZ_MAGIC_LEN))
        return 0;
    if (head->compressiontype!='A')
        return LZERROR_UNKNOWNALG;
    return 1;
}

Referenced by GetExpandedNameA(), and LZInit().

WINE_DEFAULT_DEBUG_CHANNEL()

WINE_DEFAULT_DEBUG_CHANNEL

(

file

)

Variable Documentation

LZMagic

const BYTE LZMagic[LZ_MAGIC_LEN] ={'S','Z','D','D',0x88,0xf0,0x27,0x33}

static

Definition at line 89 of file lzexpand.c.

Referenced by read_header().

lzstates

struct lzstate* lzstates[MAX_LZSTATES]

static

Definition at line 116 of file lzexpand.c.

Referenced by LZClose(), and LZInit().