Fixed bug when testing against ip-adresses

finckelt · Jul 10, 2016 · a97d754 · a97d754
1 parent 3d10cb7
commit a97d754
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/Netling.Core/Performance/HttpWorker.cs b/Netling.Core/Performance/HttpWorker.cs
@@ -28,8 +28,18 @@ public HttpWorker(string url)
             _read = 0;
             _responseType = ResponseType.Unknown;
             _uri = new Uri(url, UriKind.Absolute);
-            var host = Dns.GetHostEntry(_uri.Host);
-            var ip = host.AddressList.First(i => i.AddressFamily == AddressFamily.InterNetwork);
+            IPAddress ip;
+
+            if (_uri.HostNameType == UriHostNameType.Dns)
+            {
+                var host = Dns.GetHostEntry(_uri.Host);
+                ip = host.AddressList.First(i => i.AddressFamily == AddressFamily.InterNetwork);
+            }
+            else
+            {
+                ip = IPAddress.Parse(_uri.Host);
+            }
+
             _endPoint = new IPEndPoint(ip, _uri.Port);
             _request = Encoding.UTF8.GetBytes($"GET {_uri.PathAndQuery} HTTP/1.1\r\nAccept-Encoding: gzip, deflate, sdch\r\nHost: {_uri.Host}\r\nContent-Length: 0\r\n\r\n");
         }