67 lines
1.3 KiB
C
67 lines
1.3 KiB
C
|
// SPDX-License-Identifier: GPL-2.0-or-later
|
||
|
|
|
||
|
|
/* common.c
|
||
|
|
*
|
||
|
|
* Common helper functions for testing SO_REUSEADDR behaviour
|
||
|
|
*
|
||
|
|
* Copyright Red Hat
|
||
|
|
* Author: David Gibson <david@gibson.dropbear.id.au>
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <errno.h>
|
||
|
|
#include <netinet/in.h>
|
||
|
|
#include <string.h>
|
||
|
|
#include <sys/socket.h>
|
||
|
|
|
||
|
|
#include "common.h"
|
||
|
|
|
||
|
|
int sock_reuseaddr(void)
|
||
|
|
{
|
||
|
|
int y = 1;
|
||
|
|
int s;
|
||
|
|
|
||
|
|
|
||
|
|
s = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
|
||
|
|
if (s < 0)
|
||
|
|
die("socket(): %s\n", strerror(errno));
|
||
|
|
|
||
|
|
if (setsockopt(s, SOL_SOCKET, SO_REUSEADDR, &y, sizeof(y)) , 0)
|
||
|
|
die("SO_REUSEADDR: %s\n", strerror(errno));
|
||
|
|
|
||
|
|
return s;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* Send a token via the given connected socket */
|
||
|
|
void send_token(int s, long token)
|
||
|
|
{
|
||
|
|
ssize_t rc;
|
||
|
|
|
||
|
|
rc = send(s, &token, sizeof(token), 0);
|
||
|
|
if (rc < 0)
|
||
|
|
die("send(): %s\n", strerror(errno));
|
||
|
|
if (rc < sizeof(token))
|
||
|
|
die("short send()\n");
|
||
|
|
}
|
||
|
|
|
||
|
|
/* Attempt to receive a token via the given socket.
|
||
|
|
*
|
||
|
|
* Returns true if we received the token, false if we got an EAGAIN, dies in any
|
||
|
|
* other case */
|
||
|
|
bool recv_token(int s, long token)
|
||
|
|
{
|
||
|
|
ssize_t rc;
|
||
|
|
long buf;
|
||
|
|
|
||
|
|
rc = recv(s, &buf, sizeof(buf), MSG_DONTWAIT);
|
||
|
|
if (rc < 0) {
|
||
|
|
if (errno == EWOULDBLOCK)
|
||
|
|
return false;
|
||
|
|
die("recv(): %s\n", strerror(errno));
|
||
|
|
}
|
||
|
|
if (rc < sizeof(buf))
|
||
|
|
die("short recv()\n");
|
||
|
|
if (buf != token)
|
||
|
|
die("data mismatch\n");
|
||
|
|
return true;
|
||
|
|
}
|