/********************************************************************
* Function:
*
* Purpose:
*
* Arguments:
*
* Returns:
*
********************************************************************/
DCE2_Ret DCE2_BufferAddData(
DCE2_Buffer *buf, const uint8_t *data,
uint32_t data_len, uint32_t offset,
DCE2_BufferMinAddFlag mflag)
{
DCE2_Ret status;
if ((buf == NULL) || (data == NULL))
return DCE2_RET__ERROR;
/* Return success for this since ultimately nothing _was_ added */
if (data_len == 0)
return DCE2_RET__SUCCESS;
if ( !offset )
offset = DCE2_BufferLength(buf);
if (buf->data == NULL)
{
uint32_t size = offset + data_len;
if ((size < buf->min_add_size) && (mflag == DCE2_BUFFER_MIN_ADD_FLAG__USE))
size = buf->min_add_size;
buf->data = (uint8_t *)DCE2_Alloc(size, buf->mtype);
if (buf->data == NULL)
return DCE2_RET__ERROR;
buf->size = size;
}
else if ((offset + data_len) > buf->size)
{
uint8_t *tmp;
uint32_t new_size = offset + data_len;
if (((new_size - buf->size) < buf->min_add_size) && (mflag == DCE2_BUFFER_MIN_ADD_FLAG__USE))
new_size += buf->min_add_size;
tmp = (uint8_t *)DCE2_ReAlloc(buf->data, buf->size, new_size, buf->mtype);
if (tmp == NULL)
return DCE2_RET__ERROR;
buf->data = tmp;
buf->size = new_size;
}
status = DCE2_Memcpy(buf->data + offset, data, data_len, buf->data, buf->data + buf->size);
if (status != DCE2_RET__SUCCESS)
{
DCE2_Log(DCE2_LOG_TYPE__ERROR,
"%s(%d) Failed to copy data into buffer.", __FILE__, __LINE__);
return DCE2_RET__ERROR;
}
buf->len = offset + data_len;
return DCE2_RET__SUCCESS;
}
/********************************************************************
* Function:
*
* Purpose:
*
* Arguments:
*
* Returns:
*
********************************************************************/
void * DCE2_ReAlloc(void *old_mem, uint32_t old_size, uint32_t new_size, DCE2_MemType mtype)
{
void *new_mem;
DCE2_Ret status;
if (dce2_mem_state == DCE2_MEM_STATE__MEMCAP)
return NULL;
if (old_mem == NULL)
{
DCE2_Log(DCE2_LOG_TYPE__ERROR,
"%s(%d) Old memory passed in was NULL.",
__FILE__, __LINE__);
return NULL;
}
else if (new_size < old_size)
{
DCE2_Log(DCE2_LOG_TYPE__ERROR,
"%s(%d) New size is less than old size.",
__FILE__, __LINE__);
return NULL;
}
else if (new_size == old_size)
{
return old_mem;
}
if (DCE2_CheckMemcap(new_size - old_size, mtype) == DCE2_MEMCAP_EXCEEDED)
return NULL;
new_mem = DCE2_Alloc(new_size, mtype);
if (new_mem == NULL)
return NULL;
status = DCE2_Memcpy(new_mem, old_mem, old_size,
new_mem, (void *)((uint8_t *)new_mem + new_size));
if (status != DCE2_RET__SUCCESS)
{
DCE2_Log(DCE2_LOG_TYPE__ERROR,
"%s(%d) Failed to copy old memory into new memory.",
__FILE__, __LINE__);
DCE2_Free(new_mem, new_size, mtype);
return NULL;
}
DCE2_Free(old_mem, old_size, mtype);
return new_mem;
}
/********************************************************************
* Function: DCE2_ClFragReassemble()
*
* Reassembles fragments into reassembly buffer and copies to
* reassembly packet.
*
* Arguments:
* DCE2_SsnData *
* Pointer to the session data structure.
* DCE2_ClActTracker *
* Pointer to the connectionless activity tracker.
* DceRpcClHdr *
* Pointer to the connectionless header in the packet.
*
* Returns: None
*
********************************************************************/
static void DCE2_ClFragReassemble(DCE2_SsnData *sd, DCE2_ClActTracker *at, const DceRpcClHdr *cl_hdr)
{
DCE2_ClFragTracker *ft = &at->frag_tracker;
DCE2_ClFragNode *fnode;
uint8_t *rdata = dce2_cl_rbuf;
uint16_t rlen = sizeof(dce2_cl_rbuf);
uint32_t stub_len = 0;
const uint8_t *stub_data = NULL;
SFSnortPacket *rpkt = NULL;
PROFILE_VARS;
PREPROC_PROFILE_START(dce2_pstat_cl_reass);
for (fnode = (DCE2_ClFragNode *)DCE2_ListFirst(ft->frags);
fnode != NULL;
fnode = (DCE2_ClFragNode *)DCE2_ListNext(ft->frags))
{
if (fnode->frag_len > rlen)
{
DCE2_Log(DCE2_LOG_TYPE__WARN,
"%s(%d) Size of fragments exceeds reassembly buffer size. "
"Using as many fragments as will fit.", __FILE__, __LINE__);
break;
}
if (DCE2_Memcpy(rdata, fnode->frag_data, fnode->frag_len, rdata, rdata + rlen) != DCE2_RET__SUCCESS)
{
DCE2_Log(DCE2_LOG_TYPE__ERROR,
"%s(%d) Failed to copy data into fragment "
"reassembly buffer.", __FILE__, __LINE__);
break;
}
DCE2_MOVE(rdata, rlen, fnode->frag_len);
stub_len += fnode->frag_len;
}
switch (sd->trans)
{
case DCE2_TRANS_TYPE__UDP:
rpkt = DCE2_GetRpkt(sd->wire_pkt, DCE2_RPKT_TYPE__UDP_CL_FRAG, dce2_cl_rbuf, stub_len);
if (rpkt == NULL)
{
DCE2_Log(DCE2_LOG_TYPE__ERROR,
"%s(%d) Failed to create reassembly packet.",
__FILE__, __LINE__);
PREPROC_PROFILE_END(dce2_pstat_cl_reass);
return;
}
DCE2_ClSetRdata(at, cl_hdr, (uint8_t *)rpkt->payload,
(uint16_t)(rpkt->payload_size - DCE2_MOCK_HDR_LEN__CL));
stub_data = rpkt->payload + DCE2_MOCK_HDR_LEN__CL;
break;
default:
DCE2_Log(DCE2_LOG_TYPE__ERROR,
"%s(%d) Invalid transport type: %d",
__FILE__, __LINE__, sd->trans);
return;
}
PREPROC_PROFILE_END(dce2_pstat_cl_reass);
if (DCE2_PushPkt(rpkt) != DCE2_RET__SUCCESS)
{
DCE2_Log(DCE2_LOG_TYPE__ERROR,
"%s(%d) Failed to push packet onto packet stack.",
__FILE__, __LINE__);
return;
}
/* Cache relevant values for rule option processing */
sd->ropts.first_frag = 1;
DCE2_CopyUuid(&sd->ropts.iface, &ft->iface, DCERPC_BO_FLAG__NONE);
sd->ropts.iface_vers = ft->iface_vers;
sd->ropts.hdr_byte_order = DceRpcClByteOrder(cl_hdr);
if (ft->data_byte_order != DCE2_SENTINEL)
sd->ropts.data_byte_order = ft->data_byte_order;
else
sd->ropts.data_byte_order = DceRpcClByteOrder(cl_hdr);
if (ft->opnum != DCE2_SENTINEL)
sd->ropts.opnum = ft->opnum;
else
sd->ropts.opnum = DceRpcClOpnum(cl_hdr);
sd->ropts.stub_data = stub_data;
DCE2_Detect(sd);
DCE2_PopPkt();
dce2_stats.cl_frag_reassembled++;
}
/********************************************************************
* Function: DCE2_ClHandleFrag()
*
* Handles connectionless fragments. Creates a new fragment list
* if necessary and inserts fragment into list. Sets rule option
* values based on the fragment.
*
* Arguments:
* DCE2_SsnData *
* Pointer to the session data structure.
* DCE2_ClActTracker *
* Pointer to the connectionless activity tracker.
* DceRpcClHdr *
* Pointer to the connectionless header in the packet.
* const uint8_t *
* Pointer to current position in the packet payload.
* uint16_t
* Length of packet payload left from current pointer
* position.
*
* Returns: None
*
********************************************************************/
static void DCE2_ClHandleFrag(DCE2_SsnData *sd, DCE2_ClActTracker *at, DceRpcClHdr *cl_hdr,
const uint8_t *data_ptr, uint16_t data_len)
{
DCE2_ClFragTracker *ft = &at->frag_tracker;
DCE2_ClFragNode *fn;
uint16_t frag_len;
int status;
PROFILE_VARS;
PREPROC_PROFILE_START(dce2_pstat_cl_frag);
/* If the frag length is less than data length there might be authentication
* data that we don't want to include, otherwise just set to data len */
if (DceRpcClLen(cl_hdr) < data_len)
frag_len = DceRpcClLen(cl_hdr);
else
frag_len = data_len;
if (frag_len == 0)
{
PREPROC_PROFILE_END(dce2_pstat_cl_frag);
return;
}
if (frag_len > dce2_stats.cl_max_frag_size)
dce2_stats.cl_max_frag_size = frag_len;
if (DCE2_GcMaxFrag() && (frag_len > DCE2_GcMaxFragLen()))
frag_len = DCE2_GcMaxFragLen();
if (ft->frags == NULL)
{
/* Create new list if we don't have one already */
ft->frags = DCE2_ListNew(DCE2_LIST_TYPE__SORTED, DCE2_ClFragCompare, DCE2_ClFragDataFree,
NULL, DCE2_LIST_FLAG__NO_DUPS | DCE2_LIST_FLAG__INS_TAIL,
DCE2_MEM_TYPE__CL_FRAG);
if (ft->frags == NULL)
{
PREPROC_PROFILE_END(dce2_pstat_cl_frag);
return;
}
}
else
{
/* If we already have a fragment in the list with the same fragment number,
* that fragment will take precedence over this fragment and this fragment
* will not be used by the server */
fn = (DCE2_ClFragNode *)DCE2_ListFind(ft->frags, (void *)(uintptr_t)DceRpcClFragNum(cl_hdr));
if (fn != NULL)
{
PREPROC_PROFILE_END(dce2_pstat_cl_frag);
return;
}
}
/* Create a new frag node to insert into the list */
fn = (DCE2_ClFragNode *)DCE2_Alloc(sizeof(DCE2_ClFragNode), DCE2_MEM_TYPE__CL_FRAG);
if (fn == NULL)
{
PREPROC_PROFILE_END(dce2_pstat_cl_frag);
DCE2_ClFragReassemble(sd, at, cl_hdr);
return;
}
fn->frag_number = DceRpcClFragNum(cl_hdr);
fn->frag_len = frag_len;
/* Allocate space for the fragment data */
fn->frag_data = (uint8_t *)DCE2_Alloc(frag_len, DCE2_MEM_TYPE__CL_FRAG);
if (fn->frag_data == NULL)
{
DCE2_Free((void *)fn, sizeof(DCE2_ClFragNode), DCE2_MEM_TYPE__CL_FRAG);
PREPROC_PROFILE_END(dce2_pstat_cl_frag);
DCE2_ClFragReassemble(sd, at, cl_hdr);
return;
}
/* Copy the fragment data in the packet to the space just allocated */
status = DCE2_Memcpy(fn->frag_data, data_ptr, frag_len, fn->frag_data, fn->frag_data + frag_len);
if (status != DCE2_RET__SUCCESS)
{
DCE2_Free((void *)fn->frag_data, frag_len, DCE2_MEM_TYPE__CL_FRAG);
DCE2_Free((void *)fn, sizeof(DCE2_ClFragNode), DCE2_MEM_TYPE__CL_FRAG);
PREPROC_PROFILE_END(dce2_pstat_cl_frag);
DCE2_ClFragReassemble(sd, at, cl_hdr);
return;
}
if (DCE2_ListIsEmpty(ft->frags))
{
/* If this is the first fragment we've received, set interface uuid */
DCE2_CopyUuid(&ft->iface, DceRpcClIface(cl_hdr), DceRpcClByteOrder(cl_hdr));
ft->iface_vers = DceRpcClIfaceVers(cl_hdr);
}
if (DceRpcClLastFrag(cl_hdr))
{
/* Set number of expected frags on last frag */
ft->num_expected_frags = DceRpcClFragNum(cl_hdr) + 1;
}
else if (DceRpcClFirstFrag(cl_hdr))
{
/* Set opum and byte order on first frag */
ft->opnum = DceRpcClOpnum(cl_hdr);
ft->data_byte_order = DceRpcClByteOrder(cl_hdr);
}
/* Insert frag node into the list */
status = DCE2_ListInsert(ft->frags, (void *)(uintptr_t)fn->frag_number, (void *)fn);
if (status != DCE2_RET__SUCCESS)
{
DCE2_Free((void *)fn->frag_data, frag_len, DCE2_MEM_TYPE__CL_FRAG);
DCE2_Free((void *)fn, sizeof(DCE2_ClFragNode), DCE2_MEM_TYPE__CL_FRAG);
PREPROC_PROFILE_END(dce2_pstat_cl_frag);
DCE2_ClFragReassemble(sd, at, cl_hdr);
return;
}
/* Fragment number field in header is uint16_t */
if ((ft->num_expected_frags != DCE2_SENTINEL) &&
(uint16_t)ft->frags->num_nodes == (uint16_t)ft->num_expected_frags)
{
PREPROC_PROFILE_END(dce2_pstat_cl_frag);
/* We got all of the frags - reassemble */
DCE2_ClFragReassemble(sd, at, cl_hdr);
at->seq_num_invalid = 1;
return;
}
PREPROC_PROFILE_END(dce2_pstat_cl_frag);
/* Cache relevant values for rule option processing */
sd->ropts.first_frag = DceRpcClFirstFrag(cl_hdr);
DCE2_CopyUuid(&sd->ropts.iface, &ft->iface, DCERPC_BO_FLAG__NONE);
sd->ropts.iface_vers = ft->iface_vers;
sd->ropts.hdr_byte_order = DceRpcClByteOrder(cl_hdr);
if (ft->data_byte_order != DCE2_SENTINEL)
sd->ropts.data_byte_order = ft->data_byte_order;
else
sd->ropts.data_byte_order = DceRpcClByteOrder(cl_hdr);
if (ft->opnum != DCE2_SENTINEL)
sd->ropts.opnum = ft->opnum;
else
sd->ropts.opnum = DceRpcClOpnum(cl_hdr);
sd->ropts.stub_data = (uint8_t *)cl_hdr + sizeof(DceRpcClHdr);
DCE2_Detect(sd);
}
/********************************************************************
* Function:
*
* Must have allocated data in buffer and data_len must fit in
* buffer.
*
* Arguments:
*
* Returns:
*
********************************************************************/
DCE2_Ret DCE2_BufferMoveData(DCE2_Buffer *buf, uint32_t data_offset,
const uint8_t *move, uint32_t move_len)
{
DCE2_Ret status;
uint8_t *offset, *end;
if ((buf == NULL) || (buf->data == NULL) || (move == NULL))
return DCE2_RET__ERROR;
/* Return success for this since ultimately nothing _was_ moved */
if (move_len == 0)
return DCE2_RET__SUCCESS;
offset = buf->data + data_offset;
end = buf->data + buf->len;
/* Moved data must be within current data */
if ((move < buf->data) || ((move + move_len) > end))
return DCE2_RET__ERROR;
/* No move required */
if (move == offset)
return DCE2_RET__SUCCESS;
/* Would have to do two moves. One for the data and one to realign data
* with start of moved data. Don't want to succeed on the first and fail
* on the second and leave the buffer in a bad state. Don't want to use
* an offset in data buffer because want to keep the size the same. */
if (move == buf->data)
{
uint32_t tmp_size = buf->len;
uint8_t *tmp = (uint8_t *)DCE2_Alloc(tmp_size, buf->mtype);
uint8_t *tmp_offset, *tmp_end;
uint32_t new_len;
if (tmp == NULL)
return DCE2_RET__ERROR;
tmp_offset = tmp + data_offset;
tmp_end = tmp + tmp_size;
status = DCE2_Memcpy(tmp, buf->data, buf->len, tmp, tmp_end);
if (status != DCE2_RET__SUCCESS)
{
DCE2_Free((void *)tmp, tmp_size, buf->mtype);
DCE2_Log(DCE2_LOG_TYPE__ERROR,
"%s(%d) Failed to move data in buffer.", __FILE__, __LINE__);
return DCE2_RET__ERROR;
}
status = DCE2_Memmove(tmp_offset, tmp, move_len, tmp_offset, tmp_end);
if (status != DCE2_RET__SUCCESS)
{
DCE2_Free((void *)tmp, tmp_size, buf->mtype);
DCE2_Log(DCE2_LOG_TYPE__ERROR,
"%s(%d) Failed to move data in buffer.", __FILE__, __LINE__);
return DCE2_RET__ERROR;
}
if (tmp_offset > (tmp + move_len))
tmp_offset = tmp + move_len;
new_len = tmp_end - tmp_offset;
status = DCE2_Memcpy(buf->data, tmp_offset, new_len, buf->data, end);
if (status != DCE2_RET__SUCCESS)
{
DCE2_Free((void *)tmp, tmp_size, buf->mtype);
DCE2_Log(DCE2_LOG_TYPE__ERROR,
"%s(%d) Failed to move data in buffer.", __FILE__, __LINE__);
return DCE2_RET__ERROR;
}
buf->len = new_len;
DCE2_Free((void *)tmp, tmp_size, buf->mtype);
}
else
{
status = DCE2_Memmove(offset, move, move_len, offset, end);
if (status != DCE2_RET__SUCCESS)
{
DCE2_Log(DCE2_LOG_TYPE__ERROR,
"%s(%d) Failed to move data in buffer", __FILE__, __LINE__);
return DCE2_RET__ERROR;
}
/* If we have a new end of data, adjust length */
if ((move + move_len) == end)
buf->len = data_offset + move_len;
}
return DCE2_RET__SUCCESS;
}
/*********************************************************************
* Function:
*
* Purpose:
*
* Arguments:
*
* Returns:
*
*********************************************************************/
DCE2_Ret DCE2_AddDataToRpkt(SFSnortPacket *rpkt, DCE2_RpktType rtype,
const uint8_t *data, uint32_t data_len)
{
int hdr_overhead = 0;
const uint8_t *pkt_data_end;
const uint8_t *payload_end;
DCE2_Ret status;
if ((rpkt == NULL) || (data == NULL) || (data_len == 0))
return DCE2_RET__ERROR;
if (rpkt->payload == NULL)
return DCE2_RET__ERROR;
/* This is a check to make sure we don't overwrite header data */
switch (rtype)
{
case DCE2_RPKT_TYPE__SMB_CO_SEG:
if (DCE2_SsnFromClient(rpkt))
hdr_overhead = DCE2_MOCK_HDR_LEN__SMB_CLI;
else
hdr_overhead = DCE2_MOCK_HDR_LEN__SMB_SRV;
break;
case DCE2_RPKT_TYPE__SMB_CO_FRAG:
if (DCE2_SsnFromClient(rpkt))
hdr_overhead = DCE2_MOCK_HDR_LEN__SMB_CLI + DCE2_MOCK_HDR_LEN__CO_CLI;
else
hdr_overhead = DCE2_MOCK_HDR_LEN__SMB_SRV + DCE2_MOCK_HDR_LEN__CO_SRV;
break;
case DCE2_RPKT_TYPE__TCP_CO_FRAG:
if (DCE2_SsnFromClient(rpkt))
hdr_overhead = DCE2_MOCK_HDR_LEN__CO_CLI;
else
hdr_overhead = DCE2_MOCK_HDR_LEN__CO_SRV;
break;
case DCE2_RPKT_TYPE__UDP_CL_FRAG:
hdr_overhead = DCE2_MOCK_HDR_LEN__CL;
break;
default:
break;
}
if (rpkt->payload_size < hdr_overhead)
return DCE2_RET__ERROR;
pkt_data_end = rpkt->pkt_data + rpkt->max_payload;
payload_end = rpkt->payload + rpkt->payload_size;
if ((payload_end + data_len) > pkt_data_end)
data_len = pkt_data_end - payload_end;
status = DCE2_Memcpy((void *)payload_end, (void *)data, (size_t)data_len,
(void *)payload_end, (void *)pkt_data_end);
if (status != DCE2_RET__SUCCESS)
{
DCE2_Log(DCE2_LOG_TYPE__ERROR,
"%s(%d) Failed to copy data into reassembly packet.",
__FILE__, __LINE__);
return DCE2_RET__ERROR;
}
rpkt->payload_size += (uint16_t)data_len;
// there is room for optimization here since the update was done
// earlier - that my be eliminated, but only in this case one
// approach is to move the updates to push pkt - but don't want
// to update non-dce2 pseudo pkts; perhaps a flag check there
// will suffice.
_dpd.encodeUpdate(rpkt);
if (rpkt->family == AF_INET)
{
rpkt->ip4h->ip_len = rpkt->ip4_header->data_length;
}
else
{
IP6RawHdr* ip6h = (IP6RawHdr*)rpkt->raw_ip6_header;
if ( ip6h ) rpkt->ip6h->len = ip6h->ip6_payload_len;
}
return DCE2_RET__SUCCESS;
}
/*********************************************************************
* Function: DCE2_GetRpkt()
*
* Purpose:
*
* Arguments:
* SFSnortPacket * - pointer to packet off wire
* const uint8_t * - pointer to data to attach to reassembly packet
* uint16_t - length of data
*
* Returns:
* SFSnortPacket * - pointer to reassembly packet
*
*********************************************************************/
SFSnortPacket * DCE2_GetRpkt(const SFSnortPacket *wire_pkt, DCE2_RpktType rpkt_type,
const uint8_t *data, uint32_t data_len)
{
DCE2_Ret status;
SFSnortPacket *rpkt;
uint16_t payload_len = 0;
uint16_t data_overhead = 0;
rpkt = dce2_rpkt[rpkt_type];
switch (rpkt_type)
{
case DCE2_RPKT_TYPE__SMB_SEG:
_dpd.encodeFormat(ENC_DYN_FWD, wire_pkt, rpkt, PSEUDO_PKT_SMB_SEG);
break;
case DCE2_RPKT_TYPE__SMB_TRANS:
// TBD these memset()s could be encapsulated by the various
// init functions which should also return the data_overhead.
// Better still pass in rpkt and let the init function update
// payload, etc. Also, some memsets could probably be avoided
// by explicitly setting the unitialized header fields.
_dpd.encodeFormat(ENC_DYN_FWD, wire_pkt, rpkt, PSEUDO_PKT_SMB_TRANS);
if (DCE2_SsnFromClient(wire_pkt))
{
data_overhead = DCE2_MOCK_HDR_LEN__SMB_CLI;
memset((void*)rpkt->payload, 0, data_overhead);
DCE2_SmbInitRdata((uint8_t *)rpkt->payload, FLAG_FROM_CLIENT);
}
else
{
data_overhead = DCE2_MOCK_HDR_LEN__SMB_SRV;
memset((void*)rpkt->payload, 0, data_overhead);
DCE2_SmbInitRdata((uint8_t *)rpkt->payload, FLAG_FROM_SERVER);
}
break;
case DCE2_RPKT_TYPE__SMB_CO_SEG:
_dpd.encodeFormat(ENC_DYN_FWD, wire_pkt, rpkt, PSEUDO_PKT_DCE_SEG);
if (DCE2_SsnFromClient(wire_pkt))
{
data_overhead = DCE2_MOCK_HDR_LEN__SMB_CLI;
memset((void*)rpkt->payload, 0, data_overhead);
DCE2_SmbInitRdata((uint8_t *)rpkt->payload, FLAG_FROM_CLIENT);
}
else
{
data_overhead = DCE2_MOCK_HDR_LEN__SMB_SRV;
memset((void*)rpkt->payload, 0, data_overhead);
DCE2_SmbInitRdata((uint8_t *)rpkt->payload, FLAG_FROM_SERVER);
}
break;
case DCE2_RPKT_TYPE__SMB_CO_FRAG:
_dpd.encodeFormat(ENC_DYN_FWD, wire_pkt, rpkt, PSEUDO_PKT_DCE_FRAG);
if (DCE2_SsnFromClient(wire_pkt))
{
data_overhead = DCE2_MOCK_HDR_LEN__SMB_CLI + DCE2_MOCK_HDR_LEN__CO_CLI;
memset((void*)rpkt->payload, 0, data_overhead);
DCE2_SmbInitRdata((uint8_t *)rpkt->payload, FLAG_FROM_CLIENT);
DCE2_CoInitRdata((uint8_t *)rpkt->payload +
DCE2_MOCK_HDR_LEN__SMB_CLI, FLAG_FROM_CLIENT);
}
else
{
data_overhead = DCE2_MOCK_HDR_LEN__SMB_SRV + DCE2_MOCK_HDR_LEN__CO_SRV;
memset((void*)rpkt->payload, 0, data_overhead);
DCE2_SmbInitRdata((uint8_t *)rpkt->payload, FLAG_FROM_SERVER);
DCE2_CoInitRdata((uint8_t *)rpkt->payload +
DCE2_MOCK_HDR_LEN__SMB_SRV, FLAG_FROM_SERVER);
}
break;
case DCE2_RPKT_TYPE__TCP_CO_SEG:
_dpd.encodeFormat(ENC_DYN_FWD, wire_pkt, rpkt, PSEUDO_PKT_DCE_SEG);
break;
case DCE2_RPKT_TYPE__TCP_CO_FRAG:
_dpd.encodeFormat(ENC_DYN_FWD, wire_pkt, rpkt, PSEUDO_PKT_DCE_FRAG);
if (DCE2_SsnFromClient(wire_pkt))
{
data_overhead = DCE2_MOCK_HDR_LEN__CO_CLI;
memset((void*)rpkt->payload, 0, data_overhead);
DCE2_CoInitRdata((uint8_t *)rpkt->payload, FLAG_FROM_CLIENT);
}
else
{
data_overhead = DCE2_MOCK_HDR_LEN__CO_SRV;
memset((void*)rpkt->payload, 0, data_overhead);
DCE2_CoInitRdata((uint8_t *)rpkt->payload, FLAG_FROM_SERVER);
}
break;
case DCE2_RPKT_TYPE__UDP_CL_FRAG:
_dpd.encodeFormat(ENC_DYN_FWD, wire_pkt, rpkt, PSEUDO_PKT_DCE_FRAG);
data_overhead = DCE2_MOCK_HDR_LEN__CL;
memset((void*)rpkt->payload, 0, data_overhead);
DCE2_ClInitRdata((uint8_t *)rpkt->payload);
break;
default:
DCE2_Log(DCE2_LOG_TYPE__ERROR,
"%s(%d) Invalid reassembly packet type: %d",
__FILE__, __LINE__, rpkt_type);
return NULL;
}
payload_len = rpkt->max_payload;
if ((data_overhead + data_len) > payload_len)
data_len -= (data_overhead + data_len) - payload_len;
status = DCE2_Memcpy(
(void *)(rpkt->payload + data_overhead),
(void *)data, (size_t)data_len, (void *)rpkt->payload,
(void *)((uint8_t *)rpkt->payload + payload_len));
if (status != DCE2_RET__SUCCESS)
{
DCE2_Log(DCE2_LOG_TYPE__ERROR,
"%s(%d) Failed to copy data into reassembly packet.",
__FILE__, __LINE__);
return NULL;
}
rpkt->payload_size = (uint16_t)(data_overhead + data_len);
_dpd.encodeUpdate(rpkt);
if (wire_pkt->family == AF_INET)
{
rpkt->ip4h->ip_len = rpkt->ip4_header->data_length;
}
else
{
IP6RawHdr* ip6h = (IP6RawHdr*)rpkt->raw_ip6_header;
if ( ip6h ) rpkt->ip6h->len = ip6h->ip6_payload_len;
}
rpkt->flags |= FLAG_STREAM_EST;
if (DCE2_SsnFromClient(wire_pkt))
rpkt->flags |= FLAG_FROM_CLIENT;
else
rpkt->flags |= FLAG_FROM_SERVER;
rpkt->stream_session = wire_pkt->stream_session;
return rpkt;
}