1 files changed, 25 insertions, 5 deletions
diff --git a/SwifTools/Cocoa/CocoaUtil.h b/SwifTools/Cocoa/CocoaUtil.h
index 49ed682..83d95b6 100644
--- a/SwifTools/Cocoa/CocoaUtil.h
+++ b/SwifTools/Cocoa/CocoaUtil.h
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2011 Isode Limited.
+ * Copyright (c) 2011-2016 Isode Limited.
  * All rights reserved.
  * See the COPYING file for more information.
  */
@@ -7,16 +7,36 @@
 #pragma once
 
 // Conversion utilities
-#define NS2STDSTRING(a) (a == nil ? std::string() : std::string([a cStringUsingEncoding:NSUTF8StringEncoding]))
-#define STD2NSSTRING(a) [NSString stringWithCString:a.c_str() encoding:NSUTF8StringEncoding]
+namespace {
+
+inline std::string ns2StdString(NSString* _Nullable nsString);
+inline std::string ns2StdString(NSString* _Nullable nsString) {
+	std::string stdString;
+	if (nsString != nil) {
+		stdString = std::string([nsString cStringUsingEncoding:NSUTF8StringEncoding]);
+	}
+	return stdString;
+}
+
+inline NSString* _Nonnull std2NSString(const std::string& stdString);
+inline NSString* _Nonnull std2NSString(const std::string& stdString) {
+	NSString* _Nullable nsString = [NSString stringWithUTF8String:stdString.c_str()];
+	if (nsString == nil) {
+		nsString = @"";
+	}
+	// At this point nsString is guaranteed to be not null/nil.
+	return static_cast<NSString* _Nonnull>(nsString);
+}
+
+}
 
 // Intrusive pointer for NSObjects
 namespace boost {	
-	inline void intrusive_ptr_add_ref(NSObject* object) {
+	inline void intrusive_ptr_add_ref(NSObject* _Nonnull object) {
 		[object retain];
 	}
 	
-	inline void intrusive_ptr_release(NSObject* object) {
+	inline void intrusive_ptr_release(NSObject* _Nonnull object) {
 		[object release];
 	}
 }