iploc: Optimize IP address parsing.

Change-Id: Ieeac354aeee0ed10b8732b19183aa0320caac07a

1 files changed, 2 insertions, 2 deletions

diff --git a/iploc.py b/iploc.py
index b70c203..3b6085a 100644
--- a/iploc.py
+++ b/iploc.py
@@ -9,6 +9,7 @@ import struct
 import sys
 import os
 import optparse
+import socket
 
 from bsddb3 import db
 
@@ -49,10 +50,9 @@ temp_user = ""
 
 def get_reverse_dns(ip_address):
     # XXX: this works only with IPv4 addresses (because dnshistory works only with IPv4)
-    octets = str(ip_address).split('.')
     # The keys in the the reverse DNS db are stored as a char encoded string
     # made from the single octet of the IP address.
-    key = ''.join(chr(int(x)) for x in octets)
+    key = socket.inet_aton(ip_address)
     value = REVERSE_DNS_DB.get(key)
     struct_sz = struct.calcsize("li")
     assert len(value) > struct_sz