aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--controller/src/lib.rs9
-rw-r--r--utils/.gitignore1
-rw-r--r--utils/src/protocol.rs4
3 files changed, 12 insertions, 2 deletions
diff --git a/controller/src/lib.rs b/controller/src/lib.rs
index b53502f..60b3639 100644
--- a/controller/src/lib.rs
+++ b/controller/src/lib.rs
@@ -120,6 +120,15 @@ enum ThreadCollect {
async fn thread_handshake_handler(mut listener: TcpListener, sink: mpsc::Sender<ThreadCollect>) {
loop {
let (mut sock, _) = listener.accept().await.expect("Accept failed on TCP server socket");
+
+ // Error setting keepalive is not fatal
+ match sock.set_keepalive(Some(std::time::Duration::from_secs(60))) {
+ Ok(()) => {}
+ Err(e) => {
+ eprintln!("WARNING: Error setting keepalive on worker socket: {}", e);
+ }
+ }
+
let mut sink = sink.clone();
task::spawn(async move {
let payload = encode_message(MessageBody::Version(1));
diff --git a/utils/.gitignore b/utils/.gitignore
new file mode 100644
index 0000000..ea8c4bf
--- /dev/null
+++ b/utils/.gitignore
@@ -0,0 +1 @@
+/target
diff --git a/utils/src/protocol.rs b/utils/src/protocol.rs
index 4820f79..fce168d 100644
--- a/utils/src/protocol.rs
+++ b/utils/src/protocol.rs
@@ -23,7 +23,7 @@ pub struct RawMessage {
impl RawMessage {
pub fn receive(reader: &mut BufReader<TcpStream>) -> io::Result<Option<Self>> {
let mut header = [0u8; 17];
- if let Err(e) = reader.read(&mut header) {
+ if let Err(e) = reader.read_exact(&mut header) {
if e.kind() == ErrorKind::UnexpectedEof { return Ok(None); }
else { return Err(e); }
}
@@ -34,7 +34,7 @@ impl RawMessage {
let mut payload = Vec::new();
payload.resize(length, 0u8);
- if let Err(e) = reader.read(&mut payload) {
+ if let Err(e) = reader.read_exact(&mut payload) {
if e.kind() == ErrorKind::UnexpectedEof { return Ok(None); }
else { return Err(e); }
}