change cors config, support string config

Author: timzaak

config.release.toml

@@ -1,17 +1,18 @@
 ## directory to store static web files. if you use docker, please mount a persistence volume for it.
 file_dir = "/data"
 
-## enable cors, default is false, its implementation is simple now.
+## enable cors, default is none, if cors is [], then all cors is ok.
 ## Access-Control-Allow-Origin: $ORIGIN
 ## Access-Control-Allow-Methods: OPTION,GET,HEAD
 ## Access-Control-Max-Age: 3600
-## If you put the server behind HTTPS proxy, please enable it
-cors = false
+## If you put the server behind HTTPS proxy, please enable it, or domains.cors = ['http://www.example.com:8080']
+## Attension: domains.cors config would overwrite the cors config, rather than merge this.
+cors = []
 ## http bind, if set port <= 0 or remove http, will disable http server(need set https config)
 [http]
 port = 80
 addr = "0.0.0.0"
-## port when serving public PI,default is http port. external_port should not be 0.
+## port when serving public network,default is http port. external_port should not be 0.
 # external_port = 80
 
 ## optional, when https enabled, redirect_https default value true
@@ -21,7 +22,7 @@ addr = "0.0.0.0"
 # [https]
 # port = 443 # https bind address
 # addr = "0.0.0.0"
-## port when serving public PI,default is https port. external_port should not be 0.
+## port when serving public network,default is https port. external_port should not be 0.
 # external_port = 443
 
 ## if set true, http server(80) will send client
@@ -89,7 +90,7 @@ addr = "0.0.0.0"
 # domain = "www.example.com"
 ## optional, `example.com` would redirect to `www.example.com`
 # alias = ["example.com"]
-# cors = false
+# cors = ['https://www.example.com', 'http://www.baidu.com']
 # [domains.https]
 ## optional, when https enabled, redirect_https default value true
 ## it would the port would be https.external_port(https.external_port should be defined), otherwise is false
@@ -106,7 +107,6 @@ addr = "0.0.0.0"
 # expire = '30d' # 30day
 # extension_names = ['icon', 'gif', 'jpg', 'jpeg', 'png', 'js']
 
-[openTelemetry]
-endpoint = "http://localhost:4317"
-
 
+# [openTelemetry]
+# endpoint = "http://localhost:4317"

docs/develop/change-log.md

@@ -1,5 +1,8 @@
 # Change Log
 
+### Version 2.4.1
+- spa-server: `cors` value bool to array string.
+
 ### Version 2.4.0
 
 - improve: extract client and server common entity

docs/guide/break-changes.md

@@ -1,5 +1,7 @@
 # Break Changes
-## V2.4.0
+## V2.4.1
+* spa-server: `cors` value bool to array string.
+## V2.4.0(2024-08-04)
 * remove hocon config format, change to toml.
 ## V2.3.0(2024-07-01)
 * spa-server: `https.http_redirect_to_https` move to `http.redirect_https`, and value is bool.

docs/guide/spa-server-configuration.md

@@ -14,17 +14,18 @@ The config default path is `./config.toml`, you can change it by environment `SP
 ## directory to store static web files. if you use docker, please mount a persistence volume for it.
 file_dir = "/data"
 
-## enable cors, default is false, its implementation is simple now.
+## enable cors, default is none, if cors is [], then all cors is ok.
 ## Access-Control-Allow-Origin: $ORIGIN
 ## Access-Control-Allow-Methods: OPTION,GET,HEAD
 ## Access-Control-Max-Age: 3600
-## If you put the server behind HTTPS proxy, please enable it
-cors = false
+## If you put the server behind HTTPS proxy, please enable it, or domains.cors = ['http://www.example.com:8080']
+## Attension: domains.cors config would overwrite the cors config, rather than merge this.
+cors = []
 ## http bind, if set port <= 0 or remove http, will disable http server(need set https config)
 [http]
 port = 80
 addr = "0.0.0.0"
-## port when serving public PI,default is http port. external_port should not be 0.
+## port when serving public network,default is http port. external_port should not be 0.
 # external_port = 80
 
 ## optional, when https enabled, redirect_https default value true
@@ -34,7 +35,7 @@ addr = "0.0.0.0"
 # [https]
 # port = 443 # https bind address
 # addr = "0.0.0.0"
-## port when serving public PI,default is https port. external_port should not be 0.
+## port when serving public network,default is https port. external_port should not be 0.
 # external_port = 443
 
 ## if set true, http server(80) will send client
@@ -102,7 +103,7 @@ addr = "0.0.0.0"
 # domain = "www.example.com"
 ## optional, `example.com` would redirect to `www.example.com`
 # alias = ["example.com"]
-# cors = false
+# cors = ['https://www.example.com', 'http://www.baidu.com']
 # [domains.https]
 ## optional, when https enabled, redirect_https default value true
 ## it would the port would be https.external_port(https.external_port should be defined), otherwise is false

server/src/config.rs

@@ -1,9 +1,11 @@
 use anyhow::{bail, Context};
 use duration_str::deserialize_duration;
-use serde::Deserialize;
+use serde::{Deserialize, Deserializer};
 use small_acme::LetsEncrypt;
 use std::time::Duration;
 use std::{env, fs};
+use std::collections::HashSet;
+use headers::{HeaderValue, Origin};
 use tracing::warn;
 
 const CONFIG_PATH: &str = "/config/config.toml";
@@ -12,7 +14,7 @@ const CONFIG_PATH: &str = "/config/config.toml";
 pub struct Config {
     pub file_dir: String,
     #[serde(default)]
-    pub cors: bool,
+    pub cors: Option<HashSet<OriginWrapper>>,
     pub admin_config: Option<AdminConfig>,
     pub http: Option<HttpConfig>,
     pub https: Option<HttpsConfig>,
@@ -94,7 +96,7 @@ fn default_max_upload_size() -> u64 {
 #[derive(Deserialize, Debug, Clone, PartialEq)]
 pub struct DomainConfig {
     pub domain: String,
-    pub cors: Option<bool>,
+    pub cors: Option<HashSet<OriginWrapper>>,
     pub cache: Option<DomainCacheConfig>,
     pub https: Option<DomainHttpsConfig>,
     pub alias: Option<Vec<String>>,
@@ -226,6 +228,29 @@ pub fn get_host_path_from_domain(domain: &str) -> (&str, &str) {
     }
 }
 
+#[derive(Debug, Clone, PartialEq, Eq, Hash)]
+pub struct OriginWrapper(HeaderValue);
+
+pub(crate) fn extract_origin(data:&Option<HashSet<OriginWrapper>>) -> Option<HashSet<HeaderValue>> {
+    data.as_ref().map(|set| set.iter().map(|o| o.0.clone()).collect())
+}
+
+impl <'de> Deserialize<'de> for OriginWrapper {
+    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
+    where
+        D: Deserializer<'de>
+    {
+        let data = String::deserialize(deserializer)?;
+        let mut parts = data.splitn(2, "://");
+        let scheme = parts.next().expect("missing scheme");
+        let rest = parts.next().expect("missing scheme");
+        let origin = Origin::try_from_parts(scheme, rest, None).expect("invalid Origin");
+        
+        Ok(OriginWrapper(origin.to_string().parse()
+            .expect("Origin is always a valid HeaderValue")))
+    }
+}
+
 #[cfg(test)]
 mod test {
     use std::env;

server/src/cors.rs

@@ -30,14 +30,23 @@ pub fn cors_resp(mut res: Response<Body>, origin: HeaderValue) -> Response<Body>
     res
 }
 
+fn is_origin_allowed(origins: &Option<HashSet<HeaderValue>>, origin: &HeaderValue) -> bool {
+    if let Some(ref allowed) = origins {
+        allowed.is_empty()||allowed.contains(origin)
+    } else {
+        false
+    }
+}
+// preflight response
 pub fn resp_cors_request(
     method: &Method,
     headers: &HeaderMap,
-    allow_cors: bool,
+    origins: &Option<HashSet<HeaderValue>>,
 ) -> Either<Validated, Response<Body>> {
     match (headers.get(header::ORIGIN), method) {
         (Some(origin), &Method::OPTIONS) => {
-            if !allow_cors {
+            
+            if !is_origin_allowed(origins, origin) {
                 return Either::Right(resp(StatusCode::FORBIDDEN, "origin not allowed"));
             }
             if let Some(req_method) = headers.get(header::ACCESS_CONTROL_REQUEST_METHOD) {
@@ -59,7 +68,7 @@ pub fn resp_cors_request(
             Either::Right(res)
         }
         (Some(origin), _) => {
-            if !allow_cors {
+            if !is_origin_allowed(origins, origin) {
                 Either::Right(resp(StatusCode::FORBIDDEN, "origin not allowed"))
             } else {
                 Either::Left(Validated::Simple(origin.clone()))

server/src/service.rs

@@ -2,11 +2,11 @@ use crate::acme::{get_challenge_path, ChallengePath, ACME_CHALLENGE};
 use crate::cors::{cors_resp, resp_cors_request, Validated};
 use crate::DomainStorage;
 use futures_util::future::Either;
-use headers::HeaderMapExt;
+use headers::{HeaderMapExt, HeaderValue};
 use hyper::header::LOCATION;
 use hyper::http::uri::Authority;
 use hyper::{Body, Request, Response, StatusCode};
-use std::collections::HashMap;
+use std::collections::{HashMap, HashSet};
 use std::convert::Infallible;
 use std::str::FromStr;
 use std::sync::Arc;
@@ -24,7 +24,7 @@ pub struct ServiceConfig {
 }
 
 pub struct DomainServiceConfig {
-    pub cors: bool,
+    pub cors: Option<HashSet<HeaderValue>>,
     pub redirect_https: Option<u16>,
     pub enable_acme: bool,
 }
@@ -130,7 +130,7 @@ pub async fn create_http_service(
 
         let service_config = service_config.get_domain_service_config(host);
         // cors
-        let origin_opt = match resp_cors_request(req.method(), req.headers(), service_config.cors) {
+        let origin_opt = match resp_cors_request(req.method(), req.headers(), &service_config.cors) {
             Either::Left(x) => Some(x),
             Either::Right(v) => return Ok(v),
         };
@@ -194,7 +194,7 @@ pub async fn create_https_service(
 
         let service_config = service_config.get_domain_service_config(host);
         // cors
-        let origin_opt = match resp_cors_request(req.method(), req.headers(), service_config.cors) {
+        let origin_opt = match resp_cors_request(req.method(), req.headers(), &service_config.cors) {
             Either::Left(x) => Some(x),
             Either::Right(v) => return Ok(v),
         };

server/src/web_server.rs

@@ -15,7 +15,7 @@ use futures_util::future::Either;
 use tokio::net::TcpListener as TKTcpListener;
 use tokio::sync::oneshot::Receiver;
 
-use crate::config::{Config, HttpConfig, HttpsConfig};
+use crate::config::{extract_origin, Config, HttpConfig, HttpsConfig};
 use crate::domain_storage::DomainStorage;
 use crate::service::{create_http_service, create_https_service, DomainServiceConfig, ServiceConfig};
 use crate::tls::TlsAcceptor;
@@ -93,7 +93,7 @@ impl Server {
         };
 
         let default = DomainServiceConfig {
-            cors: conf.cors,
+            cors: extract_origin(&conf.cors),
             redirect_https: default_http_redirect_to_https,
             enable_acme: conf.https.as_ref().and_then(|x| x.acme.as_ref()).is_some(),
         };
@@ -116,7 +116,7 @@ impl Server {
                 Some(false) => None
             };
             let domain_service_config: DomainServiceConfig = DomainServiceConfig {
-                cors: domain.cors.unwrap_or(default.cors),
+                cors: extract_origin(&domain.cors).or_else(||default.cors.clone()),
                 redirect_https,
                 enable_acme: domain
                     .https

tests/data/server_config.toml

@@ -1,5 +1,5 @@
 file_dir = "./data/web"
-cors = true
+cors = []
 [http]
 port = 8080
 addr = "0.0.0.0"

tests/data/server_config_acme.toml

@@ -1,5 +1,5 @@
 file_dir = "./data/web"
-cors = true
+cors = []
 
 # http bind, if set port <= 0, will disable http server(need set https config)
 [http]