- Add get_sock_opt/set_sock_opt to Controller

src/brpc/controller.cpp

@@ -1382,6 +1382,26 @@ x509_st* Controller::get_peer_certificate() const {
     return s ? s->GetPeerCertificate() : NULL;
 }
 
+int Controller::get_sock_opt(int level, int optname, void* optval, socklen_t* optlen) {
+    Socket* s = _current_call.sending_sock.get();
+    if (s) {
+        return getsockopt(s->fd(), level, optname, optval, optlen);
+    } else {
+        LOG(WARNING) << "sock is null";
+        return EINVAL;
+    }
+}
+
+int Controller::set_sock_opt(int level, int optname, void* optval, socklen_t optlen) {
+    Socket* s = _current_call.sending_sock.get();
+    if (s) {
+        return setsockopt(s->fd(), level, optname, optval, optlen);
+    } else {
+        LOG(WARNING) << "sock is null";
+        return EINVAL;
+    }
+}
+
 #if defined(OS_MACOSX)
 typedef sig_t SignalHandler;
 #else

src/brpc/controller.h

@@ -592,6 +592,9 @@ private:
     void set_used_by_rpc() { add_flag(FLAGS_USED_BY_RPC); }
     bool is_used_by_rpc() const { return has_flag(FLAGS_USED_BY_RPC); }
 
+    // Get/set sock option. .e.g get vip info through ttm kernel module hook,
+    int get_sock_opt(int level, int optname, void* optval, socklen_t* optlen);
+    int set_sock_opt(int level, int optname, void* optval, socklen_t optlen);
 private:
     // NOTE: align and group fields to make Controller as compact as possible.