itsi 0.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +7 -0
- data/Cargo.lock +1541 -0
- data/Cargo.toml +13 -0
- data/README.md +82 -0
- data/Rakefile +87 -0
- data/crates/itsi_error/Cargo.lock +368 -0
- data/crates/itsi_error/Cargo.toml +9 -0
- data/crates/itsi_error/src/lib.rs +49 -0
- data/crates/itsi_rb_helpers/Cargo.lock +355 -0
- data/crates/itsi_rb_helpers/Cargo.toml +8 -0
- data/crates/itsi_rb_helpers/src/lib.rs +98 -0
- data/crates/itsi_scheduler/Cargo.toml +16 -0
- data/crates/itsi_scheduler/extconf.rb +6 -0
- data/crates/itsi_scheduler/src/lib.rs +17 -0
- data/crates/itsi_server/Cargo.toml +29 -0
- data/crates/itsi_server/extconf.rb +6 -0
- data/crates/itsi_server/src/lib.rs +52 -0
- data/crates/itsi_server/src/request/itsi_request.rs +143 -0
- data/crates/itsi_server/src/request/mod.rs +1 -0
- data/crates/itsi_server/src/server/bind.rs +138 -0
- data/crates/itsi_server/src/server/itsi_ca/itsi_ca.crt +32 -0
- data/crates/itsi_server/src/server/itsi_ca/itsi_ca.key +52 -0
- data/crates/itsi_server/src/server/itsi_server.rs +182 -0
- data/crates/itsi_server/src/server/listener.rs +218 -0
- data/crates/itsi_server/src/server/mod.rs +5 -0
- data/crates/itsi_server/src/server/tls.rs +138 -0
- data/crates/itsi_server/src/server/transfer_protocol.rs +23 -0
- data/crates/itsi_server/src/stream_writer/mod.rs +21 -0
- data/crates/itsi_tracing/Cargo.lock +274 -0
- data/crates/itsi_tracing/Cargo.toml +12 -0
- data/crates/itsi_tracing/src/lib.rs +11 -0
- data/gems/scheduler/.gitignore +14 -0
- data/gems/scheduler/.rubocop.yml +8 -0
- data/gems/scheduler/CHANGELOG.md +5 -0
- data/gems/scheduler/CODE_OF_CONDUCT.md +132 -0
- data/gems/scheduler/Cargo.lock +745 -0
- data/gems/scheduler/Cargo.toml +3 -0
- data/gems/scheduler/LICENSE.txt +21 -0
- data/gems/scheduler/README.md +43 -0
- data/gems/scheduler/Rakefile +22 -0
- data/gems/scheduler/bin/console +11 -0
- data/gems/scheduler/bin/setup +8 -0
- data/gems/scheduler/ext/itsi_error/Cargo.lock +368 -0
- data/gems/scheduler/ext/itsi_error/Cargo.toml +9 -0
- data/gems/scheduler/ext/itsi_error/src/lib.rs +49 -0
- data/gems/scheduler/ext/itsi_rb_helpers/Cargo.lock +355 -0
- data/gems/scheduler/ext/itsi_rb_helpers/Cargo.toml +8 -0
- data/gems/scheduler/ext/itsi_rb_helpers/src/lib.rs +98 -0
- data/gems/scheduler/ext/itsi_scheduler/Cargo.toml +16 -0
- data/gems/scheduler/ext/itsi_scheduler/extconf.rb +6 -0
- data/gems/scheduler/ext/itsi_scheduler/src/lib.rs +17 -0
- data/gems/scheduler/ext/itsi_tracing/Cargo.lock +274 -0
- data/gems/scheduler/ext/itsi_tracing/Cargo.toml +12 -0
- data/gems/scheduler/ext/itsi_tracing/src/lib.rs +11 -0
- data/gems/scheduler/itsi-scheduler.gemspec +43 -0
- data/gems/scheduler/lib/itsi/scheduler/version.rb +7 -0
- data/gems/scheduler/lib/itsi/scheduler.rb +11 -0
- data/gems/scheduler/sig/itsi_scheduler.rbs +4 -0
- data/gems/scheduler/test/test_helper.rb +6 -0
- data/gems/scheduler/test/test_itsi_scheduler.rb +9 -0
- data/gems/server/.gitignore +14 -0
- data/gems/server/.rubocop.yml +8 -0
- data/gems/server/CHANGELOG.md +5 -0
- data/gems/server/CODE_OF_CONDUCT.md +132 -0
- data/gems/server/LICENSE.txt +21 -0
- data/gems/server/README.md +43 -0
- data/gems/server/Rakefile +22 -0
- data/gems/server/bin/console +11 -0
- data/gems/server/bin/setup +8 -0
- data/gems/server/exe/itsi +84 -0
- data/gems/server/ext/itsi_error/Cargo.lock +368 -0
- data/gems/server/ext/itsi_error/Cargo.toml +9 -0
- data/gems/server/ext/itsi_error/src/lib.rs +49 -0
- data/gems/server/ext/itsi_rb_helpers/Cargo.lock +355 -0
- data/gems/server/ext/itsi_rb_helpers/Cargo.toml +8 -0
- data/gems/server/ext/itsi_rb_helpers/src/lib.rs +98 -0
- data/gems/server/ext/itsi_server/Cargo.toml +29 -0
- data/gems/server/ext/itsi_server/extconf.rb +6 -0
- data/gems/server/ext/itsi_server/src/lib.rs +52 -0
- data/gems/server/ext/itsi_server/src/request/itsi_request.rs +143 -0
- data/gems/server/ext/itsi_server/src/request/mod.rs +1 -0
- data/gems/server/ext/itsi_server/src/server/bind.rs +138 -0
- data/gems/server/ext/itsi_server/src/server/itsi_ca/itsi_ca.crt +32 -0
- data/gems/server/ext/itsi_server/src/server/itsi_ca/itsi_ca.key +52 -0
- data/gems/server/ext/itsi_server/src/server/itsi_server.rs +182 -0
- data/gems/server/ext/itsi_server/src/server/listener.rs +218 -0
- data/gems/server/ext/itsi_server/src/server/mod.rs +5 -0
- data/gems/server/ext/itsi_server/src/server/tls.rs +138 -0
- data/gems/server/ext/itsi_server/src/server/transfer_protocol.rs +23 -0
- data/gems/server/ext/itsi_server/src/stream_writer/mod.rs +21 -0
- data/gems/server/ext/itsi_tracing/Cargo.lock +274 -0
- data/gems/server/ext/itsi_tracing/Cargo.toml +12 -0
- data/gems/server/ext/itsi_tracing/src/lib.rs +11 -0
- data/gems/server/itsi-server.gemspec +45 -0
- data/gems/server/lib/itsi/request.rb +39 -0
- data/gems/server/lib/itsi/server/version.rb +7 -0
- data/gems/server/lib/itsi/server.rb +21 -0
- data/gems/server/sig/itsi_server.rbs +4 -0
- data/gems/server/test/test_helper.rb +5 -0
- data/gems/server/test/test_itsi_server.rb +9 -0
- data/lib/itsi/version.rb +3 -0
- data/lib/itsi.rb +6 -0
- metadata +172 -0
|
@@ -0,0 +1,143 @@
|
|
|
1
|
+
use std::{collections::HashMap, sync::Arc};
|
|
2
|
+
|
|
3
|
+
use crate::server::listener::{Listener, SockAddr};
|
|
4
|
+
use bytes::Bytes;
|
|
5
|
+
use http::request::Parts;
|
|
6
|
+
use http_body_util::BodyExt;
|
|
7
|
+
use hyper::{body::Incoming, Request};
|
|
8
|
+
use magnus::error::Result;
|
|
9
|
+
|
|
10
|
+
#[magnus::wrap(class = "Itsi::Request", free_immediately, size)]
|
|
11
|
+
#[derive(Debug)]
|
|
12
|
+
pub struct ItsiRequest {
|
|
13
|
+
pub path: String,
|
|
14
|
+
pub script_name: String,
|
|
15
|
+
pub query_string: String,
|
|
16
|
+
pub method: String,
|
|
17
|
+
pub version: String,
|
|
18
|
+
pub rack_protocol: Vec<String>,
|
|
19
|
+
pub host: String,
|
|
20
|
+
pub scheme: String,
|
|
21
|
+
pub headers: HashMap<String, String>,
|
|
22
|
+
pub remote_addr: String,
|
|
23
|
+
pub port: u16,
|
|
24
|
+
pub body: Bytes,
|
|
25
|
+
pub parts: Parts,
|
|
26
|
+
}
|
|
27
|
+
|
|
28
|
+
impl ItsiRequest {
|
|
29
|
+
pub(crate) async fn build_from(
|
|
30
|
+
request: Request<Incoming>,
|
|
31
|
+
sock_addr: SockAddr,
|
|
32
|
+
script_name: String,
|
|
33
|
+
listener: Arc<Listener>,
|
|
34
|
+
) -> Self {
|
|
35
|
+
let (parts, body) = request.into_parts();
|
|
36
|
+
let method = parts.method.to_string();
|
|
37
|
+
let port = parts.uri.port_u16().unwrap_or(listener.port());
|
|
38
|
+
let query_string = parts.uri.query().unwrap_or("").to_string();
|
|
39
|
+
let rack_protocol = parts
|
|
40
|
+
.headers
|
|
41
|
+
.get("upgrade")
|
|
42
|
+
.or_else(|| parts.headers.get("protocol"))
|
|
43
|
+
.map(|value| {
|
|
44
|
+
value
|
|
45
|
+
.to_str()
|
|
46
|
+
.unwrap_or("")
|
|
47
|
+
.split(',')
|
|
48
|
+
.map(|s| s.trim().to_owned())
|
|
49
|
+
.collect::<Vec<String>>()
|
|
50
|
+
})
|
|
51
|
+
.unwrap_or_else(|| vec!["http".to_string()]);
|
|
52
|
+
|
|
53
|
+
let host = parts
|
|
54
|
+
.uri
|
|
55
|
+
.host()
|
|
56
|
+
.map(ToOwned::to_owned)
|
|
57
|
+
.unwrap_or_else(|| listener.host());
|
|
58
|
+
|
|
59
|
+
let scheme = parts
|
|
60
|
+
.uri
|
|
61
|
+
.scheme()
|
|
62
|
+
.map(|s| s.to_string())
|
|
63
|
+
.unwrap_or_else(|| listener.scheme());
|
|
64
|
+
|
|
65
|
+
let headers = parts
|
|
66
|
+
.headers
|
|
67
|
+
.iter()
|
|
68
|
+
.map(|(k, v)| (k.to_string(), v.to_str().unwrap_or("").to_string()))
|
|
69
|
+
.collect();
|
|
70
|
+
|
|
71
|
+
let path = parts
|
|
72
|
+
.uri
|
|
73
|
+
.path()
|
|
74
|
+
.strip_prefix(&script_name)
|
|
75
|
+
.unwrap_or(parts.uri.path())
|
|
76
|
+
.to_string();
|
|
77
|
+
|
|
78
|
+
let version = format!("{:?}", parts.version);
|
|
79
|
+
let body = body.collect().await.unwrap().to_bytes();
|
|
80
|
+
|
|
81
|
+
Self {
|
|
82
|
+
remote_addr: sock_addr.to_string(),
|
|
83
|
+
body,
|
|
84
|
+
parts,
|
|
85
|
+
script_name,
|
|
86
|
+
query_string,
|
|
87
|
+
method,
|
|
88
|
+
headers,
|
|
89
|
+
path,
|
|
90
|
+
version,
|
|
91
|
+
rack_protocol,
|
|
92
|
+
host,
|
|
93
|
+
scheme,
|
|
94
|
+
port,
|
|
95
|
+
}
|
|
96
|
+
}
|
|
97
|
+
}
|
|
98
|
+
|
|
99
|
+
impl ItsiRequest {
|
|
100
|
+
pub(crate) fn path(&self) -> Result<String> {
|
|
101
|
+
Ok(self.path.clone())
|
|
102
|
+
}
|
|
103
|
+
|
|
104
|
+
pub(crate) fn script_name(&self) -> Result<String> {
|
|
105
|
+
Ok(self.script_name.clone())
|
|
106
|
+
}
|
|
107
|
+
|
|
108
|
+
pub(crate) fn query_string(&self) -> Result<String> {
|
|
109
|
+
Ok(self.query_string.clone())
|
|
110
|
+
}
|
|
111
|
+
|
|
112
|
+
pub(crate) fn method(&self) -> Result<String> {
|
|
113
|
+
Ok(self.method.clone())
|
|
114
|
+
}
|
|
115
|
+
|
|
116
|
+
pub(crate) fn version(&self) -> Result<String> {
|
|
117
|
+
Ok(self.version.clone())
|
|
118
|
+
}
|
|
119
|
+
|
|
120
|
+
pub(crate) fn rack_protocol(&self) -> Result<Vec<String>> {
|
|
121
|
+
Ok(self.rack_protocol.clone())
|
|
122
|
+
}
|
|
123
|
+
|
|
124
|
+
pub(crate) fn host(&self) -> Result<String> {
|
|
125
|
+
Ok(self.host.clone())
|
|
126
|
+
}
|
|
127
|
+
|
|
128
|
+
pub(crate) fn headers(&self) -> Result<HashMap<String, String>> {
|
|
129
|
+
Ok(self.headers.clone())
|
|
130
|
+
}
|
|
131
|
+
|
|
132
|
+
pub(crate) fn remote_addr(&self) -> Result<String> {
|
|
133
|
+
Ok(self.remote_addr.clone())
|
|
134
|
+
}
|
|
135
|
+
|
|
136
|
+
pub(crate) fn port(&self) -> Result<u16> {
|
|
137
|
+
Ok(self.port)
|
|
138
|
+
}
|
|
139
|
+
|
|
140
|
+
pub(crate) fn body(&self) -> Result<Bytes> {
|
|
141
|
+
Ok(self.body.clone())
|
|
142
|
+
}
|
|
143
|
+
}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
pub mod itsi_request;
|
|
@@ -0,0 +1,138 @@
|
|
|
1
|
+
use super::{tls::configure_tls, transfer_protocol::TransferProtocol};
|
|
2
|
+
use itsi_error::ItsiError;
|
|
3
|
+
use std::{
|
|
4
|
+
collections::HashMap,
|
|
5
|
+
net::{IpAddr, Ipv4Addr, Ipv6Addr, ToSocketAddrs},
|
|
6
|
+
path::PathBuf,
|
|
7
|
+
str::FromStr,
|
|
8
|
+
};
|
|
9
|
+
use tokio_rustls::rustls::ServerConfig;
|
|
10
|
+
|
|
11
|
+
// Support binding to either IP or Unix Socket
|
|
12
|
+
#[derive(Debug, Clone)]
|
|
13
|
+
pub enum BindAddress {
|
|
14
|
+
Ip(IpAddr),
|
|
15
|
+
UnixSocket(PathBuf),
|
|
16
|
+
}
|
|
17
|
+
|
|
18
|
+
impl Default for BindAddress {
|
|
19
|
+
fn default() -> Self {
|
|
20
|
+
BindAddress::Ip(IpAddr::V4(Ipv4Addr::UNSPECIFIED))
|
|
21
|
+
}
|
|
22
|
+
}
|
|
23
|
+
|
|
24
|
+
#[derive(Debug, Default, Clone)]
|
|
25
|
+
#[magnus::wrap(class = "Itsi::Bind")]
|
|
26
|
+
pub struct Bind {
|
|
27
|
+
pub address: BindAddress,
|
|
28
|
+
pub port: Option<u16>, // None for Unix Sockets
|
|
29
|
+
pub protocol: TransferProtocol,
|
|
30
|
+
pub tls_config: Option<ServerConfig>,
|
|
31
|
+
}
|
|
32
|
+
|
|
33
|
+
impl FromStr for Bind {
|
|
34
|
+
type Err = ItsiError;
|
|
35
|
+
|
|
36
|
+
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
|
37
|
+
let (protocol, remainder) = if let Some((proto, rest)) = s.split_once("://") {
|
|
38
|
+
(proto.parse::<TransferProtocol>()?, rest)
|
|
39
|
+
} else {
|
|
40
|
+
(TransferProtocol::Https, s)
|
|
41
|
+
};
|
|
42
|
+
|
|
43
|
+
let (url, options) = if let Some((base, options)) = remainder.split_once('?') {
|
|
44
|
+
(base, parse_bind_options(options))
|
|
45
|
+
} else {
|
|
46
|
+
(remainder, HashMap::new())
|
|
47
|
+
};
|
|
48
|
+
|
|
49
|
+
let (host, port) = if url.starts_with('[') {
|
|
50
|
+
// IPv6 with brackets `[::]:port`
|
|
51
|
+
if let Some(end) = url.find(']') {
|
|
52
|
+
let host = &url[1..end]; // Extract `::`
|
|
53
|
+
let port = url[end + 1..]
|
|
54
|
+
.strip_prefix(':')
|
|
55
|
+
.and_then(|p| p.parse().ok());
|
|
56
|
+
(host, port)
|
|
57
|
+
} else {
|
|
58
|
+
return Err(ItsiError::InvalidInput(
|
|
59
|
+
"Invalid IPv6 address format".to_owned(),
|
|
60
|
+
));
|
|
61
|
+
}
|
|
62
|
+
} else if let Some((h, p)) = url.rsplit_once(':') {
|
|
63
|
+
// Check if `h` is an IPv6 address before assuming it's a port
|
|
64
|
+
if h.contains('.') || h.parse::<Ipv4Addr>().is_ok() {
|
|
65
|
+
(h, p.parse::<u16>().ok()) // IPv4 case
|
|
66
|
+
} else if h.parse::<Ipv6Addr>().is_ok() {
|
|
67
|
+
// If it's IPv6, require brackets for port
|
|
68
|
+
return Err(ItsiError::InvalidInput(
|
|
69
|
+
"IPv6 addresses must use [ ] when specifying a port".to_owned(),
|
|
70
|
+
));
|
|
71
|
+
} else {
|
|
72
|
+
(h, None) // Treat as a hostname
|
|
73
|
+
}
|
|
74
|
+
} else {
|
|
75
|
+
(url, None)
|
|
76
|
+
};
|
|
77
|
+
|
|
78
|
+
let address = if let Ok(ip) = host.parse::<IpAddr>() {
|
|
79
|
+
BindAddress::Ip(ip)
|
|
80
|
+
} else {
|
|
81
|
+
resolve_hostname(host)
|
|
82
|
+
.map(BindAddress::Ip)
|
|
83
|
+
.unwrap_or(BindAddress::Ip(IpAddr::V4(Ipv4Addr::UNSPECIFIED)))
|
|
84
|
+
};
|
|
85
|
+
let (port, address) = match protocol {
|
|
86
|
+
TransferProtocol::Http => (port.or(Some(80)), address),
|
|
87
|
+
TransferProtocol::Https => (port.or(Some(443)), address),
|
|
88
|
+
TransferProtocol::Unix => (None, BindAddress::UnixSocket(host.into())),
|
|
89
|
+
};
|
|
90
|
+
|
|
91
|
+
let tls_config = if let TransferProtocol::Http = protocol {
|
|
92
|
+
None
|
|
93
|
+
} else if let TransferProtocol::Https = protocol {
|
|
94
|
+
Some(configure_tls(host, &options)?)
|
|
95
|
+
} else if options.contains_key("cert") {
|
|
96
|
+
Some(configure_tls(host, &options)?)
|
|
97
|
+
} else {
|
|
98
|
+
None
|
|
99
|
+
};
|
|
100
|
+
|
|
101
|
+
Ok(Self {
|
|
102
|
+
address,
|
|
103
|
+
port,
|
|
104
|
+
protocol,
|
|
105
|
+
tls_config,
|
|
106
|
+
})
|
|
107
|
+
}
|
|
108
|
+
}
|
|
109
|
+
|
|
110
|
+
fn parse_bind_options(query: &str) -> HashMap<String, String> {
|
|
111
|
+
query
|
|
112
|
+
.split('&')
|
|
113
|
+
.filter_map(|pair| pair.split_once('='))
|
|
114
|
+
.map(|(k, v)| (k.to_owned(), v.to_owned()))
|
|
115
|
+
.collect()
|
|
116
|
+
}
|
|
117
|
+
|
|
118
|
+
/// Attempts to resolve a hostname into an IP address.
|
|
119
|
+
fn resolve_hostname(hostname: &str) -> Option<IpAddr> {
|
|
120
|
+
(hostname, 0)
|
|
121
|
+
.to_socket_addrs()
|
|
122
|
+
.ok()?
|
|
123
|
+
.filter_map(|addr| {
|
|
124
|
+
if addr.is_ipv6() {
|
|
125
|
+
Some(addr.ip()) // Prefer IPv6
|
|
126
|
+
} else {
|
|
127
|
+
None
|
|
128
|
+
}
|
|
129
|
+
})
|
|
130
|
+
.next()
|
|
131
|
+
.or_else(|| {
|
|
132
|
+
(hostname, 0)
|
|
133
|
+
.to_socket_addrs()
|
|
134
|
+
.ok()?
|
|
135
|
+
.map(|addr| addr.ip())
|
|
136
|
+
.next()
|
|
137
|
+
}) // Fallback to IPv4
|
|
138
|
+
}
|
|
@@ -0,0 +1,32 @@
|
|
|
1
|
+
-----BEGIN CERTIFICATE-----
|
|
2
|
+
MIIFgTCCA2mgAwIBAgIUe316+G3qdkcWtRxYKlNBEAEenzAwDQYJKoZIhvcNAQEL
|
|
3
|
+
BQAwUDELMAkGA1UEBhMCVVMxDTALBgNVBAgMBEl0c2kxDTALBgNVBAcMBEl0c2kx
|
|
4
|
+
EDAOBgNVBAoMB0l0c2kgQ0ExETAPBgNVBAMMCGl0c2kuZnlpMB4XDTI1MDIyODAy
|
|
5
|
+
NTkyNVoXDTM1MDIyNjAyNTkyNVowUDELMAkGA1UEBhMCVVMxDTALBgNVBAgMBEl0
|
|
6
|
+
c2kxDTALBgNVBAcMBEl0c2kxEDAOBgNVBAoMB0l0c2kgQ0ExETAPBgNVBAMMCGl0
|
|
7
|
+
c2kuZnlpMIICIjANBgkqhkiG9w0BAQEFAAOCAg8AMIICCgKCAgEAzKAK2RIvnATa
|
|
8
|
+
pOxNYubaGjOx8z4thxwWM/KxMqI8gABmgRFTa99n1vrTenlfdX4PcoRyCXEJ3krh
|
|
9
|
+
nPsRmP1nnUS1BEIX4Ha9xVPIy0UoP0JN1a4kVInejZe2ZTcpRv8k2KlkafIw0ISt
|
|
10
|
+
/hSoopNV2SESCMDxx2e3F9d4wzHmkfs2+6k+12TTYkkifXaXgSOXm91a5ABqTfa8
|
|
11
|
+
wS6td/zmwv6W5RNBU3kS3TsqoKjc0xdu25aifJ40i/+82b5OOWhWt+psSPlTZb+Y
|
|
12
|
+
5elMhlh8Hjs52S3u48wz5joPsBi9r4yLYGhgB+AiW+b+y0uss7EiTQR24U+CroEB
|
|
13
|
+
4c1LA4qTgQPUvUIXyyFGC3lPsjKHGiEitbIT4sDQOz3nMMvT0FIP7DPtiSAuyx0I
|
|
14
|
+
J6/3+/QQ/8dzGqZskolKTSWGToOysNWcIxbbBprDAXYseOTJvREVpwC/Qra5OUqU
|
|
15
|
+
6P8K72yp2hSCu/5EQwV3kwKMw0JmjJFyaL8SC2GJnueWWCIIZjYdc0ZAA9Dvl9eo
|
|
16
|
+
SfQA4emLjUcScFpj8kv3Iu5tGJxnO1gqJ4JV+NKJ/09AxXvFYpNhaBXJ0xUOTA8/
|
|
17
|
+
vBTkAs9hTFV1IFIgJNP5CxEbkWr5FcFYUKAopFhDKfQkXGCKEn4s3wuNqUuzEEdE
|
|
18
|
+
Sx2YVV8XKbb9eKpHN5cQ6ljeFvanKpMCAwEAAaNTMFEwHQYDVR0OBBYEFIaKuU5Z
|
|
19
|
+
CCv9JAsdqvHyR5QnE1OdMB8GA1UdIwQYMBaAFIaKuU5ZCCv9JAsdqvHyR5QnE1Od
|
|
20
|
+
MA8GA1UdEwEB/wQFMAMBAf8wDQYJKoZIhvcNAQELBQADggIBAGslKyiOzrJzfVEA
|
|
21
|
+
HwJu/4CpDs7MpZuzAJ/AS0shWR3qlpHJUn6c/NIaQBfmVIo+vfxTvwV+ih6tZ+4A
|
|
22
|
+
HmIZrCQm7rn5uQm2Q9G++JQ0RgQfLngnz3FBL+aERW2wxw1vOcPwg8JWCeGUSCd4
|
|
23
|
+
Lj4jtpvlniS5uIiCD2GYxRIub401egX60XAiHdR0k6rS2dKxab9vMKW28RvRddak
|
|
24
|
+
YCk7UKSXOZlj39XVH/JCB8+4IokDLTikpoAUqiNfytO+kx9+1JHKti8NjSQJwjqO
|
|
25
|
+
JtBd1OD2ziQqd0gVKjr3J4IwIBv3Yl3GGUi3c5HVyDejriPciPrK+4G6a4IDGiyG
|
|
26
|
+
rQRTzTR98ILsPX5uvOJadF1TAyZpS8oN7xFM8BnKfdpB34x3p9KgBqk34dRIXxbF
|
|
27
|
+
nM3AJRT5dHlcHkn6z1snFqRHko4QvG0bSHlZFepogLG9yOGB0B1Hp4JPTSimEb6f
|
|
28
|
+
b+CL5o8mXzAMRDIzdjTkBM/nQg5NMoXaXvmKypw/zIYI/ffb6Kwu/Y6gWmXubQrA
|
|
29
|
+
fJ95Ssb14RKtmW6IDmHD5mNpybjhoSzwtBZyuHAyVZT/5P8/QHNkMBwpfQsevY5f
|
|
30
|
+
FmOcAZIq9bHTBvn5SNNvi2NxZfTRD7sTMogXgqKKcxwe2rs52IYecamNekQMjrPL
|
|
31
|
+
jI16bCLO/G1NaQ+N9YFL4TGfvbCe
|
|
32
|
+
-----END CERTIFICATE-----
|
|
@@ -0,0 +1,52 @@
|
|
|
1
|
+
-----BEGIN PRIVATE KEY-----
|
|
2
|
+
MIIJRAIBADANBgkqhkiG9w0BAQEFAASCCS4wggkqAgEAAoICAQDMoArZEi+cBNqk
|
|
3
|
+
7E1i5toaM7HzPi2HHBYz8rEyojyAAGaBEVNr32fW+tN6eV91fg9yhHIJcQneSuGc
|
|
4
|
+
+xGY/WedRLUEQhfgdr3FU8jLRSg/Qk3VriRUid6Nl7ZlNylG/yTYqWRp8jDQhK3+
|
|
5
|
+
FKiik1XZIRIIwPHHZ7cX13jDMeaR+zb7qT7XZNNiSSJ9dpeBI5eb3VrkAGpN9rzB
|
|
6
|
+
Lq13/ObC/pblE0FTeRLdOyqgqNzTF27blqJ8njSL/7zZvk45aFa36mxI+VNlv5jl
|
|
7
|
+
6UyGWHweOznZLe7jzDPmOg+wGL2vjItgaGAH4CJb5v7LS6yzsSJNBHbhT4KugQHh
|
|
8
|
+
zUsDipOBA9S9QhfLIUYLeU+yMocaISK1shPiwNA7Pecwy9PQUg/sM+2JIC7LHQgn
|
|
9
|
+
r/f79BD/x3MapmySiUpNJYZOg7Kw1ZwjFtsGmsMBdix45Mm9ERWnAL9Ctrk5SpTo
|
|
10
|
+
/wrvbKnaFIK7/kRDBXeTAozDQmaMkXJovxILYYme55ZYIghmNh1zRkAD0O+X16hJ
|
|
11
|
+
9ADh6YuNRxJwWmPyS/ci7m0YnGc7WConglX40on/T0DFe8Vik2FoFcnTFQ5MDz+8
|
|
12
|
+
FOQCz2FMVXUgUiAk0/kLERuRavkVwVhQoCikWEMp9CRcYIoSfizfC42pS7MQR0RL
|
|
13
|
+
HZhVXxcptv14qkc3lxDqWN4W9qcqkwIDAQABAoICADBfoVDhuLmUeC/G4SCBXIwX
|
|
14
|
+
LnlHeLHZFPKg6/0BV3YXIiRe+S6mOMEcuMPaT5PSAkrbPq42t9OCNkXLIMTfGxCV
|
|
15
|
+
volMKqLYz1IH1Y8gQTx7KzVZnqMRmLg2ZlsVKD/tb0N9AAz/wUR6KTvInHkahY/3
|
|
16
|
+
/nBtVHsEbMdJG/ZhJJXcIopp3z5CSqqQiYPJdlWEGYIyWRtPcdIOg17T7xRPiCoO
|
|
17
|
+
z5NF8wqNs8TzCMfEQ7fvcTieKrl1GQ0DnxyGna16mg2OcJzrvChwm++2MG4OGwF2
|
|
18
|
+
lN1fu3rEunwxu6Wwo58NhaF76z/RX64ENLXQpPox5N76MDRhfI5OVyiPRK2IpAfO
|
|
19
|
+
rgMY4tbiE+OMRqM553MpfcrmxmafxI7287OqraHWRaayHKARDrF42m5XXJG5sYUP
|
|
20
|
+
2RDKlt8LOOmhQWbnpzOifzyk8Rn57yIGPda2Wqc9vvH3efMcLJ/ZhDPeHBaA2JIN
|
|
21
|
+
7JWJNZ5G58muXBzh2+q4mBJTKN9RBHE+CHeNDR6Yp2/Tul0P7GesMMx78De4iNsu
|
|
22
|
+
uMiBhrx6FB5qsoBlFGD5FUyBVRYyIfZXbzGGh7dEHzOl1YVcZszjE/xm6QlLr0HD
|
|
23
|
+
pVH+Wf+y6rltz+Hw/XOJfvF1Wx42NmXX+FOGnVG0VPQB8s/oRnsUerYt9DrzV2BT
|
|
24
|
+
1oppvacvemTeryKx6LRBAoIBAQDlnMDBRoxvbJpzNcVQtZQr9GokIVockhKn2pHb
|
|
25
|
+
/2y4Y7Kl6q7LM8lEjSeCxvl2joEEDIvX6ex+mcB45D3CgKb1kCrjSsFgOgRMPjB1
|
|
26
|
+
yDDyQ9wf4UqJNKXriDy5v3HQVZKiFiDQPG1X4VEofIaNxMDSo2fx9v93bO3YeCJ9
|
|
27
|
+
2KTX5091Hx31UCZnBzE8siZtizz9iVqQ4qILbluxRQf8YMlLyT2KtCZzJ8wYfk9h
|
|
28
|
+
zHsr+v3wyrOUzbSxTOG9NfBuBUrz4gg3z08e+2ymfrl3lZdZ11iHkBMvfSIzZPUP
|
|
29
|
+
7TysZFR/DukW2Uh9szgSP8biaP8PDvqpZ+p493MwdEl2YCShAoIBAQDkJClmQ9b+
|
|
30
|
+
eRk/b2CFkpV1NNT9A5RaoI+aL4qfgOdDCBA/6lQy3P3aurIn0qAjNEsFl45mOBEy
|
|
31
|
+
GWCrU8HrccxhHVma48gxrTnQ0kJ7YEzgxTTN3ccqsly1e8mpNroWjZFqN85B9JwV
|
|
32
|
+
rzswu6pXD+673u7m/q8nIz9JSS2A0KKFr5BXVHiyxM9EgOW+v4Lx2r5f18iATTbs
|
|
33
|
+
qs8GSHMc1OPwFe/KKnrLPzpx1tGuBQkdXdv1WzSvfZdznPjMFNhapEWsvS+J5lKo
|
|
34
|
+
56t2EyxOR0d/TmyZVphw0qdD/eM57aWXxlJJoIs/NF+XG2pghK0YhoP2Y0bcCkOI
|
|
35
|
+
Hy1Hk6JbdM6zAoIBAQCiIkcF81AVGgYR1mVHMYC4bPVKH/bmd8sOlcsrIrjdlyC+
|
|
36
|
+
AfJ9cErtyhKdSO08ZzH47vcMdpTVbLI5a0mk/31lpvBx4QadcTo5sCw97yeI2pwk
|
|
37
|
+
MsyUCAYlQ+VFcEboypQpOiDfidvYEzVgtlW447cYxeQPOs93wAZPNb19Sa5U+nPk
|
|
38
|
+
Cx33bCpB1BVTe6Sg85IUoZm+9xlfowTCLzGNZ7acejSnrb/8zpxSq1ZYg7ByBOCt
|
|
39
|
+
2CRorbyq+dPo7J6iwcAEaJZO+mcvRHCbPJ6wL4RZHzPXPcgeX2j4C5D0NxwByzLT
|
|
40
|
+
KW80ACgtApFUaY6Br4xzUKt1Vfh+hJTlISgCm68hAoIBAQCEkGKh8q99hF5gVtZu
|
|
41
|
+
JwPDbCSKtEbC9mMbA574Gc3HTGssyHuOZoz3SN52d1PnwN4K7MqoqNGNG+PpCa03
|
|
42
|
+
oxNQJt7HOq47910N8u8Ag6+IN+775G9zZtqp9bjzI0K0EiS55J1sA7eifgTVx2Yu
|
|
43
|
+
Wqqs7dhBzyF1i2ydp/DR1elp5t7nb8UGk4egVYmp8dwjrqhKRrdRngxZLtNG4lhJ
|
|
44
|
+
G4crHYQNI+vgJ+hM97c09+YY8035XrrZcg/L9R04cLBa0vNNcUyrQ3MqhBtEa9Wl
|
|
45
|
+
0pM/7RD7dK71d+ILhv4+zdEXxPxRngDrhAI7aonAdbei9Z6+T2eAOlKNSRhla+q+
|
|
46
|
+
W7wzAoIBAQDGX2nnfmdzDdDyFSJj5mIfj/nMG1FQg4BlxtFn6BRJmAVxCasi8tP9
|
|
47
|
+
LLlMy4n6jYNxUUad5jwmFTcH3WBIaNtKyMhvzpY7lb6zp2i8U6GS6ctbzrWCobM5
|
|
48
|
+
ivhuz0WuayZzI9fQGBb+EjAEhDgVRAFPG+xRowxC+vUw30kTAW15ENUN3HRRBLPo
|
|
49
|
+
b2VWCatESqGIuuweu44aZcj0c7QLOQiDry4QqrpIuIR2HIzOhX8Uuxq1kdSPuXBh
|
|
50
|
+
3YQBLl7YtOi/UNpRqlYrJpVlLoDZol4HgsQBqd5dqYUYCbl5fpgxM25vww6LFFjK
|
|
51
|
+
cN5973/E4MpAKMt1shs0YWr3axraOhkg
|
|
52
|
+
-----END PRIVATE KEY-----
|
|
@@ -0,0 +1,182 @@
|
|
|
1
|
+
use super::{
|
|
2
|
+
bind::Bind,
|
|
3
|
+
listener::{Listener, SockAddr},
|
|
4
|
+
};
|
|
5
|
+
use crate::{request::itsi_request::ItsiRequest, ITSI_SERVER};
|
|
6
|
+
use bytes::Bytes;
|
|
7
|
+
use derive_more::Debug;
|
|
8
|
+
use http_body_util::{combinators::BoxBody, Empty};
|
|
9
|
+
use hyper::{
|
|
10
|
+
body::Incoming, header::HeaderName, service::service_fn, HeaderMap, Request, Response,
|
|
11
|
+
StatusCode,
|
|
12
|
+
};
|
|
13
|
+
use hyper_util::{rt::TokioExecutor, server::conn::auto::Builder};
|
|
14
|
+
use itsi_tracing::{error, info};
|
|
15
|
+
use magnus::{
|
|
16
|
+
error::Result,
|
|
17
|
+
scan_args::{get_kwargs, scan_args, Args, KwArgs},
|
|
18
|
+
value::{Opaque, ReprValue},
|
|
19
|
+
RHash, Ruby, Value,
|
|
20
|
+
};
|
|
21
|
+
use parking_lot::Mutex;
|
|
22
|
+
use std::{collections::HashMap, convert::Infallible, sync::Arc};
|
|
23
|
+
use tokio::runtime::Builder as RuntimeBuilder;
|
|
24
|
+
use tokio::task::JoinSet;
|
|
25
|
+
|
|
26
|
+
#[magnus::wrap(class = "Itsi::Server", free_immediately, size)]
|
|
27
|
+
#[derive(Debug)]
|
|
28
|
+
pub struct Server {
|
|
29
|
+
#[debug(skip)]
|
|
30
|
+
app: Opaque<Value>,
|
|
31
|
+
#[allow(unused)]
|
|
32
|
+
workers: u16,
|
|
33
|
+
#[allow(unused)]
|
|
34
|
+
threads: u16,
|
|
35
|
+
#[allow(unused)]
|
|
36
|
+
shutdown_timeout: f64,
|
|
37
|
+
script_name: String,
|
|
38
|
+
pub(crate) binds: Mutex<Vec<Bind>>,
|
|
39
|
+
}
|
|
40
|
+
|
|
41
|
+
impl Server {
|
|
42
|
+
pub fn new(args: &[Value]) -> Result<Self> {
|
|
43
|
+
type OptionalArgs = (
|
|
44
|
+
Option<u16>,
|
|
45
|
+
Option<u16>,
|
|
46
|
+
Option<f64>,
|
|
47
|
+
Option<String>,
|
|
48
|
+
Option<Vec<String>>,
|
|
49
|
+
);
|
|
50
|
+
|
|
51
|
+
let scan_args: Args<(), (), (), (), RHash, ()> = scan_args(args)?;
|
|
52
|
+
let args: KwArgs<(Value,), OptionalArgs, ()> = get_kwargs(
|
|
53
|
+
scan_args.keywords,
|
|
54
|
+
&["app"],
|
|
55
|
+
&[
|
|
56
|
+
"workers",
|
|
57
|
+
"threads",
|
|
58
|
+
"shutdown_timeout",
|
|
59
|
+
"script_name",
|
|
60
|
+
"binds",
|
|
61
|
+
],
|
|
62
|
+
)?;
|
|
63
|
+
let server = Server {
|
|
64
|
+
app: Opaque::from(args.required.0),
|
|
65
|
+
workers: args.optional.0.unwrap_or(1),
|
|
66
|
+
threads: args.optional.1.unwrap_or(1),
|
|
67
|
+
shutdown_timeout: args.optional.2.unwrap_or(5.0),
|
|
68
|
+
script_name: args.optional.3.unwrap_or("".to_string()),
|
|
69
|
+
binds: Mutex::new(
|
|
70
|
+
args.optional
|
|
71
|
+
.4
|
|
72
|
+
.unwrap_or_else(|| vec!["localhost:3000".to_string()])
|
|
73
|
+
.into_iter()
|
|
74
|
+
.map(|s| s.parse().unwrap_or_else(|_| Bind::default()))
|
|
75
|
+
.collect(),
|
|
76
|
+
),
|
|
77
|
+
};
|
|
78
|
+
Ok(server)
|
|
79
|
+
}
|
|
80
|
+
|
|
81
|
+
pub(crate) async fn process_request(
|
|
82
|
+
hyper_request: Request<Incoming>,
|
|
83
|
+
app: Opaque<Value>,
|
|
84
|
+
script_name: String,
|
|
85
|
+
listener: Arc<Listener>,
|
|
86
|
+
addr: SockAddr,
|
|
87
|
+
) -> itsi_error::Result<Response<BoxBody<Bytes, Infallible>>> {
|
|
88
|
+
let request = ItsiRequest::build_from(hyper_request, addr, script_name, listener).await;
|
|
89
|
+
let ruby = Ruby::get().unwrap();
|
|
90
|
+
let server = ruby.get_inner(&ITSI_SERVER);
|
|
91
|
+
let response: Result<(u16, HashMap<String, String>, Value)> =
|
|
92
|
+
server.funcall("call", (app, request));
|
|
93
|
+
if let Ok((status, headers_raw, body)) = response {
|
|
94
|
+
let mut body_buf = vec![];
|
|
95
|
+
for body_chunk in body.enumeratorize("each", ()) {
|
|
96
|
+
body_buf.push(body_chunk.unwrap().to_string())
|
|
97
|
+
}
|
|
98
|
+
body.check_funcall::<_, _, Value>("close", ());
|
|
99
|
+
let boxed_body = BoxBody::new(body_buf.join(""));
|
|
100
|
+
let mut response = Response::new(boxed_body);
|
|
101
|
+
let mut headers = HeaderMap::new();
|
|
102
|
+
headers_raw.into_iter().for_each(|(key, value)| {
|
|
103
|
+
let header_name: HeaderName = key.parse().unwrap();
|
|
104
|
+
headers.insert(header_name, value.parse().unwrap());
|
|
105
|
+
});
|
|
106
|
+
*response.headers_mut() = headers;
|
|
107
|
+
*response.status_mut() = StatusCode::from_u16(status).unwrap();
|
|
108
|
+
Ok(response)
|
|
109
|
+
} else {
|
|
110
|
+
let mut response = Response::new(BoxBody::new(Empty::new()));
|
|
111
|
+
*response.status_mut() = StatusCode::BAD_REQUEST;
|
|
112
|
+
Ok(response)
|
|
113
|
+
}
|
|
114
|
+
}
|
|
115
|
+
|
|
116
|
+
pub fn start(&self) {
|
|
117
|
+
let mut builder: RuntimeBuilder = RuntimeBuilder::new_current_thread();
|
|
118
|
+
let runtime = builder
|
|
119
|
+
.thread_name("itsi-server-accept-loop")
|
|
120
|
+
.thread_stack_size(3 * 1024 * 1024)
|
|
121
|
+
.enable_io()
|
|
122
|
+
.enable_time()
|
|
123
|
+
.build()
|
|
124
|
+
.expect("Failed to build Tokio runtime");
|
|
125
|
+
|
|
126
|
+
runtime.block_on(async {
|
|
127
|
+
let server = Arc::new(Builder::new(TokioExecutor::new()));
|
|
128
|
+
let listeners: Vec<Listener> = self
|
|
129
|
+
.binds
|
|
130
|
+
.lock()
|
|
131
|
+
.iter()
|
|
132
|
+
.cloned()
|
|
133
|
+
.map(Listener::from)
|
|
134
|
+
.collect::<Vec<_>>();
|
|
135
|
+
|
|
136
|
+
let mut set = JoinSet::new();
|
|
137
|
+
|
|
138
|
+
for listener in listeners {
|
|
139
|
+
let app = self.app;
|
|
140
|
+
let server_clone = server.clone();
|
|
141
|
+
let listener_clone = Arc::new(listener);
|
|
142
|
+
let script_name = self.script_name.clone();
|
|
143
|
+
|
|
144
|
+
set.spawn(async move {
|
|
145
|
+
loop {
|
|
146
|
+
let server = server_clone.clone();
|
|
147
|
+
let listener = listener_clone.clone();
|
|
148
|
+
let script_name = script_name.clone();
|
|
149
|
+
let (stream, addr) = match listener.accept().await {
|
|
150
|
+
Ok(stream) => stream,
|
|
151
|
+
Err(e) => {
|
|
152
|
+
error!("Failed to accept connection: {:?}", e);
|
|
153
|
+
continue;
|
|
154
|
+
}
|
|
155
|
+
};
|
|
156
|
+
|
|
157
|
+
tokio::spawn(async move {
|
|
158
|
+
if let Err(e) = server
|
|
159
|
+
.serve_connection_with_upgrades(
|
|
160
|
+
stream,
|
|
161
|
+
service_fn(move |hyper_request: Request<Incoming>| {
|
|
162
|
+
Server::process_request(
|
|
163
|
+
hyper_request,
|
|
164
|
+
app,
|
|
165
|
+
script_name.clone(),
|
|
166
|
+
listener.clone(),
|
|
167
|
+
addr.clone(),
|
|
168
|
+
)
|
|
169
|
+
}),
|
|
170
|
+
)
|
|
171
|
+
.await
|
|
172
|
+
{
|
|
173
|
+
info!("Closed connection due to: {:?}", e);
|
|
174
|
+
}
|
|
175
|
+
});
|
|
176
|
+
}
|
|
177
|
+
});
|
|
178
|
+
}
|
|
179
|
+
while let Some(_res) = set.join_next().await {}
|
|
180
|
+
})
|
|
181
|
+
}
|
|
182
|
+
}
|