Develop and Download Open Source Software

Tera Term

Browse Subversion Repository

/[ttssh2]/trunk/ttssh2/ttxssh/pkt.c

Contents of /trunk/ttssh2/ttxssh/pkt.c

Parent Directory Parent Directory | Revision Log Revision Log

Revision 6771 - (show annotations) (download) (as text)
Mon Jun 5 08:48:48 2017 UTC (6 years, 9 months ago) by doda
File MIME type: text/x-csrc
File size: 9098 byte(s) 
不用な条件判断およびコードを削除
1 /*
2 Copyright (c) 1998-2001, Robert O'Callahan
3 All rights reserved.
4
5 Redistribution and use in source and binary forms, with or without modification,
6 are permitted provided that the following conditions are met:
7
8 Redistributions of source code must retain the above copyright notice, this list of
9 conditions and the following disclaimer.
10
11 Redistributions in binary form must reproduce the above copyright notice, this list
12 of conditions and the following disclaimer in the documentation and/or other materials
13 provided with the distribution.
14
15 The name of Robert O'Callahan may not be used to endorse or promote products derived from
16 this software without specific prior written permission.
17
18 THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ``AS IS'' AND
19 ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
20 OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL
21 THE REGENTS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
22 EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
23 SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24 HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
25 OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
26 SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27 */
28
29 /*
30 This code is copyright (C) 1998-1999 Robert O'Callahan.
31 See LICENSE.TXT for the license.
32 */
33
34 #include "ttxssh.h"
35 #include "util.h"
36 #include "pkt.h"
37
38 //#define READAMOUNT 60000
39 // 60000 -> 65536 ���g���BSSH2����window�������s�������ASSH2��window size��
40 // �������������K�v�������B(2004.10.17 yutaka)
41 //#define READAMOUNT 65536
42 // 65536 -> 131072 ���g���B(2007.10.29 maya)
43 #define READAMOUNT CHAN_SES_WINDOW_DEFAULT
44
45 void PKT_init(PTInstVar pvar)
46 {
47 buf_create(&pvar->pkt_state.buf, &pvar->pkt_state.buflen);
48 pvar->pkt_state.datastart = 0;
49 pvar->pkt_state.datalen = 0;
50 pvar->pkt_state.seen_server_ID = FALSE;
51 pvar->pkt_state.seen_newline = FALSE;
52 pvar->pkt_state.predecrypted_packet = FALSE;
53 }
54
55 /* Read some data, leave no more than up_to_amount bytes in the buffer,
56 return the number of bytes read or -1 on error or blocking. */
57 static int recv_data(PTInstVar pvar, unsigned long up_to_amount)
58 {
59 int amount_read;
60
61 /* Shuffle data to the start of the buffer */
62 if (pvar->pkt_state.datastart != 0) {
63 memmove(pvar->pkt_state.buf,
64 pvar->pkt_state.buf + pvar->pkt_state.datastart,
65 pvar->pkt_state.datalen);
66 pvar->pkt_state.datastart = 0;
67 }
68
69 buf_ensure_size(&pvar->pkt_state.buf, &pvar->pkt_state.buflen,
70 up_to_amount);
71
72 _ASSERT(pvar->pkt_state.buf != NULL);
73
74 amount_read = (pvar->Precv) (pvar->socket,
75 pvar->pkt_state.buf +
76 pvar->pkt_state.datalen,
77 up_to_amount - pvar->pkt_state.datalen,
78 0);
79
80 if (amount_read > 0) {
81 /* Update seen_newline if necessary */
82 if (!pvar->pkt_state.seen_server_ID
83 && !pvar->pkt_state.seen_newline) {
84 int i;
85
86 for (i = 0; i < amount_read; i++) {
87 if (pvar->pkt_state.buf[pvar->pkt_state.datalen + i] ==
88 '\n') {
89 pvar->pkt_state.seen_newline = 1;
90 }
91 }
92 }
93
94 pvar->pkt_state.datalen += amount_read;
95 }
96
97 return amount_read;
98 }
99
100
101 // ���s�R�[�h���o��������������
102 static int recv_line_data(PTInstVar pvar)
103 {
104 int amount_read;
105 char buf[256];
106 size_t up_to_amount = sizeof(buf);
107 int i;
108
109 /* Shuffle data to the start of the buffer */
110 if (pvar->pkt_state.datastart != 0) {
111 memmove(pvar->pkt_state.buf,
112 pvar->pkt_state.buf + pvar->pkt_state.datastart,
113 pvar->pkt_state.datalen);
114 pvar->pkt_state.datastart = 0;
115 }
116
117 buf_ensure_size(&pvar->pkt_state.buf, &pvar->pkt_state.buflen,
118 up_to_amount);
119
120 for (i = 0 ; i < (int)up_to_amount ; i++) {
121 amount_read = (pvar->Precv) (pvar->socket,
122 &buf[i], 1, 0);
123 if (amount_read != 1) {
124 return 0; // error
125 }
126
127 pvar->pkt_state.datalen += amount_read;
128
129 if (buf[i] == '\n') { // 0x0a
130 buf[i+1] = 0;
131 break;
132 }
133 }
134 amount_read = i + 1; // ���������T�C�Y�iLF�������j
135 memcpy(pvar->pkt_state.buf, buf, amount_read);
136
137 pvar->pkt_state.seen_newline = 1;
138
139 return amount_read;
140 }
141
142
143 /* This function does two things:
144 -- reads data from the sshd and feeds the SSH protocol packets to ssh.c
145 -- copies any available decrypted session data into the application buffer
146 */
147 int PKT_recv(PTInstVar pvar, char FAR * buf, int buflen)
148 {
149 int amount_in_buf = 0;
150 BOOL connection_closed = FALSE;
151
152 while (SSH_is_any_payload(pvar) ? buflen > 0 : !connection_closed) {
153 if (SSH_is_any_payload(pvar)) {
154 /* ssh.c has some session data for us to give to Tera Term. */
155 int grabbed = SSH_extract_payload(pvar, buf, buflen);
156
157 amount_in_buf += grabbed;
158 buf += grabbed;
159 buflen -= grabbed;
160
161 } else if (!pvar->pkt_state.seen_server_ID &&
162 (pvar->pkt_state.seen_newline
163 || pvar->pkt_state.datalen >= 255)) {
164 /* We're looking for the initial ID string and either we've seen the
165 terminating newline, or we've exceeded the limit at which we should see
166 a newline. */
167 unsigned int i;
168
169 for (i = 0;
170 pvar->pkt_state.buf[i] != '\n'
171 && i < pvar->pkt_state.datalen; i++) {
172 }
173 if (pvar->pkt_state.buf[i] == '\n') {
174 i++;
175 }
176
177 // SSH�T�[�o���o�[�W�����`�F�b�N���s��
178 if (SSH_handle_server_ID(pvar, pvar->pkt_state.buf, i)) {
179 pvar->pkt_state.seen_server_ID = 1;
180
181 if (SSHv1(pvar)) {
182
183 } else { // for SSH2(yutaka)
184 // send Key Exchange Init
185 SSH2_send_kexinit(pvar);
186 }
187
188 } else {
189 // reset flag to re-read server ID (2008.1.24 yutaka)
190 pvar->pkt_state.seen_newline = 0;
191
192 }
193
194 pvar->pkt_state.datastart += i;
195 pvar->pkt_state.datalen -= i;
196
197 } else if (pvar->pkt_state.seen_server_ID
198 && pvar->pkt_state.datalen >=
199 (unsigned int) SSH_get_min_packet_size(pvar)) {
200 char FAR *data =
201 pvar->pkt_state.buf + pvar->pkt_state.datastart;
202 uint32 padding;
203 uint32 pktsize;
204 uint32 total_packet_size;
205
206 //debug_print(10, data, pvar->pkt_state.datalen);
207
208 // �������p�P�b�g�������������������B
209 if (!pvar->pkt_state.predecrypted_packet) {
210 SSH_predecrpyt_packet(pvar, data);
211 pvar->pkt_state.predecrypted_packet = TRUE;
212 }
213
214 if (SSHv1(pvar)) {
215 uint32 realpktsize = get_uint32_MSBfirst(data);
216
217 padding = 8 - (realpktsize % 8);
218 pktsize = realpktsize + padding;
219 } else {
220 // SSH2���p�P�b�g�������� packet-size(4)+padding(1)+type(1) �������B
221 pktsize = get_uint32_MSBfirst(data);
222 padding = (unsigned char) data[4];
223 }
224
225 // �p�P�b�g(TCP�y�C���[�h)���S�����T�C�Y���ASSH�y�C���[�h�{4�i�{MAC�j�������B
226 // +4���ASSH�y�C���[�h���T�C�Y���i�[�������������iint�^�j�B
227 total_packet_size = pktsize + 4 + SSH_get_clear_MAC_size(pvar);
228
229 if (total_packet_size <= pvar->pkt_state.datalen) {
230 /* the data must be 4 byte aligned. */
231 SSH_handle_packet(pvar, data, pktsize, padding);
232 pvar->pkt_state.predecrypted_packet = FALSE;
233
234 pvar->pkt_state.datastart += total_packet_size;
235 pvar->pkt_state.datalen -= total_packet_size;
236
237 } else if (total_packet_size > PACKET_MAX_SIZE) {
238 // 4MB���������������p�P�b�g�����������A�����I�������B
239 // ���������f�[�^�������������s�����A���F�����������������B
240 UTIL_get_lang_msg("MSG_PKT_OVERSIZED_ERROR", pvar,
241 "Oversized packet received from server; connection will close.");
242 notify_fatal_error(pvar, pvar->ts->UIMsg, TRUE);
243 } else {
244 int amount_read =
245 recv_data(pvar, max(total_packet_size, READAMOUNT));
246
247 if (amount_read == SOCKET_ERROR) {
248 if (amount_in_buf == 0) {
249 return SOCKET_ERROR;
250 } else {
251 return amount_in_buf;
252 }
253 } else {
254 if (amount_read == 0) {
255 connection_closed = TRUE;
256 }
257 }
258 }
259
260
261 } else {
262 // �p�P�b�g�����M�i����60KB�j
263 int amount_read;
264
265 if (pvar->pkt_state.seen_server_ID == 0) {
266 //amount_read = recv_line_data(pvar);
267 amount_read = recv_data(pvar, READAMOUNT);
268
269 } else {
270 amount_read = recv_data(pvar, READAMOUNT);
271
272 }
273
274 if (amount_read == SOCKET_ERROR) {
275 if (amount_in_buf == 0) {
276 return SOCKET_ERROR;
277 } else {
278 return amount_in_buf;
279 }
280 } else if (amount_read == 0) {
281 connection_closed = TRUE;
282 }
283 }
284
285 if (pvar->fatal_error) {
286 return amount_in_buf;
287 }
288 }
289
290 if (SSH_is_any_payload(pvar)) {
291 PostMessage(pvar->NotificationWindow, WM_USER_COMMNOTIFY,
292 pvar->socket, MAKELPARAM(FD_READ, 0));
293 }
294
295 return amount_in_buf;
296 }
297
298 void PKT_end(PTInstVar pvar)
299 {
300 buf_destroy(&pvar->pkt_state.buf, &pvar->pkt_state.buflen);
301 }
Back to OSDN">Back to OSDN
 ViewVC Help
Powered by ViewVC 1.1.26  
About OSDN
Find Software
Develop Software
Help