Noratrieb/survey
1
0
Fork 0
mirror of https://github.com/Noratrieb/survey.git synced 2026-01-14 16:35:09 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

handle connection

Browse source
This commit is contained in:
nora 2022-03-31 23:09:09 +02:00
parent e5484affdf
commit 43df60f510
1 changed files with 15 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

22
src/listener.rs
View file

@ -1,10 +1,18 @@
use std::mem::MaybeUninit; use std::mem::MaybeUninit;
use std::{io, mem}; use std::{io, mem};
const PORT: libc::in_port_t = 1234; const PORT: libc::in_port_t = 1112;
const SOCKADDR_IN_SIZE: libc::socklen_t = mem::size_of::<libc::sockaddr_in>() as _; const SOCKADDR_IN_SIZE: libc::socklen_t = mem::size_of::<libc::sockaddr_in>() as _;
macro_rules! check_zero {
($result:expr) => {
if $result != 0 {
return Err(io::Error::last_os_error());
}
};
}
pub fn listener() -> io::Result<()> { pub fn listener() -> io::Result<()> {
unsafe { unsafe {
let socket = libc::socket(libc::AF_INET, libc::SOCK_STREAM, 0); let socket = libc::socket(libc::AF_INET, libc::SOCK_STREAM, 0);
@ -17,7 +25,7 @@ pub fn listener() -> io::Result<()> {
let addr = libc::sockaddr_in { let addr = libc::sockaddr_in {
sin_family: libc::AF_INET.try_into().unwrap(), sin_family: libc::AF_INET.try_into().unwrap(),
sin_port: PORT, sin_port: PORT.to_be(),
sin_addr: libc::in_addr { sin_addr: libc::in_addr {
s_addr: libc::INADDR_ANY, s_addr: libc::INADDR_ANY,
}, },
@ -32,21 +40,21 @@ pub fn listener() -> io::Result<()> {
println!("Bound socket ({socket}) on port {PORT}"); println!("Bound socket ({socket}) on port {PORT}");
let result = libc::listen(socket, 5); check_zero!(libc::listen(socket, 5));
if result != 0 {
return Err(io::Error::last_os_error());
}
println!("Listening on socket ({socket})"); println!("Listening on socket ({socket})");
let mut peer_sockaddr = MaybeUninit::uninit(); let mut peer_sockaddr = MaybeUninit::uninit();
let mut sockaddr_size = 0; let mut sockaddr_size = 0;
let connection = libc::accept(socket, peer_sockaddr.as_mut_ptr(), &mut sockaddr_size); let connection = libc::accept(socket, peer_sockaddr.as_mut_ptr(), &mut sockaddr_size);
if connection < 0 { if connection == -1 {
return Err(io::Error::last_os_error()); return Err(io::Error::last_os_error());
} }
println!("Received connection! (connfd={connection})"); println!("Received connection! (connfd={connection})");
check_zero!(libc::close(connection));
check_zero!(libc::close(socket));
} }
Ok(()) Ok(())