feat: add support for Windows (#98)

src/tracing/error.rs

@@ -23,4 +23,6 @@ pub enum TracerError {
     AddressNotAvailable(SocketAddr),
     #[error("invalid source IP address: {0}")]
     InvalidSourceAddr(IpAddr),
+    #[error("error: {0}")]
+    ErrorString(String),
 }

src/tracing/net/ipv4.rs

@@ -154,7 +154,8 @@ pub fn dispatch_tcp_probe(
         PortDirection::FixedDest(dest_port) => (probe.sequence.0, dest_port.0),
         PortDirection::FixedBoth(_, _) | PortDirection::None => unimplemented!(),
     };
-    let socket = platform::make_stream_socket_ipv4()?;
+    #[allow(unused_mut)]
+    let mut socket = platform::make_stream_socket_ipv4()?;
     let local_addr = SocketAddr::new(IpAddr::V4(src_addr), src_port);
     socket.bind(local_addr)?;
     socket.set_ttl(u32::from(probe.ttl.0))?;
@@ -180,6 +181,7 @@ pub fn dispatch_tcp_probe(
     Ok(socket)
 }
 
+#[cfg(unix)]
 pub fn recv_icmp_probe(
     recv_socket: &mut Socket,
     protocol: TracerProtocol,
@@ -204,6 +206,20 @@ pub fn recv_icmp_probe(
     }
 }
 
+#[cfg(windows)]
+pub fn recv_icmp_probe(
+    recv_socket: &mut Socket,
+    protocol: TracerProtocol,
+    multipath_strategy: MultipathStrategy,
+    direction: PortDirection,
+) -> TraceResult<Option<ProbeResponse>> {
+    let bytes = &recv_socket.buf_bytes();
+    let ipv4 = Ipv4Packet::new_view(bytes).req()?;
+    // post the WSARecvFrom again, so that the next OVERLAPPED event can get triggered
+    recv_socket.recv_from()?;
+    extract_probe_resp(protocol, multipath_strategy, direction, &ipv4)
+}
+
 pub fn recv_tcp_socket(
     tcp_socket: &Socket,
     sequence: Sequence,

src/tracing/net/ipv6.rs

@@ -109,11 +109,12 @@ pub fn dispatch_tcp_probe(
         PortDirection::FixedDest(dest_port) => (probe.sequence.0, dest_port.0),
         PortDirection::FixedBoth(_, _) | PortDirection::None => unimplemented!(),
     };
-    let socket = platform::make_stream_socket_ipv6()?;
     let local_addr = SocketAddr::new(IpAddr::V6(src_addr), src_port);
+    let remote_addr = SocketAddr::new(IpAddr::V6(dest_addr), dest_port);
+    #[allow(unused_mut)]
+    let mut socket = platform::make_stream_socket_ipv6()?;
     socket.bind(local_addr)?;
     socket.set_unicast_hops_v6(probe.ttl.0)?;
-    let remote_addr = SocketAddr::new(IpAddr::V6(dest_addr), dest_port);
     match socket.connect(remote_addr) {
         Ok(_) => {}
         Err(err) => {
@@ -134,6 +135,7 @@ pub fn dispatch_tcp_probe(
     Ok(socket)
 }
 
+#[cfg(unix)]
 pub fn recv_icmp_probe(
     recv_socket: &mut Socket,
     protocol: TracerProtocol,
@@ -160,6 +162,25 @@ pub fn recv_icmp_probe(
     }
 }
 
+#[cfg(windows)]
+#[allow(unsafe_code)]
+pub fn recv_icmp_probe(
+    recv_socket: &mut Socket,
+    protocol: TracerProtocol,
+    direction: PortDirection,
+) -> TraceResult<Option<ProbeResponse>> {
+    let bytes = &recv_socket.buf_bytes();
+    let icmp_v6 = IcmpPacket::new_view(bytes).req()?;
+    let addr = recv_socket.from()?;
+    // post the WSARecvFrom again, so that the next OVERLAPPED event can get triggered
+    recv_socket.recv_from()?;
+    if let IpAddr::V6(src_addr) = addr {
+        extract_probe_resp(protocol, direction, &icmp_v6, src_addr)
+    } else {
+        Err(TracerError::InvalidSourceAddr(addr))
+    }
+}
+
 pub fn recv_tcp_socket(
     tcp_socket: &Socket,
     sequence: Sequence,