forked from mirrors/gecko-dev
		
	
		
			
				
	
	
		
			94 lines
		
	
	
	
		
			2.5 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			94 lines
		
	
	
	
		
			2.5 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
 | |
| /* vim: set ts=8 sts=2 et sw=2 tw=80: */
 | |
| /* This Source Code Form is subject to the terms of the Mozilla Public
 | |
|  * License, v. 2.0. If a copy of the MPL was not distributed with this
 | |
|  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
 | |
| 
 | |
| #if defined(XP_MACOSX) || defined(XP_LINUX)
 | |
| #  include <unistd.h>
 | |
| #elif defined(XP_WIN)
 | |
| #  include <winsock2.h>
 | |
| #endif
 | |
| 
 | |
| #include "nsNetworkInfoService.h"
 | |
| 
 | |
| #if defined(XP_MACOSX) || defined(XP_WIN) || defined(XP_LINUX)
 | |
| #  include "NetworkInfoServiceImpl.h"
 | |
| #else
 | |
| #  error "Unsupported platform for nsNetworkInfoService!  Check moz.build"
 | |
| #endif
 | |
| 
 | |
| namespace mozilla::net {
 | |
| 
 | |
| NS_IMPL_ISUPPORTS(nsNetworkInfoService, nsINetworkInfoService)
 | |
| 
 | |
| nsNetworkInfoService::nsNetworkInfoService() = default;
 | |
| 
 | |
| nsresult nsNetworkInfoService::Init() { return NS_OK; }
 | |
| 
 | |
| nsresult nsNetworkInfoService::ListNetworkAddresses(
 | |
|     nsIListNetworkAddressesListener* aListener) {
 | |
|   nsresult rv;
 | |
| 
 | |
|   AddrMapType addrMap;
 | |
|   rv = DoListAddresses(addrMap);
 | |
|   if (NS_WARN_IF(NS_FAILED(rv))) {
 | |
|     aListener->OnListNetworkAddressesFailed();
 | |
|     return NS_OK;
 | |
|   }
 | |
| 
 | |
|   uint32_t addrCount = addrMap.Count();
 | |
|   nsTArray<nsCString> addrStrings;
 | |
|   if (!addrStrings.SetCapacity(addrCount, fallible)) {
 | |
|     aListener->OnListNetworkAddressesFailed();
 | |
|     return NS_OK;
 | |
|   }
 | |
| 
 | |
|   for (const auto& data : addrMap.Values()) {
 | |
|     addrStrings.AppendElement(data);
 | |
|   }
 | |
|   aListener->OnListedNetworkAddresses(addrStrings);
 | |
|   return NS_OK;
 | |
| }
 | |
| 
 | |
| // TODO: Bug 1275373: https://bugzilla.mozilla.org/show_bug.cgi?id=1275373
 | |
| // Use platform-specific implementation of DoGetHostname on Cocoa and Windows.
 | |
| static nsresult DoGetHostname(nsACString& aHostname) {
 | |
|   char hostnameBuf[256];
 | |
|   int result = gethostname(hostnameBuf, 256);
 | |
|   if (result == -1) {
 | |
|     return NS_ERROR_FAILURE;
 | |
|   }
 | |
| 
 | |
|   // Ensure that there is always a terminating NUL byte.
 | |
|   hostnameBuf[255] = '\0';
 | |
| 
 | |
|   // Find the first '.', terminate string there.
 | |
|   char* dotLocation = strchr(hostnameBuf, '.');
 | |
|   if (dotLocation) {
 | |
|     *dotLocation = '\0';
 | |
|   }
 | |
| 
 | |
|   if (strlen(hostnameBuf) == 0) {
 | |
|     return NS_ERROR_FAILURE;
 | |
|   }
 | |
| 
 | |
|   aHostname.AssignASCII(hostnameBuf);
 | |
|   return NS_OK;
 | |
| }
 | |
| 
 | |
| nsresult nsNetworkInfoService::GetHostname(nsIGetHostnameListener* aListener) {
 | |
|   nsresult rv;
 | |
|   nsCString hostnameStr;
 | |
|   rv = DoGetHostname(hostnameStr);
 | |
|   if (NS_FAILED(rv)) {
 | |
|     aListener->OnGetHostnameFailed();
 | |
|     return NS_OK;
 | |
|   }
 | |
| 
 | |
|   aListener->OnGotHostname(hostnameStr);
 | |
| 
 | |
|   return NS_OK;
 | |
| }
 | |
| 
 | |
| }  // namespace mozilla::net
 | 
