support sending requests from different IP for HTTP client

Signed-off-by: liulanzheng <[email protected]>

Author: liulanzheng

net/http/client.cpp

@@ -43,6 +43,7 @@ class PooledDialer {
     photon::mutex init_mtx;
     bool initialized = false;
     bool tls_ctx_ownership = false;
+    std::vector<IPAddr> src_ips;
 
     // If there is a photon thread switch during construction, the constructor might be called
     // multiple times, even for a thread_local instance. Therefore, ensure that there is no photon
@@ -52,7 +53,7 @@ class PooledDialer {
         photon::fini_hook({this, &PooledDialer::at_photon_fini});
     }
 
-    int init(TLSContext *_tls_ctx) {
+    int init(TLSContext *_tls_ctx, std::vector<IPAddr> &src_ips) {
         if (initialized)
             return 0;
         SCOPED_LOCK(init_mtx);
@@ -63,8 +64,8 @@ class PooledDialer {
             tls_ctx_ownership = true;
             tls_ctx = new_tls_context(nullptr, nullptr, nullptr);
         }
-        auto tcp_cli = new_tcp_socket_client();
-        auto tls_cli = new_tls_client(tls_ctx, new_tcp_socket_client(), true);
+        auto tcp_cli = new_tcp_socket_client(src_ips.data(), src_ips.size());
+        auto tls_cli = new_tls_client(tls_ctx, new_tcp_socket_client(src_ips.data(), src_ips.size()), true);
         tcpsock.reset(new_tcp_socket_pool(tcp_cli, -1, true));
         tlssock.reset(new_tcp_socket_pool(tls_cli, -1, true));
         udssock.reset(new_uds_client());
@@ -158,10 +159,9 @@ class ClientImpl : public Client {
         m_tls_ctx(tls_ctx),
         m_cookie_jar(cookie_jar) {
     }
-
     PooledDialer& get_dialer() {
         thread_local PooledDialer dialer;
-        dialer.init(m_tls_ctx);
+        dialer.init(m_tls_ctx, m_bind_ips);
         return dialer;
     }
 

net/http/client.h

@@ -25,7 +25,7 @@ limitations under the License.
 #include <photon/common/stream.h>
 #include <photon/common/timeout.h>
 #include <photon/net/socket.h>
-
+#include <vector>
 
 namespace photon {
 namespace net {
@@ -129,6 +129,9 @@ class Client : public Object {
     void set_user_agent(std::string_view user_agent) {
         m_user_agent = std::string(user_agent);
     }
+    void set_bind_ips(std::vector<IPAddr> &ips) {
+        m_bind_ips = ips;
+    }
     StoredURL* get_proxy() {
         return &m_proxy_url;
     }
@@ -146,12 +149,13 @@ class Client : public Object {
     void timeout_s(uint64_t tmo) { timeout(tmo * 1000UL * 1000UL); }
 
     virtual ISocketStream* native_connect(std::string_view host, uint16_t port,
-                                    bool secure = false, uint64_t timeout = -1UL) = 0;
+                                          bool secure = false, uint64_t timeout = -1UL) = 0;
 protected:
     StoredURL m_proxy_url;
     std::string m_user_agent;
     uint64_t m_timeout = -1UL;
     bool m_proxy = false;
+    std::vector<IPAddr> m_bind_ips;
 };
 
 //A Client without cookie_jar would ignore all response-header "Set-Cookies"

net/kernel_socket.cpp

@@ -196,6 +196,13 @@ class KernelSocketStream : public SocketStreamBase {
 
 class KernelSocketClient : public SocketClientBase {
 public:
+    std::vector<sockaddr_storage> bind_ss;
+
+    KernelSocketClient(IPAddr* bind_ip = nullptr, int bind_ip_n = 0) {
+        for (int i = 0; i < bind_ip_n; i++)
+            bind_ss.emplace_back(EndPoint(bind_ip[i], 0));
+    }
+
     ISocketStream* connect(const char* path, size_t count) override {
         struct sockaddr_un addr_un;
         if (fill_uds_path(addr_un, path, count) != 0) return nullptr;
@@ -206,6 +213,10 @@ class KernelSocketClient : public SocketClientBase {
     ISocketStream* connect(const EndPoint& remote, const EndPoint* local) override {
         sockaddr_storage r(remote);
         if (likely(!local || local->is_ipv4() != remote.is_ipv4())) {
+            if (bind_ss.size() > 0) {
+                int rid = rand() % bind_ss.size();
+                return do_connect(&r, &bind_ss[rid]);
+            }
             return do_connect(&r);
         }
         sockaddr_storage l(*local);
@@ -981,8 +992,8 @@ class ETKernelSocketServer : public KernelSocketServer, public NotifyContext {
 
 /* ET Socket - End */
 
-extern "C" ISocketClient* new_tcp_socket_client() {
-    return new KernelSocketClient();
+extern "C" ISocketClient* new_tcp_socket_client(IPAddr* bind_ip, uint32_t bind_ip_n) {
+    return new KernelSocketClient(bind_ip, bind_ip_n);
 }
 extern "C" ISocketServer* new_tcp_socket_server() {
     return NewObj<KernelSocketServer>()->init();

net/socket.h

@@ -278,7 +278,7 @@ namespace net {
         virtual void terminate() = 0;
     };
 
-    extern "C" ISocketClient* new_tcp_socket_client();
+    extern "C" ISocketClient* new_tcp_socket_client(IPAddr* bind_ip = nullptr, uint32_t bind_ip_n = 0);
     extern "C" ISocketServer* new_tcp_socket_server();
     extern "C" ISocketClient* new_uds_client();
     extern "C" ISocketServer* new_uds_server(bool autoremove = false);