channel.c 31.4 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
22
23
24
25
26
27
/* 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
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */

#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
35
36
#include "service.h"
#include "ssh.h"
#include "werror.h"
#include "xalloc.h"

#include <assert.h>

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

41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
#include "channel.c.x"

/* CLASS:
   (class
     (name connection_service)
     (super ssh_service)
     (vars
       ; Supported global requests 
       (global_requests object alist)

       (channel_types object alist)

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

       (start object connection_startup)))
*/

/* CLASS:
   (class
     (name channel_handler)
     (super packet_handler)
     (vars
       (table object channel_table)))
*/

/* CLASS:
   (class
     (name global_request_handler)
     (super channel_handler super)
     (vars
       (global_requests object alist)))
*/

/* CLASS:
   (class
     (name channel_open_handler)
     (super channel_handler)
     (vars
       (channel_types object alist)))
*/

Niels Möller's avatar
Niels Möller committed
83
84
85
86
87
struct lsh_string *format_global_failure(void)
{
  return ssh_format("%c", SSH_MSG_REQUEST_FAILURE);
}

88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
struct lsh_string *format_open_confirmation(struct ssh_channel *channel,
					    UINT32 channel_number,
					    char *format, ...)
{
  va_list args;
  UINT32 l1, l2;
  struct lsh_string *packet;

#define CONFIRM_FORMAT "%c%i%i%i%i"
#define CONFIRM_ARGS SSH_MSG_CHANNEL_OPEN_CONFIRMATION, channel->channel_number, \
  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);
111
  ssh_vformat_write(format, l2, packet->data+l1, args);
112
113
114
115
116
117
118
  va_end(args);

  return packet;
#undef CONFIRM_FORMAT
#undef CONFIRM_ARGS
}

Niels Möller's avatar
Niels Möller committed
119
120
121
122
123
124
125
struct lsh_string *format_open_failure(UINT32 channel, UINT32 reason,
				       char *msg, char *language)
{
  return ssh_format("%c%i%i%z%z", SSH_MSG_CHANNEL_OPEN_FAILURE,
		    channel, reason, msg, language);
}

126
127
128
129
130
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
131
132
133
134
135
struct lsh_string *format_channel_failure(UINT32 channel)
{
  return ssh_format("%c%i", SSH_MSG_CHANNEL_FAILURE, channel);
}

136
137
138
139
140
141
142
143
144
145
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);
}

146
/* Channel objects */
Niels Möller's avatar
Niels Möller committed
147
148
149
150
151

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

Niels Möller's avatar
Niels Möller committed
152
struct channel_table *make_channel_table(void)
Niels Möller's avatar
Niels Möller committed
153
{
154
  NEW(channel_table, table);
Niels Möller's avatar
Niels Möller committed
155

156
  table->channels = lsh_space_alloc(sizeof(struct ssh_channel *)
157
				      * INITIAL_CHANNELS);
158
159
160
161
  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
162

Niels Möller's avatar
Niels Möller committed
163
164
  table->pending_close = 0;
  
165
  return table;
Niels Möller's avatar
Niels Möller committed
166
167
168
};

/* Returns -1 if allocation fails */
169
int alloc_channel(struct channel_table *table)
Niels Möller's avatar
Niels Möller committed
170
171
{
  int i;
172
  for(i = table->next_channel; i < table->used_channels; i++)
Niels Möller's avatar
Niels Möller committed
173
    {
174
      if (!table->channels[i])
Niels Möller's avatar
Niels Möller committed
175
	{
176
	  table->next_channel = i+1;
Niels Möller's avatar
Niels Möller committed
177
178
179
	  return i;
	}
    }
180
  if (i == table->max_channels)
Niels Möller's avatar
Niels Möller committed
181
    return -1;
182
  if (i == table->allocated_channels) 
Niels Möller's avatar
Niels Möller committed
183
    {
184
      int new_size = table->allocated_channels * 2;
Niels Möller's avatar
Niels Möller committed
185
      struct ssh_channel **new
186
	= lsh_space_alloc(sizeof(struct ssh_channel *) * new_size);
Niels Möller's avatar
Niels Möller committed
187

188
189
      memcpy(new, table->channels,
	     sizeof(struct ssh_channel *) * table->used_channels);
Niels Möller's avatar
Niels Möller committed
190
      
191
192
      table->channels = new;
      table->allocated_channels = new_size;
Niels Möller's avatar
Niels Möller committed
193
194
    }

195
  table->next_channel = table->used_channels = i+1;
Niels Möller's avatar
Niels Möller committed
196
197
198
199

  return i;
}

200
void dealloc_channel(struct channel_table *table, int i)
Niels Möller's avatar
Niels Möller committed
201
202
{
  assert(i >= 0);
203
  assert(i < table->used_channels);
Niels Möller's avatar
Niels Möller committed
204
  
205
206
207
208
209
210
211
212
213
214
215
216
  table->channels[i] = NULL;

  if (i < table->next_channel)
    table->next_channel = i;
}

int register_channel(struct channel_table *table, struct ssh_channel *channel)
{
  int n = alloc_channel(table);

  if (n >= 0)
    table->channels[n] = channel;
Niels Möller's avatar
Niels Möller committed
217

218
  return n;
Niels Möller's avatar
Niels Möller committed
219
220
}

221
struct ssh_channel *lookup_channel(struct channel_table *table, UINT32 i)
Niels Möller's avatar
Niels Möller committed
222
{
223
224
  return (i < table->used_channels)
    ? table->channels[i] : NULL;
Niels Möller's avatar
Niels Möller committed
225
226
}

Niels Möller's avatar
Niels Möller committed
227
228
229
230
231
232
233
234
235
236
237
static int channel_process_status(struct channel_table *table,
				  int channel,
				  int status)
{
  if (status & LSH_CHANNEL_PENDING_CLOSE)
    table->pending_close = 1;
  
  if (status & LSH_CHANNEL_FINISHED)
    {
      /* Clear this bit */
      status &= ~LSH_CHANNEL_FINISHED;
238
239
240

      if (table->channels[channel]->close)
	CHANNEL_CLOSE(table->channels[channel]);
Niels Möller's avatar
Niels Möller committed
241
242
243
244
245
246
247
248
249
250
      
      dealloc_channel(table, channel);

      /* If this was the last channel, close connection */
      if (table->pending_close && !table->next_channel)
	status |= LSH_CLOSE;
    }

  return status;
}
Niels Möller's avatar
Niels Möller committed
251

Niels Möller's avatar
Niels Möller committed
252
/* Channel related messages */
Niels Möller's avatar
Niels Möller committed
253
254
255
256
static int do_global_request(struct packet_handler *c,
			     struct ssh_connection *connection,
			     struct lsh_string *packet)
{
257
  CAST(global_request_handler, closure, c);
Niels Möller's avatar
Niels Möller committed
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289

  struct simple_buffer buffer;
  int msg_number;
  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;

      lsh_string_free(packet);
      
      if (!name || !(req = ALIST_GET(closure->global_requests, name)))
	return A_WRITE(connection->write,
		       format_global_failure());

      return GLOBAL_REQUEST(req, want_reply, &buffer);
    }
  lsh_string_free(packet);

  return LSH_FAIL | LSH_DIE;
}

static int do_channel_open(struct packet_handler *c,
			   struct ssh_connection *connection,
			   struct lsh_string *packet)
{
290
  CAST(channel_open_handler, closure, c);
Niels Möller's avatar
Niels Möller committed
291
292
293
294

  struct simple_buffer buffer;
  int msg_number;
  int type;
295
296
297
  UINT32 remote_channel_number;
  UINT32 window_size;
  UINT32 max_packet;
Niels Möller's avatar
Niels Möller committed
298
299
300
301
302
303
  
  simple_buffer_init(&buffer, packet->length, packet->data);

  if (parse_uint8(&buffer, &msg_number)
      && (msg_number == SSH_MSG_CHANNEL_OPEN)
      && parse_atom(&buffer, &type)
304
305
306
      && 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
307
308
    {
      struct channel_open *open;
309
310
311
312
313
314
315
      struct ssh_channel *channel;
      UINT32 error = 0;
      char *error_msg;
      struct lsh_string *args = NULL;
      
      int local_channel_number;
      
Niels Möller's avatar
Niels Möller committed
316
      lsh_string_free(packet);
Niels Möller's avatar
Niels Möller committed
317
318
319
320
321
322
323

      if (closure->super.table->pending_close)
	/* We are waiting for channels to close. Don't open any new ones. */
	return A_WRITE(connection->write,
		       format_open_failure(remote_channel_number,
					   SSH_OPEN_ADMINISTRATIVELY_PROHIBITED,
					   "Waiting for channels to close.", ""));
Niels Möller's avatar
Niels Möller committed
324
325
326
      
      if (!type || !(open = ALIST_GET(closure->channel_types, type)))
	return A_WRITE(connection->write,
327
		       format_open_failure(remote_channel_number,
Niels Möller's avatar
Niels Möller committed
328
329
					   SSH_OPEN_UNKNOWN_CHANNEL_TYPE,
					   "Unknown channel type", ""));
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
      
      channel = CHANNEL_OPEN(open, &buffer, &error, &error_msg, &args);
      
      if (!channel)
	{
	  if (error)
	    return A_WRITE(connection->write,
			   format_open_failure(remote_channel_number,
					       error, error_msg, ""));
	  /* The request was invalid */
	  return LSH_FAIL | LSH_DIE;
	}

      if ( (local_channel_number
	    = register_channel(closure->super.table, channel)) < 0)
	{
	  werror("Could not allocate a channel number for pened channel!\n");
	  return A_WRITE(connection->write,
			 format_open_failure(remote_channel_number,
					     SSH_OPEN_RESOURCE_SHORTAGE,
					     "Could not allocate a channel number "
					     "(shouldn't happen...)", ""));
	}
      
      channel->send_window_size = window_size;
      channel->send_max_packet = max_packet;
      channel->channel_number = remote_channel_number;

      channel->write = connection->write;

      return A_WRITE(connection->write,
		     args
		     ? format_open_confirmation(channel, local_channel_number,
						"%lfS", args)
Niels Möller's avatar
Niels Möller committed
364
365
		     : format_open_confirmation(channel, local_channel_number,
						""));
Niels Möller's avatar
Niels Möller committed
366
367
368
369
370
371
372
373
374
375
    }
  lsh_string_free(packet);

  return LSH_FAIL | LSH_DIE;
}     

static int do_channel_request(struct packet_handler *c,
			      struct ssh_connection *connection,
			      struct lsh_string *packet)
{
376
  CAST(channel_handler, closure, c);
Niels Möller's avatar
Niels Möller committed
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391

  struct simple_buffer buffer;
  int msg_number;
  UINT32 channel_number;
  int type;
  int want_reply;

  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))
    {
392
      struct ssh_channel *channel = lookup_channel(closure->table,
Niels Möller's avatar
Niels Möller committed
393
394
395
396
397
398
399
400
						   channel_number);

      lsh_string_free(packet);
      
      if (channel)
	{
	  struct channel_request *req;

401
402
	  if (type && channel->request_types 
	      && ( (req = ALIST_GET(channel->request_types, type)) ))
Niels Möller's avatar
Niels Möller committed
403
404
405
	    return channel_process_status
	      (closure->table, channel_number,
	       CHANNEL_REQUEST(req, channel, want_reply, &buffer));
406
	  else
407
408
409
410
411
	    return want_reply
	      ? A_WRITE(connection->write,
			format_channel_failure(channel->channel_number))
	      : LSH_OK | LSH_GOON;
	  
Niels Möller's avatar
Niels Möller committed
412
413
414
415
416
417
418
419
420
421
422
423
	}
      werror("SSH_MSG_CHANNEL_REQUEST on nonexistant channel %d\n",
	     channel_number);
      return LSH_FAIL | LSH_DIE;
    }
  lsh_string_free(packet);

  return LSH_FAIL | LSH_DIE;
}
      
static int do_window_adjust(struct packet_handler *c,
			    struct ssh_connection *connection,
424
			    struct lsh_string *packet)
Niels Möller's avatar
Niels Möller committed
425
{
426
  CAST(channel_handler, closure, c);
Niels Möller's avatar
Niels Möller committed
427
428
429
430
431
432
433
434
435
436

  struct simple_buffer buffer;
  int msg_number;
  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)
437
      && parse_uint32(&buffer, &channel_number)
Niels Möller's avatar
Niels Möller committed
438
439
440
      && parse_uint32(&buffer, &size)
      && parse_eod(&buffer))
    {
441
      struct ssh_channel *channel = lookup_channel(closure->table,
Niels Möller's avatar
Niels Möller committed
442
443
444
445
						   channel_number);

      lsh_string_free(packet);
      
Niels Möller's avatar
Niels Möller committed
446
447
      if (channel
	  && !(channel->flags & (CHANNEL_RECIEVED_EOF | CHANNEL_RECIEVED_CLOSE)))
Niels Möller's avatar
Niels Möller committed
448
	{
Niels Möller's avatar
Niels Möller committed
449
450
451
452
	  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
453
454
455
		return channel_process_status(closure->table,
					      channel_number,
					      CHANNEL_SEND(channel));
Niels Möller's avatar
Niels Möller committed
456
	    }
Niels Möller's avatar
Niels Möller committed
457
458
459
460
	  return LSH_OK | LSH_GOON;
	}
      /* FIXME: What to do now? Should unknown channel numbers be
       * ignored silently? */
Niels Möller's avatar
Niels Möller committed
461
      werror("SSH_MSG_CHANNEL_WINDOW_ADJUST on nonexistant or closed channel %d\n",
Niels Möller's avatar
Niels Möller committed
462
463
464
465
466
467
468
469
470
471
472
473
	     channel_number);
      return LSH_FAIL | LSH_DIE;
    }
  lsh_string_free(packet);

  return LSH_FAIL | LSH_DIE;
}

static int do_channel_data(struct packet_handler *c,
			   struct ssh_connection *connection,
			   struct lsh_string *packet)
{
474
  CAST(channel_handler, closure, c);
Niels Möller's avatar
Niels Möller committed
475
476
477
478
479
480
481
482
483
484

  struct simple_buffer buffer;
  int msg_number;
  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)
485
      && parse_uint32(&buffer, &channel_number)
Niels Möller's avatar
Niels Möller committed
486
487
488
      && ( (data = parse_string_copy(&buffer)) )
      && parse_eod(&buffer))
    {
489
      struct ssh_channel *channel = lookup_channel(closure->table,
Niels Möller's avatar
Niels Möller committed
490
491
492
493
						   channel_number);

      lsh_string_free(packet);
      
Niels Möller's avatar
Niels Möller committed
494
495
      if (channel && channel->recieve
	  && !(channel->flags & (CHANNEL_RECIEVED_EOF | CHANNEL_RECIEVED_CLOSE)))
Niels Möller's avatar
Niels Möller committed
496
	{
Niels Möller's avatar
Niels Möller committed
497
	  if (channel->flags & CHANNEL_SENT_CLOSE)
Niels Möller's avatar
Niels Möller committed
498
	    {
Niels Möller's avatar
Niels Möller committed
499
500
	      werror("Ignoring data on channel which is closing\n");
	      return LSH_OK | LSH_GOON;
Niels Möller's avatar
Niels Möller committed
501
	    }
Niels Möller's avatar
Niels Möller committed
502
503
	  else
	    {
504
505
	      int res = 0;
	      
Niels Möller's avatar
Niels Möller committed
506
507
508
509
510
511
	      if (data->length > channel->rec_window_size)
		{
		  /* Truncate data to fit window */
		  werror("Channel data overflow. Extra data ignored.\n"); 
		  data->length = channel->rec_window_size;
		}
512
513
514
515
516
517
518
519
520
521
522
	      channel->rec_window_size -= data->length;

	      if (channel->rec_window_size < channel->max_window / 2)
		{
		  res = A_WRITE(channel->write, prepare_window_adjust
				(channel,
				 channel->max_window - channel->rec_window_size));
		  if (LSH_CLOSEDP(res))
		    return res;
		}

Niels Möller's avatar
Niels Möller committed
523
524
525
526
	      return channel_process_status(
		closure->table, channel_number,
		res | CHANNEL_RECIEVE(channel, 
				      CHANNEL_DATA, data));
Niels Möller's avatar
Niels Möller committed
527
528
	    }
	  return LSH_OK | LSH_GOON;
Niels Möller's avatar
Niels Möller committed
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
	}
	  
      werror("Data on closed or non-existant channel %d\n",
	     channel_number);
      lsh_string_free(data);
      return LSH_FAIL | LSH_DIE;
    }
  lsh_string_free(packet);
  
  return LSH_FAIL | LSH_DIE;
}

static int do_channel_extended_data(struct packet_handler *c,
				    struct ssh_connection *connection,
				    struct lsh_string *packet)
{
545
  CAST(channel_handler, closure, c);
Niels Möller's avatar
Niels Möller committed
546
547
548
549
550
551
552
553
554
555
556

  struct simple_buffer buffer;
  int msg_number;
  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)
557
      && parse_uint32(&buffer, &channel_number)
Niels Möller's avatar
Niels Möller committed
558
559
560
561
      && parse_uint32(&buffer, &type)
      && ( (data = parse_string_copy(&buffer)) )
      && parse_eod(&buffer))
    {
562
      struct ssh_channel *channel = lookup_channel(closure->table,
Niels Möller's avatar
Niels Möller committed
563
564
565
566
						   channel_number);

      lsh_string_free(packet);
      
Niels Möller's avatar
Niels Möller committed
567
568
      if (channel && channel->recieve
	  && !(channel->flags & (CHANNEL_RECIEVED_EOF | CHANNEL_RECIEVED_CLOSE)))
Niels Möller's avatar
Niels Möller committed
569
	{
Niels Möller's avatar
Niels Möller committed
570
	  if (channel->flags & CHANNEL_SENT_CLOSE)
Niels Möller's avatar
Niels Möller committed
571
	    {
Niels Möller's avatar
Niels Möller committed
572
573
	      werror("Ignoring extended data on channel which is closing\n");
	      return LSH_OK | LSH_GOON;
Niels Möller's avatar
Niels Möller committed
574
	    }
Niels Möller's avatar
Niels Möller committed
575
	  else
Niels Möller's avatar
Niels Möller committed
576
	    {
577
578
	      int res = 0;
	      
Niels Möller's avatar
Niels Möller committed
579
580
581
582
583
584
585
586
587
	      if (data->length > channel->rec_window_size)
		{
		  /* Truncate data to fit window */
		  werror("Channel extended data overflow. "
			 "Extra data ignored.\n");
		  data->length = channel->rec_window_size;
		}
	      
	      channel->rec_window_size -= data->length;
588
589
590
591
592
593
594
595
596
597

	      if (channel->rec_window_size < channel->max_window / 2)
		{
		  res = A_WRITE(channel->write, prepare_window_adjust
				(channel,
				 channel->max_window - channel->rec_window_size));
		  if (LSH_CLOSEDP(res))
		    return res;
		}

Niels Möller's avatar
Niels Möller committed
598
599
600
	      switch(type)
		{
		case SSH_EXTENDED_DATA_STDERR:
Niels Möller's avatar
Niels Möller committed
601
602
603
604
		  return channel_process_status(
		    closure->table, channel_number,
		    res | CHANNEL_RECIEVE(channel, 
					  CHANNEL_STDERR_DATA, data));
Niels Möller's avatar
Niels Möller committed
605
606
607
608
609
610
		default:
		  werror("Unknown type %d of extended data.\n",
			 type);
		  lsh_string_free(data);
		  return LSH_FAIL | LSH_DIE;
		}
Niels Möller's avatar
Niels Möller committed
611
612
613
614
615
616
617
618
619
620
621
622
623
	    }
	}
      werror("Extended data on closed or non-existant channel %d\n",
	     channel_number);
      lsh_string_free(data);
      return LSH_FAIL | LSH_DIE;
    }
  lsh_string_free(packet);
  
  return LSH_FAIL | LSH_DIE;
}

static int do_channel_eof(struct packet_handler *c,
624
625
			  struct ssh_connection *connection,
			  struct lsh_string *packet)
Niels Möller's avatar
Niels Möller committed
626
{
627
  CAST(channel_handler, closure, c);
Niels Möller's avatar
Niels Möller committed
628
629
630
631
632
633
634
635
636

  struct simple_buffer buffer;
  int msg_number;
  UINT32 channel_number;
  
  simple_buffer_init(&buffer, packet->length, packet->data);

  if (parse_uint8(&buffer, &msg_number)
      && (msg_number == SSH_MSG_CHANNEL_EOF)
637
      && parse_uint32(&buffer, &channel_number)
Niels Möller's avatar
Niels Möller committed
638
639
      && parse_eod(&buffer))
    {
640
      struct ssh_channel *channel = lookup_channel(closure->table,
Niels Möller's avatar
Niels Möller committed
641
642
643
644
						   channel_number);

      lsh_string_free(packet);

Niels Möller's avatar
Niels Möller committed
645
646
      if (channel)
	{
647
648
	  int res = 0;
	  
Niels Möller's avatar
Niels Möller committed
649
650
651
652
653
654
655
656
	  if (channel->flags & (CHANNEL_RECIEVED_EOF | CHANNEL_RECIEVED_CLOSE))
	    {
	      werror("Recieving EOF on channel on closed channel.\n");
	      return LSH_FAIL | LSH_DIE;
	    }

	  channel->flags |= CHANNEL_RECIEVED_EOF;

657
658
659
660
661
662
	  if (channel->eof)
	    res = CHANNEL_EOF(channel);

	  if (!LSH_CLOSEDP(res)
	      && ! (channel->flags & CHANNEL_SENT_CLOSE)
	      && (channel->flags & CHANNEL_SENT_EOF))
Niels Möller's avatar
Niels Möller committed
663
664
665
	    {
	      /* Both parties have sent EOF. Initiate close, if we
	       * havn't done that already. */
666
667
	      
	      res |= channel_close(channel);
Niels Möller's avatar
Niels Möller committed
668
	    }
669
670
671
672
	      
	  return channel_process_status(closure->table, channel_number,
					res);

Niels Möller's avatar
Niels Möller committed
673
674
675
676
	}
      werror("EOF on non-existant channel %d\n",
	     channel_number);
      return LSH_FAIL | LSH_DIE;
Niels Möller's avatar
Niels Möller committed
677
    }
Niels Möller's avatar
Niels Möller committed
678
      
Niels Möller's avatar
Niels Möller committed
679
680
681
682
683
684
685
686
  lsh_string_free(packet);
  return LSH_FAIL | LSH_DIE;
}

static int do_channel_close(struct packet_handler *c,
			    struct ssh_connection *connection,
			    struct lsh_string *packet)
{
687
  CAST(channel_handler, closure, c);
Niels Möller's avatar
Niels Möller committed
688
689
690
691
692
693
694
695
696

  struct simple_buffer buffer;
  int msg_number;
  UINT32 channel_number;
  
  simple_buffer_init(&buffer, packet->length, packet->data);

  if (parse_uint8(&buffer, &msg_number)
      && (msg_number == SSH_MSG_CHANNEL_CLOSE)
697
      && parse_uint32(&buffer, &channel_number)
Niels Möller's avatar
Niels Möller committed
698
699
      && parse_eod(&buffer))
    {
700
      struct ssh_channel *channel = lookup_channel(closure->table,
Niels Möller's avatar
Niels Möller committed
701
702
703
704
						   channel_number);

      lsh_string_free(packet);
      
Niels Möller's avatar
Niels Möller committed
705
706
707
708
709
710
711
712
713
714
      if (channel)
	{
	  if (channel->flags & CHANNEL_RECIEVED_CLOSE)
	    {
	      werror("Recieving multiple CLOSE on channel.\n");
	      return LSH_FAIL | LSH_DIE;
	    }

	  channel->flags |= CHANNEL_RECIEVED_CLOSE;
	  
Niels Möller's avatar
Niels Möller committed
715
	  if (! (channel->flags & (CHANNEL_RECIEVED_EOF | CHANNEL_SENT_EOF)))
Niels Möller's avatar
Niels Möller committed
716
717
718
	    {
	      werror("Unexpected channel CLOSE.\n");
	    }
719
720
721
722

	  if (! (channel->flags & (CHANNEL_RECIEVED_EOF))
	      && channel->eof)
	    CHANNEL_EOF(channel);
Niels Möller's avatar
Niels Möller committed
723
	  
Niels Möller's avatar
Niels Möller committed
724
725
726
727
728
	  return channel_process_status(
	    closure->table, channel_number,
	    ( (channel->flags & (CHANNEL_SENT_CLOSE))
	      ? LSH_OK | LSH_CHANNEL_FINISHED
	      : channel_close(channel)));
Niels Möller's avatar
Niels Möller committed
729
730
731
732
	}
      werror("CLOSE on non-existant channel %d\n",
	     channel_number);
      return LSH_FAIL | LSH_DIE;
Niels Möller's avatar
Niels Möller committed
733
734
735
736
737
738
      
    }
  lsh_string_free(packet);
  return LSH_FAIL | LSH_DIE;
}

739
static int do_channel_open_confirm(struct packet_handler *c,
Niels Möller's avatar
Niels Möller committed
740
741
742
			      struct ssh_connection *connection,
			      struct lsh_string *packet)
{
743
  CAST(channel_handler, closure, c);
Niels Möller's avatar
Niels Möller committed
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761

  struct simple_buffer buffer;
  int msg_number;
  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)
      && parse_uint32(&buffer, &local_channel_number)
      && parse_uint32(&buffer, &remote_channel_number)
      && parse_uint32(&buffer, &window_size)
      && parse_uint32(&buffer, &max_packet)
      && parse_eod(&buffer))
    {
762
      struct ssh_channel *channel = lookup_channel(closure->table,
Niels Möller's avatar
Niels Möller committed
763
764
765
766
						   local_channel_number);

      lsh_string_free(packet);

767
      if (channel && channel->open_confirm)
Niels Möller's avatar
Niels Möller committed
768
769
770
771
772
	{
	  channel->channel_number = remote_channel_number;
	  channel->send_window_size = window_size;
	  channel->send_max_packet = max_packet;

Niels Möller's avatar
Niels Möller committed
773
774
	  return channel_process_status(closure->table, local_channel_number,
					CHANNEL_OPEN_CONFIRM(channel));
Niels Möller's avatar
Niels Möller committed
775
776
777
778
779
780
781
782
783
	}
      werror("Unexpected SSH_MSG_CHANNEL_OPEN_CONFIRMATION on channel %d\n",
	     local_channel_number);
      return LSH_FAIL | LSH_DIE;
    }
  lsh_string_free(packet);
  return LSH_FAIL | LSH_DIE;
}

784
static int do_channel_open_failure(struct packet_handler *c,
Niels Möller's avatar
Niels Möller committed
785
786
787
			      struct ssh_connection *connection,
			      struct lsh_string *packet)
{
788
  CAST(channel_handler, closure, c);
Niels Möller's avatar
Niels Möller committed
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810

  struct simple_buffer buffer;
  int msg_number;
  UINT32 channel_number;
  UINT32 reason;

  UINT8 *msg;
  UINT32 length;

  UINT8 *language;
  UINT32 language_length;
  
  simple_buffer_init(&buffer, packet->length, packet->data);

  if (parse_uint8(&buffer, &msg_number)
      && (msg_number == SSH_MSG_CHANNEL_OPEN_FAILURE)
      && parse_uint32(&buffer, &channel_number)
      && parse_uint32(&buffer, &reason)
      && parse_string(&buffer, &length, &msg)
      && parse_string(&buffer, &language_length, &language)
      && parse_eod(&buffer))
    {
811
      struct ssh_channel *channel = lookup_channel(closure->table,
Niels Möller's avatar
Niels Möller committed
812
813
814
815
						   channel_number);

      /* lsh_string_free(packet); */

816
      if (channel && channel->open_failure)
Niels Möller's avatar
Niels Möller committed
817
	{
818
	  int res = CHANNEL_OPEN_FAILURE(channel);
Niels Möller's avatar
Niels Möller committed
819
820

	  lsh_string_free(packet);
Niels Möller's avatar
Niels Möller committed
821
822
823

	  return channel_process_status(closure->table, channel_number,
					res | LSH_CHANNEL_FINISHED);
Niels Möller's avatar
Niels Möller committed
824
825
826
827
828
829
830
831
832
833
834
	}
      werror("Unexpected SSH_MSG_CHANNEL_OPEN_FAILURE on channel %d\n",
	     channel_number);
      lsh_string_free(packet);
      
      return LSH_FAIL | LSH_DIE;
    }
  lsh_string_free(packet);
  return LSH_FAIL | LSH_DIE;
}

835
836
837
838
static int do_channel_success(struct packet_handler *c,
			      struct ssh_connection *connection,
			      struct lsh_string *packet)
{
839
  CAST(channel_handler, closure, c);
840
841
842
843
844
845
846
847
848

  struct simple_buffer buffer;
  int msg_number;
  UINT32 channel_number;
  
  simple_buffer_init(&buffer, packet->length, packet->data);

  if (parse_uint8(&buffer, &msg_number)
      && (msg_number == SSH_MSG_CHANNEL_SUCCESS)
849
      && parse_uint32(&buffer, &channel_number)
850
851
852
853
854
855
856
      && parse_eod(&buffer))
    {
      struct ssh_channel *channel = lookup_channel(closure->table,
						   channel_number);

      lsh_string_free(packet);
      
Niels Möller's avatar
Niels Möller committed
857
      if (channel && channel->channel_success)
Niels Möller's avatar
Niels Möller committed
858
859
	return channel_process_status(closure->table, channel_number,
				      CHANNEL_SUCCESS(channel));
860
861
862
863
864
865
866
867
868
    }
  lsh_string_free(packet);
  return LSH_FAIL | LSH_DIE;
}

static int do_channel_failure(struct packet_handler *c,
			      struct ssh_connection *connection,
			      struct lsh_string *packet)
{
869
  CAST(channel_handler, closure, c);
870
871
872
873
874
875
876
877
878

  struct simple_buffer buffer;
  int msg_number;
  UINT32 channel_number;
  
  simple_buffer_init(&buffer, packet->length, packet->data);

  if (parse_uint8(&buffer, &msg_number)
      && (msg_number == SSH_MSG_CHANNEL_FAILURE)
879
      && parse_uint32(&buffer, &channel_number)
880
881
882
883
884
885
886
      && parse_eod(&buffer))
    {
      struct ssh_channel *channel = lookup_channel(closure->table,
						   channel_number);

      lsh_string_free(packet);
      
Niels Möller's avatar
Niels Möller committed
887
      if (channel && channel->channel_failure)
Niels Möller's avatar
Niels Möller committed
888
889
	return channel_process_status(closure->table, channel_number,
				      CHANNEL_FAILURE(channel));
890
891
892
893
894
    }
  lsh_string_free(packet);
  return LSH_FAIL | LSH_DIE;
}

895
static int init_connection_service(struct ssh_service *s,
Niels Möller's avatar
Niels Möller committed
896
				   struct ssh_connection *connection)
Niels Möller's avatar
Niels Möller committed
897
{
898
899
  CAST(connection_service, self, s);

900
  struct channel_table *table;
Niels Möller's avatar
Niels Möller committed
901
  
902
903
904
  NEW(global_request_handler, globals);
  NEW(channel_open_handler, open);
  NEW(channel_handler, request);
Niels Möller's avatar
Niels Möller committed
905

906
907
908
  NEW(channel_handler, adjust);
  NEW(channel_handler, data);
  NEW(channel_handler, extended);
Niels Möller's avatar
Niels Möller committed
909

910
911
  NEW(channel_handler, eof);
  NEW(channel_handler, close);
912

913
914
  NEW(channel_handler, open_confirm);
  NEW(channel_handler, open_failure);
Niels Möller's avatar
Niels Möller committed
915

916
917
  NEW(channel_handler, channel_success);
  NEW(channel_handler, channel_failure);
Niels Möller's avatar
Niels Möller committed
918

919
  table = make_channel_table();
Niels Möller's avatar
Niels Möller committed
920
921
  
  globals->super.super.handler = do_global_request;
922
  globals->super.table = table;
Niels Möller's avatar
Niels Möller committed
923
924
925
926
  globals->global_requests = self->global_requests;
  connection->dispatch[SSH_MSG_GLOBAL_REQUEST] = &globals->super.super;
    
  open->super.super.handler = do_channel_open;
927
  open->super.table = table;
Niels Möller's avatar
Niels Möller committed
928
929
930
931
  open->channel_types = self->channel_types;
  connection->dispatch[SSH_MSG_CHANNEL_OPEN] = &open->super.super;

  request->super.handler = do_channel_request;
932
  request->table = table;
Niels Möller's avatar
Niels Möller committed
933
934
935
  connection->dispatch[SSH_MSG_CHANNEL_REQUEST] = &request->super;
  
  adjust->super.handler = do_window_adjust;
936
  adjust->table = table;
Niels Möller's avatar
Niels Möller committed
937
938
939
  connection->dispatch[SSH_MSG_CHANNEL_WINDOW_ADJUST] = &adjust->super;

  data->super.handler = do_channel_data;
940
  data->table = table;
Niels Möller's avatar
Niels Möller committed
941
942
943
  connection->dispatch[SSH_MSG_CHANNEL_DATA] = &data->super;

  extended->super.handler = do_channel_extended_data;
944
  extended->table = table;
Niels Möller's avatar
Niels Möller committed
945
946
947
  connection->dispatch[SSH_MSG_CHANNEL_WINDOW_ADJUST] = &extended->super;

  eof->super.handler = do_channel_eof;
948
  eof->table = table;
Niels Möller's avatar
Niels Möller committed
949
950
951
  connection->dispatch[SSH_MSG_CHANNEL_EOF] = &eof->super;

  close->super.handler = do_channel_close;
952
  close->table = table;
Niels Möller's avatar
Niels Möller committed
953
954
  connection->dispatch[SSH_MSG_CHANNEL_CLOSE] = &close->super;

955
956
957
  open_confirm->super.handler = do_channel_open_confirm;
  open_confirm->table = table;
  connection->dispatch[SSH_MSG_CHANNEL_OPEN_CONFIRMATION] = &open_confirm->super;
Niels Möller's avatar
Niels Möller committed
958

959
960
961
  open_failure->super.handler = do_channel_open_failure;
  open_failure->table = table;
  connection->dispatch[SSH_MSG_CHANNEL_OPEN_FAILURE] = &open_failure->super;
Niels Möller's avatar
Niels Möller committed
962
  
963
964
965
966
967
968
969
970
  channel_success->super.handler = do_channel_success;
  channel_success->table = table;
  connection->dispatch[SSH_MSG_CHANNEL_SUCCESS] = &channel_success->super;

  channel_failure->super.handler = do_channel_failure;
  channel_failure->table = table;
  connection->dispatch[SSH_MSG_CHANNEL_FAILURE] = &channel_failure->super;
    
Niels Möller's avatar
Niels Möller committed
971
972
  return self->start
    ? CONNECTION_START(self->start, table, connection->write)
973
    : LSH_OK | LSH_GOON;
Niels Möller's avatar
Niels Möller committed
974
975
}

976
977
978
struct ssh_service *make_connection_service(struct alist *global_requests,
					    struct alist *channel_types,
					    struct connection_startup *start)
979
{
980
  NEW(connection_service, self);
981

982
983
984
985
986
987
  self->super.init = init_connection_service;
  self->global_requests = global_requests;
  self->channel_types = channel_types;
  self->start = start;
  
  return &self->super;
988
989
}

Niels Möller's avatar
Niels Möller committed
990
991
992
993
994
995
996
997
998
999
1000
struct lsh_string *format_channel_close(struct ssh_channel *channel)
{
  return ssh_format("%c%i",
		    SSH_MSG_CHANNEL_CLOSE,
		    channel->channel_number);
}

int channel_close(struct ssh_channel *channel)
{
  channel->flags |= CHANNEL_SENT_CLOSE;

For faster browsing, not all history is shown. View entire blame