232 lines
7.3 KiB
C
Executable File
232 lines
7.3 KiB
C
Executable File
#include <common/bk_include.h>
|
|
#include "bk_arm_arch.h"
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <netdb.h>
|
|
#include "time/ntp.h"
|
|
#include "time/time.h"
|
|
#include <driver/aon_rtc.h>
|
|
|
|
|
|
#define NTP_TIMEZONE 0
|
|
#define NTP_HOSTNAME "cn.pool.ntp.org"
|
|
#define NTP_TIMESTAMP_DELTA 2208988800ull
|
|
#define NTP_GET_TIMEOUT 1
|
|
|
|
#define LI(packet) (uint8_t) ((packet.li_vn_mode & 0xC0) >> 6) // (li & 11 000 000) >> 6
|
|
#define VN(packet) (uint8_t) ((packet.li_vn_mode & 0x38) >> 3) // (vn & 00 111 000) >> 3
|
|
#define MODE(packet) (uint8_t) ((packet.li_vn_mode & 0x07) >> 0) // (mode & 00 000 111) >> 0
|
|
|
|
// Structure that defines the 48 byte NTP packet protocol.
|
|
typedef struct {
|
|
|
|
uint8_t li_vn_mode; // Eight bits. li, vn, and mode.
|
|
// li. Two bits. Leap indicator.
|
|
// vn. Three bits. Version number of the protocol.
|
|
// mode. Three bits. Client will pick mode 3 for client.
|
|
|
|
uint8_t stratum; // Eight bits. Stratum level of the local clock.
|
|
uint8_t poll; // Eight bits. Maximum interval between successive messages.
|
|
uint8_t precision; // Eight bits. Precision of the local clock.
|
|
|
|
uint32_t rootDelay; // 32 bits. Total round trip delay time.
|
|
uint32_t rootDispersion; // 32 bits. Max error aloud from primary clock source.
|
|
uint32_t refId; // 32 bits. Reference clock identifier.
|
|
|
|
uint32_t refTm_s; // 32 bits. Reference time-stamp seconds.
|
|
uint32_t refTm_f; // 32 bits. Reference time-stamp fraction of a second.
|
|
|
|
uint32_t origTm_s; // 32 bits. Originate time-stamp seconds.
|
|
uint32_t origTm_f; // 32 bits. Originate time-stamp fraction of a second.
|
|
|
|
uint32_t rxTm_s; // 32 bits. Received time-stamp seconds.
|
|
uint32_t rxTm_f; // 32 bits. Received time-stamp fraction of a second.
|
|
|
|
uint32_t txTm_s; // 32 bits and the most important field the client cares about. Transmit time-stamp seconds.
|
|
uint32_t txTm_f; // 32 bits. Transmit time-stamp fraction of a second.
|
|
|
|
} ntp_packet; // Total: 384 bits or 48 bytes.
|
|
|
|
static ntp_packet packet = { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 };
|
|
|
|
static void ntp_error(char* msg)
|
|
{
|
|
os_printf("\033[31;22m[E/NTP]: ERROR %s\033[0m\n", msg); // Print the error message to stderr.
|
|
}
|
|
|
|
/**
|
|
* Get the UTC time from NTP server
|
|
*
|
|
* @note this function is not reentrant
|
|
*
|
|
* @return >0: success, current UTC time
|
|
* =0: get failed
|
|
*/
|
|
time_t ntp_get_time(char *host_name, uint32_t *frac_val)
|
|
{
|
|
int sockfd, n; // Socket file descriptor and the n return result from writing/reading from the socket.
|
|
|
|
int portno = 123; // NTP UDP port number.
|
|
|
|
//char* host_name = NTP_HOSTNAME; // NTP server host-name.
|
|
|
|
time_t new_time = 0;
|
|
|
|
fd_set readset;
|
|
struct timeval timeout;
|
|
|
|
// Create and zero out the packet. All 48 bytes worth.
|
|
|
|
memset(&packet, 0, sizeof(ntp_packet));
|
|
|
|
// Set the first byte's bits to 00,011,011 for li = 0, vn = 3, and mode = 3. The rest will be left set to zero.
|
|
|
|
*((char *) &packet + 0) = 0x1b; // Represents 27 in base 10 or 00011011 in base 2.
|
|
|
|
// Create a UDP socket, convert the host-name to an IP address, set the port number,
|
|
// connect to the server, send the packet, and then read in the return packet.
|
|
|
|
struct sockaddr_in serv_addr; // Server address data structure.
|
|
struct hostent *server; // Server data structure.
|
|
|
|
sockfd = socket( AF_INET, SOCK_DGRAM, IPPROTO_UDP); // Create a UDP socket.
|
|
|
|
if (sockfd < 0) {
|
|
ntp_error("opening socket");
|
|
goto __exit;
|
|
}
|
|
|
|
server = gethostbyname(host_name); // Convert URL to IP.
|
|
|
|
if (server == NULL) {
|
|
ntp_error("no such host");
|
|
goto __exit;
|
|
}
|
|
|
|
// Zero out the server address structure.
|
|
|
|
memset((char *) &serv_addr, 0, sizeof(serv_addr));
|
|
|
|
serv_addr.sin_family = AF_INET;
|
|
|
|
// Copy the server's IP address to the server address structure.
|
|
|
|
// memcpy((char *) &serv_addr.sin_addr.s_addr, (char *) server->h_addr, server->h_length);
|
|
memcpy((char *) &serv_addr.sin_addr.s_addr, (char *) server->h_addr, sizeof(serv_addr.sin_addr.s_addr));
|
|
|
|
// Convert the port number integer to network big-endian style and save it to the server address structure.
|
|
|
|
serv_addr.sin_port = htons(portno);
|
|
|
|
// Call up the server using its IP address and port number.
|
|
|
|
if (connect(sockfd, (struct sockaddr *) &serv_addr, sizeof(serv_addr)) < 0) {
|
|
ntp_error("connecting");
|
|
goto __exit;
|
|
}
|
|
|
|
// Send it the NTP packet it wants. If n == -1, it failed.
|
|
|
|
n = write(sockfd, (char*) &packet, sizeof(ntp_packet));
|
|
|
|
if (n < 0) {
|
|
ntp_error("writing to socket");
|
|
goto __exit;
|
|
}
|
|
|
|
timeout.tv_sec = NTP_GET_TIMEOUT;
|
|
timeout.tv_usec = 0;
|
|
|
|
FD_ZERO(&readset);
|
|
FD_SET(sockfd, &readset);
|
|
|
|
if (select(sockfd + 1, &readset, NULL, NULL, &timeout) <= 0) {
|
|
ntp_error("select the socket timeout(1s)");
|
|
goto __exit;
|
|
}
|
|
|
|
// Wait and receive the packet back from the server. If n == -1, it failed.
|
|
|
|
n = read(sockfd, (char*) &packet, sizeof(ntp_packet));
|
|
|
|
if (n < 0) {
|
|
ntp_error("reading from socket");
|
|
goto __exit;
|
|
}
|
|
|
|
// These two fields contain the time-stamp seconds as the packet left the NTP server.
|
|
// The number of seconds correspond to the seconds passed since 1900.
|
|
// ntohl() converts the bit/byte order from the network's to host's "endianness".
|
|
|
|
packet.txTm_s = ntohl(packet.txTm_s); // Time-stamp seconds.
|
|
packet.txTm_f = ntohl(packet.txTm_f); // Time-stamp fraction of a second.
|
|
|
|
// Extract the 32 bits that represent the time-stamp seconds (since NTP epoch) from when the packet left the server.
|
|
// Subtract 70 years worth of seconds from the seconds since 1900.
|
|
// This leaves the seconds since the UNIX epoch of 1970.
|
|
// (1900)------------------(1970)**************************************(Time Packet Left the Server)
|
|
|
|
new_time = (time_t) (packet.txTm_s - NTP_TIMESTAMP_DELTA);
|
|
if(frac_val)
|
|
*frac_val = packet.txTm_f;
|
|
|
|
__exit:
|
|
|
|
close(sockfd);
|
|
|
|
return new_time;
|
|
}
|
|
|
|
/**
|
|
* Get the local time from NTP server
|
|
*
|
|
* @return >0: success, current local time, offset timezone by NTP_TIMEZONE
|
|
* =0: get failed
|
|
*/
|
|
time_t ntp_get_local_time(uint32_t *frac_val)
|
|
{
|
|
//time_t cur_time = ntp_get_time();
|
|
|
|
char *pool[4] = {"ntp.aliyun.com","ntp.tencent.com","time.windows.com",0};
|
|
|
|
time_t cur_time = 0;
|
|
int i=0;
|
|
for(i=0; pool[i]; i++){
|
|
cur_time = ntp_get_time(pool[i], frac_val);
|
|
if (cur_time)
|
|
{
|
|
break ;
|
|
}
|
|
}
|
|
|
|
if (cur_time)
|
|
{
|
|
/* add the timezone offset for set_time/set_date */
|
|
cur_time += NTP_TIMEZONE * 3600;
|
|
}
|
|
|
|
return cur_time;
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
* Sync current local time to RTC by NTP
|
|
*
|
|
* @return >0: success, current local time, offset timezone by NTP_TIMEZONE
|
|
* =0: sync failed
|
|
*/
|
|
time_t ntp_sync_to_rtc(void)
|
|
{
|
|
uint32_t frac_val = 0;
|
|
time_t cur_time = ntp_get_local_time(&frac_val);
|
|
if (cur_time)
|
|
{
|
|
bk_printf("%s:cur_time=%u,frag=%u,us=%d\r\n", __func__, cur_time, frac_val, (uint32_t)(bk_aon_rtc_get_us()%1000000LL));
|
|
datetime_set_nano(cur_time, frac_val);
|
|
}
|
|
|
|
return cur_time;
|
|
}
|