Raise an error on syscall failure in tls_retry_write_records
[openssl.git] / util / perl / TLSProxy / Certificate.pm
1 # Copyright 2016-2024 The OpenSSL Project Authors. All Rights Reserved.
2 #
3 # Licensed under the Apache License 2.0 (the "License").  You may not use
4 # this file except in compliance with the License.  You can obtain a copy
5 # in the file LICENSE in the source distribution or at
6 # https://www.openssl.org/source/license.html
7
8 use strict;
9
10 package TLSProxy::Certificate;
11
12 use vars '@ISA';
13 push @ISA, 'TLSProxy::Message';
14
15 sub new
16 {
17     my $class = shift;
18     my ($isdtls,
19         $server,
20         $msgseq,
21         $msgfrag,
22         $msgfragoffs,
23         $data,
24         $records,
25         $startoffset,
26         $message_frag_lens) = @_;
27
28     my $self = $class->SUPER::new(
29         $isdtls,
30         $server,
31         TLSProxy::Message::MT_CERTIFICATE,
32         $msgseq,
33         $msgfrag,
34         $msgfragoffs,
35         $data,
36         $records,
37         $startoffset,
38         $message_frag_lens);
39
40     $self->{first_certificate} = "";
41     $self->{extension_data} = "";
42     $self->{remaining_certdata} = "";
43
44     return $self;
45 }
46
47 sub parse
48 {
49     my $self = shift;
50
51     if (TLSProxy::Proxy->is_tls13()) {
52         my $context_len = unpack('C', $self->data);
53         my $context = substr($self->data, 1, $context_len);
54
55         my $remdata = substr($self->data, 1 + $context_len);
56
57         my ($hicertlistlen, $certlistlen) = unpack('Cn', $remdata);
58         $certlistlen += ($hicertlistlen << 16);
59
60         $remdata = substr($remdata, 3);
61
62         die "Invalid Certificate List length"
63             if length($remdata) != $certlistlen;
64
65         my ($hicertlen, $certlen) = unpack('Cn', $remdata);
66         $certlen += ($hicertlen << 16);
67
68         die "Certificate too long" if ($certlen + 3) > $certlistlen;
69
70         $remdata = substr($remdata, 3);
71
72         my $certdata = substr($remdata, 0, $certlen);
73
74         $remdata = substr($remdata, $certlen);
75
76         my $extensions_len = unpack('n', $remdata);
77         $remdata = substr($remdata, 2);
78
79         die "Extensions too long"
80             if ($certlen + 3 + $extensions_len + 2) > $certlistlen;
81
82         my $extension_data = "";
83         if ($extensions_len != 0) {
84             $extension_data = substr($remdata, 0, $extensions_len);
85
86             if (length($extension_data) != $extensions_len) {
87                 die "Invalid extension length\n";
88             }
89         }
90         my %extensions = ();
91         while (length($extension_data) >= 4) {
92             my ($type, $size) = unpack("nn", $extension_data);
93             my $extdata = substr($extension_data, 4, $size);
94             $extension_data = substr($extension_data, 4 + $size);
95             $extensions{$type} = $extdata;
96         }
97         $remdata = substr($remdata, $extensions_len);
98
99         $self->context($context);
100         $self->first_certificate($certdata);
101         $self->extension_data(\%extensions);
102         $self->remaining_certdata($remdata);
103
104         print "    Context:".$context."\n";
105         print "    Certificate List Len:".$certlistlen."\n";
106         print "    Certificate Len:".$certlen."\n";
107         print "    Extensions Len:".$extensions_len."\n";
108     } else {
109         my ($hicertlistlen, $certlistlen) = unpack('Cn', $self->data);
110         $certlistlen += ($hicertlistlen << 16);
111
112         my $remdata = substr($self->data, 3);
113
114         die "Invalid Certificate List length"
115             if length($remdata) != $certlistlen;
116
117         my ($hicertlen, $certlen) = unpack('Cn', $remdata);
118         $certlen += ($hicertlen << 16);
119
120         die "Certificate too long" if ($certlen + 3) > $certlistlen;
121
122         $remdata = substr($remdata, 3);
123
124         my $certdata = substr($remdata, 0, $certlen);
125
126         $remdata = substr($remdata, $certlen);
127
128         $self->first_certificate($certdata);
129         $self->remaining_certdata($remdata);
130
131         print "    Certificate List Len:".$certlistlen."\n";
132         print "    Certificate Len:".$certlen."\n";
133     }
134 }
135
136 #Reconstruct the on-the-wire message data following changes
137 sub set_message_contents
138 {
139     my $self = shift;
140     my $data;
141     my $extensions = "";
142
143     if (TLSProxy::Proxy->is_tls13()) {
144         foreach my $key (keys %{$self->extension_data}) {
145             my $extdata = ${$self->extension_data}{$key};
146             $extensions .= pack("n", $key);
147             $extensions .= pack("n", length($extdata));
148             $extensions .= $extdata;
149         }
150         $data = pack('C', length($self->context()));
151         $data .= $self->context;
152         my $certlen = length($self->first_certificate);
153         my $certlistlen = $certlen + length($extensions)
154                           + length($self->remaining_certdata);
155         my $hi = $certlistlen >> 16;
156         $certlistlen = $certlistlen & 0xffff;
157         $data .= pack('Cn', $hi, $certlistlen);
158         $hi = $certlen >> 16;
159         $certlen = $certlen & 0xffff;
160         $data .= pack('Cn', $hi, $certlen);
161         $data .= pack('n', length($extensions));
162         $data .= $extensions;
163         $data .= $self->remaining_certdata();
164         $self->data($data);
165     } else {
166         my $certlen = length($self->first_certificate);
167         my $certlistlen = $certlen + length($self->remaining_certdata);
168         my $hi = $certlistlen >> 16;
169         $certlistlen = $certlistlen & 0xffff;
170         $data .= pack('Cn', $hi, $certlistlen);
171         $hi = $certlen >> 16;
172         $certlen = $certlen & 0xffff;
173         $data .= pack('Cn', $hi, $certlen);
174         $data .= $self->remaining_certdata();
175         $self->data($data);
176     }
177 }
178
179 #Read/write accessors
180 sub context
181 {
182     my $self = shift;
183     if (@_) {
184       $self->{context} = shift;
185     }
186     return $self->{context};
187 }
188 sub first_certificate
189 {
190     my $self = shift;
191     if (@_) {
192       $self->{first_certificate} = shift;
193     }
194     return $self->{first_certificate};
195 }
196 sub remaining_certdata
197 {
198     my $self = shift;
199     if (@_) {
200       $self->{remaining_certdata} = shift;
201     }
202     return $self->{remaining_certdata};
203 }
204 sub extension_data
205 {
206     my $self = shift;
207     if (@_) {
208       $self->{extension_data} = shift;
209     }
210     return $self->{extension_data};
211 }
212 sub set_extension
213 {
214     my ($self, $ext_type, $ext_data) = @_;
215     $self->{extension_data}{$ext_type} = $ext_data;
216 }
217 sub delete_extension
218 {
219     my ($self, $ext_type) = @_;
220     delete $self->{extension_data}{$ext_type};
221 }
222 1;