1
2
3
4
5
6
7
8#include <common.h>
9#include <command.h>
10#include <dm.h>
11#include <net.h>
12#include <rtc.h>
13
14#include "sntp.h"
15
16#define SNTP_TIMEOUT 10000UL
17
18static int sntp_our_port;
19
20static void sntp_send(void)
21{
22 struct sntp_pkt_t pkt;
23 int pktlen = SNTP_PACKET_LEN;
24 int sport;
25
26 debug("%s\n", __func__);
27
28 memset(&pkt, 0, sizeof(pkt));
29
30 pkt.li = NTP_LI_NOLEAP;
31 pkt.vn = NTP_VERSION;
32 pkt.mode = NTP_MODE_CLIENT;
33
34 memcpy((char *)net_tx_packet + net_eth_hdr_size() + IP_UDP_HDR_SIZE,
35 (char *)&pkt, pktlen);
36
37 sntp_our_port = 10000 + (get_timer(0) % 4096);
38 sport = NTP_SERVICE_PORT;
39
40 net_send_udp_packet(net_server_ethaddr, net_ntp_server, sport,
41 sntp_our_port, pktlen);
42}
43
44static void sntp_timeout_handler(void)
45{
46 puts("Timeout\n");
47 net_set_state(NETLOOP_FAIL);
48 return;
49}
50
51static void sntp_handler(uchar *pkt, unsigned dest, struct in_addr sip,
52 unsigned src, unsigned len)
53{
54#ifdef CONFIG_TIMESTAMP
55 struct sntp_pkt_t *rpktp = (struct sntp_pkt_t *)pkt;
56 struct rtc_time tm;
57 ulong seconds;
58#endif
59
60 debug("%s\n", __func__);
61
62 if (dest != sntp_our_port)
63 return;
64
65#ifdef CONFIG_TIMESTAMP
66
67
68
69
70 memcpy(&seconds, &rpktp->transmit_timestamp, sizeof(ulong));
71
72 rtc_to_tm(ntohl(seconds) - 2208988800UL + net_ntp_time_offset, &tm);
73#if defined(CONFIG_CMD_DATE)
74# ifdef CONFIG_DM_RTC
75 struct udevice *dev;
76 int ret;
77
78 ret = uclass_get_device(UCLASS_RTC, 0, &dev);
79 if (ret)
80 printf("SNTP: cannot find RTC: err=%d\n", ret);
81 else
82 dm_rtc_set(dev, &tm);
83# else
84 rtc_set(&tm);
85# endif
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#endif
91
92 net_set_state(NETLOOP_SUCCESS);
93}
94
95void sntp_start(void)
96{
97 debug("%s\n", __func__);
98
99 net_set_timeout_handler(SNTP_TIMEOUT, sntp_timeout_handler);
100 net_set_udp_handler(sntp_handler);
101 memset(net_server_ethaddr, 0, sizeof(net_server_ethaddr));
102
103 sntp_send();
104}
105