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.
Files changed (103) hide show
  1. checksums.yaml +7 -0
  2. data/Cargo.lock +1541 -0
  3. data/Cargo.toml +13 -0
  4. data/README.md +82 -0
  5. data/Rakefile +87 -0
  6. data/crates/itsi_error/Cargo.lock +368 -0
  7. data/crates/itsi_error/Cargo.toml +9 -0
  8. data/crates/itsi_error/src/lib.rs +49 -0
  9. data/crates/itsi_rb_helpers/Cargo.lock +355 -0
  10. data/crates/itsi_rb_helpers/Cargo.toml +8 -0
  11. data/crates/itsi_rb_helpers/src/lib.rs +98 -0
  12. data/crates/itsi_scheduler/Cargo.toml +16 -0
  13. data/crates/itsi_scheduler/extconf.rb +6 -0
  14. data/crates/itsi_scheduler/src/lib.rs +17 -0
  15. data/crates/itsi_server/Cargo.toml +29 -0
  16. data/crates/itsi_server/extconf.rb +6 -0
  17. data/crates/itsi_server/src/lib.rs +52 -0
  18. data/crates/itsi_server/src/request/itsi_request.rs +143 -0
  19. data/crates/itsi_server/src/request/mod.rs +1 -0
  20. data/crates/itsi_server/src/server/bind.rs +138 -0
  21. data/crates/itsi_server/src/server/itsi_ca/itsi_ca.crt +32 -0
  22. data/crates/itsi_server/src/server/itsi_ca/itsi_ca.key +52 -0
  23. data/crates/itsi_server/src/server/itsi_server.rs +182 -0
  24. data/crates/itsi_server/src/server/listener.rs +218 -0
  25. data/crates/itsi_server/src/server/mod.rs +5 -0
  26. data/crates/itsi_server/src/server/tls.rs +138 -0
  27. data/crates/itsi_server/src/server/transfer_protocol.rs +23 -0
  28. data/crates/itsi_server/src/stream_writer/mod.rs +21 -0
  29. data/crates/itsi_tracing/Cargo.lock +274 -0
  30. data/crates/itsi_tracing/Cargo.toml +12 -0
  31. data/crates/itsi_tracing/src/lib.rs +11 -0
  32. data/gems/scheduler/.gitignore +14 -0
  33. data/gems/scheduler/.rubocop.yml +8 -0
  34. data/gems/scheduler/CHANGELOG.md +5 -0
  35. data/gems/scheduler/CODE_OF_CONDUCT.md +132 -0
  36. data/gems/scheduler/Cargo.lock +745 -0
  37. data/gems/scheduler/Cargo.toml +3 -0
  38. data/gems/scheduler/LICENSE.txt +21 -0
  39. data/gems/scheduler/README.md +43 -0
  40. data/gems/scheduler/Rakefile +22 -0
  41. data/gems/scheduler/bin/console +11 -0
  42. data/gems/scheduler/bin/setup +8 -0
  43. data/gems/scheduler/ext/itsi_error/Cargo.lock +368 -0
  44. data/gems/scheduler/ext/itsi_error/Cargo.toml +9 -0
  45. data/gems/scheduler/ext/itsi_error/src/lib.rs +49 -0
  46. data/gems/scheduler/ext/itsi_rb_helpers/Cargo.lock +355 -0
  47. data/gems/scheduler/ext/itsi_rb_helpers/Cargo.toml +8 -0
  48. data/gems/scheduler/ext/itsi_rb_helpers/src/lib.rs +98 -0
  49. data/gems/scheduler/ext/itsi_scheduler/Cargo.toml +16 -0
  50. data/gems/scheduler/ext/itsi_scheduler/extconf.rb +6 -0
  51. data/gems/scheduler/ext/itsi_scheduler/src/lib.rs +17 -0
  52. data/gems/scheduler/ext/itsi_tracing/Cargo.lock +274 -0
  53. data/gems/scheduler/ext/itsi_tracing/Cargo.toml +12 -0
  54. data/gems/scheduler/ext/itsi_tracing/src/lib.rs +11 -0
  55. data/gems/scheduler/itsi-scheduler.gemspec +43 -0
  56. data/gems/scheduler/lib/itsi/scheduler/version.rb +7 -0
  57. data/gems/scheduler/lib/itsi/scheduler.rb +11 -0
  58. data/gems/scheduler/sig/itsi_scheduler.rbs +4 -0
  59. data/gems/scheduler/test/test_helper.rb +6 -0
  60. data/gems/scheduler/test/test_itsi_scheduler.rb +9 -0
  61. data/gems/server/.gitignore +14 -0
  62. data/gems/server/.rubocop.yml +8 -0
  63. data/gems/server/CHANGELOG.md +5 -0
  64. data/gems/server/CODE_OF_CONDUCT.md +132 -0
  65. data/gems/server/LICENSE.txt +21 -0
  66. data/gems/server/README.md +43 -0
  67. data/gems/server/Rakefile +22 -0
  68. data/gems/server/bin/console +11 -0
  69. data/gems/server/bin/setup +8 -0
  70. data/gems/server/exe/itsi +84 -0
  71. data/gems/server/ext/itsi_error/Cargo.lock +368 -0
  72. data/gems/server/ext/itsi_error/Cargo.toml +9 -0
  73. data/gems/server/ext/itsi_error/src/lib.rs +49 -0
  74. data/gems/server/ext/itsi_rb_helpers/Cargo.lock +355 -0
  75. data/gems/server/ext/itsi_rb_helpers/Cargo.toml +8 -0
  76. data/gems/server/ext/itsi_rb_helpers/src/lib.rs +98 -0
  77. data/gems/server/ext/itsi_server/Cargo.toml +29 -0
  78. data/gems/server/ext/itsi_server/extconf.rb +6 -0
  79. data/gems/server/ext/itsi_server/src/lib.rs +52 -0
  80. data/gems/server/ext/itsi_server/src/request/itsi_request.rs +143 -0
  81. data/gems/server/ext/itsi_server/src/request/mod.rs +1 -0
  82. data/gems/server/ext/itsi_server/src/server/bind.rs +138 -0
  83. data/gems/server/ext/itsi_server/src/server/itsi_ca/itsi_ca.crt +32 -0
  84. data/gems/server/ext/itsi_server/src/server/itsi_ca/itsi_ca.key +52 -0
  85. data/gems/server/ext/itsi_server/src/server/itsi_server.rs +182 -0
  86. data/gems/server/ext/itsi_server/src/server/listener.rs +218 -0
  87. data/gems/server/ext/itsi_server/src/server/mod.rs +5 -0
  88. data/gems/server/ext/itsi_server/src/server/tls.rs +138 -0
  89. data/gems/server/ext/itsi_server/src/server/transfer_protocol.rs +23 -0
  90. data/gems/server/ext/itsi_server/src/stream_writer/mod.rs +21 -0
  91. data/gems/server/ext/itsi_tracing/Cargo.lock +274 -0
  92. data/gems/server/ext/itsi_tracing/Cargo.toml +12 -0
  93. data/gems/server/ext/itsi_tracing/src/lib.rs +11 -0
  94. data/gems/server/itsi-server.gemspec +45 -0
  95. data/gems/server/lib/itsi/request.rb +39 -0
  96. data/gems/server/lib/itsi/server/version.rb +7 -0
  97. data/gems/server/lib/itsi/server.rb +21 -0
  98. data/gems/server/sig/itsi_server.rbs +4 -0
  99. data/gems/server/test/test_helper.rb +5 -0
  100. data/gems/server/test/test_itsi_server.rb +9 -0
  101. data/lib/itsi/version.rb +3 -0
  102. data/lib/itsi.rb +6 -0
  103. 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
+ }