xref: /petsc/src/sys/utils/fhost.c (revision a8031340ffb3f67c209e3ccbf1e437ef3e683964)
1 #define PETSC_DESIRE_FEATURE_TEST_MACROS /* for getdomainname() */
2 /*
3       Code for manipulating files.
4 */
5 #include <petscsys.h>
6 #if defined(PETSC_HAVE_SYS_UTSNAME_H)
7   #include <sys/utsname.h>
8 #endif
9 #if defined(PETSC_HAVE_WINDOWS_H)
10   #include <windows.h>
11 #endif
12 #if defined(PETSC_HAVE_SYS_SYSTEMINFO_H)
13   #include <sys/systeminfo.h>
14 #endif
15 #if defined(PETSC_HAVE_UNISTD_H)
16   #include <unistd.h>
17 #endif
18 #if defined(PETSC_HAVE_NETDB_H)
19   #include <netdb.h>
20 #endif
21 
22 /*@C
23     PetscGetHostName - Returns the name of the host. This attempts to
24     return the entire Internet name. It may not return the same name
25     as `MPI_Get_processor_name()`.
26 
27     Not Collective
28 
29     Input Parameter:
30 .   nlen - length of name
31 
32     Output Parameter:
33 .   name - contains host name.  Must be long enough to hold the name
34            This is the fully qualified name, including the domain.
35 
36     Level: developer
37 
38    Fortran Note:
39    In Fortran this routine has the format
40 
41 $       character*(64) name
42 $       call PetscGetHostName(name,ierr)
43 
44 .seealso: `PetscGetUserName()`, `PetscGetArchType()`
45 @*/
46 PetscErrorCode PetscGetHostName(char name[], size_t nlen)
47 {
48   char *domain = NULL;
49 #if defined(PETSC_HAVE_UNAME) && !defined(PETSC_HAVE_GETCOMPUTERNAME)
50   struct utsname utname;
51 #endif
52 
53   PetscFunctionBegin;
54 #if defined(PETSC_HAVE_GETCOMPUTERNAME)
55   {
56     size_t nnlen = nlen;
57     GetComputerName((LPTSTR)name, (LPDWORD)(&nnlen));
58   }
59 #elif defined(PETSC_HAVE_UNAME)
60   uname(&utname);
61   PetscCall(PetscStrncpy(name, utname.nodename, nlen));
62 #elif defined(PETSC_HAVE_GETHOSTNAME)
63   gethostname(name, nlen);
64 #endif
65   /* if there was not enough room then system call will not null terminate name */
66   name[nlen - 1] = 0;
67 
68   /* See if this name includes the domain */
69   PetscCall(PetscStrchr(name, '.', &domain));
70   if (!domain) {
71     size_t l, ll;
72     PetscCall(PetscStrlen(name, &l));
73     if (l == nlen - 1) PetscFunctionReturn(PETSC_SUCCESS);
74     name[l++] = '.';
75     name[l]   = 0;
76 #if defined(PETSC_HAVE_GETDOMAINNAME)
77     PetscCheck(!getdomainname(name + l, nlen - l), PETSC_COMM_SELF, PETSC_ERR_SYS, "getdomainname()");
78 #endif
79     /* check if domain name is not a dnsdomainname and nuke it */
80     PetscCall(PetscStrlen(name, &ll));
81     if (ll > 4) {
82       const char *suffixes[] = {".edu", ".com", ".net", ".org", ".mil", NULL};
83       PetscInt    index;
84       PetscCall(PetscStrendswithwhich(name, suffixes, &index));
85       if (!suffixes[index]) {
86         PetscCall(PetscInfo(NULL, "Rejecting domainname, likely is NIS %s\n", name));
87         name[l - 1] = 0;
88       }
89     }
90   }
91   PetscFunctionReturn(PETSC_SUCCESS);
92 }
93