mirror of
				https://github.com/yuezk/GlobalProtect-openconnect.git
				synced 2025-05-20 07:26:58 -04:00 
			
		
		
		
	upgrade gpauth
This commit is contained in:
		@@ -1,12 +1,13 @@
 | 
			
		||||
[package]
 | 
			
		||||
name = "gpauth"
 | 
			
		||||
rust-version.workspace = true
 | 
			
		||||
authors.workspace = true
 | 
			
		||||
version.workspace = true
 | 
			
		||||
edition.workspace = true
 | 
			
		||||
license.workspace = true
 | 
			
		||||
 | 
			
		||||
[build-dependencies]
 | 
			
		||||
tauri-build = { version = "1.5", features = [] }
 | 
			
		||||
tauri-build = { version = "2", features = [] }
 | 
			
		||||
 | 
			
		||||
[dependencies]
 | 
			
		||||
gpapi = { path = "../../crates/gpapi", features = [
 | 
			
		||||
@@ -14,6 +15,9 @@ gpapi = { path = "../../crates/gpapi", features = [
 | 
			
		||||
  "clap",
 | 
			
		||||
  "browser-auth",
 | 
			
		||||
] }
 | 
			
		||||
 | 
			
		||||
tauri = { workspace = true }
 | 
			
		||||
 | 
			
		||||
anyhow.workspace = true
 | 
			
		||||
clap.workspace = true
 | 
			
		||||
env_logger.workspace = true
 | 
			
		||||
@@ -24,6 +28,7 @@ tokio.workspace = true
 | 
			
		||||
tokio-util.workspace = true
 | 
			
		||||
tempfile.workspace = true
 | 
			
		||||
html-escape = "0.2.13"
 | 
			
		||||
webkit2gtk = "0.18.2"
 | 
			
		||||
tauri = { workspace = true, features = ["http-all"] }
 | 
			
		||||
compile-time.workspace = true
 | 
			
		||||
 | 
			
		||||
[target.'cfg(not(target_os = "macos"))'.dependencies]
 | 
			
		||||
webkit2gtk = "2"
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										108
									
								
								apps/gpauth/src/auth_messenger.rs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										108
									
								
								apps/gpauth/src/auth_messenger.rs
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,108 @@
 | 
			
		||||
use anyhow::bail;
 | 
			
		||||
use gpapi::auth::SamlAuthData;
 | 
			
		||||
use log::{error, info};
 | 
			
		||||
use tokio::sync::{mpsc, RwLock};
 | 
			
		||||
use tokio_util::sync::CancellationToken;
 | 
			
		||||
 | 
			
		||||
pub enum AuthError {
 | 
			
		||||
  /// Failed to load page due to TLS error
 | 
			
		||||
  TlsError,
 | 
			
		||||
  /// 1. Found auth data in headers/body but it's invalid
 | 
			
		||||
  /// 2. Loaded an empty page, failed to load page. etc.
 | 
			
		||||
  Invalid,
 | 
			
		||||
  /// No auth data found in headers/body
 | 
			
		||||
  NotFound,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub type AuthResult = anyhow::Result<SamlAuthData, AuthError>;
 | 
			
		||||
 | 
			
		||||
pub enum AuthEvent {
 | 
			
		||||
  Data(SamlAuthData),
 | 
			
		||||
  Error(AuthError),
 | 
			
		||||
  RaiseWindow,
 | 
			
		||||
  Close,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub struct AuthMessenger {
 | 
			
		||||
  tx: mpsc::UnboundedSender<AuthEvent>,
 | 
			
		||||
  rx: RwLock<mpsc::UnboundedReceiver<AuthEvent>>,
 | 
			
		||||
  raise_window_cancel_token: RwLock<Option<CancellationToken>>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl AuthMessenger {
 | 
			
		||||
  pub fn new() -> Self {
 | 
			
		||||
    let (tx, rx) = mpsc::unbounded_channel();
 | 
			
		||||
 | 
			
		||||
    Self {
 | 
			
		||||
      tx,
 | 
			
		||||
      rx: RwLock::new(rx),
 | 
			
		||||
      raise_window_cancel_token: Default::default(),
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  pub async fn subscribe(&self) -> anyhow::Result<AuthEvent> {
 | 
			
		||||
    let mut rx = self.rx.write().await;
 | 
			
		||||
    if let Some(event) = rx.recv().await {
 | 
			
		||||
      return Ok(event);
 | 
			
		||||
    }
 | 
			
		||||
    bail!("Failed to receive auth event");
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  pub fn send_auth_event(&self, event: AuthEvent) {
 | 
			
		||||
    if let Err(event) = self.tx.send(event) {
 | 
			
		||||
      error!("Failed to send auth event: {}", event);
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  pub fn send_auth_result(&self, result: AuthResult) {
 | 
			
		||||
    match result {
 | 
			
		||||
      Ok(data) => self.send_auth_data(data),
 | 
			
		||||
      Err(err) => self.send_auth_error(err),
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  pub fn send_auth_error(&self, err: AuthError) {
 | 
			
		||||
    self.send_auth_event(AuthEvent::Error(err));
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  pub fn send_auth_data(&self, data: SamlAuthData) {
 | 
			
		||||
    self.send_auth_event(AuthEvent::Data(data));
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  pub fn schedule_raise_window(&self, delay: u64) {
 | 
			
		||||
    let cancel_token = CancellationToken::new();
 | 
			
		||||
    let cancel_token_clone = cancel_token.clone();
 | 
			
		||||
 | 
			
		||||
    if let Ok(mut guard) = self.raise_window_cancel_token.try_write() {
 | 
			
		||||
      // Cancel the previous raise window task if it exists
 | 
			
		||||
      if let Some(token) = guard.take() {
 | 
			
		||||
        token.cancel();
 | 
			
		||||
      }
 | 
			
		||||
      *guard = Some(cancel_token_clone);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    let tx = self.tx.clone();
 | 
			
		||||
    tokio::spawn(async move {
 | 
			
		||||
      info!("Displaying the window in {} second(s)...", delay);
 | 
			
		||||
 | 
			
		||||
      tokio::select! {
 | 
			
		||||
        _ = tokio::time::sleep(tokio::time::Duration::from_secs(delay)) => {
 | 
			
		||||
          if let Err(err) = tx.send(AuthEvent::RaiseWindow) {
 | 
			
		||||
            error!("Failed to send raise window event: {}", err);
 | 
			
		||||
          }
 | 
			
		||||
        }
 | 
			
		||||
        _ = cancel_token.cancelled() => {
 | 
			
		||||
          info!("Cancelled raise window task");
 | 
			
		||||
        }
 | 
			
		||||
      }
 | 
			
		||||
    });
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  pub fn cancel_raise_window(&self) {
 | 
			
		||||
    if let Ok(mut cancel_token) = self.raise_window_cancel_token.try_write() {
 | 
			
		||||
      if let Some(token) = cancel_token.take() {
 | 
			
		||||
        token.cancel();
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
@@ -1,5 +1,8 @@
 | 
			
		||||
use std::{
 | 
			
		||||
  rc::Rc,
 | 
			
		||||
  borrow::Cow,
 | 
			
		||||
  env::temp_dir,
 | 
			
		||||
  fs,
 | 
			
		||||
  os::unix::fs::PermissionsExt,
 | 
			
		||||
  sync::Arc,
 | 
			
		||||
  time::{Duration, Instant},
 | 
			
		||||
};
 | 
			
		||||
@@ -7,517 +10,278 @@ use std::{
 | 
			
		||||
use anyhow::bail;
 | 
			
		||||
use gpapi::{
 | 
			
		||||
  auth::SamlAuthData,
 | 
			
		||||
  error::AuthDataParseError,
 | 
			
		||||
  error::PortalError,
 | 
			
		||||
  gp_params::GpParams,
 | 
			
		||||
  portal::{prelogin, Prelogin},
 | 
			
		||||
  utils::{redact::redact_uri, window::WindowExt},
 | 
			
		||||
  process::browser_authenticator::BrowserAuthenticator,
 | 
			
		||||
  utils::window::WindowExt,
 | 
			
		||||
  GP_CALLBACK_PORT_FILENAME,
 | 
			
		||||
};
 | 
			
		||||
use log::{info, warn};
 | 
			
		||||
use regex::Regex;
 | 
			
		||||
use tauri::{AppHandle, Window, WindowEvent, WindowUrl};
 | 
			
		||||
use tokio::sync::{mpsc, oneshot, RwLock};
 | 
			
		||||
use tokio_util::sync::CancellationToken;
 | 
			
		||||
use webkit2gtk::{
 | 
			
		||||
  gio::Cancellable,
 | 
			
		||||
  glib::{GString, TimeSpan},
 | 
			
		||||
  LoadEvent, SettingsExt, TLSErrorsPolicy, URIResponse, URIResponseExt, WebContextExt, WebResource, WebResourceExt,
 | 
			
		||||
  WebView, WebViewExt, WebsiteDataManagerExtManual, WebsiteDataTypes,
 | 
			
		||||
use tauri::{AppHandle, WebviewUrl, WebviewWindow, WindowEvent};
 | 
			
		||||
use tokio::{
 | 
			
		||||
  io::AsyncReadExt,
 | 
			
		||||
  net::TcpListener,
 | 
			
		||||
  sync::{oneshot, RwLock},
 | 
			
		||||
  time,
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
enum AuthDataError {
 | 
			
		||||
  /// Failed to load page due to TLS error
 | 
			
		||||
  TlsError,
 | 
			
		||||
  /// 1. Found auth data in headers/body but it's invalid
 | 
			
		||||
  /// 2. Loaded an empty page, failed to load page. etc.
 | 
			
		||||
  Invalid,
 | 
			
		||||
  /// No auth data found in headers/body
 | 
			
		||||
  NotFound,
 | 
			
		||||
}
 | 
			
		||||
use crate::{
 | 
			
		||||
  auth_messenger::{AuthError, AuthEvent, AuthMessenger},
 | 
			
		||||
  common::{AuthRequest, AuthSettings},
 | 
			
		||||
  platform_impl,
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
type AuthResult = Result<SamlAuthData, AuthDataError>;
 | 
			
		||||
 | 
			
		||||
pub(crate) struct AuthWindow<'a> {
 | 
			
		||||
  app_handle: AppHandle,
 | 
			
		||||
pub struct AuthWindow<'a> {
 | 
			
		||||
  server: &'a str,
 | 
			
		||||
  saml_request: &'a str,
 | 
			
		||||
  user_agent: &'a str,
 | 
			
		||||
  gp_params: Option<GpParams>,
 | 
			
		||||
  gp_params: &'a GpParams,
 | 
			
		||||
  auth_request: Option<&'a str>,
 | 
			
		||||
  clean: bool,
 | 
			
		||||
  is_retrying: RwLock<bool>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<'a> AuthWindow<'a> {
 | 
			
		||||
  pub fn new(app_handle: AppHandle) -> Self {
 | 
			
		||||
  pub fn new(server: &'a str, gp_params: &'a GpParams) -> Self {
 | 
			
		||||
    Self {
 | 
			
		||||
      app_handle,
 | 
			
		||||
      server: "",
 | 
			
		||||
      saml_request: "",
 | 
			
		||||
      user_agent: "",
 | 
			
		||||
      gp_params: None,
 | 
			
		||||
      server,
 | 
			
		||||
      gp_params,
 | 
			
		||||
      auth_request: None,
 | 
			
		||||
      clean: false,
 | 
			
		||||
      is_retrying: Default::default(),
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  pub fn server(mut self, server: &'a str) -> Self {
 | 
			
		||||
    self.server = server;
 | 
			
		||||
  pub fn with_auth_request(mut self, auth_request: &'a str) -> Self {
 | 
			
		||||
    if !auth_request.is_empty() {
 | 
			
		||||
      self.auth_request = Some(auth_request);
 | 
			
		||||
    }
 | 
			
		||||
    self
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  pub fn saml_request(mut self, saml_request: &'a str) -> Self {
 | 
			
		||||
    self.saml_request = saml_request;
 | 
			
		||||
    self
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  pub fn user_agent(mut self, user_agent: &'a str) -> Self {
 | 
			
		||||
    self.user_agent = user_agent;
 | 
			
		||||
    self
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  pub fn gp_params(mut self, gp_params: GpParams) -> Self {
 | 
			
		||||
    self.gp_params.replace(gp_params);
 | 
			
		||||
    self
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  pub fn clean(mut self, clean: bool) -> Self {
 | 
			
		||||
  pub fn with_clean(mut self, clean: bool) -> Self {
 | 
			
		||||
    self.clean = clean;
 | 
			
		||||
    self
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  pub async fn open(&self) -> anyhow::Result<SamlAuthData> {
 | 
			
		||||
    info!("Open auth window, user_agent: {}", self.user_agent);
 | 
			
		||||
  pub async fn browser_authenticate(&self, browser: Option<&str>) -> anyhow::Result<SamlAuthData> {
 | 
			
		||||
    let auth_request = self.initial_auth_request().await?;
 | 
			
		||||
    let browser_auth = if let Some(browser) = browser {
 | 
			
		||||
      BrowserAuthenticator::new_with_browser(&auth_request, browser)
 | 
			
		||||
    } else {
 | 
			
		||||
      BrowserAuthenticator::new(&auth_request)
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    let window = Window::builder(&self.app_handle, "auth_window", WindowUrl::default())
 | 
			
		||||
    browser_auth.authenticate()?;
 | 
			
		||||
    info!("Please continue the authentication process in the default browser");
 | 
			
		||||
 | 
			
		||||
    wait_auth_data().await
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  pub async fn webview_authenticate(&self, app_handle: &AppHandle) -> anyhow::Result<SamlAuthData> {
 | 
			
		||||
    let auth_window = WebviewWindow::builder(app_handle, "auth_window", WebviewUrl::default())
 | 
			
		||||
      .title("GlobalProtect Login")
 | 
			
		||||
      // .user_agent(self.user_agent)
 | 
			
		||||
      .focused(true)
 | 
			
		||||
      .visible(false)
 | 
			
		||||
      .center()
 | 
			
		||||
      .build()?;
 | 
			
		||||
 | 
			
		||||
    let window = Arc::new(window);
 | 
			
		||||
 | 
			
		||||
    let cancel_token = CancellationToken::new();
 | 
			
		||||
    let cancel_token_clone = cancel_token.clone();
 | 
			
		||||
 | 
			
		||||
    window.on_window_event(move |event| {
 | 
			
		||||
      if let WindowEvent::CloseRequested { .. } = event {
 | 
			
		||||
        cancel_token_clone.cancel();
 | 
			
		||||
      }
 | 
			
		||||
    });
 | 
			
		||||
 | 
			
		||||
    let window_clone = Arc::clone(&window);
 | 
			
		||||
    let timeout_secs = 15;
 | 
			
		||||
    tokio::spawn(async move {
 | 
			
		||||
      tokio::time::sleep(Duration::from_secs(timeout_secs)).await;
 | 
			
		||||
      let visible = window_clone.is_visible().unwrap_or(false);
 | 
			
		||||
      if !visible {
 | 
			
		||||
        info!("Try to raise auth window after {} seconds", timeout_secs);
 | 
			
		||||
        raise_window(&window_clone);
 | 
			
		||||
      }
 | 
			
		||||
    });
 | 
			
		||||
 | 
			
		||||
    tokio::select! {
 | 
			
		||||
      _ = cancel_token.cancelled() => {
 | 
			
		||||
        bail!("Auth cancelled");
 | 
			
		||||
      }
 | 
			
		||||
      saml_result = self.auth_loop(&window) => {
 | 
			
		||||
        window.close()?;
 | 
			
		||||
        saml_result
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
    self.auth_loop(&auth_window).await
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  async fn auth_loop(&self, window: &Arc<Window>) -> anyhow::Result<SamlAuthData> {
 | 
			
		||||
    let saml_request = self.saml_request.to_string();
 | 
			
		||||
    let (auth_result_tx, mut auth_result_rx) = mpsc::unbounded_channel::<AuthResult>();
 | 
			
		||||
    let raise_window_cancel_token: Arc<RwLock<Option<CancellationToken>>> = Default::default();
 | 
			
		||||
    let gp_params = self.gp_params.as_ref().unwrap();
 | 
			
		||||
    let tls_err_policy = if gp_params.ignore_tls_errors() {
 | 
			
		||||
      TLSErrorsPolicy::Ignore
 | 
			
		||||
    } else {
 | 
			
		||||
      TLSErrorsPolicy::Fail
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
  async fn auth_loop(&self, auth_window: &WebviewWindow) -> anyhow::Result<SamlAuthData> {
 | 
			
		||||
    if self.clean {
 | 
			
		||||
      clear_webview_cookies(window).await?;
 | 
			
		||||
      self.clear_webview_data(&auth_window).await?;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    let raise_window_cancel_token_clone = Arc::clone(&raise_window_cancel_token);
 | 
			
		||||
    window.with_webview(move |wv| {
 | 
			
		||||
      let wv = wv.inner();
 | 
			
		||||
 | 
			
		||||
      if let Some(context) = wv.context() {
 | 
			
		||||
        context.set_tls_errors_policy(tls_err_policy);
 | 
			
		||||
      }
 | 
			
		||||
 | 
			
		||||
      if let Some(settings) = wv.settings() {
 | 
			
		||||
        let ua = settings.user_agent().unwrap_or("".into());
 | 
			
		||||
        info!("Auth window user agent: {}", ua);
 | 
			
		||||
      }
 | 
			
		||||
 | 
			
		||||
      // Load the initial SAML request
 | 
			
		||||
      load_saml_request(&wv, &saml_request);
 | 
			
		||||
 | 
			
		||||
      let auth_result_tx_clone = auth_result_tx.clone();
 | 
			
		||||
      wv.connect_load_changed(move |wv, event| {
 | 
			
		||||
        if event == LoadEvent::Started {
 | 
			
		||||
          let Ok(mut cancel_token) = raise_window_cancel_token_clone.try_write() else {
 | 
			
		||||
            return;
 | 
			
		||||
          };
 | 
			
		||||
 | 
			
		||||
          // Cancel the raise window task
 | 
			
		||||
          if let Some(cancel_token) = cancel_token.take() {
 | 
			
		||||
            cancel_token.cancel();
 | 
			
		||||
          }
 | 
			
		||||
          return;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        if event != LoadEvent::Finished {
 | 
			
		||||
          return;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        if let Some(main_resource) = wv.main_resource() {
 | 
			
		||||
          let uri = main_resource.uri().unwrap_or("".into());
 | 
			
		||||
 | 
			
		||||
          if uri.is_empty() {
 | 
			
		||||
            warn!("Loaded an empty uri");
 | 
			
		||||
            send_auth_result(&auth_result_tx_clone, Err(AuthDataError::Invalid));
 | 
			
		||||
            return;
 | 
			
		||||
          }
 | 
			
		||||
 | 
			
		||||
          info!("Loaded uri: {}", redact_uri(&uri));
 | 
			
		||||
          if uri.starts_with("globalprotectcallback:") {
 | 
			
		||||
            return;
 | 
			
		||||
          }
 | 
			
		||||
 | 
			
		||||
          read_auth_data(&main_resource, auth_result_tx_clone.clone());
 | 
			
		||||
        }
 | 
			
		||||
      });
 | 
			
		||||
 | 
			
		||||
      let auth_result_tx_clone = auth_result_tx.clone();
 | 
			
		||||
      wv.connect_load_failed_with_tls_errors(move |_wv, uri, cert, err| {
 | 
			
		||||
        let redacted_uri = redact_uri(uri);
 | 
			
		||||
        warn!(
 | 
			
		||||
          "Failed to load uri: {} with error: {}, cert: {}",
 | 
			
		||||
          redacted_uri, err, cert
 | 
			
		||||
        );
 | 
			
		||||
 | 
			
		||||
        send_auth_result(&auth_result_tx_clone, Err(AuthDataError::TlsError));
 | 
			
		||||
        true
 | 
			
		||||
      });
 | 
			
		||||
 | 
			
		||||
      wv.connect_load_failed(move |_wv, _event, uri, err| {
 | 
			
		||||
        let redacted_uri = redact_uri(uri);
 | 
			
		||||
        if !uri.starts_with("globalprotectcallback:") {
 | 
			
		||||
          warn!("Failed to load uri: {} with error: {}", redacted_uri, err);
 | 
			
		||||
        }
 | 
			
		||||
        // NOTE: Don't send error here, since load_changed event will be triggered after this
 | 
			
		||||
        // send_auth_result(&auth_result_tx, Err(AuthDataError::Invalid));
 | 
			
		||||
        // true to stop other handlers from being invoked for the event. false to propagate the event further.
 | 
			
		||||
        true
 | 
			
		||||
      });
 | 
			
		||||
    })?;
 | 
			
		||||
 | 
			
		||||
    let portal = self.server.to_string();
 | 
			
		||||
    let auth_messenger = self.setup_auth_window(&auth_window).await?;
 | 
			
		||||
 | 
			
		||||
    loop {
 | 
			
		||||
      if let Some(auth_result) = auth_result_rx.recv().await {
 | 
			
		||||
        match auth_result {
 | 
			
		||||
          Ok(auth_data) => return Ok(auth_data),
 | 
			
		||||
          Err(AuthDataError::TlsError) => bail!("TLS error: certificate verify failed"),
 | 
			
		||||
          Err(AuthDataError::NotFound) => {
 | 
			
		||||
            info!("No auth data found, it may not be the /SAML20/SP/ACS endpoint");
 | 
			
		||||
 | 
			
		||||
            // The user may need to interact with the auth window, raise it in 3 seconds
 | 
			
		||||
            if !window.is_visible().unwrap_or(false) {
 | 
			
		||||
              let window = Arc::clone(window);
 | 
			
		||||
              let cancel_token = CancellationToken::new();
 | 
			
		||||
 | 
			
		||||
              raise_window_cancel_token.write().await.replace(cancel_token.clone());
 | 
			
		||||
 | 
			
		||||
              tokio::spawn(async move {
 | 
			
		||||
                let delay_secs = 1;
 | 
			
		||||
 | 
			
		||||
                info!("Raise window in {} second(s)", delay_secs);
 | 
			
		||||
                tokio::select! {
 | 
			
		||||
                  _ = tokio::time::sleep(Duration::from_secs(delay_secs)) => {
 | 
			
		||||
                    raise_window(&window);
 | 
			
		||||
                  }
 | 
			
		||||
                  _ = cancel_token.cancelled() => {
 | 
			
		||||
                    info!("Raise window cancelled");
 | 
			
		||||
                  }
 | 
			
		||||
                }
 | 
			
		||||
              });
 | 
			
		||||
            }
 | 
			
		||||
          }
 | 
			
		||||
          Err(AuthDataError::Invalid) => {
 | 
			
		||||
            info!("Got invalid auth data, retrying...");
 | 
			
		||||
 | 
			
		||||
            window.with_webview(|wv| {
 | 
			
		||||
              let wv = wv.inner();
 | 
			
		||||
              wv.run_javascript(r#"
 | 
			
		||||
                  var loading = document.createElement("div");
 | 
			
		||||
                  loading.innerHTML = '<div style="position: absolute; width: 100%; text-align: center; font-size: 20px; font-weight: bold; top: 50%; left: 50%; transform: translate(-50%, -50%);">Got invalid token, retrying...</div>';
 | 
			
		||||
                  loading.style = "position: fixed; top: 0; left: 0; width: 100%; height: 100%; background: rgba(255, 255, 255, 0.85); z-index: 99999;";
 | 
			
		||||
                  document.body.appendChild(loading);
 | 
			
		||||
              "#,
 | 
			
		||||
                  Cancellable::NONE,
 | 
			
		||||
                  |_| info!("Injected loading element successfully"),
 | 
			
		||||
              );
 | 
			
		||||
            })?;
 | 
			
		||||
 | 
			
		||||
            let saml_request = portal_prelogin(&portal, gp_params).await?;
 | 
			
		||||
            window.with_webview(move |wv| {
 | 
			
		||||
              let wv = wv.inner();
 | 
			
		||||
              load_saml_request(&wv, &saml_request);
 | 
			
		||||
            })?;
 | 
			
		||||
          }
 | 
			
		||||
      match auth_messenger.subscribe().await? {
 | 
			
		||||
        AuthEvent::Close => bail!("Authentication cancelled"),
 | 
			
		||||
        AuthEvent::RaiseWindow => self.raise_window(auth_window),
 | 
			
		||||
        AuthEvent::Error(AuthError::TlsError) => bail!(PortalError::TlsError),
 | 
			
		||||
        AuthEvent::Error(AuthError::NotFound) => self.handle_not_found(auth_window, &auth_messenger),
 | 
			
		||||
        AuthEvent::Error(AuthError::Invalid) => self.retry_auth(auth_window).await,
 | 
			
		||||
        AuthEvent::Data(auth_data) => {
 | 
			
		||||
          auth_window.close()?;
 | 
			
		||||
          return Ok(auth_data);
 | 
			
		||||
        }
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn raise_window(window: &Arc<Window>) {
 | 
			
		||||
  let visible = window.is_visible().unwrap_or(false);
 | 
			
		||||
  if !visible {
 | 
			
		||||
    if let Err(err) = window.raise() {
 | 
			
		||||
  async fn initial_auth_request(&self) -> anyhow::Result<Cow<'a, str>> {
 | 
			
		||||
    if let Some(auth_request) = self.auth_request {
 | 
			
		||||
      return Ok(Cow::Borrowed(auth_request));
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    let auth_request = portal_prelogin(&self.server, &self.gp_params).await?;
 | 
			
		||||
    Ok(Cow::Owned(auth_request))
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  async fn clear_webview_data(&self, auth_window: &WebviewWindow) -> anyhow::Result<()> {
 | 
			
		||||
    info!("Clearing webview data...");
 | 
			
		||||
 | 
			
		||||
    let (tx, rx) = oneshot::channel::<anyhow::Result<()>>();
 | 
			
		||||
    let now = Instant::now();
 | 
			
		||||
    auth_window.with_webview(|webview| {
 | 
			
		||||
      platform_impl::clear_data(&webview.inner(), |result| {
 | 
			
		||||
        if let Err(result) = tx.send(result) {
 | 
			
		||||
          warn!("Failed to send clear data result: {:?}", result);
 | 
			
		||||
        }
 | 
			
		||||
      })
 | 
			
		||||
    })?;
 | 
			
		||||
 | 
			
		||||
    rx.await??;
 | 
			
		||||
    info!("Webview data cleared in {:?}", now.elapsed());
 | 
			
		||||
 | 
			
		||||
    Ok(())
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  async fn setup_auth_window(&self, auth_window: &WebviewWindow) -> anyhow::Result<Arc<AuthMessenger>> {
 | 
			
		||||
    info!("Setting up auth window...");
 | 
			
		||||
 | 
			
		||||
    let auth_messenger = Arc::new(AuthMessenger::new());
 | 
			
		||||
    let auth_request = self.initial_auth_request().await?.into_owned();
 | 
			
		||||
    let ignore_tls_errors = self.gp_params.ignore_tls_errors();
 | 
			
		||||
 | 
			
		||||
    // Handle window close event
 | 
			
		||||
    let auth_messenger_clone = Arc::clone(&auth_messenger);
 | 
			
		||||
    auth_window.on_window_event(move |event| {
 | 
			
		||||
      if let WindowEvent::CloseRequested { .. } = event {
 | 
			
		||||
        auth_messenger_clone.send_auth_event(AuthEvent::Close);
 | 
			
		||||
      }
 | 
			
		||||
    });
 | 
			
		||||
 | 
			
		||||
    // Show the window after 10 seconds, so that the user can see the window if the auth process is stuck
 | 
			
		||||
    let auth_messenger_clone = Arc::clone(&auth_messenger);
 | 
			
		||||
    tokio::spawn(async move {
 | 
			
		||||
      time::sleep(Duration::from_secs(10)).await;
 | 
			
		||||
      auth_messenger_clone.send_auth_event(AuthEvent::RaiseWindow);
 | 
			
		||||
    });
 | 
			
		||||
 | 
			
		||||
    // setup webview
 | 
			
		||||
    let auth_messenger_clone = Arc::clone(&auth_messenger);
 | 
			
		||||
    let (tx, rx) = oneshot::channel::<anyhow::Result<()>>();
 | 
			
		||||
 | 
			
		||||
    auth_window.with_webview(move |webview| {
 | 
			
		||||
      let auth_settings = AuthSettings {
 | 
			
		||||
        auth_request: AuthRequest::new(&auth_request),
 | 
			
		||||
        auth_messenger: auth_messenger_clone,
 | 
			
		||||
        ignore_tls_errors,
 | 
			
		||||
      };
 | 
			
		||||
 | 
			
		||||
      let result = platform_impl::setup_webview(&webview.inner(), auth_settings);
 | 
			
		||||
      if let Err(result) = tx.send(result) {
 | 
			
		||||
        warn!("Failed to send setup auth window result: {:?}", result);
 | 
			
		||||
      }
 | 
			
		||||
    })?;
 | 
			
		||||
 | 
			
		||||
    rx.await??;
 | 
			
		||||
    info!("Auth window setup completed");
 | 
			
		||||
 | 
			
		||||
    Ok(auth_messenger)
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  fn handle_not_found(&self, auth_window: &WebviewWindow, auth_messenger: &Arc<AuthMessenger>) {
 | 
			
		||||
    info!("No auth data found, it may not be the /SAML20/SP/ACS endpoint");
 | 
			
		||||
 | 
			
		||||
    let visible = auth_window.is_visible().unwrap_or(false);
 | 
			
		||||
    if visible {
 | 
			
		||||
      return;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    auth_messenger.schedule_raise_window(1);
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  async fn retry_auth(&self, auth_window: &WebviewWindow) {
 | 
			
		||||
    let mut is_retrying = self.is_retrying.write().await;
 | 
			
		||||
    if *is_retrying {
 | 
			
		||||
      info!("Already retrying authentication, skipping...");
 | 
			
		||||
      return;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    *is_retrying = true;
 | 
			
		||||
    drop(is_retrying);
 | 
			
		||||
 | 
			
		||||
    if let Err(err) = self.retry_auth_impl(auth_window).await {
 | 
			
		||||
      warn!("Failed to retry authentication: {}", err);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    *self.is_retrying.write().await = false;
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  async fn retry_auth_impl(&self, auth_window: &WebviewWindow) -> anyhow::Result<()> {
 | 
			
		||||
    info!("Retrying authentication...");
 | 
			
		||||
 | 
			
		||||
    auth_window.eval( r#"
 | 
			
		||||
      var loading = document.createElement("div");
 | 
			
		||||
      loading.innerHTML = '<div style="position: absolute; width: 100%; text-align: center; font-size: 20px; font-weight: bold; top: 50%; left: 50%; transform: translate(-50%, -50%);">Got invalid token, retrying...</div>';
 | 
			
		||||
      loading.style = "position: fixed; top: 0; left: 0; width: 100%; height: 100%; background: rgba(255, 255, 255, 0.85); z-index: 99999;";
 | 
			
		||||
      document.body.appendChild(loading);
 | 
			
		||||
    "#)?;
 | 
			
		||||
 | 
			
		||||
    let auth_request = portal_prelogin(&self.server, &self.gp_params).await?;
 | 
			
		||||
    let (tx, rx) = oneshot::channel::<()>();
 | 
			
		||||
    auth_window.with_webview(move |webview| {
 | 
			
		||||
      let auth_request = AuthRequest::new(&auth_request);
 | 
			
		||||
      platform_impl::load_auth_request(&webview.inner(), &auth_request);
 | 
			
		||||
 | 
			
		||||
      tx.send(()).expect("Failed to send message to the channel")
 | 
			
		||||
    })?;
 | 
			
		||||
 | 
			
		||||
    rx.await?;
 | 
			
		||||
    Ok(())
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  fn raise_window(&self, auth_window: &WebviewWindow) {
 | 
			
		||||
    let visible = auth_window.is_visible().unwrap_or(false);
 | 
			
		||||
    if visible {
 | 
			
		||||
      return;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    info!("Raising auth window...");
 | 
			
		||||
    if let Err(err) = auth_window.raise() {
 | 
			
		||||
      warn!("Failed to raise window: {}", err);
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub async fn portal_prelogin(portal: &str, gp_params: &GpParams) -> anyhow::Result<String> {
 | 
			
		||||
async fn portal_prelogin(portal: &str, gp_params: &GpParams) -> anyhow::Result<String> {
 | 
			
		||||
  match prelogin(portal, gp_params).await? {
 | 
			
		||||
    Prelogin::Saml(prelogin) => Ok(prelogin.saml_request().to_string()),
 | 
			
		||||
    Prelogin::Standard(_) => bail!("Received non-SAML prelogin response"),
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn send_auth_result(auth_result_tx: &mpsc::UnboundedSender<AuthResult>, auth_result: AuthResult) {
 | 
			
		||||
  if let Err(err) = auth_result_tx.send(auth_result) {
 | 
			
		||||
    warn!("Failed to send auth event: {}", err);
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn load_saml_request(wv: &Rc<WebView>, saml_request: &str) {
 | 
			
		||||
  if saml_request.starts_with("http") {
 | 
			
		||||
    info!("Load the SAML request as URI...");
 | 
			
		||||
    wv.load_uri(saml_request);
 | 
			
		||||
  } else {
 | 
			
		||||
    info!("Load the SAML request as HTML...");
 | 
			
		||||
    wv.load_html(saml_request, None);
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn read_auth_data_from_headers(response: &URIResponse) -> AuthResult {
 | 
			
		||||
  response.http_headers().map_or_else(
 | 
			
		||||
    || {
 | 
			
		||||
      info!("No headers found in response");
 | 
			
		||||
      Err(AuthDataError::NotFound)
 | 
			
		||||
    },
 | 
			
		||||
    |mut headers| match headers.get("saml-auth-status") {
 | 
			
		||||
      Some(status) if status == "1" => {
 | 
			
		||||
        let username = headers.get("saml-username").map(GString::into);
 | 
			
		||||
        let prelogin_cookie = headers.get("prelogin-cookie").map(GString::into);
 | 
			
		||||
        let portal_userauthcookie = headers.get("portal-userauthcookie").map(GString::into);
 | 
			
		||||
 | 
			
		||||
        if SamlAuthData::check(&username, &prelogin_cookie, &portal_userauthcookie) {
 | 
			
		||||
          return Ok(SamlAuthData::new(
 | 
			
		||||
            username.unwrap(),
 | 
			
		||||
            prelogin_cookie,
 | 
			
		||||
            portal_userauthcookie,
 | 
			
		||||
          ));
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        info!("Found invalid auth data in headers");
 | 
			
		||||
        Err(AuthDataError::Invalid)
 | 
			
		||||
      }
 | 
			
		||||
      Some(status) => {
 | 
			
		||||
        info!("Found invalid SAML status: {} in headers", status);
 | 
			
		||||
        Err(AuthDataError::Invalid)
 | 
			
		||||
      }
 | 
			
		||||
      None => {
 | 
			
		||||
        info!("No saml-auth-status header found");
 | 
			
		||||
        Err(AuthDataError::NotFound)
 | 
			
		||||
      }
 | 
			
		||||
    },
 | 
			
		||||
  )
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn read_auth_data_from_body<F>(main_resource: &WebResource, callback: F)
 | 
			
		||||
where
 | 
			
		||||
  F: FnOnce(Result<SamlAuthData, AuthDataParseError>) + Send + 'static,
 | 
			
		||||
{
 | 
			
		||||
  main_resource.data(Cancellable::NONE, |data| match data {
 | 
			
		||||
    Ok(data) => {
 | 
			
		||||
      let html = String::from_utf8_lossy(&data);
 | 
			
		||||
      callback(read_auth_data_from_html(&html));
 | 
			
		||||
    }
 | 
			
		||||
    Err(err) => {
 | 
			
		||||
      info!("Failed to read response body: {}", err);
 | 
			
		||||
      callback(Err(AuthDataParseError::Invalid))
 | 
			
		||||
    }
 | 
			
		||||
  });
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn read_auth_data_from_html(html: &str) -> Result<SamlAuthData, AuthDataParseError> {
 | 
			
		||||
  if html.contains("Temporarily Unavailable") {
 | 
			
		||||
    info!("Found 'Temporarily Unavailable' in HTML, auth failed");
 | 
			
		||||
    return Err(AuthDataParseError::Invalid);
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  SamlAuthData::from_html(html).or_else(|err| {
 | 
			
		||||
    if let Some(gpcallback) = extract_gpcallback(html) {
 | 
			
		||||
      info!("Found gpcallback from html...");
 | 
			
		||||
      SamlAuthData::from_gpcallback(&gpcallback)
 | 
			
		||||
    } else {
 | 
			
		||||
      Err(err)
 | 
			
		||||
    }
 | 
			
		||||
  })
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn extract_gpcallback(html: &str) -> Option<String> {
 | 
			
		||||
  let re = Regex::new(r#"globalprotectcallback:[^"]+"#).unwrap();
 | 
			
		||||
  re.captures(html)
 | 
			
		||||
    .and_then(|captures| captures.get(0))
 | 
			
		||||
    .map(|m| html_escape::decode_html_entities(m.as_str()).to_string())
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn read_auth_data(main_resource: &WebResource, auth_result_tx: mpsc::UnboundedSender<AuthResult>) {
 | 
			
		||||
  let Some(response) = main_resource.response() else {
 | 
			
		||||
    info!("No response found in main resource");
 | 
			
		||||
    send_auth_result(&auth_result_tx, Err(AuthDataError::Invalid));
 | 
			
		||||
    return;
 | 
			
		||||
  };
 | 
			
		||||
 | 
			
		||||
  info!("Trying to read auth data from response headers...");
 | 
			
		||||
 | 
			
		||||
  match read_auth_data_from_headers(&response) {
 | 
			
		||||
    Ok(auth_data) => {
 | 
			
		||||
      info!("Got auth data from headers");
 | 
			
		||||
      send_auth_result(&auth_result_tx, Ok(auth_data));
 | 
			
		||||
    }
 | 
			
		||||
    Err(AuthDataError::Invalid) => {
 | 
			
		||||
      info!("Found invalid auth data in headers, trying to read from body...");
 | 
			
		||||
      read_auth_data_from_body(main_resource, move |auth_result| {
 | 
			
		||||
        // Since we have already found invalid auth data in headers, which means this could be the `/SAML20/SP/ACS` endpoint
 | 
			
		||||
        // any error result from body should be considered as invalid, and trigger a retry
 | 
			
		||||
        let auth_result = auth_result.map_err(|err| {
 | 
			
		||||
          info!("Failed to read auth data from body: {}", err);
 | 
			
		||||
          AuthDataError::Invalid
 | 
			
		||||
        });
 | 
			
		||||
        send_auth_result(&auth_result_tx, auth_result);
 | 
			
		||||
      });
 | 
			
		||||
    }
 | 
			
		||||
    Err(AuthDataError::NotFound) => {
 | 
			
		||||
      info!("No auth data found in headers, trying to read from body...");
 | 
			
		||||
 | 
			
		||||
      let is_acs_endpoint = main_resource.uri().map_or(false, |uri| uri.contains("/SAML20/SP/ACS"));
 | 
			
		||||
 | 
			
		||||
      read_auth_data_from_body(main_resource, move |auth_result| {
 | 
			
		||||
        // If the endpoint is `/SAML20/SP/ACS` and no auth data found in body, it should be considered as invalid
 | 
			
		||||
        let auth_result = auth_result.map_err(|err| {
 | 
			
		||||
          info!("Failed to read auth data from body: {}", err);
 | 
			
		||||
 | 
			
		||||
          if !is_acs_endpoint && matches!(err, AuthDataParseError::NotFound) {
 | 
			
		||||
            AuthDataError::NotFound
 | 
			
		||||
          } else {
 | 
			
		||||
            AuthDataError::Invalid
 | 
			
		||||
          }
 | 
			
		||||
        });
 | 
			
		||||
 | 
			
		||||
        send_auth_result(&auth_result_tx, auth_result)
 | 
			
		||||
      });
 | 
			
		||||
    }
 | 
			
		||||
    Err(AuthDataError::TlsError) => {
 | 
			
		||||
      // NOTE: This is unreachable
 | 
			
		||||
      info!("TLS error found in headers, trying to read from body...");
 | 
			
		||||
      send_auth_result(&auth_result_tx, Err(AuthDataError::TlsError));
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub(crate) async fn clear_webview_cookies(window: &Window) -> anyhow::Result<()> {
 | 
			
		||||
  let (tx, rx) = oneshot::channel::<Result<(), String>>();
 | 
			
		||||
 | 
			
		||||
  window.with_webview(|wv| {
 | 
			
		||||
    let send_result = move |result: Result<(), String>| {
 | 
			
		||||
      if let Err(err) = tx.send(result) {
 | 
			
		||||
        info!("Failed to send result: {:?}", err);
 | 
			
		||||
      }
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    let wv = wv.inner();
 | 
			
		||||
    let context = match wv.context() {
 | 
			
		||||
      Some(context) => context,
 | 
			
		||||
      None => {
 | 
			
		||||
        send_result(Err("No webview context found".into()));
 | 
			
		||||
        return;
 | 
			
		||||
      }
 | 
			
		||||
    };
 | 
			
		||||
    let data_manager = match context.website_data_manager() {
 | 
			
		||||
      Some(manager) => manager,
 | 
			
		||||
      None => {
 | 
			
		||||
        send_result(Err("No data manager found".into()));
 | 
			
		||||
        return;
 | 
			
		||||
      }
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    let now = Instant::now();
 | 
			
		||||
    data_manager.clear(
 | 
			
		||||
      WebsiteDataTypes::COOKIES,
 | 
			
		||||
      TimeSpan(0),
 | 
			
		||||
      Cancellable::NONE,
 | 
			
		||||
      move |result| match result {
 | 
			
		||||
        Err(err) => {
 | 
			
		||||
          send_result(Err(err.to_string()));
 | 
			
		||||
        }
 | 
			
		||||
        Ok(_) => {
 | 
			
		||||
          info!("Cookies cleared in {} ms", now.elapsed().as_millis());
 | 
			
		||||
          send_result(Ok(()));
 | 
			
		||||
        }
 | 
			
		||||
      },
 | 
			
		||||
    );
 | 
			
		||||
  })?;
 | 
			
		||||
 | 
			
		||||
  rx.await?.map_err(|err| anyhow::anyhow!(err))
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[cfg(test)]
 | 
			
		||||
mod tests {
 | 
			
		||||
  use super::*;
 | 
			
		||||
 | 
			
		||||
  #[test]
 | 
			
		||||
  fn extract_gpcallback_some() {
 | 
			
		||||
    let html = r#"
 | 
			
		||||
      <meta http-equiv="refresh" content="0; URL=globalprotectcallback:PGh0bWw+PCEtLSA8c">
 | 
			
		||||
      <meta http-equiv="refresh" content="0; URL=globalprotectcallback:PGh0bWw+PCEtLSA8c">
 | 
			
		||||
    "#;
 | 
			
		||||
 | 
			
		||||
    assert_eq!(
 | 
			
		||||
      extract_gpcallback(html).as_deref(),
 | 
			
		||||
      Some("globalprotectcallback:PGh0bWw+PCEtLSA8c")
 | 
			
		||||
    );
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  #[test]
 | 
			
		||||
  fn extract_gpcallback_cas() {
 | 
			
		||||
    let html = r#"
 | 
			
		||||
      <meta http-equiv="refresh" content="0; URL=globalprotectcallback:cas-as=1&un=xyz@email.com&token=very_long_string">
 | 
			
		||||
    "#;
 | 
			
		||||
 | 
			
		||||
    assert_eq!(
 | 
			
		||||
      extract_gpcallback(html).as_deref(),
 | 
			
		||||
      Some("globalprotectcallback:cas-as=1&un=xyz@email.com&token=very_long_string")
 | 
			
		||||
    );
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  #[test]
 | 
			
		||||
  fn extract_gpcallback_none() {
 | 
			
		||||
    let html = r#"
 | 
			
		||||
      <meta http-equiv="refresh" content="0; URL=PGh0bWw+PCEtLSA8c">
 | 
			
		||||
    "#;
 | 
			
		||||
 | 
			
		||||
    assert_eq!(extract_gpcallback(html), None);
 | 
			
		||||
  }
 | 
			
		||||
async fn wait_auth_data() -> anyhow::Result<SamlAuthData> {
 | 
			
		||||
  // Start a local server to receive the browser authentication data
 | 
			
		||||
  let listener = TcpListener::bind("127.0.0.1:0").await?;
 | 
			
		||||
  let port = listener.local_addr()?.port();
 | 
			
		||||
  let port_file = temp_dir().join(GP_CALLBACK_PORT_FILENAME);
 | 
			
		||||
 | 
			
		||||
  // Write the port to a file
 | 
			
		||||
  fs::write(&port_file, port.to_string())?;
 | 
			
		||||
  fs::set_permissions(&port_file, fs::Permissions::from_mode(0o600))?;
 | 
			
		||||
 | 
			
		||||
  // Remove the previous log file
 | 
			
		||||
  let callback_log = temp_dir().join("gpcallback.log");
 | 
			
		||||
  let _ = fs::remove_file(&callback_log);
 | 
			
		||||
 | 
			
		||||
  info!("Listening authentication data on port {}", port);
 | 
			
		||||
  info!(
 | 
			
		||||
    "If it hangs, please check the logs at `{}` for more information",
 | 
			
		||||
    callback_log.display()
 | 
			
		||||
  );
 | 
			
		||||
  let (mut socket, _) = listener.accept().await?;
 | 
			
		||||
 | 
			
		||||
  info!("Received the browser authentication data from the socket");
 | 
			
		||||
  let mut data = String::new();
 | 
			
		||||
  socket.read_to_string(&mut data).await?;
 | 
			
		||||
 | 
			
		||||
  // Remove the port file
 | 
			
		||||
  fs::remove_file(&port_file)?;
 | 
			
		||||
 | 
			
		||||
  let auth_data = SamlAuthData::from_gpcallback(&data)?;
 | 
			
		||||
  Ok(auth_data)
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -1,21 +1,16 @@
 | 
			
		||||
use std::{env::temp_dir, fs, os::unix::fs::PermissionsExt};
 | 
			
		||||
 | 
			
		||||
use clap::Parser;
 | 
			
		||||
use gpapi::{
 | 
			
		||||
  auth::{SamlAuthData, SamlAuthResult},
 | 
			
		||||
  clap::args::Os,
 | 
			
		||||
  clap::{args::Os, handle_error, Args},
 | 
			
		||||
  gp_params::{ClientOs, GpParams},
 | 
			
		||||
  process::browser_authenticator::BrowserAuthenticator,
 | 
			
		||||
  utils::{env_utils, normalize_server, openssl},
 | 
			
		||||
  GP_USER_AGENT,
 | 
			
		||||
};
 | 
			
		||||
use gpauth::auth_window::AuthWindow;
 | 
			
		||||
use log::{info, LevelFilter};
 | 
			
		||||
use serde_json::json;
 | 
			
		||||
use tauri::{App, AppHandle, RunEvent};
 | 
			
		||||
use tauri::RunEvent;
 | 
			
		||||
use tempfile::NamedTempFile;
 | 
			
		||||
use tokio::{io::AsyncReadExt, net::TcpListener};
 | 
			
		||||
 | 
			
		||||
use crate::auth_window::{portal_prelogin, AuthWindow};
 | 
			
		||||
 | 
			
		||||
const VERSION: &str = concat!(env!("CARGO_PKG_VERSION"), " (", compile_time::date_str!(), ")");
 | 
			
		||||
 | 
			
		||||
@@ -78,65 +73,17 @@ struct Cli {
 | 
			
		||||
  browser: Option<String>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Cli {
 | 
			
		||||
  async fn run(&mut self) -> anyhow::Result<()> {
 | 
			
		||||
    if self.ignore_tls_errors {
 | 
			
		||||
      info!("TLS errors will be ignored");
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    let mut openssl_conf = self.prepare_env()?;
 | 
			
		||||
 | 
			
		||||
    self.server = normalize_server(&self.server)?;
 | 
			
		||||
    let gp_params = self.build_gp_params();
 | 
			
		||||
 | 
			
		||||
    // Get the initial SAML request
 | 
			
		||||
    let saml_request = match self.saml_request {
 | 
			
		||||
      Some(ref saml_request) => saml_request.clone(),
 | 
			
		||||
      None => portal_prelogin(&self.server, &gp_params).await?,
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    let browser_auth = if let Some(browser) = &self.browser {
 | 
			
		||||
      Some(BrowserAuthenticator::new_with_browser(&saml_request, browser))
 | 
			
		||||
    } else if self.default_browser {
 | 
			
		||||
      Some(BrowserAuthenticator::new(&saml_request))
 | 
			
		||||
    } else {
 | 
			
		||||
      None
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    if let Some(browser_auth) = browser_auth {
 | 
			
		||||
      browser_auth.authenticate()?;
 | 
			
		||||
 | 
			
		||||
      info!("Please continue the authentication process in the default browser");
 | 
			
		||||
 | 
			
		||||
      let auth_result = match wait_auth_data().await {
 | 
			
		||||
        Ok(auth_data) => SamlAuthResult::Success(auth_data),
 | 
			
		||||
        Err(err) => SamlAuthResult::Failure(format!("{}", err)),
 | 
			
		||||
      };
 | 
			
		||||
 | 
			
		||||
      info!("Authentication completed");
 | 
			
		||||
 | 
			
		||||
      println!("{}", json!(auth_result));
 | 
			
		||||
 | 
			
		||||
      return Ok(());
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    self.saml_request.replace(saml_request);
 | 
			
		||||
 | 
			
		||||
    let app = create_app(self.clone())?;
 | 
			
		||||
 | 
			
		||||
    app.run(move |_app_handle, event| {
 | 
			
		||||
      if let RunEvent::Exit = event {
 | 
			
		||||
        if let Some(file) = openssl_conf.take() {
 | 
			
		||||
          if let Err(err) = file.close() {
 | 
			
		||||
            info!("Error closing OpenSSL config file: {}", err);
 | 
			
		||||
          }
 | 
			
		||||
        }
 | 
			
		||||
      }
 | 
			
		||||
    });
 | 
			
		||||
 | 
			
		||||
    Ok(())
 | 
			
		||||
impl Args for Cli {
 | 
			
		||||
  fn fix_openssl(&self) -> bool {
 | 
			
		||||
    self.fix_openssl
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  fn ignore_tls_errors(&self) -> bool {
 | 
			
		||||
    self.ignore_tls_errors
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Cli {
 | 
			
		||||
  fn prepare_env(&self) -> anyhow::Result<Option<NamedTempFile>> {
 | 
			
		||||
    env_utils::patch_gui_runtime_env(self.hidpi);
 | 
			
		||||
 | 
			
		||||
@@ -150,6 +97,64 @@ impl Cli {
 | 
			
		||||
    Ok(None)
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  async fn run(&self) -> anyhow::Result<()> {
 | 
			
		||||
    if self.ignore_tls_errors {
 | 
			
		||||
      info!("TLS errors will be ignored");
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    let mut openssl_conf = self.prepare_env()?;
 | 
			
		||||
 | 
			
		||||
    let server = normalize_server(&self.server)?;
 | 
			
		||||
    let server: &'static str = Box::leak(server.into_boxed_str());
 | 
			
		||||
    let gp_params: &'static GpParams = Box::leak(Box::new(self.build_gp_params()));
 | 
			
		||||
 | 
			
		||||
    let auth_request = self.saml_request.clone().unwrap_or_default();
 | 
			
		||||
    let auth_request: &'static str = Box::leak(Box::new(auth_request));
 | 
			
		||||
 | 
			
		||||
    let auth_window = AuthWindow::new(&server, gp_params)
 | 
			
		||||
      .with_auth_request(&auth_request)
 | 
			
		||||
      .with_clean(self.clean);
 | 
			
		||||
 | 
			
		||||
    let browser = if let Some(browser) = self.browser.as_deref() {
 | 
			
		||||
      Some(browser)
 | 
			
		||||
    } else if self.default_browser {
 | 
			
		||||
      Some("default")
 | 
			
		||||
    } else {
 | 
			
		||||
      None
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    if browser.is_some() {
 | 
			
		||||
      let auth_result = auth_window.browser_authenticate(browser).await;
 | 
			
		||||
      print_auth_result(auth_result);
 | 
			
		||||
 | 
			
		||||
      return Ok(());
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    tauri::Builder::default()
 | 
			
		||||
      .setup(move |app| {
 | 
			
		||||
        let app_handle = app.handle().clone();
 | 
			
		||||
 | 
			
		||||
        tauri::async_runtime::spawn(async move {
 | 
			
		||||
          let auth_result = auth_window.webview_authenticate(&app_handle).await;
 | 
			
		||||
          print_auth_result(auth_result);
 | 
			
		||||
        });
 | 
			
		||||
 | 
			
		||||
        Ok(())
 | 
			
		||||
      })
 | 
			
		||||
      .build(tauri::generate_context!())?
 | 
			
		||||
      .run(move |_app_handle, event| {
 | 
			
		||||
        if let RunEvent::Exit = event {
 | 
			
		||||
          if let Some(file) = openssl_conf.take() {
 | 
			
		||||
            if let Err(err) = file.close() {
 | 
			
		||||
              info!("Error closing OpenSSL config file: {}", err);
 | 
			
		||||
            }
 | 
			
		||||
          }
 | 
			
		||||
        }
 | 
			
		||||
      });
 | 
			
		||||
 | 
			
		||||
    Ok(())
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  fn build_gp_params(&self) -> GpParams {
 | 
			
		||||
    let gp_params = GpParams::builder()
 | 
			
		||||
      .user_agent(&self.user_agent)
 | 
			
		||||
@@ -161,37 +166,6 @@ impl Cli {
 | 
			
		||||
 | 
			
		||||
    gp_params
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  async fn saml_auth(&self, app_handle: AppHandle) -> anyhow::Result<SamlAuthData> {
 | 
			
		||||
    let auth_window = AuthWindow::new(app_handle)
 | 
			
		||||
      .server(&self.server)
 | 
			
		||||
      .user_agent(&self.user_agent)
 | 
			
		||||
      .gp_params(self.build_gp_params())
 | 
			
		||||
      .saml_request(self.saml_request.as_ref().unwrap())
 | 
			
		||||
      .clean(self.clean);
 | 
			
		||||
 | 
			
		||||
    auth_window.open().await
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn create_app(cli: Cli) -> anyhow::Result<App> {
 | 
			
		||||
  let app = tauri::Builder::default()
 | 
			
		||||
    .setup(|app| {
 | 
			
		||||
      let app_handle = app.handle();
 | 
			
		||||
 | 
			
		||||
      tauri::async_runtime::spawn(async move {
 | 
			
		||||
        let auth_result = match cli.saml_auth(app_handle.clone()).await {
 | 
			
		||||
          Ok(auth_data) => SamlAuthResult::Success(auth_data),
 | 
			
		||||
          Err(err) => SamlAuthResult::Failure(format!("{}", err)),
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        println!("{}", json!(auth_result));
 | 
			
		||||
      });
 | 
			
		||||
      Ok(())
 | 
			
		||||
    })
 | 
			
		||||
    .build(tauri::generate_context!())?;
 | 
			
		||||
 | 
			
		||||
  Ok(app)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn init_logger() {
 | 
			
		||||
@@ -199,53 +173,22 @@ fn init_logger() {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub async fn run() {
 | 
			
		||||
  let mut cli = Cli::parse();
 | 
			
		||||
  let cli = Cli::parse();
 | 
			
		||||
 | 
			
		||||
  init_logger();
 | 
			
		||||
  info!("gpauth started: {}", VERSION);
 | 
			
		||||
 | 
			
		||||
  if let Err(err) = cli.run().await {
 | 
			
		||||
    eprintln!("\nError: {}", err);
 | 
			
		||||
 | 
			
		||||
    if err.to_string().contains("unsafe legacy renegotiation") && !cli.fix_openssl {
 | 
			
		||||
      eprintln!("\nRe-run it with the `--fix-openssl` option to work around this issue, e.g.:\n");
 | 
			
		||||
      // Print the command
 | 
			
		||||
      let args = std::env::args().collect::<Vec<_>>();
 | 
			
		||||
      eprintln!("{} --fix-openssl {}\n", args[0], args[1..].join(" "));
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    handle_error(err, &cli);
 | 
			
		||||
    std::process::exit(1);
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
async fn wait_auth_data() -> anyhow::Result<SamlAuthData> {
 | 
			
		||||
  // Start a local server to receive the browser authentication data
 | 
			
		||||
  let listener = TcpListener::bind("127.0.0.1:0").await?;
 | 
			
		||||
  let port = listener.local_addr()?.port();
 | 
			
		||||
  let port_file = temp_dir().join("gpcallback.port");
 | 
			
		||||
fn print_auth_result(auth_result: anyhow::Result<SamlAuthData>) {
 | 
			
		||||
  let auth_result = match auth_result {
 | 
			
		||||
    Ok(auth_data) => SamlAuthResult::Success(auth_data),
 | 
			
		||||
    Err(err) => SamlAuthResult::Failure(format!("{}", err)),
 | 
			
		||||
  };
 | 
			
		||||
 | 
			
		||||
  // Write the port to a file
 | 
			
		||||
  fs::write(&port_file, port.to_string())?;
 | 
			
		||||
  fs::set_permissions(&port_file, fs::Permissions::from_mode(0o600))?;
 | 
			
		||||
 | 
			
		||||
  // Remove the previous log file
 | 
			
		||||
  let callback_log = temp_dir().join("gpcallback.log");
 | 
			
		||||
  let _ = fs::remove_file(&callback_log);
 | 
			
		||||
 | 
			
		||||
  info!("Listening authentication data on port {}", port);
 | 
			
		||||
  info!(
 | 
			
		||||
    "If it hangs, please check the logs at `{}` for more information",
 | 
			
		||||
    callback_log.display()
 | 
			
		||||
  );
 | 
			
		||||
  let (mut socket, _) = listener.accept().await?;
 | 
			
		||||
 | 
			
		||||
  info!("Received the browser authentication data from the socket");
 | 
			
		||||
  let mut data = String::new();
 | 
			
		||||
  socket.read_to_string(&mut data).await?;
 | 
			
		||||
 | 
			
		||||
  // Remove the port file
 | 
			
		||||
  fs::remove_file(&port_file)?;
 | 
			
		||||
 | 
			
		||||
  let auth_data = SamlAuthData::from_gpcallback(&data)?;
 | 
			
		||||
  Ok(auth_data)
 | 
			
		||||
  println!("{}", json!(auth_result));
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										174
									
								
								apps/gpauth/src/common.rs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										174
									
								
								apps/gpauth/src/common.rs
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,174 @@
 | 
			
		||||
use std::sync::Arc;
 | 
			
		||||
 | 
			
		||||
use gpapi::{
 | 
			
		||||
  auth::{AuthDataParseResult, SamlAuthData},
 | 
			
		||||
  error::AuthDataParseError,
 | 
			
		||||
};
 | 
			
		||||
use log::{info, warn};
 | 
			
		||||
use regex::Regex;
 | 
			
		||||
 | 
			
		||||
use crate::auth_messenger::{AuthError, AuthMessenger};
 | 
			
		||||
 | 
			
		||||
pub struct AuthSettings<'a> {
 | 
			
		||||
  pub auth_request: AuthRequest<'a>,
 | 
			
		||||
  pub auth_messenger: Arc<AuthMessenger>,
 | 
			
		||||
  pub ignore_tls_errors: bool,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub struct AuthRequest<'a>(&'a str);
 | 
			
		||||
 | 
			
		||||
impl<'a> AuthRequest<'a> {
 | 
			
		||||
  pub fn new(auth_request: &'a str) -> Self {
 | 
			
		||||
    Self(auth_request)
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  pub fn is_url(&self) -> bool {
 | 
			
		||||
    self.0.starts_with("http")
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  pub fn as_str(&self) -> &str {
 | 
			
		||||
    self.0
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/// Trait for handling authentication response
 | 
			
		||||
pub trait AuthResponse {
 | 
			
		||||
  fn get_header(&self, key: &str) -> Option<String>;
 | 
			
		||||
  fn get_body<F>(&self, cb: F)
 | 
			
		||||
  where
 | 
			
		||||
    F: FnOnce(anyhow::Result<Vec<u8>>) + 'static;
 | 
			
		||||
 | 
			
		||||
  fn url(&self) -> Option<String>;
 | 
			
		||||
 | 
			
		||||
  fn is_acs_endpoint(&self) -> bool {
 | 
			
		||||
    self.url().map_or(false, |url| url.ends_with("/SAML20/SP/ACS"))
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub fn read_auth_data(auth_response: &impl AuthResponse, auth_messenger: &Arc<AuthMessenger>) {
 | 
			
		||||
  let auth_messenger = Arc::clone(auth_messenger);
 | 
			
		||||
 | 
			
		||||
  match read_from_headers(auth_response) {
 | 
			
		||||
    Ok(auth_data) => {
 | 
			
		||||
      info!("Found auth data in headers");
 | 
			
		||||
      auth_messenger.send_auth_data(auth_data);
 | 
			
		||||
    }
 | 
			
		||||
    Err(header_err) => {
 | 
			
		||||
      info!("Failed to read auth data from headers: {}", header_err);
 | 
			
		||||
 | 
			
		||||
      let is_acs_endpoint = auth_response.is_acs_endpoint();
 | 
			
		||||
      read_from_body(auth_response, move |auth_result| {
 | 
			
		||||
        // If the endpoint is `/SAML20/SP/ACS` and no auth data found in body, it should be considered as invalid
 | 
			
		||||
        let auth_result = auth_result.map_err(move |e| {
 | 
			
		||||
          info!("Failed to read auth data from body: {}", e);
 | 
			
		||||
          if is_acs_endpoint || e.is_invalid() || header_err.is_invalid() {
 | 
			
		||||
            AuthError::Invalid
 | 
			
		||||
          } else {
 | 
			
		||||
            AuthError::NotFound
 | 
			
		||||
          }
 | 
			
		||||
        });
 | 
			
		||||
 | 
			
		||||
        auth_messenger.send_auth_result(auth_result);
 | 
			
		||||
      });
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn read_from_headers(auth_response: &impl AuthResponse) -> AuthDataParseResult {
 | 
			
		||||
  let Some(status) = auth_response.get_header("saml-auth-status") else {
 | 
			
		||||
    info!("No SAML auth status found in headers");
 | 
			
		||||
    return Err(AuthDataParseError::NotFound);
 | 
			
		||||
  };
 | 
			
		||||
 | 
			
		||||
  if status != "1" {
 | 
			
		||||
    info!("Found invalid auth status: {}", status);
 | 
			
		||||
    return Err(AuthDataParseError::Invalid);
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  let username = auth_response.get_header("saml-username");
 | 
			
		||||
  let prelogin_cookie = auth_response.get_header("prelogin-cookie");
 | 
			
		||||
  let portal_userauthcookie = auth_response.get_header("portal-userauthcookie");
 | 
			
		||||
 | 
			
		||||
  SamlAuthData::new(username, prelogin_cookie, portal_userauthcookie).map_err(|e| {
 | 
			
		||||
    warn!("Found invalid auth data: {}", e);
 | 
			
		||||
    AuthDataParseError::Invalid
 | 
			
		||||
  })
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn read_from_body<F>(auth_response: &impl AuthResponse, cb: F)
 | 
			
		||||
where
 | 
			
		||||
  F: FnOnce(AuthDataParseResult) + 'static,
 | 
			
		||||
{
 | 
			
		||||
  auth_response.get_body(|body| match body {
 | 
			
		||||
    Ok(body) => {
 | 
			
		||||
      let html = String::from_utf8_lossy(&body);
 | 
			
		||||
      cb(read_from_html(&html))
 | 
			
		||||
    }
 | 
			
		||||
    Err(err) => {
 | 
			
		||||
      info!("Failed to read body: {}", err);
 | 
			
		||||
      cb(Err(AuthDataParseError::Invalid))
 | 
			
		||||
    }
 | 
			
		||||
  });
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn read_from_html(html: &str) -> AuthDataParseResult {
 | 
			
		||||
  if html.contains("Temporarily Unavailable") {
 | 
			
		||||
    info!("Found 'Temporarily Unavailable' in HTML, auth failed");
 | 
			
		||||
    return Err(AuthDataParseError::Invalid);
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  SamlAuthData::from_html(html).or_else(|err| {
 | 
			
		||||
    if let Some(gpcallback) = extract_gpcallback(html) {
 | 
			
		||||
      info!("Found gpcallback from html...");
 | 
			
		||||
      SamlAuthData::from_gpcallback(&gpcallback)
 | 
			
		||||
    } else {
 | 
			
		||||
      Err(err)
 | 
			
		||||
    }
 | 
			
		||||
  })
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn extract_gpcallback(html: &str) -> Option<String> {
 | 
			
		||||
  let re = Regex::new(r#"globalprotectcallback:[^"]+"#).unwrap();
 | 
			
		||||
  re.captures(html)
 | 
			
		||||
    .and_then(|captures| captures.get(0))
 | 
			
		||||
    .map(|m| html_escape::decode_html_entities(m.as_str()).to_string())
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[cfg(test)]
 | 
			
		||||
mod tests {
 | 
			
		||||
  use super::*;
 | 
			
		||||
 | 
			
		||||
  #[test]
 | 
			
		||||
  fn extract_gpcallback_some() {
 | 
			
		||||
    let html = r#"
 | 
			
		||||
      <meta http-equiv="refresh" content="0; URL=globalprotectcallback:PGh0bWw+PCEtLSA8c">
 | 
			
		||||
      <meta http-equiv="refresh" content="0; URL=globalprotectcallback:PGh0bWw+PCEtLSA8c">
 | 
			
		||||
    "#;
 | 
			
		||||
 | 
			
		||||
    assert_eq!(
 | 
			
		||||
      extract_gpcallback(html).as_deref(),
 | 
			
		||||
      Some("globalprotectcallback:PGh0bWw+PCEtLSA8c")
 | 
			
		||||
    );
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  #[test]
 | 
			
		||||
  fn extract_gpcallback_cas() {
 | 
			
		||||
    let html = r#"
 | 
			
		||||
      <meta http-equiv="refresh" content="0; URL=globalprotectcallback:cas-as=1&un=xyz@email.com&token=very_long_string">
 | 
			
		||||
    "#;
 | 
			
		||||
 | 
			
		||||
    assert_eq!(
 | 
			
		||||
      extract_gpcallback(html).as_deref(),
 | 
			
		||||
      Some("globalprotectcallback:cas-as=1&un=xyz@email.com&token=very_long_string")
 | 
			
		||||
    );
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  #[test]
 | 
			
		||||
  fn extract_gpcallback_none() {
 | 
			
		||||
    let html = r#"
 | 
			
		||||
      <meta http-equiv="refresh" content="0; URL=PGh0bWw+PCEtLSA8c">
 | 
			
		||||
    "#;
 | 
			
		||||
 | 
			
		||||
    assert_eq!(extract_gpcallback(html), None);
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										7
									
								
								apps/gpauth/src/lib.rs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										7
									
								
								apps/gpauth/src/lib.rs
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,7 @@
 | 
			
		||||
mod auth_messenger;
 | 
			
		||||
mod common;
 | 
			
		||||
 | 
			
		||||
pub mod auth_window;
 | 
			
		||||
 | 
			
		||||
#[cfg_attr(not(target_os = "macos"), path = "unix.rs")]
 | 
			
		||||
mod platform_impl;
 | 
			
		||||
@@ -1,6 +1,5 @@
 | 
			
		||||
#![cfg_attr(not(debug_assertions), windows_subsystem = "windows")]
 | 
			
		||||
 | 
			
		||||
mod auth_window;
 | 
			
		||||
mod cli;
 | 
			
		||||
 | 
			
		||||
#[tokio::main]
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										133
									
								
								apps/gpauth/src/unix.rs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										133
									
								
								apps/gpauth/src/unix.rs
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,133 @@
 | 
			
		||||
use std::sync::Arc;
 | 
			
		||||
 | 
			
		||||
use anyhow::bail;
 | 
			
		||||
use gpapi::utils::redact::redact_uri;
 | 
			
		||||
use log::{info, warn};
 | 
			
		||||
use webkit2gtk::{
 | 
			
		||||
  gio::Cancellable,
 | 
			
		||||
  glib::{GString, TimeSpan},
 | 
			
		||||
  LoadEvent, TLSErrorsPolicy, URIResponseExt, WebResource, WebResourceExt, WebView, WebViewExt, WebsiteDataManagerExt,
 | 
			
		||||
  WebsiteDataManagerExtManual, WebsiteDataTypes,
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
use crate::{
 | 
			
		||||
  auth_messenger::AuthError,
 | 
			
		||||
  common::{read_auth_data, AuthRequest, AuthResponse, AuthSettings},
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
impl AuthResponse for WebResource {
 | 
			
		||||
  fn get_header(&self, key: &str) -> Option<String> {
 | 
			
		||||
    self
 | 
			
		||||
      .response()
 | 
			
		||||
      .and_then(|response| response.http_headers())
 | 
			
		||||
      .and_then(|headers| headers.one(key))
 | 
			
		||||
      .map(GString::into)
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  fn get_body<F>(&self, cb: F)
 | 
			
		||||
  where
 | 
			
		||||
    F: FnOnce(anyhow::Result<Vec<u8>>) + 'static,
 | 
			
		||||
  {
 | 
			
		||||
    let cancellable = Cancellable::NONE;
 | 
			
		||||
    self.data(cancellable, |data| cb(data.map_err(|e| anyhow::anyhow!(e))));
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  fn url(&self) -> Option<String> {
 | 
			
		||||
    self.uri().map(GString::into)
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub fn clear_data<F>(wv: &WebView, cb: F)
 | 
			
		||||
where
 | 
			
		||||
  F: FnOnce(anyhow::Result<()>) + Send + 'static,
 | 
			
		||||
{
 | 
			
		||||
  let Some(data_manager) = wv.website_data_manager() else {
 | 
			
		||||
    cb(Err(anyhow::anyhow!("Failed to get website data manager")));
 | 
			
		||||
    return;
 | 
			
		||||
  };
 | 
			
		||||
 | 
			
		||||
  data_manager.clear(
 | 
			
		||||
    WebsiteDataTypes::COOKIES,
 | 
			
		||||
    TimeSpan(0),
 | 
			
		||||
    Cancellable::NONE,
 | 
			
		||||
    move |result| {
 | 
			
		||||
      cb(result.map_err(|e| anyhow::anyhow!(e)));
 | 
			
		||||
    },
 | 
			
		||||
  );
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub fn setup_webview(wv: &WebView, auth_settings: AuthSettings) -> anyhow::Result<()> {
 | 
			
		||||
  let AuthSettings {
 | 
			
		||||
    auth_request,
 | 
			
		||||
    auth_messenger,
 | 
			
		||||
    ignore_tls_errors,
 | 
			
		||||
  } = auth_settings;
 | 
			
		||||
  let auth_messenger_clone = Arc::clone(&auth_messenger);
 | 
			
		||||
 | 
			
		||||
  let Some(data_manager) = wv.website_data_manager() else {
 | 
			
		||||
    bail!("Failed to get website data manager");
 | 
			
		||||
  };
 | 
			
		||||
 | 
			
		||||
  if ignore_tls_errors {
 | 
			
		||||
    data_manager.set_tls_errors_policy(TLSErrorsPolicy::Ignore);
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  wv.connect_load_changed(move |wv, event| {
 | 
			
		||||
    if event == LoadEvent::Started {
 | 
			
		||||
      auth_messenger_clone.cancel_raise_window();
 | 
			
		||||
      return;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    if event != LoadEvent::Finished {
 | 
			
		||||
      return;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    let Some(main_resource) = wv.main_resource() else {
 | 
			
		||||
      return;
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    let uri = main_resource.uri().unwrap_or("".into());
 | 
			
		||||
    if uri.is_empty() {
 | 
			
		||||
      warn!("Loaded an empty URI");
 | 
			
		||||
      auth_messenger_clone.send_auth_error(AuthError::Invalid);
 | 
			
		||||
      return;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    read_auth_data(&main_resource, &auth_messenger_clone);
 | 
			
		||||
  });
 | 
			
		||||
 | 
			
		||||
  wv.connect_load_failed_with_tls_errors(move |_wv, uri, cert, err| {
 | 
			
		||||
    let redacted_uri = redact_uri(uri);
 | 
			
		||||
    warn!(
 | 
			
		||||
      "Failed to load uri: {} with error: {}, cert: {}",
 | 
			
		||||
      redacted_uri, err, cert
 | 
			
		||||
    );
 | 
			
		||||
 | 
			
		||||
    auth_messenger.send_auth_error(AuthError::TlsError);
 | 
			
		||||
    true
 | 
			
		||||
  });
 | 
			
		||||
 | 
			
		||||
  wv.connect_load_failed(move |_wv, _event, uri, err| {
 | 
			
		||||
    let redacted_uri = redact_uri(uri);
 | 
			
		||||
    if !uri.starts_with("globalprotectcallback:") {
 | 
			
		||||
      warn!("Failed to load uri: {} with error: {}", redacted_uri, err);
 | 
			
		||||
    }
 | 
			
		||||
    // NOTE: Don't send error here, since load_changed event will be triggered after this
 | 
			
		||||
    // true to stop other handlers from being invoked for the event. false to propagate the event further.
 | 
			
		||||
    true
 | 
			
		||||
  });
 | 
			
		||||
 | 
			
		||||
  load_auth_request(wv, &auth_request);
 | 
			
		||||
 | 
			
		||||
  Ok(())
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub fn load_auth_request(wv: &WebView, auth_request: &AuthRequest) {
 | 
			
		||||
  if auth_request.is_url() {
 | 
			
		||||
    info!("Loading auth request as URI...");
 | 
			
		||||
    wv.load_uri(auth_request.as_str());
 | 
			
		||||
  } else {
 | 
			
		||||
    info!("Loading auth request as HTML...");
 | 
			
		||||
    wv.load_html(auth_request.as_str(), None);
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
@@ -1,47 +1,16 @@
 | 
			
		||||
{
 | 
			
		||||
  "$schema": "https://cdn.jsdelivr.net/gh/tauri-apps/tauri@tauri-v1.5.0/tooling/cli/schema.json",
 | 
			
		||||
  "$schema": "https://cdn.jsdelivr.net/gh/tauri-apps/tauri@tauri-v2.1.1/crates/tauri-cli/config.schema.json",
 | 
			
		||||
  "build": {
 | 
			
		||||
    "distDir": [
 | 
			
		||||
      "index.html"
 | 
			
		||||
    ],
 | 
			
		||||
    "devPath": [
 | 
			
		||||
      "index.html"
 | 
			
		||||
    ],
 | 
			
		||||
    "frontendDist": ["index.html"],
 | 
			
		||||
    "beforeDevCommand": "",
 | 
			
		||||
    "beforeBuildCommand": "",
 | 
			
		||||
    "withGlobalTauri": false
 | 
			
		||||
    "beforeBuildCommand": ""
 | 
			
		||||
  },
 | 
			
		||||
  "package": {
 | 
			
		||||
    "productName": "gpauth",
 | 
			
		||||
    "version": "0.0.0"
 | 
			
		||||
  },
 | 
			
		||||
  "tauri": {
 | 
			
		||||
    "allowlist": {
 | 
			
		||||
      "all": false,
 | 
			
		||||
      "http": {
 | 
			
		||||
        "all": true,
 | 
			
		||||
        "request": true,
 | 
			
		||||
        "scope": [
 | 
			
		||||
          "http://*",
 | 
			
		||||
          "https://*"
 | 
			
		||||
        ]
 | 
			
		||||
      }
 | 
			
		||||
    },
 | 
			
		||||
    "bundle": {
 | 
			
		||||
      "active": true,
 | 
			
		||||
      "targets": "deb",
 | 
			
		||||
      "identifier": "com.yuezk.gpauth",
 | 
			
		||||
      "icon": [
 | 
			
		||||
        "icons/32x32.png",
 | 
			
		||||
        "icons/128x128.png",
 | 
			
		||||
        "icons/128x128@2x.png",
 | 
			
		||||
        "icons/icon.icns",
 | 
			
		||||
        "icons/icon.ico"
 | 
			
		||||
      ]
 | 
			
		||||
    },
 | 
			
		||||
  "identifier": "com.yuezk.gpauth",
 | 
			
		||||
  "productName": "gpauth",
 | 
			
		||||
  "app": {
 | 
			
		||||
    "withGlobalTauri": false,
 | 
			
		||||
    "security": {
 | 
			
		||||
      "csp": null
 | 
			
		||||
    },
 | 
			
		||||
    "windows": []
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user