瀏覽代碼

fix await on servers, now both are listening

master
Jonathan Cobb 4 年之前
父節點
當前提交
37c92bf120
共有 2 個檔案被更改,包括 7 行新增4 行删除
  1. +4
    -3
      src/main.rs
  2. +3
    -1
      src/proxy.rs

+ 4
- 3
src/main.rs 查看文件

@@ -8,6 +8,8 @@ use std::process::exit;

use clap::{Arg, ArgMatches, App};

use futures_util::future::join;

use pnet::datalink;

use whoami;
@@ -117,8 +119,7 @@ async fn main() {
let proxy_port = args.value_of("proxy_port").unwrap().parse::<u16>().unwrap();
let proxy_ip = proxy_bind_addr.unwrap().ip();

let proxy = start_proxy(dns1_ip, dns2_ip, proxy_ip, proxy_port);
let admin = start_admin(admin_port, proxy_ip.to_string(), proxy_port, password_hash);
proxy.await;
admin.await;
let proxy = start_proxy(dns1_ip, dns2_ip, proxy_ip, proxy_port);
join(admin, proxy).await;
}

+ 3
- 1
src/proxy.rs 查看文件

@@ -66,8 +66,10 @@ pub async fn start_proxy (dns1_ip : &str,
}
});

Server::bind(&addr).serve(make_service);
let server = Server::bind(&addr).serve(make_service);
eprintln!("Proxy listening on {}", addr);
let result = server.await;
eprintln!("Proxy await result: {:?}", result);
}

async fn proxy(client: Client<HttpsConnector<HttpConnector<CacheResolver>>>,


Loading…
取消
儲存