/*
 * Copyright (c) 2006, Swedish Institute of Computer Science
 * All rights reserved. 
 *
 * Redistribution and use in source and binary forms, with or without 
 * modification, are permitted provided that the following conditions 
 * are met: 
 * 1. Redistributions of source code must retain the above copyright 
 *    notice, this list of conditions and the following disclaimer. 
 * 2. 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. 
 * 3. Neither the name of the Institute nor the names of its contributors 
 *    may be used to endorse or promote products derived from this software 
 *    without specific prior written permission. 
 *
 * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE 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 INSTITUTE 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. 
 *
 * @(#)$Id: dhclient.c,v 1.2 2006/08/09 16:13:40 bg- Exp $
 */

/*
 * Example mote configuration with one radio interface that uses DHCP
 * to configure its IP stack.
 *
 * The gateway systems must be configured to find what DHCP server to
 * use, this can be done with an extra argument to tunslip, example:
 *
 * sliphost# ./tunslip 172.16.0.1 255.255.0.0 193.10.66.195
 *
 *
 * This kernel has no application code but new modules can be uploaded
 * using the codeprop program, example:
 *
 * datan$ ./codeprop 172.16.0.1 loadable_prg.ko
 * File successfully sent (1116 bytes)
 * Reply: ok
 *
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "contiki.h"

/* Also IP output. */
#include "net/uip-fw-service.h"
#include "net/uaodv.h"
#include "net/dhcpc.h"
#include "dev/slip.h"
#include "dev/cc2420.h"

#include "dev/lpm.h"
#include "dev/ds2411.h"
#include "dev/leds.h"
#include "dev/light.h"
#include "dev/xmem.h"

#include "dev/button.h"

#include "codeprop/codeprop.h"

/* This is how we force inclusion of the psock library. */
#include "net/psock.h"
void *force_psock_inclusion = &psock_init;

void uip_log(char *msg) { puts(msg); }

struct uip_fw_netif cc2420if =
{UIP_FW_NETIF(0,0,0,0, 255,255,255,255, cc2420_send_uaodv)};

PROCESS(button_process, "Button process");
PROCESS(dhclient_process, "Dhclient process");

/* Radio stuff in network byte order. */
static u16_t panId = HTONS(0x2024);

#define RF_CHANNEL              26

int
main(int argc, char **argv)
{
  /*
   * Initalize hardware.
   */
  msp430_cpu_init();
  clock_init();
  leds_init();
  leds_toggle(LEDS_RED | LEDS_GREEN | LEDS_BLUE);
  slip_arch_init();		/* Must come before first printf */
  printf("Starting %s "
	 "($Id: dhclient.c,v 1.2 2006/08/09 16:13:40 bg- Exp $)\n", __FILE__);
  ds2411_init();
  sensors_light_init();
  cc2420_init();
  xmem_init();
  button_init(&button_process);
  leds_toggle(LEDS_RED | LEDS_GREEN | LEDS_BLUE);
  /*
   * Hardware initialization done!
   */
  
  printf("MAC %02x:%02x:%02x:%02x:%02x:%02x:%02x:%02x\n",
	 ds2411_id[0], ds2411_id[1], ds2411_id[2], ds2411_id[3],
	 ds2411_id[4], ds2411_id[5], ds2411_id[6], ds2411_id[7]);

  srand((ds2411_id[3]<<8) + (ds2411_id[4]<<6) + (ds2411_id[5]<<4) +
	(ds2411_id[6]<<2) +  ds2411_id[7]);

  /*
   * Initialize Contiki and our processes.
   */
  process_init();
  process_start(&etimer_process, NULL);

  /* Configure IP stack. */
  uip_init();
  
  /* Start IP stack. */
  process_start(&tcpip_process, NULL);
  process_start(&uip_fw_process, NULL);	/* Start IP output */
  process_start(&cc2420_process, NULL);
  cc2420_on();
  process_start(&dhclient_process, NULL);

  process_start(&button_process, NULL);
  process_start(&tcp_loader_process, NULL);

  /*
   * This is the scheduler loop.
   */
  printf("process_run()...\n");
  lpm_on();
  while (1) {
    do {
      /* Reset watchdog. */
    } while(process_run() > 0);
    /* Idle processing. */
    LPM_SLEEP();
  }

  return 0;
}

PROCESS_THREAD(button_process, ev, data)
{
  static struct etimer etimer;

  PROCESS_EXITHANDLER(goto exit);
  PROCESS_BEGIN();

  printf("button_process starting\n");

  while(1) {
    PROCESS_WAIT_EVENT();

    if(ev == PROCESS_EVENT_MSG && data != NULL
       && ((struct button_msg *)data)->type == BUTTON_MSG_TYPE) {
      printf("button press\n");

      leds_toggle(LEDS_RED | LEDS_GREEN | LEDS_BLUE);
      etimer_set(&etimer, CLOCK_SECOND);
      PROCESS_WAIT_UNTIL(etimer_expired(&etimer));
      leds_toggle(LEDS_RED | LEDS_GREEN | LEDS_BLUE);
    }
  }

 exit:
  printf("button_process exiting\n");
  PROCESS_END();
}

PROCESS_THREAD(dhclient_process, ev, data)
{
  PROCESS_EXITHANDLER(goto exit);
  PROCESS_BEGIN();

  printf("dhclient_process starting\n");

  leds_on(LEDS_GREEN);

  uip_setipid(rand());

  /* For now use 0.0.0.0 as our IP address. */
  uip_ipaddr(&uip_hostaddr, 0,0,0,0);
  cc2420_set_chan_pan_addr(RF_CHANNEL, panId, uip_hostaddr.u16[1], ds2411_id);

  /* Use only radio interface and enable forwarding. */
  uip_fw_default(&cc2420if);
  tcpip_set_forwarding(1);
  /* No default router yet. */

  dhcpc_init(ds2411_id, sizeof(ds2411_id));
  dhcpc_request();

  while(1) {
    PROCESS_WAIT_EVENT();
    dhcpc_appcall(ev, data);
  }

 exit:
  printf("dhclient_process exiting\n");
  PROCESS_END();
}

static char is_configured;

#define ip2quad(p) uip_ipaddr1(p),uip_ipaddr2(p),uip_ipaddr3(p),uip_ipaddr4(p)

void
dhcpc_configured(const struct dhcpc_state *s)
{
  if(is_configured)
    printf("dhcp reconfigure %d.%d.%d.%d\n", ip2quad(&s->ipaddr));
  else {
    printf("dhcpc_configured %d.%d.%d.%d\n", ip2quad(&s->ipaddr));
    leds_off(LEDS_GREEN);
    is_configured = 1;
    process_start(&uaodv_process, NULL);
  }

  uip_sethostaddr(&s->ipaddr);
  uip_setnetmask(&s->netmask);
  uip_setdraddr(&s->default_router);

  uip_ipaddr_copy(&cc2420if.ipaddr,  &s->ipaddr);
  uip_ipaddr_copy(&cc2420if.netmask, &s->netmask);
  /* resolv_conf(s->dnsaddr); */

  /*
   * Now we also have a new short MAC address!
   */
  cc2420_set_chan_pan_addr(RF_CHANNEL, panId, uip_hostaddr.u16[1], ds2411_id);

  /* Use only radio interface and enable forwarding. */
  uip_fw_init();
  tcpip_set_forwarding(1);
  uip_fw_default(&cc2420if);
}

void
dhcpc_unconfigured(const struct dhcpc_state *s)
{
  printf("dhcpc_unconfigured\n");
  leds_on(LEDS_GREEN);
  is_configured = 0;
  process_exit(&uaodv_process);

  uip_ipaddr(&uip_hostaddr, 0,0,0,0);
  uip_ipaddr(&uip_netmask, 0,0,0,0);
  uip_ipaddr(&uip_draddr, 0,0,0,0);

  /* New short MAC address. */
  cc2420_set_chan_pan_addr(RF_CHANNEL, panId, uip_hostaddr.u16[1], ds2411_id);
}