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
1001
1002
1003
1004
1005
1006
1007
1008
1009
  NEW(channel_success);
  channel_success->super.handler = do_channel_success;
  channel_success->table = table;
  connection->dispatch[SSH_MSG_CHANNEL_SUCCESS] = &channel_success->super;

  NEW(channel_failure);
  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
1010
1011
  return self->start
    ? CONNECTION_START(self->start, table, connection->write)
1012
    : LSH_OK | LSH_GOON;
Niels Möller's avatar
Niels Möller committed
1013
1014
}

1015
1016
1017
struct ssh_service *make_connection_service(struct alist *global_requests,
					    struct alist *channel_types,
					    struct connection_startup *start)
1018
{
1019
  struct connection_service *self;
1020

1021
  NEW(self);
1022

1023
1024
1025
1026
1027
1028
  self->super.init = init_connection_service;
  self->global_requests = global_requests;
  self->channel_types = channel_types;
  self->start = start;
  
  return &self->super;
1029
1030
}

Niels Möller's avatar
Niels Möller committed
1031
1032
1033
1034
1035
1036
1037
1038
1039
1040
1041
1042
1043
1044
1045
1046
1047
1048
1049
1050
1051
1052
1053
1054
1055
1056
1057
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;

  return A_WRITE(channel->write, format_channel_close(channel))
    | ( (channel->flags & CHANNEL_RECIEVED_CLOSE)
	? LSH_CHANNEL_FINISHED : 0);
}

struct lsh_string *format_channel_eof(struct ssh_channel *channel)
{
  return ssh_format("%c%i",
		    SSH_MSG_CHANNEL_EOF,
		    channel->channel_number);
}

int channel_eof(struct ssh_channel *channel)
{
  int res ;
  
1058
  channel->flags |= CHANNEL_SENT_EOF;
Niels Möller's avatar
Niels Möller committed
1059
1060
1061
1062
1063
  res =  A_WRITE(channel->write, format_channel_eof(channel));

  if (LSH_CLOSEDP(res))
    return res;

1064
  if (channel->flags & (CHANNEL_RECIEVED_EOF | CHANNEL_CLOSE_AT_EOF))
Niels Möller's avatar
Niels Möller committed
1065
1066
1067
1068
1069
1070
1071
1072
1073
1074
1075
1076
1077
1078
    {
      /* Initiate close */
      res |= channel_close(channel);
    }

  return res;
}

void init_channel(struct ssh_channel *channel)
{
  /* channel->super.handler = do_read_channel; */
  channel->write = NULL;

  channel->flags = 0;
1079
  channel->sources = 0;
Niels Möller's avatar
Niels Möller committed
1080
1081
1082
1083
  
  channel->request_types = NULL;
  channel->recieve = NULL;
  channel->send = NULL;
1084

Niels Möller's avatar
Niels Möller committed
1085
1086
  channel->close = NULL;
  channel->eof = NULL;
1087

Niels Möller's avatar
Niels Möller committed
1088
1089
1090
1091
1092
1093
  channel->open_confirm = NULL;
  channel->open_failure = NULL;
  channel->channel_success = NULL;
  channel->channel_failure = NULL;
}

1094
1095
struct lsh_string *channel_transmit_data(struct ssh_channel *channel,
					 struct lsh_string *data)
1096
{
1097
1098
1099
1100
1101
1102
1103
  assert(data->length <= channel->send_window_size);
  assert(data->length <= channel->send_max_packet);
  
  return ssh_format("%c%i%fS",
		    SSH_MSG_CHANNEL_DATA,
		    channel->channel_number,
		    data);
1104
1105
}

1106
1107
1108
struct lsh_string *channel_transmit_extended(struct ssh_channel *channel,
					     UINT32 type,
					     struct lsh_string *data)
1109
{
1110
1111
  assert(data->length <= channel->send_window_size);
  assert(data->length <= channel->send_max_packet);
1112
  
1113
1114
1115
1116
1117
  return ssh_format("%c%i%i%fS",
		    SSH_MSG_CHANNEL_EXTENDED_DATA,
		    type,
		    channel->channel_number,
		    data);
1118
1119
}

Niels Möller's avatar
Niels Möller committed
1120
1121
/* Writing data to a channel */
struct channel_write
Niels Möller's avatar
Niels Möller committed
1122
{
Niels Möller's avatar
Niels Möller committed
1123
1124
1125
  struct abstract_write super;
  struct ssh_channel *channel;
};
Niels Möller's avatar
Niels Möller committed
1126

Niels Möller's avatar
Niels Möller committed
1127
1128
1129
1130
1131
struct channel_write_extended
{
  struct channel_write super;
  UINT32 type;
};
Niels Möller's avatar
Niels Möller committed
1132

1133
static int do_channel_write(struct abstract_write *w,
Niels Möller's avatar
Niels Möller committed
1134
1135
			    struct lsh_string *packet)
{
1136
  struct channel_write *closure = (struct channel_write *) w;
Niels Möller's avatar
Niels Möller committed
1137

Niels Möller's avatar
Niels Möller committed
1138
1139
1140
1141
  return A_WRITE(closure->channel->write,
		 channel_transmit_data(closure->channel, packet));
}

1142
1143
static int do_channel_write_extended(struct abstract_write *w,
				     struct lsh_string *packet)
Niels Möller's avatar
Niels Möller committed
1144
{
1145
  struct channel_write_extended *closure = (struct channel_write_extended *) w;
Niels Möller's avatar
Niels Möller committed
1146
1147
1148
1149
1150

  return A_WRITE(closure->super.channel->write,
		 channel_transmit_extended(closure->super.channel,
					   closure->type,
					   packet));
1151
1152
}

Niels Möller's avatar
Niels Möller committed
1153
struct abstract_write *make_channel_write(struct ssh_channel *channel)
1154
{
Niels Möller's avatar
Niels Möller committed
1155
1156
1157
1158
1159
1160
1161
1162
1163
1164
1165
1166
1167
1168
1169
1170
1171
1172
1173
1174
  struct channel_write *closure;

  NEW(closure);

  closure->super.write = do_channel_write;
  closure->channel = channel;

  return &closure->super;
}

struct abstract_write *make_channel_write_extended(struct ssh_channel *channel,
						   UINT32 type)
{
  struct channel_write_extended *closure;

  NEW(closure);

  closure->super.super.write = do_channel_write_extended;
  closure->super.channel = channel;
  closure->type = type;
1175
  
Niels Möller's avatar
Niels Möller committed
1176
  return &closure->super.super;
Niels Möller's avatar
Niels Möller committed
1177
}
1178

Niels Möller's avatar
Niels Möller committed
1179
1180
1181
1182
1183
1184
1185
1186
1187
1188
1189
1190
struct read_handler *make_channel_read_data(struct ssh_channel *channel)
{
  return make_read_data(channel, make_channel_write(channel));
}

struct read_handler *make_channel_read_stderr(struct ssh_channel *channel)
{
  return make_read_data(channel,
			make_channel_write_extended(channel,
						    SSH_EXTENDED_DATA_STDERR));
}    

1191
1192
1193
1194
1195
1196
1197
1198
1199
1200
1201
1202
1203
1204
1205
1206
1207
1208
1209
1210
1211
1212
1213
1214
1215
1216
1217
1218
1219
struct channel_close_callback
{
  struct close_callback super;
  struct ssh_channel *channel;
};

static int channel_close_callback(struct close_callback *c, int reason)
{
  struct channel_close_callback *closure = (struct channel_close_callback *)c;

  MDEBUG(closure);

  channel_close(closure->channel);

  /* FIXME: So far, the returned value is ignored. */
  return 17;
}
  
struct close_callback *make_channel_close(struct ssh_channel *channel)
{
  struct channel_close_callback *closure;

  NEW(closure);
  closure->super.f = channel_close_callback;
  closure->channel = channel;

  return &closure->super;
}

1220
1221
1222
1223
1224
1225
1226
1227
1228
1229
1230
1231
1232
1233
1234
1235
1236
1237
1238
1239
1240
1241
1242
1243
1244
1245
1246
1247
1248
struct lsh_string *prepare_channel_open(struct channel_table *table,
					int type, struct ssh_channel *channel,
					char *format, ...)
{
  int index;
    
  va_list args;
  UINT32 l1, l2;
  struct lsh_string *packet;

#define OPEN_FORMAT "%c%a%i%i%i"
#define OPEN_ARGS SSH_MSG_CHANNEL_OPEN, type, (UINT32) index, \
  channel->rec_window_size, channel->rec_max_packet  

  index = register_channel(table, channel);
  if (index < 0)
    return NULL;

  l1 = ssh_format_length(OPEN_FORMAT, OPEN_ARGS);
  
  va_start(args, format);
  l2 = ssh_vformat_length(format, args);
  va_end(args);

  packet = lsh_string_alloc(l1 + l2);

  ssh_format_write(OPEN_FORMAT, l1, packet->data, OPEN_ARGS);

  va_start(args, format);
1249
  ssh_vformat_write(format, l2, packet->data+l1, args);
1250
1251
1252
1253
1254
1255
1256
1257
1258
1259
1260
1261
1262
1263
1264
  va_end(args);

  return packet;
#undef OPEN_FORMAT
#undef OPEN_ARGS
}
		   
struct lsh_string *format_channel_request(int type, struct ssh_channel *channel,
					  int want_reply, char *format, ...)
{
  va_list args;
  UINT32 l1, l2;
  struct lsh_string *packet;

#define REQUEST_FORMAT "%c%i%a%c"
Niels Möller's avatar
Niels Möller committed
1265
#define REQUEST_ARGS SSH_MSG_CHANNEL_REQUEST, channel->channel_number, \
1266
1267
1268
1269
1270
1271
1272
1273
1274
1275
1276
1277
1278
  type, want_reply
    
  l1 = ssh_format_length(REQUEST_FORMAT, REQUEST_ARGS);
  
  va_start(args, format);
  l2 = ssh_vformat_length(format, args);
  va_end(args);

  packet = lsh_string_alloc(l1 + l2);

  ssh_format_write(REQUEST_FORMAT, l1, packet->data, REQUEST_ARGS);

  va_start(args, format);
1279
  ssh_vformat_write(format, l2, packet->data+l1, args);
1280
1281
1282
1283
1284
1285
1286
  va_end(args);

  return packet;
#undef REQUEST_FORMAT
#undef REQUEST_ARGS
}