actube/src/capwap/conn_recv_packet.c

/*
    This file is part of libcapwap.

    libcapwap is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    libcapwap is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with Foobar.  If not, see <http://www.gnu.org/licenses/>.

*/

#include <time.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <errno.h>
#include <string.h>

#include "conn.h"

#include "sock.h"

int conn_recv_packet_(struct conn *conn, uint8_t * buf, int len, int flags)
{
	int n;
	while ((n = recv(conn->sock, (char *) buf, len, flags)) < 0) {
		if (errno != EINTR) {
			if (errno == EAGAIN)
				return n;

		}
	}
	return n;
}


int conn_recvfrom_packet(struct conn *conn, uint8_t * buf, int len,
			 struct sockaddr_storage *from)
{
	socklen_t al;

	al = sizeof(struct sockaddr_storage);
	memset(from, 0, sizeof(struct sockaddr_storage));

	int n;
	while ((n = recvfrom(conn->sock, (char *) buf, len, 0, (struct sockaddr*)from, &al)) < 0) {
		if (errno != EINTR) {
			if (errno == EAGAIN)
				return n;

		}

	}
	return n;

}

/* yes, these functions could be better defined as macros in a .h file */

int conn_recv_packet(struct conn *conn, uint8_t * buf, int len)
{
	return conn_recv_packet_(conn, buf, len, 0);
}

int conn_recv_packet_peek(struct conn *conn, uint8_t * buf, int len)
{
	int rc = conn_recv_packet_(conn, buf, len, MSG_PEEK);
	return rc;

}