diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/fastd.c | 34 |
1 files changed, 22 insertions, 12 deletions
diff --git a/src/fastd.c b/src/fastd.c index 7bcfdd8..876975f 100644 --- a/src/fastd.c +++ b/src/fastd.c @@ -237,6 +237,8 @@ static void init_peers(fastd_context *ctx) { } } + + static void* get_source_address(const fastd_context *ctx, void *buffer) { switch (ctx->conf->protocol) { case PROTOCOL_ETHERNET: @@ -375,21 +377,29 @@ static void handle_input(fastd_context *ctx) { if (len < 0) pr_warn(ctx, "recvfrom: %s", strerror(errno)); - // TODO get correct peer - fastd_peer *peer = ctx->peers; - + fastd_peer *peer; + for (peer = ctx->peers; peer; peer = peer->next) { + if (recvaddr.sin_addr.s_addr == peer->address && recvaddr.sin_port == peer->port) + break; + } - switch (packet_type) { - case 0: - buffer.len = len - 1; - ctx->conf->method->method_handle_recv(ctx, peer, buffer); - break; + if (peer) { + switch (packet_type) { + case 0: + buffer.len = len - 1; + ctx->conf->method->method_handle_recv(ctx, peer, buffer); + break; - case 1: - fastd_handshake_handle(ctx, peer, buffer); - break; + case 1: + fastd_handshake_handle(ctx, peer, buffer); + break; - default: + default: + fastd_buffer_free(buffer); + } + } + else { + pr_debug(ctx, "received packet from unknown peer"); fastd_buffer_free(buffer); } } |