Mercurial > 510Connectbot

comparison src/ch/ethz/ssh2/SCPInputStream.java @ 273:91a31873c42a ganymed

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
start conversion from trilead to ganymed
author Carl Byington <carl@five-ten-sg.com>
date Fri, 18 Jul 2014 11:21:46 -0700
parents
children 071eccdff8ea
comparison
equal deleted inserted replaced
272:ce2f4e397703 273:91a31873c42a
1 /*
2 * Copyright (c) 2011 David Kocher. All rights reserved.
3 * Please refer to the LICENSE.txt for licensing details.
4 */
5 package ch.ethz.ssh2;
6
7 import java.io.BufferedInputStream;
8 import java.io.BufferedOutputStream;
9 import java.io.IOException;
10 import java.io.OutputStream;
11
12 /**
13 * @version $Id: SCPInputStream.java 151 2014-04-28 10:03:39Z dkocher@sudo.ch $
14 */
15 public class SCPInputStream extends BufferedInputStream
16 {
17 private Session session;
18
19 /**
20 * Bytes remaining to be read from the stream
21 */
22 private long remaining;
23
24 public SCPInputStream(SCPClient client, Session session) throws IOException
25 {
26 super(session.getStdout());
27
28 this.session = session;
29
30 OutputStream os = new BufferedOutputStream(session.getStdin(), 512);
31
32 os.write(0x0);
33 os.flush();
34
35 final SCPClient.LenNamePair lnp;
36
37 while (true)
38 {
39 int c = session.getStdout().read();
40 if (c < 0)
41 {
42 throw new IOException("Remote scp terminated unexpectedly.");
43 }
44
45 String line = client.receiveLine(session.getStdout());
46
47 if (c == 'T')
48 {
49 /* Ignore modification times */
50 continue;
51 }
52
53 if ((c == 1) || (c == 2))
54 {
55 throw new IOException("Remote SCP error: " + line);
56 }
57
58 if (c == 'C')
59 {
60 lnp = client.parseCLine(line);
61 break;
62
63 }
64 throw new IOException("Remote SCP error: " + ((char) c) + line);
65 }
66
67 os.write(0x0);
68 os.flush();
69
70 this.remaining = lnp.length;
71 }
72
73 @Override
74 public int read() throws IOException
75 {
76 if (!(remaining > 0))
77 {
78 return -1;
79 }
80
81 int b = super.read();
82 if (b < 0)
83 {
84 throw new IOException("Remote scp terminated connection unexpectedly");
85 }
86
87 remaining -= 1;
88
89 return b;
90 }
91
92 @Override
93 public int read(byte b[], int off, int len) throws IOException
94 {
95 if (!(remaining > 0))
96 {
97 return -1;
98 }
99
100 int trans = (int) remaining;
101 if (remaining > len)
102 {
103 trans = len;
104 }
105
106 int read = super.read(b, off, trans);
107 if (read < 0)
108 {
109 throw new IOException("Remote scp terminated connection unexpectedly");
110 }
111
112 remaining -= read;
113
114 return read;
115 }
116
117 @Override
118 public void close() throws IOException
119 {
120 try
121 {
122 session.getStdin().write(0x0);
123 session.getStdin().flush();
124 }
125 finally
126 {
127 if (session != null)
128 {
129 session.close();
130 }
131 }
132 }
133 }