Integrate curl-impersonate to bypass Cloudflare JS challenge

Author: cjiung123

src/cache/mod.rs

@@ -352,13 +352,22 @@ impl Cache {
  let (json, [url, refer]) = self.pre_run_code(run.clone(), rfid, test_case).await?;
  trace!("Pre-run code result {:#?}, {}, {}", json, url, refer);
 
- let text = self
+ let response = self
  .0
  .clone()
  .run_code(json.clone(), url.clone(), refer.clone())
- .await?
- .text()
  .await?;
+ let response_status = response.status();
+
+ let mut text = response.text().await?;
+ if !response_status.is_success() {
+ println!("Response is not success! Error Code {:?}", response_status);
+ println!("If error code is 403, it's likely caught by Cloudflare, trying to bypass it");
+ let output = self
+ .0
+ .clone().execute_curl_impersonate(json.clone(), url.clone(), refer.clone())?;
+ text = String::from_utf8_lossy(&output.stdout).to_string();
+ }
 
  let run_res: RunCode = serde_json::from_str(&text).map_err(|e| {
  anyhow!("JSON error: {e}, please double check your session and csrf config.")

src/plugins/leetcode.rs

@@ -8,6 +8,7 @@ use reqwest::{
  Client, ClientBuilder, Response,
 };
 use std::{collections::HashMap, str::FromStr, time::Duration};
+use std::process::{Command, Output};
 
 /// LeetCode API set
 #[derive(Clone)]
@@ -43,7 +44,6 @@ impl LeetCode {
  vec![
  ("Cookie", &cookie),
  ("x-csrftoken", &csrf),
- ("x-requested-with", "XMLHttpRequest"),
  ("Origin", &conf.sys.urls.base),
  ],
  )?;
@@ -232,6 +232,40 @@ impl LeetCode {
  .send(&self.client)
  .await
  }
+ 
+ pub fn execute_curl_impersonate(self, j: Json, url: String, refer: String) -> Result<Output> {
+ let header_csrf_arg = match self.default_headers.get("x-csrftoken") {
+ Some(value) => format!("x-csrftoken: {}", value.to_str().unwrap()),
+ None => String::new()
+ };
+
+ let header_referer_arg = format!("Referer: {}", refer);
+
+ let cookie_arg = match self.default_headers.get("Cookie") {
+ Some(value) => format!("{}", value.to_str().unwrap()),
+ None => String::new()
+ }; 
+
+ let data_arg = match serde_json::to_string(&j) {
+ Ok(value) => format!("{}", value),
+ Err(_) => String::new()
+ };
+
+ let output = Command::new("curl_ff117")
+ .arg("--header")
+ .arg(&header_csrf_arg)
+ .arg("--header")
+ .arg(&header_referer_arg)
+ .arg("--cookie")
+ .arg(&cookie_arg)
+ .arg("--data")
+ .arg(&data_arg)
+ .arg(&url)
+ .output()
+ .expect("Failed to execute process");
+ trace!("{:?}", output);
+ Ok(output)
+ }
 
  /// Get the result of submission / testing
  pub async fn verify_result(self, id: String) -> Result<Response> {