itsi-server 0.1.1 → 0.1.13
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.
Potentially problematic release.
This version of itsi-server might be problematic. Click here for more details.
- checksums.yaml +4 -4
- data/CHANGELOG.md +5 -0
- data/CODE_OF_CONDUCT.md +7 -0
- data/Cargo.lock +4417 -0
- data/Cargo.toml +7 -0
- data/README.md +4 -0
- data/Rakefile +8 -1
- data/_index.md +6 -0
- data/exe/itsi +94 -45
- data/ext/itsi_error/Cargo.toml +2 -0
- data/ext/itsi_error/src/from.rs +68 -0
- data/ext/itsi_error/src/lib.rs +18 -34
- data/ext/itsi_error/target/debug/build/clang-sys-da71b0344e568175/out/common.rs +355 -0
- data/ext/itsi_error/target/debug/build/clang-sys-da71b0344e568175/out/dynamic.rs +276 -0
- data/ext/itsi_error/target/debug/build/clang-sys-da71b0344e568175/out/macros.rs +49 -0
- data/ext/itsi_error/target/debug/build/rb-sys-49f554618693db24/out/bindings-0.9.110-mri-arm64-darwin23-3.4.2.rs +8865 -0
- data/ext/itsi_error/target/debug/incremental/itsi_error-1mmt5sux7jb0i/s-h510z7m8v9-0bxu7yd.lock +0 -0
- data/ext/itsi_error/target/debug/incremental/itsi_error-2vn3jey74oiw0/s-h5113n0e7e-1v5qzs6.lock +0 -0
- data/ext/itsi_error/target/debug/incremental/itsi_error-37uv9dicz7awp/s-h510ykifhe-0tbnep2.lock +0 -0
- data/ext/itsi_error/target/debug/incremental/itsi_error-37uv9dicz7awp/s-h510yyocpj-0tz7ug7.lock +0 -0
- data/ext/itsi_error/target/debug/incremental/itsi_error-37uv9dicz7awp/s-h510z0xc8g-14ol18k.lock +0 -0
- data/ext/itsi_error/target/debug/incremental/itsi_error-3g5qf4y7d54uj/s-h5113n0e7d-1trk8on.lock +0 -0
- data/ext/itsi_error/target/debug/incremental/itsi_error-3lpfftm45d3e2/s-h510z7m8r3-1pxp20o.lock +0 -0
- data/ext/itsi_error/target/debug/incremental/itsi_error-3o4qownhl3d7n/s-h510ykifek-1uxasnk.lock +0 -0
- data/ext/itsi_error/target/debug/incremental/itsi_error-3o4qownhl3d7n/s-h510yyocki-11u37qm.lock +0 -0
- data/ext/itsi_error/target/debug/incremental/itsi_error-3o4qownhl3d7n/s-h510z0xc93-0pmy0zm.lock +0 -0
- data/ext/itsi_instrument_entry/Cargo.toml +15 -0
- data/ext/itsi_instrument_entry/src/lib.rs +31 -0
- data/ext/itsi_rb_helpers/Cargo.toml +3 -0
- data/ext/itsi_rb_helpers/src/heap_value.rs +139 -0
- data/ext/itsi_rb_helpers/src/lib.rs +140 -10
- data/ext/itsi_rb_helpers/target/debug/build/clang-sys-da71b0344e568175/out/common.rs +355 -0
- data/ext/itsi_rb_helpers/target/debug/build/clang-sys-da71b0344e568175/out/dynamic.rs +276 -0
- data/ext/itsi_rb_helpers/target/debug/build/clang-sys-da71b0344e568175/out/macros.rs +49 -0
- data/ext/itsi_rb_helpers/target/debug/build/rb-sys-eb9ed4ff3a60f995/out/bindings-0.9.110-mri-arm64-darwin23-3.4.2.rs +8865 -0
- data/ext/itsi_rb_helpers/target/debug/incremental/itsi_rb_helpers-040pxg6yhb3g3/s-h5113n7a1b-03bwlt4.lock +0 -0
- data/ext/itsi_rb_helpers/target/debug/incremental/itsi_rb_helpers-131g1u4dzkt1a/s-h51113xnh3-1eik1ip.lock +0 -0
- data/ext/itsi_rb_helpers/target/debug/incremental/itsi_rb_helpers-131g1u4dzkt1a/s-h5111704jj-0g4rj8x.lock +0 -0
- data/ext/itsi_rb_helpers/target/debug/incremental/itsi_rb_helpers-1q2d3drtxrzs5/s-h5113n79yl-0bxcqc5.lock +0 -0
- data/ext/itsi_rb_helpers/target/debug/incremental/itsi_rb_helpers-374a9h7ovycj0/s-h51113xoox-10de2hp.lock +0 -0
- data/ext/itsi_rb_helpers/target/debug/incremental/itsi_rb_helpers-374a9h7ovycj0/s-h5111704w7-0vdq7gq.lock +0 -0
- data/ext/itsi_scheduler/Cargo.toml +24 -0
- data/ext/itsi_scheduler/src/itsi_scheduler/io_helpers.rs +56 -0
- data/ext/itsi_scheduler/src/itsi_scheduler/io_waiter.rs +44 -0
- data/ext/itsi_scheduler/src/itsi_scheduler/timer.rs +44 -0
- data/ext/itsi_scheduler/src/itsi_scheduler.rs +308 -0
- data/ext/itsi_scheduler/src/lib.rs +38 -0
- data/ext/itsi_server/Cargo.lock +2956 -0
- data/ext/itsi_server/Cargo.toml +73 -13
- data/ext/itsi_server/extconf.rb +1 -1
- data/ext/itsi_server/src/env.rs +43 -0
- data/ext/itsi_server/src/lib.rs +100 -40
- data/ext/itsi_server/src/ruby_types/itsi_body_proxy/big_bytes.rs +109 -0
- data/ext/itsi_server/src/ruby_types/itsi_body_proxy/mod.rs +141 -0
- data/ext/itsi_server/src/ruby_types/itsi_grpc_request.rs +147 -0
- data/ext/itsi_server/src/ruby_types/itsi_grpc_response.rs +19 -0
- data/ext/itsi_server/src/ruby_types/itsi_grpc_stream/mod.rs +216 -0
- data/ext/itsi_server/src/ruby_types/itsi_http_request.rs +282 -0
- data/ext/itsi_server/src/ruby_types/itsi_http_response.rs +388 -0
- data/ext/itsi_server/src/ruby_types/itsi_server/file_watcher.rs +225 -0
- data/ext/itsi_server/src/ruby_types/itsi_server/itsi_server_config.rs +355 -0
- data/ext/itsi_server/src/ruby_types/itsi_server.rs +82 -0
- data/ext/itsi_server/src/ruby_types/mod.rs +55 -0
- data/ext/itsi_server/src/server/bind.rs +75 -31
- data/ext/itsi_server/src/server/bind_protocol.rs +37 -0
- data/ext/itsi_server/src/server/byte_frame.rs +32 -0
- data/ext/itsi_server/src/server/cache_store.rs +74 -0
- data/ext/itsi_server/src/server/io_stream.rs +104 -0
- data/ext/itsi_server/src/server/itsi_service.rs +172 -0
- data/ext/itsi_server/src/server/lifecycle_event.rs +12 -0
- data/ext/itsi_server/src/server/listener.rs +332 -132
- data/ext/itsi_server/src/server/middleware_stack/middleware.rs +153 -0
- data/ext/itsi_server/src/server/middleware_stack/middlewares/allow_list.rs +47 -0
- data/ext/itsi_server/src/server/middleware_stack/middlewares/auth_api_key.rs +58 -0
- data/ext/itsi_server/src/server/middleware_stack/middlewares/auth_basic.rs +82 -0
- data/ext/itsi_server/src/server/middleware_stack/middlewares/auth_jwt.rs +321 -0
- data/ext/itsi_server/src/server/middleware_stack/middlewares/cache_control.rs +139 -0
- data/ext/itsi_server/src/server/middleware_stack/middlewares/compression.rs +300 -0
- data/ext/itsi_server/src/server/middleware_stack/middlewares/cors.rs +287 -0
- data/ext/itsi_server/src/server/middleware_stack/middlewares/deny_list.rs +48 -0
- data/ext/itsi_server/src/server/middleware_stack/middlewares/error_response.rs +127 -0
- data/ext/itsi_server/src/server/middleware_stack/middlewares/etag.rs +191 -0
- data/ext/itsi_server/src/server/middleware_stack/middlewares/grpc_service.rs +72 -0
- data/ext/itsi_server/src/server/middleware_stack/middlewares/header_interpretation.rs +85 -0
- data/ext/itsi_server/src/server/middleware_stack/middlewares/intrusion_protection.rs +195 -0
- data/ext/itsi_server/src/server/middleware_stack/middlewares/log_requests.rs +82 -0
- data/ext/itsi_server/src/server/middleware_stack/middlewares/mod.rs +82 -0
- data/ext/itsi_server/src/server/middleware_stack/middlewares/proxy.rs +216 -0
- data/ext/itsi_server/src/server/middleware_stack/middlewares/rate_limit.rs +124 -0
- data/ext/itsi_server/src/server/middleware_stack/middlewares/redirect.rs +76 -0
- data/ext/itsi_server/src/server/middleware_stack/middlewares/request_headers.rs +43 -0
- data/ext/itsi_server/src/server/middleware_stack/middlewares/response_headers.rs +34 -0
- data/ext/itsi_server/src/server/middleware_stack/middlewares/ruby_app.rs +93 -0
- data/ext/itsi_server/src/server/middleware_stack/middlewares/static_assets.rs +162 -0
- data/ext/itsi_server/src/server/middleware_stack/middlewares/string_rewrite.rs +158 -0
- data/ext/itsi_server/src/server/middleware_stack/middlewares/token_source.rs +12 -0
- data/ext/itsi_server/src/server/middleware_stack/mod.rs +315 -0
- data/ext/itsi_server/src/server/mod.rs +15 -2
- data/ext/itsi_server/src/server/process_worker.rs +229 -0
- data/ext/itsi_server/src/server/rate_limiter.rs +565 -0
- data/ext/itsi_server/src/server/request_job.rs +11 -0
- data/ext/itsi_server/src/server/serve_strategy/cluster_mode.rs +337 -0
- data/ext/itsi_server/src/server/serve_strategy/mod.rs +30 -0
- data/ext/itsi_server/src/server/serve_strategy/single_mode.rs +421 -0
- data/ext/itsi_server/src/server/signal.rs +93 -0
- data/ext/itsi_server/src/server/static_file_server.rs +984 -0
- data/ext/itsi_server/src/server/thread_worker.rs +444 -0
- data/ext/itsi_server/src/server/tls/locked_dir_cache.rs +132 -0
- data/ext/itsi_server/src/server/tls.rs +187 -60
- data/ext/itsi_server/src/server/types.rs +43 -0
- data/ext/itsi_tracing/Cargo.toml +5 -0
- data/ext/itsi_tracing/src/lib.rs +225 -7
- data/ext/itsi_tracing/target/debug/incremental/itsi_tracing-0994n8rpvvt9m/s-h510hfz1f6-1kbycmq.lock +0 -0
- data/ext/itsi_tracing/target/debug/incremental/itsi_tracing-0bob7bf4yq34i/s-h5113125h5-0lh4rag.lock +0 -0
- data/ext/itsi_tracing/target/debug/incremental/itsi_tracing-2fcodulrxbbxo/s-h510h2infk-0hp5kjw.lock +0 -0
- data/ext/itsi_tracing/target/debug/incremental/itsi_tracing-2iak63r1woi1l/s-h510h2in4q-0kxfzw1.lock +0 -0
- data/ext/itsi_tracing/target/debug/incremental/itsi_tracing-2kk4qj9gn5dg2/s-h5113124kv-0enwon2.lock +0 -0
- data/ext/itsi_tracing/target/debug/incremental/itsi_tracing-2mwo0yas7dtw4/s-h510hfz1ha-1udgpei.lock +0 -0
- data/lib/itsi/http_request.rb +87 -0
- data/lib/itsi/http_response.rb +39 -0
- data/lib/itsi/server/Itsi.rb +119 -0
- data/lib/itsi/server/config/dsl.rb +506 -0
- data/lib/itsi/server/config.rb +131 -0
- data/lib/itsi/server/default_app/default_app.rb +38 -0
- data/lib/itsi/server/default_app/index.html +91 -0
- data/lib/itsi/server/grpc_interface.rb +213 -0
- data/lib/itsi/server/rack/handler/itsi.rb +27 -0
- data/lib/itsi/server/rack_interface.rb +94 -0
- data/lib/itsi/server/scheduler_interface.rb +21 -0
- data/lib/itsi/server/scheduler_mode.rb +10 -0
- data/lib/itsi/server/signal_trap.rb +29 -0
- data/lib/itsi/server/version.rb +1 -1
- data/lib/itsi/server.rb +90 -9
- data/lib/itsi/standard_headers.rb +86 -0
- metadata +122 -31
- data/ext/itsi_server/src/request/itsi_request.rs +0 -143
- data/ext/itsi_server/src/request/mod.rs +0 -1
- data/ext/itsi_server/src/server/itsi_ca/itsi_ca.crt +0 -32
- data/ext/itsi_server/src/server/itsi_ca/itsi_ca.key +0 -52
- data/ext/itsi_server/src/server/itsi_server.rs +0 -182
- data/ext/itsi_server/src/server/transfer_protocol.rs +0 -23
- data/ext/itsi_server/src/stream_writer/mod.rs +0 -21
- data/lib/itsi/request.rb +0 -39
@@ -0,0 +1,153 @@
|
|
1
|
+
use super::middlewares::*;
|
2
|
+
use crate::server::{
|
3
|
+
itsi_service::RequestContext,
|
4
|
+
types::{HttpRequest, HttpResponse},
|
5
|
+
};
|
6
|
+
use async_trait::async_trait;
|
7
|
+
use either::Either;
|
8
|
+
use magnus::error::Result;
|
9
|
+
use std::cmp::Ordering;
|
10
|
+
|
11
|
+
#[derive(Debug)]
|
12
|
+
pub enum Middleware {
|
13
|
+
AllowList(AllowList),
|
14
|
+
AuthAPIKey(AuthAPIKey),
|
15
|
+
AuthBasic(AuthBasic),
|
16
|
+
AuthJwt(Box<AuthJwt>),
|
17
|
+
CacheControl(CacheControl),
|
18
|
+
Compression(Compression),
|
19
|
+
Cors(Box<Cors>),
|
20
|
+
DenyList(DenyList),
|
21
|
+
ETag(ETag),
|
22
|
+
IntrusionProtection(IntrusionProtection),
|
23
|
+
LogRequests(LogRequests),
|
24
|
+
Proxy(Proxy),
|
25
|
+
RateLimit(RateLimit),
|
26
|
+
Redirect(Redirect),
|
27
|
+
RequestHeaders(RequestHeaders),
|
28
|
+
ResponseHeaders(ResponseHeaders),
|
29
|
+
RubyApp(RubyApp),
|
30
|
+
StaticAssets(StaticAssets),
|
31
|
+
}
|
32
|
+
|
33
|
+
#[async_trait]
|
34
|
+
impl MiddlewareLayer for Middleware {
|
35
|
+
/// Called just once, to initialize the middleware state.
|
36
|
+
async fn initialize(&self) -> Result<()> {
|
37
|
+
match self {
|
38
|
+
Middleware::DenyList(filter) => filter.initialize().await,
|
39
|
+
Middleware::AllowList(filter) => filter.initialize().await,
|
40
|
+
Middleware::AuthBasic(filter) => filter.initialize().await,
|
41
|
+
Middleware::AuthJwt(filter) => filter.initialize().await,
|
42
|
+
Middleware::AuthAPIKey(filter) => filter.initialize().await,
|
43
|
+
Middleware::IntrusionProtection(filter) => filter.initialize().await,
|
44
|
+
Middleware::RateLimit(filter) => filter.initialize().await,
|
45
|
+
Middleware::RequestHeaders(filter) => filter.initialize().await,
|
46
|
+
Middleware::ResponseHeaders(filter) => filter.initialize().await,
|
47
|
+
Middleware::CacheControl(filter) => filter.initialize().await,
|
48
|
+
Middleware::Cors(filter) => filter.initialize().await,
|
49
|
+
Middleware::ETag(filter) => filter.initialize().await,
|
50
|
+
Middleware::StaticAssets(filter) => filter.initialize().await,
|
51
|
+
Middleware::Compression(filter) => filter.initialize().await,
|
52
|
+
Middleware::LogRequests(filter) => filter.initialize().await,
|
53
|
+
Middleware::Redirect(filter) => filter.initialize().await,
|
54
|
+
Middleware::Proxy(filter) => filter.initialize().await,
|
55
|
+
Middleware::RubyApp(filter) => filter.initialize().await,
|
56
|
+
}
|
57
|
+
}
|
58
|
+
|
59
|
+
async fn before(
|
60
|
+
&self,
|
61
|
+
req: HttpRequest,
|
62
|
+
context: &mut RequestContext,
|
63
|
+
) -> Result<Either<HttpRequest, HttpResponse>> {
|
64
|
+
match self {
|
65
|
+
Middleware::DenyList(filter) => filter.before(req, context).await,
|
66
|
+
Middleware::AllowList(filter) => filter.before(req, context).await,
|
67
|
+
Middleware::AuthBasic(filter) => filter.before(req, context).await,
|
68
|
+
Middleware::AuthJwt(filter) => filter.before(req, context).await,
|
69
|
+
Middleware::AuthAPIKey(filter) => filter.before(req, context).await,
|
70
|
+
Middleware::IntrusionProtection(filter) => filter.before(req, context).await,
|
71
|
+
Middleware::RequestHeaders(filter) => filter.before(req, context).await,
|
72
|
+
Middleware::ResponseHeaders(filter) => filter.before(req, context).await,
|
73
|
+
Middleware::RateLimit(filter) => filter.before(req, context).await,
|
74
|
+
Middleware::CacheControl(filter) => filter.before(req, context).await,
|
75
|
+
Middleware::Cors(filter) => filter.before(req, context).await,
|
76
|
+
Middleware::ETag(filter) => filter.before(req, context).await,
|
77
|
+
Middleware::StaticAssets(filter) => filter.before(req, context).await,
|
78
|
+
Middleware::Compression(filter) => filter.before(req, context).await,
|
79
|
+
Middleware::LogRequests(filter) => filter.before(req, context).await,
|
80
|
+
Middleware::Redirect(filter) => filter.before(req, context).await,
|
81
|
+
Middleware::Proxy(filter) => filter.before(req, context).await,
|
82
|
+
Middleware::RubyApp(filter) => filter.before(req, context).await,
|
83
|
+
}
|
84
|
+
}
|
85
|
+
|
86
|
+
async fn after(&self, res: HttpResponse, context: &mut RequestContext) -> HttpResponse {
|
87
|
+
match self {
|
88
|
+
Middleware::DenyList(filter) => filter.after(res, context).await,
|
89
|
+
Middleware::AllowList(filter) => filter.after(res, context).await,
|
90
|
+
Middleware::AuthBasic(filter) => filter.after(res, context).await,
|
91
|
+
Middleware::AuthJwt(filter) => filter.after(res, context).await,
|
92
|
+
Middleware::AuthAPIKey(filter) => filter.after(res, context).await,
|
93
|
+
Middleware::IntrusionProtection(filter) => filter.after(res, context).await,
|
94
|
+
Middleware::RateLimit(filter) => filter.after(res, context).await,
|
95
|
+
Middleware::RequestHeaders(filter) => filter.after(res, context).await,
|
96
|
+
Middleware::ResponseHeaders(filter) => filter.after(res, context).await,
|
97
|
+
Middleware::CacheControl(filter) => filter.after(res, context).await,
|
98
|
+
Middleware::Cors(filter) => filter.after(res, context).await,
|
99
|
+
Middleware::ETag(filter) => filter.after(res, context).await,
|
100
|
+
Middleware::StaticAssets(filter) => filter.after(res, context).await,
|
101
|
+
Middleware::Compression(filter) => filter.after(res, context).await,
|
102
|
+
Middleware::LogRequests(filter) => filter.after(res, context).await,
|
103
|
+
Middleware::Redirect(filter) => filter.after(res, context).await,
|
104
|
+
Middleware::Proxy(filter) => filter.after(res, context).await,
|
105
|
+
Middleware::RubyApp(filter) => filter.after(res, context).await,
|
106
|
+
}
|
107
|
+
}
|
108
|
+
}
|
109
|
+
|
110
|
+
impl Middleware {
|
111
|
+
fn variant_order(&self) -> usize {
|
112
|
+
match self {
|
113
|
+
Middleware::DenyList(_) => 0,
|
114
|
+
Middleware::AllowList(_) => 1,
|
115
|
+
Middleware::IntrusionProtection(_) => 2,
|
116
|
+
Middleware::Redirect(_) => 3,
|
117
|
+
Middleware::LogRequests(_) => 4,
|
118
|
+
Middleware::CacheControl(_) => 5,
|
119
|
+
Middleware::RequestHeaders(_) => 6,
|
120
|
+
Middleware::ResponseHeaders(_) => 7,
|
121
|
+
Middleware::AuthBasic(_) => 8,
|
122
|
+
Middleware::AuthJwt(_) => 9,
|
123
|
+
Middleware::AuthAPIKey(_) => 10,
|
124
|
+
Middleware::RateLimit(_) => 11,
|
125
|
+
Middleware::ETag(_) => 12,
|
126
|
+
Middleware::Compression(_) => 13,
|
127
|
+
Middleware::Proxy(_) => 14,
|
128
|
+
Middleware::Cors(_) => 15,
|
129
|
+
Middleware::StaticAssets(_) => 16,
|
130
|
+
Middleware::RubyApp(_) => 17,
|
131
|
+
}
|
132
|
+
}
|
133
|
+
}
|
134
|
+
|
135
|
+
impl PartialEq for Middleware {
|
136
|
+
fn eq(&self, other: &Self) -> bool {
|
137
|
+
self.variant_order() == other.variant_order()
|
138
|
+
}
|
139
|
+
}
|
140
|
+
|
141
|
+
impl Eq for Middleware {}
|
142
|
+
|
143
|
+
impl PartialOrd for Middleware {
|
144
|
+
fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
|
145
|
+
Some(self.variant_order().cmp(&other.variant_order()))
|
146
|
+
}
|
147
|
+
}
|
148
|
+
|
149
|
+
impl Ord for Middleware {
|
150
|
+
fn cmp(&self, other: &Self) -> Ordering {
|
151
|
+
self.variant_order().cmp(&other.variant_order())
|
152
|
+
}
|
153
|
+
}
|
@@ -0,0 +1,47 @@
|
|
1
|
+
use super::{ErrorResponse, FromValue, MiddlewareLayer};
|
2
|
+
use crate::server::{
|
3
|
+
itsi_service::RequestContext,
|
4
|
+
types::{HttpRequest, HttpResponse},
|
5
|
+
};
|
6
|
+
use async_trait::async_trait;
|
7
|
+
use either::Either;
|
8
|
+
use itsi_error::ItsiError;
|
9
|
+
use magnus::error::Result;
|
10
|
+
use regex::RegexSet;
|
11
|
+
use serde::Deserialize;
|
12
|
+
use std::sync::OnceLock;
|
13
|
+
|
14
|
+
#[derive(Debug, Clone, Deserialize)]
|
15
|
+
pub struct AllowList {
|
16
|
+
#[serde(skip_deserializing)]
|
17
|
+
pub allowed_ips: OnceLock<RegexSet>,
|
18
|
+
pub allowed_patterns: Vec<String>,
|
19
|
+
pub error_response: ErrorResponse,
|
20
|
+
}
|
21
|
+
|
22
|
+
#[async_trait]
|
23
|
+
impl MiddlewareLayer for AllowList {
|
24
|
+
async fn initialize(&self) -> Result<()> {
|
25
|
+
let allowed_ips = RegexSet::new(&self.allowed_patterns).map_err(ItsiError::default)?;
|
26
|
+
self.allowed_ips
|
27
|
+
.set(allowed_ips)
|
28
|
+
.map_err(|e| ItsiError::default(format!("Failed to set allowed IPs: {:?}", e)))?;
|
29
|
+
Ok(())
|
30
|
+
}
|
31
|
+
|
32
|
+
async fn before(
|
33
|
+
&self,
|
34
|
+
req: HttpRequest,
|
35
|
+
context: &mut RequestContext,
|
36
|
+
) -> Result<Either<HttpRequest, HttpResponse>> {
|
37
|
+
if let Some(allowed_ips) = self.allowed_ips.get() {
|
38
|
+
if !allowed_ips.is_match(&context.addr) {
|
39
|
+
return Ok(Either::Right(
|
40
|
+
self.error_response.to_http_response(&req).await,
|
41
|
+
));
|
42
|
+
}
|
43
|
+
}
|
44
|
+
Ok(Either::Left(req))
|
45
|
+
}
|
46
|
+
}
|
47
|
+
impl FromValue for AllowList {}
|
@@ -0,0 +1,58 @@
|
|
1
|
+
use crate::server::{
|
2
|
+
itsi_service::RequestContext,
|
3
|
+
types::{HttpRequest, HttpResponse, RequestExt},
|
4
|
+
};
|
5
|
+
|
6
|
+
use super::{error_response::ErrorResponse, token_source::TokenSource, FromValue, MiddlewareLayer};
|
7
|
+
|
8
|
+
use async_trait::async_trait;
|
9
|
+
use either::Either;
|
10
|
+
use magnus::error::Result;
|
11
|
+
use serde::Deserialize;
|
12
|
+
|
13
|
+
/// A simple API key filter.
|
14
|
+
/// The API key can be given inside the header or a query string
|
15
|
+
/// Keys are validated against a list of allowed key values (Changing these requires a restart)
|
16
|
+
///
|
17
|
+
#[derive(Debug, Clone, Deserialize)]
|
18
|
+
pub struct AuthAPIKey {
|
19
|
+
pub valid_keys: Vec<String>,
|
20
|
+
pub token_source: TokenSource,
|
21
|
+
pub error_response: ErrorResponse,
|
22
|
+
}
|
23
|
+
|
24
|
+
#[async_trait]
|
25
|
+
impl MiddlewareLayer for AuthAPIKey {
|
26
|
+
async fn before(
|
27
|
+
&self,
|
28
|
+
req: HttpRequest,
|
29
|
+
_context: &mut RequestContext,
|
30
|
+
) -> Result<Either<HttpRequest, HttpResponse>> {
|
31
|
+
let submitted_value = match &self.token_source {
|
32
|
+
TokenSource::Header { name, prefix } => {
|
33
|
+
if let Some(header) = req.header(name) {
|
34
|
+
if let Some(prefix) = prefix {
|
35
|
+
Some(header.strip_prefix(prefix).unwrap_or("").trim_ascii())
|
36
|
+
} else {
|
37
|
+
Some(header.trim_ascii())
|
38
|
+
}
|
39
|
+
} else {
|
40
|
+
None
|
41
|
+
}
|
42
|
+
}
|
43
|
+
TokenSource::Query(query_name) => req.query_param(query_name),
|
44
|
+
};
|
45
|
+
if !self
|
46
|
+
.valid_keys
|
47
|
+
.iter()
|
48
|
+
.any(|key| submitted_value.is_some_and(|sv| sv == key))
|
49
|
+
{
|
50
|
+
Ok(Either::Right(
|
51
|
+
self.error_response.to_http_response(&req).await,
|
52
|
+
))
|
53
|
+
} else {
|
54
|
+
Ok(Either::Left(req))
|
55
|
+
}
|
56
|
+
}
|
57
|
+
}
|
58
|
+
impl FromValue for AuthAPIKey {}
|
@@ -0,0 +1,82 @@
|
|
1
|
+
use async_trait::async_trait;
|
2
|
+
use base64::{engine::general_purpose, Engine};
|
3
|
+
use bytes::Bytes;
|
4
|
+
use either::Either;
|
5
|
+
use http::{Response, StatusCode};
|
6
|
+
use http_body_util::{combinators::BoxBody, Full};
|
7
|
+
use magnus::error::Result;
|
8
|
+
use serde::{Deserialize, Serialize};
|
9
|
+
use std::collections::HashMap;
|
10
|
+
use std::str;
|
11
|
+
|
12
|
+
use crate::server::{
|
13
|
+
itsi_service::RequestContext,
|
14
|
+
types::{HttpRequest, HttpResponse, RequestExt},
|
15
|
+
};
|
16
|
+
|
17
|
+
use super::{FromValue, MiddlewareLayer};
|
18
|
+
|
19
|
+
#[derive(Debug, Clone, Serialize, Deserialize)]
|
20
|
+
pub struct AuthBasic {
|
21
|
+
pub realm: String,
|
22
|
+
/// Maps usernames to passwords.
|
23
|
+
pub credential_pairs: HashMap<String, String>,
|
24
|
+
}
|
25
|
+
|
26
|
+
impl AuthBasic {
|
27
|
+
fn basic_auth_failed_response(&self) -> HttpResponse {
|
28
|
+
Response::builder()
|
29
|
+
.status(StatusCode::UNAUTHORIZED)
|
30
|
+
.header(
|
31
|
+
"WWW-Authenticate",
|
32
|
+
format!("Basic realm=\"{}\"", self.realm),
|
33
|
+
)
|
34
|
+
.body(BoxBody::new(Full::new(Bytes::from("Unauthorized"))))
|
35
|
+
.unwrap()
|
36
|
+
}
|
37
|
+
}
|
38
|
+
#[async_trait]
|
39
|
+
impl MiddlewareLayer for AuthBasic {
|
40
|
+
async fn before(
|
41
|
+
&self,
|
42
|
+
req: HttpRequest,
|
43
|
+
_context: &mut RequestContext,
|
44
|
+
) -> Result<Either<HttpRequest, HttpResponse>> {
|
45
|
+
// Retrieve the Authorization header.
|
46
|
+
let auth_header = req.header("Authorization");
|
47
|
+
|
48
|
+
if !auth_header.is_some_and(|header| header.starts_with("Basic ")) {
|
49
|
+
return Ok(Either::Right(self.basic_auth_failed_response()));
|
50
|
+
}
|
51
|
+
|
52
|
+
let auth_header = auth_header.unwrap();
|
53
|
+
|
54
|
+
let encoded_credentials = &auth_header["Basic ".len()..];
|
55
|
+
let decoded = match general_purpose::STANDARD.decode(encoded_credentials) {
|
56
|
+
Ok(bytes) => bytes,
|
57
|
+
Err(_) => {
|
58
|
+
return Ok(Either::Right(self.basic_auth_failed_response()));
|
59
|
+
}
|
60
|
+
};
|
61
|
+
|
62
|
+
let decoded_str = match str::from_utf8(&decoded) {
|
63
|
+
Ok(s) => s,
|
64
|
+
Err(_) => {
|
65
|
+
return Ok(Either::Right(self.basic_auth_failed_response()));
|
66
|
+
}
|
67
|
+
};
|
68
|
+
|
69
|
+
let mut parts = decoded_str.splitn(2, ':');
|
70
|
+
let username = parts.next().unwrap_or("");
|
71
|
+
let password = parts.next().unwrap_or("");
|
72
|
+
|
73
|
+
match self.credential_pairs.get(username) {
|
74
|
+
Some(expected_password) if expected_password == password => Ok(Either::Left(req)),
|
75
|
+
_ => {
|
76
|
+
return Ok(Either::Right(self.basic_auth_failed_response()));
|
77
|
+
}
|
78
|
+
}
|
79
|
+
}
|
80
|
+
}
|
81
|
+
|
82
|
+
impl FromValue for AuthBasic {}
|
@@ -0,0 +1,321 @@
|
|
1
|
+
use super::{error_response::ErrorResponse, token_source::TokenSource, FromValue, MiddlewareLayer};
|
2
|
+
use crate::server::{
|
3
|
+
itsi_service::RequestContext,
|
4
|
+
types::{HttpRequest, HttpResponse, RequestExt},
|
5
|
+
};
|
6
|
+
use async_trait::async_trait;
|
7
|
+
use base64::{engine::general_purpose, Engine};
|
8
|
+
use either::Either;
|
9
|
+
use itsi_error::ItsiError;
|
10
|
+
use jwt_simple::{
|
11
|
+
claims::{self, JWTClaims, NoCustomClaims},
|
12
|
+
prelude::{
|
13
|
+
ECDSAP256PublicKeyLike, ECDSAP384PublicKeyLike, ES256PublicKey, ES384PublicKey, HS256Key,
|
14
|
+
HS384Key, HS512Key, MACLike, PS256PublicKey, PS384PublicKey, PS512PublicKey,
|
15
|
+
RS256PublicKey, RS384PublicKey, RS512PublicKey, RSAPublicKeyLike,
|
16
|
+
},
|
17
|
+
token::Token,
|
18
|
+
};
|
19
|
+
use magnus::error::Result;
|
20
|
+
use serde::Deserialize;
|
21
|
+
use std::str;
|
22
|
+
use std::{
|
23
|
+
collections::{HashMap, HashSet},
|
24
|
+
sync::OnceLock,
|
25
|
+
};
|
26
|
+
|
27
|
+
#[derive(Debug, Clone, Deserialize)]
|
28
|
+
pub struct AuthJwt {
|
29
|
+
pub token_source: TokenSource,
|
30
|
+
pub verifiers: HashMap<JwtAlgorithm, Vec<String>>,
|
31
|
+
#[serde(skip_deserializing)]
|
32
|
+
pub keys: OnceLock<HashMap<JwtAlgorithm, Vec<JwtKey>>>,
|
33
|
+
pub audiences: Option<HashSet<String>>,
|
34
|
+
pub subjects: Option<HashSet<String>>,
|
35
|
+
pub issuers: Option<HashSet<String>>,
|
36
|
+
pub leeway: Option<u64>,
|
37
|
+
pub error_response: ErrorResponse,
|
38
|
+
}
|
39
|
+
|
40
|
+
#[derive(Debug, Clone, Deserialize, PartialEq, Eq, Hash)]
|
41
|
+
pub enum JwtAlgorithm {
|
42
|
+
#[serde(rename(deserialize = "hs256"))]
|
43
|
+
Hs256,
|
44
|
+
#[serde(rename(deserialize = "hs384"))]
|
45
|
+
Hs384,
|
46
|
+
#[serde(rename(deserialize = "hs512"))]
|
47
|
+
Hs512,
|
48
|
+
#[serde(rename(deserialize = "rs256"))]
|
49
|
+
Rs256,
|
50
|
+
#[serde(rename(deserialize = "rs384"))]
|
51
|
+
Rs384,
|
52
|
+
#[serde(rename(deserialize = "rs512"))]
|
53
|
+
Rs512,
|
54
|
+
#[serde(rename(deserialize = "es256"))]
|
55
|
+
Es256,
|
56
|
+
#[serde(rename(deserialize = "es384"))]
|
57
|
+
Es384,
|
58
|
+
#[serde(rename(deserialize = "ps256"))]
|
59
|
+
Ps256,
|
60
|
+
#[serde(rename(deserialize = "ps384"))]
|
61
|
+
Ps384,
|
62
|
+
#[serde(rename(deserialize = "ps512"))]
|
63
|
+
Ps512,
|
64
|
+
}
|
65
|
+
|
66
|
+
impl JwtAlgorithm {
|
67
|
+
pub fn key_from(&self, base64: &str) -> Result<JwtKey> {
|
68
|
+
let bytes = general_purpose::STANDARD
|
69
|
+
.decode(base64)
|
70
|
+
.map_err(ItsiError::default)?;
|
71
|
+
|
72
|
+
match self {
|
73
|
+
JwtAlgorithm::Hs256 => Ok(JwtKey::Hs256(HS256Key::from_bytes(&bytes))),
|
74
|
+
JwtAlgorithm::Hs384 => Ok(JwtKey::Hs384(HS384Key::from_bytes(&bytes))),
|
75
|
+
JwtAlgorithm::Hs512 => Ok(JwtKey::Hs512(HS512Key::from_bytes(&bytes))),
|
76
|
+
JwtAlgorithm::Rs256 => Ok(RS256PublicKey::from_der(&bytes)
|
77
|
+
.or_else(|_| {
|
78
|
+
RS256PublicKey::from_pem(
|
79
|
+
&String::from_utf8(bytes.clone()).map_err(ItsiError::default)?,
|
80
|
+
)
|
81
|
+
})
|
82
|
+
.map(JwtKey::Rs256)
|
83
|
+
.map_err(ItsiError::default)?),
|
84
|
+
JwtAlgorithm::Rs384 => Ok(RS384PublicKey::from_der(&bytes)
|
85
|
+
.or_else(|_| {
|
86
|
+
RS384PublicKey::from_pem(
|
87
|
+
&String::from_utf8(bytes.clone()).map_err(ItsiError::default)?,
|
88
|
+
)
|
89
|
+
})
|
90
|
+
.map(JwtKey::Rs384)
|
91
|
+
.map_err(ItsiError::default)?),
|
92
|
+
JwtAlgorithm::Rs512 => Ok(RS512PublicKey::from_der(&bytes)
|
93
|
+
.or_else(|_| {
|
94
|
+
RS512PublicKey::from_pem(
|
95
|
+
&String::from_utf8(bytes.clone()).map_err(ItsiError::default)?,
|
96
|
+
)
|
97
|
+
})
|
98
|
+
.map(JwtKey::Rs512)
|
99
|
+
.map_err(ItsiError::default)?),
|
100
|
+
JwtAlgorithm::Es256 => Ok(ES256PublicKey::from_der(&bytes)
|
101
|
+
.or_else(|_| {
|
102
|
+
ES256PublicKey::from_pem(
|
103
|
+
&String::from_utf8(bytes.clone()).map_err(ItsiError::default)?,
|
104
|
+
)
|
105
|
+
})
|
106
|
+
.map(JwtKey::Es256)
|
107
|
+
.map_err(ItsiError::default)?),
|
108
|
+
JwtAlgorithm::Es384 => Ok(ES384PublicKey::from_der(&bytes)
|
109
|
+
.or_else(|_| {
|
110
|
+
ES384PublicKey::from_pem(
|
111
|
+
&String::from_utf8(bytes.clone()).map_err(ItsiError::default)?,
|
112
|
+
)
|
113
|
+
})
|
114
|
+
.map(JwtKey::Es384)
|
115
|
+
.map_err(ItsiError::default)?),
|
116
|
+
JwtAlgorithm::Ps256 => Ok(PS256PublicKey::from_der(&bytes)
|
117
|
+
.or_else(|_| {
|
118
|
+
PS256PublicKey::from_pem(
|
119
|
+
&String::from_utf8(bytes.clone()).map_err(ItsiError::default)?,
|
120
|
+
)
|
121
|
+
})
|
122
|
+
.map(JwtKey::Ps256)
|
123
|
+
.map_err(ItsiError::default)?),
|
124
|
+
JwtAlgorithm::Ps384 => Ok(PS384PublicKey::from_der(&bytes)
|
125
|
+
.or_else(|_| {
|
126
|
+
PS384PublicKey::from_pem(
|
127
|
+
&String::from_utf8(bytes.clone()).map_err(ItsiError::default)?,
|
128
|
+
)
|
129
|
+
})
|
130
|
+
.map(JwtKey::Ps384)
|
131
|
+
.map_err(ItsiError::default)?),
|
132
|
+
JwtAlgorithm::Ps512 => Ok(PS512PublicKey::from_der(&bytes)
|
133
|
+
.or_else(|_| {
|
134
|
+
PS512PublicKey::from_pem(
|
135
|
+
&String::from_utf8(bytes.clone()).map_err(ItsiError::default)?,
|
136
|
+
)
|
137
|
+
})
|
138
|
+
.map(JwtKey::Ps512)
|
139
|
+
.map_err(ItsiError::default)?),
|
140
|
+
}
|
141
|
+
}
|
142
|
+
}
|
143
|
+
|
144
|
+
#[derive(Debug, Clone)]
|
145
|
+
pub enum JwtKey {
|
146
|
+
Hs256(HS256Key),
|
147
|
+
Hs384(HS384Key),
|
148
|
+
Hs512(HS512Key),
|
149
|
+
Rs256(RS256PublicKey),
|
150
|
+
Rs384(RS384PublicKey),
|
151
|
+
Rs512(RS512PublicKey),
|
152
|
+
Es256(ES256PublicKey),
|
153
|
+
Es384(ES384PublicKey),
|
154
|
+
Ps256(PS256PublicKey),
|
155
|
+
Ps384(PS384PublicKey),
|
156
|
+
Ps512(PS512PublicKey),
|
157
|
+
}
|
158
|
+
|
159
|
+
impl TryFrom<&str> for JwtAlgorithm {
|
160
|
+
type Error = itsi_error::ItsiError;
|
161
|
+
|
162
|
+
fn try_from(value: &str) -> std::result::Result<Self, Self::Error> {
|
163
|
+
match value.to_ascii_lowercase().as_str() {
|
164
|
+
"hs256" => Ok(JwtAlgorithm::Hs256),
|
165
|
+
"hs384" => Ok(JwtAlgorithm::Hs384),
|
166
|
+
"hs512" => Ok(JwtAlgorithm::Hs512),
|
167
|
+
"rs256" => Ok(JwtAlgorithm::Rs256),
|
168
|
+
"rs384" => Ok(JwtAlgorithm::Rs384),
|
169
|
+
"rs512" => Ok(JwtAlgorithm::Rs512),
|
170
|
+
"es256" => Ok(JwtAlgorithm::Es256),
|
171
|
+
"es384" => Ok(JwtAlgorithm::Es384),
|
172
|
+
"ps256" => Ok(JwtAlgorithm::Ps256),
|
173
|
+
"ps384" => Ok(JwtAlgorithm::Ps384),
|
174
|
+
"ps512" => Ok(JwtAlgorithm::Ps512),
|
175
|
+
_ => Err(itsi_error::ItsiError::UnsupportedProtocol(
|
176
|
+
"Unsupported JWT Algorithm".to_string(),
|
177
|
+
)),
|
178
|
+
}
|
179
|
+
}
|
180
|
+
}
|
181
|
+
|
182
|
+
impl JwtKey {
|
183
|
+
pub fn verify(
|
184
|
+
&self,
|
185
|
+
token: &str,
|
186
|
+
) -> std::result::Result<JWTClaims<claims::NoCustomClaims>, jwt_simple::Error> {
|
187
|
+
match self {
|
188
|
+
JwtKey::Hs256(key) => key.verify_token::<NoCustomClaims>(token, None),
|
189
|
+
JwtKey::Hs384(key) => key.verify_token::<NoCustomClaims>(token, None),
|
190
|
+
JwtKey::Hs512(key) => key.verify_token::<NoCustomClaims>(token, None),
|
191
|
+
JwtKey::Rs256(key) => key.verify_token::<NoCustomClaims>(token, None),
|
192
|
+
JwtKey::Rs384(key) => key.verify_token::<NoCustomClaims>(token, None),
|
193
|
+
JwtKey::Rs512(key) => key.verify_token::<NoCustomClaims>(token, None),
|
194
|
+
JwtKey::Es256(key) => key.verify_token::<NoCustomClaims>(token, None),
|
195
|
+
JwtKey::Es384(key) => key.verify_token::<NoCustomClaims>(token, None),
|
196
|
+
JwtKey::Ps256(key) => key.verify_token::<NoCustomClaims>(token, None),
|
197
|
+
JwtKey::Ps384(key) => key.verify_token::<NoCustomClaims>(token, None),
|
198
|
+
JwtKey::Ps512(key) => key.verify_token::<NoCustomClaims>(token, None),
|
199
|
+
}
|
200
|
+
}
|
201
|
+
}
|
202
|
+
|
203
|
+
#[async_trait]
|
204
|
+
impl MiddlewareLayer for AuthJwt {
|
205
|
+
async fn initialize(&self) -> Result<()> {
|
206
|
+
let keys: HashMap<JwtAlgorithm, Vec<JwtKey>> = self
|
207
|
+
.verifiers
|
208
|
+
.iter()
|
209
|
+
.map(|(algorithm, key_strings)| {
|
210
|
+
let algo = algorithm.clone();
|
211
|
+
let keys: Result<Vec<JwtKey>> = key_strings
|
212
|
+
.iter()
|
213
|
+
.map(|key_string| algorithm.key_from(key_string))
|
214
|
+
.collect();
|
215
|
+
keys.map(|keys| (algo, keys))
|
216
|
+
})
|
217
|
+
.collect::<Result<HashMap<JwtAlgorithm, Vec<JwtKey>>>>()?;
|
218
|
+
self.keys
|
219
|
+
.set(keys)
|
220
|
+
.map_err(|e| ItsiError::default(format!("Failed to set keys: {:?}", e)))?;
|
221
|
+
Ok(())
|
222
|
+
}
|
223
|
+
|
224
|
+
async fn before(
|
225
|
+
&self,
|
226
|
+
req: HttpRequest,
|
227
|
+
_context: &mut RequestContext,
|
228
|
+
) -> Result<Either<HttpRequest, HttpResponse>> {
|
229
|
+
let token_str = match &self.token_source {
|
230
|
+
TokenSource::Header { name, prefix } => {
|
231
|
+
if let Some(header) = req.header(name) {
|
232
|
+
if let Some(prefix) = prefix {
|
233
|
+
Some(header.strip_prefix(prefix).unwrap_or("").trim_ascii())
|
234
|
+
} else {
|
235
|
+
Some(header.trim_ascii())
|
236
|
+
}
|
237
|
+
} else {
|
238
|
+
None
|
239
|
+
}
|
240
|
+
}
|
241
|
+
TokenSource::Query(query_name) => req.query_param(query_name),
|
242
|
+
};
|
243
|
+
|
244
|
+
if token_str.is_none() {
|
245
|
+
return Ok(Either::Right(
|
246
|
+
self.error_response.to_http_response(&req).await,
|
247
|
+
));
|
248
|
+
}
|
249
|
+
|
250
|
+
let token_str = token_str.unwrap();
|
251
|
+
let token_meta = Token::decode_metadata(token_str);
|
252
|
+
|
253
|
+
if token_meta.is_err() {
|
254
|
+
return Ok(Either::Right(
|
255
|
+
self.error_response.to_http_response(&req).await,
|
256
|
+
));
|
257
|
+
}
|
258
|
+
let token_meta: std::result::Result<JwtAlgorithm, ItsiError> =
|
259
|
+
token_meta.unwrap().algorithm().try_into();
|
260
|
+
if token_meta.is_err() {
|
261
|
+
return Ok(Either::Right(
|
262
|
+
self.error_response.to_http_response(&req).await,
|
263
|
+
));
|
264
|
+
}
|
265
|
+
let algorithm = token_meta.unwrap();
|
266
|
+
|
267
|
+
if !self.verifiers.contains_key(&algorithm) {
|
268
|
+
return Ok(Either::Right(
|
269
|
+
self.error_response.to_http_response(&req).await,
|
270
|
+
));
|
271
|
+
}
|
272
|
+
|
273
|
+
let keys = self.keys.get().unwrap().get(&algorithm).unwrap();
|
274
|
+
|
275
|
+
let verified_claims = keys.iter().find_map(|key| key.verify(token_str).ok());
|
276
|
+
if verified_claims.is_none() {
|
277
|
+
return Ok(Either::Right(
|
278
|
+
self.error_response.to_http_response(&req).await,
|
279
|
+
));
|
280
|
+
}
|
281
|
+
|
282
|
+
let claims = verified_claims.unwrap();
|
283
|
+
|
284
|
+
if let Some(expected_audiences) = &self.audiences {
|
285
|
+
// The aud claim may be a string or an array.
|
286
|
+
if let Some(audiences) = &claims.audiences {
|
287
|
+
if !audiences.contains(expected_audiences) {
|
288
|
+
return Ok(Either::Right(
|
289
|
+
self.error_response.to_http_response(&req).await,
|
290
|
+
));
|
291
|
+
}
|
292
|
+
}
|
293
|
+
}
|
294
|
+
|
295
|
+
if let Some(expected_subjects) = &self.subjects {
|
296
|
+
// The aud claim may be a string or an array.
|
297
|
+
if let Some(subject) = &claims.subject {
|
298
|
+
if !expected_subjects.contains(subject) {
|
299
|
+
return Ok(Either::Right(
|
300
|
+
self.error_response.to_http_response(&req).await,
|
301
|
+
));
|
302
|
+
}
|
303
|
+
}
|
304
|
+
}
|
305
|
+
|
306
|
+
if let Some(expected_issuers) = &self.issuers {
|
307
|
+
// The aud claim may be a string or an array.
|
308
|
+
if let Some(issuer) = &claims.issuer {
|
309
|
+
if !expected_issuers.contains(issuer) {
|
310
|
+
return Ok(Either::Right(
|
311
|
+
self.error_response.to_http_response(&req).await,
|
312
|
+
));
|
313
|
+
}
|
314
|
+
}
|
315
|
+
}
|
316
|
+
|
317
|
+
Ok(Either::Left(req))
|
318
|
+
}
|
319
|
+
}
|
320
|
+
|
321
|
+
impl FromValue for AuthJwt {}
|