cmSiteNameCommand.cxx 3.64 KB
Newer Older
1 2
/*=========================================================================

3
  Program:   CMake - Cross-Platform Makefile Generator
4 5 6 7 8
  Module:    $RCSfile$
  Language:  C++
  Date:      $Date$
  Version:   $Revision$

9 10
  Copyright (c) 2002 Kitware, Inc., Insight Consortium.  All rights reserved.
  See Copyright.txt or http://www.cmake.org/HTML/Copyright.html for details.
11

Will Schroeder's avatar
Will Schroeder committed
12 13 14
     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.
15 16 17 18 19

=========================================================================*/
#include "cmSiteNameCommand.h"

// cmSiteNameCommand
20
bool cmSiteNameCommand::InitialPass(std::vector<std::string> const& args)
21
{
22
  if(args.size() != 1 )
23 24 25 26
    {
    this->SetError("called with incorrect number of arguments");
    return false;
    }
27 28 29 30 31 32 33 34
  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");
  
35
  const char* cacheValue
36
    = m_Makefile->GetDefinition(args[0].c_str());
37 38 39 40 41
  if(cacheValue)
    {
    return true;
    }
  
42 43 44
  const char *temp = m_Makefile->GetDefinition("HOSTNAME");
  std::string hostname_cmd;
  if(temp)
45
    {
46
    hostname_cmd = temp;
47
    }
48
  else 
49
    {
50
    hostname_cmd = cmSystemTools::FindProgram("hostname", paths);
51
    }
52 53
  
  std::string siteName = "unknown";
Bill Hoffman's avatar
Bill Hoffman committed
54 55 56 57 58 59 60 61 62
#if defined(_WIN32) && !defined(__CYGWIN__)
  std::string host;
  if(cmSystemTools::ReadRegistryValue(
    "HKEY_LOCAL_MACHINE\\System\\CurrentControlSet\\Control\\ComputerName\\ComputerName;ComputerName",
    host))
    {
    siteName = host;
    }
#else
63
  // try to find the hostname for this computer
Bill Hoffman's avatar
Bill Hoffman committed
64
  if (!cmSystemTools::IsOff(hostname_cmd.c_str()))
65
    {
66 67 68 69 70 71
    std::string host;
    cmSystemTools::RunCommand(hostname_cmd.c_str(),
                              host);
    
    // got the hostname
    if (host.length())
72
      {
73 74 75 76 77 78 79 80
      // remove any white space from the host name
      std::string hostRegExp = "[ \t\n\r]*([^\t\n\r ]*)[ \t\n\r]*";
      cmRegularExpression hostReg (hostRegExp.c_str());
      if (hostReg.find(host.c_str()))
        {
        // strip whitespace
        host = hostReg.match(1);
        }
81

82 83 84 85 86 87 88 89 90 91 92 93
      if(host.length())
        {
        siteName = host;

        temp = m_Makefile->GetDefinition("NSLOOKUP");
        std::string nslookup_cmd;
        if(temp)
          {
          nslookup_cmd = temp;
          }
        else
          {
94
          nslookup_cmd = cmSystemTools::FindProgram("nslookup", paths);
95 96 97
          }

        // try to find the domain name for this computer
Bill Hoffman's avatar
Bill Hoffman committed
98
        if (!cmSystemTools::IsOff(nslookup_cmd.c_str()))
99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119
          {
          nslookup_cmd += " ";
          nslookup_cmd += host;
          std::string nsOutput;
          cmSystemTools::RunCommand(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:";
            cmRegularExpression reg( RegExp.c_str() );
            if(reg.find(nsOutput.c_str()))
              {
              siteName += '.' + cmSystemTools::LowerCase(reg.match(1));
              }
            }
          }
        }
120 121
      }
    }
Bill Hoffman's avatar
Bill Hoffman committed
122
#endif
123
  m_Makefile->
124
    AddCacheDefinition(args[0].c_str(),
125 126 127
                       siteName.c_str(),
                       "Name of the computer/site where compile is being run",
                       cmCacheManager::STRING);
128

129 130 131
  return true;
}