Tests: QUIC packets buffering support across encryption levels.

For simplicity, this doesn't include ACKs buffering.

Author: pluknet

lib/Test/Nginx/HTTP3.pm

@@ -1333,13 +1333,14 @@ sub dehuff {
 }
 
 sub raw_write {
-	my ($self, $message) = @_;
+	my ($self, $message, $level) = @_;
+	$level = 3 if !defined $level;
 
 	if ($self->{chaining}) {
-		return add_chain($self, $message);
+		return add_chain($self, $message, $level);
 	}
 
-	$self->{socket}->syswrite($self->encrypt_aead($message, 3));
+	$self->{socket}->syswrite($self->encrypt_aead($message, $level));
 }
 
 sub start_chain {
@@ -1349,20 +1350,24 @@ sub start_chain {
 }
 
 sub add_chain {
-	my ($self, $buf) = @_;
+	my ($self, $buf, $level) = @_;
 
-	if ($self->{chained_buf}) {
-		$self->{chained_buf} .= $buf;
+	if ($self->{chained_buf}{$level}) {
+		$self->{chained_buf}{$level} .= $buf;
 	} else {
-		$self->{chained_buf} = $buf;
+		$self->{chained_buf}{$level} = $buf;
 	}
 }
 
 sub send_chain {
 	my ($self) = @_;
 
 	undef $self->{chaining};
-	$self->raw_write($self->{chained_buf}) if $self->{chained_buf};
+	my $buf = join '', map {
+		$self->encrypt_aead($self->{chained_buf}{$_}, $_)
+			if defined $self->{chained_buf}{$_}
+	} 0 .. 3;
+	$self->{socket}->syswrite($buf) if $buf;
 	undef $self->{chained_buf};
 }