NTPClients.cpp

#include <Arduino.h>
#include "NTPClock.h"
#include "NTPClients.h"

#if LWIP_IPV6
const CLIENT_ADDR_T zero_addr = {0,0,0,0};
#else
const CLIENT_ADDR_T zero_addr = {0};
#endif

NTPClients::NTPClients() {
  for(uint8_t i = 0; i < NUMCLIENTS; i++) {
    CLIENT_ADDR_SET(&clients[i].addr, &zero_addr);
  }
}

void NTPClients::addRx(CLIENT_ADDR_T *addr, uint16_t port, uint32_t rx_s, uint32_t rx_subs) {
  // TODO: dealing with large volume addresses
  for(uint8_t i = 0; i < NUMCLIENTS; i++) {
    if(CLIENT_ADDR_CMP(&clients[i].addr, addr) || CLIENT_ADDR_CMP(&clients[i].addr, &zero_addr)) {
      CLIENT_ADDR_SET(&clients[i].addr, addr);
      clients[i].lastPort = port;
      clients[i].rx_s = rx_s;
      clients[i].rx_subs = rx_subs;
      clients[i].tx_s = 0;
      clients[i].tx_subs = 0;
      return;
    }
  }
}

void NTPClients::addTx(CLIENT_ADDR_T *addr, uint16_t port, uint32_t tx_s, uint32_t tx_subs) {
  for(uint8_t i = 0; i < NUMCLIENTS; i++) {
    if(CLIENT_ADDR_CMP(&clients[i].addr, addr) && clients[i].lastPort == port) {
      clients[i].tx_s = tx_s;
      clients[i].tx_subs = tx_subs;
      return;
    }
  }
}

struct client *NTPClients::findClient(CLIENT_ADDR_T *addr, uint32_t ts, uint32_t ts_subs) {
  for(uint8_t i = 0; i < NUMCLIENTS; i++) {
    if(CLIENT_ADDR_CMP(&clients[i].addr, addr) && clients[i].rx_s == ts && clients[i].rx_subs == ts_subs) {
      return &clients[i];
    }
  }

  return NULL;
}

void NTPClients::expireClients() {
  uint32_t sec;
  localClock.getTime(&sec, NULL);
  uint32_t expire_time = sec - 4096; // allow a 2^12 poll time

  for(uint8_t i = 0; i < NUMCLIENTS; i++) {
    if(!CLIENT_ADDR_CMP(&clients[i].addr, &zero_addr) && clients[i].rx_s < expire_time) {
      CLIENT_ADDR_SET(&clients[i].addr, &zero_addr);
      clients[i].rx_s = 0;
      clients[i].rx_subs = 0;
      clients[i].tx_s = 0;
      clients[i].tx_subs = 0;
    }
  }
}