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 "    Server Version:".$server_version."\n";
  85     print "    Ciphersuite:".$ciphersuite."\n";
  86     print "    Extensions Len:".$extensions_len."\n";
  87 }
  88
  89 #Reconstruct the on-the-wire message data following changes
  90 sub set_message_contents
  91 {
  92     my $self = shift;
  93     my $data;
  94     my $extensions = "";
  95
  96     foreach my $key (keys %{$self->extension_data}) {
  97         my $extdata = ${$self->extension_data}{$key};
  98         $extensions .= pack("n", $key);
  99         $extensions .= pack("n", length($extdata));
 100         $extensions .= $extdata;
 101         if ($key == TLSProxy::Message::EXT_DUPLICATE_EXTENSION) {
 102             $extensions .= pack("n", $key);
 103             $extensions .= pack("n", length($extdata));
 104             $extensions .= $extdata;
 105         }
 106     }
 107
 108     $data = pack('n', $self->server_version);
 109     $data .= pack('n', $self->ciphersuite);
 110     $data .= pack('n', length($extensions));
 111     $data .= $extensions;
 112     $self->data($data);
 113 }
 114
 115 #Read/write accessors
 116 sub server_version
 117 {
 118     my $self = shift;
 119     if (@_) {
 120       $self->{server_version} = shift;
 121     }
 122     return $self->{server_version};
 123 }
 124 sub ciphersuite
 125 {
 126     my $self = shift;
 127     if (@_) {
 128       $self->{ciphersuite} = shift;
 129     }
 130     return $self->{ciphersuite};
 131 }
 132 sub extension_data
 133 {
 134     my $self = shift;
 135     if (@_) {
 136         $self->{extension_data} = shift;
 137     }
 138     return $self->{extension_data};
 139 }
 140 sub set_extension
 141 {
 142     my ($self, $ext_type, $ext_data) = @_;
 143     $self->{extension_data}{$ext_type} = $ext_data;
 144 }
 145 sub delete_extension
 146 {
 147     my ($self, $ext_type) = @_;
 148     delete $self->{extension_data}{$ext_type};
 149 }
 150 1;