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

Some styling tweaks

Browse source 
- used transient declarations where appropriate
- fixed brackets
- cleaned up some comments
This commit is contained in:
Martin van Hensbergen 2016-04-11 11:35:36 +02:00
parent 04dd65cbaf
commit 034f725f3f
9 changed files with 109 additions and 113 deletions
Download patch file Download diff file Expand all files Collapse all files

60
scripts/base/protocols/rfb/main.bro
View file

@ -1,4 +1,4 @@
module Rfb;
module RFB;
export {
redef enum Log::ID += { LOG };
@ -11,17 +11,27 @@ export {
## The connection's 4-tuple of endpoint addresses/ports.
id: conn_id &log;
## Major version of the client.
client_major_version: string &log &optional;
## Minor version of the client.
client_minor_version: string &log &optional;
## Major version of the server.
server_major_version: string &log &optional;
## Major version of the client.
server_minor_version: string &log &optional;
## Identifier of authentication method used.
authentication_method: string &log &optional;
## Whether or not authentication was succesful.
auth: bool &log &optional;
## Whether the client has an exclusive or a shared session.
share_flag: bool &log &optional;
## Name of the screen that is being shared.
desktop_name: string &log &optional;
## Width of the screen that is being shared.
width: count &log &optional;
## Height of the screen that is being shared.
height: count &log &optional;
done: bool &default=F;
@ -30,7 +40,8 @@ export {
global log_rfb: event(rec: Info);
}
function friendly_auth_name(auth: count): string {
function friendly_auth_name(auth: count): string
{
switch (auth) {
case 0:
return "Invalid";
@ -56,37 +67,40 @@ function friendly_auth_name(auth: count): string {
return "Apple Remote Desktop";
}
return "RealVNC";
}
redef record connection += {
rfb_state: Info &optional;
};
event bro_init() &priority=5
{
Log::create_stream(Rfb::LOG, [$columns=Info, $ev=log_rfb, $path="rfb"]);
Log::create_stream(RFB::LOG, [$columns=Info, $ev=log_rfb, $path="rfb"]);
}
function write_log(c:connection) {
function write_log(c:connection)
{
local state = c$rfb_state;
if ( state?$done && state$done == T) {
if ( state?$done && state$done == T )
{
return;
}
Log::write(Rfb::LOG, c$rfb_state);
c$rfb_state$done = T;
}
}
function set_session(c: connection) {
if ( ! c?$rfb_state ) {
Log::write(RFB::LOG, c$rfb_state);
c$rfb_state$done = T;
}
function set_session(c: connection)
{
if ( ! c?$rfb_state )
{
local info: Info;
info$ts = network_time();
info$uid = c$uid;
info$id = c$id;
c$rfb_state = info;
}
}
}
event rfb_event(c: connection)
@ -121,13 +135,9 @@ event rfb_server_parameters(c: connection, name: string, width: count, height: c
write_log(c);
}
event rfb_auth_result(c: connection, result: count)
event rfb_auth_result(c: connection, result: bool)
{
if ( result ==0 ) {
c$rfb_state$auth = T;
} else {
c$rfb_state$auth = F;
}
c$rfb_state$auth = !result;
}
event rfb_share_flag(c: connection, flag: bool)
@ -135,8 +145,10 @@ event rfb_share_flag(c: connection, flag: bool)
c$rfb_state$share_flag = flag;
}
event connection_state_remove(c: connection) {
if ( c?$rfb_state ) {
write_log(c);
event connection_state_remove(c: connection)
{
if ( c?$rfb_state )
{
write_log(c);
}
}
}