util/TLSProxy/HelloRetryRequest.pm

   1 # Copyright 2016 The OpenSSL Project Authors. All Rights Reserved.
   2 #
   3 # Licensed under the OpenSSL license (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::HelloRetryRequest;
  11
  12 use vars '@ISA';
  13 push @ISA, 'TLSProxy::Message';
  14
  15 sub new
  16 {
  17     my $class = shift;
  18     my ($server,
  19         $data,
  20         $records,
  21         $startoffset,
  22         $message_frag_lens) = @_;
  23
  24     my $self = $class->SUPER::new(
  25         $server,
  26         TLSProxy::Message::MT_HELLO_RETRY_REQUEST,
  27         $data,
  28         $records,
  29         $startoffset,
  30         $message_frag_lens);
  31
  32     $self->{extension_data} = "";
  33
  34     return $self;
  35 }
  36
  37 sub parse
  38 {
  39     my $self = shift;
  40     my $ptr = 2;
  41
  42     TLSProxy::Proxy->is_tls13(1);
  43
  44     my ($server_version) = unpack('n', $self->data);
  45     # TODO(TLS1.3): Replace this reference to draft version before release
  46     if ($server_version == TLSProxy::Record::VERS_TLS_1_3_DRAFT) {
  47         $server_version = TLSProxy::Record::VERS_TLS_1_3;
  48     }
  49
  50     my $ciphersuite = unpack('n', substr($self->data, $ptr));
  51     $ptr += 2;
  52
  53     my $extensions_len = unpack('n', substr($self->data, $ptr));
  54     if (!defined $extensions_len) {
  55         $extensions_len = 0;
  56     }
  57
  58     $ptr += 2;
  59     my $extension_data;
  60     if ($extensions_len != 0) {
  61         $extension_data = substr($self->data, $ptr);
  62
  63         if (length($extension_data) != $extensions_len) {
  64             die "Invalid extension length\n";
  65         }
  66     } else {
  67         if (length($self->data) != 2) {
  68             die "Invalid extension length\n";
  69         }
  70         $extension_data = "";
  71     }
  72     my %extensions = ();
  73     while (length($extension_data) >= 4) {
  74         my ($type, $size) = unpack("nn", $extension_data);
  75         my $extdata = substr($extension_data, 4, $size);
  76         $extension_data = substr($extension_data, 4 + $size);
  77         $extensions{$type} = $extdata;
  78     }
  79
  80     $self->server_version($server_version);
  81     $self->ciphersuite($ciphersuite);
  82     $self->extension_data(\%extensions);
  83
  84     print "    Extensions Len:".$extensions_len."\n";
  85 }
  86
  87 #Reconstruct the on-the-wire message data following changes
  88 sub set_message_contents
  89 {
  90     my $self = shift;
  91     my $data;
  92     my $extensions = "";
  93
  94     foreach my $key (keys %{$self->extension_data}) {
  95         my $extdata = ${$self->extension_data}{$key};
  96         $extensions .= pack("n", $key);
  97         $extensions .= pack("n", length($extdata));
  98         $extensions .= $extdata;
  99         if ($key == TLSProxy::Message::EXT_DUPLICATE_EXTENSION) {
 100             $extensions .= pack("n", $key);
 101             $extensions .= pack("n", length($extdata));
 102             $extensions .= $extdata;
 103         }
 104     }
 105
 106     $data = pack('n', $self->server_version);
 107     $data .= pack('n', $self->ciphersuite);
 108     $data .= pack('n', length($extensions));
 109     $data .= $extensions;
 110     $self->data($data);
 111 }
 112
 113 #Read/write accessors
 114 sub server_version
 115 {
 116     my $self = shift;
 117     if (@_) {
 118       $self->{server_version} = shift;
 119     }
 120     return $self->{server_version};
 121 }
 122 sub ciphersuite
 123 {
 124     my $self = shift;
 125     if (@_) {
 126       $self->{ciphersuite} = shift;
 127     }
 128     return $self->{ciphersuite};
 129 }
 130 sub extension_data
 131 {
 132     my $self = shift;
 133     if (@_) {
 134         $self->{extension_data} = shift;
 135     }
 136     return $self->{extension_data};
 137 }
 138 sub set_extension
 139 {
 140     my ($self, $ext_type, $ext_data) = @_;
 141     $self->{extension_data}{$ext_type} = $ext_data;
 142 }
 143 sub delete_extension
 144 {
 145     my ($self, $ext_type) = @_;
 146     delete $self->{extension_data}{$ext_type};
 147 }
 148 1;