channel.c 47 KB
Newer Older
Niels Möller's avatar
Niels Möller committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
/* channel.c
 *
 * $Id$
 */

/* lsh, an implementation of the ssh protocol
 *
 * Copyright (C) 1998 Niels Mller
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as
 * published by the Free Software Foundation; either version 2 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
J.H.M. Dassen's avatar
J.H.M. Dassen committed
22
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
Niels Möller's avatar
Niels Möller committed
23
24
25
26
27
 */

#include "channel.h"

#include "format.h"
28
#include "io.h"
Niels Möller's avatar
Niels Möller committed
29
#include "read_data.h"
Niels Möller's avatar
Niels Möller committed
30
31
32
33
34
#include "ssh.h"
#include "werror.h"
#include "xalloc.h"

#include <assert.h>
35
#include <string.h>
Niels Möller's avatar
Niels Möller committed
36

37
#define GABA_DEFINE
38
#include "channel.h.x"
39
#undef GABA_DEFINE
40

41
42
#include "channel.c.x"

Niels Möller's avatar
Niels Möller committed
43
44
45
struct exception *make_channel_open_exception(UINT32 error_code, char *msg)
{
  NEW(channel_open_exception, self);
Niels Möller's avatar
Niels Möller committed
46
47
48
49
50
51
52
53
54
55
56
57
58
59

#define MAX_ERROR 4
  static const char *msgs[MAX_ERROR + 1] = {
    "",
    "Administratively prohibited",
    "Connect failed",
    "Unknown channel type",
    "Resource shortage"
  };

  assert(error_code > 0);
  assert(error_code <= MAX_ERROR);
#undef MAX_ERROR
  
Niels Möller's avatar
Niels Möller committed
60
  self->super.type = EXC_CHANNEL_OPEN;
Niels Möller's avatar
Niels Möller committed
61
  self->super.msg = msg ? msg : msgs[error_code];
Niels Möller's avatar
Niels Möller committed
62
63
64
65
66
  self->error_code = error_code;

  return &self->super;
}

67
/* GABA:
68
69
   (class
     (name connection_service)
70
     (super command)
71
72
73
74
     (vars
       ; Supported global requests 
       (global_requests object alist)

75
       (channel_types object alist) ))
76
77
78
79

       ; Initialize connection (for instance, request channels to be 
       ; opened or services to be forwarded).

80
       ; (start object connection_startup)))
81
82
*/

83
/* ;; GABA:
84
85
   (class
     (name global_request_handler)
86
     (super packet_handler)
87
88
89
90
     (vars
       (global_requests object alist)))
*/

91
/* ;; GABA:
92
93
   (class
     (name channel_open_handler)
94
     (super packet_handler)
95
96
97
98
     (vars
       (channel_types object alist)))
*/

Niels Möller's avatar
Niels Möller committed
99
/* ;; GABA:
100
101
102
103
104
105
106
107
108
   (class
     (name channel_open_response)
     (super channel_open_callback)
     (vars
       (remote_channel_number simple UINT32)
       (window_size simple UINT32)
       (max_packet simple UINT32)))
*/

Niels Möller's avatar
Niels Möller committed
109
110
111
112
113
struct lsh_string *format_global_failure(void)
{
  return ssh_format("%c", SSH_MSG_REQUEST_FAILURE);
}

114
115
116
117
118
struct lsh_string *format_global_success(void)
{
  return ssh_format("%c", SSH_MSG_REQUEST_SUCCESS);
}

119
120
struct lsh_string *format_open_confirmation(struct ssh_channel *channel,
					    UINT32 channel_number,
121
					    const char *format, ...)
122
123
124
125
126
127
{
  va_list args;
  UINT32 l1, l2;
  struct lsh_string *packet;

#define CONFIRM_FORMAT "%c%i%i%i%i"
128
129
#define CONFIRM_ARGS \
  SSH_MSG_CHANNEL_OPEN_CONFIRMATION, channel->channel_number, \
130
131
132
133
134
135
136
137
138
139
140
141
142
  channel_number, channel->rec_window_size, channel->rec_max_packet
    
  l1 = ssh_format_length(CONFIRM_FORMAT, CONFIRM_ARGS);

  va_start(args, format);
  l2 = ssh_vformat_length(format, args);
  va_end(args);

  packet = lsh_string_alloc(l1 + l2);

  ssh_format_write(CONFIRM_FORMAT, l1, packet->data, CONFIRM_ARGS);

  va_start(args, format);
143
  ssh_vformat_write(format, l2, packet->data+l1, args);
144
145
146
147
148
149
150
  va_end(args);

  return packet;
#undef CONFIRM_FORMAT
#undef CONFIRM_ARGS
}

Niels Möller's avatar
Niels Möller committed
151
struct lsh_string *format_open_failure(UINT32 channel, UINT32 reason,
152
				       const char *msg, const char *language)
Niels Möller's avatar
Niels Möller committed
153
154
155
156
157
{
  return ssh_format("%c%i%i%z%z", SSH_MSG_CHANNEL_OPEN_FAILURE,
		    channel, reason, msg, language);
}

158
159
160
161
162
struct lsh_string *format_channel_success(UINT32 channel)
{
  return ssh_format("%c%i", SSH_MSG_CHANNEL_SUCCESS, channel);
}

Niels Möller's avatar
Niels Möller committed
163
164
165
166
167
struct lsh_string *format_channel_failure(UINT32 channel)
{
  return ssh_format("%c%i", SSH_MSG_CHANNEL_FAILURE, channel);
}

168
169
170
171
172
173
174
175
176
177
struct lsh_string *prepare_window_adjust(struct ssh_channel *channel,
					 UINT32 add)
{
  channel->rec_window_size += add;
  
  return ssh_format("%c%i%i",
		    SSH_MSG_CHANNEL_WINDOW_ADJUST,
		    channel->channel_number, add);
}

Niels Möller's avatar
Niels Möller committed
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
/* ;; GABA:
   (class
     (name channel_exception)
     (super exception)
     (vars
       (channel object ssh_channel)
       (pending_close . int)))
*/

/* GABA:
   (class
     (name exc_finish_channel_handler)
     (super exception_handler)
     (vars
       (table object channel_table)
       ; Local channel number 
       (channel_number . UINT32)))
*/

static void do_exc_finish_channel_handler(struct exception_handler *s,
Niels Möller's avatar
Niels Möller committed
198
					  const struct exception *e)
Niels Möller's avatar
Niels Möller committed
199
200
201
202
203
{
  CAST(exc_finish_channel_handler, self, s);

  switch (e->type)
    {
Niels Möller's avatar
Niels Möller committed
204
205
206
207
208
209
210
    case EXC_FINISH_PENDING:
      self->table->pending_close = 1;

      if (!self->table->next_channel)
	EXCEPTION_RAISE(self->super.parent, &finish_read_exception);
      break;
      
Niels Möller's avatar
Niels Möller committed
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
    case EXC_FINISH_CHANNEL:
      /* NOTE: This type of exception must be handled only once.
       * Perhaps we must add a liveness flag in the ssh_channel struct
       * to avoid deallocating dead channels? */
      {
	struct ssh_channel *channel
	  = self->table->channels[self->channel_number];

	assert(channel);
	
	if (channel->close)
	  CHANNEL_CLOSE(channel);
	
	dealloc_channel(self->table, self->channel_number);

	if (self->table->pending_close && !self->table->next_channel)
	  EXCEPTION_RAISE(self->super.parent, &finish_read_exception);
      }
      break;
    default:
      EXCEPTION_RAISE(self->super.parent, e);
    }
}

static struct exception_handler *
make_exc_finish_channel_handler(struct channel_table *table,
				UINT32 channel_number,
				struct exception_handler *e)
{
  NEW(exc_finish_channel_handler, self);
  self->table = table;
  self->channel_number = channel_number;
  self->super.parent = e;
  self->super.raise = do_exc_finish_channel_handler;
Niels Möller's avatar
Niels Möller committed
245
246

  return &self->super;
Niels Möller's avatar
Niels Möller committed
247
248
249
}
				

250
/* Channel objects */
Niels Möller's avatar
Niels Möller committed
251
252
253
254
255

#define INITIAL_CHANNELS 32
/* Arbitrary limit */
#define MAX_CHANNELS (1L<<17)

Niels Möller's avatar
Niels Möller committed
256
struct channel_table *make_channel_table(void)
Niels Möller's avatar
Niels Möller committed
257
{
258
  NEW(channel_table, table);
Niels Möller's avatar
Niels Möller committed
259

260
  table->channels = lsh_space_alloc(sizeof(struct ssh_channel *)
261
				      * INITIAL_CHANNELS);
Niels Möller's avatar
Niels Möller committed
262
263
  table->in_use = lsh_space_alloc(INITIAL_CHANNELS);
  
264
265
266
267
  table->allocated_channels = INITIAL_CHANNELS;
  table->next_channel = 0;
  table->used_channels = 0;
  table->max_channels = MAX_CHANNELS;
Niels Möller's avatar
Niels Möller committed
268

Niels Möller's avatar
Niels Möller committed
269
  table->pending_close = 0;
270

271
272
273
  table->global_requests = make_alist(0, -1);
  table->channel_types = make_alist(0, -1);
  
274
275
276
  object_queue_init(&table->local_ports);
  object_queue_init(&table->remote_ports);
  
277
278
  object_queue_init(&table->active_global_requests);
  object_queue_init(&table->pending_global_requests);
Niels Möller's avatar
Niels Möller committed
279
  
280
  return table;
Niels Möller's avatar
Niels Möller committed
281
282
283
};

/* Returns -1 if allocation fails */
284
285
286
/* NOTE: This function returns locally chosen channel numbers, which
 * are always small integers. So there's no problem fitting them in
 * a signed int. */
287
int alloc_channel(struct channel_table *table)
Niels Möller's avatar
Niels Möller committed
288
{
289
  UINT32 i;
290
  
291
  for(i = table->next_channel; i < table->used_channels; i++)
Niels Möller's avatar
Niels Möller committed
292
    {
Niels Möller's avatar
Niels Möller committed
293
      if (!table->in_use[i])
Niels Möller's avatar
Niels Möller committed
294
	{
Niels Möller's avatar
Niels Möller committed
295
296
	  assert(!table->channels[i]);
	  table->in_use[i] = 1;
297
	  table->next_channel = i+1;
Niels Möller's avatar
Niels Möller committed
298
299
300
	  return i;
	}
    }
301
  if (i == table->max_channels)
Niels Möller's avatar
Niels Möller committed
302
    return -1;
303

304
  if (i == table->allocated_channels) 
Niels Möller's avatar
Niels Möller committed
305
    {
306
      int new_size = table->allocated_channels * 2;
Niels Möller's avatar
Niels Möller committed
307
      struct ssh_channel **new_channels;
Niels Möller's avatar
Niels Möller committed
308
      UINT8 *new_in_use;
Niels Möller's avatar
Niels Möller committed
309

Niels Möller's avatar
Niels Möller committed
310
311
312
      new_channels = lsh_space_alloc(sizeof(struct ssh_channel *)
				     * new_size);
      memcpy(new_channels, table->channels,
313
	     sizeof(struct ssh_channel *) * table->used_channels);
Niels Möller's avatar
Niels Möller committed
314
315
316
317
318
319
320
321
      lsh_space_free(table->channels);
      table->channels = new_channels;

      new_in_use = lsh_space_alloc(new_size);
      memcpy(new_in_use, table->in_use, table->used_channels);
      lsh_space_free(table->in_use);
      table->in_use = new_in_use;

322
      table->allocated_channels = new_size;
Niels Möller's avatar
Niels Möller committed
323
324
    }

325
  table->next_channel = table->used_channels = i+1;
Niels Möller's avatar
Niels Möller committed
326

327
  table->in_use[i] = 1;
Niels Möller's avatar
Niels Möller committed
328
329
330
  return i;
}

331
void dealloc_channel(struct channel_table *table, int i)
Niels Möller's avatar
Niels Möller committed
332
333
{
  assert(i >= 0);
334
  assert( (unsigned) i < table->used_channels);
Niels Möller's avatar
Niels Möller committed
335
  
336
  table->channels[i] = NULL;
Niels Möller's avatar
Niels Möller committed
337
338
  table->in_use[i] = 0;
  
339
  if ( (unsigned) i < table->next_channel)
340
341
342
    table->next_channel = i;
}

Niels Möller's avatar
Niels Möller committed
343
void register_channel(struct channel_table *table,
Niels Möller's avatar
Niels Möller committed
344
345
		      UINT32 local_channel_number,
		      struct ssh_channel *channel)
346
{
Niels Möller's avatar
Niels Möller committed
347
348
  assert(table->in_use[local_channel_number]);
  assert(!table->channels[local_channel_number]);
Niels Möller's avatar
Niels Möller committed
349
  
Niels Möller's avatar
Niels Möller committed
350
351
352
353
354
355
  table->channels[local_channel_number] = channel;

  /* FIXME: Is this the right place to install this exception handler? */
  channel->e = make_exc_finish_channel_handler(table,
					       local_channel_number,
					       channel->e);
Niels Möller's avatar
Niels Möller committed
356
357
}

358
struct ssh_channel *lookup_channel(struct channel_table *table, UINT32 i)
Niels Möller's avatar
Niels Möller committed
359
{
360
361
  return (i < table->used_channels)
    ? table->channels[i] : NULL;
Niels Möller's avatar
Niels Möller committed
362
363
}

364
365
366
367
368
369
370
371
372
373
/* FIXME: It seems suboptimal to send a window adjust message for *every* write that we do.
 * A better scheme might be as follows:
 *
 * Delay window adjust messages, keeping track of both the locally
 * maintained window size, which is updated after each write, and the
 * size that has been reported to the remote end. When the difference
 * between these two values gets large enough (say, larger than one
 * half or one third of the maximum window size), we send a
 * window_adjust message to sync them. */
static void adjust_rec_window(struct flow_controlled *f, UINT32 written)
374
{
375
376
  CAST_SUBTYPE(ssh_channel, channel, f);

Niels Möller's avatar
Niels Möller committed
377
  A_WRITE(channel->write,
378
	  prepare_window_adjust(channel, written));
379
380
}

Niels Möller's avatar
Niels Möller committed
381
void channel_start_receive(struct ssh_channel *channel)
382
{
Niels Möller's avatar
Niels Möller committed
383
384
  A_WRITE(channel->write,
	  prepare_window_adjust
385
	  (channel, channel->max_window - channel->rec_window_size));
386
387
}

Niels Möller's avatar
Niels Möller committed
388
#if 0
389
390
/* Process channel-related status codes. Used by the packet handlers,
 * before returning. */
Niels Möller's avatar
Niels Möller committed
391
392
393
394
static int channel_process_status(struct channel_table *table,
				  int channel,
				  int status)
{
395
396
397
398
  struct ssh_channel *c = table->channels[channel];
  
  while (!LSH_CLOSEDP(status))
    {
399
400
401
402
403
404
405
      if (status & LSH_CHANNEL_CLOSE)
	{ /* Close the channel now */ 
	  if (!c->flags & CHANNEL_SENT_CLOSE)
	    status |= channel_close(c);
	  break;
	}
      
406
407
408
409
410
411
412
413
414
      if (status & LSH_CHANNEL_READY_SEND)
	{
	  status &= ~ LSH_CHANNEL_READY_SEND;
	  if (c->send_window_size)
	    status |= CHANNEL_SEND(c);
	}
      else if (status & LSH_CHANNEL_READY_REC)
	{
	  status &= ~ LSH_CHANNEL_READY_REC;
415
	  status |= channel_start_receive(c);
416
417
418
419
420
	}
      else
	break;
    }
	
Niels Möller's avatar
Niels Möller committed
421
422
423
424
  if (status & LSH_CHANNEL_FINISHED)
    {
      /* Clear this bit */
      status &= ~LSH_CHANNEL_FINISHED;
425

426
      if (c->close)
427
	status |= CHANNEL_CLOSE(c);
Niels Möller's avatar
Niels Möller committed
428
429
430
431
      
      dealloc_channel(table, channel);
    }

432
433
434
435
436
437
438
  if (status & LSH_CHANNEL_PENDING_CLOSE)
    table->pending_close = 1;
  
  /* If this was the last channel, close connection */
  if (table->pending_close && !table->next_channel)
    status |= LSH_CLOSE;

Niels Möller's avatar
Niels Möller committed
439
440
  return status;
}
Niels Möller's avatar
Niels Möller committed
441
442
#endif

Niels Möller's avatar
Niels Möller committed
443

444
445
446
/* Ugly macros to make it a little simpler to free the input packet at
 * the right time. */

Niels Möller's avatar
Niels Möller committed
447
448
449
450
451
#define RETURN goto foo_finish
#define END(s) do { foo_finish: \
                    lsh_string_free((s)); \
                    return; } while(0)

452
#if 0
Niels Möller's avatar
Niels Möller committed
453
454
455
#define ERROR(e, msg) \
  EXCEPTION_RAISE(e, make_protocol_exception \
    (SSH_DISCONNECT_PROTOCOL_ERROR, msg))
456
#endif
457

Niels Möller's avatar
Niels Möller committed
458
/* Channel related messages */
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477

/* GABA:
   (class
     (name global_request_status)
     (vars
       ; -1 for still active requests,
       ; 0 for failure,
       ; 1 for success
       (status . int)))
*/

static struct global_request_status *make_global_request_status(void)
{
  NEW(global_request_status, self);
  self->status = -1;

  return self;
}

Niels Möller's avatar
Niels Möller committed
478
/* FIXME: Split into a continuation and an exception handler */
479
480
481
482
483
484
485
486
/* GABA:
   (class
     (name global_request_response)
     (super global_request_callback)
     (vars
       (active object global_request_status)))
*/

Niels Möller's avatar
Niels Möller committed
487
static void
488
489
490
491
do_global_request_response(struct global_request_callback *c,
			   int success)
{
  CAST(global_request_response, self, c);
Niels Möller's avatar
Niels Möller committed
492
  struct object_queue *q = &self->super.connection->table->active_global_requests;
493
494
495
496
497
498
499
500
501
502
503

  assert( self->active->status == -1);
  assert( (success == 0) || (success == 1) );
  assert( !object_queue_is_empty(q));
	  
  self->active->status = success;

  for (;;)
    {
      CAST(global_request_status, n, object_queue_peek_head(q));
      if (!n || (n->status < 0))
Niels Möller's avatar
Niels Möller committed
504
	break;
505
506

      object_queue_remove_head(q);
Niels Möller's avatar
Niels Möller committed
507
508
509

      /* FIXME: Perhaps install some exception handler that cancels
       * the queue as soon as a write failes. */
510
      C_WRITE(self->super.connection,
Niels Möller's avatar
Niels Möller committed
511
512
	      (n->status
	       ? format_global_success()
513
	       : format_global_failure()));
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
    }
}

static struct global_request_callback *
make_global_request_response(struct ssh_connection *connection,
			     struct global_request_status *active)
{
  NEW(global_request_response, self);

  self->super.connection = connection;
  self->super.response = do_global_request_response;

  self->active = active;

  return &self->super;
}
     
Niels Möller's avatar
Niels Möller committed
531
532
533
static void do_global_request(struct packet_handler *s UNUSED,
			      struct ssh_connection *connection,
			      struct lsh_string *packet)
Niels Möller's avatar
Niels Möller committed
534
{
535
  /* CAST(global_request_handler, closure, c); */
Niels Möller's avatar
Niels Möller committed
536
537

  struct simple_buffer buffer;
538
  unsigned msg_number;
Niels Möller's avatar
Niels Möller committed
539
540
541
542
543
544
545
546
547
548
549
  int name;
  int want_reply;
  
  simple_buffer_init(&buffer, packet->length, packet->data);

  if (parse_uint8(&buffer, &msg_number)
      && (msg_number == SSH_MSG_GLOBAL_REQUEST)
      && parse_atom(&buffer, &name)
      && parse_boolean(&buffer, &want_reply))
    {
      struct global_request *req;
550
551
      struct global_request_callback *c = NULL;
      
Niels Möller's avatar
Niels Möller committed
552
      if (!name || !(req = ALIST_GET(connection->table->global_requests,
553
				     name)))
554
	{
Niels Möller's avatar
Niels Möller committed
555
	  lsh_string_free(packet);
556

Niels Möller's avatar
Niels Möller committed
557
558
	  C_WRITE(connection, format_global_failure());
	  return;
Niels Möller's avatar
Niels Möller committed
559
560
561
562
563
564
565
	}
      else
	{
	  if (want_reply)
	    {
	      struct global_request_status *a = make_global_request_status();
	      
Niels Möller's avatar
Niels Möller committed
566
	      object_queue_add_tail(&connection->table->active_global_requests,
Niels Möller's avatar
Niels Möller committed
567
568
569
570
571
				    &a->super);
	      
	      c = make_global_request_response(connection, a);
	    }
	  GLOBAL_REQUEST(req, connection, &buffer, c);
572
	}
Niels Möller's avatar
Niels Möller committed
573
    }
Niels Möller's avatar
Niels Möller committed
574
575
  else
    {
576
      PROTOCOL_ERROR(connection->e, "Invalid SSH_MSG_GLOBAL_REQUEST message.");
Niels Möller's avatar
Niels Möller committed
577
578
    }
  lsh_string_free(packet);
Niels Möller's avatar
Niels Möller committed
579
580
}

Niels Möller's avatar
Niels Möller committed
581
582
583
584
static void
do_global_request_success(struct packet_handler *s UNUSED,
			  struct ssh_connection *connection,
			  struct lsh_string *packet)
585
586
{
  if (packet->length != 1)
Niels Möller's avatar
Niels Möller committed
587
    {
588
      PROTOCOL_ERROR(connection->e, "Invalid GLOBAL_REQUEST_SUCCESS message.");
Niels Möller's avatar
Niels Möller committed
589
590
      RETURN;
    }
591
592
593

  assert(packet->data[0] == SSH_MSG_REQUEST_SUCCESS);

Niels Möller's avatar
Niels Möller committed
594
  if (object_queue_is_empty(&connection->table->pending_global_requests))
595
596
    {
      werror("do_global_request_success: Unexpected message, ignoring.\n");
Niels Möller's avatar
Niels Möller committed
597
      return;
598
599
    }
  {
600
    CAST_SUBTYPE(command_context, ctx,
Niels Möller's avatar
Niels Möller committed
601
602
		 object_queue_remove_head(&connection->table->pending_global_requests));
    COMMAND_RETURN(ctx->c, connection);
603
  }
Niels Möller's avatar
Niels Möller committed
604
  END(packet);
605
606
}

607
608
609
struct exception global_request_exception =
STATIC_EXCEPTION(EXC_GLOBAL_REQUEST, "Global request failed");

Niels Möller's avatar
Niels Möller committed
610
611
612
613
static void
do_global_request_failure(struct packet_handler *s UNUSED,
			  struct ssh_connection *connection,
			  struct lsh_string *packet)
614
615
{
  if (packet->length != 1)
Niels Möller's avatar
Niels Möller committed
616
    {
617
      PROTOCOL_ERROR(connection->e, "Invalid GLOBAL_REQUEST_FAILURE message.");
Niels Möller's avatar
Niels Möller committed
618
619
      RETURN;
    }
620
621
622

  assert(packet->data[0] == SSH_MSG_REQUEST_FAILURE);

Niels Möller's avatar
Niels Möller committed
623
  if (object_queue_is_empty(&connection->table->pending_global_requests))
624
625
626
    {
      werror("do_global_request_failure: Unexpected message, ignoring.\n");
    }
Niels Möller's avatar
Niels Möller committed
627
628
629
630
631
632
633
  else
    {
      CAST_SUBTYPE(command_context, ctx,
		   object_queue_remove_head(&connection->table->pending_global_requests));
      EXCEPTION_RAISE(ctx->e, &global_request_exception);
    }
  END(packet);
634
635
}

Niels Möller's avatar
Niels Möller committed
636
637
#if 0
/* HERE!!! */
638
/* FIXME: Split into a continuation and an exception handler */
639
640
641
642
643
644
645
646
647
648
649
650
651
/* Callback given to the CHANNEL_OPEN method */
static int do_channel_open_response(struct channel_open_callback *c,
                                    struct ssh_channel *channel,
                                    UINT32 error, char *error_msg,
                                    struct lsh_string *args)
{
  CAST(channel_open_response, closure, c);
  
  int local_channel_number;

  if (!channel)
    {
      if (error)
Niels Möller's avatar
Niels Möller committed
652
653
654
655
656
657
        C_WRITE(closure->super.connection->write,
		format_open_failure(closure->remote_channel_number,
				    error, error_msg, ""));
      else
	EXCEPTION_RAISE(closure->super.connection->e,
			make_protocol_exception())
658
      return LSH_FAIL | LSH_DIE;
659
660
    }

Niels Möller's avatar
Niels Möller committed
661
662
663
  /* FIXME: It would be better to allocate or at least reserve a channel number earlier,
   * so that we can't fail at this point.
   */
664
665
666
667
668
669
670
671
672
673
674
675
  if ( (local_channel_number
            = register_channel(closure->super.connection->channels,
			       channel)) < 0)
    {
      werror("Could not allocate a channel number for opened channel!\n");
      return A_WRITE(closure->super.connection->write,
                     format_open_failure(closure->remote_channel_number,
                                         SSH_OPEN_RESOURCE_SHORTAGE,
                                         "Could not allocate a channel number "
                                         "(shouldn't happen...)", ""));
    }

676
  /* FIXME: This copying could just as well be done by the
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
   * CHANNEL_OPEN handler? Then we can remove the corresponding fields
   * from the closure as well. */
  channel->send_window_size = closure->window_size;
  channel->send_max_packet = closure->max_packet;
  channel->channel_number = closure->remote_channel_number;

  /* FIXME: Is the channel->write field really needed? */
  channel->write = closure->super.connection->write;

  return A_WRITE(closure->super.connection->write,
                 args
                 ? format_open_confirmation(channel, local_channel_number,
                                            "%lfS", args)
                 : format_open_confirmation(channel, local_channel_number,
                                            ""));
}

static struct channel_open_response *
make_channel_open_response(struct ssh_connection* connection,
			   UINT32 remote_channel_number,
			   UINT32 window_size,
			   UINT32 max_packet)
{
  NEW(channel_open_response, closure);

  closure->super.response = do_channel_open_response;
  closure->super.connection = connection;
  closure->remote_channel_number = remote_channel_number;
  closure->window_size = window_size;
  closure->max_packet = max_packet;

  return closure;
}
Niels Möller's avatar
Niels Möller committed
710
#endif
711

Niels Möller's avatar
Niels Möller committed
712
/* GABA:
713
714
715
716
717
   (class
     (name channel_open_continuation)
     (super command_continuation)
     (vars
       (connection object ssh_connection)
Niels Möller's avatar
Niels Möller committed
718
719
720
721
       (local_channel_number . UINT32)
       (remote_channel_number . UINT32)
       (window_size . UINT32)
       (max_packet . UINT32)))
722
*/
723

Niels Möller's avatar
Niels Möller committed
724
725
726
static void
do_channel_open_continue(struct command_continuation *c,
			 struct lsh_object *value)
727
728
{
  CAST(channel_open_continuation, self, c);
Niels Möller's avatar
Niels Möller committed
729
730
731
732
733
734
735
736
737
738
739
740
  CAST_SUBTYPE(ssh_channel, channel, value);

  assert(channel);

  /* FIXME: This copying could just as well be done by the
   * CHANNEL_OPEN handler? Then we can remove the corresponding fields
   * from the closure as well. */
  channel->send_window_size = self->window_size;
  channel->send_max_packet = self->max_packet;
  channel->channel_number = self->remote_channel_number;

  /* FIXME: Is the channel->write field really needed? */
Niels Möller's avatar
Niels Möller committed
741
  channel->write = self->connection->write;
Niels Möller's avatar
Niels Möller committed
742

Niels Möller's avatar
Niels Möller committed
743
744
  register_channel(self->connection->table,
		   self->local_channel_number, channel);
Niels Möller's avatar
Niels Möller committed
745
746
747
748
749

  /* FIXME: Doesn't support sending extra arguments with the
   * confirmation message. */

  C_WRITE(self->connection,
Niels Möller's avatar
Niels Möller committed
750
	  format_open_confirmation(channel, self->local_channel_number, ""));
751
}
Niels Möller's avatar
Niels Möller committed
752

Niels Möller's avatar
Niels Möller committed
753
static struct command_continuation *
Niels Möller's avatar
Niels Möller committed
754
755
756
757
758
759
760
make_channel_open_continuation(struct ssh_connection *connection,
			       UINT32 local_channel_number,
			       UINT32 remote_channel_number,
			       UINT32 window_size,
			       UINT32 max_packet)
{
  NEW(channel_open_continuation, self);
Niels Möller's avatar
Niels Möller committed
761

Niels Möller's avatar
Niels Möller committed
762
763
764
765
766
767
768
  self->super.c = do_channel_open_continue;
  self->connection = connection;
  self->local_channel_number = local_channel_number;
  self->remote_channel_number = remote_channel_number;
  self->window_size = window_size;
  self->max_packet = max_packet;

Niels Möller's avatar
Niels Möller committed
769
  return &self->super;
Niels Möller's avatar
Niels Möller committed
770
771
772
773
774
775
776
777
778
}
			       
/* GABA:
   (class
     (name exc_channel_open_handler)
     (super exception_handler)
     (vars
       (connection object ssh_connection)
       (local_channel_number . UINT32)
Niels Möller's avatar
Niels Möller committed
779
       (remote_channel_number . UINT32)))
Niels Möller's avatar
Niels Möller committed
780
781
782
*/

static void do_exc_channel_open_handler(struct exception_handler *s,
Niels Möller's avatar
Niels Möller committed
783
					const struct exception *e)
Niels Möller's avatar
Niels Möller committed
784
785
786
787
788
789
790
791
{
  CAST(exc_channel_open_handler, self, s);

  switch (e->type)
    {
    case EXC_CHANNEL_OPEN:
      {
	CAST_SUBTYPE(channel_open_exception, exc, e);
Niels Möller's avatar
Niels Möller committed
792
	struct channel_table *table = self->connection->table;
Niels Möller's avatar
Niels Möller committed
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
	
	assert(table->in_use[self->local_channel_number]);
	assert(!table->channels[self->local_channel_number]);

	dealloc_channel(table, self->local_channel_number);
	
        C_WRITE(self->connection,
		format_open_failure(self->remote_channel_number,
				    exc->error_code, e->msg, ""));
	break;
      }
    default:
      EXCEPTION_RAISE(self->super.parent, e);
    }      
}

static struct exception_handler *
make_exc_channel_open_handler(struct ssh_connection *connection,
			      UINT32 local_channel_number,
			      UINT32 remote_channel_number,
			      struct exception_handler *parent)
{
Niels Möller's avatar
Niels Möller committed
815
  NEW(exc_channel_open_handler, self);
Niels Möller's avatar
Niels Möller committed
816
  self->super.parent = parent;
Niels Möller's avatar
Niels Möller committed
817
818
  self->super.raise = do_exc_channel_open_handler;
  self->connection = connection;
Niels Möller's avatar
Niels Möller committed
819
820
821
822
823
824
825
826
827
  self->local_channel_number = local_channel_number;
  self->remote_channel_number = remote_channel_number;

  return &self->super;
}

static void do_channel_open(struct packet_handler *c UNUSED,
			    struct ssh_connection *connection,
			    struct lsh_string *packet)
Niels Möller's avatar
Niels Möller committed
828
{
829
  /* CAST(channel_open_handler, closure, c); */
Niels Möller's avatar
Niels Möller committed
830
831

  struct simple_buffer buffer;
832
  unsigned msg_number;
Niels Möller's avatar
Niels Möller committed
833
  int type;
834
835
836
  UINT32 remote_channel_number;
  UINT32 window_size;
  UINT32 max_packet;
Niels Möller's avatar
Niels Möller committed
837
838
839
840
841
842
  
  simple_buffer_init(&buffer, packet->length, packet->data);

  if (parse_uint8(&buffer, &msg_number)
      && (msg_number == SSH_MSG_CHANNEL_OPEN)
      && parse_atom(&buffer, &type)
843
844
845
      && parse_uint32(&buffer, &remote_channel_number)
      && parse_uint32(&buffer, &window_size)
      && parse_uint32(&buffer, &max_packet))
Niels Möller's avatar
Niels Möller committed
846
847
    {
      struct channel_open *open;
Niels Möller's avatar
Niels Möller committed
848

Niels Möller's avatar
Niels Möller committed
849
850
851
852
      /* NOTE: We can't free the packet yet, as the buffer is passed
       * to the CHANNEL_OPEN method later. */

      if (connection->table->pending_close)
Niels Möller's avatar
Niels Möller committed
853
854
855
856
857
858
859
860
	{
	  /* We are waiting for channels to close. Don't open any new ones. */

	  C_WRITE(connection,
		  format_open_failure(remote_channel_number,
				      SSH_OPEN_ADMINISTRATIVELY_PROHIBITED,
				      "Waiting for channels to close.", ""));
	}
Niels Möller's avatar
Niels Möller committed
861
      else if (!type || !(open = ALIST_GET(connection->table->channel_types,
862
				      type)))
Niels Möller's avatar
Niels Möller committed
863
864
865
866
867
868
869
870
	{
	  C_WRITE(connection,
		  format_open_failure(remote_channel_number,
				      SSH_OPEN_UNKNOWN_CHANNEL_TYPE,
				      "Unknown channel type", ""));
	}
      else
	{
Niels Möller's avatar
Niels Möller committed
871
      	  int local_number = alloc_channel(connection->table);
Niels Möller's avatar
Niels Möller committed
872
873
874
875
876
877
878
879

	  if (local_number < 0)
	    C_WRITE(connection,
		    format_open_failure(remote_channel_number,
					SSH_OPEN_RESOURCE_SHORTAGE,
					"Unknown channel type", ""));

	  
Niels Möller's avatar
Niels Möller committed
880
881
882
883
884
885
886
887
888
889
890
	  
	  CHANNEL_OPEN(open, connection, &buffer,
		       make_channel_open_continuation(connection,
						      local_number,
						      remote_channel_number,
						      window_size,
						      max_packet),
		       make_exc_channel_open_handler(connection,
						     local_number,
						     remote_channel_number,
						     connection->e));
Niels Möller's avatar
Niels Möller committed
891
	}
Niels Möller's avatar
Niels Möller committed
892
    }
Niels Möller's avatar
Niels Möller committed
893
  else
894
    PROTOCOL_ERROR(connection->e, "Invalid SSH_MSG_CHANNEL_OPEN message.");
Niels Möller's avatar
Niels Möller committed
895

Niels Möller's avatar
Niels Möller committed
896
  lsh_string_free(packet);
Niels Möller's avatar
Niels Möller committed
897
898
}     

Niels Möller's avatar
Niels Möller committed
899
900
901
902
static void
do_channel_request(struct packet_handler *closure UNUSED,
		   struct ssh_connection *connection,
		   struct lsh_string *packet)
Niels Möller's avatar
Niels Möller committed
903
904
{
  struct simple_buffer buffer;
905
  unsigned msg_number;
Niels Möller's avatar
Niels Möller committed
906
907
908
  UINT32 channel_number;
  int type;
  int want_reply;
909
  
Niels Möller's avatar
Niels Möller committed
910
911
912
913
914
915
916
917
  simple_buffer_init(&buffer, packet->length, packet->data);

  if (parse_uint8(&buffer, &msg_number)
      && (msg_number == SSH_MSG_CHANNEL_REQUEST)
      && parse_uint32(&buffer, &channel_number)
      && parse_atom(&buffer, &type)
      && parse_boolean(&buffer, &want_reply))
    {
Niels Möller's avatar
Niels Möller committed
918
      struct ssh_channel *channel = lookup_channel(connection->table,
Niels Möller's avatar
Niels Möller committed
919
920
						   channel_number);

921
922
923
924
      /* NOTE: We can't free packet yet, because it is not yet fully
       * parsed. There may be some more arguments, which are parsed by
       * the CHANNEL_REQUEST method below. */

Niels Möller's avatar
Niels Möller committed
925
926
927
928
      if (channel)
	{
	  struct channel_request *req;

929
930
	  if (type && channel->request_types 
	      && ( (req = ALIST_GET(channel->request_types, type)) ))
Niels Möller's avatar
Niels Möller committed
931
	    CHANNEL_REQUEST(req, channel, connection, want_reply, &buffer);
932
	  else
Niels Möller's avatar
Niels Möller committed
933
934
	    {
	      if (want_reply)
Niels Möller's avatar
Niels Möller committed
935
		C_WRITE(connection,
Niels Möller's avatar
Niels Möller committed
936
937
938
939
940
941
942
			format_channel_failure(channel->channel_number));
	    }
	}
      else
	{
	  werror("SSH_MSG_CHANNEL_REQUEST on nonexistant channel %i\n",
		 channel_number);
Niels Möller's avatar
Niels Möller committed
943
	}
944
    }
Niels Möller's avatar
Niels Möller committed
945
  else
946
    PROTOCOL_ERROR(connection->e, "Invalid SSH_MSG_CHANNEL_REQUEST message.");
Niels Möller's avatar
Niels Möller committed
947
  
Niels Möller's avatar
Niels Möller committed
948
  lsh_string_free(packet);
Niels Möller's avatar
Niels Möller committed
949
950
}
      
Niels Möller's avatar
Niels Möller committed
951
952
953
954
static void
do_window_adjust(struct packet_handler *closure UNUSED,
		 struct ssh_connection *connection,
		 struct lsh_string *packet)
Niels Möller's avatar
Niels Möller committed
955
956
{
  struct simple_buffer buffer;
957
  unsigned msg_number;
Niels Möller's avatar
Niels Möller committed
958
959
960
961
962
963
964
  UINT32 channel_number;
  UINT32 size;

  simple_buffer_init(&buffer, packet->length, packet->data);

  if (parse_uint8(&buffer, &msg_number)
      && (msg_number == SSH_MSG_CHANNEL_WINDOW_ADJUST)
965
      && parse_uint32(&buffer, &channel_number)
Niels Möller's avatar
Niels Möller committed
966
967
968
      && parse_uint32(&buffer, &size)
      && parse_eod(&buffer))
    {
Niels Möller's avatar
Niels Möller committed
969
      struct ssh_channel *channel = lookup_channel(connection->table,
Niels Möller's avatar
Niels Möller committed
970
971
972
973
						   channel_number);

      lsh_string_free(packet);
      
Niels Möller's avatar
Niels Möller committed
974
      if (channel
975
976
	  && !(channel->flags & (CHANNEL_RECEIVED_EOF
				 | CHANNEL_RECEIVED_CLOSE)))
Niels Möller's avatar
Niels Möller committed
977
	{
Niels Möller's avatar
Niels Möller committed
978
979
980
981
	  if (! (channel->flags & CHANNEL_SENT_CLOSE))
	    {
	      channel->send_window_size += size;
	      if (channel->send_window_size && channel->send)
Niels Möller's avatar
Niels Möller committed
982
		CHANNEL_SEND(channel);
Niels Möller's avatar
Niels Möller committed
983
	    }
Niels Möller's avatar
Niels Möller committed
984
	}
Niels Möller's avatar
Niels Möller committed
985
986
987
988
989
990
      else
	{
	  /* FIXME: What to do now? Should unknown channel numbers be
	   * ignored silently? */
	  werror("SSH_MSG_CHANNEL_WINDOW_ADJUST on nonexistant or closed "
		 "channel %i\n", channel_number);
991
	  PROTOCOL_ERROR(connection->e, "Unexpected CHANNEL_WINDOW_ADJUST");
Niels Möller's avatar
Niels Möller committed
992
993
994
995
996
	}
    }
  else
    {
      lsh_string_free(packet);
997
      PROTOCOL_ERROR(connection->e, "Invalid CHANNEL_WINDOW_ADJUST message.");
Niels Möller's avatar
Niels Möller committed
998
999
1000
    }
}

Niels Möller's avatar
Niels Möller committed
1001
1002
1003
1004
static void
do_channel_data(struct packet_handler *closure UNUSED,
		struct ssh_connection *connection,
		struct lsh_string *packet)
Niels Möller's avatar
Niels Möller committed
1005
1006
{
  struct simple_buffer buffer;
1007
  unsigned msg_number;
Niels Möller's avatar
Niels Möller committed
1008
1009
1010
1011
1012
1013
1014
  UINT32 channel_number;
  struct lsh_string *data;
  
  simple_buffer_init(&buffer, packet->length, packet->data);

  if (parse_uint8(&buffer, &msg_number)
      && (msg_number == SSH_MSG_CHANNEL_DATA)
1015
      && parse_uint32(&buffer, &channel_number)
Niels Möller's avatar
Niels Möller committed
1016
1017
1018
      && ( (data = parse_string_copy(&buffer)) )
      && parse_eod(&buffer))
    {
Niels Möller's avatar
Niels Möller committed
1019
      struct ssh_channel *channel = lookup_channel(connection->table,
Niels Möller's avatar
Niels Möller committed
1020
1021
1022
1023
						   channel_number);

      lsh_string_free(packet);
      
1024
      if (channel && channel->receive
1025
1026
	  && !(channel->flags & (CHANNEL_RECEIVED_EOF
				 | CHANNEL_RECEIVED_CLOSE)))
Niels Möller's avatar
Niels Möller committed
1027
	{
Niels Möller's avatar
Niels Möller committed
1028
	  if (channel->flags & CHANNEL_SENT_CLOSE)
Niels Möller's avatar
Niels Möller committed
1029
	    {
Niels Möller's avatar
Niels Möller committed
1030
	      lsh_string_free(data);
1031
	      werror("Ignoring data on channel which is closing\n");
Niels Möller's avatar
Niels Möller committed
1032
	      return;
Niels Möller's avatar
Niels Möller committed
1033
	    }
Niels Möller's avatar
Niels Möller committed
1034
1035
1036
1037
1038
	  else
	    {
	      if (data->length > channel->rec_window_size)
		{
		  /* Truncate data to fit window */
1039
		  werror("Channel data overflow. Extra data ignored.\n"); 
Niels Möller's avatar
Niels Möller committed
1040
1041
		  data->length = channel->rec_window_size;
		}
1042
1043

	      if (!data->length)
Niels Möller's avatar
Niels Möller committed
1044
1045
1046
1047
1048
		{
		  /* Ignore data packet */
		  lsh_string_free(data);
		  return;
		}
1049
1050
	      channel->rec_window_size -= data->length;

Niels Möller's avatar
Niels Möller committed
1051
	      CHANNEL_RECEIVE(channel, CHANNEL_DATA, data);
Niels Möller's avatar
Niels Möller committed
1052
	    }
Niels Möller's avatar
Niels Möller committed
1053
	}
Niels Möller's avatar
Niels Möller committed
1054
1055
1056
1057
1058
1059
1060
1061
1062
1063
      else
	{
	  werror("Data on closed or non-existant channel %i\n",
		 channel_number);
	  lsh_string_free(data);
	}
    }
  else
    {
      lsh_string_free(packet);
1064
      PROTOCOL_ERROR(connection->e, "Invalid CHANNEL_DATA message.");
Niels Möller's avatar
Niels Möller committed
1065
1066
1067
    }
}

Niels Möller's avatar
Niels Möller committed
1068
1069
1070
1071
static void
do_channel_extended_data(struct packet_handler *closure UNUSED,
			 struct ssh_connection *connection,
			 struct lsh_string *packet)
Niels Möller's avatar
Niels Möller committed
1072
1073
{
  struct simple_buffer buffer;
1074
  unsigned msg_number;
Niels Möller's avatar
Niels Möller committed
1075
1076
1077
1078
1079
1080
1081
1082
  UINT32 channel_number;
  UINT32 type;
  struct lsh_string *data;
  
  simple_buffer_init(&buffer, packet->length, packet->data);

  if (parse_uint8(&buffer, &msg_number)
      && (msg_number == SSH_MSG_CHANNEL_EXTENDED_DATA)
1083
      && parse_uint32(&buffer, &channel_number)
Niels Möller's avatar
Niels Möller committed
1084
1085
1086
1087
      && parse_uint32(&buffer, &type)
      && ( (data = parse_string_copy(&buffer)) )
      && parse_eod(&buffer))
    {
Niels Möller's avatar
Niels Möller committed
1088
      struct ssh_channel *channel = lookup_channel(connection->table,
Niels Möller's avatar
Niels Möller committed
1089
1090
1091
1092
						   channel_number);

      lsh_string_free(packet);
      
1093
      if (channel && channel->receive
1094
1095
	  && !(channel->flags & (CHANNEL_RECEIVED_EOF
				 | CHANNEL_RECEIVED_CLOSE)))
Niels Möller's avatar
Niels Möller committed
1096
	{
Niels Möller's avatar
Niels Möller committed
1097
	  if (channel->flags & CHANNEL_SENT_CLOSE)
Niels Möller's avatar
Niels Möller committed
1098
	    {
Niels Möller's avatar
Niels Möller committed
1099
	      lsh_string_free(data);
1100
	      werror("Ignoring extended data on channel which is closing\n");
Niels Möller's avatar
Niels Möller committed
1101
	      return;
Niels Möller's avatar
Niels Möller committed
1102
	    }
Niels Möller's avatar
Niels Möller committed
1103
	  else
Niels Möller's avatar
Niels Möller committed
1104
	    {
Niels Möller's avatar
Niels Möller committed
1105
1106
1107
	      if (data->length > channel->rec_window_size)
		{
		  /* Truncate data to fit window */
1108
		  werror("Channel extended data overflow. "
Niels Möller's avatar
Niels Möller committed
1109
1110
1111
1112
			 "Extra data ignored.\n");
		  data->length = channel->rec_window_size;
		}
	      
Niels Möller's avatar
Niels Möller committed
1113
1114
1115
1116
1117
1118
	      if (!data->length)
		{
		  /* Ignore data packet */
		  lsh_string_free(data);
		  return;
		}
1119

Niels Möller's avatar
Niels Möller committed
1120
	      channel->rec_window_size -= data->length;
1121

Niels Möller's avatar
Niels Möller committed
1122
1123
1124
	      switch(type)
		{
		case SSH_EXTENDED_DATA_STDERR:
Niels Möller's avatar
Niels Möller committed
1125
1126
		  CHANNEL_RECEIVE(channel, CHANNEL_STDERR_DATA, data);
		  break;
Niels Möller's avatar
Niels Möller committed
1127
		default:
1128
		  werror("Unknown type %i of extended data.\n",
Niels Möller's avatar
Niels Möller committed
1129
1130
1131
			 type);
		  lsh_string_free(data);
		}
Niels Möller's avatar
Niels Möller committed
1132
1133
	    }
	}
Niels Möller's avatar
Niels Möller committed
1134
1135
1136
1137
1138
1139
1140
1141
1142
1143
      else
	{
	  werror("Extended data on closed or non-existant channel %i\n",
		 channel_number);
	  lsh_string_free(data);
	}
    }
  else
    {
      lsh_string_free(packet);
1144
      PROTOCOL_ERROR(connection->e, "Invalid CHANNEL_EXTENDED_DATA message.");
Niels Möller's avatar
Niels Möller committed
1145
1146
1147
    }
}

Niels Möller's avatar
Niels Möller committed
1148
1149
1150
1151
static void
do_channel_eof(struct packet_handler *closure UNUSED,
	       struct ssh_connection *connection,
	       struct lsh_string *packet)
Niels Möller's avatar
Niels Möller committed
1152
1153
{
  struct simple_buffer buffer;
1154
  unsigned msg_number;
Niels Möller's avatar
Niels Möller committed
1155
1156
1157
1158
1159
1160
  UINT32 channel_number;
  
  simple_buffer_init(&buffer, packet->length, packet->data);

  if (parse_uint8(&buffer, &msg_number)
      && (msg_number == SSH_MSG_CHANNEL_EOF)
1161
      && parse_uint32(&buffer, &channel_number)
Niels Möller's avatar
Niels Möller committed
1162
1163
      && parse_eod(&buffer))
    {
Niels Möller's avatar
Niels Möller committed
1164
      struct ssh_channel *channel = lookup_channel(connection->table,
Niels Möller's avatar
Niels Möller committed
1165
1166
1167
1168
						   channel_number);

      lsh_string_free(packet);

Niels Möller's avatar
Niels Möller committed
1169
1170
      if (channel)
	{
1171
	  if (channel->flags & (CHANNEL_RECEIVED_EOF | CHANNEL_RECEIVED_CLOSE))
Niels Möller's avatar
Niels Möller committed
1172
	    {
1173
	      werror("Receiving EOF on channel on closed channel.\n");
1174
1175
	      PROTOCOL_ERROR(connection->e,
			     "Received EOF on channel on closed channel.");
Niels Möller's avatar
Niels Möller committed
1176
	    }
1177
	  else
Niels Möller's avatar
Niels Möller committed
1178
	    {
Niels Möller's avatar
Niels Möller committed
1179
	      channel->flags |= CHANNEL_RECEIVED_EOF;
1180
	      
Niels Möller's avatar
Niels Möller committed
1181
1182
1183
1184
1185
1186
1187
	      if (channel->eof)
		CHANNEL_EOF(channel);
	      else
		/* FIXME: What is a reasonable default behaviour?
		 * Closing the channel may be the right thing to do. */
		if (! (channel->flags & CHANNEL_SENT_CLOSE))
		  channel_close(channel);
Niels Möller's avatar
Niels Möller committed
1188
1189
	    }
	}
Niels Möller's avatar
Niels Möller committed
1190
1191
1192
1193
      else
	{
	  werror("EOF on non-existant channel %i\n",
		 channel_number);
1194
	  PROTOCOL_ERROR(connection->e, "EOF on non-existant channel");
Niels Möller's avatar
Niels Möller committed
1195
1196
1197
1198
1199
	}
    }
  else
    {
      lsh_string_free(packet);
1200
      PROTOCOL_ERROR(connection->e, "Invalid CHANNEL_EOF message");
Niels Möller's avatar
Niels Möller committed
1201
1202
1203
    }
}

Niels Möller's avatar
Niels Möller committed
1204
1205
1206
1207
static void
do_channel_close(struct packet_handler *closure UNUSED,
		 struct ssh_connection *connection,
		 struct lsh_string *packet)
Niels Möller's avatar
Niels Möller committed
1208
1209
{
  struct simple_buffer buffer;
1210
  unsigned msg_number;
Niels Möller's avatar
Niels Möller committed
1211
1212
1213
1214
1215
1216
  UINT32 channel_number;
  
  simple_buffer_init(&buffer, packet->length, packet->data);

  if (parse_uint8(&buffer, &msg_number)
      && (msg_number == SSH_MSG_CHANNEL_CLOSE)
1217
      && parse_uint32(&buffer, &channel_number)
Niels Möller's avatar
Niels Möller committed
1218
1219
      && parse_eod(&buffer))
    {
Niels Möller's avatar
Niels Möller committed
1220
      struct ssh_channel *channel = lookup_channel(connection->table,
Niels Möller's avatar
Niels Möller committed
1221
1222
1223
1224
						   channel_number);

      lsh_string_free(packet);
      
Niels Möller's avatar
Niels Möller committed
1225
1226
      if (channel)
	{
1227
	  if (channel->flags & CHANNEL_RECEIVED_CLOSE)
Niels Möller's avatar
Niels Möller committed
1228
	    {
1229
	      werror("Receiving multiple CLOSE on channel.\n");
1230
	      PROTOCOL_ERROR(connection->e, "Receiving multiple CLOSE on channel.");
Niels Möller's avatar
Niels Möller committed
1231
	    }
Niels Möller's avatar
Niels Möller committed
1232
	  else
Niels Möller's avatar
Niels Möller committed
1233
	    {