1
2
3
4
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
22struct in_addr net_ntp_server;
23
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
71
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
97
98
99
100
101
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