net/sntp.c

   1 /*
   2  * SNTP support driver
   3  *
   4  * Masami Komiya <mkomiya@sonare.it> 2005
   5  *
   6  */
   7
   8 #include <common.h>
   9 #include <command.h>
  10 #include <net.h>
  11 #include <rtc.h>
  12
  13 #include "sntp.h"
  14
  15 #if ((CONFIG_COMMANDS & CFG_CMD_NET) && (CONFIG_COMMANDS & CFG_CMD_SNTP))
  16
  17 #define SNTP_TIMEOUT 10
  18
  19 static int SntpOurPort;
  20
  21 static void
  22 SntpSend (void)
  23 {
  24         struct sntp_pkt_t pkt;
  25         int pktlen = SNTP_PACKET_LEN;
  26         int sport;
  27
  28         debug ("%s\n", __FUNCTION__);
  29
  30         memset (&pkt, 0, sizeof(pkt));
  31
  32         pkt.li = NTP_LI_NOLEAP;
  33         pkt.vn = NTP_VERSION;
  34         pkt.mode = NTP_MODE_CLIENT;
  35
  36         memcpy ((char *)NetTxPacket + NetEthHdrSize() + IP_HDR_SIZE, (char *)&pkt, pktlen);
  37
  38         SntpOurPort = 10000 + (get_timer(0) % 4096);
  39         sport = NTP_SERVICE_PORT;
  40
  41         NetSendUDPPacket (NetServerEther, NetNtpServerIP, sport, SntpOurPort, pktlen);
  42 }
  43
  44 static void
  45 SntpTimeout (void)
  46 {
  47         puts ("Timeout\n");
  48         NetState = NETLOOP_FAIL;
  49         return;
  50 }
  51
  52 static void
  53 SntpHandler (uchar *pkt, unsigned dest, unsigned src, unsigned len)
  54 {
  55         struct sntp_pkt_t rpkt;
  56         struct rtc_time tm;
  57
  58         debug ("%s\n", __FUNCTION__);
  59
  60         if (dest != SntpOurPort) return;
  61
  62         memcpy ((unsigned char *)&rpkt, pkt, len);
  63
  64 #if (CONFIG_COMMANDS & CFG_CMD_DATE) || defined(CONFIG_TIMESTAMP)
  65 to_tm(ntohl(rpkt.transmit_timestamp), &tm);
  66 printf ("Date: %4d-%02d-%02d Time: %2d:%02d:%02d\n",
  67 tm.tm_year, tm.tm_mon, tm.tm_mday,
  68 tm.tm_hour, tm.tm_min, tm.tm_sec);
  69         to_tm(ntohl(rpkt.transmit_timestamp) - 2208988800u + NetTimeOffset, &tm);
  70 #if (CONFIG_COMMANDS & CFG_CMD_DATE)
  71         rtc_set (&tm);
  72 #endif
  73         printf ("Date: %4d-%02d-%02d Time: %2d:%02d:%02d\n",
  74                 tm.tm_year, tm.tm_mon, tm.tm_mday,
  75                 tm.tm_hour, tm.tm_min, tm.tm_sec);
  76 #endif
  77
  78         NetState = NETLOOP_SUCCESS;
  79 }
  80
  81 void
  82 SntpStart (void)
  83 {
  84         debug ("%s\n", __FUNCTION__);
  85
  86         NetSetTimeout (SNTP_TIMEOUT * CFG_HZ, SntpTimeout);
  87         NetSetHandler(SntpHandler);
  88         memset (NetServerEther, 0, 6);
  89
  90         SntpSend ();
  91 }
  92
  93 #endif /* CONFIG_COMMANDS & CFG_CMD_SNTP */