mirror of
				https://github.com/yuezk/GlobalProtect-openconnect.git
				synced 2025-05-20 07:26:58 -04:00 
			
		
		
		
	Compare commits
	
		
			7 Commits
		
	
	
		
			feature/ha
			...
			8446874290
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| 
						 | 
					8446874290 | ||
| 
						 | 
					c347f97b95 | ||
| 
						 | 
					29cfa9e24b | ||
| 
						 | 
					1b1ce882a5 | ||
| 
						 | 
					e9f2dbf9ea | ||
| 
						 | 
					7c6ae315e1 | ||
| 
						 | 
					cec0d22dc8 | 
							
								
								
									
										68
									
								
								.github/workflows/build.yaml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										68
									
								
								.github/workflows/build.yaml
									
									
									
									
										vendored
									
									
								
							@@ -13,7 +13,6 @@ on:
 | 
			
		||||
      - feature/*
 | 
			
		||||
      - release/*
 | 
			
		||||
    tags:
 | 
			
		||||
      - latest
 | 
			
		||||
      - v*.*.*
 | 
			
		||||
jobs:
 | 
			
		||||
  # Include arm64 if ref is a tag
 | 
			
		||||
@@ -26,9 +25,9 @@ jobs:
 | 
			
		||||
        id: set-matrix
 | 
			
		||||
        run: |
 | 
			
		||||
          if [[ "${{ github.ref }}" == "refs/tags/"* ]]; then
 | 
			
		||||
            echo "matrix=[\"ubuntu-latest\", \"arm64\"]" >> $GITHUB_OUTPUT
 | 
			
		||||
            echo 'matrix=[{"runner": "ubuntu-latest", "arch": "amd64"}, {"runner": "arm64", "arch": "arm64"]' >> $GITHUB_OUTPUT
 | 
			
		||||
          else
 | 
			
		||||
            echo "matrix=[\"ubuntu-latest\"]" >> $GITHUB_OUTPUT
 | 
			
		||||
            echo 'matrix=[{"runner": "ubuntu-latest", "arch": "amd64"}]' >> $GITHUB_OUTPUT
 | 
			
		||||
          fi
 | 
			
		||||
 | 
			
		||||
  tarball:
 | 
			
		||||
@@ -50,6 +49,10 @@ jobs:
 | 
			
		||||
    - name: Create tarball
 | 
			
		||||
      run: |
 | 
			
		||||
        cd source/gp
 | 
			
		||||
        # Generate the SNAPSHOT file for non-tagged commits
 | 
			
		||||
        if [[ "${{ github.ref }}" != "refs/tags/"* ]]; then
 | 
			
		||||
          touch SNAPSHOT
 | 
			
		||||
        fi
 | 
			
		||||
        make tarball
 | 
			
		||||
    - name: Upload tarball
 | 
			
		||||
      uses: actions/upload-artifact@v3
 | 
			
		||||
@@ -66,20 +69,40 @@ jobs:
 | 
			
		||||
    strategy:
 | 
			
		||||
      matrix:
 | 
			
		||||
        os: ${{fromJson(needs.setup-matrix.outputs.matrix)}}
 | 
			
		||||
    runs-on: ${{ matrix.os }}
 | 
			
		||||
        package: [deb, rpm, pkg, binary]
 | 
			
		||||
    runs-on: ${{ matrix.os.runner }}
 | 
			
		||||
    name: build-gp (${{ matrix.package }}, ${{ matrix.os.arch }})
 | 
			
		||||
    steps:
 | 
			
		||||
    - name: Prepare workspace
 | 
			
		||||
      run: rm -rf build-gp && mkdir build-gp
 | 
			
		||||
      run: |
 | 
			
		||||
        rm -rf build-gp-${{ matrix.package }}
 | 
			
		||||
        mkdir -p build-gp-${{ matrix.package }}
 | 
			
		||||
    - name: Download tarball
 | 
			
		||||
      uses: actions/download-artifact@v3
 | 
			
		||||
      with:
 | 
			
		||||
        name: artifact-source
 | 
			
		||||
        path: build-gp
 | 
			
		||||
        path: build-gp-${{ matrix.package }}
 | 
			
		||||
    - name: Docker Login
 | 
			
		||||
      run: echo ${{ secrets.DOCKER_HUB_TOKEN }} | docker login -u ${{ secrets.DOCKER_HUB_USERNAME }} --password-stdin
 | 
			
		||||
    - name: Build gp in Docker
 | 
			
		||||
    - name: Build ${{ matrix.package }} package in Docker
 | 
			
		||||
      run: |
 | 
			
		||||
        docker run --rm -v $(pwd)/build-gp:/gp yuezk/gpdev:gp-builder
 | 
			
		||||
        docker run --rm \
 | 
			
		||||
          -v $(pwd)/build-gp-${{ matrix.package }}:/${{ matrix.package }} \
 | 
			
		||||
          yuezk/gpdev:${{ matrix.package }}-builder
 | 
			
		||||
    - name: Install ${{ matrix.package }} package in Docker
 | 
			
		||||
      run: |
 | 
			
		||||
        docker run --rm \
 | 
			
		||||
          -e GPGUI_INSTALLED=0 \
 | 
			
		||||
          -v $(pwd)/build-gp-${{ matrix.package }}:/${{ matrix.package }} \
 | 
			
		||||
          yuezk/gpdev:${{ matrix.package }}-builder \
 | 
			
		||||
          bash install.sh
 | 
			
		||||
    - name: Upload ${{ matrix.package }} package
 | 
			
		||||
      uses: actions/upload-artifact@v3
 | 
			
		||||
      with:
 | 
			
		||||
        name: artifact-gp-${{ matrix.package }}-${{ matrix.os.arch }}
 | 
			
		||||
        if-no-files-found: error
 | 
			
		||||
        path: |
 | 
			
		||||
          build-gp-${{ matrix.package }}/artifacts/*
 | 
			
		||||
 | 
			
		||||
  build-gpgui:
 | 
			
		||||
    needs:
 | 
			
		||||
@@ -87,7 +110,8 @@ jobs:
 | 
			
		||||
    strategy:
 | 
			
		||||
      matrix:
 | 
			
		||||
        os: ${{fromJson(needs.setup-matrix.outputs.matrix)}}
 | 
			
		||||
    runs-on: ${{ matrix.os }}
 | 
			
		||||
    runs-on: ${{ matrix.os.runner }}
 | 
			
		||||
    name: build-gpgui (${{ matrix.os.arch }})
 | 
			
		||||
    steps:
 | 
			
		||||
    - uses: pnpm/action-setup@v2
 | 
			
		||||
      with:
 | 
			
		||||
@@ -126,16 +150,17 @@ jobs:
 | 
			
		||||
    - name: Upload gpgui
 | 
			
		||||
      uses: actions/upload-artifact@v3
 | 
			
		||||
      with:
 | 
			
		||||
        name: artifact-gpgui-${{ matrix.os }}
 | 
			
		||||
        name: artifact-gpgui-${{ matrix.os.arch }}
 | 
			
		||||
        if-no-files-found: error
 | 
			
		||||
        path: |
 | 
			
		||||
          gpgui-source/*.bin.tar.xz
 | 
			
		||||
          gpgui-source/*.bin.tar.xz.sha256
 | 
			
		||||
 | 
			
		||||
  gh-release:
 | 
			
		||||
    if: startsWith(github.ref, 'refs/tags/')
 | 
			
		||||
    if: ${{ github.ref == 'refs/heads/dev' || startsWith(github.ref, 'refs/tags/') }}
 | 
			
		||||
    runs-on: ubuntu-latest
 | 
			
		||||
    needs:
 | 
			
		||||
      - tarball
 | 
			
		||||
      - build-gp
 | 
			
		||||
      - build-gpgui
 | 
			
		||||
 | 
			
		||||
@@ -147,10 +172,17 @@ jobs:
 | 
			
		||||
      with:
 | 
			
		||||
        path: gh-release
 | 
			
		||||
    - name: Create GH release
 | 
			
		||||
      uses: softprops/action-gh-release@v1
 | 
			
		||||
      with:
 | 
			
		||||
        token: ${{ secrets.GH_PAT }}
 | 
			
		||||
        prerelease: ${{ contains(github.ref, 'latest') }}
 | 
			
		||||
        fail_on_unmatched_files: true
 | 
			
		||||
        files: |
 | 
			
		||||
          gh-release/artifact-*/*
 | 
			
		||||
      env:
 | 
			
		||||
        GH_TOKEN: ${{ secrets.GH_PAT }}
 | 
			
		||||
        RELEASE_TAG: ${{ github.ref == 'refs/heads/dev' && 'snapshot' || github.ref_name }}
 | 
			
		||||
        REPO: ${{ github.repository }}
 | 
			
		||||
        NOTES: ${{ github.ref == 'refs/heads/dev' && '**!!! DO NOT USE THIS RELEASE IN PRODUCTION !!!**' || format('Release {0}', github.ref_name) }}
 | 
			
		||||
      run: |
 | 
			
		||||
        gh -R "$REPO" release delete $RELEASE_TAG --yes --cleanup-tag || true
 | 
			
		||||
        gh -R "$REPO" release create $RELEASE_TAG \
 | 
			
		||||
          --title "$RELEASE_TAG" \
 | 
			
		||||
          --notes "$NOTES" \
 | 
			
		||||
          --target ${{ github.ref }} \
 | 
			
		||||
          ${{ github.ref == 'refs/heads/dev' && '--prerelease' || '' }} \
 | 
			
		||||
          gh-release/artifact-source/* \
 | 
			
		||||
          gh-release/artifact-gpgui-*/*
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										11
									
								
								.github/workflows/release.yaml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										11
									
								
								.github/workflows/release.yaml
									
									
									
									
										vendored
									
									
								
							@@ -109,11 +109,16 @@ jobs:
 | 
			
		||||
      run: echo ${{ secrets.DOCKER_HUB_TOKEN }} | docker login -u ${{ secrets.DOCKER_HUB_USERNAME }} --password-stdin
 | 
			
		||||
    - name: Build ${{ matrix.package }} package in Docker
 | 
			
		||||
      run: |
 | 
			
		||||
        docker run --rm -v $(pwd)/build-${{ matrix.package }}:/${{ matrix.package }} -e INCLUDE_GUI=1 yuezk/gpdev:${{ matrix.package }}-builder
 | 
			
		||||
        docker run --rm \
 | 
			
		||||
          -v $(pwd)/build-${{ matrix.package }}:/${{ matrix.package }} \
 | 
			
		||||
          -e INCLUDE_GUI=1 \
 | 
			
		||||
          yuezk/gpdev:${{ matrix.package }}-builder
 | 
			
		||||
 | 
			
		||||
    - name: Install ${{ matrix.package }} package in Docker
 | 
			
		||||
      run: |
 | 
			
		||||
        docker run --rm -v $(pwd)/build-${{ matrix.package }}:/${{ matrix.package }} yuezk/gpdev:${{ matrix.package }}-builder \
 | 
			
		||||
        docker run --rm \
 | 
			
		||||
          -v $(pwd)/build-${{ matrix.package }}:/${{ matrix.package }} \
 | 
			
		||||
          yuezk/gpdev:${{ matrix.package }}-builder \
 | 
			
		||||
          bash install.sh
 | 
			
		||||
 | 
			
		||||
    - name: Upload ${{ matrix.package }} package
 | 
			
		||||
@@ -140,7 +145,7 @@ jobs:
 | 
			
		||||
      uses: softprops/action-gh-release@v1
 | 
			
		||||
      with:
 | 
			
		||||
        token: ${{ secrets.GH_PAT }}
 | 
			
		||||
        prerelease: ${{ contains(github.ref, 'latest') }}
 | 
			
		||||
        prerelease: ${{ contains(github.ref, 'snapshot') }}
 | 
			
		||||
        fail_on_unmatched_files: true
 | 
			
		||||
        tag_name: ${{ inputs.tag }}
 | 
			
		||||
        files: |
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										1
									
								
								.gitignore
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										1
									
								
								.gitignore
									
									
									
									
										vendored
									
									
								
							@@ -7,3 +7,4 @@
 | 
			
		||||
 | 
			
		||||
.cargo
 | 
			
		||||
.build
 | 
			
		||||
SNAPSHOT
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										16
									
								
								Cargo.lock
									
									
									
										generated
									
									
									
								
							
							
						
						
									
										16
									
								
								Cargo.lock
									
									
									
										generated
									
									
									
								
							@@ -1469,6 +1469,7 @@ dependencies = [
 | 
			
		||||
 "compile-time",
 | 
			
		||||
 "env_logger",
 | 
			
		||||
 "gpapi",
 | 
			
		||||
 "html-escape",
 | 
			
		||||
 "log",
 | 
			
		||||
 "regex",
 | 
			
		||||
 "serde_json",
 | 
			
		||||
@@ -1673,6 +1674,15 @@ version = "0.4.3"
 | 
			
		||||
source = "registry+https://github.com/rust-lang/crates.io-index"
 | 
			
		||||
checksum = "7f24254aa9a54b5c858eaee2f5bccdb46aaf0e486a595ed5fd8f86ba55232a70"
 | 
			
		||||
 | 
			
		||||
[[package]]
 | 
			
		||||
name = "html-escape"
 | 
			
		||||
version = "0.2.13"
 | 
			
		||||
source = "registry+https://github.com/rust-lang/crates.io-index"
 | 
			
		||||
checksum = "6d1ad449764d627e22bfd7cd5e8868264fc9236e07c752972b4080cd351cb476"
 | 
			
		||||
dependencies = [
 | 
			
		||||
 "utf8-width",
 | 
			
		||||
]
 | 
			
		||||
 | 
			
		||||
[[package]]
 | 
			
		||||
name = "html5ever"
 | 
			
		||||
version = "0.26.0"
 | 
			
		||||
@@ -4484,6 +4494,12 @@ version = "0.7.6"
 | 
			
		||||
source = "registry+https://github.com/rust-lang/crates.io-index"
 | 
			
		||||
checksum = "09cc8ee72d2a9becf2f2febe0205bbed8fc6615b7cb429ad062dc7b7ddd036a9"
 | 
			
		||||
 | 
			
		||||
[[package]]
 | 
			
		||||
name = "utf8-width"
 | 
			
		||||
version = "0.1.7"
 | 
			
		||||
source = "registry+https://github.com/rust-lang/crates.io-index"
 | 
			
		||||
checksum = "86bd8d4e895da8537e5315b8254664e6b769c4ff3db18321b297a1e7004392e3"
 | 
			
		||||
 | 
			
		||||
[[package]]
 | 
			
		||||
name = "utf8parse"
 | 
			
		||||
version = "0.2.1"
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										12
									
								
								Makefile
									
									
									
									
									
								
							
							
						
						
									
										12
									
								
								Makefile
									
									
									
									
									
								
							@@ -15,6 +15,13 @@ PUBLISH ?= 0
 | 
			
		||||
 | 
			
		||||
export DEBEMAIL = k3vinyue@gmail.com
 | 
			
		||||
export DEBFULLNAME = Kevin Yue
 | 
			
		||||
export SNAPSHOT = $(shell test -f SNAPSHOT && echo "true" || echo "false")
 | 
			
		||||
 | 
			
		||||
ifeq ($(SNAPSHOT), true)
 | 
			
		||||
	RELEASE_TAG = snapshot
 | 
			
		||||
else
 | 
			
		||||
	RELEASE_TAG = v$(VERSION)
 | 
			
		||||
endif
 | 
			
		||||
 | 
			
		||||
CARGO_BUILD_ARGS = --release
 | 
			
		||||
 | 
			
		||||
@@ -61,7 +68,8 @@ download-gui:
 | 
			
		||||
	if [ $(INCLUDE_GUI) -eq 1 ]; then \
 | 
			
		||||
		echo "Downloading GlobalProtect GUI..."; \
 | 
			
		||||
		mkdir -p .build/gpgui; \
 | 
			
		||||
		curl -sSL https://github.com/yuezk/GlobalProtect-openconnect/releases/download/v$(VERSION)/gpgui_$(VERSION)_$(shell uname -m).bin.tar.xz -o .build/gpgui/gpgui_$(VERSION)_x$(shell uname -m).bin.tar.xz; \
 | 
			
		||||
		curl -sSL https://github.com/yuezk/GlobalProtect-openconnect/releases/download/$(RELEASE_TAG)/gpgui_$(shell uname -m).bin.tar.xz \
 | 
			
		||||
			-o .build/gpgui/gpgui_$(shell uname -m).bin.tar.xz; \
 | 
			
		||||
		tar -xJf .build/gpgui/*.tar.xz -C .build/gpgui; \
 | 
			
		||||
	else \
 | 
			
		||||
		echo "Skipping GlobalProtect GUI download (INCLUDE_GUI=0)"; \
 | 
			
		||||
@@ -195,7 +203,7 @@ init-rpm: clean-rpm
 | 
			
		||||
	sed -i "s/@VERSION@/$(VERSION)/g" .build/rpm/globalprotect-openconnect.spec
 | 
			
		||||
	sed -i "s/@REVISION@/$(REVISION)/g" .build/rpm/globalprotect-openconnect.spec
 | 
			
		||||
	sed -i "s/@OFFLINE@/$(OFFLINE)/g" .build/rpm/globalprotect-openconnect.spec
 | 
			
		||||
	sed -i "s/@DATE@/$(shell date "+%a %b %d %Y")/g" .build/rpm/globalprotect-openconnect.spec
 | 
			
		||||
	sed -i "s/@DATE@/$(shell LC_ALL=en.US date "+%a %b %d %Y")/g" .build/rpm/globalprotect-openconnect.spec
 | 
			
		||||
 | 
			
		||||
	sed -i "s/@VERSION@/$(VERSION)/g" .build/rpm/globalprotect-openconnect.changes
 | 
			
		||||
	sed -i "s/@DATE@/$(shell LC_ALL=en.US date -u "+%a %b %e %T %Z %Y")/g" .build/rpm/globalprotect-openconnect.changes
 | 
			
		||||
 
 | 
			
		||||
@@ -18,6 +18,7 @@ serde_json.workspace = true
 | 
			
		||||
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
 | 
			
		||||
 
 | 
			
		||||
@@ -7,6 +7,7 @@ use std::{
 | 
			
		||||
use anyhow::bail;
 | 
			
		||||
use gpapi::{
 | 
			
		||||
  auth::SamlAuthData,
 | 
			
		||||
  error::AuthDataParseError,
 | 
			
		||||
  gp_params::GpParams,
 | 
			
		||||
  portal::{prelogin, Prelogin},
 | 
			
		||||
  utils::{redact::redact_uri, window::WindowExt},
 | 
			
		||||
@@ -184,6 +185,10 @@ impl<'a> AuthWindow<'a> {
 | 
			
		||||
          }
 | 
			
		||||
 | 
			
		||||
          info!("Loaded uri: {}", redact_uri(&uri));
 | 
			
		||||
          if uri.starts_with("globalprotectcallback:") {
 | 
			
		||||
            return;
 | 
			
		||||
          }
 | 
			
		||||
 | 
			
		||||
          read_auth_data(&main_resource, auth_result_tx_clone.clone());
 | 
			
		||||
        }
 | 
			
		||||
      });
 | 
			
		||||
@@ -202,7 +207,9 @@ impl<'a> AuthWindow<'a> {
 | 
			
		||||
 | 
			
		||||
      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.
 | 
			
		||||
@@ -339,7 +346,7 @@ fn read_auth_data_from_headers(response: &URIResponse) -> AuthResult {
 | 
			
		||||
 | 
			
		||||
fn read_auth_data_from_body<F>(main_resource: &WebResource, callback: F)
 | 
			
		||||
where
 | 
			
		||||
  F: FnOnce(AuthResult) + Send + 'static,
 | 
			
		||||
  F: FnOnce(Result<SamlAuthData, AuthDataParseError>) + Send + 'static,
 | 
			
		||||
{
 | 
			
		||||
  main_resource.data(Cancellable::NONE, |data| match data {
 | 
			
		||||
    Ok(data) => {
 | 
			
		||||
@@ -348,53 +355,44 @@ where
 | 
			
		||||
    }
 | 
			
		||||
    Err(err) => {
 | 
			
		||||
      info!("Failed to read response body: {}", err);
 | 
			
		||||
      callback(Err(AuthDataError::Invalid))
 | 
			
		||||
      callback(Err(AuthDataParseError::Invalid))
 | 
			
		||||
    }
 | 
			
		||||
  });
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn read_auth_data_from_html(html: &str) -> AuthResult {
 | 
			
		||||
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(AuthDataError::Invalid);
 | 
			
		||||
    return Err(AuthDataParseError::Invalid);
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  match parse_xml_tag(html, "saml-auth-status") {
 | 
			
		||||
    Some(saml_status) if saml_status == "1" => {
 | 
			
		||||
      let username = parse_xml_tag(html, "saml-username");
 | 
			
		||||
      let prelogin_cookie = parse_xml_tag(html, "prelogin-cookie");
 | 
			
		||||
      let portal_userauthcookie = parse_xml_tag(html, "portal-userauthcookie");
 | 
			
		||||
 | 
			
		||||
      if SamlAuthData::check(&username, &prelogin_cookie, &portal_userauthcookie) {
 | 
			
		||||
        return Ok(SamlAuthData::new(
 | 
			
		||||
          username.unwrap(),
 | 
			
		||||
          prelogin_cookie,
 | 
			
		||||
          portal_userauthcookie,
 | 
			
		||||
        ));
 | 
			
		||||
  match SamlAuthData::from_html(html) {
 | 
			
		||||
    Ok(auth_data) => Ok(auth_data),
 | 
			
		||||
    Err(err) => {
 | 
			
		||||
      if let Some(gpcallback) = extract_gpcallback(html) {
 | 
			
		||||
        info!("Found gpcallback from html...");
 | 
			
		||||
        SamlAuthData::from_gpcallback(&gpcallback)
 | 
			
		||||
      } else {
 | 
			
		||||
        Err(err)
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
      info!("Found invalid auth data in HTML");
 | 
			
		||||
      Err(AuthDataError::Invalid)
 | 
			
		||||
    }
 | 
			
		||||
    Some(status) => {
 | 
			
		||||
      info!("Found invalid SAML status {} in HTML", status);
 | 
			
		||||
      Err(AuthDataError::Invalid)
 | 
			
		||||
    }
 | 
			
		||||
    None => {
 | 
			
		||||
      info!("No auth data found in HTML");
 | 
			
		||||
      Err(AuthDataError::NotFound)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
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>) {
 | 
			
		||||
  if main_resource.response().is_none() {
 | 
			
		||||
  let Some(response) = main_resource.response() else {
 | 
			
		||||
    info!("No response found in main resource");
 | 
			
		||||
    send_auth_result(&auth_result_tx, Err(AuthDataError::Invalid));
 | 
			
		||||
    return;
 | 
			
		||||
  }
 | 
			
		||||
  };
 | 
			
		||||
 | 
			
		||||
  let response = main_resource.response().unwrap();
 | 
			
		||||
  info!("Trying to read auth data from response headers...");
 | 
			
		||||
 | 
			
		||||
  match read_auth_data_from_headers(&response) {
 | 
			
		||||
@@ -407,22 +405,27 @@ fn read_auth_data(main_resource: &WebResource, auth_result_tx: mpsc::UnboundedSe
 | 
			
		||||
      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(|_| AuthDataError::Invalid);
 | 
			
		||||
        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 url = main_resource.uri().unwrap_or("".into());
 | 
			
		||||
      let is_acs_endpoint = url.contains("/SAML20/SP/ACS");
 | 
			
		||||
 | 
			
		||||
      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| {
 | 
			
		||||
          if matches!(err, AuthDataError::NotFound) && is_acs_endpoint {
 | 
			
		||||
            AuthDataError::Invalid
 | 
			
		||||
          info!("Failed to read auth data from body: {}", err);
 | 
			
		||||
 | 
			
		||||
          if !is_acs_endpoint && matches!(err, AuthDataParseError::NotFound) {
 | 
			
		||||
            AuthDataError::NotFound
 | 
			
		||||
          } else {
 | 
			
		||||
            err
 | 
			
		||||
            AuthDataError::Invalid
 | 
			
		||||
          }
 | 
			
		||||
        });
 | 
			
		||||
 | 
			
		||||
@@ -437,13 +440,6 @@ fn read_auth_data(main_resource: &WebResource, auth_result_tx: mpsc::UnboundedSe
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn parse_xml_tag(html: &str, tag: &str) -> Option<String> {
 | 
			
		||||
  let re = Regex::new(&format!("<{}>(.*)</{}>", tag, tag)).unwrap();
 | 
			
		||||
  re.captures(html)
 | 
			
		||||
    .and_then(|captures| captures.get(1))
 | 
			
		||||
    .map(|m| m.as_str().to_string())
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub(crate) async fn clear_webview_cookies(window: &Window) -> anyhow::Result<()> {
 | 
			
		||||
  let (tx, rx) = oneshot::channel::<Result<(), String>>();
 | 
			
		||||
 | 
			
		||||
@@ -489,3 +485,42 @@ pub(crate) async fn clear_webview_cookies(window: &Window) -> anyhow::Result<()>
 | 
			
		||||
 | 
			
		||||
  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);
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -82,7 +82,7 @@ async fn feed_auth_data(auth_data: &str) -> anyhow::Result<()> {
 | 
			
		||||
 | 
			
		||||
  reqwest::Client::default()
 | 
			
		||||
    .post(format!("{}/auth-data", service_endpoint))
 | 
			
		||||
    .json(&auth_data)
 | 
			
		||||
    .body(auth_data.to_string())
 | 
			
		||||
    .send()
 | 
			
		||||
    .await?
 | 
			
		||||
    .error_for_status()?;
 | 
			
		||||
 
 | 
			
		||||
@@ -31,6 +31,6 @@
 | 
			
		||||
    "eslint-plugin-react-hooks": "^4.6.0",
 | 
			
		||||
    "prettier": "3.1.0",
 | 
			
		||||
    "typescript": "^5.0.2",
 | 
			
		||||
    "vite": "^4.5.2"
 | 
			
		||||
    "vite": "^4.5.3"
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										14
									
								
								apps/gpgui-helper/pnpm-lock.yaml
									
									
									
										generated
									
									
									
								
							
							
						
						
									
										14
									
								
								apps/gpgui-helper/pnpm-lock.yaml
									
									
									
										generated
									
									
									
								
							@@ -48,7 +48,7 @@ devDependencies:
 | 
			
		||||
    version: 6.12.0(eslint@8.54.0)(typescript@5.0.2)
 | 
			
		||||
  '@vitejs/plugin-react':
 | 
			
		||||
    specifier: ^4.0.3
 | 
			
		||||
    version: 4.0.3(vite@4.5.2)
 | 
			
		||||
    version: 4.0.3(vite@4.5.3)
 | 
			
		||||
  eslint:
 | 
			
		||||
    specifier: ^8.54.0
 | 
			
		||||
    version: 8.54.0
 | 
			
		||||
@@ -68,8 +68,8 @@ devDependencies:
 | 
			
		||||
    specifier: ^5.0.2
 | 
			
		||||
    version: 5.0.2
 | 
			
		||||
  vite:
 | 
			
		||||
    specifier: ^4.5.2
 | 
			
		||||
    version: 4.5.2(@types/node@20.8.10)
 | 
			
		||||
    specifier: ^4.5.3
 | 
			
		||||
    version: 4.5.3(@types/node@20.8.10)
 | 
			
		||||
 | 
			
		||||
packages:
 | 
			
		||||
 | 
			
		||||
@@ -1229,7 +1229,7 @@ packages:
 | 
			
		||||
    resolution: {integrity: sha512-zuVdFrMJiuCDQUMCzQaD6KL28MjnqqN8XnAqiEq9PNm/hCPTSGfrXCOfwj1ow4LFb/tNymJPwsNbVePc1xFqrQ==}
 | 
			
		||||
    dev: true
 | 
			
		||||
 | 
			
		||||
  /@vitejs/plugin-react@4.0.3(vite@4.5.2):
 | 
			
		||||
  /@vitejs/plugin-react@4.0.3(vite@4.5.3):
 | 
			
		||||
    resolution: {integrity: sha512-pwXDog5nwwvSIzwrvYYmA2Ljcd/ZNlcsSG2Q9CNDBwnsd55UGAyr2doXtB5j+2uymRCnCfExlznzzSFbBRcoCg==}
 | 
			
		||||
    engines: {node: ^14.18.0 || >=16.0.0}
 | 
			
		||||
    peerDependencies:
 | 
			
		||||
@@ -1239,7 +1239,7 @@ packages:
 | 
			
		||||
      '@babel/plugin-transform-react-jsx-self': 7.22.5(@babel/core@7.23.2)
 | 
			
		||||
      '@babel/plugin-transform-react-jsx-source': 7.22.5(@babel/core@7.23.2)
 | 
			
		||||
      react-refresh: 0.14.0
 | 
			
		||||
      vite: 4.5.2(@types/node@20.8.10)
 | 
			
		||||
      vite: 4.5.3(@types/node@20.8.10)
 | 
			
		||||
    transitivePeerDependencies:
 | 
			
		||||
      - supports-color
 | 
			
		||||
    dev: true
 | 
			
		||||
@@ -2979,8 +2979,8 @@ packages:
 | 
			
		||||
      punycode: 2.3.1
 | 
			
		||||
    dev: true
 | 
			
		||||
 | 
			
		||||
  /vite@4.5.2(@types/node@20.8.10):
 | 
			
		||||
    resolution: {integrity: sha512-tBCZBNSBbHQkaGyhGCDUGqeo2ph8Fstyp6FMSvTtsXeZSPpSMGlviAOav2hxVTqFcx8Hj/twtWKsMJXNY0xI8w==}
 | 
			
		||||
  /vite@4.5.3(@types/node@20.8.10):
 | 
			
		||||
    resolution: {integrity: sha512-kQL23kMeX92v3ph7IauVkXkikdDRsYMGTVl5KY2E9OY4ONLvkHf04MDTbnfo6NKxZiDLWzVpP5oTa8hQD8U3dg==}
 | 
			
		||||
    engines: {node: ^14.18.0 || >=16.0.0}
 | 
			
		||||
    hasBin: true
 | 
			
		||||
    peerDependencies:
 | 
			
		||||
 
 | 
			
		||||
@@ -9,6 +9,12 @@ use tauri::{Manager, Window};
 | 
			
		||||
 | 
			
		||||
use crate::downloader::{ChecksumFetcher, FileDownloader};
 | 
			
		||||
 | 
			
		||||
#[cfg(not(debug_assertions))]
 | 
			
		||||
const SNAPSHOT: &str = match option_env!("SNAPSHOT") {
 | 
			
		||||
    Some(val) => val,
 | 
			
		||||
    None => "false"
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
pub struct ProgressNotifier {
 | 
			
		||||
  win: Window,
 | 
			
		||||
}
 | 
			
		||||
@@ -81,9 +87,13 @@ impl GuiUpdater {
 | 
			
		||||
    info!("Update GUI, version: {}", self.version);
 | 
			
		||||
 | 
			
		||||
    #[cfg(debug_assertions)]
 | 
			
		||||
    let release_tag = "latest";
 | 
			
		||||
    let release_tag = "snapshot";
 | 
			
		||||
    #[cfg(not(debug_assertions))]
 | 
			
		||||
    let release_tag = format!("v{}", self.version);
 | 
			
		||||
    let release_tag = if SNAPSHOT == "true" {
 | 
			
		||||
      String::from("snapshot")
 | 
			
		||||
    } else {
 | 
			
		||||
      format!("v{}", self.version)
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    #[cfg(target_arch = "x86_64")]
 | 
			
		||||
    let arch = "x86_64";
 | 
			
		||||
@@ -91,8 +101,8 @@ impl GuiUpdater {
 | 
			
		||||
    let arch = "aarch64";
 | 
			
		||||
 | 
			
		||||
    let file_url = format!(
 | 
			
		||||
      "https://github.com/yuezk/GlobalProtect-openconnect/releases/download/{}/gpgui_{}_{}.bin.tar.xz",
 | 
			
		||||
      release_tag, self.version, arch
 | 
			
		||||
      "https://github.com/yuezk/GlobalProtect-openconnect/releases/download/{}/gpgui_{}.bin.tar.xz",
 | 
			
		||||
      release_tag, arch
 | 
			
		||||
    );
 | 
			
		||||
    let checksum_url = format!("{}.sha256", file_url);
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
@@ -1,13 +1,17 @@
 | 
			
		||||
use anyhow::bail;
 | 
			
		||||
use log::{info, warn};
 | 
			
		||||
use regex::Regex;
 | 
			
		||||
use serde::{Deserialize, Serialize};
 | 
			
		||||
 | 
			
		||||
use crate::{error::AuthDataParseError, utils::base64::decode_to_string};
 | 
			
		||||
 | 
			
		||||
#[derive(Debug, Serialize, Deserialize)]
 | 
			
		||||
#[serde(rename_all = "camelCase")]
 | 
			
		||||
pub struct SamlAuthData {
 | 
			
		||||
  #[serde(alias = "un")]
 | 
			
		||||
  username: String,
 | 
			
		||||
  prelogin_cookie: Option<String>,
 | 
			
		||||
  portal_userauthcookie: Option<String>,
 | 
			
		||||
  token: Option<String>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Debug, Serialize, Deserialize)]
 | 
			
		||||
@@ -32,10 +36,11 @@ impl SamlAuthData {
 | 
			
		||||
      username,
 | 
			
		||||
      prelogin_cookie,
 | 
			
		||||
      portal_userauthcookie,
 | 
			
		||||
      token: None,
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  pub fn parse_html(html: &str) -> anyhow::Result<SamlAuthData> {
 | 
			
		||||
  pub fn from_html(html: &str) -> anyhow::Result<SamlAuthData, AuthDataParseError> {
 | 
			
		||||
    match parse_xml_tag(html, "saml-auth-status") {
 | 
			
		||||
      Some(saml_status) if saml_status == "1" => {
 | 
			
		||||
        let username = parse_xml_tag(html, "saml-username");
 | 
			
		||||
@@ -43,21 +48,42 @@ impl SamlAuthData {
 | 
			
		||||
        let portal_userauthcookie = parse_xml_tag(html, "portal-userauthcookie");
 | 
			
		||||
 | 
			
		||||
        if SamlAuthData::check(&username, &prelogin_cookie, &portal_userauthcookie) {
 | 
			
		||||
          return Ok(SamlAuthData::new(
 | 
			
		||||
          Ok(SamlAuthData::new(
 | 
			
		||||
            username.unwrap(),
 | 
			
		||||
            prelogin_cookie,
 | 
			
		||||
            portal_userauthcookie,
 | 
			
		||||
          ));
 | 
			
		||||
          ))
 | 
			
		||||
        } else {
 | 
			
		||||
          Err(AuthDataParseError::Invalid)
 | 
			
		||||
        }
 | 
			
		||||
      }
 | 
			
		||||
      Some(_) => Err(AuthDataParseError::Invalid),
 | 
			
		||||
      None => Err(AuthDataParseError::NotFound),
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
        bail!("Found invalid auth data in HTML");
 | 
			
		||||
      }
 | 
			
		||||
      Some(status) => {
 | 
			
		||||
        bail!("Found invalid SAML status {} in HTML", status);
 | 
			
		||||
      }
 | 
			
		||||
      None => {
 | 
			
		||||
        bail!("No auth data found in HTML");
 | 
			
		||||
      }
 | 
			
		||||
  pub fn from_gpcallback(data: &str) -> anyhow::Result<SamlAuthData, AuthDataParseError> {
 | 
			
		||||
    let auth_data = data.trim_start_matches("globalprotectcallback:");
 | 
			
		||||
 | 
			
		||||
    if auth_data.starts_with("cas-as") {
 | 
			
		||||
      info!("Got CAS auth data from globalprotectcallback");
 | 
			
		||||
 | 
			
		||||
      let auth_data: SamlAuthData = serde_urlencoded::from_str(auth_data).map_err(|e| {
 | 
			
		||||
        warn!("Failed to parse token auth data: {}", e);
 | 
			
		||||
        AuthDataParseError::Invalid
 | 
			
		||||
      })?;
 | 
			
		||||
 | 
			
		||||
      Ok(auth_data)
 | 
			
		||||
    } else {
 | 
			
		||||
      info!("Parsing SAML auth data...");
 | 
			
		||||
 | 
			
		||||
      let auth_data = decode_to_string(auth_data).map_err(|e| {
 | 
			
		||||
        warn!("Failed to decode SAML auth data: {}", e);
 | 
			
		||||
        AuthDataParseError::Invalid
 | 
			
		||||
      })?;
 | 
			
		||||
      let auth_data = Self::from_html(&auth_data)?;
 | 
			
		||||
 | 
			
		||||
      Ok(auth_data)
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
@@ -69,6 +95,10 @@ impl SamlAuthData {
 | 
			
		||||
    self.prelogin_cookie.as_deref()
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  pub fn token(&self) -> Option<&str> {
 | 
			
		||||
    self.token.as_deref()
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  pub fn check(
 | 
			
		||||
    username: &Option<String>,
 | 
			
		||||
    prelogin_cookie: &Option<String>,
 | 
			
		||||
@@ -78,7 +108,16 @@ impl SamlAuthData {
 | 
			
		||||
    let prelogin_cookie_valid = prelogin_cookie.as_ref().is_some_and(|val| val.len() > 5);
 | 
			
		||||
    let portal_userauthcookie_valid = portal_userauthcookie.as_ref().is_some_and(|val| val.len() > 5);
 | 
			
		||||
 | 
			
		||||
    username_valid && (prelogin_cookie_valid || portal_userauthcookie_valid)
 | 
			
		||||
    let is_valid = username_valid && (prelogin_cookie_valid || portal_userauthcookie_valid);
 | 
			
		||||
 | 
			
		||||
    if !is_valid {
 | 
			
		||||
      warn!(
 | 
			
		||||
        "Invalid SAML auth data: username: {:?}, prelogin-cookie: {:?}, portal-userauthcookie: {:?}",
 | 
			
		||||
        username, prelogin_cookie, portal_userauthcookie
 | 
			
		||||
      );
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    is_valid
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
@@ -88,3 +127,28 @@ pub fn parse_xml_tag(html: &str, tag: &str) -> Option<String> {
 | 
			
		||||
    .and_then(|captures| captures.get(1))
 | 
			
		||||
    .map(|m| m.as_str().to_string())
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[cfg(test)]
 | 
			
		||||
mod tests {
 | 
			
		||||
  use super::*;
 | 
			
		||||
 | 
			
		||||
  #[test]
 | 
			
		||||
  fn auth_data_from_gpcallback_cas() {
 | 
			
		||||
    let auth_data = "globalprotectcallback:cas-as=1&un=xyz@email.com&token=very_long_string";
 | 
			
		||||
 | 
			
		||||
    let auth_data = SamlAuthData::from_gpcallback(auth_data).unwrap();
 | 
			
		||||
 | 
			
		||||
    assert_eq!(auth_data.username(), "xyz@email.com");
 | 
			
		||||
    assert_eq!(auth_data.token(), Some("very_long_string"));
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  #[test]
 | 
			
		||||
  fn auth_data_from_gpcallback_non_cas() {
 | 
			
		||||
    let auth_data = "PGh0bWw+PCEtLSA8c2FtbC1hdXRoLXN0YXR1cz4xPC9zYW1sLWF1dGgtc3RhdHVzPjxwcmVsb2dpbi1jb29raWU+cHJlbG9naW4tY29va2llPC9wcmVsb2dpbi1jb29raWU+PHNhbWwtdXNlcm5hbWU+eHl6QGVtYWlsLmNvbTwvc2FtbC11c2VybmFtZT48c2FtbC1zbG8+bm88L3NhbWwtc2xvPjxzYW1sLVNlc3Npb25Ob3RPbk9yQWZ0ZXI+PC9zYW1sLVNlc3Npb25Ob3RPbk9yQWZ0ZXI+IC0tPjwvaHRtbD4=";
 | 
			
		||||
 | 
			
		||||
    let auth_data = SamlAuthData::from_gpcallback(auth_data).unwrap();
 | 
			
		||||
 | 
			
		||||
    assert_eq!(auth_data.username(), "xyz@email.com");
 | 
			
		||||
    assert_eq!(auth_data.prelogin_cookie(), Some("prelogin-cookie"));
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -3,7 +3,7 @@ use std::collections::HashMap;
 | 
			
		||||
use serde::{Deserialize, Serialize};
 | 
			
		||||
use specta::Type;
 | 
			
		||||
 | 
			
		||||
use crate::{auth::SamlAuthData, utils::base64::decode_to_string};
 | 
			
		||||
use crate::auth::SamlAuthData;
 | 
			
		||||
 | 
			
		||||
#[derive(Debug, Serialize, Deserialize, Type, Clone)]
 | 
			
		||||
#[serde(rename_all = "camelCase")]
 | 
			
		||||
@@ -39,14 +39,16 @@ impl From<&CachedCredential> for PasswordCredential {
 | 
			
		||||
#[serde(rename_all = "camelCase")]
 | 
			
		||||
pub struct PreloginCookieCredential {
 | 
			
		||||
  username: String,
 | 
			
		||||
  prelogin_cookie: String,
 | 
			
		||||
  prelogin_cookie: Option<String>,
 | 
			
		||||
  token: Option<String>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl PreloginCookieCredential {
 | 
			
		||||
  pub fn new(username: &str, prelogin_cookie: &str) -> Self {
 | 
			
		||||
  pub fn new(username: &str, prelogin_cookie: Option<&str>, token: Option<&str>) -> Self {
 | 
			
		||||
    Self {
 | 
			
		||||
      username: username.to_string(),
 | 
			
		||||
      prelogin_cookie: prelogin_cookie.to_string(),
 | 
			
		||||
      prelogin_cookie: prelogin_cookie.map(|s| s.to_string()),
 | 
			
		||||
      token: token.map(|s| s.to_string()),
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
@@ -54,22 +56,22 @@ impl PreloginCookieCredential {
 | 
			
		||||
    &self.username
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  pub fn prelogin_cookie(&self) -> &str {
 | 
			
		||||
    &self.prelogin_cookie
 | 
			
		||||
  pub fn prelogin_cookie(&self) -> Option<&str> {
 | 
			
		||||
    self.prelogin_cookie.as_deref()
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  pub fn token(&self) -> Option<&str> {
 | 
			
		||||
    self.token.as_deref()
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl TryFrom<SamlAuthData> for PreloginCookieCredential {
 | 
			
		||||
  type Error = anyhow::Error;
 | 
			
		||||
 | 
			
		||||
  fn try_from(value: SamlAuthData) -> Result<Self, Self::Error> {
 | 
			
		||||
impl From<SamlAuthData> for PreloginCookieCredential {
 | 
			
		||||
  fn from(value: SamlAuthData) -> Self {
 | 
			
		||||
    let username = value.username().to_string();
 | 
			
		||||
    let prelogin_cookie = value
 | 
			
		||||
      .prelogin_cookie()
 | 
			
		||||
      .ok_or_else(|| anyhow::anyhow!("Missing prelogin cookie"))?
 | 
			
		||||
      .to_string();
 | 
			
		||||
    let prelogin_cookie = value.prelogin_cookie();
 | 
			
		||||
    let token = value.token();
 | 
			
		||||
 | 
			
		||||
    Ok(Self::new(&username, &prelogin_cookie))
 | 
			
		||||
    Self::new(&username, prelogin_cookie, token)
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
@@ -154,7 +156,6 @@ impl From<PasswordCredential> for CachedCredential {
 | 
			
		||||
    )
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Debug, Serialize, Deserialize, Type, Clone)]
 | 
			
		||||
#[serde(tag = "type", rename_all = "camelCase")]
 | 
			
		||||
pub enum Credential {
 | 
			
		||||
@@ -165,15 +166,12 @@ pub enum Credential {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Credential {
 | 
			
		||||
  /// Create a credential from a globalprotectcallback:<base64 encoded string>
 | 
			
		||||
  pub fn parse_gpcallback(auth_data: &str) -> anyhow::Result<Self> {
 | 
			
		||||
    // Remove the surrounding quotes
 | 
			
		||||
    let auth_data = auth_data.trim_matches('"');
 | 
			
		||||
    let auth_data = auth_data.trim_start_matches("globalprotectcallback:");
 | 
			
		||||
    let auth_data = decode_to_string(auth_data)?;
 | 
			
		||||
    let auth_data = SamlAuthData::parse_html(&auth_data)?;
 | 
			
		||||
  /// Create a credential from a globalprotectcallback:<base64 encoded string>,
 | 
			
		||||
  /// or globalprotectcallback:cas-as=1&un=user@xyz.com&token=very_long_string
 | 
			
		||||
  pub fn from_gpcallback(auth_data: &str) -> anyhow::Result<Self> {
 | 
			
		||||
    let auth_data = SamlAuthData::from_gpcallback(auth_data)?;
 | 
			
		||||
 | 
			
		||||
    Self::try_from(auth_data)
 | 
			
		||||
    Ok(Self::from(auth_data))
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  pub fn username(&self) -> &str {
 | 
			
		||||
@@ -189,20 +187,22 @@ impl Credential {
 | 
			
		||||
    let mut params = HashMap::new();
 | 
			
		||||
    params.insert("user", self.username());
 | 
			
		||||
 | 
			
		||||
    let (passwd, prelogin_cookie, portal_userauthcookie, portal_prelogonuserauthcookie) = match self {
 | 
			
		||||
      Credential::Password(cred) => (Some(cred.password()), None, None, None),
 | 
			
		||||
      Credential::PreloginCookie(cred) => (None, Some(cred.prelogin_cookie()), None, None),
 | 
			
		||||
    let (passwd, prelogin_cookie, portal_userauthcookie, portal_prelogonuserauthcookie, token) = match self {
 | 
			
		||||
      Credential::Password(cred) => (Some(cred.password()), None, None, None, None),
 | 
			
		||||
      Credential::PreloginCookie(cred) => (None, cred.prelogin_cookie(), None, None, cred.token()),
 | 
			
		||||
      Credential::AuthCookie(cred) => (
 | 
			
		||||
        None,
 | 
			
		||||
        None,
 | 
			
		||||
        Some(cred.user_auth_cookie()),
 | 
			
		||||
        Some(cred.prelogon_user_auth_cookie()),
 | 
			
		||||
        None,
 | 
			
		||||
      ),
 | 
			
		||||
      Credential::CachedCredential(cred) => (
 | 
			
		||||
        cred.password(),
 | 
			
		||||
        None,
 | 
			
		||||
        Some(cred.auth_cookie.user_auth_cookie()),
 | 
			
		||||
        Some(cred.auth_cookie.prelogon_user_auth_cookie()),
 | 
			
		||||
        None,
 | 
			
		||||
      ),
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
@@ -214,17 +214,19 @@ impl Credential {
 | 
			
		||||
      portal_prelogonuserauthcookie.unwrap_or_default(),
 | 
			
		||||
    );
 | 
			
		||||
 | 
			
		||||
    if let Some(token) = token {
 | 
			
		||||
      params.insert("token", token);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    params
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl TryFrom<SamlAuthData> for Credential {
 | 
			
		||||
  type Error = anyhow::Error;
 | 
			
		||||
impl From<SamlAuthData> for Credential {
 | 
			
		||||
  fn from(value: SamlAuthData) -> Self {
 | 
			
		||||
    let cred = PreloginCookieCredential::from(value);
 | 
			
		||||
 | 
			
		||||
  fn try_from(value: SamlAuthData) -> Result<Self, Self::Error> {
 | 
			
		||||
    let prelogin_cookie = PreloginCookieCredential::try_from(value)?;
 | 
			
		||||
 | 
			
		||||
    Ok(Self::PreloginCookie(prelogin_cookie))
 | 
			
		||||
    Self::PreloginCookie(cred)
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
@@ -9,3 +9,11 @@ pub enum PortalError {
 | 
			
		||||
  #[error("Network error: {0}")]
 | 
			
		||||
  NetworkError(String),
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[derive(Error, Debug)]
 | 
			
		||||
pub enum AuthDataParseError {
 | 
			
		||||
  #[error("No auth data found")]
 | 
			
		||||
  NotFound,
 | 
			
		||||
  #[error("Invalid auth data")]
 | 
			
		||||
  Invalid,
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -51,7 +51,6 @@ pub struct GpParams {
 | 
			
		||||
  client_version: Option<String>,
 | 
			
		||||
  computer: String,
 | 
			
		||||
  ignore_tls_errors: bool,
 | 
			
		||||
  prefer_default_browser: bool,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl GpParams {
 | 
			
		||||
@@ -79,10 +78,6 @@ impl GpParams {
 | 
			
		||||
    self.ignore_tls_errors
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  pub fn prefer_default_browser(&self) -> bool {
 | 
			
		||||
    self.prefer_default_browser
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  pub fn client_os(&self) -> &str {
 | 
			
		||||
    self.client_os.as_str()
 | 
			
		||||
  }
 | 
			
		||||
@@ -131,7 +126,6 @@ pub struct GpParamsBuilder {
 | 
			
		||||
  client_version: Option<String>,
 | 
			
		||||
  computer: String,
 | 
			
		||||
  ignore_tls_errors: bool,
 | 
			
		||||
  prefer_default_browser: bool,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl GpParamsBuilder {
 | 
			
		||||
@@ -144,7 +138,6 @@ impl GpParamsBuilder {
 | 
			
		||||
      client_version: Default::default(),
 | 
			
		||||
      computer: whoami::hostname(),
 | 
			
		||||
      ignore_tls_errors: false,
 | 
			
		||||
      prefer_default_browser: false,
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
@@ -183,11 +176,6 @@ impl GpParamsBuilder {
 | 
			
		||||
    self
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  pub fn prefer_default_browser(&mut self, prefer_default_browser: bool) -> &mut Self {
 | 
			
		||||
    self.prefer_default_browser = prefer_default_browser;
 | 
			
		||||
    self
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  pub fn build(&self) -> GpParams {
 | 
			
		||||
    GpParams {
 | 
			
		||||
      is_gateway: self.is_gateway,
 | 
			
		||||
@@ -197,7 +185,6 @@ impl GpParamsBuilder {
 | 
			
		||||
      client_version: self.client_version.clone(),
 | 
			
		||||
      computer: self.computer.clone(),
 | 
			
		||||
      ignore_tls_errors: self.ignore_tls_errors,
 | 
			
		||||
      prefer_default_browser: self.prefer_default_browser,
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -1,4 +1,4 @@
 | 
			
		||||
use anyhow::bail;
 | 
			
		||||
use anyhow::{anyhow, bail};
 | 
			
		||||
use log::{info, warn};
 | 
			
		||||
use reqwest::{Client, StatusCode};
 | 
			
		||||
use roxmltree::Document;
 | 
			
		||||
@@ -107,12 +107,13 @@ pub async fn prelogin(portal: &str, gp_params: &GpParams) -> anyhow::Result<Prel
 | 
			
		||||
  let mut params = gp_params.to_params();
 | 
			
		||||
 | 
			
		||||
  params.insert("tmp", "tmp");
 | 
			
		||||
  if gp_params.prefer_default_browser() {
 | 
			
		||||
  params.insert("default-browser", "1");
 | 
			
		||||
  }
 | 
			
		||||
  params.insert("cas-support", "yes");
 | 
			
		||||
 | 
			
		||||
  params.retain(|k, _| REQUIRED_PARAMS.iter().any(|required_param| required_param == k));
 | 
			
		||||
 | 
			
		||||
  info!("Prelogin with params: {:?}", params);
 | 
			
		||||
 | 
			
		||||
  let client = Client::builder()
 | 
			
		||||
    .danger_accept_invalid_certs(gp_params.ignore_tls_errors())
 | 
			
		||||
    .user_agent(user_agent)
 | 
			
		||||
@@ -124,8 +125,8 @@ pub async fn prelogin(portal: &str, gp_params: &GpParams) -> anyhow::Result<Prel
 | 
			
		||||
    .send()
 | 
			
		||||
    .await
 | 
			
		||||
    .map_err(|e| anyhow::anyhow!(PortalError::NetworkError(e.to_string())))?;
 | 
			
		||||
  let status = res.status();
 | 
			
		||||
 | 
			
		||||
  let status = res.status();
 | 
			
		||||
  if status == StatusCode::NOT_FOUND {
 | 
			
		||||
    bail!(PortalError::PreloginError("Prelogin endpoint not found".to_string()))
 | 
			
		||||
  }
 | 
			
		||||
@@ -196,8 +197,8 @@ fn parse_res_xml(res_xml: String, is_gateway: bool) -> anyhow::Result<Prelogin>
 | 
			
		||||
      label_password: label_password.unwrap(),
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    return Ok(Prelogin::Standard(standard_prelogin));
 | 
			
		||||
    Ok(Prelogin::Standard(standard_prelogin))
 | 
			
		||||
  } else {
 | 
			
		||||
    Err(anyhow!("Invalid prelogin response"))
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  bail!("Invalid prelogin response");
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -135,7 +135,7 @@ impl<'a> SamlAuthLauncher<'a> {
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    match auth_result {
 | 
			
		||||
      SamlAuthResult::Success(auth_data) => Credential::try_from(auth_data),
 | 
			
		||||
      SamlAuthResult::Success(auth_data) => Ok(Credential::from(auth_data)),
 | 
			
		||||
      SamlAuthResult::Failure(msg) => bail!(msg),
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
 
 | 
			
		||||
@@ -5,7 +5,10 @@ install:
 | 
			
		||||
	install -Dm755 artifacts/usr/bin/gpservice $(DESTDIR)/usr/bin/gpservice
 | 
			
		||||
	install -Dm755 artifacts/usr/bin/gpauth $(DESTDIR)/usr/bin/gpauth
 | 
			
		||||
	install -Dm755 artifacts/usr/bin/gpgui-helper $(DESTDIR)/usr/bin/gpgui-helper
 | 
			
		||||
	install -Dm755 artifacts/usr/bin/gpgui $(DESTDIR)/usr/bin/gpgui
 | 
			
		||||
 | 
			
		||||
	if [ -f artifacts/usr/bin/gpgui ]; then \
 | 
			
		||||
		install -Dm755 artifacts/usr/bin/gpgui $(DESTDIR)/usr/bin/gpgui; \
 | 
			
		||||
	fi
 | 
			
		||||
 | 
			
		||||
	install -Dm644 artifacts/usr/share/applications/gpgui.desktop $(DESTDIR)/usr/share/applications/gpgui.desktop
 | 
			
		||||
	install -Dm644 artifacts/usr/share/icons/hicolor/scalable/apps/gpgui.svg $(DESTDIR)/usr/share/icons/hicolor/scalable/apps/gpgui.svg
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user