Noratrieb/cluelessh
1
0
Fork 0
mirror of https://github.com/Noratrieb/cluelessh.git synced 2026-01-16 09:25:04 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

cleanup

Browse source
This commit is contained in:
nora 2024-08-15 15:26:50 +02:00
parent e3bf214ec6
commit d68caf8e3c
1 changed files with 11 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

33
src/main.rs
View file

@ -141,21 +141,6 @@ async fn handle_connection(
} }
} }
ChannelRequest::Shell { want_reply } => { ChannelRequest::Shell { want_reply } => {
state.do_operation(
update.number.construct_op(ChannelOperationKind::Data(
vec![b'a'; 1_000_000],
)),
);
state.do_operation(
update.number.construct_op(ChannelOperationKind::Data(
vec![b'b'; 1_000_000],
)),
);
state.do_operation(
update.number.construct_op(ChannelOperationKind::Data(
vec![b'c'; 1_000_000],
)),
);
if want_reply { if want_reply {
state.do_operation(success); state.do_operation(success);
} }
@ -191,11 +176,14 @@ async fn handle_connection(
}; };
} }
ChannelUpdateKind::Data { data } => { ChannelUpdateKind::Data { data } => {
let is_eof = data.contains(&0x03 /*EOF, Ctrl-C*/); let is_eof = data.contains(&0x04 /*EOF, Ctrl-D*/);
// echo :3 // echo :3
// state state.do_operation(
// .do_operation(update.number.construct_op(ChannelOperationKind::Data(data))); update
.number
.construct_op(ChannelOperationKind::Data(data.clone())),
);
// arbitrary limit // arbitrary limit
if total_sent_data.len() < 50_000 { if total_sent_data.len() < 50_000 {
@ -204,15 +192,16 @@ async fn handle_connection(
info!(channel = %update.number, "Reached stdin limit"); info!(channel = %update.number, "Reached stdin limit");
state.do_operation( state.do_operation(
update.number.construct_op(ChannelOperationKind::Data( update.number.construct_op(ChannelOperationKind::Data(
b"Thanks Hayley!".to_vec(), b"Thanks Hayley!\n".to_vec(),
)), )),
); );
//state.do_operation(update.number.construct_op(ChannelOperationKind::Close)); state.do_operation(update.number.construct_op(ChannelOperationKind::Close));
} }
if false && is_eof { if is_eof {
debug!(channel = %update.number, "Received Ctrl-C, closing channel"); debug!(channel = %update.number, "Received Ctrl-D, closing channel");
state.do_operation(update.number.construct_op(ChannelOperationKind::Eof));
state.do_operation(update.number.construct_op(ChannelOperationKind::Close)); state.do_operation(update.number.construct_op(ChannelOperationKind::Close));
} }
} }