socket: use safe_container_of() in ep_sock_find_in_tree()

This commit is contained in:
Bert Belder 2017-09-25 17:52:08 +02:00
parent 1e24f44238
commit b9f4578a96

View File

@ -231,11 +231,7 @@ void ep_sock_force_delete(ep_port_t* port_info, ep_sock_t* sock_info) {
}
ep_sock_t* ep_sock_find_in_tree(tree_t* tree, SOCKET socket) {
tree_node_t* tree_node = tree_find(tree, socket);
if (tree_node == NULL)
return NULL;
return container_of(tree_node, ep_sock_t, tree_node);
return safe_container_of(tree_find(tree, socket), ep_sock_t, tree_node);
}
ep_sock_t* ep_sock_from_overlapped(OVERLAPPED* overlapped) {