diff --git a/CMakeLists.txt b/CMakeLists.txt
index 5dc3be3d..cfcd7c93 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -33,6 +33,9 @@
 # USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY
 # OF SUCH DAMAGE.
 
+set(CMAKE_MODULE_PATH "${CMAKE_CURRENT_SOURCE_DIR}/cmake;${CMAKE_MODULE_PATH}")
+include(CheckFunctionExistsMayNeedLibrary)
+
 cmake_minimum_required(VERSION 3.1)
 
 set(CMAKE_MODULE_PATH ${CMAKE_CURRENT_SOURCE_DIR}/cmake)
@@ -79,8 +82,18 @@ install(
 include(max_warnings)
 include(FeatureSummary)
 
+# Add socket libraries
 if(WIN32)
   list(APPEND LIBRARIES ws2_32)
+else()
+  check_function_exists_may_need_library(socket HAVE_SOCKET socket)
+  if(NEED_LIB_SOCKET)
+    list(APPEND LIBRARIES socket)
+  endif()
+  check_function_exists_may_need_library(inet_addr HAVE_INET_ADDR nsl)
+  if(NEED_LIB_NSL)
+    list(APPEND LIBRARIES nsl)
+  endif()
 endif()
 
 add_subdirectory(src)
diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
index a9ae1191..a0322ef5 100644
--- a/src/CMakeLists.txt
+++ b/src/CMakeLists.txt
@@ -35,7 +35,6 @@
 
 include(CheckFunctionExists)
 include(CheckSymbolExists)
-include(CheckFunctionExistsMayNeedLibrary)
 include(CheckIncludeFiles)
 include(CheckTypeSize)
 include(CheckSymbolExists)