Fix SntpClient 2036 issue (1/2) am: f663ab4276 am: 87fe4ba614 am: 741fb2db90 am: 970c036774

Original change: https://android-review.googlesource.com/c/platform/frameworks/base/+/1863054

Change-Id: I9f25e665d1feac95bbd20f53b9d8f3eae3699d70
This commit is contained in:
Neil Fuller
2021-10-19 20:21:39 +00:00
committed by Automerger Merge Worker
6 changed files with 1156 additions and 51 deletions

View File

@@ -20,13 +20,18 @@ import android.compat.annotation.UnsupportedAppUsage;
import android.os.SystemClock; import android.os.SystemClock;
import android.util.Log; import android.util.Log;
import com.android.internal.annotations.VisibleForTesting;
import com.android.internal.util.TrafficStatsConstants; import com.android.internal.util.TrafficStatsConstants;
import java.net.DatagramPacket; import java.net.DatagramPacket;
import java.net.DatagramSocket; import java.net.DatagramSocket;
import java.net.InetAddress; import java.net.InetAddress;
import java.net.UnknownHostException; import java.net.UnknownHostException;
import java.time.Duration;
import java.time.Instant;
import java.util.Arrays; import java.util.Arrays;
import java.util.Objects;
import java.util.function.Supplier;
/** /**
* {@hide} * {@hide}
@@ -64,13 +69,19 @@ public class SntpClient {
// 70 years plus 17 leap days // 70 years plus 17 leap days
private static final long OFFSET_1900_TO_1970 = ((365L * 70L) + 17L) * 24L * 60L * 60L; private static final long OFFSET_1900_TO_1970 = ((365L * 70L) + 17L) * 24L * 60L * 60L;
// system time computed from NTP server response // The source of the current system clock time, replaceable for testing.
private final Supplier<Instant> mSystemTimeSupplier;
// The last offset calculated from an NTP server response
private long mClockOffset;
// The last system time computed from an NTP server response
private long mNtpTime; private long mNtpTime;
// value of SystemClock.elapsedRealtime() corresponding to mNtpTime // The value of SystemClock.elapsedRealtime() corresponding to mNtpTime / mClockOffset
private long mNtpTimeReference; private long mNtpTimeReference;
// round trip time in milliseconds // The round trip (network) time in milliseconds
private long mRoundTripTime; private long mRoundTripTime;
private static class InvalidServerReplyException extends Exception { private static class InvalidServerReplyException extends Exception {
@@ -81,6 +92,12 @@ public class SntpClient {
@UnsupportedAppUsage @UnsupportedAppUsage
public SntpClient() { public SntpClient() {
this(Instant::now);
}
@VisibleForTesting
public SntpClient(Supplier<Instant> systemTimeSupplier) {
mSystemTimeSupplier = Objects.requireNonNull(systemTimeSupplier);
} }
/** /**
@@ -126,9 +143,11 @@ public class SntpClient {
buffer[0] = NTP_MODE_CLIENT | (NTP_VERSION << 3); buffer[0] = NTP_MODE_CLIENT | (NTP_VERSION << 3);
// get current time and write it to the request packet // get current time and write it to the request packet
final long requestTime = System.currentTimeMillis(); final Instant requestTime = mSystemTimeSupplier.get();
final long requestTimestamp = requestTime.toEpochMilli();
final long requestTicks = SystemClock.elapsedRealtime(); final long requestTicks = SystemClock.elapsedRealtime();
writeTimeStamp(buffer, TRANSMIT_TIME_OFFSET, requestTime); writeTimeStamp(buffer, TRANSMIT_TIME_OFFSET, requestTimestamp);
socket.send(request); socket.send(request);
@@ -136,42 +155,42 @@ public class SntpClient {
DatagramPacket response = new DatagramPacket(buffer, buffer.length); DatagramPacket response = new DatagramPacket(buffer, buffer.length);
socket.receive(response); socket.receive(response);
final long responseTicks = SystemClock.elapsedRealtime(); final long responseTicks = SystemClock.elapsedRealtime();
final long responseTime = requestTime + (responseTicks - requestTicks); final Instant responseTime = requestTime.plusMillis(responseTicks - requestTicks);
final long responseTimestamp = responseTime.toEpochMilli();
// extract the results // extract the results
final byte leap = (byte) ((buffer[0] >> 6) & 0x3); final byte leap = (byte) ((buffer[0] >> 6) & 0x3);
final byte mode = (byte) (buffer[0] & 0x7); final byte mode = (byte) (buffer[0] & 0x7);
final int stratum = (int) (buffer[1] & 0xff); final int stratum = (int) (buffer[1] & 0xff);
final long originateTime = readTimeStamp(buffer, ORIGINATE_TIME_OFFSET); final long originateTimestamp = readTimeStamp(buffer, ORIGINATE_TIME_OFFSET);
final long receiveTime = readTimeStamp(buffer, RECEIVE_TIME_OFFSET); final long receiveTimestamp = readTimeStamp(buffer, RECEIVE_TIME_OFFSET);
final long transmitTime = readTimeStamp(buffer, TRANSMIT_TIME_OFFSET); final long transmitTimestamp = readTimeStamp(buffer, TRANSMIT_TIME_OFFSET);
final long referenceTime = readTimeStamp(buffer, REFERENCE_TIME_OFFSET); final long referenceTimestamp = readTimeStamp(buffer, REFERENCE_TIME_OFFSET);
/* Do validation according to RFC */ /* Do validation according to RFC */
// TODO: validate originateTime == requestTime. // TODO: validate originateTime == requestTime.
checkValidServerReply(leap, mode, stratum, transmitTime, referenceTime); checkValidServerReply(leap, mode, stratum, transmitTimestamp, referenceTimestamp);
long roundTripTime = responseTicks - requestTicks - (transmitTime - receiveTime); long roundTripTimeMillis = responseTicks - requestTicks
// receiveTime = originateTime + transit + skew - (transmitTimestamp - receiveTimestamp);
// responseTime = transmitTime + transit - skew
// clockOffset = ((receiveTime - originateTime) + (transmitTime - responseTime))/2 Duration clockOffsetDuration = calculateClockOffset(requestTimestamp,
// = ((originateTime + transit + skew - originateTime) + receiveTimestamp, transmitTimestamp, responseTimestamp);
// (transmitTime - (transmitTime + transit - skew)))/2 long clockOffsetMillis = clockOffsetDuration.toMillis();
// = ((transit + skew) + (transmitTime - transmitTime - transit + skew))/2
// = (transit + skew - transit + skew)/2 EventLogTags.writeNtpSuccess(
// = (2 * skew)/2 = skew address.toString(), roundTripTimeMillis, clockOffsetMillis);
long clockOffset = ((receiveTime - originateTime) + (transmitTime - responseTime))/2;
EventLogTags.writeNtpSuccess(address.toString(), roundTripTime, clockOffset);
if (DBG) { if (DBG) {
Log.d(TAG, "round trip: " + roundTripTime + "ms, " + Log.d(TAG, "round trip: " + roundTripTimeMillis + "ms, "
"clock offset: " + clockOffset + "ms"); + "clock offset: " + clockOffsetMillis + "ms");
} }
// save our results - use the times on this side of the network latency // save our results - use the times on this side of the network latency
// (response rather than request time) // (response rather than request time)
mNtpTime = responseTime + clockOffset; mClockOffset = clockOffsetMillis;
mNtpTime = responseTime.plus(clockOffsetDuration).toEpochMilli();
mNtpTimeReference = responseTicks; mNtpTimeReference = responseTicks;
mRoundTripTime = roundTripTime; mRoundTripTime = roundTripTimeMillis;
} catch (Exception e) { } catch (Exception e) {
EventLogTags.writeNtpFailure(address.toString(), e.toString()); EventLogTags.writeNtpFailure(address.toString(), e.toString());
if (DBG) Log.d(TAG, "request time failed: " + e); if (DBG) Log.d(TAG, "request time failed: " + e);
@@ -186,6 +205,24 @@ public class SntpClient {
return true; return true;
} }
/** Performs the NTP clock offset calculation. */
@VisibleForTesting
public static Duration calculateClockOffset(long clientRequestTimestamp,
long serverReceiveTimestamp, long serverTransmitTimestamp,
long clientResponseTimestamp) {
// receiveTime = originateTime + transit + skew
// responseTime = transmitTime + transit - skew
// clockOffset = ((receiveTime - originateTime) + (transmitTime - responseTime))/2
// = ((originateTime + transit + skew - originateTime) +
// (transmitTime - (transmitTime + transit - skew)))/2
// = ((transit + skew) + (transmitTime - transmitTime - transit + skew))/2
// = (transit + skew - transit + skew)/2
// = (2 * skew)/2 = skew
long clockOffsetMillis = ((serverReceiveTimestamp - clientRequestTimestamp)
+ (serverTransmitTimestamp - clientResponseTimestamp)) / 2;
return Duration.ofMillis(clockOffsetMillis);
}
@Deprecated @Deprecated
@UnsupportedAppUsage @UnsupportedAppUsage
public boolean requestTime(String host, int timeout) { public boolean requestTime(String host, int timeout) {
@@ -193,6 +230,14 @@ public class SntpClient {
return false; return false;
} }
/**
* Returns the offset calculated to apply to the client clock to arrive at {@link #getNtpTime()}
*/
@VisibleForTesting
public long getClockOffset() {
return mClockOffset;
}
/** /**
* Returns the time computed from the NTP transaction. * Returns the time computed from the NTP transaction.
* *

View File

@@ -0,0 +1,141 @@
/*
* Copyright (C) 2021 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package android.net.sntp;
import java.time.Duration;
/**
* A type similar to {@link Timestamp64} but used when calculating the difference between two
* timestamps. As such, it is a signed type, but still uses 64-bits in total and so can only
* represent half the magnitude of {@link Timestamp64}.
*
* <p>See <a href="https://www.eecis.udel.edu/~mills/time.html">4. Time Difference Calculations</a>.
*
* @hide
*/
public class Duration64 {
public static final Duration64 ZERO = new Duration64(0);
private final long mBits;
private Duration64(long bits) {
this.mBits = bits;
}
/**
* Returns the difference between two 64-bit NTP timestamps as a {@link Duration64}, as
* described in the NTP spec. The times represented by the timestamps have to be within {@link
* Timestamp64#MAX_SECONDS_IN_ERA} (~68 years) of each other for the calculation to produce a
* correct answer.
*/
public static Duration64 between(Timestamp64 startInclusive, Timestamp64 endExclusive) {
long oneBits = (startInclusive.getEraSeconds() << 32)
| (startInclusive.getFractionBits() & 0xFFFF_FFFFL);
long twoBits = (endExclusive.getEraSeconds() << 32)
| (endExclusive.getFractionBits() & 0xFFFF_FFFFL);
long resultBits = twoBits - oneBits;
return new Duration64(resultBits);
}
/**
* Add two {@link Duration64} instances together. This performs the calculation in {@link
* Duration} and returns a {@link Duration} to increase the magnitude of accepted arguments,
* since {@link Duration64} only supports signed 32-bit seconds. The use of {@link Duration}
* limits precision to nanoseconds.
*/
public Duration plus(Duration64 other) {
// From https://www.eecis.udel.edu/~mills/time.html:
// "The offset and delay calculations require sums and differences of these raw timestamp
// differences that can span no more than from 34 years in the future to 34 years in the
// past without overflow. This is a fundamental limitation in 64-bit integer calculations.
//
// In the NTPv4 reference implementation, all calculations involving offset and delay values
// use 64-bit floating double arithmetic, with the exception of raw timestamp subtraction,
// as mentioned above. The raw timestamp differences are then converted to 64-bit floating
// double format without loss of precision or chance of overflow in subsequent
// calculations."
//
// Here, we use Duration instead, which provides sufficient range, but loses precision below
// nanos.
return this.toDuration().plus(other.toDuration());
}
/**
* Returns a {@link Duration64} equivalent of the supplied duration, if the magnitude can be
* represented. Because {@link Duration64} uses a fixed point type for sub-second values it
* cannot represent all nanosecond values precisely and so the conversion can be lossy.
*
* @throws IllegalArgumentException if the supplied duration is too big to be represented
*/
public static Duration64 fromDuration(Duration duration) {
long seconds = duration.getSeconds();
if (seconds < Integer.MIN_VALUE || seconds > Integer.MAX_VALUE) {
throw new IllegalArgumentException();
}
long bits = (seconds << 32)
| (Timestamp64.nanosToFractionBits(duration.getNano()) & 0xFFFF_FFFFL);
return new Duration64(bits);
}
/**
* Returns a {@link Duration} equivalent of this duration. Because {@link Duration64} uses a
* fixed point type for sub-second values it can values smaller than nanosecond precision and so
* the conversion can be lossy.
*/
public Duration toDuration() {
int seconds = getSeconds();
int nanos = getNanos();
return Duration.ofSeconds(seconds, nanos);
}
@Override
public boolean equals(Object o) {
if (this == o) {
return true;
}
if (o == null || getClass() != o.getClass()) {
return false;
}
Duration64 that = (Duration64) o;
return mBits == that.mBits;
}
@Override
public int hashCode() {
return java.util.Objects.hash(mBits);
}
@Override
public String toString() {
Duration duration = toDuration();
return Long.toHexString(mBits)
+ "(" + duration.getSeconds() + "s " + duration.getNano() + "ns)";
}
/**
* Returns the <em>signed</em> seconds in this duration.
*/
public int getSeconds() {
return (int) (mBits >> 32);
}
/**
* Returns the <em>unsigned</em> nanoseconds in this duration (truncated).
*/
public int getNanos() {
return Timestamp64.fractionBitsToNanos((int) (mBits & 0xFFFF_FFFFL));
}
}

View File

@@ -0,0 +1,186 @@
/*
* Copyright (C) 2021 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package android.net.sntp;
import com.android.internal.annotations.VisibleForTesting;
import java.time.Instant;
import java.util.Objects;
import java.util.Random;
/**
* The 64-bit type ("timestamp") that NTP uses to represent a point in time. It only holds the
* lowest 32-bits of the number of seconds since 1900-01-01 00:00:00. Consequently, to turn an
* instance into an unambiguous point in time the era number must be known. Era zero runs from
* 1900-01-01 00:00:00 to a date in 2036.
*
* It stores sub-second values using a 32-bit fixed point type, so it can resolve values smaller
* than a nanosecond, but is imprecise (i.e. it truncates).
*
* See also <a href=https://www.eecis.udel.edu/~mills/y2k.html>NTP docs</a>.
*
* @hide
*/
public final class Timestamp64 {
public static final Timestamp64 ZERO = fromComponents(0, 0);
static final int SUB_MILLIS_BITS_TO_RANDOMIZE = 32 - 10;
// Number of seconds between Jan 1, 1900 and Jan 1, 1970
// 70 years plus 17 leap days
static final long OFFSET_1900_TO_1970 = ((365L * 70L) + 17L) * 24L * 60L * 60L;
static final long MAX_SECONDS_IN_ERA = 0xFFFF_FFFFL;
static final long SECONDS_IN_ERA = MAX_SECONDS_IN_ERA + 1;
static final int NANOS_PER_SECOND = 1_000_000_000;
/** Creates a {@link Timestamp64} from the seconds and fraction components. */
public static Timestamp64 fromComponents(long eraSeconds, int fractionBits) {
return new Timestamp64(eraSeconds, fractionBits);
}
/** Creates a {@link Timestamp64} by decoding a string in the form "e4dc720c.4d4fc9eb". */
public static Timestamp64 fromString(String string) {
final int requiredLength = 17;
if (string.length() != requiredLength || string.charAt(8) != '.') {
throw new IllegalArgumentException(string);
}
String eraSecondsString = string.substring(0, 8);
String fractionString = string.substring(9);
long eraSeconds = Long.parseLong(eraSecondsString, 16);
// Use parseLong() because the type is unsigned. Integer.parseInt() will reject 0x70000000
// or above as being out of range.
long fractionBitsAsLong = Long.parseLong(fractionString, 16);
if (fractionBitsAsLong < 0 || fractionBitsAsLong > 0xFFFFFFFFL) {
throw new IllegalArgumentException("Invalid fractionBits:" + fractionString);
}
return new Timestamp64(eraSeconds, (int) fractionBitsAsLong);
}
/**
* Converts an {@link Instant} into a {@link Timestamp64}. This is lossy: Timestamp64 only
* contains the number of seconds in a given era, but the era is not stored. Also, sub-second
* values are not stored precisely.
*/
public static Timestamp64 fromInstant(Instant instant) {
long ntpEraSeconds = instant.getEpochSecond() + OFFSET_1900_TO_1970;
if (ntpEraSeconds < 0) {
ntpEraSeconds = SECONDS_IN_ERA - (-ntpEraSeconds % SECONDS_IN_ERA);
}
ntpEraSeconds %= SECONDS_IN_ERA;
long nanos = instant.getNano();
int fractionBits = nanosToFractionBits(nanos);
return new Timestamp64(ntpEraSeconds, fractionBits);
}
private final long mEraSeconds;
private final int mFractionBits;
private Timestamp64(long eraSeconds, int fractionBits) {
if (eraSeconds < 0 || eraSeconds > MAX_SECONDS_IN_ERA) {
throw new IllegalArgumentException(
"Invalid parameters. seconds=" + eraSeconds + ", fraction=" + fractionBits);
}
this.mEraSeconds = eraSeconds;
this.mFractionBits = fractionBits;
}
/** Returns the number of seconds in the NTP era. */
public long getEraSeconds() {
return mEraSeconds;
}
/** Returns the fraction of a second as 32-bit, unsigned fixed-point bits. */
public int getFractionBits() {
return mFractionBits;
}
@Override
public String toString() {
return String.format("%08x.%08x", mEraSeconds, mFractionBits);
}
/** Returns the instant represented by this value in the specified NTP era. */
public Instant toInstant(int ntpEra) {
long secondsSinceEpoch = mEraSeconds - OFFSET_1900_TO_1970;
secondsSinceEpoch += ntpEra * SECONDS_IN_ERA;
int nanos = fractionBitsToNanos(mFractionBits);
return Instant.ofEpochSecond(secondsSinceEpoch, nanos);
}
@Override
public boolean equals(Object o) {
if (this == o) {
return true;
}
if (o == null || getClass() != o.getClass()) {
return false;
}
Timestamp64 that = (Timestamp64) o;
return mEraSeconds == that.mEraSeconds && mFractionBits == that.mFractionBits;
}
@Override
public int hashCode() {
return Objects.hash(mEraSeconds, mFractionBits);
}
static int fractionBitsToNanos(int fractionBits) {
long fractionBitsLong = fractionBits & 0xFFFF_FFFFL;
return (int) ((fractionBitsLong * NANOS_PER_SECOND) >>> 32);
}
static int nanosToFractionBits(long nanos) {
if (nanos > NANOS_PER_SECOND) {
throw new IllegalArgumentException();
}
return (int) ((nanos << 32) / NANOS_PER_SECOND);
}
/**
* Randomizes the fraction bits that represent sub-millisecond values. i.e. the randomization
* won't change the number of milliseconds represented after truncation. This is used to
* implement the part of the NTP spec that calls for clients with millisecond accuracy clocks
* to send randomized LSB values rather than zeros.
*/
public Timestamp64 randomizeSubMillis(Random random) {
int randomizedFractionBits =
randomizeLowestBits(random, this.mFractionBits, SUB_MILLIS_BITS_TO_RANDOMIZE);
return new Timestamp64(mEraSeconds, randomizedFractionBits);
}
/**
* Randomizes the specified number of LSBs in {@code value} by using replacement bits from
* {@code Random.getNextInt()}.
*/
@VisibleForTesting
public static int randomizeLowestBits(Random random, int value, int bitsToRandomize) {
if (bitsToRandomize < 1 || bitsToRandomize >= Integer.SIZE) {
// There's no point in randomizing all bits or none of the bits.
throw new IllegalArgumentException(Integer.toString(bitsToRandomize));
}
int upperBitMask = 0xFFFF_FFFF << bitsToRandomize;
int lowerBitMask = ~upperBitMask;
int randomValue = random.nextInt();
return (value & upperBitMask) | (randomValue & lowerBitMask);
}
}

View File

@@ -22,7 +22,10 @@ import static junit.framework.Assert.assertTrue;
import static org.mockito.Mockito.CALLS_REAL_METHODS; import static org.mockito.Mockito.CALLS_REAL_METHODS;
import static org.mockito.Mockito.mock; import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
import android.net.sntp.Duration64;
import android.net.sntp.Timestamp64;
import android.util.Log; import android.util.Log;
import androidx.test.runner.AndroidJUnit4; import androidx.test.runner.AndroidJUnit4;
@@ -38,7 +41,12 @@ import java.net.DatagramPacket;
import java.net.DatagramSocket; import java.net.DatagramSocket;
import java.net.InetAddress; import java.net.InetAddress;
import java.net.SocketException; import java.net.SocketException;
import java.time.Duration;
import java.time.Instant;
import java.time.LocalDateTime;
import java.time.ZoneOffset;
import java.util.Arrays; import java.util.Arrays;
import java.util.function.Supplier;
@RunWith(AndroidJUnit4.class) @RunWith(AndroidJUnit4.class)
public class SntpClientTest { public class SntpClientTest {
@@ -54,41 +62,227 @@ public class SntpClientTest {
// //
// Server, Leap indicator: (0), Stratum 2 (secondary reference), poll 6 (64s), precision -20 // Server, Leap indicator: (0), Stratum 2 (secondary reference), poll 6 (64s), precision -20
// Root Delay: 0.005447, Root dispersion: 0.002716, Reference-ID: 221.253.71.41 // Root Delay: 0.005447, Root dispersion: 0.002716, Reference-ID: 221.253.71.41
// Reference Timestamp: 3653932102.507969856 (2015/10/15 14:08:22) // Reference Timestamp:
// Originator Timestamp: 3653932113.576327741 (2015/10/15 14:08:33) // d9ca9446.820a5000 / ERA0: 2015-10-15 21:08:22 UTC / ERA1: 2151-11-22 03:36:38 UTC
// Receive Timestamp: 3653932113.581012725 (2015/10/15 14:08:33) // Originator Timestamp:
// Transmit Timestamp: 3653932113.581012725 (2015/10/15 14:08:33) // d9ca9451.938a3771 / ERA0: 2015-10-15 21:08:33 UTC / ERA1: 2151-11-22 03:36:49 UTC
// Receive Timestamp:
// d9ca9451.94bd3fff / ERA0: 2015-10-15 21:08:33 UTC / ERA1: 2151-11-22 03:36:49 UTC
// Transmit Timestamp:
// d9ca9451.94bd4001 / ERA0: 2015-10-15 21:08:33 UTC / ERA1: 2151-11-22 03:36:49 UTC
//
// Originator - Receive Timestamp: +0.004684958 // Originator - Receive Timestamp: +0.004684958
// Originator - Transmit Timestamp: +0.004684958 // Originator - Transmit Timestamp: +0.004684958
private static final String WORKING_VERSION4 = private static final String LATE_ERA_RESPONSE =
"240206ec" + "240206ec"
"00000165" + + "00000165"
"000000b2" + + "000000b2"
"ddfd4729" + + "ddfd4729"
"d9ca9446820a5000" + + "d9ca9446820a5000"
"d9ca9451938a3771" + + "d9ca9451938a3771"
"d9ca945194bd3fff" + + "d9ca945194bd3fff"
"d9ca945194bd4001"; + "d9ca945194bd4001";
/** This is the actual UTC time in the server if it is in ERA0 */
private static final Instant LATE_ERA0_SERVER_TIME =
calculateIdealServerTime("d9ca9451.94bd3fff", "d9ca9451.94bd4001", 0);
/**
* This is the Unix epoch time matches the originate timestamp from {@link #LATE_ERA_RESPONSE}
* when interpreted as an ERA0 timestamp.
*/
private static final Instant LATE_ERA0_REQUEST_TIME =
Timestamp64.fromString("d9ca9451.938a3771").toInstant(0);
// A tweaked version of the ERA0 response to represent an ERA 1 response.
//
// Server, Leap indicator: (0), Stratum 2 (secondary reference), poll 6 (64s), precision -20
// Root Delay: 0.005447, Root dispersion: 0.002716, Reference-ID: 221.253.71.41
// Reference Timestamp:
// 1db2d246.820a5000 / ERA0: 1915-10-16 21:08:22 UTC / ERA1: 2051-11-22 03:36:38 UTC
// Originate Timestamp:
// 1db2d251.938a3771 / ERA0: 1915-10-16 21:08:33 UTC / ERA1: 2051-11-22 03:36:49 UTC
// Receive Timestamp:
// 1db2d251.94bd3fff / ERA0: 1915-10-16 21:08:33 UTC / ERA1: 2051-11-22 03:36:49 UTC
// Transmit Timestamp:
// 1db2d251.94bd4001 / ERA0: 1915-10-16 21:08:33 UTC / ERA1: 2051-11-22 03:36:49 UTC
//
// Originate - Receive Timestamp: +0.004684958
// Originate - Transmit Timestamp: +0.004684958
private static final String EARLY_ERA_RESPONSE =
"240206ec"
+ "00000165"
+ "000000b2"
+ "ddfd4729"
+ "1db2d246820a5000"
+ "1db2d251938a3771"
+ "1db2d25194bd3fff"
+ "1db2d25194bd4001";
/** This is the actual UTC time in the server if it is in ERA0 */
private static final Instant EARLY_ERA1_SERVER_TIME =
calculateIdealServerTime("1db2d251.94bd3fff", "1db2d251.94bd4001", 1);
/**
* This is the Unix epoch time matches the originate timestamp from {@link #EARLY_ERA_RESPONSE}
* when interpreted as an ERA1 timestamp.
*/
private static final Instant EARLY_ERA1_REQUEST_TIME =
Timestamp64.fromString("1db2d251.938a3771").toInstant(1);
private SntpTestServer mServer; private SntpTestServer mServer;
private SntpClient mClient; private SntpClient mClient;
private Network mNetwork; private Network mNetwork;
private Supplier<Instant> mSystemTimeSupplier;
@SuppressWarnings("unchecked")
@Before @Before
public void setUp() throws Exception { public void setUp() throws Exception {
mServer = new SntpTestServer();
// A mock network has NETID_UNSET, which allows the test to run, with a loopback server, // A mock network has NETID_UNSET, which allows the test to run, with a loopback server,
// even w/o external networking. // even w/o external networking.
mNetwork = mock(Network.class, CALLS_REAL_METHODS); mNetwork = mock(Network.class, CALLS_REAL_METHODS);
mServer = new SntpTestServer();
mClient = new SntpClient(); mSystemTimeSupplier = mock(Supplier.class);
mClient = new SntpClient(mSystemTimeSupplier);
} }
/** Tests when the client and server are in ERA0. b/199481251. */
@Test @Test
public void testBasicWorkingSntpClientQuery() throws Exception { public void testRequestTime_era0ClientEra0RServer() throws Exception {
mServer.setServerReply(HexEncoding.decode(WORKING_VERSION4.toCharArray(), false)); when(mSystemTimeSupplier.get()).thenReturn(LATE_ERA0_REQUEST_TIME);
mServer.setServerReply(HexEncoding.decode(LATE_ERA_RESPONSE.toCharArray(), false));
assertTrue(mClient.requestTime(mServer.getAddress(), mServer.getPort(), 500, mNetwork)); assertTrue(mClient.requestTime(mServer.getAddress(), mServer.getPort(), 500, mNetwork));
assertEquals(1, mServer.numRequestsReceived()); assertEquals(1, mServer.numRequestsReceived());
assertEquals(1, mServer.numRepliesSent()); assertEquals(1, mServer.numRepliesSent());
checkRequestTimeCalcs(LATE_ERA0_REQUEST_TIME, LATE_ERA0_SERVER_TIME, mClient);
}
/** Tests when the client is behind the server and in the previous ERA. b/199481251. */
@Test
public void testRequestTime_era0ClientEra1Server() throws Exception {
when(mSystemTimeSupplier.get()).thenReturn(LATE_ERA0_REQUEST_TIME);
mServer.setServerReply(HexEncoding.decode(EARLY_ERA_RESPONSE.toCharArray(), false));
assertTrue(mClient.requestTime(mServer.getAddress(), mServer.getPort(), 500, mNetwork));
assertEquals(1, mServer.numRequestsReceived());
assertEquals(1, mServer.numRepliesSent());
checkRequestTimeCalcs(LATE_ERA0_REQUEST_TIME, EARLY_ERA1_SERVER_TIME, mClient);
}
/** Tests when the client is ahead of the server and in the next ERA. b/199481251. */
@Test
public void testRequestTime_era1ClientEra0Server() throws Exception {
when(mSystemTimeSupplier.get()).thenReturn(EARLY_ERA1_REQUEST_TIME);
mServer.setServerReply(HexEncoding.decode(LATE_ERA_RESPONSE.toCharArray(), false));
assertTrue(mClient.requestTime(mServer.getAddress(), mServer.getPort(), 500, mNetwork));
assertEquals(1, mServer.numRequestsReceived());
assertEquals(1, mServer.numRepliesSent());
checkRequestTimeCalcs(EARLY_ERA1_REQUEST_TIME, LATE_ERA0_SERVER_TIME, mClient);
}
/** Tests when the client and server are in ERA1. b/199481251. */
@Test
public void testRequestTime_era1ClientEra1Server() throws Exception {
when(mSystemTimeSupplier.get()).thenReturn(EARLY_ERA1_REQUEST_TIME);
mServer.setServerReply(HexEncoding.decode(EARLY_ERA_RESPONSE.toCharArray(), false));
assertTrue(mClient.requestTime(mServer.getAddress(), mServer.getPort(), 500, mNetwork));
assertEquals(1, mServer.numRequestsReceived());
assertEquals(1, mServer.numRepliesSent());
checkRequestTimeCalcs(EARLY_ERA1_REQUEST_TIME, EARLY_ERA1_SERVER_TIME, mClient);
}
private static void checkRequestTimeCalcs(
Instant clientTime, Instant serverTime, SntpClient client) {
// The tests don't attempt to control the elapsed time tracking, which influences the
// round trip time (i.e. time spent in due to the network), but they control everything
// else, so assertions are allowed some slop and round trip time just has to be >= 0.
assertTrue("getRoundTripTime()=" + client.getRoundTripTime(),
client.getRoundTripTime() >= 0);
// Calculate the ideal offset if nothing took any time.
long expectedOffset = serverTime.toEpochMilli() - clientTime.toEpochMilli();
long allowedSlop = (client.getRoundTripTime() / 2) + 1; // +1 to allow for truncation loss.
assertNearlyEquals(expectedOffset, client.getClockOffset(), allowedSlop);
assertNearlyEquals(clientTime.toEpochMilli() + expectedOffset,
client.getNtpTime(), allowedSlop);
}
/**
* Unit tests for the low-level offset calculations. More targeted / easier to write than the
* end-to-end tests above that simulate the server. b/199481251.
*/
@Test
public void testCalculateClockOffset() {
Instant era0Time1 = utcInstant(2021, 10, 5, 2, 2, 2, 2);
// Confirm what happens when the client and server are completely in sync.
checkCalculateClockOffset(era0Time1, era0Time1);
Instant era0Time2 = utcInstant(2021, 10, 6, 1, 1, 1, 1);
checkCalculateClockOffset(era0Time1, era0Time2);
checkCalculateClockOffset(era0Time2, era0Time1);
Instant era1Time1 = utcInstant(2061, 10, 5, 2, 2, 2, 2);
checkCalculateClockOffset(era1Time1, era1Time1);
Instant era1Time2 = utcInstant(2061, 10, 6, 1, 1, 1, 1);
checkCalculateClockOffset(era1Time1, era1Time2);
checkCalculateClockOffset(era1Time2, era1Time1);
// Cross-era calcs (requires they are still within 68 years of each other).
checkCalculateClockOffset(era0Time1, era1Time1);
checkCalculateClockOffset(era1Time1, era0Time1);
}
private void checkCalculateClockOffset(Instant clientTime, Instant serverTime) {
// The expected (ideal) offset is the difference between the client and server clocks. NTP
// assumes delays are symmetric, i.e. that the server time is between server
// receive/transmit time, client time is between request/response time, and send networking
// delay == receive networking delay.
Duration expectedOffset = Duration.between(clientTime, serverTime);
// Try simulating various round trip delays, including zero.
for (long totalElapsedTimeMillis : Arrays.asList(0, 20, 200, 2000, 20000)) {
// Simulate that a 10% of the elapsed time is due to time spent in the server, the rest
// is network / client processing time.
long simulatedServerElapsedTimeMillis = totalElapsedTimeMillis / 10;
long simulatedClientElapsedTimeMillis = totalElapsedTimeMillis;
// Create some symmetrical timestamps.
long clientRequestTimestamp =
clientTime.minusMillis(simulatedClientElapsedTimeMillis / 2).toEpochMilli();
long clientResponseTimestamp =
clientTime.plusMillis(simulatedClientElapsedTimeMillis / 2).toEpochMilli();
long serverReceiveTimestamp =
serverTime.minusMillis(simulatedServerElapsedTimeMillis / 2).toEpochMilli();
long serverTransmitTimestamp =
serverTime.plusMillis(simulatedServerElapsedTimeMillis / 2).toEpochMilli();
Duration actualOffset = SntpClient.calculateClockOffset(
clientRequestTimestamp, serverReceiveTimestamp,
serverTransmitTimestamp, clientResponseTimestamp);
// We allow up to 1ms variation because NTP types are lossy and the simulated elapsed
// time millis may not divide exactly.
int allowedSlopMillis = 1;
assertNearlyEquals(
expectedOffset.toMillis(), actualOffset.toMillis(), allowedSlopMillis);
}
}
private static Instant utcInstant(
int year, int monthOfYear, int day, int hour, int minute, int second, int nanos) {
return LocalDateTime.of(year, monthOfYear, day, hour, minute, second, nanos)
.toInstant(ZoneOffset.UTC);
} }
@Test @Test
@@ -98,6 +292,8 @@ public class SntpClientTest {
@Test @Test
public void testTimeoutFailure() throws Exception { public void testTimeoutFailure() throws Exception {
when(mSystemTimeSupplier.get()).thenReturn(LATE_ERA0_REQUEST_TIME);
mServer.clearServerReply(); mServer.clearServerReply();
assertFalse(mClient.requestTime(mServer.getAddress(), mServer.getPort(), 500, mNetwork)); assertFalse(mClient.requestTime(mServer.getAddress(), mServer.getPort(), 500, mNetwork));
assertEquals(1, mServer.numRequestsReceived()); assertEquals(1, mServer.numRequestsReceived());
@@ -106,7 +302,9 @@ public class SntpClientTest {
@Test @Test
public void testIgnoreLeapNoSync() throws Exception { public void testIgnoreLeapNoSync() throws Exception {
final byte[] reply = HexEncoding.decode(WORKING_VERSION4.toCharArray(), false); when(mSystemTimeSupplier.get()).thenReturn(LATE_ERA0_REQUEST_TIME);
final byte[] reply = HexEncoding.decode(LATE_ERA_RESPONSE.toCharArray(), false);
reply[0] |= (byte) 0xc0; reply[0] |= (byte) 0xc0;
mServer.setServerReply(reply); mServer.setServerReply(reply);
assertFalse(mClient.requestTime(mServer.getAddress(), mServer.getPort(), 500, mNetwork)); assertFalse(mClient.requestTime(mServer.getAddress(), mServer.getPort(), 500, mNetwork));
@@ -116,7 +314,9 @@ public class SntpClientTest {
@Test @Test
public void testAcceptOnlyServerAndBroadcastModes() throws Exception { public void testAcceptOnlyServerAndBroadcastModes() throws Exception {
final byte[] reply = HexEncoding.decode(WORKING_VERSION4.toCharArray(), false); when(mSystemTimeSupplier.get()).thenReturn(LATE_ERA0_REQUEST_TIME);
final byte[] reply = HexEncoding.decode(LATE_ERA_RESPONSE.toCharArray(), false);
for (int i = 0; i <= 7; i++) { for (int i = 0; i <= 7; i++) {
final String logMsg = "mode: " + i; final String logMsg = "mode: " + i;
reply[0] &= (byte) 0xf8; reply[0] &= (byte) 0xf8;
@@ -140,10 +340,12 @@ public class SntpClientTest {
@Test @Test
public void testAcceptableStrataOnly() throws Exception { public void testAcceptableStrataOnly() throws Exception {
when(mSystemTimeSupplier.get()).thenReturn(LATE_ERA0_REQUEST_TIME);
final int STRATUM_MIN = 1; final int STRATUM_MIN = 1;
final int STRATUM_MAX = 15; final int STRATUM_MAX = 15;
final byte[] reply = HexEncoding.decode(WORKING_VERSION4.toCharArray(), false); final byte[] reply = HexEncoding.decode(LATE_ERA_RESPONSE.toCharArray(), false);
for (int i = 0; i < 256; i++) { for (int i = 0; i < 256; i++) {
final String logMsg = "stratum: " + i; final String logMsg = "stratum: " + i;
reply[1] = (byte) i; reply[1] = (byte) i;
@@ -162,7 +364,9 @@ public class SntpClientTest {
@Test @Test
public void testZeroTransmitTime() throws Exception { public void testZeroTransmitTime() throws Exception {
final byte[] reply = HexEncoding.decode(WORKING_VERSION4.toCharArray(), false); when(mSystemTimeSupplier.get()).thenReturn(LATE_ERA0_REQUEST_TIME);
final byte[] reply = HexEncoding.decode(LATE_ERA_RESPONSE.toCharArray(), false);
Arrays.fill(reply, TRANSMIT_TIME_OFFSET, TRANSMIT_TIME_OFFSET + 8, (byte) 0x00); Arrays.fill(reply, TRANSMIT_TIME_OFFSET, TRANSMIT_TIME_OFFSET + 8, (byte) 0x00);
mServer.setServerReply(reply); mServer.setServerReply(reply);
assertFalse(mClient.requestTime(mServer.getAddress(), mServer.getPort(), 500, mNetwork)); assertFalse(mClient.requestTime(mServer.getAddress(), mServer.getPort(), 500, mNetwork));
@@ -170,6 +374,19 @@ public class SntpClientTest {
assertEquals(1, mServer.numRepliesSent()); assertEquals(1, mServer.numRepliesSent());
} }
@Test
public void testNonMatchingOriginateTime() throws Exception {
when(mSystemTimeSupplier.get()).thenReturn(LATE_ERA0_REQUEST_TIME);
final byte[] reply = HexEncoding.decode(LATE_ERA_RESPONSE.toCharArray(), false);
mServer.setServerReply(reply);
mServer.setGenerateValidOriginateTimestamp(false);
assertFalse(mClient.requestTime(mServer.getAddress(), mServer.getPort(), 500, mNetwork));
assertEquals(1, mServer.numRequestsReceived());
assertEquals(1, mServer.numRepliesSent());
}
private static class SntpTestServer { private static class SntpTestServer {
private final Object mLock = new Object(); private final Object mLock = new Object();
@@ -177,6 +394,7 @@ public class SntpClientTest {
private final InetAddress mAddress; private final InetAddress mAddress;
private final int mPort; private final int mPort;
private byte[] mReply; private byte[] mReply;
private boolean mGenerateValidOriginateTimestamp = true;
private int mRcvd; private int mRcvd;
private int mSent; private int mSent;
private Thread mListeningThread; private Thread mListeningThread;
@@ -201,10 +419,16 @@ public class SntpClientTest {
synchronized (mLock) { synchronized (mLock) {
mRcvd++; mRcvd++;
if (mReply == null) { continue; } if (mReply == null) { continue; }
// Copy transmit timestamp into originate timestamp. if (mGenerateValidOriginateTimestamp) {
// TODO: bounds checking. // Copy the transmit timestamp into originate timestamp: This is
System.arraycopy(ntpMsg.getData(), TRANSMIT_TIME_OFFSET, // validated by well-behaved clients.
mReply, ORIGINATE_TIME_OFFSET, 8); System.arraycopy(ntpMsg.getData(), TRANSMIT_TIME_OFFSET,
mReply, ORIGINATE_TIME_OFFSET, 8);
} else {
// Fill it with junk instead.
Arrays.fill(mReply, ORIGINATE_TIME_OFFSET,
ORIGINATE_TIME_OFFSET + 8, (byte) 0xFF);
}
ntpMsg.setData(mReply); ntpMsg.setData(mReply);
ntpMsg.setLength(mReply.length); ntpMsg.setLength(mReply.length);
try { try {
@@ -245,9 +469,38 @@ public class SntpClientTest {
} }
} }
/**
* Controls the test server's behavior of copying the client's transmit timestamp into the
* response's originate timestamp (which is required of a real server).
*/
public void setGenerateValidOriginateTimestamp(boolean enabled) {
synchronized (mLock) {
mGenerateValidOriginateTimestamp = enabled;
}
}
public InetAddress getAddress() { return mAddress; } public InetAddress getAddress() { return mAddress; }
public int getPort() { return mPort; } public int getPort() { return mPort; }
public int numRequestsReceived() { synchronized (mLock) { return mRcvd; } } public int numRequestsReceived() { synchronized (mLock) { return mRcvd; } }
public int numRepliesSent() { synchronized (mLock) { return mSent; } } public int numRepliesSent() { synchronized (mLock) { return mSent; } }
} }
/**
* Generates the "real" server time assuming it is exactly between the receive and transmit
* timestamp and in the NTP era specified.
*/
private static Instant calculateIdealServerTime(String receiveTimestampString,
String transmitTimestampString, int era) {
Timestamp64 receiveTimestamp = Timestamp64.fromString(receiveTimestampString);
Timestamp64 transmitTimestamp = Timestamp64.fromString(transmitTimestampString);
Duration serverProcessingTime =
Duration64.between(receiveTimestamp, transmitTimestamp).toDuration();
return receiveTimestamp.toInstant(era)
.plusMillis(serverProcessingTime.dividedBy(2).toMillis());
}
private static void assertNearlyEquals(long expected, long actual, long allowedSlop) {
assertTrue("expected=" + expected + ", actual=" + actual + ", allowedSlop=" + allowedSlop,
actual >= expected - allowedSlop && actual <= expected + allowedSlop);
}
} }

View File

@@ -0,0 +1,264 @@
/*
* Copyright (C) 2021 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package android.net.sntp;
import static android.net.sntp.Timestamp64.NANOS_PER_SECOND;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotEquals;
import static org.junit.Assert.assertTrue;
import org.junit.Test;
import java.time.Duration;
import java.time.Instant;
import java.time.LocalDateTime;
import java.time.ZoneOffset;
public class Duration64Test {
@Test
public void testBetween_rangeChecks() {
long maxDuration64Seconds = Timestamp64.MAX_SECONDS_IN_ERA / 2;
Timestamp64 zeroNoFrac = Timestamp64.fromComponents(0, 0);
assertEquals(Duration64.ZERO, Duration64.between(zeroNoFrac, zeroNoFrac));
{
Timestamp64 ceilNoFrac = Timestamp64.fromComponents(maxDuration64Seconds, 0);
assertEquals(Duration64.ZERO, Duration64.between(ceilNoFrac, ceilNoFrac));
long expectedNanos = maxDuration64Seconds * NANOS_PER_SECOND;
assertEquals(Duration.ofNanos(expectedNanos),
Duration64.between(zeroNoFrac, ceilNoFrac).toDuration());
assertEquals(Duration.ofNanos(-expectedNanos),
Duration64.between(ceilNoFrac, zeroNoFrac).toDuration());
}
{
// This value is the largest fraction of a second representable. It is 1-(1/2^32)), and
// so numerically larger than 999_999_999 nanos.
int fractionBits = 0xFFFF_FFFF;
Timestamp64 ceilWithFrac = Timestamp64
.fromComponents(maxDuration64Seconds, fractionBits);
assertEquals(Duration64.ZERO, Duration64.between(ceilWithFrac, ceilWithFrac));
long expectedNanos = maxDuration64Seconds * NANOS_PER_SECOND + 999_999_999;
assertEquals(
Duration.ofNanos(expectedNanos),
Duration64.between(zeroNoFrac, ceilWithFrac).toDuration());
// The -1 nanos demonstrates asymmetry due to the way Duration64 has different
// precision / range of sub-second fractions.
assertEquals(
Duration.ofNanos(-expectedNanos - 1),
Duration64.between(ceilWithFrac, zeroNoFrac).toDuration());
}
}
@Test
public void testBetween_smallSecondsOnly() {
long expectedNanos = 5L * NANOS_PER_SECOND;
assertEquals(Duration.ofNanos(expectedNanos),
Duration64.between(Timestamp64.fromComponents(5, 0),
Timestamp64.fromComponents(10, 0))
.toDuration());
assertEquals(Duration.ofNanos(-expectedNanos),
Duration64.between(Timestamp64.fromComponents(10, 0),
Timestamp64.fromComponents(5, 0))
.toDuration());
}
@Test
public void testBetween_smallSecondsAndFraction() {
// Choose a nanos values we know can be represented exactly with fixed point binary (1/2
// second, 1/4 second, etc.).
{
long expectedNanos = 5L * NANOS_PER_SECOND + 500_000_000L;
int fractionHalfSecond = 0x8000_0000;
assertEquals(Duration.ofNanos(expectedNanos),
Duration64.between(
Timestamp64.fromComponents(5, 0),
Timestamp64.fromComponents(10, fractionHalfSecond)).toDuration());
assertEquals(Duration.ofNanos(-expectedNanos),
Duration64.between(
Timestamp64.fromComponents(10, fractionHalfSecond),
Timestamp64.fromComponents(5, 0)).toDuration());
}
{
long expectedNanos = 5L * NANOS_PER_SECOND + 250_000_000L;
int fractionHalfSecond = 0x8000_0000;
int fractionQuarterSecond = 0x4000_0000;
assertEquals(Duration.ofNanos(expectedNanos),
Duration64.between(
Timestamp64.fromComponents(5, fractionQuarterSecond),
Timestamp64.fromComponents(10, fractionHalfSecond)).toDuration());
assertEquals(Duration.ofNanos(-expectedNanos),
Duration64.between(
Timestamp64.fromComponents(10, fractionHalfSecond),
Timestamp64.fromComponents(5, fractionQuarterSecond)).toDuration());
}
}
@Test
public void testBetween_sameEra0() {
int arbitraryEra0Year = 2021;
Instant one = utcInstant(arbitraryEra0Year, 1, 1, 0, 0, 0, 500);
assertNtpEraOfInstant(one, 0);
checkDuration64Behavior(one, one);
Instant two = utcInstant(arbitraryEra0Year + 1, 1, 1, 0, 0, 0, 250);
assertNtpEraOfInstant(two, 0);
checkDuration64Behavior(one, two);
checkDuration64Behavior(two, one);
}
@Test
public void testBetween_sameEra1() {
int arbitraryEra1Year = 2037;
Instant one = utcInstant(arbitraryEra1Year, 1, 1, 0, 0, 0, 500);
assertNtpEraOfInstant(one, 1);
checkDuration64Behavior(one, one);
Instant two = utcInstant(arbitraryEra1Year + 1, 1, 1, 0, 0, 0, 250);
assertNtpEraOfInstant(two, 1);
checkDuration64Behavior(one, two);
checkDuration64Behavior(two, one);
}
/**
* Tests that two timestamps can originate from times in different eras, and the works
* calculation still works providing the two times aren't more than 68 years apart (half of the
* 136 years representable using an unsigned 32-bit seconds representation).
*/
@Test
public void testBetween_adjacentEras() {
int yearsSeparation = 68;
// This year just needs to be < 68 years before the end of NTP timestamp era 0.
int arbitraryYearInEra0 = 2021;
Instant one = utcInstant(arbitraryYearInEra0, 1, 1, 0, 0, 0, 500);
assertNtpEraOfInstant(one, 0);
checkDuration64Behavior(one, one);
Instant two = utcInstant(arbitraryYearInEra0 + yearsSeparation, 1, 1, 0, 0, 0, 250);
assertNtpEraOfInstant(two, 1);
checkDuration64Behavior(one, two);
checkDuration64Behavior(two, one);
}
/**
* This test confirms that duration calculations fail in the expected fashion if two
* Timestamp64s are more than 2^31 seconds apart.
*
* <p>The types / math specified by NTP for timestamps deliberately takes place in 64-bit signed
* arithmetic for the bits used to represent timestamps (32-bit unsigned integer seconds,
* 32-bits fixed point for fraction of seconds). Timestamps can therefore represent ~136 years
* of seconds.
* When subtracting one timestamp from another, we end up with a signed 32-bit seconds value.
* This means the max duration representable is ~68 years before numbers will over or underflow.
* i.e. the client and server are in the same or adjacent NTP eras and the difference in their
* clocks isn't more than ~68 years. >= ~68 years and things break down.
*/
@Test
public void testBetween_tooFarApart() {
int tooManyYearsSeparation = 68 + 1;
Instant one = utcInstant(2021, 1, 1, 0, 0, 0, 500);
assertNtpEraOfInstant(one, 0);
Instant two = utcInstant(2021 + tooManyYearsSeparation, 1, 1, 0, 0, 0, 250);
assertNtpEraOfInstant(two, 1);
checkDuration64OverflowBehavior(one, two);
checkDuration64OverflowBehavior(two, one);
}
private static void checkDuration64Behavior(Instant one, Instant two) {
// This is the answer if we perform the arithmetic in a lossless fashion.
Duration expectedDuration = Duration.between(one, two);
Duration64 expectedDuration64 = Duration64.fromDuration(expectedDuration);
// Sub-second precision is limited in Timestamp64, so we can lose 1ms.
assertEqualsOrSlightlyLessThan(
expectedDuration.toMillis(), expectedDuration64.toDuration().toMillis());
Timestamp64 one64 = Timestamp64.fromInstant(one);
Timestamp64 two64 = Timestamp64.fromInstant(two);
// This is the answer if we perform the arithmetic in a lossy fashion.
Duration64 actualDuration64 = Duration64.between(one64, two64);
assertEquals(expectedDuration64.getSeconds(), actualDuration64.getSeconds());
assertEqualsOrSlightlyLessThan(expectedDuration64.getNanos(), actualDuration64.getNanos());
}
private static void checkDuration64OverflowBehavior(Instant one, Instant two) {
// This is the answer if we perform the arithmetic in a lossless fashion.
Duration trueDuration = Duration.between(one, two);
// Confirm the maths is expected to overflow / underflow.
assertTrue(trueDuration.getSeconds() > Integer.MAX_VALUE / 2
|| trueDuration.getSeconds() < Integer.MIN_VALUE / 2);
// Now perform the arithmetic as specified for NTP: do subtraction using the 64-bit
// timestamp.
Timestamp64 one64 = Timestamp64.fromInstant(one);
Timestamp64 two64 = Timestamp64.fromInstant(two);
Duration64 actualDuration64 = Duration64.between(one64, two64);
assertNotEquals(trueDuration.getSeconds(), actualDuration64.getSeconds());
}
/**
* Asserts the instant provided is in the specified NTP timestamp era. Used to confirm /
* document values picked for tests have the properties needed.
*/
private static void assertNtpEraOfInstant(Instant one, int ntpEra) {
long expectedSeconds = one.getEpochSecond();
// The conversion to Timestamp64 is lossy (it loses the era). We then supply the expected
// era. If the era was correct, we will end up with the value we started with (modulo nano
// precision loss). If the era is wrong, we won't.
Instant roundtrippedInstant = Timestamp64.fromInstant(one).toInstant(ntpEra);
long actualSeconds = roundtrippedInstant.getEpochSecond();
assertEquals(expectedSeconds, actualSeconds);
}
/**
* Used to account for the fact that NTP types used 32-bit fixed point storage, so cannot store
* all values precisely. The value we get out will always be the value we put in, or one that is
* one unit smaller (due to truncation).
*/
private static void assertEqualsOrSlightlyLessThan(long expected, long actual) {
assertTrue("expected=" + expected + ", actual=" + actual,
expected == actual || expected == actual - 1);
}
private static Instant utcInstant(
int year, int monthOfYear, int day, int hour, int minute, int second, int nanos) {
return LocalDateTime.of(year, monthOfYear, day, hour, minute, second, nanos)
.toInstant(ZoneOffset.UTC);
}
}

View File

@@ -0,0 +1,216 @@
/*
* Copyright (C) 2021 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package android.net.sntp;
import static android.net.sntp.Timestamp64.NANOS_PER_SECOND;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
import org.junit.Test;
import java.time.Instant;
public class Timestamp64Test {
@Test
public void testFromComponents() {
long minNtpEraSeconds = 0;
long maxNtpEraSeconds = 0xFFFFFFFFL;
expectIllegalArgumentException(() -> Timestamp64.fromComponents(minNtpEraSeconds - 1, 0));
expectIllegalArgumentException(() -> Timestamp64.fromComponents(maxNtpEraSeconds + 1, 0));
assertComponentCreation(minNtpEraSeconds, 0);
assertComponentCreation(maxNtpEraSeconds, 0);
assertComponentCreation(maxNtpEraSeconds, Integer.MIN_VALUE);
assertComponentCreation(maxNtpEraSeconds, Integer.MAX_VALUE);
}
private static void assertComponentCreation(long ntpEraSeconds, int fractionBits) {
Timestamp64 value = Timestamp64.fromComponents(ntpEraSeconds, fractionBits);
assertEquals(ntpEraSeconds, value.getEraSeconds());
assertEquals(fractionBits, value.getFractionBits());
}
@Test
public void testEqualsAndHashcode() {
assertEqualsAndHashcode(0, 0);
assertEqualsAndHashcode(1, 0);
assertEqualsAndHashcode(0, 1);
}
private static void assertEqualsAndHashcode(int eraSeconds, int fractionBits) {
Timestamp64 one = Timestamp64.fromComponents(eraSeconds, fractionBits);
Timestamp64 two = Timestamp64.fromComponents(eraSeconds, fractionBits);
assertEquals(one, two);
assertEquals(one.hashCode(), two.hashCode());
}
@Test
public void testStringForm() {
expectIllegalArgumentException(() -> Timestamp64.fromString(""));
expectIllegalArgumentException(() -> Timestamp64.fromString("."));
expectIllegalArgumentException(() -> Timestamp64.fromString("1234567812345678"));
expectIllegalArgumentException(() -> Timestamp64.fromString("12345678?12345678"));
expectIllegalArgumentException(() -> Timestamp64.fromString("12345678..12345678"));
expectIllegalArgumentException(() -> Timestamp64.fromString("1.12345678"));
expectIllegalArgumentException(() -> Timestamp64.fromString("12.12345678"));
expectIllegalArgumentException(() -> Timestamp64.fromString("123456.12345678"));
expectIllegalArgumentException(() -> Timestamp64.fromString("1234567.12345678"));
expectIllegalArgumentException(() -> Timestamp64.fromString("12345678.1"));
expectIllegalArgumentException(() -> Timestamp64.fromString("12345678.12"));
expectIllegalArgumentException(() -> Timestamp64.fromString("12345678.123456"));
expectIllegalArgumentException(() -> Timestamp64.fromString("12345678.1234567"));
expectIllegalArgumentException(() -> Timestamp64.fromString("X2345678.12345678"));
expectIllegalArgumentException(() -> Timestamp64.fromString("12345678.X2345678"));
assertStringCreation("00000000.00000000", 0, 0);
assertStringCreation("00000001.00000001", 1, 1);
assertStringCreation("ffffffff.ffffffff", 0xFFFFFFFFL, 0xFFFFFFFF);
}
private static void assertStringCreation(
String string, long expectedSeconds, int expectedFractionBits) {
Timestamp64 timestamp64 = Timestamp64.fromString(string);
assertEquals(string, timestamp64.toString());
assertEquals(expectedSeconds, timestamp64.getEraSeconds());
assertEquals(expectedFractionBits, timestamp64.getFractionBits());
}
@Test
public void testStringForm_lenientHexCasing() {
Timestamp64 mixedCaseValue = Timestamp64.fromString("AaBbCcDd.EeFf1234");
assertEquals(0xAABBCCDDL, mixedCaseValue.getEraSeconds());
assertEquals(0xEEFF1234, mixedCaseValue.getFractionBits());
}
@Test
public void testFromInstant_secondsHandling() {
final int era0 = 0;
final int eraNeg1 = -1;
final int eraNeg2 = -2;
final int era1 = 1;
assertInstantCreationOnlySeconds(-Timestamp64.OFFSET_1900_TO_1970, 0, era0);
assertInstantCreationOnlySeconds(
-Timestamp64.OFFSET_1900_TO_1970 - Timestamp64.SECONDS_IN_ERA, 0, eraNeg1);
assertInstantCreationOnlySeconds(
-Timestamp64.OFFSET_1900_TO_1970 + Timestamp64.SECONDS_IN_ERA, 0, era1);
assertInstantCreationOnlySeconds(
-Timestamp64.OFFSET_1900_TO_1970 - 1, Timestamp64.MAX_SECONDS_IN_ERA, -1);
assertInstantCreationOnlySeconds(
-Timestamp64.OFFSET_1900_TO_1970 - Timestamp64.SECONDS_IN_ERA - 1,
Timestamp64.MAX_SECONDS_IN_ERA, eraNeg2);
assertInstantCreationOnlySeconds(
-Timestamp64.OFFSET_1900_TO_1970 + Timestamp64.SECONDS_IN_ERA - 1,
Timestamp64.MAX_SECONDS_IN_ERA, era0);
assertInstantCreationOnlySeconds(-Timestamp64.OFFSET_1900_TO_1970 + 1, 1, era0);
assertInstantCreationOnlySeconds(
-Timestamp64.OFFSET_1900_TO_1970 - Timestamp64.SECONDS_IN_ERA + 1, 1, eraNeg1);
assertInstantCreationOnlySeconds(
-Timestamp64.OFFSET_1900_TO_1970 + Timestamp64.SECONDS_IN_ERA + 1, 1, era1);
assertInstantCreationOnlySeconds(0, Timestamp64.OFFSET_1900_TO_1970, era0);
assertInstantCreationOnlySeconds(
-Timestamp64.SECONDS_IN_ERA, Timestamp64.OFFSET_1900_TO_1970, eraNeg1);
assertInstantCreationOnlySeconds(
Timestamp64.SECONDS_IN_ERA, Timestamp64.OFFSET_1900_TO_1970, era1);
assertInstantCreationOnlySeconds(1, Timestamp64.OFFSET_1900_TO_1970 + 1, era0);
assertInstantCreationOnlySeconds(
-Timestamp64.SECONDS_IN_ERA + 1, Timestamp64.OFFSET_1900_TO_1970 + 1, eraNeg1);
assertInstantCreationOnlySeconds(
Timestamp64.SECONDS_IN_ERA + 1, Timestamp64.OFFSET_1900_TO_1970 + 1, era1);
assertInstantCreationOnlySeconds(-1, Timestamp64.OFFSET_1900_TO_1970 - 1, era0);
assertInstantCreationOnlySeconds(
-Timestamp64.SECONDS_IN_ERA - 1, Timestamp64.OFFSET_1900_TO_1970 - 1, eraNeg1);
assertInstantCreationOnlySeconds(
Timestamp64.SECONDS_IN_ERA - 1, Timestamp64.OFFSET_1900_TO_1970 - 1, era1);
}
private static void assertInstantCreationOnlySeconds(
long epochSeconds, long expectedNtpEraSeconds, int ntpEra) {
int nanosOfSecond = 0;
Instant instant = Instant.ofEpochSecond(epochSeconds, nanosOfSecond);
Timestamp64 timestamp = Timestamp64.fromInstant(instant);
assertEquals(expectedNtpEraSeconds, timestamp.getEraSeconds());
int expectedFractionBits = 0;
assertEquals(expectedFractionBits, timestamp.getFractionBits());
// Confirm the Instant can be round-tripped if we know the era. Also assumes the nanos can
// be stored precisely; 0 can be.
Instant roundTrip = timestamp.toInstant(ntpEra);
assertEquals(instant, roundTrip);
}
@Test
public void testFromInstant_fractionHandling() {
// Try some values we know can be represented exactly.
assertInstantCreationOnlyFractionExact(0x0, 0);
assertInstantCreationOnlyFractionExact(0x80000000, 500_000_000L);
assertInstantCreationOnlyFractionExact(0x40000000, 250_000_000L);
// Test the limits of precision.
assertInstantCreationOnlyFractionExact(0x00000006, 1L);
assertInstantCreationOnlyFractionExact(0x00000005, 1L);
assertInstantCreationOnlyFractionExact(0x00000004, 0L);
assertInstantCreationOnlyFractionExact(0x00000002, 0L);
assertInstantCreationOnlyFractionExact(0x00000001, 0L);
// Confirm nanosecond storage / precision is within 1ns.
final boolean exhaustive = false;
for (int i = 0; i < NANOS_PER_SECOND; i++) {
Instant instant = Instant.ofEpochSecond(0, i);
Instant roundTripped = Timestamp64.fromInstant(instant).toInstant(0);
assertNanosWithTruncationAllowed(i, roundTripped);
if (!exhaustive) {
i += 999_999;
}
}
}
private static void assertInstantCreationOnlyFractionExact(
int fractionBits, long expectedNanos) {
Timestamp64 timestamp64 = Timestamp64.fromComponents(0, fractionBits);
final int ntpEra = 0;
Instant instant = timestamp64.toInstant(ntpEra);
assertEquals(expectedNanos, instant.getNano());
}
private static void assertNanosWithTruncationAllowed(long expectedNanos, Instant instant) {
// Allow for < 1ns difference due to truncation.
long actualNanos = instant.getNano();
assertTrue("expectedNanos=" + expectedNanos + ", actualNanos=" + actualNanos,
actualNanos == expectedNanos || actualNanos == expectedNanos - 1);
}
private static void expectIllegalArgumentException(Runnable r) {
try {
r.run();
fail();
} catch (IllegalArgumentException e) {
// Expected
}
}
}