Merge pull request #2 from adaminglehart/configurable-max-connection-…

…attempts

Make the maximum number of connection attempts configurable

addon.d.ts

@@ -13,4 +13,5 @@ export class VsockSocket {
   startRecv(): void
   writeText(data: string): void
   writeBuffer(data: Buffer): void
+  setMaxConnectionAttempts(maxConnectionAttempts: number): void
 }

index.ts

@@ -93,6 +93,10 @@ export class VsockSocket extends EventEmitter {
     this.on('end', this.onEnd)
   }
 
+  setMaxConnectionAttempts(maxConnectionAttempts: number) {
+    this.socket.setMaxConnectionAttempts(maxConnectionAttempts)
+  }
+
   connect(cid: number, port: number, connectCallback?: Callback) {
     this.checkDestroyed()
     this.connecting = true

src/socket.rs

@@ -23,7 +23,7 @@ const BUF_MAX_LEN: usize = 8192;
 const BACKLOG: usize = 128;
 
 // Maximum number of connection attempts
-const MAX_CONNECTION_ATTEMPTS: usize = 5;
+const MAX_CONNECTION_ATTEMPTS: u32 = 5;
 
 #[derive(Eq, Ord, PartialEq, PartialOrd, Copy, Clone)]
 enum State {
@@ -38,6 +38,7 @@ pub struct VsockSocket {
   fd: RawFd,
   emitter: Emitter,
   state: State,
+  max_connection_attempts: u32,
 }
 
 #[napi]
@@ -59,6 +60,7 @@ impl VsockSocket {
       fd,
       emitter: Emitter::new(env, emit_fn)?,
       state: State::Initialized,
+      max_connection_attempts: MAX_CONNECTION_ATTEMPTS,
     })
   }
 
@@ -101,6 +103,20 @@ impl VsockSocket {
     Ok(())
   }
 
+  #[napi]
+  pub fn set_max_connection_attempts(&mut self, max_connection_attempts: JsNumber) -> Result<()> {
+    let max_connection_attempts = max_connection_attempts.get_uint32()?;
+    if max_connection_attempts < 1 {
+      return Err(error(
+        "max_connection_attempts must be greater than 0".to_string(),
+      ));
+    }
+
+    self.max_connection_attempts = max_connection_attempts;
+
+    Ok(())
+  }
+
   #[napi]
   pub fn connect(&mut self, cid: JsNumber, port: JsNumber) -> Result<()> {
     let cid = cid.get_uint32()?;
@@ -110,9 +126,11 @@ impl VsockSocket {
     let emit_error = self.thread_safe_emit_error()?;
     let emit_connect = self.thread_safe_emit_connect()?;
 
+    let max_connection_attempts = self.max_connection_attempts;
+
     thread::spawn(move || {
       let mut err_msg: Option<String> = None;
-      for i in 0..MAX_CONNECTION_ATTEMPTS {
+      for i in 0..max_connection_attempts {
         match connect(fd, &sockaddr) {
           Ok(_) => {
             emit_connect.call(Ok(()), ThreadsafeFunctionCallMode::Blocking);