mirror of
https://github.com/DragonOS-Community/DragonOS.git
synced 2025-06-09 11:16:47 +00:00
make fmt
This commit is contained in:
parent
7cebb88792
commit
8fe49e190e
@ -133,7 +133,7 @@ pub extern "sysv64" fn syscall_handler(frame: &mut TrapFrame) {
|
|||||||
show &= false;
|
show &= false;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
show = false;
|
show &= false;
|
||||||
if show {
|
if show {
|
||||||
debug!("[SYS] [Pid: {:?}] [Call: {:?}]", pid, to_print);
|
debug!("[SYS] [Pid: {:?}] [Call: {:?}]", pid, to_print);
|
||||||
}
|
}
|
||||||
|
@ -1,8 +1,10 @@
|
|||||||
/// 引入Module
|
/// 引入Module
|
||||||
use crate::{driver::{
|
use crate::{
|
||||||
|
driver::{
|
||||||
base::{
|
base::{
|
||||||
device::{
|
device::{
|
||||||
device_number::{DeviceNumber, Major}, Device, DeviceError, IdTable, BLOCKDEVS
|
device_number::{DeviceNumber, Major},
|
||||||
|
Device, DeviceError, IdTable, BLOCKDEVS,
|
||||||
},
|
},
|
||||||
map::{
|
map::{
|
||||||
DeviceStruct, DEV_MAJOR_DYN_END, DEV_MAJOR_DYN_EXT_END, DEV_MAJOR_DYN_EXT_START,
|
DeviceStruct, DEV_MAJOR_DYN_END, DEV_MAJOR_DYN_EXT_END, DEV_MAJOR_DYN_EXT_START,
|
||||||
@ -10,7 +12,9 @@ use crate::{driver::{
|
|||||||
},
|
},
|
||||||
},
|
},
|
||||||
block::cache::{cached_block_device::BlockCache, BlockCacheError, BLOCK_SIZE},
|
block::cache::{cached_block_device::BlockCache, BlockCacheError, BLOCK_SIZE},
|
||||||
}, filesystem::sysfs::AttributeGroup};
|
},
|
||||||
|
filesystem::sysfs::AttributeGroup,
|
||||||
|
};
|
||||||
|
|
||||||
use alloc::{string::String, sync::Arc, vec::Vec};
|
use alloc::{string::String, sync::Arc, vec::Vec};
|
||||||
use core::{any::Any, fmt::Display, ops::Deref};
|
use core::{any::Any, fmt::Display, ops::Deref};
|
||||||
|
@ -157,7 +157,11 @@ impl Attribute for UeventAttr {
|
|||||||
writeln!(&mut uevent_content, "DEVTYPE=char").unwrap();
|
writeln!(&mut uevent_content, "DEVTYPE=char").unwrap();
|
||||||
}
|
}
|
||||||
DeviceType::Net => {
|
DeviceType::Net => {
|
||||||
let net_device = device.clone().cast::<dyn Iface>().map_err(|e: Arc<dyn Device>| {
|
let net_device =
|
||||||
|
device
|
||||||
|
.clone()
|
||||||
|
.cast::<dyn Iface>()
|
||||||
|
.map_err(|e: Arc<dyn Device>| {
|
||||||
warn!("device:{:?} is not a net device!", e);
|
warn!("device:{:?} is not a net device!", e);
|
||||||
SystemError::EINVAL
|
SystemError::EINVAL
|
||||||
})?;
|
})?;
|
||||||
@ -200,7 +204,6 @@ impl Attribute for UeventAttr {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
/// 将设备的基本信息写入 uevent 文件
|
/// 将设备的基本信息写入 uevent 文件
|
||||||
fn sysfs_emit_str(buf: &mut [u8], content: &str) -> Result<usize, SystemError> {
|
fn sysfs_emit_str(buf: &mut [u8], content: &str) -> Result<usize, SystemError> {
|
||||||
log::info!("sysfs_emit_str");
|
log::info!("sysfs_emit_str");
|
||||||
|
@ -3,6 +3,7 @@ use core::sync::atomic::{AtomicU32, AtomicUsize};
|
|||||||
use crate::libs::rwlock::RwLock;
|
use crate::libs::rwlock::RwLock;
|
||||||
use crate::net::socket::EPollEventType;
|
use crate::net::socket::EPollEventType;
|
||||||
use crate::net::socket::{self, inet::Types};
|
use crate::net::socket::{self, inet::Types};
|
||||||
|
use alloc::boxed::Box;
|
||||||
use alloc::vec::Vec;
|
use alloc::vec::Vec;
|
||||||
use smoltcp;
|
use smoltcp;
|
||||||
use system_error::SystemError::{self, *};
|
use system_error::SystemError::{self, *};
|
||||||
@ -30,13 +31,13 @@ where
|
|||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub enum Init {
|
pub enum Init {
|
||||||
Unbound(smoltcp::socket::tcp::Socket<'static>),
|
Unbound(Box<smoltcp::socket::tcp::Socket<'static>>),
|
||||||
Bound((socket::inet::BoundInner, smoltcp::wire::IpEndpoint)),
|
Bound((socket::inet::BoundInner, smoltcp::wire::IpEndpoint)),
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Init {
|
impl Init {
|
||||||
pub(super) fn new() -> Self {
|
pub(super) fn new() -> Self {
|
||||||
Init::Unbound(new_smoltcp_socket())
|
Init::Unbound(Box::new(new_smoltcp_socket()))
|
||||||
}
|
}
|
||||||
|
|
||||||
/// 传入一个已经绑定的socket
|
/// 传入一个已经绑定的socket
|
||||||
@ -55,7 +56,7 @@ impl Init {
|
|||||||
) -> Result<Self, SystemError> {
|
) -> Result<Self, SystemError> {
|
||||||
match self {
|
match self {
|
||||||
Init::Unbound(socket) => {
|
Init::Unbound(socket) => {
|
||||||
let bound = socket::inet::BoundInner::bind(socket, &local_endpoint.addr)?;
|
let bound = socket::inet::BoundInner::bind(*socket, &local_endpoint.addr)?;
|
||||||
bound
|
bound
|
||||||
.port_manager()
|
.port_manager()
|
||||||
.bind_port(Types::Tcp, local_endpoint.port)?;
|
.bind_port(Types::Tcp, local_endpoint.port)?;
|
||||||
@ -73,7 +74,7 @@ impl Init {
|
|||||||
match self {
|
match self {
|
||||||
Init::Unbound(socket) => {
|
Init::Unbound(socket) => {
|
||||||
let (bound, address) =
|
let (bound, address) =
|
||||||
socket::inet::BoundInner::bind_ephemeral(socket, remote_endpoint.addr)
|
socket::inet::BoundInner::bind_ephemeral(*socket, remote_endpoint.addr)
|
||||||
.map_err(|err| (Self::new(), err))?;
|
.map_err(|err| (Self::new(), err))?;
|
||||||
let bound_port = bound
|
let bound_port = bound
|
||||||
.port_manager()
|
.port_manager()
|
||||||
|
@ -185,7 +185,10 @@ impl TcpSocket {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub fn try_recv(&self, buf: &mut [u8]) -> Result<usize, SystemError> {
|
pub fn try_recv(&self, buf: &mut [u8]) -> Result<usize, SystemError> {
|
||||||
self.inner.read().as_ref().map(|inner| {
|
self.inner
|
||||||
|
.read()
|
||||||
|
.as_ref()
|
||||||
|
.map(|inner| {
|
||||||
inner.iface().unwrap().poll();
|
inner.iface().unwrap().poll();
|
||||||
let result = match inner {
|
let result = match inner {
|
||||||
Inner::Established(inner) => inner.recv_slice(buf),
|
Inner::Established(inner) => inner.recv_slice(buf),
|
||||||
@ -193,7 +196,8 @@ impl TcpSocket {
|
|||||||
};
|
};
|
||||||
inner.iface().unwrap().poll();
|
inner.iface().unwrap().poll();
|
||||||
result
|
result
|
||||||
}).unwrap()
|
})
|
||||||
|
.unwrap()
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn try_send(&self, buf: &[u8]) -> Result<usize, SystemError> {
|
pub fn try_send(&self, buf: &[u8]) -> Result<usize, SystemError> {
|
||||||
@ -238,7 +242,7 @@ impl Socket for TcpSocket {
|
|||||||
fn get_name(&self) -> Result<Endpoint, SystemError> {
|
fn get_name(&self) -> Result<Endpoint, SystemError> {
|
||||||
match self.inner.read().as_ref().expect("Tcp Inner is None") {
|
match self.inner.read().as_ref().expect("Tcp Inner is None") {
|
||||||
Inner::Init(Init::Unbound(_)) => Ok(Endpoint::Ip(UNSPECIFIED_LOCAL_ENDPOINT)),
|
Inner::Init(Init::Unbound(_)) => Ok(Endpoint::Ip(UNSPECIFIED_LOCAL_ENDPOINT)),
|
||||||
Inner::Init(Init::Bound((_, local))) => Ok(Endpoint::Ip(local.clone())),
|
Inner::Init(Init::Bound((_, local))) => Ok(Endpoint::Ip(*local)),
|
||||||
Inner::Connecting(connecting) => Ok(Endpoint::Ip(connecting.get_name())),
|
Inner::Connecting(connecting) => Ok(Endpoint::Ip(connecting.get_name())),
|
||||||
Inner::Established(established) => Ok(Endpoint::Ip(established.local_endpoint())),
|
Inner::Established(established) => Ok(Endpoint::Ip(established.local_endpoint())),
|
||||||
Inner::Listening(listening) => Ok(Endpoint::Ip(listening.get_name())),
|
Inner::Listening(listening) => Ok(Endpoint::Ip(listening.get_name())),
|
||||||
|
@ -62,11 +62,7 @@ pub(super) struct Listener {
|
|||||||
impl Listener {
|
impl Listener {
|
||||||
pub(super) fn new(inode: Endpoint, backlog: usize) -> Self {
|
pub(super) fn new(inode: Endpoint, backlog: usize) -> Self {
|
||||||
log::debug!("backlog {}", backlog);
|
log::debug!("backlog {}", backlog);
|
||||||
let back = if backlog > 1024 {
|
let back = if backlog > 1024 { 1024_usize } else { backlog };
|
||||||
1024 as usize
|
|
||||||
} else {
|
|
||||||
backlog
|
|
||||||
};
|
|
||||||
return Self {
|
return Self {
|
||||||
inode,
|
inode,
|
||||||
backlog: AtomicUsize::new(back),
|
backlog: AtomicUsize::new(back),
|
||||||
@ -82,7 +78,7 @@ impl Listener {
|
|||||||
log::debug!(" incom len {}", incoming_conns.len());
|
log::debug!(" incom len {}", incoming_conns.len());
|
||||||
let conn = incoming_conns
|
let conn = incoming_conns
|
||||||
.pop_front()
|
.pop_front()
|
||||||
.ok_or_else(|| SystemError::EAGAIN_OR_EWOULDBLOCK)?;
|
.ok_or(SystemError::EAGAIN_OR_EWOULDBLOCK)?;
|
||||||
let socket =
|
let socket =
|
||||||
Arc::downcast::<SeqpacketSocket>(conn.inner()).map_err(|_| SystemError::EINVAL)?;
|
Arc::downcast::<SeqpacketSocket>(conn.inner()).map_err(|_| SystemError::EINVAL)?;
|
||||||
let peer = match &*socket.inner.read() {
|
let peer = match &*socket.inner.read() {
|
||||||
@ -190,7 +186,7 @@ impl Connected {
|
|||||||
if self.can_send()? {
|
if self.can_send()? {
|
||||||
return self.send_slice(buf);
|
return self.send_slice(buf);
|
||||||
} else {
|
} else {
|
||||||
log::debug!("can not send {:?}", String::from_utf8_lossy(&buf[..]));
|
log::debug!("can not send {:?}", String::from_utf8_lossy(buf));
|
||||||
return Err(SystemError::ENOBUFS);
|
return Err(SystemError::ENOBUFS);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -230,11 +230,7 @@ impl Socket for SeqpacketSocket {
|
|||||||
if !self.is_nonblocking() {
|
if !self.is_nonblocking() {
|
||||||
loop {
|
loop {
|
||||||
wq_wait_event_interruptible!(self.wait_queue, self.is_acceptable(), {})?;
|
wq_wait_event_interruptible!(self.wait_queue, self.is_acceptable(), {})?;
|
||||||
match self
|
match self.try_accept() {
|
||||||
.try_accept()
|
|
||||||
.map(|(seqpacket_socket, remote_endpoint)| {
|
|
||||||
(seqpacket_socket, Endpoint::from(remote_endpoint))
|
|
||||||
}) {
|
|
||||||
Ok((socket, epoint)) => return Ok((socket, epoint)),
|
Ok((socket, epoint)) => return Ok((socket, epoint)),
|
||||||
Err(_) => continue,
|
Err(_) => continue,
|
||||||
}
|
}
|
||||||
@ -274,7 +270,7 @@ impl Socket for SeqpacketSocket {
|
|||||||
};
|
};
|
||||||
|
|
||||||
if let Some(endpoint) = endpoint {
|
if let Some(endpoint) = endpoint {
|
||||||
return Ok(Endpoint::from(endpoint));
|
return Ok(endpoint);
|
||||||
} else {
|
} else {
|
||||||
return Err(SystemError::EAGAIN_OR_EWOULDBLOCK);
|
return Err(SystemError::EAGAIN_OR_EWOULDBLOCK);
|
||||||
}
|
}
|
||||||
@ -289,7 +285,7 @@ impl Socket for SeqpacketSocket {
|
|||||||
};
|
};
|
||||||
|
|
||||||
if let Some(endpoint) = endpoint {
|
if let Some(endpoint) = endpoint {
|
||||||
return Ok(Endpoint::from(endpoint));
|
return Ok(endpoint);
|
||||||
} else {
|
} else {
|
||||||
return Err(SystemError::EAGAIN_OR_EWOULDBLOCK);
|
return Err(SystemError::EAGAIN_OR_EWOULDBLOCK);
|
||||||
}
|
}
|
||||||
|
@ -231,10 +231,7 @@ impl Socket for StreamSocket {
|
|||||||
//目前只实现了阻塞式实现
|
//目前只实现了阻塞式实现
|
||||||
loop {
|
loop {
|
||||||
wq_wait_event_interruptible!(self.wait_queue, self.is_acceptable(), {})?;
|
wq_wait_event_interruptible!(self.wait_queue, self.is_acceptable(), {})?;
|
||||||
match self
|
match self.try_accept() {
|
||||||
.try_accept()
|
|
||||||
.map(|(stream_socket, remote_endpoint)| (stream_socket, remote_endpoint))
|
|
||||||
{
|
|
||||||
Ok((socket, endpoint)) => {
|
Ok((socket, endpoint)) => {
|
||||||
debug!("server accept!:{:?}", endpoint);
|
debug!("server accept!:{:?}", endpoint);
|
||||||
return Ok((socket, endpoint));
|
return Ok((socket, endpoint));
|
||||||
|
@ -312,7 +312,7 @@ impl From<Endpoint> for SockAddr {
|
|||||||
}
|
}
|
||||||
let addr_un = SockAddrUn {
|
let addr_un = SockAddrUn {
|
||||||
sun_family: AddressFamily::Unix as u16,
|
sun_family: AddressFamily::Unix as u16,
|
||||||
sun_path: sun_path,
|
sun_path,
|
||||||
};
|
};
|
||||||
return SockAddr { addr_un };
|
return SockAddr { addr_un };
|
||||||
}
|
}
|
||||||
|
@ -1,7 +1,10 @@
|
|||||||
use libc::{sockaddr, sockaddr_storage, recvfrom, bind, sendto, socket, AF_NETLINK, SOCK_DGRAM, SOCK_CLOEXEC, getpid, c_void};
|
use libc::{
|
||||||
|
bind, c_void, getpid, recvfrom, sendto, sockaddr, sockaddr_storage, socket, AF_NETLINK,
|
||||||
|
SOCK_CLOEXEC, SOCK_DGRAM,
|
||||||
|
};
|
||||||
use nix::libc;
|
use nix::libc;
|
||||||
use std::os::unix::io::RawFd;
|
use std::os::unix::io::RawFd;
|
||||||
use std::{ mem, io};
|
use std::{io, mem};
|
||||||
|
|
||||||
#[repr(C)]
|
#[repr(C)]
|
||||||
struct Nlmsghdr {
|
struct Nlmsghdr {
|
||||||
@ -14,7 +17,11 @@ struct Nlmsghdr {
|
|||||||
|
|
||||||
fn create_netlink_socket() -> io::Result<RawFd> {
|
fn create_netlink_socket() -> io::Result<RawFd> {
|
||||||
let sockfd = unsafe {
|
let sockfd = unsafe {
|
||||||
socket(AF_NETLINK, SOCK_DGRAM | SOCK_CLOEXEC, libc::NETLINK_KOBJECT_UEVENT)
|
socket(
|
||||||
|
AF_NETLINK,
|
||||||
|
SOCK_DGRAM | SOCK_CLOEXEC,
|
||||||
|
libc::NETLINK_KOBJECT_UEVENT,
|
||||||
|
)
|
||||||
};
|
};
|
||||||
|
|
||||||
if sockfd < 0 {
|
if sockfd < 0 {
|
||||||
@ -33,7 +40,11 @@ fn bind_netlink_socket(sock: RawFd) -> io::Result<()> {
|
|||||||
addr.nl_groups = 0;
|
addr.nl_groups = 0;
|
||||||
|
|
||||||
let ret = unsafe {
|
let ret = unsafe {
|
||||||
bind(sock, &addr as *const _ as *const sockaddr, mem::size_of::<libc::sockaddr_nl>() as u32)
|
bind(
|
||||||
|
sock,
|
||||||
|
&addr as *const _ as *const sockaddr,
|
||||||
|
mem::size_of::<libc::sockaddr_nl>() as u32,
|
||||||
|
)
|
||||||
};
|
};
|
||||||
|
|
||||||
if ret < 0 {
|
if ret < 0 {
|
||||||
@ -90,7 +101,10 @@ fn receive_uevent(sock: RawFd) -> io::Result<String> {
|
|||||||
// 检查套接字文件描述符是否有效
|
// 检查套接字文件描述符是否有效
|
||||||
if sock < 0 {
|
if sock < 0 {
|
||||||
println!("Invalid socket file descriptor: {}", sock);
|
println!("Invalid socket file descriptor: {}", sock);
|
||||||
return Err(io::Error::new(io::ErrorKind::InvalidInput, "Invalid socket file descriptor"));
|
return Err(io::Error::new(
|
||||||
|
io::ErrorKind::InvalidInput,
|
||||||
|
"Invalid socket file descriptor",
|
||||||
|
));
|
||||||
}
|
}
|
||||||
|
|
||||||
let mut buf = [0u8; 1024];
|
let mut buf = [0u8; 1024];
|
||||||
@ -100,7 +114,10 @@ fn receive_uevent(sock: RawFd) -> io::Result<String> {
|
|||||||
// 检查缓冲区指针和长度是否有效
|
// 检查缓冲区指针和长度是否有效
|
||||||
if buf.is_empty() {
|
if buf.is_empty() {
|
||||||
println!("Buffer is empty");
|
println!("Buffer is empty");
|
||||||
return Err(io::Error::new(io::ErrorKind::InvalidInput, "Buffer is empty"));
|
return Err(io::Error::new(
|
||||||
|
io::ErrorKind::InvalidInput,
|
||||||
|
"Buffer is empty",
|
||||||
|
));
|
||||||
}
|
}
|
||||||
let len = unsafe {
|
let len = unsafe {
|
||||||
recvfrom(
|
recvfrom(
|
||||||
@ -122,13 +139,19 @@ fn receive_uevent(sock: RawFd) -> io::Result<String> {
|
|||||||
let nlmsghdr_size = mem::size_of::<Nlmsghdr>();
|
let nlmsghdr_size = mem::size_of::<Nlmsghdr>();
|
||||||
if (len as usize) < nlmsghdr_size {
|
if (len as usize) < nlmsghdr_size {
|
||||||
println!("Received message is too short");
|
println!("Received message is too short");
|
||||||
return Err(io::Error::new(io::ErrorKind::InvalidData, "Received message is too short"));
|
return Err(io::Error::new(
|
||||||
|
io::ErrorKind::InvalidData,
|
||||||
|
"Received message is too short",
|
||||||
|
));
|
||||||
}
|
}
|
||||||
|
|
||||||
let nlmsghdr = unsafe { &*(buf.as_ptr() as *const Nlmsghdr) };
|
let nlmsghdr = unsafe { &*(buf.as_ptr() as *const Nlmsghdr) };
|
||||||
if nlmsghdr.nlmsg_len as isize > len {
|
if nlmsghdr.nlmsg_len as isize > len {
|
||||||
println!("Received message is incomplete");
|
println!("Received message is incomplete");
|
||||||
return Err(io::Error::new(io::ErrorKind::InvalidData, "Received message is incomplete"));
|
return Err(io::Error::new(
|
||||||
|
io::ErrorKind::InvalidData,
|
||||||
|
"Received message is incomplete",
|
||||||
|
));
|
||||||
}
|
}
|
||||||
|
|
||||||
let message_data = &buf[nlmsghdr_size..nlmsghdr.nlmsg_len as usize];
|
let message_data = &buf[nlmsghdr_size..nlmsghdr.nlmsg_len as usize];
|
||||||
|
@ -1,8 +1,8 @@
|
|||||||
mod seq_socket;
|
|
||||||
mod seq_pair;
|
mod seq_pair;
|
||||||
|
mod seq_socket;
|
||||||
|
|
||||||
use seq_socket::test_seq_socket;
|
|
||||||
use seq_pair::test_seq_pair;
|
use seq_pair::test_seq_pair;
|
||||||
|
use seq_socket::test_seq_socket;
|
||||||
|
|
||||||
fn main() -> Result<(), std::io::Error> {
|
fn main() -> Result<(), std::io::Error> {
|
||||||
if let Err(e) = test_seq_socket() {
|
if let Err(e) = test_seq_socket() {
|
||||||
|
@ -1,6 +1,6 @@
|
|||||||
use nix::sys::socket::{socketpair, AddressFamily, SockFlag, SockType};
|
use nix::sys::socket::{socketpair, AddressFamily, SockFlag, SockType};
|
||||||
use std::fs::File;
|
use std::fs::File;
|
||||||
use std::io::{Read, Write,Error};
|
use std::io::{Error, Read, Write};
|
||||||
use std::os::fd::FromRawFd;
|
use std::os::fd::FromRawFd;
|
||||||
|
|
||||||
pub fn test_seq_pair() -> Result<(), Error> {
|
pub fn test_seq_pair() -> Result<(), Error> {
|
||||||
@ -10,7 +10,8 @@ pub fn test_seq_pair()->Result<(),Error>{
|
|||||||
SockType::SeqPacket, // 使用 SeqPacket 类型
|
SockType::SeqPacket, // 使用 SeqPacket 类型
|
||||||
None, // 协议默认
|
None, // 协议默认
|
||||||
SockFlag::empty(),
|
SockFlag::empty(),
|
||||||
).expect("Failed to create socket pair");
|
)
|
||||||
|
.expect("Failed to create socket pair");
|
||||||
|
|
||||||
let mut socket1 = unsafe { File::from_raw_fd(sock1) };
|
let mut socket1 = unsafe { File::from_raw_fd(sock1) };
|
||||||
let mut socket2 = unsafe { File::from_raw_fd(sock2) };
|
let mut socket2 = unsafe { File::from_raw_fd(sock2) };
|
||||||
|
@ -1,16 +1,14 @@
|
|||||||
|
|
||||||
use libc::*;
|
use libc::*;
|
||||||
use std::{fs, str};
|
|
||||||
use std::ffi::CString;
|
use std::ffi::CString;
|
||||||
use std::io::Error;
|
use std::io::Error;
|
||||||
use std::mem;
|
use std::mem;
|
||||||
use std::os::unix::io::RawFd;
|
use std::os::unix::io::RawFd;
|
||||||
|
use std::{fs, str};
|
||||||
|
|
||||||
const SOCKET_PATH: &str = "/test.seqpacket";
|
const SOCKET_PATH: &str = "/test.seqpacket";
|
||||||
const MSG1: &str = "Hello, Unix SEQPACKET socket from Client!";
|
const MSG1: &str = "Hello, Unix SEQPACKET socket from Client!";
|
||||||
const MSG2: &str = "Hello, Unix SEQPACKET socket from Server!";
|
const MSG2: &str = "Hello, Unix SEQPACKET socket from Server!";
|
||||||
|
|
||||||
|
|
||||||
fn create_seqpacket_socket() -> Result<RawFd, Error> {
|
fn create_seqpacket_socket() -> Result<RawFd, Error> {
|
||||||
unsafe {
|
unsafe {
|
||||||
let fd = socket(AF_UNIX, SOCK_SEQPACKET, 0);
|
let fd = socket(AF_UNIX, SOCK_SEQPACKET, 0);
|
||||||
@ -33,7 +31,12 @@ fn bind_socket(fd: RawFd) -> Result<(), Error> {
|
|||||||
addr.sun_path[i] = byte as i8;
|
addr.sun_path[i] = byte as i8;
|
||||||
}
|
}
|
||||||
|
|
||||||
if bind(fd, &addr as *const _ as *const sockaddr, mem::size_of_val(&addr) as socklen_t) == -1 {
|
if bind(
|
||||||
|
fd,
|
||||||
|
&addr as *const _ as *const sockaddr,
|
||||||
|
mem::size_of_val(&addr) as socklen_t,
|
||||||
|
) == -1
|
||||||
|
{
|
||||||
return Err(Error::last_os_error());
|
return Err(Error::last_os_error());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -68,7 +71,13 @@ fn accept_connection(fd: RawFd) -> Result<RawFd, Error> {
|
|||||||
fn send_message(fd: RawFd, msg: &str) -> Result<(), Error> {
|
fn send_message(fd: RawFd, msg: &str) -> Result<(), Error> {
|
||||||
unsafe {
|
unsafe {
|
||||||
let msg_bytes = msg.as_bytes();
|
let msg_bytes = msg.as_bytes();
|
||||||
if send(fd, msg_bytes.as_ptr() as *const libc::c_void, msg_bytes.len(), 0) == -1 {
|
if send(
|
||||||
|
fd,
|
||||||
|
msg_bytes.as_ptr() as *const libc::c_void,
|
||||||
|
msg_bytes.len(),
|
||||||
|
0,
|
||||||
|
) == -1
|
||||||
|
{
|
||||||
return Err(Error::last_os_error());
|
return Err(Error::last_os_error());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -78,7 +87,12 @@ fn send_message(fd: RawFd, msg: &str) -> Result<(), Error> {
|
|||||||
fn receive_message(fd: RawFd) -> Result<String, Error> {
|
fn receive_message(fd: RawFd) -> Result<String, Error> {
|
||||||
let mut buffer = [0; 1024];
|
let mut buffer = [0; 1024];
|
||||||
unsafe {
|
unsafe {
|
||||||
let len = recv(fd, buffer.as_mut_ptr() as *mut libc::c_void, buffer.len(), 0);
|
let len = recv(
|
||||||
|
fd,
|
||||||
|
buffer.as_mut_ptr() as *mut libc::c_void,
|
||||||
|
buffer.len(),
|
||||||
|
0,
|
||||||
|
);
|
||||||
if len == -1 {
|
if len == -1 {
|
||||||
return Err(Error::last_os_error());
|
return Err(Error::last_os_error());
|
||||||
}
|
}
|
||||||
@ -121,7 +135,12 @@ pub fn test_seq_socket() ->Result<(), Error>{
|
|||||||
for (i, &byte) in path_bytes.iter().enumerate() {
|
for (i, &byte) in path_bytes.iter().enumerate() {
|
||||||
addr.sun_path[i] = byte as i8;
|
addr.sun_path[i] = byte as i8;
|
||||||
}
|
}
|
||||||
if connect(client_fd, &addr as *const _ as *const sockaddr, mem::size_of_val(&addr) as socklen_t) == -1 {
|
if connect(
|
||||||
|
client_fd,
|
||||||
|
&addr as *const _ as *const sockaddr,
|
||||||
|
mem::size_of_val(&addr) as socklen_t,
|
||||||
|
) == -1
|
||||||
|
{
|
||||||
return Err(Error::last_os_error());
|
return Err(Error::last_os_error());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -140,9 +159,16 @@ pub fn test_seq_socket() ->Result<(), Error>{
|
|||||||
return Err(Error::last_os_error());
|
return Err(Error::last_os_error());
|
||||||
}
|
}
|
||||||
let sun_path = addrss.sun_path.clone();
|
let sun_path = addrss.sun_path.clone();
|
||||||
let peer_path:[u8;108] = sun_path.iter().map(|&x| x as u8).collect::<Vec<u8>>().try_into().unwrap();
|
let peer_path: [u8; 108] = sun_path
|
||||||
println!("Client: Connected to server at path: {}", String::from_utf8_lossy(&peer_path));
|
.iter()
|
||||||
|
.map(|&x| x as u8)
|
||||||
|
.collect::<Vec<u8>>()
|
||||||
|
.try_into()
|
||||||
|
.unwrap();
|
||||||
|
println!(
|
||||||
|
"Client: Connected to server at path: {}",
|
||||||
|
String::from_utf8_lossy(&peer_path)
|
||||||
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
server_thread.join().expect("Server thread panicked");
|
server_thread.join().expect("Server thread panicked");
|
||||||
|
@ -1,9 +1,9 @@
|
|||||||
use std::io::Error;
|
|
||||||
use std::os::fd::RawFd;
|
|
||||||
use std::fs;
|
|
||||||
use libc::*;
|
use libc::*;
|
||||||
use std::ffi::CString;
|
use std::ffi::CString;
|
||||||
|
use std::fs;
|
||||||
|
use std::io::Error;
|
||||||
use std::mem;
|
use std::mem;
|
||||||
|
use std::os::fd::RawFd;
|
||||||
|
|
||||||
const SOCKET_PATH: &str = "/test.stream";
|
const SOCKET_PATH: &str = "/test.stream";
|
||||||
const MSG1: &str = "Hello, unix stream socket from Client!";
|
const MSG1: &str = "Hello, unix stream socket from Client!";
|
||||||
@ -13,7 +13,7 @@ fn create_stream_socket() -> Result<RawFd, Error>{
|
|||||||
unsafe {
|
unsafe {
|
||||||
let fd = socket(AF_UNIX, SOCK_STREAM, 0);
|
let fd = socket(AF_UNIX, SOCK_STREAM, 0);
|
||||||
if fd == -1 {
|
if fd == -1 {
|
||||||
return Err(Error::last_os_error())
|
return Err(Error::last_os_error());
|
||||||
}
|
}
|
||||||
Ok(fd)
|
Ok(fd)
|
||||||
}
|
}
|
||||||
@ -31,7 +31,12 @@ fn bind_socket(fd: RawFd) -> Result<(), Error> {
|
|||||||
addr.sun_path[i] = byte as i8;
|
addr.sun_path[i] = byte as i8;
|
||||||
}
|
}
|
||||||
|
|
||||||
if bind(fd, &addr as *const _ as *const sockaddr, mem::size_of_val(&addr) as socklen_t) == -1 {
|
if bind(
|
||||||
|
fd,
|
||||||
|
&addr as *const _ as *const sockaddr,
|
||||||
|
mem::size_of_val(&addr) as socklen_t,
|
||||||
|
) == -1
|
||||||
|
{
|
||||||
return Err(Error::last_os_error());
|
return Err(Error::last_os_error());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -61,7 +66,13 @@ fn accept_conn(fd: RawFd) -> Result<RawFd, Error> {
|
|||||||
fn send_message(fd: RawFd, msg: &str) -> Result<(), Error> {
|
fn send_message(fd: RawFd, msg: &str) -> Result<(), Error> {
|
||||||
unsafe {
|
unsafe {
|
||||||
let msg_bytes = msg.as_bytes();
|
let msg_bytes = msg.as_bytes();
|
||||||
if send(fd, msg_bytes.as_ptr() as *const libc::c_void, msg_bytes.len(), 0)== -1 {
|
if send(
|
||||||
|
fd,
|
||||||
|
msg_bytes.as_ptr() as *const libc::c_void,
|
||||||
|
msg_bytes.len(),
|
||||||
|
0,
|
||||||
|
) == -1
|
||||||
|
{
|
||||||
return Err(Error::last_os_error());
|
return Err(Error::last_os_error());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -71,7 +82,12 @@ fn send_message(fd: RawFd, msg: &str) -> Result<(), Error> {
|
|||||||
fn recv_message(fd: RawFd) -> Result<String, Error> {
|
fn recv_message(fd: RawFd) -> Result<String, Error> {
|
||||||
let mut buffer = [0; 1024];
|
let mut buffer = [0; 1024];
|
||||||
unsafe {
|
unsafe {
|
||||||
let len = recv(fd, buffer.as_mut_ptr() as *mut libc::c_void, buffer.len(),0);
|
let len = recv(
|
||||||
|
fd,
|
||||||
|
buffer.as_mut_ptr() as *mut libc::c_void,
|
||||||
|
buffer.len(),
|
||||||
|
0,
|
||||||
|
);
|
||||||
if len == -1 {
|
if len == -1 {
|
||||||
return Err(Error::last_os_error());
|
return Err(Error::last_os_error());
|
||||||
}
|
}
|
||||||
@ -111,7 +127,12 @@ fn test_stream() -> Result<(), Error> {
|
|||||||
addr.sun_path[i] = byte as i8;
|
addr.sun_path[i] = byte as i8;
|
||||||
}
|
}
|
||||||
|
|
||||||
if connect(client_fd, &addr as *const _ as *const sockaddr, mem::size_of_val(&addr) as socklen_t) == -1 {
|
if connect(
|
||||||
|
client_fd,
|
||||||
|
&addr as *const _ as *const sockaddr,
|
||||||
|
mem::size_of_val(&addr) as socklen_t,
|
||||||
|
) == -1
|
||||||
|
{
|
||||||
return Err(Error::last_os_error());
|
return Err(Error::last_os_error());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -129,9 +150,16 @@ fn test_stream() -> Result<(), Error> {
|
|||||||
return Err(Error::last_os_error());
|
return Err(Error::last_os_error());
|
||||||
}
|
}
|
||||||
let sun_path = addrss.sun_path.clone();
|
let sun_path = addrss.sun_path.clone();
|
||||||
let peer_path:[u8;108] = sun_path.iter().map(|&x| x as u8).collect::<Vec<u8>>().try_into().unwrap();
|
let peer_path: [u8; 108] = sun_path
|
||||||
println!("Client: Connected to server at path: {}", String::from_utf8_lossy(&peer_path));
|
.iter()
|
||||||
|
.map(|&x| x as u8)
|
||||||
|
.collect::<Vec<u8>>()
|
||||||
|
.try_into()
|
||||||
|
.unwrap();
|
||||||
|
println!(
|
||||||
|
"Client: Connected to server at path: {}",
|
||||||
|
String::from_utf8_lossy(&peer_path)
|
||||||
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
server_thread.join().expect("Server thread panicked");
|
server_thread.join().expect("Server thread panicked");
|
||||||
@ -148,6 +176,6 @@ fn test_stream() -> Result<(), Error> {
|
|||||||
fn main() {
|
fn main() {
|
||||||
match test_stream() {
|
match test_stream() {
|
||||||
Ok(_) => println!("test for unix stream success"),
|
Ok(_) => println!("test for unix stream success"),
|
||||||
Err(_) => println!("test for unix stream failed")
|
Err(_) => println!("test for unix stream failed"),
|
||||||
}
|
}
|
||||||
}
|
}
|
Loading…
x
Reference in New Issue
Block a user