fix(proxy-nodes): allow management tokens with json null allowed_ips (#317)

* fix(proxy-nodes): allow management tokens with json null allowed_ips

* style(proxy-nodes): format regression test
This commit is contained in:
RWDai
2026-04-20 21:50:48 +08:00
committed by GitHub
parent 226a6e58d5
commit cf7d129595
2 changed files with 75 additions and 0 deletions

View File

@@ -120,6 +120,9 @@ fn remote_ip_allowed(allowed_ips: Option<&serde_json::Value>, remote_ip: std::ne
let Some(allowed_ips) = allowed_ips else { let Some(allowed_ips) = allowed_ips else {
return true; return true;
}; };
if allowed_ips.is_null() {
return true;
}
let Some(items) = allowed_ips.as_array() else { let Some(items) = allowed_ips.as_array() else {
return false; return false;
}; };

View File

@@ -763,6 +763,78 @@ async fn gateway_registers_and_unregisters_proxy_nodes_locally_with_management_t
upstream_handle.abort(); upstream_handle.abort();
} }
#[tokio::test]
async fn gateway_registers_proxy_node_with_management_token_when_allowed_ips_is_json_null() {
let raw_token = "ae_proxy_register_json_null";
let proxy_node_repository = Arc::new(InMemoryProxyNodeRepository::default());
let state = AppState::new().expect("gateway should build");
let admin_user = state
.create_local_auth_user_with_settings(
Some("proxy-admin@example.com".to_string()),
true,
"admin".to_string(),
"hash".to_string(),
"admin".to_string(),
None,
None,
None,
None,
)
.await
.expect("admin user should be created")
.expect("admin user should exist");
let mut management_token = sample_management_token(
"token-proxy-register-json-null",
&admin_user.id,
"proxy-admin",
true,
);
management_token.token.allowed_ips = Some(serde_json::Value::Null);
let management_token_repository =
Arc::new(InMemoryManagementTokenRepository::seed_with_hashes(
vec![management_token],
vec![(
hash_management_token(raw_token),
"token-proxy-register-json-null".to_string(),
)],
));
let state = state.with_data_state_for_tests(
GatewayDataState::with_management_token_repository_for_tests(management_token_repository)
.attach_proxy_node_repository_for_tests(proxy_node_repository),
);
let gateway = build_router_with_state(state);
let (gateway_url, gateway_handle) = start_server(gateway).await;
let register_response = reqwest::Client::new()
.post(format!("{gateway_url}/api/admin/proxy-nodes/register"))
.header(GATEWAY_HEADER, "rust-phase3b")
.bearer_auth(raw_token)
.json(&json!({
"name": "proxy-json-null",
"ip": "1.1.1.1",
"port": 0,
"heartbeat_interval": 30,
"tunnel_mode": true
}))
.send()
.await
.expect("request should succeed");
assert_eq!(register_response.status(), StatusCode::OK);
let register_payload: serde_json::Value = register_response
.json()
.await
.expect("json body should parse");
assert_eq!(register_payload["node"]["name"], "proxy-json-null");
assert_eq!(
register_payload["node"]["registered_by"],
json!(admin_user.id)
);
gateway_handle.abort();
}
#[tokio::test] #[tokio::test]
async fn gateway_creates_updates_and_tests_manual_proxy_nodes_locally() { async fn gateway_creates_updates_and_tests_manual_proxy_nodes_locally() {
let proxy_auths = Arc::new(Mutex::new(Vec::<Option<String>>::new())); let proxy_auths = Arc::new(Mutex::new(Vec::<Option<String>>::new()));