denyhosts/clamscan/libclamav/xz_iface.c

94 lines
2.7 KiB
C
Raw Normal View History

2022-10-22 18:41:00 +08:00
/*
* Copyright (C) 2013-2022 Cisco Systems, Inc. and/or its affiliates. All rights reserved.
* Copyright (C) 2013 Sourcefire, Inc.
*
* Authors: Steven Morgan (smorgan@sourcefire.com)
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License version 2 as
* published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
* MA 02110-1301, USA.
*/
#if HAVE_CONFIG_H
#include "clamav-config.h"
#endif
#include "clamav.h"
#include "7z/XzCrc64.h"
#include "xz_iface.h"
void *__xz_wrap_alloc(void *unused, size_t size);
void __xz_wrap_free(void *unused, void *freeme);
void *__xz_wrap_alloc(void *unused, size_t size)
{
UNUSEDPARAM(unused);
if (!size || size > CLI_MAX_ALLOCATION)
return NULL;
if (!size || size > CLI_MAX_ALLOCATION) {
cli_dbgmsg("xz_iface: Attempt to allocate %lu bytes exceeds CLI_MAX_ALLOCATION.\n",
(unsigned long int)size);
return NULL;
}
return cli_malloc(size);
}
void __xz_wrap_free(void *unused, void *freeme)
{
UNUSEDPARAM(unused);
free(freeme);
}
static ISzAlloc g_Alloc = {__xz_wrap_alloc, __xz_wrap_free};
int cli_XzInit(struct CLI_XZ *XZ)
{
if (SZ_OK != XzUnpacker_Create(&XZ->state, &g_Alloc))
return XZ_RESULT_DATA_ERROR;
if (g_Crc64Table[1] == 0)
Crc64GenerateTable();
return XZ_RESULT_OK;
}
void cli_XzShutdown(struct CLI_XZ *XZ)
{
if (!XZ)
return;
XzUnpacker_Free(&XZ->state);
}
int cli_XzDecode(struct CLI_XZ *XZ)
{
SRes res;
SizeT outbytes, inbytes;
inbytes = XZ->avail_in;
outbytes = XZ->avail_out;
res = XzUnpacker_Code(&XZ->state, XZ->next_out, &outbytes,
XZ->next_in, &inbytes, CODER_FINISH_ANY, &XZ->status);
XZ->avail_in -= inbytes;
XZ->next_in += inbytes;
XZ->avail_out -= outbytes;
XZ->next_out += outbytes;
if (XZ->status == CODER_STATUS_FINISHED_WITH_MARK || XzUnpacker_IsStreamWasFinished(&XZ->state))
return XZ_STREAM_END;
if (XZ->status == CODER_STATUS_NOT_FINISHED && XZ->avail_out == 0)
return XZ_RESULT_OK;
if (((inbytes == 0) && (outbytes == 0)) || res != SZ_OK) {
if (res == SZ_ERROR_MEM) {
return XZ_DIC_HEURISTIC;
}
return XZ_RESULT_DATA_ERROR;
}
return XZ_RESULT_OK;
}