YGGverse/aquatic
1
0
Fork 0
mirror of https://github.com/YGGverse/aquatic.git synced 2026-04-02 10:45:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Move things to aquatic_common_tcp

Browse source
This commit is contained in:
Joakim Frostegård 2020-07-02 16:54:27 +02:00
parent 720596dfb4
commit 82378e71f6
8 changed files with 61 additions and 45 deletions
Download patch file Download diff file Expand all files Collapse all files

4
aquatic_http/src/lib/network/mod.rs
View file

@ -10,6 +10,8 @@ use native_tls::TlsAcceptor;
use mio::{Events, Poll, Interest, Token};
use mio::net::TcpListener;
use aquatic_common_tcp::network::create_listener;
use crate::common::*;
use crate::config::Config;
use crate::protocol::*;
@ -67,7 +69,7 @@ pub fn run_socket_worker(
response_channel_receiver: ResponseChannelReceiver,
opt_tls_acceptor: Option<TlsAcceptor>,
){
match create_listener(&config){
match create_listener(&config.network.socket){
Ok(listener) => {
socket_worker_statuses.lock()[socket_worker_index] = Some(Ok(()));

32
aquatic_http/src/lib/network/utils.rs
View file

@ -2,42 +2,10 @@ use std::time::Instant;
use anyhow::Context;
use mio::Token;
use socket2::{Socket, Domain, Type, Protocol};
use crate::config::Config;
use super::*;
// will be almost identical to ws version
pub fn create_listener(
config: &Config
) -> ::anyhow::Result<::std::net::TcpListener> {
let builder = if config.network.address.is_ipv4(){
Socket::new(Domain::ipv4(), Type::stream(), Some(Protocol::tcp()))
} else {
Socket::new(Domain::ipv6(), Type::stream(), Some(Protocol::tcp()))
}.context("Couldn't create socket2::Socket")?;
if config.network.ipv6_only {
builder.set_only_v6(true)
.context("Couldn't put socket in ipv6 only mode")?
}
builder.set_nonblocking(true)
.context("Couldn't put socket in non-blocking mode")?;
builder.set_reuse_port(true)
.context("Couldn't put socket in reuse_port mode")?;
builder.bind(&config.network.address.into()).with_context(||
format!("Couldn't bind socket to address {}", config.network.address)
)?;
builder.listen(128)
.context("Couldn't listen for connections on socket")?;
Ok(builder.into_tcp_listener())
}
/// Don't bother with deregistering from Poll. In my understanding, this is
/// done automatically when the stream is dropped, as long as there are no
/// other references to the file descriptor, such as when it is accessed