channel.c 32.3 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

Niels Möller's avatar
Niels Möller committed
41
42
43
44
45
46
47
struct connection_service
{
  struct ssh_service super;

  /* Supported global requests */
  struct alist *global_requests;
  struct alist *channel_types;
48
49
50
51

  /* Initialize connection (for instance, request channels to be opened
   * or services to be forwarded. */
  struct connection_startup *start;
Niels Möller's avatar
Niels Möller committed
52
53
};

54
struct channel_handler
Niels Möller's avatar
Niels Möller committed
55
56
57
{
  struct packet_handler super;
  
58
  struct channel_table *table;
Niels Möller's avatar
Niels Möller committed
59
60
61
62
};

struct global_request_handler
{
63
  struct channel_handler super;
Niels Möller's avatar
Niels Möller committed
64
65
66
67
68
69
  
  struct alist *global_requests;
};

struct channel_open_handler
{
70
  struct channel_handler super;
Niels Möller's avatar
Niels Möller committed
71
72
73
74
75
76
77

  struct alist *channel_types;
};

#if 0
struct channel_request_handler
{
78
  struct channel_handler *super;
Niels Möller's avatar
Niels Möller committed
79
80
81
82
83
84
85
86
87
88

  struct alist *request_types;
};
#endif

struct lsh_string *format_global_failure(void)
{
  return ssh_format("%c", SSH_MSG_REQUEST_FAILURE);
}

89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
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);
112
  ssh_vformat_write(format, l2, packet->data+l1, args);
113
114
115
116
117
118
119
  va_end(args);

  return packet;
#undef CONFIRM_FORMAT
#undef CONFIRM_ARGS
}

Niels Möller's avatar
Niels Möller committed
120
121
122
123
124
125
126
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);
}

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

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

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

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

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

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

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

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

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

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

  return i;
}

202
void dealloc_channel(struct channel_table *table, int i)
Niels Möller's avatar
Niels Möller committed
203
204
{
  assert(i >= 0);
205
  assert(i < table->used_channels);
Niels Möller's avatar
Niels Möller committed
206
  
207
208
209
210
211
212
213
214
215
216
217
218
  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
219

220
  return n;
Niels Möller's avatar
Niels Möller committed
221
222
}

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

Niels Möller's avatar
Niels Möller committed
229
230
231
232
233
234
235
236
237
238
239
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;
240
241
242

      if (table->channels[channel]->close)
	CHANNEL_CLOSE(table->channels[channel]);
Niels Möller's avatar
Niels Möller committed
243
244
245
246
247
248
249
250
251
252
      
      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
253

Niels Möller's avatar
Niels Möller committed
254
/* Channel related messages */
Niels Möller's avatar
Niels Möller committed
255
256
257
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
290
291
292
293
294
295
296
297
298
static int do_global_request(struct packet_handler *c,
			     struct ssh_connection *connection,
			     struct lsh_string *packet)
{
  struct global_request_handler *closure = (struct global_request_handler *) c;

  struct simple_buffer buffer;
  int msg_number;
  int name;
  int want_reply;
  
  MDEBUG(closure);

  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)
{
  struct channel_open_handler *closure = (struct channel_open_handler *) c;

  struct simple_buffer buffer;
  int msg_number;
  int type;
299
300
301
  UINT32 remote_channel_number;
  UINT32 window_size;
  UINT32 max_packet;
Niels Möller's avatar
Niels Möller committed
302
303
304
305
306
307
308
309
  
  MDEBUG(closure);

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

  if (parse_uint8(&buffer, &msg_number)
      && (msg_number == SSH_MSG_CHANNEL_OPEN)
      && parse_atom(&buffer, &type)
310
311
312
      && 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
313
314
    {
      struct channel_open *open;
315
316
317
318
319
320
321
      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
322
      lsh_string_free(packet);
Niels Möller's avatar
Niels Möller committed
323
324
325
326
327
328
329

      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
330
331
332
      
      if (!type || !(open = ALIST_GET(closure->channel_types, type)))
	return A_WRITE(connection->write,
333
		       format_open_failure(remote_channel_number,
Niels Möller's avatar
Niels Möller committed
334
335
					   SSH_OPEN_UNKNOWN_CHANNEL_TYPE,
					   "Unknown channel type", ""));
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
364
365
366
367
368
369
      
      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
370
371
		     : format_open_confirmation(channel, local_channel_number,
						""));
Niels Möller's avatar
Niels Möller committed
372
373
374
375
376
377
378
379
380
381
    }
  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)
{
382
  struct channel_handler *closure = (struct channel_handler *) c;
Niels Möller's avatar
Niels Möller committed
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399

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

  MDEBUG(closure);

  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))
    {
400
      struct ssh_channel *channel = lookup_channel(closure->table,
Niels Möller's avatar
Niels Möller committed
401
402
403
404
405
406
407
408
						   channel_number);

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

409
410
	  if (type && channel->request_types 
	      && ( (req = ALIST_GET(channel->request_types, type)) ))
Niels Möller's avatar
Niels Möller committed
411
412
413
	    return channel_process_status
	      (closure->table, channel_number,
	       CHANNEL_REQUEST(req, channel, want_reply, &buffer));
414
	  else
415
416
417
418
419
	    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
420
421
422
423
424
425
426
427
428
429
430
431
	}
      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,
432
			    struct lsh_string *packet)
Niels Möller's avatar
Niels Möller committed
433
{
434
  struct channel_handler *closure = (struct channel_handler *) c;
Niels Möller's avatar
Niels Möller committed
435
436
437
438
439
440
441
442
443
444
445
446

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

  MDEBUG(closure);

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

  if (parse_uint8(&buffer, &msg_number)
      && (msg_number == SSH_MSG_CHANNEL_WINDOW_ADJUST)
447
      && parse_uint32(&buffer, &channel_number)
Niels Möller's avatar
Niels Möller committed
448
449
450
      && parse_uint32(&buffer, &size)
      && parse_eod(&buffer))
    {
451
      struct ssh_channel *channel = lookup_channel(closure->table,
Niels Möller's avatar
Niels Möller committed
452
453
454
455
						   channel_number);

      lsh_string_free(packet);
      
Niels Möller's avatar
Niels Möller committed
456
457
      if (channel
	  && !(channel->flags & (CHANNEL_RECIEVED_EOF | CHANNEL_RECIEVED_CLOSE)))
Niels Möller's avatar
Niels Möller committed
458
	{
Niels Möller's avatar
Niels Möller committed
459
460
461
462
	  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
463
464
465
		return channel_process_status(closure->table,
					      channel_number,
					      CHANNEL_SEND(channel));
Niels Möller's avatar
Niels Möller committed
466
	    }
Niels Möller's avatar
Niels Möller committed
467
468
469
470
	  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
471
      werror("SSH_MSG_CHANNEL_WINDOW_ADJUST on nonexistant or closed channel %d\n",
Niels Möller's avatar
Niels Möller committed
472
473
474
475
476
477
478
479
480
481
482
483
	     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)
{
484
  struct channel_handler *closure = (struct channel_handler *) c;
Niels Möller's avatar
Niels Möller committed
485
486
487
488
489
490
491
492
493
494
495
496

  struct simple_buffer buffer;
  int msg_number;
  UINT32 channel_number;
  struct lsh_string *data;
  
  MDEBUG(closure);

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

  if (parse_uint8(&buffer, &msg_number)
      && (msg_number == SSH_MSG_CHANNEL_DATA)
497
      && parse_uint32(&buffer, &channel_number)
Niels Möller's avatar
Niels Möller committed
498
499
500
      && ( (data = parse_string_copy(&buffer)) )
      && parse_eod(&buffer))
    {
501
      struct ssh_channel *channel = lookup_channel(closure->table,
Niels Möller's avatar
Niels Möller committed
502
503
504
505
						   channel_number);

      lsh_string_free(packet);
      
Niels Möller's avatar
Niels Möller committed
506
507
      if (channel && channel->recieve
	  && !(channel->flags & (CHANNEL_RECIEVED_EOF | CHANNEL_RECIEVED_CLOSE)))
Niels Möller's avatar
Niels Möller committed
508
	{
Niels Möller's avatar
Niels Möller committed
509
	  if (channel->flags & CHANNEL_SENT_CLOSE)
Niels Möller's avatar
Niels Möller committed
510
	    {
Niels Möller's avatar
Niels Möller committed
511
512
	      werror("Ignoring data on channel which is closing\n");
	      return LSH_OK | LSH_GOON;
Niels Möller's avatar
Niels Möller committed
513
	    }
Niels Möller's avatar
Niels Möller committed
514
515
	  else
	    {
516
517
	      int res = 0;
	      
Niels Möller's avatar
Niels Möller committed
518
519
520
521
522
523
	      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;
		}
524
525
526
527
528
529
530
531
532
533
534
	      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
535
536
537
538
	      return channel_process_status(
		closure->table, channel_number,
		res | CHANNEL_RECIEVE(channel, 
				      CHANNEL_DATA, data));
Niels Möller's avatar
Niels Möller committed
539
540
	    }
	  return LSH_OK | LSH_GOON;
Niels Möller's avatar
Niels Möller committed
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
	}
	  
      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)
{
557
  struct channel_handler *closure = (struct channel_handler *) c;
Niels Möller's avatar
Niels Möller committed
558
559
560
561
562
563
564
565
566
567
568
569
570

  struct simple_buffer buffer;
  int msg_number;
  UINT32 channel_number;
  UINT32 type;
  struct lsh_string *data;
  
  MDEBUG(closure);

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

  if (parse_uint8(&buffer, &msg_number)
      && (msg_number == SSH_MSG_CHANNEL_EXTENDED_DATA)
571
      && parse_uint32(&buffer, &channel_number)
Niels Möller's avatar
Niels Möller committed
572
573
574
575
      && parse_uint32(&buffer, &type)
      && ( (data = parse_string_copy(&buffer)) )
      && parse_eod(&buffer))
    {
576
      struct ssh_channel *channel = lookup_channel(closure->table,
Niels Möller's avatar
Niels Möller committed
577
578
579
580
						   channel_number);

      lsh_string_free(packet);
      
Niels Möller's avatar
Niels Möller committed
581
582
      if (channel && channel->recieve
	  && !(channel->flags & (CHANNEL_RECIEVED_EOF | CHANNEL_RECIEVED_CLOSE)))
Niels Möller's avatar
Niels Möller committed
583
	{
Niels Möller's avatar
Niels Möller committed
584
	  if (channel->flags & CHANNEL_SENT_CLOSE)
Niels Möller's avatar
Niels Möller committed
585
	    {
Niels Möller's avatar
Niels Möller committed
586
587
	      werror("Ignoring extended data on channel which is closing\n");
	      return LSH_OK | LSH_GOON;
Niels Möller's avatar
Niels Möller committed
588
	    }
Niels Möller's avatar
Niels Möller committed
589
	  else
Niels Möller's avatar
Niels Möller committed
590
	    {
591
592
	      int res = 0;
	      
Niels Möller's avatar
Niels Möller committed
593
594
595
596
597
598
599
600
601
	      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;
602
603
604
605
606
607
608
609
610
611

	      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
612
613
614
	      switch(type)
		{
		case SSH_EXTENDED_DATA_STDERR:
Niels Möller's avatar
Niels Möller committed
615
616
617
618
		  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
619
620
621
622
623
624
		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
625
626
627
628
629
630
631
632
633
634
635
636
637
	    }
	}
      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,
638
639
			  struct ssh_connection *connection,
			  struct lsh_string *packet)
Niels Möller's avatar
Niels Möller committed
640
{
641
  struct channel_handler *closure = (struct channel_handler *) c;
Niels Möller's avatar
Niels Möller committed
642
643
644
645
646
647
648
649
650
651
652

  struct simple_buffer buffer;
  int msg_number;
  UINT32 channel_number;
  
  MDEBUG(closure);

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

  if (parse_uint8(&buffer, &msg_number)
      && (msg_number == SSH_MSG_CHANNEL_EOF)
653
      && parse_uint32(&buffer, &channel_number)
Niels Möller's avatar
Niels Möller committed
654
655
      && parse_eod(&buffer))
    {
656
      struct ssh_channel *channel = lookup_channel(closure->table,
Niels Möller's avatar
Niels Möller committed
657
658
659
660
						   channel_number);

      lsh_string_free(packet);

Niels Möller's avatar
Niels Möller committed
661
662
      if (channel)
	{
663
664
	  int res = 0;
	  
Niels Möller's avatar
Niels Möller committed
665
666
667
668
669
670
671
672
	  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;

673
674
675
676
677
678
	  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
679
680
681
	    {
	      /* Both parties have sent EOF. Initiate close, if we
	       * havn't done that already. */
682
683
	      
	      res |= channel_close(channel);
Niels Möller's avatar
Niels Möller committed
684
	    }
685
686
687
688
	      
	  return channel_process_status(closure->table, channel_number,
					res);

Niels Möller's avatar
Niels Möller committed
689
690
691
692
	}
      werror("EOF on non-existant channel %d\n",
	     channel_number);
      return LSH_FAIL | LSH_DIE;
Niels Möller's avatar
Niels Möller committed
693
    }
Niels Möller's avatar
Niels Möller committed
694
      
Niels Möller's avatar
Niels Möller committed
695
696
697
698
699
700
701
702
  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)
{
703
  struct channel_handler *closure = (struct channel_handler *) c;
Niels Möller's avatar
Niels Möller committed
704
705
706
707
708
709
710
711
712
713
714

  struct simple_buffer buffer;
  int msg_number;
  UINT32 channel_number;
  
  MDEBUG(closure);

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

  if (parse_uint8(&buffer, &msg_number)
      && (msg_number == SSH_MSG_CHANNEL_CLOSE)
715
      && parse_uint32(&buffer, &channel_number)
Niels Möller's avatar
Niels Möller committed
716
717
      && parse_eod(&buffer))
    {
718
      struct ssh_channel *channel = lookup_channel(closure->table,
Niels Möller's avatar
Niels Möller committed
719
720
721
722
						   channel_number);

      lsh_string_free(packet);
      
Niels Möller's avatar
Niels Möller committed
723
724
725
726
727
728
729
730
731
732
      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
733
	  if (! (channel->flags & (CHANNEL_RECIEVED_EOF | CHANNEL_SENT_EOF)))
Niels Möller's avatar
Niels Möller committed
734
735
736
	    {
	      werror("Unexpected channel CLOSE.\n");
	    }
737
738
739
740

	  if (! (channel->flags & (CHANNEL_RECIEVED_EOF))
	      && channel->eof)
	    CHANNEL_EOF(channel);
Niels Möller's avatar
Niels Möller committed
741
	  
Niels Möller's avatar
Niels Möller committed
742
743
744
745
746
	  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
747
748
749
750
	}
      werror("CLOSE on non-existant channel %d\n",
	     channel_number);
      return LSH_FAIL | LSH_DIE;
Niels Möller's avatar
Niels Möller committed
751
752
753
754
755
756
      
    }
  lsh_string_free(packet);
  return LSH_FAIL | LSH_DIE;
}

757
static int do_channel_open_confirm(struct packet_handler *c,
Niels Möller's avatar
Niels Möller committed
758
759
760
			      struct ssh_connection *connection,
			      struct lsh_string *packet)
{
761
  struct channel_handler *closure = (struct channel_handler *) c;
Niels Möller's avatar
Niels Möller committed
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781

  struct simple_buffer buffer;
  int msg_number;
  UINT32 local_channel_number;
  UINT32 remote_channel_number;  
  UINT32 window_size;
  UINT32 max_packet;
  
  MDEBUG(closure);

  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))
    {
782
      struct ssh_channel *channel = lookup_channel(closure->table,
Niels Möller's avatar
Niels Möller committed
783
784
785
786
						   local_channel_number);

      lsh_string_free(packet);

787
      if (channel && channel->open_confirm)
Niels Möller's avatar
Niels Möller committed
788
789
790
791
792
	{
	  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
793
794
	  return channel_process_status(closure->table, local_channel_number,
					CHANNEL_OPEN_CONFIRM(channel));
Niels Möller's avatar
Niels Möller committed
795
796
797
798
799
800
801
802
803
	}
      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;
}

804
static int do_channel_open_failure(struct packet_handler *c,
Niels Möller's avatar
Niels Möller committed
805
806
807
			      struct ssh_connection *connection,
			      struct lsh_string *packet)
{
808
  struct channel_handler *closure = (struct channel_handler *) c;
Niels Möller's avatar
Niels Möller committed
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832

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

  UINT8 *msg;
  UINT32 length;

  UINT8 *language;
  UINT32 language_length;
  
  MDEBUG(closure);

  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))
    {
833
      struct ssh_channel *channel = lookup_channel(closure->table,
Niels Möller's avatar
Niels Möller committed
834
835
836
837
						   channel_number);

      /* lsh_string_free(packet); */

838
      if (channel && channel->open_failure)
Niels Möller's avatar
Niels Möller committed
839
	{
840
	  int res = CHANNEL_OPEN_FAILURE(channel);
Niels Möller's avatar
Niels Möller committed
841
842

	  lsh_string_free(packet);
Niels Möller's avatar
Niels Möller committed
843
844
845

	  return channel_process_status(closure->table, channel_number,
					res | LSH_CHANNEL_FINISHED);
Niels Möller's avatar
Niels Möller committed
846
847
848
849
850
851
852
853
854
855
856
	}
      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;
}

857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
static int do_channel_success(struct packet_handler *c,
			      struct ssh_connection *connection,
			      struct lsh_string *packet)
{
  struct channel_handler *closure = (struct channel_handler *) c;

  struct simple_buffer buffer;
  int msg_number;
  UINT32 channel_number;
  
  MDEBUG(closure);

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

  if (parse_uint8(&buffer, &msg_number)
      && (msg_number == SSH_MSG_CHANNEL_SUCCESS)
873
      && parse_uint32(&buffer, &channel_number)
874
875
876
877
878
879
880
      && 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
881
      if (channel && channel->channel_success)
Niels Möller's avatar
Niels Möller committed
882
883
	return channel_process_status(closure->table, channel_number,
				      CHANNEL_SUCCESS(channel));
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
    }
  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)
{
  struct channel_handler *closure = (struct channel_handler *) c;

  struct simple_buffer buffer;
  int msg_number;
  UINT32 channel_number;
  
  MDEBUG(closure);

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

  if (parse_uint8(&buffer, &msg_number)
      && (msg_number == SSH_MSG_CHANNEL_FAILURE)
905
      && parse_uint32(&buffer, &channel_number)
906
907
908
909
910
911
912
      && 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
913
      if (channel && channel->channel_failure)
Niels Möller's avatar
Niels Möller committed
914
915
	return channel_process_status(closure->table, channel_number,
				      CHANNEL_FAILURE(channel));
916
917
918
919
920
    }
  lsh_string_free(packet);
  return LSH_FAIL | LSH_DIE;
}

921
static int init_connection_service(struct ssh_service *s,
Niels Möller's avatar
Niels Möller committed
922
				   struct ssh_connection *connection)
Niels Möller's avatar
Niels Möller committed
923
924
{
  struct connection_service *self = (struct connection_service *) s;
925
  struct channel_table *table;
Niels Möller's avatar
Niels Möller committed
926
927
928
  
  struct global_request_handler *globals;
  struct channel_open_handler *open;
929
  struct channel_handler *request;
Niels Möller's avatar
Niels Möller committed
930

931
932
933
  struct channel_handler *adjust;
  struct channel_handler *data;
  struct channel_handler *extended;
Niels Möller's avatar
Niels Möller committed
934

935
936
  struct channel_handler *eof;
  struct channel_handler *close;
Niels Möller's avatar
Niels Möller committed
937

938
939
940
941
942
  struct channel_handler *open_confirm;
  struct channel_handler *open_failure;

  struct channel_handler *channel_success;
  struct channel_handler *channel_failure;
Niels Möller's avatar
Niels Möller committed
943
944
945

  MDEBUG(self);

946
  table = make_channel_table();
Niels Möller's avatar
Niels Möller committed
947
948
949
  
  NEW(globals);
  globals->super.super.handler = do_global_request;
950
  globals->super.table = table;
Niels Möller's avatar
Niels Möller committed
951
952
953
954
955
  globals->global_requests = self->global_requests;
  connection->dispatch[SSH_MSG_GLOBAL_REQUEST] = &globals->super.super;
    
  NEW(open);
  open->super.super.handler = do_channel_open;
956
  open->super.table = table;
Niels Möller's avatar
Niels Möller committed
957
958
959
960
961
  open->channel_types = self->channel_types;
  connection->dispatch[SSH_MSG_CHANNEL_OPEN] = &open->super.super;

  NEW(request);
  request->super.handler = do_channel_request;
962
  request->table = table;
Niels Möller's avatar
Niels Möller committed
963
964
965
966
  connection->dispatch[SSH_MSG_CHANNEL_REQUEST] = &request->super;
  
  NEW(adjust);
  adjust->super.handler = do_window_adjust;
967
  adjust->table = table;
Niels Möller's avatar
Niels Möller committed
968
969
970
971
  connection->dispatch[SSH_MSG_CHANNEL_WINDOW_ADJUST] = &adjust->super;

  NEW(data);
  data->super.handler = do_channel_data;
972
  data->table = table;
Niels Möller's avatar
Niels Möller committed
973
974
975
976
  connection->dispatch[SSH_MSG_CHANNEL_DATA] = &data->super;

  NEW(extended);
  extended->super.handler = do_channel_extended_data;
977
  extended->table = table;
Niels Möller's avatar
Niels Möller committed
978
979
980
981
  connection->dispatch[SSH_MSG_CHANNEL_WINDOW_ADJUST] = &extended->super;

  NEW(eof);
  eof->super.handler = do_channel_eof;
982
  eof->table = table;
Niels Möller's avatar
Niels Möller committed
983
984
985
986
  connection->dispatch[SSH_MSG_CHANNEL_EOF] = &eof->super;

  NEW(close);
  close->super.handler = do_channel_close;
987
  close->table = table;
Niels Möller's avatar
Niels Möller committed
988
989
  connection->dispatch[SSH_MSG_CHANNEL_CLOSE] = &close->super;

990
991
992
993
  NEW(open_confirm);
  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
994

995
996
997
998
  NEW(open_failure);
  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
999
  
1000
  NEW(channel_success);
For faster browsing, not all history is shown. View entire blame