CMake/Source/cmSiteNameCommand.cxx

134 lines
3.7 KiB
C++

/*=========================================================================
Program: CMake - Cross-Platform Makefile Generator
Module: $RCSfile$
Language: C++
Date: $Date$
Version: $Revision$
Copyright (c) 2002 Kitware, Inc., Insight Consortium. All rights reserved.
See Copyright.txt or http://www.cmake.org/HTML/Copyright.html for details.
This software is distributed WITHOUT ANY WARRANTY; without even
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE. See the above copyright notices for more information.
=========================================================================*/
#include "cmSiteNameCommand.h"
#include <cmsys/RegularExpression.hxx>
// cmSiteNameCommand
bool cmSiteNameCommand::InitialPass(std::vector<std::string> const& args)
{
if(args.size() != 1 )
{
this->SetError("called with incorrect number of arguments");
return false;
}
std::vector<std::string> paths;
paths.push_back("/usr/bsd");
paths.push_back("/usr/sbin");
paths.push_back("/usr/bin");
paths.push_back("/bin");
paths.push_back("/sbin");
paths.push_back("/usr/local/bin");
const char* cacheValue
= m_Makefile->GetDefinition(args[0].c_str());
if(cacheValue)
{
return true;
}
const char *temp = m_Makefile->GetDefinition("HOSTNAME");
std::string hostname_cmd;
if(temp)
{
hostname_cmd = temp;
}
else
{
hostname_cmd = cmSystemTools::FindProgram("hostname", paths);
}
std::string siteName = "unknown";
#if defined(_WIN32) && !defined(__CYGWIN__)
std::string host;
if(cmSystemTools::ReadRegistryValue(
"HKEY_LOCAL_MACHINE\\System\\CurrentControlSet\\Control\\ComputerName\\ComputerName;ComputerName",
host))
{
siteName = host;
}
#else
// try to find the hostname for this computer
if (!cmSystemTools::IsOff(hostname_cmd.c_str()))
{
std::string host;
cmSystemTools::RunSingleCommand(hostname_cmd.c_str(),
&host);
// got the hostname
if (host.length())
{
// remove any white space from the host name
std::string hostRegExp = "[ \t\n\r]*([^\t\n\r ]*)[ \t\n\r]*";
cmsys::RegularExpression hostReg (hostRegExp.c_str());
if (hostReg.find(host.c_str()))
{
// strip whitespace
host = hostReg.match(1);
}
if(host.length())
{
siteName = host;
temp = m_Makefile->GetDefinition("NSLOOKUP");
std::string nslookup_cmd;
if(temp)
{
nslookup_cmd = temp;
}
else
{
nslookup_cmd = cmSystemTools::FindProgram("nslookup", paths);
}
// try to find the domain name for this computer
if (!cmSystemTools::IsOff(nslookup_cmd.c_str()))
{
nslookup_cmd += " ";
nslookup_cmd += host;
std::string nsOutput;
cmSystemTools::RunSingleCommand(nslookup_cmd.c_str(),
&nsOutput);
// got the domain name
if (nsOutput.length())
{
std::string RegExp = ".*Name:[ \t\n]*";
RegExp += host;
RegExp += "\\.([^ \t\n\r]*)[ \t\n\r]*Address:";
cmsys::RegularExpression reg( RegExp.c_str() );
if(reg.find(nsOutput.c_str()))
{
siteName += '.' + cmSystemTools::LowerCase(reg.match(1));
}
}
}
}
}
}
#endif
m_Makefile->
AddCacheDefinition(args[0].c_str(),
siteName.c_str(),
"Name of the computer/site where compile is being run",
cmCacheManager::STRING);
return true;
}