command.c 13.1 KB
Newer Older
Niels Möller's avatar
Niels Möller committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
/* command.h
 *
 * $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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

#include "command.h"

26
27
#include "connection.h"
#include "io.h"
28
#include "werror.h"
29
30
#include "xalloc.h"

31
32
#include <assert.h>

33
#define GABA_DEFINE
34
#include "command.h.x"
35
#undef GABA_DEFINE
36
37
38

#include "command.c.x"

39
/* GABA:
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
   (class
     (name command_apply)
     (super command_frame)
     (vars
       (f object command)))
*/

static int do_command_apply(struct command_continuation *s,
			    struct lsh_object *value)
{
  CAST(command_apply, self, s);
  return COMMAND_CALL(self->f, value, self->super.up);
}

struct command_continuation *
make_apply(struct command *f, struct command_continuation *c)
{
  NEW(command_apply, res);
  res->f = f;
  res->super.up = c;
  res->super.super.c = do_command_apply;

  return &res->super.super;
}
64

Niels Möller's avatar
Niels Möller committed
65
66
struct lsh_object *gaba_apply(struct lsh_object *f,
			      struct lsh_object *x)
67
68
69
70
71
{
  CAST_SUBTYPE(command_simple, cf, f);
  return COMMAND_SIMPLE(cf, x);
}

72
int do_call_simple_command(struct command *s,
73
74
75
			   struct lsh_object *arg,
			   struct command_continuation *c)
{
76
  CAST_SUBTYPE(command_simple, self, s);
77
78
79
  return COMMAND_RETURN(c, COMMAND_SIMPLE(self, arg));
}

80

81
82
83
84
85
86
/* Unimplemented command */
static int
do_command_unimplemented(struct command *s UNUSED,
			 struct lsh_object *o UNUSED,
			 struct command_continuation *c UNUSED)
{ fatal("command.c: Unimplemented command.\n"); }
Niels Möller's avatar
Niels Möller committed
87

88
89
90
91
static struct lsh_object *
do_command_simple_unimplemented(struct command_simple *s UNUSED,
				struct lsh_object *o UNUSED)
{ fatal("command.c: Unimplemented simple command.\n"); }
Niels Möller's avatar
Niels Möller committed
92

93
94
95
96
97
98
99
100
101
struct command_simple command_unimplemented =
{ { STATIC_HEADER, do_command_unimplemented}, do_command_simple_unimplemented};

static struct lsh_object *
do_collect_1(struct command_simple *s, struct lsh_object *a)
{
  CAST(collect_info_1, self, s);
  return self->f(self, a);
}
Niels Möller's avatar
Niels Möller committed
102
103
104

/* GABA:
   (class
105
106
     (name collect_state_1)
     (super command_simple)
Niels Möller's avatar
Niels Möller committed
107
     (vars
108
       (info object collect_info_2)
Niels Möller's avatar
Niels Möller committed
109
110
111
112
113
       (a object lsh_object)))
*/

/* GABA:
   (class
114
115
     (name collect_state_2)
     (super command_simple)
Niels Möller's avatar
Niels Möller committed
116
     (vars
117
118
       (info object collect_info_3)
       (a object lsh_object)
Niels Möller's avatar
Niels Möller committed
119
120
       (b object lsh_object)))
*/
121

Niels Möller's avatar
Niels Möller committed
122
123
/* GABA:
   (class
124
125
     (name collect_state_3)
     (super command_simple)
Niels Möller's avatar
Niels Möller committed
126
     (vars
127
128
129
       (info object collect_info_4)
       (a object lsh_object)
       (b object lsh_object)
Niels Möller's avatar
Niels Möller committed
130
131
132
       (c object lsh_object)))
*/

133
134
135
static struct lsh_object *
do_collect_2(struct command_simple *s,
	     struct lsh_object *x)
Niels Möller's avatar
Niels Möller committed
136
{
137
138
  CAST(collect_state_1, self, s);
  return self->info->f(self->info, self->a, x);
Niels Möller's avatar
Niels Möller committed
139
140
}

141
142
143
struct lsh_object *
make_collect_state_1(struct collect_info_1 *info,
		     struct lsh_object *a)
Niels Möller's avatar
Niels Möller committed
144
{
145
146
147
148
149
150
  NEW(collect_state_1, self);
  self->info = info->next;
  self->a = a;

  self->super.call_simple = do_collect_2;
  self->super.super.call = do_call_simple_command;
Niels Möller's avatar
Niels Möller committed
151
  
152
  return &self->super.super.super;
Niels Möller's avatar
Niels Möller committed
153
154
155
}

static struct lsh_object *
156
157
do_collect_3(struct command_simple *s,
	     struct lsh_object *x)
Niels Möller's avatar
Niels Möller committed
158
{
159
160
  CAST(collect_state_2, self, s);
  return self->info->f(self->info, self->a, self->b, x);
Niels Möller's avatar
Niels Möller committed
161
162
}

163
164
165
166
struct lsh_object *
make_collect_state_2(struct collect_info_2 *info,
		     struct lsh_object *a,
		     struct lsh_object *b)
167
{
168
169
170
171
  NEW(collect_state_2, self);
  self->info = info->next;
  self->a = a;
  self->b = b;
Niels Möller's avatar
Niels Möller committed
172
  
173
174
175
176
  self->super.call_simple = do_collect_3;
  self->super.super.call = do_call_simple_command;
  
  return &self->super.super.super;
177
}
Niels Möller's avatar
Niels Möller committed
178

179
180
181
static struct lsh_object *
do_collect_4(struct command_simple *s,
	     struct lsh_object *x)
Niels Möller's avatar
Niels Möller committed
182
{
183
184
185
  CAST(collect_state_3, self, s);
  return self->info->f(self->info, self->a, self->b, self->c, x);
}
Niels Möller's avatar
Niels Möller committed
186

187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
struct lsh_object *
make_collect_state_3(struct collect_info_3 *info,
		     struct lsh_object *a,
		     struct lsh_object *b,
		     struct lsh_object *c)
{
  NEW(collect_state_3, self);
  self->info = info->next;
  self->a = a;
  self->b = b;
  self->c = c;
  
  self->super.call_simple = do_collect_4;
  self->super.super.call = do_call_simple_command;
  
  return &self->super.super.super;
Niels Möller's avatar
Niels Möller committed
203
204
205
206
207
}

/* Combinators */

/* Ix == x */
208

209
210
211
static struct lsh_object *
do_simple_command_I(struct command_simple *ignored UNUSED,
		    struct lsh_object *arg)
212
213
214
215
{
  return arg;
}

216
217
struct command_simple command_I =
STATIC_COMMAND_SIMPLE(do_simple_command_I);
218

219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
/* ((K x) y) == x */

/* Represents (K x) */
/* GABA:
   (class
     (name command_K_1)
     (super command_simple)
     (vars
       (x object lsh_object)))
*/

static struct lsh_object *
do_simple_command_K_1(struct command_simple *s,
		      struct lsh_object *ignored UNUSED)
{
  CAST(command_K_1, self, s);
  return self->x;
}

struct command *make_command_K_1(struct lsh_object *x)
{
  NEW(command_K_1, res);
  res->x = x;
  res->super.super.call = do_call_simple_command;
  res->super.call_simple = do_simple_command_K_1;

  return &res->super.super;
}

static struct lsh_object *
do_simple_command_K(struct command_simple *ignored UNUSED,
		    struct lsh_object *a)
{
  return &make_command_K_1(a)->super;
}

struct command_simple command_K = STATIC_COMMAND_SIMPLE(do_simple_command_K);

257
258
259
/* ((S f) g)x == (f x)(g x) */

/* Continuation called after evaluating (f x) */
260
/* GABA:
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
   (class
     (name command_S_continuation)
     (super command_frame)
     (vars
       (g object command)
       (x object lsh_object)))
*/

static int do_command_S_continuation(struct command_continuation *c,
				     struct lsh_object *value)
{
  CAST(command_S_continuation, self, c);
  CAST_SUBTYPE(command, op, value);
  return COMMAND_CALL(self->g, self->x, make_apply(op, self->super.up));
}

277
/* Represents ((S f) g) */
278
/* GABA:
279
280
281
282
283
284
285
286
   (class
     (name command_S_2)
     (super command_simple)
     (vars
       (f object command)
       (g object command)))
*/

287
288
289
290
291
292
293
294
295
296
297
298
299
300
static int do_command_S_2(struct command *s,
			  struct lsh_object *x,
			  struct command_continuation *up)
{
  CAST(command_S_2, self, s);
  NEW(command_S_continuation, c);
  c->g = self->g;
  c->x = x;
  c->super.up = up;
  c->super.super.c = do_command_S_continuation;
  
  return COMMAND_CALL(self->f, x, &c->super.super);
}

301
302
303
static struct lsh_object *
do_simple_command_S_2(struct command_simple *s,
		      struct lsh_object *x)
304
{
305
306
307
308
  CAST(command_S_2, self, s);
  CAST_SUBTYPE(command_simple, fs, self->f);
  CAST_SUBTYPE(command_simple, gs, self->g);
  CAST_SUBTYPE(command_simple, op, COMMAND_SIMPLE(fs, x));
309
  
310
311
  return COMMAND_SIMPLE(op, COMMAND_SIMPLE(gs, x));
}
312

313
314
struct command *make_command_S_2(struct command *f,
				 struct command *g)
315
316
317
318
319
320
321
322
{
  NEW(command_S_2, res);
  res->f = f;
  res->g = g;
  res->super.super.call = do_command_S_2;
  res->super.call_simple = do_simple_command_S_2;
  
  return &res->super.super;
323
324
}

325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
static struct lsh_object *collect_S_2(struct collect_info_2 *info,
				      struct lsh_object *f,
				      struct lsh_object *g)
{
  CAST_SUBTYPE(command, cf, f);
  CAST_SUBTYPE(command, cg, g);
  assert(!info);
  
  return &make_command_S_2(cf, cg)->super;
}

struct collect_info_2 collect_info_S_2 =
STATIC_COLLECT_2_FINAL(collect_S_2);

struct collect_info_1 command_S =
STATIC_COLLECT_1(&collect_info_S_2);

#if 0
343
/* Represents (S f) */
344
/* GABA:
345
346
347
348
349
350
351
352
353
354
   (class
     (name command_S_1)
     (super command_simple)
     (vars
       (f object command)))
*/

static struct lsh_object *
do_simple_command_S_1(struct command_simple *s,
		      struct lsh_object *a)
355
{
356
  CAST(command_S_1, self, s);
357
  CAST_SUBTYPE(command, arg, a);
358
359
360
361

  return &make_command_S_2(self->f, arg)->super;
}

362
struct command *make_command_S_1(struct command *f)
363
{
364
  NEW(command_S_1, res);
365
366
367
  res->f = f;
  res->super.super.call = do_call_simple_command;
  res->super.call_simple = do_simple_command_S_1;
368

369
  return &res->super.super;
370
371
}

372
373
374
375
376
377
378
static struct lsh_object *
do_simple_command_S(struct command_simple *ignored UNUSED,
		    struct lsh_object *a)
{
  CAST_SUBTYPE(command, arg, a);
  return &make_command_S_1(arg)->super;
}
379

380
struct command_simple command_S = STATIC_COMMAND_SIMPLE(do_simple_command_S);
381

382
383
384
385
386
struct lsh_object *gaba_apply_S_1(struct lsh_object *f)
{
  CAST_SUBTYPE(command, cf, f);
  return &make_command_S_1(cf)->super;
}
387
388
#endif
#if 0
389

390
/* ((B f) g) x == (f (g x)) */
391

392
/* Represents ((B f) g) */
393
/* GABA:
394
395
   (class
     (name command_B_2)
396
     (super command_simple)
397
     (vars
398
399
       (f object command)
       (g object command)))
400
401
402
*/

static int do_command_B_2(struct command *s,
403
			  struct lsh_object *x,
404
405
406
			  struct command_continuation *c)
{
  CAST(command_B_2, self, s);
407
  return COMMAND_CALL(self->g, x, make_apply(self->f, c));
408
409
}

410
411
static struct lsh_object *do_simple_command_B_2(struct command_simple *s,
						struct lsh_object *x)
412
{
413
414
415
416
  CAST(command_B_2, self, s);
  CAST_SUBTYPE(command_simple, fs, self->f);
  CAST_SUBTYPE(command_simple, gs, self->g);
  return COMMAND_SIMPLE(fs, COMMAND_SIMPLE(gs, x));
417
418
}

419
420
421
static struct command *
make_command_B_2(struct command *f,
		 struct command *g)
422
{
423
424
425
426
427
  NEW(command_B_2, res);
  res->f = f;
  res->g = g;
  res->super.super.call = do_command_B_2;
  res->super.call_simple = do_simple_command_B_2;
428

429
  return &res->super.super;
430
431
}

432
/* Represents (B f) */
433
/* GABA:
434
   (class
435
436
     (name command_B_1)
     (super command_simple)
Niels Möller's avatar
Niels Möller committed
437
     (vars
438
       (f object command)))
Niels Möller's avatar
Niels Möller committed
439
440
*/

441
442
443
static struct lsh_object *
do_simple_command_B_1(struct command_simple *s,
		      struct lsh_object *a)
Niels Möller's avatar
Niels Möller committed
444
{
445
446
447
448
  CAST(command_B_1, self, s);
  CAST_SUBTYPE(command, g, a);

  return &make_command_B_2(self->f, g)->super;
Niels Möller's avatar
Niels Möller committed
449
450
}

451
static struct command *make_command_B_1(struct command *f)
Niels Möller's avatar
Niels Möller committed
452
{
453
454
455
456
  NEW(command_B_1, res);
  res->f = f;
  res->super.super.call = do_call_simple_command;
  res->super.call_simple = do_simple_command_B_1;
Niels Möller's avatar
Niels Möller committed
457

458
  return &res->super.super;
Niels Möller's avatar
Niels Möller committed
459
460
}

461
462
463
static struct lsh_object *
do_simple_command_B(struct command_simple *ignored UNUSED,
		    struct lsh_object *a)
Niels Möller's avatar
Niels Möller committed
464
{
465
466
  CAST_SUBTYPE(command, f, a);
  return &make_command_B_1(f)->super;
Niels Möller's avatar
Niels Möller committed
467
}
468
469

struct command_simple command_B = STATIC_COMMAND_SIMPLE(do_simple_command_B);
470

471

472
/* Returned by listen */
473
/* GABA:
474
475
476
477
478
479
480
481
   (class
     (name listen_value)
     (vars
       (fd . int)
       (peername string)
       (peerport . int)))
*/

482
/* GABA:
483
484
485
486
487
488
489
490
   (class
     (name listen_command)
     (super command)
     (vars
       (backend object io_backend)
       (address . "struct sockaddr_in")))
*/

491
/* GABA:
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
   (class
     (name listen_command_callback)
     (super fd_listen_callback)
     (vars
       (c object command_continuation)))
*/

static int do_listen_continue(struct fd_listen_callback *s, int fd,
			      size_t addr_len UNUSED,
			      struct sockaddr *peer UNUSED )
{
  CAST(listen_command_callback, self, s);
  NEW(listen_value, res);
  res->fd = fd;
  /* Parse peer addr */
  return COMMAND_RETURN(self->c, &res->super);
}

static struct fd_listen_callback *
make_listen_command_callback(struct command_continuation *c)
{
  NEW(listen_command_callback, closure);
  closure->c = c;
  closure->super.f = do_listen_continue;
  
  return &closure->super;
}

static int do_listen_call(struct command *s,
			  struct lsh_object *arg,
			  struct command_continuation *c)		  
{
  CAST(listen_command, self, s);
  CAST(ssh_connection, connection, arg);
  struct listen_fd *fd = io_listen(self->backend,
				   &self->address,
				   make_listen_command_callback(c));

  if (!fd)
    COMMAND_RETURN(c, NULL);
  
  REMEMBER_RESOURCE(connection->resources,
		    &fd->super.super);

  return LSH_OK | LSH_GOON;
}
Niels Möller's avatar
Niels Möller committed
538

539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
struct command *make_listen_command(struct io_backend *backend,
				    struct lsh_string *interface,
				    UINT32 port)
{
  NEW(listen_command, self);
  self->backend = backend;
  if (!tcp_addr(&self->address,
		interface->length,
		interface->data,
		port))
    {
      KILL(self);
      return NULL;
    }
  self->super.call = do_listen_call;

  return &self->super;
}
557
#endif
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
#if 0
/* xxCLASS:
   (class
     (name command_compose_continuation)
     (super command_continuation)
     (vars
       (f object command)
       (c object command_continuation)))
*/

static int do_continue_compose(struct command_continuation *c,
			       struct lsh_object *value)
{
  CAST(compose_continuation, self, c);
  return COMMAND_CALL(self->f, value, self->c);
}

static struct command_continuation *
make_compose_continuation(struct command *f, struct command_continuation *c)
{
  NEW(compose_continuation, self);
  self->f = f;
  self->c = c;

  return &self->super;
}

/* xxCLASS:
   (class
     (name command_compose)
     (super command)
     (vars
       (arg object command)
       (f object command)))
*/

int do_compose_call(struct command_continuation *c, lsh_object *value)
{
  CAST(command_compose, self, c);
  return COMMAND_CALL(self->arg,
		      make_compose_continuation(self->f, c));
}
#endif