/* arp.c - arp ping checking
 *
 * Copyright (c) 2010-2014 Nicholas J. Kain <njkain at gmail dot com>
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * - Redistributions of source code must retain the above copyright notice,
 *   this list of conditions and the following disclaimer.
 *
 * - Redistributions in binary form must reproduce the above copyright notice,
 *   this list of conditions and the following disclaimer in the documentation
 *   and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */
#include <stdbool.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <limits.h>
#include <netinet/if_ether.h>
#include <arpa/inet.h>
#include <linux/if_packet.h>
#include <linux/filter.h>
#include <errno.h>
#include "nk/log.h"
#include "nk/io.h"
#include "arp.h"
#include "state.h"
#include "dhcp.h"
#include "sys.h"
#include "ifchange.h"
#include "options.h"
#include "leasefile.h"
#include "sockd.h"

#define ARP_MSG_SIZE 0x2a
#define ARP_RETRANS_DELAY 5000 // ms

// From RFC5227
int arp_probe_wait = 1000;         // initial random delay (ms)
int arp_probe_num = 3;             // number of probe packets
int arp_probe_min = 1000;          // minimum delay until repeated probe (ms)
int arp_probe_max = 2000;          // maximum delay until repeated probe (ms)
#define ANNOUNCE_WAIT 2000         // delay before announcing
#define ANNOUNCE_NUM 2             // number of Announcement packets
#define ANNOUNCE_INTERVAL 2000     // time between Announcement packets
#define MAX_CONFLICTS 10           // max conflicts before rate-limiting
#define RATE_LIMIT_INTERVAL 60000  // delay between successive attempts
#define DEFEND_INTERVAL 10000      // minimum interval between defensive ARPs


typedef enum {
    AS_NONE = 0,        // Nothing to react to wrt ARP
    AS_COLLISION_CHECK, // Checking to see if another host has our IP before
                        // accepting a new lease.
    AS_GW_CHECK,        // Seeing if the default GW still exists on the local
                        // segment after the hardware link was lost.
    AS_GW_QUERY,        // Finding the default GW MAC address.
    AS_DEFENSE,         // Defending our IP address (RFC5227)
    AS_MAX,
} arp_state_t;

static long long arp_wake_ts[AS_MAX] = { -1, -1, -1, -1, -1 };

typedef enum {
    ASEND_COLLISION_CHECK,
    ASEND_GW_PING,
    ASEND_ANNOUNCE,
    ASEND_MAX,
} arp_send_t;

static arp_state_t arpState;
struct arp_stats {
    long long ts;
    int count;
};
static struct arp_stats arp_send_stats[ASEND_MAX];
static int using_arp_bpf; // Is a BPF installed on the ARP socket?

int arp_relentless_def; // Don't give up defense no matter what.
static long long last_conflict_ts; // TS of the last conflicting ARP seen.

static int gw_check_init_pingcount; // Initial count of ASEND_GW_PING when
                                    // AS_GW_CHECK was entered.

static uint16_t probe_wait_time; // Time to wait for a COLLISION_CHECK reply.
static long long arp_check_start_ts; // TS of when we started the
                                     // AS_COLLISION_CHECK state.

static unsigned int total_conflicts; // Total number of address conflicts on
                                     // the interface.  Never decreases.

static struct dhcpmsg arp_dhcp_packet; // Used only for AS_COLLISION_CHECK

static char arp_router_has_replied;
static char arp_server_has_replied;

static struct arpMsg arpreply;
static size_t arpreply_offset;
static void arpreply_clear(void)
{
    memset(&arpreply, 0, sizeof arpreply);
    arpreply_offset = 0;
}

void arp_reset_send_stats(void)
{
    for (int i = 0; i < ASEND_MAX; ++i) {
        arp_send_stats[i].ts = 0;
        arp_send_stats[i].count = 0;
    }
}

static int get_arp_basic_socket(void)
{
    char resp;
    int fd = request_sockd_fd("a", 1, &resp);
    switch (resp) {
        case 'A': using_arp_bpf = 1; break;
        case 'a': using_arp_bpf = 0; break;
        default: suicide("%s: (%s) expected a or A sockd reply but got %c",
                         client_config.interface, __func__, resp);
    }
    return fd;
}

static int get_arp_defense_socket(struct client_state_t *cs)
{
    char buf[32];
    size_t buflen = 0;
    buf[0] = 'd';
    buflen += 1;
    memcpy(buf + buflen, &cs->clientAddr, sizeof cs->clientAddr);
    buflen += sizeof cs->clientAddr;
    memcpy(buf + buflen, client_config.arp, 6);
    buflen += 6;
    char resp;
    int fd = request_sockd_fd(buf, buflen, &resp);
    switch (resp) {
        case 'D': using_arp_bpf = 1; break;
        case 'd': using_arp_bpf = 0; break;
        default: suicide("%s: (%s) expected d or D sockd reply but got %c",
                         client_config.interface, __func__, resp);
    }
    return fd;
}

static int arp_open_fd(struct client_state_t *cs, arp_state_t state)
{
    if (cs->arpFd >= 0) {
        log_warning("%s: (%s) called but fd already exists",
                    client_config.interface, __func__);
        return 0;
    }
    switch (state) {
    default:
        log_warning("%s: (%s) called for 'default' state",
                    client_config.interface, __func__);
        return 0;
    case AS_COLLISION_CHECK:
    case AS_GW_QUERY:
    case AS_GW_CHECK: cs->arpFd = get_arp_basic_socket(); break;
    case AS_DEFENSE: cs->arpFd = get_arp_defense_socket(cs); break;
    }
    if (cs->arpFd < 0) {
        log_error("arp: Failed to create socket: %s", strerror(errno));
        return -1;
    }
    epoll_add(cs->epollFd, cs->arpFd);
    arpreply_clear();
    return 0;
}

static void arp_min_close_fd(struct client_state_t *cs)
{
    if (cs->arpFd < 0)
        return;
    epoll_del(cs->epollFd, cs->arpFd);
    close(cs->arpFd);
    cs->arpFd = -1;
    arpState = AS_NONE;
}

static void arp_switch_state(struct client_state_t *cs, arp_state_t state)
{
    if (arpState == state || arpState >= AS_MAX)
        return;
    if (state == AS_NONE) {
        arp_close_fd(cs);
        return;
    }
    bool force_reopen = state == AS_DEFENSE || arpState == AS_DEFENSE;
    if (force_reopen)
        arp_min_close_fd(cs);
    if (cs->arpFd < 0 || force_reopen) {
        if (arp_open_fd(cs, state) < 0)
            suicide("arp: Failed to open arpFd when changing state %u -> %u",
                    arpState, state);
    }
    arpState = state;
}

void arp_close_fd(struct client_state_t *cs)
{
    arp_min_close_fd(cs);
    for (int i = 0; i < AS_MAX; ++i)
        arp_wake_ts[i] = -1;
}

static void arp_reopen_fd(struct client_state_t *cs)
{
    arp_state_t prev_state = arpState;
    arp_min_close_fd(cs);
    arp_switch_state(cs, prev_state);
}

static int arp_send(struct client_state_t *cs, struct arpMsg *arp)
{
    struct sockaddr_ll addr = {
        .sll_family = AF_PACKET,
        .sll_ifindex = client_config.ifindex,
        .sll_halen = 6,
    };
    memcpy(addr.sll_addr, client_config.arp, 6);

    if (cs->arpFd < 0) {
        log_warning("arp: Send attempted when no ARP fd is open.");
        return -1;
    }

    if (safe_sendto(cs->arpFd, (const char *)arp, sizeof *arp,
                    0, (struct sockaddr *)&addr, sizeof addr) < 0) {
        log_error("arp: sendto failed: %s", strerror(errno));
        arp_reopen_fd(cs);
        return -1;
    }
    return 0;
}

#define BASE_ARPMSG() struct arpMsg arp = {                             \
        .h_proto = htons(ETH_P_ARP),                                    \
        .htype = htons(ARPHRD_ETHER),                                   \
        .ptype = htons(ETH_P_IP),                                       \
        .hlen = 6, .plen = 4,                                           \
        .operation = htons(ARPOP_REQUEST),                              \
        .smac = {0},                                                    \
    };                                                                  \
    memcpy(arp.h_source, client_config.arp, 6);                         \
    memset(arp.h_dest, 0xff, 6);                                        \
    memcpy(arp.smac, client_config.arp, 6)

// Returns 0 on success, -1 on failure.
static int arp_ping(struct client_state_t *cs, uint32_t test_ip)
{
    BASE_ARPMSG();
    memcpy(arp.sip4, &cs->clientAddr, sizeof cs->clientAddr);
    memcpy(arp.dip4, &test_ip, sizeof test_ip);
    if (arp_send(cs, &arp) < 0)
        return -1;
    arp_send_stats[ASEND_GW_PING].count++;
    arp_send_stats[ASEND_GW_PING].ts = curms();
    return 0;
}

// Returns 0 on success, -1 on failure.
static int arp_ip_anon_ping(struct client_state_t *cs, uint32_t test_ip)
{
    BASE_ARPMSG();
    memcpy(arp.dip4, &test_ip, sizeof test_ip);
    log_line("arp: Probing for hosts that may conflict with our lease...");
    if (arp_send(cs, &arp) < 0)
        return -1;
    arp_send_stats[ASEND_COLLISION_CHECK].count++;
    arp_send_stats[ASEND_COLLISION_CHECK].ts = curms();
    return 0;
}

static int arp_announcement(struct client_state_t *cs)
{
    BASE_ARPMSG();
    memcpy(arp.sip4, &cs->clientAddr, 4);
    memcpy(arp.dip4, &cs->clientAddr, 4);
    if (arp_send(cs, &arp) < 0)
        return -1;
    arp_send_stats[ASEND_ANNOUNCE].count++;
    arp_send_stats[ASEND_ANNOUNCE].ts = curms();
    return 0;
}
#undef BASE_ARPMSG

// Callable from DS_REQUESTING, DS_RENEWING, or DS_REBINDING via an_packet()
int arp_check(struct client_state_t *cs, struct dhcpmsg *packet)
{
    memcpy(&arp_dhcp_packet, packet, sizeof (struct dhcpmsg));
    arp_switch_state(cs, AS_COLLISION_CHECK);
    if (arp_ip_anon_ping(cs, arp_dhcp_packet.yiaddr) < 0)
        return -1;
    cs->arpPrevState = cs->dhcpState;
    cs->dhcpState = DS_COLLISION_CHECK;
    arp_check_start_ts = arp_send_stats[ASEND_COLLISION_CHECK].ts;
    probe_wait_time = arp_probe_wait;
    arp_wake_ts[AS_COLLISION_CHECK] = arp_check_start_ts + probe_wait_time;
    return 0;
}

// Callable only from DS_BOUND via state.c:ifup_action().
int arp_gw_check(struct client_state_t *cs)
{
    if (arpState == AS_GW_CHECK)  // Guard against state bounce.
        return 0;
    gw_check_init_pingcount = arp_send_stats[ASEND_GW_PING].count;
    arp_server_has_replied = 0;
    if (arp_ping(cs, cs->serverAddr) < 0)
        return -1;
    if (cs->routerAddr) {
        arp_router_has_replied = 0;
        if (arp_ping(cs, cs->routerAddr) < 0)
            return -1;
    } else
        arp_router_has_replied = 1;
    arp_switch_state(cs, AS_GW_CHECK);
    cs->arpPrevState = cs->dhcpState;
    cs->dhcpState = DS_BOUND_GW_CHECK;
    arp_wake_ts[AS_GW_CHECK] =
        arp_send_stats[ASEND_GW_PING].ts + ARP_RETRANS_DELAY + 250;
    return 0;
}

// Should only be called from DS_BOUND state.
static int arp_get_gw_hwaddr(struct client_state_t *cs)
{
    if (cs->dhcpState != DS_BOUND)
        log_error("arp_get_gw_hwaddr: called when state != DS_BOUND");
    arp_switch_state(cs, AS_GW_QUERY);
    if (cs->routerAddr)
        log_line("arp: Searching for dhcp server and gw addresses...");
    else
        log_line("arp: Searching for dhcp server address...");
    cs->got_server_arp = 0;
    if (arp_ping(cs, cs->serverAddr) < 0)
        return -1;
    if (cs->routerAddr) {
        cs->got_router_arp = 0;
        if (arp_ping(cs, cs->routerAddr) < 0)
            return -1;
    } else
        cs->got_router_arp = 1;
    arp_wake_ts[AS_GW_QUERY] =
        arp_send_stats[ASEND_GW_PING].ts + ARP_RETRANS_DELAY + 250;
    return 0;
}

static void arp_failed(struct client_state_t *cs)
{
    log_line("arp: Offered address is in use.  Declining.");
    send_decline(cs, arp_dhcp_packet.yiaddr);
    arp_wake_ts[AS_COLLISION_CHECK] = -1;
    reinit_selecting(cs, total_conflicts < MAX_CONFLICTS ?
                     0 : RATE_LIMIT_INTERVAL);
}

static void arp_gw_failed(struct client_state_t *cs)
{
    arp_wake_ts[AS_GW_CHECK] = -1;
    reinit_selecting(cs, 0);
}

static int act_if_arp_gw_failed(struct client_state_t *cs)
{
    if (arp_send_stats[ASEND_GW_PING].count >= gw_check_init_pingcount + 6) {
        if (arp_router_has_replied && !arp_server_has_replied)
            log_line("arp: DHCP server didn't reply.  Getting new lease.");
        else if (!arp_router_has_replied && arp_server_has_replied)
            log_line("arp: Gateway didn't reply.  Getting new lease.");
        else
            log_line("arp: DHCP server and gateway didn't reply.  Getting new lease.");
        arp_gw_failed(cs);
        return 1;
    }
    return 0;
}

void arp_set_defense_mode(struct client_state_t *cs)
{
    arp_switch_state(cs, AS_DEFENSE);
}

void arp_success(struct client_state_t *cs)
{
    char clibuf[INET_ADDRSTRLEN];
    struct in_addr temp_addr = {.s_addr = arp_dhcp_packet.yiaddr};
    inet_ntop(AF_INET, &temp_addr, clibuf, sizeof clibuf);
    log_line("Lease of %s obtained.  Lease time is %ld seconds.",
             clibuf, cs->lease);
    cs->clientAddr = arp_dhcp_packet.yiaddr;
    cs->dhcpState = DS_BOUND;
    cs->init = 0;
    last_conflict_ts = 0;
    arp_wake_ts[AS_COLLISION_CHECK] = -1;
    ifchange_bind(cs, &arp_dhcp_packet);
    if (cs->arpPrevState == DS_RENEWING || cs->arpPrevState == DS_REBINDING) {
        arp_switch_state(cs, AS_DEFENSE);
    } else {
        cs->routerAddr = get_option_router(&arp_dhcp_packet);
        arp_get_gw_hwaddr(cs);
    }
    set_listen_none(cs);
    write_leasefile(temp_addr);
    arp_announcement(cs);
    if (client_config.quit_after_lease)
        exit(EXIT_SUCCESS);
    if (!client_config.foreground)
        background();
}

static void arp_gw_success(struct client_state_t *cs)
{
    log_line("arp: Network seems unchanged.  Resuming normal operation.");
    arp_switch_state(cs, AS_DEFENSE);
    arp_announcement(cs);

    arp_wake_ts[AS_GW_CHECK] = -1;
    cs->dhcpState = cs->arpPrevState;
}

// ARP validation functions that will be performed by the BPF if it is
// installed.
static int arp_validate_bpf(struct arpMsg *am)
{
    if (am->h_proto != htons(ETH_P_ARP)) {
        log_warning("arp: IP header does not indicate ARP protocol");
        return 0;
    }
    if (am->htype != htons(ARPHRD_ETHER)) {
        log_warning("arp: ARP hardware type field invalid");
        return 0;
    }
    if (am->ptype != htons(ETH_P_IP)) {
        log_warning("arp: ARP protocol type field invalid");
        return 0;
    }
    if (am->hlen != 6) {
        log_warning("arp: ARP hardware address length invalid");
        return 0;
    }
    if (am->plen != 4) {
        log_warning("arp: ARP protocol address length invalid");
        return 0;
    }
    return 1;
}

// ARP validation functions that will be performed by the BPF if it is
// installed.
static int arp_validate_bpf_defense(struct client_state_t *cs,
                                    struct arpMsg *am)
{
    if (memcmp(am->sip4, &cs->clientAddr, 4))
        return 0;
    if (!memcmp(am->smac, client_config.arp, 6))
        return 0;
    return 1;
}

static int arp_is_query_reply(struct arpMsg *am)
{
    if (am->operation != htons(ARPOP_REPLY))
        return 0;
    if (memcmp(am->h_dest, client_config.arp, 6))
        return 0;
    if (memcmp(am->dmac, client_config.arp, 6))
        return 0;
    return 1;
}

static int arp_gen_probe_wait(struct client_state_t *cs)
{
    // This is not a uniform distribution but it doesn't matter here.
    return arp_probe_min + (nk_random_u32(&cs->rnd32_state) & 0x7fffffffu)
        % (arp_probe_max - arp_probe_min);
}

static void arp_defense_timeout(struct client_state_t *cs, long long nowts)
{
    (void)nowts; // Suppress warning; parameter necessary but unused.
    if (arp_wake_ts[AS_DEFENSE] != -1) {
        log_line("arp: Defending our lease IP.");
        arp_announcement(cs);
        arp_wake_ts[AS_DEFENSE] = -1;
    }
}

static void arp_gw_check_timeout(struct client_state_t *cs, long long nowts)
{
    arp_defense_timeout(cs, nowts);

    if (act_if_arp_gw_failed(cs))
        return;
    long long rtts = arp_send_stats[ASEND_GW_PING].ts + ARP_RETRANS_DELAY;
    if (nowts < rtts) {
        arp_wake_ts[AS_GW_CHECK] = rtts;
        return;
    }
    if (!arp_router_has_replied) {
        log_line("arp: Still waiting for gateway to reply to arp ping...");
        if (arp_ping(cs, cs->routerAddr) < 0)
            log_warning("arp: Failed to send ARP ping in retransmission.");
    }
    if (!arp_server_has_replied) {
        log_line("arp: Still waiting for DHCP server to reply to arp ping...");
        if (arp_ping(cs, cs->serverAddr) < 0)
            log_warning("arp: Failed to send ARP ping in retransmission.");
    }
    arp_wake_ts[AS_GW_CHECK] =
        arp_send_stats[ASEND_GW_PING].ts + ARP_RETRANS_DELAY;
}

static void arp_gw_query_timeout(struct client_state_t *cs, long long nowts)
{
    arp_defense_timeout(cs, nowts);

    long long rtts = arp_send_stats[ASEND_GW_PING].ts + ARP_RETRANS_DELAY;
    if (nowts < rtts) {
        arp_wake_ts[AS_GW_QUERY] = rtts;
        return;
    }
    if (!cs->got_router_arp) {
        log_line("arp: Still looking for gateway hardware address...");
        if (arp_ping(cs, cs->routerAddr) < 0)
            log_warning("arp: Failed to send ARP ping in retransmission.");
    }
    if (!cs->got_server_arp) {
        log_line("arp: Still looking for DHCP server hardware address...");
        if (arp_ping(cs, cs->serverAddr) < 0)
            log_warning("arp: Failed to send ARP ping in retransmission.");
    }
    arp_wake_ts[AS_GW_QUERY] =
        arp_send_stats[ASEND_GW_PING].ts + ARP_RETRANS_DELAY;
}

static void arp_collision_timeout(struct client_state_t *cs, long long nowts)
{
    arp_defense_timeout(cs, nowts);

    if (nowts >= arp_check_start_ts + ANNOUNCE_WAIT ||
        arp_send_stats[ASEND_COLLISION_CHECK].count >= arp_probe_num) {
        arp_success(cs);
        return;
    }
    long long rtts = arp_send_stats[ASEND_COLLISION_CHECK].ts +
        probe_wait_time;
    if (nowts < rtts) {
        arp_wake_ts[AS_COLLISION_CHECK] = rtts;
        return;
    }
    if (arp_ip_anon_ping(cs, arp_dhcp_packet.yiaddr) < 0)
        log_warning("arp: Failed to send ARP ping in retransmission.");
    probe_wait_time = arp_gen_probe_wait(cs);
    arp_wake_ts[AS_COLLISION_CHECK] =
        arp_send_stats[ASEND_COLLISION_CHECK].ts + probe_wait_time;
}

static void arp_do_defense(struct client_state_t *cs)
{
    // Even though the BPF will usually catch this case, sometimes there are
    // packets still in the socket buffer that arrived before the defense
    // BPF was installed, so it's necessary to check here.
    if (!arp_validate_bpf_defense(cs, &arpreply))
        return;

    log_line("arp: Detected a peer attempting to use our IP!");
    long long nowts = curms();
    arp_wake_ts[AS_DEFENSE] = -1;
    if (!last_conflict_ts ||
        nowts - last_conflict_ts < DEFEND_INTERVAL) {
        log_line("arp: Defending our lease IP.");
        arp_announcement(cs);
    } else if (!arp_relentless_def) {
        log_line("arp: Conflicting peer is persistent.  Requesting new lease.");
        send_release(cs);
        reinit_selecting(cs, 0);
    } else {
        arp_wake_ts[AS_DEFENSE] =
            arp_send_stats[ASEND_ANNOUNCE].ts + DEFEND_INTERVAL;
    }
    total_conflicts++;
    last_conflict_ts = nowts;
}

static void arp_do_gw_query_done(struct client_state_t *cs)
{
    arp_wake_ts[AS_GW_QUERY] = -1;
    arp_switch_state(cs, AS_DEFENSE);
    arp_announcement(cs);  // Do a second announcement.
}

static void arp_do_gw_query(struct client_state_t *cs)
{
    if (!arp_is_query_reply(&arpreply)) {
        arp_do_defense(cs);
        return;
    }
    if (!memcmp(arpreply.sip4, &cs->routerAddr, 4)) {
        memcpy(cs->routerArp, arpreply.smac, 6);
        log_line("arp: Gateway hardware address %02x:%02x:%02x:%02x:%02x:%02x",
                 cs->routerArp[0], cs->routerArp[1],
                 cs->routerArp[2], cs->routerArp[3],
                 cs->routerArp[4], cs->routerArp[5]);
        cs->got_router_arp = 1;
        if (cs->routerAddr == cs->serverAddr)
            goto server_is_router;
        if (cs->got_server_arp)
            arp_do_gw_query_done(cs);
        return;
    }
    if (!memcmp(arpreply.sip4, &cs->serverAddr, 4)) {
server_is_router:
        memcpy(cs->serverArp, arpreply.smac, 6);
        log_line("arp: DHCP Server hardware address %02x:%02x:%02x:%02x:%02x:%02x",
                 cs->serverArp[0], cs->serverArp[1],
                 cs->serverArp[2], cs->serverArp[3],
                 cs->serverArp[4], cs->serverArp[5]);
        cs->got_server_arp = 1;
        if (cs->got_router_arp)
            arp_do_gw_query_done(cs);
        return;
    }
    arp_do_defense(cs);
}

static void arp_do_collision_check(struct client_state_t *cs)
{
    if (!arp_is_query_reply(&arpreply))
        return;
    // If this packet was sent from our lease IP, and does not have a
    // MAC address matching our own (the latter check guards against stupid
    // hubs or repeaters), then it's a conflict and thus a failure.
    if (!memcmp(arpreply.sip4, &arp_dhcp_packet.yiaddr, 4) &&
        !memcmp(client_config.arp, arpreply.smac, 6)) {
        total_conflicts++;
        arp_failed(cs);
    }
}

static void arp_do_gw_check(struct client_state_t *cs)
{
    if (!arp_is_query_reply(&arpreply))
        return;
    if (!memcmp(arpreply.sip4, &cs->routerAddr, 4)) {
        // Success only if the router/gw MAC matches stored value
        if (!memcmp(cs->routerArp, arpreply.smac, 6)) {
            arp_router_has_replied = 1;
            if (cs->routerAddr == cs->serverAddr)
                goto server_is_router;
            if (arp_server_has_replied)
                arp_gw_success(cs);
        } else {
            log_line("arp: Gateway is different.  Getting a new lease.");
            arp_gw_failed(cs);
        }
        return;
    }
    if (!memcmp(arpreply.sip4, &cs->serverAddr, 4)) {
server_is_router:
        // Success only if the server MAC matches stored value
        if (!memcmp(cs->serverArp, arpreply.smac, 6)) {
            arp_server_has_replied = 1;
            if (arp_router_has_replied)
                arp_gw_success(cs);
        } else {
            log_line("arp: DHCP server is different.  Getting a new lease.");
            arp_gw_failed(cs);
        }
    }
}

static void arp_do_invalid(struct client_state_t *cs)
{
    log_error("handle_arp_response: called in invalid state %u", arpState);
    arp_close_fd(cs);
}

typedef struct {
    void (*packet_fn)(struct client_state_t *cs);
    void (*timeout_fn)(struct client_state_t *cs, long long nowts);
} arp_state_fn_t;

static const arp_state_fn_t arp_states[] = {
    { arp_do_invalid, 0 }, // AS_NONE
    { arp_do_collision_check, arp_collision_timeout }, // AS_COLLISION_CHECK
    { arp_do_gw_check, arp_gw_check_timeout }, // AS_GW_CHECK
    { arp_do_gw_query, arp_gw_query_timeout }, // AS_GW_QUERY
    { arp_do_defense, arp_defense_timeout }, // AS_DEFENSE
    { arp_do_invalid, 0 }, // AS_MAX
};

void handle_arp_response(struct client_state_t *cs)
{
    ssize_t r = 0;
    if (arpreply_offset < sizeof arpreply) {
        r = safe_read(cs->arpFd, (char *)&arpreply + arpreply_offset,
                      sizeof arpreply - arpreply_offset);
        if (r < 0) {
            log_error("arp: ARP response read failed: %s", strerror(errno));
            switch (arpState) {
            case AS_COLLISION_CHECK: arp_failed(cs); break;
            case AS_GW_CHECK: arp_gw_failed(cs); break;
            default: arp_reopen_fd(cs); break;
            }
        } else
            arpreply_offset += (size_t)r;
    }

    if (r <= 0) {
        handle_arp_timeout(cs, curms());
        return;
    }

    if (arpreply_offset < ARP_MSG_SIZE)
        return;

    // Emulate the BPF filters if they are not in use.
    if (!using_arp_bpf && (!arp_validate_bpf(&arpreply) ||
                           (arpState == AS_DEFENSE &&
                            !arp_validate_bpf_defense(cs, &arpreply)))) {
        arpreply_clear();
        return;
    }

    if (arp_states[arpState].packet_fn)
        arp_states[arpState].packet_fn(cs);
    arpreply_clear();
}

// Perform retransmission if necessary.
void handle_arp_timeout(struct client_state_t *cs, long long nowts)
{
    if (arp_states[arpState].timeout_fn)
        arp_states[arpState].timeout_fn(cs, nowts);
}

long long arp_get_wake_ts(void)
{
    long long mt = -1;
    for (int i = 0; i < AS_MAX; ++i) {
        if (arp_wake_ts[i] < 0)
            continue;
        if (mt < 0 || mt > arp_wake_ts[i])
            mt = arp_wake_ts[i];
    }
    return mt;
}