#include <vnet/pg/pg.h>
#include <vnet/ip/ip.h> /* for unformat_udp_udp_port */
+#include <vnet/udp/udp.h>
#define UDP_PG_EDIT_LENGTH (1 << 0)
#define UDP_PG_EDIT_CHECKSUM (1 << 1)
ip0 = (void *) (p0->data + ip_offset);
udp0 = (void *) (p0->data + udp_offset);
- udp_len0 = clib_net_to_host_u16 (ip0->length) - sizeof (ip0[0]);
+ udp_len0 = vlib_buffer_length_in_chain (vm, p0) - udp_offset;
if (flags & UDP_PG_EDIT_LENGTH)
- udp0->length =
- clib_net_to_host_u16 (vlib_buffer_length_in_chain (vm, p0)
- - ip_offset);
+ udp0->length = clib_host_to_net_u16 (udp_len0);
+ else
+ udp_len0 = clib_host_to_net_u16 (udp0->length);
/* Initialize checksum with header. */
if (flags & UDP_PG_EDIT_CHECKSUM)
u16 dst_port;
tcp_udp_port_info_t *pi;
+ /* For the pg format of applications over UDP local */
+ udp_dst_port_info_t *pi2 = NULL;
+
pi = 0;
if (p->dst_port.type == PG_EDIT_FIXED)
{
dst_port = pg_edit_get_value (&p->dst_port, PG_EDIT_LO);
pi = ip_get_tcp_udp_port_info (im, dst_port);
+ pi2 = udp_get_dst_port_info (&udp_main, dst_port, UDP_IP4);
+ if (!pi2)
+ pi2 = udp_get_dst_port_info (&udp_main, dst_port, UDP_IP6);
}
if (pi && pi->unformat_pg_edit
&& unformat_user (input, pi->unformat_pg_edit, s))
;
-
+ else if (pi2 && pi2->unformat_pg_edit
+ && unformat_user (input, pi2->unformat_pg_edit, s))
+ ;
else if (!unformat_user (input, unformat_pg_payload, s))
goto error;