mir/criu
2
0
Fork 0
mirror of https://github.com/checkpoint-restore/criu synced 2025-08-29 13:28:27 +00:00
Code Issues Releases Wiki Activity

inet: Factor out sockaddr-s preparations

Browse Source 
Signed-off-by: Pavel Emelyanov <xemul@parallels.com>
This commit is contained in:
Pavel Emelyanov 2012-08-20 18:11:35 +04:00
parent 7cb674109b
commit a8e056bfc8
1 changed files with 39 additions and 44 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

83
sk-inet.c
View File

@ -517,32 +517,44 @@ err:
return -1; return -1;
} }
union sockaddr_inet {
struct sockaddr_in v4;
struct sockaddr_in6 v6;
};
static int restore_sockaddr(union sockaddr_inet *sa,
int family, uint32_t pb_port, uint32_t *pb_addr)
{
BUILD_BUG_ON(sizeof(sa->v4.sin_addr.s_addr) > PB_ALEN_INET * sizeof(uint32_t));
BUILD_BUG_ON(sizeof(sa->v6.sin6_addr.s6_addr) > PB_ALEN_INET6 * sizeof(uint32_t));
memzero(sa, sizeof(*sa));
if (family == AF_INET) {
sa->v4.sin_family = AF_INET;
sa->v4.sin_port = htons(pb_port);
memcpy(&sa->v4.sin_addr.s_addr, pb_addr, sizeof(sa->v4.sin_addr.s_addr));
return sizeof(sa->v4);
}
if (family == AF_INET6) {
sa->v6.sin6_family = AF_INET6;
sa->v6.sin6_port = htons(pb_port);
memcpy(sa->v6.sin6_addr.s6_addr, pb_addr, sizeof(sa->v6.sin6_addr.s6_addr));
return sizeof(sa->v6);
}
BUG();
return -1;
}
int inet_bind(int sk, struct inet_sk_info *ii) int inet_bind(int sk, struct inet_sk_info *ii)
{ {
union { union sockaddr_inet addr;
struct sockaddr_in v4; int addr_size;
struct sockaddr_in6 v6;
} addr;
int addr_size = 0;
BUILD_BUG_ON(sizeof(addr.v4.sin_addr.s_addr) > PB_ALEN_INET * sizeof(uint32_t)); addr_size = restore_sockaddr(&addr, ii->ie->family,
BUILD_BUG_ON(sizeof(addr.v6.sin6_addr.s6_addr) > PB_ALEN_INET6 * sizeof(uint32_t)); ii->ie->src_port, ii->ie->src_addr);
memzero(&addr, sizeof(addr));
if (ii->ie->family == AF_INET) {
addr.v4.sin_family = ii->ie->family;
addr.v4.sin_port = htons(ii->ie->src_port);
memcpy(&addr.v4.sin_addr.s_addr, ii->ie->src_addr, sizeof(addr.v4.sin_addr.s_addr));
addr_size = sizeof(addr.v4);
} else if (ii->ie->family == AF_INET6) {
addr.v6.sin6_family = ii->ie->family;
addr.v6.sin6_port = htons(ii->ie->src_port);
memcpy(&addr.v6.sin6_addr.s6_addr, ii->ie->src_addr, sizeof(addr.v6.sin6_addr.s6_addr));
addr_size = sizeof(addr.v6);
} else {
pr_perror("Unsupported address family: %d\n", ii->ie->family);
return -1;
}
if (bind(sk, (struct sockaddr *)&addr, addr_size) == -1) { if (bind(sk, (struct sockaddr *)&addr, addr_size) == -1) {
pr_perror("Can't bind inet socket"); pr_perror("Can't bind inet socket");
@ -554,28 +566,11 @@ int inet_bind(int sk, struct inet_sk_info *ii)
int inet_connect(int sk, struct inet_sk_info *ii) int inet_connect(int sk, struct inet_sk_info *ii)
{ {
union { union sockaddr_inet addr;
struct sockaddr_in v4; int addr_size;
struct sockaddr_in6 v6;
} addr;
int addr_size = 0;
addr_size = restore_sockaddr(&addr, ii->ie->family,
memzero(&addr, sizeof(addr)); ii->ie->dst_port, ii->ie->dst_addr);
if (ii->ie->family == AF_INET) {
addr.v4.sin_family = ii->ie->family;
addr.v4.sin_port = htons(ii->ie->dst_port);
memcpy(&addr.v4.sin_addr.s_addr, ii->ie->dst_addr, sizeof(addr.v4.sin_addr.s_addr));
addr_size = sizeof(addr.v4);
} else if (ii->ie->family == AF_INET6) {
addr.v6.sin6_family = ii->ie->family;
addr.v6.sin6_port = htons(ii->ie->dst_port);
memcpy(&addr.v6.sin6_addr.s6_addr, ii->ie->dst_addr, sizeof(addr.v6.sin6_addr.s6_addr));
addr_size = sizeof(addr.v6);
} else {
pr_perror("Unsupported address family: %d\n", ii->ie->family);
return -1;
}
if (connect(sk, (struct sockaddr *)&addr, addr_size) == -1) { if (connect(sk, (struct sockaddr *)&addr, addr_size) == -1) {
pr_perror("Can't connect inet socket back"); pr_perror("Can't connect inet socket back");