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 <dm.h>
  11 #include <log.h>
  12 #include <net.h>
  13 #include <rtc.h>
  14
  15 #include <net/sntp.h>
  16
  17 #define SNTP_TIMEOUT 10000UL
  18
  19 static int sntp_our_port;
  20
  21 /* NTP server IP address */
  22 struct in_addr  net_ntp_server;
  23 /* offset time from UTC */
  24 int             net_ntp_time_offset;
  25
  26 static void sntp_send(void)
  27 {
  28         struct sntp_pkt_t pkt;
  29         int pktlen = SNTP_PACKET_LEN;
  30         int sport;
  31
  32         debug("%s\n", __func__);
  33
  34         memset(&pkt, 0, sizeof(pkt));
  35
  36         pkt.li = NTP_LI_NOLEAP;
  37         pkt.vn = NTP_VERSION;
  38         pkt.mode = NTP_MODE_CLIENT;
  39
  40         memcpy((char *)net_tx_packet + net_eth_hdr_size() + IP_UDP_HDR_SIZE,
  41                (char *)&pkt, pktlen);
  42
  43         sntp_our_port = 10000 + (get_timer(0) % 4096);
  44         sport = NTP_SERVICE_PORT;
  45
  46         net_send_udp_packet(net_server_ethaddr, net_ntp_server, sport,
  47                             sntp_our_port, pktlen);
  48 }
  49
  50 static void sntp_timeout_handler(void)
  51 {
  52         puts("Timeout\n");
  53         net_set_state(NETLOOP_FAIL);
  54         return;
  55 }
  56
  57 static void sntp_handler(uchar *pkt, unsigned dest, struct in_addr sip,
  58                          unsigned src, unsigned len)
  59 {
  60 #ifdef CONFIG_TIMESTAMP
  61         struct sntp_pkt_t *rpktp = (struct sntp_pkt_t *)pkt;
  62         struct rtc_time tm;
  63         ulong seconds;
  64 #endif
  65
  66         debug("%s\n", __func__);
  67
  68         if (dest != sntp_our_port)
  69                 return;
  70
  71 #ifdef CONFIG_TIMESTAMP
  72         /*
  73          * As the RTC's used in U-Boot support second resolution only
  74          * we simply ignore the sub-second field.
  75          */
  76         memcpy(&seconds, &rpktp->transmit_timestamp, sizeof(ulong));
  77
  78         rtc_to_tm(ntohl(seconds) - 2208988800UL + net_ntp_time_offset, &tm);
  79 #if defined(CONFIG_CMD_DATE)
  80 #  ifdef CONFIG_DM_RTC
  81         struct udevice *dev;
  82         int ret;
  83
  84         ret = uclass_get_device(UCLASS_RTC, 0, &dev);
  85         if (ret)
  86                 printf("SNTP: cannot find RTC: err=%d\n", ret);
  87         else
  88                 dm_rtc_set(dev, &tm);
  89 #  else
  90         rtc_set(&tm);
  91 #  endif
  92 #endif
  93         printf("Date: %4d-%02d-%02d Time: %2d:%02d:%02d\n",
  94                tm.tm_year, tm.tm_mon, tm.tm_mday,
  95                tm.tm_hour, tm.tm_min, tm.tm_sec);
  96 #endif
  97
  98         net_set_state(NETLOOP_SUCCESS);
  99 }
 100
 101 /*
 102  * SNTP:
 103  *
 104  *      Prerequisites:  - own ethernet address
 105  *                      - own IP address
 106  *      We want:        - network time
 107  *      Next step:      none
 108  */
 109 int sntp_prereq(void *data)
 110 {
 111         if (net_ntp_server.s_addr == 0) {
 112                 puts("*** ERROR: NTP server address not given\n");
 113                 return 1;
 114         }
 115
 116         return 0;
 117 }
 118
 119 int sntp_start(void *data)
 120 {
 121         debug("%s\n", __func__);
 122
 123         net_set_timeout_handler(SNTP_TIMEOUT, sntp_timeout_handler);
 124         net_set_udp_handler(sntp_handler);
 125         memset(net_server_ethaddr, 0, sizeof(net_server_ethaddr));
 126
 127         sntp_send();
 128
 129         return 0;
 130 }