|  | /* peer.c: Rx RPC peer management | 
|  | * | 
|  | * Copyright (C) 2002 Red Hat, Inc. All Rights Reserved. | 
|  | * Written by David Howells (dhowells@redhat.com) | 
|  | * | 
|  | * This program 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 | 
|  | * 2 of the License, or (at your option) any later version. | 
|  | */ | 
|  |  | 
|  | #include <linux/sched.h> | 
|  | #include <linux/slab.h> | 
|  | #include <linux/module.h> | 
|  | #include <rxrpc/rxrpc.h> | 
|  | #include <rxrpc/transport.h> | 
|  | #include <rxrpc/peer.h> | 
|  | #include <rxrpc/connection.h> | 
|  | #include <rxrpc/call.h> | 
|  | #include <rxrpc/message.h> | 
|  | #include <linux/udp.h> | 
|  | #include <linux/ip.h> | 
|  | #include <net/sock.h> | 
|  | #include <asm/uaccess.h> | 
|  | #include <asm/div64.h> | 
|  | #include "internal.h" | 
|  |  | 
|  | __RXACCT_DECL(atomic_t rxrpc_peer_count); | 
|  | LIST_HEAD(rxrpc_peers); | 
|  | DECLARE_RWSEM(rxrpc_peers_sem); | 
|  | unsigned long rxrpc_peer_timeout = 12 * 60 * 60; | 
|  |  | 
|  | static void rxrpc_peer_do_timeout(struct rxrpc_peer *peer); | 
|  |  | 
|  | static void __rxrpc_peer_timeout(rxrpc_timer_t *timer) | 
|  | { | 
|  | struct rxrpc_peer *peer = | 
|  | list_entry(timer, struct rxrpc_peer, timeout); | 
|  |  | 
|  | _debug("Rx PEER TIMEOUT [%p{u=%d}]", peer, atomic_read(&peer->usage)); | 
|  |  | 
|  | rxrpc_peer_do_timeout(peer); | 
|  | } | 
|  |  | 
|  | static const struct rxrpc_timer_ops rxrpc_peer_timer_ops = { | 
|  | .timed_out	= __rxrpc_peer_timeout, | 
|  | }; | 
|  |  | 
|  | /*****************************************************************************/ | 
|  | /* | 
|  | * create a peer record | 
|  | */ | 
|  | static int __rxrpc_create_peer(struct rxrpc_transport *trans, __be32 addr, | 
|  | struct rxrpc_peer **_peer) | 
|  | { | 
|  | struct rxrpc_peer *peer; | 
|  |  | 
|  | _enter("%p,%08x", trans, ntohl(addr)); | 
|  |  | 
|  | /* allocate and initialise a peer record */ | 
|  | peer = kzalloc(sizeof(struct rxrpc_peer), GFP_KERNEL); | 
|  | if (!peer) { | 
|  | _leave(" = -ENOMEM"); | 
|  | return -ENOMEM; | 
|  | } | 
|  |  | 
|  | atomic_set(&peer->usage, 1); | 
|  |  | 
|  | INIT_LIST_HEAD(&peer->link); | 
|  | INIT_LIST_HEAD(&peer->proc_link); | 
|  | INIT_LIST_HEAD(&peer->conn_idlist); | 
|  | INIT_LIST_HEAD(&peer->conn_active); | 
|  | INIT_LIST_HEAD(&peer->conn_graveyard); | 
|  | spin_lock_init(&peer->conn_gylock); | 
|  | init_waitqueue_head(&peer->conn_gy_waitq); | 
|  | rwlock_init(&peer->conn_idlock); | 
|  | rwlock_init(&peer->conn_lock); | 
|  | atomic_set(&peer->conn_count, 0); | 
|  | spin_lock_init(&peer->lock); | 
|  | rxrpc_timer_init(&peer->timeout, &rxrpc_peer_timer_ops); | 
|  |  | 
|  | peer->addr.s_addr = addr; | 
|  |  | 
|  | peer->trans = trans; | 
|  | peer->ops = trans->peer_ops; | 
|  |  | 
|  | __RXACCT(atomic_inc(&rxrpc_peer_count)); | 
|  | *_peer = peer; | 
|  | _leave(" = 0 (%p)", peer); | 
|  |  | 
|  | return 0; | 
|  | } /* end __rxrpc_create_peer() */ | 
|  |  | 
|  | /*****************************************************************************/ | 
|  | /* | 
|  | * find a peer record on the specified transport | 
|  | * - returns (if successful) with peer record usage incremented | 
|  | * - resurrects it from the graveyard if found there | 
|  | */ | 
|  | int rxrpc_peer_lookup(struct rxrpc_transport *trans, __be32 addr, | 
|  | struct rxrpc_peer **_peer) | 
|  | { | 
|  | struct rxrpc_peer *peer, *candidate = NULL; | 
|  | struct list_head *_p; | 
|  | int ret; | 
|  |  | 
|  | _enter("%p{%hu},%08x", trans, trans->port, ntohl(addr)); | 
|  |  | 
|  | /* [common case] search the transport's active list first */ | 
|  | read_lock(&trans->peer_lock); | 
|  | list_for_each(_p, &trans->peer_active) { | 
|  | peer = list_entry(_p, struct rxrpc_peer, link); | 
|  | if (peer->addr.s_addr == addr) | 
|  | goto found_active; | 
|  | } | 
|  | read_unlock(&trans->peer_lock); | 
|  |  | 
|  | /* [uncommon case] not active - create a candidate for a new record */ | 
|  | ret = __rxrpc_create_peer(trans, addr, &candidate); | 
|  | if (ret < 0) { | 
|  | _leave(" = %d", ret); | 
|  | return ret; | 
|  | } | 
|  |  | 
|  | /* search the active list again, just in case it appeared whilst we | 
|  | * were busy */ | 
|  | write_lock(&trans->peer_lock); | 
|  | list_for_each(_p, &trans->peer_active) { | 
|  | peer = list_entry(_p, struct rxrpc_peer, link); | 
|  | if (peer->addr.s_addr == addr) | 
|  | goto found_active_second_chance; | 
|  | } | 
|  |  | 
|  | /* search the transport's graveyard list */ | 
|  | spin_lock(&trans->peer_gylock); | 
|  | list_for_each(_p, &trans->peer_graveyard) { | 
|  | peer = list_entry(_p, struct rxrpc_peer, link); | 
|  | if (peer->addr.s_addr == addr) | 
|  | goto found_in_graveyard; | 
|  | } | 
|  | spin_unlock(&trans->peer_gylock); | 
|  |  | 
|  | /* we can now add the new candidate to the list | 
|  | * - tell the application layer that this peer has been added | 
|  | */ | 
|  | rxrpc_get_transport(trans); | 
|  | peer = candidate; | 
|  | candidate = NULL; | 
|  |  | 
|  | if (peer->ops && peer->ops->adding) { | 
|  | ret = peer->ops->adding(peer); | 
|  | if (ret < 0) { | 
|  | write_unlock(&trans->peer_lock); | 
|  | __RXACCT(atomic_dec(&rxrpc_peer_count)); | 
|  | kfree(peer); | 
|  | rxrpc_put_transport(trans); | 
|  | _leave(" = %d", ret); | 
|  | return ret; | 
|  | } | 
|  | } | 
|  |  | 
|  | atomic_inc(&trans->peer_count); | 
|  |  | 
|  | make_active: | 
|  | list_add_tail(&peer->link, &trans->peer_active); | 
|  |  | 
|  | success_uwfree: | 
|  | write_unlock(&trans->peer_lock); | 
|  |  | 
|  | if (candidate) { | 
|  | __RXACCT(atomic_dec(&rxrpc_peer_count)); | 
|  | kfree(candidate); | 
|  | } | 
|  |  | 
|  | if (list_empty(&peer->proc_link)) { | 
|  | down_write(&rxrpc_peers_sem); | 
|  | list_add_tail(&peer->proc_link, &rxrpc_peers); | 
|  | up_write(&rxrpc_peers_sem); | 
|  | } | 
|  |  | 
|  | success: | 
|  | *_peer = peer; | 
|  |  | 
|  | _leave(" = 0 (%p{u=%d cc=%d})", | 
|  | peer, | 
|  | atomic_read(&peer->usage), | 
|  | atomic_read(&peer->conn_count)); | 
|  | return 0; | 
|  |  | 
|  | /* handle the peer being found in the active list straight off */ | 
|  | found_active: | 
|  | rxrpc_get_peer(peer); | 
|  | read_unlock(&trans->peer_lock); | 
|  | goto success; | 
|  |  | 
|  | /* handle resurrecting a peer from the graveyard */ | 
|  | found_in_graveyard: | 
|  | rxrpc_get_peer(peer); | 
|  | rxrpc_get_transport(peer->trans); | 
|  | rxrpc_krxtimod_del_timer(&peer->timeout); | 
|  | list_del_init(&peer->link); | 
|  | spin_unlock(&trans->peer_gylock); | 
|  | goto make_active; | 
|  |  | 
|  | /* handle finding the peer on the second time through the active | 
|  | * list */ | 
|  | found_active_second_chance: | 
|  | rxrpc_get_peer(peer); | 
|  | goto success_uwfree; | 
|  |  | 
|  | } /* end rxrpc_peer_lookup() */ | 
|  |  | 
|  | /*****************************************************************************/ | 
|  | /* | 
|  | * finish with a peer record | 
|  | * - it gets sent to the graveyard from where it can be resurrected or timed | 
|  | *   out | 
|  | */ | 
|  | void rxrpc_put_peer(struct rxrpc_peer *peer) | 
|  | { | 
|  | struct rxrpc_transport *trans = peer->trans; | 
|  |  | 
|  | _enter("%p{cc=%d a=%08x}", | 
|  | peer, | 
|  | atomic_read(&peer->conn_count), | 
|  | ntohl(peer->addr.s_addr)); | 
|  |  | 
|  | /* sanity check */ | 
|  | if (atomic_read(&peer->usage) <= 0) | 
|  | BUG(); | 
|  |  | 
|  | write_lock(&trans->peer_lock); | 
|  | spin_lock(&trans->peer_gylock); | 
|  | if (likely(!atomic_dec_and_test(&peer->usage))) { | 
|  | spin_unlock(&trans->peer_gylock); | 
|  | write_unlock(&trans->peer_lock); | 
|  | _leave(""); | 
|  | return; | 
|  | } | 
|  |  | 
|  | /* move to graveyard queue */ | 
|  | list_del(&peer->link); | 
|  | write_unlock(&trans->peer_lock); | 
|  |  | 
|  | list_add_tail(&peer->link, &trans->peer_graveyard); | 
|  |  | 
|  | BUG_ON(!list_empty(&peer->conn_active)); | 
|  |  | 
|  | rxrpc_krxtimod_add_timer(&peer->timeout, rxrpc_peer_timeout * HZ); | 
|  |  | 
|  | spin_unlock(&trans->peer_gylock); | 
|  |  | 
|  | rxrpc_put_transport(trans); | 
|  |  | 
|  | _leave(" [killed]"); | 
|  | } /* end rxrpc_put_peer() */ | 
|  |  | 
|  | /*****************************************************************************/ | 
|  | /* | 
|  | * handle a peer timing out in the graveyard | 
|  | * - called from krxtimod | 
|  | */ | 
|  | static void rxrpc_peer_do_timeout(struct rxrpc_peer *peer) | 
|  | { | 
|  | struct rxrpc_transport *trans = peer->trans; | 
|  |  | 
|  | _enter("%p{u=%d cc=%d a=%08x}", | 
|  | peer, | 
|  | atomic_read(&peer->usage), | 
|  | atomic_read(&peer->conn_count), | 
|  | ntohl(peer->addr.s_addr)); | 
|  |  | 
|  | BUG_ON(atomic_read(&peer->usage) < 0); | 
|  |  | 
|  | /* remove from graveyard if still dead */ | 
|  | spin_lock(&trans->peer_gylock); | 
|  | if (atomic_read(&peer->usage) == 0) | 
|  | list_del_init(&peer->link); | 
|  | else | 
|  | peer = NULL; | 
|  | spin_unlock(&trans->peer_gylock); | 
|  |  | 
|  | if (!peer) { | 
|  | _leave(""); | 
|  | return; /* resurrected */ | 
|  | } | 
|  |  | 
|  | /* clear all connections on this peer */ | 
|  | rxrpc_conn_clearall(peer); | 
|  |  | 
|  | BUG_ON(!list_empty(&peer->conn_active)); | 
|  | BUG_ON(!list_empty(&peer->conn_graveyard)); | 
|  |  | 
|  | /* inform the application layer */ | 
|  | if (peer->ops && peer->ops->discarding) | 
|  | peer->ops->discarding(peer); | 
|  |  | 
|  | if (!list_empty(&peer->proc_link)) { | 
|  | down_write(&rxrpc_peers_sem); | 
|  | list_del(&peer->proc_link); | 
|  | up_write(&rxrpc_peers_sem); | 
|  | } | 
|  |  | 
|  | __RXACCT(atomic_dec(&rxrpc_peer_count)); | 
|  | kfree(peer); | 
|  |  | 
|  | /* if the graveyard is now empty, wake up anyone waiting for that */ | 
|  | if (atomic_dec_and_test(&trans->peer_count)) | 
|  | wake_up(&trans->peer_gy_waitq); | 
|  |  | 
|  | _leave(" [destroyed]"); | 
|  | } /* end rxrpc_peer_do_timeout() */ | 
|  |  | 
|  | /*****************************************************************************/ | 
|  | /* | 
|  | * clear all peer records from a transport endpoint | 
|  | */ | 
|  | void rxrpc_peer_clearall(struct rxrpc_transport *trans) | 
|  | { | 
|  | DECLARE_WAITQUEUE(myself,current); | 
|  |  | 
|  | struct rxrpc_peer *peer; | 
|  | int err; | 
|  |  | 
|  | _enter("%p",trans); | 
|  |  | 
|  | /* there shouldn't be any active peers remaining */ | 
|  | BUG_ON(!list_empty(&trans->peer_active)); | 
|  |  | 
|  | /* manually timeout all peers in the graveyard */ | 
|  | spin_lock(&trans->peer_gylock); | 
|  | while (!list_empty(&trans->peer_graveyard)) { | 
|  | peer = list_entry(trans->peer_graveyard.next, | 
|  | struct rxrpc_peer, link); | 
|  | _debug("Clearing peer %p\n", peer); | 
|  | err = rxrpc_krxtimod_del_timer(&peer->timeout); | 
|  | spin_unlock(&trans->peer_gylock); | 
|  |  | 
|  | if (err == 0) | 
|  | rxrpc_peer_do_timeout(peer); | 
|  |  | 
|  | spin_lock(&trans->peer_gylock); | 
|  | } | 
|  | spin_unlock(&trans->peer_gylock); | 
|  |  | 
|  | /* wait for the the peer graveyard to be completely cleared */ | 
|  | set_current_state(TASK_UNINTERRUPTIBLE); | 
|  | add_wait_queue(&trans->peer_gy_waitq, &myself); | 
|  |  | 
|  | while (atomic_read(&trans->peer_count) != 0) { | 
|  | schedule(); | 
|  | set_current_state(TASK_UNINTERRUPTIBLE); | 
|  | } | 
|  |  | 
|  | remove_wait_queue(&trans->peer_gy_waitq, &myself); | 
|  | set_current_state(TASK_RUNNING); | 
|  |  | 
|  | _leave(""); | 
|  | } /* end rxrpc_peer_clearall() */ | 
|  |  | 
|  | /*****************************************************************************/ | 
|  | /* | 
|  | * calculate and cache the Round-Trip-Time for a message and its response | 
|  | */ | 
|  | void rxrpc_peer_calculate_rtt(struct rxrpc_peer *peer, | 
|  | struct rxrpc_message *msg, | 
|  | struct rxrpc_message *resp) | 
|  | { | 
|  | unsigned long long rtt; | 
|  | int loop; | 
|  |  | 
|  | _enter("%p,%p,%p", peer, msg, resp); | 
|  |  | 
|  | /* calculate the latest RTT */ | 
|  | rtt = resp->stamp.tv_sec - msg->stamp.tv_sec; | 
|  | rtt *= 1000000UL; | 
|  | rtt += resp->stamp.tv_usec - msg->stamp.tv_usec; | 
|  |  | 
|  | /* add to cache */ | 
|  | peer->rtt_cache[peer->rtt_point] = rtt; | 
|  | peer->rtt_point++; | 
|  | peer->rtt_point %= RXRPC_RTT_CACHE_SIZE; | 
|  |  | 
|  | if (peer->rtt_usage < RXRPC_RTT_CACHE_SIZE) | 
|  | peer->rtt_usage++; | 
|  |  | 
|  | /* recalculate RTT */ | 
|  | rtt = 0; | 
|  | for (loop = peer->rtt_usage - 1; loop >= 0; loop--) | 
|  | rtt += peer->rtt_cache[loop]; | 
|  |  | 
|  | do_div(rtt, peer->rtt_usage); | 
|  | peer->rtt = rtt; | 
|  |  | 
|  | _leave(" RTT=%lu.%lums", | 
|  | (long) (peer->rtt / 1000), (long) (peer->rtt % 1000)); | 
|  |  | 
|  | } /* end rxrpc_peer_calculate_rtt() */ |