Develop and Download Open Source Software

Tera Term

Browse Subversion Repository

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

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

Parent Directory Parent Directory | Revision Log Revision Log

Revision 6966 - (hide annotations) (download) (as text)
Thu Nov 2 11:37:28 2017 UTC (6 years, 4 months ago) by doda
File MIME type: text/x-csrc
File size: 8478 byte(s) 
コード整理。動作には影響しないはず。

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