IP address from as NSString from NSData.

gabriel · Jan 24, 2015 · a239f4f · a239f4f
1 parent c333ccf
commit a239f4f
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/Classes/GHNSData+Utils.h b/Classes/GHNSData+Utils.h
@@ -15,4 +15,9 @@
  */
 - (NSString *)gh_hexString;
 
+/*!
+ Get IP address string from data object.
+ */
+- (NSString *)gh_IPAddressAsString;
+
 @end
diff --git a/Classes/GHNSData+Utils.m b/Classes/GHNSData+Utils.m
@@ -7,6 +7,7 @@
 //
 
 #import "GHNSData+Utils.h"
+#import <arpa/inet.h>
 
 @implementation NSData (GHUtils)
 
@@ -19,4 +20,27 @@ - (NSString *)gh_hexString {
   return hex;
 }
 
+- (NSString *)gh_IPAddressAsString {
+  struct sockaddr *addr = (struct sockaddr*)[self bytes];
+  if (addr->sa_family == AF_INET) {
+    struct sockaddr_in *addr4 = (struct sockaddr_in*) addr;
+    char addr4CString[INET_ADDRSTRLEN];
+    if (!inet_ntop(AF_INET, &addr4->sin_addr, addr4CString, INET_ADDRSTRLEN)) {
+      return nil;
+    } else {
+      return [[NSString alloc] initWithUTF8String:addr4CString];
+    }
+  } else if (addr->sa_family == AF_INET6) {
+    struct sockaddr_in6* addr6 = (struct sockaddr_in6*) addr;
+    char addr6CString[INET6_ADDRSTRLEN];
+    if (!inet_ntop(AF_INET6, &addr6->sin6_addr, addr6CString, INET6_ADDRSTRLEN)) {
+      return nil;
+    } else {
+      return [[NSString alloc] initWithUTF8String:addr6CString];
+    }
+  } else {
+    return nil;
+  }
+}
+
 @end