Browse Source

set zeroidc.running = false on token exchange error

Grant Limberg 2 years ago
parent
commit
9826c20d1a
1 changed files with 17 additions and 5 deletions
  1. 17 5
      zeroidc/src/lib.rs

+ 17 - 5
zeroidc/src/lib.rs

@@ -498,7 +498,8 @@ impl ZeroIDC {
                             let n = match i.nonce.clone() {
                             let n = match i.nonce.clone() {
                                 Some(n) => n,
                                 Some(n) => n,
                                 None => {
                                 None => {
-                                    println!("no noce");
+                                    println!("no nonce");
+                                    i.running = false;
                                     return None;
                                     return None;
                                 }
                                 }
                             };
                             };
@@ -507,6 +508,7 @@ impl ZeroIDC {
                                 Some(t) => t,
                                 Some(t) => t,
                                 None => {
                                 None => {
                                     println!("no id token");
                                     println!("no id token");
+                                    i.running = false;
                                     return None;
                                     return None;
                                 }
                                 }
                             };
                             };
@@ -515,6 +517,7 @@ impl ZeroIDC {
                                 Ok(c) => c,
                                 Ok(c) => c,
                                 Err(_e) => {
                                 Err(_e) => {
                                     println!("no claims");
                                     println!("no claims");
+                                    i.running = false;
                                     return None;
                                     return None;
                                 }
                                 }
                             };
                             };
@@ -523,6 +526,7 @@ impl ZeroIDC {
                                 Ok(s) => s,
                                 Ok(s) => s,
                                 Err(_) => {
                                 Err(_) => {
                                     println!("no signing algorithm");
                                     println!("no signing algorithm");
+                                    i.running = false;
                                     return None;
                                     return None;
                                 }
                                 }
                             };
                             };
@@ -535,12 +539,14 @@ impl ZeroIDC {
                                     Ok(h) => h,
                                     Ok(h) => h,
                                     Err(e) => {
                                     Err(e) => {
                                         println!("Error hashing access token: {}", e);
                                         println!("Error hashing access token: {}", e);
+                                        i.running = false;
                                         return None;
                                         return None;
                                     }
                                     }
                                 };
                                 };
 
 
                                 if actual_hash != *expected_hash {
                                 if actual_hash != *expected_hash {
                                     println!("token hash error");
                                     println!("token hash error");
+                                    i.running = false;
                                     return None;
                                     return None;
                                 }
                                 }
                             }
                             }
@@ -549,7 +555,7 @@ impl ZeroIDC {
                         Err(e) => {
                         Err(e) => {
                             println!("token response error: {:?}", e.to_string());
                             println!("token response error: {:?}", e.to_string());
                             println!("\t {:?}", e.source());
                             println!("\t {:?}", e.source());
-
+                            i.running = false;
                             None
                             None
                         }
                         }
                     }
                     }
@@ -634,10 +640,12 @@ impl ZeroIDC {
 
 
                                     Ok(bytes)
                                     Ok(bytes)
                                 } else if res.status() == 402 {
                                 } else if res.status() == 402 {
-                                        Err(SSOExchangeError::new(
-                                            "additional license seats required. Please contact your network administrator.".to_string(),
-                                        ))
+                                    i.running = false;
+                                    Err(SSOExchangeError::new(
+                                        "additional license seats required. Please contact your network administrator.".to_string(),
+                                    ))
                                 } else {
                                 } else {
+                                    i.running = false;
                                     Err(SSOExchangeError::new(
                                     Err(SSOExchangeError::new(
                                         "error from central endpoint".to_string(),
                                         "error from central endpoint".to_string(),
                                     ))
                                     ))
@@ -649,20 +657,24 @@ impl ZeroIDC {
                                 println!("Status: {}", res.status().unwrap());
                                 println!("Status: {}", res.status().unwrap());
                                 println!("Post error: {}", res);
                                 println!("Post error: {}", res);
                                 i.exp_time = 0;
                                 i.exp_time = 0;
+                                i.running = false;
                                 Err(SSOExchangeError::new(
                                 Err(SSOExchangeError::new(
                                     "error from central endpoint".to_string(),
                                     "error from central endpoint".to_string(),
                                 ))
                                 ))
                             }
                             }
                         }
                         }
                     } else {
                     } else {
+                        i.running = false;
                         Err(SSOExchangeError::new(
                         Err(SSOExchangeError::new(
                             "error splitting state token".to_string(),
                             "error splitting state token".to_string(),
                         ))
                         ))
                     }
                     }
                 } else {
                 } else {
+                    i.running = false;
                     Err(SSOExchangeError::new("invalid token response".to_string()))
                     Err(SSOExchangeError::new("invalid token response".to_string()))
                 }
                 }
             } else {
             } else {
+                i.running = false;
                 Err(SSOExchangeError::new("invalid pkce verifier".to_string()))
                 Err(SSOExchangeError::new("invalid pkce verifier".to_string()))
             }
             }
         });
         });