debug.c 3.22 KB
Newer Older
1
2
/* debug.c
 *
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
 *
 *
 * $Id$ */

/* lsh, an implementation of the ssh protocol
 *
 * Copyright (C) 1998 Niels Mller
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as
 * published by the Free Software Foundation; either version 2 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
J.H.M. Dassen's avatar
J.H.M. Dassen committed
23
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
24
25
 */

Niels Möller's avatar
Niels Möller committed
26
#include "debug.h"
27
28
29
30
31

#include "connection.h"
#include "format.h"
#include "parse.h"
#include "ssh.h"
Niels Möller's avatar
Niels Möller committed
32
#include "xalloc.h"
33
#include "werror.h"
Niels Möller's avatar
Niels Möller committed
34

35
36
#include "debug.c.x"

37
/* GABA:
38
39
40
41
   (class
     (name packet_debug)
     (super abstract_write_pipe)
     (vars
42
       (prefix simple "const char *")))
43
44
*/

Niels Möller's avatar
Niels Möller committed
45
46
47
48
static void
do_debug(struct abstract_write *w,
	 struct lsh_string *packet,
	 struct exception_handler *e)
49
{
50
  CAST(packet_debug, closure, w);
51
  
52
  debug("DEBUG: received packet %xS\n", packet);
53
  
Niels Möller's avatar
Niels Möller committed
54
  A_WRITE(closure->super.next, packet, e);
55
56
}

Niels Möller's avatar
Niels Möller committed
57
struct abstract_write *
58
make_packet_debug(struct abstract_write *continuation, const char *prefix)
59
{
60
  NEW(packet_debug, closure);
61

62
  closure->super.super.write = do_debug;
Niels Möller's avatar
Niels Möller committed
63
  closure->super.next = continuation;
64
  closure->prefix = prefix;
65

66
  return &closure->super.super;
67
68
}

69
static struct lsh_string *make_debug_packet(const char *msg, int always_display)
70
71
72
73
74
75
76
77
78
79
{
  return ssh_format("%c%c%z%z",
		    SSH_MSG_DEBUG,
		    always_display,
		    msg,
		    /* Empty language tag */ 
		    "");
}

/* Send a debug message to the other end. */
Niels Möller's avatar
Niels Möller committed
80
void send_debug(struct ssh_connection *connection, const char *msg, int always_display)
81
{
Niels Möller's avatar
Niels Möller committed
82
83
  if (debug_flag)
    C_WRITE(connection, make_debug_packet(msg, always_display));
84
85
}

Niels Möller's avatar
Niels Möller committed
86
void send_verbose(struct ssh_connection *connection, const char *msg, int always_display)
87
{
Niels Möller's avatar
Niels Möller committed
88
89
  if (verbose_flag)
    C_WRITE(connection, make_debug_packet(msg, always_display));
90
91
}

Niels Möller's avatar
Niels Möller committed
92
93
94
95
static void
do_rec_debug(struct packet_handler *self UNUSED,
	     struct ssh_connection *connection UNUSED,
	     struct lsh_string *packet)
96
97
{
  struct simple_buffer buffer;
Niels Möller's avatar
Niels Möller committed
98
99
  unsigned msg_number;
  unsigned always_display;
100
101
102
103
104
105
106
107
108
109
110
111
112
  UINT32 length;
  UINT8 *msg;
  int language;
  
  simple_buffer_init(&buffer, packet->length, packet->data);

  if (!(parse_uint8(&buffer, &msg_number)
	&& parse_uint8(&buffer, &always_display)
	&& parse_string(&buffer, &length, &msg)
	&& parse_atom(&buffer, &language)
	&& parse_eod(&buffer)))
    {
      lsh_string_free(packet);
Niels Möller's avatar
Niels Möller committed
113
114
115
116
117
      EXCEPTION_RAISE
	(connection->e,
	 make_protocol_exception(SSH_DISCONNECT_PROTOCOL_ERROR,
				 "Invalid DEBUG message."));
				 
118
119
    }
  else
Niels Möller's avatar
Niels Möller committed
120
121
122
    {
      if (always_display)
	werror("Received debug: %ups\n", length, msg);
123

Niels Möller's avatar
Niels Möller committed
124
125
126
127
128
      else
	verbose("Received debug: %ups\n", length, msg);
      
      lsh_string_free(packet);
    }
129
130
131
132
}

struct packet_handler *make_rec_debug_handler(void)
{
133
  NEW(packet_handler, self);
134
135
136
137
138

  self->handler = do_rec_debug;

  return self;
}