Commit d87b503f authored by Jean-Philippe Brucker's avatar Jean-Philippe Brucker Committed by Will Deacon
Browse files

net/uip: Add exit function



When resetting the virtio-net queues, the UIP state needs to be reset as
well. Stop all threads (one per TCP stream and UDP connection) and free
memory.
Signed-off-by: default avatarJean-Philippe Brucker <jean-philippe.brucker@arm.com>
Signed-off-by: default avatarJulien Thierry <julien.thierry@arm.com>
Signed-off-by: default avatarWill Deacon <will.deacon@arm.com>
parent eb34a8c2
......@@ -196,6 +196,7 @@ struct uip_info {
struct list_head buf_head;
struct mutex buf_lock;
pthread_t udp_thread;
u8 *udp_buf;
int udp_epollfd;
int buf_free_nr;
int buf_used_nr;
......@@ -249,6 +250,7 @@ struct uip_tcp_socket {
int read_done;
u32 dip, sip;
u8 *payload;
u8 *buf;
int fd;
};
......@@ -336,6 +338,9 @@ int uip_tx(struct iovec *iov, u16 out, struct uip_info *info);
int uip_rx(struct iovec *iov, u16 in, struct uip_info *info);
void uip_static_init(struct uip_info *info);
int uip_init(struct uip_info *info);
void uip_exit(struct uip_info *info);
void uip_tcp_exit(struct uip_info *info);
void uip_udp_exit(struct uip_info *info);
int uip_tx_do_ipv4_udp_dhcp(struct uip_tx_arg *arg);
int uip_tx_do_ipv4_icmp(struct uip_tx_arg *arg);
......@@ -359,4 +364,5 @@ int uip_udp_make_pkg(struct uip_info *info, struct uip_udp_socket *sk, struct ui
bool uip_udp_is_dhcp(struct uip_udp *udp);
int uip_dhcp_get_dns(struct uip_info *info);
void uip_dhcp_exit(struct uip_info *info);
#endif /* KVM__UIP_H */
......@@ -154,3 +154,20 @@ int uip_init(struct uip_info *info)
return 0;
}
void uip_exit(struct uip_info *info)
{
struct uip_buf *buf, *next;
uip_udp_exit(info);
uip_tcp_exit(info);
uip_dhcp_exit(info);
list_for_each_entry_safe(buf, next, &info->buf_head, list) {
free(buf->vnet);
free(buf->eth);
list_del(&buf->list);
free(buf);
}
uip_static_init(info);
}
......@@ -200,3 +200,9 @@ int uip_tx_do_ipv4_udp_dhcp(struct uip_tx_arg *arg)
return 0;
}
void uip_dhcp_exit(struct uip_info *info)
{
free(info->domain_name);
info->domain_name = NULL;
}
......@@ -18,6 +18,7 @@ static int uip_tcp_socket_close(struct uip_tcp_socket *sk, int how)
list_del(&sk->list);
mutex_unlock(sk->lock);
free(sk->buf);
free(sk);
}
......@@ -94,6 +95,24 @@ static struct uip_tcp_socket *uip_tcp_socket_alloc(struct uip_tx_arg *arg, u32 s
return sk;
}
/* Caller holds the sk lock */
static void uip_tcp_socket_free(struct uip_tcp_socket *sk)
{
/*
* Here we assume that the virtqueues are already inactive so we don't
* race with uip_tx_do_ipv4_tcp. We are racing with
* uip_tcp_socket_thread though, but holding the sk lock ensures that it
* cannot free data concurrently.
*/
if (sk->thread) {
pthread_cancel(sk->thread);
pthread_join(sk->thread, NULL);
}
sk->write_done = sk->read_done = 1;
uip_tcp_socket_close(sk, SHUT_RDWR);
}
static int uip_tcp_payload_send(struct uip_tcp_socket *sk, u8 flag, u16 payload_len)
{
struct uip_info *info;
......@@ -175,20 +194,16 @@ static void *uip_tcp_socket_thread(void *p)
{
struct uip_tcp_socket *sk;
int len, left, ret;
u8 *payload, *pos;
u8 *pos;
kvm__set_thread_name("uip-tcp");
sk = p;
payload = malloc(UIP_MAX_TCP_PAYLOAD);
if (!payload)
goto out;
while (1) {
pos = payload;
pos = sk->buf;
ret = read(sk->fd, payload, UIP_MAX_TCP_PAYLOAD);
ret = read(sk->fd, sk->buf, UIP_MAX_TCP_PAYLOAD);
if (ret <= 0 || ret > UIP_MAX_TCP_PAYLOAD)
goto out;
......@@ -224,7 +239,6 @@ out:
sk->read_done = 1;
free(payload);
pthread_exit(NULL);
return NULL;
......@@ -232,8 +246,18 @@ out:
static int uip_tcp_socket_receive(struct uip_tcp_socket *sk)
{
if (sk->thread == 0)
return pthread_create(&sk->thread, NULL, uip_tcp_socket_thread, (void *)sk);
int ret;
if (sk->thread == 0) {
sk->buf = malloc(UIP_MAX_TCP_PAYLOAD);
if (!sk->buf)
return -ENOMEM;
ret = pthread_create(&sk->thread, NULL, uip_tcp_socket_thread,
(void *)sk);
if (ret)
free(sk->buf);
return ret;
}
return 0;
}
......@@ -346,3 +370,13 @@ int uip_tx_do_ipv4_tcp(struct uip_tx_arg *arg)
out:
return 0;
}
void uip_tcp_exit(struct uip_info *info)
{
struct uip_tcp_socket *sk, *next;
mutex_lock(&info->tcp_socket_lock);
list_for_each_entry_safe(sk, next, &info->tcp_socket_head, list)
uip_tcp_socket_free(sk);
mutex_unlock(&info->tcp_socket_lock);
}
......@@ -164,10 +164,7 @@ static void *uip_udp_socket_thread(void *p)
kvm__set_thread_name("uip-udp");
info = p;
do {
payload = malloc(UIP_MAX_UDP_PAYLOAD);
} while (!payload);
payload = info->udp_buf;
while (1) {
nfds = epoll_wait(info->udp_epollfd, events, UIP_UDP_MAX_EVENTS, -1);
......@@ -196,7 +193,11 @@ static void *uip_udp_socket_thread(void *p)
}
}
free(payload);
mutex_lock(&info->udp_socket_lock);
free(info->udp_buf);
info->udp_buf = NULL;
mutex_unlock(&info->udp_socket_lock);
pthread_exit(NULL);
return NULL;
}
......@@ -232,8 +233,36 @@ int uip_tx_do_ipv4_udp(struct uip_tx_arg *arg)
if (ret)
return -1;
if (!info->udp_thread)
if (!info->udp_thread) {
info->udp_buf = malloc(UIP_MAX_UDP_PAYLOAD);
if (!info->udp_buf)
return -1;
pthread_create(&info->udp_thread, NULL, uip_udp_socket_thread, (void *)info);
}
return 0;
}
void uip_udp_exit(struct uip_info *info)
{
struct uip_udp_socket *sk, *next;
mutex_lock(&info->udp_socket_lock);
if (info->udp_thread) {
pthread_cancel(info->udp_thread);
pthread_join(info->udp_thread, NULL);
info->udp_thread = 0;
free(info->udp_buf);
}
if (info->udp_epollfd > 0) {
close(info->udp_epollfd);
info->udp_epollfd = 0;
}
list_for_each_entry_safe(sk, next, &info->udp_socket_head, list) {
close(sk->fd);
free(sk);
}
mutex_unlock(&info->udp_socket_lock);
}
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment