liyunjia/chuanyue-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修改公钥解析过程

Browse Source
This commit is contained in:
李运家 2024-05-29 12:34:46 +08:00
parent 47df2cf3cd
commit 4e0c5fbfdc
1 changed files with 26 additions and 46 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

72
library/src/social/google.rs
View File

@ -3,6 +3,7 @@ use std::collections::HashMap;
use chrono::Utc;
use jsonwebtoken::{decode, Algorithm, DecodingKey, Validation};
use reqwest::Client;
use serde::Deserialize;
use serde_json::Value;
use crate::resp::response::ResErr;
@ -92,53 +93,34 @@ impl From<Value> for GoogleJwtProfile {
}
}
// 异步获取并解析Google公钥
async fn fetch_and_parse_google_public_keys() -> SocialResult<HashMap<String, Value>> {
let response = Client::new().get(GOOGLE_PUBLIC_CERT_URL).send().await?;
let google_keys: Value = response.json().await?;
#[derive(Deserialize, Debug, Clone)]
struct GooglePublicKey {
e: String,
n: String,
kty: String,
use_: String,
alg: String,
kid: String,
}
/* Google公钥结构如下：
{
"keys": [
{
"e": "AQAB",
"n": "xjWd1j8GmmWzuz732haG9HECXsSZBvxOBLph3FQhk_tplhWloI1ywx-RdopUZt1lndbOM9n99lZJkpQyNJ1sdy7JFgYLjqj-wtHdEaQlBGEQtmkW8zUjr_N3bmpsxGbPzOzlKe3qddtoxXvn9rI_RvHfJD1YY-6kayQeyPOBz_4ML1lvI_JHV-Bb1MSmSk3WaAh5PzeqleusmUT87Gqfu02cOPrY8cwugqo65D6-wzAEeVvceV8-c36TMoLU5csU05GBVplgd6Ouuw35ZsETG4si4QQJztC3KsZ4jhYM-aJ3jeFPt0r3cQooiXdZBp3JkXSpE-UUaOVPsXo7WiVmww",
"kty": "RSA",
"use": "sig",
"alg": "RS256",
"kid": "323b214ae6975a0f034ea77354dc0c25d03642dc"
},
{
"kid": "6719678351a5faedc2e70274bbea62da2a8c4a12",
"n": "oAP5OnSzKfkEV2QMm2XCuu4G8VGRBOyhKg-4H04WzYzPqM_Tmqi60Vod96JTo7SfM0OoGeNnlkWNjjBWkSS66alNLrvTNLi0A-KGeBsZiIFmrbsP6HHJfFzPd0Mci7-e11fNKecZgbC1me9PtRXFZb9JprZGFOvBiMwU0rRvh0GWYmTFj1HFjOIMAwTGOKOVGNuPjv0b3V0YaAkUNklzi4MM6qgzUb0tE0so1Ii7kBe7roMScS2USPeeJkeoPjLEbQcrT8MxOSxH-JgPLfq-zOnEJ6ERW3mtXdZCNzqmVLn5yjX5lKr5E2vgkPAHx9NLZ09fo_L9woeX_5epl6cIkQ",
"use": "sig",
"alg": "RS256",
"kty": "RSA",
"e": "AQAB"
}
]
}
*/
tracing::info!("Google公钥获取成功, {}", google_keys);
#[derive(Deserialize, Debug, Clone)]
struct GooglePublicKeys {
keys: Vec<GooglePublicKey>,
}
// 异步获取并解析Google公钥
async fn fetch_and_parse_google_public_keys() -> SocialResult<HashMap<String, GooglePublicKey>> {
let response = Client::new().get(GOOGLE_PUBLIC_CERT_URL).send().await?;
let google_keys: GooglePublicKeys = response.json().await?;
tracing::info!("Google公钥获取成功, {:?}", google_keys);
let mut key_map = HashMap::new();
// 解析公钥
google_keys
.get("keys")
.and_then(serde_json::Value::as_array)
.unwrap()
.iter()
.for_each(|key| {
if key
.get("kty")
.map(|kty| kty.as_str() == Some("RSA"))
.unwrap()
{
let n = key.get("n").unwrap().as_str().unwrap();
let kid = key.get("kid").unwrap().as_str().unwrap();
key_map.insert(kid.to_string(), key.clone());
}
});
for key in google_keys.keys.iter() {
if key.kty == "RSA" {
key_map.insert(key.kid.to_owned(), key.to_owned());
}
}
tracing::info!("Google公钥解析成功, {:?}", key_map);
Ok(key_map)
@ -165,8 +147,6 @@ pub async fn verify_id_token(id_token: &str) -> SocialResult<GoogleJwtProfile> {
.ok_or_else(|| Box::new(ResErr::social("校验Token失败,未找到正确的公钥")))?;
tracing::info!("public key : {:?}", key);
let n = key.get("n").unwrap().as_str().unwrap();
let e = &key.get("e").unwrap().as_str().unwrap();
// 验证Token
let mut validation: Validation = Validation::new(Algorithm::RS256);
@ -175,7 +155,7 @@ pub async fn verify_id_token(id_token: &str) -> SocialResult<GoogleJwtProfile> {
let decoded = decode::<Value>(
id_token,
&DecodingKey::from_rsa_components(n, e).unwrap(),
&DecodingKey::from_rsa_components(key.n.as_str(), key.e.as_str()).unwrap(),
&validation,
)?;