Added pkb_unput() and pkb_len().

1 files changed, 21 insertions, 0 deletions

diff --git a/src/include/gpxe/pkbuff.h b/src/include/gpxe/pkbuff.h
index 54d8511c..bcc30497 100644
--- a/src/include/gpxe/pkbuff.h
+++ b/src/include/gpxe/pkbuff.h
@@ -69,6 +69,17 @@ static inline void * pkb_put ( struct pk_buff *pkb, size_t len ) {
 }
 
 /**
+ * Remove data from end of packet buffer
+ *
+ * @v pkb	Packet buffer
+ * @v len	Length to remove
+ */
+static inline void pkb_unput ( struct pk_buff *pkb, size_t len ) {
+	pkb->tail -= len;
+	assert ( pkb->tail >= pkb->data );
+}
+
+/**
  * Empty a packet buffer
  *
  * @v pkb	Packet buffer
@@ -77,4 +88,14 @@ static inline void pkb_empty ( struct pk_buff *pkb ) {
 	pkb->tail = pkb->data;
 }
 
+/**
+ * Calculate length of data in a packet buffer
+ *
+ * @v pkb	Packet buffer
+ * @ret len	Length of data in buffer
+ */
+static inline size_t pkb_len ( struct pk_buff *pkb ) {
+	return ( pkb->tail - pkb->data );
+}
+
 #endif /* _PKBUFF_H */