golang.org/x/net@v0.25.1-0.20240516223405-c87a5b62e243/quic/packet_protection_test.go (about)

     1  // Copyright 2023 The Go Authors. All rights reserved.
     2  // Use of this source code is governed by a BSD-style
     3  // license that can be found in the LICENSE file.
     4  
     5  //go:build go1.21
     6  
     7  package quic
     8  
     9  import (
    10  	"bytes"
    11  	"crypto/tls"
    12  	"testing"
    13  )
    14  
    15  func TestPacketProtection(t *testing.T) {
    16  	// Test cases from:
    17  	// https://www.rfc-editor.org/rfc/rfc9001#section-appendix.a
    18  	cid := unhex(`8394c8f03e515708`)
    19  	k := initialKeys(cid, clientSide)
    20  	initialClientKeys, initialServerKeys := k.w, k.r
    21  	for _, test := range []struct {
    22  		name string
    23  		k    fixedKeys
    24  		pnum packetNumber
    25  		hdr  []byte
    26  		pay  []byte
    27  		prot []byte
    28  	}{{
    29  		name: "Client Initial",
    30  		k:    initialClientKeys,
    31  		pnum: 2,
    32  		hdr: unhex(`
    33  			c300000001088394c8f03e5157080000 449e00000002
    34  		`),
    35  		pay: pad(1162, unhex(`
    36  			060040f1010000ed0303ebf8fa56f129 39b9584a3896472ec40bb863cfd3e868
    37  			04fe3a47f06a2b69484c000004130113 02010000c000000010000e00000b6578
    38  			616d706c652e636f6dff01000100000a 00080006001d00170018001000070005
    39  			04616c706e0005000501000000000033 00260024001d00209370b2c9caa47fba
    40  			baf4559fedba753de171fa71f50f1ce1 5d43e994ec74d748002b000302030400
    41  			0d0010000e0403050306030203080408 050806002d00020101001c0002400100
    42  			3900320408ffffffffffffffff050480 00ffff07048000ffff08011001048000
    43  			75300901100f088394c8f03e51570806 048000ffff
    44  		`)),
    45  		prot: unhex(`
    46  			c000000001088394c8f03e5157080000 449e7b9aec34d1b1c98dd7689fb8ec11
    47  			d242b123dc9bd8bab936b47d92ec356c 0bab7df5976d27cd449f63300099f399
    48  			1c260ec4c60d17b31f8429157bb35a12 82a643a8d2262cad67500cadb8e7378c
    49  			8eb7539ec4d4905fed1bee1fc8aafba1 7c750e2c7ace01e6005f80fcb7df6212
    50  			30c83711b39343fa028cea7f7fb5ff89 eac2308249a02252155e2347b63d58c5
    51  			457afd84d05dfffdb20392844ae81215 4682e9cf012f9021a6f0be17ddd0c208
    52  			4dce25ff9b06cde535d0f920a2db1bf3 62c23e596d11a4f5a6cf3948838a3aec
    53  			4e15daf8500a6ef69ec4e3feb6b1d98e 610ac8b7ec3faf6ad760b7bad1db4ba3
    54  			485e8a94dc250ae3fdb41ed15fb6a8e5 eba0fc3dd60bc8e30c5c4287e53805db
    55  			059ae0648db2f64264ed5e39be2e20d8 2df566da8dd5998ccabdae053060ae6c
    56  			7b4378e846d29f37ed7b4ea9ec5d82e7 961b7f25a9323851f681d582363aa5f8
    57  			9937f5a67258bf63ad6f1a0b1d96dbd4 faddfcefc5266ba6611722395c906556
    58  			be52afe3f565636ad1b17d508b73d874 3eeb524be22b3dcbc2c7468d54119c74
    59  			68449a13d8e3b95811a198f3491de3e7 fe942b330407abf82a4ed7c1b311663a
    60  			c69890f4157015853d91e923037c227a 33cdd5ec281ca3f79c44546b9d90ca00
    61  			f064c99e3dd97911d39fe9c5d0b23a22 9a234cb36186c4819e8b9c5927726632
    62  			291d6a418211cc2962e20fe47feb3edf 330f2c603a9d48c0fcb5699dbfe58964
    63  			25c5bac4aee82e57a85aaf4e2513e4f0 5796b07ba2ee47d80506f8d2c25e50fd
    64  			14de71e6c418559302f939b0e1abd576 f279c4b2e0feb85c1f28ff18f58891ff
    65  			ef132eef2fa09346aee33c28eb130ff2 8f5b766953334113211996d20011a198
    66  			e3fc433f9f2541010ae17c1bf202580f 6047472fb36857fe843b19f5984009dd
    67  			c324044e847a4f4a0ab34f719595de37 252d6235365e9b84392b061085349d73
    68  			203a4a13e96f5432ec0fd4a1ee65accd d5e3904df54c1da510b0ff20dcc0c77f
    69  			cb2c0e0eb605cb0504db87632cf3d8b4 dae6e705769d1de354270123cb11450e
    70  			fc60ac47683d7b8d0f811365565fd98c 4c8eb936bcab8d069fc33bd801b03ade
    71  			a2e1fbc5aa463d08ca19896d2bf59a07 1b851e6c239052172f296bfb5e724047
    72  			90a2181014f3b94a4e97d117b4381303 68cc39dbb2d198065ae3986547926cd2
    73  			162f40a29f0c3c8745c0f50fba3852e5 66d44575c29d39a03f0cda721984b6f4
    74  			40591f355e12d439ff150aab7613499d bd49adabc8676eef023b15b65bfc5ca0
    75  			6948109f23f350db82123535eb8a7433 bdabcb909271a6ecbcb58b936a88cd4e
    76  			8f2e6ff5800175f113253d8fa9ca8885 c2f552e657dc603f252e1a8e308f76f0
    77  			be79e2fb8f5d5fbbe2e30ecadd220723 c8c0aea8078cdfcb3868263ff8f09400
    78  			54da48781893a7e49ad5aff4af300cd8 04a6b6279ab3ff3afb64491c85194aab
    79  			760d58a606654f9f4400e8b38591356f bf6425aca26dc85244259ff2b19c41b9
    80  			f96f3ca9ec1dde434da7d2d392b905dd f3d1f9af93d1af5950bd493f5aa731b4
    81  			056df31bd267b6b90a079831aaf579be 0a39013137aac6d404f518cfd4684064
    82  			7e78bfe706ca4cf5e9c5453e9f7cfd2b 8b4c8d169a44e55c88d4a9a7f9474241
    83  			e221af44860018ab0856972e194cd934
    84  		`),
    85  	}, {
    86  		name: "Server Initial",
    87  		k:    initialServerKeys,
    88  		pnum: 1,
    89  		hdr: unhex(`
    90  			c1000000010008f067a5502a4262b500 40750001
    91  		`),
    92  		pay: unhex(`
    93  			02000000000600405a020000560303ee fce7f7b37ba1d1632e96677825ddf739
    94  			88cfc79825df566dc5430b9a045a1200 130100002e00330024001d00209d3c94
    95  			0d89690b84d08a60993c144eca684d10 81287c834d5311bcf32bb9da1a002b00
    96  			020304
    97  		`),
    98  		prot: unhex(`
    99  			cf000000010008f067a5502a4262b500 4075c0d95a482cd0991cd25b0aac406a
   100  			5816b6394100f37a1c69797554780bb3 8cc5a99f5ede4cf73c3ec2493a1839b3
   101  			dbcba3f6ea46c5b7684df3548e7ddeb9 c3bf9c73cc3f3bded74b562bfb19fb84
   102  			022f8ef4cdd93795d77d06edbb7aaf2f 58891850abbdca3d20398c276456cbc4
   103  			2158407dd074ee
   104  		`),
   105  	}, {
   106  		name: "ChaCha20_Poly1305 Short Header",
   107  		k: func() fixedKeys {
   108  			secret := unhex(`
   109  				9ac312a7f877468ebe69422748ad00a1
   110  				5443f18203a07d6060f688f30f21632b
   111  			`)
   112  			var k fixedKeys
   113  			k.init(tls.TLS_CHACHA20_POLY1305_SHA256, secret)
   114  			return k
   115  		}(),
   116  		pnum: 654360564,
   117  		hdr:  unhex(`4200bff4`),
   118  		pay:  unhex(`01`),
   119  		prot: unhex(`
   120  			4cfe4189655e5cd55c41f69080575d79 99c25a5bfb
   121  		`),
   122  	}} {
   123  		test := test
   124  		t.Run(test.name, func(t *testing.T) {
   125  			pnumLen := int(test.hdr[0]&0x03) + 1
   126  			pnumOff := len(test.hdr) - pnumLen
   127  
   128  			b := append([]byte{}, test.hdr...)
   129  			gotProt := test.k.protect(b, test.pay, pnumOff, test.pnum)
   130  			if got, want := gotProt, test.prot; !bytes.Equal(got, want) {
   131  				t.Errorf("Protected payload does not match:")
   132  				t.Errorf("got:  %x", got)
   133  				t.Errorf("want: %x", want)
   134  			}
   135  
   136  			pkt := append([]byte{}, test.prot...)
   137  			gotPay, gotNum, err := test.k.unprotect(pkt, pnumOff, test.pnum-1)
   138  			if err != nil {
   139  				t.Fatalf("Unexpected error unprotecting packet: %v", err)
   140  			}
   141  			if got, want := pkt[:len(test.hdr)], test.hdr; !bytes.Equal(got, want) {
   142  				t.Errorf("Unprotected header does not match:")
   143  				t.Errorf("got:  %x", got)
   144  				t.Errorf("want: %x", want)
   145  			}
   146  			if got, want := gotPay, test.pay; !bytes.Equal(got, want) {
   147  				t.Errorf("Unprotected payload does not match:")
   148  				t.Errorf("got:  %x", got)
   149  				t.Errorf("want: %x", want)
   150  			}
   151  			if got, want := gotNum, test.pnum; got != want {
   152  				t.Errorf("Unprotected packet number does not match: got %v, want %v", got, want)
   153  			}
   154  		})
   155  	}
   156  }
   157  
   158  func pad(n int, b []byte) []byte {
   159  	for len(b) < n {
   160  		b = append(b, 0)
   161  	}
   162  	return b
   163  }