Mercurial > 510Connectbot
annotate src/ch/ethz/ssh2/crypto/cipher/CipherOutputStream.java @ 357:46c2115ae1c8
update jzlib to a21be20213d66eff15904d925e9b721956a01ef7
| author | Carl Byington <carl@five-ten-sg.com> |
|---|---|
| date | Fri, 01 Aug 2014 13:34:58 -0700 |
| parents | 071eccdff8ea |
| children |
| 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.crypto.cipher; |
|
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.OutputStream; |
|
91a31873c42a
start conversion from trilead to ganymed
Carl Byington <carl@five-ten-sg.com>
parents:
diff
changeset
|
9 |
|
91a31873c42a
start conversion from trilead to ganymed
Carl Byington <carl@five-ten-sg.com>
parents:
diff
changeset
|
10 /** |
|
91a31873c42a
start conversion from trilead to ganymed
Carl Byington <carl@five-ten-sg.com>
parents:
diff
changeset
|
11 * CipherOutputStream. |
|
91a31873c42a
start conversion from trilead to ganymed
Carl Byington <carl@five-ten-sg.com>
parents:
diff
changeset
|
12 * |
|
91a31873c42a
start conversion from trilead to ganymed
Carl Byington <carl@five-ten-sg.com>
parents:
diff
changeset
|
13 * @author Christian Plattner |
|
91a31873c42a
start conversion from trilead to ganymed
Carl Byington <carl@five-ten-sg.com>
parents:
diff
changeset
|
14 * @version $Id: CipherOutputStream.java 85 2014-04-07 14:05:09Z dkocher@sudo.ch $ |
|
91a31873c42a
start conversion from trilead to ganymed
Carl Byington <carl@five-ten-sg.com>
parents:
diff
changeset
|
15 */ |
| 307 | 16 public class CipherOutputStream { |
| 17 BlockCipher currentCipher; | |
| 18 OutputStream bo; | |
| 19 byte[] buffer; | |
| 20 byte[] enc; | |
| 21 int blockSize; | |
| 22 int pos; | |
|
273
91a31873c42a
start conversion from trilead to ganymed
Carl Byington <carl@five-ten-sg.com>
parents:
diff
changeset
|
23 |
| 307 | 24 /* |
| 25 * We cannot use java.io.BufferedOutputStream, since that is not available | |
| 26 * in J2ME. Everything could be improved here alot. | |
| 27 */ | |
|
273
91a31873c42a
start conversion from trilead to ganymed
Carl Byington <carl@five-ten-sg.com>
parents:
diff
changeset
|
28 |
| 307 | 29 private static final int BUFF_SIZE = 8192; |
| 30 byte[] out_buffer = new byte[BUFF_SIZE]; | |
| 31 int out_buffer_pos = 0; | |
|
273
91a31873c42a
start conversion from trilead to ganymed
Carl Byington <carl@five-ten-sg.com>
parents:
diff
changeset
|
32 |
| 307 | 33 public CipherOutputStream(BlockCipher tc, OutputStream bo) { |
| 34 this.bo = bo; | |
| 35 changeCipher(tc); | |
| 36 } | |
|
273
91a31873c42a
start conversion from trilead to ganymed
Carl Byington <carl@five-ten-sg.com>
parents:
diff
changeset
|
37 |
| 307 | 38 private void internal_write(byte[] src, int off, int len) throws IOException { |
| 39 while (len > 0) { | |
| 40 int space = BUFF_SIZE - out_buffer_pos; | |
| 41 int copy = (len > space) ? space : len; | |
| 42 System.arraycopy(src, off, out_buffer, out_buffer_pos, copy); | |
| 43 off += copy; | |
| 44 out_buffer_pos += copy; | |
| 45 len -= copy; | |
|
273
91a31873c42a
start conversion from trilead to ganymed
Carl Byington <carl@five-ten-sg.com>
parents:
diff
changeset
|
46 |
| 307 | 47 if (out_buffer_pos >= BUFF_SIZE) { |
| 48 bo.write(out_buffer, 0, BUFF_SIZE); | |
| 49 out_buffer_pos = 0; | |
| 50 } | |
| 51 } | |
| 52 } | |
| 53 | |
| 54 private void internal_write(int b) throws IOException { | |
| 55 out_buffer[out_buffer_pos++] = (byte) b; | |
|
273
91a31873c42a
start conversion from trilead to ganymed
Carl Byington <carl@five-ten-sg.com>
parents:
diff
changeset
|
56 |
| 307 | 57 if (out_buffer_pos >= BUFF_SIZE) { |
| 58 bo.write(out_buffer, 0, BUFF_SIZE); | |
| 59 out_buffer_pos = 0; | |
| 60 } | |
| 61 } | |
|
273
91a31873c42a
start conversion from trilead to ganymed
Carl Byington <carl@five-ten-sg.com>
parents:
diff
changeset
|
62 |
| 307 | 63 public void flush() throws IOException { |
| 64 if (pos != 0) { | |
| 65 throw new IOException("FATAL: cannot flush since crypto buffer is not aligned."); | |
| 66 } | |
|
273
91a31873c42a
start conversion from trilead to ganymed
Carl Byington <carl@five-ten-sg.com>
parents:
diff
changeset
|
67 |
| 307 | 68 if (out_buffer_pos > 0) { |
| 69 bo.write(out_buffer, 0, out_buffer_pos); | |
| 70 out_buffer_pos = 0; | |
| 71 } | |
| 72 | |
| 73 bo.flush(); | |
| 74 } | |
|
273
91a31873c42a
start conversion from trilead to ganymed
Carl Byington <carl@five-ten-sg.com>
parents:
diff
changeset
|
75 |
| 307 | 76 public void changeCipher(BlockCipher bc) { |
| 77 this.currentCipher = bc; | |
| 78 blockSize = bc.getBlockSize(); | |
| 79 buffer = new byte[blockSize]; | |
| 80 enc = new byte[blockSize]; | |
| 81 pos = 0; | |
| 82 } | |
|
273
91a31873c42a
start conversion from trilead to ganymed
Carl Byington <carl@five-ten-sg.com>
parents:
diff
changeset
|
83 |
| 307 | 84 private void writeBlock() throws IOException { |
| 85 try { | |
| 86 currentCipher.transformBlock(buffer, 0, enc, 0); | |
| 87 } | |
| 88 catch (Exception e) { | |
| 89 throw new IOException("Error while decrypting block.", e); | |
| 90 } | |
|
273
91a31873c42a
start conversion from trilead to ganymed
Carl Byington <carl@five-ten-sg.com>
parents:
diff
changeset
|
91 |
| 307 | 92 internal_write(enc, 0, blockSize); |
| 93 pos = 0; | |
| 94 } | |
|
273
91a31873c42a
start conversion from trilead to ganymed
Carl Byington <carl@five-ten-sg.com>
parents:
diff
changeset
|
95 |
| 307 | 96 public void write(byte[] src, int off, int len) throws IOException { |
| 97 while (len > 0) { | |
| 98 int avail = blockSize - pos; | |
| 99 int copy = Math.min(avail, len); | |
| 100 System.arraycopy(src, off, buffer, pos, copy); | |
| 101 pos += copy; | |
| 102 off += copy; | |
| 103 len -= copy; | |
|
273
91a31873c42a
start conversion from trilead to ganymed
Carl Byington <carl@five-ten-sg.com>
parents:
diff
changeset
|
104 |
| 307 | 105 if (pos >= blockSize) { |
| 106 writeBlock(); | |
| 107 } | |
| 108 } | |
| 109 } | |
|
273
91a31873c42a
start conversion from trilead to ganymed
Carl Byington <carl@five-ten-sg.com>
parents:
diff
changeset
|
110 |
| 307 | 111 public void write(int b) throws IOException { |
| 112 buffer[pos++] = (byte) b; | |
| 113 | |
| 114 if (pos >= blockSize) { | |
| 115 writeBlock(); | |
| 116 } | |
| 117 } | |
|
273
91a31873c42a
start conversion from trilead to ganymed
Carl Byington <carl@five-ten-sg.com>
parents:
diff
changeset
|
118 |
| 307 | 119 public void writePlain(int b) throws IOException { |
| 120 if (pos != 0) { | |
| 121 throw new IOException("Cannot write plain since crypto buffer is not aligned."); | |
| 122 } | |
| 123 | |
| 124 internal_write(b); | |
| 125 } | |
|
273
91a31873c42a
start conversion from trilead to ganymed
Carl Byington <carl@five-ten-sg.com>
parents:
diff
changeset
|
126 |
| 307 | 127 public void writePlain(byte[] b, int off, int len) throws IOException { |
| 128 if (pos != 0) { | |
| 129 throw new IOException("Cannot write plain since crypto buffer is not aligned."); | |
| 130 } | |
|
273
91a31873c42a
start conversion from trilead to ganymed
Carl Byington <carl@five-ten-sg.com>
parents:
diff
changeset
|
131 |
| 307 | 132 internal_write(b, off, len); |
| 133 } | |
|
273
91a31873c42a
start conversion from trilead to ganymed
Carl Byington <carl@five-ten-sg.com>
parents:
diff
changeset
|
134 } |