commit a4d2aad3315e7f72dc92ec9d0e0ba16a6507b181
parent 3a1321a5a9e866d18e38c403c6475c1f31abf6bd
Author: Daniel García <dani1861994@hotmail.com>
Date: Tue, 24 Apr 2018 20:53:12 +0200
Removed unused proxy functionality
Diffstat:
2 files changed, 1 insertion(+), 168 deletions(-)
diff --git a/README.md b/README.md
@@ -34,7 +34,7 @@ on other systems use their respective package managers.
Then run:
```
-cargo run --bin bitwarden_rs
+cargo run
# or
cargo build
```
diff --git a/src/bin/proxy.rs b/src/bin/proxy.rs
@@ -1,167 +0,0 @@
-#![feature(plugin)]
-
-#![plugin(rocket_codegen)]
-extern crate rocket;
-extern crate rocket_contrib;
-extern crate reqwest;
-
-use std::io::{self, Cursor};
-use std::str::FromStr;
-use std::path::{Path, PathBuf};
-
-use rocket::{Request, Response};
-use rocket::config::Config;
-use rocket::fairing::{Fairing, Info, Kind};
-use rocket::http;
-use rocket::response::NamedFile;
-
-use reqwest::header::{self, Headers};
-
-/**
-** These routes are here to avoid showing errors in the console,
-** redirect the body data to the fairing and show the web vault.
-**/
-#[get("/")]
-fn index() -> io::Result<NamedFile> {
- NamedFile::open(Path::new("web-vault").join("index.html"))
-}
-
-#[get("/<p..>")] // Only match this if the other routes don't match
-fn get(p: PathBuf) -> io::Result<NamedFile> {
- NamedFile::open(Path::new("web-vault").join(p))
-}
-
-#[delete("/<_p..>")]
-fn delete(_p: PathBuf) {}
-
-#[put("/<_p..>", data = "<d>")]
-fn put(_p: PathBuf, d: Vec<u8>) -> Vec<u8> { d }
-
-#[post("/<_p..>", data = "<d>")]
-fn post(_p: PathBuf, d: Vec<u8>) -> Vec<u8> { d }
-
-
-fn main() {
- let config = Config::development().unwrap();
-
- rocket::custom(config, false)
- .mount("/", routes![get, put, post, delete, index])
- .attach(ProxyFairing { client: reqwest::Client::new() })
- .launch();
-}
-
-struct ProxyFairing {
- client: reqwest::Client
-}
-
-impl Fairing for ProxyFairing {
- fn info(&self) -> Info {
- Info {
- name: "Proxy Fairing",
- kind: Kind::Launch | Kind::Response,
- }
- }
-
- fn on_launch(&self, _rocket: &rocket::Rocket) {
- println!("Started proxy on locahost:8000");
- }
-
- fn on_response(&self, req: &Request, res: &mut Response) {
- // Prepare the data to make the request
- // -------------------------------------
-
- let url = {
- let url = req.uri().as_str();
-
- // Check if we are outside the API paths
- if !url.starts_with("/api/")
- && !url.starts_with("/identity/") {
- return;
- }
-
- // Replace the path with the real server URL
- url.replacen("/api/", "https://api.bitwarden.com/", 1)
- .replacen("/identity/", "https://identity.bitwarden.com/", 1)
- };
-
- let host = url.split("/").collect::<Vec<_>>()[2];
- let headers = headers_rocket_to_reqwest(req.headers(), host);
- let method = reqwest::Method::from_str(req.method().as_str()).unwrap();
- let body = res.body_bytes();
-
- println!("\n\nREQ. {} {}", req.method().as_str(), url);
- println!("HEADERS. {:#?}", headers);
- if let Some(ref body) = body {
- let body_string = String::from_utf8_lossy(body);
- if !body_string.contains("<!DOCTYPE html>") {
- println!("BODY. {:?}", body_string);
- }
- }
-
-
- // Execute the request
- // -------------------------------------
- let mut client = self.client.request(method, &url);
- let request_builder = client.headers(headers);
-
- if let Some(body_vec) = body {
- request_builder.body(body_vec);
- }
-
- let mut server_res = match request_builder.send() {
- Ok(response) => response,
- Err(e) => {
- res.set_status(http::Status::BadRequest);
- res.set_sized_body(Cursor::new(e.to_string()));
- return;
- }
- };
-
- // Get the response values
- // -------------------------------------
- let mut res_body: Vec<u8> = vec![];
- server_res.copy_to(&mut res_body).unwrap();
-
- let res_status = server_res.status().as_u16();
- let mut res_headers = server_res.headers().clone();
-
- // These headers break stuff
- res_headers.remove::<header::TransferEncoding>();
- res_headers.remove::<header::ContentLength>();
-
- println!("\n\nRES. {} {}", res_status, url);
- // Nothing interesting here
- // println!("HEADERS. {:#?}", res_headers);
- println!("BODY. {:?}", String::from_utf8_lossy(&res_body));
-
- // Prepare the response
- // -------------------------------------
- res.set_status(http::Status::from_code(res_status).unwrap_or(http::Status::BadRequest));
-
- headers_reqwest_to_rocket(&res_headers, res);
- res.set_sized_body(Cursor::new(res_body));
- }
-}
-
-fn headers_rocket_to_reqwest(headers: &http::HeaderMap, host: &str) -> Headers {
- let mut new_headers = Headers::new();
-
- for header in headers.iter() {
- let name = header.name().to_string();
-
- let value = if name.to_lowercase() != "host" {
- header.value().to_string()
- } else {
- host.to_string()
- };
-
- new_headers.set_raw(name, value);
- }
- new_headers
-}
-
-fn headers_reqwest_to_rocket(headers: &Headers, res: &mut Response) {
- for header in headers.iter() {
- res.set_raw_header(header.name().to_string(), header.value_string());
- }
-}
