/*
   * SNTP support driver
   *
   * Masami Komiya <mkomiya@sonare.it> 2005
   *
   */
  
  #include <common.h>
  #include <command.h>
  #include <net.h>
  #include <rtc.h>
  
  #include "sntp.h"

  #define SNTP_TIMEOUT 10000UL

  
  static int SntpOurPort;
  
  static void

  SntpSend(void)

  {
  	struct sntp_pkt_t pkt;
  	int pktlen = SNTP_PACKET_LEN;
  	int sport;

  	debug("%s
  ", __func__);

  	memset(&pkt, 0, sizeof(pkt));

  
  	pkt.li = NTP_LI_NOLEAP;
  	pkt.vn = NTP_VERSION;
  	pkt.mode = NTP_MODE_CLIENT;

  	memcpy((char *)NetTxPacket + NetEthHdrSize() + IP_UDP_HDR_SIZE,

  		(char *)&pkt, pktlen);

  
  	SntpOurPort = 10000 + (get_timer(0) % 4096);
  	sport = NTP_SERVICE_PORT;

  	NetSendUDPPacket(NetServerEther, NetNtpServerIP, sport, SntpOurPort,
  		pktlen);

  }
  
  static void

  SntpTimeout(void)

  {

  	puts("Timeout
  ");

  	net_set_state(NETLOOP_FAIL);

  	return;
  }
  
  static void

  SntpHandler(uchar *pkt, unsigned dest, IPaddr_t sip, unsigned src,
  	    unsigned len)

  {

  	struct sntp_pkt_t *rpktp = (struct sntp_pkt_t *)pkt;

  	struct rtc_time tm;

  	ulong seconds;

  	debug("%s
  ", __func__);

  	if (dest != SntpOurPort)
  		return;

  	/*
  	 * As the RTC's used in U-Boot sepport second resolution only
  	 * we simply ignore the sub-second field.
  	 */

  	memcpy(&seconds, &rpktp->transmit_timestamp, sizeof(ulong));

  	to_tm(ntohl(seconds) - 2208988800UL + NetTimeOffset, &tm);

  #if defined(CONFIG_CMD_DATE)

  	rtc_set(&tm);

  #endif

  	printf("Date: %4d-%02d-%02d Time: %2d:%02d:%02d
  ",

  		tm.tm_year, tm.tm_mon, tm.tm_mday,
  		tm.tm_hour, tm.tm_min, tm.tm_sec);

  	net_set_state(NETLOOP_SUCCESS);

  }
  
  void

  SntpStart(void)

  {

  	debug("%s
  ", __func__);

  	NetSetTimeout(SNTP_TIMEOUT, SntpTimeout);

  	net_set_udp_handler(SntpHandler);

  	memset(NetServerEther, 0, sizeof(NetServerEther));

  	SntpSend();

  }