net/sntp.c

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