channel.c 47.1 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
328
329

  return i;
}

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

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

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

363
364
365
366
367
368
369
370
371
372
/* 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)
373
{
374
375
  CAST_SUBTYPE(ssh_channel, channel, f);

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

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

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

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

434
435
436
437
438
439
440
  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
441
442
  return status;
}
Niels Möller's avatar
Niels Möller committed
443
444
#endif

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

446
447
448
/* 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
449
450
451
452
453
454
455
456
#define RETURN goto foo_finish
#define END(s) do { foo_finish: \
                    lsh_string_free((s)); \
                    return; } while(0)

#define ERROR(e, msg) \
  EXCEPTION_RAISE(e, make_protocol_exception \
    (SSH_DISCONNECT_PROTOCOL_ERROR, msg))
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
510
511
512
513
514

      /* FIXME: Perhaps install some exception handler that cancels
       * the queue as soon as a write failes. */
      A_WRITE(self->super.connection->write,
	      (n->status
	       ? format_global_success()
	       : format_global_failure()),
	      self->super.connection->e);
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
    }
}

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
532
533
534
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
535
{
536
  /* CAST(global_request_handler, closure, c); */
Niels Möller's avatar
Niels Möller committed
537
538

  struct simple_buffer buffer;
539
  unsigned msg_number;
Niels Möller's avatar
Niels Möller committed
540
541
542
543
544
545
546
547
548
549
550
  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;
551
552
      struct global_request_callback *c = NULL;
      
Niels Möller's avatar
Niels Möller committed
553
      if (!name || !(req = ALIST_GET(connection->table->global_requests,
554
				     name)))
555
	{
Niels Möller's avatar
Niels Möller committed
556
	  lsh_string_free(packet);
557

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

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

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

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

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

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

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

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

Niels Möller's avatar
Niels Möller committed
637
638
#if 0
/* HERE!!! */
639
/* FIXME: Split into a continuation and an exception handler */
640
641
642
643
644
645
646
647
648
649
650
651
652
/* 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
653
654
655
656
657
658
        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())
659
      return LSH_FAIL | LSH_DIE;
660
661
    }

Niels Möller's avatar
Niels Möller committed
662
663
664
  /* FIXME: It would be better to allocate or at least reserve a channel number earlier,
   * so that we can't fail at this point.
   */
665
666
667
668
669
670
671
672
673
674
675
676
  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...)", ""));
    }

677
  /* FIXME: This copying could just as well be done by the
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
710
   * 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
711
#endif
712

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

Niels Möller's avatar
Niels Möller committed
725
726
727
static void
do_channel_open_continue(struct command_continuation *c,
			 struct lsh_object *value)
728
729
{
  CAST(channel_open_continuation, self, c);
Niels Möller's avatar
Niels Möller committed
730
731
732
733
734
735
736
737
738
739
740
741
  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
742
  channel->write = self->connection->write;
Niels Möller's avatar
Niels Möller committed
743

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

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

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

Niels Möller's avatar
Niels Möller committed
754
static struct command_continuation *
Niels Möller's avatar
Niels Möller committed
755
756
757
758
759
760
761
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
762

Niels Möller's avatar
Niels Möller committed
763
764
765
766
767
768
769
  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
770
  return &self->super;
Niels Möller's avatar
Niels Möller committed
771
772
773
774
775
776
777
778
779
}
			       
/* 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
780
       (remote_channel_number . UINT32)))
Niels Möller's avatar
Niels Möller committed
781
782
783
*/

static void do_exc_channel_open_handler(struct exception_handler *s,
Niels Möller's avatar
Niels Möller committed
784
					const struct exception *e)
Niels Möller's avatar
Niels Möller committed
785
786
787
788
789
790
791
792
{
  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
793
	struct channel_table *table = self->connection->table;
Niels Möller's avatar
Niels Möller committed
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
	
	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
816
  NEW(exc_channel_open_handler, self);
Niels Möller's avatar
Niels Möller committed
817
  self->super.parent = parent;
Niels Möller's avatar
Niels Möller committed
818
819
  self->super.raise = do_exc_channel_open_handler;
  self->connection = connection;
Niels Möller's avatar
Niels Möller committed
820
821
822
823
824
825
826
827
828
  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
829
{
830
  /* CAST(channel_open_handler, closure, c); */
Niels Möller's avatar
Niels Möller committed
831
832

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

  if (parse_uint8(&buffer, &msg_number)
      && (msg_number == SSH_MSG_CHANNEL_OPEN)
      && parse_atom(&buffer, &type)
844
845
846
      && 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
847
848
    {
      struct channel_open *open;
Niels Möller's avatar
Niels Möller committed
849

Niels Möller's avatar
Niels Möller committed
850
851
852
853
      /* 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
854
855
856
857
858
859
860
861
	{
	  /* 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
862
      else if (!type || !(open = ALIST_GET(connection->table->channel_types,
863
				      type)))
Niels Möller's avatar
Niels Möller committed
864
865
866
867
868
869
870
871
	{
	  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
872
      	  int local_number = alloc_channel(connection->table);
Niels Möller's avatar
Niels Möller committed
873
874
875
876
877
878
879
880

	  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
881
882
883
884
885
886
887
888
889
890
891
	  
	  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
892
	}
Niels Möller's avatar
Niels Möller committed
893
    }
Niels Möller's avatar
Niels Möller committed
894
  else
Niels Möller's avatar
Niels Möller committed
895
896
    ERROR(connection->e, "Invalid SSH_MSG_CHANNEL_OPEN message.");

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

Niels Möller's avatar
Niels Möller committed
900
901
902
903
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
904
905
{
  struct simple_buffer buffer;
906
  unsigned msg_number;
Niels Möller's avatar
Niels Möller committed
907
908
909
  UINT32 channel_number;
  int type;
  int want_reply;
910
  
Niels Möller's avatar
Niels Möller committed
911
912
913
914
915
916
917
918
  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
919
      struct ssh_channel *channel = lookup_channel(connection->table,
Niels Möller's avatar
Niels Möller committed
920
921
						   channel_number);

922
923
924
925
      /* 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
926
927
928
929
      if (channel)
	{
	  struct channel_request *req;

930
931
	  if (type && channel->request_types 
	      && ( (req = ALIST_GET(channel->request_types, type)) ))
Niels Möller's avatar
Niels Möller committed
932
	    CHANNEL_REQUEST(req, channel, connection, want_reply, &buffer);
933
	  else
Niels Möller's avatar
Niels Möller committed
934
935
	    {
	      if (want_reply)
Niels Möller's avatar
Niels Möller committed
936
		C_WRITE(connection,
Niels Möller's avatar
Niels Möller committed
937
938
939
940
941
942
943
			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
944
	}
945
    }
Niels Möller's avatar
Niels Möller committed
946
  else
Niels Möller's avatar
Niels Möller committed
947
948
    ERROR(connection->e, "Invalid SSH_MSG_CHANNEL_REQUEST message.");
  
Niels Möller's avatar
Niels Möller committed
949
  lsh_string_free(packet);
Niels Möller's avatar
Niels Möller committed
950
951
}
      
Niels Möller's avatar
Niels Möller committed
952
953
954
955
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
956
957
{
  struct simple_buffer buffer;
958
  unsigned msg_number;
Niels Möller's avatar
Niels Möller committed
959
960
961
962
963
964
965
  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)
966
      && parse_uint32(&buffer, &channel_number)
Niels Möller's avatar
Niels Möller committed
967
968
969
      && parse_uint32(&buffer, &size)
      && parse_eod(&buffer))
    {
Niels Möller's avatar
Niels Möller committed
970
      struct ssh_channel *channel = lookup_channel(connection->table,
Niels Möller's avatar
Niels Möller committed
971
972
973
974
						   channel_number);

      lsh_string_free(packet);
      
Niels Möller's avatar
Niels Möller committed
975
      if (channel
976
977
	  && !(channel->flags & (CHANNEL_RECEIVED_EOF
				 | CHANNEL_RECEIVED_CLOSE)))
Niels Möller's avatar
Niels Möller committed
978
	{
Niels Möller's avatar
Niels Möller committed
979
980
981
982
	  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
983
		CHANNEL_SEND(channel);
Niels Möller's avatar
Niels Möller committed
984
	    }
Niels Möller's avatar
Niels Möller committed
985
	}
Niels Möller's avatar
Niels Möller committed
986
987
988
989
990
991
992
993
994
995
996
997
998
      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);
	  ERROR(connection->e, "Unexpected CHANNEL_WINDOW_ADJUST");
	}
    }
  else
    {
      lsh_string_free(packet);
      ERROR(connection->e, "Invalid CHANNEL_WINDOW_ADJUST message.");
Niels Möller's avatar
Niels Möller committed
999
1000
1001
    }
}

Niels Möller's avatar
Niels Möller committed
1002
1003
1004
1005
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
1006
1007
{
  struct simple_buffer buffer;
1008
  unsigned msg_number;
Niels Möller's avatar
Niels Möller committed
1009
1010
1011
1012
1013
1014
1015
  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)
1016
      && parse_uint32(&buffer, &channel_number)
Niels Möller's avatar
Niels Möller committed
1017
1018
1019
      && ( (data = parse_string_copy(&buffer)) )
      && parse_eod(&buffer))
    {
Niels Möller's avatar
Niels Möller committed
1020
      struct ssh_channel *channel = lookup_channel(connection->table,
Niels Möller's avatar
Niels Möller committed
1021
1022
1023
1024
						   channel_number);

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

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

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

Niels Möller's avatar
Niels Möller committed
1069
1070
1071
1072
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
1073
1074
{
  struct simple_buffer buffer;
1075
  unsigned msg_number;
Niels Möller's avatar
Niels Möller committed
1076
1077
1078
1079
1080
1081
1082
1083
  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)
1084
      && parse_uint32(&buffer, &channel_number)
Niels Möller's avatar
Niels Möller committed
1085
1086
1087
1088
      && parse_uint32(&buffer, &type)
      && ( (data = parse_string_copy(&buffer)) )
      && parse_eod(&buffer))
    {
Niels Möller's avatar
Niels Möller committed
1089
      struct ssh_channel *channel = lookup_channel(connection->table,
Niels Möller's avatar
Niels Möller committed
1090
1091
1092
1093
						   channel_number);

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

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

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

Niels Möller's avatar
Niels Möller committed
1149
1150
1151
1152
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
1153
1154
{
  struct simple_buffer buffer;
1155
  unsigned msg_number;
Niels Möller's avatar
Niels Möller committed
1156
1157
1158
1159
1160
1161
  UINT32 channel_number;
  
  simple_buffer_init(&buffer, packet->length, packet->data);

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

      lsh_string_free(packet);

Niels Möller's avatar
Niels Möller committed
1170
1171
      if (channel)
	{
1172
	  if (channel->flags & (CHANNEL_RECEIVED_EOF | CHANNEL_RECEIVED_CLOSE))
Niels Möller's avatar
Niels Möller committed
1173
	    {
1174
	      werror("Receiving EOF on channel on closed channel.\n");
Niels Möller's avatar
Niels Möller committed
1175
1176
1177
1178
	      EXCEPTION_RAISE
		(connection->e,
		 make_protocol_exception(SSH_DISCONNECT_PROTOCOL_ERROR,
					 "Received EOF on channel on closed channel."));
Niels Möller's avatar
Niels Möller committed
1179
	    }
1180
	  else
Niels Möller's avatar
Niels Möller committed
1181
	    {
Niels Möller's avatar
Niels Möller committed
1182
	      channel->flags |= CHANNEL_RECEIVED_EOF;
1183
	      
Niels Möller's avatar
Niels Möller committed
1184
1185
1186
1187
1188
1189
1190
	      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
1191
1192
	    }
	}
Niels Möller's avatar
Niels Möller committed
1193
1194
1195
1196
1197
1198
1199
1200
1201
1202
1203
      else
	{
	  werror("EOF on non-existant channel %i\n",
		 channel_number);
	  ERROR(connection->e, "EOF on non-existant channel");
	}
    }
  else
    {
      lsh_string_free(packet);
      ERROR(connection->e, "Invalid CHANNEL_EOF message");
Niels Möller's avatar
Niels Möller committed
1204
1205
1206
    }
}

Niels Möller's avatar
Niels Möller committed
1207
1208
1209
1210
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
1211
1212
{
  struct simple_buffer buffer;
1213
  unsigned msg_number;
Niels Möller's avatar
Niels Möller committed
1214
1215
1216
1217
1218
1219
  UINT32 channel_number;
  
  simple_buffer_init(&buffer, packet->length, packet->data);

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

      lsh_string_free(packet);
      
Niels Möller's avatar
Niels Möller committed
1228
1229
      if (channel)
	{
1230
	  if (channel->flags & CHANNEL_RECEIVED_CLOSE)
Niels Möller's avatar
Niels Möller committed
1231
	    {
1232
	      werror("Receiving multiple CLOSE on channel.\n");
Niels Möller's avatar
Niels Möller committed
1233
	      ERROR(connection->e, "Receiving multiple CLOSE on channel.");
Niels Möller's avatar
Niels Möller committed
1234
	    }
Niels Möller's avatar
Niels Möller committed
1235
	  else
Niels Möller's avatar
Niels Möller committed
1236
	    {
Niels Möller's avatar
Niels Möller committed
1237
	      channel->flags |= CHANNEL_RECEIVED_CLOSE;
Niels Möller's avatar
Niels Möller committed
1238
	  
Niels Möller's avatar
Niels Möller committed
1239
1240
1241
1242
1243
1244
1245
1246
1247
1248
1249
1250
1251
1252
1253
1254
1255
1256
1257
1258
1259
1260
1261
1262
1263
1264
1265
	      if (! (channel->flags & (CHANNEL_RECEIVED_EOF | CHANNEL_SENT_EOF
				       | CHANNEL_SENT_CLOSE)))
		{
		  werror("Unexpected channel CLOSE.\n");
		}

	      if (! (channel->flags & (CHANNEL_RECEIVED_EOF))
		  && channel->eof)
		CHANNEL_EOF(channel);

	      if (channel->flags & (CHANNEL_SENT_CLOSE))
		{
		  static const struct exception finish_exception
		    = STATIC_EXCEPTION(EXC_FINISH_CHANNEL, "Received CLOSE message.");
	      
		  EXCEPTION_RAISE(channel->e,
				  &finish_exception);
		}
	      else
		channel_close(channel);
	    }
	}
      else
	{
	  werror("CLOSE on non-existant channel %i\n",
		 channel_number);
	  ERROR(connection->e, "CLOSE on non-existant channel");
Niels Möller's avatar
Niels Möller committed
1266
	}
Niels Möller's avatar
Niels Möller committed
1267
    }
Niels Möller's avatar
Niels Möller committed
1268
1269
1270
1271
1272
  else
    {
      lsh_string_free(packet);
      ERROR(connection->e, "Invalid CHANNEL_CLOSE message");
    }
Niels Möller's avatar
Niels Möller committed
1273
1274
}

Niels Möller's avatar
Niels Möller committed
1275
1276
1277
1278
static void
do_channel_open_confirm(struct packet_handler *closure UNUSED,
			struct ssh_connection *connection,
			struct lsh_string *packet)
Niels Möller's avatar
Niels Möller committed
1279
1280
{
  struct simple_buffer buffer;
1281
  unsigned msg_number;
Niels Möller's avatar
Niels Möller committed
1282
1283
1284
1285
1286
1287
1288
1289
1290
1291
1292
1293
1294
1295
1296
  UINT32 local_channel_number;
  UINT32 remote_channel_number;  
  UINT32 window_size;
  UINT32 max_packet;
  
  simple_buffer_init(&buffer, packet->length, packet->data);

  if (parse_uint8(&buffer, &msg_number)
      && (msg_number == SSH_MSG_CHANNEL_OPEN_CONFIRMATION