510Connectbot: src/ch/ethz/ssh2/transport/TransportConnection.java annotate

annotate src/ch/ethz/ssh2/transport/TransportConnection.java @ 328:459eb9b6b84e ganymed

pickup compression from trilead

author	Carl Byington <carl@five-ten-sg.com>
date	Thu, 31 Jul 2014 11:27:40 -0700
parents	071eccdff8ea
children	20d0a1356c43

rev	line source
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	1 /*
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	2 * Copyright (c) 2006-2011 Christian Plattner. All rights reserved.
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	3 * Please refer to the LICENSE.txt for licensing details.
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	4 */
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	5 package ch.ethz.ssh2.transport;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	6
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	7 import java.io.IOException;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	8 import java.io.InputStream;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	9 import java.io.OutputStream;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	10 import java.security.SecureRandom;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	11
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	12 import ch.ethz.ssh2.PacketFormatException;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	13 import ch.ethz.ssh2.compression.Compressor;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	14 import ch.ethz.ssh2.crypto.cipher.BlockCipher;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	15 import ch.ethz.ssh2.crypto.cipher.CipherInputStream;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	16 import ch.ethz.ssh2.crypto.cipher.CipherOutputStream;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	17 import ch.ethz.ssh2.crypto.cipher.NullCipher;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	18 import ch.ethz.ssh2.crypto.digest.MAC;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	19 import ch.ethz.ssh2.log.Logger;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	20 import ch.ethz.ssh2.packets.Packets;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	21
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	22 /**
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	23 * TransportConnection.
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	24 *
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	25 * @author Christian Plattner
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	26 * @version $Id: TransportConnection.java 144 2014-04-25 12:48:25Z dkocher@sudo.ch $
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	27 */
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	28 public class TransportConnection {
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	29 private static final Logger log = Logger.getLogger(TransportConnection.class);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	30
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	31 int send_seq_number = 0;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	32
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	33 int recv_seq_number = 0;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	34
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	35 CipherInputStream cis;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	36
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	37 CipherOutputStream cos;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	38
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	39 boolean useRandomPadding;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	40
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	41 /* Depends on current MAC and CIPHER */
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	42
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	43 MAC send_mac;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	44
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	45 byte[] send_mac_buffer;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	46
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	47 int send_padd_blocksize = 8;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	48
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	49 MAC recv_mac;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	50
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	51 byte[] recv_mac_buffer;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	52
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	53 byte[] recv_mac_buffer_cmp;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	54
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	55 int recv_padd_blocksize = 8;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	56
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	57 Compressor recv_comp;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	58
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	59 Compressor send_comp;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	60
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	61 boolean can_compress;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	62
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	63 byte[] recv_comp_buffer;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	64
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	65 byte[] send_comp_buffer;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	66
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	67 /* won't change */
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	68
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	69 final byte[] send_padding_buffer = new byte[256];
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	70
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	71 final byte[] send_packet_header_buffer = new byte[5];
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	72
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	73 final byte[] recv_padding_buffer = new byte[256];
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	74
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	75 final byte[] recv_packet_header_buffer = new byte[5];
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	76
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	77 boolean recv_packet_header_present = false;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	78
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	79 ClientServerHello csh;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	80
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	81 final SecureRandom rnd;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	82
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	83 public TransportConnection(InputStream is, OutputStream os, SecureRandom rnd) {
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	84 this.cis = new CipherInputStream(new NullCipher(), is);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	85 this.cos = new CipherOutputStream(new NullCipher(), os);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	86 this.rnd = rnd;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	87 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	88
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	89 public void changeRecvCipher(BlockCipher bc, MAC mac) {
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	90 cis.changeCipher(bc);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	91 recv_mac = mac;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	92 recv_mac_buffer = (mac != null) ? new byte[mac.size()] : null;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	93 recv_mac_buffer_cmp = (mac != null) ? new byte[mac.size()] : null;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	94 recv_padd_blocksize = bc.getBlockSize();
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	95
071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	96 if (recv_padd_blocksize < 8) {
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	97 recv_padd_blocksize = 8;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	98 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	99 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	100
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	101 public void changeSendCipher(BlockCipher bc, MAC mac) {
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	102 if ((bc instanceof NullCipher) == false) {
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	103 /* Only use zero byte padding for the first few packets */
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	104 useRandomPadding = true;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	105 /* Once we start encrypting, there is no way back */
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	106 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	107
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	108 cos.changeCipher(bc);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	109 send_mac = mac;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	110 send_mac_buffer = (mac != null) ? new byte[mac.size()] : null;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	111 send_padd_blocksize = bc.getBlockSize();
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	112
071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	113 if (send_padd_blocksize < 8) {
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	114 send_padd_blocksize = 8;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	115 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	116 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	117
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	118 public void changeRecvCompression(Compressor comp) {
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	119 recv_comp = comp;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	120
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	121 if (comp != null) {
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	122 recv_comp_buffer = new byte[comp.getBufferSize()];
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	123 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	124 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	125
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	126 public void changeSendCompression(Compressor comp) {
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	127 send_comp = comp;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	128
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	129 if (comp != null) {
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	130 send_comp_buffer = new byte[comp.getBufferSize()];
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	131 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	132 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	133
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	134 public void sendMessage(byte[] message) throws IOException {
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	135 sendMessage(message, 0, message.length, 0);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	136 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	137
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	138 public void sendMessage(byte[] message, int off, int len) throws IOException {
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	139 sendMessage(message, off, len, 0);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	140 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	141
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	142 public int getPacketOverheadEstimate() {
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	143 // return an estimate for the paket overhead (for send operations)
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	144 return 5 + 4 + (send_padd_blocksize - 1) + send_mac_buffer.length;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	145 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	146
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	147 public void sendMessage(byte[] message, int off, int len, int padd) throws IOException {
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	148 if (padd < 4) {
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	149 padd = 4;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	150 }
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	151 else if (padd > 64) {
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	152 padd = 64;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	153 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	154
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	155 if (send_comp != null && can_compress) {
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	156 len = send_comp.compress(message, off, len, send_comp_buffer);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	157 message = send_comp_buffer;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	158 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	159
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	160 int packet_len = 5 + len + padd; /* Minimum allowed padding is 4 */
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	161 int slack = packet_len % send_padd_blocksize;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	162
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	163 if (slack != 0) {
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	164 packet_len += (send_padd_blocksize - slack);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	165 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	166
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	167 if (packet_len < 16) {
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	168 packet_len = 16;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	169 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	170
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	171 int padd_len = packet_len - (5 + len);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	172
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	173 if (useRandomPadding) {
071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	174 for (int i = 0; i < padd_len; i = i + 4) {
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	175 /*
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	176 * don't waste calls to rnd.nextInt() (by using only 8bit of the
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	177 * output). just believe me: even though we may write here up to 3
071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	178 * bytes which won't be used, there is no "buffer overflow" (i.e.,
071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	179 * arrayindexoutofbounds). the padding buffer is big enough =) (256
071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	180 * bytes, and that is bigger than any current cipher block size + 64).
071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	181 */
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	182 int r = rnd.nextInt();
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	183 send_padding_buffer[i] = (byte) r;
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	184 send_padding_buffer[i + 1] = (byte)(r >> 8);
071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	185 send_padding_buffer[i + 2] = (byte)(r >> 16);
071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	186 send_padding_buffer[i + 3] = (byte)(r >> 24);
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	187 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	188 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	189 else {
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	190 /* use zero padding for unencrypted traffic */
071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	191 for (int i = 0; i < padd_len; i++) {
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	192 send_padding_buffer[i] = 0;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	193 }
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	194
071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	195 /* Actually this code is paranoid: we never filled any
071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	196 * bytes into the padding buffer so far, therefore it should
071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	197 * consist of zeros only.
071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	198 */
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	199 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	200
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	201 send_packet_header_buffer[0] = (byte)((packet_len - 4) >> 24);
071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	202 send_packet_header_buffer[1] = (byte)((packet_len - 4) >> 16);
071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	203 send_packet_header_buffer[2] = (byte)((packet_len - 4) >> 8);
071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	204 send_packet_header_buffer[3] = (byte)((packet_len - 4));
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	205 send_packet_header_buffer[4] = (byte) padd_len;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	206 cos.write(send_packet_header_buffer, 0, 5);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	207 cos.write(message, off, len);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	208 cos.write(send_padding_buffer, 0, padd_len);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	209
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	210 if (send_mac != null) {
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	211 send_mac.initMac(send_seq_number);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	212 send_mac.update(send_packet_header_buffer, 0, 5);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	213 send_mac.update(message, off, len);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	214 send_mac.update(send_padding_buffer, 0, padd_len);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	215 send_mac.getMac(send_mac_buffer, 0);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	216 cos.writePlain(send_mac_buffer, 0, send_mac_buffer.length);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	217 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	218
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	219 cos.flush();
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	220
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	221 if (log.isDebugEnabled()) {
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	222 log.debug("Sent " + Packets.getMessageName(message[off] & 0xff) + " " + len + " bytes payload");
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	223 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	224
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	225 send_seq_number++;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	226 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	227
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	228 public int peekNextMessageLength() throws IOException {
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	229 if (recv_packet_header_present == false) {
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	230 cis.read(recv_packet_header_buffer, 0, 5);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	231 recv_packet_header_present = true;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	232 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	233
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	234 int packet_length = ((recv_packet_header_buffer[0] & 0xff) << 24)
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	235 \| ((recv_packet_header_buffer[1] & 0xff) << 16) \| ((recv_packet_header_buffer[2] & 0xff) << 8)
071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	236 \| ((recv_packet_header_buffer[3] & 0xff));
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	237 int padding_length = recv_packet_header_buffer[4] & 0xff;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	238
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	239 if (packet_length > TransportManager.MAX_PACKET_SIZE \|\| packet_length < 12) {
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	240 throw new PacketFormatException(String.format("Illegal packet size (%d)", packet_length));
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	241 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	242
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	243 int payload_length = packet_length - padding_length - 1;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	244
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	245 if (payload_length < 0) {
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	246 throw new PacketFormatException(String.format("Illegal padding_length in packet from remote (%d)", padding_length));
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	247 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	248
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	249 return payload_length;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	250 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	251
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	252 public int receiveMessage(byte buffer[], int off, int len) throws IOException {
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	253 if (recv_packet_header_present == false) {
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	254 cis.read(recv_packet_header_buffer, 0, 5);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	255 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	256 else {
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	257 recv_packet_header_present = false;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	258 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	259
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	260 int packet_length = ((recv_packet_header_buffer[0] & 0xff) << 24)
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	261 \| ((recv_packet_header_buffer[1] & 0xff) << 16) \| ((recv_packet_header_buffer[2] & 0xff) << 8)
071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	262 \| ((recv_packet_header_buffer[3] & 0xff));
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	263 int padding_length = recv_packet_header_buffer[4] & 0xff;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	264
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	265 if (packet_length > TransportManager.MAX_PACKET_SIZE \|\| packet_length < 12) {
071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	266 throw new PacketFormatException(String.format("Illegal packet size (%d)", packet_length));
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	267 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	268
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	269 int payload_length = packet_length - padding_length - 1;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	270
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	271 if (payload_length < 0) {
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	272 throw new PacketFormatException(String.format("Illegal padding_length in packet from remote (%d)", padding_length));
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	273 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	274
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	275 if (payload_length >= len) {
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	276 throw new IOException("Receive buffer too small (" + len + ", need " + payload_length + ")");
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	277 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	278
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	279 cis.read(buffer, off, payload_length);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	280 cis.read(recv_padding_buffer, 0, padding_length);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	281
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	282 if (recv_mac != null) {
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	283 cis.readPlain(recv_mac_buffer, 0, recv_mac_buffer.length);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	284 recv_mac.initMac(recv_seq_number);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	285 recv_mac.update(recv_packet_header_buffer, 0, 5);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	286 recv_mac.update(buffer, off, payload_length);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	287 recv_mac.update(recv_padding_buffer, 0, padding_length);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	288 recv_mac.getMac(recv_mac_buffer_cmp, 0);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	289
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	290 for (int i = 0; i < recv_mac_buffer.length; i++) {
071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	291 if (recv_mac_buffer[i] != recv_mac_buffer_cmp[i]) {
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	292 throw new IOException("Remote sent corrupt MAC.");
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	293 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	294 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	295 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	296
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	297 recv_seq_number++;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	298
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	299 if (log.isDebugEnabled()) {
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	300 log.debug("Received " + Packets.getMessageName(buffer[off] & 0xff) + " " + payload_length
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	301 + " bytes payload");
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	302 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	303
307 071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	304 if (recv_comp != null && can_compress) {
071eccdff8ea fix java formatting Carl Byington <carl@five-ten-sg.com> parents: 273 diff changeset	305 int[] uncomp_len = new int[] {payload_length};
273 91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	306 buffer = recv_comp.uncompress(buffer, off, uncomp_len);
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	307 return uncomp_len[0];
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	308 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	309 else {
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	310 return payload_length;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	311 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	312 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	313
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	314 public void startCompression() {
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	315 can_compress = true;
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	316 }
91a31873c42a start conversion from trilead to ganymed Carl Byington <carl@five-ten-sg.com> parents: diff changeset	317 }

Mercurial > 510Connectbot

annotate src/ch/ethz/ssh2/transport/TransportConnection.java @ 328:459eb9b6b84e ganymed