/*
Serval Distributed Numbering Architecture (DNA)
Copyright (C) 2010 Paul Gardner-Stephen
 
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.
 
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.
 
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
*/

/*
  Serval Overlay Mesh Network.

  Basically we use UDP broadcast to send link-local, and then implement a BATMAN-like protocol over the top of that.
  
  Each overlay packet can contain one or more encapsulated packets each addressed using Serval DNA SIDs, with source, 
  destination and next-hop addresses.

  The use of an overlay also lets us be a bit clever about using irregular transports, such as an ISM915 modem attached via ethernet
  (which we are planning to build in coming months), by paring off the IP and UDP headers that would otherwise dominate.  Even on
  regular WiFi and ethernet we can aggregate packets in a way similar to IAX, but not just for voice frames.

  The use of long (relative to IPv4 or even IPv6) 256 bit Curve25519 addresses means that it is a really good idea to
  have neighbouring nodes exchange lists of peer aliases so that addresses can be summarised, possibly using less space than IPv4
  would have.
  
  One approach to handle address shortening is to have the periodic TTL=255 BATMAN-style hello packets include an epoch number.  
  This epoch number can be used by immediate neighbours of the originator to reference the neighbours listed in that packet by
  their ordinal position in the packet instead of by their full address.  This gets us address shortening to 1 byte in most cases 
  in return for no new packets, but the periodic hello packets will now be larger.  We might deal with this issue by having these
  hello packets reference the previous epoch for common neighbours.  Unresolved neighbour addresses could be resolved by a simple
  DNA request, which should only need to occur ocassionally, and other link-local neighbours could sniff and cache the responses
  to avoid duplicated traffic.  Indeed, during quiet times nodes could preemptively advertise address resolutions if they wished,
  or similarly advertise the full address of a few (possibly randomly selected) neighbours in each epoch.

  Byzantine Robustness is a goal, so we have to think about all sorts of malicious failure modes.

  One approach to help byzantine robustness is to have multiple signature shells for each hop for mesh topology packets.
  Thus forging a report of closeness requires forging a signature.  As such frames are forwarded, the outermost signature
  shell is removed. This is really only needed for more paranoid uses.

  We want to have different traffic classes for voice/video calls versus regular traffic, e.g., MeshMS frames.  Thus we need to have
  separate traffic queues for these items.  Aside from allowing us to prioritise isochronous data, it also allows us to expire old
  isochronous frames that are in-queue once there is no longer any point delivering them (e.g after holding them more than 200ms).
  We can also be clever about round-robin fair-sharing or even prioritising among isochronous streams.  Since we also know about the
  DNA isochronous protocols and the forward error correction and other redundancy measures we also get smart about dropping, say, 1 in 3
  frames from every call if we know that this can be safely done.  That is, when traffic is low, we maximise redundancy, and when we
  start to hit the limit of traffic, we start to throw away some of the redundancy.  This of course relies on us knowing when the
  network channel is getting too full.

  Smart-flooding of broadcast information is also a requirement.  The long addresses help here, as we can make any address that begins
  with the first 192 bits all ones be broadcast, and use the remaining 64 bits as a "broadcast packet identifier" (BPI).  
  Nodes can remember recently seen BPIs and not forward broadcast frames that have been seen recently.  This should get us smart flooding
  of the majority of a mesh (with some node mobility issues being a factor).  We could refine this later, but it will do for now, especially
  since for things like number resolution we are happy to send repeat requests.

  This file currently seems to exist solely to contain this introduction, which is fine with me. Functions land in here until their
  proper place becomes apparent.
  
*/

#include "serval.h"
#include "rhizome.h"
#include "strbuf.h"

int overlayMode=0;

overlay_txqueue overlay_tx[OQ_MAX];

keyring_file *keyring=NULL;

int overlayServerMode()
{
  /* In overlay mode we need to listen to all of our sockets, and also to
     send periodic traffic. This means we need to */
  INFO("Running in overlay mode.");

  /* Make sure rhizome configured settings are known. */
  if (rhizome_fetch_interval_ms < 1)
    rhizome_configure();

  /* Get keyring available for use.
     Required for MDP, and very soon as a complete replacement for the
     HLR for DNA lookups, even in non-overlay mode. */
  keyring=keyring_open_with_pins("");
  if (!keyring) {
    return WHY("Could not open serval keyring file.");
  }
  /* put initial identity in if we don't have any visible */
  keyring_seed(keyring);

  /* Set default congestion levels for queues */
  int i;
  for(i=0;i<OQ_MAX;i++) {
    overlay_tx[i].maxLength=100;
    overlay_tx[i].latencyTarget=1000; /* Keep packets in queue for 1 second by default */
    overlay_tx[i].transmit_delay=10; /* Hold onto packets for 10ms before trying to send a full packet */
    overlay_tx[i].grace_period=100; /* Delay sending a packet for up to 100ms if servald has other processing to do */
  }
  /* expire voice/video call packets much sooner, as they just aren't any use if late */
  overlay_tx[OQ_ISOCHRONOUS_VOICE].latencyTarget=500;
  overlay_tx[OQ_ISOCHRONOUS_VIDEO].latencyTarget=500;

  /* try to send voice packets without any delay, and before other background processing */
  overlay_tx[OQ_ISOCHRONOUS_VOICE].transmit_delay=0;
  overlay_tx[OQ_ISOCHRONOUS_VOICE].grace_period=0;

  /* opportunistic traffic can be significantly delayed */
  overlay_tx[OQ_OPPORTUNISTIC].transmit_delay=200;
  overlay_tx[OQ_OPPORTUNISTIC].grace_period=500;
  
  /* Get the set of socket file descriptors we need to monitor.
     Note that end-of-file will trigger select(), so we cannot run select() if we 
     have any dummy interfaces running. So we do an ugly hack of just waiting no more than
     5ms between checks if we have a dummy interface running.  This is a reasonable simulation
     of wifi latency anyway, so we'll live with it.  Larger values will affect voice transport,
     and smaller values would affect CPU and energy use, and make the simulation less realistic. */

  /* Create structures to use 1MB of RAM for testing */
  overlay_route_init(1);

#define SCHEDULE(X, Y, D) { \
static struct sched_ent _sched_##X; \
static struct profile_total _stats_##X; \
bzero(&_sched_##X, sizeof(struct sched_ent)); \
bzero(&_stats_##X, sizeof(struct profile_total)); \
_sched_##X.stats = &_stats_##X; \
_sched_##X.function=X;\
_stats_##X.name="" #X "";\
_sched_##X.alarm=gettime_ms()+Y;\
_sched_##X.deadline=_sched_##X.alarm+D;\
schedule(&_sched_##X); }
  
  /* Periodically check for server shut down */
  SCHEDULE(server_shutdown_check, 0, 100);
  
  /* Setup up MDP & monitor interface unix domain sockets */
  overlay_mdp_setup_sockets();
  monitor_setup_sockets();

  /* Get rhizome server started BEFORE populating fd list so that
     the server's listen socket is in the list for poll() */
  if (rhizome_enabled()) 
    /* Rhizome http server needs to know which callback to attach
       to client sockets, so provide it here, along with the name to
       appear in time accounting statistics. */
    rhizome_http_server_start(rhizome_server_parse_http_request,
			      "rhizome_server_parse_http_request",
			      RHIZOME_HTTP_PORT,RHIZOME_HTTP_PORT_MAX);
  
  /* Pick next rhizome files to grab every few seconds
     from the priority list continuously being built from observed
     bundle announcements */
  SCHEDULE(rhizome_enqueue_suggestions, rhizome_fetch_interval_ms, rhizome_fetch_interval_ms*3);

  /* Periodically check for new interfaces */
  SCHEDULE(overlay_interface_discover, 1, 100);

  /* Periodically update route table. */
  SCHEDULE(overlay_route_tick, 100, 100);

  /* Show CPU usage stats periodically */
  if (debug&DEBUG_TIMING){
    SCHEDULE(fd_periodicstats, 3000, 500);
  }

#undef SCHEDULE
  
  while(1) {
    /* Check for activitiy and respond to it */
    fd_poll();
  }

  return 0;
}

int overlay_frame_process(struct overlay_interface *interface,overlay_frame *f)
{
  IN();
  if (!f) RETURN(WHY("f==NULL"));

  time_ms_t now = gettime_ms();

  if (debug&DEBUG_OVERLAYFRAMES)
    DEBUGF(">>> Received frame (type=%02x, bytes=%d)",f->type,f->payload?f->payload->length:-1);

  // only examine payloads that are broadcasts, or where I'm the next hop
  if (overlay_address_is_broadcast(f->nexthop)) {
    if (overlay_broadcast_drop_check(f->nexthop)){
      if (debug&DEBUG_OVERLAYFRAMES)
	DEBUGF("Dropping frame, duplicate broadcast %s", alloca_tohex_sid(f->nexthop));
      RETURN(0);
    }
  }else if (!overlay_address_is_local(f->nexthop)){
    if (debug&DEBUG_OVERLAYFRAMES)
      DEBUGF("Dropping frame, not addressed to me %s", alloca_tohex_sid(f->nexthop));
    RETURN(0);
  }
  
  int broadcast=overlay_address_is_broadcast(f->destination);
  int ultimatelyForMe=0;
  
  if (broadcast){
    ultimatelyForMe = 1;
    // Note that we assume a broadcast destination address is the same as the broadcast nexthop address
    // we should decide to drop the packet based on the nexthop address.
  }else{
    if (overlay_address_is_local(f->destination))
      ultimatelyForMe = 1;
  }
  
  f->ttl--;
  
  // Never ever forward these types
  if ((f->type==OF_TYPE_SELFANNOUNCE)
      ||(f->type==OF_TYPE_RHIZOME_ADVERT))
    f->ttl=0;

  /* Is this a frame we have to forward on? */
  if (((!ultimatelyForMe)||broadcast)&&(f->ttl>0))
    {
      /* Yes, it is. */

      int forward=1;
      
      if (!broadcast)
      {
	if (overlay_get_nexthop(f->destination,f->nexthop,&f->nexthop_interface))
	  WHYF("Could not find next hop for %s* - dropping frame",
	       alloca_tohex(f->destination, 7));
	forward=0;
      }

      if (0)
	DEBUGF("considering forwarding frame to %s (forme=%d, bcast=%d)",
	       alloca_tohex_sid(f->destination),ultimatelyForMe,broadcast);


      if (forward) {
	if (debug&DEBUG_OVERLAYFRAMES)
	  DEBUG("Forwarding frame");
	
	/* Queue frame for dispatch.
	   Don't forget to put packet in the correct queue based on type.
	   (e.g., mesh management, voice, video, ordinary or opportunistic).

	   But the really important bit is to clone the frame, since the
	   structure we are looking at here must be left as is and returned
	   to the caller to do as they please */	  
	overlay_frame *qf=op_dup(f);
	if (!qf) WHY("Could not clone frame for queuing");
	else {
	  int qn=OQ_ORDINARY;
	  /* Make sure voice traffic gets priority */
	  if ((qf->type&OF_TYPE_BITS)==OF_TYPE_DATA_VOICE) {
	    qn=OQ_ISOCHRONOUS_VOICE;
	    rhizome_saw_voice_traffic();
	  }
	  if (0) WHY("queuing frame for forwarding");
	  if (overlay_payload_enqueue(qn,qf,0)) {
	    WHY("failed to enqueue forwarded payload");
	    op_free(qf);
	  }
	}
      }
    }

  // process payloads with broadcast or our sid as destination
  if (ultimatelyForMe){
    int id = (interface - overlay_interfaces);
    switch(f->type)
      {
      case OF_TYPE_SELFANNOUNCE:
	overlay_route_saw_selfannounce(f,now);
	break;
      case OF_TYPE_SELFANNOUNCE_ACK:
	overlay_route_saw_selfannounce_ack(f,now);
	break;
      case OF_TYPE_NODEANNOUNCE:
	overlay_route_saw_advertisements(id,f,now);
	break;
      case OF_TYPE_RHIZOME_ADVERT:
	overlay_rhizome_saw_advertisements(id,f,now);
	break;
      case OF_TYPE_DATA:
      case OF_TYPE_DATA_VOICE:
	if (0) {
	  DEBUG("saw mdp containing frame");
	  DEBUGF("  src = %s\n", alloca_tohex_sid(f->source));
	  DEBUGF("  nxt = %s\n", alloca_tohex_sid(f->nexthop));
	  DEBUGF("  dst = %s\n", alloca_tohex_sid(f->destination));
	  dump("payload", f->payload->bytes, f->payload->length);
	}
	overlay_saw_mdp_containing_frame(f,now);
	break;
      default:
	DEBUGF("Unsupported f->type=0x%x",f->type);
	RETURN(WHY("Support for that f->type not yet implemented"));
	break;
      }
  }

  RETURN(0);
}