Merge tag 'u-boot-atmel-2021.01-a' of https://gitlab.denx.de/u-boot/custodians/u...
[platform/kernel/u-boot.git] / 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 }