Rust-related/RustPython
1
0
Fork 1
mirror of https://github.com/RustPython/RustPython.git synced 2026-06-09 22:49:57 +09:00
Code Issues Packages Projects Releases Wiki Activity

sock.recv_into takes any rw buffer

Browse Source
This commit is contained in:
Noah
2020-12-17 12:57:05 -06:00
parent 5a56dd1a77
commit 13ea0efa9f
1 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
vm/src/stdlib/socket.rs
View File

@@ -8,12 +8,11 @@ use std::io::{self, prelude::*};
use std::net::{Ipv4Addr, Ipv6Addr, Shutdown, SocketAddr, ToSocketAddrs};
use std::time::Duration;
use crate::builtins::bytearray::PyByteArrayRef;
use crate::builtins::bytes::PyBytesRef;
use crate::builtins::pystr::{PyStr, PyStrRef};
use crate::builtins::pytype::PyTypeRef;
use crate::builtins::tuple::PyTupleRef;
use crate::byteslike::PyBytesLike;
use crate::byteslike::{PyBytesLike, PyRwBytesLike};
use crate::common::lock::{PyRwLock, PyRwLockReadGuard, PyRwLockWriteGuard};
use crate::exceptions::{IntoPyException, PyBaseExceptionRef};
use crate::function::{FuncArgs, OptionalArg};
@@ -179,10 +178,8 @@ impl PySocket {
}
#[pymethod]
fn recv_into(&self, buf: PyByteArrayRef, vm: &VirtualMachine) -> PyResult<usize> {
let mut buffer = buf.borrow_value_mut();
self.sock()
.recv(&mut buffer.elements)
fn recv_into(&self, buf: PyRwBytesLike, vm: &VirtualMachine) -> PyResult<usize> {
buf.with_ref(|buf| self.sock().recv(buf))
.map_err(|err| convert_sock_error(vm, err))
}