Drop for PySSLSocket by youknowone · Pull Request #6791 · RustPython/RustPython
// Check if connection has been shut down // After unwrap()/shutdown(), read operations should fail with SSLError // Only block after shutdown is COMPLETED, not during shutdown process let shutdown_state = *self.shutdown_state.lock(); if shutdown_state != ShutdownState::NotStarted { if shutdown_state == ShutdownState::Completed { return Err(vm .new_os_subtype_error( PySSLError::class(&vm.ctx).to_owned(),
// Check shutdown state if *self.shutdown_state.lock() != ShutdownState::NotStarted { // Only block after shutdown is COMPLETED, not during shutdown process if *self.shutdown_state.lock() == ShutdownState::Completed { return Err(vm .new_os_subtype_error( PySSLError::class(&vm.ctx).to_owned(),
// Set shutdown_state first to ensure atomic visibility // This prevents read/write race conditions during shutdown *self.shutdown_state.lock() = ShutdownState::Completed; *self.connection.lock() = None; return Ok(self.sock.clone());
// Clean up SSL socket resources on drop impl Drop for PySSLSocket { fn drop(&mut self) { // Clear connection state let _ = self.connection.lock().take();
// Clear pending buffers self.pending_tls_output.lock().clear(); *self.write_buffered_len.lock() = 0;
// Reset shutdown state *self.shutdown_state.lock() = ShutdownState::NotStarted; } }
// MemoryBIO - provides in-memory buffer for SSL/TLS I/O #[pyattr] #[pyclass(name = "MemoryBIO", module = "ssl")]