uboot/net/sntp.c
<<
>>
Prefs
   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
  19static int sntp_our_port;
  20
  21/* NTP server IP address */
  22struct in_addr  net_ntp_server;
  23/* offset time from UTC */
  24int             net_ntp_time_offset;
  25
  26static 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
  50static void sntp_timeout_handler(void)
  51{
  52        puts("Timeout\n");
  53        net_set_state(NETLOOP_FAIL);
  54        return;
  55}
  56
  57static void sntp_handler(uchar *pkt, unsigned dest, struct in_addr sip,
  58                         unsigned src, unsigned len)
  59{
  60        struct sntp_pkt_t *rpktp = (struct sntp_pkt_t *)pkt;
  61        struct rtc_time tm;
  62        ulong seconds;
  63
  64        debug("%s\n", __func__);
  65
  66        if (dest != sntp_our_port)
  67                return;
  68
  69        /*
  70         * As the RTC's used in U-Boot support second resolution only
  71         * we simply ignore the sub-second field.
  72         */
  73        memcpy(&seconds, &rpktp->transmit_timestamp, sizeof(ulong));
  74
  75        rtc_to_tm(ntohl(seconds) - 2208988800UL + net_ntp_time_offset, &tm);
  76#ifdef CONFIG_DM_RTC
  77        struct udevice *dev;
  78        int ret;
  79
  80        ret = uclass_get_device(UCLASS_RTC, 0, &dev);
  81        if (ret)
  82                printf("SNTP: cannot find RTC: err=%d\n", ret);
  83        else
  84                dm_rtc_set(dev, &tm);
  85#elif defined(CONFIG_CMD_DATE)
  86        rtc_set(&tm);
  87#endif
  88        printf("Date: %4d-%02d-%02d Time: %2d:%02d:%02d\n",
  89               tm.tm_year, tm.tm_mon, tm.tm_mday,
  90               tm.tm_hour, tm.tm_min, tm.tm_sec);
  91
  92        net_set_state(NETLOOP_SUCCESS);
  93}
  94
  95/*
  96 * SNTP:
  97 *
  98 *      Prerequisites:  - own ethernet address
  99 *                      - own IP address
 100 *      We want:        - network time
 101 *      Next step:      none
 102 */
 103int sntp_prereq(void *data)
 104{
 105        if (net_ntp_server.s_addr == 0) {
 106                puts("*** ERROR: NTP server address not given\n");
 107                return 1;
 108        }
 109
 110        return 0;
 111}
 112
 113int sntp_start(void *data)
 114{
 115        debug("%s\n", __func__);
 116
 117        net_set_timeout_handler(SNTP_TIMEOUT, sntp_timeout_handler);
 118        net_set_udp_handler(sntp_handler);
 119        memset(net_server_ethaddr, 0, sizeof(net_server_ethaddr));
 120
 121        sntp_send();
 122
 123        return 0;
 124}
 125