77 lines
1.8 KiB
C
77 lines
1.8 KiB
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;
|
|
|
|
}
|