mirror of
https://git.proxmox.com/git/mirror_frr
synced 2025-08-09 21:33:55 +00:00
Merge pull request #5624 from qlyoung/fix-zebra-ptm-buffer-overrun
Fix PTM ZAPI stream parsing
This commit is contained in:
commit
6e882c5c55
@ -2622,6 +2622,14 @@ void zserv_handle_commands(struct zserv *client, struct stream *msg)
|
|||||||
struct zmsghdr hdr;
|
struct zmsghdr hdr;
|
||||||
struct zebra_vrf *zvrf;
|
struct zebra_vrf *zvrf;
|
||||||
|
|
||||||
|
if (STREAM_READABLE(msg) > ZEBRA_MAX_PACKET_SIZ) {
|
||||||
|
if (IS_ZEBRA_DEBUG_PACKET && IS_ZEBRA_DEBUG_RECV)
|
||||||
|
zlog_debug(
|
||||||
|
"ZAPI message is %zu bytes long but the maximum packet size is %u; dropping",
|
||||||
|
STREAM_READABLE(msg), ZEBRA_MAX_PACKET_SIZ);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
zapi_parse_header(msg, &hdr);
|
zapi_parse_header(msg, &hdr);
|
||||||
|
|
||||||
if (IS_ZEBRA_DEBUG_PACKET && IS_ZEBRA_DEBUG_RECV)
|
if (IS_ZEBRA_DEBUG_PACKET && IS_ZEBRA_DEBUG_RECV)
|
||||||
|
@ -1400,37 +1400,25 @@ static void _zebra_ptm_reroute(struct zserv *zs, struct zebra_vrf *zvrf,
|
|||||||
struct stream *msg, uint32_t command)
|
struct stream *msg, uint32_t command)
|
||||||
{
|
{
|
||||||
struct stream *msgc;
|
struct stream *msgc;
|
||||||
size_t zmsglen, zhdrlen;
|
char buf[ZEBRA_MAX_PACKET_SIZ];
|
||||||
pid_t ppid;
|
pid_t ppid;
|
||||||
|
|
||||||
/*
|
/* Create BFD header */
|
||||||
* Don't modify message in the zebra API. In order to do that we
|
|
||||||
* need to allocate a new message stream and copy the message
|
|
||||||
* provided by zebra.
|
|
||||||
*/
|
|
||||||
msgc = stream_new(ZEBRA_MAX_PACKET_SIZ);
|
msgc = stream_new(ZEBRA_MAX_PACKET_SIZ);
|
||||||
if (msgc == NULL) {
|
|
||||||
zlog_debug("%s: not enough memory", __func__);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Calculate our header size plus the message contents. */
|
|
||||||
zhdrlen = ZEBRA_HEADER_SIZE + sizeof(uint32_t);
|
|
||||||
zmsglen = msg->endp - msg->getp;
|
|
||||||
memcpy(msgc->data + zhdrlen, msg->data + msg->getp, zmsglen);
|
|
||||||
|
|
||||||
/*
|
|
||||||
* The message type will be BFD_DEST_REPLY so we can use only
|
|
||||||
* one callback at the `bfdd` side, however the real command
|
|
||||||
* number will be included right after the zebra header.
|
|
||||||
*/
|
|
||||||
zclient_create_header(msgc, ZEBRA_BFD_DEST_REPLAY, zvrf->vrf->vrf_id);
|
zclient_create_header(msgc, ZEBRA_BFD_DEST_REPLAY, zvrf->vrf->vrf_id);
|
||||||
stream_putl(msgc, command);
|
stream_putl(msgc, command);
|
||||||
|
|
||||||
/* Update the data pointers. */
|
if (STREAM_READABLE(msg) > STREAM_WRITEABLE(msgc)) {
|
||||||
msgc->getp = 0;
|
zlog_warn("Cannot fit extended BFD header plus original message contents into ZAPI packet; dropping message");
|
||||||
msgc->endp = zhdrlen + zmsglen;
|
goto stream_failure;
|
||||||
stream_putw_at(msgc, 0, stream_get_endp(msgc));
|
}
|
||||||
|
|
||||||
|
/* Copy original message, excluding header, into new message */
|
||||||
|
stream_get_from(buf, msg, stream_get_getp(msg), STREAM_READABLE(msg));
|
||||||
|
stream_put(msgc, buf, STREAM_READABLE(msg));
|
||||||
|
|
||||||
|
/* Update length field */
|
||||||
|
stream_putw_at(msgc, 0, STREAM_READABLE(msgc));
|
||||||
|
|
||||||
zebra_ptm_send_bfdd(msgc);
|
zebra_ptm_send_bfdd(msgc);
|
||||||
|
|
||||||
@ -1441,6 +1429,7 @@ static void _zebra_ptm_reroute(struct zserv *zs, struct zebra_vrf *zvrf,
|
|||||||
return;
|
return;
|
||||||
|
|
||||||
stream_failure:
|
stream_failure:
|
||||||
|
stream_free(msgc);
|
||||||
zlog_err("%s:%d failed to registrate client pid", __FILE__, __LINE__);
|
zlog_err("%s:%d failed to registrate client pid", __FILE__, __LINE__);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user