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 2922 - (show annotations) (download) (as text)
Sun Oct 29 17:15:47 2006 UTC (17 years, 4 months ago) by yutakapon
Original Path: ttssh2/trunk/ttxssh/pkt.c
File MIME type: text/x-csrc
File size: 8623 byte(s) 
DEBUG_PRINT_TO_FILEマクロを追加。
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
37 //#define READAMOUNT 60000
38 // 60000 -> 65536 ���g���BSSH2����window�������s�������ASSH2��window size��
39 // �������������K�v�������B(2004.10.17 yutaka)
40 #define READAMOUNT 65536
41
42 void PKT_init(PTInstVar pvar)
43 {
44 buf_create(&pvar->pkt_state.buf, &pvar->pkt_state.buflen);
45 pvar->pkt_state.datastart = 0;
46 pvar->pkt_state.datalen = 0;
47 pvar->pkt_state.seen_server_ID = FALSE;
48 pvar->pkt_state.seen_newline = FALSE;
49 pvar->pkt_state.predecrypted_packet = FALSE;
50 }
51
52 /* Read some data, leave no more than up_to_amount bytes in the buffer,
53 return the number of bytes read or -1 on error or blocking. */
54 static int recv_data(PTInstVar pvar, unsigned long up_to_amount)
55 {
56 int amount_read;
57
58 /* Shuffle data to the start of the buffer */
59 if (pvar->pkt_state.datastart != 0) {
60 memmove(pvar->pkt_state.buf,
61 pvar->pkt_state.buf + pvar->pkt_state.datastart,
62 pvar->pkt_state.datalen);
63 pvar->pkt_state.datastart = 0;
64 }
65
66 buf_ensure_size(&pvar->pkt_state.buf, &pvar->pkt_state.buflen,
67 up_to_amount);
68
69 _ASSERT(pvar->pkt_state.buf != NULL);
70
71 amount_read = (pvar->Precv) (pvar->socket,
72 pvar->pkt_state.buf +
73 pvar->pkt_state.datalen,
74 up_to_amount - pvar->pkt_state.datalen,
75 0);
76
77 if (amount_read > 0) {
78 /* Update seen_newline if necessary */
79 if (!pvar->pkt_state.seen_server_ID
80 && !pvar->pkt_state.seen_newline) {
81 int i;
82
83 for (i = 0; i < amount_read; i++) {
84 if (pvar->pkt_state.buf[pvar->pkt_state.datalen + i] ==
85 '\n') {
86 pvar->pkt_state.seen_newline = 1;
87 }
88 }
89 }
90
91 pvar->pkt_state.datalen += amount_read;
92 }
93
94 return amount_read;
95 }
96
97
98 // ���s�R�[�h���o��������������
99 static int recv_line_data(PTInstVar pvar)
100 {
101 int amount_read;
102 char buf[256];
103 size_t up_to_amount = sizeof(buf);
104 int i;
105
106 /* Shuffle data to the start of the buffer */
107 if (pvar->pkt_state.datastart != 0) {
108 memmove(pvar->pkt_state.buf,
109 pvar->pkt_state.buf + pvar->pkt_state.datastart,
110 pvar->pkt_state.datalen);
111 pvar->pkt_state.datastart = 0;
112 }
113
114 buf_ensure_size(&pvar->pkt_state.buf, &pvar->pkt_state.buflen,
115 up_to_amount);
116
117 for (i = 0 ; i < (int)up_to_amount ; i++) {
118 amount_read = (pvar->Precv) (pvar->socket,
119 &buf[i],
120 1,
121 0);
122 if (amount_read != 1) {
123 return 0; // error
124 }
125
126 pvar->pkt_state.datalen += amount_read;
127
128 if (buf[i] == '\n') { // 0x0a
129 buf[i+1] = 0;
130 break;
131 }
132 }
133 amount_read = i + 1; // ���������T�C�Y�iLF�������j
134 memcpy(pvar->pkt_state.buf, buf, amount_read);
135
136 pvar->pkt_state.seen_newline = 1;
137
138 return amount_read;
139 }
140
141
142 /* This function does two things:
143 -- reads data from the sshd and feeds the SSH protocol packets to ssh.c
144 -- copies any available decrypted session data into the application buffer
145 */
146 int PKT_recv(PTInstVar pvar, char FAR * buf, int buflen)
147 {
148 int amount_in_buf = 0;
149 BOOL connection_closed = FALSE;
150
151 while (SSH_is_any_payload(pvar) ? buflen > 0 : !connection_closed) {
152 if (SSH_is_any_payload(pvar)) {
153 /* ssh.c has some session data for us to give to Teraterm. */
154 int grabbed = SSH_extract_payload(pvar, buf, buflen);
155
156 amount_in_buf += grabbed;
157 buf += grabbed;
158 buflen -= grabbed;
159
160 } else if (!pvar->pkt_state.seen_server_ID &&
161 (pvar->pkt_state.seen_newline
162 || pvar->pkt_state.datalen >= 255)) {
163 /* We're looking for the initial ID string and either we've seen the
164 terminating newline, or we've exceeded the limit at which we should see
165 a newline. */
166 unsigned int i;
167
168 for (i = 0;
169 pvar->pkt_state.buf[i] != '\n'
170 && i < pvar->pkt_state.datalen; i++) {
171 }
172 if (pvar->pkt_state.buf[i] == '\n') {
173 i++;
174 }
175
176 // SSH�T�[�o���o�[�W�����`�F�b�N���s��
177 if (SSH_handle_server_ID(pvar, pvar->pkt_state.buf, i)) {
178 pvar->pkt_state.seen_server_ID = 1;
179
180 if (SSHv1(pvar)) {
181
182 } else { // for SSH2(yutaka)
183 // send Key Exchange Init
184 SSH2_send_kexinit(pvar);
185 }
186
187 }
188
189 pvar->pkt_state.datastart += i;
190 pvar->pkt_state.datalen -= i;
191
192 } else if (pvar->pkt_state.seen_server_ID
193 && pvar->pkt_state.datalen >=
194 (unsigned int) SSH_get_min_packet_size(pvar)) {
195 char FAR *data =
196 pvar->pkt_state.buf + pvar->pkt_state.datastart;
197 uint32 padding;
198 uint32 pktsize;
199 uint32 total_packet_size;
200
201 //debug_print(10, data, pvar->pkt_state.datalen);
202
203 // SSH2�����������p�P�b�g�������������������B
204 if (!pvar->pkt_state.predecrypted_packet) {
205 //DEBUG_PRINT_TO_FILE(0, data, pvar->pkt_state.datalen);
206 SSH_predecrpyt_packet(pvar, data);
207
208 if (SSHv1(pvar)) {
209 pvar->pkt_state.predecrypted_packet = TRUE;
210 } else { // for SSH2(yutaka)
211 // do nothing
212 pvar->pkt_state.predecrypted_packet = TRUE;
213 }
214 }
215
216 if (SSHv1(pvar)) {
217 uint32 realpktsize = get_uint32_MSBfirst(data);
218
219 padding = 8 - (realpktsize % 8);
220 pktsize = realpktsize + padding;
221 } else {
222 // SSH2���p�P�b�g�������� packet-size(4)+padding(1)+type(1) �������B
223 pktsize = get_uint32_MSBfirst(data);
224 padding = (unsigned char) data[4];
225 }
226
227 // �p�P�b�g(TCP�y�C���[�h)���S�����T�C�Y���ASSH�y�C���[�h�{4�i�{MAC�j�������B
228 // +4���ASSH�y�C���[�h���T�C�Y���i�[�������������iint�^�j�B
229 total_packet_size = pktsize + 4 + SSH_get_clear_MAC_size(pvar);
230
231 if (total_packet_size <= pvar->pkt_state.datalen) {
232 /* the data must be 4 byte aligned. */
233 SSH_handle_packet(pvar, data, pktsize, padding);
234 pvar->pkt_state.predecrypted_packet = FALSE;
235
236 pvar->pkt_state.datastart += total_packet_size;
237 pvar->pkt_state.datalen -= total_packet_size;
238
239 } else if (total_packet_size > 4 * 1024 * 1024) {
240 // 4MB���������������p�P�b�g�����������A�����I�������B
241 // ���������f�[�^�������������s�����A���F�����������������B
242 notify_fatal_error(pvar,
243 "Oversized packet received from server; connection will close.");
244 } else {
245 int amount_read =
246 recv_data(pvar, max(total_packet_size, READAMOUNT));
247
248 if (amount_read == SOCKET_ERROR) {
249 if (amount_in_buf == 0) {
250 return SOCKET_ERROR;
251 } else {
252 return amount_in_buf;
253 }
254 } else {
255 if (amount_read == 0) {
256 connection_closed = TRUE;
257 }
258 }
259 }
260
261
262 } else {
263 // �p�P�b�g�����M�i����60KB�j
264 int amount_read;
265
266 if (pvar->pkt_state.seen_server_ID == 0) {
267 //amount_read = recv_line_data(pvar);
268 amount_read = recv_data(pvar, READAMOUNT);
269
270 } else {
271 amount_read = recv_data(pvar, READAMOUNT);
272
273 }
274
275 if (amount_read == SOCKET_ERROR) {
276 if (amount_in_buf == 0) {
277 return SOCKET_ERROR;
278 } else {
279 return amount_in_buf;
280 }
281 } else if (amount_read == 0) {
282 connection_closed = TRUE;
283 }
284 }
285
286 if (pvar->fatal_error) {
287 return amount_in_buf;
288 }
289 }
290
291 if (SSH_is_any_payload(pvar)) {
292 PostMessage(pvar->NotificationWindow, WM_USER_COMMNOTIFY,
293 pvar->socket, MAKELPARAM(FD_READ, 0));
294 }
295
296 return amount_in_buf;
297 }
298
299 void PKT_end(PTInstVar pvar)
300 {
301 buf_destroy(&pvar->pkt_state.buf, &pvar->pkt_state.buflen);
302 }
303
304 /*
305 * $Log: not supported by cvs2svn $
306 * Revision 1.2 2004/12/19 15:39:58 yutakakn
307 * CVS LogID������
308 *
309 */
Back to OSDN">Back to OSDN
 ViewVC Help
Powered by ViewVC 1.1.26  
About OSDN
Find Software
Develop Software
Help