Mirror/zeek
1
0
Fork 0
mirror of https://github.com/zeek/zeek.git synced 2025-10-08 01:28:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

SSH: Cleanup code style.

Browse source
This commit is contained in:
Vlad Grigorescu 2015-03-03 16:22:22 -05:00
parent 3190ca275e
commit b76f7d9fa7
5 changed files with 278 additions and 285 deletions
Download patch file Download diff file Expand all files Collapse all files

30
src/analyzer/protocol/ssh/Plugin.cc
View file

@ -1,26 +1,24 @@
// See the file in the main distribution directory for copyright. // See the file in the main distribution directory for copyright.
#include "plugin/Plugin.h" #include "plugin/Plugin.h"
#include "SSH.h" #include "SSH.h"
namespace plugin { namespace plugin {
namespace Bro_SSH { namespace Bro_SSH {
class Plugin : public plugin::Plugin { class Plugin : public plugin::Plugin {
public: public:
plugin::Configuration Configure() plugin::Configuration Configure()
{ {
AddComponent(new ::analyzer::Component("SSH", ::analyzer::SSH::SSH_Analyzer::Instantiate)); AddComponent(new ::analyzer::Component("SSH", ::analyzer::SSH::SSH_Analyzer::Instantiate));
plugin::Configuration config;
config.name = "Bro::SSH";
config.description = "Secure Shell analyzer";
return config;
}
} plugin;
plugin::Configuration config;
config.name = "Bro::SSH";
config.description = "Secure Shell analyzer";
return config;
} }
} plugin; }
}
}

10
src/analyzer/protocol/ssh/SSH.cc
View file

@ -12,9 +12,7 @@
using namespace analyzer::SSH; using namespace analyzer::SSH;
SSH_Analyzer::SSH_Analyzer(Connection* c) SSH_Analyzer::SSH_Analyzer(Connection* c)
: tcp::TCP_ApplicationAnalyzer("SSH", c)
: tcp::TCP_ApplicationAnalyzer("SSH", c)
{ {
interp = new binpac::SSH::SSH_Conn(this); interp = new binpac::SSH::SSH_Conn(this);
had_gap = false; had_gap = false;
@ -22,7 +20,7 @@ SSH_Analyzer::SSH_Analyzer(Connection* c)
skipped_banner = false; skipped_banner = false;
service_accept_size = 0; service_accept_size = 0;
userauth_failure_size = 0; userauth_failure_size = 0;
} }
SSH_Analyzer::~SSH_Analyzer() SSH_Analyzer::~SSH_Analyzer()
{ {
@ -31,12 +29,10 @@ SSH_Analyzer::~SSH_Analyzer()
void SSH_Analyzer::Done() void SSH_Analyzer::Done()
{ {
tcp::TCP_ApplicationAnalyzer::Done(); tcp::TCP_ApplicationAnalyzer::Done();
interp->FlowEOF(true); interp->FlowEOF(true);
interp->FlowEOF(false); interp->FlowEOF(false);
} }
void SSH_Analyzer::EndpointEOF(bool is_orig) void SSH_Analyzer::EndpointEOF(bool is_orig)
@ -62,7 +58,7 @@ void SSH_Analyzer::DeliverStream(int len, const u_char* data, bool orig)
{ {
if ( ssh_encrypted_packet ) if ( ssh_encrypted_packet )
BifEvent::generate_ssh_encrypted_packet(interp->bro_analyzer(), interp->bro_analyzer()->Conn(), BifEvent::generate_ssh_encrypted_packet(interp->bro_analyzer(), interp->bro_analyzer()->Conn(),
orig, len); orig, len);
if ( !auth_decision_made ) if ( !auth_decision_made )
ProcessEncrypted(len, orig); ProcessEncrypted(len, orig);

52
src/analyzer/protocol/ssh/SSH.h
View file

@ -8,41 +8,41 @@
#include "analyzer/protocol/tcp/TCP.h" #include "analyzer/protocol/tcp/TCP.h"
#include "ssh_pac.h" #include "ssh_pac.h"
namespace analyzer { namespace SSH { namespace analyzer {
namespace SSH {
class SSH_Analyzer : public tcp::TCP_ApplicationAnalyzer {
class SSH_Analyzer : public tcp::TCP_ApplicationAnalyzer { public:
SSH_Analyzer(Connection* conn);
virtual ~SSH_Analyzer();
public: // Overriden from Analyzer.
SSH_Analyzer(Connection* conn); virtual void Done();
virtual ~SSH_Analyzer(); virtual void DeliverStream(int len, const u_char* data, bool orig);
virtual void Undelivered(uint64 seq, int len, bool orig);
// Overriden from Analyzer. // Overriden from tcp::TCP_ApplicationAnalyzer.
virtual void Done(); virtual void EndpointEOF(bool is_orig);
virtual void DeliverStream(int len, const u_char* data, bool orig);
virtual void Undelivered(uint64 seq, int len, bool orig);
// Overriden from tcp::TCP_ApplicationAnalyzer. static analyzer::Analyzer* Instantiate(Connection* conn)
virtual void EndpointEOF(bool is_orig); { return new SSH_Analyzer(conn); }
static analyzer::Analyzer* Instantiate(Connection* conn) protected:
{ return new SSH_Analyzer(conn); } binpac::SSH::SSH_Conn* interp;
protected: void ProcessEncrypted(int len, bool orig);
binpac::SSH::SSH_Conn* interp;
void ProcessEncrypted(int len, bool orig); bool had_gap;
bool had_gap; // Packet analysis stuff
bool auth_decision_made;
bool skipped_banner;
// Packet analysis stuff int service_accept_size;
bool auth_decision_made; int userauth_failure_size;
bool skipped_banner;
int service_accept_size; };
int userauth_failure_size;
};
} } // namespace analyzer::*
}
}
#endif #endif

24
src/analyzer/protocol/ssh/ssh-analyzer.pac
View file

@ -54,14 +54,14 @@ refine flow SSH_Flow += {
if ( ssh_client_version && ${msg.is_orig } ) if ( ssh_client_version && ${msg.is_orig } )
{ {
BifEvent::generate_ssh_client_version(connection()->bro_analyzer(), BifEvent::generate_ssh_client_version(connection()->bro_analyzer(),
connection()->bro_analyzer()->Conn(), connection()->bro_analyzer()->Conn(),
bytestring_to_val(${msg.version})); bytestring_to_val(${msg.version}));
} }
else if ( ssh_server_version ) else if ( ssh_server_version )
{ {
BifEvent::generate_ssh_server_version(connection()->bro_analyzer(), BifEvent::generate_ssh_server_version(connection()->bro_analyzer(),
connection()->bro_analyzer()->Conn(), connection()->bro_analyzer()->Conn(),
bytestring_to_val(${msg.version})); bytestring_to_val(${msg.version}));
} }
return true; return true;
%} %}
@ -86,8 +86,8 @@ refine flow SSH_Flow += {
result->Assign(10, new Val(${msg.is_orig}, TYPE_BOOL)); result->Assign(10, new Val(${msg.is_orig}, TYPE_BOOL));
BifEvent::generate_ssh_capabilities(connection()->bro_analyzer(), BifEvent::generate_ssh_capabilities(connection()->bro_analyzer(),
connection()->bro_analyzer()->Conn(), bytestring_to_val(${msg.cookie}), connection()->bro_analyzer()->Conn(), bytestring_to_val(${msg.cookie}),
result); result);
} }
return true; return true;
%} %}
@ -97,8 +97,8 @@ refine flow SSH_Flow += {
if ( ssh_server_host_key ) if ( ssh_server_host_key )
{ {
BifEvent::generate_ssh_server_host_key(connection()->bro_analyzer(), BifEvent::generate_ssh_server_host_key(connection()->bro_analyzer(),
connection()->bro_analyzer()->Conn(), connection()->bro_analyzer()->Conn(),
bytestring_to_val(${key})); bytestring_to_val(${key}));
} }
return true; return true;
%} %}
@ -108,9 +108,9 @@ refine flow SSH_Flow += {
if ( ssh1_server_host_key ) if ( ssh1_server_host_key )
{ {
BifEvent::generate_ssh1_server_host_key(connection()->bro_analyzer(), BifEvent::generate_ssh1_server_host_key(connection()->bro_analyzer(),
connection()->bro_analyzer()->Conn(), connection()->bro_analyzer()->Conn(),
bytestring_to_val(${p}), bytestring_to_val(${p}),
bytestring_to_val(${e})); bytestring_to_val(${e}));
} }
return true; return true;
%} %}
@ -148,5 +148,5 @@ refine typeattr SSH_ECC_REPLY += &let {
}; };
refine typeattr SSH1_PUBLIC_KEY += &let { refine typeattr SSH1_PUBLIC_KEY += &let {
proc: bool = $context.flow.proc_ssh1_server_host_key(host_key_p.val, host_key_e.val); proc: bool = $context.flow.proc_ssh1_server_host_key(host_key_p.val, host_key_e.val);
}; };

411
src/analyzer/protocol/ssh/ssh-protocol.pac
View file

@ -5,14 +5,14 @@ enum version {
}; };
enum state { enum state {
VERSION_EXCHANGE = 0, VERSION_EXCHANGE = 0,
KEX_INIT = 1, KEX_INIT = 1,
KEX_DH_GEX = 2, KEX_DH_GEX = 2,
KEX_DH = 3, KEX_DH = 3,
KEX_ECC = 4, KEX_ECC = 4,
KEX_GSS = 5, KEX_GSS = 5,
KEX_RSA = 6, KEX_RSA = 6,
ENCRYPTED = 7, ENCRYPTED = 7,
}; };
# diffie-hellman-group1-sha1 [RFC4253] Section 8.1 # diffie-hellman-group1-sha1 [RFC4253] Section 8.1
@ -25,19 +25,19 @@ enum KEX_DH_message_id {
# diffie-hellman-group-exchange-sha1 [RFC4419] Section 4.1 # diffie-hellman-group-exchange-sha1 [RFC4419] Section 4.1
# diffie-hellman-group-exchange-sha256 [RFC4419] Section 4.2 # diffie-hellman-group-exchange-sha256 [RFC4419] Section 4.2
enum KEX_DH_GEX_message_id { enum KEX_DH_GEX_message_id {
SSH_MSG_KEX_DH_GEX_REQUEST_OLD = 30, SSH_MSG_KEX_DH_GEX_REQUEST_OLD = 30,
SSH_MSG_KEX_DH_GEX_GROUP = 31, SSH_MSG_KEX_DH_GEX_GROUP = 31,
SSH_MSG_KEX_DH_GEX_INIT = 32, SSH_MSG_KEX_DH_GEX_INIT = 32,
SSH_MSG_KEX_DH_GEX_REPLY = 33, SSH_MSG_KEX_DH_GEX_REPLY = 33,
SSH_MSG_KEX_DH_GEX_REQUEST = 34, SSH_MSG_KEX_DH_GEX_REQUEST = 34,
}; };
# rsa1024-sha1 [RFC4432] # rsa1024-sha1 [RFC4432]
# rsa2048-sha256 [RFC4432] # rsa2048-sha256 [RFC4432]
enum KEX_RSA_message_id { enum KEX_RSA_message_id {
SSH_MSG_KEXRSA_PUBKEY = 30, SSH_MSG_KEXRSA_PUBKEY = 30,
SSH_MSG_KEXRSA_SECRET = 31, SSH_MSG_KEXRSA_SECRET = 31,
SSH_MSG_KEXRSA_DONE = 32, SSH_MSG_KEXRSA_DONE = 32,
}; };
# gss-group1-sha1-* [RFC4462] Section 2.3 # gss-group1-sha1-* [RFC4462] Section 2.3
@ -45,174 +45,175 @@ enum KEX_RSA_message_id {
# gss-gex-sha1-* [RFC4462] Section 2.5 # gss-gex-sha1-* [RFC4462] Section 2.5
# gss-* [RFC4462] Section 2.6 # gss-* [RFC4462] Section 2.6
enum KEX_GSS_message_id { enum KEX_GSS_message_id {
SSH_MSG_KEXGSS_INIT = 30, SSH_MSG_KEXGSS_INIT = 30,
SSH_MSG_KEXGSS_CONTINUE = 31, SSH_MSG_KEXGSS_CONTINUE = 31,
SSH_MSG_KEXGSS_COMPLETE = 32, SSH_MSG_KEXGSS_COMPLETE = 32,
SSH_MSG_KEXGSS_HOSTKEY = 33, SSH_MSG_KEXGSS_HOSTKEY = 33,
SSH_MSG_KEXGSS_ERROR = 34, SSH_MSG_KEXGSS_ERROR = 34,
SSH_MSG_KEXGSS_GROUPREQ = 40, SSH_MSG_KEXGSS_GROUPREQ = 40,
SSH_MSG_KEXGSS_GROUP = 41, SSH_MSG_KEXGSS_GROUP = 41,
}; };
# ecdh-sha2-* [RFC5656] # ecdh-sha2-* [RFC5656]
enum KEX_ECDH_message_id { enum KEX_ECDH_message_id {
SSH_MSG_KEX_ECDH_INIT = 30, SSH_MSG_KEX_ECDH_INIT = 30,
SSH_MSG_KEX_ECDH_REPLY = 31, SSH_MSG_KEX_ECDH_REPLY = 31,
}; };
# ecmqv-sha2 [RFC5656] # ecmqv-sha2 [RFC5656]
enum KEX_ECMQV_message_id { enum KEX_ECMQV_message_id {
SSH_MSG_ECMQV_INIT = 30, SSH_MSG_ECMQV_INIT = 30,
SSH_MSG_ECMQV_REPLY = 31, SSH_MSG_ECMQV_REPLY = 31,
}; };
enum ssh1_message_id { enum ssh1_message_id {
SSH_MSG_NONE = 0, SSH_MSG_NONE = 0,
SSH_MSG_DISCONNECT = 1, SSH_MSG_DISCONNECT = 1,
SSH_SMSG_PUBLIC_KEY = 2, SSH_SMSG_PUBLIC_KEY = 2,
SSH_CMSG_SESSION_KEY = 3, SSH_CMSG_SESSION_KEY = 3,
SSH_CMSG_USER = 4, SSH_CMSG_USER = 4,
SSH_CMSG_AUTH_RHOSTS = 5, SSH_CMSG_AUTH_RHOSTS = 5,
SSH_CMSG_AUTH_RSA = 6, SSH_CMSG_AUTH_RSA = 6,
SSH_SMSG_AUTH_RSA_CHALLENGE = 7, SSH_SMSG_AUTH_RSA_CHALLENGE = 7,
SSH_CMSG_AUTH_RSA_RESPONSE = 8, SSH_CMSG_AUTH_RSA_RESPONSE = 8,
SSH_CMSG_AUTH_PASSWORD = 9, SSH_CMSG_AUTH_PASSWORD = 9,
SSH_CMSG_REQUEST_PTY = 10, SSH_CMSG_REQUEST_PTY = 10,
SSH_CMSG_WINDOW_SIZE = 11, SSH_CMSG_WINDOW_SIZE = 11,
SSH_CMSG_EXEC_SHELL = 12, SSH_CMSG_EXEC_SHELL = 12,
SSH_CMSG_EXEC_CMD = 13, SSH_CMSG_EXEC_CMD = 13,
SSH_SMSG_SUCCESS = 14, SSH_SMSG_SUCCESS = 14,
SSH_SMSG_FAILURE = 15, SSH_SMSG_FAILURE = 15,
SSH_CMSG_STDIN_DATA = 16, SSH_CMSG_STDIN_DATA = 16,
SSH_SMSG_STDOUT_DATA = 17, SSH_SMSG_STDOUT_DATA = 17,
SSH_SMSG_STDERR_DATA = 18, SSH_SMSG_STDERR_DATA = 18,
SSH_CMSG_EOF = 19, SSH_CMSG_EOF = 19,
SSH_SMSG_EXITSTATUS = 20, SSH_SMSG_EXITSTATUS = 20,
SSH_MSG_CHANNEL_OPEN_CONFIRMATION = 21, SSH_MSG_CHANNEL_OPEN_CONFIRMATION = 21,
SSH_MSG_CHANNEL_OPEN_FAILURE = 22, SSH_MSG_CHANNEL_OPEN_FAILURE = 22,
SSH_MSG_CHANNEL_DATA = 23, SSH_MSG_CHANNEL_DATA = 23,
SSH_MSG_CHANNEL_CLOSE = 24, SSH_MSG_CHANNEL_CLOSE = 24,
SSH_MSG_CHANNEL_CLOSE_CONFIRMATION = 25, SSH_MSG_CHANNEL_CLOSE_CONFIRMATION = 25,
SSH_CMSG_X11_REQUEST_FORWARDING_OLD = 26, SSH_CMSG_X11_REQUEST_FORWARDING_OLD = 26,
SSH_SMSG_X11_OPEN = 27, SSH_SMSG_X11_OPEN = 27,
SSH_CMSG_PORT_FORWARD_REQUEST = 28, SSH_CMSG_PORT_FORWARD_REQUEST = 28,
SSH_MSG_PORT_OPEN = 29, SSH_MSG_PORT_OPEN = 29,
SSH_CMSG_AGENT_REQUEST_FORWARDING = 30, SSH_CMSG_AGENT_REQUEST_FORWARDING = 30,
SSH_SMSG_AGENT_OPEN = 31, SSH_SMSG_AGENT_OPEN = 31,
SSH_MSG_IGNORE = 32, SSH_MSG_IGNORE = 32,
SSH_CMSG_EXIT_CONFIRMATION = 33, SSH_CMSG_EXIT_CONFIRMATION = 33,
SSH_CMSG_X11_REQUEST_FORWARDING = 34, SSH_CMSG_X11_REQUEST_FORWARDING = 34,
SSH_CMSG_AUTH_RHOSTS_RSA = 35, SSH_CMSG_AUTH_RHOSTS_RSA = 35,
SSH_MSG_DEBUG = 36, SSH_MSG_DEBUG = 36,
SSH_CMSG_REQUEST_COMPRESSION = 37, SSH_CMSG_REQUEST_COMPRESSION = 37,
SSH_CMSG_MAX_PACKET_SIZE = 38, SSH_CMSG_MAX_PACKET_SIZE = 38,
SSH_CMSG_AUTH_TIS = 39, SSH_CMSG_AUTH_TIS = 39,
SSH_SMSG_AUTH_TIS_CHALLENGE = 40, SSH_SMSG_AUTH_TIS_CHALLENGE = 40,
SSH_CMSG_AUTH_TIS_RESPONSE = 41, SSH_CMSG_AUTH_TIS_RESPONSE = 41,
SSH_CMSG_AUTH_KERBEROS = 42, SSH_CMSG_AUTH_KERBEROS = 42,
SSH_SMSG_AUTH_KERBEROS_RESPONSE = 43, SSH_SMSG_AUTH_KERBEROS_RESPONSE = 43,
SSH_CMSG_HAVE_KERBEROS_TGT = 44, SSH_CMSG_HAVE_KERBEROS_TGT = 44,
}; };
enum ssh2_message_id { enum ssh2_message_id {
MSG_DISCONNECT = 1, MSG_DISCONNECT = 1,
MSG_IGNORE = 2, MSG_IGNORE = 2,
MSG_UNIMPLEMENTED = 3, MSG_UNIMPLEMENTED = 3,
MSG_DEBUG = 4, MSG_DEBUG = 4,
MSG_SERVICE_REQUEST = 5, MSG_SERVICE_REQUEST = 5,
MSG_SERVICE_ACCEPT = 6, MSG_SERVICE_ACCEPT = 6,
MSG_KEXINIT = 20, MSG_KEXINIT = 20,
MSG_NEWKEYS = 21, MSG_NEWKEYS = 21,
}; };
## SSH Generic ## SSH Generic
type SSH_PDU(is_orig: bool) = case $context.connection.get_state(is_orig) of { type SSH_PDU(is_orig: bool) = case $context.connection.get_state(is_orig) of {
VERSION_EXCHANGE -> version: SSH_Version(is_orig); VERSION_EXCHANGE -> version : SSH_Version(is_orig);
KEX_INIT -> kex: SSH_Key_Exchange(is_orig); KEX_INIT -> kex : SSH_Key_Exchange(is_orig);
KEX_DH_GEX -> kex_dh_gex: SSH_Key_Exchange_DH_GEX(is_orig); KEX_DH_GEX -> kex_dh_gex : SSH_Key_Exchange_DH_GEX(is_orig);
KEX_DH -> kex_dh: SSH_Key_Exchange_DH(is_orig); KEX_DH -> kex_dh : SSH_Key_Exchange_DH(is_orig);
KEX_ECC -> kex_ecc: SSH_Key_Exchange_ECC(is_orig); KEX_ECC -> kex_ecc : SSH_Key_Exchange_ECC(is_orig);
KEX_GSS -> kex_gss: SSH_Key_Exchange_GSS(is_orig); KEX_GSS -> kex_gss : SSH_Key_Exchange_GSS(is_orig);
KEX_RSA -> kex_rsa: SSH_Key_Exchange_RSA(is_orig); KEX_RSA -> kex_rsa : SSH_Key_Exchange_RSA(is_orig);
} &byteorder=bigendian; } &byteorder=bigendian;
type SSH_Version(is_orig: bool) = record { type SSH_Version(is_orig: bool) = record {
version: bytestring &oneline; version : bytestring &oneline;
pad: bytestring &length=0 &transient; pad : bytestring &length=0 &transient;
} &let { } &let {
update_state : bool = $context.connection.update_state(KEX_INIT, is_orig); update_state : bool = $context.connection.update_state(KEX_INIT, is_orig);
update_version: bool = $context.connection.update_version(version, is_orig); update_version : bool = $context.connection.update_version(version, is_orig);
}; };
type SSH_Key_Exchange(is_orig: bool) = case $context.connection.get_version() of { type SSH_Key_Exchange(is_orig: bool) = case $context.connection.get_version() of {
SSH1 -> ssh1_msg: SSH1_Key_Exchange(is_orig); SSH1 -> ssh1_msg : SSH1_Key_Exchange(is_orig);
SSH2 -> ssh2_msg: SSH2_Key_Exchange(is_orig); SSH2 -> ssh2_msg : SSH2_Key_Exchange(is_orig);
}; };
## SSH1 ## SSH1
type SSH1_Key_Exchange(is_orig: bool) = record { type SSH1_Key_Exchange(is_orig: bool) = record {
packet_length: uint32; packet_length : uint32;
pad_fill : bytestring &length = 8 - (packet_length % 8); pad_fill : bytestring &length = 8 - (packet_length % 8);
msg_type : uint8; msg_type : uint8;
message : SSH1_Message(is_orig, msg_type, packet_length-5); message : SSH1_Message(is_orig, msg_type, packet_length - 5);
crc : uint32; crc : uint32;
} &length = packet_length + 4 + 8 - (packet_length % 8); } &length = packet_length + 4 + 8 - (packet_length % 8);
type SSH1_Message(is_orig: bool, msg_type: uint8, length: uint32) = case msg_type of { type SSH1_Message(is_orig: bool, msg_type: uint8, length: uint32) = case msg_type of {
SSH_SMSG_PUBLIC_KEY -> public_key: SSH1_PUBLIC_KEY(length); SSH_SMSG_PUBLIC_KEY -> public_key : SSH1_PUBLIC_KEY(length);
SSH_CMSG_SESSION_KEY -> session_key: SSH1_SESSION_KEY(length); SSH_CMSG_SESSION_KEY -> session_key : SSH1_SESSION_KEY(length);
} &let { } &let {
detach: bool = $context.connection.update_state(ENCRYPTED, is_orig); detach : bool=$context.connection.update_state(ENCRYPTED, is_orig);
}; };
type SSH1_PUBLIC_KEY(length: uint32) = record { type SSH1_PUBLIC_KEY(length: uint32) = record {
cookie : bytestring &length=8; cookie : bytestring &length=8;
server_key : uint32; server_key : uint32;
server_key_p : ssh1_mp_int; server_key_p : ssh1_mp_int;
server_key_e : ssh1_mp_int; server_key_e : ssh1_mp_int;
host_key : uint32; host_key : uint32;
host_key_p : ssh1_mp_int; host_key_p : ssh1_mp_int;
host_key_e : ssh1_mp_int; host_key_e : ssh1_mp_int;
flags : uint32; flags : uint32;
supported_ciphers : uint32; supported_ciphers : uint32;
supported_auths : uint32; supported_auths : uint32;
} &length=length; } &length=length;
type SSH1_SESSION_KEY(length: uint32) = record { type SSH1_SESSION_KEY(length: uint32) = record {
cipher : uint8; cipher : uint8;
cookie : bytestring &length=8; cookie : bytestring &length=8;
session_key : ssh1_mp_int; session_key : ssh1_mp_int;
flags : uint32; flags : uint32;
} &length=length; } &length=length;
type ssh1_mp_int = record { type ssh1_mp_int = record {
len: uint16; len : uint16;
val: bytestring &length=(len+7)/8; val : bytestring &length=(len+7)/8;
}; };
## SSH2 ## SSH2
type SSH2_Key_Exchange_Header = record { type SSH2_Key_Exchange_Header = record {
packet_length : uint32; packet_length : uint32;
padding_length: uint8; padding_length : uint8;
msg_type : uint8; msg_type : uint8;
} &let { } &let {
payload_length: uint32 = packet_length - padding_length - 2; payload_length : uint32 = packet_length - padding_length - 2;
} &length=6; } &length=6;
type SSH2_Key_Exchange(is_orig: bool) = record { type SSH2_Key_Exchange(is_orig: bool) = record {
header : SSH2_Key_Exchange_Header; header : SSH2_Key_Exchange_Header;
payload : SSH2_Message(is_orig, header.msg_type, header.payload_length); payload : SSH2_Message(is_orig, header.msg_type, header.payload_length);
pad : bytestring &length=header.padding_length; pad : bytestring &length=header.padding_length;
} &length=header.packet_length + 4; } &length=header.packet_length + 4;
type SSH2_Message(is_orig: bool, msg_type: uint8, length: uint32) = case msg_type of { type SSH2_Message(is_orig: bool, msg_type: uint8, length: uint32) = case msg_type of {
MSG_KEXINIT -> kexinit: SSH_KEXINIT(length, is_orig); MSG_KEXINIT -> kexinit : SSH_KEXINIT(length, is_orig);
default -> unknown: bytestring &length=length; default -> unknown : bytestring &length=length;
} &let { } &let {
detach: bool = $context.connection.update_state(ENCRYPTED, is_orig) &if(msg_type == MSG_NEWKEYS); detach : bool = $context.connection.update_state(ENCRYPTED, is_orig) &if(msg_type == MSG_NEWKEYS);
}; };
type SSH_KEXINIT(length: uint32, is_orig: bool) = record { type SSH_KEXINIT(length: uint32, is_orig: bool) = record {
@ -225,12 +226,12 @@ type SSH_KEXINIT(length: uint32, is_orig: bool) = record {
mac_algorithms_server_to_client : ssh_string; mac_algorithms_server_to_client : ssh_string;
compression_algorithms_client_to_server : ssh_string; compression_algorithms_client_to_server : ssh_string;
compression_algorithms_server_to_client : ssh_string; compression_algorithms_server_to_client : ssh_string;
languages_client_to_server : ssh_string; languages_client_to_server : ssh_string;
languages_server_to_client : ssh_string; languages_server_to_client : ssh_string;
first_kex_packet_follows : uint8; first_kex_packet_follows : uint8;
reserved : uint32; reserved : uint32;
} &let { } &let {
proc_kex : bool = $context.connection.update_kex(kex_algorithms.val, is_orig); proc_kex : bool= $context.connection.update_kex(kex_algorithms.val, is_orig);
} &length=length; } &length=length;
# KEX_DH exchanges # KEX_DH exchanges
@ -238,12 +239,12 @@ type SSH_KEXINIT(length: uint32, is_orig: bool) = record {
type SSH_Key_Exchange_DH(is_orig: bool) = record { type SSH_Key_Exchange_DH(is_orig: bool) = record {
header : SSH2_Key_Exchange_Header; header : SSH2_Key_Exchange_Header;
payload : SSH_Key_Exchange_DH_Message(is_orig, header.msg_type, header.payload_length); payload : SSH_Key_Exchange_DH_Message(is_orig, header.msg_type, header.payload_length);
pad : bytestring &length=header.padding_length; pad : bytestring &length=header.padding_length;
} &length=header.packet_length + 4; } &length=header.packet_length + 4;
type SSH_Key_Exchange_DH_Message(is_orig: bool, msg_type: uint8, length: uint32) = case msg_type of { type SSH_Key_Exchange_DH_Message(is_orig: bool, msg_type: uint8, length: uint32) = case msg_type of {
SSH_MSG_KEXDH_INIT -> init : SSH_DH_GEX_INIT(length); SSH_MSG_KEXDH_INIT -> init : SSH_DH_GEX_INIT(length);
SSH_MSG_KEXDH_REPLY -> reply : SSH_DH_GEX_REPLY(length); SSH_MSG_KEXDH_REPLY -> reply : SSH_DH_GEX_REPLY(length);
}; };
# KEX_DH_GEX exchanges # KEX_DH_GEX exchanges
@ -251,40 +252,40 @@ type SSH_Key_Exchange_DH_Message(is_orig: bool, msg_type: uint8, length: uint32)
type SSH_Key_Exchange_DH_GEX(is_orig: bool) = record { type SSH_Key_Exchange_DH_GEX(is_orig: bool) = record {
header : SSH2_Key_Exchange_Header; header : SSH2_Key_Exchange_Header;
payload : SSH_Key_Exchange_DH_GEX_Message(is_orig, header.msg_type, header.payload_length); payload : SSH_Key_Exchange_DH_GEX_Message(is_orig, header.msg_type, header.payload_length);
pad : bytestring &length=header.padding_length; pad : bytestring &length=header.padding_length;
} &length=header.packet_length + 4; } &length=header.packet_length + 4;
type SSH_Key_Exchange_DH_GEX_Message(is_orig: bool, msg_type: uint8, length: uint32) = case msg_type of { type SSH_Key_Exchange_DH_GEX_Message(is_orig: bool, msg_type: uint8, length: uint32) = case msg_type of {
SSH_MSG_KEX_DH_GEX_REQUEST_OLD -> request_old : SSH_DH_GEX_REQUEST_OLD; SSH_MSG_KEX_DH_GEX_REQUEST_OLD -> request_old : SSH_DH_GEX_REQUEST_OLD;
SSH_MSG_KEX_DH_GEX_REQUEST -> request : SSH_DH_GEX_REQUEST; SSH_MSG_KEX_DH_GEX_REQUEST -> request : SSH_DH_GEX_REQUEST;
SSH_MSG_KEX_DH_GEX_GROUP -> group : SSH_DH_GEX_GROUP(length); SSH_MSG_KEX_DH_GEX_GROUP -> group : SSH_DH_GEX_GROUP(length);
SSH_MSG_KEX_DH_GEX_INIT -> init : SSH_DH_GEX_INIT(length); SSH_MSG_KEX_DH_GEX_INIT -> init : SSH_DH_GEX_INIT(length);
SSH_MSG_KEX_DH_GEX_REPLY -> reply : SSH_DH_GEX_REPLY(length); SSH_MSG_KEX_DH_GEX_REPLY -> reply : SSH_DH_GEX_REPLY(length);
}; };
type SSH_DH_GEX_REQUEST = record { type SSH_DH_GEX_REQUEST = record {
min: uint32; min : uint32;
n : uint32; n : uint32;
max: uint32; max : uint32;
} &length=12; } &length=12;
type SSH_DH_GEX_REQUEST_OLD = record { type SSH_DH_GEX_REQUEST_OLD = record {
n: uint32; n : uint32;
} &length=4; } &length=4;
type SSH_DH_GEX_GROUP(length: uint32) = record { type SSH_DH_GEX_GROUP(length: uint32) = record {
p: ssh_string; p : ssh_string;
g: ssh_string; g : ssh_string;
} &length=length; } &length=length;
type SSH_DH_GEX_INIT(length: uint32) = record { type SSH_DH_GEX_INIT(length: uint32) = record {
e: ssh_string; e : ssh_string;
} &length=length; } &length=length;
type SSH_DH_GEX_REPLY(length: uint32) = record { type SSH_DH_GEX_REPLY(length: uint32) = record {
k_s : ssh_string; k_s : ssh_string;
f : ssh_string; f : ssh_string;
signature: ssh_string; signature : ssh_string;
} &length=length; } &length=length;
# KEX_RSA exchanges # KEX_RSA exchanges
@ -292,26 +293,26 @@ type SSH_DH_GEX_REPLY(length: uint32) = record {
type SSH_Key_Exchange_RSA(is_orig: bool) = record { type SSH_Key_Exchange_RSA(is_orig: bool) = record {
header : SSH2_Key_Exchange_Header; header : SSH2_Key_Exchange_Header;
payload : SSH_Key_Exchange_RSA_Message(is_orig, header.msg_type, header.payload_length); payload : SSH_Key_Exchange_RSA_Message(is_orig, header.msg_type, header.payload_length);
pad : bytestring &length=header.padding_length; pad : bytestring &length=header.padding_length;
} &length=header.packet_length + 4; } &length=header.packet_length + 4;
type SSH_Key_Exchange_RSA_Message(is_orig: bool, msg_type: uint8, length: uint32) = case msg_type of { type SSH_Key_Exchange_RSA_Message(is_orig: bool, msg_type: uint8, length: uint32) = case msg_type of {
SSH_MSG_KEXRSA_PUBKEY -> pubkey : SSH_RSA_PUBKEY(length); SSH_MSG_KEXRSA_PUBKEY -> pubkey : SSH_RSA_PUBKEY(length);
SSH_MSG_KEXRSA_SECRET -> secret : SSH_RSA_SECRET(length); SSH_MSG_KEXRSA_SECRET -> secret : SSH_RSA_SECRET(length);
SSH_MSG_KEXRSA_DONE -> done : SSH_RSA_DONE(length); SSH_MSG_KEXRSA_DONE -> done : SSH_RSA_DONE(length);
}; };
type SSH_RSA_PUBKEY(length: uint32) = record { type SSH_RSA_PUBKEY(length: uint32) = record {
k_s: ssh_string; k_s : ssh_string;
k_t: ssh_string; k_t : ssh_string;
} &length=length; } &length=length;
type SSH_RSA_SECRET(length: uint32) = record { type SSH_RSA_SECRET(length: uint32) = record {
encrypted_payload: ssh_string; encrypted_payload : ssh_string;
} &length=length; } &length=length;
type SSH_RSA_DONE(length: uint32) = record { type SSH_RSA_DONE(length: uint32) = record {
signature: ssh_string; signature : ssh_string;
} &length=length; } &length=length;
# KEX_GSS exchanges # KEX_GSS exchanges
@ -319,47 +320,47 @@ type SSH_RSA_DONE(length: uint32) = record {
type SSH_Key_Exchange_GSS(is_orig: bool) = record { type SSH_Key_Exchange_GSS(is_orig: bool) = record {
header : SSH2_Key_Exchange_Header; header : SSH2_Key_Exchange_Header;
payload : SSH_Key_Exchange_GSS_Message(is_orig, header.msg_type, header.payload_length); payload : SSH_Key_Exchange_GSS_Message(is_orig, header.msg_type, header.payload_length);
pad : bytestring &length=header.padding_length; pad : bytestring &length=header.padding_length;
} &length=header.packet_length + 4; } &length=header.packet_length + 4;
type SSH_Key_Exchange_GSS_Message(is_orig: bool, msg_type: uint8, length: uint32) = case msg_type of { type SSH_Key_Exchange_GSS_Message(is_orig: bool, msg_type: uint8, length: uint32) = case msg_type of {
SSH_MSG_KEXGSS_INIT -> init : SSH_GSS_INIT(length); SSH_MSG_KEXGSS_INIT -> init : SSH_GSS_INIT(length);
SSH_MSG_KEXGSS_CONTINUE -> cont : SSH_GSS_CONTINUE(length); SSH_MSG_KEXGSS_CONTINUE -> cont : SSH_GSS_CONTINUE(length);
SSH_MSG_KEXGSS_COMPLETE -> complete : SSH_GSS_COMPLETE(length); SSH_MSG_KEXGSS_COMPLETE -> complete : SSH_GSS_COMPLETE(length);
SSH_MSG_KEXGSS_HOSTKEY -> hostkey : SSH_GSS_HOSTKEY(length); SSH_MSG_KEXGSS_HOSTKEY -> hostkey : SSH_GSS_HOSTKEY(length);
SSH_MSG_KEXGSS_ERROR -> error : SSH_GSS_ERROR(length); SSH_MSG_KEXGSS_ERROR -> error : SSH_GSS_ERROR(length);
SSH_MSG_KEXGSS_GROUPREQ -> groupreq : SSH_DH_GEX_REQUEST; SSH_MSG_KEXGSS_GROUPREQ -> groupreq : SSH_DH_GEX_REQUEST;
SSH_MSG_KEXGSS_GROUP -> group : SSH_DH_GEX_GROUP(length); SSH_MSG_KEXGSS_GROUP -> group : SSH_DH_GEX_GROUP(length);
}; };
type SSH_GSS_INIT(length: uint32) = record { type SSH_GSS_INIT(length: uint32) = record {
output_token: ssh_string; output_token : ssh_string;
e : ssh_string; e : ssh_string;
} &length=length; } &length=length;
type SSH_GSS_CONTINUE(length: uint32) = record { type SSH_GSS_CONTINUE(length: uint32) = record {
output_token: ssh_string; output_token : ssh_string;
} &length=length; } &length=length;
type SSH_GSS_COMPLETE(length: uint32) = record { type SSH_GSS_COMPLETE(length: uint32) = record {
f : ssh_string; f : ssh_string;
per_msg_token : ssh_string; per_msg_token : ssh_string;
have_token : uint8; have_token : uint8;
parse_token : case have_token of { parse_token : case have_token of {
0 -> no_token: empty; 0 -> no_token : empty;
default -> token: ssh_string; default -> token : ssh_string;
}; };
} &length=length; } &length=length;
type SSH_GSS_HOSTKEY(length: uint32) = record { type SSH_GSS_HOSTKEY(length: uint32) = record {
k_s: ssh_string; k_s : ssh_string;
} &length=length; } &length=length;
type SSH_GSS_ERROR(length: uint32) = record { type SSH_GSS_ERROR(length: uint32) = record {
major_status: uint32; major_status : uint32;
minor_status: uint32; minor_status : uint32;
message : ssh_string; message : ssh_string;
language : ssh_string; language : ssh_string;
} &length=length; } &length=length;
# KEX_ECDH and KEX_ECMQV exchanges # KEX_ECDH and KEX_ECMQV exchanges
@ -367,37 +368,37 @@ type SSH_GSS_ERROR(length: uint32) = record {
type SSH_Key_Exchange_ECC(is_orig: bool) = record { type SSH_Key_Exchange_ECC(is_orig: bool) = record {
header : SSH2_Key_Exchange_Header; header : SSH2_Key_Exchange_Header;
payload : SSH_Key_Exchange_ECC_Message(is_orig, header.msg_type, header.payload_length); payload : SSH_Key_Exchange_ECC_Message(is_orig, header.msg_type, header.payload_length);
pad : bytestring &length=header.padding_length; pad : bytestring &length=header.padding_length;
} &length=header.packet_length + 4; } &length=header.packet_length + 4;
type SSH_Key_Exchange_ECC_Message(is_orig: bool, msg_type: uint8, length: uint32) = case msg_type of { type SSH_Key_Exchange_ECC_Message(is_orig: bool, msg_type: uint8, length: uint32) = case msg_type of {
SSH_MSG_KEX_ECDH_INIT -> init : SSH_ECC_INIT(length); SSH_MSG_KEX_ECDH_INIT -> init : SSH_ECC_INIT(length);
SSH_MSG_KEX_ECDH_REPLY -> reply : SSH_ECC_REPLY(length); SSH_MSG_KEX_ECDH_REPLY -> reply : SSH_ECC_REPLY(length);
}; };
# This deviates from the RFC. SSH_MSG_KEX_ECDH_INIT and # This deviates from the RFC. SSH_MSG_KEX_ECDH_INIT and
# SSH_MSG_KEX_ECMQV_INIT can be parsed the same way. # SSH_MSG_KEX_ECMQV_INIT can be parsed the same way.
type SSH_ECC_INIT(length: uint32) = record { type SSH_ECC_INIT(length: uint32) = record {
q_c: ssh_string; q_c : ssh_string;
}; };
# This deviates from the RFC. SSH_MSG_KEX_ECDH_REPLY and # This deviates from the RFC. SSH_MSG_KEX_ECDH_REPLY and
# SSH_MSG_KEX_ECMQV_REPLY can be parsed the same way. # SSH_MSG_KEX_ECMQV_REPLY can be parsed the same way.
type SSH_ECC_REPLY(length: uint32) = record { type SSH_ECC_REPLY(length: uint32) = record {
k_s : ssh_string; k_s : ssh_string;
q_s : ssh_string; q_s : ssh_string;
signature : ssh_string; signature : ssh_string;
}; };
type ssh_string = record { type ssh_string = record {
len: uint32; len : uint32;
val: bytestring &length=len; val : bytestring &length=len;
}; };
type ssh_host_key = record { type ssh_host_key = record {
len: uint32; len : uint32;
key_type: ssh_string; key_type : ssh_string;
key: ssh_string; key : ssh_string;
} &length=(len + 4); } &length=(len + 4);
## Done with types ## Done with types
@ -430,7 +431,7 @@ refine connection SSH_Conn += {
kex_algs_cache_.free(); kex_algs_cache_.free();
%} %}
function get_state(is_orig: bool): int function get_state(is_orig: bool) : int
%{ %{
if ( is_orig ) if ( is_orig )
{ {
@ -442,7 +443,7 @@ refine connection SSH_Conn += {
} }
%} %}
function update_state(s: state, is_orig: bool): bool function update_state(s: state, is_orig: bool) : bool
%{ %{
if ( is_orig ) if ( is_orig )
state_up_ = s; state_up_ = s;
@ -451,12 +452,12 @@ refine connection SSH_Conn += {
return true; return true;
%} %}
function get_version(): int function get_version() : int
%{ %{
return version_; return version_;
%} %}
function update_version(v: bytestring, is_orig: bool): bool function update_version(v: bytestring, is_orig: bool) : bool
%{ %{
if ( is_orig && ( v.length() >= 4 ) ) if ( is_orig && ( v.length() >= 4 ) )
{ {
@ -468,7 +469,7 @@ refine connection SSH_Conn += {
return true; return true;
%} %}
function update_kex_state_if_equal(s: string, to_state: state): bool function update_kex_state_if_equal(s: string, to_state: state) : bool
%{ %{
if ( strcmp(c_str(kex_algorithm_), s.c_str()) == 0 ) if ( strcmp(c_str(kex_algorithm_), s.c_str()) == 0 )
{ {
@ -479,7 +480,7 @@ refine connection SSH_Conn += {
return false; return false;
%} %}
function update_kex_state_if_startswith(s: string, to_state: state): bool function update_kex_state_if_startswith(s: string, to_state: state) : bool
%{ %{
if ( (uint) kex_algorithm_.length() < s.length() ) if ( (uint) kex_algorithm_.length() < s.length() )
return false; return false;
@ -493,7 +494,7 @@ refine connection SSH_Conn += {
return false; return false;
%} %}
function update_kex(algs: bytestring, orig: bool): bool function update_kex(algs: bytestring, orig: bool) : bool
%{ %{
if ( !kex_seen_ ) if ( !kex_seen_ )
{ {
@ -506,18 +507,18 @@ refine connection SSH_Conn += {
else if ( kex_orig_ == orig ) else if ( kex_orig_ == orig )
return false; return false;
VectorVal* client_list = name_list_to_vector(orig ? algs : kex_algs_cache_); VectorVal* client_list = name_list_to_vector(orig ? algs : kex_algs_cache_);
VectorVal* server_list = name_list_to_vector(orig ? kex_algs_cache_ : algs); VectorVal* server_list = name_list_to_vector(orig ? kex_algs_cache_ : algs);
for ( unsigned int i = 0; i < client_list->Size(); ++i) for ( unsigned int i = 0; i < client_list->Size(); ++i)
{ {
for ( unsigned int j = 0; j < server_list->Size(); ++j) for ( unsigned int j = 0; j < server_list->Size(); ++j)
{ {
if ( strcmp((const char *) client_list->Lookup(i)->AsStringVal()->Bytes(), if ( strcmp((const char *) client_list->Lookup(i)->AsStringVal()->Bytes(),
(const char *) server_list->Lookup(j)->AsStringVal()->Bytes()) == 0 ) (const char *) server_list->Lookup(j)->AsStringVal()->Bytes()) == 0 )
{ {
kex_algorithm_.init((const uint8 *) client_list->Lookup(i)->AsStringVal()->Bytes(), kex_algorithm_.init((const uint8 *) client_list->Lookup(i)->AsStringVal()->Bytes(),
client_list->Lookup(i)->AsStringVal()->Len()); client_list->Lookup(i)->AsStringVal()->Len());
Unref(client_list); Unref(client_list);
Unref(server_list); Unref(server_list);
@ -569,8 +570,6 @@ refine connection SSH_Conn += {
Unref(server_list); Unref(server_list);
return true; return true;
%} %}
}; };