Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fix(windows): don't send key twice #280

Merged
merged 3 commits into from
Sep 1, 2023
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 5 additions & 1 deletion .github/workflows/ci.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,8 @@ jobs:
run: cargo clippy --all-targets --all-features -- -D warnings

- name: Install npcap on Windows
if: matrix.os == 'windows-latest'
# PRs cannot not be trusted with repository secrets
if: (matrix.os == 'windows-latest') && (github.event_name != 'pull_request')
env:
NPCAP_OEM_URL: ${{ secrets.NPCAP_OEM_URL }}
run: |
Expand All @@ -62,6 +63,9 @@ jobs:
& "$env:TEMP/npcap-oem.exe" /S

- name: Run tests
# npcap is needed to run tests on Windows, so this is an unfortunate
# sacrifice we have to make in the name of security
if: (matrix.os != 'windows-latest') || (github.event_name != 'pull_request')
run: cargo test --verbose

- name: Upload unix binary
Expand Down
2 changes: 1 addition & 1 deletion .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -11,4 +11,4 @@ debian/*
!debian/control
!debian/copyright
!debian/rules
!debian/source
!debian/source
1 change: 1 addition & 0 deletions Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -50,5 +50,6 @@ pnet_base = "0.34.0"
packet-builder = { version = "0.7.0", git = "https://github.com/cyqsimon/packet_builder.git", branch = "patch-update" }

[target.'cfg(target_os="windows")'.build-dependencies]
anyhow = "1.0.75"
http_req = "0.9.2"
zip = "0.6.6"
78 changes: 53 additions & 25 deletions build.rs
Original file line number Diff line number Diff line change
@@ -1,50 +1,78 @@
fn main() {
#[cfg(target_os = "windows")]
download_windows_pcap_sdk()
download_windows_npcap_sdk().unwrap();
}

#[cfg(target_os = "windows")]
fn download_windows_pcap_sdk() {
fn download_windows_npcap_sdk() -> anyhow::Result<()> {
use std::{
env, fs,
io::{self, Write},
path::PathBuf,
};

use anyhow::anyhow;
use http_req::request;
use zip::ZipArchive;

println!("cargo:rerun-if-changed=build.rs");

let out_dir = env::var("OUT_DIR").unwrap();
// get npcap SDK
const NPCAP_SDK: &str = "npcap-sdk-1.13.zip";

let mut pcap_zip = Vec::new();
let res = request::get("https://npcap.com/dist/npcap-sdk-1.13.zip", &mut pcap_zip).unwrap();
eprintln!("{res:?}");
let npcap_sdk_download_url = format!("https://npcap.com/dist/{NPCAP_SDK}");
let cache_dir = PathBuf::from(env::var("CARGO_MANIFEST_DIR")?).join("target");
let npcap_sdk_cache_path = cache_dir.join(NPCAP_SDK);

let lib_dir = if cfg!(target_arch = "aarch64") {
"Lib/ARM64"
let npcap_zip = match fs::read(&npcap_sdk_cache_path) {
// use cached
Ok(zip_data) => {
eprintln!("Found cached npcap SDK");
zip_data
}
// download SDK
Err(_) => {
eprintln!("Downloading npcap SDK");

// download
let mut zip_data = vec![];
let _res = request::get(npcap_sdk_download_url, &mut zip_data)?;

// write cache
fs::create_dir_all(cache_dir)?;
let mut cache = fs::File::create(npcap_sdk_cache_path)?;
cache.write_all(&zip_data)?;

zip_data
}
};

// extract DLL
let lib_path = if cfg!(target_arch = "aarch64") {
"Lib/ARM64/Packet.lib"
} else if cfg!(target_arch = "x86_64") {
"Lib/x64"
"Lib/x64/Packet.lib"
} else if cfg!(target_arch = "x86") {
"Lib"
"Lib/Packet.lib"
} else {
panic!("Unsupported target!")
};
let lib_name = "Packet.lib";
let lib_path = format!("{lib_dir}/{lib_name}");

let mut archive = ZipArchive::new(io::Cursor::new(pcap_zip)).unwrap();
let mut pcap_lib = match archive.by_name(&lib_path) {
Ok(lib) => lib,
Err(err) => {
panic!("{err}");
}
};
let mut archive = ZipArchive::new(io::Cursor::new(npcap_zip))?;
let mut npcap_lib = archive.by_name(lib_path)?;

// write DLL
let lib_dir = PathBuf::from(env::var("OUT_DIR")?).join("npcap_sdk");
let lib_path = lib_dir.join("Packet.lib");
fs::create_dir_all(&lib_dir)?;
let mut lib_file = fs::File::create(lib_path)?;
io::copy(&mut npcap_lib, &mut lib_file)?;

fs::create_dir_all(format!("{out_dir}/{lib_dir}")).unwrap();
let mut pcap_lib_file = fs::File::create(format!("{out_dir}/{lib_path}")).unwrap();
io::copy(&mut pcap_lib, &mut pcap_lib_file).unwrap();
pcap_lib_file.flush().unwrap();
println!(
"cargo:rustc-link-search=native={}",
lib_dir
.to_str()
.ok_or(anyhow!("{lib_dir:?} is not valid UTF-8"))?
);

println!("cargo:rustc-link-search=native={out_dir}/{lib_dir}");
Ok(())
}
6 changes: 5 additions & 1 deletion src/main.rs
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ use std::{
};

use crossterm::{
event::{Event, KeyCode, KeyEvent, KeyModifiers},
event::{Event, KeyCode, KeyEvent, KeyEventKind, KeyModifiers},
terminal,
};
use display::{elapsed_time, RawTerminalBackend, Ui};
Expand Down Expand Up @@ -229,11 +229,13 @@ where
Event::Key(KeyEvent {
modifiers: KeyModifiers::CONTROL,
code: KeyCode::Char('c'),
kind: KeyEventKind::Press,
..
})
| Event::Key(KeyEvent {
modifiers: KeyModifiers::NONE,
code: KeyCode::Char('q'),
kind: KeyEventKind::Press,
..
}) => {
running.store(false, Ordering::Release);
Expand All @@ -253,6 +255,7 @@ where
Event::Key(KeyEvent {
modifiers: KeyModifiers::NONE,
code: KeyCode::Char(' '),
kind: KeyEventKind::Press,
..
}) => {
let restarting = paused.fetch_xor(true, Ordering::SeqCst);
Expand All @@ -272,6 +275,7 @@ where
Event::Key(KeyEvent {
modifiers: KeyModifiers::NONE,
code: KeyCode::Tab,
kind: KeyEventKind::Press,
..
}) => {
let paused = paused.load(Ordering::SeqCst);
Expand Down