Comparison

net.stun: Factor out address unpack, an operation common to multiple attributes

comparison

equal deleted inserted replaced

-:f2ae9c6d1d9f
+:9889b1815d31
 			return attribute:sub(5);
 		end
 	end
 end
-function packet_methods:get_mapped_address()
+function packet_methods:_unpack_address(data, xor)
-	local data = self:get_attribute("mapped-address");
-	if not data then return; end
 	local family, port = struct.unpack("x>BI2", data);
 	local addr = data:sub(5);
+	if xor then
+		port = bit32.bxor(port, 0x2112);
+		addr = sxor(addr, magic_cookie..self.transaction_id);
+	end
 	return {
 		family = addr_families[family] or "unknown";
 		port = port;
 		address = net.ntop(addr);
 	};
 end
+function packet_methods:get_mapped_address()
+	local data = self:get_attribute("mapped-address");
+	if not data then return; end
+	return self:_unpack_address(data, false);
+end
 function packet_methods:get_xor_mapped_address()
 	local data = self:get_attribute("xor-mapped-address");
 	if not data then return; end
-	local family, port = struct.unpack("x>BI2", data);
+	return self:_unpack_address(data, true);
-	local addr = sxor(data:sub(5), magic_cookie..self.transaction_id);
-	return {
-		family = addr_families[family] or "unknown";
-		port = bit32.bxor(port, 0x2112);
-		address = net.ntop(addr);
-		address_raw = data:sub(5);
-	};
 end
 function packet_methods:add_message_integrity(key)
 	-- Add attribute with a dummy value so we can artificially increase
 	-- the packet 'length'