Allow customizations for the socket configurations

[lqiu96]

Allow users to modify the the TCP options/ flags: https://docs.oracle.com/javase/8/docs/api/java/net/SocketOptions.html. Specifically, allow setting the SO_KEEPALIVE value to use OS keep-alive values.

See b/325293479 for more information.