From 2f81a97f5c55cf23d6de045ba1468f2aa69abf0c Mon Sep 17 00:00:00 2001 From: Matthias Schiffer Date: Fri, 2 Nov 2012 09:02:35 +0100 Subject: Directly use peer when receiving on associated sockets --- src/fastd.c | 18 +++++++++++++----- 1 file changed, 13 insertions(+), 5 deletions(-) (limited to 'src/fastd.c') diff --git a/src/fastd.c b/src/fastd.c index 1d74cb3..f121e49 100644 --- a/src/fastd.c +++ b/src/fastd.c @@ -256,7 +256,7 @@ static void bind_sockets(fastd_context *ctx) { } } -fastd_socket* fastd_socket_open(fastd_context *ctx, const fastd_peer *peer, int af) { +fastd_socket* fastd_socket_open(fastd_context *ctx, fastd_peer *peer, int af) { const fastd_bind_address any_address = { .addr.sa.sa_family = af }; int fd = bind_socket(ctx, &any_address, true); @@ -697,10 +697,18 @@ static void handle_socket(fastd_context *ctx, fastd_socket *sock) { fastd_buffer_push_head(&buffer, 1); - fastd_peer *peer; - for (peer = ctx->peers; peer; peer = peer->next) { - if (fastd_peer_address_equal(&peer->address, &recvaddr)) - break; + fastd_peer *peer = NULL; + + if (sock->peer) { + if (fastd_peer_address_equal(&sock->peer->address, &recvaddr)) { + peer = sock->peer; + } + } + else { + for (peer = ctx->peers; peer; peer = peer->next) { + if (fastd_peer_address_equal(&peer->address, &recvaddr)) + break; + } } if (peer) { -- cgit v1.2.3