From 015d93a71527b41c5de22ed6b0efa04636ed4462 Mon Sep 17 00:00:00 2001
From: James <jhc@apache.org>
Date: Wed, 30 Mar 2016 15:33:48 -0400
Subject: [PATCH] HDFS-10222. libhdfs++: Shutdown sockets to avoid 'Connection
 reset by peer'.  Contributed by James Clampffer

---
 .../lib/connection/datanodeconnection.cc      |  4 +--
 .../lib/connection/datanodeconnection.h       | 31 ++++++++++++++++++-
 2 files changed, 31 insertions(+), 4 deletions(-)

diff --git a/hadoop-hdfs-project/hadoop-hdfs-native-client/src/main/native/libhdfspp/lib/connection/datanodeconnection.cc b/hadoop-hdfs-project/hadoop-hdfs-native-client/src/main/native/libhdfspp/lib/connection/datanodeconnection.cc
index 19878ab228..be36fceae3 100644
--- a/hadoop-hdfs-project/hadoop-hdfs-native-client/src/main/native/libhdfspp/lib/connection/datanodeconnection.cc
+++ b/hadoop-hdfs-project/hadoop-hdfs-native-client/src/main/native/libhdfspp/lib/connection/datanodeconnection.cc
@@ -55,9 +55,7 @@ void DataNodeConnectionImpl::Connect(
 }
 
 void DataNodeConnectionImpl::Cancel() {
-  // best to do a shutdown() first for portability
-  conn_->shutdown(asio::ip::tcp::socket::shutdown_both);
-  conn_->close();
+  conn_.reset();
 }
 
 
diff --git a/hadoop-hdfs-project/hadoop-hdfs-native-client/src/main/native/libhdfspp/lib/connection/datanodeconnection.h b/hadoop-hdfs-project/hadoop-hdfs-native-client/src/main/native/libhdfspp/lib/connection/datanodeconnection.h
index 6cb7f4a8a1..96f26599b0 100644
--- a/hadoop-hdfs-project/hadoop-hdfs-native-client/src/main/native/libhdfspp/lib/connection/datanodeconnection.h
+++ b/hadoop-hdfs-project/hadoop-hdfs-native-client/src/main/native/libhdfspp/lib/connection/datanodeconnection.h
@@ -22,9 +22,12 @@
 #include "common/async_stream.h"
 #include "ClientNamenodeProtocol.pb.h"
 #include "common/libhdfs_events_impl.h"
+#include "common/logging.h"
 
 #include "asio.hpp"
 
+#include <exception>
+
 namespace hdfs {
 
 class DataNodeConnection : public AsyncStream {
@@ -38,9 +41,35 @@ public:
 };
 
 
+struct SocketDeleter {
+  inline void operator()(asio::ip::tcp::socket *sock) {
+    if(sock->is_open()) {
+      /**
+       *  Even though we just checked that the socket is open it's possible
+       *  it isn't in a state where it can properly send or receive.  If that's
+       *  the case asio will turn the underlying error codes from shutdown()
+       *  and close() into unhelpfully named std::exceptions.  Due to the
+       *  relatively innocuous nature of most of these error codes it's better
+       *  to just catch, give a warning, and move on with life.
+       **/
+      try {
+        sock->shutdown(asio::ip::tcp::socket::shutdown_both);
+      } catch (const std::exception &e) {
+        LOG_WARN(kBlockReader, << "Error calling socket->shutdown");
+      }
+      try {
+        sock->close();
+      } catch (const std::exception &e) {
+        LOG_WARN(kBlockReader, << "Error calling socket->close");
+      }
+    }
+    delete sock;
+  }
+};
+
 class DataNodeConnectionImpl : public DataNodeConnection, public std::enable_shared_from_this<DataNodeConnectionImpl>{
 public:
-  std::unique_ptr<asio::ip::tcp::socket> conn_;
+  std::unique_ptr<asio::ip::tcp::socket, SocketDeleter> conn_;
   std::array<asio::ip::tcp::endpoint, 1> endpoints_;
   std::string uuid_;
   LibhdfsEvents *event_handlers_;