Add alloca_packet() macro

1 files changed, 9 insertions, 0 deletions

diff --git a/src/packet.h b/src/packet.h
index 286192e..07c0f3b 100644
--- a/src/packet.h
+++ b/src/packet.h
@@ -29,6 +29,8 @@
 
 #include "types.h"
 
+#include <alloca.h>
+
 #include <arpa/inet.h>
 
 
@@ -41,6 +43,13 @@ struct __attribute__((packed)) gp_babel_packet {
 	uint8_t tlv[];
 };
 
+#define alloca_packet(size) ({ \
+			gp_babel_packet_t *__packet = alloca(sizeof(gp_babel_packet_t)+size); \
+			__packet->version = htons(GP_BABEL_VERSION);	\
+			__packet->len = 0;				\
+			__packet;					\
+		})
+
 static inline size_t gp_babel_packet_size(const gp_babel_packet_t *packet) {
 	return sizeof(gp_babel_packet_t)+ntohs(packet->len);
 }