source: rtems/cpukit/libnetworking/lib/rtems_bsdnet_ntp.c @ 8b02581

4.104.114.84.95
Last change on this file since 8b02581 was 8b02581, checked in by Ralf Corsepius <ralf.corsepius@…>, on 05/10/05 at 08:06:58

socklen_t.
  • Property mode set to 100644
File size: 5.5 KB
Line 
1/*
2 * Synchronize with an NTP server
3 *
4 * This program may be distributed and used for any purpose.
5 * I ask only that you:
6 *      1. Leave this author information intact.
7 *      2. Document any changes you make.
8 *
9 * W. Eric Norum
10 * Canadian Light Source
11 * University of Saskatchewan
12 * Saskatoon, Saskatchewan, CANADA
13 * eric@cls.usask.ca
14 *
15 *  $Id$
16 */
17
18#include <stdio.h>
19#include <string.h>
20#include <errno.h>
21#include <time.h>
22#include <limits.h>
23#include <rtems.h>
24#include <rtems/rtems_bsdnet.h>
25#include <rtems/error.h>
26#include <sys/types.h>
27#include <sys/socket.h>
28#include <netinet/in.h>
29
30#include <rtems/bsdnet/servers.h>
31
32/*
33 * RTEMS base: 1988, January 1
34 *  UNIX base: 1970, January 1
35 *   NTP base: 1900, January 1
36 */
37#define UNIX_BASE_TO_NTP_BASE (((70UL*365UL)+17UL) * (24*60*60))
38
39struct ntpPacket {
40        struct ntpPacketSmall   ntp;
41        char                    authenticator[96];
42};
43
44static int
45processPacket (struct ntpPacketSmall *p, int state, void *unused)
46{
47        time_t tbuf;
48        struct tm *lt;
49        rtems_time_of_day rt;
50        rtems_interval ticks_per_second;
51
52        if ( state )
53                return 0;
54
55        rtems_clock_get (RTEMS_CLOCK_GET_TICKS_PER_SECOND, &ticks_per_second);
56        tbuf = ntohl (p->transmit_timestamp.integer) - UNIX_BASE_TO_NTP_BASE - rtems_bsdnet_timeoffset;
57        lt = gmtime (&tbuf);
58        rt.year = lt->tm_year + 1900;
59        rt.month = lt->tm_mon + 1;
60        rt.day = lt->tm_mday;
61        rt.hour = lt->tm_hour;
62        rt.minute = lt->tm_min;
63        rt.second = lt->tm_sec;
64        rt.ticks = ntohl (p->transmit_timestamp.fraction) / (ULONG_MAX / ticks_per_second);
65        if (rt.ticks >= ticks_per_second)
66                rt.ticks = ticks_per_second - 1;
67        rtems_clock_set (&rt);
68        return 0;
69}
70
71static int
72getServerTimespec(struct ntpPacketSmall *p, int state, void *usr_data)
73{
74struct timespec *ts = usr_data;
75unsigned long long tmp;
76
77        if ( 0 == state ) {
78                ts->tv_sec  = ntohl( p->transmit_timestamp.integer );
79                ts->tv_sec -= rtems_bsdnet_timeoffset + UNIX_BASE_TO_NTP_BASE;
80
81                tmp  = 1000000000 * (unsigned long long)ntohl(p->transmit_timestamp.fraction);
82
83                ts->tv_nsec = (unsigned long) (tmp>>32);
84        }
85        return 0;
86}
87
88int rtems_bsdnet_ntp_retry_count  = 5;
89int rtems_bsdnet_ntp_timeout_secs = 5;
90int rtems_bsdnet_ntp_bcast_timeout_secs = 80;
91
92static int
93tryServer (int i, int s, rtems_bsdnet_ntp_callback_t callback, void *usr_data)
94{
95        int l = 0;
96        struct timeval tv;
97        socklen_t farlen;
98        struct sockaddr_in farAddr;
99        struct ntpPacketSmall packet;
100
101        if (i < 0)
102                tv.tv_sec = rtems_bsdnet_ntp_bcast_timeout_secs;
103        else
104                tv.tv_sec = rtems_bsdnet_ntp_timeout_secs;
105        tv.tv_usec = 0;
106        if (setsockopt (s, SOL_SOCKET, SO_RCVTIMEO, (char *)&tv, sizeof tv) < 0) {
107                fprintf (stderr, "rtems_bsdnet_get_ntp() Can't set socket receive timeout: %s\n", strerror (errno));
108                close (s);
109                return -1;
110        }
111        if (i >= 0) {
112                memset (&farAddr, 0, sizeof farAddr);
113                farAddr.sin_family = AF_INET;
114                farAddr.sin_port = htons (123);
115                farAddr.sin_addr = rtems_bsdnet_ntpserver[i];
116                memset (&packet, 0, sizeof packet);
117                packet.li_vn_mode = (3 << 3) | 3; /* NTP version 3, client */
118                if ( callback( &packet, 1, usr_data ) )
119                        return -1;
120                l = sendto (s, &packet, sizeof packet, 0, (struct sockaddr *)&farAddr, sizeof farAddr);
121                if (l != sizeof packet) {
122                        fprintf (stderr, "rtems_bsdnet_get_ntp() Can't send: %s\n", strerror (errno));
123                        return -1;
124                }
125        } else {
126                if ( callback( &packet, -1, usr_data ) )
127                        return -1;
128        }
129        farlen = sizeof farAddr;
130        i = recvfrom (s, &packet, sizeof packet, 0, (struct sockaddr *)&farAddr, &farlen);
131        if (i == 0)
132                fprintf (stderr, "rtems_bsdnet_get_ntp() Unexpected EOF");
133        if (i < 0) {
134                if ((errno == EWOULDBLOCK) || (errno == EAGAIN))
135                        return -1;
136                fprintf (stderr, "rtems_bsdnet_get_ntp() Can't receive: %s\n", strerror (errno));
137        }
138
139        if ( i >= sizeof packet &&
140                ((packet.li_vn_mode & (0x7 << 3)) == (3 << 3)) &&
141            ((packet.transmit_timestamp.integer != 0) || (packet.transmit_timestamp.fraction != 0)) &&
142                0 == callback( &packet, 0 , usr_data) )
143                return 0;
144
145        return -1;
146}
147
148int rtems_bsdnet_get_ntp(int sock, rtems_bsdnet_ntp_callback_t callback, void *usr_data)
149{
150int s = -1;
151int i;
152int retry;
153struct sockaddr_in myAddr;
154int reuseFlag;
155int ret;
156
157        if ( !callback )
158                callback = getServerTimespec;
159
160        if ( sock < 0 ) {
161        s = socket (AF_INET, SOCK_DGRAM, 0);
162        if (s < 0) {
163                fprintf (stderr, "rtems_bsdnet_get_ntp() Can't create socket: %s\n", strerror (errno));
164                return -1;
165        }
166        reuseFlag = 1;
167        if (setsockopt (s, SOL_SOCKET, SO_REUSEADDR, (char *)&reuseFlag, sizeof reuseFlag) < 0) {
168                fprintf (stderr, "rtems_bsdnet_get_ntp() Can't set socket reuse: %s\n", strerror (errno));
169                close (s);
170                return -1;
171        }
172        memset (&myAddr, 0, sizeof myAddr);
173        myAddr.sin_family = AF_INET;
174        myAddr.sin_port = htons (123);
175        myAddr.sin_addr.s_addr = htonl (INADDR_ANY);
176        if (bind (s, (struct sockaddr *)&myAddr, sizeof myAddr) < 0) {
177                fprintf (stderr, "rtems_bsdnet_get_ntp() Can't bind socket: %s\n", strerror (errno));
178                close (s);
179                return -1;
180        }
181        sock = s;
182        }
183        ret = -1;
184        for (retry = 0 ; (ret == -1) && (retry < rtems_bsdnet_ntp_retry_count) ; retry++) {
185                /*
186                 * If there's no server we just have to wait
187                 * and hope that there's an NTP broadcast
188                 * server out there somewhere.
189                 */
190                if (rtems_bsdnet_ntpserver_count < 0) {
191                        ret = tryServer (-1, sock, callback, usr_data);
192                }
193                else {
194                        for (i = 0 ; (ret == -1) && (i < rtems_bsdnet_ntpserver_count) ; i++) {
195                                ret = tryServer (i, sock, callback, usr_data);
196                        }
197                }
198        }
199        if ( s >= 0 )
200                close (s);
201        return ret;
202}
203
204int
205rtems_bsdnet_synchronize_ntp (int interval, rtems_task_priority priority)
206{
207        if (interval != 0) {
208                fprintf (stderr, "Daemon-mode note yet supported.\n");
209                errno = EINVAL;
210                return -1;
211        }
212        return rtems_bsdnet_get_ntp( -1, processPacket, 0);
213}
Note: See TracBrowser for help on using the repository browser.