Develop and Download Open Source Software

Tera Term

Browse Subversion Repository

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

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

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 3221 by maya, Tue Mar 24 09:37:20 2009 UTC revision 6969 by doda, Thu Nov 2 11:37:41 2017 UTC
# Line 1  Line 1 
1  /*  /*
2  Copyright (c) 1998-2001, Robert O'Callahan   * Copyright (c) 1998-2001, Robert O'Callahan
3  All rights reserved.   * (C) 2004-2017 TeraTerm Project
4     * All rights reserved.
5  Redistribution and use in source and binary forms, with or without modification,   *
6  are permitted provided that the following conditions are met:   * Redistribution and use in source and binary forms, with or without
7     * modification, are permitted provided that the following conditions
8  Redistributions of source code must retain the above copyright notice, this list of   * are met:
9  conditions and the following disclaimer.   *
10     * 1. Redistributions of source code must retain the above copyright
11  Redistributions in binary form must reproduce the above copyright notice, this list   *    notice, this list of conditions and the following disclaimer.
12  of conditions and the following disclaimer in the documentation and/or other materials   * 2. Redistributions in binary form must reproduce the above copyright
13  provided with the distribution.   *    notice, this list of conditions and the following disclaimer in the
14     *    documentation and/or other materials provided with the distribution.
15  The name of Robert O'Callahan may not be used to endorse or promote products derived from   * 3. The name of the author may not be used to endorse or promote products
16  this software without specific prior written permission.   *    derived from this software without specific prior written permission.
17     *
18  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ``AS IS'' AND   * THIS SOFTWARE IS PROVIDED BY THE AUTHORS ``AS IS'' AND ANY EXPRESS OR
19  ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES   * 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   * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
21  THE REGENTS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,   * IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY DIRECT, INDIRECT,
22  EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF   * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
23  SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)   * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
24  HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,   * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
25  OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS   * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26  SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.   * (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  /*  
30  This code is copyright (C) 1998-1999 Robert O'Callahan.  /*
31  See LICENSE.TXT for the license.  This code is copyright (C) 1998-1999 Robert O'Callahan.
32  */  See LICENSE.TXT for the license.
33    */
34  #include "ttxssh.h"  
35  #include "util.h"  #include "ttxssh.h"
36  #include "pkt.h"  #include "util.h"
37    #include "pkt.h"
38  //#define READAMOUNT 60000  
39  // 60000 -> 65536 へ拡張。SSH2ではwindow制御を行うため、SSH2のwindow sizeと  #define READAMOUNT CHAN_SES_WINDOW_DEFAULT
40  // 合わせておく必要がある。(2004.10.17 yutaka)  
41  //#define READAMOUNT 65536  void PKT_init(PTInstVar pvar)
42  // 65536 -> 131072 へ拡張。(2007.10.29 maya)  {
43  #define READAMOUNT CHAN_SES_WINDOW_DEFAULT          buf_create(&pvar->pkt_state.buf, &pvar->pkt_state.buflen);
44            pvar->pkt_state.datastart = 0;
45  void PKT_init(PTInstVar pvar)          pvar->pkt_state.datalen = 0;
46  {          pvar->pkt_state.seen_server_ID = FALSE;
47          buf_create(&pvar->pkt_state.buf, &pvar->pkt_state.buflen);          pvar->pkt_state.seen_newline = FALSE;
48          pvar->pkt_state.datastart = 0;          pvar->pkt_state.predecrypted_packet = FALSE;
49          pvar->pkt_state.datalen = 0;  }
50          pvar->pkt_state.seen_server_ID = FALSE;  
51          pvar->pkt_state.seen_newline = FALSE;  /* Read some data, leave no more than up_to_amount bytes in the buffer,
52          pvar->pkt_state.predecrypted_packet = FALSE;     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  /* Read some data, leave no more than up_to_amount bytes in the buffer,          int amount_read;
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)          /* Shuffle data to the start of the buffer */
58  {          if (pvar->pkt_state.datastart != 0) {
59          int amount_read;                  memmove(pvar->pkt_state.buf,
60                            pvar->pkt_state.buf + pvar->pkt_state.datastart,
61          /* Shuffle data to the start of the buffer */                          pvar->pkt_state.datalen);
62          if (pvar->pkt_state.datastart != 0) {                  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);          buf_ensure_size(&pvar->pkt_state.buf, &pvar->pkt_state.buflen, up_to_amount);
66                  pvar->pkt_state.datastart = 0;  
67          }          _ASSERT(pvar->pkt_state.buf != NULL);
68    
69          buf_ensure_size(&pvar->pkt_state.buf, &pvar->pkt_state.buflen,          amount_read = (pvar->Precv) (pvar->socket,
70                          up_to_amount);                                       pvar->pkt_state.buf + pvar->pkt_state.datalen,
71                                         up_to_amount - pvar->pkt_state.datalen,
72          _ASSERT(pvar->pkt_state.buf != NULL);                                       0);
73    
74          amount_read = (pvar->Precv) (pvar->socket,          if (amount_read > 0) {
75                                       pvar->pkt_state.buf +                  /* Update seen_newline if necessary */
76                                       pvar->pkt_state.datalen,                  if (!pvar->pkt_state.seen_server_ID && !pvar->pkt_state.seen_newline) {
77                                       up_to_amount - pvar->pkt_state.datalen,                          int i;
78                                       0);  
79                            for (i = 0; i < amount_read; i++) {
80          if (amount_read > 0) {                                  if (pvar->pkt_state.buf[pvar->pkt_state.datalen + i] == '\n') {
81                  /* Update seen_newline if necessary */                                          pvar->pkt_state.seen_newline = 1;
82                  if (!pvar->pkt_state.seen_server_ID                                  }
83                   && !pvar->pkt_state.seen_newline) {                          }
84                          int i;                  }
85                    pvar->pkt_state.datalen += amount_read;
86                          for (i = 0; i < amount_read; i++) {          }
87                                  if (pvar->pkt_state.buf[pvar->pkt_state.datalen + i] ==  
88                                      '\n') {          return amount_read;
89                                          pvar->pkt_state.seen_newline = 1;  }
90                                  }  
91                          }  
92                  }  // 改行コードが出てくるまで読む
93    static int recv_line_data(PTInstVar pvar)
94                  pvar->pkt_state.datalen += amount_read;  {
95          }          int amount_read;
96            char buf[256];
97          return amount_read;          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  static int recv_line_data(PTInstVar pvar)                  memmove(pvar->pkt_state.buf,
103  {                          pvar->pkt_state.buf + pvar->pkt_state.datastart,
104          int amount_read;                          pvar->pkt_state.datalen);
105          char buf[256];                  pvar->pkt_state.datastart = 0;
106          size_t up_to_amount = sizeof(buf);          }
107          int i;  
108            buf_ensure_size(&pvar->pkt_state.buf, &pvar->pkt_state.buflen, up_to_amount);
109          /* Shuffle data to the start of the buffer */  
110          if (pvar->pkt_state.datastart != 0) {          for (i = 0 ; i < (int)up_to_amount ; i++) {
111                  memmove(pvar->pkt_state.buf,                  amount_read = (pvar->Precv) (pvar->socket, &buf[i], 1, 0);
112                          pvar->pkt_state.buf + pvar->pkt_state.datastart,                  if (amount_read != 1) {
113                          pvar->pkt_state.datalen);                          return 0; // error
114                  pvar->pkt_state.datastart = 0;                  }
115          }  
116                    pvar->pkt_state.datalen += amount_read;
117          buf_ensure_size(&pvar->pkt_state.buf, &pvar->pkt_state.buflen,  
118                          up_to_amount);                  if (buf[i] == '\n') { // 0x0a
119                            buf[i+1] = 0;
120          for (i = 0 ; i < (int)up_to_amount ; i++) {                          break;
121                  amount_read = (pvar->Precv) (pvar->socket,                  }
122                                               &buf[i], 1, 0);          }
123                  if (amount_read != 1) {          amount_read = i + 1; // 読み込みサイズ（LFも含む）
124                          return 0; // error          memcpy(pvar->pkt_state.buf, buf, amount_read);
125                  }  
126            pvar->pkt_state.seen_newline = 1;
127                  pvar->pkt_state.datalen += amount_read;  
128            return amount_read;
129                  if (buf[i] == '\n') { // 0x0a  }
130                          buf[i+1] = 0;  
131                          break;  
132                  }  /* This function does two things:
133          }     -- reads data from the sshd and feeds the SSH protocol packets to ssh.c
134          amount_read = i + 1; // 読み込みサイズ（LFも含む）     -- copies any available decrypted session data into the application buffer
135          memcpy(pvar->pkt_state.buf, buf, amount_read);  */
136    int PKT_recv(PTInstVar pvar, char *buf, int buflen)
137          pvar->pkt_state.seen_newline = 1;  {
138            int amount_in_buf = 0;
139          return amount_read;          BOOL connection_closed = FALSE;
140  }  
141            while (SSH_is_any_payload(pvar) ? buflen > 0 : !connection_closed) {
142                    if (SSH_is_any_payload(pvar)) {
143  /* This function does two things:                          /* ssh.c has some session data for us to give to Tera Term. */
144     -- reads data from the sshd and feeds the SSH protocol packets to ssh.c                          int grabbed = SSH_extract_payload(pvar, buf, buflen);
145     -- copies any available decrypted session data into the application buffer  
146  */                          amount_in_buf += grabbed;
147  int PKT_recv(PTInstVar pvar, char FAR * buf, int buflen)                          buf += grabbed;
148  {                          buflen -= grabbed;
149          int amount_in_buf = 0;                  }
150          BOOL connection_closed = FALSE;                  else if (!pvar->pkt_state.seen_server_ID && (pvar->pkt_state.seen_newline || pvar->pkt_state.datalen >= 255)) {
151                            /*
152          while (SSH_is_any_payload(pvar) ? buflen > 0 : !connection_closed) {                           * We're looking for the initial ID string and either we've seen the
153                  if (SSH_is_any_payload(pvar)) {                           * terminating newline, or we've exceeded the limit at which we should see a newline.
154                          /* ssh.c has some session data for us to give to Tera Term. */                           */
155                          int grabbed = SSH_extract_payload(pvar, buf, buflen);                          unsigned int i;
156    
157                          amount_in_buf += grabbed;                          for (i = 0; pvar->pkt_state.buf[i] != '\n' && i < pvar->pkt_state.datalen; i++) {
158                          buf += grabbed;                          }
159                          buflen -= grabbed;                          if (pvar->pkt_state.buf[i] == '\n') {
160                                    i++;
161                  } else if (!pvar->pkt_state.seen_server_ID &&                          }
162                             (pvar->pkt_state.seen_newline  
163                              || pvar->pkt_state.datalen >= 255)) {                          // SSHサーバのバージョンチェックを行う
164                          /* We're looking for the initial ID string and either we've seen the                          if (SSH_handle_server_ID(pvar, pvar->pkt_state.buf, i)) {
165                             terminating newline, or we've exceeded the limit at which we should see                                  pvar->pkt_state.seen_server_ID = 1;
166                             a newline. */  
167                          unsigned int i;                                  if (SSHv2(pvar)) {
168                                            // send Key Exchange Init
169                          for (i = 0;                                          SSH2_send_kexinit(pvar);
170                               pvar->pkt_state.buf[i] != '\n'                                  }
171                               && i < pvar->pkt_state.datalen; i++) {                          } else {
172                          }                                  // reset flag to re-read server ID (2008.1.24 yutaka)
173                          if (pvar->pkt_state.buf[i] == '\n') {                                  pvar->pkt_state.seen_newline = 0;
174                                  i++;                          }
175                          }  
176                            pvar->pkt_state.datastart += i;
177                          // SSHサーバのバージョンチェックを行う                          pvar->pkt_state.datalen -= i;
178                          if (SSH_handle_server_ID(pvar, pvar->pkt_state.buf, i)) {                  }
179                                  pvar->pkt_state.seen_server_ID = 1;                  else if (pvar->pkt_state.seen_server_ID && pvar->pkt_state.datalen >= (unsigned int) SSH_get_min_packet_size(pvar)) {
180                            char *data = pvar->pkt_state.buf + pvar->pkt_state.datastart;
181                                  if (SSHv1(pvar)) {                          uint32 padding;
182                            uint32 pktsize;
183                                  } else { // for SSH2(yutaka)                          uint32 total_packet_size;
184                                          // send Key Exchange Init                          struct Mac *mac = &pvar->ssh2_keys[MODE_IN].mac;
185                                          SSH2_send_kexinit(pvar);                          int etm;
186                                  }  
187                            etm = mac && mac->enabled && mac->etm;
188                          } else {  
189                                  // reset flag to re-read server ID (2008.1.24 yutaka)                          // 暗号化パケットの一部を復号化する。
190                                  pvar->pkt_state.seen_newline = 0;                          if (!pvar->pkt_state.predecrypted_packet && !etm) {
191                                    SSH_predecrpyt_packet(pvar, data);
192                          }                                  pvar->pkt_state.predecrypted_packet = TRUE;
193                            }
194                          pvar->pkt_state.datastart += i;  
195                          pvar->pkt_state.datalen -= i;                          if (SSHv1(pvar)) {
196                                    uint32 realpktsize = get_uint32_MSBfirst(data);
197                  } else if (pvar->pkt_state.seen_server_ID  
198                             && pvar->pkt_state.datalen >=                                  padding = 8 - (realpktsize % 8);
199                             (unsigned int) SSH_get_min_packet_size(pvar)) {                                  pktsize = realpktsize + padding;
200                          char FAR *data =                          } else {
201                                  pvar->pkt_state.buf + pvar->pkt_state.datastart;                                  // SSH2のパケットは先頭に packet-size(4)+padding(1)+type(1) が続く。
202                          uint32 padding;                                  pktsize = get_uint32_MSBfirst(data);
203                          uint32 pktsize;                                  if (etm) {
204                          uint32 total_packet_size;                                          padding = 0;
205                                    }
206                          //debug_print(10, data, pvar->pkt_state.datalen);                                  else {
207                                            padding = (unsigned char) data[4];
208                          // SSH2なら暗号化パケットの一部を復号化する。                                  }
209                          if (!pvar->pkt_state.predecrypted_packet) {                          }
210                                  //DEBUG_PRINT_TO_FILE(0, data, pvar->pkt_state.datalen);  
211                                  SSH_predecrpyt_packet(pvar, data);                          // パケット(TCPペイロード)の全体のサイズは、SSHペイロード＋4（＋MAC）となる。
212                            // +4は、SSHペイロードのサイズを格納している部分（int型）。
213                                  if (SSHv1(pvar)) {                          total_packet_size = pktsize + 4 + SSH_get_clear_MAC_size(pvar);
214                                          pvar->pkt_state.predecrypted_packet = TRUE;  
215                                  } else { // for SSH2(yutaka)                          if (total_packet_size <= pvar->pkt_state.datalen) {
216                                          // do nothing                                  /* the data must be 4 byte aligned. */
217                                          pvar->pkt_state.predecrypted_packet = TRUE;                                  if (SSHv1(pvar)) {
218                                  }                                          SSH_handle_packet1(pvar, data, pktsize, padding);
219                          }                                  }
220                                    else {
221                          if (SSHv1(pvar)) {                                          SSH_handle_packet2(pvar, data, pktsize, padding, etm);
222                                  uint32 realpktsize = get_uint32_MSBfirst(data);                                  }
223                                    pvar->pkt_state.predecrypted_packet = FALSE;
224                                  padding = 8 - (realpktsize % 8);  
225                                  pktsize = realpktsize + padding;                                  pvar->pkt_state.datastart += total_packet_size;
226                          } else {                                  pvar->pkt_state.datalen -= total_packet_size;
227                                  // SSH2のパケットは先頭に packet-size(4)+padding(1)+type(1) が続く。  
228                                  pktsize = get_uint32_MSBfirst(data);                          } else if (total_packet_size > PACKET_MAX_SIZE) {
229                                  padding = (unsigned char) data[4];                                  // 4MBを超える巨大なパケットが届いたら、異常終了する。
230                          }                                  // 実際にはデータ化けで復号失敗時に、誤認識することが多い。
231                                    UTIL_get_lang_msg("MSG_PKT_OVERSIZED_ERROR", pvar,
232                          // パケット(TCPペイロード)の全体のサイズは、SSHペイロード＋4（＋MAC）となる。                                                    "Oversized packet received from server; connection will close.");
233                          // +4は、SSHペイロードのサイズを格納している部分（int型）。                                  notify_fatal_error(pvar, pvar->ts->UIMsg, TRUE);
234                          total_packet_size = pktsize + 4 + SSH_get_clear_MAC_size(pvar);                          } else {
235                                    int amount_read = recv_data(pvar, max(total_packet_size, READAMOUNT));
236                          if (total_packet_size <= pvar->pkt_state.datalen) {  
237                                  /* the data must be 4 byte aligned. */                                  if (amount_read == SOCKET_ERROR) {
238                                  SSH_handle_packet(pvar, data, pktsize, padding);                                          if (amount_in_buf == 0) {
239                                  pvar->pkt_state.predecrypted_packet = FALSE;                                                  return SOCKET_ERROR;
240                                            } else {
241                                  pvar->pkt_state.datastart += total_packet_size;                                                  return amount_in_buf;
242                                  pvar->pkt_state.datalen -= total_packet_size;                                          }
243                                    } else {
244                          } else if (total_packet_size > PACKET_MAX_SIZE) {                                          if (amount_read == 0) {
245                                  // 4MBを超える巨大なパケットが届いたら、異常終了する。                                                  connection_closed = TRUE;
246                                  // 実際にはデータ化けで復号失敗時に、誤認識することが多い。                                          }
247                                  UTIL_get_lang_msg("MSG_PKT_OVERSIZED_ERROR", pvar,                                  }
248                                                                    "Oversized packet received from server; connection will close.");                          }
249                                  notify_fatal_error(pvar, pvar->ts->UIMsg);                  } else {
250                          } else {                          // パケットの受信
251                                  int amount_read =                          int amount_read;
252                                          recv_data(pvar, max(total_packet_size, READAMOUNT));  
253                            amount_read = recv_data(pvar, READAMOUNT);
254                                  if (amount_read == SOCKET_ERROR) {  
255                                          if (amount_in_buf == 0) {                          if (amount_read == SOCKET_ERROR) {
256                                                  return SOCKET_ERROR;                                  if (amount_in_buf == 0) {
257                                          } else {                                          return SOCKET_ERROR;
258                                                  return amount_in_buf;                                  } else {
259                                          }                                          return amount_in_buf;
260                                  } else {                                  }
261                                          if (amount_read == 0) {                          } else if (amount_read == 0) {
262                                                  connection_closed = TRUE;                                  connection_closed = TRUE;
263                                          }                          }
264                                  }                  }
265                          }  
266                    if (pvar->fatal_error) {
267                            return amount_in_buf;
268                  } else {                  }
269                          // パケットの受信（最大60KB）          }
270                          int amount_read;  
271            if (SSH_is_any_payload(pvar)) {
272                          if (pvar->pkt_state.seen_server_ID == 0) {                  PostMessage(pvar->NotificationWindow, WM_USER_COMMNOTIFY, pvar->socket, MAKELPARAM(FD_READ, 0));
273                                  //amount_read = recv_line_data(pvar);          }
274                                  amount_read = recv_data(pvar, READAMOUNT);  
275            return amount_in_buf;
276                          } else {  }
277                                  amount_read = recv_data(pvar, READAMOUNT);  
278    void PKT_end(PTInstVar pvar)
279                          }  {
280            buf_destroy(&pvar->pkt_state.buf, &pvar->pkt_state.buflen);
281                          if (amount_read == SOCKET_ERROR) {  }
                                 if (amount_in_buf == 0) {  
                                         return SOCKET_ERROR;  
                                 } else {  
                                         return amount_in_buf;  
                                 }  
                         } else if (amount_read == 0) {  
                                 connection_closed = TRUE;  
                         }  
                 }  
   
                 if (pvar->fatal_error) {  
                         return amount_in_buf;  
                 }  
         }  
   
         if (SSH_is_any_payload(pvar)) {  
                 PostMessage(pvar->NotificationWindow, WM_USER_COMMNOTIFY,  
                             pvar->socket, MAKELPARAM(FD_READ, 0));  
         }  
   
         return amount_in_buf;  
 }  
   
 void PKT_end(PTInstVar pvar)  
 {  
         buf_destroy(&pvar->pkt_state.buf, &pvar->pkt_state.buflen);  
 }  
Legend:
Removed from v.3221  
changed lines
  Added in v.6969
Back to OSDN">Back to OSDN
 ViewVC Help
Powered by ViewVC 1.1.26  
About OSDN
Find Software
Develop Software
Help