1 /*
2  * SNTP support driver
3  *
4  * Masami Komiya <mkomiya@sonare.it> 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 
sntp_send(void)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 
sntp_timeout_handler(void)49 static void sntp_timeout_handler(void)
50 {
51 	puts("Timeout\n");
52 	net_set_state(NETLOOP_FAIL);
53 	return;
54 }
55 
sntp_handler(uchar * pkt,unsigned dest,struct in_addr sip,unsigned src,unsigned len)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 	u32 seconds;
61 
62 	debug("%s\n", __func__);
63 
64 	if (dest != sntp_our_port)
65 		return;
66 
67 	/*
68 	 * As the RTC's used in U-Boot support second resolution only
69 	 * we simply ignore the sub-second field.
70 	 */
71 	memcpy(&seconds, &rpktp->transmit_timestamp, sizeof(seconds));
72 	net_sntp_set_rtc(ntohl(seconds) - 2208988800UL + net_ntp_time_offset);
73 
74 	net_set_state(NETLOOP_SUCCESS);
75 }
76 
77 /*
78  * SNTP:
79  *
80  *	Prerequisites:	- own ethernet address
81  *			- own IP address
82  *	We want:	- network time
83  *	Next step:	none
84  */
sntp_prereq(void * data)85 int sntp_prereq(void *data)
86 {
87 	if (net_ntp_server.s_addr == 0) {
88 		puts("*** ERROR: NTP server address not given\n");
89 		return 1;
90 	}
91 
92 	return 0;
93 }
94 
sntp_start(void * data)95 int sntp_start(void *data)
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 	return 0;
106 }
107