From 462b27e7a492911fa19ea64b7f5c380071945ea0 Mon Sep 17 00:00:00 2001
From: Josh Blum <josh@joshknows.com>
Date: Wed, 14 Mar 2012 10:41:14 -0700
Subject: uhd: make atlbase options for msvc build

---
 host/lib/transport/udp_zero_copy.cpp | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

(limited to 'host/lib/transport/udp_zero_copy.cpp')

diff --git a/host/lib/transport/udp_zero_copy.cpp b/host/lib/transport/udp_zero_copy.cpp
index e43a96dd0..0ccc92b82 100644
--- a/host/lib/transport/udp_zero_copy.cpp
+++ b/host/lib/transport/udp_zero_copy.cpp
@@ -35,7 +35,8 @@ static const size_t DEFAULT_NUM_FRAMES = 32;
 /***********************************************************************
  * Check registry for correct fast-path setting (windows only)
  **********************************************************************/
-#ifdef UHD_PLATFORM_WIN32
+#ifdef HAVE_ATLBASE_H
+#define CHECK_REG_SEND_THRESH
 #include <atlbase.h> //CRegKey
 static void check_registry_for_fast_send_threshold(const size_t mtu){
     static bool warned = false;
@@ -56,7 +57,7 @@ static void check_registry_for_fast_send_threshold(const size_t mtu){
     }
     reg_key.Close();
 }
-#endif /*UHD_PLATFORM_WIN32*/
+#endif /*HAVE_ATLBASE_H*/
 
 /***********************************************************************
  * Reusable managed receiver buffer:
@@ -149,9 +150,9 @@ public:
     {
         UHD_LOG << boost::format("Creating udp transport for %s %s") % addr % port << std::endl;
 
-        #ifdef UHD_PLATFORM_WIN32
+        #ifdef CHECK_REG_SEND_THRESH
         check_registry_for_fast_send_threshold(this->get_send_frame_size());
-        #endif /*UHD_PLATFORM_WIN32*/
+        #endif /*CHECK_REG_SEND_THRESH*/
 
         //resolve the address
         asio::ip::udp::resolver resolver(_io_service);
-- 
cgit v1.2.3