From be69873d1c2c9a4afe5aaed804d3a1edb564e024 Mon Sep 17 00:00:00 2001
From: "7u83@mail.ru" <7u83@mail.ru@noemail.net>
Date: Sat, 7 Feb 2015 23:50:24 +0000
Subject: [PATCH] Peek option for conn_q_recv_packet.

FossilOrigin-Name: 07b2fd4eeb2a5d3858567a1c2168efdc1e4b3e782b7ce10153cb7773bb093ac1
---
 src/capwap/conn_q_recv_packet.c | 26 +++++++++++++++++++++++---
 1 file changed, 23 insertions(+), 3 deletions(-)

diff --git a/src/capwap/conn_q_recv_packet.c b/src/capwap/conn_q_recv_packet.c
index bf215a24..cab52cd9 100644
--- a/src/capwap/conn_q_recv_packet.c
+++ b/src/capwap/conn_q_recv_packet.c
@@ -22,11 +22,10 @@
 
 #include "conn.h"
 
-int conn_q_recv_packet(struct conn * conn, uint8_t * buffer,int len)
+static int conn_q_recv_packet_(struct conn * conn, uint8_t * buffer,int len,int peek)
 {
 	if ( !conn->cur_packet)
 	{
-/*		while ((conn->cur_packet = conn_q_get_packet(conn)) == 0){*/
 		if ((conn->cur_packet = conn_q_get_packet(conn)) == 0){
 			errno = EAGAIN;
 			return -1;
@@ -38,14 +37,35 @@ int conn_q_recv_packet(struct conn * conn, uint8_t * buffer,int len)
 	if (conn->cur_packet_len > len )
 	{
 		memcpy(buffer,conn->cur_packet+conn->cur_packet_pos,len);
+		if (peek)
+			return len;
+
 		conn->cur_packet_pos+=len;
 		conn->cur_packet_len-=len;
+		if (conn->cur_packet_len==0){
+			free(conn->cur_packet);
+			conn->cur_packet=0;
+		}	
 		return len;
-
 	}
 
 	memcpy(buffer,conn->cur_packet+conn->cur_packet_pos,conn->cur_packet_len);
+	if (peek)
+		return conn->cur_packet_len;
+
 	free (conn->cur_packet);
 	conn->cur_packet=0;
 	return conn->cur_packet_len;	
 }
+
+int conn_q_recv_packet(struct conn * conn, uint8_t * buffer,int len)
+{
+	return conn_q_recv_packet_(conn,buffer,len,0);
+}
+
+int conn_q_recv_packet_peek(struct conn * conn, uint8_t * buffer,int len)
+{
+	return conn_q_recv_packet_(conn,buffer,len,1);
+}
+
+