github.com/noisysockets/noisysockets@v0.21.2-0.20240515114641-7f467e651c90/internal/tai64n/tai64n_test.go (about)

     1  // SPDX-License-Identifier: MPL-2.0
     2  /*
     3   * Copyright (C) 2024 The Noisy Sockets Authors.
     4   *
     5   * This Source Code Form is subject to the terms of the Mozilla Public
     6   * License, v. 2.0. If a copy of the MPL was not distributed with this
     7   * file, You can obtain one at http://mozilla.org/MPL/2.0/.
     8   *
     9   * Portions of this file are based on code originally from wireguard-go,
    10   *
    11   * Copyright (C) 2017-2023 WireGuard LLC. All Rights Reserved.
    12   *
    13   * Permission is hereby granted, free of charge, to any person obtaining a copy of
    14   * this software and associated documentation files (the "Software"), to deal in
    15   * the Software without restriction, including without limitation the rights to
    16   * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
    17   * of the Software, and to permit persons to whom the Software is furnished to do
    18   * so, subject to the following conditions:
    19   *
    20   * The above copyright notice and this permission notice shall be included in all
    21   * copies or substantial portions of the Software.
    22   *
    23   * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
    24   * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
    25   * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
    26   * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
    27   * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
    28   * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
    29   * SOFTWARE.
    30   */
    31  
    32  package tai64n
    33  
    34  import (
    35  	"testing"
    36  	"time"
    37  )
    38  
    39  // Test that timestamps are monotonic as required by Wireguard and that
    40  // nanosecond-level information is whitened to prevent side channel attacks.
    41  func TestMonotonic(t *testing.T) {
    42  	startTime := time.Unix(0, 123456789) // a nontrivial bit pattern
    43  	// Whitening should reduce timestamp granularity
    44  	// to more than 10 but fewer than 20 milliseconds.
    45  	tests := []struct {
    46  		name      string
    47  		t1, t2    time.Time
    48  		wantAfter bool
    49  	}{
    50  		{"after_10_ns", startTime, startTime.Add(10 * time.Nanosecond), false},
    51  		{"after_10_us", startTime, startTime.Add(10 * time.Microsecond), false},
    52  		{"after_1_ms", startTime, startTime.Add(time.Millisecond), false},
    53  		{"after_10_ms", startTime, startTime.Add(10 * time.Millisecond), false},
    54  		{"after_20_ms", startTime, startTime.Add(20 * time.Millisecond), true},
    55  	}
    56  
    57  	for _, tt := range tests {
    58  		t.Run(tt.name, func(t *testing.T) {
    59  			ts1, ts2 := stamp(tt.t1), stamp(tt.t2)
    60  			got := ts2.After(ts1)
    61  			if got != tt.wantAfter {
    62  				t.Errorf("after = %v; want %v", got, tt.wantAfter)
    63  			}
    64  		})
    65  	}
    66  }