Address Windows network review feedback

This commit is contained in:
David Wiesen
2026-07-09 13:10:46 -07:00
parent 7f5bfee155
commit 54abd9e90d
7 changed files with 54 additions and 23 deletions
+1
View File
@@ -1372,6 +1372,7 @@ fn analytics_rpc_transport(transport: &AppServerTransport) -> AppServerRpcTransp
#[cfg(test)]
mod tests {
use super::LogFormat;
#[cfg(target_os = "windows")]
use super::is_unrecoverable_windows_network_config_error;
#[cfg(debug_assertions)]
use super::loader_overrides_with_test_user_config_file;
+7 -4
View File
@@ -756,10 +756,13 @@ impl WindowsRequirementsToml {
}
pub fn allows_only_elevated_sandbox(&self) -> bool {
matches!(
self.allowed_sandbox_implementations.as_deref(),
Some([WindowsSandboxModeToml::Elevated])
)
let Some(modes) = self.allowed_sandbox_implementations.as_deref() else {
return false;
};
!modes.is_empty()
&& modes
.iter()
.all(|&mode| mode == WindowsSandboxModeToml::Elevated)
}
}
@@ -212,22 +212,26 @@ pub(crate) fn push_xml_escaped_text(rendered: &mut String, value: &str) {
#[derive(Debug, Clone, PartialEq, Eq, Default)]
pub(crate) struct NetworkContext {
enabled: bool,
allowed_domains: Vec<String>,
denied_domains: Vec<String>,
}
impl NetworkContext {
pub(crate) fn new(allowed_domains: Vec<String>, denied_domains: Vec<String>) -> Self {
pub(crate) fn new(enabled: bool, allowed: Vec<String>, denied: Vec<String>) -> Self {
Self {
allowed_domains,
denied_domains,
enabled,
allowed_domains: allowed,
denied_domains: denied,
}
}
pub(super) fn render(&self) -> String {
let mut rendered = "<network enabled=\"true\">".to_string();
Self::push_rendered_domain_element(&mut rendered, "allowed", &self.allowed_domains);
Self::push_rendered_domain_element(&mut rendered, "denied", &self.denied_domains);
let mut rendered = format!("<network enabled=\"{}\">", self.enabled);
if self.enabled {
Self::push_rendered_domain_element(&mut rendered, "allowed", &self.allowed_domains);
Self::push_rendered_domain_element(&mut rendered, "denied", &self.denied_domains);
}
rendered.push_str("</network>");
rendered
}
@@ -369,6 +369,7 @@ fn network_from_turn_context(turn_context: &TurnContext) -> Option<NetworkContex
.as_ref()?;
Some(NetworkContext::new(
turn_context.network.is_some(),
network
.domains
.as_ref()
@@ -110,6 +110,7 @@ fn serialize_environment_context_with_foreign_windows_cwd() {
#[test]
fn serialize_environment_context_with_network() {
let network = NetworkContext::new(
true,
vec!["api.example.com".to_string(), "*.openai.com".to_string()],
vec!["blocked.example.com".to_string()],
);
@@ -117,6 +117,7 @@ fn persisted_turn_context_values_render_a_diff() -> Result<()> {
current_date: Some("2026-06-19".to_string()),
timezone: Some("UTC".to_string()),
network: Some(NetworkContext::new(
true,
vec!["old.example.com".to_string()],
vec![],
)),
@@ -131,6 +132,7 @@ fn persisted_turn_context_values_render_a_diff() -> Result<()> {
current_date: Some("2026-06-20".to_string()),
timezone: Some("America/Los_Angeles".to_string()),
network: Some(NetworkContext::new(
true,
vec!["new.example.com".to_string()],
vec!["blocked.example.com".to_string()],
)),
+32 -13
View File
@@ -8020,17 +8020,30 @@ async fn spawn_task_does_not_update_previous_turn_settings_for_non_run_turn_task
}
#[tokio::test]
async fn record_context_updates_emits_environment_item_for_network_changes() {
let (session, previous_context) = make_session_and_context().await;
let previous_context = Arc::new(previous_context);
let mut current_context = previous_context
.with_model(
previous_context.model_info.slug.clone(),
&session.services.models_manager,
async fn record_context_updates_emits_disabled_network_transition() {
let (session, mut previous_context) = make_session_and_context().await;
let permission_profile = PermissionProfile::workspace_write();
let mut network_config = NetworkProxyConfig::default();
network_config.enabled = true;
let network_spec = crate::config::NetworkProxySpec::from_config_and_constraints(
network_config,
/*requirements*/ None,
&permission_profile,
)
.expect("build network proxy spec");
let started_proxy = network_spec
.start_proxy(
&permission_profile,
/*policy_decider*/ None,
/*blocked_request_observer*/ None,
/*enable_network_approval_flow*/ false,
crate::config::NetworkProxyAuditMetadata::default(),
)
.await;
.await
.expect("start network proxy");
previous_context.network = Some(started_proxy.proxy());
let mut config = (*current_context.config).clone();
let mut config = (*previous_context.config).clone();
let mut requirements = config.config_layer_stack.requirements().clone();
requirements.network = Some(Sourced::new(
NetworkConstraints {
@@ -8065,7 +8078,15 @@ async fn record_context_updates_emits_environment_item_for_network_changes() {
config.config_layer_stack.requirements_toml().clone(),
)
.expect("rebuild config layer stack with network requirements");
current_context.config = Arc::new(config);
previous_context.config = Arc::new(config);
let previous_context = Arc::new(previous_context);
let mut current_context = previous_context
.with_model(
previous_context.model_info.slug.clone(),
&session.services.models_manager,
)
.await;
current_context.network = None;
let update_items =
record_context_update_items(&session, previous_context, current_context).await;
@@ -8074,9 +8095,7 @@ async fn record_context_updates_emits_environment_item_for_network_changes() {
.into_iter()
.find(|text| text.contains("<environment_context>"))
.expect("environment update item should be emitted");
assert!(environment_update.contains(
"<network enabled=\"true\"><allowed>api.example.com</allowed><denied>blocked.example.com</denied></network>"
));
assert!(environment_update.contains("<network enabled=\"false\"></network>"));
}
#[tokio::test]