termux-packages/packages/tergent/0001-Automatically-unlock-k...

107 lines
4.0 KiB
Diff

From 87ab008deb2bd7701da6e91f3998e51e68ec2bcf Mon Sep 17 00:00:00 2001
From: Daniil Gentili <daniil@daniil.it>
Date: Wed, 30 Sep 2020 18:21:05 +0200
Subject: [PATCH 1/2] Automatically unlock keystore using fingerprint, if
possible
---
src/bridge/mod.rs | 51 ++++++++++++++++++++++++++++++++++++++++-------
1 file changed, 44 insertions(+), 7 deletions(-)
diff --git a/src/bridge/mod.rs b/src/bridge/mod.rs
index 4b00164..9e6a8bc 100644
--- a/src/bridge/mod.rs
+++ b/src/bridge/mod.rs
@@ -7,12 +7,15 @@ use std::error::Error;
use std::io::{Read, Write};
use std::process::{Command, Stdio};
+use serde_json;
+use serde_json::{Value};
+
use base64;
/// Send a request to `termux-api` to list all the keys.
/// Returns a string that contains a JSON object.
pub fn list_keys() -> Result<String, Box<dyn Error>> {
- Ok(communicate(&["list", "--ez", "detailed", "true"], &[0; 0])?)
+ Ok(communicate(&"Keystore", &["-e", "command", "list", "--ez", "detailed", "true"], &[0; 0])?)
}
/// Send some data to `termux-api` to be signed.
@@ -23,16 +26,48 @@ pub fn list_keys() -> Result<String, Box<dyn Error>> {
///
/// [Signature algorithms]:
/// https://docs.oracle.com/javase/8/docs/technotes/guides/security/StandardNames.html#Signature
+fn sign_internal(alias: &str, algorithm: &str, data: &[u8]) -> Result<Vec<u8>, Box<dyn Error>> {
+ let args = [
+ "-e", "command", "sign",
+ "-e", "alias", alias,
+ "-e", "algorithm", algorithm
+ ];
+ let output = communicate(&"Keystore", &args, data)?;
+ let res = base64::decode(output)?;
+ if res.len() == 0 {
+ return Err("Could not sign!".into())
+ }
+ Ok(res)
+}
+
+pub fn unlock() -> Result<(), Box<dyn Error>> {
+ let args = [
+ "--es", "title", "Tergent",
+ "--es", "description", "Use your fingerprint to unlock the keystore",
+ ];
+ let json = communicate(&"Fingerprint", &args, &[0; 0])?;
+ let decoded: Value = serde_json::from_str::<serde_json::Value>(&json)?;
+ let res = decoded["auth_result"].as_str().ok_or("Invalid result")?;
+ if res == "AUTH_RESULT_FAILURE" {
+ return Err("Fingerprint authentication failed".into())
+ }
+ Ok(())
+}
+
pub fn sign(alias: &str, algorithm: &str, data: &[u8]) -> Result<Vec<u8>, Box<dyn Error>> {
- let args = ["sign", "-e", "alias", alias, "-e", "algorithm", algorithm];
- let output = communicate(&args, data)?;
- return Ok(base64::decode(output)?);
+ match sign_internal(&alias, &algorithm, &data) {
+ Ok(res) => Ok(res),
+ Err(_) => match unlock() {
+ Ok(_) => Ok(sign_internal(&alias, &algorithm, &data)?),
+ Err(e) => Err(e)
+ }
+ }
}
/// Performs a generic call to `termux-api`, providing `args` to its receiver.
/// Sets up proper sockets so that the `input` is provided to `termux-api` and
/// its output is returned from this function.
-fn communicate(args: &[&str], input: &[u8]) -> Result<String, Box<dyn Error>> {
+fn communicate(method: &str, args: &[&str], input: &[u8]) -> Result<String, Box<dyn Error>> {
let mut input_socket = socket::Socket::new()?;
let mut output_socket = socket::Socket::new()?;
@@ -43,7 +78,7 @@ fn communicate(args: &[&str], input: &[u8]) -> Result<String, Box<dyn Error>> {
.args(&["-n", "com.termux.api/.TermuxApiReceiver"])
.args(&["--es", "socket_input", &output_socket.address()])
.args(&["--es", "socket_output", &input_socket.address()])
- .args(&["--es", "api_method", "Keystore", "-e", "command"])
+ .args(&["--es", "api_method", method])
.args(args)
.stdin(Stdio::null())
.stdout(Stdio::null())
@@ -65,6 +100,8 @@ fn communicate(args: &[&str], input: &[u8]) -> Result<String, Box<dyn Error>> {
input_socket.close()?;
// We need to reap our children otherwise they will stay as zombies.
- command.wait()?;
+ // Ignore result, since this may error if the process has already closed
+ command.wait();
+
Ok(output)
}
--
2.30.1